xref: /freebsd/usr.bin/mail/edit.c (revision 2e3507c25e42292b45a5482e116d278f5515d04d)
1 /*-
2  * SPDX-License-Identifier: BSD-3-Clause
3  *
4  * Copyright (c) 1980, 1993
5  *	The Regents of the University of California.  All rights reserved.
6  *
7  * Redistribution and use in source and binary forms, with or without
8  * modification, are permitted provided that the following conditions
9  * are met:
10  * 1. Redistributions of source code must retain the above copyright
11  *    notice, this list of conditions and the following disclaimer.
12  * 2. Redistributions in binary form must reproduce the above copyright
13  *    notice, this list of conditions and the following disclaimer in the
14  *    documentation and/or other materials provided with the distribution.
15  * 3. Neither the name of the University nor the names of its contributors
16  *    may be used to endorse or promote products derived from this software
17  *    without specific prior written permission.
18  *
19  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
20  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
21  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
22  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
23  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
24  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
25  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
26  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
27  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
28  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
29  * SUCH DAMAGE.
30  */
31 
32 #include "rcv.h"
33 #include <fcntl.h>
34 #include "extern.h"
35 
36 /*
37  * Mail -- a mail program
38  *
39  * Perform message editing functions.
40  */
41 
42 /*
43  * Edit a message list.
44  */
45 int
46 editor(void *msgvec)
47 {
48 
49 	return (edit1(msgvec, 'e'));
50 }
51 
52 /*
53  * Invoke the visual editor on a message list.
54  */
55 int
56 visual(void *msgvec)
57 {
58 
59 	return (edit1(msgvec, 'v'));
60 }
61 
62 /*
63  * Edit a message by writing the message into a funnily-named file
64  * (which should not exist) and forking an editor on it.
65  * We get the editor from the stuff above.
66  */
67 int
68 edit1(int *msgvec, int type)
69 {
70 	int c, i;
71 	FILE *fp;
72 	struct message *mp;
73 	off_t size;
74 
75 	/*
76 	 * Deal with each message to be edited . . .
77 	 */
78 	for (i = 0; i < msgCount && msgvec[i]; i++) {
79 		sig_t sigint;
80 
81 		if (i > 0) {
82 			char buf[100];
83 			char *p;
84 
85 			printf("Edit message %d [ynq]? ", msgvec[i]);
86 			if (fgets(buf, sizeof(buf), stdin) == NULL)
87 				break;
88 			for (p = buf; *p == ' ' || *p == '\t'; p++)
89 				;
90 			if (*p == 'q')
91 				break;
92 			if (*p == 'n')
93 				continue;
94 		}
95 		dot = mp = &message[msgvec[i] - 1];
96 		touch(mp);
97 		sigint = signal(SIGINT, SIG_IGN);
98 		fp = run_editor(setinput(mp), mp->m_size, type, readonly);
99 		if (fp != NULL) {
100 			(void)fseeko(otf, (off_t)0, SEEK_END);
101 			size = ftello(otf);
102 			mp->m_block = blockof(size);
103 			mp->m_offset = boffsetof(size);
104 			mp->m_size = (long)fsize(fp);
105 			mp->m_lines = 0;
106 			mp->m_flag |= MODIFY;
107 			rewind(fp);
108 			while ((c = getc(fp)) != EOF) {
109 				if (c == '\n')
110 					mp->m_lines++;
111 				if (putc(c, otf) == EOF)
112 					break;
113 			}
114 			if (ferror(otf))
115 				warnx("/tmp");
116 			(void)Fclose(fp);
117 		}
118 		(void)signal(SIGINT, sigint);
119 	}
120 	return (0);
121 }
122 
123 /*
124  * Run an editor on the file at "fpp" of "size" bytes,
125  * and return a new file pointer.
126  * Signals must be handled by the caller.
127  * "Type" is 'e' for _PATH_EX, 'v' for _PATH_VI.
128  */
129 FILE *
130 run_editor(FILE *fp, off_t size, int type, int readonly)
131 {
132 	FILE *nf = NULL;
133 	int t;
134 	time_t modtime;
135 	char *edit, tempname[PATHSIZE];
136 	struct stat statb;
137 
138 	(void)snprintf(tempname, sizeof(tempname),
139 	    "%s/mail.ReXXXXXXXXXX", tmpdir);
140 	if ((t = mkstemp(tempname)) == -1 ||
141 	    (nf = Fdopen(t, "w")) == NULL) {
142 		warn("%s", tempname);
143 		goto out;
144 	}
145 	if (readonly && fchmod(t, 0400) == -1) {
146 		warn("%s", tempname);
147 		(void)rm(tempname);
148 		goto out;
149 	}
150 	if (size >= 0)
151 		while (--size >= 0 && (t = getc(fp)) != EOF)
152 			(void)putc(t, nf);
153 	else
154 		while ((t = getc(fp)) != EOF)
155 			(void)putc(t, nf);
156 	(void)fflush(nf);
157 	if (fstat(fileno(nf), &statb) < 0)
158 		modtime = 0;
159 	else
160 		modtime = statb.st_mtime;
161 	if (ferror(nf)) {
162 		(void)Fclose(nf);
163 		warnx("%s", tempname);
164 		(void)rm(tempname);
165 		nf = NULL;
166 		goto out;
167 	}
168 	if (Fclose(nf) < 0) {
169 		warn("%s", tempname);
170 		(void)rm(tempname);
171 		nf = NULL;
172 		goto out;
173 	}
174 	nf = NULL;
175 	if ((edit = value(type == 'e' ? "EDITOR" : "VISUAL")) == NULL)
176 		edit = type == 'e' ? _PATH_EX : _PATH_VI;
177 	if (run_command(edit, 0, -1, -1, tempname, NULL) < 0) {
178 		(void)rm(tempname);
179 		goto out;
180 	}
181 	/*
182 	 * If in read only mode or file unchanged, just remove the editor
183 	 * temporary and return.
184 	 */
185 	if (readonly) {
186 		(void)rm(tempname);
187 		goto out;
188 	}
189 	if (stat(tempname, &statb) < 0) {
190 		warn("%s", tempname);
191 		goto out;
192 	}
193 	if (modtime == statb.st_mtime) {
194 		(void)rm(tempname);
195 		goto out;
196 	}
197 	/*
198 	 * Now switch to new file.
199 	 */
200 	if ((nf = Fopen(tempname, "a+")) == NULL) {
201 		warn("%s", tempname);
202 		(void)rm(tempname);
203 		goto out;
204 	}
205 	(void)rm(tempname);
206 out:
207 	return (nf);
208 }
209