device: fix comment typo and shorten state.mu.Lock to state.Lock

Signed-off-by: Jason A. Donenfeld <Jason@zx2c4.com>
This commit is contained in:
Jason A. Donenfeld 2021-02-09 15:35:43 +01:00
parent 02138f1f81
commit a816e8511e
2 changed files with 12 additions and 13 deletions

View File

@ -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
}

View File

@ -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