X-Git-Url: https://git.xinqibao.xyz/st.git/blobdiff_plain/0cbcedb89ffe6e29b3c0640f371bed578c6e0c07..d61a2a8fce2638594644feb664ce04398825b0f3:/std.c?ds=inline

diff --git a/std.c b/std.c
index 5bc27b8..a9a4f8c 100644
--- a/std.c
+++ b/std.c
@@ -3,6 +3,10 @@
 #include <sys/wait.h>
 #include <ctype.h>
 #include <err.h>
+#include <fcntl.h>
+#if !(_POSIX_C_SOURCE >= 200112L || _XOPEN_SOURCE >= 600)
+#include <pty.h>
+#endif
 #include <signal.h>
 #include <stdarg.h>
 #include <stdio.h>
@@ -14,19 +18,6 @@
 #define MAX(a,b)	(((a) > (b)) ? (a) : (b))
 #define MIN(a,b)	(((a) < (b)) ? (a) : (b))
 
-static void buffer(char c);
-static void cmd(const char *cmdstr, ...);
-static int getch();
-void getpty(void);
-static void movea(int x, int y);
-static void mover(int x, int y);
-static void parseesc(void);
-static void scroll(int l);
-static void shell(void);
-static void sigchld(int n);
-static char unbuffer(void);
-static void ungetch(int c);
-
 typedef struct {
 	unsigned char data[BUFSIZ];
 	int s, e;
@@ -36,8 +27,23 @@ typedef struct {
 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);
+static void parsecmd(void);
+static void parseesc(void);
+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;
 static int c;
@@ -45,7 +51,7 @@ static int ptm, pts;
 static _Bool bold, digit, qmark;
 static pid_t pid;
 static RingBuffer buf;
-static ReadBuffer rbuf;
+static ReadBuffer cmdbuf, ptmbuf;
 
 void
 buffer(char c) {
@@ -69,14 +75,14 @@ cmd(const char *cmdstr, ...) {
 }
 
 int
-getch() {
-	if(rbuf.i++ >= rbuf.n) {
-		rbuf.n = read(ptm, rbuf.data, LENGTH(rbuf.data));
-		if(rbuf.n == -1)
-			err(EXIT_FAILURE, "cannot read from slave pty");
-		rbuf.i = 0;
+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 rbuf.data[rbuf.i];
+	return buf->data[buf->i];
 }
 
 void
@@ -93,16 +99,20 @@ mover(int x, int y) {
 	movea(cx + x, cy + y);
 }
 
+void
+parsecmd(void) {
+}
+
 void
 parseesc(void) {
 	int i, j;
 	int arg[16];
 
 	memset(arg, 0, LENGTH(arg));
-	c = getch();
+	c = getch(&ptmbuf);
 	switch(c) {
 	case '[':
-		c = getch();
+		c = getch(&ptmbuf);
 		for(j = 0; j < LENGTH(arg);) {
 			if(isdigit(c)) {
 				digit = 1;
@@ -124,7 +134,7 @@ parseesc(void) {
 				}
 				break;
 			}
-			c = getch();
+			c = getch(&ptmbuf);
 		}
 		switch(c) {
 		case '@':
@@ -198,7 +208,7 @@ parseesc(void) {
 		break;
 	default:
 		putchar('\033');
-		ungetch(c);
+		ungetch(&ptmbuf, c);
 	}
 }
 
@@ -207,6 +217,38 @@ scroll(int l) {
 	cmd("seek(%d,%d)", cx, cy + l);
 }
 
+void
+getpty(void) {
+	char *ptsdev;
+
+#if defined(_GNU_SOURCE)
+	ptm = getpt();
+#elif _POSIX_C_SOURCE >= 200112L || _XOPEN_SOURCE >= 600
+	ptm = posix_openpt(O_RDWR);
+#else
+	ptm = open("/dev/ptmx", O_RDWR);
+	if(ptm == -1)
+		if(openpty(&ptm, &pts, NULL, NULL, NULL) == -1)
+			err(EXIT_FAILURE, "cannot open pty");
+#endif
+#if defined(_XOPEN_SOURCE)
+	if(ptm != -1) {
+		if(grantpt(ptm) == -1)
+			err(EXIT_FAILURE, "cannot grant access to pty");
+		if(unlockpt(ptm) == -1)
+			err(EXIT_FAILURE, "cannot unlock pty");
+		ptsdev = ptsname(ptm);
+		if(!ptsdev)
+			err(EXIT_FAILURE, "slave pty name undefined");
+		pts = open(ptsdev, O_RDWR);
+		if(pts == -1)
+			err(EXIT_FAILURE, "cannot open slave pty");
+	}
+	else
+		err(EXIT_FAILURE, "cannot open pty");
+#endif
+}
+
 void
 shell(void) {
 	static char *shell = NULL;
@@ -255,35 +297,44 @@ unbuffer(void) {
 }
 
 void
-ungetch(int c) {
-	if(rbuf.i + 1 >= rbuf.n)
-		errx(EXIT_FAILURE, "read buffer full");
-	rbuf.data[rbuf.i++] = c;
+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;
 
-	if(argc == 2 && !strcmp("-v", argv[1])) {
-		fprintf(stderr, "std-"VERSION", © 2008 Matthias-Christian Ott\n");
-		exit(EXIT_SUCCESS);
-	}
-	else if(argc == 1) {
-		fprintf(stderr, "usage: st [-v]\n");
-		exit(EXIT_FAILURE);
-	}
+	if(argc == 2 && !strcmp("-v", argv[1]))
+		errx(EXIT_SUCCESS, "std-"VERSION", © 2008 Matthias-Christian Ott");
+	else if(argc == 1)
+		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);
 	for(;;) {
 		if(select(ptm + 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:
+					break;
+				}
+			} while(cmdbuf.i < cmdbuf.n);
 		if(FD_ISSET(ptm, &rfds)) {
 			do {
-				c = getch();
+				c = getch(&ptmbuf);
 				switch(c) {
 				case '\033':
 					parseesc();
@@ -291,7 +342,7 @@ main(int argc, char *argv[]) {
 				default:
 					putchar(c);
 				}
-			} while(rbuf.i < rbuf.n);
+			} while(ptmbuf.i < ptmbuf.n);
 			fflush(stdout);
 		}
 	}