]> CyberLeo.Net >> Repos - FreeBSD/releng/10.0.git/blob - contrib/sendmail/src/milter.c
- Copy stable/10 (r259064) to releng/10.0 as part of the
[FreeBSD/releng/10.0.git] / contrib / sendmail / src / milter.c
1 /*
2  * Copyright (c) 1999-2009, 2012, 2013 Sendmail, Inc. and its suppliers.
3  *      All rights reserved.
4  *
5  * By using this file, you agree to the terms and conditions set
6  * forth in the LICENSE file which can be found at the top level of
7  * the sendmail distribution.
8  *
9  */
10
11 #include <sendmail.h>
12
13 SM_RCSID("@(#)$Id: milter.c,v 8.280 2013/01/16 18:48:36 ca Exp $")
14
15 #if MILTER
16 # include <sm/sendmail.h>
17 # include <libmilter/mfapi.h>
18 # include <libmilter/mfdef.h>
19
20 # include <errno.h>
21 # include <sm/time.h>
22 # include <sys/uio.h>
23
24 # if NETINET || NETINET6
25 #  include <arpa/inet.h>
26 #  if MILTER_NO_NAGLE
27 #   include <netinet/tcp.h>
28 #  endif /* MILTER_NO_NAGLE */
29 # endif /* NETINET || NETINET6 */
30
31 # include <sm/fdset.h>
32
33 static void     milter_connect_timeout __P((int));
34 static void     milter_error __P((struct milter *, ENVELOPE *));
35 static int      milter_open __P((struct milter *, bool, ENVELOPE *));
36 static void     milter_parse_timeouts __P((char *, struct milter *));
37 static char     *milter_sysread __P((struct milter *, char *, ssize_t, time_t,
38                         ENVELOPE *, const char *));
39 static char     *milter_read __P((struct milter *, char *, ssize_t *, time_t,
40                         ENVELOPE *, const char *));
41 static char     *milter_write __P((struct milter *, int, char *, ssize_t,
42                         time_t, ENVELOPE *, const char *));
43 static char     *milter_send_command __P((struct milter *, int, void *,
44                         ssize_t, ENVELOPE *, char *, const char *));
45 static char     *milter_command __P((int, void *, ssize_t, int,
46                         ENVELOPE *, char *, const char *, bool));
47 static char     *milter_body __P((struct milter *, ENVELOPE *, char *));
48 static int      milter_reopen_df __P((ENVELOPE *));
49 static int      milter_reset_df __P((ENVELOPE *));
50 static void     milter_quit_filter __P((struct milter *, ENVELOPE *));
51 static void     milter_abort_filter __P((struct milter *, ENVELOPE *));
52 static void     milter_send_macros __P((struct milter *, char **, int,
53                         ENVELOPE *));
54 static int      milter_negotiate __P((struct milter *, ENVELOPE *,
55                         milters_T *));
56 static void     milter_per_connection_check __P((ENVELOPE *));
57 static char     *milter_headers __P((struct milter *, ENVELOPE *, char *));
58 static void     milter_addheader __P((struct milter *, char *, ssize_t,
59                         ENVELOPE *));
60 static void     milter_insheader __P((struct milter *, char *, ssize_t,
61                         ENVELOPE *));
62 static void     milter_changeheader __P((struct milter *, char *, ssize_t,
63                         ENVELOPE *));
64 static void     milter_chgfrom __P((char *, ssize_t, ENVELOPE *));
65 static void     milter_addrcpt __P((char *, ssize_t, ENVELOPE *));
66 static void     milter_addrcpt_par __P((char *, ssize_t, ENVELOPE *));
67 static void     milter_delrcpt __P((char *, ssize_t, ENVELOPE *));
68 static int      milter_replbody __P((char *, ssize_t, bool, ENVELOPE *));
69 static int      milter_set_macros __P((char *, char **, char *, int));
70
71
72 /* milter states */
73 # define SMFS_CLOSED            'C'     /* closed for all further actions */
74 # define SMFS_OPEN              'O'     /* connected to remote milter filter */
75 # define SMFS_INMSG             'M'     /* currently servicing a message */
76 # define SMFS_DONE              'D'     /* done with current message */
77 # define SMFS_CLOSABLE          'Q'     /* done with current connection */
78 # define SMFS_ERROR             'E'     /* error state */
79 # define SMFS_READY             'R'     /* ready for action */
80 # define SMFS_SKIP              'S'     /* skip body */
81
82 /*
83 **  MilterMacros contains the milter macros for each milter and each stage.
84 **  indices are (in order): stages, milter-index, macro
85 **  milter-index == 0: "global" macros (not for a specific milter).
86 */
87
88 static char *MilterMacros[SMFIM_LAST + 1][MAXFILTERS + 1][MAXFILTERMACROS + 1];
89 static size_t MilterMaxDataSize = MILTER_MAX_DATA_SIZE;
90
91 # define MILTER_CHECK_DONE_MSG() \
92         if (*state == SMFIR_REPLYCODE || \
93             *state == SMFIR_REJECT || \
94             *state == SMFIR_DISCARD || \
95             *state == SMFIR_TEMPFAIL) \
96         { \
97                 /* Abort the filters to let them know we are done with msg */ \
98                 milter_abort(e); \
99         }
100
101 /* set state in case of an error */
102 # define MILTER_SET_STATE       \
103         if (bitnset(SMF_TEMPFAIL, m->mf_flags)) \
104                 *state = SMFIR_TEMPFAIL; \
105         else if (bitnset(SMF_TEMPDROP, m->mf_flags)) \
106                 *state = SMFIR_SHUTDOWN; \
107         else if (bitnset(SMF_REJECT, m->mf_flags)) \
108                 *state = SMFIR_REJECT
109
110 /* flow through code maybe using continue; don't wrap in do {} while */
111 # define MILTER_CHECK_ERROR(initial, action) \
112         if (!initial && tTd(71, 100)) \
113         { \
114                 if (e->e_quarmsg == NULL) \
115                 { \
116                         e->e_quarmsg = sm_rpool_strdup_x(e->e_rpool, \
117                                                          "filter failure"); \
118                         macdefine(&e->e_macro, A_PERM, macid("{quarantine}"), \
119                                   e->e_quarmsg); \
120                 } \
121         } \
122         else if (tTd(71, 101)) \
123         { \
124                 if (e->e_quarmsg == NULL) \
125                 { \
126                         e->e_quarmsg = sm_rpool_strdup_x(e->e_rpool, \
127                                                          "filter failure"); \
128                         macdefine(&e->e_macro, A_PERM, macid("{quarantine}"), \
129                                   e->e_quarmsg); \
130                 } \
131         } \
132         else MILTER_SET_STATE;  \
133         else \
134                 action;
135
136 # define MILTER_CHECK_REPLYCODE(default) \
137         if (response == NULL || \
138             strlen(response) + 1 != (size_t) rlen || \
139             rlen < 3 || \
140             (response[0] != '4' && response[0] != '5') || \
141             !isascii(response[1]) || !isdigit(response[1]) || \
142             !isascii(response[2]) || !isdigit(response[2])) \
143         { \
144                 if (response != NULL) \
145                         sm_free(response); /* XXX */ \
146                 response = newstr(default); \
147         } \
148         else \
149         { \
150                 char *ptr = response; \
151  \
152                 /* Check for unprotected %'s in the string */ \
153                 while (*ptr != '\0') \
154                 { \
155                         if (*ptr == '%' && *++ptr != '%') \
156                         { \
157                                 sm_free(response); /* XXX */ \
158                                 response = newstr(default); \
159                                 break; \
160                         } \
161                         ptr++; \
162                 } \
163         }
164
165 # define MILTER_DF_ERROR(msg) \
166 { \
167         int save_errno = errno; \
168  \
169         if (tTd(64, 5)) \
170         { \
171                 sm_dprintf(msg, dfname, sm_errstring(save_errno)); \
172                 sm_dprintf("\n"); \
173         } \
174         if (MilterLogLevel > 0) \
175                 sm_syslog(LOG_ERR, e->e_id, msg, dfname, sm_errstring(save_errno)); \
176         if (SuperSafe == SAFE_REALLY) \
177         { \
178                 if (e->e_dfp != NULL) \
179                 { \
180                         (void) sm_io_close(e->e_dfp, SM_TIME_DEFAULT); \
181                         e->e_dfp = NULL; \
182                 } \
183                 e->e_flags &= ~EF_HAS_DF; \
184         } \
185         errno = save_errno; \
186 }
187
188 /*
189 **  MILTER_TIMEOUT -- make sure socket is ready in time
190 **
191 **      Parameters:
192 **              routine -- routine name for debug/logging
193 **              secs -- number of seconds in timeout
194 **              write -- waiting to read or write?
195 **              started -- whether this is part of a previous sequence
196 **
197 **      Assumes 'm' is a milter structure for the current socket.
198 */
199
200 # define MILTER_TIMEOUT(routine, secs, write, started, function) \
201 { \
202         int ret; \
203         int save_errno; \
204         fd_set fds; \
205         struct timeval tv; \
206  \
207         if (SM_FD_SETSIZE > 0 && m->mf_sock >= SM_FD_SETSIZE) \
208         { \
209                 if (tTd(64, 5)) \
210                         sm_dprintf("milter_%s(%s): socket %d is larger than FD_SETSIZE %d\n", \
211                                    (routine), m->mf_name, m->mf_sock, \
212                                    SM_FD_SETSIZE); \
213                 if (MilterLogLevel > 0) \
214                         sm_syslog(LOG_ERR, e->e_id, \
215                                   "Milter (%s): socket(%s) %d is larger than FD_SETSIZE %d", \
216                                   m->mf_name, (routine), m->mf_sock, \
217                                   SM_FD_SETSIZE); \
218                 milter_error(m, e); \
219                 return NULL; \
220         } \
221  \
222         do \
223         { \
224                 FD_ZERO(&fds); \
225                 SM_FD_SET(m->mf_sock, &fds); \
226                 tv.tv_sec = (secs); \
227                 tv.tv_usec = 0; \
228                 ret = select(m->mf_sock + 1, \
229                              (write) ? NULL : &fds, \
230                              (write) ? &fds : NULL, \
231                              NULL, &tv); \
232         } while (ret < 0 && errno == EINTR); \
233  \
234         switch (ret) \
235         { \
236           case 0: \
237                 if (tTd(64, 5)) \
238                         sm_dprintf("milter_%s(%s): timeout, where=%s\n", \
239                                 (routine), m->mf_name, (function)); \
240                 if (MilterLogLevel > 0) \
241                         sm_syslog(LOG_ERR, e->e_id, \
242                                   "Milter (%s): timeout %s data %s, where=%s", \
243                                   m->mf_name, \
244                                   started ? "during" : "before", \
245                                   (routine), (function)); \
246                 milter_error(m, e); \
247                 return NULL; \
248  \
249           case -1: \
250                 save_errno = errno; \
251                 if (tTd(64, 5)) \
252                         sm_dprintf("milter_%s(%s): select: %s\n", (routine), \
253                                    m->mf_name, sm_errstring(save_errno)); \
254                 if (MilterLogLevel > 0) \
255                 { \
256                         sm_syslog(LOG_ERR, e->e_id, \
257                                   "Milter (%s): select(%s): %s", \
258                                   m->mf_name, (routine), \
259                                   sm_errstring(save_errno)); \
260                 } \
261                 milter_error(m, e); \
262                 return NULL; \
263  \
264           default: \
265                 if (SM_FD_ISSET(m->mf_sock, &fds)) \
266                         break; \
267                 if (tTd(64, 5)) \
268                         sm_dprintf("milter_%s(%s): socket not ready\n", \
269                                 (routine), m->mf_name); \
270                 if (MilterLogLevel > 0) \
271                 { \
272                         sm_syslog(LOG_ERR, e->e_id, \
273                                   "Milter (%s): socket(%s) not ready", \
274                                   m->mf_name, (routine)); \
275                 } \
276                 milter_error(m, e); \
277                 return NULL; \
278         } \
279 }
280
281 /*
282 **  Low level functions
283 */
284
285 /*
286 **  MILTER_READ -- read from a remote milter filter
287 **
288 **      Parameters:
289 **              m -- milter to read from.
290 **              cmd -- return param for command read.
291 **              rlen -- return length of response string.
292 **              to -- timeout in seconds.
293 **              e -- current envelope.
294 **
295 **      Returns:
296 **              response string (may be NULL)
297 */
298
299 static char *
300 milter_sysread(m, buf, sz, to, e, where)
301         struct milter *m;
302         char *buf;
303         ssize_t sz;
304         time_t to;
305         ENVELOPE *e;
306         const char *where;
307 {
308         time_t readstart = 0;
309         ssize_t len, curl;
310         bool started = false;
311
312         curl = 0;
313
314         if (to > 0)
315                 readstart = curtime();
316
317         for (;;)
318         {
319                 if (to > 0)
320                 {
321                         time_t now;
322
323                         now = curtime();
324                         if (now - readstart >= to)
325                         {
326                                 if (tTd(64, 5))
327                                         sm_dprintf("milter_sys_read (%s): timeout %s data read in %s",
328                                                   m->mf_name,
329                                                   started ? "during" : "before",
330                                                   where);
331                                 if (MilterLogLevel > 0)
332                                         sm_syslog(LOG_ERR, e->e_id,
333                                                   "Milter (%s): timeout %s data read in %s",
334                                                   m->mf_name,
335                                                   started ? "during" : "before",
336                                                   where);
337                                 milter_error(m, e);
338                                 return NULL;
339                         }
340                         to -= now - readstart;
341                         readstart = now;
342                         MILTER_TIMEOUT("read", to, false, started, where);
343                 }
344
345                 len = read(m->mf_sock, buf + curl, sz - curl);
346
347                 if (len < 0)
348                 {
349                         int save_errno = errno;
350
351                         if (tTd(64, 5))
352                                 sm_dprintf("milter_sys_read(%s): read returned %ld: %s\n",
353                                         m->mf_name, (long) len,
354                                         sm_errstring(save_errno));
355                         if (MilterLogLevel > 0)
356                                 sm_syslog(LOG_ERR, e->e_id,
357                                           "Milter (%s): read returned %ld: %s",
358                                           m->mf_name, (long) len,
359                                           sm_errstring(save_errno));
360                         milter_error(m, e);
361                         return NULL;
362                 }
363
364                 started = true;
365                 curl += len;
366                 if (len == 0 || curl >= sz)
367                         break;
368
369         }
370
371         if (curl != sz)
372         {
373                 if (tTd(64, 5))
374                         sm_dprintf("milter_sys_read(%s): cmd read returned %ld, expecting %ld\n",
375                                 m->mf_name, (long) curl, (long) sz);
376                 if (MilterLogLevel > 0)
377                         sm_syslog(LOG_ERR, e->e_id,
378                                   "milter_sys_read(%s): cmd read returned %ld, expecting %ld",
379                                   m->mf_name, (long) curl, (long) sz);
380                 milter_error(m, e);
381                 return NULL;
382         }
383         return buf;
384 }
385
386 static char *
387 milter_read(m, cmd, rlen, to, e, where)
388         struct milter *m;
389         char *cmd;
390         ssize_t *rlen;
391         time_t to;
392         ENVELOPE *e;
393         const char *where;
394 {
395         time_t readstart = 0;
396         ssize_t expl;
397         mi_int32 i;
398 # if MILTER_NO_NAGLE && defined(TCP_CORK)
399         int cork = 0;
400 # endif /* MILTER_NO_NAGLE && defined(TCP_CORK) */
401         char *buf;
402         char data[MILTER_LEN_BYTES + 1];
403
404         if (m->mf_sock < 0)
405         {
406                 if (MilterLogLevel > 0)
407                         sm_syslog(LOG_ERR, e->e_id,
408                                   "milter_read(%s): socket closed, where=%s",
409                                   m->mf_name, where);
410                 milter_error(m, e);
411                 return NULL;
412         }
413
414         *rlen = 0;
415         *cmd = '\0';
416
417         if (to > 0)
418                 readstart = curtime();
419
420 # if MILTER_NO_NAGLE && defined(TCP_CORK)
421         setsockopt(m->mf_sock, IPPROTO_TCP, TCP_CORK, (char *)&cork,
422                    sizeof(cork));
423 # endif /* MILTER_NO_NAGLE && defined(TCP_CORK) */
424
425         if (milter_sysread(m, data, sizeof(data), to, e, where) == NULL)
426                 return NULL;
427
428 # if MILTER_NO_NAGLE && defined(TCP_CORK)
429         cork = 1;
430         setsockopt(m->mf_sock, IPPROTO_TCP, TCP_CORK, (char *)&cork,
431                    sizeof(cork));
432 # endif /* MILTER_NO_NAGLE && defined(TCP_CORK) */
433
434         /* reset timeout */
435         if (to > 0)
436         {
437                 time_t now;
438
439                 now = curtime();
440                 if (now - readstart >= to)
441                 {
442                         if (tTd(64, 5))
443                                 sm_dprintf("milter_read(%s): timeout before data read, where=%s\n",
444                                         m->mf_name, where);
445                         if (MilterLogLevel > 0)
446                                 sm_syslog(LOG_ERR, e->e_id,
447                                           "Milter read(%s): timeout before data read, where=%s",
448                                           m->mf_name, where);
449                         milter_error(m, e);
450                         return NULL;
451                 }
452                 to -= now - readstart;
453         }
454
455         *cmd = data[MILTER_LEN_BYTES];
456         data[MILTER_LEN_BYTES] = '\0';
457         (void) memcpy(&i, data, MILTER_LEN_BYTES);
458         expl = ntohl(i) - 1;
459
460         if (tTd(64, 25))
461                 sm_dprintf("milter_read(%s): expecting %ld bytes\n",
462                         m->mf_name, (long) expl);
463
464         if (expl < 0)
465         {
466                 if (tTd(64, 5))
467                         sm_dprintf("milter_read(%s): read size %ld out of range, where=%s\n",
468                                 m->mf_name, (long) expl, where);
469                 if (MilterLogLevel > 0)
470                         sm_syslog(LOG_ERR, e->e_id,
471                                   "milter_read(%s): read size %ld out of range, where=%s",
472                                   m->mf_name, (long) expl, where);
473                 milter_error(m, e);
474                 return NULL;
475         }
476
477         if (expl == 0)
478                 return NULL;
479
480         buf = (char *) xalloc(expl);
481
482         if (milter_sysread(m, buf, expl, to, e, where) == NULL)
483         {
484                 sm_free(buf); /* XXX */
485                 return NULL;
486         }
487
488         if (tTd(64, 50))
489                 sm_dprintf("milter_read(%s): Returning %*s\n",
490                         m->mf_name, (int) expl, buf);
491         *rlen = expl;
492         return buf;
493 }
494
495 /*
496 **  MILTER_WRITE -- write to a remote milter filter
497 **
498 **      Parameters:
499 **              m -- milter to read from.
500 **              cmd -- command to send.
501 **              buf -- optional command data.
502 **              len -- length of buf.
503 **              to -- timeout in seconds.
504 **              e -- current envelope.
505 **
506 **      Returns:
507 **              buf if successful, NULL otherwise
508 **              Not actually used anywhere but function prototype
509 **                      must match milter_read()
510 */
511
512 static char *
513 milter_write(m, cmd, buf, len, to, e, where)
514         struct milter *m;
515         int cmd;
516         char *buf;
517         ssize_t len;
518         time_t to;
519         ENVELOPE *e;
520         const char *where;
521 {
522         ssize_t sl, i;
523         int num_vectors;
524         mi_int32 nl;
525         char command = (char) cmd;
526         char data[MILTER_LEN_BYTES + 1];
527         bool started = false;
528         struct iovec vector[2];
529
530         /*
531         **  At most two buffers will be written, though
532         **  only one may actually be used (see num_vectors).
533         **  The first is the size/command and the second is the command data.
534         */
535
536         if (len < 0 || len > MilterMaxDataSize)
537         {
538                 if (tTd(64, 5))
539                 {
540                         sm_dprintf("milter_write(%s): length %ld out of range, cmd=%c\n",
541                                 m->mf_name, (long) len, command);
542                         sm_dprintf("milter_write(%s): buf=%s\n",
543                                 m->mf_name, str2prt(buf));
544                 }
545                 if (MilterLogLevel > 0)
546                         sm_syslog(LOG_ERR, e->e_id,
547                                   "milter_write(%s): length %ld out of range, cmd=%c",
548                                   m->mf_name, (long) len, command);
549                 milter_error(m, e);
550                 return NULL;
551         }
552         if (m->mf_sock < 0)
553         {
554                 if (MilterLogLevel > 0)
555                         sm_syslog(LOG_ERR, e->e_id,
556                                   "milter_write(%s): socket closed",
557                                   m->mf_name);
558                 milter_error(m, e);
559                 return NULL;
560         }
561
562         if (tTd(64, 20))
563                 sm_dprintf("milter_write(%s): cmd %c, len %ld\n",
564                            m->mf_name, command, (long) len);
565
566         nl = htonl(len + 1);    /* add 1 for the command char */
567         (void) memcpy(data, (char *) &nl, MILTER_LEN_BYTES);
568         data[MILTER_LEN_BYTES] = command;
569         sl = MILTER_LEN_BYTES + 1;
570
571         /* set up the vector for the size / command */
572         vector[0].iov_base = (void *) data;
573         vector[0].iov_len  = sl;
574
575         /*
576         **  Determine if there is command data.  If so, there will be two
577         **  vectors.  If not, there will be only one.  The vectors are set
578         **  up here and 'num_vectors' and 'sl' are set appropriately.
579         */
580
581         /* NOTE:  len<0 has already been checked for.  Pedantic */
582         if (len <= 0 || buf == NULL)
583         {
584                 /* There is no command data -- only a size / command data */
585                 num_vectors = 1;
586         }
587         else
588         {
589                 /*
590                 **  There is both size / command and command data.
591                 **  Set up the vector for the command data.
592                 */
593
594                 num_vectors = 2;
595                 sl += len;
596                 vector[1].iov_base = (void *) buf;
597                 vector[1].iov_len  = len;
598
599                 if (tTd(64, 50))
600                         sm_dprintf("milter_write(%s): Sending %*s\n",
601                                    m->mf_name, (int) len, buf);
602         }
603
604         if (to > 0)
605                 MILTER_TIMEOUT("write", to, true, started, where);
606
607         /* write the vector(s) */
608         i = writev(m->mf_sock, vector, num_vectors);
609         if (i != sl)
610         {
611                 int save_errno = errno;
612
613                 if (tTd(64, 5))
614                         sm_dprintf("milter_write(%s): write(%c) returned %ld, expected %ld: %s\n",
615                                    m->mf_name, command, (long) i, (long) sl,
616                                    sm_errstring(save_errno));
617                 if (MilterLogLevel > 0)
618                         sm_syslog(LOG_ERR, e->e_id,
619                                   "Milter (%s): write(%c) returned %ld, expected %ld: %s",
620                                   m->mf_name, command, (long) i, (long) sl,
621                                   sm_errstring(save_errno));
622                 milter_error(m, e);
623                 return NULL;
624         }
625         return buf;
626 }
627
628 /*
629 **  Utility functions
630 */
631
632 /*
633 **  MILTER_OPEN -- connect to remote milter filter
634 **
635 **      Parameters:
636 **              m -- milter to connect to.
637 **              parseonly -- parse but don't connect.
638 **              e -- current envelope.
639 **
640 **      Returns:
641 **              connected socket if successful && !parseonly,
642 **              0 upon parse success if parseonly,
643 **              -1 otherwise.
644 */
645
646 static jmp_buf  MilterConnectTimeout;
647
648 static int
649 milter_open(m, parseonly, e)
650         struct milter *m;
651         bool parseonly;
652         ENVELOPE *e;
653 {
654         int sock = 0;
655         SOCKADDR_LEN_T addrlen = 0;
656         int addrno = 0;
657         int save_errno;
658         char *p;
659         char *colon;
660         char *at;
661         struct hostent *hp = NULL;
662         SOCKADDR addr;
663
664         if (m->mf_conn == NULL || m->mf_conn[0] == '\0')
665         {
666                 if (tTd(64, 5))
667                         sm_dprintf("X%s: empty or missing socket information\n",
668                                    m->mf_name);
669                 if (parseonly)
670                         syserr("X%s: empty or missing socket information",
671                                m->mf_name);
672                 else if (MilterLogLevel > 0)
673                         sm_syslog(LOG_ERR, e->e_id,
674                                   "Milter (%s): empty or missing socket information",
675                                   m->mf_name);
676                 milter_error(m, e);
677                 return -1;
678         }
679
680         /* protocol:filename or protocol:port@host */
681         memset(&addr, '\0', sizeof(addr));
682         p = m->mf_conn;
683         colon = strchr(p, ':');
684         if (colon != NULL)
685         {
686                 *colon = '\0';
687
688                 if (*p == '\0')
689                 {
690 # if NETUNIX
691                         /* default to AF_UNIX */
692                         addr.sa.sa_family = AF_UNIX;
693 # else /* NETUNIX */
694 #  if NETINET
695                         /* default to AF_INET */
696                         addr.sa.sa_family = AF_INET;
697 #  else /* NETINET */
698 #   if NETINET6
699                         /* default to AF_INET6 */
700                         addr.sa.sa_family = AF_INET6;
701 #   else /* NETINET6 */
702                         /* no protocols available */
703                         if (MilterLogLevel > 0)
704                                 sm_syslog(LOG_ERR, e->e_id,
705                                           "Milter (%s): no valid socket protocols available",
706                                           m->mf_name);
707                         milter_error(m, e);
708                         return -1;
709 #   endif /* NETINET6 */
710 #  endif /* NETINET */
711 # endif /* NETUNIX */
712                 }
713 # if NETUNIX
714                 else if (sm_strcasecmp(p, "unix") == 0 ||
715                          sm_strcasecmp(p, "local") == 0)
716                         addr.sa.sa_family = AF_UNIX;
717 # endif /* NETUNIX */
718 # if NETINET
719                 else if (sm_strcasecmp(p, "inet") == 0)
720                         addr.sa.sa_family = AF_INET;
721 # endif /* NETINET */
722 # if NETINET6
723                 else if (sm_strcasecmp(p, "inet6") == 0)
724                         addr.sa.sa_family = AF_INET6;
725 # endif /* NETINET6 */
726                 else
727                 {
728 # ifdef EPROTONOSUPPORT
729                         errno = EPROTONOSUPPORT;
730 # else /* EPROTONOSUPPORT */
731                         errno = EINVAL;
732 # endif /* EPROTONOSUPPORT */
733                         if (tTd(64, 5))
734                                 sm_dprintf("X%s: unknown socket type %s\n",
735                                         m->mf_name, p);
736                         if (parseonly)
737                                 syserr("X%s: unknown socket type %s",
738                                        m->mf_name, p);
739                         else if (MilterLogLevel > 0)
740                                 sm_syslog(LOG_ERR, e->e_id,
741                                           "Milter (%s): unknown socket type %s",
742                                           m->mf_name, p);
743                         milter_error(m, e);
744                         return -1;
745                 }
746                 *colon++ = ':';
747         }
748         else
749         {
750                 /* default to AF_UNIX */
751                 addr.sa.sa_family = AF_UNIX;
752                 colon = p;
753         }
754
755 # if NETUNIX
756         if (addr.sa.sa_family == AF_UNIX)
757         {
758                 long sff = SFF_SAFEDIRPATH|SFF_OPENASROOT|SFF_NOLINK|SFF_EXECOK;
759
760                 at = colon;
761                 if (strlen(colon) >= sizeof(addr.sunix.sun_path))
762                 {
763                         if (tTd(64, 5))
764                                 sm_dprintf("X%s: local socket name %s too long\n",
765                                         m->mf_name, colon);
766                         errno = EINVAL;
767                         if (parseonly)
768                                 syserr("X%s: local socket name %s too long",
769                                        m->mf_name, colon);
770                         else if (MilterLogLevel > 0)
771                                 sm_syslog(LOG_ERR, e->e_id,
772                                           "Milter (%s): local socket name %s too long",
773                                           m->mf_name, colon);
774                         milter_error(m, e);
775                         return -1;
776                 }
777                 errno = safefile(colon, RunAsUid, RunAsGid, RunAsUserName, sff,
778                                  S_IRUSR|S_IWUSR, NULL);
779
780                 /* if just parsing .cf file, socket doesn't need to exist */
781                 if (parseonly && errno == ENOENT)
782                 {
783                         if (OpMode == MD_DAEMON ||
784                             OpMode == MD_FGDAEMON)
785                                 (void) sm_io_fprintf(smioerr, SM_TIME_DEFAULT,
786                                                      "WARNING: X%s: local socket name %s missing\n",
787                                                      m->mf_name, colon);
788                 }
789                 else if (errno != 0)
790                 {
791                         /* if not safe, don't create */
792                         save_errno = errno;
793                         if (tTd(64, 5))
794                                 sm_dprintf("X%s: local socket name %s unsafe\n",
795                                         m->mf_name, colon);
796                         errno = save_errno;
797                         if (parseonly)
798                         {
799                                 if (OpMode == MD_DAEMON ||
800                                     OpMode == MD_FGDAEMON ||
801                                     OpMode == MD_SMTP)
802                                         syserr("X%s: local socket name %s unsafe",
803                                                m->mf_name, colon);
804                         }
805                         else if (MilterLogLevel > 0)
806                                 sm_syslog(LOG_ERR, e->e_id,
807                                           "Milter (%s): local socket name %s unsafe",
808                                           m->mf_name, colon);
809                         milter_error(m, e);
810                         return -1;
811                 }
812
813                 (void) sm_strlcpy(addr.sunix.sun_path, colon,
814                                sizeof(addr.sunix.sun_path));
815                 addrlen = sizeof(struct sockaddr_un);
816         }
817         else
818 # endif /* NETUNIX */
819 # if NETINET || NETINET6
820         if (false
821 #  if NETINET
822                  || addr.sa.sa_family == AF_INET
823 #  endif /* NETINET */
824 #  if NETINET6
825                  || addr.sa.sa_family == AF_INET6
826 #  endif /* NETINET6 */
827                  )
828         {
829                 unsigned short port;
830
831                 /* Parse port@host */
832                 at = strchr(colon, '@');
833                 if (at == NULL)
834                 {
835                         if (tTd(64, 5))
836                                 sm_dprintf("X%s: bad address %s (expected port@host)\n",
837                                         m->mf_name, colon);
838                         if (parseonly)
839                                 syserr("X%s: bad address %s (expected port@host)",
840                                        m->mf_name, colon);
841                         else if (MilterLogLevel > 0)
842                                 sm_syslog(LOG_ERR, e->e_id,
843                                           "Milter (%s): bad address %s (expected port@host)",
844                                           m->mf_name, colon);
845                         milter_error(m, e);
846                         return -1;
847                 }
848                 *at = '\0';
849                 if (isascii(*colon) && isdigit(*colon))
850                         port = htons((unsigned short) atoi(colon));
851                 else
852                 {
853 #  ifdef NO_GETSERVBYNAME
854                         if (tTd(64, 5))
855                                 sm_dprintf("X%s: invalid port number %s\n",
856                                         m->mf_name, colon);
857                         if (parseonly)
858                                 syserr("X%s: invalid port number %s",
859                                        m->mf_name, colon);
860                         else if (MilterLogLevel > 0)
861                                 sm_syslog(LOG_ERR, e->e_id,
862                                           "Milter (%s): invalid port number %s",
863                                           m->mf_name, colon);
864                         milter_error(m, e);
865                         return -1;
866 #  else /* NO_GETSERVBYNAME */
867                         struct servent *sp;
868
869                         sp = getservbyname(colon, "tcp");
870                         if (sp == NULL)
871                         {
872                                 save_errno = errno;
873                                 if (tTd(64, 5))
874                                         sm_dprintf("X%s: unknown port name %s\n",
875                                                 m->mf_name, colon);
876                                 errno = save_errno;
877                                 if (parseonly)
878                                         syserr("X%s: unknown port name %s",
879                                                m->mf_name, colon);
880                                 else if (MilterLogLevel > 0)
881                                         sm_syslog(LOG_ERR, e->e_id,
882                                                   "Milter (%s): unknown port name %s",
883                                                   m->mf_name, colon);
884                                 milter_error(m, e);
885                                 return -1;
886                         }
887                         port = sp->s_port;
888 #  endif /* NO_GETSERVBYNAME */
889                 }
890                 *at++ = '@';
891                 if (*at == '[')
892                 {
893                         char *end;
894
895                         end = strchr(at, ']');
896                         if (end != NULL)
897                         {
898                                 bool found = false;
899 #  if NETINET
900                                 unsigned long hid = INADDR_NONE;
901 #  endif /* NETINET */
902 #  if NETINET6
903                                 struct sockaddr_in6 hid6;
904 #  endif /* NETINET6 */
905
906                                 *end = '\0';
907 #  if NETINET
908                                 if (addr.sa.sa_family == AF_INET &&
909                                     (hid = inet_addr(&at[1])) != INADDR_NONE)
910                                 {
911                                         addr.sin.sin_addr.s_addr = hid;
912                                         addr.sin.sin_port = port;
913                                         found = true;
914                                 }
915 #  endif /* NETINET */
916 #  if NETINET6
917                                 (void) memset(&hid6, '\0', sizeof(hid6));
918                                 if (addr.sa.sa_family == AF_INET6 &&
919                                     anynet_pton(AF_INET6, &at[1],
920                                                 &hid6.sin6_addr) == 1)
921                                 {
922                                         addr.sin6.sin6_addr = hid6.sin6_addr;
923                                         addr.sin6.sin6_port = port;
924                                         found = true;
925                                 }
926 #  endif /* NETINET6 */
927                                 *end = ']';
928                                 if (!found)
929                                 {
930                                         if (tTd(64, 5))
931                                                 sm_dprintf("X%s: Invalid numeric domain spec \"%s\"\n",
932                                                         m->mf_name, at);
933                                         if (parseonly)
934                                                 syserr("X%s: Invalid numeric domain spec \"%s\"",
935                                                        m->mf_name, at);
936                                         else if (MilterLogLevel > 0)
937                                                 sm_syslog(LOG_ERR, e->e_id,
938                                                           "Milter (%s): Invalid numeric domain spec \"%s\"",
939                                                           m->mf_name, at);
940                                         milter_error(m, e);
941                                         return -1;
942                                 }
943                         }
944                         else
945                         {
946                                 if (tTd(64, 5))
947                                         sm_dprintf("X%s: Invalid numeric domain spec \"%s\"\n",
948                                                 m->mf_name, at);
949                                 if (parseonly)
950                                         syserr("X%s: Invalid numeric domain spec \"%s\"",
951                                                m->mf_name, at);
952                                 else if (MilterLogLevel > 0)
953                                         sm_syslog(LOG_ERR, e->e_id,
954                                                   "Milter (%s): Invalid numeric domain spec \"%s\"",
955                                                   m->mf_name, at);
956                                 milter_error(m, e);
957                                 return -1;
958                         }
959                 }
960                 else
961                 {
962                         hp = sm_gethostbyname(at, addr.sa.sa_family);
963                         if (hp == NULL)
964                         {
965                                 save_errno = errno;
966                                 if (tTd(64, 5))
967                                         sm_dprintf("X%s: Unknown host name %s\n",
968                                                    m->mf_name, at);
969                                 errno = save_errno;
970                                 if (parseonly)
971                                         syserr("X%s: Unknown host name %s",
972                                                m->mf_name, at);
973                                 else if (MilterLogLevel > 0)
974                                         sm_syslog(LOG_ERR, e->e_id,
975                                                   "Milter (%s): Unknown host name %s",
976                                                   m->mf_name, at);
977                                 milter_error(m, e);
978                                 return -1;
979                         }
980                         addr.sa.sa_family = hp->h_addrtype;
981                         switch (hp->h_addrtype)
982                         {
983 #  if NETINET
984                           case AF_INET:
985                                 memmove(&addr.sin.sin_addr,
986                                         hp->h_addr, INADDRSZ);
987                                 addr.sin.sin_port = port;
988                                 addrlen = sizeof(struct sockaddr_in);
989                                 addrno = 1;
990                                 break;
991 #  endif /* NETINET */
992
993 #  if NETINET6
994                           case AF_INET6:
995                                 memmove(&addr.sin6.sin6_addr,
996                                         hp->h_addr, IN6ADDRSZ);
997                                 addr.sin6.sin6_port = port;
998                                 addrlen = sizeof(struct sockaddr_in6);
999                                 addrno = 1;
1000                                 break;
1001 #  endif /* NETINET6 */
1002
1003                           default:
1004                                 if (tTd(64, 5))
1005                                         sm_dprintf("X%s: Unknown protocol for %s (%d)\n",
1006                                                    m->mf_name, at,
1007                                                    hp->h_addrtype);
1008                                 if (parseonly)
1009                                         syserr("X%s: Unknown protocol for %s (%d)",
1010                                                m->mf_name, at, hp->h_addrtype);
1011                                 else if (MilterLogLevel > 0)
1012                                         sm_syslog(LOG_ERR, e->e_id,
1013                                                   "Milter (%s): Unknown protocol for %s (%d)",
1014                                                   m->mf_name, at,
1015                                                   hp->h_addrtype);
1016                                 milter_error(m, e);
1017 #  if NETINET6
1018                                 freehostent(hp);
1019 #  endif /* NETINET6 */
1020                                 return -1;
1021                         }
1022                 }
1023         }
1024         else
1025 # endif /* NETINET || NETINET6 */
1026         {
1027                 if (tTd(64, 5))
1028                         sm_dprintf("X%s: unknown socket protocol\n",
1029                                    m->mf_name);
1030                 if (parseonly)
1031                         syserr("X%s: unknown socket protocol", m->mf_name);
1032                 else if (MilterLogLevel > 0)
1033                         sm_syslog(LOG_ERR, e->e_id,
1034                                   "Milter (%s): unknown socket protocol",
1035                                   m->mf_name);
1036                 milter_error(m, e);
1037                 return -1;
1038         }
1039
1040         /* just parsing through? */
1041         if (parseonly)
1042         {
1043                 m->mf_state = SMFS_READY;
1044 # if NETINET6
1045                 if (hp != NULL)
1046                         freehostent(hp);
1047 # endif /* NETINET6 */
1048                 return 0;
1049         }
1050
1051         /* sanity check */
1052         if (m->mf_state != SMFS_READY &&
1053             m->mf_state != SMFS_CLOSED)
1054         {
1055                 /* shouldn't happen */
1056                 if (tTd(64, 1))
1057                         sm_dprintf("Milter (%s): Trying to open filter in state %c\n",
1058                                    m->mf_name, (char) m->mf_state);
1059                 milter_error(m, e);
1060 # if NETINET6
1061                 if (hp != NULL)
1062                         freehostent(hp);
1063 # endif /* NETINET6 */
1064                 return -1;
1065         }
1066
1067         /* nope, actually connecting */
1068         for (;;)
1069         {
1070                 sock = socket(addr.sa.sa_family, SOCK_STREAM, 0);
1071                 if (sock < 0)
1072                 {
1073                         save_errno = errno;
1074                         if (tTd(64, 5))
1075                                 sm_dprintf("Milter (%s): error creating socket: %s\n",
1076                                            m->mf_name,
1077                                            sm_errstring(save_errno));
1078                         if (MilterLogLevel > 0)
1079                                 sm_syslog(LOG_ERR, e->e_id,
1080                                           "Milter (%s): error creating socket: %s",
1081                                           m->mf_name, sm_errstring(save_errno));
1082                         milter_error(m, e);
1083 # if NETINET6
1084                         if (hp != NULL)
1085                                 freehostent(hp);
1086 # endif /* NETINET6 */
1087                         return -1;
1088                 }
1089
1090                 if (setjmp(MilterConnectTimeout) == 0)
1091                 {
1092                         SM_EVENT *ev = NULL;
1093                         int i;
1094
1095                         if (m->mf_timeout[SMFTO_CONNECT] > 0)
1096                                 ev = sm_setevent(m->mf_timeout[SMFTO_CONNECT],
1097                                                  milter_connect_timeout, 0);
1098
1099                         i = connect(sock, (struct sockaddr *) &addr, addrlen);
1100                         save_errno = errno;
1101                         if (ev != NULL)
1102                                 sm_clrevent(ev);
1103                         errno = save_errno;
1104                         if (i >= 0)
1105                                 break;
1106                 }
1107
1108                 /* couldn't connect.... try next address */
1109                 save_errno = errno;
1110                 p = CurHostName;
1111                 CurHostName = at;
1112                 if (tTd(64, 5))
1113                         sm_dprintf("milter_open (%s): open %s failed: %s\n",
1114                                    m->mf_name, at, sm_errstring(save_errno));
1115                 if (MilterLogLevel > 13)
1116                         sm_syslog(LOG_INFO, e->e_id,
1117                                   "Milter (%s): open %s failed: %s",
1118                                   m->mf_name, at, sm_errstring(save_errno));
1119                 CurHostName = p;
1120                 (void) close(sock);
1121
1122                 /* try next address */
1123                 if (hp != NULL && hp->h_addr_list[addrno] != NULL)
1124                 {
1125                         switch (addr.sa.sa_family)
1126                         {
1127 # if NETINET
1128                           case AF_INET:
1129                                 memmove(&addr.sin.sin_addr,
1130                                         hp->h_addr_list[addrno++],
1131                                         INADDRSZ);
1132                                 break;
1133 # endif /* NETINET */
1134
1135 # if NETINET6
1136                           case AF_INET6:
1137                                 memmove(&addr.sin6.sin6_addr,
1138                                         hp->h_addr_list[addrno++],
1139                                         IN6ADDRSZ);
1140                                 break;
1141 # endif /* NETINET6 */
1142
1143                           default:
1144                                 if (tTd(64, 5))
1145                                         sm_dprintf("X%s: Unknown protocol for %s (%d)\n",
1146                                                    m->mf_name, at,
1147                                                    hp->h_addrtype);
1148                                 if (MilterLogLevel > 0)
1149                                         sm_syslog(LOG_ERR, e->e_id,
1150                                                   "Milter (%s): Unknown protocol for %s (%d)",
1151                                                   m->mf_name, at,
1152                                                   hp->h_addrtype);
1153                                 milter_error(m, e);
1154 # if NETINET6
1155                                 freehostent(hp);
1156 # endif /* NETINET6 */
1157                                 return -1;
1158                         }
1159                         continue;
1160                 }
1161                 p = CurHostName;
1162                 CurHostName = at;
1163                 if (tTd(64, 5))
1164                         sm_dprintf("X%s: error connecting to filter: %s\n",
1165                                    m->mf_name, sm_errstring(save_errno));
1166                 if (MilterLogLevel > 0)
1167                         sm_syslog(LOG_ERR, e->e_id,
1168                                   "Milter (%s): error connecting to filter: %s",
1169                                   m->mf_name, sm_errstring(save_errno));
1170                 CurHostName = p;
1171                 milter_error(m, e);
1172 # if NETINET6
1173                 if (hp != NULL)
1174                         freehostent(hp);
1175 # endif /* NETINET6 */
1176                 return -1;
1177         }
1178         m->mf_state = SMFS_OPEN;
1179 # if NETINET6
1180         if (hp != NULL)
1181         {
1182                 freehostent(hp);
1183                 hp = NULL;
1184         }
1185 # endif /* NETINET6 */
1186 # if MILTER_NO_NAGLE && !defined(TCP_CORK)
1187         {
1188                 int nodelay = 1;
1189
1190                 setsockopt(m->mf_sock, IPPROTO_TCP, TCP_NODELAY,
1191                            (char *)&nodelay, sizeof(nodelay));
1192         }
1193 # endif /* MILTER_NO_NAGLE && !defined(TCP_CORK) */
1194         return sock;
1195 }
1196
1197 static void
1198 milter_connect_timeout(ignore)
1199         int ignore;
1200 {
1201         /*
1202         **  NOTE: THIS CAN BE CALLED FROM A SIGNAL HANDLER.  DO NOT ADD
1203         **      ANYTHING TO THIS ROUTINE UNLESS YOU KNOW WHAT YOU ARE
1204         **      DOING.
1205         */
1206
1207         errno = ETIMEDOUT;
1208         longjmp(MilterConnectTimeout, 1);
1209 }
1210
1211 /*
1212 **  MILTER_SETUP -- setup structure for a mail filter
1213 **
1214 **      Parameters:
1215 **              line -- the options line.
1216 **
1217 **      Returns:
1218 **              none
1219 */
1220
1221 void
1222 milter_setup(line)
1223         char *line;
1224 {
1225         char fcode;
1226         char *p;
1227         struct milter *m;
1228         STAB *s;
1229         static int idx = 0;
1230
1231         /* collect the filter name */
1232         for (p = line;
1233              *p != '\0' && *p != ',' && !(isascii(*p) && isspace(*p));
1234              p++)
1235                 continue;
1236         if (*p != '\0')
1237                 *p++ = '\0';
1238         if (line[0] == '\0')
1239         {
1240                 syserr("name required for mail filter");
1241                 return;
1242         }
1243         m = (struct milter *) xalloc(sizeof(*m));
1244         memset((char *) m, '\0', sizeof(*m));
1245         m->mf_name = newstr(line);
1246         m->mf_state = SMFS_READY;
1247         m->mf_sock = -1;
1248         m->mf_timeout[SMFTO_CONNECT] = (time_t) 300;
1249         m->mf_timeout[SMFTO_WRITE] = (time_t) 10;
1250         m->mf_timeout[SMFTO_READ] = (time_t) 10;
1251         m->mf_timeout[SMFTO_EOM] = (time_t) 300;
1252 #if _FFR_MILTER_CHECK
1253         m->mf_mta_prot_version = SMFI_PROT_VERSION;
1254         m->mf_mta_prot_flags = SMFI_CURR_PROT;
1255         m->mf_mta_actions = SMFI_CURR_ACTS;
1256 #endif /* _FFR_MILTER_CHECK */
1257
1258         /* now scan through and assign info from the fields */
1259         while (*p != '\0')
1260         {
1261                 char *delimptr;
1262
1263                 while (*p != '\0' &&
1264                        (*p == ',' || (isascii(*p) && isspace(*p))))
1265                         p++;
1266
1267                 /* p now points to field code */
1268                 fcode = *p;
1269                 while (*p != '\0' && *p != '=' && *p != ',')
1270                         p++;
1271                 if (*p++ != '=')
1272                 {
1273                         syserr("X%s: `=' expected", m->mf_name);
1274                         return;
1275                 }
1276                 while (isascii(*p) && isspace(*p))
1277                         p++;
1278
1279                 /* p now points to the field body */
1280                 p = munchstring(p, &delimptr, ',');
1281
1282                 /* install the field into the filter struct */
1283                 switch (fcode)
1284                 {
1285                   case 'S':             /* socket */
1286                         if (p == NULL)
1287                                 m->mf_conn = NULL;
1288                         else
1289                                 m->mf_conn = newstr(p);
1290                         break;
1291
1292                   case 'F':             /* Milter flags configured on MTA */
1293                         for (; *p != '\0'; p++)
1294                         {
1295                                 if (!(isascii(*p) && isspace(*p)))
1296                                         setbitn(bitidx(*p), m->mf_flags);
1297                         }
1298                         break;
1299
1300                   case 'T':             /* timeouts */
1301                         milter_parse_timeouts(p, m);
1302                         break;
1303
1304 #if _FFR_MILTER_CHECK
1305                   case 'a':
1306                         m->mf_mta_actions = strtoul(p, NULL, 0);
1307                         break;
1308                   case 'f':
1309                         m->mf_mta_prot_flags = strtoul(p, NULL, 0);
1310                         break;
1311                   case 'v':
1312                         m->mf_mta_prot_version = strtoul(p, NULL, 0);
1313                         break;
1314 #endif /* _FFR_MILTER_CHECK */
1315
1316                   default:
1317                         syserr("X%s: unknown filter equate %c=",
1318                                m->mf_name, fcode);
1319                         break;
1320                 }
1321                 p = delimptr;
1322         }
1323
1324         /* early check for errors */
1325         (void) milter_open(m, true, CurEnv);
1326
1327         /* enter the filter into the symbol table */
1328         s = stab(m->mf_name, ST_MILTER, ST_ENTER);
1329         if (s->s_milter != NULL)
1330                 syserr("X%s: duplicate filter definition", m->mf_name);
1331         else
1332         {
1333                 s->s_milter = m;
1334                 m->mf_idx = ++idx;
1335         }
1336 }
1337
1338 /*
1339 **  MILTER_CONFIG -- parse option list into an array and check config
1340 **
1341 **      Called when reading configuration file.
1342 **
1343 **      Parameters:
1344 **              spec -- the filter list.
1345 **              list -- the array to fill in.
1346 **              max -- the maximum number of entries in list.
1347 **
1348 **      Returns:
1349 **              none
1350 */
1351
1352 void
1353 milter_config(spec, list, max)
1354         char *spec;
1355         struct milter **list;
1356         int max;
1357 {
1358         int numitems = 0;
1359         char *p;
1360
1361         /* leave one for the NULL signifying the end of the list */
1362         max--;
1363
1364         for (p = spec; p != NULL; )
1365         {
1366                 STAB *s;
1367
1368                 while (isascii(*p) && isspace(*p))
1369                         p++;
1370                 if (*p == '\0')
1371                         break;
1372                 spec = p;
1373
1374                 if (numitems >= max)
1375                 {
1376                         syserr("Too many filters defined, %d max", max);
1377                         if (max > 0)
1378                                 list[0] = NULL;
1379                         return;
1380                 }
1381                 p = strpbrk(p, ";,");
1382                 if (p != NULL)
1383                         *p++ = '\0';
1384
1385                 s = stab(spec, ST_MILTER, ST_FIND);
1386                 if (s == NULL)
1387                 {
1388                         syserr("InputFilter %s not defined", spec);
1389                         ExitStat = EX_CONFIG;
1390                         return;
1391                 }
1392                 list[numitems++] = s->s_milter;
1393         }
1394         list[numitems] = NULL;
1395
1396         /* if not set, set to LogLevel */
1397         if (MilterLogLevel == -1)
1398                 MilterLogLevel = LogLevel;
1399 }
1400
1401 /*
1402 **  MILTER_PARSE_TIMEOUTS -- parse timeout list
1403 **
1404 **      Called when reading configuration file.
1405 **
1406 **      Parameters:
1407 **              spec -- the timeout list.
1408 **              m -- milter to set.
1409 **
1410 **      Returns:
1411 **              none
1412 */
1413
1414 static void
1415 milter_parse_timeouts(spec, m)
1416         char *spec;
1417         struct milter *m;
1418 {
1419         char fcode;
1420         int tcode;
1421         char *p;
1422
1423         p = spec;
1424
1425         /* now scan through and assign info from the fields */
1426         while (*p != '\0')
1427         {
1428                 char *delimptr;
1429
1430                 while (*p != '\0' &&
1431                        (*p == ';' || (isascii(*p) && isspace(*p))))
1432                         p++;
1433
1434                 /* p now points to field code */
1435                 fcode = *p;
1436                 while (*p != '\0' && *p != ':')
1437                         p++;
1438                 if (*p++ != ':')
1439                 {
1440                         syserr("X%s, T=: `:' expected", m->mf_name);
1441                         return;
1442                 }
1443                 while (isascii(*p) && isspace(*p))
1444                         p++;
1445
1446                 /* p now points to the field body */
1447                 p = munchstring(p, &delimptr, ';');
1448                 tcode = -1;
1449
1450                 /* install the field into the filter struct */
1451                 switch (fcode)
1452                 {
1453                   case 'C':
1454                         tcode = SMFTO_CONNECT;
1455                         break;
1456
1457                   case 'S':
1458                         tcode = SMFTO_WRITE;
1459                         break;
1460
1461                   case 'R':
1462                         tcode = SMFTO_READ;
1463                         break;
1464
1465                   case 'E':
1466                         tcode = SMFTO_EOM;
1467                         break;
1468
1469                   default:
1470                         if (tTd(64, 5))
1471                                 sm_dprintf("X%s: %c unknown\n",
1472                                            m->mf_name, fcode);
1473                         syserr("X%s: unknown filter timeout %c",
1474                                m->mf_name, fcode);
1475                         break;
1476                 }
1477                 if (tcode >= 0)
1478                 {
1479                         m->mf_timeout[tcode] = convtime(p, 's');
1480                         if (tTd(64, 5))
1481                                 sm_dprintf("X%s: %c=%ld\n",
1482                                            m->mf_name, fcode,
1483                                            (u_long) m->mf_timeout[tcode]);
1484                 }
1485                 p = delimptr;
1486         }
1487 }
1488
1489 /*
1490 **  MILTER_SET_MACROS -- set milter macros
1491 **
1492 **      Parameters:
1493 **              name -- name of milter.
1494 **              macros -- where to store macros.
1495 **              val -- the value of the option.
1496 **              nummac -- current number of macros
1497 **
1498 **      Returns:
1499 **              new number of macros
1500 */
1501
1502 static int
1503 milter_set_macros(name, macros, val, nummac)
1504         char *name;
1505         char **macros;
1506         char *val;
1507         int nummac;
1508 {
1509         char *p;
1510
1511         p = newstr(val);
1512         while (*p != '\0')
1513         {
1514                 char *macro;
1515
1516                 /* Skip leading commas, spaces */
1517                 while (*p != '\0' &&
1518                        (*p == ',' || (isascii(*p) && isspace(*p))))
1519                         p++;
1520
1521                 if (*p == '\0')
1522                         break;
1523
1524                 /* Find end of macro */
1525                 macro = p;
1526                 while (*p != '\0' && *p != ',' &&
1527                        isascii(*p) && !isspace(*p))
1528                         p++;
1529                 if (*p != '\0')
1530                         *p++ = '\0';
1531
1532                 if (nummac >= MAXFILTERMACROS)
1533                 {
1534                         syserr("milter_set_option: too many macros in Milter.%s (max %d)",
1535                                name, MAXFILTERMACROS);
1536                         macros[nummac] = NULL;
1537                         return -1;
1538                 }
1539                 macros[nummac++] = macro;
1540         }
1541         macros[nummac] = NULL;
1542         return nummac;
1543 }
1544
1545 /*
1546 **  MILTER_SET_OPTION -- set an individual milter option
1547 **
1548 **      Parameters:
1549 **              name -- the name of the option.
1550 **              val -- the value of the option.
1551 **              sticky -- if set, don't let other setoptions override
1552 **                      this value.
1553 **
1554 **      Returns:
1555 **              none.
1556 */
1557
1558 /* set if Milter sub-option is stuck */
1559 static BITMAP256        StickyMilterOpt;
1560
1561 static struct milteropt
1562 {
1563         char            *mo_name;       /* long name of milter option */
1564         unsigned char   mo_code;        /* code for option */
1565 } MilterOptTab[] =
1566 {
1567         { "macros.connect",             SMFIM_CONNECT                   },
1568         { "macros.helo",                SMFIM_HELO                      },
1569         { "macros.envfrom",             SMFIM_ENVFROM                   },
1570         { "macros.envrcpt",             SMFIM_ENVRCPT                   },
1571         { "macros.data",                SMFIM_DATA                      },
1572         { "macros.eom",                 SMFIM_EOM                       },
1573         { "macros.eoh",                 SMFIM_EOH                       },
1574
1575 # define MO_LOGLEVEL                    0x07
1576         { "loglevel",                   MO_LOGLEVEL                     },
1577 # if _FFR_MAXDATASIZE || _FFR_MDS_NEGOTIATE
1578 #  define MO_MAXDATASIZE                0x08
1579         { "maxdatasize",                MO_MAXDATASIZE                  },
1580 # endif /* _FFR_MAXDATASIZE || _FFR_MDS_NEGOTIATE */
1581         { NULL,                         (unsigned char)-1               },
1582 };
1583
1584 void
1585 milter_set_option(name, val, sticky)
1586         char *name;
1587         char *val;
1588         bool sticky;
1589 {
1590         int nummac, r;
1591         struct milteropt *mo;
1592         char **macros = NULL;
1593
1594         nummac = 0;
1595         if (tTd(37, 2) || tTd(64, 5))
1596                 sm_dprintf("milter_set_option(%s = %s)", name, val);
1597
1598         if (name == NULL)
1599         {
1600                 syserr("milter_set_option: invalid Milter option, must specify suboption");
1601                 return;
1602         }
1603
1604         for (mo = MilterOptTab; mo->mo_name != NULL; mo++)
1605         {
1606                 if (sm_strcasecmp(mo->mo_name, name) == 0)
1607                         break;
1608         }
1609
1610         if (mo->mo_name == NULL)
1611         {
1612                 syserr("milter_set_option: invalid Milter option %s", name);
1613                 return;
1614         }
1615
1616         /*
1617         **  See if this option is preset for us.
1618         */
1619
1620         if (!sticky && bitnset(mo->mo_code, StickyMilterOpt))
1621         {
1622                 if (tTd(37, 2) || tTd(64,5))
1623                         sm_dprintf(" (ignored)\n");
1624                 return;
1625         }
1626
1627         if (tTd(37, 2) || tTd(64,5))
1628                 sm_dprintf("\n");
1629
1630         switch (mo->mo_code)
1631         {
1632           case MO_LOGLEVEL:
1633                 MilterLogLevel = atoi(val);
1634                 break;
1635
1636 # if _FFR_MAXDATASIZE || _FFR_MDS_NEGOTIATE
1637           case MO_MAXDATASIZE:
1638 #  if _FFR_MDS_NEGOTIATE
1639                 MilterMaxDataSize = (size_t)atol(val);
1640                 if (MilterMaxDataSize != MILTER_MDS_64K &&
1641                     MilterMaxDataSize != MILTER_MDS_256K &&
1642                     MilterMaxDataSize != MILTER_MDS_1M)
1643                 {
1644                         sm_syslog(LOG_WARNING, NOQID,
1645                                 "WARNING: Milter.%s=%d, allowed are only %d, %d, and %d",
1646                                 name, MilterMaxDataSize,
1647                                 MILTER_MDS_64K, MILTER_MDS_256K,
1648                                 MILTER_MDS_1M);
1649                         if (MilterMaxDataSize < MILTER_MDS_64K)
1650                                 MilterMaxDataSize = MILTER_MDS_64K;
1651                         else if (MilterMaxDataSize < MILTER_MDS_256K)
1652                                 MilterMaxDataSize = MILTER_MDS_256K;
1653                         else
1654                                 MilterMaxDataSize = MILTER_MDS_1M;
1655                 }
1656 #  endif /* _FFR_MDS_NEGOTIATE */
1657                 break;
1658 # endif /* _FFR_MAXDATASIZE || _FFR_MDS_NEGOTIATE */
1659
1660           case SMFIM_CONNECT:
1661           case SMFIM_HELO:
1662           case SMFIM_ENVFROM:
1663           case SMFIM_ENVRCPT:
1664           case SMFIM_EOH:
1665           case SMFIM_EOM:
1666           case SMFIM_DATA:
1667                 macros = MilterMacros[mo->mo_code][0];
1668
1669                 r = milter_set_macros(name, macros, val, nummac);
1670                 if (r >= 0)
1671                         nummac = r;
1672                 break;
1673
1674           default:
1675                 syserr("milter_set_option: invalid Milter option %s", name);
1676                 break;
1677         }
1678         if (sticky)
1679                 setbitn(mo->mo_code, StickyMilterOpt);
1680 }
1681
1682 /*
1683 **  MILTER_REOPEN_DF -- open & truncate the data file (for replbody)
1684 **
1685 **      Parameters:
1686 **              e -- current envelope.
1687 **
1688 **      Returns:
1689 **              0 if succesful, -1 otherwise
1690 */
1691
1692 static int
1693 milter_reopen_df(e)
1694         ENVELOPE *e;
1695 {
1696         char dfname[MAXPATHLEN];
1697
1698         (void) sm_strlcpy(dfname, queuename(e, DATAFL_LETTER), sizeof(dfname));
1699
1700         /*
1701         **  In SuperSafe == SAFE_REALLY mode, e->e_dfp is a read-only FP so
1702         **  close and reopen writable (later close and reopen
1703         **  read only again).
1704         **
1705         **  In SuperSafe != SAFE_REALLY mode, e->e_dfp still points at the
1706         **  buffered file I/O descriptor, still open for writing so there
1707         **  isn't any work to do here (except checking for consistency).
1708         */
1709
1710         if (SuperSafe == SAFE_REALLY)
1711         {
1712                 /* close read-only data file */
1713                 if (bitset(EF_HAS_DF, e->e_flags) && e->e_dfp != NULL)
1714                 {
1715                         (void) sm_io_close(e->e_dfp, SM_TIME_DEFAULT);
1716                         e->e_flags &= ~EF_HAS_DF;
1717                 }
1718
1719                 /* open writable */
1720                 if ((e->e_dfp = sm_io_open(SmFtStdio, SM_TIME_DEFAULT, dfname,
1721                                            SM_IO_RDWR_B, NULL)) == NULL)
1722                 {
1723                         MILTER_DF_ERROR("milter_reopen_df: sm_io_open %s: %s");
1724                         return -1;
1725                 }
1726         }
1727         else if (e->e_dfp == NULL)
1728         {
1729                 /* shouldn't happen */
1730                 errno = ENOENT;
1731                 MILTER_DF_ERROR("milter_reopen_df: NULL e_dfp (%s: %s)");
1732                 return -1;
1733         }
1734         return 0;
1735 }
1736
1737 /*
1738 **  MILTER_RESET_DF -- re-open read-only the data file (for replbody)
1739 **
1740 **      Parameters:
1741 **              e -- current envelope.
1742 **
1743 **      Returns:
1744 **              0 if succesful, -1 otherwise
1745 */
1746
1747 static int
1748 milter_reset_df(e)
1749         ENVELOPE *e;
1750 {
1751         int afd;
1752         char dfname[MAXPATHLEN];
1753
1754         (void) sm_strlcpy(dfname, queuename(e, DATAFL_LETTER), sizeof(dfname));
1755
1756         if (sm_io_flush(e->e_dfp, SM_TIME_DEFAULT) != 0 ||
1757             sm_io_error(e->e_dfp))
1758         {
1759                 MILTER_DF_ERROR("milter_reset_df: error writing/flushing %s: %s");
1760                 return -1;
1761         }
1762         else if (SuperSafe != SAFE_REALLY)
1763         {
1764                 /* skip next few clauses */
1765                 /* EMPTY */
1766         }
1767         else if ((afd = sm_io_getinfo(e->e_dfp, SM_IO_WHAT_FD, NULL)) >= 0
1768                  && fsync(afd) < 0)
1769         {
1770                 MILTER_DF_ERROR("milter_reset_df: error sync'ing %s: %s");
1771                 return -1;
1772         }
1773         else if (sm_io_close(e->e_dfp, SM_TIME_DEFAULT) < 0)
1774         {
1775                 MILTER_DF_ERROR("milter_reset_df: error closing %s: %s");
1776                 return -1;
1777         }
1778         else if ((e->e_dfp = sm_io_open(SmFtStdio, SM_TIME_DEFAULT, dfname,
1779                                         SM_IO_RDONLY_B, NULL)) == NULL)
1780         {
1781                 MILTER_DF_ERROR("milter_reset_df: error reopening %s: %s");
1782                 return -1;
1783         }
1784         else
1785                 e->e_flags |= EF_HAS_DF;
1786         return 0;
1787 }
1788
1789 /*
1790 **  MILTER_QUIT_FILTER -- close down a single filter
1791 **
1792 **      Parameters:
1793 **              m -- milter structure of filter to close down.
1794 **              e -- current envelope.
1795 **
1796 **      Returns:
1797 **              none
1798 */
1799
1800 static void
1801 milter_quit_filter(m, e)
1802         struct milter *m;
1803         ENVELOPE *e;
1804 {
1805         if (tTd(64, 10))
1806                 sm_dprintf("milter_quit_filter(%s)\n", m->mf_name);
1807         if (MilterLogLevel > 18)
1808                 sm_syslog(LOG_INFO, e->e_id, "Milter (%s): quit filter",
1809                           m->mf_name);
1810
1811         /* Never replace error state */
1812         if (m->mf_state == SMFS_ERROR)
1813                 return;
1814
1815         if (m->mf_sock < 0 ||
1816             m->mf_state == SMFS_CLOSED ||
1817             m->mf_state == SMFS_READY)
1818         {
1819                 m->mf_sock = -1;
1820                 m->mf_state = SMFS_CLOSED;
1821                 return;
1822         }
1823
1824         (void) milter_write(m, SMFIC_QUIT, (char *) NULL, 0,
1825                             m->mf_timeout[SMFTO_WRITE], e, "quit_filter");
1826         if (m->mf_sock >= 0)
1827         {
1828                 (void) close(m->mf_sock);
1829                 m->mf_sock = -1;
1830         }
1831         if (m->mf_state != SMFS_ERROR)
1832                 m->mf_state = SMFS_CLOSED;
1833 }
1834
1835 /*
1836 **  MILTER_ABORT_FILTER -- tell filter to abort current message
1837 **
1838 **      Parameters:
1839 **              m -- milter structure of filter to abort.
1840 **              e -- current envelope.
1841 **
1842 **      Returns:
1843 **              none
1844 */
1845
1846 static void
1847 milter_abort_filter(m, e)
1848         struct milter *m;
1849         ENVELOPE *e;
1850 {
1851         if (tTd(64, 10))
1852                 sm_dprintf("milter_abort_filter(%s)\n", m->mf_name);
1853         if (MilterLogLevel > 10)
1854                 sm_syslog(LOG_INFO, e->e_id, "Milter (%s): abort filter",
1855                           m->mf_name);
1856
1857         if (m->mf_sock < 0 ||
1858             m->mf_state != SMFS_INMSG)
1859                 return;
1860
1861         (void) milter_write(m, SMFIC_ABORT, (char *) NULL, 0,
1862                             m->mf_timeout[SMFTO_WRITE], e, "abort_filter");
1863         if (m->mf_state != SMFS_ERROR)
1864                 m->mf_state = SMFS_DONE;
1865 }
1866
1867 /*
1868 **  MILTER_SEND_MACROS -- provide macros to the filters
1869 **
1870 **      Parameters:
1871 **              m -- milter to send macros to.
1872 **              macros -- macros to send for filter smfi_getsymval().
1873 **              cmd -- which command the macros are associated with.
1874 **              e -- current envelope (for macro access).
1875 **
1876 **      Returns:
1877 **              none
1878 */
1879
1880 static void
1881 milter_send_macros(m, macros, cmd, e)
1882         struct milter *m;
1883         char **macros;
1884         int cmd;
1885         ENVELOPE *e;
1886 {
1887         int i;
1888         int mid;
1889         char command = (char) cmd;
1890         char *v;
1891         char *buf, *bp;
1892         char exp[MAXLINE];
1893         ssize_t s;
1894
1895         /* sanity check */
1896         if (macros == NULL || macros[0] == NULL)
1897                 return;
1898
1899         /* put together data */
1900         s = 1;                  /* for the command character */
1901         for (i = 0; macros[i] != NULL; i++)
1902         {
1903                 mid = macid(macros[i]);
1904                 if (mid == 0)
1905                         continue;
1906                 v = macvalue(mid, e);
1907                 if (v == NULL)
1908                         continue;
1909                 expand(v, exp, sizeof(exp), e);
1910                 s += strlen(macros[i]) + 1 + strlen(exp) + 1;
1911         }
1912
1913         if (s < 0)
1914                 return;
1915
1916         buf = (char *) xalloc(s);
1917         bp = buf;
1918         *bp++ = command;
1919         for (i = 0; macros[i] != NULL; i++)
1920         {
1921                 mid = macid(macros[i]);
1922                 if (mid == 0)
1923                         continue;
1924                 v = macvalue(mid, e);
1925                 if (v == NULL)
1926                         continue;
1927                 expand(v, exp, sizeof(exp), e);
1928
1929                 if (tTd(64, 10))
1930                         sm_dprintf("milter_send_macros(%s, %c): %s=%s\n",
1931                                 m->mf_name, command, macros[i], exp);
1932
1933                 (void) sm_strlcpy(bp, macros[i], s - (bp - buf));
1934                 bp += strlen(bp) + 1;
1935                 (void) sm_strlcpy(bp, exp, s - (bp - buf));
1936                 bp += strlen(bp) + 1;
1937         }
1938         (void) milter_write(m, SMFIC_MACRO, buf, s,
1939                             m->mf_timeout[SMFTO_WRITE], e, "send_macros");
1940         sm_free(buf);
1941 }
1942
1943 /*
1944 **  MILTER_SEND_COMMAND -- send a command and return the response for a filter
1945 **
1946 **      Parameters:
1947 **              m -- current milter filter
1948 **              cmd -- command to send.
1949 **              data -- optional command data.
1950 **              sz -- length of buf.
1951 **              e -- current envelope (for e->e_id).
1952 **              state -- return state word.
1953 **
1954 **      Returns:
1955 **              response string (may be NULL)
1956 */
1957
1958 static char *
1959 milter_send_command(m, cmd, data, sz, e, state, where)
1960         struct milter *m;
1961         int cmd;
1962         void *data;
1963         ssize_t sz;
1964         ENVELOPE *e;
1965         char *state;
1966         const char *where;
1967 {
1968         char rcmd;
1969         ssize_t rlen;
1970         unsigned long skipflag;
1971         unsigned long norespflag = 0;
1972         char command = (char) cmd;
1973         char *action;
1974         char *defresponse;
1975         char *response;
1976
1977         if (tTd(64, 10))
1978                 sm_dprintf("milter_send_command(%s): cmd %c len %ld\n",
1979                         m->mf_name, (char) command, (long) sz);
1980
1981         /* find skip flag and default failure */
1982         switch (command)
1983         {
1984           case SMFIC_CONNECT:
1985                 skipflag = SMFIP_NOCONNECT;
1986                 norespflag = SMFIP_NR_CONN;
1987                 action = "connect";
1988                 defresponse = "554 Command rejected";
1989                 break;
1990
1991           case SMFIC_HELO:
1992                 skipflag = SMFIP_NOHELO;
1993                 norespflag = SMFIP_NR_HELO;
1994                 action = "helo";
1995                 defresponse = "550 Command rejected";
1996                 break;
1997
1998           case SMFIC_MAIL:
1999                 skipflag = SMFIP_NOMAIL;
2000                 norespflag = SMFIP_NR_MAIL;
2001                 action = "mail";
2002                 defresponse = "550 5.7.1 Command rejected";
2003                 break;
2004
2005           case SMFIC_RCPT:
2006                 skipflag = SMFIP_NORCPT;
2007                 norespflag = SMFIP_NR_RCPT;
2008                 action = "rcpt";
2009                 defresponse = "550 5.7.1 Command rejected";
2010                 break;
2011
2012           case SMFIC_HEADER:
2013                 skipflag = SMFIP_NOHDRS;
2014                 norespflag = SMFIP_NR_HDR;
2015                 action = "header";
2016                 defresponse = "550 5.7.1 Command rejected";
2017                 break;
2018
2019           case SMFIC_BODY:
2020                 skipflag = SMFIP_NOBODY;
2021                 norespflag = SMFIP_NR_BODY;
2022                 action = "body";
2023                 defresponse = "554 5.7.1 Command rejected";
2024                 break;
2025
2026           case SMFIC_EOH:
2027                 skipflag = SMFIP_NOEOH;
2028                 norespflag = SMFIP_NR_EOH;
2029                 action = "eoh";
2030                 defresponse = "550 5.7.1 Command rejected";
2031                 break;
2032
2033           case SMFIC_UNKNOWN:
2034                 skipflag = SMFIP_NOUNKNOWN;
2035                 norespflag = SMFIP_NR_UNKN;
2036                 action = "unknown";
2037                 defresponse = "550 5.7.1 Command rejected";
2038                 break;
2039
2040           case SMFIC_DATA:
2041                 skipflag = SMFIP_NODATA;
2042                 norespflag = SMFIP_NR_DATA;
2043                 action = "data";
2044                 defresponse = "550 5.7.1 Command rejected";
2045                 break;
2046
2047           case SMFIC_BODYEOB:
2048           case SMFIC_OPTNEG:
2049           case SMFIC_MACRO:
2050           case SMFIC_ABORT:
2051           case SMFIC_QUIT:
2052                 /* NOTE: not handled by milter_send_command() */
2053                 /* FALLTHROUGH */
2054
2055           default:
2056                 skipflag = 0;
2057                 action = "default";
2058                 defresponse = "550 5.7.1 Command rejected";
2059                 break;
2060         }
2061
2062         if (tTd(64, 10))
2063                 sm_dprintf("milter_send_command(%s): skip=%lx, pflags=%x\n",
2064                         m->mf_name, skipflag, m->mf_pflags);
2065
2066         /* check if filter wants this command */
2067         if (skipflag != 0 && bitset(skipflag, m->mf_pflags))
2068                 return NULL;
2069
2070         /* send the command to the filter */
2071         (void) milter_write(m, command, data, sz,
2072                             m->mf_timeout[SMFTO_WRITE], e, where);
2073         if (m->mf_state == SMFS_ERROR)
2074         {
2075                 MILTER_CHECK_ERROR(false, return NULL);
2076                 return NULL;
2077         }
2078
2079         /* check if filter sends response to this command */
2080         if (norespflag != 0 && bitset(norespflag, m->mf_pflags))
2081                 return NULL;
2082
2083         /* get the response from the filter */
2084         response = milter_read(m, &rcmd, &rlen,
2085                                m->mf_timeout[SMFTO_READ], e, where);
2086         if (m->mf_state == SMFS_ERROR)
2087         {
2088                 MILTER_CHECK_ERROR(false, return NULL);
2089                 return NULL;
2090         }
2091
2092         if (tTd(64, 10))
2093                 sm_dprintf("milter_send_command(%s): returned %c\n",
2094                            m->mf_name, (char) rcmd);
2095
2096         switch (rcmd)
2097         {
2098           case SMFIR_REPLYCODE:
2099                 MILTER_CHECK_REPLYCODE(defresponse);
2100                 if (MilterLogLevel > 10)
2101                         sm_syslog(LOG_INFO, e->e_id,
2102                                   "milter=%s, action=%s, reject=%s",
2103                                   m->mf_name, action, response);
2104                 *state = rcmd;
2105                 break;
2106
2107           case SMFIR_REJECT:
2108                 if (MilterLogLevel > 10)
2109                         sm_syslog(LOG_INFO, e->e_id,
2110                                   "milter=%s, action=%s, reject",
2111                                   m->mf_name, action);
2112                 *state = rcmd;
2113                 break;
2114
2115           case SMFIR_DISCARD:
2116                 if (MilterLogLevel > 10)
2117                         sm_syslog(LOG_INFO, e->e_id,
2118                                   "milter=%s, action=%s, discard",
2119                                   m->mf_name, action);
2120                 *state = rcmd;
2121                 break;
2122
2123           case SMFIR_TEMPFAIL:
2124                 if (MilterLogLevel > 10)
2125                         sm_syslog(LOG_INFO, e->e_id,
2126                                   "milter=%s, action=%s, tempfail",
2127                                   m->mf_name, action);
2128                 *state = rcmd;
2129                 break;
2130
2131           case SMFIR_ACCEPT:
2132                 /* this filter is done with message/connection */
2133                 if (command == SMFIC_HELO ||
2134                     command == SMFIC_CONNECT)
2135                         m->mf_state = SMFS_CLOSABLE;
2136                 else
2137                         m->mf_state = SMFS_DONE;
2138                 if (MilterLogLevel > 10)
2139                         sm_syslog(LOG_INFO, e->e_id,
2140                                   "milter=%s, action=%s, accepted",
2141                                   m->mf_name, action);
2142                 break;
2143
2144           case SMFIR_CONTINUE:
2145                 /* if MAIL command is ok, filter is in message state */
2146                 if (command == SMFIC_MAIL)
2147                         m->mf_state = SMFS_INMSG;
2148                 if (MilterLogLevel > 12)
2149                         sm_syslog(LOG_INFO, e->e_id,
2150                                   "milter=%s, action=%s, continue",
2151                                   m->mf_name, action);
2152                 break;
2153
2154           case SMFIR_SKIP:
2155                 if (MilterLogLevel > 12)
2156                         sm_syslog(LOG_INFO, e->e_id,
2157                                   "milter=%s, action=%s, skip",
2158                                   m->mf_name, action);
2159                 m->mf_state = SMFS_SKIP;
2160                 break;
2161
2162           default:
2163                 /* Invalid response to command */
2164                 if (MilterLogLevel > 0)
2165                         sm_syslog(LOG_ERR, e->e_id,
2166                                   "milter_send_command(%s): action=%s returned bogus response %c",
2167                                   m->mf_name, action, rcmd);
2168                 milter_error(m, e); /* NO ERROR CHECK? */
2169                 break;
2170         }
2171
2172         if (*state != SMFIR_REPLYCODE && response != NULL)
2173         {
2174                 sm_free(response); /* XXX */
2175                 response = NULL;
2176         }
2177         return response;
2178 }
2179
2180 /*
2181 **  MILTER_COMMAND -- send a command and return the response for each filter
2182 **
2183 **      Parameters:
2184 **              cmd -- command to send.
2185 **              data -- optional command data.
2186 **              sz -- length of buf.
2187 **              stage -- index of macros to send for filter smfi_getsymval().
2188 **              e -- current envelope (for macro access).
2189 **              state -- return state word.
2190 **              where -- description of calling function (logging).
2191 **              cmd_error -- did the SMTP command cause an error?
2192 **
2193 **      Returns:
2194 **              response string (may be NULL)
2195 */
2196
2197 static char *
2198 milter_command(cmd, data, sz, stage, e, state, where, cmd_error)
2199         int cmd;
2200         void *data;
2201         ssize_t sz;
2202         int stage;
2203         ENVELOPE *e;
2204         char *state;
2205         const char *where;
2206         bool cmd_error;
2207 {
2208         int i;
2209         char command = (char) cmd;
2210         char *response = NULL;
2211         time_t tn = 0;
2212
2213         if (tTd(64, 10))
2214                 sm_dprintf("milter_command: cmd %c len %ld\n",
2215                         command, (long) sz);
2216
2217         *state = SMFIR_CONTINUE;
2218         for (i = 0; InputFilters[i] != NULL; i++)
2219         {
2220                 struct milter *m = InputFilters[i];
2221
2222                 /* previous problem? */
2223                 if (m->mf_state == SMFS_ERROR)
2224                 {
2225                         MILTER_CHECK_ERROR(false, continue);
2226                         break;
2227                 }
2228
2229                 /* sanity check */
2230                 if (m->mf_sock < 0 ||
2231                     (m->mf_state != SMFS_OPEN && m->mf_state != SMFS_INMSG))
2232                         continue;
2233
2234                 if (stage >= SMFIM_FIRST && stage <= SMFIM_LAST)
2235                 {
2236                         int idx;
2237                         char **macros;
2238
2239                         if ((m->mf_lflags & MI_LFLAGS_SYM(stage)) != 0)
2240                                 idx = m->mf_idx;
2241                         else
2242                                 idx = 0;
2243                         SM_ASSERT(idx >= 0 && idx <= MAXFILTERS);
2244                         macros = MilterMacros[stage][idx];
2245
2246                         /* send macros (regardless of whether we send cmd) */
2247                         if (macros != NULL && macros[0] != NULL)
2248                         {
2249                                 milter_send_macros(m, macros, command, e);
2250                                 if (m->mf_state == SMFS_ERROR)
2251                                 {
2252                                         MILTER_CHECK_ERROR(false, continue);
2253                                         break;
2254                                 }
2255                         }
2256                 }
2257
2258                 if (MilterLogLevel > 21)
2259                         tn = curtime();
2260
2261                 /*
2262                 **  send the command if
2263                 **      there is no error
2264                 **      or it's RCPT and the client asked for it:
2265                 **      !cmd_error ||
2266                 **      where == "rcpt" && m->mf_pflags & SMFIP_RCPT_REJ != 0
2267                 **  negate that condition and use continue
2268                 */
2269
2270                 if (cmd_error &&
2271                     (strcmp(where, "rcpt") != 0 ||
2272                      (m->mf_pflags & SMFIP_RCPT_REJ) == 0))
2273                         continue;
2274
2275                 response = milter_send_command(m, command, data, sz, e, state,
2276                                                 where);
2277
2278                 if (MilterLogLevel > 21)
2279                 {
2280                         /* log the time it took for the command per filter */
2281                         sm_syslog(LOG_INFO, e->e_id,
2282                                   "Milter (%s): time command (%c), %d",
2283                                   m->mf_name, command, (int) (tn - curtime()));
2284                 }
2285
2286                 if (*state != SMFIR_CONTINUE)
2287                         break;
2288         }
2289         return response;
2290 }
2291
2292 static int milter_getsymlist __P((struct milter *, char *, int, int));
2293
2294 static int
2295 milter_getsymlist(m, buf, rlen, offset)
2296         struct milter *m;
2297         char *buf;
2298         int rlen;
2299         int offset;
2300 {
2301         int i, r, nummac;
2302         mi_int32 v;
2303
2304         SM_ASSERT(m != NULL);
2305         SM_ASSERT(buf != NULL);
2306
2307         while (offset + MILTER_LEN_BYTES < rlen)
2308         {
2309                 size_t len;
2310                 char **macros;
2311
2312                 nummac = 0;
2313                 (void) memcpy((char *) &v, buf + offset, MILTER_LEN_BYTES);
2314                 i = ntohl(v);
2315                 if (i < SMFIM_FIRST || i > SMFIM_LAST)
2316                         return -1;
2317                 offset += MILTER_LEN_BYTES;
2318                 macros = NULL;
2319
2320                 switch (i)
2321                 {
2322                   case SMFIM_CONNECT:
2323                   case SMFIM_HELO:
2324                   case SMFIM_ENVFROM:
2325                   case SMFIM_ENVRCPT:
2326                   case SMFIM_EOH:
2327                   case SMFIM_EOM:
2328                   case SMFIM_DATA:
2329                         SM_ASSERT(m->mf_idx > 0 && m->mf_idx < MAXFILTERS);
2330                         macros = MilterMacros[i][m->mf_idx];
2331                         m->mf_lflags |= MI_LFLAGS_SYM(i);
2332                         len = strlen(buf + offset);
2333                         if (len > 0)
2334                         {
2335                                 r = milter_set_macros(m->mf_name, macros,
2336                                                 buf + offset, nummac);
2337                                 if (r >= 0)
2338                                         nummac = r;
2339                                 if (tTd(64, 5))
2340                                         sm_dprintf("milter_getsymlist(%s, %s)=%d\n",
2341                                                 m->mf_name, buf + offset, r);
2342                         }
2343                         break;
2344
2345                   default:
2346                         return -1;
2347                 }
2348                 if (len == 0)
2349                         return -1;
2350                 offset += len + 1;
2351         }
2352
2353         return 0;
2354 }
2355
2356 /*
2357 **  MILTER_NEGOTIATE -- get version and flags from filter
2358 **
2359 **      Parameters:
2360 **              m -- milter filter structure.
2361 **              e -- current envelope.
2362 **              milters -- milters structure.
2363 **
2364 **      Returns:
2365 **              0 on success, -1 otherwise
2366 */
2367
2368 static int
2369 milter_negotiate(m, e, milters)
2370         struct milter *m;
2371         ENVELOPE *e;
2372         milters_T *milters;
2373 {
2374         char rcmd;
2375         mi_int32 fvers, fflags, pflags;
2376         mi_int32 mta_prot_vers, mta_prot_flags, mta_actions;
2377         ssize_t rlen;
2378         char *response;
2379         char data[MILTER_OPTLEN];
2380
2381         /* sanity check */
2382         if (m->mf_sock < 0 || m->mf_state != SMFS_OPEN)
2383         {
2384                 if (MilterLogLevel > 0)
2385                         sm_syslog(LOG_ERR, e->e_id,
2386                                   "Milter (%s): negotiate, impossible state",
2387                                   m->mf_name);
2388                 milter_error(m, e);
2389                 return -1;
2390         }
2391
2392 #if _FFR_MILTER_CHECK
2393         mta_prot_vers = m->mf_mta_prot_version;
2394         mta_prot_flags = m->mf_mta_prot_flags;
2395         mta_actions = m->mf_mta_actions;
2396 #else /* _FFR_MILTER_CHECK */
2397         mta_prot_vers = SMFI_PROT_VERSION;
2398         mta_prot_flags = SMFI_CURR_PROT;
2399         mta_actions = SMFI_CURR_ACTS;
2400 #endif /* _FFR_MILTER_CHECK */
2401 #if _FFR_MDS_NEGOTIATE
2402         if (MilterMaxDataSize == MILTER_MDS_256K)
2403                 mta_prot_flags |= SMFIP_MDS_256K;
2404         else if (MilterMaxDataSize == MILTER_MDS_1M)
2405                 mta_prot_flags |= SMFIP_MDS_1M;
2406 #endif /* _FFR_MDS_NEGOTIATE */
2407
2408         fvers = htonl(mta_prot_vers);
2409         pflags = htonl(mta_prot_flags);
2410         fflags = htonl(mta_actions);
2411         (void) memcpy(data, (char *) &fvers, MILTER_LEN_BYTES);
2412         (void) memcpy(data + MILTER_LEN_BYTES,
2413                       (char *) &fflags, MILTER_LEN_BYTES);
2414         (void) memcpy(data + (MILTER_LEN_BYTES * 2),
2415                       (char *) &pflags, MILTER_LEN_BYTES);
2416         (void) milter_write(m, SMFIC_OPTNEG, data, sizeof(data),
2417                             m->mf_timeout[SMFTO_WRITE], e, "negotiate");
2418
2419         if (m->mf_state == SMFS_ERROR)
2420                 return -1;
2421
2422         if (tTd(64, 5))
2423                 sm_dprintf("milter_negotiate(%s): send: version %lu, fflags 0x%lx, pflags 0x%lx\n",
2424                         m->mf_name, ntohl(fvers), ntohl(fflags), ntohl(pflags));
2425
2426         response = milter_read(m, &rcmd, &rlen, m->mf_timeout[SMFTO_READ], e,
2427                                 "negotiate");
2428         if (m->mf_state == SMFS_ERROR)
2429                 return -1;
2430
2431         if (rcmd != SMFIC_OPTNEG)
2432         {
2433                 if (tTd(64, 5))
2434                         sm_dprintf("milter_negotiate(%s): returned %c instead of %c\n",
2435                                 m->mf_name, rcmd, SMFIC_OPTNEG);
2436                 if (MilterLogLevel > 0)
2437                         sm_syslog(LOG_ERR, e->e_id,
2438                                   "Milter (%s): negotiate: returned %c instead of %c",
2439                                   m->mf_name, rcmd, SMFIC_OPTNEG);
2440                 if (response != NULL)
2441                         sm_free(response); /* XXX */
2442                 milter_error(m, e);
2443                 return -1;
2444         }
2445
2446         /* Make sure we have enough bytes for the version */
2447         if (response == NULL || rlen < MILTER_LEN_BYTES)
2448         {
2449                 if (tTd(64, 5))
2450                         sm_dprintf("milter_negotiate(%s): did not return valid info\n",
2451                                 m->mf_name);
2452                 if (MilterLogLevel > 0)
2453                         sm_syslog(LOG_ERR, e->e_id,
2454                                   "Milter (%s): negotiate: did not return valid info",
2455                                   m->mf_name);
2456                 if (response != NULL)
2457                         sm_free(response); /* XXX */
2458                 milter_error(m, e);
2459                 return -1;
2460         }
2461
2462         /* extract information */
2463         (void) memcpy((char *) &fvers, response, MILTER_LEN_BYTES);
2464
2465         /* Now make sure we have enough for the feature bitmap */
2466         if (rlen < MILTER_OPTLEN)
2467         {
2468                 if (tTd(64, 5))
2469                         sm_dprintf("milter_negotiate(%s): did not return enough info\n",
2470                                 m->mf_name);
2471                 if (MilterLogLevel > 0)
2472                         sm_syslog(LOG_ERR, e->e_id,
2473                                   "Milter (%s): negotiate: did not return enough info",
2474                                   m->mf_name);
2475                 if (response != NULL)
2476                         sm_free(response); /* XXX */
2477                 milter_error(m, e);
2478                 return -1;
2479         }
2480
2481         (void) memcpy((char *) &fflags, response + MILTER_LEN_BYTES,
2482                       MILTER_LEN_BYTES);
2483         (void) memcpy((char *) &pflags, response + (MILTER_LEN_BYTES * 2),
2484                       MILTER_LEN_BYTES);
2485
2486         m->mf_fvers = ntohl(fvers);
2487         m->mf_fflags = ntohl(fflags);
2488         m->mf_pflags = ntohl(pflags);
2489
2490         /* check for version compatibility */
2491         if (m->mf_fvers == 1 ||
2492             m->mf_fvers > SMFI_VERSION)
2493         {
2494                 if (tTd(64, 5))
2495                         sm_dprintf("milter_negotiate(%s): version %d != MTA milter version %d\n",
2496                                 m->mf_name, m->mf_fvers, SMFI_VERSION);
2497                 if (MilterLogLevel > 0)
2498                         sm_syslog(LOG_ERR, e->e_id,
2499                                   "Milter (%s): negotiate: version %d != MTA milter version %d",
2500                                   m->mf_name, m->mf_fvers, SMFI_VERSION);
2501                 milter_error(m, e);
2502                 goto error;
2503         }
2504
2505         /* check for filter feature mismatch */
2506         if ((m->mf_fflags & mta_actions) != m->mf_fflags)
2507         {
2508                 if (tTd(64, 5))
2509                         sm_dprintf("milter_negotiate(%s): filter abilities 0x%x != MTA milter abilities 0x%lx\n",
2510                                 m->mf_name, m->mf_fflags,
2511                                 (unsigned long) mta_actions);
2512                 if (MilterLogLevel > 0)
2513                         sm_syslog(LOG_ERR, e->e_id,
2514                                   "Milter (%s): negotiate: filter abilities 0x%x != MTA milter abilities 0x%lx",
2515                                   m->mf_name, m->mf_fflags,
2516                                   (unsigned long) mta_actions);
2517                 milter_error(m, e);
2518                 goto error;
2519         }
2520
2521 #if _FFR_MDS_NEGOTIATE
2522         /* use a table instead of sequence? */
2523         if (bitset(SMFIP_MDS_1M, m->mf_pflags))
2524         {
2525                 if (MilterMaxDataSize != MILTER_MDS_1M)
2526                 {
2527                         /* this should not happen... */
2528                         sm_syslog(LOG_WARNING, NOQID,
2529                                   "WARNING: Milter.maxdatasize: configured=%d, set by libmilter=%d",
2530                                   MilterMaxDataSize, MILTER_MDS_1M);
2531                         MilterMaxDataSize = MILTER_MDS_1M;
2532                 }
2533         }
2534         else if (bitset(SMFIP_MDS_256K, m->mf_pflags))
2535         {
2536                 if (MilterMaxDataSize != MILTER_MDS_256K)
2537                 {
2538                         sm_syslog(LOG_WARNING, NOQID,
2539                                   "WARNING: Milter.maxdatasize: configured=%d, set by libmilter=%d",
2540                                   MilterMaxDataSize, MILTER_MDS_256K);
2541                         MilterMaxDataSize = MILTER_MDS_256K;
2542                 }
2543         }
2544         else if (MilterMaxDataSize != MILTER_MDS_64K)
2545         {
2546                 sm_syslog(LOG_WARNING, NOQID,
2547                           "WARNING: Milter.maxdatasize: configured=%d, set by libmilter=%d",
2548                           MilterMaxDataSize, MILTER_MDS_64K);
2549                 MilterMaxDataSize = MILTER_MDS_64K;
2550         }
2551         m->mf_pflags &= ~SMFI_INTERNAL;
2552 #endif /* _FFR_MDS_NEGOTIATE */
2553
2554         /* check for protocol feature mismatch */
2555         if ((m->mf_pflags & mta_prot_flags) != m->mf_pflags)
2556         {
2557                 if (tTd(64, 5))
2558                         sm_dprintf("milter_negotiate(%s): protocol abilities 0x%x != MTA milter abilities 0x%lx\n",
2559                                 m->mf_name, m->mf_pflags,
2560                                 (unsigned long) mta_prot_flags);
2561                 if (MilterLogLevel > 0)
2562                         sm_syslog(LOG_ERR, e->e_id,
2563                                   "Milter (%s): negotiate: protocol abilities 0x%x != MTA milter abilities 0x%lx",
2564                                   m->mf_name, m->mf_pflags,
2565                                   (unsigned long) mta_prot_flags);
2566                 milter_error(m, e);
2567                 goto error;
2568         }
2569
2570         if (m->mf_fvers <= 2)
2571                 m->mf_pflags |= SMFIP_NOUNKNOWN;
2572         if (m->mf_fvers <= 3)
2573                 m->mf_pflags |= SMFIP_NODATA;
2574
2575         if (rlen > MILTER_OPTLEN)
2576         {
2577                 milter_getsymlist(m, response, rlen, MILTER_OPTLEN);
2578         }
2579
2580         if (bitset(SMFIF_DELRCPT, m->mf_fflags))
2581                 milters->mis_flags |= MIS_FL_DEL_RCPT;
2582         if (!bitset(SMFIP_NORCPT, m->mf_pflags) &&
2583             !bitset(SMFIP_NR_RCPT, m->mf_pflags))
2584                 milters->mis_flags |= MIS_FL_REJ_RCPT;
2585
2586         if (tTd(64, 5))
2587                 sm_dprintf("milter_negotiate(%s): received: version %u, fflags 0x%x, pflags 0x%x\n",
2588                         m->mf_name, m->mf_fvers, m->mf_fflags, m->mf_pflags);
2589         return 0;
2590
2591   error:
2592         if (response != NULL)
2593                 sm_free(response); /* XXX */
2594         return -1;
2595 }
2596
2597 /*
2598 **  MILTER_PER_CONNECTION_CHECK -- checks on per-connection commands
2599 **
2600 **      Reduce code duplication by putting these checks in one place
2601 **
2602 **      Parameters:
2603 **              e -- current envelope.
2604 **
2605 **      Returns:
2606 **              none
2607 */
2608
2609 static void
2610 milter_per_connection_check(e)
2611         ENVELOPE *e;
2612 {
2613         int i;
2614
2615         /* see if we are done with any of the filters */
2616         for (i = 0; InputFilters[i] != NULL; i++)
2617         {
2618                 struct milter *m = InputFilters[i];
2619
2620                 if (m->mf_state == SMFS_CLOSABLE)
2621                         milter_quit_filter(m, e);
2622         }
2623 }
2624
2625 /*
2626 **  MILTER_ERROR -- Put a milter filter into error state
2627 **
2628 **      Parameters:
2629 **              m -- the broken filter.
2630 **              e -- current envelope.
2631 **
2632 **      Returns:
2633 **              none
2634 */
2635
2636 static void
2637 milter_error(m, e)
2638         struct milter *m;
2639         ENVELOPE *e;
2640 {
2641         /*
2642         **  We could send a quit here but we may have gotten here due to
2643         **  an I/O error so we don't want to try to make things worse.
2644         */
2645
2646         if (m->mf_sock >= 0)
2647         {
2648                 (void) close(m->mf_sock);
2649                 m->mf_sock = -1;
2650         }
2651         m->mf_state = SMFS_ERROR;
2652
2653         if (MilterLogLevel > 0)
2654                 sm_syslog(LOG_INFO, e->e_id, "Milter (%s): to error state",
2655                           m->mf_name);
2656 }
2657
2658 /*
2659 **  MILTER_HEADERS -- send headers to a single milter filter
2660 **
2661 **      Parameters:
2662 **              m -- current filter.
2663 **              e -- current envelope.
2664 **              state -- return state from response.
2665 **
2666 **      Returns:
2667 **              response string (may be NULL)
2668 */
2669
2670 static char *
2671 milter_headers(m, e, state)
2672         struct milter *m;
2673         ENVELOPE *e;
2674         char *state;
2675 {
2676         char *response = NULL;
2677         HDR *h;
2678
2679         if (MilterLogLevel > 17)
2680                 sm_syslog(LOG_INFO, e->e_id, "Milter (%s): headers, send",
2681                           m->mf_name);
2682
2683         for (h = e->e_header; h != NULL; h = h->h_link)
2684         {
2685                 int len_n, len_v, len_t, len_f;
2686                 char *buf, *hv;
2687
2688                 /* don't send over deleted headers */
2689                 if (h->h_value == NULL)
2690                 {
2691                         /* strip H_USER so not counted in milter_changeheader() */
2692                         h->h_flags &= ~H_USER;
2693                         continue;
2694                 }
2695
2696                 /* skip auto-generated */
2697                 if (!bitset(H_USER, h->h_flags))
2698                         continue;
2699
2700                 if (tTd(64, 10))
2701                         sm_dprintf("milter_headers: %s:%s\n",
2702                                 h->h_field, h->h_value);
2703                 if (MilterLogLevel > 21)
2704                         sm_syslog(LOG_INFO, e->e_id, "Milter (%s): header, %s",
2705                                   m->mf_name, h->h_field);
2706
2707                 if (bitset(SMFIP_HDR_LEADSPC, m->mf_pflags)
2708                     || *(h->h_value) != ' ')
2709                         hv = h->h_value;
2710                 else
2711                         hv = h->h_value + 1;
2712                 len_f = strlen(h->h_field) + 1;
2713                 len_t = len_f + strlen(hv) + 1;
2714                 if (len_t < 0)
2715                         continue;
2716                 buf = (char *) xalloc(len_t);
2717
2718                 /*
2719                 **  Note: currently the call to dequote_internal_chars()
2720                 **  is not required as h_field is supposed to be 7-bit US-ASCII.
2721                 */
2722
2723                 len_n = dequote_internal_chars(h->h_field, buf, len_f);
2724                 SM_ASSERT(len_n < len_f);
2725                 len_v = dequote_internal_chars(hv, buf + len_n + 1,
2726                                                 len_t - len_n - 1);
2727                 SM_ASSERT(len_t >= len_n + 1 + len_v + 1);
2728                 len_t = len_n + 1 + len_v + 1;
2729
2730                 /* send it over */
2731                 response = milter_send_command(m, SMFIC_HEADER, buf,
2732                                                len_t, e, state, "header");
2733                 sm_free(buf);
2734                 if (m->mf_state == SMFS_ERROR ||
2735                     m->mf_state == SMFS_DONE ||
2736                     *state != SMFIR_CONTINUE)
2737                         break;
2738         }
2739         if (MilterLogLevel > 17)
2740                 sm_syslog(LOG_INFO, e->e_id, "Milter (%s): headers, sent",
2741                           m->mf_name);
2742         return response;
2743 }
2744
2745 /*
2746 **  MILTER_BODY -- send the body to a filter
2747 **
2748 **      Parameters:
2749 **              m -- current filter.
2750 **              e -- current envelope.
2751 **              state -- return state from response.
2752 **
2753 **      Returns:
2754 **              response string (may be NULL)
2755 */
2756
2757 static char *
2758 milter_body(m, e, state)
2759         struct milter *m;
2760         ENVELOPE *e;
2761         char *state;
2762 {
2763         char bufchar = '\0';
2764         char prevchar = '\0';
2765         int c;
2766         char *response = NULL;
2767         char *bp;
2768         char buf[MILTER_CHUNK_SIZE];
2769
2770         if (tTd(64, 10))
2771                 sm_dprintf("milter_body\n");
2772
2773         if (bfrewind(e->e_dfp) < 0)
2774         {
2775                 ExitStat = EX_IOERR;
2776                 *state = SMFIR_TEMPFAIL;
2777                 syserr("milter_body: %s/%cf%s: rewind error",
2778                        qid_printqueue(e->e_qgrp, e->e_qdir),
2779                        DATAFL_LETTER, e->e_id);
2780                 return NULL;
2781         }
2782
2783         if (MilterLogLevel > 17)
2784                 sm_syslog(LOG_INFO, e->e_id, "Milter (%s): body, send",
2785                           m->mf_name);
2786         bp = buf;
2787         while ((c = sm_io_getc(e->e_dfp, SM_TIME_DEFAULT)) != SM_IO_EOF)
2788         {
2789                 /*  Change LF to CRLF */
2790                 if (c == '\n')
2791                 {
2792 #if !_FFR_MILTER_CONVERT_ALL_LF_TO_CRLF
2793                         /* Not a CRLF already? */
2794                         if (prevchar != '\r')
2795 #endif /* !_FFR_MILTER_CONVERT_ALL_LF_TO_CRLF */
2796                         {
2797                                 /* Room for CR now? */
2798                                 if (bp + 2 > &buf[sizeof(buf)])
2799                                 {
2800                                         /* No room, buffer LF */
2801                                         bufchar = c;
2802
2803                                         /* and send CR now */
2804                                         c = '\r';
2805                                 }
2806                                 else
2807                                 {
2808                                         /* Room to do it now */
2809                                         *bp++ = '\r';
2810                                         prevchar = '\r';
2811                                 }
2812                         }
2813                 }
2814                 *bp++ = (char) c;
2815                 prevchar = c;
2816                 if (bp >= &buf[sizeof(buf)])
2817                 {
2818                         /* send chunk */
2819                         response = milter_send_command(m, SMFIC_BODY, buf,
2820                                                        bp - buf, e, state,
2821                                                         "body chunk");
2822                         bp = buf;
2823                         if (bufchar != '\0')
2824                         {
2825                                 *bp++ = bufchar;
2826                                 bufchar = '\0';
2827                                 prevchar = bufchar;
2828                         }
2829                 }
2830                 if (m->mf_state == SMFS_ERROR ||
2831                     m->mf_state == SMFS_DONE ||
2832                     m->mf_state == SMFS_SKIP ||
2833                     *state != SMFIR_CONTINUE)
2834                         break;
2835         }
2836
2837         /* check for read errors */
2838         if (sm_io_error(e->e_dfp))
2839         {
2840                 ExitStat = EX_IOERR;
2841                 if (*state == SMFIR_CONTINUE ||
2842                     *state == SMFIR_ACCEPT ||
2843                     m->mf_state == SMFS_SKIP)
2844                 {
2845                         *state = SMFIR_TEMPFAIL;
2846                         if (response != NULL)
2847                         {
2848                                 sm_free(response); /* XXX */
2849                                 response = NULL;
2850                         }
2851                 }
2852                 syserr("milter_body: %s/%cf%s: read error",
2853                        qid_printqueue(e->e_qgrp, e->e_qdir),
2854                        DATAFL_LETTER, e->e_id);
2855                 return response;
2856         }
2857
2858         /* send last body chunk */
2859         if (bp > buf &&
2860             m->mf_state != SMFS_ERROR &&
2861             m->mf_state != SMFS_DONE &&
2862             m->mf_state != SMFS_SKIP &&
2863             *state == SMFIR_CONTINUE)
2864         {
2865                 /* send chunk */
2866                 response = milter_send_command(m, SMFIC_BODY, buf, bp - buf,
2867                                                e, state, "last body chunk");
2868                 bp = buf;
2869         }
2870         if (MilterLogLevel > 17)
2871                 sm_syslog(LOG_INFO, e->e_id, "Milter (%s): body, sent",
2872                           m->mf_name);
2873         if (m->mf_state == SMFS_SKIP)
2874         {
2875                 *state = SMFIR_CONTINUE;
2876                 m->mf_state = SMFS_READY;
2877         }
2878
2879         return response;
2880 }
2881
2882 /*
2883 **  Actions
2884 */
2885
2886 /*
2887 **  ADDLEADINGSPACE -- Add a leading space to a string
2888 **
2889 **      Parameters:
2890 **              str -- string
2891 **              rp -- resource pool for allocations
2892 **
2893 **      Returns:
2894 **              pointer to new string
2895 */
2896
2897 static char *addleadingspace __P((char *, SM_RPOOL_T *));
2898
2899 static char *
2900 addleadingspace(str, rp)
2901         char *str;
2902         SM_RPOOL_T *rp;
2903 {
2904         size_t l;
2905         char *new;
2906
2907         SM_ASSERT(str != NULL);
2908         l = strlen(str);
2909         SM_ASSERT(l + 2 > l);
2910         new = sm_rpool_malloc_x(rp, l + 2);
2911         new[0] = ' ';
2912         new[1] = '\0';
2913         sm_strlcpy(new + 1, str, l + 1);
2914         return new;
2915 }
2916
2917 /*
2918 **  MILTER_ADDHEADER -- Add the supplied header to the message
2919 **
2920 **      Parameters:
2921 **              m -- current filter.
2922 **              response -- encoded form of header/value.
2923 **              rlen -- length of response.
2924 **              e -- current envelope.
2925 **
2926 **      Returns:
2927 **              none
2928 */
2929
2930 static void
2931 milter_addheader(m, response, rlen, e)
2932         struct milter *m;
2933         char *response;
2934         ssize_t rlen;
2935         ENVELOPE *e;
2936 {
2937         int mh_v_len;
2938         char *val, *mh_value;
2939         HDR *h;
2940
2941         if (tTd(64, 10))
2942                 sm_dprintf("milter_addheader: ");
2943
2944         /* sanity checks */
2945         if (response == NULL)
2946         {
2947                 if (tTd(64, 10))
2948                         sm_dprintf("NULL response\n");
2949                 return;
2950         }
2951
2952         if (rlen < 2 || strlen(response) + 1 >= (size_t) rlen)
2953         {
2954                 if (tTd(64, 10))
2955                         sm_dprintf("didn't follow protocol (total len %d != rlen %d)\n",
2956                                    (int) strlen(response), (int) (rlen - 1));
2957                 return;
2958         }
2959
2960         /* Find separating NUL */
2961         val = response + strlen(response) + 1;
2962
2963         /* another sanity check */
2964         if (strlen(response) + strlen(val) + 2 != (size_t) rlen)
2965         {
2966                 if (tTd(64, 10))
2967                         sm_dprintf("didn't follow protocol (part len)\n");
2968                 return;
2969         }
2970
2971         if (*response == '\0')
2972         {
2973                 if (tTd(64, 10))
2974                         sm_dprintf("empty field name\n");
2975                 return;
2976         }
2977
2978         for (h = e->e_header; h != NULL; h = h->h_link)
2979         {
2980                 if (sm_strcasecmp(h->h_field, response) == 0 &&
2981                     !bitset(H_USER, h->h_flags) &&
2982                     !bitset(H_TRACE, h->h_flags))
2983                         break;
2984         }
2985
2986         mh_v_len = 0;
2987         mh_value = quote_internal_chars(val, NULL, &mh_v_len);
2988
2989         /* add to e_msgsize */
2990         e->e_msgsize += strlen(response) + 2 + strlen(val);
2991
2992         if (h != NULL)
2993         {
2994                 if (tTd(64, 10))
2995                         sm_dprintf("Replace default header %s value with %s\n",
2996                                    h->h_field, mh_value);
2997                 if (MilterLogLevel > 8)
2998                         sm_syslog(LOG_INFO, e->e_id,
2999                                   "Milter change: default header %s value with %s",
3000                                   h->h_field, mh_value);
3001                 if (bitset(SMFIP_HDR_LEADSPC, m->mf_pflags))
3002                         h->h_value = mh_value;
3003                 else
3004                 {
3005                         h->h_value = addleadingspace(mh_value, e->e_rpool);
3006                         SM_FREE(mh_value);
3007                 }
3008                 h->h_flags |= H_USER;
3009         }
3010         else
3011         {
3012                 if (tTd(64, 10))
3013                         sm_dprintf("Add %s: %s\n", response, mh_value);
3014                 if (MilterLogLevel > 8)
3015                         sm_syslog(LOG_INFO, e->e_id,
3016                                   "Milter add: header: %s: %s",
3017                                   response, mh_value);
3018                 addheader(newstr(response), mh_value, H_USER, e,
3019                         !bitset(SMFIP_HDR_LEADSPC, m->mf_pflags));
3020                 SM_FREE(mh_value);
3021         }
3022 }
3023
3024 /*
3025 **  MILTER_INSHEADER -- Insert the supplied header
3026 **
3027 **      Parameters:
3028 **              m -- current filter.
3029 **              response -- encoded form of header/value.
3030 **              rlen -- length of response.
3031 **              e -- current envelope.
3032 **
3033 **      Returns:
3034 **              none
3035 **
3036 **      Notes:
3037 **              Unlike milter_addheader(), this does not attempt to determine
3038 **              if the header already exists in the envelope, even a
3039 **              deleted version.  It just blindly inserts.
3040 */
3041
3042 static void
3043 milter_insheader(m, response, rlen, e)
3044         struct milter *m;
3045         char *response;
3046         ssize_t rlen;
3047         ENVELOPE *e;
3048 {
3049         mi_int32 idx, i;
3050         int mh_v_len;
3051         char *field, *val, *mh_value;
3052
3053         if (tTd(64, 10))
3054                 sm_dprintf("milter_insheader: ");
3055
3056         /* sanity checks */
3057         if (response == NULL)
3058         {
3059                 if (tTd(64, 10))
3060                         sm_dprintf("NULL response\n");
3061                 return;
3062         }
3063
3064         if (rlen < 2 || strlen(response) + 1 >= (size_t) rlen)
3065         {
3066                 if (tTd(64, 10))
3067                         sm_dprintf("didn't follow protocol (total len)\n");
3068                 return;
3069         }
3070
3071         /* decode */
3072         (void) memcpy((char *) &i, response, MILTER_LEN_BYTES);
3073         idx = ntohl(i);
3074         field = response + MILTER_LEN_BYTES;
3075         val = field + strlen(field) + 1;
3076
3077         /* another sanity check */
3078         if (MILTER_LEN_BYTES + strlen(field) + 1 +
3079             strlen(val) + 1 != (size_t) rlen)
3080         {
3081                 if (tTd(64, 10))
3082                         sm_dprintf("didn't follow protocol (part len)\n");
3083                 return;
3084         }
3085
3086         if (*field == '\0')
3087         {
3088                 if (tTd(64, 10))
3089                         sm_dprintf("empty field name\n");
3090                 return;
3091         }
3092
3093         /* add to e_msgsize */
3094         e->e_msgsize += strlen(response) + 2 + strlen(val);
3095
3096         if (tTd(64, 10))
3097                 sm_dprintf("Insert (%d) %s: %s\n", idx, field, val);
3098         if (MilterLogLevel > 8)
3099                 sm_syslog(LOG_INFO, e->e_id,
3100                           "Milter insert (%d): header: %s: %s",
3101                           idx, field, val);
3102         mh_v_len = 0;
3103         mh_value = quote_internal_chars(val, NULL, &mh_v_len);
3104         insheader(idx, newstr(field), mh_value, H_USER, e,
3105                 !bitset(SMFIP_HDR_LEADSPC, m->mf_pflags));
3106         SM_FREE(mh_value);
3107 }
3108
3109 /*
3110 **  MILTER_CHANGEHEADER -- Change the supplied header in the message
3111 **
3112 **      Parameters:
3113 **              m -- current filter.
3114 **              response -- encoded form of header/index/value.
3115 **              rlen -- length of response.
3116 **              e -- current envelope.
3117 **
3118 **      Returns:
3119 **              none
3120 */
3121
3122 static void
3123 milter_changeheader(m, response, rlen, e)
3124         struct milter *m;
3125         char *response;
3126         ssize_t rlen;
3127         ENVELOPE *e;
3128 {
3129         mi_int32 i, index;
3130         int mh_v_len;
3131         char *field, *val, *mh_value;
3132         HDR *h, *sysheader;
3133
3134         if (tTd(64, 10))
3135                 sm_dprintf("milter_changeheader: ");
3136
3137         /* sanity checks */
3138         if (response == NULL)
3139         {
3140                 if (tTd(64, 10))
3141                         sm_dprintf("NULL response\n");
3142                 return;
3143         }
3144
3145         if (rlen < 2 || strlen(response) + 1 >= (size_t) rlen)
3146         {
3147                 if (tTd(64, 10))
3148                         sm_dprintf("didn't follow protocol (total len)\n");
3149                 return;
3150         }
3151
3152         /* Find separating NUL */
3153         (void) memcpy((char *) &i, response, MILTER_LEN_BYTES);
3154         index = ntohl(i);
3155         field = response + MILTER_LEN_BYTES;
3156         val = field + strlen(field) + 1;
3157
3158         /* another sanity check */
3159         if (MILTER_LEN_BYTES + strlen(field) + 1 +
3160             strlen(val) + 1 != (size_t) rlen)
3161         {
3162                 if (tTd(64, 10))
3163                         sm_dprintf("didn't follow protocol (part len)\n");
3164                 return;
3165         }
3166
3167         if (*field == '\0')
3168         {
3169                 if (tTd(64, 10))
3170                         sm_dprintf("empty field name\n");
3171                 return;
3172         }
3173
3174         mh_v_len = 0;
3175         mh_value = quote_internal_chars(val, NULL, &mh_v_len);
3176
3177         sysheader = NULL;
3178         for (h = e->e_header; h != NULL; h = h->h_link)
3179         {
3180                 if (sm_strcasecmp(h->h_field, field) == 0)
3181                 {
3182                         if (bitset(H_USER, h->h_flags) && --index <= 0)
3183                         {
3184                                 sysheader = NULL;
3185                                 break;
3186                         }
3187                         else if (!bitset(H_USER, h->h_flags) &&
3188                                  !bitset(H_TRACE, h->h_flags))
3189                         {
3190                                 /*
3191                                 **  DRUMS msg-fmt draft says can only have
3192                                 **  multiple occurences of trace fields,
3193                                 **  so make sure we replace any non-trace,
3194                                 **  non-user field.
3195                                 */
3196
3197                                 sysheader = h;
3198                         }
3199                 }
3200         }
3201
3202         /* if not found as user-provided header at index, use sysheader */
3203         if (h == NULL)
3204                 h = sysheader;
3205
3206         if (h == NULL)
3207         {
3208                 if (*val == '\0')
3209                 {
3210                         if (tTd(64, 10))
3211                                 sm_dprintf("Delete (noop) %s\n", field);
3212                         if (MilterLogLevel > 8)
3213                                 sm_syslog(LOG_INFO, e->e_id,
3214                                         "Milter delete (noop): header: %s"
3215                                         , field);
3216                 }
3217                 else
3218                 {
3219                         /* treat modify value with no existing header as add */
3220                         if (tTd(64, 10))
3221                                 sm_dprintf("Add %s: %s\n", field, mh_value);
3222                         if (MilterLogLevel > 8)
3223                                 sm_syslog(LOG_INFO, e->e_id,
3224                                         "Milter change (add): header: %s: %s"
3225                                         , field, mh_value);
3226                         addheader(newstr(field), mh_value, H_USER, e,
3227                                 !bitset(SMFIP_HDR_LEADSPC, m->mf_pflags));
3228                 }
3229                 return;
3230         }
3231
3232         if (tTd(64, 10))
3233         {
3234                 if (*val == '\0')
3235                 {
3236                         sm_dprintf("Delete%s %s:%s\n",
3237                                    h == sysheader ? " (default header)" : "",
3238                                    field,
3239                                    h->h_value == NULL ? "<NULL>" : h->h_value);
3240                 }
3241                 else
3242                 {
3243                         sm_dprintf("Change%s %s: from %s to %s\n",
3244                                    h == sysheader ? " (default header)" : "",
3245                                    field,
3246                                    h->h_value == NULL ? "<NULL>" : h->h_value,
3247                                    mh_value);
3248                 }
3249         }
3250
3251         if (MilterLogLevel > 8)
3252         {
3253                 if (*val == '\0')
3254                 {
3255                         sm_syslog(LOG_INFO, e->e_id,
3256                                   "Milter delete: header%s %s:%s",
3257                                   h == sysheader ? " (default header)" : "",
3258                                   field,
3259                                   h->h_value == NULL ? "<NULL>" : h->h_value);
3260                 }
3261                 else
3262                 {
3263                         sm_syslog(LOG_INFO, e->e_id,
3264                                   "Milter change: header%s %s: from %s to %s",
3265                                   h == sysheader ? " (default header)" : "",
3266                                   field,
3267                                   h->h_value == NULL ? "<NULL>" : h->h_value,
3268                                   mh_value);
3269                 }
3270         }
3271
3272         if (h != sysheader && h->h_value != NULL)
3273         {
3274                 size_t l;
3275
3276                 l = strlen(h->h_value);
3277                 if (l > e->e_msgsize)
3278                         e->e_msgsize = 0;
3279                 else
3280                         e->e_msgsize -= l;
3281                 /* rpool, don't free: sm_free(h->h_value); XXX */
3282         }
3283
3284         if (*val == '\0')
3285         {
3286                 /* Remove "Field: " from message size */
3287                 if (h != sysheader)
3288                 {
3289                         size_t l;
3290
3291                         l = strlen(h->h_field) + 2;
3292                         if (l > e->e_msgsize)
3293                                 e->e_msgsize = 0;
3294                         else
3295                                 e->e_msgsize -= l;
3296                 }
3297                 h->h_value = NULL;
3298                 SM_FREE(mh_value);
3299         }
3300         else
3301         {
3302                 if (bitset(SMFIP_HDR_LEADSPC, m->mf_pflags))
3303                         h->h_value = mh_value;
3304                 else
3305                 {
3306                         h->h_value = addleadingspace(mh_value, e->e_rpool);
3307                         SM_FREE(mh_value);
3308                 }
3309                 h->h_flags |= H_USER;
3310                 e->e_msgsize += strlen(h->h_value);
3311         }
3312 }
3313
3314 /*
3315 **  MILTER_SPLIT_RESPONSE -- Split response into fields.
3316 **
3317 **      Parameters:
3318 **              response -- encoded repsonse.
3319 **              rlen -- length of response.
3320 **              pargc -- number of arguments (ouput)
3321 **
3322 **      Returns:
3323 **              array of pointers to the individual strings
3324 */
3325
3326 static char **milter_split_response __P((char *, ssize_t, int *));
3327
3328 static char **
3329 milter_split_response(response, rlen, pargc)
3330         char *response;
3331         ssize_t rlen;
3332         int *pargc;
3333 {
3334         char **s;
3335         size_t i;
3336         int elem, nelem;
3337
3338         SM_ASSERT(response != NULL);
3339         SM_ASSERT(pargc != NULL);
3340         *pargc = 0;
3341         if (rlen < 2 || strlen(response) >= (size_t) rlen)
3342         {
3343                 if (tTd(64, 10))
3344                         sm_dprintf("didn't follow protocol (total len %d != rlen %d)\n",
3345                                    (int) strlen(response), (int) (rlen - 1));
3346                 return NULL;
3347         }
3348
3349         nelem = 0;
3350         for (i = 0; i < rlen; i++)
3351         {
3352                 if (response[i] == '\0')
3353                         ++nelem;
3354         }
3355         if (nelem == 0)
3356                 return NULL;
3357
3358         /* last entry is only for the name */
3359         s = (char **)malloc((nelem + 1) * (sizeof(*s)));
3360         if (s == NULL)
3361                 return NULL;
3362         s[0] = response;
3363         for (i = 0, elem = 0; i < rlen && elem < nelem; i++)
3364         {
3365                 if (response[i] == '\0')
3366                 {
3367                         ++elem;
3368                         if (i + 1 >= rlen)
3369                                 s[elem] = NULL;
3370                         else
3371                                 s[elem] = &(response[i + 1]);
3372                 }
3373         }
3374         *pargc = nelem;
3375
3376         if (tTd(64, 10))
3377         {
3378                 for (elem = 0; elem < nelem; elem++)
3379                         sm_dprintf("argv[%d]=\"%s\"\n", elem, s[elem]);
3380         }
3381
3382         /* overwrite last entry (already done above, just paranoia) */
3383         s[elem] = NULL;
3384         return s;
3385 }
3386
3387 /*
3388 **  MILTER_CHGFROM -- Change the envelope sender address
3389 **
3390 **      Parameters:
3391 **              response -- encoded form of recipient address.
3392 **              rlen -- length of response.
3393 **              e -- current envelope.
3394 **
3395 **      Returns:
3396 **              none
3397 */
3398
3399 static void
3400 milter_chgfrom(response, rlen, e)
3401         char *response;
3402         ssize_t rlen;
3403         ENVELOPE *e;
3404 {
3405         int olderrors, argc;
3406         char **argv;
3407
3408         if (tTd(64, 10))
3409                 sm_dprintf("milter_chgfrom: ");
3410
3411         /* sanity checks */
3412         if (response == NULL)
3413         {
3414                 if (tTd(64, 10))
3415                         sm_dprintf("NULL response\n");
3416                 return;
3417         }
3418
3419         if (*response == '\0' ||
3420             strlen(response) + 1 > (size_t) rlen)
3421         {
3422                 if (tTd(64, 10))
3423                         sm_dprintf("didn't follow protocol (total len %d != rlen %d)\n",
3424                                    (int) strlen(response), (int) (rlen - 1));
3425                 return;
3426         }
3427
3428         if (tTd(64, 10))
3429                 sm_dprintf("%s\n", response);
3430         if (MilterLogLevel > 8)
3431                 sm_syslog(LOG_INFO, e->e_id, "Milter chgfrom: %s", response);
3432         argv = milter_split_response(response, rlen, &argc);
3433         if (argc < 1 || argc > 2)
3434         {
3435                 if (tTd(64, 10))
3436                         sm_dprintf("didn't follow protocol argc=%d\n", argc);
3437                 return;
3438         }
3439
3440         olderrors = Errors;
3441         setsender(argv[0], e, NULL, '\0', false);
3442         if (argc == 2)
3443         {
3444                 reset_mail_esmtp_args(e);
3445
3446                 /*
3447                 **  need "features" here: how to get those? via e?
3448                 **  "fake" it for now: allow everything.
3449                 */
3450
3451                 parse_esmtp_args(e, NULL, argv[0], argv[1], "MAIL", NULL,
3452                                 mail_esmtp_args);
3453         }
3454         Errors = olderrors;
3455         return;
3456 }
3457
3458 /*
3459 **  MILTER_ADDRCPT_PAR -- Add the supplied recipient to the message
3460 **
3461 **      Parameters:
3462 **              response -- encoded form of recipient address.
3463 **              rlen -- length of response.
3464 **              e -- current envelope.
3465 **
3466 **      Returns:
3467 **              none
3468 */
3469
3470 static void
3471 milter_addrcpt_par(response, rlen, e)
3472         char *response;
3473         ssize_t rlen;
3474         ENVELOPE *e;
3475 {
3476         int olderrors, argc;
3477         char *delimptr;
3478         char **argv;
3479         ADDRESS *a;
3480
3481         if (tTd(64, 10))
3482                 sm_dprintf("milter_addrcpt_par: ");
3483
3484         /* sanity checks */
3485         if (response == NULL)
3486         {
3487                 if (tTd(64, 10))
3488                         sm_dprintf("NULL response\n");
3489                 return;
3490         }
3491
3492         if (tTd(64, 10))
3493                 sm_dprintf("%s\n", response);
3494         if (MilterLogLevel > 8)
3495                 sm_syslog(LOG_INFO, e->e_id, "Milter add: rcpt: %s", response);
3496
3497         argv = milter_split_response(response, rlen, &argc);
3498         if (argc < 1 || argc > 2)
3499         {
3500                 if (tTd(64, 10))
3501                         sm_dprintf("didn't follow protocol argc=%d\n", argc);
3502                 return;
3503         }
3504         olderrors = Errors;
3505
3506         /* how to set ESMTP arguments? */
3507         a = parseaddr(argv[0], NULLADDR, RF_COPYALL, ' ', &delimptr, e, true);
3508
3509         if (a != NULL && olderrors == Errors)
3510         {
3511                 parse_esmtp_args(e, a, argv[0], argv[1], "RCPT", NULL,
3512                                 rcpt_esmtp_args);
3513                 if (olderrors == Errors)
3514                         a = recipient(a, &e->e_sendqueue, 0, e);
3515                 else
3516                         sm_dprintf("olderrors=%d, Errors=%d\n",
3517                                 olderrors, Errors);
3518         }
3519         else
3520         {
3521                 sm_dprintf("a=%p, olderrors=%d, Errors=%d\n",
3522                         a, olderrors, Errors);
3523         }
3524
3525         Errors = olderrors;
3526         return;
3527 }
3528
3529 /*
3530 **  MILTER_ADDRCPT -- Add the supplied recipient to the message
3531 **
3532 **      Parameters:
3533 **              response -- encoded form of recipient address.
3534 **              rlen -- length of response.
3535 **              e -- current envelope.
3536 **
3537 **      Returns:
3538 **              none
3539 */
3540
3541 static void
3542 milter_addrcpt(response, rlen, e)
3543         char *response;
3544         ssize_t rlen;
3545         ENVELOPE *e;
3546 {
3547         int olderrors;
3548
3549         if (tTd(64, 10))
3550                 sm_dprintf("milter_addrcpt: ");
3551
3552         /* sanity checks */
3553         if (response == NULL)
3554         {
3555                 if (tTd(64, 10))
3556                         sm_dprintf("NULL response\n");
3557                 return;
3558         }
3559
3560         if (*response == '\0' ||
3561             strlen(response) + 1 != (size_t) rlen)
3562         {
3563                 if (tTd(64, 10))
3564                         sm_dprintf("didn't follow protocol (total len %d != rlen %d)\n",
3565                                    (int) strlen(response), (int) (rlen - 1));
3566                 return;
3567         }
3568
3569         if (tTd(64, 10))
3570                 sm_dprintf("%s\n", response);
3571         if (MilterLogLevel > 8)
3572                 sm_syslog(LOG_INFO, e->e_id, "Milter add: rcpt: %s", response);
3573         olderrors = Errors;
3574         (void) sendtolist(response, NULLADDR, &e->e_sendqueue, 0, e);
3575         Errors = olderrors;
3576         return;
3577 }
3578
3579 /*
3580 **  MILTER_DELRCPT -- Delete the supplied recipient from the message
3581 **
3582 **      Parameters:
3583 **              response -- encoded form of recipient address.
3584 **              rlen -- length of response.
3585 **              e -- current envelope.
3586 **
3587 **      Returns:
3588 **              none
3589 */
3590
3591 static void
3592 milter_delrcpt(response, rlen, e)
3593         char *response;
3594         ssize_t rlen;
3595         ENVELOPE *e;
3596 {
3597         if (tTd(64, 10))
3598                 sm_dprintf("milter_delrcpt: ");
3599
3600         /* sanity checks */
3601         if (response == NULL)
3602         {
3603                 if (tTd(64, 10))
3604                         sm_dprintf("NULL response\n");
3605                 return;
3606         }
3607
3608         if (*response == '\0' ||
3609             strlen(response) + 1 != (size_t) rlen)
3610         {
3611                 if (tTd(64, 10))
3612                         sm_dprintf("didn't follow protocol (total len %d != rlen %d)\n",
3613                                    (int) strlen(response), (int) (rlen - 1));
3614                 return;
3615         }
3616
3617         if (tTd(64, 10))
3618                 sm_dprintf("%s\n", response);
3619         if (MilterLogLevel > 8)
3620                 sm_syslog(LOG_INFO, e->e_id, "Milter delete: rcpt %s",
3621                           response);
3622         (void) removefromlist(response, &e->e_sendqueue, e);
3623         return;
3624 }
3625
3626 /*
3627 **  MILTER_REPLBODY -- Replace the current data file with new body
3628 **
3629 **      Parameters:
3630 **              response -- encoded form of new body.
3631 **              rlen -- length of response.
3632 **              newfilter -- if first time called by a new filter
3633 **              e -- current envelope.
3634 **
3635 **      Returns:
3636 **              0 upon success, -1 upon failure
3637 */
3638
3639 static int
3640 milter_replbody(response, rlen, newfilter, e)
3641         char *response;
3642         ssize_t rlen;
3643         bool newfilter;
3644         ENVELOPE *e;
3645 {
3646         static char prevchar;
3647         int i;
3648
3649         if (tTd(64, 10))
3650                 sm_dprintf("milter_replbody\n");
3651
3652         /* If a new filter, reset previous character and truncate data file */
3653         if (newfilter)
3654         {
3655                 off_t prevsize;
3656                 char dfname[MAXPATHLEN];
3657
3658                 (void) sm_strlcpy(dfname, queuename(e, DATAFL_LETTER),
3659                                   sizeof(dfname));
3660
3661                 /* Reset prevchar */
3662                 prevchar = '\0';
3663
3664                 /* Get the current data file information */
3665                 prevsize = sm_io_getinfo(e->e_dfp, SM_IO_WHAT_SIZE, NULL);
3666                 if (prevsize < 0)
3667                         prevsize = 0;
3668
3669                 /* truncate current data file */
3670                 if (sm_io_getinfo(e->e_dfp, SM_IO_WHAT_ISTYPE, BF_FILE_TYPE))
3671                 {
3672                         if (sm_io_setinfo(e->e_dfp, SM_BF_TRUNCATE, NULL) < 0)
3673                         {
3674                                 MILTER_DF_ERROR("milter_replbody: sm_io truncate %s: %s");
3675                                 return -1;
3676                         }
3677                 }
3678                 else
3679                 {
3680                         int err;
3681
3682                         err = sm_io_error(e->e_dfp);
3683                         (void) sm_io_flush(e->e_dfp, SM_TIME_DEFAULT);
3684
3685                         /*
3686                         **  Clear error if tried to fflush()
3687                         **  a read-only file pointer and
3688                         **  there wasn't a previous error.
3689                         */
3690
3691                         if (err == 0)
3692                                 sm_io_clearerr(e->e_dfp);
3693
3694                         /* errno is set implicitly by fseek() before return */
3695                         err = sm_io_seek(e->e_dfp, SM_TIME_DEFAULT,
3696                                          0, SEEK_SET);
3697                         if (err < 0)
3698                         {
3699                                 MILTER_DF_ERROR("milter_replbody: sm_io_seek %s: %s");
3700                                 return -1;
3701                         }
3702 # if NOFTRUNCATE
3703                         /* XXX: Not much we can do except rewind it */
3704                         errno = EINVAL;
3705                         MILTER_DF_ERROR("milter_replbody: ftruncate not available on this platform (%s:%s)");
3706                         return -1;
3707 # else /* NOFTRUNCATE */
3708                         err = ftruncate(sm_io_getinfo(e->e_dfp,
3709                                                       SM_IO_WHAT_FD, NULL),
3710                                         0);
3711                         if (err < 0)
3712                         {
3713                                 MILTER_DF_ERROR("milter_replbody: sm_io ftruncate %s: %s");
3714                                 return -1;
3715                         }
3716 # endif /* NOFTRUNCATE */
3717                 }
3718
3719                 if (prevsize > e->e_msgsize)
3720                         e->e_msgsize = 0;
3721                 else
3722                         e->e_msgsize -= prevsize;
3723         }
3724
3725         if (newfilter && MilterLogLevel > 8)
3726                 sm_syslog(LOG_INFO, e->e_id, "Milter message: body replaced");
3727
3728         if (response == NULL)
3729         {
3730                 /* Flush the buffered '\r' */
3731                 if (prevchar == '\r')
3732                 {
3733                         (void) sm_io_putc(e->e_dfp, SM_TIME_DEFAULT, prevchar);
3734                         e->e_msgsize++;
3735                 }
3736                 return 0;
3737         }
3738
3739         for (i = 0; i < rlen; i++)
3740         {
3741                 /* Buffered char from last chunk */
3742                 if (i == 0 && prevchar == '\r')
3743                 {
3744                         /* Not CRLF, output prevchar */
3745                         if (response[i] != '\n')
3746                         {
3747                                 (void) sm_io_putc(e->e_dfp, SM_TIME_DEFAULT,
3748                                                   prevchar);
3749                                 e->e_msgsize++;
3750                         }
3751                         prevchar = '\0';
3752                 }
3753
3754                 /* Turn CRLF into LF */
3755                 if (response[i] == '\r')
3756                 {
3757                         /* check if at end of chunk */
3758                         if (i + 1 < rlen)
3759                         {
3760                                 /* If LF, strip CR */
3761                                 if (response[i + 1] == '\n')
3762                                         i++;
3763                         }
3764                         else
3765                         {
3766                                 /* check next chunk */
3767                                 prevchar = '\r';
3768                                 continue;
3769                         }
3770                 }
3771                 (void) sm_io_putc(e->e_dfp, SM_TIME_DEFAULT, response[i]);
3772                 e->e_msgsize++;
3773         }
3774         return 0;
3775 }
3776
3777 /*
3778 **  MTA callouts
3779 */
3780
3781 /*
3782 **  MILTER_INIT -- open and negotiate with all of the filters
3783 **
3784 **      Parameters:
3785 **              e -- current envelope.
3786 **              state -- return state from response.
3787 **              milters -- milters structure.
3788 **
3789 **      Returns:
3790 **              true iff at least one filter is active
3791 */
3792
3793 /* ARGSUSED */
3794 bool
3795 milter_init(e, state, milters)
3796         ENVELOPE *e;
3797         char *state;
3798         milters_T *milters;
3799 {
3800         int i;
3801
3802         if (tTd(64, 10))
3803                 sm_dprintf("milter_init\n");
3804
3805         memset(milters, '\0', sizeof(*milters));
3806         *state = SMFIR_CONTINUE;
3807         if (InputFilters[0] == NULL)
3808         {
3809                 if (MilterLogLevel > 10)
3810                         sm_syslog(LOG_INFO, e->e_id,
3811                                   "Milter: no active filter");
3812                 return false;
3813         }
3814
3815         for (i = 0; InputFilters[i] != NULL; i++)
3816         {
3817                 struct milter *m = InputFilters[i];
3818
3819                 m->mf_sock = milter_open(m, false, e);
3820                 if (m->mf_state == SMFS_ERROR)
3821                 {
3822                         MILTER_CHECK_ERROR(true, continue);
3823                         break;
3824                 }
3825
3826                 if (m->mf_sock < 0 ||
3827                     milter_negotiate(m, e, milters) < 0 ||
3828                     m->mf_state == SMFS_ERROR)
3829                 {
3830                         if (tTd(64, 5))
3831                                 sm_dprintf("milter_init(%s): failed to %s\n",
3832                                            m->mf_name,
3833                                            m->mf_sock < 0 ? "open" :
3834                                                             "negotiate");
3835                         if (MilterLogLevel > 0)
3836                                 sm_syslog(LOG_ERR, e->e_id,
3837                                           "Milter (%s): init failed to %s",
3838                                           m->mf_name,
3839                                           m->mf_sock < 0 ? "open" :
3840                                                            "negotiate");
3841
3842                         /* if negotiation failure, close socket */
3843                         milter_error(m, e);
3844                         MILTER_CHECK_ERROR(true, continue);
3845                         continue;
3846                 }
3847                 if (MilterLogLevel > 9)
3848                         sm_syslog(LOG_INFO, e->e_id,
3849                                   "Milter (%s): init success to %s",
3850                                   m->mf_name,
3851                                   m->mf_sock < 0 ? "open" : "negotiate");
3852         }
3853
3854         /*
3855         **  If something temp/perm failed with one of the filters,
3856         **  we won't be using any of them, so clear any existing
3857         **  connections.
3858         */
3859
3860         if (*state != SMFIR_CONTINUE)
3861                 milter_quit(e);
3862
3863         return true;
3864 }
3865
3866 /*
3867 **  MILTER_CONNECT -- send connection info to milter filters
3868 **
3869 **      Parameters:
3870 **              hostname -- hostname of remote machine.
3871 **              addr -- address of remote machine.
3872 **              e -- current envelope.
3873 **              state -- return state from response.
3874 **
3875 **      Returns:
3876 **              response string (may be NULL)
3877 */
3878
3879 char *
3880 milter_connect(hostname, addr, e, state)
3881         char *hostname;
3882         SOCKADDR addr;
3883         ENVELOPE *e;
3884         char *state;
3885 {
3886         char family;
3887         unsigned short port;
3888         char *buf, *bp;
3889         char *response;
3890         char *sockinfo = NULL;
3891         ssize_t s;
3892 # if NETINET6
3893         char buf6[INET6_ADDRSTRLEN];
3894 # endif /* NETINET6 */
3895
3896         if (tTd(64, 10))
3897                 sm_dprintf("milter_connect(%s)\n", hostname);
3898         if (MilterLogLevel > 9)
3899                 sm_syslog(LOG_INFO, e->e_id, "Milter: connect to filters");
3900
3901         /* gather data */
3902         switch (addr.sa.sa_family)
3903         {
3904 # if NETUNIX
3905           case AF_UNIX:
3906                 family = SMFIA_UNIX;
3907                 port = htons(0);
3908                 sockinfo = addr.sunix.sun_path;
3909                 break;
3910 # endif /* NETUNIX */
3911
3912 # if NETINET
3913           case AF_INET:
3914                 family = SMFIA_INET;
3915                 port = addr.sin.sin_port;
3916                 sockinfo = (char *) inet_ntoa(addr.sin.sin_addr);
3917                 break;
3918 # endif /* NETINET */
3919
3920 # if NETINET6
3921           case AF_INET6:
3922                 if (IN6_IS_ADDR_V4MAPPED(&addr.sin6.sin6_addr))
3923                         family = SMFIA_INET;
3924                 else
3925                         family = SMFIA_INET6;
3926                 port = addr.sin6.sin6_port;
3927                 sockinfo = anynet_ntop(&addr.sin6.sin6_addr, buf6,
3928                                        sizeof(buf6));
3929                 if (sockinfo == NULL)
3930                         sockinfo = "";
3931                 break;
3932 # endif /* NETINET6 */
3933
3934           default:
3935                 family = SMFIA_UNKNOWN;
3936                 break;
3937         }
3938
3939         s = strlen(hostname) + 1 + sizeof(family);
3940         if (family != SMFIA_UNKNOWN)
3941                 s += sizeof(port) + strlen(sockinfo) + 1;
3942
3943         buf = (char *) xalloc(s);
3944         bp = buf;
3945
3946         /* put together data */
3947         (void) memcpy(bp, hostname, strlen(hostname));
3948         bp += strlen(hostname);
3949         *bp++ = '\0';
3950         (void) memcpy(bp, &family, sizeof(family));
3951         bp += sizeof(family);
3952         if (family != SMFIA_UNKNOWN)
3953         {
3954                 (void) memcpy(bp, &port, sizeof(port));
3955                 bp += sizeof(port);
3956
3957                 /* include trailing '\0' */
3958                 (void) memcpy(bp, sockinfo, strlen(sockinfo) + 1);
3959         }
3960
3961         response = milter_command(SMFIC_CONNECT, buf, s, SMFIM_CONNECT,
3962                                 e, state, "connect", false);
3963         sm_free(buf); /* XXX */
3964
3965         /*
3966         **  If this message connection is done for,
3967         **  close the filters.
3968         */
3969
3970         if (*state != SMFIR_CONTINUE)
3971         {
3972                 if (MilterLogLevel > 9)
3973                         sm_syslog(LOG_INFO, e->e_id, "Milter: connect, ending");
3974                 milter_quit(e);
3975         }
3976         else
3977                 milter_per_connection_check(e);
3978
3979         /*
3980         **  SMFIR_REPLYCODE can't work with connect due to
3981         **  the requirements of SMTP.  Therefore, ignore the
3982         **  reply code text but keep the state it would reflect.
3983         */
3984
3985         if (*state == SMFIR_REPLYCODE)
3986         {
3987                 if (response != NULL &&
3988                     *response == '4')
3989                 {
3990                         if (strncmp(response, "421 ", 4) == 0)
3991                                 *state = SMFIR_SHUTDOWN;
3992                         else
3993                                 *state = SMFIR_TEMPFAIL;
3994                 }
3995                 else
3996                         *state = SMFIR_REJECT;
3997                 if (response != NULL)
3998                 {
3999                         sm_free(response); /* XXX */
4000                         response = NULL;
4001                 }
4002         }
4003         return response;
4004 }
4005
4006 /*
4007 **  MILTER_HELO -- send SMTP HELO/EHLO command info to milter filters
4008 **
4009 **      Parameters:
4010 **              helo -- argument to SMTP HELO/EHLO command.
4011 **              e -- current envelope.
4012 **              state -- return state from response.
4013 **
4014 **      Returns:
4015 **              response string (may be NULL)
4016 */
4017
4018 char *
4019 milter_helo(helo, e, state)
4020         char *helo;
4021         ENVELOPE *e;
4022         char *state;
4023 {
4024         int i;
4025         char *response;
4026
4027         if (tTd(64, 10))
4028                 sm_dprintf("milter_helo(%s)\n", helo);
4029
4030         /* HELO/EHLO can come at any point */
4031         for (i = 0; InputFilters[i] != NULL; i++)
4032         {
4033                 struct milter *m = InputFilters[i];
4034
4035                 switch (m->mf_state)
4036                 {
4037                   case SMFS_INMSG:
4038                         /* abort in message filters */
4039                         milter_abort_filter(m, e);
4040                         /* FALLTHROUGH */
4041
4042                   case SMFS_DONE:
4043                         /* reset done filters */
4044                         m->mf_state = SMFS_OPEN;
4045                         break;
4046                 }
4047         }
4048
4049         response = milter_command(SMFIC_HELO, helo, strlen(helo) + 1,
4050                                   SMFIM_HELO, e, state, "helo", false);
4051         milter_per_connection_check(e);
4052         return response;
4053 }
4054
4055 /*
4056 **  MILTER_ENVFROM -- send SMTP MAIL command info to milter filters
4057 **
4058 **      Parameters:
4059 **              args -- SMTP MAIL command args (args[0] == sender).
4060 **              e -- current envelope.
4061 **              state -- return state from response.
4062 **
4063 **      Returns:
4064 **              response string (may be NULL)
4065 */
4066
4067 char *
4068 milter_envfrom(args, e, state)
4069         char **args;
4070         ENVELOPE *e;
4071         char *state;
4072 {
4073         int i;
4074         char *buf, *bp;
4075         char *response;
4076         ssize_t s;
4077
4078         if (tTd(64, 10))
4079         {
4080                 sm_dprintf("milter_envfrom:");
4081                 for (i = 0; args[i] != NULL; i++)
4082                         sm_dprintf(" %s", args[i]);
4083                 sm_dprintf("\n");
4084         }
4085
4086         /* sanity check */
4087         if (args[0] == NULL)
4088         {
4089                 *state = SMFIR_REJECT;
4090                 if (MilterLogLevel > 10)
4091                         sm_syslog(LOG_INFO, e->e_id,
4092                                   "Milter: reject, no sender");
4093                 return NULL;
4094         }
4095
4096         /* new message, so ... */
4097         for (i = 0; InputFilters[i] != NULL; i++)
4098         {
4099                 struct milter *m = InputFilters[i];
4100
4101                 switch (m->mf_state)
4102                 {
4103                   case SMFS_INMSG:
4104                         /* abort in message filters */
4105                         milter_abort_filter(m, e);
4106                         /* FALLTHROUGH */
4107
4108                   case SMFS_DONE:
4109                         /* reset done filters */
4110                         m->mf_state = SMFS_OPEN;
4111                         break;
4112                 }
4113         }
4114
4115         /* put together data */
4116         s = 0;
4117         for (i = 0; args[i] != NULL; i++)
4118                 s += strlen(args[i]) + 1;
4119
4120         if (s < 0)
4121         {
4122                 *state = SMFIR_TEMPFAIL;
4123                 return NULL;
4124         }
4125
4126         buf = (char *) xalloc(s);
4127         bp = buf;
4128         for (i = 0; args[i] != NULL; i++)
4129         {
4130                 (void) sm_strlcpy(bp, args[i], s - (bp - buf));
4131                 bp += strlen(bp) + 1;
4132         }
4133
4134         if (MilterLogLevel > 14)
4135                 sm_syslog(LOG_INFO, e->e_id, "Milter: sender: %s", buf);
4136
4137         /* send it over */
4138         response = milter_command(SMFIC_MAIL, buf, s, SMFIM_ENVFROM,
4139                                 e, state, "mail", false);
4140         sm_free(buf); /* XXX */
4141
4142         /*
4143         **  If filter rejects/discards a per message command,
4144         **  abort the other filters since we are done with the
4145         **  current message.
4146         */
4147
4148         MILTER_CHECK_DONE_MSG();
4149         if (MilterLogLevel > 10 && *state == SMFIR_REJECT)
4150                 sm_syslog(LOG_INFO, e->e_id, "Milter: reject, sender");
4151         return response;
4152 }
4153
4154 /*
4155 **  MILTER_ENVRCPT -- send SMTP RCPT command info to milter filters
4156 **
4157 **      Parameters:
4158 **              args -- SMTP MAIL command args (args[0] == recipient).
4159 **              e -- current envelope.
4160 **              state -- return state from response.
4161 **              rcpt_error -- does RCPT have an error?
4162 **
4163 **      Returns:
4164 **              response string (may be NULL)
4165 */
4166
4167 char *
4168 milter_envrcpt(args, e, state, rcpt_error)
4169         char **args;
4170         ENVELOPE *e;
4171         char *state;
4172         bool rcpt_error;
4173 {
4174         int i;
4175         char *buf, *bp;
4176         char *response;
4177         ssize_t s;
4178
4179         if (tTd(64, 10))
4180         {
4181                 sm_dprintf("milter_envrcpt:");
4182                 for (i = 0; args[i] != NULL; i++)
4183                         sm_dprintf(" %s", args[i]);
4184                 sm_dprintf("\n");
4185         }
4186
4187         /* sanity check */
4188         if (args[0] == NULL)
4189         {
4190                 *state = SMFIR_REJECT;
4191                 if (MilterLogLevel > 10)
4192                         sm_syslog(LOG_INFO, e->e_id, "Milter: reject, no rcpt");
4193                 return NULL;
4194         }
4195
4196         /* put together data */
4197         s = 0;
4198         for (i = 0; args[i] != NULL; i++)
4199                 s += strlen(args[i]) + 1;
4200
4201         if (s < 0)
4202         {
4203                 *state = SMFIR_TEMPFAIL;
4204                 return NULL;
4205         }
4206
4207         buf = (char *) xalloc(s);
4208         bp = buf;
4209         for (i = 0; args[i] != NULL; i++)
4210         {
4211                 (void) sm_strlcpy(bp, args[i], s - (bp - buf));
4212                 bp += strlen(bp) + 1;
4213         }
4214
4215         if (MilterLogLevel > 14)
4216                 sm_syslog(LOG_INFO, e->e_id, "Milter: rcpts: %s", buf);
4217
4218         /* send it over */
4219         response = milter_command(SMFIC_RCPT, buf, s, SMFIM_ENVRCPT,
4220                                 e, state, "rcpt", rcpt_error);
4221         sm_free(buf); /* XXX */
4222         return response;
4223 }
4224
4225 /*
4226 **  MILTER_DATA_CMD -- send SMTP DATA command info to milter filters
4227 **
4228 **      Parameters:
4229 **              e -- current envelope.
4230 **              state -- return state from response.
4231 **
4232 **      Returns:
4233 **              response string (may be NULL)
4234 */
4235
4236 char *
4237 milter_data_cmd(e, state)
4238         ENVELOPE *e;
4239         char *state;
4240 {
4241         if (tTd(64, 10))
4242                 sm_dprintf("milter_data_cmd\n");
4243
4244         /* send it over */
4245         return milter_command(SMFIC_DATA, NULL, 0, SMFIM_DATA,
4246                                 e, state, "data", false);
4247 }
4248
4249 /*
4250 **  MILTER_DATA -- send message headers/body and gather final message results
4251 **
4252 **      Parameters:
4253 **              e -- current envelope.
4254 **              state -- return state from response.
4255 **
4256 **      Returns:
4257 **              response string (may be NULL)
4258 **
4259 **      Side effects:
4260 **              - Uses e->e_dfp for access to the body
4261 **              - Can call the various milter action routines to
4262 **                modify the envelope or message.
4263 */
4264
4265 /* flow through code using continue; don't wrap in do {} while */
4266 # define MILTER_CHECK_RESULTS() \
4267         if (m->mf_state == SMFS_ERROR && *state == SMFIR_CONTINUE) \
4268         { \
4269                         MILTER_SET_STATE;       \
4270         } \
4271         if (*state == SMFIR_ACCEPT || \
4272             m->mf_state == SMFS_DONE || \
4273             m->mf_state == SMFS_ERROR) \
4274         { \
4275                 if (m->mf_state != SMFS_ERROR) \
4276                         m->mf_state = SMFS_DONE; \
4277                 continue;       /* to next filter */ \
4278         } \
4279         if (*state != SMFIR_CONTINUE) \
4280         { \
4281                 m->mf_state = SMFS_DONE; \
4282                 goto finishup; \
4283         }
4284
4285 char *
4286 milter_data(e, state)
4287         ENVELOPE *e;
4288         char *state;
4289 {
4290         bool replbody = false;          /* milter_replbody() called? */
4291         bool replfailed = false;        /* milter_replbody() failed? */
4292         bool rewind = false;            /* rewind data file? */
4293         bool dfopen = false;            /* data file open for writing? */
4294         bool newfilter;                 /* reset on each new filter */
4295         char rcmd;
4296         int i;
4297         int save_errno;
4298         char *response = NULL;
4299         time_t eomsent;
4300         ssize_t rlen;
4301
4302         if (tTd(64, 10))
4303                 sm_dprintf("milter_data\n");
4304
4305         *state = SMFIR_CONTINUE;
4306
4307         /*
4308         **  XXX: Should actually send body chunks to each filter
4309         **  a chunk at a time instead of sending the whole body to
4310         **  each filter in turn.  However, only if the filters don't
4311         **  change the body.
4312         */
4313
4314         for (i = 0; InputFilters[i] != NULL; i++)
4315         {
4316                 int idx;
4317                 char **macros;
4318                 struct milter *m = InputFilters[i];
4319
4320                 if (*state != SMFIR_CONTINUE &&
4321                     *state != SMFIR_ACCEPT)
4322                 {
4323                         /*
4324                         **  A previous filter has dealt with the message,
4325                         **  safe to stop processing the filters.
4326                         */
4327
4328                         break;
4329                 }
4330
4331                 /* Now reset state for later evaluation */
4332                 *state = SMFIR_CONTINUE;
4333                 newfilter = true;
4334
4335                 /* previous problem? */
4336                 if (m->mf_state == SMFS_ERROR)
4337                 {
4338                         MILTER_CHECK_ERROR(false, continue);
4339                         break;
4340                 }
4341
4342                 /* sanity checks */
4343                 if (m->mf_sock < 0 ||
4344                     (m->mf_state != SMFS_OPEN && m->mf_state != SMFS_INMSG))
4345                         continue;
4346
4347                 m->mf_state = SMFS_INMSG;
4348
4349                 /* check if filter wants the headers */
4350                 if (!bitset(SMFIP_NOHDRS, m->mf_pflags))
4351                 {
4352                         response = milter_headers(m, e, state);
4353                         MILTER_CHECK_RESULTS();
4354                 }
4355
4356                 /* check if filter wants EOH */
4357                 if (!bitset(SMFIP_NOEOH, m->mf_pflags))
4358                 {
4359                         if (tTd(64, 10))
4360                                 sm_dprintf("milter_data: eoh\n");
4361
4362                         if ((m->mf_lflags & MI_LFLAGS_SYM(SMFIM_EOH)) != 0)
4363                                 idx = m->mf_idx;
4364                         else
4365                                 idx = 0;
4366                         SM_ASSERT(idx >= 0 && idx <= MAXFILTERS);
4367                         macros = MilterMacros[SMFIM_EOH][idx];
4368
4369                         if (macros != NULL)
4370                         {
4371                                 milter_send_macros(m, macros, SMFIC_EOH, e);
4372                                 MILTER_CHECK_RESULTS();
4373                         }
4374
4375                         /* send it over */
4376                         response = milter_send_command(m, SMFIC_EOH, NULL, 0,
4377                                                        e, state, "eoh");
4378                         MILTER_CHECK_RESULTS();
4379                 }
4380
4381                 /* check if filter wants the body */
4382                 if (!bitset(SMFIP_NOBODY, m->mf_pflags) &&
4383                     e->e_dfp != NULL)
4384                 {
4385                         rewind = true;
4386                         response = milter_body(m, e, state);
4387                         MILTER_CHECK_RESULTS();
4388                 }
4389
4390                 if ((m->mf_lflags & MI_LFLAGS_SYM(SMFIM_EOH)) != 0)
4391                         idx = m->mf_idx;
4392                 else
4393                         idx = 0;
4394                 SM_ASSERT(idx >= 0 && idx <= MAXFILTERS);
4395                 macros = MilterMacros[SMFIM_EOM][idx];
4396                 if (macros != NULL)
4397                 {
4398                         milter_send_macros(m, macros, SMFIC_BODYEOB, e);
4399                         MILTER_CHECK_RESULTS();
4400                 }
4401
4402                 /* send the final body chunk */
4403                 (void) milter_write(m, SMFIC_BODYEOB, NULL, 0,
4404                                     m->mf_timeout[SMFTO_WRITE], e, "eom");
4405
4406                 /* Get time EOM sent for timeout */
4407                 eomsent = curtime();
4408
4409                 /* deal with the possibility of multiple responses */
4410                 while (*state == SMFIR_CONTINUE)
4411                 {
4412                         /* Check total timeout from EOM to final ACK/NAK */
4413                         if (m->mf_timeout[SMFTO_EOM] > 0 &&
4414                             curtime() - eomsent >= m->mf_timeout[SMFTO_EOM])
4415                         {
4416                                 if (tTd(64, 5))
4417                                         sm_dprintf("milter_data(%s): EOM ACK/NAK timeout\n",
4418                                                 m->mf_name);
4419                                 if (MilterLogLevel > 0)
4420                                         sm_syslog(LOG_ERR, e->e_id,
4421                                                   "milter_data(%s): EOM ACK/NAK timeout",
4422                                                   m->mf_name);
4423                                 milter_error(m, e);
4424                                 MILTER_CHECK_ERROR(false, break);
4425                                 break;
4426                         }
4427
4428                         response = milter_read(m, &rcmd, &rlen,
4429                                                m->mf_timeout[SMFTO_READ], e,
4430                                                 "eom");
4431                         if (m->mf_state == SMFS_ERROR)
4432                                 break;
4433
4434                         if (tTd(64, 10))
4435                                 sm_dprintf("milter_data(%s): state %c\n",
4436                                            m->mf_name, (char) rcmd);
4437
4438                         switch (rcmd)
4439                         {
4440                           case SMFIR_REPLYCODE:
4441                                 MILTER_CHECK_REPLYCODE("554 5.7.1 Command rejected");
4442                                 if (MilterLogLevel > 12)
4443                                         sm_syslog(LOG_INFO, e->e_id, "milter=%s, reject=%s",
4444                                                   m->mf_name, response);
4445                                 *state = rcmd;
4446                                 m->mf_state = SMFS_DONE;
4447                                 break;
4448
4449                           case SMFIR_REJECT: /* log msg at end of function */
4450                                 if (MilterLogLevel > 12)
4451                                         sm_syslog(LOG_INFO, e->e_id, "milter=%s, reject",
4452                                                   m->mf_name);
4453                                 *state = rcmd;
4454                                 m->mf_state = SMFS_DONE;
4455                                 break;
4456
4457                           case SMFIR_DISCARD:
4458                                 if (MilterLogLevel > 12)
4459                                         sm_syslog(LOG_INFO, e->e_id, "milter=%s, discard",
4460                                                   m->mf_name);
4461                                 *state = rcmd;
4462                                 m->mf_state = SMFS_DONE;
4463                                 break;
4464
4465                           case SMFIR_TEMPFAIL:
4466                                 if (MilterLogLevel > 12)
4467                                         sm_syslog(LOG_INFO, e->e_id, "milter=%s, tempfail",
4468                                                   m->mf_name);
4469                                 *state = rcmd;
4470                                 m->mf_state = SMFS_DONE;
4471                                 break;
4472
4473                           case SMFIR_CONTINUE:
4474                           case SMFIR_ACCEPT:
4475                                 /* this filter is done with message */
4476                                 if (replfailed)
4477                                         *state = SMFIR_TEMPFAIL;
4478                                 else
4479                                         *state = SMFIR_ACCEPT;
4480                                 m->mf_state = SMFS_DONE;
4481                                 break;
4482
4483                           case SMFIR_PROGRESS:
4484                                 break;
4485
4486                           case SMFIR_QUARANTINE:
4487                                 if (!bitset(SMFIF_QUARANTINE, m->mf_fflags))
4488                                 {
4489                                         if (MilterLogLevel > 9)
4490                                                 sm_syslog(LOG_WARNING, e->e_id,
4491                                                           "milter_data(%s): lied about quarantining, honoring request anyway",
4492                                                           m->mf_name);
4493                                 }
4494                                 if (response == NULL)
4495                                         response = newstr("");
4496                                 if (MilterLogLevel > 3)
4497                                         sm_syslog(LOG_INFO, e->e_id,
4498                                                   "milter=%s, quarantine=%s",
4499                                                   m->mf_name, response);
4500                                 e->e_quarmsg = sm_rpool_strdup_x(e->e_rpool,
4501                                                                  response);
4502                                 macdefine(&e->e_macro, A_PERM,
4503                                           macid("{quarantine}"), e->e_quarmsg);
4504                                 break;
4505
4506                           case SMFIR_ADDHEADER:
4507                                 if (!bitset(SMFIF_ADDHDRS, m->mf_fflags))
4508                                 {
4509                                         if (MilterLogLevel > 9)
4510                                                 sm_syslog(LOG_WARNING, e->e_id,
4511                                                           "milter_data(%s): lied about adding headers, honoring request anyway",
4512                                                           m->mf_name);
4513                                 }
4514                                 milter_addheader(m, response, rlen, e);
4515                                 break;
4516
4517                           case SMFIR_INSHEADER:
4518                                 if (!bitset(SMFIF_ADDHDRS, m->mf_fflags))
4519                                 {
4520                                         if (MilterLogLevel > 9)
4521                                                 sm_syslog(LOG_WARNING, e->e_id,
4522                                                           "milter_data(%s): lied about adding headers, honoring request anyway",
4523                                                           m->mf_name);
4524                                 }
4525                                 milter_insheader(m, response, rlen, e);
4526                                 break;
4527
4528                           case SMFIR_CHGHEADER:
4529                                 if (!bitset(SMFIF_CHGHDRS, m->mf_fflags))
4530                                 {
4531                                         if (MilterLogLevel > 9)
4532                                                 sm_syslog(LOG_WARNING, e->e_id,
4533                                                           "milter_data(%s): lied about changing headers, honoring request anyway",
4534                                                           m->mf_name);
4535                                 }
4536                                 milter_changeheader(m, response, rlen, e);
4537                                 break;
4538
4539                           case SMFIR_CHGFROM:
4540                                 if (!bitset(SMFIF_CHGFROM, m->mf_fflags))
4541                                 {
4542                                         if (MilterLogLevel > 9)
4543                                                 sm_syslog(LOG_WARNING, e->e_id,
4544                                                           "milter_data(%s) lied about changing sender, honoring request anyway",
4545                                                           m->mf_name);
4546                                 }
4547                                 milter_chgfrom(response, rlen, e);
4548                                 break;
4549
4550                           case SMFIR_ADDRCPT:
4551                                 if (!bitset(SMFIF_ADDRCPT, m->mf_fflags))
4552                                 {
4553                                         if (MilterLogLevel > 9)
4554                                                 sm_syslog(LOG_WARNING, e->e_id,
4555                                                           "milter_data(%s) lied about adding recipients, honoring request anyway",
4556                                                           m->mf_name);
4557                                 }
4558                                 milter_addrcpt(response, rlen, e);
4559                                 break;
4560
4561                           case SMFIR_ADDRCPT_PAR:
4562                                 if (!bitset(SMFIF_ADDRCPT_PAR, m->mf_fflags))
4563                                 {
4564                                         if (MilterLogLevel > 9)
4565                                                 sm_syslog(LOG_WARNING, e->e_id,
4566                                                           "milter_data(%s) lied about adding recipients with parameters, honoring request anyway",
4567                                                           m->mf_name);
4568                                 }
4569                                 milter_addrcpt_par(response, rlen, e);
4570                                 break;
4571
4572                           case SMFIR_DELRCPT:
4573                                 if (!bitset(SMFIF_DELRCPT, m->mf_fflags))
4574                                 {
4575                                         if (MilterLogLevel > 9)
4576                                                 sm_syslog(LOG_WARNING, e->e_id,
4577                                                           "milter_data(%s): lied about removing recipients, honoring request anyway",
4578                                                           m->mf_name);
4579                                 }
4580                                 milter_delrcpt(response, rlen, e);
4581                                 break;
4582
4583                           case SMFIR_REPLBODY:
4584                                 if (!bitset(SMFIF_MODBODY, m->mf_fflags))
4585                                 {
4586                                         if (MilterLogLevel > 0)
4587                                                 sm_syslog(LOG_ERR, e->e_id,
4588                                                           "milter_data(%s): lied about replacing body, rejecting request and tempfailing message",
4589                                                           m->mf_name);
4590                                         replfailed = true;
4591                                         break;
4592                                 }
4593
4594                                 /* already failed in attempt */
4595                                 if (replfailed)
4596                                         break;
4597
4598                                 if (!dfopen)
4599                                 {
4600                                         if (milter_reopen_df(e) < 0)
4601                                         {
4602                                                 replfailed = true;
4603                                                 break;
4604                                         }
4605                                         dfopen = true;
4606                                         rewind = true;
4607                                 }
4608
4609                                 if (milter_replbody(response, rlen,
4610                                                     newfilter, e) < 0)
4611                                         replfailed = true;
4612                                 newfilter = false;
4613                                 replbody = true;
4614                                 break;
4615
4616                           default:
4617                                 /* Invalid response to command */
4618                                 if (MilterLogLevel > 0)
4619                                         sm_syslog(LOG_ERR, e->e_id,
4620                                                   "milter_data(%s): returned bogus response %c",
4621                                                   m->mf_name, rcmd);
4622                                 milter_error(m, e);
4623                                 break;
4624                         }
4625                         if (rcmd != SMFIR_REPLYCODE && response != NULL)
4626                         {
4627                                 sm_free(response); /* XXX */
4628                                 response = NULL;
4629                         }
4630
4631                         if (m->mf_state == SMFS_ERROR)
4632                                 break;
4633                 }
4634
4635                 if (replbody && !replfailed)
4636                 {
4637                         /* flush possible buffered character */
4638                         milter_replbody(NULL, 0, !replbody, e);
4639                         replbody = false;
4640                 }
4641
4642                 if (m->mf_state == SMFS_ERROR)
4643                 {
4644                         MILTER_CHECK_ERROR(false, continue);
4645                         goto finishup;
4646                 }
4647         }
4648
4649 finishup:
4650         /* leave things in the expected state if we touched it */
4651         if (replfailed)
4652         {
4653                 if (*state == SMFIR_CONTINUE ||
4654                     *state == SMFIR_ACCEPT)
4655                 {
4656                         *state = SMFIR_TEMPFAIL;
4657                         SM_FREE_CLR(response);
4658                 }
4659
4660                 if (dfopen)
4661                 {
4662                         (void) sm_io_close(e->e_dfp, SM_TIME_DEFAULT);
4663                         e->e_dfp = NULL;
4664                         e->e_flags &= ~EF_HAS_DF;
4665                         dfopen = false;
4666                 }
4667                 rewind = false;
4668         }
4669
4670         if ((dfopen && milter_reset_df(e) < 0) ||
4671             (rewind && bfrewind(e->e_dfp) < 0))
4672         {
4673                 save_errno = errno;
4674                 ExitStat = EX_IOERR;
4675
4676                 /*
4677                 **  If filter told us to keep message but we had
4678                 **  an error, we can't really keep it, tempfail it.
4679                 */
4680
4681                 if (*state == SMFIR_CONTINUE ||
4682                     *state == SMFIR_ACCEPT)
4683                 {
4684                         *state = SMFIR_TEMPFAIL;
4685                         SM_FREE_CLR(response);
4686                 }
4687
4688                 errno = save_errno;
4689                 syserr("milter_data: %s/%cf%s: read error",
4690                        qid_printqueue(e->e_qgrp, e->e_qdir),
4691                        DATAFL_LETTER, e->e_id);
4692         }
4693
4694         MILTER_CHECK_DONE_MSG();
4695         if (MilterLogLevel > 10 && *state == SMFIR_REJECT)
4696                 sm_syslog(LOG_INFO, e->e_id, "Milter: reject, data");
4697         return response;
4698 }
4699
4700 /*
4701 **  MILTER_UNKNOWN -- send any unrecognized or unimplemented command
4702 **                      string to milter filters
4703 **
4704 **      Parameters:
4705 **              smtpcmd -- the string itself.
4706 **              e -- current envelope.
4707 **              state -- return state from response.
4708 **
4709 **
4710 **      Returns:
4711 **              response string (may be NULL)
4712 */
4713
4714 char *
4715 milter_unknown(smtpcmd, e, state)
4716         char *smtpcmd;
4717         ENVELOPE *e;
4718         char *state;
4719 {
4720         if (tTd(64, 10))
4721                 sm_dprintf("milter_unknown(%s)\n", smtpcmd);
4722
4723         return milter_command(SMFIC_UNKNOWN, smtpcmd, strlen(smtpcmd) + 1,
4724                                 SMFIM_NOMACROS, e, state, "unknown", false);
4725 }
4726
4727 /*
4728 **  MILTER_QUIT -- informs the filter(s) we are done and closes connection(s)
4729 **
4730 **      Parameters:
4731 **              e -- current envelope.
4732 **
4733 **      Returns:
4734 **              none
4735 */
4736
4737 void
4738 milter_quit(e)
4739         ENVELOPE *e;
4740 {
4741         int i;
4742
4743         if (tTd(64, 10))
4744                 sm_dprintf("milter_quit(%s)\n", e->e_id);
4745
4746         for (i = 0; InputFilters[i] != NULL; i++)
4747                 milter_quit_filter(InputFilters[i], e);
4748 }
4749
4750 /*
4751 **  MILTER_ABORT -- informs the filter(s) that we are aborting current message
4752 **
4753 **      Parameters:
4754 **              e -- current envelope.
4755 **
4756 **      Returns:
4757 **              none
4758 */
4759
4760 void
4761 milter_abort(e)
4762         ENVELOPE *e;
4763 {
4764         int i;
4765
4766         if (tTd(64, 10))
4767                 sm_dprintf("milter_abort\n");
4768
4769         for (i = 0; InputFilters[i] != NULL; i++)
4770         {
4771                 struct milter *m = InputFilters[i];
4772
4773                 /* sanity checks */
4774                 if (m->mf_sock < 0 || m->mf_state != SMFS_INMSG)
4775                         continue;
4776
4777                 milter_abort_filter(m, e);
4778         }
4779 }
4780 #endif /* MILTER */