From 88af60fea4c6127b3a745a24c6a3b573a0d62c6a Mon Sep 17 00:00:00 2001 From: ReaJason Date: Wed, 27 Aug 2025 12:47:38 +0800 Subject: [PATCH] fix: probe custom shellClassName not work --- .../com/reajason/javaweb/probe/generator/DnsLogGenerator.java | 4 ++-- .../com/reajason/javaweb/probe/generator/SleepGenerator.java | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/generator/src/main/java/com/reajason/javaweb/probe/generator/DnsLogGenerator.java b/generator/src/main/java/com/reajason/javaweb/probe/generator/DnsLogGenerator.java index 74fd73b6..22deb638 100644 --- a/generator/src/main/java/com/reajason/javaweb/probe/generator/DnsLogGenerator.java +++ b/generator/src/main/java/com/reajason/javaweb/probe/generator/DnsLogGenerator.java @@ -36,13 +36,13 @@ public class DnsLogGenerator extends ByteBuddyShellGenerator { switch (detectContent) { case Server: return buddy.redefine(DnsLogServer.class) - .name(CommonUtil.generateShellClassName()) + .name(probeConfig.getShellClassName()) .visit(TargetJreVersionVisitorWrapper.DEFAULT) .field(named("host")).value(probeContentConfig.getHost()) .visit(Advice.to(ServerProbe.class).on(named("getServer"))); case JDK: return buddy.redefine(DnsLogJdk.class) - .name(CommonUtil.generateShellClassName()) + .name(probeConfig.getShellClassName()) .visit(TargetJreVersionVisitorWrapper.DEFAULT) .field(named("host")).value(probeContentConfig.getHost()) .visit(Advice.to(JdkProbe.class).on(named("getJdk"))); diff --git a/generator/src/main/java/com/reajason/javaweb/probe/generator/SleepGenerator.java b/generator/src/main/java/com/reajason/javaweb/probe/generator/SleepGenerator.java index e38d85dc..24066a14 100644 --- a/generator/src/main/java/com/reajason/javaweb/probe/generator/SleepGenerator.java +++ b/generator/src/main/java/com/reajason/javaweb/probe/generator/SleepGenerator.java @@ -37,7 +37,7 @@ public class SleepGenerator extends ByteBuddyShellGenerator { throw new GenerationException("sleepProbe server must be specified"); } return buddy.redefine(SleepServer.class) - .name(CommonUtil.generateShellClassName()) + .name(probeConfig.getShellClassName()) .visit(TargetJreVersionVisitorWrapper.DEFAULT) .field(named("server")).value(probeContentConfig.getServer()) .field(named("seconds")).value(probeContentConfig.getSeconds())