aboutsummaryrefslogtreecommitdiffstatshomepage
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--include/pdb/symbol.hpp26
1 files changed, 24 insertions, 2 deletions
diff --git a/include/pdb/symbol.hpp b/include/pdb/symbol.hpp
index 8cb66d1..12061bd 100644
--- a/include/pdb/symbol.hpp
+++ b/include/pdb/symbol.hpp
@@ -5,13 +5,18 @@
#pragma once
+#include "address.hpp"
#include "module.hpp"
+#include <safeint.h>
+
#include <Windows.h>
#include <DbgHelp.h>
+#include <cstddef>
#include <cstring>
+#include <stdexcept>
#include <string>
namespace pdb
@@ -31,7 +36,9 @@ namespace pdb
explicit SymbolInfo(const Raw& raw)
: SymbolInfo{}
{
- std::memcpy(buffer, &raw, raw.SizeOfStruct + raw.NameLen - 1);
+ if (raw.SizeOfStruct != sizeof(raw))
+ throw std::runtime_error{"unexpected symbol structure size"};
+ std::memcpy(buffer, &raw, calc_size(raw));
}
explicit operator Raw&() { return raw; }
@@ -66,7 +73,22 @@ namespace pdb
bool is_function() const { return get_type() == Type::Function; }
private:
- unsigned char buffer[sizeof(Raw) + MAX_SYM_NAME - 1];
+ static std::size_t calc_size(const Raw& raw)
+ {
+ try
+ {
+ msl::utilities::SafeInt<std::size_t> size{raw.SizeOfStruct};
+ size += raw.NameLen;
+ size -= 1;
+ return size;
+ }
+ catch (const msl::utilities::SafeIntException&)
+ {
+ throw std::runtime_error{"symbol name is too long"};
+ }
+ }
+
+ unsigned char buffer[sizeof(Raw) + MAX_SYM_NAME - 1] = {0};
Address displacement = 0;
protected: