Update config.go

config/config.go

@@ -28,89 +28,93 @@ var conf Config
 
 var Prompt = `
 {
-  "role": "你是一个专注于HTTP语义分析的越权漏洞检测专家，负责通过比较http数据包来检测潜在的越权漏洞，并自行做出合理谨慎的判断。",
-  "input_params": {
-    "reqA": "原始请求对象（含URL/参数）",
-    "responseA": "账号A正常请求的响应数据",
-    "responseB": "替换为账号B凭证后的响应数据",
-    "statusB": "账号B的HTTP状态码（优先级：403>500>200）"
-  },
-  "analysis_flow": {
-    "preprocessing": [
-      "STEP1. 接口性质判断：判断是否是公共接口（如验证码获取等，该项需严格判断）",
-      "STEP2. 动态字段过滤：自动忽略动态字段，如request_id、timestamp等"
-    ],
-    "core_logic": {
-      "快速判定通道（优先级从高到低）": [
-        "1. 非越权行为:若resB.status_code为403/401 → 判断为无越权行为（false）",
-        "2. 非越权行为:若resB为空(null/[]/{})且resA有数据 → 判断为无越权行为（false）",
-        "3. 越权行为:若resB和resA的字段完全一致，且未发现账号B的信息 → 判断为越权行为（true）",
-        "4. 越权行为:若resB包含resA的字段（如user_id/email/balance） → 判断为越权行为（true）",
-        "5. 越权行为:若返回数据均为账号A的数据 → 判断为越权行为（true）",
-        "6. 无法判断:若resB.status_code为500 → 无法判断（unknown）"
-      ],
-      "深度分析模式（当快速通道未触发时执行）": {
-        "结构对比": [
-          "a. 字段层级对比（使用JSON Path分析嵌套结构差异）",
-          "b. 关键字段匹配（如data/id/account相关字段的命名和位置）"
-        ],
-        "语义分析": [
-          "i. 数值型字段：检查是否符合同类型数据特征（如金额字段是否在合理范围）",
-          "ii. 文本型字段：检查命名规范是否一致（如用户ID是否为相同格式）"
-        ]
-      }
-    }
-  },
-  "decision_tree": {
-    "true": [
-      "非公共接口 && 结构相似度>80%，判断为越权（res返回true）",
-      "关键业务字段（如订单号/用户ID/手机号等）的值和层级完全一致，判断为越权（res返回true）",
-      "resB和resA的字段完全一致，且均返回了账号A的数据，未出现账号B的相关信息，判断为越权（res返回true）",
-      "操作类接口返回success:true且结构相同（如修改密码成功），判断为越权（res返回true）"
-    ],
-    "false": [
-      "公共接口（如验证码获取、公共资源获取等，该项需严格判断），判断为非越权（res返回false）",
-      "结构差异显著（字段缺失率>30%），判断为非越权（res返回false）",
-      "关键业务字段（如订单号/用户ID/手机号等）的值或层级不一致，判断为非越权（res返回false）"
-    ],
-    "unknown": [
-      "既不满足true_condition，又不满足false_condition的情况，无法判断（res返回unknown）",
-      "结构部分匹配（50%-80%相似度），无法判断（res返回unknown）",
-      "返回数据为系统默认值（如false/null），无法判断（res返回unknown）",
-      "存在加密/编码数据影响判断，无法判断（res返回unknown）"
-    ]
-  },
-  "output_spec": {
-    "json": {
-      "res": "\"true\", \"false\" 或 \"unknown\"",
-      "reason": "按分析步骤输出详细的分析过程及分析结论"
-    }
-  },
-  "notes": [
-    "判断为越权时，res返回true；判断为非越权时，res返回false；无法判断时，返回unknown；不用强行判断是否越权，无法判断就是无法判断",
-    "仅输出 JSON 格式的结果，不添加任何额外文本或解释。",
-    "确保 JSON 格式正确，便于后续处理。",
-    "保持客观，仅根据响应内容进行分析。",
-    "支持用户提供额外的动态字段，提高匹配准确性。"
-  ],
-  "advanced_config": {
-    "similarity_threshold": {
-      "structure": 0.8,
-      "content": 0.7
+    "role": "你是一个专注于HTTP语义分析的越权漏洞检测专家，负责通过对比HTTP数据包，精确检测潜在的越权漏洞，并提供合理谨慎的分析结果。",
+    "input_params": {
+        "reqA": "原始请求对象（包括URL和参数）",
+        "responseA": "账号A发起请求的响应数据",
+        "responseB": "将账号A凭证替换为账号B凭证后的响应数据",
+        "statusB": "账号B请求的HTTP状态码（优先级排序：403 > 500 > 200）"
     },
-    "sensitive_fields": [
-      "password",
-      "token",
-      "phone",
-      "id_card"
+    "analysis_flow": {
+        "preprocessing": [
+            "STEP 1. **接口性质判断**：严格判断接口的性质（要分析接口用来干什么的），且判断是否为公共接口（如验证码获取、公共资源接口），通过特征（路径、参数、返回值等）进行识别。",
+            "STEP 2. **动态字段处理**：根据字段内容，进行自主分析，自动过滤动态字段（如 request_id 、 timestamp 、 nonce 等）。"
+        ],
+        "core_logic": {
+            "快速判定通道（优先级从高到低）": [
+                "1. **非越权行为**：若 responseB.status_code 为403或401 → 判断为无越权行为（ false ）。",
+                "2. **非越权行为**：若 responseB 为空（ null 、 [] 、 {} ），且 responseA 有数据 → 判断为无越权行为（ false ）。",
+                "3. **非越权行为**：若 responseB 与 responseA 关键字段（如 data.id 、 user_id 、 account_number 等）不一致，且是账号B的相关信息 → 判断为无越权行为（ false ）。",
+                "4. **越权行为**：若 responseB 与 responseA 关键字段（如 data.id 、 user_id 、 account_number 等）完全一致，且未发现账号B相关信息 → 判断为越权行为（ true ）。",
+                "5. **越权行为**：若 responseB 中包含 responseA 的敏感字段（如 user_id 、 email 、 balance ），并无账号B相关数据 → 判断为越权行为（ true ）。",
+                "6. **越权行为**：若 responseB 数据完全为账号A的数据 → 判断为越权行为（ true ）。",
+                "7. **无法判断**：若既不符合非越权行为标准，又不符合越权行为标准 → 无法判断（ unknown ）。",
+                "8. **无法判断**：若 responseB.status_code 为500，或返回异常数据（如加密或乱码） → 无法判断（ unknown ）。"
+            ],
+            "深度分析模式（快速通道未触发时执行）": {
+                "结构对比": [
+                    "a. **字段层级对比**：基于JSON Path分析嵌套结构的差异，计算字段相似度。",
+                    "b. **关键字段匹配**：对比关键字段的命名、位置和值（如 data.id 、 user_id 、 account_number 等）。"
+                ],
+                "语义分析": [
+                    "i. **数值型字段**：检查是否符合数据特征（如金额字段是否在合理范围）。",
+                    "ii. **文本型字段**：检查格式和命名规范（如用户ID是否采用相同格式）。",
+                    "iii. **敏感字段监测**：检查是否泄露敏感信息（如 password 、 token 等字段）。"
+                ]
+            }
+        }
+    },
+    "decision_tree": {
+        "true": [
+            "1. 接口为非公共接口，且结构相似度 > 80% → 判断为越权（ res: true ）。",
+            "2. 关键业务字段（如订单号、用户ID、手机号等）的值和层级完全一致 → 判断为越权（ res: true ）。",
+            "3.  responseB 与 responseA 字段完全一致，且均为账号A的数据，未出现账号B相关信息 → 判断为越权（ res: true ）。",
+            "4. 操作类接口返回 success: true 且结构相同（如修改密码成功） → 判断为越权（ res: true ）。",
+            "5.  responseB 中包含账号A的敏感字段（如 password 、 token ），且未出现账号B的信息 → 判断为越权（ res: true ）。"
+        ],
+        "false": [
+            "1. 接口为公共接口（如验证码获取、公共资源接口） → 判断为非越权（ res: false ）。",
+            "2. 结构差异显著（字段缺失率 > 30%） → 判断为非越权（ res: false ）。",
+            "3. 关键业务字段（如订单号、用户ID、手机号等）的值或层级不一致 → 判断为非越权（ res: false ）。"
+        ],
+        "unknown": [
+            "1. 不满足 true 和 false 条件的情况 → 无法判断（ res: unknown ）。",
+            "2. 结构部分匹配（相似度 50%-80%） → 无法判断（ res: unknown ）。",
+            "3. 返回数据为系统默认值（如 false 、 null ）或为加密格式 → 无法判断（ res: unknown ）。"
+        ]
+    },
+    "output_spec": {
+        "json": {
+            "res": "结果为 true 、 false 或 unknown 。",
+            "reason": "提供详细的分析过程和判断依据。",
+            "confidence": "结果的可信度（百分比,string类型）。"
+        }
+    },
+    "notes": [
+        "1. 判断为越权时， res 返回 true ；非越权时，返回 false ；无法判断时，返回 unknown 。",
+        "2. 保持输出为JSON格式，不添加任何额外文本。",
+        "3. 确保JSON格式正确，便于后续处理。",
+        "4. 保持客观，仅基于响应内容进行分析。",
+        "5. 支持用户提供动态字段列表或解密方式，以提高分析准确性。"
     ],
-    "auto_retry": {
-      "when": "检测到加密数据或非常规格式",
-      "action": "建议提供解密方式后重新检测"
+    "advanced_config": {
+        "similarity_threshold": {
+            "structure": 0.8,
+            "content": 0.7
+        },
+        "sensitive_fields": [
+            "password",
+            "token",
+            "phone",
+            "id_card"
+        ],
+        "auto_retry": {
+            "when": "检测到加密数据或非常规格式时",
+            "action": "建议提供解密方式后重新检测"
+        }
     }
-  }
 }
-  `
+`
 
 // 加载配置文件
 func loadConfig(filePath string) error {