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

Update scan.go

Browse Source
This commit is contained in:
公明 2025-02-27 10:07:35 +08:00 committed by GitHub
parent eccaae7045
commit cc8be2845b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
scan.go
View File

@ -10,6 +10,7 @@ import (
"strings"
"time"
aiapis "yuequanScan/AIAPIS"
"yuequanScan/config"
)
type Result struct {
@ -115,14 +116,14 @@ func sendHTTPAndKimi(r *RequestResponseLog) (result string, respA string, respB
RawQuery: r.Request.URL.RawQuery,
}
if isNotSuffix(r.Request.URL.Path, suffixes) && !containsString(r.Response.Header.Get("Content-Type"), allowedRespHeaders) {
if isNotSuffix(r.Request.URL.Path, config.GetConfig().Suffixes) && !containsString(r.Response.Header.Get("Content-Type"), config.GetConfig().AllowedRespHeaders) {
req, err := http.NewRequest(r.Request.Method, fullURL.String(), strings.NewReader(string(r.Request.Body)))
if err != nil {
fmt.Println("创建请求失败:", err)
return "", "", "", err
}
req.Header = r.Request.Header
req.Header.Set("Cookie", cookie2)
req.Header.Set("Cookie", config.GetConfig().Cookie2)
client := &http.Client{}
resp, err := client.Do(req)
if err != nil {
@ -141,7 +142,7 @@ func sendHTTPAndKimi(r *RequestResponseLog) (result string, respA string, respB
// fmt.Println("Response1 Body:", resp1)
// fmt.Println("Response2 Body:", resp2)
if len(resp1+resp2) < 65535 {
result, err := detectPrivilegeEscalation(AI, fullURL.String(), resp1, resp2)
result, err := detectPrivilegeEscalation(config.GetConfig().AI, fullURL.String(), resp1, resp2)
if err != nil {
fmt.Println("Error:", err)
return "", "", "", err