/* 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;
}