From 661e4b2d3e8f89b5a32116af005550fa25471724 Mon Sep 17 00:00:00 2001
From: catbestme <bestme@me.com>
Date: Tue, 2 Jul 2024 19:22:37 +0800
Subject: [PATCH]  enable use cdn edge ip to accelerate cdn transport when wss

---
 cmd/gost/route.go |  1 +
 ws.go             | 12 ++++++++++--
 2 files changed, 11 insertions(+), 2 deletions(-)

diff --git a/cmd/gost/route.go b/cmd/gost/route.go
index 360bc2d..4b3a47c 100644
--- a/cmd/gost/route.go
+++ b/cmd/gost/route.go
@@ -170,6 +170,7 @@ func parseChainNode(ns string) (nodes []gost.Node, err error) {
 	wsOpts.WriteBufferSize = node.GetInt("wbuf")
 	wsOpts.UserAgent = node.Get("agent")
 	wsOpts.Path = node.Get("path")
+	wsOpts.Host = node.Get("host")
 
 	timeout := node.GetDuration("timeout")
 
diff --git a/ws.go b/ws.go
index 9dc8f0d..322adf5 100644
--- a/ws.go
+++ b/ws.go
@@ -5,15 +5,15 @@ import (
 	"crypto/sha1"
 	"crypto/tls"
 	"encoding/base64"
+	"fmt"
 	"io"
 	"net"
 	"net/http"
 	"net/http/httputil"
+	"net/url"
 	"sync"
 	"time"
 
-	"net/url"
-
 	"github.com/go-log/log"
 	"github.com/gorilla/websocket"
 	smux "github.com/xtaci/smux"
@@ -31,6 +31,7 @@ type WSOptions struct {
 	EnableCompression bool
 	UserAgent         string
 	Path              string
+	Host              string
 }
 
 type wsTransporter struct {
@@ -749,6 +750,13 @@ func websocketClientConn(url string, conn net.Conn, tlsConfig *tls.Config, optio
 	if options.UserAgent != "" {
 		header.Set("User-Agent", options.UserAgent)
 	}
+	if options.Host != "" {
+		fmt.Println("获取到命令行传参host==>", options.Host)
+		dialer.TLSClientConfig.ServerName = options.Host
+		header.Set("Host", options.Host)
+	}
+
+	fmt.Printf("TLSClientConfig==>%+v\n", dialer.TLSClientConfig)
 	c, resp, err := dialer.Dial(url, header)
 	if err != nil {
 		return nil, err