Browse Source

state: panic on ResponsePrepareProposal validation error

pull/8145/head
William Banfield 3 years ago
parent
commit
8f592576c6
No known key found for this signature in database GPG Key ID: EFAD3442BF29E3AC
2 changed files with 59 additions and 1 deletions
  1. +1
    -1
      internal/state/execution.go
  2. +58
    -0
      internal/state/execution_test.go

+ 1
- 1
internal/state/execution.go View File

@ -147,7 +147,7 @@ func (blockExec *BlockExecutor) CreateProposalBlock(
txrSet := types.NewTxRecordSet(rpp.TxRecords) txrSet := types.NewTxRecordSet(rpp.TxRecords)
if err := txrSet.Validate(maxDataBytes, block.Txs); err != nil { if err := txrSet.Validate(maxDataBytes, block.Txs); err != nil {
return nil, err
panic(fmt.Errorf("ResponsePrepareProposal validation: %w", err))
} }
for _, rtx := range txrSet.RemovedTxs() { for _, rtx := range txrSet.RemovedTxs() {


+ 58
- 0
internal/state/execution_test.go View File

@ -650,6 +650,64 @@ func TestEmptyPrepareProposal(t *testing.T) {
require.NoError(t, err) require.NoError(t, err)
} }
// TestPrepareProposalPanicOnInvalid tests that the block creation logic panics
// if the ResponsePrepareProposal returned from the application is invalid.
func TestPrepareProposalPanicOnInvalid(t *testing.T) {
const height = 2
ctx, cancel := context.WithCancel(context.Background())
defer cancel()
logger := log.TestingLogger()
eventBus := eventbus.NewDefault(logger)
require.NoError(t, eventBus.Start(ctx))
state, stateDB, privVals := makeState(t, 1, height)
stateStore := sm.NewStore(stateDB)
evpool := &mocks.EvidencePool{}
evpool.On("PendingEvidence", mock.Anything).Return([]types.Evidence{}, int64(0))
mp := &mpmocks.Mempool{}
mp.On("ReapMaxBytesMaxGas", mock.Anything, mock.Anything).Return(types.Txs{})
app := abcimocks.NewBaseMock()
// create an invalid ResponsePrepareProposal
rpp := abci.ResponsePrepareProposal{
ModifiedTx: true,
TxRecords: []*abci.TxRecord{
{
Action: abci.TxRecord_REMOVED,
Tx: []byte("new tx"),
},
},
}
app.On("PrepareProposal", mock.Anything).Return(rpp, nil)
cc := abciclient.NewLocalClient(logger, app)
proxyApp := proxy.New(cc, logger, proxy.NopMetrics())
err := proxyApp.Start(ctx)
require.NoError(t, err)
blockExec := sm.NewBlockExecutor(
stateStore,
logger,
proxyApp,
mp,
evpool,
nil,
eventBus,
)
pa, _ := state.Validators.GetByIndex(0)
commit := makeValidCommit(ctx, t, height, types.BlockID{}, state.Validators, privVals)
require.Panics(t,
func() {
blockExec.CreateProposalBlock(ctx, height, state, commit, pa, nil)
})
mp.AssertExpectations(t)
}
// TestPrepareProposalRemoveTxs tests that any transactions marked as REMOVED // TestPrepareProposalRemoveTxs tests that any transactions marked as REMOVED
// are not included in the block produced by CreateProposalBlock. The test also // are not included in the block produced by CreateProposalBlock. The test also
// ensures that any transactions removed are also removed from the mempool. // ensures that any transactions removed are also removed from the mempool.


Loading…
Cancel
Save