#include "core/BasicBlock.hxx"
#include "core/InformationManager.hxx"
#include "core/events/RenameFunctionEvent.hxx"
+#include "core/events/NewFunctionEvent.hxx"
#include "widgets/FridaDock.hxx"
#include "widgets/LogDock.hxx"
connect(guileLoad, &QAction::triggered,
[&]() {
QString fileName = QFileDialog::getOpenFileName(this, tr("Open Script"), "",
- tr("Binaries") + " (*." +
+ tr("Scripts") + " (*." +
manager->getInterpreter("GUILE")->fileExtension().c_str() + ")");
- std::stringstream a, b;
- std::string c;
- manager->getInterpreter("GUILE")->loadFile(fileName.toStdString(), a, b, c);
+ if(! fileName.isNull()) {
+ std::stringstream a, b;
+ std::string c;
+ manager->getInterpreter("GUILE")->loadFile(fileName.toStdString(), a, b, c);
+ }
});
listWidget = new QTreeWidget();
setWindowTitle(tr("FRIDA"));
- QTreeWidgetItem * external = new QTreeWidgetItem(listWidget, QStringList("External Functions"));
+ external = new QTreeWidgetItem(listWidget, QStringList("External Functions"));
external->setChildIndicatorPolicy(QTreeWidgetItem::ShowIndicator);
external->setBackground(0, QBrush(QColor(0xff, 0xdd, 0xdd)));
- mgr->connect_new_function_signal([&] (Function* fun) {addFunction(fun);});
- mgr->connect_new_dyn_symbol_signal([=] (const std::string& name) {
- auto item = new QTreeWidgetItem(external, QStringList(name.c_str()));
- item->setBackground(0, QBrush(QColor(0xff, 0xdd, 0xdd)));
- });
- mgr->connect_rename_function_signal([&](RenameFunctionEvent* event) {
- if (objects_list_by_address.find(event->address) == objects_list_by_address.end())
- return;
- auto item = objects_list_by_address[event->address];
- if (item) item->setText(0, event->new_name.c_str());
- });
+ connect(mgr, &InformationManager::resetEvent,
+ [this,mgr]() {
+ connect(this, SIGNAL(requestNewFunctionByAddress(uint64_t)),
+ mgr->getDisassembler(), SLOT(disassembleFunctionAt(uint64_t)));
+ });
+ connect(mgr, &InformationManager::newFunctionEvent,
+ this, &Mainwindow::handleNewFunctionEvent);
+ connect(mgr, &InformationManager::renameFunctionEvent,
+ [&](RenameFunctionEvent* event) {
+ if (objects_list_by_address.find(event->address) == objects_list_by_address.end())
+ return;
+ auto item = objects_list_by_address[event->address];
+ if (item) item->setText(0, event->new_name.c_str());
+ });
setGlobalHotkeys();
}
});
}
+void Mainwindow::handleNewFunctionEvent(NewFunctionEvent event) {
+ std::string name = event.function->getName();
+ if (event.function->isDynamic()) {
+ auto item = new QTreeWidgetItem(external, QStringList(name.c_str()));
+ item->setBackground(0, QBrush(QColor(0xff, 0xdd, 0xdd)));
+ } else {
+ addFunction(event.function);
+ }
+}
+
void Mainwindow::quit()
{
QMessageBox messageBox;
}
void Mainwindow::open() {
- QString fileName = QFileDialog::getOpenFileName(this, tr("Open File"), "",
- tr("Binaries (*)"));
- manager->reset(fileName.toStdString());
+ QFileDialog dialog(this, tr("Open bianry"), "", tr("Binaries (*)"));
+
+ if (dialog.exec()) {
+ QStringList files = dialog.selectedFiles();
+ if(1 != files.size()) {
+ LOG4CXX_ERROR(logger, "Needs exactly one file name")
+ } else {
+ manager->reset(files[0].toStdString());
+ }
+ }
}
void Mainwindow::load() {
- QString fileName = QFileDialog::getOpenFileName(this, tr("Open File"), "",
- tr("Frida Archives (*.frida)"));
- manager->load(fileName.toStdString());
+ QFileDialog dialog(this, tr("Open saved FrIDa file"), "", tr("Frida Archives (*.frida)"));
+
+ if (dialog.exec()) {
+ QStringList files = dialog.selectedFiles();
+ if(1 != files.size()) {
+ LOG4CXX_ERROR(logger, "Needs exactly one file name")
+ } else {
+ manager->load(files[0].toStdString());
+ }
+ }
}
void Mainwindow::save() {
NewFunctionDialog dialog;
int result = dialog.exec();
if (QDialog::Accepted == result) {
- requestNewFunctionByAddress(dialog.result());
+ emit requestNewFunctionByAddress(dialog.result());
} else {
LOG4CXX_DEBUG(logger, "requestNewFunction aborted");
}
}
}
-void Mainwindow::requestNewFunctionByAddress(uint64_t address) {
- LOG4CXX_DEBUG(logger, "requesting Function at " << std::hex << address);
- manager->getDisassembler()->disassembleFunctionAt(address);
- switchMainPlaneToAddress(address);
-}
-
void Mainwindow::renameFunction(Function* function) {
SimpleStringDialog dialog("New name");
int result = dialog.exec();