]>
git.siccegge.de Git - frida/frida.git/blob - src/gui/Mainwindow.cxx
1c8ae0bb2225b49e2132b8cb9dcc188c2c2fcb25
1 #include "Mainwindow.hxx"
2 #include "widgets/BasicBlockWidget.hxx"
4 #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 listWidget
= new QListWidget();
28 stackedWidget
= new QStackedWidget();
29 dockWidget
= new QDockWidget(tr("Functions"), this);
30 dockWidget
->setAllowedAreas(Qt::LeftDockWidgetArea
|
31 Qt::RightDockWidgetArea
);
32 dockWidget
->setWidget(listWidget
);
33 addDockWidget(Qt::LeftDockWidgetArea
, dockWidget
);
34 setCentralWidget(stackedWidget
);
36 connect(listWidget
, SIGNAL(currentRowChanged(int)),
37 stackedWidget
, SLOT(setCurrentIndex(int)));
39 setWindowTitle(tr("FRIDA"));
44 void Mainwindow::quit()
46 QMessageBox messageBox
;
47 messageBox
.setWindowTitle(tr("Notepad"));
48 messageBox
.setText(tr("Do you really want to quit?"));
49 messageBox
.setStandardButtons(QMessageBox::Yes
| QMessageBox::No
);
50 messageBox
.setDefaultButton(QMessageBox::No
);
51 if (messageBox
.exec() == QMessageBox::Yes
)
55 void Mainwindow::open() {
56 QString fileName
= QFileDialog::getOpenFileName(this, tr("Open File"), "",
59 openBinary(fileName
.toStdString());
62 void Mainwindow::openBinary(const std::string
& filename
) {
64 disassembler
.reset(new LLVMDisassembler(filename
));
65 disassembler
->forEachFunction([&](uint64_t address
, Function
* fun
) {
66 populateSymbolInformation(fun
);
72 void local__add_basic_block(BasicBlock
* block
, Disassembler
* dis
, QGraphicsScene
* scene
,
73 uint64_t starty
, uint64_t startx
) {
75 s
<< "BLOCK_" << std::hex
<< block
->getStartAddress();
76 BasicBlockWidget
* widget
= new BasicBlockWidget(s
.str().c_str());
77 scene
->addItem(widget
);
78 widget
->setFlag(QGraphicsItem::ItemIsMovable
, true);
79 widget
->moveBy(100*startx
, 10*(block
->getStartAddress() - starty
));
81 dis
->printEachInstruction(block
->getStartAddress(), block
->getEndAddress(),
82 [&](const std::string
& line
) {
83 widget
->addItem(line
.c_str());
86 if (block
->getNextBlock(0) != 0)
87 local__add_basic_block(dis
->getBasicBlock(block
->getNextBlock(0)), dis
, scene
, starty
, startx
+1);
88 if (block
->getNextBlock(1) != 0)
89 local__add_basic_block(dis
->getBasicBlock(block
->getNextBlock(1)), dis
, scene
, starty
, startx
-1);
93 void Mainwindow::populateSymbolInformation(Function
* fun
) {
94 QTabWidget
* w
= new QTabWidget();
97 QGraphicsScene
* scene
= new QGraphicsScene
;
99 BasicBlock
* block
= disassembler
->getBasicBlock(fun
->getStartAddress());
101 local__add_basic_block(block
, disassembler
.get(), scene
, block
->getStartAddress(), 100);
103 QGraphicsView
* view
= new QGraphicsView(scene
);
104 w
->addTab(view
, "CFG");
107 QTableWidget
* t
= new QTableWidget();
108 t
->setColumnCount(3);
109 t
->horizontalHeader()->setSectionResizeMode(QHeaderView::ResizeToContents
);
110 // curBin->for_each_instruction(sym, [&t](long add, std::string bytes, std::string mnemonic) {
111 // int row = t->rowCount();
112 // std::stringstream s;
113 // t->setRowCount(t->rowCount() + 1);
114 // s << std::hex << add;
115 // t->setItem(row,0,new QTableWidgetItem(s.str().c_str()));
118 // for_each(bytes.begin(), bytes.end(), [&s](char c){s << (unsigned int)((unsigned char)c) << ' ';});
119 // t->setItem(row,1,new QTableWidgetItem(s.str().c_str()));
120 // t->setItem(row,2,new QTableWidgetItem(mnemonic.c_str() + 1));
122 w
->addTab(t
, "Listing");
124 listWidget
->addItem(fun
->getName().c_str());
125 stackedWidget
->addWidget(w
);