130 lines
3.3 KiB
Go
130 lines
3.3 KiB
Go
package main
|
|
|
|
import (
|
|
"os"
|
|
|
|
"github.com/urfave/cli/v2"
|
|
)
|
|
|
|
var (
|
|
// Version and date of the build. This is injected at build-time.
|
|
buildVersion = "unknown"
|
|
buildDate = "unknown"
|
|
)
|
|
|
|
func main() {
|
|
var (
|
|
logger = initLogger()
|
|
app = cli.NewApp()
|
|
)
|
|
// Initialize hub.
|
|
hub := NewHub(logger, buildVersion)
|
|
|
|
// Configure CLI app.
|
|
app.Name = "doggo"
|
|
app.Usage = "Command-line DNS Client"
|
|
app.Version = buildVersion
|
|
|
|
// Register command line flags.
|
|
app.Flags = []cli.Flag{
|
|
&cli.StringSliceFlag{
|
|
Name: "query",
|
|
Usage: "Domain name to query",
|
|
Destination: hub.QueryFlags.QNames,
|
|
},
|
|
&cli.StringSliceFlag{
|
|
Name: "type",
|
|
Usage: "Type of DNS record to be queried (A, AAAA, MX etc)",
|
|
Destination: hub.QueryFlags.QTypes,
|
|
},
|
|
&cli.StringSliceFlag{
|
|
Name: "nameserver",
|
|
Usage: "Address of the nameserver to send packets to",
|
|
Destination: hub.QueryFlags.Nameservers,
|
|
},
|
|
&cli.StringSliceFlag{
|
|
Name: "class",
|
|
Usage: "Network class of the DNS record to be queried (IN, CH, HS etc)",
|
|
Destination: hub.QueryFlags.QClasses,
|
|
},
|
|
&cli.BoolFlag{
|
|
Name: "udp",
|
|
Usage: "Use the DNS protocol over UDP",
|
|
Aliases: []string{"U"},
|
|
},
|
|
&cli.BoolFlag{
|
|
Name: "tcp",
|
|
Usage: "Use the DNS protocol over TCP",
|
|
Aliases: []string{"T"},
|
|
Destination: &hub.QueryFlags.UseTCP,
|
|
},
|
|
&cli.BoolFlag{
|
|
Name: "https",
|
|
Usage: "Use the DNS-over-HTTPS protocol",
|
|
Aliases: []string{"H"},
|
|
Destination: &hub.QueryFlags.IsDOH,
|
|
},
|
|
&cli.BoolFlag{
|
|
Name: "tls",
|
|
Usage: "Use the DNS-over-TLS",
|
|
Aliases: []string{"S"},
|
|
Destination: &hub.QueryFlags.IsDOT,
|
|
},
|
|
&cli.BoolFlag{
|
|
Name: "ipv6",
|
|
Aliases: []string{"6"},
|
|
Usage: "Use IPv6 only",
|
|
Destination: &hub.QueryFlags.UseIPv6,
|
|
},
|
|
&cli.BoolFlag{
|
|
Name: "ipv4",
|
|
Aliases: []string{"4"},
|
|
Usage: "Use IPv4 only",
|
|
Destination: &hub.QueryFlags.UseIPv4,
|
|
},
|
|
&cli.BoolFlag{
|
|
Name: "time",
|
|
Usage: "Display how long it took for the response to arrive",
|
|
Destination: &hub.QueryFlags.DisplayTimeTaken,
|
|
},
|
|
&cli.BoolFlag{
|
|
Name: "search",
|
|
Usage: "Use the search list provided in resolv.conf. It sets the `ndots` parameter as well unless overriden by `ndots` flag.",
|
|
Destination: &hub.QueryFlags.UseSearchList,
|
|
},
|
|
&cli.IntFlag{
|
|
Name: "ndots",
|
|
Usage: "Specify the ndots paramter",
|
|
DefaultText: "Default value is that set in `/etc/resolv.conf` or 1 if no `ndots` statement is present.",
|
|
Destination: &hub.QueryFlags.Ndots,
|
|
},
|
|
&cli.BoolFlag{
|
|
Name: "json",
|
|
Aliases: []string{"J"},
|
|
Usage: "Set the output format as JSON",
|
|
Destination: &hub.QueryFlags.ShowJSON,
|
|
},
|
|
&cli.BoolFlag{
|
|
Name: "debug",
|
|
Usage: "Enable verbose logging",
|
|
Destination: &hub.QueryFlags.Verbose,
|
|
DefaultText: "false",
|
|
},
|
|
}
|
|
app.Before = hub.loadQueryArgs
|
|
app.Action = func(c *cli.Context) error {
|
|
if len(hub.QueryFlags.QNames.Value()) == 0 {
|
|
cli.ShowAppHelpAndExit(c, 0)
|
|
}
|
|
hub.Lookup(c)
|
|
return nil
|
|
}
|
|
|
|
// Run the app.
|
|
hub.Logger.Debug("Starting doggo...")
|
|
err := app.Run(os.Args)
|
|
if err != nil {
|
|
logger.Errorf("oops! we encountered an issue: %s", err)
|
|
}
|
|
}
|