diff options
Diffstat (limited to 'usr.sbin/ctm/ctm_dequeue/ctm_dequeue.c')
| -rw-r--r-- | usr.sbin/ctm/ctm_dequeue/ctm_dequeue.c | 209 |
1 files changed, 209 insertions, 0 deletions
diff --git a/usr.sbin/ctm/ctm_dequeue/ctm_dequeue.c b/usr.sbin/ctm/ctm_dequeue/ctm_dequeue.c new file mode 100644 index 0000000..2dd6e86 --- /dev/null +++ b/usr.sbin/ctm/ctm_dequeue/ctm_dequeue.c @@ -0,0 +1,209 @@ +/* + * Copyright (c) 1996, Gary J. Palmer + * All rights reserved. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions + * are met: + * + * 1. Redistributions of source code must retain the above copyright + * notice, this list of conditions and the following disclaimer, + * verbatim and that no modifications are made prior to this + * point in the file. + * 2. Redistributions in binary form must reproduce the above copyright + * notice, this list of conditions and the following disclaimer in the + * documentation and/or other materials provided with the distribution. + * + * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND + * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE + * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE + * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE + * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL + * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS + * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) + * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT + * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY + * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF + * SUCH DAMAGE. + * + * $FreeBSD$ + */ + +/* + * ctm_dequeue: Dequeue queued delta pieces and mail them. + * + * The pieces have already been packaged up as mail messages by ctm_smail, + * and will be simply passed to sendmail in alphabetical order. + */ + +#include <stdio.h> +#include <stdlib.h> +#include <string.h> +#include <unistd.h> +#include <fcntl.h> +#include <sys/types.h> +#include <sys/stat.h> +#include <sys/wait.h> +#include <fts.h> +#include <limits.h> +#include <errno.h> +#include <paths.h> +#include "error.h" +#include "options.h" + +#define DEFAULT_NUM 1 /* Default number of pieces mailed per run. */ + +int fts_sort(const FTSENT * const *, const FTSENT * const *); +int run_sendmail(int ifd); + +int +main(int argc, char **argv) +{ + char *log_file = NULL; + char *queue_dir = NULL; + char *list[2]; + int num_to_send = DEFAULT_NUM, chunk; + int fd; + FTS *fts; + FTSENT *ftsent; + int piece, npieces; + char filename[PATH_MAX]; + + err_prog_name(argv[0]); + + OPTIONS("[-l log] [-n num] queuedir") + NUMBER('n', num_to_send) + STRING('l', log_file) + ENDOPTS + + if (argc != 2) + usage(); + + if (log_file) + err_set_log(log_file); + + queue_dir = argv[1]; + list[0] = queue_dir; + list[1] = NULL; + + fts = fts_open(list, FTS_PHYSICAL|FTS_COMFOLLOW, fts_sort); + if (fts == NULL) + { + err("fts failed on `%s'", queue_dir); + exit(1); + } + + ftsent = fts_read(fts); + if (ftsent == NULL || ftsent->fts_info != FTS_D) + { + err("not a directory: %s", queue_dir); + exit(1); + } + + ftsent = fts_children(fts, 0); + if (ftsent == NULL && errno) + { + err("*ftschildren failed"); + exit(1); + } + + for (chunk = 1; ftsent != NULL; ftsent = ftsent->fts_link) + { + /* + * Skip non-files and ctm_smail tmp files (ones starting with `.') + */ + if (ftsent->fts_info != FTS_F || ftsent->fts_name[0] == '.') + continue; + + sprintf(filename, "%s/%s", queue_dir, ftsent->fts_name); + fd = open(filename, O_RDONLY); + if (fd < 0) + { + err("*open: %s", filename); + exit(1); + } + + if (run_sendmail(fd)) + exit(1); + + close(fd); + + if (unlink(filename) < 0) + { + err("*unlink: %s", filename); + exit(1); + } + + /* + * Deduce the delta, piece number, and number of pieces from + * the name of the file in the queue. Ideally, we should be + * able to get the mail alias name too. + * + * NOTE: This depends intimately on the queue name used in ctm_smail. + */ + npieces = atoi(&ftsent->fts_name[ftsent->fts_namelen-3]); + piece = atoi(&ftsent->fts_name[ftsent->fts_namelen-7]); + err("%.*s %d/%d sent", (int)(ftsent->fts_namelen-8), ftsent->fts_name, + piece, npieces); + + if (chunk++ == num_to_send) + break; + } + + fts_close(fts); + + return(0); +} + +int +fts_sort(const FTSENT * const * a, const FTSENT * const * b) +{ + if ((*a)->fts_info != FTS_F) + return(0); + if ((*a)->fts_info != FTS_F) + return(0); + + return (strcmp((*a)->fts_name, (*b)->fts_name)); +} + +/* + * Run sendmail with the given file descriptor as standard input. + * Sendmail will decode the destination from the message contents. + * Returns 0 on success, 1 on failure. + */ +int +run_sendmail(int ifd) +{ + pid_t child, pid; + int status; + + switch (child = fork()) + { + case -1: + err("*fork"); + return(1); + + case 0: /* Child */ + dup2(ifd, 0); + execl(_PATH_SENDMAIL, _PATH_SENDMAIL, "-odq", "-t", (char *)NULL); + err("*exec: %s", _PATH_SENDMAIL); + _exit(1); + + default: /* Parent */ + while ((pid = wait(&status)) != child) + { + if (pid == -1 && errno != EINTR) + { + err("*wait"); + return(1); + } + } + if (!WIFEXITED(status) || WEXITSTATUS(status) != 0) + { + err("sendmail failed"); + return(1); + } + } + + return(0); +} |
