events fn

Signed-off-by: xermicus <cyrill@parity.io>

crates/llvm-context/src/polkavm/context/mod.rs

@@ -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`.

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

@@ -7,11 +7,6 @@ use crate::polkavm::Dependency;
 use crate::polkavm_const::runtime_api;
 
 /// Translates a log or event call.
-///
-/// TODO: Splitting up into dedicated functions (log0..log4)
-/// could potentially decrease code sizes (LLVM can still decide to inline).
-/// However, passing many i256 parameters could also hurt a bit.
-/// Should be reviewed after 64bit support.
 pub fn log<'ctx, D>(
     context: &mut Context<'ctx, D>,
     input_offset: inkwell::values::IntValue<'ctx>,
@@ -29,14 +24,48 @@ where
         "event_input_offset",
     )?;
 
-    let arguments = if topics.is_empty() {
-        [
-            context.xlen_type().const_zero().as_basic_value_enum(),
-            context.xlen_type().const_zero().as_basic_value_enum(),
-            input_pointer.as_basic_value_enum(),
-            input_length.as_basic_value_enum(),
-        ]
-    } else {
+    if topics.is_empty() {
+        let _ = context.build_runtime_call(
+            runtime_api::DEPOSIT_EVENT,
+            &[
+                context.xlen_type().const_zero().as_basic_value_enum(),
+                context.xlen_type().const_zero().as_basic_value_enum(),
+                input_pointer.as_basic_value_enum(),
+                input_length.as_basic_value_enum(),
+            ],
+        );
+        return Ok(());
+    }
+
+    let name = match topics.len() {
+        1 => "__log_1",
+        2 => "__log_2",
+        3 => "__log_3",
+        4 => "__log_4",
+        _ => unreachable!(),
+    };
+    let function = context.module().get_function(name).unwrap_or_else(|| {
+        let position = context.basic_block();
+
+        let mut parameters = vec![context.xlen_type().into(), context.xlen_type().into()];
+        parameters.extend_from_slice(
+            &topics
+                .iter()
+                .map(|_| context.word_type().into())
+                .collect::<Vec<_>>(),
+        );
+        let function = context.module().add_function(
+            name,
+            context.void_type().fn_type(&parameters, false),
+            None,
+        );
+        let block_entry = context.llvm().append_basic_block(function, "entry");
+        context.set_basic_block(block_entry);
+
+        let parameters = function.get_param_iter().collect::<Vec<_>>();
+        let topics = &parameters[2..];
+        let input_offset = parameters.first().unwrap();
+        let input_length = parameters.get(1).unwrap();
         let topics_buffer_size = topics.len() * revive_common::BYTE_LENGTH_WORD;
         let topics_buffer_pointer = context.build_alloca(
             context.byte_type().array_type(topics_buffer_size as u32),
@@ -46,35 +75,59 @@ where
             let topic_buffer_offset = context
                 .xlen_type()
                 .const_int((n * revive_common::BYTE_LENGTH_WORD) as u64, false);
-            context.build_store(
-                context.build_gep(
-                    topics_buffer_pointer,
-                    &[context.xlen_type().const_zero(), topic_buffer_offset],
-                    context.byte_type(),
-                    "topic_buffer_gep",
-                ),
-                context.build_byte_swap(topic.as_basic_value_enum())?,
-            )?;
+            context
+                .build_store(
+                    context.build_gep(
+                        topics_buffer_pointer,
+                        &[context.xlen_type().const_zero(), topic_buffer_offset],
+                        context.byte_type(),
+                        "topic_buffer_gep",
+                    ),
+                    context
+                        .build_byte_swap(topic.as_basic_value_enum())
+                        .unwrap(),
+                )
+                .unwrap();
         }
-        [
+        let arguments = [
             context
                 .builder()
                 .build_ptr_to_int(
                     topics_buffer_pointer.value,
                     context.xlen_type(),
                     "event_topics_offset",
-                )?
+                )
+                .unwrap()
                 .as_basic_value_enum(),
             context
                 .xlen_type()
                 .const_int(topics_buffer_size as u64, false)
                 .as_basic_value_enum(),
-            input_pointer.as_basic_value_enum(),
+            input_offset.as_basic_value_enum(),
             input_length.as_basic_value_enum(),
-        ]
-    };
+        ];
 
-    let _ = context.build_runtime_call(runtime_api::DEPOSIT_EVENT, &arguments);
+        let _ = context.build_runtime_call(runtime_api::DEPOSIT_EVENT, &arguments);
+        context.builder().build_return(None).unwrap();
+
+        context.set_basic_block(position);
+
+        function
+    });
+
+    let mut arguments = vec![
+        input_pointer.as_basic_value_enum().into(),
+        input_length.as_basic_value_enum().into(),
+    ];
+    arguments.extend_from_slice(
+        &topics
+            .iter()
+            .map(|value| value.as_basic_value_enum().into())
+            .collect::<Vec<_>>(),
+    );
+    let _ = context
+        .builder()
+        .build_direct_call(function, &arguments[..], "call_log");
 
     Ok(())
 }