Annotation of src/usr.bin/cu/cu.c, Revision 1.5
1.5 ! nicm 1: /* $OpenBSD: cu.c,v 1.4 2012/07/10 10:28:05 nicm Exp $ */
1.1 nicm 2:
3: /*
4: * Copyright (c) 2012 Nicholas Marriott <nicm@openbsd.org>
5: *
6: * Permission to use, copy, modify, and distribute this software for any
7: * purpose with or without fee is hereby granted, provided that the above
8: * copyright notice and this permission notice appear in all copies.
9: *
10: * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
11: * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
12: * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
13: * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
14: * WHATSOEVER RESULTING FROM LOSS OF MIND, USE, DATA OR PROFITS, WHETHER
15: * IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING
16: * OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
17: */
18:
19: #include <sys/param.h>
20: #include <sys/ioctl.h>
21:
22: #include <err.h>
23: #include <event.h>
24: #include <fcntl.h>
25: #include <getopt.h>
26: #include <paths.h>
27: #include <pwd.h>
28: #include <signal.h>
29: #include <stdio.h>
30: #include <stdlib.h>
31: #include <string.h>
32: #include <termios.h>
33: #include <unistd.h>
34:
35: #include "cu.h"
36:
37: extern char *__progname;
38:
39: struct termios saved_tio;
40: struct bufferevent *input_ev;
41: struct bufferevent *output_ev;
42: int line_fd;
43: struct bufferevent *line_ev;
44: struct event sigterm_ev;
45: enum {
46: STATE_NONE,
47: STATE_NEWLINE,
48: STATE_TILDE
49: } last_state = STATE_NEWLINE;
50:
51: __dead void usage(void);
52: void signal_event(int, short, void *);
53: void stream_read(struct bufferevent *, void *);
54: void stream_error(struct bufferevent *, short, void *);
55: void line_read(struct bufferevent *, void *);
56: void line_error(struct bufferevent *, short, void *);
57:
58: __dead void
59: usage(void)
60: {
61: fprintf(stderr, "usage: %s [-l line] [-s speed]\n", __progname);
62: exit(1);
63: }
64:
65: int
66: main(int argc, char **argv)
67: {
68: const char *line, *errstr;
69: char *tmp;
70: int opt, speed, i, ch;
71: static char sbuf[12];
72:
73: line = "/dev/cua00";
74: speed = 9600;
75:
76: /*
77: * Convert obsolescent -### speed to modern -s### syntax which getopt()
78: * can handle.
79: */
80: for (i = 1; i < argc; i++) {
81: if (argv[i][0] == '-') {
82: switch (argv[i][1]) {
83: case '0': case '1': case '2': case '3': case '4':
84: case '5': case '6': case '7': case '8': case '9':
85: ch = snprintf(sbuf, sizeof(sbuf), "-s%s",
86: &argv[i][1]);
87: if (ch <= 0 || ch >= (int)sizeof(sbuf)) {
88: errx(1, "invalid speed: %s",
89: &argv[i][1]);
90: }
91: argv[i] = sbuf;
92: break;
93: case '-':
94: /* if we get "--" stop processing args */
95: if (argv[i][2] == '\0')
96: goto getopt;
97: break;
98: }
99: }
100: }
101:
102: getopt:
103: while ((opt = getopt(argc, argv, "l:s:")) != -1) {
104: switch (opt) {
105: case 'l':
106: line = optarg;
107: break;
108: case 's':
109: speed = strtonum(optarg, 0, UINT_MAX, &errstr);
110: if (errstr != NULL)
111: errx(1, "speed is %s: %s", errstr, optarg);
112: break;
113: default:
114: usage();
115: }
116: }
117: argc -= optind;
118: argv += optind;
119: if (argc != 0)
120: usage();
121:
122: if (strchr(line, '/') == NULL) {
123: if (asprintf(&tmp, "%s%s", _PATH_DEV, line) == -1)
124: err(1, "asprintf");
125: line = tmp;
126: }
127:
128: line_fd = open(line, O_RDWR);
129: if (line_fd < 0)
130: err(1, "open(\"%s\")", line);
131: if (ioctl(line_fd, TIOCEXCL) != 0)
132: err(1, "ioctl(TIOCEXCL)");
133:
1.2 nicm 134: if (set_line(speed) != 0)
1.4 nicm 135: err(1, "tcsetattr");
1.1 nicm 136:
137: if (isatty(STDIN_FILENO) && tcgetattr(STDIN_FILENO, &saved_tio) != 0)
138: err(1, "tcgetattr");
139:
140: event_init();
1.3 nicm 141:
1.1 nicm 142: signal_set(&sigterm_ev, SIGTERM, signal_event, NULL);
143: signal_add(&sigterm_ev, NULL);
1.3 nicm 144: if (signal(SIGINT, SIG_IGN) == SIG_ERR)
145: err(1, "signal");
146: if (signal(SIGQUIT, SIG_IGN) == SIG_ERR)
147: err(1, "signal");
1.1 nicm 148:
1.4 nicm 149: set_termios(); /* after this use cu_err and friends */
1.1 nicm 150:
151: /* stdin and stdout get separate events */
152: input_ev = bufferevent_new(STDIN_FILENO, stream_read, NULL,
153: stream_error, NULL);
154: bufferevent_enable(input_ev, EV_READ);
155: output_ev = bufferevent_new(STDOUT_FILENO, NULL, NULL, stream_error,
156: NULL);
157: bufferevent_enable(output_ev, EV_WRITE);
158:
159: line_ev = bufferevent_new(line_fd, line_read, NULL, line_error,
160: NULL);
161: bufferevent_enable(line_ev, EV_READ|EV_WRITE);
162:
163: printf("Connected (speed %u)\r\n", speed);
164: event_dispatch();
165:
1.5 ! nicm 166: restore_termios();
1.1 nicm 167: printf("\r\n[EOT]\n");
168:
169: exit(0);
170: }
171:
172: void
173: signal_event(int fd, short events, void *data)
174: {
1.5 ! nicm 175: restore_termios();
1.1 nicm 176: printf("\r\n[SIG%s]\n", sys_signame[fd]);
177:
178: exit(0);
179: }
180:
181: void
182: set_termios(void)
183: {
184: struct termios tio;
185:
186: if (!isatty(STDIN_FILENO))
187: return;
188:
189: memcpy(&tio, &saved_tio, sizeof(tio));
190: tio.c_lflag &= ~(ICANON|IEXTEN|ECHO);
191: tio.c_iflag &= ~(INPCK|ICRNL);
192: tio.c_oflag &= ~OPOST;
193: tio.c_cc[VMIN] = 1;
194: tio.c_cc[VTIME] = 0;
195: tio.c_cc[VDISCARD] = _POSIX_VDISABLE;
196: tio.c_cc[VDSUSP] = _POSIX_VDISABLE;
197: tio.c_cc[VINTR] = _POSIX_VDISABLE;
198: tio.c_cc[VLNEXT] = _POSIX_VDISABLE;
199: tio.c_cc[VQUIT] = _POSIX_VDISABLE;
200: tio.c_cc[VSUSP] = _POSIX_VDISABLE;
201: if (tcsetattr(STDIN_FILENO, TCSAFLUSH, &tio) != 0)
1.4 nicm 202: cu_err(1, "tcsetattr");
1.1 nicm 203: }
204:
205: void
206: restore_termios(void)
207: {
1.4 nicm 208: if (isatty(STDIN_FILENO))
209: tcsetattr(STDIN_FILENO, TCSAFLUSH, &saved_tio);
1.2 nicm 210: }
211:
212: int
213: set_line(int speed)
214: {
215: struct termios tio;
216:
217: cfmakeraw(&tio);
218: tio.c_iflag = 0;
219: tio.c_oflag = 0;
220: tio.c_lflag = 0;
221: tio.c_cflag = CREAD|CS8;
222: tio.c_cc[VMIN] = 1;
223: tio.c_cc[VTIME] = 0;
224: cfsetspeed(&tio, speed);
1.4 nicm 225: if (tcsetattr(line_fd, TCSAFLUSH, &tio) != 0)
1.2 nicm 226: return (-1);
227: return (0);
1.1 nicm 228: }
229:
230: void
231: stream_read(struct bufferevent *bufev, void *data)
232: {
233: char *new_data, *ptr;
234: size_t new_size;
235: int state_change;
236:
237: new_data = EVBUFFER_DATA(input_ev->input);
238: new_size = EVBUFFER_LENGTH(input_ev->input);
239: if (new_size == 0)
240: return;
241:
242: state_change = isatty(STDIN_FILENO);
243: for (ptr = new_data; ptr < new_data + new_size; ptr++) {
244: switch (last_state) {
245: case STATE_NONE:
246: if (state_change && *ptr == '\r')
247: last_state = STATE_NEWLINE;
248: break;
249: case STATE_NEWLINE:
250: if (state_change && *ptr == '~') {
251: last_state = STATE_TILDE;
252: continue;
253: }
254: if (*ptr != '\r')
255: last_state = STATE_NONE;
256: break;
257: case STATE_TILDE:
258: do_command(*ptr);
259: last_state = STATE_NEWLINE;
260: continue;
261: }
262:
263: bufferevent_write(line_ev, ptr, 1);
264: }
265:
266: evbuffer_drain(input_ev->input, new_size);
267: }
268:
269: void
270: stream_error(struct bufferevent *bufev, short what, void *data)
271: {
272: event_loopexit(NULL);
273: }
274:
275: void
276: line_read(struct bufferevent *bufev, void *data)
277: {
278: char *new_data;
279: size_t new_size;
280:
281: new_data = EVBUFFER_DATA(line_ev->input);
282: new_size = EVBUFFER_LENGTH(line_ev->input);
283: if (new_size == 0)
284: return;
285:
286: bufferevent_write(output_ev, new_data, new_size);
287:
288: evbuffer_drain(line_ev->input, new_size);
289: }
290:
291: void
292: line_error(struct bufferevent *bufev, short what, void *data)
293: {
294: event_loopexit(NULL);
295: }
296:
297: /* Expands tildes in the file name. Based on code from ssh/misc.c. */
298: char *
299: tilde_expand(const char *filename1)
300: {
301: const char *filename, *path;
302: char user[128], ret[MAXPATHLEN], *out;
303: struct passwd *pw;
304: u_int len, slash;
305:
306: if (*filename1 != '~')
307: goto no_change;
308: filename = filename1 + 1;
309:
310: path = strchr(filename, '/');
311: if (path != NULL && path > filename) { /* ~user/path */
312: slash = path - filename;
313: if (slash > sizeof(user) - 1)
314: goto no_change;
315: memcpy(user, filename, slash);
316: user[slash] = '\0';
317: if ((pw = getpwnam(user)) == NULL)
318: goto no_change;
319: } else if ((pw = getpwuid(getuid())) == NULL) /* ~/path */
320: goto no_change;
321:
322: if (strlcpy(ret, pw->pw_dir, sizeof(ret)) >= sizeof(ret))
323: goto no_change;
324:
325: /* Make sure directory has a trailing '/' */
326: len = strlen(pw->pw_dir);
327: if ((len == 0 || pw->pw_dir[len - 1] != '/') &&
328: strlcat(ret, "/", sizeof(ret)) >= sizeof(ret))
329: goto no_change;
330:
331: /* Skip leading '/' from specified path */
332: if (path != NULL)
333: filename = path + 1;
334: if (strlcat(ret, filename, sizeof(ret)) >= sizeof(ret))
335: goto no_change;
336:
337: out = strdup(ret);
338: if (out == NULL)
1.4 nicm 339: cu_err(1, "strdup");
1.1 nicm 340: return (out);
341:
342: no_change:
343: out = strdup(filename1);
344: if (out == NULL)
1.4 nicm 345: cu_err(1, "strdup");
1.1 nicm 346: return (out);
347: }