forked from TrueCloudLab/neoneo-go
vm: implement storage slots
This commit is contained in:
parent
8e916f5ab8
commit
81cbf183af
2 changed files with 50 additions and 0 deletions
29
pkg/vm/slot.go
Normal file
29
pkg/vm/slot.go
Normal file
|
@ -0,0 +1,29 @@
|
||||||
|
package vm
|
||||||
|
|
||||||
|
// Slot is a fixed-size slice of stack items.
|
||||||
|
type Slot struct {
|
||||||
|
storage []StackItem
|
||||||
|
}
|
||||||
|
|
||||||
|
// newSlot returns new slot of n items.
|
||||||
|
func newSlot(n int) *Slot {
|
||||||
|
return &Slot{
|
||||||
|
storage: make([]StackItem, n),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Set sets i-th storage slot.
|
||||||
|
func (s *Slot) Set(i int, item StackItem) {
|
||||||
|
s.storage[i] = item
|
||||||
|
}
|
||||||
|
|
||||||
|
// Get returns item contained in i-th slot.
|
||||||
|
func (s *Slot) Get(i int) StackItem {
|
||||||
|
if item := s.storage[i]; item != nil {
|
||||||
|
return item
|
||||||
|
}
|
||||||
|
return NullItem{}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Size returns slot size.
|
||||||
|
func (s *Slot) Size() int { return len(s.storage) }
|
21
pkg/vm/slot_test.go
Normal file
21
pkg/vm/slot_test.go
Normal file
|
@ -0,0 +1,21 @@
|
||||||
|
package vm
|
||||||
|
|
||||||
|
import (
|
||||||
|
"math/big"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"github.com/stretchr/testify/require"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestSlot_Get(t *testing.T) {
|
||||||
|
s := newSlot(3)
|
||||||
|
require.NotNil(t, s)
|
||||||
|
require.Equal(t, 3, s.Size())
|
||||||
|
|
||||||
|
// NullItem is the default
|
||||||
|
item := s.Get(2)
|
||||||
|
require.Equal(t, NullItem{}, item)
|
||||||
|
|
||||||
|
s.Set(1, NewBigIntegerItem(big.NewInt(42)))
|
||||||
|
require.Equal(t, NewBigIntegerItem(big.NewInt(42)), s.Get(1))
|
||||||
|
}
|
Loading…
Reference in a new issue