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