xgo/vm.go

884 lines
20 KiB
Go
Raw Normal View History

2019-12-20 22:40:38 +03:00
package tengo
2019-01-09 10:17:42 +03:00
import (
"fmt"
"sync/atomic"
2019-01-09 10:17:42 +03:00
2019-12-30 00:38:51 +03:00
"github.com/d5/tengo/v2/parser"
"github.com/d5/tengo/v2/token"
2019-01-09 10:17:42 +03:00
)
2019-12-20 22:40:38 +03:00
// frame represents a function call frame.
type frame struct {
fn *CompiledFunction
freeVars []*ObjectPtr
ip int
basePointer int
}
2019-01-09 10:17:42 +03:00
// VM is a virtual machine that executes the bytecode compiled by Compiler.
2019-01-09 10:17:42 +03:00
type VM struct {
2019-12-20 22:40:38 +03:00
constants []Object
stack [StackSize]Object
sp int
2019-12-20 22:40:38 +03:00
globals []Object
fileSet *parser.SourceFileSet
2019-12-20 22:40:38 +03:00
frames [MaxFrames]frame
framesIndex int
2019-12-20 22:40:38 +03:00
curFrame *frame
curInsts []byte
ip int
aborting int64
maxAllocs int64
allocs int64
err error
2019-01-09 10:17:42 +03:00
}
// NewVM creates a VM.
2019-12-20 22:40:38 +03:00
func NewVM(
bytecode *Bytecode,
globals []Object,
maxAllocs int64,
) *VM {
2019-01-09 10:17:42 +03:00
if globals == nil {
2019-12-20 22:40:38 +03:00
globals = make([]Object, GlobalsSize)
2019-01-09 10:17:42 +03:00
}
v := &VM{
constants: bytecode.Constants,
sp: 0,
globals: globals,
fileSet: bytecode.FileSet,
framesIndex: 1,
ip: -1,
maxAllocs: maxAllocs,
2019-01-09 10:17:42 +03:00
}
v.frames[0].fn = bytecode.MainFunction
v.frames[0].ip = -1
v.curFrame = &v.frames[0]
v.curInsts = v.curFrame.fn.Instructions
return v
2019-01-09 10:17:42 +03:00
}
// Abort aborts the execution.
2019-01-09 10:17:42 +03:00
func (v *VM) Abort() {
atomic.StoreInt64(&v.aborting, 1)
2019-01-09 10:17:42 +03:00
}
// Run starts the execution.
func (v *VM) Run() (err error) {
// reset VM states
v.sp = 0
v.curFrame = &(v.frames[0])
v.curInsts = v.curFrame.fn.Instructions
v.framesIndex = 1
v.ip = -1
v.allocs = v.maxAllocs + 1
v.run()
2019-04-01 18:02:11 +03:00
atomic.StoreInt64(&v.aborting, 0)
err = v.err
if err != nil {
2019-12-20 22:40:38 +03:00
filePos := v.fileSet.Position(
v.curFrame.fn.SourcePos(v.ip - 1))
err = fmt.Errorf("Runtime Error: %w\n\tat %s",
err, filePos)
for v.framesIndex > 1 {
v.framesIndex--
v.curFrame = &v.frames[v.framesIndex-1]
2019-12-20 22:40:38 +03:00
filePos = v.fileSet.Position(
v.curFrame.fn.SourcePos(v.curFrame.ip - 1))
err = fmt.Errorf("%w\n\tat %s", err, filePos)
}
return err
}
return nil
}
func (v *VM) run() {
2019-04-01 18:02:11 +03:00
for atomic.LoadInt64(&v.aborting) == 0 {
2019-01-20 18:41:57 +03:00
v.ip++
2019-01-09 10:17:42 +03:00
switch v.curInsts[v.ip] {
case parser.OpConstant:
2019-01-20 18:41:57 +03:00
v.ip += 2
cidx := int(v.curInsts[v.ip]) | int(v.curInsts[v.ip-1])<<8
v.stack[v.sp] = v.constants[cidx]
v.sp++
case parser.OpNull:
2019-12-20 22:40:38 +03:00
v.stack[v.sp] = UndefinedValue
v.sp++
case parser.OpBinaryOp:
v.ip++
right := v.stack[v.sp-1]
left := v.stack[v.sp-2]
tok := token.Token(v.curInsts[v.ip])
res, e := left.BinaryOp(tok, right)
if e != nil {
v.sp -= 2
2019-12-20 22:40:38 +03:00
if e == ErrInvalidOperator {
v.err = fmt.Errorf("invalid operation: %s %s %s",
left.TypeName(), tok.String(), right.TypeName())
return
}
v.err = e
return
}
v.allocs--
if v.allocs == 0 {
v.err = ErrObjectAllocLimit
return
}
v.stack[v.sp-2] = res
v.sp--
case parser.OpEqual:
right := v.stack[v.sp-1]
left := v.stack[v.sp-2]
v.sp -= 2
if left.Equals(right) {
2019-12-20 22:40:38 +03:00
v.stack[v.sp] = TrueValue
2019-01-09 10:17:42 +03:00
} else {
2019-12-20 22:40:38 +03:00
v.stack[v.sp] = FalseValue
2019-01-09 10:17:42 +03:00
}
v.sp++
case parser.OpNotEqual:
right := v.stack[v.sp-1]
left := v.stack[v.sp-2]
v.sp -= 2
if left.Equals(right) {
2019-12-20 22:40:38 +03:00
v.stack[v.sp] = FalseValue
2019-01-09 10:17:42 +03:00
} else {
2019-12-20 22:40:38 +03:00
v.stack[v.sp] = TrueValue
2019-01-09 10:17:42 +03:00
}
v.sp++
case parser.OpPop:
v.sp--
case parser.OpTrue:
2019-12-20 22:40:38 +03:00
v.stack[v.sp] = TrueValue
v.sp++
case parser.OpFalse:
2019-12-20 22:40:38 +03:00
v.stack[v.sp] = FalseValue
v.sp++
case parser.OpLNot:
operand := v.stack[v.sp-1]
v.sp--
if operand.IsFalsy() {
2019-12-20 22:40:38 +03:00
v.stack[v.sp] = TrueValue
2019-01-09 10:17:42 +03:00
} else {
2019-12-20 22:40:38 +03:00
v.stack[v.sp] = FalseValue
2019-01-09 10:17:42 +03:00
}
v.sp++
case parser.OpBComplement:
operand := v.stack[v.sp-1]
v.sp--
2019-01-09 10:17:42 +03:00
switch x := operand.(type) {
2019-12-20 22:40:38 +03:00
case *Int:
var res Object = &Int{Value: ^x.Value}
v.allocs--
if v.allocs == 0 {
v.err = ErrObjectAllocLimit
return
}
v.stack[v.sp] = res
v.sp++
2019-01-09 10:17:42 +03:00
default:
2019-12-20 22:40:38 +03:00
v.err = fmt.Errorf("invalid operation: ^%s",
operand.TypeName())
return
2019-01-09 10:17:42 +03:00
}
case parser.OpMinus:
operand := v.stack[v.sp-1]
v.sp--
2019-01-09 10:17:42 +03:00
switch x := operand.(type) {
2019-12-20 22:40:38 +03:00
case *Int:
var res Object = &Int{Value: -x.Value}
v.allocs--
if v.allocs == 0 {
v.err = ErrObjectAllocLimit
return
}
v.stack[v.sp] = res
v.sp++
2019-12-20 22:40:38 +03:00
case *Float:
var res Object = &Float{Value: -x.Value}
v.allocs--
if v.allocs == 0 {
v.err = ErrObjectAllocLimit
return
}
v.stack[v.sp] = res
v.sp++
2019-01-09 10:17:42 +03:00
default:
2019-12-20 22:40:38 +03:00
v.err = fmt.Errorf("invalid operation: -%s",
operand.TypeName())
return
2019-01-09 10:17:42 +03:00
}
case parser.OpJumpFalsy:
2019-01-20 18:41:57 +03:00
v.ip += 2
v.sp--
if v.stack[v.sp].IsFalsy() {
pos := int(v.curInsts[v.ip]) | int(v.curInsts[v.ip-1])<<8
2019-01-20 18:41:57 +03:00
v.ip = pos - 1
2019-01-09 10:17:42 +03:00
}
case parser.OpAndJump:
2019-01-20 18:41:57 +03:00
v.ip += 2
if v.stack[v.sp-1].IsFalsy() {
pos := int(v.curInsts[v.ip]) | int(v.curInsts[v.ip-1])<<8
2019-01-20 18:41:57 +03:00
v.ip = pos - 1
2019-01-09 10:17:42 +03:00
} else {
v.sp--
2019-01-09 10:17:42 +03:00
}
case parser.OpOrJump:
2019-01-20 18:41:57 +03:00
v.ip += 2
if v.stack[v.sp-1].IsFalsy() {
v.sp--
} else {
pos := int(v.curInsts[v.ip]) | int(v.curInsts[v.ip-1])<<8
v.ip = pos - 1
2019-01-09 10:17:42 +03:00
}
case parser.OpJump:
2019-01-20 18:41:57 +03:00
pos := int(v.curInsts[v.ip+2]) | int(v.curInsts[v.ip+1])<<8
v.ip = pos - 1
case parser.OpSetGlobal:
2019-01-20 18:41:57 +03:00
v.ip += 2
v.sp--
globalIndex := int(v.curInsts[v.ip]) | int(v.curInsts[v.ip-1])<<8
v.globals[globalIndex] = v.stack[v.sp]
case parser.OpSetSelGlobal:
2019-01-20 18:41:57 +03:00
v.ip += 3
globalIndex := int(v.curInsts[v.ip-1]) | int(v.curInsts[v.ip-2])<<8
numSelectors := int(v.curInsts[v.ip])
2019-01-09 10:17:42 +03:00
// selectors and RHS value
2019-12-20 22:40:38 +03:00
selectors := make([]Object, numSelectors)
for i := 0; i < numSelectors; i++ {
selectors[i] = v.stack[v.sp-numSelectors+i]
}
val := v.stack[v.sp-numSelectors-1]
v.sp -= numSelectors + 1
2019-12-20 22:40:38 +03:00
e := indexAssign(v.globals[globalIndex], val, selectors)
if e != nil {
v.err = e
return
2019-01-09 10:17:42 +03:00
}
case parser.OpGetGlobal:
2019-01-20 18:41:57 +03:00
v.ip += 2
globalIndex := int(v.curInsts[v.ip]) | int(v.curInsts[v.ip-1])<<8
2019-01-09 10:17:42 +03:00
val := v.globals[globalIndex]
v.stack[v.sp] = val
v.sp++
case parser.OpArray:
2019-01-20 18:41:57 +03:00
v.ip += 2
numElements := int(v.curInsts[v.ip]) | int(v.curInsts[v.ip-1])<<8
2019-01-09 10:17:42 +03:00
2019-12-20 22:40:38 +03:00
var elements []Object
2019-01-09 10:17:42 +03:00
for i := v.sp - numElements; i < v.sp; i++ {
elements = append(elements, v.stack[i])
2019-01-09 10:17:42 +03:00
}
v.sp -= numElements
2019-12-20 22:40:38 +03:00
var arr Object = &Array{Value: elements}
v.allocs--
if v.allocs == 0 {
v.err = ErrObjectAllocLimit
return
}
v.stack[v.sp] = arr
v.sp++
case parser.OpMap:
2019-01-20 18:41:57 +03:00
v.ip += 2
numElements := int(v.curInsts[v.ip]) | int(v.curInsts[v.ip-1])<<8
2019-12-20 22:40:38 +03:00
kv := make(map[string]Object)
2019-01-09 10:17:42 +03:00
for i := v.sp - numElements; i < v.sp; i += 2 {
key := v.stack[i]
value := v.stack[i+1]
2019-12-20 22:40:38 +03:00
kv[key.(*String).Value] = value
2019-01-09 10:17:42 +03:00
}
v.sp -= numElements
2019-12-20 22:40:38 +03:00
var m Object = &Map{Value: kv}
v.allocs--
if v.allocs == 0 {
v.err = ErrObjectAllocLimit
return
}
v.stack[v.sp] = m
v.sp++
case parser.OpError:
value := v.stack[v.sp-1]
2019-12-20 22:40:38 +03:00
var e Object = &Error{
Value: value,
}
v.allocs--
if v.allocs == 0 {
v.err = ErrObjectAllocLimit
return
}
v.stack[v.sp-1] = e
case parser.OpImmutable:
2019-01-26 01:54:58 +03:00
value := v.stack[v.sp-1]
switch value := value.(type) {
2019-12-20 22:40:38 +03:00
case *Array:
var immutableArray Object = &ImmutableArray{
2019-01-26 01:54:58 +03:00
Value: value.Value,
}
v.allocs--
if v.allocs == 0 {
v.err = ErrObjectAllocLimit
return
}
v.stack[v.sp-1] = immutableArray
2019-12-20 22:40:38 +03:00
case *Map:
var immutableMap Object = &ImmutableMap{
2019-01-26 01:54:58 +03:00
Value: value.Value,
}
v.allocs--
if v.allocs == 0 {
v.err = ErrObjectAllocLimit
return
}
v.stack[v.sp-1] = immutableMap
2019-01-26 01:54:58 +03:00
}
case parser.OpIndex:
index := v.stack[v.sp-1]
left := v.stack[v.sp-2]
v.sp -= 2
2019-01-09 10:17:42 +03:00
2019-12-20 22:40:38 +03:00
val, err := left.IndexGet(index)
if err != nil {
if err == ErrNotIndexable {
v.err = fmt.Errorf("not indexable: %s", index.TypeName())
return
2019-01-18 12:43:46 +03:00
}
2019-12-20 22:40:38 +03:00
if err == ErrInvalidIndexType {
v.err = fmt.Errorf("invalid index type: %s",
index.TypeName())
return
}
2019-12-20 22:40:38 +03:00
v.err = err
return
2019-01-09 10:17:42 +03:00
}
2019-12-20 22:40:38 +03:00
if val == nil {
val = UndefinedValue
}
v.stack[v.sp] = val
v.sp++
case parser.OpSliceIndex:
high := v.stack[v.sp-1]
low := v.stack[v.sp-2]
left := v.stack[v.sp-3]
v.sp -= 3
var lowIdx int64
2019-12-20 22:40:38 +03:00
if low != UndefinedValue {
if low, ok := low.(*Int); ok {
lowIdx = low.Value
} else {
2019-12-20 22:40:38 +03:00
v.err = fmt.Errorf("invalid slice index type: %s",
low.TypeName())
return
}
2019-01-09 10:17:42 +03:00
}
switch left := left.(type) {
2019-12-20 22:40:38 +03:00
case *Array:
numElements := int64(len(left.Value))
var highIdx int64
2019-12-20 22:40:38 +03:00
if high == UndefinedValue {
highIdx = numElements
2019-12-20 22:40:38 +03:00
} else if high, ok := high.(*Int); ok {
highIdx = high.Value
} else {
2019-12-20 22:40:38 +03:00
v.err = fmt.Errorf("invalid slice index type: %s",
high.TypeName())
return
}
if lowIdx > highIdx {
2019-12-20 22:40:38 +03:00
v.err = fmt.Errorf("invalid slice index: %d > %d",
lowIdx, highIdx)
return
}
if lowIdx < 0 {
lowIdx = 0
} else if lowIdx > numElements {
lowIdx = numElements
}
if highIdx < 0 {
highIdx = 0
} else if highIdx > numElements {
highIdx = numElements
}
2019-12-20 22:40:38 +03:00
var val Object = &Array{
Value: left.Value[lowIdx:highIdx],
}
v.allocs--
if v.allocs == 0 {
v.err = ErrObjectAllocLimit
return
}
v.stack[v.sp] = val
v.sp++
2019-12-20 22:40:38 +03:00
case *ImmutableArray:
2019-01-26 01:54:58 +03:00
numElements := int64(len(left.Value))
var highIdx int64
2019-12-20 22:40:38 +03:00
if high == UndefinedValue {
2019-01-26 01:54:58 +03:00
highIdx = numElements
2019-12-20 22:40:38 +03:00
} else if high, ok := high.(*Int); ok {
highIdx = high.Value
} else {
2019-12-20 22:40:38 +03:00
v.err = fmt.Errorf("invalid slice index type: %s",
high.TypeName())
return
2019-01-26 01:54:58 +03:00
}
if lowIdx > highIdx {
2019-12-20 22:40:38 +03:00
v.err = fmt.Errorf("invalid slice index: %d > %d",
lowIdx, highIdx)
return
2019-01-26 01:54:58 +03:00
}
if lowIdx < 0 {
lowIdx = 0
} else if lowIdx > numElements {
lowIdx = numElements
}
if highIdx < 0 {
highIdx = 0
} else if highIdx > numElements {
highIdx = numElements
}
2019-12-20 22:40:38 +03:00
var val Object = &Array{
Value: left.Value[lowIdx:highIdx],
}
v.allocs--
if v.allocs == 0 {
v.err = ErrObjectAllocLimit
return
}
v.stack[v.sp] = val
2019-01-26 01:54:58 +03:00
v.sp++
2019-12-20 22:40:38 +03:00
case *String:
numElements := int64(len(left.Value))
var highIdx int64
2019-12-20 22:40:38 +03:00
if high == UndefinedValue {
highIdx = numElements
2019-12-20 22:40:38 +03:00
} else if high, ok := high.(*Int); ok {
highIdx = high.Value
} else {
2019-12-20 22:40:38 +03:00
v.err = fmt.Errorf("invalid slice index type: %s",
high.TypeName())
return
2019-01-09 10:17:42 +03:00
}
if lowIdx > highIdx {
2019-12-20 22:40:38 +03:00
v.err = fmt.Errorf("invalid slice index: %d > %d",
lowIdx, highIdx)
return
}
if lowIdx < 0 {
lowIdx = 0
} else if lowIdx > numElements {
lowIdx = numElements
}
if highIdx < 0 {
highIdx = 0
} else if highIdx > numElements {
highIdx = numElements
}
2019-12-20 22:40:38 +03:00
var val Object = &String{
Value: left.Value[lowIdx:highIdx],
}
v.allocs--
if v.allocs == 0 {
v.err = ErrObjectAllocLimit
return
}
v.stack[v.sp] = val
v.sp++
2019-12-20 22:40:38 +03:00
case *Bytes:
2019-01-18 12:43:46 +03:00
numElements := int64(len(left.Value))
var highIdx int64
2019-12-20 22:40:38 +03:00
if high == UndefinedValue {
2019-01-18 12:43:46 +03:00
highIdx = numElements
2019-12-20 22:40:38 +03:00
} else if high, ok := high.(*Int); ok {
highIdx = high.Value
} else {
2019-12-20 22:40:38 +03:00
v.err = fmt.Errorf("invalid slice index type: %s",
high.TypeName())
return
2019-01-18 12:43:46 +03:00
}
if lowIdx > highIdx {
2019-12-20 22:40:38 +03:00
v.err = fmt.Errorf("invalid slice index: %d > %d",
lowIdx, highIdx)
return
2019-01-18 12:43:46 +03:00
}
if lowIdx < 0 {
lowIdx = 0
} else if lowIdx > numElements {
lowIdx = numElements
}
if highIdx < 0 {
highIdx = 0
} else if highIdx > numElements {
highIdx = numElements
}
2019-12-20 22:40:38 +03:00
var val Object = &Bytes{
Value: left.Value[lowIdx:highIdx],
}
v.allocs--
if v.allocs == 0 {
v.err = ErrObjectAllocLimit
return
}
v.stack[v.sp] = val
2019-01-18 12:43:46 +03:00
v.sp++
2019-01-09 10:17:42 +03:00
}
case parser.OpCall:
2019-01-20 18:41:57 +03:00
numArgs := int(v.curInsts[v.ip+1])
v.ip++
value := v.stack[v.sp-1-numArgs]
2019-12-20 22:40:38 +03:00
if !value.CanCall() {
v.err = fmt.Errorf("not callable: %s", value.TypeName())
return
}
if callee, ok := value.(*CompiledFunction); ok {
if callee.VarArgs {
// if the closure is variadic,
// roll up all variadic parameters into an array
realArgs := callee.NumParameters - 1
varArgs := numArgs - realArgs
if varArgs >= 0 {
numArgs = realArgs + 1
2019-12-20 22:40:38 +03:00
args := make([]Object, varArgs)
spStart := v.sp - varArgs
for i := spStart; i < v.sp; i++ {
args[i-spStart] = v.stack[i]
}
2019-12-20 22:40:38 +03:00
v.stack[spStart] = &Array{Value: args}
v.sp = spStart + 1
}
}
if numArgs != callee.NumParameters {
if callee.VarArgs {
2019-12-20 22:40:38 +03:00
v.err = fmt.Errorf(
"wrong number of arguments: want>=%d, got=%d",
callee.NumParameters-1, numArgs)
} else {
2019-12-20 22:40:38 +03:00
v.err = fmt.Errorf(
"wrong number of arguments: want=%d, got=%d",
callee.NumParameters, numArgs)
}
return
}
// test if it's tail-call
if callee == v.curFrame.fn { // recursion
nextOp := v.curInsts[v.ip+1]
if nextOp == parser.OpReturn ||
(nextOp == parser.OpPop &&
parser.OpReturn == v.curInsts[v.ip+2]) {
for p := 0; p < numArgs; p++ {
2019-12-20 22:40:38 +03:00
v.stack[v.curFrame.basePointer+p] =
v.stack[v.sp-numArgs+p]
}
v.sp -= numArgs + 1
v.ip = -1 // reset IP to beginning of the frame
continue
}
}
2019-12-20 22:40:38 +03:00
if v.framesIndex >= MaxFrames {
v.err = ErrStackOverflow
return
}
// update call frame
v.curFrame.ip = v.ip // store current ip before call
v.curFrame = &(v.frames[v.framesIndex])
v.curFrame.fn = callee
2019-12-20 22:40:38 +03:00
v.curFrame.freeVars = callee.Free
v.curFrame.basePointer = v.sp - numArgs
v.curInsts = callee.Instructions
v.ip = -1
v.framesIndex++
v.sp = v.sp - numArgs + callee.NumLocals
2019-12-20 22:40:38 +03:00
} else {
var args []Object
args = append(args, v.stack[v.sp-numArgs:v.sp]...)
ret, e := value.Call(args...)
v.sp -= numArgs + 1
// runtime error
if e != nil {
2019-12-20 22:40:38 +03:00
if e == ErrWrongNumArguments {
v.err = fmt.Errorf(
"wrong number of arguments in call to '%s'",
value.TypeName())
return
}
2019-12-20 22:40:38 +03:00
if e, ok := e.(ErrInvalidArgumentType); ok {
v.err = fmt.Errorf(
"invalid type for argument '%s' in call to '%s': "+
"expected %s, found %s",
e.Name, value.TypeName(), e.Expected, e.Found)
return
}
v.err = e
return
}
// nil return -> undefined
if ret == nil {
2019-12-20 22:40:38 +03:00
ret = UndefinedValue
}
v.allocs--
if v.allocs == 0 {
v.err = ErrObjectAllocLimit
return
}
v.stack[v.sp] = ret
v.sp++
2019-01-09 10:17:42 +03:00
}
case parser.OpReturn:
v.ip++
2019-12-20 22:40:38 +03:00
var retVal Object
if int(v.curInsts[v.ip]) == 1 {
retVal = v.stack[v.sp-1]
} else {
2019-12-20 22:40:38 +03:00
retVal = UndefinedValue
}
//v.sp--
2019-01-09 10:17:42 +03:00
v.framesIndex--
v.curFrame = &v.frames[v.framesIndex-1]
v.curInsts = v.curFrame.fn.Instructions
2019-01-20 18:41:57 +03:00
v.ip = v.curFrame.ip
//v.sp = lastFrame.basePointer - 1
v.sp = v.frames[v.framesIndex].basePointer
// skip stack overflow check because (newSP) <= (oldSP)
v.stack[v.sp-1] = retVal
//v.sp++
case parser.OpDefineLocal:
2019-01-20 18:41:57 +03:00
v.ip++
localIndex := int(v.curInsts[v.ip])
2019-01-20 18:41:57 +03:00
sp := v.curFrame.basePointer + localIndex
// local variables can be mutated by other actions
// so always store the copy of popped value
val := v.stack[v.sp-1]
v.sp--
v.stack[sp] = val
case parser.OpSetLocal:
2019-01-20 18:41:57 +03:00
localIndex := int(v.curInsts[v.ip+1])
v.ip++
sp := v.curFrame.basePointer + localIndex
2019-12-20 22:40:38 +03:00
// update pointee of v.stack[sp] instead of replacing the pointer
// itself. this is needed because there can be free variables
// referencing the same local variables.
val := v.stack[v.sp-1]
v.sp--
2019-12-20 22:40:38 +03:00
if obj, ok := v.stack[sp].(*ObjectPtr); ok {
*obj.Value = val
val = obj
}
v.stack[sp] = val // also use a copy of popped value
case parser.OpSetSelLocal:
2019-01-20 18:41:57 +03:00
localIndex := int(v.curInsts[v.ip+1])
numSelectors := int(v.curInsts[v.ip+2])
v.ip += 2
2019-01-09 10:17:42 +03:00
// selectors and RHS value
2019-12-20 22:40:38 +03:00
selectors := make([]Object, numSelectors)
for i := 0; i < numSelectors; i++ {
selectors[i] = v.stack[v.sp-numSelectors+i]
}
val := v.stack[v.sp-numSelectors-1]
v.sp -= numSelectors + 1
dst := v.stack[v.curFrame.basePointer+localIndex]
2019-12-20 22:40:38 +03:00
if obj, ok := dst.(*ObjectPtr); ok {
dst = *obj.Value
}
if e := indexAssign(dst, val, selectors); e != nil {
v.err = e
return
2019-01-09 10:17:42 +03:00
}
case parser.OpGetLocal:
2019-01-20 18:41:57 +03:00
v.ip++
localIndex := int(v.curInsts[v.ip])
2019-01-20 18:41:57 +03:00
val := v.stack[v.curFrame.basePointer+localIndex]
2019-12-20 22:40:38 +03:00
if obj, ok := val.(*ObjectPtr); ok {
val = *obj.Value
}
v.stack[v.sp] = val
v.sp++
case parser.OpGetBuiltin:
2019-01-20 18:41:57 +03:00
v.ip++
builtinIndex := int(v.curInsts[v.ip])
2019-12-20 22:40:38 +03:00
v.stack[v.sp] = builtinFuncs[builtinIndex]
v.sp++
case parser.OpClosure:
2019-01-20 18:41:57 +03:00
v.ip += 3
constIndex := int(v.curInsts[v.ip-1]) | int(v.curInsts[v.ip-2])<<8
numFree := int(v.curInsts[v.ip])
2019-12-20 22:40:38 +03:00
fn, ok := v.constants[constIndex].(*CompiledFunction)
if !ok {
v.err = fmt.Errorf("not function: %s", fn.TypeName())
return
}
2019-12-20 22:40:38 +03:00
free := make([]*ObjectPtr, numFree)
for i := 0; i < numFree; i++ {
switch freeVar := (v.stack[v.sp-numFree+i]).(type) {
2019-12-20 22:40:38 +03:00
case *ObjectPtr:
free[i] = freeVar
default:
2019-12-20 22:40:38 +03:00
free[i] = &ObjectPtr{
Value: &v.stack[v.sp-numFree+i],
}
}
}
v.sp -= numFree
2019-12-20 22:40:38 +03:00
cl := &CompiledFunction{
Instructions: fn.Instructions,
NumLocals: fn.NumLocals,
NumParameters: fn.NumParameters,
VarArgs: fn.VarArgs,
Free: free,
2019-01-09 10:17:42 +03:00
}
v.allocs--
if v.allocs == 0 {
v.err = ErrObjectAllocLimit
return
}
v.stack[v.sp] = cl
v.sp++
case parser.OpGetFreePtr:
2019-01-20 18:41:57 +03:00
v.ip++
freeIndex := int(v.curInsts[v.ip])
val := v.curFrame.freeVars[freeIndex]
v.stack[v.sp] = val
v.sp++
case parser.OpGetFree:
v.ip++
freeIndex := int(v.curInsts[v.ip])
val := *v.curFrame.freeVars[freeIndex].Value
v.stack[v.sp] = val
v.sp++
case parser.OpSetFree:
v.ip++
freeIndex := int(v.curInsts[v.ip])
*v.curFrame.freeVars[freeIndex].Value = v.stack[v.sp-1]
v.sp--
case parser.OpGetLocalPtr:
v.ip++
localIndex := int(v.curInsts[v.ip])
sp := v.curFrame.basePointer + localIndex
val := v.stack[sp]
2019-12-20 22:40:38 +03:00
var freeVar *ObjectPtr
if obj, ok := val.(*ObjectPtr); ok {
freeVar = obj
} else {
2019-12-20 22:40:38 +03:00
freeVar = &ObjectPtr{Value: &val}
v.stack[sp] = freeVar
}
v.stack[v.sp] = freeVar
v.sp++
case parser.OpSetSelFree:
2019-01-20 18:41:57 +03:00
v.ip += 2
freeIndex := int(v.curInsts[v.ip-1])
numSelectors := int(v.curInsts[v.ip])
2019-01-09 10:17:42 +03:00
// selectors and RHS value
2019-12-20 22:40:38 +03:00
selectors := make([]Object, numSelectors)
for i := 0; i < numSelectors; i++ {
selectors[i] = v.stack[v.sp-numSelectors+i]
}
val := v.stack[v.sp-numSelectors-1]
v.sp -= numSelectors + 1
2019-12-20 22:40:38 +03:00
e := indexAssign(*v.curFrame.freeVars[freeIndex].Value,
val, selectors)
if e != nil {
v.err = e
return
2019-01-09 10:17:42 +03:00
}
case parser.OpIteratorInit:
2019-12-20 22:40:38 +03:00
var iterator Object
dst := v.stack[v.sp-1]
v.sp--
2019-12-20 22:40:38 +03:00
if !dst.CanIterate() {
v.err = fmt.Errorf("not iterable: %s", dst.TypeName())
return
2019-01-09 10:17:42 +03:00
}
2019-12-20 22:40:38 +03:00
iterator = dst.Iterate()
v.allocs--
if v.allocs == 0 {
v.err = ErrObjectAllocLimit
return
}
v.stack[v.sp] = iterator
v.sp++
case parser.OpIteratorNext:
iterator := v.stack[v.sp-1]
v.sp--
2019-12-20 22:40:38 +03:00
hasMore := iterator.(Iterator).Next()
2019-01-24 00:36:03 +03:00
if hasMore {
2019-12-20 22:40:38 +03:00
v.stack[v.sp] = TrueValue
2019-01-09 10:17:42 +03:00
} else {
2019-12-20 22:40:38 +03:00
v.stack[v.sp] = FalseValue
2019-01-09 10:17:42 +03:00
}
v.sp++
case parser.OpIteratorKey:
iterator := v.stack[v.sp-1]
v.sp--
2019-12-20 22:40:38 +03:00
val := iterator.(Iterator).Key()
v.stack[v.sp] = val
v.sp++
case parser.OpIteratorValue:
iterator := v.stack[v.sp-1]
v.sp--
2019-12-20 22:40:38 +03:00
val := iterator.(Iterator).Value()
v.stack[v.sp] = val
v.sp++
case parser.OpSuspend:
2019-12-20 22:40:38 +03:00
return
2019-01-09 10:17:42 +03:00
default:
v.err = fmt.Errorf("unknown opcode: %d", v.curInsts[v.ip])
return
2019-01-09 10:17:42 +03:00
}
}
}
// IsStackEmpty tests if the stack is empty or not.
func (v *VM) IsStackEmpty() bool {
return v.sp == 0
2019-01-09 23:39:40 +03:00
}
2019-12-20 22:40:38 +03:00
func indexAssign(dst, src Object, selectors []Object) error {
2019-01-09 10:17:42 +03:00
numSel := len(selectors)
for sidx := numSel - 1; sidx > 0; sidx-- {
2019-12-20 22:40:38 +03:00
next, err := dst.IndexGet(selectors[sidx])
if err != nil {
2019-12-20 22:40:38 +03:00
if err == ErrNotIndexable {
return fmt.Errorf("not indexable: %s", dst.TypeName())
}
if err == ErrInvalidIndexType {
return fmt.Errorf("invalid index type: %s",
selectors[sidx].TypeName())
}
return err
}
dst = next
}
2019-01-09 10:17:42 +03:00
2019-12-20 22:40:38 +03:00
if err := dst.IndexSet(selectors[0], src); err != nil {
if err == ErrNotIndexAssignable {
return fmt.Errorf("not index-assignable: %s", dst.TypeName())
}
if err == ErrInvalidIndexValueType {
return fmt.Errorf("invaid index value type: %s", src.TypeName())
}
return err
}
return nil
2019-01-09 10:17:42 +03:00
}