neo-go/pkg/vm/context.go

package vm

import (
	"errors"

	"github.com/CityOfZion/neo-go/pkg/io"
)

// Context represent the current execution context of the VM.
type Context struct {
	// Instruction pointer.
	ip int

	// The next instruction pointer.
	nextip int

	// The raw program script.
	prog []byte

	// Breakpoints
	breakPoints []int
}

// NewContext return a new Context object.
func NewContext(b []byte) *Context {
	return &Context{
		prog:        b,
		breakPoints: []int{},
	}
}

// Next returns the next instruction to execute with its parameter if any. After
// its invocation the instruction pointer points to the instruction being
// returned.
func (c *Context) Next() (Instruction, []byte, error) {
	c.ip = c.nextip
	if c.ip >= len(c.prog) {
		return RET, nil, nil
	}
	r := io.NewBinReaderFromBuf(c.prog[c.ip:])

	var instrbyte byte
	r.ReadLE(&instrbyte)
	instr := Instruction(instrbyte)
	c.nextip++

	var numtoread int
	switch instr {
	case PUSHDATA1, SYSCALL:
		var n byte
		r.ReadLE(&n)
		numtoread = int(n)
		c.nextip++
	case PUSHDATA2:
		var n uint16
		r.ReadLE(&n)
		numtoread = int(n)
		c.nextip += 2
	case PUSHDATA4:
		var n uint32
		r.ReadLE(&n)
		numtoread = int(n)
		c.nextip += 4
	case JMP, JMPIF, JMPIFNOT, CALL:
		numtoread = 2
	case APPCALL, TAILCALL:
		numtoread = 20
	default:
		if instr >= PUSHBYTES1 && instr <= PUSHBYTES75 {
			numtoread = int(instr)
		} else {
			// No parameters, can just return.
			return instr, nil, nil
		}
	}
	parameter := make([]byte, numtoread)
	r.ReadLE(parameter)
	if r.Err != nil {
		return instr, nil, errors.New("failed to read instruction parameter")
	}
	c.nextip += numtoread
	return instr, parameter, nil
}

// IP returns the absolute instruction without taking 0 into account.
// If that program starts the ip = 0 but IP() will return 1, cause its
// the first instruction.
func (c *Context) IP() int {
	return c.ip + 1
}

// LenInstr returns the number of instructions loaded.
func (c *Context) LenInstr() int {
	return len(c.prog)
}

// CurrInstr returns the current instruction and opcode.
func (c *Context) CurrInstr() (int, Instruction) {
	return c.ip, Instruction(c.prog[c.ip])
}

// Copy returns an new exact copy of c.
func (c *Context) Copy() *Context {
	ctx := new(Context)
	*ctx = *c
	return ctx
}

// Program returns the loaded program.
func (c *Context) Program() []byte {
	return c.prog
}

// Value implements StackItem interface.
func (c *Context) Value() interface{} {
	return c
}

func (c *Context) atBreakPoint() bool {
	for _, n := range c.breakPoints {
		if n == c.ip {
			return true
		}
	}
	return false
}

func (c *Context) String() string {
	return "execution context"
}