// Copyright (c) 2020 Egor Tensin <Egor.Tensin@gmail.com>
// This file is part of the "PDB repository" project.
// For details, see https://github.com/egor-tensin/pdb-repo.
// Distributed under the MIT License.

#include "pdb/all.hpp"

#include <SafeInt.hpp>

#include <DbgHelp.h>
#include <Windows.h>

#include <cstddef>
#include <cstring>
#include <stdexcept>

namespace pdb {
namespace {

std::size_t calc_size(const SymbolInfo::Impl& impl) {
    try {
        return SafeInt<std::size_t>{impl.SizeOfStruct} + impl.NameLen - 1;
    } catch (const SafeIntException&) {
        throw std::runtime_error{"invalid SYMBOL_INFO size"};
    }
}

unsigned long cast_line_number(DWORD impl) {
    unsigned long dest = 0;

    if (!SafeCast(impl, dest))
        throw std::runtime_error{"invalid line number"};

    return dest;
}

} // namespace

SymbolInfo::SymbolInfo() : buffer{}, impl{*reinterpret_cast<Impl*>(buffer.data())} {
    buffer.fill(0);
    impl.SizeOfStruct = sizeof(Impl);
    impl.MaxNameLen = MAX_SYM_NAME;
}

SymbolInfo::SymbolInfo(const Impl& impl) : SymbolInfo{} {
    if (impl.SizeOfStruct != sizeof(impl))
        throw std::runtime_error{"invalid SYMBOL_INFO.SizeOfStruct"};
    const auto impl_size = calc_size(impl);
    if (impl_size > buffer.size())
        throw std::runtime_error{"SYMBOL_INFO is too large"};
    std::memcpy(buffer.data(), &impl, impl_size);
}

LineInfo::LineInfo(const Impl& impl)
    : file_path{impl.FileName}, line_number{cast_line_number(impl.LineNumber)} {}

} // namespace pdb