mirror of
https://github.com/SagerNet/sing-box.git
synced 2024-09-20 10:37:51 +00:00
54 lines
1.3 KiB
Go
54 lines
1.3 KiB
Go
package sniff
|
|
|
|
import (
|
|
"context"
|
|
"encoding/binary"
|
|
"io"
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/sagernet/sing-box/adapter"
|
|
C "github.com/sagernet/sing-box/constant"
|
|
"github.com/sagernet/sing/common"
|
|
"github.com/sagernet/sing/common/buf"
|
|
"github.com/sagernet/sing/common/task"
|
|
|
|
mDNS "github.com/miekg/dns"
|
|
)
|
|
|
|
func StreamDomainNameQuery(readCtx context.Context, reader io.Reader) (*adapter.InboundContext, error) {
|
|
var length uint16
|
|
err := binary.Read(reader, binary.BigEndian, &length)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if length == 0 {
|
|
return nil, os.ErrInvalid
|
|
}
|
|
_buffer := buf.StackNewSize(int(length))
|
|
defer common.KeepAlive(_buffer)
|
|
buffer := common.Dup(_buffer)
|
|
defer buffer.Release()
|
|
|
|
readCtx, cancel := context.WithTimeout(readCtx, time.Millisecond*100)
|
|
var readTask task.Group
|
|
readTask.Append0(func(ctx context.Context) error {
|
|
return common.Error(buffer.ReadFullFrom(reader, buffer.FreeLen()))
|
|
})
|
|
err = readTask.Run(readCtx)
|
|
cancel()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return DomainNameQuery(readCtx, buffer.Bytes())
|
|
}
|
|
|
|
func DomainNameQuery(ctx context.Context, packet []byte) (*adapter.InboundContext, error) {
|
|
var msg mDNS.Msg
|
|
err := msg.Unpack(packet)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &adapter.InboundContext{Protocol: C.ProtocolDNS}, nil
|
|
}
|