libexec/dma/spool.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
  35 #include <sys/stat.h>
  36
  37 #include <dirent.h>
  38 #include <err.h>
  39 #include <errno.h>
  40 #include <fcntl.h>
  41 #include <inttypes.h>
  42 #include <unistd.h>
  43 #include <syslog.h>
  44
  45 #include "dma.h"
  46
  47 /*
  48  * Spool file format:
  49  *
  50  * 'Q'id files (queue):
  51  *   id envelope-to
  52  *
  53  * 'M'id files (data):
  54  *   envelope-from
  55  *   mail data
  56  *
  57  * Each queue file needs to have a corresponding data file.
  58  * One data file might be shared by linking it several times.
  59  *
  60  * Queue ids are unique, formed from the inode of the data file
  61  * and a unique identifier.
  62  */
  63
  64 int
  65 newspoolf(struct queue *queue)
  66 {
  67         char fn[PATH_MAX+1];
  68         struct stat st;
  69         struct stritem *t;
  70         struct qitem *it;
  71         off_t hdrlen;
  72         int fd;
  73
  74         if (snprintf(fn, sizeof(fn), "%s/%s", config->spooldir, "tmp_XXXXXXXXXX") <= 0)
  75                 return (-1);
  76
  77         fd = mkstemp(fn);
  78         if (fd < 0)
  79                 return (-1);
  80         if (flock(fd, LOCK_EX) == -1)
  81                 goto fail;
  82         queue->tmpf = strdup(fn);
  83         if (queue->tmpf == NULL)
  84                 goto fail;
  85
  86         /*
  87          * Assign queue id
  88          */
  89         if (fstat(fd, &st) != 0)
  90                 goto fail;
  91         if (asprintf(&queue->id, "%"PRIxMAX, st.st_ino) < 0)
  92                 goto fail;
  93
  94         queue->mailf = fdopen(fd, "r+");
  95         if (queue->mailf == NULL)
  96                 goto fail;
  97
  98         if (fprintf(queue->mailf, "%s\n", queue->sender) < 0)
  99                 goto fail;
 100
 101         hdrlen = ftello(queue->mailf);
 102
 103         LIST_FOREACH(it, &queue->queue, next) {
 104                 it->hdrlen = hdrlen;
 105         }
 106
 107         t = malloc(sizeof(*t));
 108         if (t != NULL) {
 109                 t->str = queue->tmpf;
 110                 SLIST_INSERT_HEAD(&tmpfs, t, next);
 111         }
 112         return (0);
 113
 114 fail:
 115         if (queue->mailf != NULL)
 116                 fclose(queue->mailf);
 117         close(fd);
 118         unlink(fn);
 119         return (-1);
 120 }
 121
 122 int
 123 linkspool(struct queue *queue)
 124 {
 125         char line[1000];        /* by RFC2822 */
 126         struct stat st;
 127         size_t error;
 128         int queuefd;
 129         struct qitem *it;
 130
 131         if (fflush(queue->mailf) != 0 || fsync(fileno(queue->mailf)) != 0)
 132                 goto delfiles;
 133
 134         syslog(LOG_INFO, "new mail from user=%s uid=%d envelope_from=<%s>",
 135                username, getuid(), queue->sender);
 136
 137         LIST_FOREACH(it, &queue->queue, next) {
 138                 if (asprintf(&it->queueid, "%s.%"PRIxPTR, queue->id, (uintptr_t)it) <= 0)
 139                         goto delfiles;
 140                 if (asprintf(&it->queuefn, "%s/Q%s", config->spooldir, it->queueid) <= 0)
 141                         goto delfiles;
 142                 if (asprintf(&it->mailfn, "%s/M%s", config->spooldir, it->queueid) <= 0)
 143                         goto delfiles;
 144
 145                 /* Neither file may not exist yet */
 146                 if (stat(it->queuefn, &st) == 0 || stat(it->mailfn, &st) == 0)
 147                         goto delfiles;
 148
 149                 error = snprintf(line, sizeof(line), "%s %s\n", it->queueid, it->addr);
 150                 if ((ssize_t)error < 0 || error >= sizeof(line))
 151                         goto delfiles;
 152
 153                 queuefd = open_locked(it->queuefn, O_CREAT|O_EXCL|O_RDWR, 0600);
 154                 if (queuefd == -1)
 155                         goto delfiles;
 156                 it->queuef = fdopen(queuefd, "w+");
 157                 if (it->queuef == NULL)
 158                         goto delfiles;
 159
 160                 if (fwrite(line, strlen(line), 1, it->queuef) != 1)
 161                         goto delfiles;
 162                 if (fflush(it->queuef) != 0 || fsync(fileno(it->queuef)) != 0)
 163                         goto delfiles;
 164
 165                 if (link(queue->tmpf, it->mailfn) != 0)
 166                         goto delfiles;
 167         }
 168
 169         LIST_FOREACH(it, &queue->queue, next) {
 170                 syslog(LOG_INFO, "mail to=<%s> queued as %s",
 171                        it->addr, it->queueid);
 172         }
 173
 174         unlink(queue->tmpf);
 175         return (0);
 176
 177 delfiles:
 178         LIST_FOREACH(it, &queue->queue, next) {
 179                 unlink(it->mailfn);
 180                 unlink(it->queuefn);
 181         }
 182         return (-1);
 183 }
 184
 185 int
 186 load_queue(struct queue *queue)
 187 {
 188         struct qitem *it;
 189         //struct queue queue, itmqueue;
 190         struct queue itmqueue;
 191         DIR *spooldir;
 192         struct dirent *de;
 193         char line[1000];
 194         FILE *queuef;
 195         FILE *mailf;
 196         char *sender;
 197         char *addr;
 198         char *queueid;
 199         char *queuefn;
 200         char *mailfn;
 201         off_t hdrlen;
 202
 203         bzero(queue, sizeof(queue));
 204         LIST_INIT(&queue->queue);
 205
 206         spooldir = opendir(config->spooldir);
 207         if (spooldir == NULL)
 208                 err(1, "reading queue");
 209
 210         while ((de = readdir(spooldir)) != NULL) {
 211                 sender = NULL;
 212                 queuef = NULL;
 213                 mailf = NULL;
 214                 queueid = NULL;
 215                 queuefn = NULL;
 216                 LIST_INIT(&itmqueue.queue);
 217
 218                 /* ignore temp files */
 219                 if (strncmp(de->d_name, "tmp_", 4) == 0 || de->d_type != DT_REG)
 220                         continue;
 221                 if (de->d_name[0] != 'Q')
 222                         continue;
 223                 if (asprintf(&queuefn, "%s/Q%s", config->spooldir, de->d_name + 1) < 0)
 224                         goto fail;
 225                 if (asprintf(&mailfn, "%s/M%s", config->spooldir, de->d_name + 1) < 0)
 226                         goto fail;
 227
 228                 mailf = fopen(mailfn, "r");
 229                 if (mailf == NULL)
 230                         goto skip_item;
 231                 if (fgets(line, sizeof(line), mailf) == NULL || line[0] == 0)
 232                         goto skip_item;
 233                 line[strlen(line) - 1] = 0;     /* chop newline */
 234                 sender = strdup(line);
 235                 if (sender == NULL)
 236                         goto skip_item;
 237
 238                 hdrlen = ftell(mailf);
 239
 240                 queuef = fopen(queuefn, "r");
 241                 if (queuef == NULL)
 242                         goto skip_item;
 243
 244                 if (fgets(line, sizeof(line), queuef) == NULL || line[0] == 0)
 245                         goto skip_item;
 246                 line[strlen(line) - 1] = 0;
 247                 queueid = strdup(line);
 248                 if (queueid == NULL)
 249                         goto skip_item;
 250                 addr = strchr(queueid, ' ');
 251                 if (addr == NULL)
 252                         goto skip_item;
 253                 *addr++ = 0;
 254
 255                 if (add_recp(&itmqueue, addr, 0) != 0)
 256                         goto skip_item;
 257
 258                 it = LIST_FIRST(&itmqueue.queue);
 259                 it->sender = sender;
 260                 it->queueid = queueid;
 261                 it->queuefn = queuefn;
 262                 it->mailfn = mailfn;
 263                 it->hdrlen = hdrlen;
 264                 LIST_INSERT_HEAD(&queue->queue, it, next);
 265
 266                 if (queuef != NULL)
 267                         fclose(queuef);
 268                 if (mailf != NULL)
 269                         fclose(mailf);
 270                 continue;
 271
 272 skip_item:
 273                 syslog(LOG_INFO, "could not pick up queue file: `%s'/`%s': %m", queuefn, mailfn);
 274                 if (sender != NULL)
 275                         free(sender);
 276                 if (queuefn != NULL)
 277                         free(queuefn);
 278                 if (mailfn != NULL)
 279                         free(queuefn);
 280                 if (queueid != NULL)
 281                         free(queueid);
 282                 if (queuef != NULL)
 283                         fclose(queuef);
 284                 if (mailf != NULL)
 285                         fclose(mailf);
 286         }
 287         closedir(spooldir);
 288         return (0);
 289
 290 fail:
 291         return (-1);
 292 }
 293
 294 void
 295 delqueue(struct qitem *it)
 296 {
 297         unlink(it->mailfn);
 298         unlink(it->queuefn);
 299         if (it->queuef != NULL)
 300                 fclose(it->queuef);
 301         if (it->mailf != NULL)
 302                 fclose(it->mailf);
 303         free(it);
 304 }
 305
 306 int
 307 aquirespool(struct qitem *it)
 308 {
 309         int queuefd;
 310
 311         if (it->queuef == NULL) {
 312                 queuefd = open_locked(it->queuefn, O_RDWR|O_NONBLOCK);
 313                 if (queuefd < 0)
 314                         goto fail;
 315                 it->queuef = fdopen(queuefd, "r+");
 316                 if (it->queuef == NULL)
 317                         goto fail;
 318         }
 319
 320         if (it->mailf == NULL) {
 321                 it->mailf = fopen(it->mailfn, "r");
 322                 if (it->mailf == NULL)
 323                         goto fail;
 324         }
 325
 326         return (0);
 327
 328 fail:
 329         syslog(LOG_INFO, "could not aquire queue file: %m");
 330         return (-1);
 331 }
 332
 333 void
 334 dropspool(struct queue *queue, struct qitem *keep)
 335 {
 336         struct qitem *it;
 337
 338         LIST_FOREACH(it, &queue->queue, next) {
 339                 if (it == keep)
 340                         continue;
 341
 342                 if (it->queuef != NULL)
 343                         fclose(it->queuef);
 344                 if (it->mailf != NULL)
 345                         fclose(it->mailf);
 346         }
 347 }