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 34 #ifndef lint 35 #if 0 36 static char sccsid[] = "@(#)collect.c 8.2 (Berkeley) 4/19/94"; 37 #endif 38 static const char rcsid[] = 39 "$FreeBSD$"; 40 #endif /* not lint */ 41 42 /* 43 * Mail -- a mail program 44 * 45 * Collect input from standard input, handling 46 * ~ escapes. 47 */ 48 49 #include "rcv.h" 50 #include "extern.h" 51 52 /* 53 * Read a message from standard output and return a read file to it 54 * or NULL on error. 55 */ 56 57 /* 58 * The following hokiness with global variables is so that on 59 * receipt of an interrupt signal, the partial message can be salted 60 * away on dead.letter. 61 */ 62 63 static sig_t saveint; /* Previous SIGINT value */ 64 static sig_t savehup; /* Previous SIGHUP value */ 65 static sig_t savetstp; /* Previous SIGTSTP value */ 66 static sig_t savettou; /* Previous SIGTTOU value */ 67 static sig_t savettin; /* Previous SIGTTIN value */ 68 static FILE *collf; /* File for saving away */ 69 static int hadintr; /* Have seen one SIGINT so far */ 70 71 static jmp_buf colljmp; /* To get back to work */ 72 static int colljmp_p; /* whether to long jump */ 73 static jmp_buf collabort; /* To end collection with error */ 74 75 FILE * 76 collect(hp, printheaders) 77 struct header *hp; 78 int printheaders; 79 { 80 FILE *fbuf; 81 int lc, cc, escape, eofcount, fd, c, t; 82 char linebuf[LINESIZE], tempname[PATHSIZE], *cp, getsub; 83 int omask; 84 85 collf = NULL; 86 /* 87 * Start catching signals from here, but we're still die on interrupts 88 * until we're in the main loop. 89 */ 90 omask = sigblock(sigmask(SIGINT) | sigmask(SIGHUP)); 91 if ((saveint = signal(SIGINT, SIG_IGN)) != SIG_IGN) 92 (void)signal(SIGINT, collint); 93 if ((savehup = signal(SIGHUP, SIG_IGN)) != SIG_IGN) 94 (void)signal(SIGHUP, collhup); 95 savetstp = signal(SIGTSTP, collstop); 96 savettou = signal(SIGTTOU, collstop); 97 savettin = signal(SIGTTIN, collstop); 98 if (setjmp(collabort) || setjmp(colljmp)) { 99 (void)rm(tempname); 100 goto err; 101 } 102 sigsetmask(omask & ~(sigmask(SIGINT) | sigmask(SIGHUP))); 103 104 noreset++; 105 (void)snprintf(tempname, sizeof(tempname), 106 "%s/mail.RsXXXXXXXXXX", tmpdir); 107 if ((fd = mkstemp(tempname)) == -1 || 108 (collf = Fdopen(fd, "w+")) == NULL) { 109 warn("%s", tempname); 110 goto err; 111 } 112 (void)rm(tempname); 113 114 /* 115 * If we are going to prompt for a subject, 116 * refrain from printing a newline after 117 * the headers (since some people mind). 118 */ 119 t = GTO|GSUBJECT|GCC|GNL; 120 getsub = 0; 121 if (hp->h_subject == NULL && value("interactive") != NULL && 122 (value("ask") != NULL || value("asksub") != NULL)) 123 t &= ~GNL, getsub++; 124 if (printheaders) { 125 puthead(hp, stdout, t); 126 (void)fflush(stdout); 127 } 128 if ((cp = value("escape")) != NULL) 129 escape = *cp; 130 else 131 escape = ESCAPE; 132 eofcount = 0; 133 hadintr = 0; 134 135 if (!setjmp(colljmp)) { 136 if (getsub) 137 grabh(hp, GSUBJECT); 138 } else { 139 /* 140 * Come here for printing the after-signal message. 141 * Duplicate messages won't be printed because 142 * the write is aborted if we get a SIGTTOU. 143 */ 144 cont: 145 if (hadintr) { 146 (void)fflush(stdout); 147 fprintf(stderr, 148 "\n(Interrupt -- one more to kill letter)\n"); 149 } else { 150 printf("(continue)\n"); 151 (void)fflush(stdout); 152 } 153 } 154 for (;;) { 155 colljmp_p = 1; 156 c = readline(stdin, linebuf, LINESIZE); 157 colljmp_p = 0; 158 if (c < 0) { 159 if (value("interactive") != NULL && 160 value("ignoreeof") != NULL && ++eofcount < 25) { 161 printf("Use \".\" to terminate letter\n"); 162 continue; 163 } 164 break; 165 } 166 eofcount = 0; 167 hadintr = 0; 168 if (linebuf[0] == '.' && linebuf[1] == '\0' && 169 value("interactive") != NULL && 170 (value("dot") != NULL || value("ignoreeof") != NULL)) 171 break; 172 if (linebuf[0] != escape || value("interactive") == NULL) { 173 if (putline(collf, linebuf) < 0) 174 goto err; 175 continue; 176 } 177 c = linebuf[1]; 178 switch (c) { 179 default: 180 /* 181 * On double escape, just send the single one. 182 * Otherwise, it's an error. 183 */ 184 if (c == escape) { 185 if (putline(collf, &linebuf[1]) < 0) 186 goto err; 187 else 188 break; 189 } 190 printf("Unknown tilde escape.\n"); 191 break; 192 case 'C': 193 /* 194 * Dump core. 195 */ 196 core(); 197 break; 198 case '!': 199 /* 200 * Shell escape, send the balance of the 201 * line to sh -c. 202 */ 203 shell(&linebuf[2]); 204 break; 205 case ':': 206 /* 207 * Escape to command mode, but be nice! 208 */ 209 execute(&linebuf[2], 1); 210 goto cont; 211 case '.': 212 /* 213 * Simulate end of file on input. 214 */ 215 goto out; 216 case 'q': 217 /* 218 * Force a quit of sending mail. 219 * Act like an interrupt happened. 220 */ 221 hadintr++; 222 collint(SIGINT); 223 exit(1); 224 case 'h': 225 /* 226 * Grab a bunch of headers. 227 */ 228 grabh(hp, GTO|GSUBJECT|GCC|GBCC); 229 goto cont; 230 case 't': 231 /* 232 * Add to the To list. 233 */ 234 hp->h_to = cat(hp->h_to, extract(&linebuf[2], GTO)); 235 break; 236 case 's': 237 /* 238 * Set the Subject line. 239 */ 240 cp = &linebuf[2]; 241 while (isspace(*cp)) 242 cp++; 243 hp->h_subject = savestr(cp); 244 break; 245 case 'R': 246 /* 247 * Set the Reply-To line. 248 */ 249 cp = &linebuf[2]; 250 while (isspace(*cp)) 251 cp++; 252 hp->h_replyto = savestr(cp); 253 break; 254 case 'c': 255 /* 256 * Add to the CC list. 257 */ 258 hp->h_cc = cat(hp->h_cc, extract(&linebuf[2], GCC)); 259 break; 260 case 'b': 261 /* 262 * Add stuff to blind carbon copies list. 263 */ 264 hp->h_bcc = cat(hp->h_bcc, extract(&linebuf[2], GBCC)); 265 break; 266 case 'd': 267 if (strlcpy(linebuf + 2, getdeadletter(), sizeof(linebuf) - 2) 268 >= sizeof(linebuf) - 2) { 269 printf("Line buffer overflow\n"); 270 break; 271 } 272 /* fall into . . . */ 273 case 'r': 274 /* 275 * Invoke a file: 276 * Search for the file name, 277 * then open it and copy the contents to collf. 278 */ 279 cp = &linebuf[2]; 280 while (isspace(*cp)) 281 cp++; 282 if (*cp == '\0') { 283 printf("Interpolate what file?\n"); 284 break; 285 } 286 cp = expand(cp); 287 if (cp == NULL) 288 break; 289 if (isdir(cp)) { 290 printf("%s: Directory\n", cp); 291 break; 292 } 293 if ((fbuf = Fopen(cp, "r")) == NULL) { 294 warn("%s", cp); 295 break; 296 } 297 printf("\"%s\" ", cp); 298 (void)fflush(stdout); 299 lc = 0; 300 cc = 0; 301 while (readline(fbuf, linebuf, LINESIZE) >= 0) { 302 lc++; 303 if ((t = putline(collf, linebuf)) < 0) { 304 (void)Fclose(fbuf); 305 goto err; 306 } 307 cc += t; 308 } 309 (void)Fclose(fbuf); 310 printf("%d/%d\n", lc, cc); 311 break; 312 case 'w': 313 /* 314 * Write the message on a file. 315 */ 316 cp = &linebuf[2]; 317 while (*cp == ' ' || *cp == '\t') 318 cp++; 319 if (*cp == '\0') { 320 fprintf(stderr, "Write what file!?\n"); 321 break; 322 } 323 if ((cp = expand(cp)) == NULL) 324 break; 325 rewind(collf); 326 exwrite(cp, collf, 1); 327 break; 328 case 'm': 329 case 'M': 330 case 'f': 331 case 'F': 332 /* 333 * Interpolate the named messages, if we 334 * are in receiving mail mode. Does the 335 * standard list processing garbage. 336 * If ~f is given, we don't shift over. 337 */ 338 if (forward(linebuf + 2, collf, tempname, c) < 0) 339 goto err; 340 goto cont; 341 case '?': 342 if ((fbuf = Fopen(_PATH_TILDE, "r")) == NULL) { 343 warn("%s", _PATH_TILDE); 344 break; 345 } 346 while ((t = getc(fbuf)) != EOF) 347 (void)putchar(t); 348 (void)Fclose(fbuf); 349 break; 350 case 'p': 351 /* 352 * Print out the current state of the 353 * message without altering anything. 354 */ 355 rewind(collf); 356 printf("-------\nMessage contains:\n"); 357 puthead(hp, stdout, GTO|GSUBJECT|GCC|GBCC|GNL); 358 while ((t = getc(collf)) != EOF) 359 (void)putchar(t); 360 goto cont; 361 case '|': 362 /* 363 * Pipe message through command. 364 * Collect output as new message. 365 */ 366 rewind(collf); 367 mespipe(collf, &linebuf[2]); 368 goto cont; 369 case 'v': 370 case 'e': 371 /* 372 * Edit the current message. 373 * 'e' means to use EDITOR 374 * 'v' means to use VISUAL 375 */ 376 rewind(collf); 377 mesedit(collf, c); 378 goto cont; 379 } 380 } 381 goto out; 382 err: 383 if (collf != NULL) { 384 (void)Fclose(collf); 385 collf = NULL; 386 } 387 out: 388 if (collf != NULL) 389 rewind(collf); 390 noreset--; 391 (void)sigblock(sigmask(SIGINT) | sigmask(SIGHUP)); 392 (void)signal(SIGINT, saveint); 393 (void)signal(SIGHUP, savehup); 394 (void)signal(SIGTSTP, savetstp); 395 (void)signal(SIGTTOU, savettou); 396 (void)signal(SIGTTIN, savettin); 397 (void)sigsetmask(omask); 398 return (collf); 399 } 400 401 /* 402 * Write a file, ex-like if f set. 403 */ 404 int 405 exwrite(name, fp, f) 406 char name[]; 407 FILE *fp; 408 int f; 409 { 410 FILE *of; 411 int c, lc; 412 long cc; 413 struct stat junk; 414 415 if (f) { 416 printf("\"%s\" ", name); 417 (void)fflush(stdout); 418 } 419 if (stat(name, &junk) >= 0 && S_ISREG(junk.st_mode)) { 420 if (!f) 421 fprintf(stderr, "%s: ", name); 422 fprintf(stderr, "File exists\n"); 423 return (-1); 424 } 425 if ((of = Fopen(name, "w")) == NULL) { 426 warn((char *)NULL); 427 return (-1); 428 } 429 lc = 0; 430 cc = 0; 431 while ((c = getc(fp)) != EOF) { 432 cc++; 433 if (c == '\n') 434 lc++; 435 (void)putc(c, of); 436 if (ferror(of)) { 437 warnx("%s", name); 438 (void)Fclose(of); 439 return (-1); 440 } 441 } 442 (void)Fclose(of); 443 printf("%d/%ld\n", lc, cc); 444 (void)fflush(stdout); 445 return (0); 446 } 447 448 /* 449 * Edit the message being collected on fp. 450 * On return, make the edit file the new temp file. 451 */ 452 void 453 mesedit(fp, c) 454 FILE *fp; 455 int c; 456 { 457 sig_t sigint = signal(SIGINT, SIG_IGN); 458 FILE *nf = run_editor(fp, (off_t)-1, c, 0); 459 460 if (nf != NULL) { 461 (void)fseek(nf, 0L, 2); 462 collf = nf; 463 (void)Fclose(fp); 464 } 465 (void)signal(SIGINT, sigint); 466 } 467 468 /* 469 * Pipe the message through the command. 470 * Old message is on stdin of command; 471 * New message collected from stdout. 472 * Sh -c must return 0 to accept the new message. 473 */ 474 void 475 mespipe(fp, cmd) 476 FILE *fp; 477 char cmd[]; 478 { 479 FILE *nf; 480 int fd; 481 sig_t sigint = signal(SIGINT, SIG_IGN); 482 char *sh, tempname[PATHSIZE]; 483 484 (void)snprintf(tempname, sizeof(tempname), 485 "%s/mail.ReXXXXXXXXXX", tmpdir); 486 if ((fd = mkstemp(tempname)) == -1 || 487 (nf = Fdopen(fd, "w+")) == NULL) { 488 warn("%s", tempname); 489 goto out; 490 } 491 (void)rm(tempname); 492 /* 493 * stdin = current message. 494 * stdout = new message. 495 */ 496 if ((sh = value("SHELL")) == NULL) 497 sh = _PATH_CSHELL; 498 if (run_command(sh, 499 0, fileno(fp), fileno(nf), "-c", cmd, NULL) < 0) { 500 (void)Fclose(nf); 501 goto out; 502 } 503 if (fsize(nf) == 0) { 504 fprintf(stderr, "No bytes from \"%s\" !?\n", cmd); 505 (void)Fclose(nf); 506 goto out; 507 } 508 /* 509 * Take new files. 510 */ 511 (void)fseek(nf, 0L, 2); 512 collf = nf; 513 (void)Fclose(fp); 514 out: 515 (void)signal(SIGINT, sigint); 516 } 517 518 /* 519 * Interpolate the named messages into the current 520 * message, preceding each line with a tab. 521 * Return a count of the number of characters now in 522 * the message, or -1 if an error is encountered writing 523 * the message temporary. The flag argument is 'm' if we 524 * should shift over and 'f' if not. 525 */ 526 int 527 forward(ms, fp, fn, f) 528 char ms[]; 529 FILE *fp; 530 char *fn; 531 int f; 532 { 533 int *msgvec; 534 struct ignoretab *ig; 535 char *tabst; 536 537 msgvec = (int *)salloc((msgCount+1) * sizeof(*msgvec)); 538 if (msgvec == NULL) 539 return (0); 540 if (getmsglist(ms, msgvec, 0) < 0) 541 return (0); 542 if (*msgvec == 0) { 543 *msgvec = first(0, MMNORM); 544 if (*msgvec == 0) { 545 printf("No appropriate messages\n"); 546 return (0); 547 } 548 msgvec[1] = 0; 549 } 550 if (f == 'f' || f == 'F') 551 tabst = NULL; 552 else if ((tabst = value("indentprefix")) == NULL) 553 tabst = "\t"; 554 ig = isupper(f) ? NULL : ignore; 555 printf("Interpolating:"); 556 for (; *msgvec != 0; msgvec++) { 557 struct message *mp = message + *msgvec - 1; 558 559 touch(mp); 560 printf(" %d", *msgvec); 561 if (sendmessage(mp, fp, ig, tabst) < 0) { 562 warnx("%s", fn); 563 return (-1); 564 } 565 } 566 printf("\n"); 567 return (0); 568 } 569 570 /* 571 * Print (continue) when continued after ^Z. 572 */ 573 /*ARGSUSED*/ 574 void 575 collstop(s) 576 int s; 577 { 578 sig_t old_action = signal(s, SIG_DFL); 579 580 (void)sigsetmask(sigblock(0) & ~sigmask(s)); 581 (void)kill(0, s); 582 (void)sigblock(sigmask(s)); 583 (void)signal(s, old_action); 584 if (colljmp_p) { 585 colljmp_p = 0; 586 hadintr = 0; 587 longjmp(colljmp, 1); 588 } 589 } 590 591 /* 592 * On interrupt, come here to save the partial message in ~/dead.letter. 593 * Then jump out of the collection loop. 594 */ 595 /*ARGSUSED*/ 596 void 597 collint(s) 598 int s; 599 { 600 /* 601 * the control flow is subtle, because we can be called from ~q. 602 */ 603 if (!hadintr) { 604 if (value("ignore") != NULL) { 605 printf("@"); 606 (void)fflush(stdout); 607 clearerr(stdin); 608 return; 609 } 610 hadintr = 1; 611 longjmp(colljmp, 1); 612 } 613 rewind(collf); 614 if (value("nosave") == NULL) 615 savedeadletter(collf); 616 longjmp(collabort, 1); 617 } 618 619 /*ARGSUSED*/ 620 void 621 collhup(s) 622 int s; 623 { 624 rewind(collf); 625 savedeadletter(collf); 626 /* 627 * Let's pretend nobody else wants to clean up, 628 * a true statement at this time. 629 */ 630 exit(1); 631 } 632 633 void 634 savedeadletter(fp) 635 FILE *fp; 636 { 637 FILE *dbuf; 638 int c; 639 char *cp; 640 641 if (fsize(fp) == 0) 642 return; 643 cp = getdeadletter(); 644 c = umask(077); 645 dbuf = Fopen(cp, "a"); 646 (void)umask(c); 647 if (dbuf == NULL) 648 return; 649 while ((c = getc(fp)) != EOF) 650 (void)putc(c, dbuf); 651 (void)Fclose(dbuf); 652 rewind(fp); 653 } 654