git.marmaro.de Git - mmh/blob - uip/burst.c

   1 /*
   2 ** burst.c -- explode digests into individual messages
   3 **
   4 ** This code is Copyright (c) 2002, by the authors of nmh.  See the
   5 ** COPYRIGHT file in the root directory of the nmh distribution for
   6 ** complete copyright information.
   7 */
   8
   9 #include <h/mh.h>
  10
  11 static struct swit switches[] = {
  12 #define INPLSW 0
  13         { "inplace", 0 },
  14 #define NINPLSW 1
  15         { "noinplace", 0 },
  16 #define QIETSW 2
  17         { "quiet", 0 },
  18 #define NQIETSW 3
  19         { "noquiet", 0 },
  20 #define VERBSW 4
  21         { "verbose", 0 },
  22 #define NVERBSW 5
  23         { "noverbose", 0 },
  24 #define VERSIONSW 6
  25         { "version", 0 },
  26 #define HELPSW 7
  27         { "help", 0 },
  28         { NULL, 0 }
  29 };
  30
  31 static char delim3[] = "-------";
  32
  33 struct smsg {
  34         long s_start;
  35         long s_stop;
  36 };
  37
  38 /*
  39 ** static prototypes
  40 */
  41 static int find_delim(int, struct smsg *);
  42 static void burst(struct msgs **, int, struct smsg *, int, int, int, char *);
  43 static void cpybrst(FILE *, FILE *, char *, char *, int);
  44
  45
  46 int
  47 main(int argc, char **argv)
  48 {
  49         int inplace = 0, quietsw = 0, verbosw = 0;
  50         int msgp = 0, hi, msgnum, numburst;
  51         char *cp, *maildir, *folder = NULL, buf[BUFSIZ];
  52         char **argp, **arguments, *msgs[MAXARGS];
  53         struct smsg *smsgs;
  54         struct msgs *mp;
  55
  56 #ifdef LOCALE
  57         setlocale(LC_ALL, "");
  58 #endif
  59         invo_name = mhbasename(argv[0]);
  60
  61         /* read user profile/context */
  62         context_read();
  63
  64         arguments = getarguments(invo_name, argc, argv, 1);
  65         argp = arguments;
  66
  67         while ((cp = *argp++)) {
  68                 if (*cp == '-') {
  69                         switch (smatch(++cp, switches)) {
  70                         case AMBIGSW:
  71                                 ambigsw(cp, switches);
  72                                 done(1);
  73                         case UNKWNSW:
  74                                 adios(NULL, "-%s unknown\n", cp);
  75
  76                         case HELPSW:
  77                                 snprintf(buf, sizeof(buf), "%s [+folder] [msgs] [switches]", invo_name);
  78                                 print_help(buf, switches, 1);
  79                                 done(1);
  80                         case VERSIONSW:
  81                                 print_version(invo_name);
  82                                 done(1);
  83
  84                         case INPLSW:
  85                                 inplace++;
  86                                 continue;
  87                         case NINPLSW:
  88                                 inplace = 0;
  89                                 continue;
  90
  91                         case QIETSW:
  92                                 quietsw++;
  93                                 continue;
  94                         case NQIETSW:
  95                                 quietsw = 0;
  96                                 continue;
  97
  98                         case VERBSW:
  99                                 verbosw++;
 100                                 continue;
 101                         case NVERBSW:
 102                                 verbosw = 0;
 103                                 continue;
 104                         }
 105                 }
 106                 if (*cp == '+' || *cp == '@') {
 107                         if (folder)
 108                                 adios(NULL, "only one folder at a time!");
 109                         else
 110                                 folder = getcpy(expandfol(cp));
 111                 } else {
 112                         msgs[msgp++] = cp;
 113                 }
 114         }
 115
 116         if (!msgp)
 117                 msgs[msgp++] = seq_cur;
 118         if (!folder)
 119                 folder = getcurfol();
 120         maildir = toabsdir(folder);
 121
 122         if (chdir(maildir) == NOTOK)
 123                 adios(maildir, "unable to change directory to");
 124
 125         /* read folder and create message structure */
 126         if (!(mp = folder_read(folder)))
 127                 adios(NULL, "unable to read folder %s", folder);
 128
 129         /* check for empty folder */
 130         if (mp->nummsg == 0)
 131                 adios(NULL, "no messages in %s", folder);
 132
 133         /* parse all the message ranges/sequences and set SELECTED */
 134         for (msgnum = 0; msgnum < msgp; msgnum++)
 135                 if (!m_convert(mp, msgs[msgnum]))
 136                         done(1);
 137         seq_setprev(mp);  /* set the previous-sequence */
 138
 139         smsgs = (struct smsg *)
 140                 calloc((size_t) (MAXFOLDER + 2), sizeof(*smsgs));
 141         if (smsgs == NULL)
 142                 adios(NULL, "unable to allocate burst storage");
 143
 144         hi = mp->hghmsg + 1;
 145
 146         /* burst all the SELECTED messages */
 147         for (msgnum = mp->lowsel; msgnum <= mp->hghsel; msgnum++) {
 148                 if (is_selected(mp, msgnum)) {
 149                         if ((numburst = find_delim(msgnum, smsgs)) >= 1) {
 150                                 if (verbosw)
 151                                         printf("%d message%s exploded from digest %d\n", numburst, numburst > 1 ? "s" : "", msgnum);
 152                                 burst(&mp, msgnum, smsgs, numburst, inplace, verbosw, maildir);
 153                         } else {
 154                                 if (numburst == 0) {
 155                                         if (!quietsw)
 156                                                 admonish(NULL, "message %d not in digest format", msgnum);
 157                                 }  /* this pair of braces was missing before 1999-07-15 */
 158                                 else
 159                                         adios(NULL, "burst() botch -- you lose big");
 160                         }
 161                 }
 162         }
 163
 164         free((char *) smsgs);
 165         context_replace(curfolder, folder);  /* update current folder */
 166
 167         /*
 168         ** If -inplace is given, then the first message burst becomes
 169         ** the current message (which will now show a table of contents).
 170         ** Otherwise, the first message extracted from the first digest
 171         ** becomes the current message.
 172         */
 173         if (inplace) {
 174                 if (mp->lowsel != mp->curmsg)
 175                         seq_setcur(mp, mp->lowsel);
 176         } else {
 177                 if (hi <= mp->hghmsg)
 178                         seq_setcur(mp, hi);
 179         }
 180
 181         seq_save(mp);  /* synchronize message sequences */
 182         context_save(); /* save the context file */
 183         folder_free(mp); /* free folder/message structure */
 184         done(0);
 185         return 1;
 186 }
 187
 188
 189 /*
 190 ** Scan the message and find the beginning and
 191 ** end of all the messages in the digest.
 192 */
 193
 194 static int
 195 find_delim(int msgnum, struct smsg *smsgs)
 196 {
 197         int ld3, wasdlm, msgp;
 198         long pos;
 199         char c, *msgnam;
 200         int cc;
 201         char buffer[BUFSIZ];
 202         FILE *in;
 203
 204         ld3 = strlen(delim3);
 205
 206         if ((in = fopen(msgnam = m_name(msgnum), "r")) == NULL)
 207                 adios(msgnam, "unable to read message");
 208
 209         for (msgp = 0, pos = 0L; msgp <= MAXFOLDER;) {
 210                 while (fgets(buffer, sizeof(buffer), in) && buffer[0] == '\n')
 211                         pos += (long) strlen(buffer);
 212                 if (feof(in))
 213                         break;
 214                 fseek(in, pos, SEEK_SET);
 215                 smsgs[msgp].s_start = pos;
 216
 217                 for (c = 0; fgets(buffer, sizeof(buffer), in); c = buffer[0]) {
 218                         if (strncmp(buffer, delim3, ld3) == 0
 219                                         && (msgp == 1 || c == '\n')) {
 220                                 cc = getc(in);
 221                                 ungetc(cc, in);
 222                                 if (cc == '\n' || cc == EOF) {
 223                                         break;
 224                                 }
 225                         } else
 226                                 pos += (long) strlen(buffer);
 227                 }
 228
 229                 wasdlm = strncmp(buffer, delim3, ld3) == 0;
 230                 if (smsgs[msgp].s_start != pos)
 231                         smsgs[msgp++].s_stop = (c == '\n' && wasdlm) ?
 232                                         pos - 1 : pos;
 233                 if (feof(in)) {
 234                         break;
 235                 }
 236                 pos += (long) strlen(buffer);
 237         }
 238
 239         fclose(in);
 240         return (msgp - 1);  /* toss "End of XXX Digest" */
 241 }
 242
 243
 244 /*
 245 ** Burst out the messages in the digest into the folder
 246 */
 247
 248 static void
 249 burst(struct msgs **mpp, int msgnum, struct smsg *smsgs, int numburst,
 250         int inplace, int verbosw, char *maildir)
 251 {
 252         int i, j, mode;
 253         char *msgnam;
 254         char f1[BUFSIZ], f2[BUFSIZ], f3[BUFSIZ];
 255         FILE *in, *out;
 256         struct stat st;
 257         struct msgs *mp;
 258
 259         if ((in = fopen(msgnam = m_name(msgnum), "r")) == NULL)
 260                 adios(msgnam, "unable to read message");
 261
 262         mode = fstat(fileno(in), &st) != NOTOK ?
 263                         (int)(st.st_mode & 0777) : m_gmprot();
 264         mp = *mpp;
 265
 266         /*
 267         ** See if we have enough space in the folder
 268         ** structure for all the new messages.
 269         */
 270         if ((mp->hghmsg + numburst > mp->hghoff) &&
 271                 !(mp = folder_realloc(mp, mp->lowoff, mp->hghmsg + numburst)))
 272                 adios(NULL, "unable to allocate folder storage");
 273         *mpp = mp;
 274
 275         j = mp->hghmsg;  /* old value */
 276         mp->hghmsg += numburst;
 277         mp->nummsg += numburst;
 278
 279         /*
 280         ** If this is not the highest SELECTED message, then
 281         ** increment mp->hghsel by numburst, since the highest
 282         ** SELECTED is about to be slid down by that amount.
 283         */
 284         if (msgnum < mp->hghsel)
 285                 mp->hghsel += numburst;
 286
 287         /*
 288         ** If -inplace is given, renumber the messages after the
 289         ** source message, to make room for each of the messages
 290         ** contained within the digest.
 291         **
 292         ** This is equivalent to refiling a message from the point
 293         ** of view of the external hooks.
 294         */
 295         if (inplace) {
 296                 for (i = mp->hghmsg; j > msgnum; i--, j--) {
 297                         strncpy(f1, m_name(i), sizeof(f1));
 298                         strncpy(f2, m_name(j), sizeof(f2));
 299                         if (does_exist(mp, j)) {
 300                                 if (verbosw)
 301                                         printf("message %d becomes message %d\n", j, i);
 302
 303                                 if (rename(f2, f1) == NOTOK)
 304                                         admonish(f1, "unable to rename %s to", f2);
 305
 306                                 snprintf(f1, sizeof (f1), "%s/%d", maildir, i);
 307                                 snprintf(f2, sizeof (f2), "%s/%d", maildir, j);
 308                                 ext_hook("ref-hook", f1, f2);
 309
 310                                 copy_msg_flags(mp, i, j);
 311                                 clear_msg_flags(mp, j);
 312                                 mp->msgflags |= SEQMOD;
 313                         }
 314                 }
 315         }
 316
 317         unset_selected(mp, msgnum);
 318
 319         /* new hghmsg is hghmsg + numburst
 320         **
 321         ** At this point, there is an array of numburst smsgs, each
 322         ** element of which contains the starting and stopping offsets
 323         ** (seeks) of the message in the digest.  The inplace flag is set
 324         ** if the original digest is replaced by a message containing
 325         ** the table of contents.  smsgs[0] is that table of contents.
 326         ** Go through the message numbers in reverse order (high to low).
 327         **
 328         ** Set f1 to the name of the destination message, f2 to the name
 329         ** of a scratch file.  Extract a message from the digest to the
 330         ** scratch file.  Move the original message to a backup file if
 331         ** the destination message number is the same as the number of
 332         ** the original message, which only happens if the inplace flag
 333         ** is set.  Then move the scratch file to the destination message.
 334         **
 335         ** Moving the original message to the backup file is equivalent
 336         ** to deleting the message from the point of view of the external
 337         ** hooks.  And bursting each message is equivalent to adding a
 338         ** new message.
 339         */
 340
 341         i = inplace ? msgnum + numburst : mp->hghmsg;
 342         for (j = numburst; j >= (inplace ? 0 : 1); i--, j--) {
 343                 strncpy(f1, m_name(i), sizeof(f1));
 344                 strncpy(f2, m_mktemp(invo_name, NULL, &out), sizeof(f2));
 345
 346                 if (verbosw && i != msgnum)
 347                         printf("message %d of digest %d becomes message %d\n",
 348                                         j, msgnum, i);
 349
 350                 chmod(f2, mode);
 351                 fseek(in, smsgs[j].s_start, SEEK_SET);
 352                 cpybrst(in, out, msgnam, f2,
 353                                 (int) (smsgs[j].s_stop - smsgs[j].s_start));
 354                 fclose(out);
 355
 356                 if (i == msgnum) {
 357                         strncpy(f3, m_backup(f1), sizeof(f3));
 358                         if (rename(f1, f3) == NOTOK)
 359                                 admonish(f3, "unable to rename %s to", f1);
 360
 361                         snprintf(f3, sizeof (f3), "%s/%d", maildir, i);
 362                         ext_hook("del-hook", f3, NULL);
 363                 }
 364                 if (rename(f2, f1) == NOTOK)
 365                         admonish(f1, "unable to rename %s to", f2);
 366
 367                 snprintf(f3, sizeof (f3), "%s/%d", maildir, i);
 368                 ext_hook("add-hook", f3, NULL);
 369
 370                 copy_msg_flags(mp, i, msgnum);
 371                 mp->msgflags |= SEQMOD;
 372         }
 373
 374         fclose(in);
 375 }
 376
 377
 378 #define S1  0
 379 #define S2  1
 380 #define S3  2
 381
 382 /*
 383 ** Copy a mesage which is being burst out of a digest.
 384 ** It will remove any "dashstuffing" in the message.
 385 */
 386
 387 static void
 388 cpybrst(FILE *in, FILE *out, char *ifile, char *ofile, int len)
 389 {
 390         register int c, state;
 391
 392         for (state = S1; (c = fgetc(in)) != EOF && len > 0; len--) {
 393                 if (c == 0)
 394                         continue;
 395                 switch (state) {
 396                 case S1:
 397                         switch (c) {
 398                         case '-':
 399                                 state = S3;
 400                                 break;
 401
 402                         default:
 403                                 state = S2;
 404                         case '\n':
 405                                 fputc(c, out);
 406                                 break;
 407                         }
 408                         break;
 409
 410                 case S2:
 411                         switch (c) {
 412                         case '\n':
 413                                 state = S1;
 414                         default:
 415                                 fputc(c, out);
 416                                 break;
 417                         }
 418                         break;
 419
 420                 case S3:
 421                         switch (c) {
 422                         case ' ':
 423                                 state = S2;
 424                                 break;
 425
 426                         default:
 427                                 state = (c == '\n') ? S1 : S2;
 428                                 fputc('-', out);
 429                                 fputc(c, out);
 430                                 break;
 431                         }
 432                         break;
 433                 }
 434         }
 435
 436         if (ferror(in) && !feof(in))
 437                 adios(ifile, "error reading");
 438         if (ferror(out))
 439                 adios(ofile, "error writing");
 440 }