Changeset 97cdbaf5 for filterproc.c

Timestamp:

Mar 11, 2012, 10:57:35 PM (12 years ago)

Author:

David Benjamin <davidben@mit.edu>

Branches:

master, release-1.10, release-1.9

Children:

a03a409

Parents:

1f39ded

git-author:

David Benjamin <davidben@mit.edu> (01/23/12 00:38:29)

git-committer:

David Benjamin <davidben@mit.edu> (03/11/12 22:57:35)

Message:

Rewrite call_filter to use g_spawn_async_with_pipes

This simplifies the error-handling code. Also fixes a bug where file
descriptors get double-closed in call_filter. Also adds a unit test. The
separate prog argument is removed to avoid having to deal with
G_SPAWN_FILE_AND_ARGV_ZERO, and since we don't really use it anyway.

File:

• filterproc.c (modified) (4 diffs)

filterproc.c

@@ -3 +3 @@
 #include <poll.h>
 
-int send_receive(int rfd, int wfd, const char *out, char **in)
+/* Even in case of error, send_receive is responsible for closing wfd
+ * (to EOF the child) and rfd (for consistency). */
+static int send_receive(int rfd, int wfd, const char *out, char **in)
 {
   GString *str = g_string_new("");
@@ -21 +23 @@
   if(!out || !*out) {
     /* Nothing to write. Close our end so the child doesn't hang waiting. */
-    close(wfd);
+    close(wfd); wfd = -1;
     out = NULL;
   }
@@ -46 +48 @@
       }
       if(!out || !*out || fds[1].revents & (POLLERR | POLLHUP)) {
-        close(wfd);
+        close(wfd); wfd = -1;
         out = NULL;
       }
@@ -62 +64 @@
   }
 
+  if (wfd >= 0) close(wfd);
+  close(rfd);
   *in = g_string_free(str, err < 0);
   return err;
 }
 
-int call_filter(const char *prog, const char *const *argv, const char *in, char **out, int *status)
+int call_filter(const char *const *argv, const char *in, char **out, int *status)
 {
-  int err = 0;
-  pid_t pid;
-  int rfd[2];
-  int wfd[2];
+  int err;
+  GPid child_pid;
+  int child_stdin, child_stdout;
 
-  if((err = pipe(rfd))) goto out;
-  if((err = pipe(wfd))) goto out_close_rfd;
-
-  pid = fork();
-  if(pid < 0) {
-    err = pid;
-    goto out_close_all;
-  }
-  if(pid) {
-    /* parent */
-    close(rfd[1]);
-    close(wfd[0]);
-    err = send_receive(rfd[0], wfd[1], in, out);
-    if(err == 0) {
-      waitpid(pid, status, 0);
-    }
-  } else {
-    /* child */
-    close(rfd[0]);
-    close(wfd[1]);
-    dup2(rfd[1], 1);
-    dup2(wfd[0], 0);
-    close(rfd[1]);
-    close(wfd[0]);
-
-    if(execvp(prog, (char *const *)argv)) {
-      _exit(-1);
-    }
+  if (!g_spawn_async_with_pipes(NULL, (char**)argv, NULL,
+                                G_SPAWN_SEARCH_PATH | G_SPAWN_DO_NOT_REAP_CHILD,
+                                NULL, NULL,
+                                &child_pid, &child_stdin, &child_stdout, NULL,
+                                NULL)) {
+    return 1;
   }
 
- out_close_all:
-  close(wfd[0]);
-  close(wfd[1]);
- out_close_rfd:
-  close(rfd[0]);
-  close(rfd[1]);
- out:
+  err = send_receive(child_stdout, child_stdin, in, out);
+  if (err == 0) {
+    waitpid(child_pid, status, 0);
+  }
   return err;
 }