Added nssm_exe().

New function to retrieve argv[0].
puppetlabs · Jul 15, 2016 · caaaa12 · caaaa12
1 parent 6adc886
commit caaaa12
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/nssm.cpp b/nssm.cpp
@@ -6,6 +6,7 @@ extern imports_t imports;
 
 static TCHAR unquoted_imagepath[PATH_LENGTH];
 static TCHAR imagepath[PATH_LENGTH];
+static TCHAR imageargv0[PATH_LENGTH];
 
 /* Are two strings case-insensitively equivalent? */
 int str_equiv(const TCHAR *a, const TCHAR *b) {
@@ -104,6 +105,10 @@ const TCHAR *nssm_imagepath() {
   return imagepath;
 }
 
+const TCHAR *nssm_exe() {
+  return imageargv0;
+}
+
 int _tmain(int argc, TCHAR **argv) {
   check_console();
 
@@ -123,6 +128,8 @@ int _tmain(int argc, TCHAR **argv) {
   if (get_imports()) exit(111);
 
   /* Remember our path for later. */
+  _sntprintf_s(imageargv0, _countof(imageargv0), _TRUNCATE, _T("%s"), argv[0]);
+  PathQuoteSpaces(imageargv0);
   GetModuleFileName(0, unquoted_imagepath, _countof(unquoted_imagepath));
   GetModuleFileName(0, imagepath, _countof(imagepath));
   PathQuoteSpaces(imagepath);

diff --git a/nssm.h b/nssm.h
@@ -69,6 +69,7 @@ int num_cpus();
 int usage(int);
 const TCHAR *nssm_unquoted_imagepath();
 const TCHAR *nssm_imagepath();
+const TCHAR *nssm_exe();
 
 #define NSSM _T("NSSM")
 #ifdef _WIN64