From 1a7f5f7b314c213026e1addf64c3ad63fd7a5b11 Mon Sep 17 00:00:00 2001
From: georgemoralis <giorgosmrls@gmail.com>
Date: Tue, 26 Mar 2024 18:13:27 +0200
Subject: [PATCH] function to start modules

---
 src/core/linker.cpp | 30 ++++++++++++++++++++++++++++--
 src/core/linker.h   |  4 +++-
 2 files changed, 31 insertions(+), 3 deletions(-)

diff --git a/src/core/linker.cpp b/src/core/linker.cpp
index 653d853a..23c60054 100644
--- a/src/core/linker.cpp
+++ b/src/core/linker.cpp
@@ -604,6 +604,27 @@ void Linker::Resolve(const std::string& name, Loader::SymbolType sym_type, Modul
 
 using exit_func_t = PS4_SYSV_ABI void (*)();
 using entry_func_t = PS4_SYSV_ABI void (*)(EntryParams* params, exit_func_t atexit_func);
+using module_ini_func_t = PS4_SYSV_ABI int (*)(size_t args, const void* argp, module_func_t func);
+
+static PS4_SYSV_ABI int run_module(uint64_t addr, size_t args, const void* argp,
+                                   module_func_t func) {
+    return reinterpret_cast<module_ini_func_t>(addr)(args, argp, func);
+}
+
+int Linker::StartModule(Module* m, size_t args, const void* argp, module_func_t func) {
+    LOG_INFO(Core_Linker, "Module started : {}\n", m->file_name);
+    return run_module(m->dynamic_info.init_virtual_addr + m->base_virtual_addr, args, argp, func);
+}
+
+void Linker::StartAllModules() {
+    std::scoped_lock lock{m_mutex};
+
+    for (auto& m : m_modules) {
+        if (m->elf.IsSharedLib()) {
+            StartModule(m.get(), 0, nullptr, nullptr);
+        }
+    }
+}
 
 static PS4_SYSV_ABI void ProgramExitFunc() {
     fmt::print("exit function called\n");
@@ -638,15 +659,20 @@ void Linker::Execute() {
     }
 
     Core::Libraries::LibKernel::pthreadInitSelfMainThread();
+    //relocate all modules
     for (const auto& m : m_modules) {
         Relocate(m.get());
     }
+    StartAllModules();
     EntryParams p{};
     p.argc = 1;
     p.argv[0] = "eboot.bin"; // hmm should be ok?
 
-    const auto& module = m_modules.at(0);
-    RunMainEntry(module->elf.GetElfEntry() + module->base_virtual_addr, &p, ProgramExitFunc);
+    for (auto& m : m_modules) {
+        if (!m->elf.IsSharedLib()) {
+            RunMainEntry(m->elf.GetElfEntry() + m->base_virtual_addr, &p, ProgramExitFunc);
+        }
+    }
 }
 
 void Linker::DebugDump() {
diff --git a/src/core/linker.h b/src/core/linker.h
index be637f31..de4048aa 100644
--- a/src/core/linker.h
+++ b/src/core/linker.h
@@ -9,7 +9,7 @@
 #include "core/loader/symbols_resolver.h"
 
 namespace Core {
-
+using module_func_t = int (*)(size_t args, const void* argp);
 struct DynamicModuleInfo;
 class Linker;
 
@@ -142,6 +142,8 @@ private:
     const ModuleInfo* FindModule(const Module& m, const std::string& id);
     const LibraryInfo* FindLibrary(const Module& program, const std::string& id);
     Module* FindExportedModule(const ModuleInfo& m, const LibraryInfo& l);
+    int StartModule(Module* m, size_t args, const void* argp, module_func_t func);
+    void StartAllModules();
 
     std::vector<std::unique_ptr<Module>> m_modules;
     Loader::SymbolsResolver m_hle_symbols{};