]> Sergey Matveev's repositories - btrtrc.git/commitdiff
Add default DHT servers when UTP disabled
authorMatt Joiner <anacrolix@gmail.com>
Sun, 4 Nov 2018 06:43:47 +0000 (17:43 +1100)
committerMatt Joiner <anacrolix@gmail.com>
Sun, 4 Nov 2018 06:43:47 +0000 (17:43 +1100)
Also ensures that UDP and TCP sockets are reserved, even if not used, and fixes tests. Fixes #289.

client.go
client_test.go

index 036154ab092e8606725a86b806ba7a02e187eee3..2c402198fcc1c65e3647bf5ef7ded546be43028f 100644 (file)
--- a/client.go
+++ b/client.go
@@ -230,7 +230,7 @@ func NewClient(cfg *ClientConfig) (cl *Client, err error) {
                }
        }
 
-       cl.conns, err = listenAll(cl.enabledPeerNetworks(), cl.config.ListenHost, cl.config.ListenPort, cl.config.ProxyURL, cl.firewallCallback)
+       cl.conns, err = listenAll(allPeerNetworks, cl.config.ListenHost, cl.config.ListenPort, cl.config.ProxyURL, cl.firewallCallback)
        if err != nil {
                return
        }
index af3457da4e9b564147b91c7bec1970f66a01cf48..4b74a64babfb2120ffc7aa906fa68fb4ceb90d70 100644 (file)
@@ -873,37 +873,21 @@ func TestClientDynamicListenPortAllProtocols(t *testing.T) {
 func TestClientDynamicListenTCPOnly(t *testing.T) {
        cfg := TestingConfig()
        cfg.DisableUTP = true
+       cfg.DisableTCP = false
        cl, err := NewClient(cfg)
        require.NoError(t, err)
        defer cl.Close()
        assert.NotEqual(t, 0, cl.LocalPort())
-       cl.eachListener(func(s socket) bool {
-               assert.True(t, isTcpNetwork(s.Addr().Network()))
-               return true
-       })
 }
 
 func TestClientDynamicListenUTPOnly(t *testing.T) {
        cfg := TestingConfig()
        cfg.DisableTCP = true
+       cfg.DisableUTP = false
        cl, err := NewClient(cfg)
        require.NoError(t, err)
        defer cl.Close()
        assert.NotEqual(t, 0, cl.LocalPort())
-       cl.eachListener(func(s socket) bool {
-               assert.True(t, isUtpNetwork(s.Addr().Network()))
-               return true
-       })
-}
-
-func TestClientDynamicListenPortNoProtocols(t *testing.T) {
-       cfg := TestingConfig()
-       cfg.DisableTCP = true
-       cfg.DisableUTP = true
-       cl, err := NewClient(cfg)
-       require.NoError(t, err)
-       defer cl.Close()
-       assert.Equal(t, 0, cl.LocalPort())
 }
 
 func totalConns(tts []*Torrent) (ret int) {