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

1、whitelist增加whiteListCommandsCheck、whiteListStaticRulesCheck。支持前端发送staticRuleIds列表，查询静态规则是否存在白名单。记录白名单check指令的问题

Browse Source
This commit is contained in:
Hao Miao
2024-01-25 01:25:42 +08:00
parent 02dd20743f
commit f368c312ba
8 changed files with 205 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
src/main/java/com/realtime/protection/server/whitelist/WhiteListController.java
View File

@@ -171,5 +171,28 @@ public class WhiteListController implements WhiteListControllerApi {
}
/*
白名单审核思路：
（1）按现在界面：任务创建时提示白名单命中。
①静态任务：任务还没通过审核进入RUNNING状态，指令也没有生成，目前只能check静态规则在不在白名单
②动态任务：任务还没通过审核进入RUNNING状态，告警信息没有接收、指令也没有生成，check不了白名单
（2）若对指令进行白名单检查：（直接剔除指令中存在的白名单对象）
存在的问题：指令如果是含有掩码的子网IP，就需要把子网拆分成一个个IP，从中剔除白名单IP，然后把一个个IP变成指令。
也就是把含有掩码的指令拆分成不包含掩码的多个指令，此时指令展示会很多。
①静态任务：指令生成后，自动剔除，若出现掩码，存在以上问题
②实时任务：告警接收后，经策略模板指令生成后，自动剔除白名单对象，若出现掩码，存在以上问题
③研判后认务：告警接收后，经策略模板指令生成后，自动剔除白名单对象，若出现掩码，包含掩码的指令都拆分多个指令了，
告警信息如果不拆的话，告警信息只存一个指令的UUID。无法查询拆分后的多个指令对应哪个告警信息。
拆分成多个指令也不好研判下发。
如果说直接研判告警信息的话，多个告警信息还可能对应一个指令，下发了这个告警信息，另一个下发吗？所以研判的应该是指令
*/
@Override
@GetMapping("/staticrule/check/{staticRuleIds}")
public ResponseResult whiteListStaticRulesCheck(@PathVariable List<Integer> staticRuleIds) {
return ResponseResult.ok()
.setData("whiteobj_list", whiteListService.whiteListStaticRulesCheck(staticRuleIds));
}
}

35
src/main/java/com/realtime/protection/server/whitelist/WhiteListControllerApi.java
View File

@@ -9,6 +9,7 @@ import io.swagger.v3.oas.annotations.media.ExampleObject;
import io.swagger.v3.oas.annotations.media.Schema;
import io.swagger.v3.oas.annotations.tags.Tag;
import jakarta.servlet.http.HttpServletResponse;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestParam;
@@ -205,4 +206,38 @@ public interface WhiteListControllerApi {
@PathVariable Integer auditStatus);
/*
白名单审核思路：
（1）按现在界面：任务创建时提示白名单命中。
①静态任务：任务还没通过审核进入RUNNING状态，指令也没有生成，目前只能check静态规则在不在白名单
②动态任务：任务还没通过审核进入RUNNING状态，告警信息没有接收、指令也没有生成，check不了白名单
（2）若对指令进行白名单检查：（直接剔除指令中存在的白名单对象）
存在的问题：指令如果是含有掩码的子网IP，就需要把子网拆分成一个个IP，从中剔除白名单IP，然后把一个个IP变成指令。
也就是把含有掩码的指令拆分成不包含掩码的多个指令，此时指令展示会很多。
①静态任务：指令生成后，自动剔除，若出现掩码，存在以上问题
②实时任务：告警接收后，经策略模板指令生成后，自动剔除白名单对象，若出现掩码，存在以上问题
③研判后认务：告警接收后，经策略模板指令生成后，自动剔除白名单对象，若出现掩码，包含掩码的指令都拆分多个指令了，
告警信息如果不拆的话，告警信息只存一个指令的UUID。无法查询拆分后的多个指令对应哪个告警信息。
拆分成多个指令也不好研判下发。
如果说直接研判告警信息的话，多个告警信息还可能对应一个指令，下发了这个告警信息，另一个下发吗？所以研判的应该是指令
*/
@Operation(
summary = "查询静态规则命中白名单",
description = "用于在创建任务的第三页中，查询静态规则命中的白名单",
responses = {
@io.swagger.v3.oas.annotations.responses.ApiResponse(
description = "返回静态规则命中的所有白名单",
content = @Content(
mediaType = "application/json",
schema = @Schema(title = "ResponseResult和WhiteListObject的属性",
anyOf = {ResponseResult.class, WhiteListObject.class})
)
)
},
parameters = {
@Parameter(name = "staticRuleIds", description = "静态规则id")
}
)
ResponseResult whiteListStaticRulesCheck(@PathVariable List<Integer> staticRuleIds);
}

7
src/main/java/com/realtime/protection/server/whitelist/WhiteListMapper.java
View File

@@ -2,6 +2,7 @@ package com.realtime.protection.server.whitelist;
import com.realtime.protection.configuration.entity.rule.staticrule.StaticRuleObject;
import com.realtime.protection.configuration.entity.task.FiveTupleWithMask;
import com.realtime.protection.configuration.entity.task.TaskCommandInfo;
import com.realtime.protection.configuration.entity.whitelist.WhiteListObject;
import org.apache.ibatis.annotations.Delete;
import org.apache.ibatis.annotations.Mapper;
@@ -37,7 +38,11 @@ public interface WhiteListMapper {
void deleteWhiteListObjects(@Param("whiteListIds") List<Integer> whiteListBatch);
List<WhiteListObject> whiteListCommandJudge(@Param("command") FiveTupleWithMask fiveTupleWithMaskInCommand);
List<WhiteListObject> whiteListCommandCheck(@Param("command") FiveTupleWithMask fiveTupleWithMaskInCommand);
Integer queryWhiteListTotalNum(String whiteListName, Integer whiteListId);
List<WhiteListObject> whiteListCommandsCheck(List<TaskCommandInfo> taskCommandInfos);
List<WhiteListObject> whiteListCStaticRulesCheck(@Param("staticRuleObjects") List<StaticRuleObject> staticRuleObjects);
}

20
src/main/java/com/realtime/protection/server/whitelist/WhiteListService.java
View File

@@ -129,10 +129,26 @@ public class WhiteListService {
}
public List<WhiteListObject> whiteListCommandJudge(TaskCommandInfo taskCommandInfo) {
public List<WhiteListObject> whiteListCommandCheck(TaskCommandInfo taskCommandInfo) {
//参数应该是指令，不管动态静态
// 命中的whitelist列表：每一列包含ip port url
return whiteListMapper.whiteListCommandJudge(taskCommandInfo.getFiveTupleWithMask());
return whiteListMapper.whiteListCommandCheck(taskCommandInfo.getFiveTupleWithMask());
}
public List<WhiteListObject> whiteListCommandsCheck(List<TaskCommandInfo> taskCommandInfoList) {
//参数应该是指令，不管动态静态
// 命中的whitelist列表：每一列包含ip port url
return whiteListMapper.whiteListCommandsCheck(taskCommandInfoList);
}
public List<WhiteListObject> whiteListStaticRulesCheck(List<Integer> staticRuleId) {
//参数应该是指令，不管动态静态
// 命中的whitelist列表：每一列包含ip port url
List<StaticRuleObject> staticRuleObjects = staticRuleMapper.queryStaticRuleByIds(staticRuleId);
return whiteListMapper.whiteListCStaticRulesCheck(staticRuleObjects);
}