From b1ec92a16f0dea99b45bd571d14b320acaf29a51 Mon Sep 17 00:00:00 2001
From: xDark <19853368+xxDark@users.noreply.github.com>
Date: Mon, 22 Feb 2021 22:41:14 +0300
Subject: [PATCH] Replace permit library

---
 pom.xml                                       | 17 +++--
 src/main/java/java/lang/Module.java           | 16 +++++
 src/main/java/java/lang/ModuleLayer.java      | 20 ++++++
 src/main/java/java/lang/module/Module.java    |  8 ---
 .../java/java/lang/module/ModuleFinder.java   |  2 +-
 src/main/java/me/coley/recaf/Recaf.java       |  2 +-
 .../java/me/coley/recaf/util/Java9Util.java   | 65 +++++++++++++++++++
 src/main/java/me/coley/recaf/util/VMUtil.java | 35 +++++++++-
 .../workspace/InstrumentationResource.java    |  1 -
 9 files changed, 145 insertions(+), 21 deletions(-)
 create mode 100644 src/main/java/java/lang/Module.java
 create mode 100644 src/main/java/java/lang/ModuleLayer.java
 delete mode 100644 src/main/java/java/lang/module/Module.java
 create mode 100644 src/main/java/me/coley/recaf/util/Java9Util.java

diff --git a/pom.xml b/pom.xml
index ff9ba79f6..b14153859 100644
--- a/pom.xml
+++ b/pom.xml
@@ -3,7 +3,7 @@
 	<groupId>me.coley</groupId>
 	<artifactId>recaf</artifactId>
 	<url>https://github.com/Col-E/Recaf/</url>
-	<version>2.17.5</version>
+	<version>2.17.6</version>
 	<name>Recaf</name>
 	<description>A modern java bytecode editor</description>
 	<!-- Variables -->
@@ -300,12 +300,6 @@
 			<artifactId>maven-jdk-tools-wrapper</artifactId>
 			<version>0.1</version>
 		</dependency>
-		<!-- https://mvnrepository.com/artifact/com.nqzero/permit-reflect -->
-		<dependency>
-			<groupId>com.nqzero</groupId>
-			<artifactId>permit-reflect</artifactId>
-			<version>0.4</version>
-		</dependency>
 		<!--- Testing -->
 		<!-- https://mvnrepository.com/artifact/org.junit.jupiter/junit-jupiter-api -->
 		<dependency>
@@ -525,6 +519,15 @@
                     <target>8</target>
                 </configuration>
             </plugin>
+			<plugin>
+				<groupId>org.apache.maven.plugins</groupId>
+				<artifactId>maven-jar-plugin</artifactId>
+				<configuration>
+					<excludes>
+						<exclude>java/**</exclude>
+					</excludes>
+				</configuration>
+			</plugin>
             <!-- Bug/quality enforcement
                  mvn spotbugs:spotbugs - create report
                  mvn spotbugs:gui - show gui
diff --git a/src/main/java/java/lang/Module.java b/src/main/java/java/lang/Module.java
new file mode 100644
index 000000000..c4180f357
--- /dev/null
+++ b/src/main/java/java/lang/Module.java
@@ -0,0 +1,16 @@
+package java.lang;
+
+import java.util.Set;
+
+/**
+ * Dummy java compatibility class
+ *
+ * @author Matt
+ */
+public abstract class Module {
+
+  //CHECKSTYLE:OFF
+  public ModuleLayer getLayer() { throw new UnsupportedOperationException(); }
+  public Set<String> getPackages() { throw new UnsupportedOperationException(); }
+  //CHECKSTYLE:ON
+}
diff --git a/src/main/java/java/lang/ModuleLayer.java b/src/main/java/java/lang/ModuleLayer.java
new file mode 100644
index 000000000..c4e418033
--- /dev/null
+++ b/src/main/java/java/lang/ModuleLayer.java
@@ -0,0 +1,20 @@
+package java.lang;
+
+import java.util.Set;
+
+/**
+ * Dummy java compatibility class
+ *
+ * @author xxDark
+ */
+public abstract class ModuleLayer {
+
+  //CHECKSTYLE:OFF
+  public Set<Module> modules() {
+    throw new UnsupportedOperationException();
+  }
+  public static ModuleLayer boot() {
+    throw new UnsupportedOperationException();
+  }
+  //CHECKSTYLE:ON
+}
diff --git a/src/main/java/java/lang/module/Module.java b/src/main/java/java/lang/module/Module.java
deleted file mode 100644
index 75ec498df..000000000
--- a/src/main/java/java/lang/module/Module.java
+++ /dev/null
@@ -1,8 +0,0 @@
-package java.lang.module;
-
-/**
- * Dummy java compatibility class
- *
- * @author Matt
- */
-public interface Module {}
diff --git a/src/main/java/java/lang/module/ModuleFinder.java b/src/main/java/java/lang/module/ModuleFinder.java
index cb35e9745..f5a52745a 100644
--- a/src/main/java/java/lang/module/ModuleFinder.java
+++ b/src/main/java/java/lang/module/ModuleFinder.java
@@ -10,7 +10,7 @@
 public interface ModuleFinder {
     //CHECKSTYLE:OFF
     static ModuleFinder ofSystem() {
-        return null;
+        throw new UnsupportedOperationException();
     }
     Set<ModuleReference> findAll();
     //CHECKSTYLE:ON
diff --git a/src/main/java/me/coley/recaf/Recaf.java b/src/main/java/me/coley/recaf/Recaf.java
index 82323819b..8c3687800 100644
--- a/src/main/java/me/coley/recaf/Recaf.java
+++ b/src/main/java/me/coley/recaf/Recaf.java
@@ -30,7 +30,7 @@
  * @author Matt
  */
 public class Recaf {
-	public static final String VERSION = "2.17.5";
+	public static final String VERSION = "2.17.6";
 	public static final String DOC_URL = "https://col-e.github.io/Recaf-documentation/";
 	public static final int ASM_VERSION = Opcodes.ASM9;
 	private static Controller currentController;
diff --git a/src/main/java/me/coley/recaf/util/Java9Util.java b/src/main/java/me/coley/recaf/util/Java9Util.java
new file mode 100644
index 000000000..e241c7853
--- /dev/null
+++ b/src/main/java/me/coley/recaf/util/Java9Util.java
@@ -0,0 +1,65 @@
+package me.coley.recaf.util;
+
+import java.lang.invoke.MethodHandle;
+import java.lang.invoke.MethodHandles;
+import java.lang.invoke.MethodHandles.Lookup;
+import java.lang.invoke.MethodType;
+import java.lang.reflect.Field;
+
+/**
+ * Package-private util to deal with modules.
+ *
+ * @author xxDark
+ */
+final class Java9Util {
+
+    private static final MethodHandle CLASS_MODULE;
+    private static final MethodHandle CLASS_LOADER_MDOULE;
+
+    /**
+     * Deny all constructions.
+     */
+    private Java9Util() {
+    }
+
+    /**
+     * @param klass {@link Class} to get module from.
+     * @return {@link Module} of the class.
+     */
+    static Module getClassModule(Class<?> klass) {
+        try {
+            return (Module) CLASS_MODULE.invokeExact(klass);
+        } catch (Throwable t) {
+            // That should never happen.
+            throw new AssertionError(t);
+        }
+    }
+
+
+    /**
+     * @param loader {@link ClassLoader} to get module from.
+     * @return {@link Module} of the class.
+     */
+    static Module getLoaderModule(ClassLoader loader) {
+        try {
+            return (Module) CLASS_LOADER_MDOULE.invokeExact(loader);
+        } catch (Throwable t) {
+            // That should never happen.
+            throw new AssertionError(t);
+        }
+    }
+
+    static {
+        try {
+            Field field = Lookup.class.getDeclaredField("IMPL_LOOKUP");
+            field.setAccessible(true);
+            MethodHandles.publicLookup();
+            Lookup lookup = (Lookup) field.get(null);
+            MethodType type = MethodType.methodType(Module.class);
+            CLASS_MODULE = lookup.findVirtual(Class.class, "getModule", type);
+            CLASS_LOADER_MDOULE = lookup.findVirtual(ClassLoader.class, "getUnnamedModule", type);
+        } catch (NoSuchMethodException | IllegalAccessException | NoSuchFieldException ex) {
+            throw new ExceptionInInitializerError(ex);
+        }
+    }
+}
diff --git a/src/main/java/me/coley/recaf/util/VMUtil.java b/src/main/java/me/coley/recaf/util/VMUtil.java
index 15acb614b..fae4f7338 100644
--- a/src/main/java/me/coley/recaf/util/VMUtil.java
+++ b/src/main/java/me/coley/recaf/util/VMUtil.java
@@ -1,7 +1,7 @@
 package me.coley.recaf.util;
 
-import com.nqzero.permit.Permit;
 import com.sun.javafx.application.PlatformImpl;
+import java.util.HashSet;
 import javafx.application.Platform;
 
 import java.io.IOException;
@@ -212,12 +212,41 @@ public static Path getJavaPath() {
      */
     public static void patch() {
         if (getVmVersion() > 8) {
-            Permit.godMode();
-            Permit.unLog();
+            openPackages();
             patchReflectionFilters();
         }
     }
 
+    /**
+     * Opens all packages.
+     */
+    private static void openPackages() {
+        try {
+            Method export = Module.class.getDeclaredMethod("implAddOpens",String.class);
+            export.setAccessible(true);
+            HashSet<Module> modules = new HashSet<>();
+            Class<?> classBase = VMUtil.class;
+            Module base = Java9Util.getClassModule(classBase);
+            if (base.getLayer() != null)
+                modules.addAll(base.getLayer().modules());
+            modules.addAll(ModuleLayer.boot().modules());
+            for (ClassLoader cl = classBase.getClassLoader(); cl != null; cl = cl.getParent()) {
+                modules.add(Java9Util.getLoaderModule(cl));
+            }
+            for (Module module : modules) {
+                for (String name : module.getPackages()) {
+                    try {
+                        export.invoke(module, name);
+                    } catch (Exception ex) {
+                        Log.error(ex, "Could not export package {} in module {}", name, module);
+                    }
+                }
+            }
+        } catch (Exception ex) {
+            Log.error(ex, "Could not export packages");
+        }
+    }
+
     /**
      * Patches reflection filters.
      */
diff --git a/src/main/java/me/coley/recaf/workspace/InstrumentationResource.java b/src/main/java/me/coley/recaf/workspace/InstrumentationResource.java
index 0e7bc91ed..49dc4c3fc 100644
--- a/src/main/java/me/coley/recaf/workspace/InstrumentationResource.java
+++ b/src/main/java/me/coley/recaf/workspace/InstrumentationResource.java
@@ -19,7 +19,6 @@
 import java.lang.instrument.ClassFileTransformer;
 import java.lang.instrument.Instrumentation;
 import java.lang.instrument.UnmodifiableClassException;
-import java.lang.module.Module;
 import java.security.ProtectionDomain;
 import java.util.*;
 import java.util.concurrent.ConcurrentHashMap;