mirror of
https://github.com/SagerNet/sing-box.git
synced 2024-12-25 02:25:39 +08:00
122 lines
3.3 KiB
Go
122 lines
3.3 KiB
Go
package adapter
|
|
|
|
import (
|
|
"context"
|
|
"net"
|
|
"net/http"
|
|
"net/netip"
|
|
"sync"
|
|
|
|
"github.com/sagernet/sing-box/common/geoip"
|
|
C "github.com/sagernet/sing-box/constant"
|
|
"github.com/sagernet/sing-dns"
|
|
M "github.com/sagernet/sing/common/metadata"
|
|
N "github.com/sagernet/sing/common/network"
|
|
"github.com/sagernet/sing/common/x/list"
|
|
|
|
mdns "github.com/miekg/dns"
|
|
"go4.org/netipx"
|
|
)
|
|
|
|
type Router interface {
|
|
Lifecycle
|
|
|
|
FakeIPStore() FakeIPStore
|
|
|
|
ConnectionRouter
|
|
PreMatch(metadata InboundContext) error
|
|
ConnectionRouterEx
|
|
|
|
GeoIPReader() *geoip.Reader
|
|
LoadGeosite(code string) (Rule, error)
|
|
RuleSet(tag string) (RuleSet, bool)
|
|
NeedWIFIState() bool
|
|
|
|
Exchange(ctx context.Context, message *mdns.Msg) (*mdns.Msg, error)
|
|
Lookup(ctx context.Context, domain string, strategy dns.DomainStrategy) ([]netip.Addr, error)
|
|
LookupDefault(ctx context.Context, domain string) ([]netip.Addr, error)
|
|
ClearDNSCache()
|
|
Rules() []Rule
|
|
|
|
SetTracker(tracker ConnectionTracker)
|
|
|
|
ResetNetwork()
|
|
}
|
|
|
|
type ConnectionTracker interface {
|
|
RoutedConnection(ctx context.Context, conn net.Conn, metadata InboundContext, matchedRule Rule, matchOutbound Outbound) net.Conn
|
|
RoutedPacketConnection(ctx context.Context, conn N.PacketConn, metadata InboundContext, matchedRule Rule, matchOutbound Outbound) N.PacketConn
|
|
}
|
|
|
|
// Deprecated: Use ConnectionRouterEx instead.
|
|
type ConnectionRouter interface {
|
|
RouteConnection(ctx context.Context, conn net.Conn, metadata InboundContext) error
|
|
RoutePacketConnection(ctx context.Context, conn N.PacketConn, metadata InboundContext) error
|
|
}
|
|
|
|
type ConnectionRouterEx interface {
|
|
ConnectionRouter
|
|
RouteConnectionEx(ctx context.Context, conn net.Conn, metadata InboundContext, onClose N.CloseHandlerFunc)
|
|
RoutePacketConnectionEx(ctx context.Context, conn N.PacketConn, metadata InboundContext, onClose N.CloseHandlerFunc)
|
|
}
|
|
|
|
type RuleSet interface {
|
|
Name() string
|
|
StartContext(ctx context.Context, startContext *HTTPStartContext) error
|
|
PostStart() error
|
|
Metadata() RuleSetMetadata
|
|
ExtractIPSet() []*netipx.IPSet
|
|
IncRef()
|
|
DecRef()
|
|
Cleanup()
|
|
RegisterCallback(callback RuleSetUpdateCallback) *list.Element[RuleSetUpdateCallback]
|
|
UnregisterCallback(element *list.Element[RuleSetUpdateCallback])
|
|
Close() error
|
|
HeadlessRule
|
|
}
|
|
|
|
type RuleSetUpdateCallback func(it RuleSet)
|
|
|
|
type RuleSetMetadata struct {
|
|
ContainsProcessRule bool
|
|
ContainsWIFIRule bool
|
|
ContainsIPCIDRRule bool
|
|
}
|
|
type HTTPStartContext struct {
|
|
access sync.Mutex
|
|
httpClientCache map[string]*http.Client
|
|
}
|
|
|
|
func NewHTTPStartContext() *HTTPStartContext {
|
|
return &HTTPStartContext{
|
|
httpClientCache: make(map[string]*http.Client),
|
|
}
|
|
}
|
|
|
|
func (c *HTTPStartContext) HTTPClient(detour string, dialer N.Dialer) *http.Client {
|
|
c.access.Lock()
|
|
defer c.access.Unlock()
|
|
if httpClient, loaded := c.httpClientCache[detour]; loaded {
|
|
return httpClient
|
|
}
|
|
httpClient := &http.Client{
|
|
Transport: &http.Transport{
|
|
ForceAttemptHTTP2: true,
|
|
TLSHandshakeTimeout: C.TCPTimeout,
|
|
DialContext: func(ctx context.Context, network, addr string) (net.Conn, error) {
|
|
return dialer.DialContext(ctx, network, M.ParseSocksaddr(addr))
|
|
},
|
|
},
|
|
}
|
|
c.httpClientCache[detour] = httpClient
|
|
return httpClient
|
|
}
|
|
|
|
func (c *HTTPStartContext) Close() {
|
|
c.access.Lock()
|
|
defer c.access.Unlock()
|
|
for _, client := range c.httpClientCache {
|
|
client.CloseIdleConnections()
|
|
}
|
|
}
|