Browse Source

Use libraries syntax sugar for revoking grants logic

Jorge Izquierdo 8 years ago
parent
commit
22f60eace0
2 changed files with 5 additions and 6 deletions
  1. 2 2
      contracts/token/BasicToken.sol
  2. 3 4
      contracts/token/VestedToken.sol

+ 2 - 2
contracts/token/BasicToken.sol

@@ -15,7 +15,7 @@ contract BasicToken is ERC20Basic {
   mapping(address => uint) balances;
 
   /*
-   * Fix for the ERC20 short address attack  
+   * Fix for the ERC20 short address attack
    */
   modifier onlyPayloadSize(uint size) {
      if(msg.data.length < size + 4) {
@@ -33,5 +33,5 @@ contract BasicToken is ERC20Basic {
   function balanceOf(address _owner) constant returns (uint balance) {
     return balances[_owner];
   }
-  
+
 }

+ 3 - 4
contracts/token/VestedToken.sol

@@ -4,7 +4,6 @@ import "./StandardToken.sol";
 import "./LimitedTransferToken.sol";
 
 contract VestedToken is StandardToken, LimitedTransferToken {
-
   struct TokenGrant {
     address granter;     // 20 bytes
     uint256 value;       // 32 bytes
@@ -68,11 +67,11 @@ contract VestedToken is StandardToken, LimitedTransferToken {
 
     // remove grant from array
     delete grants[_holder][_grantId];
-    grants[_holder][_grantId] = grants[_holder][grants[_holder].length - 1];
+    grants[_holder][_grantId] = grants[_holder][grants[_holder].length.sub(1)];
     grants[_holder].length -= 1;
 
-    balances[receiver] = SafeMath.add(balances[receiver], nonVested);
-    balances[_holder] = SafeMath.sub(balances[_holder], nonVested);
+    balances[receiver] = balances[receiver].add(nonVested);
+    balances[_holder] = balances[_holder].sub(nonVested);
     Transfer(_holder, receiver, nonVested);
   }