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

refactor: simplify memshell generate map register

Browse Source
This commit is contained in:
ReaJason
2025-02-23 02:33:09 +08:00
parent bba2b2d211
commit d9e3e03def
177 changed files with 3761 additions and 10859 deletions
Download Patch File Download Diff File Expand all files Collapse all files
generator/src/test/java/com/reajason/javaweb/memshell/generator/GodzillaGeneratorTest.java
+4 -1
View File
@@ -1,5 +1,8 @@
package com.reajason.javaweb.memshell.generator;
import com.reajason.javaweb.memshell.ShellType;
import com.reajason.javaweb.memshell.Server;
import com.reajason.javaweb.memshell.ShellTool;
import com.reajason.javaweb.memshell.config.*;
import com.reajason.javaweb.memshell.shelltool.godzilla.GodzillaServlet;
import com.reajason.javaweb.memshell.utils.CommonUtil;
@@ -23,7 +26,7 @@ class GodzillaGeneratorTest {
ShellConfig shellConfig = ShellConfig.builder()
.server(Server.Tomcat)
.shellTool(ShellTool.Godzilla)
.shellType(Constants.SERVLET)
.shellType(ShellType.SERVLET)
.targetJreVersion(Opcodes.V1_6)
.debug(true)
.build();
generator/src/test/java/com/reajason/javaweb/memshell/packer/spel/SpELPackerTest.java
-27
View File
@@ -1,27 +0,0 @@
package com.reajason.javaweb.memshell.packer.spel;
import com.alibaba.fastjson2.JSON;
import com.alibaba.fastjson2.JSONWriter;
import com.reajason.javaweb.memshell.config.GenerateResult;
import com.reajason.javaweb.memshell.packer.AggregatePacker;
import com.reajason.javaweb.memshell.packer.Packers;
import org.junit.jupiter.api.Test;
import java.util.Map;
/**
* @author ReaJason
* @since 2025/1/26
*/
class SpELPackerTest {
@Test
void pack() {
GenerateResult generateResult = GenerateResult.builder()
.injectorClassName("name")
.injectorBytes("name".getBytes()).build();
AggregatePacker spELPacker = (AggregatePacker) Packers.SpEL.getInstance();
Map<String, String> stringStringMap = spELPacker.packAll(generateResult);
System.out.println(JSON.toJSONString(stringStringMap, JSONWriter.Feature.PrettyFormat));
}
}
generator/src/test/java/com/reajason/javaweb/memshell/tomcat/command/CommandFilterTest.java
+4 -4
View File
@@ -1,7 +1,7 @@
package com.reajason.javaweb.memshell.tomcat.command;
import com.reajason.javaweb.memshell.config.CommandConfig;
import com.reajason.javaweb.memshell.config.Constants;
import com.reajason.javaweb.memshell.ShellType;
import com.reajason.javaweb.memshell.config.ShellConfig;
import com.reajason.javaweb.memshell.generator.CommandGenerator;
import com.reajason.javaweb.memshell.shelltool.command.CommandFilter;
@@ -24,9 +24,9 @@ class CommandFilterTest {
static Stream<Arguments> casesProvider() {
return Stream.of(
arguments(Constants.FILTER, CommandFilter.class, "org.apache.utils.CommandFilter"),
arguments(Constants.LISTENER, CommandListener.class, "org.apache.utils.CommandListener"),
arguments(Constants.VALVE, CommandValve.class, "org.apache.utils.CommandValve")
arguments(ShellType.FILTER, CommandFilter.class, "org.apache.utils.CommandFilter"),
arguments(ShellType.LISTENER, CommandListener.class, "org.apache.utils.CommandListener"),
arguments(ShellType.VALVE, CommandValve.class, "org.apache.utils.CommandValve")
);
}
generator/src/test/java/com/reajason/javaweb/memshell/tomcat/godzilla/GodzillaTest.java
+4 -4
View File
@@ -1,6 +1,6 @@
package com.reajason.javaweb.memshell.tomcat.godzilla;
import com.reajason.javaweb.memshell.config.Constants;
import com.reajason.javaweb.memshell.ShellType;
import com.reajason.javaweb.memshell.config.GodzillaConfig;
import com.reajason.javaweb.memshell.config.ShellConfig;
import com.reajason.javaweb.memshell.generator.GodzillaGenerator;
@@ -32,9 +32,9 @@ class GodzillaTest {
static Stream<Arguments> casesProvider() {
return Stream.of(
arguments(Constants.FILTER, GodzillaFilter.class, "org.apache.utils.GodzillaFilter"),
arguments(Constants.LISTENER, GodzillaListener.class, "org.apache.utils.GodzillaListener"),
arguments(Constants.VALVE, GodzillaValve.class, "org.apache.utils.GodzillaValve")
arguments(ShellType.FILTER, GodzillaFilter.class, "org.apache.utils.GodzillaFilter"),
arguments(ShellType.LISTENER, GodzillaListener.class, "org.apache.utils.GodzillaListener"),
arguments(ShellType.VALVE, GodzillaValve.class, "org.apache.utils.GodzillaValve")
);
}