2020-02-18 23:11:01 +08:00
|
|
|
#define WIN32_LEAN_AND_MEAN
|
|
|
|
#include <Windows.h>
|
|
|
|
#include <shellapi.h>
|
|
|
|
|
|
|
|
#include <string_view>
|
|
|
|
|
2020-04-21 15:30:12 +08:00
|
|
|
#include <common/common.h>
|
|
|
|
#include <common/updating/updating.h>
|
2020-02-18 23:11:01 +08:00
|
|
|
|
|
|
|
#include <winrt/Windows.ApplicationModel.h>
|
|
|
|
#include <winrt/Windows.Storage.h>
|
2020-04-21 15:30:12 +08:00
|
|
|
#include <Msi.h>
|
|
|
|
|
|
|
|
#include "../runner/tray_icon.h"
|
|
|
|
#include "../runner/action_runner_utils.h"
|
2020-02-18 23:11:01 +08:00
|
|
|
|
2020-05-07 22:39:32 +08:00
|
|
|
#include "resource.h"
|
|
|
|
|
|
|
|
extern "C" IMAGE_DOS_HEADER __ImageBase;
|
|
|
|
|
2020-02-18 23:11:01 +08:00
|
|
|
int uninstall_msi_action()
|
|
|
|
{
|
2020-04-21 15:30:12 +08:00
|
|
|
const auto package_path = updating::get_msi_package_path();
|
2020-02-18 23:11:01 +08:00
|
|
|
if (package_path.empty())
|
|
|
|
{
|
|
|
|
return 0;
|
|
|
|
}
|
2020-04-21 15:30:12 +08:00
|
|
|
if (!updating::uninstall_msi_version(package_path))
|
2020-02-18 23:11:01 +08:00
|
|
|
{
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
// Launch PowerToys again, since it's been terminated by the MSI uninstaller
|
|
|
|
std::wstring runner_path{ winrt::Windows::ApplicationModel::Package::Current().InstalledLocation().Path() };
|
|
|
|
runner_path += L"\\PowerToys.exe";
|
|
|
|
SHELLEXECUTEINFOW sei{ sizeof(sei) };
|
|
|
|
sei.fMask = { SEE_MASK_FLAG_NO_UI | SEE_MASK_NOASYNC };
|
|
|
|
sei.lpFile = runner_path.c_str();
|
|
|
|
sei.nShow = SW_SHOWNORMAL;
|
|
|
|
ShellExecuteExW(&sei);
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2020-04-21 15:30:12 +08:00
|
|
|
namespace fs = std::filesystem;
|
|
|
|
|
|
|
|
std::optional<fs::path> copy_self_to_temp_dir()
|
|
|
|
{
|
|
|
|
std::error_code error;
|
|
|
|
auto dst_path = fs::temp_directory_path() / "action_runner.exe";
|
|
|
|
fs::copy_file(get_module_filename(), dst_path, fs::copy_options::overwrite_existing, error);
|
|
|
|
if (error)
|
|
|
|
{
|
|
|
|
return std::nullopt;
|
|
|
|
}
|
|
|
|
return std::move(dst_path);
|
|
|
|
}
|
|
|
|
|
|
|
|
bool install_new_version_stage_1(const bool must_restart = false)
|
|
|
|
{
|
|
|
|
std::optional<fs::path> installer;
|
|
|
|
for (auto path : fs::directory_iterator{ updating::get_pending_updates_path() })
|
|
|
|
{
|
|
|
|
if (path.path().native().find(updating::installer_filename_pattern) != std::wstring::npos)
|
|
|
|
{
|
|
|
|
installer.emplace(std::move(path));
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if (!installer)
|
|
|
|
{
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (auto copy_in_temp = copy_self_to_temp_dir())
|
|
|
|
{
|
|
|
|
// detect if PT was running
|
|
|
|
const auto pt_main_window = FindWindowW(pt_tray_icon_window_class, nullptr);
|
|
|
|
const bool launch_powertoys = must_restart || pt_main_window != nullptr;
|
|
|
|
if (pt_main_window != nullptr)
|
|
|
|
{
|
|
|
|
SendMessageW(pt_main_window, WM_CLOSE, 0, 0);
|
|
|
|
}
|
|
|
|
|
|
|
|
std::wstring arguments{ UPDATE_NOW_LAUNCH_STAGE2_CMDARG };
|
|
|
|
arguments += L" \"";
|
|
|
|
arguments += installer->c_str();
|
|
|
|
arguments += L"\" \"";
|
|
|
|
arguments += get_module_folderpath();
|
|
|
|
arguments += L"\" ";
|
|
|
|
arguments += launch_powertoys ? UPDATE_STAGE2_RESTART_PT_CMDARG : UPDATE_STAGE2_DONT_START_PT_CMDARG;
|
|
|
|
SHELLEXECUTEINFOW sei{ sizeof(sei) };
|
|
|
|
sei.fMask = { SEE_MASK_FLAG_NO_UI | SEE_MASK_NOASYNC };
|
|
|
|
sei.lpFile = copy_in_temp->c_str();
|
|
|
|
sei.nShow = SW_SHOWNORMAL;
|
|
|
|
|
|
|
|
sei.lpParameters = arguments.c_str();
|
|
|
|
return ShellExecuteExW(&sei) == TRUE;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
bool install_new_version_stage_2(std::wstring_view installer_path, std::wstring_view install_path, const bool launch_powertoys)
|
|
|
|
{
|
|
|
|
if (MsiInstallProductW(installer_path.data(), nullptr) != ERROR_SUCCESS)
|
|
|
|
{
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
std::error_code _;
|
|
|
|
fs::remove(installer_path, _);
|
|
|
|
if (launch_powertoys)
|
|
|
|
{
|
|
|
|
std::wstring new_pt_path{ install_path };
|
|
|
|
new_pt_path += L"\\PowerToys.exe";
|
|
|
|
SHELLEXECUTEINFOW sei{ sizeof(sei) };
|
|
|
|
sei.fMask = { SEE_MASK_FLAG_NO_UI | SEE_MASK_NOASYNC };
|
|
|
|
sei.lpFile = new_pt_path.c_str();
|
|
|
|
sei.nShow = SW_SHOWNORMAL;
|
|
|
|
sei.lpParameters = UPDATE_REPORT_SUCCESS;
|
|
|
|
return ShellExecuteExW(&sei) == TRUE;
|
|
|
|
}
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2020-05-07 22:39:32 +08:00
|
|
|
bool dotnet_is_installed()
|
|
|
|
{
|
|
|
|
auto runtimes = exec_and_read_output(LR"(dotnet --list-runtimes)");
|
|
|
|
if (!runtimes)
|
|
|
|
{
|
|
|
|
return false;
|
|
|
|
}
|
2020-05-14 04:05:13 +08:00
|
|
|
const char DESKTOP_DOTNET_RUNTIME_STRING[] = "Microsoft.WindowsDesktop.App 3.1.";
|
2020-05-07 22:39:32 +08:00
|
|
|
return runtimes->find(DESKTOP_DOTNET_RUNTIME_STRING) != std::string::npos;
|
|
|
|
}
|
|
|
|
|
2020-05-14 04:05:13 +08:00
|
|
|
bool install_dotnet()
|
2020-05-07 22:39:32 +08:00
|
|
|
{
|
2020-05-15 00:20:06 +08:00
|
|
|
const wchar_t DOTNET_DESKTOP_DOWNLOAD_LINK[] = L"https://download.visualstudio.microsoft.com/download/pr/d8cf1fe3-21c2-4baf-988f-f0152996135e/0c00b94713ee93e7ad5b4f82e2b86607/windowsdesktop-runtime-3.1.4-win-x64.exe";
|
|
|
|
const wchar_t DOTNET_DESKTOP_FILENAME[] = L"windowsdesktop-runtime-3.1.4-win-x64.exe";
|
2020-05-14 04:05:13 +08:00
|
|
|
|
|
|
|
auto dotnet_download_path = fs::temp_directory_path() / DOTNET_DESKTOP_FILENAME;
|
|
|
|
winrt::Windows::Foundation::Uri download_link{ DOTNET_DESKTOP_DOWNLOAD_LINK };
|
2020-05-07 22:39:32 +08:00
|
|
|
|
|
|
|
const size_t max_attempts = 3;
|
|
|
|
bool download_success = false;
|
|
|
|
for (size_t i = 0; i < max_attempts; ++i)
|
|
|
|
{
|
|
|
|
try
|
|
|
|
{
|
|
|
|
updating::try_download_file(dotnet_download_path, download_link).wait();
|
|
|
|
download_success = true;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
catch (...)
|
|
|
|
{
|
|
|
|
// couldn't download
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if (!download_success)
|
|
|
|
{
|
|
|
|
MessageBoxW(nullptr,
|
|
|
|
GET_RESOURCE_STRING(IDS_DOTNET_CORE_DOWNLOAD_FAILURE).c_str(),
|
|
|
|
GET_RESOURCE_STRING(IDS_DOTNET_CORE_DOWNLOAD_FAILURE_TITLE).c_str(),
|
|
|
|
MB_OK | MB_ICONERROR);
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
SHELLEXECUTEINFOW sei{ sizeof(sei) };
|
2020-05-14 04:05:13 +08:00
|
|
|
sei.fMask = { SEE_MASK_NOASYNC };
|
2020-05-07 22:39:32 +08:00
|
|
|
sei.lpFile = dotnet_download_path.c_str();
|
|
|
|
sei.nShow = SW_SHOWNORMAL;
|
|
|
|
sei.lpParameters = L"/install /passive";
|
2020-05-14 04:05:13 +08:00
|
|
|
return ShellExecuteExW(&sei) == TRUE;
|
2020-05-07 22:39:32 +08:00
|
|
|
}
|
|
|
|
|
2020-02-18 23:11:01 +08:00
|
|
|
int WINAPI WinMain(HINSTANCE, HINSTANCE, LPSTR, int)
|
|
|
|
{
|
|
|
|
int nArgs = 0;
|
|
|
|
LPWSTR* args = CommandLineToArgvW(GetCommandLineW(), &nArgs);
|
|
|
|
if (!args || nArgs < 2)
|
|
|
|
{
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
std::wstring_view action{ args[1] };
|
|
|
|
|
2020-05-22 01:44:32 +08:00
|
|
|
if (action == L"-start_PowerLauncher")
|
|
|
|
{
|
|
|
|
HANDLE hMapFile = OpenFileMappingW(FILE_MAP_WRITE, FALSE, POWER_LAUNCHER_PID_SHARED_FILE);
|
2020-05-30 01:02:37 +08:00
|
|
|
if (hMapFile)
|
2020-05-22 01:44:32 +08:00
|
|
|
{
|
2020-05-30 01:02:37 +08:00
|
|
|
PDWORD pidBuffer = reinterpret_cast<PDWORD>(MapViewOfFile(hMapFile, FILE_MAP_ALL_ACCESS, 0, 0, sizeof(DWORD)));
|
|
|
|
if (pidBuffer)
|
|
|
|
{
|
|
|
|
*pidBuffer = 0;
|
|
|
|
run_same_elevation(L"modules\\launcher\\PowerLauncher.exe", L"", pidBuffer);
|
|
|
|
FlushViewOfFile(pidBuffer, sizeof(DWORD));
|
|
|
|
UnmapViewOfFile(pidBuffer);
|
|
|
|
}
|
|
|
|
|
|
|
|
FlushFileBuffers(hMapFile);
|
|
|
|
CloseHandle(hMapFile);
|
2020-05-22 01:44:32 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
else if (action == L"-install_dotnet")
|
2020-05-07 22:39:32 +08:00
|
|
|
{
|
|
|
|
if (dotnet_is_installed())
|
|
|
|
{
|
|
|
|
return 0;
|
|
|
|
}
|
2020-05-14 04:05:13 +08:00
|
|
|
return !install_dotnet();
|
2020-05-07 22:39:32 +08:00
|
|
|
}
|
|
|
|
else if (action == L"-uninstall_msi")
|
2020-02-18 23:11:01 +08:00
|
|
|
{
|
|
|
|
return uninstall_msi_action();
|
|
|
|
}
|
2020-04-21 15:30:12 +08:00
|
|
|
else if (action == UPDATE_NOW_LAUNCH_STAGE1_CMDARG)
|
|
|
|
{
|
|
|
|
return !install_new_version_stage_1();
|
|
|
|
}
|
|
|
|
else if (action == UPDATE_NOW_LAUNCH_STAGE1_START_PT_CMDARG)
|
|
|
|
{
|
|
|
|
return !install_new_version_stage_1(true);
|
|
|
|
}
|
|
|
|
else if (action == UPDATE_NOW_LAUNCH_STAGE2_CMDARG)
|
|
|
|
{
|
|
|
|
using namespace std::string_view_literals;
|
|
|
|
return !install_new_version_stage_2(args[2], args[3], args[4] == std::wstring_view{ UPDATE_STAGE2_RESTART_PT_CMDARG });
|
|
|
|
}
|
2020-02-18 23:11:01 +08:00
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|