feat: not eoa (#52)

* feat: not eoa * fix: comments * fix: comments
defi-wonderland · Jul 24, 2024 · fbdedb5 · fbdedb5
1 parent 07e210d
commit fbdedb5
Show file tree

Hide file tree

Showing 8 changed files with 215 additions and 131 deletions.
diff --git a/package.json b/package.json
@@ -38,7 +38,7 @@
     "package.json": "sort-package-json"
   },
   "dependencies": {
-    "@defi-wonderland/prophet-core-contracts": "0.0.0-4d4a4487",
+    "@defi-wonderland/prophet-core-contracts": "0.0.0-ad40b65b",
     "@openzeppelin/contracts": "4.9.5",
     "solmate": "https://github.com/transmissions11/solmate.git#bfc9c25865a274a7827fea5abf6e4fb64fc64e6c"
   },

diff --git a/solidity/contracts/modules/dispute/CircuitResolverModule.sol b/solidity/contracts/modules/dispute/CircuitResolverModule.sol
@@ -94,12 +94,25 @@ contract CircuitResolverModule is Module, ICircuitResolverModule {
   /// @inheritdoc IModule
   function validateParameters(bytes calldata _encodedParameters)
     external
-    pure
+    view
     override(Module, IModule)
     returns (bool _valid)
   {
     RequestParameters memory _params = decodeRequestData(_encodedParameters);
     _valid = address(_params.accountingExtension) != address(0) && address(_params.bondToken) != address(0)
-      && _params.bondSize != 0 && address(_params.verifier) != address(0) && _params.callData.length != 0;
+      && _params.bondSize != 0 && _targetHasBytecode(_params.verifier) && _params.callData.length != 0;
+  }
+
+  /**
+   * @notice Checks if a target address has bytecode
+   * @param _target The address to check
+   * @return _hasBytecode Whether the target has bytecode or not
+   */
+  function _targetHasBytecode(address _target) private view returns (bool _hasBytecode) {
+    uint256 _size;
+    assembly {
+      _size := extcodesize(_target)
+    }
+    _hasBytecode = _size > 0;
   }
 }
diff --git a/solidity/contracts/modules/finality/CallbackModule.sol b/solidity/contracts/modules/finality/CallbackModule.sol
@@ -27,6 +27,7 @@ contract CallbackModule is Module, ICallbackModule {
     address _finalizer
   ) external override(Module, ICallbackModule) onlyOracle {
     RequestParameters memory _params = decodeRequestData(_request.finalityModuleData);
+
     _params.target.call(_params.data);
     emit Callback(_response.requestId, _params.target, _params.data);
     emit RequestFinalized(_response.requestId, _response, _finalizer);
@@ -35,11 +36,24 @@ contract CallbackModule is Module, ICallbackModule {
   /// @inheritdoc IModule
   function validateParameters(bytes calldata _encodedParameters)
     external
-    pure
+    view
     override(Module, IModule)
     returns (bool _valid)
   {
     RequestParameters memory _params = decodeRequestData(_encodedParameters);
-    _valid = address(_params.target) != address(0) && _params.data.length != 0;
+    _valid = _params.data.length != 0 && _targetHasBytecode(_params.target);
+  }
+
+  /**
+   * @notice Checks if a target address has bytecode
+   * @param _target The address to check
+   * @return _hasBytecode Whether the target has bytecode or not
+   */
+  function _targetHasBytecode(address _target) private view returns (bool _hasBytecode) {
+    uint256 _size;
+    assembly {
+      _size := extcodesize(_target)
+    }
+    _hasBytecode = _size > 0;
   }
 }
diff --git a/solidity/contracts/modules/finality/MultipleCallbacksModule.sol b/solidity/contracts/modules/finality/MultipleCallbacksModule.sol
@@ -43,15 +43,15 @@ contract MultipleCallbacksModule is Module, IMultipleCallbacksModule {
   /// @inheritdoc IModule
   function validateParameters(bytes calldata _encodedParameters)
     external
-    pure
+    view
     override(Module, IModule)
     returns (bool _valid)
   {
     RequestParameters memory _params = decodeRequestData(_encodedParameters);
     _valid = true;
 
     for (uint256 _i; _i < _params.targets.length; ++_i) {
-      if (_params.targets[_i] == address(0)) {
+      if (!_targetHasBytecode(_params.targets[_i])) {
         _valid = false;
         break;
       }
@@ -64,4 +64,17 @@ contract MultipleCallbacksModule is Module, IMultipleCallbacksModule {
       }
     }
   }
+
+  /**
+   * @notice Checks if a target address has bytecode
+   * @param _target The address to check
+   * @return _hasBytecode Whether the target has bytecode or not
+   */
+  function _targetHasBytecode(address _target) private view returns (bool _hasBytecode) {
+    uint256 _size;
+    assembly {
+      _size := extcodesize(_target)
+    }
+    _hasBytecode = _size > 0;
+  }
 }
diff --git a/solidity/test/unit/modules/dispute/CircuitResolverModule.t.sol b/solidity/test/unit/modules/dispute/CircuitResolverModule.t.sol
@@ -60,6 +60,14 @@ contract BaseTest is Test, Helpers {
 
     circuitResolverModule = new ForTest_CircuitResolverModule(oracle);
   }
+
+  function targetHasBytecode(address _target) public view returns (bool _hasBytecode) {
+    uint256 _size;
+    assembly {
+      _size := extcodesize(_target)
+    }
+    _hasBytecode = _size > 0;
+  }
 }
 
 contract CircuitResolverModule_Unit_ModuleData is BaseTest {
@@ -109,7 +117,7 @@ contract CircuitResolverModule_Unit_ModuleData is BaseTest {
   function test_validateParameters(ICircuitResolverModule.RequestParameters calldata _params) public {
     if (
       address(_params.accountingExtension) == address(0) || address(_params.bondToken) == address(0)
-        || _params.bondSize == 0 || address(_params.verifier) == address(0) || _params.callData.length == 0
+        || _params.bondSize == 0 || !targetHasBytecode(_params.verifier) || _params.callData.length == 0
     ) {
       assertFalse(circuitResolverModule.validateParameters(abi.encode(_params)));
     } else {

diff --git a/solidity/test/unit/modules/finality/CallbackModule.t.sol b/solidity/test/unit/modules/finality/CallbackModule.t.sol
@@ -31,6 +31,14 @@ contract BaseTest is Test, Helpers {
 
     callbackModule = new CallbackModule(oracle);
   }
+
+  function targetHasBytecode(address _target) public view returns (bool _hasBytecode) {
+    uint256 _size;
+    assembly {
+      _size := extcodesize(_target)
+    }
+    _hasBytecode = _size > 0;
+  }
 }
 
 contract CallbackModule_Unit_ModuleData is BaseTest {
@@ -60,7 +68,7 @@ contract CallbackModule_Unit_ModuleData is BaseTest {
    * @notice Test that the validateParameters function correctly checks the parameters
    */
   function test_validateParameters(ICallbackModule.RequestParameters calldata _params) public {
-    if (address(_params.target) == address(0) || _params.data.length == 0) {
+    if (address(_params.target) == address(0) || _params.data.length == 0 || !targetHasBytecode(_params.target)) {
       assertFalse(callbackModule.validateParameters(abi.encode(_params)));
     } else {
       assertTrue(callbackModule.validateParameters(abi.encode(_params)));
@@ -73,6 +81,7 @@ contract CallbackModule_Unit_FinalizeRequest is BaseTest {
    * @notice Test that finalizeRequest emits events
    */
   function test_emitsEvents(address _proposer, address _target, bytes calldata _data) public assumeFuzzable(_target) {
+    vm.etch(_target, '0xabcdef');
     mockRequest.finalityModuleData = abi.encode(ICallbackModule.RequestParameters({target: _target, data: _data}));
     mockResponse.requestId = _getId(mockRequest);
 

diff --git a/solidity/test/unit/modules/finality/MultipleCallbacksModule.t.sol b/solidity/test/unit/modules/finality/MultipleCallbacksModule.t.sol
@@ -31,6 +31,14 @@ contract BaseTest is Test, Helpers {
 
     multipleCallbackModule = new MultipleCallbacksModule(oracle);
   }
+
+  function targetHasBytecode(address _target) public view returns (bool _hasBytecode) {
+    uint256 _size;
+    assembly {
+      _size := extcodesize(_target)
+    }
+    _hasBytecode = _size > 0;
+  }
 }
 
 /**
@@ -50,7 +58,7 @@ contract MultipleCallbacksModule_Unit_ModuleData is BaseTest {
   function test_validateParameters(IMultipleCallbacksModule.RequestParameters calldata _params) public {
     bool _valid = true;
     for (uint256 _i; _i < _params.targets.length; ++_i) {
-      if (_params.targets[_i] == address(0)) {
+      if (_params.targets[_i] == address(0) || !targetHasBytecode(_params.targets[_i])) {
         _valid = false;
       }
     }