sing-box/adapter/inbound.go

114 lines
2.7 KiB
Go
Raw Normal View History

2022-06-30 13:27:56 +00:00
package adapter
2022-07-01 11:34:02 +00:00
import (
2022-07-07 13:47:21 +00:00
"context"
2022-08-29 11:43:13 +00:00
"net"
2022-07-07 15:36:32 +00:00
"net/netip"
2022-07-07 13:47:21 +00:00
"github.com/sagernet/sing-box/common/process"
2022-10-07 12:30:27 +00:00
"github.com/sagernet/sing-box/option"
2022-07-08 15:03:57 +00:00
M "github.com/sagernet/sing/common/metadata"
2022-08-29 11:43:13 +00:00
N "github.com/sagernet/sing/common/network"
2022-07-01 11:34:02 +00:00
)
2022-06-30 13:27:56 +00:00
type Inbound interface {
Service
Type() string
Tag() string
}
2022-07-01 11:34:02 +00:00
2022-08-29 11:43:13 +00:00
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
}
2022-07-01 11:34:02 +00:00
type InboundContext struct {
Inbound string
2022-07-19 14:16:49 +00:00
InboundType string
2023-06-07 12:28:21 +00:00
IPVersion uint8
2022-07-01 11:34:02 +00:00
Network string
2022-07-02 06:07:50 +00:00
Source M.Socksaddr
Destination M.Socksaddr
2022-07-01 11:34:02 +00:00
Domain string
2022-07-02 06:07:50 +00:00
Protocol string
2022-07-17 07:11:26 +00:00
User string
2022-07-07 13:47:21 +00:00
Outbound string
2022-07-02 14:55:10 +00:00
// cache
2022-10-07 12:30:27 +00:00
InboundDetour string
LastInbound string
OriginDestination M.Socksaddr
InboundOptions option.InboundOptions
DestinationAddresses []netip.Addr
SourceGeoIPCode string
GeoIPCode string
ProcessInfo *process.Info
2023-12-01 05:24:12 +00:00
QueryType uint16
2023-06-10 08:26:40 +00:00
FakeIP bool
2023-02-08 08:18:40 +00:00
2023-12-01 05:24:12 +00:00
// rule cache
2023-02-08 08:18:40 +00:00
2024-06-24 01:41:00 +00:00
IPCIDRMatchSource bool
IPCIDRAcceptEmpty bool
SourceAddressMatch bool
SourcePortMatch bool
DestinationAddressMatch bool
DestinationPortMatch bool
DidMatch bool
IgnoreDestinationIPCIDRMatch bool
2023-12-01 05:24:12 +00:00
}
func (c *InboundContext) ResetRuleCache() {
c.IPCIDRMatchSource = false
2024-06-24 01:41:00 +00:00
c.IPCIDRAcceptEmpty = false
2023-12-01 05:24:12 +00:00
c.SourceAddressMatch = false
c.SourcePortMatch = false
c.DestinationAddressMatch = false
c.DestinationPortMatch = false
c.DidMatch = false
2022-07-01 11:34:02 +00:00
}
2022-07-07 13:47:21 +00:00
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)
2022-07-07 15:36:32 +00:00
return WithContext(ctx, metadata), metadata
2022-07-07 13:47:21 +00:00
}
2023-10-25 04:00:00 +00:00
func ExtendContext(ctx context.Context) (context.Context, *InboundContext) {
var newMetadata InboundContext
if metadata := ContextFrom(ctx); metadata != nil {
newMetadata = *metadata
}
return WithContext(ctx, &newMetadata), &newMetadata
}
2024-04-29 03:55:25 +00:00
func OverrideContext(ctx context.Context) context.Context {
if metadata := ContextFrom(ctx); metadata != nil {
var newMetadata InboundContext
newMetadata = *metadata
return WithContext(ctx, &newMetadata)
}
return ctx
}