updatehub: lock state change mutex in cancel operation

Signed-off-by: Luis Gustavo S. Barreto <gustavo@ossystems.com.br>
UpdateHub · Jul 31, 2018 · f5c07be · f5c07be
1 parent 7bc6511
commit f5c07be
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/updatehub/updatehub.go b/updatehub/updatehub.go
@@ -141,6 +141,9 @@ func NewUpdateHub(
 }
 
 func (uh *UpdateHub) Cancel(nextState State) {
+	uh.stateMutex.Lock()
+	defer uh.stateMutex.Unlock()
+
 	uh.state.Cancel(true, nextState)
 }