add testcase

terra-money · Oct 19, 2021 · 27e18ac · 27e18ac
1 parent d2e72d4
commit 27e18ac
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 2 deletions.
diff --git a/x/wasm/keeper/recursive_test.go b/x/wasm/keeper/recursive_test.go
@@ -342,3 +342,20 @@ func TestLimitRecursiveQueryGas(t *testing.T) {
 		})
 	}
 }
+
+func TestLimitRecursiveQueryDepth(t *testing.T) {
+
+	contractAddr, _, ctx, keeper, _ := initRecurseContract(t)
+	msg := buildQuery(t, Recurse{
+		Depth: types.ContractMaxQueryDepth - 1, // need to exclude first query
+	})
+
+	_, err := keeper.queryToContract(ctx, contractAddr, msg)
+	require.NoError(t, err)
+
+	msg = buildQuery(t, Recurse{
+		Depth: types.ContractMaxQueryDepth,
+	})
+	_, err = keeper.queryToContract(ctx, contractAddr, msg)
+	require.Error(t, err)
+}
diff --git a/x/wasm/types/wasmer_engine.go b/x/wasm/types/wasmer_engine.go
@@ -5,7 +5,8 @@ import (
 	wasmvmtypes "github.com/CosmWasm/wasmvm/types"
 )
 
-const maxQueryDepth = 20
+// ContractMaxQueryDepth maximum recursive query depth allowed
+const ContractMaxQueryDepth = 20
 
 var _ WasmerEngine = &WasmerEngineWithQueryDepth{}
 
@@ -25,7 +26,7 @@ func NewWasmerEngineWithQueryDepth(wasmVM *wasmvm.VM) *WasmerEngineWithQueryDept
 // IncreaseQueryDepth increase execution depth by 1 and check whether
 // the depth exceeds max one or not
 func (wasmer *WasmerEngineWithQueryDepth) IncreaseQueryDepth() error {
-	if wasmer.QueryDepth >= maxQueryDepth {
+	if wasmer.QueryDepth >= ContractMaxQueryDepth {
 		return ErrExceedMaxQueryDepth
 	}