2022-07-08 09:28:37 +00:00
|
|
|
import sys
|
|
|
|
|
|
|
|
sys.path.append(".")
|
|
|
|
|
|
|
|
from androguard.core import dex
|
|
|
|
|
2017-12-17 21:46:54 +00:00
|
|
|
from binascii import hexlify
|
|
|
|
import parse_dex
|
|
|
|
import unittest
|
|
|
|
from difflib import Differ
|
|
|
|
|
|
|
|
|
|
|
|
class TestDexCodeParsing(unittest.TestCase):
|
|
|
|
|
|
|
|
def testcode(self):
|
2017-12-17 23:14:58 +00:00
|
|
|
skipped_methods = []
|
2017-12-17 21:46:54 +00:00
|
|
|
|
|
|
|
fname = "examples/android/TestsAndroguard/bin/classes.dex"
|
|
|
|
|
|
|
|
parsed = parse_dex.read_dex(fname)
|
|
|
|
|
|
|
|
with open(fname, "rb") as f:
|
2022-07-08 09:28:37 +00:00
|
|
|
d = dex.DEX(f.read())
|
2017-12-17 21:46:54 +00:00
|
|
|
|
2019-02-24 22:34:54 +00:00
|
|
|
dif = Differ()
|
|
|
|
|
|
|
|
for m in d.get_methods():
|
|
|
|
if not m.get_code():
|
|
|
|
continue
|
|
|
|
|
|
|
|
if m.get_method_idx() in skipped_methods:
|
|
|
|
continue
|
|
|
|
|
|
|
|
code = hexlify(m.get_code().get_raw())
|
|
|
|
self.assertEqual(parsed.methods[m.get_method_idx()],
|
|
|
|
code,
|
|
|
|
"incorrect code for "
|
|
|
|
"[{}]: {} --> {}:\n"
|
|
|
|
"{}\ntries_size: {}, insns_size: {}\nSHOULD BE {}\n{}\n{}".format(m.get_method_idx(),
|
|
|
|
m.get_class_name(),
|
|
|
|
m.get_name(),
|
|
|
|
"".join(dif.compare(parsed.methods[m.get_method_idx()],
|
|
|
|
code)),
|
|
|
|
m.get_code().tries_size,
|
|
|
|
m.get_code().insns_size,
|
|
|
|
hexlify(m.get_code().get_raw()),
|
|
|
|
parsed.methods[m.get_method_idx()],
|
|
|
|
hexlify(m.get_code().code.get_raw())))
|
|
|
|
|
|
|
|
def testClassManager(self):
|
|
|
|
"""Test if the classmanager has the same items"""
|
|
|
|
|
2019-05-09 19:20:57 +00:00
|
|
|
from androguard.core.mutf8 import decode
|
2019-02-24 22:34:54 +00:00
|
|
|
|
|
|
|
fname = "examples/android/TestsAndroguard/bin/classes.dex"
|
|
|
|
|
|
|
|
parsed = parse_dex.read_dex(fname)
|
|
|
|
|
|
|
|
with open(fname, "rb") as f:
|
2022-07-08 09:28:37 +00:00
|
|
|
d = dex.DEX(f.read())
|
2019-02-24 22:34:54 +00:00
|
|
|
|
|
|
|
cm = d.get_class_manager()
|
|
|
|
|
|
|
|
self.assertFalse(cm.get_odex_format())
|
|
|
|
|
|
|
|
ERR_STR = 'AG:IS: invalid string'
|
|
|
|
|
|
|
|
## Testing Strings...
|
|
|
|
for idx in range(parsed.string_ids_size):
|
|
|
|
self.assertNotEqual(cm.get_string(idx), ERR_STR)
|
|
|
|
self.assertNotEqual(cm.get_raw_string(idx), ERR_STR)
|
2019-05-01 19:57:43 +00:00
|
|
|
self.assertEqual(cm.get_raw_string(idx), decode(parsed.str_raw[idx]))
|
2019-02-24 22:34:54 +00:00
|
|
|
|
|
|
|
self.assertEqual(cm.get_string(parsed.string_ids_size), ERR_STR)
|
|
|
|
self.assertEqual(cm.get_raw_string(parsed.string_ids_size), ERR_STR)
|
|
|
|
|
|
|
|
self.assertEqual(cm.get_string(parsed.string_ids_size + 100), ERR_STR)
|
|
|
|
self.assertEqual(cm.get_raw_string(parsed.string_ids_size + 100), ERR_STR)
|
|
|
|
|
2017-12-17 21:46:54 +00:00
|
|
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
unittest.main()
|