]>
git.siccegge.de Git - frida/frida.git/blob - src/gui/Mainwindow.cxx
1 #include "Mainwindow.hxx"
3 #include "disassembler/llvm/LLVMDisassembler.hxx"
11 Mainwindow::Mainwindow(const std::string
& filename
)
13 openAction
= new QAction(tr("&Open"), this);
14 // saveAction = new QAction(tr("&Save"), this);
15 exitAction
= new QAction(tr("E&xit"), this);
17 connect(openAction
, SIGNAL(triggered()), this, SLOT(open()));
18 // connect(saveAction, SIGNAL(triggered()), this, SLOT(save()));
19 connect(exitAction
, SIGNAL(triggered()), qApp
, SLOT(quit()));
21 fileMenu
= menuBar()->addMenu(tr("&File"));
22 fileMenu
->addAction(openAction
);
23 // fileMenu->addAction(saveAction);
24 fileMenu
->addSeparator();
25 fileMenu
->addAction(exitAction
);
27 scripting
= new ScriptingDock(tr("Scripting"), this);
28 scripting
->setAllowedAreas(Qt::BottomDockWidgetArea
);
29 addDockWidget(Qt::BottomDockWidgetArea
, scripting
);
31 listWidget
= new QListWidget();
32 stackedWidget
= new QStackedWidget();
33 dockWidget
= new QDockWidget(tr("Functions"), this);
34 dockWidget
->setAllowedAreas(Qt::LeftDockWidgetArea
|
35 Qt::RightDockWidgetArea
);
36 dockWidget
->setWidget(listWidget
);
37 addDockWidget(Qt::LeftDockWidgetArea
, dockWidget
);
38 setCentralWidget(stackedWidget
);
40 connect(listWidget
, SIGNAL(currentRowChanged(int)),
41 stackedWidget
, SLOT(setCurrentIndex(int)));
43 setWindowTitle(tr("FRIDA"));
48 void Mainwindow::quit()
50 QMessageBox messageBox
;
51 messageBox
.setWindowTitle(tr("Notepad"));
52 messageBox
.setText(tr("Do you really want to quit?"));
53 messageBox
.setStandardButtons(QMessageBox::Yes
| QMessageBox::No
);
54 messageBox
.setDefaultButton(QMessageBox::No
);
55 if (messageBox
.exec() == QMessageBox::Yes
)
59 void Mainwindow::open() {
60 QString fileName
= QFileDialog::getOpenFileName(this, tr("Open File"), "",
63 openBinary(fileName
.toStdString());
66 void Mainwindow::openBinary(const std::string
& filename
) {
68 disassembler
.reset(new LLVMDisassembler(filename
));
69 disassembler
->forEachFunction([&](uint64_t address
, Function
* fun
) {
70 populateSymbolInformation(fun
);
77 local__add_basic_block(BasicBlock
* block
, Disassembler
* dis
,
78 std::map
<uint64_t, BasicBlockWidget
*>& known_blocks
,
79 QGraphicsScene
* scene
, uint64_t starty
, uint64_t startx
) {
81 decltype(known_blocks
.begin()) old
;
82 if ((old
= known_blocks
.find(block
->getStartAddress())) != known_blocks
.end())
86 s
<< "BLOCK_" << std::hex
<< block
->getStartAddress()
87 << "_" << block
->getEndAddress();
88 BasicBlockWidget
* widget
= new BasicBlockWidget(s
.str().c_str(), block
);
90 known_blocks
.insert(std::make_pair(block
->getStartAddress(), widget
));
92 scene
->addItem(widget
);
93 widget
->setFlag(QGraphicsItem::ItemIsMovable
, true);
94 widget
->moveBy(100*startx
, 10*(block
->getStartAddress() - starty
));
96 dis
->printEachInstruction(block
->getStartAddress(), block
->getEndAddress(),
97 [&](uint8_t* bytes
, size_t byte_count
, const std::string
& line
) {
98 widget
->addItem(bytes
, byte_count
, line
.c_str() + 1);
101 BasicBlockWidget
*tmp
, *nextl(NULL
), *nextr(NULL
);
102 BasicBlock
* tmpblock
;
103 if (block
->getNextBlock(0) != 0) {
104 tmpblock
= dis
->getBasicBlock(block
->getNextBlock(0));
105 tmp
= local__add_basic_block(tmpblock
, dis
,
107 scene
, starty
, startx
+1);
109 tmp
->addPrevious(widget
);
111 if (block
->getNextBlock(1) != 0) {
112 tmpblock
= dis
->getBasicBlock(block
->getNextBlock(1));
113 tmp
= local__add_basic_block(tmpblock
, dis
,
115 scene
, starty
, startx
-1);
117 tmp
->addPrevious(widget
);
119 widget
->addNext(nextl
, nextr
);
124 void Mainwindow::populateSymbolInformation(Function
* fun
) {
125 QTabWidget
* w
= new QTabWidget();
128 QGraphicsScene
* scene
= new QGraphicsScene
;
130 BasicBlock
* block
= disassembler
->getBasicBlock(fun
->getStartAddress());
132 local__add_basic_block(block
, disassembler
.get(), blocks
, scene
, block
->getStartAddress(), 100);
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 listWidget
->addItem(fun
->getName().c_str());
145 stackedWidget
->addWidget(w
);