ff835db955
bencode: * Handle omitempty on non-trivial types. cmd/dht-ping: * Handle timeouts in ping transactions. dht: * Propagate failed transaction responses properly. * Msg related tests are moved into their own file. * In some places, IPs in binary form are shorted to 4 bytes if IPv4.
86 lines
2.4 KiB
Go
86 lines
2.4 KiB
Go
package dht
|
||
|
||
import (
|
||
"net"
|
||
"testing"
|
||
|
||
"github.com/anacrolix/torrent/bencode"
|
||
"github.com/anacrolix/torrent/util"
|
||
"github.com/stretchr/testify/assert"
|
||
"github.com/stretchr/testify/require"
|
||
)
|
||
|
||
func testMarshalUnmarshalMsg(t *testing.T, m Msg, expected string) {
|
||
b, err := bencode.Marshal(m)
|
||
require.NoError(t, err)
|
||
assert.Equal(t, expected, string(b))
|
||
var _m Msg
|
||
err = bencode.Unmarshal([]byte(expected), &_m)
|
||
assert.NoError(t, err)
|
||
assert.EqualValues(t, m, _m)
|
||
assert.EqualValues(t, m.A, _m.A)
|
||
assert.EqualValues(t, m.R, _m.R)
|
||
}
|
||
|
||
func TestMarshalUnmarshalMsg(t *testing.T) {
|
||
testMarshalUnmarshalMsg(t, Msg{}, "d1:t0:1:y0:e")
|
||
testMarshalUnmarshalMsg(t, Msg{
|
||
Y: "q",
|
||
Q: "ping",
|
||
T: "hi",
|
||
}, "d1:q4:ping1:t2:hi1:y1:qe")
|
||
testMarshalUnmarshalMsg(t, Msg{
|
||
Y: "e",
|
||
T: "42",
|
||
E: &KRPCError{Code: 200, Msg: "fuck"},
|
||
}, "d1:eli200e4:fucke1:t2:421:y1:ee")
|
||
testMarshalUnmarshalMsg(t, Msg{
|
||
Y: "r",
|
||
T: "\x8c%",
|
||
R: &Return{},
|
||
}, "d1:rd2:id0:e1:t2:\x8c%1:y1:re")
|
||
testMarshalUnmarshalMsg(t, Msg{
|
||
Y: "r",
|
||
T: "\x8c%",
|
||
R: &Return{
|
||
Nodes: CompactIPv4NodeInfo{
|
||
NodeInfo{
|
||
Addr: newDHTAddr(&net.UDPAddr{
|
||
IP: net.IPv4(1, 2, 3, 4).To4(),
|
||
Port: 0x1234,
|
||
}),
|
||
},
|
||
},
|
||
},
|
||
}, "d1:rd2:id0:5:nodes26:\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x01\x02\x03\x04\x124e1:t2:\x8c%1:y1:re")
|
||
testMarshalUnmarshalMsg(t, Msg{
|
||
Y: "r",
|
||
T: "\x8c%",
|
||
R: &Return{
|
||
Values: []util.CompactPeer{
|
||
util.CompactPeer{
|
||
IP: net.IPv4(1, 2, 3, 4).To4(),
|
||
Port: 0x5678,
|
||
},
|
||
},
|
||
},
|
||
}, "d1:rd2:id0:6:valuesl6:\x01\x02\x03\x04\x56\x78ee1:t2:\x8c%1:y1:re")
|
||
testMarshalUnmarshalMsg(t, Msg{
|
||
Y: "r",
|
||
T: "\x03",
|
||
R: &Return{
|
||
ID: "\xeb\xff6isQ\xffJ\xec)ͺ\xab\xf2\xfb\xe3F|\xc2g",
|
||
},
|
||
IP: util.CompactPeer{net.IPv4(124, 168, 180, 8).To4(), 62844},
|
||
}, "d2:ip6:|\xa8\xb4\b\xf5|1:rd2:id20:\xeb\xff6isQ\xffJ\xec)ͺ\xab\xf2\xfb\xe3F|\xc2ge1:t1:\x031:y1:re")
|
||
}
|
||
|
||
func TestUnmarshalGetPeersResponse(t *testing.T) {
|
||
var msg Msg
|
||
err := bencode.Unmarshal([]byte("d1:rd6:valuesl6:\x01\x02\x03\x04\x05\x066:\x07\x08\x09\x0a\x0b\x0ce5:nodes52:\x02\x03\x04\x05\x06\x07\x08\x09\x0a\x0b\x0c\x0d\x02\x03\x04\x05\x06\x07\x08\x09\x02\x03\x04\x05\x06\x07\x02\x03\x04\x05\x06\x07\x08\x09\x0a\x0b\x0c\x0d\x02\x03\x04\x05\x06\x07\x08\x09\x02\x03\x04\x05\x06\x07ee"), &msg)
|
||
require.NoError(t, err)
|
||
assert.Len(t, msg.R.Values, 2)
|
||
assert.Len(t, msg.R.Nodes, 2)
|
||
assert.Nil(t, msg.E)
|
||
}
|