From a816e8511eb45975e8e0e4199c3481989953fd56 Mon Sep 17 00:00:00 2001
From: "Jason A. Donenfeld" <Jason@zx2c4.com>
Date: Tue, 9 Feb 2021 15:35:43 +0100
Subject: [PATCH] device: fix comment typo and shorten state.mu.Lock to
 state.Lock

Signed-off-by: Jason A. Donenfeld <Jason@zx2c4.com>
---
 device/device.go | 15 +++++++--------
 device/peer.go   | 10 +++++-----
 2 files changed, 12 insertions(+), 13 deletions(-)

diff --git a/device/device.go b/device/device.go
index 935f41b..0417397 100644
--- a/device/device.go
+++ b/device/device.go
@@ -24,9 +24,8 @@ type Device struct {
 	state struct {
 		// state holds the device's state. It is accessed atomically.
 		// Use the device.deviceState method to read it.
-		// If state.mu is (r)locked, state is the current state of the device.
-		// Without state.mu (r)locked, state is either the current state
-		// of the device or the intended future state of the device.
+		// If state is not locked, state is the current state of the device.
+		// If state is locked, state is the current state or intended future state of the device.
 		// For example, while executing a call to Up, state will be deviceStateUp.
 		// There is no guarantee that that intended future state of the device
 		// will become the actual state; Up can fail.
@@ -36,7 +35,7 @@ type Device struct {
 		// stopping blocks until all inputs to Device have been closed.
 		stopping sync.WaitGroup
 		// mu protects state changes.
-		mu sync.Mutex
+		sync.Mutex
 	}
 
 	net struct {
@@ -145,8 +144,8 @@ func unsafeRemovePeer(device *Device, peer *Peer, key NoisePublicKey) {
 
 // changeState attempts to change the device state to match want.
 func (device *Device) changeState(want deviceState) {
-	device.state.mu.Lock()
-	defer device.state.mu.Unlock()
+	device.state.Lock()
+	defer device.state.Unlock()
 	old := device.deviceState()
 	if old == deviceStateClosed {
 		// once closed, always closed
@@ -353,8 +352,8 @@ func (device *Device) RemoveAllPeers() {
 }
 
 func (device *Device) Close() {
-	device.state.mu.Lock()
-	defer device.state.mu.Unlock()
+	device.state.Lock()
+	defer device.state.Unlock()
 	if device.isClosed() {
 		return
 	}
diff --git a/device/peer.go b/device/peer.go
index 69238a6..96481ee 100644
--- a/device/peer.go
+++ b/device/peer.go
@@ -52,7 +52,7 @@ type Peer struct {
 	}
 
 	state struct {
-		mu sync.Mutex // protects against concurrent Start/Stop
+		sync.Mutex // protects against concurrent Start/Stop
 	}
 
 	queue struct {
@@ -161,8 +161,8 @@ func (peer *Peer) Start() {
 	}
 
 	// prevent simultaneous start/stop operations
-	peer.state.mu.Lock()
-	defer peer.state.mu.Unlock()
+	peer.state.Lock()
+	defer peer.state.Unlock()
 
 	if peer.isRunning.Get() {
 		return
@@ -242,8 +242,8 @@ func (peer *Peer) ExpireCurrentKeypairs() {
 }
 
 func (peer *Peer) Stop() {
-	peer.state.mu.Lock()
-	defer peer.state.mu.Unlock()
+	peer.state.Lock()
+	defer peer.state.Unlock()
 
 	if !peer.isRunning.Swap(false) {
 		return