diff options
Diffstat (limited to 'main.go')
-rw-r--r-- | main.go | 8 |
1 files changed, 6 insertions, 2 deletions
@@ -97,14 +97,18 @@ func DoAddress(host, network, address string) Point { return NewPoint("public", tags, result) } +var dialer = net.Dialer{ + Timeout: 10 * time.Second, +} + func Hello(addr *net.TCPAddr) (name, version string, err error) { - conn, err := net.DialTCP(addr.Network(), nil, addr) + conn, err := dialer.Dial(addr.Network(), addr.String()) if err != nil { return "", "", err } defer conn.Close() conn.SetDeadline(time.Now().Add(10 * time.Second)) - conn.CloseWrite() + conn.(*net.TCPConn).CloseWrite() all, _ := ioutil.ReadAll(conn) line := strings.TrimRight(string(all), "\n") parts := strings.Split(line, " ") |