update ProgramPrepare to accept wasm instead of statedb and code

Signed-off-by: Igor Braga <5835477+bragaigor@users.noreply.github.com>

Author: bragaigor

arbos/programs/wasm.go

@@ -7,6 +7,7 @@ package programs
 
 import (
 	"errors"
+	"fmt"
 	"unsafe"
 
 	"github.com/ethereum/go-ethereum/arbitrum/multigas"
@@ -124,10 +125,9 @@ func programRequiresPrepare(
 
 //go:wasmimport programs program_prepare
 func ProgramPrepare(
-	statedbPtr unsafe.Pointer,
+	wasmPtr unsafe.Pointer,
 	moduleHashPtr unsafe.Pointer,
 	addressForLoggingPtr unsafe.Pointer,
-	codePtr unsafe.Pointer,
 	codeSize uint64,
 	codehashPtr unsafe.Pointer,
 	maxWasmSize uint32,
@@ -156,21 +156,25 @@ func startProgram(module uint32) uint32
 //go:wasmimport programs send_response
 func sendResponse(req_id uint32) uint32
 
-func handleProgramPrepare(statedb vm.StateDB, moduleHash common.Hash, addressForLogging common.Address, code []byte, codehash common.Hash, params *StylusParams, time uint64, debugMode bool, program Program, runCtx *core.MessageRunContext) []byte {
+func handleProgramPrepare(statedb vm.StateDB, moduleHash common.Hash, programAddress common.Address, code []byte, codehash common.Hash, params *StylusParams, time uint64, debugMode bool, program Program, runCtx *core.MessageRunContext) []byte {
 	requiresPrepare := programRequiresPrepare(unsafe.Pointer(&moduleHash[0]))
 	if requiresPrepare != 0 {
 		var debugInt uint32
 		if debugMode {
 			debugInt = 1
 		}
 
+		wasm, err := getWasm(statedb, programAddress, params)
+		if err != nil {
+			panic(fmt.Sprintf("failed to get wasm for program, program address: %v, err: %v", programAddress.Hex(), err))
+		}
+
 		codeSize := uint64(len(code))
 
 		ProgramPrepare(
-			unsafe.Pointer(&statedb),
+			unsafe.Pointer(&wasm),
 			unsafe.Pointer(&moduleHash),
-			unsafe.Pointer(&addressForLogging),
-			unsafe.Pointer(&code),
+			unsafe.Pointer(&programAddress),
 			codeSize,
 			unsafe.Pointer(&codehash),
 			params.MaxWasmSize,

changelog/bragaigor-nit-4407.md

@@ -0,0 +1,2 @@
+### Changed
+- update ProgramPrepare to accept wasm instead of statedb and code

crates/jit/src/program.rs

@@ -188,10 +188,9 @@ pub fn program_requires_prepare(
 /// program_prepare
 pub fn program_prepare(
     mut _env: WasmEnvMut,
-    _state_ptr: GuestPtr,
+    _wasm_ptr: GuestPtr,
     _module_hash_ptr: GuestPtr,
     _address_for_logging_ptr: GuestPtr,
-    _code_ptr: GuestPtr,
     _code_size: u64,
     _code_hash_ptr: GuestPtr,
     _max_wasm_size: u32,

crates/wasm-libraries/user-host/src/link.rs

@@ -147,10 +147,9 @@ pub unsafe extern "C" fn programs__program_requires_prepare(_module_hash_ptr: Gu
 /// consumes activated program module hash and wasm code
 #[no_mangle]
 pub unsafe extern "C" fn programs__program_prepare(
-    _state_ptr: GuestPtr,
+    _wasm_ptr: GuestPtr,
     _module_hash_ptr: GuestPtr,
     _address_for_logging_ptr: GuestPtr,
-    _code_ptr: GuestPtr,
     _code_size: u64,
     _code_hash_ptr: GuestPtr,
     _max_wasm_size: u32,