From 0b391b6e3a76228f5312235ee9a9af8f55b4d6f1 Mon Sep 17 00:00:00 2001
From: Arceliar <Arceliar@users.noreply.github.com>
Date: Sun, 6 May 2018 19:01:52 -0500
Subject: [PATCH] debugging and cleanup

---
 src/yggdrasil/debug.go |  2 +-
 src/yggdrasil/peer.go  | 35 ++++++++++++++++++++++++++++++-----
 2 files changed, 31 insertions(+), 6 deletions(-)

diff --git a/src/yggdrasil/debug.go b/src/yggdrasil/debug.go
index 5358379e..55a896c6 100644
--- a/src/yggdrasil/debug.go
+++ b/src/yggdrasil/debug.go
@@ -400,7 +400,7 @@ func (c *Core) DEBUG_setIfceExpr(expr *regexp.Regexp) {
 func (c *Core) DEBUG_addAuthBoxPub(boxBytes []byte) {
 	var box boxPubKey
 	copy(box[:], boxBytes)
-	c.peers.authBoxPubs[box] = struct{}{}
+	c.peers.addAuthBoxPub(&box)
 }
 
 ////////////////////////////////////////////////////////////////////////////////
diff --git a/src/yggdrasil/peer.go b/src/yggdrasil/peer.go
index 5e3f2f1d..024d9bab 100644
--- a/src/yggdrasil/peer.go
+++ b/src/yggdrasil/peer.go
@@ -30,11 +30,12 @@ import "math"
 //import "fmt"
 
 type peers struct {
-	core        *Core
-	authBoxPubs map[boxPubKey]struct{}
-	mutex       sync.Mutex   // Synchronize writes to atomic
-	ports       atomic.Value //map[Port]*peer, use CoW semantics
+	core  *Core
+	mutex sync.Mutex   // Synchronize writes to atomic
+	ports atomic.Value //map[Port]*peer, use CoW semantics
 	//ports map[Port]*peer
+	authMutex   sync.RWMutex
+	authBoxPubs map[boxPubKey]struct{}
 }
 
 func (ps *peers) init(c *Core) {
@@ -46,10 +47,34 @@ func (ps *peers) init(c *Core) {
 }
 
 func (ps *peers) isAuthBoxPub(box *boxPubKey) bool {
+	ps.authMutex.RLock()
+	defer ps.authMutex.RUnlock()
 	_, isIn := ps.authBoxPubs[*box]
 	return isIn || len(ps.authBoxPubs) == 0
 }
 
+func (ps *peers) addAuthBoxPub(box *boxPubKey) {
+	ps.authMutex.Lock()
+	defer ps.authMutex.Unlock()
+	ps.authBoxPubs[*box] = struct{}{}
+}
+
+func (ps *peers) removeAuthBoxPub(box *boxPubKey) {
+	ps.authMutex.Lock()
+	defer ps.authMutex.Unlock()
+	delete(ps.authBoxPubs, *box)
+}
+
+func (ps *peers) getAuthBoxPubs() []boxPubKey {
+	ps.authMutex.RLock()
+	defer ps.authMutex.RUnlock()
+	keys := make([]boxPubKey, 0, len(ps.authBoxPubs))
+	for key := range ps.authBoxPubs {
+		keys = append(keys, key)
+	}
+	return keys
+}
+
 func (ps *peers) getPorts() map[switchPort]*peer {
 	return ps.ports.Load().(map[switchPort]*peer)
 }
@@ -218,7 +243,7 @@ func (p *peer) handlePacket(packet []byte, linkIn chan<- []byte) {
 }
 
 func (p *peer) handleTraffic(packet []byte, pTypeLen int) {
-	if p.msgAnc == nil {
+	if p.port != 0 && p.msgAnc == nil {
 		// Drop traffic until the peer manages to send us at least one anc
 		return
 	}