githubmirror/syncthing

lib/osutil/ping.go

40 строк1 KB

Jakob Borg

lib/dialer: Preserve nilness in error handling (fixes #6368) (#6369)

26 фев 2020, 15:16

26 фев 2020, 15:164e4b9a8

// Copyright (C) 2015 The Syncthing Authors.
//
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this file,
// You can obtain one at https://mozilla.org/MPL/2.0/.

package osutil

import (
	"context"
	"net/url"
	"time"

	"github.com/syncthing/syncthing/lib/dialer"
)

// TCPPing returns the duration required to establish a TCP connection
// to the given host. ICMP packets require root privileges, hence why we use
// tcp.
func TCPPing(ctx context.Context, address string) (time.Duration, error) {
	start := time.Now()
	ctx, cancel := context.WithTimeout(ctx, time.Second)
	defer cancel()
	conn, err := dialer.DialContext(ctx, "tcp", address)
	if err == nil {
		conn.Close()
	}
	return time.Since(start), err
}

// GetLatencyForURL parses the given URL, tries opening a TCP connection to it
// and returns the time it took to establish the connection.
func GetLatencyForURL(ctx context.Context, addr string) (time.Duration, error) {
	uri, err := url.Parse(addr)
	if err != nil {
		return 0, err
	}

	return TCPPing(ctx, uri.Host)
}