run() can now redirect stderr

Author: Daniel Kroening

src/goto-cc/as_mode.cpp

@@ -267,7 +267,7 @@ int as_modet::run_as()
   std::cout << '\n';
   #endif
 
-  return run(new_argv[0], new_argv, cmdline.stdin_file, "");
+  return run(new_argv[0], new_argv, cmdline.stdin_file);
 }
 
 int as_modet::as_hybrid_binary()
@@ -314,7 +314,7 @@ int as_modet::as_hybrid_binary()
     objcopy_argv.push_back("--remove-section=goto-cc");
     objcopy_argv.push_back(output_file);
 
-    result=run(objcopy_argv[0], objcopy_argv, "", "");
+    result = run(objcopy_argv[0], objcopy_argv);
   }
 
   if(result==0)
@@ -327,7 +327,7 @@ int as_modet::as_hybrid_binary()
     objcopy_argv.push_back("goto-cc="+saved);
     objcopy_argv.push_back(output_file);
 
-    result=run(objcopy_argv[0], objcopy_argv, "", "");
+    result = run(objcopy_argv[0], objcopy_argv);
   }
 
   int remove_result=remove(saved.c_str());
@@ -354,7 +354,7 @@ int as_modet::as_hybrid_binary()
     lipo_argv.push_back("-output");
     lipo_argv.push_back(output_file);
 
-    result=run(lipo_argv[0], lipo_argv, "", "");
+    result = run(lipo_argv[0], lipo_argv);
   }
 
   int remove_result=remove(saved.c_str());

src/goto-cc/gcc_mode.cpp

@@ -831,7 +831,7 @@ int gcc_modet::run_gcc(const compilet &compiler)
     debug() << " " << new_argv[i];
   debug() << eom;
 
-  return run(new_argv[0], new_argv, cmdline.stdin_file, "");
+  return run(new_argv[0], new_argv, cmdline.stdin_file);
 }
 
 int gcc_modet::gcc_hybrid_binary(compilet &compiler)

src/util/run.cpp

@@ -10,8 +10,6 @@ Date: August 2012
 
 #include "run.h"
 
-#include <cassert>
-
 #ifdef _WIN32
 #include <process.h>
 #else
@@ -30,6 +28,7 @@ Date: August 2012
 
 #endif
 
+#include <util/invariant.h>
 #include <util/unicode.h>
 #include <util/signal_catcher.h>
 
@@ -39,19 +38,21 @@ int run_shell(const std::string &command)
   std::vector<std::string> argv;
   argv.push_back(shell);
   argv.push_back(command);
-  return run(shell, argv, "", "");
+  return run(shell, argv, "", "", "");
 }
 
 int run(
   const std::string &what,
   const std::vector<std::string> &argv,
   const std::string &std_input,
-  const std::string &std_output)
+  const std::string &std_output,
+  const std::string &std_error)
 {
   #ifdef _WIN32
-  // we don't support stdin/stdout redirection on Windows
-  assert(std_input.empty());
-  assert(std_output.empty());
+  // we don't support stdin/stdout/stderr redirection on Windows
+  PRECONDITION(std_input.empty());
+  PRECONDITION(std_output.empty());
+  PRECONDITION(std_error.empty());
 
   // unicode version of the arguments
   std::vector<std::wstring> wargv;
@@ -101,6 +102,18 @@ int run(
     }
   }
 
+  int stderr_fd = STDERR_FILENO;
+
+  if(!std_error.empty())
+  {
+    stderr_fd = open(std_error.c_str(), O_CREAT | O_WRONLY, S_IRUSR | S_IWUSR);
+    if(stderr_fd == -1)
+    {
+      perror("Failed to open stderr copy");
+      return 1;
+    }
+  }
+
   // temporarily suspend all signals
   sigset_t new_mask, old_mask;
   sigemptyset(&new_mask);
@@ -127,6 +140,8 @@ int run(
         dup2(stdin_fd, STDIN_FILENO);
       if(stdout_fd!=STDOUT_FILENO)
         dup2(stdout_fd, STDOUT_FILENO);
+      if(stderr_fd != STDERR_FILENO)
+        dup2(stderr_fd, STDERR_FILENO);
 
       errno=0;
       execvp(what.c_str(), _argv.data());
@@ -153,13 +168,17 @@ int run(
             close(stdin_fd);
           if(stdout_fd!=STDOUT_FILENO)
             close(stdout_fd);
+          if(stderr_fd != STDERR_FILENO)
+            close(stderr_fd);
           return 1;
         }
 
       if(stdin_fd!=STDIN_FILENO)
         close(stdin_fd);
       if(stdout_fd!=STDOUT_FILENO)
         close(stdout_fd);
+      if(stderr_fd != STDERR_FILENO)
+        close(stderr_fd);
 
       return WEXITSTATUS(status);
     }
@@ -173,6 +192,8 @@ int run(
       close(stdin_fd);
     if(stdout_fd!=STDOUT_FILENO)
       close(stdout_fd);
+    if(stderr_fd != STDERR_FILENO)
+      close(stderr_fd);
 
     return 1;
   }

src/util/run.h

@@ -18,8 +18,9 @@ Date: August 2012
 int run(
   const std::string &what,
   const std::vector<std::string> &argv,
-  const std::string &std_input,
-  const std::string &std_output);
+  const std::string &std_input = "",
+  const std::string &std_output = "",
+  const std::string &std_error = "");
 
 int run_shell(const std::string &command);