mirror of
https://github.com/pezkuwichain/revive.git
synced 2026-06-14 21:31:05 +00:00
xermicus
53 lines
1.9 KiB
Rust
53 lines
1.9 KiB
Rust
//! This crate vendors the [PolkaVM][0] C API and provides a LLVM module for interacting
|
|
//! with the `pallet-contracts` runtime API.
|
|
//! At present, the contracts pallet requires blobs to export `call` and `deploy`,
|
|
//! and offers a bunch of [runtime API methods][1]. The provided [module] implements
|
|
//! those exports and imports.
|
|
//! [0]: [https://crates.io/crates/polkavm]
|
|
//! [1]: [https://docs.rs/pallet-contracts/26.0.0/pallet_contracts/api_doc/index.html]
|
|
|
|
use inkwell::{context::Context, memory_buffer::MemoryBuffer, module::Module, support::LLVMString};
|
|
|
|
include!(concat!(env!("OUT_DIR"), "/polkavm_guest.rs"));
|
|
|
|
/// Creates a LLVM module from the [BITCODE].
|
|
/// The module does:
|
|
/// - Export the `call` and `deploy` functions (which are named thereafter).
|
|
/// - Import (most) `pallet-contracts` runtime API functions.
|
|
/// Returns `Error` if the bitcode fails to parse, which should never happen.
|
|
pub fn module<'context>(
|
|
context: &'context Context,
|
|
module_name: &str,
|
|
) -> Result<Module<'context>, LLVMString> {
|
|
let buf = MemoryBuffer::create_from_memory_range(BITCODE, module_name);
|
|
Module::parse_bitcode_from_buffer(&buf, context)
|
|
}
|
|
|
|
/// Creates a module that sets the PolkaVM minimum stack size to [`size`] if linked in.
|
|
pub fn min_stack_size<'context>(
|
|
context: &'context Context,
|
|
module_name: &str,
|
|
size: u32,
|
|
) -> Module<'context> {
|
|
let module = context.create_module(module_name);
|
|
module.set_inline_assembly(&format!(
|
|
".pushsection .polkavm_min_stack_size,\"\",@progbits
|
|
.word {size}
|
|
.popsection"
|
|
));
|
|
module
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
#[test]
|
|
fn it_works() {
|
|
inkwell::targets::Target::initialize_riscv(&Default::default());
|
|
let context = inkwell::context::Context::create();
|
|
let module = crate::module(&context, "polkavm_guest").unwrap();
|
|
|
|
assert!(module.get_function("call").is_some());
|
|
assert!(module.get_function("deploy").is_some());
|
|
}
|
|
}
|