Browse Source

[TokenVesting] Add missing safemath ops

Martín Triay 8 years ago
parent
commit
b60e434e0b
1 changed files with 2 additions and 2 deletions
  1. 2 2
      contracts/token/TokenVesting.sol

+ 2 - 2
contracts/token/TokenVesting.sol

@@ -106,10 +106,10 @@ contract TokenVesting is Ownable {
 
     if (now < cliff) {
       return 0;
-    } else if (now >= start + duration || revoked[token]) {
+    } else if (now >= start.add(duration) || revoked[token]) {
       return totalBalance;
     } else {
-      return totalBalance.mul(now - start).div(duration);
+      return totalBalance.mul(now.sub(start)).div(duration);
     }
   }
 }