Fix: SignServer TCP ping for custom port (#2353)

Mrs4s · Aug 23, 2023 · d85d697 · d85d697
1 parent da9f03f
commit d85d697
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/cmd/gocq/login.go b/cmd/gocq/login.go
@@ -7,6 +7,7 @@ import (
  "fmt"
  "image"
  "image/png"
+ "net"
  "net/http"
  "net/url"
  "os"
@@ -541,7 +542,18 @@ func signWaitServer() bool {
  log.Warnf("连接到签名服务器出现错误: %v", err)
  continue
  }
- r := utils.RunTCPPingLoop(u.Host, 4)
+ host := u.Hostname()
+ port := u.Port()
+ if port == "" {
+ switch u.Scheme {
+ case "https":
+ port = "443"
+ case "http":
+ port = "80"
+ }
+ }
+ hostPort := net.JoinHostPort(host, port)
+ r := utils.RunTCPPingLoop(hostPort, 4)
  if r.PacketsLoss > 0 {
  log.Warnf("连接到签名服务器出现错误: 丢包%d/%d 时延%dms", r.PacketsLoss, r.PacketsSent, r.AvgTimeMill)
  continue