xref: /netbsd-src/lib/libutil/ttymsg.c (revision 5e4c038a45edbc7d63b7c2daa76e29f88b64a4e3)
1 /*	$NetBSD: ttymsg.c,v 1.15 2000/07/05 11:46:42 ad Exp $	*/
2 
3 /*
4  * Copyright (c) 1989, 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. All advertising materials mentioning features or use of this software
16  *    must display the following acknowledgement:
17  *	This product includes software developed by the University of
18  *	California, Berkeley and its contributors.
19  * 4. Neither the name of the University nor the names of its contributors
20  *    may be used to endorse or promote products derived from this software
21  *    without specific prior written permission.
22  *
23  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
24  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
25  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
26  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
27  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
28  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
29  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
30  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
31  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
32  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
33  * SUCH DAMAGE.
34  */
35 
36 #include <sys/cdefs.h>
37 #if defined(LIBC_SCCS) && !defined(lint)
38 #if 0
39 static char sccsid[] = "@(#)ttymsg.c	8.2 (Berkeley) 11/16/93";
40 #else
41 __RCSID("$NetBSD: ttymsg.c,v 1.15 2000/07/05 11:46:42 ad Exp $");
42 #endif
43 #endif /* LIBC_SCCS and not lint */
44 
45 #include <sys/types.h>
46 #include <sys/uio.h>
47 
48 #include <assert.h>
49 #include <dirent.h>
50 #include <errno.h>
51 #include <fcntl.h>
52 #include <paths.h>
53 #include <signal.h>
54 #include <stdio.h>
55 #include <stdlib.h>
56 #include <string.h>
57 #include <unistd.h>
58 #include <util.h>
59 
60 /*
61  * Display the contents of a uio structure on a terminal.  Used by wall(1),
62  * syslogd(8), and talkd(8).  Forks and finishes in child if write would block,
63  * waiting up to tmout seconds.  Returns pointer to error string on unexpected
64  * error; string is not newline-terminated.  Various "normal" errors are
65  * ignored (exclusive-use, lack of permission, etc.).
66  */
67 char *
68 ttymsg(struct iovec *iov, int iovcnt, const char *line, int tmout)
69 {
70 	static char device[MAXNAMLEN] = _PATH_DEV;
71 	static char errbuf[1024];
72 	int cnt, fd, left, wret;
73 	struct iovec localiov[6];
74 	sigset_t nset;
75 	int forked = 0;
76 
77 	_DIAGASSERT(iov != NULL);
78 	_DIAGASSERT(iovcnt >= 0);
79 	_DIAGASSERT(line != NULL);
80 
81 	if (iovcnt > sizeof(localiov) / sizeof(localiov[0]))
82 		return ("too many iov's (change code in libutil/ttymsg.c)");
83 
84 	(void)strncpy(device + sizeof(_PATH_DEV) - 1, line,
85 	    sizeof(device) - sizeof(_PATH_DEV));
86 	if (strchr(device + sizeof(_PATH_DEV) - 1, '/')) {
87 		/* A slash is an attempt to break security... */
88 		(void) snprintf(errbuf, sizeof(errbuf), "'/' in \"%s\"",
89 		    device);
90 		return (errbuf);
91 	}
92 
93 	/*
94 	 * open will fail on slip lines or exclusive-use lines
95 	 * if not running as root; not an error.
96 	 */
97 	if ((fd = open(device, O_WRONLY|O_NONBLOCK, 0)) < 0) {
98 		if (errno == EBUSY || errno == EACCES)
99 			return (NULL);
100 		(void) snprintf(errbuf, sizeof(errbuf),
101 		    "%s: %s", device, strerror(errno));
102 		return (errbuf);
103 	}
104 
105 	for (cnt = left = 0; cnt < iovcnt; ++cnt)
106 		left += iov[cnt].iov_len;
107 
108 	for (;;) {
109 		wret = writev(fd, iov, iovcnt);
110 		if (wret >= left)
111 			break;
112 		if (wret >= 0) {
113 			left -= wret;
114 			if (iov != localiov) {
115 				memcpy(localiov, iov,
116 				    iovcnt * sizeof(struct iovec));
117 				iov = localiov;
118 			}
119 			for (cnt = 0; wret >= iov->iov_len; ++cnt) {
120 				wret -= iov->iov_len;
121 				++iov;
122 				--iovcnt;
123 			}
124 			if (wret) {
125 				iov->iov_base =
126 				    (char *)iov->iov_base + wret;
127 				iov->iov_len -= wret;
128 			}
129 			continue;
130 		}
131 		if (errno == EWOULDBLOCK) {
132 			int cpid, off = 0;
133 
134 			if (forked) {
135 				(void) close(fd);
136 				_exit(1);
137 			}
138 			cpid = fork();
139 			if (cpid < 0) {
140 				(void) snprintf(errbuf, sizeof(errbuf),
141 				    "fork: %s", strerror(errno));
142 				(void) close(fd);
143 				return (errbuf);
144 			}
145 			if (cpid) {	/* parent */
146 				(void) close(fd);
147 				return (NULL);
148 			}
149 			forked++;
150 			/* wait at most tmout seconds */
151 			(void) signal(SIGALRM, SIG_DFL);
152 			(void) signal(SIGTERM, SIG_DFL); /* XXX */
153 			sigfillset(&nset);
154 			(void) sigprocmask(SIG_UNBLOCK, &nset, NULL);
155 			(void) alarm((u_int)tmout);
156 			(void) fcntl(fd, O_NONBLOCK, &off);
157 			continue;
158 		}
159 		/*
160 		 * We get ENODEV on a slip line if we're running as root,
161 		 * and EIO if the line just went away.
162 		 */
163 		if (errno == ENODEV || errno == EIO)
164 			break;
165 		(void) close(fd);
166 		if (forked)
167 			_exit(1);
168 		(void) snprintf(errbuf, sizeof(errbuf),
169 		    "%s: %s", device, strerror(errno));
170 		return (errbuf);
171 	}
172 
173 	(void) close(fd);
174 	if (forked)
175 		_exit(0);
176 	return (NULL);
177 }
178