usr.sbin/ctm/ctm_dequeue/ctm_dequeue.c

   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
  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
 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
 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 }