PDBReader.h

Go to the documentation of this file.

#pragma once
 
#include <dia2.h>
#include <unordered_set>
 
#include "json.hpp"
 
#include <API/Fields.h>
 
namespace API
{
    class PdbReader
    {
    public:
        PdbReader() = default;
        ~PdbReader() = default;
 
        void Read(const std::wstring& path, std::unordered_map<std::string, intptr_t>* offsets_dump,
                  std::unordered_map<std::string, BitField>* bitfields_dump, const std::unordered_set<std::string> filter_set);
 
    private:
        static void LoadDataFromPdb(const std::wstring& /*path*/, IDiaDataSource** /*dia_source*/, IDiaSession**
                                    /*session*/, IDiaSymbol** /*symbol*/);
 
        void DumpStructs(IDiaSymbol* /*g_symbol*/);
        void DumpFunctions(IDiaSymbol* /*g_symbol*/);
        void DumpGlobalVariables(IDiaSymbol* /*g_symbol*/);
        void DumpType(IDiaSymbol* /*symbol*/, const std::string& /*structure*/, int /*indent*/) const;
        void DumpData(IDiaSymbol* /*symbol*/, const std::string& /*structure*/) const;
 
        bool FilterSymbols(const std::string input);
        static std::string GetSymbolNameString(IDiaSymbol* /*symbol*/);
        static uint32_t GetSymbolId(IDiaSymbol* /*symbol*/);
        static void Cleanup(IDiaSymbol* /*symbol*/, IDiaSession* /*session*/, IDiaDataSource* /*source*/);
        static std::string GetFunctionSymbolParams(IDiaSymbol* /*symbol*/);
 
        std::unordered_map<std::string, intptr_t>* offsets_dump_{nullptr};
        std::unordered_map<std::string, BitField>* bitfields_dump_{nullptr};
 
        std::unordered_set<std::string> filter_set_;
        std::unordered_set<uint32_t> visited_;
    };
} // namespace API