From 724e6cb37d8a32835d50b98543b6df50ce44d6a0 Mon Sep 17 00:00:00 2001
From: Jeromy <jeromyj@gmail.com>
Date: Sun, 8 Nov 2015 11:04:37 -0800
Subject: [PATCH] reuseport tcp is now a dial creation option

And other assorted PR feedback

License: MIT
Signed-off-by: Jeromy <jeromyj@gmail.com>
---
 p2p/net/conn/dial_test.go      |  2 +-
 p2p/net/swarm/swarm.go         |  4 ++--
 p2p/net/swarm/swarm_listen.go  |  6 +++---
 p2p/net/transport/tcp.go       | 33 ++++++++++++++++++---------------
 p2p/net/transport/transport.go |  8 ++++++--
 5 files changed, 30 insertions(+), 23 deletions(-)

diff --git a/p2p/net/conn/dial_test.go b/p2p/net/conn/dial_test.go
index 3206b16cb..78c9d1d12 100644
--- a/p2p/net/conn/dial_test.go
+++ b/p2p/net/conn/dial_test.go
@@ -54,7 +54,7 @@ func setupSingleConn(t *testing.T, ctx context.Context) (a, b Conn, p1, p2 tu.Pe
 }
 
 func Listen(ctx context.Context, addr ma.Multiaddr, local peer.ID, sk ic.PrivKey) (Listener, error) {
-	list, err := transport.NewTCPTransport().Listener(addr)
+	list, err := transport.NewTCPTransport().Listen(addr)
 	if err != nil {
 		return nil, err
 	}
diff --git a/p2p/net/swarm/swarm.go b/p2p/net/swarm/swarm.go
index 4a1eaa40d..56c268fb9 100644
--- a/p2p/net/swarm/swarm.go
+++ b/p2p/net/swarm/swarm.go
@@ -114,7 +114,7 @@ func NewSwarm(ctx context.Context, listenAddrs []ma.Multiaddr,
 	prom.MustRegisterOrGet(peersTotal)
 	s.Notify((*metricsNotifiee)(s))
 
-	return s, s.setupAddresses(listenAddrs)
+	return s, s.setupInterfaces(listenAddrs)
 }
 
 func (s *Swarm) teardown() error {
@@ -147,7 +147,7 @@ func (s *Swarm) Listen(addrs ...ma.Multiaddr) error {
 		return err
 	}
 
-	return s.setupAddresses(addrs)
+	return s.setupInterfaces(addrs)
 }
 
 // Process returns the Process of the swarm
diff --git a/p2p/net/swarm/swarm_listen.go b/p2p/net/swarm/swarm_listen.go
index 099e49e5f..143ff0e94 100644
--- a/p2p/net/swarm/swarm_listen.go
+++ b/p2p/net/swarm/swarm_listen.go
@@ -15,21 +15,21 @@ import (
 )
 
 // Open listeners and reuse-dialers for the given addresses
-func (s *Swarm) setupAddresses(addrs []ma.Multiaddr) error {
+func (s *Swarm) setupInterfaces(addrs []ma.Multiaddr) error {
 	for _, a := range addrs {
 		tpt := s.transportForAddr(a)
 		if tpt == nil {
 			return fmt.Errorf("no transport for address: %s", a)
 		}
 
-		d, err := tpt.Dialer(a, transport.TimeoutOpt(DialTimeout))
+		d, err := tpt.Dialer(a, transport.TimeoutOpt(DialTimeout), transport.ReusePorts)
 		if err != nil {
 			return err
 		}
 
 		s.dialer.AddDialer(d)
 
-		list, err := tpt.Listener(a)
+		list, err := tpt.Listen(a)
 		if err != nil {
 			return err
 		}
diff --git a/p2p/net/transport/tcp.go b/p2p/net/transport/tcp.go
index 2f01be681..8b200bbd0 100644
--- a/p2p/net/transport/tcp.go
+++ b/p2p/net/transport/tcp.go
@@ -38,16 +38,19 @@ func (t *TcpTransport) Dialer(laddr ma.Multiaddr, opts ...DialOpt) (Dialer, erro
 	}
 	var base manet.Dialer
 
+	var doReuse bool
 	for _, o := range opts {
 		switch o := o.(type) {
 		case TimeoutOpt:
-			base.Timeout = o.(time.Duration)
+			base.Timeout = time.Duration(o)
+		case ReuseportOpt:
+			doReuse = bool(o)
 		default:
 			return nil, fmt.Errorf("unrecognized option: %#v", o)
 		}
 	}
 
-	tcpd, err := t.newTcpDialer(base, laddr)
+	tcpd, err := t.newTcpDialer(base, laddr, doReuse)
 	if err != nil {
 		return nil, err
 	}
@@ -56,7 +59,7 @@ func (t *TcpTransport) Dialer(laddr ma.Multiaddr, opts ...DialOpt) (Dialer, erro
 	return tcpd, nil
 }
 
-func (t *TcpTransport) Listener(laddr ma.Multiaddr) (Listener, error) {
+func (t *TcpTransport) Listen(laddr ma.Multiaddr) (Listener, error) {
 	t.llock.Lock()
 	defer t.llock.Unlock()
 	s := laddr.String()
@@ -114,33 +117,33 @@ type tcpDialer struct {
 	transport Transport
 }
 
-func (t *TcpTransport) newTcpDialer(base manet.Dialer, laddr ma.Multiaddr) (*tcpDialer, error) {
+func (t *TcpTransport) newTcpDialer(base manet.Dialer, laddr ma.Multiaddr, doReuse bool) (*tcpDialer, error) {
 	// get the local net.Addr manually
 	la, err := manet.ToNetAddr(laddr)
 	if err != nil {
 		return nil, err // something wrong with laddr.
 	}
 
-	if !ReuseportIsAvailable() {
+	if doReuse && ReuseportIsAvailable() {
+		rd := reuseport.Dialer{
+			D: net.Dialer{
+				LocalAddr: la,
+				Timeout:   base.Timeout,
+			},
+		}
+
 		return &tcpDialer{
-			doReuse:   false,
+			doReuse:   true,
 			laddr:     laddr,
+			rd:        rd,
 			madialer:  base,
 			transport: t,
 		}, nil
 	}
 
-	rd := reuseport.Dialer{
-		D: net.Dialer{
-			LocalAddr: la,
-			Timeout:   base.Timeout,
-		},
-	}
-
 	return &tcpDialer{
-		doReuse:   true,
+		doReuse:   false,
 		laddr:     laddr,
-		rd:        rd,
 		madialer:  base,
 		transport: t,
 	}, nil
diff --git a/p2p/net/transport/transport.go b/p2p/net/transport/transport.go
index 9ffef74b2..e75f43029 100644
--- a/p2p/net/transport/transport.go
+++ b/p2p/net/transport/transport.go
@@ -2,6 +2,7 @@ package transport
 
 import (
 	"net"
+	"time"
 
 	ma "github.com/ipfs/go-ipfs/Godeps/_workspace/src/github.com/jbenet/go-multiaddr"
 	manet "github.com/ipfs/go-ipfs/Godeps/_workspace/src/github.com/jbenet/go-multiaddr-net"
@@ -18,7 +19,7 @@ type Conn interface {
 
 type Transport interface {
 	Dialer(laddr ma.Multiaddr, opts ...DialOpt) (Dialer, error)
-	Listener(laddr ma.Multiaddr) (Listener, error)
+	Listen(laddr ma.Multiaddr) (Listener, error)
 	Matches(ma.Multiaddr) bool
 }
 
@@ -44,7 +45,10 @@ func (cw *connWrap) Transport() Transport {
 }
 
 type DialOpt interface{}
-type TimeoutOpt interface{}
+type TimeoutOpt time.Duration
+type ReuseportOpt bool
+
+var ReusePorts ReuseportOpt = true
 
 func IsTcpMultiaddr(a ma.Multiaddr) bool {
 	p := a.Protocols()