From 6cbf6f05a171d297b18d929978604e1f65d42b77 Mon Sep 17 00:00:00 2001
From: Koncord <koncord@rwa.su>
Date: Sun, 23 Dec 2018 23:44:22 +0800
Subject: [PATCH] [Server] Add --debug option to launch options.

Add Soft-Mode Debugger for Mono.
---
 apps/openmw-mp/Script/LangLua/LangLua.cpp     | 10 ++++++
 apps/openmw-mp/Script/LangLua/LangLua.hpp     |  3 ++
 apps/openmw-mp/Script/LangMono/LangMono.cpp   | 35 +++++++++++++++----
 apps/openmw-mp/Script/LangMono/LangMono.hpp   |  6 ++--
 .../Script/LangNative/LangNative.cpp          | 10 ++++++
 .../Script/LangNative/LangNative.hpp          |  2 ++
 apps/openmw-mp/Script/Script.cpp              | 33 +++++++++++++++++
 apps/openmw-mp/Script/Script.hpp              |  5 +++
 apps/openmw-mp/main.cpp                       |  5 +++
 9 files changed, 100 insertions(+), 9 deletions(-)

diff --git a/apps/openmw-mp/Script/LangLua/LangLua.cpp b/apps/openmw-mp/Script/LangLua/LangLua.cpp
index e853e7aaa..2b264a06f 100644
--- a/apps/openmw-mp/Script/LangLua/LangLua.cpp
+++ b/apps/openmw-mp/Script/LangLua/LangLua.cpp
@@ -292,3 +292,13 @@ void LangLua::AddPackageCPath(const std::string& path)
 {
     packageCPath.emplace(path);
 }
+
+void LangLua::Init()
+{
+
+}
+
+void LangLua::Free()
+{
+
+}
diff --git a/apps/openmw-mp/Script/LangLua/LangLua.hpp b/apps/openmw-mp/Script/LangLua/LangLua.hpp
index adf0cc2b6..2de5ae2f5 100644
--- a/apps/openmw-mp/Script/LangLua/LangLua.hpp
+++ b/apps/openmw-mp/Script/LangLua/LangLua.hpp
@@ -48,6 +48,9 @@ public:
 private:
     static std::set<std::string> packageCPath;
     static std::set<std::string> packagePath;
+
+    static void Init();
+    static void Free();
 };
 
 
diff --git a/apps/openmw-mp/Script/LangMono/LangMono.cpp b/apps/openmw-mp/Script/LangMono/LangMono.cpp
index 65c37f514..8f1c9cede 100644
--- a/apps/openmw-mp/Script/LangMono/LangMono.cpp
+++ b/apps/openmw-mp/Script/LangMono/LangMono.cpp
@@ -8,8 +8,9 @@
 #include <mono/jit/jit.h>
 #include <mono/metadata/assembly.h>
 #include <mono/metadata/mono-config.h>
-#include <apps/openmw-mp/Script/ScriptFunctions.hpp>
-#include <apps/openmw-mp/Script/API/TimerAPI.hpp>
+#include <mono/metadata/mono-debug.h>
+#include <Script/ScriptFunctions.hpp>
+#include <Script/API/TimerAPI.hpp>
 #include "LangMono.hpp"
 
 static MonoDomain *domain = nullptr; // shared domain
@@ -114,13 +115,10 @@ lib_t LangMono::GetInterface()
 LangMono::LangMono()
 {
     instance = new MonoInstance;
-
-    Init();
 }
 
 LangMono::LangMono(MonoInstance *instance) : instance(instance)
 {
-    Init();
 }
 
 LangMono::~LangMono()
@@ -133,12 +131,37 @@ void LangMono::Init()
     if (domain != nullptr)
         return;
 
-    domain = mono_jit_init("TES3MP Mono VM"); // will leak :P
+    if (Script::IsDebugMode())
+    {
+        std::string debugServerAddress = "127.0.0.1:10000";
+        LOG_MESSAGE_SIMPLE(Log::LOG_INFO, "Waiting for debugger on %s...", debugServerAddress.c_str());
+        std::string connectionString = "--debugger-agent=transport=dt_socket,server=y,address=" + debugServerAddress;
+        std::vector<const char*> options {
+                "--soft-breakpoints",
+                connectionString.c_str()
+        };
+        mono_jit_parse_options(options.size(), (char**) options.data());
+
+        mono_debug_init(MONO_DEBUG_FORMAT_MONO);
+    }
+
+    domain = mono_jit_init("TES3MP Mono VM");
+
+    if (Script::IsDebugMode())
+    {
+        mono_debug_domain_create(domain);
+    }
+
     mono_add_internal_call("TES3MPSharp.TES3MP::CreateTimerEx", (void*) &LangMono::CreateTimerEx);
     mono_add_internal_call("TES3MPSharp.TES3MP::MakePublic", (void*) &LangMono::MakePublic);
     mono_add_internal_call("TES3MPSharp.TES3MP::CallPublic", (void*) &LangMono::CallPublic);
 }
 
+void LangMono::Free()
+{
+    mono_domain_free(domain, 0);
+}
+
 std::vector<MonoClass *> getInstanceClassList(MonoImage *image, const std::string &parentName)
 {
     std::vector<MonoClass *> classes;
diff --git a/apps/openmw-mp/Script/LangMono/LangMono.hpp b/apps/openmw-mp/Script/LangMono/LangMono.hpp
index 00269d228..e98a448a5 100644
--- a/apps/openmw-mp/Script/LangMono/LangMono.hpp
+++ b/apps/openmw-mp/Script/LangMono/LangMono.hpp
@@ -66,7 +66,7 @@ public:
     static int CreateTimerEx(MonoObject *delegate, long msec, MonoString *monoStr, MonoArray *args);
     static void MakePublic(MonoObject *delegate, const char *name) noexcept;
     static MonoObject *CallPublic(const char *name, MonoArray *args);
-
-private:
-    void Init();
+    
+    static void Init();
+    static void Free();
 };
diff --git a/apps/openmw-mp/Script/LangNative/LangNative.cpp b/apps/openmw-mp/Script/LangNative/LangNative.cpp
index 616557eb0..338052d82 100644
--- a/apps/openmw-mp/Script/LangNative/LangNative.cpp
+++ b/apps/openmw-mp/Script/LangNative/LangNative.cpp
@@ -100,3 +100,13 @@ LangNative::~LangNative()
 {
 
 }
+
+void LangNative::Init()
+{
+
+}
+
+void LangNative::Free()
+{
+
+}
diff --git a/apps/openmw-mp/Script/LangNative/LangNative.hpp b/apps/openmw-mp/Script/LangNative/LangNative.hpp
index bc1346ca1..7ac1a5f47 100644
--- a/apps/openmw-mp/Script/LangNative/LangNative.hpp
+++ b/apps/openmw-mp/Script/LangNative/LangNative.hpp
@@ -22,6 +22,8 @@ public:
     virtual boost::any Call(const char *name, const char *argl, int buf, ...) override;
     virtual boost::any Call(const char *name, const char *argl, const std::vector<boost::any> &args) override;
 
+    static void Init();
+    static void Free();
 };
 
 
diff --git a/apps/openmw-mp/Script/Script.cpp b/apps/openmw-mp/Script/Script.cpp
index 30f41a0ce..b660f4525 100644
--- a/apps/openmw-mp/Script/Script.cpp
+++ b/apps/openmw-mp/Script/Script.cpp
@@ -17,6 +17,7 @@ using namespace std;
 
 Script::ScriptList Script::scripts;
 std::string Script::moddir;
+bool Script::debugMode = false;
 
 inline bool Load(Language *lang, const std::string &path)
 {
@@ -133,3 +134,35 @@ const char* Script::GetModDir()
 {
     return moddir.c_str();
 }
+
+bool Script::IsDebugMode()
+{
+    return debugMode;
+}
+
+void Script::EnableDebugMode()
+{
+    debugMode = true;
+}
+
+void Script::Init()
+{
+#ifdef ENABLE_MONO
+    LangMono::Init();
+#endif
+#ifdef ENABLE_LUA
+    LangLua::Init();
+#endif
+    LangNative::Init();
+}
+
+void Script::Free()
+{
+#ifdef ENABLE_MONO
+    LangMono::Free();
+#endif
+#ifdef ENABLE_LUA
+    LangLua::Free();
+#endif
+    LangNative::Free();
+}
diff --git a/apps/openmw-mp/Script/Script.hpp b/apps/openmw-mp/Script/Script.hpp
index 360452707..916b63e9e 100644
--- a/apps/openmw-mp/Script/Script.hpp
+++ b/apps/openmw-mp/Script/Script.hpp
@@ -57,9 +57,14 @@ private:
 
 protected:
     static std::string moddir;
+    static bool debugMode;
 public:
     ~Script();
 
+    static void Init();
+    static void Free();
+    static bool IsDebugMode();
+    static void EnableDebugMode();
     static void LoadScript(const char *script, const char* base);
     static void LoadScripts(char* scripts, const char* base);
     static void UnloadScripts();
diff --git a/apps/openmw-mp/main.cpp b/apps/openmw-mp/main.cpp
index 279f2d56d..06210b738 100644
--- a/apps/openmw-mp/main.cpp
+++ b/apps/openmw-mp/main.cpp
@@ -134,6 +134,7 @@ boost::program_options::variables_map launchOptions(int argc, char *argv[], File
 
     desc.add_options()
             ("resources", bpo::value<Files::EscapeHashString>()->default_value("resources"), "set resources directory")
+            ("debug", bpo::value<bool>()->implicit_value(true)->default_value(false), "Enable debug mode for scripts (if supported)")
             ("no-logs", bpo::value<bool>()->implicit_value(true)->default_value(false),
              "Do not write logs. Useful for daemonizing.");
 
@@ -264,6 +265,9 @@ int main(int argc, char *argv[])
 
     try
     {
+        if (variables["debug"].as<bool>())
+            Script::EnableDebugMode();
+        Script::Init();
         for (auto plugin : plugins)
             Script::LoadScript(plugin.c_str(), plugin_home.c_str());
 
@@ -331,6 +335,7 @@ int main(int argc, char *argv[])
         code = networking.mainLoop();
 
         networking.getMasterClient()->Stop();
+        Script::Free();
     }
     catch (std::exception &e)
     {