改进判定

PBH-BTN · Nov 11, 2024 · f90ea1d · f90ea1d
1 parent 3d4d487
commit f90ea1d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/src/main/java/com/ghostchu/btn/sparkle/module/analyse/AnalyseService.java b/src/main/java/com/ghostchu/btn/sparkle/module/analyse/AnalyseService.java
@@ -52,6 +52,8 @@ public class AnalyseService {
     private long highRiskIpv6IdentityOffset;
     @Value("${analyse.ipv6.prefix-length}")
     private int ipv6ConvertToPrefixLength;
+    @Value("${analyse.highriskips.traffic-from-peer-less-than}")
+    private long trafficFromPeerLessThan;
     @Autowired
     private AnalysedRuleRepository analysedRuleRepository;
     @PersistenceContext
@@ -110,18 +112,21 @@ public void cronHighRiskIps() {
         list.addAll(banHistoryRepository
                 .findDistinctByPeerClientNameAndModuleLikeAndInsertTimeBetween("Transmission 2.93", "%ProgressCheatBlocker%", pastTimestamp(highRiskIpsOffset), nowTimestamp())
                 .stream()
+                .filter(banHistory -> banHistory.getFromPeerTraffic() != -1 && banHistory.getFromPeerTraffic() < trafficFromPeerLessThan)
                 .map(ban -> IPUtil.toIPAddress(ban.getPeerIp().getHostAddress()))
                 .distinct()
                 .toList());
         list.addAll(banHistoryRepository
                 .findDistinctByPeerClientNameAndModuleLikeAndInsertTimeBetween("Transmission 3.00", "%ProgressCheatBlocker%", pastTimestamp(highRiskIpsOffset), nowTimestamp())
                 .stream()
+                .filter(banHistory -> banHistory.getFromPeerTraffic() != -1 && banHistory.getFromPeerTraffic() < trafficFromPeerLessThan)
                 .map(ban -> IPUtil.toIPAddress(ban.getPeerIp().getHostAddress()))
                 .distinct()
                 .toList());
         list.addAll(banHistoryRepository
                 .findDistinctByPeerClientNameAndModuleLikeAndInsertTimeBetween("aria2/%", "%ProgressCheatBlocker%", pastTimestamp(highRiskIpsOffset), nowTimestamp())
                 .stream()
+                .filter(banHistory -> banHistory.getFromPeerTraffic() != -1 && banHistory.getFromPeerTraffic() < trafficFromPeerLessThan)
                 .map(ban -> IPUtil.toIPAddress(ban.getPeerIp().getHostAddress()))
                 .distinct()
                 .toList());

diff --git a/src/main/resources/application.properties b/src/main/resources/application.properties
@@ -113,5 +113,6 @@ analyse.overdownload.offset=3888000000
 analyse.overdownload.threshold=2.5
 analyse.highriskips.interval=14400000
 analyse.highriskips.offset=3888000000
+analyse.highriskips.traffic-from-peer-less-than=50000000
 analyse.highriskipv6identity.interval=14400000
 analyse.highriskipv6identity.offset=3888000000