summaryrefslogtreecommitdiff
path: root/src/trie_rand_test.go
diff options
context:
space:
mode:
authorMathias Hall-Andersen <mathias@hall-andersen.dk>2018-02-04 16:08:26 +0100
committerMathias Hall-Andersen <mathias@hall-andersen.dk>2018-02-04 16:08:26 +0100
commita0f54cbe5ac2cd8b8296c2c57c30029dd349cff0 (patch)
tree64574090d79ff3899c5c18e5268e450028e4656b /src/trie_rand_test.go
parent5871ec04deb8f4715cab37146940baa35c08cbee (diff)
downloadwireguard-go-a0f54cbe5ac2cd8b8296c2c57c30029dd349cff0.tar.gz
wireguard-go-a0f54cbe5ac2cd8b8296c2c57c30029dd349cff0.zip
Align with go library layout
Diffstat (limited to 'src/trie_rand_test.go')
-rw-r--r--src/trie_rand_test.go126
1 files changed, 0 insertions, 126 deletions
diff --git a/src/trie_rand_test.go b/src/trie_rand_test.go
deleted file mode 100644
index 840d269..0000000
--- a/src/trie_rand_test.go
+++ /dev/null
@@ -1,126 +0,0 @@
-package main
-
-import (
- "math/rand"
- "sort"
- "testing"
-)
-
-const (
- NumberOfPeers = 100
- NumberOfAddresses = 250
- NumberOfTests = 10000
-)
-
-type SlowNode struct {
- peer *Peer
- cidr uint
- bits []byte
-}
-
-type SlowRouter []*SlowNode
-
-func (r SlowRouter) Len() int {
- return len(r)
-}
-
-func (r SlowRouter) Less(i, j int) bool {
- return r[i].cidr > r[j].cidr
-}
-
-func (r SlowRouter) Swap(i, j int) {
- r[i], r[j] = r[j], r[i]
-}
-
-func (r SlowRouter) Insert(addr []byte, cidr uint, peer *Peer) SlowRouter {
- for _, t := range r {
- if t.cidr == cidr && commonBits(t.bits, addr) >= cidr {
- t.peer = peer
- t.bits = addr
- return r
- }
- }
- r = append(r, &SlowNode{
- cidr: cidr,
- bits: addr,
- peer: peer,
- })
- sort.Sort(r)
- return r
-}
-
-func (r SlowRouter) Lookup(addr []byte) *Peer {
- for _, t := range r {
- common := commonBits(t.bits, addr)
- if common >= t.cidr {
- return t.peer
- }
- }
- return nil
-}
-
-func TestTrieRandomIPv4(t *testing.T) {
- var trie *Trie
- var slow SlowRouter
- var peers []*Peer
-
- rand.Seed(1)
-
- const AddressLength = 4
-
- for n := 0; n < NumberOfPeers; n += 1 {
- peers = append(peers, &Peer{})
- }
-
- for n := 0; n < NumberOfAddresses; n += 1 {
- var addr [AddressLength]byte
- rand.Read(addr[:])
- cidr := uint(rand.Uint32() % (AddressLength * 8))
- index := rand.Int() % NumberOfPeers
- trie = trie.Insert(addr[:], cidr, peers[index])
- slow = slow.Insert(addr[:], cidr, peers[index])
- }
-
- for n := 0; n < NumberOfTests; n += 1 {
- var addr [AddressLength]byte
- rand.Read(addr[:])
- peer1 := slow.Lookup(addr[:])
- peer2 := trie.Lookup(addr[:])
- if peer1 != peer2 {
- t.Error("Trie did not match naive implementation, for:", addr)
- }
- }
-}
-
-func TestTrieRandomIPv6(t *testing.T) {
- var trie *Trie
- var slow SlowRouter
- var peers []*Peer
-
- rand.Seed(1)
-
- const AddressLength = 16
-
- for n := 0; n < NumberOfPeers; n += 1 {
- peers = append(peers, &Peer{})
- }
-
- for n := 0; n < NumberOfAddresses; n += 1 {
- var addr [AddressLength]byte
- rand.Read(addr[:])
- cidr := uint(rand.Uint32() % (AddressLength * 8))
- index := rand.Int() % NumberOfPeers
- trie = trie.Insert(addr[:], cidr, peers[index])
- slow = slow.Insert(addr[:], cidr, peers[index])
- }
-
- for n := 0; n < NumberOfTests; n += 1 {
- var addr [AddressLength]byte
- rand.Read(addr[:])
- peer1 := slow.Lookup(addr[:])
- peer2 := trie.Lookup(addr[:])
- if peer1 != peer2 {
- t.Error("Trie did not match naive implementation, for:", addr)
- }
- }
-}