diff options
| author |   Josh Bleecher Snyder <josh@tailscale.com> | 2020-12-15 17:44:21 -0800 |
|---|---|---|
| committer | Josh Bleecher Snyder <josh@tailscale.com> | 2020-12-16 16:16:26 -0800 |
| commit | 26b0dca4f569ccd1fb02fe3742cbed8595b24728 (patch) | |
| tree | b2605ff0dd0a12be4055d8b936abdb666a88ae9a | |
| parent | device: fix races from changing private_key (diff) | |
| download | wireguard-go-26b0dca4f569ccd1fb02fe3742cbed8595b24728.tar.xz wireguard-go-26b0dca4f569ccd1fb02fe3742cbed8595b24728.zip | |
device: fix data race in peer.timersActive
Found by the race detector and existing tests.
To avoid introducing a lock into this hot path,
calculate and cache whether any peers exist.
Signed-off-by: Josh Bleecher Snyder <josh@tailscale.com>
| -rw-r--r-- | device/device.go | 6 | ||||
| -rw-r--r-- | device/peer.go | 1 | ||||
| -rw-r--r-- | device/timers.go | 2 |
3 files changed, 6 insertions, 3 deletions
diff --git a/device/device.go b/device/device.go index 99f5e60..e7d70c4 100644 --- a/device/device.go +++ b/device/device.go @@ -49,8 +49,9 @@ type Device struct { } peers struct { - sync.RWMutex - keyMap map[NoisePublicKey]*Peer + empty AtomicBool // empty reports whether len(keyMap) == 0 + sync.RWMutex // protects keyMap + keyMap map[NoisePublicKey]*Peer } // unprotected / "self-synchronising resources" @@ -129,6 +130,7 @@ func unsafeRemovePeer(device *Device, peer *Peer, key NoisePublicKey) { // remove from peer map delete(device.peers.keyMap, key) + device.peers.empty.Set(len(device.peers.keyMap) == 0) } func deviceUpdateState(device *Device) { diff --git a/device/peer.go b/device/peer.go index fe6de33..73e6083 100644 --- a/device/peer.go +++ b/device/peer.go @@ -125,6 +125,7 @@ func (device *Device) NewPeer(pk NoisePublicKey) (*Peer, error) { // add device.peers.keyMap[pk] = peer + device.peers.empty.Set(false) // start peer diff --git a/device/timers.go b/device/timers.go index e94da36..dbd4c66 100644 --- a/device/timers.go +++ b/device/timers.go @@ -73,7 +73,7 @@ func (timer *Timer) IsPending() bool { } func (peer *Peer) timersActive() bool { - return peer.isRunning.Get() && peer.device != nil && peer.device.isUp.Get() && len(peer.device.peers.keyMap) > 0 + return peer.isRunning.Get() && peer.device != nil && peer.device.isUp.Get() && !peer.device.peers.empty.Get() } func expiredRetransmitHandshake(peer *Peer) { |