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

feat: support boot-ui

Browse Source
This commit is contained in:
ReaJason
2024-12-20 01:15:12 +08:00
parent 2b00c96a06
commit 534db31fa0
81 changed files with 1556 additions and 929 deletions
Download Patch File Download Diff File Expand all files Collapse all files
boot/Dockerfile
+12
View File
@@ -0,0 +1,12 @@
FROM eclipse-temurin:17-jre
LABEL authors="ReaJason<[email protected]>"
WORKDIR /app
COPY build/libs/boot.jar app.jar
ENV JAVA_OPTS="" \
INTER_JAVA_OPTS="--add-opens=java.base/java.util=ALL-UNNAMED --add-opens=java.xml/com.sun.org.apache.xalan.internal.xsltc.trax=ALL-UNNAMED --add-opens=java.xml/com.sun.org.apache.xalan.internal.xsltc.runtime=ALL-UNNAMED"
EXPOSE 8080
ENTRYPOINT java $JAVA_OPTS $INTER_JAVA_OPTS -jar app.jar
boot/build.gradle
+33 -7
View File
@@ -5,7 +5,7 @@ plugins {
}
group = 'com.reajason.javaweb'
version = '0.0.1-SNAPSHOT'
version = ''
java {
toolchain {
@@ -13,21 +13,47 @@ java {
}
}
def runtimeJvmArgs = [
'--add-opens=java.base/java.util=ALL-UNNAMED',
'--add-opens=java.xml/com.sun.org.apache.xalan.internal.xsltc.trax=ALL-UNNAMED',
'--add-opens=java.xml/com.sun.org.apache.xalan.internal.xsltc.runtime=ALL-UNNAMED'
]
tasks.withType(JavaCompile).configureEach {
options.compilerArgs += [
'--add-exports=java.xml/com.sun.org.apache.xalan.internal.xsltc.trax=ALL-UNNAMED',
'--add-exports=java.xml/com.sun.org.apache.xalan.internal.xsltc.runtime=ALL-UNNAMED'
]
}
tasks.withType(Test).configureEach {
jvmArgs += runtimeJvmArgs
}
// For running the application
tasks.withType(JavaExec).configureEach {
jvmArgs += runtimeJvmArgs
}
configurations {
compileOnly {
extendsFrom annotationProcessor
}
}
repositories {
mavenCentral()
}
dependencies {
implementation project(":deserialize")
implementation project(":generator")
implementation(project(":generator")) {
exclude group: 'org.apache.tomcat', module: 'tomcat-catalina'
exclude group: 'commons-logging', module: 'commons-logging'
}
implementation(project(":deserialize")) {
exclude group: 'commons-logging', module: 'commons-logging'
}
implementation 'org.apache.bcel:bcel:5.2'
implementation 'org.springframework.boot:spring-boot-starter-thymeleaf'
implementation 'org.springframework.boot:spring-boot-starter-web'
implementation 'org.springframework.boot:spring-boot-starter-tomcat'
implementation 'com.google.code.gson:gson:2.11.0'
compileOnly 'org.projectlombok:lombok'
developmentOnly 'org.springframework.boot:spring-boot-devtools'
annotationProcessor 'org.springframework.boot:spring-boot-configuration-processor'
boot/src/main/java/com/reajason/javaweb/boot/BootApplication.java
+1 -1
View File
@@ -13,4 +13,4 @@ public class BootApplication {
SpringApplication.run(BootApplication.class, args);
}
}
}
boot/src/main/java/com/reajason/javaweb/boot/api/ErrorResponse.java
+16
View File
@@ -0,0 +1,16 @@
package com.reajason.javaweb.boot.api;
import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;
/**
* @author ReaJason
* @since 2024/5/30
*/
@Data
@NoArgsConstructor
@AllArgsConstructor
public class ErrorResponse {
private String error;
}
boot/src/main/java/com/reajason/javaweb/boot/api/GlobalExceptionHandler.java
+22
View File
@@ -0,0 +1,22 @@
package com.reajason.javaweb.boot.api;
import lombok.extern.slf4j.Slf4j;
import org.springframework.http.HttpStatus;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.ResponseStatus;
import org.springframework.web.bind.annotation.RestControllerAdvice;
/**
* @author ReaJason
* @since 2024/5/30
*/
@RestControllerAdvice
@Slf4j
public class GlobalExceptionHandler {
@ResponseStatus(HttpStatus.INTERNAL_SERVER_ERROR)
@ExceptionHandler(Throwable.class)
public ErrorResponse handleThrowable(Throwable throwable) {
log.error("请求出错", throwable);
return new ErrorResponse(throwable.getMessage());
}
}
boot/src/main/java/com/reajason/javaweb/boot/controller/ConfigController.java
+19 -18
View File
@@ -3,28 +3,27 @@ package com.reajason.javaweb.boot.controller;
import com.reajason.javaweb.boot.entity.Config;
import com.reajason.javaweb.config.Server;
import com.reajason.javaweb.config.ShellTool;
import com.reajason.javaweb.deserialize.PayloadType;
import com.reajason.javaweb.memsell.AbstractShell;
import com.reajason.javaweb.memsell.packer.Packer;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.CrossOrigin;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.*;
import java.util.stream.Collectors;
/**
* @author ReaJason
* @since 2024/12/13
*/
@RestController("/config")
@RestController
@RequestMapping("/config")
@CrossOrigin("*")
public class ConfigController {
@RequestMapping
public ResponseEntity<?> config() {
Map<String, Map<?, ?>> coreMap = new HashMap<>(16);
List<String> servers = new ArrayList<>();
for (Server value : Server.values()) {
AbstractShell shell = value.getShell();
if (shell != null) {
@@ -34,22 +33,24 @@ public class ConfigController {
List<String> supportedShellTypes = shell.getSupportedShellTypes(shellTool);
map.put(shellTool.name(), supportedShellTypes);
}
servers.add(value.name());
coreMap.put(value.name(), map);
}
}
Config config = new Config();
config.setServers(servers);
config.setServers(
Arrays.stream(Server.values())
.filter(s -> s.getShell() != null)
.map(Server::name)
.collect(Collectors.toList())
);
config.setCore(coreMap);
Map<String, Map<?, ?>> packerMap = new HashMap<>(16);
for (Packer.INSTANCE value : Packer.INSTANCE.values()) {
if (value.equals(Packer.INSTANCE.Deserialize)) {
packerMap.put(value.name(), Map.of("payloads", PayloadType.values()));
} else {
packerMap.put(value.name(), null);
}
}
config.setPacker(packerMap);
config.setPackers(Arrays.stream(Packer.INSTANCE.values())
.collect(Collectors.toMap(
Packer.INSTANCE::getDesc,
Packer.INSTANCE::name,
(e1, e2) -> e1,
LinkedHashMap::new
)));
return ResponseEntity.ok(config);
}
}
boot/src/main/java/com/reajason/javaweb/boot/controller/GeneratorController.java
+30
View File
@@ -0,0 +1,30 @@
package com.reajason.javaweb.boot.controller;
import com.reajason.javaweb.GeneratorMain;
import com.reajason.javaweb.boot.dto.GenerateRequest;
import com.reajason.javaweb.boot.dto.GenerateResponse;
import com.reajason.javaweb.config.GenerateResult;
import com.reajason.javaweb.config.InjectorConfig;
import com.reajason.javaweb.config.ShellConfig;
import com.reajason.javaweb.config.ShellToolConfig;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.*;
/**
* @author ReaJason
* @since 2024/12/18
*/
@RestController
@RequestMapping("/generate")
@CrossOrigin("*")
public class GeneratorController {
@PostMapping
public ResponseEntity<?> generate(@RequestBody GenerateRequest request) {
ShellConfig shellConfig = request.getShellConfig();
ShellToolConfig shellToolConfig = request.parseShellToolConfig();
InjectorConfig injectorConfig = request.getInjectorConfig();
GenerateResult generateResult = GeneratorMain.generate(shellConfig, injectorConfig, shellToolConfig);
String packResult = request.getPacker().getPacker().pack(generateResult);
return ResponseEntity.ok(new GenerateResponse(generateResult, packResult));
}
}
boot/src/main/java/com/reajason/javaweb/boot/controller/ViewController.java
+16
View File
@@ -0,0 +1,16 @@
package com.reajason.javaweb.boot.controller;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.GetMapping;
/**
* @author ReaJason
* @since 2024/12/19
*/
@Controller
public class ViewController {
@GetMapping("/")
public String index() {
return "index";
}
}
boot/src/main/java/com/reajason/javaweb/boot/dto/GenerateRequest.java
+46
View File
@@ -0,0 +1,46 @@
package com.reajason.javaweb.boot.dto;
import com.reajason.javaweb.config.*;
import com.reajason.javaweb.memsell.packer.Packer;
import lombok.Data;
/**
* @author ReaJason
* @since 2024/12/18
*/
@Data
public class GenerateRequest {
private ShellConfig shellConfig;
private ShellToolConfigDTO shellToolConfig;
private InjectorConfig injectorConfig;
private Packer.INSTANCE packer;
public ShellToolConfig parseShellToolConfig() {
if (shellConfig.getShellTool().equals(ShellTool.Godzilla)) {
return GodzillaConfig.builder()
.shellClassName(shellToolConfig.getShellClassName())
.pass(shellToolConfig.getGodzillaPass())
.key(shellToolConfig.getGodzillaKey())
.headerName(shellToolConfig.getGodzillaHeaderName())
.headerValue(shellToolConfig.getGodzillaHeaderValue())
.build();
}
if (shellConfig.getShellTool().equals(ShellTool.Command)) {
return CommandConfig.builder()
.shellClassName(shellToolConfig.getShellClassName())
.paramName(shellToolConfig.getCommandParamName())
.build();
}
throw new UnsupportedOperationException("unknown shell tool " + shellConfig.getShellTool());
}
@Data
static class ShellToolConfigDTO {
private String shellClassName;
private String godzillaPass;
private String godzillaKey;
private String godzillaHeaderName;
private String godzillaHeaderValue;
private String commandParamName;
}
}
boot/src/main/java/com/reajason/javaweb/boot/dto/GenerateResponse.java
+16
View File
@@ -0,0 +1,16 @@
package com.reajason.javaweb.boot.dto;
import com.reajason.javaweb.config.GenerateResult;
import lombok.AllArgsConstructor;
import lombok.Data;
/**
* @author ReaJason
* @since 2024/12/18
*/
@Data
@AllArgsConstructor
public class GenerateResponse {
private GenerateResult generateResult;
private String packResult;
}
boot/src/main/java/com/reajason/javaweb/boot/entity/Config.java
+1 -1
View File
@@ -13,5 +13,5 @@ import java.util.Map;
public class Config {
private List<String> servers;
private Map<String, Map<?, ?>> core;
private Map<String, Map<?, ?>> packer;
private Map<String, String> packers;
}