add pipestream ctor overload for vectors of string
[libi2ncommon] / src / pipestream.hxx
index 03ee60e..d13f312 100644 (file)
@@ -1,9 +1,27 @@
+ /*
+The software in this package is distributed under the GNU General
+Public License version 2 (with a special exception described below).
+
+A copy of GNU General Public License (GPL) is included in this distribution,
+in the file COPYING.GPL.
+
+As a special exception, if other files instantiate templates or use macros
+or inline functions from this file, or you compile this file and link it
+with other works to produce a work based on this file, this file
+does not by itself cause the resulting work to be covered
+by the GNU General Public License.
+
+However the source code for this file must still be made available
+in accordance with section (3) of the GNU General Public License.
+
+This exception does not invalidate any other reasons why a work based
+on this file might be covered by the GNU General Public License.
+*/
 /***************************************************************************
               inpipestream.hxx  -  C++ streambuffer wrapper 
                              -------------------
     begin                : Thu Dec 27 2001
     copyright            : (C) 2001 by Intra2net AG
-    email                : intranator@intra2net.com
  ***************************************************************************/
 
 #ifndef _PIPESTREAM
 #include <streambuf>
 #include <istream>
 #include <ostream>
-#include <cstdio>
+#include <vector>
 
-#include "exception.hxx"
+struct ExecResult
+{
+    /** if the program exited normally and returned a return code */
+    bool normal_exit;
+
+    /** the real return code of the program, only set when normal_exit true */
+    char return_code;
 
-// ATTENTION: A lot of mysterious STL bugs occured
-//            with a "real" buffer (buffer larger than 1 byte and up to 100 bytes)
-//            -> Keep it slow and working!
+    /** if the program was terminated by a signal */
+    bool terminated_by_signal;
 
+    /** number of the signal that terminated the program, only valid when terminated_by_signal true */
+    int signal;
+
+    /** errormessage if we have one */
+    std::string error_message;
+};
+typedef struct ExecResult ExecResult;
+
+std::string capture_exec(const std::string& command, ExecResult &rescode);
+std::string capture_exec(const char *const *command, ExecResult &rescode);
+std::string capture_exec(const std::vector<std::string>& command, ExecResult &rescode);
+
+inline std::string capture_exec (const std::string &command)
+{
+    ExecResult r;
+    return capture_exec(command,r);
+}
+
+inline std::string capture_exec(const char *const *command)
+{
+    ExecResult r;
+    return capture_exec(command,r);
+}
+
+inline std::string capture_exec(const std::vector<std::string>& command)
+{
+    ExecResult r;
+    return capture_exec(command,r);
+}
+
+/** @brief runs command and provides buffered input for it through pipe
+ *
+ * opens pipe to command using popen; exit status available after destruction
+ * (use WEXITSTATUS to get the "regular" return code (lowest byte))
+ *
+ * ATTENTION: A lot of mysterious STL bugs occured
+ *            with a "real" buffer (buffer larger than 1 byte and up to 100 bytes)
+ *            -> Keep it slow and working!
+ */
 class inpipebuf : public std::streambuf
 {
 protected:
     char buffer;
+
     FILE *pipe;
 
     // "callback" variables for destructor to store exit status
@@ -34,55 +97,22 @@ protected:
     int *exit_status;
 
 public:
-    inpipebuf(const std::string& command)
-    {
-        status_set = NULL;
-        exit_status = NULL;
-
-        pipe = popen (command.c_str(), "r");
-        if (pipe == NULL)
-            throw EXCEPTION (pipestream_error, "can't open program or permission denied");
-
-        // force underflow
-        setg (&buffer, &buffer, &buffer);
-    }
-
-    ~inpipebuf()
-    {
-        if (pipe != NULL) {
-            int pclose_exit = pclose (pipe);
+    inpipebuf(const std::string& command);
+    inpipebuf(const char *const *command);
+    inpipebuf(const std::vector<std::string> &command);
 
-            if (exit_status && pclose_exit != -1) {
-                *status_set = true;
-                *exit_status = pclose_exit;
-            }
+    ~inpipebuf();
 
-            pipe = NULL;
-        }
-    }
-
-    void store_exit_status(bool *_status_set, int *_exit_status)
-    { status_set = _status_set; exit_status = _exit_status; }
+    void store_exit_status(bool *_status_set, int *_exit_status);
 
 protected:
-    virtual int_type underflow()
-    {
-        if (gptr() < egptr())
-            return traits_type::to_int_type(*gptr());
-
-        buffer = fgetc (pipe);
-        if (feof (pipe))
-        {
-            // ERROR or EOF
-            return EOF;
-        }
-
-        setg (&buffer, &buffer, &buffer+sizeof(char));
-
-        return traits_type::to_int_type(*gptr());
-    }
+    virtual int_type underflow();
+
+private:
+    FILE *init_without_shell (const char *const *argv) const;
 };
 
+/** @brief stream around inpipebuf -- see comment there */
 class inpipestream : public std::istream
 {
 protected:
@@ -90,13 +120,25 @@ protected:
 
 public:
     inpipestream(const std::string& command)
-            : buf(command), std::istream(&buf)
+            : std::istream(&buf), buf(command)
+    {}
+    inpipestream(const char *const command[])
+            : std::istream(&buf), buf(command)
+    {}
+
+    inpipestream(const std::vector<std::string> &command)
+            : std::istream(&buf), buf(command)
     {}
 
     void store_exit_status(bool *_status_set, int *_exit_status)
     { buf.store_exit_status(_status_set, _exit_status); }
 };
 
+/** @brief runs command and provides buffered ouptput from it through pipe
+ *
+ * opens pipe to command using popen; exit status available after destruction
+ * (use WEXITSTATUS to get the "regular" return code (lowest byte))
+ */
 class outpipebuf : public std::streambuf
 {
 protected:
@@ -107,57 +149,28 @@ protected:
     int *exit_status;
 
 public:
-    outpipebuf(const std::string& command)
-    {
-        status_set = NULL;
-        exit_status = NULL;
-
-        pipe = popen (command.c_str(), "w");
-        if (pipe == NULL)
-            throw EXCEPTION (pipestream_error, "can't open program or permission denied");
-    }
-
-    ~outpipebuf()
-    {
-        if (pipe != NULL) {
-            int pclose_exit = pclose (pipe);
-
-            if (exit_status && pclose_exit != -1) {
-                *status_set = true;
-                *exit_status = pclose_exit;
-            }
-
-            pipe = NULL;
-        }
-    }
+    outpipebuf(const std::string& command);
 
-    void store_exit_status(bool *_status_set, int *_exit_status)
-    { status_set = _status_set; exit_status = _exit_status; }
+    ~outpipebuf();
+
+    /** note: exit status only available after destruction */
+    void store_exit_status(bool *_status_set, int *_exit_status);
 
 protected:
-    virtual int_type overflow(int_type c)
-    {
-        if (c != EOF)
-        {
-            if (fputc(c,pipe)==EOF)
-                return EOF;
-        }
-        return c;
-    }
-
-    virtual std::streamsize xsputn(const char* s, std::streamsize num)
-    {
-        return fwrite(s,num,1,pipe);
-    }
+    virtual int_type overflow(int_type c);
+
+    virtual std::streamsize xsputn(const char* s, std::streamsize num);
 };
 
+
+/** @brief stream around outpipebuf -- see comment there */
 class outpipestream : public std::ostream
 {
 protected:
     outpipebuf buf;
 public:
     outpipestream(const std::string& command)
-            : buf(command), std::ostream(&buf)
+            : std::ostream(&buf), buf(command)
     {}
 
     void store_exit_status(bool *_status_set, int *_exit_status)