Fix: new Forge Installer will terminate main process

xfl03 · Feb 2, 2023 · d17808e · d17808e
1 parent 5bf1b95
commit d17808e
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/...der/src/main/java/org/to2mbn/jmccc/mcdownloader/provider/forge/ForgeInstallerTweaker.java b/...der/src/main/java/org/to2mbn/jmccc/mcdownloader/provider/forge/ForgeInstallerTweaker.java
@@ -35,10 +35,21 @@ protected MainMethodVisitor(int api, MethodVisitor methodVisitor) {
 
         @Override
         public void visitFieldInsn(int opcode, String owner, String name, String descriptor) {
+            // Change Actions.SERVER -> Actions.CLIENT
             if (opcode == Opcodes.GETSTATIC && name.equals("SERVER")) {
                 name = "CLIENT";
             }
             super.visitFieldInsn(opcode, owner, name, descriptor);
         }
+
+        @Override
+        public void visitMethodInsn(int opcode, String owner, String name, String descriptor, boolean isInterface) {
+            // Remove System.exit
+            if (opcode == Opcodes.INVOKESTATIC && name.equals("exit")) {
+                super.visitInsn(Opcodes.POP);
+                return;
+            }
+            super.visitMethodInsn(opcode, owner, name, descriptor, isInterface);
+        }
     }
 }