fix nil deref

nicocha30 · Aug 7, 2024 · f31ef9c · f31ef9c
1 parent ced0e48
commit f31ef9c
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/cmd/agent/main.go b/cmd/agent/main.go
@@ -99,9 +99,11 @@ func main() {
 	}
 
 	serverUrl, err := url.Parse(*serverAddr)
-	if serverUrl.Scheme == "https" && err == nil {
-		//websocket https connection
-		tlsConfig.ServerName = serverUrl.Hostname()
+	if err != nil && serverUrl != nil {
+		if serverUrl.Scheme == "https" {
+			//websocket https connection
+			tlsConfig.ServerName = serverUrl.Hostname()
+		}
 	} else {
 		//direct connection. try to parse as host:port
 		host, _, err := net.SplitHostPort(*serverAddr)
@@ -120,7 +122,7 @@ func main() {
 
 	for {
 		var err error
-		if serverUrl.Scheme == "https" {
+		if serverUrl != nil && serverUrl.Scheme == "https" {
 			*serverAddr = strings.Replace(*serverAddr, "https://", "wss://", 1)
 			//websocket
 			err = wsconnect(&tlsConfig, *serverAddr, *socksProxy, *userAgent)