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

   1 /*
   2 ** dropsbr.c -- create/read/manipulate mail drops
   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/nmh.h>
  10 #include <h/utils.h>
  11
  12 #include <h/mh.h>
  13 #include <h/dropsbr.h>
  14 #include <h/mts.h>
  15 #include <h/tws.h>
  16
  17 #ifdef HAVE_ERRNO_H
  18 # include <errno.h>
  19 #endif
  20
  21 #ifdef NTOHLSWAP
  22 # include <netinet/in.h>
  23 #else
  24 # undef ntohl
  25 # define ntohl(n) (n)
  26 #endif
  27
  28 #include <fcntl.h>
  29
  30 /*
  31 ** static prototypes
  32 */
  33 static int mbx_chk_mbox(int);
  34 static int mbx_chk_mmdf(int);
  35 static int map_open(char *, int);
  36
  37
  38 /*
  39 ** Main entry point to open/create and lock
  40 ** a file or maildrop.
  41 */
  42
  43 int
  44 mbx_open(char *file, int mbx_style, uid_t uid, gid_t gid, mode_t mode)
  45 {
  46         int j, count, fd;
  47         struct stat st;
  48
  49         j = 0;
  50
  51         /* attempt to open and lock file */
  52         for (count = 4; count > 0; count--) {
  53                 if ((fd = lkopen(file, O_RDWR | O_CREAT | O_NONBLOCK, mode))
  54                                 == NOTOK) {
  55                         switch (errno) {
  56 #if defined(FCNTL_LOCKING) || defined(LOCKF_LOCKING)
  57                         case EACCES:
  58                         case EAGAIN:
  59 #endif
  60
  61 #ifdef FLOCK_LOCKING
  62                         case EWOULDBLOCK:
  63 #endif
  64                         case ETXTBSY:
  65                                 j = errno;
  66                                 sleep(5);
  67                                 break;
  68
  69                         default:
  70                                 /* just return error */
  71                                 return NOTOK;
  72                         }
  73                 }
  74
  75                 /* good file descriptor */
  76                 break;
  77         }
  78
  79         errno = j;
  80
  81         /*
  82         ** Return if we still failed after 4 attempts,
  83         ** or we just want to skip the sanity checks.
  84         */
  85         if (fd == NOTOK || mbx_style == OTHER_FORMAT)
  86                 return fd;
  87
  88         /*
  89         ** Do sanity checks on maildrop.
  90         */
  91         if (fstat(fd, &st) == NOTOK) {
  92                 /*
  93                 ** The stat failed.  So we make sure file
  94                 ** has right ownership/modes
  95                 */
  96                 chown(file, uid, gid);
  97                 chmod(file, mode);
  98         } else if (st.st_size > (off_t) 0) {
  99                 int status;
 100
 101                 /* check the maildrop */
 102                 switch (mbx_style) {
 103                 case MMDF_FORMAT:
 104                 default:
 105                         status = mbx_chk_mmdf(fd);
 106                         break;
 107
 108                 case MBOX_FORMAT:
 109                         status = mbx_chk_mbox(fd);
 110                         break;
 111                 }
 112
 113                 /* if error, attempt to close it */
 114                 if (status == NOTOK) {
 115                         close(fd);
 116                         return NOTOK;
 117                 }
 118         }
 119
 120         return fd;
 121 }
 122
 123
 124 /*
 125 ** Check/prepare MBOX style maildrop for appending.
 126 */
 127
 128 static int
 129 mbx_chk_mbox(int fd)
 130 {
 131         /* just seek to the end */
 132         if (lseek(fd, (off_t) 0, SEEK_END) == (off_t) NOTOK)
 133                 return NOTOK;
 134
 135         return OK;
 136 }
 137
 138
 139 /*
 140 ** Check/prepare MMDF style maildrop for appending.
 141 */
 142
 143 static int
 144 mbx_chk_mmdf(int fd)
 145 {
 146         size_t count;
 147         char ldelim[BUFSIZ];
 148
 149         count = strlen(mmdlm2);
 150
 151         /* casting -count to off_t, seem to break FreeBSD 2.2.6 */
 152         if (lseek(fd, (long) (-count), SEEK_END) == (off_t) NOTOK)
 153                 return NOTOK;
 154         if (read(fd, ldelim, count) != count)
 155                 return NOTOK;
 156
 157         ldelim[count] = 0;
 158
 159         if (strcmp(ldelim, mmdlm2)!=0 && write(fd, "\n", 1) != 1
 160                         && write(fd, mmdlm2, count) != count)
 161                 return NOTOK;
 162
 163         return OK;
 164 }
 165
 166
 167 int
 168 mbx_read(FILE *fp, long pos, struct drop **drops, int noisy)
 169 {
 170         register int len, size;
 171         register long ld1, ld2;
 172         register char *bp;
 173         char buffer[BUFSIZ];
 174         register struct drop *cp, *dp, *ep, *pp;
 175
 176         pp = (struct drop *) calloc((size_t) (len = MAXFOLDER), sizeof(*dp));
 177         if (pp == NULL) {
 178                 if (noisy)
 179                         admonish(NULL, "unable to allocate drop storage");
 180                 return NOTOK;
 181         }
 182
 183         ld1 = (long) strlen(mmdlm1);
 184         ld2 = (long) strlen(mmdlm2);
 185
 186         fseek(fp, pos, SEEK_SET);
 187         for (ep = (dp = pp) + len - 1; fgets(buffer, sizeof(buffer), fp);) {
 188                 size = 0;
 189                 if (strcmp(buffer, mmdlm1) == 0)
 190                         pos += ld1, dp->d_start = (long) pos;
 191                 else {
 192                         dp->d_start = (long)pos , pos += (long) strlen(buffer);
 193                         for (bp = buffer; *bp; bp++, size++)
 194                                 if (*bp == '\n')
 195                                         size++;
 196                 }
 197
 198                 while (fgets(buffer, sizeof(buffer), fp) != NULL)
 199                         if (strcmp(buffer, mmdlm2) == 0)
 200                                 break;
 201                         else {
 202                                 pos += (long) strlen(buffer);
 203                                 for (bp = buffer; *bp; bp++, size++)
 204                                         if (*bp == '\n')
 205                                                 size++;
 206                         }
 207
 208                 if (dp->d_start != (long) pos) {
 209                         dp->d_id = 0;
 210                         dp->d_size = (long) size;
 211                         dp->d_stop = pos;
 212                         dp++;
 213                 }
 214                 pos += ld2;
 215
 216                 if (dp >= ep) {
 217                         register int curlen = dp - pp;
 218
 219                         cp = (struct drop *) mh_xrealloc((char *) pp,
 220                                         (size_t) (len += MAXFOLDER) *
 221                                         sizeof(*pp));
 222                         dp = cp + curlen, ep = (pp = cp) + len - 1;
 223                 }
 224         }
 225
 226         if (dp == pp)
 227                 free((char *) pp);
 228         else
 229                 *drops = pp;
 230         return(dp - pp);
 231 }
 232
 233
 234 int
 235 mbx_write(char *mailbox, int md, FILE *fp, int id, long last,
 236                    long pos, off_t stop, int mapping, int noisy)
 237 {
 238         register int i, j, size;
 239         off_t start;
 240         long off;
 241         register char *cp;
 242         char buffer[BUFSIZ];
 243
 244         off = (long) lseek(md, (off_t) 0, SEEK_CUR);
 245         j = strlen(mmdlm1);
 246         if (write(md, mmdlm1, j) != j)
 247                 return NOTOK;
 248         start = lseek(md, (off_t) 0, SEEK_CUR);
 249         size = 0;
 250
 251         fseek(fp, pos, SEEK_SET);
 252         while (fgets(buffer, sizeof(buffer), fp) && (pos < stop)) {
 253                 i = strlen(buffer);
 254                 for (j = 0; (j = stringdex(mmdlm1, buffer)) >= 0; buffer[j]++)
 255                         continue;
 256                 for (j = 0; (j = stringdex(mmdlm2, buffer)) >= 0; buffer[j]++)
 257                         continue;
 258                 if (write(md, buffer, i) != i)
 259                         return NOTOK;
 260                 pos += (long) i;
 261                 if (mapping)
 262                         for (cp = buffer; i-- > 0; size++)
 263                                 if (*cp++ == '\n')
 264                                         size++;
 265         }
 266
 267         stop = lseek(md, (off_t) 0, SEEK_CUR);
 268         j = strlen(mmdlm2);
 269         if (write(md, mmdlm2, j) != j)
 270                 return NOTOK;
 271         if (mapping)
 272                 map_write(mailbox, md, id, last, start, stop, off, size,
 273                                 noisy);
 274
 275         return OK;
 276 }
 277
 278
 279 /*
 280 ** Append message to end of file or maildrop.
 281 */
 282
 283 int
 284 mbx_copy(char *mailbox, int mbx_style, int md, int fd, int mapping,
 285                 char *text, int noisy)
 286 {
 287         int i, j, size;
 288         off_t start, stop;
 289         long pos;
 290         char *cp, buffer[BUFSIZ];
 291         FILE *fp;
 292
 293         pos = (long) lseek(md, (off_t) 0, SEEK_CUR);
 294         size = 0;
 295
 296         switch (mbx_style) {
 297         case MMDF_FORMAT:
 298         default:
 299                 j = strlen(mmdlm1);
 300                 if (write(md, mmdlm1, j) != j)
 301                         return NOTOK;
 302                 start = lseek(md, (off_t) 0, SEEK_CUR);
 303
 304                 if (text) {
 305                         i = strlen(text);
 306                         if (write(md, text, i) != i)
 307                                 return NOTOK;
 308                         for (cp = text; *cp++; size++)
 309                                 if (*cp == '\n')
 310                                         size++;
 311                 }
 312
 313                 while ((i = read(fd, buffer, sizeof(buffer))) > 0) {
 314                         for (j = 0; (j = stringdex(mmdlm1, buffer)) >= 0; buffer[j]++)
 315                                 continue;
 316                         for (j = 0; (j = stringdex(mmdlm2, buffer)) >= 0; buffer[j]++)
 317                                 continue;
 318                         if (write(md, buffer, i) != i)
 319                                 return NOTOK;
 320                         if (mapping)
 321                                 for (cp = buffer; i-- > 0; size++)
 322                                         if (*cp++ == '\n')
 323                                                 size++;
 324                 }
 325
 326                 stop = lseek(md, (off_t) 0, SEEK_CUR);
 327                 j = strlen(mmdlm2);
 328                 if (write(md, mmdlm2, j) != j)
 329                         return NOTOK;
 330                 if (mapping)
 331                         map_write(mailbox, md, 0, (long) 0, start, stop, pos,
 332                                         size, noisy);
 333
 334                 return (i != NOTOK ? OK : NOTOK);
 335
 336         case MBOX_FORMAT:
 337                 if ((j = dup(fd)) == NOTOK)
 338                         return NOTOK;
 339                 if ((fp = fdopen(j, "r")) == NULL) {
 340                         close(j);
 341                         return NOTOK;
 342                 }
 343                 start = lseek(md, (off_t) 0, SEEK_CUR);
 344
 345                 /* If text is given, we add it to top of message */
 346                 if (text) {
 347                         i = strlen(text);
 348                         if (write(md, text, i) != i)
 349                                 return NOTOK;
 350                         for (cp = text; *cp++; size++)
 351                                 if (*cp == '\n')
 352                                         size++;
 353                 }
 354
 355                 for (j = 0; fgets(buffer, sizeof(buffer), fp) != NULL; j++) {
 356                         /*
 357                         ** Check the first line, and make some changes.
 358                         */
 359                         if (j == 0 && !text) {
 360                                 /*
 361                                 ** Change the "Return-Path:" field
 362                                 ** (if in first line) back to "From ".
 363                                 */
 364                                 if (strncmp(buffer, "Return-Path:", 12)==0) {
 365                                         char tmpbuffer[BUFSIZ];
 366                                         char *tp, *ep, *fp;
 367
 368                                         strncpy(tmpbuffer, buffer,
 369                                                         sizeof(tmpbuffer));
 370                                         ep = tmpbuffer + 13;
 371                                         if (!(fp = strchr(ep + 1, ' ')))
 372                                                 fp = strchr(ep + 1, '\n');
 373                                         tp = dctime(dlocaltimenow());
 374                                                 snprintf(buffer, sizeof(buffer), "From %.*s  %s", (int)(fp - ep), ep, tp);
 375                                 } else if (strncmp(buffer, "X-Envelope-From:",
 376                                                 16)==0) {
 377                                         /*
 378                                         ** Change the "X-Envelope-From:"
 379                                         ** field (if first line) back
 380                                         ** to "From ".
 381                                         */
 382                                         char tmpbuffer[BUFSIZ];
 383                                         char *ep;
 384
 385                                         strncpy(tmpbuffer, buffer,
 386                                                         sizeof(tmpbuffer));
 387                                         ep = tmpbuffer + 17;
 388                                         snprintf(buffer, sizeof(buffer),
 389                                                         "From %s", ep);
 390                                 } else if (strncmp(buffer, "From ", 5)!=0) {
 391                                         /*
 392                                         ** If there is already a "From "
 393                                         ** line, then leave it alone.
 394                                         ** Else we add one.
 395                                         */
 396                                         char tmpbuffer[BUFSIZ];
 397                                         char *tp, *ep;
 398
 399                                         strncpy(tmpbuffer, buffer,
 400                                                         sizeof(tmpbuffer));
 401                                         ep = "nobody@nowhere";
 402                                         tp = dctime(dlocaltimenow());
 403                                         snprintf(buffer, sizeof(buffer),
 404                                                         "From %s  %s%s",
 405                                                         ep, tp, tmpbuffer);
 406                                 }
 407                         }
 408
 409                         /*
 410                         ** If this is not first line, and begins with
 411                         ** "From ", then prepend line with ">".
 412                         */
 413                         if (j != 0 && strncmp(buffer, "From ", 5) == 0) {
 414                                 write(md, ">", 1);
 415                                 size++;
 416                         }
 417                         i = strlen(buffer);
 418                         if (write(md, buffer, i) != i) {
 419                                 fclose(fp);
 420                                 return NOTOK;
 421                         }
 422                         if (mapping)
 423                                 for (cp = buffer; i-- > 0; size++)
 424                                         if (*cp++ == '\n')
 425                                                 size++;
 426                 }
 427                 if (write(md, "\n", 1) != 1) {
 428                         fclose(fp);
 429                         return NOTOK;
 430                 }
 431                 if (mapping)
 432                         size += 2;
 433
 434                 fclose(fp);
 435                 lseek(fd, (off_t) 0, SEEK_END);
 436                 stop = lseek(md, (off_t) 0, SEEK_CUR);
 437                 if (mapping)
 438                         map_write(mailbox, md, 0, (long) 0, start, stop, pos,
 439                                         size, noisy);
 440
 441                 return OK;
 442         }
 443 }
 444
 445
 446 int
 447 mbx_size(int md, off_t start, off_t stop)
 448 {
 449         register int i, fd;
 450         register long pos;
 451         register FILE *fp;
 452
 453         if ((fd = dup(md)) == NOTOK || (fp = fdopen(fd, "r")) == NULL) {
 454                 if (fd != NOTOK)
 455                         close(fd);
 456                 return NOTOK;
 457         }
 458
 459         fseek(fp, start, SEEK_SET);
 460         for (i = 0, pos = stop - start; pos-- > 0; i++)
 461                 if (fgetc(fp) == '\n')
 462                         i++;
 463
 464         fclose(fp);
 465         return i;
 466 }
 467
 468
 469 /*
 470 ** Close and unlock file/maildrop.
 471 */
 472
 473 int
 474 mbx_close(char *mailbox, int md)
 475 {
 476         if (lkclose(md, mailbox) == 0)
 477                 return OK;
 478         return NOTOK;
 479 }
 480
 481
 482 /*
 483 ** This function is performed implicitly by getbbent.c:
 484 **     bb->bb_map = map_name(bb->bb_file);
 485 */
 486
 487 char *
 488 map_name(char *file)
 489 {
 490         register char *cp, *dp;
 491         static char buffer[BUFSIZ];
 492
 493         if ((dp = strchr(cp = mhbasename(file), '.')) == NULL)
 494                 dp = cp + strlen(cp);
 495         if (cp == file)
 496                 snprintf(buffer, sizeof(buffer), ".%.*s%s",
 497                                 (int)(dp - cp), cp, ".map");
 498         else
 499                 snprintf (buffer, sizeof(buffer), "%.*s.%.*s%s",
 500                                 (int)(cp - file), file, (int)(dp - cp),
 501                                 cp, ".map");
 502
 503         return buffer;
 504 }
 505
 506
 507 int
 508 map_read(char *file, long pos, struct drop **drops, int noisy)
 509 {
 510         register int i, md, msgp;
 511         register char *cp;
 512         struct drop d;
 513         register struct drop *mp, *dp;
 514
 515         if ((md = open(cp = map_name(file), O_RDONLY)) == NOTOK
 516                         || map_chk(cp, md, mp = &d, pos, noisy)) {
 517                 if (md != NOTOK)
 518                         close(md);
 519                 return 0;
 520         }
 521
 522         msgp = mp->d_id;
 523         dp = (struct drop *) calloc((size_t) (msgp + 1), sizeof(*dp));
 524         if (dp == NULL) {
 525                 close(md);
 526                 return 0;
 527         }
 528
 529         memcpy((char *) dp, (char *) mp, sizeof(*dp));
 530
 531         lseek(md, (off_t) sizeof(*mp), SEEK_SET);
 532         if ((i = read(md, (char *) (dp + 1), msgp * sizeof(*dp))) <
 533                         sizeof(*dp)) {
 534                 i = 0;
 535                 free((char *) dp);
 536         } else {
 537 #ifdef NTOHLSWAP
 538                 register struct drop *tdp;
 539                 int j;
 540
 541                 for (j = 0, tdp = dp; j < i / sizeof(*dp); j++, tdp++) {
 542                         tdp->d_id = ntohl(tdp->d_id);
 543                         tdp->d_size = ntohl(tdp->d_size);
 544                         tdp->d_start = ntohl(tdp->d_start);
 545                         tdp->d_stop = ntohl(tdp->d_stop);
 546                 }
 547 #endif
 548                 *drops = dp;
 549         }
 550
 551         close(md);
 552
 553         return (i / sizeof(*dp));
 554 }
 555
 556
 557 int
 558 map_write(char *mailbox, int md, int id, long last, off_t start,
 559                    off_t stop, long pos, int size, int noisy)
 560 {
 561         register int i;
 562         int clear, fd, td;
 563         char *file;
 564         register struct drop *dp;
 565         struct drop d1, d2, *rp;
 566         register FILE *fp;
 567         struct stat st;
 568
 569         if ((fd = map_open(file = map_name(mailbox), md)) == NOTOK)
 570                 return NOTOK;
 571
 572         if ((fstat(fd, &st) == OK) && (st.st_size > 0))
 573                 clear = 0;
 574         else
 575                 clear = 1;
 576
 577         if (!clear && map_chk(file, fd, &d1, pos, noisy)) {
 578                 unlink(file);
 579                 mbx_close(file, fd);
 580                 if ((fd = map_open(file, md)) == NOTOK)
 581                         return NOTOK;
 582                 clear++;
 583         }
 584
 585         if (clear) {
 586                 if ((td = dup(md)) == NOTOK ||
 587                                 (fp = fdopen(td, "r")) == NULL) {
 588                         if (noisy)
 589                                 admonish(file, "unable to %s", td != NOTOK ?
 590                                                 "fdopen" : "dup");
 591                         if (td != NOTOK)
 592                                 close(td);
 593                         mbx_close(file, fd);
 594                         return NOTOK;
 595                 }
 596
 597                 switch (i = mbx_read(fp, 0, &rp, noisy)) {
 598                 case NOTOK:
 599                         fclose(fp);
 600                         mbx_close(file, fd);
 601                         return NOTOK;
 602
 603                 case OK:
 604                         fclose(fp);
 605                         break;
 606
 607                 default:
 608                         d1.d_id = 0;
 609                         for (dp = rp; i-- >0; dp++) {
 610                                 if (dp->d_start == start)
 611                                         dp->d_id = id;
 612                                 lseek(fd, (off_t) (++d1.d_id * sizeof(*dp)),
 613                                                 SEEK_SET);
 614                                 if (write(fd, (char *)dp, sizeof(*dp)) != sizeof(*dp)) {
 615                                         if (noisy)
 616                                                 admonish(file, "write error");
 617                                         mbx_close(file, fd);
 618                                         fclose(fp);
 619                                         return NOTOK;
 620                                 }
 621                         }
 622                         free((char *) rp);
 623                         fclose(fp);
 624                         break;
 625                 }
 626         } else {
 627                 if (last == 0)
 628                         last = d1.d_start;
 629                 dp = &d2;
 630                 dp->d_id = id;
 631                 dp->d_size = (long) (size ? size : mbx_size(fd, start, stop));
 632                 dp->d_start = start;
 633                 dp->d_stop = stop;
 634                 lseek(fd, (off_t) (++d1.d_id * sizeof(*dp)), SEEK_SET);
 635                 if (write(fd, (char *) dp, sizeof(*dp)) != sizeof(*dp)) {
 636                         if (noisy)
 637                                 admonish(file, "write error");
 638                         mbx_close(file, fd);
 639                         return NOTOK;
 640                 }
 641         }
 642
 643         dp = &d1;
 644         dp->d_size = DRVRSN;
 645         dp->d_start = (long) last;
 646         dp->d_stop = lseek(md, (off_t) 0, SEEK_CUR);
 647
 648         lseek(fd, (off_t) 0, SEEK_SET);
 649         if (write(fd, (char *) dp, sizeof(*dp)) != sizeof(*dp)) {
 650                 if (noisy)
 651                         admonish(file, "write error");
 652                 mbx_close(file, fd);
 653                 return NOTOK;
 654         }
 655
 656         mbx_close(file, fd);
 657
 658         return OK;
 659 }
 660
 661
 662 static int
 663 map_open(char *file, int md)
 664 {
 665         mode_t mode;
 666         struct stat st;
 667
 668         mode = fstat(md, &st) != NOTOK ?
 669                         (mode_t) (st.st_mode & 0777) : m_gmprot();
 670         return mbx_open(file, OTHER_FORMAT, st.st_uid, st.st_gid, mode);
 671 }
 672
 673
 674 int
 675 map_chk(char *file, int fd, struct drop *dp, long pos, int noisy)
 676 {
 677         long count;
 678         struct drop d, tmpd;
 679         register struct drop *dl;
 680
 681         if (read(fd, (char *) &tmpd, sizeof(*dp)) != sizeof(*dp)) {
 682 #ifdef notdef
 683                 admonish(NULL, "%s: missing or partial index", file);
 684 #endif /* notdef */
 685                 return NOTOK;
 686         }
 687 #ifndef NTOHLSWAP
 688         *dp = tmpd;  /* if ntohl(n)=(n), can use struct assign */
 689 #else
 690         dp->d_id = ntohl(tmpd.d_id);
 691         dp->d_size  = ntohl(tmpd.d_size);
 692         dp->d_start = ntohl(tmpd.d_start);
 693         dp->d_stop  = ntohl(tmpd.d_stop);
 694 #endif
 695
 696         if (dp->d_size != DRVRSN) {
 697                 if (noisy)
 698                         admonish(NULL, "%s: version mismatch (%d != %d)",
 699                                         file, dp->d_size, DRVRSN);
 700                 return NOTOK;
 701         }
 702
 703         if (dp->d_stop != pos) {
 704                 if (noisy && pos != (long) 0)
 705                         admonish(NULL, "%s: pointer mismatch or incomplete index (%ld!=%ld)", file, dp->d_stop, (long) pos);
 706                 return NOTOK;
 707         }
 708
 709         if ((long) ((dp->d_id + 1) * sizeof(*dp)) != (long) lseek(fd, (off_t) 0, SEEK_END)) {
 710                 if (noisy)
 711                         admonish(NULL, "%s: corrupt index(1)", file);
 712                 return NOTOK;
 713         }
 714
 715         dl = &d;
 716         count = (long) strlen(mmdlm2);
 717         lseek(fd, (off_t) (dp->d_id * sizeof(*dp)), SEEK_SET);
 718         if (read(fd, (char *) dl, sizeof(*dl)) != sizeof(*dl)
 719                         || (ntohl(dl->d_stop) != dp->d_stop
 720                                 && ntohl(dl->d_stop) + count != dp->d_stop)) {
 721                 if (noisy)
 722                         admonish(NULL, "%s: corrupt index(2)", file);
 723                 return NOTOK;
 724         }
 725
 726         return OK;
 727 }