X-Git-Url: https://git.siccegge.de//index.cgi?p=frida%2Ffrida.git;a=blobdiff_plain;f=src%2Fgui%2FMainwindow.cxx;h=269f3e7359d7c82b89e613c011d6f614097cee0a;hp=3e26227ad6acd086854dfe7ab08a1d347c2b605f;hb=5514e0b76ec9af2fa67abebe77cfe26052f2326f;hpb=9023eb3885faa52eb9729b61b401e5b131199c0d diff --git a/src/gui/Mainwindow.cxx b/src/gui/Mainwindow.cxx index 3e26227..269f3e7 100644 --- a/src/gui/Mainwindow.cxx +++ b/src/gui/Mainwindow.cxx @@ -1,46 +1,266 @@ -#include "Mainwindow.h++" +#include "Mainwindow.hxx" +#include "qt.hxx" +#include "bindings/Guile.hxx" +#include "disassembler/llvm/LLVMDisassembler.hxx" +#include "core/Function.hxx" +#include "core/BasicBlock.hxx" +#include "core/InformationManager.hxx" +#include "core/events/RenameFunctionEvent.hxx" -#include +#include "widgets/FridaDock.hxx" +#include "widgets/LogDock.hxx" +#include "widgets/ScriptingDock.hxx" +#include "widgets/CFGScene.hxx" +#include "widgets/FunctionWidget.hxx" +#include "dialogs/NewFunctionDialog.hxx" +#include "dialogs/SimpleStringDialog.hxx" -Mainwindow::Mainwindow() -{ - // openAction = new QAction(tr("&Open"), this); - // saveAction = new QAction(tr("&Save"), this); - exitAction = new QAction(tr("E&xit"), this); +#include + +Mainwindow::Mainwindow(InformationManager* mgr) + : manager(mgr) + , logger(log4cxx::Logger::getLogger("Mainwindow")) { + openAction = new QAction(tr("&Open"), this); + loadAction = new QAction(tr("&Load"), this); + saveAction = new QAction(tr("&Save"), this); + exitAction = new QAction(tr("E&xit"), this); + + connect(openAction, &QAction::triggered, + this, &Mainwindow::open); + connect(loadAction, &QAction::triggered, + this, &Mainwindow::load); + connect(saveAction, &QAction::triggered, + this, &Mainwindow::save); + connect(exitAction, &QAction::triggered, + qApp, &QApplication::quit); + + fileMenu = menuBar()->addMenu(tr("&File")); + fileMenu->addAction(openAction); + fileMenu->addAction(loadAction); + fileMenu->addAction(saveAction); + fileMenu->addSeparator(); + fileMenu->addAction(exitAction); + + QMenu* interpretermenu = menuBar()->addMenu(tr("&Interpreter")); + + QPluginLoader* loader = new QPluginLoader("libguilePlugin", this); + if (!loader->load()) + LOG4CXX_ERROR(logger, "Loading plugin failed: " << loader->errorString().toStdString()); + interpreter["GUILE"] = qobject_cast(loader->instance()); + fdock = new FridaDock(tr("Frida Dock"), this); - // connect(openAction, SIGNAL(triggered()), this, SLOT(open())); - // connect(saveAction, SIGNAL(triggered()), this, SLOT(save())); - connect(exitAction, SIGNAL(triggered()), qApp, SLOT(quit())); + fdock->addTab(new LogDock(fdock), "Log"); - fileMenu = menuBar()->addMenu(tr("&File")); - // fileMenu->addAction(openAction); - // fileMenu->addAction(saveAction); - // fileMenu->addSeparator(); - fileMenu->addAction(exitAction); + fdock->addTab(new ScriptingDock(interpreter["GUILE"], fdock), "guile"); + fdock->setAllowedAreas(Qt::BottomDockWidgetArea); + addDockWidget(Qt::BottomDockWidgetArea, fdock); + QAction* guileLoad = new QAction(tr("&GUILE"), this); + interpretermenu->addAction(guileLoad); + connect(guileLoad, &QAction::triggered, + [&]() { + QString fileName = QFileDialog::getOpenFileName(this, tr("Open Script"), "", + tr("Binaries") + " (*." + + interpreter["GUILE"]->fileExtension().c_str() + ")"); + std::stringstream a, b; + std::string c; + interpreter["GUILE"]->loadFile(fileName.toStdString(), a, b, c); + }); - // - // setCentralWidget(textEdit); + listWidget = new QTreeWidget(); + listWidget->setColumnCount(1); + listWidget->setDragDropMode(QAbstractItemView::InternalMove); + listWidget->setContextMenuPolicy(Qt::CustomContextMenu); + connect(listWidget, SIGNAL(customContextMenuRequested(const QPoint&)), + this, SLOT(showListContextMenu(const QPoint&))); - tabwidget = new QTabWidget; - setCentralWidget(tabwidget); + stackedWidget = new QStackedWidget(); + dockWidget = new QDockWidget(tr("Functions"), this); + dockWidget->setAllowedAreas(Qt::LeftDockWidgetArea | + Qt::RightDockWidgetArea); + dockWidget->setWidget(listWidget); + addDockWidget(Qt::LeftDockWidgetArea, dockWidget); + setCentralWidget(stackedWidget); - textEdit = new QTextEdit; - tabwidget->addTab(textEdit, "edit"); + connect(listWidget, &QTreeWidget::currentItemChanged, + [=] (QTreeWidgetItem* current, QTreeWidgetItem*) { + switchMainPlane(current); + }); - setWindowTitle(tr("Notepad")); + setWindowTitle(tr("FRIDA")); + + QTreeWidgetItem * external = new QTreeWidgetItem(listWidget, QStringList("External Functions")); + external->setChildIndicatorPolicy(QTreeWidgetItem::ShowIndicator); + external->setBackground(0, QBrush(QColor(0xff, 0xdd, 0xdd))); + mgr->connect_new_function_signal([&] (Function* fun) {addFunction(fun);}); + mgr->connect_new_dyn_symbol_signal([=] (const std::string& name) { + auto item = new QTreeWidgetItem(external, QStringList(name.c_str())); + item->setBackground(0, QBrush(QColor(0xff, 0xdd, 0xdd))); + }); + mgr->connect_rename_function_signal([&](RenameFunctionEvent* event) { + if (objects_list_by_address.find(event->address) == objects_list_by_address.end()) + return; + auto item = objects_list_by_address[event->address]; + if (item) item->setText(0, event->new_name.c_str()); + }); + setGlobalHotkeys(); +} + +void Mainwindow::setGlobalHotkeys() { + QShortcut *shortcut = new QShortcut(QKeySequence("f"), this); + connect(shortcut, &QShortcut::activated, this, &Mainwindow::requestNewFunction); + + shortcut = new QShortcut(QKeySequence("r"), listWidget); + connect(shortcut, &QShortcut::activated, [=]() { + QTreeWidgetItem * item = listWidget->currentItem(); + if (item) renameFunction(objects_list[item]->getFunction()); + }); } void Mainwindow::quit() { - QMessageBox messageBox; - messageBox.setWindowTitle(tr("Notepad")); - messageBox.setText(tr("Do you really want to quit?")); - messageBox.setStandardButtons(QMessageBox::Yes | QMessageBox::No); - messageBox.setDefaultButton(QMessageBox::No); - if (messageBox.exec() == QMessageBox::Yes) - qApp->quit(); + QMessageBox messageBox; + messageBox.setWindowTitle(tr("Frida")); + messageBox.setText(tr("Do you really want to quit?")); + messageBox.setStandardButtons(QMessageBox::Yes | QMessageBox::No); + messageBox.setDefaultButton(QMessageBox::No); + if (messageBox.exec() == QMessageBox::Yes) + qApp->quit(); } void Mainwindow::open() { + QString fileName = QFileDialog::getOpenFileName(this, tr("Open File"), "", + tr("Binaries (*)")); + manager->reset(fileName.toStdString()); +} + +void Mainwindow::load() { + QString fileName = QFileDialog::getOpenFileName(this, tr("Open File"), "", + tr("Frida Archives (*.frida)")); + manager->load(fileName.toStdString()); +} + +void Mainwindow::save() { + QString filename = QFileDialog::getSaveFileName(this, tr("Save File"), "", tr("Frida Archives (*.frida)")); + manager->save(filename.toStdString()); +} + +void Mainwindow::switchMainPlaneToAddress(uint64_t address) { + if (objects_list_by_address.find(address) != objects_list_by_address.end()) { + LOG4CXX_DEBUG(logger, "Switching to function " << std::hex << address); + QTreeWidgetItem * item = objects_list_by_address[address]; + listWidget->setCurrentItem(item); + stackedWidget->setCurrentWidget(objects_list[item]); + } else { + LOG4CXX_DEBUG(logger, "No function at " << std::hex << address + << " -- it's probably an imported Symbol"); + } +} + +void Mainwindow::switchMainPlane(QTreeWidgetItem* to) { + if (objects_list.end() != objects_list.find(to)) + stackedWidget->setCurrentWidget(objects_list[to]); +} + +void Mainwindow::showListContextMenu(const QPoint& point) { + QAction * act; + QTreeWidgetItem * item = listWidget->itemAt(point); + QMenu menu(this); + + act = menu.addAction("Add Function"); + connect(act, &QAction::triggered, this, &Mainwindow::requestNewFunction); + + act = menu.addAction("Add Group"); + connect(act, &QAction::triggered, this, &Mainwindow::requestNewGroup); + + if (item) { + if (objects_list.find(item) != objects_list.end()) { + act = menu.addAction("Rename Function"); + connect(act, &QAction::triggered, [=]() {this->renameFunction(objects_list[item]->getFunction());}); + } else { + act = menu.addAction("Rename Group"); + connect(act, &QAction::triggered, [=]() {renameGroup(item);}); + } + + + QMenu* submenu = menu.addMenu("Move to group"); + + for (QTreeWidgetItem* groupitem : group_list) { + act = submenu->addAction(groupitem->text(0)); + connect(act, &QAction::triggered, + [=] () { + listWidget->invisibleRootItem()->removeChild(item); + groupitem->addChild(item); + }); + } + } + + menu.exec(listWidget->mapToGlobal(point)); +} + +void Mainwindow::requestNewFunction() { + NewFunctionDialog dialog; + int result = dialog.exec(); + if (QDialog::Accepted == result) { + requestNewFunctionByAddress(dialog.result()); + } else { + LOG4CXX_DEBUG(logger, "requestNewFunction aborted"); + } +} + +void Mainwindow::requestNewGroup() { + SimpleStringDialog dialog("New Group"); + int result = dialog.exec(); + if (QDialog::Accepted == result) { + QTreeWidgetItem * external = new QTreeWidgetItem(listWidget, QStringList(dialog.result())); + external->setChildIndicatorPolicy(QTreeWidgetItem::ShowIndicator); + group_list.push_back(external); + } else { + LOG4CXX_DEBUG(logger, "requestNewGroup aborted"); + } +} + +void Mainwindow::requestNewFunctionByAddress(uint64_t address) { + LOG4CXX_DEBUG(logger, "requesting Function at " << std::hex << address); + manager->getDisassembler()->disassembleFunctionAt(address); + switchMainPlaneToAddress(address); +} + +void Mainwindow::renameFunction(Function* function) { + SimpleStringDialog dialog("New name"); + int result = dialog.exec(); + if (QDialog::Accepted == result) { + LOG4CXX_DEBUG(logger, "renaming Function " << function->getName() + << " to " << dialog.result().toStdString()); + function->setName(dialog.result().toStdString()); + } else { + LOG4CXX_DEBUG(logger, "renameFunction aborted"); + } +} + +void Mainwindow::renameGroup(QTreeWidgetItem* item) { + SimpleStringDialog dialog("New name"); + int result = dialog.exec(); + if (QDialog::Accepted == result) { + LOG4CXX_DEBUG(logger, "renaming group " << item->text(0).toStdString() + << " to " << dialog.result().toStdString()); + item->setText(0, dialog.result()); + } else { + LOG4CXX_DEBUG(logger, "renameFunction aborted"); + } +} + +void Mainwindow::addFunction(Function* fun) { + if (functions.find(fun->getStartAddress()) != functions.end()) + return; + + functions.insert(std::make_pair(fun->getStartAddress(), fun)); + + FunctionWidget * w = new FunctionWidget(fun, this); + QTreeWidgetItem * item = new QTreeWidgetItem(listWidget, QStringList(fun->getName().c_str())); + stackedWidget->addWidget(w); + objects_list.insert(std::make_pair(item, w)); + LOG4CXX_DEBUG(logger, "Adding function widget at " << std::hex + << fun->getStartAddress()); + objects_list_by_address.insert(std::make_pair(fun->getStartAddress(), item)); }