doggo/cmd/lookup.go

package main

import (
	"errors"
	"strings"

	"github.com/miekg/dns"
	"github.com/mr-karan/doggo/pkg/resolvers"
	"github.com/sirupsen/logrus"
)

// Lookup sends the DNS queries to the server.
func (hub *Hub) Lookup() error {
	err := hub.prepareQuestions()
	if err != nil {
		return err
	}
	// for each type of resolver do a DNS lookup
	responses := make([][]resolvers.Response, 0, len(hub.Questions))
	for _, r := range hub.Resolver {
		resp, err := r.Lookup(hub.Questions)
		if err != nil {
			return err
		}
		responses = append(responses, resp)
	}
	if len(responses) == 0 {
		return errors.New(`no DNS records found`)
	}
	hub.Output(responses)
	return nil
}

// prepareQuestions iterates on list of domain names
// and prepare a list of questions
// sent to the server with all possible combinations.
func (hub *Hub) prepareQuestions() error {
	var (
		question dns.Question
	)
	for _, name := range hub.QueryFlags.QNames {
		var (
			domains []string
			ndots   int
		)
		ndots = hub.QueryFlags.Ndots
		// If `search` flag is specified then fetch the search list
		// from `resolv.conf` and set the
		if hub.QueryFlags.UseSearchList {
			list, n, err := fetchDomainList(name, ndots)
			if err != nil {
				return err
			}
			domains = list
			ndots = n
		} else {
			domains = []string{dns.Fqdn(name)}
		}
		for _, d := range domains {
			hub.Logger.WithFields(logrus.Fields{
				"domain": d,
				"ndots":  ndots,
			}).Debug("Attmepting to resolve")
			question.Name = d
			// iterate on a list of query types.
			for _, q := range hub.QueryFlags.QTypes {
				question.Qtype = dns.StringToType[strings.ToUpper(q)]
				// iterate on a list of query classes.
				for _, c := range hub.QueryFlags.QClasses {
					question.Qclass = dns.StringToClass[strings.ToUpper(c)]
					// append a new question for each possible pair.
					hub.Questions = append(hub.Questions, question)
				}
			}
		}
	}
	return nil
}

func fetchDomainList(d string, ndots int) ([]string, int, error) {
	cfg, err := dns.ClientConfigFromFile(resolvers.DefaultResolvConfPath)
	if err != nil {
		return nil, 0, err
	}
	// if it's the default value
	if cfg.Ndots == 1 {
		// override what the user gave. If the user didn't give any setting then it's 1 by default.
		cfg.Ndots = ndots
	}
	return cfg.NameList(d), cfg.Ndots, nil
}