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
|
import pytest
from shared_vars import *
def test_asm_large_file():
_, _, stderr = run_process(
[CECCOMP, 'asm', '/dev/zero'],
)
assert stderr == '[ERROR]: The input file is greater than 4 MiB!\n'
def test_asm_no_lf():
_, _, stderr = run_process(
[CECCOMP, 'asm', '-'], stdin='1' * 0x400,
)
assert stderr == "[ERROR]: No line break in source file, perhaps it's not a text file?\n"
def test_asm_long_line():
_, _, stderr = run_process(
[CECCOMP, 'asm', '-'], stdin='\n\n' + '1' * 0x400 + '\n',
)
assert stderr == '[ERROR]: Line 3 has more than 384 bytes, perhaps the input is not a text file?\n'
def test_asm_many_lines():
_, _, stderr = run_process(
[CECCOMP, 'asm', '-'], stdin='\n' * 0x4001,
)
assert stderr == "[ERROR]: Found more than 16384 lines of text, perhaps it's not for ceccomp?\n"
def test_asm_0_file():
_, _, stderr = run_process(
[CECCOMP, 'asm', '-'], stdin='\n\n\0\n',
)
assert stderr == "[ERROR]: Found '\\0' at file offset 2, perhaps it's not a text file?\n"
def test_asm_empty_file():
_, _, stderr = run_process(
[CECCOMP, 'asm', '-'], stdin='# \n',
)
assert stderr == '[ERROR]: The input does not contain any valid statement\n'
def test_asm_4096_statements(errns: SimpleNamespace):
exit_code, _, stderr = run_process(
[CECCOMP, 'asm', '-'], stdin='return ALLOW\n\n' * 4096,
)
errns.stderr = stderr
assert exit_code == 0
def test_asm_4097_statements():
_, _, stderr = run_process(
[CECCOMP, 'asm', '-'], stdin='return ALLOW\n' * 4097,
)
assert stderr == '[ERROR]: Input file has more than 4096 statements!\n'
def test_disasm_large_file():
_, _, stderr = run_process(
[CECCOMP, 'disasm', '/dev/zero'],
)
assert stderr == '[ERROR]: The input is larger than 4096 filters! Perhaps inputting a wrong file?\n'
EDGE_IDS = sorted([p.stem[1:] for p in ERR_CASE_DIR.glob('e*')])
@pytest.mark.parametrize('edgeid', EDGE_IDS)
def test_edge_cases(errns: SimpleNamespace, edgeid: str):
chunk_file = ERR_CASE_DIR / f'e{edgeid}'
with chunk_file.open() as f:
blob = f.read()
cli_idx = blob.find('CLI')
in_idx = blob.find('STDIN')
out_idx = blob.find('STDOUT')
err_idx = blob.find('STDERR')
assert cli_idx != -1 and in_idx != -1 and out_idx != -1
cli = blob[cli_idx + 4 : in_idx].strip().split()
stdin = blob[in_idx + 6 : out_idx]
if err_idx == -1:
stdout, stderr = blob[out_idx + 7:], None
else:
stdout, stderr = blob[out_idx + 7 : err_idx], blob[err_idx + 7:]
is_disasm = cli[0] == 'disasm'
if is_disasm:
stdin = bytes.fromhex(stdin)
_, real_out, real_err = run_process(
[CECCOMP, *cli], stdin=stdin, is_binary=is_disasm,
)
errns.stderr = real_err
if is_disasm:
assert real_out.decode() == stdout
else:
assert real_out == stdout
if stderr:
assert real_err == stderr
|
