Bladeren bron

Fix ERC20Snapshot#_beforeTokenTransfer (#2328)

sunhurts 5 jaren geleden
bovenliggende
commit
6d5a73815d
1 gewijzigde bestanden met toevoegingen van 6 en 2 verwijderingen
  1. 6 2
      contracts/token/ERC20/ERC20Snapshot.sol

+ 6 - 2
contracts/token/ERC20/ERC20Snapshot.sol

@@ -110,8 +110,12 @@ abstract contract ERC20Snapshot is ERC20 {
     function _beforeTokenTransfer(address from, address to, uint256 amount) internal virtual override {
       super._beforeTokenTransfer(from, to, amount);
 
-      if (from == address(0) || to == address(0)) {
-        // mint or burn
+      if (from == address(0)) {
+        // mint
+        _updateAccountSnapshot(to);
+        _updateTotalSupplySnapshot();
+      } else if (to == address(0)) {
+        // burn
         _updateAccountSnapshot(from);
         _updateTotalSupplySnapshot();
       } else {