admin/PrivHunterAI
1
0
Fork 0
mirror of https://github.com/Ed1s0nZ/PrivHunterAI.git synced 2025-09-18 04:51:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update scan.go

Browse Source
This commit is contained in:
公明 2025-03-21 10:57:01 +08:00 committed by GitHub
parent 015653ba15
commit 7656705bc6
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 21 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
scan.go
View File

@ -11,6 +11,8 @@ import (
"time" "time"
aiapis "yuequanScan/AIAPIS" aiapis "yuequanScan/AIAPIS"
"yuequanScan/config" "yuequanScan/config"
"github.com/lqqyt2423/go-mitmproxy/proxy"
) )
type Result struct { type Result struct {
@ -93,7 +95,7 @@ func scan() {
} }
//--- //---
fmt.Println(PrintYuequan(resultOutput.Result, resultOutput.Method, resultOutput.Host+resultOutput.Path, resultOutput.Reason)) log.Println(PrintYuequan(resultOutput.Result, resultOutput.Method, resultOutput.Host+resultOutput.Path, resultOutput.Reason))
logs.Delete(key) logs.Delete(key)
return true // 返回true继续遍历，返回false停止遍历 return true // 返回true继续遍历，返回false停止遍历
} }
@ -108,6 +110,7 @@ func scan() {
} }
func sendHTTPAndKimi(r *RequestResponseLog) (result string, respA string, respB string, err error) { func sendHTTPAndKimi(r *RequestResponseLog) (result string, respA string, respB string, err error) {
jsonDataReq, err := json.Marshal(r.Request) jsonDataReq, err := json.Marshal(r.Request)
if err != nil { if err != nil {
fmt.Println("Error marshaling:", err) fmt.Println("Error marshaling:", err)
@ -124,6 +127,7 @@ func sendHTTPAndKimi(r *RequestResponseLog) (result string, respA string, respB
RawQuery: r.Request.URL.RawQuery, RawQuery: r.Request.URL.RawQuery,
} }
if isNotSuffix(r.Request.URL.Path, config.GetConfig().Suffixes) && !containsString(r.Response.Header.Get("Content-Type"), config.GetConfig().AllowedRespHeaders) { if isNotSuffix(r.Request.URL.Path, config.GetConfig().Suffixes) && !containsString(r.Response.Header.Get("Content-Type"), config.GetConfig().AllowedRespHeaders) {
log.Println("Req1:", req1)
req, err := http.NewRequest(r.Request.Method, fullURL.String(), strings.NewReader(string(r.Request.Body))) req, err := http.NewRequest(r.Request.Method, fullURL.String(), strings.NewReader(string(r.Request.Body)))
if err != nil { if err != nil {
fmt.Println("创建请求失败:", err) fmt.Println("创建请求失败:", err)
@ -139,6 +143,22 @@ func sendHTTPAndKimi(r *RequestResponseLog) (result string, respA string, respB
// 2025 02 27 end // 2025 02 27 end
// req.Header.Set("Cookie", config.GetConfig().Cookie2) // req.Header.Set("Cookie", config.GetConfig().Cookie2)
// log.Println(req.Header) // log.Println(req.Header)
requestInfo2 := proxy.Request{
Method: req.Method,
URL: req.URL,
Proto: req.Proto,
Header: req.Header,
Body: r.Request.Body,
}
jsonDataReq2, err := json.Marshal(requestInfo2)
if err != nil {
fmt.Println("Error marshaling:", err)
return "", "", "", err // 返回错误
}
req2 := string(jsonDataReq2)
log.Println("Req2:", req2)
client := &http.Client{} client := &http.Client{}
resp, err := client.Do(req) resp, err := client.Do(req)
if err != nil { if err != nil {