777edb701f662e3999765ed87fa92e6022ae5b68
[citadel] / citadel / utils / citmail.c
1 /*
2  * This program attempts to act like a local MDA if you're using sendmail or
3  * some other non-Citadel MTA.  It basically just contacts the Citadel LMTP
4  * listener on a unix domain socket and transmits the message.
5  *
6  * Copyright (c) 1987-2021 by the citadel.org team
7  *
8  * This program is open source software; you can redistribute it and/or modify
9  * it under the terms of the GNU General Public License version 3.
10  *
11  * This program is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  * GNU General Public License for more details.
15  */
16
17 #include "sysdep.h"
18 #include <stdlib.h>
19 #include <unistd.h>
20 #include <ctype.h>
21 #include <stdio.h>
22 #include <signal.h>
23 #include <sys/types.h>
24 #include <sys/socket.h>
25 #include <sys/un.h>
26 #include <netdb.h>
27 #include <string.h>
28 #include <pwd.h>
29 #include <errno.h>
30 #include <stdarg.h>
31 #include <limits.h>
32 #include <libcitadel.h>
33 #include "citadel.h"
34 #include "citadel_dirs.h"
35
36 int serv_sock;
37 int debug = 0;
38
39 void strip_trailing_nonprint(char *buf)
40 {
41         while ( (!IsEmptyStr(buf)) && (!isprint(buf[strlen(buf) - 1])) )
42                 buf[strlen(buf) - 1] = 0;
43 }
44
45
46 void timeout(int signum)
47 {
48         exit(signum);
49 }
50
51
52 int uds_connectsock(char *sockpath)
53 {
54         int s;
55         struct sockaddr_un addr;
56
57         memset(&addr, 0, sizeof(addr));
58         addr.sun_family = AF_UNIX;
59         strncpy(addr.sun_path, sockpath, sizeof addr.sun_path);
60
61         s = socket(AF_UNIX, SOCK_STREAM, 0);
62         if (s < 0) {
63                 fprintf(stderr, "citmail: Can't create socket: %s\n",
64                         strerror(errno));
65                 exit(3);
66         }
67
68         if (connect(s, (struct sockaddr *) &addr, sizeof(addr)) < 0) {
69                 fprintf(stderr, "citmail: can't connect: %s\n",
70                         strerror(errno));
71                 close(s);
72                 exit(3);
73         }
74
75         return s;
76 }
77
78
79 /*
80  * input binary data from socket
81  */
82 void serv_read(char *buf, int bytes)
83 {
84         int len, rlen;
85
86         len = 0;
87         while (len < bytes) {
88                 rlen = read(serv_sock, &buf[len], bytes - len);
89                 if (rlen < 1) {
90                         return;
91                 }
92                 len = len + rlen;
93         }
94 }
95
96
97 /*
98  * send binary to server
99  */
100 void serv_write(char *buf, int nbytes)
101 {
102         int bytes_written = 0;
103         int retval;
104         while (bytes_written < nbytes) {
105                 retval = write(serv_sock, &buf[bytes_written],
106                                nbytes - bytes_written);
107                 if (retval < 1) {
108                         return;
109                 }
110                 bytes_written = bytes_written + retval;
111         }
112 }
113
114
115
116 /*
117  * input string from socket - implemented in terms of serv_read()
118  */
119 void serv_gets(char *buf)
120 {
121         int i;
122
123         /* Read one character at a time.
124          */
125         for (i = 0;; i++) {
126                 serv_read(&buf[i], 1);
127                 if (buf[i] == '\n' || i == (SIZ-1))
128                         break;
129         }
130
131         /* If we got a long line, discard characters until the newline.
132          */
133         if (i == (SIZ-1))
134                 while (buf[i] != '\n')
135                         serv_read(&buf[i], 1);
136
137         /* Strip all trailing nonprintables (crlf)
138          */
139         buf[i] = 0;
140         strip_trailing_nonprint(buf);
141         if (debug) fprintf(stderr, "> %s\n", buf);
142 }
143
144
145 /*
146  * send line to server - implemented in terms of serv_write()
147  */
148 void serv_puts(char *buf)
149 {
150         if (debug) fprintf(stderr, "< %s\n", buf);
151         serv_write(buf, strlen(buf));
152         serv_write("\n", 1);
153 }
154
155
156
157 void cleanup(int exitcode) {
158         char buf[1024];
159
160         if (exitcode != 0) {
161                 fprintf(stderr, "citmail: error #%d occurred while sending mail.\n", exitcode);
162                 fprintf(stderr, "Please check your Citadel configuration.\n");
163         }
164         serv_puts("QUIT");
165         serv_gets(buf);
166         exit(exitcode);
167 }
168
169
170
171 int main(int argc, char **argv) {
172         char buf[1024];
173         char fromline[1024];
174         FILE *fp;
175         int i;
176         struct passwd *pw;
177         int from_header = 0;
178         int in_body = 0;
179         char ctdldir[PATH_MAX]=CTDLDIR;
180         char *sp, *ep;
181         char hostname[256];
182         char **recipients = NULL;
183         int num_recipients = 0;
184         int to_or_cc = 0;
185         int read_recipients_from_headers = 0;
186         char *add_these_recipients = NULL;
187
188         for (i=1; i<argc; ++i) {
189                 if (!strcmp(argv[i], "-d")) {
190                         debug = 1;
191                 }
192                 else if (!strcmp(argv[i], "-t")) {
193                         read_recipients_from_headers = 1;
194                 }
195                 else if (!strncmp(argv[i], "-h", 2)) {
196                         safestrncpy(ctdldir, &argv[i][2], sizeof ctdldir);
197                 }
198                 else if (argv[i][0] != '-') {
199                         ++num_recipients;
200                         recipients = realloc(recipients, (num_recipients * sizeof (char *)));
201                         recipients[num_recipients - 1] = strdup(argv[i]);
202                 }
203         }
204
205         if (chdir(ctdldir) != 0) {
206                 fprintf(stderr, "sendcommand: %s: %s\n", ctdldir, strerror(errno));
207                 exit(errno);
208         }
209                
210         pw = getpwuid(getuid());
211
212         fp = tmpfile();
213         if (fp == NULL) return(errno);
214         serv_sock = uds_connectsock(file_lmtp_socket);
215         serv_gets(buf);
216         if (buf[0] != '2') {
217                 fprintf(stderr, "%s\n", &buf[4]);
218                 if (debug) fprintf(stderr, "citmail: could not connect to LMTP socket.\n");
219                 cleanup(1);
220         }
221
222         sp = strchr (buf, ' ');
223         if (sp == NULL) {
224                 if (debug) fprintf(stderr, "citmail: could not calculate hostname.\n");
225                 cleanup(2);
226         }
227         sp ++;
228         ep = strchr (sp, ' ');
229         if (ep == NULL) {
230                 if (debug) fprintf(stderr, "citmail: error parsing hostname\n");
231                 cleanup(3);
232         }
233         else
234                 *ep = '\0';
235
236         strncpy(hostname, sp, sizeof hostname);
237
238         snprintf(fromline, sizeof fromline, "From: %s@%s", pw->pw_name, hostname);
239         while (fgets(buf, 1024, stdin) != NULL) {
240                 if ( ( (buf[0] == 13) || (buf[0] == 10)) && (in_body == 0) ) {
241                         in_body = 1;
242                         if (from_header == 0) {
243                                 fprintf(fp, "%s%s", fromline, buf);
244                         }
245                 }
246                 if (in_body == 0 && !strncasecmp(buf, "From:", 5)) {
247                         strcpy(fromline, buf);
248                         from_header = 1;
249                 }
250
251                 if (read_recipients_from_headers) {
252                         add_these_recipients = NULL;
253                         if ((isspace(buf[0])) && (to_or_cc)) {
254                                 add_these_recipients = buf;
255                         }
256                         else {
257                                 if ((!strncasecmp(buf, "To:", 3)) || (!strncasecmp(buf, "Cc:", 3))) {
258                                         to_or_cc = 1;
259                                 }
260                                 else {
261                                         to_or_cc = 0;
262                                 }
263                                 if (to_or_cc) {
264                                         add_these_recipients = &buf[3];
265                                 }
266                         }
267
268                         if (add_these_recipients) {
269                                 int num_recp_on_this_line;
270                                 char this_recp[256];
271
272                                 num_recp_on_this_line = num_tokens(add_these_recipients, ',');
273                                 for (i=0; i<num_recp_on_this_line; ++i) {
274                                         extract_token(this_recp, add_these_recipients,
275                                                 i, ',', sizeof this_recp);
276                                         striplt(this_recp);
277                                         if (!IsEmptyStr(this_recp)) {
278                                                 ++num_recipients;
279                                                 recipients = realloc(recipients,
280                                                         (num_recipients * sizeof (char *)));
281                                                 recipients[num_recipients - 1] = strdup(this_recp);
282                                         }
283                                 }
284                         }
285                 }
286
287                 fprintf(fp, "%s", buf);
288         }
289         strip_trailing_nonprint(fromline);
290
291         sprintf(buf, "LHLO %s", hostname);
292         serv_puts(buf);
293         do {
294                 serv_gets(buf);
295                 strcat(buf, "    ");
296         } while (buf[3] == '-');
297         if (buf[0] != '2') {
298                 if (debug) fprintf(stderr, "citmail: LHLO command failed\n");
299                 cleanup(4);
300         }
301
302         snprintf(buf, sizeof buf, "MAIL %s", fromline);
303         serv_puts(buf);
304         serv_gets(buf);
305         if (buf[0] != '2') {
306                 if (debug) fprintf(stderr, "citmail: MAIL command failed\n");
307                 cleanup(5);
308         }
309
310         for (i=0; i<num_recipients; ++i) {
311                 snprintf(buf, sizeof buf, "RCPT To: %s", recipients[i]);
312                 serv_puts(buf);
313                 serv_gets(buf);
314                 free(recipients[i]);
315         }
316         free(recipients);
317
318         serv_puts("DATA");
319         serv_gets(buf);
320         if (buf[0]!='3') {
321                 if (debug) fprintf(stderr, "citmail: DATA command failed\n");
322                 cleanup(6);
323         }
324
325         rewind(fp);
326         while (fgets(buf, sizeof buf, fp) != NULL) {
327                 strip_trailing_nonprint(buf);
328                 serv_puts(buf);
329         }
330         serv_puts(".");
331         serv_gets(buf);
332         if (buf[0] != '2') {
333                 fprintf(stderr, "%s\n", &buf[4]);
334                 cleanup(7);
335         }
336         else {
337                 cleanup(0);
338         }
339
340         /* We won't actually reach this statement but the compiler will
341          * display a spurious warning about an invalid return type if
342          * we don't return an int.
343          */
344         return(0);
345 }