79 lines
1.9 KiB
Go
79 lines
1.9 KiB
Go
package whois
|
||
|
||
import (
|
||
"b612.me/starlog"
|
||
"b612.me/staros"
|
||
"github.com/likexian/whois"
|
||
"github.com/spf13/cobra"
|
||
"golang.org/x/net/proxy"
|
||
"os"
|
||
"strings"
|
||
"time"
|
||
)
|
||
|
||
var timeout int
|
||
var output string
|
||
var whoisServer []string
|
||
var socks5 string
|
||
var socks5Auth string
|
||
|
||
func init() {
|
||
Cmd.Flags().IntVarP(&timeout, "timeout", "t", 20, "超时时间")
|
||
Cmd.Flags().StringVarP(&output, "output", "o", "", "输出文件夹")
|
||
Cmd.Flags().StringSliceVarP(&whoisServer, "server", "s", nil, "whois服务器")
|
||
Cmd.Flags().StringVarP(&socks5, "socks5", "p", "", "socks5代理，示例：127.0.0.1:1080")
|
||
Cmd.Flags().StringVarP(&socks5Auth, "socks5-auth", "A", "", "socks5代理认证，示例：username:password")
|
||
}
|
||
|
||
var Cmd = &cobra.Command{
|
||
Use: "whois",
|
||
Short: "whois查询",
|
||
Long: "whois查询",
|
||
Run: func(cmd *cobra.Command, args []string) {
|
||
if len(args) == 0 {
|
||
cmd.Help()
|
||
return
|
||
}
|
||
if !staros.Exists(output) {
|
||
cmd.Println("输出文件夹不存在，将使用标准输出")
|
||
output = ""
|
||
}
|
||
c := whois.NewClient()
|
||
if socks5 != "" {
|
||
var auth *proxy.Auth
|
||
if socks5Auth != "" {
|
||
up := strings.SplitN(socks5Auth, ":", 2)
|
||
if len(up) == 2 {
|
||
auth = &proxy.Auth{
|
||
User: up[0],
|
||
Password: up[1],
|
||
}
|
||
} else {
|
||
starlog.Errorln("socks5认证格式错误")
|
||
return
|
||
}
|
||
}
|
||
s5Dial, err := proxy.SOCKS5("tcp", socks5, auth, proxy.Direct)
|
||
if err == nil {
|
||
c.SetDialer(s5Dial)
|
||
} else {
|
||
starlog.Errorln("socks5代理错误:", err)
|
||
return
|
||
}
|
||
}
|
||
c.SetTimeout(time.Second * time.Duration(timeout))
|
||
for _, v := range args {
|
||
data, err := c.Whois(v, whoisServer...)
|
||
cmd.Println("Query:", v)
|
||
if err != nil {
|
||
cmd.Println("查询失败:", err)
|
||
cmd.Println("-----------------------------------------------------")
|
||
continue
|
||
}
|
||
cmd.Println(data)
|
||
cmd.Println("-----------------------------------------------------")
|
||
os.WriteFile(output+"/"+v+".txt", []byte(data), 0644)
|
||
}
|
||
},
|
||
}
|