update: dns utils

Makefile

@@ -3,14 +3,18 @@ BUILD_DIR = bin
 
 MAIN_PROGRAM_NAME = k8spider
 
-default: build build-static
+default: build build-static check-size
 
 # build
 build:
 	go	build -o $(BUILD_DIR)/$(MAIN_PROGRAM_NAME) main.go
 
 build-static:
 	GOOS=linux GOARCH=amd64 go build -ldflags="-s -w" -o $(BUILD_DIR)/$(MAIN_PROGRAM_NAME)-linux-static main.go
-	upx $(BUILD_DIR)/$(MAIN_PROGRAM_NAME)-linux-static
+	upx -9 $(BUILD_DIR)/$(MAIN_PROGRAM_NAME)-linux-static
+
+check-size:
+	ls -alh $(BUILD_DIR)/$(MAIN_PROGRAM_NAME)*
+
 clean:
 	rm -rf $(BUILD_DIR)

cmd/all/all.go

@@ -48,13 +48,15 @@ var AllCmd = &cobra.Command{
 			log.Warnf("ParseStringToIPNet failed: %v", err)
 			return
 		}
+
 		var finalRecord define.Records
 		if command.Opts.MultiThreadingMode {
 			finalRecord = RunMultiThread(ipNets, command.Opts.ThreadingNum)
 		} else {
 			finalRecord = Run(ipNets)
 		}
 		printer.PrintResult(finalRecord, command.Opts.OutputFile)
+
 		PostRun(finalRecord)
 	},
 }
@@ -78,6 +80,9 @@ func RunMultiThread(net *net.IPNet, count int) (finalRecord define.Records) {
 }
 
 func PostRun(finalRecord define.Records) {
+	if finalRecord == nil || len(finalRecord) == 0 {
+		return
+	}
 	log.Info("Extract Namespaces: ")
 	list := post.RecordsDumpNameSpace(finalRecord, command.Opts.Zone)
 	for _, ns := range list {

cmd/dnsutils/dns.go

@@ -0,0 +1,46 @@
+package dnsutils
+
+import (
+	"strings"
+
+	command "github.com/esonhugh/k8spider/cmd"
+	"github.com/esonhugh/k8spider/pkg"
+	log "github.com/sirupsen/logrus"
+	"github.com/spf13/cobra"
+)
+
+var queryType string
+
+func init() {
+	DNSCmd.PersistentFlags().StringVarP(&queryType, "type", "t", "A", "query type")
+	command.RootCmd.AddCommand(DNSCmd)
+}
+
+var DNSCmd = &cobra.Command{
+	Use:     "dns",
+	Aliases: []string{"dig"},
+	Short:   "dns is a command to query dns server",
+	Run: func(cmd *cobra.Command, args []string) {
+		var querier pkg.DnsQuery
+		switch strings.ToLower(queryType) {
+		case "a", "aaaa":
+			querier = pkg.QueryA
+		case "ptr":
+			querier = pkg.QueryPTR
+		case "srv":
+			querier = pkg.QuerySRV
+		case "txt":
+			querier = pkg.QueryTXT
+		default:
+			querier = pkg.QueryA
+		}
+		for _, query := range args {
+			res, err := querier(query)
+			if err != nil {
+				log.Warnf("Query %s failed: %v", query, err)
+				continue
+			}
+			log.Infof("Query [%d] %s: %v", queryType, query, res)
+		}
+	},
+}

main.go

@@ -4,6 +4,7 @@ import (
 	"github.com/esonhugh/k8spider/cmd"
 	_ "github.com/esonhugh/k8spider/cmd/all"
 	_ "github.com/esonhugh/k8spider/cmd/axfr"
+	_ "github.com/esonhugh/k8spider/cmd/dnsutils"
 	_ "github.com/esonhugh/k8spider/cmd/neighbor"
 	_ "github.com/esonhugh/k8spider/cmd/service"
 	_ "github.com/esonhugh/k8spider/cmd/subnet"

pkg/query_utils.go

@@ -2,6 +2,7 @@ package pkg
 
 import (
 	"context"
+	"fmt"
 	"net"
 	"regexp"
 	"strings"
@@ -89,7 +90,9 @@ func WarpDnsServer(dnsServer string) *SpiderResolver {
 				return d.DialContext(ctx, network, dnsServer)
 			},
 		},
-		ctx: ctx,
+		ctx:      ctx,
+		filter:   []*regexp.Regexp{},
+		contains: []string{},
 	}
 }
 
@@ -146,3 +149,28 @@ func (s *SpiderResolver) TXTRecord(domain string) ([]string, error) {
 func TXTRecord(domain string) (txts []string, err error) {
 	return NetResolver.TXTRecord(domain)
 }
+
+type DnsQuery func(domain string) ([]string, error)
+
+var (
+	QueryPTR DnsQuery = func(domain string) ([]string, error) {
+		return PTRRecord(net.ParseIP(domain)), nil
+	}
+	QueryA DnsQuery = func(domain string) ([]string, error) {
+		res, err := ARecord(domain)
+		var ret []string
+		for _, r := range res {
+			ret = append(ret, r.String())
+		}
+		return ret, err
+	}
+	QueryTXT DnsQuery = TXTRecord
+	QuerySRV DnsQuery = func(domain string) ([]string, error) {
+		_, res, err := SRVRecord(domain)
+		var ret []string
+		for _, r := range res {
+			ret = append(ret, fmt.Sprintf("%s:%d", r.Target, r.Port))
+		}
+		return ret, err
+	}
+)