HttpGet support Alt-Svc

README.md

@@ -6,4 +6,7 @@ Agent of Nezha Monitoring
 
 <!--GAMFC_DELIMITER--><a href="https://github.com/naiba" title="naiba">
   <img src="https://avatars.githubusercontent.com/u/29243953?v=4" width="50;" alt="naiba"/>
+</a>
+<a href="https://github.com/zhangnew" title="zhangnew">
+  <img src="https://avatars.githubusercontent.com/u/9146834?v=4" width="50;" alt="zhangnew"/>
 </a><!--GAMFC_DELIMITER_END-->

go.mod

@@ -9,6 +9,7 @@ require (
 	github.com/blang/semver v3.5.1+incompatible
 	github.com/creack/pty v1.1.18
 	github.com/dean2021/goss v0.0.0-20230129073947-df90431348f1
+	github.com/ebi-yade/altsvc-go v0.1.1
 	github.com/go-ping/ping v1.1.0
 	github.com/gorilla/websocket v1.5.0
 	github.com/iamacarpet/go-winpty v1.0.4

go.sum

@@ -70,6 +70,8 @@ github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c
 github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
 github.com/dean2021/goss v0.0.0-20230129073947-df90431348f1 h1:5UiJ324LiCdOF/3w/5IeXrKVjdnwHoalvLG2smb3wi4=
 github.com/dean2021/goss v0.0.0-20230129073947-df90431348f1/go.mod h1:NiLueuVb3hYcdF4ta+2ezcKJh6BEjhrBz9Hts6XJ5Sc=
+github.com/ebi-yade/altsvc-go v0.1.1 h1:HmZDNb5ZOPlkyXhi34LnRckawFCux7yPYw+dtInIixo=
+github.com/ebi-yade/altsvc-go v0.1.1/go.mod h1:K/U20bLcsOVrbTeDhqRjp+e3tgNT5iAqSiQzPoU0/Q0=
 github.com/envoyproxy/go-control-plane v0.9.0/go.mod h1:YTl/9mNaCwkRvm6d1a2C3ymFceY/DCBVvsKhRF0iEA4=
 github.com/envoyproxy/go-control-plane v0.9.1-0.20191026205805-5f8ba28d4473/go.mod h1:YTl/9mNaCwkRvm6d1a2C3ymFceY/DCBVvsKhRF0iEA4=
 github.com/envoyproxy/go-control-plane v0.9.4/go.mod h1:6rpuAdCZL397s3pYoYcLgu1mIlRU8Am5FuJP05cCM98=

main.go

@@ -8,6 +8,7 @@ import (
 	"io"
 	"net"
 	"net/http"
+	"net/url"
 	"os"
 	"os/exec"
 	"path/filepath"
@@ -18,6 +19,7 @@ import (
 	"github.com/AlecAivazis/survey/v2"
 	bpc "github.com/DaRealFreak/cloudflare-bp-go"
 	"github.com/blang/semver"
+	"github.com/ebi-yade/altsvc-go"
 	"github.com/go-ping/ping"
 	"github.com/gorilla/websocket"
 	"github.com/nezhahq/go-github-selfupdate/selfupdate"
@@ -48,6 +50,7 @@ type AgentCliParam struct {
 	ClientSecret          string // 客户端密钥
 	ReportDelay           int    // 报告间隔
 	TLS                   bool   // 是否使用TLS加密传输至服务端
+	Version               bool   // 当前版本号
 }
 
 var (
@@ -155,8 +158,14 @@ func main() {
 	flag.BoolVar(&agentCliParam.DisableAutoUpdate, "disable-auto-update", false, "禁用自动升级")
 	flag.BoolVar(&agentCliParam.DisableForceUpdate, "disable-force-update", false, "禁用强制升级")
 	flag.BoolVar(&agentCliParam.TLS, "tls", false, "启用SSL/TLS加密")
+	flag.BoolVarP(&agentCliParam.Version, "version", "v", false, "查看当前版本号")
 	flag.Parse()
 
+	if agentCliParam.Version {
+		fmt.Println(version)
+		return
+	}
+
 	if isEditAgentConfig {
 		editAgentConfig()
 		return
@@ -399,11 +408,63 @@ func handleHttpGetTask(task *pb.Task, result *pb.TaskResult) {
 			c := resp.TLS.PeerCertificates[0]
 			result.Data = c.Issuer.CommonName + "|" + c.NotAfter.String()
 		}
-		result.Successful = true
+		altSvc := resp.Header.Get("Alt-Svc")
+		parsedUrl, _ := url.Parse(task.Data)
+		originalHost := parsedUrl.Hostname()
+		if checkAltSvc(altSvc, originalHost, result) {
+			result.Successful = true
+		}
+	} else {
+		// HTTP 请求失败
+		result.Data = err.Error()
+	}
+}
+
+func checkAltSvc(altSvcStr string, originalHost string, result *pb.TaskResult) bool {
+	if altSvcStr == "" {
+		return true
+	}
+	altSvcList, _ := altsvc.Parse(altSvcStr)
+	altAuthorityHost := ""
+	altAuthorityPort := ""
+	for _, altSvc := range altSvcList {
+		if altSvc.AltAuthority.Host != "" && altSvc.AltAuthority.Host != originalHost {
+			altAuthorityHost = altSvc.AltAuthority.Host
+			altAuthorityPort = altSvc.AltAuthority.Port
+		}
+	}
+	if altAuthorityHost == "" {
+		return true
+	}
+	altAuthorityUrl := "https://" + altAuthorityHost + ":" + altAuthorityPort + "/"
+	println("checkAltSvc originalHost: ", originalHost, " AltSvc: ", altAuthorityUrl)
+	successful := false
+	req, _ := http.NewRequest("GET", altAuthorityUrl, nil)
+	req.Host = originalHost
+	req.Header.Add("Upgrade", originalHost)
+	req.Header.Add("Connection", "Upgrade")
+
+	start := time.Now()
+	resp, err := httpClient.Do(req)
+	if err == nil {
+		// 检查 HTTP Response 状态
+		result.Delay = float32(time.Since(start).Microseconds()) / 1000.0
+		if resp.StatusCode > 399 || resp.StatusCode < 200 {
+			err = errors.New("\n应用错误：" + resp.Status)
+		}
+	}
+	if err == nil {
+		// 检查 SSL 证书信息
+		if resp.TLS != nil && len(resp.TLS.PeerCertificates) > 0 {
+			c := resp.TLS.PeerCertificates[0]
+			result.Data = c.Issuer.CommonName + "|" + c.NotAfter.String()
+		}
+		successful = true
 	} else {
 		// HTTP 请求失败
 		result.Data = err.Error()
 	}
+	return successful
 }
 
 func handleCommandTask(task *pb.Task, result *pb.TaskResult) {