Add dns transports

dns/transport_test.go

@@ -0,0 +1,88 @@
+package dns
+
+import (
+	"context"
+	"testing"
+	"time"
+
+	M "github.com/sagernet/sing/common/metadata"
+	N "github.com/sagernet/sing/common/network"
+
+	C "github.com/sagernet/sing-box/constant"
+	"github.com/sagernet/sing-box/log"
+
+	"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
+}