Properly handle an error return from udev2dev().
[dragonfly.git] / usr.sbin / ctm / ctm_dequeue / ctm_dequeue.c
1 /*
2  * Copyright (c) 1996, Gary J. Palmer
3  * All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  *
9  * 1. Redistributions of source code must retain the above copyright
10  *    notice, this list of conditions and the following disclaimer,
11  *    verbatim and that no modifications are made prior to this
12  *    point in the file.
13  * 2. Redistributions in binary form must reproduce the above copyright
14  *    notice, this list of conditions and the following disclaimer in the
15  *    documentation and/or other materials provided with the distribution.
16  *
17  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
18  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
21  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
23  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
24  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
26  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
27  * SUCH DAMAGE.
28  *
29  * $FreeBSD: src/usr.sbin/ctm/ctm_dequeue/ctm_dequeue.c,v 1.9 1999/08/28 01:16:01 peter Exp $
30  * $DragonFly: src/usr.sbin/ctm/ctm_dequeue/Attic/ctm_dequeue.c,v 1.2 2003/06/17 04:29:53 dillon Exp $
31  */
32
33 /*
34  * ctm_dequeue:  Dequeue queued delta pieces and mail them.
35  *
36  * The pieces have already been packaged up as mail messages by ctm_smail,
37  * and will be simply passed to sendmail in alphabetical order.
38  */
39
40 #include <stdio.h>
41 #include <stdlib.h>
42 #include <string.h>
43 #include <unistd.h>
44 #include <fcntl.h>
45 #include <sys/types.h>
46 #include <sys/stat.h>
47 #include <sys/wait.h>
48 #include <fts.h>
49 #include <limits.h>
50 #include <errno.h>
51 #include <paths.h>
52 #include "error.h"
53 #include "options.h"
54
55 #define DEFAULT_NUM     1       /* Default number of pieces mailed per run. */
56
57 int fts_sort(const FTSENT **, const FTSENT **);
58 int run_sendmail(int ifd);
59
60 int
61 main(int argc, char **argv)
62 {
63     char *log_file = NULL;
64     char *queue_dir = NULL;
65     char *list[2];
66     int num_to_send = DEFAULT_NUM, chunk;
67     int fd;
68     FTS *fts;
69     FTSENT *ftsent;
70     int piece, npieces;
71     char filename[PATH_MAX];
72
73     err_prog_name(argv[0]);
74
75     OPTIONS("[-l log] [-n num] queuedir")
76         NUMBER('n', num_to_send)
77         STRING('l', log_file)
78     ENDOPTS
79
80     if (argc != 2)
81         usage();
82
83     if (log_file)
84         err_set_log(log_file);
85
86     queue_dir = argv[1];
87     list[0] = queue_dir;
88     list[1] = NULL;
89
90     fts = fts_open(list, FTS_PHYSICAL|FTS_COMFOLLOW, fts_sort);
91     if (fts == NULL)
92     {
93         err("fts failed on `%s'", queue_dir);
94         exit(1);
95     }
96
97     ftsent = fts_read(fts);
98     if (ftsent == NULL || ftsent->fts_info != FTS_D)
99     {
100         err("not a directory: %s", queue_dir);
101         exit(1);
102     }
103
104     ftsent = fts_children(fts, 0);
105     if (ftsent == NULL && errno)
106     {
107         err("*ftschildren failed");
108         exit(1);
109     }
110
111     for (chunk = 1; ftsent != NULL; ftsent = ftsent->fts_link)
112     {
113         /*
114          * Skip non-files and ctm_smail tmp files (ones starting with `.')
115          */
116         if (ftsent->fts_info != FTS_F || ftsent->fts_name[0] == '.')
117             continue;
118
119         sprintf(filename, "%s/%s", queue_dir, ftsent->fts_name);
120         fd = open(filename, O_RDONLY);
121         if (fd < 0)
122         {
123             err("*open: %s", filename);
124             exit(1);
125         }
126
127         if (run_sendmail(fd))
128             exit(1);
129
130         close(fd);
131         
132         if (unlink(filename) < 0)
133         {
134             err("*unlink: %s", filename);
135             exit(1);
136         }
137         
138         /*
139          * Deduce the delta, piece number, and number of pieces from
140          * the name of the file in the queue.  Ideally, we should be
141          * able to get the mail alias name too.
142          *
143          * NOTE: This depends intimately on the queue name used in ctm_smail.
144          */
145         npieces = atoi(&ftsent->fts_name[ftsent->fts_namelen-3]);
146         piece = atoi(&ftsent->fts_name[ftsent->fts_namelen-7]);
147         err("%.*s %d/%d sent", ftsent->fts_namelen-8, ftsent->fts_name,
148                 piece, npieces);
149
150         if (chunk++ == num_to_send)
151             break;
152     }
153
154     fts_close(fts);
155
156     return(0);
157 }
158
159 int
160 fts_sort(const FTSENT ** a, const FTSENT ** b)
161 {
162     if ((*a)->fts_info != FTS_F)
163         return(0);
164     if ((*a)->fts_info != FTS_F)
165         return(0);
166
167     return (strcmp((*a)->fts_name, (*b)->fts_name));
168 }
169
170 /*
171  * Run sendmail with the given file descriptor as standard input.
172  * Sendmail will decode the destination from the message contents.
173  * Returns 0 on success, 1 on failure.
174  */
175 int
176 run_sendmail(int ifd)
177 {
178     pid_t child, pid;
179     int status;
180
181     switch (child = fork())
182     {
183     case -1:
184         err("*fork");
185         return(1);
186
187     case 0:     /* Child */
188         dup2(ifd, 0);
189         execl(_PATH_SENDMAIL, _PATH_SENDMAIL, "-odq", "-t", NULL);
190         err("*exec: %s", _PATH_SENDMAIL);
191         _exit(1);
192
193     default:    /* Parent */
194         while ((pid = wait(&status)) != child)
195         {
196             if (pid == -1 && errno != EINTR)
197             {
198                 err("*wait");
199                 return(1);
200             }
201         }
202         if (!WIFEXITED(status) || WEXITSTATUS(status) != 0)
203         {
204             err("sendmail failed");
205             return(1);
206         }
207     }
208
209     return(0);
210 }