Add Evm Allowlist Feature to SBT Pallet

pallets/manta-sbt/Cargo.toml

@@ -7,7 +7,7 @@ license = 'GPL-3.0'
 name = 'pallet-manta-sbt'
 readme = 'README.md'
 repository = 'https://github.com/Manta-Network/Manta/'
-version = '4.0.2'
+version = '4.0.4'
 
 [[bin]]
 name = "precompute_coins"
@@ -20,6 +20,7 @@ default = ["std"]
 # Runtime Benchmarks
 runtime-benchmarks = [
   "frame-benchmarking",
+  "libsecp256k1/hmac",
   "frame-support/runtime-benchmarks",
   "frame-system/runtime-benchmarks",
   "manta-primitives/runtime-benchmarks",
@@ -28,6 +29,8 @@ runtime-benchmarks = [
 # Standard Library
 std = [
   "anyhow/std",
+  "sha3/std",
+  "libsecp256k1/std",
   "frame-benchmarking/std",
   "frame-support/std",
   "frame-system/std",
@@ -36,6 +39,8 @@ std = [
   "sp-runtime/std",
   "sp-api/std",
   "sp-std/std",
+  "sp-core/std",
+  "sp-io/std",
   "manta-primitives/std",
   "manta-accounting/std",
   "manta-crypto/std",
@@ -83,7 +88,9 @@ precompute-coins = [
 anyhow = { version = "1.0.55", optional = true, default-features = false }
 base64 = { version = "0.20", default-features = false, features = ["alloc"] }
 indoc = { version = "1.0.3", optional = true, default-features = false }
+libsecp256k1 = { version = "0.7", default-features = false, features = ["hmac", "static-context"], optional = true }
 rand_chacha = { version = "0.3.1", optional = true, default-features = false }
+sha3 = { version = "0.10.6", default-features = false }
 tempfile = { version = "3.3.0", optional = true, default-features = false }
 
 # substrate dependencies
@@ -94,6 +101,8 @@ scale-codec = { package = "parity-scale-codec", version = '3.1.2', default-featu
 scale-info = { version = "2.1.2", default-features = false, features = ["derive"] }
 sp-api = { git = 'https://github.com/paritytech/substrate.git', branch = "polkadot-v0.9.28", optional = true, default-features = false }
 sp-blockchain = { git = 'https://github.com/paritytech/substrate.git', branch = "polkadot-v0.9.28", optional = true, default-features = false }
+sp-core = { git = 'https://github.com/paritytech/substrate.git', branch = "polkadot-v0.9.28", default-features = false }
+sp-io = { git = 'https://github.com/paritytech/substrate.git', branch = "polkadot-v0.9.28", default-features = false }
 sp-runtime = { git = "https://github.com/paritytech/substrate.git", branch = "polkadot-v0.9.28", default-features = false }
 sp-std = { git = 'https://github.com/paritytech/substrate.git', branch = "polkadot-v0.9.28", default-features = false }
 
@@ -116,8 +125,7 @@ pallet-asset-manager = { path = "../asset-manager" }
 pallet-assets = { git = "https://github.com/paritytech/substrate.git", branch = "polkadot-v0.9.28" }
 pallet-balances = { git = 'https://github.com/paritytech/substrate.git', branch = "polkadot-v0.9.28" }
 pallet-manta-pay = { path = "../manta-pay" }
+pallet-timestamp = { git = "https://github.com/paritytech/substrate.git", branch = "polkadot-v0.9.28" }
 pallet-tx-pause = { path = "../tx-pause" }
-sp-core = { git = "https://github.com/paritytech/substrate.git", branch = "polkadot-v0.9.28" }
-sp-io = { git = "https://github.com/paritytech/substrate.git", branch = "polkadot-v0.9.28" }
 tempfile = "3.3.0"
 xcm = { git = 'https://github.com/paritytech/polkadot.git', branch = "release-v0.9.28" }

pallets/manta-sbt/src/benchmark/mod.rs

@@ -15,15 +15,22 @@
 // along with Manta.  If not, see <http://www.gnu.org/licenses/>.
 
 use crate::{
-    benchmark::precomputed_coins::TO_PRIVATE, AccountId, Box, Call, Config, Pallet, TransferPost,
+    benchmark::precomputed_coins::TO_PRIVATE, AccountId, Box, Call, Config, EvmAddressType,
+    MintType, Pallet, Pallet as MantaSBTPallet, TransferPost,
 };
 use frame_benchmarking::{benchmarks, impl_benchmark_test_suite, vec, whitelisted_caller};
 use frame_support::traits::{Currency, Get};
 use frame_system::RawOrigin;
 use scale_codec::Decode;
+use sp_core::H160;
+use sp_io::hashing::keccak_256;
 
 mod precomputed_coins;
 
+fn alice() -> libsecp256k1::SecretKey {
+    libsecp256k1::SecretKey::parse(&keccak_256(b"Alice")).unwrap()
+}
+
 benchmarks! {
     where_clause {  where T::AccountId: From<AccountId> + Into<AccountId> }
     to_private {
@@ -45,6 +52,66 @@ benchmarks! {
     }: reserve_sbt (
         RawOrigin::Signed(caller)
     )
+
+    change_allowlist_account{
+        let caller: T::AccountId = whitelisted_caller();
+    }: change_allowlist_account (
+        RawOrigin::Root,
+        Some(caller)
+    )
+
+    allowlist_evm_account {
+        let caller: T::AccountId = whitelisted_caller();
+        MantaSBTPallet::<T>::change_allowlist_account(
+            RawOrigin::Root.into(),
+            Some(caller.clone())
+        )?;
+    }: allowlist_evm_account (
+        RawOrigin::Signed(caller),
+        EvmAddressType::Bab(H160::default())
+    )
+
+    set_mint_chain_info {
+    }: set_mint_chain_info (
+        RawOrigin::Root,
+        MintType::Bab,
+        0,
+        5u32.into(),
+        Some(10u32.into())
+    )
+
+    mint_sbt_eth {
+        let caller: T::AccountId = whitelisted_caller();
+        MantaSBTPallet::<T>::change_allowlist_account(
+            RawOrigin::Root.into(),
+            Some(caller.clone())
+        )?;
+        let bab_alice = EvmAddressType::Bab(MantaSBTPallet::<T>::eth_address(&alice()));
+        MantaSBTPallet::<T>::set_mint_chain_info(
+            RawOrigin::Root.into(),
+            MintType::Bab,
+            0,
+            0_u32.into(),
+            None
+        )?;
+
+        MantaSBTPallet::<T>::allowlist_evm_account(
+            RawOrigin::Signed(caller.clone()).into(),
+            bab_alice,
+        )?;
+        let mint_post = TransferPost::decode(&mut &*TO_PRIVATE).unwrap();
+
+        let signature = MantaSBTPallet::<T>::eth_sign(&alice(), &mint_post.proof, 0);
+
+    }: mint_sbt_eth(
+        RawOrigin::Signed(caller),
+        Box::new(mint_post),
+        signature,
+        bab_alice,
+        Some(0),
+        Some(0),
+        Some(vec![0].try_into().unwrap())
+    )
 }
 
 impl_benchmark_test_suite!(Pallet, crate::mock::new_test_ext(), crate::mock::Test);