148 lines
3.5 KiB
Go
148 lines
3.5 KiB
Go
package parsing
|
|
|
|
import (
|
|
"github.com/go-gost/core/chain"
|
|
"github.com/go-gost/core/connector"
|
|
"github.com/go-gost/core/dialer"
|
|
"github.com/go-gost/core/logger"
|
|
"github.com/go-gost/x/config"
|
|
tls_util "github.com/go-gost/x/internal/util/tls"
|
|
"github.com/go-gost/x/metadata"
|
|
"github.com/go-gost/x/registry"
|
|
)
|
|
|
|
func ParseChain(cfg *config.ChainConfig) (chain.Chainer, error) {
|
|
if cfg == nil {
|
|
return nil, nil
|
|
}
|
|
|
|
chainLogger := logger.Default().WithFields(map[string]any{
|
|
"kind": "chain",
|
|
"chain": cfg.Name,
|
|
})
|
|
|
|
c := chain.NewChain(cfg.Name)
|
|
selector := parseSelector(cfg.Selector)
|
|
for _, hop := range cfg.Hops {
|
|
group := &chain.NodeGroup{}
|
|
for _, v := range hop.Nodes {
|
|
nodeLogger := chainLogger.WithFields(map[string]any{
|
|
"kind": "node",
|
|
"connector": v.Connector.Type,
|
|
"dialer": v.Dialer.Type,
|
|
"hop": hop.Name,
|
|
"node": v.Name,
|
|
})
|
|
connectorLogger := nodeLogger.WithFields(map[string]any{
|
|
"kind": "connector",
|
|
})
|
|
|
|
tlsCfg := v.Connector.TLS
|
|
if tlsCfg == nil {
|
|
tlsCfg = &config.TLSConfig{}
|
|
}
|
|
tlsConfig, err := tls_util.LoadClientConfig(
|
|
tlsCfg.CertFile, tlsCfg.KeyFile, tlsCfg.CAFile,
|
|
tlsCfg.Secure, tlsCfg.ServerName)
|
|
if err != nil {
|
|
chainLogger.Error(err)
|
|
return nil, err
|
|
}
|
|
|
|
cr := registry.ConnectorRegistry().Get(v.Connector.Type)(
|
|
connector.AuthOption(parseAuth(v.Connector.Auth)),
|
|
connector.TLSConfigOption(tlsConfig),
|
|
connector.LoggerOption(connectorLogger),
|
|
)
|
|
|
|
if v.Connector.Metadata == nil {
|
|
v.Connector.Metadata = make(map[string]any)
|
|
}
|
|
if err := cr.Init(metadata.NewMetadata(v.Connector.Metadata)); err != nil {
|
|
connectorLogger.Error("init: ", err)
|
|
return nil, err
|
|
}
|
|
|
|
dialerLogger := nodeLogger.WithFields(map[string]any{
|
|
"kind": "dialer",
|
|
})
|
|
|
|
tlsCfg = v.Dialer.TLS
|
|
if tlsCfg == nil {
|
|
tlsCfg = &config.TLSConfig{}
|
|
}
|
|
tlsConfig, err = tls_util.LoadClientConfig(
|
|
tlsCfg.CertFile, tlsCfg.KeyFile, tlsCfg.CAFile,
|
|
tlsCfg.Secure, tlsCfg.ServerName)
|
|
if err != nil {
|
|
chainLogger.Error(err)
|
|
return nil, err
|
|
}
|
|
|
|
d := registry.DialerRegistry().Get(v.Dialer.Type)(
|
|
dialer.AuthOption(parseAuth(v.Dialer.Auth)),
|
|
dialer.TLSConfigOption(tlsConfig),
|
|
dialer.LoggerOption(dialerLogger),
|
|
)
|
|
|
|
if v.Dialer.Metadata == nil {
|
|
v.Dialer.Metadata = make(map[string]any)
|
|
}
|
|
if err := d.Init(metadata.NewMetadata(v.Dialer.Metadata)); err != nil {
|
|
dialerLogger.Error("init: ", err)
|
|
return nil, err
|
|
}
|
|
|
|
if v.Bypass == "" {
|
|
v.Bypass = hop.Bypass
|
|
}
|
|
if v.Resolver == "" {
|
|
v.Resolver = hop.Resolver
|
|
}
|
|
if v.Hosts == "" {
|
|
v.Hosts = hop.Hosts
|
|
}
|
|
if v.Interface == "" {
|
|
v.Interface = hop.Interface
|
|
}
|
|
if v.SockOpts == nil {
|
|
v.SockOpts = hop.SockOpts
|
|
}
|
|
|
|
var sockOpts *chain.SockOpts
|
|
if v.SockOpts != nil {
|
|
sockOpts = &chain.SockOpts{
|
|
Mark: v.SockOpts.Mark,
|
|
}
|
|
}
|
|
|
|
tr := (&chain.Transport{}).
|
|
WithConnector(cr).
|
|
WithDialer(d).
|
|
WithAddr(v.Addr).
|
|
WithInterface(v.Interface).
|
|
WithSockOpts(sockOpts)
|
|
|
|
node := &chain.Node{
|
|
Name: v.Name,
|
|
Addr: v.Addr,
|
|
Bypass: registry.BypassRegistry().Get(v.Bypass),
|
|
Resolver: registry.ResolverRegistry().Get(v.Resolver),
|
|
Hosts: registry.HostsRegistry().Get(v.Hosts),
|
|
Marker: &chain.FailMarker{},
|
|
Transport: tr,
|
|
}
|
|
group.AddNode(node)
|
|
}
|
|
|
|
sel := selector
|
|
if s := parseSelector(hop.Selector); s != nil {
|
|
sel = s
|
|
}
|
|
group.WithSelector(sel)
|
|
c.AddNodeGroup(group)
|
|
}
|
|
|
|
return c, nil
|
|
}
|