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 }