Rearranged whitespace (and comments) in all the code!
[mmh] / uip / replsbr.c
1 /*
2  * replsbr.c -- routines to help repl along...
3  *
4  * This code is Copyright (c) 2002, by the authors of nmh.  See the
5  * COPYRIGHT file in the root directory of the nmh distribution for
6  * complete copyright information.
7  */
8
9 #include <h/mh.h>
10 #include <h/addrsbr.h>
11 #include <h/fmt_scan.h>
12 #include <h/utils.h>
13 #include <sys/file.h>  /* L_SET */
14 #include <errno.h>
15
16 extern short ccto;  /* from repl.c */
17 extern short cccc;
18 extern short ccme;
19 extern short querysw;
20
21 static int dftype=0;
22
23 static char *badaddrs = NULL;
24 static char *dfhost = NULL;
25
26 static struct mailname mq = { NULL };
27
28 /*
29  * Buffer size for content part of header fields.
30  * We want this to be large enough so that we don't
31  * do a lot of extra FLDPLUS calls on m_getfld but
32  * small enough so that we don't snarf the entire
33  * message body when we're not going to use any of it.
34  */
35 #define SBUFSIZ 256
36
37 static struct format *fmt;
38
39 static int ncomps = 0;  /* # of interesting components */
40 static char **compbuffers = NULL;  /* buffers for component text */
41 static struct comp **used_buf = NULL;  /* stack for comp that use buffers */
42
43 static int dat[5];  /* aux. data for format routine */
44
45 static char *addrcomps[] = {
46         "from",
47         "sender",
48         "reply-to",
49         "to",
50         "cc",
51         "bcc",
52         "resent-from",
53         "resent-sender",
54         "resent-reply-to",
55         "resent-to",
56         "resent-cc",
57         "resent-bcc",
58         NULL
59 };
60
61 /*
62  * static prototypes
63  */
64 static int insert (struct mailname *);
65 static void replfilter (FILE *, FILE *, char *);
66
67
68 void
69 replout (FILE *inb, char *msg, char *drft, struct msgs *mp, int outputlinelen,
70         int mime, char *form, char *filter, char *fcc)
71 {
72         register int state, i;
73         register struct comp *cptr;
74         register char *tmpbuf;
75         register char **nxtbuf;
76         register char **ap;
77         register struct comp **savecomp;
78         int char_read = 0, format_len, mask;
79         char name[NAMESZ], *scanl;
80         unsigned char *cp;
81         FILE *out;
82
83         mask = umask(~m_gmprot());
84         if ((out = fopen (drft, "w")) == NULL)
85                 adios (drft, "unable to create");
86
87         umask(mask);
88
89         /* get new format string */
90         cp = new_fs (form, NULL, NULL);
91         format_len = strlen (cp);
92
93         /* compile format string */
94         ncomps = fmt_compile (cp, &fmt) + 1;
95
96         if (!(nxtbuf = compbuffers = (char **)
97                         calloc((size_t) ncomps, sizeof(char *))))
98                 adios (NULL, "unable to allocate component buffers");
99         if (!(savecomp = used_buf = (struct comp **)
100                         calloc((size_t) (ncomps+1), sizeof(struct comp *))))
101                 adios (NULL, "unable to allocate component buffer stack");
102         savecomp += ncomps + 1;
103         *--savecomp = NULL;  /* point at zero'd end minus 1 */
104
105         for (i = ncomps; i--; )
106                 *nxtbuf++ = mh_xmalloc(SBUFSIZ);
107
108         nxtbuf = compbuffers;  /* point at start */
109         tmpbuf = *nxtbuf++;
110
111         for (ap = addrcomps; *ap; ap++) {
112                 FINDCOMP (cptr, *ap);
113                 if (cptr)
114                         cptr->c_type |= CT_ADDR;
115         }
116
117         /*
118          * ignore any components killed by command line switches
119          */
120         if (!ccto) {
121                 FINDCOMP (cptr, "to");
122                 if (cptr)
123                         cptr->c_name = "";
124         }
125         if (!cccc) {
126                 FINDCOMP (cptr, "cc");
127                 if (cptr)
128                         cptr->c_name = "";
129         }
130         /* set up the "fcc" pseudo-component */
131         if (fcc) {
132                 FINDCOMP (cptr, "fcc");
133                 if (cptr)
134                         cptr->c_text = getcpy (fcc);
135         }
136         if ((cp = getenv("USER"))) {
137                 FINDCOMP (cptr, "user");
138                 if (cptr)
139                         cptr->c_text = getcpy(cp);
140         }
141         if (!ccme)
142                 ismymbox (NULL);
143
144         /*
145          * pick any interesting stuff out of msg "inb"
146          */
147         for (state = FLD;;) {
148                 state = m_getfld (state, name, tmpbuf, SBUFSIZ, inb);
149                 switch (state) {
150                         case FLD:
151                         case FLDPLUS:
152                                 /*
153                                  * if we're interested in this component,
154                                  * save a pointer to the component text,
155                                  * then start using our next free buffer
156                                  * as the component temp buffer (buffer
157                                  * switching saves an extra copy of the
158                                  * component text).
159                                  */
160                                 if ((cptr = wantcomp[CHASH(name)]))
161                                         do {
162                                                 if (!mh_strcasecmp(name, cptr->c_name)) {
163                                                         char_read += msg_count;
164                                                         if (! cptr->c_text) {
165                                                                 i = strlen(cptr->c_text = tmpbuf) - 1;
166                                                                 if (tmpbuf[i] == '\n')
167                                                                         tmpbuf[i] = '\0';
168                                                                 *--savecomp = cptr;
169                                                                 tmpbuf = *nxtbuf++;
170                                                         } else {
171                                                                 i = strlen (cp = cptr->c_text) - 1;
172                                                                 if (cp[i] == '\n') {
173                                                                         if (cptr->c_type & CT_ADDR) {
174                                                                                 cp[i] = '\0';
175                                                                                 cp = add (",\n\t", cp);
176                                                                         } else {
177                                                                                 cp = add ("\t", cp);
178                                                                         }
179                                                                 }
180                                                                 cptr->c_text = add (tmpbuf, cp);
181                                                         }
182                                                         while (state == FLDPLUS) {
183                                                                 state = m_getfld (state, name, tmpbuf,
184                                                                                                   SBUFSIZ, inb);
185                                                                 cptr->c_text = add (tmpbuf, cptr->c_text);
186                                                                 char_read += msg_count;
187                                                         }
188                                                         break;
189                                                 }
190                                         } while ((cptr = cptr->c_next));
191
192                                 while (state == FLDPLUS)
193                                         state = m_getfld (state, name, tmpbuf, SBUFSIZ, inb);
194                                 break;
195
196                         case LENERR:
197                         case FMTERR:
198                         case BODY:
199                         case FILEEOF:
200                                 goto finished;
201
202                         default:
203                                 adios (NULL, "m_getfld() returned %d", state);
204                 }
205         }
206
207         /*
208          * format and output the header lines.
209          */
210 finished:
211
212         /*
213          * if there's a "Subject" component, strip any "Re:"s off it
214          */
215         FINDCOMP (cptr, "subject")
216         if (cptr && (cp = cptr->c_text)) {
217                 register char *sp = cp;
218
219                 for (;;) {
220                         while (isspace(*cp))
221                                 cp++;
222                         if(uprf(cp, "re:"))
223                                 cp += 3;
224                         else
225                                 break;
226                         sp = cp;
227                 }
228                 if (sp != cptr->c_text) {
229                         cp = cptr->c_text;
230                         cptr->c_text = getcpy (sp);
231                         free (cp);
232                 }
233         }
234         i = format_len + char_read + 256;
235         scanl = mh_xmalloc ((size_t) i + 2);
236         dat[0] = 0;
237         dat[1] = 0;
238         dat[2] = 0;
239         dat[3] = outputlinelen;
240         dat[4] = 0;
241         fmt_scan (fmt, scanl, i, dat);
242         fputs (scanl, out);
243         if (badaddrs) {
244                 fputs ("\nrepl: bad addresses:\n", out);
245                 fputs ( badaddrs, out);
246         }
247
248         /*
249          * Check if we should filter the message
250          * or add mhbuild directives
251          */
252         if (filter) {
253                 fflush(out);
254                 if (ferror (out))
255                         adios (drft, "error writing");
256
257                 replfilter (inb, out, filter);
258         } else if (mime && mp) {
259                         fprintf (out, "#forw [original message] +%s %s\n",
260                                          mp->foldpath, m_name (mp->lowsel));
261         }
262
263         fflush(out);
264         if (ferror (out))
265                 adios (drft, "error writing");
266         fclose (out);
267
268         /* return dynamically allocated buffers */
269         free (scanl);
270         for (nxtbuf = compbuffers, i = ncomps; (cptr = *savecomp++); nxtbuf++, i--)
271                 free (cptr->c_text);  /* if not nxtbuf, nxtbuf already freed */
272         while ( i-- > 0)
273                 free (*nxtbuf++);  /* free unused nxtbufs */
274         free ((char *) compbuffers);
275         free ((char *) used_buf);
276 }
277
278 static char *buf;  /* our current working buffer */
279 static char *bufend;  /* end of working buffer */
280 static char *last_dst;  /* buf ptr at end of last call */
281 static unsigned int bufsiz=0;  /* current size of buf */
282
283 #define BUFINCR 512  /* how much to expand buf when if fills */
284
285 #define CPY(s) { cp = (s); while ((*dst++ = *cp++)) ; --dst; }
286
287 /*
288  * check if there's enough room in buf for str.
289  * add more mem if needed
290  */
291 #define CHECKMEM(str) \
292         if ((len = strlen (str)) >= bufend - dst) {\
293                 int i = dst - buf;\
294                 int n = last_dst - buf;\
295                 bufsiz += ((dst + len - bufend) / BUFINCR + 1) * BUFINCR;\
296                 buf = mh_xrealloc (buf, bufsiz);\
297                 dst = buf + i;\
298                 last_dst = buf + n;\
299                 bufend = buf + bufsiz;\
300         }
301
302
303 /*
304  * fmt_scan will call this routine if the user includes the function
305  * "(formataddr {component})" in a format string.  "orig" is the
306  * original contents of the string register.  "str" is the address
307  * string to be formatted and concatenated onto orig.  This routine
308  * returns a pointer to the concatenated address string.
309  *
310  * We try to not do a lot of malloc/copy/free's (which is why we
311  * don't call "getcpy") but still place no upper limit on the
312  * length of the result string.
313  */
314 char *
315 formataddr (char *orig, char *str)
316 {
317         register int len;
318         char baddr[BUFSIZ], error[BUFSIZ];
319         register int isgroup;
320         register char *dst;
321         register char *cp;
322         register char *sp;
323         register struct mailname *mp = NULL;
324
325         /* if we don't have a buffer yet, get one */
326         if (bufsiz == 0) {
327                 buf = mh_xmalloc (BUFINCR);
328                 last_dst = buf;  /* XXX */
329                 bufsiz = BUFINCR - 6;  /* leave some slop */
330                 bufend = buf + bufsiz;
331         }
332         /*
333          * If "orig" points to our buffer we can just pick up where we
334          * left off.  Otherwise we have to copy orig into our buffer.
335          */
336         if (orig == buf)
337                 dst = last_dst;
338         else if (!orig || !*orig) {
339                 dst = buf;
340                 *dst = '\0';
341         } else {
342                 dst = last_dst;  /* XXX */
343                 CHECKMEM (orig);
344                 CPY (orig);
345         }
346
347         /* concatenate all the new addresses onto 'buf' */
348         for (isgroup = 0; (cp = getname (str)); ) {
349                 if ((mp = getm (cp, dfhost, dftype, AD_NAME, error)) == NULL) {
350                         snprintf (baddr, sizeof(baddr), "\t%s -- %s\n", cp, error);
351                         badaddrs = add (baddr, badaddrs);
352                         continue;
353                 }
354                 if (isgroup && (mp->m_gname || !mp->m_ingrp)) {
355                         *dst++ = ';';
356                         isgroup = 0;
357                 }
358                 if (insert (mp)) {
359                         /* if we get here we're going to add an address */
360                         if (dst != buf) {
361                                 *dst++ = ',';
362                                 *dst++ = ' ';
363                         }
364                         if (mp->m_gname) {
365                                 CHECKMEM (mp->m_gname);
366                                 CPY (mp->m_gname);
367                                 isgroup++;
368                         }
369                         sp = adrformat (mp);
370                         CHECKMEM (sp);
371                         CPY (sp);
372                 }
373         }
374
375         if (isgroup)
376                 *dst++ = ';';
377
378         *dst = '\0';
379         last_dst = dst;
380         return (buf);
381 }
382
383
384 static int
385 insert (struct mailname *np)
386 {
387         char buffer[BUFSIZ];
388         register struct mailname *mp;
389
390         if (np->m_mbox == NULL)
391                 return 0;
392
393         for (mp = &mq; mp->m_next; mp = mp->m_next) {
394                 if (!mh_strcasecmp (np->m_host, mp->m_next->m_host)
395                                 && !mh_strcasecmp (np->m_mbox, mp->m_next->m_mbox))
396                         return 0;
397         }
398         if (!ccme && ismymbox (np))
399                 return 0;
400
401         if (querysw) {
402                 snprintf (buffer, sizeof(buffer), "Reply to %s? ", adrformat (np));
403                 if (!gans (buffer, anoyes))
404                 return 0;
405         }
406         mp->m_next = np;
407
408 #ifdef ISI
409         if (ismymbox (np))
410                 ccme = 0;
411 #endif
412
413         return 1;
414 }
415
416
417 /*
418  * Call the mhlproc
419  *
420  * This function expects that argument out has been fflushed by the caller.
421  */
422
423 static void
424 replfilter (FILE *in, FILE *out, char *filter)
425 {
426         int pid;
427         char *mhl;
428         char *errstr;
429
430         if (filter == NULL)
431                 return;
432
433         if (access (filter, R_OK) == NOTOK)
434                 adios (filter, "unable to read");
435
436         mhl = r1bindex (mhlproc, '/');
437
438         rewind (in);
439         lseek (fileno(in), (off_t) 0, SEEK_SET);
440
441         switch (pid = vfork ()) {
442                 case NOTOK:
443                         adios ("fork", "unable to");
444
445                 case OK:
446                         dup2 (fileno (in), fileno (stdin));
447                         dup2 (fileno (out), fileno (stdout));
448                         closefds (3);
449
450                         execlp (mhlproc, mhl, "-form", filter, "-noclear", NULL);
451                         errstr = strerror(errno);
452                         write(2, "unable to exec ", 15);
453                         write(2, mhlproc, strlen(mhlproc));
454                         write(2, ": ", 2);
455                         write(2, errstr, strlen(errstr));
456                         write(2, "\n", 1);
457                         _exit (-1);
458
459                 default:
460                         if (pidXwait (pid, mhl))
461                                 done (1);
462                         fseek (out, 0L, SEEK_END);
463                         break;
464         }
465 }