KJZH001/nezhahq-agent
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix checkAltSvc check condition logic error

Browse Source
This commit is contained in:
naiba 2023-10-17 08:50:23 +08:00
parent 2b565e8f12
commit b885bd8343
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
main.go
View File

@ -415,7 +415,7 @@ func checkHttpResp(taskUrl string, start time.Time, resp *http.Response, err err
} }
altSvc := resp.Header.Get("Alt-Svc") altSvc := resp.Header.Get("Alt-Svc")
if altSvc != "" { if altSvc != "" {
checkAltSvc(altSvc, taskUrl, result) checkAltSvc(start, altSvc, taskUrl, result)
} else { } else {
result.Successful = true result.Successful = true
} }
@ -425,7 +425,7 @@ func checkHttpResp(taskUrl string, start time.Time, resp *http.Response, err err
} }
} }
func checkAltSvc(altSvcStr string, taskUrl string, result *pb.TaskResult) { func checkAltSvc(start time.Time, altSvcStr string, taskUrl string, result *pb.TaskResult) {
altSvcList, err := altsvc.Parse(altSvcStr) altSvcList, err := altsvc.Parse(altSvcStr)
if err != nil { if err != nil {
result.Data = err.Error() result.Data = err.Error()
@ -467,9 +467,9 @@ func checkAltSvc(altSvcStr string, taskUrl string, result *pb.TaskResult) {
req.Header.Add("Upgrade", originalHost) req.Header.Add("Upgrade", originalHost)
req.Header.Add("Connection", "Upgrade") req.Header.Add("Connection", "Upgrade")
start := time.Now()
resp, err := httpClient.Do(req) resp, err := httpClient.Do(req)
checkHttpResp(taskUrl, start, resp, err, result)
checkHttpResp(altAuthorityUrl, start, resp, err, result)
} }
func handleCommandTask(task *pb.Task, result *pb.TaskResult) { func handleCommandTask(task *pb.Task, result *pb.TaskResult) {