示例#1
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
示例#2
0
文件: DexFile.py 项目: rewhy/happer
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
def ArtQuickToInterpreterBridge():
	# # at the very beginning, we are required to ensure that the DoCall breakpoint has been enabled
	# for idx in range(0, execution_state.getBreakpointService().getBreakpointCount()):
		# brk_object = execution_state.getBreakpointService().getBreakpoint(idx)
		# enable the DoCall breakpoint
		# if (int(brk_object.getAddresses()[0]) & 0xffffffff) == config.brk_DoCall:
			# brk_object.enable()

	# -1- for caller
	
	# get the "sp" parameter
	sp_ptr = int(execution_state.getRegisterService().getValue("R2"))
	if config.debug:
		print "[ArtQuickToInterpreterBridge] sp = %0#10x" % sp_ptr
	
	# get the pointer that refers to ArtMethod
	art_method_ptr = memory.readMemory32(sp_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
	
	caller_method_signature = "unknown"
	if class_ptr == 0x0:
		pass
	else:
		# 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 "[ArtQuickToInterpreterBridge] 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 "[ArtQuickToInterpreterBridge] 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 "[ArtQuickToInterpreterBridge] ArtMethod::declaring_class_::dex_cache_::dex_file_::location_ = %s" % dex_file_location_string_val
			
		# # we only focus on the Compiled Code -> 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 "[ArtQuickToInterpreterBridge] 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 "[ArtQuickToInterpreterBridge] 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 "[ArtQuickToInterpreterBridge] 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 "[ArtQuickToInterpreterBridge] 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 "[ArtQuickToInterpreterBridge] 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 "[ArtQuickToInterpreterBridge] 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 "[ArtQuickToInterpreterBridge] 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 "[ArtQuickToInterpreterBridge] caller signature = %s" % caller_method_signature
	
	# -2- for callee
	
	# get the "method" parameter
	method_ptr = int(execution_state.getRegisterService().getValue("R0"))
	if config.debug:
		print "[ArtQuickToInterpreterBridge] method = %0#10x" % method_ptr
		
	# get the pointer that refers to ArtMethod
	art_method_ptr = 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 "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 "[ArtQuickToInterpreterBridge] 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 "[ArtQuickToInterpreterBridge] 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 "[ArtQuickToInterpreterBridge] ArtMethod::declaring_class_::dex_cache_::dex_file_::location_ = %s" % dex_file_location_string_val
		
	# we only focus on the Compiled Code -> 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 "[ArtQuickToInterpreterBridge] 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 "[ArtQuickToInterpreterBridge] 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 "[ArtQuickToInterpreterBridge] 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 "[ArtQuickToInterpreterBridge] 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 "[ArtQuickToInterpreterBridge] 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 "[ArtQuickToInterpreterBridge] 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 "[ArtQuickToInterpreterBridge] 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 "[ArtQuickToInterpreterBridge] callee signature = %s" % callee_method_signature
	
	config.log_print("[ArtQuickToInterpreterBridge] caller signature = %s" % caller_method_signature)
	config.log_print("[ArtQuickToInterpreterBridge] callee signature = %s" % callee_method_signature)
	
	# continue the execution of the target application
	execution_state.getExecutionService().resume()
	cleanup()
	return
示例#4
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
示例#5
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
示例#6
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(
                60000)  # wait for 60s
        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]))

    # 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()

    # continue the execution of the target application
    execution_state.getExecutionService().resume()
    cleanup()
    return
def ArtInterpreterToInterpreterBridge():
    # get the "shadow_frame" parameter
    shadow_frame_ptr = int(execution_state.getRegisterService().getValue("R2"))
    if config.debug:
        print "[ArtInterpreterToInterpreterBridge] shadow_frame = %0#10x" % shadow_frame_ptr

    # -1- for caller

    # retrieve the "link_" field of ShadowFrame structure
    shadow_frame_link_ptr = memory.readMemory32(
        shadow_frame_ptr + config.offset_ShadowFrame_link_)

    caller_method_signature = None
    if memory.readMemory32(shadow_frame_link_ptr) != 0x0:
        # a previous shadow frame is existed

        # retrieve the "method_" field of ShadowFrame structure
        shadow_frame_method_ptr = memory.readMemory32(
            shadow_frame_link_ptr + config.offset_ShadowFrame_method_)

        # get the pointer that refers to ArtMethod
        art_method_ptr = shadow_frame_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

        # 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 "[ArtInterpreterToInterpreterBridge] 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 "[ArtInterpreterToInterpreterBridge] 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 "[ArtInterpreterToInterpreterBridge] 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 "[ArtInterpreterToInterpreterBridge] 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 "[ArtInterpreterToInterpreterBridge] 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 "[ArtInterpreterToInterpreterBridge] 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 "[ArtInterpreterToInterpreterBridge] 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 "[ArtInterpreterToInterpreterBridge] 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 "[ArtInterpreterToInterpreterBridge] 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 "[ArtInterpreterToInterpreterBridge] 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)
    else:
        # cannot find a previous shadow frame
        caller_method_signature = "unknown"

    if config.debug:
        print "[ArtInterpreterToInterpreterBridge] caller signature = %s" % caller_method_signature

    # -2- for callee

    # 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

    # 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 "[ArtInterpreterToInterpreterBridge] 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 "[ArtInterpreterToInterpreterBridge] 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 "[ArtInterpreterToInterpreterBridge] ArtMethod::declaring_class_::dex_cache_::dex_file_::location_ = %s" % dex_file_location_string_val

    # 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 "[ArtInterpreterToInterpreterBridge] 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 "[ArtInterpreterToInterpreterBridge] 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 "[ArtInterpreterToInterpreterBridge] 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 "[ArtInterpreterToInterpreterBridge] 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 "[ArtInterpreterToInterpreterBridge] 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 "[ArtInterpreterToInterpreterBridge] 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 "[ArtInterpreterToInterpreterBridge] 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 "[ArtInterpreterToInterpreterBridge] callee signature = %s" % callee_method_signature

    config.log_print(
        "[ArtInterpreterToInterpreterBridge] caller signature = %s" %
        caller_method_signature)
    config.log_print(
        "[ArtInterpreterToInterpreterBridge] callee signature = %s" %
        callee_method_signature)

    # continue the execution of the target application
    execution_state.getExecutionService().resume()
    cleanup()
    return
示例#8
0
def init_art():
    # get the PID
    pid_val = int(execution_state.getVariableService().readValue(
        "$AARCH64::$System::$Memory::$CONTEXTIDR_EL1.PROCID")) & 0xffffffff

    # -1-
    # read the "name" parameter
    name_ptr = int(
        execution_state.getRegisterService().getValue("R0")) & 0xffffffff
    name_val = memory.retrieve_char_array(name_ptr)
    # read the "flags" parameter
    flags_val = int(
        execution_state.getRegisterService().getValue("R1")) & 0xffffffff
    # read the "extinfo" parameter
    extinfo_ptr = int(
        execution_state.getRegisterService().getValue("R2")) & 0xffffffff

    if config.debug:
        print "[entrypoint] [do_dlopen] pid = %#x" % pid_val
        print "[entrypoint] [do_dlopen] name = %s" % name_val
        print "[entrypoint] [do_dlopen] flags_val = %#x" % flags_val
        print "[entrypoint] [do_dlopen] extinfo_ptr = %0#10x" % extinfo_ptr

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

    config.log_print("[entrypoint] [do_dlopen] pid = %#x" % pid_val)

    # -2-
    # goto the invocation of the find_library method
    brk_find_library_offset = config.linker_base + find_library_offset - config.linker_file_offset + config.linker_memory_offset
    execution_state.getExecutionService().resumeTo(brk_find_library_offset)
    try:
        execution_state.getExecutionService().waitForStop(
            120000)  # wait for 120s
    except DebugException:
        raise RuntimeError("wtf !!!")

    # retrieve soinfo pointer
    si_ptr = int(
        execution_state.getRegisterService().getValue("R0")) & 0xffffffff
    if config.debug:
        print "[entrypoint] [do_dlopen] si = %0#10x" % si_ptr

    base_ptr = si_ptr + config.offset_soinfo_base
    base_val = memory.readMemory32(base_ptr)
    size_ptr = si_ptr + config.offset_soinfo_size
    size_val = memory.readMemory32(size_ptr)

    if base_val == 0x0 or size_val == 0x0:
        # -3-
        # goto the invocation of the call_constructors method
        brk_call_constructors_offset = config.linker_base + call_constructors_offset - config.linker_file_offset + config.linker_memory_offset
        execution_state.getExecutionService().resumeTo(
            brk_call_constructors_offset)
        try:
            execution_state.getExecutionService().waitForStop(
                120000)  # wait for 120s
        except DebugException:
            raise RuntimeError("wtf !!!")

        base_ptr = si_ptr + config.offset_soinfo_base
        base_val = memory.readMemory32(base_ptr)
        size_ptr = si_ptr + config.offset_soinfo_size
        size_val = memory.readMemory32(size_ptr)

    if config.debug:
        print "[entrypoint] [do_dlopen] si->base = %0#10x" % base_val
        print "[entrypoint] [do_dlopen] si->size = %#x" % size_val

    config.log_print(
        "[entrypoint] [do_dlopen] name = %s, si->base = %0#10x, si->size = %#x"
        % (name_val, base_val, size_val))
    if not base_val == 0x0:
        config.save_range_info(base_val, base_val + size_val - 0x1)

    # initialization
    setup(pid_val)

    # continue the execution of the target application
    execution_state.getExecutionService().resume()
    cleanup()
    return
示例#9
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
def ArtQuickGenericJniTrampoline():
    # -1- for callee

    # get the pointer that refers to ArtMethod*
    art_method_ptr_ptr = int(
        execution_state.getRegisterService().getValue("R1"))
    # 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 "[ArtQuickGenericJniTrampoline] 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 "[ArtQuickGenericJniTrampoline] 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 "[ArtQuickGenericJniTrampoline] 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 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 "[ArtQuickGenericJniTrampoline] 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 "[ArtQuickGenericJniTrampoline] 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 "[ArtQuickGenericJniTrampoline] 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 "[ArtQuickGenericJniTrampoline] 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 "[ArtQuickGenericJniTrampoline] 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 "[ArtQuickGenericJniTrampoline] 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 "[ArtQuickGenericJniTrampoline] 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 "[ArtQuickGenericJniTrampoline] method signature = %s" % callee_method_signature

    config.log_print("[ArtQuickGenericJniTrampoline] callee signature = %s" %
                     callee_method_signature)

    # it is unnecessary to dump the executable segments of libc.so and libart.so here

    # # start tracing
    # if config.trace:
    # trace_cmd = "trace start DSTREAM"
    # execution_state.executeDSCommand(trace_cmd)

    # continue the execution of the target application
    execution_state.getExecutionService().resume()
    cleanup()
    return
示例#11
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