Fix executor middleware

vmhost/hosttest/bad_test.go

@@ -1,12 +1,17 @@
 package hostCoretest
 
 import (
+	"context"
 	"testing"
+	"time"
 
 	"github.com/multiversx/mx-chain-vm-go/executor"
 	contextmock "github.com/multiversx/mx-chain-vm-go/mock/context"
 	test "github.com/multiversx/mx-chain-vm-go/testcommon"
 	"github.com/multiversx/mx-chain-vm-go/vmhost"
+	"github.com/multiversx/mx-chain-vm-go/wasmer"
+	"github.com/multiversx/mx-chain-vm-go/wasmer2"
+	"github.com/stretchr/testify/require"
 )
 
 func TestBadContract_NoPanic_Memoryfault(t *testing.T) {
@@ -251,3 +256,42 @@ func TestBadContract_NoPanic_NonExistingFunction(t *testing.T) {
 				HasRuntimeErrorAndInfo(executor.ErrInvalidFunction.Error(), "thisDoesNotExist")
 		})
 }
+
+func TestBadContractExtra_LongIntLoop_Wasmer1(t *testing.T) {
+	testBadContractExtraLongIntLoop(t, wasmer.ExecutorFactory())
+}
+
+func TestBadContractExtra_LongIntLoop_Wasmer2(t *testing.T) {
+	testBadContractExtraLongIntLoop(t, wasmer2.ExecutorFactory())
+}
+
+func testBadContractExtraLongIntLoop(t *testing.T, executorFactory executor.ExecutorAbstractFactory) {
+	testCase := test.BuildInstanceCallTest(t).WithContracts(
+		test.CreateInstanceContract(test.ParentAddress).
+			WithCode(test.GetTestSCCode("bad-extra", "../../"))).
+		WithInput(test.CreateTestContractCallInputBuilder().
+			WithRecipientAddr(test.ParentAddress).
+			WithGasProvided(test.GasProvided).
+			WithFunction("bigLoop").
+			Build()).
+		WithExecutorFactory(executorFactory).
+		WithWasmerSIGSEGVPassthrough(false)
+
+	ctx, cancel := context.WithTimeout(context.Background(), time.Second)
+	defer cancel()
+
+	done := make(chan struct{})
+	go func() {
+		testCase.AndAssertResults(func(_ vmhost.VMHost, _ *contextmock.BlockchainHookStub, verify *test.VMOutputVerifier) {
+			verify.OutOfGas()
+		})
+		close(done)
+	}()
+
+	select {
+	case <-done:
+		return
+	case <-ctx.Done():
+		require.FailNow(t, "test timed out")
+	}
+}