X-Git-Url: http://git.madism.org/?a=blobdiff_plain;f=madtty%2Fmadtty.c;h=8fbc4695cd3e533868807926db7f59535c10f090;hb=d0e48d929c19d39c4a13e57e0f8a8dd6194d6dde;hp=2002e18e5d4d07ff205a3b677875e751f4c317e0;hpb=5618d7a42ce62103e5f29da3756b81f57198b22f;p=apps%2Fmadtty.git diff --git a/madtty/madtty.c b/madtty/madtty.c index 2002e18..8fbc469 100644 --- a/madtty/madtty.c +++ b/madtty/madtty.c @@ -19,12 +19,12 @@ Copyright © 2006 Pierre Habouzit */ +#include #include -#ifdef USE_PTY -#include -#endif +#include #include #include +#include #include "madtty.h" #include "roteprivate.h" @@ -73,7 +73,7 @@ RoteTerm *rote_vt_create(int rows, int cols) /* allocate private data */ rt->pd = (RoteTermPrivate*)calloc(sizeof(RoteTermPrivate), 1); - rt->pd->pty = -1; /* no pty for now */ + rt->pty = -1; /* no pty for now */ /* initial scrolling area is the whole window */ rt->pd->scrolltop = 0; @@ -134,58 +134,112 @@ void rote_vt_draw(RoteTerm *rt, WINDOW *win, int srow, int scol, #endif -#ifdef USE_PTY +/******************************************************/ + +#define PTYCHAR1 "pqrstuvwxyz" +#define PTYCHAR2 "0123456789abcdef" + +/* allocate one pty/tty pair */ +static int get_pty(char *tty_str) +{ + int fd; + char ptydev[] = "/dev/pty??"; + char ttydev[] = "/dev/tty??"; + int len = strlen(ttydev); + const char *c1, *c2; + + for (c1 = PTYCHAR1; *c1; c1++) { + ptydev[len-2] = ttydev[len-2] = *c1; + for (c2 = PTYCHAR2; *c2; c2++) { + ptydev[len-1] = ttydev[len-1] = *c2; + if ((fd = open(ptydev, O_RDWR)) >= 0) { + if (access(ttydev, R_OK|W_OK) == 0) { + strcpy(tty_str, ttydev); + return fd; + } + close(fd); + } + } + } + return -1; +} -pid_t rote_vt_forkpty(RoteTerm *rt, const char *command) +static int +run_process(const char *path, const char **argv, int *fd_ptr, int *pid_ptr) { + int pty_fd, pid, i, nb_fds; + char tty_name[32]; struct winsize ws; - pid_t childpid; - ws.ws_row = rt->rows; - ws.ws_col = rt->cols; - ws.ws_xpixel = ws.ws_ypixel = 0; + pty_fd = get_pty(tty_name); + if (pty_fd < 0) + return -1; - childpid = forkpty(&rt->pd->pty, NULL, NULL, &ws); - if (childpid < 0) + fcntl(pty_fd, F_SETFL, O_NONBLOCK); + + /* set dummy screen size */ + ws.ws_col = 80; + ws.ws_row = 25; + ws.ws_xpixel = ws.ws_col; + ws.ws_ypixel = ws.ws_row; + ioctl(pty_fd, TIOCSWINSZ, &ws); + + pid = fork(); + if (pid < 0) return -1; - if (childpid == 0) { - /* we are the child, running under the slave side of the pty. */ + if (pid == 0) { + /* child process */ + nb_fds = getdtablesize(); + for (i = 0; i < nb_fds; i++) + close(i); + /* open pseudo tty for standard i/o */ + open(tty_name, O_RDWR); + dup(0); + dup(0); + + setsid(); - /* Cajole application into using linux-console-compatible escape - * sequences (which is what we are prepared to interpret) */ setenv("TERM", "linux", 1); + execv(path, (char *const*)argv); + fprintf(stderr, "\nexecv() failed.\nCommand: '%s'\n", argv[0]); + exit(1); + } + /* return file info */ + *fd_ptr = pty_fd; + *pid_ptr = pid; + return 0; +} - /* Now we will exec /bin/sh -c command. */ - execl("/bin/sh", "/bin/sh", "-c", command, NULL); +pid_t rote_vt_forkpty(RoteTerm *rt, const char *path, const char *argv[]) +{ + struct winsize ws; - fprintf(stderr, "\nexecl() failed.\nCommand: '%s'\n", command); - exit(127); /* error exec'ing */ + ws.ws_row = rt->rows; + ws.ws_col = rt->cols; + ws.ws_xpixel = ws.ws_ypixel = 0; + + if (run_process(path, argv, &rt->pty, &rt->childpid)) { + return -1; } - /* if we got here we are the parent process */ - return (rt->childpid = childpid); + ioctl(rt->pty, TIOCSWINSZ, &ws); + return rt->childpid; } void rote_vt_forsake_child(RoteTerm *rt) { - if (rt->pd->pty >= 0) - close(rt->pd->pty); - rt->pd->pty = -1; + if (rt->pty >= 0) + close(rt->pty); + rt->pty = -1; rt->childpid = 0; } -#endif - void rote_vt_update(RoteTerm *rt) { - fd_set ifs; - struct timeval tvzero; char buf[512]; - int bytesread; + int nbread; int n = ROTE_VT_UPDATE_ITERATIONS; - if (rt->pd->pty < 0) - return; /* nothing to pump */ while (n--) { /* iterate at most ROVE_VT_UPDATE_ITERATIONS times. * As Phil Endecott pointed out, if we don't restrict this, @@ -195,26 +249,28 @@ void rote_vt_update(RoteTerm *rt) * calling rote_vt_update often, as the documentation * recommends :-) */ - /* check if pty has something to say */ - FD_ZERO(&ifs); FD_SET(rt->pd->pty, &ifs); - tvzero.tv_sec = 0; tvzero.tv_usec = 0; - - if (select(rt->pd->pty + 1, &ifs, NULL, NULL, &tvzero) <= 0) - return; /* nothing to read, or select() failed */ - - /* read what we can. This is guaranteed not to block, since - * select() told us there was something to read. */ - bytesread = read(rt->pd->pty, buf, 512); - if (bytesread <= 0) return; + nbread = rote_vt_read(rt, buf, sizeof(buf)); + if (nbread <= 0) + return; /* inject the data into the terminal */ - rote_vt_inject(rt, buf, bytesread); + rote_vt_inject(rt, buf, nbread); } } +int rote_vt_read(RoteTerm *rt, char *buf, int buflen) +{ + if (rt->pty < 0) { + errno = EINVAL; + return -1; + } + + return read(rt->pty, buf, buflen); +} + void rote_vt_write(RoteTerm *rt, const char *data, int len) { - if (rt->pd->pty < 0) { + if (rt->pty < 0) { /* no pty, so just inject the data plain and simple */ rote_vt_inject(rt, data, len); return; @@ -223,7 +279,7 @@ void rote_vt_write(RoteTerm *rt, const char *data, int len) /* write data to pty. Keep calling write() until we have written * everything. */ while (len > 0) { - int byteswritten = write(rt->pd->pty, data, len); + int byteswritten = write(rt->pty, data, len); if (byteswritten < 0) { /* very ugly way to inform the error. Improvements welcome! */ static char errormsg[] = "\n(ROTE: pty write() error)\n"; @@ -261,11 +317,6 @@ void rote_vt_restore_snapshot(RoteTerm *rt, void *snapbuf) } } -int rote_vt_get_pty_fd(RoteTerm *rt) -{ - return rt->pd->pty; -} - static const char *keytable[KEY_MAX+1]; static void keytable_init() @@ -301,7 +352,7 @@ void rote_vt_keypress(RoteTerm *rt, int keycode) { char c = (char) keycode; - if (keytable['\n'] != NULL) + if (keytable['\n'] == NULL) keytable_init(); if (keycode >= 0 && keycode < KEY_MAX && keytable[keycode]) {