From 1de0c27c365d2045bb43b9ba04af18b260af6abe Mon Sep 17 00:00:00 2001
From: wangkuan <wangkuan@geedgenetworks.com>
Date: Mon, 30 Sep 2024 10:40:01 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8DTSG-22710=20DoS=20Event=20?=
 =?UTF-8?q?=E5=9F=BA=E4=BA=8E=E9=9D=99=E6=80=81=E6=95=8F=E6=84=9F=E9=98=88?=
 =?UTF-8?q?=E5=80=BC=E8=BE=93=E5=87=BA=E7=9A=84=E5=91=8A=E8=AD=A6=E4=BA=8B?=
 =?UTF-8?q?=E4=BB=B6=E4=B8=8E=E5=88=A4=E5=AE=9A=E6=9D=A1=E4=BB=B6=E4=B8=8D?=
 =?UTF-8?q?=E7=AC=A6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 pom.xml                                              |  4 ++--
 .../com/zdjizhi/function/DosDetectionFunction.java   | 12 ++++++------
 src/test/java/com/zdjizhi/etl/DosDetectionTest.java  |  2 +-
 3 files changed, 9 insertions(+), 9 deletions(-)

diff --git a/pom.xml b/pom.xml
index 9d51f4b..8659031 100644
--- a/pom.xml
+++ b/pom.xml
@@ -6,7 +6,7 @@
 
     <groupId>com.zdjizhi</groupId>
     <artifactId>flink-dos-detection</artifactId>
-    <version>24-04-21</version>
+    <version>24-04-22</version>
 
     <name>flink-dos-detection</name>
     <url>http://www.example.com</url>
@@ -80,7 +80,7 @@
                         </goals>
 
                         <configuration>
-                            <finalName>flink-dos-detection-24-04-21</finalName>
+                            <finalName>flink-dos-detection-24-04-22</finalName>
                             <relocations>
                                 <relocation>
                                     <pattern>org.apache.http</pattern>
diff --git a/src/main/java/com/zdjizhi/function/DosDetectionFunction.java b/src/main/java/com/zdjizhi/function/DosDetectionFunction.java
index 9db50b9..6fd95f8 100644
--- a/src/main/java/com/zdjizhi/function/DosDetectionFunction.java
+++ b/src/main/java/com/zdjizhi/function/DosDetectionFunction.java
@@ -101,18 +101,18 @@ public class DosDetectionFunction extends ProcessFunction<DosSketchLog, DosEvent
 
 
     private DosEventLog getDosEventLogBySensitivityThreshold(DosSketchLog value) {
-        long sketchSessions = value.getSessions();
+        long sketchSessionsRate = value.getSession_rate();
         Integer staticSensitivityThreshold = configuration.get(STATIC_SENSITIVITY_THRESHOLD);
-        long diff = sketchSessions - staticSensitivityThreshold;
+        long diff = sketchSessionsRate - staticSensitivityThreshold;
         return getDosEventLog(value, staticSensitivityThreshold, diff, 0, SENSITIVITY_CONDITION_TYPE, SESSIONS_TAG);
     }
 
     private DosEventLog getDosEventLogByBaseline(DosSketchLog value, String key) {
         String attackType = value.getAttack_type();
-        long sketchSessions = value.getSessions();
+        long sketchSessionsRate = value.getSession_rate();
         DosBaselineThreshold dosBaselineThreshold = baselineMap.get(key).get(attackType);
         Integer base = getBaseValue(dosBaselineThreshold, value);
-        long diff = sketchSessions - base;
+        long diff = sketchSessionsRate - base;
         return getDosEventLog(value, base, diff, 0, BASELINE_CONDITION_TYPE, SESSIONS_TAG);
     }
 
@@ -127,7 +127,7 @@ public class DosDetectionFunction extends ProcessFunction<DosSketchLog, DosEvent
             if (severity != Severity.NORMAL) {
                 if (type == BASELINE_CONDITION_TYPE && percent < configuration.get(BASELINE_SENSITIVITY_THRESHOLD)) {
                     logger.debug("当前server IP:{},类型:{},基线值{}百分比{}未超过基线敏感阈值,日志详情\n{}", destinationIp, attackType, base, percent, value);
-                } else if ((type == BASELINE_CONDITION_TYPE || type == SENSITIVITY_CONDITION_TYPE) && value.getSessions() < staticSensitivityThreshold) {
+                } else if ((type == BASELINE_CONDITION_TYPE || type == SENSITIVITY_CONDITION_TYPE) && value.getSession_rate() < staticSensitivityThreshold) {
                     logger.debug("当前server IP:{},类型:{},基线值{}百分比{}未超过静态敏感阈值,日志详情\n{}", destinationIp, attackType, base, percent, value);
                 } else {
                     result = getResult(value, base, profileId, severity, percent, type, tag);
@@ -155,7 +155,7 @@ public class DosDetectionFunction extends ProcessFunction<DosSketchLog, DosEvent
         dosEventLog.setAttack_type(value.getAttack_type());
         if(base != 0) {
             dosEventLog.setSeverity(severity.severity);
-            dosEventLog.setConditions(getConditions(PERCENT_INSTANCE.format(percent), base, value.getSessions(), type, tag, dosEventLog));
+            dosEventLog.setConditions(getConditions(PERCENT_INSTANCE.format(percent), base, value.getSession_rate(), type, tag, dosEventLog));
         }
         else{
             dosEventLog.setSeverity(severity.severity);
diff --git a/src/test/java/com/zdjizhi/etl/DosDetectionTest.java b/src/test/java/com/zdjizhi/etl/DosDetectionTest.java
index 97b2384..1daae81 100644
--- a/src/test/java/com/zdjizhi/etl/DosDetectionTest.java
+++ b/src/test/java/com/zdjizhi/etl/DosDetectionTest.java
@@ -94,7 +94,7 @@ public class DosDetectionTest {
             if (severity != Severity.NORMAL) {
                 if (type == BASELINE_CONDITION_TYPE && percent < 0.2) {
 //                    logger.debug("当前server IP:{},类型:{},基线值{}百分比{}未超过基线敏感阈值,日志详情\n{}", destinationIp, attackType, base, percent, value);
-                }else if ((type == BASELINE_CONDITION_TYPE || type == SENSITIVITY_CONDITION_TYPE) && value.getSessions() < staticSensitivityThreshold){
+                }else if ((type == BASELINE_CONDITION_TYPE || type == SENSITIVITY_CONDITION_TYPE) && value.getSession_rate() < staticSensitivityThreshold){
 //                    logger.debug("当前server IP:{},类型:{},基线值{}百分比{}未超过静态敏感阈值,日志详情\n{}",destinationIp, attackType, base, percent, value);
                 }else {
                     result = getResult(value, base, profileId, severity, percent+1, type, tag);