mirror of
https://github.com/Sagit-chu/flvx.git
synced 2026-10-02 17:06:36 +08:00
sagitchu
102 lines
2.5 KiB
Go
102 lines
2.5 KiB
Go
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.Clone()
|
|
} else {
|
|
d.md.config = d.md.config.Clone()
|
|
}
|
|
|
|
if md != nil {
|
|
if md.IsExists("kcp.tcp") {
|
|
d.md.config.TCP = mdutil.GetBool(md, "kcp.tcp")
|
|
}
|
|
if md.IsExists("kcp.key") {
|
|
d.md.config.Key = mdutil.GetString(md, "kcp.key")
|
|
}
|
|
if md.IsExists("kcp.crypt") {
|
|
d.md.config.Crypt = mdutil.GetString(md, "kcp.crypt")
|
|
}
|
|
if md.IsExists("kcp.mode") {
|
|
d.md.config.Mode = mdutil.GetString(md, "kcp.mode")
|
|
}
|
|
if md.IsExists("kcp.keepalive") {
|
|
d.md.config.KeepAlive = mdutil.GetInt(md, "kcp.keepalive")
|
|
}
|
|
if md.IsExists("kcp.interval") {
|
|
d.md.config.Interval = mdutil.GetInt(md, "kcp.interval")
|
|
}
|
|
if md.IsExists("kcp.mtu") {
|
|
d.md.config.MTU = mdutil.GetInt(md, "kcp.mtu")
|
|
}
|
|
if md.IsExists("kcp.rcvwnd") {
|
|
d.md.config.RcvWnd = mdutil.GetInt(md, "kcp.rcvwnd")
|
|
}
|
|
if md.IsExists("kcp.sndwnd") {
|
|
d.md.config.SndWnd = mdutil.GetInt(md, "kcp.sndwnd")
|
|
}
|
|
if md.IsExists("kcp.sockbuf") {
|
|
d.md.config.SockBuf = mdutil.GetInt(md, "kcp.sockbuf")
|
|
}
|
|
if md.IsExists("kcp.smuxver") {
|
|
d.md.config.SmuxVer = mdutil.GetInt(md, "kcp.smuxver")
|
|
}
|
|
if md.IsExists("kcp.smuxbuf") {
|
|
d.md.config.SmuxBuf = mdutil.GetInt(md, "kcp.smuxbuf")
|
|
}
|
|
if md.IsExists("kcp.streambuf") {
|
|
d.md.config.StreamBuf = mdutil.GetInt(md, "kcp.streambuf")
|
|
}
|
|
if md.IsExists("kcp.nocomp") {
|
|
d.md.config.NoComp = mdutil.GetBool(md, "kcp.nocomp")
|
|
}
|
|
if md.IsExists("kcp.datashard") {
|
|
d.md.config.DataShard = mdutil.GetInt(md, "kcp.datashard")
|
|
}
|
|
if md.IsExists("kcp.parityshard") {
|
|
d.md.config.ParityShard = mdutil.GetInt(md, "kcp.parityshard")
|
|
}
|
|
}
|
|
|
|
d.md.handshakeTimeout = mdutil.GetDuration(md, handshakeTimeout)
|
|
return
|
|
}
|