mirror of
https://github.com/NohamR/Reclass.git
synced 2026-05-10 19:59:21 +00:00
1d7d384b93
Replace DIA SDK COM-based PDB importer with RawPDB (MolecularMatters) which reads PDB files directly via memory-mapped I/O. Adds File menu "Import PDB..." dialog with type filtering, selection, and progress. - Vendor raw_pdb into third_party/ - Two-phase API: enumeratePdbTypes() + importPdbSelected() - Full recursive import of structs/unions/arrays/pointers/bitfields - PDB import dialog with name filter, select-all, type count - Benchmark: 1654 types from ntkrnlmp.pdb in 16ms - Reorganize import/export files into src/imports/
69 lines
2.5 KiB
C++
69 lines
2.5 KiB
C++
// Copyright 2011-2022, Molecular Matters GmbH <office@molecular-matters.com>
|
|
// See LICENSE.txt for licensing details (2-clause BSD License: https://opensource.org/licenses/BSD-2-Clause)
|
|
|
|
#include "PDB_PCH.h"
|
|
#include "PDB_SourceFileStream.h"
|
|
|
|
|
|
// ------------------------------------------------------------------------------------------------
|
|
// ------------------------------------------------------------------------------------------------
|
|
PDB::SourceFileStream::SourceFileStream(void) PDB_NO_EXCEPT
|
|
: m_stream()
|
|
, m_moduleCount(0u)
|
|
, m_moduleIndices(nullptr)
|
|
, m_moduleFileCounts(nullptr)
|
|
, m_fileNameOffsets(nullptr)
|
|
, m_stringTable(nullptr)
|
|
{
|
|
}
|
|
|
|
|
|
// ------------------------------------------------------------------------------------------------
|
|
// ------------------------------------------------------------------------------------------------
|
|
PDB::SourceFileStream::SourceFileStream(const DirectMSFStream& directStream, uint32_t size, uint32_t offset) PDB_NO_EXCEPT
|
|
: m_stream(directStream, size, offset)
|
|
, m_moduleCount(0u)
|
|
, m_moduleIndices(nullptr)
|
|
, m_moduleFileCounts(nullptr)
|
|
, m_fileNameOffsets(nullptr)
|
|
, m_stringTable(nullptr)
|
|
{
|
|
// we are going to consume the whole source info sub-stream, so create a coalesced stream for faster read operations and direct access.
|
|
// the sub-stream has the following layout:
|
|
// struct SourceInfoSubstream
|
|
// {
|
|
// uint16_t moduleCount;
|
|
// uint16_t sourceFileCount;
|
|
// uint16_t moduleIndices[moduleCount];
|
|
// uint16_t moduleFileCounts[moduleCount];
|
|
// uint32_t fileNameOffsets[realSourceFileCount];
|
|
// char stringTable[][realSourceFileCount];
|
|
// };
|
|
m_moduleCount = *m_stream.GetDataAtOffset<uint16_t>(0u);
|
|
size_t readOffset = sizeof(uint16_t);
|
|
|
|
// skip number of source files. this would only support 64k unique files and is no longer used.
|
|
// the number of source files is computed dynamically instead.
|
|
readOffset += sizeof(uint16_t);
|
|
|
|
// grab direct pointers into the stream data
|
|
m_moduleIndices = m_stream.GetDataAtOffset<uint16_t>(readOffset);
|
|
readOffset += sizeof(uint16_t) * m_moduleCount;
|
|
|
|
m_moduleFileCounts = m_stream.GetDataAtOffset<uint16_t>(readOffset);
|
|
readOffset += sizeof(uint16_t) * m_moduleCount;
|
|
|
|
// count the actual number of source files
|
|
size_t sourceFileCount = 0u;
|
|
for (unsigned int i = 0u; i < m_moduleCount; ++i)
|
|
{
|
|
sourceFileCount += m_moduleFileCounts[i];
|
|
}
|
|
|
|
m_fileNameOffsets = m_stream.GetDataAtOffset<uint32_t>(readOffset);
|
|
readOffset += sizeof(uint32_t) * sourceFileCount;
|
|
|
|
// grab a pointer into the string table
|
|
m_stringTable = m_stream.GetDataAtOffset<char>(readOffset);
|
|
}
|