Merge pull request #122 from techotaku/service-url-improvement

Improve HTTPS certificate checking logic.

Co-authored-by: techotaku <1948179+techotaku@users.noreply.github.com>

cmd/agent/main.go

@@ -9,9 +9,9 @@ import (
 	"log"
 	"net"
 	"net/http"
+	"net/url"
 	"os"
 	"os/exec"
-	"strings"
 	"time"
 
 	"github.com/blang/semver"
@@ -186,8 +186,10 @@ func doTask(task *pb.Task) {
 		}
 		if err == nil {
 			// 检查 SSL 证书信息
-			if strings.HasPrefix(task.GetData(), "https://") {
+			serviceUrl, err := url.Parse(task.GetData())
-				c := cert.NewCert(task.GetData()[8:])
+			if err == nil {
+				if serviceUrl.Scheme == "https" {
+					c := cert.NewCert(serviceUrl.Host)
 					if c.Error != "" {
 						result.Data = "SSL证书错误：" + c.Error
 					} else {
@@ -197,6 +199,9 @@ func doTask(task *pb.Task) {
 				} else {
 					result.Successful = true
 				}
+			} else {
+				result.Data = "URL解析错误：" + err.Error()
+			}
 		} else {
 			// HTTP 请求失败
 			result.Data = err.Error()