12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049 |
- package lua
- import (
- "fmt"
- "math"
- "strings"
- )
- func mainLoop(L *LState, baseframe *callFrame) {
- var inst uint32
- var cf *callFrame
- if L.stack.IsEmpty() {
- return
- }
- L.currentFrame = L.stack.Last()
- if L.currentFrame.Fn.IsG {
- callGFunction(L, false)
- return
- }
- for {
- cf = L.currentFrame
- inst = cf.Fn.Proto.Code[cf.Pc]
- cf.Pc++
- if jumpTable[int(inst>>26)](L, inst, baseframe) == 1 {
- return
- }
- }
- }
- func mainLoopWithContext(L *LState, baseframe *callFrame) {
- var inst uint32
- var cf *callFrame
- if L.stack.IsEmpty() {
- return
- }
- L.currentFrame = L.stack.Last()
- if L.currentFrame.Fn.IsG {
- callGFunction(L, false)
- return
- }
- for {
- cf = L.currentFrame
- inst = cf.Fn.Proto.Code[cf.Pc]
- cf.Pc++
- select {
- case <-L.ctx.Done():
- L.RaiseError(L.ctx.Err().Error())
- return
- default:
- if jumpTable[int(inst>>26)](L, inst, baseframe) == 1 {
- return
- }
- }
- }
- }
- // regv is the first target register to copy the return values to.
- // It can be reg.top, indicating that the copied values are going into new registers, or it can be below reg.top
- // Indicating that the values should be within the existing registers.
- // b is the available number of return values + 1.
- // n is the desired number of return values.
- // If n more than the available return values then the extra values are set to nil.
- // When this function returns the top of the registry will be set to regv+n.
- func copyReturnValues(L *LState, regv, start, n, b int) { // +inline-start
- if b == 1 {
- // +inline-call L.reg.FillNil regv n
- } else {
- // +inline-call L.reg.CopyRange regv start -1 n
- if b > 1 && n > (b-1) {
- // +inline-call L.reg.FillNil regv+b-1 n-(b-1)
- }
- }
- } // +inline-end
- func switchToParentThread(L *LState, nargs int, haserror bool, kill bool) {
- parent := L.Parent
- if parent == nil {
- L.RaiseError("can not yield from outside of a coroutine")
- }
- L.G.CurrentThread = parent
- L.Parent = nil
- if !L.wrapped {
- if haserror {
- parent.Push(LFalse)
- } else {
- parent.Push(LTrue)
- }
- }
- L.XMoveTo(parent, nargs)
- L.stack.Pop()
- offset := L.currentFrame.LocalBase - L.currentFrame.ReturnBase
- L.currentFrame = L.stack.Last()
- L.reg.SetTop(L.reg.Top() - offset) // remove 'yield' function(including tailcalled functions)
- if kill {
- L.kill()
- }
- }
- func callGFunction(L *LState, tailcall bool) bool {
- frame := L.currentFrame
- gfnret := frame.Fn.GFunction(L)
- if tailcall {
- L.currentFrame = L.RemoveCallerFrame()
- }
- if gfnret < 0 {
- switchToParentThread(L, L.GetTop(), false, false)
- return true
- }
- wantret := frame.NRet
- if wantret == MultRet {
- wantret = gfnret
- }
- if tailcall && L.Parent != nil && L.stack.Sp() == 1 {
- switchToParentThread(L, wantret, false, true)
- return true
- }
- // +inline-call L.reg.CopyRange frame.ReturnBase L.reg.Top()-gfnret -1 wantret
- L.stack.Pop()
- L.currentFrame = L.stack.Last()
- return false
- }
- func threadRun(L *LState) {
- if L.stack.IsEmpty() {
- return
- }
- defer func() {
- if rcv := recover(); rcv != nil {
- var lv LValue
- if v, ok := rcv.(*ApiError); ok {
- lv = v.Object
- } else {
- lv = LString(fmt.Sprint(rcv))
- }
- if parent := L.Parent; parent != nil {
- if L.wrapped {
- L.Push(lv)
- parent.Panic(L)
- } else {
- L.SetTop(0)
- L.Push(lv)
- switchToParentThread(L, 1, true, true)
- }
- } else {
- panic(rcv)
- }
- }
- }()
- L.mainLoop(L, nil)
- }
- type instFunc func(*LState, uint32, *callFrame) int
- var jumpTable [opCodeMax + 1]instFunc
- func init() {
- jumpTable = [opCodeMax + 1]instFunc{
- func(L *LState, inst uint32, baseframe *callFrame) int { //OP_MOVE
- reg := L.reg
- cf := L.currentFrame
- lbase := cf.LocalBase
- A := int(inst>>18) & 0xff //GETA
- RA := lbase + A
- B := int(inst & 0x1ff) //GETB
- v := reg.Get(lbase + B)
- // +inline-call reg.Set RA v
- return 0
- },
- func(L *LState, inst uint32, baseframe *callFrame) int { //OP_MOVEN
- reg := L.reg
- cf := L.currentFrame
- lbase := cf.LocalBase
- A := int(inst>>18) & 0xff //GETA
- B := int(inst & 0x1ff) //GETB
- C := int(inst>>9) & 0x1ff //GETC
- v := reg.Get(lbase + B)
- // +inline-call reg.Set lbase+A v
- code := cf.Fn.Proto.Code
- pc := cf.Pc
- for i := 0; i < C; i++ {
- inst = code[pc]
- pc++
- A = int(inst>>18) & 0xff //GETA
- B = int(inst & 0x1ff) //GETB
- v := reg.Get(lbase + B)
- // +inline-call reg.Set lbase+A v
- }
- cf.Pc = pc
- return 0
- },
- func(L *LState, inst uint32, baseframe *callFrame) int { //OP_LOADK
- reg := L.reg
- cf := L.currentFrame
- lbase := cf.LocalBase
- A := int(inst>>18) & 0xff //GETA
- RA := lbase + A
- Bx := int(inst & 0x3ffff) //GETBX
- v := cf.Fn.Proto.Constants[Bx]
- // +inline-call reg.Set RA v
- return 0
- },
- func(L *LState, inst uint32, baseframe *callFrame) int { //OP_LOADBOOL
- reg := L.reg
- cf := L.currentFrame
- lbase := cf.LocalBase
- A := int(inst>>18) & 0xff //GETA
- RA := lbase + A
- B := int(inst & 0x1ff) //GETB
- C := int(inst>>9) & 0x1ff //GETC
- if B != 0 {
- // +inline-call reg.Set RA LTrue
- } else {
- // +inline-call reg.Set RA LFalse
- }
- if C != 0 {
- cf.Pc++
- }
- return 0
- },
- func(L *LState, inst uint32, baseframe *callFrame) int { //OP_LOADNIL
- reg := L.reg
- cf := L.currentFrame
- lbase := cf.LocalBase
- A := int(inst>>18) & 0xff //GETA
- RA := lbase + A
- B := int(inst & 0x1ff) //GETB
- for i := RA; i <= lbase+B; i++ {
- // +inline-call reg.Set i LNil
- }
- return 0
- },
- func(L *LState, inst uint32, baseframe *callFrame) int { //OP_GETUPVAL
- reg := L.reg
- cf := L.currentFrame
- lbase := cf.LocalBase
- A := int(inst>>18) & 0xff //GETA
- RA := lbase + A
- B := int(inst & 0x1ff) //GETB
- v := cf.Fn.Upvalues[B].Value()
- // +inline-call reg.Set RA v
- return 0
- },
- func(L *LState, inst uint32, baseframe *callFrame) int { //OP_GETGLOBAL
- reg := L.reg
- cf := L.currentFrame
- lbase := cf.LocalBase
- A := int(inst>>18) & 0xff //GETA
- RA := lbase + A
- Bx := int(inst & 0x3ffff) //GETBX
- //reg.Set(RA, L.getField(cf.Fn.Env, cf.Fn.Proto.Constants[Bx]))
- v := L.getFieldString(cf.Fn.Env, cf.Fn.Proto.stringConstants[Bx])
- // +inline-call reg.Set RA v
- return 0
- },
- func(L *LState, inst uint32, baseframe *callFrame) int { //OP_GETTABLE
- reg := L.reg
- cf := L.currentFrame
- lbase := cf.LocalBase
- A := int(inst>>18) & 0xff //GETA
- RA := lbase + A
- B := int(inst & 0x1ff) //GETB
- C := int(inst>>9) & 0x1ff //GETC
- v := L.getField(reg.Get(lbase+B), L.rkValue(C))
- // +inline-call reg.Set RA v
- return 0
- },
- func(L *LState, inst uint32, baseframe *callFrame) int { //OP_GETTABLEKS
- reg := L.reg
- cf := L.currentFrame
- lbase := cf.LocalBase
- A := int(inst>>18) & 0xff //GETA
- RA := lbase + A
- B := int(inst & 0x1ff) //GETB
- C := int(inst>>9) & 0x1ff //GETC
- v := L.getFieldString(reg.Get(lbase+B), L.rkString(C))
- // +inline-call reg.Set RA v
- return 0
- },
- func(L *LState, inst uint32, baseframe *callFrame) int { //OP_SETGLOBAL
- reg := L.reg
- cf := L.currentFrame
- lbase := cf.LocalBase
- A := int(inst>>18) & 0xff //GETA
- RA := lbase + A
- Bx := int(inst & 0x3ffff) //GETBX
- //L.setField(cf.Fn.Env, cf.Fn.Proto.Constants[Bx], reg.Get(RA))
- L.setFieldString(cf.Fn.Env, cf.Fn.Proto.stringConstants[Bx], reg.Get(RA))
- return 0
- },
- func(L *LState, inst uint32, baseframe *callFrame) int { //OP_SETUPVAL
- reg := L.reg
- cf := L.currentFrame
- lbase := cf.LocalBase
- A := int(inst>>18) & 0xff //GETA
- RA := lbase + A
- B := int(inst & 0x1ff) //GETB
- cf.Fn.Upvalues[B].SetValue(reg.Get(RA))
- return 0
- },
- func(L *LState, inst uint32, baseframe *callFrame) int { //OP_SETTABLE
- reg := L.reg
- cf := L.currentFrame
- lbase := cf.LocalBase
- A := int(inst>>18) & 0xff //GETA
- RA := lbase + A
- B := int(inst & 0x1ff) //GETB
- C := int(inst>>9) & 0x1ff //GETC
- L.setField(reg.Get(RA), L.rkValue(B), L.rkValue(C))
- return 0
- },
- func(L *LState, inst uint32, baseframe *callFrame) int { //OP_SETTABLEKS
- reg := L.reg
- cf := L.currentFrame
- lbase := cf.LocalBase
- A := int(inst>>18) & 0xff //GETA
- RA := lbase + A
- B := int(inst & 0x1ff) //GETB
- C := int(inst>>9) & 0x1ff //GETC
- L.setFieldString(reg.Get(RA), L.rkString(B), L.rkValue(C))
- return 0
- },
- func(L *LState, inst uint32, baseframe *callFrame) int { //OP_NEWTABLE
- reg := L.reg
- cf := L.currentFrame
- lbase := cf.LocalBase
- A := int(inst>>18) & 0xff //GETA
- RA := lbase + A
- B := int(inst & 0x1ff) //GETB
- C := int(inst>>9) & 0x1ff //GETC
- v := newLTable(B, C)
- // +inline-call reg.Set RA v
- return 0
- },
- func(L *LState, inst uint32, baseframe *callFrame) int { //OP_SELF
- reg := L.reg
- cf := L.currentFrame
- lbase := cf.LocalBase
- A := int(inst>>18) & 0xff //GETA
- RA := lbase + A
- B := int(inst & 0x1ff) //GETB
- C := int(inst>>9) & 0x1ff //GETC
- selfobj := reg.Get(lbase + B)
- v := L.getFieldString(selfobj, L.rkString(C))
- // +inline-call reg.Set RA v
- // +inline-call reg.Set RA+1 selfobj
- return 0
- },
- opArith, // OP_ADD
- opArith, // OP_SUB
- opArith, // OP_MUL
- opArith, // OP_DIV
- opArith, // OP_MOD
- opArith, // OP_POW
- func(L *LState, inst uint32, baseframe *callFrame) int { //OP_UNM
- reg := L.reg
- cf := L.currentFrame
- lbase := cf.LocalBase
- A := int(inst>>18) & 0xff //GETA
- RA := lbase + A
- B := int(inst & 0x1ff) //GETB
- unaryv := L.rkValue(B)
- if nm, ok := unaryv.(LNumber); ok {
- // +inline-call reg.Set RA -nm
- } else {
- op := L.metaOp1(unaryv, "__unm")
- if op.Type() == LTFunction {
- reg.Push(op)
- reg.Push(unaryv)
- L.Call(1, 1)
- // +inline-call reg.Set RA reg.Pop()
- } else if str, ok1 := unaryv.(LString); ok1 {
- if num, err := parseNumber(string(str)); err == nil {
- // +inline-call reg.Set RA -num
- } else {
- L.RaiseError("__unm undefined")
- }
- } else {
- L.RaiseError("__unm undefined")
- }
- }
- return 0
- },
- func(L *LState, inst uint32, baseframe *callFrame) int { //OP_NOT
- reg := L.reg
- cf := L.currentFrame
- lbase := cf.LocalBase
- A := int(inst>>18) & 0xff //GETA
- RA := lbase + A
- B := int(inst & 0x1ff) //GETB
- if LVIsFalse(reg.Get(lbase + B)) {
- // +inline-call reg.Set RA LTrue
- } else {
- // +inline-call reg.Set RA LFalse
- }
- return 0
- },
- func(L *LState, inst uint32, baseframe *callFrame) int { //OP_LEN
- reg := L.reg
- cf := L.currentFrame
- lbase := cf.LocalBase
- A := int(inst>>18) & 0xff //GETA
- RA := lbase + A
- B := int(inst & 0x1ff) //GETB
- switch lv := L.rkValue(B).(type) {
- case LString:
- // +inline-call reg.SetNumber RA LNumber(len(lv))
- default:
- op := L.metaOp1(lv, "__len")
- if op.Type() == LTFunction {
- reg.Push(op)
- reg.Push(lv)
- L.Call(1, 1)
- ret := reg.Pop()
- if ret.Type() == LTNumber {
- v, _ := ret.(LNumber)
- // +inline-call reg.SetNumber RA v
- } else {
- // +inline-call reg.Set RA ret
- }
- } else if lv.Type() == LTTable {
- // +inline-call reg.SetNumber RA LNumber(lv.(*LTable).Len())
- } else {
- L.RaiseError("__len undefined")
- }
- }
- return 0
- },
- func(L *LState, inst uint32, baseframe *callFrame) int { //OP_CONCAT
- reg := L.reg
- cf := L.currentFrame
- lbase := cf.LocalBase
- A := int(inst>>18) & 0xff //GETA
- RA := lbase + A
- B := int(inst & 0x1ff) //GETB
- C := int(inst>>9) & 0x1ff //GETC
- RC := lbase + C
- RB := lbase + B
- v := stringConcat(L, RC-RB+1, RC)
- // +inline-call reg.Set RA v
- return 0
- },
- func(L *LState, inst uint32, baseframe *callFrame) int { //OP_JMP
- cf := L.currentFrame
- Sbx := int(inst&0x3ffff) - opMaxArgSbx //GETSBX
- cf.Pc += Sbx
- return 0
- },
- func(L *LState, inst uint32, baseframe *callFrame) int { //OP_EQ
- cf := L.currentFrame
- A := int(inst>>18) & 0xff //GETA
- B := int(inst & 0x1ff) //GETB
- C := int(inst>>9) & 0x1ff //GETC
- ret := equals(L, L.rkValue(B), L.rkValue(C), false)
- v := 1
- if ret {
- v = 0
- }
- if v == A {
- cf.Pc++
- }
- return 0
- },
- func(L *LState, inst uint32, baseframe *callFrame) int { //OP_LT
- cf := L.currentFrame
- A := int(inst>>18) & 0xff //GETA
- B := int(inst & 0x1ff) //GETB
- C := int(inst>>9) & 0x1ff //GETC
- ret := lessThan(L, L.rkValue(B), L.rkValue(C))
- v := 1
- if ret {
- v = 0
- }
- if v == A {
- cf.Pc++
- }
- return 0
- },
- func(L *LState, inst uint32, baseframe *callFrame) int { //OP_LE
- cf := L.currentFrame
- A := int(inst>>18) & 0xff //GETA
- B := int(inst & 0x1ff) //GETB
- C := int(inst>>9) & 0x1ff //GETC
- lhs := L.rkValue(B)
- rhs := L.rkValue(C)
- ret := false
- if v1, ok1 := lhs.(LNumber); ok1 {
- if v2, ok2 := rhs.(LNumber); ok2 {
- ret = v1 <= v2
- } else {
- L.RaiseError("attempt to compare %v with %v", lhs.Type().String(), rhs.Type().String())
- }
- } else {
- if lhs.Type() != rhs.Type() {
- L.RaiseError("attempt to compare %v with %v", lhs.Type().String(), rhs.Type().String())
- }
- switch lhs.Type() {
- case LTString:
- ret = strCmp(string(lhs.(LString)), string(rhs.(LString))) <= 0
- default:
- switch objectRational(L, lhs, rhs, "__le") {
- case 1:
- ret = true
- case 0:
- ret = false
- default:
- ret = !objectRationalWithError(L, rhs, lhs, "__lt")
- }
- }
- }
- v := 1
- if ret {
- v = 0
- }
- if v == A {
- cf.Pc++
- }
- return 0
- },
- func(L *LState, inst uint32, baseframe *callFrame) int { //OP_TEST
- reg := L.reg
- cf := L.currentFrame
- lbase := cf.LocalBase
- A := int(inst>>18) & 0xff //GETA
- RA := lbase + A
- C := int(inst>>9) & 0x1ff //GETC
- if LVAsBool(reg.Get(RA)) == (C == 0) {
- cf.Pc++
- }
- return 0
- },
- func(L *LState, inst uint32, baseframe *callFrame) int { //OP_TESTSET
- reg := L.reg
- cf := L.currentFrame
- lbase := cf.LocalBase
- A := int(inst>>18) & 0xff //GETA
- RA := lbase + A
- B := int(inst & 0x1ff) //GETB
- C := int(inst>>9) & 0x1ff //GETC
- if value := reg.Get(lbase + B); LVAsBool(value) != (C == 0) {
- // +inline-call reg.Set RA value
- } else {
- cf.Pc++
- }
- return 0
- },
- func(L *LState, inst uint32, baseframe *callFrame) int { //OP_CALL
- reg := L.reg
- cf := L.currentFrame
- lbase := cf.LocalBase
- A := int(inst>>18) & 0xff //GETA
- RA := lbase + A
- B := int(inst & 0x1ff) //GETB
- C := int(inst>>9) & 0x1ff //GETC
- nargs := B - 1
- if B == 0 {
- nargs = reg.Top() - (RA + 1)
- }
- lv := reg.Get(RA)
- nret := C - 1
- var callable *LFunction
- var meta bool
- if fn, ok := lv.(*LFunction); ok {
- callable = fn
- meta = false
- } else {
- callable, meta = L.metaCall(lv)
- }
- // +inline-call L.pushCallFrame callFrame{Fn:callable,Pc:0,Base:RA,LocalBase:RA+1,ReturnBase:RA,NArgs:nargs,NRet:nret,Parent:cf,TailCall:0} lv meta
- if callable.IsG && callGFunction(L, false) {
- return 1
- }
- return 0
- },
- func(L *LState, inst uint32, baseframe *callFrame) int { //OP_TAILCALL
- reg := L.reg
- cf := L.currentFrame
- lbase := cf.LocalBase
- A := int(inst>>18) & 0xff //GETA
- RA := lbase + A
- B := int(inst & 0x1ff) //GETB
- nargs := B - 1
- if B == 0 {
- nargs = reg.Top() - (RA + 1)
- }
- lv := reg.Get(RA)
- var callable *LFunction
- var meta bool
- if fn, ok := lv.(*LFunction); ok {
- callable = fn
- meta = false
- } else {
- callable, meta = L.metaCall(lv)
- }
- if callable == nil {
- L.RaiseError("attempt to call a non-function object")
- }
- // +inline-call L.closeUpvalues lbase
- if callable.IsG {
- luaframe := cf
- L.pushCallFrame(callFrame{
- Fn: callable,
- Pc: 0,
- Base: RA,
- LocalBase: RA + 1,
- ReturnBase: cf.ReturnBase,
- NArgs: nargs,
- NRet: cf.NRet,
- Parent: cf,
- TailCall: 0,
- }, lv, meta)
- if callGFunction(L, true) {
- return 1
- }
- if L.currentFrame == nil || L.currentFrame.Fn.IsG || luaframe == baseframe {
- return 1
- }
- } else {
- base := cf.Base
- cf.Fn = callable
- cf.Pc = 0
- cf.Base = RA
- cf.LocalBase = RA + 1
- cf.ReturnBase = cf.ReturnBase
- cf.NArgs = nargs
- cf.NRet = cf.NRet
- cf.TailCall++
- lbase := cf.LocalBase
- if meta {
- cf.NArgs++
- L.reg.Insert(lv, cf.LocalBase)
- }
- // +inline-call L.initCallFrame cf
- // +inline-call L.reg.CopyRange base RA -1 reg.Top()-RA-1
- cf.Base = base
- cf.LocalBase = base + (cf.LocalBase - lbase + 1)
- }
- return 0
- },
- func(L *LState, inst uint32, baseframe *callFrame) int { //OP_RETURN
- reg := L.reg
- cf := L.currentFrame
- lbase := cf.LocalBase
- A := int(inst>>18) & 0xff //GETA
- RA := lbase + A
- B := int(inst & 0x1ff) //GETB
- // +inline-call L.closeUpvalues lbase
- nret := B - 1
- if B == 0 {
- nret = reg.Top() - RA
- }
- n := cf.NRet
- if cf.NRet == MultRet {
- n = nret
- }
- if L.Parent != nil && L.stack.Sp() == 1 {
- // +inline-call copyReturnValues L reg.Top() RA n B
- switchToParentThread(L, n, false, true)
- return 1
- }
- islast := baseframe == L.stack.Pop() || L.stack.IsEmpty()
- // +inline-call copyReturnValues L cf.ReturnBase RA n B
- L.currentFrame = L.stack.Last()
- if islast || L.currentFrame == nil || L.currentFrame.Fn.IsG {
- return 1
- }
- return 0
- },
- func(L *LState, inst uint32, baseframe *callFrame) int { //OP_FORLOOP
- reg := L.reg
- cf := L.currentFrame
- lbase := cf.LocalBase
- A := int(inst>>18) & 0xff //GETA
- RA := lbase + A
- if init, ok1 := reg.Get(RA).(LNumber); ok1 {
- if limit, ok2 := reg.Get(RA + 1).(LNumber); ok2 {
- if step, ok3 := reg.Get(RA + 2).(LNumber); ok3 {
- init += step
- v := LNumber(init)
- // +inline-call reg.SetNumber RA v
- if (step > 0 && init <= limit) || (step <= 0 && init >= limit) {
- Sbx := int(inst&0x3ffff) - opMaxArgSbx //GETSBX
- cf.Pc += Sbx
- // +inline-call reg.SetNumber RA+3 v
- } else {
- // +inline-call reg.SetTop RA+1
- }
- } else {
- L.RaiseError("for statement step must be a number")
- }
- } else {
- L.RaiseError("for statement limit must be a number")
- }
- } else {
- L.RaiseError("for statement init must be a number")
- }
- return 0
- },
- func(L *LState, inst uint32, baseframe *callFrame) int { //OP_FORPREP
- reg := L.reg
- cf := L.currentFrame
- lbase := cf.LocalBase
- A := int(inst>>18) & 0xff //GETA
- RA := lbase + A
- Sbx := int(inst&0x3ffff) - opMaxArgSbx //GETSBX
- if init, ok1 := reg.Get(RA).(LNumber); ok1 {
- if step, ok2 := reg.Get(RA + 2).(LNumber); ok2 {
- // +inline-call reg.SetNumber RA LNumber(init-step)
- } else {
- L.RaiseError("for statement step must be a number")
- }
- } else {
- L.RaiseError("for statement init must be a number")
- }
- cf.Pc += Sbx
- return 0
- },
- func(L *LState, inst uint32, baseframe *callFrame) int { //OP_TFORLOOP
- reg := L.reg
- cf := L.currentFrame
- lbase := cf.LocalBase
- A := int(inst>>18) & 0xff //GETA
- RA := lbase + A
- C := int(inst>>9) & 0x1ff //GETC
- nret := C
- // +inline-call reg.SetTop RA+3+2
- // +inline-call reg.Set RA+3+2 reg.Get(RA+2)
- // +inline-call reg.Set RA+3+1 reg.Get(RA+1)
- // +inline-call reg.Set RA+3 reg.Get(RA)
- L.callR(2, nret, RA+3)
- if value := reg.Get(RA + 3); value != LNil {
- // +inline-call reg.Set RA+2 value
- pc := cf.Fn.Proto.Code[cf.Pc]
- cf.Pc += int(pc&0x3ffff) - opMaxArgSbx
- }
- cf.Pc++
- return 0
- },
- func(L *LState, inst uint32, baseframe *callFrame) int { //OP_SETLIST
- reg := L.reg
- cf := L.currentFrame
- lbase := cf.LocalBase
- A := int(inst>>18) & 0xff //GETA
- RA := lbase + A
- B := int(inst & 0x1ff) //GETB
- C := int(inst>>9) & 0x1ff //GETC
- if C == 0 {
- C = int(cf.Fn.Proto.Code[cf.Pc])
- cf.Pc++
- }
- offset := (C - 1) * FieldsPerFlush
- table := reg.Get(RA).(*LTable)
- nelem := B
- if B == 0 {
- nelem = reg.Top() - RA - 1
- }
- for i := 1; i <= nelem; i++ {
- table.RawSetInt(offset+i, reg.Get(RA+i))
- }
- return 0
- },
- func(L *LState, inst uint32, baseframe *callFrame) int { //OP_CLOSE
- cf := L.currentFrame
- lbase := cf.LocalBase
- A := int(inst>>18) & 0xff //GETA
- RA := lbase + A
- // +inline-call L.closeUpvalues RA
- return 0
- },
- func(L *LState, inst uint32, baseframe *callFrame) int { //OP_CLOSURE
- reg := L.reg
- cf := L.currentFrame
- lbase := cf.LocalBase
- A := int(inst>>18) & 0xff //GETA
- RA := lbase + A
- Bx := int(inst & 0x3ffff) //GETBX
- proto := cf.Fn.Proto.FunctionPrototypes[Bx]
- closure := newLFunctionL(proto, cf.Fn.Env, int(proto.NumUpvalues))
- // +inline-call reg.Set RA closure
- for i := 0; i < int(proto.NumUpvalues); i++ {
- inst = cf.Fn.Proto.Code[cf.Pc]
- cf.Pc++
- B := opGetArgB(inst)
- switch opGetOpCode(inst) {
- case OP_MOVE:
- closure.Upvalues[i] = L.findUpvalue(lbase + B)
- case OP_GETUPVAL:
- closure.Upvalues[i] = cf.Fn.Upvalues[B]
- }
- }
- return 0
- },
- func(L *LState, inst uint32, baseframe *callFrame) int { //OP_VARARG
- reg := L.reg
- cf := L.currentFrame
- lbase := cf.LocalBase
- A := int(inst>>18) & 0xff //GETA
- RA := lbase + A
- B := int(inst & 0x1ff) //GETB
- nparams := int(cf.Fn.Proto.NumParameters)
- nvarargs := cf.NArgs - nparams
- if nvarargs < 0 {
- nvarargs = 0
- }
- nwant := B - 1
- if B == 0 {
- nwant = nvarargs
- }
- // +inline-call reg.CopyRange RA cf.Base+nparams+1 cf.LocalBase nwant
- return 0
- },
- func(L *LState, inst uint32, baseframe *callFrame) int { //OP_NOP
- return 0
- },
- }
- }
- func opArith(L *LState, inst uint32, baseframe *callFrame) int { //OP_ADD, OP_SUB, OP_MUL, OP_DIV, OP_MOD, OP_POW
- reg := L.reg
- cf := L.currentFrame
- lbase := cf.LocalBase
- A := int(inst>>18) & 0xff //GETA
- RA := lbase + A
- opcode := int(inst >> 26) //GETOPCODE
- B := int(inst & 0x1ff) //GETB
- C := int(inst>>9) & 0x1ff //GETC
- lhs := L.rkValue(B)
- rhs := L.rkValue(C)
- v1, ok1 := lhs.(LNumber)
- v2, ok2 := rhs.(LNumber)
- if ok1 && ok2 {
- v := numberArith(L, opcode, LNumber(v1), LNumber(v2))
- // +inline-call reg.SetNumber RA v
- } else {
- v := objectArith(L, opcode, lhs, rhs)
- // +inline-call reg.Set RA v
- }
- return 0
- }
- func luaModulo(lhs, rhs LNumber) LNumber {
- flhs := float64(lhs)
- frhs := float64(rhs)
- v := math.Mod(flhs, frhs)
- if frhs > 0 && v < 0 || frhs < 0 && v > 0 {
- v += frhs
- }
- return LNumber(v)
- }
- func numberArith(L *LState, opcode int, lhs, rhs LNumber) LNumber {
- switch opcode {
- case OP_ADD:
- return lhs + rhs
- case OP_SUB:
- return lhs - rhs
- case OP_MUL:
- return lhs * rhs
- case OP_DIV:
- return lhs / rhs
- case OP_MOD:
- return luaModulo(lhs, rhs)
- case OP_POW:
- flhs := float64(lhs)
- frhs := float64(rhs)
- return LNumber(math.Pow(flhs, frhs))
- }
- panic("should not reach here")
- return LNumber(0)
- }
- func objectArith(L *LState, opcode int, lhs, rhs LValue) LValue {
- event := ""
- switch opcode {
- case OP_ADD:
- event = "__add"
- case OP_SUB:
- event = "__sub"
- case OP_MUL:
- event = "__mul"
- case OP_DIV:
- event = "__div"
- case OP_MOD:
- event = "__mod"
- case OP_POW:
- event = "__pow"
- }
- op := L.metaOp2(lhs, rhs, event)
- if _, ok := op.(*LFunction); ok {
- L.reg.Push(op)
- L.reg.Push(lhs)
- L.reg.Push(rhs)
- L.Call(2, 1)
- return L.reg.Pop()
- }
- if str, ok := lhs.(LString); ok {
- if lnum, err := parseNumber(string(str)); err == nil {
- lhs = lnum
- }
- }
- if str, ok := rhs.(LString); ok {
- if rnum, err := parseNumber(string(str)); err == nil {
- rhs = rnum
- }
- }
- if v1, ok1 := lhs.(LNumber); ok1 {
- if v2, ok2 := rhs.(LNumber); ok2 {
- return numberArith(L, opcode, LNumber(v1), LNumber(v2))
- }
- }
- L.RaiseError(fmt.Sprintf("cannot perform %v operation between %v and %v",
- strings.TrimLeft(event, "_"), lhs.Type().String(), rhs.Type().String()))
- return LNil
- }
- func stringConcat(L *LState, total, last int) LValue {
- rhs := L.reg.Get(last)
- total--
- for i := last - 1; total > 0; {
- lhs := L.reg.Get(i)
- if !(LVCanConvToString(lhs) && LVCanConvToString(rhs)) {
- op := L.metaOp2(lhs, rhs, "__concat")
- if op.Type() == LTFunction {
- L.reg.Push(op)
- L.reg.Push(lhs)
- L.reg.Push(rhs)
- L.Call(2, 1)
- rhs = L.reg.Pop()
- total--
- i--
- } else {
- L.RaiseError("cannot perform concat operation between %v and %v", lhs.Type().String(), rhs.Type().String())
- return LNil
- }
- } else {
- buf := make([]string, total+1)
- buf[total] = LVAsString(rhs)
- for total > 0 {
- lhs = L.reg.Get(i)
- if !LVCanConvToString(lhs) {
- break
- }
- buf[total-1] = LVAsString(lhs)
- i--
- total--
- }
- rhs = LString(strings.Join(buf, ""))
- }
- }
- return rhs
- }
- func lessThan(L *LState, lhs, rhs LValue) bool {
- // optimization for numbers
- if v1, ok1 := lhs.(LNumber); ok1 {
- if v2, ok2 := rhs.(LNumber); ok2 {
- return v1 < v2
- }
- L.RaiseError("attempt to compare %v with %v", lhs.Type().String(), rhs.Type().String())
- }
- if lhs.Type() != rhs.Type() {
- L.RaiseError("attempt to compare %v with %v", lhs.Type().String(), rhs.Type().String())
- return false
- }
- ret := false
- switch lhs.Type() {
- case LTString:
- ret = strCmp(string(lhs.(LString)), string(rhs.(LString))) < 0
- default:
- ret = objectRationalWithError(L, lhs, rhs, "__lt")
- }
- return ret
- }
- func equals(L *LState, lhs, rhs LValue, raw bool) bool {
- lt := lhs.Type()
- if lt != rhs.Type() {
- return false
- }
- ret := false
- switch lt {
- case LTNil:
- ret = true
- case LTNumber:
- v1, _ := lhs.(LNumber)
- v2, _ := rhs.(LNumber)
- ret = v1 == v2
- case LTBool:
- ret = bool(lhs.(LBool)) == bool(rhs.(LBool))
- case LTString:
- ret = string(lhs.(LString)) == string(rhs.(LString))
- case LTUserData, LTTable:
- if lhs == rhs {
- ret = true
- } else if !raw {
- switch objectRational(L, lhs, rhs, "__eq") {
- case 1:
- ret = true
- default:
- ret = false
- }
- }
- default:
- ret = lhs == rhs
- }
- return ret
- }
- func objectRationalWithError(L *LState, lhs, rhs LValue, event string) bool {
- switch objectRational(L, lhs, rhs, event) {
- case 1:
- return true
- case 0:
- return false
- }
- L.RaiseError("attempt to compare %v with %v", lhs.Type().String(), rhs.Type().String())
- return false
- }
- func objectRational(L *LState, lhs, rhs LValue, event string) int {
- m1 := L.metaOp1(lhs, event)
- m2 := L.metaOp1(rhs, event)
- if m1.Type() == LTFunction && m1 == m2 {
- L.reg.Push(m1)
- L.reg.Push(lhs)
- L.reg.Push(rhs)
- L.Call(2, 1)
- if LVAsBool(L.reg.Pop()) {
- return 1
- }
- return 0
- }
- return -1
- }
|