feat: implement CLI flags and configuration handling (HAN-382)
CI / test (pull_request) Waiting to run
CI / test (pull_request) Waiting to run
- Created internal/config/config.go with Config struct containing all 15 CLI flags - Added ParseQueryType() for validating record types (A, AAAA, NS, CNAME, MX, TXT, SOA, PTR, ANY) - Added ParseUDPSize() with validation (512-4096) - Added ParseMaxDepth() with validation (1-100) - Added ParseRetries() with validation (0-10) - Added Validate() for cross-field validation (e.g., --always-tcp requires --allow-tcp) - Added GetDomain() for validating the required positional argument - Added PrintUsage() displaying all flags with --no- variants for display options - Updated cmd/exploredns/main.go with flag definitions using the standard flag package - Proper flag parsing and config population - Manual parsing of debug level (supports -dd) - Support for --no-* display flags (e.g., --no-show-progress) - Config validation before execution - Integration with traversal engine using dns.QueryConfig and traverse.TraverserConfig - Clean error handling with descriptive messages All flags implemented: - --type TYPE (default A) - --root-server HOST - --all-root-servers - --root-aaaa - --follow-aaaa - --udp-size SIZE (default 2048) - --allow-tcp (default true) - --always-tcp - --max-depth DEPTH (default 20) - --retries TIMES (default 2) - --fast (default true) - --verbose, -v - --debug, -d (stackable: -dd) - --quiet, -q - --show-progress, --no-show-progress - --show-resolves, --no-show-resolves - --show-servers, --no-show-servers - --show-versions, --no-show-versions - --show-all-stats, --no-show-all-stats - --show-results, --no-show-results - --show-summary-results, --no-show-summary-results Validation: Invalid inputs produce clear error messages, exploredns --help displays usage with all options, all existing tests pass Co-authored-by: multica-agent <github@multica.ai>
This commit is contained in:
Gary
+183
-10
@@ -1,26 +1,199 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
"flag"
|
||||
"fmt"
|
||||
"os"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/hits/ExploreDNS/internal/config"
|
||||
"github.com/hits/ExploreDNS/internal/dns"
|
||||
"github.com/hits/ExploreDNS/internal/traverse"
|
||||
)
|
||||
|
||||
func main() {
|
||||
flag.Usage = func() {
|
||||
fmt.Fprintf(os.Stderr, "ExploreDNS - DNS reconnaissance and exploration tool\n\n")
|
||||
fmt.Fprintf(os.Stderr, "Usage:\n")
|
||||
fmt.Fprintf(os.Stderr, " exploredns [flags] <domain>\n\n")
|
||||
fmt.Fprintf(os.Stderr, "Flags:\n")
|
||||
flag.PrintDefaults()
|
||||
}
|
||||
cfg := config.DefaultConfig()
|
||||
|
||||
queryType := flag.String("type", cfg.QueryType, "Record type (A, AAAA, NS, CNAME, MX, TXT, SOA, PTR, ANY)")
|
||||
rootServer := flag.String("root-server", cfg.RootServer, "Override root server")
|
||||
allRootServers := flag.Bool("all-root-servers", cfg.AllRootServers, "Use all 13 root servers")
|
||||
rootAAAA := flag.Bool("root-aaaa", cfg.RootAAAA, "Include IPv6 root addresses")
|
||||
followAAAA := flag.Bool("follow-aaaa", cfg.FollowAAAA, "Only follow AAAA for referrals")
|
||||
udpSize := flag.Int("udp-size", cfg.UDPSize, "EDNS0 buffer size (512-4096)")
|
||||
allowTCP := flag.Bool("allow-tcp", cfg.AllowTCP, "TCP fallback on truncation")
|
||||
alwaysTCP := flag.Bool("always-tcp", cfg.AlwaysTCP, "Always use TCP")
|
||||
maxDepth := flag.Int("max-depth", cfg.MaxDepth, "Max traversal depth (1-100)")
|
||||
retries := flag.Int("retries", cfg.Retries, "Retry count (0-10)")
|
||||
fast := flag.Bool("fast", cfg.Fast, "Fast mode: reuse earlier branch cache")
|
||||
verbose := flag.Bool("verbose", cfg.Verbose, "Verbose output")
|
||||
debugFlag := flag.Bool("debug", false, "Debug mode (stackable: -dd for library debug)")
|
||||
quiet := flag.Bool("quiet", cfg.Quiet, "Suppress supplementary info")
|
||||
|
||||
showProgress := flag.Bool("show-progress", cfg.ShowProgress, "Show traversal progress")
|
||||
noShowProgress := flag.Bool("no-show-progress", false, "Hide traversal progress")
|
||||
showResolves := flag.Bool("show-resolves", cfg.ShowResolves, "Show glue resolution details")
|
||||
noShowResolves := flag.Bool("no-show-resolves", false, "Hide glue resolution details")
|
||||
showServers := flag.Bool("show-servers", cfg.ShowServers, "Show servers queried")
|
||||
noShowServers := flag.Bool("no-show-servers", false, "Hide servers queried")
|
||||
showVersions := flag.Bool("show-versions", cfg.ShowVersions, "Show server versions")
|
||||
noShowVersions := flag.Bool("no-show-versions", false, "Hide server versions")
|
||||
showAllStats := flag.Bool("show-all-stats", cfg.ShowAllStats, "Show all statistics")
|
||||
noShowAllStats := flag.Bool("no-show-all-stats", false, "Hide all statistics")
|
||||
showResults := flag.Bool("show-results", cfg.ShowResults, "Show query results")
|
||||
noShowResults := flag.Bool("no-show-results", false, "Hide query results")
|
||||
showSummaryResults := flag.Bool("show-summary-results", cfg.ShowSummaryResults, "Show summary of results")
|
||||
noShowSummaryResults := flag.Bool("no-show-summary-results", false, "Hide summary of results")
|
||||
|
||||
flag.Usage = config.PrintUsage
|
||||
|
||||
flag.Parse()
|
||||
|
||||
if flag.NArg() == 0 {
|
||||
flag.Usage()
|
||||
args := flag.Args()
|
||||
|
||||
debugLevel := 0
|
||||
for _, arg := range os.Args[1:] {
|
||||
if strings.Count(arg, "-d") > 0 || strings.Count(arg, "-dd") > 0 {
|
||||
debugLevel = strings.Count(arg, "-d") + strings.Count(arg, "-dd")
|
||||
}
|
||||
}
|
||||
if *debugFlag {
|
||||
debugLevel++
|
||||
}
|
||||
|
||||
cfg.QueryType = *queryType
|
||||
cfg.RootServer = *rootServer
|
||||
cfg.AllRootServers = *allRootServers
|
||||
cfg.RootAAAA = *rootAAAA
|
||||
cfg.FollowAAAA = *followAAAA
|
||||
cfg.UDPSize = *udpSize
|
||||
cfg.AllowTCP = *allowTCP
|
||||
cfg.AlwaysTCP = *alwaysTCP
|
||||
cfg.MaxDepth = *maxDepth
|
||||
cfg.Retries = *retries
|
||||
cfg.Fast = *fast
|
||||
cfg.Verbose = *verbose
|
||||
cfg.Debug = debugLevel
|
||||
cfg.Quiet = *quiet
|
||||
|
||||
if *noShowProgress {
|
||||
cfg.ShowProgress = false
|
||||
} else if *showProgress {
|
||||
cfg.ShowProgress = true
|
||||
}
|
||||
if *noShowResolves {
|
||||
cfg.ShowResolves = false
|
||||
} else if *showResolves {
|
||||
cfg.ShowResolves = true
|
||||
}
|
||||
if *noShowServers {
|
||||
cfg.ShowServers = false
|
||||
} else if *showServers {
|
||||
cfg.ShowServers = true
|
||||
}
|
||||
if *noShowVersions {
|
||||
cfg.ShowVersions = false
|
||||
} else if *showVersions {
|
||||
cfg.ShowVersions = true
|
||||
}
|
||||
if *noShowAllStats {
|
||||
cfg.ShowAllStats = false
|
||||
} else if *showAllStats {
|
||||
cfg.ShowAllStats = true
|
||||
}
|
||||
if *noShowResults {
|
||||
cfg.ShowResults = false
|
||||
} else if *showResults {
|
||||
cfg.ShowResults = true
|
||||
}
|
||||
if *noShowSummaryResults {
|
||||
cfg.ShowSummaryResults = false
|
||||
} else if *showSummaryResults {
|
||||
cfg.ShowSummaryResults = true
|
||||
}
|
||||
|
||||
if err := cfg.Validate(); err != nil {
|
||||
fmt.Fprintf(os.Stderr, "Error: %v\n", err)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
fmt.Printf("ExploreDNS - exploring: %s\n", flag.Arg(0))
|
||||
domain, err := cfg.GetDomain(args)
|
||||
if err != nil {
|
||||
fmt.Fprintf(os.Stderr, "Error: %v\n", err)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
rootIP, err := cfg.ParseRootServer()
|
||||
if err != nil {
|
||||
fmt.Fprintf(os.Stderr, "Error: invalid root server: %v\n", err)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
queryTypeValue, err := config.ParseQueryType(cfg.QueryType)
|
||||
if err != nil {
|
||||
fmt.Fprintf(os.Stderr, "Error: %v\n", err)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
var rootServerAddr string
|
||||
if rootIP != nil {
|
||||
rootServerAddr = rootIP.String()
|
||||
}
|
||||
|
||||
queryConfig := &dns.QueryConfig{
|
||||
UDPSize: cfg.UDPSize,
|
||||
Timeout: 5 * time.Second,
|
||||
Retries: cfg.Retries,
|
||||
UseTCP: cfg.AlwaysTCP,
|
||||
}
|
||||
|
||||
rootConfig := &dns.RootDiscoveryConfig{
|
||||
IncludeAAAA: cfg.RootAAAA,
|
||||
Server: rootServerAddr,
|
||||
AllRoots: cfg.AllRootServers,
|
||||
}
|
||||
|
||||
traverserConfig := &traverse.TraverserConfig{
|
||||
MaxDepth: cfg.MaxDepth,
|
||||
QueryType: queryTypeValue,
|
||||
RootConfig: rootConfig,
|
||||
QueryConfig: queryConfig,
|
||||
}
|
||||
|
||||
traverser := traverse.NewTraverser(traverserConfig)
|
||||
|
||||
if cfg.Debug > 0 {
|
||||
fmt.Fprintf(os.Stderr, "Debug: Config loaded\n")
|
||||
fmt.Fprintf(os.Stderr, " Domain: %s\n", domain)
|
||||
fmt.Fprintf(os.Stderr, " Query Type: %s\n", cfg.QueryType)
|
||||
fmt.Fprintf(os.Stderr, " Max Depth: %d\n", cfg.MaxDepth)
|
||||
fmt.Fprintf(os.Stderr, " UDP Size: %d\n", cfg.UDPSize)
|
||||
fmt.Fprintf(os.Stderr, " Retries: %d\n", cfg.Retries)
|
||||
fmt.Fprintf(os.Stderr, " Allow TCP: %v\n", cfg.AllowTCP)
|
||||
fmt.Fprintf(os.Stderr, " Always TCP: %v\n", cfg.AlwaysTCP)
|
||||
fmt.Fprintf(os.Stderr, " Fast Mode: %v\n", cfg.Fast)
|
||||
}
|
||||
|
||||
if !cfg.Quiet {
|
||||
fmt.Printf("ExploreDNS - exploring: %s (type: %s)\n", domain, cfg.QueryType)
|
||||
}
|
||||
|
||||
ctx := context.Background()
|
||||
results, err := traverser.Traverse(ctx, domain)
|
||||
if err != nil {
|
||||
fmt.Fprintf(os.Stderr, "Error: traversal failed: %v\n", err)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
if cfg.ShowResults && len(results) > 0 {
|
||||
fmt.Printf("\nResults:\n")
|
||||
for i, result := range results {
|
||||
fmt.Printf(" [%d] %s -> %s\n", i+1, result.Referral.Name, result.Response.Type)
|
||||
}
|
||||
}
|
||||
|
||||
if cfg.Debug > 0 {
|
||||
fmt.Fprintf(os.Stderr, "Debug: Traversal completed with %d results\n", len(results))
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user