Create ProxyContract.sol

46_ProxyContract/ProxyContract.sol

@@ -0,0 +1,96 @@
+// SPDX-License-Identifier: MIT
+// wtf.academy
+pragma solidity ^0.8.4;
+
+/**
+ * @dev Proxy合约的所有调用都通过`delegatecall`操作码委托给另一个合约执行。后者被称为逻辑合约（Implementation）。
+ *
+ * 委托调用的返回值，会直接返回给Proxy的调用者
+ */
+contract Proxy {
+    address immutable implementation; // 逻辑合约地址。implementation合约同一个位置的状态变量类型必须和Proxy合约的相同，不然会报错。
+
+    /**
+     * @dev 初始化逻辑合约地址
+     */
+    constructor(address implementation_){
+        implementation = implementation_;
+    }
+
+    /**
+     * @dev 回调函数，调用`_delegate()`函数将本合约的调用委托给 `implementation` 合约
+     */
+    fallback() external payable {
+        _delegate();
+    }
+
+    /**
+     * @dev 将调用委托给逻辑合约运行
+     */
+    function _delegate() internal {
+        assembly {
+            // Copy msg.data. We take full control of memory in this inline assembly
+            // block because it will not return to Solidity code. We overwrite the
+            // 读取位置为0的storage，也就是implementation地址。
+            let _implementation := sload(0)
+
+            // 利用delegatecall调用implementation合约
+            // delegatecall操作码的参数分别为：gas, 目标合约地址，input mem起始位置，input mem长度，output area mem起始位置，output area mem长度
+            // output area起始位置和长度位置，所以设为0
+            // delegatecall成功返回1，失败返回0
+            let result := delegatecall(gas(), _implementation, 0, calldatasize(), 0, 0)
+
+            // 将起始位置为0，长度为returndatasize()的returndata复制到mem位置0
+            returndatacopy(0, 0, returndatasize())
+
+            switch result
+            // 如果delegate call失败，revert
+            case 0 {
+                revert(0, returndatasize())
+            }
+            // 如果delegate call成功，返回mem起始位置为0，长度为returndatasize()的数据（格式为bytes）
+            default {
+                return(0, returndatasize())
+            }
+        }
+    }
+}
+
+/**
+ * @dev 逻辑合约，执行被委托的调用
+ */
+contract Logic {
+    address public implementation; // 与Proxy保持一致，防止插槽冲突
+    uint public x = 99; 
+    event LogicCalled();
+    event FallbackCalled();
+
+    // 回调函数，释放FallbackCalled事件
+    fallback() external payable {
+        emit FallbackCalled();
+    }
+
+    // 这个函数会释放LogicCalled并返回一个uint。
+    // 函数selector: 0xd09de08a
+    function increment() external returns(uint) {
+        emit LogicCalled();
+        return x + 1;
+    }
+}
+
+/**
+ * @dev Caller合约，调用代理合约，并获取执行结果
+ */
+contract Caller{
+    address public proxy; // 代理合约地址
+
+    constructor(address proxy_){
+        proxy = proxy_;
+    }
+
+    // 通过代理合约调用 increase()函数
+    function increase() external returns(bytes memory) {
+        ( , bytes memory data) = proxy.call("0xd09de08a");
+        return data;
+    }
+}