try delegate call

Author: open-junius

Cargo.lock

@@ -35,6 +35,7 @@ pallet-drand.workspace = true
 subtensor-swap-interface.workspace = true
 num_enum.workspace = true
 substrate-fixed.workspace = true
+sp-debug-derive = { workspace = true, features = ["force-debug"] }
 
 [lints]
 workspace = true

chain-extensions/Cargo.toml

@@ -739,6 +739,15 @@ where
 
                 Ok(RetVal::Converging(Output::Success as u32))
             }
+            FunctionId::DummyV1 => {
+                let weight = Weight::from_parts(1_000, 0);
+                let caller = env.caller();
+                let origin = env.origin();
+                log::error!("dummy: contract address as caller = {:?}", caller);
+                log::error!("dummy: original account as origin = {:?}", origin);
+                env.charge_weight(weight)?;
+                Ok(RetVal::Converging(Output::Success as u32))
+            }
         }
     }
 }
@@ -817,6 +826,9 @@ where
     }
 
     fn origin(&mut self) -> pallet_contracts::Origin<T> {
+        if let pallet_contracts::Origin::Signed(account_id) = self.env.ext().caller() {
+            log::error!("origin: signed account id = {:?}", account_id);
+        }
         self.env.ext().caller()
     }
 }

chain-extensions/src/lib.rs

@@ -35,6 +35,8 @@ pub enum FunctionId {
     CallerSetColdkeyAutoStakeHotkeyV1 = 27,
     CallerAddProxyV1 = 28,
     CallerRemoveProxyV1 = 29,
+    /// No-op extension for ink integration tests (`direct_call` / `delegate_call` contracts).
+    DummyV1 = 100,
 }
 
 #[derive(PartialEq, Eq, Copy, Clone, Encode, Decode, Debug)]
@@ -134,6 +136,7 @@ mod function_id_tests {
         assert_eq!(FunctionId::CallerSetColdkeyAutoStakeHotkeyV1 as u16, 27);
         assert_eq!(FunctionId::CallerAddProxyV1 as u16, 28);
         assert_eq!(FunctionId::CallerRemoveProxyV1 as u16, 29);
+        assert_eq!(FunctionId::DummyV1 as u16, 100);
     }
 
     #[test]

chain-extensions/src/types.rs

@@ -0,0 +1,9 @@
+# Ignore build artifacts from the local tests sub-crate.
+/target/
+
+# Ignore backup files creates by cargo fmt.
+**/*.rs.bk
+
+# Remove Cargo.lock when creating an executable, leave it for libraries
+# More information here http://doc.crates.io/guide.html#cargotoml-vs-cargolock
+Cargo.lock

contract-tests/delegate-call/.gitignore

@@ -0,0 +1,28 @@
+[workspace]
+
+[package]
+name = "delegate_call"
+version = "0.1.0"
+authors = ["[your_name] <[your_email]>"]
+edition = "2021"
+
+[dependencies]
+ink = { version = "5.1.1", default-features = false }
+parity-scale-codec = { version = "3.0.0", default-features = false }
+serde = { version = "1.0.228", default-features = false }
+[dev-dependencies]
+ink_e2e = { version = "5.1.1" }
+
+[lib]
+path = "lib.rs"
+
+[features]
+default = ["std"]
+std = [
+    "ink/std",
+    "parity-scale-codec/std",
+    "serde/std",
+]
+
+ink-as-dependency = []
+e2e-tests = []

contract-tests/delegate-call/Cargo.toml

@@ -0,0 +1,83 @@
+#![cfg_attr(not(feature = "std"), no_std, no_main)]
+
+use ink::env::call::{ExecutionInput, Selector, build_call};
+#[derive(Debug, Clone)]
+pub struct CustomEnvironment;
+
+pub enum FunctionId {
+    Dummy = 100,
+}
+
+#[ink::chain_extension(extension = 0x1000)]
+pub trait RuntimeReadWrite {
+    type ErrorCode = ReadWriteErrorCode;
+
+    #[ink(function = 100)]
+    fn dummy();
+}
+
+#[ink::scale_derive(Encode, Decode, TypeInfo)]
+pub enum ReadWriteErrorCode {
+    ReadFailed,
+    WriteFailed,
+}
+
+impl ink::env::chain_extension::FromStatusCode for ReadWriteErrorCode {
+    fn from_status_code(status_code: u32) -> Result<(), Self> {
+        match status_code {
+            0 => Ok(()),
+            1 => Err(ReadWriteErrorCode::ReadFailed),
+            2 => Err(ReadWriteErrorCode::WriteFailed),
+            _ => Err(ReadWriteErrorCode::ReadFailed),
+        }
+    }
+}
+
+impl ink::env::Environment for CustomEnvironment {
+    const MAX_EVENT_TOPICS: usize = 4;
+
+    type AccountId = ink::primitives::AccountId;
+    type Balance = u64;
+    type Hash = ink::primitives::Hash;
+    type BlockNumber = u32;
+    type Timestamp = u64;
+
+    type ChainExtension = RuntimeReadWrite;
+}
+
+#[ink::contract(env = crate::CustomEnvironment)]
+mod delegate_call {
+    use super::*;
+
+    #[ink(storage)]
+    pub struct Bittensor {}
+
+    impl Bittensor {
+        #[ink(constructor)]
+        pub fn new() -> Self {
+            Self {}
+        }
+
+        /// Constructor
+        #[ink(constructor)]
+        pub fn default() -> Self {
+            Self::new()
+        }
+
+        #[ink(message)]
+        pub fn dummy(&self, code_hash: [u8; 32]) -> Result<(), ReadWriteErrorCode> {
+            // let code_hash = self
+            //     .delegate_code_hash
+            //     .unwrap_or(ink::primitives::Hash::from(hash));
+            let my_return_value = build_call::<CustomEnvironment>()
+                .delegate(ink::primitives::Hash::from(code_hash))
+                .exec_input(ExecutionInput::new(Selector::new(ink::selector_bytes!(
+                    "dummy"
+                ))))
+                .returns::<Result<(), ReadWriteErrorCode>>()
+                .invoke();
+
+            my_return_value.map_err(|_e| ReadWriteErrorCode::WriteFailed)
+        }
+    }
+}

contract-tests/delegate-call/lib.rs

@@ -0,0 +1,9 @@
+# Ignore build artifacts from the local tests sub-crate.
+/target/
+
+# Ignore backup files creates by cargo fmt.
+**/*.rs.bk
+
+# Remove Cargo.lock when creating an executable, leave it for libraries
+# More information here http://doc.crates.io/guide.html#cargotoml-vs-cargolock
+Cargo.lock

contract-tests/direct-call/.gitignore

@@ -0,0 +1,28 @@
+[workspace]
+
+[package]
+name = "direct_call"
+version = "0.1.0"
+authors = ["[your_name] <[your_email]>"]
+edition = "2021"
+
+[dependencies]
+ink = { version = "5.1.1", default-features = false }
+parity-scale-codec = { version = "3.0.0", default-features = false }
+serde = { version = "1.0.228", default-features = false }
+[dev-dependencies]
+ink_e2e = { version = "5.1.1" }
+
+[lib]
+path = "lib.rs"
+
+[features]
+default = ["std"]
+std = [
+    "ink/std",
+    "parity-scale-codec/std",
+    "serde/std",
+]
+
+ink-as-dependency = []
+e2e-tests = []

contract-tests/direct-call/Cargo.toml

@@ -0,0 +1,80 @@
+#![cfg_attr(not(feature = "std"), no_std, no_main)]
+
+use parity_scale_codec::Compact;
+
+#[derive(Debug, Clone)]
+pub struct CustomEnvironment;
+
+pub enum FunctionId {
+    Dummy = 100,
+}
+
+#[ink::chain_extension(extension = 0x1000)]
+pub trait RuntimeReadWrite {
+    type ErrorCode = ReadWriteErrorCode;
+
+    #[ink(function = 100)]
+    fn dummy();
+}
+
+#[ink::scale_derive(Encode, Decode, TypeInfo)]
+pub enum ReadWriteErrorCode {
+    ReadFailed,
+    WriteFailed,
+}
+
+impl ink::env::chain_extension::FromStatusCode for ReadWriteErrorCode {
+    fn from_status_code(status_code: u32) -> Result<(), Self> {
+        match status_code {
+            0 => Ok(()),
+            1 => Err(ReadWriteErrorCode::ReadFailed),
+            2 => Err(ReadWriteErrorCode::WriteFailed),
+            _ => Err(ReadWriteErrorCode::ReadFailed),
+        }
+    }
+}
+
+impl ink::env::Environment for CustomEnvironment {
+    const MAX_EVENT_TOPICS: usize = 4;
+
+    type AccountId = ink::primitives::AccountId;
+    type Balance = u64;
+    type Hash = ink::primitives::Hash;
+    type BlockNumber = u32;
+    type Timestamp = u64;
+
+    type ChainExtension = RuntimeReadWrite;
+}
+
+#[ink::contract(env = crate::CustomEnvironment)]
+mod direct_call {
+    use super::*;
+
+    /// Defines the storage of your contract.
+    /// Add new fields to the below struct in order
+    /// to add new static storage fields to your contract.
+    #[ink(storage)]
+    pub struct Bittensor {}
+
+    impl Bittensor {
+        /// Constructor
+        #[ink(constructor)]
+        pub fn new() -> Self {
+            Self {}
+        }
+
+        /// Constructor
+        #[ink(constructor)]
+        pub fn default() -> Self {
+            Self::new()
+        }
+
+        #[ink(message)]
+        pub fn dummy(&self) -> Result<(), ReadWriteErrorCode> {
+            self.env()
+                .extension()
+                .dummy()
+                .map_err(|_e| ReadWriteErrorCode::WriteFailed)
+        }
+    }
+}