76 lines
2.2 KiB
Go
76 lines
2.2 KiB
Go
package edns
|
|
|
|
import (
|
|
"io"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"golang.org/x/net/dns/dnsmessage"
|
|
)
|
|
|
|
func TestExchangeDoHGETAndPOST(t *testing.T) {
|
|
server := httptest.NewTLSServer(http.HandlerFunc(func(writer http.ResponseWriter, request *http.Request) {
|
|
var payload []byte
|
|
var err error
|
|
if request.Method == http.MethodGet {
|
|
payload, err = decodeGETQuery(request.URL.Query().Get("dns"))
|
|
} else {
|
|
payload, err = io.ReadAll(request.Body)
|
|
}
|
|
if err != nil {
|
|
http.Error(writer, err.Error(), http.StatusBadRequest)
|
|
return
|
|
}
|
|
if request.Header.Get("Accept") != "application/dns-message" {
|
|
http.Error(writer, "missing accept", http.StatusNotAcceptable)
|
|
return
|
|
}
|
|
|
|
var query dnsmessage.Message
|
|
if err := query.Unpack(payload); err != nil {
|
|
http.Error(writer, err.Error(), http.StatusBadRequest)
|
|
return
|
|
}
|
|
response := dnsmessage.Message{
|
|
Header: dnsmessage.Header{ID: query.Header.ID, Response: true, RecursionAvailable: true},
|
|
Questions: query.Questions,
|
|
}
|
|
responseWire, err := response.Pack()
|
|
if err != nil {
|
|
http.Error(writer, err.Error(), http.StatusInternalServerError)
|
|
return
|
|
}
|
|
writer.Header().Set("Content-Type", "application/dns-message")
|
|
_, _ = writer.Write(responseWire)
|
|
}))
|
|
defer server.Close()
|
|
|
|
wire, _, _, err := BuildQuery("example.com", "A")
|
|
if err != nil {
|
|
t.Fatalf("build query: %v", err)
|
|
}
|
|
for _, method := range []string{"get", "post"} {
|
|
t.Run(method, func(t *testing.T) {
|
|
response, info, err := exchangeDoHWithClient(t.Context(), server.Client(), server.URL, wire, method)
|
|
if err != nil {
|
|
t.Fatalf("exchange DoH: %v", err)
|
|
}
|
|
if len(response) == 0 || !info.Encrypted || !info.ServerAuthenticated {
|
|
t.Fatalf("unexpected result: response=%d info=%#v", len(response), info)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestExchangeDoHRejectsHTTPError(t *testing.T) {
|
|
server := httptest.NewTLSServer(http.HandlerFunc(func(writer http.ResponseWriter, _ *http.Request) {
|
|
http.Error(writer, "unavailable", http.StatusServiceUnavailable)
|
|
}))
|
|
defer server.Close()
|
|
|
|
if _, _, err := exchangeDoHWithClient(t.Context(), server.Client(), server.URL, []byte{1}, "post"); err == nil {
|
|
t.Fatal("HTTP error was accepted")
|
|
}
|
|
}
|