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
128
129
130
131
132
133
134
135
136
137
138
|
from declarativeunittest import *
from construct import *
from construct.lib import *
def test_path():
path = Path("path")
x = ~((path.foo * 2 + 3 << 2) % 11)
assert repr(x) == "not ((((path.foo * 2) + 3) << 2) % 11)"
assert str(x) == "not ((((path['foo'] * 2) + 3) << 2) % 11)"
assert x(dict(foo=7)) == False
def test_this():
assert repr(this.x) == "this.x"
assert str(this.x) == "this['x']"
assert repr(this.x == 0) == "(this.x == 0)"
assert str(this.x == 0) == "(this['x'] == 0)"
this_example = Struct(
# straight-forward usage: instead of passing (lambda ctx: ctx["length"]) use this.length
"length" / Int8ub,
"value" / Bytes(this.length),
# an example of nesting: '_' refers to the parent's scope
"nested" / Struct(
"b1" / Int8ub,
"b2" / Int8ub,
"b3" / Computed(this.b1 * this.b2 + this._.length),
),
# and conditions work as expected
"condition" / IfThenElse(
this.nested.b1 > 50,
"c1" / Int32ub,
"c2" / Int8ub,
),
)
common(this_example, b"\x05helloABXXXX", Container(length=5)(value=b'hello')(nested=Container(b1=65)(b2=66)(b3=4295))(condition=1482184792))
def test_this_getitem():
d = Struct(
"num" / Int8ub,
"dup1" / Computed(this.num),
"dup2" / Computed(this["num"]),
Check(this.num == 0),
Check(this.dup1 == 0),
Check(this.dup2 == 0),
)
common(d, b"\x00", Container(num=0, dup1=0, dup2=0), 1)
def test_functions():
assert repr(len_(this.x)) == "len_(this.x)"
assert str(len_(this.x)) == "len_(this['x'])"
assert repr(len_) == "len_"
assert str(len_) == "len_"
example = Struct(
"items" / Byte[2],
Check(len_(this.items) == 2),
Check(sum_(this.items) == 10),
Check(min_(this.items) == 3),
Check(max_(this.items) == 7),
"nega" / Int8sb,
Check(this.nega == -1),
Check(abs_(this.nega) == 1),
)
common(example, b"\x03\x07\xff", Container(items=[3,7], nega=-1), 3)
def test_obj():
assert repr(obj_) == "obj_"
assert repr(obj_ + 1 == 12) == "((obj_ + 1) == 12)"
assert str(obj_) == "obj_"
assert str(obj_ + 1 == 12) == "((obj_ + 1) == 12)"
assert (obj_)(1,{}) == 1
assert (obj_ + 10)(1,{}) == 11
assert (obj_ == 12)(12,{})
assert (obj_ != 12)(13,{})
assert (obj_)(1,[],{}) == 1
assert (obj_ + 10)(1,[],{}) == 11
assert (obj_ == 12)(12,[],{})
assert (obj_ != 12)(13,[],{})
example = Struct(
"items" / RepeatUntil(obj_ == 255, Byte),
)
common(example, b"\x03\x07\xff", Container(items=[3,7,255]))
@xfail(reason="faulty implementation, needs fixing")
def test_list():
assert repr(list_) == "list_"
assert str(list_) == "list_"
assert repr(list_ == [0, 1, 2]) == "(list_ == [0, 1, 2])"
assert str(list_ == [0, 1, 2]) == "(list_ == [0, 1, 2])"
assert repr(list_[-1]) == "list_[-1]"
assert str(list_[-1]) == "list_[-1]"
assert repr(list_[-1] == 0) == "(list_[-1] == 0)"
assert str(list_[-1] == 0) == "(list_[-1] == 0)"
assert repr(list_[-1] + 1) == "(list_[-1] + 1)"
assert str(list_[-1] + 1) == "(list_[-1] + 1)"
# missing str
assert (list_)(1,[],{}) == []
assert (list_[-1])(1,[2,3,4],{}) == 4
# below fail
assert (list_[-1] + 1)(1,[2,3,4],{}) == 5
assert (list_[-1] == 4)(1,[2,3,4],{}) == True
assert (len_(list_))(1,[2,3,4],{}) == 3
assert (len_(list_[:]))(1,[2,3,4],{}) == 3
example = Struct(
"items" / RepeatUntil(list_[-1] == 255, Byte),
)
common(example, b"\x03\x07\xff", Container(items=[3,7,255]))
@xfail(reason="this expression does not support in operator")
def test_this_in_operator():
d = Struct(
"if" / If(this.data in [1,2,3], Const(b"4")),
"not" / If(this.data not in [1,2,3], Const(b"5")),
)
assert d.build(dict(data=1)) == b'4'
assert d.build(dict(data=7)) == b'5'
def test_lambda_in_operator():
d = Struct(
"if" / If(lambda ctx: ctx.data in [1,2,3], Const(b"4")),
"not" / If(lambda ctx: ctx.data not in [1,2,3], Const(b"5")),
)
assert d.build(dict(data=1)) == b'4'
assert d.build(dict(data=7)) == b'5'
def test_this_shift_operator():
d = Struct(
"a" / Byte,
"ls" / Computed(this.a << 1),
"rs" / Computed(this.a >> 1),
)
assert d.parse(b"\x02") == Container(a=2)(ls=4)(rs=1)
|
