archive/MemShellParty
1
0
Fork 0
mirror of https://github.com/ReaJason/MemShellParty.git synced 2026-09-24 07:51:52 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: probe custom shellClassName not work

Browse Source
This commit is contained in:
ReaJason
2025-09-12 22:51:50 +08:00
parent 7b95367e4e
commit 8eea712f4d
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
generator/src/main/java/com/reajason/javaweb/probe/generator/DnsLogGenerator.java
+2 -2
View File
@@ -36,13 +36,13 @@ public class DnsLogGenerator extends ByteBuddyShellGenerator<DnsLogConfig> {
switch (detectContent) { switch (detectContent) {
case Server: case Server:
return buddy.redefine(DnsLogServer.class) return buddy.redefine(DnsLogServer.class)
.name(CommonUtil.generateShellClassName()) .name(probeConfig.getShellClassName())
.visit(TargetJreVersionVisitorWrapper.DEFAULT) .visit(TargetJreVersionVisitorWrapper.DEFAULT)
.field(named("host")).value(probeContentConfig.getHost()) .field(named("host")).value(probeContentConfig.getHost())
.visit(Advice.to(ServerProbe.class).on(named("getServer"))); .visit(Advice.to(ServerProbe.class).on(named("getServer")));
case JDK: case JDK:
return buddy.redefine(DnsLogJdk.class) return buddy.redefine(DnsLogJdk.class)
.name(CommonUtil.generateShellClassName()) .name(probeConfig.getShellClassName())
.visit(TargetJreVersionVisitorWrapper.DEFAULT) .visit(TargetJreVersionVisitorWrapper.DEFAULT)
.field(named("host")).value(probeContentConfig.getHost()) .field(named("host")).value(probeContentConfig.getHost())
.visit(Advice.to(JdkProbe.class).on(named("getJdk"))); .visit(Advice.to(JdkProbe.class).on(named("getJdk")));
generator/src/main/java/com/reajason/javaweb/probe/generator/SleepGenerator.java
+1 -1
View File
@@ -37,7 +37,7 @@ public class SleepGenerator extends ByteBuddyShellGenerator<SleepConfig> {
throw new GenerationException("sleepProbe server must be specified"); throw new GenerationException("sleepProbe server must be specified");
} }
return buddy.redefine(SleepServer.class) return buddy.redefine(SleepServer.class)
.name(CommonUtil.generateShellClassName()) .name(probeConfig.getShellClassName())
.visit(TargetJreVersionVisitorWrapper.DEFAULT) .visit(TargetJreVersionVisitorWrapper.DEFAULT)
.field(named("server")).value(probeContentConfig.getServer()) .field(named("server")).value(probeContentConfig.getServer())
.field(named("seconds")).value(probeContentConfig.getSeconds()) .field(named("seconds")).value(probeContentConfig.getSeconds())