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