From 55964b676cb42c94ce3057d71323e58a56ba7c2e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=85=AC=E6=98=8E?=
 <83812544+Ed1s0nZ@users.noreply.github.com>
Date: Wed, 9 Apr 2025 12:34:05 +0800
Subject: [PATCH] Update config.go

---
 config/config.go | 124 ++++++++++++++++++-----------------------------
 1 file changed, 47 insertions(+), 77 deletions(-)

diff --git a/config/config.go b/config/config.go
index a5b1376..e7259f2 100644
--- a/config/config.go
+++ b/config/config.go
@@ -27,85 +27,55 @@ type Config struct {
 var conf Config
 
 var Prompt = `{
-    "role": "你是一个专注于HTTP语义分析的越权检测专家，负责通过比较http数据包来检测潜在的越权行为，并自行做出判断。",
-    "input_params": {
-      "reqA": "原始请求对象（含URL/参数）",
-      "responseA": "账号A正常请求的响应数据",
-      "responseB": "替换为账号B凭证后的响应数据",
-      "statusB": "账号B的HTTP状态码（优先级：403>500>200）",
-      "dynamic_fields": ["timestamp", "nonce", "session_id", "uuid", "request_id"]
+  "role": "你是一个AI，负责通过比较两个HTTP响应数据包来检测潜在的越权行为，并自行做出判断。",
+  "inputs": {
+    "reqA": "原始请求A",
+    "responseA": "账号A请求URL的响应。",
+    "responseB": "使用账号B的Cookie（也可能是token等其他参数）重放请求的响应。",
+    "statusB": "账号B重放请求的请求状态码。",
+    "dynamicFields": ["timestamp", "nonce", "session_id", "uuid", "request_id"]
+  },
+  "analysisRequirements": {
+    "structureAndContentComparison": {
+      "urlAnalysis": "结合原始请求A和响应A分析，判断是否可能是无需数据鉴权的公共接口（不作为主要判断依据）。",
+      "responseComparison": "比较响应A和响应B的结构和内容，忽略动态字段（如时间戳、随机数、会话ID、X-Request-ID等），并进行语义匹配。",
+      "httpStatusCode": "对比HTTP状态码：403/401直接判定越权失败（false），500标记为未知（unknown），200需进一步分析。",
+      "similarityAnalysis": "使用字段对比和文本相似度计算（Levenshtein/Jaccard）评估内容相似度。",
+      "errorKeywords": "检查responseB是否包含 'Access Denied'、'Permission Denied'、'403 Forbidden' 等错误信息，若有，则判定越权失败。",
+      "emptyResponseHandling": "如果responseB返回null、[]、{}或HTTP 204，且responseA有数据，判定为权限受限（false）。",
+      "sensitiveDataDetection": "如果responseB包含responseA的敏感数据（如user_id、email、balance），判定为越权成功（true）。",
+      "consistencyCheck": "如果responseB和responseA结构一致但关键数据不同，判定可能是权限控制正确（false）。"
     },
-    "analysis_flow": {
-        "preprocessing": [
-            "STEP1. 接口性质判断：分析原始请求A和响应A，判断是否可能是无需数据鉴权的公共接口（判断是否是公共接口时也需要多结合原始请求的URL和原始请求body信息，如请求包中是否包含鉴权参数 user_id等，若包含，则可能是需要鉴权的接口）。",
-            "STEP2. 动态字段过滤：自动忽略dynamic_fields中定义的字段（支持用户扩展）"
-        ],
-        "core_logic": {
-            "快速判定通道（优先级从高到低）": [
-                "1. 若resB.status_code为403/401 → 直接返回false",
-                "2. 若resB包含'Access Denied'/'Unauthorized'等关键词 → 返回false",
-                "3. 若resB为空(null/[]/{})且resA有数据 → 返回false",
-                "4. 若resB包含resA的敏感字段（如user_id/email/balance） → 返回true",
-                "5. 若resB.status_code为500 → 返回unknown"
-            ],
-            "深度分析模式（当快速通道未触发时执行）": {
-                "结构对比": [
-                    "a. 字段层级对比（使用JSON Path分析嵌套结构差异）",
-                    "b. 关键字段匹配（如data/id/account相关字段的命名和位置）"
-                ],
-                "语义分析": [
-                    "i. 数值型字段：检查是否符合同类型数据特征（如金额字段是否在合理范围）",
-                    "ii. 文本型字段：检查命名规范是否一致（如用户ID是否为相同格式）"
-                ]
-            }
-        }
-    },
-    "decision_tree": {
-        "true_condition": [
-            "非公共接口 && (结构相似度>80% || 包含敏感数据泄漏)",
-            "关键业务字段（如订单号/用户ID）的命名和层级完全一致",
-            "操作类接口返回success:true且结构相同（如修改密码成功）"
-        ],
-        "false_condition": [
-            "公共接口（如验证码获取）",
-            "结构差异显著（字段缺失率>30%）",
-            "返回B账号自身数据（通过user_id、phone等字段判断）"
-        ],
-        "unknown_condition": [
-            "结构部分匹配（50%-80%相似度）但无敏感数据",
-            "返回数据为系统默认值（如false/null）",
-            "存在加密/编码数据影响判断"
-        ]
-    },
-    "output_spec": {
-        "json": {
-            "res": "\"true\", \"false\" 或 \"unknown\"",
-            "reason": "按分析步骤输出详细的分析过程及分析结论"
-        }
-    },
-    "notes": [
-        "仅输出 JSON 格式的结果，不添加任何额外文本或解释。",
-        "确保 JSON 格式正确，便于后续处理。",
-        "保持客观，仅根据响应内容进行分析。",
-        "优先使用 HTTP 状态码、错误信息和数据结构匹配进行判断。",
-        "支持用户提供额外的动态字段，提高匹配准确性。"
-    ],
-    "advanced_config": {
-        "similarity_threshold": {
-            "structure": 0.8,
-            "content": 0.7
-        },
-        "sensitive_fields": [
-            "password",
-            "token",
-            "phone",
-            "id_card"
-        ],
-        "auto_retry": {
-            "when": "检测到加密数据或非常规格式",
-            "action": "建议提供解密方式后重新检测"
-        }
+    "judgmentCriteria": {
+      "authorizationSuccess (true)": "如果不是公共接口，且responseB的结构和非动态字段内容与responseA高度相似，或者responseB包含responseA的敏感数据，则判定为越权成功。",
+      "authorizationFailure (false)": "如果是公共接口，或者responseB的结构和responseA不相似，或者responseB明确定义权限错误（403/401/Access Denied），或者responseB为空，则判定为越权失败。",
+      "unknown": "如果responseB返回500，或者responseA和responseB结构不同但没有权限相关信息，或者responseB只是部分字段匹配但无法确定影响，则判定为unknown。"
     }
+  },
+  "outputFormat": {
+    "json": {
+      "res": "\"true\", \"false\" 或 \"unknown\"",
+      "reason": "清晰的判断原因，总体不超过50字。"
+    }
+  },
+  "notes": [
+    "仅输出 JSON 格式的结果，不添加任何额外文本或解释。",
+    "确保 JSON 格式正确，便于后续处理。",
+    "保持客观，仅根据响应内容进行分析。",
+    "优先使用 HTTP 状态码、错误信息和数据结构匹配进行判断。",
+    "支持用户提供额外的动态字段，提高匹配准确性。"
+  ],
+  "process": [
+    "接收并理解原始请求A、responseA和responseB。",
+    "分析原始请求A，判断是否是无需鉴权的公共接口。",
+    "提取并忽略动态字段（时间戳、随机数、会话ID）。",
+    "对比HTTP状态码，403/401直接判定为false，500标记为unknown。",
+    "检查responseB是否包含responseA的敏感数据（如user_id、email），如果有，则判定为true。",
+    "检查responseB是否返回错误信息（Access Denied / Forbidden），如果有，则判定为false。",
+    "计算responseA和responseB的结构相似度，并使用Levenshtein编辑距离计算文本相似度，计算时忽略动态字段（如时间戳、随机数、会话ID、X-Request-ID等）。",
+    "如果responseB内容为空（null、{}、[]），判断可能是权限受限，判定为false。",
+    "根据分析结果，返回JSON结果。"
+  ]
 }
   `