def test_accessor_set_array():
mem = set_memory_struct(">BBBBB", 0, 0, 0, 0, 0)
s = partial_struct(
dump_struct_layout("struct x { char arr[5] };", "x")["x"])(MEM_BASE)
for i in range(0, len(s.arr)):
s.arr[i] = i + 1
assert mem == b"\x01\x02\x03\x04\x05"
def test_accessor_pointer():
set_memory_struct(">QL", MEM_BASE + 8, 5)
s = partial_struct(
dump_struct_layout("struct x { int *ptr; int x; };",
"x")["x"])(MEM_BASE)
assert s.x == 5
assert s.ptr == Ptr(Scalar(32, "int", True), MEM_BASE + 8)
assert s.ptr.p() == 5
def test_accessor_scalar():
set_memory_struct(">LBBh", 12345678, 5, 8, -4387)
s = partial_struct(
dump_struct_layout("struct x { int n; char a; char b; short sign; };",
"x")["x"])(MEM_BASE)
assert s.n == 12345678
assert s.a == 5
assert s.b == 8
assert s.sign == -4387
def test_accessor_pointer_to_array():
set_memory_struct(">Q3L", MEM_BASE + 8, 0, 1, 2)
s = partial_struct(
dump_struct_layout("struct x { int (*aptr)[3]; };",
"x")["x"])(MEM_BASE)
assert s.aptr == ArrayPtr(MEM_BASE + 8, 3, Scalar(32, "int", True))
assert s.aptr[0] == 0
assert s.aptr[1] == 1
assert s.aptr[2] == 2
def test_accessor_array_scalar():
set_memory_struct(">LHHHH", 1, 0, 1, 2, 3)
s = partial_struct(
dump_struct_layout("struct x { int n; short arr[4]; };",
"x")["x"])(MEM_BASE)
assert s.arr[0] == 0
assert s.arr[1] == 1
assert s.arr[2] == 2
assert s.arr[3] == 3
def test_accessor_set_invalid():
set_memory_struct("")
s = partial_struct(
dump_struct_layout("struct x { struct { int a; } b; int arr[5]; };",
"x")["x"])(MEM_BASE)
with pytest.raises(TypeError):
s.b = 5
with pytest.raises(TypeError):
s.arr = 5
def test_accessor_set_pointer():
set_memory_struct(">QL", 0, 5)
s = partial_struct(
dump_struct_layout("struct x { int *ptr; int x; };",
"x")["x"])(MEM_BASE)
assert s.ptr == Ptr(Scalar(32, "int", True), 0)
s.ptr = MEM_BASE + 8
assert s.ptr[0] == s.x
def test_accessor_array_struct():
set_memory_struct(">QLHBxLHBxLHBx", 0, 3, 2, 1, 30, 20, 10, 300, 200, 100)
s = partial_struct(
dump_struct_layout(
"struct x { long n; struct { int n; short s; char c; } a[3]; };",
"x")["x"])(MEM_BASE)
for i in range(0, 3):
assert s.a[i].n == 3 * 10**i
assert s.a[i].s == 2 * 10**i
assert s.a[i].c == 1 * 10**i
def test_accessor_set_array_buffer():
mem = set_memory_struct(">BBBBB", 0, 0, 0, 0, 0)
s = partial_struct(
dump_struct_layout("struct x { char arr[5] };", "x")["x"])(MEM_BASE)
assert mem == b"\x00" * 5
s.arr = "hello"
assert mem == b"hello"
s.arr = "bye"
assert mem == b"byelo"
with pytest.raises(ValueError):
s.arr = b"123456"
def test_accessor_set_scalar():
set_memory_struct(">Bxh", 0, 0)
s = partial_struct(
dump_struct_layout("struct x { unsigned char a; short sign; };",
"x")["x"])(MEM_BASE)
assert s.a == 0
assert s.sign == 0
s.a = 200 # unsigned
s.sign = -3254 # signed
assert s.a == 200
assert s.sign == -3254
def test_accessor_bitfield():
set_memory_struct(">LHHH", 0xa00b112f, 0x9876, 0x0000, 0x0000)
def swap32(n):
return struct.unpack("<L", struct.pack(">L", n))[0]
s = partial_struct(
dump_struct_layout(
"""
struct x {
unsigned int bf1: 2;
unsigned int bf2: 14;
char z1;
int bf3: 4;
int bf4: 5;
int bf5: 3;
int bf6: 2;
int bf7: 7;
char z2;
// cross-word (since it's packed)
int bf8: 9;
char z3;
int z4: 7;
unsigned int bf9: 2;
char z5;
} __attribute__((packed));
""", "x")["x"])(MEM_BASE)
assert s.bf1 == 0x2
assert s.bf2 == 0x200b
assert s.bf3 == 0x2
# read with p64 since it spans across two 32-bit words
assert s.bf4 == -1 # signed 0b11111
assert s.bf5 == 1
assert s.bf6 == -2 # signed 0b10
assert s.bf7 == 0b0001110
with pytest.raises(NotImplementedError):
# cross-word is not implemented (requires 2 accesses, what a hassle)
s.bf8
def test_accessor_invalid_pointer_deref():
set_memory_struct(">QQQ", 4, 8, 0)
s = partial_struct(
dump_struct_layout(
"struct x { int (*fptr)(void); void *void_ptr; void *nullptr; };",
"x")["x"])(MEM_BASE)
assert s.fptr == Ptr(Function(), 4)
with pytest.raises(TypeError):
# can't deref a function pointer
s.fptr.p()
assert s.void_ptr == Ptr(Void(), 8)
with pytest.raises(TypeError):
# can't deref a void pointer
s.void_ptr.p()
assert s.nullptr == Ptr(Void(), 0)
with pytest.raises(ValueError):
# can't deref a null pointer
s.nullptr.p()
