diff options
| -rw-r--r-- | script-beta/vm/eachRef.lua | 577 | ||||
| -rw-r--r-- | script-beta/vm/init.lua | 1 | ||||
| -rw-r--r-- | script-beta/vm/refOf.lua | 466 |
3 files changed, 269 insertions, 775 deletions
diff --git a/script-beta/vm/eachRef.lua b/script-beta/vm/eachRef.lua index f08423a5..dc97c04d 100644 --- a/script-beta/vm/eachRef.lua +++ b/script-beta/vm/eachRef.lua @@ -3,8 +3,153 @@ local files = require 'files' local vm = require 'vm.vm' local library = require 'library' local await = require 'await' +local vm = require 'vm.vm' +local guide = require 'parser.guide' +local files = require 'files' +local library = require 'library' + +local function ofSelf(state, loc, callback) + -- self 的2个特殊引用位置： + -- 1. 当前方法定义时的对象（mt） + local method = loc.method + local node = method.node + callback(node) + -- 2. 调用该方法时传入的对象 +end + +local function ofLocal(state, loc, callback) + if state[loc] then + return + end + state[loc] = true + -- 方法中的 self 使用了一个虚拟的定义位置 + if loc.tag ~= 'self' then + callback(loc, 'declare') + end + local refs = loc.ref + if refs then + for i = 1, #refs do + local ref = refs[i] + if ref.type == 'getlocal' then + callback(ref, 'get') + if loc.tag == '_ENV' then + local parent = ref.parent + if parent.type == 'getfield' + or parent.type == 'getindex' then + if guide.getKeyName(parent) == '_G' then + callback(parent, 'declare') + end + end + end + elseif ref.type == 'setlocal' then + callback(ref, 'set') + elseif ref.type == 'getglobal' then + if loc.tag == '_ENV' then + if guide.getName(ref) == '_G' then + callback(ref, 'get') + end + end + end + end + end + if loc.tag == 'self' then + ofSelf(state, loc, callback) + end +end + +local function ofGlobal(state, source, callback) + if state[source] then + return + end + local key = guide.getKeyName(source) + local node = source.node + if node.tag == '_ENV' then + local uris = files.findGlobals(key) + for i = 1, #uris do + local uri = uris[i] + local ast = files.getAst(uri) + local globals = vm.getGlobals(ast.ast) + if globals and globals[key] then + for _, info in ipairs(globals[key]) do + state[info.source] = true + callback(info) + end + end + end + else + vm.eachField(node, function (info) + if key == info.key then + state[info.source] = true + callback(info) + end + end) + end +end + +local function ofField(state, source, callback) + if state[source] then + return + end + local parent = source.parent + local key = guide.getKeyName(source) + if parent.type == 'tablefield' + or parent.type == 'tableindex' then + local tbl = parent.parent + vm.eachField(tbl, function (info) + if key == info.key then + state[info.source] = true + callback(info) + end + end) + else + local node = parent.node + vm.eachField(node, function (info) + if key == info.key then + state[info.source] = true + callback(info) + end + end) + end +end + +local function ofLabel(state, source, callback) + if state[source] then + return + end + state[source] = true + callback(source, 'set') + if source.ref then + for _, ref in ipairs(source.ref) do + callback(ref, 'get') + end + end +end + +local function ofGoTo(state, source, callback) + local name = source[1] + local label = guide.getLabel(source, name) + if label then + ofLabel(state, label, callback) + end +end + +local function ofValue(state, source, callback) + callback(source, 'value') +end -local function ofCall(func, index, callback, offset) +local function ofIndex(state, source, callback) + local parent = source.parent + if not parent then + return + end + if parent.type == 'setindex' + or parent.type == 'getindex' + or parent.type == 'tableindex' then + ofField(state, source, callback) + end +end + +local function ofCall(state, func, index, callback, offset) offset = offset or 0 vm.eachRef(func, function (info) local src = info.source @@ -14,53 +159,30 @@ local function ofCall(func, index, callback, offset) end if returns then -- 搜索函数第 index 个返回值 - for _, rtn in ipairs(returns) do + for i = 1, #returns do + local rtn = returns[i] local val = rtn[index-offset] if val then - callback { - source = val, - mode = 'return', - } - vm.eachRef(val, callback) + callback(val) end end end end) end -local function ofCallSelect(call, index, callback) - local slc = call.parent - if slc.index == index then - vm.eachRef(slc.parent, callback) - return - end - if call.extParent then - for i = 1, #call.extParent do - slc = call.extParent[i] - if slc.index == index then - vm.eachRef(slc.parent, callback) - return - end - end - end -end - -local function ofSpecialCall(call, func, index, callback, offset) +local function ofSpecialCall(state, call, func, index, callback, offset) local name = func.special offset = offset or 0 if name == 'setmetatable' then if index == 1 + offset then local args = call.args if args[1+offset] then - vm.eachRef(args[1+offset], callback) + callback(args[1+offset]) end if args[2+offset] then vm.eachField(args[2+offset], function (info) if info.key == 's|__index' then - vm.eachRef(info.source, callback) - if info.value then - vm.eachRef(info.value, callback) - end + callback(info.source) end end) end @@ -71,11 +193,12 @@ local function ofSpecialCall(call, func, index, callback, offset) local result = vm.getLinkUris(call) if result then local myUri = guide.getRoot(call).uri - for _, uri in ipairs(result) do + for i = 1, #result do + local uri = result[i] if not files.eq(uri, myUri) then local ast = files.getAst(uri) if ast then - ofCall(ast.ast, 1, callback) + ofCall(state, ast.ast, 1, callback) end end end @@ -87,10 +210,7 @@ local function ofSpecialCall(call, func, index, callback, offset) local objName = args[1+offset][1] local lib = library.library[objName] if lib then - callback { - source = lib, - mode = 'value', - } + callback(lib) end end end @@ -103,8 +223,8 @@ local function ofSpecialCall(call, func, index, callback, offset) vm.eachRef(args[1+offset], function (info) local src = info.source if src.type == 'function' then - ofCall(src, index, callback, 1+offset) - ofSpecialCall(call, src, index, callback, 1+offset) + ofCall(state, src, index, callback, 1+offset) + ofSpecialCall(state, call, src, index, callback, 1+offset) end end) end @@ -112,59 +232,17 @@ local function ofSpecialCall(call, func, index, callback, offset) end end -local function asSetValue(value, callback) - if value.type == 'field' - or value.type == 'method' then - value = value.parent - end - local parent = value.parent - if not parent then - return - end - if parent.type == 'local' - or parent.type == 'setglobal' - or parent.type == 'setlocal' - or parent.type == 'setfield' - or parent.type == 'setmethod' - or parent.type == 'setindex' - or parent.type == 'tablefield' - or parent.type == 'tableindex' then - if parent.value == value then - vm.eachRef(parent, callback) - if guide.getName(parent) == '__index' then - if parent.type == 'tablefield' - or parent.type == 'tableindex' then - local t = parent.parent - local args = t.parent - if args[2] == t then - local call = args.parent - local func = call.node - if func.special == 'setmetatable' then - vm.eachRef(args[1], callback) - end - end - end - end - end - end -end - -local function ofSelect(source, callback) +local function ofSelect(state, source, callback) -- 检查函数返回值 local call = source.vararg if call.type == 'call' then - ofCall(call.node, source.index, callback) - ofSpecialCall(call, call.node, source.index, callback) + ofCall(state, call.node, source.index, callback) + ofSpecialCall(state, call, call.node, source.index, callback) end end -local function ofSelf(loc, callback) - -- self 的2个特殊引用位置： - -- 1. 当前方法定义时的对象（mt） - local method = loc.method - local node = method.node - vm.eachRef(node, callback) - -- 2. 调用该方法时传入的对象 +local function ofMain(state, source, callback) + callback(source, 'main') end local function getCallRecvs(call) @@ -176,7 +254,8 @@ local function getCallRecvs(call) local recvs = {} recvs[1] = parent.parent if extParent then - for _, p in ipairs(extParent) do + for i = 1, #extParent do + local p = extParent[i] recvs[#recvs+1] = p.parent end end @@ -184,7 +263,7 @@ local function getCallRecvs(call) end --- 自己作为函数的参数 -local function asArg(source, callback) +local function checkAsArg(state, source, callback) local parent = source.parent if not parent then return @@ -198,16 +277,13 @@ local function asArg(source, callback) if parent[2] then vm.eachField(parent[2], function (info) if info.key == 's|__index' then - vm.eachRef(info.source, callback) - if info.value then - vm.eachRef(info.value, callback) - end + callback(info) end end) end local recvs = getCallRecvs(call) if recvs and recvs[1] then - vm.eachRef(recvs[1], callback) + callback(recvs[1]) end vm.setMeta(source, parent[2]) end @@ -215,8 +291,25 @@ local function asArg(source, callback) end end +local function ofCallSelect(state, call, index, callback) + local slc = call.parent + if slc.index == index then + callback(slc.parent) + return + end + if call.extParent then + for i = 1, #call.extParent do + slc = call.extParent[i] + if slc.index == index then + callback(slc.parent) + return + end + end + end +end + --- 自己作为函数的返回值 -local function asReturn(source, callback) +local function checkAsReturn(state, source, callback) local parent = source.parent if source.type == 'field' or source.type == 'method' then @@ -232,15 +325,16 @@ local function asReturn(source, callback) if not uris then return end - for _, uri in ipairs(uris) do + for i = 1, #uris do + local uri = uris[i] local ast = files.getAst(uri) if ast then local links = vm.getLinks(ast.ast) if links then for linkUri, calls in pairs(links) do if files.eq(linkUri, myUri) then - for i = 1, #calls do - ofCallSelect(calls[i], 1, callback) + for j = 1, #calls do + ofCallSelect(state, calls[j], 1, callback) end end end @@ -266,262 +360,114 @@ local function asReturn(source, callback) end local recvs = getCallRecvs(call) if recvs and recvs[index] then - vm.eachRef(recvs[index], callback) + callback(recvs[index]) elseif index == 1 then - callback { - type = 'call', - source = call, - } + callback(call, 'call') end end) end end -local function ofLocal(loc, callback) - -- 方法中的 self 使用了一个虚拟的定义位置 - if loc.tag ~= 'self' then - callback { - source = loc, - mode = 'declare', - } - end - if loc.ref then - for _, ref in ipairs(loc.ref) do - if ref.type == 'getlocal' then - callback { - source = ref, - mode = 'get', - } - vm.eachRef(ref, callback) - elseif ref.type == 'setlocal' then - callback { - source = ref, - mode = 'set', - } - vm.eachRef(ref, callback) - if ref.value then - vm.eachRef(ref.value, callback) - end - end - end - end - if loc.tag == 'self' then - ofSelf(loc, callback) - end - if loc.value then - vm.eachRef(loc.value, callback) +local function checkAsParen(state, source, callback) + if state[source] then + return end - if loc.tag == '_ENV' and loc.ref then - for _, ref in ipairs(loc.ref) do - if ref.type == 'getlocal' then - local parent = ref.parent - if parent.type == 'getfield' - or parent.type == 'getindex' then - if guide.getKeyName(parent) == '_G' then - callback { - source = parent, - mode = 'get', - } - end - end - elseif ref.type == 'getglobal' then - if guide.getName(ref) == '_G' then - callback { - source = ref, - mode = 'get', - } - end - end - end + state[source] = true + if source.parent and source.parent.type == 'paren' then + vm.refOf(state, source.parent, callback) end end -local function ofGlobal(source, callback) - local key = guide.getKeyName(source) - local node = source.node - if node.tag == '_ENV' then - local uris = files.findGlobals(key) - for _, uri in ipairs(uris) do - local ast = files.getAst(uri) - local globals = vm.getGlobals(ast.ast) - if globals and globals[key] then - for _, info in ipairs(globals[key]) do - callback(info) - if info.value then - vm.eachRef(info.value, callback) - end - end - end - end - else - vm.eachField(node, function (info) - if key == info.key then - callback { - source = info.source, - mode = info.mode, - } - if info.value then - vm.eachRef(info.value, callback) - end - end - end) +local function checkValue(state, source, callback) + if source.value then + callback(source.value) end end -local function ofField(source, callback) - if not source then - return - end - local parent = source.parent - local key = guide.getKeyName(source) - if parent.type == 'tablefield' - or parent.type == 'tableindex' then - local tbl = parent.parent - vm.eachField(tbl, function (info) - if key == info.key then - callback { - source = info.source, - mode = info.mode, - } - vm.eachRef(info.source, callback) - if info.value then - vm.eachRef(info.value, callback) - end - end - end) - else - local node = parent.node - vm.eachField(node, function (info) - if key == info.key then - callback { - source = info.source, - mode = info.mode, - } - vm.eachRef(info.source, callback) - if info.value then - vm.eachRef(info.value, callback) - end - end - end) +local function checkSetValue(state, value, callback) + if value.type == 'field' + or value.type == 'method' then + value = value.parent end -end - -local function ofLiteral(source, callback) - local parent = source.parent + local parent = value.parent if not parent then return end - if parent.type == 'setindex' - or parent.type == 'getindex' + if parent.type == 'local' + or parent.type == 'setglobal' + or parent.type == 'setlocal' + or parent.type == 'setfield' + or parent.type == 'setmethod' + or parent.type == 'setindex' + or parent.type == 'tablefield' or parent.type == 'tableindex' then - ofField(source, callback) - end -end - -local function ofLabel(source, callback) - callback { - source = source, - mode = 'set', - } - if source.ref then - for _, ref in ipairs(source.ref) do - callback { - source = ref, - mode = 'get', - } + if parent.value == value then + callback(parent) + if guide.getName(parent) == '__index' then + if parent.type == 'tablefield' + or parent.type == 'tableindex' then + local t = parent.parent + local args = t.parent + if args[2] == t then + local call = args.parent + local func = call.node + if func.special == 'setmetatable' then + callback(args[1]) + end + end + end + end end end end -local function ofGoTo(source, callback) - local name = source[1] - local label = guide.getLabel(source, name) - if label then - ofLabel(label, callback) - end -end - -local function ofMain(source, callback) - callback { - source = source, - mode = 'main', - } -end - -local function asParen(source, callback) - if source.parent and source.parent.type == 'paren' then - vm.eachRef(source.parent, callback) - end -end - -local function ofSelfValue(source, callback) - callback { - source = source, - mode = 'value', - } -end - -local function eachRef(source, callback) +function vm.refOf(state, source, callback) local stype = source.type - if stype == 'local' then - ofLocal(source, callback) + if stype == 'local' then + ofLocal(state, source, callback) elseif stype == 'getlocal' or stype == 'setlocal' then - ofLocal(source.node, callback) + ofLocal(state, source.node, callback) elseif stype == 'setglobal' or stype == 'getglobal' then - ofGlobal(source, callback) + ofGlobal(state, source, callback) elseif stype == 'field' or stype == 'method' then - ofField(source, callback) + ofField(state, source, callback) elseif stype == 'setfield' or stype == 'getfield' or stype == 'tablefield' then - ofField(source.field, callback) + ofField(state, source.field, callback) elseif stype == 'setmethod' or stype == 'getmethod' then - ofField(source.method, callback) + ofField(state, source.method, callback) + elseif stype == 'goto' then + ofGoTo(state, source, callback) + elseif stype == 'label' then + ofLabel(state, source, callback) elseif stype == 'number' or stype == 'boolean' or stype == 'string' then - ofLiteral(source, callback) - ofSelfValue(source, callback) - elseif stype == 'goto' then - ofGoTo(source, callback) - elseif stype == 'label' then - ofLabel(source, callback) + ofIndex(state, source, callback) + ofValue(state, source, callback) elseif stype == 'table' or stype == 'function' then - ofSelfValue(source, callback) + ofValue(state, source, callback) elseif stype == 'select' then - ofSelect(source, callback) + ofSelect(state, source, callback) elseif stype == 'call' then - ofCall(source.node, 1, callback) - ofSpecialCall(source, source.node, 1, callback) + ofCall(state, source.node, 1, callback) + ofSpecialCall(state, source, source.node, 1, callback) elseif stype == 'main' then - ofMain(source, callback) + ofMain(state, source, callback) elseif stype == 'paren' then - eachRef(source.exp, callback) - end - asArg(source, callback) - asReturn(source, callback) - asParen(source, callback) - asSetValue(source, callback) -end - ---- 判断2个对象是否拥有相同的引用 -function vm.isSameRef(a, b) - local cache = vm.cache.eachRef[a] - if cache then - -- 相同引用的source共享同一份cache - return cache == vm.cache.eachRef[b] - else - return vm.eachRef(a, function (info) - if info.source == b then - return true - end - end) or false + vm.refOf(state, source.exp, callback) end + checkValue(state, source, callback) + checkAsArg(state, source, callback) + checkAsReturn(state, source, callback) + checkAsParen(state, source, callback) + checkSetValue(state, source, callback) end local function applyCache(cache, callback, max) @@ -578,6 +524,21 @@ local function eachRef(source, callback) return result end +--- 判断2个对象是否拥有相同的引用 +function vm.isSameRef(a, b) + local cache = vm.cache.eachRef[a] + if cache then + -- 相同引用的source共享同一份cache + return cache == vm.cache.eachRef[b] + else + return vm.eachRef(a, function (info) + if info.source == b then + return true + end + end) or false + end +end + --- 获取所有的引用 function vm.eachRef(source, callback, max) local cache = vm.cache.eachRef[source] diff --git a/script-beta/vm/init.lua b/script-beta/vm/init.lua index 3645e77a..a4f81d07 100644 --- a/script-beta/vm/init.lua +++ b/script-beta/vm/init.lua @@ -1,6 +1,5 @@ local vm = require 'vm.vm' require 'vm.eachField' -require 'vm.refOf' require 'vm.eachRef' require 'vm.eachDef' require 'vm.getGlobals' diff --git a/script-beta/vm/refOf.lua b/script-beta/vm/refOf.lua deleted file mode 100644 index f31c0ec8..00000000 --- a/script-beta/vm/refOf.lua +++ /dev/null @@ -1,466 +0,0 @@ -local vm = require 'vm.vm' -local guide = require 'parser.guide' -local files = require 'files' -local library = require 'library' - -local function ofSelf(state, loc, callback) - -- self 的2个特殊引用位置： - -- 1. 当前方法定义时的对象（mt） - local method = loc.method - local node = method.node - callback(node) - -- 2. 调用该方法时传入的对象 -end - -local function ofLocal(state, loc, callback) - if state[loc] then - return - end - state[loc] = true - -- 方法中的 self 使用了一个虚拟的定义位置 - if loc.tag ~= 'self' then - callback(loc, 'declare') - end - local refs = loc.ref - if refs then - for i = 1, #refs do - local ref = refs[i] - if ref.type == 'getlocal' then - callback(ref, 'get') - if loc.tag == '_ENV' then - local parent = ref.parent - if parent.type == 'getfield' - or parent.type == 'getindex' then - if guide.getKeyName(parent) == '_G' then - callback(parent, 'declare') - end - end - end - elseif ref.type == 'setlocal' then - callback(ref, 'set') - elseif ref.type == 'getglobal' then - if loc.tag == '_ENV' then - if guide.getName(ref) == '_G' then - callback(ref, 'get') - end - end - end - end - end - if loc.tag == 'self' then - ofSelf(state, loc, callback) - end -end - -local function ofGlobal(state, source, callback) - if state[source] then - return - end - local key = guide.getKeyName(source) - local node = source.node - if node.tag == '_ENV' then - local uris = files.findGlobals(key) - for i = 1, #uris do - local uri = uris[i] - local ast = files.getAst(uri) - local globals = vm.getGlobals(ast.ast) - if globals and globals[key] then - for _, info in ipairs(globals[key]) do - state[info.source] = true - callback(info) - end - end - end - else - vm.eachField(node, function (info) - if key == info.key then - state[info.source] = true - callback(info) - end - end) - end -end - -local function ofField(state, source, callback) - if state[source] then - return - end - local parent = source.parent - local key = guide.getKeyName(source) - if parent.type == 'tablefield' - or parent.type == 'tableindex' then - local tbl = parent.parent - vm.eachField(tbl, function (info) - if key == info.key then - state[info.source] = true - callback(info) - end - end) - else - local node = parent.node - vm.eachField(node, function (info) - if key == info.key then - state[info.source] = true - callback(info) - end - end) - end -end - -local function ofLabel(state, source, callback) - if state[source] then - return - end - state[source] = true - callback(source, 'set') - if source.ref then - for _, ref in ipairs(source.ref) do - callback(ref, 'get') - end - end -end - -local function ofGoTo(state, source, callback) - local name = source[1] - local label = guide.getLabel(source, name) - if label then - ofLabel(state, label, callback) - end -end - -local function ofValue(state, source, callback) - callback(source, 'value') -end - -local function ofIndex(state, source, callback) - local parent = source.parent - if not parent then - return - end - if parent.type == 'setindex' - or parent.type == 'getindex' - or parent.type == 'tableindex' then - ofField(state, source, callback) - end -end - -local function ofCall(state, func, index, callback, offset) - offset = offset or 0 - vm.eachRef(func, function (info) - local src = info.source - local returns - if src.type == 'main' or src.type == 'function' then - returns = src.returns - end - if returns then - -- 搜索函数第 index 个返回值 - for i = 1, #returns do - local rtn = returns[i] - local val = rtn[index-offset] - if val then - callback(val) - end - end - end - end) -end - -local function ofSpecialCall(state, call, func, index, callback, offset) - local name = func.special - offset = offset or 0 - if name == 'setmetatable' then - if index == 1 + offset then - local args = call.args - if args[1+offset] then - callback(args[1+offset]) - end - if args[2+offset] then - vm.eachField(args[2+offset], function (info) - if info.key == 's|__index' then - callback(info.source) - end - end) - end - vm.setMeta(args[1+offset], args[2+offset]) - end - elseif name == 'require' then - if index == 1 + offset then - local result = vm.getLinkUris(call) - if result then - local myUri = guide.getRoot(call).uri - for i = 1, #result do - local uri = result[i] - if not files.eq(uri, myUri) then - local ast = files.getAst(uri) - if ast then - ofCall(state, ast.ast, 1, callback) - end - end - end - end - - local args = call.args - if args[1+offset] then - if args[1+offset].type == 'string' then - local objName = args[1+offset][1] - local lib = library.library[objName] - if lib then - callback(lib) - end - end - end - end - elseif name == 'pcall' - or name == 'xpcall' then - if index >= 2-offset then - local args = call.args - if args[1+offset] then - vm.eachRef(args[1+offset], function (info) - local src = info.source - if src.type == 'function' then - ofCall(state, src, index, callback, 1+offset) - ofSpecialCall(state, call, src, index, callback, 1+offset) - end - end) - end - end - end -end - -local function ofSelect(state, source, callback) - -- 检查函数返回值 - local call = source.vararg - if call.type == 'call' then - ofCall(state, call.node, source.index, callback) - ofSpecialCall(state, call, call.node, source.index, callback) - end -end - -local function ofMain(state, source, callback) - callback(source, 'main') -end - -local function getCallRecvs(call) - local parent = call.parent - if parent.type ~= 'select' then - return nil - end - local extParent = call.extParent - local recvs = {} - recvs[1] = parent.parent - if extParent then - for i = 1, #extParent do - local p = extParent[i] - recvs[#recvs+1] = p.parent - end - end - return recvs -end - ---- 自己作为函数的参数 -local function checkAsArg(state, source, callback) - local parent = source.parent - if not parent then - return - end - if parent.type == 'callargs' then - local call = parent.parent - local func = call.node - local name = func.special - if name == 'setmetatable' then - if parent[1] == source then - if parent[2] then - vm.eachField(parent[2], function (info) - if info.key == 's|__index' then - callback(info) - end - end) - end - local recvs = getCallRecvs(call) - if recvs and recvs[1] then - callback(recvs[1]) - end - vm.setMeta(source, parent[2]) - end - end - end -end - -local function ofCallSelect(state, call, index, callback) - local slc = call.parent - if slc.index == index then - callback(slc.parent) - return - end - if call.extParent then - for i = 1, #call.extParent do - slc = call.extParent[i] - if slc.index == index then - callback(slc.parent) - return - end - end - end -end - ---- 自己作为函数的返回值 -local function checkAsReturn(state, source, callback) - local parent = source.parent - if source.type == 'field' - or source.type == 'method' then - parent = parent.parent - end - if not parent or parent.type ~= 'return' then - return - end - local func = guide.getParentFunction(source) - if func.type == 'main' then - local myUri = func.uri - local uris = files.findLinkTo(myUri) - if not uris then - return - end - for i = 1, #uris do - local uri = uris[i] - local ast = files.getAst(uri) - if ast then - local links = vm.getLinks(ast.ast) - if links then - for linkUri, calls in pairs(links) do - if files.eq(linkUri, myUri) then - for j = 1, #calls do - ofCallSelect(state, calls[j], 1, callback) - end - end - end - end - end - end - else - local index - for i = 1, #parent do - if parent[i] == source then - index = i - break - end - end - if not index then - return - end - vm.eachRef(func, function (info) - local src = info.source - local call = src.parent - if not call or call.type ~= 'call' then - return - end - local recvs = getCallRecvs(call) - if recvs and recvs[index] then - callback(recvs[index]) - elseif index == 1 then - callback(call, 'call') - end - end) - end -end - -local function checkAsParen(state, source, callback) - if state[source] then - return - end - state[source] = true - if source.parent and source.parent.type == 'paren' then - vm.refOf(state, source.parent, callback) - end -end - -local function checkValue(state, source, callback) - if source.value then - callback(source.value) - end -end - -local function checkSetValue(state, value, callback) - if value.type == 'field' - or value.type == 'method' then - value = value.parent - end - local parent = value.parent - if not parent then - return - end - if parent.type == 'local' - or parent.type == 'setglobal' - or parent.type == 'setlocal' - or parent.type == 'setfield' - or parent.type == 'setmethod' - or parent.type == 'setindex' - or parent.type == 'tablefield' - or parent.type == 'tableindex' then - if parent.value == value then - callback(parent) - if guide.getName(parent) == '__index' then - if parent.type == 'tablefield' - or parent.type == 'tableindex' then - local t = parent.parent - local args = t.parent - if args[2] == t then - local call = args.parent - local func = call.node - if func.special == 'setmetatable' then - callback(args[1]) - end - end - end - end - end - end -end - -function vm.refOf(state, source, callback) - local stype = source.type - if stype == 'local' then - ofLocal(state, source, callback) - elseif stype == 'getlocal' - or stype == 'setlocal' then - ofLocal(state, source.node, callback) - elseif stype == 'setglobal' - or stype == 'getglobal' then - ofGlobal(state, source, callback) - elseif stype == 'field' - or stype == 'method' then - ofField(state, source, callback) - elseif stype == 'setfield' - or stype == 'getfield' - or stype == 'tablefield' then - ofField(state, source.field, callback) - elseif stype == 'setmethod' - or stype == 'getmethod' then - ofField(state, source.method, callback) - elseif stype == 'goto' then - ofGoTo(state, source, callback) - elseif stype == 'label' then - ofLabel(state, source, callback) - elseif stype == 'number' - or stype == 'boolean' - or stype == 'string' then - ofIndex(state, source, callback) - ofValue(state, source, callback) - elseif stype == 'table' - or stype == 'function' then - ofValue(state, source, callback) - elseif stype == 'select' then - ofSelect(state, source, callback) - elseif stype == 'call' then - ofCall(state, source.node, 1, callback) - ofSpecialCall(state, source, source.node, 1, callback) - elseif stype == 'main' then - ofMain(state, source, callback) - elseif stype == 'paren' then - vm.refOf(state, source.exp, callback) - end - checkValue(state, source, callback) - checkAsArg(state, source, callback) - checkAsReturn(state, source, callback) - checkAsParen(state, source, callback) - checkSetValue(state, source, callback) -end |