results_dumper.h

Go to the documentation of this file.

#pragma once
 
#include <windows.h>
 
#include "report_formatter.h"
#include "dump_report.h"
#include "pe_buffer.h"
 
namespace pesieve {
 
    class ResultsDumper
    {
    public:
 
        ResultsDumper(std::string _baseDir, bool _quiet)
            : baseDir(_baseDir), quiet(_quiet)
        {
        }
 
        // dump all modules detected as suspicious during the process scan
        ProcessDumpReport* dumpDetectedModules(
            HANDLE hProcess, 
            bool isRefl, 
            ProcessScanReport &process_report, 
            const pesieve::t_dump_mode dump_mode,
            const t_imprec_mode imprec_mode,
            const bool rebase
        );
 
        // dump JSON report from the process scan
        bool dumpJsonReport(ProcessScanReport &process_report, const t_results_filter &filter, const pesieve::t_json_level &jdetails);
 
        bool dumpJsonReport(ProcessDumpReport &process_report);
 
        bool dumpJsonReport(ErrorReport& error_report, const t_results_filter& filter);
 
        std::string getOutputDir()
        {
            return this->dumpDir;
        }
 
        std::string makeOutPath(const std::string &fname, const std::string& defaultExtension = "");
 
    protected:
        bool dumpModule(
            IN HANDLE processHandle,
            IN bool isRefl,
            IN const ModulesInfo &modulesInfo,
            IN ModuleScanReport* modReport,
            IN const peconv::ExportsMapper *exportsMap,
            IN const pesieve::t_dump_mode dump_mode,
            IN const pesieve::t_imprec_mode imprec_mode,
            IN bool rebase,
            OUT ProcessDumpReport &dumpReport
        );

        std::string makeModuleDumpPath(ULONGLONG modBaseAddr, const std::string &fname, const std::string &defaultExtension);
 
        std::string makeDirName(const DWORD process_id);
 
        void makeAndJoinDirectories(std::stringstream& name_stream);
 
        bool fillModuleCopy(IN ModuleScanReport* mod, IN OUT PeBuffer& module_buf);
 
        std::string dumpDir; // dump directory
        std::string baseDir; // base directory
        bool quiet;
    };
 
}; //namespace pesieve