mirror of
https://github.com/Ed1s0nZ/PrivHunterAI.git
synced 2025-09-17 20:41:37 +00:00
Update scan.go
This commit is contained in:
parent
ed2513648f
commit
618c1edb37
6
scan.go
6
scan.go
@ -199,6 +199,8 @@ func sendHTTPAndKimi(r *RequestResponseLog) (result, reqA, reqB, respA, respB st
|
||||
|
||||
if len(resp1+resp2) < 65535 {
|
||||
if !MatchString(config.GetConfig().RespBodyBWhiteList, resp2) {
|
||||
similarity := StringSimilarity(resp1, resp2)
|
||||
if similarity > 0.5 {
|
||||
// 初始值
|
||||
var resultDetect string
|
||||
var detectErr error
|
||||
@ -219,10 +221,12 @@ func sendHTTPAndKimi(r *RequestResponseLog) (result, reqA, reqB, respA, respB st
|
||||
}
|
||||
|
||||
return resultDetect, req1, req2, resp1, resp2, nil
|
||||
} else {
|
||||
return `{"res": "false", "reason": "相似度小于0.5(` + fmt.Sprint(similarity) + `),判断为未越权(未消耗AI tokens)","confidence":"100%"}`, req1, req2, resp1, resp2, nil
|
||||
}
|
||||
} else {
|
||||
return `{"res": "false", "reason": "匹配到关键字,判断为无越权(未消耗AI tokens)","confidence":"100%"}`, req1, req2, resp1, resp2, nil
|
||||
}
|
||||
|
||||
} else {
|
||||
return `{"res": "white", "reason": "请求包太大","confidence":"100%"}`, req1, req2, resp1, resp2, nil
|
||||
}
|
||||
|
Loading…
x
Reference in New Issue
Block a user