87 lines
1.7 KiB
Go
87 lines
1.7 KiB
Go
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
|
|
}
|