updatehub: Fix race condition in GetState

Signed-off-by: Luis Gustavo S. Barreto <gustavo@ossystems.com.br>
UpdateHub · Jul 12, 2018 · 5de10ab · 5de10ab
1 parent 7320d3c
commit 5de10ab
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/updatehub/updatehub.go b/updatehub/updatehub.go
@@ -145,6 +145,9 @@ func (uh *UpdateHub) Cancel(nextState State) {
 }
 
 func (uh *UpdateHub) GetState() State {
+	uh.stateMutex.Lock()
+	defer uh.stateMutex.Unlock()
+
 	return uh.state
 }