crowetic/protocol
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

@0x/zero-ex: Always offset storage bucket ID by 1. (#2579)

Browse Source 
Co-authored-by: Lawrence Forman <me@merklejerk.com>
This commit is contained in:
Lawrence Forman 2020-05-08 20:13:45 -04:00 committed by GitHub
parent ff1811ef90
commit b348084e2a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 17 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
contracts/zero-ex/contracts/src/storage/LibOwnableStorage.sol
View File

@ -33,9 +33,9 @@ library LibOwnableStorage {
/// @dev Get the storage bucket for this contract.
function getStorage() internal pure returns (Storage storage stor) {
uint256 storageOffset = LibStorage.getStorageOffset(
uint256 storageSlot = LibStorage.getStorageSlot(
LibStorage.StorageId.Ownable
);
assembly { stor_slot := storageOffset }
assembly { stor_slot := storageSlot }
}
}

4
contracts/zero-ex/contracts/src/storage/LibProxyStorage.sol
View File

@ -33,9 +33,9 @@ library LibProxyStorage {
/// @dev Get the storage bucket for this contract.
function getStorage() internal pure returns (Storage storage stor) {
uint256 storageOffset = LibStorage.getStorageOffset(
uint256 storageSlot = LibStorage.getStorageSlot(
LibStorage.StorageId.Proxy
);
assembly { stor_slot := storageOffset }
assembly { stor_slot := storageSlot }
}
}

4
contracts/zero-ex/contracts/src/storage/LibSimpleFunctionRegistryStorage.sol
View File

@ -33,9 +33,9 @@ library LibSimpleFunctionRegistryStorage {
/// @dev Get the storage bucket for this contract.
function getStorage() internal pure returns (Storage storage stor) {
uint256 storageOffset = LibStorage.getStorageOffset(
uint256 storageSlot = LibStorage.getStorageSlot(
LibStorage.StorageId.SimpleFunctionRegistry
);
assembly { stor_slot := storageOffset }
assembly { stor_slot := storageSlot }
}
}

22
contracts/zero-ex/contracts/src/storage/LibStorage.sol
View File

@ -23,29 +23,29 @@ pragma experimental ABIEncoderV2;
/// @dev Common storage helpers
library LibStorage {
/// @dev What to multiply a storage ID by to get its offset.
/// This is also the maximum number of fields inside a storage
/// bucket.
uint256 internal constant STORAGE_OFFSET_MULTIPLIER = 1e18;
/// @dev What to bit-shift a storage ID by to get its slot.
/// This gives us a maximum of 2**128 inline fields in each bucket.
uint256 private constant STORAGE_SLOT_EXP = 128;
/// @dev Storage IDs for feature storage buckets.
enum StorageId {
Unused, // Unused buffer for state accidents.
Proxy,
SimpleFunctionRegistry,
Ownable
}
/// @dev Get the storage offset given a storage ID.
/// @dev Get the storage slot given a storage ID. We assign unique, well-spaced
/// slots to storage bucket variables to ensure they do not overlap.
/// See: https://solidity.readthedocs.io/en/v0.6.6/assembly.html#access-to-external-variables-functions-and-libraries
/// @param storageId An entry in `StorageId`
/// @return offset The storage offset.
function getStorageOffset(StorageId storageId)
/// @return slot The storage slot.
function getStorageSlot(StorageId storageId)
internal
pure
returns (uint256 offset)
returns (uint256 slot)
{
// This should never overflow with a reasonable `STORAGE_OFFSET_MULTIPLIER`
// This should never overflow with a reasonable `STORAGE_SLOT_EXP`
// because Solidity will do a range check on `storageId` during the cast.
return uint256(storageId) * STORAGE_OFFSET_MULTIPLIER;
return (uint256(storageId) + 1) << STORAGE_SLOT_EXP;
}
}