#include "disassembler/llvm/LLVMDisassembler.hxx"
#include "core/Function.hxx"
#include "core/BasicBlock.hxx"
+#include "core/Comment.hxx"
#include "gui/qt.hxx"
#include <quazip/quazip.h>
#include <quazip/quazipfile.h>
+#include <QTemporaryFile>
+
+InformationManager* current_information_manager;
+
+InformationManager::InformationManager()
+ : logger(log4cxx::Logger::getLogger("InformationManager"))
+{
+ current_information_manager = this;
+}
+
InformationManager::~InformationManager() {
- for (BasicBlock * b : blocks)
- delete b;
+ for (auto b : blocks)
+ delete b.second;
- for (Function * f : functions)
- delete f;
+ for (auto f : functions)
+ delete f.second;
}
void InformationManager::reset(const std::string& filename) {
+ this->filename = filename;
disassembler.reset(createLLVMDisassembler(filename, this));
if (disassembler.get() != NULL)
disassembler.get()->start();
}
-void InformationManager::save(const QString& filename) {
- QuaZip zip(filename);
+void InformationManager::load(const std::string& filename) {
+ QuaZip zip(filename.c_str());
+ QuaZipFile file(&zip);
+ QuaZipFileInfo info;
+
+ zip.open(QuaZip::mdUnzip);
+ tmpfile.reset(new QTemporaryFile());
+
+ {
+ LOG4CXX_INFO(logger, "Loading binary from archive");
+ zip.setCurrentFile("binary");
+ tmpfile->open();
+ file.open(QIODevice::ReadOnly);
+ QByteArray buffer;
+ while (!file.atEnd()) {
+ buffer = file.read(4096);
+ tmpfile->write(buffer);
+ }
+ tmpfile->flush();
+ file.close();
+ disassembler.reset(createLLVMDisassembler(tmpfile->fileName().toStdString(), this));
+ }
+
+ for (bool more = zip.goToFirstFile(); more; more = zip.goToNextFile()) {
+ zip.getCurrentFileInfo(&info);
+ 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);
+ }
+ file.close();
+ }
+}
+
+void InformationManager::save(const std::string& filename) {
+ QuaZip zip(filename.c_str());
zip.open(QuaZip::mdCreate);
zip.setComment("FRIDA 0.0");
QuaZipFile outZipFile(&zip);
- for (Function* fun : functions) {
+ {
+ QFile binary(this->filename.c_str());
+ binary.open(QIODevice::ReadOnly);
+ QuaZipNewInfo zipinfo("binary");
+ zipinfo.setPermissions(static_cast<QFile::Permissions>(0x6444));
+ outZipFile.open(QIODevice::WriteOnly, zipinfo);
+ QByteArray buffer;
+ while (!binary.atEnd()) {
+ buffer = binary.read(4096);
+ outZipFile.write(buffer);
+ }
+ outZipFile.close();
+ }
+
+ for (auto funpair : functions) {
+ Function* fun = funpair.second;
QuaZipNewInfo zipinfo(fun->getName().c_str());
zipinfo.setPermissions(static_cast<QFile::Permissions>(0x6444));
outZipFile.open(QIODevice::WriteOnly, zipinfo);
stream.setAutoFormatting(true);
stream.setAutoFormattingIndent(-1);
stream.writeStartDocument();
- stream.writeStartElement("function");
- stream.writeAttribute("name", fun->getName().c_str());
- stream.writeAttribute("entry", QString::number(fun->getStartAddress(), 16));
-
- for (auto& blockentry : fun->blocks()) {
- stream.writeStartElement("block");
- stream.writeAttribute("id", blockentry.second->getName().c_str());
- stream.writeTextElement("start", QString::number(blockentry.second->getStartAddress(), 16));
- stream.writeTextElement("end", QString::number(blockentry.second->getEndAddress(), 16));
- if (0 != blockentry.second->getNextBlock(0))
- stream.writeTextElement("next", QString::number(blockentry.second->getNextBlock(0), 16));
- if (0 != blockentry.second->getNextBlock(1))
- stream.writeTextElement("next", QString::number(blockentry.second->getNextBlock(1), 16));
- stream.writeEndElement(); // "block"
- }
- stream.writeEndElement(); // "function"
+ fun->serialize(stream);
+
stream.writeEndDocument();
outZipFile.close();
}
}
void InformationManager::signal_new_function(Function* fun) {
- functions.insert(fun);
- for (auto b : fun->blocks())
- blocks.insert(b.second);
+}
+
+Function* InformationManager::getFunction(uint64_t address) {
+ auto it = functions.find(address);
+ if (it != functions.end())
+ return it->second;
+ else
+ return NULL;
+}
+
+BasicBlock* InformationManager::getBasicBlock(uint64_t address) {
+ auto it = blocks.find(address);
+ if (it != blocks.end())
+ return it->second;
+ else
+ return NULL;
+}
+
+Function* InformationManager::newFunction(uint64_t address) {
+ Function* fun = new Function(address, this);
+ functions.insert(std::make_pair(address, fun));
+ return fun;
+}
+
+BasicBlock* InformationManager::newBasicBlock(uint64_t address) {
+ BasicBlock* block = new BasicBlock(address, this);
+ blocks.insert(std::make_pair(address, block));
+ return block;
+}
+
+Comment* InformationManager::newGlobalComment(uint64_t address) {
+ return NULL;
+}
+
+Comment* InformationManager::newLocalComment(uint64_t address, Function* f) {
+ return NULL;
+}
+
+void InformationManager::finishFunction(Function* fun) {
+ LOG4CXX_DEBUG(logger, "Finishing function " << fun->getName());
+ for (auto b : fun->blocks()) {
+ BasicBlock* bl = b.second;
+ blocks.insert(std::make_pair(bl->getStartAddress(), bl));
+ }
new_function_signal(fun);
}
+
+void InformationManager::finishBasicBlock(BasicBlock* b) {
+}
+
+void InformationManager::finnishComment(Comment* c) {
+}
+
+void InformationManager::deleteFunction(Function* f) {
+ functions.erase(f->getStartAddress());
+ delete f;
+}
+
+void InformationManager::deleteBasicBlock(BasicBlock* b) {
+ blocks.erase(b->getStartAddress());
+ delete b;
+}
+
+void InformationManager::deleteComment(Comment* c) {
+ delete c;
+}