/
api_reboot.c
127 lines (105 loc) · 3.23 KB
/
api_reboot.c
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
121
122
123
124
125
126
127
/* liblxcapi
*
* Copyright © 2017 Christian Brauner <christian.brauner@ubuntu.com>.
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License version 2, 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, write to the Free Software Foundation, Inc.,
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
*/
#include <alloca.h>
#include <stdio.h>
#include <sched.h>
#include <unistd.h>
#include <signal.h>
#include <errno.h>
#include <string.h>
#include <sys/reboot.h>
#include <sys/types.h>
#include <sys/wait.h>
#include "lxc/lxccontainer.h"
#include "lxctest.h"
int main(int argc, char *argv[])
{
int i;
struct lxc_container *c;
int ret = EXIT_FAILURE;
/* Test that the reboot() API function properly waits for containers to
* restart.
*/
c = lxc_container_new("reboot", NULL);
if (!c) {
lxc_error("%s", "Failed to create container \"reboot\"");
exit(ret);
}
if (c->is_defined(c)) {
lxc_error("%s\n", "Container \"reboot\" is defined");
goto on_error_put;
}
if (!c->createl(c, "busybox", NULL, NULL, 0, NULL)) {
lxc_error("%s\n", "Failed to create busybox container \"reboot\"");
goto on_error_put;
}
if (!c->is_defined(c)) {
lxc_error("%s\n", "Container \"reboot\" is not defined");
goto on_error_put;
}
c->clear_config(c);
if (!c->load_config(c, NULL)) {
lxc_error("%s\n", "Failed to load config for container \"reboot\"");
goto on_error_stop;
}
if (!c->want_daemonize(c, true)) {
lxc_error("%s\n", "Failed to mark container \"reboot\" daemonized");
goto on_error_stop;
}
if (!c->startl(c, 0, NULL)) {
lxc_error("%s\n", "Failed to start container \"reboot\" daemonized");
goto on_error_stop;
}
/* reboot 10 times */
for (i = 0; i < 10; i++) {
/* Give the init system some time to setup it's signal handlers
* otherwise we will hang indefinitely.
*/
sleep(5);
if (!c->reboot2(c, 60)) {
lxc_error("%s\n", "Failed to reboot container \"reboot\"");
goto on_error_stop;
}
if (!c->is_running(c)) {
lxc_error("%s\n", "Failed to reboot container \"reboot\"");
goto on_error_stop;
}
lxc_debug("%s\n", "Container \"reboot\" rebooted successfully");
}
/* Give the init system some time to setup it's signal handlers
* otherwise we will hang indefinitely.
*/
sleep(5);
/* Test non-blocking reboot2() */
if (!c->reboot2(c, 0)) {
lxc_error("%s\n", "Failed to request non-blocking reboot of container \"reboot\"");
goto on_error_stop;
}
lxc_debug("%s\n", "Non-blocking reboot of container \"reboot\" succeeded");
ret = EXIT_SUCCESS;
on_error_stop:
if (c->is_running(c) && !c->stop(c))
lxc_error("%s\n", "Failed to stop container \"reboot\"");
if (!c->destroy(c))
lxc_error("%s\n", "Failed to destroy container \"reboot\"");
on_error_put:
lxc_container_put(c);
if (ret == EXIT_SUCCESS)
lxc_debug("%s\n", "All reboot tests passed");
exit(ret);
}