zink/primitives/
u256.rs

1#![allow(clippy::should_implement_trait)]
2
3use crate::{
4    ffi,
5    primitives::Bytes32,
6    storage::{StorageValue, TransientStorageValue},
7    Asm,
8};
9use core::ops::Sub;
10
11/// Account address
12#[repr(C)]
13#[derive(Clone, Copy, PartialEq, PartialOrd, Debug)]
14pub struct U256(Bytes32);
15
16impl U256 {
17    /// Returns empty value  
18    pub const fn empty() -> Self {
19        U256(Bytes32::empty())
20    }
21
22    /// u256 add
23    #[inline(always)]
24    pub fn add(self, other: Self) -> Self {
25        unsafe { ffi::u256_add(self, other) }
26    }
27
28    /// u256 less than
29    #[inline(always)]
30    pub fn lt(self, other: Self) -> bool {
31        unsafe { ffi::u256_lt(other, self) }
32    }
33
34    /// u256 eq
35    #[inline(always)]
36    pub fn eq(self, other: Self) -> bool {
37        self.0.eq(other.0)
38    }
39
40    /// u256 sub
41    #[inline(always)]
42    pub fn sub(self, other: Self) -> Self {
43        unsafe { ffi::u256_sub(other, self) }
44    }
45
46    /// u256 div
47    #[inline(always)]
48    pub fn div(self, other: Self) -> Self {
49        unsafe { ffi::u256_div(self, other) }
50    }
51
52    /// max of u256
53    #[inline(always)]
54    pub fn max() -> Self {
55        unsafe { ffi::u256_max() }
56    }
57
58    pub fn to_bytes32(&self) -> Bytes32 {
59        self.0
60    }
61
62    #[cfg(not(target_family = "wasm"))]
63    pub fn bytes32(&self) -> [u8; 32] {
64        self.0 .0 // [u8; 32] in non-WASM
65    }
66
67    #[inline(always)]
68    pub fn addmod(self, other: Self, modulus: Self) -> Self {
69        unsafe { ffi::u256_addmod(modulus, other, self) }
70    }
71
72    /// Mulmod for U256
73    #[inline(always)]
74    pub fn mulmod(self, other: Self, modulus: Self) -> Self {
75        unsafe { ffi::u256_mulmod(modulus, other, self) }
76    }
77}
78
79impl Sub for U256 {
80    type Output = Self;
81
82    /// u256 sub
83    #[inline(always)]
84    fn sub(self, other: Self) -> Self::Output {
85        unsafe { ffi::u256_sub(self, other) }
86    }
87}
88
89impl Asm for U256 {
90    #[inline(always)]
91    fn push(self) {
92        unsafe { ffi::bytes::push_bytes32(self.0) }
93    }
94
95    #[cfg(not(target_family = "wasm"))]
96    fn bytes32(&self) -> [u8; 32] {
97        self.bytes32() // Delegate to the instance method
98    }
99}
100
101impl StorageValue for U256 {
102    #[inline(always)]
103    fn sload() -> Self {
104        Self(unsafe { ffi::bytes::sload_bytes32() })
105    }
106}
107
108impl TransientStorageValue for U256 {
109    #[inline(always)]
110    fn tload() -> Self {
111        Self(unsafe { ffi::bytes::tload_bytes32() })
112    }
113}
114
115impl From<u64> for U256 {
116    fn from(value: u64) -> Self {
117        #[cfg(target_family = "wasm")]
118        {
119            U256(Bytes32(value as i32))
120        }
121        #[cfg(not(target_family = "wasm"))]
122        {
123            // On non-WASM, Bytes32 is [u8; 32]
124            let mut bytes = [0u8; 32];
125            bytes[24..32].copy_from_slice(&value.to_be_bytes());
126            U256(Bytes32(bytes))
127        }
128    }
129}
zink/primitives/u256.rs

zink/primitives/
u256.rs
