Skip to content

Commit 0bde8cd

Browse files
wzv5wzv5
committed
添加 quic ping
1 parent 7b1b99f commit 0bde8cd

File tree

7 files changed

+266
-25
lines changed

7 files changed

+266
-25
lines changed

‎cmd/pping/cmd/quic.go‎

Lines changed: 55 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,55 @@
1+
package cmd
2+
3+
import (
4+
"errors"
5+
"fmt"
6+
"net"
7+
"time"
8+
9+
"github.com/quic-go/quic-go/http3"
10+
"github.com/spf13/cobra"
11+
"github.com/wzv5/pping/pkg/ping"
12+
)
13+
14+
type quicFlags struct {
15+
timeout time.Duration
16+
port uint16
17+
insecure bool
18+
alpn string
19+
}
20+
21+
var quicflag quicFlags
22+
23+
func addQuicCommand() {
24+
var cmd = &cobra.Command{
25+
Use: "quic <host> [ip]",
26+
Short: "quic ping",
27+
Long: "quic ping",
28+
Args: cobra.RangeArgs(1, 2),
29+
RunE: runquic,
30+
}
31+
32+
cmd.Flags().DurationVarP(&quicflag.timeout, "timeout", "w", time.Second*4, "timeout")
33+
cmd.Flags().Uint16VarP(&quicflag.port, "port", "p", 443, "port")
34+
cmd.Flags().BoolVarP(&quicflag.insecure, "insecure", "k", false, "allow insecure server connections")
35+
cmd.Flags().StringVarP(&quicflag.alpn, "alpn", "a", http3.NextProtoH3, "ALPN")
36+
rootCmd.AddCommand(cmd)
37+
}
38+
39+
func runquic(cmd *cobra.Command, args []string) error {
40+
host := args[0]
41+
var ip net.IP
42+
if len(args) == 2 {
43+
ip = net.ParseIP(args[1])
44+
if ip == nil {
45+
return errors.New("parse IP failed")
46+
}
47+
}
48+
49+
fmt.Printf("Ping %s (%d):\n", host, quicflag.port)
50+
p := ping.NewQuicPing(host, quicflag.port, quicflag.timeout)
51+
p.Insecure = quicflag.insecure
52+
p.ALPN = quicflag.alpn
53+
p.IP = ip
54+
return RunPing(p)
55+
}

‎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
addHttpCommand()
5757
addIcmpCommand()
5858
addDnsCommand()
59+
addQuicCommand()
5960
}
6061

6162
func Execute() error {

‎go.mod‎

Lines changed: 15 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -3,15 +3,26 @@ module github.com/wzv5/pping
33
go 1.19
44

55
require (
6-
github.com/miekg/dns v1.1.53
6+
github.com/miekg/dns v1.1.54
7+
github.com/quic-go/quic-go v0.34.0
78
github.com/spf13/cobra v1.7.0
8-
golang.org/x/net v0.9.0
9+
golang.org/x/net v0.10.0
910
)
1011

1112
require (
13+
github.com/go-task/slim-sprig v0.0.0-20230315185526-52ccab3ef572 // indirect
14+
github.com/golang/mock v1.6.0 // indirect
15+
github.com/google/pprof v0.0.0-20230510103437-eeec1cb781c3 // indirect
1216
github.com/inconshreveable/mousetrap v1.1.0 // indirect
17+
github.com/onsi/ginkgo/v2 v2.9.4 // indirect
18+
github.com/quic-go/qpack v0.4.0 // indirect
19+
github.com/quic-go/qtls-go1-19 v0.3.2 // indirect
20+
github.com/quic-go/qtls-go1-20 v0.2.2 // indirect
1321
github.com/spf13/pflag v1.0.5 // indirect
22+
golang.org/x/crypto v0.9.0 // indirect
23+
golang.org/x/exp v0.0.0-20230510235704-dd950f8aeaea // indirect
1424
golang.org/x/mod v0.10.0 // indirect
15-
golang.org/x/sys v0.7.0 // indirect
16-
golang.org/x/tools v0.8.0 // indirect
25+
golang.org/x/sys v0.8.0 // indirect
26+
golang.org/x/text v0.9.0 // indirect
27+
golang.org/x/tools v0.9.1 // indirect
1728
)

‎go.sum‎

Lines changed: 69 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1,21 +1,81 @@
11
github.com/cpuguy83/go-md2man/v2 v2.0.2/go.mod h1:tgQtvFlXSQOSOSIRvRPT7W67SCa46tRHOmNcaadrF8o=
2+
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
3+
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
4+
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
5+
github.com/go-logr/logr v1.2.4 h1:g01GSCwiDw2xSZfjJ2/T9M+S6pFdcNtFYsp+Y43HYDQ=
6+
github.com/go-task/slim-sprig v0.0.0-20230315185526-52ccab3ef572 h1:tfuBGBXKqDEevZMzYi5KSi8KkcZtzBcTgAUUtapy0OI=
7+
github.com/go-task/slim-sprig v0.0.0-20230315185526-52ccab3ef572/go.mod h1:9Pwr4B2jHnOSGXyyzV8ROjYa2ojvAY6HCGYYfMoC3Ls=
8+
github.com/golang/mock v1.6.0 h1:ErTB+efbowRARo13NNdxyJji2egdxLGQhRaY+DUumQc=
9+
github.com/golang/mock v1.6.0/go.mod h1:p6yTPP+5HYm5mzsMV8JkE6ZKdX+/wYM6Hr+LicevLPs=
10+
github.com/golang/protobuf v1.5.3 h1:KhyjKVUg7Usr/dYsdSqoFveMYd5ko72D+zANwlG1mmg=
11+
github.com/google/go-cmp v0.5.9 h1:O2Tfq5qg4qc4AmwVlvv0oLiVAGB7enBSJ2x2DqQFi38=
12+
github.com/google/pprof v0.0.0-20230510103437-eeec1cb781c3 h1:2XF1Vzq06X+inNqgJ9tRnGuw+ZVCB3FazXODD6JE1R8=
13+
github.com/google/pprof v0.0.0-20230510103437-eeec1cb781c3/go.mod h1:79YE0hCXdHag9sBkw2o+N/YnZtTkXi0UT9Nnixa5eYk=
214
github.com/inconshreveable/mousetrap v1.1.0 h1:wN+x4NVGpMsO7ErUn/mUI3vEoE6Jt13X2s0bqwp9tc8=
315
github.com/inconshreveable/mousetrap v1.1.0/go.mod h1:vpF70FUmC8bwa3OWnCshd2FqLfsEA9PFc4w1p2J65bw=
4-
github.com/miekg/dns v1.1.53 h1:ZBkuHr5dxHtB1caEOlZTLPo7D3L3TWckgUUs/RHfDxw=
5-
github.com/miekg/dns v1.1.53/go.mod h1:uInx36IzPl7FYnDcMeVWxj9byh7DutNykX4G9Sj60FY=
16+
github.com/miekg/dns v1.1.54 h1:5jon9mWcb0sFJGpnI99tOMhCPyJ+RPVz5b63MQG0VWI=
17+
github.com/miekg/dns v1.1.54/go.mod h1:uInx36IzPl7FYnDcMeVWxj9byh7DutNykX4G9Sj60FY=
18+
github.com/onsi/ginkgo/v2 v2.9.4 h1:xR7vG4IXt5RWx6FfIjyAtsoMAtnc3C/rFXBBd2AjZwE=
19+
github.com/onsi/ginkgo/v2 v2.9.4/go.mod h1:gCQYp2Q+kSoIj7ykSVb9nskRSsR6PUj4AiLywzIhbKM=
20+
github.com/onsi/gomega v1.27.6 h1:ENqfyGeS5AX/rlXDd/ETokDz93u0YufY1Pgxuy/PvWE=
21+
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
22+
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
23+
github.com/quic-go/qpack v0.4.0 h1:Cr9BXA1sQS2SmDUWjSofMPNKmvF6IiIfDRmgU0w1ZCo=
24+
github.com/quic-go/qpack v0.4.0/go.mod h1:UZVnYIfi5GRk+zI9UMaCPsmZ2xKJP7XBUvVyT1Knj9A=
25+
github.com/quic-go/qtls-go1-19 v0.3.2 h1:tFxjCFcTQzK+oMxG6Zcvp4Dq8dx4yD3dDiIiyc86Z5U=
26+
github.com/quic-go/qtls-go1-19 v0.3.2/go.mod h1:ySOI96ew8lnoKPtSqx2BlI5wCpUVPT05RMAlajtnyOI=
27+
github.com/quic-go/qtls-go1-20 v0.2.2 h1:WLOPx6OY/hxtTxKV1Zrq20FtXtDEkeY00CGQm8GEa3E=
28+
github.com/quic-go/qtls-go1-20 v0.2.2/go.mod h1:JKtK6mjbAVcUTN/9jZpvLbGxvdWIKS8uT7EiStoU1SM=
29+
github.com/quic-go/quic-go v0.34.0 h1:OvOJ9LFjTySgwOTYUZmNoq0FzVicP8YujpV0kB7m2lU=
30+
github.com/quic-go/quic-go v0.34.0/go.mod h1:+4CVgVppm0FNjpG3UcX8Joi/frKOH7/ciD5yGcwOO1g=
631
github.com/russross/blackfriday/v2 v2.1.0/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQDYRxCVz55jmeOWTM=
732
github.com/spf13/cobra v1.7.0 h1:hyqWnYt1ZQShIddO5kBpj3vu05/++x6tJ6dg8EC572I=
833
github.com/spf13/cobra v1.7.0/go.mod h1:uLxZILRyS/50WlhOIKD7W6V5bgeIt+4sICxh6uRMrb0=
934
github.com/spf13/pflag v1.0.5 h1:iy+VFUOCP1a+8yFto/drg2CJ5u0yRoB7fZw3DKv/JXA=
1035
github.com/spf13/pflag v1.0.5/go.mod h1:McXfInJRrz4CZXVZOBLb0bTZqETkiAhM9Iw0y3An2Bg=
36+
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
37+
github.com/stretchr/testify v1.6.1 h1:hDPOHmpOpP40lSULcqw7IrRb/u7w6RpDC9399XyoNd0=
38+
github.com/stretchr/testify v1.6.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
39+
github.com/yuin/goldmark v1.3.5/go.mod h1:mwnBkeHKe2W/ZEtQ+71ViKU8L12m81fl3OWwC1Zlc8k=
40+
golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
41+
golang.org/x/crypto v0.0.0-20191011191535-87dc89f01550/go.mod h1:yigFU9vqHzYiE8UmvKecakEJjdnWj3jj499lnFckfCI=
42+
golang.org/x/crypto v0.9.0 h1:LF6fAI+IutBocDJ2OT0Q1g8plpYljMZ4+lty+dsqw3g=
43+
golang.org/x/crypto v0.9.0/go.mod h1:yrmDGqONDYtNj3tH8X9dzUun2m2lzPa9ngI6/RUPGR0=
44+
golang.org/x/exp v0.0.0-20230510235704-dd950f8aeaea h1:vLCWI/yYrdEHyN2JzIzPO3aaQJHQdp89IZBA/+azVC4=
45+
golang.org/x/exp v0.0.0-20230510235704-dd950f8aeaea/go.mod h1:V1LtkGg67GoY2N1AnLN78QLrzxkLyJw7RJb1gzOOz9w=
46+
golang.org/x/mod v0.4.2/go.mod h1:s0Qsj1ACt9ePp/hMypM3fl4fZqREWJwdYDEqhRiZZUA=
1147
golang.org/x/mod v0.10.0 h1:lFO9qtOdlre5W1jxS3r/4szv2/6iXxScdzjoBMXNhYk=
1248
golang.org/x/mod v0.10.0/go.mod h1:iBbtSCu2XBx23ZKBPSOrRkjjQPZFPuis4dIYUhu/chs=
13-
golang.org/x/net v0.9.0 h1:aWJ/m6xSmxWBx+V0XRHTlrYrPG56jKsLdTFmsSsCzOM=
14-
golang.org/x/net v0.9.0/go.mod h1:d48xBJpPfHeWQsugry2m+kC02ZBRGRgulfHnEXEuWns=
15-
golang.org/x/sync v0.1.0 h1:wsuoTGHzEhffawBOhz5CYhcrV4IdKZbEyZjBMuTp12o=
16-
golang.org/x/sys v0.7.0 h1:3jlCCIQZPdOYu1h8BkNvLz8Kgwtae2cagcG/VamtZRU=
17-
golang.org/x/sys v0.7.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
18-
golang.org/x/tools v0.8.0 h1:vSDcovVPld282ceKgDimkRSC8kpaH1dgyc9UMzlt84Y=
19-
golang.org/x/tools v0.8.0/go.mod h1:JxBZ99ISMI5ViVkT1tr6tdNmXeTrcpVSD3vZ1RsRdN4=
49+
golang.org/x/net v0.0.0-20190404232315-eb5bcb51f2a3/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=
50+
golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=
51+
golang.org/x/net v0.0.0-20210405180319-a5a99cb37ef4/go.mod h1:p54w0d4576C0XHj96bSt6lcn1PtDYWL6XObtHCRCNQM=
52+
golang.org/x/net v0.10.0 h1:X2//UzNDwYmtCLn7To6G58Wr6f5ahEAQgKNzv9Y951M=
53+
golang.org/x/net v0.10.0/go.mod h1:0qNGK6F8kojg2nk9dLZ2mShWaEBan6FAoqfSigmmuDg=
54+
golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
55+
golang.org/x/sync v0.0.0-20210220032951-036812b2e83c/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
56+
golang.org/x/sync v0.2.0 h1:PUR+T4wwASmuSTYdKjYHI5TD22Wy5ogLU5qZCOLxBrI=
57+
golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
58+
golang.org/x/sys v0.0.0-20190412213103-97732733099d/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
59+
golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
60+
golang.org/x/sys v0.0.0-20210330210617-4fbd30eecc44/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
61+
golang.org/x/sys v0.0.0-20210510120138-977fb7262007/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
62+
golang.org/x/sys v0.8.0 h1:EBmGv8NaZBZTWvrbjNoL6HVt+IVy3QDQpJs7VRIw3tU=
63+
golang.org/x/sys v0.8.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
64+
golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=
65+
golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
66+
golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
67+
golang.org/x/text v0.9.0 h1:2sjJmO8cDvYveuX97RDLsxlyUxLl+GHoLxBiRdHllBE=
68+
golang.org/x/text v0.9.0/go.mod h1:e1OnstbJyHTd6l/uOt8jFFHp6TRDWZR/bV3emEE/zU8=
69+
golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
70+
golang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=
71+
golang.org/x/tools v0.1.1/go.mod h1:o0xws9oXOQQZyjljx8fwUC0k7L1pTE6eaCbjGeHmOkk=
72+
golang.org/x/tools v0.9.1 h1:8WMNJAz3zrtPmnYC7ISf5dEn3MT0gY7jBJfw27yrrLo=
73+
golang.org/x/tools v0.9.1/go.mod h1:owI94Op576fPu3cIGQeHs3joujW/2Oc6MtlxbF5dfNc=
74+
golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
75+
golang.org/x/xerrors v0.0.0-20191011141410-1b5146add898/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
76+
golang.org/x/xerrors v0.0.0-20200804184101-5ec99f83aff1/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
77+
google.golang.org/protobuf v1.28.0 h1:w43yiav+6bVFTBQFZX0r7ipe9JQ1QsbMgHwbBziscLw=
2078
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
79+
gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
80+
gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
2181
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=

‎pkg/ping/icmp.go‎

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -99,13 +99,6 @@ func (this *IcmpPing) rawping(network string) IPingResult {
9999
}
100100
defer conn.Close()
101101
conn.SetDeadline(time.Now().Add(this.Timeout))
102-
if this.TTL > 0 {
103-
if isipv6 {
104-
conn.IPv6PacketConn().SetHopLimit(this.TTL)
105-
} else {
106-
conn.IPv4PacketConn().SetTTL(this.TTL)
107-
}
108-
}
109102

110103
// 发送
111104
r := rand.New(rand.NewSource(time.Now().UnixNano()))
@@ -167,18 +160,18 @@ func (this *IcmpPing) rawping(network string) IPingResult {
167160
return this.errorResult(err)
168161
}
169162
recvData, recvID, recvType := this.parserecvmsg(isipv6, recvMsg)
170-
// 修正数据长度
163+
// 是 echo 回复，但 ID 不一致，继续接收
164+
if recvType == 1 && network == "ip" && recvID != id {
165+
continue
166+
}
167+
// 无脑修正数据长度，开头数据其实为 IP 包头
171168
if len(recvData) > len(sendData) {
172169
recvData = recvData[len(recvData)-len(sendData):]
173170
}
174171
// 收到的数据和发送的数据不一致，继续接收
175172
if !bytes.Equal(recvData, sendData) {
176173
continue
177174
}
178-
// 是 echo 回复，但 ID 不一致，继续接收
179-
if recvType == 1 && network == "ip" && recvID != id {
180-
continue
181-
}
182175

183176
if peer != nil {
184177
if _ip := net.ParseIP(peer.String()); _ip != nil {
@@ -241,6 +234,13 @@ func (this *IcmpPing) getconn(network string, ip net.IP, isipv6 bool) (*icmp.Pac
241234
} else {
242235
conn.IPv4PacketConn().SetControlMessage(ipv4.FlagTTL, true)
243236
}
237+
if this.TTL > 0 {
238+
if isipv6 {
239+
conn.IPv6PacketConn().SetHopLimit(this.TTL)
240+
} else {
241+
conn.IPv4PacketConn().SetTTL(this.TTL)
242+
}
243+
}
244244
return conn, nil
245245
}
246246

‎pkg/ping/quic.go‎

Lines changed: 106 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,106 @@
1+
package ping
2+
3+
import (
4+
"context"
5+
"crypto/tls"
6+
"fmt"
7+
"net"
8+
"time"
9+
10+
"github.com/quic-go/quic-go"
11+
"github.com/quic-go/quic-go/http3"
12+
)
13+
14+
type QuicPingResult struct {
15+
Time int
16+
Err error
17+
IP net.IP
18+
QUICVersion uint32
19+
TLSVersion uint16
20+
}
21+
22+
func (this *QuicPingResult) Result() int {
23+
return this.Time
24+
}
25+
26+
func (this *QuicPingResult) Error() error {
27+
return this.Err
28+
}
29+
30+
func (this *QuicPingResult) String() string {
31+
if this.Err != nil {
32+
return fmt.Sprintf("%s", this.Err)
33+
} else {
34+
return fmt.Sprintf("%s: quic=%s, tls=%s, time=%d ms", this.IP.String(), quic.VersionNumber(this.QUICVersion).String(), tlsVersionToString(this.TLSVersion), this.Time)
35+
}
36+
}
37+
38+
type QuicPing struct {
39+
Host string
40+
Port uint16
41+
Timeout time.Duration
42+
43+
// 以下为可选参数
44+
Insecure bool
45+
ALPN string
46+
IP net.IP
47+
}
48+
49+
func (this *QuicPing) Ping() IPingResult {
50+
return this.PingContext(context.Background())
51+
}
52+
53+
func (this *QuicPing) PingContext(ctx context.Context) IPingResult {
54+
ip := cloneIP(this.IP)
55+
if ip == nil {
56+
var err error
57+
ip, err = LookupFunc(this.Host)
58+
if err != nil {
59+
return this.errorResult(err)
60+
}
61+
}
62+
addr := net.JoinHostPort(ip.String(), fmt.Sprint(this.Port))
63+
64+
alpn := http3.NextProtoH3
65+
if this.ALPN != "" {
66+
alpn = this.ALPN
67+
}
68+
tlsconf := tls.Config{
69+
ServerName: this.Host,
70+
InsecureSkipVerify: this.Insecure,
71+
NextProtos: []string{alpn},
72+
}
73+
quicconf := quic.Config{
74+
HandshakeIdleTimeout: this.Timeout,
75+
}
76+
t0 := time.Now()
77+
conn, err := quic.DialAddrContext(ctx, addr, &tlsconf, &quicconf)
78+
if err != nil {
79+
return this.errorResult(err)
80+
}
81+
closecode := uint64(http3.ErrCodeNoError)
82+
if alpn != http3.NextProtoH3 {
83+
closecode = 0
84+
}
85+
defer conn.CloseWithError(quic.ApplicationErrorCode(closecode), "")
86+
return &QuicPingResult{int(time.Since(t0).Milliseconds()), nil, ip, uint32(conn.ConnectionState().Version), conn.ConnectionState().TLS.Version}
87+
}
88+
89+
func NewQuicPing(host string, port uint16, timeout time.Duration) *QuicPing {
90+
return &QuicPing{
91+
Host: host,
92+
Port: port,
93+
Timeout: timeout,
94+
}
95+
}
96+
97+
func (this *QuicPing) errorResult(err error) *QuicPingResult {
98+
r := &QuicPingResult{}
99+
r.Err = err
100+
return r
101+
}
102+
103+
var (
104+
_ IPing = (*QuicPing)(nil)
105+
_ IPingResult = (*QuicPingResult)(nil)
106+
)

‎test/pping_test.go‎

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -91,3 +91,11 @@ func TestIcmp_size_root(t *testing.T) {
9191
t.Fatal(result.Error())
9292
}
9393
}
94+
95+
func TestQuic(t *testing.T) {
96+
p := ping.NewQuicPing("quic.nginx.org", 443, time.Second*5)
97+
result := p.Ping()
98+
if result.Error() != nil {
99+
t.Fatal(result.Error())
100+
}
101+
}

0 commit comments

Comments
 (0)