#!/usr/bin/env vpython3 # # Copyright 2017 The Chromium Authors # Use of this source code is governed by a BSD-style license that can be # found in the LICENSE file. """Tests for java_deobfuscate.""" import argparse import os import subprocess import sys import tempfile import unittest # Set by command-line argument. _JAVA_DEOBFUSCATE_PATH = None LINE_PREFIXES = [ '', # logcat -v threadtime '09-08 14:38:35.535 18029 18084 E qcom_sensors_hal: ', # logcat 'W/GCM (15158): ', 'W/GCM ( 158): ', ] TEST_MAP = """\ this.was.Deobfuscated -> FOO: int[] mFontFamily -> a 1:3:void someMethod(int,android.os.Bundle):65:67 -> bar never.Deobfuscated -> NOTFOO: int[] mFontFamily -> a 1:3:void someMethod(int,android.os.Bundle):65:67 -> bar """ TEST_DATA = [ '', 'FOO', 'FOO.bar', 'Here is a FOO', 'Here is a class FOO', 'Here is a class FOO baz', 'Here is a "FOO" baz', 'Here is a type "FOO" baz', 'Here is a "FOO.bar" baz', 'SomeError: SomeFrameworkClass in isTestClass for FOO', 'Here is a FOO.bar', 'Here is a FOO.bar baz', 'END FOO#bar', 'new-instance 3810 (LSome/Framework/Class;) in LFOO;', 'FOO: Error message', 'Caused by: FOO: Error message', '\tat FOO.bar(PG:1)', '\t at\t FOO.bar\t (\t PG:\t 1\t )', '0xfff \t( \tPG:\t 1 \t)\tFOO.bar', ( 'Unable to start activity ComponentInfo{garbage.in/here.test}:' ' java.lang.NullPointerException: Attempt to invoke interface method' ' \'void FOO.bar(int,android.os.Bundle)\' on a null object reference' ), ( 'Caused by: java.lang.NullPointerException: Attempt to read from field' ' \'int[] FOO.a\' on a null object reference' ), 'java.lang.VerifyError: FOO', ( 'java.lang.NoSuchFieldError: No instance field a of type ' 'Ljava/lang/Class; in class LFOO;' ), 'NOTFOO: Object of type FOO was not destroyed...', ] EXPECTED_OUTPUT = [ '', 'this.was.Deobfuscated', 'this.was.Deobfuscated.someMethod', 'Here is a FOO', 'Here is a class this.was.Deobfuscated', 'Here is a class FOO baz', 'Here is a "FOO" baz', 'Here is a type "this.was.Deobfuscated" baz', 'Here is a "this.was.Deobfuscated.someMethod" baz', 'SomeError: SomeFrameworkClass in isTestClass for this.was.Deobfuscated', 'Here is a this.was.Deobfuscated.someMethod', 'Here is a FOO.bar baz', 'END this.was.Deobfuscated#someMethod', 'new-instance 3810 (LSome/Framework/Class;) in Lthis/was/Deobfuscated;', 'this.was.Deobfuscated: Error message', 'Caused by: this.was.Deobfuscated: Error message', '\tat this.was.Deobfuscated.someMethod(Deobfuscated.java:65)', ( '\t at\t this.was.Deobfuscated.someMethod\t ' '(\t Deobfuscated.java:\t 65\t )' ), '0xfff \t( \tDeobfuscated.java:\t 65 \t)\tthis.was.Deobfuscated.someMethod', ( 'Unable to start activity ComponentInfo{garbage.in/here.test}:' ' java.lang.NullPointerException: Attempt to invoke interface method' ' \'void this.was.Deobfuscated.someMethod(int,android.os.Bundle)\' on a' ' null object reference' ), ( 'Caused by: java.lang.NullPointerException: Attempt to read from field' ' \'int[] this.was.Deobfuscated.mFontFamily\' on a null object reference' ), 'java.lang.VerifyError: this.was.Deobfuscated', ( 'java.lang.NoSuchFieldError: No instance field mFontFamily of type ' 'Ljava/lang/Class; in class Lthis/was/Deobfuscated;' ), 'NOTFOO: Object of type this.was.Deobfuscated was not destroyed...', ] TEST_DATA = [s + '\n' for s in TEST_DATA] EXPECTED_OUTPUT = [s + '\n' for s in EXPECTED_OUTPUT] class JavaDeobfuscateTest(unittest.TestCase): def __init__(self, *args, **kwargs): super().__init__(*args, **kwargs) self._map_file = None def setUp(self): self._map_file = tempfile.NamedTemporaryFile() self._map_file.write(TEST_MAP.encode('utf-8')) self._map_file.flush() def tearDown(self): if self._map_file: self._map_file.close() def _testImpl( self, input_lines=None, expected_output_lines=None, prefix='' ): self.assertTrue(bool(input_lines) == bool(expected_output_lines)) if not input_lines: input_lines = [prefix + x for x in TEST_DATA] if not expected_output_lines: expected_output_lines = [prefix + x for x in EXPECTED_OUTPUT] cmd = [_JAVA_DEOBFUSCATE_PATH, self._map_file.name] proc = subprocess.Popen( cmd, stdin=subprocess.PIPE, stdout=subprocess.PIPE ) proc_output, _ = proc.communicate(''.join(input_lines).encode()) actual_output_lines = proc_output.decode().splitlines(True) for actual, expected in zip(actual_output_lines, expected_output_lines): self.assertTrue( actual == expected or actual.replace('bar', 'someMethod') == expected, msg=''.join( [ 'Deobfuscation failed.\n', ' actual: %s' % actual, ' expected: %s' % expected, ] ), ) def testNoPrefix(self): self._testImpl(prefix='') def testThreadtimePrefix(self): self._testImpl( prefix='09-08 14:38:35.535 18029 18084 E qcom_sensors_hal: ' ) def testStandardPrefix(self): self._testImpl(prefix='W/GCM (15158): ') def testStandardPrefixWithPadding(self): self._testImpl(prefix='W/GCM ( 158): ') @unittest.skip('causes java_deobfuscate to hang, see crbug.com/876539') def testIndefiniteHang(self): # Test for crbug.com/876539. self._testImpl( input_lines=[ 'VFY: unable to resolve virtual method 2: LFOO;' + '.onDescendantInvalidated ' + '(Landroid/view/View;Landroid/view/View;)V', ], expected_output_lines=[ 'VFY: unable to resolve virtual method 2: Lthis.was.Deobfuscated;' + '.onDescendantInvalidated ' + '(Landroid/view/View;Landroid/view/View;)V', ], ) if __name__ == '__main__': parser = argparse.ArgumentParser() parser.add_argument( '--java-deobfuscate-path', type=os.path.realpath, required=True ) known_args, unittest_args = parser.parse_known_args() _JAVA_DEOBFUSCATE_PATH = known_args.java_deobfuscate_path unittest_args = [sys.argv[0]] + unittest_args unittest.main(argv=unittest_args)