#include <errno.h>
#include <stdio.h>
#include <string.h>
+#include <fcntl.h>
#include <sys/wait.h>
#include <unistd.h>
/** @brief runs command and returns it's output as string
* @param command the full command with all parameters
* @param rescode struct containing the return code, if the program exited normally and so on
+ * @param out Whether to collect \c stdout.
+ * @param err Whether to collect \c stderr; combines with \c out.
+ * @param path Wether to look up the executable in \c $PATH.
* @returns the output (stdout) of the called program
*/
template <typename CmdT>
std::string capture_exec(CmdT command, ExecResult &rescode,
- const bool out, const bool err)
+ const bool out, const bool err,
+ const bool path)
{
std::string output;
try
{
{
- inpipestream ips(command, out, err);
+ inpipestream ips(command, out, err, path);
ips.store_exit_status(&exit_set, &exit_status_waitpid);
* passed.
*/
std::string capture_exec (const std::string &command, ExecResult &res)
-{ return capture_exec<const std::string &>(command, res, true, false); }
+{ return capture_exec<const std::string &>(command, res, true, false, false); }
/** @brief Instantiation of \c capture_exec for argument lists. The
* pipestream used to run the command will not shell out.
* state in this struct.
* @param out Whether to collect \c stdout.
* @param err Whether to collect \c stderr; combines with \c out.
+ * @param path Wether to look up the executable in \c $PATH.
*
* @returns Captured output, combined into one string.
*/
std::string capture_exec (const char *const *command, ExecResult &res,
- const bool out, const bool err)
-{ return capture_exec<const char *const *>(command, res, out, err); }
+ const bool out, const bool err, const bool path)
+{ return capture_exec<const char *const *>(command, res, out, err, path); }
/** @brief Instantiation of \c capture_exec for argument lists. The
* pipestream used to run the command will not shell out.
* state in this struct.
* @param out Whether to collect \c stdout.
* @param err Whether to collect \c stderr; combines with \c out.
+ * @param path Wether to look up the executable in \c $PATH.
*
* @returns Captured output, combined into one string.
*/
std::string capture_exec (const std::vector<std::string> &command, ExecResult &res,
- const bool out, const bool err)
+ const bool out, const bool err, const bool path)
{
return capture_exec<const std::vector<std::string> &>
- (command, res, out, err);
+ (command, res, out, err, path);
}
#define PIPE_CTOR_FAIL(where) \
return boost::shared_array<char *> (ret);
}
+/** @brief Helper for redirecting a file descriptor to \c /dev/null.
+ * This will only acquire an fd the first time it is called
+ * or if it is called after unsuccessfully attempting to
+ * acquire one.
+ *
+ * @param fd The open file descriptor to operate on.
+ *
+ * @returns \c true on success, \c false otherwise (the call to
+ * either \c open(2) or \c dup2(2) failed).
+ */
+static bool
+redirect_devnull (const int fd)
+{
+ static int nullfd = -1;
+
+ errno = 0;
+ if (nullfd == -1 && (nullfd = open ("/dev/null", O_RDWR)) == -1) {
+ return false;
+ }
+
+ errno = 0;
+ if (dup2 (nullfd, fd) == -1) {
+ return false;
+ }
+
+ return true;
+}
/** @brief Helper aggregating common code for the shell-free ctors.
*
* @returns A \c FILE* handle for streaming if successful, \c NULL
* otherwise.
*/
-FILE *
+std::pair <pid_t, FILE *>
inpipebuf::init_without_shell (const char *const *argv,
const bool out,
- const bool err) const
+ const bool err,
+ const bool path) const
{
FILE *pipeobj = NULL;
int pipefd [2];
close (pipefd [0]);
if (!out) {
- close (STDOUT_FILENO);
+ if (!redirect_devnull (STDOUT_FILENO)) {
+ fprintf(stderr, "redirect_devnull/stdout: %m\n");
+ /* XXX should we bail here? */
+ }
} else if (dup2 (pipefd[1], STDOUT_FILENO) == -1) {
- PIPE_CTOR_FAIL("dup2/stdout");
+ fprintf(stderr, "dup2/stdout: %m\n");
+ exit(EXIT_FAILURE);
}
if (!err) {
- close (STDERR_FILENO);
+ if (!redirect_devnull (STDERR_FILENO)) {
+ fprintf(stderr, "redirect_devnull/stderr: %m\n");
+ /* XXX should we bail here? */
+ }
} else if (dup2 (pipefd[1], STDERR_FILENO) == -1) {
- PIPE_CTOR_FAIL("dup2/stderr");
+ fprintf(stderr, "dup2/stderr: %m\n");
+ exit(EXIT_FAILURE);
}
+ close (pipefd [1]);
+
errno = 0;
- if (execve (argv [0], const_cast <char *const *>(argv), NULL) == -1) {
- PIPE_CTOR_FAIL("exec");
+ if (path) {
+ execvpe (argv [0], const_cast <char *const *>(argv), environ);
+ } else {
+ execve (argv [0], const_cast <char *const *>(argv), NULL);
}
+ exit(EXIT_FAILURE);
break;
}
default: {
}
}
- return pipeobj;
+ return std::make_pair (childpid, pipeobj);
}
inpipebuf::inpipebuf(const char *const *command,
const bool out,
- const bool err)
+ const bool err,
+ const bool path)
: pipe (NULL) /* brr: shadowing global ident */
+ , pid (-1)
, status_set (NULL)
, exit_status (NULL)
{
PIPE_CTOR_FAIL("command");
}
- this->pipe = this->init_without_shell (command, out, err);
+ std::pair <pid_t, FILE *> tmp =
+ this->init_without_shell (command, out, err, path);
+ this->pid = tmp.first; /* no std::tie :/ */
+ this->pipe = tmp.second;
setg (&buffer, &buffer, &buffer);
}
inpipebuf::inpipebuf(const std::vector<std::string> &command,
const bool out,
- const bool err)
+ const bool err,
+ const bool path)
: pipe (NULL) /* brr: shadowing global ident */
+ , pid (-1)
, status_set (NULL)
, exit_status (NULL)
{
PIPE_CTOR_FAIL("malloc");
}
- this->pipe = this->init_without_shell (argv.get (), out, err);
+ std::pair <pid_t, FILE *> tmp =
+ this->init_without_shell (argv.get (), out, err, path);
+ this->pid = tmp.first;
+ this->pipe = tmp.second;
setg (&buffer, &buffer, &buffer);
}
inpipebuf::inpipebuf(const std::string& command,
const bool _ignored_out,
- const bool _ignored_err)
+ const bool _ignored_err,
+ const bool _ignored_path)
+ : pid (-1)
+ , status_set (NULL)
+ , exit_status (NULL)
{
- 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");
inpipebuf::~inpipebuf()
{
if (pipe != NULL) {
- int pclose_exit = pclose (pipe);
+ int status;
- if (exit_status && pclose_exit != -1)
+ if (this->pid == -1)
{
- if (status_set)
- *status_set = true;
- *exit_status = pclose_exit;
+ errno = 0;
+ status = pclose (pipe);
+ if (status != -1) {
+ if (exit_status != NULL) {
+ *exit_status = status;
+ if (status_set != NULL) {
+ *status_set = true;
+ }
+ }
+ }
+ }
+ else
+ {
+ errno = 0;
+ status = fclose (pipe);
+ if (status != EOF) {
+ if (exit_status != NULL) {
+ *exit_status = status; /* might be overwritten below */
+ if (status_set != NULL) {
+ *status_set = true;
+ }
+ }
+ }
+
+ errno = 0;
+ while (waitpid (this->pid, &status, 0) == -1) {
+ if (errno != EINTR) {
+ status = -1;
+ break;
+ }
+ }
+ if (status != 0 && exit_status != NULL) {
+ *exit_status = status; /* might overwrite pipe status above */
+ if (status_set != NULL) {
+ *status_set = true;
+ }
+ }
}
pipe = NULL;
git://developer.intra2net.com / libi2ncommon / blobdiff