1 files changed, 11 insertions, 9 deletions

diff --git a/src/cuchaz/enigma/bytecode/InnerClassWriter.java b/src/cuchaz/enigma/bytecode/InnerClassWriter.java
index a061792..5e59307 100644
--- a/src/cuchaz/enigma/bytecode/InnerClassWriter.java
+++ b/src/cuchaz/enigma/bytecode/InnerClassWriter.java
@@ -18,6 +18,7 @@ import javassist.bytecode.ConstPool;
 import javassist.bytecode.Descriptor;
 import javassist.bytecode.EnclosingMethodAttribute;
 import javassist.bytecode.InnerClassesAttribute;
+import cuchaz.enigma.Constants;
 import cuchaz.enigma.analysis.JarIndex;
 import cuchaz.enigma.mapping.BehaviorEntry;
 import cuchaz.enigma.mapping.ClassEntry;
@@ -33,21 +34,21 @@ public class InnerClassWriter
         
         public void write( CtClass c )
         {
-                // get the outer class name
-                String obfClassName = Descriptor.toJvmName( c.getName() );
-                String obfOuterClassName = m_jarIndex.getOuterClass( obfClassName );
+                // is this an inner or outer class?
+                String obfInnerClassName = new ClassEntry( Descriptor.toJvmName( c.getName() ) ).getSimpleName();
+                String obfOuterClassName = m_jarIndex.getOuterClass( obfInnerClassName );
                 if( obfOuterClassName == null )
                 {
                         // this is an outer class
-                        obfOuterClassName = obfClassName;
+                        obfOuterClassName = Descriptor.toJvmName( c.getName() );
                 }
                 else
                 {
                         // this is an inner class, rename it to outer$inner
-                        ClassEntry obfClassEntry = new ClassEntry( obfOuterClassName + "$" + new ClassEntry( obfClassName ).getSimpleName() );
+                        ClassEntry obfClassEntry = new ClassEntry( obfOuterClassName + "$" + obfInnerClassName );
                         c.setName( obfClassEntry.getName() );
                         
-                        BehaviorEntry caller = m_jarIndex.getAnonymousClassCaller( obfClassName );
+                        BehaviorEntry caller = m_jarIndex.getAnonymousClassCaller( obfInnerClassName );
                         if( caller != null )
                         {
                                 // write the enclosing method attribute
@@ -85,7 +86,7 @@ public class InnerClassWriter
                 for( String obfInnerClassName : obfInnerClassNames )
                 {
                         // get the new inner class name
-                        ClassEntry obfClassEntry = new ClassEntry( obfOuterClassName + "$" + new ClassEntry( obfInnerClassName ).getSimpleName() );
+                        ClassEntry obfClassEntry = new ClassEntry( obfOuterClassName + "$" + obfInnerClassName );
                         
                         // here's what the JVM spec says about the InnerClasses attribute
                         // append( inner, outer of inner if inner is member of outer 0 ow, name after $ if inner not anonymous 0 ow, flags ); 
@@ -114,12 +115,13 @@ public class InnerClassWriter
                                 attr.outerClass( attr.tableLength() - 1 ),
                                 attr.innerClass( attr.tableLength() - 1 ),
                                 attr.innerName( attr.tableLength() - 1 ),
-                                obfInnerClassName, obfClassEntry.getName()
+                                Constants.NonePackage + "/" + obfInnerClassName,
+                                obfClassEntry.getName()
                         ) );
                         */
                         
                         // make sure the outer class references only the new inner class names
-                        c.replaceClassName( obfInnerClassName, obfClassEntry.getName() );
+                        c.replaceClassName( Constants.NonePackage + "/" + obfInnerClassName, obfClassEntry.getName() );
                 }
         }
 }