difftreelog
fix check errors in erc
in: master
1 file changed
pallets/common/src/erc.rsdiffbeforeafterboth1// Copyright 2019-2022 Unique Network (Gibraltar) Ltd.2// This file is part of Unique Network.34// Unique Network is free software: you can redistribute it and/or modify5// it under the terms of the GNU General Public License as published by6// the Free Software Foundation, either version 3 of the License, or7// (at your option) any later version.89// Unique Network is distributed in the hope that it will be useful,10// but WITHOUT ANY WARRANTY; without even the implied warranty of11// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the12// GNU General Public License for more details.1314// You should have received a copy of the GNU General Public License15// along with Unique Network. If not, see <http://www.gnu.org/licenses/>.1617use evm_coder::{18 solidity_interface, solidity, ToLog,19 types::*,20 execution::{Result, Error},21};22pub use pallet_evm::{PrecompileOutput, PrecompileResult, PrecompileHandle, account::CrossAccountId};23use pallet_evm_coder_substrate::dispatch_to_evm;24use sp_core::{H160, U256, H256};25use sp_std::vec::Vec;26use up_data_structs::{Property, SponsoringRateLimit, NestingRule, OwnerRestrictedSet, AccessMode};27use alloc::format;2829use crate::{Pallet, CollectionHandle, Config, CollectionProperties};3031#[derive(ToLog)]32pub enum CollectionHelpersEvents {33 CollectionCreated {34 #[indexed]35 owner: address,36 #[indexed]37 collection_id: address,38 },39}4041/// Does not always represent a full collection, for RFT it is either42/// collection (Implementing ERC721), or specific collection token (Implementing ERC20)43pub trait CommonEvmHandler {44 const CODE: &'static [u8];4546 fn call(self, handle: &mut impl PrecompileHandle) -> Option<PrecompileResult>;47}4849#[solidity_interface(name = "Collection")]50impl<T: Config> CollectionHandle<T>51// where52// T::AccountId: From<H256>53{54 fn set_collection_property(&mut self, caller: caller, key: string, value: bytes) -> Result<()> {55 let caller = T::CrossAccountId::from_eth(caller);56 let key = <Vec<u8>>::from(key)57 .try_into()58 .map_err(|_| "key too large")?;59 let value = value.try_into().map_err(|_| "value too large")?;6061 <Pallet<T>>::set_collection_property(self, &caller, Property { key, value })62 .map_err(dispatch_to_evm::<T>)63 }6465 fn delete_collection_property(&mut self, caller: caller, key: string) -> Result<()> {66 let caller = T::CrossAccountId::from_eth(caller);67 let key = <Vec<u8>>::from(key)68 .try_into()69 .map_err(|_| "key too large")?;7071 <Pallet<T>>::delete_collection_property(self, &caller, key).map_err(dispatch_to_evm::<T>)72 }7374 /// Throws error if key not found75 fn collection_property(&self, key: string) -> Result<bytes> {76 let key = <Vec<u8>>::from(key)77 .try_into()78 .map_err(|_| "key too large")?;7980 let props = <CollectionProperties<T>>::get(self.id);81 let prop = props.get(&key).ok_or("key not found")?;8283 Ok(prop.to_vec())84 }8586 fn set_collection_sponsor(&mut self, caller: caller, sponsor: address) -> Result<void> {87 check_is_owner(caller, self)?;8889 let sponsor = T::CrossAccountId::from_eth(sponsor);90 self.set_sponsor(sponsor.as_sub().clone())91 .map_err(dispatch_to_evm::<T>)?;92 save(self)93 }9495 fn confirm_collection_sponsorship(&mut self, caller: caller) -> Result<void> {96 let caller = T::CrossAccountId::from_eth(caller);97 if !self98 .confirm_sponsorship(caller.as_sub())99 .map_err(dispatch_to_evm::<T>)?100 {101 return Err(Error::Revert("Caller is not set as sponsor".into()));102 }103 save(self)104 }105106 #[solidity(rename_selector = "setCollectionLimit")]107 fn set_int_limit(&mut self, caller: caller, limit: string, value: uint32) -> Result<void> {108 check_is_owner(caller, self)?;109 let mut limits = self.limits.clone();110111 match limit.as_str() {112 "accountTokenOwnershipLimit" => {113 limits.account_token_ownership_limit = Some(value);114 }115 "sponsoredDataSize" => {116 limits.sponsored_data_size = Some(value);117 }118 "sponsoredDataRateLimit" => {119 limits.sponsored_data_rate_limit = Some(SponsoringRateLimit::Blocks(value));120 }121 "tokenLimit" => {122 limits.token_limit = Some(value);123 }124 "sponsorTransferTimeout" => {125 limits.sponsor_transfer_timeout = Some(value);126 }127 "sponsorApproveTimeout" => {128 limits.sponsor_approve_timeout = Some(value);129 }130 _ => {131 return Err(Error::Revert(format!(132 "Unknown integer limit \"{}\"",133 limit134 )))135 }136 }137 self.limits = <Pallet<T>>::clamp_limits(self.mode.clone(), &self.limits, limits)138 .map_err(dispatch_to_evm::<T>)?;139 save(self)140 }141142 #[solidity(rename_selector = "setCollectionLimit")]143 fn set_bool_limit(&mut self, caller: caller, limit: string, value: bool) -> Result<void> {144 check_is_owner(caller, self)?;145 let mut limits = self.limits.clone();146147 match limit.as_str() {148 "ownerCanTransfer" => {149 limits.owner_can_transfer = Some(value);150 }151 "ownerCanDestroy" => {152 limits.owner_can_destroy = Some(value);153 }154 "transfersEnabled" => {155 limits.transfers_enabled = Some(value);156 }157 _ => {158 return Err(Error::Revert(format!(159 "Unknown boolean limit \"{}\"",160 limit161 )))162 }163 }164 self.limits = <Pallet<T>>::clamp_limits(self.mode.clone(), &self.limits, limits)165 .map_err(dispatch_to_evm::<T>)?;166 save(self)167 }168169 fn contract_address(&self, _caller: caller) -> Result<address> {170 Ok(crate::eth::collection_id_to_address(self.id))171 }172173 // fn add_admin_substrate(&self, caller: caller, new_admin: uint256) -> Result<void> {174 // let mut new_admin_h256 = H256::default();175 // new_admin.to_little_endian(&mut new_admin_h256.0);176 // let account_id = T::AccountId::from(new_admin_h256);177 // let caller = T::CrossAccountId::from_eth(caller);178 // let new_admin = T::CrossAccountId::from_sub(account_id);179 // <Pallet<T>>::toggle_admin(&self, &caller, &new_admin, true)180 // .map_err(dispatch_to_evm::<T>)?;181 // Ok(())182 // }183184 // fn remove_admin_substrate(&self, caller: caller, new_admin: uint256) -> Result<void> {185 // let mut new_admin_h256 = H256::default();186 // new_admin.to_little_endian(&mut new_admin_h256.0);187 // let account_id = T::AccountId::from(new_admin_h256);188 // let caller = T::CrossAccountId::from_eth(caller);189 // let new_admin = T::CrossAccountId::from_sub(account_id);190 // <Pallet<T>>::toggle_admin(&self, &caller, &new_admin, false)191 // .map_err(dispatch_to_evm::<T>)?;192 // Ok(())193 // }194195 fn add_collection_admin(&self, caller: caller, new_admin: address) -> Result<void> {196 let caller = T::CrossAccountId::from_eth(caller);197 self.check_is_owner_or_admin(&caller)198 .map_err(dispatch_to_evm::<T>)?;199 let new_admin = T::CrossAccountId::from_eth(new_admin);200 <Pallet<T>>::toggle_admin(&self, &caller, &new_admin, true)201 .map_err(dispatch_to_evm::<T>)?;202 Ok(())203 }204205 fn remove_collection_admin(&self, caller: caller, admin: address) -> Result<void> {206 let caller = T::CrossAccountId::from_eth(caller);207 self.check_is_owner_or_admin(&caller)208 .map_err(dispatch_to_evm::<T>)?;209 let admin = T::CrossAccountId::from_eth(admin);210 <Pallet<T>>::toggle_admin(&self, &caller, &admin, false).map_err(dispatch_to_evm::<T>)?;211 Ok(())212 }213214 #[solidity(rename_selector = "setCollectionNesting")]215 fn set_nesting_bool(&mut self, caller: caller, enable: bool) -> Result<void> {216 let caller = T::CrossAccountId::from_eth(caller);217 self.check_is_owner_or_admin(&caller)218 .map_err(dispatch_to_evm::<T>)?;219 self.collection.permissions.nesting = Some(match enable {220 false => NestingRule::Disabled,221 true => NestingRule::Owner,222 });223 save(self);224 Ok(())225 }226227 #[solidity(rename_selector = "setCollectionNesting")]228 fn set_nesting(229 &mut self,230 caller: caller,231 enable: bool,232 collections: Vec<address>,233 ) -> Result<void> {234 if collections.is_empty() {235 return Err("No addresses provided".into());236 }237 if collections.len() >= OwnerRestrictedSet::bound() {238 return Err(Error::Revert(format!(239 "Out of bound: {} >= {}",240 collections.len(),241 OwnerRestrictedSet::bound()242 )));243 }244 let caller = T::CrossAccountId::from_eth(caller);245 self.check_is_owner_or_admin(&caller)246 .map_err(dispatch_to_evm::<T>)?;247 self.collection.permissions.nesting = Some(match enable {248 false => NestingRule::Disabled,249 true => {250 let mut bv = OwnerRestrictedSet::new();251 for i in collections {252 bv.try_insert(crate::eth::map_eth_to_id(&i).ok_or(Error::Revert(253 "Can't convert address into collection id".into(),254 ))?)255 .map_err(|e| Error::Revert(format!("{:?}", e)))?;256 }257 NestingRule::OwnerRestricted(bv)258 }259 });260 save(self);261 Ok(())262 }263264 fn set_collection_access(&mut self, caller: caller, mode: uint8) -> Result<void> {265 let caller = T::CrossAccountId::from_eth(caller);266 self.check_is_owner_or_admin(&caller)267 .map_err(dispatch_to_evm::<T>)?;268 self.collection.permissions.access = Some(match mode {269 0 => AccessMode::Normal,270 1 => AccessMode::AllowList,271 _ => return Err("Not supported access mode".into()),272 });273 save(self);274 Ok(())275 }276277 fn add_to_collection_allow_list(&self, caller: caller, user: address) -> Result<void> {278 let caller = check_is_owner_or_admin(caller, self)?;279 let user = T::CrossAccountId::from_eth(user);280 <Pallet<T>>::toggle_allowlist(self, &caller, &user, true).map_err(dispatch_to_evm::<T>)?;281 Ok(())282 }283284 fn remove_from_collection_allow_list(&self, caller: caller, user: address) -> Result<void> {285 let caller = check_is_owner_or_admin(caller, self)?;286 let user = T::CrossAccountId::from_eth(user);287 <Pallet<T>>::toggle_allowlist(self, &caller, &user, false).map_err(dispatch_to_evm::<T>)?;288 Ok(())289 }290291 fn set_collection_mint_mode(&mut self, caller: caller, mode: bool) -> Result<void> {292 check_is_owner_or_admin(caller, self)?;293 self.collection.permissions.mint_mode = Some(mode);294 save(self);295 Ok(())296 }297}298299fn check_is_owner<T: Config>(caller: caller, collection: &CollectionHandle<T>) -> Result<void> {300 let caller = T::CrossAccountId::from_eth(caller);301 collection302 .check_is_owner(&caller)303 .map_err(pallet_evm_coder_substrate::dispatch_to_evm::<T>)?;304 Ok(())305}306307fn check_is_owner_or_admin<T: Config>(308 caller: caller,309 collection: &CollectionHandle<T>,310) -> Result<T::CrossAccountId> {311 let caller = T::CrossAccountId::from_eth(caller);312 collection313 .check_is_owner_or_admin(&caller)314 .map_err(pallet_evm_coder_substrate::dispatch_to_evm::<T>)?;315 Ok(caller)316}317318fn save<T: Config>(collection: &CollectionHandle<T>) -> Result<void> {319 // TODO possibly delete for the lack of transaction320 collection321 .check_is_internal()322 .map_err(dispatch_to_evm::<T>)?;323 <crate::CollectionById<T>>::insert(collection.id, collection.collection.clone());324 Ok(())325}326327pub fn token_uri_key() -> up_data_structs::PropertyKey {328 b"tokenURI"329 .to_vec()330 .try_into()331 .expect("length < limit; qed")332}1// Copyright 2019-2022 Unique Network (Gibraltar) Ltd.2// This file is part of Unique Network.34// Unique Network is free software: you can redistribute it and/or modify5// it under the terms of the GNU General Public License as published by6// the Free Software Foundation, either version 3 of the License, or7// (at your option) any later version.89// Unique Network is distributed in the hope that it will be useful,10// but WITHOUT ANY WARRANTY; without even the implied warranty of11// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the12// GNU General Public License for more details.1314// You should have received a copy of the GNU General Public License15// along with Unique Network. If not, see <http://www.gnu.org/licenses/>.1617use evm_coder::{18 solidity_interface, solidity, ToLog,19 types::*,20 execution::{Result, Error},21};22pub use pallet_evm::{PrecompileOutput, PrecompileResult, PrecompileHandle, account::CrossAccountId};23use pallet_evm_coder_substrate::dispatch_to_evm;24use sp_std::vec::Vec;25use up_data_structs::{Property, SponsoringRateLimit, NestingRule, OwnerRestrictedSet, AccessMode};26use alloc::format;2728use crate::{Pallet, CollectionHandle, Config, CollectionProperties};2930#[derive(ToLog)]31pub enum CollectionHelpersEvents {32 CollectionCreated {33 #[indexed]34 owner: address,35 #[indexed]36 collection_id: address,37 },38}3940/// Does not always represent a full collection, for RFT it is either41/// collection (Implementing ERC721), or specific collection token (Implementing ERC20)42pub trait CommonEvmHandler {43 const CODE: &'static [u8];4445 fn call(self, handle: &mut impl PrecompileHandle) -> Option<PrecompileResult>;46}4748#[solidity_interface(name = "Collection")]49impl<T: Config> CollectionHandle<T>50// where51// T::AccountId: From<H256>52{53 fn set_collection_property(&mut self, caller: caller, key: string, value: bytes) -> Result<()> {54 let caller = T::CrossAccountId::from_eth(caller);55 let key = <Vec<u8>>::from(key)56 .try_into()57 .map_err(|_| "key too large")?;58 let value = value.try_into().map_err(|_| "value too large")?;5960 <Pallet<T>>::set_collection_property(self, &caller, Property { key, value })61 .map_err(dispatch_to_evm::<T>)62 }6364 fn delete_collection_property(&mut self, caller: caller, key: string) -> Result<()> {65 let caller = T::CrossAccountId::from_eth(caller);66 let key = <Vec<u8>>::from(key)67 .try_into()68 .map_err(|_| "key too large")?;6970 <Pallet<T>>::delete_collection_property(self, &caller, key).map_err(dispatch_to_evm::<T>)71 }7273 /// Throws error if key not found74 fn collection_property(&self, key: string) -> Result<bytes> {75 let key = <Vec<u8>>::from(key)76 .try_into()77 .map_err(|_| "key too large")?;7879 let props = <CollectionProperties<T>>::get(self.id);80 let prop = props.get(&key).ok_or("key not found")?;8182 Ok(prop.to_vec())83 }8485 fn set_collection_sponsor(&mut self, caller: caller, sponsor: address) -> Result<void> {86 check_is_owner(caller, self)?;8788 let sponsor = T::CrossAccountId::from_eth(sponsor);89 self.set_sponsor(sponsor.as_sub().clone())90 .map_err(dispatch_to_evm::<T>)?;91 save(self)92 }9394 fn confirm_collection_sponsorship(&mut self, caller: caller) -> Result<void> {95 let caller = T::CrossAccountId::from_eth(caller);96 if !self97 .confirm_sponsorship(caller.as_sub())98 .map_err(dispatch_to_evm::<T>)?99 {100 return Err(Error::Revert("Caller is not set as sponsor".into()));101 }102 save(self)103 }104105 #[solidity(rename_selector = "setCollectionLimit")]106 fn set_int_limit(&mut self, caller: caller, limit: string, value: uint32) -> Result<void> {107 check_is_owner(caller, self)?;108 let mut limits = self.limits.clone();109110 match limit.as_str() {111 "accountTokenOwnershipLimit" => {112 limits.account_token_ownership_limit = Some(value);113 }114 "sponsoredDataSize" => {115 limits.sponsored_data_size = Some(value);116 }117 "sponsoredDataRateLimit" => {118 limits.sponsored_data_rate_limit = Some(SponsoringRateLimit::Blocks(value));119 }120 "tokenLimit" => {121 limits.token_limit = Some(value);122 }123 "sponsorTransferTimeout" => {124 limits.sponsor_transfer_timeout = Some(value);125 }126 "sponsorApproveTimeout" => {127 limits.sponsor_approve_timeout = Some(value);128 }129 _ => {130 return Err(Error::Revert(format!(131 "Unknown integer limit \"{}\"",132 limit133 )))134 }135 }136 self.limits = <Pallet<T>>::clamp_limits(self.mode.clone(), &self.limits, limits)137 .map_err(dispatch_to_evm::<T>)?;138 save(self)139 }140141 #[solidity(rename_selector = "setCollectionLimit")]142 fn set_bool_limit(&mut self, caller: caller, limit: string, value: bool) -> Result<void> {143 check_is_owner(caller, self)?;144 let mut limits = self.limits.clone();145146 match limit.as_str() {147 "ownerCanTransfer" => {148 limits.owner_can_transfer = Some(value);149 }150 "ownerCanDestroy" => {151 limits.owner_can_destroy = Some(value);152 }153 "transfersEnabled" => {154 limits.transfers_enabled = Some(value);155 }156 _ => {157 return Err(Error::Revert(format!(158 "Unknown boolean limit \"{}\"",159 limit160 )))161 }162 }163 self.limits = <Pallet<T>>::clamp_limits(self.mode.clone(), &self.limits, limits)164 .map_err(dispatch_to_evm::<T>)?;165 save(self)166 }167168 fn contract_address(&self, _caller: caller) -> Result<address> {169 Ok(crate::eth::collection_id_to_address(self.id))170 }171172 // fn add_admin_substrate(&self, caller: caller, new_admin: uint256) -> Result<void> {173 // let mut new_admin_h256 = H256::default();174 // new_admin.to_little_endian(&mut new_admin_h256.0);175 // let account_id = T::AccountId::from(new_admin_h256);176 // let caller = T::CrossAccountId::from_eth(caller);177 // let new_admin = T::CrossAccountId::from_sub(account_id);178 // <Pallet<T>>::toggle_admin(&self, &caller, &new_admin, true)179 // .map_err(dispatch_to_evm::<T>)?;180 // Ok(())181 // }182183 // fn remove_admin_substrate(&self, caller: caller, new_admin: uint256) -> Result<void> {184 // let mut new_admin_h256 = H256::default();185 // new_admin.to_little_endian(&mut new_admin_h256.0);186 // let account_id = T::AccountId::from(new_admin_h256);187 // let caller = T::CrossAccountId::from_eth(caller);188 // let new_admin = T::CrossAccountId::from_sub(account_id);189 // <Pallet<T>>::toggle_admin(&self, &caller, &new_admin, false)190 // .map_err(dispatch_to_evm::<T>)?;191 // Ok(())192 // }193194 fn add_collection_admin(&self, caller: caller, new_admin: address) -> Result<void> {195 let caller = T::CrossAccountId::from_eth(caller);196 self.check_is_owner_or_admin(&caller)197 .map_err(dispatch_to_evm::<T>)?;198 let new_admin = T::CrossAccountId::from_eth(new_admin);199 <Pallet<T>>::toggle_admin(&self, &caller, &new_admin, true)200 .map_err(dispatch_to_evm::<T>)?;201 Ok(())202 }203204 fn remove_collection_admin(&self, caller: caller, admin: address) -> Result<void> {205 let caller = T::CrossAccountId::from_eth(caller);206 self.check_is_owner_or_admin(&caller)207 .map_err(dispatch_to_evm::<T>)?;208 let admin = T::CrossAccountId::from_eth(admin);209 <Pallet<T>>::toggle_admin(&self, &caller, &admin, false).map_err(dispatch_to_evm::<T>)?;210 Ok(())211 }212213 #[solidity(rename_selector = "setCollectionNesting")]214 fn set_nesting_bool(&mut self, caller: caller, enable: bool) -> Result<void> {215 let caller = T::CrossAccountId::from_eth(caller);216 self.check_is_owner_or_admin(&caller)217 .map_err(dispatch_to_evm::<T>)?;218 self.collection.permissions.nesting = Some(match enable {219 false => NestingRule::Disabled,220 true => NestingRule::Owner,221 });222 save(self)?;223 Ok(())224 }225226 #[solidity(rename_selector = "setCollectionNesting")]227 fn set_nesting(228 &mut self,229 caller: caller,230 enable: bool,231 collections: Vec<address>,232 ) -> Result<void> {233 if collections.is_empty() {234 return Err("No addresses provided".into());235 }236 if collections.len() >= OwnerRestrictedSet::bound() {237 return Err(Error::Revert(format!(238 "Out of bound: {} >= {}",239 collections.len(),240 OwnerRestrictedSet::bound()241 )));242 }243 let caller = T::CrossAccountId::from_eth(caller);244 self.check_is_owner_or_admin(&caller)245 .map_err(dispatch_to_evm::<T>)?;246 self.collection.permissions.nesting = Some(match enable {247 false => NestingRule::Disabled,248 true => {249 let mut bv = OwnerRestrictedSet::new();250 for i in collections {251 bv.try_insert(crate::eth::map_eth_to_id(&i).ok_or(Error::Revert(252 "Can't convert address into collection id".into(),253 ))?)254 .map_err(|e| Error::Revert(format!("{:?}", e)))?;255 }256 NestingRule::OwnerRestricted(bv)257 }258 });259 save(self)?;260 Ok(())261 }262263 fn set_collection_access(&mut self, caller: caller, mode: uint8) -> Result<void> {264 let caller = T::CrossAccountId::from_eth(caller);265 self.check_is_owner_or_admin(&caller)266 .map_err(dispatch_to_evm::<T>)?;267 self.collection.permissions.access = Some(match mode {268 0 => AccessMode::Normal,269 1 => AccessMode::AllowList,270 _ => return Err("Not supported access mode".into()),271 });272 save(self)?;273 Ok(())274 }275276 fn add_to_collection_allow_list(&self, caller: caller, user: address) -> Result<void> {277 let caller = check_is_owner_or_admin(caller, self)?;278 let user = T::CrossAccountId::from_eth(user);279 <Pallet<T>>::toggle_allowlist(self, &caller, &user, true).map_err(dispatch_to_evm::<T>)?;280 Ok(())281 }282283 fn remove_from_collection_allow_list(&self, caller: caller, user: address) -> Result<void> {284 let caller = check_is_owner_or_admin(caller, self)?;285 let user = T::CrossAccountId::from_eth(user);286 <Pallet<T>>::toggle_allowlist(self, &caller, &user, false).map_err(dispatch_to_evm::<T>)?;287 Ok(())288 }289290 fn set_collection_mint_mode(&mut self, caller: caller, mode: bool) -> Result<void> {291 check_is_owner_or_admin(caller, self)?;292 self.collection.permissions.mint_mode = Some(mode);293 save(self)?;294 Ok(())295 }296}297298fn check_is_owner<T: Config>(caller: caller, collection: &CollectionHandle<T>) -> Result<void> {299 let caller = T::CrossAccountId::from_eth(caller);300 collection301 .check_is_owner(&caller)302 .map_err(pallet_evm_coder_substrate::dispatch_to_evm::<T>)?;303 Ok(())304}305306fn check_is_owner_or_admin<T: Config>(307 caller: caller,308 collection: &CollectionHandle<T>,309) -> Result<T::CrossAccountId> {310 let caller = T::CrossAccountId::from_eth(caller);311 collection312 .check_is_owner_or_admin(&caller)313 .map_err(pallet_evm_coder_substrate::dispatch_to_evm::<T>)?;314 Ok(caller)315}316317fn save<T: Config>(collection: &CollectionHandle<T>) -> Result<void> {318 // TODO possibly delete for the lack of transaction319 collection320 .check_is_internal()321 .map_err(dispatch_to_evm::<T>)?;322 <crate::CollectionById<T>>::insert(collection.id, collection.collection.clone());323 Ok(())324}325326pub fn token_uri_key() -> up_data_structs::PropertyKey {327 b"tokenURI"328 .to_vec()329 .try_into()330 .expect("length < limit; qed")331}