diff options
Diffstat (limited to 'device/allowedips_rand_test.go')
| -rw-r--r-- | device/allowedips_rand_test.go | 13 |
1 files changed, 11 insertions, 2 deletions
diff --git a/device/allowedips_rand_test.go b/device/allowedips_rand_test.go index 0d3eecb..8c17d02 100644 --- a/device/allowedips_rand_test.go +++ b/device/allowedips_rand_test.go @@ -40,9 +40,18 @@ func (r SlowRouter) Swap(i, j int) { r[i], r[j] = r[j], r[i] } +func commonBitsSlice(addr1, addr2 []byte) uint8 { + if len(addr1) == 4 { + return commonBits4(*(*[4]byte)(addr1), *(*[4]byte)(addr2)) + } else if len(addr1) == 16 { + return commonBits16(*(*[16]byte)(addr1), *(*[16]byte)(addr2)) + } + return 0 +} + func (r SlowRouter) Insert(addr []byte, cidr uint8, peer *Peer) SlowRouter { for _, t := range r { - if t.cidr == cidr && commonBits(t.bits, addr) >= cidr { + if t.cidr == cidr && commonBitsSlice(t.bits, addr) >= cidr { t.peer = peer t.bits = addr return r @@ -59,7 +68,7 @@ func (r SlowRouter) Insert(addr []byte, cidr uint8, peer *Peer) SlowRouter { func (r SlowRouter) Lookup(addr []byte) *Peer { for _, t := range r { - common := commonBits(t.bits, addr) + common := commonBitsSlice(t.bits, addr) if common >= t.cidr { return t.peer } |