dump_report.h

Go to the documentation of this file.

#pragma once
 
#include <windows.h>
 
#include <iostream>
#include <sstream>
#include <string>
#include <vector>
 
#include <peconv.h>
#include "../utils/path_util.h"
#include "../utils/path_converter.h"
 
namespace pesieve {
 
    class ModuleDumpReport
    {
    public:
 
        ModuleDumpReport(ULONGLONG module_start, size_t module_size)
            : moduleStart(module_start), moduleSize(module_size), rebasedTo(module_start),
            isDumped(false), isReportDumped(false),
            is_corrupt_pe(false),
            is_shellcode(false)
        {
        }
 
        const virtual bool toJSON(std::stringstream &outs, size_t level);
 
        ULONGLONG moduleStart;
        size_t moduleSize;
        ULONGLONG rebasedTo;
        bool is_corrupt_pe;
        bool is_shellcode;
        std::string impRecMode;
        bool isReportDumped;
        bool isDumped;
        std::string mode_info;
        std::string dumpFileName;
        std::string hooksTagFileName;
        std::string patternsTagFileName;
        std::string impListFileName;
        std::string notRecoveredFileName;
        std::string iatHooksFileName;
    };
 
    class ProcessDumpReport
    {
    public:
        ProcessDumpReport(DWORD _pid)
            : pid(_pid)
        {
        }
 
        ~ProcessDumpReport()
        {
            deleteModuleReports();
        }
 
        void appendReport(ModuleDumpReport *report)
        {
            if (!report) return;
            moduleReports.push_back(report);
        }
 
        size_t countTotal() const
        {
            return moduleReports.size();
        }
 
        bool isFilled() const
        {
            if (countTotal()) return true;
            if (this->minidumpPath.length()) return true;
            return false;
        }
 
        size_t countDumped() const
        {
            size_t dumped = 0;
            std::vector<ModuleDumpReport*>::const_iterator itr = moduleReports.begin();
            for (; itr != moduleReports.end(); ++itr) {
                ModuleDumpReport* module = *itr;
                if (module->isDumped) {
                    dumped++;
                }
            }
            return dumped;
        }
 
        bool hasModule(const ULONGLONG modBase, const size_t modSize) const
        {
            if (!modBase) return false;
 
            for (auto itr = moduleReports.begin(); itr != moduleReports.end(); ++itr) {
                const ModuleDumpReport* report = *itr;
                if (!report->isDumped) continue; // dumping failed
                if (report->moduleStart == modBase && report->moduleSize == modSize) {
                    return true;
                }
            }
            return false;
        }
 
        virtual bool toJSON(std::stringstream &stream, size_t level) const;
 
        DWORD getPid() const { return pid; }
 
        std::string outputDir;
        std::string minidumpPath;
 
    protected:
 
        std::string list_dumped_modules(size_t level) const;
 
        void deleteModuleReports()
        {
            std::vector<ModuleDumpReport*>::iterator itr = moduleReports.begin();
            for (; itr != moduleReports.end(); ++itr) {
                ModuleDumpReport* module = *itr;
                delete module;
            }
            moduleReports.clear();
        }
 
        DWORD pid;
        std::vector<ModuleDumpReport*> moduleReports;
 
        friend class ResultsDumper;
    };
 
};