diff options
| author |   Jason A. Donenfeld <Jason@zx2c4.com> | 2021-06-04 16:33:28 +0200 |
|---|---|---|
| committer | Jason A. Donenfeld <Jason@zx2c4.com> | 2021-06-04 16:33:28 +0200 |
| commit | f9b48a961cd271bcc58c4c76b61a84a139e76167 (patch) | |
| tree | 7a8f380838ba5844c04e75269e075ddfe373256d /device/allowedips_test.go | |
| parent | device: limit allowedip fuzzer a to 4 times through (diff) | |
| download | wireguard-go-f9b48a961cd271bcc58c4c76b61a84a139e76167.tar.xz wireguard-go-f9b48a961cd271bcc58c4c76b61a84a139e76167.zip | |
device: zero out allowedip node pointers when removing
This should make it a bit easier for the garbage collector.
Signed-off-by: Jason A. Donenfeld <Jason@zx2c4.com>
Diffstat (limited to '')
| -rw-r--r-- | device/allowedips_test.go | 11 |
1 files changed, 10 insertions, 1 deletions
diff --git a/device/allowedips_test.go b/device/allowedips_test.go index 7701cde..2059a88 100644 --- a/device/allowedips_test.go +++ b/device/allowedips_test.go @@ -159,7 +159,16 @@ func TestTrieIPv4(t *testing.T) { assertNEQ(a, 192, 0, 0, 0) assertNEQ(a, 255, 0, 0, 0) - allowedIPs = AllowedIPs{} + allowedIPs.RemoveByPeer(a) + allowedIPs.RemoveByPeer(b) + allowedIPs.RemoveByPeer(c) + allowedIPs.RemoveByPeer(d) + allowedIPs.RemoveByPeer(e) + allowedIPs.RemoveByPeer(g) + allowedIPs.RemoveByPeer(h) + if allowedIPs.IPv4 != nil || allowedIPs.IPv6 != nil { + t.Error("Expected removing all the peers to empty trie, but it did not") + } insert(a, 192, 168, 0, 0, 16) insert(a, 192, 168, 0, 0, 24) |