initial commit

2 files changed, 78 insertions, 0 deletions

diff --git a/cmd/root.go b/cmd/root.go
new file mode 100644
index 0000000..6597d83
--- /dev/null
+++ b/cmd/root.go
@@ -0,0 +1,46 @@
+/*
+Copyright © 2023 NAME HERE <EMAIL ADDRESS>
+*/
+package cmd
+
+import (
+	"os"
+
+	"github.com/spf13/cobra"
+)
+
+// rootCmd represents the base command when called without any subcommands
+var rootCmd = &cobra.Command{
+	Use:   "dtool",
+	Short: "dns resolver probing tool",
+	Long: `A tool for probing open resolvers for various purpose. The following modules are implemented.
+
+Modules:
+	upstream	probe to get the upstream recursive resolvers of the given resolver as many as possible
+	ttl			probe to test if resolvers on the resolution path change the authoritative TTL value
+	cache		probe to enumerate the cache`,
+	// Uncomment the following line if your bare application
+	// has an action associated with it:
+	// Run: func(cmd *cobra.Command, args []string) { },
+}
+
+// Execute adds all child commands to the root command and sets flags appropriately.
+// This is called by main.main(). It only needs to happen once to the rootCmd.
+func Execute() {
+	err := rootCmd.Execute()
+	if err != nil {
+		os.Exit(1)
+	}
+}
+
+func init() {
+	// Here you will define your flags and configuration settings.
+	// Cobra supports persistent flags, which, if defined here,
+	// will be global for your application.
+
+	// rootCmd.PersistentFlags().StringVar(&cfgFile, "config", "", "config file (default is $HOME/.dtool.yaml)")
+
+	// Cobra also supports local flags, which will only run
+	// when this action is called directly.
+	rootCmd.Flags().BoolP("toggle", "t", false, "Help message for toggle")
+}
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)
+}