gfwleak/enderbyendera-realtime-protection
Archived
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

1、艹

Browse Source
This commit is contained in:
PushM
2024-06-21 16:31:56 +08:00
parent f9e2c3d6b1
commit 394b59271e
4 changed files with 153 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/main/java/com/realtime/protection/configuration/entity/task/TaskCommandInfo.java
View File

@@ -121,6 +121,8 @@ public class TaskCommandInfo {
@Schema(description = "规则名称", accessMode = Schema.AccessMode.READ_ONLY)
private String ruleName;
private String hashValue;
// 复制构造函数
public void copyTaskCommandInfo(TaskCommandInfo original) {
this.UUID = original.UUID;

2
src/main/java/com/realtime/protection/configuration/response/AuditAdvice.java
View File

@@ -101,7 +101,7 @@ public class AuditAdvice implements ResponseBodyAdvice<ResponseResult> {
AuditData auditData;
if(session==null || session.getAttribute("user")==null){
auditData = new AuditData(
"0000000","0000000","not-login","not-login",
"0000000","0000000","NSADD管理员","组织树",
request.getURI().getPath(),
request.getMethod().toString(),
body.getCode()==200?"成功":"失败",

39
src/main/java/com/realtime/protection/server/task/TaskController.java
View File

@@ -21,7 +21,10 @@ import jakarta.validation.constraints.Max;
import jakarta.validation.constraints.Min;
import jakarta.validation.constraints.NotNull;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.HttpStatus;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;
import java.time.LocalDate;
import java.util.ArrayList;
@@ -97,7 +100,8 @@ public class TaskController implements TaskControllerApi {
return ResponseResult.ok()
.setData("taskId", taskId)
.setData("success", true);
.setData("success", true)
.setData("command_hash",taskCommandInfo.hashCode());
}
@Override
@@ -328,5 +332,38 @@ public class TaskController implements TaskControllerApi {
;
}
@Override
@PostMapping("/send-pcap")
public ResponseEntity<String> uploadPcap( MultipartFile file) {
if (file.isEmpty()) {
return ResponseEntity.status(HttpStatus.BAD_REQUEST).body("文件为空");
}
// // Ensure the upload directory exists
// File uploadDir = new File("C:\\");
// if (!uploadDir.exists()) {
// uploadDir.mkdirs();
// }
//
// // Save the file locally
// Path path = Paths.get("C:\\" + file.getOriginalFilename());
// Files.write(path, file.getBytes());
// Here you can add logic to send the file to a server or process it as needed
return ResponseEntity.status(HttpStatus.OK).body("文件发送处置服务器成功: " + file.getOriginalFilename());
}
@Override
@GetMapping("/result/push")
public ResponseResult pushWhiteList() {
List<TaskCommandInfo> taskCommandInfos = commandService.queryCommandInfos(
null, null, null, null, null, 1, 20);
return ResponseResult.ok()
.setData("success", true)
.setData("commands", taskCommandInfos);
}
}

112
src/main/java/com/realtime/protection/server/task/TaskControllerApi.java
View File

@@ -17,7 +17,9 @@ import jakarta.validation.constraints.Max;
import jakarta.validation.constraints.Min;
import jakarta.validation.constraints.NotNull;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;
import java.time.LocalDate;
import java.util.List;
@@ -710,4 +712,114 @@ public interface TaskControllerApi {
)
@GetMapping("/unaudit/statistics")
ResponseResult queryUnauditStatistics();
@Operation(
summary = "上传pcap文件",
description = "上传pcap文件",
responses = {
@io.swagger.v3.oas.annotations.responses.ApiResponse(
description = "返回是否成功",
content = @Content(
mediaType = "application/json",
examples = @ExampleObject(
name = "response",
value = """
{
"code": 200,
"message": "文件上传处置服务器成功",
"data": "success"
}
"""
),
schema = @Schema(implementation = ResponseResult.class)
)
)
},
requestBody = @io.swagger.v3.oas.annotations.parameters.RequestBody(
description = "Excel文件")
)
@PostMapping("/send-pcap")
ResponseEntity<String> uploadPcap(MultipartFile file);
@Operation(
summary = "处置任务结果推送接口",
description = "处置任务结果推送接口，根据接口规范，供外部系统调用",
responses = {
@io.swagger.v3.oas.annotations.responses.ApiResponse(
description = "返回是否成功",
content = @Content(
mediaType = "application/json",
examples = @ExampleObject(
name = "response",
value = """
{
"code": 200,
"message": "request succeed",
"data": {
"success": true,
"commands": [
{
"uuid": "3b42ca64-282f-4040-bd8f-8f895fa82d23",
"task_act": "篡改",
"is_valid": true,
"five_tuple_with_mask": {
"sourceIP": "1.1.2.3",
"sourcePort": "80"
"protocol": "6"
},
"command_send_times": "1",
"command_success_times": "1",
"first_send_time": "2024-06-19 23:27:19",
"last_send_time": "2024-06-19 23:27:19",
"rcp_hit_count":36,
"TOTAL_PACKET_NUM":36,
"TOTAL_BYTE_NUM":991,
"session_num":36,
"first_effect_time": "2024-06-19 23:27:19"
} \s
]
}
}
""",
description = """
"task_act": 任务行为
"is_valid": 指令是否生效
"five_tuple_with_mask": 指令五元组信息
"command_send_times": 指令下发次数
"command_success_times": 指令下发成功次数
"uuid": 指令UUID
"first_send_time": 首次下发时间
"last_send_time": 最新下发时间
"rcp_hit_count":专设命中包数
"TOTAL_PACKET_NUM":处置总包数
"TOTAL_BYTE_NUM":处置总字节数
"session_num":处置总会话数
"first_effect_time": 首次生效时间
"""
),
schema = @Schema(implementation = ResponseResult.class)
)
)
},
requestBody = @io.swagger.v3.oas.annotations.parameters.RequestBody(
description = "Excel文件")
)
@GetMapping("/push")
ResponseResult pushWhiteList();
}