Invent a value for argv[0] if not present

[libs/core.git] / source / core / application.cpp

diff --git a/source/core/application.cpp b/source/core/application.cpp
index d5745ed8a36ae2cbe139a14de0f1ba76c1b22d5c..aedaa4b6f25c027429f8bff1a5c069078206e7df 100644 (file)
--- a/source/core/application.cpp
+++ b/source/core/application.cpp
@@ -1,7 +1,11 @@
+#include <cstring>
 #include <typeinfo>
 #include <signal.h>
 #include <msp/debug/demangle.h>
 #include <msp/debug/errorreporter.h>
+#include <msp/fs/dir.h>
+#include <msp/fs/path.h>
+#include <msp/fs/utils.h>
 #include <msp/io/print.h>
 #include "application.h"
 #include "getopt.h"
@@ -12,16 +16,23 @@ namespace Msp {
 
 Application *Application::app_ = 0;
 Application::Starter *Application::starter_ = 0;
+const char *Application::argv0_ = 0;
+string Application::name_;
 void *Application::data_ = 0;
 
-Application::Application():
+Application::Application(const string &n):
        exit_code(0)
 {
        if(app_)
                throw logic_error("instance already exists");
+
+       if(!n.empty())
+               name_ = n;
+       else
+               name_ = FS::basename(argv0_);
 }
 
-int Application::run(int argc, char **argv, void *data)
+int Application::run(int argc, char **argv, void *data, void (*created_callback)(void *))
 {
        if(!starter_)
        {
@@ -29,7 +40,7 @@ int Application::run(int argc, char **argv, void *data)
                return 126;
        }
 
-       data_ = data;
+       set_startup_info(argv[0], data);
 
        try
        {
@@ -43,6 +54,9 @@ int Application::run(int argc, char **argv, void *data)
                        return 1;
                }
 
+               if(created_callback)
+                       created_callback(data);
+
                int result = app_->main();
                Application *a = app_;
                app_ = 0;
@@ -69,6 +83,32 @@ int Application::run(int argc, char **argv, void *data)
        }
 }
 
+void Application::set_startup_info(const char *argv0, void *data)
+{
+       if(argv0_)
+               throw logic_error("startup info already set");
+
+       static FS::Path exe;
+
+       if(!argv0 || !*argv0)
+       {
+#ifdef _WIN32
+               argv0 = "application.exe";
+#else
+               argv0 = "./application";
+#endif
+       }
+
+       bool has_slash = strchr(argv0, FS::DIRSEP);
+       if(!has_slash)
+               exe = FS::path_lookup(argv0);
+       if(exe.empty())
+               exe = FS::realpath(argv0);
+
+       argv0_ = exe.c_str();
+       data_ = data;
+}
+
 int Application::main()
 {
        done = false;