Fix race condition

datarhei · Jan 15, 2024 · 0bd1188 · 0bd1188
1 parent 4bffbea
commit 0bd1188
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/process/process.go b/process/process.go
@@ -589,15 +589,19 @@ func (p *process) stop(wait bool) error {
 			p.callbacks.onExit = func() {
 				wg.Done()
 
+				p.callbacks.lock.Lock()
 				p.callbacks.onExit = nil
+				p.callbacks.lock.Unlock()
 			}
 		} else {
 			cb := p.callbacks.onExit
 			p.callbacks.onExit = func() {
 				cb()
 				wg.Done()
 
+				p.callbacks.lock.Lock()
 				p.callbacks.onExit = cb
+				p.callbacks.lock.Unlock()
 			}
 		}
 		p.callbacks.lock.Unlock()