Adding early checks for ELF file

Change-Id: I1dc833fc2b017edae5c5456b14c376e88ca2bef7


[ROCm/rocprofiler commit: 698b6eb641]
This commit is contained in:
Giovanni LB
2024-08-26 17:41:20 -03:00
orang tua 5a6c6a379c
melakukan 3b53fb2c47
2 mengubah file dengan 18 tambahan dan 0 penghapusan
@@ -162,6 +162,15 @@ DisassemblyInstance::DisassemblyInstance(
uint64_t gpu_id
)
{
if (
codeobj_size <= 4 ||
codeobj_data[0] != ELFMAG0 ||
codeobj_data[1] != ELFMAG1 ||
codeobj_data[2] != ELFMAG2 ||
codeobj_data[3] != ELFMAG3
)
throw std::invalid_argument("Invalid ELF file");
buffer = std::vector<char>(codeobj_size, 0);
std::memcpy(buffer.data(), codeobj_data, codeobj_size);