git.sur5r.net Git - bacula/bacula/blob - bacula/src/tools/bsmtp.c

   1 /*
   2    Copyright (C) 2001-2005 Kern Sibbald
   3
   4    This program is free software; you can redistribute it and/or
   5    modify it under the terms of the GNU General Public License
   6    version 2 as amended with additional clauses defined in the
   7    file LICENSE in the main source directory.
   8
   9    This program is distributed in the hope that it will be useful,
  10    but WITHOUT ANY WARRANTY; without even the implied warranty of
  11    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  12    the file LICENSE for additional details.
  13
  14  */
  15 /*
  16    Derived from a SMTPclient:
  17
  18        SMTPclient -- simple SMTP client
  19
  20        Copyright (C) 1997 Ralf S. Engelschall, All Rights Reserved.
  21        rse@engelschall.com
  22        www.engelschall.com
  23
  24    Kern Sibbald, July 2001
  25
  26    Version $Id$
  27
  28  */
  29
  30 #ifdef APCUPSD
  31
  32 #include "apc.h"
  33 #undef main
  34 #define my_name_is(x)
  35 #define bstrdup(x) strdup(x)
  36 UPSINFO myUPS;
  37 UPSINFO *core_ups = &myUPS;
  38 #define MY_NAME "smtp"
  39
  40 #else
  41
  42 #include "bacula.h"
  43 #include "jcr.h"
  44 #define MY_NAME "bsmtp"
  45
  46 #endif
  47
  48 /* Dummy functions */
  49 int generate_daemon_event(JCR *jcr, const char *event)
  50    { return 1; }
  51
  52 #ifndef MAXSTRING
  53 #define MAXSTRING 254
  54 #endif
  55
  56 static FILE *sfp;
  57 static FILE *rfp;
  58
  59 static char *from_addr = NULL;
  60 static char *cc_addr = NULL;
  61 static char *subject = NULL;
  62 static char *err_addr = NULL;
  63 static const char *mailhost = NULL;
  64 static char *reply_addr = NULL;
  65 static int mailport = 25;
  66 static char my_hostname[MAXSTRING];
  67
  68
  69 /*
  70  *  examine message from server
  71  */
  72 static void get_response(void)
  73 {
  74     char buf[MAXSTRING];
  75
  76     Dmsg0(50, "Calling fgets on read socket rfp.\n");
  77     buf[3] = 0;
  78     while (fgets(buf, sizeof(buf), rfp)) {
  79         int len = strlen(buf);
  80         if (len > 0) {
  81            buf[len-1] = 0;
  82         }
  83         Dmsg2(10, "%s --> %s\n", mailhost, buf);
  84         if (!isdigit((int)buf[0]) || buf[0] > '3') {
  85             Pmsg2(0, "Fatal malformed reply from %s: %s\n", mailhost, buf);
  86             exit(1);
  87         }
  88         if (buf[3] != '-') {
  89             break;
  90         }
  91     }
  92     return;
  93 }
  94
  95 /*
  96  *  say something to server and check the response
  97  */
  98 static void chat(const char *fmt, ...)
  99 {
 100     va_list ap;
 101
 102     va_start(ap, fmt);
 103     vfprintf(sfp, fmt, ap);
 104     if (debug_level >= 10) {
 105        fprintf(stdout, "%s --> ", my_hostname);
 106        vfprintf(stdout, fmt, ap);
 107     }
 108     va_end(ap);
 109
 110     fflush(sfp);
 111     if (debug_level >= 10) {
 112        fflush(stdout);
 113     }
 114     get_response();
 115 }
 116
 117
 118 static void usage()
 119 {
 120    fprintf(stderr,
 121 "\n"
 122 "Usage: %s [-f from] [-h mailhost] [-s subject] [-c copy] [recipient ...]\n"
 123 "       -c          set the Cc: field\n"
 124 "       -dnn        set debug level to nn\n"
 125 "       -f          set the From: field\n"
 126 "       -h          use mailhost:port as the SMTP server\n"
 127 "       -s          set the Subject: field\n"
 128 "       -?          print this message.\n"
 129 "\n", MY_NAME);
 130
 131    exit(1);
 132 }
 133
 134
 135 /*********************************************************************
 136  *
 137  *  Program to send email
 138  */
 139 int main (int argc, char *argv[])
 140 {
 141     char buf[MAXSTRING];
 142     struct sockaddr_in sin;
 143     struct hostent *hp;
 144     int s, r, i, ch;
 145     struct passwd *pwd;
 146     char *cp, *p;
 147     time_t now = time(NULL);
 148     struct tm tm;
 149
 150    my_name_is(argc, argv, "bsmtp");
 151
 152    while ((ch = getopt(argc, argv, "c:d:f:h:r:s:?")) != -1) {
 153       switch (ch) {
 154       case 'c':
 155          Dmsg1(20, "cc=%s\n", optarg);
 156          cc_addr = optarg;
 157          break;
 158
 159       case 'd':                    /* set debug level */
 160          debug_level = atoi(optarg);
 161          if (debug_level <= 0) {
 162             debug_level = 1;
 163          }
 164          Dmsg1(20, "Debug level = %d\n", debug_level);
 165          break;
 166
 167       case 'f':                    /* from */
 168          from_addr = optarg;
 169          break;
 170
 171       case 'h':                    /* smtp host */
 172          Dmsg1(20, "host=%s\n", optarg);
 173          p = strchr(optarg, ':');
 174          if (p) {
 175             *p++ = 0;
 176             mailport = atoi(p);
 177          }
 178          mailhost = optarg;
 179          break;
 180
 181       case 's':                    /* subject */
 182          Dmsg1(20, "subject=%s\n", optarg);
 183          subject = optarg;
 184          break;
 185
 186       case 'r':                    /* reply address */
 187          reply_addr = optarg;
 188          break;
 189
 190       case '?':
 191       default:
 192          usage();
 193
 194       }
 195    }
 196    argc -= optind;
 197    argv += optind;
 198
 199    if (argc < 1) {
 200       Pmsg0(0, "Fatal error: no recipient given.\n");
 201       usage();
 202       exit(1);
 203    }
 204
 205    /*
 206     *  Determine SMTP server
 207     */
 208    if (mailhost == NULL) {
 209       if ((cp = getenv("SMTPSERVER")) != NULL) {
 210          mailhost = cp;
 211       } else {
 212          mailhost = "localhost";
 213       }
 214    }
 215
 216    /*
 217     *  Find out my own host name for HELO;
 218     *  if possible, get the fully qualified domain name
 219     */
 220    if (gethostname(my_hostname, sizeof(my_hostname) - 1) < 0) {
 221       Pmsg1(0, "Fatal gethostname error: ERR=%s\n", strerror(errno));
 222       exit(1);
 223    }
 224    if ((hp = gethostbyname(my_hostname)) == NULL) {
 225       Pmsg2(0, "Fatal gethostbyname for myself failed \"%s\": ERR=%s\n", my_hostname,
 226          strerror(errno));
 227       exit(1);
 228    }
 229    strcpy(my_hostname, hp->h_name);
 230    Dmsg1(20, "My hostname is: %s\n", my_hostname);
 231
 232    /*
 233     *  Determine from address.
 234     */
 235    if (from_addr == NULL) {
 236       if ((pwd = getpwuid(getuid())) == 0) {
 237          sprintf(buf, "userid-%d@%s", (int)getuid(), my_hostname);
 238       } else {
 239          sprintf(buf, "%s@%s", pwd->pw_name, my_hostname);
 240       }
 241       from_addr = bstrdup(buf);
 242    }
 243    Dmsg1(20, "From addr=%s\n", from_addr);
 244
 245    /*
 246     *  Connect to smtp daemon on mailhost.
 247     */
 248 hp:
 249    if ((hp = gethostbyname(mailhost)) == NULL) {
 250       Pmsg2(0, "Error unknown mail host \"%s\": ERR=%s\n", mailhost,
 251          strerror(errno));
 252       if (strcasecmp(mailhost, "localhost") != 0) {
 253          Pmsg0(0, "Retrying connection using \"localhost\".\n");
 254          mailhost = "localhost";
 255          goto hp;
 256       }
 257       exit(1);
 258    }
 259
 260    if (hp->h_addrtype != AF_INET) {
 261       Pmsg1(0, "Fatal error: Unknown address family for smtp host: %d\n", hp->h_addrtype);
 262       exit(1);
 263    }
 264    memset((char *)&sin, 0, sizeof(sin));
 265    memcpy((char *)&sin.sin_addr, hp->h_addr, hp->h_length);
 266    sin.sin_family = hp->h_addrtype;
 267    sin.sin_port = htons(mailport);
 268    if ((s = socket(AF_INET, SOCK_STREAM, 0)) < 0) {
 269       Pmsg1(0, "Fatal socket error: ERR=%s\n", strerror(errno));
 270       exit(1);
 271    }
 272    if (connect(s, (struct sockaddr *)&sin, sizeof(sin)) < 0) {
 273       Pmsg2(0, "Fatal connect error to %s: ERR=%s\n", mailhost, strerror(errno));
 274       exit(1);
 275    }
 276    Dmsg0(20, "Connected\n");
 277    if ((r = dup(s)) < 0) {
 278       Pmsg1(0, "Fatal dup error: ERR=%s\n", strerror(errno));
 279       exit(1);
 280    }
 281    if ((sfp = fdopen(s, "w")) == 0) {
 282       Pmsg1(0, "Fatal fdopen error: ERR=%s\n", strerror(errno));
 283       exit(1);
 284    }
 285    if ((rfp = fdopen(r, "r")) == 0) {
 286       Pmsg1(0, "Fatal fdopen error: ERR=%s\n", strerror(errno));
 287       exit(1);
 288    }
 289
 290    /*
 291     *  Send SMTP headers
 292     */
 293    get_response(); /* banner */
 294    chat("helo %s\r\n", my_hostname);
 295    chat("mail from:<%s>\r\n", from_addr);
 296
 297    for (i = 0; i < argc; i++) {
 298       Dmsg1(20, "rcpt to: %s\n", argv[i]);
 299       chat("rcpt to:<%s>\r\n", argv[i]);
 300    }
 301
 302    if (cc_addr) {
 303       chat("rcpt to:<%s>\r\n", cc_addr);
 304    }
 305    Dmsg0(20, "Data\n");
 306    chat("data\r\n");
 307
 308    /*
 309     *  Send message header
 310     */
 311    fprintf(sfp, "From: %s\r\n", from_addr);
 312    if (subject) {
 313       fprintf(sfp, "Subject: %s\r\n", subject);
 314    }
 315    if (reply_addr) {
 316       fprintf(sfp, "Reply-To: %s\r\n", reply_addr);
 317    }
 318    if (err_addr) {
 319       fprintf(sfp, "Errors-To: %s\r\n", err_addr);
 320    }
 321    if ((pwd = getpwuid(getuid())) == 0) {
 322       fprintf(sfp, "Sender: userid-%d@%s\r\n", (int)getuid(), my_hostname);
 323    } else {
 324       fprintf(sfp, "Sender: %s@%s\r\n", pwd->pw_name, my_hostname);
 325    }
 326
 327    fprintf(sfp, "To: %s", argv[0]);
 328    for (i = 1; i < argc; i++) {
 329       fprintf(sfp, ",%s", argv[i]);
 330    }
 331
 332    fprintf(sfp, "\r\n");
 333    if (cc_addr) {
 334       fprintf(sfp, "Cc: %s\r\n", cc_addr);
 335    }
 336
 337    /* Add RFC822 date */
 338    localtime_r(&now, &tm);
 339    strftime(buf, sizeof(buf), "%a, %d %b %Y %H:%M:%S %z", &tm);
 340    fprintf(sfp, "Date: %s\r\n", buf);
 341
 342    fprintf(sfp, "\r\n");
 343
 344    /*
 345     *  Send message body
 346     */
 347    while (fgets(buf, sizeof(buf), stdin)) {
 348       buf[strlen(buf)-1] = 0;
 349       if (strcmp(buf, ".") == 0) { /* quote lone dots */
 350          fprintf(sfp, "..\r\n");
 351       } else {                     /* pass body through unchanged */
 352          fprintf(sfp, "%s\r\n", buf);
 353       }
 354    }
 355
 356    /*
 357     *  Send SMTP quit command
 358     */
 359    chat(".\r\n");
 360    chat("quit\r\n");
 361
 362    /*
 363     *  Go away gracefully ...
 364     */
 365    exit(0);
 366 }