Allow addition of new functions

[frida/frida.git] / src / gui / Mainwindow.cxx

diff --git a/src/gui/Mainwindow.cxx b/src/gui/Mainwindow.cxx
index 9f1d353ab2a568c14609c051133ebf87143e62ce..764eb0376d1ca4acb12f56fb0faa7b37c27c98d1 100644 (file)
--- a/src/gui/Mainwindow.cxx
+++ b/src/gui/Mainwindow.cxx
@@ -3,7 +3,7 @@
 #include "disassembler/llvm/LLVMDisassembler.hxx"
 
 #include "widgets/CFGScene.hxx"
-
+#include "dialogs/NewFunctionDialog.hxx"
 #include <iostream>
 #include <sstream>
 #include <map>
@@ -41,6 +41,10 @@ Mainwindow::Mainwindow(InformationManager* mgr)
        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 |
@@ -55,12 +59,16 @@ Mainwindow::Mainwindow(InformationManager* mgr)
        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)));
+               });
 }
 
 void Mainwindow::quit()
 {
        QMessageBox messageBox;
-       messageBox.setWindowTitle(tr("Notepad"));
+       messageBox.setWindowTitle(tr("Frida"));
        messageBox.setText(tr("Do you really want to quit?"));
        messageBox.setStandardButtons(QMessageBox::Yes | QMessageBox::No);
        messageBox.setDefaultButton(QMessageBox::No);
@@ -75,6 +83,30 @@ void Mainwindow::open() {
        manager->reset(fileName.toStdString());
 }
 
+void Mainwindow::showListContextMenu(const QPoint& point) {
+       QListWidgetItem * item = listWidget->itemAt(point);
+       if (item) {
+               LOG4CXX_DEBUG(logger, "WOHO " << item->text().toStdString());
+       } else {
+               QMenu menu(this);
+               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");
+       }
+}
+
 void Mainwindow::addFunction(Function* fun) {
        if (functions.find(fun) != functions.end())
                return;
@@ -137,8 +169,11 @@ namespace {
                                          block->getEndAddress(),
                                          [&](uint8_t* bytes,
                                              size_t byte_count,
-                                             const std::string& line) {
-                                                 widget->addItem(bytes, byte_count, line.c_str() + 1);
+                                             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);