diplodocus.org Git - nmh/blob - uip/mhbuild.c

   1
   2 /*
   3  * mhbuild.c -- expand/translate MIME composition files
   4  *
   5  * This code is Copyright (c) 2002, by the authors of nmh.  See the
   6  * COPYRIGHT file in the root directory of the nmh distribution for
   7  * complete copyright information.
   8  */
   9
  10 #include <h/mh.h>
  11 #include <fcntl.h>
  12 #include <h/md5.h>
  13 #include <h/mts.h>
  14 #include <h/tws.h>
  15 #include <h/mime.h>
  16 #include <h/mhparse.h>
  17 #include <h/mhcachesbr.h>
  18 #include <h/utils.h>
  19
  20 #define MHBUILD_SWITCHES \
  21     X("auto", 0, AUTOSW) \
  22     X("noauto", 0, NAUTOSW) \
  23     X("check", 0, CHECKSW) \
  24     X("nocheck", 0, NCHECKSW) \
  25     X("directives", 0, DIRECTIVES) \
  26     X("nodirectives", 0, NDIRECTIVES) \
  27     X("headers", 0, HEADSW) \
  28     X("noheaders", 0, NHEADSW) \
  29     X("list", 0, LISTSW) \
  30     X("nolist", 0, NLISTSW) \
  31     X("realsize", 0, SIZESW) \
  32     X("norealsize", 0, NSIZESW) \
  33     X("rfc934mode", 0, RFC934SW) \
  34     X("norfc934mode", 0, NRFC934SW) \
  35     X("verbose", 0, VERBSW) \
  36     X("noverbose", 0, NVERBSW) \
  37     X("disposition", 0, DISPOSW) \
  38     X("nodisposition", 0, NDISPOSW) \
  39     X("rcache policy", 0, RCACHESW) \
  40     X("wcache policy", 0, WCACHESW) \
  41     X("contentid", 0, CONTENTIDSW) \
  42     X("nocontentid", 0, NCONTENTIDSW) \
  43     X("headerencoding encoding-algorithm", 0, HEADERENCSW) \
  44     X("autoheaderencoding", 0, AUTOHEADERENCSW) \
  45     X("maxunencoded", 0, MAXUNENCSW) \
  46     X("version", 0, VERSIONSW) \
  47     X("help", 0, HELPSW) \
  48     X("debug", -5, DEBUGSW) \
  49     X("dist", -4, DISTSW) \
  50
  51 #define X(sw, minchars, id) id,
  52 DEFINE_SWITCH_ENUM(MHBUILD);
  53 #undef X
  54
  55 #define X(sw, minchars, id) { sw, minchars, id },
  56 DEFINE_SWITCH_ARRAY(MHBUILD, switches);
  57 #undef X
  58
  59 #define MIMEENCODING_SWITCHES \
  60     X("base64", 0, BASE64SW) \
  61     X("quoted-printable", 0, QUOTEDPRINTSW) \
  62
  63 #define X(sw, minchars, id) id,
  64 DEFINE_SWITCH_ENUM(MIMEENCODING);
  65 #undef X
  66
  67 #define X(sw, minchars, id) { sw, minchars, id },
  68 DEFINE_SWITCH_ARRAY(MIMEENCODING, encodingswitches);
  69 #undef X
  70
  71 /* mhcachesbr.c */
  72 extern int rcachesw;
  73 extern int wcachesw;
  74 extern char *cache_public;
  75 extern char *cache_private;
  76
  77 int debugsw = 0;
  78
  79 int listsw   = 0;
  80 int rfc934sw = 0;
  81 int contentidsw = 1;
  82
  83 /*
  84  * Temporary files
  85  */
  86 static char infile[BUFSIZ];
  87 static int unlink_infile  = 0;
  88
  89 static char outfile[BUFSIZ];
  90 static int unlink_outfile = 0;
  91
  92 static void unlink_done (int) NORETURN;
  93
  94 /* mhbuildsbr.c */
  95 int output_message (CT, char *);
  96 int output_message_fp (CT, FILE *, char*);
  97
  98
  99 int
 100 main (int argc, char **argv)
 101 {
 102     int sizesw = 1, headsw = 1, directives = 1, autobuild = 0, dist = 0;
 103     int verbosw = 0, dispo = 0;
 104     size_t maxunencoded = MAXTEXTPERLN;
 105     int *icachesw;
 106     char *cp, buf[BUFSIZ];
 107     char buffer[BUFSIZ], *compfile = NULL;
 108     char **argp, **arguments;
 109     CT ct, cts[2];
 110     FILE *fp = NULL;
 111     FILE *fp_out = NULL;
 112     int header_encoding = CE_UNKNOWN;
 113     size_t n;
 114
 115     if (nmh_init(argv[0], 1)) { return 1; }
 116
 117     done=unlink_done;
 118
 119     arguments = getarguments (invo_name, argc, argv, 1);
 120     argp = arguments;
 121
 122     while ((cp = *argp++)) {
 123         if (cp[0] == '-' && cp[1] == '\0') {
 124             if (compfile)
 125                 adios (NULL, "cannot specify both standard input and a file");
 126             else
 127                 compfile = cp;
 128             listsw = 0;         /* turn off -list if using standard in/out */
 129             verbosw = 0;        /* turn off -verbose listings */
 130             break;
 131         }
 132         if (*cp == '-') {
 133             switch (smatch (++cp, switches)) {
 134             case AMBIGSW:
 135                 ambigsw (cp, switches);
 136                 done (1);
 137             case UNKWNSW:
 138                 adios (NULL, "-%s unknown", cp);
 139
 140             case HELPSW:
 141                 snprintf (buf, sizeof(buf), "%s [switches] file", invo_name);
 142                 print_help (buf, switches, 1);
 143                 done (0);
 144             case VERSIONSW:
 145                 print_version(invo_name);
 146                 done (0);
 147
 148             case AUTOSW:
 149                 /* -auto implies -nodirectives */
 150                 autobuild = 1;
 151                 directives = 0;
 152                 continue;
 153             case NAUTOSW:
 154                 /*
 155                  * We're turning directives back on since this is likely here
 156                  * to override a profile entry
 157                  */
 158                 autobuild = 0;
 159                 directives = 1;
 160                 continue;
 161
 162             case RCACHESW:
 163                 icachesw = &rcachesw;
 164                 goto do_cache;
 165             case WCACHESW:
 166                 icachesw = &wcachesw;
 167             do_cache: ;
 168                 if (!(cp = *argp++) || *cp == '-')
 169                     adios (NULL, "missing argument to %s", argp[-2]);
 170                 switch (*icachesw = smatch (cp, caches)) {
 171                 case AMBIGSW:
 172                     ambigsw (cp, caches);
 173                     done (1);
 174                 case UNKWNSW:
 175                     adios (NULL, "%s unknown", cp);
 176                 default:
 177                     break;
 178                 }
 179                 continue;
 180
 181             case CHECKSW:
 182                 checksw++;
 183                 continue;
 184             case NCHECKSW:
 185                 checksw = 0;
 186                 continue;
 187
 188             case HEADSW:
 189                 headsw++;
 190                 continue;
 191             case NHEADSW:
 192                 headsw = 0;
 193                 continue;
 194
 195             case DIRECTIVES:
 196                 directives = 1;
 197                 continue;
 198             case NDIRECTIVES:
 199                 directives = 0;
 200                 continue;
 201
 202             case LISTSW:
 203                 listsw++;
 204                 continue;
 205             case NLISTSW:
 206                 listsw = 0;
 207                 continue;
 208
 209             case RFC934SW:
 210                 rfc934sw++;
 211                 continue;
 212             case NRFC934SW:
 213                 rfc934sw = 0;
 214                 continue;
 215
 216             case SIZESW:
 217                 sizesw++;
 218                 continue;
 219             case NSIZESW:
 220                 sizesw = 0;
 221                 continue;
 222
 223             case CONTENTIDSW:
 224                 contentidsw = 1;
 225                 continue;
 226             case NCONTENTIDSW:
 227                 contentidsw = 0;
 228                 continue;
 229
 230             case HEADERENCSW: {
 231                 int encoding;
 232
 233                 if (!(cp = *argp++) || *cp == '-')
 234                     adios (NULL, "missing argument to %s", argp[-2]);
 235                 switch (encoding = smatch (cp, encodingswitches)) {
 236                 case AMBIGSW:
 237                     ambigsw (cp, encodingswitches);
 238                     done (1);
 239                 case UNKWNSW:
 240                     adios (NULL, "%s unknown encoding algorithm", cp);
 241                 case BASE64SW:
 242                     header_encoding = CE_BASE64;
 243                     break;
 244                 case QUOTEDPRINTSW:
 245                     header_encoding = CE_QUOTED;
 246                     break;
 247                 default:
 248                     adios (NULL, "Internal error: algorithm %s", cp);
 249                 }
 250                 continue;
 251             }
 252
 253             case AUTOHEADERENCSW:
 254                 header_encoding = CE_UNKNOWN;
 255                 continue;
 256
 257             case MAXUNENCSW:
 258                 if (!(cp = *argp++) || *cp == '-')
 259                     adios (NULL, "missing argument to %s", argp[-2]);
 260                 if ((maxunencoded = atoi(cp)) < 1)
 261                     adios (NULL, "Invalid argument for %s: %s", argp[-2], cp);
 262                 if (maxunencoded > 998)
 263                     adios (NULL, "limit of -maxunencoded is 998");
 264                 continue;
 265
 266             case VERBSW:
 267                 verbosw++;
 268                 continue;
 269             case NVERBSW:
 270                 verbosw = 0;
 271                 continue;
 272             case DISPOSW:
 273                 dispo = 1;
 274                 continue;
 275             case NDISPOSW:
 276                 dispo = 0;
 277                 continue;
 278             case DEBUGSW:
 279                 debugsw = 1;
 280                 continue;
 281             case DISTSW:
 282                 dist = 1;
 283                 continue;
 284             }
 285         }
 286         if (compfile)
 287             adios (NULL, "only one composition file allowed");
 288         else
 289             compfile = cp;
 290     }
 291
 292     /*
 293      * Check if we've specified an additional profile
 294      */
 295     if ((cp = getenv ("MHBUILD"))) {
 296         if ((fp = fopen (cp, "r"))) {
 297             readconfig ((struct node **) 0, fp, cp, 0);
 298             fclose (fp);
 299         } else {
 300             admonish ("", "unable to read $MHBUILD profile (%s)", cp);
 301         }
 302     }
 303
 304     /*
 305      * Read the standard profile setup
 306      */
 307     if ((fp = fopen (cp = etcpath ("mhn.defaults"), "r"))) {
 308         readconfig ((struct node **) 0, fp, cp, 0);
 309         fclose (fp);
 310     }
 311
 312     /* Check for public cache location */
 313     if ((cache_public = context_find (nmhcache)) && *cache_public != '/')
 314         cache_public = NULL;
 315
 316     /* Check for private cache location */
 317     if (!(cache_private = context_find (nmhprivcache)))
 318         cache_private = ".cache";
 319     cache_private = getcpy (m_maildir (cache_private));
 320
 321     if (!context_find ("path"))
 322         free (path ("./", TFOLDER));
 323
 324     /* Check if we have a file to process */
 325     if (!compfile)
 326         adios (NULL, "need to specify a %s composition file", invo_name);
 327
 328     /*
 329      * Process the composition file from standard input.
 330      */
 331     if (compfile[0] == '-' && compfile[1] == '\0') {
 332         if ((cp = m_mktemp2(NULL, invo_name, NULL, &fp)) == NULL) {
 333             adios(NULL, "unable to create temporary file in %s",
 334                   get_temp_dir());
 335         }
 336
 337         /* save a copy of the name for later removal */
 338         strncpy (infile, cp, sizeof(infile));
 339         unlink_infile = 1;
 340
 341         /* copy standard input to temporary file */
 342         while ((n = fread(buffer, 1, sizeof(buffer), stdin)) > 0) {
 343             if (fwrite(buffer, 1, n, fp) != n) {
 344                 adios (NULL, "error copying to temporary file");
 345             }
 346         }
 347         fclose (fp);
 348
 349         /* build the content structures for MIME message */
 350         ct = build_mime (infile, autobuild, dist, directives, header_encoding,
 351                          maxunencoded, verbosw);
 352
 353         /*
 354          * If ct == NULL, that means that -auto was set and a MIME version
 355          * header was already seen.  Just use the input file as the output
 356          */
 357
 358         if (!ct) {
 359             if (! (fp = fopen(infile, "r"))) {
 360                 adios(NULL, "Unable to open %s for reading", infile);
 361             }
 362             while ((n = fread(buffer, 1, sizeof(buffer), fp)) > 0) {
 363                 if (fwrite(buffer, 1, n, stdout) != n) {
 364                     adios(NULL, "error copying %s to stdout", infile);
 365                 }
 366             }
 367         } else {
 368             /* output the message */
 369             output_message_fp (ct, stdout, NULL);
 370             free_content (ct);
 371         }
 372
 373         done (0);
 374     }
 375
 376     /*
 377      * Process the composition file from a file.
 378      */
 379
 380     /* build the content structures for MIME message */
 381     ct = build_mime (compfile, autobuild, dist, directives, header_encoding,
 382                      maxunencoded, verbosw);
 383
 384     /*
 385      * If ct == NULL, that means -auto was set and we found a MIME version
 386      * header.  Simply exit and do nothing.
 387      */
 388
 389     if (! ct)
 390         done(0);
 391
 392     cts[0] = ct;
 393     cts[1] = NULL;
 394
 395     /* output MIME message to this temporary file */
 396     if ((cp = m_mktemp2(compfile, invo_name, NULL, &fp_out)) == NULL) {
 397         adios(NULL, "unable to create temporary file in %s", get_temp_dir());
 398     }
 399     strncpy(outfile, cp, sizeof(outfile));
 400     unlink_outfile = 1;
 401
 402     /* output the message */
 403     output_message_fp (ct, fp_out, outfile);
 404     fclose(fp_out);
 405
 406     /*
 407      * List the message info
 408      */
 409     if (listsw)
 410         list_all_messages (cts, headsw, sizesw, verbosw, debugsw, dispo);
 411
 412     /* Rename composition draft */
 413     snprintf (buffer, sizeof(buffer), "%s.orig", m_backup (compfile));
 414     if (rename (compfile, buffer) == NOTOK) {
 415         adios (compfile, "unable to rename comp draft %s to", buffer);
 416     }
 417
 418     /* Rename output file to take its place */
 419     if (rename (outfile, compfile) == NOTOK) {
 420         advise (outfile, "unable to rename output %s to", compfile);
 421         rename (buffer, compfile);
 422         done (1);
 423     }
 424     unlink_outfile = 0;
 425
 426     free_content (ct);
 427     done (0);
 428     return 1;
 429 }
 430
 431
 432 static void
 433 unlink_done (int status)
 434 {
 435     /*
 436      * Check if we need to remove stray
 437      * temporary files.
 438      */
 439     if (unlink_infile)
 440         (void) m_unlink (infile);
 441     if (unlink_outfile)
 442         (void) m_unlink (outfile);
 443
 444     exit (status);
 445 }