111 lines
2.6 KiB
Go
111 lines
2.6 KiB
Go
package adapter
|
|
|
|
import (
|
|
"context"
|
|
"net"
|
|
"net/netip"
|
|
|
|
"github.com/sagernet/sing-box/common/process"
|
|
"github.com/sagernet/sing-box/option"
|
|
M "github.com/sagernet/sing/common/metadata"
|
|
N "github.com/sagernet/sing/common/network"
|
|
)
|
|
|
|
type Inbound interface {
|
|
Service
|
|
Type() string
|
|
Tag() string
|
|
}
|
|
|
|
type InjectableInbound interface {
|
|
Inbound
|
|
Network() []string
|
|
NewConnection(ctx context.Context, conn net.Conn, metadata InboundContext) error
|
|
NewPacketConnection(ctx context.Context, conn N.PacketConn, metadata InboundContext) error
|
|
}
|
|
|
|
type InboundContext struct {
|
|
Inbound string
|
|
InboundType string
|
|
IPVersion uint8
|
|
Network string
|
|
Source M.Socksaddr
|
|
Destination M.Socksaddr
|
|
Domain string
|
|
Protocol string
|
|
User string
|
|
Outbound string
|
|
|
|
// cache
|
|
|
|
InboundDetour string
|
|
LastInbound string
|
|
OriginDestination M.Socksaddr
|
|
InboundOptions option.InboundOptions
|
|
DestinationAddresses []netip.Addr
|
|
SourceGeoIPCode string
|
|
GeoIPCode string
|
|
ProcessInfo *process.Info
|
|
QueryType uint16
|
|
FakeIP bool
|
|
|
|
// rule cache
|
|
|
|
IPCIDRMatchSource bool
|
|
SourceAddressMatch bool
|
|
SourcePortMatch bool
|
|
DestinationAddressMatch bool
|
|
DestinationPortMatch bool
|
|
DidMatch bool
|
|
IgnoreDestinationIPCIDRMatch bool
|
|
}
|
|
|
|
func (c *InboundContext) ResetRuleCache() {
|
|
c.IPCIDRMatchSource = false
|
|
c.SourceAddressMatch = false
|
|
c.SourcePortMatch = false
|
|
c.DestinationAddressMatch = false
|
|
c.DestinationPortMatch = false
|
|
c.DidMatch = false
|
|
}
|
|
|
|
type inboundContextKey struct{}
|
|
|
|
func WithContext(ctx context.Context, inboundContext *InboundContext) context.Context {
|
|
return context.WithValue(ctx, (*inboundContextKey)(nil), inboundContext)
|
|
}
|
|
|
|
func ContextFrom(ctx context.Context) *InboundContext {
|
|
metadata := ctx.Value((*inboundContextKey)(nil))
|
|
if metadata == nil {
|
|
return nil
|
|
}
|
|
return metadata.(*InboundContext)
|
|
}
|
|
|
|
func AppendContext(ctx context.Context) (context.Context, *InboundContext) {
|
|
metadata := ContextFrom(ctx)
|
|
if metadata != nil {
|
|
return ctx, metadata
|
|
}
|
|
metadata = new(InboundContext)
|
|
return WithContext(ctx, metadata), metadata
|
|
}
|
|
|
|
func ExtendContext(ctx context.Context) (context.Context, *InboundContext) {
|
|
var newMetadata InboundContext
|
|
if metadata := ContextFrom(ctx); metadata != nil {
|
|
newMetadata = *metadata
|
|
}
|
|
return WithContext(ctx, &newMetadata), &newMetadata
|
|
}
|
|
|
|
func OverrideContext(ctx context.Context) context.Context {
|
|
if metadata := ContextFrom(ctx); metadata != nil {
|
|
var newMetadata InboundContext
|
|
newMetadata = *metadata
|
|
return WithContext(ctx, &newMetadata)
|
|
}
|
|
return ctx
|
|
}
|