/*
* sscr_insert --
* Take a line from the shell and insert it into the file.
*/
static int
sscr_insert(SCR *sp)
{
CHAR_T *endp, *p, *t;
SCRIPT *sc;
struct pollfd pfd[1];
recno_t lno;
size_t blen, len, tlen;
u_int value;
int nr, rval;
char *bp;
/* Find out where the end of the file is. */
if (db_last(sp, &lno))
return (1);
#define MINREAD 1024
GET_SPACE_RET(sp, bp, blen, MINREAD);
endp = bp;
/* Read the characters. */
rval = 1;
sc = sp->script;
more: switch (nr = read(sc->sh_master, endp, MINREAD)) {
case 0: /* EOF; shell just exited. */
sscr_end(sp);
rval = 0;
goto ret;
case -1: /* Error or interrupt. */
msgq(sp, M_SYSERR, "shell");
goto ret;
default:
endp += nr;
break;
}
/* Append the lines into the file. */
for (p = t = bp; p < endp; ++p) {
value = KEY_VAL(sp, *p);
if (value == K_CR || value == K_NL) {
len = p - t;
if (db_append(sp, 1, lno++, t, len))
goto ret;
t = p + 1;
}
}
if (p > t) {
len = p - t;
/*
* If the last thing from the shell isn't another prompt, wait
* up to 1/10 of a second for more stuff to show up, so that
* we don't break the output into two separate lines. Don't
* want to hang indefinitely because some program is hanging,
* confused the shell, or whatever.
*/
if (!sscr_matchprompt(sp, t, len, &tlen) || tlen != 0) {
pfd[0].fd = sc->sh_master;
pfd[0].events = POLLIN;
if (poll(pfd, 1, 100) > 0) {
memmove(bp, t, len);
endp = bp + len;
goto more;
}
}
if (sscr_setprompt(sp, t, len))
return (1);
if (db_append(sp, 1, lno++, t, len))
goto ret;
}
/* The cursor moves to EOF. */
sp->lno = lno;
sp->cno = len ? len - 1 : 0;
rval = vs_refresh(sp, 1);
ret: FREE_SPACE(sp, bp, blen);
return (rval);
}
/*
* sscr_insert --
* Take a line from the shell and insert it into the file.
*/
static int
sscr_insert(SCR *sp)
{
struct timeval tv;
char *endp, *p, *t;
SCRIPT *sc;
fd_set rdfd;
db_recno_t lno;
size_t len;
ssize_t nr;
char bp[1024];
const CHAR_T *ip;
size_t ilen = 0;
/* Find out where the end of the file is. */
if (db_last(sp, &lno))
return (1);
endp = bp;
/* Read the characters. */
sc = sp->script;
more: switch (nr = read(sc->sh_master, endp, bp + sizeof(bp) - endp)) {
case 0: /* EOF; shell just exited. */
sscr_end(sp);
return (0);
case -1: /* Error or interrupt. */
msgq(sp, M_SYSERR, "shell");
return (1);
default:
endp += nr;
break;
}
/* Append the lines into the file. */
for (p = t = bp; p < endp; ++p) {
if (*p == '\r' || *p == '\n') {
len = p - t;
if (CHAR2INT(sp, t, len, ip, ilen) ||
db_append(sp, 1, lno++, ip, ilen))
return (1);
t = p + 1;
}
}
/*
* If the last thing from the shell isn't another prompt, wait up to
* 1/10 of a second for more stuff to show up, so that we don't break
* the output into two separate lines. Don't want to hang indefinitely
* because some program is hanging, confused the shell, or whatever.
* Note that sc->sh_prompt can be NULL here.
*/
len = p - t;
if (sc->sh_prompt == NULL || len != sc->sh_prompt_len ||
memcmp(t, sc->sh_prompt, len) != 0) {
tv.tv_sec = 0;
tv.tv_usec = 100000;
FD_ZERO(&rdfd);
FD_SET(sc->sh_master, &rdfd);
if (select(sc->sh_master + 1, &rdfd, NULL, NULL, &tv) == 1) {
if (len == sizeof(bp)) {
if (CHAR2INT(sp, t, len, ip, ilen) ||
db_append(sp, 1, lno++, ip, ilen))
return (1);
endp = bp;
} else {
memmove(bp, t, len);
endp = bp + len;
}
goto more;
}
if (sscr_setprompt(sp, t, len))
return (1);
}
/* Append the remains into the file, and the cursor moves to EOF. */
if (len > 0) {
if (CHAR2INT(sp, t, len, ip, ilen) ||
db_append(sp, 1, lno++, ip, ilen))
return (1);
sp->cno = ilen - 1;
} else
sp->cno = 0;
sp->lno = lno;
return (vs_refresh(sp, 1));
}
/*
* sscr_getprompt --
* Eat lines printed by the shell until a line with no trailing
* carriage return comes; set the prompt from that line.
*/
static int
sscr_getprompt(SCR *sp)
{
CHAR_T *endp, *p, *t, buf[1024];
SCRIPT *sc;
struct pollfd pfd[1];
recno_t lline;
size_t llen, len;
u_int value;
int nr;
endp = buf;
len = sizeof(buf);
/* Wait up to a second for characters to read. */
sc = sp->script;
pfd[0].fd = sc->sh_master;
pfd[0].events = POLLIN;
switch (poll(pfd, 1, 5 * 1000)) {
case -1: /* Error or interrupt. */
msgq(sp, M_SYSERR, "poll");
goto prompterr;
case 0: /* Timeout */
msgq(sp, M_ERR, "Error: timed out");
goto prompterr;
default: /* Characters to read. */
break;
}
/* Read the characters. */
more: len = sizeof(buf) - (endp - buf);
switch (nr = read(sc->sh_master, endp, len)) {
case 0: /* EOF. */
msgq(sp, M_ERR, "Error: shell: EOF");
goto prompterr;
case -1: /* Error or interrupt. */
msgq(sp, M_SYSERR, "shell");
goto prompterr;
default:
endp += nr;
break;
}
/* If any complete lines, push them into the file. */
for (p = t = buf; p < endp; ++p) {
value = KEY_VAL(sp, *p);
if (value == K_CR || value == K_NL) {
if (db_last(sp, &lline) ||
db_append(sp, 0, lline, t, p - t))
goto prompterr;
t = p + 1;
}
}
if (p > buf) {
memmove(buf, t, endp - t);
endp = buf + (endp - t);
}
if (endp == buf)
goto more;
/* Wait up 1/10 of a second to make sure that we got it all. */
switch (poll(pfd, 1, 100)) {
case -1: /* Error or interrupt. */
msgq(sp, M_SYSERR, "poll");
goto prompterr;
case 0: /* Timeout */
break;
default: /* Characters to read. */
goto more;
}
/* Timed out, so theoretically we have a prompt. */
llen = endp - buf;
endp = buf;
/* Append the line into the file. */
if (db_last(sp, &lline) || db_append(sp, 0, lline, buf, llen)) {
prompterr: sscr_end(sp);
return (1);
}
return (sscr_setprompt(sp, buf, llen));
}
/*
* sscr_getprompt --
* Eat lines printed by the shell until a line with no trailing
* carriage return comes; set the prompt from that line.
*/
static int
sscr_getprompt(SCR *sp)
{
struct timeval tv;
CHAR_T *endp, *p, *t, buf[1024];
SCRIPT *sc;
fd_set fdset;
db_recno_t lline;
size_t llen, len;
e_key_t value;
int nr;
FD_ZERO(&fdset);
endp = buf;
len = sizeof(buf);
/* Wait up to a second for characters to read. */
tv.tv_sec = 5;
tv.tv_usec = 0;
sc = sp->script;
FD_SET(sc->sh_master, &fdset);
switch (select(sc->sh_master + 1, &fdset, NULL, NULL, &tv)) {
case -1: /* Error or interrupt. */
msgq(sp, M_SYSERR, "select");
goto prompterr;
case 0: /* Timeout */
msgq(sp, M_ERR, "Error: timed out");
goto prompterr;
case 1: /* Characters to read. */
break;
}
/* Read the characters. */
more: len = sizeof(buf) - (endp - buf);
switch (nr = read(sc->sh_master, endp, len)) {
case 0: /* EOF. */
msgq(sp, M_ERR, "Error: shell: EOF");
goto prompterr;
case -1: /* Error or interrupt. */
msgq(sp, M_SYSERR, "shell");
goto prompterr;
default:
endp += nr;
break;
}
/* If any complete lines, push them into the file. */
for (p = t = buf; p < endp; ++p) {
value = KEY_VAL(sp, *p);
if (value == K_CR || value == K_NL) {
if (db_last(sp, &lline) ||
db_append(sp, 0, lline, t, p - t))
goto prompterr;
t = p + 1;
}
}
if (p > buf) {
MEMMOVE(buf, t, endp - t);
endp = buf + (endp - t);
}
if (endp == buf)
goto more;
/* Wait up 1/10 of a second to make sure that we got it all. */
tv.tv_sec = 0;
tv.tv_usec = 100000;
switch (select(sc->sh_master + 1, &fdset, NULL, NULL, &tv)) {
case -1: /* Error or interrupt. */
msgq(sp, M_SYSERR, "select");
goto prompterr;
case 0: /* Timeout */
break;
case 1: /* Characters to read. */
goto more;
}
/* Timed out, so theoretically we have a prompt. */
llen = endp - buf;
endp = buf;
/* Append the line into the file. */
if (db_last(sp, &lline) || db_append(sp, 0, lline, buf, llen)) {
prompterr: sscr_end(sp);
return (1);
}
return (sscr_setprompt(sp, buf, llen));
}
