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
|
import unittest
import fstracer
class simple_tests(unittest.TestCase):
def test_open_unexists(self):
self.assertEqual(fstracer.getfsevents('/bin/cat',
['/bin/cat','f1','f2']),
[['open', 'f1'], ['open', 'f2']])
def test_open_exists(self):
self.assertEqual(fstracer.getfsevents('/bin/cat',
['/bin/cat','/etc/passwd']),
[['open', '/etc/passwd']])
def test_open_many(self):
filesnum=200
self.assertEqual(fstracer.getfsevents('/bin/cat',
['/bin/cat']+map(lambda x: 'file'+str(x),range(0,filesnum))),
map(lambda x: ['open','file'+str(x)],range(0,filesnum)))
def test_parralel(self):
filesnum=200
procnum=6
# create command
command=""
for p in xrange(0,procnum):
command+="/bin/cat "
for f in xrange(0,filesnum):
command+="file_%d_%d " % (p,f)
command+="& "
command+=" >/dev/null 2>&1"
command+=" "+"A"*65536
resultarray=fstracer.getfsevents('/bin/sh', ['/bin/sh','-c',command])
self.assertTrue(resultarray.count(['execve', '/bin/cat'])==procnum)
print resultarray
for p in xrange(0,procnum):
for f in xrange(0,filesnum):
self.assertTrue(resultarray.count(['open', 'file_%d_%d' % (p,f)])==1)
#self.assertEqual(resultarray,
# map(lambda x: ['open','file'+str(x)],range(0,filesnum)))
if __name__ == '__main__':
unittest.main()
|
GitWeb
