]>
git.siccegge.de Git - frida/frida.git/blob - src/gui/Mainwindow.cxx
1 #include "Mainwindow.hxx"
3 #include "disassembler/llvm/LLVMDisassembler.hxx"
5 #include "widgets/CFGScene.hxx"
15 local__add_basic_block(BasicBlock
* block
, Disassembler
* dis
,
16 std::map
<uint64_t, BasicBlockWidget
*>& known_blocks
,
17 CFGScene
* scene
, uint64_t starty
, uint64_t startx
);
20 Mainwindow::Mainwindow(InformationManager
* mgr
)
22 openAction
= new QAction(tr("&Open"), this);
23 // saveAction = new QAction(tr("&Save"), this);
24 exitAction
= new QAction(tr("E&xit"), this);
26 connect(openAction
, SIGNAL(triggered()),
28 // connect(saveAction, SIGNAL(triggered()), this, SLOT(save()));
29 connect(exitAction
, SIGNAL(triggered()),
32 fileMenu
= menuBar()->addMenu(tr("&File"));
33 fileMenu
->addAction(openAction
);
34 // fileMenu->addAction(saveAction);
35 fileMenu
->addSeparator();
36 fileMenu
->addAction(exitAction
);
38 scripting
= new ScriptingDock(tr("Scripting"), this);
39 scripting
->setAllowedAreas(Qt::BottomDockWidgetArea
);
40 addDockWidget(Qt::BottomDockWidgetArea
, scripting
);
42 listWidget
= new QListWidget();
43 stackedWidget
= new QStackedWidget();
44 dockWidget
= new QDockWidget(tr("Functions"), this);
45 dockWidget
->setAllowedAreas(Qt::LeftDockWidgetArea
|
46 Qt::RightDockWidgetArea
);
47 dockWidget
->setWidget(listWidget
);
48 addDockWidget(Qt::LeftDockWidgetArea
, dockWidget
);
49 setCentralWidget(stackedWidget
);
51 connect(listWidget
, SIGNAL(currentRowChanged(int)),
52 stackedWidget
, SLOT(setCurrentIndex(int)));
54 setWindowTitle(tr("FRIDA"));
56 mgr
->connect_new_function_signal([&] (Function
* fun
) {addFunction(fun
);});
59 void Mainwindow::quit()
61 QMessageBox messageBox
;
62 messageBox
.setWindowTitle(tr("Notepad"));
63 messageBox
.setText(tr("Do you really want to quit?"));
64 messageBox
.setStandardButtons(QMessageBox::Yes
| QMessageBox::No
);
65 messageBox
.setDefaultButton(QMessageBox::No
);
66 if (messageBox
.exec() == QMessageBox::Yes
)
70 void Mainwindow::open() {
71 QString fileName
= QFileDialog::getOpenFileName(this, tr("Open File"), "",
74 manager
->reset(fileName
.toStdString());
77 void Mainwindow::addFunction(Function
* fun
) {
78 if (functions
.find(fun
) != functions
.end())
81 functions
.insert(fun
);
83 QTabWidget
* w
= new QTabWidget();
86 CFGScene
* scene
= new CFGScene
;
88 Disassembler
* dis
= manager
->getDisassembler();
89 BasicBlock
* block
= dis
->getBasicBlock(fun
->getStartAddress());
91 local__add_basic_block(block
, manager
->getDisassembler(), blocks
, scene
,
92 block
->getStartAddress(), 100);
94 QGraphicsView
* view
= new QGraphicsView(scene
);
95 w
->addTab(view
, "CFG");
98 QTableWidget
* t
= new QTableWidget();
100 t
->horizontalHeader()->setSectionResizeMode(QHeaderView::ResizeToContents
);
102 w
->addTab(t
, "Listing");
104 listWidget
->addItem(fun
->getName().c_str());
105 stackedWidget
->addWidget(w
);
110 local__add_basic_block(BasicBlock
* block
, Disassembler
* dis
,
111 std::map
<uint64_t, BasicBlockWidget
*>& known_blocks
,
112 CFGScene
* scene
, uint64_t starty
, uint64_t startx
) {
114 decltype(known_blocks
.begin()) old
;
115 if ((old
= known_blocks
.find(block
->getStartAddress())) != known_blocks
.end())
119 s
<< "BLOCK_" << std::hex
<< block
->getStartAddress()
120 << "_" << block
->getEndAddress();
121 BasicBlockWidget
* widget
= new BasicBlockWidget(s
.str().c_str(), block
);
123 known_blocks
.insert(std::make_pair(block
->getStartAddress(), widget
));
125 scene
->addItem(widget
);
126 widget
->setFlag(QGraphicsItem::ItemIsMovable
, true);
127 widget
->moveBy(100*startx
, block
->getStartAddress() - starty
);
129 dis
->printEachInstruction(block
->getStartAddress(),
130 block
->getEndAddress(),
133 const std::string
& line
) {
134 widget
->addItem(bytes
, byte_count
, line
.c_str() + 1);
137 BasicBlockWidget
*tmp
, *nextl(NULL
), *nextr(NULL
);
138 BasicBlock
* tmpblock
;
139 if (block
->getNextBlock(0) != 0) {
141 if (block
->getNextBlock(1) != 0)
143 tmpblock
= dis
->getBasicBlock(block
->getNextBlock(0));
144 tmp
= local__add_basic_block(tmpblock
, dis
,
146 scene
, starty
, startx
+xshift
);
148 tmp
->addPrevious(widget
);
150 if (block
->getNextBlock(1) != 0) {
151 tmpblock
= dis
->getBasicBlock(block
->getNextBlock(1));
152 tmp
= local__add_basic_block(tmpblock
, dis
,
154 scene
, starty
, startx
-1);
156 tmp
->addPrevious(widget
);
158 widget
->addNext(nextl
, nextr
);