Added nssm_exe().

New function to retrieve argv[0].
master
Iain Patterson 9 years ago
parent 6adc886e1f
commit caaaa1208f

@ -6,6 +6,7 @@ extern imports_t imports;
static TCHAR unquoted_imagepath[PATH_LENGTH]; static TCHAR unquoted_imagepath[PATH_LENGTH];
static TCHAR imagepath[PATH_LENGTH]; static TCHAR imagepath[PATH_LENGTH];
static TCHAR imageargv0[PATH_LENGTH];
/* Are two strings case-insensitively equivalent? */ /* Are two strings case-insensitively equivalent? */
int str_equiv(const TCHAR *a, const TCHAR *b) { int str_equiv(const TCHAR *a, const TCHAR *b) {
@ -104,6 +105,10 @@ const TCHAR *nssm_imagepath() {
return imagepath; return imagepath;
} }
const TCHAR *nssm_exe() {
return imageargv0;
}
int _tmain(int argc, TCHAR **argv) { int _tmain(int argc, TCHAR **argv) {
check_console(); check_console();
@ -123,6 +128,8 @@ int _tmain(int argc, TCHAR **argv) {
if (get_imports()) exit(111); if (get_imports()) exit(111);
/* Remember our path for later. */ /* 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, unquoted_imagepath, _countof(unquoted_imagepath));
GetModuleFileName(0, imagepath, _countof(imagepath)); GetModuleFileName(0, imagepath, _countof(imagepath));
PathQuoteSpaces(imagepath); PathQuoteSpaces(imagepath);

@ -69,6 +69,7 @@ int num_cpus();
int usage(int); int usage(int);
const TCHAR *nssm_unquoted_imagepath(); const TCHAR *nssm_unquoted_imagepath();
const TCHAR *nssm_imagepath(); const TCHAR *nssm_imagepath();
const TCHAR *nssm_exe();
#define NSSM _T("NSSM") #define NSSM _T("NSSM")
#ifdef _WIN64 #ifdef _WIN64

Loading…
Cancel
Save