From 4c88eb48c0558626495d07939ede0cde8cd14e79 Mon Sep 17 00:00:00 2001 From: Bryan Boreham Date: Thu, 11 Dec 2014 00:50:45 +0000 Subject: [PATCH] Move call to Topology.RebuildRoutes It doesn't belong in handleBroadcastTCP(), so moved it up to the two places that cause that function to be called. --- router/connection.go | 1 + router/peer.go | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/router/connection.go b/router/connection.go index 72efc333..5713a67e 100644 --- a/router/connection.go +++ b/router/connection.go @@ -492,6 +492,7 @@ func (conn *LocalConnection) receiveTCP(decoder *gob.Decoder, usingPassword bool if len(newUpdate) == 0 { continue } + conn.Router.Topology.RebuildRoutes() conn.local.BroadcastTCP(Concat(ProtocolUpdateByte, newUpdate)) } else if msg[0] == ProtocolPMTUVerified { conn.verifyPMTU <- int(binary.BigEndian.Uint16(msg[1:])) diff --git a/router/peer.go b/router/peer.go index 6c08afd3..d99145e7 100644 --- a/router/peer.go +++ b/router/peer.go @@ -358,13 +358,13 @@ func (peer *Peer) handleConnectionEstablished(conn *LocalConnection) { } func (peer *Peer) handleBroadcastTCP(msg []byte) { - peer.Router.Topology.RebuildRoutes() peer.ForEachConnection(func(_ PeerName, conn Connection) { conn.(*LocalConnection).SendTCP(msg) }) } func (peer *Peer) broadcastPeerUpdate(peers ...*Peer) { + peer.Router.Topology.RebuildRoutes() peer.handleBroadcastTCP(Concat(ProtocolUpdateByte, EncodePeers(append(peers, peer)...))) }