Quellcode durchsuchen

check number of socks servers

Darien Raymond vor 8 Jahren
Ursprung
Commit
ac8d882a55
1 geänderte Dateien mit 5 neuen und 3 gelöschten Zeilen
  1. 5 3
      proxy/socks/client.go

+ 5 - 3
proxy/socks/client.go

@@ -27,11 +27,13 @@ func NewClient(ctx context.Context, config *ClientConfig) (*Client, error) {
 	for _, rec := range config.Server {
 		serverList.AddServer(protocol.NewServerSpecFromPB(*rec))
 	}
-	client := &Client{
-		serverPicker: protocol.NewRoundRobinServerPicker(serverList),
+	if serverList.Size() == 0 {
+		return nil, newError("0 target server")
 	}
 
-	return client, nil
+	return &Client{
+		serverPicker: protocol.NewRoundRobinServerPicker(serverList),
+	}, nil
 }
 
 // Process implements proxy.Outbound.Process.