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
|
#!/usr/bin/env python3
import os
import random
import subprocess as sp
import time
from runtest import TestBase
TDIR = 'xxx'
class TestCase(TestBase):
def __init__(self):
TestBase.__init__(self, 'openclose', serial=True, result="""
# DURATION TID FUNCTION
1.088 us [18343] | __monstartup();
0.640 us [18343] | __cxa_atexit();
[18343] | main() {
[18343] | fopen() {
86.790 us [18343] | sys_open();
89.018 us [18343] | } /* fopen */
[18343] | fclose() {
10.781 us [18343] | sys_close();
37.325 us [18343] | } /* fclose */
128.387 us [18343] | } /* main */
""")
self.recv_p = None
def prerun(self, timeout):
if os.geteuid() != 0:
return TestBase.TEST_SKIP
if os.path.exists('/.dockerenv'):
return TestBase.TEST_SKIP
self.gen_port()
self.subcmd = 'recv'
self.option = '-d %s --port %s' % (TDIR, self.port)
self.exearg = ''
recv_cmd = self.runcmd()
self.pr_debug('prerun command: ' + recv_cmd)
self.recv_p = sp.Popen(recv_cmd.split())
time.sleep(0.1)
self.dirname = 'dir-%d' % random.randint(100000, 999999)
self.subcmd = 'record'
self.option = '--host %s --port %s -d %s -k' % ('localhost', self.port, self.dirname)
self.exearg = 't-' + self.name
record_cmd = self.runcmd()
self.pr_debug('prerun command: ' + record_cmd)
sp.call(record_cmd.split())
return TestBase.TEST_SUCCESS
def setup(self):
self.subcmd = 'replay'
self.option = '-d %s' % os.path.join(TDIR, self.dirname)
def postrun(self, ret):
self.recv_p.terminate()
return ret
def fixup(self, cflags, result):
uname = os.uname()
# Later version changed syscall routines
major, minor, release = uname[2].split('.', 2)
if uname[0] == 'Linux' and uname[4] == 'x86_64':
if int(major) == 6 and int(minor) >= 9:
result = result.replace('sys_open', 'x64_sys_call')
result = result.replace('sys_close', 'x64_sys_call')
elif int(major) >= 5 or (int(major) == 4 and int(minor) >= 17):
result = result.replace('sys_', '__x64_sys_')
if uname[0] == 'Linux' and uname[4] == 'aarch64' and \
int(major) >= 5 or (int(major) == 4 and int(minor) >= 19):
result = result.replace('sys_', '__arm64_sys_')
return result.replace(' sys_open', ' sys_openat')
|
