文件 1 的 10:Address.sol
pragma solidity ^0.8.1;
library Address {
function isContract(address account) internal view returns (bool) {
return account.code.length > 0;
}
function sendValue(address payable recipient, uint256 amount) internal {
require(address(this).balance >= amount, "Address: insufficient balance");
(bool success, ) = recipient.call{value: amount}("");
require(success, "Address: unable to send value, recipient may have reverted");
}
function functionCall(address target, bytes memory data) internal returns (bytes memory) {
return functionCall(target, data, "Address: low-level call failed");
}
function functionCall(
address target,
bytes memory data,
string memory errorMessage
) internal returns (bytes memory) {
return functionCallWithValue(target, data, 0, errorMessage);
}
function functionCallWithValue(
address target,
bytes memory data,
uint256 value
) internal returns (bytes memory) {
return functionCallWithValue(target, data, value, "Address: low-level call with value failed");
}
function functionCallWithValue(
address target,
bytes memory data,
uint256 value,
string memory errorMessage
) internal returns (bytes memory) {
require(address(this).balance >= value, "Address: insufficient balance for call");
require(isContract(target), "Address: call to non-contract");
(bool success, bytes memory returndata) = target.call{value: value}(data);
return verifyCallResult(success, returndata, errorMessage);
}
function functionStaticCall(address target, bytes memory data) internal view returns (bytes memory) {
return functionStaticCall(target, data, "Address: low-level static call failed");
}
function functionStaticCall(
address target,
bytes memory data,
string memory errorMessage
) internal view returns (bytes memory) {
require(isContract(target), "Address: static call to non-contract");
(bool success, bytes memory returndata) = target.staticcall(data);
return verifyCallResult(success, returndata, errorMessage);
}
function functionDelegateCall(address target, bytes memory data) internal returns (bytes memory) {
return functionDelegateCall(target, data, "Address: low-level delegate call failed");
}
function functionDelegateCall(
address target,
bytes memory data,
string memory errorMessage
) internal returns (bytes memory) {
require(isContract(target), "Address: delegate call to non-contract");
(bool success, bytes memory returndata) = target.delegatecall(data);
return verifyCallResult(success, returndata, errorMessage);
}
function verifyCallResult(
bool success,
bytes memory returndata,
string memory errorMessage
) internal pure returns (bytes memory) {
if (success) {
return returndata;
} else {
if (returndata.length > 0) {
assembly {
let returndata_size := mload(returndata)
revert(add(32, returndata), returndata_size)
}
} else {
revert(errorMessage);
}
}
}
}
文件 2 的 10:BytesLib.sol
pragma solidity >=0.8.0 <0.9.0;
library BytesLib {
function concat(
bytes memory _preBytes,
bytes memory _postBytes
)
internal
pure
returns (bytes memory)
{
bytes memory tempBytes;
assembly {
tempBytes := mload(0x40)
let length := mload(_preBytes)
mstore(tempBytes, length)
let mc := add(tempBytes, 0x20)
let end := add(mc, length)
for {
let cc := add(_preBytes, 0x20)
} lt(mc, end) {
mc := add(mc, 0x20)
cc := add(cc, 0x20)
} {
mstore(mc, mload(cc))
}
length := mload(_postBytes)
mstore(tempBytes, add(length, mload(tempBytes)))
mc := end
end := add(mc, length)
for {
let cc := add(_postBytes, 0x20)
} lt(mc, end) {
mc := add(mc, 0x20)
cc := add(cc, 0x20)
} {
mstore(mc, mload(cc))
}
mstore(0x40, and(
add(add(end, iszero(add(length, mload(_preBytes)))), 31),
not(31)
))
}
return tempBytes;
}
function concatStorage(bytes storage _preBytes, bytes memory _postBytes) internal {
assembly {
let fslot := sload(_preBytes.slot)
let slength := div(and(fslot, sub(mul(0x100, iszero(and(fslot, 1))), 1)), 2)
let mlength := mload(_postBytes)
let newlength := add(slength, mlength)
switch add(lt(slength, 32), lt(newlength, 32))
case 2 {
sstore(
_preBytes.slot,
add(
fslot,
add(
mul(
div(
mload(add(_postBytes, 0x20)),
exp(0x100, sub(32, mlength))
),
exp(0x100, sub(32, newlength))
),
mul(mlength, 2)
)
)
)
}
case 1 {
mstore(0x0, _preBytes.slot)
let sc := add(keccak256(0x0, 0x20), div(slength, 32))
sstore(_preBytes.slot, add(mul(newlength, 2), 1))
let submod := sub(32, slength)
let mc := add(_postBytes, submod)
let end := add(_postBytes, mlength)
let mask := sub(exp(0x100, submod), 1)
sstore(
sc,
add(
and(
fslot,
0xffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffff00
),
and(mload(mc), mask)
)
)
for {
mc := add(mc, 0x20)
sc := add(sc, 1)
} lt(mc, end) {
sc := add(sc, 1)
mc := add(mc, 0x20)
} {
sstore(sc, mload(mc))
}
mask := exp(0x100, sub(mc, end))
sstore(sc, mul(div(mload(mc), mask), mask))
}
default {
mstore(0x0, _preBytes.slot)
let sc := add(keccak256(0x0, 0x20), div(slength, 32))
sstore(_preBytes.slot, add(mul(newlength, 2), 1))
let slengthmod := mod(slength, 32)
let mlengthmod := mod(mlength, 32)
let submod := sub(32, slengthmod)
let mc := add(_postBytes, submod)
let end := add(_postBytes, mlength)
let mask := sub(exp(0x100, submod), 1)
sstore(sc, add(sload(sc), and(mload(mc), mask)))
for {
sc := add(sc, 1)
mc := add(mc, 0x20)
} lt(mc, end) {
sc := add(sc, 1)
mc := add(mc, 0x20)
} {
sstore(sc, mload(mc))
}
mask := exp(0x100, sub(mc, end))
sstore(sc, mul(div(mload(mc), mask), mask))
}
}
}
function slice(
bytes memory _bytes,
uint256 _start,
uint256 _length
)
internal
pure
returns (bytes memory)
{
require(_length + 31 >= _length, "slice_overflow");
require(_bytes.length >= _start + _length, "slice_outOfBounds");
bytes memory tempBytes;
assembly {
switch iszero(_length)
case 0 {
tempBytes := mload(0x40)
let lengthmod := and(_length, 31)
let mc := add(add(tempBytes, lengthmod), mul(0x20, iszero(lengthmod)))
let end := add(mc, _length)
for {
let cc := add(add(add(_bytes, lengthmod), mul(0x20, iszero(lengthmod))), _start)
} lt(mc, end) {
mc := add(mc, 0x20)
cc := add(cc, 0x20)
} {
mstore(mc, mload(cc))
}
mstore(tempBytes, _length)
mstore(0x40, and(add(mc, 31), not(31)))
}
default {
tempBytes := mload(0x40)
mstore(tempBytes, 0)
mstore(0x40, add(tempBytes, 0x20))
}
}
return tempBytes;
}
function toAddress(bytes memory _bytes, uint256 _start) internal pure returns (address) {
require(_bytes.length >= _start + 20, "toAddress_outOfBounds");
address tempAddress;
assembly {
tempAddress := div(mload(add(add(_bytes, 0x20), _start)), 0x1000000000000000000000000)
}
return tempAddress;
}
function toUint8(bytes memory _bytes, uint256 _start) internal pure returns (uint8) {
require(_bytes.length >= _start + 1 , "toUint8_outOfBounds");
uint8 tempUint;
assembly {
tempUint := mload(add(add(_bytes, 0x1), _start))
}
return tempUint;
}
function toUint16(bytes memory _bytes, uint256 _start) internal pure returns (uint16) {
require(_bytes.length >= _start + 2, "toUint16_outOfBounds");
uint16 tempUint;
assembly {
tempUint := mload(add(add(_bytes, 0x2), _start))
}
return tempUint;
}
function toUint32(bytes memory _bytes, uint256 _start) internal pure returns (uint32) {
require(_bytes.length >= _start + 4, "toUint32_outOfBounds");
uint32 tempUint;
assembly {
tempUint := mload(add(add(_bytes, 0x4), _start))
}
return tempUint;
}
function toUint64(bytes memory _bytes, uint256 _start) internal pure returns (uint64) {
require(_bytes.length >= _start + 8, "toUint64_outOfBounds");
uint64 tempUint;
assembly {
tempUint := mload(add(add(_bytes, 0x8), _start))
}
return tempUint;
}
function toUint96(bytes memory _bytes, uint256 _start) internal pure returns (uint96) {
require(_bytes.length >= _start + 12, "toUint96_outOfBounds");
uint96 tempUint;
assembly {
tempUint := mload(add(add(_bytes, 0xc), _start))
}
return tempUint;
}
function toUint128(bytes memory _bytes, uint256 _start) internal pure returns (uint128) {
require(_bytes.length >= _start + 16, "toUint128_outOfBounds");
uint128 tempUint;
assembly {
tempUint := mload(add(add(_bytes, 0x10), _start))
}
return tempUint;
}
function toUint256(bytes memory _bytes, uint256 _start) internal pure returns (uint256) {
require(_bytes.length >= _start + 32, "toUint256_outOfBounds");
uint256 tempUint;
assembly {
tempUint := mload(add(add(_bytes, 0x20), _start))
}
return tempUint;
}
function toBytes32(bytes memory _bytes, uint256 _start) internal pure returns (bytes32) {
require(_bytes.length >= _start + 32, "toBytes32_outOfBounds");
bytes32 tempBytes32;
assembly {
tempBytes32 := mload(add(add(_bytes, 0x20), _start))
}
return tempBytes32;
}
function equal(bytes memory _preBytes, bytes memory _postBytes) internal pure returns (bool) {
bool success = true;
assembly {
let length := mload(_preBytes)
switch eq(length, mload(_postBytes))
case 1 {
let cb := 1
let mc := add(_preBytes, 0x20)
let end := add(mc, length)
for {
let cc := add(_postBytes, 0x20)
} eq(add(lt(mc, end), cb), 2) {
mc := add(mc, 0x20)
cc := add(cc, 0x20)
} {
if iszero(eq(mload(mc), mload(cc))) {
success := 0
cb := 0
}
}
}
default {
success := 0
}
}
return success;
}
function equalStorage(
bytes storage _preBytes,
bytes memory _postBytes
)
internal
view
returns (bool)
{
bool success = true;
assembly {
let fslot := sload(_preBytes.slot)
let slength := div(and(fslot, sub(mul(0x100, iszero(and(fslot, 1))), 1)), 2)
let mlength := mload(_postBytes)
switch eq(slength, mlength)
case 1 {
if iszero(iszero(slength)) {
switch lt(slength, 32)
case 1 {
fslot := mul(div(fslot, 0x100), 0x100)
if iszero(eq(fslot, mload(add(_postBytes, 0x20)))) {
success := 0
}
}
default {
let cb := 1
mstore(0x0, _preBytes.slot)
let sc := keccak256(0x0, 0x20)
let mc := add(_postBytes, 0x20)
let end := add(mc, mlength)
for {} eq(add(lt(mc, end), cb), 2) {
sc := add(sc, 1)
mc := add(mc, 0x20)
} {
if iszero(eq(sload(sc), mload(mc))) {
success := 0
cb := 0
}
}
}
}
}
default {
success := 0
}
}
return success;
}
}
文件 3 的 10:IERC20.sol
pragma solidity ^0.8.0;
interface IERC20 {
event Transfer(address indexed from, address indexed to, uint256 value);
event Approval(address indexed owner, address indexed spender, uint256 value);
function totalSupply() external view returns (uint256);
function balanceOf(address account) external view returns (uint256);
function transfer(address to, uint256 amount) external returns (bool);
function allowance(address owner, address spender) external view returns (uint256);
function approve(address spender, uint256 amount) external returns (bool);
function transferFrom(
address from,
address to,
uint256 amount
) external returns (bool);
}
文件 4 的 10:ITokenBridge.sol
pragma solidity ^0.8.0;
import "./IWETH.sol";
import "./IWormhole.sol";
interface ITokenBridge {
struct Transfer {
uint8 payloadID;
uint256 amount;
bytes32 tokenAddress;
uint16 tokenChain;
bytes32 to;
uint16 toChain;
uint256 fee;
}
struct TransferWithPayload {
uint8 payloadID;
uint256 amount;
bytes32 tokenAddress;
uint16 tokenChain;
bytes32 to;
uint16 toChain;
bytes32 fromAddress;
bytes payload;
}
struct AssetMeta {
uint8 payloadID;
bytes32 tokenAddress;
uint16 tokenChain;
uint8 decimals;
bytes32 symbol;
bytes32 name;
}
struct RegisterChain {
bytes32 module;
uint8 action;
uint16 chainId;
uint16 emitterChainID;
bytes32 emitterAddress;
}
struct UpgradeContract {
bytes32 module;
uint8 action;
uint16 chainId;
bytes32 newContract;
}
struct RecoverChainId {
bytes32 module;
uint8 action;
uint256 evmChainId;
uint16 newChainId;
}
event ContractUpgraded(address indexed oldContract, address indexed newContract);
function _parseTransferCommon(bytes memory encoded) external pure returns (Transfer memory transfer);
function attestToken(address tokenAddress, uint32 nonce) external payable returns (uint64 sequence);
function wrapAndTransferETH(uint16 recipientChain, bytes32 recipient, uint256 arbiterFee, uint32 nonce) external payable returns (uint64 sequence);
function wrapAndTransferETHWithPayload(uint16 recipientChain, bytes32 recipient, uint32 nonce, bytes memory payload) external payable returns (uint64 sequence);
function transferTokens(address token, uint256 amount, uint16 recipientChain, bytes32 recipient, uint256 arbiterFee, uint32 nonce) external payable returns (uint64 sequence);
function transferTokensWithPayload(address token, uint256 amount, uint16 recipientChain, bytes32 recipient, uint32 nonce, bytes memory payload) external payable returns (uint64 sequence);
function updateWrapped(bytes memory encodedVm) external returns (address token);
function createWrapped(bytes memory encodedVm) external returns (address token);
function completeTransferWithPayload(bytes memory encodedVm) external returns (bytes memory);
function completeTransferAndUnwrapETHWithPayload(bytes memory encodedVm) external returns (bytes memory);
function completeTransfer(bytes memory encodedVm) external;
function completeTransferAndUnwrapETH(bytes memory encodedVm) external;
function encodeAssetMeta(AssetMeta memory meta) external pure returns (bytes memory encoded);
function encodeTransfer(Transfer memory transfer) external pure returns (bytes memory encoded);
function encodeTransferWithPayload(TransferWithPayload memory transfer) external pure returns (bytes memory encoded);
function parsePayloadID(bytes memory encoded) external pure returns (uint8 payloadID);
function parseAssetMeta(bytes memory encoded) external pure returns (AssetMeta memory meta);
function parseTransfer(bytes memory encoded) external pure returns (Transfer memory transfer);
function parseTransferWithPayload(bytes memory encoded) external pure returns (TransferWithPayload memory transfer);
function governanceActionIsConsumed(bytes32 hash) external view returns (bool);
function isInitialized(address impl) external view returns (bool);
function isTransferCompleted(bytes32 hash) external view returns (bool);
function wormhole() external view returns (IWormhole);
function chainId() external view returns (uint16);
function evmChainId() external view returns (uint256);
function isFork() external view returns (bool);
function governanceChainId() external view returns (uint16);
function governanceContract() external view returns (bytes32);
function wrappedAsset(uint16 tokenChainId, bytes32 tokenAddress) external view returns (address);
function bridgeContracts(uint16 chainId_) external view returns (bytes32);
function tokenImplementation() external view returns (address);
function WETH() external view returns (IWETH);
function outstandingBridged(address token) external view returns (uint256);
function isWrappedAsset(address token) external view returns (bool);
function finality() external view returns (uint8);
function implementation() external view returns (address);
function initialize() external;
function registerChain(bytes memory encodedVM) external;
function upgrade(bytes memory encodedVM) external;
function submitRecoverChainId(bytes memory encodedVM) external;
function parseRegisterChain(bytes memory encoded) external pure returns (RegisterChain memory chain);
function parseUpgrade(bytes memory encoded) external pure returns (UpgradeContract memory chain);
function parseRecoverChainId(bytes memory encodedRecoverChainId) external pure returns (RecoverChainId memory rci);
}
文件 5 的 10:IWETH.sol
pragma solidity ^0.8.0;
import "@openzeppelin/contracts/token/ERC20/IERC20.sol";
interface IWETH is IERC20 {
function deposit() external payable;
function withdraw(uint amount) external;
}
文件 6 的 10:IWormhole.sol
pragma solidity ^0.8.0;
interface IWormhole {
struct GuardianSet {
address[] keys;
uint32 expirationTime;
}
struct Signature {
bytes32 r;
bytes32 s;
uint8 v;
uint8 guardianIndex;
}
struct VM {
uint8 version;
uint32 timestamp;
uint32 nonce;
uint16 emitterChainId;
bytes32 emitterAddress;
uint64 sequence;
uint8 consistencyLevel;
bytes payload;
uint32 guardianSetIndex;
Signature[] signatures;
bytes32 hash;
}
struct ContractUpgrade {
bytes32 module;
uint8 action;
uint16 chain;
address newContract;
}
struct GuardianSetUpgrade {
bytes32 module;
uint8 action;
uint16 chain;
GuardianSet newGuardianSet;
uint32 newGuardianSetIndex;
}
struct SetMessageFee {
bytes32 module;
uint8 action;
uint16 chain;
uint256 messageFee;
}
struct TransferFees {
bytes32 module;
uint8 action;
uint16 chain;
uint256 amount;
bytes32 recipient;
}
struct RecoverChainId {
bytes32 module;
uint8 action;
uint256 evmChainId;
uint16 newChainId;
}
event LogMessagePublished(address indexed sender, uint64 sequence, uint32 nonce, bytes payload, uint8 consistencyLevel);
event ContractUpgraded(address indexed oldContract, address indexed newContract);
event GuardianSetAdded(uint32 indexed index);
function publishMessage(
uint32 nonce,
bytes memory payload,
uint8 consistencyLevel
) external payable returns (uint64 sequence);
function initialize() external;
function parseAndVerifyVM(bytes calldata encodedVM) external view returns (VM memory vm, bool valid, string memory reason);
function verifyVM(VM memory vm) external view returns (bool valid, string memory reason);
function verifySignatures(bytes32 hash, Signature[] memory signatures, GuardianSet memory guardianSet) external pure returns (bool valid, string memory reason);
function parseVM(bytes memory encodedVM) external pure returns (VM memory vm);
function quorum(uint numGuardians) external pure returns (uint numSignaturesRequiredForQuorum);
function getGuardianSet(uint32 index) external view returns (GuardianSet memory);
function getCurrentGuardianSetIndex() external view returns (uint32);
function getGuardianSetExpiry() external view returns (uint32);
function governanceActionIsConsumed(bytes32 hash) external view returns (bool);
function isInitialized(address impl) external view returns (bool);
function chainId() external view returns (uint16);
function isFork() external view returns (bool);
function governanceChainId() external view returns (uint16);
function governanceContract() external view returns (bytes32);
function messageFee() external view returns (uint256);
function evmChainId() external view returns (uint256);
function nextSequence(address emitter) external view returns (uint64);
function parseContractUpgrade(bytes memory encodedUpgrade) external pure returns (ContractUpgrade memory cu);
function parseGuardianSetUpgrade(bytes memory encodedUpgrade) external pure returns (GuardianSetUpgrade memory gsu);
function parseSetMessageFee(bytes memory encodedSetMessageFee) external pure returns (SetMessageFee memory smf);
function parseTransferFees(bytes memory encodedTransferFees) external pure returns (TransferFees memory tf);
function parseRecoverChainId(bytes memory encodedRecoverChainId) external pure returns (RecoverChainId memory rci);
function submitContractUpgrade(bytes memory _vm) external;
function submitSetMessageFee(bytes memory _vm) external;
function submitNewGuardianSet(bytes memory _vm) external;
function submitTransferFees(bytes memory _vm) external;
function submitRecoverChainId(bytes memory _vm) external;
}
文件 7 的 10:MayanStructs.sol
pragma solidity ^0.8.0;
contract MayanStructs {
struct Swap {
uint8 payloadId;
bytes32 tokenAddr;
uint16 tokenChainId;
bytes32 destAddr;
uint16 destChainId;
bytes32 sourceAddr;
uint16 sourceChainId;
uint64 sequence;
uint64 amountOutMin;
uint64 deadline;
uint64 swapFee;
uint64 redeemFee;
uint64 refundFee;
bytes32 auctionAddr;
bool unwrapRedeem;
bool unwrapRefund;
}
struct Redeem {
uint8 payloadId;
bytes32 recipient;
uint64 relayerFee;
bool unwrap;
uint64 gasDrop;
bytes customPayload;
}
}
文件 8 的 10:MayanSwap.sol
pragma solidity ^0.8.0;
import "@openzeppelin/contracts/token/ERC20/IERC20.sol";
import "@openzeppelin/contracts/token/ERC20/utils/SafeERC20.sol";
import "./interfaces/ITokenBridge.sol";
import "./interfaces/IWormhole.sol";
import "./MayanStructs.sol";
import "./libs/BytesLib.sol";
contract MayanSwap {
event Redeemed(uint16 indexed emitterChainId, bytes32 indexed emitterAddress, uint64 indexed sequence);
using SafeERC20 for IERC20;
using BytesLib for bytes;
ITokenBridge tokenBridge;
address guardian;
address nextGuardian;
bool paused;
IWETH weth;
uint16 homeChainId;
struct RelayerFees {
uint64 swapFee;
uint64 redeemFee;
uint64 refundFee;
}
struct Criteria {
uint256 transferDeadline;
uint64 swapDeadline;
uint64 amountOutMin;
bool unwrap;
uint64 gasDrop;
bytes customPayload;
}
struct Recepient {
bytes32 mayanAddr;
uint16 mayanChainId;
bytes32 auctionAddr;
bytes32 destAddr;
uint16 destChainId;
bytes32 referrer;
bytes32 refundAddr;
}
constructor(address _tokenBridge, address _weth) {
tokenBridge = ITokenBridge(_tokenBridge);
homeChainId = tokenBridge.chainId();
guardian = msg.sender;
weth = IWETH(_weth);
}
function swap(RelayerFees memory relayerFees, Recepient memory recipient, bytes32 tokenOutAddr, uint16 tokenOutChainId, Criteria memory criteria, address tokenIn, uint256 amountIn) public payable returns (uint64 sequence) {
require(paused == false, 'contract is paused');
require(block.timestamp <= criteria.transferDeadline, 'deadline passed');
if (criteria.unwrap) {
require(criteria.gasDrop == 0, 'gas drop not allowed');
}
uint8 decimals = decimalsOf(tokenIn);
uint256 normalizedAmount = normalizeAmount(amountIn, decimals);
require(relayerFees.swapFee + relayerFees.refundFee < normalizedAmount, 'fees exceed amount');
require(relayerFees.redeemFee < criteria.amountOutMin, 'redeem fee exceeds min output');
amountIn = deNormalizeAmount(normalizedAmount, decimals);
IERC20(tokenIn).safeTransferFrom(msg.sender, address(this), amountIn);
IERC20(tokenIn).safeIncreaseAllowance(address(tokenBridge), amountIn);
uint64 seq1 = tokenBridge.transferTokens{ value: msg.value/2 }(tokenIn, amountIn, recipient.mayanChainId, recipient.mayanAddr, 0, 0);
MayanStructs.Swap memory swapStruct = MayanStructs.Swap({
payloadId: criteria.customPayload.length > 0 ? 2 : 1,
tokenAddr: tokenOutAddr,
tokenChainId: tokenOutChainId,
destAddr: recipient.destAddr,
destChainId: recipient.destChainId,
sourceAddr: recipient.refundAddr,
sourceChainId: homeChainId,
sequence: seq1,
amountOutMin: criteria.amountOutMin,
deadline: criteria.swapDeadline,
swapFee: relayerFees.swapFee,
redeemFee: relayerFees.redeemFee,
refundFee: relayerFees.refundFee,
auctionAddr: recipient.auctionAddr,
unwrapRedeem: criteria.unwrap,
unwrapRefund: false
});
bytes memory encoded = encodeSwap(swapStruct)
.concat(abi.encodePacked(swapStruct.unwrapRedeem, swapStruct.unwrapRefund, recipient.referrer, criteria.gasDrop));
if (swapStruct.payloadId == 2) {
require(swapStruct.destChainId == recipient.mayanChainId, 'invalid chain id with payload');
encoded = encoded.concat(abi.encodePacked(criteria.customPayload));
}
sequence = tokenBridge.wormhole().publishMessage{
value : msg.value/2
}(0, encoded, tokenBridge.finality());
}
function wrapAndSwapETH(RelayerFees memory relayerFees, Recepient memory recipient, bytes32 tokenOutAddr, uint16 tokenOutChainId, Criteria memory criteria) public payable returns (uint64 sequence) {
require(paused == false, 'contract is paused');
require(block.timestamp <= criteria.transferDeadline, 'deadline passed');
if (criteria.unwrap) {
require(criteria.gasDrop == 0, 'gas drop not allowed');
}
uint wormholeFee = tokenBridge.wormhole().messageFee();
uint256 normalizedAmount = normalizeAmount(msg.value - 2*wormholeFee, 18);
require(relayerFees.swapFee + relayerFees.refundFee < normalizedAmount, 'fees exceed amount');
require(relayerFees.redeemFee < criteria.amountOutMin, 'redeem fee exceeds min output');
uint256 amountIn = deNormalizeAmount(normalizedAmount, 18);
uint64 seq1 = tokenBridge.wrapAndTransferETH{ value: amountIn + wormholeFee }(recipient.mayanChainId, recipient.mayanAddr, 0, 0);
uint dust = msg.value - 2*wormholeFee - amountIn;
if (dust > 0) {
payable(msg.sender).transfer(dust);
}
MayanStructs.Swap memory swapStruct = MayanStructs.Swap({
payloadId: criteria.customPayload.length > 0 ? 2 : 1,
tokenAddr: tokenOutAddr,
tokenChainId: tokenOutChainId,
destAddr: recipient.destAddr,
destChainId: recipient.destChainId,
sourceAddr: recipient.refundAddr,
sourceChainId: homeChainId,
sequence: seq1,
amountOutMin: criteria.amountOutMin,
deadline: criteria.swapDeadline,
swapFee: relayerFees.swapFee,
redeemFee: relayerFees.redeemFee,
refundFee: relayerFees.refundFee,
auctionAddr: recipient.auctionAddr,
unwrapRedeem: criteria.unwrap,
unwrapRefund: true
});
bytes memory encoded = encodeSwap(swapStruct)
.concat(abi.encodePacked(swapStruct.unwrapRedeem, swapStruct.unwrapRefund, recipient.referrer, criteria.gasDrop));
if (swapStruct.payloadId == 2) {
require(swapStruct.destChainId == recipient.mayanChainId, 'invalid chain id with payload');
encoded = encoded.concat(abi.encodePacked(criteria.customPayload));
}
sequence = tokenBridge.wormhole().publishMessage{
value : wormholeFee
}(0, encoded, tokenBridge.finality());
}
function redeem(bytes memory encodedVm) public payable {
IWormhole.VM memory vm = tokenBridge.wormhole().parseVM(encodedVm);
ITokenBridge.TransferWithPayload memory transferPayload = tokenBridge.parseTransferWithPayload(vm.payload);
MayanStructs.Redeem memory redeemPayload = parseRedeemPayload(transferPayload.payload);
address recipient = truncateAddress(redeemPayload.recipient);
if (redeemPayload.payloadId == 2) {
require(msg.sender == recipient, 'not recipient');
}
address tokenAddr;
if (transferPayload.tokenChain == homeChainId) {
tokenAddr = truncateAddress(transferPayload.tokenAddress);
} else {
tokenAddr = tokenBridge.wrappedAsset(transferPayload.tokenChain, transferPayload.tokenAddress);
}
uint256 amount = IERC20(tokenAddr).balanceOf(address(this));
tokenBridge.completeTransferWithPayload(encodedVm);
amount = IERC20(tokenAddr).balanceOf(address(this)) - amount;
uint256 relayerFee = deNormalizeAmount(uint256(redeemPayload.relayerFee), decimalsOf(tokenAddr));
require(amount > relayerFee, 'relayer fee exeeds amount');
if (redeemPayload.gasDrop > 0) {
uint256 gasDrop = deNormalizeAmount(uint256(redeemPayload.gasDrop), decimalsOf(address(weth)));
require(msg.value == gasDrop, 'incorrect gas drop');
payable(recipient).transfer(gasDrop);
}
if (redeemPayload.unwrap && tokenAddr == address(weth)) {
weth.withdraw(amount);
payable(msg.sender).transfer(relayerFee);
payable(recipient).transfer(amount - relayerFee);
} else {
IERC20(tokenAddr).safeTransfer(msg.sender, relayerFee);
IERC20(tokenAddr).safeTransfer(recipient, amount - relayerFee);
}
emit Redeemed(vm.emitterChainId, vm.emitterAddress, vm.sequence);
}
function redeemAndUnwrap(bytes memory encodedVm) public {
IWormhole.VM memory vm = tokenBridge.wormhole().parseVM(encodedVm);
ITokenBridge.TransferWithPayload memory transferPayload = tokenBridge.parseTransferWithPayload(vm.payload);
require(transferPayload.tokenChain == homeChainId, 'not home chain');
address tokenAddr = truncateAddress(transferPayload.tokenAddress);
require(tokenAddr == address(weth), 'not weth');
MayanStructs.Redeem memory redeemPayload = parseRedeemPayload(transferPayload.payload);
require(redeemPayload.unwrap, 'not unwrap');
address recipient = truncateAddress(redeemPayload.recipient);
if (redeemPayload.payloadId == 2) {
require(msg.sender == recipient, 'not recipient');
}
uint256 amount = address(this).balance;
tokenBridge.completeTransferAndUnwrapETHWithPayload(encodedVm);
amount = address(this).balance - amount;
uint256 relayerFee = deNormalizeAmount(uint256(redeemPayload.relayerFee), 18);
require(amount > relayerFee, 'relayer fee exeeds amount');
payable(msg.sender).transfer(relayerFee);
payable(recipient).transfer(amount - relayerFee);
emit Redeemed(vm.emitterChainId, vm.emitterAddress, vm.sequence);
}
function parseRedeemPayload(bytes memory encoded) public pure returns (MayanStructs.Redeem memory r) {
uint index = 0;
r.payloadId = encoded.toUint8(index);
index += 1;
require(r.payloadId == 1 || r.payloadId == 2, 'payload id not supported');
r.recipient = encoded.toBytes32(index);
index += 32;
r.relayerFee = encoded.toUint64(index);
index += 8;
r.unwrap = encoded[index] != bytes1(0);
index += 1;
r.gasDrop = encoded.toUint64(index);
index += 8;
if (r.payloadId == 2) {
r.customPayload = encoded.slice(index, encoded.length - index);
} else {
require(index == encoded.length, 'invalid payload length');
}
}
function truncateAddress(bytes32 b) internal pure returns (address) {
require(bytes12(b) == 0, 'invalid EVM address');
return address(uint160(uint256(b)));
}
function decimalsOf(address token) internal view returns(uint8) {
(,bytes memory queriedDecimals) = token.staticcall(abi.encodeWithSignature('decimals()'));
return abi.decode(queriedDecimals, (uint8));
}
function normalizeAmount(uint256 amount, uint8 decimals) internal pure returns(uint256) {
if (decimals > 8) {
amount /= 10 ** (decimals - 8);
}
return amount;
}
function deNormalizeAmount(uint256 amount, uint8 decimals) internal pure returns(uint256) {
if (decimals > 8) {
amount *= 10 ** (decimals - 8);
}
return amount;
}
function encodeSwap(MayanStructs.Swap memory s) public pure returns(bytes memory encoded) {
encoded = abi.encodePacked(
s.payloadId,
s.tokenAddr,
s.tokenChainId,
s.destAddr,
s.destChainId,
s.sourceAddr,
s.sourceChainId,
s.sequence,
s.amountOutMin,
s.deadline,
s.swapFee,
s.redeemFee,
s.refundFee,
s.auctionAddr
);
}
function setPause(bool _pause) public {
require(msg.sender == guardian, 'only guardian');
paused = _pause;
}
function isPaused() public view returns(bool) {
return paused;
}
function changeGuardian(address newGuardian) public {
require(msg.sender == guardian, 'only guardian');
nextGuardian = newGuardian;
}
function claimGuardian() public {
require(msg.sender == nextGuardian, 'only next guardian');
guardian = nextGuardian;
}
function sweepToken(address token, uint256 amount, address to) public {
require(msg.sender == guardian, 'only guardian');
IERC20(token).safeTransfer(to, amount);
}
function sweepEth(uint256 amount, address payable to) public {
require(msg.sender == guardian, 'only guardian');
require(to != address(0), 'transfer to the zero address');
to.transfer(amount);
}
function getWeth() public view returns(address) {
return address(weth);
}
receive() external payable {}
}
文件 9 的 10:SafeERC20.sol
pragma solidity ^0.8.0;
import "../IERC20.sol";
import "../extensions/draft-IERC20Permit.sol";
import "../../../utils/Address.sol";
library SafeERC20 {
using Address for address;
function safeTransfer(
IERC20 token,
address to,
uint256 value
) internal {
_callOptionalReturn(token, abi.encodeWithSelector(token.transfer.selector, to, value));
}
function safeTransferFrom(
IERC20 token,
address from,
address to,
uint256 value
) internal {
_callOptionalReturn(token, abi.encodeWithSelector(token.transferFrom.selector, from, to, value));
}
function safeApprove(
IERC20 token,
address spender,
uint256 value
) internal {
require(
(value == 0) || (token.allowance(address(this), spender) == 0),
"SafeERC20: approve from non-zero to non-zero allowance"
);
_callOptionalReturn(token, abi.encodeWithSelector(token.approve.selector, spender, value));
}
function safeIncreaseAllowance(
IERC20 token,
address spender,
uint256 value
) internal {
uint256 newAllowance = token.allowance(address(this), spender) + value;
_callOptionalReturn(token, abi.encodeWithSelector(token.approve.selector, spender, newAllowance));
}
function safeDecreaseAllowance(
IERC20 token,
address spender,
uint256 value
) internal {
unchecked {
uint256 oldAllowance = token.allowance(address(this), spender);
require(oldAllowance >= value, "SafeERC20: decreased allowance below zero");
uint256 newAllowance = oldAllowance - value;
_callOptionalReturn(token, abi.encodeWithSelector(token.approve.selector, spender, newAllowance));
}
}
function safePermit(
IERC20Permit token,
address owner,
address spender,
uint256 value,
uint256 deadline,
uint8 v,
bytes32 r,
bytes32 s
) internal {
uint256 nonceBefore = token.nonces(owner);
token.permit(owner, spender, value, deadline, v, r, s);
uint256 nonceAfter = token.nonces(owner);
require(nonceAfter == nonceBefore + 1, "SafeERC20: permit did not succeed");
}
function _callOptionalReturn(IERC20 token, bytes memory data) private {
bytes memory returndata = address(token).functionCall(data, "SafeERC20: low-level call failed");
if (returndata.length > 0) {
require(abi.decode(returndata, (bool)), "SafeERC20: ERC20 operation did not succeed");
}
}
}
文件 10 的 10:draft-IERC20Permit.sol
pragma solidity ^0.8.0;
interface IERC20Permit {
function permit(
address owner,
address spender,
uint256 value,
uint256 deadline,
uint8 v,
bytes32 r,
bytes32 s
) external;
function nonces(address owner) external view returns (uint256);
function DOMAIN_SEPARATOR() external view returns (bytes32);
}
{
"compilationTarget": {
"src/MayanSwap.sol": "MayanSwap"
},
"evmVersion": "istanbul",
"libraries": {},
"metadata": {
"bytecodeHash": "ipfs"
},
"optimizer": {
"enabled": true,
"runs": 200
},
"remappings": []
}
[{"inputs":[{"internalType":"address","name":"_tokenBridge","type":"address"},{"internalType":"address","name":"_weth","type":"address"}],"stateMutability":"nonpayable","type":"constructor"},{"anonymous":false,"inputs":[{"indexed":true,"internalType":"uint16","name":"emitterChainId","type":"uint16"},{"indexed":true,"internalType":"bytes32","name":"emitterAddress","type":"bytes32"},{"indexed":true,"internalType":"uint64","name":"sequence","type":"uint64"}],"name":"Redeemed","type":"event"},{"inputs":[{"internalType":"address","name":"newGuardian","type":"address"}],"name":"changeGuardian","outputs":[],"stateMutability":"nonpayable","type":"function"},{"inputs":[],"name":"claimGuardian","outputs":[],"stateMutability":"nonpayable","type":"function"},{"inputs":[{"components":[{"internalType":"uint8","name":"payloadId","type":"uint8"},{"internalType":"bytes32","name":"tokenAddr","type":"bytes32"},{"internalType":"uint16","name":"tokenChainId","type":"uint16"},{"internalType":"bytes32","name":"destAddr","type":"bytes32"},{"internalType":"uint16","name":"destChainId","type":"uint16"},{"internalType":"bytes32","name":"sourceAddr","type":"bytes32"},{"internalType":"uint16","name":"sourceChainId","type":"uint16"},{"internalType":"uint64","name":"sequence","type":"uint64"},{"internalType":"uint64","name":"amountOutMin","type":"uint64"},{"internalType":"uint64","name":"deadline","type":"uint64"},{"internalType":"uint64","name":"swapFee","type":"uint64"},{"internalType":"uint64","name":"redeemFee","type":"uint64"},{"internalType":"uint64","name":"refundFee","type":"uint64"},{"internalType":"bytes32","name":"auctionAddr","type":"bytes32"},{"internalType":"bool","name":"unwrapRedeem","type":"bool"},{"internalType":"bool","name":"unwrapRefund","type":"bool"}],"internalType":"struct MayanStructs.Swap","name":"s","type":"tuple"}],"name":"encodeSwap","outputs":[{"internalType":"bytes","name":"encoded","type":"bytes"}],"stateMutability":"pure","type":"function"},{"inputs":[],"name":"getWeth","outputs":[{"internalType":"address","name":"","type":"address"}],"stateMutability":"view","type":"function"},{"inputs":[],"name":"isPaused","outputs":[{"internalType":"bool","name":"","type":"bool"}],"stateMutability":"view","type":"function"},{"inputs":[{"internalType":"bytes","name":"encoded","type":"bytes"}],"name":"parseRedeemPayload","outputs":[{"components":[{"internalType":"uint8","name":"payloadId","type":"uint8"},{"internalType":"bytes32","name":"recipient","type":"bytes32"},{"internalType":"uint64","name":"relayerFee","type":"uint64"},{"internalType":"bool","name":"unwrap","type":"bool"},{"internalType":"uint64","name":"gasDrop","type":"uint64"},{"internalType":"bytes","name":"customPayload","type":"bytes"}],"internalType":"struct MayanStructs.Redeem","name":"r","type":"tuple"}],"stateMutability":"pure","type":"function"},{"inputs":[{"internalType":"bytes","name":"encodedVm","type":"bytes"}],"name":"redeem","outputs":[],"stateMutability":"payable","type":"function"},{"inputs":[{"internalType":"bytes","name":"encodedVm","type":"bytes"}],"name":"redeemAndUnwrap","outputs":[],"stateMutability":"nonpayable","type":"function"},{"inputs":[{"internalType":"bool","name":"_pause","type":"bool"}],"name":"setPause","outputs":[],"stateMutability":"nonpayable","type":"function"},{"inputs":[{"components":[{"internalType":"uint64","name":"swapFee","type":"uint64"},{"internalType":"uint64","name":"redeemFee","type":"uint64"},{"internalType":"uint64","name":"refundFee","type":"uint64"}],"internalType":"struct MayanSwap.RelayerFees","name":"relayerFees","type":"tuple"},{"components":[{"internalType":"bytes32","name":"mayanAddr","type":"bytes32"},{"internalType":"uint16","name":"mayanChainId","type":"uint16"},{"internalType":"bytes32","name":"auctionAddr","type":"bytes32"},{"internalType":"bytes32","name":"destAddr","type":"bytes32"},{"internalType":"uint16","name":"destChainId","type":"uint16"},{"internalType":"bytes32","name":"referrer","type":"bytes32"},{"internalType":"bytes32","name":"refundAddr","type":"bytes32"}],"internalType":"struct MayanSwap.Recepient","name":"recipient","type":"tuple"},{"internalType":"bytes32","name":"tokenOutAddr","type":"bytes32"},{"internalType":"uint16","name":"tokenOutChainId","type":"uint16"},{"components":[{"internalType":"uint256","name":"transferDeadline","type":"uint256"},{"internalType":"uint64","name":"swapDeadline","type":"uint64"},{"internalType":"uint64","name":"amountOutMin","type":"uint64"},{"internalType":"bool","name":"unwrap","type":"bool"},{"internalType":"uint64","name":"gasDrop","type":"uint64"},{"internalType":"bytes","name":"customPayload","type":"bytes"}],"internalType":"struct MayanSwap.Criteria","name":"criteria","type":"tuple"},{"internalType":"address","name":"tokenIn","type":"address"},{"internalType":"uint256","name":"amountIn","type":"uint256"}],"name":"swap","outputs":[{"internalType":"uint64","name":"sequence","type":"uint64"}],"stateMutability":"payable","type":"function"},{"inputs":[{"internalType":"uint256","name":"amount","type":"uint256"},{"internalType":"address payable","name":"to","type":"address"}],"name":"sweepEth","outputs":[],"stateMutability":"nonpayable","type":"function"},{"inputs":[{"internalType":"address","name":"token","type":"address"},{"internalType":"uint256","name":"amount","type":"uint256"},{"internalType":"address","name":"to","type":"address"}],"name":"sweepToken","outputs":[],"stateMutability":"nonpayable","type":"function"},{"inputs":[{"components":[{"internalType":"uint64","name":"swapFee","type":"uint64"},{"internalType":"uint64","name":"redeemFee","type":"uint64"},{"internalType":"uint64","name":"refundFee","type":"uint64"}],"internalType":"struct MayanSwap.RelayerFees","name":"relayerFees","type":"tuple"},{"components":[{"internalType":"bytes32","name":"mayanAddr","type":"bytes32"},{"internalType":"uint16","name":"mayanChainId","type":"uint16"},{"internalType":"bytes32","name":"auctionAddr","type":"bytes32"},{"internalType":"bytes32","name":"destAddr","type":"bytes32"},{"internalType":"uint16","name":"destChainId","type":"uint16"},{"internalType":"bytes32","name":"referrer","type":"bytes32"},{"internalType":"bytes32","name":"refundAddr","type":"bytes32"}],"internalType":"struct MayanSwap.Recepient","name":"recipient","type":"tuple"},{"internalType":"bytes32","name":"tokenOutAddr","type":"bytes32"},{"internalType":"uint16","name":"tokenOutChainId","type":"uint16"},{"components":[{"internalType":"uint256","name":"transferDeadline","type":"uint256"},{"internalType":"uint64","name":"swapDeadline","type":"uint64"},{"internalType":"uint64","name":"amountOutMin","type":"uint64"},{"internalType":"bool","name":"unwrap","type":"bool"},{"internalType":"uint64","name":"gasDrop","type":"uint64"},{"internalType":"bytes","name":"customPayload","type":"bytes"}],"internalType":"struct MayanSwap.Criteria","name":"criteria","type":"tuple"}],"name":"wrapAndSwapETH","outputs":[{"internalType":"uint64","name":"sequence","type":"uint64"}],"stateMutability":"payable","type":"function"},{"stateMutability":"payable","type":"receive"}]