Files
deepseek-harness/apps/desktop/tests/windows-uninstall-data.cpp
T
winewill dbbecd2461 feat(desktop): opt-in Windows uninstall data removal
The Windows uninstaller keeps every user file by default and adds a custom
page with unchecked options for desktop browser data and update caches, the
desktop plugin profile, and the complete Harness home. Selecting the whole
home requires a native warning that names the actual folder and the CLI
impact, defaults to Cancel, and forces the desktop child selection. Silent
and upgrade uninstalls never remove data, including through the legacy
--delete-app-data flag, which the adapted upstream template no longer honors.

Packaged Desktop records its resolved home in uninstall.ini under the
Electron user-data directory so a custom DSH_HOME is offered instead of the
default; a length field detects truncated NSIS reads and an invalid record
disables whole-home removal. The native remover in window-frame.dll rejects
protected roots and overlap with retained data, holds directory handles
during traversal, and unlinks reparse points without entering their targets.

Installation now also writes InstallLocation on the Windows uninstall entry,
which Explorer needs to associate the Start menu shortcut with this entry so
that the context-menu Uninstall launches the uninstaller directly instead of
opening the installed-apps list. The upstream template only records the
value under its private key.

Closes #3993
2026-09-21 14:22:39 +08:00

51 lines
2.6 KiB
C++

// Native cleanup regressions operate only in the caller's private test directory.
#define WIN32_LEAN_AND_MEAN
#define NOMINMAX
#define UNICODE
#include <windows.h>
#include <cassert>
#include "../installer/uninstall-data.h"
static void File(const std::wstring& path) {
HANDLE file = CreateFileW(path.c_str(), GENERIC_WRITE, 0, nullptr, CREATE_NEW, FILE_ATTRIBUTE_NORMAL, nullptr);
assert(file != INVALID_HANDLE_VALUE);
DWORD count;
assert(WriteFile(file, "retained", 8, &count, nullptr));
CloseHandle(file);
}
int wmain(int argc, wchar_t** argv) {
assert(argc == 2);
const std::wstring root = argv[1];
const std::wstring installation = root + L"\\application";
const std::wstring home = root + L"\\home";
assert(CreateDirectoryW(root.c_str(), nullptr));
assert(CreateDirectoryW(installation.c_str(), nullptr));
assert(CreateDirectoryW(home.c_str(), nullptr));
assert(CreateDirectoryW((home + L"\\profiles").c_str(), nullptr));
const std::wstring desktop = home + L"\\profiles\\desktop";
assert(CreateDirectoryW(desktop.c_str(), nullptr));
File(home + L"\\session.jsonl");
File(desktop + L"\\plugin.json");
assert(uninstall_data::Remove(root.c_str(), installation.c_str()) == ERROR_ACCESS_DENIED);
assert(uninstall_data::Remove(L"C:\\", installation.c_str()) == ERROR_INVALID_NAME);
assert(uninstall_data::Remove(L"relative", installation.c_str()) == ERROR_INVALID_NAME);
assert(uninstall_data::Remove(L"C:\\bad*", installation.c_str()) == ERROR_INVALID_NAME);
PWSTR profile = nullptr;
assert(SUCCEEDED(SHGetKnownFolderPath(FOLDERID_Profile, 0, nullptr, &profile)));
assert(uninstall_data::Remove(profile, installation.c_str()) == ERROR_ACCESS_DENIED);
CoTaskMemFree(profile);
HANDLE held = CreateFileW((desktop + L"\\plugin.json").c_str(), GENERIC_READ, FILE_SHARE_READ,
nullptr, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, nullptr);
assert(held != INVALID_HANDLE_VALUE);
assert(uninstall_data::Remove(desktop.c_str(), installation.c_str()) == ERROR_SHARING_VIOLATION);
CloseHandle(held);
assert(uninstall_data::Remove(desktop.c_str(), installation.c_str()) == ERROR_SUCCESS);
assert(GetFileAttributesW((home + L"\\session.jsonl").c_str()) != INVALID_FILE_ATTRIBUTES);
assert(uninstall_data::Remove(desktop.c_str(), installation.c_str()) == ERROR_SUCCESS);
assert(uninstall_data::Remove(home.c_str(), installation.c_str()) == ERROR_SUCCESS);
assert(GetFileAttributesW(home.c_str()) == INVALID_FILE_ATTRIBUTES);
assert(RemoveDirectoryW(installation.c_str()));
assert(RemoveDirectoryW(root.c_str()));
}