astar_runtime/
precompiles.rs

1// This file is part of Astar.
2
3// Copyright (C) Stake Technologies Pte.Ltd.
4// SPDX-License-Identifier: GPL-3.0-or-later
5
6// Astar is free software: you can redistribute it and/or modify
7// it under the terms of the GNU General Public License as published by
8// the Free Software Foundation, either version 3 of the License, or
9// (at your option) any later version.
10
11// Astar is distributed in the hope that it will be useful,
12// but WITHOUT ANY WARRANTY; without even the implied warranty of
13// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14// GNU General Public License for more details.
15
16// You should have received a copy of the GNU General Public License
17// along with Astar. If not, see <http://www.gnu.org/licenses/>.
18
19//! The Astar Network EVM precompiles. This can be compiled with ``#[no_std]`, ready for Wasm.
20
21use crate::{Runtime, RuntimeCall};
22use astar_primitives::precompiles::DispatchFilterValidate;
23use frame_support::{parameter_types, traits::Contains};
24use pallet_evm_precompile_assets_erc20::Erc20AssetsPrecompileSet;
25use pallet_evm_precompile_blake2::Blake2F;
26use pallet_evm_precompile_bn128::{Bn128Add, Bn128Mul, Bn128Pairing};
27use pallet_evm_precompile_dapp_staking::DappStakingV3Precompile;
28use pallet_evm_precompile_dispatch::Dispatch;
29use pallet_evm_precompile_dispatch_lockdrop::DispatchLockdrop;
30use pallet_evm_precompile_ed25519::Ed25519Verify;
31use pallet_evm_precompile_modexp::Modexp;
32use pallet_evm_precompile_sha3fips::Sha3FIPS256;
33use pallet_evm_precompile_simple::{ECRecover, ECRecoverPublicKey, Identity, Ripemd160, Sha256};
34use pallet_evm_precompile_sr25519::Sr25519Precompile;
35use pallet_evm_precompile_substrate_ecdsa::SubstrateEcdsaPrecompile;
36use pallet_evm_precompile_xcm::XcmPrecompile;
37use precompile_utils::precompile_set::*;
38use sp_core::ConstU32;
39use sp_std::fmt::Debug;
40
41/// The asset precompile address prefix. Addresses that match against this prefix will be routed
42/// to Erc20AssetsPrecompileSet
43pub const ASSET_PRECOMPILE_ADDRESS_PREFIX: &[u8] = &[255u8; 4];
44parameter_types! {
45    pub AssetPrefix: &'static [u8] = ASSET_PRECOMPILE_ADDRESS_PREFIX;
46}
47
48/// Precompile checks for ethereum spec precompiles
49/// We allow DELEGATECALL to stay compliant with Ethereum behavior.
50type EthereumPrecompilesChecks = (AcceptDelegateCall, CallableByContract, CallableByPrecompile);
51
52/// Filter that only allows whitelisted runtime call to pass through dispatch precompile
53pub struct WhitelistedCalls;
54
55impl Contains<RuntimeCall> for WhitelistedCalls {
56    fn contains(t: &RuntimeCall) -> bool {
57        match t {
58            RuntimeCall::Utility(pallet_utility::Call::batch { calls })
59            | RuntimeCall::Utility(pallet_utility::Call::batch_all { calls }) => {
60                calls.iter().all(|call| WhitelistedCalls::contains(call))
61            }
62            RuntimeCall::DappStaking(_) => true,
63            RuntimeCall::Assets(pallet_assets::Call::transfer { .. }) => true,
64            RuntimeCall::XTokens(orml_xtokens::Call::transfer_multiasset_with_fee { .. }) => true,
65            RuntimeCall::XTokens(orml_xtokens::Call::transfer_multiasset { .. }) => true,
66            // Governance related calls
67            RuntimeCall::Democracy(_)
68            | RuntimeCall::Treasury(_)
69            | RuntimeCall::CommunityTreasury(_)
70            | RuntimeCall::Preimage(_) => true,
71            _ => false,
72        }
73    }
74}
75
76/// Filter that only allows whitelisted runtime call to pass through dispatch-lockdrop precompile
77pub struct WhitelistedLockdropCalls;
78
79impl Contains<RuntimeCall> for WhitelistedLockdropCalls {
80    fn contains(t: &RuntimeCall) -> bool {
81        match t {
82            RuntimeCall::Utility(pallet_utility::Call::batch { calls })
83            | RuntimeCall::Utility(pallet_utility::Call::batch_all { calls }) => calls
84                .iter()
85                .all(|call| WhitelistedLockdropCalls::contains(call)),
86            RuntimeCall::DappStaking(pallet_dapp_staking::Call::unbond_and_unstake { .. }) => true,
87            RuntimeCall::DappStaking(pallet_dapp_staking::Call::withdraw_unbonded { .. }) => true,
88            RuntimeCall::Balances(pallet_balances::Call::transfer_all { .. }) => true,
89            RuntimeCall::Balances(pallet_balances::Call::transfer_keep_alive { .. }) => true,
90            RuntimeCall::Balances(pallet_balances::Call::transfer_allow_death { .. }) => true,
91            RuntimeCall::Assets(pallet_assets::Call::transfer { .. }) => true,
92            _ => false,
93        }
94    }
95}
96
97/// The PrecompileSet installed in the Astar runtime.
98#[precompile_utils::precompile_name_from_address]
99pub type AstarPrecompilesSetAt<R, C> = (
100    // Ethereum precompiles:
101    // We allow DELEGATECALL to stay compliant with Ethereum behavior.
102    PrecompileAt<AddressU64<1>, ECRecover, EthereumPrecompilesChecks>,
103    PrecompileAt<AddressU64<2>, Sha256, EthereumPrecompilesChecks>,
104    PrecompileAt<AddressU64<3>, Ripemd160, EthereumPrecompilesChecks>,
105    PrecompileAt<AddressU64<4>, Identity, EthereumPrecompilesChecks>,
106    PrecompileAt<AddressU64<5>, Modexp, EthereumPrecompilesChecks>,
107    PrecompileAt<AddressU64<6>, Bn128Add, EthereumPrecompilesChecks>,
108    PrecompileAt<AddressU64<7>, Bn128Mul, EthereumPrecompilesChecks>,
109    PrecompileAt<AddressU64<8>, Bn128Pairing, EthereumPrecompilesChecks>,
110    PrecompileAt<AddressU64<9>, Blake2F, EthereumPrecompilesChecks>,
111    // Non-Astar specific nor Ethereum precompiles :
112    PrecompileAt<
113        AddressU64<1024>,
114        Sha3FIPS256<Runtime, ()>,
115        (CallableByContract, CallableByPrecompile),
116    >,
117    PrecompileAt<
118        AddressU64<1025>,
119        Dispatch<R, DispatchFilterValidate<RuntimeCall, WhitelistedCalls>>,
120        // Not callable from smart contract nor precompiles, only EOA accounts
121        // TODO: test this without the gensis hack for blacklisted
122        (),
123    >,
124    PrecompileAt<AddressU64<1026>, ECRecoverPublicKey, (CallableByContract, CallableByPrecompile)>,
125    PrecompileAt<AddressU64<1027>, Ed25519Verify, (CallableByContract, CallableByPrecompile)>,
126    // Astar specific precompiles:
127    PrecompileAt<
128        AddressU64<20481>,
129        DappStakingV3Precompile<R>,
130        (CallableByContract, CallableByPrecompile),
131    >,
132    PrecompileAt<
133        AddressU64<20482>,
134        Sr25519Precompile<R>,
135        (CallableByContract, CallableByPrecompile),
136    >,
137    PrecompileAt<
138        AddressU64<20483>,
139        SubstrateEcdsaPrecompile<R>,
140        (CallableByContract, CallableByPrecompile),
141    >,
142    PrecompileAt<
143        AddressU64<20484>,
144        XcmPrecompile<R, C>,
145        (
146            SubcallWithMaxNesting<1>,
147            CallableByContract,
148            CallableByPrecompile,
149        ),
150    >,
151    // Skipping 20485 and 20486 to make sure all network have consistent
152    // precompiles address
153    PrecompileAt<
154        AddressU64<20487>,
155        DispatchLockdrop<
156            R,
157            DispatchFilterValidate<RuntimeCall, WhitelistedLockdropCalls>,
158            ConstU32<8>,
159        >,
160        // Not callable from smart contract nor precompiled, only EOA accounts
161        (),
162    >,
163);
164
165pub type AstarPrecompiles<R, C> = PrecompileSetBuilder<
166    R,
167    (
168        // Skip precompiles if out of range.
169        PrecompilesInRangeInclusive<
170            // We take range as last precompile index, UPDATE this once new prcompile is added
171            (AddressU64<1>, AddressU64<20487>),
172            AstarPrecompilesSetAt<R, C>,
173        >,
174        // Prefixed precompile sets (XC20)
175        PrecompileSetStartingWith<AssetPrefix, Erc20AssetsPrecompileSet<R>, CallableByContract>,
176    ),
177>;