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

   1 /* mhn.c -- display, list, cache, or store the contents of MIME messages
   2  *
   3  * This code is Copyright (c) 2002, by the authors of nmh.  See the
   4  * COPYRIGHT file in the root directory of the nmh distribution for
   5  * complete copyright information.
   6  */
   7
   8 #include "h/mh.h"
   9 #include "sbr/print_help.h"
  10 #include "sbr/error.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/fmt_scan.h"
  17 #include "h/mime.h"
  18 #include "h/mhparse.h"
  19 #include "h/mhcachesbr.h"
  20 #include "h/done.h"
  21 #include "h/utils.h"
  22 #include "mhmisc.h"
  23 #include "sbr/m_maildir.h"
  24 #include "mhfree.h"
  25 #include "mhshowsbr.h"
  26
  27 #define MHN_SWITCHES \
  28     X("auto", 0, AUTOSW) \
  29     X("noauto", 0, NAUTOSW) \
  30     X("cache", 0, CACHESW) \
  31     X("nocache", 0, NCACHESW) \
  32     X("check", 0, CHECKSW) \
  33     X("nocheck", 0, NCHECKSW) \
  34     X("headers", 0, HEADSW) \
  35     X("noheaders", 0, NHEADSW) \
  36     X("list", 0, LISTSW) \
  37     X("nolist", 0, NLISTSW) \
  38     X("realsize", 0, SIZESW) \
  39     X("norealsize", 0, NSIZESW) \
  40     X("show", 0, SHOWSW) \
  41     X("noshow", 0, NSHOWSW) \
  42     X("store", 0, STORESW) \
  43     X("nostore", 0, NSTORESW) \
  44     X("verbose", 0, VERBSW) \
  45     X("noverbose", 0, NVERBSW) \
  46     X("file file", 0, FILESW) \
  47     X("form formfile", 0, FORMSW) \
  48     X("part number", 0, PARTSW) \
  49     X("type content", 0, TYPESW) \
  50     X("rcache policy", 0, RCACHESW) \
  51     X("wcache policy", 0, WCACHESW) \
  52     X("version", 0, VERSIONSW) \
  53     X("help", 0, HELPSW) \
  54     /*                                  \
  55      * for debugging                    \
  56      */                                 \
  57     X("debug", -5, DEBUGSW) \
  58     /*                                  \
  59      * switches for moreproc/mhlproc    \
  60      */                                 \
  61     X("moreproc program", -4, PROGSW) \
  62     X("nomoreproc", -3, NPROGSW) \
  63     X("length lines", -4, LENSW) \
  64     X("width columns", -4, WIDTHSW) \
  65     /*                                  \
  66      * switches for mhbuild             \
  67      */                                 \
  68     X("build", -5, BUILDSW) \
  69     X("nobuild", -7, NBUILDSW) \
  70     X("rfc934mode", 0, RFC934SW) \
  71     X("norfc934mode", 0, NRFC934SW) \
  72
  73 #define X(sw, minchars, id) id,
  74 DEFINE_SWITCH_ENUM(MHN);
  75 #undef X
  76
  77 #define X(sw, minchars, id) { sw, minchars, id },
  78 DEFINE_SWITCH_ARRAY(MHN, switches);
  79 #undef X
  80
  81
  82 int debugsw = 0;
  83 int verbosw = 0;
  84
  85 /*
  86  * variables for mhbuild (mhn -build)
  87  */
  88 static bool buildsw;
  89 static int rfc934sw = 0;
  90
  91 /*
  92  * what action to take?
  93  */
  94 static bool cachesw;
  95 static bool listsw;
  96 static bool showsw;
  97 static bool storesw;
  98
  99 #define quitser pipeser
 100
 101 /*
 102  * static prototypes
 103  */
 104 static void pipeser (int);
 105
 106
 107 int
 108 main (int argc, char **argv)
 109 {
 110     bool sizesw = true;
 111     bool headsw = true;
 112     bool autosw = false;
 113     int msgnum, *icachesw;
 114     char *cp, *file = NULL, *folder = NULL;
 115     char *maildir, buf[100], **argp;
 116     char **arguments;
 117     char *cwd;
 118     struct msgs_array msgs = { 0, 0, NULL };
 119     struct msgs *mp = NULL;
 120     CT ct, *ctp;
 121     FILE *fp;
 122     mhstoreinfo_t info;
 123
 124     if (nmh_init(argv[0], true, true)) { return 1; }
 125
 126     set_done(freects_done);
 127
 128     arguments = getarguments (invo_name, argc, argv, 1);
 129     argp = arguments;
 130
 131     /*
 132      * Parse arguments
 133      */
 134     while ((cp = *argp++)) {
 135         if (*cp == '-') {
 136             switch (smatch (++cp, switches)) {
 137             case AMBIGSW:
 138                 ambigsw (cp, switches);
 139                 done (1);
 140             case UNKWNSW:
 141                 die("-%s unknown", cp);
 142
 143             case HELPSW:
 144                 snprintf (buf, sizeof(buf), "%s [+folder] [msgs] [switches]",
 145                         invo_name);
 146                 print_help (buf, switches, 1);
 147                 done (0);
 148             case VERSIONSW:
 149                 print_version(invo_name);
 150                 done (0);
 151
 152             case AUTOSW:
 153                 autosw = true;
 154                 continue;
 155             case NAUTOSW:
 156                 autosw = false;
 157                 continue;
 158
 159             case CACHESW:
 160                 cachesw = true;
 161                 continue;
 162             case NCACHESW:
 163                 cachesw = false;
 164                 continue;
 165
 166             case RCACHESW:
 167                 icachesw = &rcachesw;
 168                 goto do_cache;
 169             case WCACHESW:
 170                 icachesw = &wcachesw;
 171 do_cache:
 172                 if (!(cp = *argp++) || *cp == '-')
 173                     die("missing argument to %s", argp[-2]);
 174                 switch (*icachesw = smatch (cp, cache_policy)) {
 175                 case AMBIGSW:
 176                     ambigsw (cp, cache_policy);
 177                     done (1);
 178                 case UNKWNSW:
 179                     die("%s unknown", cp);
 180                 default:
 181                     break;
 182                 }
 183                 continue;
 184
 185             case CHECKSW:
 186                 checksw++;
 187                 continue;
 188             case NCHECKSW:
 189                 checksw = 0;
 190                 continue;
 191
 192             case HEADSW:
 193                 headsw = true;
 194                 continue;
 195             case NHEADSW:
 196                 headsw = false;
 197                 continue;
 198
 199             case LISTSW:
 200                 listsw = true;
 201                 continue;
 202             case NLISTSW:
 203                 listsw = false;
 204                 continue;
 205
 206             case SHOWSW:
 207                 showsw = true;
 208                 continue;
 209             case NSHOWSW:
 210                 showsw = false;
 211                 continue;
 212
 213             case SIZESW:
 214                 sizesw = true;
 215                 continue;
 216             case NSIZESW:
 217                 sizesw = false;
 218                 continue;
 219
 220             case STORESW:
 221                 storesw = true;
 222                 continue;
 223             case NSTORESW:
 224                 storesw = false;
 225                 continue;
 226
 227             case PARTSW:
 228                 if (!(cp = *argp++) || *cp == '-')
 229                     die("missing argument to %s", argp[-2]);
 230                 if (npart >= NPARTS)
 231                     die("too many parts (starting with %s), %d max",
 232                            cp, NPARTS);
 233                 parts[npart++] = cp;
 234                 continue;
 235
 236             case TYPESW:
 237                 if (!(cp = *argp++) || *cp == '-')
 238                     die("missing argument to %s", argp[-2]);
 239                 if (ntype >= NTYPES)
 240                     die("too many types (starting with %s), %d max",
 241                            cp, NTYPES);
 242                 types[ntype++] = cp;
 243                 continue;
 244
 245             case FILESW:
 246                 if (!(cp = *argp++) || (*cp == '-' && cp[1]))
 247                     die("missing argument to %s", argp[-2]);
 248                 file = *cp == '-' ? cp : path (cp, TFILE);
 249                 continue;
 250
 251             case FORMSW:
 252                 if (!(cp = *argp++) || *cp == '-')
 253                     die("missing argument to %s", argp[-2]);
 254                 free(formsw);
 255                 formsw = mh_xstrdup(etcpath(cp));
 256                 continue;
 257
 258             /*
 259              * Switches for moreproc/mhlproc
 260              */
 261             case PROGSW:
 262                 if (!(progsw = *argp++) || *progsw == '-')
 263                     die("missing argument to %s", argp[-2]);
 264                 continue;
 265             case NPROGSW:
 266                 nomore++;
 267                 continue;
 268
 269             case LENSW:
 270             case WIDTHSW:
 271                 if (!(cp = *argp++) || *cp == '-')
 272                     die("missing argument to %s", argp[-2]);
 273                 continue;
 274
 275             /*
 276              * Switches for mhbuild
 277              */
 278             case BUILDSW:
 279                 buildsw = true;
 280                 continue;
 281             case NBUILDSW:
 282                 buildsw = false;
 283                 continue;
 284             case RFC934SW:
 285                 rfc934sw = 1;
 286                 continue;
 287             case NRFC934SW:
 288                 rfc934sw = -1;
 289                 continue;
 290
 291             case VERBSW:
 292                 verbosw = 1;
 293                 continue;
 294             case NVERBSW:
 295                 verbosw = 0;
 296                 continue;
 297             case DEBUGSW:
 298                 debugsw = 1;
 299                 continue;
 300             }
 301         }
 302         if (*cp == '+' || *cp == '@') {
 303             if (folder)
 304                 die("only one folder at a time!");
 305             folder = pluspath (cp);
 306         } else
 307                 app_msgarg(&msgs, cp);
 308     }
 309
 310     /* null terminate the list of acceptable parts/types */
 311     parts[npart] = NULL;
 312     types[ntype] = NULL;
 313
 314     /*
 315      * Check if we've specified an additional profile
 316      */
 317     if ((cp = getenv ("MHN"))) {
 318         if ((fp = fopen (cp, "r"))) {
 319             readconfig(NULL, fp, cp, 0);
 320             fclose (fp);
 321         } else {
 322             admonish ("", "unable to read $MHN profile (%s)", cp);
 323         }
 324     }
 325
 326     /*
 327      * Read the standard profile setup
 328      */
 329     if ((fp = fopen (cp = etcpath ("mhn.defaults"), "r"))) {
 330         readconfig(NULL, fp, cp, 0);
 331         fclose (fp);
 332     }
 333
 334     /* Check for public cache location */
 335     if ((cache_public = context_find (nmhcache)) && *cache_public != '/')
 336         cache_public = NULL;
 337
 338     /* Check for private cache location */
 339     if (!(cache_private = context_find (nmhprivcache)))
 340         cache_private = ".cache";
 341     cache_private = mh_xstrdup(m_maildir(cache_private));
 342
 343     /*
 344      * Cache the current directory before we do any chdirs()'s.
 345      */
 346     cwd = mh_xstrdup(pwd());
 347
 348     if (!context_find ("path"))
 349         free (path ("./", TFOLDER));
 350
 351     /*
 352      * Process a mhn composition file (mhn -build)
 353      */
 354     if (buildsw) {
 355         char *vec[MAXARGS];
 356         int vecp;
 357
 358         if (showsw || storesw || cachesw)
 359             die("cannot use -build with -show, -store, -cache");
 360         if (msgs.size < 1)
 361             die("need to specify a %s composition file", invo_name);
 362         if (msgs.size > 1)
 363             die("only one %s composition file at a time", invo_name);
 364
 365         vecp = 0;
 366         vec[vecp++] = "mhbuild";
 367
 368         if (rfc934sw == 1)
 369             vec[vecp++] = "-rfc934mode";
 370         else if (rfc934sw == -1)
 371             vec[vecp++] = "-norfc934mode";
 372
 373         vec[vecp++] = msgs.msgs[0];
 374         vec[vecp] = NULL;
 375
 376         execvp ("mhbuild", vec);
 377         fprintf (stderr, "unable to exec ");
 378         _exit(1);
 379     }
 380
 381     /*
 382      * Process a mhn composition file (old MH style)
 383      */
 384     if (msgs.size == 1 && !folder && !npart && !cachesw
 385         && !showsw && !storesw && !ntype && !file
 386         && (cp = getenv ("mhdraft"))
 387         && strcmp (cp, msgs.msgs[0]) == 0) {
 388
 389         char *vec[MAXARGS];
 390         int vecp;
 391
 392         vecp = 0;
 393         vec[vecp++] = "mhbuild";
 394
 395         if (rfc934sw == 1)
 396             vec[vecp++] = "-rfc934mode";
 397         else if (rfc934sw == -1)
 398             vec[vecp++] = "-norfc934mode";
 399
 400         vec[vecp++] = cp;
 401         vec[vecp] = NULL;
 402
 403         execvp ("mhbuild", vec);
 404         fprintf (stderr, "unable to exec ");
 405         _exit(1);
 406     }
 407
 408     if (file && msgs.size)
 409         die("cannot specify msg and file at same time!");
 410
 411     /*
 412      * check if message is coming from file
 413      */
 414     if (file) {
 415         cts = mh_xcalloc(2, sizeof *cts);
 416         ctp = cts;
 417
 418         if ((ct = parse_mime (file)))
 419             *ctp++ = ct;
 420     } else {
 421         /*
 422          * message(s) are coming from a folder
 423          */
 424         if (!msgs.size)
 425             app_msgarg(&msgs, "cur");
 426         if (!folder)
 427             folder = getfolder (1);
 428         maildir = m_maildir (folder);
 429
 430         if (chdir (maildir) == NOTOK)
 431             adios (maildir, "unable to change directory to");
 432
 433         /* read folder and create message structure */
 434         if (!(mp = folder_read (folder, 1)))
 435             die("unable to read folder %s", folder);
 436
 437         /* check for empty folder */
 438         if (mp->nummsg == 0)
 439             die("no messages in %s", folder);
 440
 441         /* parse all the message ranges/sequences and set SELECTED */
 442         for (msgnum = 0; msgnum < msgs.size; msgnum++)
 443             if (!m_convert (mp, msgs.msgs[msgnum]))
 444                 done (1);
 445         seq_setprev (mp);       /* set the previous-sequence */
 446
 447         cts = mh_xcalloc(mp->numsel + 1, sizeof *cts);
 448         ctp = cts;
 449
 450         for (msgnum = mp->lowsel; msgnum <= mp->hghsel; msgnum++) {
 451             if (is_selected(mp, msgnum)) {
 452                 char *msgnam;
 453
 454                 msgnam = m_name (msgnum);
 455                 if ((ct = parse_mime (msgnam)))
 456                     *ctp++ = ct;
 457             }
 458         }
 459     }
 460
 461     if (!*cts)
 462         done (1);
 463
 464     /*
 465      * You can't give more than one of these flags
 466      * at a time.
 467      */
 468     if (showsw + listsw + storesw + cachesw > 1)
 469         die("can only use one of -show, -list, -store, -cache at same time");
 470
 471     /* If no action is specified, assume -show */
 472     if (!listsw && !showsw && !storesw && !cachesw)
 473         showsw = true;
 474
 475     userrs = true;
 476     SIGNAL (SIGQUIT, quitser);
 477     SIGNAL (SIGPIPE, pipeser);
 478
 479     /*
 480      * Get the associated umask for the relevant contents.
 481      */
 482     for (ctp = cts; *ctp; ctp++) {
 483         struct stat st;
 484
 485         ct = *ctp;
 486         if (type_ok (ct, 1) && !ct->c_umask) {
 487             if (stat (ct->c_file, &st) != NOTOK)
 488                 ct->c_umask = ~(st.st_mode & 0777);
 489             else
 490                 ct->c_umask = ~m_gmprot();
 491         }
 492     }
 493
 494     /*
 495      * List the message content
 496      */
 497     if (listsw)
 498         list_all_messages (cts, headsw, sizesw, verbosw, debugsw, 0);
 499
 500     /*
 501      * Store the message content
 502      */
 503     if (storesw) {
 504         info = mhstoreinfo_create (cts, cwd, "always", autosw, verbosw);;
 505         store_all_messages (info);
 506         mhstoreinfo_free (info);
 507     }
 508
 509     /* If reading from a folder, do some updating */
 510     if (mp) {
 511         context_replace (pfolder, folder);/* update current folder  */
 512         seq_setcur (mp, mp->hghsel);      /* update current message */
 513         seq_save (mp);                    /* synchronize sequences  */
 514         context_save ();                  /* save the context file  */
 515     }
 516
 517     /*
 518      * Cache the message content
 519      */
 520     if (cachesw)
 521         cache_all_messages (cts);
 522
 523     /*
 524      * Show the message content
 525      */
 526     if (showsw)
 527         show_all_messages (cts, 0, 0, 0);
 528
 529     /* Now free all the structures for the content */
 530     for (ctp = cts; *ctp; ctp++)
 531         free_content (*ctp);
 532
 533     free (cts);
 534     cts = NULL;
 535
 536     done (0);
 537     return 1;
 538 }
 539
 540
 541 static void
 542 pipeser (int i)
 543 {
 544     if (i == SIGQUIT) {
 545         fflush (stdout);
 546         fprintf (stderr, "\n");
 547         fflush (stderr);
 548     }
 549
 550     done (1);
 551     /* NOTREACHED */
 552 }