Clean up process wait to use POSIX waitpid() interface.
[mmh] / uip / whom.c
1
2 /*
3  * whom.c -- report to whom a message would be sent
4  *
5  * This code is Copyright (c) 2002, by the authors of nmh.  See the
6  * COPYRIGHT file in the root directory of the nmh distribution for
7  * complete copyright information.
8  */
9
10 #include <h/mh.h>
11 #include <h/signals.h>
12 #include <signal.h>
13
14 #ifndef CYRUS_SASL
15 # define SASLminc(a) (a)
16 #else /* CYRUS_SASL */
17 # define SASLminc(a)  0
18 #endif /* CYRUS_SASL */
19
20 #ifndef TLS_SUPPORT
21 # define TLSminc(a)  (a)
22 #else /* TLS_SUPPORT */
23 # define TLSminc(a)   0
24 #endif /* TLS_SUPPORT */
25
26 static struct swit switches[] = {
27 #define ALIASW              0
28     { "alias aliasfile", 0 },
29 #define CHKSW               1
30     { "check", 0 },
31 #define NOCHKSW             2
32     { "nocheck", 0 },
33 #define DRAFTSW             3
34     { "draft", 0 },
35 #define DFOLDSW             4
36     { "draftfolder +folder", 6 },
37 #define DMSGSW              5
38     { "draftmessage msg", 6 },
39 #define NDFLDSW             6
40     { "nodraftfolder", 0 },
41 #define VERSIONSW           7
42     { "version", 0 },
43 #define HELPSW              8
44     { "help", 0 },
45 #define CLIESW              9
46     { "client host", -6 },
47 #define SERVSW             10
48     { "server host", -6 },
49 #define SNOOPSW            11
50     { "snoop", -5 },
51 #define SASLSW             12
52     { "sasl", SASLminc(4) },
53 #define SASLMECHSW         13
54     { "saslmech mechanism", SASLminc(-5) },
55 #define USERSW             14
56     { "user username", SASLminc(-4) },
57 #define PORTSW             15
58     { "port server port name/number", 4 },
59 #define TLSSW              16
60     { "tls", TLSminc(-3) },
61     { NULL, 0 }
62 };
63
64
65 int
66 main (int argc, char **argv)
67 {
68     pid_t child_id;
69     int i, status, isdf = 0;
70     int distsw = 0, vecp = 0;
71     char *cp, *dfolder = NULL, *dmsg = NULL;
72     char *msg = NULL, **ap, **argp, backup[BUFSIZ];
73     char buf[BUFSIZ], **arguments, *vec[MAXARGS];
74
75 #ifdef LOCALE
76     setlocale(LC_ALL, "");
77 #endif
78     invo_name = r1bindex (argv[0], '/');
79
80     /* read user profile/context */
81     context_read();
82
83     arguments = getarguments (invo_name, argc, argv, 1);
84     argp = arguments;
85
86     vec[vecp++] = invo_name;
87     vec[vecp++] = "-whom";
88     vec[vecp++] = "-library";
89     vec[vecp++] = getcpy (m_maildir (""));
90
91     while ((cp = *argp++)) {
92         if (*cp == '-') {
93             switch (smatch (++cp, switches)) {
94                 case AMBIGSW: 
95                     ambigsw (cp, switches);
96                     done (1);
97                 case UNKWNSW: 
98                     adios (NULL, "-%s unknown", cp);
99
100                 case HELPSW: 
101                     snprintf (buf, sizeof(buf), "%s [switches] [file]", invo_name);
102                     print_help (buf, switches, 1);
103                     done (1);
104                 case VERSIONSW:
105                     print_version(invo_name);
106                     done (1);
107
108                 case CHKSW: 
109                 case NOCHKSW: 
110                 case SNOOPSW:
111                 case SASLSW:
112                     vec[vecp++] = --cp;
113                     continue;
114
115                 case DRAFTSW:
116                     msg = draft;
117                     continue;
118
119                 case DFOLDSW: 
120                     if (dfolder)
121                         adios (NULL, "only one draft folder at a time!");
122                     if (!(cp = *argp++) || *cp == '-')
123                         adios (NULL, "missing argument to %s", argp[-2]);
124                     dfolder = path (*cp == '+' || *cp == '@' ? cp + 1 : cp,
125                             *cp != '@' ? TFOLDER : TSUBCWF);
126                     continue;
127                 case DMSGSW: 
128                     if (dmsg)
129                         adios (NULL, "only one draft message at a time!");
130                     if (!(dmsg = *argp++) || *dmsg == '-')
131                         adios (NULL, "missing argument to %s", argp[-2]);
132                     continue;
133                 case NDFLDSW: 
134                     dfolder = NULL;
135                     isdf = NOTOK;
136                     continue;
137
138                 case ALIASW: 
139                 case CLIESW: 
140                 case SERVSW: 
141                 case USERSW:
142                 case PORTSW:
143                 case SASLMECHSW:
144                     vec[vecp++] = --cp;
145                     if (!(cp = *argp++) || *cp == '-')
146                         adios (NULL, "missing argument to %s", argp[-2]);
147                     vec[vecp++] = cp;
148                     continue;
149             }
150         }
151         if (msg)
152             adios (NULL, "only one draft at a time!");
153         else
154             vec[vecp++] = msg = cp;
155     }
156
157     /* allow Aliasfile: profile entry */
158     if ((cp = context_find ("Aliasfile"))) {
159         char *dp = NULL;
160
161         for (ap = brkstring(dp = getcpy(cp), " ", "\n"); ap && *ap; ap++) {
162             vec[vecp++] = "-alias";
163             vec[vecp++] = *ap;
164         }
165     }
166
167     if (msg == NULL) {
168 #ifdef  WHATNOW
169         if (dfolder || (cp = getenv ("mhdraft")) == NULL || *cp == '\0')
170 #endif  /* WHATNOW */
171             cp  = getcpy (m_draft (dfolder, dmsg, 1, &isdf));
172         msg = vec[vecp++] = cp;
173     }
174     if ((cp = getenv ("mhdist"))
175             && *cp
176             && (distsw = atoi (cp))
177             && (cp = getenv ("mhaltmsg"))
178             && *cp) {
179         if (distout (msg, cp, backup) == NOTOK)
180             done (1);
181         vec[vecp++] = "-dist";
182         distsw++;
183     }
184     vec[vecp] = NULL;
185
186     closefds (3);
187
188     if (distsw) {
189         for (i = 0; (child_id = fork()) == NOTOK && i < 5; i++)
190             sleep (5);
191     }
192
193     switch (distsw ? child_id : OK) {
194         case NOTOK:
195             advise (NULL, "unable to fork, so checking directly...");
196         case OK:
197             execvp (postproc, vec);
198             fprintf (stderr, "unable to exec ");
199             perror (postproc);
200             _exit (-1);
201
202         default:
203             SIGNAL (SIGHUP, SIG_IGN);
204             SIGNAL (SIGINT, SIG_IGN);
205             SIGNAL (SIGQUIT, SIG_IGN);
206             SIGNAL (SIGTERM, SIG_IGN);
207
208             status = pidwait(child_id, OK);
209
210             unlink (msg);
211             if (rename (backup, msg) == NOTOK)
212                 adios (msg, "unable to rename %s to", backup);
213             done (status);
214     }
215
216     return 0;  /* dead code to satisfy the compiler */
217 }