1 files changed, 21 insertions, 13 deletions

diff --git a/src/main/java/cuchaz/enigma/analysis/ParsedJar.java b/src/main/java/cuchaz/enigma/analysis/ParsedJar.java
index ad3aceb..ddcda3e 100644
--- a/src/main/java/cuchaz/enigma/analysis/ParsedJar.java
+++ b/src/main/java/cuchaz/enigma/analysis/ParsedJar.java
@@ -12,9 +12,12 @@
 package cuchaz.enigma.analysis;
 
 import com.google.common.io.ByteStreams;
+import cuchaz.enigma.CompiledSource;
+import cuchaz.enigma.bytecode.translators.LocalVariableFixVisitor;
 import cuchaz.enigma.translation.representation.entry.ClassEntry;
 import org.objectweb.asm.ClassReader;
 import org.objectweb.asm.ClassVisitor;
+import org.objectweb.asm.Opcodes;
 import org.objectweb.asm.tree.ClassNode;
 
 import javax.annotation.Nullable;
@@ -28,12 +31,12 @@ import java.util.jar.JarEntry;
 import java.util.jar.JarFile;
 import java.util.jar.JarInputStream;
 
-public class ParsedJar {
+public class ParsedJar implements CompiledSource {
         private final Map<String, byte[]> classBytes;
         private final Map<String, ClassNode> nodeCache = new HashMap<>();
 
         public ParsedJar(JarFile jar) throws IOException {
-                Map<String, byte[]> uClassBytes = new LinkedHashMap<>();;
+                Map<String, byte[]> uClassBytes = new LinkedHashMap<>();
                 try {
                         // get the jar entries that correspond to classes
                         Enumeration<JarEntry> entries = jar.entries();
@@ -93,29 +96,34 @@ public class ParsedJar {
                 return classBytes.size();
         }
 
-        public List<ClassEntry> getClassEntries() {
-                List<ClassEntry> entries = new ArrayList<>(classBytes.size());
-                for (String s : classBytes.keySet()) {
-                        entries.add(new ClassEntry(s));
-                }
-                return entries;
-        }
-
         @Nullable
+        @Override
         public ClassNode getClassNode(String name) {
                 return nodeCache.computeIfAbsent(name, (n) -> {
                         byte[] bytes = classBytes.get(name);
                         if (bytes == null) {
                                 return null;
                         }
+
                         ClassReader reader = new ClassReader(bytes);
                         ClassNode node = new ClassNode();
-                        reader.accept(node, 0);
+
+                        LocalVariableFixVisitor visitor = new LocalVariableFixVisitor(Opcodes.ASM5, node);
+                        reader.accept(visitor, 0);
+
                         return node;
                 });
         }
 
-    public Map<String, byte[]> getClassDataMap() {
+        public List<ClassEntry> getClassEntries() {
+                List<ClassEntry> entries = new ArrayList<>(classBytes.size());
+                for (String s : classBytes.keySet()) {
+                        entries.add(new ClassEntry(s));
+                }
+                return entries;
+        }
+
+        public Map<String, byte[]> getClassDataMap() {
                 return classBytes;
-    }
+        }
 }