-
Notifications
You must be signed in to change notification settings - Fork 0
/
load_test.py
executable file
·114 lines (90 loc) · 3.58 KB
/
load_test.py
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
#!/usr/bin/python3
import requests
import argparse
import docker
import os
from time import sleep
from docker import DockerClient
from docker.models.containers import Container
def start_container(client: DockerClient, image: str, basedir: str) -> Container:
volumes = {
basedir: {
'bind': '/app',
},
'/tmp/data/dump': {
'bind': '/data/dump',
},
'/tmp/data/dconf': {
'bind': '/data/dconf',
},
'/lib/modules': {
'bind': '/lib/modules',
'mode': 'ro'
},
}
tmpfs = {'/run': ''}
container = client.containers.run(image, command='/bin/sleep infinity',
init=True, tty=True,
volumes=volumes, tmpfs=tmpfs, privileged=True, working_dir='/app',
detach=True, remove=True)
return container
def test_software(i: int, url: str):
for _ in range(0, 2):
try:
result = requests.get(url)
if result.status_code != 200:
print(f'''Iteration {i} failed. Application test failed with code {result.status_code}.''')
except Exception:
print('Failed request.')
def main(args):
client: DockerClient = docker.from_env()
print(f'Docker: Starting container...')
container = start_container(client, args.image, args.basedir)
print(f'CRIU: Check...')
exit_code, output = container.exec_run('criu check --all', privileged=True, tty=True)
print(exit_code, output.decode('utf-8'))
if exit_code != 0:
print(f'CRIU: ERROR: Failed kernel checks.')
container.stop()
exit(1)
print(f'Docker: Starting processes...')
exit_code, output = container.exec_run('./02_set_process_id.sh ./03_start_processes.sh', privileged=True, tty=True)
print(exit_code, output.decode('utf-8'))
if exit_code != 0:
print(f'Docker: ERROR: Failed to start processes.')
container.stop()
exit(1)
# wait for the application to start
sleep(2)
# test_software(0, args.url)
for i in range(1, args.loops):
print(f'CRIU: Dumping...')
exit_code, output = container.exec_run('./04_dump_processes.sh', privileged=True, tty=True)
print(exit_code, output.decode('utf-8'))
if exit_code != 0:
print(f'CRIU: ERROR: Failed to dump.')
container.stop()
exit(1)
print(f'Docker: Stopping container...')
container.stop()
print(f'Docker: Starting container...')
container = start_container(client, args.image, args.basedir)
print(f'CRIU: Restoring process...')
exit_code, output = container.exec_run('./05_restore_processes.sh', privileged=True, tty=True)
print(exit_code, output.decode('utf-8'))
if exit_code != 0:
print(f'''CRIU: ERROR: Iteration {i} failed. Couldn't restore.''')
exit(1)
exit_code, output = container.exec_run('ps axufwww', privileged=True, tty=True)
print(exit_code, output.decode('utf-8'))
# test_software(i, args.url)
container.stop()
def parse_arguments():
parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter)
parser.add_argument('-l', '--loops', type=int, help='Loops', default=8)
parser.add_argument('-i', '--image', help='Docker image', default='criu-x11-poc')
parser.add_argument('-b', '--basedir', help='Base directory', default=os.getcwd())
parser.add_argument('-u', '--url', help='App test URL', default='http://localhost:8080/')
return parser.parse_args()
if __name__ == '__main__':
main(parse_arguments())