#include "Windows.h" #include "Richedit.h" #include "Version.h" #include #include #include #include #include "Memory.h" #include "Random.h" #include "Randomizer.h" // Heartbeat is defined to 0x401 by Memory.h #define RANDOMIZE_READY 0x402 #define RANDOMIZING 0403 #define RANDOMIZE_DONE 0x404 #define RANDOMIZE_CHALLENGE_DONE 0x405 #define CHALLENGE_ONLY 0x406 #define DISABLE_SNIPES 0x407 #define SPEED_UP_AUTOSCROLLERS 0x408 /* ------- Temp ------- */ #define TMP1 0x501 #define TMP2 0x502 #define TMP3 0x503 #define TMP4 0x504 #include "Panel.h" int panel = 0x33D4; std::shared_ptr g_panel; /* ------- Temp ------- */ // Globals HWND g_hwnd; HWND g_seed; HWND g_randomizerStatus; HINSTANCE g_hInstance; auto g_witnessProc = std::make_shared(L"witness64_d3d11.exe"); std::shared_ptr g_randomizer; void SetRandomSeed(); LRESULT CALLBACK WndProc(HWND hwnd, UINT message, WPARAM wParam, LPARAM lParam) { if (message == WM_DESTROY) { PostQuitMessage(0); } else if (message == WM_NOTIFY) { MSGFILTER* m = (MSGFILTER *)lParam; if (m->msg == WM_KEYDOWN && m->wParam == VK_RETURN) { if (IsWindowEnabled(g_randomizerStatus) == TRUE) { PostMessage(g_hwnd, WM_COMMAND, RANDOMIZING, NULL); return 1; // Non-zero to indicate that message was handled } } } else if (message == WM_COMMAND || message == WM_TIMER || message == WM_NOTIFY) { switch (LOWORD(wParam)) { case HEARTBEAT: switch ((ProcStatus)lParam) { case ProcStatus::NotRunning: // Shut down randomizer, wait for startup if (g_randomizer) { g_randomizer = nullptr; EnableWindow(g_randomizerStatus, FALSE); } break; case ProcStatus::Running: if (!g_randomizer) { g_randomizer = std::make_shared(g_witnessProc); PostMessage(g_hwnd, WM_COMMAND, RANDOMIZE_READY, NULL); } break; case ProcStatus::NewGame: // This status will fire only once per new game SetWindowText(g_seed, L""); PostMessage(g_hwnd, WM_COMMAND, RANDOMIZE_READY, NULL); break; } break; case RANDOMIZE_READY: EnableWindow(g_randomizerStatus, TRUE); if (IsDlgButtonChecked(hwnd, CHALLENGE_ONLY)) { SetWindowText(g_randomizerStatus, L"Randomize Challenge"); } else { SetWindowText(g_randomizerStatus, L"Randomize"); } break; case RANDOMIZING: if (!g_randomizer) break; // E.g. an enter press at the wrong time EnableWindow(g_randomizerStatus, FALSE); SetRandomSeed(); std::thread([]{ if (IsDlgButtonChecked(g_hwnd, DISABLE_SNIPES)) { g_randomizer->PreventSnipes(); } if (IsDlgButtonChecked(g_hwnd, SPEED_UP_AUTOSCROLLERS)) { g_randomizer->AdjustSpeed(); } if (IsDlgButtonChecked(g_hwnd, CHALLENGE_ONLY)) { SetWindowText(g_randomizerStatus, L"Randomizing Challenge..."); g_randomizer->RandomizeChallenge(); PostMessage(g_hwnd, WM_COMMAND, RANDOMIZE_CHALLENGE_DONE, NULL); } else { SetWindowText(g_randomizerStatus, L"Randomizing..."); g_randomizer->Randomize(); PostMessage(g_hwnd, WM_COMMAND, RANDOMIZE_DONE, NULL); } }).detach(); break; case RANDOMIZE_DONE: EnableWindow(g_randomizerStatus, FALSE); SetWindowText(g_randomizerStatus, L"Randomized!"); break; case RANDOMIZE_CHALLENGE_DONE: EnableWindow(g_randomizerStatus, FALSE); SetWindowText(g_randomizerStatus, L"Randomized Challenge!"); std::thread([]{ // Allow re-randomization for challenge -- it doesn't break the rest of the game. std::this_thread::sleep_for(std::chrono::seconds(10)); PostMessage(g_hwnd, WM_COMMAND, RANDOMIZE_READY, NULL); }).detach(); break; case CHALLENGE_ONLY: CheckDlgButton(hwnd, CHALLENGE_ONLY, !IsDlgButtonChecked(hwnd, CHALLENGE_ONLY)); if (IsWindowEnabled(g_randomizerStatus)) { PostMessage(g_hwnd, WM_COMMAND, RANDOMIZE_READY, NULL); } break; case DISABLE_SNIPES: CheckDlgButton(hwnd, DISABLE_SNIPES, !IsDlgButtonChecked(hwnd, DISABLE_SNIPES)); break; case SPEED_UP_AUTOSCROLLERS: CheckDlgButton(hwnd, SPEED_UP_AUTOSCROLLERS, !IsDlgButtonChecked(hwnd, SPEED_UP_AUTOSCROLLERS)); break; case TMP1: g_panel = std::make_shared(g_witnessProc, panel); break; case TMP2: if(g_panel) g_panel->Write(panel); break; case TMP4: if(g_panel) g_panel->Serialize(); break; } } return DefWindowProc(hwnd, message, wParam, lParam); } void SetRandomSeed() { std::wstring text(128, L'\0'); int length = GetWindowText(g_seed, text.data(), static_cast(text.size())); if (length > 0) { // Set seed text.resize(length); Random::SetSeed(_wtoi(text.c_str())); } else { // Random seed int seed = Random::RandInt(0, 999999); SetWindowText(g_seed, std::to_wstring(seed).c_str()); RedrawWindow(g_seed, NULL, NULL, RDW_UPDATENOW); Random::SetSeed(seed); } } HWND CreateLabel(int x, int y, int width, LPCWSTR text) { return CreateWindow(L"STATIC", text, WS_TABSTOP | WS_VISIBLE | WS_CHILD | SS_LEFT, x, y, width, 16, g_hwnd, NULL, g_hInstance, NULL); } HWND CreateText(int x, int y, int width, LPCWSTR defaultText = L"") { return CreateWindow(MSFTEDIT_CLASS, defaultText, WS_TABSTOP | WS_VISIBLE | WS_CHILD | WS_BORDER, x, y, width, 26, g_hwnd, NULL, g_hInstance, NULL); } #pragma warning(push) #pragma warning(disable: 4312) HWND CreateButton(int x, int y, int width, LPCWSTR text, int message) { return CreateWindow(L"BUTTON", text, WS_TABSTOP | WS_VISIBLE | WS_CHILD | BS_DEFPUSHBUTTON, x, y, width, 26, g_hwnd, (HMENU)message, g_hInstance, NULL); } HWND CreateCheckbox(int x, int y, int message) { return CreateWindow(L"BUTTON", L"", WS_VISIBLE | WS_CHILD | BS_CHECKBOX, x, y, 12, 12, g_hwnd, (HMENU)message, g_hInstance, NULL); } #pragma warning(pop) int APIENTRY wWinMain(_In_ HINSTANCE hInstance, _In_opt_ HINSTANCE hPrevInstance, _In_ LPWSTR lpCmdLine, _In_ int nCmdShow) { LoadLibrary(L"Msftedit.dll"); WNDCLASSW wndClass = { CS_HREDRAW | CS_VREDRAW, WndProc, 0, 0, hInstance, NULL, LoadCursor(nullptr, IDC_ARROW), (HBRUSH)(COLOR_WINDOW+1), WINDOW_CLASS, WINDOW_CLASS, }; RegisterClassW(&wndClass); g_hInstance = hInstance; RECT rect; GetClientRect(GetDesktopWindow(), &rect); g_hwnd = CreateWindow(WINDOW_CLASS, PRODUCT_NAME, WS_OVERLAPPEDWINDOW, rect.right - 550, 200, 500, 500, nullptr, nullptr, hInstance, nullptr); CreateLabel(390, 15, 90, L"Version: " VERSION_STR); g_seed = CreateText(10, 10, 100); PostMessage(g_seed, EM_SETEVENTMASK, 0, ENM_KEYEVENTS); g_randomizerStatus = CreateButton(120, 10, 180, L"Randomize", RANDOMIZING); EnableWindow(g_randomizerStatus, FALSE); CreateCheckbox(10, 300, CHALLENGE_ONLY); CreateLabel(30, 300, 200, L"Randomize the challenge only"); CreateCheckbox(10, 320, DISABLE_SNIPES); CheckDlgButton(g_hwnd, DISABLE_SNIPES, TRUE); CreateLabel(30, 320, 240, L"Disable Swamp and Shadows snipes"); CreateCheckbox(10, 340, SPEED_UP_AUTOSCROLLERS); CreateLabel(30, 340, 205, L"Speed up various autoscrollers"); // CreateButton(200, 100, 100, L"Read", TMP1); // CreateButton(200, 130, 100, L"Write", TMP2); // CreateButton(200, 160, 100, L"Random", TMP3); // CreateButton(200, 190, 100, L"Dump", TMP4); g_witnessProc->StartHeartbeat(g_hwnd); ShowWindow(g_hwnd, nCmdShow); UpdateWindow(g_hwnd); MSG msg; while (GetMessage(&msg, nullptr, 0, 0) == TRUE) { TranslateMessage(&msg); DispatchMessage(&msg); } return (int) msg.wParam; }