feat: overwrite _deposit and _withdraw

SafeYields · Dec 8, 2023 · d193902 · d193902
1 parent 59f7301
commit d193902
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/contracts/SafeVault.sol b/contracts/SafeVault.sol
@@ -65,4 +65,24 @@ contract SafeVault is ISafeVault, ERC4626, Ownable, Pausable {
     function unpause() external onlyOwner {
         _unpause();
     }
+
+    /**
+     * @dev See {ERC4626-_deposit}.
+     */
+    function _deposit(address caller, address receiver, uint256 assets, uint256 shares) internal override {
+        super._deposit(caller, receiver, assets, shares);
+    }
+
+    /**
+     * @dev See {ERC4626-_withdraw}.
+     */
+    function _withdraw(
+        address caller,
+        address receiver,
+        address owner,
+        uint256 assets,
+        uint256 shares
+    ) internal override {
+        super._withdraw(caller, receiver, owner, assets, shares);
+    }
 }