]>
git.siccegge.de Git - frida/frida.git/blob - src/gui/Mainwindow.cxx
8a62ef8d2257386cf17b6df7074d48b82a414c29
1 #include "Mainwindow.hxx"
3 #include "disassembler/llvm/LLVMDisassembler.hxx"
5 #include "widgets/CFGScene.hxx"
6 #include "dialogs/NewFunctionDialog.hxx"
12 local__add_basic_block(BasicBlock
* block
, Disassembler
* dis
,
13 std::map
<uint64_t, BasicBlockWidget
*>& known_blocks
,
14 CFGScene
* scene
, uint64_t starty
, uint64_t startx
);
17 Mainwindow::Mainwindow(InformationManager
* mgr
)
19 , logger(log4cxx::Logger::getLogger("Mainwindow")) {
20 openAction
= new QAction(tr("&Open"), this);
21 // saveAction = new QAction(tr("&Save"), this);
22 exitAction
= new QAction(tr("E&xit"), this);
24 connect(openAction
, SIGNAL(triggered()),
26 // connect(saveAction, SIGNAL(triggered()), this, SLOT(save()));
27 connect(exitAction
, SIGNAL(triggered()),
30 fileMenu
= menuBar()->addMenu(tr("&File"));
31 fileMenu
->addAction(openAction
);
32 // fileMenu->addAction(saveAction);
33 fileMenu
->addSeparator();
34 fileMenu
->addAction(exitAction
);
36 scripting
= new ScriptingDock(tr("Scripting"), this);
37 scripting
->setAllowedAreas(Qt::BottomDockWidgetArea
);
38 addDockWidget(Qt::BottomDockWidgetArea
, scripting
);
40 listWidget
= new QListWidget();
41 listWidget
->setContextMenuPolicy(Qt::CustomContextMenu
);
42 connect(listWidget
, SIGNAL(customContextMenuRequested(const QPoint
&)),
43 this, SLOT(showListContextMenu(const QPoint
&)));
45 stackedWidget
= new QStackedWidget();
46 dockWidget
= new QDockWidget(tr("Functions"), this);
47 dockWidget
->setAllowedAreas(Qt::LeftDockWidgetArea
|
48 Qt::RightDockWidgetArea
);
49 dockWidget
->setWidget(listWidget
);
50 addDockWidget(Qt::LeftDockWidgetArea
, dockWidget
);
51 setCentralWidget(stackedWidget
);
53 connect(listWidget
, SIGNAL(currentRowChanged(int)),
54 this, SLOT(switchMainPlane(int)));
56 setWindowTitle(tr("FRIDA"));
58 mgr
->connect_new_function_signal([&] (Function
* fun
) {addFunction(fun
);});
59 mgr
->connect_new_dyn_symbol_signal([&] (const std::string
& name
) {
60 auto item
= new QListWidgetItem(name
.c_str(), listWidget
);
61 item
->setBackground(QBrush(QColor(0xff, 0xdd, 0xdd)));
65 void Mainwindow::quit()
67 QMessageBox messageBox
;
68 messageBox
.setWindowTitle(tr("Frida"));
69 messageBox
.setText(tr("Do you really want to quit?"));
70 messageBox
.setStandardButtons(QMessageBox::Yes
| QMessageBox::No
);
71 messageBox
.setDefaultButton(QMessageBox::No
);
72 if (messageBox
.exec() == QMessageBox::Yes
)
76 void Mainwindow::open() {
77 QString fileName
= QFileDialog::getOpenFileName(this, tr("Open File"), "",
80 manager
->reset(fileName
.toStdString());
83 void Mainwindow::switchMainPlane(int index
) {
84 stackedWidget
->setCurrentWidget(objects_list
[listWidget
->currentItem()]);
87 void Mainwindow::showListContextMenu(const QPoint
& point
) {
88 QListWidgetItem
* item
= listWidget
->itemAt(point
);
90 LOG4CXX_DEBUG(logger
, "WOHO " << item
->text().toStdString());
93 QAction
* act
= menu
.addAction("AddFunction");
94 connect(act
, SIGNAL(triggered()), this, SLOT(requestNewFunction()));
96 menu
.exec(listWidget
->mapToGlobal(point
));
100 void Mainwindow::requestNewFunction() {
101 NewFunctionDialog dialog
;
102 int result
= dialog
.exec();
103 if (QDialog::Accepted
== result
) {
104 LOG4CXX_DEBUG(logger
, "requesting Function at " << std::hex
<< dialog
.result());
105 manager
->getDisassembler()->disassembleFunctionAt(dialog
.result());
107 LOG4CXX_DEBUG(logger
, "requestNewFunction aborted");
111 void Mainwindow::addFunction(Function
* fun
) {
112 if (functions
.find(fun
) != functions
.end())
115 functions
.insert(fun
);
117 QTabWidget
* w
= new QTabWidget();
120 CFGScene
* scene
= new CFGScene
;
122 Disassembler
* dis
= manager
->getDisassembler();
123 BasicBlock
* block
= dis
->getBasicBlock(fun
->getStartAddress());
125 uint64_t start_address(std::numeric_limits
<uint64_t>::max());
126 for (auto b
: fun
->blocks()) {
127 if (b
.first
< start_address
)
128 start_address
= b
.first
;
131 local__add_basic_block(block
, manager
->getDisassembler(), blocks
, scene
,
134 QGraphicsView
* view
= new QGraphicsView(scene
);
135 w
->addTab(view
, "CFG");
138 QTableWidget
* t
= new QTableWidget();
139 t
->setColumnCount(3);
140 t
->horizontalHeader()->setSectionResizeMode(QHeaderView::ResizeToContents
);
142 w
->addTab(t
, "Listing");
144 QListWidgetItem
* item
= new QListWidgetItem(fun
->getName().c_str(), listWidget
);
145 stackedWidget
->addWidget(w
);
146 objects_list
.insert(std::make_pair(item
, w
));
151 local__add_basic_block(BasicBlock
* block
, Disassembler
* dis
,
152 std::map
<uint64_t, BasicBlockWidget
*>& known_blocks
,
153 CFGScene
* scene
, uint64_t starty
, uint64_t startx
) {
155 decltype(known_blocks
.begin()) old
;
156 if ((old
= known_blocks
.find(block
->getStartAddress())) != known_blocks
.end())
160 s
<< "BLOCK_" << std::hex
<< block
->getStartAddress()
161 << "_" << block
->getEndAddress();
162 BasicBlockWidget
* widget
= new BasicBlockWidget(s
.str().c_str(), block
);
164 known_blocks
.insert(std::make_pair(block
->getStartAddress(), widget
));
166 scene
->addItem(widget
);
167 widget
->setFlag(QGraphicsItem::ItemIsMovable
, true);
168 widget
->moveBy(100*startx
, block
->getStartAddress() - starty
);
170 dis
->printEachInstruction(block
->getStartAddress(),
171 block
->getEndAddress(),
174 const std::string
& line
,
175 const std::string
& ref
) {
176 widget
->addItem(bytes
, byte_count
,
177 line
.c_str() + 1, // remove \t
181 BasicBlockWidget
*tmp
, *nextl(NULL
), *nextr(NULL
);
182 BasicBlock
* tmpblock
;
183 if (block
->getNextBlock(0) != 0) {
185 if (block
->getNextBlock(1) != 0)
187 tmpblock
= dis
->getBasicBlock(block
->getNextBlock(0));
188 tmp
= local__add_basic_block(tmpblock
, dis
,
190 scene
, starty
, startx
+xshift
);
192 tmp
->addPrevious(widget
);
194 if (block
->getNextBlock(1) != 0) {
195 tmpblock
= dis
->getBasicBlock(block
->getNextBlock(1));
196 tmp
= local__add_basic_block(tmpblock
, dis
,
198 scene
, starty
, startx
-1);
200 tmp
->addPrevious(widget
);
202 widget
->addNext(nextl
, nextr
);