summary refs log tree commit diff
diff options
context:
space:
mode:
authorJune McEnroe <june@causal.agency>2016-09-17 00:35:22 -0400
committerJune McEnroe <june@causal.agency>2016-09-17 00:35:22 -0400
commit61e37f3654a3e5b2a28382d7dec883288ac18c5d (patch)
tree7b7e52475b66af02de446600a5593ecb2f0228a4
parentAdd initial pbd implementation (diff)
downloadsrc-61e37f3654a3e5b2a28382d7dec883288ac18c5d.tar.gz
src-61e37f3654a3e5b2a28382d7dec883288ac18c5d.zip
Factor out spawn in pbd
-rwxr-xr-x.bin/pbd.c49
1 files changed, 19 insertions, 30 deletions
diff --git a/.bin/pbd.c b/.bin/pbd.c
index 713800d4..3f924f2c 100755
--- a/.bin/pbd.c
+++ b/.bin/pbd.c
@@ -1,8 +1,7 @@
 #if 0
-exec clang -Weverything $@ -o $(dirname $0)/pbd $0
+exec clang -Weverything -Wno-missing-prototypes $@ -o $(dirname $0)/pbd $0
 #endif
 
-#include <arpa/inet.h>
 #include <err.h>
 #include <netinet/in.h>
 #include <stdint.h>
@@ -12,6 +11,22 @@ exec clang -Weverything $@ -o $(dirname $0)/pbd $0
 #include <sysexits.h>
 #include <unistd.h>
 
+void spawn(const char *cmd, int child_fd, int parent_fd) {
+    pid_t pid = fork();
+    if (pid < 0) err(EX_OSERR, "fork");
+
+    if (pid) {
+        if (waitpid(pid, NULL, 0) < 0)
+            err(EX_OSERR, "waitpid");
+        // TODO: Check child status.
+    } else {
+        if (dup2(parent_fd, child_fd) < 0)
+            err(EX_OSERR, "dup2");
+        if (execlp(cmd, cmd) < 0)
+            err(EX_OSERR, "execlp");
+    }
+}
+
 int main() {
     int server = socket(PF_INET, SOCK_STREAM, 0);
     if (server < 0) err(EX_OSERR, "socket");
@@ -32,39 +47,13 @@ int main() {
         int client = accept(server, NULL, NULL);
         if (client < 0) err(EX_OSERR, "accept");
 
-        pid_t pid_paste = fork();
-        if (pid_paste < 0) err(EX_OSERR, "fork");
-
-        if (pid_paste) {
-            if (waitpid(pid_paste, NULL, 0) < 0)
-                warn("waitpid");
-            // TODO: Check child status.
-        } else {
-            if (dup2(client, STDOUT_FILENO) < 0)
-                err(EX_OSERR, "dup2");
-            if (execlp("pbpaste", "pbpaste") < 0)
-                err(EX_OSERR, "execlp");
-        }
+        spawn("pbpaste", STDOUT_FILENO, client);
 
         uint8_t x;
         ssize_t peek = recv(client, &x, 1, MSG_PEEK);
         if (peek < 0) err(EX_IOERR, "recv");
 
-        if (peek) {
-            pid_t pid_copy = fork();
-            if (pid_copy < 0) err(EX_OSERR, "fork");
-
-            if (pid_copy) {
-                if (waitpid(pid_copy, NULL, 0) < 0)
-                    warn("waitpid");
-                // TODO: Check child status.
-            } else {
-                if (dup2(client, STDIN_FILENO) < 0)
-                    err(EX_OSERR, "dup2");
-                if (execlp("pbcopy", "pbcopy") < 0)
-                    err(EX_OSERR, "execlp");
-            }
-        }
+        if (peek) spawn("pbcopy", STDIN_FILENO, client);
 
         if (close(client) < 0) warn("close");
     }