fixed back transfers on callBack.

integrationTests/json/scenariosContracts_test.go

@@ -119,3 +119,10 @@ func TestCAttestation(t *testing.T) {
 		Run().
 		CheckNoError()
 }
+
+func TestRustPaymaster(t *testing.T) {
+	ScenariosTest(t).
+		Folder("paymaster/scenarios").
+		Run().
+		CheckNoError()
+}

vmhost/contexts/managedType.go

@@ -5,6 +5,7 @@ import (
 	"crypto/elliptic"
 	"encoding/binary"
 	"errors"
+	"github.com/multiversx/mx-chain-core-go/data/vm"
 	"io"
 	basicMath "math"
 	"math/big"
@@ -139,7 +140,7 @@ func (context *managedTypesContext) InitState() {
 // PushState appends the values map to the state stack
 func (context *managedTypesContext) PushState() {
 	newBigIntState, newBigFloatState, newEcState, newmBufferState, newmMapState := context.clone()
-	newTransfers := context.cloneBackTransfers()
+	newTransfers := cloneBackTransfers(context.managedTypesValues.backTransfers)
 	context.managedTypesStack = append(context.managedTypesStack, managedTypesState{
 		bigIntValues:   newBigIntState,
 		bigFloatValues: newBigFloatState,
@@ -150,6 +151,21 @@ func (context *managedTypesContext) PushState() {
 	})
 }
 
+// PopBackTransferIfAsyncCallBack copies the back transfer from the top of the stack in case of callbacks
+func (context *managedTypesContext) PopBackTransferIfAsyncCallBack(vmInput *vmcommon.ContractCallInput) {
+	if vmInput.CallType != vm.AsynchronousCallBack {
+		return
+	}
+
+	managedTypesStackLen := len(context.managedTypesStack)
+	if managedTypesStackLen == 0 {
+		return
+	}
+
+	prevState := context.managedTypesStack[managedTypesStackLen-1]
+	context.managedTypesValues.backTransfers = cloneBackTransfers(prevState.backTransfers)
+}
+
 // PopSetActiveState removes the latest entry from the state stack and sets it as the current values map
 func (context *managedTypesContext) PopSetActiveState() {
 	managedTypesStackLen := len(context.managedTypesStack)
@@ -797,7 +813,7 @@ func (context *managedTypesContext) AddValueOnlyBackTransfer(value *big.Int) {
 
 // GetBackTransfers returns all ESDT transfers and accumulated value as well, will clean accumulated values
 func (context *managedTypesContext) GetBackTransfers() ([]*vmcommon.ESDTTransfer, *big.Int) {
-	clonedTransfers := context.cloneBackTransfers()
+	clonedTransfers := cloneBackTransfers(context.managedTypesValues.backTransfers)
 	context.managedTypesValues.backTransfers = backTransfers{
 		ESDTTransfers: make([]*vmcommon.ESDTTransfer, 0),
 		CallValue:     big.NewInt(0),
@@ -806,9 +822,7 @@ func (context *managedTypesContext) GetBackTransfers() ([]*vmcommon.ESDTTransfer
 	return clonedTransfers.ESDTTransfers, clonedTransfers.CallValue
 }
 
-func (context *managedTypesContext) cloneBackTransfers() backTransfers {
-	currentBackTransfers := context.managedTypesValues.backTransfers
-
+func cloneBackTransfers(currentBackTransfers backTransfers) backTransfers {
 	newBackTransfers := backTransfers{
 		ESDTTransfers: make([]*vmcommon.ESDTTransfer, len(currentBackTransfers.ESDTTransfers)),
 		CallValue:     big.NewInt(0).Set(currentBackTransfers.CallValue),

vmhost/hostCore/execution.go

@@ -448,6 +448,7 @@ func (host *vmHost) executeOnDestContextNoBuiltinFunction(input *vmcommon.Contra
 	managedTypes, _, metering, output, runtime, async, storage := host.GetContexts()
 	managedTypes.PushState()
 	managedTypes.InitState()
+	managedTypes.PopBackTransferIfAsyncCallBack(input)
 
 	output.PushState()
 	output.CensorVMOutput()

vmhost/interface.go

@@ -218,6 +218,7 @@ type ManagedTypesContext interface {
 	GetBackTransfers() ([]*vmcommon.ESDTTransfer, *big.Int)
 	AddValueOnlyBackTransfer(value *big.Int)
 	AddBackTransfers(transfers []*vmcommon.ESDTTransfer)
+	PopBackTransferIfAsyncCallBack(vmInput *vmcommon.ContractCallInput)
 }
 
 // OutputContext defines the functionality needed for interacting with the output context