init

go-gost/x/dialer/kcp/metadata.go

@@ -0,0 +1,61 @@
+package kcp
+
+import (
+	"encoding/json"
+	"time"
+
+	mdata "github.com/go-gost/core/metadata"
+	mdutil "github.com/go-gost/x/metadata/util"
+	kcp_util "github.com/go-gost/x/internal/util/kcp"
+)
+
+type metadata struct {
+	handshakeTimeout time.Duration
+	config           *kcp_util.Config
+}
+
+func (d *kcpDialer) parseMetadata(md mdata.Metadata) (err error) {
+	const (
+		config           = "config"
+		configFile       = "c"
+		handshakeTimeout = "handshakeTimeout"
+	)
+
+	if file := mdutil.GetString(md, "kcp.configFile", "configFile", "c"); file != "" {
+		d.md.config, err = kcp_util.ParseFromFile(file)
+		if err != nil {
+			return
+		}
+	}
+
+	if m := mdutil.GetStringMap(md, "kcp.config", "config"); len(m) > 0 {
+		b, err := json.Marshal(m)
+		if err != nil {
+			return err
+		}
+		cfg := &kcp_util.Config{}
+		if err := json.Unmarshal(b, cfg); err != nil {
+			return err
+		}
+		d.md.config = cfg
+	}
+	if d.md.config == nil {
+		d.md.config = kcp_util.DefaultConfig
+	}
+	d.md.config.TCP = mdutil.GetBool(md, "kcp.tcp", "tcp")
+	d.md.config.Key = mdutil.GetString(md, "kcp.key")
+	d.md.config.Crypt = mdutil.GetString(md, "kcp.crypt")
+	d.md.config.Mode = mdutil.GetString(md, "kcp.mode")
+	d.md.config.KeepAlive = mdutil.GetInt(md, "kcp.keepalive")
+	d.md.config.Interval = mdutil.GetInt(md, "kcp.interval")
+	d.md.config.MTU = mdutil.GetInt(md, "kcp.mtu")
+	d.md.config.RcvWnd = mdutil.GetInt(md, "kcp.rcvwnd")
+	d.md.config.SndWnd = mdutil.GetInt(md, "kcp.sndwnd")
+	d.md.config.SmuxVer = mdutil.GetInt(md, "kcp.smuxver")
+	d.md.config.SmuxBuf = mdutil.GetInt(md, "kcp.smuxbuf")
+	d.md.config.StreamBuf = mdutil.GetInt(md, "kcp.streambuf")
+	d.md.config.NoComp = mdutil.GetBool(md, "kcp.nocomp")
+
+	d.md.handshakeTimeout = mdutil.GetDuration(md, handshakeTimeout)
+	return
+}