archive/MemShellParty
1
0
Fork 0
mirror of https://github.com/ReaJason/MemShellParty.git synced 2026-09-22 07:00:43 +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 9b7d244837
commit 88af60fea4
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) {
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")));
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");
}
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())