TrueCloudLab/neo-go
6
0
Fork 0
mirror of https://github.com/nspcc-dev/neo-go.git synced 2024-10-16 15:43:25 +00:00
Code Issues Projects Releases Packages Wiki Activity

Add popTwoIntegers convenience func

Browse source
This commit is contained in:
BlockChainDev 2019-03-15 22:54:52 +00:00
parent baf9d2b768
commit abc3b46f1c
1 changed files with 36 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

45
pkg/vm/vm_ops_maths.go
View file

@ -1,19 +1,16 @@
package vm
import "github.com/CityOfZion/neo-go/pkg/vm/stack"
import (
"github.com/CityOfZion/neo-go/pkg/vm/stack"
)
// Add adds two stack Items together.
// Returns an error if either items cannot be casted to an integer
// or if integers cannot be added together
func Add(ctx *stack.Context, istack *stack.Invocation) error {
operandA, err := ctx.Estack.PopInt()
if err != nil {
return err
}
operandB, err := ctx.Estack.PopInt()
if err != nil {
return err
}
operandA, operandB, err := popTwoIntegers(ctx)
res, err := operandA.Add(operandB)
if err != nil {
return err
@ -23,3 +20,33 @@ func Add(ctx *stack.Context, istack *stack.Invocation) error {
return nil
}
// Sub subtracts two stack Items.
// Returns an error if either items cannot be casted to an integer
// or if integers cannot be subtracted together
func Sub(ctx *stack.Context, istack *stack.Invocation) error {
operandA, operandB, err := popTwoIntegers(ctx)
res, err := operandB.Sub(operandA)
if err != nil {
return err
}
ctx.Estack.Push(res)
return nil
}
func popTwoIntegers(ctx *stack.Context) (*stack.Int, *stack.Int, error) {
operandA, err := ctx.Estack.PopInt()
if err != nil {
return nil, nil, err
}
operandB, err := ctx.Estack.PopInt()
if err != nil {
return nil, nil, err
}
return operandA, operandB, nil
}