Skip to content

Commit 9e774fb

Browse files
wzv5wzv5
committed
dns ping
1 parent 4c6b4d4 commit 9e774fb

File tree

5 files changed

+199
-1
lines changed

5 files changed

+199
-1
lines changed

‎cmd/pping/cmd/dns.go‎

Lines changed: 67 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,67 @@
1+
package cmd
2+
3+
import (
4+
"fmt"
5+
"net"
6+
"strconv"
7+
"time"
8+
9+
"github.com/wzv5/pping/pkg/ping"
10+
11+
"github.com/spf13/cobra"
12+
)
13+
14+
type dnsFlags struct {
15+
port uint16
16+
timeout time.Duration
17+
tcp bool
18+
tls bool
19+
qtype string
20+
domain string
21+
}
22+
23+
var dnsflag dnsFlags
24+
25+
func addDnsCommand() {
26+
var cmd = &cobra.Command{
27+
Use: "dns <host>",
28+
Short: "dns ping",
29+
Long: "dns ping",
30+
Args: cobra.ExactArgs(1),
31+
RunE: rundns,
32+
}
33+
34+
cmd.Flags().DurationVarP(&dnsflag.timeout, "timeout", "w", time.Second*4, "timeout")
35+
cmd.Flags().Uint16VarP(&dnsflag.port, "port", "p", 0, "port")
36+
cmd.Flags().BoolVar(&dnsflag.tcp, "tcp", false, "use TCP")
37+
cmd.Flags().BoolVar(&dnsflag.tls, "tls", false, "use DNS-over-TLS")
38+
cmd.Flags().StringVar(&dnsflag.qtype, "type", "NS", "A, AAAA, NS, ...")
39+
cmd.Flags().StringVar(&dnsflag.domain, "domain", ".", "domain")
40+
41+
rootCmd.AddCommand(cmd)
42+
}
43+
44+
func rundns(cmd *cobra.Command, args []string) error {
45+
host := args[0]
46+
Net := "udp"
47+
if dnsflag.tls {
48+
Net = "tcp-tls"
49+
} else if dnsflag.tcp {
50+
Net = "tcp"
51+
}
52+
if dnsflag.port == 0 {
53+
switch Net {
54+
case "udp", "tcp":
55+
dnsflag.port = 53
56+
case "tcp-tls":
57+
dnsflag.port = 853
58+
}
59+
}
60+
fmt.Printf("Ping %s://%s:\n", Net, net.JoinHostPort(host, strconv.Itoa(int(dnsflag.port))))
61+
p := ping.NewDnsPing(host, dnsflag.timeout)
62+
p.Port = dnsflag.port
63+
p.Net = Net
64+
p.Type = dnsflag.qtype
65+
p.Domain = dnsflag.domain
66+
return RunPing(p)
67+
}

‎cmd/pping/cmd/root.go‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -56,6 +56,7 @@ func init() {
5656
addTlsCommand()
5757
addHttpCommand()
5858
addIcmpCommand()
59+
addDnsCommand()
5960
}
6061

6162
func Execute() error {

‎go.mod‎

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@ module github.com/wzv5/pping
33
go 1.14
44

55
require (
6+
github.com/miekg/dns v1.1.31
67
github.com/spf13/cobra v1.0.0
7-
golang.org/x/net v0.0.0-20200904194848-62affa334b73
8+
golang.org/x/net v0.0.0-20201002202402-0a1ea396d57c
89
)

‎go.sum‎

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -50,6 +50,8 @@ github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ=
5050
github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI=
5151
github.com/magiconair/properties v1.8.0/go.mod h1:PppfXfuXeibc/6YijjN8zIbojt8czPbwD3XqdrwzmxQ=
5252
github.com/matttproud/golang_protobuf_extensions v1.0.1/go.mod h1:D8He9yQNgCq6Z5Ld7szi9bcBfOoFv/3dc6xSMkL2PC0=
53+
github.com/miekg/dns v1.1.31 h1:sJFOl9BgwbYAWOGEwr61FU28pqsBNdpRBnhGXtO06Oo=
54+
github.com/miekg/dns v1.1.31/go.mod h1:KNUDUusw/aVsxyTYZM1oqvCicbwhgbNgztCETuNZ7xM=
5355
github.com/mitchellh/go-homedir v1.1.0/go.mod h1:SfyaCUpYCn1Vlf4IUYiD9fPX4A5wJrkLzIz1N1q0pr0=
5456
github.com/mitchellh/mapstructure v1.1.2/go.mod h1:FVVH3fgwuzCH5S8UJGiWEs2h04kUh9fWfEaFds41c1Y=
5557
github.com/mwitkow/go-conntrack v0.0.0-20161129095857-cc309e4a2223/go.mod h1:qRWi+5nqEBWmkhHvq77mSJWrCKwh8bxhgT7d/eI7P4U=
@@ -92,34 +94,47 @@ go.uber.org/multierr v1.1.0/go.mod h1:wR5kodmAFQ0UK8QlbwjlSNy0Z68gJhDJUG5sjR94q/
9294
go.uber.org/zap v1.10.0/go.mod h1:vwi/ZaCAaUcBkycHslxD9B2zi4UTXhF60s6SWpuDF0Q=
9395
golang.org/x/crypto v0.0.0-20180904163835-0709b304e793/go.mod h1:6SG95UA2DQfeDnfUPMdvaQW0Q7yPrPDi9nlGo2tz2b4=
9496
golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
97+
golang.org/x/crypto v0.0.0-20191011191535-87dc89f01550/go.mod h1:yigFU9vqHzYiE8UmvKecakEJjdnWj3jj499lnFckfCI=
98+
golang.org/x/crypto v0.0.0-20200622213623-75b288015ac9 h1:psW17arqaxU48Z5kZ0CQnkZWQJsqcURM6tKiBApRjXI=
9599
golang.org/x/crypto v0.0.0-20200622213623-75b288015ac9/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto=
96100
golang.org/x/lint v0.0.0-20181026193005-c67002cb31c3/go.mod h1:UVdnD1Gm6xHRNCYTkRU2/jEulfH38KcIWyp/GAMgvoE=
97101
golang.org/x/lint v0.0.0-20190313153728-d0100b6bd8b3/go.mod h1:6SW0HCj/g11FgYtHlgUYUwCkIfeOF89ocIRzGO/8vkc=
102+
golang.org/x/mod v0.1.1-0.20191105210325-c90efee705ee/go.mod h1:QqPTAvyqsEbceGzBzNggFXnrqF1CaUcvgkdR5Ot7KZg=
98103
golang.org/x/net v0.0.0-20180826012351-8a410e7b638d/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
99104
golang.org/x/net v0.0.0-20181114220301-adae6a3d119a/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
100105
golang.org/x/net v0.0.0-20181220203305-927f97764cc3/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
101106
golang.org/x/net v0.0.0-20190311183353-d8887717615a/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=
102107
golang.org/x/net v0.0.0-20190404232315-eb5bcb51f2a3/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=
103108
golang.org/x/net v0.0.0-20190522155817-f3200d17e092/go.mod h1:HSz+uSET+XFnRR8LxR5pz3Of3rY3CfYBVs4xY44aLks=
109+
golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=
110+
golang.org/x/net v0.0.0-20190923162816-aa69164e4478/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=
104111
golang.org/x/net v0.0.0-20200904194848-62affa334b73 h1:MXfv8rhZWmFeqX3GNZRsd6vOLoaCHjYEX3qkRo3YBUA=
105112
golang.org/x/net v0.0.0-20200904194848-62affa334b73/go.mod h1:/O7V0waA8r7cgGh81Ro3o1hOxt32SMVPicZroKQ2sZA=
113+
golang.org/x/net v0.0.0-20201002202402-0a1ea396d57c h1:dk0ukUIHmGHqASjP0iue2261isepFCC6XRCSd1nHgDw=
114+
golang.org/x/net v0.0.0-20201002202402-0a1ea396d57c/go.mod h1:iQL9McJNjoIa5mjH6nYTCTZXUN6RP+XW3eib7Ya3XcI=
106115
golang.org/x/oauth2 v0.0.0-20180821212333-d2e6202438be/go.mod h1:N/0e6XlmueqKjAGxoOufVs8QHGRruUQn6yWY3a++T0U=
107116
golang.org/x/sync v0.0.0-20180314180146-1d60e4601c6f/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
108117
golang.org/x/sync v0.0.0-20181108010431-42b317875d0f/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
109118
golang.org/x/sync v0.0.0-20181221193216-37e7f081c4d4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
119+
golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
110120
golang.org/x/sys v0.0.0-20180830151530-49385e6e1522/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
111121
golang.org/x/sys v0.0.0-20180905080454-ebe1bf3edb33/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
112122
golang.org/x/sys v0.0.0-20181107165924-66b7b1311ac8/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
113123
golang.org/x/sys v0.0.0-20181116152217-5ac8a444bdc5/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
114124
golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
115125
golang.org/x/sys v0.0.0-20190412213103-97732733099d/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
126+
golang.org/x/sys v0.0.0-20190924154521-2837fb4f24fe/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
116127
golang.org/x/sys v0.0.0-20200323222414-85ca7c5b95cd h1:xhmwyvizuTgC2qz7ZlMluP20uW+C3Rm0FD/WLDX8884=
117128
golang.org/x/sys v0.0.0-20200323222414-85ca7c5b95cd/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
129+
golang.org/x/sys v0.0.0-20200930185726-fdedc70b468f h1:+Nyd8tzPX9R7BWHguqsrbFdRx3WQ/1ib8I44HXV5yTA=
130+
golang.org/x/sys v0.0.0-20200930185726-fdedc70b468f/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
118131
golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
119132
golang.org/x/time v0.0.0-20190308202827-9d24e82272b4/go.mod h1:tRJNPiyCQ0inRvYxbN9jk5I+vvW/OXSQhTDSoE431IQ=
120133
golang.org/x/tools v0.0.0-20180221164845-07fd8470d635/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
121134
golang.org/x/tools v0.0.0-20190114222345-bf090417da8b/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
122135
golang.org/x/tools v0.0.0-20190311212946-11955173bddd/go.mod h1:LCzVGOaR6xXOjkQ3onu1FJEFr0SW1gC7cKk1uF8kGRs=
136+
golang.org/x/tools v0.0.0-20191216052735-49a3e744a425/go.mod h1:TB2adYChydJhpapKDTa4BR/hXlZSLoq2Wpct/0txZ28=
137+
golang.org/x/xerrors v0.0.0-20191011141410-1b5146add898/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
123138
google.golang.org/appengine v1.1.0/go.mod h1:EbEs0AVv82hx2wNQdGPgUI5lhzA/G0D9YwlJXL52JkM=
124139
google.golang.org/genproto v0.0.0-20180817151627-c66870c02cf8/go.mod h1:JiN7NxoALGmiZfu7CAH4rXhgtRTLTxftemlI0sWmxmc=
125140
google.golang.org/grpc v1.19.0/go.mod h1:mqu4LbDTu4XGKhr4mRzUsmM4RtVoemTSY81AxZiDr8c=

‎pkg/ping/dns.go‎

Lines changed: 114 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,114 @@
1+
package ping
2+
3+
import (
4+
"context"
5+
"errors"
6+
"fmt"
7+
"net"
8+
"strconv"
9+
"time"
10+
11+
"github.com/miekg/dns"
12+
)
13+
14+
type DnsPingResult struct {
15+
Time int
16+
Err error
17+
IP net.IP
18+
}
19+
20+
func (this *DnsPingResult) Result() int {
21+
return this.Time
22+
}
23+
24+
func (this *DnsPingResult) Error() error {
25+
return this.Err
26+
}
27+
28+
func (this *DnsPingResult) String() string {
29+
if this.Err != nil {
30+
return fmt.Sprintf("%s", this.Err)
31+
} else {
32+
return fmt.Sprintf("%s: time=%d ms", this.IP.String(), this.Time)
33+
}
34+
}
35+
36+
type DnsPing struct {
37+
host string
38+
Port uint16
39+
Timeout time.Duration
40+
41+
// udp, tcp, tcp-tls，默认 udp
42+
Net string
43+
44+
// A, AAAA, NS, ...，默认 NS
45+
Type string
46+
47+
// 查询域名，默认 .
48+
Domain string
49+
50+
ip net.IP
51+
}
52+
53+
func (this *DnsPing) SetHost(host string) {
54+
this.host = host
55+
this.ip = net.ParseIP(host)
56+
}
57+
58+
func (this *DnsPing) Host() string {
59+
return this.host
60+
}
61+
62+
func (this *DnsPing) Ping() IPingResult {
63+
return this.PingContext(context.Background())
64+
}
65+
66+
func (this *DnsPing) PingContext(ctx context.Context) IPingResult {
67+
ip := cloneIP(this.ip)
68+
if ip == nil {
69+
var err error
70+
ip, err = LookupFunc(this.host)
71+
if err != nil {
72+
return &DnsPingResult{0, err, nil}
73+
}
74+
}
75+
76+
msg := &dns.Msg{}
77+
qtype, ok := dns.StringToType[this.Type]
78+
if !ok {
79+
return &DnsPingResult{0, errors.New("unknown type"), nil}
80+
}
81+
msg.SetQuestion(this.Domain, qtype)
82+
msg.MsgHdr.RecursionDesired = true
83+
84+
client := &dns.Client{}
85+
client.Net = this.Net
86+
client.Timeout = this.Timeout
87+
88+
t0 := time.Now()
89+
r, _, err := client.ExchangeContext(ctx, msg, net.JoinHostPort(ip.String(), strconv.Itoa(int(this.Port))))
90+
if err != nil {
91+
return &DnsPingResult{0, err, nil}
92+
}
93+
if r == nil || r.Response == false || r.Opcode != dns.OpcodeQuery {
94+
return &DnsPingResult{0, errors.New("response error"), nil}
95+
}
96+
return &DnsPingResult{int(time.Now().Sub(t0).Milliseconds()), nil, ip}
97+
}
98+
99+
func NewDnsPing(host string, timeout time.Duration) *DnsPing {
100+
return &DnsPing{
101+
host: host,
102+
Port: 53,
103+
Timeout: timeout,
104+
Net: "udp",
105+
Type: "NS",
106+
Domain: ".",
107+
ip: net.ParseIP(host),
108+
}
109+
}
110+
111+
var (
112+
_ IPing = (*DnsPing)(nil)
113+
_ IPingResult = (*DnsPingResult)(nil)
114+
)

0 commit comments

Comments
 (0)