awl/query/struct.go

// SPDX-License-Identifier: BSD-3-Clause

package query

import (
	"strconv"
	"strings"
	"time"

	"git.froth.zone/sam/awl/cli"
	"git.froth.zone/sam/awl/internal/helpers"

	"github.com/miekg/dns"
)

type Message struct {
	Header   dns.MsgHdr `json:",omitempty" xml:",omitempty" yaml:",omitempty"`
	Question []Question `json:",omitempty" xml:",omitempty" yaml:",omitempty"`
	Answer   []Answer   `json:",omitempty" xml:",omitempty" yaml:",omitempty"`
	Ns       []Answer   `json:",omitempty" xml:",omitempty" yaml:",omitempty"`
	Extra    []Answer   `json:",omitempty" xml:",omitempty" yaml:",omitempty"`
}

type Question struct {
	Name  string `json:",omitempty" xml:",omitempty" yaml:",omitempty"`
	Type  string `json:",omitempty" xml:",omitempty" yaml:",omitempty"`
	Class string `json:",omitempty" xml:",omitempty" yaml:",omitempty"`
}

type RRHeader struct {
	Name     string `json:",omitempty" xml:",omitempty" yaml:",omitempty"`
	Type     string `json:",omitempty" xml:",omitempty" yaml:",omitempty"`
	Class    string `json:",omitempty" xml:",omitempty" yaml:",omitempty"`
	TTL      string `json:",omitempty" xml:",omitempty" yaml:",omitempty"`
	Rdlength uint16 `json:"-" xml:"-" yaml:"-"`
}

type Answer struct {
	RRHeader `json:"Header,omitempty" xml:"Header,omitempty" yaml:"header,omitempty"`
	Value    string `json:"Response,omitempty" xml:"Response,omitempty" yaml:"response,omitempty"`
}

// Turn the response into dig
//
// Much of this is taken from https://github.com/miekg/dns/blob/master/msg.go#L900
func ToString(res helpers.Response, opts cli.Options) string {

	if res.DNS == nil {
		return "<nil> MsgHdr"
	}
	var s string
	var opt *dns.OPT

	if !opts.Short {

		if opts.Display.Comments {
			s += res.DNS.MsgHdr.String() + " "
			s += "QUERY: " + strconv.Itoa(len(res.DNS.Question)) + ", "
			s += "ANSWER: " + strconv.Itoa(len(res.DNS.Answer)) + ", "
			s += "AUTHORITY: " + strconv.Itoa(len(res.DNS.Ns)) + ", "
			s += "ADDITIONAL: " + strconv.Itoa(len(res.DNS.Extra)) + "\n"
			opt = res.DNS.IsEdns0()
			if opt != nil && opts.Display.Opt {
				// OPT PSEUDOSECTION
				s += opt.String() + "\n"
			}
		}
		if opts.Display.Question {
			if len(res.DNS.Question) > 0 {
				if opts.Display.Comments {
					s += "\n;; QUESTION SECTION:\n"
				}
				for _, r := range res.DNS.Question {
					s += r.String() + "\n"
				}
			}
		}
		if opts.Display.Answer {
			if len(res.DNS.Answer) > 0 {
				if opts.Display.Comments {
					s += "\n;; ANSWER SECTION:\n"
				}
				for _, r := range res.DNS.Answer {
					if r != nil {
						s += r.String() + "\n"
					}
				}
			}
		}
		if opts.Display.Authority {
			if len(res.DNS.Ns) > 0 {
				if opts.Display.Comments {
					s += "\n;; AUTHORITY SECTION:\n"
				}
				for _, r := range res.DNS.Ns {
					if r != nil {
						s += r.String() + "\n"
					}
				}
			}
		}
		if opts.Display.Additional {
			if len(res.DNS.Extra) > 0 && (opt == nil || len(res.DNS.Extra) > 1) {
				if opts.Display.Comments {
					s += "\n;; ADDITIONAL SECTION:\n"
				}
				for _, r := range res.DNS.Extra {
					if r != nil && r.Header().Rrtype != dns.TypeOPT {
						s += r.String() + "\n"
					}
				}
			}
		}
		if opts.Display.Statistics {
			s += "\n;; Query time: " + res.RTT.String()
			// Add extra information to server string
			var extra string
			switch {
			case opts.TCP:
				extra = ":" + strconv.Itoa(opts.Port) + " (TCP)"
			case opts.TLS:
				extra = ":" + strconv.Itoa(opts.Port) + " (TLS)"
			case opts.HTTPS, opts.DNSCrypt:
				extra = ""
			case opts.QUIC:
				extra = ":" + strconv.Itoa(opts.Port) + " (QUIC)"
			default:
				extra = ":" + strconv.Itoa(opts.Port) + " (UDP)"
			}

			s += "\n;; SERVER: " + opts.Request.Server + extra
			s += "\n;; WHEN: " + time.Now().Format(time.RFC1123Z)
			s += "\n;; MSG SIZE  rcvd: " + strconv.Itoa(res.DNS.Len()) + "\n"
		}
	} else {
		// Print just the responses, nothing else
		for _, res := range res.DNS.Answer {
			temp := strings.Split(res.String(), "\t")
			s += temp[len(temp)-1]
		}
		if opts.Identify {
			s += " from server " + opts.Request.Server + " in " + res.RTT.String()
		}
		// s += "\n"
	}

	return s
}