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    setlocale(LC_ALL, "");
 151    bindtextdomain("bacula", LOCALEDIR);
 152    textdomain("bacula");
 153
 154    my_name_is(argc, argv, "bsmtp");
 155
 156    while ((ch = getopt(argc, argv, "c:d:f:h:r:s:?")) != -1) {
 157       switch (ch) {
 158       case 'c':
 159          Dmsg1(20, "cc=%s\n", optarg);
 160          cc_addr = optarg;
 161          break;
 162
 163       case 'd':                    /* set debug level */
 164          debug_level = atoi(optarg);
 165          if (debug_level <= 0) {
 166             debug_level = 1;
 167          }
 168          Dmsg1(20, "Debug level = %d\n", debug_level);
 169          break;
 170
 171       case 'f':                    /* from */
 172          from_addr = optarg;
 173          break;
 174
 175       case 'h':                    /* smtp host */
 176          Dmsg1(20, "host=%s\n", optarg);
 177          p = strchr(optarg, ':');
 178          if (p) {
 179             *p++ = 0;
 180             mailport = atoi(p);
 181          }
 182          mailhost = optarg;
 183          break;
 184
 185       case 's':                    /* subject */
 186          Dmsg1(20, "subject=%s\n", optarg);
 187          subject = optarg;
 188          break;
 189
 190       case 'r':                    /* reply address */
 191          reply_addr = optarg;
 192          break;
 193
 194       case '?':
 195       default:
 196          usage();
 197
 198       }
 199    }
 200    argc -= optind;
 201    argv += optind;
 202
 203    if (argc < 1) {
 204       Pmsg0(0, _("Fatal error: no recipient given.\n"));
 205       usage();
 206       exit(1);
 207    }
 208
 209    /*
 210     *  Determine SMTP server
 211     */
 212    if (mailhost == NULL) {
 213       if ((cp = getenv("SMTPSERVER")) != NULL) {
 214          mailhost = cp;
 215       } else {
 216          mailhost = "localhost";
 217       }
 218    }
 219
 220    /*
 221     *  Find out my own host name for HELO;
 222     *  if possible, get the fully qualified domain name
 223     */
 224    if (gethostname(my_hostname, sizeof(my_hostname) - 1) < 0) {
 225       Pmsg1(0, _("Fatal gethostname error: ERR=%s\n"), strerror(errno));
 226       exit(1);
 227    }
 228    if ((hp = gethostbyname(my_hostname)) == NULL) {
 229       Pmsg2(0, _("Fatal gethostbyname for myself failed \"%s\": ERR=%s\n"), my_hostname,
 230          strerror(errno));
 231       exit(1);
 232    }
 233    strcpy(my_hostname, hp->h_name);
 234    Dmsg1(20, "My hostname is: %s\n", my_hostname);
 235
 236    /*
 237     *  Determine from address.
 238     */
 239    if (from_addr == NULL) {
 240       if ((pwd = getpwuid(getuid())) == 0) {
 241          sprintf(buf, "userid-%d@%s", (int)getuid(), my_hostname);
 242       } else {
 243          sprintf(buf, "%s@%s", pwd->pw_name, my_hostname);
 244       }
 245       from_addr = bstrdup(buf);
 246    }
 247    Dmsg1(20, "From addr=%s\n", from_addr);
 248
 249    /*
 250     *  Connect to smtp daemon on mailhost.
 251     */
 252 hp:
 253    if ((hp = gethostbyname(mailhost)) == NULL) {
 254       Pmsg2(0, _("Error unknown mail host \"%s\": ERR=%s\n"), mailhost,
 255          strerror(errno));
 256       if (strcasecmp(mailhost, "localhost") != 0) {
 257          Pmsg0(0, _("Retrying connection using \"localhost\".\n"));
 258          mailhost = "localhost";
 259          goto hp;
 260       }
 261       exit(1);
 262    }
 263
 264    if (hp->h_addrtype != AF_INET) {
 265       Pmsg1(0, _("Fatal error: Unknown address family for smtp host: %d\n"), hp->h_addrtype);
 266       exit(1);
 267    }
 268    memset((char *)&sin, 0, sizeof(sin));
 269    memcpy((char *)&sin.sin_addr, hp->h_addr, hp->h_length);
 270    sin.sin_family = hp->h_addrtype;
 271    sin.sin_port = htons(mailport);
 272    if ((s = socket(AF_INET, SOCK_STREAM, 0)) < 0) {
 273       Pmsg1(0, _("Fatal socket error: ERR=%s\n"), strerror(errno));
 274       exit(1);
 275    }
 276    if (connect(s, (struct sockaddr *)&sin, sizeof(sin)) < 0) {
 277       Pmsg2(0, _("Fatal connect error to %s: ERR=%s\n"), mailhost, strerror(errno));
 278       exit(1);
 279    }
 280    Dmsg0(20, "Connected\n");
 281    if ((r = dup(s)) < 0) {
 282       Pmsg1(0, _("Fatal dup error: ERR=%s\n"), strerror(errno));
 283       exit(1);
 284    }
 285    if ((sfp = fdopen(s, "w")) == 0) {
 286       Pmsg1(0, _("Fatal fdopen error: ERR=%s\n"), strerror(errno));
 287       exit(1);
 288    }
 289    if ((rfp = fdopen(r, "r")) == 0) {
 290       Pmsg1(0, _("Fatal fdopen error: ERR=%s\n"), strerror(errno));
 291       exit(1);
 292    }
 293
 294    /*
 295     *  Send SMTP headers
 296     */
 297    get_response(); /* banner */
 298    chat("helo %s\r\n", my_hostname);
 299    chat("mail from:<%s>\r\n", from_addr);
 300
 301    for (i = 0; i < argc; i++) {
 302       Dmsg1(20, "rcpt to: %s\n", argv[i]);
 303       chat("rcpt to:<%s>\r\n", argv[i]);
 304    }
 305
 306    if (cc_addr) {
 307       chat("rcpt to:<%s>\r\n", cc_addr);
 308    }
 309    Dmsg0(20, "Data\n");
 310    chat("data\r\n");
 311
 312    /*
 313     *  Send message header
 314     */
 315    fprintf(sfp, "From: %s\r\n", from_addr);
 316    if (subject) {
 317       fprintf(sfp, "Subject: %s\r\n", subject);
 318    }
 319    if (reply_addr) {
 320       fprintf(sfp, "Reply-To: %s\r\n", reply_addr);
 321    }
 322    if (err_addr) {
 323       fprintf(sfp, "Errors-To: %s\r\n", err_addr);
 324    }
 325    if ((pwd = getpwuid(getuid())) == 0) {
 326       fprintf(sfp, "Sender: userid-%d@%s\r\n", (int)getuid(), my_hostname);
 327    } else {
 328       fprintf(sfp, "Sender: %s@%s\r\n", pwd->pw_name, my_hostname);
 329    }
 330
 331    fprintf(sfp, "To: %s", argv[0]);
 332    for (i = 1; i < argc; i++) {
 333       fprintf(sfp, ",%s", argv[i]);
 334    }
 335
 336    fprintf(sfp, "\r\n");
 337    if (cc_addr) {
 338       fprintf(sfp, "Cc: %s\r\n", cc_addr);
 339    }
 340
 341    /* Add RFC822 date */
 342    localtime_r(&now, &tm);
 343    strftime(buf, sizeof(buf), "%a, %d %b %Y %H:%M:%S %z", &tm);
 344    fprintf(sfp, "Date: %s\r\n", buf);
 345
 346    fprintf(sfp, "\r\n");
 347
 348    /*
 349     *  Send message body
 350     */
 351    while (fgets(buf, sizeof(buf), stdin)) {
 352       buf[strlen(buf)-1] = 0;
 353       if (strcmp(buf, ".") == 0) { /* quote lone dots */
 354          fprintf(sfp, "..\r\n");
 355       } else {                     /* pass body through unchanged */
 356          fprintf(sfp, "%s\r\n", buf);
 357       }
 358    }
 359
 360    /*
 361     *  Send SMTP quit command
 362     */
 363    chat(".\r\n");
 364    chat("quit\r\n");
 365
 366    /*
 367     *  Go away gracefully ...
 368     */
 369    exit(0);
 370 }