source: zwrite.c @ ecc958e

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