udp_test.go 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505
  1. // Copyright 2016 The go-ethereum Authors
  2. // This file is part of the go-ethereum library.
  3. //
  4. // The go-ethereum library is free software: you can redistribute it and/or modify
  5. // it under the terms of the GNU Lesser General Public License as published by
  6. // the Free Software Foundation, either version 3 of the License, or
  7. // (at your option) any later version.
  8. //
  9. // The go-ethereum library is distributed in the hope that it will be useful,
  10. // but WITHOUT ANY WARRANTY; without even the implied warranty of
  11. // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  12. // GNU Lesser General Public License for more details.
  13. //
  14. // You should have received a copy of the GNU Lesser General Public License
  15. // along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
  16. package discv5
  17. import (
  18. "encoding/hex"
  19. "errors"
  20. "io"
  21. "net"
  22. "reflect"
  23. "sync"
  24. "testing"
  25. "time"
  26. "github.com/davecgh/go-spew/spew"
  27. "github.com/ethereum/go-ethereum/common"
  28. "github.com/ethereum/go-ethereum/crypto"
  29. "github.com/ethereum/go-ethereum/rlp"
  30. )
  31. func init() {
  32. spew.Config.DisableMethods = true
  33. }
  34. // This test checks that isPacketTooBig correctly identifies
  35. // errors that result from receiving a UDP packet larger
  36. // than the supplied receive buffer.
  37. func TestIsPacketTooBig(t *testing.T) {
  38. listener, err := net.ListenPacket("udp", "127.0.0.1:0")
  39. if err != nil {
  40. t.Fatal(err)
  41. }
  42. defer listener.Close()
  43. sender, err := net.Dial("udp", listener.LocalAddr().String())
  44. if err != nil {
  45. t.Fatal(err)
  46. }
  47. defer sender.Close()
  48. sendN := 1800
  49. recvN := 300
  50. for i := 0; i < 20; i++ {
  51. go func() {
  52. buf := make([]byte, sendN)
  53. for i := range buf {
  54. buf[i] = byte(i)
  55. }
  56. sender.Write(buf)
  57. }()
  58. buf := make([]byte, recvN)
  59. listener.SetDeadline(time.Now().Add(1 * time.Second))
  60. n, _, err := listener.ReadFrom(buf)
  61. if err != nil {
  62. if nerr, ok := err.(net.Error); ok && nerr.Timeout() {
  63. continue
  64. }
  65. if !isPacketTooBig(err) {
  66. t.Fatal("unexpected read error:", spew.Sdump(err))
  67. }
  68. continue
  69. }
  70. if n != recvN {
  71. t.Fatalf("short read: %d, want %d", n, recvN)
  72. }
  73. for i := range buf {
  74. if buf[i] != byte(i) {
  75. t.Fatalf("error in pattern")
  76. break
  77. }
  78. }
  79. }
  80. }
  81. // shared test variables
  82. var (
  83. futureExp = uint64(time.Now().Add(10 * time.Hour).Unix())
  84. testTarget = NodeID{0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1}
  85. testRemote = rpcEndpoint{IP: net.ParseIP("1.1.1.1").To4(), UDP: 1, TCP: 2}
  86. testLocalAnnounced = rpcEndpoint{IP: net.ParseIP("2.2.2.2").To4(), UDP: 3, TCP: 4}
  87. testLocal = rpcEndpoint{IP: net.ParseIP("3.3.3.3").To4(), UDP: 5, TCP: 6}
  88. )
  89. // type udpTest struct {
  90. // t *testing.T
  91. // pipe *dgramPipe
  92. // table *Table
  93. // udp *udp
  94. // sent [][]byte
  95. // localkey, remotekey *ecdsa.PrivateKey
  96. // remoteaddr *net.UDPAddr
  97. // }
  98. //
  99. // func newUDPTest(t *testing.T) *udpTest {
  100. // test := &udpTest{
  101. // t: t,
  102. // pipe: newpipe(),
  103. // localkey: newkey(),
  104. // remotekey: newkey(),
  105. // remoteaddr: &net.UDPAddr{IP: net.IP{1, 2, 3, 4}, Port: 30303},
  106. // }
  107. // test.table, test.udp, _ = newUDP(test.localkey, test.pipe, nil, "")
  108. // return test
  109. // }
  110. //
  111. // // handles a packet as if it had been sent to the transport.
  112. // func (test *udpTest) packetIn(wantError error, ptype byte, data packet) error {
  113. // enc, err := encodePacket(test.remotekey, ptype, data)
  114. // if err != nil {
  115. // return test.errorf("packet (%d) encode error: %v", ptype, err)
  116. // }
  117. // test.sent = append(test.sent, enc)
  118. // if err = test.udp.handlePacket(test.remoteaddr, enc); err != wantError {
  119. // return test.errorf("error mismatch: got %q, want %q", err, wantError)
  120. // }
  121. // return nil
  122. // }
  123. //
  124. // // waits for a packet to be sent by the transport.
  125. // // validate should have type func(*udpTest, X) error, where X is a packet type.
  126. // func (test *udpTest) waitPacketOut(validate interface{}) error {
  127. // dgram := test.pipe.waitPacketOut()
  128. // p, _, _, err := decodePacket(dgram)
  129. // if err != nil {
  130. // return test.errorf("sent packet decode error: %v", err)
  131. // }
  132. // fn := reflect.ValueOf(validate)
  133. // exptype := fn.Type().In(0)
  134. // if reflect.TypeOf(p) != exptype {
  135. // return test.errorf("sent packet type mismatch, got: %v, want: %v", reflect.TypeOf(p), exptype)
  136. // }
  137. // fn.Call([]reflect.Value{reflect.ValueOf(p)})
  138. // return nil
  139. // }
  140. //
  141. // func (test *udpTest) errorf(format string, args ...interface{}) error {
  142. // _, file, line, ok := runtime.Caller(2) // errorf + waitPacketOut
  143. // if ok {
  144. // file = filepath.Base(file)
  145. // } else {
  146. // file = "???"
  147. // line = 1
  148. // }
  149. // err := fmt.Errorf(format, args...)
  150. // fmt.Printf("\t%s:%d: %v\n", file, line, err)
  151. // test.t.Fail()
  152. // return err
  153. // }
  154. //
  155. // func TestUDP_packetErrors(t *testing.T) {
  156. // test := newUDPTest(t)
  157. // defer test.table.Close()
  158. //
  159. // test.packetIn(errExpired, pingPacket, &ping{From: testRemote, To: testLocalAnnounced, Version: Version})
  160. // test.packetIn(errUnsolicitedReply, pongPacket, &pong{ReplyTok: []byte{}, Expiration: futureExp})
  161. // test.packetIn(errUnknownNode, findnodePacket, &findnode{Expiration: futureExp})
  162. // test.packetIn(errUnsolicitedReply, neighborsPacket, &neighbors{Expiration: futureExp})
  163. // }
  164. //
  165. // func TestUDP_findnode(t *testing.T) {
  166. // test := newUDPTest(t)
  167. // defer test.table.Close()
  168. //
  169. // // put a few nodes into the table. their exact
  170. // // distribution shouldn't matter much, altough we need to
  171. // // take care not to overflow any bucket.
  172. // targetHash := crypto.Keccak256Hash(testTarget[:])
  173. // nodes := &nodesByDistance{target: targetHash}
  174. // for i := 0; i < bucketSize; i++ {
  175. // nodes.push(nodeAtDistance(test.table.self.sha, i+2), bucketSize)
  176. // }
  177. // test.table.stuff(nodes.entries)
  178. //
  179. // // ensure there's a bond with the test node,
  180. // // findnode won't be accepted otherwise.
  181. // test.table.db.updateNode(NewNode(
  182. // PubkeyID(&test.remotekey.PublicKey),
  183. // test.remoteaddr.IP,
  184. // uint16(test.remoteaddr.Port),
  185. // 99,
  186. // ))
  187. // // check that closest neighbors are returned.
  188. // test.packetIn(nil, findnodePacket, &findnode{Target: testTarget, Expiration: futureExp})
  189. // expected := test.table.closest(targetHash, bucketSize)
  190. //
  191. // waitNeighbors := func(want []*Node) {
  192. // test.waitPacketOut(func(p *neighbors) {
  193. // if len(p.Nodes) != len(want) {
  194. // t.Errorf("wrong number of results: got %d, want %d", len(p.Nodes), bucketSize)
  195. // }
  196. // for i := range p.Nodes {
  197. // if p.Nodes[i].ID != want[i].ID {
  198. // t.Errorf("result mismatch at %d:\n got: %v\n want: %v", i, p.Nodes[i], expected.entries[i])
  199. // }
  200. // }
  201. // })
  202. // }
  203. // waitNeighbors(expected.entries[:maxNeighbors])
  204. // waitNeighbors(expected.entries[maxNeighbors:])
  205. // }
  206. //
  207. // func TestUDP_findnodeMultiReply(t *testing.T) {
  208. // test := newUDPTest(t)
  209. // defer test.table.Close()
  210. //
  211. // // queue a pending findnode request
  212. // resultc, errc := make(chan []*Node), make(chan error)
  213. // go func() {
  214. // rid := PubkeyID(&test.remotekey.PublicKey)
  215. // ns, err := test.udp.findnode(rid, test.remoteaddr, testTarget)
  216. // if err != nil && len(ns) == 0 {
  217. // errc <- err
  218. // } else {
  219. // resultc <- ns
  220. // }
  221. // }()
  222. //
  223. // // wait for the findnode to be sent.
  224. // // after it is sent, the transport is waiting for a reply
  225. // test.waitPacketOut(func(p *findnode) {
  226. // if p.Target != testTarget {
  227. // t.Errorf("wrong target: got %v, want %v", p.Target, testTarget)
  228. // }
  229. // })
  230. //
  231. // // send the reply as two packets.
  232. // list := []*Node{
  233. // MustParseNode("enode://ba85011c70bcc5c04d8607d3a0ed29aa6179c092cbdda10d5d32684fb33ed01bd94f588ca8f91ac48318087dcb02eaf36773a7a453f0eedd6742af668097b29c@10.0.1.16:30303?discport=30304"),
  234. // MustParseNode("enode://81fa361d25f157cd421c60dcc28d8dac5ef6a89476633339c5df30287474520caca09627da18543d9079b5b288698b542d56167aa5c09111e55acdbbdf2ef799@10.0.1.16:30303"),
  235. // MustParseNode("enode://9bffefd833d53fac8e652415f4973bee289e8b1a5c6c4cbe70abf817ce8a64cee11b823b66a987f51aaa9fba0d6a91b3e6bf0d5a5d1042de8e9eeea057b217f8@10.0.1.36:30301?discport=17"),
  236. // MustParseNode("enode://1b5b4aa662d7cb44a7221bfba67302590b643028197a7d5214790f3bac7aaa4a3241be9e83c09cf1f6c69d007c634faae3dc1b1221793e8446c0b3a09de65960@10.0.1.16:30303"),
  237. // }
  238. // rpclist := make([]rpcNode, len(list))
  239. // for i := range list {
  240. // rpclist[i] = nodeToRPC(list[i])
  241. // }
  242. // test.packetIn(nil, neighborsPacket, &neighbors{Expiration: futureExp, Nodes: rpclist[:2]})
  243. // test.packetIn(nil, neighborsPacket, &neighbors{Expiration: futureExp, Nodes: rpclist[2:]})
  244. //
  245. // // check that the sent neighbors are all returned by findnode
  246. // select {
  247. // case result := <-resultc:
  248. // if !reflect.DeepEqual(result, list) {
  249. // t.Errorf("neighbors mismatch:\n got: %v\n want: %v", result, list)
  250. // }
  251. // case err := <-errc:
  252. // t.Errorf("findnode error: %v", err)
  253. // case <-time.After(5 * time.Second):
  254. // t.Error("findnode did not return within 5 seconds")
  255. // }
  256. // }
  257. //
  258. // func TestUDP_successfulPing(t *testing.T) {
  259. // test := newUDPTest(t)
  260. // added := make(chan *Node, 1)
  261. // test.table.nodeAddedHook = func(n *Node) { added <- n }
  262. // defer test.table.Close()
  263. //
  264. // // The remote side sends a ping packet to initiate the exchange.
  265. // go test.packetIn(nil, pingPacket, &ping{From: testRemote, To: testLocalAnnounced, Version: Version, Expiration: futureExp})
  266. //
  267. // // the ping is replied to.
  268. // test.waitPacketOut(func(p *pong) {
  269. // pinghash := test.sent[0][:macSize]
  270. // if !bytes.Equal(p.ReplyTok, pinghash) {
  271. // t.Errorf("got pong.ReplyTok %x, want %x", p.ReplyTok, pinghash)
  272. // }
  273. // wantTo := rpcEndpoint{
  274. // // The mirrored UDP address is the UDP packet sender
  275. // IP: test.remoteaddr.IP, UDP: uint16(test.remoteaddr.Port),
  276. // // The mirrored TCP port is the one from the ping packet
  277. // TCP: testRemote.TCP,
  278. // }
  279. // if !reflect.DeepEqual(p.To, wantTo) {
  280. // t.Errorf("got pong.To %v, want %v", p.To, wantTo)
  281. // }
  282. // })
  283. //
  284. // // remote is unknown, the table pings back.
  285. // test.waitPacketOut(func(p *ping) error {
  286. // if !reflect.DeepEqual(p.From, test.udp.ourEndpoint) {
  287. // t.Errorf("got ping.From %v, want %v", p.From, test.udp.ourEndpoint)
  288. // }
  289. // wantTo := rpcEndpoint{
  290. // // The mirrored UDP address is the UDP packet sender.
  291. // IP: test.remoteaddr.IP, UDP: uint16(test.remoteaddr.Port),
  292. // TCP: 0,
  293. // }
  294. // if !reflect.DeepEqual(p.To, wantTo) {
  295. // t.Errorf("got ping.To %v, want %v", p.To, wantTo)
  296. // }
  297. // return nil
  298. // })
  299. // test.packetIn(nil, pongPacket, &pong{Expiration: futureExp})
  300. //
  301. // // the node should be added to the table shortly after getting the
  302. // // pong packet.
  303. // select {
  304. // case n := <-added:
  305. // rid := PubkeyID(&test.remotekey.PublicKey)
  306. // if n.ID != rid {
  307. // t.Errorf("node has wrong ID: got %v, want %v", n.ID, rid)
  308. // }
  309. // if !bytes.Equal(n.IP, test.remoteaddr.IP) {
  310. // t.Errorf("node has wrong IP: got %v, want: %v", n.IP, test.remoteaddr.IP)
  311. // }
  312. // if int(n.UDP) != test.remoteaddr.Port {
  313. // t.Errorf("node has wrong UDP port: got %v, want: %v", n.UDP, test.remoteaddr.Port)
  314. // }
  315. // if n.TCP != testRemote.TCP {
  316. // t.Errorf("node has wrong TCP port: got %v, want: %v", n.TCP, testRemote.TCP)
  317. // }
  318. // case <-time.After(2 * time.Second):
  319. // t.Errorf("node was not added within 2 seconds")
  320. // }
  321. // }
  322. var testPackets = []struct {
  323. input string
  324. wantPacket interface{}
  325. }{
  326. {
  327. input: "71dbda3a79554728d4f94411e42ee1f8b0d561c10e1e5f5893367948c6a7d70bb87b235fa28a77070271b6c164a2dce8c7e13a5739b53b5e96f2e5acb0e458a02902f5965d55ecbeb2ebb6cabb8b2b232896a36b737666c55265ad0a68412f250001ea04cb847f000001820cfa8215a8d790000000000000000000000000000000018208ae820d058443b9a355",
  328. wantPacket: &ping{
  329. Version: 4,
  330. From: rpcEndpoint{net.ParseIP("127.0.0.1").To4(), 3322, 5544},
  331. To: rpcEndpoint{net.ParseIP("::1"), 2222, 3333},
  332. Expiration: 1136239445,
  333. Rest: []rlp.RawValue{},
  334. },
  335. },
  336. {
  337. input: "e9614ccfd9fc3e74360018522d30e1419a143407ffcce748de3e22116b7e8dc92ff74788c0b6663aaa3d67d641936511c8f8d6ad8698b820a7cf9e1be7155e9a241f556658c55428ec0563514365799a4be2be5a685a80971ddcfa80cb422cdd0101ec04cb847f000001820cfa8215a8d790000000000000000000000000000000018208ae820d058443b9a3550102",
  338. wantPacket: &ping{
  339. Version: 4,
  340. From: rpcEndpoint{net.ParseIP("127.0.0.1").To4(), 3322, 5544},
  341. To: rpcEndpoint{net.ParseIP("::1"), 2222, 3333},
  342. Expiration: 1136239445,
  343. Rest: []rlp.RawValue{{0x01}, {0x02}},
  344. },
  345. },
  346. {
  347. input: "577be4349c4dd26768081f58de4c6f375a7a22f3f7adda654d1428637412c3d7fe917cadc56d4e5e7ffae1dbe3efffb9849feb71b262de37977e7c7a44e677295680e9e38ab26bee2fcbae207fba3ff3d74069a50b902a82c9903ed37cc993c50001f83e82022bd79020010db83c4d001500000000abcdef12820cfa8215a8d79020010db885a308d313198a2e037073488208ae82823a8443b9a355c5010203040531b9019afde696e582a78fa8d95ea13ce3297d4afb8ba6433e4154caa5ac6431af1b80ba76023fa4090c408f6b4bc3701562c031041d4702971d102c9ab7fa5eed4cd6bab8f7af956f7d565ee1917084a95398b6a21eac920fe3dd1345ec0a7ef39367ee69ddf092cbfe5b93e5e568ebc491983c09c76d922dc3",
  348. wantPacket: &ping{
  349. Version: 555,
  350. From: rpcEndpoint{net.ParseIP("2001:db8:3c4d:15::abcd:ef12"), 3322, 5544},
  351. To: rpcEndpoint{net.ParseIP("2001:db8:85a3:8d3:1319:8a2e:370:7348"), 2222, 33338},
  352. Expiration: 1136239445,
  353. Rest: []rlp.RawValue{{0xC5, 0x01, 0x02, 0x03, 0x04, 0x05}},
  354. },
  355. },
  356. {
  357. input: "09b2428d83348d27cdf7064ad9024f526cebc19e4958f0fdad87c15eb598dd61d08423e0bf66b2069869e1724125f820d851c136684082774f870e614d95a2855d000f05d1648b2d5945470bc187c2d2216fbe870f43ed0909009882e176a46b0102f846d79020010db885a308d313198a2e037073488208ae82823aa0fbc914b16819237dcd8801d7e53f69e9719adecb3cc0e790c57e91ca4461c9548443b9a355c6010203c2040506a0c969a58f6f9095004c0177a6b47f451530cab38966a25cca5cb58f055542124e",
  358. wantPacket: &pong{
  359. To: rpcEndpoint{net.ParseIP("2001:db8:85a3:8d3:1319:8a2e:370:7348"), 2222, 33338},
  360. ReplyTok: common.Hex2Bytes("fbc914b16819237dcd8801d7e53f69e9719adecb3cc0e790c57e91ca4461c954"),
  361. Expiration: 1136239445,
  362. Rest: []rlp.RawValue{{0xC6, 0x01, 0x02, 0x03, 0xC2, 0x04, 0x05}, {0x06}},
  363. },
  364. },
  365. {
  366. input: "c7c44041b9f7c7e41934417ebac9a8e1a4c6298f74553f2fcfdcae6ed6fe53163eb3d2b52e39fe91831b8a927bf4fc222c3902202027e5e9eb812195f95d20061ef5cd31d502e47ecb61183f74a504fe04c51e73df81f25c4d506b26db4517490103f84eb840ca634cae0d49acb401d8a4c6b6fe8c55b70d115bf400769cc1400f3258cd31387574077f301b421bc84df7266c44e9e6d569fc56be00812904767bf5ccd1fc7f8443b9a35582999983999999280dc62cc8255c73471e0a61da0c89acdc0e035e260add7fc0c04ad9ebf3919644c91cb247affc82b69bd2ca235c71eab8e49737c937a2c396",
  367. wantPacket: &findnode{
  368. Target: MustHexID("ca634cae0d49acb401d8a4c6b6fe8c55b70d115bf400769cc1400f3258cd31387574077f301b421bc84df7266c44e9e6d569fc56be00812904767bf5ccd1fc7f"),
  369. Expiration: 1136239445,
  370. Rest: []rlp.RawValue{{0x82, 0x99, 0x99}, {0x83, 0x99, 0x99, 0x99}},
  371. },
  372. },
  373. {
  374. input: "c679fc8fe0b8b12f06577f2e802d34f6fa257e6137a995f6f4cbfc9ee50ed3710faf6e66f932c4c8d81d64343f429651328758b47d3dbc02c4042f0fff6946a50f4a49037a72bb550f3a7872363a83e1b9ee6469856c24eb4ef80b7535bcf99c0004f9015bf90150f84d846321163782115c82115db8403155e1427f85f10a5c9a7755877748041af1bcd8d474ec065eb33df57a97babf54bfd2103575fa829115d224c523596b401065a97f74010610fce76382c0bf32f84984010203040101b840312c55512422cf9b8a4097e9a6ad79402e87a15ae909a4bfefa22398f03d20951933beea1e4dfa6f968212385e829f04c2d314fc2d4e255e0d3bc08792b069dbf8599020010db83c4d001500000000abcdef12820d05820d05b84038643200b172dcfef857492156971f0e6aa2c538d8b74010f8e140811d53b98c765dd2d96126051913f44582e8c199ad7c6d6819e9a56483f637feaac9448aacf8599020010db885a308d313198a2e037073488203e78203e8b8408dcab8618c3253b558d459da53bd8fa68935a719aff8b811197101a4b2b47dd2d47295286fc00cc081bb542d760717d1bdd6bec2c37cd72eca367d6dd3b9df738443b9a355010203b525a138aa34383fec3d2719a0",
  375. wantPacket: &neighbors{
  376. Nodes: []rpcNode{
  377. {
  378. ID: MustHexID("3155e1427f85f10a5c9a7755877748041af1bcd8d474ec065eb33df57a97babf54bfd2103575fa829115d224c523596b401065a97f74010610fce76382c0bf32"),
  379. IP: net.ParseIP("99.33.22.55").To4(),
  380. UDP: 4444,
  381. TCP: 4445,
  382. },
  383. {
  384. ID: MustHexID("312c55512422cf9b8a4097e9a6ad79402e87a15ae909a4bfefa22398f03d20951933beea1e4dfa6f968212385e829f04c2d314fc2d4e255e0d3bc08792b069db"),
  385. IP: net.ParseIP("1.2.3.4").To4(),
  386. UDP: 1,
  387. TCP: 1,
  388. },
  389. {
  390. ID: MustHexID("38643200b172dcfef857492156971f0e6aa2c538d8b74010f8e140811d53b98c765dd2d96126051913f44582e8c199ad7c6d6819e9a56483f637feaac9448aac"),
  391. IP: net.ParseIP("2001:db8:3c4d:15::abcd:ef12"),
  392. UDP: 3333,
  393. TCP: 3333,
  394. },
  395. {
  396. ID: MustHexID("8dcab8618c3253b558d459da53bd8fa68935a719aff8b811197101a4b2b47dd2d47295286fc00cc081bb542d760717d1bdd6bec2c37cd72eca367d6dd3b9df73"),
  397. IP: net.ParseIP("2001:db8:85a3:8d3:1319:8a2e:370:7348"),
  398. UDP: 999,
  399. TCP: 1000,
  400. },
  401. },
  402. Expiration: 1136239445,
  403. Rest: []rlp.RawValue{{0x01}, {0x02}, {0x03}},
  404. },
  405. },
  406. }
  407. func TestForwardCompatibility(t *testing.T) {
  408. t.Skip("skipped while working on discovery v5")
  409. testkey, _ := crypto.HexToECDSA("b71c71a67e1177ad4e901695e1b4b9ee17ae16c6668d313eac2f96dbcda3f291")
  410. wantNodeID := PubkeyID(&testkey.PublicKey)
  411. for _, test := range testPackets {
  412. input, err := hex.DecodeString(test.input)
  413. if err != nil {
  414. t.Fatalf("invalid hex: %s", test.input)
  415. }
  416. var pkt ingressPacket
  417. if err := decodePacket(input, &pkt); err != nil {
  418. t.Errorf("did not accept packet %s\n%v", test.input, err)
  419. continue
  420. }
  421. if !reflect.DeepEqual(pkt.data, test.wantPacket) {
  422. t.Errorf("got %s\nwant %s", spew.Sdump(pkt.data), spew.Sdump(test.wantPacket))
  423. }
  424. if pkt.remoteID != wantNodeID {
  425. t.Errorf("got id %v\nwant id %v", pkt.remoteID, wantNodeID)
  426. }
  427. }
  428. }
  429. // dgramPipe is a fake UDP socket. It queues all sent datagrams.
  430. type dgramPipe struct {
  431. mu *sync.Mutex
  432. cond *sync.Cond
  433. closing chan struct{}
  434. closed bool
  435. queue [][]byte
  436. }
  437. func newpipe() *dgramPipe {
  438. mu := new(sync.Mutex)
  439. return &dgramPipe{
  440. closing: make(chan struct{}),
  441. cond: &sync.Cond{L: mu},
  442. mu: mu,
  443. }
  444. }
  445. // WriteToUDP queues a datagram.
  446. func (c *dgramPipe) WriteToUDP(b []byte, to *net.UDPAddr) (n int, err error) {
  447. msg := make([]byte, len(b))
  448. copy(msg, b)
  449. c.mu.Lock()
  450. defer c.mu.Unlock()
  451. if c.closed {
  452. return 0, errors.New("closed")
  453. }
  454. c.queue = append(c.queue, msg)
  455. c.cond.Signal()
  456. return len(b), nil
  457. }
  458. // ReadFromUDP just hangs until the pipe is closed.
  459. func (c *dgramPipe) ReadFromUDP(b []byte) (n int, addr *net.UDPAddr, err error) {
  460. <-c.closing
  461. return 0, nil, io.EOF
  462. }
  463. func (c *dgramPipe) Close() error {
  464. c.mu.Lock()
  465. defer c.mu.Unlock()
  466. if !c.closed {
  467. close(c.closing)
  468. c.closed = true
  469. }
  470. return nil
  471. }
  472. func (c *dgramPipe) LocalAddr() net.Addr {
  473. return &net.UDPAddr{IP: testLocal.IP, Port: int(testLocal.UDP)}
  474. }
  475. func (c *dgramPipe) waitPacketOut() []byte {
  476. c.mu.Lock()
  477. defer c.mu.Unlock()
  478. for len(c.queue) == 0 {
  479. c.cond.Wait()
  480. }
  481. p := c.queue[0]
  482. copy(c.queue, c.queue[1:])
  483. c.queue = c.queue[:len(c.queue)-1]
  484. return p
  485. }