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
|
#!/usr/bin/env python3
# Copyright 2022 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
import os
import unittest
import r8_disassembly
import test_util
_TEST_DATA_DIR = test_util.TEST_DATA_DIR
class DexDisassemblyTest(unittest.TestCase):
@classmethod
def setUpClass(cls):
with open(os.path.join(_TEST_DATA_DIR, 'R8_Disassembler_Output.txt'),
'r') as f:
cls.class_obj_map, _ = r8_disassembly.Parse(f)
def testParseClassName(self):
"""Test parsing the class names."""
expected_info_list = [
'org.chromium.chrome.browser.customtabs.' +
'CustomTabDelegateFactory$$Lambda$5',
'org.chromium.chrome.browser.app.appmenu.IncognitoMenuItemViewBinder$1',
'com.youtube.elements.fbs.AnimatedVectorType'
]
self.assertEqual(expected_info_list, list(self.class_obj_map))
def testParseMethodList(self):
"""Test parsing the method names for a class."""
expected_info_list_class1 = ['<init>', 'get']
expected_info_list_class2 = []
expected_info_list_class3 = [
'<init>', 'animation', 'frameState', 'progressState'
]
self.assertEqual(expected_info_list_class1, [
method.name for method in
self.class_obj_map['org.chromium.chrome.browser.customtabs.' +
'CustomTabDelegateFactory$$Lambda$5'].methods
])
self.assertEqual(expected_info_list_class2, [
method.name for method in
self.class_obj_map['org.chromium.chrome.browser.app.appmenu.' +
'IncognitoMenuItemViewBinder$1'].methods
])
self.assertEqual(expected_info_list_class3, [
method.name for method in self.
class_obj_map['com.youtube.elements.fbs.AnimatedVectorType'].methods
])
def testParseMethodReturnType(self):
"""Test parsing the return type for each method."""
# Note the return types are obfuscated.
expected_info_list_class1 = ['void', 'java.lang.Object']
expected_info_list_class2 = []
expected_info_list_class3 = ['void', 'bb', 'Va', 'Wa']
self.assertEqual(expected_info_list_class1, [
method.return_type for method in self.class_obj_map[
'org.chromium.chrome.browser.customtabs.CustomTabDelegateFactory' +
'$$Lambda$5'].methods
])
self.assertEqual(expected_info_list_class2, [
method.return_type for method in
self.class_obj_map['org.chromium.chrome.browser.app.appmenu' +
'.IncognitoMenuItemViewBinder$1'].methods
])
self.assertEqual(expected_info_list_class3, [
method.return_type for method in self.
class_obj_map['com.youtube.elements.fbs.AnimatedVectorType'].methods
])
def testParseMethodParamType(self):
"""Test parsing the parameters type for each method."""
# Note the return types are obfuscated.
expected_info_list_class1 = [[], []]
expected_info_list_class2 = []
expected_info_list_class3 = [[], [], ['java.lang.Object', 'int', 'byte[]'],
['java.lang.Object']]
self.assertEqual(expected_info_list_class1, [
method.param_types for method in self.class_obj_map[
'org.chromium.chrome.browser.customtabs.CustomTabDelegateFactory' +
'$$Lambda$5'].methods
])
self.assertEqual(expected_info_list_class2, [
method.param_types for method in
self.class_obj_map['org.chromium.chrome.browser.app.appmenu.' +
'IncognitoMenuItemViewBinder$1'].methods
])
self.assertEqual(expected_info_list_class3, [
method.param_types for method in self.
class_obj_map['com.youtube.elements.fbs.AnimatedVectorType'].methods
])
def testParseMethodBytecode(self):
"""Test parsing a stand alone class."""
# Note the return types are obfuscated.
expected_info = [
'registers: 1, inputs: 1, outputs: 1\n',
'------------------------------------------------------------\n',
'inst# offset instruction arguments\n',
'------------------------------------------------------------\n',
' 0: 0x00: InvokeDirect { v0 } org.chromium.base.' +
'supplier.Supplier$$CC void <init>()\n', ' 1: 0x03: ReturnVoid\n'
]
self.assertEqual(
expected_info, self.class_obj_map[
'org.chromium.chrome.browser.customtabs.' +
'CustomTabDelegateFactory$$Lambda$5'].FindMethodByteCode(
'org.chromium.chrome.browser.customtabs.' +
'CustomTabDelegateFactory$$Lambda$5', '<init>', [], 'void'))
if __name__ == '__main__':
unittest.main()
|
