fix: dns command bug

Author: Esonhugh

cmd/dnsutils/dns.go

@@ -12,7 +12,7 @@ import (
 var queryType string
 
 func init() {
-	DNSCmd.PersistentFlags().StringVarP(&queryType, "type", "t", "A", "query type")
+	DNSCmd.PersistentFlags().StringVarP(&queryType, "type", "x", "A", "query type")
 	command.RootCmd.AddCommand(DNSCmd)
 }
 
@@ -40,7 +40,7 @@ var DNSCmd = &cobra.Command{
 				log.Warnf("Query %s failed: %v", query, err)
 				continue
 			}
-			log.Infof("Query [%d] %s: %v", queryType, query, res)
+			log.Infof("Query [%s] %s: %v", queryType, query, res)
 		}
 	},
 }

cmd/root.go

@@ -3,15 +3,19 @@ package cmd
 import (
 	"encoding/json"
 	"fmt"
+	"net"
 	"os"
+	"strings"
 
 	"github.com/esonhugh/k8spider/pkg"
 	log "github.com/sirupsen/logrus"
 	"github.com/spf13/cobra"
 )
 
 var Opts = struct {
-	Cidr       string
+	Cidr    string
+	PodCidr string
+
 	DnsServer  string
 	SvcDomains []string
 	Zone       string
@@ -27,9 +31,24 @@ var Opts = struct {
 	FilterStrings []string
 }{}
 
+func defaultPodCidr() string {
+	interfaces, _ := net.Interfaces()
+	for _, i := range interfaces {
+		if i.Name == "eth0" {
+			addrs, _ := i.Addrs()
+			if addrs != nil || len(addrs) > 0 {
+				ip := strings.Split(addrs[0].String(), "/")[0]
+				return fmt.Sprintf("%v/16", ip)
+			}
+		}
+	}
+	return "10.0.0.1/16"
+}
+
 func init() {
 
 	RootCmd.PersistentFlags().StringVarP(&Opts.Cidr, "cidr", "c", os.Getenv("KUBERNETES_SERVICE_HOST")+"/16", "cidr like: 192.168.0.1/16")
+	RootCmd.PersistentFlags().StringVarP(&Opts.PodCidr, "pod-cidr", "p", defaultPodCidr(), "pod cidr list, watch out for the network interface name, default is eth0")
 
 	RootCmd.PersistentFlags().StringVarP(&Opts.DnsServer, "dns-server", "d", "", "dns server")
 	RootCmd.PersistentFlags().IntVarP(&pkg.DnsTimeout, "dns-timeout", "i", 2, "dns timeout")