增加流量监测和假种检测功能
Change-Id: I808ce14b6f08565f797f4681a6f72db9c730d011
diff --git a/src/main/java/cheat/Cheat.java b/src/main/java/cheat/Cheat.java
index cb35c05..1141c5b 100644
--- a/src/main/java/cheat/Cheat.java
+++ b/src/main/java/cheat/Cheat.java
@@ -4,6 +4,7 @@
import java.util.HashMap;
import java.util.List;
import java.util.Map;
+import java.util.HashMap;
import javax.persistence.EntityManager;
import javax.persistence.EntityManagerFactory;
@@ -20,6 +21,7 @@
import com.querydsl.jpa.impl.JPAQueryFactory;
+import entity.Seed;
import entity.Appeal;
import entity.QAppeal;
import entity.QSeed;
@@ -27,6 +29,8 @@
import entity.config;
import entity.QUser;
import entity.TTorent;
+import entity.QTransRecord;
+import entity.TransRecord;
public class Cheat implements CheatInterfnterface {
@@ -180,50 +184,230 @@
@Override
public void DetectTrans() {
- // JPAQueryFactory queryFactory = new JPAQueryFactory(entityManager);
- // QSeed qSeed = QSeed.seed;
- // List<Seed> seeds = queryFactory
- // .selectFrom(qSeed)
- // .fetch();
- // for (Seed seed : seeds) {
- // QTransRecord qTransRecord = QTransRecord.transRecord;
- // List<TransRecord> tasks = queryFactory
- // .selectFrom(qTransRecord)
- // .where(qTransRecord.seedid.eq(seed.seedid))
- // .fetch();
+ EntityManager entityManager = emf.createEntityManager();
+ try {
+ JPAQueryFactory queryFactory = new JPAQueryFactory(entityManager);
+ QSeed qSeed = QSeed.seed;
+ QTransRecord qTransRecord = QTransRecord.transRecord;
+
+ // 获取所有种子
+ List<Seed> seeds = queryFactory
+ .selectFrom(qSeed)
+ .fetch();
- // int n = tasks.size();
- // if (n == 0) continue;
+ // 存储每个种子的delta值
+ List<Double> allDeltas = new ArrayList<>();
+ Map<String, Double> seedDeltaMap = new HashMap<>();
+
+ // 计算每个种子的delta值（只考虑account_status不为1的用户）
+ for (Seed seed : seeds) {
+ List<TransRecord> tasks = queryFactory
+ .selectFrom(qTransRecord)
+ .leftJoin(qTransRecord.uploader).fetchJoin()
+ .leftJoin(qTransRecord.downloader).fetchJoin()
+ .where(qTransRecord.seedid.eq(seed.seedid))
+ .fetch();
- // double[] xArr = new double[n];
- // for (int i = 0; i < n; i++) {
- // TransRecord t = tasks.get(i);
- // xArr[i] = Math.max(0, t.upload - t.download);
- // }
+ // 过滤掉account_status为1的用户的记录
+ List<TransRecord> validTasks = new ArrayList<>();
+ for (TransRecord task : tasks) {
+ boolean shouldInclude = true;
+
+ // 检查上传者
+ if (task.uploaduserid != null) {
+ User uploader = entityManager.find(User.class, task.uploaduserid);
+ if (uploader != null && uploader.accountstate) {
+ shouldInclude = false;
+ }
+ }
+
+ // 检查下载者
+ if (task.downloaduserid != null) {
+ User downloader = entityManager.find(User.class, task.downloaduserid);
+ if (downloader != null && downloader.accountstate) {
+ shouldInclude = false;
+ }
+ }
+
+ if (shouldInclude) {
+ validTasks.add(task);
+ }
+ }
- // double sum = 0;
- // for (double x : xArr) sum += x;
- // double mu = sum / n;
+ if (validTasks.isEmpty()) continue;
- // double sqSum = 0;
- // for (double x : xArr) sqSum += (x - mu) * (x - mu);
- // double sigma = Math.sqrt(sqSum / n);
+ // 计算该种子的总delta
+ double totalDelta = 0;
+ for (TransRecord task : validTasks) {
+ long upload = task.upload != null ? task.upload : 0L;
+ long download = task.download != null ? task.download : 0L;
+ totalDelta += (upload - download);
+ }
- // for (int i = 0; i < n; i++) {
- // if (Math.abs(xArr[i] - mu) > 3 * sigma) {
- // User user = entityManager.find(User.class, tasks.get(i).downloaduserid);
- // if (user != null) {
- // user.detectedCount++;
- // user.lastDetectedTime = new java.util.Date();
- // entityManager.merge(user);
- // }
- // }
- // }
- // }
+ allDeltas.add(totalDelta);
+ seedDeltaMap.put(seed.seedid, totalDelta);
+ }
+
+ if (allDeltas.size() < 2) return; // 需要至少2个种子才能计算标准差
+
+ // 计算平均值和标准差
+ double sum = 0;
+ for (double delta : allDeltas) {
+ sum += delta;
+ }
+ double mean = sum / allDeltas.size();
+
+ double variance = 0;
+ for (double delta : allDeltas) {
+ variance += Math.pow(delta - mean, 2);
+ }
+ double stdDev = Math.sqrt(variance / allDeltas.size());
+
+ // 检测异常种子并处理参与者
+ for (Seed seed : seeds) {
+ Double seedDelta = seedDeltaMap.get(seed.seedid);
+ if (seedDelta == null) continue;
+
+ // 检查是否偏离1个标准差
+ if (Math.abs(seedDelta - mean) > stdDev) {
+ // 获取该种子的所有传输记录参与者
+ List<TransRecord> suspiciousTasks = queryFactory
+ .selectFrom(qTransRecord)
+ .where(qTransRecord.seedid.eq(seed.seedid))
+ .fetch();
+
+ // 更新所有参与者的detectedCount
+ for (TransRecord task : suspiciousTasks) {
+ // 处理上传者
+ if (task.uploaduserid != null) {
+ User uploader = entityManager.find(User.class, task.uploaduserid);
+ if (uploader != null && !uploader.accountstate) {
+ uploader.detectedCount++;
+ uploader.lastDetectedTime = new java.util.Date();
+ entityManager.merge(uploader);
+ }
+ }
+
+ // 处理下载者
+ if (task.downloaduserid != null) {
+ User downloader = entityManager.find(User.class, task.downloaduserid);
+ if (downloader != null && !downloader.accountstate) {
+ downloader.detectedCount++;
+ downloader.lastDetectedTime = new java.util.Date();
+ entityManager.merge(downloader);
+ }
+ }
+ }
+ }
+ }
+ } finally {
+ if (entityManager != null) {
+ entityManager.close();
+ }
+ }
}
@Override
+ @Transactional
public void DetectFakeSeed(){
+ EntityManager entityManager = emf.createEntityManager();
+ try {
+ String torrentDir = entity.config.TORRENT_STORAGE_DIR;
+ String scriptPath = "./src/main/java/cheat/fakeSeed.bash";
+
+ ProcessBuilder processBuilder = new ProcessBuilder("bash", scriptPath, torrentDir);
+ processBuilder.inheritIO();
+
+ Process process = processBuilder.start();
+ int exitCode = process.waitFor();
+
+ if (exitCode == 0) {
+ System.out.println("FakeSeed detection completed successfully");
+
+ // 处理检测结果
+ String failedDir = torrentDir + "/failed_torrents";
+ java.io.File failedDirFile = new java.io.File(failedDir);
+
+ // 获取失败种子文件列表
+ java.util.Set<String> failedTorrentNames = new java.util.HashSet<>();
+ if (failedDirFile.exists() && failedDirFile.isDirectory()) {
+ java.io.File[] failedFiles = failedDirFile.listFiles((dir, name) -> name.endsWith(".torrent"));
+ if (failedFiles != null) {
+ for (java.io.File file : failedFiles) {
+ failedTorrentNames.add(file.getName());
+ }
+ }
+ }
+
+ JPAQueryFactory queryFactory = new JPAQueryFactory(entityManager);
+ QSeed qSeed = QSeed.seed;
+
+ // 收集假种子的拥有者
+ java.util.Set<String> fakeOwners = new java.util.HashSet<>();
+
+ // 更新数据库中的fake_hits字段
+ List<Seed> allSeeds = queryFactory
+ .selectFrom(qSeed)
+ .where(qSeed.url.isNotNull())
+ .fetch();
+
+ for (Seed seed : allSeeds) {
+ java.io.File seedFile = new java.io.File(seed.url);
+ String fileName = seedFile.getName();
+
+ if (failedTorrentNames.contains(fileName)) {
+ // 失败种子，fake_hits + 1
+ seed.faketime++;
+ seed.lastfakecheck = new java.util.Date();
+ fakeOwners.add(seed.seeduserid);
+ System.out.println("Found fake seed: " + seed.seedid + " (fake_hits: " + seed.faketime + ")");
+ } else {
+ // 正常种子，fake_hits 清零
+ if (seed.faketime > 0) {
+ seed.faketime = 0;
+ seed.lastfakecheck = new java.util.Date();
+ System.out.println("Reset fake_hits for seed: " + seed.seedid);
+ }
+ }
+ entityManager.merge(seed);
+ }
+
+ // 更新所有用户的 fakeDetectedCount 和 fakeLastDetectedTime
+ entity.QUser qUser = entity.QUser.user;
+ List<User> allUsers = queryFactory
+ .selectFrom(qUser)
+ .fetch();
+
+ java.util.Date now = new java.util.Date();
+ for (User user : allUsers) {
+ if (fakeOwners.contains(user.userid)) {
+ // 假种子拥有者，fakeDetectedCount + 1
+ user.fakeDetectedCount++;
+ System.out.println("Increased fake detection count for user: " + user.userid + " (count: " + user.fakeDetectedCount + ")");
+ } else {
+ // 其他用户，fakeDetectedCount 清零
+ if (user.fakeDetectedCount > 0) {
+ user.fakeDetectedCount = 0;
+ System.out.println("Reset fake detection count for user: " + user.userid);
+ }
+ }
+ // 更新所有用户的检测时间
+ user.fakeLastDetectedTime = now;
+ entityManager.merge(user);
+ }
+
+ System.out.println("Fake seed detection and database update completed");
+ } else {
+ System.err.println("FakeSeed detection failed with exit code: " + exitCode);
+ }
+ } catch (Exception e) {
+ System.err.println("Error running FakeSeed detection script: " + e.getMessage());
+ e.printStackTrace();
+ } finally {
+ if (entityManager != null) {
+ entityManager.close();
+ }
+ }
}
@Override
diff --git a/src/main/java/cheat/fakeSeed.bash b/src/main/java/cheat/fakeSeed.bash
new file mode 100644
index 0000000..a2aae52
--- /dev/null
+++ b/src/main/java/cheat/fakeSeed.bash
@@ -0,0 +1,182 @@
+#!/usr/bin/env bash
+#
+# batch_test_torrents.sh — 批量检测 .torrent 文件是否能正确启动下载
+#
+# 依赖：curl, jq
+#
+# 用法：./batch_test_torrents.sh /path/to/torrent_dir
+
+set -euo pipefail
+
+QB_URL="http://127.0.0.1:8080"
+QB_USER="admin"
+QB_PASS="9H6k8VpcM"
+COOKIE_JAR="$(mktemp)"
+TORRENT_DIR="${1:-.}"
+TIMEOUT_SECS=60 # 最长等多久才判超时
+POLL_INTERVAL=2 # 每次轮询间隔
+FAILED_DIR="${TORRENT_DIR}/failed_torrents" # 失败种子存放目录
+
+
+# 登录
+_login() {
+ curl -s -c "$COOKIE_JAR" \
+ -d "username=$QB_USER&password=$QB_PASS" \
+ "$QB_URL/api/v2/auth/login" \
+ | grep -q "Ok." || {
+ echo "❌ 登录失败" >&2
+ exit 1
+ }
+}
+
+# 登出
+_logout() {
+ curl -s -b "$COOKIE_JAR" "$QB_URL/api/v2/auth/logout" >/dev/null
+ rm -f "$COOKIE_JAR"
+}
+
+# 添加 torrent，返回 infoHash
+# $1 = .torrent 文件路径
+_add_torrent() {
+ local file="$1"
+ # 丢弃 “Ok.”，只留下后续 info-hash
+ curl -s -b "$COOKIE_JAR" -X POST \
+ -F "torrents=@${file}" \
+ "$QB_URL/api/v2/torrents/add" >/dev/null
+
+ # 等待 qBittorrent 收到任务
+ sleep 3
+
+ # 取最新添加的那个 torrent（按 added_on 降序，limit=1）
+ local info
+ info=$(curl -s -b "$COOKIE_JAR" \
+ "$QB_URL/api/v2/torrents/info?limit=1&sort=added_on&reverse=true")
+
+ # 检查是否获取到有效的 JSON 响应
+ if ! echo "$info" | jq empty 2>/dev/null; then
+ echo "ERROR: Invalid JSON response from qBittorrent API" >&2
+ echo "Response: $info" >&2
+ return 1
+ fi
+
+ # 检查是否有 torrent 记录
+ local count
+ count=$(echo "$info" | jq 'length')
+ if [[ "$count" == "0" ]]; then
+ echo "ERROR: No torrents found after adding" >&2
+ return 1
+ fi
+
+ # 只输出 hash
+ echo "$info" | jq -r '.[0].hash'
+}
+
+# 删除 torrent，同时删除已下载的文件
+# $1 = infoHash
+_delete_torrent() {
+ local hash="$1"
+ curl -s -b "$COOKIE_JAR" \
+ -G --data-urlencode "hashes=${hash}" \
+ "$QB_URL/api/v2/torrents/delete?deleteFiles=true" >/dev/null
+}
+
+# 等待并检测状态
+# $1 = infoHash
+_wait_for_progress() {
+ local hash="$1"
+ local waited=0
+
+ while (( waited < TIMEOUT_SECS )); do
+ # 获取特定种子信息
+ local info
+ info=$(curl -s -b "$COOKIE_JAR" \
+ -G --data-urlencode "hashes=${hash}" \
+ "$QB_URL/api/v2/torrents/info")
+
+ # 检查 JSON 响应
+ if ! echo "$info" | jq empty 2>/dev/null; then
+ echo "⚠️ ${hash}: Invalid API response, retrying..."
+ sleep $POLL_INTERVAL
+ waited=$(( waited + POLL_INTERVAL ))
+ continue
+ fi
+
+ # 检查是否返回了数据
+ local count
+ count=$(echo "$info" | jq 'length')
+ if [[ "$count" == "0" ]]; then
+ echo "⚠️ ${hash}: Torrent not found, retrying..."
+ sleep $POLL_INTERVAL
+ waited=$(( waited + POLL_INTERVAL ))
+ continue
+ fi
+
+ local state progress
+ state=$(echo "$info" | jq -r '.[0].state // "unknown"')
+ progress=$(echo "$info" | jq -r '.[0].progress // 0')
+
+ # 成功开始下载（progress > 0）
+ if awk "BEGIN {exit !($progress > 0)}"; then
+ local progress_percent
+ progress_percent=$(awk "BEGIN {printf \"%.2f\", $progress * 100}")
+ echo "✅ ${hash}: started downloading (progress=${progress_percent}%)"
+ return 0
+ fi
+
+ # 出错状态
+ if [[ "$state" == "error" ]]; then
+ echo "❌ ${hash}: entered error state"
+ return 1
+ fi
+
+ sleep $POLL_INTERVAL
+ waited=$(( waited + POLL_INTERVAL ))
+ done
+
+ echo "⚠️ ${hash}: no progress after ${TIMEOUT_SECS}s timeout"
+ return 2
+}
+
+main() {
+ if [[ ! -d "$TORRENT_DIR" ]]; then
+ echo "Usage: $0 /path/to/torrent_dir" >&2
+ exit 1
+ fi
+
+ # 创建失败种子目录（如果不存在）
+ mkdir -p "$FAILED_DIR"
+
+ # 清空失败种子目录
+ if [[ -d "$FAILED_DIR" ]]; then
+ rm -f "$FAILED_DIR"/*.torrent 2>/dev/null || true
+ echo "已清空失败种子目录：$FAILED_DIR"
+ fi
+
+ _login
+ echo "开始批量测试目录：$TORRENT_DIR"
+ echo "失败的种子将被复制到：$FAILED_DIR"
+
+ for file in "$TORRENT_DIR"/*.torrent; do
+ [[ -e "$file" ]] || { echo "目录中没有 .torrent 文件"; break; }
+
+ echo "---- 测试 $file ----"
+ hash=$(_add_torrent "$file")
+ echo "添加成功，infoHash=$hash"
+
+ if _wait_for_progress "$hash"; then
+ echo ">>> $file 下载检测通过"
+ else
+ echo ">>> $file 下载检测失败"
+ cp "$file" "$FAILED_DIR/"
+ echo "已将失败种子复制到：$FAILED_DIR/$(basename "$file")"
+ fi
+
+ _delete_torrent "$hash"
+ echo
+ done
+
+ _logout
+ echo "全部完成。失败的种子文件已保存在：$FAILED_DIR"
+}
+
+main "$@"
diff --git a/src/main/java/cheat/faketest.py b/src/main/java/cheat/faketest.py
new file mode 100644
index 0000000..e69de29
--- /dev/null
+++ b/src/main/java/cheat/faketest.py