diff options
| author |   Jason A. Donenfeld <Jason@zx2c4.com> | 2021-06-03 13:51:03 +0200 |
|---|---|---|
| committer | Jason A. Donenfeld <Jason@zx2c4.com> | 2021-06-03 13:51:03 +0200 |
| commit | 4a57024b94edf23a20f1e4289052d0717227683b (patch) | |
| tree | 1449ee35b5f8e8d585504547cb855acc78153dff /device/allowedips_rand_test.go | |
| parent | go.mod: bump golang.org/x/sys again (diff) | |
| download | wireguard-go-4a57024b94edf23a20f1e4289052d0717227683b.tar.xz wireguard-go-4a57024b94edf23a20f1e4289052d0717227683b.zip | |
device: reduce size of trie struct
Signed-off-by: Jason A. Donenfeld <Jason@zx2c4.com>
Diffstat (limited to 'device/allowedips_rand_test.go')
| -rw-r--r-- | device/allowedips_rand_test.go | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/device/allowedips_rand_test.go b/device/allowedips_rand_test.go index bb3fb43..2da8795 100644 --- a/device/allowedips_rand_test.go +++ b/device/allowedips_rand_test.go @@ -19,7 +19,7 @@ const ( type SlowNode struct { peer *Peer - cidr uint + cidr uint8 bits []byte } @@ -37,7 +37,7 @@ 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 { +func (r SlowRouter) Insert(addr []byte, cidr uint8, peer *Peer) SlowRouter { for _, t := range r { if t.cidr == cidr && commonBits(t.bits, addr) >= cidr { t.peer = peer @@ -80,7 +80,7 @@ func TestTrieRandomIPv4(t *testing.T) { for n := 0; n < NumberOfAddresses; n++ { var addr [AddressLength]byte rand.Read(addr[:]) - cidr := uint(rand.Uint32() % (AddressLength * 8)) + cidr := uint8(rand.Uint32() % (AddressLength * 8)) index := rand.Int() % NumberOfPeers trie = trie.insert(addr[:], cidr, peers[index]) slow = slow.Insert(addr[:], cidr, peers[index]) @@ -113,7 +113,7 @@ func TestTrieRandomIPv6(t *testing.T) { for n := 0; n < NumberOfAddresses; n++ { var addr [AddressLength]byte rand.Read(addr[:]) - cidr := uint(rand.Uint32() % (AddressLength * 8)) + cidr := uint8(rand.Uint32() % (AddressLength * 8)) index := rand.Int() % NumberOfPeers trie = trie.insert(addr[:], cidr, peers[index]) slow = slow.Insert(addr[:], cidr, peers[index]) |