Textractor/GUI/mainwindow.cpp

256 lines
8.6 KiB
C++
Raw Normal View History

2018-07-21 15:40:16 -07:00
#include "mainwindow.h"
#include "ui_mainwindow.h"
2018-07-26 22:42:21 -07:00
#include "extensions.h"
2018-07-25 21:48:18 -07:00
#include "misc.h"
2018-08-22 15:11:40 -04:00
#include <QCoreApplication>
#include <QInputDialog>
#include <QFileDialog>
2018-07-23 12:25:02 -07:00
2018-07-21 15:40:16 -07:00
MainWindow::MainWindow(QWidget *parent) :
2018-07-25 10:46:59 -07:00
QMainWindow(parent),
2018-08-21 22:43:30 -04:00
ui(new Ui::MainWindow)
2018-07-21 15:40:16 -07:00
{
2018-07-25 10:46:59 -07:00
ui->setupUi(this);
2018-09-21 22:25:37 -04:00
if (settings.contains("Window")) this->setGeometry(settings.value("Window").toRect());
2018-09-23 22:29:33 -04:00
// TODO: add GUI for changing these
2018-10-09 02:09:52 -04:00
if (settings.contains("Flush_Delay")) TextThread::flushDelay = settings.value("Flush_Delay").toInt();
if (settings.contains("Max_Buffer_Size")) TextThread::maxBufferSize = settings.value("Max_Buffer_Size").toInt();
2018-07-23 12:25:02 -07:00
2018-08-21 22:43:30 -04:00
processCombo = findChild<QComboBox*>("processCombo");
ttCombo = findChild<QComboBox*>("ttCombo");
extenCombo = findChild<QComboBox*>("extenCombo");
textOutput = findChild<QPlainTextEdit*>("textOutput");
connect(this, &MainWindow::SigAddProcess, this, &MainWindow::AddProcess);
connect(this, &MainWindow::SigRemoveProcess, this, &MainWindow::RemoveProcess);
connect(this, &MainWindow::SigAddThread, this, &MainWindow::AddThread);
connect(this, &MainWindow::SigRemoveThread, this, &MainWindow::RemoveThread);
connect(this, &MainWindow::SigThreadOutput, this, &MainWindow::ThreadOutput);
Host::Start(
[&](DWORD processId) { emit SigAddProcess(processId); },
[&](DWORD processId) { emit SigRemoveProcess(processId); },
[&](TextThread* thread) { emit SigAddThread(thread); },
2018-10-08 00:26:43 -04:00
[&](TextThread* thread) { emit SigRemoveThread(thread); },
[&](TextThread* thread, std::wstring& output) { return ProcessThreadOutput(thread, output); }
2018-08-21 22:43:30 -04:00
);
2018-08-17 14:37:37 -04:00
ReloadExtensions();
2018-09-29 20:24:57 -04:00
Host::AddConsoleOutput(L"Textractor beta v3.2.2 by Artikash\r\nSource code and more information available under GPLv3 at https://github.com/Artikash/Textractor");
2018-07-21 15:40:16 -07:00
}
MainWindow::~MainWindow()
{
2018-09-21 22:25:37 -04:00
settings.setValue("Window", this->geometry());
2018-10-09 02:09:52 -04:00
settings.setValue("Flush_Delay", TextThread::flushDelay);
settings.setValue("Max_Buffer_Size", TextThread::maxBufferSize);
2018-09-21 22:25:37 -04:00
settings.sync();
2018-08-22 01:22:34 -04:00
Host::Close();
2018-07-25 10:46:59 -07:00
delete ui;
2018-07-21 15:40:16 -07:00
}
2018-07-22 16:53:51 -07:00
2018-10-08 00:26:43 -04:00
void MainWindow::AddProcess(unsigned processId)
2018-07-23 22:57:54 -07:00
{
processCombo->addItem(QString::number(processId, 16).toUpper() + ": " + GetModuleName(processId));
2018-07-26 22:42:21 -07:00
QFile file("SavedHooks.txt");
if (!file.open(QIODevice::ReadOnly)) return;
QString processName = GetFullModuleName(processId);
2018-10-10 08:16:14 -04:00
QStringList allProcesses = QString(file.readAll()).split("\r", QString::SkipEmptyParts);
for (auto hooks = allProcesses.rbegin(); hooks != allProcesses.rend(); ++hooks)
if (hooks->contains(processName))
2018-07-26 22:42:21 -07:00
{
2018-10-10 08:16:14 -04:00
for (auto hook : hooks->split(" , "))
if (auto hp = ParseCode(hook)) Host::InsertHook(processId, hp.value());
2018-07-26 22:42:21 -07:00
return;
}
2018-07-23 22:57:54 -07:00
}
2018-10-08 00:26:43 -04:00
void MainWindow::RemoveProcess(unsigned processId)
2018-07-23 22:57:54 -07:00
{
processCombo->removeItem(processCombo->findText(QString::number(processId, 16).toUpper() + ":", Qt::MatchStartsWith));
2018-07-23 22:57:54 -07:00
}
void MainWindow::AddThread(TextThread* thread)
2018-07-23 22:57:54 -07:00
{
2018-07-25 10:46:59 -07:00
ttCombo->addItem(
TextThreadString(thread) +
2018-09-20 22:32:47 -04:00
QString::fromStdWString(thread->name) +
2018-07-25 21:48:18 -07:00
" (" +
2018-09-20 22:32:47 -04:00
GenerateCode(Host::GetHookParam(thread->tp), thread->tp.pid) +
2018-07-25 21:48:18 -07:00
")"
2018-07-25 10:46:59 -07:00
);
2018-07-23 22:57:54 -07:00
}
void MainWindow::RemoveThread(TextThread* thread)
2018-07-23 22:57:54 -07:00
{
2018-08-21 22:43:30 -04:00
int threadIndex = ttCombo->findText(TextThreadString(thread), Qt::MatchStartsWith);
2018-07-25 10:46:59 -07:00
if (threadIndex == ttCombo->currentIndex())
{
ttCombo->setCurrentIndex(0);
on_ttCombo_activated(0);
}
2018-08-02 02:17:20 -04:00
ttCombo->removeItem(threadIndex);
2018-08-21 22:43:30 -04:00
delete thread;
}
void MainWindow::ThreadOutput(TextThread* thread, QString output)
{
2018-07-25 10:46:59 -07:00
if (ttCombo->currentText().startsWith(TextThreadString(thread)))
{
textOutput->moveCursor(QTextCursor::End);
textOutput->insertPlainText(output);
textOutput->moveCursor(QTextCursor::End);
}
2018-07-23 22:57:54 -07:00
}
2018-10-08 00:26:43 -04:00
bool MainWindow::ProcessThreadOutput(TextThread* thread, std::wstring& output)
{
if (DispatchSentenceToExtensions(output, GetInfoForExtensions(thread)))
{
output += L"\r\n";
emit SigThreadOutput(thread, QString::fromStdWString(output));
return true;
}
return false;
}
2018-08-22 15:11:40 -04:00
QString MainWindow::TextThreadString(TextThread* thread)
{
2018-09-20 22:32:47 -04:00
ThreadParam tp = thread->tp;
2018-09-21 22:25:37 -04:00
return QString("%1:%2:%3:%4:%5: ").arg(
2018-09-22 17:13:53 -04:00
QString::number(thread->handle, 16),
QString::number(tp.pid, 16),
2018-08-22 15:11:40 -04:00
QString::number(tp.hook, 16),
QString::number(tp.retn, 16),
QString::number(tp.spl, 16)
2018-08-22 15:11:40 -04:00
).toUpper();
}
2018-08-23 11:53:23 -04:00
ThreadParam MainWindow::ParseTextThreadString(QString textThreadString)
2018-08-22 15:11:40 -04:00
{
QStringList threadParam = textThreadString.split(":");
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()
{
return processCombo->currentText().split(":")[0].toULong(nullptr, 16);
2018-08-22 15:11:40 -04:00
}
2018-08-17 14:37:37 -04:00
void MainWindow::ReloadExtensions()
{
extenCombo->clear();
for (auto extension : LoadExtensions()) extenCombo->addItem(QString::number(extension.number) + ": " + extension.name);
2018-08-17 14:37:37 -04:00
}
2018-09-20 22:32:47 -04:00
std::unordered_map<std::string, int64_t> MainWindow::GetInfoForExtensions(TextThread* thread)
{
return
{
{ "current select", ttCombo->currentText().startsWith(TextThreadString(thread)) },
{ "text number", thread->handle },
2018-09-20 22:32:47 -04:00
{ "process id", thread->tp.pid },
{ "hook address", thread->tp.hook },
{ "text handle", thread->handle },
2018-09-20 22:32:47 -04:00
{ "text name", (int64_t)thread->name.c_str() }
};
}
2018-07-25 21:48:18 -07:00
QVector<HookParam> MainWindow::GetAllHooks(DWORD processId)
{
2018-08-22 15:11:40 -04:00
QSet<DWORD> addresses;
2018-07-25 21:48:18 -07:00
QVector<HookParam> hooks;
for (int i = 0; i < ttCombo->count(); ++i)
2018-08-21 22:43:30 -04:00
{
2018-08-23 11:53:23 -04:00
ThreadParam tp = ParseTextThreadString(ttCombo->itemText(i));
2018-08-22 15:11:40 -04:00
if (tp.pid == processId && !addresses.contains(tp.hook))
2018-07-25 21:48:18 -07:00
{
2018-08-21 22:43:30 -04:00
addresses.insert(tp.hook);
hooks.push_back(Host::GetHookParam(tp));
2018-07-25 21:48:18 -07:00
}
2018-08-21 22:43:30 -04:00
}
2018-07-25 21:48:18 -07:00
return hooks;
}
2018-07-23 12:25:02 -07:00
void MainWindow::on_attachButton_clicked()
2018-07-22 16:53:51 -07:00
{
2018-09-09 22:37:48 -04:00
QMultiHash<QString, DWORD> allProcesses = GetAllProcesses();
QStringList processList(allProcesses.uniqueKeys());
2018-08-20 14:30:50 -04:00
processList.sort(Qt::CaseInsensitive);
2018-07-25 21:48:18 -07:00
bool ok;
2018-07-30 20:25:08 -07:00
QString process = QInputDialog::getItem(this, "Select Process",
2018-09-09 22:37:48 -04:00
"If you don't see the process you want to inject, try running with admin rights\r\nYou can also type in the process id if you know it",
2018-08-20 14:30:50 -04:00
processList, 0, true, &ok);
2018-09-13 11:59:15 -04:00
bool injected = false;
2018-07-30 20:25:08 -07:00
if (!ok) return;
2018-09-13 11:59:15 -04:00
if (process.toInt(nullptr, 0)) injected |= Host::InjectProcess(process.toInt(nullptr, 0));
2018-10-10 08:16:14 -04:00
else for (auto processId : allProcesses.values(process)) injected |= Host::InjectProcess(processId);
2018-09-13 11:59:15 -04:00
if (!injected) Host::AddConsoleOutput(L"failed to inject");
2018-07-23 22:57:54 -07:00
}
void MainWindow::on_detachButton_clicked()
{
2018-08-21 22:43:30 -04:00
Host::DetachProcess(GetSelectedProcessId());
2018-07-23 22:57:54 -07:00
}
2018-07-25 21:48:18 -07:00
void MainWindow::on_hookButton_clicked()
{
bool ok;
2018-08-04 18:01:59 -04:00
QString hookCode = QInputDialog::getText(this, "Add Hook", CodeInfoDump, QLineEdit::Normal, "", &ok);
2018-07-30 20:25:08 -07:00
if (!ok) return;
2018-08-24 14:24:46 -04:00
if (auto hp = ParseCode(hookCode)) Host::InsertHook(GetSelectedProcessId(), hp.value());
else Host::AddConsoleOutput(L"invalid code");
2018-07-25 21:48:18 -07:00
}
void MainWindow::on_unhookButton_clicked()
{
2018-08-21 22:43:30 -04:00
QVector<HookParam> hooks = GetAllHooks(GetSelectedProcessId());
2018-10-10 08:16:14 -04:00
if (hooks.size() == 0) return Host::AddConsoleOutput(L"no hooks detected");
2018-07-25 21:48:18 -07:00
QStringList hookList;
2018-10-10 08:16:14 -04:00
for (auto hook : hooks)
hookList.push_back(
QString::fromStdWString(Host::GetHookName(GetSelectedProcessId(), hook.address)) +
": " +
GenerateCode(hook, GetSelectedProcessId())
);
2018-07-25 21:48:18 -07:00
bool ok;
QString hook = QInputDialog::getItem(this, "Unhook", "Which hook to remove?", hookList, 0, false, &ok);
2018-08-21 22:43:30 -04:00
if (ok) Host::RemoveHook(GetSelectedProcessId(), hooks.at(hookList.indexOf(hook)).address);
2018-07-25 21:48:18 -07:00
}
2018-07-26 22:42:21 -07:00
void MainWindow::on_saveButton_clicked()
{
2018-08-21 22:43:30 -04:00
QVector<HookParam> hooks = GetAllHooks(GetSelectedProcessId());
QString hookList = GetFullModuleName(GetSelectedProcessId());
2018-10-10 08:16:14 -04:00
for (auto hook : hooks)
if (!(hook.type & HOOK_ENGINE))
hookList += " , " + GenerateCode(hook, GetSelectedProcessId());
2018-07-26 22:42:21 -07:00
QFile file("SavedHooks.txt");
if (!file.open(QIODevice::Append | QIODevice::Text)) return;
file.write((hookList + "\r\n").toUtf8());
}
2018-07-23 22:57:54 -07:00
void MainWindow::on_ttCombo_activated(int index)
{
2018-10-08 00:26:43 -04:00
textOutput->setPlainText(QString::fromStdWString(Host::GetThread(ParseTextThreadString(ttCombo->itemText(index)))->GetStorage()));
2018-07-25 10:46:59 -07:00
textOutput->moveCursor(QTextCursor::End);
2018-07-22 16:53:51 -07:00
}
2018-07-26 22:42:21 -07:00
void MainWindow::on_addExtenButton_clicked()
{
2018-08-03 00:07:25 -04:00
QString extenFileName = QFileDialog::getOpenFileName(this, "Select Extension dll", "C:\\", "Extensions (*.dll)");
2018-08-25 17:08:45 -04:00
if (!extenFileName.size()) return;
2018-07-28 12:41:21 -07:00
QString extenName = extenFileName.split("/")[extenFileName.split("/").count() - 1];
2018-08-25 17:08:45 -04:00
QString copyTo = QString::number(extenCombo->itemText(extenCombo->count() - 1).split(":")[0].toInt() + 1) + "_" + extenName;
2018-07-28 12:41:21 -07:00
QFile::copy(extenFileName, copyTo);
2018-08-17 14:37:37 -04:00
ReloadExtensions();
2018-07-28 12:41:21 -07:00
}
void MainWindow::on_rmvExtenButton_clicked()
{
2018-08-03 00:07:25 -04:00
if (extenCombo->currentText().size() == 0) return;
2018-08-25 17:08:45 -04:00
QString extenFileName = extenCombo->currentText().split(":")[0] + "_" + extenCombo->currentText().split(": ")[1] + ".dll";
2018-07-28 12:41:21 -07:00
FreeLibrary(GetModuleHandleW(extenFileName.toStdWString().c_str()));
2018-08-19 17:55:50 -04:00
QFile::remove(extenFileName);
2018-08-17 14:37:37 -04:00
ReloadExtensions();
2018-07-26 22:42:21 -07:00
}