refactor: simplify memshell generate map register

2026-09-22 15:10:43 +08:00 · 2025-02-23 02:33:09 +08:00
parent f27b06fc0b
commit 5084a0b850
177 changed files with 3761 additions and 10859 deletions
@@ -0,0 +1,62 @@
+package com.reajason.javaweb.buddy;
+
+import net.bytebuddy.ByteBuddy;
+import net.bytebuddy.asm.AsmVisitorWrapper;
+import net.bytebuddy.dynamic.DynamicType;
+import org.junit.jupiter.api.Test;
+
+import java.util.Collections;
+
+import static net.bytebuddy.matcher.ElementMatchers.named;
+import static org.junit.jupiter.api.Assertions.assertEquals;
+
+/**
+ * @author ReaJason
+ * @since 2025/2/22
+ */
+class MethodCallReplaceVisitorWrapperTest {
+
+    public static class ExternalClass {
+        public static String externalMethod(String input) {
+            return "External: " + input;
+        }
+
+        public static String replacementMethod(String input) {
+            return "Replaced: " + input;
+        }
+    }
+
+    public static class TargetClass {
+        public String targetMethod(String input) {
+            System.out.println("targetMethod");
+            return ExternalClass.replacementMethod(input);
+        }
+
+        public static String replacementMethod(String input) {
+            return "Replaced: " + input;
+        }
+    }
+
+    @Test
+    void test() throws Exception {
+        String newClassName = TargetClass.class.getName() + "Redefinition";
+        DynamicType.Unloaded<TargetClass> dynamicType = new ByteBuddy()
+                .redefine(TargetClass.class)
+                .name(newClassName)
+                .visit(new AsmVisitorWrapper
+                        .ForDeclaredMethods()
+                        .method(named("targetMethod"),
+                                new MethodCallReplaceVisitorWrapper(
+                                        newClassName,
+                                        Collections.singleton(ExternalClass.class.getName())
+                                )
+                        )
+                )
+                .make();
+        Class<?> redefinedClass = dynamicType.load(MethodCallReplaceVisitorWrapperTest.class.getClassLoader())
+                .getLoaded();
+        Object object = redefinedClass.newInstance();
+        Object result = object.getClass().getMethod("targetMethod", String.class).invoke(object, "xixi");
+        assertEquals("Replaced: xixi", result);
+    }
+}