X-Git-Url: https://git.siccegge.de//index.cgi?p=frida%2Ffrida.git;a=blobdiff_plain;f=src%2Fgui%2FMainwindow.cxx;h=55bc86dd4fca2576bb9fa168ec82e2f8754f87c8;hp=fd0c24595da1bfe0efc675ac395970ccce764ece;hb=10d250c78f36a41d98a6dca768f4e2dafec64e57;hpb=f3a965524eae344fbd25ba486e9fa32e1262fcc4 diff --git a/src/gui/Mainwindow.cxx b/src/gui/Mainwindow.cxx index fd0c245..55bc86d 100644 --- a/src/gui/Mainwindow.cxx +++ b/src/gui/Mainwindow.cxx @@ -1,125 +1,250 @@ - #include "Mainwindow.hxx" -#include "widgets/BasicBlockWidget.hxx" +#include "Mainwindow.hxx" #include "qt.hxx" #include "disassembler/llvm/LLVMDisassembler.hxx" -#include +#include "widgets/CFGScene.hxx" +#include "dialogs/NewFunctionDialog.hxx" +#include "dialogs/RenameFunctionDialog.hxx" + #include -#include +namespace { + BasicBlockWidget * + local__add_basic_block(BasicBlock * block, Disassembler * dis, + Mainwindow * mainwindow, + std::map& known_blocks, + CFGScene * scene, uint64_t starty, uint64_t startx); +} -Mainwindow::Mainwindow(const std::string& filename) -{ - openAction = new QAction(tr("&Open"), this); - // saveAction = new QAction(tr("&Save"), this); - exitAction = new QAction(tr("E&xit"), this); - - connect(openAction, SIGNAL(triggered()), this, SLOT(open())); - // connect(saveAction, SIGNAL(triggered()), this, SLOT(save())); - connect(exitAction, SIGNAL(triggered()), qApp, SLOT(quit())); - - fileMenu = menuBar()->addMenu(tr("&File")); - fileMenu->addAction(openAction); - // fileMenu->addAction(saveAction); - fileMenu->addSeparator(); - fileMenu->addAction(exitAction); - - listWidget = new QListWidget(); - stackedWidget = new QStackedWidget(); - dockWidget = new QDockWidget(tr("Functions"), this); - dockWidget->setAllowedAreas(Qt::LeftDockWidgetArea | - Qt::RightDockWidgetArea); - dockWidget->setWidget(listWidget); - addDockWidget(Qt::LeftDockWidgetArea, dockWidget); - setCentralWidget(stackedWidget); - - connect(listWidget, SIGNAL(currentRowChanged(int)), - stackedWidget, SLOT(setCurrentIndex(int))); - - setWindowTitle(tr("Notepad")); - - openBinary(filename); +Mainwindow::Mainwindow(InformationManager* mgr) + : manager(mgr) + , logger(log4cxx::Logger::getLogger("Mainwindow")) { + openAction = new QAction(tr("&Open"), this); + // saveAction = new QAction(tr("&Save"), this); + exitAction = new QAction(tr("E&xit"), this); + + connect(openAction, SIGNAL(triggered()), + this, SLOT(open())); + // connect(saveAction, SIGNAL(triggered()), this, SLOT(save())); + connect(exitAction, SIGNAL(triggered()), + qApp, SLOT(quit())); + + fileMenu = menuBar()->addMenu(tr("&File")); + fileMenu->addAction(openAction); + // fileMenu->addAction(saveAction); + fileMenu->addSeparator(); + fileMenu->addAction(exitAction); + + scripting = new ScriptingDock(tr("Scripting"), this); + scripting->setAllowedAreas(Qt::BottomDockWidgetArea); + addDockWidget(Qt::BottomDockWidgetArea, scripting); + + listWidget = new QListWidget(); + listWidget->setContextMenuPolicy(Qt::CustomContextMenu); + connect(listWidget, SIGNAL(customContextMenuRequested(const QPoint&)), + this, SLOT(showListContextMenu(const QPoint&))); + + stackedWidget = new QStackedWidget(); + dockWidget = new QDockWidget(tr("Functions"), this); + dockWidget->setAllowedAreas(Qt::LeftDockWidgetArea | + Qt::RightDockWidgetArea); + dockWidget->setWidget(listWidget); + addDockWidget(Qt::LeftDockWidgetArea, dockWidget); + setCentralWidget(stackedWidget); + + connect(listWidget, SIGNAL(currentRowChanged(int)), + this, SLOT(switchMainPlane(int))); + + setWindowTitle(tr("FRIDA")); + + mgr->connect_new_function_signal([&] (Function* fun) {addFunction(fun);}); + mgr->connect_new_dyn_symbol_signal([&] (const std::string& name) { + auto item = new QListWidgetItem(name.c_str(), listWidget); + item->setBackground(QBrush(QColor(0xff, 0xdd, 0xdd))); + }); + 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, [=]() { + QListWidgetItem * item = listWidget->currentItem(); + if (item) renameFunction(item); + }); } 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 (*)")); + QString fileName = QFileDialog::getOpenFileName(this, tr("Open File"), "", + tr("Binaries (*)")); + + manager->reset(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); + QListWidgetItem * 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(int index) { + stackedWidget->setCurrentWidget(objects_list[listWidget->currentItem()]); +} + +void Mainwindow::showListContextMenu(const QPoint& point) { + QListWidgetItem * item = listWidget->itemAt(point); + QMenu menu(this); + if (item) { + QAction * act = menu.addAction("Rename Function"); + connect(act, &QAction::triggered, [=]() {this->renameFunction(item);}); + } else { + QAction * act = menu.addAction("AddFunction"); + connect(act, SIGNAL(triggered()), this, SLOT(requestNewFunction())); + } + menu.exec(listWidget->mapToGlobal(point)); +} + +void Mainwindow::requestNewFunction() { + NewFunctionDialog dialog; + int result = dialog.exec(); + if (QDialog::Accepted == result) { + LOG4CXX_DEBUG(logger, "requesting Function at " << std::hex << dialog.result()); + manager->getDisassembler()->disassembleFunctionAt(dialog.result()); + } else { + LOG4CXX_DEBUG(logger, "requestNewFunction aborted"); + } +} - openBinary(fileName.toStdString()); +void Mainwindow::renameFunction(QListWidgetItem * item) { + RenameFunctionDialog dialog; + int result = dialog.exec(); + if (QDialog::Accepted == result) { + LOG4CXX_DEBUG(logger, "renaming Function " << item->text().toStdString() + << " to " << dialog.result().toStdString()); + item->setText(dialog.result()); + } else { + LOG4CXX_DEBUG(logger, "renameFunction aborted"); + } } -void Mainwindow::openBinary(const std::string& filename) { - if (filename != "") { - disassembler.reset(new LLVMDisassembler(filename)); - // curBin = new Binary(fileName.toStdString()); - - // std::vector symbols = curBin->getSymbols(); - // if (0 == symbols.size()) - // populateSymbolInformation(".text"); - // for (auto it = symbols.begin(); it != symbols.end(); ++it) { - // populateSymbolInformation(*it); - // } - } +void Mainwindow::addFunction(Function* fun) { + if (functions.find(fun) != functions.end()) + return; + + functions.insert(fun); + + QTabWidget * w = new QTabWidget(); + + // CFG + CFGScene * scene = new CFGScene; + + Disassembler * dis = manager->getDisassembler(); + BasicBlock * block = dis->getBasicBlock(fun->getStartAddress()); + + uint64_t start_address(std::numeric_limits::max()); + for (auto b : fun->blocks()) { + if (b.first < start_address) + start_address = b.first; + } + + local__add_basic_block(block, manager->getDisassembler(), this, + blocks, scene, start_address, 100); + + QGraphicsView * view = new QGraphicsView(scene); + w->addTab(view, "CFG"); + + // Listing + QTableWidget * t = new QTableWidget(); + t->setColumnCount(3); + t->horizontalHeader()->setSectionResizeMode(QHeaderView::ResizeToContents); + + w->addTab(t, "Listing"); + + QListWidgetItem * item = new QListWidgetItem(fun->getName().c_str(), listWidget); + 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)); } -void Mainwindow::populateSymbolInformation(const std::string& sym) { - QTabWidget * w = new QTabWidget(); - - // Listing - QTableWidget * t = new QTableWidget(); - t->setColumnCount(3); - t->horizontalHeader()->setSectionResizeMode(QHeaderView::ResizeToContents); - // curBin->for_each_instruction(sym, [&t](long add, std::string bytes, std::string mnemonic) { - // int row = t->rowCount(); - // std::stringstream s; - // t->setRowCount(t->rowCount() + 1); - // s << std::hex << add; - // t->setItem(row,0,new QTableWidgetItem(s.str().c_str())); - // s.str(""); - // s << std::hex; - // for_each(bytes.begin(), bytes.end(), [&s](char c){s << (unsigned int)((unsigned char)c) << ' ';}); - // t->setItem(row,1,new QTableWidgetItem(s.str().c_str())); - // t->setItem(row,2,new QTableWidgetItem(mnemonic.c_str() + 1)); - // }); - w->addTab(t, "Listing"); - - // CFG - QGraphicsScene * scene = new QGraphicsScene; - - BasicBlockWidget * s1 = new BasicBlockWidget; - scene->addItem(s1); - s1->setFlag(QGraphicsItem::ItemIsMovable, true); - - BasicBlockWidget * s2 = new BasicBlockWidget; - scene->addItem(s2); - s2->setFlag(QGraphicsItem::ItemIsMovable, true); - s2->moveBy(-200, 350); - - BasicBlockWidget * s3 = new BasicBlockWidget; - scene->addItem(s3); - s3->setFlag(QGraphicsItem::ItemIsMovable, true); - s3->moveBy(100, 350); - - BasicBlockWidget * s4 = new BasicBlockWidget; - scene->addItem(s4); - s4->setFlag(QGraphicsItem::ItemIsMovable, true); - s4->moveBy(400, 350); - - - QGraphicsView * view = new QGraphicsView(scene); - w->addTab(view, "CFG"); - - listWidget->addItem(sym.c_str()); - stackedWidget->addWidget(w); +namespace { + BasicBlockWidget * + local__add_basic_block(BasicBlock * block, Disassembler * dis, + Mainwindow * mainwindow, + std::map& known_blocks, + CFGScene * scene, uint64_t starty, uint64_t startx) { + + decltype(known_blocks.begin()) old; + if ((old = known_blocks.find(block->getStartAddress())) != known_blocks.end()) + return old->second; + + std::stringstream s; + s << "BLOCK_" << std::hex << block->getStartAddress() + << "_" << block->getEndAddress(); + BasicBlockWidget * widget = new BasicBlockWidget(s.str().c_str(), + block, mainwindow); + + known_blocks.insert(std::make_pair(block->getStartAddress(), widget)); + + scene->addItem(widget); + widget->setFlag(QGraphicsItem::ItemIsMovable, true); + widget->moveBy(100*startx, block->getStartAddress() - starty); + + dis->printEachInstruction(block->getStartAddress(), + block->getEndAddress(), + [&](uint8_t* bytes, + size_t byte_count, + const std::string& line, + const std::string& ref) { + widget->addItem(bytes, byte_count, + line.c_str() + 1, // remove \t + ref.c_str()); + }); + + BasicBlockWidget *tmp, *nextl(NULL), *nextr(NULL); + BasicBlock * tmpblock; + if (block->getNextBlock(0) != 0) { + int xshift = 0; + if (block->getNextBlock(1) != 0) + xshift = 1; + tmpblock = dis->getBasicBlock(block->getNextBlock(0)); + tmp = local__add_basic_block(tmpblock, dis, + mainwindow, + known_blocks, + scene, starty, startx+xshift); + nextl = tmp; + tmp->addPrevious(widget); + } + if (block->getNextBlock(1) != 0) { + tmpblock = dis->getBasicBlock(block->getNextBlock(1)); + tmp = local__add_basic_block(tmpblock, dis, + mainwindow, + known_blocks, + scene, starty, startx-1); + nextr = tmp; + tmp->addPrevious(widget); + } + widget->addNext(nextl, nextr); + return widget; + } }