certusone
/
tpuproxy
mirror of https://github.com/certusone/tpuproxy.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

cmd/quictpu/ping: add -count and -delay

This commit is contained in:
Leopold Schabel 2022-06-19 13:14:16 +02:00
parent 74ef2c4801
commit 6787f534c7
1 changed files with 24 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
cmd/quictpu/ping/ping.go
View File

@ -22,6 +22,8 @@ import (
var (
flagDebug = flag.Bool("debug", false, "Enable debug logging")
flagCount = flag.Int("count", 1, "Number of pings to send, -1 for infinite")
flagDelay = flag.Duration("delay", 1*time.Second, "Delay between pings")
)
func init() {
@ -62,20 +64,30 @@ func main() {
KeyLogWriter: dbg,
}
t := time.Now()
conn, err := quic.DialAddrContext(ctx, flag.Args()[0], tlsConf, &qconf)
if err != nil {
klog.Exitf("Failed to dial: %v", err)
}
c := 0
klog.Infof("Connected to %s (%dms)", flag.Args()[0], time.Since(t).Milliseconds())
for c < *flagCount || *flagCount == -1 {
t := time.Now()
conn, err := quic.DialAddrContext(ctx, flag.Args()[0], tlsConf, &qconf)
if err != nil {
klog.Exitf("Failed to dial: %v", err)
}
for _, cert := range conn.ConnectionState().TLS.PeerCertificates {
klog.Infof("Certificate: %s", cert.Subject)
klog.Infof("Public key: %s", base58.Encode(cert.PublicKey.(ed25519.PublicKey)))
}
klog.Infof("Connected to %s (in %dms, %d/%d)",
flag.Args()[0], time.Since(t).Milliseconds(),
c+1, *flagCount)
if err := conn.CloseWithError(0, ""); err != nil {
klog.Exitf("Failed to close: %v", err)
for _, cert := range conn.ConnectionState().TLS.PeerCertificates {
klog.Infof("Certificate: %s", cert.Subject)
klog.Infof("Public key: %s", base58.Encode(cert.PublicKey.(ed25519.PublicKey)))
}
if err := conn.CloseWithError(0, ""); err != nil {
klog.Exitf("Failed to close: %v", err)
}
time.Sleep(*flagDelay)
c++
}
}