zink/storage/
dkmapping.rs

1//! Double key mapping
2
3use crate::{
4    ffi,
5    storage::{StorageValue, TransientStorageValue},
6    Asm,
7};
8
9/// Storage mapping interface
10pub trait DoubleKeyMapping {
11    const STORAGE_SLOT: i32;
12
13    type Key1: Asm;
14    type Key2: Asm;
15    type Value: StorageValue;
16
17    #[cfg(not(target_family = "wasm"))]
18    fn storage_key(key1: Self::Key1, key2: Self::Key2) -> [u8; 32];
19
20    /// Get value from storage key.
21    #[inline(always)]
22    fn get(key1: Self::Key1, key2: Self::Key2) -> Self::Value {
23        load_double_key(key1, key2, Self::STORAGE_SLOT);
24        Self::Value::sload()
25    }
26
27    /// Set key and value
28    #[inline(always)]
29    fn set(key1: Self::Key1, key2: Self::Key2, value: Self::Value) {
30        value.push();
31        load_double_key(key1, key2, Self::STORAGE_SLOT);
32        unsafe {
33            ffi::evm::sstore();
34        }
35    }
36}
37
38/// Transient storage mapping interface
39pub trait DoubleKeyTransientMapping {
40    const STORAGE_SLOT: i32;
41
42    type Key1: Asm;
43    type Key2: Asm;
44    type Value: TransientStorageValue;
45
46    #[cfg(not(target_family = "wasm"))]
47    fn storage_key(key1: Self::Key1, key2: Self::Key2) -> [u8; 32];
48
49    /// Get value from transient storage key.
50    #[inline(always)]
51    fn get(key1: Self::Key1, key2: Self::Key2) -> Self::Value {
52        load_double_key(key1, key2, Self::STORAGE_SLOT);
53        Self::Value::tload()
54    }
55
56    /// Set key and value in transient storage
57    #[inline(always)]
58    fn set(key1: Self::Key1, key2: Self::Key2, value: Self::Value) {
59        value.push();
60        load_double_key(key1, key2, Self::STORAGE_SLOT);
61        unsafe {
62            ffi::evm::tstore();
63        }
64    }
65}
66
67/// Load storage key to stack
68#[inline(always)]
69pub fn load_double_key(key1: impl Asm, key2: impl Asm, index: i32) {
70    unsafe {
71        ffi::label_reserve_mem_64();
72
73        // write key1 to memory
74        key1.push();
75        ffi::evm::push0();
76        ffi::evm::mstore();
77
78        // write index to memory
79        index.push();
80        ffi::asm::push_u8(0x20);
81        ffi::evm::mstore();
82
83        // hash key
84        ffi::asm::push_u8(0x40);
85        ffi::evm::push0();
86        ffi::evm::keccak256();
87
88        // stores the hash
89        ffi::evm::push0();
90        ffi::evm::mstore();
91
92        // write index to memory
93        key2.push();
94        ffi::asm::push_u8(0x20);
95        ffi::evm::mstore();
96
97        // hash key
98        ffi::asm::push_u8(0x40);
99        ffi::evm::push0();
100        ffi::evm::keccak256();
101    }
102}
zink/storage/dkmapping.rs

zink/storage/
dkmapping.rs
