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
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
|
/*
* Copyright © Canonical Ltd.
*
* This program is free software: you can redistribute it and/or modify it
* under the terms of the GNU General Public License version 2 or 3,
* as published by the Free Software Foundation.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#ifndef MIR_TEST_FRAMEWORK_PROCESS_H_
#define MIR_TEST_FRAMEWORK_PROCESS_H_
#include <chrono>
#include <cstdlib>
#include <functional>
#include <iosfwd>
#include <memory>
#include <stdexcept>
#include <unistd.h>
namespace mir_test_framework
{
enum class TerminationReason
{
unknown,
child_terminated_normally,
child_terminated_by_signal,
child_terminated_with_core_dump,
child_stopped_by_signal,
child_resumed_by_signal
};
// Aggregated results of running a process to completion
struct Result
{
Result();
// Did the process exit without error?
bool succeeded() const;
// Was the process terminated by a signal?
bool signalled() const;
// Was the process terminated normally?
bool exited_normally() const;
TerminationReason reason;
int exit_code;
int signal;
};
// Posix process control class.
class Process
{
public:
// Construct a process with the supplied pid
Process(pid_t pid);
// Destroy the process cleanly, by terminating it and waiting for
// the pid.
~Process();
// Wait for the process to terminate, and return the results.
Result wait_for_termination(const std::chrono::milliseconds& timeout = std::chrono::minutes(4));
void kill();
void terminate();
void stop();
void cont();
void detach();
protected:
Process() = delete;
Process(const Process&) = delete;
Process& operator=(const Process&) = delete;
private:
pid_t pid;
bool terminated;
bool detached;
};
// Stream print helper
std::ostream& operator<<(std::ostream& out, const Result& result);
// Fork a child process to run the supplied main function, calling
// the exit function when done.
// Returns the parent process.
template<typename Callable>
std::shared_ptr<Process> fork_and_run_in_a_different_process(
Callable&& main_fn, std::function<int()> exit_fn)
{
pid_t pid = fork();
if (pid < 0)
{
throw std::runtime_error("Failed to fork process");
}
if (pid == 0)
{
main_fn();
exit(exit_fn());
}
return std::shared_ptr<Process>(new Process(pid));
}
}
#endif // MIR_TEST_FRAMEWORK_PROCESS_H_
|
