edit.c revision 74769
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 * $FreeBSD: head/usr.bin/mail/edit.c 74769 2001-03-25 04:57:05Z mikeh $
34 */
35
36#ifndef lint
37#if 0
38static char sccsid[] = "@(#)edit.c	8.1 (Berkeley) 6/6/93";
39#endif
40static const char rcsid[] =
41  "$FreeBSD: head/usr.bin/mail/edit.c 74769 2001-03-25 04:57:05Z mikeh $";
42#endif /* not lint */
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	register int c;
87	int i;
88	FILE *fp;
89	register struct message *mp;
90	off_t size;
91
92	/*
93	 * Deal with each message to be edited . . .
94	 */
95	for (i = 0; msgvec[i] && i < msgCount; i++) {
96		sig_t sigint;
97
98		if (i > 0) {
99			char buf[100];
100			char *p;
101
102			printf("Edit message %d [ynq]? ", msgvec[i]);
103			if (fgets(buf, sizeof buf, stdin) == 0)
104				break;
105			for (p = buf; *p == ' ' || *p == '\t'; p++)
106				;
107			if (*p == 'q')
108				break;
109			if (*p == 'n')
110				continue;
111		}
112		dot = mp = &message[msgvec[i] - 1];
113		touch(mp);
114		sigint = signal(SIGINT, SIG_IGN);
115		fp = run_editor(setinput(mp), mp->m_size, type, readonly);
116		if (fp != NULL) {
117			(void) fseek(otf, 0L, 2);
118			size = ftell(otf);
119			mp->m_block = blockof(size);
120			mp->m_offset = boffsetof(size);
121			mp->m_size = fsize(fp);
122			mp->m_lines = 0;
123			mp->m_flag |= MODIFY;
124			rewind(fp);
125			while ((c = getc(fp)) != EOF) {
126				if (c == '\n')
127					mp->m_lines++;
128				if (putc(c, otf) == EOF)
129					break;
130			}
131			if (ferror(otf))
132				warnx("/tmp");
133			(void) Fclose(fp);
134		}
135		(void) signal(SIGINT, sigint);
136	}
137	return 0;
138}
139
140/*
141 * Run an editor on the file at "fpp" of "size" bytes,
142 * and return a new file pointer.
143 * Signals must be handled by the caller.
144 * "Type" is 'e' for _PATH_EX, 'v' for _PATH_VI.
145 */
146FILE *
147run_editor(fp, size, type, readonly)
148	register FILE *fp;
149	off_t size;
150	int type, readonly;
151{
152	register FILE *nf = NULL;
153	register int t;
154	time_t modtime;
155	char *edit, tempname[PATHSIZE];
156	struct stat statb;
157
158	snprintf(tempname, sizeof(tempname), "%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")) == NOSTR)
195		edit = type == 'e' ? _PATH_EX : _PATH_VI;
196	if (run_command(edit, 0, -1, -1, tempname, NOSTR, NOSTR) < 0) {
197		(void) rm(tempname);
198		goto out;
199	}
200	/*
201	 * If in read only mode or file unchanged, just remove the editor
202	 * temporary and return.
203	 */
204	if (readonly) {
205		(void) rm(tempname);
206		goto out;
207	}
208	if (stat(tempname, &statb) < 0) {
209		warn("%s", tempname);
210		goto out;
211	}
212	if (modtime == statb.st_mtime) {
213		(void) rm(tempname);
214		goto out;
215	}
216	/*
217	 * Now switch to new file.
218	 */
219	if ((nf = Fopen(tempname, "a+")) == NULL) {
220		warn("%s", tempname);
221		(void) rm(tempname);
222		goto out;
223	}
224	(void) rm(tempname);
225out:
226	return nf;
227}
228