add mws dialer

pkg/dialer/ws/metadata.go

@@ -0,0 +1,86 @@
+package ws
+
+import (
+	"crypto/tls"
+	"fmt"
+	"net"
+	"net/http"
+	"time"
+
+	tls_util "github.com/go-gost/gost/pkg/common/util/tls"
+	md "github.com/go-gost/gost/pkg/metadata"
+)
+
+const (
+	defaultPath = "/ws"
+)
+
+type metadata struct {
+	path      string
+	host      string
+	tlsConfig *tls.Config
+
+	handshakeTimeout  time.Duration
+	readHeaderTimeout time.Duration
+	readBufferSize    int
+	writeBufferSize   int
+	enableCompression bool
+
+	header http.Header
+}
+
+func (d *wsDialer) parseMetadata(md md.Metadata) (err error) {
+	const (
+		path = "path"
+		host = "host"
+
+		certFile   = "certFile"
+		keyFile    = "keyFile"
+		caFile     = "caFile"
+		secure     = "secure"
+		serverName = "serverName"
+
+		handshakeTimeout  = "handshakeTimeout"
+		readHeaderTimeout = "readHeaderTimeout"
+		readBufferSize    = "readBufferSize"
+		writeBufferSize   = "writeBufferSize"
+		enableCompression = "enableCompression"
+
+		header = "header"
+	)
+
+	d.md.path = md.GetString(path)
+	if d.md.path == "" {
+		d.md.path = defaultPath
+	}
+
+	d.md.host = md.GetString(host)
+
+	sn, _, _ := net.SplitHostPort(md.GetString(serverName))
+	if sn == "" {
+		sn = "localhost"
+	}
+	d.md.tlsConfig, err = tls_util.LoadClientConfig(
+		md.GetString(certFile),
+		md.GetString(keyFile),
+		md.GetString(caFile),
+		md.GetBool(secure),
+		sn,
+	)
+
+	d.md.handshakeTimeout = md.GetDuration(handshakeTimeout)
+	d.md.readHeaderTimeout = md.GetDuration(readHeaderTimeout)
+	d.md.readBufferSize = md.GetInt(readBufferSize)
+	d.md.writeBufferSize = md.GetInt(writeBufferSize)
+	d.md.enableCompression = md.GetBool(enableCompression)
+
+	if mm, _ := md.Get(header).(map[interface{}]interface{}); len(mm) > 0 {
+		h := http.Header{}
+		for k, v := range mm {
+			h.Add(fmt.Sprintf("%v", k), fmt.Sprintf("%v", v))
+		}
+		d.md.header = h
+	}
+
+	return
+}