added ownership

CanzaApps · Apr 21, 2024 · 3f876c2 · 3f876c2
1 parent f5e85cb
commit 3f876c2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/contracts/Vault.sol b/contracts/Vault.sol
@@ -171,7 +171,7 @@ contract Vault is
 
     event SetOracleAddress(address _address);
 
-    function resetGlobalState(uint256 _collateral, uint256 _globalMint, uint256 _swapVolume, uint256 _fees) external {
+    function resetGlobalState(uint256 _collateral, uint256 _globalMint, uint256 _swapVolume, uint256 _fees) external onlyOwner {
         require( globalResetState == false, "State already reset" );
 
         totalCollateral = _collateral;
@@ -182,7 +182,7 @@ contract Vault is
         globalResetState = true;
     }
 
-    function resetUserState(address _user, uint256 _collateral, uint256 _netMint, uint256 _zusd, uint256 _zngn, uint256 _zxaf, uint256 _zzar) external {
+    function resetUserState(address _user, uint256 _collateral, uint256 _netMint, uint256 _zusd, uint256 _zngn, uint256 _zxaf, uint256 _zzar) external onlyOwner {
         require( userResetState[_user] == false, "User state already reset" );
 
         uint256 amountInUSDC = _collateral / USDC_DIVISOR;