diff --git a/src/TokenDivider.sol b/src/TokenDivider.sol index 15dcfa0..b02abd9 100644 --- a/src/TokenDivider.sol +++ b/src/TokenDivider.sol @@ -223,6 +223,8 @@ contract TokenDivider is IERC721Receiver, Ownable { revert TokenDivider__AmountCantBeZero(); } + // @audit: no checks for 0 price? + ERC20Info memory tokenInfo = nftToErc20Info[nftPegged]; if (balances[msg.sender][tokenInfo.erc20Address] < amount) { revert TokenDivider__InsuficientBalance(); @@ -236,6 +238,7 @@ contract TokenDivider is IERC721Receiver, Ownable { emit OrderPublished(amount, msg.sender, nftPegged); + // @audit: what if I cancel IERC20(tokenInfo.erc20Address).transferFrom(msg.sender, address(this), amount); }