JasonWoof Got questions, comments, patches, etc.? Contact Jason Woofenden
correct year of copyright notice
[st.git] / std.c
diff --git a/std.c b/std.c
index 5b430b8..ef9e57a 100644 (file)
--- 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>
@@ -17,7 +21,7 @@
 static void buffer(char c);
 static void cmd(const char *cmdstr, ...);
 static int getch();
-void getpty(void);
+static void getpty(void);
 static void movea(int x, int y);
 static void mover(int x, int y);
 static void parseesc(void);
@@ -41,7 +45,7 @@ typedef struct {
 static int cols = 80, lines = 25;
 static int cx = 0, cy = 0;
 static int c;
-int ptm, pts;
+static int ptm, pts;
 static _Bool bold, digit, qmark;
 static pid_t pid;
 static RingBuffer buf;
@@ -208,6 +212,38 @@ scroll(int 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;
 
@@ -264,24 +300,18 @@ ungetch(int c) {
 int
 main(int argc, char *argv[]) {
        fd_set rfds;
-       int r;
 
-       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();
        FD_ZERO(&rfds);
        FD_SET(STDIN_FILENO, &rfds);
        FD_SET(ptm, &rfds);
        for(;;) {
-               r = select(ptm + 1, &rfds, NULL, NULL, NULL);
-               if(r == -1)
+               if(select(ptm + 1, &rfds, NULL, NULL, NULL) == -1)
                        err(EXIT_FAILURE, "cannot select");
                if(FD_ISSET(ptm, &rfds)) {
                        do {