forked from mirror/ledisdb
1788 lines
37 KiB
Go
1788 lines
37 KiB
Go
package lua
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"github.com/yuin/gopher-lua/parse"
|
|
"io"
|
|
"math"
|
|
"os"
|
|
"runtime"
|
|
"strings"
|
|
"sync/atomic"
|
|
"time"
|
|
)
|
|
|
|
const MultRet = -1
|
|
const RegistryIndex = -10000
|
|
const EnvironIndex = -10001
|
|
const GlobalsIndex = -10002
|
|
|
|
/* ApiError {{{ */
|
|
|
|
type ApiError struct {
|
|
Type ApiErrorType
|
|
Object LValue
|
|
StackTrace string
|
|
// Underlying error. This attribute is set only if the Type is ApiErrorFile or ApiErrorSyntax
|
|
Cause error
|
|
}
|
|
|
|
func newApiError(code ApiErrorType, object LValue) *ApiError {
|
|
return &ApiError{code, object, "", nil}
|
|
}
|
|
|
|
func newApiErrorS(code ApiErrorType, message string) *ApiError {
|
|
return newApiError(code, LString(message))
|
|
}
|
|
|
|
func newApiErrorE(code ApiErrorType, err error) *ApiError {
|
|
return &ApiError{code, LString(err.Error()), "", err}
|
|
}
|
|
|
|
func (e *ApiError) Error() string {
|
|
if len(e.StackTrace) > 0 {
|
|
return fmt.Sprintf("%s\n%s", e.Object.String(), e.StackTrace)
|
|
}
|
|
return e.Object.String()
|
|
}
|
|
|
|
type ApiErrorType int
|
|
|
|
const (
|
|
ApiErrorSyntax ApiErrorType = iota
|
|
ApiErrorFile
|
|
ApiErrorRun
|
|
ApiErrorError
|
|
ApiErrorPanic
|
|
)
|
|
|
|
/* }}} */
|
|
|
|
/* ResumeState {{{ */
|
|
|
|
type ResumeState int
|
|
|
|
const (
|
|
ResumeOK ResumeState = iota
|
|
ResumeYield
|
|
ResumeError
|
|
)
|
|
|
|
/* }}} */
|
|
|
|
/* P {{{ */
|
|
|
|
type P struct {
|
|
Fn LValue
|
|
NRet int
|
|
Protect bool
|
|
Handler *LFunction
|
|
}
|
|
|
|
/* }}} */
|
|
|
|
/* Options {{{ */
|
|
|
|
// Options is a configuration that is used to create a new LState.
|
|
type Options struct {
|
|
// Call stack size. This defaults to `lua.CallStackSize`.
|
|
CallStackSize int
|
|
// Data stack size. This defaults to `lua.RegistrySize`.
|
|
RegistrySize int
|
|
// Controls whether or not libraries are opened by default
|
|
SkipOpenLibs bool
|
|
// Tells whether a Go stacktrace should be included in a Lua stacktrace when panics occur.
|
|
IncludeGoStackTrace bool
|
|
}
|
|
|
|
/* }}} */
|
|
|
|
/* Debug {{{ */
|
|
|
|
type Debug struct {
|
|
frame *callFrame
|
|
Name string
|
|
What string
|
|
Source string
|
|
CurrentLine int
|
|
NUpvalues int
|
|
LineDefined int
|
|
LastLineDefined int
|
|
}
|
|
|
|
/* }}} */
|
|
|
|
/* callFrame {{{ */
|
|
|
|
type callFrame struct {
|
|
Idx int
|
|
Fn *LFunction
|
|
Parent *callFrame
|
|
Pc int
|
|
Base int
|
|
LocalBase int
|
|
ReturnBase int
|
|
NArgs int
|
|
NRet int
|
|
TailCall int
|
|
}
|
|
|
|
type callFrameStack struct {
|
|
array []callFrame
|
|
sp int
|
|
}
|
|
|
|
func newCallFrameStack(size int) *callFrameStack {
|
|
return &callFrameStack{
|
|
array: make([]callFrame, size),
|
|
sp: 0,
|
|
}
|
|
}
|
|
|
|
func (cs *callFrameStack) IsEmpty() bool { return cs.sp == 0 }
|
|
|
|
func (cs *callFrameStack) Clear() {
|
|
cs.sp = 0
|
|
}
|
|
|
|
func (cs *callFrameStack) Push(v callFrame) { // +inline-start
|
|
cs.array[cs.sp] = v
|
|
cs.array[cs.sp].Idx = cs.sp
|
|
cs.sp++
|
|
} // +inline-end
|
|
|
|
func (cs *callFrameStack) Remove(sp int) {
|
|
psp := sp - 1
|
|
nsp := sp + 1
|
|
var pre *callFrame
|
|
var next *callFrame
|
|
if psp > 0 {
|
|
pre = &cs.array[psp]
|
|
}
|
|
if nsp < cs.sp {
|
|
next = &cs.array[nsp]
|
|
}
|
|
if next != nil {
|
|
next.Parent = pre
|
|
}
|
|
for i := sp; i+1 < cs.sp; i++ {
|
|
cs.array[i] = cs.array[i+1]
|
|
cs.array[i].Idx = i
|
|
cs.sp = i
|
|
}
|
|
cs.sp++
|
|
}
|
|
|
|
func (cs *callFrameStack) Sp() int {
|
|
return cs.sp
|
|
}
|
|
|
|
func (cs *callFrameStack) SetSp(sp int) {
|
|
cs.sp = sp
|
|
}
|
|
|
|
func (cs *callFrameStack) Last() *callFrame {
|
|
if cs.sp == 0 {
|
|
return nil
|
|
}
|
|
return &cs.array[cs.sp-1]
|
|
}
|
|
|
|
func (cs *callFrameStack) At(sp int) *callFrame {
|
|
return &cs.array[sp]
|
|
}
|
|
|
|
func (cs *callFrameStack) Pop() *callFrame {
|
|
cs.sp--
|
|
return &cs.array[cs.sp]
|
|
}
|
|
|
|
/* }}} */
|
|
|
|
/* registry {{{ */
|
|
|
|
type registry struct {
|
|
array []LValue
|
|
top int
|
|
alloc *allocator
|
|
}
|
|
|
|
func newRegistry(size int, alloc *allocator) *registry {
|
|
return ®istry{make([]LValue, size), 0, alloc}
|
|
}
|
|
|
|
func (rg *registry) SetTop(top int) {
|
|
oldtop := rg.top
|
|
rg.top = top
|
|
for i := oldtop; i < rg.top; i++ {
|
|
rg.array[i] = LNil
|
|
}
|
|
for i := rg.top; i < oldtop; i++ {
|
|
rg.array[i] = LNil
|
|
}
|
|
}
|
|
|
|
func (rg *registry) Top() int {
|
|
return rg.top
|
|
}
|
|
|
|
func (rg *registry) Push(v LValue) {
|
|
rg.array[rg.top] = v
|
|
rg.top++
|
|
}
|
|
|
|
func (rg *registry) Pop() LValue {
|
|
v := rg.array[rg.top-1]
|
|
rg.array[rg.top-1] = LNil
|
|
rg.top--
|
|
return v
|
|
}
|
|
|
|
func (rg *registry) Get(reg int) LValue {
|
|
return rg.array[reg]
|
|
}
|
|
|
|
func (rg *registry) CopyRange(regv, start, limit, n int) { // +inline-start
|
|
for i := 0; i < n; i++ {
|
|
if tidx := start + i; tidx >= rg.top || limit > -1 && tidx >= limit || tidx < 0 {
|
|
rg.array[regv+i] = LNil
|
|
} else {
|
|
rg.array[regv+i] = rg.array[tidx]
|
|
}
|
|
}
|
|
rg.top = regv + n
|
|
} // +inline-end
|
|
|
|
func (rg *registry) FillNil(regm, n int) { // +inline-start
|
|
for i := 0; i < n; i++ {
|
|
rg.array[regm+i] = LNil
|
|
}
|
|
rg.top = regm + n
|
|
} // +inline-end
|
|
|
|
func (rg *registry) Insert(value LValue, reg int) {
|
|
top := rg.Top()
|
|
if reg >= top {
|
|
rg.Set(reg, value)
|
|
return
|
|
}
|
|
top--
|
|
for ; top >= reg; top-- {
|
|
rg.Set(top+1, rg.Get(top))
|
|
}
|
|
rg.Set(reg, value)
|
|
}
|
|
|
|
func (rg *registry) Set(reg int, val LValue) {
|
|
rg.array[reg] = val
|
|
if reg >= rg.top {
|
|
rg.top = reg + 1
|
|
}
|
|
}
|
|
|
|
func (rg *registry) SetNumber(reg int, val LNumber) {
|
|
rg.array[reg] = rg.alloc.LNumber2I(val)
|
|
if reg >= rg.top {
|
|
rg.top = reg + 1
|
|
}
|
|
} /* }}} */
|
|
|
|
/* Global {{{ */
|
|
|
|
func newGlobal() *Global {
|
|
return &Global{
|
|
MainThread: nil,
|
|
Registry: newLTable(0, 32),
|
|
Global: newLTable(0, 64),
|
|
builtinMts: make(map[int]LValue),
|
|
tempFiles: make([]*os.File, 0, 10),
|
|
}
|
|
}
|
|
|
|
/* }}} */
|
|
|
|
/* package local methods {{{ */
|
|
|
|
func panicWithTraceback(L *LState) {
|
|
err := newApiError(ApiErrorRun, L.Get(-1))
|
|
err.StackTrace = L.stackTrace(0)
|
|
panic(err)
|
|
}
|
|
|
|
func panicWithoutTraceback(L *LState) {
|
|
err := newApiError(ApiErrorRun, L.Get(-1))
|
|
panic(err)
|
|
}
|
|
|
|
func newLState(options Options) *LState {
|
|
al := newAllocator(32)
|
|
ls := &LState{
|
|
G: newGlobal(),
|
|
Parent: nil,
|
|
Panic: panicWithTraceback,
|
|
Dead: false,
|
|
Options: options,
|
|
|
|
stop: 0,
|
|
reg: newRegistry(options.RegistrySize, al),
|
|
stack: newCallFrameStack(options.CallStackSize),
|
|
alloc: al,
|
|
currentFrame: nil,
|
|
wrapped: false,
|
|
uvcache: nil,
|
|
hasErrorFunc: false,
|
|
mainLoop: mainLoop,
|
|
ctx: nil,
|
|
}
|
|
ls.Env = ls.G.Global
|
|
return ls
|
|
}
|
|
|
|
func (ls *LState) printReg() {
|
|
println("-------------------------")
|
|
println("thread:", ls)
|
|
println("top:", ls.reg.Top())
|
|
if ls.currentFrame != nil {
|
|
println("function base:", ls.currentFrame.Base)
|
|
println("return base:", ls.currentFrame.ReturnBase)
|
|
} else {
|
|
println("(vm not started)")
|
|
}
|
|
println("local base:", ls.currentLocalBase())
|
|
for i := 0; i < ls.reg.Top(); i++ {
|
|
println(i, ls.reg.Get(i).String())
|
|
}
|
|
println("-------------------------")
|
|
}
|
|
|
|
func (ls *LState) printCallStack() {
|
|
println("-------------------------")
|
|
for i := 0; i < ls.stack.Sp(); i++ {
|
|
print(i)
|
|
print(" ")
|
|
frame := ls.stack.At(i)
|
|
if frame == nil {
|
|
break
|
|
}
|
|
if frame.Fn.IsG {
|
|
println("IsG:", true, "Frame:", frame, "Fn:", frame.Fn)
|
|
} else {
|
|
println("IsG:", false, "Frame:", frame, "Fn:", frame.Fn, "pc:", frame.Pc)
|
|
}
|
|
}
|
|
println("-------------------------")
|
|
}
|
|
|
|
func (ls *LState) closeAllUpvalues() { // +inline-start
|
|
for cf := ls.currentFrame; cf != nil; cf = cf.Parent {
|
|
if !cf.Fn.IsG {
|
|
ls.closeUpvalues(cf.LocalBase)
|
|
}
|
|
}
|
|
} // +inline-end
|
|
|
|
func (ls *LState) raiseError(level int, format string, args ...interface{}) {
|
|
if !ls.hasErrorFunc {
|
|
ls.closeAllUpvalues()
|
|
}
|
|
message := format
|
|
if len(args) > 0 {
|
|
message = fmt.Sprintf(format, args...)
|
|
}
|
|
if level > 0 {
|
|
message = fmt.Sprintf("%v %v", ls.where(level-1, true), message)
|
|
}
|
|
ls.reg.Push(LString(message))
|
|
ls.Panic(ls)
|
|
}
|
|
|
|
func (ls *LState) findLocal(frame *callFrame, no int) string {
|
|
fn := frame.Fn
|
|
if !fn.IsG {
|
|
if name, ok := fn.LocalName(no, frame.Pc-1); ok {
|
|
return name
|
|
}
|
|
}
|
|
var top int
|
|
if ls.currentFrame == frame {
|
|
top = ls.reg.Top()
|
|
} else if frame.Idx+1 < ls.stack.Sp() {
|
|
top = ls.stack.At(frame.Idx + 1).Base
|
|
} else {
|
|
return ""
|
|
}
|
|
if top-frame.LocalBase >= no {
|
|
return "(*temporary)"
|
|
}
|
|
return ""
|
|
}
|
|
|
|
func (ls *LState) where(level int, skipg bool) string {
|
|
dbg, ok := ls.GetStack(level)
|
|
if !ok {
|
|
return ""
|
|
}
|
|
cf := dbg.frame
|
|
proto := cf.Fn.Proto
|
|
sourcename := "[G]"
|
|
if proto != nil {
|
|
sourcename = proto.SourceName
|
|
} else if skipg {
|
|
return ls.where(level+1, skipg)
|
|
}
|
|
line := ""
|
|
if proto != nil {
|
|
line = fmt.Sprintf("%v:", proto.DbgSourcePositions[cf.Pc-1])
|
|
}
|
|
return fmt.Sprintf("%v:%v", sourcename, line)
|
|
}
|
|
|
|
func (ls *LState) stackTrace(level int) string {
|
|
buf := []string{}
|
|
header := "stack traceback:"
|
|
if ls.currentFrame != nil {
|
|
i := 0
|
|
for dbg, ok := ls.GetStack(i); ok; dbg, ok = ls.GetStack(i) {
|
|
cf := dbg.frame
|
|
buf = append(buf, fmt.Sprintf("\t%v in %v", ls.Where(i), ls.formattedFrameFuncName(cf)))
|
|
if !cf.Fn.IsG && cf.TailCall > 0 {
|
|
for tc := cf.TailCall; tc > 0; tc-- {
|
|
buf = append(buf, "\t(tailcall): ?")
|
|
i++
|
|
}
|
|
}
|
|
i++
|
|
}
|
|
}
|
|
buf = append(buf, fmt.Sprintf("\t%v: %v", "[G]", "?"))
|
|
buf = buf[intMax(0, intMin(level, len(buf))):len(buf)]
|
|
if len(buf) > 20 {
|
|
newbuf := make([]string, 0, 20)
|
|
newbuf = append(newbuf, buf[0:7]...)
|
|
newbuf = append(newbuf, "\t...")
|
|
newbuf = append(newbuf, buf[len(buf)-7:len(buf)]...)
|
|
buf = newbuf
|
|
}
|
|
return fmt.Sprintf("%s\n%s", header, strings.Join(buf, "\n"))
|
|
}
|
|
|
|
func (ls *LState) formattedFrameFuncName(fr *callFrame) string {
|
|
name, ischunk := ls.frameFuncName(fr)
|
|
if ischunk {
|
|
return name
|
|
}
|
|
if name[0] != '(' && name[0] != '<' {
|
|
return fmt.Sprintf("function '%s'", name)
|
|
}
|
|
return fmt.Sprintf("function %s", name)
|
|
}
|
|
|
|
func (ls *LState) rawFrameFuncName(fr *callFrame) string {
|
|
name, _ := ls.frameFuncName(fr)
|
|
return name
|
|
}
|
|
|
|
func (ls *LState) frameFuncName(fr *callFrame) (string, bool) {
|
|
frame := fr.Parent
|
|
if frame == nil {
|
|
if ls.Parent == nil {
|
|
return "main chunk", true
|
|
} else {
|
|
return "corountine", true
|
|
}
|
|
}
|
|
if !frame.Fn.IsG {
|
|
pc := frame.Pc - 1
|
|
for _, call := range frame.Fn.Proto.DbgCalls {
|
|
if call.Pc == pc {
|
|
name := call.Name
|
|
if (name == "?" || fr.TailCall > 0) && !fr.Fn.IsG {
|
|
name = fmt.Sprintf("<%v:%v>", fr.Fn.Proto.SourceName, fr.Fn.Proto.LineDefined)
|
|
}
|
|
return name, false
|
|
}
|
|
}
|
|
}
|
|
if !fr.Fn.IsG {
|
|
return fmt.Sprintf("<%v:%v>", fr.Fn.Proto.SourceName, fr.Fn.Proto.LineDefined), false
|
|
}
|
|
return "(anonymous)", false
|
|
}
|
|
|
|
func (ls *LState) isStarted() bool {
|
|
return ls.currentFrame != nil
|
|
}
|
|
|
|
func (ls *LState) kill() {
|
|
ls.Dead = true
|
|
}
|
|
|
|
func (ls *LState) indexToReg(idx int) int {
|
|
base := ls.currentLocalBase()
|
|
if idx > 0 {
|
|
return base + idx - 1
|
|
} else if idx == 0 {
|
|
return -1
|
|
} else {
|
|
tidx := ls.reg.Top() + idx
|
|
if tidx < base {
|
|
return -1
|
|
}
|
|
return tidx
|
|
}
|
|
}
|
|
|
|
func (ls *LState) currentLocalBase() int {
|
|
base := 0
|
|
if ls.currentFrame != nil {
|
|
base = ls.currentFrame.LocalBase
|
|
}
|
|
return base
|
|
}
|
|
|
|
func (ls *LState) currentEnv() *LTable {
|
|
return ls.Env
|
|
/*
|
|
if ls.currentFrame == nil {
|
|
return ls.Env
|
|
}
|
|
return ls.currentFrame.Fn.Env
|
|
*/
|
|
}
|
|
|
|
func (ls *LState) rkValue(idx int) LValue {
|
|
/*
|
|
if OpIsK(idx) {
|
|
return ls.currentFrame.Fn.Proto.Constants[opIndexK(idx)]
|
|
}
|
|
return ls.reg.Get(ls.currentFrame.LocalBase + idx)
|
|
*/
|
|
if (idx & opBitRk) != 0 {
|
|
return ls.currentFrame.Fn.Proto.Constants[idx & ^opBitRk]
|
|
}
|
|
return ls.reg.array[ls.currentFrame.LocalBase+idx]
|
|
}
|
|
|
|
func (ls *LState) rkString(idx int) string {
|
|
if (idx & opBitRk) != 0 {
|
|
return ls.currentFrame.Fn.Proto.stringConstants[idx & ^opBitRk]
|
|
}
|
|
return string(ls.reg.array[ls.currentFrame.LocalBase+idx].(LString))
|
|
}
|
|
|
|
func (ls *LState) closeUpvalues(idx int) { // +inline-start
|
|
if ls.uvcache != nil {
|
|
var prev *Upvalue
|
|
for uv := ls.uvcache; uv != nil; uv = uv.next {
|
|
if uv.index >= idx {
|
|
if prev != nil {
|
|
prev.next = nil
|
|
} else {
|
|
ls.uvcache = nil
|
|
}
|
|
uv.Close()
|
|
}
|
|
prev = uv
|
|
}
|
|
}
|
|
} // +inline-end
|
|
|
|
func (ls *LState) findUpvalue(idx int) *Upvalue {
|
|
var prev *Upvalue
|
|
var next *Upvalue
|
|
if ls.uvcache != nil {
|
|
for uv := ls.uvcache; uv != nil; uv = uv.next {
|
|
if uv.index == idx {
|
|
return uv
|
|
}
|
|
if uv.index > idx {
|
|
next = uv
|
|
break
|
|
}
|
|
prev = uv
|
|
}
|
|
}
|
|
uv := &Upvalue{reg: ls.reg, index: idx, closed: false}
|
|
if prev != nil {
|
|
prev.next = uv
|
|
} else {
|
|
ls.uvcache = uv
|
|
}
|
|
if next != nil {
|
|
uv.next = next
|
|
}
|
|
return uv
|
|
}
|
|
|
|
func (ls *LState) metatable(lvalue LValue, rawget bool) LValue {
|
|
var metatable LValue = LNil
|
|
switch obj := lvalue.(type) {
|
|
case *LTable:
|
|
metatable = obj.Metatable
|
|
case *LUserData:
|
|
metatable = obj.Metatable
|
|
default:
|
|
if table, ok := ls.G.builtinMts[int(obj.Type())]; ok {
|
|
metatable = table
|
|
}
|
|
}
|
|
|
|
if !rawget && metatable != LNil {
|
|
oldmt := metatable
|
|
if tb, ok := metatable.(*LTable); ok {
|
|
metatable = tb.RawGetString("__metatable")
|
|
if metatable == LNil {
|
|
metatable = oldmt
|
|
}
|
|
}
|
|
}
|
|
|
|
return metatable
|
|
}
|
|
|
|
func (ls *LState) metaOp1(lvalue LValue, event string) LValue {
|
|
if mt := ls.metatable(lvalue, true); mt != LNil {
|
|
if tb, ok := mt.(*LTable); ok {
|
|
return tb.RawGetString(event)
|
|
}
|
|
}
|
|
return LNil
|
|
}
|
|
|
|
func (ls *LState) metaOp2(value1, value2 LValue, event string) LValue {
|
|
if mt := ls.metatable(value1, true); mt != LNil {
|
|
if tb, ok := mt.(*LTable); ok {
|
|
if ret := tb.RawGetString(event); ret != LNil {
|
|
return ret
|
|
}
|
|
}
|
|
}
|
|
if mt := ls.metatable(value2, true); mt != LNil {
|
|
if tb, ok := mt.(*LTable); ok {
|
|
return tb.RawGetString(event)
|
|
}
|
|
}
|
|
return LNil
|
|
}
|
|
|
|
func (ls *LState) metaCall(lvalue LValue) (*LFunction, bool) {
|
|
if fn, ok := lvalue.(*LFunction); ok {
|
|
return fn, false
|
|
}
|
|
if fn, ok := ls.metaOp1(lvalue, "__call").(*LFunction); ok {
|
|
return fn, true
|
|
}
|
|
return nil, false
|
|
}
|
|
|
|
func (ls *LState) initCallFrame(cf *callFrame) { // +inline-start
|
|
if cf.Fn.IsG {
|
|
ls.reg.SetTop(cf.LocalBase + cf.NArgs)
|
|
} else {
|
|
proto := cf.Fn.Proto
|
|
nargs := cf.NArgs
|
|
np := int(proto.NumParameters)
|
|
for i := nargs; i < np; i++ {
|
|
ls.reg.array[cf.LocalBase+i] = LNil
|
|
nargs = np
|
|
}
|
|
|
|
if (proto.IsVarArg & VarArgIsVarArg) == 0 {
|
|
if nargs < int(proto.NumUsedRegisters) {
|
|
nargs = int(proto.NumUsedRegisters)
|
|
}
|
|
for i := np; i < nargs; i++ {
|
|
ls.reg.array[cf.LocalBase+i] = LNil
|
|
}
|
|
ls.reg.top = cf.LocalBase + int(proto.NumUsedRegisters)
|
|
} else {
|
|
/* swap vararg positions:
|
|
closure
|
|
namedparam1 <- lbase
|
|
namedparam2
|
|
vararg1
|
|
vararg2
|
|
|
|
TO
|
|
|
|
closure
|
|
nil
|
|
nil
|
|
vararg1
|
|
vararg2
|
|
namedparam1 <- lbase
|
|
namedparam2
|
|
*/
|
|
nvarargs := nargs - np
|
|
if nvarargs < 0 {
|
|
nvarargs = 0
|
|
}
|
|
|
|
ls.reg.SetTop(cf.LocalBase + nargs + np)
|
|
for i := 0; i < np; i++ {
|
|
//ls.reg.Set(cf.LocalBase+nargs+i, ls.reg.Get(cf.LocalBase+i))
|
|
ls.reg.array[cf.LocalBase+nargs+i] = ls.reg.array[cf.LocalBase+i]
|
|
//ls.reg.Set(cf.LocalBase+i, LNil)
|
|
ls.reg.array[cf.LocalBase+i] = LNil
|
|
}
|
|
|
|
if CompatVarArg {
|
|
ls.reg.SetTop(cf.LocalBase + nargs + np + 1)
|
|
if (proto.IsVarArg & VarArgNeedsArg) != 0 {
|
|
argtb := newLTable(nvarargs, 0)
|
|
for i := 0; i < nvarargs; i++ {
|
|
argtb.RawSetInt(i+1, ls.reg.Get(cf.LocalBase+np+i))
|
|
}
|
|
argtb.RawSetString("n", LNumber(nvarargs))
|
|
//ls.reg.Set(cf.LocalBase+nargs+np, argtb)
|
|
ls.reg.array[cf.LocalBase+nargs+np] = argtb
|
|
} else {
|
|
ls.reg.array[cf.LocalBase+nargs+np] = LNil
|
|
}
|
|
}
|
|
cf.LocalBase += nargs
|
|
maxreg := cf.LocalBase + int(proto.NumUsedRegisters)
|
|
ls.reg.SetTop(maxreg)
|
|
}
|
|
}
|
|
} // +inline-end
|
|
|
|
func (ls *LState) pushCallFrame(cf callFrame, fn LValue, meta bool) { // +inline-start
|
|
if meta {
|
|
cf.NArgs++
|
|
ls.reg.Insert(fn, cf.LocalBase)
|
|
}
|
|
if cf.Fn == nil {
|
|
ls.RaiseError("attempt to call a non-function object")
|
|
}
|
|
if ls.stack.sp == ls.Options.CallStackSize {
|
|
ls.RaiseError("stack overflow")
|
|
}
|
|
// +inline-call ls.stack.Push cf
|
|
newcf := ls.stack.Last()
|
|
// +inline-call ls.initCallFrame newcf
|
|
ls.currentFrame = newcf
|
|
} // +inline-end
|
|
|
|
func (ls *LState) callR(nargs, nret, rbase int) {
|
|
base := ls.reg.Top() - nargs - 1
|
|
if rbase < 0 {
|
|
rbase = base
|
|
}
|
|
lv := ls.reg.Get(base)
|
|
fn, meta := ls.metaCall(lv)
|
|
ls.pushCallFrame(callFrame{
|
|
Fn: fn,
|
|
Pc: 0,
|
|
Base: base,
|
|
LocalBase: base + 1,
|
|
ReturnBase: rbase,
|
|
NArgs: nargs,
|
|
NRet: nret,
|
|
Parent: ls.currentFrame,
|
|
TailCall: 0,
|
|
}, lv, meta)
|
|
if ls.G.MainThread == nil {
|
|
ls.G.MainThread = ls
|
|
ls.G.CurrentThread = ls
|
|
ls.mainLoop(ls, nil)
|
|
} else {
|
|
ls.mainLoop(ls, ls.currentFrame)
|
|
}
|
|
if nret != MultRet {
|
|
ls.reg.SetTop(rbase + nret)
|
|
}
|
|
}
|
|
|
|
func (ls *LState) getField(obj LValue, key LValue) LValue {
|
|
curobj := obj
|
|
for i := 0; i < MaxTableGetLoop; i++ {
|
|
tb, istable := curobj.(*LTable)
|
|
if istable {
|
|
ret := tb.RawGet(key)
|
|
if ret != LNil {
|
|
return ret
|
|
}
|
|
}
|
|
metaindex := ls.metaOp1(curobj, "__index")
|
|
if metaindex == LNil {
|
|
if !istable {
|
|
ls.RaiseError("attempt to index a non-table object(%v)", curobj.Type().String())
|
|
}
|
|
return LNil
|
|
}
|
|
if metaindex.Type() == LTFunction {
|
|
ls.reg.Push(metaindex)
|
|
ls.reg.Push(curobj)
|
|
ls.reg.Push(key)
|
|
ls.Call(2, 1)
|
|
return ls.reg.Pop()
|
|
} else {
|
|
curobj = metaindex
|
|
}
|
|
}
|
|
ls.RaiseError("too many recursions in gettable")
|
|
return nil
|
|
}
|
|
|
|
func (ls *LState) getFieldString(obj LValue, key string) LValue {
|
|
curobj := obj
|
|
for i := 0; i < MaxTableGetLoop; i++ {
|
|
tb, istable := curobj.(*LTable)
|
|
if istable {
|
|
ret := tb.RawGetString(key)
|
|
if ret != LNil {
|
|
return ret
|
|
}
|
|
}
|
|
metaindex := ls.metaOp1(curobj, "__index")
|
|
if metaindex == LNil {
|
|
if !istable {
|
|
ls.RaiseError("attempt to index a non-table object(%v)", curobj.Type().String())
|
|
}
|
|
return LNil
|
|
}
|
|
if metaindex.Type() == LTFunction {
|
|
ls.reg.Push(metaindex)
|
|
ls.reg.Push(curobj)
|
|
ls.reg.Push(LString(key))
|
|
ls.Call(2, 1)
|
|
return ls.reg.Pop()
|
|
} else {
|
|
curobj = metaindex
|
|
}
|
|
}
|
|
ls.RaiseError("too many recursions in gettable")
|
|
return nil
|
|
}
|
|
|
|
func (ls *LState) setField(obj LValue, key LValue, value LValue) {
|
|
curobj := obj
|
|
for i := 0; i < MaxTableGetLoop; i++ {
|
|
tb, istable := curobj.(*LTable)
|
|
if istable {
|
|
if tb.RawGet(key) != LNil {
|
|
ls.RawSet(tb, key, value)
|
|
return
|
|
}
|
|
}
|
|
metaindex := ls.metaOp1(curobj, "__newindex")
|
|
if metaindex == LNil {
|
|
if !istable {
|
|
ls.RaiseError("attempt to index a non-table object(%v)", curobj.Type().String())
|
|
}
|
|
ls.RawSet(tb, key, value)
|
|
return
|
|
}
|
|
if metaindex.Type() == LTFunction {
|
|
ls.reg.Push(metaindex)
|
|
ls.reg.Push(curobj)
|
|
ls.reg.Push(key)
|
|
ls.reg.Push(value)
|
|
ls.Call(3, 0)
|
|
return
|
|
} else {
|
|
curobj = metaindex
|
|
}
|
|
}
|
|
ls.RaiseError("too many recursions in settable")
|
|
}
|
|
|
|
func (ls *LState) setFieldString(obj LValue, key string, value LValue) {
|
|
curobj := obj
|
|
for i := 0; i < MaxTableGetLoop; i++ {
|
|
tb, istable := curobj.(*LTable)
|
|
if istable {
|
|
if tb.RawGetString(key) != LNil {
|
|
tb.RawSetString(key, value)
|
|
return
|
|
}
|
|
}
|
|
metaindex := ls.metaOp1(curobj, "__newindex")
|
|
if metaindex == LNil {
|
|
if !istable {
|
|
ls.RaiseError("attempt to index a non-table object(%v)", curobj.Type().String())
|
|
}
|
|
tb.RawSetString(key, value)
|
|
return
|
|
}
|
|
if metaindex.Type() == LTFunction {
|
|
ls.reg.Push(metaindex)
|
|
ls.reg.Push(curobj)
|
|
ls.reg.Push(LString(key))
|
|
ls.reg.Push(value)
|
|
ls.Call(3, 0)
|
|
return
|
|
} else {
|
|
curobj = metaindex
|
|
}
|
|
}
|
|
ls.RaiseError("too many recursions in settable")
|
|
}
|
|
|
|
/* }}} */
|
|
|
|
/* api methods {{{ */
|
|
|
|
func NewState(opts ...Options) *LState {
|
|
var ls *LState
|
|
if len(opts) == 0 {
|
|
ls = newLState(Options{
|
|
CallStackSize: CallStackSize,
|
|
RegistrySize: RegistrySize,
|
|
})
|
|
ls.OpenLibs()
|
|
} else {
|
|
if opts[0].CallStackSize < 1 {
|
|
opts[0].CallStackSize = CallStackSize
|
|
}
|
|
if opts[0].RegistrySize < 128 {
|
|
opts[0].RegistrySize = RegistrySize
|
|
}
|
|
ls = newLState(opts[0])
|
|
if !opts[0].SkipOpenLibs {
|
|
ls.OpenLibs()
|
|
}
|
|
}
|
|
return ls
|
|
}
|
|
|
|
func (ls *LState) Close() {
|
|
atomic.AddInt32(&ls.stop, 1)
|
|
for _, file := range ls.G.tempFiles {
|
|
// ignore errors in these operations
|
|
file.Close()
|
|
os.Remove(file.Name())
|
|
}
|
|
}
|
|
|
|
/* registry operations {{{ */
|
|
|
|
func (ls *LState) GetTop() int {
|
|
return ls.reg.Top() - ls.currentLocalBase()
|
|
}
|
|
|
|
func (ls *LState) SetTop(idx int) {
|
|
base := ls.currentLocalBase()
|
|
newtop := ls.indexToReg(idx) + 1
|
|
if newtop < base {
|
|
ls.reg.SetTop(base)
|
|
} else {
|
|
ls.reg.SetTop(newtop)
|
|
}
|
|
}
|
|
|
|
func (ls *LState) Replace(idx int, value LValue) {
|
|
base := ls.currentLocalBase()
|
|
if idx > 0 {
|
|
reg := base + idx - 1
|
|
if reg < ls.reg.Top() {
|
|
ls.reg.Set(reg, value)
|
|
}
|
|
} else if idx == 0 {
|
|
} else if idx > RegistryIndex {
|
|
if tidx := ls.reg.Top() + idx; tidx >= base {
|
|
ls.reg.Set(tidx, value)
|
|
}
|
|
} else {
|
|
switch idx {
|
|
case RegistryIndex:
|
|
if tb, ok := value.(*LTable); ok {
|
|
ls.G.Registry = tb
|
|
} else {
|
|
ls.RaiseError("registry must be a table(%v)", value.Type().String())
|
|
}
|
|
case EnvironIndex:
|
|
if ls.currentFrame == nil {
|
|
ls.RaiseError("no calling environment")
|
|
}
|
|
if tb, ok := value.(*LTable); ok {
|
|
ls.currentFrame.Fn.Env = tb
|
|
} else {
|
|
ls.RaiseError("environment must be a table(%v)", value.Type().String())
|
|
}
|
|
case GlobalsIndex:
|
|
if tb, ok := value.(*LTable); ok {
|
|
ls.G.Global = tb
|
|
} else {
|
|
ls.RaiseError("_G must be a table(%v)", value.Type().String())
|
|
}
|
|
default:
|
|
fn := ls.currentFrame.Fn
|
|
index := GlobalsIndex - idx - 1
|
|
if index < len(fn.Upvalues) {
|
|
fn.Upvalues[index].SetValue(value)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func (ls *LState) Get(idx int) LValue {
|
|
base := ls.currentLocalBase()
|
|
if idx > 0 {
|
|
reg := base + idx - 1
|
|
if reg < ls.reg.Top() {
|
|
return ls.reg.Get(reg)
|
|
}
|
|
return LNil
|
|
} else if idx == 0 {
|
|
return LNil
|
|
} else if idx > RegistryIndex {
|
|
tidx := ls.reg.Top() + idx
|
|
if tidx < base {
|
|
return LNil
|
|
}
|
|
return ls.reg.Get(tidx)
|
|
} else {
|
|
switch idx {
|
|
case RegistryIndex:
|
|
return ls.G.Registry
|
|
case EnvironIndex:
|
|
if ls.currentFrame == nil {
|
|
return ls.Env
|
|
}
|
|
return ls.currentFrame.Fn.Env
|
|
case GlobalsIndex:
|
|
return ls.G.Global
|
|
default:
|
|
fn := ls.currentFrame.Fn
|
|
index := GlobalsIndex - idx - 1
|
|
if index < len(fn.Upvalues) {
|
|
return fn.Upvalues[index].Value()
|
|
}
|
|
return LNil
|
|
}
|
|
}
|
|
return LNil
|
|
}
|
|
|
|
func (ls *LState) Push(value LValue) {
|
|
ls.reg.Push(value)
|
|
}
|
|
|
|
func (ls *LState) Pop(n int) {
|
|
for i := 0; i < n; i++ {
|
|
if ls.GetTop() == 0 {
|
|
ls.RaiseError("register underflow")
|
|
}
|
|
ls.reg.Pop()
|
|
}
|
|
}
|
|
|
|
func (ls *LState) Insert(value LValue, index int) {
|
|
reg := ls.indexToReg(index)
|
|
top := ls.reg.Top()
|
|
if reg >= top {
|
|
ls.reg.Set(reg, value)
|
|
return
|
|
}
|
|
if reg <= ls.currentLocalBase() {
|
|
reg = ls.currentLocalBase()
|
|
}
|
|
top--
|
|
for ; top >= reg; top-- {
|
|
ls.reg.Set(top+1, ls.reg.Get(top))
|
|
}
|
|
ls.reg.Set(reg, value)
|
|
}
|
|
|
|
func (ls *LState) Remove(index int) {
|
|
reg := ls.indexToReg(index)
|
|
top := ls.reg.Top()
|
|
switch {
|
|
case reg >= top:
|
|
return
|
|
case reg < ls.currentLocalBase():
|
|
return
|
|
case reg == top-1:
|
|
ls.Pop(1)
|
|
return
|
|
}
|
|
for i := reg; i < top-1; i++ {
|
|
ls.reg.Set(i, ls.reg.Get(i+1))
|
|
}
|
|
ls.reg.SetTop(top - 1)
|
|
}
|
|
|
|
/* }}} */
|
|
|
|
/* object allocation {{{ */
|
|
|
|
func (ls *LState) NewTable() *LTable {
|
|
return newLTable(defaultArrayCap, defaultHashCap)
|
|
}
|
|
|
|
func (ls *LState) CreateTable(acap, hcap int) *LTable {
|
|
return newLTable(acap, hcap)
|
|
}
|
|
|
|
// NewThread returns a new LState that shares with the original state all global objects.
|
|
// If the original state has context.Context, the new state has a new child context of the original state and this function returns its cancel function.
|
|
func (ls *LState) NewThread() (*LState, context.CancelFunc) {
|
|
thread := newLState(ls.Options)
|
|
thread.G = ls.G
|
|
thread.Env = ls.Env
|
|
var f context.CancelFunc = nil
|
|
if ls.ctx != nil {
|
|
thread.mainLoop = mainLoopWithContext
|
|
thread.ctx, f = context.WithCancel(ls.ctx)
|
|
}
|
|
return thread, f
|
|
}
|
|
|
|
func (ls *LState) NewUserData() *LUserData {
|
|
return &LUserData{
|
|
Env: ls.currentEnv(),
|
|
Metatable: LNil,
|
|
}
|
|
}
|
|
|
|
func (ls *LState) NewFunction(fn LGFunction) *LFunction {
|
|
return newLFunctionG(fn, ls.currentEnv(), 0)
|
|
}
|
|
|
|
func (ls *LState) NewClosure(fn LGFunction, upvalues ...LValue) *LFunction {
|
|
cl := newLFunctionG(fn, ls.currentEnv(), len(upvalues))
|
|
for i, lv := range upvalues {
|
|
cl.Upvalues[i] = &Upvalue{}
|
|
cl.Upvalues[i].Close()
|
|
cl.Upvalues[i].SetValue(lv)
|
|
}
|
|
return cl
|
|
}
|
|
|
|
/* }}} */
|
|
|
|
/* toType {{{ */
|
|
|
|
func (ls *LState) ToBool(n int) bool {
|
|
return LVAsBool(ls.Get(n))
|
|
}
|
|
|
|
func (ls *LState) ToInt(n int) int {
|
|
if lv, ok := ls.Get(n).(LNumber); ok {
|
|
return int(lv)
|
|
}
|
|
if lv, ok := ls.Get(n).(LString); ok {
|
|
if num, err := parseNumber(string(lv)); err == nil {
|
|
return int(num)
|
|
}
|
|
}
|
|
return 0
|
|
}
|
|
|
|
func (ls *LState) ToInt64(n int) int64 {
|
|
if lv, ok := ls.Get(n).(LNumber); ok {
|
|
return int64(lv)
|
|
}
|
|
if lv, ok := ls.Get(n).(LString); ok {
|
|
if num, err := parseNumber(string(lv)); err == nil {
|
|
return int64(num)
|
|
}
|
|
}
|
|
return 0
|
|
}
|
|
|
|
func (ls *LState) ToNumber(n int) LNumber {
|
|
return LVAsNumber(ls.Get(n))
|
|
}
|
|
|
|
func (ls *LState) ToString(n int) string {
|
|
return LVAsString(ls.Get(n))
|
|
}
|
|
|
|
func (ls *LState) ToTable(n int) *LTable {
|
|
if lv, ok := ls.Get(n).(*LTable); ok {
|
|
return lv
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (ls *LState) ToFunction(n int) *LFunction {
|
|
if lv, ok := ls.Get(n).(*LFunction); ok {
|
|
return lv
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (ls *LState) ToUserData(n int) *LUserData {
|
|
if lv, ok := ls.Get(n).(*LUserData); ok {
|
|
return lv
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (ls *LState) ToThread(n int) *LState {
|
|
if lv, ok := ls.Get(n).(*LState); ok {
|
|
return lv
|
|
}
|
|
return nil
|
|
}
|
|
|
|
/* }}} */
|
|
|
|
/* error & debug operations {{{ */
|
|
|
|
// This function is equivalent to luaL_error( http://www.lua.org/manual/5.1/manual.html#luaL_error ).
|
|
func (ls *LState) RaiseError(format string, args ...interface{}) {
|
|
ls.raiseError(1, format, args...)
|
|
}
|
|
|
|
// This function is equivalent to lua_error( http://www.lua.org/manual/5.1/manual.html#lua_error ).
|
|
func (ls *LState) Error(lv LValue, level int) {
|
|
if str, ok := lv.(LString); ok {
|
|
ls.raiseError(level, string(str))
|
|
} else {
|
|
if !ls.hasErrorFunc {
|
|
ls.closeAllUpvalues()
|
|
}
|
|
ls.Push(lv)
|
|
ls.Panic(ls)
|
|
}
|
|
}
|
|
|
|
func (ls *LState) GetInfo(what string, dbg *Debug, fn LValue) (LValue, error) {
|
|
if !strings.HasPrefix(what, ">") {
|
|
fn = dbg.frame.Fn
|
|
} else {
|
|
what = what[1:]
|
|
}
|
|
f, ok := fn.(*LFunction)
|
|
if !ok {
|
|
return LNil, newApiErrorS(ApiErrorRun, "can not get debug info(an object in not a function)")
|
|
}
|
|
|
|
retfn := false
|
|
for _, c := range what {
|
|
switch c {
|
|
case 'f':
|
|
retfn = true
|
|
case 'S':
|
|
if dbg.frame != nil && dbg.frame.Parent == nil {
|
|
dbg.What = "main"
|
|
} else if f.IsG {
|
|
dbg.What = "G"
|
|
} else if dbg.frame != nil && dbg.frame.TailCall > 0 {
|
|
dbg.What = "tail"
|
|
} else {
|
|
dbg.What = "Lua"
|
|
}
|
|
if !f.IsG {
|
|
dbg.Source = f.Proto.SourceName
|
|
dbg.LineDefined = f.Proto.LineDefined
|
|
dbg.LastLineDefined = f.Proto.LastLineDefined
|
|
}
|
|
case 'l':
|
|
if !f.IsG && dbg.frame != nil {
|
|
if dbg.frame.Pc > 0 {
|
|
dbg.CurrentLine = f.Proto.DbgSourcePositions[dbg.frame.Pc-1]
|
|
}
|
|
} else {
|
|
dbg.CurrentLine = -1
|
|
}
|
|
case 'u':
|
|
dbg.NUpvalues = len(f.Upvalues)
|
|
case 'n':
|
|
if dbg.frame != nil {
|
|
dbg.Name = ls.rawFrameFuncName(dbg.frame)
|
|
}
|
|
default:
|
|
return LNil, newApiErrorS(ApiErrorRun, "invalid what: "+string(c))
|
|
}
|
|
}
|
|
|
|
if retfn {
|
|
return f, nil
|
|
}
|
|
return LNil, nil
|
|
|
|
}
|
|
|
|
func (ls *LState) GetStack(level int) (*Debug, bool) {
|
|
frame := ls.currentFrame
|
|
for ; level > 0 && frame != nil; frame = frame.Parent {
|
|
level--
|
|
if !frame.Fn.IsG {
|
|
level -= frame.TailCall
|
|
}
|
|
}
|
|
|
|
if level == 0 && frame != nil {
|
|
return &Debug{frame: frame}, true
|
|
} else if level < 0 && ls.stack.Sp() > 0 {
|
|
return &Debug{frame: ls.stack.At(0)}, true
|
|
}
|
|
return &Debug{}, false
|
|
}
|
|
|
|
func (ls *LState) GetLocal(dbg *Debug, no int) (string, LValue) {
|
|
frame := dbg.frame
|
|
if name := ls.findLocal(frame, no); len(name) > 0 {
|
|
return name, ls.reg.Get(frame.LocalBase + no - 1)
|
|
}
|
|
return "", LNil
|
|
}
|
|
|
|
func (ls *LState) SetLocal(dbg *Debug, no int, lv LValue) string {
|
|
frame := dbg.frame
|
|
if name := ls.findLocal(frame, no); len(name) > 0 {
|
|
ls.reg.Set(frame.LocalBase+no-1, lv)
|
|
return name
|
|
}
|
|
return ""
|
|
}
|
|
|
|
func (ls *LState) GetUpvalue(fn *LFunction, no int) (string, LValue) {
|
|
if fn.IsG {
|
|
return "", LNil
|
|
}
|
|
|
|
no--
|
|
if no >= 0 && no < len(fn.Upvalues) {
|
|
return fn.Proto.DbgUpvalues[no], fn.Upvalues[no].Value()
|
|
}
|
|
return "", LNil
|
|
}
|
|
|
|
func (ls *LState) SetUpvalue(fn *LFunction, no int, lv LValue) string {
|
|
if fn.IsG {
|
|
return ""
|
|
}
|
|
|
|
no--
|
|
if no >= 0 && no < len(fn.Upvalues) {
|
|
fn.Upvalues[no].SetValue(lv)
|
|
return fn.Proto.DbgUpvalues[no]
|
|
}
|
|
return ""
|
|
}
|
|
|
|
/* }}} */
|
|
|
|
/* env operations {{{ */
|
|
|
|
func (ls *LState) GetFEnv(obj LValue) LValue {
|
|
switch lv := obj.(type) {
|
|
case *LFunction:
|
|
return lv.Env
|
|
case *LUserData:
|
|
return lv.Env
|
|
case *LState:
|
|
return lv.Env
|
|
}
|
|
return LNil
|
|
}
|
|
|
|
func (ls *LState) SetFEnv(obj LValue, env LValue) {
|
|
tb, ok := env.(*LTable)
|
|
if !ok {
|
|
ls.RaiseError("cannot use %v as an environment", env.Type().String())
|
|
}
|
|
|
|
switch lv := obj.(type) {
|
|
case *LFunction:
|
|
lv.Env = tb
|
|
case *LUserData:
|
|
lv.Env = tb
|
|
case *LState:
|
|
lv.Env = tb
|
|
}
|
|
/* do nothing */
|
|
}
|
|
|
|
/* }}} */
|
|
|
|
/* table operations {{{ */
|
|
|
|
func (ls *LState) RawGet(tb *LTable, key LValue) LValue {
|
|
return tb.RawGet(key)
|
|
}
|
|
|
|
func (ls *LState) RawGetInt(tb *LTable, key int) LValue {
|
|
return tb.RawGetInt(key)
|
|
}
|
|
|
|
func (ls *LState) GetField(obj LValue, skey string) LValue {
|
|
return ls.getFieldString(obj, skey)
|
|
}
|
|
|
|
func (ls *LState) GetTable(obj LValue, key LValue) LValue {
|
|
return ls.getField(obj, key)
|
|
}
|
|
|
|
func (ls *LState) RawSet(tb *LTable, key LValue, value LValue) {
|
|
if n, ok := key.(LNumber); ok && math.IsNaN(float64(n)) {
|
|
ls.RaiseError("table index is NaN")
|
|
} else if key == LNil {
|
|
ls.RaiseError("table index is nil")
|
|
}
|
|
tb.RawSet(key, value)
|
|
}
|
|
|
|
func (ls *LState) RawSetInt(tb *LTable, key int, value LValue) {
|
|
tb.RawSetInt(key, value)
|
|
}
|
|
|
|
func (ls *LState) SetField(obj LValue, key string, value LValue) {
|
|
ls.setFieldString(obj, key, value)
|
|
}
|
|
|
|
func (ls *LState) SetTable(obj LValue, key LValue, value LValue) {
|
|
ls.setField(obj, key, value)
|
|
}
|
|
|
|
func (ls *LState) ForEach(tb *LTable, cb func(LValue, LValue)) {
|
|
tb.ForEach(cb)
|
|
}
|
|
|
|
func (ls *LState) GetGlobal(name string) LValue {
|
|
return ls.GetField(ls.Get(GlobalsIndex), name)
|
|
}
|
|
|
|
func (ls *LState) SetGlobal(name string, value LValue) {
|
|
ls.SetField(ls.Get(GlobalsIndex), name, value)
|
|
}
|
|
|
|
func (ls *LState) Next(tb *LTable, key LValue) (LValue, LValue) {
|
|
return tb.Next(key)
|
|
}
|
|
|
|
/* }}} */
|
|
|
|
/* unary operations {{{ */
|
|
|
|
func (ls *LState) ObjLen(v1 LValue) int {
|
|
if v1.Type() == LTString {
|
|
return len(string(v1.(LString)))
|
|
}
|
|
op := ls.metaOp1(v1, "__len")
|
|
if op.Type() == LTFunction {
|
|
ls.Push(op)
|
|
ls.Push(v1)
|
|
ls.Call(1, 1)
|
|
ret := ls.reg.Pop()
|
|
if ret.Type() == LTNumber {
|
|
return int(ret.(LNumber))
|
|
}
|
|
} else if v1.Type() == LTTable {
|
|
return v1.(*LTable).Len()
|
|
}
|
|
return 0
|
|
}
|
|
|
|
/* }}} */
|
|
|
|
/* binary operations {{{ */
|
|
|
|
func (ls *LState) Concat(values ...LValue) string {
|
|
top := ls.reg.Top()
|
|
for _, value := range values {
|
|
ls.reg.Push(value)
|
|
}
|
|
ret := stringConcat(ls, len(values), ls.reg.Top()-1)
|
|
ls.reg.SetTop(top)
|
|
return LVAsString(ret)
|
|
}
|
|
|
|
func (ls *LState) LessThan(lhs, rhs LValue) bool {
|
|
return lessThan(ls, lhs, rhs)
|
|
}
|
|
|
|
func (ls *LState) Equal(lhs, rhs LValue) bool {
|
|
return equals(ls, lhs, rhs, false)
|
|
}
|
|
|
|
func (ls *LState) RawEqual(lhs, rhs LValue) bool {
|
|
return equals(ls, lhs, rhs, true)
|
|
}
|
|
|
|
/* }}} */
|
|
|
|
/* register operations {{{ */
|
|
|
|
func (ls *LState) Register(name string, fn LGFunction) {
|
|
ls.SetGlobal(name, ls.NewFunction(fn))
|
|
}
|
|
|
|
/* }}} */
|
|
|
|
/* load and function call operations {{{ */
|
|
|
|
func (ls *LState) Load(reader io.Reader, name string) (*LFunction, error) {
|
|
chunk, err := parse.Parse(reader, name)
|
|
if err != nil {
|
|
return nil, newApiErrorE(ApiErrorSyntax, err)
|
|
}
|
|
proto, err := Compile(chunk, name)
|
|
if err != nil {
|
|
return nil, newApiErrorE(ApiErrorSyntax, err)
|
|
}
|
|
return newLFunctionL(proto, ls.currentEnv(), 0), nil
|
|
}
|
|
|
|
func (ls *LState) Call(nargs, nret int) {
|
|
ls.callR(nargs, nret, -1)
|
|
}
|
|
|
|
func (ls *LState) PCall(nargs, nret int, errfunc *LFunction) (err error) {
|
|
err = nil
|
|
sp := ls.stack.Sp()
|
|
base := ls.reg.Top() - nargs - 1
|
|
oldpanic := ls.Panic
|
|
ls.Panic = panicWithoutTraceback
|
|
if errfunc != nil {
|
|
ls.hasErrorFunc = true
|
|
}
|
|
defer func() {
|
|
ls.Panic = oldpanic
|
|
ls.hasErrorFunc = false
|
|
rcv := recover()
|
|
if rcv != nil {
|
|
if _, ok := rcv.(*ApiError); !ok {
|
|
err = newApiErrorS(ApiErrorPanic, fmt.Sprint(rcv))
|
|
if ls.Options.IncludeGoStackTrace {
|
|
buf := make([]byte, 4096)
|
|
runtime.Stack(buf, false)
|
|
err.(*ApiError).StackTrace = strings.Trim(string(buf), "\000") + "\n" + ls.stackTrace(0)
|
|
}
|
|
} else {
|
|
err = rcv.(*ApiError)
|
|
}
|
|
if errfunc != nil {
|
|
ls.Push(errfunc)
|
|
ls.Push(err.(*ApiError).Object)
|
|
ls.Panic = panicWithoutTraceback
|
|
defer func() {
|
|
ls.Panic = oldpanic
|
|
rcv := recover()
|
|
if rcv != nil {
|
|
if _, ok := rcv.(*ApiError); !ok {
|
|
err = newApiErrorS(ApiErrorPanic, fmt.Sprint(rcv))
|
|
if ls.Options.IncludeGoStackTrace {
|
|
buf := make([]byte, 4096)
|
|
runtime.Stack(buf, false)
|
|
err.(*ApiError).StackTrace = strings.Trim(string(buf), "\000") + ls.stackTrace(0)
|
|
}
|
|
} else {
|
|
err = rcv.(*ApiError)
|
|
err.(*ApiError).StackTrace = ls.stackTrace(0)
|
|
}
|
|
}
|
|
}()
|
|
ls.Call(1, 1)
|
|
err = newApiError(ApiErrorError, ls.Get(-1))
|
|
} else if len(err.(*ApiError).StackTrace) == 0 {
|
|
err.(*ApiError).StackTrace = ls.stackTrace(0)
|
|
}
|
|
ls.stack.SetSp(sp)
|
|
ls.currentFrame = ls.stack.Last()
|
|
ls.reg.SetTop(base)
|
|
}
|
|
ls.stack.SetSp(sp)
|
|
if sp == 0 {
|
|
ls.currentFrame = nil
|
|
}
|
|
}()
|
|
|
|
ls.Call(nargs, nret)
|
|
|
|
return
|
|
}
|
|
|
|
func (ls *LState) GPCall(fn LGFunction, data LValue) error {
|
|
ls.Push(newLFunctionG(fn, ls.currentEnv(), 0))
|
|
ls.Push(data)
|
|
return ls.PCall(1, MultRet, nil)
|
|
}
|
|
|
|
func (ls *LState) CallByParam(cp P, args ...LValue) error {
|
|
ls.Push(cp.Fn)
|
|
for _, arg := range args {
|
|
ls.Push(arg)
|
|
}
|
|
|
|
if cp.Protect {
|
|
return ls.PCall(len(args), cp.NRet, cp.Handler)
|
|
}
|
|
ls.Call(len(args), cp.NRet)
|
|
return nil
|
|
}
|
|
|
|
/* }}} */
|
|
|
|
/* metatable operations {{{ */
|
|
|
|
func (ls *LState) GetMetatable(obj LValue) LValue {
|
|
return ls.metatable(obj, false)
|
|
}
|
|
|
|
func (ls *LState) SetMetatable(obj LValue, mt LValue) {
|
|
switch mt.(type) {
|
|
case *LNilType, *LTable:
|
|
default:
|
|
ls.RaiseError("metatable must be a table or nil, but got %v", mt.Type().String())
|
|
}
|
|
|
|
switch v := obj.(type) {
|
|
case *LTable:
|
|
v.Metatable = mt
|
|
case *LUserData:
|
|
v.Metatable = mt
|
|
default:
|
|
ls.G.builtinMts[int(obj.Type())] = mt
|
|
}
|
|
}
|
|
|
|
/* }}} */
|
|
|
|
/* coroutine operations {{{ */
|
|
|
|
func (ls *LState) Status(th *LState) string {
|
|
status := "suspended"
|
|
if th.Dead {
|
|
status = "dead"
|
|
} else if ls.G.CurrentThread == th {
|
|
status = "running"
|
|
} else if ls.Parent == th {
|
|
status = "normal"
|
|
}
|
|
return status
|
|
}
|
|
|
|
func (ls *LState) Resume(th *LState, fn *LFunction, args ...LValue) (ResumeState, error, []LValue) {
|
|
isstarted := th.isStarted()
|
|
if !isstarted {
|
|
base := 0
|
|
th.stack.Push(callFrame{
|
|
Fn: fn,
|
|
Pc: 0,
|
|
Base: base,
|
|
LocalBase: base + 1,
|
|
ReturnBase: base,
|
|
NArgs: 0,
|
|
NRet: MultRet,
|
|
Parent: nil,
|
|
TailCall: 0,
|
|
})
|
|
}
|
|
|
|
if ls.G.CurrentThread == th {
|
|
return ResumeError, newApiErrorS(ApiErrorRun, "can not resume a running thread"), nil
|
|
}
|
|
if th.Dead {
|
|
return ResumeError, newApiErrorS(ApiErrorRun, "can not resume a dead thread"), nil
|
|
}
|
|
th.Parent = ls
|
|
ls.G.CurrentThread = th
|
|
if !isstarted {
|
|
cf := th.stack.Last()
|
|
th.currentFrame = cf
|
|
th.SetTop(0)
|
|
for _, arg := range args {
|
|
th.Push(arg)
|
|
}
|
|
cf.NArgs = len(args)
|
|
th.initCallFrame(cf)
|
|
th.Panic = panicWithoutTraceback
|
|
} else {
|
|
for _, arg := range args {
|
|
th.Push(arg)
|
|
}
|
|
}
|
|
top := ls.GetTop()
|
|
threadRun(th)
|
|
haserror := LVIsFalse(ls.Get(top + 1))
|
|
ret := make([]LValue, 0, ls.GetTop())
|
|
for idx := top + 2; idx <= ls.GetTop(); idx++ {
|
|
ret = append(ret, ls.Get(idx))
|
|
}
|
|
if len(ret) == 0 {
|
|
ret = append(ret, LNil)
|
|
}
|
|
ls.SetTop(top)
|
|
|
|
if haserror {
|
|
return ResumeError, newApiError(ApiErrorRun, ret[0]), nil
|
|
} else if th.stack.IsEmpty() {
|
|
return ResumeOK, nil, ret
|
|
}
|
|
return ResumeYield, nil, ret
|
|
}
|
|
|
|
func (ls *LState) Yield(values ...LValue) int {
|
|
ls.SetTop(0)
|
|
for _, lv := range values {
|
|
ls.Push(lv)
|
|
}
|
|
return -1
|
|
}
|
|
|
|
func (ls *LState) XMoveTo(other *LState, n int) {
|
|
if ls == other {
|
|
return
|
|
}
|
|
top := ls.GetTop()
|
|
n = intMin(n, top)
|
|
for i := n; i > 0; i-- {
|
|
other.Push(ls.Get(top - i + 1))
|
|
}
|
|
ls.SetTop(top - n)
|
|
}
|
|
|
|
/* }}} */
|
|
|
|
/* GopherLua original APIs {{{ */
|
|
|
|
// Set maximum memory size. This function can only be called from the main thread.
|
|
func (ls *LState) SetMx(mx int) {
|
|
if ls.Parent != nil {
|
|
ls.RaiseError("sub threads are not allowed to set a memory limit")
|
|
}
|
|
go func() {
|
|
limit := uint64(mx * 1024 * 1024) //MB
|
|
var s runtime.MemStats
|
|
for ls.stop == 0 {
|
|
runtime.ReadMemStats(&s)
|
|
if s.Alloc >= limit {
|
|
fmt.Println("out of memory")
|
|
os.Exit(3)
|
|
}
|
|
time.Sleep(100 * time.Millisecond)
|
|
}
|
|
}()
|
|
}
|
|
|
|
// SetContext set a context ctx to this LState. The provided ctx must be non-nil.
|
|
func (ls *LState) SetContext(ctx context.Context) {
|
|
ls.mainLoop = mainLoopWithContext
|
|
ls.ctx = ctx
|
|
}
|
|
|
|
// Context returns the LState's context. To change the context, use WithContext.
|
|
func (ls *LState) Context() context.Context {
|
|
return ls.ctx
|
|
}
|
|
|
|
// RemoveContext removes the context associated with this LState and returns this context.
|
|
func (ls *LState) RemoveContext() context.Context {
|
|
oldctx := ls.ctx
|
|
ls.mainLoop = mainLoop
|
|
ls.ctx = nil
|
|
return oldctx
|
|
}
|
|
|
|
// Converts the Lua value at the given acceptable index to the chan LValue.
|
|
func (ls *LState) ToChannel(n int) chan LValue {
|
|
if lv, ok := ls.Get(n).(LChannel); ok {
|
|
return (chan LValue)(lv)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
/* }}} */
|
|
|
|
/* }}} */
|
|
|
|
//
|