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
|
/*
* Copyright © 2012-2013 Canonical Ltd.
*
* This program is free software: you can redistribute it and/or modify it
* under the terms of the GNU Lesser General Public License version 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 Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*
* Authored by: Thomas Voß <thomas.voss@canonical.com>
*/
#ifndef FORK_AND_RUN_H_
#define FORK_AND_RUN_H_
#include <cstring>
#include <functional>
#include <stdexcept>
#include <sys/types.h>
#include <signal.h>
#include <unistd.h>
namespace test
{
bool is_child(pid_t pid)
{
return pid == 0;
}
int wait_for_child(int pid)
{
int status;
if (0 > waitpid( pid, &status, 0))
return -1;
if (WIFEXITED(status))
return WEXITSTATUS(status);
else
return -2;
}
int fork_and_run(std::function<bool()> child, std::function<void()> parent)
{
auto pid = fork();
if (pid < 0)
{
throw std::runtime_error(std::string("Could not fork child: ") + std::strerror(errno));
}
if (is_child(pid))
{
int err = child();
_exit(err);
return err ? EXIT_FAILURE : EXIT_SUCCESS;
}
else
{
parent();
int err = wait_for_child(pid);
kill(pid, SIGKILL);
return err ? EXIT_FAILURE : EXIT_SUCCESS;
}
return EXIT_FAILURE;
}
}
#endif // FORK_AND_RUN_H_
|
