Mirrors/velopack
1
0
Fork 0
mirror of https://github.com/velopack/velopack.git synced 2025-10-25 15:19:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

Added support for environment variable SQUIRREL_TEMP to Setup.exe

Browse Source 
Allows to control where Setup.exe will extract and execute the package
This commit is contained in:
Lennart Jansson
2016-10-05 12:57:03 +02:00
parent 6722c66f77
commit c3def2a7c3
1 changed files with 28 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/Setup/UpdateRunner.cpp
View File

@@ -137,6 +137,11 @@ int CUpdateRunner::ExtractUpdaterAndRun(wchar_t* lpCommandLine, bool useFallback
wchar_t logFile[MAX_PATH]; wchar_t logFile[MAX_PATH];
std::vector<CString> to_delete; std::vector<CString> to_delete;
wchar_t *envSquirrelTemp = _wgetenv(L"SQUIRREL_TEMP");
if (envSquirrelTemp) {
_swprintf_c(targetDir, _countof(targetDir), L"%s", envSquirrelTemp);
}
else {
if (!useFallbackDir) { if (!useFallbackDir) {
SHGetFolderPath(NULL, CSIDL_LOCAL_APPDATA, NULL, SHGFP_TYPE_CURRENT, targetDir); SHGetFolderPath(NULL, CSIDL_LOCAL_APPDATA, NULL, SHGFP_TYPE_CURRENT, targetDir);
} else { } else {
@@ -161,6 +166,7 @@ int CUpdateRunner::ExtractUpdaterAndRun(wchar_t* lpCommandLine, bool useFallback
} }
wcscat_s(targetDir, _countof(targetDir), L"\\SquirrelTemp"); wcscat_s(targetDir, _countof(targetDir), L"\\SquirrelTemp");
}
if (!CreateDirectory(targetDir, NULL) && GetLastError() != ERROR_ALREADY_EXISTS) { if (!CreateDirectory(targetDir, NULL) && GetLastError() != ERROR_ALREADY_EXISTS) {
wchar_t err[4096]; wchar_t err[4096];