Compare commits
3 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| fa2f6d2ce4 | |||
| 30e70192dc | |||
| 94cab4c112 |
+13
-1
@@ -1,5 +1,5 @@
|
|||||||
cmake_minimum_required(VERSION 3.24)
|
cmake_minimum_required(VERSION 3.24)
|
||||||
project(KSwitcher VERSION 0.6.35 LANGUAGES CXX RC)
|
project(KSwitcher VERSION 0.6.38 LANGUAGES CXX RC)
|
||||||
if(NOT WIN32)
|
if(NOT WIN32)
|
||||||
message(FATAL_ERROR "KSwitcher uses native Windows APIs")
|
message(FATAL_ERROR "KSwitcher uses native Windows APIs")
|
||||||
endif()
|
endif()
|
||||||
@@ -48,6 +48,7 @@ else()
|
|||||||
endif()
|
endif()
|
||||||
add_executable(KSwitcherOcrTests tests/ocr.cpp src/screenshot_launcher.cpp src/screenshot_activation.cpp src/app.cpp src/screenshot_notifications.cpp src/screenshot.cpp src/configuration.cpp src/ui.cpp resources/app.rc)
|
add_executable(KSwitcherOcrTests tests/ocr.cpp src/screenshot_launcher.cpp src/screenshot_activation.cpp src/app.cpp src/screenshot_notifications.cpp src/screenshot.cpp src/configuration.cpp src/ui.cpp resources/app.rc)
|
||||||
target_link_libraries(KSwitcherOcrTests PRIVATE kscore)
|
target_link_libraries(KSwitcherOcrTests PRIVATE kscore)
|
||||||
|
add_dependencies(KSwitcherOcrTests KSwitcherScreenshot)
|
||||||
if(MSVC)
|
if(MSVC)
|
||||||
set_property(TARGET kscore KSwitcher KSwitcherConfig KSwitcherScreenshot KSwitcherTests KSwitcherScrollTests KSwitcherIntegration KSwitcherInputTests KSwitcherOcrTests PROPERTY MSVC_RUNTIME_LIBRARY "MultiThreaded$<$<CONFIG:Debug>:Debug>")
|
set_property(TARGET kscore KSwitcher KSwitcherConfig KSwitcherScreenshot KSwitcherTests KSwitcherScrollTests KSwitcherIntegration KSwitcherInputTests KSwitcherOcrTests PROPERTY MSVC_RUNTIME_LIBRARY "MultiThreaded$<$<CONFIG:Debug>:Debug>")
|
||||||
target_compile_options(kscore PRIVATE /W4 /utf-8)
|
target_compile_options(kscore PRIVATE /W4 /utf-8)
|
||||||
@@ -65,6 +66,17 @@ else()
|
|||||||
target_link_options(KSwitcherOcrTests PRIVATE -static -static-libgcc -static-libstdc++)
|
target_link_options(KSwitcherOcrTests PRIVATE -static -static-libgcc -static-libstdc++)
|
||||||
endif()
|
endif()
|
||||||
enable_testing()
|
enable_testing()
|
||||||
|
add_executable(KSwitcherImageCaptureTests tests/image_capture.cpp)
|
||||||
|
target_link_libraries(KSwitcherImageCaptureTests PRIVATE kscore)
|
||||||
|
add_dependencies(KSwitcherImageCaptureTests KSwitcherScreenshot)
|
||||||
|
if(MSVC)
|
||||||
|
target_compile_options(KSwitcherImageCaptureTests PRIVATE /utf-8)
|
||||||
|
set_property(TARGET KSwitcherImageCaptureTests PROPERTY MSVC_RUNTIME_LIBRARY "MultiThreaded$<$<CONFIG:Debug>:Debug>")
|
||||||
|
else()
|
||||||
|
target_link_options(KSwitcherImageCaptureTests PRIVATE -static -static-libgcc -static-libstdc++)
|
||||||
|
endif()
|
||||||
|
# Explicit interactive test: stop the resident first so its clipboard listener
|
||||||
|
# does not capture the test fixtures. Like KSwitcherIntegration, not part of ctest.
|
||||||
add_executable(KSwitcherDictionaryTests tests/dictionary.cpp)
|
add_executable(KSwitcherDictionaryTests tests/dictionary.cpp)
|
||||||
target_link_libraries(KSwitcherDictionaryTests PRIVATE kscore psapi)
|
target_link_libraries(KSwitcherDictionaryTests PRIVATE kscore psapi)
|
||||||
if(MSVC)
|
if(MSVC)
|
||||||
|
|||||||
@@ -707,3 +707,21 @@ supporting old clients; their immutable update code only reads that page.
|
|||||||
### Editors that omit copied trailing spaces
|
### Editors that omit copied trailing spaces
|
||||||
|
|
||||||
From v0.6.31, KSwitcher recovers trailing ordinary spaces omitted by Ctrl+C only when Windows accessibility proves the complete selection. The same focused control and range are rechecked before paste, and the result is read back even without Write log. Recent-text conversion, automatic correction, repeat conversion and Undo retain exact verification. Missing letters, tabs, line breaks and unavailable or ambiguous selections are not reconstructed. No setting or shortcut has changed.
|
From v0.6.31, KSwitcher recovers trailing ordinary spaces omitted by Ctrl+C only when Windows accessibility proves the complete selection. The same focused control and range are rechecked before paste, and the result is read back even without Write log. Recent-text conversion, automatic correction, repeat conversion and Undo retain exact verification. Missing letters, tabs, line breaks and unavailable or ambiguous selections are not reconstructed. No setting or shortcut has changed.
|
||||||
|
|
||||||
|
### Notification registration and Viber — v0.6.36
|
||||||
|
|
||||||
|
Notification registration runs in KSwitcher.Screenshot.exe; keep all three executables together. Startup registration is asynchronous. Viber pastes allow 75 ms for clipboard notifications before Ctrl+V, retaining focus, ownership and selection checks. This mitigates an intermittent clipboard timing issue; it does not guarantee all Viber failures are resolved. UI Automation is unchanged.
|
||||||
|
|
||||||
|
### Image history, OCR and duplicate fingerprints — v0.6.37
|
||||||
|
|
||||||
|
Image clipboard capture and automatic/uncached manual history OCR run in short-lived KSwitcher.Screenshot.exe processes. Keep all three executables together. Original clipboard formats, image limits, privacy exclusions, Windows OCR, optional Tesseract and searchable encrypted OCR results are preserved. Background OCR is cancelled on shutdown or an OCR-profile change; late results do not recreate deleted entries.
|
||||||
|
|
||||||
|
Consecutive duplicates use SHA-256 fingerprints inside the encrypted history index, including pixel fingerprints for equivalent image encodings. Older entries acquire fingerprints on first comparison without a bulk migration. Pins and OCR caches survive duplicate detection; non-consecutive copies still create separate entries. New text no longer loads the preceding image to rule out a duplicate. No shortcut, setting or default changed.
|
||||||
|
|
||||||
|
Moving OCR isolates its temporary memory rather than reducing the combined processing peak. A controlled three-run comparison retained about 0.8 MiB less private RAM in the caller after recognition; combined peak RAM was about 55.6 MiB before and 56.6 MiB with the helper. Results depend on the image, OCR engine and system.
|
||||||
|
|
||||||
|
### History previews and Escape — v0.6.38
|
||||||
|
|
||||||
|
History initializes RichEdit only for HTML/RTF previews. Plain text keeps Unicode and line breaks without RTF conversion; empty history and image-only previews do not load RichEdit unnecessarily. Plain and formatted text previews share the existing bounded text-preview cache.
|
||||||
|
|
||||||
|
Esc clears hidden rows/caches and cancels obsolete queued preview work, while retaining the selected/top entry for reopening. Already-needed RichEdit controls are cleared and reused. Clipboard contents, paste formatting, OCR, settings and shortcuts are unchanged. A local isolated comparison saved approximately 0.48–0.72 MiB private working set after Esc depending on content; this is not a fixed application footprint or a guarantee that all UI memory returns to the startup level.
|
||||||
|
|||||||
Binary file not shown.
+7
-8
@@ -1,18 +1,17 @@
|
|||||||
# Latest Windows release
|
# Latest Windows release
|
||||||
|
|
||||||
## v0.6.35 — 2026-11-04
|
## v0.6.38 — 2026-11-04
|
||||||
|
|
||||||
## Changes
|
## Changes
|
||||||
- Reduced resident memory: the input predictor and correction worker share immutable dictionaries. Compact indexed blocks retain all 140,939 bundled words and their frequency ranks while reducing the EN/UK/RU lookup payload from approximately 3.06 MiB per copy to 0.90 MiB shared. Custom UTF-8 dictionary files, duplicate handling, frequency limits and personal dictionaries remain compatible.
|
- Clipboard history loads RichEdit only when an HTML/RTF preview actually needs it. Empty history, plain text and image-only previews no longer initialize it unnecessarily.
|
||||||
- Closing clipboard history releases its duplicate row list and Windows list strings, retains selection/scroll identifiers for reopening, and rejects late search results and timers. Preview caches now have payload budgets of 1 MiB for thumbnails and 256 KiB for formatted text, plus the existing item-count limit. Visible entries are retained even if an unusually large viewport exceeds a budget.
|
- Plain text previews retain Unicode and line breaks without converting the text to RTF. Plain and formatted preview payloads share the existing bounded text-preview cache.
|
||||||
- PNG decoding reads the existing input buffer without making another full PNG copy. DIB decoding allocates its temporary BMP buffer once at the required size.
|
- Esc still clears hidden history rows and preview caches while preserving the remembered selection/top item. Closing or replacing a search now cancels obsolete preview work before loading/decoding where possible, and discards completed stale results.
|
||||||
- Idle maintenance releases diagnostic context older than 30 seconds and expired Undo text after its existing two-minute eligibility window, without interrupting an active operation. Completed sync buffers are released on success, failure and stale responses; unresolved conflicts remain available.
|
- A RichEdit control that was needed for formatted text is cleared and reused, not repeatedly destroyed/recreated. This avoids additional initialization on reopening. Settings, shortcuts, stored clipboard contents, formatting on paste and OCR behavior are unchanged.
|
||||||
- No shortcut, default, on-disk history format or screenshot-editing behavior changed. The v0.6.34 four-direction movement cursors remain included.
|
|
||||||
|
|
||||||
## Validation
|
## Validation
|
||||||
The full `build.ps1 -Stage -Test` suite covers complete bundled dictionary word/rank equivalence, malformed/custom dictionary boundaries, 105 sync checks, shared wheel batching, 304 native regressions, 559 input/UI checks, screenshot editor UI and PNG saving, isolated screenshot registration, 48 updater regressions and updater progress UI checks. Cache assertions now run once outside the exception helper rather than being repeated inside it. All three executable versions and all 12 package manifest file sizes/SHA-256 hashes were verified; no personal data is packaged.
|
Staged build and tests cover dictionary equivalence for 140,939 words, 105 sync checks, wheel batching, 315 native checks, 561 input checks, screenshot UI/PNG/notification registration, 48 updater checks and updater progress UI. Added checks for plain previews without RichEdit, retained HTML rendering, lazy control creation and cleanup through Esc. Isolated real-window tests also verified formatted previews after reopening.
|
||||||
|
|
||||||
Controlled local measurements of the final dictionary step, after equal restart/wait periods with three dictionaries enabled, reduced private resident memory from 7.08 to 4.65 MiB and private committed memory from 8.34 to 5.70 MiB. These compare consecutive development builds, not the published v0.6.34 with the full cumulative change. Randomized lookup tests were approximately 20% faster. Results depend on settings, workload, Windows and allocator state; these are not fixed memory limits. The 0.90 MiB dictionary figure excludes process/allocator overhead.
|
A controlled local comparison measured 0.72 MiB less private working set after Esc for empty history, 0.66 MiB for plain text, 0.48 MiB for formatted text and 0.53 MiB for images. These are isolated test-process comparisons, not the application's total RAM or a guaranteed reduction on another computer. The reported 7 MiB plateau on a different laptop was not reproduced. No working-set trimming was used; closing the history window does not promise an exact return to startup memory.
|
||||||
|
|
||||||
## Upgrade
|
## Upgrade
|
||||||
Choose **Update** in the tray menu. Legacy v0.3.0–v0.4.1 clients first receive the v0.4.2 compatibility bridge, then the current full release. v0.2.3 and earlier require manual installation.
|
Choose **Update** in the tray menu. Legacy v0.3.0–v0.4.1 clients first receive the v0.4.2 compatibility bridge, then the current full release. v0.2.3 and earlier require manual installation.
|
||||||
|
|||||||
@@ -1 +1 @@
|
|||||||
6e36a582eec59b3cc4f0f9d6b3e05cd699371c05cae30a95c6eae51a838f7bfe KSwitcher-win-x64.zip
|
f430767d440581018e6adb33c153fb332318fa1eee5a162ddf32690d0b5a3391 KSwitcher-win-x64.zip
|
||||||
|
|||||||
@@ -37,7 +37,7 @@ try {
|
|||||||
if ($LASTEXITCODE) { throw 'Configurator link failed' }
|
if ($LASTEXITCODE) { throw 'Configurator link failed' }
|
||||||
Copy-Item -LiteralPath $configPath -Destination build/release/KSwitcher.Config.exe -Force
|
Copy-Item -LiteralPath $configPath -Destination build/release/KSwitcher.Config.exe -Force
|
||||||
$screenshotPath = Join-Path (Split-Path $executablePath) 'KSwitcher.Screenshot.exe'
|
$screenshotPath = Join-Path (Split-Path $executablePath) 'KSwitcher.Screenshot.exe'
|
||||||
& $compiler -mwindows -municode -static -s '-Wl,--gc-sections' build/core.o build/platform.o build/ocr.o build/screenshot.o build/screenshot_notifications.o build/screenshot_activation.o build/screenshot_main.o build/screenshot-res.o @libs -o $screenshotPath
|
& $compiler -mwindows -municode -static -s '-Wl,--gc-sections' build/core.o build/platform.o build/storage.o build/ocr.o build/screenshot.o build/screenshot_notifications.o build/screenshot_activation.o build/screenshot_main.o build/screenshot-res.o @libs -o $screenshotPath
|
||||||
if ($LASTEXITCODE) { throw 'Screenshot link failed' }
|
if ($LASTEXITCODE) { throw 'Screenshot link failed' }
|
||||||
Copy-Item -LiteralPath $screenshotPath -Destination build/release/KSwitcher.Screenshot.exe -Force
|
Copy-Item -LiteralPath $screenshotPath -Destination build/release/KSwitcher.Screenshot.exe -Force
|
||||||
if ($Test) {
|
if ($Test) {
|
||||||
@@ -81,15 +81,21 @@ try {
|
|||||||
if ($LASTEXITCODE) { throw 'Updater progress UI tests failed' }
|
if ($LASTEXITCODE) { throw 'Updater progress UI tests failed' }
|
||||||
}
|
}
|
||||||
if ($Integration) {
|
if ($Integration) {
|
||||||
|
$captureTest = Join-Path (Split-Path $executablePath) 'KSwitcherImageCaptureTests.exe'
|
||||||
|
& $compiler @common tests/image_capture.cpp build/core.o build/platform.o build/storage.o build/ocr.o @libs -static -s -o $captureTest
|
||||||
|
if ($LASTEXITCODE) { throw 'Image capture test build failed' }
|
||||||
|
& $captureTest
|
||||||
|
if ($LASTEXITCODE) { throw 'Image capture tests failed' }
|
||||||
& $compiler @common tests/integration.cpp build/core.o build/platform.o build/autostart.o build/autocorrect.o build/storage.o build/sync.o build/ocr.o build/richtext.o build/update.o build/screenshot.o build/screenshot_launcher.o build/screenshot_activation.o build/screenshot_notifications.o build/app.o build/configuration.o build/ui.o build/screenshot-res.o @libs -static -s -o build/KSwitcherIntegration.exe
|
& $compiler @common tests/integration.cpp build/core.o build/platform.o build/autostart.o build/autocorrect.o build/storage.o build/sync.o build/ocr.o build/richtext.o build/update.o build/screenshot.o build/screenshot_launcher.o build/screenshot_activation.o build/screenshot_notifications.o build/app.o build/configuration.o build/ui.o build/screenshot-res.o @libs -static -s -o build/KSwitcherIntegration.exe
|
||||||
if ($LASTEXITCODE) { throw 'Integration test build failed' }
|
if ($LASTEXITCODE) { throw 'Integration test build failed' }
|
||||||
& .\build\KSwitcherIntegration.exe
|
& .\build\KSwitcherIntegration.exe
|
||||||
if ($LASTEXITCODE) { throw 'Integration tests failed' }
|
if ($LASTEXITCODE) { throw 'Integration tests failed' }
|
||||||
}
|
}
|
||||||
if ($Ocr) {
|
if ($Ocr) {
|
||||||
& $compiler @common tests/ocr.cpp build/core.o build/platform.o build/autostart.o build/autocorrect.o build/storage.o build/sync.o build/ocr.o build/richtext.o build/update.o build/screenshot.o build/screenshot_launcher.o build/screenshot_activation.o build/screenshot_notifications.o build/app.o build/configuration.o build/ui.o build/screenshot-res.o @libs -static -s -o build/KSwitcherOcrTests.exe
|
$ocrTest = Join-Path (Split-Path $executablePath) 'KSwitcherOcrTests.exe'
|
||||||
|
& $compiler @common tests/ocr.cpp build/core.o build/platform.o build/autostart.o build/autocorrect.o build/storage.o build/sync.o build/ocr.o build/richtext.o build/update.o build/screenshot.o build/screenshot_launcher.o build/screenshot_activation.o build/screenshot_notifications.o build/app.o build/configuration.o build/ui.o build/screenshot-res.o @libs -static -s -o $ocrTest
|
||||||
if ($LASTEXITCODE) { throw 'OCR test build failed' }
|
if ($LASTEXITCODE) { throw 'OCR test build failed' }
|
||||||
& .\build\KSwitcherOcrTests.exe
|
& $ocrTest
|
||||||
if ($LASTEXITCODE) { throw 'OCR tests failed' }
|
if ($LASTEXITCODE) { throw 'OCR tests failed' }
|
||||||
}
|
}
|
||||||
foreach ($folder in @('build/dictionaries','build/release/dictionaries','build/staged/dictionaries')) {
|
foreach ($folder in @('build/dictionaries','build/release/dictionaries','build/staged/dictionaries')) {
|
||||||
|
|||||||
+3
-3
@@ -9,8 +9,8 @@ IDC_SCREENSHOT_PENCIL CURSOR "icons/lucide/pencil.cur"
|
|||||||
1 RT_MANIFEST "app.manifest"
|
1 RT_MANIFEST "app.manifest"
|
||||||
201 RCDATA "update.ps1"
|
201 RCDATA "update.ps1"
|
||||||
1 VERSIONINFO
|
1 VERSIONINFO
|
||||||
FILEVERSION 0,6,35,0
|
FILEVERSION 0,6,38,0
|
||||||
PRODUCTVERSION 0,6,35,0
|
PRODUCTVERSION 0,6,38,0
|
||||||
FILEOS VOS_NT_WINDOWS32
|
FILEOS VOS_NT_WINDOWS32
|
||||||
FILETYPE VFT_APP
|
FILETYPE VFT_APP
|
||||||
BEGIN
|
BEGIN
|
||||||
@@ -20,7 +20,7 @@ BEGIN
|
|||||||
BEGIN
|
BEGIN
|
||||||
VALUE "FileDescription", "KSwitcher — native keyboard and clipboard utility"
|
VALUE "FileDescription", "KSwitcher — native keyboard and clipboard utility"
|
||||||
VALUE "ProductName", "KSwitcher"
|
VALUE "ProductName", "KSwitcher"
|
||||||
VALUE "FileVersion", "0.6.35"
|
VALUE "FileVersion", "0.6.38"
|
||||||
END
|
END
|
||||||
END
|
END
|
||||||
BLOCK "VarFileInfo"
|
BLOCK "VarFileInfo"
|
||||||
|
|||||||
+36
-3
@@ -358,6 +358,14 @@ class Transaction
|
|||||||
if (!valid())
|
if (!valid())
|
||||||
return false;
|
return false;
|
||||||
write(clip);
|
write(clip);
|
||||||
|
// Viber's Qt editor observes clipboard changes asynchronously. Give it
|
||||||
|
// a chance to discard the data object from Ctrl+C before requesting paste.
|
||||||
|
// Recheck ownership and input below; never retry a possibly applied paste.
|
||||||
|
if (lower(processName(target)) == L"viber")
|
||||||
|
{
|
||||||
|
Sleep(75);
|
||||||
|
diagnosticEvent(L"Paste: Viber clipboard notification settling interval completed");
|
||||||
|
}
|
||||||
for (int i = 0; i < 2; ++i)
|
for (int i = 0; i < 2; ++i)
|
||||||
{
|
{
|
||||||
check();
|
check();
|
||||||
@@ -433,6 +441,7 @@ App::App(HINSTANCE h, bool test, const std::filesystem::path &dataRoot) : instan
|
|||||||
}
|
}
|
||||||
App::~App()
|
App::~App()
|
||||||
{
|
{
|
||||||
|
if (notificationRegistration) CloseHandle(notificationRegistration);
|
||||||
if (syncWorker.joinable()) syncWorker.join();
|
if (syncWorker.joinable()) syncWorker.join();
|
||||||
if (updateWorker.joinable())
|
if (updateWorker.joinable())
|
||||||
updateWorker.join();
|
updateWorker.join();
|
||||||
@@ -527,7 +536,10 @@ void App::scheduleHistoryOcr(const Settings &config)
|
|||||||
if (!entry.image || (entry.ocrState != 1 && entry.ocrProfile == profile)) continue;
|
if (!entry.image || (entry.ocrState != 1 && entry.ocrProfile == profile)) continue;
|
||||||
try
|
try
|
||||||
{
|
{
|
||||||
auto text = recognizeText(history->load(entry.id), config);
|
auto text = history->recognizeImage(entry.id, config, [this, &profile] {
|
||||||
|
std::lock_guard lock(jobsMutex);
|
||||||
|
return stop || ocrProfile(historyOcrSettings) != profile;
|
||||||
|
});
|
||||||
std::lock_guard lock(jobsMutex);
|
std::lock_guard lock(jobsMutex);
|
||||||
if (stop || ocrProfile(historyOcrSettings) != profile) continue;
|
if (stop || ocrProfile(historyOcrSettings) != profile) continue;
|
||||||
if (history->saveOcr(entry.id, text, config))
|
if (history->saveOcr(entry.id, text, config))
|
||||||
@@ -1576,6 +1588,17 @@ void App::captureHistory()
|
|||||||
for (auto &name : config.excluded)
|
for (auto &name : config.excluded)
|
||||||
if (lower(name) == lower(app))
|
if (lower(name) == lower(app))
|
||||||
return;
|
return;
|
||||||
|
if (IsClipboardFormatAvailable(CF_DIB) || IsClipboardFormatAvailable(CF_DIBV5) ||
|
||||||
|
IsClipboardFormatAvailable(CF_BITMAP) ||
|
||||||
|
IsClipboardFormatAvailable(RegisterClipboardFormatW(L"PNG")))
|
||||||
|
{
|
||||||
|
if (history->captureImage(sequence, app, config))
|
||||||
|
{
|
||||||
|
scheduleHistoryOcr(config);
|
||||||
|
PostMessageW(window, HistoryMessage, 0, 0);
|
||||||
|
}
|
||||||
|
return;
|
||||||
|
}
|
||||||
auto clip = readClipboard(window, (size_t)config.maxImageMiB * 1024 * 1024);
|
auto clip = readClipboard(window, (size_t)config.maxImageMiB * 1024 * 1024);
|
||||||
if (sequence != GetClipboardSequenceNumber())
|
if (sequence != GetClipboardSequenceNumber())
|
||||||
return;
|
return;
|
||||||
@@ -1613,7 +1636,9 @@ void App::recognizeEntry(std::wstring id)
|
|||||||
try
|
try
|
||||||
{
|
{
|
||||||
auto cached = history->ocrText(id, ocrProfile(config));
|
auto cached = history->ocrText(id, ocrProfile(config));
|
||||||
auto text = cached ? *cached : recognizeText(history->load(id), config);
|
auto text = cached ? *cached : history->recognizeImage(id, config, [this] {
|
||||||
|
std::lock_guard lock(jobsMutex); return stop;
|
||||||
|
});
|
||||||
if (!cached) history->saveOcr(id, text, config);
|
if (!cached) history->saveOcr(id, text, config);
|
||||||
if (text.empty())
|
if (text.empty())
|
||||||
{
|
{
|
||||||
@@ -1893,6 +1918,14 @@ LRESULT CALLBACK App::windowProc(HWND hwnd, UINT msg, WPARAM wp, LPARAM lp)
|
|||||||
case WM_TIMER:
|
case WM_TIMER:
|
||||||
if (wp == 6)
|
if (wp == 6)
|
||||||
{
|
{
|
||||||
|
if (app->notificationRegistration && WaitForSingleObject(app->notificationRegistration, 0) == WAIT_OBJECT_0)
|
||||||
|
{
|
||||||
|
DWORD code = ERROR_GEN_FAILURE;
|
||||||
|
GetExitCodeProcess(app->notificationRegistration, &code);
|
||||||
|
if (code) log(L"Screenshot notification registration helper failed: " + std::to_wstring(code));
|
||||||
|
CloseHandle(app->notificationRegistration);
|
||||||
|
app->notificationRegistration = nullptr;
|
||||||
|
}
|
||||||
app->releaseIdleMemory(GetTickCount64());
|
app->releaseIdleMemory(GetTickCount64());
|
||||||
app->pollSync();
|
app->pollSync();
|
||||||
return 0;
|
return 0;
|
||||||
@@ -2012,7 +2045,7 @@ int App::execute()
|
|||||||
log(std::wstring(L"Update completed; running version ") + AppVersion);
|
log(std::wstring(L"Update completed; running version ") + AppVersion);
|
||||||
if (startCommand)
|
if (startCommand)
|
||||||
PostMessageW(window, WM_HOTKEY, startCommand, 0);
|
PostMessageW(window, WM_HOTKEY, startCommand, 0);
|
||||||
try { registerScreenshotNotifications(); }
|
try { notificationRegistration = startScreenshotNotificationRegistration(); }
|
||||||
catch(const std::exception &e) { log(L"Screenshot notification registration: "+wide(e.what())); }
|
catch(const std::exception &e) { log(L"Screenshot notification registration: "+wide(e.what())); }
|
||||||
auto screenshotActivation=startScreenshotActivationListener();
|
auto screenshotActivation=startScreenshotActivationListener();
|
||||||
MSG msg;
|
MSG msg;
|
||||||
|
|||||||
@@ -92,6 +92,7 @@ class App
|
|||||||
std::condition_variable jobsReady;
|
std::condition_variable jobsReady;
|
||||||
std::deque<std::function<void()>> jobs;
|
std::deque<std::function<void()>> jobs;
|
||||||
std::thread worker;
|
std::thread worker;
|
||||||
|
HANDLE notificationRegistration{};
|
||||||
std::thread historySearchWorker;
|
std::thread historySearchWorker;
|
||||||
std::condition_variable historySearchReady;
|
std::condition_variable historySearchReady;
|
||||||
std::function<void()> pendingHistorySearch;
|
std::function<void()> pendingHistorySearch;
|
||||||
|
|||||||
+2
-1
@@ -252,7 +252,7 @@ DWORD CALLBACK readStream(DWORD_PTR cookie, LPBYTE buffer, LONG size, LONG *read
|
|||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
} // namespace
|
} // namespace
|
||||||
std::string richPreview(const Clip &clip)
|
std::string richPreview(const Clip &clip, bool plainFallback)
|
||||||
{
|
{
|
||||||
for (const auto &f : clip.formats)
|
for (const auto &f : clip.formats)
|
||||||
if (f.name == L"Rich Text Format" && f.data.size() <= 65536)
|
if (f.name == L"Rich Text Format" && f.data.size() <= 65536)
|
||||||
@@ -276,6 +276,7 @@ std::string richPreview(const Clip &clip)
|
|||||||
{
|
{
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
if (!plainFallback) return {};
|
||||||
auto text = clip.text();
|
auto text = clip.text();
|
||||||
if (text.size() > 4096)
|
if (text.size() > 4096)
|
||||||
text.resize(4096);
|
text.resize(4096);
|
||||||
|
|||||||
+1
-1
@@ -2,7 +2,7 @@
|
|||||||
#include "platform.h"
|
#include "platform.h"
|
||||||
namespace ks
|
namespace ks
|
||||||
{
|
{
|
||||||
std::string richPreview(const Clip &);
|
std::string richPreview(const Clip &, bool plainFallback = true);
|
||||||
HWND createRichText(HWND parent, int id, bool visible);
|
HWND createRichText(HWND parent, int id, bool visible);
|
||||||
bool setRichText(HWND, const std::string &);
|
bool setRichText(HWND, const std::string &);
|
||||||
} // namespace ks
|
} // namespace ks
|
||||||
|
|||||||
@@ -50,6 +50,30 @@ void registerScreenshotNotifications()
|
|||||||
// Notify the shell that notification identity/activation registration changed.
|
// Notify the shell that notification identity/activation registration changed.
|
||||||
SHChangeNotify(SHCNE_ASSOCCHANGED,SHCNF_IDLIST,nullptr,nullptr);
|
SHChangeNotify(SHCNE_ASSOCCHANGED,SHCNF_IDLIST,nullptr,nullptr);
|
||||||
}
|
}
|
||||||
|
HANDLE startScreenshotNotificationRegistration()
|
||||||
|
{
|
||||||
|
auto path=std::filesystem::path(executable()).parent_path()/L"KSwitcher.Screenshot.exe";
|
||||||
|
auto command=L"\""+path.wstring()+L"\" --register-notifications";
|
||||||
|
STARTUPINFOW startup{sizeof(startup)};
|
||||||
|
startup.dwFlags=STARTF_USESHOWWINDOW; startup.wShowWindow=SW_HIDE;
|
||||||
|
PROCESS_INFORMATION process{};
|
||||||
|
if(!CreateProcessW(path.c_str(),command.data(),nullptr,nullptr,FALSE,CREATE_NO_WINDOW,
|
||||||
|
nullptr,path.parent_path().c_str(),&startup,&process))
|
||||||
|
throw std::runtime_error(utf8(lastError(L"Cannot start notification registration helper")));
|
||||||
|
CloseHandle(process.hThread);
|
||||||
|
return process.hProcess;
|
||||||
|
}
|
||||||
|
void registerScreenshotNotificationsInHelper()
|
||||||
|
{
|
||||||
|
auto process=startScreenshotNotificationRegistration();
|
||||||
|
auto wait=WaitForSingleObject(process,10000);
|
||||||
|
DWORD code=ERROR_GEN_FAILURE;
|
||||||
|
bool completed=wait==WAIT_OBJECT_0 && GetExitCodeProcess(process,&code);
|
||||||
|
CloseHandle(process);
|
||||||
|
if(!completed || code)
|
||||||
|
throw std::runtime_error(completed ? "Notification registration helper failed: "+std::to_string(code)
|
||||||
|
: "Notification registration helper did not complete within 10 seconds");
|
||||||
|
}
|
||||||
bool validScreenshotToken(const std::wstring &id)
|
bool validScreenshotToken(const std::wstring &id)
|
||||||
{
|
{
|
||||||
if(id.size()!=36) return false;
|
if(id.size()!=36) return false;
|
||||||
|
|||||||
@@ -1,5 +1,6 @@
|
|||||||
#include "screenshot.h"
|
#include "screenshot.h"
|
||||||
#include "screenshot_notifications.h"
|
#include "screenshot_notifications.h"
|
||||||
|
#include "storage.h"
|
||||||
#include <commctrl.h>
|
#include <commctrl.h>
|
||||||
#include <shellapi.h>
|
#include <shellapi.h>
|
||||||
#include <roapi.h>
|
#include <roapi.h>
|
||||||
@@ -49,6 +50,14 @@ int WINAPI wWinMain(HINSTANCE instance,HINSTANCE,PWSTR,int)
|
|||||||
std::wstring token;
|
std::wstring token;
|
||||||
bool autoOpenEditor=false, writeLog=false;
|
bool autoOpenEditor=false, writeLog=false;
|
||||||
int argc{}; auto argv=CommandLineToArgvW(GetCommandLineW(),&argc);
|
int argc{}; auto argv=CommandLineToArgvW(GetCommandLineW(),&argc);
|
||||||
|
if(argc==3 && (std::wstring(argv[1])==L"--capture-history" || std::wstring(argv[1])==L"--history-ocr")) {
|
||||||
|
bool ocr=std::wstring(argv[1])==L"--history-ocr";
|
||||||
|
std::filesystem::path request=argv[2]; LocalFree(argv);
|
||||||
|
auto initialized=CoInitializeEx(nullptr,COINIT_MULTITHREADED);
|
||||||
|
if(FAILED(initialized)) return 1;
|
||||||
|
int result=ocr ? ks::runHistoryOcr(request) : ks::runImageHistoryCapture(request);
|
||||||
|
CoUninitialize(); return result;
|
||||||
|
}
|
||||||
if(argc>1 && std::wstring(argv[argc-1])==L"--write-log") {
|
if(argc>1 && std::wstring(argv[argc-1])==L"--write-log") {
|
||||||
writeLog=true; --argc;
|
writeLog=true; --argc;
|
||||||
ks::setLoggingEnabled(true);
|
ks::setLoggingEnabled(true);
|
||||||
|
|||||||
@@ -206,7 +206,7 @@ bool notifyUpdateAvailable(const std::wstring &version)
|
|||||||
{
|
{
|
||||||
try {
|
try {
|
||||||
// Reuse the screenshot notification identity, Windows settings and COM activator.
|
// Reuse the screenshot notification identity, Windows settings and COM activator.
|
||||||
registerScreenshotNotifications();
|
registerScreenshotNotificationsInHelper();
|
||||||
ComPtr<Notifications::IToastNotificationManagerStatics> manager;
|
ComPtr<Notifications::IToastNotificationManagerStatics> manager;
|
||||||
String managerClass(L"Windows.UI.Notifications.ToastNotificationManager");
|
String managerClass(L"Windows.UI.Notifications.ToastNotificationManager");
|
||||||
check(RoGetActivationFactory(managerClass.value,IID_PPV_ARGS(&manager)));
|
check(RoGetActivationFactory(managerClass.value,IID_PPV_ARGS(&manager)));
|
||||||
|
|||||||
@@ -4,6 +4,10 @@ namespace ks {
|
|||||||
bool notifyScreenshot(const Clip &);
|
bool notifyScreenshot(const Clip &);
|
||||||
std::string screenshotNotificationError();
|
std::string screenshotNotificationError();
|
||||||
void registerScreenshotNotifications();
|
void registerScreenshotNotifications();
|
||||||
|
// Caller owns the returned process handle; the helper exits after registration.
|
||||||
|
HANDLE startScreenshotNotificationRegistration();
|
||||||
|
// For background notification workers only; bounded wait, never on the UI thread.
|
||||||
|
void registerScreenshotNotificationsInHelper();
|
||||||
int runScreenshotToastActivator();
|
int runScreenshotToastActivator();
|
||||||
DWORD startScreenshotActivationListener();
|
DWORD startScreenshotActivationListener();
|
||||||
bool validScreenshotToken(const std::wstring &);
|
bool validScreenshotToken(const std::wstring &);
|
||||||
|
|||||||
+235
-18
@@ -4,6 +4,7 @@
|
|||||||
#include <algorithm>
|
#include <algorithm>
|
||||||
#include <set>
|
#include <set>
|
||||||
#include <sstream>
|
#include <sstream>
|
||||||
|
#include <wincrypt.h>
|
||||||
|
|
||||||
namespace ks
|
namespace ks
|
||||||
{
|
{
|
||||||
@@ -374,9 +375,17 @@ Clip decodeClip(const Bytes &b)
|
|||||||
History::History(const std::filesystem::path &folder) : folder_(folder)
|
History::History(const std::filesystem::path &folder) : folder_(folder)
|
||||||
{
|
{
|
||||||
std::filesystem::create_directories(folder_);
|
std::filesystem::create_directories(folder_);
|
||||||
|
loadIndex();
|
||||||
|
}
|
||||||
|
void History::loadIndex()
|
||||||
|
{
|
||||||
|
std::vector<Entry> loaded;
|
||||||
auto index = folder_ / L"index.dat";
|
auto index = folder_ / L"index.dat";
|
||||||
if (!std::filesystem::exists(index))
|
if (!std::filesystem::exists(index))
|
||||||
|
{
|
||||||
|
entries_.clear();
|
||||||
return;
|
return;
|
||||||
|
}
|
||||||
auto j = parse(unprotect(readFile(index, 16 * 1024 * 1024)));
|
auto j = parse(unprotect(readFile(index, 16 * 1024 * 1024)));
|
||||||
if (!j.is_array() || j.size() > 10000)
|
if (!j.is_array() || j.size() > 10000)
|
||||||
throw std::runtime_error("Invalid history index");
|
throw std::runtime_error("Invalid history index");
|
||||||
@@ -391,8 +400,88 @@ History::History(const std::filesystem::path &folder) : folder_(folder)
|
|||||||
e.value("image", false),
|
e.value("image", false),
|
||||||
e.value("files", false), e.value("ocrState", e.value("image", false) ? 1 : 0),
|
e.value("files", false), e.value("ocrState", e.value("image", false) ? 1 : 0),
|
||||||
str(e, "ocrProfile", L"windows")};
|
str(e, "ocrProfile", L"windows")};
|
||||||
|
if (e.value("fingerprintVersion", 0) == 1) {
|
||||||
|
item.contentHash = e.value("contentHash", std::string{});
|
||||||
|
item.imageHash = e.value("imageHash", std::string{});
|
||||||
|
item.pixelHash = e.value("pixelHash", std::string{});
|
||||||
|
}
|
||||||
if (validId(item.id) && std::filesystem::exists(folder_ / (item.id + L".dat")))
|
if (validId(item.id) && std::filesystem::exists(folder_ / (item.id + L".dat")))
|
||||||
entries_.push_back(std::move(item));
|
loaded.push_back(std::move(item));
|
||||||
|
}
|
||||||
|
entries_ = std::move(loaded);
|
||||||
|
}
|
||||||
|
bool History::captureImage(DWORD sequence, const std::wstring &app, const Settings &settings)
|
||||||
|
{
|
||||||
|
// Serialize helper index changes with all resident history mutations. Only
|
||||||
|
// metadata is reloaded: GetClipboardData's image allocations die with the child.
|
||||||
|
std::lock_guard lock(mutex_);
|
||||||
|
if (sequence != GetClipboardSequenceNumber()) return false;
|
||||||
|
auto request = std::filesystem::absolute(folder_) / (L"capture-" + newSnippetId() + L".request");
|
||||||
|
struct Cleanup {
|
||||||
|
std::filesystem::path path;
|
||||||
|
~Cleanup() { std::error_code ignored; std::filesystem::remove(path, ignored); }
|
||||||
|
} cleanup{request};
|
||||||
|
Json job{{"schema", 1}, {"sequence", sequence}, {"app", utf8(app)},
|
||||||
|
{"settings", Json::parse(encodeSettings(settings))}};
|
||||||
|
atomicWrite(request, protect(jsonBytes(job)));
|
||||||
|
auto helper = std::filesystem::path(executable()).parent_path() / L"KSwitcher.Screenshot.exe";
|
||||||
|
auto command = L"\"" + helper.wstring() + L"\" --capture-history \"" + request.wstring() + L"\"";
|
||||||
|
STARTUPINFOW startup{sizeof(startup)};
|
||||||
|
startup.dwFlags = STARTF_USESHOWWINDOW; startup.wShowWindow = SW_HIDE;
|
||||||
|
PROCESS_INFORMATION process{};
|
||||||
|
if (!CreateProcessW(helper.c_str(), command.data(), nullptr, nullptr, FALSE, CREATE_NO_WINDOW,
|
||||||
|
nullptr, helper.parent_path().c_str(), &startup, &process))
|
||||||
|
throw std::runtime_error(utf8(lastError(L"Cannot start image history helper")));
|
||||||
|
CloseHandle(process.hThread);
|
||||||
|
auto wait = WaitForSingleObject(process.hProcess, 30000);
|
||||||
|
DWORD result = ERROR_GEN_FAILURE;
|
||||||
|
if (wait != WAIT_OBJECT_0)
|
||||||
|
{
|
||||||
|
// Never release the index lock while a timed-out child can still write it.
|
||||||
|
TerminateProcess(process.hProcess, ERROR_TIMEOUT);
|
||||||
|
WaitForSingleObject(process.hProcess, INFINITE);
|
||||||
|
}
|
||||||
|
GetExitCodeProcess(process.hProcess, &result);
|
||||||
|
CloseHandle(process.hProcess);
|
||||||
|
loadIndex(); // Also reconcile any completed atomic save before a child failure.
|
||||||
|
if (wait != WAIT_OBJECT_0) throw std::runtime_error("Image history helper timed out");
|
||||||
|
if (result == 2) return false;
|
||||||
|
if (result == 3) throw std::runtime_error("Clipboard exceeds the configured size limit");
|
||||||
|
if (result) throw std::runtime_error("Image history helper failed: " + std::to_string(result));
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
int runImageHistoryCapture(const std::filesystem::path &request)
|
||||||
|
{
|
||||||
|
try
|
||||||
|
{
|
||||||
|
if (!request.is_absolute() || !request.filename().wstring().starts_with(L"capture-") ||
|
||||||
|
request.extension() != L".request") return 1;
|
||||||
|
auto job = parse(unprotect(readFile(request, 4 * 1024 * 1024)));
|
||||||
|
if (job.at("schema") != 1) return 1;
|
||||||
|
auto sequence = job.at("sequence").get<DWORD>();
|
||||||
|
auto settings = decodeSettings(jsonBytes(job.at("settings")), readLayouts());
|
||||||
|
auto valid = [&] {
|
||||||
|
return settings.history && sequence == GetClipboardSequenceNumber() &&
|
||||||
|
!IsClipboardFormatAvailable(RegisterClipboardFormatW(L"ExcludeClipboardContentFromMonitorProcessing")) &&
|
||||||
|
!IsClipboardFormatAvailable(RegisterClipboardFormatW(L"Clipboard Viewer Ignore"));
|
||||||
|
};
|
||||||
|
if (!valid()) return 2;
|
||||||
|
if (!IsClipboardFormatAvailable(CF_DIB) && !IsClipboardFormatAvailable(CF_DIBV5) &&
|
||||||
|
!IsClipboardFormatAvailable(CF_BITMAP) &&
|
||||||
|
!IsClipboardFormatAvailable(RegisterClipboardFormatW(L"PNG"))) return 2;
|
||||||
|
auto clip = readClipboard(nullptr, size_t(settings.maxImageMiB) * 1024 * 1024);
|
||||||
|
if (!valid()) return 2;
|
||||||
|
for (const auto &format : clip.formats)
|
||||||
|
if (format.name == L"CanIncludeInClipboardHistory" && format.data.size() >= 4 &&
|
||||||
|
!format.data[0] && !format.data[1] && !format.data[2] && !format.data[3]) return 2;
|
||||||
|
History history(request.parent_path());
|
||||||
|
if (!valid()) return 2;
|
||||||
|
history.add(clip, str(job, "app"), settings);
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
catch (const std::exception &e)
|
||||||
|
{
|
||||||
|
return std::string(e.what()) == "Clipboard exceeds the configured size limit" ? 3 : 1;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
void History::saveIndex()
|
void History::saveIndex()
|
||||||
@@ -406,7 +495,9 @@ void History::saveIndex()
|
|||||||
{"at", e.at},
|
{"at", e.at},
|
||||||
{"pinned", e.pinned},
|
{"pinned", e.pinned},
|
||||||
{"image", e.image},
|
{"image", e.image},
|
||||||
{"files", e.files}, {"ocrState", e.ocrState}, {"ocrProfile", utf8(e.ocrProfile)}});
|
{"files", e.files}, {"ocrState", e.ocrState}, {"ocrProfile", utf8(e.ocrProfile)},
|
||||||
|
{"fingerprintVersion", 1}, {"contentHash", e.contentHash},
|
||||||
|
{"imageHash", e.imageHash}, {"pixelHash", e.pixelHash}});
|
||||||
atomicWrite(folder_ / L"index.dat", protect(jsonBytes(j)));
|
atomicWrite(folder_ / L"index.dat", protect(jsonBytes(j)));
|
||||||
}
|
}
|
||||||
std::vector<Entry> History::entries() const
|
std::vector<Entry> History::entries() const
|
||||||
@@ -439,12 +530,10 @@ void History::prune(const Settings &s)
|
|||||||
entries_.erase(entries_.begin() + i);
|
entries_.erase(entries_.begin() + i);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
bool sameClipboardImage(const Clip &a, const Clip &b)
|
static const Format *clipboardImageFormat(const Clip &c)
|
||||||
{
|
{
|
||||||
// OLE DataObject / Ole Private Data change when Snipping Tool republishes
|
// OLE DataObject / Ole Private Data change when Snipping Tool republishes
|
||||||
// the same capture. They are transport metadata, not image identity.
|
// the same capture. They are transport metadata, not image identity.
|
||||||
auto image = [](const Clip &c) -> const Format *
|
|
||||||
{
|
|
||||||
if (!c.text().empty())
|
if (!c.text().empty())
|
||||||
return nullptr;
|
return nullptr;
|
||||||
for (const auto &f : c.formats)
|
for (const auto &f : c.formats)
|
||||||
@@ -458,8 +547,42 @@ bool sameClipboardImage(const Clip &a, const Clip &b)
|
|||||||
if (f.id == id && !f.data.empty())
|
if (f.id == id && !f.data.empty())
|
||||||
return &f;
|
return &f;
|
||||||
return nullptr;
|
return nullptr;
|
||||||
};
|
}
|
||||||
const auto *left = image(a), *right = image(b);
|
static std::string clipboardHash(const Bytes &bytes)
|
||||||
|
{
|
||||||
|
struct Hash {
|
||||||
|
HCRYPTPROV provider{}; HCRYPTHASH hash{};
|
||||||
|
~Hash() { if (hash) CryptDestroyHash(hash); if (provider) CryptReleaseContext(provider, 0); }
|
||||||
|
} state;
|
||||||
|
if (!CryptAcquireContextW(&state.provider, nullptr, nullptr, PROV_RSA_AES, CRYPT_VERIFYCONTEXT) ||
|
||||||
|
!CryptCreateHash(state.provider, CALG_SHA_256, 0, 0, &state.hash) ||
|
||||||
|
!CryptHashData(state.hash, bytes.data(), static_cast<DWORD>(bytes.size()), 0))
|
||||||
|
throw std::runtime_error("Cannot hash clipboard data");
|
||||||
|
BYTE digest[32]{}; DWORD size = sizeof(digest);
|
||||||
|
if (!CryptGetHashParam(state.hash, HP_HASHVAL, digest, &size, 0) || size != sizeof(digest))
|
||||||
|
throw std::runtime_error("Cannot read clipboard hash");
|
||||||
|
static constexpr char hex[] = "0123456789abcdef";
|
||||||
|
std::string result; result.reserve(64);
|
||||||
|
for (auto b : digest) { result += hex[b >> 4]; result += hex[b & 15]; }
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
static void imageFingerprints(Entry &entry, const Clip &clip, bool decodePixels = true)
|
||||||
|
{
|
||||||
|
const auto *image = clipboardImageFormat(clip);
|
||||||
|
if (!image) return;
|
||||||
|
if (entry.imageHash.empty())
|
||||||
|
entry.imageHash = std::to_string(image->id) + ":" + utf8(image->name) + ":" + clipboardHash(image->data);
|
||||||
|
if (!decodePixels) return;
|
||||||
|
try {
|
||||||
|
auto pixels = decodeImage(clip);
|
||||||
|
entry.pixelHash = std::to_string(pixels.width) + "x" + std::to_string(pixels.height) + ":" + clipboardHash(pixels.bgra);
|
||||||
|
} catch (const std::exception &) {
|
||||||
|
// Undecodable images can still be compared by their original bytes.
|
||||||
|
}
|
||||||
|
}
|
||||||
|
bool sameClipboardImage(const Clip &a, const Clip &b)
|
||||||
|
{
|
||||||
|
const auto *left = clipboardImageFormat(a), *right = clipboardImageFormat(b);
|
||||||
if (!left || !right)
|
if (!left || !right)
|
||||||
return false;
|
return false;
|
||||||
if (left->id == right->id && left->name == right->name && left->data == right->data)
|
if (left->id == right->id && left->name == right->name && left->data == right->data)
|
||||||
@@ -483,15 +606,40 @@ void History::add(const Clip &clip, const std::wstring &app, const Settings &s)
|
|||||||
auto data = encodeClip(clip);
|
auto data = encodeClip(clip);
|
||||||
if (data.size() > uint64_t(s.maxImageMiB) * 1024 * 1024)
|
if (data.size() > uint64_t(s.maxImageMiB) * 1024 * 1024)
|
||||||
return;
|
return;
|
||||||
|
bool image = false, files = false;
|
||||||
|
for (const auto &f : clip.formats)
|
||||||
|
{
|
||||||
|
image |= f.id == CF_DIB || f.id == CF_DIBV5 || f.name == L"PNG";
|
||||||
|
files |= f.id == CF_HDROP;
|
||||||
|
}
|
||||||
std::lock_guard lock(mutex_);
|
std::lock_guard lock(mutex_);
|
||||||
// Compare only the newest record; never keep image payloads in the resident index.
|
Entry fingerprints;
|
||||||
if (!entries_.empty())
|
fingerprints.contentHash = clipboardHash(data);
|
||||||
|
if (!entries_.empty() && entries_.front().contentHash == fingerprints.contentHash)
|
||||||
|
return;
|
||||||
|
imageFingerprints(fingerprints, clip, false);
|
||||||
|
if (!entries_.empty() && !fingerprints.imageHash.empty() && entries_.front().imageHash == fingerprints.imageHash)
|
||||||
|
return;
|
||||||
|
imageFingerprints(fingerprints, clip);
|
||||||
|
// Different payload kinds cannot be duplicates. Consult the index first so
|
||||||
|
// copying text after a screenshot never loads/decrypts the previous image.
|
||||||
|
if (!entries_.empty() && entries_.front().image == image && entries_.front().files == files)
|
||||||
{
|
{
|
||||||
try
|
try
|
||||||
{
|
{
|
||||||
auto previous = load(entries_.front().id);
|
auto &previous = entries_.front();
|
||||||
if (encodeClip(previous) == data || sameClipboardImage(previous, clip))
|
bool migrated = previous.contentHash.empty();
|
||||||
|
if (migrated) {
|
||||||
|
auto oldClip = load(previous.id);
|
||||||
|
previous.contentHash = clipboardHash(encodeClip(oldClip));
|
||||||
|
imageFingerprints(previous, oldClip);
|
||||||
|
}
|
||||||
|
if (previous.contentHash == fingerprints.contentHash ||
|
||||||
|
(!fingerprints.imageHash.empty() && previous.imageHash == fingerprints.imageHash) ||
|
||||||
|
(!fingerprints.pixelHash.empty() && previous.pixelHash == fingerprints.pixelHash)) {
|
||||||
|
if (migrated) saveIndex();
|
||||||
return;
|
return;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
catch (...)
|
catch (...)
|
||||||
{
|
{
|
||||||
@@ -506,12 +654,6 @@ void History::add(const Clip &clip, const std::wstring &app, const Settings &s)
|
|||||||
auto encrypted = protect(data);
|
auto encrypted = protect(data);
|
||||||
atomicWrite(folder_ / (id + L".dat"), encrypted);
|
atomicWrite(folder_ / (id + L".dat"), encrypted);
|
||||||
auto text = clip.text();
|
auto text = clip.text();
|
||||||
bool image = false, files = false;
|
|
||||||
for (auto &f : clip.formats)
|
|
||||||
{
|
|
||||||
image |= f.id == CF_DIB || f.id == CF_DIBV5 || f.name == L"PNG";
|
|
||||||
files |= f.id == CF_HDROP;
|
|
||||||
}
|
|
||||||
if (text.empty())
|
if (text.empty())
|
||||||
text = files ? L"[Files]" : image ? L"[Image]" : L"[Clipboard data]";
|
text = files ? L"[Files]" : image ? L"[Image]" : L"[Clipboard data]";
|
||||||
for (auto &c : text)
|
for (auto &c : text)
|
||||||
@@ -524,7 +666,8 @@ void History::add(const Clip &clip, const std::wstring &app, const Settings &s)
|
|||||||
entries_.insert(entries_.begin(),
|
entries_.insert(entries_.begin(),
|
||||||
Entry{id, text, app, encrypted.size(),
|
Entry{id, text, app, encrypted.size(),
|
||||||
(uint64_t(now.dwHighDateTime) << 32) | now.dwLowDateTime, false, image, files,
|
(uint64_t(now.dwHighDateTime) << 32) | now.dwLowDateTime, false, image, files,
|
||||||
image ? 1 : 0, ocrProfile(s)});
|
image ? 1 : 0, ocrProfile(s), std::move(fingerprints.contentHash),
|
||||||
|
std::move(fingerprints.imageHash), std::move(fingerprints.pixelHash)});
|
||||||
prune(s);
|
prune(s);
|
||||||
saveIndex();
|
saveIndex();
|
||||||
}
|
}
|
||||||
@@ -563,6 +706,80 @@ void History::clear()
|
|||||||
std::filesystem::remove(folder_ / (e.id + L".ocr"), ec);
|
std::filesystem::remove(folder_ / (e.id + L".ocr"), ec);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
std::wstring History::recognizeImage(const std::wstring &id, const Settings &settings,
|
||||||
|
const std::function<bool()> &cancelled) const
|
||||||
|
{
|
||||||
|
if (!validId(id)) throw std::runtime_error("Invalid history ID");
|
||||||
|
if (cancelled && cancelled()) throw std::runtime_error("History OCR cancelled");
|
||||||
|
auto request = std::filesystem::absolute(folder_) / (L"ocr-" + newSnippetId() + L".request");
|
||||||
|
auto response = request; response += L".result";
|
||||||
|
struct Cleanup {
|
||||||
|
std::filesystem::path request, response;
|
||||||
|
~Cleanup() {
|
||||||
|
std::error_code ignored;
|
||||||
|
std::filesystem::remove(request, ignored);
|
||||||
|
std::filesystem::remove(response, ignored);
|
||||||
|
}
|
||||||
|
} cleanup{request, response};
|
||||||
|
// Send only OCR options and the entry ID. The resident never loads image bytes.
|
||||||
|
Json job{{"schema", 1}, {"id", utf8(id)}, {"tesseract", settings.useTesseract},
|
||||||
|
{"path", utf8(settings.tesseractPath)}, {"languages", utf8(settings.tesseractLanguages)}};
|
||||||
|
atomicWrite(request, protect(jsonBytes(job)));
|
||||||
|
auto helper = std::filesystem::path(executable()).parent_path() / L"KSwitcher.Screenshot.exe";
|
||||||
|
auto command = L"\"" + helper.wstring() + L"\" --history-ocr \"" + request.wstring() + L"\"";
|
||||||
|
STARTUPINFOW startup{sizeof(startup)};
|
||||||
|
startup.dwFlags = STARTF_USESHOWWINDOW; startup.wShowWindow = SW_HIDE;
|
||||||
|
PROCESS_INFORMATION process{};
|
||||||
|
if (!CreateProcessW(helper.c_str(), command.data(), nullptr, nullptr, FALSE, CREATE_NO_WINDOW,
|
||||||
|
nullptr, helper.parent_path().c_str(), &startup, &process))
|
||||||
|
throw std::runtime_error(utf8(lastError(L"Cannot start history OCR helper")));
|
||||||
|
CloseHandle(process.hThread);
|
||||||
|
struct Child {
|
||||||
|
HANDLE handle;
|
||||||
|
~Child() {
|
||||||
|
if (WaitForSingleObject(handle, 0) != WAIT_OBJECT_0) {
|
||||||
|
TerminateProcess(handle, ERROR_CANCELLED);
|
||||||
|
WaitForSingleObject(handle, INFINITE);
|
||||||
|
}
|
||||||
|
CloseHandle(handle);
|
||||||
|
}
|
||||||
|
} child{process.hProcess};
|
||||||
|
auto deadline = GetTickCount64() + 60000;
|
||||||
|
for (;;) {
|
||||||
|
auto wait = WaitForSingleObject(child.handle, 100);
|
||||||
|
if (wait == WAIT_OBJECT_0) break;
|
||||||
|
if (wait != WAIT_TIMEOUT) throw std::runtime_error("Cannot wait for history OCR helper");
|
||||||
|
if (cancelled && cancelled()) throw std::runtime_error("History OCR cancelled");
|
||||||
|
if (GetTickCount64() >= deadline) throw std::runtime_error("History OCR helper timed out");
|
||||||
|
}
|
||||||
|
DWORD result{};
|
||||||
|
if (!GetExitCodeProcess(child.handle, &result) || result)
|
||||||
|
throw std::runtime_error("History OCR helper failed: " + std::to_string(result));
|
||||||
|
auto reply = parse(unprotect(readFile(response, 8 * 1024 * 1024)));
|
||||||
|
if (reply.at("schema") != 1 || str(reply, "id") != id)
|
||||||
|
throw std::runtime_error("Invalid history OCR response");
|
||||||
|
return str(reply, "text");
|
||||||
|
}
|
||||||
|
int runHistoryOcr(const std::filesystem::path &request)
|
||||||
|
{
|
||||||
|
try {
|
||||||
|
if (!request.is_absolute() || !request.filename().wstring().starts_with(L"ocr-") ||
|
||||||
|
request.extension() != L".request") return 1;
|
||||||
|
auto job = parse(unprotect(readFile(request, 1024 * 1024)));
|
||||||
|
auto id = str(job, "id");
|
||||||
|
if (job.at("schema") != 1 || !validId(id)) return 1;
|
||||||
|
Settings settings;
|
||||||
|
settings.useTesseract = job.at("tesseract").get<bool>();
|
||||||
|
settings.tesseractPath = str(job, "path");
|
||||||
|
settings.tesseractLanguages = str(job, "languages");
|
||||||
|
auto clip = decodeClip(unprotect(readFile(request.parent_path() / (id + L".dat"), 80 * 1024 * 1024)));
|
||||||
|
auto text = utf8(recognizeText(clip, settings));
|
||||||
|
if (text.size() > 1024 * 1024) return 1;
|
||||||
|
auto response = request; response += L".result";
|
||||||
|
atomicWrite(response, protect(jsonBytes(Json{{"schema", 1}, {"id", utf8(id)}, {"text", text}})));
|
||||||
|
return 0;
|
||||||
|
} catch (...) { return 1; }
|
||||||
|
}
|
||||||
bool History::saveOcr(const std::wstring &id, const std::optional<std::wstring> &text, const Settings &settings)
|
bool History::saveOcr(const std::wstring &id, const std::optional<std::wstring> &text, const Settings &settings)
|
||||||
{
|
{
|
||||||
std::lock_guard lock(mutex_);
|
std::lock_guard lock(mutex_);
|
||||||
|
|||||||
@@ -1,6 +1,7 @@
|
|||||||
#pragma once
|
#pragma once
|
||||||
#include "platform.h"
|
#include "platform.h"
|
||||||
#include <mutex>
|
#include <mutex>
|
||||||
|
#include <functional>
|
||||||
namespace ks
|
namespace ks
|
||||||
{
|
{
|
||||||
Settings loadSettings(const std::filesystem::path &, const std::vector<Layout> &);
|
Settings loadSettings(const std::filesystem::path &, const std::vector<Layout> &);
|
||||||
@@ -15,6 +16,7 @@ struct Entry
|
|||||||
bool pinned{}, image{}, files{};
|
bool pinned{}, image{}, files{};
|
||||||
int ocrState{}; // 0: legacy/not applicable, 1: pending, 2: saved, 3: failed
|
int ocrState{}; // 0: legacy/not applicable, 1: pending, 2: saved, 3: failed
|
||||||
std::wstring ocrProfile = L"windows";
|
std::wstring ocrProfile = L"windows";
|
||||||
|
std::string contentHash, imageHash, pixelHash;
|
||||||
};
|
};
|
||||||
class History
|
class History
|
||||||
{
|
{
|
||||||
@@ -22,6 +24,7 @@ class History
|
|||||||
std::vector<Entry> entries_;
|
std::vector<Entry> entries_;
|
||||||
mutable std::mutex mutex_;
|
mutable std::mutex mutex_;
|
||||||
void saveIndex();
|
void saveIndex();
|
||||||
|
void loadIndex();
|
||||||
void prune(const Settings &);
|
void prune(const Settings &);
|
||||||
|
|
||||||
public:
|
public:
|
||||||
@@ -29,6 +32,9 @@ class History
|
|||||||
std::vector<Entry> entries() const;
|
std::vector<Entry> entries() const;
|
||||||
Clip load(const std::wstring &id) const;
|
Clip load(const std::wstring &id) const;
|
||||||
void add(const Clip &, const std::wstring &app, const Settings &);
|
void add(const Clip &, const std::wstring &app, const Settings &);
|
||||||
|
bool captureImage(DWORD sequence, const std::wstring &app, const Settings &);
|
||||||
|
std::wstring recognizeImage(const std::wstring &id, const Settings &,
|
||||||
|
const std::function<bool()> &cancelled = {}) const;
|
||||||
bool saveOcr(const std::wstring &id, const std::optional<std::wstring> &, const Settings &);
|
bool saveOcr(const std::wstring &id, const std::optional<std::wstring> &, const Settings &);
|
||||||
std::optional<std::wstring> ocrText(const std::wstring &id, const std::wstring &profile = L"") const;
|
std::optional<std::wstring> ocrText(const std::wstring &id, const std::wstring &profile = L"") const;
|
||||||
void remove(const std::wstring &);
|
void remove(const std::wstring &);
|
||||||
@@ -39,4 +45,6 @@ class History
|
|||||||
Bytes encodeClip(const Clip &);
|
Bytes encodeClip(const Clip &);
|
||||||
Clip decodeClip(const Bytes &);
|
Clip decodeClip(const Bytes &);
|
||||||
bool sameClipboardImage(const Clip &, const Clip &);
|
bool sameClipboardImage(const Clip &, const Clip &);
|
||||||
|
int runImageHistoryCapture(const std::filesystem::path &request);
|
||||||
|
int runHistoryOcr(const std::filesystem::path &request);
|
||||||
} // namespace ks
|
} // namespace ks
|
||||||
|
|||||||
+34
-14
@@ -468,12 +468,17 @@ struct SearchResult
|
|||||||
uint64_t generation;
|
uint64_t generation;
|
||||||
std::vector<Entry> items;
|
std::vector<Entry> items;
|
||||||
};
|
};
|
||||||
|
struct TextPreview
|
||||||
|
{
|
||||||
|
std::string rtf;
|
||||||
|
std::wstring plain;
|
||||||
|
};
|
||||||
struct PreviewResult
|
struct PreviewResult
|
||||||
{
|
{
|
||||||
uint64_t generation;
|
uint64_t generation;
|
||||||
std::wstring id;
|
std::wstring id;
|
||||||
Pixels pixels;
|
Pixels pixels;
|
||||||
std::string rtf;
|
TextPreview text;
|
||||||
};
|
};
|
||||||
struct HistoryView
|
struct HistoryView
|
||||||
{
|
{
|
||||||
@@ -481,8 +486,9 @@ struct HistoryView
|
|||||||
std::vector<Entry> items;
|
std::vector<Entry> items;
|
||||||
std::wstring rememberedSelection, rememberedTop;
|
std::wstring rememberedSelection, rememberedTop;
|
||||||
uint64_t generation{};
|
uint64_t generation{};
|
||||||
|
std::shared_ptr<std::atomic<bool>> previewActive = std::make_shared<std::atomic<bool>>(true);
|
||||||
std::map<std::wstring, Pixels> thumbnails;
|
std::map<std::wstring, Pixels> thumbnails;
|
||||||
std::map<std::wstring, std::string> richTexts;
|
std::map<std::wstring, TextPreview> richTexts;
|
||||||
HWND rich{};
|
HWND rich{};
|
||||||
std::set<std::wstring> pending;
|
std::set<std::wstring> pending;
|
||||||
bool scrolling{};
|
bool scrolling{};
|
||||||
@@ -504,6 +510,8 @@ std::wstring selected(HWND w, HistoryView *v)
|
|||||||
}
|
}
|
||||||
void search(HWND w, HistoryView *v)
|
void search(HWND w, HistoryView *v)
|
||||||
{
|
{
|
||||||
|
v->previewActive->store(false);
|
||||||
|
v->previewActive = std::make_shared<std::atomic<bool>>(true);
|
||||||
v->closed = false;
|
v->closed = false;
|
||||||
auto q = value(w, 101);
|
auto q = value(w, 101);
|
||||||
auto generation = ++v->generation;
|
auto generation = ++v->generation;
|
||||||
@@ -532,23 +540,31 @@ void requestPreview(HWND w, HistoryView *v, const Entry &entry)
|
|||||||
if (v->thumbnails.contains(id) || v->richTexts.contains(id) || !v->pending.insert(id).second)
|
if (v->thumbnails.contains(id) || v->richTexts.contains(id) || !v->pending.insert(id).second)
|
||||||
return;
|
return;
|
||||||
auto generation = v->generation;
|
auto generation = v->generation;
|
||||||
|
auto active = v->previewActive;
|
||||||
auto *app = v->app;
|
auto *app = v->app;
|
||||||
app->enqueue(
|
app->enqueue(
|
||||||
[w, app, id, generation, image]
|
[w, app, id, generation, image, active]
|
||||||
{
|
{
|
||||||
|
if (!active->load()) return;
|
||||||
auto result = new PreviewResult{generation, id, {}, {}};
|
auto result = new PreviewResult{generation, id, {}, {}};
|
||||||
try
|
try
|
||||||
{
|
{
|
||||||
auto clip = app->history->load(id);
|
auto clip = app->history->load(id);
|
||||||
|
if (!active->load()) { delete result; return; }
|
||||||
if (image)
|
if (image)
|
||||||
result->pixels = decodeImage(clip, 208, 128);
|
result->pixels = decodeImage(clip, 208, 128);
|
||||||
else
|
else {
|
||||||
result->rtf = richPreview(clip);
|
result->text.rtf = richPreview(clip, false);
|
||||||
|
if (result->text.rtf.empty()) {
|
||||||
|
result->text.plain = clip.text();
|
||||||
|
if (result->text.plain.size() > 4096) result->text.plain.resize(4096);
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
catch (const std::exception &)
|
catch (const std::exception &)
|
||||||
{
|
{
|
||||||
}
|
}
|
||||||
if (!PostMessageW(w, PreviewMessage, 0, (LPARAM)result))
|
if (!active->load() || !PostMessageW(w, PreviewMessage, 0, (LPARAM)result))
|
||||||
delete result;
|
delete result;
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
@@ -602,9 +618,13 @@ void drawHistoryRow(HWND w, HistoryView *v, const DRAWITEMSTRUCT &draw)
|
|||||||
if (!entry.image && !entry.files)
|
if (!entry.image && !entry.files)
|
||||||
{
|
{
|
||||||
auto rich = v->richTexts.find(entry.id);
|
auto rich = v->richTexts.find(entry.id);
|
||||||
if (rich != v->richTexts.end() && v->rich && !rich->second.empty())
|
if (rich != v->richTexts.end() && rich->second.rtf.empty())
|
||||||
|
text = (entry.pinned ? L"★ " : L"") + rich->second.plain;
|
||||||
|
if (rich != v->richTexts.end() && !rich->second.rtf.empty() && !v->rich)
|
||||||
|
v->rich = createRichText(w, 105, false);
|
||||||
|
if (rich != v->richTexts.end() && v->rich && !rich->second.rtf.empty())
|
||||||
{
|
{
|
||||||
setRichText(v->rich, rich->second);
|
setRichText(v->rich, rich->second.rtf);
|
||||||
CHARFORMAT2W format{};
|
CHARFORMAT2W format{};
|
||||||
format.cbSize = sizeof(format);
|
format.cbSize = sizeof(format);
|
||||||
format.dwMask = CFM_COLOR;
|
format.dwMask = CFM_COLOR;
|
||||||
@@ -825,14 +845,14 @@ LRESULT CALLBACK historyProc(HWND w, UINT msg, WPARAM wp, LPARAM lp)
|
|||||||
visible.insert(v->items[i].id);
|
visible.insert(v->items[i].id);
|
||||||
if(v->items[i].id==result->id) { changed=intersection; repaint=true; }
|
if(v->items[i].id==result->id) { changed=intersection; repaint=true; }
|
||||||
}
|
}
|
||||||
if (!result->rtf.empty())
|
if (!result->text.rtf.empty() || !result->text.plain.empty())
|
||||||
v->richTexts[result->id] = std::move(result->rtf);
|
v->richTexts[result->id] = std::move(result->text);
|
||||||
else
|
else
|
||||||
v->thumbnails[result->id] = std::move(result->pixels);
|
v->thumbnails[result->id] = std::move(result->pixels);
|
||||||
trimHistoryPreviews(v->thumbnails,visible,32,HistoryThumbnailBudget,
|
trimHistoryPreviews(v->thumbnails,visible,32,HistoryThumbnailBudget,
|
||||||
[](const Pixels &p) { return p.bgra.capacity(); });
|
[](const Pixels &p) { return p.bgra.capacity(); });
|
||||||
trimHistoryPreviews(v->richTexts,visible,32,HistoryRichTextBudget,
|
trimHistoryPreviews(v->richTexts,visible,32,HistoryRichTextBudget,
|
||||||
[](const std::string &s) { return s.capacity(); });
|
[](const TextPreview &s) { return s.rtf.capacity() + s.plain.capacity() * sizeof(wchar_t); });
|
||||||
if(repaint) InvalidateRect(list,&changed,FALSE);
|
if(repaint) InvalidateRect(list,&changed,FALSE);
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
@@ -889,18 +909,19 @@ LRESULT CALLBACK historyProc(HWND w, UINT msg, WPARAM wp, LPARAM lp)
|
|||||||
v->rememberedTop = top >= 0 && top < (int)v->items.size() ? v->items[top].id : L"";
|
v->rememberedTop = top >= 0 && top < (int)v->items.size() ? v->items[top].id : L"";
|
||||||
KillTimer(w, 1);
|
KillTimer(w, 1);
|
||||||
v->closed = true;
|
v->closed = true;
|
||||||
|
v->previewActive->store(false);
|
||||||
++v->generation; // Discard search/preview results queued before closing.
|
++v->generation; // Discard search/preview results queued before closing.
|
||||||
ShowWindow(w, SW_HIDE);
|
ShowWindow(w, SW_HIDE);
|
||||||
SendDlgItemMessageW(w,102,LB_RESETCONTENT,0,0);
|
SendDlgItemMessageW(w,102,LB_RESETCONTENT,0,0);
|
||||||
std::vector<Entry>().swap(v->items);
|
std::vector<Entry>().swap(v->items);
|
||||||
v->thumbnails.clear();
|
v->thumbnails.clear();
|
||||||
v->richTexts.clear();
|
v->richTexts.clear();
|
||||||
if (v->rich)
|
if (v->rich) SetWindowTextW(v->rich, L"");
|
||||||
SetWindowTextW(v->rich, L"");
|
|
||||||
v->pending.clear();
|
v->pending.clear();
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
case WM_NCDESTROY:
|
case WM_NCDESTROY:
|
||||||
|
v->previewActive->store(false);
|
||||||
disposeFont(w);
|
disposeFont(w);
|
||||||
v->app->historyWindow = nullptr;
|
v->app->historyWindow = nullptr;
|
||||||
delete v;
|
delete v;
|
||||||
@@ -1702,7 +1723,6 @@ void showHistory(App &app, bool visible)
|
|||||||
HWND w = makeWindow(app, L"KSwitcher.History", L"Clipboard history", historyProc, 760, 500);
|
HWND w = makeWindow(app, L"KSwitcher.History", L"Clipboard history", historyProc, 760, 500);
|
||||||
app.historyWindow = w;
|
app.historyWindow = w;
|
||||||
windowLayoutState(w)->history = true;
|
windowLayoutState(w)->history = true;
|
||||||
((HistoryView *)GetWindowLongPtrW(w, GWLP_USERDATA))->rich = createRichText(w, 105, false);
|
|
||||||
label(w, L"&Search clipboard history", 16, 12, 500);
|
label(w, L"&Search clipboard history", 16, 12, 500);
|
||||||
edit(w, 101, 16, 38, 728);
|
edit(w, 101, 16, 38, 728);
|
||||||
auto list = control(w, L"LISTBOX", L"",
|
auto list = control(w, L"LISTBOX", L"",
|
||||||
|
|||||||
+1
-1
@@ -2,7 +2,7 @@
|
|||||||
#include "platform.h"
|
#include "platform.h"
|
||||||
namespace ks
|
namespace ks
|
||||||
{
|
{
|
||||||
inline constexpr wchar_t AppVersion[] = L"0.6.35";
|
inline constexpr wchar_t AppVersion[] = L"0.6.38";
|
||||||
inline constexpr UINT UpdateMessage = WM_APP + 7;
|
inline constexpr UINT UpdateMessage = WM_APP + 7;
|
||||||
inline constexpr UINT UpdateCheckIntervalMs = 6 * 60 * 60 * 1000;
|
inline constexpr UINT UpdateCheckIntervalMs = 6 * 60 * 60 * 1000;
|
||||||
inline constexpr UINT UpdateNotificationRetryMs = 5 * 60 * 1000;
|
inline constexpr UINT UpdateNotificationRetryMs = 5 * 60 * 1000;
|
||||||
|
|||||||
@@ -0,0 +1,68 @@
|
|||||||
|
// Interactive clipboard test. Run with the resident stopped; the original
|
||||||
|
// clipboard is restored. The helper must be beside this test executable.
|
||||||
|
#include "storage.h"
|
||||||
|
#include "ocr.h"
|
||||||
|
#include <algorithm>
|
||||||
|
#include <iostream>
|
||||||
|
static void require(bool ok,const char *name) {
|
||||||
|
if(!ok) throw std::runtime_error(name);
|
||||||
|
std::cout << "PASS " << name << '\n';
|
||||||
|
}
|
||||||
|
int main() {
|
||||||
|
CoInitializeEx(nullptr,COINIT_MULTITHREADED);
|
||||||
|
HWND owner=CreateWindowExW(0,L"STATIC",L"Capture test",0,0,0,0,0,HWND_MESSAGE,nullptr,nullptr,nullptr);
|
||||||
|
ks::Clip saved; bool captured=false; int result=0;
|
||||||
|
auto folder=std::filesystem::temp_directory_path()/(L"KSwitcher-image-capture-"+std::to_wstring(GetCurrentProcessId()));
|
||||||
|
try {
|
||||||
|
saved=ks::readClipboard(owner,128*1024*1024); captured=true;
|
||||||
|
ks::Settings settings; settings.history=true;
|
||||||
|
ks::History history(folder);
|
||||||
|
history.add(ks::textClip(L"keep pinned"),L"test",settings);
|
||||||
|
auto pinned=history.entries().front().id; history.pin(pinned);
|
||||||
|
ks::Pixels pixels{32,16,ks::Bytes(32*16*4,255)};
|
||||||
|
ks::Clip image{{ks::imageDib(pixels),{0,L"PNG",ks::encodePng(pixels)}}};
|
||||||
|
auto sequence=ks::writeClipboard(owner,image);
|
||||||
|
require(history.captureImage(sequence,L"capture test",settings),"helper captures image");
|
||||||
|
auto entries=history.entries();
|
||||||
|
require(entries.size()==2 && entries.front().image && entries.front().ocrState==1,"resident reloads metadata with pending OCR");
|
||||||
|
require(entries.front().contentHash.size()==64 && !entries.front().pixelHash.empty(),
|
||||||
|
"helper persists fingerprints and resident reloads them");
|
||||||
|
require(entries.back().id==pinned && entries.back().pinned,"helper preserves existing pinned entry");
|
||||||
|
auto restored=history.load(entries.front().id);
|
||||||
|
auto has=[&](const ks::Format &expected) {
|
||||||
|
return std::any_of(restored.formats.begin(),restored.formats.end(),[&](const ks::Format &actual) {
|
||||||
|
return (expected.name.empty()?actual.id==expected.id:actual.name==expected.name) && actual.data==expected.data;
|
||||||
|
});
|
||||||
|
};
|
||||||
|
require(has(image.formats[0]) && has(image.formats[1]),
|
||||||
|
"encrypted history preserves both image representations");
|
||||||
|
require(ks::readFile(folder/(entries.front().id+L".dat"),1024*1024)!=ks::encodeClip(restored),"saved image is encrypted");
|
||||||
|
require(GetClipboardSequenceNumber()==sequence,"capture leaves clipboard unchanged");
|
||||||
|
require(history.captureImage(sequence,L"capture test",settings) && history.entries().size()==2,"duplicate capture is not added");
|
||||||
|
ks::writeClipboard(owner,ks::textClip(L"new clipboard"));
|
||||||
|
require(!history.captureImage(sequence,L"stale",settings) && history.entries().size()==2,"stale request does not save another clipboard");
|
||||||
|
for(auto name : {L"ExcludeClipboardContentFromMonitorProcessing",L"Clipboard Viewer Ignore",L"CanIncludeInClipboardHistory"}) {
|
||||||
|
auto blocked=image; blocked.formats.push_back({0,name,ks::Bytes(4,0)});
|
||||||
|
sequence=ks::writeClipboard(owner,blocked);
|
||||||
|
require(!history.captureImage(sequence,L"excluded",settings) && history.entries().size()==2,"clipboard privacy flag prevents capture");
|
||||||
|
}
|
||||||
|
pixels={1024,512,ks::Bytes(1024*512*4,255)};
|
||||||
|
sequence=ks::writeClipboard(owner,ks::Clip{{ks::imageDib(pixels)}});
|
||||||
|
settings.maxImageMiB=1;
|
||||||
|
bool rejected=false;
|
||||||
|
try {history.captureImage(sequence,L"oversized",settings);} catch(const std::exception &e) {
|
||||||
|
rejected=std::string(e.what())=="Clipboard exceeds the configured size limit";
|
||||||
|
}
|
||||||
|
require(rejected && history.entries().size()==2,"oversized image is rejected without changing index");
|
||||||
|
require(GetClipboardSequenceNumber()==sequence,"rejection leaves clipboard unchanged");
|
||||||
|
for(const auto &file:std::filesystem::directory_iterator(folder))
|
||||||
|
require(file.path().extension()!=L".request","encrypted capture request is cleaned up");
|
||||||
|
} catch(const std::exception &e) {std::cerr << "FAIL " << e.what() << '\n';result=1;}
|
||||||
|
if(captured) try {ks::writeClipboard(owner,saved);} catch(...) {result=1;}
|
||||||
|
DestroyWindow(owner); CoUninitialize();
|
||||||
|
// Remove only the exact PID-specific fixture under the system temp directory.
|
||||||
|
std::error_code ignored;
|
||||||
|
if(std::filesystem::weakly_canonical(folder).parent_path()==std::filesystem::weakly_canonical(std::filesystem::temp_directory_path()))
|
||||||
|
std::filesystem::remove_all(folder,ignored);
|
||||||
|
return result;
|
||||||
|
}
|
||||||
+4
-1
@@ -912,6 +912,7 @@ int main(int argc, char **argv) {
|
|||||||
ks::showHistory(app,false);
|
ks::showHistory(app,false);
|
||||||
auto history=app.historyWindow;
|
auto history=app.historyWindow;
|
||||||
require(history&&!IsWindowVisible(history),"history layout tested without changing focus");
|
require(history&&!IsWindowVisible(history),"history layout tested without changing focus");
|
||||||
|
require(!GetDlgItem(history,105),"history creates no RichEdit control before formatted content is displayed");
|
||||||
{
|
{
|
||||||
app.history=std::make_unique<ks::History>(temp/L"history-search");
|
app.history=std::make_unique<ks::History>(temp/L"history-search");
|
||||||
app.history->add(ks::textClip(L"згенеруй приклад"),L"test",app.settings);
|
app.history->add(ks::textClip(L"згенеруй приклад"),L"test",app.settings);
|
||||||
@@ -941,7 +942,9 @@ int main(int argc, char **argv) {
|
|||||||
require(GetFocus()==originalFocus,"history search preserves focus");
|
require(GetFocus()==originalFocus,"history search preserves focus");
|
||||||
SendDlgItemMessageW(history,102,LB_SETCURSEL,1,0);
|
SendDlgItemMessageW(history,102,LB_SETCURSEL,1,0);
|
||||||
ks::refreshHistory(app); // A result may still be in flight when closing.
|
ks::refreshHistory(app); // A result may still be in flight when closing.
|
||||||
SendMessageW(history,WM_CLOSE,0,0);
|
SendDlgItemMessageW(history,101,WM_KEYDOWN,VK_ESCAPE,0);
|
||||||
|
require(!IsWindowVisible(history) && !GetDlgItem(history,105),
|
||||||
|
"Escape closes history and leaves no RichEdit control");
|
||||||
SendMessageW(history,WM_TIMER,1,0); // KillTimer cannot retract a posted timer message.
|
SendMessageW(history,WM_TIMER,1,0); // KillTimer cannot retract a posted timer message.
|
||||||
require(pumpUntil(0),"closing history releases list strings");
|
require(pumpUntil(0),"closing history releases list strings");
|
||||||
auto until=GetTickCount64()+250;
|
auto until=GetTickCount64()+250;
|
||||||
|
|||||||
+17
-1
@@ -529,6 +529,9 @@ LRESULT CALLBACK procedure(HWND w, UINT msg, WPARAM wp, LPARAM lp)
|
|||||||
require(text() == L"wrong ", "unrelated selection remains rejected with trimmed clipboard");
|
require(text() == L"wrong ", "unrelated selection remains rejected with trimmed clipboard");
|
||||||
app->settings.autoCorrect = true;
|
app->settings.autoCorrect = true;
|
||||||
app->settings.autoLanguages = {L"en", L"ru"};
|
app->settings.autoLanguages = {L"en", L"ru"};
|
||||||
|
// The worker now shares the predictor's immutable dictionaries.
|
||||||
|
// This fixture changes settings directly instead of applySettings.
|
||||||
|
app->autoPredictor.configure(L"dictionaries", app->settings.autoLanguages);
|
||||||
app->autoCorrector = ks::AutoCorrector{};
|
app->autoCorrector = ks::AutoCorrector{};
|
||||||
ActivateKeyboardLayout(app->layouts[en].handle, 0);
|
ActivateKeyboardLayout(app->layouts[en].handle, 0);
|
||||||
prepare(L"ghbdtn ", false);
|
prepare(L"ghbdtn ", false);
|
||||||
@@ -556,9 +559,22 @@ LRESULT CALLBACK procedure(HWND w, UINT msg, WPARAM wp, LPARAM lp)
|
|||||||
require(trimmedCopies >= 7, "fixture actually stripped spaces during recovery scenarios");
|
require(trimmedCopies >= 7, "fixture actually stripped spaces during recovery scenarios");
|
||||||
pasteBehavior = 0;
|
pasteBehavior = 0;
|
||||||
trimCopiedSpaces = false;
|
trimCopiedSpaces = false;
|
||||||
finish();
|
// User-selected middle range, followed by the same command with
|
||||||
|
// no selection: retain both surrounding text and trailing spaces.
|
||||||
|
prepare(L"OLDghbdtn NEXT", false);
|
||||||
|
SendMessageW(field, EM_SETSEL, 3, 11);
|
||||||
|
app->transform(ks::Direct + ru);
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
|
case 58:
|
||||||
|
require(text() == L"OLDпривет NEXT", "explicit middle selection preserves surrounding text");
|
||||||
|
app->transform(ks::Direct + en);
|
||||||
|
break;
|
||||||
|
case 59:
|
||||||
|
require(text() == L"OLDghbdtn NEXT", "hotkey without selection reconverts only the previous middle range");
|
||||||
|
require(ks::readClipboard(app->window).text() == sentinel, "selected and repeated conversions restore clipboard");
|
||||||
|
finish();
|
||||||
|
break;
|
||||||
case 30:
|
case 30:
|
||||||
case 32:
|
case 32:
|
||||||
case 34:
|
case 34:
|
||||||
|
|||||||
@@ -59,6 +59,24 @@ int main(int argc, char **argv)
|
|||||||
app.settings.history = true;
|
app.settings.history = true;
|
||||||
app.history->add(clip, L"OCR fixture", app.settings);
|
app.history->add(clip, L"OCR fixture", app.settings);
|
||||||
auto id = app.history->entries().front().id;
|
auto id = app.history->entries().front().id;
|
||||||
|
auto isolated = app.history->recognizeImage(id, config);
|
||||||
|
if (isolated != text) throw std::runtime_error("Helper and in-process OCR results differ");
|
||||||
|
bool cancelled = false;
|
||||||
|
try { app.history->recognizeImage(id, config, [] { return true; }); }
|
||||||
|
catch (const std::exception &) { cancelled = true; }
|
||||||
|
if (!cancelled) throw std::runtime_error("Cancelled OCR request was executed");
|
||||||
|
auto invalid = config;
|
||||||
|
invalid.useTesseract = true;
|
||||||
|
invalid.tesseractPath = (temp / L"missing-tesseract.exe").wstring();
|
||||||
|
bool failed = false;
|
||||||
|
try { app.history->recognizeImage(id, invalid); }
|
||||||
|
catch (const std::exception &) { failed = true; }
|
||||||
|
if (!failed || app.history->entries().front().ocrState != 1)
|
||||||
|
throw std::runtime_error("Failed helper modified history metadata");
|
||||||
|
for (const auto &file : std::filesystem::directory_iterator(temp / L"history"))
|
||||||
|
if (file.path().filename().wstring().starts_with(L"ocr-"))
|
||||||
|
throw std::runtime_error("OCR helper left temporary files");
|
||||||
|
std::cout << "PASS isolated OCR equivalence, cancellation, failure and encrypted IPC cleanup\n";
|
||||||
if (tesseract)
|
if (tesseract)
|
||||||
{
|
{
|
||||||
app.history->saveOcr(id, L"old Windows result", app.settings);
|
app.history->saveOcr(id, L"old Windows result", app.settings);
|
||||||
|
|||||||
@@ -5,9 +5,11 @@
|
|||||||
static std::wstring programs;
|
static std::wstring programs;
|
||||||
static std::wstring activationExecutable, activationCommand;
|
static std::wstring activationExecutable, activationCommand;
|
||||||
static int activationCalls{};
|
static int activationCalls{};
|
||||||
|
static DWORD launchFlags{}; static bool inheritsHandles{};
|
||||||
static BOOL WINAPI testCreateProcess(LPCWSTR path, LPWSTR command, LPSECURITY_ATTRIBUTES,
|
static BOOL WINAPI testCreateProcess(LPCWSTR path, LPWSTR command, LPSECURITY_ATTRIBUTES,
|
||||||
LPSECURITY_ATTRIBUTES, BOOL, DWORD, LPVOID, LPCWSTR, LPSTARTUPINFOW, LPPROCESS_INFORMATION)
|
LPSECURITY_ATTRIBUTES, BOOL inherit, DWORD flags, LPVOID, LPCWSTR, LPSTARTUPINFOW, LPPROCESS_INFORMATION)
|
||||||
{
|
{
|
||||||
|
launchFlags=flags; inheritsHandles=inherit;
|
||||||
++activationCalls; activationExecutable=path; activationCommand=command;
|
++activationCalls; activationExecutable=path; activationCommand=command;
|
||||||
SetLastError(ERROR_CANCELLED); return FALSE; // Inspect activation without starting any program.
|
SetLastError(ERROR_CANCELLED); return FALSE; // Inspect activation without starting any program.
|
||||||
}
|
}
|
||||||
@@ -55,6 +57,14 @@ int main()
|
|||||||
valid=id.vt==VT_CLSID && *id.puuid==ks::ActivatorId; PropVariantClear(&id);
|
valid=id.vt==VT_CLSID && *id.puuid==ks::ActivatorId; PropVariantClear(&id);
|
||||||
if(!valid) throw std::runtime_error("shortcut activator missing");
|
if(!valid) throw std::runtime_error("shortcut activator missing");
|
||||||
ks::registerScreenshotNotifications(); // Registration is safe to repeat.
|
ks::registerScreenshotNotifications(); // Registration is safe to repeat.
|
||||||
|
bool launchFailed=false;
|
||||||
|
try { ks::startScreenshotNotificationRegistration(); }
|
||||||
|
catch(const std::exception &) { launchFailed=true; }
|
||||||
|
if(!launchFailed || activationExecutable!=helper.wstring() ||
|
||||||
|
activationCommand!=L"\""+helper.wstring()+L"\" --register-notifications" ||
|
||||||
|
!(launchFlags&CREATE_NO_WINDOW) || inheritsHandles)
|
||||||
|
throw std::runtime_error("registration must launch the sibling helper and propagate failure");
|
||||||
|
activationCalls=0;
|
||||||
ks::Activator activator(helper.wstring(),false);
|
ks::Activator activator(helper.wstring(),false);
|
||||||
activator.Activate(L"HrynCo.KSwitcher.Screenshot",L"check-update",nullptr,0);
|
activator.Activate(L"HrynCo.KSwitcher.Screenshot",L"check-update",nullptr,0);
|
||||||
auto main=(helper.parent_path()/L"KSwitcher.exe").wstring();
|
auto main=(helper.parent_path()/L"KSwitcher.exe").wstring();
|
||||||
|
|||||||
@@ -4,6 +4,7 @@
|
|||||||
#include "richtext.h"
|
#include "richtext.h"
|
||||||
#include "storage.h"
|
#include "storage.h"
|
||||||
#include "diagnostics.h"
|
#include "diagnostics.h"
|
||||||
|
#include "json.hpp"
|
||||||
#include <algorithm>
|
#include <algorithm>
|
||||||
#include <iostream>
|
#include <iostream>
|
||||||
#include <map>
|
#include <map>
|
||||||
@@ -189,6 +190,8 @@ int main(int argc, char **argv)
|
|||||||
"HTML entities and Unicode survive without script text");
|
"HTML entities and Unicode survive without script text");
|
||||||
require((effects(0, 6) & CFE_BOLD) != 0, "inline CSS bold is rendered");
|
require((effects(0, 6) & CFE_BOLD) != 0, "inline CSS bold is rendered");
|
||||||
auto plain = ks::textClip(L"{\\literal} Привіт");
|
auto plain = ks::textClip(L"{\\literal} Привіт");
|
||||||
|
require(ks::richPreview(plain, false).empty(), "plain history preview does not require RichEdit");
|
||||||
|
require(!ks::richPreview(styled, false).empty(), "formatted history preview retains HTML rendering");
|
||||||
ks::setRichText(rich, ks::richPreview(plain));
|
ks::setRichText(rich, ks::richPreview(plain));
|
||||||
GetWindowTextW(rich, rendered, 256);
|
GetWindowTextW(rich, rendered, 256);
|
||||||
require(std::wstring(rendered) == plain.text(), "plain Unicode and RTF metacharacters are escaped");
|
require(std::wstring(rendered) == plain.text(), "plain Unicode and RTF metacharacters are escaped");
|
||||||
@@ -635,6 +638,55 @@ int main(int argc, char **argv)
|
|||||||
"Snipping Tool republish retains one pinned history entry");
|
"Snipping Tool republish retains one pinned history entry");
|
||||||
history.add(changedImage, L"SnippingTool", defaults);
|
history.add(changedImage, L"SnippingTool", defaults);
|
||||||
require(history.entries().size() == 2, "a new screenshot is retained");
|
require(history.entries().size() == 2, "a new screenshot is retained");
|
||||||
|
auto copiedText = ks::textClip(L"text after screenshot");
|
||||||
|
history.add(copiedText, L"Editor", defaults);
|
||||||
|
history.add(copiedText, L"Editor", defaults);
|
||||||
|
require(history.entries().size() == 3 && !history.entries().front().image,
|
||||||
|
"text after screenshot is retained and consecutive text duplicates are skipped");
|
||||||
|
history.add(changedImage, L"SnippingTool", defaults);
|
||||||
|
history.add(changedImage, L"SnippingTool", defaults);
|
||||||
|
require(history.entries().size() == 4 && history.entries().front().image,
|
||||||
|
"image after text is retained and consecutive image duplicates are skipped");
|
||||||
|
history.add(withText, L"Editor", defaults);
|
||||||
|
require(history.entries().size() == 5,
|
||||||
|
"mixed image and text payload is not discarded by metadata prefilter");
|
||||||
|
}
|
||||||
|
{
|
||||||
|
auto folder = temp / L"fingerprints";
|
||||||
|
ks::Settings defaults;
|
||||||
|
ks::History history(folder);
|
||||||
|
history.add(firstImage, L"test", defaults);
|
||||||
|
auto id = history.entries().front().id;
|
||||||
|
history.pin(id);
|
||||||
|
history.saveOcr(id, L"retained OCR", defaults);
|
||||||
|
ks::History reopened(folder);
|
||||||
|
require(reopened.entries().front().contentHash.size() == 64,
|
||||||
|
"SHA-256 fingerprint survives encrypted index reload");
|
||||||
|
// Deny payload reads: duplicates must still be detected using only the index.
|
||||||
|
auto handle = CreateFileW((folder / (id + L".dat")).c_str(), GENERIC_READ, 0, nullptr,
|
||||||
|
OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, nullptr);
|
||||||
|
require(handle != INVALID_HANDLE_VALUE, "lock image payload for no-read regression");
|
||||||
|
reopened.add(firstImage, L"test", defaults);
|
||||||
|
reopened.add(secondImage, L"test", defaults);
|
||||||
|
reopened.add(recoded, L"test", defaults);
|
||||||
|
CloseHandle(handle);
|
||||||
|
require(reopened.entries().size() == 1 && reopened.entries().front().pinned &&
|
||||||
|
reopened.ocrText(id) == std::optional<std::wstring>(L"retained OCR"),
|
||||||
|
"exact, OLE and pixel-equivalent duplicates need no payload read and retain pin/OCR");
|
||||||
|
auto index = nlohmann::json::parse(ks::unprotect(ks::readFile(folder / L"index.dat", 1024*1024)));
|
||||||
|
for (auto &entry : index)
|
||||||
|
for (auto key : {"fingerprintVersion", "contentHash", "imageHash", "pixelHash"}) entry.erase(key);
|
||||||
|
auto legacy = index.dump();
|
||||||
|
ks::atomicWrite(folder / L"index.dat", ks::protect(ks::Bytes(legacy.begin(), legacy.end())));
|
||||||
|
ks::History migrated(folder);
|
||||||
|
migrated.add(secondImage, L"test", defaults);
|
||||||
|
ks::History persisted(folder);
|
||||||
|
require(persisted.entries().size() == 1 && persisted.entries().front().contentHash.size() == 64,
|
||||||
|
"legacy duplicate migrates fingerprints without adding or losing an entry");
|
||||||
|
persisted.add(changedImage, L"test", defaults);
|
||||||
|
require(persisted.entries().size() == 2, "pixel fingerprint preserves genuinely changed images");
|
||||||
|
persisted.add(firstImage, L"test", defaults);
|
||||||
|
require(persisted.entries().size() == 3, "hash deduplication remains consecutive rather than global");
|
||||||
}
|
}
|
||||||
ks::Settings settings;
|
ks::Settings settings;
|
||||||
settings.writeLog = false;
|
settings.writeLog = false;
|
||||||
|
|||||||
@@ -0,0 +1,61 @@
|
|||||||
|
param([ValidateRange(1,20)][int]$Runs=3,[ValidateRange(100,10000)][int]$SettleMilliseconds=1500)
|
||||||
|
# Private resident pages are counted externally, so the measurement buffer is
|
||||||
|
# allocated in PowerShell, not in the measured process. Never trim working sets.
|
||||||
|
$ErrorActionPreference='Stop'
|
||||||
|
$root=Split-Path $PSScriptRoot -Parent
|
||||||
|
$output=Join-Path $root 'build/memory-components'
|
||||||
|
New-Item -ItemType Directory -Force $output | Out-Null
|
||||||
|
$exe=Join-Path $output 'probe.exe'
|
||||||
|
& "$root/.tools/w64devkit/bin/g++.exe" -std=c++20 -Os -static -s -DUNICODE -D_UNICODE -D_WIN32_WINNT=0x0A00 "$PSScriptRoot/memory-components.cpp" -o $exe -lole32 -loleaut32 -lshell32 -lpropsys -lshlwapi -luuid -luiautomationcore
|
||||||
|
if($LASTEXITCODE) { throw 'Probe compilation failed' }
|
||||||
|
Add-Type @'
|
||||||
|
using System;
|
||||||
|
using System.Runtime.InteropServices;
|
||||||
|
public static class ComponentWorkingSet {
|
||||||
|
[DllImport("psapi.dll", SetLastError=true)] static extern bool QueryWorkingSet(IntPtr process,IntPtr buffer,int size);
|
||||||
|
public static long PrivateBytes(IntPtr process) {
|
||||||
|
int size=4*1024*1024;
|
||||||
|
IntPtr buffer=Marshal.AllocHGlobal(size);
|
||||||
|
try {
|
||||||
|
if(!QueryWorkingSet(process,buffer,size)) throw new System.ComponentModel.Win32Exception();
|
||||||
|
long count=Marshal.ReadIntPtr(buffer).ToInt64(), pages=0;
|
||||||
|
if(count>(size/IntPtr.Size)-1) throw new InvalidOperationException("Working set buffer is too small");
|
||||||
|
for(int i=0;i<count;i++) if((Marshal.ReadIntPtr(buffer,(i+1)*IntPtr.Size).ToInt64() & 256)==0) pages++;
|
||||||
|
return pages*Environment.SystemPageSize;
|
||||||
|
} finally { Marshal.FreeHGlobal(buffer); }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
'@
|
||||||
|
$rows=[Collections.Generic.List[object]]::new()
|
||||||
|
foreach($run in 1..$Runs) {
|
||||||
|
foreach($mode in @('baseline','sta','mta','shell','uia','focused','combined')) {
|
||||||
|
$start=[Diagnostics.ProcessStartInfo]::new($exe,$mode)
|
||||||
|
$start.UseShellExecute=$false; $start.CreateNoWindow=$true
|
||||||
|
$start.RedirectStandardOutput=$true; $start.RedirectStandardInput=$true; $start.RedirectStandardError=$true
|
||||||
|
$process=[Diagnostics.Process]::Start($start)
|
||||||
|
$errors=$process.StandardError.ReadToEndAsync()
|
||||||
|
try {
|
||||||
|
foreach($expected in @('baseline','com','active','released','uninitialized')) {
|
||||||
|
$pending=$process.StandardOutput.ReadLineAsync()
|
||||||
|
if(!$pending.Wait(15000)) { throw "Timeout: $mode/$expected" }
|
||||||
|
$phase=$pending.Result
|
||||||
|
if($phase -ne $expected) { throw "Unexpected phase '$phase', expected $expected" }
|
||||||
|
Start-Sleep -Milliseconds $SettleMilliseconds
|
||||||
|
$process.Refresh()
|
||||||
|
$rows.Add([pscustomobject]@{Run=$run;Mode=$mode;Phase=$phase;PrivateBytes=$process.PrivateMemorySize64;PrivateWorkingSet=[ComponentWorkingSet]::PrivateBytes($process.Handle);WorkingSet=$process.WorkingSet64;Threads=$process.Threads.Count;Modules=@($process.Modules | ForEach-Object ModuleName)})
|
||||||
|
$process.StandardInput.Write("`n"); $process.StandardInput.Flush()
|
||||||
|
}
|
||||||
|
if(!$process.WaitForExit(15000)) { throw 'Exit timeout' }
|
||||||
|
if($process.ExitCode) { throw "Probe failed: $($errors.Result)" }
|
||||||
|
$errors.Result | Set-Content (Join-Path $output "$run-$mode.txt")
|
||||||
|
} finally {
|
||||||
|
if(!$process.HasExited) { $process.Kill(); $process.WaitForExit() }
|
||||||
|
$process.Dispose()
|
||||||
|
$rows | ConvertTo-Json -Depth 5 | Set-Content (Join-Path $output 'results.json')
|
||||||
|
}
|
||||||
|
Write-Host "Completed $run/$Runs $mode"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
$rows | Group-Object Mode,Phase | ForEach-Object {
|
||||||
|
[pscustomobject]@{Case=$_.Name;PrivateMiB=[math]::Round(($_.Group.PrivateBytes | Measure-Object -Average).Average/1MB,3);PrivateWorkingSetMiB=[math]::Round(($_.Group.PrivateWorkingSet | Measure-Object -Average).Average/1MB,3)}
|
||||||
|
} | Format-Table
|
||||||
@@ -0,0 +1,65 @@
|
|||||||
|
// Standalone attribution probe. No clipboard, registry or user input mutations.
|
||||||
|
#include <windows.h>
|
||||||
|
#include <shlobj.h>
|
||||||
|
#include <propkey.h>
|
||||||
|
#include <propvarutil.h>
|
||||||
|
#include <uiautomation.h>
|
||||||
|
#include <cstdio>
|
||||||
|
#include <cstring>
|
||||||
|
#include <cstdlib>
|
||||||
|
|
||||||
|
static void check(HRESULT hr) {
|
||||||
|
if (FAILED(hr)) { fprintf(stderr,"HRESULT=%08lx\n",(unsigned long)hr); exit(2); }
|
||||||
|
}
|
||||||
|
static void phase(const char *name) {
|
||||||
|
printf("%s\n",name); fflush(stdout);
|
||||||
|
if (getchar()==EOF) exit(3);
|
||||||
|
}
|
||||||
|
int main(int argc,char **argv) {
|
||||||
|
const char *mode=argc>1?argv[1]:"baseline";
|
||||||
|
const bool baseline=!strcmp(mode,"baseline");
|
||||||
|
const bool shell=!strcmp(mode,"shell") || !strcmp(mode,"combined");
|
||||||
|
const bool uia=!strcmp(mode,"uia") || !strcmp(mode,"focused") || !strcmp(mode,"combined");
|
||||||
|
phase("baseline");
|
||||||
|
if(!baseline) check(CoInitializeEx(nullptr,!strcmp(mode,"sta") || shell ? COINIT_APARTMENTTHREADED : COINIT_MULTITHREADED));
|
||||||
|
phase("com");
|
||||||
|
IShellLinkW *link{}; IPropertyStore *store{}; IPersistFile *file{};
|
||||||
|
IUIAutomation2 *automation{}; IUIAutomationElement *element{}; IUIAutomationTextPattern *pattern{};
|
||||||
|
wchar_t temporary[MAX_PATH]{},shortcut[MAX_PATH]{},exe[MAX_PATH]{};
|
||||||
|
if(shell) {
|
||||||
|
check(CoCreateInstance(CLSID_ShellLink,nullptr,CLSCTX_INPROC_SERVER,IID_PPV_ARGS(&link)));
|
||||||
|
GetModuleFileNameW(nullptr,exe,MAX_PATH);
|
||||||
|
check(link->SetPath(exe)); check(link->SetIconLocation(exe,0));
|
||||||
|
check(link->QueryInterface(IID_PPV_ARGS(&store)));
|
||||||
|
PROPVARIANT value{}; check(InitPropVariantFromString(L"KSwitcher.MemoryProbe",&value));
|
||||||
|
check(store->SetValue(PKEY_AppUserModel_ID,value)); PropVariantClear(&value);
|
||||||
|
GUID id{0xe3f78047,0xf0ad,0x49b6,{0x90,0x74,0x41,0x19,0x31,0x70,0x61,0x67}};
|
||||||
|
value.vt=VT_CLSID; value.puuid=&id;
|
||||||
|
check(store->SetValue(PKEY_AppUserModel_ToastActivatorCLSID,value)); check(store->Commit());
|
||||||
|
PWSTR programs{}; check(SHGetKnownFolderPath(FOLDERID_Programs,0,nullptr,&programs)); CoTaskMemFree(programs);
|
||||||
|
GetTempPathW(MAX_PATH,temporary); if(!GetTempFileNameW(temporary,L"ksm",0,shortcut)) return 4;
|
||||||
|
check(link->QueryInterface(IID_PPV_ARGS(&file))); check(file->Save(shortcut,TRUE));
|
||||||
|
}
|
||||||
|
if(uia) {
|
||||||
|
check(CoCreateInstance(CLSID_CUIAutomation8,nullptr,CLSCTX_INPROC_SERVER,IID_PPV_ARGS(&automation)));
|
||||||
|
check(automation->put_ConnectionTimeout(200)); check(automation->put_TransactionTimeout(200));
|
||||||
|
if(strcmp(mode,"uia")) {
|
||||||
|
HRESULT focused=automation->GetFocusedElement(&element);
|
||||||
|
fprintf(stderr,"focus=%08lx\n",(unsigned long)focused);
|
||||||
|
if(SUCCEEDED(focused) && element) {
|
||||||
|
BOOL password=TRUE,enabled=FALSE; CONTROLTYPEID type{};
|
||||||
|
element->get_CurrentIsPassword(&password); element->get_CurrentIsEnabled(&enabled); element->get_CurrentControlType(&type);
|
||||||
|
fprintf(stderr,"type=%d enabled=%d protected=%d\n",type,enabled,password);
|
||||||
|
if(!password) fprintf(stderr,"pattern=%08lx\n",(unsigned long)element->GetCurrentPatternAs(UIA_TextPatternId,IID_PPV_ARGS(&pattern)));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
phase("active");
|
||||||
|
if(pattern) pattern->Release(); if(element) element->Release(); if(automation) automation->Release();
|
||||||
|
if(file) file->Release(); if(store) store->Release(); if(link) link->Release();
|
||||||
|
if(shortcut[0]) DeleteFileW(shortcut);
|
||||||
|
phase("released");
|
||||||
|
if(!baseline) { CoFreeUnusedLibrariesEx(0,0); CoUninitialize(); }
|
||||||
|
phase("uninitialized");
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user