#include "InformationManager.hxx"
+#include "bindings/Interpreter.hxx"
#include "disassembler/llvm/LLVMDisassembler.hxx"
#include "core/Function.hxx"
#include "core/BasicBlock.hxx"
#include "core/Comment.hxx"
+#include "core/events/ChangeCommentEvent.hxx"
-#include "gui/qt.hxx"
+#include "qt.hxx"
#include <quazip/quazip.h>
#include <quazip/quazipfile.h>
: logger(log4cxx::Logger::getLogger("core.InformationManager"))
{
current_information_manager = this;
+ qRegisterMetaType<NewFunctionEvent>("NewFunctionEvent");
+
+ QPluginLoader* loader = new QPluginLoader("libguilePlugin", NULL);
+ if (!loader->load())
+ LOG4CXX_ERROR(logger, "Loading plugin failed: " << loader->errorString().toStdString());
+ interpreters["GUILE"] = qobject_cast<Interpreter*>(loader->instance());
+ plugins.push_back(loader);
}
InformationManager::~InformationManager() {
+ disassemblerThread.quit();
for (auto b : blocks)
delete b.second;
for (auto f : functions)
delete f.second;
+
+ for (auto i : plugins)
+ delete i;
+ disassemblerThread.wait();
}
void InformationManager::reset(const std::string& filename) {
this->filename = filename;
disassembler.reset(createLLVMDisassembler(filename, this));
- if (disassembler.get() != NULL)
+ if (disassembler.get() != NULL) {
+ emit resetEvent();
+#ifndef CONFIG_TEST
+ disassembler.get()->moveToThread(&disassemblerThread);
+ connect(&disassemblerThread, &QThread::started,
+ disassembler.get(), &Disassembler::start);
+ disassemblerThread.start();
+#else
disassembler.get()->start();
+#endif
+ }
}
void InformationManager::load(const std::string& filename) {
QuaZipFile file(&zip);
QuaZipFileInfo info;
- zip.open(QuaZip::mdUnzip);
+ if (!zip.open(QuaZip::mdUnzip)) {
+ LOG4CXX_ERROR(logger, "Failed to open archive " << filename);
+ return;
+ }
tmpfile.reset(new QTemporaryFile());
{
file.open(QIODevice::ReadOnly);
if(info.name != "binary") {
- QXmlStreamReader reader(&file);
- assert(QXmlStreamReader::StartDocument == reader.readNext());
- assert(QXmlStreamReader::StartElement == reader.readNext());
- Function * fun = Function::deserialize(reader, this);
+ if (info.name.startsWith("comment:")) {
+ QXmlStreamReader reader(&file);
+ auto starttoken = reader.readNext();
+ auto elementtoken = reader.readNext();
+ assert(QXmlStreamReader::StartDocument == starttoken);
+ assert(QXmlStreamReader::StartElement == elementtoken);
+ Comment::deserialize(reader, this);
+ } else {
+ QXmlStreamReader reader(&file);
+ auto starttoken = reader.readNext();
+ auto elementtoken = reader.readNext();
+ assert(QXmlStreamReader::StartDocument == starttoken);
+ assert(QXmlStreamReader::StartElement == elementtoken);
+ Function::deserialize(reader, this);
+ }
}
file.close();
}
stream.writeEndDocument();
outZipFile.close();
}
+ for (auto commentpair : comments) {
+ Comment* comment = commentpair.second;
+ if (!comment->isLocal()) {
+ QuaZipNewInfo zipinfo(QString("comment:%1").arg(comment->getAddress(), 0, 16));
+ zipinfo.setPermissions(static_cast<QFile::Permissions>(0x6444));
+ outZipFile.open(QIODevice::WriteOnly, zipinfo);
+ QXmlStreamWriter stream(&outZipFile);
+ stream.setAutoFormatting(true);
+ stream.setAutoFormattingIndent(-1);
+ stream.writeStartDocument();
+
+ comment->serialize(stream);
+
+ stream.writeEndDocument();
+ outZipFile.close();
+ }
+ }
zip.close();
}
-void InformationManager::signal_new_function(Function* fun) {
-}
+
+/* *******************************
+ * Accessors for the Functions map
+ */
Function* InformationManager::getFunction(uint64_t address) {
auto it = functions.find(address);
return NULL;
}
+std::map<uint64_t, Function*>::const_iterator InformationManager::beginFunctions() {
+ return functions.begin();
+}
+std::map<uint64_t, Function*>::const_iterator InformationManager::endFunctions() {
+ return functions.end();
+}
+
+
+/* *********************************
+ * Accessors for the BasicBlocks map
+ */
+
BasicBlock* InformationManager::getBasicBlock(uint64_t address) {
auto it = blocks.find(address);
if (it != blocks.end())
return NULL;
}
+std::map<uint64_t, BasicBlock*>::const_iterator InformationManager::beginBasicBlocks() {
+ return blocks.begin();
+}
+std::map<uint64_t, BasicBlock*>::const_iterator InformationManager::endBasicBlocks() {
+ return blocks.end();
+}
+
+/* *********************************
+ * Accessors for the Comments map
+ */
+std::pair<
+ std::multimap<uint64_t, Comment*>::const_iterator,
+ std::multimap<uint64_t, Comment*>::const_iterator>
+InformationManager::getComments(uint64_t address) {
+ return comments.equal_range(address);
+}
+
+std::multimap<uint64_t, Comment*>::const_iterator InformationManager::beginComments() {
+ return comments.begin();
+}
+std::multimap<uint64_t, Comment*>::const_iterator InformationManager::endComments() {
+ return comments.end();
+}
+
+/* *********************************
+ * Accessors for the Interpreter map
+ */
+
+Interpreter* InformationManager::getInterpreter(const std::string& name) {
+ auto it = interpreters.find(name);
+ if (it != interpreters.end())
+ return it->second;
+ else
+ return NULL;
+}
+
+std::map<std::string, Interpreter*>::const_iterator InformationManager::beginInterpreters() {
+ return interpreters.begin();
+}
+std::map<std::string, Interpreter*>::const_iterator InformationManager::endInterpreters() {
+ return interpreters.end();
+}
+
+
+/* ********************************
+ * Factory methods for data classes
+ */
+
Function* InformationManager::newFunction(uint64_t address) {
- Function* fun = new Function(address, this);
+ Function* fun = new Function(address, false, this);
+ functions.insert(std::make_pair(address, fun));
+ return fun;
+}
+
+Function* InformationManager::newDynamicFunction(uint64_t address) {
+ Function* fun = new Function(address, true, this);
functions.insert(std::make_pair(address, fun));
return fun;
}
}
Comment* InformationManager::newGlobalComment(uint64_t address) {
- return NULL;
+ Comment* comment = new Comment(address, this);
+ comments.insert(std::make_pair(address, comment));
+ return comment;
}
Comment* InformationManager::newLocalComment(uint64_t address, Function* f) {
- return NULL;
+ Comment* comment = new Comment(address, f, this);
+ comments.insert(std::make_pair(address, comment));
+ return comment;
}
void InformationManager::finishFunction(Function* fun) {
BasicBlock* bl = b.second;
blocks.insert(std::make_pair(bl->getStartAddress(), bl));
}
- new_function_signal(fun);
+ NewFunctionEvent event(fun->getStartAddress(), fun);
+ emit newFunctionEvent(event);
}
-void InformationManager::finishBasicBlock(BasicBlock* b) {
+void InformationManager::finishBasicBlock(BasicBlock*) {
}
-void InformationManager::finnishComment(Comment* c) {
+void InformationManager::finishComment(Comment* c) {
+ LOG4CXX_DEBUG(logger, "Finishing comment " << c->getAddress());
+ ChangeCommentEvent event(c->getAddress(), c->getLocation(), c);
+ emit changeCommentEvent(&event);
}
void InformationManager::deleteFunction(Function* f) {
}
void InformationManager::deleteComment(Comment* c) {
+ auto range = comments.equal_range(c->getAddress());
+ for (auto it = range.first; it != range.second; ++it) {
+ if (it->second == c) {
+ comments.erase(it);
+ break;
+ }
+ }
delete c;
}
projects / frida / frida.git / blobdiff