Exemple #1
0
def unpack_baidu_2016(file_path, dex_file_base, dex_file_size):
	dex = Dex.Dex(file_path)
	for class_def_item in dex.class_defs.class_def_items:
		class_data_off = class_def_item.class_data_off
		if (class_data_off > dex_file_size) or (class_data_off < 0):
			# parse class_data_item
			static_fields_size_off = 0x0
			static_fields_size, length_static_fields_size = class_data_item.get_static_fields_size(dex_file_base, class_data_off, static_fields_size_off)
			config.log_print("[DexFile] [class_data_item] static_fields_size = %#x" % static_fields_size)
			
			instance_fields_size_off = static_fields_size_off + length_static_fields_size
			instance_fields_size, length_instance_fields_size = class_data_item.get_instance_fields_size(dex_file_base, class_data_off, instance_fields_size_off)
			config.log_print("[DexFile] [class_data_item] instance_fields_size = %#x" % instance_fields_size)
			
			direct_methods_size_off = instance_fields_size_off + length_instance_fields_size
			direct_methods_size, length_direct_methods_size = class_data_item.get_direct_methods_size(dex_file_base, class_data_off, direct_methods_size_off)
			config.log_print("[DexFile] [class_data_item] direct_methods_size = %#x" % direct_methods_size)
			
			virtual_methods_size_off = direct_methods_size_off + length_direct_methods_size
			virtual_methods_size, length_virtual_methods_size = class_data_item.get_virtual_methods_size(dex_file_base, class_data_off, virtual_methods_size_off)
			config.log_print("[DexFile] [class_data_item] virtual_methods_size = %#x" % virtual_methods_size)
			
			static_fields_off = virtual_methods_size_off + length_virtual_methods_size
			static_fields, length_static_fields = class_data_item.get_static_fields(dex_file_base, class_data_off, static_fields_off, static_fields_size)
			for idx in range(static_fields_size):
				config.log_print("[DexFile] [class_data_item] static_fields[%d].field_idx_diff = %#x" % (idx, static_fields[idx][0]))
				config.log_print("[DexFile] [class_data_item] static_fields[%d].access_flags = %0#10x" % (idx, static_fields[idx][1]))
			
			instance_fields_off = static_fields_off + length_static_fields
			instance_fields, length_instance_fields = class_data_item.get_instance_fields(dex_file_base, class_data_off, instance_fields_off, instance_fields_size)
			for idx in range(instance_fields_size):
				config.log_print("[DexFile] [class_data_item] instance_fields[%d].field_idx_diff = %#x" % (idx, instance_fields[idx][0]))
				config.log_print("[DexFile] [class_data_item] instance_fields[%d].access_flags = %0#10x" % (idx, instance_fields[idx][1]))

			direct_methods_off = instance_fields_off + length_instance_fields
			direct_methods, length_direct_methods = class_data_item.get_direct_methods(dex_file_base, class_data_off, direct_methods_off, direct_methods_size)
			for idx in range(direct_methods_size):
				config.log_print("[DexFile] [class_data_item] direct_methods[%d].method_idx_diff = %#x" % (idx, direct_methods[idx][0]))
				config.log_print("[DexFile] [class_data_item] direct_methods[%d].access_flags = %0#10x" % (idx, direct_methods[idx][1]))
				config.log_print("[DexFile] [class_data_item] direct_methods[%d].code_off = %0#10x" % (idx, direct_methods[idx][2]))
			
			virtual_methods_off = direct_methods_off + length_direct_methods
			virtual_methods, length_virtual_methods = class_data_item.get_virtual_methods(dex_file_base, class_data_off, virtual_methods_off, virtual_methods_size)
			for idx in range(virtual_methods_size):
				config.log_print("[DexFile] [class_data_item] virtual_methods[%d].method_idx_diff = %#x" % (idx, virtual_methods[idx][0]))
				config.log_print("[DexFile] [class_data_item] virtual_methods[%d].access_flags = %0#10x" % (idx, virtual_methods[idx][1]))
				config.log_print("[DexFile] [class_data_item] virtual_methods[%d].code_off = %0#10x" % (idx, virtual_methods[idx][2]))
			
			class_data_size = virtual_methods_off + length_virtual_methods
			
			file_path = os.path.join(config.workspace, config.dex_directory, "class_data_item_%0#10x.bin" % (class_data_off if class_data_off > 0 else (0xffffffff + class_data_off)))
			if not os.path.exists(file_path):
				file_format = "binary"
				file_vtl_start_address = (dex_file_base + class_data_off) & 0xffffffff
				file_vtl_end_address = ((dex_file_base + class_data_off) & 0xffffffff) + class_data_size - 0x1
				memory.dump(file_path, file_format, file_vtl_start_address, file_vtl_end_address)
Exemple #2
0
def init_dvm():
    # get the PID
    pid_val = int(execution_state.getVariableService().readValue(
        "$AARCH64::$System::$Memory::$CONTEXTIDR_EL1.PROCID")) & 0xffffffff

    # get the pointer that refers to the DexFile structure
    dex_file_ptr = int(
        execution_state.getRegisterService().getValue("R0")) & 0xffffffff

    # read the "begin_" field
    dex_file_begin_val = int(
        execution_state.getRegisterService().getValue("R1")) & 0xffffffff

    # read the "size_" field
    dex_file_size_val = int(
        execution_state.getRegisterService().getValue("R2")) & 0xffffffff

    # read the "location_" field
    dex_file_location_ptr = int(
        execution_state.getRegisterService().getValue("R3")) & 0xffffffff
    # retrieve the value of the std::string structure
    dex_file_location_string_val = retrieve_string_value(dex_file_location_ptr)

    if config.debug:
        print "[entrypoint] [DexFile] pid = %#x" % pid_val
        print "[entrypoint] [DexFile] begin_ = %0#10x" % dex_file_begin_val
        print "[entrypoint] [DexFile] size_ = %#x" % dex_file_size_val
        print "[entrypoint] [DexFile] location_ = %s" % dex_file_location_string_val

    # the first loaded Dex file must be our target
    if not config.package_filter(dex_file_location_string_val):
        # should not be reached
        assert False

    config.log_print("[DexFile] pid = %#x" % pid_val)
    config.log_print(
        "[DexFile] begin_ = %0#10x, size_ = %#x, location_ = %s" %
        (dex_file_begin_val, dex_file_size_val, dex_file_location_string_val))

    # initialization
    setup(pid_val)

    # dump the in-memory DexFile
    file_path = os.path.join(config.workspace, config.dex_directory,
                             dex_file_location_string_val.split("/")[-1])
    file_format = "binary"
    file_vtl_start_address = dex_file_begin_val
    file_vtl_end_address = dex_file_begin_val + dex_file_size_val - 0x1
    memory.dump(file_path, file_format, file_vtl_start_address,
                file_vtl_end_address)

    # continue the execution of the target application
    execution_state.getExecutionService().resume()
    cleanup()
    return
def dump_dex_file():
	# dump the dex file
	dex_info_list = config.load_dex_info()
	for dex_file_base, dex_file_size, dex_file_name in dex_info_list:
		config.log_print("[JNI_onLoad] begin_ = %0#10x, size_ = %#x, location_ = %s" % (dex_file_base, dex_file_size, dex_file_name))
		file_path = os.path.join(config.workspace, config.dex_directory, dex_file_name)
		file_format = "binary"
		file_vtl_start_address = dex_file_base
		file_vtl_end_address = dex_file_base + dex_file_size - 0x1
		file_path = memory.dump(file_path, file_format, file_vtl_start_address, file_vtl_end_address)
Exemple #4
0
def DexFile():
	# get the pointer that refers to the DexFile structure
	dex_file_ptr = int(execution_state.getRegisterService().getValue("R0")) & 0xffffffff
	
	# read the "begin_" field
	dex_file_begin_val = int(execution_state.getRegisterService().getValue("R1")) & 0xffffffff
		
	# read the "size_" field
	dex_file_size_val = int(execution_state.getRegisterService().getValue("R2")) & 0xffffffff
	
	# read the "location_" field
	dex_file_location_ptr = int(execution_state.getRegisterService().getValue("R3")) & 0xffffffff
	# retrieve the value of the std::string structure
	dex_file_location_string_val = retrieve_string_value(dex_file_location_ptr)
	
	if config.debug:
		print "[DexFile] begin_ = %0#10x" % dex_file_begin_val
		print "[DexFile] size_ = %#x" % dex_file_size_val
		print "[DexFile] location_ = %s" % dex_file_location_string_val
		
	# we only focus on the DexFile whose location is suspicious
	if not config.package_filter(dex_file_location_string_val):
		# continue the execution of the target application
		# execution_state.getExecutionService().resume()
		cleanup()
		return
		
	config.log_print("[DexFile] begin_ = %0#10x, size_ = %#x, location_ = %s" % (dex_file_begin_val, dex_file_size_val, dex_file_location_string_val))
	
	# dump the in-memory DexFile
	file_path = os.path.join(config.workspace, config.dex_directory, dex_file_location_string_val.split("/")[-1])
	file_format = "binary"
	file_vtl_start_address = dex_file_begin_val
	file_vtl_end_address = dex_file_begin_val + dex_file_size_val - 0x1
	memory.dump(file_path, file_format, file_vtl_start_address, file_vtl_end_address)
		
	# continue the execution of the target application
	execution_state.getExecutionService().resume()
	cleanup()
	return
Exemple #5
0
def init_DexFile():
	# get the pointer that refers to the DexFile structure
	dex_file_ptr = int(execution_state.getRegisterService().getValue("R0")) & 0xffffffff
	
	# read the "begin_" field
	dex_file_begin_val = int(execution_state.getRegisterService().getValue("R1")) & 0xffffffff
	if config.debug:
		print "[DexFile] begin_ = %0#10x" % dex_file_begin_val
		
	# read the "size_" field
	dex_file_size_val = int(execution_state.getRegisterService().getValue("R2")) & 0xffffffff
	if config.debug:
		print "[DexFile] size_ = %#x" % dex_file_size_val
	
	# read the "location_" field
	dex_file_location_ptr = int(execution_state.getRegisterService().getValue("R3")) & 0xffffffff
	# retrieve the value of the std::string structure
	dex_file_location_string_val = retrieve_string_value(dex_file_location_ptr)
	if config.debug:
		print "[DexFile] location_ = %s" % dex_file_location_string_val
		
	# if config.package_filter(dex_file_location_string_val) and dex_file_location_string_val.endswith("base.apk"):
	if config.package_filter(dex_file_location_string_val):
		pid_val = int(execution_state.getVariableService().readValue("$AARCH64::$System::$Memory::$CONTEXTIDR_EL1.PROCID")) & 0xffffffff
		if config.debug:
			print "[DexFile] pid = %#x" % pid_val
		config.log_print("[DexFile] pid = %#x" % pid_val)
			
		setup(pid_val)
		
	# we only focus on the DexFile whose location is suspicious
	if not config.package_filter(dex_file_location_string_val):
		# continue the execution of the target application
		execution_state.getExecutionService().resume()
		cleanup()
		return
		
	# # parse the header_item of the dex file
	# header_item.parse_header_item(dex_file_begin_val)
	
	# # calculate the "size_" value from the "map_off" field of the header_item
	# dex_file_size_val_calc = 0x0
	# if config.package_filter(dex_file_location_string_val):
		# map_off = header_item.get_map_off(dex_file_begin_val)
		# map_list_ptr = dex_file_begin_val + map_off
		# map_list_size_val = memory.readMemory32(map_list_ptr + 0x0)
		# dex_file_size_val_calc = map_off + (0x4) + map_list_size_val * (0x2 + 0x2 + 0x4 + 0x4)
		
	# config.log_print("[DexFile] begin_ = %0#10x, size_ = %#x (inferring size = %#x), location_ = %s" % (dex_file_begin_val, dex_file_size_val, dex_file_size_val_calc, dex_file_location_string_val))
	config.log_print("[DexFile] begin_ = %0#10x, size_ = %#x, location_ = %s" % (dex_file_begin_val, dex_file_size_val, dex_file_location_string_val))
	config.save_dex_info(dex_file_begin_val, dex_file_size_val, dex_file_location_string_val.split("/")[-1])
	
	# dump the in-memory DexFile
	file_path = os.path.join(config.workspace, config.dex_directory, dex_file_location_string_val.split("/")[-1])
	file_format = "binary"
	file_vtl_start_address = dex_file_begin_val
	file_vtl_end_address = dex_file_begin_val + dex_file_size_val - 0x1
	file_path = memory.dump(file_path, file_format, file_vtl_start_address, file_vtl_end_address)
	
	# unpack_baidu_2016(file_path, dex_file_begin_val, dex_file_size_val)
		
	# continue the execution of the target application
	execution_state.getExecutionService().resume()
	cleanup()
	return
Exemple #6
0
def LoadClassMembers():
    # at the very beginning, we are required to ensure that the LoadMethod breakpoint is enabled
    for idx in range(
            0,
            execution_state.getBreakpointService().getBreakpointCount()):
        brk_object = execution_state.getBreakpointService().getBreakpoint(idx)
        if (int(brk_object.getAddresses()[0])
                & 0xffffffff) == config.brk_LoadMethod:
            brk_object.enable()

    # get the "dex_file" parameter
    dex_file_param = int(
        execution_state.getRegisterService().getValue("R2")) & 0xffffffff
    dex_file_ptr = dex_file_param

    # read the "location_" field
    dex_file_location_ptr = dex_file_ptr + config.offset_DexFile_location_
    # retrieve the value of the std::string structure
    dex_file_location_string_val = retrieve_string_value(dex_file_location_ptr)
    if config.debug:
        print "[LoadClassMembers] location_ = %s" % dex_file_location_string_val

    # we only focus on the DexFile whose location is suspicious
    if not config.package_filter(dex_file_location_string_val):
        # disable the LoadMethod breakpoint here
        for idx in range(
                0,
                execution_state.getBreakpointService().getBreakpointCount()):
            brk_object = execution_state.getBreakpointService().getBreakpoint(
                idx)
            if (int(brk_object.getAddresses()[0])
                    & 0xffffffff) == config.brk_LoadMethod:
                brk_object.disable()

        # continue the execution of the target application
        execution_state.getExecutionService().resume()
        cleanup()
        return

    # read the "begin_" field
    dex_file_begin_ptr = dex_file_ptr + config.offset_DexFile_begin_
    dex_file_begin_val = memory.readMemory32(dex_file_begin_ptr)
    if config.debug:
        print "[LoadClassMembers] begin_ = %0#10x" % dex_file_begin_val

    # read the "size_" field
    dex_file_size_ptr = dex_file_ptr + config.offset_DexFile_size_
    dex_file_size_val = memory.readMemory32(dex_file_size_ptr)
    if config.debug:
        print "[LoadClassMembers] size_ = %#x" % dex_file_size_val

    # parse the header_item of the dex file
    header_item.parse_header_item(dex_file_begin_val)

    # # calculate the "size_" value from the "map_off" field of the header_item
    # dex_file_size_val_calc = 0x0
    # if config.package_filter(dex_file_location_string_val):
    # map_off = header_item.get_map_off(dex_file_begin_val)
    # map_list_ptr = dex_file_begin_val + map_off
    # map_list_size_val = memory.readMemory32(map_list_ptr + 0x0)
    # dex_file_size_val_calc = map_off + (0x4) + map_list_size_val * (0x2 + 0x2 + 0x4 + 0x4)

    # config.log_print("[LoadClassMembers] begin_ = %0#10x, size_ = %#x (inferring size = %#x), location_ = %s" % (dex_file_begin_val, dex_file_size_val, dex_file_size_val_calc, dex_file_location_string_val))
    config.log_print(
        "[LoadClassMembers] begin_ = %0#10x, size_ = %#x, location_ = %s" %
        (dex_file_begin_val, dex_file_size_val, dex_file_location_string_val))

    # get the "class_data" parameter
    class_data_param = int(
        execution_state.getRegisterService().getValue("R3")) & 0xffffffff
    class_data_off = class_data_param - dex_file_begin_val

    # parse class_data_item
    static_fields_size_off = 0x0
    static_fields_size, length_static_fields_size = class_data_item.get_static_fields_size(
        dex_file_begin_val, class_data_off, static_fields_size_off)
    config.log_print(
        "[LoadClassMembers] [class_data_item] static_fields_size = %#x" %
        static_fields_size)

    instance_fields_size_off = static_fields_size_off + length_static_fields_size
    instance_fields_size, length_instance_fields_size = class_data_item.get_instance_fields_size(
        dex_file_begin_val, class_data_off, instance_fields_size_off)
    config.log_print(
        "[LoadClassMembers] [class_data_item] instance_fields_size = %#x" %
        instance_fields_size)

    direct_methods_size_off = instance_fields_size_off + length_instance_fields_size
    direct_methods_size, length_direct_methods_size = class_data_item.get_direct_methods_size(
        dex_file_begin_val, class_data_off, direct_methods_size_off)
    config.log_print(
        "[LoadClassMembers] [class_data_item] direct_methods_size = %#x" %
        direct_methods_size)

    virtual_methods_size_off = direct_methods_size_off + length_direct_methods_size
    virtual_methods_size, length_virtual_methods_size = class_data_item.get_virtual_methods_size(
        dex_file_begin_val, class_data_off, virtual_methods_size_off)
    config.log_print(
        "[LoadClassMembers] [class_data_item] virtual_methods_size = %#x" %
        virtual_methods_size)

    static_fields_off = virtual_methods_size_off + length_virtual_methods_size
    static_fields, length_static_fields = class_data_item.get_static_fields(
        dex_file_begin_val, class_data_off, static_fields_off,
        static_fields_size)
    for idx in range(static_fields_size):
        config.log_print(
            "[LoadClassMembers] [class_data_item] static_fields[%d].field_idx_diff = %#x"
            % (idx, static_fields[idx][0]))
        config.log_print(
            "[LoadClassMembers] [class_data_item] static_fields[%d].access_flags = %0#10x"
            % (idx, static_fields[idx][1]))

    instance_fields_off = static_fields_off + length_static_fields
    instance_fields, length_instance_fields = class_data_item.get_instance_fields(
        dex_file_begin_val, class_data_off, instance_fields_off,
        instance_fields_size)
    for idx in range(instance_fields_size):
        config.log_print(
            "[LoadClassMembers] [class_data_item] instance_fields[%d].field_idx_diff = %#x"
            % (idx, instance_fields[idx][0]))
        config.log_print(
            "[LoadClassMembers] [class_data_item] instance_fields[%d].access_flags = %0#10x"
            % (idx, instance_fields[idx][1]))

    direct_methods_off = instance_fields_off + length_instance_fields
    direct_methods, length_direct_methods = class_data_item.get_direct_methods(
        dex_file_begin_val, class_data_off, direct_methods_off,
        direct_methods_size)
    for idx in range(direct_methods_size):
        config.log_print(
            "[LoadClassMembers] [class_data_item] direct_methods[%d].method_idx_diff = %#x"
            % (idx, direct_methods[idx][0]))
        config.log_print(
            "[LoadClassMembers] [class_data_item] direct_methods[%d].access_flags = %0#10x"
            % (idx, direct_methods[idx][1]))
        config.log_print(
            "[LoadClassMembers] [class_data_item] direct_methods[%d].code_off = %0#10x"
            % (idx, direct_methods[idx][2]))

    virtual_methods_off = direct_methods_off + length_direct_methods
    virtual_methods, length_virtual_methods = class_data_item.get_virtual_methods(
        dex_file_begin_val, class_data_off, virtual_methods_off,
        virtual_methods_size)
    for idx in range(virtual_methods_size):
        config.log_print(
            "[LoadClassMembers] [class_data_item] virtual_methods[%d].method_idx_diff = %#x"
            % (idx, virtual_methods[idx][0]))
        config.log_print(
            "[LoadClassMembers] [class_data_item] virtual_methods[%d].access_flags = %0#10x"
            % (idx, virtual_methods[idx][1]))
        config.log_print(
            "[LoadClassMembers] [class_data_item] virtual_methods[%d].code_off = %0#10x"
            % (idx, virtual_methods[idx][2]))

    class_data_size = virtual_methods_off + length_virtual_methods
    if class_data_off < 0:
        config.log_print(
            "[LoadClassMembers] class_data = %0#10x (offset = %#x, size = %#x)"
            % (class_data_param, 0xffffffff + class_data_off, class_data_size))
    else:
        config.log_print(
            "[LoadClassMembers] class_data = %0#10x (offset = %#x, size = %#x)"
            % (class_data_param, class_data_off, class_data_size))

    # dump the class_data_item
    if (class_data_off < 0) or (class_data_off > dex_file_size_val):
        file_path = os.path.join(
            config.workspace, config.dex_directory,
            "class_data_item_%0#10x.bin" %
            (class_data_off if class_data_off > 0 else
             (0xffffffff + class_data_off)))
        file_format = "binary"
        file_vtl_start_address = class_data_param
        file_vtl_end_address = class_data_param + class_data_size - 0x1
        memory.dump(file_path, file_format, file_vtl_start_address,
                    file_vtl_end_address)

    # dump the in-memory DexFile
    file_path = os.path.join(config.workspace, config.dex_directory,
                             dex_file_location_string_val.split("/")[-1])
    if os.path.exists(file_path):
        os.remove(file_path)
    file_format = "binary"
    file_vtl_start_address = dex_file_begin_val
    file_vtl_end_address = dex_file_begin_val + dex_file_size_val - 0x1
    file_path = memory.dump(file_path, file_format, file_vtl_start_address,
                            file_vtl_end_address)

    # continue the execution of the target application
    execution_state.getExecutionService().resume()
    cleanup()
    return
Exemple #7
0
# MemRead  -> 1868
# MemWrite -> 18e8
if __name__ == '__main__':
    '''
	reg = int(execution_state.getRegisterService().getValue("R0")) & 0xffffffff
	reg = int(execution_state.getRegisterService().getValue("R1")) & 0xffffffff
	reg = int(execution_state.getRegisterService().getValue("R2")) & 0xffffffff
	reg = int(execution_state.getRegisterService().getValue("R3")) & 0xffffffff
	reg = int(execution_state.getRegisterService().getValue("R4")) & 0xffffffff
	reg = int(execution_state.getRegisterService().getValue("R5")) & 0xffffffff
	reg = int(execution_state.getRegisterService().getValue("R6")) & 0xffffffff
	reg = int(execution_state.getRegisterService().getValue("R7")) & 0xffffffff
	reg = int(execution_state.getRegisterService().getValue("R8")) & 0xffffffff
	reg = int(execution_state.getRegisterService().getValue("R9")) & 0xffffffff
	reg = int(execution_state.getRegisterService().getValue("R10")) & 0xffffffff
	reg = int(execution_state.getRegisterService().getValue("R11")) & 0xffffffff
	reg = int(execution_state.getRegisterService().getValue("R12")) & 0xffffffff
	reg = int(execution_state.getRegisterService().getValue("SP")) & 0xffffffff
	pc = int(execution_state.getRegisterService().getValue("PC")) & 0xffffffff
	lr = int(execution_state.getRegisterService().getValue("LR")) & 0xffffffff
	'''
    loop = 2
    while loop > 0:
        memory.dump("D:\\tmp.bin", "binary", 0xd49ef000,
                    0xd49ef000 + 0x6000 - 0x1)
        os.remove("D:\\tmp.bin")
        loop = loop - 1

    # cleanup()
    execution_state.getExecutionService().resume()
Exemple #8
0
def LoadMethod():
    # -1-

    # get the "dst" parameter
    dst_ptr = int(execution_state.getRegisterService().getValue("SP")) + 0x4
    dst_val = memory.readMemory32(dst_ptr)
    if config.debug:
        print "[LoadMethod] dst = %0#10x" % dst_val

    # very strange !!!
    if (dst_val == 0x0) or (dst_val == 0x00000001):
        execution_state.getExecutionService().resume()
        return

    # get the return address of the LoadMethod
    brk_return_address = int(
        execution_state.getRegisterService().getValue("LR")) & 0xfffffffe
    if config.debug:
        print "[LoadMethod] return address = %0#10x" % brk_return_address

    # -2-

    # resume to the end of the LoadMethod
    brk_LoadMethod_end = config.libart_base + config.LoadMethod_end - config.libart_file_offset + config.libart_memory_offset
    execution_state.getExecutionService().resumeTo(brk_LoadMethod_end)
    try:
        execution_state.getExecutionService().waitForStop(
            120000)  # wait for 120s
    except DebugException:
        raise RuntimeError("wtf !!!")

    # get the pointer that refers to ArtMethod
    art_method_ptr = dst_val

    # read the "declaring_class_" field of ArtMethod
    art_method_declaring_class_ptr = art_method_ptr + config.offset_ArtMethod_declaring_class_
    # read the "root_" field of GcRoot
    art_method_declaring_class_root_ptr = art_method_declaring_class_ptr + config.offset_GcRoot_root_
    # read the "refeence_" field of CompressesReference
    art_method_declaring_class_root_reference_ptr = art_method_declaring_class_root_ptr + config.offset_CompressesReference_reference_
    art_method_declaring_class_root_reference_val = memory.readMemory32(
        art_method_declaring_class_root_reference_ptr)

    class_ptr = art_method_declaring_class_root_reference_val

    # read the "dex_cache_" field of Class
    class_dex_cache_ptr = class_ptr + config.offset_Class_dex_cache_
    # read the "reference_" field of HeapReference
    class_dex_cache_reference_ptr = class_dex_cache_ptr + config.offset_HeapReference_reference_
    class_dex_cache_reference_val = memory.readMemory32(
        class_dex_cache_reference_ptr)
    dex_cache_ptr = class_dex_cache_reference_val
    # read the "dex_file_" field of DexCache
    dex_cache_dex_file_ptr = dex_cache_ptr + config.offset_DexCache_dex_file_
    dex_cache_dex_file_val = memory.readMemory32(dex_cache_dex_file_ptr)

    dex_file_ptr = dex_cache_dex_file_val

    # read the "begin_" field of DexFile
    dex_file_begin_ptr = dex_file_ptr + config.offset_DexFile_begin_
    dex_file_begin_val = memory.readMemory32(dex_file_begin_ptr)
    if config.debug:
        print "[LoadMethod] ArtMethod::declaring_class_::dex_cache_::dex_file_::begin_ = %0#10x" % dex_file_begin_val

    # read the "size_" field of DexFile
    dex_file_size_ptr = dex_file_ptr + config.offset_DexFile_size_
    dex_file_size_val = memory.readMemory32(dex_file_size_ptr)
    if config.debug:
        print "[LoadMethod] ArtMethod::declaring_class_::dex_cache_::dex_file_::size_ = %#x" % dex_file_size_val

    # read the "location_" field of DexFile
    dex_file_location_ptr = dex_file_ptr + config.offset_DexFile_location_
    # retrieve the value of std::string
    dex_file_location_string_val = retrieve_string_value(dex_file_location_ptr)
    if config.debug:
        print "[LoadMethod] ArtMethod::declaring_class_::dex_cache_::dex_file_::location_ = %s" % dex_file_location_string_val

    # we only focus on the Java invocation appeared in the target package
    if not config.package_filter(dex_file_location_string_val):
        # continue the execution of the target application
        execution_state.getExecutionService().resume()
        cleanup()
        return

    # read the "access_flags_" field of ArtMethod
    art_method_access_flags_ptr = art_method_ptr + config.offset_ArtMethod_access_flags_
    art_method_access_flags_value = memory.readMemory32(
        art_method_access_flags_ptr)
    if config.debug:
        print "[LoadMethod] ArtMethod::access_flags_ = %#x (%s)" % (
            art_method_access_flags_value,
            config.resolve_access_flags(art_method_access_flags_value))

    # read the "dex_code_item_offset_" field of ArtMethod
    art_method_dex_code_item_offset_ptr = art_method_ptr + config.offset_ArtMethod_dex_code_item_offset_
    art_method_dex_code_item_offset_value = memory.readMemory32(
        art_method_dex_code_item_offset_ptr)
    if config.debug:
        print "[LoadMethod] ArtMethod::dex_code_item_offset_ = %#x" % art_method_dex_code_item_offset_value

    # read the "tries_size" field of the "DexFile::CodeItem" structure
    tries_size_off = (
        (dex_file_begin_val + art_method_dex_code_item_offset_value)
        & 0xffffffff) + config.offset_DexFile_CodeItem_tries_size_
    tries_size_val = memory.readMemory16(tries_size_off)
    if config.debug:
        print "[LoadMethod] ArtMethod::CodeItem::tries_size_ = %#x" % tries_size_val

    # read the "insns_size_in_code_units_" field of the "DexFile::CodeItem" structure
    code_item_insns_size_in_code_units_off = (
        (dex_file_begin_val + art_method_dex_code_item_offset_value) &
        0xffffffff) + config.offset_DexFile_CodeItem_insns_size_in_code_units_
    code_item_insns_size_in_code_units_val = memory.readMemory32(
        code_item_insns_size_in_code_units_off)
    if config.debug:
        print "[LoadMethod] ArtMethod::CodeItem::insns_size_in_code_units_ = %#x" % code_item_insns_size_in_code_units_val

    # read the "dex_method_index_" field of ArtMethod
    art_method_dex_method_index_ptr = art_method_ptr + config.offset_ArtMethod_dex_method_index_
    art_method_dex_method_index_val = memory.readMemory32(
        art_method_dex_method_index_ptr)
    if config.debug:
        print "[LoadMethod] ArtMethod::dex_method_index_ = %#x" % art_method_dex_method_index_val

    # resolve
    string_ids_off = header_item.get_string_ids_off(dex_file_begin_val)
    type_ids_off = header_item.get_type_ids_off(dex_file_begin_val)
    proto_ids_off = header_item.get_proto_ids_off(dex_file_begin_val)
    method_ids_off = header_item.get_method_ids_off(dex_file_begin_val)

    class_idx = method_id_item.get_class_idx(dex_file_begin_val,
                                             method_ids_off,
                                             art_method_dex_method_index_val)
    class_descriptor_idx = type_id_item.get_descriptor_idx(
        dex_file_begin_val, type_ids_off, class_idx)
    class_descriptor_content = string_id_item.get_string_id_item_data(
        dex_file_begin_val, string_ids_off, class_descriptor_idx)
    # if config.debug:
    # print "[LoadMethod] class name = %s" % class_descriptor_content

    name_idx = method_id_item.get_name_idx(dex_file_begin_val, method_ids_off,
                                           art_method_dex_method_index_val)
    name_content = string_id_item.get_string_id_item_data(
        dex_file_begin_val, string_ids_off, name_idx)
    # if config.debug:
    # print "[LoadMethod] method name = %s" % name_content

    proto_idx = method_id_item.get_proto_idx(dex_file_begin_val,
                                             method_ids_off,
                                             art_method_dex_method_index_val)
    proto_return_type_idx = proto_id_item.get_return_type_idx(
        dex_file_begin_val, proto_ids_off, proto_idx)
    proto_return_type_descriptor_idx = type_id_item.get_descriptor_idx(
        dex_file_begin_val, type_ids_off, proto_return_type_idx)
    proto_return_type_descriptor_content = string_id_item.get_string_id_item_data(
        dex_file_begin_val, string_ids_off, proto_return_type_descriptor_idx)
    # if config.debug:
    # print "[LoadMethod] return type = %s" % proto_return_type_descriptor_content

    parameters_content = ""
    proto_parameters_list = proto_id_item.get_parameters_list(
        dex_file_begin_val, proto_ids_off, proto_idx)
    if len(proto_parameters_list) == 0:
        parameters_content = "()"
    else:
        for parameter_idx in range(len(proto_parameters_list)):
            parameter_type_idx = proto_parameters_list[parameter_idx]
            parameter_type_descriptor_idx = type_id_item.get_descriptor_idx(
                dex_file_begin_val, type_ids_off, parameter_type_idx)
            parameter_type_descriptor_content = string_id_item.get_string_id_item_data(
                dex_file_begin_val, string_ids_off,
                parameter_type_descriptor_idx)

            if len(proto_parameters_list) == 1:
                parameters_content = parameters_content + "(" + parameter_type_descriptor_content + ")"
            else:
                if parameter_idx == 0:
                    parameters_content = parameters_content + "(" + parameter_type_descriptor_content
                elif parameter_idx == (len(proto_parameters_list) - 1):
                    parameters_content = parameters_content + "," + parameter_type_descriptor_content + ")"
                else:
                    parameters_content = parameters_content + "," + parameter_type_descriptor_content
    # if config.debug:
    # print "[DoCall] parameters = %s" % parameters_content

    method_signature = "%s %s->%s %s%s" % (
        config.resolve_method_access_flags(art_method_access_flags_value),
        class_descriptor_content, proto_return_type_descriptor_content,
        name_content, parameters_content)
    if config.debug:
        print "[LoadMethod] method signature = %s" % method_signature

    config.log_print("[LoadMethod] origin     method signature = %s" %
                     method_signature)

    origin_ArtMethod_declaring_class = class_ptr
    origin_ArtMethod_access_flags = art_method_access_flags_value
    origin_ArtMethod_dex_code_item_offset = art_method_dex_code_item_offset_value
    origin_ArtMethod_dex_method_index = art_method_dex_method_index_val
    origin_ArtMethod_signature = method_signature

    # -3-

    # resume to the return address of the LoadMethod
    execution_state.getExecutionService().resumeTo(brk_return_address)
    try:
        execution_state.getExecutionService().waitForStop(
            120000)  # wait for 120s
    except DebugException:
        raise RuntimeError("wtf !!!")

    # read the "declaring_class_" field of ArtMethod
    art_method_declaring_class_ptr = art_method_ptr + config.offset_ArtMethod_declaring_class_
    # read the "root_" field of GcRoot
    art_method_declaring_class_root_ptr = art_method_declaring_class_ptr + config.offset_GcRoot_root_
    # read the "refeence_" field of CompressesReference
    art_method_declaring_class_root_reference_ptr = art_method_declaring_class_root_ptr + config.offset_CompressesReference_reference_
    art_method_declaring_class_root_reference_val = memory.readMemory32(
        art_method_declaring_class_root_reference_ptr)

    class_ptr = art_method_declaring_class_root_reference_val

    # read the "dex_cache_" field of Class
    class_dex_cache_ptr = class_ptr + config.offset_Class_dex_cache_
    # read the "reference_" field of HeapReference
    class_dex_cache_reference_ptr = class_dex_cache_ptr + config.offset_HeapReference_reference_
    class_dex_cache_reference_val = memory.readMemory32(
        class_dex_cache_reference_ptr)
    dex_cache_ptr = class_dex_cache_reference_val
    # read the "dex_file_" field of DexCache
    dex_cache_dex_file_ptr = dex_cache_ptr + config.offset_DexCache_dex_file_
    dex_cache_dex_file_val = memory.readMemory32(dex_cache_dex_file_ptr)

    dex_file_ptr = dex_cache_dex_file_val

    # read the "begin_" field of DexFile
    dex_file_begin_ptr = dex_file_ptr + config.offset_DexFile_begin_
    dex_file_begin_val = memory.readMemory32(dex_file_begin_ptr)
    if config.debug:
        print "[LoadMethod] ArtMethod::declaring_class_::dex_cache_::dex_file_::begin_ = %0#10x" % dex_file_begin_val

    # read the "size_" field of DexFile
    dex_file_size_ptr = dex_file_ptr + config.offset_DexFile_size_
    dex_file_size_val = memory.readMemory32(dex_file_size_ptr)
    if config.debug:
        print "[LoadMethod] ArtMethod::declaring_class_::dex_cache_::dex_file_::size_ = %#x" % dex_file_size_val

    # read the "location_" field of DexFile
    dex_file_location_ptr = dex_file_ptr + config.offset_DexFile_location_
    # retrieve the value of std::string
    dex_file_location_string_val = retrieve_string_value(dex_file_location_ptr)
    if config.debug:
        print "[LoadMethod] ArtMethod::declaring_class_::dex_cache_::dex_file_::location_ = %s" % dex_file_location_string_val

    # # we only focus on the Java invocation appeared in the target package
    # if not config.package_filter(dex_file_location_string_val):
    # # continue the execution of the target application
    # execution_state.getExecutionService().resume()
    # cleanup()
    # return

    # read the "access_flags_" field of ArtMethod
    art_method_access_flags_ptr = art_method_ptr + config.offset_ArtMethod_access_flags_
    art_method_access_flags_value = memory.readMemory32(
        art_method_access_flags_ptr)
    if config.debug:
        print "[LoadMethod] ArtMethod::access_flags_ = %#x (%s)" % (
            art_method_access_flags_value,
            config.resolve_access_flags(art_method_access_flags_value))

    # read the "dex_code_item_offset_" field of ArtMethod
    art_method_dex_code_item_offset_ptr = art_method_ptr + config.offset_ArtMethod_dex_code_item_offset_
    art_method_dex_code_item_offset_value = memory.readMemory32(
        art_method_dex_code_item_offset_ptr)
    if config.debug:
        print "[LoadMethod] ArtMethod::dex_code_item_offset_ = %#x" % art_method_dex_code_item_offset_value

    # read the "tries_size" field of the "DexFile::CodeItem" structure
    tries_size_off = (
        (dex_file_begin_val + art_method_dex_code_item_offset_value)
        & 0xffffffff) + config.offset_DexFile_CodeItem_tries_size_
    tries_size_val = memory.readMemory16(tries_size_off)
    if config.debug:
        print "[LoadMethod] ArtMethod::CodeItem::tries_size_ = %#x" % tries_size_val

    # read the "insns_size_in_code_units_" field of the "DexFile::CodeItem" structure
    code_item_insns_size_in_code_units_off = (
        (dex_file_begin_val + art_method_dex_code_item_offset_value) &
        0xffffffff) + config.offset_DexFile_CodeItem_insns_size_in_code_units_
    code_item_insns_size_in_code_units_val = memory.readMemory32(
        code_item_insns_size_in_code_units_off)
    if config.debug:
        print "[LoadMethod] ArtMethod::CodeItem::insns_size_in_code_units_ = %#x" % code_item_insns_size_in_code_units_val

    # read the "dex_method_index_" field of ArtMethod
    art_method_dex_method_index_ptr = art_method_ptr + config.offset_ArtMethod_dex_method_index_
    art_method_dex_method_index_val = memory.readMemory32(
        art_method_dex_method_index_ptr)
    if config.debug:
        print "[LoadMethod] ArtMethod::dex_method_index_ = %#x" % art_method_dex_method_index_val

    # resolve
    string_ids_off = header_item.get_string_ids_off(dex_file_begin_val)
    type_ids_off = header_item.get_type_ids_off(dex_file_begin_val)
    proto_ids_off = header_item.get_proto_ids_off(dex_file_begin_val)
    method_ids_off = header_item.get_method_ids_off(dex_file_begin_val)

    class_idx = method_id_item.get_class_idx(dex_file_begin_val,
                                             method_ids_off,
                                             art_method_dex_method_index_val)
    class_descriptor_idx = type_id_item.get_descriptor_idx(
        dex_file_begin_val, type_ids_off, class_idx)
    class_descriptor_content = string_id_item.get_string_id_item_data(
        dex_file_begin_val, string_ids_off, class_descriptor_idx)
    # if config.debug:
    # print "[LoadMethod] class name = %s" % class_descriptor_content

    name_idx = method_id_item.get_name_idx(dex_file_begin_val, method_ids_off,
                                           art_method_dex_method_index_val)
    name_content = string_id_item.get_string_id_item_data(
        dex_file_begin_val, string_ids_off, name_idx)
    # if config.debug:
    # print "[LoadMethod] method name = %s" % name_content

    proto_idx = method_id_item.get_proto_idx(dex_file_begin_val,
                                             method_ids_off,
                                             art_method_dex_method_index_val)
    proto_return_type_idx = proto_id_item.get_return_type_idx(
        dex_file_begin_val, proto_ids_off, proto_idx)
    proto_return_type_descriptor_idx = type_id_item.get_descriptor_idx(
        dex_file_begin_val, type_ids_off, proto_return_type_idx)
    proto_return_type_descriptor_content = string_id_item.get_string_id_item_data(
        dex_file_begin_val, string_ids_off, proto_return_type_descriptor_idx)
    # if config.debug:
    # print "[LoadMethod] return type = %s" % proto_return_type_descriptor_content

    parameters_content = ""
    proto_parameters_list = proto_id_item.get_parameters_list(
        dex_file_begin_val, proto_ids_off, proto_idx)
    if len(proto_parameters_list) == 0:
        parameters_content = "()"
    else:
        for parameter_idx in range(len(proto_parameters_list)):
            parameter_type_idx = proto_parameters_list[parameter_idx]
            parameter_type_descriptor_idx = type_id_item.get_descriptor_idx(
                dex_file_begin_val, type_ids_off, parameter_type_idx)
            parameter_type_descriptor_content = string_id_item.get_string_id_item_data(
                dex_file_begin_val, string_ids_off,
                parameter_type_descriptor_idx)

            if len(proto_parameters_list) == 1:
                parameters_content = parameters_content + "(" + parameter_type_descriptor_content + ")"
            else:
                if parameter_idx == 0:
                    parameters_content = parameters_content + "(" + parameter_type_descriptor_content
                elif parameter_idx == (len(proto_parameters_list) - 1):
                    parameters_content = parameters_content + "," + parameter_type_descriptor_content + ")"
                else:
                    parameters_content = parameters_content + "," + parameter_type_descriptor_content
    # if config.debug:
    # print "[DoCall] parameters = %s" % parameters_content

    method_signature = "%s %s->%s %s%s" % (
        config.resolve_method_access_flags(art_method_access_flags_value),
        class_descriptor_content, proto_return_type_descriptor_content,
        name_content, parameters_content)
    if config.debug:
        print "[LoadMethod] method signature = %s" % method_signature

    config.log_print("[LoadMethod] suspicious method signature = %s" %
                     method_signature)

    suspicious_ArtMethod_declaring_class = class_ptr
    suspicious_ArtMethod_access_flags = art_method_access_flags_value
    suspicious_ArtMethod_dex_code_item_offset = art_method_dex_code_item_offset_value
    suspicious_ArtMethod_dex_method_index = art_method_dex_method_index_val
    suspicious_ArtMethod_signature = method_signature

    # if (origin_ArtMethod_declaring_class != suspicious_ArtMethod_declaring_class) or \
    # (origin_ArtMethod_access_flags != suspicious_ArtMethod_access_flags) or \
    # (origin_ArtMethod_dex_code_item_offset != suspicious_ArtMethod_dex_code_item_offset):
    # config.log_print("[LoadMethod] hooked !!!")
    # assert origin_ArtMethod_dex_method_index == suspicious_ArtMethod_dex_method_index
    # config.log_print("[LoadMethod][origin]     declaring_class_ = %0#10x, access_flags_ = %#x, dex_code_item_offset_ = %0#10x" % (origin_ArtMethod_declaring_class, origin_ArtMethod_access_flags, origin_ArtMethod_dex_code_item_offset))
    # config.log_print("[LoadMethod][suspicious] declaring_class_ = %0#10x, access_flags_ = %#x, dex_code_item_offset_ = %0#10x" % (suspicious_ArtMethod_declaring_class, suspicious_ArtMethod_access_flags, suspicious_ArtMethod_dex_code_item_offset))

    # # dump the code_item on-demand
    # padding = 0x2 if (tries_size_val > 0) and (code_item_insns_size_in_code_units_val % 2 == 1) else 0x0
    # code_item_size = config.offset_DexFile_CodeItem_insns_size_in_code_units_ + 0x4 + 0x2 * code_item_insns_size_in_code_units_val + padding + 0x8 * tries_size_val
    # code_item_size = code_item_size + encoded_catch_handler_list.parse_encoded_catch_handler_list(dex_file_begin_val, suspicious_ArtMethod_dex_code_item_offset + code_item_size, tries_size_val)
    # file_path = os.path.join(config.workspace, config.dex_directory, "code_item_%#x_%0#10x.bin" % (origin_ArtMethod_dex_method_index, origin_ArtMethod_dex_code_item_offset))
    # file_format = "binary"
    # file_vtl_start_address = (dex_file_begin_val + suspicious_ArtMethod_dex_code_item_offset) & 0xffffffff
    # file_vtl_end_address = ((dex_file_begin_val + suspicious_ArtMethod_dex_code_item_offset) & 0xffffffff) + code_item_size - 0x1
    # memory.dump(file_path, file_format, file_vtl_start_address, file_vtl_end_address)

    # Notice: for ijiami-16, we force dump the code_item
    if (art_method_dex_code_item_offset_value != 0x0):
        padding = 0x2 if (tries_size_val > 0) and (
            code_item_insns_size_in_code_units_val % 2 == 1) else 0x0
        code_item_size = config.offset_DexFile_CodeItem_insns_size_in_code_units_ + 0x4 + 0x2 * code_item_insns_size_in_code_units_val + padding + 0x8 * tries_size_val
        code_item_size = code_item_size + encoded_catch_handler_list.parse_encoded_catch_handler_list(
            dex_file_begin_val, suspicious_ArtMethod_dex_code_item_offset +
            code_item_size, tries_size_val)
        file_path = os.path.join(
            config.workspace, config.dex_directory,
            "code_item_%#x_%0#10x.bin" %
            (origin_ArtMethod_dex_method_index,
             origin_ArtMethod_dex_code_item_offset))
        file_format = "binary"
        file_vtl_start_address = (
            dex_file_begin_val +
            suspicious_ArtMethod_dex_code_item_offset) & 0xffffffff
        file_vtl_end_address = (
            (dex_file_begin_val + suspicious_ArtMethod_dex_code_item_offset)
            & 0xffffffff) + code_item_size - 0x1
        memory.dump(file_path, file_format, file_vtl_start_address,
                    file_vtl_end_address)

    # continue the execution of the target application
    execution_state.getExecutionService().resume()
    cleanup()
    return
Exemple #9
0
def dump_modified_method(origin_method_list, suspicious_method_list):
	assert len(origin_method_list) == len(suspicious_method_list)
	
	for idx in range(len(origin_method_list)):
		origin_method = origin_method_list[idx]
		suspicious_method = suspicious_method_list[idx]
		if origin_method != suspicious_method:
			assert origin_method[0] == suspicious_method[0]
			assert origin_method[3] == suspicious_method[3]
			
			# read the "declaring_class_" field of ArtMethod
			art_method_declaring_class_ptr = origin_method[0]
			# read the "root_" field of GcRoot
			art_method_declaring_class_root_ptr = art_method_declaring_class_ptr + config.offset_GcRoot_root_
			# read the "refeence_" field of CompressesReference
			art_method_declaring_class_root_reference_ptr = art_method_declaring_class_root_ptr + config.offset_CompressesReference_reference_
			art_method_declaring_class_root_reference_val = memory.readMemory32(art_method_declaring_class_root_reference_ptr)
			
			class_ptr = art_method_declaring_class_root_reference_val
			
			# read the "dex_cache_" field of Class
			class_dex_cache_ptr = class_ptr + config.offset_Class_dex_cache_
			# read the "reference_" field of HeapReference
			class_dex_cache_reference_ptr = class_dex_cache_ptr + config.offset_HeapReference_reference_
			class_dex_cache_reference_val = memory.readMemory32(class_dex_cache_reference_ptr)
			dex_cache_ptr = class_dex_cache_reference_val
			# read the "dex_file_" field of DexCache
			dex_cache_dex_file_ptr = dex_cache_ptr + config.offset_DexCache_dex_file_
			dex_cache_dex_file_val = memory.readMemory32(dex_cache_dex_file_ptr)
			
			dex_file_ptr = dex_cache_dex_file_val
			
			# read the "begin_" field of DexFile
			dex_file_begin_ptr = dex_file_ptr + config.offset_DexFile_begin_
			dex_file_begin_val = memory.readMemory32(dex_file_begin_ptr)
			
			art_method_dex_method_index_val = origin_method[3]
			
			# resolve 
			string_ids_off = header_item.get_string_ids_off(dex_file_begin_val)
			type_ids_off = header_item.get_type_ids_off(dex_file_begin_val)
			proto_ids_off = header_item.get_proto_ids_off(dex_file_begin_val)
			method_ids_off = header_item.get_method_ids_off(dex_file_begin_val)
			
			class_idx = method_id_item.get_class_idx(dex_file_begin_val, method_ids_off, art_method_dex_method_index_val)
			class_descriptor_idx = type_id_item.get_descriptor_idx(dex_file_begin_val, type_ids_off, class_idx)
			class_descriptor_content = string_id_item.get_string_id_item_data(dex_file_begin_val, string_ids_off, class_descriptor_idx)
				
			name_idx = method_id_item.get_name_idx(dex_file_begin_val, method_ids_off, art_method_dex_method_index_val)
			name_content = string_id_item.get_string_id_item_data(dex_file_begin_val, string_ids_off, name_idx)
			
			proto_idx = method_id_item.get_proto_idx(dex_file_begin_val, method_ids_off, art_method_dex_method_index_val)
			proto_return_type_idx = proto_id_item.get_return_type_idx(dex_file_begin_val, proto_ids_off, proto_idx)
			proto_return_type_descriptor_idx = type_id_item.get_descriptor_idx(dex_file_begin_val, type_ids_off, proto_return_type_idx)
			proto_return_type_descriptor_content = string_id_item.get_string_id_item_data(dex_file_begin_val, string_ids_off, proto_return_type_descriptor_idx)
			
			parameters_content = ""
			proto_parameters_list = proto_id_item.get_parameters_list(dex_file_begin_val, proto_ids_off, proto_idx)
			if len(proto_parameters_list) == 0:
				parameters_content = "()"
			else:
				for parameter_idx in range(len(proto_parameters_list)):
					parameter_type_idx = proto_parameters_list[parameter_idx]
					parameter_type_descriptor_idx = type_id_item.get_descriptor_idx(dex_file_begin_val, type_ids_off, parameter_type_idx)
					parameter_type_descriptor_content = string_id_item.get_string_id_item_data(dex_file_begin_val, string_ids_off, parameter_type_descriptor_idx)
				
					if len(proto_parameters_list) == 1:
						parameters_content = parameters_content + "(" + parameter_type_descriptor_content + ")"
					else:
						if parameter_idx == 0:
							parameters_content = parameters_content + "(" + parameter_type_descriptor_content
						elif parameter_idx == (len(proto_parameters_list) - 1):
							parameters_content = parameters_content + "," + parameter_type_descriptor_content + ")"
						else:
							parameters_content = parameters_content + "," + parameter_type_descriptor_content
			
			method_signature = "?? %s->%s %s%s" % (class_descriptor_content, proto_return_type_descriptor_content, name_content, parameters_content)
			
			config.log_print("[ClassModification] mirror::Class has been modified !!!")
			config.log_print("[ClassModification][origin]     signature = %s, access_flags_ = %#x, dex_code_item_offset_ = %0#10x, entry_point_from_quick_compiled_code = %0#10x" % (method_signature, origin_method[1], origin_method[2], origin_method[8]))
			config.log_print("[ClassModification][suspicious] signature = %s, access_flags_ = %#x, dex_code_item_offset_ = %0#10x, entry_point_from_quick_compiled_code = %0#10x" % (method_signature, suspicious_method[1], suspicious_method[2], suspicious_method[8]))
			
			if origin_method[2] != suspicious_method[2]:
				# dump the code_item on-demand
				tries_size_off = ((dex_file_begin_val + suspicious_method[2]) & 0xffffffff) + config.offset_DexFile_CodeItem_tries_size_
				print "[ClassModification] tries_size_off = %#x" % tries_size_off
				tries_size_val = memory.readMemory16(tries_size_off)
				code_item_insns_size_in_code_units_off = ((dex_file_begin_val + suspicious_method[2]) & 0xffffffff) + config.offset_DexFile_CodeItem_insns_size_in_code_units_
				print "[ClassModification] code_item_insns_size_in_code_units_off = %#x" % code_item_insns_size_in_code_units_off
				code_item_insns_size_in_code_units_val = memory.readMemory32(code_item_insns_size_in_code_units_off)
				padding = 0x2 if (tries_size_val > 0) and (code_item_insns_size_in_code_units_val % 2 == 1) else 0x0
				code_item_size = config.offset_DexFile_CodeItem_insns_size_in_code_units_ + 0x4 + 0x2 * code_item_insns_size_in_code_units_val + padding + 0x8 * tries_size_val
				code_item_size = code_item_size + encoded_catch_handler_list.parse_encoded_catch_handler_list(dex_file_begin_val, suspicious_method[2] + code_item_size, tries_size_val)
				file_path = os.path.join(config.workspace, config.dex_directory, "code_item_%#x_%0#10x.bin" % (origin_method[3], origin_method[2]))
				file_format = "binary"
				file_vtl_start_address = (dex_file_begin_val + suspicious_method[2]) & 0xffffffff
				file_vtl_end_address = ((dex_file_begin_val + suspicious_method[2]) & 0xffffffff) + code_item_size - 0x1
				memory.dump(file_path, file_format, file_vtl_start_address, file_vtl_end_address)
Exemple #10
0
def LoadClassMembers():
    # -- HEAD -- #
    start_prolog()

    # ensure that the LoadMethod breakpoint is enabled
    for idx in range(
            0,
            execution_state.getBreakpointService().getBreakpointCount()):
        brk_object = execution_state.getBreakpointService().getBreakpoint(idx)
        if (int(brk_object.getAddresses()[0])
                & 0xffffffff) == config.brk_LoadMethod:
            brk_object.enable()

    # -- BODY -- #

    # get the "dex_file" parameter
    dex_file_param = int(
        execution_state.getRegisterService().getValue("R2")) & 0xffffffff
    dex_file_ptr = dex_file_param

    # read the "location_" field
    dex_file_location_ptr = dex_file_ptr + config.offset_DexFile_location_
    # retrieve the value of the std::string structure
    dex_file_location_string_val = retrieve_string_value(dex_file_location_ptr)
    if config.debug:
        print "[LoadClassMembers] location_ = %s" % dex_file_location_string_val

    # we only focus on the DexFile whose location is suspicious
    if not config.package_filter(dex_file_location_string_val):
        # disable the LoadMethod breakpoint here
        for idx in range(
                0,
                execution_state.getBreakpointService().getBreakpointCount()):
            brk_object = execution_state.getBreakpointService().getBreakpoint(
                idx)
            if (int(brk_object.getAddresses()[0])
                    & 0xffffffff) == config.brk_LoadMethod:
                brk_object.disable()

        # -- TAIL -- #
        end_prolog()

        # continue the execution of the target application
        execution_state.getExecutionService().resume()
        cleanup()
        return

    # read the "begin_" field
    dex_file_begin_ptr = dex_file_ptr + config.offset_DexFile_begin_
    dex_file_begin_val = memory.readMemory32(dex_file_begin_ptr)
    if config.debug:
        print "[LoadClassMembers] begin_ = %0#10x" % dex_file_begin_val

    # read the "size_" field
    dex_file_size_ptr = dex_file_ptr + config.offset_DexFile_size_
    dex_file_size_val = memory.readMemory32(dex_file_size_ptr)
    if config.debug:
        print "[LoadClassMembers] size_ = %#x" % dex_file_size_val

    config.log_print(
        "[LoadClassMembers] begin_ = %0#10x, size_ = %#x, location_ = %s" %
        (dex_file_begin_val, dex_file_size_val, dex_file_location_string_val))

    # get the "class_data" parameter
    class_data_param = int(
        execution_state.getRegisterService().getValue("R3")) & 0xffffffff
    class_data_off = class_data_param - dex_file_begin_val

    config.log_print("[LoadClassMembers] class_data = %0#10x (offset = %#x)" %
                     (class_data_param, class_data_off))

    # parse class_data_item
    static_fields_size_off = 0x0
    static_fields_size, length_static_fields_size = class_data_item.get_static_fields_size(
        dex_file_begin_val, class_data_off, static_fields_size_off)
    config.log_print(
        "[LoadClassMembers] [class_data_item] static_fields_size = %#x" %
        static_fields_size)

    instance_fields_size_off = static_fields_size_off + length_static_fields_size
    instance_fields_size, length_instance_fields_size = class_data_item.get_instance_fields_size(
        dex_file_begin_val, class_data_off, instance_fields_size_off)
    config.log_print(
        "[LoadClassMembers] [class_data_item] instance_fields_size = %#x" %
        instance_fields_size)

    direct_methods_size_off = instance_fields_size_off + length_instance_fields_size
    direct_methods_size, length_direct_methods_size = class_data_item.get_direct_methods_size(
        dex_file_begin_val, class_data_off, direct_methods_size_off)
    config.log_print(
        "[LoadClassMembers] [class_data_item] direct_methods_size = %#x" %
        direct_methods_size)

    virtual_methods_size_off = direct_methods_size_off + length_direct_methods_size
    virtual_methods_size, length_virtual_methods_size = class_data_item.get_virtual_methods_size(
        dex_file_begin_val, class_data_off, virtual_methods_size_off)
    config.log_print(
        "[LoadClassMembers] [class_data_item] virtual_methods_size = %#x" %
        virtual_methods_size)

    static_fields_off = virtual_methods_size_off + length_virtual_methods_size
    static_fields, length_static_fields = class_data_item.get_static_fields(
        dex_file_begin_val, class_data_off, static_fields_off,
        static_fields_size)
    for idx in range(static_fields_size):
        config.log_print(
            "[LoadClassMembers] [class_data_item] static_fields[%d].field_idx_diff = %#x"
            % (idx, static_fields[idx][0]))
        config.log_print(
            "[LoadClassMembers] [class_data_item] static_fields[%d].access_flags = %0#10x"
            % (idx, static_fields[idx][1]))

    instance_fields_off = static_fields_off + length_static_fields
    instance_fields, length_instance_fields = class_data_item.get_instance_fields(
        dex_file_begin_val, class_data_off, instance_fields_off,
        instance_fields_size)
    for idx in range(instance_fields_size):
        config.log_print(
            "[LoadClassMembers] [class_data_item] instance_fields[%d].field_idx_diff = %#x"
            % (idx, instance_fields[idx][0]))
        config.log_print(
            "[LoadClassMembers] [class_data_item] instance_fields[%d].access_flags = %0#10x"
            % (idx, instance_fields[idx][1]))

    direct_methods_off = instance_fields_off + length_instance_fields
    direct_methods, length_direct_methods = class_data_item.get_direct_methods(
        dex_file_begin_val, class_data_off, direct_methods_off,
        direct_methods_size)
    for idx in range(direct_methods_size):
        config.log_print(
            "[LoadClassMembers] [class_data_item] direct_methods[%d].method_idx_diff = %#x"
            % (idx, direct_methods[idx][0]))
        config.log_print(
            "[LoadClassMembers] [class_data_item] direct_methods[%d].access_flags = %0#10x"
            % (idx, direct_methods[idx][1]))
        config.log_print(
            "[LoadClassMembers] [class_data_item] direct_methods[%d].code_off = %0#10x"
            % (idx, direct_methods[idx][2]))

    virtual_methods_off = direct_methods_off + length_direct_methods
    virtual_methods, length_virtual_methods = class_data_item.get_virtual_methods(
        dex_file_begin_val, class_data_off, virtual_methods_off,
        virtual_methods_off)
    for idx in range(virtual_methods_size):
        config.log_print(
            "[LoadClassMembers] [class_data_item] virtual_methods[%d].method_idx_diff = %#x"
            % (idx, virtual_methods[idx][0]))
        config.log_print(
            "[LoadClassMembers] [class_data_item] virtual_methods[%d].access_flags = %0#10x"
            % (idx, virtual_methods[idx][1]))
        config.log_print(
            "[LoadClassMembers] [class_data_item] virtual_methods[%d].code_off = %0#10x"
            % (idx, virtual_methods[idx][2]))

    class_data_size = virtual_methods_off + length_virtual_methods
    if class_data_off < 0:
        config.log_print(
            "[LoadClassMembers] class_data = %0#10x (offset = %#x, size = %#x)"
            % (class_data_param, 0xffffffff + class_data_off, class_data_size))
    else:
        config.log_print(
            "[LoadClassMembers] class_data = %0#10x (offset = %#x, size = %#x)"
            % (class_data_param, class_data_off, class_data_size))

    # dump the class_data_item on-demand
    if (class_data_off < 0) or (class_data_off > dex_file_size_val):
        file_path = os.path.join(
            config.workspace, config.dex_directory,
            "class_data_item_%0#10x.bin" %
            (class_data_off if class_data_off > 0 else
             (0xffffffff + class_data_off)))
        file_format = "binary"
        file_vtl_start_address = class_data_param
        file_vtl_end_address = class_data_param + class_data_size - 0x1
        memory.dump(file_path, file_format, file_vtl_start_address,
                    file_vtl_end_address)

    # -- TAIL -- #
    end_prolog()

    # continue the execution of the target application
    execution_state.getExecutionService().resume()
    cleanup()
    return
Exemple #11
0
def ClassModification():
	# at the very beginning, we remove the DexFile breakpoint
	rm_brks = []
	for breakpoint_index in range(0, execution_state.getBreakpointService().getBreakpointCount()):
		breakpoint_object = execution_state.getBreakpointService().getBreakpoint(breakpoint_index)
		if (int(breakpoint_object.getAddresses()[0]) & 0xffffffff) == config.brk_DexFile:
			rm_brks.append(breakpoint_object)
	for brk_obj in rm_brks:
		brk_obj.remove()

	# -1- DoCall
	
	# get the "shadow_frame" parameter
	shadow_frame_ptr = int(execution_state.getRegisterService().getValue("R2"))
	if config.debug:
		print "[ClassModification] shadow_frame = %0#10x" % shadow_frame_ptr
	
	# retrieve the "method_" field of ShadowFrame structure
	shadow_frame_method_ptr = memory.readMemory32(shadow_frame_ptr + config.offset_ShadowFrame_method_)
	
	# get the pointer that refers to ArtMethod
	art_method_ptr = shadow_frame_method_ptr
	if config.debug:
		print "[ClassModification] caller = %0#10x" % art_method_ptr

	# read the "declaring_class_" field of ArtMethod
	art_method_declaring_class_ptr = art_method_ptr + config.offset_ArtMethod_declaring_class_
	# read the "root_" field of GcRoot
	art_method_declaring_class_root_ptr = art_method_declaring_class_ptr + config.offset_GcRoot_root_
	# read the "refeence_" field of CompressesReference
	art_method_declaring_class_root_reference_ptr = art_method_declaring_class_root_ptr + config.offset_CompressesReference_reference_
	art_method_declaring_class_root_reference_val = memory.readMemory32(art_method_declaring_class_root_reference_ptr)
	
	class_ptr = art_method_declaring_class_root_reference_val
	caller_class_ptr = class_ptr
	
	# read the "dex_cache_" field of Class
	class_dex_cache_ptr = class_ptr + config.offset_Class_dex_cache_
	# read the "reference_" field of HeapReference
	class_dex_cache_reference_ptr = class_dex_cache_ptr + config.offset_HeapReference_reference_
	class_dex_cache_reference_val = memory.readMemory32(class_dex_cache_reference_ptr)
	dex_cache_ptr = class_dex_cache_reference_val
	# read the "dex_file_" field of DexCache
	dex_cache_dex_file_ptr = dex_cache_ptr + config.offset_DexCache_dex_file_
	dex_cache_dex_file_val = memory.readMemory32(dex_cache_dex_file_ptr)
	
	dex_file_ptr = dex_cache_dex_file_val
	
	# read the "begin_" field of DexFile
	dex_file_begin_ptr = dex_file_ptr + config.offset_DexFile_begin_
	dex_file_begin_val = memory.readMemory32(dex_file_begin_ptr)
	if config.debug:
		print "[ClassModification] ArtMethod::declaring_class_::dex_cache_::dex_file_::begin_ = %0#10x" % dex_file_begin_val
		
	# read the "size_" field of DexFile
	dex_file_size_ptr = dex_file_ptr + config.offset_DexFile_size_
	dex_file_size_val = memory.readMemory32(dex_file_size_ptr)
	if config.debug:
		print "[ClassModification] ArtMethod::declaring_class_::dex_cache_::dex_file_::size_ = %#x" % dex_file_size_val

	# read the "location_" field of DexFile
	dex_file_location_ptr = dex_file_ptr + config.offset_DexFile_location_
	# retrieve the value of std::string
	dex_file_location_string_val = retrieve_string_value(dex_file_location_ptr)
	if config.debug:
		print "[ClassModification] ArtMethod::declaring_class_::dex_cache_::dex_file_::location_ = %s" % dex_file_location_string_val
	
	# we only focus on the Java invocation appeared in the target package
	if not config.package_filter(dex_file_location_string_val):
		# continue the execution of the target application
		execution_state.getExecutionService().resume()
		cleanup()
		return
	
	# read the "access_flags_" field of ArtMethod
	art_method_access_flags_ptr = art_method_ptr + config.offset_ArtMethod_access_flags_
	art_method_access_flags_value = memory.readMemory32(art_method_access_flags_ptr)
	if config.debug:
		print "[ClassModification] ArtMethod::access_flags_ = %#x (%s)" % (art_method_access_flags_value, config.resolve_access_flags(art_method_access_flags_value))
	
	# read the "dex_code_item_offset_" field of ArtMethod
	art_method_dex_code_item_offset_ptr = art_method_ptr + config.offset_ArtMethod_dex_code_item_offset_
	art_method_dex_code_item_offset_value = memory.readMemory32(art_method_dex_code_item_offset_ptr)
	if config.debug:
		print "[ClassModification] ArtMethod::dex_code_item_offset_ = %#x" % art_method_dex_code_item_offset_value
	
	# read the "dex_method_index_" field of ArtMethod
	art_method_dex_method_index_ptr = art_method_ptr + config.offset_ArtMethod_dex_method_index_
	art_method_dex_method_index_val = memory.readMemory32(art_method_dex_method_index_ptr)
	if config.debug:
		print "[ClassModification] ArtMethod::dex_method_index_ = %#x" % art_method_dex_method_index_val
		
	# resolve 
	string_ids_off = header_item.get_string_ids_off(dex_file_begin_val)
	type_ids_off = header_item.get_type_ids_off(dex_file_begin_val)
	proto_ids_off = header_item.get_proto_ids_off(dex_file_begin_val)
	method_ids_off = header_item.get_method_ids_off(dex_file_begin_val)
	
	class_idx = method_id_item.get_class_idx(dex_file_begin_val, method_ids_off, art_method_dex_method_index_val)
	class_descriptor_idx = type_id_item.get_descriptor_idx(dex_file_begin_val, type_ids_off, class_idx)
	class_descriptor_content = string_id_item.get_string_id_item_data(dex_file_begin_val, string_ids_off, class_descriptor_idx)
	# if config.debug:
		# print "[ClassModification] class name = %s" % class_descriptor_content
		
	name_idx = method_id_item.get_name_idx(dex_file_begin_val, method_ids_off, art_method_dex_method_index_val)
	name_content = string_id_item.get_string_id_item_data(dex_file_begin_val, string_ids_off, name_idx)
	# if config.debug:
		# print "[ClassModification] method name = %s" % name_content
	
	proto_idx = method_id_item.get_proto_idx(dex_file_begin_val, method_ids_off, art_method_dex_method_index_val)
	proto_return_type_idx = proto_id_item.get_return_type_idx(dex_file_begin_val, proto_ids_off, proto_idx)
	proto_return_type_descriptor_idx = type_id_item.get_descriptor_idx(dex_file_begin_val, type_ids_off, proto_return_type_idx)
	proto_return_type_descriptor_content = string_id_item.get_string_id_item_data(dex_file_begin_val, string_ids_off, proto_return_type_descriptor_idx)
	# if config.debug:
		# print "[ClassModification] return type = %s" % proto_return_type_descriptor_content
	
	parameters_content = ""
	proto_parameters_list = proto_id_item.get_parameters_list(dex_file_begin_val, proto_ids_off, proto_idx)
	if len(proto_parameters_list) == 0:
		parameters_content = "()"
	else:
		for parameter_idx in range(len(proto_parameters_list)):
			parameter_type_idx = proto_parameters_list[parameter_idx]
			parameter_type_descriptor_idx = type_id_item.get_descriptor_idx(dex_file_begin_val, type_ids_off, parameter_type_idx)
			parameter_type_descriptor_content = string_id_item.get_string_id_item_data(dex_file_begin_val, string_ids_off, parameter_type_descriptor_idx)
		
			if len(proto_parameters_list) == 1:
				parameters_content = parameters_content + "(" + parameter_type_descriptor_content + ")"
			else:
				if parameter_idx == 0:
					parameters_content = parameters_content + "(" + parameter_type_descriptor_content
				elif parameter_idx == (len(proto_parameters_list) - 1):
					parameters_content = parameters_content + "," + parameter_type_descriptor_content + ")"
				else:
					parameters_content = parameters_content + "," + parameter_type_descriptor_content
	# if config.debug:
		# print "[ClassModification] parameters = %s" % parameters_content
	
	caller_method_signature = "%s %s->%s %s%s" % (config.resolve_method_access_flags(art_method_access_flags_value), class_descriptor_content, proto_return_type_descriptor_content, name_content, parameters_content)
	if config.debug:
		print "[ClassModification] caller signature = %s" % caller_method_signature
	
	# -2- for callee
	
	# get the "called_method" parameter
	callee_ptr = int(execution_state.getRegisterService().getValue("R0"))
	
	# get the pointer that refers to ArtMethod
	art_method_ptr = callee_ptr
	if config.debug:
		print "[ClassModification] callee = %0#10x" % art_method_ptr
	
	# read the "declaring_class_" field of ArtMethod
	art_method_declaring_class_ptr = art_method_ptr + config.offset_ArtMethod_declaring_class_
	# read the "root_" field of GcRoot
	art_method_declaring_class_root_ptr = art_method_declaring_class_ptr + config.offset_GcRoot_root_
	# read the "refeence_" field of CompressesReference
	art_method_declaring_class_root_reference_ptr = art_method_declaring_class_root_ptr + config.offset_CompressesReference_reference_
	art_method_declaring_class_root_reference_val = memory.readMemory32(art_method_declaring_class_root_reference_ptr)
	
	class_ptr = art_method_declaring_class_root_reference_val
	callee_class_ptr = class_ptr
	
	# read the "dex_cache_" field of Class
	class_dex_cache_ptr = class_ptr + config.offset_Class_dex_cache_
	# read the "reference_" field of HeapReference
	class_dex_cache_reference_ptr = class_dex_cache_ptr + config.offset_HeapReference_reference_
	class_dex_cache_reference_val = memory.readMemory32(class_dex_cache_reference_ptr)
	dex_cache_ptr = class_dex_cache_reference_val
	# read the "dex_file_" field of DexCache
	dex_cache_dex_file_ptr = dex_cache_ptr + config.offset_DexCache_dex_file_
	dex_cache_dex_file_val = memory.readMemory32(dex_cache_dex_file_ptr)
	
	dex_file_ptr = dex_cache_dex_file_val
	
	# read the "begin_" field of DexFile
	dex_file_begin_ptr = dex_file_ptr + config.offset_DexFile_begin_
	dex_file_begin_val = memory.readMemory32(dex_file_begin_ptr)
	if config.debug:
		print "[ClassModification] ArtMethod::declaring_class_::dex_cache_::dex_file_::begin_ = %0#10x" % dex_file_begin_val
		
	# read the "size_" field of DexFile
	dex_file_size_ptr = dex_file_ptr + config.offset_DexFile_size_
	dex_file_size_val = memory.readMemory32(dex_file_size_ptr)
	if config.debug:
		print "[ClassModification] ArtMethod::declaring_class_::dex_cache_::dex_file_::size_ = %#x" % dex_file_size_val

	# read the "location_" field of DexFile
	dex_file_location_ptr = dex_file_ptr + config.offset_DexFile_location_
	# retrieve the value of std::string
	dex_file_location_string_val = retrieve_string_value(dex_file_location_ptr)
	if config.debug:
		print "[ClassModification] ArtMethod::declaring_class_::dex_cache_::dex_file_::location_ = %s" % dex_file_location_string_val
	
	# we only focus on the Java invocation appeared in the target package
	if not config.package_filter(dex_file_location_string_val):
		# continue the execution of the target application
		execution_state.getExecutionService().resume()
		cleanup()
		return
	
	# read the "access_flags_" field of ArtMethod
	art_method_access_flags_ptr = art_method_ptr + config.offset_ArtMethod_access_flags_
	art_method_access_flags_value = memory.readMemory32(art_method_access_flags_ptr)
	if config.debug:
		print "[ClassModification] ArtMethod::access_flags_ = %#x (%s)" % (art_method_access_flags_value, config.resolve_access_flags(art_method_access_flags_value))
	
	# we only focus on the Java -> native invocation
	if not (art_method_access_flags_value & config.ACC_NATIVE == config.ACC_NATIVE):
		# continue the execution of the target application
		execution_state.getExecutionService().resume()
		cleanup()
		return
	
	# read the "dex_code_item_offset_" field of ArtMethod
	art_method_dex_code_item_offset_ptr = art_method_ptr + config.offset_ArtMethod_dex_code_item_offset_
	art_method_dex_code_item_offset_value = memory.readMemory32(art_method_dex_code_item_offset_ptr)
	if config.debug:
		print "[ClassModification] ArtMethod::dex_code_item_offset_ = %#x" % art_method_dex_code_item_offset_value
	
	# read the "dex_method_index_" field of ArtMethod
	art_method_dex_method_index_ptr = art_method_ptr + config.offset_ArtMethod_dex_method_index_
	art_method_dex_method_index_val = memory.readMemory32(art_method_dex_method_index_ptr)
	if config.debug:
		print "[ClassModification] ArtMethod::dex_method_index_ = %#x" % art_method_dex_method_index_val
		
	# resolve 
	string_ids_off = header_item.get_string_ids_off(dex_file_begin_val)
	type_ids_off = header_item.get_type_ids_off(dex_file_begin_val)
	proto_ids_off = header_item.get_proto_ids_off(dex_file_begin_val)
	method_ids_off = header_item.get_method_ids_off(dex_file_begin_val)
	
	class_idx = method_id_item.get_class_idx(dex_file_begin_val, method_ids_off, art_method_dex_method_index_val)
	class_descriptor_idx = type_id_item.get_descriptor_idx(dex_file_begin_val, type_ids_off, class_idx)
	class_descriptor_content = string_id_item.get_string_id_item_data(dex_file_begin_val, string_ids_off, class_descriptor_idx)
	# if config.debug:
		# print "[ClassModification] class name = %s" % class_descriptor_content
		
	name_idx = method_id_item.get_name_idx(dex_file_begin_val, method_ids_off, art_method_dex_method_index_val)
	name_content = string_id_item.get_string_id_item_data(dex_file_begin_val, string_ids_off, name_idx)
	# if config.debug:
		# print "[ClassModification] method name = %s" % name_content
	
	proto_idx = method_id_item.get_proto_idx(dex_file_begin_val, method_ids_off, art_method_dex_method_index_val)
	proto_return_type_idx = proto_id_item.get_return_type_idx(dex_file_begin_val, proto_ids_off, proto_idx)
	proto_return_type_descriptor_idx = type_id_item.get_descriptor_idx(dex_file_begin_val, type_ids_off, proto_return_type_idx)
	proto_return_type_descriptor_content = string_id_item.get_string_id_item_data(dex_file_begin_val, string_ids_off, proto_return_type_descriptor_idx)
	# if config.debug:
		# print "[ClassModification] return type = %s" % proto_return_type_descriptor_content
	
	parameters_content = ""
	proto_parameters_list = proto_id_item.get_parameters_list(dex_file_begin_val, proto_ids_off, proto_idx)
	if len(proto_parameters_list) == 0:
		parameters_content = "()"
	else:
		for parameter_idx in range(len(proto_parameters_list)):
			parameter_type_idx = proto_parameters_list[parameter_idx]
			parameter_type_descriptor_idx = type_id_item.get_descriptor_idx(dex_file_begin_val, type_ids_off, parameter_type_idx)
			parameter_type_descriptor_content = string_id_item.get_string_id_item_data(dex_file_begin_val, string_ids_off, parameter_type_descriptor_idx)
		
			if len(proto_parameters_list) == 1:
				parameters_content = parameters_content + "(" + parameter_type_descriptor_content + ")"
			else:
				if parameter_idx == 0:
					parameters_content = parameters_content + "(" + parameter_type_descriptor_content
				elif parameter_idx == (len(proto_parameters_list) - 1):
					parameters_content = parameters_content + "," + parameter_type_descriptor_content + ")"
				else:
					parameters_content = parameters_content + "," + parameter_type_descriptor_content
	# if config.debug:
		# print "[ClassModification] parameters = %s" % parameters_content
	
	callee_method_signature = "%s %s->%s %s%s" % (config.resolve_method_access_flags(art_method_access_flags_value), class_descriptor_content, proto_return_type_descriptor_content, name_content, parameters_content)
	if config.debug:
		print "[ClassModification] callee signature = %s" % callee_method_signature
	
	config.log_print("[ClassModification]          caller signature = %s" % caller_method_signature)
	config.log_print("[ClassModification] previous callee signature = %s" % callee_method_signature)
	
	# resolve the mirror::Class structure
	origin_method_list = resolve_mirror_Class(caller_class_ptr)
	suspicious_method_list = None  # we will fill this list afterwards
	
	# -2- GenericJniMethodEnd
	
	# disable the DoCall breakpoint to avoid noisy
	for breakpoint_index in range(0, execution_state.getBreakpointService().getBreakpointCount()):
		breakpoint_object = execution_state.getBreakpointService().getBreakpoint(breakpoint_index)
		if (int(breakpoint_object.getAddresses()[0]) & 0xffffffff) == config.brk_DoCall:
			breakpoint_object.disable()
			
	# resume to the start address of the GenericJniMethodEnd method
	previous_callee_method_signature = callee_method_signature
	brk_GenericJniMethodEnd	= config.libart_base + config.GenericJniMethodEnd_start - config.libart_file_offset + config.libart_memory_offset
	while True:
		execution_state.getExecutionService().resumeTo(brk_GenericJniMethodEnd)
		try:
			execution_state.getExecutionService().waitForStop(10 * 60 * 1000)  # wait for 10mins
		except DebugException:
			raise RuntimeError("wtf !!!")
		
		# get the pointer that refers to ArtMethod*
		art_method_ptr_ptr = int(execution_state.getRegisterService().getValue("SP")) + 0x8
		# get the pointer that refers to ArtMethod
		art_method_ptr = memory.readMemory32(art_method_ptr_ptr)
		
		# read the "declaring_class_" field of ArtMethod
		art_method_declaring_class_ptr = art_method_ptr + config.offset_ArtMethod_declaring_class_
		# read the "root_" field of GcRoot
		art_method_declaring_class_root_ptr = art_method_declaring_class_ptr + config.offset_GcRoot_root_
		# read the "reference_" field of CompressesReference
		art_method_declaring_class_root_reference_ptr = art_method_declaring_class_root_ptr + config.offset_CompressesReference_reference_
		art_method_declaring_class_root_reference_val = memory.readMemory32(art_method_declaring_class_root_reference_ptr)
		
		class_ptr = art_method_declaring_class_root_reference_val
		
		# read the "dex_cache_" field of Class
		class_dex_cache_ptr = class_ptr + config.offset_Class_dex_cache_
		# read the "reference_" field of HeapReference
		class_dex_cache_reference_ptr = class_dex_cache_ptr + config.offset_HeapReference_reference_
		class_dex_cache_reference_val = memory.readMemory32(class_dex_cache_reference_ptr)
		dex_cache_ptr = class_dex_cache_reference_val
		# read the "dex_file_" field of DexCache
		dex_cache_dex_file_ptr = dex_cache_ptr + config.offset_DexCache_dex_file_
		dex_cache_dex_file_val = memory.readMemory32(dex_cache_dex_file_ptr)
		
		dex_file_ptr = dex_cache_dex_file_val
		
		# read the "begin_" field of DexFile
		dex_file_begin_ptr = dex_file_ptr + config.offset_DexFile_begin_
		dex_file_begin_val = memory.readMemory32(dex_file_begin_ptr)
		if config.debug:
			print "[ClassModification] ArtMethod::declaring_class_::dex_cache_::dex_file_::begin_ = %0#10x" % dex_file_begin_val
			
		callee_method_signature = "unknown"
		if dex_file_begin_val == 0x0:
			pass
		else:
			# read the "size_" field of DexFile
			dex_file_size_ptr = dex_file_ptr + config.offset_DexFile_size_
			dex_file_size_val = memory.readMemory32(dex_file_size_ptr)
			if config.debug:
				print "[ClassModification] ArtMethod::declaring_class_::dex_cache_::dex_file_::size_ = %#x" % dex_file_size_val

			# read the "location_" field of DexFile
			dex_file_location_ptr = dex_file_ptr + config.offset_DexFile_location_
			# retrieve the value of std::string
			dex_file_location_string_val = retrieve_string_value(dex_file_location_ptr)
			if config.debug:
				print "[ClassModification] ArtMethod::declaring_class_::dex_cache_::dex_file_::location_ = %s" % dex_file_location_string_val
			
			# we focus on the JNI method defined in the target packages
			if not config.package_filter(dex_file_location_string_val):
				# continue the while-loop
				continue
				
			# read the "access_flags_" field of ArtMethod
			art_method_access_flags_ptr = art_method_ptr + config.offset_ArtMethod_access_flags_
			art_method_access_flags_value = memory.readMemory32(art_method_access_flags_ptr)
			if config.debug:
				print "[ClassModification] ArtMethod::access_flags_ = %#x (%s)" % (art_method_access_flags_value, config.resolve_access_flags(art_method_access_flags_value))
				
			# read the "dex_code_item_offset_" field of ArtMethod
			art_method_dex_code_item_offset_ptr = art_method_ptr + config.offset_ArtMethod_dex_code_item_offset_
			art_method_dex_code_item_offset_value = memory.readMemory32(art_method_dex_code_item_offset_ptr)
			if config.debug:
				print "[ClassModification] ArtMethod::dex_code_item_offset_ = %#x" % art_method_dex_code_item_offset_value
			
			# read the "dex_method_index_" field of ArtMethod
			art_method_dex_method_index_ptr = art_method_ptr + config.offset_ArtMethod_dex_method_index_
			art_method_dex_method_index_val = memory.readMemory32(art_method_dex_method_index_ptr)
			if config.debug:
				print "[ClassModification] ArtMethod::dex_method_index_ = %#x" % art_method_dex_method_index_val
				
			# resolve 
			string_ids_off = header_item.get_string_ids_off(dex_file_begin_val)
			type_ids_off = header_item.get_type_ids_off(dex_file_begin_val)
			proto_ids_off = header_item.get_proto_ids_off(dex_file_begin_val)
			method_ids_off = header_item.get_method_ids_off(dex_file_begin_val)
			
			class_idx = method_id_item.get_class_idx(dex_file_begin_val, method_ids_off, art_method_dex_method_index_val)
			class_descriptor_idx = type_id_item.get_descriptor_idx(dex_file_begin_val, type_ids_off, class_idx)
			class_descriptor_content = string_id_item.get_string_id_item_data(dex_file_begin_val, string_ids_off, class_descriptor_idx)
			# if config.debug:
				# print "[ClassModification] class name = %s" % class_descriptor_content
				
			name_idx = method_id_item.get_name_idx(dex_file_begin_val, method_ids_off, art_method_dex_method_index_val)
			name_content = string_id_item.get_string_id_item_data(dex_file_begin_val, string_ids_off, name_idx)
			# if config.debug:
				# print "[ClassModification] method name = %s" % name_content
			
			proto_idx = method_id_item.get_proto_idx(dex_file_begin_val, method_ids_off, art_method_dex_method_index_val)
			proto_return_type_idx = proto_id_item.get_return_type_idx(dex_file_begin_val, proto_ids_off, proto_idx)
			proto_return_type_descriptor_idx = type_id_item.get_descriptor_idx(dex_file_begin_val, type_ids_off, proto_return_type_idx)
			proto_return_type_descriptor_content = string_id_item.get_string_id_item_data(dex_file_begin_val, string_ids_off, proto_return_type_descriptor_idx)
			# if config.debug:
				# print "[ClassModification] return type = %s" % proto_return_type_descriptor_content
			
			parameters_content = ""
			proto_parameters_list = proto_id_item.get_parameters_list(dex_file_begin_val, proto_ids_off, proto_idx)
			if len(proto_parameters_list) == 0:
				parameters_content = "()"
			else:
				for parameter_idx in range(len(proto_parameters_list)):
					parameter_type_idx = proto_parameters_list[parameter_idx]
					parameter_type_descriptor_idx = type_id_item.get_descriptor_idx(dex_file_begin_val, type_ids_off, parameter_type_idx)
					parameter_type_descriptor_content = string_id_item.get_string_id_item_data(dex_file_begin_val, string_ids_off, parameter_type_descriptor_idx)
				
					if len(proto_parameters_list) == 1:
						parameters_content = parameters_content + "(" + parameter_type_descriptor_content + ")"
					else:
						if parameter_idx == 0:
							parameters_content = parameters_content + "(" + parameter_type_descriptor_content
						elif parameter_idx == (len(proto_parameters_list) - 1):
							parameters_content = parameters_content + "," + parameter_type_descriptor_content + ")"
						else:
							parameters_content = parameters_content + "," + parameter_type_descriptor_content
			# if config.debug:
				# print "[ClassModification] parameters = %s" % parameters_content
			
			callee_method_signature = "%s %s->%s %s%s" % (config.resolve_method_access_flags(art_method_access_flags_value), class_descriptor_content, proto_return_type_descriptor_content, name_content, parameters_content)
			if config.debug:
				print "[ClassModification] method signature = %s" % callee_method_signature
		
		config.log_print("[ClassModification] current  callee signature = %s" % callee_method_signature)
		
		if callee_method_signature == previous_callee_method_signature:
			# resolve the mirror::Class structure
			suspicious_method_list = resolve_mirror_Class(caller_class_ptr)
			break  # jump out of the while-loop
	
	# -3-
	
	# make a comparison between the origin_method_list and the suspicous_method_list
	assert suspicious_method_list is not None
	for idx in range(len(origin_method_list)):
		origin_method = origin_method_list[idx]
		suspicious_method = suspicious_method_list[idx]
		if origin_method != suspicious_method:
			assert origin_method[0] == suspicious_method[0]
			assert origin_method[3] == suspicious_method[3]
			
			# read the "declaring_class_" field of ArtMethod
			art_method_declaring_class_ptr = origin_method[0]
			# read the "root_" field of GcRoot
			art_method_declaring_class_root_ptr = art_method_declaring_class_ptr + config.offset_GcRoot_root_
			# read the "refeence_" field of CompressesReference
			art_method_declaring_class_root_reference_ptr = art_method_declaring_class_root_ptr + config.offset_CompressesReference_reference_
			art_method_declaring_class_root_reference_val = memory.readMemory32(art_method_declaring_class_root_reference_ptr)
			
			class_ptr = art_method_declaring_class_root_reference_val
			
			# read the "dex_cache_" field of Class
			class_dex_cache_ptr = class_ptr + config.offset_Class_dex_cache_
			# read the "reference_" field of HeapReference
			class_dex_cache_reference_ptr = class_dex_cache_ptr + config.offset_HeapReference_reference_
			class_dex_cache_reference_val = memory.readMemory32(class_dex_cache_reference_ptr)
			dex_cache_ptr = class_dex_cache_reference_val
			# read the "dex_file_" field of DexCache
			dex_cache_dex_file_ptr = dex_cache_ptr + config.offset_DexCache_dex_file_
			dex_cache_dex_file_val = memory.readMemory32(dex_cache_dex_file_ptr)
			
			dex_file_ptr = dex_cache_dex_file_val
			
			# read the "begin_" field of DexFile
			dex_file_begin_ptr = dex_file_ptr + config.offset_DexFile_begin_
			dex_file_begin_val = memory.readMemory32(dex_file_begin_ptr)
			
			art_method_dex_method_index_val = origin_method[3]
			
			# resolve 
			string_ids_off = header_item.get_string_ids_off(dex_file_begin_val)
			type_ids_off = header_item.get_type_ids_off(dex_file_begin_val)
			proto_ids_off = header_item.get_proto_ids_off(dex_file_begin_val)
			method_ids_off = header_item.get_method_ids_off(dex_file_begin_val)
			
			class_idx = method_id_item.get_class_idx(dex_file_begin_val, method_ids_off, art_method_dex_method_index_val)
			class_descriptor_idx = type_id_item.get_descriptor_idx(dex_file_begin_val, type_ids_off, class_idx)
			class_descriptor_content = string_id_item.get_string_id_item_data(dex_file_begin_val, string_ids_off, class_descriptor_idx)
				
			name_idx = method_id_item.get_name_idx(dex_file_begin_val, method_ids_off, art_method_dex_method_index_val)
			name_content = string_id_item.get_string_id_item_data(dex_file_begin_val, string_ids_off, name_idx)
			
			proto_idx = method_id_item.get_proto_idx(dex_file_begin_val, method_ids_off, art_method_dex_method_index_val)
			proto_return_type_idx = proto_id_item.get_return_type_idx(dex_file_begin_val, proto_ids_off, proto_idx)
			proto_return_type_descriptor_idx = type_id_item.get_descriptor_idx(dex_file_begin_val, type_ids_off, proto_return_type_idx)
			proto_return_type_descriptor_content = string_id_item.get_string_id_item_data(dex_file_begin_val, string_ids_off, proto_return_type_descriptor_idx)
			
			parameters_content = ""
			proto_parameters_list = proto_id_item.get_parameters_list(dex_file_begin_val, proto_ids_off, proto_idx)
			if len(proto_parameters_list) == 0:
				parameters_content = "()"
			else:
				for parameter_idx in range(len(proto_parameters_list)):
					parameter_type_idx = proto_parameters_list[parameter_idx]
					parameter_type_descriptor_idx = type_id_item.get_descriptor_idx(dex_file_begin_val, type_ids_off, parameter_type_idx)
					parameter_type_descriptor_content = string_id_item.get_string_id_item_data(dex_file_begin_val, string_ids_off, parameter_type_descriptor_idx)
				
					if len(proto_parameters_list) == 1:
						parameters_content = parameters_content + "(" + parameter_type_descriptor_content + ")"
					else:
						if parameter_idx == 0:
							parameters_content = parameters_content + "(" + parameter_type_descriptor_content
						elif parameter_idx == (len(proto_parameters_list) - 1):
							parameters_content = parameters_content + "," + parameter_type_descriptor_content + ")"
						else:
							parameters_content = parameters_content + "," + parameter_type_descriptor_content
			
			method_signature = "%s %s->%s %s%s" % (config.resolve_method_access_flags(art_method_access_flags_value), class_descriptor_content, proto_return_type_descriptor_content, name_content, parameters_content)
			
			config.log_print("[ClassModification] mirror::Class has been modified !!!")
			config.log_print("[ClassModification][origin]     signature = %s, access_flags_ = %#x, dex_code_item_offset_ = %0#10x, entry_point_from_quick_compiled_code = %0#10x" % (method_signature, origin_method[1], origin_method[2], origin_method[8]))
			config.log_print("[ClassModification][suspicious] signature = %s, access_flags_ = %#x, dex_code_item_offset_ = %0#10x, entry_point_from_quick_compiled_code = %0#10x" % (method_signature, suspicious_method[1], suspicious_method[2], suspicious_method[8]))
			
			if origin_method[2] != suspicious_method[2]:
				# dump the code_item on-demand
				tries_size_off = ((dex_file_begin_val + suspicious_method[2]) & 0xffffffff) + config.offset_DexFile_CodeItem_tries_size_
				print "[ClassModification] tries_size_off = %#x" % tries_size_off
				tries_size_val = memory.readMemory16(tries_size_off)
				code_item_insns_size_in_code_units_off = ((dex_file_begin_val + suspicious_method[2]) & 0xffffffff) + config.offset_DexFile_CodeItem_insns_size_in_code_units_
				print "[ClassModification] code_item_insns_size_in_code_units_off = %#x" % code_item_insns_size_in_code_units_off
				code_item_insns_size_in_code_units_val = memory.readMemory32(code_item_insns_size_in_code_units_off)
				padding = 0x2 if (tries_size_val > 0) and (code_item_insns_size_in_code_units_val % 2 == 1) else 0x0
				code_item_size = config.offset_DexFile_CodeItem_insns_size_in_code_units_ + 0x4 + 0x2 * code_item_insns_size_in_code_units_val + padding + 0x8 * tries_size_val
				code_item_size = code_item_size + encoded_catch_handler_list.parse_encoded_catch_handler_list(dex_file_begin_val, suspicious_method[2] + code_item_size, tries_size_val)
				file_path = os.path.join(config.workspace, config.dex_directory, "code_item_%#x_%0#10x.bin" % (origin_method[3], origin_method[2]))
				file_format = "binary"
				file_vtl_start_address = (dex_file_begin_val + suspicious_method[2]) & 0xffffffff
				file_vtl_end_address = ((dex_file_begin_val + suspicious_method[2]) & 0xffffffff) + code_item_size - 0x1
				memory.dump(file_path, file_format, file_vtl_start_address, file_vtl_end_address)
	
	# enable the DoCall breakpoint
	for breakpoint_index in range(0, execution_state.getBreakpointService().getBreakpointCount()):
		breakpoint_object = execution_state.getBreakpointService().getBreakpoint(breakpoint_index)
		if (int(breakpoint_object.getAddresses()[0]) & 0xffffffff) == config.brk_DoCall:
			breakpoint_object.enable()
	
	'''	
	# enable the InitClass breakpoint
	for breakpoint_index in range(0, execution_state.getBreakpointService().getBreakpointCount()):
		breakpoint_object = execution_state.getBreakpointService().getBreakpoint(breakpoint_index)
		if (int(breakpoint_object.getAddresses()[0]) & 0xffffffff) == config.brk_InitClass:
			breakpoint_object.enable()
	'''
	
	# continue the execution of the target application
	execution_state.getExecutionService().resume()
	cleanup()
	return
Exemple #12
0
def dump_dex_memory(dex_file_base, dex_file_size, dex_file_name):
    file_path = os.path.join(config.workspace, config.dex_directory,
                             dex_file_name)
    file_format = "binary"
    file_vtl_start_address = dex_file_base
    file_vtl_end_address = dex_file_base + dex_file_size - 0x1
    file_path = memory.dump(file_path, file_format, file_vtl_start_address,
                            file_vtl_end_address)

    while not os.path.exists(file_path):
        time.sleep(1)

    dex = Dex.Dex(file_path)
    for class_def_item in dex.class_defs.class_def_items:
        class_data_item = class_def_item.class_data_item
        # direct_method
        method_idx = None
        for direct_method in class_data_item.direct_methods:
            method_idx_diff = direct_method.method_idx_diff
            if method_idx is None:
                method_idx = method_idx_diff
            else:
                method_idx = method_idx + method_idx_diff

            code_off = direct_method.code_off
            if (code_off > dex_file_size) or (code_off < 0):
                # dump the code_item on-demand
                tries_size_off = (
                    (dex_file_base + code_off)
                    & 0xffffffff) + config.offset_DexFile_CodeItem_tries_size_
                tries_size_val = memory.readMemory16(tries_size_off)
                code_item_insns_size_in_code_units_off = (
                    (dex_file_base + code_off) & 0xffffffff
                ) + config.offset_DexFile_CodeItem_insns_size_in_code_units_
                code_item_insns_size_in_code_units_val = memory.readMemory32(
                    code_item_insns_size_in_code_units_off)
                padding = 0x2 if (tries_size_val > 0) and (
                    code_item_insns_size_in_code_units_val % 2 == 1) else 0x0
                code_item_size = config.offset_DexFile_CodeItem_insns_size_in_code_units_ + 0x4 + 0x2 * code_item_insns_size_in_code_units_val + padding + 0x8 * tries_size_val
                code_item_size = code_item_size + encoded_catch_handler_list.parse_encoded_catch_handler_list(
                    dex_file_base, code_off + code_item_size, tries_size_val)
                file_path = os.path.join(
                    config.workspace, config.dex_directory,
                    "code_item_%#x_%0#10x.bin" % (method_idx, code_off))
                file_format = "binary"
                file_vtl_start_address = (dex_file_base +
                                          code_off) & 0xffffffff
                file_vtl_end_address = ((dex_file_base + code_off)
                                        & 0xffffffff) + code_item_size - 0x1
                if not os.path.exists(file_path):
                    memory.dump(file_path, file_format, file_vtl_start_address,
                                file_vtl_end_address)
        # virtual_method
        method_idx = None
        for virtual_method in class_data_item.virtual_methods:
            method_idx_diff = virtual_method.method_idx_diff
            if method_idx is None:
                method_idx = method_idx_diff
            else:
                method_idx = method_idx + method_idx_diff

            code_off = virtual_method.code_off
            if (code_off > dex_file_size) or (code_off < 0):
                # dump the code_item on-demand
                tries_size_off = (
                    (dex_file_base + code_off)
                    & 0xffffffff) + config.offset_DexFile_CodeItem_tries_size_
                tries_size_val = memory.readMemory16(tries_size_off)
                code_item_insns_size_in_code_units_off = (
                    (dex_file_base + code_off) & 0xffffffff
                ) + config.offset_DexFile_CodeItem_insns_size_in_code_units_
                code_item_insns_size_in_code_units_val = memory.readMemory32(
                    code_item_insns_size_in_code_units_off)
                padding = 0x2 if (tries_size_val > 0) and (
                    code_item_insns_size_in_code_units_val % 2 == 1) else 0x0
                code_item_size = config.offset_DexFile_CodeItem_insns_size_in_code_units_ + 0x4 + 0x2 * code_item_insns_size_in_code_units_val + padding + 0x8 * tries_size_val
                code_item_size = code_item_size + encoded_catch_handler_list.parse_encoded_catch_handler_list(
                    dex_file_base, code_off + code_item_size, tries_size_val)
                file_path = os.path.join(
                    config.workspace, config.dex_directory,
                    "code_item_%#x_%0#10x.bin" % (method_idx, code_off))
                file_format = "binary"
                file_vtl_start_address = (dex_file_base +
                                          code_off) & 0xffffffff
                file_vtl_end_address = ((dex_file_base + code_off)
                                        & 0xffffffff) + code_item_size - 0x1
                if not os.path.exists(file_path):
                    memory.dump(file_path, file_format, file_vtl_start_address,
                                file_vtl_end_address)