add launcher config

.gitmodules

@@ -10,4 +10,6 @@
 [submodule "3rd/frida_gum/gumpp"]
 	path = 3rd/frida_gum/gumpp
 	url = https://github.com/fesily/gumpp
-
+[submodule "3rd/json"]
+	path = 3rd/json
+	url = https://github.com/nlohmann/json

compile/common/launcher.lua

@@ -20,6 +20,7 @@ lm:lua_source 'launcher_source' {
         "3rd/bee.lua",
         "3rd/frida_gum/gumpp",
         "3rd/lua/lua54",
+        "3rd/json/single_include",
         "src/launcher",
     },
     sources = {

compile/common/package_json.lua

@@ -252,6 +252,11 @@ Debugger address.
         },
         type = "string",
     },
+    module = {
+        default = "null",
+        markdownDescription = "specify lua module path/name",
+        type = "string",
+    }
 }
 
 if OS == "win32" then
@@ -471,6 +476,10 @@ for k, v in pairs(attributes.common) do
     attributes.attach[k] = v
     attributes.launch[k] = v
 end
+
+table.insert(attributes.attach.luaVersion.enum, "unknown")
+attributes.attach.luaVersion.default = "unknown"
+
 json.contributes.debuggers[1].configurationAttributes = {
     launch = { properties = attributes.launch },
     attach = { properties = attributes.attach },

extension/script/attach.lua

@@ -1,4 +1,4 @@
-local path, pid = ...
+local path, pid, debugger_path = ...
 if _VERSION == nil
     or type == nil
     or assert == nil
@@ -19,27 +19,22 @@ if is_luajit and jit == nil then
     return "wait initialized"
 end
 
+if debugger_path == "" then
+    debugger_path = nil
+end
+
 local function dofile(filename, ...)
     local load = _VERSION == "Lua 5.1" and loadstring or load
     local f = assert(io.open(filename))
     local str = f:read "*a"
     f:close()
     return assert(load(str, "=(debugger.lua)"))(...)
 end
-local function isLatest()
-    local ipc = dofile(path.."/script/common/ipc.lua")
-    local fd = ipc(path, pid, "luaVersion")
-    local result = false
-    if fd then
-        result = "lua-latest" == fd:read "a"
-        fd:close()
-    end
-    return result
-end
+
 local dbg = dofile(path.."/script/debugger.lua", path)
 dbg:start {
     address = ("@%s/tmp/pid_%s"):format(path, pid),
-    latest = isLatest(),
+    debugger_path = debugger_path,
 }
 dbg:event "wait"
-return "ok"
+return "ok"

extension/script/debugger.lua

@@ -101,26 +101,40 @@ local function detectLuaDebugPath(cfg)
         end
     end
 
-    local rt = "/runtime/"..PLATFORM
-    if cfg.latest then
-        rt = rt.."/lua-latest"
-    elseif _VERSION == "Lua 5.4" then
-        rt = rt.."/lua54"
-    elseif _VERSION == "Lua 5.3" then
-        rt = rt.."/lua53"
-    elseif _VERSION == "Lua 5.2" then
-        rt = rt.."/lua52"
-    elseif _VERSION == "Lua 5.1" then
-        if (tostring(assert):match('builtin') ~= nil) then
-            rt = rt.."/luajit"
-            jit.off()
-        else
-            rt = rt.."/lua51"
+    local function get_luadebug_dir()
+        local version = cfg.version
+        if not version then
+            if tostring(assert):match('builtin') ~= nil then
+                version = "luajit"
+                jit.off()
+            else
+                local version_t = {
+                    ["Lua 5.4"] = "lua54",
+                    ["Lua 5.3"] = "lua53",
+                    ["Lua 5.2"] = "lua52",
+                    ["Lua 5.1"] = "lua51",
+                }
+                version = version_t[_VERSION]
+            end
         end
-    else
-        error(_VERSION.." is not supported.")
+
+        local valid = {
+            ["lua-latest"] = true,
+            lua54 = true,
+            lua53 = true,
+            lua52 = true,
+            lua51 = true,
+            luajit = true,
+        }
+
+        if not valid[version] then
+            error(tostring(version).." is not supported.")
+        end
+        return version
     end
 
+    local rt = "/runtime/"..PLATFORM.."/"..get_luadebug_dir()
+
     local ext = isWindows() and "dll" or "so"
     return root..rt..'/luadebug.'..ext
 end
@@ -133,7 +147,7 @@ local function initDebugger(dbg, cfg)
     local luadebug = os.getenv "LUA_DEBUG_CORE"
     local updateenv = false
     if not luadebug then
-        luadebug = detectLuaDebugPath(cfg)
+        luadebug = cfg.debugger_path or detectLuaDebugPath(cfg)
         updateenv = true
     end
     local isWindows = package.config:sub(1, 1) == "\\"
@@ -212,7 +226,7 @@ function dbg:event(...)
 end
 
 function dbg:set_wait(name, f)
-    _G[name] = function(...)
+    _G[name] = function (...)
         _G[name] = nil
         f(...)
         self:event 'wait'
@@ -224,7 +238,7 @@ function dbg:setup_patch()
     local rawxpcall = xpcall
     function pcall(f, ...)
         return rawxpcall(f,
-            function(msg)
+            function (msg)
                 self:event("exception", msg)
                 return msg
             end,
@@ -233,7 +247,7 @@ function dbg:setup_patch()
 
     function xpcall(f, msgh, ...)
         return rawxpcall(f,
-            function(msg)
+            function (msg)
                 self:event("exception", msg)
                 return msgh and msgh(msg) or msg
             end
@@ -254,7 +268,7 @@ function dbg:setup_patch()
     function coroutine.wrap(f)
         local wf = rawcoroutinewrap(f)
         local _, co = debug.getupvalue(wf, 1)
-        return function(...)
+        return function (...)
             self:event("thread", co, 0)
             return coreturn(co, wf(...))
         end

extension/script/frontend/debuger_factory.lua

@@ -208,6 +208,9 @@ local function create_process_in_console(args, callback)
     if not process then
         return nil, err
     end
+    if callback then
+        callback(process)
+    end
     if args.inject ~= "none" then
 		local ok, errmsg = process_inject.inject(process, "launch", args)
         if not ok then
@@ -218,9 +221,6 @@ local function create_process_in_console(args, callback)
             end
         end
     end
-    if callback then
-        callback(process)
-    end
     if need_resume then
         process:resume()
     end

extension/script/frontend/proxy.lua

@@ -4,6 +4,7 @@ local fs = require 'bee.filesystem'
 local sp = require 'bee.subprocess'
 local platform_os = require 'frontend.platform_os'
 local process_inject = require 'frontend.process_inject'
+local json = require 'common.json'
 local server
 local client
 local initReq
@@ -15,11 +16,16 @@ local function getUnixAddress(pid)
     return "@"..(path / ("pid_%d"):format(pid)):string()
 end
 
-local function ipc_send_latest(pid)
+local function ipc_send_config(pid, args)
+    --TODO: clear config file
     fs.create_directories(WORKDIR / "tmp")
     local ipc = require "common.ipc"
-    local fd = assert(ipc(WORKDIR, pid, "luaVersion", "w"))
-    fd:write("latest")
+    local fd = assert(ipc(WORKDIR, pid, "config", "w"))
+    local config = {
+        version = args.luaVersion,
+        module = args.module
+    }
+    fd:write(json.encode(config))
     fd:close()
 end
 
@@ -56,13 +62,11 @@ end
 
 local function attach_process(pkg, pid)
     local args = pkg.arguments
-    if args.luaVersion == "lua-latest" then
-        ipc_send_latest(pid)
-    end
+    ipc_send_config(pid, args)
     local ok, errmsg = process_inject.inject(pid, "attach", args)
     if not ok then
-		return false, errmsg
-	end
+        return false, errmsg
+    end
 
     server = network(getUnixAddress(pid), true)
     server.sendmsg(initReq)
@@ -80,8 +84,8 @@ local function proxy_attach(pkg)
     local args = pkg.arguments
     platform_os.init(args)
     if platform_os() ~= "Windows" and platform_os() ~= "macOS" then
-		attach_tcp(pkg, args)
-		return
+        attach_tcp(pkg, args)
+        return
     end
     if args.processId then
         local processId = tonumber(args.processId)
@@ -92,7 +96,7 @@ local function proxy_attach(pkg)
         return
     end
     if args.processName then
-        local pids = require "frontend.query_process"(args.processName)
+        local pids = require "frontend.query_process" (args.processName)
         if #pids == 0 then
             response_error(pkg, ('Cannot found process `%s`.'):format(args.processName))
             return
@@ -113,7 +117,7 @@ local function create_server(args, pid)
     local s, address
     if args.address ~= nil then
         s = network(args.address, args.client)
-        address = (args.client and "s:" or "c:") .. args.address
+        address = (args.client and "s:" or "c:")..args.address
     else
         pid = pid or sp.get_id()
         s = network(getUnixAddress(pid), true)
@@ -158,11 +162,12 @@ local function proxy_launch_console(pkg)
             response_error(pkg, "`runtimeExecutable` need specify `inject` or `address`.")
             return
         end
-        local process, err = debuger_factory.create_process_in_console(args, function (process)
+        local process, err = debuger_factory.create_process_in_console(args, function(process)
             local address
             server, address = create_server(args, process:get_id())
-            if args.luaVersion == "lua-latest" and type(address) == "number" then
-                ipc_send_latest(address)
+
+            if type(address) == "number" then
+                ipc_send_config(address, args)
             end
         end)
         if not process then

src/launcher/autoattach/autoattach.cpp

@@ -1,19 +1,26 @@
 #include <autoattach/autoattach.h>
 #include <autoattach/ctx.h>
-#include <autoattach/lua_module.h>
 #include <autoattach/wait_dll.h>
-#include <bee/nonstd/format.h>
-#include <resolver/lua_resolver.h>
+#include <config/config.h>
 #include <util/log.h>
 
 #include <array>
 #include <atomic>
 #include <gumpp.hpp>
-#include <memory>
 #include <string>
 #include <thread>
 
 namespace luadebug::autoattach {
+#ifdef _WIN32
+#    define EXT ".dll"
+#else
+#    define EXT ".so"
+#endif
+    constexpr auto lua_module_backlist = {
+        "launcher" EXT,
+        "luadebug" EXT,
+    };
+
     constexpr auto find_lua_module_key = "lua_newstate";
     constexpr auto lua_module_strings  = std::array<const char*, 3> {
         "luaJIT_BC_%s",  // luajit
@@ -27,6 +34,22 @@ namespace luadebug::autoattach {
          " $",  // others
     };
     static bool is_lua_module(const char* module_path, bool check_export = true, bool check_strings = false) {
+        auto str = std::string_view(module_path);
+        // blacklist module
+        auto root = config::get_plugin_root();
+        if (root) {
+            if (str.find((*root).string()) != std::string_view::npos) {
+                // in luadebug root dir
+                for (auto& s : lua_module_backlist) {
+                    if (str.find(s) != std::string_view::npos)
+                        return false;
+                }
+            }
+        }
+
+        const auto& target_lua_module = ctx::get()->config->lua_module;
+        if (!target_lua_module.empty() && str.find(target_lua_module) != std::string_view::npos) return true;
+
         if (check_export && Gum::Process::module_find_export_by_name(module_path, find_lua_module_key)) return true;
         if (Gum::Process::module_find_symbol_by_name(module_path, find_lua_module_key)) return true;
         // TODO: when signature mode, check strings
@@ -62,6 +85,11 @@ namespace luadebug::autoattach {
             return;
         }
 
+        ctx->config = config::init_from_file();
+        if (!ctx->config) {
+            log::info("can't load config");
+        }
+
         bool found = false;
         lua_module rm(ctx->mode);
         Gum::Process::enumerate_modules([&rm, &found](const Gum::ModuleDetails& details) -> bool {

src/launcher/autoattach/ctx.h

@@ -1,6 +1,7 @@
 #pragma once
 #include <autoattach/autoattach.h>
 #include <autoattach/lua_module.h>
+#include <config/config.h>
 
 #include <atomic>
 #include <gumpp.hpp>
@@ -15,6 +16,7 @@ namespace luadebug::autoattach {
         work_mode mode;
         std::optional<struct lua_module> lua_module;
         bool wait_dll = false;
+        std::optional<config::Config> config;
 
         static ctx* get() {
             static ctx obj;