feat: add protocol tweaks to dns.Message

pkg/models/models.go

@@ -39,6 +39,7 @@ type QueryFlags struct {
 	Strategy           string        `koanf:"strategy" strategy:"-"`
 	InsecureSkipVerify bool          `koanf:"skip-hostname-verification" skip-hostname-verification:"-"`
 	TLSHostname        string        `koanf:"tls-hostname" tls-hostname:"-"`
+	RetryCount         int           `koanf:"retry" retry:"-"`
 }
 
 // Nameserver represents the type of Nameserver

pkg/resolvers/classic.go

@@ -63,14 +63,14 @@ func NewClassicResolver(server string, classicOpts ClassicResolverOpts, resolver
 func (r *ClassicResolver) Lookup(question dns.Question) (Response, error) {
 	var (
 		rsp      Response
-		messages = prepareMessages(question, r.resolverOptions.Ndots, r.resolverOptions.SearchList)
+		messages = prepareMessages(question, r.resolverOptions)
 	)
 	for _, msg := range messages {
 		r.resolverOptions.Logger.WithFields(logf.Fields{
 			"domain":     msg.Question[0].Name,
 			"ndots":      r.resolverOptions.Ndots,
 			"nameserver": r.server,
-		}).Debug("Attempting to resolve")
+		}).Debug("attempting to resolve")
 
 		// Since the library doesn't include tcp.Dial time,
 		// it's better to not rely on `rtt` provided here and calculate it ourselves.

pkg/resolvers/dnscrypt.go

@@ -46,14 +46,14 @@ func NewDNSCryptResolver(server string, dnscryptOpts DNSCryptResolverOpts, resol
 func (r *DNSCryptResolver) Lookup(question dns.Question) (Response, error) {
 	var (
 		rsp      Response
-		messages = prepareMessages(question, r.resolverOptions.Ndots, r.resolverOptions.SearchList)
+		messages = prepareMessages(question, r.resolverOptions)
 	)
 	for _, msg := range messages {
 		r.resolverOptions.Logger.WithFields(logf.Fields{
 			"domain":     msg.Question[0].Name,
 			"ndots":      r.resolverOptions.Ndots,
 			"nameserver": r.server,
-		}).Debug("Attempting to resolve")
+		}).Debug("attempting to resolve")
 		now := time.Now()
 		in, err := r.client.Exchange(&msg, r.resolverInfo)
 		if err != nil {

pkg/resolvers/doh.go

@@ -45,7 +45,7 @@ func NewDOHResolver(server string, resolverOpts Options) (Resolver, error) {
 func (r *DOHResolver) Lookup(question dns.Question) (Response, error) {
 	var (
 		rsp      Response
-		messages = prepareMessages(question, r.resolverOptions.Ndots, r.resolverOptions.SearchList)
+		messages = prepareMessages(question, r.resolverOptions)
 	)
 
 	for _, msg := range messages {
@@ -53,7 +53,7 @@ func (r *DOHResolver) Lookup(question dns.Question) (Response, error) {
 			"domain":     msg.Question[0].Name,
 			"ndots":      r.resolverOptions.Ndots,
 			"nameserver": r.server,
-		}).Debug("Attempting to resolve")
+		}).Debug("attempting to resolve")
 		// get the DNS Message in wire format.
 		b, err := msg.Pack()
 		if err != nil {

pkg/resolvers/doq.go

@@ -37,7 +37,7 @@ func NewDOQResolver(server string, resolverOpts Options) (Resolver, error) {
 func (r *DOQResolver) Lookup(question dns.Question) (Response, error) {
 	var (
 		rsp      Response
-		messages = prepareMessages(question, r.resolverOptions.Ndots, r.resolverOptions.SearchList)
+		messages = prepareMessages(question, r.resolverOptions)
 	)
 
 	session, err := quic.DialAddr(r.server, r.tls, nil)
@@ -51,7 +51,7 @@ func (r *DOQResolver) Lookup(question dns.Question) (Response, error) {
 			"domain":     msg.Question[0].Name,
 			"ndots":      r.resolverOptions.Ndots,
 			"nameserver": r.server,
-		}).Debug("Attempting to resolve")
+		}).Debug("attempting to resolve")
 
 		// ref: https://www.rfc-editor.org/rfc/rfc9250.html#name-dns-message-ids
 		msg.Id = 0

pkg/resolvers/resolver.go

@@ -22,6 +22,12 @@ type Options struct {
 	Strategy           string
 	InsecureSkipVerify bool
 	TLSHostname        string
+
+	// DNS Protocol Flags.
+	Authoritative     bool
+	AuthenticatedData bool
+	CheckingDisabled  bool
+	RecursionDesired  bool
 }
 
 // Resolver implements the configuration for a DNS

pkg/resolvers/utils.go

@@ -11,17 +11,22 @@ import (
 
 // prepareMessages takes a  DNS Question and returns the
 // corresponding DNS messages for the same.
-func prepareMessages(q dns.Question, ndots int, searchList []string) []dns.Msg {
+func prepareMessages(q dns.Question, opts Options) []dns.Msg {
 	var (
-		possibleQNames = constructPossibleQuestions(q.Name, ndots, searchList)
+		possibleQNames = constructPossibleQuestions(q.Name, opts.Ndots, opts.SearchList)
 		messages       = make([]dns.Msg, 0, len(possibleQNames))
 	)
 
 	for _, qName := range possibleQNames {
-		msg := dns.Msg{}
-		// generate a random id for the transaction.
-		msg.Id = dns.Id()
-		msg.RecursionDesired = true
+		msg := dns.Msg{
+			MsgHdr: dns.MsgHdr{
+				Id:                dns.Id(),
+				Authoritative:     opts.Authoritative,
+				AuthenticatedData: opts.AuthenticatedData,
+				CheckingDisabled:  opts.CheckingDisabled,
+				RecursionDesired:  opts.RecursionDesired,
+			},
+		}
 		// It's recommended to only send 1 question for 1 DNS message.
 		msg.Question = []dns.Question{{
 			Name:   qName,