mirror of
https://github.com/SagerNet/sing-box.git
synced 2024-12-01 19:06:43 +00:00
88 lines
2.6 KiB
Go
88 lines
2.6 KiB
Go
package dns
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
C "github.com/sagernet/sing-box/constant"
|
|
"github.com/sagernet/sing-box/log"
|
|
M "github.com/sagernet/sing/common/metadata"
|
|
N "github.com/sagernet/sing/common/network"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
"golang.org/x/net/dns/dnsmessage"
|
|
)
|
|
|
|
func TestTCPDNS(t *testing.T) {
|
|
ctx, cancel := context.WithTimeout(context.Background(), time.Second*5)
|
|
transport := NewTCPTransport(ctx, N.SystemDialer, log.NewNopLogger(), M.ParseSocksaddr("1.0.0.1:53"))
|
|
response, err := transport.Exchange(ctx, makeQuery())
|
|
cancel()
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, response.Answers, "no answers")
|
|
for _, answer := range response.Answers {
|
|
t.Log(answer)
|
|
}
|
|
}
|
|
|
|
func TestTLSDNS(t *testing.T) {
|
|
ctx, cancel := context.WithTimeout(context.Background(), time.Second*5)
|
|
transport := NewTLSTransport(ctx, N.SystemDialer, log.NewNopLogger(), M.ParseSocksaddr("1.0.0.1:853"))
|
|
response, err := transport.Exchange(ctx, makeQuery())
|
|
cancel()
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, response.Answers, "no answers")
|
|
for _, answer := range response.Answers {
|
|
t.Log(answer)
|
|
}
|
|
}
|
|
|
|
func TestHTTPSDNS(t *testing.T) {
|
|
ctx, cancel := context.WithTimeout(context.Background(), time.Second*5)
|
|
transport := NewHTTPSTransport(N.SystemDialer, "https://1.0.0.1:443/dns-query")
|
|
response, err := transport.Exchange(ctx, makeQuery())
|
|
cancel()
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, response.Answers, "no answers")
|
|
for _, answer := range response.Answers {
|
|
t.Log(answer)
|
|
}
|
|
}
|
|
|
|
func TestUDPDNS(t *testing.T) {
|
|
ctx, cancel := context.WithTimeout(context.Background(), time.Second*5)
|
|
transport := NewUDPTransport(ctx, N.SystemDialer, log.NewNopLogger(), M.ParseSocksaddr("1.0.0.1:53"))
|
|
response, err := transport.Exchange(ctx, makeQuery())
|
|
cancel()
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, response.Answers, "no answers")
|
|
for _, answer := range response.Answers {
|
|
t.Log(answer)
|
|
}
|
|
}
|
|
|
|
func TestLocalDNS(t *testing.T) {
|
|
ctx, cancel := context.WithTimeout(context.Background(), time.Second*5)
|
|
transport := NewLocalTransport()
|
|
response, err := transport.Lookup(ctx, "google.com", C.DomainStrategyAsIS)
|
|
cancel()
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, response, "no answers")
|
|
for _, answer := range response {
|
|
t.Log(answer)
|
|
}
|
|
}
|
|
|
|
func makeQuery() *dnsmessage.Message {
|
|
message := &dnsmessage.Message{}
|
|
message.Header.ID = 1
|
|
message.Header.RecursionDesired = true
|
|
message.Questions = append(message.Questions, dnsmessage.Question{
|
|
Name: dnsmessage.MustNewName("google.com."),
|
|
Type: dnsmessage.TypeA,
|
|
Class: dnsmessage.ClassINET,
|
|
})
|
|
return message
|
|
}
|