1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
|
#ifndef TINY_PROCESS_LIBRARY_HPP_
#define TINY_PROCESS_LIBRARY_HPP_
#include <string>
#include <functional>
#include <vector>
#include <mutex>
#include <thread>
#include <memory>
#ifndef _WIN32
#include <sys/wait.h>
#endif
namespace TinyProcessLib {
///Platform independent class for creating processes
class Process {
public:
#ifdef _WIN32
typedef unsigned long id_type; //Process id type
typedef void *fd_type; //File descriptor type
#ifdef UNICODE
typedef std::wstring string_type;
#else
typedef std::string string_type;
#endif
#else
typedef pid_t id_type;
typedef int fd_type;
typedef std::string string_type;
#endif
private:
class Data {
public:
Data() noexcept;
id_type id;
#ifdef _WIN32
void *handle;
#endif
};
public:
///Note on Windows: it seems not possible to specify which pipes to redirect.
///Thus, at the moment, if read_stdout==nullptr, read_stderr==nullptr and open_stdin==false,
///the stdout, stderr and stdin are sent to the parent process instead.
Process(const string_type &command, const string_type &path=string_type(),
std::function<void(const char *bytes, size_t n)> read_stdout=nullptr,
std::function<void(const char *bytes, size_t n)> read_stderr=nullptr,
bool open_stdin=false,
size_t buffer_size=131072) noexcept;
#ifndef _WIN32
/// Supported on Unix-like systems only.
Process(std::function<void()> function,
std::function<void(const char *bytes, size_t n)> read_stdout=nullptr,
std::function<void(const char *bytes, size_t n)> read_stderr=nullptr,
bool open_stdin=false,
size_t buffer_size=131072) noexcept;
#endif
~Process() noexcept;
///Get the process id of the started process.
id_type get_id() const noexcept;
///Wait until process is finished, and return exit status.
int get_exit_status() noexcept;
///If process is finished, returns true and sets the exit status. Returns false otherwise.
bool try_get_exit_status(int &exit_status) noexcept;
///Write to stdin.
bool write(const char *bytes, size_t n);
///Write to stdin. Convenience function using write(const char *, size_t).
bool write(const std::string &data);
///Close stdin. If the process takes parameters from stdin, use this to notify that all parameters have been sent.
void close_stdin() noexcept;
///Kill the process. force=true is only supported on Unix-like systems.
void kill(bool force=false) noexcept;
///Kill a given process id. Use kill(bool force) instead if possible. force=true is only supported on Unix-like systems.
static void kill(id_type id, bool force=false) noexcept;
private:
Data data;
bool closed;
std::mutex close_mutex;
std::function<void(const char* bytes, size_t n)> read_stdout;
std::function<void(const char* bytes, size_t n)> read_stderr;
std::thread stdout_thread, stderr_thread;
bool open_stdin;
std::mutex stdin_mutex;
size_t buffer_size;
std::unique_ptr<fd_type> stdout_fd, stderr_fd, stdin_fd;
id_type open(const string_type &command, const string_type &path) noexcept;
#ifndef _WIN32
id_type open(std::function<void()> function) noexcept;
#endif
void async_read() noexcept;
void close_fds() noexcept;
};
} // TinyProsessLib
#endif // TINY_PROCESS_LIBRARY_HPP_
|
