48 lines
1.2 KiB
Go
48 lines
1.2 KiB
Go
package cmd
|
|
|
|
import (
|
|
"dtool/prober"
|
|
"dtool/utils"
|
|
|
|
"fmt"
|
|
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
var record_input string
|
|
var record_output string
|
|
var record_type string
|
|
var record_domain string
|
|
var recordCmd = &cobra.Command{
|
|
Use: "record",
|
|
Short: "get specific record response",
|
|
Long: "get specific record response",
|
|
Run: record_probe,
|
|
}
|
|
|
|
func record_probe(cmd *cobra.Command, args []string) {
|
|
if len(args) == 1 {
|
|
if utils.IsValidIP(args[0]) {
|
|
result, err := prober.SVCBProbeOnce(args[0], record_domain)
|
|
if err == nil {
|
|
if output_str, err := prober.OutputHandler(result); err == nil {
|
|
fmt.Println(output_str)
|
|
}
|
|
} else {
|
|
fmt.Println(err)
|
|
}
|
|
}
|
|
} else {
|
|
prober.CreateTask(prober.SVCBProbe, record_domain, record_input, record_output, 500)
|
|
}
|
|
}
|
|
|
|
func init() {
|
|
recordCmd.Flags().StringVarP(&record_input, "input", "i", "", "input file(optional)")
|
|
recordCmd.Flags().StringVarP(&record_output, "output", "o", "", "output file(optional)")
|
|
recordCmd.MarkFlagsRequiredTogether("input", "output")
|
|
recordCmd.Flags().StringVarP(&record_type, "type", "t", "A", "request record type")
|
|
recordCmd.Flags().StringVarP(&record_domain, "domain", "d", "example.com", "requested domain")
|
|
rootCmd.AddCommand(recordCmd)
|
|
}
|