NetBSD/usr.bin/mail/collect.c

/*	$NetBSD: collect.c,v 1.39 2007/06/05 17:50:22 christos Exp $	*/

/*
 * Copyright (c) 1980, 1993
 *	The Regents of the University of California.  All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 * 3. Neither the name of the University nor the names of its contributors
 *    may be used to endorse or promote products derived from this software
 *    without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 */

#include <sys/cdefs.h>
#ifndef lint
#if 0
static char sccsid[] = "@(#)collect.c	8.2 (Berkeley) 4/19/94";
#else
__RCSID("$NetBSD: collect.c,v 1.39 2007/06/05 17:50:22 christos Exp $");
#endif
#endif /* not lint */

/*
 * Mail -- a mail program
 *
 * Collect input from standard input, handling
 * ~ escapes.
 */

#include <assert.h>
#include <util.h>

#include "rcv.h"
#include "extern.h"
#include "format.h"
#ifdef MIME_SUPPORT
#include "mime.h"
#endif
#include "thread.h"


/*
 * Read a message from standard input and return a read file to it
 * or NULL on error.
 */

/*
 * The following hokiness with global variables is so that on
 * receipt of an interrupt signal, the partial message can be salted
 * away on dead.letter.
 */

static	sig_t	saveint;		/* Previous SIGINT value */
static	sig_t	savehup;		/* Previous SIGHUP value */
static	sig_t	savetstp;		/* Previous SIGTSTP value */
static	sig_t	savettou;		/* Previous SIGTTOU value */
static	sig_t	savettin;		/* Previous SIGTTIN value */
static	FILE	*collf;			/* File for saving away */
static	int	hadintr;		/* Have seen one SIGINT so far */

static	jmp_buf	colljmp;		/* To get back to work */
static	int	colljmp_p;		/* whether to long jump */
static	jmp_buf	collabort;		/* To end collection with error */


/*
 * Write a file, ex-like if f set.
 */
static int
exwrite(const char name[], FILE *fp, int f)
{
	FILE *of;
	int c;
	long cc;
	int lc;
	struct stat junk;

	if (f) {
		(void)printf("\"%s\" ", name);
		(void)fflush(stdout);
	}
	if (stat(name, &junk) >= 0 && S_ISREG(junk.st_mode)) {
		if (!f)
			(void)fprintf(stderr, "%s: ", name);
		(void)fprintf(stderr, "File exists\n");
		return -1;
	}
	if ((of = Fopen(name, "w")) == NULL) {
		warn("%s", name);
		return -1;
	}
	lc = 0;
	cc = 0;
	while ((c = getc(fp)) != EOF) {
		cc++;
		if (c == '\n')
			lc++;
		(void)putc(c, of);
		if (ferror(of)) {
			warn("%s", name);
			(void)Fclose(of);
			return -1;
		}
	}
	(void)Fclose(of);
	(void)printf("%d/%ld\n", lc, cc);
	(void)fflush(stdout);
	return 0;
}

/*
 * Edit the message being collected on fp.
 * On return, make the edit file the new temp file.
 */
static void
mesedit(FILE *fp, int c)
{
	sig_t sigint = signal(SIGINT, SIG_IGN);
	FILE *nf = run_editor(fp, (off_t)-1, c, 0);

	if (nf != NULL) {
		(void)fseek(nf, 0L, 2);
		collf = nf;
		(void)Fclose(fp);
	}
	(void)signal(SIGINT, sigint);
}

/*
 * Pipe the message through the command.
 * Old message is on stdin of command;
 * New message collected from stdout.
 * Sh -c must return 0 to accept the new message.
 */
static void
mespipe(FILE *fp, char cmd[])
{
	FILE *nf;
	sig_t sigint = signal(SIGINT, SIG_IGN);
	const char *shellcmd;
	int fd;
	char tempname[PATHSIZE];

	(void)snprintf(tempname, sizeof(tempname),
	    "%s/mail.ReXXXXXXXXXX", tmpdir);
	if ((fd = mkstemp(tempname)) == -1 ||
	    (nf = Fdopen(fd, "w+")) == NULL) {
		if (fd != -1)
			(void)close(fd);
		warn("%s", tempname);
		goto out;
	}
	(void)unlink(tempname);
	/*
	 * stdin = current message.
	 * stdout = new message.
	 */
	if ((shellcmd = value(ENAME_SHELL)) == NULL)
		shellcmd = _PATH_CSHELL;
	if (run_command(shellcmd,
	    0, fileno(fp), fileno(nf), "-c", cmd, NULL) < 0) {
		(void)Fclose(nf);
		goto out;
	}
	if (fsize(nf) == 0) {
		(void)fprintf(stderr, "No bytes from \"%s\" !?\n", cmd);
		(void)Fclose(nf);
		goto out;
	}
	/*
	 * Take new files.
	 */
	(void)fseek(nf, 0L, 2);
	collf = nf;
	(void)Fclose(fp);
out:
	(void)signal(SIGINT, sigint);
}

/*
 * Interpolate the named messages into the current
 * message, preceding each line with a tab.
 * Return a count of the number of characters now in
 * the message, or -1 if an error is encountered writing
 * the message temporary.  The flag argument is 'm' if we
 * should shift over and 'f' if not.
 */
static int
forward(char ms[], FILE *fp, char *fn, int f)
{
	int *msgvec;
	struct ignoretab *ig;
	const char *tabst;
#ifdef MIME_SUPPORT
	struct mime_info *mip;
	int retval;
#endif
	msgvec = salloc((get_msgCount() + 1) * sizeof *msgvec);
	if (msgvec == NULL)
		return 0;
	if (getmsglist(ms, msgvec, 0) < 0)
		return 0;
	if (*msgvec == 0) {
		*msgvec = first(0, MMNORM);
		if (*msgvec == 0) {
			(void)printf("No appropriate messages\n");
			return 0;
		}
		msgvec[1] = 0;
	}
	if (f == 'f' || f == 'F')
		tabst = NULL;
	else if ((tabst = value(ENAME_INDENTPREFIX)) == NULL)
		tabst = "\t";
	ig = isupper(f) ? NULL : ignore;
	(void)printf("Interpolating:");
	for (; *msgvec != 0; msgvec++) {
		struct message *mp;
		char *fmtstr;

		mp = get_message(*msgvec);
		touch(mp);
		(void)printf(" %d", *msgvec);
		(void)fflush(stdout);	/* flush stdout and the above */

		if (tabst && (fmtstr = value(ENAME_INDENT_PREAMBLE)) != NULL)
			fmsgprintf(collf, fmtstr, mp);
#ifdef MIME_SUPPORT
		mip = NULL;
		if (value(ENAME_MIME_DECODE_MSG)) {
			if ((tabst == NULL && value(ENAME_MIME_DECODE_INSERT)) ||
			    (tabst != NULL && value(ENAME_MIME_DECODE_QUOTE)))
				mip = mime_decode_open(mp);
		}
		retval = mime_sendmessage(mp, fp, ig, tabst, mip);
		mime_decode_close(mip);
		if (retval < 0)
#else
		if (sendmessage(mp, fp, ig, tabst, NULL) < 0)
#endif
		{
			warn("%s", fn);
			return -1;
		}
		if (tabst && (fmtstr = value(ENAME_INDENT_POSTSCRIPT)) != NULL)
			fmsgprintf(collf, fmtstr, mp);
			
	}
	(void)printf("\n");
	return 0;
}

/*
 * Print (continue) when continued after ^Z.
 */
/*ARGSUSED*/
static void
collstop(int s)
{
	sig_t old_action = signal(s, SIG_DFL);
	sigset_t nset;

	(void)sigemptyset(&nset);
	(void)sigaddset(&nset, s);
	(void)sigprocmask(SIG_UNBLOCK, &nset, NULL);
	(void)kill(0, s);
	(void)sigprocmask(SIG_BLOCK, &nset, NULL);
	(void)signal(s, old_action);
	if (colljmp_p) {
		colljmp_p = 0;
		hadintr = 0;
		longjmp(colljmp, 1);
	}
}

/*
 * Append the contents of the file to the end of the deadletter file.
 */
PUBLIC void
savedeadletter(FILE *fp)
{
	FILE *dbuf;
	mode_t m;
	int c;
	const char *cp;

	if (fsize(fp) == 0)
		return;
	cp = getdeadletter();
	m = umask(077);
	dbuf = Fopen(cp, "a");
	(void)umask(m);
	if (dbuf == NULL)
		return;
	while ((c = getc(fp)) != EOF)
		(void)putc(c, dbuf);
	(void)Fclose(dbuf);
	rewind(fp);
}

/*
 * On interrupt, come here to save the partial message in ~/dead.letter.
 * Then jump out of the collection loop.
 */
/*ARGSUSED*/
static void
collint(int s __unused)
{
	/*
	 * the control flow is subtle, because we can be called from ~q.
	 */
	if (!hadintr) {
		if (value(ENAME_IGNORE) != NULL) {
			(void)puts("@");
			(void)fflush(stdout);
			clearerr(stdin);
			return;
		}
		hadintr = 1;
		longjmp(colljmp, 1);
	}
	rewind(collf);
	if (value(ENAME_NOSAVE) == NULL)
		savedeadletter(collf);
	longjmp(collabort, 1);
}

/*ARGSUSED*/
static void
collhup(int s __unused)
{
	rewind(collf);
	savedeadletter(collf);
	/*
	 * Let's pretend nobody else wants to clean up,
	 * a true statement at this time.
	 */
	exit(1);
}

PUBLIC FILE *
collect(struct header *hp, int printheaders)
{
	FILE *fbuf;
	int lc, cc;
	int c, fd, t;
	char linebuf[LINESIZE];
	const char *cp;
	char tempname[PATHSIZE];
	char mailtempname[PATHSIZE];
	int lastlong, rc;	/* So we don't make 2 or more lines
					   out of a long input line. */
	int eofcount;
	int longline;
	sigset_t  nset;

	/* The following are declared volatile to avoid longjmp clobbering. */
	char volatile getsub;
	int volatile escape;

	(void)memset(mailtempname, 0, sizeof(mailtempname));
	collf = NULL;
	/*
	 * Start catching signals from here, but we're still die on interrupts
	 * until we're in the main loop.
	 */
	(void)sigemptyset(&nset);
	(void)sigaddset(&nset, SIGINT);
	(void)sigaddset(&nset, SIGHUP);
	(void)sigprocmask(SIG_BLOCK, &nset, NULL);
	if ((saveint = signal(SIGINT, SIG_IGN)) != SIG_IGN)
		(void)signal(SIGINT, collint);
	if ((savehup = signal(SIGHUP, SIG_IGN)) != SIG_IGN)
		(void)signal(SIGHUP, collhup);
	savetstp = signal(SIGTSTP, collstop);
	savettou = signal(SIGTTOU, collstop);
	savettin = signal(SIGTTIN, collstop);
	if (setjmp(collabort) || setjmp(colljmp)) {
		(void)rm(mailtempname);
		goto err;
	}
	(void)sigprocmask(SIG_UNBLOCK, &nset, NULL);

	noreset++;
	(void)snprintf(mailtempname, sizeof(mailtempname),
	    "%s/mail.RsXXXXXXXXXX", tmpdir);
	if ((fd = mkstemp(mailtempname)) == -1 ||
	    (collf = Fdopen(fd, "w+")) == NULL) {
		if (fd != -1)
			(void)close(fd);
		warn("%s", mailtempname);
		goto err;
	}
	(void)rm(mailtempname);

	/*
	 * If we are going to prompt for a subject,
	 * refrain from printing a newline after
	 * the headers (since some people mind).
	 */
	t = GTO|GSUBJECT|GCC|GNL|GSMOPTS;
	getsub = 0;
	if (hp->h_subject == NULL && value(ENAME_INTERACTIVE) != NULL &&
	    (value(ENAME_ASK) != NULL || value(ENAME_ASKSUB) != NULL))
		t &= ~GNL, getsub++;
	if (printheaders) {
		(void)puthead(hp, stdout, t);
		(void)fflush(stdout);
	}
	if ((cp = value(ENAME_ESCAPE)) != NULL)
		escape = *cp;
	else
		escape = ESCAPE;
	hadintr = 0;	/* static - no longjmp problem */
	if (!setjmp(colljmp)) {
		if (getsub)
			(void)grabh(hp, GSUBJECT);
	} else {
		/*
		 * Come here for printing the after-signal message.
		 * Duplicate messages won't be printed because
		 * the write is aborted if we get a SIGTTOU.
		 */
cont:
		if (hadintr) {
			(void)fflush(stdout);
			(void)fprintf(stderr,
			"\n(Interrupt -- one more to kill letter)\n");
		} else {
			(void)printf("(continue)\n");
			(void)fflush(stdout);
		}
	}
	eofcount = 0;	/* reset after possible longjmp */
	longline = 0;	/* reset after possible longjmp */
	for (;;) {
		colljmp_p = 1;
		c = mail_readline(stdin, linebuf, LINESIZE);
		colljmp_p = 0;
#ifdef USE_EDITLINE
		if (c < 0) {
			char *p;
			if (value(ENAME_INTERACTIVE) != NULL &&
			    (p = value(ENAME_IGNOREEOF)) != NULL &&
			    ++eofcount < (*p == 0 ? 25 : atoi(p))) {
				(void)printf("Use \".\" to terminate letter\n");
				continue;
			}
			break;
		}
#else
		if (c < 0) {
			if (value(ENAME_INTERACTIVE) != NULL &&
			    value(ENAME_IGNOREEOF) != NULL && ++eofcount < 25) {
				(void)printf("Use \".\" to terminate letter\n");
				continue;
			}
			break;
		}
#endif
		lastlong = longline;
		longline = c == LINESIZE-1;
		eofcount = 0;
		hadintr = 0;
		if (linebuf[0] == '.' && linebuf[1] == '\0' &&
		    value(ENAME_INTERACTIVE) != NULL && !lastlong &&
		    (value(ENAME_DOT) != NULL || value(ENAME_IGNOREEOF) != NULL))
			break;
		if (linebuf[0] != escape || value(ENAME_INTERACTIVE) == NULL ||
		    lastlong) {
			if (putline(collf, linebuf, !longline) < 0)
				goto err;
			continue;
		}
		c = linebuf[1];
		switch (c) {
		default:
			/*
			 * On double escape, just send the single one.
			 * Otherwise, it's an error.
			 */
			if (c == escape) {
				if (putline(collf, &linebuf[1], !longline) < 0)
					goto err;
				else
					break;
			}
			(void)printf("Unknown tilde escape.\n");
			break;
#ifdef MIME_SUPPORT
		case '@':
			hp->h_attach = mime_attach_files(hp->h_attach, &linebuf[2]);
			break;
#endif
		case 'C':
			/*
			 * Dump core.
			 */
			(void)core(NULL);
			break;
		case '!':
			/*
			 * Shell escape, send the balance of the
			 * line to sh -c.
			 */
			(void)shell(&linebuf[2]);
			break;
		case ':':
		case '_':
			/*
			 * Escape to command mode, but be nice!
			 */
			(void)execute(&linebuf[2], ec_composing);
			goto cont;
		case '.':
			/*
			 * Simulate end of file on input.
			 */
			goto out;
		case 'q':
			/*
			 * Force a quit of sending mail.
			 * Act like an interrupt happened.
			 */
			hadintr++;
			collint(SIGINT);
			exit(1);
			/*NOTREACHED*/

		case 'x':	/* exit, do not save in dead.letter */
			goto err;

		case 'h':
			/*
			 * Grab a bunch of headers.
			 */
			(void)grabh(hp, GTO|GSUBJECT|GCC|GBCC|GSMOPTS);
			goto cont;
		case 't':
			/*
			 * Add to the To list.
			 */
			hp->h_to = cat(hp->h_to, extract(&linebuf[2], GTO));
			break;
		case 's':
			/*
			 * Set the Subject list.
			 */
			cp = &linebuf[2];
			while (isspace((unsigned char)*cp))
				cp++;
			hp->h_subject = savestr(cp);
			break;
		case 'c':
			/*
			 * Add to the CC list.
			 */
			hp->h_cc = cat(hp->h_cc, extract(&linebuf[2], GCC));
			break;
		case 'b':
			/*
			 * Add stuff to blind carbon copies list.
			 */
			hp->h_bcc = cat(hp->h_bcc, extract(&linebuf[2], GBCC));
			break;
		case 'i':
		case 'A':
		case 'a':
			/*
			 * Insert named variable in message
			 */

			switch(c) {
				case 'i':			
					cp = &linebuf[2];
					while(isspace((unsigned char) *cp))
						cp++;

					break;
				case 'a':
					cp = "sign";
					break;
				case 'A':
					cp = "Sign";
					break;
				default:
					goto err;
			}

			if (*cp && (cp = value(cp)) != NULL) {
				(void)printf("%s\n", cp);			
				if (putline(collf, cp, 1) < 0)
					goto err;
			}

			break;

		case 'd':
			(void)strcpy(linebuf + 2, getdeadletter());
			/* FALLTHROUGH */
		case 'r':
		case '<':
			/*
			 * Invoke a file:
			 * Search for the file name,
			 * then open it and copy the contents to collf.
			 */
			cp = &linebuf[2];
			while (isspace((unsigned char)*cp))
				cp++;
			if (*cp == '\0') {
				(void)printf("Interpolate what file?\n");
				break;
			}

			cp = expand(cp);
			if (cp == NULL)
				break;

			if (*cp == '!') {	/* insert stdout of command */
				const char *shellcmd;
				int nullfd;
				int rc2;

				if ((nullfd = open("/dev/null", O_RDONLY, 0)) == -1) {
					warn("/dev/null");
					break;
				}

				(void)snprintf(tempname, sizeof(tempname),
				    "%s/mail.ReXXXXXXXXXX", tmpdir);
				if ((fd = mkstemp(tempname)) == -1 ||
				    (fbuf = Fdopen(fd, "w+")) == NULL) {
					if (fd != -1)
						(void)close(fd);
					warn("%s", tempname);
					break;
				}
				(void)unlink(tempname);

				if ((shellcmd = value(ENAME_SHELL)) == NULL)
					shellcmd = _PATH_CSHELL;

				rc2 = run_command(shellcmd, 0, nullfd, fileno(fbuf), "-c", cp + 1, NULL);

				(void)close(nullfd);

				if (rc2 < 0) {
					(void)Fclose(fbuf);
					break;
				}

				if (fsize(fbuf) == 0) {
					(void)fprintf(stderr, "No bytes from command \"%s\"\n", cp + 1);
					(void)Fclose(fbuf);
					break;
				}

				rewind(fbuf);
			}
			else if (isdir(cp)) {
				(void)printf("%s: Directory\n", cp);
				break;
			}
			else if ((fbuf = Fopen(cp, "r")) == NULL) {
				warn("%s", cp);
				break;
			}
			(void)printf("\"%s\" ", cp);
			(void)fflush(stdout);
			lc = 0;
			cc = 0;
			while ((rc = mail_readline(fbuf, linebuf, LINESIZE)) >= 0) {
				if (rc != LINESIZE-1) lc++;
				if ((t = putline(collf, linebuf,
						 rc != LINESIZE-1)) < 0) {
					(void)Fclose(fbuf);
					goto err;
				}
				cc += t;
			}
			(void)Fclose(fbuf);
			(void)printf("%d/%d\n", lc, cc);
			break;
		case 'w':
			/*
			 * Write the message on a file.
			 */
			cp = &linebuf[2];
			while (*cp == ' ' || *cp == '\t')
				cp++;
			if (*cp == '\0') {
				(void)fprintf(stderr, "Write what file!?\n");
				break;
			}
			if ((cp = expand(cp)) == NULL)
				break;
			rewind(collf);
			(void)exwrite(cp, collf, 1);
			break;
		case 'm':
		case 'M':
		case 'f':
		case 'F':
			/*
			 * Interpolate the named messages, if we
			 * are in receiving mail mode.  Does the
			 * standard list processing garbage.
			 * If ~f is given, we don't shift over.
			 */
			if (forward(linebuf + 2, collf, mailtempname, c) < 0)
				goto err;
			goto cont;
		case '?':
			cathelp(_PATH_TILDE);
			break;
		case 'p':
			/*
			 * Print out the current state of the
			 * message without altering anything.
			 */
			rewind(collf);
			(void)printf("-------\nMessage contains:\n");
			(void)puthead(hp, stdout, GTO|GSUBJECT|GCC|GBCC|GNL);
			while ((t = getc(collf)) != EOF)
				(void)putchar(t);
			goto cont;
		case '|':
			/*
			 * Pipe message through command.
			 * Collect output as new message.
			 */
			rewind(collf);
			mespipe(collf, &linebuf[2]);
			goto cont;
		case 'v':
		case 'e':
			/*
			 * Edit the current message.
			 * 'e' means to use EDITOR
			 * 'v' means to use VISUAL
			 */
			rewind(collf);
			mesedit(collf, c);
			goto cont;
		}
	}
	goto out;
err:
	if (collf != NULL) {
		(void)Fclose(collf);
		collf = NULL;
	}
out:
	if (collf != NULL)
		rewind(collf);
	noreset--;
	(void)sigprocmask(SIG_BLOCK, &nset, NULL);
	(void)signal(SIGINT, saveint);
	(void)signal(SIGHUP, savehup);
	(void)signal(SIGTSTP, savetstp);
	(void)signal(SIGTTOU, savettou);
	(void)signal(SIGTTIN, savettin);
	(void)sigprocmask(SIG_UNBLOCK, &nset, NULL);
	return collf;
}