initial commit

1 files changed, 32 insertions, 0 deletions

diff --git a/cmd/upstream.go b/cmd/upstream.go
new file mode 100644
index 0000000..e7d4071
--- /dev/null
+++ b/cmd/upstream.go
@@ -0,0 +1,32 @@
+package cmd
+
+import (
+	_ "dtool/prober"
+	"dtool/utils"
+
+	"github.com/spf13/cobra"
+)
+
+var filename string
+var thread_num int
+var upstreamCmd = &cobra.Command{
+	Use: "upstream",
+	//Aliases: []string{"up", "stream"},
+	Short: "probe upstream recursive resolvers",
+	Long: `get the upstream recursive resolvers of the input resolvers
+input target can be added as an argument or as a file
+
+	-f	input file with limited ip addresses (limit=50)
+	-t	number of goroutine`,
+	Args: cobra.ExactArgs(1),
+	Run: func(cmd *cobra.Command, args []string) {
+		//prober.Get_upstream_ip(args[0])
+		utils.SendTencentHttpdnsQuery()
+	},
+}
+
+func init() {
+	upstreamCmd.Flags().StringVarP(&filename, "file", "f", "", "input filename")
+	upstreamCmd.Flags().IntVarP(&thread_num, "threads", "t", 10, "number of concurrent threads")
+	rootCmd.AddCommand(upstreamCmd)
+}