pezkuwichain/revive
1
0
Fork 0
mirror of https://github.com/pezkuwichain/revive.git synced 2026-04-24 21:47:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

truncate to xlen fn

Browse Source 
Signed-off-by: xermicus <cyrill@parity.io>
This commit is contained in:
xermicus
2024-05-14 10:59:39 +02:00
parent 83bf9d6041
commit 3eadcb1c40
1 changed files with 52 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files
crates/llvm-context/src/polkavm/context/mod.rs
+52 -20
View File
@@ -1198,29 +1198,61 @@ where
"expected XLEN or WORD sized int type for memory offset",
);
let truncated =
self.builder()
.build_int_truncate(value, self.xlen_type(), "offset_truncated")?;
let extended =
self.builder()
.build_int_z_extend(truncated, self.word_type(), "offset_extended")?;
let is_overflow = self.builder().build_int_compare(
inkwell::IntPredicate::NE,
value,
extended,
"compare_truncated_extended",
)?;
let name = "__safe_trunc_xlen";
let function = self.module().get_function(name).unwrap_or_else(|| {
let position = self.basic_block();
let block_continue = self.append_basic_block("offset_pointer_ok");
let block_trap = self.append_basic_block("offset_pointer_overflow");
self.build_conditional_branch(is_overflow, block_trap, block_continue)?;
let function = self.module().add_function(
name,
self.xlen_type().fn_type(&[self.word_type().into()], false),
None,
);
let block_entry = self.llvm().append_basic_block(function, "entry");
self.set_basic_block(block_entry);
self.set_basic_block(block_trap);
self.build_call(self.intrinsics().trap, &[], "invalid_trap");
self.build_unreachable();
let value = function.get_first_param().unwrap().into_int_value();
let truncated = self
.builder()
.build_int_truncate(value, self.xlen_type(), "offset_truncated")
.unwrap();
let extended = self
.builder()
.build_int_z_extend(truncated, self.word_type(), "offset_extended")
.unwrap();
let is_overflow = self
.builder()
.build_int_compare(
inkwell::IntPredicate::NE,
value,
extended,
"compare_truncated_extended",
)
.unwrap();
self.set_basic_block(block_continue);
Ok(truncated)
let block_continue = self.llvm().append_basic_block(function, "offset_ok");
let block_trap = self.llvm().append_basic_block(function, "offset_overflow");
self.build_conditional_branch(is_overflow, block_trap, block_continue)
.unwrap();
self.set_basic_block(block_trap);
self.build_call(self.intrinsics().trap, &[], "invalid_trap");
self.build_unreachable();
self.set_basic_block(block_continue);
self.builder().build_return(Some(&truncated)).unwrap();
self.set_basic_block(position);
function
});
Ok(self
.builder()
.build_direct_call(function, &[value.into()], name)?
.try_as_basic_value()
.left()
.expect("returns a value")
.into_int_value())
}
/// Build a call to PolkaVM `sbrk` for extending the heap by `size`.