source: zwrite.c @ b9df757

release-1.4release-1.5release-1.6release-1.7release-1.8release-1.9
Last change on this file since b9df757 was 24ccc01, checked in by Nelson Elhage <nelhage@mit.edu>, 12 years ago
Replace owl_message_create_from_zwriteline with owl_message_create_from_zwrite. By passing in a zwrite structure instead of a line, we make it easier to construct faked messages without having to go through generaring and parsing a zwrite command line. We change owl_zephyr_handle_ack to fake a zwrite structure, fixing the segfault on zephyring users with quotes (fixes #94), as well as logging errors sending instanced personals slightly better.
  • Property mode set to 100644
File size: 9.1 KB
Line 
1#include <string.h>
2#include <pwd.h>
3#include <sys/types.h>
4#include <unistd.h>
5#include "owl.h"
6
7int owl_zwrite_create_from_line(owl_zwrite *z, const char *line)
8{
9  int argc, badargs, myargc, i, len;
10  char **argv;
11  const char *const *myargv;
12  char *msg = NULL;
13
14  badargs=0;
15 
16  /* start with null entries */
17  z->realm=NULL;
18  z->class=NULL;
19  z->inst=NULL;
20  z->opcode=NULL;
21  z->zsig=NULL;
22  z->message=NULL;
23  z->cc=0;
24  z->noping=0;
25  owl_list_create(&(z->recips));
26  z->zwriteline = owl_strdup(line);
27
28  /* parse the command line for options */
29  argv=owl_parseline(line, &argc);
30  myargv=strs(argv);
31  if (argc<0) {
32    owl_function_error("Unbalanced quotes in zwrite");
33    return(-1);
34  }
35  myargc=argc;
36  if (myargc && *(myargv[0])!='-') {
37    myargc--;
38    myargv++;
39  }
40  while (myargc) {
41    if (!strcmp(myargv[0], "-c")) {
42      if (myargc<2) {
43        badargs=1;
44        break;
45      }
46      z->class=owl_validate_utf8(myargv[1]);
47      myargv+=2;
48      myargc-=2;
49    } else if (!strcmp(myargv[0], "-i")) {
50      if (myargc<2) {
51        badargs=1;
52        break;
53      }
54      z->inst=owl_validate_utf8(myargv[1]);
55      myargv+=2;
56      myargc-=2;
57    } else if (!strcmp(myargv[0], "-r")) {
58      if (myargc<2) {
59        badargs=1;
60        break;
61      }
62      z->realm=owl_validate_utf8(myargv[1]);
63      myargv+=2;
64      myargc-=2;
65    } else if (!strcmp(myargv[0], "-s")) {
66      if (myargc<2) {
67        badargs=1;
68        break;
69      }
70      z->zsig=owl_validate_utf8(myargv[1]);
71      myargv+=2;
72      myargc-=2;
73    } else if (!strcmp(myargv[0], "-O")) {
74      if (myargc<2) {
75        badargs=1;
76        break;
77      }
78      z->opcode=owl_validate_utf8(myargv[1]);
79      myargv+=2;
80      myargc-=2;
81    } else if (!strcmp(myargv[0], "-m")) {
82      if (myargc<2) {
83        badargs=1;
84        break;
85      }
86      /* we must already have users or a class or an instance */
87      if (owl_list_get_size(&(z->recips))<1 && (!z->class) && (!z->inst)) {
88        badargs=1;
89        break;
90      }
91
92      /* Once we have -m, gobble up everything else on the line */
93      myargv++;
94      myargc--;
95      len = 0;
96      for (i=0;i<myargc;i++) {
97        len += strlen(myargv[i]) + 1;
98      }
99      len++;    /* NULL after the last trailing ' ' in the loop below. */
100      msg = owl_malloc(len);
101      msg[0] = '\0';
102      while (myargc) {
103        strcat(msg, myargv[0]);
104        strcat(msg, " ");
105        myargc--;
106        myargv++;
107      }
108      msg[strlen(msg)-1] = '\0';
109      break;
110    } else if (!strcmp(myargv[0], "-C")) {
111      z->cc=1;
112      myargv++;
113      myargc--;
114    } else if (!strcmp(myargv[0], "-n")) {
115      z->noping=1;
116      myargv++;
117      myargc--;
118    } else {
119      /* anything unattached is a recipient */
120      owl_list_append_element(&(z->recips), owl_validate_utf8(myargv[0]));
121      myargv++;
122      myargc--;
123    }
124  }
125
126  owl_parsefree(argv, argc);
127
128  if (badargs) {
129    return(-1);
130  }
131
132  if (z->class == NULL &&
133      z->inst == NULL &&
134      owl_list_get_size(&(z->recips))==0) {
135    owl_function_error("You must specify a recipient for zwrite");
136    return(-1);
137  }
138
139  /* now deal with defaults */
140  if (z->class==NULL) z->class=owl_strdup("message");
141  if (z->inst==NULL) z->inst=owl_strdup("personal");
142  if (z->realm==NULL) z->realm=owl_strdup("");
143  if (z->opcode==NULL) z->opcode=owl_strdup("");
144  /* z->message is allowed to stay NULL */
145
146  if(msg) {
147    owl_zwrite_set_message(z, msg);
148    owl_free(msg);
149  }
150
151  return(0);
152}
153
154void owl_zwrite_populate_zsig(owl_zwrite *z)
155{
156  const char *zsigproc, *zsigowlvar, *zsigzvar;
157  char *ptr;
158  struct passwd *pw;
159
160  /* get a zsig, if not given */
161  if (z->zsig==NULL) {
162    zsigproc = owl_global_get_zsigproc(&g);
163    zsigowlvar = owl_global_get_zsig(&g);
164    zsigzvar = owl_zephyr_get_variable("zwrite-signature");
165
166    if (zsigowlvar && *zsigowlvar) {
167      z->zsig=owl_validate_utf8(zsigowlvar);
168    } else if (zsigproc && *zsigproc) {
169      FILE *file;
170      char buff[LINE], *openline;
171     
172      /* simple hack for now to nuke stderr */
173#if OWL_STDERR_REDIR
174      openline = owl_strdup(zsigproc);
175#else
176      openline = owl_sprintf("%s 2> /dev/null", zsigproc);
177#endif
178      file=popen(openline, "r");
179      owl_free(openline);
180      if (!file) {
181        if (zsigzvar && *zsigzvar) {
182          z->zsig=owl_validate_utf8(zsigzvar);
183        }
184      } else {
185        z->zsig=owl_malloc(LINE*5);
186        strcpy(z->zsig, "");
187        while (fgets(buff, LINE, file)) { /* wrong sizing */
188          strcat(z->zsig, buff);
189        }
190        pclose(file);
191        if (z->zsig[0] != '\0' && z->zsig[strlen(z->zsig) - 1] == '\n') {
192          z->zsig[strlen(z->zsig)-1]='\0';
193        }
194      }
195    } else if (zsigzvar) {
196      z->zsig=owl_validate_utf8(zsigzvar);
197    } else if (((pw=getpwuid(getuid()))!=NULL) && (pw->pw_gecos)) {
198      z->zsig=owl_validate_utf8(pw->pw_gecos);
199      ptr=strchr(z->zsig, ',');
200      if (ptr) {
201        ptr[0]='\0';
202      }
203    }
204  }
205}
206
207void owl_zwrite_send_ping(const owl_zwrite *z)
208{
209  int i, j;
210  char *to;
211
212  if (z->noping) return;
213 
214  if (strcasecmp(z->class, "message")) {
215    return;
216  }
217
218  /* if there are no recipients we won't send a ping, which
219     is what we want */
220  j=owl_list_get_size(&(z->recips));
221  for (i=0; i<j; i++) {
222    if (strcmp(z->realm, "")) {
223      to = owl_sprintf("%s@%s", (const char *) owl_list_get_element(&(z->recips), i), z->realm);
224    } else {
225      to = owl_strdup(owl_list_get_element(&(z->recips), i));
226    }
227    send_ping(to, z->class, z->inst);
228    owl_free(to);
229  }
230
231}
232
233void owl_zwrite_set_message(owl_zwrite *z, const char *msg)
234{
235  int i, j;
236  char *toline = NULL;
237  char *tmp = NULL, *tmp2;
238
239  if (z->message) owl_free(z->message);
240
241  j=owl_list_get_size(&(z->recips));
242  if (j>0 && z->cc) {
243    toline = owl_strdup( "CC: ");
244    for (i=0; i<j; i++) {
245      tmp = toline;
246      if (strcmp(z->realm, "")) {
247        toline = owl_sprintf( "%s%s@%s ", toline, (const char *) owl_list_get_element(&(z->recips), i), z->realm);
248      } else {
249        toline = owl_sprintf( "%s%s ", toline, (const char *) owl_list_get_element(&(z->recips), i));
250      }
251      owl_free(tmp);
252      tmp = NULL;
253    }
254    tmp = owl_validate_utf8(msg);
255    tmp2 = owl_text_expand_tabs(tmp);
256    z->message=owl_sprintf("%s\n%s", toline, tmp2);
257    owl_free(toline);
258    owl_free(tmp);
259    owl_free(tmp2);
260  } else {
261    tmp=owl_validate_utf8(msg);
262    z->message=owl_text_expand_tabs(tmp);
263    owl_free(tmp);
264  }
265}
266
267const char *owl_zwrite_get_message(const owl_zwrite *z)
268{
269  if (z->message) return(z->message);
270  return("");
271}
272
273int owl_zwrite_is_message_set(const owl_zwrite *z)
274{
275  if (z->message) return(1);
276  return(0);
277}
278
279int owl_zwrite_send_message(const owl_zwrite *z)
280{
281  int i, j;
282  char *to = NULL;
283
284  if (z->message==NULL) return(-1);
285
286  j=owl_list_get_size(&(z->recips));
287  if (j>0) {
288    for (i=0; i<j; i++) {
289      if (strcmp(z->realm, "")) {
290        to = owl_sprintf("%s@%s", (const char *) owl_list_get_element(&(z->recips), i), z->realm);
291      } else {
292        to = owl_strdup( owl_list_get_element(&(z->recips), i));
293      }
294      send_zephyr(z->opcode, z->zsig, z->class, z->inst, to, z->message);
295      owl_free(to);
296      to = NULL;
297    }
298  } else {
299    to = owl_sprintf( "@%s", z->realm);
300    send_zephyr(z->opcode, z->zsig, z->class, z->inst, to, z->message);
301  }
302  owl_free(to);
303  return(0);
304}
305
306int owl_zwrite_create_and_send_from_line(const char *cmd, const char *msg)
307{
308  owl_zwrite z;
309  int rv;
310  rv=owl_zwrite_create_from_line(&z, cmd);
311  if (rv) return(rv);
312  if (!owl_zwrite_is_message_set(&z)) {
313    owl_zwrite_set_message(&z, msg);
314  }
315  owl_zwrite_populate_zsig(&z);
316  owl_zwrite_send_message(&z);
317  owl_zwrite_free(&z);
318  return(0);
319}
320
321const char *owl_zwrite_get_class(const owl_zwrite *z)
322{
323  return(z->class);
324}
325
326const char *owl_zwrite_get_instance(const owl_zwrite *z)
327{
328  return(z->inst);
329}
330
331const char *owl_zwrite_get_opcode(const owl_zwrite *z)
332{
333  return(z->opcode);
334}
335
336void owl_zwrite_set_opcode(owl_zwrite *z, const char *opcode)
337{
338  if (z->opcode) owl_free(z->opcode);
339  z->opcode=owl_validate_utf8(opcode);
340}
341
342const char *owl_zwrite_get_realm(const owl_zwrite *z)
343{
344  return(z->realm);
345}
346
347const char *owl_zwrite_get_zsig(const owl_zwrite *z)
348{
349  if (z->zsig) return(z->zsig);
350  return("");
351}
352
353void owl_zwrite_set_zsig(owl_zwrite *z, const char *zsig)
354{
355  if(z->zsig) owl_free(z->zsig);
356  z->zsig = owl_strdup(zsig);
357}
358
359void owl_zwrite_get_recipstr(const owl_zwrite *z, char *buff)
360{
361  int i, j;
362
363  strcpy(buff, "");
364  j=owl_list_get_size(&(z->recips));
365  for (i=0; i<j; i++) {
366    strcat(buff, owl_list_get_element(&(z->recips), i));
367    strcat(buff, " ");
368  }
369  buff[strlen(buff)-1]='\0';
370}
371
372int owl_zwrite_get_numrecips(const owl_zwrite *z)
373{
374  return(owl_list_get_size(&(z->recips)));
375}
376
377const char *owl_zwrite_get_recip_n(const owl_zwrite *z, int n)
378{
379  return(owl_list_get_element(&(z->recips), n));
380}
381
382int owl_zwrite_is_personal(const owl_zwrite *z)
383{
384  /* return true if at least one of the recipients is personal */
385  int i, j;
386  char *foo;
387
388  j=owl_list_get_size(&(z->recips));
389  for (i=0; i<j; i++) {
390    foo=owl_list_get_element(&(z->recips), i);
391    if (foo[0]!='@') return(1);
392  }
393  return(0);
394}
395
396void owl_zwrite_free(owl_zwrite *z)
397{
398  owl_list_free_all(&(z->recips), &owl_free);
399  if (z->zwriteline) owl_free(z->zwriteline);
400  if (z->class) owl_free(z->class);
401  if (z->inst) owl_free(z->inst);
402  if (z->opcode) owl_free(z->opcode);
403  if (z->realm) owl_free(z->realm);
404  if (z->message) owl_free(z->message);
405  if (z->zsig) owl_free(z->zsig);
406}
Note: See TracBrowser for help on using the repository browser.