46 lines
1.0 KiB
Go
46 lines
1.0 KiB
Go
|
package torrent
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"net"
|
||
|
|
|
||
|
|
"github.com/anacrolix/missinggo/perf"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Dialer interface {
|
||
|
|
// The network is implied by the instance.
|
||
|
|
Dial(_ context.Context, addr string) (net.Conn, error)
|
||
|
|
// This is required for registering with the connection tracker (router connection table
|
||
|
|
// emulating rate-limiter) before dialing. TODO: What about connections that wouldn't infringe
|
||
|
|
// on routers, like localhost or unix sockets.
|
||
|
|
LocalAddr() net.Addr
|
||
|
|
}
|
||
|
|
|
||
|
|
type NetDialer struct {
|
||
|
|
Network string
|
||
|
|
Dialer net.Dialer
|
||
|
|
}
|
||
|
|
|
||
|
|
func (me NetDialer) Dial(ctx context.Context, addr string) (_ net.Conn, err error) {
|
||
|
|
defer perf.ScopeTimerErr(&err)()
|
||
|
|
return me.Dialer.DialContext(ctx, me.Network, addr)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (me NetDialer) LocalAddr() net.Addr {
|
||
|
|
return netDialerLocalAddr{me.Network, me.Dialer.LocalAddr}
|
||
|
|
}
|
||
|
|
|
||
|
|
type netDialerLocalAddr struct {
|
||
|
|
network string
|
||
|
|
addr net.Addr
|
||
|
|
}
|
||
|
|
|
||
|
|
func (me netDialerLocalAddr) Network() string { return me.network }
|
||
|
|
|
||
|
|
func (me netDialerLocalAddr) String() string {
|
||
|
|
if me.addr == nil {
|
||
|
|
return ""
|
||
|
|
}
|
||
|
|
return me.addr.String()
|
||
|
|
}
|