Add binary to saved archive

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

diff --git a/src/core/InformationManager.cxx b/src/core/InformationManager.cxx
index 3eebc3422300cfde2f68323192d7f20b2a46f203..73ea5ef1cda9bfdb61fda0fe59e807ed7d11d524 100644 (file)
--- a/src/core/InformationManager.cxx
+++ b/src/core/InformationManager.cxx
@@ -8,6 +8,10 @@
 #include <quazip/quazip.h>
 #include <quazip/quazipfile.h>
 
+InformationManager::InformationManager()
+       : logger(log4cxx::Logger::getLogger("InformationManager"))
+{}
+
 InformationManager::~InformationManager() {
        for (auto b : blocks)
                delete b.second;
@@ -17,6 +21,7 @@ InformationManager::~InformationManager() {
 }
 
 void InformationManager::reset(const std::string& filename) {
+       this->filename = filename;
        disassembler.reset(createLLVMDisassembler(filename, this));
        if (disassembler.get() != NULL)
                disassembler.get()->start();
@@ -28,6 +33,20 @@ void InformationManager::save(const QString& filename) {
        zip.setComment("FRIDA 0.0");
        QuaZipFile outZipFile(&zip);
 
+       {
+               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());
@@ -87,6 +106,7 @@ Comment* InformationManager::newLocalComment(uint64_t address, Function* f) {
 }
 
 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));