megenginelite-sys 1.8.2

A safe megenginelite wrapper in Rust
Documentation 
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

#!/usr/bin/env python3
# -*-coding=utf-8-*-

# This tool is used to analyze the file generated by compare_binary_iodump.py.
# parse() can build a dependency tree with those varnodes
# where accuracy shake occurs and show the root varnodes.
# get_varNode()/get_dependence_list()/get_reference_list()/show_src_info()
# are some functions which are used to query dependencies between varnodes.
import argparse
import os


class varNode:
    var_node_dict = {}
    var_node_root_dict = {}

    def __init__(self, id, dependence_list, src_info):
        self.src_info = src_info
        if not id in varNode.var_node_dict.keys():
            self.id = id
            self.dependence_list = []
            self.reference_list = []
        else:
            self = varNode.var_node_dict[id]

        if dependence_list:
            self.vitrual = False
            self.is_root = True
        else:
            self.vitrual = True
            self.is_root = False

        for i in dependence_list:
            if not i in varNode.var_node_dict.keys():
                varNode.var_node_dict[i] = varNode(i, [], "")

            dv = varNode.var_node_dict[i]
            self.dependence_list.append(dv)
            if not dv.vitrual:
                self.is_root = False
            dv.reference_list.append(self)

        for i in self.reference_list:
            i.is_root = False
            varNode.var_node_root_dict.pop[i.id]

        if self.is_root:
            varNode.var_node_root_dict[id] = self

        varNode.var_node_dict[id] = self

    @staticmethod
    def get_varNode(id):
        return varNode.var_node_dict[id]

    def get_dependence_list(self):
        return self.dependence_list

    def get_reference_list(self):
        return self.reference_list

    def show_src_info(self):
        print(self.src_info)


def get_dependence(string, src_info):
    start1 = "id:"
    end1 = ","
    e = 0

    count = string.count(start1)
    dependence_list = []
    for x in range(0, count):
        s = string.find(start1, e)
        e = string.find(end1, s)
        sub_str = string[s:e]
        if x == 0:
            var = sub_str
        else:
            dependence_list.append(sub_str)
    varNode(var, dependence_list, src_info)


def parse(filename):
    with open(filename) as f:
        varNode.var_node_dict.clear()
        varNode.var_node_root_dict.clear()
        line = f.readline()
        s = ["", "", ""]
        idx = 1
        while line:
            if line.find("not equal: ") != -1:
                s[2] = line
                src_info = s[0] + "\n" + s[1] + "\n" + s[2]
                get_dependence(s[0], src_info)
            else:
                if line.find("var={id:") != -1:
                    idx = idx ^ 1
                    s[idx] = ""
                s[idx] = s[idx] + line.strip()
            line = f.readline()

    return varNode.var_node_root_dict


def main():
    parser = argparse.ArgumentParser(
        description=(
            "Analyze the outputs of compare_binary_iodump.py"
            "Should save the outputs of compare_binary_iodump.py"
            "as a file"
        ),
        formatter_class=argparse.ArgumentDefaultsHelpFormatter,
    )
    parser.add_argument(
        "filename", help="file which save the outputs of compare_binary_iodump.py"
    )
    args = parser.parse_args()

    parse(args.filename)

    print("varnode root:")
    for key, value in varNode.var_node_root_dict.items():
        print(key)
        print("detail info:")
        value.show_src_info()


if __name__ == "__main__":
    main()