recover action handling panic

iotexproject · Dec 5, 2024 · ba0ca99 · ba0ca99
1 parent 7840b13
commit ba0ca99
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/action/const.go b/action/const.go
@@ -44,6 +44,7 @@ var (
 	ErrGasTipOverFeeCap   = errors.New("tip cap is greater than fee cap")
 	ErrMissRequiredField  = errors.New("missing required field")
 	ErrValueVeryHigh      = errors.New("value is very high")
+	ErrPanic              = errors.New("panic")
 )
 
 // LoadErrorDescription loads corresponding description related to the error

diff --git a/state/factory/workingset.go b/state/factory/workingset.go
@@ -168,7 +168,13 @@ func withActionCtx(ctx context.Context, selp *action.SealedEnvelope) (context.Co
 func (ws *workingSet) runAction(
 	ctx context.Context,
 	selp *action.SealedEnvelope,
-) (*action.Receipt, error) {
+) (receipt *action.Receipt, err error) {
+	defer func() {
+		if r := recover(); r != nil {
+			receipt = nil
+			err = errors.Wrapf(action.ErrPanic, "panic when running action: %v", r)
+		}
+	}()
 	actCtx := protocol.MustGetActionCtx(ctx)
 	if protocol.MustGetBlockCtx(ctx).GasLimit < actCtx.IntrinsicGas {
 		return nil, action.ErrGasLimit