TrueCloudLab/neo-go
8
0
Fork 0
mirror of https://github.com/nspcc-dev/neo-go.git synced 2024-12-23 13:41:37 +00:00
Code Issues Projects Releases Packages Wiki Activity

Merge pull request #1225 from nspcc-dev/fix/equal

Browse source 
vm: make EQUAL type strict
This commit is contained in:
Roman Khimov 2020-07-23 23:42:22 +03:00 committed by GitHub
commit a45c160f10
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 10 additions and 17 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
pkg/vm/stackitem/item.go
View file

@ -356,11 +356,7 @@ func (i *BigInteger) Equals(s Item) bool {
return false
}
val, ok := s.(*BigInteger)
if ok {
return i.value.Cmp(val.value) == 0
}
bs, err := s.TryBytes()
return err == nil && bytes.Equal(i.Bytes(), bs)
return ok && i.value.Cmp(val.value) == 0
}
// Value implements Item interface.
@ -454,11 +450,7 @@ func (i *Bool) Equals(s Item) bool {
return false
}
val, ok := s.(*Bool)
if ok {
return i.value == val.value
}
bs, err := s.TryBytes()
return err == nil && bytes.Equal(i.Bytes(), bs)
return ok && i.value == val.value
}
// Type implements Item interface.
@ -530,8 +522,8 @@ func (i *ByteArray) Equals(s Item) bool {
} else if s == nil {
return false
}
bs, err := s.TryBytes()
return err == nil && bytes.Equal(i.value, bs)
val, ok := s.(*ByteArray)
return ok && bytes.Equal(i.value, val.value)
}
// Dup implements Item interface.

4
pkg/vm/stackitem/item_test.go
View file

@ -211,7 +211,7 @@ var equalsTestCases = map[string][]struct {
{
item1: NewBool(true),
item2: NewBigInteger(big.NewInt(1)),
result: true,
result: false,
},
{
item1: NewBool(true),
@ -238,7 +238,7 @@ var equalsTestCases = map[string][]struct {
{
item1: NewByteArray([]byte{1}),
item2: NewBigInteger(big.NewInt(1)),
result: true,
result: false,
},
{
item1: NewByteArray([]byte{1, 2, 3}),

7
pkg/vm/vm_test.go
View file

@ -1105,7 +1105,8 @@ func TestEQUAL(t *testing.T) {
t.Run("NoArgs", getTestFuncForVM(prog, nil))
t.Run("OneArgument", getTestFuncForVM(prog, nil, 1))
t.Run("Integer", getTestFuncForVM(prog, true, 5, 5))
t.Run("IntegerByteArray", getTestFuncForVM(prog, true, []byte{16}, 16))
t.Run("IntegerByteArray", getTestFuncForVM(prog, false, []byte{16}, 16))
t.Run("BooleanInteger", getTestFuncForVM(prog, false, true, 1))
t.Run("Map", getTestFuncForVM(prog, false, stackitem.NewMap(), stackitem.NewMap()))
t.Run("Array", getTestFuncForVM(prog, false, []stackitem.Item{}, []stackitem.Item{}))
t.Run("Buffer", getTestFuncForVM(prog, false, stackitem.NewBuffer([]byte{42}), stackitem.NewBuffer([]byte{42})))
@ -1339,14 +1340,14 @@ func TestPICKITEMDupMap(t *testing.T) {
prog := makeProgram(opcode.DUP, opcode.PUSHINT8, 42, opcode.PICKITEM, opcode.ABS)
vm := load(prog)
m := stackitem.NewMap()
m.Add(stackitem.Make([]byte{42}), stackitem.Make(-1))
m.Add(stackitem.Make(42), stackitem.Make(-1))
vm.estack.Push(&Element{value: m})
runVM(t, vm)
assert.Equal(t, 2, vm.estack.Len())
assert.Equal(t, int64(1), vm.estack.Pop().BigInt().Int64())
items := vm.estack.Pop().Value().([]stackitem.MapElement)
assert.Equal(t, 1, len(items))
assert.Equal(t, []byte{42}, items[0].Key.Value())
assert.Equal(t, big.NewInt(42), items[0].Key.Value())
assert.Equal(t, big.NewInt(-1), items[0].Value.Value())
}