Commit | Line | Data |
---|---|---|
0e23f538 TJ |
1 | /* |
2 | The software in this package is distributed under the GNU General | |
3 | Public License version 2 (with a special exception described below). | |
4 | ||
5 | A copy of GNU General Public License (GPL) is included in this distribution, | |
6 | in the file COPYING.GPL. | |
7 | ||
8 | As a special exception, if other files instantiate templates or use macros | |
9 | or inline functions from this file, or you compile this file and link it | |
10 | with other works to produce a work based on this file, this file | |
11 | does not by itself cause the resulting work to be covered | |
12 | by the GNU General Public License. | |
13 | ||
14 | However the source code for this file must still be made available | |
15 | in accordance with section (3) of the GNU General Public License. | |
16 | ||
17 | This exception does not invalidate any other reasons why a work based | |
18 | on this file might be covered by the GNU General Public License. | |
19 | */ | |
e93545dd GE |
20 | /*************************************************************************** |
21 | inpipestream.hxx - C++ streambuffer wrapper | |
22 | ------------------- | |
23 | begin : Thu Dec 27 2001 | |
24 | copyright : (C) 2001 by Intra2net AG | |
e93545dd GE |
25 | ***************************************************************************/ |
26 | ||
27 | #ifndef _PIPESTREAM | |
28 | #define _PIPESTREAM | |
29 | ||
be3dbe32 TJ |
30 | #include <stdio.h> |
31 | ||
ad4490f1 | 32 | #include <cstring> |
e93545dd GE |
33 | #include <string> |
34 | #include <streambuf> | |
4a56f9ac GE |
35 | #include <istream> |
36 | #include <ostream> | |
c2c29997 | 37 | #include <vector> |
e93545dd | 38 | |
8621037f PG |
39 | #include <stringfunc.hxx> |
40 | ||
f8f119bf GE |
41 | struct ExecResult |
42 | { | |
43 | /** if the program exited normally and returned a return code */ | |
44 | bool normal_exit; | |
45 | ||
46 | /** the real return code of the program, only set when normal_exit true */ | |
47 | char return_code; | |
48 | ||
49 | /** if the program was terminated by a signal */ | |
50 | bool terminated_by_signal; | |
51 | ||
52 | /** number of the signal that terminated the program, only valid when terminated_by_signal true */ | |
53 | int signal; | |
54 | ||
55 | /** errormessage if we have one */ | |
56 | std::string error_message; | |
8621037f PG |
57 | |
58 | inline std::string format (void) const | |
59 | { | |
60 | return std::string ("(") | |
61 | + "(normal_exit " + (this->normal_exit ? "T" : "F") + ") " | |
62 | "(return_code '" + I2n::to_string ((int)this->return_code) + "') " | |
63 | "(signal " + (this->terminated_by_signal | |
64 | ? strsignal (this->signal) | |
65 | : "<nil>") + "))" | |
66 | ; | |
67 | }; | |
f8f119bf GE |
68 | }; |
69 | typedef struct ExecResult ExecResult; | |
70 | ||
71 | std::string capture_exec(const std::string& command, ExecResult &rescode); | |
cc917897 | 72 | std::string capture_exec(const char *const *command, ExecResult &rescode, |
ad4490f1 | 73 | const bool out=true, const bool err=false, |
cdc166d2 | 74 | const bool path=false, const bool env=false); |
cc917897 | 75 | std::string capture_exec(const std::vector<std::string>& command, ExecResult &rescode, |
ad4490f1 | 76 | const bool out=true, const bool err=false, |
cdc166d2 | 77 | const bool path=false, const bool env=false); |
f8f119bf | 78 | |
9e322fb7 PG |
79 | inline std::string capture_exec (const std::string &command) |
80 | { | |
81 | ExecResult r; | |
82 | return capture_exec(command,r); | |
83 | } | |
84 | ||
85 | inline std::string capture_exec(const char *const *command) | |
f8f119bf GE |
86 | { |
87 | ExecResult r; | |
88 | return capture_exec(command,r); | |
89 | } | |
90 | ||
c2c29997 PG |
91 | inline std::string capture_exec(const std::vector<std::string>& command) |
92 | { | |
93 | ExecResult r; | |
94 | return capture_exec(command,r); | |
95 | } | |
96 | ||
bf4c487c CH |
97 | /** @brief runs command and provides buffered input for it through pipe |
98 | * | |
99 | * opens pipe to command using popen; exit status available after destruction | |
100 | * (use WEXITSTATUS to get the "regular" return code (lowest byte)) | |
101 | * | |
102 | * ATTENTION: A lot of mysterious STL bugs occured | |
103 | * with a "real" buffer (buffer larger than 1 byte and up to 100 bytes) | |
104 | * -> Keep it slow and working! | |
105 | */ | |
e93545dd GE |
106 | class inpipebuf : public std::streambuf |
107 | { | |
20a79471 TJ |
108 | protected: |
109 | char buffer; | |
9e322fb7 | 110 | |
20a79471 | 111 | FILE *pipe; |
17b459b3 | 112 | pid_t pid; |
20a79471 TJ |
113 | |
114 | // "callback" variables for destructor to store exit status | |
115 | bool *status_set; | |
116 | int *exit_status; | |
117 | ||
118 | public: | |
cc917897 | 119 | inpipebuf(const std::string& command, |
cdc166d2 | 120 | const bool out, const bool err, const bool path, const bool env); |
cc917897 | 121 | inpipebuf(const char *const *command, |
cdc166d2 | 122 | const bool out, const bool err, const bool path, const bool env); |
cc917897 | 123 | inpipebuf(const std::vector<std::string> &command, |
cdc166d2 | 124 | const bool out, const bool err, const bool path, const bool env); |
20a79471 | 125 | |
7e606af5 | 126 | ~inpipebuf(); |
20a79471 | 127 | |
7e606af5 | 128 | void store_exit_status(bool *_status_set, int *_exit_status); |
20a79471 TJ |
129 | |
130 | protected: | |
7e606af5 | 131 | virtual int_type underflow(); |
9e322fb7 PG |
132 | |
133 | private: | |
17b459b3 PG |
134 | std::pair <pid_t, FILE *> |
135 | init_without_shell (const char *const *argv, | |
ad4490f1 | 136 | const bool out, const bool err, |
cdc166d2 | 137 | const bool path, const bool env) const; |
20a79471 | 138 | }; |
e93545dd | 139 | |
bf4c487c | 140 | /** @brief stream around inpipebuf -- see comment there */ |
e93545dd GE |
141 | class inpipestream : public std::istream |
142 | { | |
20a79471 TJ |
143 | protected: |
144 | inpipebuf buf; | |
145 | ||
146 | public: | |
cc917897 | 147 | inpipestream(const std::string& command, |
ad4490f1 | 148 | const bool out=true, const bool err=false, |
cdc166d2 PG |
149 | const bool path=false, const bool env=false) |
150 | : std::istream(&buf), buf(command, out, err, path, env) | |
20a79471 | 151 | {} |
cc917897 PG |
152 | |
153 | inpipestream(const char *const command[], | |
ad4490f1 | 154 | const bool out=true, const bool err=false, |
cdc166d2 PG |
155 | const bool path=false, const bool env=false) |
156 | : std::istream(&buf), buf(command, out, err, path, env) | |
9e322fb7 | 157 | {} |
20a79471 | 158 | |
cc917897 | 159 | inpipestream(const std::vector<std::string> &command, |
ad4490f1 | 160 | const bool out=true, const bool err=false, |
cdc166d2 PG |
161 | const bool path=false, const bool env=false) |
162 | : std::istream(&buf), buf(command, out, err, path, env) | |
c2c29997 PG |
163 | {} |
164 | ||
20a79471 TJ |
165 | void store_exit_status(bool *_status_set, int *_exit_status) |
166 | { buf.store_exit_status(_status_set, _exit_status); } | |
e93545dd GE |
167 | }; |
168 | ||
bf4c487c CH |
169 | /** @brief runs command and provides buffered ouptput from it through pipe |
170 | * | |
171 | * opens pipe to command using popen; exit status available after destruction | |
172 | * (use WEXITSTATUS to get the "regular" return code (lowest byte)) | |
173 | */ | |
e93545dd GE |
174 | class outpipebuf : public std::streambuf |
175 | { | |
20a79471 TJ |
176 | protected: |
177 | FILE *pipe; | |
178 | ||
179 | // "callback" variables for destructor to store exit status | |
180 | bool *status_set; | |
181 | int *exit_status; | |
182 | ||
183 | public: | |
7e606af5 GE |
184 | outpipebuf(const std::string& command); |
185 | ||
186 | ~outpipebuf(); | |
20a79471 | 187 | |
bf4c487c | 188 | /** note: exit status only available after destruction */ |
7e606af5 | 189 | void store_exit_status(bool *_status_set, int *_exit_status); |
20a79471 TJ |
190 | |
191 | protected: | |
7e606af5 GE |
192 | virtual int_type overflow(int_type c); |
193 | ||
194 | virtual std::streamsize xsputn(const char* s, std::streamsize num); | |
e93545dd GE |
195 | }; |
196 | ||
bf4c487c CH |
197 | |
198 | /** @brief stream around outpipebuf -- see comment there */ | |
e93545dd GE |
199 | class outpipestream : public std::ostream |
200 | { | |
20a79471 TJ |
201 | protected: |
202 | outpipebuf buf; | |
203 | public: | |
204 | outpipestream(const std::string& command) | |
d8ffcdb0 | 205 | : std::ostream(&buf), buf(command) |
20a79471 TJ |
206 | {} |
207 | ||
208 | void store_exit_status(bool *_status_set, int *_exit_status) | |
209 | { buf.store_exit_status(_status_set, _exit_status); } | |
e93545dd GE |
210 | }; |
211 | ||
212 | #endif |