forked from cerc-io/ipld-eth-server
Add function for adding hashed keys to mapping
- Tool to facilitate parsing diffs from Geth patch that emits hashed versions of storage keys
This commit is contained in:
parent
3fb8e13979
commit
8111f4ec5e
@ -46,6 +46,21 @@ const (
|
|||||||
IndexEleven = "000000000000000000000000000000000000000000000000000000000000000b"
|
IndexEleven = "000000000000000000000000000000000000000000000000000000000000000b"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
func AddHashedKeys(currentMappings map[common.Hash]utils.StorageValueMetadata) map[common.Hash]utils.StorageValueMetadata {
|
||||||
|
copyOfCurrentMappings := make(map[common.Hash]utils.StorageValueMetadata)
|
||||||
|
for k, v := range currentMappings {
|
||||||
|
copyOfCurrentMappings[k] = v
|
||||||
|
}
|
||||||
|
for k, v := range copyOfCurrentMappings {
|
||||||
|
currentMappings[hashKey(k)] = v
|
||||||
|
}
|
||||||
|
return currentMappings
|
||||||
|
}
|
||||||
|
|
||||||
|
func hashKey(key common.Hash) common.Hash {
|
||||||
|
return common.BytesToHash(crypto.Keccak256(key.Bytes()))
|
||||||
|
}
|
||||||
|
|
||||||
func GetMapping(indexOnContract, key string) common.Hash {
|
func GetMapping(indexOnContract, key string) common.Hash {
|
||||||
keyBytes := common.FromHex(key + indexOnContract)
|
keyBytes := common.FromHex(key + indexOnContract)
|
||||||
encoded := crypto.Keccak256(keyBytes)
|
encoded := crypto.Keccak256(keyBytes)
|
||||||
|
@ -5,9 +5,31 @@ import (
|
|||||||
. "github.com/onsi/ginkgo"
|
. "github.com/onsi/ginkgo"
|
||||||
. "github.com/onsi/gomega"
|
. "github.com/onsi/gomega"
|
||||||
"github.com/vulcanize/vulcanizedb/libraries/shared/storage"
|
"github.com/vulcanize/vulcanizedb/libraries/shared/storage"
|
||||||
|
"github.com/vulcanize/vulcanizedb/libraries/shared/storage/utils"
|
||||||
)
|
)
|
||||||
|
|
||||||
var _ = Describe("Mappings", func() {
|
var _ = Describe("Mappings", func() {
|
||||||
|
Describe("AddHashedKeys", func() {
|
||||||
|
It("returns a copy of the map with an additional slot for the hashed version of every key", func() {
|
||||||
|
fakeMap := map[common.Hash]utils.StorageValueMetadata{}
|
||||||
|
fakeStorageKey := common.HexToHash("72c72de6b203d67cb6cd54fc93300109fcc6fd6eac88e390271a3d548794d800")
|
||||||
|
var fakeMappingKey utils.Key = "fakeKey"
|
||||||
|
fakeMetadata := utils.StorageValueMetadata{
|
||||||
|
Name: "fakeName",
|
||||||
|
Keys: map[utils.Key]string{fakeMappingKey: "fakeValue"},
|
||||||
|
Type: utils.Uint48,
|
||||||
|
}
|
||||||
|
fakeMap[fakeStorageKey] = fakeMetadata
|
||||||
|
|
||||||
|
result := storage.AddHashedKeys(fakeMap)
|
||||||
|
|
||||||
|
Expect(len(result)).To(Equal(2))
|
||||||
|
expectedHashedStorageKey := common.HexToHash("2165edb4e1c37b99b60fa510d84f939dd35d5cd1d1c8f299d6456ea09df65a76")
|
||||||
|
Expect(fakeMap[fakeStorageKey]).To(Equal(fakeMetadata))
|
||||||
|
Expect(fakeMap[expectedHashedStorageKey]).To(Equal(fakeMetadata))
|
||||||
|
})
|
||||||
|
})
|
||||||
|
|
||||||
Describe("GetMapping", func() {
|
Describe("GetMapping", func() {
|
||||||
It("returns the storage key for a mapping when passed the mapping's index on the contract and the desired value's key", func() {
|
It("returns the storage key for a mapping when passed the mapping's index on the contract and the desired value's key", func() {
|
||||||
// ex. solidity:
|
// ex. solidity:
|
||||||
|
Loading…
Reference in New Issue
Block a user