Sam
07728cffdb
All checks were successful
continuous-integration/drone/push Build is passing
Reviewed-on: #55
73 lines
1.7 KiB
Go
73 lines
1.7 KiB
Go
// SPDX-License-Identifier: BSD-3-Clause
|
|
|
|
package query
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"io"
|
|
"net/http"
|
|
"time"
|
|
|
|
"git.froth.zone/sam/awl/cli"
|
|
"git.froth.zone/sam/awl/internal/helpers"
|
|
"github.com/miekg/dns"
|
|
)
|
|
|
|
type HTTPSResolver struct {
|
|
opts cli.Options
|
|
}
|
|
|
|
// LookUp performs a DNS query.
|
|
func (r *HTTPSResolver) LookUp(msg *dns.Msg) (helpers.Response, error) {
|
|
var resp helpers.Response
|
|
httpR := &http.Client{
|
|
Timeout: r.opts.Request.Timeout,
|
|
}
|
|
buf, err := msg.Pack()
|
|
if err != nil {
|
|
return helpers.Response{}, fmt.Errorf("doh: packing error error: %w", err)
|
|
}
|
|
r.opts.Logger.Debug("making DoH request")
|
|
//
|
|
req, err := http.NewRequest("POST", r.opts.Request.Server, bytes.NewBuffer(buf))
|
|
if err != nil {
|
|
return helpers.Response{}, fmt.Errorf("doh request creation error: %w", err)
|
|
}
|
|
req.Header.Set("Content-Type", "application/dns-message")
|
|
req.Header.Set("Accept", "application/dns-message")
|
|
|
|
now := time.Now()
|
|
res, err := httpR.Do(req)
|
|
resp.RTT = time.Since(now)
|
|
|
|
if err != nil {
|
|
return helpers.Response{}, fmt.Errorf("doh HTTP request error: %w", err)
|
|
}
|
|
defer res.Body.Close()
|
|
|
|
if res.StatusCode != http.StatusOK {
|
|
return helpers.Response{}, &errHTTPStatus{res.StatusCode}
|
|
}
|
|
|
|
fullRes, err := io.ReadAll(res.Body)
|
|
if err != nil {
|
|
return helpers.Response{}, fmt.Errorf("doh body read error: %w", err)
|
|
}
|
|
resp.DNS = &dns.Msg{}
|
|
r.opts.Logger.Debug("unpacking response")
|
|
err = resp.DNS.Unpack(fullRes)
|
|
if err != nil {
|
|
return helpers.Response{}, fmt.Errorf("doh dns message unpack error: %w", err)
|
|
}
|
|
|
|
return resp, nil
|
|
}
|
|
|
|
type errHTTPStatus struct {
|
|
code int
|
|
}
|
|
|
|
func (e *errHTTPStatus) Error() string {
|
|
return fmt.Sprintf("doh server responded with HTTP %d", e.code)
|
|
}
|