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
36static char sccsid[] = "@(#)edit.c	8.1 (Berkeley) 6/6/93";
37#endif
38static const char rcsid[] =
39  "$FreeBSD: src/usr.bin/mail/edit.c,v 1.9 2002/06/30 05:25:06 obrien Exp $";
40#endif /* not lint */
41
42#include <sys/cdefs.h>
43
44#include "rcv.h"
45#include <fcntl.h>
46#include "extern.h"
47
48/*
49 * Mail -- a mail program
50 *
51 * Perform message editing functions.
52 */
53
54/*
55 * Edit a message list.
56 */
57int
58editor(msgvec)
59	int *msgvec;
60{
61
62	return (edit1(msgvec, 'e'));
63}
64
65/*
66 * Invoke the visual editor on a message list.
67 */
68int
69visual(msgvec)
70	int *msgvec;
71{
72
73	return (edit1(msgvec, 'v'));
74}
75
76/*
77 * Edit a message by writing the message into a funnily-named file
78 * (which should not exist) and forking an editor on it.
79 * We get the editor from the stuff above.
80 */
81int
82edit1(msgvec, type)
83	int *msgvec;
84	int type;
85{
86	int c, i;
87	FILE *fp;
88	struct message *mp;
89	off_t size;
90
91	/*
92	 * Deal with each message to be edited . . .
93	 */
94	for (i = 0; msgvec[i] && i < msgCount; i++) {
95		sig_t sigint;
96
97		if (i > 0) {
98			char buf[100];
99			char *p;
100
101			printf("Edit message %d [ynq]? ", msgvec[i]);
102			if (fgets(buf, sizeof(buf), stdin) == 0)
103				break;
104			for (p = buf; *p == ' ' || *p == '\t'; p++)
105				;
106			if (*p == 'q')
107				break;
108			if (*p == 'n')
109				continue;
110		}
111		dot = mp = &message[msgvec[i] - 1];
112		touch(mp);
113		sigint = signal(SIGINT, SIG_IGN);
114		fp = run_editor(setinput(mp), mp->m_size, type, readonly);
115		if (fp != NULL) {
116			(void)fseeko(otf, (off_t)0, SEEK_END);
117			size = ftello(otf);
118			mp->m_block = blockof(size);
119			mp->m_offset = boffsetof(size);
120			mp->m_size = (long)fsize(fp);
121			mp->m_lines = 0;
122			mp->m_flag |= MODIFY;
123			rewind(fp);
124			while ((c = getc(fp)) != EOF) {
125				if (c == '\n')
126					mp->m_lines++;
127				if (putc(c, otf) == EOF)
128					break;
129			}
130			if (ferror(otf))
131				warnx("/tmp");
132			(void)Fclose(fp);
133		}
134		(void)signal(SIGINT, sigint);
135	}
136	return (0);
137}
138
139/*
140 * Run an editor on the file at "fpp" of "size" bytes,
141 * and return a new file pointer.
142 * Signals must be handled by the caller.
143 * "Type" is 'e' for _PATH_EX, 'v' for _PATH_VI.
144 */
145FILE *
146run_editor(fp, size, type, readonly)
147	FILE *fp;
148	off_t size;
149	int type, readonly;
150{
151	FILE *nf = NULL;
152	int t;
153	time_t modtime;
154	char *edit, tempname[PATHSIZE];
155	struct stat statb;
156
157	(void)snprintf(tempname, sizeof(tempname),
158	    "%s/mail.ReXXXXXXXXXX", tmpdir);
159	if ((t = mkstemp(tempname)) == -1 ||
160	    (nf = Fdopen(t, "w")) == NULL) {
161		warn("%s", tempname);
162		goto out;
163	}
164	if (readonly && fchmod(t, 0400) == -1) {
165		warn("%s", tempname);
166		(void)rm(tempname);
167		goto out;
168	}
169	if (size >= 0)
170		while (--size >= 0 && (t = getc(fp)) != EOF)
171			(void)putc(t, nf);
172	else
173		while ((t = getc(fp)) != EOF)
174			(void)putc(t, nf);
175	(void)fflush(nf);
176	if (fstat(fileno(nf), &statb) < 0)
177		modtime = 0;
178	else
179		modtime = statb.st_mtime;
180	if (ferror(nf)) {
181		(void)Fclose(nf);
182		warnx("%s", tempname);
183		(void)rm(tempname);
184		nf = NULL;
185		goto out;
186	}
187	if (Fclose(nf) < 0) {
188		warn("%s", tempname);
189		(void)rm(tempname);
190		nf = NULL;
191		goto out;
192	}
193	nf = NULL;
194	if ((edit = value(type == 'e' ? "EDITOR" : "VISUAL")) == NULL)
195		edit = type == 'e' ? _PATH_EX : _PATH_VI;
196	else {
197		if (*edit=='\0')
198			edit = type == 'e' ? _PATH_EX : _PATH_VI;
199	}
200	if (run_command(edit, 0, -1, -1, tempname, NULL, NULL) < 0) {
201		(void)rm(tempname);
202		goto out;
203	}
204	/*
205	 * If in read only mode or file unchanged, just remove the editor
206	 * temporary and return.
207	 */
208	if (readonly) {
209		(void)rm(tempname);
210		goto out;
211	}
212	if (stat(tempname, &statb) < 0) {
213		warn("%s", tempname);
214		goto out;
215	}
216	if (modtime == statb.st_mtime) {
217		(void)rm(tempname);
218		goto out;
219	}
220	/*
221	 * Now switch to new file.
222	 */
223	if ((nf = Fopen(tempname, "a+")) == NULL) {
224		warn("%s", tempname);
225		(void)rm(tempname);
226		goto out;
227	}
228	(void)rm(tempname);
229out:
230	return (nf);
231}
232