Minor audit fixes (#245)

* Minor fixes * Replace `Ownable` with `Ownable2StepUpgradeable`
Near-One · May 30, 2024 · 7dfa6bf · 7dfa6bf
1 parent 34ab34f
commit 7dfa6bf
Show file tree

Hide file tree

Showing 4 changed files with 3 additions and 7 deletions.
diff --git a/erc20-bridge-token/contracts/BridgeToken.sol b/erc20-bridge-token/contracts/BridgeToken.sol
@@ -2,15 +2,15 @@
 pragma solidity ^0.8.24;
 
 import {ERC20Upgradeable} from "@openzeppelin/contracts-upgradeable/token/ERC20/ERC20Upgradeable.sol";
-import {OwnableUpgradeable} from "@openzeppelin/contracts-upgradeable/access/OwnableUpgradeable.sol";
+import {Ownable2StepUpgradeable} from "@openzeppelin/contracts-upgradeable/access/Ownable2StepUpgradeable.sol";
 import {Initializable} from "@openzeppelin/contracts-upgradeable/proxy/utils/Initializable.sol";
 import {UUPSUpgradeable} from "@openzeppelin/contracts-upgradeable/proxy/utils/UUPSUpgradeable.sol";
 
 contract BridgeToken is
     Initializable,
     UUPSUpgradeable,
     ERC20Upgradeable,
-    OwnableUpgradeable
+    Ownable2StepUpgradeable
 {
     string private _name;
     string private _symbol;

diff --git a/erc20-bridge-token/contracts/BridgeTokenFactory.sol b/erc20-bridge-token/contracts/BridgeTokenFactory.sol
@@ -184,7 +184,7 @@ contract BridgeTokenFactory is
 
     function withdraw(
         string memory token,
-        uint256 amount,
+        uint128 amount,
         string memory recipient
     ) external whenNotPaused(PAUSED_WITHDRAW) {
         _checkWhitelistedToken(token, msg.sender);

diff --git a/erc20-bridge-token/contracts/ProofConsumer.sol b/erc20-bridge-token/contracts/ProofConsumer.sol
@@ -1,8 +1,6 @@
 // SPDX-License-Identifier: GPL-3.0-or-later
 pragma solidity ^0.8.24;
 
-import "@openzeppelin/contracts/access/Ownable.sol";
-
 import "rainbow-bridge-sol/nearprover/contracts/INearProver.sol";
 import "rainbow-bridge-sol/nearprover/contracts/ProofDecoder.sol";
 import "rainbow-bridge-sol/nearbridge/contracts/Borsh.sol";

diff --git a/erc20-bridge-token/contracts/SelectivePausableUpgradable.sol b/erc20-bridge-token/contracts/SelectivePausableUpgradable.sol
@@ -38,8 +38,6 @@ abstract contract SelectivePausableUpgradable is Initializable, ContextUpgradeab
      */
     event Paused(address account, uint flags);
 
-    uint private _pausedFlags;
-
     /**
      * @dev Initializes the contract in unpaused state.
      */