go-gost/gost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix metadata

Browse Source
This commit is contained in:
ginuerzh 2022-01-09 00:07:26 +08:00
parent f45dc93e92
commit 37e77bdce5
12 changed files with 118 additions and 159 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

72
cmd/gost/cmd.go
View File

@ -8,6 +8,7 @@ import (
"strings"
"github.com/go-gost/gost/pkg/config"
"github.com/go-gost/gost/pkg/metadata"
"github.com/go-gost/gost/pkg/registry"
)
@ -67,8 +68,12 @@ func buildConfigFromCmd(services, nodes stringList) (*config.Config, error) {
}
service.Name = fmt.Sprintf("service-%d", i)
if chain != nil {
if service.Listener.Type == "rtcp" || service.Listener.Type == "rudp" {
service.Listener.Chain = chain.Name
} else {
service.Handler.Chain = chain.Name
}
}
cfg.Services = append(cfg.Services, service)
}
@ -125,37 +130,30 @@ func buildServiceConfig(url *url.URL) (*config.ServiceConfig, error) {
auths = append(auths, auth)
}
md := make(map[string]interface{})
md := metadata.MapMetadata{}
for k, v := range url.Query() {
if len(v) > 0 {
md[k] = v[0]
}
}
if sauth := md["auth"]; sauth != nil {
if sa, _ := sauth.(string); sa != "" {
if sa := metadata.GetString(md, "auth"); sa != "" {
au, err := parseAuthFromCmd(sa)
if err != nil {
return nil, err
}
auths = append(auths, au)
}
}
delete(md, "auth")
md.Del("auth")
var tlsConfig *config.TLSConfig
if certs := md["cert"]; certs != nil {
cert, _ := certs.(string)
key, _ := md["key"].(string)
ca, _ := md["ca"].(string)
tlsConfig = &config.TLSConfig{
Cert: cert,
Key: key,
CA: ca,
tlsConfig := &config.TLSConfig{
Cert: metadata.GetString(md, "cert"),
Key: metadata.GetString(md, "key"),
CA: metadata.GetString(md, "ca"),
}
}
delete(md, "cert")
delete(md, "key")
delete(md, "ca")
md.Del("cert")
md.Del("key")
md.Del("ca")
svc.Handler = &config.HandlerConfig{
Type: handler,
@ -205,45 +203,33 @@ func buildNodeConfig(url *url.URL) (*config.NodeConfig, error) {
auth.Password, _ = url.User.Password()
}
md := make(map[string]interface{})
md := metadata.MapMetadata{}
for k, v := range url.Query() {
if len(v) > 0 {
md[k] = v[0]
}
}
md["serverName"] = url.Host
if sauth := md["auth"]; sauth != nil && auth == nil {
if sa, _ := sauth.(string); sa != "" {
au, err := parseAuthFromCmd(sa)
if sauth := metadata.GetString(md, "auth"); sauth != "" && auth == nil {
au, err := parseAuthFromCmd(sauth)
if err != nil {
return nil, err
}
auth = au
}
}
delete(md, "auth")
md.Del("auth")
var tlsConfig *config.TLSConfig
if certs := md["cert"]; certs != nil {
cert, _ := certs.(string)
key, _ := md["key"].(string)
ca, _ := md["ca"].(string)
secure, _ := md["secure"].(bool)
serverName, _ := md["serverName"].(string)
tlsConfig = &config.TLSConfig{
Cert: cert,
Key: key,
CA: ca,
Secure: secure,
ServerName: serverName,
tlsConfig := &config.TLSConfig{
CA: metadata.GetString(md, "ca"),
Secure: metadata.GetBool(md, "secure"),
ServerName: metadata.GetString(md, "serverName"),
}
if tlsConfig.ServerName == "" {
tlsConfig.ServerName = url.Hostname()
}
delete(md, "cert")
delete(md, "key")
delete(md, "ca")
delete(md, "secure")
delete(md, "serverName")
md.Del("ca")
md.Del("secure")
md.Del("serverName")
node.Connector = &config.ConnectorConfig{
Type: connector,

19
cmd/gost/config.go
View File

@ -56,8 +56,15 @@ func buildService(cfg *config.Config) (services []*service.Service) {
}
for _, svc := range cfg.Services {
if svc.Listener == nil || svc.Handler == nil {
continue
if svc.Listener == nil {
svc.Listener = &config.ListenerConfig{
Type: "tcp",
}
}
if svc.Handler == nil {
svc.Handler = &config.HandlerConfig{
Type: "auto",
}
}
serviceLogger := log.WithFields(map[string]interface{}{
"kind": "service",
@ -89,10 +96,6 @@ func buildService(cfg *config.Config) (services []*service.Service) {
listener.LoggerOption(listenerLogger),
)
if chainable, ok := ln.(chain.Chainable); ok {
chainable.WithChain(chains[svc.Listener.Chain])
}
if svc.Listener.Metadata == nil {
svc.Listener.Metadata = make(map[string]interface{})
}
@ -292,9 +295,9 @@ func logFromConfig(cfg *config.LogConfig) logger.Logger {
switch cfg.Output {
case "none":
return logger.Nop()
case "stdout", "":
case "stdout":
out = os.Stdout
case "stderr":
case "stderr", "":
out = os.Stderr
default:
f, err := os.OpenFile(cfg.Output, os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0666)

4
pkg/chain/chain.go
View File

@ -1,9 +1,5 @@
package chain
type Chainable interface {
WithChain(chain *Chain)
}
type Chain struct {
groups []*NodeGroup
}

8
pkg/config/config.go
View File

@ -31,11 +31,11 @@ type ProfilingConfig struct {
}
type TLSConfig struct {
Cert string
Key string
Cert string `yaml:",omitempty"`
Key string `yaml:",omitempty"`
CA string `yaml:",omitempty"`
Secure bool `yaml:",omitempty"`
ServerName string `yaml:",omitempty"`
ServerName string `yaml:"serverName,omitempty"`
}
type AuthConfig struct {
@ -59,7 +59,7 @@ type NameserverConfig struct {
Addr string
Chain string
Prefer string
ClientIP string
ClientIP string `yaml:"clientIP"`
Hostname string
TTL time.Duration
Timeout time.Duration

7
pkg/listener/dns/metadata.go
View File

@ -20,14 +20,17 @@ type metadata struct {
func (l *dnsListener) parseMetadata(md mdata.Metadata) (err error) {
const (
backlog = "backlog"
mode = "mode"
readBufferSize = "readBufferSize"
backlog = "backlog"
readTimeout = "readTimeout"
writeTimeout = "writeTimeout"
)
l.md.mode = mdata.GetString(md, mode)
l.md.readBufferSize = mdata.GetInt(md, readBufferSize)
l.md.readTimeout = mdata.GetDuration(md, readTimeout)
l.md.writeTimeout = mdata.GetDuration(md, writeTimeout)
l.md.backlog = mdata.GetInt(md, backlog)
if l.md.backlog <= 0 {

15
pkg/listener/http2/metadata.go
View File

@ -1,9 +1,6 @@
package http2
import (
"net/http"
"time"
mdata "github.com/go-gost/gost/pkg/metadata"
)
@ -12,23 +9,11 @@ const (
)
type metadata struct {
path string
handshakeTimeout time.Duration
readHeaderTimeout time.Duration
readBufferSize int
writeBufferSize int
enableCompression bool
responseHeader http.Header
backlog int
}
func (l *http2Listener) parseMetadata(md mdata.Metadata) (err error) {
const (
path = "path"
handshakeTimeout = "handshakeTimeout"
readHeaderTimeout = "readHeaderTimeout"
readBufferSize = "readBufferSize"
writeBufferSize = "writeBufferSize"
backlog = "backlog"
)

8
pkg/listener/option.go
View File

@ -4,6 +4,7 @@ import (
"crypto/tls"
"net/url"
"github.com/go-gost/gost/pkg/chain"
"github.com/go-gost/gost/pkg/logger"
)
@ -11,6 +12,7 @@ type Options struct {
Addr string
Auths []*url.Userinfo
TLSConfig *tls.Config
Chain *chain.Chain
Logger logger.Logger
}
@ -34,6 +36,12 @@ func TLSConfigOption(tlsConfig *tls.Config) Option {
}
}
func ChainOption(chain *chain.Chain) Option {
return func(opts *Options) {
opts.Chain = chain
}
}
func LoggerOption(logger logger.Logger) Option {
return func(opts *Options) {
opts.Logger = logger

23
pkg/listener/rtcp/listener.go
View File

@ -17,47 +17,42 @@ func init() {
}
type rtcpListener struct {
addr string
laddr net.Addr
chain *chain.Chain
ln net.Listener
md metadata
router *chain.Router
logger logger.Logger
closed chan struct{}
options listener.Options
}
func NewListener(opts ...listener.Option) listener.Listener {
options := &listener.Options{}
options := listener.Options{}
for _, opt := range opts {
opt(options)
opt(&options)
}
return &rtcpListener{
addr: options.Addr,
closed: make(chan struct{}),
router: &chain.Router{
Logger: options.Logger,
},
logger: options.Logger,
options: options,
}
}
// implements chain.Chainable interface
func (l *rtcpListener) WithChain(chain *chain.Chain) {
l.router.Chain = chain
}
func (l *rtcpListener) Init(md md.Metadata) (err error) {
if err = l.parseMetadata(md); err != nil {
return
}
laddr, err := net.ResolveTCPAddr("tcp", l.addr)
laddr, err := net.ResolveTCPAddr("tcp", l.options.Addr)
if err != nil {
return
}
l.laddr = laddr
l.router = &chain.Router{
Chain: l.options.Chain,
Logger: l.logger,
}
return
}

16
pkg/listener/rtcp/metadata.go
View File

@ -12,24 +12,8 @@ const (
)
type metadata struct {
enableMux bool
backlog int
retryCount int
}
func (l *rtcpListener) parseMetadata(md mdata.Metadata) (err error) {
const (
enableMux = "mux"
backlog = "backlog"
retryCount = "retry"
)
l.md.enableMux = mdata.GetBool(md, enableMux)
l.md.retryCount = mdata.GetInt(md, retryCount)
l.md.backlog = mdata.GetInt(md, backlog)
if l.md.backlog <= 0 {
l.md.backlog = defaultBacklog
}
return
}

27
pkg/listener/rudp/listener.go
View File

@ -17,47 +17,42 @@ func init() {
}
type rudpListener struct {
addr string
laddr *net.UDPAddr
chain *chain.Chain
ln net.Listener
md metadata
router *chain.Router
logger logger.Logger
closed chan struct{}
logger logger.Logger
md metadata
options listener.Options
}
func NewListener(opts ...listener.Option) listener.Listener {
options := &listener.Options{}
options := listener.Options{}
for _, opt := range opts {
opt(options)
opt(&options)
}
return &rudpListener{
addr: options.Addr,
closed: make(chan struct{}),
router: &chain.Router{
Logger: options.Logger,
},
logger: options.Logger,
options: options,
}
}
// implements chain.Chainable interface
func (l *rudpListener) WithChain(chain *chain.Chain) {
l.router.Chain = chain
}
func (l *rudpListener) Init(md md.Metadata) (err error) {
if err = l.parseMetadata(md); err != nil {
return
}
laddr, err := net.ResolveUDPAddr("udp", l.addr)
laddr, err := net.ResolveUDPAddr("udp", l.options.Addr)
if err != nil {
return
}
l.laddr = laddr
l.router = &chain.Router{
Chain: l.options.Chain,
Logger: l.logger,
}
return
}

10
pkg/listener/tun/listener.go
View File

@ -15,22 +15,22 @@ func init() {
}
type tunListener struct {
saddr string
addr net.Addr
cqueue chan net.Conn
closed chan struct{}
logger logger.Logger
md metadata
options listener.Options
}
func NewListener(opts ...listener.Option) listener.Listener {
options := &listener.Options{}
options := listener.Options{}
for _, opt := range opts {
opt(options)
opt(&options)
}
return &tunListener{
saddr: options.Addr,
logger: options.Logger,
options: options,
}
}
@ -39,7 +39,7 @@ func (l *tunListener) Init(md md.Metadata) (err error) {
return
}
l.addr, err = net.ResolveUDPAddr("udp", l.saddr)
l.addr, err = net.ResolveUDPAddr("udp", l.options.Addr)
if err != nil {
return
}

4
pkg/metadata/metadata.go
View File

@ -30,6 +30,10 @@ func (m MapMetadata) Get(key string) interface{} {
return nil
}
func (m MapMetadata) Del(key string) {
delete(m, key)
}
func GetBool(md Metadata, key string) (v bool) {
if md == nil || !md.IsExists(key) {
return