summary refs log tree commit diff
diff options
context:
space:
mode:
authorJune McEnroe <june@causal.agency>2018-03-06 12:55:33 -0500
committerJune McEnroe <june@causal.agency>2018-03-06 12:55:33 -0500
commit6dc9f89da7ecbc9ee859f7e2438e62eb0c9b5a52 (patch)
tree00a9207f411954b02d64cac735cc829b834b1672
parentTrigger redraw on atch with 1x1 winsize (diff)
downloadsrc-6dc9f89da7ecbc9ee859f7e2438e62eb0c9b5a52.tar.gz
src-6dc9f89da7ecbc9ee859f7e2438e62eb0c9b5a52.zip
Yet more tweaking of dtch
-rw-r--r--bin/dtch.c59
1 files changed, 23 insertions, 36 deletions
diff --git a/bin/dtch.c b/bin/dtch.c
index fb5bc89a..7ec71801 100644
--- a/bin/dtch.c
+++ b/bin/dtch.c
@@ -132,14 +132,9 @@ static int dtch(int argc, char *argv[]) {
     addr = sockAddr(user->pw_dir, name);
     error = bind(server, (struct sockaddr *)&addr, sizeof(addr));
     if (error) err(EX_CANTCREAT, "%s", addr.sun_path);
+    fcntl(server, F_SETFD, FD_CLOEXEC);
     atexit(unlinkAddr);
 
-    error = chmod(addr.sun_path, 0600);
-    if (error) err(EX_IOERR, "%s", addr.sun_path);
-
-    error = fcntl(server, F_SETFD, FD_CLOEXEC);
-    if (error) err(EX_IOERR, "fcntl");
-
     int pty;
     pid_t pid = forkpty(&pty, NULL, NULL, NULL);
     if (pid < 0) err(EX_OSERR, "forkpty");
@@ -173,9 +168,8 @@ static int dtch(int argc, char *argv[]) {
 
 static struct termios saveTerm;
 static void restoreTerm(void) {
-    tcsetattr(STDERR_FILENO, TCSADRAIN, &saveTerm);
-    fprintf(
-        stderr,
+    tcsetattr(STDIN_FILENO, TCSADRAIN, &saveTerm);
+    printf(
         "\x1b[?1049l" // rmcup
         "\x1b\x63\x1b[!p\x1b[?3;4l\x1b[4l\x1b>" // reset
     );
@@ -198,27 +192,24 @@ static int atch(int argc, char *argv[]) {
     if (pty < 0) err(EX_IOERR, "recvmsg");
 
     struct winsize window;
-    error = ioctl(STDERR_FILENO, TIOCGWINSZ, &window);
-    if (error) err(EX_IOERR, "ioctl(%d, TIOCGWINSZ)", STDERR_FILENO);
-
-    struct winsize redraw = window;
-    redraw.ws_row = 1;
-    redraw.ws_col = 1;
+    error = ioctl(STDIN_FILENO, TIOCGWINSZ, &window);
+    if (error) err(EX_IOERR, "TIOCGWINSZ");
 
+    struct winsize redraw = { .ws_row = 1, .ws_col = 1 };
     error = ioctl(pty, TIOCSWINSZ, &redraw);
-    if (error) err(EX_IOERR, "iotctl(%d, TIOCSWINSZ)", pty);
+    if (error) err(EX_IOERR, "TIOCSWINSZ");
 
     error = ioctl(pty, TIOCSWINSZ, &window);
-    if (error) err(EX_IOERR, "ioctl(%d, TIOCSWINSZ)", pty);
+    if (error) err(EX_IOERR, "TIOCSWINSZ");
 
-    error = tcgetattr(STDERR_FILENO, &saveTerm);
-    if (error) err(EX_IOERR, "tcgetattr(%d)", STDERR_FILENO);
+    error = tcgetattr(STDIN_FILENO, &saveTerm);
+    if (error) err(EX_IOERR, "tcgetattr");
     atexit(restoreTerm);
 
     struct termios raw;
     cfmakeraw(&raw);
-    error = tcsetattr(STDERR_FILENO, TCSADRAIN, &raw);
-    if (error) err(EX_IOERR, "tcsetattr(%d)", STDERR_FILENO);
+    error = tcsetattr(STDIN_FILENO, TCSADRAIN, &raw);
+    if (error) err(EX_IOERR, "tcsetattr");
 
     char buf[4096];
     struct pollfd fds[2] = {
@@ -226,26 +217,22 @@ static int atch(int argc, char *argv[]) {
         { .fd = pty, .events = POLLIN },
     };
     while (0 < poll(fds, 2, -1)) {
-        if (fds[0].revents & POLLIN) {
-            ssize_t readSize = read(STDIN_FILENO, buf, sizeof(buf));
-            if (readSize < 0) err(EX_IOERR, "read(%d)", STDIN_FILENO);
+        if (fds[0].revents) {
+            ssize_t size = read(STDIN_FILENO, buf, sizeof(buf));
+            if (size < 0) err(EX_IOERR, "read(%d)", STDIN_FILENO);
 
-            if (readSize == 1 && buf[0] == CTRL('Q')) return EX_OK;
+            if (size == 1 && buf[0] == CTRL('Q')) return EX_OK;
 
-            ssize_t writeSize = write(pty, buf, readSize);
-            if (writeSize < 0) err(EX_IOERR, "write(%d)", pty);
-            if (writeSize < readSize) errx(EX_IOERR, "short write(%d)", pty);
+            size = write(pty, buf, size);
+            if (size < 0) err(EX_IOERR, "write(%d)", pty);
         }
 
-        if (fds[1].revents & POLLIN) {
-            ssize_t readSize = read(pty, buf, sizeof(buf));
-            if (readSize < 0) err(EX_IOERR, "read(%d)", pty);
+        if (fds[1].revents) {
+            ssize_t size = read(pty, buf, sizeof(buf));
+            if (size < 0) err(EX_IOERR, "read(%d)", pty);
 
-            ssize_t writeSize = write(STDOUT_FILENO, buf, readSize);
-            if (writeSize < 0) err(EX_IOERR, "write(%d)", STDOUT_FILENO);
-            if (writeSize < readSize) {
-                errx(EX_IOERR, "short write(%d)", STDOUT_FILENO);
-            }
+            size = write(STDOUT_FILENO, buf, size);
+            if (size < 0) err(EX_IOERR, "write(%d)", STDOUT_FILENO);
         }
     }
     err(EX_IOERR, "poll");