如何通过提取智能合约启用正确的用户识别？

Question

有了这份智能合约，用户可以存入以太并撤回以太。合同应存储每个用户和存款，并随时撤回该金额。在代码中没有错误，但是不幸的是，取消提取的功能无法正常工作，实际上，以太词保留在合同中，也不会退还给所有者。

   contract vaultCoin {
        uint amount; 
        address user;
        uint amountDeposit;
    
        function vaultCoin() public {
            dev = msg.sender;                    
        }
    
        address public dev;                          
        address[] public users;                   
        uint[] public totalDeposited;                
    
        mapping(address => uint) balances;
        
        function sendToken() public payable {
            balances[msg.sender] = amount;
            require(msg.value > 0.001 ether); 
            
            user = msg.sender;               
            amountDeposit = msg.value;      
            
            users.push(user);                     
            totalDeposited.push(amountDeposit);     
        }
    
        function getUsers() public view returns (address[]) {    
            return users;
        }
    
        function getAmount() public view returns (uint[]) {
            return totalDeposited;
        }
    
        function retireMyCoins() public payable {
            require(msg.sender == user);
            require(msg.value == amountDeposit);
            balances[msg.sender] = 0; 
            msg.sender.transfer(balances[msg.sender]); 
        }
    }

Answer 1

balances[msg.sender] = 0;

您的代码首先将余额设置为0

msg.sender.transfer(balances[msg.sender]);

，然后将已经零的值传输给用户。

---

解决方案：

1. 记录余额在存储变量中
2. 记录存储值
3. 从存储器变量转移

uint256 balance = balances[msg.sender];
balances[msg.sender] = 0; 
msg.sender.transfer(balance); 

值此中间步骤重新输入攻击。