Make ERC4626 _deposit and _withdraw internal virtual (OpenZeppelin#3504)

ronhuafeng · Sep 9, 2022 · 8ea4f56 · 8ea4f56
1 parent 6930321
commit 8ea4f56
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/contracts/token/ERC20/extensions/ERC4626.sol b/contracts/token/ERC20/extensions/ERC4626.sol
@@ -175,7 +175,7 @@ abstract contract ERC4626 is ERC20, IERC4626 {
         address receiver,
         uint256 assets,
         uint256 shares
-    ) private {
+    ) internal virtual {
         // If _asset is ERC777, `transferFrom` can trigger a reenterancy BEFORE the transfer happens through the
         // `tokensToSend` hook. On the other hand, the `tokenReceived` hook, that is triggered after the transfer,
         // calls the vault, which is assumed not malicious.
@@ -198,7 +198,7 @@ abstract contract ERC4626 is ERC20, IERC4626 {
         address owner,
         uint256 assets,
         uint256 shares
-    ) private {
+    ) internal virtual {
         if (caller != owner) {
             _spendAllowance(owner, caller, shares);
         }