2 * Copyright (c) 1995 Ugen J.S.Antsilevich
4 * Redistribution and use in source forms, with and without modification,
5 * are permitted provided that this entire comment appears intact.
7 * Redistribution in binary form may occur without any restrictions.
8 * Obviously, it would be nice if you gave credit where credit is due
9 * but requiring it would be too onerous.
11 * This software is provided ``AS IS'' without any warranties of any kind.
16 #include <sys/cdefs.h>
17 __FBSDID("$FreeBSD$");
19 #include <sys/param.h>
20 #include <sys/fcntl.h>
21 #include <sys/filio.h>
22 #include <sys/snoop.h>
24 #include <sys/linker.h>
25 #include <sys/module.h>
41 #define MSG_INIT "Snoop started."
42 #define MSG_OFLOW "Snoop stopped due to overflow. Reconnecting."
43 #define MSG_CLOSED "Snoop stopped due to tty close. Reconnecting."
44 #define MSG_CHANGE "Snoop device change by user request."
45 #define MSG_NOWRITE "Snoop device change due to write failure."
47 #define DEV_NAME_LEN 1024 /* for /dev/ttyXX++ */
50 #define CHR_SWITCH 24 /* Ctrl+X */
51 #define CHR_CLEAR 23 /* Ctrl+V */
53 static void clear(void);
54 static void timestamp(const char *);
55 static void set_tty(void);
56 static void unset_tty(void);
57 static void fatal(int, const char *);
58 static int open_snp(void);
59 static void cleanup(int);
60 static void usage(void) __dead2;
61 static void setup_scr(void);
62 static void attach_snp(void);
63 static void detach_snp(void);
64 static void set_dev(const char *);
65 static void ask_dev(char *, const char *);
67 int opt_reconn_close = 0;
68 int opt_reconn_oflow = 0;
69 int opt_interactive = 1;
70 int opt_timestamp = 0;
72 int opt_no_switch = 0;
73 const char *opt_snpdev;
75 char dev_name[DEV_NAME_LEN];
77 int std_in = 0, std_out = 1;
81 char tbuf[1024], gbuf[1024];
88 tputs(gbuf, 1, putchar);
93 timestamp(const char *buf)
99 printf("\n---------------------------------------------\n");
101 strftime(btmp, 1024, "Time: %d %b %H:%M", localtime(&t));
102 printf("%s\n", btmp);
104 printf("---------------------------------------------\n");
114 ntty.c_lflag &= ~ICANON; /* disable canonical operation */
115 ntty.c_lflag &= ~ECHO;
117 ntty.c_lflag &= ~FLUSHO;
120 ntty.c_lflag &= ~PENDIN;
123 ntty.c_lflag &= ~IEXTEN;
125 ntty.c_cc[VMIN] = 1; /* minimum of one character */
126 ntty.c_cc[VTIME] = 0; /* timeout value */
128 ntty.c_cc[VINTR] = 07; /* ^G */
129 ntty.c_cc[VQUIT] = 07; /* ^G */
130 tcsetattr(std_in, TCSANOW, &ntty);
137 tcsetattr(std_in, TCSANOW, &otty);
141 fatal(int error, const char *buf)
146 errx(error, "fatal: %s", buf);
161 if (opt_snpdev == NULL)
162 f = open(_PATH_DEV "snp", mode);
164 f = open(opt_snpdev, mode);
166 fatal(EX_OSFILE, "cannot open snoop device");
172 cleanup(int signo __unused)
176 timestamp("Logging Exited.");
186 fprintf(stderr, "usage: watch [-ciotnW] [tty name]\n");
193 char *cbuf = gbuf, *term;
195 if (!opt_interactive)
197 if ((term = getenv("TERM")))
198 if (tgetent(tbuf, term) == 1)
199 if (tgetstr("cl", &cbuf))
211 ioctl(snp_io, SNPSTTY, &fd);
219 snp_tty = open(dev_name, O_RDONLY | O_NONBLOCK);
221 fatal(EX_DATAERR, "can't open device");
222 if (ioctl(snp_io, SNPSTTY, &snp_tty) != 0)
223 fatal(EX_UNAVAILABLE, "cannot attach to tty");
226 timestamp("Logging Started.");
230 set_dev(const char *name)
232 char buf[DEV_NAME_LEN];
235 if (strlen(name) > 5 && !strncmp(name, _PATH_DEV, sizeof _PATH_DEV - 1)) {
236 snprintf(buf, sizeof buf, "%s", name);
238 if (strlen(name) == 2)
239 sprintf(buf, "%s%s", _PATH_TTY, name);
241 sprintf(buf, "%s%s", _PATH_DEV, name);
244 if (*name == '\0' || stat(buf, &sb) < 0)
245 fatal(EX_DATAERR, "bad device name");
247 if ((sb.st_mode & S_IFMT) != S_IFCHR)
248 fatal(EX_DATAERR, "must be a character device");
250 strlcpy(dev_name, buf, sizeof(dev_name));
256 ask_dev(char *dbuf, const char *msg)
258 char buf[DEV_NAME_LEN];
267 printf("Enter device name [%s]:", dbuf);
269 printf("Enter device name:");
271 if (fgets(buf, DEV_NAME_LEN - 1, stdin)) {
273 if (buf[len - 1] == '\n')
275 if (buf[0] != '\0' && buf[0] != ' ')
284 main(int ac, char *av[])
286 int ch, res, rv, nread;
287 size_t b_size = MIN_SIZE;
288 char *buf, chb[READB_LEN];
291 (void) setlocale(LC_TIME, "");
298 while ((ch = getopt(ac, av, "Wciotnf:")) != -1)
304 opt_reconn_close = 1;
310 opt_reconn_oflow = 1;
326 tcgetattr(std_in, &otty);
328 if (modfind("snp") == -1)
329 if (kldload("snp") == -1 || modfind("snp") == -1)
330 warn("snp module not available");
332 signal(SIGINT, cleanup);
337 if (*(av += optind) == NULL) {
338 if (opt_interactive && !opt_no_switch)
339 ask_dev(dev_name, MSG_INIT);
341 fatal(EX_DATAERR, "no device name given");
343 strlcpy(dev_name, *av, sizeof(dev_name));
347 if (!(buf = (char *) malloc(b_size)))
348 fatal(EX_UNAVAILABLE, "malloc failed");
354 FD_SET(std_in, &fd_s);
355 FD_SET(snp_io, &fd_s);
356 res = select(snp_io + 1, &fd_s, NULL, NULL, NULL);
357 if (opt_interactive && FD_ISSET(std_in, &fd_s)) {
359 if ((res = ioctl(std_in, FIONREAD, &nread)) != 0)
360 fatal(EX_OSERR, "ioctl(FIONREAD)");
361 if (nread > READB_LEN)
363 rv = read(std_in, chb, nread);
364 if (rv == -1 || rv != nread)
365 fatal(EX_IOERR, "read (stdin) failed");
372 if (!opt_no_switch) {
374 ask_dev(dev_name, MSG_CHANGE);
380 rv = write(snp_io, chb, nread);
381 if (rv == -1 || rv != nread) {
386 ask_dev(dev_name, MSG_NOWRITE);
393 if (!FD_ISSET(snp_io, &fd_s))
396 if ((res = ioctl(snp_io, FIONREAD, &nread)) != 0)
397 fatal(EX_OSERR, "ioctl(FIONREAD)");
401 if (opt_reconn_oflow)
403 else if (opt_interactive && !opt_no_switch) {
404 ask_dev(dev_name, MSG_OFLOW);
411 if (opt_reconn_close)
413 else if (opt_interactive && !opt_no_switch) {
414 ask_dev(dev_name, MSG_CLOSED);
420 if (nread < (b_size / 2) && (b_size / 2) > MIN_SIZE) {
422 if (!(buf = (char *) malloc(b_size / 2)))
423 fatal(EX_UNAVAILABLE, "malloc failed");
426 if (nread > b_size) {
427 b_size = (nread % 2) ? (nread + 1) : (nread);
429 if (!(buf = (char *) malloc(b_size)))
430 fatal(EX_UNAVAILABLE, "malloc failed");
432 rv = read(snp_io, buf, nread);
433 if (rv == -1 || rv != nread)
434 fatal(EX_IOERR, "read failed");
435 rv = write(std_out, buf, nread);
436 if (rv == -1 || rv != nread)
437 fatal(EX_IOERR, "write failed");