From 5cd942fa42f449f6bee6208e952073d73a823ebf Mon Sep 17 00:00:00 2001
From: kakysha <kakysha@me.com>
Date: Wed, 12 Jun 2024 18:26:03 +0300
Subject: [PATCH] feat: set DoNotFailFast inside context in consensus code

so permissions module fails do not panic
---
 baseapp/baseapp.go | 16 ++++++++++------
 1 file changed, 10 insertions(+), 6 deletions(-)

diff --git a/baseapp/baseapp.go b/baseapp/baseapp.go
index 3cc88aa7dfc1..b13b6e35d0a0 100644
--- a/baseapp/baseapp.go
+++ b/baseapp/baseapp.go
@@ -43,6 +43,8 @@ type (
 	// particular, if a module changed the substore key name (or removed a substore)
 	// between two versions of the software.
 	StoreLoader func(ms storetypes.CommitMultiStore) error
+
+	contextKeyT string
 )
 
 const (
@@ -54,6 +56,8 @@ const (
 	execModeVoteExtension                       // Extend or verify a pre-commit vote
 	execModeVerifyVoteExtension                 // Verify a vote extension
 	execModeFinalize                            // Finalize a block proposal
+
+	DoNotFailFastContextKey contextKeyT = "DoNotFailFast"
 )
 
 var _ servertypes.ABCI = (*BaseApp)(nil)
@@ -708,7 +712,7 @@ func (app *BaseApp) cacheTxContext(ctx sdk.Context, txBytes []byte) (sdk.Context
 
 func (app *BaseApp) preBlock(req *abci.RequestFinalizeBlock) error {
 	if app.preBlocker != nil {
-		ctx := app.finalizeBlockState.Context()
+		ctx := app.finalizeBlockState.Context().WithValue(DoNotFailFastContextKey, struct{}{})
 		rsp, err := app.preBlocker(ctx, req)
 		if err != nil {
 			return err
@@ -733,7 +737,8 @@ func (app *BaseApp) beginBlock(req *abci.RequestFinalizeBlock) (sdk.BeginBlock,
 	)
 
 	if app.beginBlocker != nil {
-		resp, err = app.beginBlocker(app.finalizeBlockState.Context())
+		ctx := app.finalizeBlockState.Context().WithValue(DoNotFailFastContextKey, struct{}{})
+		resp, err = app.beginBlocker(ctx)
 		if err != nil {
 			return resp, err
 		}
@@ -746,7 +751,6 @@ func (app *BaseApp) beginBlock(req *abci.RequestFinalizeBlock) (sdk.BeginBlock,
 			)
 		}
 
-		ctx := app.finalizeBlockState.ctx
 		app.AddStreamEvents(ctx.BlockHeight(), ctx.BlockTime(), resp.Events, true)
 
 		resp.Events = sdk.MarkEventsToIndex(resp.Events, app.indexEvents)
@@ -797,11 +801,12 @@ func (app *BaseApp) deliverTx(tx []byte) *abci.ExecTxResult {
 
 // endBlock is an application-defined function that is called after transactions
 // have been processed in FinalizeBlock.
-func (app *BaseApp) endBlock(ctx context.Context) (sdk.EndBlock, error) {
+func (app *BaseApp) endBlock(_ context.Context) (sdk.EndBlock, error) {
 	var endblock sdk.EndBlock
 
 	if app.endBlocker != nil {
-		eb, err := app.endBlocker(app.finalizeBlockState.Context())
+		ctx := app.finalizeBlockState.Context().WithValue(DoNotFailFastContextKey, struct{}{})
+		eb, err := app.endBlocker(ctx)
 		if err != nil {
 			return endblock, err
 		}
@@ -814,7 +819,6 @@ func (app *BaseApp) endBlock(ctx context.Context) (sdk.EndBlock, error) {
 			)
 		}
 
-		ctx := app.finalizeBlockState.ctx
 		app.AddStreamEvents(ctx.BlockHeight(), ctx.BlockTime(), eb.Events, true)
 
 		eb.Events = sdk.MarkEventsToIndex(eb.Events, app.indexEvents)