130 lines
2.8 KiB
Go
130 lines
2.8 KiB
Go
// SPDX-License-Identifier: BSD-3-Clause
|
|
|
|
package main
|
|
|
|
import (
|
|
"encoding/json"
|
|
"encoding/xml"
|
|
"errors"
|
|
"fmt"
|
|
"os"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
|
|
"git.froth.zone/sam/awl/cli"
|
|
"git.froth.zone/sam/awl/internal/helpers"
|
|
"git.froth.zone/sam/awl/query"
|
|
|
|
"gopkg.in/yaml.v2"
|
|
)
|
|
|
|
var version = "DEV"
|
|
|
|
func main() {
|
|
opts, err := cli.ParseCLI(version)
|
|
if err != nil {
|
|
// TODO: Make not ew
|
|
if errors.Is(err, cli.ErrNotError) || strings.Contains(err.Error(), "help requested") {
|
|
os.Exit(0)
|
|
}
|
|
opts.Logger.Error(err)
|
|
os.Exit(1)
|
|
}
|
|
var resp helpers.Response
|
|
|
|
// Retry queries if a query fails
|
|
for i := 0; i < opts.Request.Retries; i++ {
|
|
resp, err = query.CreateQuery(opts)
|
|
if err == nil {
|
|
break
|
|
} else {
|
|
opts.Logger.Warn("Retrying")
|
|
}
|
|
}
|
|
|
|
// Query failed, make it fail
|
|
if err != nil {
|
|
opts.Logger.Error(err)
|
|
os.Exit(9)
|
|
}
|
|
switch {
|
|
case opts.JSON:
|
|
opts.Logger.Info("Printing as JSON")
|
|
json, err := json.MarshalIndent(resp.DNS, "", " ")
|
|
if err != nil {
|
|
opts.Logger.Error(err)
|
|
os.Exit(10)
|
|
}
|
|
fmt.Println(string(json))
|
|
case opts.XML:
|
|
opts.Logger.Info("Printing as XML")
|
|
xml, err := xml.MarshalIndent(resp.DNS, "", " ")
|
|
if err != nil {
|
|
opts.Logger.Error(err)
|
|
os.Exit(10)
|
|
}
|
|
fmt.Println(string(xml))
|
|
case opts.YAML:
|
|
opts.Logger.Info("Printing as YAML")
|
|
yaml, err := yaml.Marshal(resp.DNS)
|
|
if err != nil {
|
|
opts.Logger.Error(err)
|
|
os.Exit(10)
|
|
}
|
|
fmt.Println(string(yaml))
|
|
default:
|
|
if !opts.Short {
|
|
// Print everything
|
|
|
|
if !opts.Display.Question {
|
|
resp.DNS.Question = nil
|
|
opts.Logger.Info("Disabled question display")
|
|
}
|
|
if !opts.Display.Answer {
|
|
resp.DNS.Answer = nil
|
|
opts.Logger.Info("Disabled answer display")
|
|
}
|
|
if !opts.Display.Authority {
|
|
resp.DNS.Ns = nil
|
|
opts.Logger.Info("Disabled authority display")
|
|
}
|
|
if !opts.Display.Additional {
|
|
resp.DNS.Extra = nil
|
|
opts.Logger.Info("Disabled additional display")
|
|
}
|
|
|
|
fmt.Println(resp.DNS)
|
|
|
|
if opts.Display.Statistics {
|
|
fmt.Println(";; Query time:", resp.RTT)
|
|
|
|
// 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)"
|
|
}
|
|
|
|
fmt.Println(";; SERVER:", opts.Request.Server+extra)
|
|
fmt.Println(";; WHEN:", time.Now().Format(time.RFC1123Z))
|
|
fmt.Println(";; MSG SIZE rcvd:", resp.DNS.Len())
|
|
}
|
|
|
|
} else {
|
|
// Print just the responses, nothing else
|
|
for _, res := range resp.DNS.Answer {
|
|
temp := strings.Split(res.String(), "\t")
|
|
fmt.Println(temp[len(temp)-1])
|
|
}
|
|
}
|
|
}
|
|
}
|