Textractor_test/GUI/mainwindow.cpp

627 lines
25 KiB
C++
Raw Normal View History

2019-04-23 01:35:40 +08:00
#include "mainwindow.h"
2018-07-22 06:40:16 +08:00
#include "ui_mainwindow.h"
2018-12-19 01:14:54 +08:00
#include "defs.h"
#include "host/util.h"
2019-02-19 11:39:04 +08:00
#include <shellapi.h>
#include <QStringListModel>
2019-09-12 02:27:57 +08:00
#include <QScrollBar>
#include <QMenu>
#include <QDialogButtonBox>
#include <QFileDialog>
#include <QFontDialog>
2018-07-24 03:25:02 +08:00
2019-02-28 00:33:17 +08:00
extern const char* ATTACH;
extern const char* LAUNCH;
extern const char* DETACH;
extern const char* FORGET;
2019-02-28 00:33:17 +08:00
extern const char* ADD_HOOK;
2019-06-10 13:49:11 +08:00
extern const char* REMOVE_HOOKS;
2019-02-28 00:33:17 +08:00
extern const char* SAVE_HOOKS;
extern const char* SEARCH_FOR_HOOKS;
2019-02-28 00:33:17 +08:00
extern const char* SETTINGS;
extern const char* EXTENSIONS;
extern const char* FONT;
2019-02-28 00:33:17 +08:00
extern const char* SELECT_PROCESS;
extern const char* ATTACH_INFO;
extern const char* SELECT_PROCESS_INFO;
extern const char* FROM_COMPUTER;
2019-02-28 00:33:17 +08:00
extern const char* PROCESSES;
extern const char* CODE_INFODUMP;
2019-06-09 18:19:54 +08:00
extern const char* HOOK_SEARCH_UNSTABLE_WARNING;
extern const char* SEARCH_CJK;
2019-06-02 14:09:17 +08:00
extern const char* SEARCH_PATTERN;
extern const char* SEARCH_DURATION;
extern const char* SEARCH_MODULE;
2019-06-02 14:09:17 +08:00
extern const char* PATTERN_OFFSET;
extern const char* MIN_ADDRESS;
extern const char* MAX_ADDRESS;
extern const char* STRING_OFFSET;
extern const char* MAX_HOOK_SEARCH_RECORDS;
extern const char* HOOK_SEARCH_FILTER;
extern const char* SEARCH_FOR_TEXT;
extern const char* TEXT;
extern const char* CODEPAGE;
2019-06-02 14:09:17 +08:00
extern const char* START_HOOK_SEARCH;
extern const char* SAVE_SEARCH_RESULTS;
extern const char* TEXT_FILES;
2019-06-11 10:47:58 +08:00
extern const char* DOUBLE_CLICK_TO_REMOVE_HOOK;
2019-02-28 00:33:17 +08:00
extern const char* SAVE_SETTINGS;
extern const char* USE_JP_LOCALE;
extern const char* FILTER_REPETITION;
extern const char* AUTO_ATTACH;
extern const char* ATTACH_SAVED_ONLY;
2019-08-10 16:18:30 +08:00
extern const char* SHOW_SYSTEM_PROCESSES;
2019-02-28 00:33:17 +08:00
extern const char* DEFAULT_CODEPAGE;
extern const char* FLUSH_DELAY;
extern const char* MAX_BUFFER_SIZE;
2019-08-12 23:10:33 +08:00
extern const char* MAX_HISTORY_SIZE;
2019-02-28 00:33:17 +08:00
extern const wchar_t* ABOUT;
extern const wchar_t* CL_OPTIONS;
extern const wchar_t* LAUNCH_FAILED;
extern const wchar_t* INVALID_CODE;
2018-07-22 06:40:16 +08:00
MainWindow::MainWindow(QWidget *parent) :
2018-07-26 01:46:59 +08:00
QMainWindow(parent),
2018-11-01 22:38:14 +08:00
ui(new Ui::MainWindow),
2018-11-10 14:17:02 +08:00
extenWindow(new ExtenWindow(this))
2018-07-22 06:40:16 +08:00
{
2018-07-26 01:46:59 +08:00
ui->setupUi(this);
for (auto [text, slot] : Array<std::tuple<const char*, void(MainWindow::*)()>>{
2018-12-19 01:14:54 +08:00
{ ATTACH, &MainWindow::AttachProcess },
{ LAUNCH, &MainWindow::LaunchProcess },
2018-12-19 01:14:54 +08:00
{ DETACH, &MainWindow::DetachProcess },
{ FORGET, &MainWindow::ForgetProcess },
2018-12-19 01:14:54 +08:00
{ ADD_HOOK, &MainWindow::AddHook },
2019-06-10 13:49:11 +08:00
{ REMOVE_HOOKS, &MainWindow::RemoveHooks },
2018-12-19 01:14:54 +08:00
{ SAVE_HOOKS, &MainWindow::SaveHooks },
{ SEARCH_FOR_HOOKS, &MainWindow::FindHooks },
2018-12-19 01:14:54 +08:00
{ SETTINGS, &MainWindow::Settings },
{ EXTENSIONS, &MainWindow::Extensions }
2018-12-15 11:26:49 +08:00
})
{
auto button = new QPushButton(text, ui->processFrame);
2018-12-15 11:26:49 +08:00
connect(button, &QPushButton::clicked, this, slot);
ui->processLayout->addWidget(button);
2018-12-15 11:26:49 +08:00
}
ui->processLayout->addItem(new QSpacerItem(0, 0, QSizePolicy::Minimum, QSizePolicy::Expanding));
2018-08-22 10:43:30 +08:00
connect(ui->ttCombo, qOverload<int>(&QComboBox::activated), this, &MainWindow::ViewThread);
2019-01-23 03:57:13 +08:00
connect(ui->textOutput, &QPlainTextEdit::selectionChanged, [this] { if (!(QApplication::mouseButtons() & Qt::LeftButton)) ui->textOutput->copy(); });
connect(ui->textOutput, &QPlainTextEdit::customContextMenuRequested, this, &MainWindow::OutputContextMenu);
2018-12-19 01:14:54 +08:00
QSettings settings(CONFIG_FILE, QSettings::IniFormat);
2019-01-06 13:07:20 +08:00
if (settings.contains(WINDOW)) setGeometry(settings.value(WINDOW).toRect());
2019-09-12 02:27:57 +08:00
SetOutputFont(settings.value(FONT, ui->textOutput->font().toString()).toString());
2019-02-05 04:53:13 +08:00
TextThread::filterRepetition = settings.value(FILTER_REPETITION, TextThread::filterRepetition).toBool();
autoAttach = settings.value(AUTO_ATTACH, autoAttach).toBool();
autoAttachSavedOnly = settings.value(ATTACH_SAVED_ONLY, autoAttachSavedOnly).toBool();
2019-08-10 16:18:30 +08:00
showSystemProcesses = settings.value(SHOW_SYSTEM_PROCESSES, showSystemProcesses).toBool();
2019-01-01 04:06:47 +08:00
TextThread::flushDelay = settings.value(FLUSH_DELAY, TextThread::flushDelay).toInt();
TextThread::maxBufferSize = settings.value(MAX_BUFFER_SIZE, TextThread::maxBufferSize).toInt();
2019-08-12 23:10:33 +08:00
TextThread::maxHistorySize = settings.value(MAX_HISTORY_SIZE, TextThread::maxHistorySize).toInt();
2019-01-05 16:47:32 +08:00
Host::defaultCodepage = settings.value(DEFAULT_CODEPAGE, Host::defaultCodepage).toInt();
2018-08-22 10:43:30 +08:00
Host::Start(
2018-12-19 01:55:11 +08:00
[this](DWORD processId) { ProcessConnected(processId); },
[this](DWORD processId) { ProcessDisconnected(processId); },
2019-02-05 04:18:47 +08:00
[this](TextThread& thread) { ThreadAdded(thread); },
[this](TextThread& thread) { ThreadRemoved(thread); },
[this](TextThread& thread, std::wstring& output) { return SentenceReceived(thread, output); }
2018-08-22 10:43:30 +08:00
);
2019-02-09 13:30:38 +08:00
current = &Host::GetThread(Host::console);
2018-11-04 15:13:51 +08:00
Host::AddConsoleOutput(ABOUT);
2019-02-22 02:18:37 +08:00
AttachConsole(ATTACH_PARENT_PROCESS);
2019-06-04 05:58:30 +08:00
WriteConsoleW(GetStdHandle(STD_OUTPUT_HANDLE), CL_OPTIONS, wcslen(CL_OPTIONS), DUMMY, NULL);
auto processes = Util::GetAllProcesses();
2019-02-19 11:39:04 +08:00
int argc;
std::unique_ptr<LPWSTR[], Functor<LocalFree>> argv(CommandLineToArgvW(GetCommandLineW(), &argc));
for (int i = 0; i < argc; ++i)
if (std::wstring arg = argv[i]; arg[0] == L'/' || arg[0] == L'-')
2019-06-14 00:06:15 +08:00
if (arg[1] == L'p' || arg[1] == L'P')
2019-02-19 11:39:04 +08:00
if (DWORD processId = _wtoi(arg.substr(2).c_str())) Host::InjectProcess(processId);
else for (auto [processId, processName] : processes)
if (processName.value_or(L"").find(L"\\" + arg.substr(2)) != std::wstring::npos) Host::InjectProcess(processId);
2019-02-19 11:39:04 +08:00
std::thread([this]
{
for (; ; Sleep(10000))
{
std::unordered_set<std::wstring> attachTargets;
if (autoAttach)
for (auto process : QString(QTextFile(GAME_SAVE_FILE, QIODevice::ReadOnly).readAll()).split("\n", QString::SkipEmptyParts))
attachTargets.insert(S(process));
if (autoAttachSavedOnly)
for (auto process : QString(QTextFile(HOOK_SAVE_FILE, QIODevice::ReadOnly).readAll()).split("\n", QString::SkipEmptyParts))
attachTargets.insert(S(process.split(" , ")[0]));
if (!attachTargets.empty())
for (auto [processId, processName] : Util::GetAllProcesses())
if (processName && attachTargets.count(processName.value()) > 0 && alreadyAttached.count(processId) == 0) Host::InjectProcess(processId);
}
}).detach();
2018-07-22 06:40:16 +08:00
}
MainWindow::~MainWindow()
{
QSettings(CONFIG_FILE, QSettings::IniFormat).setValue(WINDOW, geometry());
2019-06-21 13:29:48 +08:00
CleanupExtensions();
2018-12-28 07:52:59 +08:00
SetErrorMode(SEM_NOGPFAULTERRORBOX);
2018-12-04 07:31:00 +08:00
ExitProcess(0);
2018-07-22 06:40:16 +08:00
}
2018-07-23 07:53:51 +08:00
void MainWindow::closeEvent(QCloseEvent*)
{
QCoreApplication::quit(); // Need to do this to kill any windows that might've been made by extensions
}
void MainWindow::ProcessConnected(DWORD processId)
2018-07-24 13:57:54 +08:00
{
alreadyAttached.insert(processId);
2019-02-03 21:49:58 +08:00
QString process = S(Util::GetModuleFilename(processId).value_or(L"???"));
QMetaObject::invokeMethod(this, [this, process, processId]
{
ui->processCombo->addItem(QString::number(processId, 16).toUpper() + ": " + QFileInfo(process).fileName());
2019-02-03 21:49:58 +08:00
});
if (process == "???") return;
2019-04-23 01:35:40 +08:00
// This does add (potentially tons of) duplicates to the file, but as long as I don't perform Ω(N^2) operations it shouldn't be an issue
2019-02-03 21:49:58 +08:00
QTextFile(GAME_SAVE_FILE, QIODevice::WriteOnly | QIODevice::Append).write((process + "\n").toUtf8());
2019-02-03 21:49:58 +08:00
QStringList allProcesses = QString(QTextFile(HOOK_SAVE_FILE, QIODevice::ReadOnly).readAll()).split("\n", QString::SkipEmptyParts);
auto hookList = std::find_if(allProcesses.rbegin(), allProcesses.rend(), [&](QString hookList) { return hookList.contains(process); });
if (hookList != allProcesses.rend())
for (auto hookInfo : hookList->split(" , "))
2019-02-05 04:18:47 +08:00
if (auto hp = Util::ParseCode(S(hookInfo))) Host::InsertHook(processId, hp.value());
else swscanf_s(S(hookInfo).c_str(), L"|%I64d:%I64d:%[^\n]", &savedThreadCtx, &savedThreadCtx2, savedThreadCode, (unsigned)std::size(savedThreadCode));
2018-07-24 13:57:54 +08:00
}
void MainWindow::ProcessDisconnected(DWORD processId)
2018-07-24 13:57:54 +08:00
{
2019-01-06 15:57:52 +08:00
QMetaObject::invokeMethod(this, [this, processId]
{
ui->processCombo->removeItem(ui->processCombo->findText(QString::number(processId, 16).toUpper() + ":", Qt::MatchStartsWith));
}, Qt::BlockingQueuedConnection);
2018-07-24 13:57:54 +08:00
}
2019-02-05 04:18:47 +08:00
void MainWindow::ThreadAdded(TextThread& thread)
2018-07-24 13:57:54 +08:00
{
2019-02-05 04:18:47 +08:00
std::wstring threadCode = Util::GenerateCode(thread.hp, thread.tp.processId);
bool savedMatch = savedThreadCtx == thread.tp.ctx && savedThreadCtx2 == thread.tp.ctx2 && savedThreadCode == threadCode;
if (savedMatch) savedThreadCtx = savedThreadCtx2 = savedThreadCode[0] = 0;
2019-09-12 02:27:57 +08:00
QMetaObject::invokeMethod(this, [this, savedMatch, ttString = TextThreadString(thread) + S(FormatString(L" (%s)", threadCode))]
2019-01-06 15:57:52 +08:00
{
ui->ttCombo->addItem(ttString);
2019-02-03 21:49:58 +08:00
if (savedMatch) ViewThread(ui->ttCombo->count() - 1);
2019-01-06 15:57:52 +08:00
});
}
2019-02-05 04:18:47 +08:00
void MainWindow::ThreadRemoved(TextThread& thread)
{
2019-09-12 02:27:57 +08:00
QMetaObject::invokeMethod(this, [this, ttString = TextThreadString(thread)]
2018-07-26 01:46:59 +08:00
{
int threadIndex = ui->ttCombo->findText(ttString, Qt::MatchStartsWith);
2019-02-03 21:49:58 +08:00
if (threadIndex == ui->ttCombo->currentIndex()) ViewThread(0);
ui->ttCombo->removeItem(threadIndex);
2019-01-06 15:57:52 +08:00
}, Qt::BlockingQueuedConnection);
2018-07-24 13:57:54 +08:00
}
2019-02-05 04:18:47 +08:00
bool MainWindow::SentenceReceived(TextThread& thread, std::wstring& sentence)
2018-10-08 12:26:43 +08:00
{
2019-09-05 00:29:48 +08:00
if (!DispatchSentenceToExtensions(sentence, GetSentenceInfo(thread).data())) return false;
2019-02-09 13:30:38 +08:00
sentence += L'\n';
2019-09-12 02:27:57 +08:00
if (current == &thread) QMetaObject::invokeMethod(this, [this, sentence = S(sentence)]
2018-10-08 12:26:43 +08:00
{
2019-09-12 02:27:57 +08:00
auto scrollbar = ui->textOutput->verticalScrollBar();
2019-12-27 08:18:02 +08:00
bool atBottom = scrollbar->value() + 3 > scrollbar->maximum() || (double)scrollbar->value() / scrollbar->maximum() > 0.975; // arbitrary
2019-09-12 02:27:57 +08:00
QTextCursor cursor(ui->textOutput->document());
cursor.movePosition(QTextCursor::End);
cursor.insertText(sentence);
if (atBottom) scrollbar->setValue(scrollbar->maximum());
2019-02-09 13:30:38 +08:00
});
return true;
2018-10-08 12:26:43 +08:00
}
void MainWindow::OutputContextMenu(QPoint point)
{
std::unique_ptr<QMenu> menu(ui->textOutput->createStandardContextMenu());
menu->addAction(FONT, [this] { if (QString font = QFontDialog::getFont(&ok, ui->textOutput->font(), this, FONT).toString(); ok) SetOutputFont(font); });
menu->exec(ui->textOutput->mapToGlobal(point));
}
2019-02-05 04:18:47 +08:00
QString MainWindow::TextThreadString(TextThread& thread)
2018-08-23 03:11:40 +08:00
{
return QString("%1:%2:%3:%4:%5: %6").arg(
2019-02-05 04:18:47 +08:00
QString::number(thread.handle, 16),
QString::number(thread.tp.processId, 16),
QString::number(thread.tp.addr, 16),
QString::number(thread.tp.ctx, 16),
QString::number(thread.tp.ctx2, 16)
).toUpper().arg(S(thread.name));
2018-08-23 03:11:40 +08:00
}
ThreadParam MainWindow::ParseTextThreadString(QString ttString)
2018-08-23 03:11:40 +08:00
{
QStringList threadParam = ttString.split(":");
return { threadParam[1].toUInt(nullptr, 16), threadParam[2].toULongLong(nullptr, 16), threadParam[3].toULongLong(nullptr, 16), threadParam[4].toULongLong(nullptr, 16) };
2018-08-23 03:11:40 +08:00
}
DWORD MainWindow::GetSelectedProcessId()
{
return ui->processCombo->currentText().split(":")[0].toULong(nullptr, 16);
2018-08-23 03:11:40 +08:00
}
2019-09-05 00:29:48 +08:00
std::array<InfoForExtension, 10> MainWindow::GetSentenceInfo(TextThread& thread)
{
void(*AddSentence)(MainWindow*, int64_t, const wchar_t*) = [](MainWindow* This, int64_t number, const wchar_t* sentence)
{
// pointer from Host::GetThread may not stay valid unless on main thread
QMetaObject::invokeMethod(This, [number, sentence = std::wstring(sentence)] { if (TextThread* thread = Host::GetThread(number)) thread->AddSentence(sentence); });
};
2018-12-29 01:14:56 +08:00
return
2019-02-09 13:30:38 +08:00
{ {
{ "current select", &thread == current },
2019-02-05 04:18:47 +08:00
{ "text number", thread.handle },
{ "process id", thread.tp.processId },
2019-02-09 13:30:38 +08:00
{ "hook address", (int64_t)thread.tp.addr },
2019-02-05 04:18:47 +08:00
{ "text handle", thread.handle },
2019-02-09 13:30:38 +08:00
{ "text name", (int64_t)thread.name.c_str() },
{ "this", (int64_t)this },
{ "void (*AddSentence)(void* this, int64_t number, const wchar_t* sentence)", (int64_t)AddSentence },
2019-02-09 13:30:38 +08:00
{ nullptr, 0 } // nullptr marks end of info array
} };
}
std::optional<std::wstring> MainWindow::UserSelectedProcess()
{
QStringList savedProcesses = QString::fromUtf8(QTextFile(GAME_SAVE_FILE, QIODevice::ReadOnly).readAll()).split("\n", QString::SkipEmptyParts);
std::reverse(savedProcesses.begin(), savedProcesses.end());
savedProcesses.removeDuplicates();
savedProcesses.insert(1, FROM_COMPUTER);
QString process = QInputDialog::getItem(this, SELECT_PROCESS, SELECT_PROCESS_INFO, savedProcesses, 0, true, &ok, Qt::WindowCloseButtonHint);
if (process == FROM_COMPUTER) process = QDir::toNativeSeparators(QFileDialog::getOpenFileName(this, SELECT_PROCESS, "C:\\", PROCESSES));
if (ok && process.contains('\\')) return S(process);
return {};
}
2018-12-19 01:14:54 +08:00
void MainWindow::AttachProcess()
2018-07-23 07:53:51 +08:00
{
QMultiHash<QString, DWORD> allProcesses;
for (auto [processId, processName] : Util::GetAllProcesses())
2019-08-10 16:18:30 +08:00
if (processName && (showSystemProcesses || processName->find(L":\\Windows\\") == std::wstring::npos))
allProcesses.insert(QFileInfo(S(processName.value())).fileName(), processId);
2018-09-10 10:37:48 +08:00
QStringList processList(allProcesses.uniqueKeys());
2018-08-21 02:30:50 +08:00
processList.sort(Qt::CaseInsensitive);
if (QString process = QInputDialog::getItem(this, SELECT_PROCESS, ATTACH_INFO, processList, 0, true, &ok, Qt::WindowCloseButtonHint); ok)
if (process.toInt(nullptr, 0)) Host::InjectProcess(process.toInt(nullptr, 0));
else for (auto processId : allProcesses.values(process)) Host::InjectProcess(processId);
2018-07-24 13:57:54 +08:00
}
void MainWindow::LaunchProcess()
{
std::wstring process;
if (auto selected = UserSelectedProcess()) process = selected.value();
else return;
2019-01-23 00:23:35 +08:00
std::wstring path = std::wstring(process).erase(process.rfind(L'\\'));
PROCESS_INFORMATION info = {};
2019-06-13 15:19:08 +08:00
if (!x64 && QMessageBox::question(this, SELECT_PROCESS, USE_JP_LOCALE) == QMessageBox::Yes)
2019-06-10 13:49:11 +08:00
{
2019-09-17 05:14:59 +08:00
if (HMODULE localeEmulator = LoadLibraryW(L"LoaderDll"))
{
// see https://github.com/xupefei/Locale-Emulator/blob/aa99dec3b25708e676c90acf5fed9beaac319160/LEProc/LoaderWrapper.cs#L252
struct
{
ULONG AnsiCodePage = SHIFT_JIS;
ULONG OemCodePage = SHIFT_JIS;
ULONG LocaleID = LANG_JAPANESE;
ULONG DefaultCharset = SHIFTJIS_CHARSET;
ULONG HookUiLanguageApi = FALSE;
WCHAR DefaultFaceName[LF_FACESIZE] = {};
TIME_ZONE_INFORMATION Timezone;
ULONG64 Unused = 0;
} LEB;
GetTimeZoneInformation(&LEB.Timezone);
((LONG(__stdcall*)(decltype(&LEB), LPCWSTR appName, LPWSTR commandLine, LPCWSTR currentDir, void*, void*, PROCESS_INFORMATION*, void*, void*, void*, void*))
2019-01-23 00:23:35 +08:00
GetProcAddress(localeEmulator, "LeCreateProcess"))(&LEB, process.c_str(), NULL, path.c_str(), NULL, NULL, &info, NULL, NULL, NULL, NULL);
}
2019-06-10 13:49:11 +08:00
}
if (info.hProcess == NULL)
{
STARTUPINFOW DUMMY = { sizeof(DUMMY) };
2019-01-23 00:23:35 +08:00
CreateProcessW(process.c_str(), NULL, nullptr, nullptr, FALSE, 0, nullptr, path.c_str(), &DUMMY, &info);
}
if (info.hProcess == NULL) return Host::AddConsoleOutput(LAUNCH_FAILED);
Host::InjectProcess(info.dwProcessId);
CloseHandle(info.hProcess);
CloseHandle(info.hThread);
}
2018-12-19 01:14:54 +08:00
void MainWindow::DetachProcess()
2018-07-24 13:57:54 +08:00
{
2019-06-22 10:55:29 +08:00
try { Host::DetachProcess(GetSelectedProcessId()); } catch (std::out_of_range) {}
2018-07-24 13:57:54 +08:00
}
void MainWindow::ForgetProcess()
{
std::optional<std::wstring> processName = Util::GetModuleFilename(GetSelectedProcessId());
if (!processName) processName = UserSelectedProcess();
DetachProcess();
if (!processName) return;
for (auto file : { GAME_SAVE_FILE, HOOK_SAVE_FILE })
{
QStringList lines = QString::fromUtf8(QTextFile(file, QIODevice::ReadOnly).readAll()).split("\n", QString::SkipEmptyParts);
lines.erase(std::remove_if(lines.begin(), lines.end(), [&](auto line) { return line.contains(S(processName.value())); }), lines.end());
QTextFile(file, QIODevice::WriteOnly | QIODevice::Truncate).write(lines.join("\n").append("\n").toUtf8());
}
}
2018-12-19 01:14:54 +08:00
void MainWindow::AddHook()
2018-07-26 12:48:18 +08:00
{
2019-08-06 23:51:17 +08:00
AddHook("");
}
void MainWindow::AddHook(QString hook)
{
if (QString hookCode = QInputDialog::getText(this, ADD_HOOK, CODE_INFODUMP, QLineEdit::Normal, hook, &ok, Qt::WindowCloseButtonHint); ok)
if (hookCode.startsWith("S") || hookCode.startsWith("/S")) FindHooks();
else if (auto hp = Util::ParseCode(S(hookCode))) try { Host::InsertHook(GetSelectedProcessId(), hp.value()); } catch (std::out_of_range) {}
else Host::AddConsoleOutput(INVALID_CODE);
2018-07-26 12:48:18 +08:00
}
2019-06-10 13:49:11 +08:00
void MainWindow::RemoveHooks()
{
DWORD processId = GetSelectedProcessId();
std::unordered_map<uint64_t, HookParam> hooks;
for (int i = 0; i < ui->ttCombo->count(); ++i)
{
ThreadParam tp = ParseTextThreadString(ui->ttCombo->itemText(i));
if (tp.processId == GetSelectedProcessId()) hooks[tp.addr] = Host::GetThread(tp).hp;
2019-06-10 13:49:11 +08:00
}
auto hookList = new QListWidget(this);
2019-06-11 10:47:58 +08:00
hookList->setWindowFlags(Qt::Window | Qt::WindowCloseButtonHint);
hookList->setAttribute(Qt::WA_DeleteOnClose);
2019-06-11 10:47:58 +08:00
hookList->setMinimumSize({ 300, 50 });
hookList->setWindowTitle(DOUBLE_CLICK_TO_REMOVE_HOOK);
for (auto [address, hp] : hooks)
2019-06-10 13:49:11 +08:00
new QListWidgetItem(QString(hp.name) + "@" + QString::number(address, 16), hookList);
connect(hookList, &QListWidget::itemDoubleClicked, [processId, hookList](QListWidgetItem* item)
{
try
{
Host::RemoveHook(processId, item->text().split("@")[1].toULongLong(nullptr, 16));
delete item;
}
catch (std::out_of_range) { hookList->close(); }
});
hookList->show();
}
2018-12-19 01:14:54 +08:00
void MainWindow::SaveHooks()
2018-07-27 13:42:21 +08:00
{
auto processName = Util::GetModuleFilename(GetSelectedProcessId());
if (!processName) return;
QHash<uint64_t, QString> hookCodes;
for (int i = 0; i < ui->ttCombo->count(); ++i)
2018-11-28 05:57:47 +08:00
{
ThreadParam tp = ParseTextThreadString(ui->ttCombo->itemText(i));
if (tp.processId == GetSelectedProcessId())
{
HookParam hp = Host::GetThread(tp).hp;
if (!(hp.type & HOOK_ENGINE)) hookCodes[tp.addr] = S(Util::GenerateCode(hp, tp.processId));
}
2018-11-28 05:57:47 +08:00
}
auto hookInfo = QStringList() << S(processName.value()) << hookCodes.values();
ThreadParam tp = current->tp;
if (tp.processId == GetSelectedProcessId()) hookInfo << QString("|%1:%2:%3").arg(tp.ctx).arg(tp.ctx2).arg(S(Util::GenerateCode(current->hp, tp.processId)));
QTextFile(HOOK_SAVE_FILE, QIODevice::WriteOnly | QIODevice::Append).write((hookInfo.join(" , ") + "\n").toUtf8());
2018-07-27 13:42:21 +08:00
}
2019-06-02 14:09:17 +08:00
void MainWindow::FindHooks()
{
QMessageBox::information(this, SEARCH_FOR_HOOKS, HOOK_SEARCH_UNSTABLE_WARNING);
DWORD processId = GetSelectedProcessId();
SearchParam sp = {};
sp.codepage = Host::defaultCodepage;
bool searchForText = false, customSettings = false;
std::wregex filter(L".");
QDialog dialog(this, Qt::WindowCloseButtonHint);
QFormLayout layout(&dialog);
QCheckBox cjkCheckbox(&dialog);
layout.addRow(SEARCH_CJK, &cjkCheckbox);
QDialogButtonBox confirm(QDialogButtonBox::Ok | QDialogButtonBox::Help | QDialogButtonBox::Retry, &dialog);
layout.addRow(&confirm);
confirm.button(QDialogButtonBox::Ok)->setText(START_HOOK_SEARCH);
confirm.button(QDialogButtonBox::Retry)->setText(SEARCH_FOR_TEXT);
confirm.button(QDialogButtonBox::Help)->setText(SETTINGS);
connect(&confirm, &QDialogButtonBox::clicked, [&](QAbstractButton* button)
{
if (button == confirm.button(QDialogButtonBox::Retry)) searchForText = true;
if (button == confirm.button(QDialogButtonBox::Help)) customSettings = true;
dialog.accept();
});
dialog.setWindowTitle(SEARCH_FOR_HOOKS);
2019-06-21 13:29:48 +08:00
if (!dialog.exec()) return;
if (searchForText)
{
QDialog dialog(this, Qt::WindowCloseButtonHint);
QFormLayout layout(&dialog);
QLineEdit textInput(&dialog);
layout.addRow(TEXT, &textInput);
QSpinBox codepageInput(&dialog);
codepageInput.setMaximum(INT_MAX);
codepageInput.setValue(sp.codepage);
layout.addRow(CODEPAGE, &codepageInput);
QDialogButtonBox confirm(QDialogButtonBox::Ok);
connect(&confirm, &QDialogButtonBox::accepted, &dialog, &QDialog::accept);
layout.addRow(&confirm);
if (!dialog.exec()) return;
wcsncpy_s(sp.text, S(textInput.text()).c_str(), PATTERN_SIZE - 1);
try { Host::FindHooks(GetSelectedProcessId(), sp); }
catch (std::out_of_range) {}
return;
}
if (customSettings)
2019-06-02 14:09:17 +08:00
{
QDialog dialog(this, Qt::WindowCloseButtonHint);
QFormLayout layout(&dialog);
QLineEdit patternInput(x64 ? "CC CC 48 89" : "55 8B EC", &dialog);
assert(QByteArray::fromHex(patternInput.text().toUtf8()) == QByteArray((const char*)sp.pattern, sp.length));
layout.addRow(SEARCH_PATTERN, &patternInput);
for (auto [value, label] : Array<std::tuple<int&, const char*>>{
{ sp.searchTime, SEARCH_DURATION },
{ sp.offset, PATTERN_OFFSET },
{ sp.maxRecords, MAX_HOOK_SEARCH_RECORDS },
{ sp.codepage, CODEPAGE },
})
2019-06-02 14:09:17 +08:00
{
auto spinBox = new QSpinBox(&dialog);
spinBox->setMaximum(INT_MAX);
spinBox->setValue(value);
layout.addRow(label, spinBox);
connect(spinBox, qOverload<int>(&QSpinBox::valueChanged), [&value](int newValue) { value = newValue; });
}
QLineEdit boundInput(QFileInfo(S(Util::GetModuleFilename(GetSelectedProcessId()).value_or(L""))).fileName(), &dialog);
layout.addRow(SEARCH_MODULE, &boundInput);
for (auto [value, label] : Array<std::tuple<uintptr_t&, const char*>>{
{ sp.minAddress, MIN_ADDRESS },
{ sp.maxAddress, MAX_ADDRESS },
{ sp.padding, STRING_OFFSET },
})
{
auto input = new QLineEdit(QString::number(value, 16), &dialog);
layout.addRow(label, input);
connect(input, &QLineEdit::textEdited, [&value](QString text) { if (uintptr_t newValue = text.toULongLong(&ok, 16); ok) value = newValue; });
2019-06-02 14:09:17 +08:00
}
QLineEdit filterInput(".", &dialog);
layout.addRow(HOOK_SEARCH_FILTER, &filterInput);
QPushButton startButton(START_HOOK_SEARCH, &dialog);
layout.addWidget(&startButton);
connect(&startButton, &QPushButton::clicked, &dialog, &QDialog::accept);
2019-06-21 13:29:48 +08:00
if (!dialog.exec()) return;
if (patternInput.text().contains('.'))
{
wcsncpy_s(sp.exportModule, S(patternInput.text()).c_str(), MAX_MODULE_SIZE - 1);
sp.length = 1;
}
else
{
QByteArray pattern = QByteArray::fromHex(patternInput.text().replace("??", QString::number(XX, 16)).toUtf8());
memcpy(sp.pattern, pattern.data(), sp.length = min(pattern.size(), PATTERN_SIZE));
}
wcsncpy_s(sp.boundaryModule, S(boundInput.text()).c_str(), MAX_MODULE_SIZE - 1);
2019-06-29 11:16:31 +08:00
try { filter = S(filterInput.text()); } catch (std::regex_error) {}
}
else
{
sp.length = 0; // use default
2019-06-29 11:16:31 +08:00
filter = std::wregex(cjkCheckbox.isChecked() ? L"[\\u3000-\\ua000]{4,}" : L"[\\u0020-\\u1000]{4,}");
}
2019-06-02 14:09:17 +08:00
auto hooks = std::make_shared<QStringList>();
try
{
2019-09-13 01:19:02 +08:00
Host::FindHooks(processId, sp, [=](HookParam hp, std::wstring text) { if (std::regex_search(text, filter)) *hooks << S(Util::GenerateCode(hp) + L" => " + text); });
2019-06-18 12:39:50 +08:00
}
catch (std::out_of_range) { return; }
std::thread([this, hooks]
{
DWORD64 cleanupTime = GetTickCount64() + 500'000;
for (int lastSize = 0; hooks->size() == 0 || hooks->size() != lastSize; Sleep(2000))
if (GetTickCount64() > cleanupTime) return;
else lastSize = hooks->size();
2019-09-13 01:19:02 +08:00
QString saveFileName;
QMetaObject::invokeMethod(this, [&]
{
auto hookList = new QListView(this);
hookList->setWindowFlags(Qt::Window | Qt::WindowCloseButtonHint);
hookList->setAttribute(Qt::WA_DeleteOnClose);
hookList->resize({ 750, 300 });
hookList->setWindowTitle(SEARCH_FOR_HOOKS);
if (hooks->size() > 5'000)
{
hookList->setUniformItemSizes(true); // they aren't actually uniform, but this improves performance
hooks->push_back(QString(2000, '-')); // dumb hack: with uniform item sizes, the last item is assumed to be the largest
}
hookList->setModel(new QStringListModel(*hooks, hookList));
2019-09-13 01:19:02 +08:00
connect(hookList, &QListView::clicked, [this](QModelIndex i) { AddHook(i.data().toString().split(" => ")[0]); });
hookList->show();
2019-07-20 22:03:28 +08:00
2019-09-13 01:19:02 +08:00
saveFileName = QFileDialog::getSaveFileName(this, SAVE_SEARCH_RESULTS, "./results.txt", TEXT_FILES);
}, Qt::BlockingQueuedConnection);
if (!saveFileName.isEmpty())
{
QTextFile saveFile(saveFileName, QIODevice::WriteOnly | QIODevice::Truncate);
for (auto hook = hooks->cbegin(); hook != hooks->cend(); ++hook) saveFile.write(hook->toUtf8().append('\n')); // QStringList::begin() makes a copy
}
hooks->clear();
}).detach();
2019-06-02 14:09:17 +08:00
}
2018-12-19 01:14:54 +08:00
void MainWindow::Settings()
2018-11-10 18:13:59 +08:00
{
QDialog dialog(this, Qt::WindowCloseButtonHint);
QSettings settings(CONFIG_FILE, QSettings::IniFormat, &dialog);
QFormLayout layout(&dialog);
QPushButton saveButton(SAVE_SETTINGS, &dialog);
for (auto [value, label] : Array<std::tuple<bool&, const char*>>{
{ TextThread::filterRepetition, FILTER_REPETITION },
{ autoAttach, AUTO_ATTACH },
{ autoAttachSavedOnly, ATTACH_SAVED_ONLY },
2019-08-10 16:18:30 +08:00
{ showSystemProcesses, SHOW_SYSTEM_PROCESSES },
})
{
auto checkBox = new QCheckBox(&dialog);
checkBox->setChecked(value);
layout.addRow(label, checkBox);
connect(&saveButton, &QPushButton::clicked, [checkBox, label, &settings, &value] { settings.setValue(label, value = checkBox->isChecked()); });
}
for (auto [value, label] : Array<std::tuple<int&, const char*>>{
{ TextThread::maxBufferSize, MAX_BUFFER_SIZE },
{ TextThread::flushDelay, FLUSH_DELAY },
2019-08-12 23:10:33 +08:00
{ TextThread::maxHistorySize, MAX_HISTORY_SIZE },
{ Host::defaultCodepage, DEFAULT_CODEPAGE },
})
{
auto spinBox = new QSpinBox(&dialog);
spinBox->setMaximum(INT_MAX);
spinBox->setValue(value);
layout.addRow(label, spinBox);
connect(&saveButton, &QPushButton::clicked, [spinBox, label, &settings, &value] { settings.setValue(label, value = spinBox->value()); });
}
layout.addWidget(&saveButton);
connect(&saveButton, &QPushButton::clicked, &dialog, &QDialog::accept);
dialog.setWindowTitle(SETTINGS);
dialog.exec();
2018-11-10 18:13:59 +08:00
}
2018-12-19 01:14:54 +08:00
void MainWindow::Extensions()
2018-07-24 13:57:54 +08:00
{
extenWindow->activateWindow();
extenWindow->showNormal();
2018-07-23 07:53:51 +08:00
}
2018-07-27 13:42:21 +08:00
2018-12-19 01:14:54 +08:00
void MainWindow::ViewThread(int index)
2018-07-27 13:42:21 +08:00
{
2019-02-03 21:49:58 +08:00
ui->ttCombo->setCurrentIndex(index);
2019-02-09 13:30:38 +08:00
ui->textOutput->setPlainText(S((current = &Host::GetThread(ParseTextThreadString(ui->ttCombo->itemText(index))))->storage->c_str()));
ui->textOutput->moveCursor(QTextCursor::End);
2018-07-27 13:42:21 +08:00
}
void MainWindow::SetOutputFont(QString fontString)
{
QFont font = ui->textOutput->font();
font.fromString(fontString);
font.setStyleStrategy(QFont::NoFontMerging);
ui->textOutput->setFont(font);
QSettings(CONFIG_FILE, QSettings::IniFormat).setValue(FONT, font.toString());
}