xref: /netbsd-src/usr.bin/mail/tty.c (revision 404fbe5fb94ca1e054339640cabb2801ce52dd30)
1 /*	$NetBSD: tty.c,v 1.27 2006/11/28 18:45:32 christos Exp $	*/
2 
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 <sys/cdefs.h>
33 #ifndef lint
34 #if 0
35 static char sccsid[] = "@(#)tty.c	8.2 (Berkeley) 6/6/93";
36 #else
37 __RCSID("$NetBSD: tty.c,v 1.27 2006/11/28 18:45:32 christos Exp $");
38 #endif
39 #endif /* not lint */
40 
41 /*
42  * Mail -- a mail program
43  *
44  * Generally useful tty stuff.
45  */
46 
47 #include "rcv.h"
48 #include "extern.h"
49 #ifdef USE_EDITLINE
50 #include "complete.h"
51 #endif
52 
53 #if !defined(USE_EDITLINE) || !defined(TIOCSTI)
54 static	cc_t	c_erase;		/* Current erase char */
55 static	cc_t	c_kill;			/* Current kill char */
56 #endif
57 #ifndef USE_EDITLINE
58 static	jmp_buf	rewrite;		/* Place to go when continued */
59 #endif
60 static	jmp_buf	intjmp;			/* Place to go when interrupted */
61 #ifndef TIOCSTI
62 static	int	ttyset;			/* We must now do erase/kill */
63 #endif
64 
65 
66 /*
67  * Read up a header from standard input.
68  * The source string has the preliminary contents to
69  * be read.
70  *
71  */
72 #ifdef USE_EDITLINE
73 static char *
74 readtty(const char pr[], char src[])
75 {
76 	char *line;
77 	line = my_gets(&elm.string, pr, src);
78 #if 0
79 	return line ? savestr(line) : __UNCONST("");
80 #else
81 	if (line)
82 		return savestr(line);
83 	else
84 		return __UNCONST("");
85 #endif
86 }
87 
88 #else /* USE_EDITLINE */
89 
90 /*
91  * Receipt continuation.
92  */
93 static void
94 ttystop(int s)
95 {
96 	sig_t old_action = signal(s, SIG_DFL);
97 	sigset_t nset;
98 
99 	(void)sigemptyset(&nset);
100 	(void)sigaddset(&nset, s);
101 	(void)sigprocmask(SIG_BLOCK, &nset, NULL);
102 	(void)kill(0, s);
103 	(void)sigprocmask(SIG_UNBLOCK, &nset, NULL);
104 	(void)signal(s, old_action);
105 	longjmp(rewrite, 1);
106 }
107 
108 static char *
109 readtty(const char pr[], char src[])
110 {
111 	/* XXX - watch for potential setjmp/longjmp clobbering!
112 	 * Currently there appear to be none.
113 	 */
114 	char canonb[LINESIZE];
115 	int c;
116 	char *cp, *cp2;
117 #ifdef TIOCSTI
118 	char ch;
119 	static char empty[] = "";
120 #endif
121 	(void)fputs(pr, stdout);
122 	(void)fflush(stdout);
123 	if (src != NULL && strlen(src) > sizeof(canonb) - 2) {
124 		(void)printf("too long to edit\n");
125 		return src;
126 	}
127 #ifndef TIOCSTI
128 	if (src != NULL)
129 		cp = copy(src, canonb);
130 	else
131 		cp = copy("", canonb);
132 	(void)fputs(canonb, stdout);
133 	(void)fflush(stdout);
134 #else
135 	cp = src == NULL ? empty : src;
136 	while ((c = *cp++) != '\0') {
137 		if ((c_erase != _POSIX_VDISABLE && c == c_erase) ||
138 		    (c_kill != _POSIX_VDISABLE && c == c_kill)) {
139 			ch = '\\';
140 			(void)ioctl(0, TIOCSTI, &ch);
141 		}
142 		ch = c;
143 		(void)ioctl(0, TIOCSTI, &ch);
144 	}
145 	cp = canonb;
146 	*cp = 0;
147 #endif
148 	cp2 = cp;
149 	while (cp2 < canonb + sizeof(canonb))
150 		*cp2++ = 0;
151 	cp2 = cp;
152 	if (setjmp(rewrite))
153 		goto redo;
154 	(void)signal(SIGTSTP, ttystop);
155 	(void)signal(SIGTTOU, ttystop);
156 	(void)signal(SIGTTIN, ttystop);
157 	clearerr(stdin);
158 	while (cp2 < canonb + sizeof(canonb)) {
159 		c = getc(stdin);
160 		if (c == EOF || c == '\n')
161 			break;
162 		*cp2++ = c;
163 	}
164 	*cp2 = 0;
165 	(void)signal(SIGTSTP, SIG_DFL);
166 	(void)signal(SIGTTOU, SIG_DFL);
167 	(void)signal(SIGTTIN, SIG_DFL);
168 	if (c == EOF && ferror(stdin)) {
169 redo:
170 		cp = strlen(canonb) > 0 ? canonb : NULL;
171 		clearerr(stdin);
172 		return readtty(pr, cp);
173 	}
174 #ifndef TIOCSTI
175 	if (cp == NULL || *cp == '\0')
176 		return src;
177 	cp2 = cp;
178 	if (!ttyset)
179 		return strlen(canonb) > 0 ? savestr(canonb) : NULL;
180 	while (*cp != '\0') {
181 		c = *cp++;
182 		if (c_erase != _POSIX_VDISABLE && c == c_erase) {
183 			if (cp2 == canonb)
184 				continue;
185 			if (cp2[-1] == '\\') {
186 				cp2[-1] = c;
187 				continue;
188 			}
189 			cp2--;
190 			continue;
191 		}
192 		if (c_kill != _POSIX_VDISABLE && c == c_kill) {
193 			if (cp2 == canonb)
194 				continue;
195 			if (cp2[-1] == '\\') {
196 				cp2[-1] = c;
197 				continue;
198 			}
199 			cp2 = canonb;
200 			continue;
201 		}
202 		*cp2++ = c;
203 	}
204 	*cp2 = '\0';
205 #endif
206 	if (equal("", canonb))
207 		return NULL;
208 	return savestr(canonb);
209 }
210 #endif /* USE_EDITLINE */
211 
212 
213 /*ARGSUSED*/
214 static void
215 ttyint(int s __unused)
216 {
217 	longjmp(intjmp, 1);
218 }
219 
220 /*
221  * Read all relevant header fields.
222  */
223 PUBLIC int
224 grabh(struct header *hp, int gflags)
225 {
226 	struct termios ttybuf;
227 
228 	/* The following are declared volatile to avoid longjmp
229 	 * clobbering, though they seem safe without it! */
230 	sig_t volatile saveint;
231 	sig_t volatile savetstp;
232 	sig_t volatile savettou;
233 	sig_t volatile savettin;
234 #ifndef TIOCSTI
235 	sig_t volatile savequit;
236 #else
237 # ifdef TIOCEXT
238 	int volatile extproc;
239 # endif /* TIOCEXT */
240 #endif /* TIOCSTI */
241 	int retval;
242 
243 	savetstp = signal(SIGTSTP, SIG_DFL);
244 	savettou = signal(SIGTTOU, SIG_DFL);
245 	savettin = signal(SIGTTIN, SIG_DFL);
246 #ifndef TIOCSTI
247 	ttyset = 0;
248 #endif
249 	if (tcgetattr(fileno(stdin), &ttybuf) < 0) {
250 		warn("tcgetattr");
251 		return -1;
252 	}
253 #if !defined(USE_EDITLINE) || !defined(TIOCSTI)
254 	c_erase = ttybuf.c_cc[VERASE];
255 	c_kill = ttybuf.c_cc[VKILL];
256 #endif
257 #ifndef TIOCSTI
258 	ttybuf.c_cc[VERASE] = _POSIX_VDISABLE;
259 	ttybuf.c_cc[VKILL] = _POSIX_VDISABLE;
260 	if ((saveint = signal(SIGINT, SIG_IGN)) == SIG_DFL)
261 		(void)signal(SIGINT, SIG_DFL);
262 	if ((savequit = signal(SIGQUIT, SIG_IGN)) == SIG_DFL)
263 		(void)signal(SIGQUIT, SIG_DFL);
264 #else
265 # ifdef		TIOCEXT
266 	extproc = ((ttybuf.c_lflag & EXTPROC) ? 1 : 0);
267 	if (extproc) {
268 		int flag;
269 		flag = 0;
270 		if (ioctl(fileno(stdin), TIOCEXT, &flag) < 0)
271 			warn("TIOCEXT: off");
272 	}
273 # endif	/* TIOCEXT */
274 	saveint = signal(SIGINT, ttyint); /* must precede setjmp to be saved */
275 	if ((retval = setjmp(intjmp)) != 0) {
276 		(void)fputc('\n', stdout);
277 		goto out;
278 	}
279 #endif
280 	if (gflags & GTO) {
281 #ifndef TIOCSTI
282 		if (!ttyset && hp->h_to != NULL)
283 			ttyset++, tcsetattr(fileno(stdin), TCSADRAIN, &ttybuf);
284 #endif
285 		hp->h_to =
286 			extract(readtty("To: ", detract(hp->h_to, 0)), GTO);
287 	}
288 	if (gflags & GSUBJECT) {
289 #ifndef TIOCSTI
290 		if (!ttyset && hp->h_subject != NULL)
291 			ttyset++, tcsetattr(fileno(stdin), TCSADRAIN, &ttybuf);
292 #endif
293 		hp->h_subject = readtty("Subject: ", hp->h_subject);
294 	}
295 	if (gflags & GCC) {
296 #ifndef TIOCSTI
297 		if (!ttyset && hp->h_cc != NULL)
298 			ttyset++, tcsetattr(fileno(stdin), TCSADRAIN, &ttybuf);
299 #endif
300 		hp->h_cc =
301 			extract(readtty("Cc: ", detract(hp->h_cc, 0)), GCC);
302 	}
303 	if (gflags & GBCC) {
304 #ifndef TIOCSTI
305 		if (!ttyset && hp->h_bcc != NULL)
306 			ttyset++, tcsetattr(fileno(stdin), TCSADRAIN, &ttybuf);
307 #endif
308 		hp->h_bcc =
309 			extract(readtty("Bcc: ", detract(hp->h_bcc, 0)), GBCC);
310 	}
311 	if (gflags & GSMOPTS) {
312 		char *smopts;
313 #ifndef TIOCSTI
314 		if (!ttyset && hp->h_smopts != NULL)
315 			ttyset++, tcsetattr(fileno(stdin), TCSADRAIN, &ttybuf);
316 #endif
317 		smopts = readtty("Smopts: ", detract(hp->h_smopts, GSMOPTS));
318 
319 		/* Parse smopts with getrawlist() rather than expand()
320 		 * to get a shell-like expansion.
321 		 */
322 		hp->h_smopts = NULL;
323 		if (smopts) {
324 			struct name *np, *t;
325 			char *argv[MAXARGC];
326 			int argc, i;
327 
328 			np = NULL;
329 			argc = getrawlist(smopts, argv, sizeof(argv)/sizeof(*argv));
330 			for (i = 0; i < argc; i++) {
331 				t = nalloc(argv[i], GSMOPTS);
332 				if (hp->h_smopts == NULL)
333 			hp->h_smopts = t;
334 				else
335 			np->n_flink = t;
336 				t->n_blink = np;
337 				np = t;
338 			}
339 		}
340 #ifdef MIME_SUPPORT
341 		if (hp->h_attach) {
342 			struct attachment *ap;
343 			int i;
344 			i = 0;
345 			for (ap = hp->h_attach; ap; ap = ap->a_flink)
346 				i++;
347 			(void)printf("Attachment%s: %d\n", i > 1 ? "s" : "", i);
348 		}
349 #endif
350 	}
351 #ifdef TIOCSTI
352 out:
353 #endif
354 	(void)signal(SIGTSTP, savetstp);
355 	(void)signal(SIGTTOU, savettou);
356 	(void)signal(SIGTTIN, savettin);
357 #ifndef TIOCSTI
358 	ttybuf.c_cc[VERASE] = c_erase;
359 	ttybuf.c_cc[VKILL] = c_kill;
360 	if (ttyset)
361 		tcsetattr(fileno(stdin), TCSADRAIN, &ttybuf);
362 	(void)signal(SIGQUIT, savequit);
363 #else
364 # ifdef		TIOCEXT
365 	if (extproc) {
366 		int flag;
367 		flag = 1;
368 		if (ioctl(fileno(stdin), TIOCEXT, &flag) < 0)
369 			warn("TIOCEXT: on");
370 	}
371 # endif	/* TIOCEXT */
372 #endif
373 	(void)signal(SIGINT, saveint);
374 	return retval;
375 }
376