fixed in-jar detection for bridge-related methods

4 files changed, 43 insertions, 56 deletions

diff --git a/src/cuchaz/enigma/analysis/BridgeFixer.java b/src/cuchaz/enigma/analysis/BridgeFixer.java
index aeaf871..112b864 100644
--- a/src/cuchaz/enigma/analysis/BridgeFixer.java
+++ b/src/cuchaz/enigma/analysis/BridgeFixer.java
@@ -15,6 +15,8 @@ import javassist.CtMethod;
 import javassist.bytecode.ConstPool;
 import javassist.bytecode.Descriptor;
 import cuchaz.enigma.bytecode.ConstPoolEditor;
+import cuchaz.enigma.mapping.BehaviorEntry;
+import cuchaz.enigma.mapping.BehaviorEntryFactory;
 import cuchaz.enigma.mapping.ClassEntry;
 import cuchaz.enigma.mapping.MethodEntry;
 
@@ -57,17 +59,23 @@ public class BridgeFixer
                                 case ConstPool.CONST_Methodref:
                                 case ConstPool.CONST_InterfaceMethodref:
                                 {
-                                        // translate the name and type
-                                        MethodEntry methodEntry = new MethodEntry(
-                                                new ClassEntry( Descriptor.toJvmName( editor.getMemberrefClassname( i ) ) ),
+                                        BehaviorEntry behaviorEntry = BehaviorEntryFactory.create(
+                                                Descriptor.toJvmName( editor.getMemberrefClassname( i ) ),
                                                 editor.getMemberrefName( i ),
                                                 editor.getMemberrefType( i )
                                         );
-                                        MethodEntry bridgeMethodEntry = m_index.getBridgeMethod( methodEntry );
-                                        if( bridgeMethodEntry != null )
+                                        
+                                        if( behaviorEntry instanceof MethodEntry )
                                         {
-                                                // FIXIT FIXIT FIXIT FIXIT FIXIT FIXIT FIXIT
-                                                editor.changeMemberrefNameAndType( i, bridgeMethodEntry.getName(), bridgeMethodEntry.getSignature() );
+                                                MethodEntry methodEntry = (MethodEntry)behaviorEntry;
+                                                
+                                                // translate the name and type
+                                                MethodEntry bridgeMethodEntry = m_index.getBridgeMethod( methodEntry );
+                                                if( bridgeMethodEntry != null )
+                                                {
+                                                        // FIXIT FIXIT FIXIT FIXIT FIXIT FIXIT FIXIT
+                                                        editor.changeMemberrefNameAndType( i, bridgeMethodEntry.getName(), bridgeMethodEntry.getSignature() );
+                                                }
                                         }
                                 }
                                 break;
diff --git a/src/cuchaz/enigma/analysis/EntryRenamer.java b/src/cuchaz/enigma/analysis/EntryRenamer.java
index 44e0220..2d59fe9 100644
--- a/src/cuchaz/enigma/analysis/EntryRenamer.java
+++ b/src/cuchaz/enigma/analysis/EntryRenamer.java
@@ -11,7 +11,6 @@
 package cuchaz.enigma.analysis;
 
 import java.util.AbstractMap;
-import java.util.Iterator;
 import java.util.List;
 import java.util.Map;
 import java.util.Set;
@@ -80,16 +79,32 @@ public class EntryRenamer
         {
                 // for each key/value pair...
                 Set<Map.Entry<Key,Val>> entriesToAdd = Sets.newHashSet();
-                Iterator<Map.Entry<Key,Val>> iter = map.entries().iterator();
-                while( iter.hasNext() )
+                for( Map.Entry<Key,Val> entry : map.entries() )
                 {
-                        Map.Entry<Key,Val> entry = iter.next();
-                        iter.remove();
                         entriesToAdd.add( new AbstractMap.SimpleEntry<Key,Val>(
                                 renameMethodsInThing( renames, entry.getKey() ),
                                 renameMethodsInThing( renames, entry.getValue() )
                         ) );
                 }
+                map.clear();
+                for( Map.Entry<Key,Val> entry : entriesToAdd )
+                {
+                        map.put( entry.getKey(), entry.getValue() );
+                }
+        }
+        
+        public static <Key,Val> void renameMethodsInMap( Map<MethodEntry,MethodEntry> renames, Map<Key,Val> map )
+        {
+                // for each key/value pair...
+                Set<Map.Entry<Key,Val>> entriesToAdd = Sets.newHashSet();
+                for( Map.Entry<Key,Val> entry : map.entrySet() )
+                {
+                        entriesToAdd.add( new AbstractMap.SimpleEntry<Key,Val>(
+                                renameMethodsInThing( renames, entry.getKey() ),
+                                renameMethodsInThing( renames, entry.getValue() )
+                        ) );
+                }
+                map.clear();
                 for( Map.Entry<Key,Val> entry : entriesToAdd )
                 {
                         map.put( entry.getKey(), entry.getValue() );
diff --git a/src/cuchaz/enigma/analysis/JarIndex.java b/src/cuchaz/enigma/analysis/JarIndex.java
index 9f309ce..a2f6bf3 100644
--- a/src/cuchaz/enigma/analysis/JarIndex.java
+++ b/src/cuchaz/enigma/analysis/JarIndex.java
@@ -44,6 +44,7 @@ import cuchaz.enigma.Constants;
 import cuchaz.enigma.bytecode.ClassRenamer;
 import cuchaz.enigma.mapping.ArgumentEntry;
 import cuchaz.enigma.mapping.BehaviorEntry;
+import cuchaz.enigma.mapping.BehaviorEntryFactory;
 import cuchaz.enigma.mapping.ClassEntry;
 import cuchaz.enigma.mapping.ConstructorEntry;
 import cuchaz.enigma.mapping.Entry;
@@ -203,6 +204,7 @@ public class JarIndex
                 EntryRenamer.renameMethodsInMultimap( m_bridgeMethods, m_methodImplementations );
                 EntryRenamer.renameMethodsInMultimap( m_bridgeMethods, m_behaviorReferences );
                 EntryRenamer.renameMethodsInMultimap( m_bridgeMethods, m_fieldReferences );
+                EntryRenamer.renameMethodsInMap( m_bridgeMethods, m_access );
         }
         
         private void indexField( CtField field )
@@ -224,21 +226,20 @@ public class JarIndex
         private void indexBehavior( CtBehavior behavior )
         {
                 // get the behavior entry
-                String className = Descriptor.toJvmName( behavior.getDeclaringClass().getName() );
-                final BehaviorEntry behaviorEntry = getBehaviorEntry( behavior );
+                final BehaviorEntry behaviorEntry = BehaviorEntryFactory.create( behavior );
                 if( behaviorEntry instanceof MethodEntry )
                 {
                         MethodEntry methodEntry = (MethodEntry)behaviorEntry;
                         
                         // index implementation
-                        m_methodImplementations.put( className, methodEntry );
+                        m_methodImplementations.put( behaviorEntry.getClassName(), methodEntry );
                         
                         // look for bridge methods
                         CtMethod bridgedMethod = getBridgedMethod( (CtMethod)behavior );
                         if( bridgedMethod != null )
                         {
                                 MethodEntry bridgedMethodEntry = new MethodEntry(
-                                        new ClassEntry( className ),
+                                        behaviorEntry.getClassEntry(),
                                         bridgedMethod.getName(),
                                         bridgedMethod.getSignature()
                                 );
@@ -251,7 +252,7 @@ public class JarIndex
         private void indexBehaviorReferences( CtBehavior behavior )
         {
                 // index method calls
-                final BehaviorEntry behaviorEntry = getBehaviorEntry( behavior );
+                final BehaviorEntry behaviorEntry = BehaviorEntryFactory.create( behavior );
                 try
                 {
                         behavior.instrument( new ExprEditor( )
@@ -344,35 +345,6 @@ public class JarIndex
                 }
         }
         
-        private BehaviorEntry getBehaviorEntry( CtBehavior behavior )
-        {
-                String className = Descriptor.toJvmName( behavior.getDeclaringClass().getName() );
-                if( behavior instanceof CtMethod )
-                {
-                        return new MethodEntry(
-                                new ClassEntry( className ),
-                                behavior.getName(),
-                                behavior.getSignature()
-                        );
-                }
-                else if( behavior instanceof CtConstructor )
-                {
-                        boolean isStatic = behavior.getName().equals( "<clinit>" );
-                        if( isStatic )
-                        {
-                                return new ConstructorEntry( new ClassEntry( className ) );
-                        }
-                        else
-                        {
-                                return new ConstructorEntry( new ClassEntry( className ), behavior.getSignature() );
-                        }
-                }
-                else
-                {
-                        throw new IllegalArgumentException( "behavior must be a method or a constructor!" );
-                }
-        }
-        
         public ClassEntry resolveEntryClass( Entry obfEntry )
         {
                 // this entry could refer to a method on a class where the method is not actually implemented
diff --git a/src/cuchaz/enigma/analysis/SourceIndexClassVisitor.java b/src/cuchaz/enigma/analysis/SourceIndexClassVisitor.java
index b789726..5d8a383 100644
--- a/src/cuchaz/enigma/analysis/SourceIndexClassVisitor.java
+++ b/src/cuchaz/enigma/analysis/SourceIndexClassVisitor.java
@@ -25,11 +25,11 @@ import com.strobel.decompiler.languages.java.ast.TypeDeclaration;
 import com.strobel.decompiler.languages.java.ast.VariableInitializer;
 
 import cuchaz.enigma.mapping.BehaviorEntry;
+import cuchaz.enigma.mapping.BehaviorEntryFactory;
 import cuchaz.enigma.mapping.ClassEntry;
 import cuchaz.enigma.mapping.ConstructorEntry;
 import cuchaz.enigma.mapping.Entry;
 import cuchaz.enigma.mapping.FieldEntry;
-import cuchaz.enigma.mapping.MethodEntry;
 
 public class SourceIndexClassVisitor extends SourceIndexVisitor
 {
@@ -77,15 +77,7 @@ public class SourceIndexClassVisitor extends SourceIndexVisitor
         {
                 MethodDefinition def = node.getUserData( Keys.METHOD_DEFINITION );
                 ClassEntry classEntry = new ClassEntry( def.getDeclaringType().getInternalName() );
-                BehaviorEntry behaviorEntry;
-                if( def.getName().equals( "<clinit>" ) )
-                {
-                        behaviorEntry = new ConstructorEntry( classEntry );
-                }
-                else
-                {
-                        behaviorEntry = new MethodEntry( classEntry, def.getName(), def.getSignature() );
-                }
+                BehaviorEntry behaviorEntry = BehaviorEntryFactory.create( classEntry, def.getName(), def.getSignature() );
                 index.addDeclaration( node.getNameToken(), behaviorEntry );
                 return node.acceptVisitor( new SourceIndexBehaviorVisitor( behaviorEntry ), index );
         }