sing-box/outbound/dns.go

150 lines
3.8 KiB
Go
Raw Normal View History

2022-07-23 01:15:47 +00:00
package outbound
2022-07-10 01:15:01 +00:00
import (
"context"
"encoding/binary"
"net"
2022-07-23 01:15:47 +00:00
"os"
2022-07-10 01:15:01 +00:00
"github.com/sagernet/sing-box/adapter"
2022-07-15 00:42:02 +00:00
C "github.com/sagernet/sing-box/constant"
2022-08-26 05:13:44 +00:00
"github.com/sagernet/sing-dns"
2022-07-10 01:15:01 +00:00
"github.com/sagernet/sing/common"
"github.com/sagernet/sing/common/buf"
2023-03-23 07:14:21 +00:00
"github.com/sagernet/sing/common/canceler"
2022-07-23 01:15:47 +00:00
M "github.com/sagernet/sing/common/metadata"
2022-07-10 01:15:01 +00:00
N "github.com/sagernet/sing/common/network"
2022-07-24 13:25:41 +00:00
"github.com/sagernet/sing/common/task"
2022-07-10 01:15:01 +00:00
2022-09-13 08:18:39 +00:00
mDNS "github.com/miekg/dns"
2022-07-10 01:15:01 +00:00
)
2022-07-23 01:15:47 +00:00
var _ adapter.Outbound = (*DNS)(nil)
2022-07-15 00:42:02 +00:00
type DNS struct {
2022-07-23 01:15:47 +00:00
myOutboundAdapter
2022-07-15 00:42:02 +00:00
}
2022-08-03 10:55:39 +00:00
func NewDNS(router adapter.Router, tag string) *DNS {
2022-07-23 01:15:47 +00:00
return &DNS{
myOutboundAdapter{
protocol: C.TypeDNS,
2022-07-29 16:29:22 +00:00
network: []string{N.NetworkTCP, N.NetworkUDP},
2022-07-23 01:15:47 +00:00
router: router,
tag: tag,
2022-07-15 00:42:02 +00:00
},
}
}
2022-07-23 01:15:47 +00:00
func (d *DNS) DialContext(ctx context.Context, network string, destination M.Socksaddr) (net.Conn, error) {
return nil, os.ErrInvalid
2022-07-15 00:42:02 +00:00
}
2022-07-23 01:15:47 +00:00
func (d *DNS) ListenPacket(ctx context.Context, destination M.Socksaddr) (net.PacketConn, error) {
return nil, os.ErrInvalid
2022-07-15 00:42:02 +00:00
}
2022-07-23 01:15:47 +00:00
func (d *DNS) NewConnection(ctx context.Context, conn net.Conn, metadata adapter.InboundContext) error {
2022-07-24 13:25:41 +00:00
defer conn.Close()
2022-07-11 04:56:57 +00:00
ctx = adapter.WithContext(ctx, &metadata)
2022-07-10 01:15:01 +00:00
for {
2022-08-26 05:13:44 +00:00
err := d.handleConnection(ctx, conn, metadata)
2022-07-10 01:15:01 +00:00
if err != nil {
return err
}
2022-08-26 05:13:44 +00:00
}
}
func (d *DNS) handleConnection(ctx context.Context, conn net.Conn, metadata adapter.InboundContext) error {
var queryLength uint16
err := binary.Read(conn, binary.BigEndian, &queryLength)
if err != nil {
return err
}
if queryLength == 0 {
return dns.RCodeFormatError
}
_buffer := buf.StackNewSize(int(queryLength))
defer common.KeepAlive(_buffer)
buffer := common.Dup(_buffer)
defer buffer.Release()
_, err = buffer.ReadFullFrom(conn, int(queryLength))
if err != nil {
return err
}
2022-09-13 08:18:39 +00:00
var message mDNS.Msg
2022-08-26 05:13:44 +00:00
err = message.Unpack(buffer.Bytes())
if err != nil {
return err
}
2022-09-04 04:39:43 +00:00
metadataInQuery := metadata
2022-08-26 05:13:44 +00:00
go func() error {
2022-09-04 04:39:43 +00:00
response, err := d.router.Exchange(adapter.WithContext(ctx, &metadataInQuery), &message)
2022-07-10 01:15:01 +00:00
if err != nil {
return err
}
2022-08-26 05:13:44 +00:00
_responseBuffer := buf.StackNewPacket()
defer common.KeepAlive(_responseBuffer)
responseBuffer := common.Dup(_responseBuffer)
defer responseBuffer.Release()
responseBuffer.Resize(2, 0)
2022-09-13 08:18:39 +00:00
n, err := response.PackBuffer(responseBuffer.FreeBytes())
2022-07-10 01:15:01 +00:00
if err != nil {
return err
}
2022-08-26 05:13:44 +00:00
responseBuffer.Truncate(len(n))
binary.BigEndian.PutUint16(responseBuffer.ExtendHeader(2), uint16(len(n)))
_, err = conn.Write(responseBuffer.Bytes())
return err
}()
return nil
2022-07-10 01:15:01 +00:00
}
2022-07-23 01:15:47 +00:00
func (d *DNS) NewPacketConnection(ctx context.Context, conn N.PacketConn, metadata adapter.InboundContext) error {
2022-07-11 04:56:57 +00:00
ctx = adapter.WithContext(ctx, &metadata)
2022-07-24 13:25:41 +00:00
fastClose, cancel := context.WithCancel(ctx)
2022-07-25 14:02:39 +00:00
timeout := canceler.New(fastClose, cancel, C.DNSTimeout)
2022-08-03 09:11:53 +00:00
var group task.Group
group.Append0(func(ctx context.Context) error {
2022-07-25 14:02:39 +00:00
defer cancel()
2022-08-26 05:13:44 +00:00
_buffer := buf.StackNewSize(dns.FixedPacketSize)
2022-07-25 14:02:39 +00:00
defer common.KeepAlive(_buffer)
buffer := common.Dup(_buffer)
defer buffer.Release()
2022-07-24 13:25:41 +00:00
for {
buffer.FullReset()
destination, err := conn.ReadPacket(buffer)
2022-07-10 01:15:01 +00:00
if err != nil {
return err
}
2022-09-13 08:18:39 +00:00
var message mDNS.Msg
2022-07-24 13:25:41 +00:00
err = message.Unpack(buffer.Bytes())
2022-07-10 01:15:01 +00:00
if err != nil {
return err
}
2022-07-25 14:02:39 +00:00
timeout.Update()
2022-09-04 04:39:43 +00:00
metadataInQuery := metadata
2022-07-24 13:25:41 +00:00
go func() error {
2022-09-04 04:39:43 +00:00
response, err := d.router.Exchange(adapter.WithContext(ctx, &metadataInQuery), &message)
2022-07-24 13:25:41 +00:00
if err != nil {
return err
}
2022-07-25 14:02:39 +00:00
timeout.Update()
2022-08-02 05:43:25 +00:00
responseBuffer := buf.NewPacket()
2022-09-13 08:18:39 +00:00
n, err := response.PackBuffer(responseBuffer.FreeBytes())
2022-07-24 13:25:41 +00:00
if err != nil {
2022-08-02 05:43:25 +00:00
responseBuffer.Release()
2022-07-24 13:25:41 +00:00
return err
}
responseBuffer.Truncate(len(n))
err = conn.WritePacket(responseBuffer, destination)
return err
}()
}
})
2022-08-03 09:11:53 +00:00
group.Cleanup(func() {
conn.Close()
})
2022-08-12 04:13:57 +00:00
return group.Run(fastClose)
2022-07-10 01:15:01 +00:00
}