]>
git.siccegge.de Git - frida/frida.git/blob - src/gui/Mainwindow.cxx
a15aa66eb299da46b1f5a3da3735fa9c95849813
1 #include "Mainwindow.hxx"
3 #include "disassembler/llvm/LLVMDisassembler.hxx"
4 #include "core/Function.hxx"
5 #include "core/BasicBlock.hxx"
6 #include "core/InformationManager.hxx"
7 #include "widgets/ScriptingDock.hxx"
8 #include "widgets/CFGScene.hxx"
9 #include "widgets/FunctionWidget.hxx"
10 #include "dialogs/NewFunctionDialog.hxx"
11 #include "dialogs/SimpleStringDialog.hxx"
17 local__add_basic_block(BasicBlock
* block
,
18 Mainwindow
* mainwindow
, InformationManager
* manager
,
19 std::map
<uint64_t, BasicBlockWidget
*>& known_blocks
,
20 CFGScene
* scene
, uint64_t starty
, uint64_t startx
);
23 Mainwindow::Mainwindow(InformationManager
* mgr
)
25 , logger(log4cxx::Logger::getLogger("Mainwindow")) {
26 openAction
= new QAction(tr("&Open"), this);
27 saveAction
= new QAction(tr("&Save"), this);
28 exitAction
= new QAction(tr("E&xit"), this);
30 connect(openAction
, SIGNAL(triggered()),
32 connect(saveAction
, SIGNAL(triggered()),
34 connect(exitAction
, SIGNAL(triggered()),
37 fileMenu
= menuBar()->addMenu(tr("&File"));
38 fileMenu
->addAction(openAction
);
39 fileMenu
->addAction(saveAction
);
40 fileMenu
->addSeparator();
41 fileMenu
->addAction(exitAction
);
43 scripting
= new ScriptingDock(tr("Scripting"), this);
44 scripting
->setAllowedAreas(Qt::BottomDockWidgetArea
);
45 addDockWidget(Qt::BottomDockWidgetArea
, scripting
);
47 listWidget
= new QTreeWidget();
48 listWidget
->setColumnCount(1);
49 listWidget
->setContextMenuPolicy(Qt::CustomContextMenu
);
50 connect(listWidget
, SIGNAL(customContextMenuRequested(const QPoint
&)),
51 this, SLOT(showListContextMenu(const QPoint
&)));
53 stackedWidget
= new QStackedWidget();
54 dockWidget
= new QDockWidget(tr("Functions"), this);
55 dockWidget
->setAllowedAreas(Qt::LeftDockWidgetArea
|
56 Qt::RightDockWidgetArea
);
57 dockWidget
->setWidget(listWidget
);
58 addDockWidget(Qt::LeftDockWidgetArea
, dockWidget
);
59 setCentralWidget(stackedWidget
);
61 connect(listWidget
, &QTreeWidget::currentItemChanged
,
62 [=] (QTreeWidgetItem
* current
, QTreeWidgetItem
*) {
63 switchMainPlane(current
);
66 setWindowTitle(tr("FRIDA"));
68 QTreeWidgetItem
* external
= new QTreeWidgetItem(listWidget
, QStringList("External Functions"));
69 external
->setChildIndicatorPolicy(QTreeWidgetItem::ShowIndicator
);
70 external
->setBackground(0, QBrush(QColor(0xff, 0xdd, 0xdd)));
71 mgr
->connect_new_function_signal([&] (Function
* fun
) {addFunction(fun
);});
72 mgr
->connect_new_dyn_symbol_signal([=] (const std::string
& name
) {
73 auto item
= new QTreeWidgetItem(external
, QStringList(name
.c_str()));
74 item
->setBackground(0, QBrush(QColor(0xff, 0xdd, 0xdd)));
79 void Mainwindow::setGlobalHotkeys() {
80 QShortcut
*shortcut
= new QShortcut(QKeySequence("f"), this);
81 connect(shortcut
, &QShortcut::activated
, this, &Mainwindow::requestNewFunction
);
83 shortcut
= new QShortcut(QKeySequence("r"), listWidget
);
84 connect(shortcut
, &QShortcut::activated
, [=]() {
85 QTreeWidgetItem
* item
= listWidget
->currentItem();
86 if (item
) renameFunction(objects_list
[item
]->getFunction());
90 void Mainwindow::quit()
92 QMessageBox messageBox
;
93 messageBox
.setWindowTitle(tr("Frida"));
94 messageBox
.setText(tr("Do you really want to quit?"));
95 messageBox
.setStandardButtons(QMessageBox::Yes
| QMessageBox::No
);
96 messageBox
.setDefaultButton(QMessageBox::No
);
97 if (messageBox
.exec() == QMessageBox::Yes
)
101 void Mainwindow::open() {
102 QString fileName
= QFileDialog::getOpenFileName(this, tr("Open File"), "",
105 manager
->reset(fileName
.toStdString());
108 void Mainwindow::save() {
109 QString filename
= QFileDialog::getSaveFileName(this, tr("Save File"), "", tr("Frida Archives (*.frida)"));
110 manager
->save(filename
);
113 void Mainwindow::switchMainPlaneToAddress(uint64_t address
) {
114 if (objects_list_by_address
.find(address
) != objects_list_by_address
.end()) {
115 LOG4CXX_DEBUG(logger
, "Switching to function " << std::hex
<< address
);
116 QTreeWidgetItem
* item
= objects_list_by_address
[address
];
117 listWidget
->setCurrentItem(item
);
118 stackedWidget
->setCurrentWidget(objects_list
[item
]);
120 LOG4CXX_DEBUG(logger
, "No function at " << std::hex
<< address
121 << " -- it's probably an imported Symbol");
125 void Mainwindow::switchMainPlane(QTreeWidgetItem
* to
) {
126 if (objects_list
.end() != objects_list
.find(to
))
127 stackedWidget
->setCurrentWidget(objects_list
[to
]);
130 void Mainwindow::showListContextMenu(const QPoint
& point
) {
131 QTreeWidgetItem
* item
= listWidget
->itemAt(point
);
134 QAction
* act
= menu
.addAction("Rename Function");
135 connect(act
, &QAction::triggered
, [=]() {this->renameFunction(objects_list
[item
]->getFunction());});
138 QAction
* act
= menu
.addAction("AddFunction");
139 connect(act
, SIGNAL(triggered()), this, SLOT(requestNewFunction()));
141 menu
.exec(listWidget
->mapToGlobal(point
));
144 void Mainwindow::requestNewFunction() {
145 NewFunctionDialog dialog
;
146 int result
= dialog
.exec();
147 if (QDialog::Accepted
== result
) {
148 requestNewFunctionByAddress(dialog
.result());
150 LOG4CXX_DEBUG(logger
, "requestNewFunction aborted");
154 void Mainwindow::requestNewFunctionByAddress(uint64_t address
) {
155 LOG4CXX_DEBUG(logger
, "requesting Function at " << std::hex
<< address
);
156 manager
->getDisassembler()->disassembleFunctionAt(address
);
157 switchMainPlaneToAddress(address
);
160 void Mainwindow::renameFunction(Function
* function
) {
161 SimpleStringDialog
dialog("New name");
162 int result
= dialog
.exec();
163 if (QDialog::Accepted
== result
) {
164 LOG4CXX_DEBUG(logger
, "renaming Function " << function
->getName()
165 << " to " << dialog
.result().toStdString());
166 function
->setName(dialog
.result().toStdString());
167 objects_list_by_address
[function
->getStartAddress()]->setText(0, dialog
.result());
169 LOG4CXX_DEBUG(logger
, "renameFunction aborted");
173 void Mainwindow::addFunction(Function
* fun
) {
174 if (functions
.find(fun
->getStartAddress()) != functions
.end())
177 functions
.insert(std::make_pair(fun
->getStartAddress(), fun
));
179 FunctionWidget
* w
= new FunctionWidget(fun
);
182 CFGScene
* scene
= new CFGScene
;
184 BasicBlock
* block
= manager
->getBasicBlock(fun
->getStartAddress());
186 uint64_t start_address(std::numeric_limits
<uint64_t>::max());
187 for (auto b
: fun
->blocks()) {
188 if (b
.first
< start_address
)
189 start_address
= b
.first
;
192 local__add_basic_block(block
, this,
193 manager
, blocks
, scene
, start_address
, 100);
195 QGraphicsView
* view
= new QGraphicsView(scene
);
196 w
->addTab(view
, "CFG");
199 QTableWidget
* t
= new QTableWidget();
200 t
->setColumnCount(3);
201 t
->horizontalHeader()->setSectionResizeMode(QHeaderView::ResizeToContents
);
203 w
->addTab(t
, "Listing");
205 QTreeWidgetItem
* item
= new QTreeWidgetItem(listWidget
, QStringList(fun
->getName().c_str()));
206 stackedWidget
->addWidget(w
);
207 objects_list
.insert(std::make_pair(item
, w
));
208 LOG4CXX_DEBUG(logger
, "Adding function widget at " << std::hex
209 << fun
->getStartAddress());
210 objects_list_by_address
.insert(std::make_pair(fun
->getStartAddress(), item
));
215 local__add_basic_block(BasicBlock
* block
,
216 Mainwindow
* mainwindow
, InformationManager
* manager
,
217 std::map
<uint64_t, BasicBlockWidget
*>& known_blocks
,
218 CFGScene
* scene
, uint64_t starty
, uint64_t startx
) {
220 decltype(known_blocks
.begin()) old
;
221 if ((old
= known_blocks
.find(block
->getStartAddress())) != known_blocks
.end())
225 s
<< "BLOCK_" << std::hex
<< block
->getStartAddress()
226 << "_" << block
->getEndAddress();
227 BasicBlockWidget
* widget
= new BasicBlockWidget(s
.str().c_str(),
230 known_blocks
.insert(std::make_pair(block
->getStartAddress(), widget
));
232 scene
->addItem(widget
);
233 widget
->setFlag(QGraphicsItem::ItemIsMovable
, true);
234 widget
->moveBy(100*startx
, block
->getStartAddress() - starty
);
236 manager
->getDisassembler()
237 ->printEachInstruction(block
->getStartAddress(),
238 block
->getEndAddress(),
241 const std::string
& line
,
242 const std::string
& ref
) {
243 widget
->addItem(bytes
, byte_count
,
244 line
.c_str() + 1, // remove \t
248 BasicBlockWidget
*tmp
, *nextl(NULL
), *nextr(NULL
);
249 BasicBlock
* tmpblock
;
250 if (block
->getNextBlock(0) != 0) {
252 if (block
->getNextBlock(1) != 0)
254 tmpblock
= manager
->getBasicBlock(block
->getNextBlock(0));
255 tmp
= local__add_basic_block(tmpblock
, mainwindow
, manager
,
257 scene
, starty
, startx
+xshift
);
259 tmp
->addPrevious(widget
);
261 if (block
->getNextBlock(1) != 0) {
262 tmpblock
= manager
->getBasicBlock(block
->getNextBlock(1));
263 tmp
= local__add_basic_block(tmpblock
, mainwindow
, manager
,
265 scene
, starty
, startx
-1);
267 tmp
->addPrevious(widget
);
269 widget
->addNext(nextl
, nextr
);