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

@@ -133,7 +133,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 +212,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 +224,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 +233,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 +254,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

@@ -221,6 +221,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
@@ -231,9 +234,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,27 @@
 #include <autoattach/autoattach.h>
 #include <autoattach/ctx.h>
-#include <autoattach/lua_module.h>
+#include <autoattach/session.h>
 #include <autoattach/wait_dll.h>
-#include <bee/nonstd/format.h>
-#include <resolver/lua_resolver.h>
 #include <util/log.h>
+#include <util/path.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
@@ -26,7 +34,22 @@ namespace luadebug::autoattach {
          "R. Ierusalimschy, L. H. de Figueiredo, W. Celes"
          " $",  // others
     };
-    static bool is_lua_module(const char* module_path, bool check_export = true, bool check_strings = false) {
+
+    static bool is_lua_module(session& session, const char* module_path, bool check_export, bool check_strings) {
+        auto str = std::string_view(module_path);
+        // blacklist module
+        auto root = 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 = session.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
@@ -49,7 +72,7 @@ namespace luadebug::autoattach {
             false
 #endif
             ;
-        if (!is_lua_module(path.c_str(), check_export, true)) {
+        if (!is_lua_module(*ctx::get()->session, path.c_str(), check_export, true)) {
             return false;
         }
         // find lua module lazy
@@ -61,22 +84,22 @@ namespace luadebug::autoattach {
         if (ctx->lua_module) {
             return;
         }
-
-        bool found = false;
-        lua_module rm(ctx->mode);
-        Gum::Process::enumerate_modules([&rm, &found](const Gum::ModuleDetails& details) -> bool {
-            if (is_lua_module(details.path())) {
-                auto range        = details.range();
-                rm.memory_address = range.base_address;
-                rm.memory_size    = range.size;
-                rm.path           = details.path();
-                rm.name           = details.name();
-                found             = true;
+        Gum::Process::enumerate_modules([&ctx](const Gum::ModuleDetails& details) -> bool {
+            auto& session = *ctx->session;
+            if (is_lua_module(session, details.path(), true, false)) {
+                ctx->lua_module                 = lua_module { session.mode };
+                auto range                      = details.range();
+                ctx->lua_module->memory_address = range.base_address;
+                ctx->lua_module->memory_size    = range.size;
+                ctx->lua_module->path           = details.path();
+                ctx->lua_module->name           = details.name();
+                ctx->lua_module->version        = session.version;
+                ctx->session.reset();
                 return false;
             }
             return true;
         });
-        if (!found) {
+        if (!ctx->lua_module) {
             if (ctx->wait_dll)
                 return;
             if (!wait_dll(load_lua_module)) {
@@ -88,23 +111,27 @@ namespace luadebug::autoattach {
             return;
         }
 
-        log::info("find lua module path:{}", rm.path);
-        if (!rm.initialize()) {
+        log::info("find lua module path:{}", ctx->lua_module->path);
+        if (!ctx->lua_module->initialize()) {
+            ctx->lua_module.reset();
             return;
         }
-        ctx->lua_module = std::move(rm);
     }
 
-    void start(work_mode mode) {
+    void start() {
         auto ctx = ctx::get();
         std::lock_guard guard(ctx->mtx);
-        ctx->mode = mode;
         start_impl(ctx);
     }
 
-    void start() {
+    void start(work_mode mode) {
         auto ctx = ctx::get();
         std::lock_guard guard(ctx->mtx);
+        ctx->session = init_session(mode);
+        if (!ctx->session) {
+            log::fatal("can't load config");
+            return;
+        }
         start_impl(ctx);
     }
 

src/launcher/autoattach/autoattach.h

@@ -12,6 +12,7 @@ namespace luadebug::autoattach {
         launch,
         attach,
     };
-    attach_status attach_lua(lua::state L);
+    struct lua_module;
+    attach_status attach_lua(lua::state L, lua_module &module);
     void initialize(work_mode mode);
 }