]>
git.siccegge.de Git - frida/frida.git/blob - src/gui/Mainwindow.cxx
2036c486b705bb872ba51e037deeb3bb1cfb649b
1 #include "Mainwindow.hxx"
3 #include "bindings/Guile.hxx"
4 #include "disassembler/llvm/LLVMDisassembler.hxx"
5 #include "core/Function.hxx"
6 #include "core/BasicBlock.hxx"
7 #include "core/InformationManager.hxx"
8 #include "core/events/RenameFunctionEvent.hxx"
9 #include "widgets/ScriptingDock.hxx"
10 #include "widgets/CFGScene.hxx"
11 #include "widgets/FunctionWidget.hxx"
12 #include "dialogs/NewFunctionDialog.hxx"
13 #include "dialogs/SimpleStringDialog.hxx"
17 Mainwindow::Mainwindow(InformationManager
* mgr
)
19 , logger(log4cxx::Logger::getLogger("Mainwindow")) {
20 openAction
= new QAction(tr("&Open"), this);
21 loadAction
= new QAction(tr("&Load"), this);
22 saveAction
= new QAction(tr("&Save"), this);
23 exitAction
= new QAction(tr("E&xit"), this);
25 connect(openAction
, &QAction::triggered
,
26 this, &Mainwindow::open
);
27 connect(loadAction
, &QAction::triggered
,
28 this, &Mainwindow::load
);
29 connect(saveAction
, &QAction::triggered
,
30 this, &Mainwindow::save
);
31 connect(exitAction
, &QAction::triggered
,
32 qApp
, &QApplication::quit
);
34 fileMenu
= menuBar()->addMenu(tr("&File"));
35 fileMenu
->addAction(openAction
);
36 fileMenu
->addAction(loadAction
);
37 fileMenu
->addAction(saveAction
);
38 fileMenu
->addSeparator();
39 fileMenu
->addAction(exitAction
);
41 QMenu
* interpretermenu
= menuBar()->addMenu(tr("&Interpreter"));
43 QPluginLoader
* loader
= new QPluginLoader("libguilePlugin", this);
45 LOG4CXX_ERROR(logger
, "Loading plugin failed: " << loader
->errorString().toStdString());
46 interpreter
["GUILE"] = qobject_cast
<Interpreter
*>(loader
->instance());
47 scripting
= new ScriptingDock(interpreter
["GUILE"], tr("Scripting"), this);
48 scripting
->setAllowedAreas(Qt::BottomDockWidgetArea
);
49 addDockWidget(Qt::BottomDockWidgetArea
, scripting
);
50 QAction
* guileLoad
= new QAction(tr("&GUILE"), this);
51 interpretermenu
->addAction(guileLoad
);
52 connect(guileLoad
, &QAction::triggered
,
54 QString fileName
= QFileDialog::getOpenFileName(this, tr("Open Script"), "",
55 tr("Binaries") + " (*." +
56 interpreter
["GUILE"]->fileExtension().c_str() + ")");
57 std::stringstream a
, b
;
59 interpreter
["GUILE"]->loadFile(fileName
.toStdString(), a
, b
, c
);
62 listWidget
= new QTreeWidget();
63 listWidget
->setColumnCount(1);
64 listWidget
->setDragDropMode(QAbstractItemView::InternalMove
);
65 listWidget
->setContextMenuPolicy(Qt::CustomContextMenu
);
66 connect(listWidget
, SIGNAL(customContextMenuRequested(const QPoint
&)),
67 this, SLOT(showListContextMenu(const QPoint
&)));
69 stackedWidget
= new QStackedWidget();
70 dockWidget
= new QDockWidget(tr("Functions"), this);
71 dockWidget
->setAllowedAreas(Qt::LeftDockWidgetArea
|
72 Qt::RightDockWidgetArea
);
73 dockWidget
->setWidget(listWidget
);
74 addDockWidget(Qt::LeftDockWidgetArea
, dockWidget
);
75 setCentralWidget(stackedWidget
);
77 connect(listWidget
, &QTreeWidget::currentItemChanged
,
78 [=] (QTreeWidgetItem
* current
, QTreeWidgetItem
*) {
79 switchMainPlane(current
);
82 setWindowTitle(tr("FRIDA"));
84 QTreeWidgetItem
* external
= new QTreeWidgetItem(listWidget
, QStringList("External Functions"));
85 external
->setChildIndicatorPolicy(QTreeWidgetItem::ShowIndicator
);
86 external
->setBackground(0, QBrush(QColor(0xff, 0xdd, 0xdd)));
87 mgr
->connect_new_function_signal([&] (Function
* fun
) {addFunction(fun
);});
88 mgr
->connect_new_dyn_symbol_signal([=] (const std::string
& name
) {
89 auto item
= new QTreeWidgetItem(external
, QStringList(name
.c_str()));
90 item
->setBackground(0, QBrush(QColor(0xff, 0xdd, 0xdd)));
92 mgr
->connect_rename_function_signal([&](RenameFunctionEvent
* event
) {
93 if (objects_list_by_address
.find(event
->address
) == objects_list_by_address
.end())
95 auto item
= objects_list_by_address
[event
->address
];
96 if (item
) item
->setText(0, event
->new_name
.c_str());
101 void Mainwindow::setGlobalHotkeys() {
102 QShortcut
*shortcut
= new QShortcut(QKeySequence("f"), this);
103 connect(shortcut
, &QShortcut::activated
, this, &Mainwindow::requestNewFunction
);
105 shortcut
= new QShortcut(QKeySequence("r"), listWidget
);
106 connect(shortcut
, &QShortcut::activated
, [=]() {
107 QTreeWidgetItem
* item
= listWidget
->currentItem();
108 if (item
) renameFunction(objects_list
[item
]->getFunction());
112 void Mainwindow::quit()
114 QMessageBox messageBox
;
115 messageBox
.setWindowTitle(tr("Frida"));
116 messageBox
.setText(tr("Do you really want to quit?"));
117 messageBox
.setStandardButtons(QMessageBox::Yes
| QMessageBox::No
);
118 messageBox
.setDefaultButton(QMessageBox::No
);
119 if (messageBox
.exec() == QMessageBox::Yes
)
123 void Mainwindow::open() {
124 QString fileName
= QFileDialog::getOpenFileName(this, tr("Open File"), "",
126 manager
->reset(fileName
.toStdString());
129 void Mainwindow::load() {
130 QString fileName
= QFileDialog::getOpenFileName(this, tr("Open File"), "",
131 tr("Frida Archives (*.frida)"));
132 manager
->load(fileName
.toStdString());
135 void Mainwindow::save() {
136 QString filename
= QFileDialog::getSaveFileName(this, tr("Save File"), "", tr("Frida Archives (*.frida)"));
137 manager
->save(filename
.toStdString());
140 void Mainwindow::switchMainPlaneToAddress(uint64_t address
) {
141 if (objects_list_by_address
.find(address
) != objects_list_by_address
.end()) {
142 LOG4CXX_DEBUG(logger
, "Switching to function " << std::hex
<< address
);
143 QTreeWidgetItem
* item
= objects_list_by_address
[address
];
144 listWidget
->setCurrentItem(item
);
145 stackedWidget
->setCurrentWidget(objects_list
[item
]);
147 LOG4CXX_DEBUG(logger
, "No function at " << std::hex
<< address
148 << " -- it's probably an imported Symbol");
152 void Mainwindow::switchMainPlane(QTreeWidgetItem
* to
) {
153 if (objects_list
.end() != objects_list
.find(to
))
154 stackedWidget
->setCurrentWidget(objects_list
[to
]);
157 void Mainwindow::showListContextMenu(const QPoint
& point
) {
159 QTreeWidgetItem
* item
= listWidget
->itemAt(point
);
162 act
= menu
.addAction("Add Function");
163 connect(act
, &QAction::triggered
, this, &Mainwindow::requestNewFunction
);
165 act
= menu
.addAction("Add Group");
166 connect(act
, &QAction::triggered
, this, &Mainwindow::requestNewGroup
);
169 if (objects_list
.find(item
) != objects_list
.end()) {
170 act
= menu
.addAction("Rename Function");
171 connect(act
, &QAction::triggered
, [=]() {this->renameFunction(objects_list
[item
]->getFunction());});
173 act
= menu
.addAction("Rename Group");
174 connect(act
, &QAction::triggered
, [=]() {renameGroup(item
);});
178 QMenu
* submenu
= menu
.addMenu("Move to group");
180 for (QTreeWidgetItem
* groupitem
: group_list
) {
181 act
= submenu
->addAction(groupitem
->text(0));
182 connect(act
, &QAction::triggered
,
184 listWidget
->invisibleRootItem()->removeChild(item
);
185 groupitem
->addChild(item
);
190 menu
.exec(listWidget
->mapToGlobal(point
));
193 void Mainwindow::requestNewFunction() {
194 NewFunctionDialog dialog
;
195 int result
= dialog
.exec();
196 if (QDialog::Accepted
== result
) {
197 requestNewFunctionByAddress(dialog
.result());
199 LOG4CXX_DEBUG(logger
, "requestNewFunction aborted");
203 void Mainwindow::requestNewGroup() {
204 SimpleStringDialog
dialog("New Group");
205 int result
= dialog
.exec();
206 if (QDialog::Accepted
== result
) {
207 QTreeWidgetItem
* external
= new QTreeWidgetItem(listWidget
, QStringList(dialog
.result()));
208 external
->setChildIndicatorPolicy(QTreeWidgetItem::ShowIndicator
);
209 group_list
.push_back(external
);
211 LOG4CXX_DEBUG(logger
, "requestNewGroup aborted");
215 void Mainwindow::requestNewFunctionByAddress(uint64_t address
) {
216 LOG4CXX_DEBUG(logger
, "requesting Function at " << std::hex
<< address
);
217 manager
->getDisassembler()->disassembleFunctionAt(address
);
218 switchMainPlaneToAddress(address
);
221 void Mainwindow::renameFunction(Function
* function
) {
222 SimpleStringDialog
dialog("New name");
223 int result
= dialog
.exec();
224 if (QDialog::Accepted
== result
) {
225 LOG4CXX_DEBUG(logger
, "renaming Function " << function
->getName()
226 << " to " << dialog
.result().toStdString());
227 function
->setName(dialog
.result().toStdString());
229 LOG4CXX_DEBUG(logger
, "renameFunction aborted");
233 void Mainwindow::renameGroup(QTreeWidgetItem
* item
) {
234 SimpleStringDialog
dialog("New name");
235 int result
= dialog
.exec();
236 if (QDialog::Accepted
== result
) {
237 LOG4CXX_DEBUG(logger
, "renaming group " << item
->text(0).toStdString()
238 << " to " << dialog
.result().toStdString());
239 item
->setText(0, dialog
.result());
241 LOG4CXX_DEBUG(logger
, "renameFunction aborted");
245 void Mainwindow::addFunction(Function
* fun
) {
246 if (functions
.find(fun
->getStartAddress()) != functions
.end())
249 functions
.insert(std::make_pair(fun
->getStartAddress(), fun
));
251 FunctionWidget
* w
= new FunctionWidget(fun
, this);
253 QTreeWidgetItem
* item
= new QTreeWidgetItem(listWidget
, QStringList(fun
->getName().c_str()));
254 stackedWidget
->addWidget(w
);
255 objects_list
.insert(std::make_pair(item
, w
));
256 LOG4CXX_DEBUG(logger
, "Adding function widget at " << std::hex
257 << fun
->getStartAddress());
258 objects_list_by_address
.insert(std::make_pair(fun
->getStartAddress(), item
));