TrueCloudLab/neoneo-go
7
0
Fork 2
Code Issues Pull requests Projects Releases Packages Wiki Activity

dao: restrict GetStorageItems by prefix

Browse source 
All storage items can still be retrived via zero-length prefix.
This commit is contained in:
Evgenii Stratonikov 2020-05-27 10:56:47 +03:00
parent 776bd85ded
commit 503442a60d
5 changed files with 10 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

8
pkg/core/dao/cacheddao.go
View file

@ -321,7 +321,7 @@ type StorageIteratorFunc func() ([]byte, []byte, error)
// GetStorageItemsIterator returns iterator over all storage items.
// Function returned can be called until first error.
func (cd *Cached) GetStorageItemsIterator(hash util.Uint160, prefix []byte) (StorageIteratorFunc, error) {
items, err := cd.DAO.GetStorageItems(hash)
items, err := cd.DAO.GetStorageItems(hash, prefix)
if err != nil {
return nil, err
}
@ -352,7 +352,7 @@ func (cd *Cached) GetStorageItemsIterator(hash util.Uint160, prefix []byte) (Sto
index++
for ; index < len(items); index++ {
_, ok := cache[string(items[index].Key)]
if !ok && bytes.HasPrefix(items[index].Key, prefix) {
if !ok {
return items[index].Key, items[index].Value, nil
}
}
@ -362,8 +362,8 @@ func (cd *Cached) GetStorageItemsIterator(hash util.Uint160, prefix []byte) (Sto
}
// GetStorageItems returns all storage items for a given scripthash.
func (cd *Cached) GetStorageItems(hash util.Uint160) ([]StorageItemWithKey, error) {
items, err := cd.DAO.GetStorageItems(hash)
func (cd *Cached) GetStorageItems(hash util.Uint160, prefix []byte) ([]StorageItemWithKey, error) {
items, err := cd.DAO.GetStorageItems(hash, prefix)
if err != nil {
return nil, err
}