/* Copyright (C) 2010-2012 kaosu (qiupf2000@gmail.com) * This file is part of the Interactive Text Hooker. * Interactive Text Hooker is free software: you can redistribute it and/or * modify it under the terms of the GNU General Public License as published * by the Free Software Foundation, either version 3 of the License, or * (at your option) any later version. * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * You should have received a copy of the GNU General Public License * along with this program. If not, see . */ #include "window.h" #include "ProcessWindow.h" #include "resource.h" #include "language.h" #include "host/host.h" #include "host/hookman.h" #include "vnrhook/include/const.h" #include "version.h" #include "ProfileManager.h" #include "host/settings.h" #include "CustomFilter.h" #include "profile/Profile.h" #include "TextBuffer.h" #include "profile/misc.h" #define CMD_SIZE 512 static WNDPROC proc, proccmd, procChar; static WCHAR last_cmd[CMD_SIZE]; extern HINSTANCE hIns; // main.cpp HWND hMainWnd, hwndCombo, hwndProcessComboBox, hwndEdit, hwndCmd; HWND hwndProcess; HWND hwndOption, hwndTop, hwndClear, hwndSave, hwndRemoveLink, hwndRemoveHook; HWND hProcDlg, hOptionDlg; HBRUSH hWhiteBrush; DWORD background; ProcessWindow* pswnd; TextBuffer* texts; extern ProfileManager* pfman; // ProfileManager.cpp extern HookManager* man; // main.cpp extern CustomFilter* mb_filter; // main.cpp extern CustomFilter* uni_filter; // main.cpp extern Settings* setman; // main.cpp #define COMMENT_BUFFER_LENGTH 512 static WCHAR comment_buffer[COMMENT_BUFFER_LENGTH]; LRESULT CALLBACK WndProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam); void SaveSettings(); // main.cpp extern LONG split_time, process_time, inject_delay, insert_delay, auto_inject, auto_insert, clipboard_flag, cyclic_remove, global_filter; //main.cpp static int last_select, last_edit; ATOM MyRegisterClass(HINSTANCE hInstance) { WNDCLASSEX wcex; wcex.cbSize = sizeof(WNDCLASSEX); wcex.style = CS_HREDRAW | CS_VREDRAW; wcex.lpfnWndProc = WndProc; wcex.cbClsExtra = 0; wcex.cbWndExtra = 0; wcex.hInstance = hInstance; wcex.hIcon = NULL; wcex.hCursor = NULL; wcex.hbrBackground = GetStockBrush(WHITE_BRUSH); wcex.lpszMenuName = NULL; wcex.lpszClassName = ClassName; wcex.hIconSm = LoadIcon(hInstance, (LPWSTR)IDI_ICON1); return RegisterClassEx(&wcex); } BOOL InitInstance(HINSTANCE hInstance, DWORD nAdmin, RECT* rc) { hIns = hInstance; LPCWSTR name = (nAdmin) ? ClassNameAdmin : ClassName; hMainWnd = CreateWindow(ClassName, name, WS_OVERLAPPEDWINDOW | WS_CLIPCHILDREN, rc->left, rc->top, rc->right - rc->left, rc->bottom - rc->top, NULL, NULL, hInstance, 0); if (!hMainWnd) return FALSE; ShowWindow(hMainWnd, SW_SHOWNORMAL); UpdateWindow(hMainWnd); return TRUE; } DWORD SaveProcessProfile(DWORD pid); // ProfileManager.cpp BOOL CALLBACK OptionDlgProc(HWND hDlg, UINT uMsg, WPARAM wParam, LPARAM lParam) { switch (uMsg) { case WM_INITDIALOG: { SetWindowText(GetDlgItem(hDlg, IDC_EDIT1), std::to_wstring((long long)split_time).c_str()); SetWindowText(GetDlgItem(hDlg, IDC_EDIT2), std::to_wstring((long long)process_time).c_str()); SetWindowText(GetDlgItem(hDlg, IDC_EDIT3), std::to_wstring((long long)inject_delay).c_str()); SetWindowText(GetDlgItem(hDlg, IDC_EDIT4), std::to_wstring((long long)insert_delay).c_str()); CheckDlgButton(hDlg, IDC_CHECK1, auto_inject); CheckDlgButton(hDlg, IDC_CHECK2, auto_insert); CheckDlgButton(hDlg, IDC_CHECK3, clipboard_flag); CheckDlgButton(hDlg, IDC_CHECK4, cyclic_remove); CheckDlgButton(hDlg, IDC_CHECK5, global_filter); } return TRUE; case WM_COMMAND: { DWORD wmId = LOWORD(wParam); DWORD wmEvent = HIWORD(wParam); switch (wmId) { case IDOK: { WCHAR str[128]; GetWindowText(GetDlgItem(hDlg, IDC_EDIT1), str, 0x80); DWORD st = std::stoul(str); split_time = st > 100 ? st : 100; GetWindowText(GetDlgItem(hDlg, IDC_EDIT2), str, 0x80); DWORD pt = std::stoul(str); process_time = pt > 50 ? pt : 50; GetWindowText(GetDlgItem(hDlg, IDC_EDIT3), str, 0x80); DWORD jd = std::stoul(str); inject_delay = jd > 1000 ? jd : 1000; GetWindowText(GetDlgItem(hDlg, IDC_EDIT4), str, 0x80); DWORD sd = std::stoul(str); insert_delay = sd > 200 ? sd : 200; if (IsDlgButtonChecked(hDlg, IDC_CHECK6)) { man->ResetRepeatStatus(); } auto_inject = IsDlgButtonChecked(hDlg, IDC_CHECK1); auto_insert = IsDlgButtonChecked(hDlg, IDC_CHECK2); clipboard_flag = IsDlgButtonChecked(hDlg, IDC_CHECK3); cyclic_remove = IsDlgButtonChecked(hDlg, IDC_CHECK4); global_filter = IsDlgButtonChecked(hDlg, IDC_CHECK5); setman->clipboardFlag = clipboard_flag; setman->splittingInterval = split_time; if (auto_inject == 0) auto_insert = 0; } case IDCANCEL: EndDialog(hDlg, 0); hOptionDlg = NULL; break; } return TRUE; } default: return FALSE; } return FALSE; } BOOL CALLBACK ProcessDlgProc(HWND hDlg, UINT uMsg, WPARAM wParam, LPARAM lParam) { switch (uMsg) { case WM_INITDIALOG: { pswnd = new ProcessWindow(hDlg); return TRUE; } case WM_COMMAND: { DWORD wmId, wmEvent; wmId = LOWORD(wParam); wmEvent = HIWORD(wParam); switch (wmId) { case WM_DESTROY: case IDOK: EndDialog(hDlg, NULL); hProcDlg = NULL; delete pswnd; pswnd = NULL; break; case IDC_BUTTON1: pswnd->RefreshProcess(); break; case IDC_BUTTON2: pswnd->AttachProcess(); break; case IDC_BUTTON3: pswnd->DetachProcess(); break; case IDC_BUTTON5: pswnd->CreateProfileForSelectedProcess(); break; case IDC_BUTTON6: pswnd->DeleteProfileForSelectedProcess(); break; } } return TRUE; case WM_NOTIFY: { LPNMHDR dr = (LPNMHDR)lParam; switch (dr->code) { case LVN_ITEMCHANGED: if (dr->idFrom == IDC_LIST1) { NMLISTVIEW *nmlv = (LPNMLISTVIEW)lParam; if (nmlv->uNewState & LVIS_SELECTED) pswnd->RefreshThread(nmlv->iItem); } break; } } return TRUE; default: return FALSE; } } LRESULT CALLBACK EditProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam) { switch (message) { case WM_CHAR: //Filter user input. if (GetKeyState(VK_CONTROL) & 0x8000) { if (wParam == 1) { Edit_SetSel(hwndEdit, 0, -1); SendMessage(hwndEdit, WM_COPY, 0, 0); } } return 0; case WM_LBUTTONUP: if (hwndEdit) SendMessage(hwndEdit, WM_COPY, 0, 0); default: { return proc(hWnd, message, wParam, lParam); } } } LRESULT CALLBACK EditCmdProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam) { switch (message) { case WM_KEYDOWN: if (wParam == VK_UP) { SetWindowText(hWnd, last_cmd); SetFocus(hWnd); return 0; } break; case WM_CHAR: if (wParam == VK_RETURN) { DWORD s = 0, pid = 0; WCHAR str[32]; if (GetWindowTextLength(hWnd) == 0) break; GetWindowText(hWnd, last_cmd, CMD_SIZE); //IthBreak(); if (GetWindowText(hwndProcessComboBox, str, 32)) pid = std::stoul(str); ProcessCommand(last_cmd, pid); Edit_SetSel(hWnd, 0, -1); Edit_ReplaceSel(hWnd, &s); SetFocus(hWnd); return 0; } default: break; } return CallWindowProc(proccmd, hWnd, message, wParam, lParam); } void CreateButtons(HWND hWnd) { hwndProcess = CreateWindow(L"Button", L"프로세스", WS_CHILD | WS_VISIBLE, 0, 0, 0, 0, hWnd, 0, hIns, NULL); hwndOption = CreateWindow(L"Button", L"옵션", WS_CHILD | WS_VISIBLE, 0, 0, 0, 0, hWnd, 0, hIns, NULL); hwndClear = CreateWindow(L"Button", L"지우기", WS_CHILD | WS_VISIBLE, 0, 0, 0, 0, hWnd, 0, hIns, NULL); hwndSave = CreateWindow(L"Button", L"저장", WS_CHILD | WS_VISIBLE, 0, 0, 0, 0, hWnd, 0, hIns, NULL); hwndRemoveLink = CreateWindow(L"Button", L"링크해제", WS_CHILD | WS_VISIBLE, 0, 0, 0, 0, hWnd, 0, hIns, NULL); hwndRemoveHook = CreateWindow(L"Button", L"후킹해제", WS_CHILD | WS_VISIBLE, 0, 0, 0, 0, hWnd, 0, hIns, NULL); hwndTop = CreateWindow(L"Button", L"항상위", WS_CHILD | WS_VISIBLE | BS_PUSHLIKE | BS_CHECKBOX, 0, 0, 0, 0, hWnd, 0, hIns, NULL); hwndProcessComboBox = CreateWindow(L"ComboBox", NULL, WS_CHILD | WS_VISIBLE | CBS_DROPDOWNLIST | CBS_SORT | WS_VSCROLL | WS_TABSTOP, 0, 0, 0, 0, hWnd, 0, hIns, NULL); hwndCmd = CreateWindowEx(WS_EX_CLIENTEDGE, L"Edit", NULL, WS_CHILD | WS_VISIBLE | ES_NOHIDESEL | ES_LEFT | ES_AUTOHSCROLL, 0, 0, 0, 0, hWnd, 0, hIns, NULL); hwndEdit = CreateWindowEx(WS_EX_CLIENTEDGE, L"Edit", NULL, WS_CHILD | WS_VISIBLE | ES_NOHIDESEL | WS_VSCROLL | ES_LEFT | ES_MULTILINE | ES_AUTOVSCROLL, 0, 0, 0, 0, hWnd, 0, hIns, NULL); } void ClickButton(HWND hWnd, HWND h) { if (h == hwndProcess) { if (hProcDlg) SetForegroundWindow(hProcDlg); else hProcDlg = CreateDialog(hIns, (LPWSTR)IDD_DIALOG2, 0, ProcessDlgProc); } else if (h == hwndOption) { if (hOptionDlg) SetForegroundWindow(hOptionDlg); else hOptionDlg = CreateDialog(hIns, (LPWSTR)IDD_DIALOG4, 0, OptionDlgProc); } else if (h == hwndClear) { WCHAR pwcEntry[128] = {}; DWORD dwId = ComboBox_GetCurSel(hwndCombo); int len = ComboBox_GetLBText(hwndCombo, dwId, pwcEntry); dwId = std::stoul(pwcEntry, NULL, 16); if (dwId == 0) man->ClearCurrent(); else man->RemoveSingleThread(dwId); } else if (h == hwndTop) { if (Button_GetCheck(h) == BST_CHECKED) { Button_SetCheck(h, BST_UNCHECKED); SetWindowPos(hWnd, HWND_NOTOPMOST, 0, 0, 0, 0, SWP_NOSIZE | SWP_NOMOVE); if (hProcDlg) SetWindowPos(hProcDlg, HWND_NOTOPMOST, 0, 0, 0, 0, SWP_NOSIZE | SWP_NOMOVE); if (hOptionDlg) SetWindowPos(hOptionDlg, HWND_NOTOPMOST, 0, 0, 0, 0, SWP_NOSIZE | SWP_NOMOVE); } else { Button_SetCheck(h, BST_CHECKED); SetWindowPos(hWnd, HWND_TOPMOST, 0, 0, 0, 0, SWP_NOSIZE | SWP_NOMOVE); if (hProcDlg) SetWindowPos(hProcDlg, HWND_TOPMOST, 0, 0, 0, 0, SWP_NOSIZE | SWP_NOMOVE); if (hOptionDlg) SetWindowPos(hOptionDlg, HWND_TOPMOST, 0, 0, 0, 0, SWP_NOSIZE | SWP_NOMOVE); } } else if (h == hwndSave) { WCHAR str[32]; if (GetWindowText(hwndProcessComboBox, str, 32)) { DWORD pid = std::stoul(str); SaveProcessProfile(pid); } pfman->SaveProfiles(); } else if (h == hwndRemoveLink) { WCHAR str[32]; if (GetWindowText(hwndCombo, str, 32)) { DWORD from = std::stoul(str, NULL, 16); if (from != 0) Host_UnLink(from); } } else if (h == hwndRemoveHook) { WCHAR str[32]; if (GetWindowText(hwndCombo, str, 32)) { std::wstring entry(str); std::size_t i; DWORD threadNumber = std::stoul(entry, &i, 16); entry = entry.substr(i + 1); DWORD pid = std::stoul(entry, &i); entry = entry.substr(i + 1); DWORD addr = std::stoul(entry, NULL, 16); if (threadNumber != 0) Host_RemoveHook(pid, addr); } } } DWORD ThreadFilter(TextThread* thread, BYTE* out, DWORD len, DWORD new_line, PVOID data, bool space) { DWORD status = thread->Status(); if (global_filter && !new_line && thread->Number() != 0) { if (status & USING_UNICODE) { DWORD i, j; len /= 2; LPWSTR str = (LPWSTR)out; for (i = 0, j = 0; i < len; i++) { WCHAR c = str[i]; if (!uni_filter->Find(c)) str[j++] = c; } memset(str + j, 0, (len - j) * 2); len = j * 2; } else { DWORD i, j; for (i = 0, j = 0; i < len; i++) { WORD c = out[i]; if (!IsDBCSLeadByte(c & 0xFF)) { if (!mb_filter->Find(c)) out[j++] = c & 0xFF; } else if (i + 1 < len) { c = out[i + 1]; c <<= 8; c |= out[i]; if (!mb_filter->Find(c)) { out[j++] = c & 0xFF; out[j++] = c >> 8; } i++; } } memset(out + j, 0, len - j); len = j; } } return len; } DWORD ThreadOutput(TextThread* thread, BYTE* out, DWORD len, DWORD new_line, PVOID data, bool space) { if (len == 0) return len; DWORD status = thread->Status(); if (status & CURRENT_SELECT) { if (new_line) { if (thread->Number() == 0) texts->AddText(L"\r\n", 2, true); else texts->AddText(L"\r\n\r\n", 4, true); } else if (status & USING_UNICODE) { texts->AddText((LPWSTR)out, len / 2, false); } else { int uni_len = MB_WC_count((char*)out, len); LPWSTR str = new WCHAR[uni_len + 1]; MB_WC((char*)out, str, uni_len + 1); str[uni_len] = L'\0'; texts->AddText(str, uni_len, false); delete str; } } return len; } bool GetHookParam(DWORD pid, DWORD hook_addr, HookParam& hp) { if (!pid) return false; ProcessRecord *pr = ::man->GetProcessRecord(pid); if (!pr) return false; bool result = false; WaitForSingleObject(pr->hookman_mutex, 0); const Hook *hks = (Hook *)pr->hookman_map; for (int i = 0; i < MAX_HOOK; i++) { if (hks[i].Address() == hook_addr) { hp = hks[i].hp; result = true; break; } } ReleaseMutex(pr->hookman_mutex); return result; } std::wstring GetEntryString(TextThread& thread) { CHAR entry[512]; thread.GetEntryString(entry, 512); return toUnicodeString(entry); } std::wstring CreateEntryWithLink(TextThread& thread, std::wstring& entry) { std::wstring entryWithLink = entry; if (thread.Link()) entryWithLink += L"->" + ToHexString(thread.LinkNumber()); if (thread.PID() == 0) entryWithLink += L"ConsoleOutput"; HookParam hp = {}; if (GetHookParam(thread.PID(), thread.Addr(), hp)) entryWithLink += L" (" + GetCode(hp, thread.PID()) + L")"; return entryWithLink; } void AddToCombo(TextThread& thread, bool replace) { std::wstring entry = GetEntryString(thread); std::wstring entryWithLink = CreateEntryWithLink(thread, entry); int i = ComboBox_FindString(hwndCombo, -1, entry.c_str()); if (replace) { int sel = ComboBox_GetCurSel(hwndCombo); if (i != CB_ERR) ComboBox_DeleteString(hwndCombo, i); ComboBox_AddString(hwndCombo, entryWithLink.c_str()); ComboBox_SetCurSel(hwndCombo, sel); } else { if (i == CB_ERR) ComboBox_AddString(hwndCombo, entryWithLink.c_str()); // Why set current selection to 0 when the new thread is selected? if (thread.Status() & CURRENT_SELECT) ComboBox_SetCurSel(hwndCombo, 0); } } void RemoveFromCombo(TextThread* thread) { CHAR entry[512]; thread->GetEntryString(entry, 512); std::wstring unicodeEntry = toUnicodeString(entry); if (thread->PID() == 0) unicodeEntry += L"ConsoleOutput"; int i = ComboBox_FindString(hwndCombo, 0, unicodeEntry.c_str()); if (i != CB_ERR) { if (ComboBox_DeleteString(hwndCombo, i) == CB_ERR) ConsoleOutput(ErrorDeleteCombo); } } void ComboSelectCurrent(TextThread* thread) { ComboBox_SetCurSel(hwndCombo, thread->Number()); } DWORD SetEditText(LPWSTR wc) { DWORD line; Edit_SetText(hwndEdit, wc); line = Edit_GetLineCount(hwndEdit); SendMessage(hwndEdit, EM_LINESCROLL, 0, line); return 0; } DWORD ThreadReset(TextThread* thread) { texts->ClearBuffer(); man->SetCurrent(thread); thread->LockVector(); DWORD uni = thread->Status() & USING_UNICODE; if (uni) { DWORD len = 0; LPWSTR wc = (LPWSTR)thread->GetStore(&len); len /= 2; wc[len] = L'\0'; SetEditText(wc); } else { DWORD len = MB_WC_count((char*)thread->Storage(), thread->Used()); LPWSTR wc = new WCHAR[len + 1]; MB_WC((char*)thread->Storage(), wc, len + 1); wc[len] = L'\0'; SetEditText(wc); delete wc; } WCHAR buffer[16]; std::swprintf(buffer, L"%04X", thread->Number()); DWORD tmp = ComboBox_FindString(hwndCombo, 0, buffer); if (tmp != CB_ERR) ComboBox_SetCurSel(hwndCombo, tmp); thread->UnlockVector(); return 0; } DWORD AddRemoveLink(TextThread* thread) { AddToCombo(*thread, true); return 0; } bool IsUnicodeHook(const ProcessRecord& pr, DWORD hook); void AddLinksToHookManager(const Profile* pf, size_t thread_index, const TextThread* thread); DWORD ThreadCreate(TextThread* thread) { thread->RegisterOutputCallBack(ThreadOutput, 0); thread->RegisterFilterCallBack(ThreadFilter, 0); AddToCombo(*thread, false); const auto& tp = thread->GetThreadParameter(); auto pr = man->GetProcessRecord(tp->pid); if (pr == NULL) return 0; if (IsUnicodeHook(*pr, tp->hook)) thread->Status() |= USING_UNICODE; auto pf = pfman->GetProfile(tp->pid); if (!pf) return 0; const std::wstring& hook_name = GetHookNameByAddress(*pr, thread->GetThreadParameter()->hook); auto thread_profile = pf->FindThread(thread->GetThreadParameter(), hook_name); if (thread_profile != pf->Threads().end()) { (*thread_profile)->HookManagerIndex() = thread->Number(); auto thread_index = thread_profile - pf->Threads().begin(); AddLinksToHookManager(pf, thread_index, thread); if (pf->IsThreadSelected(thread_profile)) ThreadReset(thread); } return 0; } bool IsUnicodeHook(const ProcessRecord& pr, DWORD hook) { bool res = false; WaitForSingleObject(pr.hookman_mutex, 0); auto hooks = (const Hook*)pr.hookman_map; for (DWORD i = 0; i < MAX_HOOK; i++) { if (hooks[i].Address() == hook) { res = hooks[i].Type() & USING_UNICODE; break; } } ReleaseMutex(pr.hookman_mutex); return res; } void AddLinksToHookManager(const Profile* pf, size_t thread_index, const TextThread* thread) { for (auto lp = pf->Links().begin(); lp != pf->Links().end(); ++lp) { if ((*lp)->FromIndex() == thread_index) { WORD to_index = pf->Threads()[(*lp)->ToIndex()]->HookManagerIndex(); if (to_index != 0) man->AddLink(thread->Number(), to_index); } if ((*lp)->ToIndex() == thread_index) { WORD from_index = pf->Threads()[(*lp)->FromIndex()]->HookManagerIndex(); if (from_index != 0) man->AddLink(from_index, thread->Number()); } } } DWORD ThreadRemove(TextThread* thread) { RemoveFromCombo(thread); return 0; } DWORD RegisterProcessList(DWORD pid) { auto path = GetProcessPath(pid); if (!path.empty()) { WCHAR str[MAX_PATH]; std::swprintf(str, L"%04d:%s", pid, path.substr(path.rfind(L'\\') + 1).c_str()); ComboBox_AddString(hwndProcessComboBox, str); if (ComboBox_GetCount(hwndProcessComboBox) == 1) ComboBox_SetCurSel(hwndProcessComboBox, 0); } return 0; } DWORD RemoveProcessList(DWORD pid) { WCHAR str[MAX_PATH]; std::swprintf(str, L"%04d", pid); DWORD i = ComboBox_FindString(hwndProcessComboBox, 0, str); DWORD j = ComboBox_GetCurSel(hwndProcessComboBox); if (i != CB_ERR) { DWORD k = ComboBox_DeleteString(hwndProcessComboBox, i); if (i == j) ComboBox_SetCurSel(hwndProcessComboBox, 0); } return 0; } DWORD RefreshProfileOnNewHook(DWORD pid) { return 0; } LRESULT CALLBACK WndProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam) { switch (message) { case WM_CREATE: CreateButtons(hWnd); // Add text to the window. Edit_LimitText(hwndEdit, -1); SendMessage(hwndEdit, WM_INPUTLANGCHANGEREQUEST, 0, 0x411); proc = (WNDPROC)SetWindowLong(hwndEdit, GWL_WNDPROC, (LONG)EditProc); proccmd = (WNDPROC)SetWindowLong(hwndCmd, GWL_WNDPROC, (LONG)EditCmdProc); hwndCombo = CreateWindow(L"ComboBox", NULL, WS_CHILD | WS_VISIBLE | CBS_DROPDOWNLIST | CBS_SORT | WS_VSCROLL | WS_TABSTOP, 0, 0, 0, 0, hWnd, 0, hIns, NULL); { HDC hDC = GetDC(hWnd); int nHeight = -MulDiv(12, GetDeviceCaps(hDC, LOGPIXELSY), 72); ReleaseDC(hWnd, hDC); HFONT hf = CreateFont(nHeight, 0, 0, 0, FW_LIGHT, FALSE, FALSE, FALSE, SHIFTJIS_CHARSET, OUT_DEFAULT_PRECIS, CLIP_DEFAULT_PRECIS, CLEARTYPE_QUALITY, DEFAULT_PITCH | FF_DONTCARE, L"MS Gothic"); hWhiteBrush = GetStockBrush(WHITE_BRUSH); SendMessage(hwndCmd, WM_SETFONT, (WPARAM)hf, 0); SendMessage(hwndEdit, WM_SETFONT, (WPARAM)hf, 0); SendMessage(hwndCombo, WM_SETFONT, (WPARAM)hf, 0); SendMessage(hwndProcessComboBox, WM_SETFONT, (WPARAM)hf, 0); texts = new TextBuffer(hwndEdit); man->RegisterThreadCreateCallback(ThreadCreate); man->RegisterThreadRemoveCallback(ThreadRemove); man->RegisterThreadResetCallback(ThreadReset); TextThread* console = man->FindSingle(0); console->RegisterOutputCallBack(ThreadOutput, NULL); AddToCombo(*console, false); man->RegisterProcessAttachCallback(RegisterProcessList); man->RegisterProcessDetachCallback(RemoveProcessList); man->RegisterProcessNewHookCallback(RefreshProfileOnNewHook); man->RegisterAddRemoveLinkCallback(AddRemoveLink); man->RegisterConsoleCallback(ConsoleOutput); Host_Start(); { static const WCHAR program_name[] = L"Interactive Text Hooker"; //static const WCHAR program_version[] = L"3.0"; static WCHAR version_info[256]; std::swprintf(version_info, L"%s %s (%s)", program_name, program_version, build_date); man->AddConsoleOutput(version_info); man->AddConsoleOutput(InitMessage); } if (background == 0) man->AddConsoleOutput(BackgroundMsg); } return 0; case WM_COMMAND: { DWORD wmId, wmEvent, dwId; wmId = LOWORD(wParam); wmEvent = HIWORD(wParam); switch (wmEvent) { case EN_VSCROLL: { SCROLLBARINFO info = { sizeof(info) }; GetScrollBarInfo(hwndEdit, OBJID_VSCROLL, &info); InvalidateRect(hwndEdit, 0, 1); ValidateRect(hwndEdit, &info.rcScrollBar); RedrawWindow(hwndEdit, 0, 0, RDW_ERASE); } break; case CBN_SELENDOK: { if ((HWND)lParam == hwndProcessComboBox) return 0; dwId = ComboBox_GetCurSel(hwndCombo); int len = ComboBox_GetLBTextLen(hwndCombo, dwId); if (len > 0) { LPWSTR pwcEntry = new WCHAR[len + 1]; len = ComboBox_GetLBText(hwndCombo, dwId, pwcEntry); DWORD num = std::stoul(pwcEntry, NULL, 16); man->SelectCurrent(num); delete[] pwcEntry; } } return 0; case BN_CLICKED: ClickButton(hWnd, (HWND)lParam); break; default: break; } } break; case WM_SETFOCUS: SetFocus(hwndEdit); return 0; case WM_SIZE: { WORD width = LOWORD(lParam); WORD height = HIWORD(lParam); DWORD l = width / 7; WORD h = HIWORD(GetDialogBaseUnits()); // height of the system font h = h + (h / 2); HDC hDC = GetDC(hWnd); RECT rc; GetClientRect(hWnd, &rc); FillRect(hDC, &rc, hWhiteBrush); ReleaseDC(hWnd, hDC); MoveWindow(hwndProcess, 0, 0, l, h, TRUE); MoveWindow(hwndOption, l * 1, 0, l, h, TRUE); MoveWindow(hwndTop, l * 2, 0, l, h, TRUE); MoveWindow(hwndClear, l * 3, 0, l, h, TRUE); MoveWindow(hwndRemoveLink, l * 4, 0, l, h, TRUE); MoveWindow(hwndRemoveHook, l * 5, 0, l, h, TRUE); MoveWindow(hwndSave, l * 6, 0, width - 6 * l, h, TRUE); l *= 2; MoveWindow(hwndProcessComboBox, 0, h, l, 200, TRUE); MoveWindow(hwndCmd, l, h, width - l, h, TRUE); MoveWindow(hwndCombo, 0, h * 2, width, 200, TRUE); h *= 3; MoveWindow(hwndEdit, 0, h, width, height - h, TRUE); } return 0; case WM_DESTROY: man->RegisterThreadCreateCallback(0); man->RegisterThreadRemoveCallback(0); man->RegisterThreadResetCallback(0); man->RegisterProcessAttachCallback(0); man->RegisterProcessDetachCallback(0); //delete texts; SaveSettings(); PostQuitMessage(0); return 0; default: return DefWindowProc(hWnd, message, wParam, lParam); } return 0; } DWORD WINAPI FlushThread(LPVOID lParam) { TextBuffer* t = (TextBuffer*)lParam; while (t->Running()) { t->Flush(); Sleep(10); } return 0; }