revive/crates/benchmarks/src/lib.rs

use polkavm::{BackendKind, Config, Engine, ExportIndex, Instance, SandboxKind};
use revive_integration::mock_runtime;
use revive_integration::mock_runtime::State;

pub fn prepare_pvm(
    code: &[u8],
    input: &[u8],
    backend: BackendKind,
) -> (State, Instance<State>, ExportIndex) {
    let mut config = Config::new();
    config.set_backend(Some(backend));
    config.set_sandbox(Some(SandboxKind::Linux));

    let (instance, export_index) = mock_runtime::prepare(code, Some(config));

    (State::new(input.to_vec()), instance, export_index)
}

pub fn instantiate_engine(backend: BackendKind) -> Engine {
    let mut config = Config::new();
    config.set_backend(Some(backend));
    config.set_sandbox(Some(SandboxKind::Linux));
    mock_runtime::setup(Some(config))
}