Add basic save support (Infrastructure)

[frida/frida.git] / src / core / InformationManager.cxx

diff --git a/src/core/InformationManager.cxx b/src/core/InformationManager.cxx
index 8f916ef6d2f5e24d3bf62778e39976aeeee70b3f..0907cf9d2d7f7eeb5a191781c4345e1fc3e5a72c 100644 (file)
--- a/src/core/InformationManager.cxx
+++ b/src/core/InformationManager.cxx
@@ -1,7 +1,49 @@
 #include "InformationManager.hxx"
 #include "disassembler/llvm/LLVMDisassembler.hxx"
+#include "gui/qt.hxx"
+#include "quazip/quazip.h"
+#include "quazip/quazipfile.h"
 
 void InformationManager::reset(const std::string& filename) {
        disassembler.reset(createLLVMDisassembler(filename, this));
-       disassembler.get()->start();
+       if (disassembler.get() != NULL)
+               disassembler.get()->start();
+}
+
+void InformationManager::save(const QString& filename) {
+       QuaZip zip(filename);
+       zip.open(QuaZip::mdCreate);
+       zip.setComment("FRIDA 0.0");
+       QuaZipFile outZipFile(&zip);
+
+       for (Function* fun : functions) {
+               QuaZipNewInfo zipinfo(fun->getName().c_str());
+               zipinfo.setPermissions(static_cast<QFile::Permissions>(0x6444));
+               outZipFile.open(QIODevice::WriteOnly, zipinfo);
+               QXmlStreamWriter stream(&outZipFile);
+               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"
+               stream.writeEndDocument();
+               outZipFile.close();
+       }
+
+       zip.close();
 }