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