From 63844d1cce0e1ff69fa7524c53f955904744d5e5 Mon Sep 17 00:00:00 2001 From: icanci <1845666903@qq.com> Date: Tue, 20 May 2025 16:14:54 +0800 Subject: [PATCH 1/2] mock ly~ --- .../damc/cache/ArgsValidateResult.java | 36 ++--- .../cache/DamcCenterConfigRepository.java | 8 +- damc-core/pom.xml | 55 +++++-- .../loopstack/damc/core/DamcCoreStart.java | 16 ++- .../damc/core/dynamic/DamcCoreConstant.java | 1 + .../damc/core/dynamic/DynamicAgent.java | 135 +++++++++++++++--- .../core/dynamic/DynamicAgentClassCache.java | 21 ++- .../dynamic/visitor/ModifierClassVisitor.java | 7 +- .../core/dynamic/visitor/VisitorModifier.java | 10 +- spring-web-application-test/pom.xml | 36 ++--- 10 files changed, 239 insertions(+), 86 deletions(-) diff --git a/damc-api/damc-api-cache/src/main/java/cn/icanci/loopstack/damc/cache/ArgsValidateResult.java b/damc-api/damc-api-cache/src/main/java/cn/icanci/loopstack/damc/cache/ArgsValidateResult.java index 8878fd0..088dac1 100644 --- a/damc-api/damc-api-cache/src/main/java/cn/icanci/loopstack/damc/cache/ArgsValidateResult.java +++ b/damc-api/damc-api-cache/src/main/java/cn/icanci/loopstack/damc/cache/ArgsValidateResult.java @@ -22,19 +22,23 @@ package cn.icanci.loopstack.damc.cache; */ public class ArgsValidateResult { /** DAMC 中心的注册地址 例如: 127.0.0.1;127.0.0.1 */ - private String damcCenterAddress; + private java.lang.String damcCenterAddress; /** DAMC 中心的注册port 例如: 9090 */ - private int damcCenterPort; + private int damcCenterPort; /** 被Mock的项目的本地netty服务端口 */ - private int mockedProjectPort; - /** 被Mock的项目的环境值，可选值: QA\UAT\STAGE\STAGE_TEST2 @see EnvEnum */ - private EnvEnum mockedProjectEnv; + private int mockedProjectPort; + /** 被Mock的项目的环境值，可选值: QA\UAT\STAGE\STAGE_TEST2 @see String */ + private String mockedProjectEnv; /** 项目名称 */ - private String mockedProjectName; + private java.lang.String mockedProjectName; /** 项目Code */ - private String mockedProjectCode; + private java.lang.String mockedProjectCode; - public ArgsValidateResult(String damcCenterAddress, int damcCenterPort, int mockedProjectPort, EnvEnum mockedProjectEnv, String mockedProjectName, String mockedProjectCode) { + public ArgsValidateResult() { + } + + public ArgsValidateResult(java.lang.String damcCenterAddress, int damcCenterPort, int mockedProjectPort, String mockedProjectEnv, java.lang.String mockedProjectName, + java.lang.String mockedProjectCode) { this.damcCenterAddress = damcCenterAddress; this.damcCenterPort = damcCenterPort; this.mockedProjectPort = mockedProjectPort; @@ -43,11 +47,11 @@ public class ArgsValidateResult { this.mockedProjectCode = mockedProjectCode; } - public String getDamcCenterAddress() { + public java.lang.String getDamcCenterAddress() { return damcCenterAddress; } - public void setDamcCenterAddress(String damcCenterAddress) { + public void setDamcCenterAddress(java.lang.String damcCenterAddress) { this.damcCenterAddress = damcCenterAddress; } @@ -67,27 +71,27 @@ public class ArgsValidateResult { this.mockedProjectPort = mockedProjectPort; } - public EnvEnum getMockedProjectEnv() { + public String getMockedProjectEnv() { return mockedProjectEnv; } - public void setMockedProjectEnv(EnvEnum mockedProjectEnv) { + public void setMockedProjectEnv(String mockedProjectEnv) { this.mockedProjectEnv = mockedProjectEnv; } - public String getMockedProjectName() { + public java.lang.String getMockedProjectName() { return mockedProjectName; } - public void setMockedProjectName(String mockedProjectName) { + public void setMockedProjectName(java.lang.String mockedProjectName) { this.mockedProjectName = mockedProjectName; } - public String getMockedProjectCode() { + public java.lang.String getMockedProjectCode() { return mockedProjectCode; } - public void setMockedProjectCode(String mockedProjectCode) { + public void setMockedProjectCode(java.lang.String mockedProjectCode) { this.mockedProjectCode = mockedProjectCode; } diff --git a/damc-api/damc-api-cache/src/main/java/cn/icanci/loopstack/damc/cache/DamcCenterConfigRepository.java b/damc-api/damc-api-cache/src/main/java/cn/icanci/loopstack/damc/cache/DamcCenterConfigRepository.java index fddf877..cfc9c8b 100644 --- a/damc-api/damc-api-cache/src/main/java/cn/icanci/loopstack/damc/cache/DamcCenterConfigRepository.java +++ b/damc-api/damc-api-cache/src/main/java/cn/icanci/loopstack/damc/cache/DamcCenterConfigRepository.java @@ -29,7 +29,7 @@ public class DamcCenterConfigRepository { DamcCenterConfigRepository.argsValidateResult = argsValidateResult; } - public static String getDamcCenterAddress() { + public static java.lang.String getDamcCenterAddress() { return argsValidateResult.getDamcCenterAddress(); } @@ -41,15 +41,15 @@ public class DamcCenterConfigRepository { return argsValidateResult.getMockedProjectPort(); } - public static EnvEnum getMockedProjectEnv() { + public static String getMockedProjectEnv() { return argsValidateResult.getMockedProjectEnv(); } - public static String getMockedProjectName() { + public static java.lang.String getMockedProjectName() { return argsValidateResult.getMockedProjectName(); } - public static String getMockedProjectCode() { + public static java.lang.String getMockedProjectCode() { return argsValidateResult.getMockedProjectCode(); } diff --git a/damc-core/pom.xml b/damc-core/pom.xml index 13013c5..d5e842f 100644 --- a/damc-core/pom.xml +++ b/damc-core/pom.xml @@ -22,16 +22,11 @@ - - org.ow2.asm - asm - ${asm.version} - - - org.ow2.asm - asm-commons - ${asm.version} - + + + + + org.codehaus.groovy groovy-all @@ -89,6 +84,46 @@ + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + org.apache.maven.plugins maven-jar-plugin diff --git a/damc-core/src/main/java/cn/icanci/loopstack/damc/core/DamcCoreStart.java b/damc-core/src/main/java/cn/icanci/loopstack/damc/core/DamcCoreStart.java index a78261f..2be8e0b 100644 --- a/damc-core/src/main/java/cn/icanci/loopstack/damc/core/DamcCoreStart.java +++ b/damc-core/src/main/java/cn/icanci/loopstack/damc/core/DamcCoreStart.java @@ -16,6 +16,7 @@ */ package cn.icanci.loopstack.damc.core; +import java.lang.instrument.Instrumentation; import java.util.List; import java.util.concurrent.TimeUnit; @@ -34,10 +35,9 @@ import cn.hutool.json.JSONUtil; import cn.icanci.loopstack.damc.api.groovy.GroovyScriptRepository; import cn.icanci.loopstack.damc.api.http.Client; import cn.icanci.loopstack.damc.api.http.HttpClientImpl; +import cn.icanci.loopstack.damc.cache.ArgsValidateResult; import cn.icanci.loopstack.damc.cache.DamcCenterConfigRepository; -import cn.icanci.loopstack.damc.core.dynamic.AttachTool; -import cn.icanci.loopstack.damc.core.dynamic.ReTransformTrigger; -import cn.icanci.loopstack.damc.core.dynamic.ReTransformTriggerMBean; +import cn.icanci.loopstack.damc.core.dynamic.*; import cn.icanci.loopstack.damc.core.server.RegisterServer; import cn.icanci.loopstack.damc.core.service.MockRunnerService; import cn.icanci.loopstack.damc.spi.model.UriConstant; @@ -60,6 +60,12 @@ public class DamcCoreStart implements ApplicationContextAware { private static final Logger logger = LoggerFactory.getLogger(DamcCoreStart.class); + public static void setInitParam(ArgsValidateResult result, Instrumentation inst) { + // 初始化到仓储 + DynamicAgent.setInstrumentation(inst); + DynamicAgent.setResult(result); + } + /** * Set the ApplicationContext that this object runs in. * Normally this call will be used to initialize the object. @@ -77,6 +83,10 @@ public class DamcCoreStart implements ApplicationContextAware { @Override public void setApplicationContext(ApplicationContext applicationContext) throws BeansException { try { + // 获取类加载器 + ClassLoader classLoader = this.getClass().getClassLoader(); + System.getProperties().put(DamcCoreConstant.APP_RUNNER_CLASS_LOADER, classLoader.getClass().getName()); + AttachTool.attachAgent(); MockRunnerService.setContext(applicationContext); diff --git a/damc-core/src/main/java/cn/icanci/loopstack/damc/core/dynamic/DamcCoreConstant.java b/damc-core/src/main/java/cn/icanci/loopstack/damc/core/dynamic/DamcCoreConstant.java index 1ef9e62..f9ca949 100644 --- a/damc-core/src/main/java/cn/icanci/loopstack/damc/core/dynamic/DamcCoreConstant.java +++ b/damc-core/src/main/java/cn/icanci/loopstack/damc/core/dynamic/DamcCoreConstant.java @@ -24,4 +24,5 @@ public interface DamcCoreConstant { String DAMC_ARGS_KEY = "DAMC_ARGS"; String DAMC_ARGS_INIT_KEY = "DAMC_ARGS_INIT"; String DAMC_ARGS_AGENT_PATH_KEY = "DAMC_ARGS_AGENT_PATH"; + String APP_RUNNER_CLASS_LOADER = "APP_RUNNER_CLASS_LOADER"; } diff --git a/damc-core/src/main/java/cn/icanci/loopstack/damc/core/dynamic/DynamicAgent.java b/damc-core/src/main/java/cn/icanci/loopstack/damc/core/dynamic/DynamicAgent.java index 86466f4..8c4ee3e 100644 --- a/damc-core/src/main/java/cn/icanci/loopstack/damc/core/dynamic/DynamicAgent.java +++ b/damc-core/src/main/java/cn/icanci/loopstack/damc/core/dynamic/DynamicAgent.java @@ -18,11 +18,16 @@ package cn.icanci.loopstack.damc.core.dynamic; import java.lang.instrument.ClassFileTransformer; import java.lang.instrument.Instrumentation; +import java.lang.reflect.Field; +import java.lang.reflect.Method; import java.security.ProtectionDomain; +import java.util.Set; import org.apache.commons.lang3.StringUtils; import org.apache.commons.lang3.math.NumberUtils; +import com.google.common.collect.Sets; + import cn.icanci.loopstack.damc.cache.ArgsValidateResult; import cn.icanci.loopstack.damc.cache.DamcCenterConfigRepository; import cn.icanci.loopstack.damc.cache.EnvEnum; @@ -41,12 +46,12 @@ public class DynamicAgent { * @param args args，例如: 127.0.0.1;127.0.0.2|9191|9292|QA|DAMC测试项目|DAMC-TEST-CORE^/Users/icanci/ideaProjects/loopstack/damc/damc-core/target/damc-core-1.0-SNAPSHOT-jar-with-dependencies.jar * @param inst inst */ - public static void premain(String args, Instrumentation inst) { + public static void premain(java.lang.String args, Instrumentation inst) { if (StringUtils.isBlank(args)) { return; } - String[] argsArray = args.split("\\^"); + java.lang.String[] argsArray = args.split("\\^"); if (argsArray.length != 2) { return; } @@ -61,15 +66,24 @@ public class DynamicAgent { } // 动态 Attach 时加载 Agent 的入口方法 - public static void agentmain(String args, Instrumentation inst) { + public static void agentmain(java.lang.String args, Instrumentation inst) { // doAgentTransformer - doAgentTransformer(inst, String.valueOf(System.getProperties().get(DamcCoreConstant.DAMC_ARGS_INIT_KEY))); + doAgentTransformer(inst, java.lang.String.valueOf(System.getProperties().get(DamcCoreConstant.DAMC_ARGS_INIT_KEY))); } - private static Instrumentation getInstrumentation() { + public static Instrumentation getInstrumentation() { return instrumentation; } + public static void setInstrumentation(Instrumentation instrumentation) { + DynamicAgent.instrumentation = instrumentation; + DynamicAgent.instrumentation.addTransformer(new DynamicTransformer(), true); + } + + public static void setResult(ArgsValidateResult result) { + DamcCenterConfigRepository.setArgsValidateResult(result); + } + /** * 是否完成了加载 * @@ -85,18 +99,99 @@ public class DynamicAgent { * @param inst inst * @param argsArray argsArray */ - private static void doAgentTransformer(Instrumentation inst, String args) { - // ArgsValidateResult - ArgsValidateResult result = getArgsValidate(args); - if (result == null) { - return; + private static void doAgentTransformer(Instrumentation inst, java.lang.String args) { + try { + // ArgsValidateResult + ArgsValidateResult result = getArgsValidate(args); + if (result == null) { + return; + } + DamcCenterConfigRepository.setArgsValidateResult(result); + // instrumentation + instrumentation = inst; + // 注册一个可重用的 Transformer + // inst.addTransformer(new DynamicTransformer(), true); + + // ======================================== 运行时处理 ======================================== + // 获取项目中的目标ClassLoader 和 这个进行匹配 + Object appRunnerClassLoaderName = System.getProperties().get(DamcCoreConstant.APP_RUNNER_CLASS_LOADER); + if (appRunnerClassLoaderName == null) { + return; + } + + java.lang.String className = java.lang.String.valueOf(appRunnerClassLoaderName); + if (StringUtils.isEmpty(className)) { + return; + } + Set classLoaders = listClassLoaders(); + for (ClassLoader clazzLoader : classLoaders) { + if (clazzLoader == null) { + continue; + } + if (!StringUtils.equals(clazzLoader.getClass().getName(), className)) { + continue; + } + + Class damcCoreStartClazz = clazzLoader.loadClass("cn.icanci.loopstack.damc.core.DamcCoreStart"); + Object start = damcCoreStartClazz.newInstance(); + Class argsValidateResultClazz = clazzLoader.loadClass("cn.icanci.loopstack.damc.cache.ArgsValidateResult"); + Object startResult = argsValidateResultClazz.newInstance(); + + Field[] fields = startResult.getClass().getDeclaredFields(); + for (Field field : fields) { + if (StringUtils.equals(field.getName(), "damcCenterAddress")) { + field.setAccessible(true); + field.set(startResult, result.getDamcCenterAddress()); + } + if (StringUtils.equals(field.getName(), "damcCenterPort")) { + field.setAccessible(true); + field.set(startResult, result.getDamcCenterPort()); + } + if (StringUtils.equals(field.getName(), "mockedProjectPort")) { + field.setAccessible(true); + field.set(startResult, result.getMockedProjectPort()); + } + if (StringUtils.equals(field.getName(), "mockedProjectEnv")) { + field.setAccessible(true); + field.set(startResult, result.getMockedProjectEnv()); + } + if (StringUtils.equals(field.getName(), "mockedProjectName")) { + field.setAccessible(true); + field.set(startResult, result.getMockedProjectName()); + } + if (StringUtils.equals(field.getName(), "mockedProjectCode")) { + field.setAccessible(true); + field.set(startResult, result.getMockedProjectCode()); + } + + } + + Method[] methods = start.getClass().getMethods(); + for (Method method : methods) { + if (StringUtils.equals(method.getName(), "setInitParam")) { + method.invoke(start, startResult, inst); + } + } + } + } catch (Exception e) { + // TODO logs + System.out.println(e); } - DamcCenterConfigRepository.setArgsValidateResult(result); + } - // instrumentation - instrumentation = inst; - // 注册一个可重用的 Transformer - inst.addTransformer(new DynamicTransformer(), true); + public static Set listClassLoaders() { + if (instrumentation == null) { + System.out.println("Instrumentation is not initialized!"); + return Sets.newHashSet(); + } + + // 获取所有已加载的类 + Set classLoaders = Sets.newHashSet(); + for (Class clazz : instrumentation.getAllLoadedClasses()) { + ClassLoader classLoader = clazz.getClassLoader(); + classLoaders.add(classLoader); + } + return classLoaders; } /** @@ -106,7 +201,7 @@ public class DynamicAgent { * @param args args * @return ArgsValidateResult */ - private static ArgsValidateResult getArgsValidate(String args) { + private static ArgsValidateResult getArgsValidate(java.lang.String args) { // 启动参数为空，不处理 if (StringUtils.isEmpty(args)) { return null; @@ -118,7 +213,7 @@ public class DynamicAgent { } // 127.0.0.1;127.0.0.2 - String damcCenterAddress = arg[0]; + java.lang.String damcCenterAddress = arg[0]; // 9191 if (!NumberUtils.isNumber(arg[1])) { return null; @@ -141,12 +236,12 @@ public class DynamicAgent { return null; } - String mockedProjectCode = arg[5]; + java.lang.String mockedProjectCode = arg[5]; if (StringUtils.isBlank(mockedProjectCode)) { return null; } - return new ArgsValidateResult(damcCenterAddress, damcCenterPort, mockedProjectPort, mockedProjectEnv, mockedProjectName, mockedProjectCode); + return new ArgsValidateResult(damcCenterAddress, damcCenterPort, mockedProjectPort, mockedProjectEnv.getEnvCode(), mockedProjectName, mockedProjectCode); } /** @@ -169,7 +264,7 @@ public class DynamicAgent { */ static class DynamicTransformer implements ClassFileTransformer { @Override - public byte[] transform(ClassLoader loader, String clazzName, Class classBeingRedefined, ProtectionDomain protectionDomain, byte[] classfileBuffer) { + public byte[] transform(ClassLoader loader, java.lang.String clazzName, Class classBeingRedefined, ProtectionDomain protectionDomain, byte[] classfileBuffer) { return DynamicAgentClassCache.doTransform(classBeingRedefined, classfileBuffer); } } diff --git a/damc-core/src/main/java/cn/icanci/loopstack/damc/core/dynamic/DynamicAgentClassCache.java b/damc-core/src/main/java/cn/icanci/loopstack/damc/core/dynamic/DynamicAgentClassCache.java index 388ba43..2d552da 100644 --- a/damc-core/src/main/java/cn/icanci/loopstack/damc/core/dynamic/DynamicAgentClassCache.java +++ b/damc-core/src/main/java/cn/icanci/loopstack/damc/core/dynamic/DynamicAgentClassCache.java @@ -16,19 +16,19 @@ */ package cn.icanci.loopstack.damc.core.dynamic; -import static org.objectweb.asm.ClassReader.EXPAND_FRAMES; +import static damc.objectweb.asm.ClassReader.EXPAND_FRAMES; import java.io.File; import java.io.FileOutputStream; import java.io.IOException; import java.util.*; -import org.objectweb.asm.ClassReader; -import org.objectweb.asm.ClassVisitor; -import org.objectweb.asm.ClassWriter; +import damc.objectweb.asm.ClassReader; +import damc.objectweb.asm.ClassVisitor; +import damc.objectweb.asm.ClassWriter; -import cn.icanci.loopstack.damc.spi.model.refresh.DynamicConditionWrapper; import cn.icanci.loopstack.damc.core.dynamic.visitor.ModifierClassVisitor; +import cn.icanci.loopstack.damc.spi.model.refresh.DynamicConditionWrapper; /** * @author icanci @@ -72,7 +72,7 @@ public class DynamicAgentClassCache { return defaultTransform(classfileBuffer); } Set wrappers = CLASS_CONDITIONS.get(classBeingRedefined); - ClassReader cr = new ClassReader(classBeingRedefined.getTypeName()); + ClassReader cr = getClassReader(classBeingRedefined); ClassWriter cw = new ClassWriter(cr, ClassWriter.COMPUTE_MAXS); for (DynamicConditionWrapper wrapper : wrappers) { @@ -92,6 +92,15 @@ public class DynamicAgentClassCache { } } + private static ClassReader getClassReader(Class classBeingRedefined) throws IOException { + try { + return new ClassReader(classBeingRedefined.getTypeName()); + } catch (Throwable e) { + String clazzName = classBeingRedefined.getTypeName().replace('.', '/') + ".class"; + return new ClassReader(classBeingRedefined.getClassLoader().getResourceAsStream(clazzName)); + } + } + /** * printNewByteArray * diff --git a/damc-core/src/main/java/cn/icanci/loopstack/damc/core/dynamic/visitor/ModifierClassVisitor.java b/damc-core/src/main/java/cn/icanci/loopstack/damc/core/dynamic/visitor/ModifierClassVisitor.java index 7edfd67..8d52f70 100644 --- a/damc-core/src/main/java/cn/icanci/loopstack/damc/core/dynamic/visitor/ModifierClassVisitor.java +++ b/damc-core/src/main/java/cn/icanci/loopstack/damc/core/dynamic/visitor/ModifierClassVisitor.java @@ -16,14 +16,13 @@ */ package cn.icanci.loopstack.damc.core.dynamic.visitor; -import static org.objectweb.asm.Opcodes.ASM9; +import static damc.objectweb.asm.Opcodes.ASM9; import java.util.List; -import org.objectweb.asm.ClassVisitor; -import org.objectweb.asm.MethodVisitor; - import cn.icanci.loopstack.damc.spi.model.refresh.DynamicCondition; +import damc.objectweb.asm.ClassVisitor; +import damc.objectweb.asm.MethodVisitor; public class ModifierClassVisitor extends ClassVisitor { private final List conditions; diff --git a/damc-core/src/main/java/cn/icanci/loopstack/damc/core/dynamic/visitor/VisitorModifier.java b/damc-core/src/main/java/cn/icanci/loopstack/damc/core/dynamic/visitor/VisitorModifier.java index b6e619e..25df5c6 100644 --- a/damc-core/src/main/java/cn/icanci/loopstack/damc/core/dynamic/visitor/VisitorModifier.java +++ b/damc-core/src/main/java/cn/icanci/loopstack/damc/core/dynamic/visitor/VisitorModifier.java @@ -16,16 +16,16 @@ */ package cn.icanci.loopstack.damc.core.dynamic.visitor; -import static org.objectweb.asm.Opcodes.ASM9; +import static damc.objectweb.asm.Opcodes.ASM9; import java.util.HashSet; import java.util.List; import java.util.Set; -import org.objectweb.asm.Label; -import org.objectweb.asm.MethodVisitor; -import org.objectweb.asm.Opcodes; -import org.objectweb.asm.Type; +import damc.objectweb.asm.Label; +import damc.objectweb.asm.MethodVisitor; +import damc.objectweb.asm.Opcodes; +import damc.objectweb.asm.Type; import cn.icanci.loopstack.damc.spi.model.refresh.DynamicCondition; diff --git a/spring-web-application-test/pom.xml b/spring-web-application-test/pom.xml index e46bac1..be2834a 100644 --- a/spring-web-application-test/pom.xml +++ b/spring-web-application-test/pom.xml @@ -47,16 +47,16 @@ aspectjweaver ${aspectjweaver.version} - - org.ow2.asm - asm - ${asm.version} - - - org.ow2.asm - asm-commons - ${asm.version} - + + + + + + + + + + cn.hutool hutool-all @@ -86,14 +86,14 @@ org.aspectj aspectjweaver - - org.ow2.asm - asm - - - org.ow2.asm - asm-commons - + + + + + + + + cn.hutool hutool-all -- Gitee From a5858789633efdb94389fb7b35f50adb4956c801 Mon Sep 17 00:00:00 2001 From: icanci <1845666903@qq.com> Date: Tue, 20 May 2025 19:02:59 +0800 Subject: [PATCH 2/2] =?UTF-8?q?=E5=90=84=E7=A7=8D=E7=8E=AF=E5=A2=83mock?= =?UTF-8?q?=E5=85=BC=E5=AE=B9=E5=AE=9E=E7=8E=B0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../icanci/loopstack/damc/core/dynamic/DynamicAgent.java | 2 +- .../damc/core/dynamic/DynamicAgentClassCache.java | 7 +++---- .../damc/core/dynamic/visitor/VisitorModifier.java | 3 +-- 3 files changed, 5 insertions(+), 7 deletions(-) diff --git a/damc-core/src/main/java/cn/icanci/loopstack/damc/core/dynamic/DynamicAgent.java b/damc-core/src/main/java/cn/icanci/loopstack/damc/core/dynamic/DynamicAgent.java index 8c4ee3e..0b8af45 100644 --- a/damc-core/src/main/java/cn/icanci/loopstack/damc/core/dynamic/DynamicAgent.java +++ b/damc-core/src/main/java/cn/icanci/loopstack/damc/core/dynamic/DynamicAgent.java @@ -110,7 +110,7 @@ public class DynamicAgent { // instrumentation instrumentation = inst; // 注册一个可重用的 Transformer - // inst.addTransformer(new DynamicTransformer(), true); + inst.addTransformer(new DynamicTransformer(), true); // ======================================== 运行时处理 ======================================== // 获取项目中的目标ClassLoader 和 这个进行匹配 diff --git a/damc-core/src/main/java/cn/icanci/loopstack/damc/core/dynamic/DynamicAgentClassCache.java b/damc-core/src/main/java/cn/icanci/loopstack/damc/core/dynamic/DynamicAgentClassCache.java index 2d552da..e757a55 100644 --- a/damc-core/src/main/java/cn/icanci/loopstack/damc/core/dynamic/DynamicAgentClassCache.java +++ b/damc-core/src/main/java/cn/icanci/loopstack/damc/core/dynamic/DynamicAgentClassCache.java @@ -23,13 +23,12 @@ import java.io.FileOutputStream; import java.io.IOException; import java.util.*; +import cn.icanci.loopstack.damc.core.dynamic.visitor.ModifierClassVisitor; +import cn.icanci.loopstack.damc.spi.model.refresh.DynamicConditionWrapper; import damc.objectweb.asm.ClassReader; import damc.objectweb.asm.ClassVisitor; import damc.objectweb.asm.ClassWriter; -import cn.icanci.loopstack.damc.core.dynamic.visitor.ModifierClassVisitor; -import cn.icanci.loopstack.damc.spi.model.refresh.DynamicConditionWrapper; - /** * @author icanci * @since 1.0 Created in 2025/04/21 15:05 @@ -73,7 +72,7 @@ public class DynamicAgentClassCache { } Set wrappers = CLASS_CONDITIONS.get(classBeingRedefined); ClassReader cr = getClassReader(classBeingRedefined); - ClassWriter cw = new ClassWriter(cr, ClassWriter.COMPUTE_MAXS); + ClassWriter cw = new ClassWriter(cr, ClassWriter.COMPUTE_FRAMES | ClassWriter.COMPUTE_MAXS); for (DynamicConditionWrapper wrapper : wrappers) { if (wrapper.getClassType() == ModifierClassVisitor.class) { diff --git a/damc-core/src/main/java/cn/icanci/loopstack/damc/core/dynamic/visitor/VisitorModifier.java b/damc-core/src/main/java/cn/icanci/loopstack/damc/core/dynamic/visitor/VisitorModifier.java index 25df5c6..a624fce 100644 --- a/damc-core/src/main/java/cn/icanci/loopstack/damc/core/dynamic/visitor/VisitorModifier.java +++ b/damc-core/src/main/java/cn/icanci/loopstack/damc/core/dynamic/visitor/VisitorModifier.java @@ -22,13 +22,12 @@ import java.util.HashSet; import java.util.List; import java.util.Set; +import cn.icanci.loopstack.damc.spi.model.refresh.DynamicCondition; import damc.objectweb.asm.Label; import damc.objectweb.asm.MethodVisitor; import damc.objectweb.asm.Opcodes; import damc.objectweb.asm.Type; -import cn.icanci.loopstack.damc.spi.model.refresh.DynamicCondition; - /** * http://localhost:9090/refresh/refresh * http://localhost:9090/order/create -- Gitee