From 0d2f6a26b7d26bf7e53878645d66f3908d1aa5e8 Mon Sep 17 00:00:00 2001 From: ReaJason Date: Mon, 27 Jan 2025 17:06:59 +0800 Subject: [PATCH] test: build error --- .../javaweb/integration/ShellAssertionTool.java | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/integration-test/src/test/java/com/reajason/javaweb/integration/ShellAssertionTool.java b/integration-test/src/test/java/com/reajason/javaweb/integration/ShellAssertionTool.java index 18ed122c..0050e3c4 100644 --- a/integration-test/src/test/java/com/reajason/javaweb/integration/ShellAssertionTool.java +++ b/integration-test/src/test/java/com/reajason/javaweb/integration/ShellAssertionTool.java @@ -4,7 +4,7 @@ import com.reajason.javaweb.GeneratorMain; import com.reajason.javaweb.memshell.SpringWebFluxShell; import com.reajason.javaweb.memshell.SpringWebMvcShell; import com.reajason.javaweb.memshell.config.*; -import com.reajason.javaweb.memshell.packer.JarPacker; +import com.reajason.javaweb.memshell.packer.jar.JarPacker; import com.reajason.javaweb.memshell.packer.Packers; import lombok.SneakyThrows; import lombok.extern.slf4j.Slf4j; @@ -50,7 +50,7 @@ public class ShellAssertionTool { String content = null; if (packer.getInstance() instanceof JarPacker) { - byte[] bytes = packer.getInstance().packBytes(generateResult); + byte[] bytes = ((JarPacker) packer.getInstance()).packBytes(generateResult); Path tempJar = Files.createTempFile("temp", "jar"); Files.write(tempJar, bytes); String jarPath = "/" + shellTool + shellType + packer.name() + ".jar"; @@ -135,6 +135,13 @@ public class ShellAssertionTool { VulTool.uploadJspFileToServer(uploadEntry, filename, content); VulTool.urlIsOk(shellUrl); } + case JSPX -> { + String uploadEntry = url + "/upload"; + String filename = shellType + shellTool + ".jspx"; + String shellUrl = url + "/" + filename; + VulTool.uploadJspFileToServer(uploadEntry, filename, content); + VulTool.urlIsOk(shellUrl); + } case ScriptEngine -> VulTool.postData(url + "/js", content); case EL -> VulTool.postData(url + "/el", content); case SpEL -> VulTool.postData(url + "/spel", content);