logs

zink-lang · Dec 6, 2024 · a67fde1 · a67fde1
1 parent 16ecb5c
commit a67fde1
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 8 deletions.
diff --git a/zink/src/ffi/asm.rs b/zink/src/ffi/asm.rs
@@ -1,5 +1,7 @@
 //! Assembly FFI.
 
+use crate::primitives::{Bytes32, U256};
+
 #[link(wasm_import_module = "asm")]
 #[allow(improper_ctypes)]
 extern "C" {
@@ -79,6 +81,8 @@ extern "C" {
     /// Revert with message in 128 bytes
     pub fn revert4(message: &'static str);
 
+    pub fn cast_bytes32(a: U256) -> Bytes32;
+
     /// Load a 8-bit signed integer from the storage.
     pub fn sload_i8() -> i8;
 

diff --git a/zink/src/primitives/u256.rs b/zink/src/primitives/u256.rs
@@ -48,18 +48,11 @@ impl U256 {
         unsafe { ffi::u256_max() }
     }
 
-
-    #[cfg(target_family = "wasm")]
+    /// U256 to bytes32
     pub fn to_bytes32(&self) -> Bytes32 {
         unsafe { ffi::asm::cast_bytes32(*self) }
     }
 
-    #[cfg(not(target_family = "wasm"))]
-    pub fn to_bytes32(&self) -> Bytes32 {
-        // Bytes32::from(self.0)
-        todo!()
-    }
-
     /// Addmod for U256
     #[inline(always)]
     pub fn addmod(self, other: Self, modulus: Self) -> Self {