X-Git-Url: https://diplodocus.org/git/nmh/blobdiff_plain/ea1fadd88568e97c6891da8ebe09d02515282e66..94187a80bd60baab4b9c4b949ad820d730578123:/test/runpty.c

diff --git a/test/runpty.c b/test/runpty.c
index e8db35f2..dff70a15 100644
--- a/test/runpty.c
+++ b/test/runpty.c
@@ -21,14 +21,15 @@
 
 #define COMMAND_TIMEOUT 30
 
+static void run_command(char *argv[], int master_in, int master_out);
+static int open_master_pty(const char *desc);
 static void die(const char *fmt, ...);
 
 int
 main(int argc, char *argv[])
 {
-    int master_in, master_out, slave, cc, status;
+    int master_in, master_out, cc, status;
     time_t starttime, now;
-    const char *slavename;
     pid_t child;
     unsigned char readbuf[1024];
     FILE *output;
@@ -39,75 +40,14 @@ main(int argc, char *argv[])
             *argv, *argv);
     }
 
-    if ((master_in = posix_openpt(O_RDWR | O_NOCTTY)) < 0) {
-	die("Unable to open master pseudo-tty: %s\n", strerror(errno));
-    }
-
-    if ((master_out = posix_openpt(O_RDWR | O_NOCTTY)) < 0) {
-	die("Unable to open master pseudo-tty: %s\n", strerror(errno));
-    }
-
-    if (grantpt(master_in) < 0) {
-	die("Unable to grant permissions to master pty: %s\n",
-            strerror(errno));
-    }
-
-    if (grantpt(master_out) < 0) {
-	die("Unable to grant permissions to master pty: %s\n",
-            strerror(errno));
-    }
-
-    if (unlockpt(master_in) < 0) {
-	die("Unable to unlock master pty: %s\n", strerror(errno));
-    }
+    master_in = open_master_pty("input");
+    master_out = open_master_pty("output");
 
-    if (unlockpt(master_out) < 0) {
-	die("Unable to unlock master pty: %s\n", strerror(errno));
+    if ((child = fork()) == -1) {
+	die("fork() failed: %s\n", strerror(errno));
     }
-
-    child = fork();
-
-    /*
-     * Start the child process if we are in the child; open the two
-     * slave pseudo-ttys and close the masters after we are done with them.
-     */
-
     if (child == 0) {
-	if (!(slavename = ptsname(master_in))) {
-	    die("Unable to determine name of slave pty: %s\n",
-                strerror(errno));
-	}
-
-	if ((slave = open(slavename, O_RDWR)) < 0) {
-	    die("Unable to open slave pty \"%s\": %s\n", slavename,
-                strerror(errno));
-	}
-
-	dup2(slave, STDIN_FILENO);
-	close(slave);
-	close(master_in);
-
-	if (!(slavename = ptsname(master_out))) {
-	    die("Unable to determine name of slave pty: %s\n",
-                strerror(errno));
-	}
-
-	if ((slave = open(slavename, O_RDWR | O_NOCTTY)) < 0) {
-	    die("Unable to open slave pty \"%s\": %s\n", slavename,
-                strerror(errno));
-	}
-
-	dup2(slave, STDOUT_FILENO);
-	dup2(slave, STDERR_FILENO);
-	close(slave);
-	close(master_out);
-
-	execvp(argv[2], argv + 2);
-
-	die("execvp(%s) failed: %s\n", argv[2], strerror(errno));
-
-    } else if (child < 0) {
-	die("fork() failed: %s\n", strerror(errno));
+        run_command(argv + 2, master_in, master_out); /* Does not return. */
     }
 
     if (!(output = fopen(argv[1], "w"))) {
@@ -174,6 +114,68 @@ main(int argc, char *argv[])
     exit(0);
 }
 
+static void
+run_command(char *argv[], int master_in, int master_out)
+{
+    const char *slavename;
+    int slave;
+
+    /* Open the two slave pseudo-ttys and close the masters after we are
+     * done with them. */
+
+    if (!(slavename = ptsname(master_in))) {
+        die("Unable to determine name of slave pty: %s\n",
+            strerror(errno));
+    }
+
+    if ((slave = open(slavename, O_RDWR)) == -1) {
+        die("Unable to open slave pty \"%s\": %s\n", slavename,
+            strerror(errno));
+    }
+
+    dup2(slave, STDIN_FILENO);
+    close(slave);
+    close(master_in);
+
+    if (!(slavename = ptsname(master_out))) {
+        die("Unable to determine name of slave pty: %s\n",
+            strerror(errno));
+    }
+
+    if ((slave = open(slavename, O_RDWR | O_NOCTTY)) < 0) {
+        die("Unable to open slave pty \"%s\": %s\n", slavename,
+            strerror(errno));
+    }
+
+    dup2(slave, STDOUT_FILENO);
+    dup2(slave, STDERR_FILENO);
+    close(slave);
+    close(master_out);
+
+    execvp(*argv, argv);
+
+    die("execvp(%s) failed: %s\n", *argv, strerror(errno));
+}
+
+static int
+open_master_pty(const char *desc)
+{
+    int fd;
+
+    if ((fd = posix_openpt(O_RDWR | O_NOCTTY)) == -1) {
+	die("Unable to open master %s pseudo-tty: %s\n", desc, strerror(errno));
+    }
+    if (grantpt(fd) == -1) {
+	die("Unable to grant permissions to master %s pty: %s\n", desc,
+            strerror(errno));
+    }
+    if (unlockpt(fd) == -1) {
+	die("Unable to unlock master %s pty: %s\n", desc, strerror(errno));
+    }
+
+    return fd;
+}
+
 static void
 die(const char *fmt, ...)
 {