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