xref: /freebsd/usr.bin/mail/edit.c (revision 0c3a8314c00f58f16823e8cd6186757d5e8bcdcc)
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$
34  */
35 
36 #ifndef lint
37 #if 0
38 static char sccsid[] = "@(#)edit.c	8.1 (Berkeley) 6/6/93";
39 #endif
40 static const char rcsid[] =
41   "$FreeBSD$";
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  */
57 int
58 editor(msgvec)
59 	int *msgvec;
60 {
61 
62 	return edit1(msgvec, 'e');
63 }
64 
65 /*
66  * Invoke the visual editor on a message list.
67  */
68 int
69 visual(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  */
81 int
82 edit1(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  */
146 FILE *
147 run_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);
225 out:
226 	return nf;
227 }
228