X-Git-Url: https://git.siccegge.de//index.cgi?p=frida%2Ffrida.git;a=blobdiff_plain;f=src%2Fgui%2FMainwindow.cxx;h=55bc86dd4fca2576bb9fa168ec82e2f8754f87c8;hp=46ce2e98562a24cca3b63c460a935529a979de48;hb=10d250c78f36a41d98a6dca768f4e2dafec64e57;hpb=6375df7c896a8a0e4be6959392848f28b021073c

diff --git a/src/gui/Mainwindow.cxx b/src/gui/Mainwindow.cxx
index 46ce2e9..55bc86d 100644
--- a/src/gui/Mainwindow.cxx
+++ b/src/gui/Mainwindow.cxx
@@ -3,16 +3,15 @@
 #include "disassembler/llvm/LLVMDisassembler.hxx"
 
 #include "widgets/CFGScene.hxx"
+#include "dialogs/NewFunctionDialog.hxx"
+#include "dialogs/RenameFunctionDialog.hxx"
 
-#include <iostream>
 #include <sstream>
-#include <map>
-
-#include <QtGui>
 
 namespace {
 	BasicBlockWidget *
 	local__add_basic_block(BasicBlock * block, Disassembler * dis,
+	                       Mainwindow * mainwindow,
 	                       std::map<uint64_t, BasicBlockWidget*>& known_blocks,
 	                       CFGScene * scene, uint64_t starty, uint64_t startx);
 }
@@ -41,6 +40,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 |
@@ -50,7 +53,7 @@ Mainwindow::Mainwindow(InformationManager* mgr)
 	setCentralWidget(stackedWidget);
 
 	connect(listWidget, SIGNAL(currentRowChanged(int)),
-	        stackedWidget, SLOT(setCurrentIndex(int)));
+	        this, SLOT(switchMainPlane(int)));
 
 	setWindowTitle(tr("FRIDA"));
 
@@ -59,12 +62,24 @@ Mainwindow::Mainwindow(InformationManager* mgr)
 			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.setWindowTitle(tr("Frida"));
 	messageBox.setText(tr("Do you really want to quit?"));
 	messageBox.setStandardButtons(QMessageBox::Yes | QMessageBox::No);
 	messageBox.setDefaultButton(QMessageBox::No);
@@ -79,6 +94,58 @@ void Mainwindow::open() {
 	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");
+	}
+}
+
+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::addFunction(Function* fun) {
 	if (functions.find(fun) != functions.end())
 		return;
@@ -99,8 +166,8 @@ void Mainwindow::addFunction(Function* fun) {
 			start_address = b.first;
 	}
 
-	local__add_basic_block(block, manager->getDisassembler(), blocks, scene,
-	                       start_address, 100);
+	local__add_basic_block(block, manager->getDisassembler(), this,
+	                       blocks, scene, start_address, 100);
 
 	QGraphicsView * view = new QGraphicsView(scene);
 	w->addTab(view, "CFG");
@@ -112,13 +179,18 @@ void Mainwindow::addFunction(Function* fun) {
 
 	w->addTab(t, "Listing");
 
-	listWidget->addItem(fun->getName().c_str());
+	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));
 }
 
 namespace {
 	BasicBlockWidget *
 	local__add_basic_block(BasicBlock * block, Disassembler * dis,
+	                       Mainwindow * mainwindow,
 	                       std::map<uint64_t, BasicBlockWidget*>& known_blocks,
 	                       CFGScene * scene, uint64_t starty, uint64_t startx) {
 
@@ -129,7 +201,8 @@ namespace {
 		std::stringstream s;
 		s << "BLOCK_" << std::hex << block->getStartAddress()
 		  << "_" << block->getEndAddress();
-		BasicBlockWidget * widget = new BasicBlockWidget(s.str().c_str(), block);
+		BasicBlockWidget * widget = new BasicBlockWidget(s.str().c_str(),
+		                                                 block, mainwindow);
 
 		known_blocks.insert(std::make_pair(block->getStartAddress(), widget));
 
@@ -156,6 +229,7 @@ namespace {
 				xshift = 1;
 			tmpblock = dis->getBasicBlock(block->getNextBlock(0));
 			tmp = local__add_basic_block(tmpblock, dis,
+			                             mainwindow,
 			                             known_blocks,
 			                             scene, starty, startx+xshift);
 			nextl = tmp;
@@ -164,6 +238,7 @@ namespace {
 		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;