ida2py/tests/basic_test.py at main · junron/ida2py · GitHub

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
import idapro
import os
import sys

cur_dir = os.path.dirname(__file__)

sys.path.append(f"{cur_dir}/..")
idapro.open_database(f"{cur_dir}/binaries/basic_test", True)

try:
    import ida2py

    ida2py.hook(globals())

    print("arr_ptr", arr_ptr)

    print("names", names)
    assert type(names) is ida2py.PointerWrapper
    assert names.pyval() == b"hell"

    # Variable types will update when set in ida`
    import idc
    idc.SetType(names.address, "char * names[4];")
    print("names2", names)
    assert type(names) is ida2py.ArrayWrapper
    assert len(names) == 4
    assert names[0].pyval() == b"hell"
    assert names.pyval() == [b'hell', b'ow', b'world', b'dl']
    assert names[1][0] == ord('o')
    assert names[-1].pyval() == b"dl"

    array = (uint * 5) @ arr.address
    print("array", array)
    assert type(array) is ida2py.ArrayWrapper
    assert type(array[0]) is ida2py.IntWrapper
    assert array[0] == 1
    assert array[1] + 3 == 5
    assert array[array[0]] == 2, array[array[0]]


    idc.SetType(arr_ptr.address, "int* arr_ptr;")
    assert arr_ptr[0] == 1
    assert arr_ptr[1] == 2


    arr_ptr2 = (ulong * 2).ptr() @ arr_ptr.address

    print(arr_ptr2)
    assert arr_ptr2.pyval() == [8589934593, 17179869187]

    idc.parse_decls("""
    struct Node{
        unsigned long val;
        Node* left;
        Node* right;
    };
    """, 0)
    idc.SetType(n1.address, "Node")

    print("n1", n1)
    assert n1.val == 90
    assert n1.pyval() == {'val': 0x5a, 'left': {'val': 0x56, 'left': {'val': 0x1, 'left': None, 'right': None}, 'right': None}, 'right': {'val': 0x41, 'left': None, 'right': None}}

    # Autocomplete tests
    print(dir(_ida))
    assert all(x in dir(_ida) for x in ["n1", "names", "arr_ptr", "main"])
    # TODO: Will be fixed in next version with type detection
    # assert "arr" in dir(_ida)
    def build_hints():
        assert _ida.n1.__class__.__name__ == "Node", _ida.n1.__class__.__name__
    build_hints()
    _ida.n1
    _ida.n1
    # Test adaptive ordering
    assert dir(_ida)[0] == "n1", dir(_ida)
finally:
    idapro.close_database(False)