src/arch.c :use Excutable file name to TargetName

yuk7 · Oct 19, 2017 · 19e1c92 · 19e1c92
1 parent 6711935
commit 19e1c92
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/src/arch.c b/src/arch.c
@@ -15,10 +15,17 @@ typedef int (WINAPI *REGISTERDISTRIBUTION)(PCWSTR,PCWSTR);
 wchar_t *GetLxUID(wchar_t *DistributionName,wchar_t *LxUID);
 
 
-int main()
+int main(int argc,char *argv[])
 {
-    //Set Target Name
-    wchar_t TargetName[] = L"Arch";
+    //Get file name of exe
+    char efpath[300];
+    if(GetModuleFileName(NULL,efpath,300) == 0)
+        return 1;
+    char efName[50];
+    _splitpath(efpath,NULL,NULL,efName,NULL);
+
+    wchar_t TargetName[30];
+    mbstowcs_s(NULL,TargetName,30,efName,_TRUNCATE);
 
 
     HMODULE hmod;