You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

158 lines
3.8 KiB

/* Test program for extracting ELF Note headers and getting whole notes.
Copyright (C) 2019 Red Hat, Inc.
This file is part of elfutils.
This file is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 3 of the License, or
(at your option) any later version.
elfutils is distributed in the hope that it will be useful, but
WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>. */
#ifdef HAVE_CONFIG_H
# include <config.h>
#endif
#include <errno.h>
#include <fcntl.h>
#include <inttypes.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include ELFUTILS_HEADER(elf)
#include <gelf.h>
int
main (int argc, char *argv[])
{
if (argc != 2)
{
printf ("No ELF file given as argument\n");
exit (1);
}
const char *fname = argv[1];
// Initialize libelf.
elf_version (EV_CURRENT);
/* Read the ELF from disk now. */
int fd = open (fname, O_RDONLY);
if (fd == -1)
{
printf ("cannot open '%s': %s\n", fname, strerror (errno));
exit (1);
}
Elf *elf = elf_begin (fd, ELF_C_READ, NULL);
if (elf == NULL)
{
printf ("cannot create ELF descriptor: %s\n", elf_errmsg (-1));
exit (1);
}
GElf_Ehdr ehdr;
if (gelf_getehdr (elf, &ehdr) == NULL)
{
printf ("cannot get Ehdr: %s\n", elf_errmsg (-1));
exit (1);
}
/* Search for all SHT_NOTE sections. */
Elf_Scn *scn = NULL;
while ((scn = elf_nextscn (elf, scn)) != NULL)
{
/* Get the header. */
GElf_Shdr shdr;
if (gelf_getshdr (scn, &shdr) == NULL)
{
printf ("couldn't get shdr: %s\n", elf_errmsg (-1));
exit (1);
}
if (shdr.sh_type == SHT_NOTE)
{
printf ("Notes in section %zd:\n", elf_ndxscn (scn));
Elf_Data *raw = elf_rawdata (scn, NULL);
if (raw == NULL)
{
printf ("couldn't get raw data: %s\n", elf_errmsg (-1));
exit (1);
}
Elf_Data *data = elf_getdata (scn, NULL);
if (data == NULL)
{
printf ("couldn't get data: %s\n", elf_errmsg (-1));
exit (1);
}
size_t off = 0;
size_t next;
GElf_Nhdr nhdr;
size_t n_off;
size_t d_off;
while ((next = gelf_getnote (data, off, &nhdr, &n_off, &d_off)) > 0)
{
/* Now just get the note header "raw" (don't
copy/translate the note data). This only handles
traditional GNU ELF Notes, so we still use the next
from gelf_getnote (padding is different for new style
ELF_T_NHDR8 notes). */
Elf32_Nhdr nh;
Elf_Data src =
{
.d_version = EV_CURRENT, .d_type = ELF_T_NHDR,
.d_size = sizeof nh
};
Elf_Data dst = src;
src.d_buf = raw->d_buf + off;
dst.d_buf = &nh;
if (elf32_xlatetom (&dst, &src, ehdr.e_ident[EI_DATA]) == NULL)
{
printf ("couldn't xlate note: %s\n", elf_errmsg (-1));
exit (1);
}
printf ("type: %" PRId32 ",%" PRId32
", namesz: %" PRId32 ",%" PRId32
", descsz: %" PRId32 ",%" PRId32 "\n",
nhdr.n_type, nh.n_type,
nhdr.n_namesz, nh.n_namesz,
nhdr.n_descsz, nh.n_descsz);
if (nhdr.n_type != nh.n_type
|| nhdr.n_namesz != nh.n_namesz
|| nhdr.n_descsz != nh.n_descsz)
{
printf ("Nhdrs not equal!\n");
exit (1);
}
off = next;
}
}
}
if (elf_end (elf) != 0)
{
printf ("failure in elf_end: %s\n", elf_errmsg (-1));
exit (1);
}
close (fd);
return 0;
}