From bfdff671388645708a09c1757891fd3660ec971b Mon Sep 17 00:00:00 2001
From: Frederic Branczyk <fbranczyk@gmail.com>
Date: Sat, 9 Dec 2017 08:22:07 -0600
Subject: [PATCH] nflog: Copy and replace gossipData instead of modifying it in
 place (#1121)

---
 nflog/nflog.go      | 34 ++++++++++++++++++++--------------
 nflog/nflog_test.go |  7 ++++---
 2 files changed, 24 insertions(+), 17 deletions(-)

diff --git a/nflog/nflog.go b/nflog/nflog.go
index 4cbf121d..5066f918 100644
--- a/nflog/nflog.go
+++ b/nflog/nflog.go
@@ -491,7 +491,8 @@ func (l *nlog) OnGossip(msg []byte) (mesh.GossipData, error) {
 	l.mtx.Lock()
 	defer l.mtx.Unlock()
 
-	if delta := l.st.mergeDelta(gd); len(delta) > 0 {
+	var delta gossipData
+	if l.st, delta = l.st.mergeDelta(gd); len(delta) > 0 {
 		return delta, nil
 	}
 	return nil, nil
@@ -506,7 +507,10 @@ func (l *nlog) OnGossipBroadcast(src mesh.PeerName, msg []byte) (mesh.GossipData
 	l.mtx.Lock()
 	defer l.mtx.Unlock()
 
-	return l.st.mergeDelta(gd), nil
+	var delta mesh.GossipData
+	l.st, delta = l.st.mergeDelta(gd)
+
+	return delta, nil
 }
 
 // OnGossipUnicast implements the mesh.Gossiper interface.
@@ -577,36 +581,38 @@ func (gd gossipData) clone() gossipData {
 // TODO(fabxc): can we just return the receiver. Does it have to remain
 // unmodified. Needs to be clarified upstream.
 func (gd gossipData) Merge(other mesh.GossipData) mesh.GossipData {
+	merged := gd.clone()
 	for k, e := range other.(gossipData) {
-		prev, ok := gd[k]
+		prev, ok := merged[k]
 		if !ok {
-			gd[k] = e
+			merged[k] = e
 			continue
 		}
 		if prev.Entry.Timestamp.Before(e.Entry.Timestamp) {
-			gd[k] = e
+			merged[k] = e
 		}
 	}
-	return gd
+	return merged
 }
 
-// mergeDelta behaves like Merge but returns a gossipData only containing
-// things that have changed.
-func (gd gossipData) mergeDelta(od gossipData) gossipData {
-	delta := gossipData{}
+// mergeDelta behaves like Merge but in addition returns a gossipData only
+// containing things that have changed.
+func (gd gossipData) mergeDelta(od gossipData) (merged gossipData, delta gossipData) {
+	merged = gd.clone()
+	delta = gossipData{}
 	for k, e := range od {
-		prev, ok := gd[k]
+		prev, ok := merged[k]
 		if !ok {
-			gd[k] = e
+			merged[k] = e
 			delta[k] = e
 			continue
 		}
 		if prev.Entry.Timestamp.Before(e.Entry.Timestamp) {
-			gd[k] = e
+			merged[k] = e
 			delta[k] = e
 		}
 	}
-	return delta
+	return merged, delta
 }
 
 // replaceFile wraps a file that is moved to another filename on closing.
diff --git a/nflog/nflog_test.go b/nflog/nflog_test.go
index 9ee4fa43..dd6af9d9 100644
--- a/nflog/nflog_test.go
+++ b/nflog/nflog_test.go
@@ -195,16 +195,17 @@ func TestGossipDataMerge(t *testing.T) {
 		res := ca.Merge(cb)
 
 		require.Equal(t, c.final, res, "Merge result should match expectation")
-		require.Equal(t, c.final, ca, "Merge should apply changes to original state")
 		require.Equal(t, c.b, cb, "Merged state should remain unmodified")
+		require.NotEqual(t, c.final, ca, "Merge should not change original state")
 
 		ca, cb = c.a.clone(), c.b.clone()
 
-		delta := ca.mergeDelta(cb)
+		res, delta := ca.mergeDelta(cb)
 
 		require.Equal(t, c.delta, delta, "Merge delta should match expectation")
-		require.Equal(t, c.final, ca, "Merge should apply changes to original state")
+		require.Equal(t, c.final, res, "Merge should apply changes to original state")
 		require.Equal(t, c.b, cb, "Merged state should remain unmodified")
+		require.NotEqual(t, res, ca, "Merge should not change original state")
 	}
 }