use stdio(3)
This commit is contained in:
parent
07d1edcd76
commit
a7f50ebb64
95
std.c
95
std.c
@ -15,6 +15,7 @@
|
||||
#include <sys/wait.h>
|
||||
#include <ctype.h>
|
||||
#include <err.h>
|
||||
#include <errno.h>
|
||||
#include <fcntl.h>
|
||||
#if !(_POSIX_C_SOURCE >= 200112L || _XOPEN_SOURCE >= 600)
|
||||
#include <pty.h>
|
||||
@ -36,15 +37,8 @@ typedef struct {
|
||||
int n;
|
||||
} RingBuffer;
|
||||
|
||||
typedef struct {
|
||||
unsigned char data[BUFSIZ];
|
||||
int i, n;
|
||||
int fd;
|
||||
} ReadBuffer;
|
||||
|
||||
static void buffer(char c);
|
||||
static void cmd(const char *cmdstr, ...);
|
||||
static int getch(ReadBuffer *buf);
|
||||
static void getpty(void);
|
||||
static void movea(int x, int y);
|
||||
static void mover(int x, int y);
|
||||
@ -54,7 +48,6 @@ static void scroll(int l);
|
||||
static void shell(void);
|
||||
static void sigchld(int n);
|
||||
static char unbuffer(void);
|
||||
static void ungetch(ReadBuffer *buf, int c);
|
||||
|
||||
static int cols = 80, lines = 25;
|
||||
static int cx = 0, cy = 0;
|
||||
@ -63,7 +56,7 @@ static int ptm, pts;
|
||||
static _Bool bold, digit, qmark;
|
||||
static pid_t pid;
|
||||
static RingBuffer buf;
|
||||
static ReadBuffer cmdbuf, ptmbuf;
|
||||
static FILE *fptm;
|
||||
|
||||
void
|
||||
buffer(char c) {
|
||||
@ -86,17 +79,6 @@ cmd(const char *cmdstr, ...) {
|
||||
va_end(ap);
|
||||
}
|
||||
|
||||
int
|
||||
getch(ReadBuffer *buf) {
|
||||
if(buf->i++ >= buf->n) {
|
||||
buf->n = read(buf->fd, buf->data, BUFSIZ);
|
||||
if(buf->n == -1)
|
||||
err(EXIT_FAILURE, "cannot read");
|
||||
buf->i = 0;
|
||||
}
|
||||
return buf->data[buf->i];
|
||||
}
|
||||
|
||||
void
|
||||
movea(int x, int y) {
|
||||
x = MAX(x, cols);
|
||||
@ -121,10 +103,10 @@ parseesc(void) {
|
||||
int arg[16];
|
||||
|
||||
memset(arg, 0, LENGTH(arg));
|
||||
c = getch(&ptmbuf);
|
||||
c = getc(fptm);
|
||||
switch(c) {
|
||||
case '[':
|
||||
c = getch(&ptmbuf);
|
||||
c = getc(fptm);
|
||||
for(j = 0; j < LENGTH(arg);) {
|
||||
if(isdigit(c)) {
|
||||
digit = 1;
|
||||
@ -146,7 +128,7 @@ parseesc(void) {
|
||||
}
|
||||
break;
|
||||
}
|
||||
c = getch(&ptmbuf);
|
||||
c = getc(fptm);
|
||||
}
|
||||
switch(c) {
|
||||
case '@':
|
||||
@ -220,7 +202,7 @@ parseesc(void) {
|
||||
break;
|
||||
default:
|
||||
putchar('\033');
|
||||
ungetch(&ptmbuf, c);
|
||||
ungetc(c, fptm);
|
||||
}
|
||||
}
|
||||
|
||||
@ -308,13 +290,6 @@ unbuffer(void) {
|
||||
return c;
|
||||
}
|
||||
|
||||
void
|
||||
ungetch(ReadBuffer *buf, int c) {
|
||||
if(buf->i + 1 >= buf->n)
|
||||
errx(EXIT_FAILURE, "buffer full");
|
||||
buf->data[buf->i++] = c;
|
||||
}
|
||||
|
||||
int
|
||||
main(int argc, char *argv[]) {
|
||||
fd_set rfds;
|
||||
@ -325,28 +300,31 @@ main(int argc, char *argv[]) {
|
||||
errx(EXIT_FAILURE, "usage: std [-v]");
|
||||
getpty();
|
||||
shell();
|
||||
cmdbuf.fd = STDIN_FILENO;
|
||||
ptmbuf.fd = ptm;
|
||||
FD_ZERO(&rfds);
|
||||
FD_SET(STDIN_FILENO, &rfds);
|
||||
FD_SET(ptm, &rfds);
|
||||
if(!(fptm = fdopen(ptm, "r+")))
|
||||
err(EXIT_FAILURE, "cannot open pty");
|
||||
if(fcntl(ptm, F_SETFL, O_NONBLOCK) == -1)
|
||||
err(EXIT_FAILURE, "cannot set pty to non-blocking mode");
|
||||
if(fcntl(STDIN_FILENO, F_SETFL, O_NONBLOCK) == -1)
|
||||
err(EXIT_FAILURE, "cannot set stdin to non-blocking mode");
|
||||
for(;;) {
|
||||
if(select(ptm + 1, &rfds, NULL, NULL, NULL) == -1)
|
||||
if(select(MAX(ptm, STDIN_FILENO) + 1, &rfds, NULL, NULL, NULL) == -1)
|
||||
err(EXIT_FAILURE, "cannot select");
|
||||
if(FD_ISSET(STDIN_FILENO, &rfds))
|
||||
do {
|
||||
c = getch(&cmdbuf);
|
||||
switch(c) {
|
||||
case ':':
|
||||
parsecmd();
|
||||
break;
|
||||
default:
|
||||
if(FD_ISSET(ptm, &rfds)) {
|
||||
for(;;) {
|
||||
if((c = getc(fptm)) == EOF) {
|
||||
if(feof(fptm)) {
|
||||
FD_CLR(ptm, &rfds);
|
||||
fflush(fptm);
|
||||
break;
|
||||
}
|
||||
if(errno != EAGAIN)
|
||||
err(EXIT_FAILURE, "cannot read from pty");
|
||||
fflush(stdout);
|
||||
break;
|
||||
}
|
||||
} while(cmdbuf.i < cmdbuf.n);
|
||||
if(FD_ISSET(ptm, &rfds)) {
|
||||
do {
|
||||
c = getch(&ptmbuf);
|
||||
switch(c) {
|
||||
case '\033':
|
||||
parseesc();
|
||||
@ -354,9 +332,32 @@ main(int argc, char *argv[]) {
|
||||
default:
|
||||
putchar(c);
|
||||
}
|
||||
} while(ptmbuf.i < ptmbuf.n);
|
||||
}
|
||||
fflush(stdout);
|
||||
}
|
||||
if(FD_ISSET(STDIN_FILENO, &rfds)) {
|
||||
for(;;) {
|
||||
if((c = getchar()) == EOF) {
|
||||
if(feof(stdin)) {
|
||||
FD_CLR(STDIN_FILENO, &rfds);
|
||||
fflush(fptm);
|
||||
break;
|
||||
}
|
||||
if(errno != EAGAIN)
|
||||
err(EXIT_FAILURE, "cannot read from stdin");
|
||||
fflush(fptm);
|
||||
break;
|
||||
}
|
||||
switch(c) {
|
||||
case ':':
|
||||
parsecmd();
|
||||
break;
|
||||
default:
|
||||
putc(c, fptm);
|
||||
}
|
||||
}
|
||||
fflush(fptm);
|
||||
}
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user