mirror of
https://github.com/shtorm-7/sing-box-extended.git
synced 2026-06-21 02:12:11 +03:00
32f91a4faa
Return SUCCESS with empty answers instead of an error when the queried address family has no range configured. Reject configurations where neither inet4_range nor inet6_range is set.
80 lines
2.3 KiB
Go
80 lines
2.3 KiB
Go
package fakeip
|
|
|
|
import (
|
|
"context"
|
|
"net/netip"
|
|
|
|
"github.com/sagernet/sing-box/adapter"
|
|
C "github.com/sagernet/sing-box/constant"
|
|
"github.com/sagernet/sing-box/dns"
|
|
"github.com/sagernet/sing-box/log"
|
|
"github.com/sagernet/sing-box/option"
|
|
E "github.com/sagernet/sing/common/exceptions"
|
|
"github.com/sagernet/sing/common/logger"
|
|
|
|
mDNS "github.com/miekg/dns"
|
|
)
|
|
|
|
func RegisterTransport(registry *dns.TransportRegistry) {
|
|
dns.RegisterTransport[option.FakeIPDNSServerOptions](registry, C.DNSTypeFakeIP, NewTransport)
|
|
}
|
|
|
|
var _ adapter.FakeIPTransport = (*Transport)(nil)
|
|
|
|
type Transport struct {
|
|
dns.TransportAdapter
|
|
logger logger.ContextLogger
|
|
store adapter.FakeIPStore
|
|
inet4Enabled bool
|
|
inet6Enabled bool
|
|
}
|
|
|
|
func NewTransport(ctx context.Context, logger log.ContextLogger, tag string, options option.FakeIPDNSServerOptions) (adapter.DNSTransport, error) {
|
|
inet4Range := options.Inet4Range.Build(netip.Prefix{})
|
|
inet6Range := options.Inet6Range.Build(netip.Prefix{})
|
|
if !inet4Range.IsValid() && !inet6Range.IsValid() {
|
|
return nil, E.New("at least one of inet4_range or inet6_range must be set")
|
|
}
|
|
store := NewStore(ctx, logger, inet4Range, inet6Range)
|
|
return &Transport{
|
|
TransportAdapter: dns.NewTransportAdapter(C.DNSTypeFakeIP, tag, nil),
|
|
logger: logger,
|
|
store: store,
|
|
inet4Enabled: inet4Range.IsValid(),
|
|
inet6Enabled: inet6Range.IsValid(),
|
|
}, nil
|
|
}
|
|
|
|
func (t *Transport) Start(stage adapter.StartStage) error {
|
|
if stage != adapter.StartStateStart {
|
|
return nil
|
|
}
|
|
return t.store.Start()
|
|
}
|
|
|
|
func (t *Transport) Close() error {
|
|
return t.store.Close()
|
|
}
|
|
|
|
func (t *Transport) Reset() {
|
|
}
|
|
|
|
func (t *Transport) Exchange(ctx context.Context, message *mDNS.Msg) (*mDNS.Msg, error) {
|
|
question := message.Question[0]
|
|
if question.Qtype != mDNS.TypeA && question.Qtype != mDNS.TypeAAAA {
|
|
return nil, E.New("only IP queries are supported by fakeip")
|
|
}
|
|
if question.Qtype == mDNS.TypeA && !t.inet4Enabled || question.Qtype == mDNS.TypeAAAA && !t.inet6Enabled {
|
|
return dns.FixedResponseStatus(message, mDNS.RcodeSuccess), nil
|
|
}
|
|
address, err := t.store.Create(dns.FqdnToDomain(question.Name), question.Qtype == mDNS.TypeAAAA)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return dns.FixedResponse(message.Id, question, []netip.Addr{address}, C.DefaultDNSTTL), nil
|
|
}
|
|
|
|
func (t *Transport) Store() adapter.FakeIPStore {
|
|
return t.store
|
|
}
|