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
|
import os
import re
import unittest
from androguard.decompiler.decompile import DvClass, DvMethod
from androguard.misc import AnalyzeAPK, AnalyzeDex
test_dir = os.path.dirname(os.path.abspath(__file__))
class DecompilerTest(unittest.TestCase):
def testSimplification(self):
h, d, dx = AnalyzeDex(os.path.join(test_dir, 'data/APK/Test.dex'))
(z,) = d.get_classes()
self.assertIn("return ((23 - p3) | ((p3 + 66) & 26));", z.get_source())
def testArrays(self):
h, d, dx = AnalyzeDex(
os.path.join(test_dir, 'data/APK/FillArrays.dex')
)
(z,) = d.get_classes()
self.assertIn("{20, 30, 40, 50};", z.get_source())
self.assertIn("{1, 2, 3, 4, 5, 999, 10324234};", z.get_source())
self.assertIn("{97, 98, 120, 122, 99};", z.get_source())
self.assertIn("{5, 10, 15, 20};", z.get_source())
# Failed version of char array
self.assertNotIn("{97, 0, 98, 0, 120};", z.get_source())
# Failed version of short array
self.assertNotIn("{5, 0, 10, 0};", z.get_source())
def test_all_decompiler(self):
# Generate test cases for this APK:
a, d, dx = AnalyzeAPK(
os.path.join(test_dir, 'data/APK/hello-world.apk')
)
for c in d[0].get_classes():
test_name = re.sub("[^a-zA-Z0-9_]", "_", str(c.get_name())[1:-1])
# Test the decompilation of a single class
# disable for now, as testing all DvMethods has the same effect as
# testing all DvClasses.
# yield dvclass, c, dx
# Test the decompilation of all single methods in the class
# if methods are in the class
if len(c.get_methods()) == 0:
# But we test on all classes that have no methods.
dvclass(c, dx)
continue
dvmethod(c, dx, False)
# Disable tests for doAST=True for now...
dvmethod(c, dx, True)
def dvmethod(c, dx, doAST=False):
for m in c.get_methods():
mx = dx.get_method(m)
ms = DvMethod(mx)
ms.process(doAST=doAST)
if doAST:
assert ms.get_ast() is not None
assert isinstance(ms.get_ast(), dict)
assert 'body' in ms.get_ast()
else:
assert ms.get_source() is not None
def dvclass(c, dx):
dc = DvClass(c, dx)
dc.process()
assert dc.get_source() is not None
if __name__ == '__main__':
unittest.main()
|
