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