From 18c5f638b979ff407a367f2b8f57448e390a84ef Mon Sep 17 00:00:00 2001 From: Anna Shaleva Date: Mon, 4 Apr 2022 19:07:32 +0300 Subject: [PATCH] dao: adjust usages of `Internal DB inconsistency` error --- pkg/core/dao/dao.go | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) diff --git a/pkg/core/dao/dao.go b/pkg/core/dao/dao.go index f7e6f065e..e5e27d4ba 100644 --- a/pkg/core/dao/dao.go +++ b/pkg/core/dao/dao.go @@ -25,6 +25,9 @@ var ( // ErrHasConflicts is returned when transaction is in the list of conflicting // transactions which are already in dao. ErrHasConflicts = errors.New("transaction has conflicts") + // ErrInternalDBInconsistency is returned when the format of retrieved DAO + // record is unexpected. + ErrInternalDBInconsistency = errors.New("internal DB inconsistency") ) // Simple is memCached wrapper around DB, simple DAO implementation. @@ -252,7 +255,7 @@ func (dao *Simple) GetAppExecResults(hash util.Uint256, trig trigger.Type) ([]st return nil, err } r := io.NewBinReaderFromBuf(bs) - switch r.ReadB() { + switch pref := r.ReadB(); pref { case storage.ExecBlock: _, err = block.NewTrimmedFromReader(dao.Version.StateRootInHeader, r) if err != nil { @@ -265,6 +268,8 @@ func (dao *Simple) GetAppExecResults(hash util.Uint256, trig trigger.Type) ([]st _ = r.ReadU32LE() tx := &transaction.Transaction{} tx.DecodeBinary(r) + default: + return nil, fmt.Errorf("%w: unexpected executable prefix %d", ErrInternalDBInconsistency, pref) } if r.Err != nil { return nil, r.Err @@ -358,7 +363,8 @@ func (dao *Simple) getBlock(key []byte) (*block.Block, error) { r := io.NewBinReaderFromBuf(b) if r.ReadB() != storage.ExecBlock { - return nil, errors.New("internal DB inconsistency") + // It may be a transaction. + return nil, storage.ErrKeyNotFound } block, err := block.NewTrimmedFromReader(dao.Version.StateRootInHeader, r) if err != nil { @@ -522,7 +528,8 @@ func (dao *Simple) GetTransaction(hash util.Uint256) (*transaction.Transaction, return nil, 0, errors.New("bad transaction bytes") } if b[0] != storage.ExecTransaction { - return nil, 0, errors.New("internal DB inconsistency") + // It may be a block. + return nil, 0, storage.ErrKeyNotFound } if b[5] == transaction.DummyVersion { return nil, 0, storage.ErrKeyNotFound