mirror of
https://github.com/JakeHillion/drgn.git
synced 2024-12-24 02:03:05 +00:00
63 lines
1.4 KiB
C
63 lines
1.4 KiB
C
|
// Copyright 2018-2019 - Omar Sandoval
|
||
|
// SPDX-License-Identifier: GPL-3.0+
|
||
|
|
||
|
#include "internal.h"
|
||
|
|
||
|
/* This definition was added to elf.h in glibc 2.22. */
|
||
|
#ifndef SHF_COMPRESSED
|
||
|
#define SHF_COMPRESSED (1 << 11)
|
||
|
#endif
|
||
|
|
||
|
/*
|
||
|
* glibc added reallocarray() in 2.26, but since it's so trivial, it's easier to
|
||
|
* duplicate it here than it is to do feature detection.
|
||
|
*/
|
||
|
void *realloc_array(void *ptr, size_t nmemb, size_t size)
|
||
|
{
|
||
|
size_t bytes;
|
||
|
|
||
|
if (__builtin_mul_overflow(nmemb, size, &bytes)) {
|
||
|
errno = ENOMEM;
|
||
|
return NULL;
|
||
|
}
|
||
|
return realloc(ptr, bytes);
|
||
|
}
|
||
|
|
||
|
void *malloc_array(size_t nmemb, size_t size)
|
||
|
{
|
||
|
size_t bytes;
|
||
|
|
||
|
if (__builtin_mul_overflow(nmemb, size, &bytes)) {
|
||
|
errno = ENOMEM;
|
||
|
return NULL;
|
||
|
}
|
||
|
return malloc(bytes);
|
||
|
}
|
||
|
|
||
|
struct drgn_error *read_elf_section(Elf_Scn *scn, Elf_Data **ret)
|
||
|
{
|
||
|
GElf_Shdr shdr_mem, *shdr;
|
||
|
Elf_Data *data;
|
||
|
|
||
|
shdr = gelf_getshdr(scn, &shdr_mem);
|
||
|
if (!shdr)
|
||
|
return drgn_error_libelf();
|
||
|
if (shdr->sh_flags & SHF_COMPRESSED) {
|
||
|
#if _ELFUTILS_PREREQ(0, 165)
|
||
|
if (elf_compress(scn, 0, 0) < 0)
|
||
|
return drgn_error_libelf();
|
||
|
shdr = gelf_getshdr(scn, &shdr_mem);
|
||
|
if (!shdr)
|
||
|
return drgn_error_libelf();
|
||
|
#else
|
||
|
return drgn_error_create(DRGN_ERROR_ELF_FORMAT,
|
||
|
"section is compressed but libelf does not support decompression");
|
||
|
#endif
|
||
|
}
|
||
|
data = elf_getdata(scn, NULL);
|
||
|
if (!data)
|
||
|
return drgn_error_libelf();
|
||
|
*ret = data;
|
||
|
return NULL;
|
||
|
}
|