2019-04-22 13:35:40 -04:00
|
|
|
|
#include "mainwindow.h"
|
2018-07-21 15:40:16 -07:00
|
|
|
|
#include "ui_mainwindow.h"
|
2018-12-18 12:14:54 -05:00
|
|
|
|
#include "defs.h"
|
2018-11-01 14:07:42 -04:00
|
|
|
|
#include "extenwindow.h"
|
2018-12-01 15:55:32 -05:00
|
|
|
|
#include "host/util.h"
|
2019-02-18 22:39:04 -05:00
|
|
|
|
#include <shellapi.h>
|
2018-12-13 08:37:37 -05:00
|
|
|
|
#include <winhttp.h>
|
2018-12-21 14:18:43 -05:00
|
|
|
|
#include <QFormLayout>
|
2019-06-02 02:09:17 -04:00
|
|
|
|
#include <QLabel>
|
2018-12-14 22:26:49 -05:00
|
|
|
|
#include <QPushButton>
|
2019-02-04 15:53:13 -05:00
|
|
|
|
#include <QCheckBox>
|
2018-12-21 14:18:43 -05:00
|
|
|
|
#include <QSpinBox>
|
2019-06-10 01:49:11 -04:00
|
|
|
|
#include <QListWidget>
|
2019-06-16 15:28:59 -04:00
|
|
|
|
#include <QDialogButtonBox>
|
2019-01-13 03:40:00 -05:00
|
|
|
|
#include <QMessageBox>
|
2018-08-22 15:11:40 -04:00
|
|
|
|
#include <QInputDialog>
|
2019-01-11 17:14:49 -05:00
|
|
|
|
#include <QFileDialog>
|
2018-07-23 12:25:02 -07:00
|
|
|
|
|
2019-02-27 11:33:17 -05:00
|
|
|
|
extern const char* ATTACH;
|
|
|
|
|
extern const char* LAUNCH;
|
|
|
|
|
extern const char* DETACH;
|
|
|
|
|
extern const char* ADD_HOOK;
|
2019-06-10 01:49:11 -04:00
|
|
|
|
extern const char* REMOVE_HOOKS;
|
2019-02-27 11:33:17 -05:00
|
|
|
|
extern const char* SAVE_HOOKS;
|
2019-06-16 15:28:59 -04:00
|
|
|
|
extern const char* SEARCH_FOR_HOOKS;
|
2019-02-27 11:33:17 -05:00
|
|
|
|
extern const char* SETTINGS;
|
|
|
|
|
extern const char* EXTENSIONS;
|
|
|
|
|
extern const char* SELECT_PROCESS;
|
|
|
|
|
extern const char* ATTACH_INFO;
|
|
|
|
|
extern const char* SEARCH_GAME;
|
|
|
|
|
extern const char* PROCESSES;
|
|
|
|
|
extern const char* CODE_INFODUMP;
|
2019-06-09 06:19:54 -04:00
|
|
|
|
extern const char* HOOK_SEARCH_UNSTABLE_WARNING;
|
2019-06-16 15:28:59 -04:00
|
|
|
|
extern const char* SEARCH_CJK;
|
2019-06-02 02:09:17 -04:00
|
|
|
|
extern const char* SEARCH_PATTERN;
|
|
|
|
|
extern const char* SEARCH_DURATION;
|
|
|
|
|
extern const char* PATTERN_OFFSET;
|
|
|
|
|
extern const char* MIN_ADDRESS;
|
|
|
|
|
extern const char* MAX_ADDRESS;
|
2019-06-09 00:48:30 -04:00
|
|
|
|
extern const char* STRING_OFFSET;
|
2019-06-06 23:53:37 -04:00
|
|
|
|
extern const char* HOOK_SEARCH_FILTER;
|
2019-06-02 02:09:17 -04:00
|
|
|
|
extern const char* START_HOOK_SEARCH;
|
|
|
|
|
extern const char* SAVE_SEARCH_RESULTS;
|
|
|
|
|
extern const char* TEXT_FILES;
|
2019-06-10 22:47:58 -04:00
|
|
|
|
extern const char* DOUBLE_CLICK_TO_REMOVE_HOOK;
|
2019-02-27 11:33:17 -05:00
|
|
|
|
extern const char* SAVE_SETTINGS;
|
|
|
|
|
extern const char* USE_JP_LOCALE;
|
|
|
|
|
extern const char* FILTER_REPETITION;
|
|
|
|
|
extern const char* DEFAULT_CODEPAGE;
|
|
|
|
|
extern const char* FLUSH_DELAY;
|
|
|
|
|
extern const char* MAX_BUFFER_SIZE;
|
|
|
|
|
extern const wchar_t* ABOUT;
|
|
|
|
|
extern const wchar_t* CL_OPTIONS;
|
|
|
|
|
extern const wchar_t* UPDATE_AVAILABLE;
|
|
|
|
|
extern const wchar_t* LAUNCH_FAILED;
|
|
|
|
|
extern const wchar_t* INVALID_CODE;
|
|
|
|
|
|
2018-07-21 15:40:16 -07:00
|
|
|
|
MainWindow::MainWindow(QWidget *parent) :
|
2018-07-25 10:46:59 -07:00
|
|
|
|
QMainWindow(parent),
|
2018-11-01 10:38:14 -04:00
|
|
|
|
ui(new Ui::MainWindow),
|
2018-11-10 01:17:02 -05:00
|
|
|
|
extenWindow(new ExtenWindow(this))
|
2018-07-21 15:40:16 -07:00
|
|
|
|
{
|
2018-07-25 10:46:59 -07:00
|
|
|
|
ui->setupUi(this);
|
2019-06-16 15:28:59 -04:00
|
|
|
|
for (auto [text, slot] : Array<std::tuple<QString, void(MainWindow::*)()>>{
|
2018-12-18 12:14:54 -05:00
|
|
|
|
{ ATTACH, &MainWindow::AttachProcess },
|
2019-01-11 17:14:49 -05:00
|
|
|
|
{ LAUNCH, &MainWindow::LaunchProcess },
|
2018-12-18 12:14:54 -05:00
|
|
|
|
{ DETACH, &MainWindow::DetachProcess },
|
|
|
|
|
{ ADD_HOOK, &MainWindow::AddHook },
|
2019-06-10 01:49:11 -04:00
|
|
|
|
{ REMOVE_HOOKS, &MainWindow::RemoveHooks },
|
2018-12-18 12:14:54 -05:00
|
|
|
|
{ SAVE_HOOKS, &MainWindow::SaveHooks },
|
2019-06-16 15:28:59 -04:00
|
|
|
|
{ SEARCH_FOR_HOOKS, &MainWindow::FindHooks },
|
2018-12-18 12:14:54 -05:00
|
|
|
|
{ SETTINGS, &MainWindow::Settings },
|
|
|
|
|
{ EXTENSIONS, &MainWindow::Extensions }
|
2018-12-14 22:26:49 -05:00
|
|
|
|
})
|
|
|
|
|
{
|
2019-02-25 01:06:35 -05:00
|
|
|
|
auto button = new QPushButton(ui->processFrame);
|
2018-12-14 22:26:49 -05:00
|
|
|
|
connect(button, &QPushButton::clicked, this, slot);
|
|
|
|
|
button->setText(text);
|
2018-12-14 23:14:30 -05:00
|
|
|
|
ui->processLayout->addWidget(button);
|
2018-12-14 22:26:49 -05:00
|
|
|
|
}
|
2018-12-14 23:14:30 -05:00
|
|
|
|
ui->processLayout->addItem(new QSpacerItem(0, 0, QSizePolicy::Minimum, QSizePolicy::Expanding));
|
2018-08-21 22:43:30 -04:00
|
|
|
|
|
2019-01-10 21:47:16 -05:00
|
|
|
|
connect(ui->ttCombo, qOverload<int>(&QComboBox::activated), this, &MainWindow::ViewThread);
|
2019-01-22 14:57:13 -05:00
|
|
|
|
connect(ui->textOutput, &QPlainTextEdit::selectionChanged, [this] { if (!(QApplication::mouseButtons() & Qt::LeftButton)) ui->textOutput->copy(); });
|
2018-12-18 12:14:54 -05:00
|
|
|
|
|
|
|
|
|
QSettings settings(CONFIG_FILE, QSettings::IniFormat);
|
2019-01-06 00:07:20 -05:00
|
|
|
|
if (settings.contains(WINDOW)) setGeometry(settings.value(WINDOW).toRect());
|
2019-02-04 15:53:13 -05:00
|
|
|
|
TextThread::filterRepetition = settings.value(FILTER_REPETITION, TextThread::filterRepetition).toBool();
|
2018-12-31 15:06:47 -05:00
|
|
|
|
TextThread::flushDelay = settings.value(FLUSH_DELAY, TextThread::flushDelay).toInt();
|
|
|
|
|
TextThread::maxBufferSize = settings.value(MAX_BUFFER_SIZE, TextThread::maxBufferSize).toInt();
|
2019-01-05 03:47:32 -05:00
|
|
|
|
Host::defaultCodepage = settings.value(DEFAULT_CODEPAGE, Host::defaultCodepage).toInt();
|
2018-10-28 02:27:24 -04:00
|
|
|
|
|
2018-08-21 22:43:30 -04:00
|
|
|
|
Host::Start(
|
2018-12-18 12:55:11 -05:00
|
|
|
|
[this](DWORD processId) { ProcessConnected(processId); },
|
|
|
|
|
[this](DWORD processId) { ProcessDisconnected(processId); },
|
2019-02-04 15:18:47 -05:00
|
|
|
|
[this](TextThread& thread) { ThreadAdded(thread); },
|
|
|
|
|
[this](TextThread& thread) { ThreadRemoved(thread); },
|
|
|
|
|
[this](TextThread& thread, std::wstring& output) { return SentenceReceived(thread, output); }
|
2018-08-21 22:43:30 -04:00
|
|
|
|
);
|
2019-02-09 00:30:38 -05:00
|
|
|
|
current = &Host::GetThread(Host::console);
|
2018-11-04 02:13:51 -05:00
|
|
|
|
Host::AddConsoleOutput(ABOUT);
|
2018-12-13 08:37:37 -05:00
|
|
|
|
|
2019-02-21 13:18:37 -05:00
|
|
|
|
AttachConsole(ATTACH_PARENT_PROCESS);
|
2019-06-03 17:58:30 -04:00
|
|
|
|
WriteConsoleW(GetStdHandle(STD_OUTPUT_HANDLE), CL_OPTIONS, wcslen(CL_OPTIONS), DUMMY, NULL);
|
2019-02-18 22:39:04 -05:00
|
|
|
|
std::vector<DWORD> processIds = Util::GetAllProcessIds();
|
|
|
|
|
std::vector<std::wstring> processNames;
|
|
|
|
|
for (auto processId : processIds) processNames.emplace_back(Util::GetModuleFilename(processId).value_or(L""));
|
|
|
|
|
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-13 12:06:15 -04:00
|
|
|
|
if (arg[1] == L'p' || arg[1] == L'P')
|
2019-02-18 22:39:04 -05:00
|
|
|
|
if (DWORD processId = _wtoi(arg.substr(2).c_str())) Host::InjectProcess(processId);
|
|
|
|
|
else for (int i = 0; i < processIds.size(); ++i)
|
|
|
|
|
if (processNames[i].find(L"\\" + arg.substr(2)) != std::wstring::npos) Host::InjectProcess(processIds[i]);
|
|
|
|
|
|
2018-12-13 08:37:37 -05:00
|
|
|
|
std::thread([]
|
|
|
|
|
{
|
2019-01-09 22:35:01 -05:00
|
|
|
|
using InternetHandle = AutoHandle<Functor<WinHttpCloseHandle>>;
|
2018-12-13 08:37:37 -05:00
|
|
|
|
// Queries GitHub releases API https://developer.github.com/v3/repos/releases/ and checks the last release tag to check if it's the same
|
2019-01-09 22:35:01 -05:00
|
|
|
|
if (InternetHandle internet = WinHttpOpen(L"Mozilla/5.0 Textractor", WINHTTP_ACCESS_TYPE_DEFAULT_PROXY, NULL, NULL, 0))
|
|
|
|
|
if (InternetHandle connection = WinHttpConnect(internet, L"api.github.com", INTERNET_DEFAULT_HTTPS_PORT, 0))
|
|
|
|
|
if (InternetHandle request = WinHttpOpenRequest(connection, L"GET", L"/repos/Artikash/Textractor/releases", NULL, NULL, NULL, WINHTTP_FLAG_SECURE))
|
2018-12-13 08:37:37 -05:00
|
|
|
|
if (WinHttpSendRequest(request, NULL, 0, NULL, 0, 0, NULL))
|
|
|
|
|
{
|
2019-06-03 17:58:30 -04:00
|
|
|
|
char buffer[1000] = {};
|
2018-12-13 08:37:37 -05:00
|
|
|
|
WinHttpReceiveResponse(request, NULL);
|
2019-06-03 17:58:30 -04:00
|
|
|
|
WinHttpReadData(request, buffer, 1000, DUMMY);
|
2019-05-31 14:48:07 -04:00
|
|
|
|
if (abs(strstr(buffer, "/tag/") - strstr(buffer, VERSION)) > 10) MESSAGE(UPDATE_AVAILABLE);
|
2018-12-13 08:37:37 -05:00
|
|
|
|
}
|
|
|
|
|
}).detach();
|
2018-07-21 15:40:16 -07:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
MainWindow::~MainWindow()
|
|
|
|
|
{
|
2019-02-16 22:51:10 -05:00
|
|
|
|
QSettings(CONFIG_FILE, QSettings::IniFormat).setValue(WINDOW, geometry());
|
2019-06-21 01:29:48 -04:00
|
|
|
|
CleanupExtensions();
|
2018-12-27 18:52:59 -05:00
|
|
|
|
SetErrorMode(SEM_NOGPFAULTERRORBOX);
|
2018-12-03 18:31:00 -05:00
|
|
|
|
ExitProcess(0);
|
2018-07-21 15:40:16 -07:00
|
|
|
|
}
|
2018-07-22 16:53:51 -07:00
|
|
|
|
|
2018-11-01 18:47:46 -04:00
|
|
|
|
void MainWindow::closeEvent(QCloseEvent*)
|
|
|
|
|
{
|
|
|
|
|
QCoreApplication::quit(); // Need to do this to kill any windows that might've been made by extensions
|
|
|
|
|
}
|
|
|
|
|
|
2018-11-27 15:54:04 -05:00
|
|
|
|
void MainWindow::ProcessConnected(DWORD processId)
|
2018-07-23 22:57:54 -07:00
|
|
|
|
{
|
2018-11-27 15:54:04 -05:00
|
|
|
|
if (processId == 0) return;
|
2019-02-03 08:49:58 -05:00
|
|
|
|
QString process = S(Util::GetModuleFilename(processId).value_or(L"???"));
|
|
|
|
|
QMetaObject::invokeMethod(this, [this, process, processId]
|
2018-11-27 15:54:04 -05:00
|
|
|
|
{
|
2018-12-14 23:14:30 -05:00
|
|
|
|
ui->processCombo->addItem(QString::number(processId, 16).toUpper() + ": " + QFileInfo(process).fileName());
|
2019-02-03 08:49:58 -05:00
|
|
|
|
});
|
|
|
|
|
if (process == "???") return;
|
2019-01-11 17:14:49 -05:00
|
|
|
|
|
2019-04-22 13:35:40 -04: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 08:49:58 -05:00
|
|
|
|
QTextFile(GAME_SAVE_FILE, QIODevice::WriteOnly | QIODevice::Append).write((process + "\n").toUtf8());
|
2018-12-01 15:55:32 -05:00
|
|
|
|
|
2019-02-03 08:49:58 -05:00
|
|
|
|
QStringList allProcesses = QString(QTextFile(HOOK_SAVE_FILE, QIODevice::ReadOnly).readAll()).split("\n", QString::SkipEmptyParts);
|
|
|
|
|
// Can't use QFileInfo::absoluteFilePath since hook save file has '\\' as path separator
|
|
|
|
|
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-04 15:18:47 -05:00
|
|
|
|
if (auto hp = Util::ParseCode(S(hookInfo))) Host::InsertHook(processId, hp.value());
|
2019-02-27 11:33:17 -05:00
|
|
|
|
else swscanf_s(S(hookInfo).c_str(), L"|%I64d:%I64d:%[^\n]", &savedThreadCtx.first, &savedThreadCtx.second, savedThreadCode, (unsigned)std::size(savedThreadCode));
|
2018-07-23 22:57:54 -07:00
|
|
|
|
}
|
|
|
|
|
|
2018-11-27 15:54:04 -05:00
|
|
|
|
void MainWindow::ProcessDisconnected(DWORD processId)
|
2018-07-23 22:57:54 -07:00
|
|
|
|
{
|
2019-01-06 02:57:52 -05:00
|
|
|
|
QMetaObject::invokeMethod(this, [this, processId]
|
|
|
|
|
{
|
|
|
|
|
ui->processCombo->removeItem(ui->processCombo->findText(QString::number(processId, 16).toUpper() + ":", Qt::MatchStartsWith));
|
|
|
|
|
}, Qt::BlockingQueuedConnection);
|
2018-07-23 22:57:54 -07:00
|
|
|
|
}
|
|
|
|
|
|
2019-02-04 15:18:47 -05:00
|
|
|
|
void MainWindow::ThreadAdded(TextThread& thread)
|
2018-07-23 22:57:54 -07:00
|
|
|
|
{
|
2019-02-04 15:18:47 -05:00
|
|
|
|
std::wstring threadCode = Util::GenerateCode(thread.hp, thread.tp.processId);
|
|
|
|
|
QString ttString = TextThreadString(thread) + S(thread.name) + " (" + S(threadCode) + ")";
|
2019-06-21 01:29:48 -04:00
|
|
|
|
bool savedMatch = savedThreadCtx == std::pair(thread.tp.ctx, thread.tp.ctx2) && savedThreadCode == threadCode;
|
2019-02-03 08:49:58 -05:00
|
|
|
|
if (savedMatch) savedThreadCtx.first = savedThreadCtx.second = savedThreadCode[0] = 0;
|
|
|
|
|
QMetaObject::invokeMethod(this, [this, ttString, savedMatch]
|
2019-01-06 02:57:52 -05:00
|
|
|
|
{
|
|
|
|
|
ui->ttCombo->addItem(ttString);
|
2019-02-03 08:49:58 -05:00
|
|
|
|
if (savedMatch) ViewThread(ui->ttCombo->count() - 1);
|
2019-01-06 02:57:52 -05:00
|
|
|
|
});
|
2018-07-24 10:39:02 -07:00
|
|
|
|
}
|
|
|
|
|
|
2019-02-04 15:18:47 -05:00
|
|
|
|
void MainWindow::ThreadRemoved(TextThread& thread)
|
2018-07-24 10:39:02 -07:00
|
|
|
|
{
|
2018-11-27 15:54:04 -05:00
|
|
|
|
QString ttString = TextThreadString(thread);
|
2018-12-18 12:55:11 -05:00
|
|
|
|
QMetaObject::invokeMethod(this, [this, ttString]
|
2018-07-25 10:46:59 -07:00
|
|
|
|
{
|
2018-12-14 23:14:30 -05:00
|
|
|
|
int threadIndex = ui->ttCombo->findText(ttString, Qt::MatchStartsWith);
|
2019-02-03 08:49:58 -05:00
|
|
|
|
if (threadIndex == ui->ttCombo->currentIndex()) ViewThread(0);
|
2018-12-14 23:14:30 -05:00
|
|
|
|
ui->ttCombo->removeItem(threadIndex);
|
2019-01-06 02:57:52 -05:00
|
|
|
|
}, Qt::BlockingQueuedConnection);
|
2018-07-23 22:57:54 -07:00
|
|
|
|
}
|
|
|
|
|
|
2019-02-04 15:18:47 -05:00
|
|
|
|
bool MainWindow::SentenceReceived(TextThread& thread, std::wstring& sentence)
|
2018-10-08 00:26:43 -04:00
|
|
|
|
{
|
2019-02-09 00:30:38 -05:00
|
|
|
|
if (!DispatchSentenceToExtensions(sentence, GetMiscInfo(thread).data())) return false;
|
|
|
|
|
sentence += L'\n';
|
|
|
|
|
if (current == &thread) QMetaObject::invokeMethod(this, [this, sentence]
|
2018-10-08 00:26:43 -04:00
|
|
|
|
{
|
2019-02-09 00:30:38 -05:00
|
|
|
|
ui->textOutput->moveCursor(QTextCursor::End);
|
|
|
|
|
ui->textOutput->insertPlainText(S(sentence));
|
|
|
|
|
ui->textOutput->moveCursor(QTextCursor::End);
|
|
|
|
|
});
|
|
|
|
|
return true;
|
2018-10-08 00:26:43 -04:00
|
|
|
|
}
|
|
|
|
|
|
2019-02-04 15:18:47 -05:00
|
|
|
|
QString MainWindow::TextThreadString(TextThread& thread)
|
2018-08-22 15:11:40 -04:00
|
|
|
|
{
|
2018-09-21 22:25:37 -04:00
|
|
|
|
return QString("%1:%2:%3:%4:%5: ").arg(
|
2019-02-04 15:18:47 -05: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)
|
2018-08-22 15:11:40 -04:00
|
|
|
|
).toUpper();
|
|
|
|
|
}
|
|
|
|
|
|
2018-11-27 15:54:04 -05:00
|
|
|
|
ThreadParam MainWindow::ParseTextThreadString(QString ttString)
|
2018-08-22 15:11:40 -04:00
|
|
|
|
{
|
2018-11-27 15:54:04 -05:00
|
|
|
|
QStringList threadParam = ttString.split(":");
|
2018-09-22 17:13:06 -04:00
|
|
|
|
return { threadParam[1].toUInt(nullptr, 16), threadParam[2].toULongLong(nullptr, 16), threadParam[3].toULongLong(nullptr, 16), threadParam[4].toULongLong(nullptr, 16) };
|
2018-08-22 15:11:40 -04:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
DWORD MainWindow::GetSelectedProcessId()
|
|
|
|
|
{
|
2018-12-14 23:14:30 -05:00
|
|
|
|
return ui->processCombo->currentText().split(":")[0].toULong(nullptr, 16);
|
2018-08-22 15:11:40 -04:00
|
|
|
|
}
|
|
|
|
|
|
2019-02-09 00:30:38 -05:00
|
|
|
|
std::array<InfoForExtension, 10> MainWindow::GetMiscInfo(TextThread& thread)
|
2018-08-04 03:16:14 -04:00
|
|
|
|
{
|
2019-06-16 22:57:41 -04:00
|
|
|
|
void(*AddSentence)(MainWindow*, int64_t, const wchar_t*) = [](MainWindow* This, int64_t number, const wchar_t* sentence)
|
|
|
|
|
{
|
2019-06-21 01:29:48 -04:00
|
|
|
|
std::wstring copy = sentence;
|
2019-06-16 22:57:41 -04:00
|
|
|
|
// pointer from Host::GetThread may not stay valid unless on main thread
|
2019-06-21 01:29:48 -04:00
|
|
|
|
QMetaObject::invokeMethod(This, [=]() mutable { if (TextThread* thread = Host::GetThread(number)) thread->AddSentence(std::move(copy)); });
|
2019-06-16 22:57:41 -04:00
|
|
|
|
};
|
|
|
|
|
|
2018-12-28 12:14:56 -05:00
|
|
|
|
return
|
2019-02-09 00:30:38 -05:00
|
|
|
|
{ {
|
|
|
|
|
{ "current select", &thread == current },
|
2019-02-04 15:18:47 -05:00
|
|
|
|
{ "text number", thread.handle },
|
|
|
|
|
{ "process id", thread.tp.processId },
|
2019-02-09 00:30:38 -05:00
|
|
|
|
{ "hook address", (int64_t)thread.tp.addr },
|
2019-02-04 15:18:47 -05:00
|
|
|
|
{ "text handle", thread.handle },
|
2019-02-09 00:30:38 -05:00
|
|
|
|
{ "text name", (int64_t)thread.name.c_str() },
|
2019-06-16 22:57:41 -04:00
|
|
|
|
{ "this", (int64_t)this },
|
|
|
|
|
{ "void (*AddSentence)(void* this, int64_t number, const wchar_t* sentence)", (int64_t)AddSentence },
|
2019-02-09 00:30:38 -05:00
|
|
|
|
{ nullptr, 0 } // nullptr marks end of info array
|
|
|
|
|
} };
|
2018-08-04 03:16:14 -04:00
|
|
|
|
}
|
|
|
|
|
|
2018-12-18 12:14:54 -05:00
|
|
|
|
void MainWindow::AttachProcess()
|
2018-07-22 16:53:51 -07:00
|
|
|
|
{
|
2018-12-13 08:37:37 -05:00
|
|
|
|
QMultiHash<QString, DWORD> allProcesses;
|
2019-02-18 22:39:04 -05:00
|
|
|
|
for (auto processId : Util::GetAllProcessIds())
|
|
|
|
|
if (auto processName = Util::GetModuleFilename(processId)) allProcesses.insert(QFileInfo(S(processName.value())).fileName(), processId);
|
2018-12-13 08:37:37 -05:00
|
|
|
|
|
2018-09-09 22:37:48 -04:00
|
|
|
|
QStringList processList(allProcesses.uniqueKeys());
|
2018-08-20 14:30:50 -04:00
|
|
|
|
processList.sort(Qt::CaseInsensitive);
|
2019-01-13 03:40:00 -05:00
|
|
|
|
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-23 22:57:54 -07:00
|
|
|
|
}
|
|
|
|
|
|
2019-01-11 17:14:49 -05:00
|
|
|
|
void MainWindow::LaunchProcess()
|
|
|
|
|
{
|
|
|
|
|
QStringList savedProcesses = QString::fromUtf8(QTextFile(GAME_SAVE_FILE, QIODevice::ReadOnly).readAll()).split("\n", QString::SkipEmptyParts);
|
2019-02-02 16:50:28 -05:00
|
|
|
|
std::reverse(savedProcesses.begin(), savedProcesses.end());
|
2019-02-13 16:22:49 -05:00
|
|
|
|
savedProcesses.removeDuplicates();
|
2019-01-11 17:14:49 -05:00
|
|
|
|
savedProcesses.push_back(SEARCH_GAME);
|
|
|
|
|
std::wstring process = S(QInputDialog::getItem(this, SELECT_PROCESS, "", savedProcesses, 0, true, &ok, Qt::WindowCloseButtonHint));
|
|
|
|
|
if (!ok) return;
|
|
|
|
|
if (S(process) == SEARCH_GAME) process = S(QDir::toNativeSeparators(QFileDialog::getOpenFileName(this, SELECT_PROCESS, "C:\\", PROCESSES)));
|
2019-01-22 11:23:35 -05:00
|
|
|
|
if (process.find(L'\\') == std::wstring::npos) return;
|
|
|
|
|
std::wstring path = std::wstring(process).erase(process.rfind(L'\\'));
|
2019-01-13 03:40:00 -05:00
|
|
|
|
|
2019-01-11 17:14:49 -05:00
|
|
|
|
PROCESS_INFORMATION info = {};
|
2019-06-13 03:19:08 -04:00
|
|
|
|
if (!x64 && QMessageBox::question(this, SELECT_PROCESS, USE_JP_LOCALE) == QMessageBox::Yes)
|
2019-06-10 01:49:11 -04:00
|
|
|
|
{
|
2019-01-13 03:40:00 -05:00
|
|
|
|
if (HMODULE localeEmulator = LoadLibraryOnce(L"LoaderDll"))
|
2019-01-11 17:14:49 -05:00
|
|
|
|
{
|
2019-01-13 03:40:00 -05:00
|
|
|
|
// 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-22 11:23:35 -05:00
|
|
|
|
GetProcAddress(localeEmulator, "LeCreateProcess"))(&LEB, process.c_str(), NULL, path.c_str(), NULL, NULL, &info, NULL, NULL, NULL, NULL);
|
2019-01-13 03:40:00 -05:00
|
|
|
|
}
|
2019-06-10 01:49:11 -04:00
|
|
|
|
}
|
2019-01-11 17:14:49 -05:00
|
|
|
|
if (info.hProcess == NULL)
|
|
|
|
|
{
|
|
|
|
|
STARTUPINFOW DUMMY = { sizeof(DUMMY) };
|
2019-01-22 11:23:35 -05:00
|
|
|
|
CreateProcessW(process.c_str(), NULL, nullptr, nullptr, FALSE, 0, nullptr, path.c_str(), &DUMMY, &info);
|
2019-01-11 17:14:49 -05:00
|
|
|
|
}
|
|
|
|
|
if (info.hProcess == NULL) return Host::AddConsoleOutput(LAUNCH_FAILED);
|
|
|
|
|
Host::InjectProcess(info.dwProcessId);
|
|
|
|
|
CloseHandle(info.hProcess);
|
|
|
|
|
CloseHandle(info.hThread);
|
|
|
|
|
}
|
|
|
|
|
|
2018-12-18 12:14:54 -05:00
|
|
|
|
void MainWindow::DetachProcess()
|
2018-07-23 22:57:54 -07:00
|
|
|
|
{
|
2018-08-21 22:43:30 -04:00
|
|
|
|
Host::DetachProcess(GetSelectedProcessId());
|
2018-07-23 22:57:54 -07:00
|
|
|
|
}
|
|
|
|
|
|
2018-12-18 12:14:54 -05:00
|
|
|
|
void MainWindow::AddHook()
|
2018-07-25 21:48:18 -07:00
|
|
|
|
{
|
2019-01-13 03:40:00 -05:00
|
|
|
|
if (QString hookCode = QInputDialog::getText(this, ADD_HOOK, CODE_INFODUMP, QLineEdit::Normal, "", &ok, Qt::WindowCloseButtonHint); ok)
|
2019-06-13 19:29:30 -04:00
|
|
|
|
if (auto hp = Util::ParseCode(S(hookCode))) try { Host::InsertHook(GetSelectedProcessId(), hp.value()); } catch (std::out_of_range) {}
|
2019-01-13 03:40:00 -05:00
|
|
|
|
else Host::AddConsoleOutput(INVALID_CODE);
|
2018-07-25 21:48:18 -07:00
|
|
|
|
}
|
|
|
|
|
|
2019-06-10 01:49:11 -04: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));
|
2019-06-16 22:57:41 -04:00
|
|
|
|
if (tp.processId == GetSelectedProcessId()) hooks[tp.addr] = Host::GetThread(tp).hp;
|
2019-06-10 01:49:11 -04:00
|
|
|
|
}
|
|
|
|
|
auto hookList = new QListWidget(this);
|
2019-06-10 22:47:58 -04:00
|
|
|
|
hookList->setWindowFlags(Qt::Window | Qt::WindowCloseButtonHint);
|
2019-06-16 15:28:59 -04:00
|
|
|
|
hookList->setAttribute(Qt::WA_DeleteOnClose);
|
2019-06-10 22:47:58 -04:00
|
|
|
|
hookList->setMinimumSize({ 300, 50 });
|
|
|
|
|
hookList->setWindowTitle(DOUBLE_CLICK_TO_REMOVE_HOOK);
|
2019-06-16 15:28:59 -04:00
|
|
|
|
for (auto [address, hp] : hooks)
|
2019-06-10 01:49:11 -04: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-18 12:14:54 -05:00
|
|
|
|
void MainWindow::SaveHooks()
|
2018-07-26 22:42:21 -07:00
|
|
|
|
{
|
2019-06-16 22:57:41 -04: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-27 16:57:47 -05:00
|
|
|
|
{
|
2019-06-16 22:57:41 -04:00
|
|
|
|
ThreadParam tp = ParseTextThreadString(ui->ttCombo->itemText(i));
|
|
|
|
|
if (tp.processId == GetSelectedProcessId())
|
2018-12-14 23:14:30 -05:00
|
|
|
|
{
|
2019-06-16 22:57:41 -04:00
|
|
|
|
HookParam hp = Host::GetThread(tp).hp;
|
|
|
|
|
if (!(hp.type & HOOK_ENGINE)) hookCodes[tp.addr] = S(Util::GenerateCode(hp, tp.processId));
|
2018-12-14 23:14:30 -05:00
|
|
|
|
}
|
2018-11-27 16:57:47 -05:00
|
|
|
|
}
|
2019-06-16 22:57:41 -04:00
|
|
|
|
auto hookInfo = QStringList() << S(processName.value()) << hookCodes.values();
|
|
|
|
|
ThreadParam tp = current.load()->tp;
|
|
|
|
|
if (tp.processId == GetSelectedProcessId()) hookInfo << QString("|%1:%2:%3").arg(tp.ctx).arg(tp.ctx2).arg(S(Util::GenerateCode(Host::GetThread(tp).hp, tp.processId)));
|
|
|
|
|
QTextFile(HOOK_SAVE_FILE, QIODevice::WriteOnly | QIODevice::Append).write((hookInfo.join(" , ") + "\n").toUtf8());
|
2018-07-26 22:42:21 -07:00
|
|
|
|
}
|
|
|
|
|
|
2019-06-02 02:09:17 -04:00
|
|
|
|
void MainWindow::FindHooks()
|
|
|
|
|
{
|
2019-06-16 15:28:59 -04:00
|
|
|
|
QMessageBox::information(this, SEARCH_FOR_HOOKS, HOOK_SEARCH_UNSTABLE_WARNING);
|
|
|
|
|
|
|
|
|
|
DWORD processId = GetSelectedProcessId();
|
|
|
|
|
SearchParam sp = {};
|
|
|
|
|
bool 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, &dialog);
|
|
|
|
|
layout.addRow(&confirm);
|
|
|
|
|
confirm.button(QDialogButtonBox::Ok)->setText(START_HOOK_SEARCH);
|
|
|
|
|
confirm.button(QDialogButtonBox::Help)->setText(SETTINGS);
|
|
|
|
|
connect(&confirm, &QDialogButtonBox::helpRequested, [&customSettings] { customSettings = true; });
|
|
|
|
|
connect(&confirm, &QDialogButtonBox::accepted, &dialog, &QDialog::accept);
|
|
|
|
|
connect(&confirm, &QDialogButtonBox::helpRequested, &dialog, &QDialog::accept);
|
|
|
|
|
dialog.setWindowTitle(SEARCH_FOR_HOOKS);
|
2019-06-21 01:29:48 -04:00
|
|
|
|
if (!dialog.exec()) return;
|
2019-06-16 15:28:59 -04:00
|
|
|
|
|
|
|
|
|
if (customSettings)
|
2019-06-02 02:09:17 -04:00
|
|
|
|
{
|
2019-06-16 15:28:59 -04:00
|
|
|
|
QDialog dialog(this, Qt::WindowCloseButtonHint);
|
|
|
|
|
QFormLayout layout(&dialog);
|
|
|
|
|
QLineEdit patternInput(x64 ? "CC CC 48 89" : "CC CC 55 8B EC", &dialog);
|
|
|
|
|
layout.addRow(SEARCH_PATTERN, &patternInput);
|
|
|
|
|
for (auto [value, label] : Array<std::tuple<int&, const char*>>{
|
|
|
|
|
{ sp.searchTime = 20000, SEARCH_DURATION },
|
|
|
|
|
{ sp.offset = 2, PATTERN_OFFSET },
|
|
|
|
|
})
|
2019-06-02 02:09:17 -04:00
|
|
|
|
{
|
2019-06-16 15:28:59 -04: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; });
|
|
|
|
|
}
|
|
|
|
|
for (auto [value, label] : Array<std::tuple<uintptr_t&, const char*>>{
|
|
|
|
|
{ sp.minAddress = 0, MIN_ADDRESS },
|
|
|
|
|
{ sp.maxAddress = -1ULL, MAX_ADDRESS },
|
|
|
|
|
{ sp.padding = 0, STRING_OFFSET }
|
|
|
|
|
})
|
|
|
|
|
{
|
|
|
|
|
auto input = new QLineEdit(QString::number(value, 16), &dialog);
|
|
|
|
|
layout.addRow(label, input);
|
|
|
|
|
connect(input, &QLineEdit::textEdited, [&value](QString input)
|
2019-06-02 02:09:17 -04:00
|
|
|
|
{
|
2019-06-16 15:28:59 -04:00
|
|
|
|
bool ok;
|
|
|
|
|
if (uintptr_t newValue = input.toULongLong(&ok, 16); ok) value = newValue;
|
2019-06-02 02:09:17 -04:00
|
|
|
|
});
|
|
|
|
|
}
|
2019-06-16 15:28:59 -04: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 01:29:48 -04:00
|
|
|
|
if (!dialog.exec()) return;
|
2019-06-16 15:28:59 -04:00
|
|
|
|
QByteArray pattern = QByteArray::fromHex(patternInput.text().replace("??", QString::number(XX, 16)).toUtf8());
|
|
|
|
|
memcpy(sp.pattern, pattern.data(), sp.length = min(pattern.size(), 25));
|
|
|
|
|
try { filter = std::wregex(S(filterInput.text())); } catch (std::regex_error) {};
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
// sp.length is 0 in this branch, so default will be used
|
|
|
|
|
filter = cjkCheckbox.isChecked() ? std::wregex(L"[\\u3000-\\ua000]{4,}") : std::wregex(L"[\\u0020-\\u1000]{4,}");
|
|
|
|
|
}
|
2019-06-02 02:09:17 -04:00
|
|
|
|
|
2019-06-16 15:28:59 -04:00
|
|
|
|
auto hooks = std::make_shared<QString>();
|
|
|
|
|
try
|
|
|
|
|
{
|
|
|
|
|
Host::FindHooks(processId, sp, [processId, hooks, filter](HookParam hp, const std::wstring& text)
|
|
|
|
|
{
|
|
|
|
|
if (std::regex_search(text, filter)) hooks->append(S(Util::GenerateCode(hp, processId)) + ": " + S(text) + "\n");
|
|
|
|
|
});
|
2019-06-18 00:39:50 -04:00
|
|
|
|
}
|
|
|
|
|
catch (std::out_of_range) { return; }
|
2019-06-16 15:28:59 -04:00
|
|
|
|
QString saveFile = QFileDialog::getSaveFileName(this, SAVE_SEARCH_RESULTS, "./Hooks.txt", TEXT_FILES);
|
|
|
|
|
if (saveFile.isEmpty()) saveFile = "Hooks.txt";
|
|
|
|
|
std::thread([hooks, saveFile]
|
|
|
|
|
{
|
|
|
|
|
for (int lastSize = 0; hooks->size() == 0 || hooks->size() != lastSize; Sleep(2000)) lastSize = hooks->size();
|
|
|
|
|
QTextFile(saveFile, QIODevice::WriteOnly | QIODevice::Truncate).write(hooks->toUtf8());
|
|
|
|
|
hooks->clear();
|
|
|
|
|
}).detach();
|
2019-06-02 02:09:17 -04:00
|
|
|
|
}
|
|
|
|
|
|
2018-12-18 12:14:54 -05:00
|
|
|
|
void MainWindow::Settings()
|
2018-11-10 05:13:59 -05:00
|
|
|
|
{
|
2019-06-16 15:28:59 -04:00
|
|
|
|
QDialog dialog(this, Qt::WindowCloseButtonHint);
|
|
|
|
|
QSettings settings(CONFIG_FILE, QSettings::IniFormat, &dialog);
|
|
|
|
|
QFormLayout layout(&dialog);
|
|
|
|
|
QPushButton saveButton(SAVE_SETTINGS, &dialog);
|
|
|
|
|
layout.addWidget(&saveButton);
|
|
|
|
|
for (auto [value, label] : Array<std::tuple<int&, const char*>>{
|
|
|
|
|
{ Host::defaultCodepage, DEFAULT_CODEPAGE },
|
|
|
|
|
{ TextThread::maxBufferSize, MAX_BUFFER_SIZE },
|
|
|
|
|
{ TextThread::flushDelay, FLUSH_DELAY },
|
|
|
|
|
})
|
2018-12-21 14:18:43 -05:00
|
|
|
|
{
|
2019-06-16 15:28:59 -04:00
|
|
|
|
auto spinBox = new QSpinBox(&dialog);
|
|
|
|
|
spinBox->setMaximum(INT_MAX);
|
|
|
|
|
spinBox->setValue(value);
|
|
|
|
|
layout.insertRow(0, label, spinBox);
|
|
|
|
|
connect(&saveButton, &QPushButton::clicked, [spinBox, label, &settings, &value] { settings.setValue(label, value = spinBox->value()); });
|
|
|
|
|
}
|
|
|
|
|
for (auto [value, label] : Array<std::tuple<bool&, const char*>>{
|
|
|
|
|
{ TextThread::filterRepetition, FILTER_REPETITION },
|
|
|
|
|
})
|
|
|
|
|
{
|
|
|
|
|
auto checkBox = new QCheckBox(&dialog);
|
|
|
|
|
checkBox->setChecked(value);
|
|
|
|
|
layout.insertRow(0, label, checkBox);
|
|
|
|
|
connect(&saveButton, &QPushButton::clicked, [checkBox, label, &settings, &value] { settings.setValue(label, value = checkBox->isChecked()); });
|
|
|
|
|
}
|
|
|
|
|
connect(&saveButton, &QPushButton::clicked, &dialog, &QDialog::accept);
|
|
|
|
|
dialog.setWindowTitle(SETTINGS);
|
|
|
|
|
dialog.exec();
|
2018-11-10 05:13:59 -05:00
|
|
|
|
}
|
|
|
|
|
|
2018-12-18 12:14:54 -05:00
|
|
|
|
void MainWindow::Extensions()
|
2018-07-23 22:57:54 -07:00
|
|
|
|
{
|
2018-11-01 17:02:52 -04:00
|
|
|
|
extenWindow->activateWindow();
|
|
|
|
|
extenWindow->showNormal();
|
2018-07-22 16:53:51 -07:00
|
|
|
|
}
|
2018-07-26 22:42:21 -07:00
|
|
|
|
|
2018-12-18 12:14:54 -05:00
|
|
|
|
void MainWindow::ViewThread(int index)
|
2018-07-26 22:42:21 -07:00
|
|
|
|
{
|
2019-02-03 08:49:58 -05:00
|
|
|
|
ui->ttCombo->setCurrentIndex(index);
|
2019-02-09 00:30:38 -05:00
|
|
|
|
ui->textOutput->setPlainText(S((current = &Host::GetThread(ParseTextThreadString(ui->ttCombo->itemText(index))))->storage->c_str()));
|
2018-12-14 23:14:30 -05:00
|
|
|
|
ui->textOutput->moveCursor(QTextCursor::End);
|
2018-07-26 22:42:21 -07:00
|
|
|
|
}
|