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

fix: custom shell generate failed

Browse Source
This commit is contained in:
ReaJason
2025-11-20 00:36:25 +08:00
parent dee1c4a5d0
commit 33944dfd32
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
generator/src/main/java/com/reajason/javaweb/memshell/generator/CustomShellGenerator.java
+6 -3
View File
@@ -31,13 +31,16 @@ public class CustomShellGenerator extends ByteBuddyShellGenerator<CustomConfig>
if (StringUtils.isBlank(shellToolConfig.getShellClassName())) { if (StringUtils.isBlank(shellToolConfig.getShellClassName())) {
shellToolConfig.setShellClassName(className); shellToolConfig.setShellClassName(className);
} }
ClassFileLocator classFileLocator = ClassFileLocator.Simple.of(className, classBytes); ClassFileLocator compoundLocator = new ClassFileLocator.Compound(
ClassFileLocator.Simple.of(className, classBytes),
ClassFileLocator.ForClassLoader.of(this.getClass().getClassLoader())
);
TypeDescription typeDescription = new TypePool.Default( TypeDescription typeDescription = new TypePool.Default(
new TypePool.CacheProvider.Simple(), classFileLocator, new TypePool.CacheProvider.Simple(), compoundLocator,
TypePool.Default.ReaderMode.FAST, TypePool.Default.ofSystemLoader() TypePool.Default.ReaderMode.FAST, TypePool.Default.ofSystemLoader()
).describe(className).resolve(); ).describe(className).resolve();
shellToolConfig.setShellTypeDescription(typeDescription); shellToolConfig.setShellTypeDescription(typeDescription);
return new ByteBuddy() return new ByteBuddy()
.redefine(typeDescription, classFileLocator); .redefine(typeDescription, compoundLocator);
} }
} }