NoBey/openzeppelin-contracts
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Improved multiple token transfer batch transfer equivalence rule

Browse Source
This commit is contained in:
Thomas Adams
2022-06-06 13:02:12 -07:00
parent 6a4fc6acb8
commit ee2f0ecb68
1 changed files with 18 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
certora/specs/ERC1155Supply.spec
View File

@ -77,6 +77,9 @@ rule multipleTokenSafeTransferFromSafeBatchTransferFromEquivalence {
mathint holderStartingBalanceA = balanceOf(holder, tokenA); mathint holderStartingBalanceA = balanceOf(holder, tokenA);
mathint holderStartingBalanceB = balanceOf(holder, tokenB); mathint holderStartingBalanceB = balanceOf(holder, tokenB);
mathint holderStartingBalanceC = balanceOf(holder, tokenC); mathint holderStartingBalanceC = balanceOf(holder, tokenC);
mathint recipientStartingBalanceA = balanceOf(recipient, tokenA);
mathint recipientStartingBalanceB = balanceOf(recipient, tokenB);
mathint recipientStartingBalanceC = balanceOf(recipient, tokenC);
require tokens.length == 3; require transferAmounts.length == 3; require tokens.length == 3; require transferAmounts.length == 3;
require tokens[0] == tokenA; require transferAmounts[0] == transferAmountA; require tokens[0] == tokenA; require transferAmounts[0] == transferAmountA;
@ -87,19 +90,28 @@ rule multipleTokenSafeTransferFromSafeBatchTransferFromEquivalence {
safeTransferFrom(e, holder, recipient, tokenA, transferAmountA, data) at beforeTransfers; safeTransferFrom(e, holder, recipient, tokenA, transferAmountA, data) at beforeTransfers;
safeTransferFrom(e, holder, recipient, tokenB, transferAmountB, data); safeTransferFrom(e, holder, recipient, tokenB, transferAmountB, data);
safeTransferFrom(e, holder, recipient, tokenC, transferAmountC, data); safeTransferFrom(e, holder, recipient, tokenC, transferAmountC, data);
mathint holderSafeTransferFromChangeA = holderStartingBalanceA - balanceOf(holder, tokenA); mathint holderSafeTransferFromBalanceChangeA = holderStartingBalanceA - balanceOf(holder, tokenA);
mathint holderSafeTransferFromChangeB = holderStartingBalanceB - balanceOf(holder, tokenB); mathint holderSafeTransferFromBalanceChangeB = holderStartingBalanceB - balanceOf(holder, tokenB);
mathint holderSafeTransferFromChangeC = holderStartingBalanceC - balanceOf(holder, tokenC); mathint holderSafeTransferFromBalanceChangeC = holderStartingBalanceC - balanceOf(holder, tokenC);
mathint recipientSafeTransferFromBalanceChangeA = balanceOf(recipient, tokenA) - recipientStartingBalanceA;
mathint recipientSafeTransferFromBalanceChangeB = balanceOf(recipient, tokenB) - recipientStartingBalanceB;
mathint recipientSafeTransferFromBalanceChangeC = balanceOf(recipient, tokenC) - recipientStartingBalanceC;
// transferring via safeBatchTransferFrom // transferring via safeBatchTransferFrom
safeBatchTransferFrom(e, holder, recipient, tokens, transferAmounts, data) at beforeTransfers; safeBatchTransferFrom(e, holder, recipient, tokens, transferAmounts, data) at beforeTransfers;
mathint holderSafeBatchTransferFromBalanceChangeA = holderStartingBalanceA - balanceOf(holder, tokenA); mathint holderSafeBatchTransferFromBalanceChangeA = holderStartingBalanceA - balanceOf(holder, tokenA);
mathint holderSafeBatchTransferFromBalanceChangeB = holderStartingBalanceB - balanceOf(holder, tokenB); mathint holderSafeBatchTransferFromBalanceChangeB = holderStartingBalanceB - balanceOf(holder, tokenB);
mathint holderSafeBatchTransferFromBalanceChangeC = holderStartingBalanceC - balanceOf(holder, tokenC); mathint holderSafeBatchTransferFromBalanceChangeC = holderStartingBalanceC - balanceOf(holder, tokenC);
mathint recipientSafeBatchTransferFromBalanceChangeA = balanceOf(recipient, tokenA) - recipientStartingBalanceA;
mathint recipientSafeBatchTransferFromBalanceChangeB = balanceOf(recipient, tokenB) - recipientStartingBalanceB;
mathint recipientSafeBatchTransferFromBalanceChangeC = balanceOf(recipient, tokenC) - recipientStartingBalanceC;
assert holderSafeTransferFromChangeA == holderSafeBatchTransferFromBalanceChangeA assert holderSafeTransferFromBalanceChangeA == holderSafeBatchTransferFromBalanceChangeA
&& holderSafeTransferFromChangeB == holderSafeBatchTransferFromBalanceChangeB && holderSafeTransferFromBalanceChangeB == holderSafeBatchTransferFromBalanceChangeB
&& holderSafeTransferFromChangeC == holderSafeBatchTransferFromBalanceChangeC, && holderSafeTransferFromBalanceChangeC == holderSafeBatchTransferFromBalanceChangeC
&& recipientSafeTransferFromBalanceChangeA == recipientSafeBatchTransferFromBalanceChangeA
&& recipientSafeTransferFromBalanceChangeB == recipientSafeBatchTransferFromBalanceChangeB
&& recipientSafeTransferFromBalanceChangeC == recipientSafeBatchTransferFromBalanceChangeC,
"Transferring multiple tokens via safeTransferFrom or safeBatchTransferFrom must be equivalent"; "Transferring multiple tokens via safeTransferFrom or safeBatchTransferFrom must be equivalent";
} }