Cancel parent context when aborting bootstrapping

Signed-off-by: Yacov Manevich <yacov.manevich@avalabs.org>

Author: yacovm

chains/manager.go

@@ -262,6 +262,8 @@ type manager struct {
 	// processed.
 	chainCreatorShutdownCh chan struct{}
 	chainCreatorExited     sync.WaitGroup
+	context                context.Context
+	cancelContext          context.CancelFunc
 
 	chainsLock sync.Mutex
 	// Key: Chain's ID
@@ -324,7 +326,11 @@ func New(config *ManagerConfig) (Manager, error) {
 		return nil, err
 	}
 
+	context, cancelContext := context.WithCancel(context.Background())
+
 	return &manager{
+		context:                context,
+		cancelContext:          cancelContext,
 		Aliaser:                ids.NewAliaser(),
 		ManagerConfig:          *config,
 		chains:                 make(map[ids.ID]handler.Handler),
@@ -468,7 +474,7 @@ func (m *manager) createChain(chainParams ChainParameters) {
 
 	// Tell the chain to start processing messages.
 	// If the X, P, or C Chain panics, do not attempt to recover
-	chain.Handler.Start(context.TODO(), !m.CriticalChains.Contains(chainParams.ID))
+	chain.Handler.Start(m.context, !m.CriticalChains.Contains(chainParams.ID))
 }
 
 // Create a chain
@@ -1523,6 +1529,7 @@ func (m *manager) Shutdown() {
 	m.Log.Info("shutting down chain manager")
 	m.chainsQueue.Close()
 	close(m.chainCreatorShutdownCh)
+	m.cancelContext()
 	m.chainCreatorExited.Wait()
 	m.ManagerConfig.Router.Shutdown(context.TODO())
 }