admin/PrivHunterAI
1
0
Fork 0
mirror of https://github.com/Ed1s0nZ/PrivHunterAI.git synced 2025-09-17 12:31:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update scan.go

Browse Source
This commit is contained in:
公明 2025-02-06 15:37:33 +08:00 committed by GitHub
parent c77e879e3e
commit 3e4b9ccbe4
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 24 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

44
scan.go
View File

@ -68,33 +68,35 @@ func sendHTTPAndKimi(r *RequestResponseLog) {
// 输出响应体字符串
fmt.Println("Response1 Body:", resp1)
fmt.Println("Response2 Body:", resp2)
switch AI {
case "kimi":
result, err := kimi(resp1, resp2) //调用kimi检测是否越权
result, err := detectPrivilegeEscalation(AI, resp1, resp2)
if err != nil {
fmt.Println(err)
}
log.Println("kimi:")
log.Println(result)
case "deepseek":
result, err := deepSeek(resp1, resp2) //调用deepSeek检测是否越权
if err != nil {
fmt.Println(err)
}
log.Println("deepseek:")
log.Println(result)
default:
result, err := kimi(resp1, resp2) //调用kimi检测是否越权
if err != nil {
fmt.Println(err)
}
log.Println("kimi:")
log.Println(result)
fmt.Println("Error:", err)
} else {
log.Println("Result:", result)
}
}
}
func detectPrivilegeEscalation(AI string, resp1, resp2 string) (string, error) {
var result string
var err error
switch AI {
case "kimi":
result, err = kimi(resp1, resp2) // 调用 kimi 检测是否越权
case "deepseek":
result, err = deepSeek(resp1, resp2) // 调用 deepSeek 检测是否越权
default:
result, err = kimi(resp1, resp2) // 默认调用 kimi 检测是否越权
}
if err != nil {
return "", err
}
return result, nil
}
func isNotSuffix(s string, suffixes []string) bool {
for _, suffix := range suffixes {
if strings.HasSuffix(s, suffix) {