citadel/citmail.c

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