diff options
author | Mathias Hall-Andersen <mathias@hall-andersen.dk> | 2017-06-29 14:39:21 +0200 |
---|---|---|
committer | Mathias Hall-Andersen <mathias@hall-andersen.dk> | 2017-06-29 14:39:21 +0200 |
commit | 7e185db1418635a28e5aacbd17b1f17b9ab89e35 (patch) | |
tree | e617d1dbca6c3d45207f9db145ec270bf8496132 /src/trie.go | |
parent | 1f0976a26c1d0a6b5eb2c0aa993f12d89f96eed2 (diff) | |
download | wireguard-go-7e185db1418635a28e5aacbd17b1f17b9ab89e35.tar.gz wireguard-go-7e185db1418635a28e5aacbd17b1f17b9ab89e35.zip |
Completed get/set configuration
For debugging of "outbound flow"
Mostly, a few things still missing
Diffstat (limited to 'src/trie.go')
-rw-r--r-- | src/trie.go | 10 |
1 files changed, 7 insertions, 3 deletions
diff --git a/src/trie.go b/src/trie.go index 4049167..c2304b2 100644 --- a/src/trie.go +++ b/src/trie.go @@ -195,7 +195,10 @@ func (node *Trie) Count() uint { return l + r } -func (node *Trie) AllowedIPs(p *Peer, results []net.IPNet) { +func (node *Trie) AllowedIPs(p *Peer, results []net.IPNet) []net.IPNet { + if node == nil { + return results + } if node.peer == p { var mask net.IPNet mask.Mask = net.CIDRMask(int(node.cidr), len(node.bits)*8) @@ -213,6 +216,7 @@ func (node *Trie) AllowedIPs(p *Peer, results []net.IPNet) { } results = append(results, mask) } - node.child[0].AllowedIPs(p, results) - node.child[1].AllowedIPs(p, results) + results = node.child[0].AllowedIPs(p, results) + results = node.child[1].AllowedIPs(p, results) + return results } |