division functions

Signed-off-by: Cyrill Leutwiler <bigcyrill@hotmail.com>

crates/llvm-context/src/polkavm/evm/arithmetic.rs

@@ -2,13 +2,13 @@
 
 use inkwell::values::BasicValue;
 
-use crate::polkavm::context::runtime::RuntimeFunction;
+//use crate::polkavm::context::runtime::RuntimeFunction;
 use crate::polkavm::context::Context;
 use crate::polkavm::Dependency;
-use crate::PolkaVMDivisionFunction;
-use crate::PolkaVMRemainderFunction;
-use crate::PolkaVMSignedDivisionFunction;
-use crate::PolkaVMSignedRemainderFunction;
+//use crate::PolkaVMDivisionFunction;
+//use crate::PolkaVMRemainderFunction;
+//use crate::PolkaVMSignedDivisionFunction;
+//use crate::PolkaVMSignedRemainderFunction;
 
 /// Translates the arithmetic addition.
 pub fn addition<'ctx, D>(
@@ -64,11 +64,21 @@ pub fn division<'ctx, D>(
 where
     D: Dependency + Clone,
 {
-    let name = <PolkaVMDivisionFunction as RuntimeFunction<D>>::NAME;
-    let declaration = <PolkaVMDivisionFunction as RuntimeFunction<D>>::declaration(context);
-    Ok(context
-        .build_call(declaration, &[operand_1.into(), operand_2.into()], "div")
-        .unwrap_or_else(|| panic!("revive runtime function {name} should return a value",)))
+    let result_pointer = context.build_alloca_at_entry(context.word_type(), "div_result_pointer");
+    let operand_1_pointer = context.build_alloca_at_entry(context.word_type(), "operand_1_pointer");
+    let operand_2_pointer = context.build_alloca_at_entry(context.word_type(), "operand_2_pointer");
+
+    context.build_store(operand_1_pointer, operand_1)?;
+    context.build_store(operand_2_pointer, operand_2)?;
+
+    let arguments = &[
+        result_pointer.to_int(context).into(),
+        operand_1_pointer.to_int(context).into(),
+        operand_2_pointer.to_int(context).into(),
+    ];
+
+    context.build_runtime_call(revive_runtime_api::polkavm_imports::DIV, arguments);
+    context.build_load(result_pointer, "div_result")
 }
 
 /// Translates the arithmetic remainder.
@@ -80,11 +90,21 @@ pub fn remainder<'ctx, D>(
 where
     D: Dependency + Clone,
 {
-    let name = <PolkaVMRemainderFunction as RuntimeFunction<D>>::NAME;
-    let declaration = <PolkaVMRemainderFunction as RuntimeFunction<D>>::declaration(context);
-    Ok(context
-        .build_call(declaration, &[operand_1.into(), operand_2.into()], "rem")
-        .unwrap_or_else(|| panic!("revive runtime function {name} should return a value",)))
+    let result_pointer = context.build_alloca_at_entry(context.word_type(), "rem_result_pointer");
+    let operand_1_pointer = context.build_alloca_at_entry(context.word_type(), "operand_1_pointer");
+    let operand_2_pointer = context.build_alloca_at_entry(context.word_type(), "operand_2_pointer");
+
+    context.build_store(operand_1_pointer, operand_1)?;
+    context.build_store(operand_2_pointer, operand_2)?;
+
+    let arguments = &[
+        result_pointer.to_int(context).into(),
+        operand_1_pointer.to_int(context).into(),
+        operand_2_pointer.to_int(context).into(),
+    ];
+
+    context.build_runtime_call(revive_runtime_api::polkavm_imports::MOD, arguments);
+    context.build_load(result_pointer, "rem_result")
 }
 
 /// Translates the signed arithmetic division.
@@ -99,11 +119,21 @@ pub fn division_signed<'ctx, D>(
 where
     D: Dependency + Clone,
 {
-    let name = <PolkaVMSignedDivisionFunction as RuntimeFunction<D>>::NAME;
-    let declaration = <PolkaVMSignedDivisionFunction as RuntimeFunction<D>>::declaration(context);
-    Ok(context
-        .build_call(declaration, &[operand_1.into(), operand_2.into()], "sdiv")
-        .unwrap_or_else(|| panic!("revive runtime function {name} should return a value",)))
+    let result_pointer = context.build_alloca_at_entry(context.word_type(), "sdiv_result_pointer");
+    let operand_1_pointer = context.build_alloca_at_entry(context.word_type(), "operand_1_pointer");
+    let operand_2_pointer = context.build_alloca_at_entry(context.word_type(), "operand_2_pointer");
+
+    context.build_store(operand_1_pointer, operand_1)?;
+    context.build_store(operand_2_pointer, operand_2)?;
+
+    let arguments = &[
+        result_pointer.to_int(context).into(),
+        operand_1_pointer.to_int(context).into(),
+        operand_2_pointer.to_int(context).into(),
+    ];
+
+    context.build_runtime_call(revive_runtime_api::polkavm_imports::SDIV, arguments);
+    context.build_load(result_pointer, "sdiv_result")
 }
 
 /// Translates the signed arithmetic remainder.
@@ -115,9 +145,19 @@ pub fn remainder_signed<'ctx, D>(
 where
     D: Dependency + Clone,
 {
-    let name = <PolkaVMSignedRemainderFunction as RuntimeFunction<D>>::NAME;
-    let declaration = <PolkaVMSignedRemainderFunction as RuntimeFunction<D>>::declaration(context);
-    Ok(context
-        .build_call(declaration, &[operand_1.into(), operand_2.into()], "srem")
-        .unwrap_or_else(|| panic!("revive runtime function {name} should return a value",)))
+    let result_pointer = context.build_alloca_at_entry(context.word_type(), "srem_result_pointer");
+    let operand_1_pointer = context.build_alloca_at_entry(context.word_type(), "operand_1_pointer");
+    let operand_2_pointer = context.build_alloca_at_entry(context.word_type(), "operand_2_pointer");
+
+    context.build_store(operand_1_pointer, operand_1)?;
+    context.build_store(operand_2_pointer, operand_2)?;
+
+    let arguments = &[
+        result_pointer.to_int(context).into(),
+        operand_1_pointer.to_int(context).into(),
+        operand_2_pointer.to_int(context).into(),
+    ];
+
+    context.build_runtime_call(revive_runtime_api::polkavm_imports::SMOD, arguments);
+    context.build_load(result_pointer, "rsem_result")
 }

crates/llvm-context/src/polkavm/evm/math.rs

@@ -1,7 +1,5 @@
 //! Translates the mathematical operations.
 
-use inkwell::values::BasicValue;
-
 use crate::polkavm::context::Context;
 use crate::polkavm::Dependency;
 
@@ -15,17 +13,26 @@ pub fn add_mod<'ctx, D>(
 where
     D: Dependency + Clone,
 {
-    Ok(context
-        .build_call(
-            context.llvm_runtime().add_mod,
-            &[
-                operand_1.as_basic_value_enum(),
-                operand_2.as_basic_value_enum(),
-                modulo.as_basic_value_enum(),
-            ],
-            "add_mod_call",
-        )
-        .expect("Always exists"))
+    let result_pointer =
+        context.build_alloca_at_entry(context.word_type(), "addmod_result_pointer");
+    let operand_1_pointer = context.build_alloca_at_entry(context.word_type(), "addmod_operand_1");
+    let operand_2_pointer = context.build_alloca_at_entry(context.word_type(), "addmod_operand_2");
+    let modulo_pointer =
+        context.build_alloca_at_entry(context.word_type(), "addmod_modulo_operand");
+
+    context.build_store(operand_2_pointer, operand_2)?;
+    context.build_store(operand_1_pointer, operand_1)?;
+    context.build_store(modulo_pointer, modulo)?;
+
+    let arguments = &[
+        result_pointer.to_int(context).into(),
+        operand_1_pointer.to_int(context).into(),
+        operand_2_pointer.to_int(context).into(),
+        modulo_pointer.to_int(context).into(),
+    ];
+
+    context.build_runtime_call(revive_runtime_api::polkavm_imports::ADDMOD, arguments);
+    context.build_load(result_pointer, "addmod_result")
 }
 
 /// Translates the `mulmod` instruction.
@@ -38,17 +45,26 @@ pub fn mul_mod<'ctx, D>(
 where
     D: Dependency + Clone,
 {
-    Ok(context
-        .build_call(
-            context.llvm_runtime().mul_mod,
-            &[
-                operand_1.as_basic_value_enum(),
-                operand_2.as_basic_value_enum(),
-                modulo.as_basic_value_enum(),
-            ],
-            "mul_mod_call",
-        )
-        .expect("Always exists"))
+    let result_pointer =
+        context.build_alloca_at_entry(context.word_type(), "mulmod_result_pointer");
+    let operand_1_pointer = context.build_alloca_at_entry(context.word_type(), "mulmod_operand_1");
+    let operand_2_pointer = context.build_alloca_at_entry(context.word_type(), "mulmod_operand_2");
+    let modulo_pointer =
+        context.build_alloca_at_entry(context.word_type(), "mulmod_modulo_operand");
+
+    context.build_store(operand_2_pointer, operand_2)?;
+    context.build_store(operand_1_pointer, operand_1)?;
+    context.build_store(modulo_pointer, modulo)?;
+
+    let arguments = &[
+        result_pointer.to_int(context).into(),
+        operand_1_pointer.to_int(context).into(),
+        operand_2_pointer.to_int(context).into(),
+        modulo_pointer.to_int(context).into(),
+    ];
+
+    context.build_runtime_call(revive_runtime_api::polkavm_imports::MULMOD, arguments);
+    context.build_load(result_pointer, "addmod_result")
 }
 
 /// Translates the `exp` instruction.
@@ -60,13 +76,22 @@ pub fn exponent<'ctx, D>(
 where
     D: Dependency + Clone,
 {
-    Ok(context
-        .build_call(
-            context.llvm_runtime().exp,
-            &[value.as_basic_value_enum(), exponent.as_basic_value_enum()],
-            "exp_call",
-        )
-        .expect("Always exists"))
+    let result_pointer = context.build_alloca_at_entry(context.word_type(), "exp_result_pointer");
+    let value_pointer = context.build_alloca_at_entry(context.word_type(), "exp_value_pointer");
+    let exponent_pointer =
+        context.build_alloca_at_entry(context.word_type(), "exp_exponent_pointer");
+
+    context.build_store(value_pointer, value)?;
+    context.build_store(exponent_pointer, exponent)?;
+
+    let arguments = &[
+        result_pointer.to_int(context).into(),
+        value_pointer.to_int(context).into(),
+        exponent_pointer.to_int(context).into(),
+    ];
+
+    context.build_runtime_call(revive_runtime_api::polkavm_imports::EXP, arguments);
+    context.build_load(result_pointer, "exponent_result")
 }
 
 /// Translates the `signextend` instruction.
@@ -78,11 +103,20 @@ pub fn sign_extend<'ctx, D>(
 where
     D: Dependency + Clone,
 {
-    Ok(context
-        .build_call(
-            context.llvm_runtime().sign_extend,
-            &[bytes.as_basic_value_enum(), value.as_basic_value_enum()],
-            "sign_extend_call",
-        )
-        .expect("Always exists"))
+    let result_pointer =
+        context.build_alloca_at_entry(context.word_type(), "signext_result_pointer");
+    let bytes_pointer = context.build_alloca_at_entry(context.word_type(), "bytes_pointer");
+    let value_pointer = context.build_alloca_at_entry(context.word_type(), "signext_value_pointer");
+
+    context.build_store(bytes_pointer, bytes)?;
+    context.build_store(value_pointer, value)?;
+
+    let arguments = &[
+        result_pointer.to_int(context).into(),
+        bytes_pointer.to_int(context).into(),
+        value_pointer.to_int(context).into(),
+    ];
+
+    context.build_runtime_call(revive_runtime_api::polkavm_imports::EXP, arguments);
+    context.build_load(result_pointer, "signext_mod_result")
 }

crates/runtime-api/src/polkavm_imports.c

@@ -104,3 +104,19 @@ POLKAVM_IMPORT(uint64_t, set_storage, uint32_t, uint32_t, uint32_t, uint32_t, ui
 POLKAVM_IMPORT(void, value_transferred, uint32_t)
 
 POLKAVM_IMPORT(void, weight_to_fee, uint64_t, uint64_t, uint32_t);
+
+POLKAVM_IMPORT(void, div, uint32_t, uint32_t, uint32_t);
+
+POLKAVM_IMPORT(void, sdiv, uint32_t, uint32_t, uint32_t);
+
+POLKAVM_IMPORT(void, addmod, uint32_t, uint32_t, uint32_t, uint32_t);
+
+POLKAVM_IMPORT(void, mulmod, uint32_t, uint32_t, uint32_t, uint32_t);
+
+POLKAVM_IMPORT(void, mod, uint32_t, uint32_t, uint32_t);
+
+POLKAVM_IMPORT(void, smod, uint32_t, uint32_t, uint32_t);
+
+POLKAVM_IMPORT(void, exp, uint32_t, uint32_t, uint32_t);
+
+POLKAVM_IMPORT(void, signext, uint32_t, uint32_t, uint32_t);

crates/runtime-api/src/polkavm_imports.rs

@@ -68,9 +68,27 @@ pub static VALUE_TRANSFERRED: &str = "value_transferred";
 
 pub static WEIGHT_TO_FEE: &str = "weight_to_fee";
 
+pub static DIV: &str = "div";
+pub static SDIV: &str = "sdiv";
+pub static ADDMOD: &str = "addmod";
+pub static MULMOD: &str = "mulmod";
+pub static MOD: &str = "mod";
+pub static SMOD: &str = "smod";
+pub static EXP: &str = "exp";
+pub static SIGNEXT: &str = "signext";
+
 /// All imported runtime API symbols.
 /// Useful for configuring common attributes and linkage.
-pub static IMPORTS: [&str; 33] = [
+pub static IMPORTS: [&str; 41] = [
+    DIV,
+    SDIV,
+    ADDMOD,
+    MULMOD,
+    MOD,
+    SMOD,
+    EXP,
+    SIGNEXT,
+    //
     ADDRESS,
     BALANCE,
     BALANCE_OF,

crates/yul/src/parser/statement/object.rs

@@ -204,10 +204,10 @@ where
         revive_llvm_context::PolkaVMEventLogFunction::<3>.declare(context)?;
         revive_llvm_context::PolkaVMEventLogFunction::<4>.declare(context)?;
 
-        revive_llvm_context::PolkaVMDivisionFunction.declare(context)?;
-        revive_llvm_context::PolkaVMSignedDivisionFunction.declare(context)?;
-        revive_llvm_context::PolkaVMRemainderFunction.declare(context)?;
-        revive_llvm_context::PolkaVMSignedRemainderFunction.declare(context)?;
+        //revive_llvm_context::PolkaVMDivisionFunction.declare(context)?;
+        //revive_llvm_context::PolkaVMSignedDivisionFunction.declare(context)?;
+        //revive_llvm_context::PolkaVMRemainderFunction.declare(context)?;
+        //revive_llvm_context::PolkaVMSignedRemainderFunction.declare(context)?;
 
         revive_llvm_context::PolkaVMSbrkFunction.declare(context)?;
 
@@ -258,10 +258,10 @@ where
         revive_llvm_context::PolkaVMEventLogFunction::<3>.into_llvm(context)?;
         revive_llvm_context::PolkaVMEventLogFunction::<4>.into_llvm(context)?;
 
-        revive_llvm_context::PolkaVMDivisionFunction.into_llvm(context)?;
-        revive_llvm_context::PolkaVMSignedDivisionFunction.into_llvm(context)?;
-        revive_llvm_context::PolkaVMRemainderFunction.into_llvm(context)?;
-        revive_llvm_context::PolkaVMSignedRemainderFunction.into_llvm(context)?;
+        //revive_llvm_context::PolkaVMDivisionFunction.into_llvm(context)?;
+        //revive_llvm_context::PolkaVMSignedDivisionFunction.into_llvm(context)?;
+        //revive_llvm_context::PolkaVMRemainderFunction.into_llvm(context)?;
+        //revive_llvm_context::PolkaVMSignedRemainderFunction.into_llvm(context)?;
 
         revive_llvm_context::PolkaVMSbrkFunction.into_llvm(context)?;