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
|
import os
import unittest
from androguard.core.analysis.analysis import ExternalMethod
from androguard.misc import AnalyzeAPK
test_dir = os.path.dirname(os.path.abspath(__file__))
class TestCallgraph(unittest.TestCase):
@classmethod
def setUpClass(cls):
test_apk_path = os.path.join(test_dir, 'data/APK/TestActivity.apk')
cls.a, cls.d, cls.dx = AnalyzeAPK(test_apk_path)
def _get_num_nodes(self, callgraph):
num_external = 0
num_internal = 0
for n in callgraph:
if isinstance(n, ExternalMethod):
num_external += 1
else:
num_internal += 1
return callgraph.number_of_nodes(), num_external, num_internal
def testCallgraphTotalNodesEdges(self):
"""test callgraph generated with default parameter values"""
callgraph = self.dx.get_call_graph()
total_nodes, total_external_nodes, total_internal_nodes = (
self._get_num_nodes(callgraph)
)
# ensure total of internal and external nodes equals the total nodes
self.assertEqual(
total_nodes, total_external_nodes + total_internal_nodes
)
# total num nodes
self.assertEqual(total_nodes, 3600)
# num external nodes
self.assertEqual(total_external_nodes, 1000)
# num internal nodes
self.assertEqual(total_internal_nodes, 2600)
# total num edges
self.assertEqual(callgraph.number_of_edges(), 4490)
def testCallgraphFilterClassname(self):
"""test callgraph with classname filter parameter"""
callgraph = self.dx.get_call_graph(classname='Ltests/androguard/*')
total_nodes, total_external_nodes, total_internal_nodes = (
self._get_num_nodes(callgraph)
)
self.assertEqual(total_nodes, 165)
self.assertEqual(total_external_nodes, 41)
self.assertEqual(total_internal_nodes, 124)
self.assertEqual(callgraph.number_of_edges(), 197)
def testCallgraphFilterMethodname(self):
"""test callgraph with methodname filter parameter"""
callgraph = self.dx.get_call_graph(methodname='Test*')
total_nodes, total_external_nodes, total_internal_nodes = (
self._get_num_nodes(callgraph)
)
self.assertEqual(total_nodes, 36)
self.assertEqual(total_external_nodes, 12)
self.assertEqual(total_internal_nodes, 24)
self.assertEqual(callgraph.number_of_edges(), 42)
def testCallgraphFilterDescriptor(self):
"""test callgraph with descriptor filter parameter"""
callgraph = self.dx.get_call_graph(
descriptor=r'\(LTestDefaultPackage;\sI\sI\sLTestDefaultPackage\$TestInnerClass;\)V'
)
total_nodes, total_external_nodes, total_internal_nodes = (
self._get_num_nodes(callgraph)
)
self.assertEqual(total_nodes, 2)
self.assertEqual(total_external_nodes, 0)
self.assertEqual(total_internal_nodes, 2)
self.assertEqual(callgraph.number_of_edges(), 1)
# since this is a small graph, let's check some additional values
# get the source and destination node of the only edge
src_node, dst_node = list(callgraph.edges)[0]
# check source node
self.assertEqual(
src_node.get_class_name(), 'LTestDefaultPackage$TestInnerClass;'
)
self.assertEqual(src_node.get_name(), '<init>')
self.assertEqual(
src_node.get_access_flags_string(), 'synthetic constructor'
)
self.assertEqual(
src_node.get_descriptor(),
'(LTestDefaultPackage; I I LTestDefaultPackage$TestInnerClass;)V',
)
# check dest node
self.assertEqual(
dst_node.get_class_name(), 'LTestDefaultPackage$TestInnerClass;'
)
self.assertEqual(dst_node.get_name(), '<init>')
self.assertEqual(
dst_node.get_access_flags_string(), 'private constructor'
)
self.assertEqual(
dst_node.get_descriptor(), '(LTestDefaultPackage; I I)V'
)
if __name__ == '__main__':
unittest.main()
|
