support abort compiling

Author: Islam0mar

src/rcrl/rcrl.cpp

@@ -29,7 +29,7 @@ typedef HMODULE RCRL_Dynlib;
 
 #include <dlfcn.h>
 typedef void* RCRL_Dynlib;
-#define RDRL_LoadDynlib(lib) dlopen(lib, RTLD_LAZY | RTLD_GLOBAL)
+#define RDRL_LoadDynlib(lib) dlopen(lib, RTLD_NOW | RTLD_GLOBAL)
 #define RCRL_CloseDynlib dlclose
 #endif
 
@@ -53,7 +53,9 @@ auto CopyFileToString(const string& f_name) {
   return out;
 }
 Plugin::Plugin(fs::path file, std::vector<string> flags)
-    : is_compiling_(false), parser_(file.string() + ".cpp", flags) {
+    : is_compiling_(false),
+      parser_(file.string() + ".cpp", flags),
+      abort_(false) {
   auto header = parser_.get_file().replace_extension(".hpp");
   std::ofstream f(header, std::fstream::trunc | std::fstream::out);
   f << "#pragma once\n";
@@ -181,15 +183,29 @@ bool Plugin::CompileCode(string code) {
       return lambda_impl(ec, size, lambda_impl);
     };
     ap.async_read_some(output_buffer, OnStdout);
-    ios.run();
-    c.join();
+    // support aborting compiling
+    while (!ios.stopped() && !abort_) {
+      ios.run_for(std::chrono::milliseconds(500));
+    }
+    if (abort_) {
+      ios.stop();
+      c.terminate();
+      abort_ = false;
+    } else {
+      c.join();
+    }
     is_compiling_ = false;
     return c.exit_code();
   });
   return true;
 }
 
 bool Plugin::IsCompiling() { return is_compiling_; }
+void Plugin::AbortCompiling() {
+  if (IsCompiling()) {
+    abort_ = true;
+  }
+}
 
 bool Plugin::TryGetExitStatusFromCompile(int& exit_code) {
   if (compiler_process_.valid() && !IsCompiling()) {
@@ -231,18 +247,18 @@ string Plugin::CopyAndLoadNewPlugin(bool redirect_stdout) {
     fd = dup(fileno(stdout));
     freopen(kRcrlOutputFile.c_str(), "w", stdout);
   }
+
+  string out;
+
   // load the plugin
   auto plugin = RDRL_LoadDynlib(name_copied.c_str());
+
   if (!plugin) {
-    fprintf(stderr, "%s\n", dlerror());
-    exit(EXIT_FAILURE);
+    compiler_output_ = string("Library loading error: ") + dlerror();
+  } else {
+    // add the plugin to the list of loaded ones - for later unloading
+    plugins_.push_back({name_copied, plugin});
   }
-  assert(plugin);
-
-  // add the plugin to the list of loaded ones - for later unloading
-  plugins_.push_back({name_copied, plugin});
-
-  string out;
 
   if (redirect_stdout) {
     // Flush stdout so any buffered messages are delivered

src/rcrl/rcrl.h

@@ -22,6 +22,7 @@ class Plugin {
   string CleanupPlugins(bool redirect_stdout = false);
   bool CompileCode(string code);
   bool IsCompiling();
+  void AbortCompiling();
 
   bool TryGetExitStatusFromCompile(int& exitcode);
   string CopyAndLoadNewPlugin(bool redirect_stdout = false);
@@ -37,6 +38,7 @@ class Plugin {
   std::future<int> compiler_process_;
   bool last_compile_successful_ = false;
   PluginParser parser_;
+  std::atomic_bool abort_;
 };
 
 }  // namespace rcrl