libexec/dma/dma.c

   1 /*
   2  * Copyright (c) 2008 The DragonFly Project.  All rights reserved.
   3  *
   4  * This code is derived from software contributed to The DragonFly Project
   5  * by Simon 'corecode' Schubert <corecode@fs.ei.tum.de>.
   6  *
   7  * Redistribution and use in source and binary forms, with or without
   8  * modification, are permitted provided that the following conditions
   9  * are met:
  10  *
  11  * 1. Redistributions of source code must retain the above copyright
  12  *    notice, this list of conditions and the following disclaimer.
  13  * 2. Redistributions in binary form must reproduce the above copyright
  14  *    notice, this list of conditions and the following disclaimer in
  15  *    the documentation and/or other materials provided with the
  16  *    distribution.
  17  * 3. Neither the name of The DragonFly Project nor the names of its
  18  *    contributors may be used to endorse or promote products derived
  19  *    from this software without specific, prior written permission.
  20  *
  21  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  22  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  23  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
  24  * FOR A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE
  25  * COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
  26  * INCIDENTAL, SPECIAL, EXEMPLARY OR CONSEQUENTIAL DAMAGES (INCLUDING,
  27  * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  28  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
  29  * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
  30  * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
  31  * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  32  * SUCH DAMAGE.
  33  *
  34  * $DragonFly: src/libexec/dma/dma.c,v 1.5 2008/09/30 17:47:21 swildner Exp $
  35  */
  36
  37 #include <sys/param.h>
  38 #include <sys/queue.h>
  39 #include <sys/stat.h>
  40 #include <sys/types.h>
  41 #include <sys/wait.h>
  42
  43 #ifdef HAVE_CRYPTO
  44 #include <openssl/ssl.h>
  45 #endif /* HAVE_CRYPTO */
  46
  47 #include <dirent.h>
  48 #include <err.h>
  49 #include <errno.h>
  50 #include <fcntl.h>
  51 #include <inttypes.h>
  52 #include <netdb.h>
  53 #include <paths.h>
  54 #include <pwd.h>
  55 #include <signal.h>
  56 #include <stdarg.h>
  57 #include <stdio.h>
  58 #include <stdlib.h>
  59 #include <string.h>
  60 #include <syslog.h>
  61 #include <unistd.h>
  62
  63 #include "dma.h"
  64
  65
  66
  67 static void deliver(struct qitem *);
  68 static int add_recp(struct queue *, const char *, const char *, int);
  69
  70 struct aliases aliases = LIST_HEAD_INITIALIZER(aliases);
  71 static struct strlist tmpfs = SLIST_HEAD_INITIALIZER(tmpfs);
  72 struct virtusers virtusers = LIST_HEAD_INITIALIZER(virtusers);
  73 struct authusers authusers = LIST_HEAD_INITIALIZER(authusers);
  74 static int daemonize = 1;
  75 struct config *config;
  76 static struct strlist seenmsg[16][16];
  77
  78
  79 char *
  80 hostname(void)
  81 {
  82         static char name[MAXHOSTNAMELEN+1];
  83
  84         if (gethostname(name, sizeof(name)) != 0)
  85                 strcpy(name, "(unknown hostname)");
  86
  87         return name;
  88 }
  89
  90 static char *
  91 set_from(const char *osender)
  92 {
  93         struct virtuser *v;
  94         char *sender;
  95
  96         if ((config->features & VIRTUAL) != 0) {
  97                 SLIST_FOREACH(v, &virtusers, next) {
  98                         if (strcmp(v->login, getlogin()) == 0) {
  99                                 sender = strdup(v->address);
 100                                 if (sender == NULL)
 101                                         return(NULL);
 102                                 goto out;
 103                         }
 104                 }
 105         }
 106
 107         if (osender) {
 108                 sender = strdup(osender);
 109                 if (sender == NULL)
 110                         return (NULL);
 111         } else {
 112                 if (asprintf(&sender, "%s@%s", getlogin(), hostname()) <= 0)
 113                         return (NULL);
 114         }
 115
 116         if (strchr(sender, '\n') != NULL) {
 117                 errno = EINVAL;
 118                 return (NULL);
 119         }
 120
 121 out:
 122         return (sender);
 123 }
 124
 125 static int
 126 read_aliases(void)
 127 {
 128         yyin = fopen(config->aliases, "r");
 129         if (yyin == NULL)
 130                 return (0);     /* not fatal */
 131         if (yyparse())
 132                 return (-1);    /* fatal error, probably malloc() */
 133         fclose(yyin);
 134         return (0);
 135 }
 136
 137 static int
 138 add_recp(struct queue *queue, const char *str, const char *sender, int expand)
 139 {
 140         struct qitem *it, *tit;
 141         struct stritem *sit;
 142         struct alias *al;
 143         struct passwd *pw;
 144         char *host;
 145         int aliased = 0;
 146
 147         it = calloc(1, sizeof(*it));
 148         if (it == NULL)
 149                 return (-1);
 150         it->addr = strdup(str);
 151         if (it->addr == NULL)
 152                 return (-1);
 153
 154         it->sender = sender;
 155         host = strrchr(it->addr, '@');
 156         if (host != NULL &&
 157             (strcmp(host + 1, hostname()) == 0 ||
 158              strcmp(host + 1, "localhost") == 0)) {
 159                 *host = 0;
 160         }
 161         LIST_FOREACH(tit, &queue->queue, next) {
 162                 /* weed out duplicate dests */
 163                 if (strcmp(tit->addr, it->addr) == 0) {
 164                         free(it->addr);
 165                         free(it);
 166                         return (0);
 167                 }
 168         }
 169         LIST_INSERT_HEAD(&queue->queue, it, next);
 170         if (strrchr(it->addr, '@') == NULL) {
 171                 it->remote = 0;
 172                 if (expand) {
 173                         LIST_FOREACH(al, &aliases, next) {
 174                                 if (strcmp(al->alias, it->addr) != 0)
 175                                         continue;
 176                                 SLIST_FOREACH(sit, &al->dests, next) {
 177                                         if (add_recp(queue, sit->str, sender, 1) != 0)
 178                                                 return (-1);
 179                                 }
 180                                 aliased = 1;
 181                         }
 182                         if (aliased) {
 183                                 LIST_REMOVE(it, next);
 184                         } else {
 185                                 /* Local destination, check */
 186                                 pw = getpwnam(it->addr);
 187                                 if (pw == NULL)
 188                                         goto out;
 189                                 endpwent();
 190                         }
 191                 }
 192         } else {
 193                 it->remote = 1;
 194         }
 195
 196         return (0);
 197
 198 out:
 199         free(it->addr);
 200         free(it);
 201         return (-1);
 202 }
 203
 204 static void
 205 deltmp(void)
 206 {
 207         struct stritem *t;
 208
 209         SLIST_FOREACH(t, &tmpfs, next) {
 210                 unlink(t->str);
 211         }
 212 }
 213
 214 static int
 215 gentempf(struct queue *queue)
 216 {
 217         char fn[PATH_MAX+1];
 218         struct stritem *t;
 219         int fd;
 220
 221         if (snprintf(fn, sizeof(fn), "%s/%s", config->spooldir, "tmp_XXXXXXXXXX") <= 0)
 222                 return (-1);
 223         fd = mkstemp(fn);
 224         if (fd < 0)
 225                 return (-1);
 226         queue->mailfd = fd;
 227         queue->tmpf = strdup(fn);
 228         if (queue->tmpf == NULL) {
 229                 unlink(fn);
 230                 return (-1);
 231         }
 232         t = malloc(sizeof(*t));
 233         if (t != NULL) {
 234                 t->str = queue->tmpf;
 235                 SLIST_INSERT_HEAD(&tmpfs, t, next);
 236         }
 237         return (0);
 238 }
 239
 240 static int
 241 open_locked(const char *fname, int flags)
 242 {
 243 #ifndef O_EXLOCK
 244         int fd, save_errno;
 245
 246         fd = open(fname, flags, 0);
 247         if (fd < 0)
 248                 return(fd);
 249         if (flock(fd, LOCK_EX|((flags & O_NONBLOCK)? LOCK_NB: 0)) < 0) {
 250                 save_errno = errno;
 251                 close(fd);
 252                 errno = save_errno;
 253                 return(-1);
 254         }
 255         return(fd);
 256 #else
 257         return(open(fname, flags|O_EXLOCK));
 258 #endif
 259 }
 260
 261 /*
 262  * spool file format:
 263  *
 264  * envelope-from
 265  * queue-id1 envelope-to1
 266  * queue-id2 envelope-to2
 267  * ...
 268  * <empty line>
 269  * mail data
 270  *
 271  * queue ids are unique, formed from the inode of the spool file
 272  * and a unique identifier.
 273  */
 274 static int
 275 preparespool(struct queue *queue, const char *sender)
 276 {
 277         char line[1000];        /* by RFC2822 */
 278         struct stat st;
 279         int error;
 280         struct qitem *it;
 281         FILE *queuef;
 282         off_t hdrlen;
 283
 284         error = snprintf(line, sizeof(line), "%s\n", sender);
 285         if (error < 0 || (size_t)error >= sizeof(line)) {
 286                 errno = E2BIG;
 287                 return (-1);
 288         }
 289         if (write(queue->mailfd, line, error) != error)
 290                 return (-1);
 291
 292         queuef = fdopen(queue->mailfd, "r+");
 293         if (queuef == NULL)
 294                 return (-1);
 295
 296         /*
 297          * Assign queue id to each dest.
 298          */
 299         if (fstat(queue->mailfd, &st) != 0)
 300                 return (-1);
 301         queue->id = st.st_ino;
 302         LIST_FOREACH(it, &queue->queue, next) {
 303                 if (asprintf(&it->queueid, "%"PRIxMAX".%"PRIxPTR,
 304                              queue->id, (uintptr_t)it) <= 0)
 305                         return (-1);
 306                 if (asprintf(&it->queuefn, "%s/%s",
 307                              config->spooldir, it->queueid) <= 0)
 308                         return (-1);
 309                 /* File may not exist yet */
 310                 if (stat(it->queuefn, &st) == 0)
 311                         return (-1);
 312                 it->queuef = queuef;
 313                 error = snprintf(line, sizeof(line), "%s %s\n",
 314                                it->queueid, it->addr);
 315                 if (error < 0 || (size_t)error >= sizeof(line))
 316                         return (-1);
 317                 if (write(queue->mailfd, line, error) != error)
 318                         return (-1);
 319         }
 320         line[0] = '\n';
 321         if (write(queue->mailfd, line, 1) != 1)
 322                 return (-1);
 323
 324         hdrlen = lseek(queue->mailfd, 0, SEEK_CUR);
 325         LIST_FOREACH(it, &queue->queue, next) {
 326                 it->hdrlen = hdrlen;
 327         }
 328         return (0);
 329 }
 330
 331 static char *
 332 rfc822date(void)
 333 {
 334         static char str[50];
 335         size_t error;
 336         time_t now;
 337
 338         now = time(NULL);
 339         error = strftime(str, sizeof(str), "%a, %d %b %Y %T %z",
 340                        localtime(&now));
 341         if (error == 0)
 342                 strcpy(str, "(date fail)");
 343         return (str);
 344 }
 345
 346 static int
 347 readmail(struct queue *queue, const char *sender, int nodot)
 348 {
 349         char line[1000];        /* by RFC2822 */
 350         size_t linelen;
 351         int error;
 352
 353         error = snprintf(line, sizeof(line), "\
 354 Received: from %s (uid %d)\n\
 355 \t(envelope-from %s)\n\
 356 \tid %"PRIxMAX"\n\
 357 \tby %s (%s)\n\
 358 \t%s\n",
 359                 getlogin(), getuid(),
 360                 sender,
 361                 queue->id,
 362                 hostname(), VERSION,
 363                 rfc822date());
 364         if (error < 0 || (size_t)error >= sizeof(line))
 365                 return (-1);
 366         if (write(queue->mailfd, line, error) != error)
 367                 return (-1);
 368
 369         while (!feof(stdin)) {
 370                 if (fgets(line, sizeof(line), stdin) == NULL)
 371                         break;
 372                 linelen = strlen(line);
 373                 if (linelen == 0 || line[linelen - 1] != '\n') {
 374                         errno = EINVAL;         /* XXX mark permanent errors */
 375                         return (-1);
 376                 }
 377                 if (!nodot && linelen == 2 && line[0] == '.')
 378                         break;
 379                 if ((size_t)write(queue->mailfd, line, linelen) != linelen)
 380                         return (-1);
 381         }
 382         if (fsync(queue->mailfd) != 0)
 383                 return (-1);
 384         return (0);
 385 }
 386
 387 static int
 388 linkspool(struct queue *queue)
 389 {
 390         struct qitem *it;
 391
 392         LIST_FOREACH(it, &queue->queue, next) {
 393                 if (link(queue->tmpf, it->queuefn) != 0)
 394                         goto delfiles;
 395         }
 396         unlink(queue->tmpf);
 397         return (0);
 398
 399 delfiles:
 400         LIST_FOREACH(it, &queue->queue, next) {
 401                 unlink(it->queuefn);
 402         }
 403         return (-1);
 404 }
 405
 406 static struct qitem *
 407 go_background(struct queue *queue)
 408 {
 409         struct sigaction sa;
 410         struct qitem *it;
 411         pid_t pid;
 412
 413         if (daemonize && daemon(0, 0) != 0) {
 414                 syslog(LOG_ERR, "can not daemonize: %m");
 415                 exit(1);
 416         }
 417         daemonize = 0;
 418
 419         bzero(&sa, sizeof(sa));
 420         sa.sa_flags = SA_NOCLDWAIT;
 421         sa.sa_handler = SIG_IGN;
 422         sigaction(SIGCHLD, &sa, NULL);
 423
 424         LIST_FOREACH(it, &queue->queue, next) {
 425                 /* No need to fork for the last dest */
 426                 if (LIST_NEXT(it, next) == NULL)
 427                         return (it);
 428
 429                 pid = fork();
 430                 switch (pid) {
 431                 case -1:
 432                         syslog(LOG_ERR, "can not fork: %m");
 433                         exit(1);
 434                         break;
 435
 436                 case 0:
 437                         /*
 438                          * Child:
 439                          *
 440                          * return and deliver mail
 441                          */
 442                         return (it);
 443
 444                 default:
 445                         /*
 446                          * Parent:
 447                          *
 448                          * fork next child
 449                          */
 450                         break;
 451                 }
 452         }
 453
 454         syslog(LOG_CRIT, "reached dead code");
 455         exit(1);
 456 }
 457
 458 static void
 459 bounce(struct qitem *it, const char *reason)
 460 {
 461         struct queue bounceq;
 462         struct qitem *bit;
 463         char line[1000];
 464         int error;
 465
 466         /* Don't bounce bounced mails */
 467         if (it->sender[0] == 0) {
 468                 syslog(LOG_CRIT, "%s: delivery panic: can't bounce a bounce",
 469                        it->queueid);
 470                 exit(1);
 471         }
 472
 473         syslog(LOG_ERR, "%s: delivery failed, bouncing",
 474                it->queueid);
 475
 476         LIST_INIT(&bounceq.queue);
 477         if (add_recp(&bounceq, it->sender, "", 1) != 0)
 478                 goto fail;
 479         if (gentempf(&bounceq) != 0)
 480                 goto fail;
 481         if (preparespool(&bounceq, "") != 0)
 482                 goto fail;
 483
 484         bit = LIST_FIRST(&bounceq.queue);
 485         error = fprintf(bit->queuef, "\
 486 Received: from MAILER-DAEMON\n\
 487 \tid %"PRIxMAX"\n\
 488 \tby %s (%s)\n\
 489 \t%s\n\
 490 X-Original-To: <%s>\n\
 491 From: MAILER-DAEMON <>\n\
 492 To: %s\n\
 493 Subject: Mail delivery failed\n\
 494 Message-Id: <%"PRIxMAX"@%s>\n\
 495 Date: %s\n\
 496 \n\
 497 This is the %s at %s.\n\
 498 \n\
 499 There was an error delivering your mail to <%s>.\n\
 500 \n\
 501 %s\n\
 502 \n\
 503 Message headers follow.\n\
 504 \n\
 505 ",
 506                 bounceq.id,
 507                 hostname(), VERSION,
 508                 rfc822date(),
 509                 it->addr,
 510                 it->sender,
 511                 bounceq.id, hostname(),
 512                 rfc822date(),
 513                 VERSION, hostname(),
 514                 it->addr,
 515                 reason);
 516         if (error < 0)
 517                 goto fail;
 518         if (fflush(bit->queuef) != 0)
 519                 goto fail;
 520
 521         if (fseek(it->queuef, it->hdrlen, SEEK_SET) != 0)
 522                 goto fail;
 523         while (!feof(it->queuef)) {
 524                 if (fgets(line, sizeof(line), it->queuef) == NULL)
 525                         break;
 526                 if (line[0] == '\n')
 527                         break;
 528                 if ((size_t)write(bounceq.mailfd, line, strlen(line)) != strlen(line))
 529                         goto fail;
 530         }
 531         if (fsync(bounceq.mailfd) != 0)
 532                 goto fail;
 533         if (linkspool(&bounceq) != 0)
 534                 goto fail;
 535         /* bounce is safe */
 536
 537         unlink(it->queuefn);
 538         fclose(it->queuef);
 539
 540         bit = go_background(&bounceq);
 541         deliver(bit);
 542         /* NOTREACHED */
 543
 544 fail:
 545         syslog(LOG_CRIT, "%s: error creating bounce: %m", it->queueid);
 546         unlink(it->queuefn);
 547         exit(1);
 548 }
 549
 550 static int
 551 deliver_local(struct qitem *it, const char **errmsg)
 552 {
 553         char fn[PATH_MAX+1];
 554         char line[1000];
 555         size_t linelen;
 556         int mbox;
 557         int error;
 558         off_t mboxlen;
 559         time_t now = time(NULL);
 560
 561         error = snprintf(fn, sizeof(fn), "%s/%s", _PATH_MAILDIR, it->addr);
 562         if (error < 0 || (size_t)error >= sizeof(fn)) {
 563                 syslog(LOG_ERR, "%s: local delivery deferred: %m",
 564                        it->queueid);
 565                 return (1);
 566         }
 567
 568         /* mailx removes users mailspool file if empty, so open with O_CREAT */
 569         mbox = open_locked(fn, O_WRONLY | O_APPEND | O_CREAT);
 570         if (mbox < 0) {
 571                 syslog(LOG_ERR, "%s: local delivery deferred: can not open `%s': %m",
 572                        it->queueid, fn);
 573                 return (1);
 574         }
 575         mboxlen = lseek(mbox, 0, SEEK_CUR);
 576
 577         if (fseek(it->queuef, it->hdrlen, SEEK_SET) != 0) {
 578                 syslog(LOG_ERR, "%s: local delivery deferred: can not seek: %m",
 579                        it->queueid);
 580                 return (1);
 581         }
 582
 583         error = snprintf(line, sizeof(line), "From %s\t%s", it->sender, ctime(&now));
 584         if (error < 0 || (size_t)error >= sizeof(line)) {
 585                 syslog(LOG_ERR, "%s: local delivery deferred: can not write header: %m",
 586                        it->queueid);
 587                 return (1);
 588         }
 589         if (write(mbox, line, error) != error)
 590                 goto wrerror;
 591
 592         while (!feof(it->queuef)) {
 593                 if (fgets(line, sizeof(line), it->queuef) == NULL)
 594                         break;
 595                 linelen = strlen(line);
 596                 if (linelen == 0 || line[linelen - 1] != '\n') {
 597                         syslog(LOG_CRIT, "%s: local delivery failed: corrupted queue file",
 598                                it->queueid);
 599                         *errmsg = "corrupted queue file";
 600                         error = -1;
 601                         goto chop;
 602                 }
 603
 604                 if (strncmp(line, "From ", 5) == 0) {
 605                         const char *gt = ">";
 606
 607                         if (write(mbox, gt, 1) != 1)
 608                                 goto wrerror;
 609                 }
 610                 if ((size_t)write(mbox, line, linelen) != linelen)
 611                         goto wrerror;
 612         }
 613         line[0] = '\n';
 614         if (write(mbox, line, 1) != 1)
 615                 goto wrerror;
 616         close(mbox);
 617         return (0);
 618
 619 wrerror:
 620         syslog(LOG_ERR, "%s: local delivery failed: write error: %m",
 621                it->queueid);
 622         error = 1;
 623 chop:
 624         if (ftruncate(mbox, mboxlen) != 0)
 625                 syslog(LOG_WARNING, "%s: error recovering mbox `%s': %m",
 626                        it->queueid, fn);
 627         close(mbox);
 628         return (error);
 629 }
 630
 631 static void
 632 deliver(struct qitem *it)
 633 {
 634         int error;
 635         unsigned int backoff = MIN_RETRY;
 636         const char *errmsg = "unknown bounce reason";
 637         struct timeval now;
 638         struct stat st;
 639
 640         syslog(LOG_INFO, "%s: mail from=<%s> to=<%s>",
 641                it->queueid, it->sender, it->addr);
 642
 643 retry:
 644         syslog(LOG_INFO, "%s: trying delivery",
 645                it->queueid);
 646
 647         if (it->remote)
 648                 error = deliver_remote(it, &errmsg);
 649         else
 650                 error = deliver_local(it, &errmsg);
 651
 652         switch (error) {
 653         case 0:
 654                 unlink(it->queuefn);
 655                 syslog(LOG_INFO, "%s: delivery successful",
 656                        it->queueid);
 657                 exit(0);
 658
 659         case 1:
 660                 if (stat(it->queuefn, &st) != 0) {
 661                         syslog(LOG_ERR, "%s: lost queue file `%s'",
 662                                it->queueid, it->queuefn);
 663                         exit(1);
 664                 }
 665                 if (gettimeofday(&now, NULL) == 0 &&
 666                     (now.tv_sec - st.st_mtimespec.tv_sec > MAX_TIMEOUT)) {
 667                         char *msg;
 668
 669                         if (asprintf(&msg,
 670                                  "Could not deliver for the last %d seconds. Giving up.",
 671                                  MAX_TIMEOUT) > 0)
 672                                 errmsg = msg;
 673                         goto bounce;
 674                 }
 675                 sleep(backoff);
 676                 backoff *= 2;
 677                 if (backoff > MAX_RETRY)
 678                         backoff = MAX_RETRY;
 679                 goto retry;
 680
 681         case -1:
 682         default:
 683                 break;
 684         }
 685
 686 bounce:
 687         bounce(it, errmsg);
 688         /* NOTREACHED */
 689 }
 690
 691 static int
 692 c2x(char c)
 693 {
 694         if (c <= '9')
 695                 return (c - '0');
 696         else if (c <= 'F')
 697                 return (c - 'A' + 10);
 698         else
 699                 return (c - 'a' + 10);
 700 }
 701
 702 static void
 703 seen_init(void)
 704 {
 705         int i, j;
 706
 707         for (i = 0; i < 16; i++)
 708                 for (j = 0; j < 16; j++)
 709                         SLIST_INIT(&seenmsg[i][j]);
 710 }
 711
 712 static int
 713 seen(const char *msgid)
 714 {
 715         const char *p;
 716         size_t len;
 717         int i, j;
 718         struct stritem *t;
 719
 720         p = strchr(msgid, '.');
 721         if (p == NULL)
 722                 return (0);
 723         len = p - msgid;
 724         if (len >= 2) {
 725                 i = c2x(msgid[len - 2]);
 726                 j = c2x(msgid[len - 1]);
 727         } else if (len == 1) {
 728                 i = c2x(msgid[0]);
 729                 j = 0;
 730         } else {
 731                 i = j = 0;
 732         }
 733         if (i < 0 || i >= 16 || j < 0 || j >= 16)
 734                 errx(1, "INTERNAL ERROR: bad seen code for msgid %s", msgid);
 735         SLIST_FOREACH(t, &seenmsg[i][j], next)
 736                 if (!strncmp(t->str, msgid, len))
 737                         return (1);
 738         t = malloc(sizeof(*t));
 739         if (t == NULL)
 740                 errx(1, "Could not allocate %lu bytes",
 741                     (unsigned long)(sizeof(*t)));
 742         t->str = strdup(msgid);
 743         if (t->str == NULL)
 744                 errx(1, "Could not duplicate msgid %s", msgid);
 745         SLIST_INSERT_HEAD(&seenmsg[i][j], t, next);
 746         return (0);
 747 }
 748
 749 static void
 750 load_queue(struct queue *queue, int ignorelock)
 751 {
 752         struct stat st;
 753         struct qitem *it;
 754         //struct queue queue, itmqueue;
 755         struct queue itmqueue;
 756         DIR *spooldir;
 757         struct dirent *de;
 758         char line[1000];
 759         char *fn;
 760         FILE *queuef;
 761         char *sender;
 762         char *addr;
 763         char *queueid;
 764         char *queuefn;
 765         off_t hdrlen;
 766         int fd, locked, seenit;
 767
 768         LIST_INIT(&queue->queue);
 769
 770         spooldir = opendir(config->spooldir);
 771         if (spooldir == NULL)
 772                 err(1, "reading queue");
 773
 774         seen_init();
 775         while ((de = readdir(spooldir)) != NULL) {
 776                 sender = NULL;
 777                 queuef = NULL;
 778                 queueid = NULL;
 779                 queuefn = NULL;
 780                 fn = NULL;
 781                 LIST_INIT(&itmqueue.queue);
 782
 783                 /* ignore temp files */
 784                 if (strncmp(de->d_name, "tmp_", 4) == 0 ||
 785                     de->d_type != DT_REG)
 786                         continue;
 787                 if (asprintf(&queuefn, "%s/%s", config->spooldir, de->d_name) < 0)
 788                         goto fail;
 789                 seenit = seen(de->d_name);
 790                 locked = 0;
 791                 fd = open_locked(queuefn, O_RDONLY|O_NONBLOCK);
 792                 if (fd < 0) {
 793                         /* Ignore locked files */
 794                         if (errno != EWOULDBLOCK)
 795                                 goto skip_item;
 796                         if (!ignorelock || seenit)
 797                                 continue;
 798                         fd = open(queuefn, O_RDONLY);
 799                         if (fd < 0)
 800                                 goto skip_item;
 801                         locked = 1;
 802                 }
 803
 804                 queuef = fdopen(fd, "r");
 805                 if (queuef == NULL)
 806                         goto skip_item;
 807                 if (fgets(line, sizeof(line), queuef) == NULL ||
 808                     line[0] == 0)
 809                         goto skip_item;
 810                 line[strlen(line) - 1] = 0;     /* chop newline */
 811                 sender = strdup(line);
 812                 if (sender == NULL)
 813                         goto skip_item;
 814
 815                 for (;;) {
 816                         if (fgets(line, sizeof(line), queuef) == NULL ||
 817                             line[0] == 0)
 818                                 goto skip_item;
 819                         if (line[0] == '\n')
 820                                 break;
 821                         line[strlen(line) - 1] = 0;
 822                         queueid = strdup(line);
 823                         if (queueid == NULL)
 824                                 goto skip_item;
 825                         addr = strchr(queueid, ' ');
 826                         if (addr == NULL)
 827                                 goto skip_item;
 828                         *addr++ = 0;
 829                         if (fn != NULL)
 830                                 free(fn);
 831                         if (asprintf(&fn, "%s/%s", config->spooldir, queueid) < 0)
 832                                 goto skip_item;
 833                         /* Item has already been delivered? */
 834                         if (stat(fn, &st) != 0)
 835                                 continue;
 836                         if (add_recp(&itmqueue, addr, sender, 0) != 0)
 837                                 goto skip_item;
 838                         it = LIST_FIRST(&itmqueue.queue);
 839                         it->queuef = queuef;
 840                         it->queueid = queueid;
 841                         it->queuefn = fn;
 842                         it->locked = locked;
 843                         fn = NULL;
 844                 }
 845                 if (LIST_EMPTY(&itmqueue.queue)) {
 846                         warnx("queue file without items: `%s'", queuefn);
 847                         goto skip_item2;
 848                 }
 849                 hdrlen = ftell(queuef);
 850                 while ((it = LIST_FIRST(&itmqueue.queue)) != NULL) {
 851                         it->hdrlen = hdrlen;
 852                         LIST_REMOVE(it, next);
 853                         LIST_INSERT_HEAD(&queue->queue, it, next);
 854                 }
 855                 continue;
 856
 857 skip_item:
 858                 warn("reading queue: `%s'", queuefn);
 859 skip_item2:
 860                 if (sender != NULL)
 861                         free(sender);
 862                 if (queuefn != NULL)
 863                         free(queuefn);
 864                 if (fn != NULL)
 865                         free(fn);
 866                 if (queueid != NULL)
 867                         free(queueid);
 868                 close(fd);
 869         }
 870         closedir(spooldir);
 871         return;
 872
 873 fail:
 874         err(1, "reading queue");
 875 }
 876
 877 static void
 878 run_queue(struct queue *queue)
 879 {
 880         struct qitem *it;
 881
 882         if (LIST_EMPTY(&queue->queue))
 883                 return;
 884
 885         it = go_background(queue);
 886         deliver(it);
 887         /* NOTREACHED */
 888 }
 889
 890 static void
 891 show_queue(struct queue *queue)
 892 {
 893         struct qitem *it;
 894
 895         if (LIST_EMPTY(&queue->queue)) {
 896                 printf("Mail queue is empty\n");
 897                 return;
 898         }
 899
 900         LIST_FOREACH(it, &queue->queue, next) {
 901                 printf("\
 902 ID\t: %s%s\n\
 903 From\t: %s\n\
 904 To\t: %s\n--\n", it->queueid, it->locked? "*": "", it->sender, it->addr);
 905         }
 906 }
 907
 908 /*
 909  * TODO:
 910  *
 911  * - alias processing
 912  * - use group permissions
 913  * - proper sysexit codes
 914  */
 915
 916 int
 917 main(int argc, char **argv)
 918 {
 919         char *sender = NULL;
 920         char tag[255];
 921         struct qitem *it;
 922         struct queue queue;
 923         struct queue lqueue;
 924         int i, ch;
 925         int nodot = 0, doqueue = 0, showq = 0;
 926
 927         atexit(deltmp);
 928         LIST_INIT(&queue.queue);
 929         snprintf(tag, 254, "dma");
 930
 931         opterr = 0;
 932         while ((ch = getopt(argc, argv, "A:b:Df:iL:o:O:q:r:")) != -1) {
 933                 switch (ch) {
 934                 case 'A':
 935                         /* -AX is being ignored, except for -A{c,m} */
 936                         if (optarg[0] == 'c' || optarg[0] == 'm') {
 937                                 break;
 938                         }
 939                         /* else FALLTRHOUGH */
 940                 case 'b':
 941                         /* -bX is being ignored, except for -bp */
 942                         if (optarg[0] == 'p') {
 943                                 showq = 1;
 944                                 break;
 945                         }
 946                         /* else FALLTRHOUGH */
 947                 case 'D':
 948                         daemonize = 0;
 949                         break;
 950                 case 'L':
 951                         if (optarg != NULL)
 952                                 snprintf(tag, 254, "%s", optarg);
 953                         break;
 954                 case 'f':
 955                 case 'r':
 956                         sender = optarg;
 957                         break;
 958
 959                 case 'o':
 960                         /* -oX is being ignored, except for -oi */
 961                         if (optarg[0] != 'i')
 962                                 break;
 963                         /* else FALLTRHOUGH */
 964                 case 'O':
 965                         break;
 966                 case 'i':
 967                         nodot = 1;
 968                         break;
 969
 970                 case 'q':
 971                         doqueue = 1;
 972                         break;
 973
 974                 default:
 975                         exit(1);
 976                 }
 977         }
 978         argc -= optind;
 979         argv += optind;
 980         opterr = 1;
 981
 982         openlog(tag, LOG_PID | LOG_PERROR, LOG_MAIL);
 983
 984         config = malloc(sizeof(struct config));
 985         if (config == NULL)
 986                 errx(1, "Cannot allocate enough memory");
 987
 988         memset(config, 0, sizeof(struct config));
 989         if (parse_conf(CONF_PATH, config) < 0) {
 990                 free(config);
 991                 errx(1, "reading config file");
 992         }
 993
 994         if (config->features & VIRTUAL)
 995                 if (parse_virtuser(config->virtualpath) < 0)
 996                         errx(1, "error reading virtual user file: %s",
 997                                 config->virtualpath);
 998
 999         if (parse_authfile(config->authpath) < 0)
1000                 err(1, "reading SMTP authentication file");
1001
1002         if (showq) {
1003                 if (argc != 0)
1004                         errx(1, "sending mail and displaying queue is"
1005                                 " mutually exclusive");
1006                 load_queue(&lqueue, 1);
1007                 show_queue(&lqueue);
1008                 return (0);
1009         }
1010
1011         if (doqueue) {
1012                 if (argc != 0)
1013                         errx(1, "sending mail and queue pickup is mutually exclusive");
1014                 load_queue(&lqueue, 0);
1015                 run_queue(&lqueue);
1016                 return (0);
1017         }
1018
1019         if (read_aliases() != 0)
1020                 err(1, "reading aliases");
1021
1022         if ((sender = set_from(sender)) == NULL)
1023                 err(1, "setting from address");
1024
1025         for (i = 0; i < argc; i++) {
1026                 if (add_recp(&queue, argv[i], sender, 1) != 0)
1027                         errx(1, "invalid recipient `%s'\n", argv[i]);
1028         }
1029
1030         if (LIST_EMPTY(&queue.queue))
1031                 errx(1, "no recipients");
1032
1033         if (gentempf(&queue) != 0)
1034                 err(1, "create temp file");
1035
1036         if (preparespool(&queue, sender) != 0)
1037                 err(1, "creating spools (1)");
1038
1039         if (readmail(&queue, sender, nodot) != 0)
1040                 err(1, "reading mail");
1041
1042         if (linkspool(&queue) != 0)
1043                 err(1, "creating spools (2)");
1044
1045         /* From here on the mail is safe. */
1046
1047         if (config->features & DEFER)
1048                 return (0);
1049
1050         it = go_background(&queue);
1051         deliver(it);
1052
1053         /* NOTREACHED */
1054         return (0);
1055 }