feat: custom response format

pkg/resolvers/classic.go

@@ -98,22 +98,26 @@ func NewResolverFromResolvFile(resolvFilePath string) (Resolver, error) {
 // Lookup prepare a list of DNS messages to be sent to the server.
 // It's possible to send multiple question in one message
 // but some nameservers are not able to
-func (c *ClassicResolver) Lookup(questions []dns.Question) error {
-	messages := prepareMessages(questions)
+func (c *ClassicResolver) Lookup(questions []dns.Question) ([]Response, error) {
+	var (
+		messages  = prepareMessages(questions)
+		responses []Response
+	)
 
 	for _, msg := range messages {
 		for _, srv := range c.servers {
 			in, rtt, err := c.client.Exchange(&msg, srv)
 			if err != nil {
-				return err
+				return nil, err
 			}
-			for _, ans := range in.Answer {
-				if t, ok := ans.(*dns.A); ok {
-					fmt.Println(t.String())
-				}
+			msg.Answer = in.Answer
+			rsp := Response{
+				Message:    msg,
+				RTT:        rtt,
+				Nameserver: srv,
 			}
-			fmt.Println("rtt is", rtt)
+			responses = append(responses, rsp)
 		}
 	}
-	return nil
+	return responses, nil
 }

pkg/resolvers/doh.go

@@ -2,7 +2,6 @@ package resolvers
 
 import (
 	"bytes"
-	"fmt"
 	"io/ioutil"
 	"net/http"
 	"time"
@@ -27,39 +26,46 @@ func NewDOHResolver(servers []string) (Resolver, error) {
 	}, nil
 }
 
-func (r *DOHResolver) Lookup(questions []dns.Question) error {
-	messages := prepareMessages(questions)
+func (r *DOHResolver) Lookup(questions []dns.Question) ([]Response, error) {
+	var (
+		messages  = prepareMessages(questions)
+		responses []Response
+	)
 
-	for _, m := range messages {
-		b, err := m.Pack()
+	for _, msg := range messages {
+		// get the DNS Message in wire format.
+		b, err := msg.Pack()
 		if err != nil {
-			return err
+			return nil, err
 		}
 		for _, srv := range r.servers {
+			now := time.Now()
+			// Make an HTTP POST request to the DNS server with the DNS message as wire format bytes in the body.
 			resp, err := r.client.Post(srv, "application/dns-message", bytes.NewBuffer(b))
 			if err != nil {
-				return err
+				return nil, err
 			}
 			if resp.StatusCode != http.StatusOK {
-				return err
+				return nil, err
 			}
-
+			rtt := time.Since(now)
+			// extract the binary response in DNS Message.
 			body, err := ioutil.ReadAll(resp.Body)
 			if err != nil {
-				return err
+				return nil, err
 			}
 
-			r := &dns.Msg{}
-			err = r.Unpack(body)
+			err = msg.Unpack(body)
 			if err != nil {
-				return err
+				return nil, err
 			}
-			for _, ans := range r.Answer {
-				if t, ok := ans.(*dns.A); ok {
-					fmt.Println(t.String())
-				}
+			rsp := Response{
+				Message:    msg,
+				RTT:        rtt,
+				Nameserver: srv,
 			}
+			responses = append(responses, rsp)
 		}
 	}
-	return nil
+	return responses, nil
 }

pkg/resolvers/resolver.go

@@ -1,10 +1,23 @@
 package resolvers
 
-import "github.com/miekg/dns"
+import (
+	"time"
+
+	"github.com/miekg/dns"
+)
 
 // Resolver implements the configuration for a DNS
 // Client. Different types of client like (UDP/TCP/DOH/DOT)
 // can be initialised.
 type Resolver interface {
-	Lookup([]dns.Question) error
+	Lookup([]dns.Question) ([]Response, error)
+}
+
+// Response represents a custom output format
+// which wraps certain metadata about the DNS query
+// and the DNS Answer as well.
+type Response struct {
+	Message    dns.Msg
+	RTT        time.Duration
+	Nameserver string
 }