Build Recompile Environment

Author: Ecdcaeb

src/main/java/cn/enaium/joe/JavaOctetEditor.java

@@ -41,6 +41,7 @@
  * @author Enaium
  */
 public class JavaOctetEditor {
+    public static final ClassLoader classLoader = JavaOctetEditor.class.getClassLoader();
     private static JavaOctetEditor instance;
 
     public static final String TITLE = "JavaOctetEditor";
@@ -143,4 +144,13 @@ public void setJar(Jar jar) {
     public static JavaOctetEditor getInstance() {
         return instance;
     }
+
+    public static boolean isClassExist(String canonicalName){
+        try {
+            Class.forName(canonicalName, false, classLoader);
+            return true;
+        } catch (ClassNotFoundException e) {
+            return false;
+        }
+    }
 }

src/main/java/cn/enaium/joe/gui/panel/file/tabbed/tab/classes/ASMifierTablePanel.java

@@ -17,21 +17,19 @@
 package cn.enaium.joe.gui.panel.file.tabbed.tab.classes;
 
 import cn.enaium.joe.JavaOctetEditor;
-import cn.enaium.joe.compiler.Compiler;
+import cn.enaium.joe.util.compiler.Compiler;
 import cn.enaium.joe.config.extend.KeymapConfig;
 import cn.enaium.joe.event.events.EditSaveSuccessEvent;
 import cn.enaium.joe.gui.panel.CodeAreaPanel;
 import cn.enaium.joe.util.*;
+import cn.enaium.joe.util.classes.ASMClassLoader;
 import org.fife.ui.rsyntaxtextarea.SyntaxConstants;
 import org.objectweb.asm.ClassReader;
 import org.objectweb.asm.tree.ClassNode;
 import org.objectweb.asm.util.ASMifier;
 import org.objectweb.asm.util.TraceClassVisitor;
 
-import javax.swing.*;
 import java.awt.*;
-import java.awt.event.InputEvent;
-import java.awt.event.KeyEvent;
 import java.io.PrintWriter;
 import java.io.StringWriter;
 
@@ -46,7 +44,9 @@ public ASMifierTablePanel(ClassNode classNode) {
         CodeAreaPanel codeAreaPanel = this.codeAreaPanel = new CodeAreaPanel() {{
             KeyStrokeUtil.register(getTextArea(), JavaOctetEditor.getInstance().config.getByClass(KeymapConfig.class).save.getValue(), () -> {
                 try {
-                    String stringBuilder = "import org.objectweb.asm.AnnotationVisitor;" +
+                    String className = "ASMifier" + Integer.toHexString(classNode.name.hashCode()) + Integer.toHexString(getTextArea().getText().hashCode());
+                    String stringBuilder =
+                            "import org.objectweb.asm.AnnotationVisitor;" +
                             "import org.objectweb.asm.Attribute;" +
                             "import org.objectweb.asm.ClassReader;" +
                             "import org.objectweb.asm.ClassWriter;" +
@@ -60,26 +60,15 @@ public ASMifierTablePanel(ClassNode classNode) {
                             "import org.objectweb.asm.ModuleVisitor;" +
                             "import org.objectweb.asm.Type;" +
                             "import org.objectweb.asm.TypePath;" +
-                            "public class" + " " + ASMifier.class.getSimpleName() + " " + "implements Opcodes" +
+                            "public class " + className + " implements Opcodes" +
                             "{" +
                             "public static byte[] dump() throws Exception {" +
                             getTextArea().getText() +
                             "return classWriter.toByteArray();" +
                             "}" +
                             "}";
-                    Compiler compiler = new Compiler();
-                    compiler.addSource(ASMifier.class.getSimpleName(), stringBuilder);
-                    compiler.compile();
 
-                    ClassLoader loader = new ClassLoader() {
-                        @Override
-                        protected Class<?> findClass(String name) {
-                            byte[] bytes = compiler.getClasses().get(ASMifier.class.getSimpleName());
-                            return defineClass(name, bytes, 0, bytes.length);
-                        }
-                    };
-
-                    byte[] dumps = (byte[]) loader.loadClass(ASMifier.class.getSimpleName()).getMethod("dump").invoke(null);
+                    byte[] dumps = (byte[])new ASMClassLoader().defineClass(className, Compiler.compileSingle(className, stringBuilder)).getMethod("dump").invoke(null);
                     ReflectUtil.copyAllMember(classNode, ASMUtil.acceptClassNode(new ClassReader(dumps)));
                     MessageUtil.info(LangUtil.i18n("success"));
                     EditSaveSuccessEvent.trigger(classNode.name);
@@ -94,6 +83,13 @@ protected Class<?> findClass(String name) {
         add(codeAreaPanel);
     }
 
+    public static String getSimpleName(String name){
+        int idx = name.lastIndexOf('/');
+        if (idx != -1){
+            return name.substring(idx + 1, name.length() - 1);
+        } else return name.substring(name.lastIndexOf('.') + 1, name.length() - 1);
+    }
+
     public void update(){
         StringWriter stringWriter = new StringWriter();
 

src/main/java/cn/enaium/joe/gui/panel/file/tabbed/tab/classes/ClassTabPanel.java

@@ -66,7 +66,7 @@ public ClassTabPanel(ClassNode classNode) {
     public void update(boolean forced){
         if (forced || classTabIndex != 0) traceBytecodeTabPanel.update();
         if (forced || classTabIndex != 1) decompileTabPanel.update();
-        if (forced || cclassTabIndex != 2) asmTablePanel.update();
+        if (forced || classTabIndex != 2) asmTablePanel.update();
     }
 
     static {

src/main/java/cn/enaium/joe/gui/panel/file/tabbed/tab/classes/DecompileTabPanel.java

@@ -17,7 +17,7 @@
 package cn.enaium.joe.gui.panel.file.tabbed.tab.classes;
 
 import cn.enaium.joe.JavaOctetEditor;
-import cn.enaium.joe.compiler.Compiler;
+import cn.enaium.joe.util.compiler.Compiler;
 import cn.enaium.joe.config.extend.KeymapConfig;
 import cn.enaium.joe.event.events.EditSaveSuccessEvent;
 import cn.enaium.joe.gui.panel.CodeAreaPanel;
@@ -26,10 +26,7 @@
 import org.objectweb.asm.ClassReader;
 import org.objectweb.asm.tree.ClassNode;
 
-import javax.swing.*;
 import java.awt.*;
-import java.awt.event.InputEvent;
-import java.awt.event.KeyEvent;
 
 /**
  * @author Enaium

src/main/java/cn/enaium/joe/jar/Jar.java

@@ -16,21 +16,10 @@
 
 package cn.enaium.joe.jar;
 
-import cn.enaium.joe.JavaOctetEditor;
-import cn.enaium.joe.config.extend.ApplicationConfig;
-import cn.enaium.joe.util.Util;
-import org.objectweb.asm.ClassReader;
 import org.objectweb.asm.tree.ClassNode;
 
-import java.io.ByteArrayOutputStream;
-import java.io.File;
-import java.io.IOException;
-import java.io.InputStream;
-import java.util.Enumeration;
 import java.util.LinkedHashMap;
 import java.util.Map;
-import java.util.jar.JarEntry;
-import java.util.jar.JarFile;
 
 /**
  * @author Enaium

src/main/java/cn/enaium/joe/task/AbstractTask.java

@@ -16,7 +16,6 @@
 
 package cn.enaium.joe.task;
 
-import java.util.concurrent.Callable;
 import java.util.function.Supplier;
 
 /**

src/main/java/cn/enaium/joe/task/BuildEnvironmentTask.java

@@ -0,0 +1,21 @@
+package cn.enaium.joe.task;
+
+import cn.enaium.joe.jar.Jar;
+import cn.enaium.joe.util.compiler.environment.RecompileEnvironment;
+
+import java.util.Map;
+
+public class BuildEnvironmentTask extends AbstractTask<Map<String, byte[]>> {
+    Jar jar;
+    public BuildEnvironmentTask(Jar jar) {
+        super(" BuildEnvironment");
+        this.jar = jar;
+    }
+
+    @Override
+    public Map<String, byte[]> get() {
+        RecompileEnvironment.environment = RecompileEnvironment.build(jar, this::setProgress);
+        this.setProgress(100);
+        return RecompileEnvironment.getEnvironment();
+    }
+}

src/main/java/cn/enaium/joe/task/InputJarTask.java

@@ -72,6 +72,9 @@ public Jar get() {
         JavaOctetEditor.getInstance().config.getByClass(ApplicationConfig.class).loadRecent.getValue().add(file.getAbsolutePath());
         JavaOctetEditor.getInstance().setJar(jar);
         JavaOctetEditor.getInstance().window.setTitle(JavaOctetEditor.TITLE + "-" + file.getName());
+
+        JavaOctetEditor.getInstance().task.submit(new BuildEnvironmentTask(jar));
+
         return jar;
     }
 }

src/main/java/cn/enaium/joe/util/asm/ClassVersionVisitor.java

@@ -0,0 +1,26 @@
+package cn.enaium.joe.util.asm;
+
+import org.objectweb.asm.ClassReader;
+import org.objectweb.asm.ClassVisitor;
+import org.objectweb.asm.ClassWriter;
+import org.objectweb.asm.Opcodes;
+
+public class ClassVersionVisitor extends ClassVisitor {
+    public final int version;
+    public ClassVersionVisitor(int jdkVersion) {
+        super(Opcodes.ASM9);
+        this.version = jdkVersion;
+    }
+
+    @Override
+    public void visit(int version, int access, String name, String signature, String superName, String[] interfaces) {
+        super.visit(this.version, access, name, signature, superName, interfaces);
+    }
+
+    public static byte[] processVersion(byte[] clazz, int version){
+        ClassReader classReader = new ClassReader(clazz);
+        classReader.accept(new ClassVersionVisitor(version), 0);
+        ClassWriter writer = new ClassWriter(classReader, 0);
+        return writer.toByteArray();
+    }
+}

src/main/java/cn/enaium/joe/util/classes/ASMClassLoader.java

@@ -0,0 +1,8 @@
+package cn.enaium.joe.util.classes;
+
+public class ASMClassLoader extends ClassLoader{
+    public Class<?> defineClass(String canonicalName, byte[] clazz){
+        return defineClass(canonicalName, clazz, 0, clazz.length);
+    }
+
+}