1 /* 2 * Copyright (c) 1980, 1993 3 * The Regents of the University of California. All rights reserved. 4 * 5 * Redistribution and use in source and binary forms, with or without 6 * modification, are permitted provided that the following conditions 7 * are met: 8 * 1. Redistributions of source code must retain the above copyright 9 * notice, this list of conditions and the following disclaimer. 10 * 2. Redistributions in binary form must reproduce the above copyright 11 * notice, this list of conditions and the following disclaimer in the 12 * documentation and/or other materials provided with the distribution. 13 * 3. All advertising materials mentioning features or use of this software 14 * must display the following acknowledgement: 15 * This product includes software developed by the University of 16 * California, Berkeley and its contributors. 17 * 4. Neither the name of the University nor the names of its contributors 18 * may be used to endorse or promote products derived from this software 19 * without specific prior written permission. 20 * 21 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND 22 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 23 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 24 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE 25 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 26 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 27 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 28 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 29 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 30 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 31 * SUCH DAMAGE. 32 * 33 * @(#)collect.c 8.2 (Berkeley) 4/19/94 34 * $FreeBSD: src/usr.bin/mail/collect.c,v 1.4.6.3 2003/01/06 05:46:03 mikeh Exp $ 35 * $DragonFly: src/usr.bin/mail/collect.c,v 1.2 2003/06/17 04:29:28 dillon Exp $ 36 */ 37 38 /* 39 * Mail -- a mail program 40 * 41 * Collect input from standard input, handling 42 * ~ escapes. 43 */ 44 45 #include "rcv.h" 46 #include <fcntl.h> 47 #include "extern.h" 48 49 /* 50 * Read a message from standard output and return a read file to it 51 * or NULL on error. 52 */ 53 54 /* 55 * The following hokiness with global variables is so that on 56 * receipt of an interrupt signal, the partial message can be salted 57 * away on dead.letter. 58 */ 59 60 static sig_t saveint; /* Previous SIGINT value */ 61 static sig_t savehup; /* Previous SIGHUP value */ 62 static sig_t savetstp; /* Previous SIGTSTP value */ 63 static sig_t savettou; /* Previous SIGTTOU value */ 64 static sig_t savettin; /* Previous SIGTTIN value */ 65 static FILE *collf; /* File for saving away */ 66 static int hadintr; /* Have seen one SIGINT so far */ 67 68 static jmp_buf colljmp; /* To get back to work */ 69 static int colljmp_p; /* whether to long jump */ 70 static jmp_buf collabort; /* To end collection with error */ 71 72 FILE * 73 collect(hp, printheaders) 74 struct header *hp; 75 int printheaders; 76 { 77 FILE *fbuf; 78 int lc, cc, escape, eofcount, fd, c, t; 79 char linebuf[LINESIZE], tempname[PATHSIZE], *cp, getsub; 80 sigset_t nset; 81 int longline, lastlong, rc; /* So we don't make 2 or more lines 82 out of a long input line. */ 83 84 collf = NULL; 85 /* 86 * Start catching signals from here, but we're still die on interrupts 87 * until we're in the main loop. 88 */ 89 (void)sigemptyset(&nset); 90 (void)sigaddset(&nset, SIGINT); 91 (void)sigaddset(&nset, SIGHUP); 92 (void)sigprocmask(SIG_BLOCK, &nset, NULL); 93 if ((saveint = signal(SIGINT, SIG_IGN)) != SIG_IGN) 94 (void)signal(SIGINT, collint); 95 if ((savehup = signal(SIGHUP, SIG_IGN)) != SIG_IGN) 96 (void)signal(SIGHUP, collhup); 97 savetstp = signal(SIGTSTP, collstop); 98 savettou = signal(SIGTTOU, collstop); 99 savettin = signal(SIGTTIN, collstop); 100 if (setjmp(collabort) || setjmp(colljmp)) { 101 (void)rm(tempname); 102 goto err; 103 } 104 (void)sigprocmask(SIG_UNBLOCK, &nset, NULL); 105 106 noreset++; 107 (void)snprintf(tempname, sizeof(tempname), 108 "%s/mail.RsXXXXXXXXXX", tmpdir); 109 if ((fd = mkstemp(tempname)) == -1 || 110 (collf = Fdopen(fd, "w+")) == NULL) { 111 warn("%s", tempname); 112 goto err; 113 } 114 (void)rm(tempname); 115 116 /* 117 * If we are going to prompt for a subject, 118 * refrain from printing a newline after 119 * the headers (since some people mind). 120 */ 121 t = GTO|GSUBJECT|GCC|GNL; 122 getsub = 0; 123 if (hp->h_subject == NULL && value("interactive") != NULL && 124 (value("ask") != NULL || value("asksub") != NULL)) 125 t &= ~GNL, getsub++; 126 if (printheaders) { 127 puthead(hp, stdout, t); 128 (void)fflush(stdout); 129 } 130 if ((cp = value("escape")) != NULL) 131 escape = *cp; 132 else 133 escape = ESCAPE; 134 eofcount = 0; 135 hadintr = 0; 136 lastlong = 0; 137 longline = 0; 138 139 if (!setjmp(colljmp)) { 140 if (getsub) 141 grabh(hp, GSUBJECT); 142 } else { 143 /* 144 * Come here for printing the after-signal message. 145 * Duplicate messages won't be printed because 146 * the write is aborted if we get a SIGTTOU. 147 */ 148 cont: 149 if (hadintr) { 150 (void)fflush(stdout); 151 fprintf(stderr, 152 "\n(Interrupt -- one more to kill letter)\n"); 153 } else { 154 printf("(continue)\n"); 155 (void)fflush(stdout); 156 } 157 } 158 for (;;) { 159 colljmp_p = 1; 160 c = readline(stdin, linebuf, LINESIZE); 161 colljmp_p = 0; 162 if (c < 0) { 163 if (value("interactive") != NULL && 164 value("ignoreeof") != NULL && ++eofcount < 25) { 165 printf("Use \".\" to terminate letter\n"); 166 continue; 167 } 168 break; 169 } 170 lastlong = longline; 171 longline = c == LINESIZE - 1; 172 eofcount = 0; 173 hadintr = 0; 174 if (linebuf[0] == '.' && linebuf[1] == '\0' && 175 value("interactive") != NULL && !lastlong && 176 (value("dot") != NULL || value("ignoreeof") != NULL)) 177 break; 178 if (linebuf[0] != escape || value("interactive") == NULL || 179 lastlong) { 180 if (putline(collf, linebuf, !longline) < 0) 181 goto err; 182 continue; 183 } 184 c = linebuf[1]; 185 switch (c) { 186 default: 187 /* 188 * On double escape, just send the single one. 189 * Otherwise, it's an error. 190 */ 191 if (c == escape) { 192 if (putline(collf, &linebuf[1], !longline) < 0) 193 goto err; 194 else 195 break; 196 } 197 printf("Unknown tilde escape.\n"); 198 break; 199 case 'C': 200 /* 201 * Dump core. 202 */ 203 core(); 204 break; 205 case '!': 206 /* 207 * Shell escape, send the balance of the 208 * line to sh -c. 209 */ 210 shell(&linebuf[2]); 211 break; 212 case ':': 213 case '_': 214 /* 215 * Escape to command mode, but be nice! 216 */ 217 execute(&linebuf[2], 1); 218 goto cont; 219 case '.': 220 /* 221 * Simulate end of file on input. 222 */ 223 goto out; 224 case 'q': 225 /* 226 * Force a quit of sending mail. 227 * Act like an interrupt happened. 228 */ 229 hadintr++; 230 collint(SIGINT); 231 exit(1); 232 case 'x': 233 /* 234 * Exit, do not save in dead.letter. 235 */ 236 goto err; 237 case 'h': 238 /* 239 * Grab a bunch of headers. 240 */ 241 grabh(hp, GTO|GSUBJECT|GCC|GBCC); 242 goto cont; 243 case 't': 244 /* 245 * Add to the To list. 246 */ 247 hp->h_to = cat(hp->h_to, extract(&linebuf[2], GTO)); 248 break; 249 case 's': 250 /* 251 * Set the Subject line. 252 */ 253 cp = &linebuf[2]; 254 while (isspace((unsigned char)*cp)) 255 cp++; 256 hp->h_subject = savestr(cp); 257 break; 258 case 'R': 259 /* 260 * Set the Reply-To line. 261 */ 262 cp = &linebuf[2]; 263 while (isspace((unsigned char)*cp)) 264 cp++; 265 hp->h_replyto = savestr(cp); 266 break; 267 case 'c': 268 /* 269 * Add to the CC list. 270 */ 271 hp->h_cc = cat(hp->h_cc, extract(&linebuf[2], GCC)); 272 break; 273 case 'b': 274 /* 275 * Add to the BCC list. 276 */ 277 hp->h_bcc = cat(hp->h_bcc, extract(&linebuf[2], GBCC)); 278 break; 279 case 'i': 280 case 'A': 281 case 'a': 282 /* 283 * Insert named variable in message. 284 */ 285 switch(c) { 286 case 'i': 287 cp = &linebuf[2]; 288 while(isspace((unsigned char)*cp)) 289 cp++; 290 break; 291 case 'a': 292 cp = "sign"; 293 break; 294 case 'A': 295 cp = "Sign"; 296 break; 297 default: 298 goto err; 299 } 300 301 if(*cp != '\0' && (cp = value(cp)) != NULL) { 302 printf("%s\n", cp); 303 if(putline(collf, cp, 1) < 0) 304 goto err; 305 } 306 307 break; 308 case 'd': 309 /* 310 * Read in the dead letter file. 311 */ 312 if (strlcpy(linebuf + 2, getdeadletter(), 313 sizeof(linebuf) - 2) 314 >= sizeof(linebuf) - 2) { 315 printf("Line buffer overflow\n"); 316 break; 317 } 318 /* FALLTHROUGH */ 319 case 'r': 320 case '<': 321 /* 322 * Invoke a file: 323 * Search for the file name, 324 * then open it and copy the contents to collf. 325 */ 326 cp = &linebuf[2]; 327 while (isspace((unsigned char)*cp)) 328 cp++; 329 if (*cp == '\0') { 330 printf("Interpolate what file?\n"); 331 break; 332 } 333 cp = expand(cp); 334 if (cp == NULL) 335 break; 336 if (*cp == '!') { 337 /* 338 * Insert stdout of command. 339 */ 340 char *sh; 341 int nullfd, tempfd, rc; 342 char tempname2[PATHSIZE]; 343 344 if ((nullfd = open("/dev/null", O_RDONLY, 0)) 345 == -1) { 346 warn("/dev/null"); 347 break; 348 } 349 350 (void)snprintf(tempname2, sizeof(tempname2), 351 "%s/mail.ReXXXXXXXXXX", tmpdir); 352 if ((tempfd = mkstemp(tempname2)) == -1 || 353 (fbuf = Fdopen(tempfd, "w+")) == NULL) { 354 warn("%s", tempname2); 355 break; 356 } 357 (void)unlink(tempname2); 358 359 if ((sh = value("SHELL")) == NULL) 360 sh = _PATH_CSHELL; 361 362 rc = run_command(sh, 0, nullfd, fileno(fbuf), 363 "-c", cp+1, NULL); 364 365 close(nullfd); 366 367 if (rc < 0) { 368 (void)Fclose(fbuf); 369 break; 370 } 371 372 if (fsize(fbuf) == 0) { 373 fprintf(stderr, 374 "No bytes from command \"%s\"\n", 375 cp+1); 376 (void)Fclose(fbuf); 377 break; 378 } 379 380 rewind(fbuf); 381 } else if (isdir(cp)) { 382 printf("%s: Directory\n", cp); 383 break; 384 } else if ((fbuf = Fopen(cp, "r")) == NULL) { 385 warn("%s", cp); 386 break; 387 } 388 printf("\"%s\" ", cp); 389 (void)fflush(stdout); 390 lc = 0; 391 cc = 0; 392 while ((rc = readline(fbuf, linebuf, LINESIZE)) >= 0) { 393 if (rc != LINESIZE - 1) 394 lc++; 395 if ((t = putline(collf, linebuf, 396 rc != LINESIZE - 1)) < 0) { 397 (void)Fclose(fbuf); 398 goto err; 399 } 400 cc += t; 401 } 402 (void)Fclose(fbuf); 403 printf("%d/%d\n", lc, cc); 404 break; 405 case 'w': 406 /* 407 * Write the message on a file. 408 */ 409 cp = &linebuf[2]; 410 while (*cp == ' ' || *cp == '\t') 411 cp++; 412 if (*cp == '\0') { 413 fprintf(stderr, "Write what file!?\n"); 414 break; 415 } 416 if ((cp = expand(cp)) == NULL) 417 break; 418 rewind(collf); 419 exwrite(cp, collf, 1); 420 break; 421 case 'm': 422 case 'M': 423 case 'f': 424 case 'F': 425 /* 426 * Interpolate the named messages, if we 427 * are in receiving mail mode. Does the 428 * standard list processing garbage. 429 * If ~f is given, we don't shift over. 430 */ 431 if (forward(linebuf + 2, collf, tempname, c) < 0) 432 goto err; 433 goto cont; 434 case '?': 435 if ((fbuf = Fopen(_PATH_TILDE, "r")) == NULL) { 436 warn("%s", _PATH_TILDE); 437 break; 438 } 439 while ((t = getc(fbuf)) != EOF) 440 (void)putchar(t); 441 (void)Fclose(fbuf); 442 break; 443 case 'p': 444 /* 445 * Print out the current state of the 446 * message without altering anything. 447 */ 448 rewind(collf); 449 printf("-------\nMessage contains:\n"); 450 puthead(hp, stdout, GTO|GSUBJECT|GCC|GBCC|GNL); 451 while ((t = getc(collf)) != EOF) 452 (void)putchar(t); 453 goto cont; 454 case '|': 455 /* 456 * Pipe message through command. 457 * Collect output as new message. 458 */ 459 rewind(collf); 460 mespipe(collf, &linebuf[2]); 461 goto cont; 462 case 'v': 463 case 'e': 464 /* 465 * Edit the current message. 466 * 'e' means to use EDITOR 467 * 'v' means to use VISUAL 468 */ 469 rewind(collf); 470 mesedit(collf, c); 471 goto cont; 472 } 473 } 474 goto out; 475 err: 476 if (collf != NULL) { 477 (void)Fclose(collf); 478 collf = NULL; 479 } 480 out: 481 if (collf != NULL) 482 rewind(collf); 483 noreset--; 484 (void)sigprocmask(SIG_BLOCK, &nset, NULL); 485 (void)signal(SIGINT, saveint); 486 (void)signal(SIGHUP, savehup); 487 (void)signal(SIGTSTP, savetstp); 488 (void)signal(SIGTTOU, savettou); 489 (void)signal(SIGTTIN, savettin); 490 (void)sigprocmask(SIG_UNBLOCK, &nset, NULL); 491 return (collf); 492 } 493 494 /* 495 * Write a file, ex-like if f set. 496 */ 497 int 498 exwrite(name, fp, f) 499 char name[]; 500 FILE *fp; 501 int f; 502 { 503 FILE *of; 504 int c, lc; 505 long cc; 506 struct stat junk; 507 508 if (f) { 509 printf("\"%s\" ", name); 510 (void)fflush(stdout); 511 } 512 if (stat(name, &junk) >= 0 && S_ISREG(junk.st_mode)) { 513 if (!f) 514 fprintf(stderr, "%s: ", name); 515 fprintf(stderr, "File exists\n"); 516 return (-1); 517 } 518 if ((of = Fopen(name, "w")) == NULL) { 519 warn((char *)NULL); 520 return (-1); 521 } 522 lc = 0; 523 cc = 0; 524 while ((c = getc(fp)) != EOF) { 525 cc++; 526 if (c == '\n') 527 lc++; 528 (void)putc(c, of); 529 if (ferror(of)) { 530 warnx("%s", name); 531 (void)Fclose(of); 532 return (-1); 533 } 534 } 535 (void)Fclose(of); 536 printf("%d/%ld\n", lc, cc); 537 (void)fflush(stdout); 538 return (0); 539 } 540 541 /* 542 * Edit the message being collected on fp. 543 * On return, make the edit file the new temp file. 544 */ 545 void 546 mesedit(fp, c) 547 FILE *fp; 548 int c; 549 { 550 sig_t sigint = signal(SIGINT, SIG_IGN); 551 FILE *nf = run_editor(fp, (off_t)-1, c, 0); 552 553 if (nf != NULL) { 554 (void)fseeko(nf, (off_t)0, SEEK_END); 555 collf = nf; 556 (void)Fclose(fp); 557 } 558 (void)signal(SIGINT, sigint); 559 } 560 561 /* 562 * Pipe the message through the command. 563 * Old message is on stdin of command; 564 * New message collected from stdout. 565 * Sh -c must return 0 to accept the new message. 566 */ 567 void 568 mespipe(fp, cmd) 569 FILE *fp; 570 char cmd[]; 571 { 572 FILE *nf; 573 int fd; 574 sig_t sigint = signal(SIGINT, SIG_IGN); 575 char *sh, tempname[PATHSIZE]; 576 577 (void)snprintf(tempname, sizeof(tempname), 578 "%s/mail.ReXXXXXXXXXX", tmpdir); 579 if ((fd = mkstemp(tempname)) == -1 || 580 (nf = Fdopen(fd, "w+")) == NULL) { 581 warn("%s", tempname); 582 goto out; 583 } 584 (void)rm(tempname); 585 /* 586 * stdin = current message. 587 * stdout = new message. 588 */ 589 if ((sh = value("SHELL")) == NULL) 590 sh = _PATH_CSHELL; 591 if (run_command(sh, 592 0, fileno(fp), fileno(nf), "-c", cmd, NULL) < 0) { 593 (void)Fclose(nf); 594 goto out; 595 } 596 if (fsize(nf) == 0) { 597 fprintf(stderr, "No bytes from \"%s\" !?\n", cmd); 598 (void)Fclose(nf); 599 goto out; 600 } 601 /* 602 * Take new files. 603 */ 604 (void)fseeko(nf, (off_t)0, SEEK_END); 605 collf = nf; 606 (void)Fclose(fp); 607 out: 608 (void)signal(SIGINT, sigint); 609 } 610 611 /* 612 * Interpolate the named messages into the current 613 * message, preceding each line with a tab. 614 * Return a count of the number of characters now in 615 * the message, or -1 if an error is encountered writing 616 * the message temporary. The flag argument is 'm' if we 617 * should shift over and 'f' if not. 618 */ 619 int 620 forward(ms, fp, fn, f) 621 char ms[]; 622 FILE *fp; 623 char *fn; 624 int f; 625 { 626 int *msgvec; 627 struct ignoretab *ig; 628 char *tabst; 629 630 msgvec = (int *)salloc((msgCount+1) * sizeof(*msgvec)); 631 if (msgvec == NULL) 632 return (0); 633 if (getmsglist(ms, msgvec, 0) < 0) 634 return (0); 635 if (*msgvec == 0) { 636 *msgvec = first(0, MMNORM); 637 if (*msgvec == 0) { 638 printf("No appropriate messages\n"); 639 return (0); 640 } 641 msgvec[1] = 0; 642 } 643 if (f == 'f' || f == 'F') 644 tabst = NULL; 645 else if ((tabst = value("indentprefix")) == NULL) 646 tabst = "\t"; 647 ig = isupper((unsigned char)f) ? NULL : ignore; 648 printf("Interpolating:"); 649 for (; *msgvec != 0; msgvec++) { 650 struct message *mp = message + *msgvec - 1; 651 652 touch(mp); 653 printf(" %d", *msgvec); 654 if (sendmessage(mp, fp, ig, tabst) < 0) { 655 warnx("%s", fn); 656 return (-1); 657 } 658 } 659 printf("\n"); 660 return (0); 661 } 662 663 /* 664 * Print (continue) when continued after ^Z. 665 */ 666 /*ARGSUSED*/ 667 void 668 collstop(s) 669 int s; 670 { 671 sig_t old_action = signal(s, SIG_DFL); 672 sigset_t nset; 673 674 (void)sigemptyset(&nset); 675 (void)sigaddset(&nset, s); 676 (void)sigprocmask(SIG_UNBLOCK, &nset, NULL); 677 (void)kill(0, s); 678 (void)sigprocmask(SIG_BLOCK, &nset, NULL); 679 (void)signal(s, old_action); 680 if (colljmp_p) { 681 colljmp_p = 0; 682 hadintr = 0; 683 longjmp(colljmp, 1); 684 } 685 } 686 687 /* 688 * On interrupt, come here to save the partial message in ~/dead.letter. 689 * Then jump out of the collection loop. 690 */ 691 /*ARGSUSED*/ 692 void 693 collint(s) 694 int s; 695 { 696 /* 697 * the control flow is subtle, because we can be called from ~q. 698 */ 699 if (!hadintr) { 700 if (value("ignore") != NULL) { 701 printf("@"); 702 (void)fflush(stdout); 703 clearerr(stdin); 704 return; 705 } 706 hadintr = 1; 707 longjmp(colljmp, 1); 708 } 709 rewind(collf); 710 if (value("nosave") == NULL) 711 savedeadletter(collf); 712 longjmp(collabort, 1); 713 } 714 715 /*ARGSUSED*/ 716 void 717 collhup(s) 718 int s; 719 { 720 rewind(collf); 721 savedeadletter(collf); 722 /* 723 * Let's pretend nobody else wants to clean up, 724 * a true statement at this time. 725 */ 726 exit(1); 727 } 728 729 void 730 savedeadletter(fp) 731 FILE *fp; 732 { 733 FILE *dbuf; 734 int c; 735 char *cp; 736 737 if (fsize(fp) == 0) 738 return; 739 cp = getdeadletter(); 740 c = umask(077); 741 dbuf = Fopen(cp, "a"); 742 (void)umask(c); 743 if (dbuf == NULL) 744 return; 745 while ((c = getc(fp)) != EOF) 746 (void)putc(c, dbuf); 747 (void)Fclose(dbuf); 748 rewind(fp); 749 } 750