1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
#![cfg_attr(not(feature = "std"), no_std)]
#[cfg(test)]
mod tests;
use core::marker::PhantomData;
use darwinia_evm_precompile_utils::{prelude::*, revert, PrecompileHelper};
use fp_evm::{
Context, ExitRevert, ExitSucceed, Precompile, PrecompileFailure, PrecompileOutput,
PrecompileResult,
};
const PALLET_PREFIX_LENGTH: usize = 16;
#[selector]
pub enum Action {
StateGetStorage = "state_storage(bytes)",
}
pub trait StorageFilterT {
fn allow(prefix: &[u8]) -> bool;
}
pub struct StateStorage<T, F> {
_marker: PhantomData<(T, F)>,
}
impl<T, F> Precompile for StateStorage<T, F>
where
T: darwinia_evm::Config,
F: StorageFilterT,
{
fn execute(
input: &[u8],
target_gas: Option<u64>,
context: &Context,
is_static: bool,
) -> PrecompileResult {
let mut helper = PrecompileHelper::<T>::new(input, target_gas, context, is_static);
let (selector, _data) = helper.split_input()?;
let action = Action::from_u32(selector)?;
helper.check_state_modifier(StateMutability::View)?;
let output = match action {
Action::StateGetStorage => {
let mut reader = helper.reader()?;
reader.expect_arguments(1)?;
let key: Bytes = reader.read()?;
if key.0.len() < PALLET_PREFIX_LENGTH || !F::allow(&key.0[0..PALLET_PREFIX_LENGTH])
{
return Err(revert("Read restriction"));
}
helper.record_db_gas(1, 0)?;
frame_support::storage::unhashed::get_raw(&key.0)
},
};
Ok(PrecompileOutput {
exit_status: ExitSucceed::Returned,
cost: helper.used_gas(),
output: EvmDataWriter::new()
.write::<Bytes>(output.unwrap_or_default().as_slice().into())
.build(),
logs: Default::default(),
})
}
}