add rtcp
This commit is contained in:
parent
3fe5084629
commit
ce3d62759a
@ -47,6 +47,7 @@ func buildService(cfg *config.Config) (services []*service.Service) {
|
|||||||
})
|
})
|
||||||
ln := registry.GetListener(svc.Listener.Type)(
|
ln := registry.GetListener(svc.Listener.Type)(
|
||||||
listener.AddrOption(svc.Addr),
|
listener.AddrOption(svc.Addr),
|
||||||
|
listener.ChainOption(chains[svc.Listener.Chain]),
|
||||||
listener.LoggerOption(listenerLogger),
|
listener.LoggerOption(listenerLogger),
|
||||||
)
|
)
|
||||||
if err := ln.Init(metadata.MapMetadata(svc.Listener.Metadata)); err != nil {
|
if err := ln.Init(metadata.MapMetadata(svc.Listener.Metadata)); err != nil {
|
||||||
@ -59,8 +60,8 @@ func buildService(cfg *config.Config) (services []*service.Service) {
|
|||||||
})
|
})
|
||||||
|
|
||||||
h := registry.GetHandler(svc.Handler.Type)(
|
h := registry.GetHandler(svc.Handler.Type)(
|
||||||
handler.ChainOption(chains[svc.Chain]),
|
handler.ChainOption(chains[svc.Handler.Chain]),
|
||||||
handler.BypassOption(bypasses[svc.Bypass]),
|
handler.BypassOption(bypasses[svc.Handler.Bypass]),
|
||||||
handler.LoggerOption(handlerLogger),
|
handler.LoggerOption(handlerLogger),
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -13,6 +13,8 @@ services:
|
|||||||
addr: ":28000"
|
addr: ":28000"
|
||||||
handler:
|
handler:
|
||||||
type: http
|
type: http
|
||||||
|
chain: chain01
|
||||||
|
# bypass: bypass01
|
||||||
metadata:
|
metadata:
|
||||||
proxyAgent: "gost/3.0"
|
proxyAgent: "gost/3.0"
|
||||||
retry: 3
|
retry: 3
|
||||||
@ -25,13 +27,13 @@ services:
|
|||||||
type: tcp
|
type: tcp
|
||||||
metadata:
|
metadata:
|
||||||
keepAlive: 15s
|
keepAlive: 15s
|
||||||
chain: chain01
|
|
||||||
# bypass: bypass01
|
|
||||||
- name: ss
|
- name: ss
|
||||||
url: "ss://chacha20:gost@:8000"
|
url: "ss://chacha20:gost@:8000"
|
||||||
addr: ":28338"
|
addr: ":28338"
|
||||||
handler:
|
handler:
|
||||||
type: ss
|
type: ss
|
||||||
|
# chain: chain01
|
||||||
|
# bypass: bypass01
|
||||||
metadata:
|
metadata:
|
||||||
method: chacha20-ietf
|
method: chacha20-ietf
|
||||||
password: gost
|
password: gost
|
||||||
@ -43,26 +45,26 @@ services:
|
|||||||
type: tcp
|
type: tcp
|
||||||
metadata:
|
metadata:
|
||||||
keepAlive: 15s
|
keepAlive: 15s
|
||||||
# chain: chain01
|
|
||||||
# bypass: bypass01
|
|
||||||
- name: socks5
|
- name: socks5
|
||||||
url: "socks5://gost:gost@:1080"
|
url: "socks5://gost:gost@:1080"
|
||||||
addr: ":21080"
|
addr: ":21080"
|
||||||
handler:
|
handler:
|
||||||
type: socks5
|
type: socks5
|
||||||
|
# chain: chain-ss
|
||||||
|
# bypass: bypass01
|
||||||
metadata:
|
metadata:
|
||||||
auths:
|
auths:
|
||||||
- gost:gost
|
- gost:gost
|
||||||
readTimeout: 5s
|
readTimeout: 5s
|
||||||
retry: 3
|
retry: 3
|
||||||
notls: true
|
notls: true
|
||||||
|
bind: true
|
||||||
|
udp: true
|
||||||
# udpBufferSize: 4096 # range [512, 66560]
|
# udpBufferSize: 4096 # range [512, 66560]
|
||||||
listener:
|
listener:
|
||||||
type: tcp
|
type: tcp
|
||||||
metadata:
|
metadata:
|
||||||
keepAlive: 15s
|
keepAlive: 15s
|
||||||
chain: chain-ss
|
|
||||||
# bypass: bypass01
|
|
||||||
- name: socks5+tcp
|
- name: socks5+tcp
|
||||||
url: "socks5://gost:gost@:1080"
|
url: "socks5://gost:gost@:1080"
|
||||||
addr: ":21081"
|
addr: ":21081"
|
||||||
@ -93,6 +95,7 @@ services:
|
|||||||
failTimeout: 30s
|
failTimeout: 30s
|
||||||
handler:
|
handler:
|
||||||
type: forward
|
type: forward
|
||||||
|
chain: chain-ss
|
||||||
metadata:
|
metadata:
|
||||||
readTimeout: 5s
|
readTimeout: 5s
|
||||||
retry: 3
|
retry: 3
|
||||||
@ -100,7 +103,38 @@ services:
|
|||||||
type: udp
|
type: udp
|
||||||
metadata:
|
metadata:
|
||||||
keepAlive: 15s
|
keepAlive: 15s
|
||||||
chain: chain-ss
|
|
||||||
|
- name: kcp-forward-tunnel
|
||||||
|
addr: ":8388"
|
||||||
|
forwarder:
|
||||||
|
targets:
|
||||||
|
- 127.0.0.1:28338
|
||||||
|
handler:
|
||||||
|
type: forward
|
||||||
|
metadata:
|
||||||
|
readTimeout: 5s
|
||||||
|
retry: 3
|
||||||
|
listener:
|
||||||
|
type: kcp
|
||||||
|
metadata:
|
||||||
|
keepAlive: 15s
|
||||||
|
|
||||||
|
- name: rtcp
|
||||||
|
addr: ":28100"
|
||||||
|
forwarder:
|
||||||
|
targets:
|
||||||
|
- 192.168.8.8:80
|
||||||
|
handler:
|
||||||
|
type: forward
|
||||||
|
metadata:
|
||||||
|
readTimeout: 5s
|
||||||
|
retry: 3
|
||||||
|
listener:
|
||||||
|
type: rtcp
|
||||||
|
chain: chain-socks5
|
||||||
|
metadata:
|
||||||
|
keepAlive: 15s
|
||||||
|
mux: true
|
||||||
|
|
||||||
chains:
|
chains:
|
||||||
- name: chain01
|
- name: chain01
|
||||||
@ -179,7 +213,7 @@ chains:
|
|||||||
- name: hop01
|
- name: hop01
|
||||||
nodes:
|
nodes:
|
||||||
- name: node01
|
- name: node01
|
||||||
addr: ":11080"
|
addr: ":21080"
|
||||||
url: "http://gost:gost@:8081"
|
url: "http://gost:gost@:8081"
|
||||||
# bypass: bypass01
|
# bypass: bypass01
|
||||||
connector:
|
connector:
|
||||||
|
@ -26,6 +26,7 @@ import (
|
|||||||
_ "github.com/go-gost/gost/pkg/listener/obfs/http"
|
_ "github.com/go-gost/gost/pkg/listener/obfs/http"
|
||||||
_ "github.com/go-gost/gost/pkg/listener/obfs/tls"
|
_ "github.com/go-gost/gost/pkg/listener/obfs/tls"
|
||||||
_ "github.com/go-gost/gost/pkg/listener/quic"
|
_ "github.com/go-gost/gost/pkg/listener/quic"
|
||||||
|
_ "github.com/go-gost/gost/pkg/listener/rtcp"
|
||||||
_ "github.com/go-gost/gost/pkg/listener/tcp"
|
_ "github.com/go-gost/gost/pkg/listener/tcp"
|
||||||
_ "github.com/go-gost/gost/pkg/listener/tls"
|
_ "github.com/go-gost/gost/pkg/listener/tls"
|
||||||
_ "github.com/go-gost/gost/pkg/listener/tls/mux"
|
_ "github.com/go-gost/gost/pkg/listener/tls/mux"
|
||||||
|
@ -1,4 +1,4 @@
|
|||||||
package handler
|
package chain
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"bytes"
|
"bytes"
|
||||||
@ -6,17 +6,16 @@ import (
|
|||||||
"fmt"
|
"fmt"
|
||||||
"net"
|
"net"
|
||||||
|
|
||||||
"github.com/go-gost/gost/pkg/chain"
|
|
||||||
"github.com/go-gost/gost/pkg/logger"
|
"github.com/go-gost/gost/pkg/logger"
|
||||||
)
|
)
|
||||||
|
|
||||||
type Router struct {
|
type Router struct {
|
||||||
chain *chain.Chain
|
chain *Chain
|
||||||
retries int
|
retries int
|
||||||
logger logger.Logger
|
logger logger.Logger
|
||||||
}
|
}
|
||||||
|
|
||||||
func (r *Router) WithChain(chain *chain.Chain) *Router {
|
func (r *Router) WithChain(chain *Chain) *Router {
|
||||||
r.chain = chain
|
r.chain = chain
|
||||||
return r
|
return r
|
||||||
}
|
}
|
@ -42,11 +42,14 @@ type BypassConfig struct {
|
|||||||
}
|
}
|
||||||
type ListenerConfig struct {
|
type ListenerConfig struct {
|
||||||
Type string
|
Type string
|
||||||
|
Chain string
|
||||||
Metadata map[string]interface{}
|
Metadata map[string]interface{}
|
||||||
}
|
}
|
||||||
|
|
||||||
type HandlerConfig struct {
|
type HandlerConfig struct {
|
||||||
Type string
|
Type string
|
||||||
|
Chain string
|
||||||
|
Bypass string
|
||||||
Metadata map[string]interface{}
|
Metadata map[string]interface{}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -72,8 +75,6 @@ type ServiceConfig struct {
|
|||||||
Listener *ListenerConfig
|
Listener *ListenerConfig
|
||||||
Handler *HandlerConfig
|
Handler *HandlerConfig
|
||||||
Forwarder *ForwarderConfig
|
Forwarder *ForwarderConfig
|
||||||
Chain string
|
|
||||||
Bypass string
|
|
||||||
}
|
}
|
||||||
|
|
||||||
type ChainConfig struct {
|
type ChainConfig struct {
|
||||||
|
@ -9,7 +9,6 @@ import (
|
|||||||
"net/http"
|
"net/http"
|
||||||
"net/http/httputil"
|
"net/http/httputil"
|
||||||
"net/url"
|
"net/url"
|
||||||
"strings"
|
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/go-gost/gost/pkg/connector"
|
"github.com/go-gost/gost/pkg/connector"
|
||||||
@ -111,22 +110,3 @@ func (c *httpConnector) Connect(ctx context.Context, conn net.Conn, network, add
|
|||||||
|
|
||||||
return conn, nil
|
return conn, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *httpConnector) parseMetadata(md md.Metadata) (err error) {
|
|
||||||
c.md.connectTimeout = md.GetDuration(connectTimeout)
|
|
||||||
c.md.UserAgent, _ = md.Get(userAgent).(string)
|
|
||||||
if c.md.UserAgent == "" {
|
|
||||||
c.md.UserAgent = defaultUserAgent
|
|
||||||
}
|
|
||||||
|
|
||||||
if v := md.GetString(auth); v != "" {
|
|
||||||
ss := strings.SplitN(v, ":", 2)
|
|
||||||
if len(ss) == 1 {
|
|
||||||
c.md.User = url.User(ss[0])
|
|
||||||
} else {
|
|
||||||
c.md.User = url.UserPassword(ss[0], ss[1])
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
@ -2,13 +2,10 @@ package http
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"net/url"
|
"net/url"
|
||||||
|
"strings"
|
||||||
"time"
|
"time"
|
||||||
)
|
|
||||||
|
|
||||||
const (
|
md "github.com/go-gost/gost/pkg/metadata"
|
||||||
connectTimeout = "timeout"
|
|
||||||
userAgent = "userAgent"
|
|
||||||
auth = "auth"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
const (
|
const (
|
||||||
@ -20,3 +17,28 @@ type metadata struct {
|
|||||||
UserAgent string
|
UserAgent string
|
||||||
User *url.Userinfo
|
User *url.Userinfo
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (c *httpConnector) parseMetadata(md md.Metadata) (err error) {
|
||||||
|
const (
|
||||||
|
connectTimeout = "timeout"
|
||||||
|
userAgent = "userAgent"
|
||||||
|
auth = "auth"
|
||||||
|
)
|
||||||
|
|
||||||
|
c.md.connectTimeout = md.GetDuration(connectTimeout)
|
||||||
|
c.md.UserAgent, _ = md.Get(userAgent).(string)
|
||||||
|
if c.md.UserAgent == "" {
|
||||||
|
c.md.UserAgent = defaultUserAgent
|
||||||
|
}
|
||||||
|
|
||||||
|
if v := md.GetString(auth); v != "" {
|
||||||
|
ss := strings.SplitN(v, ":", 2)
|
||||||
|
if len(ss) == 1 {
|
||||||
|
c.md.User = url.User(ss[0])
|
||||||
|
} else {
|
||||||
|
c.md.User = url.UserPassword(ss[0], ss[1])
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
@ -5,7 +5,6 @@ import (
|
|||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"net"
|
"net"
|
||||||
"net/url"
|
|
||||||
"strconv"
|
"strconv"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
@ -114,13 +113,3 @@ func (c *socks4Connector) Connect(ctx context.Context, conn net.Conn, network, a
|
|||||||
|
|
||||||
return conn, nil
|
return conn, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *socks4Connector) parseMetadata(md md.Metadata) (err error) {
|
|
||||||
if v := md.GetString(auth); v != "" {
|
|
||||||
c.md.User = url.User(v)
|
|
||||||
}
|
|
||||||
c.md.connectTimeout = md.GetDuration(connectTimeout)
|
|
||||||
c.md.disable4a = md.GetBool(disable4a)
|
|
||||||
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
@ -3,12 +3,8 @@ package v4
|
|||||||
import (
|
import (
|
||||||
"net/url"
|
"net/url"
|
||||||
"time"
|
"time"
|
||||||
)
|
|
||||||
|
|
||||||
const (
|
md "github.com/go-gost/gost/pkg/metadata"
|
||||||
connectTimeout = "timeout"
|
|
||||||
auth = "auth"
|
|
||||||
disable4a = "disable4a"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
type metadata struct {
|
type metadata struct {
|
||||||
@ -16,3 +12,19 @@ type metadata struct {
|
|||||||
User *url.Userinfo
|
User *url.Userinfo
|
||||||
disable4a bool
|
disable4a bool
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (c *socks4Connector) parseMetadata(md md.Metadata) (err error) {
|
||||||
|
const (
|
||||||
|
connectTimeout = "timeout"
|
||||||
|
auth = "auth"
|
||||||
|
disable4a = "disable4a"
|
||||||
|
)
|
||||||
|
|
||||||
|
if v := md.GetString(auth); v != "" {
|
||||||
|
c.md.User = url.User(v)
|
||||||
|
}
|
||||||
|
c.md.connectTimeout = md.GetDuration(connectTimeout)
|
||||||
|
c.md.disable4a = md.GetBool(disable4a)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
@ -6,8 +6,6 @@ import (
|
|||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"net"
|
"net"
|
||||||
"net/url"
|
|
||||||
"strings"
|
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/go-gost/gosocks5"
|
"github.com/go-gost/gosocks5"
|
||||||
@ -172,19 +170,3 @@ func (c *socks5Connector) connectUDP(ctx context.Context, conn net.Conn, network
|
|||||||
|
|
||||||
return socks.UDPTunClientConn(conn, addr), nil
|
return socks.UDPTunClientConn(conn, addr), nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *socks5Connector) parseMetadata(md md.Metadata) (err error) {
|
|
||||||
if v := md.GetString(auth); v != "" {
|
|
||||||
ss := strings.SplitN(v, ":", 2)
|
|
||||||
if len(ss) == 1 {
|
|
||||||
c.md.User = url.User(ss[0])
|
|
||||||
} else {
|
|
||||||
c.md.User = url.UserPassword(ss[0], ss[1])
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
c.md.connectTimeout = md.GetDuration(connectTimeout)
|
|
||||||
c.md.noTLS = md.GetBool(noTLS)
|
|
||||||
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
@ -3,13 +3,10 @@ package v5
|
|||||||
import (
|
import (
|
||||||
"crypto/tls"
|
"crypto/tls"
|
||||||
"net/url"
|
"net/url"
|
||||||
|
"strings"
|
||||||
"time"
|
"time"
|
||||||
)
|
|
||||||
|
|
||||||
const (
|
md "github.com/go-gost/gost/pkg/metadata"
|
||||||
connectTimeout = "timeout"
|
|
||||||
auth = "auth"
|
|
||||||
noTLS = "notls"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
type metadata struct {
|
type metadata struct {
|
||||||
@ -18,3 +15,25 @@ type metadata struct {
|
|||||||
tlsConfig *tls.Config
|
tlsConfig *tls.Config
|
||||||
noTLS bool
|
noTLS bool
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (c *socks5Connector) parseMetadata(md md.Metadata) (err error) {
|
||||||
|
const (
|
||||||
|
connectTimeout = "timeout"
|
||||||
|
auth = "auth"
|
||||||
|
noTLS = "notls"
|
||||||
|
)
|
||||||
|
|
||||||
|
if v := md.GetString(auth); v != "" {
|
||||||
|
ss := strings.SplitN(v, ":", 2)
|
||||||
|
if len(ss) == 1 {
|
||||||
|
c.md.User = url.User(ss[0])
|
||||||
|
} else {
|
||||||
|
c.md.User = url.UserPassword(ss[0], ss[1])
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
c.md.connectTimeout = md.GetDuration(connectTimeout)
|
||||||
|
c.md.noTLS = md.GetBool(noTLS)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
@ -75,7 +75,7 @@ func (h *localForwardHandler) Handle(ctx context.Context, conn net.Conn) {
|
|||||||
|
|
||||||
h.logger.Infof("%s >> %s", conn.RemoteAddr(), target.Addr())
|
h.logger.Infof("%s >> %s", conn.RemoteAddr(), target.Addr())
|
||||||
|
|
||||||
r := (&handler.Router{}).
|
r := (&chain.Router{}).
|
||||||
WithChain(h.chain).
|
WithChain(h.chain).
|
||||||
WithRetry(h.md.retryCount).
|
WithRetry(h.md.retryCount).
|
||||||
WithLogger(h.logger)
|
WithLogger(h.logger)
|
||||||
@ -105,9 +105,3 @@ func (h *localForwardHandler) Handle(ctx context.Context, conn net.Conn) {
|
|||||||
}).
|
}).
|
||||||
Infof("%s >-< %s", conn.RemoteAddr(), target.Addr())
|
Infof("%s >-< %s", conn.RemoteAddr(), target.Addr())
|
||||||
}
|
}
|
||||||
|
|
||||||
func (h *localForwardHandler) parseMetadata(md md.Metadata) (err error) {
|
|
||||||
h.md.readTimeout = md.GetDuration(readTimeout)
|
|
||||||
h.md.retryCount = md.GetInt(retryCount)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
@ -2,14 +2,22 @@ package local
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"time"
|
"time"
|
||||||
)
|
|
||||||
|
|
||||||
const (
|
md "github.com/go-gost/gost/pkg/metadata"
|
||||||
readTimeout = "readTimeout"
|
|
||||||
retryCount = "retry"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
type metadata struct {
|
type metadata struct {
|
||||||
readTimeout time.Duration
|
readTimeout time.Duration
|
||||||
retryCount int
|
retryCount int
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (h *localForwardHandler) parseMetadata(md md.Metadata) (err error) {
|
||||||
|
const (
|
||||||
|
readTimeout = "readTimeout"
|
||||||
|
retryCount = "retry"
|
||||||
|
)
|
||||||
|
|
||||||
|
h.md.readTimeout = md.GetDuration(readTimeout)
|
||||||
|
h.md.retryCount = md.GetInt(retryCount)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
@ -15,7 +15,6 @@ import (
|
|||||||
"strings"
|
"strings"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/go-gost/gost/pkg/auth"
|
|
||||||
"github.com/go-gost/gost/pkg/bypass"
|
"github.com/go-gost/gost/pkg/bypass"
|
||||||
"github.com/go-gost/gost/pkg/chain"
|
"github.com/go-gost/gost/pkg/chain"
|
||||||
"github.com/go-gost/gost/pkg/handler"
|
"github.com/go-gost/gost/pkg/handler"
|
||||||
@ -52,38 +51,6 @@ func (h *httpHandler) Init(md md.Metadata) error {
|
|||||||
return h.parseMetadata(md)
|
return h.parseMetadata(md)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (h *httpHandler) parseMetadata(md md.Metadata) error {
|
|
||||||
h.md.proxyAgent = md.GetString(proxyAgentKey)
|
|
||||||
|
|
||||||
if v, _ := md.Get(authsKey).([]interface{}); len(v) > 0 {
|
|
||||||
authenticator := auth.NewLocalAuthenticator(nil)
|
|
||||||
for _, auth := range v {
|
|
||||||
if s, _ := auth.(string); s != "" {
|
|
||||||
ss := strings.SplitN(s, ":", 2)
|
|
||||||
if len(ss) == 1 {
|
|
||||||
authenticator.Add(ss[0], "")
|
|
||||||
} else {
|
|
||||||
authenticator.Add(ss[0], ss[1])
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
h.md.authenticator = authenticator
|
|
||||||
}
|
|
||||||
|
|
||||||
if v := md.GetString(probeResistKey); v != "" {
|
|
||||||
if ss := strings.SplitN(v, ":", 2); len(ss) == 2 {
|
|
||||||
h.md.probeResist = &probeResist{
|
|
||||||
Type: ss[0],
|
|
||||||
Value: ss[1],
|
|
||||||
Knock: md.GetString(knockKey),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
h.md.retryCount = md.GetInt(retryCount)
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (h *httpHandler) Handle(ctx context.Context, conn net.Conn) {
|
func (h *httpHandler) Handle(ctx context.Context, conn net.Conn) {
|
||||||
defer conn.Close()
|
defer conn.Close()
|
||||||
|
|
||||||
@ -209,7 +176,7 @@ func (h *httpHandler) handleRequest(ctx context.Context, conn net.Conn, req *htt
|
|||||||
|
|
||||||
req.Header.Del("Proxy-Authorization")
|
req.Header.Del("Proxy-Authorization")
|
||||||
|
|
||||||
r := (&handler.Router{}).
|
r := (&chain.Router{}).
|
||||||
WithChain(h.chain).
|
WithChain(h.chain).
|
||||||
WithRetry(h.md.retryCount).
|
WithRetry(h.md.retryCount).
|
||||||
WithLogger(h.logger)
|
WithLogger(h.logger)
|
||||||
|
@ -1,13 +1,10 @@
|
|||||||
package http
|
package http
|
||||||
|
|
||||||
import "github.com/go-gost/gost/pkg/auth"
|
import (
|
||||||
|
"strings"
|
||||||
|
|
||||||
const (
|
"github.com/go-gost/gost/pkg/auth"
|
||||||
proxyAgentKey = "proxyAgent"
|
md "github.com/go-gost/gost/pkg/metadata"
|
||||||
authsKey = "auths"
|
|
||||||
probeResistKey = "probeResist"
|
|
||||||
knockKey = "knock"
|
|
||||||
retryCount = "retry"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
type metadata struct {
|
type metadata struct {
|
||||||
@ -17,6 +14,46 @@ type metadata struct {
|
|||||||
probeResist *probeResist
|
probeResist *probeResist
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (h *httpHandler) parseMetadata(md md.Metadata) error {
|
||||||
|
const (
|
||||||
|
proxyAgentKey = "proxyAgent"
|
||||||
|
authsKey = "auths"
|
||||||
|
probeResistKey = "probeResist"
|
||||||
|
knockKey = "knock"
|
||||||
|
retryCount = "retry"
|
||||||
|
)
|
||||||
|
|
||||||
|
h.md.proxyAgent = md.GetString(proxyAgentKey)
|
||||||
|
|
||||||
|
if v, _ := md.Get(authsKey).([]interface{}); len(v) > 0 {
|
||||||
|
authenticator := auth.NewLocalAuthenticator(nil)
|
||||||
|
for _, auth := range v {
|
||||||
|
if s, _ := auth.(string); s != "" {
|
||||||
|
ss := strings.SplitN(s, ":", 2)
|
||||||
|
if len(ss) == 1 {
|
||||||
|
authenticator.Add(ss[0], "")
|
||||||
|
} else {
|
||||||
|
authenticator.Add(ss[0], ss[1])
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
h.md.authenticator = authenticator
|
||||||
|
}
|
||||||
|
|
||||||
|
if v := md.GetString(probeResistKey); v != "" {
|
||||||
|
if ss := strings.SplitN(v, ":", 2); len(ss) == 2 {
|
||||||
|
h.md.probeResist = &probeResist{
|
||||||
|
Type: ss[0],
|
||||||
|
Value: ss[1],
|
||||||
|
Knock: md.GetString(knockKey),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
h.md.retryCount = md.GetInt(retryCount)
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
type probeResist struct {
|
type probeResist struct {
|
||||||
Type string
|
Type string
|
||||||
Value string
|
Value string
|
||||||
|
@ -6,7 +6,6 @@ import (
|
|||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/go-gost/gosocks4"
|
"github.com/go-gost/gosocks4"
|
||||||
"github.com/go-gost/gost/pkg/auth"
|
|
||||||
"github.com/go-gost/gost/pkg/bypass"
|
"github.com/go-gost/gost/pkg/bypass"
|
||||||
"github.com/go-gost/gost/pkg/chain"
|
"github.com/go-gost/gost/pkg/chain"
|
||||||
"github.com/go-gost/gost/pkg/handler"
|
"github.com/go-gost/gost/pkg/handler"
|
||||||
@ -108,7 +107,7 @@ func (h *socks4Handler) handleConnect(ctx context.Context, conn net.Conn, req *g
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
r := (&handler.Router{}).
|
r := (&chain.Router{}).
|
||||||
WithChain(h.chain).
|
WithChain(h.chain).
|
||||||
WithRetry(h.md.retryCount).
|
WithRetry(h.md.retryCount).
|
||||||
WithLogger(h.logger)
|
WithLogger(h.logger)
|
||||||
@ -142,19 +141,3 @@ func (h *socks4Handler) handleConnect(ctx context.Context, conn net.Conn, req *g
|
|||||||
func (h *socks4Handler) handleBind(ctx context.Context, conn net.Conn, req *gosocks4.Request) {
|
func (h *socks4Handler) handleBind(ctx context.Context, conn net.Conn, req *gosocks4.Request) {
|
||||||
// TODO: bind
|
// TODO: bind
|
||||||
}
|
}
|
||||||
|
|
||||||
func (h *socks4Handler) parseMetadata(md md.Metadata) (err error) {
|
|
||||||
if v, _ := md.Get(authsKey).([]interface{}); len(v) > 0 {
|
|
||||||
authenticator := auth.NewLocalAuthenticator(nil)
|
|
||||||
for _, auth := range v {
|
|
||||||
if v, _ := auth.(string); v != "" {
|
|
||||||
authenticator.Add(v, "")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
h.md.authenticator = authenticator
|
|
||||||
}
|
|
||||||
|
|
||||||
h.md.readTimeout = md.GetDuration(readTimeout)
|
|
||||||
h.md.retryCount = md.GetInt(retryCount)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
@ -4,12 +4,7 @@ import (
|
|||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/go-gost/gost/pkg/auth"
|
"github.com/go-gost/gost/pkg/auth"
|
||||||
)
|
md "github.com/go-gost/gost/pkg/metadata"
|
||||||
|
|
||||||
const (
|
|
||||||
authsKey = "auths"
|
|
||||||
readTimeout = "readTimeout"
|
|
||||||
retryCount = "retry"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
type metadata struct {
|
type metadata struct {
|
||||||
@ -17,3 +12,25 @@ type metadata struct {
|
|||||||
readTimeout time.Duration
|
readTimeout time.Duration
|
||||||
retryCount int
|
retryCount int
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (h *socks4Handler) parseMetadata(md md.Metadata) (err error) {
|
||||||
|
const (
|
||||||
|
authsKey = "auths"
|
||||||
|
readTimeout = "readTimeout"
|
||||||
|
retryCount = "retry"
|
||||||
|
)
|
||||||
|
|
||||||
|
if v, _ := md.Get(authsKey).([]interface{}); len(v) > 0 {
|
||||||
|
authenticator := auth.NewLocalAuthenticator(nil)
|
||||||
|
for _, auth := range v {
|
||||||
|
if v, _ := auth.(string); v != "" {
|
||||||
|
authenticator.Add(v, "")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
h.md.authenticator = authenticator
|
||||||
|
}
|
||||||
|
|
||||||
|
h.md.readTimeout = md.GetDuration(readTimeout)
|
||||||
|
h.md.retryCount = md.GetInt(retryCount)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
@ -6,6 +6,7 @@ import (
|
|||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/go-gost/gosocks5"
|
"github.com/go-gost/gosocks5"
|
||||||
|
"github.com/go-gost/gost/pkg/chain"
|
||||||
"github.com/go-gost/gost/pkg/handler"
|
"github.com/go-gost/gost/pkg/handler"
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -19,12 +20,20 @@ func (h *socks5Handler) handleBind(ctx context.Context, conn net.Conn, req *goso
|
|||||||
|
|
||||||
h.logger.Infof("%s >> %s", conn.RemoteAddr(), addr)
|
h.logger.Infof("%s >> %s", conn.RemoteAddr(), addr)
|
||||||
|
|
||||||
|
if !h.md.enableBind {
|
||||||
|
reply := gosocks5.NewReply(gosocks5.NotAllowed, nil)
|
||||||
|
reply.Write(conn)
|
||||||
|
h.logger.Debug(reply)
|
||||||
|
h.logger.Error("BIND is diabled")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
if h.chain.IsEmpty() {
|
if h.chain.IsEmpty() {
|
||||||
h.bindLocal(ctx, conn, addr)
|
h.bindLocal(ctx, conn, addr)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
r := (&handler.Router{}).
|
r := (&chain.Router{}).
|
||||||
WithChain(h.chain).
|
WithChain(h.chain).
|
||||||
WithRetry(h.md.retryCount).
|
WithRetry(h.md.retryCount).
|
||||||
WithLogger(h.logger)
|
WithLogger(h.logger)
|
||||||
@ -83,7 +92,7 @@ func (h *socks5Handler) bindLocal(ctx context.Context, conn net.Conn, addr strin
|
|||||||
h.logger = h.logger.WithFields(map[string]interface{}{
|
h.logger = h.logger.WithFields(map[string]interface{}{
|
||||||
"bind": socksAddr.String(),
|
"bind": socksAddr.String(),
|
||||||
})
|
})
|
||||||
h.logger.Infof("bind on %s OK", socksAddr.String())
|
h.logger.Debugf("bind on %s OK", &socksAddr)
|
||||||
|
|
||||||
h.serveBind(ctx, conn, ln)
|
h.serveBind(ctx, conn, ln)
|
||||||
}
|
}
|
||||||
@ -127,10 +136,19 @@ func (h *socks5Handler) serveBind(ctx context.Context, conn net.Conn, ln net.Lis
|
|||||||
case err := <-accept():
|
case err := <-accept():
|
||||||
if err != nil {
|
if err != nil {
|
||||||
h.logger.Error(err)
|
h.logger.Error(err)
|
||||||
|
|
||||||
|
reply := gosocks5.NewReply(gosocks5.Failure, nil)
|
||||||
|
if err := reply.Write(pc2); err != nil {
|
||||||
|
h.logger.Error(err)
|
||||||
|
}
|
||||||
|
h.logger.Debug(reply)
|
||||||
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
defer rc.Close()
|
defer rc.Close()
|
||||||
|
|
||||||
|
h.logger.Debugf("peer %s accepted", rc.RemoteAddr())
|
||||||
|
|
||||||
raddr := gosocks5.Addr{}
|
raddr := gosocks5.Addr{}
|
||||||
raddr.ParseFrom(rc.RemoteAddr().String())
|
raddr.ParseFrom(rc.RemoteAddr().String())
|
||||||
reply := gosocks5.NewReply(gosocks5.Succeeded, &raddr)
|
reply := gosocks5.NewReply(gosocks5.Succeeded, &raddr)
|
||||||
@ -138,7 +156,6 @@ func (h *socks5Handler) serveBind(ctx context.Context, conn net.Conn, ln net.Lis
|
|||||||
h.logger.Error(err)
|
h.logger.Error(err)
|
||||||
}
|
}
|
||||||
h.logger.Debug(reply)
|
h.logger.Debug(reply)
|
||||||
h.logger.Infof("peer accepted: %s", raddr.String())
|
|
||||||
|
|
||||||
start := time.Now()
|
start := time.Now()
|
||||||
h.logger.Infof("%s <-> %s", conn.RemoteAddr(), raddr.String())
|
h.logger.Infof("%s <-> %s", conn.RemoteAddr(), raddr.String())
|
||||||
|
@ -6,6 +6,7 @@ import (
|
|||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/go-gost/gosocks5"
|
"github.com/go-gost/gosocks5"
|
||||||
|
"github.com/go-gost/gost/pkg/chain"
|
||||||
"github.com/go-gost/gost/pkg/handler"
|
"github.com/go-gost/gost/pkg/handler"
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -24,7 +25,7 @@ func (h *socks5Handler) handleConnect(ctx context.Context, conn net.Conn, addr s
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
r := (&handler.Router{}).
|
r := (&chain.Router{}).
|
||||||
WithChain(h.chain).
|
WithChain(h.chain).
|
||||||
WithRetry(h.md.retryCount).
|
WithRetry(h.md.retryCount).
|
||||||
WithLogger(h.logger)
|
WithLogger(h.logger)
|
||||||
|
@ -90,29 +90,13 @@ func (h *socks5Handler) Handle(ctx context.Context, conn net.Conn) {
|
|||||||
case gosocks5.CmdConnect:
|
case gosocks5.CmdConnect:
|
||||||
h.handleConnect(ctx, conn, req.Addr.String())
|
h.handleConnect(ctx, conn, req.Addr.String())
|
||||||
case gosocks5.CmdBind:
|
case gosocks5.CmdBind:
|
||||||
if h.md.enableBind {
|
|
||||||
h.handleBind(ctx, conn, req)
|
h.handleBind(ctx, conn, req)
|
||||||
} else {
|
|
||||||
h.logger.Error("BIND is diabled")
|
|
||||||
}
|
|
||||||
case socks.CmdMuxBind:
|
case socks.CmdMuxBind:
|
||||||
if h.md.enableBind {
|
|
||||||
h.handleMuxBind(ctx, conn, req)
|
h.handleMuxBind(ctx, conn, req)
|
||||||
} else {
|
|
||||||
h.logger.Error("BIND is diabled")
|
|
||||||
}
|
|
||||||
case gosocks5.CmdUdp:
|
case gosocks5.CmdUdp:
|
||||||
if h.md.enableUDP {
|
|
||||||
h.handleUDP(ctx, conn, req)
|
h.handleUDP(ctx, conn, req)
|
||||||
} else {
|
|
||||||
h.logger.Error("UDP relay is diabled")
|
|
||||||
}
|
|
||||||
case socks.CmdUDPTun:
|
case socks.CmdUDPTun:
|
||||||
if h.md.enableUDP {
|
|
||||||
h.handleUDPTun(ctx, conn, req)
|
h.handleUDPTun(ctx, conn, req)
|
||||||
} else {
|
|
||||||
h.logger.Error("UDP relay is diabled")
|
|
||||||
}
|
|
||||||
default:
|
default:
|
||||||
h.logger.Errorf("unknown cmd: %d", req.Cmd)
|
h.logger.Errorf("unknown cmd: %d", req.Cmd)
|
||||||
resp := gosocks5.NewReply(gosocks5.CmdUnsupported, nil)
|
resp := gosocks5.NewReply(gosocks5.CmdUnsupported, nil)
|
||||||
|
@ -6,6 +6,7 @@ import (
|
|||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/go-gost/gosocks5"
|
"github.com/go-gost/gosocks5"
|
||||||
|
"github.com/go-gost/gost/pkg/chain"
|
||||||
"github.com/go-gost/gost/pkg/handler"
|
"github.com/go-gost/gost/pkg/handler"
|
||||||
"github.com/go-gost/gost/pkg/internal/utils/mux"
|
"github.com/go-gost/gost/pkg/internal/utils/mux"
|
||||||
)
|
)
|
||||||
@ -20,12 +21,20 @@ func (h *socks5Handler) handleMuxBind(ctx context.Context, conn net.Conn, req *g
|
|||||||
|
|
||||||
h.logger.Infof("%s >> %s", conn.RemoteAddr(), addr)
|
h.logger.Infof("%s >> %s", conn.RemoteAddr(), addr)
|
||||||
|
|
||||||
|
if !h.md.enableBind {
|
||||||
|
reply := gosocks5.NewReply(gosocks5.NotAllowed, nil)
|
||||||
|
reply.Write(conn)
|
||||||
|
h.logger.Debug(reply)
|
||||||
|
h.logger.Error("BIND is diabled")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
if h.chain.IsEmpty() {
|
if h.chain.IsEmpty() {
|
||||||
h.muxBindLocal(ctx, conn, addr)
|
h.muxBindLocal(ctx, conn, addr)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
r := (&handler.Router{}).
|
r := (&chain.Router{}).
|
||||||
WithChain(h.chain).
|
WithChain(h.chain).
|
||||||
WithRetry(h.md.retryCount).
|
WithRetry(h.md.retryCount).
|
||||||
WithLogger(h.logger)
|
WithLogger(h.logger)
|
||||||
@ -90,14 +99,14 @@ func (h *socks5Handler) muxBindLocal(ctx context.Context, conn net.Conn, addr st
|
|||||||
h.logger = h.logger.WithFields(map[string]interface{}{
|
h.logger = h.logger.WithFields(map[string]interface{}{
|
||||||
"bind": socksAddr.String(),
|
"bind": socksAddr.String(),
|
||||||
})
|
})
|
||||||
h.logger.Infof("bind on: %s OK", socksAddr.String())
|
h.logger.Debugf("bind on %s OK", &socksAddr)
|
||||||
|
|
||||||
h.serveMuxBind(ctx, conn, ln)
|
h.serveMuxBind(ctx, conn, ln)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (h *socks5Handler) serveMuxBind(ctx context.Context, conn net.Conn, ln net.Listener) {
|
func (h *socks5Handler) serveMuxBind(ctx context.Context, conn net.Conn, ln net.Listener) {
|
||||||
// Upgrade connection to multiplex stream.
|
// Upgrade connection to multiplex stream.
|
||||||
session, err := mux.NewMuxSession(conn)
|
session, err := mux.ClientSession(conn)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
h.logger.Error(err)
|
h.logger.Error(err)
|
||||||
return
|
return
|
||||||
@ -122,7 +131,7 @@ func (h *socks5Handler) serveMuxBind(ctx context.Context, conn net.Conn, ln net.
|
|||||||
h.logger.Error(err)
|
h.logger.Error(err)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
h.logger.Infof("peer accepted: %s", rc.RemoteAddr().String())
|
h.logger.Debugf("peer %s accepted", rc.RemoteAddr())
|
||||||
|
|
||||||
go func(c net.Conn) {
|
go func(c net.Conn) {
|
||||||
defer c.Close()
|
defer c.Close()
|
||||||
@ -134,6 +143,18 @@ func (h *socks5Handler) serveMuxBind(ctx context.Context, conn net.Conn, ln net.
|
|||||||
}
|
}
|
||||||
defer sc.Close()
|
defer sc.Close()
|
||||||
|
|
||||||
|
// incompatible with GOST v2.x
|
||||||
|
if !h.md.compatibilityMode {
|
||||||
|
addr := gosocks5.Addr{}
|
||||||
|
addr.ParseFrom(c.RemoteAddr().String())
|
||||||
|
reply := gosocks5.NewReply(gosocks5.Succeeded, &addr)
|
||||||
|
if err := reply.Write(sc); err != nil {
|
||||||
|
h.logger.Error(err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
h.logger.Debug(reply)
|
||||||
|
}
|
||||||
|
|
||||||
t := time.Now()
|
t := time.Now()
|
||||||
h.logger.Infof("%s <-> %s", conn.RemoteAddr(), c.RemoteAddr().String())
|
h.logger.Infof("%s <-> %s", conn.RemoteAddr(), c.RemoteAddr().String())
|
||||||
handler.Transport(sc, c)
|
handler.Transport(sc, c)
|
||||||
|
@ -20,6 +20,7 @@ type metadata struct {
|
|||||||
enableBind bool
|
enableBind bool
|
||||||
enableUDP bool
|
enableUDP bool
|
||||||
udpBufferSize int
|
udpBufferSize int
|
||||||
|
compatibilityMode bool
|
||||||
}
|
}
|
||||||
|
|
||||||
func (h *socks5Handler) parseMetadata(md md.Metadata) error {
|
func (h *socks5Handler) parseMetadata(md md.Metadata) error {
|
||||||
@ -35,6 +36,7 @@ func (h *socks5Handler) parseMetadata(md md.Metadata) error {
|
|||||||
enableBind = "bind"
|
enableBind = "bind"
|
||||||
enableUDP = "udp"
|
enableUDP = "udp"
|
||||||
udpBufferSize = "udpBufferSize"
|
udpBufferSize = "udpBufferSize"
|
||||||
|
compatibilityMode = "comp"
|
||||||
)
|
)
|
||||||
|
|
||||||
var err error
|
var err error
|
||||||
@ -81,5 +83,7 @@ func (h *socks5Handler) parseMetadata(md md.Metadata) error {
|
|||||||
h.md.udpBufferSize = 4096 // default buffer size
|
h.md.udpBufferSize = 4096 // default buffer size
|
||||||
}
|
}
|
||||||
|
|
||||||
|
h.md.compatibilityMode = md.GetBool(compatibilityMode)
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
@ -9,7 +9,7 @@ import (
|
|||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/go-gost/gosocks5"
|
"github.com/go-gost/gosocks5"
|
||||||
"github.com/go-gost/gost/pkg/handler"
|
"github.com/go-gost/gost/pkg/chain"
|
||||||
"github.com/go-gost/gost/pkg/internal/bufpool"
|
"github.com/go-gost/gost/pkg/internal/bufpool"
|
||||||
"github.com/go-gost/gost/pkg/internal/utils/socks"
|
"github.com/go-gost/gost/pkg/internal/utils/socks"
|
||||||
)
|
)
|
||||||
@ -19,6 +19,14 @@ func (h *socks5Handler) handleUDP(ctx context.Context, conn net.Conn, req *gosoc
|
|||||||
"cmd": "udp",
|
"cmd": "udp",
|
||||||
})
|
})
|
||||||
|
|
||||||
|
if !h.md.enableUDP {
|
||||||
|
reply := gosocks5.NewReply(gosocks5.NotAllowed, nil)
|
||||||
|
reply.Write(conn)
|
||||||
|
h.logger.Debug(reply)
|
||||||
|
h.logger.Error("UDP relay is diabled")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
relay, err := net.ListenUDP("udp", nil)
|
relay, err := net.ListenUDP("udp", nil)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
h.logger.Error(err)
|
h.logger.Error(err)
|
||||||
@ -43,7 +51,7 @@ func (h *socks5Handler) handleUDP(ctx context.Context, conn net.Conn, req *gosoc
|
|||||||
h.logger = h.logger.WithFields(map[string]interface{}{
|
h.logger = h.logger.WithFields(map[string]interface{}{
|
||||||
"bind": saddr.String(),
|
"bind": saddr.String(),
|
||||||
})
|
})
|
||||||
h.logger.Infof("bind on %s OK", saddr.String())
|
h.logger.Debugf("bind on %s OK", &saddr)
|
||||||
|
|
||||||
if h.chain.IsEmpty() {
|
if h.chain.IsEmpty() {
|
||||||
// serve as standard socks5 udp relay.
|
// serve as standard socks5 udp relay.
|
||||||
@ -81,7 +89,7 @@ func (h *socks5Handler) handleUDP(ctx context.Context, conn net.Conn, req *gosoc
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (h *socks5Handler) getUDPTun(ctx context.Context) (conn net.Conn, err error) {
|
func (h *socks5Handler) getUDPTun(ctx context.Context) (conn net.Conn, err error) {
|
||||||
r := (&handler.Router{}).
|
r := (&chain.Router{}).
|
||||||
WithChain(h.chain).
|
WithChain(h.chain).
|
||||||
WithRetry(h.md.retryCount).
|
WithRetry(h.md.retryCount).
|
||||||
WithLogger(h.logger)
|
WithLogger(h.logger)
|
||||||
|
@ -6,6 +6,7 @@ import (
|
|||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/go-gost/gosocks5"
|
"github.com/go-gost/gosocks5"
|
||||||
|
"github.com/go-gost/gost/pkg/chain"
|
||||||
"github.com/go-gost/gost/pkg/handler"
|
"github.com/go-gost/gost/pkg/handler"
|
||||||
"github.com/go-gost/gost/pkg/internal/bufpool"
|
"github.com/go-gost/gost/pkg/internal/bufpool"
|
||||||
"github.com/go-gost/gost/pkg/internal/utils/socks"
|
"github.com/go-gost/gost/pkg/internal/utils/socks"
|
||||||
@ -16,6 +17,14 @@ func (h *socks5Handler) handleUDPTun(ctx context.Context, conn net.Conn, req *go
|
|||||||
"cmd": "udp-tun",
|
"cmd": "udp-tun",
|
||||||
})
|
})
|
||||||
|
|
||||||
|
if !h.md.enableUDP {
|
||||||
|
reply := gosocks5.NewReply(gosocks5.NotAllowed, nil)
|
||||||
|
reply.Write(conn)
|
||||||
|
h.logger.Debug(reply)
|
||||||
|
h.logger.Error("UDP relay is diabled")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
if h.chain.IsEmpty() {
|
if h.chain.IsEmpty() {
|
||||||
addr := req.Addr.String()
|
addr := req.Addr.String()
|
||||||
|
|
||||||
@ -56,7 +65,7 @@ func (h *socks5Handler) handleUDPTun(ctx context.Context, conn net.Conn, req *go
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
r := (&handler.Router{}).
|
r := (&chain.Router{}).
|
||||||
WithChain(h.chain).
|
WithChain(h.chain).
|
||||||
WithRetry(h.md.retryCount).
|
WithRetry(h.md.retryCount).
|
||||||
WithLogger(h.logger)
|
WithLogger(h.logger)
|
||||||
|
@ -123,7 +123,7 @@ func (h *ssHandler) Handle(ctx context.Context, conn net.Conn) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
r := (&handler.Router{}).
|
r := (&chain.Router{}).
|
||||||
WithChain(h.chain).
|
WithChain(h.chain).
|
||||||
WithRetry(h.md.retryCount).
|
WithRetry(h.md.retryCount).
|
||||||
WithLogger(h.logger)
|
WithLogger(h.logger)
|
||||||
|
@ -5,7 +5,7 @@ import (
|
|||||||
"net"
|
"net"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/go-gost/gost/pkg/handler"
|
"github.com/go-gost/gost/pkg/chain"
|
||||||
"github.com/go-gost/gost/pkg/internal/bufpool"
|
"github.com/go-gost/gost/pkg/internal/bufpool"
|
||||||
"github.com/go-gost/gost/pkg/internal/utils/socks"
|
"github.com/go-gost/gost/pkg/internal/utils/socks"
|
||||||
"github.com/go-gost/gost/pkg/internal/utils/ss"
|
"github.com/go-gost/gost/pkg/internal/utils/ss"
|
||||||
@ -17,7 +17,7 @@ func (h *ssHandler) handleUDP(ctx context.Context, raddr net.Addr, conn net.Pack
|
|||||||
}
|
}
|
||||||
|
|
||||||
// obtain a udp connection
|
// obtain a udp connection
|
||||||
r := (&handler.Router{}).
|
r := (&chain.Router{}).
|
||||||
WithChain(h.chain).
|
WithChain(h.chain).
|
||||||
WithRetry(h.md.retryCount).
|
WithRetry(h.md.retryCount).
|
||||||
WithLogger(h.logger)
|
WithLogger(h.logger)
|
||||||
@ -51,7 +51,7 @@ func (h *ssHandler) handleUDP(ctx context.Context, raddr net.Addr, conn net.Pack
|
|||||||
|
|
||||||
func (h *ssHandler) handleUDPTun(ctx context.Context, conn net.Conn) {
|
func (h *ssHandler) handleUDPTun(ctx context.Context, conn net.Conn) {
|
||||||
// obtain a udp connection
|
// obtain a udp connection
|
||||||
r := (&handler.Router{}).
|
r := (&chain.Router{}).
|
||||||
WithChain(h.chain).
|
WithChain(h.chain).
|
||||||
WithRetry(h.md.retryCount).
|
WithRetry(h.md.retryCount).
|
||||||
WithLogger(h.logger)
|
WithLogger(h.logger)
|
||||||
|
@ -6,70 +6,80 @@ import (
|
|||||||
smux "github.com/xtaci/smux"
|
smux "github.com/xtaci/smux"
|
||||||
)
|
)
|
||||||
|
|
||||||
type MuxSession struct {
|
type Session struct {
|
||||||
conn net.Conn
|
conn net.Conn
|
||||||
session *smux.Session
|
session *smux.Session
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewMuxSession(conn net.Conn) (*MuxSession, error) {
|
func ClientSession(conn net.Conn) (*Session, error) {
|
||||||
// Upgrade connection to multiplex stream.
|
|
||||||
s, err := smux.Client(conn, smux.DefaultConfig())
|
s, err := smux.Client(conn, smux.DefaultConfig())
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
return &MuxSession{
|
return &Session{
|
||||||
conn: conn,
|
conn: conn,
|
||||||
session: s,
|
session: s,
|
||||||
}, nil
|
}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (session *MuxSession) GetConn() (net.Conn, error) {
|
func ServerSession(conn net.Conn) (*Session, error) {
|
||||||
|
s, err := smux.Server(conn, smux.DefaultConfig())
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return &Session{
|
||||||
|
conn: conn,
|
||||||
|
session: s,
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (session *Session) GetConn() (net.Conn, error) {
|
||||||
stream, err := session.session.OpenStream()
|
stream, err := session.session.OpenStream()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
return &muxStreamConn{Conn: session.conn, stream: stream}, nil
|
return &StreamConn{Conn: session.conn, stream: stream}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (session *MuxSession) Accept() (net.Conn, error) {
|
func (session *Session) Accept() (net.Conn, error) {
|
||||||
stream, err := session.session.AcceptStream()
|
stream, err := session.session.AcceptStream()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
return &muxStreamConn{Conn: session.conn, stream: stream}, nil
|
return &StreamConn{Conn: session.conn, stream: stream}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (session *MuxSession) Close() error {
|
func (session *Session) Close() error {
|
||||||
if session.session == nil {
|
if session.session == nil {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
return session.session.Close()
|
return session.session.Close()
|
||||||
}
|
}
|
||||||
|
|
||||||
func (session *MuxSession) IsClosed() bool {
|
func (session *Session) IsClosed() bool {
|
||||||
if session.session == nil {
|
if session.session == nil {
|
||||||
return true
|
return true
|
||||||
}
|
}
|
||||||
return session.session.IsClosed()
|
return session.session.IsClosed()
|
||||||
}
|
}
|
||||||
|
|
||||||
func (session *MuxSession) NumStreams() int {
|
func (session *Session) NumStreams() int {
|
||||||
return session.session.NumStreams()
|
return session.session.NumStreams()
|
||||||
}
|
}
|
||||||
|
|
||||||
type muxStreamConn struct {
|
type StreamConn struct {
|
||||||
net.Conn
|
net.Conn
|
||||||
stream *smux.Stream
|
stream *smux.Stream
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *muxStreamConn) Read(b []byte) (n int, err error) {
|
func (c *StreamConn) Read(b []byte) (n int, err error) {
|
||||||
return c.stream.Read(b)
|
return c.stream.Read(b)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *muxStreamConn) Write(b []byte) (n int, err error) {
|
func (c *StreamConn) Write(b []byte) (n int, err error) {
|
||||||
return c.stream.Write(b)
|
return c.stream.Write(b)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *muxStreamConn) Close() error {
|
func (c *StreamConn) Close() error {
|
||||||
return c.stream.Close()
|
return c.stream.Close()
|
||||||
}
|
}
|
||||||
|
@ -48,6 +48,7 @@ func (l *kcpListener) Init(md md.Metadata) (err error) {
|
|||||||
config = DefaultConfig
|
config = DefaultConfig
|
||||||
}
|
}
|
||||||
config.Init()
|
config.Init()
|
||||||
|
l.md.config = config
|
||||||
|
|
||||||
var ln *kcp.Listener
|
var ln *kcp.Listener
|
||||||
|
|
||||||
@ -168,7 +169,3 @@ func (l *kcpListener) mux(conn net.Conn) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (l *kcpListener) parseMetadata(md md.Metadata) (err error) {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
@ -1,8 +1,6 @@
|
|||||||
package kcp
|
package kcp
|
||||||
|
|
||||||
const (
|
import md "github.com/go-gost/gost/pkg/metadata"
|
||||||
connQueueSize = "connQueueSize"
|
|
||||||
)
|
|
||||||
|
|
||||||
const (
|
const (
|
||||||
defaultQueueSize = 128
|
defaultQueueSize = 128
|
||||||
@ -13,3 +11,13 @@ type metadata struct {
|
|||||||
|
|
||||||
connQueueSize int
|
connQueueSize int
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (l *kcpListener) parseMetadata(md md.Metadata) (err error) {
|
||||||
|
const (
|
||||||
|
connQueueSize = "connQueueSize"
|
||||||
|
)
|
||||||
|
|
||||||
|
l.md.connQueueSize = md.GetInt(connQueueSize)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
@ -1,11 +1,13 @@
|
|||||||
package listener
|
package listener
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"github.com/go-gost/gost/pkg/chain"
|
||||||
"github.com/go-gost/gost/pkg/logger"
|
"github.com/go-gost/gost/pkg/logger"
|
||||||
)
|
)
|
||||||
|
|
||||||
type Options struct {
|
type Options struct {
|
||||||
Addr string
|
Addr string
|
||||||
|
Chain *chain.Chain
|
||||||
Logger logger.Logger
|
Logger logger.Logger
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -17,6 +19,12 @@ func AddrOption(addr string) Option {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func ChainOption(chain *chain.Chain) Option {
|
||||||
|
return func(opts *Options) {
|
||||||
|
opts.Chain = chain
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
func LoggerOption(logger logger.Logger) Option {
|
func LoggerOption(logger logger.Logger) Option {
|
||||||
return func(opts *Options) {
|
return func(opts *Options) {
|
||||||
opts.Logger = logger
|
opts.Logger = logger
|
||||||
|
17
pkg/listener/rtcp/conn.go
Normal file
17
pkg/listener/rtcp/conn.go
Normal file
@ -0,0 +1,17 @@
|
|||||||
|
package rtcp
|
||||||
|
|
||||||
|
import "net"
|
||||||
|
|
||||||
|
type peerConn struct {
|
||||||
|
net.Conn
|
||||||
|
localAddr net.Addr
|
||||||
|
remoteAddr net.Addr
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *peerConn) LocalAddr() net.Addr {
|
||||||
|
return c.localAddr
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *peerConn) RemoteAddr() net.Addr {
|
||||||
|
return c.remoteAddr
|
||||||
|
}
|
214
pkg/listener/rtcp/listener.go
Normal file
214
pkg/listener/rtcp/listener.go
Normal file
@ -0,0 +1,214 @@
|
|||||||
|
package rtcp
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"fmt"
|
||||||
|
"net"
|
||||||
|
"sync"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/go-gost/gosocks5"
|
||||||
|
"github.com/go-gost/gost/pkg/chain"
|
||||||
|
"github.com/go-gost/gost/pkg/internal/utils/mux"
|
||||||
|
"github.com/go-gost/gost/pkg/listener"
|
||||||
|
"github.com/go-gost/gost/pkg/logger"
|
||||||
|
md "github.com/go-gost/gost/pkg/metadata"
|
||||||
|
"github.com/go-gost/gost/pkg/registry"
|
||||||
|
)
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
registry.RegisterListener("rtcp", NewListener)
|
||||||
|
}
|
||||||
|
|
||||||
|
type rtcpListener struct {
|
||||||
|
addr string
|
||||||
|
laddr net.Addr
|
||||||
|
chain *chain.Chain
|
||||||
|
md metadata
|
||||||
|
ln net.Listener
|
||||||
|
connChan chan net.Conn
|
||||||
|
session *mux.Session
|
||||||
|
sessionMux sync.Mutex
|
||||||
|
logger logger.Logger
|
||||||
|
closed chan struct{}
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewListener(opts ...listener.Option) listener.Listener {
|
||||||
|
options := &listener.Options{}
|
||||||
|
for _, opt := range opts {
|
||||||
|
opt(options)
|
||||||
|
}
|
||||||
|
return &rtcpListener{
|
||||||
|
addr: options.Addr,
|
||||||
|
chain: options.Chain,
|
||||||
|
closed: make(chan struct{}),
|
||||||
|
logger: options.Logger,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (l *rtcpListener) Init(md md.Metadata) (err error) {
|
||||||
|
if err = l.parseMetadata(md); err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
laddr, err := net.ResolveTCPAddr("tcp", l.addr)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
l.laddr = laddr
|
||||||
|
l.connChan = make(chan net.Conn, l.md.connQueueSize)
|
||||||
|
|
||||||
|
if l.chain.IsEmpty() {
|
||||||
|
l.ln, err = net.ListenTCP("tcp", laddr)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
go l.listenLoop()
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func (l *rtcpListener) Addr() net.Addr {
|
||||||
|
return l.laddr
|
||||||
|
}
|
||||||
|
|
||||||
|
func (l *rtcpListener) Close() error {
|
||||||
|
if l.ln != nil {
|
||||||
|
return l.ln.Close()
|
||||||
|
}
|
||||||
|
|
||||||
|
select {
|
||||||
|
case <-l.closed:
|
||||||
|
default:
|
||||||
|
close(l.closed)
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (l *rtcpListener) Accept() (conn net.Conn, err error) {
|
||||||
|
if l.ln != nil {
|
||||||
|
return l.ln.Accept()
|
||||||
|
}
|
||||||
|
|
||||||
|
select {
|
||||||
|
case conn = <-l.connChan:
|
||||||
|
case <-l.closed:
|
||||||
|
err = net.ErrClosed
|
||||||
|
}
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func (l *rtcpListener) listenLoop() {
|
||||||
|
var tempDelay time.Duration
|
||||||
|
|
||||||
|
for {
|
||||||
|
select {
|
||||||
|
case <-l.closed:
|
||||||
|
return
|
||||||
|
default:
|
||||||
|
}
|
||||||
|
|
||||||
|
conn, err := l.accept()
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
if tempDelay == 0 {
|
||||||
|
tempDelay = 1000 * time.Millisecond
|
||||||
|
} else {
|
||||||
|
tempDelay *= 2
|
||||||
|
}
|
||||||
|
if max := 6 * time.Second; tempDelay > max {
|
||||||
|
tempDelay = max
|
||||||
|
}
|
||||||
|
l.logger.Warnf("accept: %v, retrying in %v", err, tempDelay)
|
||||||
|
time.Sleep(tempDelay)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
tempDelay = 0
|
||||||
|
|
||||||
|
select {
|
||||||
|
case l.connChan <- conn:
|
||||||
|
default:
|
||||||
|
conn.Close()
|
||||||
|
l.logger.Warnf("connection queue is full, client %s discarded", conn.RemoteAddr().String())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (l *rtcpListener) accept() (net.Conn, error) {
|
||||||
|
if l.md.enableMux {
|
||||||
|
return l.muxAccept()
|
||||||
|
}
|
||||||
|
|
||||||
|
r := (&chain.Router{}).
|
||||||
|
WithChain(l.chain).
|
||||||
|
WithRetry(l.md.retryCount).
|
||||||
|
WithLogger(l.logger)
|
||||||
|
cc, err := r.Connect(context.Background())
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
conn, err := l.waitPeer(cc)
|
||||||
|
if err != nil {
|
||||||
|
l.logger.Error(err)
|
||||||
|
cc.Close()
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
l.logger.Debugf("peer %s accepted", conn.RemoteAddr())
|
||||||
|
|
||||||
|
return conn, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (l *rtcpListener) waitPeer(conn net.Conn) (net.Conn, error) {
|
||||||
|
addr := gosocks5.Addr{}
|
||||||
|
addr.ParseFrom(l.addr)
|
||||||
|
req := gosocks5.NewRequest(gosocks5.CmdBind, &addr)
|
||||||
|
if err := req.Write(conn); err != nil {
|
||||||
|
l.logger.Error(err)
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// first reply, bind status
|
||||||
|
rep, err := gosocks5.ReadReply(conn)
|
||||||
|
if err != nil {
|
||||||
|
l.logger.Error(err)
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
l.logger.Debug(rep)
|
||||||
|
|
||||||
|
if rep.Rep != gosocks5.Succeeded {
|
||||||
|
err = fmt.Errorf("bind on %s failed", l.addr)
|
||||||
|
l.logger.Error(err)
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
l.logger.Debugf("bind on %s OK", rep.Addr)
|
||||||
|
|
||||||
|
// second reply, peer connected
|
||||||
|
rep, err = gosocks5.ReadReply(conn)
|
||||||
|
if err != nil {
|
||||||
|
l.logger.Error(err)
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if rep.Rep != gosocks5.Succeeded {
|
||||||
|
err = fmt.Errorf("peer connect failed")
|
||||||
|
l.logger.Error(err)
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
raddr, err := net.ResolveTCPAddr("tcp", rep.Addr.String())
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return &peerConn{
|
||||||
|
Conn: conn,
|
||||||
|
localAddr: l.laddr,
|
||||||
|
remoteAddr: raddr,
|
||||||
|
}, nil
|
||||||
|
}
|
35
pkg/listener/rtcp/metadata.go
Normal file
35
pkg/listener/rtcp/metadata.go
Normal file
@ -0,0 +1,35 @@
|
|||||||
|
package rtcp
|
||||||
|
|
||||||
|
import (
|
||||||
|
"time"
|
||||||
|
|
||||||
|
md "github.com/go-gost/gost/pkg/metadata"
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
defaultKeepAlivePeriod = 180 * time.Second
|
||||||
|
defaultConnQueueSize = 128
|
||||||
|
)
|
||||||
|
|
||||||
|
type metadata struct {
|
||||||
|
enableMux bool
|
||||||
|
connQueueSize int
|
||||||
|
retryCount int
|
||||||
|
}
|
||||||
|
|
||||||
|
func (l *rtcpListener) parseMetadata(md md.Metadata) (err error) {
|
||||||
|
const (
|
||||||
|
enableMux = "mux"
|
||||||
|
connQueueSize = "connQueueSize"
|
||||||
|
retryCount = "retry"
|
||||||
|
)
|
||||||
|
|
||||||
|
l.md.enableMux = md.GetBool(enableMux)
|
||||||
|
l.md.retryCount = md.GetInt(retryCount)
|
||||||
|
|
||||||
|
l.md.connQueueSize = md.GetInt(connQueueSize)
|
||||||
|
if l.md.connQueueSize <= 0 {
|
||||||
|
l.md.connQueueSize = defaultConnQueueSize
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
109
pkg/listener/rtcp/mux.go
Normal file
109
pkg/listener/rtcp/mux.go
Normal file
@ -0,0 +1,109 @@
|
|||||||
|
package rtcp
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"fmt"
|
||||||
|
"net"
|
||||||
|
|
||||||
|
"github.com/go-gost/gosocks5"
|
||||||
|
"github.com/go-gost/gost/pkg/chain"
|
||||||
|
"github.com/go-gost/gost/pkg/internal/utils/mux"
|
||||||
|
"github.com/go-gost/gost/pkg/internal/utils/socks"
|
||||||
|
)
|
||||||
|
|
||||||
|
func (l *rtcpListener) muxAccept() (net.Conn, error) {
|
||||||
|
session, err := l.getSession()
|
||||||
|
if err != nil {
|
||||||
|
l.logger.Error(err)
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
cc, err := session.Accept()
|
||||||
|
if err != nil {
|
||||||
|
session.Close()
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
conn, err := l.getPeerConn(cc)
|
||||||
|
if err != nil {
|
||||||
|
l.logger.Error(err)
|
||||||
|
cc.Close()
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
l.logger.Debugf("peer %s accepted", conn.RemoteAddr())
|
||||||
|
|
||||||
|
return conn, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (l *rtcpListener) getPeerConn(conn net.Conn) (net.Conn, error) {
|
||||||
|
// second reply, peer connected
|
||||||
|
rep, err := gosocks5.ReadReply(conn)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if rep.Rep != gosocks5.Succeeded {
|
||||||
|
err = fmt.Errorf("peer connect failed")
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
raddr, err := net.ResolveTCPAddr("tcp", rep.Addr.String())
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return &peerConn{
|
||||||
|
Conn: conn,
|
||||||
|
localAddr: l.laddr,
|
||||||
|
remoteAddr: raddr,
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (l *rtcpListener) getSession() (s *mux.Session, err error) {
|
||||||
|
l.sessionMux.Lock()
|
||||||
|
defer l.sessionMux.Unlock()
|
||||||
|
|
||||||
|
if l.session != nil && !l.session.IsClosed() {
|
||||||
|
return l.session, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
r := (&chain.Router{}).
|
||||||
|
WithChain(l.chain).
|
||||||
|
WithRetry(l.md.retryCount).
|
||||||
|
WithLogger(l.logger)
|
||||||
|
conn, err := r.Connect(context.Background())
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
l.session, err = l.initSession(conn)
|
||||||
|
if err != nil {
|
||||||
|
conn.Close()
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
return l.session, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (l *rtcpListener) initSession(conn net.Conn) (*mux.Session, error) {
|
||||||
|
addr := gosocks5.Addr{}
|
||||||
|
addr.ParseFrom(l.addr)
|
||||||
|
req := gosocks5.NewRequest(socks.CmdMuxBind, &addr)
|
||||||
|
if err := req.Write(conn); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// first reply, bind status
|
||||||
|
rep, err := gosocks5.ReadReply(conn)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
if rep.Rep != gosocks5.Succeeded {
|
||||||
|
err = fmt.Errorf("bind on %s failed", l.addr)
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
l.logger.Debugf("bind on %s OK", rep.Addr)
|
||||||
|
|
||||||
|
return mux.ServerSession(conn)
|
||||||
|
}
|
@ -57,10 +57,3 @@ func (l *tcpListener) Init(md md.Metadata) (err error) {
|
|||||||
l.Listener = ln
|
l.Listener = ln
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
func (l *tcpListener) parseMetadata(md md.Metadata) (err error) {
|
|
||||||
l.md.keepAlive = md.GetBool(keepAlive)
|
|
||||||
l.md.keepAlivePeriod = md.GetDuration(keepAlivePeriod)
|
|
||||||
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
@ -1,10 +1,9 @@
|
|||||||
package tcp
|
package tcp
|
||||||
|
|
||||||
import "time"
|
import (
|
||||||
|
"time"
|
||||||
|
|
||||||
const (
|
md "github.com/go-gost/gost/pkg/metadata"
|
||||||
keepAlive = "keepAlive"
|
|
||||||
keepAlivePeriod = "keepAlivePeriod"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
const (
|
const (
|
||||||
@ -15,3 +14,15 @@ type metadata struct {
|
|||||||
keepAlive bool
|
keepAlive bool
|
||||||
keepAlivePeriod time.Duration
|
keepAlivePeriod time.Duration
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (l *tcpListener) parseMetadata(md md.Metadata) (err error) {
|
||||||
|
const (
|
||||||
|
keepAlive = "keepAlive"
|
||||||
|
keepAlivePeriod = "keepAlivePeriod"
|
||||||
|
)
|
||||||
|
|
||||||
|
l.md.keepAlive = md.GetBool(keepAlive)
|
||||||
|
l.md.keepAlivePeriod = md.GetDuration(keepAlivePeriod)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
@ -20,7 +20,7 @@ type udpListener struct {
|
|||||||
conn net.PacketConn
|
conn net.PacketConn
|
||||||
connChan chan net.Conn
|
connChan chan net.Conn
|
||||||
errChan chan error
|
errChan chan error
|
||||||
closeChan chan struct{}
|
closed chan struct{}
|
||||||
connPool *connPool
|
connPool *connPool
|
||||||
logger logger.Logger
|
logger logger.Logger
|
||||||
}
|
}
|
||||||
@ -33,7 +33,7 @@ func NewListener(opts ...listener.Option) listener.Listener {
|
|||||||
return &udpListener{
|
return &udpListener{
|
||||||
addr: options.Addr,
|
addr: options.Addr,
|
||||||
errChan: make(chan error, 1),
|
errChan: make(chan error, 1),
|
||||||
closeChan: make(chan struct{}),
|
closed: make(chan struct{}),
|
||||||
logger: options.Logger,
|
logger: options.Logger,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -75,13 +75,14 @@ func (l *udpListener) Accept() (conn net.Conn, err error) {
|
|||||||
|
|
||||||
func (l *udpListener) Close() error {
|
func (l *udpListener) Close() error {
|
||||||
select {
|
select {
|
||||||
case <-l.closeChan:
|
case <-l.closed:
|
||||||
return nil
|
|
||||||
default:
|
default:
|
||||||
close(l.closeChan)
|
close(l.closed)
|
||||||
l.connPool.Close()
|
l.connPool.Close()
|
||||||
return l.conn.Close()
|
return l.conn.Close()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (l *udpListener) Addr() net.Addr {
|
func (l *udpListener) Addr() net.Addr {
|
||||||
@ -126,26 +127,3 @@ func (l *udpListener) getConn(addr net.Addr) *conn {
|
|||||||
}
|
}
|
||||||
return c
|
return c
|
||||||
}
|
}
|
||||||
|
|
||||||
func (l *udpListener) parseMetadata(md md.Metadata) (err error) {
|
|
||||||
l.md.ttl = md.GetDuration(ttl)
|
|
||||||
if l.md.ttl <= 0 {
|
|
||||||
l.md.ttl = defaultTTL
|
|
||||||
}
|
|
||||||
l.md.readBufferSize = md.GetInt(readBufferSize)
|
|
||||||
if l.md.readBufferSize <= 0 {
|
|
||||||
l.md.readBufferSize = defaultReadBufferSize
|
|
||||||
}
|
|
||||||
|
|
||||||
l.md.readQueueSize = md.GetInt(readQueueSize)
|
|
||||||
if l.md.readQueueSize <= 0 {
|
|
||||||
l.md.readQueueSize = defaultReadQueueSize
|
|
||||||
}
|
|
||||||
|
|
||||||
l.md.connQueueSize = md.GetInt(connQueueSize)
|
|
||||||
if l.md.connQueueSize <= 0 {
|
|
||||||
l.md.connQueueSize = defaultConnQueueSize
|
|
||||||
}
|
|
||||||
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
@ -1,6 +1,10 @@
|
|||||||
package udp
|
package udp
|
||||||
|
|
||||||
import "time"
|
import (
|
||||||
|
"time"
|
||||||
|
|
||||||
|
md "github.com/go-gost/gost/pkg/metadata"
|
||||||
|
)
|
||||||
|
|
||||||
const (
|
const (
|
||||||
defaultTTL = 60 * time.Second
|
defaultTTL = 60 * time.Second
|
||||||
@ -9,13 +13,6 @@ const (
|
|||||||
defaultConnQueueSize = 128
|
defaultConnQueueSize = 128
|
||||||
)
|
)
|
||||||
|
|
||||||
const (
|
|
||||||
ttl = "ttl"
|
|
||||||
readBufferSize = "readBufferSize"
|
|
||||||
readQueueSize = "readQueueSize"
|
|
||||||
connQueueSize = "connQueueSize"
|
|
||||||
)
|
|
||||||
|
|
||||||
type metadata struct {
|
type metadata struct {
|
||||||
ttl time.Duration
|
ttl time.Duration
|
||||||
|
|
||||||
@ -23,3 +20,33 @@ type metadata struct {
|
|||||||
readQueueSize int
|
readQueueSize int
|
||||||
connQueueSize int
|
connQueueSize int
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (l *udpListener) parseMetadata(md md.Metadata) (err error) {
|
||||||
|
const (
|
||||||
|
ttl = "ttl"
|
||||||
|
readBufferSize = "readBufferSize"
|
||||||
|
readQueueSize = "readQueueSize"
|
||||||
|
connQueueSize = "connQueueSize"
|
||||||
|
)
|
||||||
|
|
||||||
|
l.md.ttl = md.GetDuration(ttl)
|
||||||
|
if l.md.ttl <= 0 {
|
||||||
|
l.md.ttl = defaultTTL
|
||||||
|
}
|
||||||
|
l.md.readBufferSize = md.GetInt(readBufferSize)
|
||||||
|
if l.md.readBufferSize <= 0 {
|
||||||
|
l.md.readBufferSize = defaultReadBufferSize
|
||||||
|
}
|
||||||
|
|
||||||
|
l.md.readQueueSize = md.GetInt(readQueueSize)
|
||||||
|
if l.md.readQueueSize <= 0 {
|
||||||
|
l.md.readQueueSize = defaultReadQueueSize
|
||||||
|
}
|
||||||
|
|
||||||
|
l.md.connQueueSize = md.GetInt(connQueueSize)
|
||||||
|
if l.md.connQueueSize <= 0 {
|
||||||
|
l.md.connQueueSize = defaultConnQueueSize
|
||||||
|
}
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user