cellar/include/subprocess.hpp
2017-03-28 12:41:25 -07:00

171 lines
4.4 KiB
C++

//
// subprocess C++ library - https://github.com/tsaarni/cpp-subprocess
//
// The MIT License (MIT)
//
// Copyright (c) 2015 Tero Saarni
//
#pragma once
#include <string>
#include <vector>
#include <iostream>
#include <ext/stdio_filebuf.h>
#include <cstdio>
#include <system_error>
#include <unistd.h>
#include <sys/types.h>
#include <sys/wait.h>
namespace subprocess
{
class popen
{
public:
popen(const std::string& cmd, std::vector<std::string> argv)
: in_filebuf(nullptr), out_filebuf(nullptr), err_filebuf(nullptr), in_stream(nullptr), out_stream(nullptr), err_stream(nullptr)
{
if (pipe(in_pipe) == -1 ||
pipe(out_pipe) == -1 ||
pipe(err_pipe) == -1 )
{
throw std::system_error(errno, std::system_category());
}
run(cmd, argv);
}
popen(const std::string& cmd, std::vector<std::string> argv, std::ostream& pipe_stdout)
: in_filebuf(nullptr), out_filebuf(nullptr), err_filebuf(nullptr), in_stream(nullptr), out_stream(nullptr), err_stream(nullptr)
{
auto filebuf = dynamic_cast<__gnu_cxx::stdio_filebuf<char>*>(pipe_stdout.rdbuf());
out_pipe[READ] = -1;
out_pipe[WRITE] = filebuf->fd();
if (pipe(in_pipe) == -1 ||
pipe(err_pipe) == -1 )
{
throw std::system_error(errno, std::system_category());
}
run(cmd, argv);
}
~popen()
{
delete in_filebuf;
delete in_stream;
if (out_filebuf != nullptr) delete out_filebuf;
if (out_stream != nullptr) delete out_stream;
delete err_filebuf;
delete err_stream;
}
std::ostream& stdin() { return *in_stream; };
std::istream& stdout()
{
if (out_stream == nullptr) throw std::system_error(EBADF, std::system_category());
return *out_stream;
};
std::istream& stderr() { return *err_stream; };
int wait()
{
int status = 0;
waitpid(pid, &status, 0);
return WEXITSTATUS(status);
};
void close()
{
in_filebuf->close();
}
private:
enum ends_of_pipe { READ = 0, WRITE = 1 };
struct raii_char_str
{
raii_char_str(std::string s) : buf(s.c_str(), s.c_str() + s.size() + 1) { };
operator char*() const { return &buf[0]; };
mutable std::vector<char> buf;
};
void run(const std::string& cmd, std::vector<std::string> argv)
{
argv.insert(argv.begin(), cmd);
pid = ::fork();
if (pid == 0) child(argv);
::close(in_pipe[READ]);
::close(out_pipe[WRITE]);
::close(err_pipe[WRITE]);
in_filebuf = new __gnu_cxx::stdio_filebuf<char>(in_pipe[WRITE], std::ios_base::out, 1);
in_stream = new std::ostream(in_filebuf);
if (out_pipe[READ] != -1)
{
out_filebuf = new __gnu_cxx::stdio_filebuf<char>(out_pipe[READ], std::ios_base::in, 1);
out_stream = new std::istream(out_filebuf);
}
err_filebuf = new __gnu_cxx::stdio_filebuf<char>(err_pipe[READ], std::ios_base::in, 1);
err_stream = new std::istream(err_filebuf);
}
void child(const std::vector<std::string>& argv)
{
if (dup2(in_pipe[READ], STDIN_FILENO) == -1 ||
dup2(out_pipe[WRITE], STDOUT_FILENO) == -1 ||
dup2(err_pipe[WRITE], STDERR_FILENO) == -1 )
{
std::perror("subprocess: dup2() failed");
return;
}
::close(in_pipe[READ]);
::close(in_pipe[WRITE]);
if (out_pipe[READ] != -1) ::close(out_pipe[READ]);
::close(out_pipe[WRITE]);
::close(err_pipe[READ]);
::close(err_pipe[WRITE]);
std::vector<raii_char_str> real_args(argv.begin(), argv.end());
std::vector<char*> cargs(real_args.begin(), real_args.end());
cargs.push_back(nullptr);
if (execvp(cargs[0], &cargs[0]) == -1)
{
std::perror("subprocess: execvp() failed");
return;
}
}
pid_t pid;
int in_pipe[2];
int out_pipe[2];
int err_pipe[2];
__gnu_cxx::stdio_filebuf<char>* in_filebuf;
__gnu_cxx::stdio_filebuf<char>* out_filebuf;
__gnu_cxx::stdio_filebuf<char>* err_filebuf;
std::ostream* in_stream;
std::istream* out_stream;
std::istream* err_stream;
};
} // namespace: subprocess