mirror of
https://github.com/libp2p/go-reuseport.git
synced 2025-10-05 14:56:51 +08:00
avoid returning nil RemoteAddr -- horrible hack
This commit is contained in:
19
interface.go
19
interface.go
@@ -20,6 +20,7 @@ package reuseport
|
||||
import (
|
||||
"errors"
|
||||
"net"
|
||||
"time"
|
||||
)
|
||||
|
||||
// ErrUnsuportedProtocol signals that the protocol is not currently
|
||||
@@ -57,7 +58,23 @@ func Dial(network, laddr, raddr string) (net.Conn, error) {
|
||||
d.D.LocalAddr = netladdr
|
||||
}
|
||||
|
||||
return dial(d.D, network, raddr)
|
||||
// there's a rare case where dial returns successfully but for some reason the
|
||||
// RemoteAddr is not yet set. We wait here a while until it is, and if too long
|
||||
// passes, we fail.
|
||||
c, err := dial(d.D, network, raddr)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
for start := time.Now(); c.RemoteAddr() == nil; {
|
||||
if time.Now().Sub(start) > time.Second {
|
||||
c.Close()
|
||||
return nil, ErrReuseFailed
|
||||
}
|
||||
|
||||
<-time.After(20 * time.Microsecond)
|
||||
}
|
||||
return c, nil
|
||||
}
|
||||
|
||||
// Dialer is used to specify the Dial options, much like net.Dialer.
|
||||
|
Reference in New Issue
Block a user