network_test.go

   1 package torrent
   2
   3 import (
   4         "net"
   5         "testing"
   6
   7         "github.com/anacrolix/missinggo/v2"
   8         "github.com/stretchr/testify/assert"
   9         "github.com/stretchr/testify/require"
  10 )
  11
  12 func testListenerNetwork(
  13         t *testing.T,
  14         listenFunc func(net, addr string) (net.Listener, error),
  15         expectedNet, givenNet, addr string, validIp4 bool,
  16 ) {
  17         l, err := listenFunc(givenNet, addr)
  18         require.NoError(t, err)
  19         defer l.Close()
  20         assert.EqualValues(t, expectedNet, l.Addr().Network())
  21         ip := missinggo.AddrIP(l.Addr())
  22         assert.Equal(t, validIp4, ip.To4() != nil, ip)
  23 }
  24
  25 func listenUtpListener(net, addr string) (l net.Listener, err error) {
  26         l, err = NewUtpSocket(net, addr, nil)
  27         return
  28 }
  29
  30 func testAcceptedConnAddr(
  31         t *testing.T,
  32         network string, valid4 bool,
  33         dial func(addr string) (net.Conn, error),
  34         listen func() (net.Listener, error),
  35 ) {
  36         l, err := listen()
  37         require.NoError(t, err)
  38         defer l.Close()
  39         done := make(chan struct{})
  40         defer close(done)
  41         go func() {
  42                 c, err := dial(l.Addr().String())
  43                 require.NoError(t, err)
  44                 <-done
  45                 c.Close()
  46         }()
  47         c, err := l.Accept()
  48         require.NoError(t, err)
  49         defer c.Close()
  50         assert.EqualValues(t, network, c.RemoteAddr().Network())
  51         assert.Equal(t, valid4, missinggo.AddrIP(c.RemoteAddr()).To4() != nil)
  52 }
  53
  54 func listenClosure(rawListenFunc func(string, string) (net.Listener, error), network, addr string) func() (net.Listener, error) {
  55         return func() (net.Listener, error) {
  56                 return rawListenFunc(network, addr)
  57         }
  58 }
  59
  60 func dialClosure(f func(net, addr string) (net.Conn, error), network string) func(addr string) (net.Conn, error) {
  61         return func(addr string) (net.Conn, error) {
  62                 return f(network, addr)
  63         }
  64 }
  65
  66 func TestListenLocalhostNetwork(t *testing.T) {
  67         testListenerNetwork(t, net.Listen, "tcp", "tcp", "0.0.0.0:0", false)
  68         testListenerNetwork(t, net.Listen, "tcp", "tcp", "[::1]:0", false)
  69         testListenerNetwork(t, listenUtpListener, "udp", "udp6", "[::1]:0", false)
  70         testListenerNetwork(t, listenUtpListener, "udp", "udp6", "[::]:0", false)
  71         testListenerNetwork(t, listenUtpListener, "udp", "udp4", "localhost:0", true)
  72
  73         testAcceptedConnAddr(t, "tcp", false, dialClosure(net.Dial, "tcp"), listenClosure(net.Listen, "tcp6", ":0"))
  74 }