Skip to content

Commit f9d68d9

Browse files
committed
Checking user existence is done by accounts set
1 parent 2f8d497 commit f9d68d9

File tree

1 file changed

+7
-4
lines changed

1 file changed

+7
-4
lines changed

contracts/registry/Registry.sol

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -118,7 +118,10 @@ contract Registry is Context, AdminRole {
118118
) external onlyAdmin {
119119
require(_adrs.length == _cnt, "Invalid number of addresses");
120120
require(_trusts.length == _cnt, "Invalid number of trust values");
121-
require(_pendingBalances.length == _cnt, "Invalid number of pending balance values");
121+
require(
122+
_pendingBalances.length == _cnt,
123+
"Invalid number of pending balance values"
124+
);
122125

123126
for (uint256 i = 0; i < _cnt; i++) {
124127
_register(_adrs[i], _trusts[i], _pendingBalances[i]);
@@ -285,7 +288,7 @@ contract Registry is Context, AdminRole {
285288

286289
function _remove(address _adr) internal {
287290
require(_adr != address(0), "Cannot remove zero address");
288-
require(maxTrusts[_adr] != 0, "Address is not a contributor");
291+
require(EnumerableSet.contains(accounts, _adr), "Address is not a contributor");
289292

290293
EnumerableSet.remove(accounts, _adr);
291294
delete maxTrusts[_adr];
@@ -301,7 +304,7 @@ contract Registry is Context, AdminRole {
301304
_adr != address(0),
302305
"Cannot set pending balance for zero balance"
303306
);
304-
require(maxTrusts[_adr] != 0, "Address is not a contributor");
307+
require(EnumerableSet.contains(accounts, _adr), "Address is not a contributor");
305308
require(
306309
cstkToken.balanceOf(_adr) == 0,
307310
"User has activated his membership"
@@ -317,7 +320,7 @@ contract Registry is Context, AdminRole {
317320
_adr != address(0),
318321
"Cannot set pending balance for zero balance"
319322
);
320-
require(maxTrusts[_adr] != 0, "Address is not a contributor");
323+
require(EnumerableSet.contains(accounts, _adr), "Address is not a contributor");
321324
require(
322325
cstkToken.balanceOf(_adr) == 0,
323326
"User has activated his membership"

0 commit comments

Comments
 (0)