安全智能合约工具：端到端开发人员指南

// SPDX-License-Identifier: GPL-3.0
 pragma solidity >=0.8.2 < 0.9.0;
/// @title A simulator for trees
 /// @author Larry A. Gardner
 /// @notice You can use this contract for only the most basic simulation
 contract Tree {
     /// @notice Calculate tree age in years, rounded up, for live trees
     /// @dev The Alexandr N. Tetearing algorithm could increase precision
    /// @param rings The number of rings from dendrochronological sample
    /// @return Age in years, rounded up for partial years
    function age(uint256 rings) external virtual pure returns (uint256) {
        return rings + 1;
    }
 }

// SPDX-License-Identifier: Unlicense
 pragma solidity 0.8.10;
 import "ds-test/test.sol";
 import "../StakeContract.sol";
 import "./mocks/MockERC20.sol";

 contract StakeContractTest is DSTest {
    StakeContract public stakeContract;
    MockERC20 public mockToken;
    function setUp() public {
        stakeContract = new StakeContract();
        mockToken = new MockERC20();
    }
   /// @notice Test token staking with different amount
    function test_staking_tokens() public {
        uint256 amount = 10e18;
        mockToken.approve(address(stakeContract), amount);
        bool stakePassed = stakeContract.stake(amount, address(mockToken));
        assertTrue(stakePassed);
    }
}

// SPDX-License-Identifier: MIT
pragma solidity ^0.8.10;
import "@openzeppelin/contracts/token/ERC20/IERC20.sol";
error TransferFailed();
contract StakeContract {
   mapping(address => uint256) public s_balances;
   /// #if_succeeds {:msg "Stake created successfully"} $result == true;
   function stake(uint256 amount, address token) external returns (bool){
       s_balances[msg.sender] += amount;
       bool success = IERC20(token).transferFrom(msg.sender, address(this), amount);
       if (!success) revert TransferFailed();
       return success;
   }
}