paraspace: paraspace/injection.py comparison

comparison paraspace/injection.py @ 108:18be67af7f1e

Use method redirection map for defining redirection

author	Thinker K.F. Li <thinker@codemud.net>
date	Mon, 01 Aug 2011 12:27:28 +0800
parents	4b3573d039af
children	835336632aba

comparison

equal deleted inserted replaced

-:4b3573d039af
+:18be67af7f1e
 def dexfile_insert_class(dex, class_def):
 clone = _clone_classdef(dex, class_def)
 return clone
-def method_redirect_types(dex, method, redirect_map):
+## \brief Redirect types and methods for the code of given method.
+def method_redirect_types(dex, method, types_redir, methods_redir):
 from paraspace.dalvik_opcodes import decode_insn_blk, all_opcodes
 from paraspace.dalvik_opcodes import encode_opcode_vectors
 from paraspace.dexfile import DEXFile_linked
 if not method.codeOffRef.is_true:
 op_vectors = decode_insn_blk(insns_blk)
 def redirect(opcode, args):
 if opcode == all_opcodes.OP_NEW_INSTANCE:
 typeidx = args[1]
-if typeidx in redirect_map:
+if typeidx in types_redir:
-to_type = redirect_map[typeidx]
+to_type = types_redir[typeidx]
 return opcode, (args[0], to_type)
 pass
 elif opcode == all_opcodes.OP_INVOKE_DIRECT:
 methodidx = args[2]
+if methodidx not in methods_redir:
+return opcode, args
+return opcode, (args[0], args[1], methods_redir[methodidx],
+args[3], args[4], args[5], args[6])
 methodid = dex.find_methodid_idx(methodidx)
 method_typeid = methodid.classIdx
 method_typeidx = dex.get_idx_typeid(method_typeid)
-if method_typeidx not in redirect_map:
+if method_typeidx not in types_redir:
 method_typeid = dex.find_typeid_idx(method_typeidx - 1)
 return opcode, args
-new_method_typeidx = redirect_map[method_typeidx]
+new_method_typeidx = types_redir[method_typeidx]
 new_method_typeid = dex.find_typeid_idx(new_method_typeidx)
 classdef = dex.find_class_typeid(new_method_typeid)
 method_name = DEXFile_linked.get_methodid_name(methodid)
 method_proto = methodid.protoIdx
 code.insns.data = new_insns_blk
 pass
-def class_redirect_types(dex, classdef, redirect_map):
+## \brief Make a map for methods from source type ID to ones from desinate.
+def make_redir_classes_methods_map(dex_src, typeid_src,
+dex_dst, typeid_dst):
+from paraspace.dexfile import DEXFile_linked
+methods_src = [(idx, methodid)
+for idx, methodid in enumerate(dex_src.methodIds.items)
+if methodid.classIdx == typeid_src]
+def make_map_methodid(methodid_src):
+name = DEXFile_linked.get_methodid_name(methodid_src)
+proto = methodid_src.protoIdx
+try:
+methodid_dst = \
+dex_dst.find_methodid_name_proto(name, proto, typeid_dst)
+except ValueError:
+return -1
+methodidx_dst = dex_dst.get_idx_methodid(methodid_dst)
+return methodidx_dst
+methods_map = [(methodidx_src, make_map_methodid(methodid_src))
+for methodidx_src, methodid_src in methods_src]
+methods_map = [(methodidx_src, methodidx_dst)
+for methodidx_src, methodidx_dst in methods_map
+if methodidx_dst != -1]
+methods_map = dict(methods_map)
+return methods_map
+## \brief Redirect types and methods mentioned in the code of a class.
+#
+# For code of given class definition, Every mentions of types and
+# methods are rewrote to types and methods according types_redir and
+# methods_redir respectively.
+#
+# \param dex is a DEXFile_linked.
+# \param classdef is a class definition.
+# \param types_redir is a map of types.
+# \param methods_redir is a map of methods.
+#
+def class_redirect_types(dex, classdef, types_redir, methods_redir):
 if not classdef.classDataOffRef.is_true:
 return
 classdata = classdef.classDataOffRef.value
 for method in classdata.directMethods.items:
-method_redirect_types(dex, method, redirect_map)
+method_redirect_types(dex, method, types_redir, methods_redir)
 pass
 for method in classdata.virtualMethods.items:
-method_redirect_types(dex, method, redirect_map)
+method_redirect_types(dex, method, types_redir, methods_redir)
 pass
 pass
-def dexfile_redirect_types(dex, redirect_map, excludes=set([])):
+## \brief Make a map to map methods from source types to destinate types.
+#
+# This function create a map to map methods from source types to
+# methods from destinate types in \ref types_redir.
+#
+# \param dex is a DEXFile_linked that owns source and destinate types.
+# \param types_redir is a map of types for redirecting types.
+# \return a map of method indices.
+#
+def _make_methods_redir_for_types_redir(dex, types_redir):
+methods_map = {}
+for typeidx_src, typeidx_dst in types_redir.items():
+typeid_src = dex.find_typeid_idx(typeidx_src)
+typeid_dst = dex.find_typeid_idx(typeidx_dst)
+class_methods_map =  make_redir_classes_methods_map(dex,
+typeid_src,
+dex,
+typeid_dst)
+methods_map.update(class_methods_map)
+pass
+return methods_map
+## \biref Redirect types of all code in given DEXFile_linked.
+def dexfile_redirect_types(dex, types_redir, excludes=set([])):
+methods_redir = _make_methods_redir_for_types_redir(dex, types_redir)
 for classdef in dex.classDefs.items:
 typeid = classdef.classIdx
 idx = dex.get_idx_typeid(typeid)
 if idx in excludes:
 continue
-class_redirect_types(dex, classdef, redirect_map)
+class_redirect_types(dex, classdef, types_redir, methods_redir)
 pass
 pass

Mercurial > paraspace

comparison paraspace/injection.py @ 108:18be67af7f1e