# HG changeset patch
# User Thinker K.F. Li <thinker@codemud.net>
# Date 1311728856 -28800
# Node ID 7fcd555d802b3587ef0dcce2198576c4c1a9f1c6
# Parent  b2db11aed6b85d170cf972f6baabbbe33bda5311
Add the function for finding method with a name

diff -r b2db11aed6b8 -r 7fcd555d802b paraspace/dexfile.py
--- a/paraspace/dexfile.py	Tue Jul 26 12:46:24 2011 +0800
+++ b/paraspace/dexfile.py	Wed Jul 27 09:07:36 2011 +0800
@@ -1741,6 +1741,34 @@
 
         clone = dexfile_insert_class(self, classdef)
         return clone
+
+    ## \brief Get name string of given method.
+    @staticmethod
+    def get_method_name(method):
+        methodid = method.methodIdx
+        namestrid = methodid.nameIdx
+        namestrdata = namestrid.stringDataOff
+        name_str = namestrdata.data.data
+        return name_str
+
+    ## \brief Find the method of given method name and class definition.
+    #
+    # \param method_name is the method name.
+    # \param classdef is a _DEX_ClassDef.
+    # \return the corresponding _DEX_Method of given method_name and classdef.
+    #
+    def find_method_name(self, method_name, classdef):
+        if not classdef.classDataOffRef.is_true:
+            return
+
+        classdata = classdef.classDataOffRef.value
+        for wmethod in classdata.directMethods.items + \
+                classdata.virtualMethods.items:
+            wmethod_name = DEXFile_linked.get_method_name(wmethod)
+            if method_name == wmethod_name:
+                return wmethod
+            pass
+        pass
     pass
 
 
diff -r b2db11aed6b8 -r 7fcd555d802b paraspace/tests/dexfile_test.py
--- a/paraspace/tests/dexfile_test.py	Tue Jul 26 12:46:24 2011 +0800
+++ b/paraspace/tests/dexfile_test.py	Wed Jul 27 09:07:36 2011 +0800
@@ -375,3 +375,29 @@
     # assert dex.header.checksum == saved_checksum
     assert dex.header.checksum == 0xc5efe529
     pass
+
+
+def find_method_name_test():
+    from paraspace.dex_deptracker import prepare_dep_decls
+    
+    _install_dexfile_4_deptracker()
+    
+    all_dep_decls = prepare_dep_decls()
+    
+    srcdir = os.path.dirname(__file__)
+    srcroot = os.path.join(srcdir, '..', '..')
+    
+    fakefile_fn = os.path.join(srcroot, 'data', 'fakefile.dex')
+    fakefile_dex = dexfile.DEXFile.open(fakefile_fn)
+    fakefile_linked = \
+        dexfile.DEXFile_linked. \
+        build_dependencies(fakefile_dex, all_dep_decls)
+    
+    fakefile_def = fakefile_linked. \
+        find_class_name('Lcom/codemud/fakefile/fakefile;')
+    fakefile_cstr = fakefile_linked.find_method_name('<init>', fakefile_def)
+    assert fakefile_cstr
+
+    method_name = dexfile.DEXFile_linked.get_method_name(fakefile_cstr)
+    assert method_name == '<init>'
+    pass