Blame tests/addsections.c

Packit Service 97d2fb
/* Test program for adding (more than SHN_LORESERVE) sections.
Packit Service 97d2fb
   Copyright (C) 2018 Red Hat, Inc.
Packit Service 97d2fb
   This file is part of elfutils.
Packit Service 97d2fb
Packit Service 97d2fb
   This file is free software; you can redistribute it and/or modify
Packit Service 97d2fb
   it under the terms of the GNU General Public License as published by
Packit Service 97d2fb
   the Free Software Foundation; either version 3 of the License, or
Packit Service 97d2fb
   (at your option) any later version.
Packit Service 97d2fb
Packit Service 97d2fb
   elfutils is distributed in the hope that it will be useful, but
Packit Service 97d2fb
   WITHOUT ANY WARRANTY; without even the implied warranty of
Packit Service 97d2fb
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
Packit Service 97d2fb
   GNU General Public License for more details.
Packit Service 97d2fb
Packit Service 97d2fb
   You should have received a copy of the GNU General Public License
Packit Service 97d2fb
   along with this program.  If not, see <http://www.gnu.org/licenses/>.  */
Packit Service 97d2fb
Packit Service 97d2fb
Packit Service 97d2fb
#ifdef HAVE_CONFIG_H
Packit Service 97d2fb
# include <config.h>
Packit Service 97d2fb
#endif
Packit Service 97d2fb
Packit Service 97d2fb
#include <errno.h>
Packit Service 97d2fb
#include <fcntl.h>
Packit Service 97d2fb
#include <inttypes.h>
Packit Service 97d2fb
#include <stdbool.h>
Packit Service 97d2fb
#include <stdio.h>
Packit Service 97d2fb
#include <stdlib.h>
Packit Service 97d2fb
#include <string.h>
Packit Service 97d2fb
#include <unistd.h>
Packit Service 97d2fb
#include <sys/types.h>
Packit Service 97d2fb
#include <sys/stat.h>
Packit Service 97d2fb
Packit Service 97d2fb
#include ELFUTILS_HEADER(elf)
Packit Service 97d2fb
#include <gelf.h>
Packit Service 97d2fb
Packit Service 97d2fb
Packit Service 97d2fb
/* shstrndx is special, might overflow into section zero header sh_link.  */
Packit Service 97d2fb
static int
Packit Service 97d2fb
setshstrndx (Elf *elf, size_t ndx)
Packit Service 97d2fb
{
Packit Service 97d2fb
  printf ("setshstrndx: %zd\n", ndx);
Packit Service 97d2fb
Packit Service 97d2fb
  GElf_Ehdr ehdr_mem;
Packit Service 97d2fb
  GElf_Ehdr *ehdr = gelf_getehdr (elf, &ehdr_mem);
Packit Service 97d2fb
  if (ehdr == NULL)
Packit Service 97d2fb
    return -1;
Packit Service 97d2fb
Packit Service 97d2fb
  if (ndx < SHN_LORESERVE)
Packit Service 97d2fb
    ehdr->e_shstrndx = ndx;
Packit Service 97d2fb
  else
Packit Service 97d2fb
    {
Packit Service 97d2fb
      ehdr->e_shstrndx = SHN_XINDEX;
Packit Service 97d2fb
      Elf_Scn *zscn = elf_getscn (elf, 0);
Packit Service 97d2fb
      GElf_Shdr zshdr_mem;
Packit Service 97d2fb
      GElf_Shdr *zshdr = gelf_getshdr (zscn, &zshdr_mem);
Packit Service 97d2fb
      if (zshdr == NULL)
Packit Service 97d2fb
	return -1;
Packit Service 97d2fb
      zshdr->sh_link = ndx;
Packit Service 97d2fb
      if (gelf_update_shdr (zscn, zshdr) == 0)
Packit Service 97d2fb
	return -1;
Packit Service 97d2fb
    }
Packit Service 97d2fb
Packit Service 97d2fb
  if (gelf_update_ehdr (elf, ehdr) == 0)
Packit Service 97d2fb
    return -1;
Packit Service 97d2fb
Packit Service 97d2fb
  return 0;
Packit Service 97d2fb
}
Packit Service 97d2fb
Packit Service 97d2fb
/* Will add nr new '.extra' sections and a new '.new_shstrtab' section
Packit Service 97d2fb
   at the end.  */
Packit Service 97d2fb
static void
Packit Service 97d2fb
add_sections (const char *name, size_t nr, int use_mmap, size_t sec_size)
Packit Service 97d2fb
{
Packit Service 97d2fb
  printf ("add_sections '%s': %zd (sec_size: %zd)\n", name, nr, sec_size);
Packit Service 97d2fb
Packit Service 97d2fb
  int fd = open (name, O_RDWR);
Packit Service 97d2fb
  if (fd < 0)
Packit Service 97d2fb
    {
Packit Service 97d2fb
      fprintf (stderr, "Couldn't open file '%s': %s\n",
Packit Service 97d2fb
	       name, strerror (errno));
Packit Service 97d2fb
      exit (1);
Packit Service 97d2fb
    }
Packit Service 97d2fb
Packit Service 97d2fb
  Elf *elf = elf_begin (fd, use_mmap ? ELF_C_RDWR_MMAP : ELF_C_RDWR, NULL);
Packit Service 97d2fb
  if (elf == NULL)
Packit Service 97d2fb
    {
Packit Service 97d2fb
      fprintf (stderr, "Couldn't open ELF file '%s': %s\n",
Packit Service 97d2fb
	       name, elf_errmsg (-1));
Packit Service 97d2fb
      exit (1);
Packit Service 97d2fb
    }
Packit Service 97d2fb
Packit Service 97d2fb
  /* We will add a new shstrtab section with two new names at the end.
Packit Service 97d2fb
     Just get the current shstrtab table and add two entries '.extra'
Packit Service 97d2fb
     and '.old_shstrtab' at the end of the table, so all existing indexes
Packit Service 97d2fb
     are still valid.  */
Packit Service 97d2fb
  size_t shstrndx;
Packit Service 97d2fb
  if (elf_getshdrstrndx (elf, &shstrndx) < 0)
Packit Service 97d2fb
    {
Packit Service 97d2fb
      printf ("cannot get shstrndx: %s\n", elf_errmsg (-1));
Packit Service 97d2fb
      exit (1);
Packit Service 97d2fb
    }
Packit Service 97d2fb
Packit Service 97d2fb
  Elf_Scn *shstrtab_scn = elf_getscn (elf, shstrndx);
Packit Service 97d2fb
  if (shstrtab_scn == NULL)
Packit Service 97d2fb
    {
Packit Service 97d2fb
      printf ("couldn't get shstrtab scn: %s\n", elf_errmsg (-1));
Packit Service 97d2fb
      exit (1);
Packit Service 97d2fb
    }
Packit Service 97d2fb
  Elf_Data *shstrtab_data = elf_getdata (shstrtab_scn, NULL);
Packit Service 97d2fb
  if (shstrtab_data == NULL)
Packit Service 97d2fb
    {
Packit Service 97d2fb
      printf ("couldn't get shstrtab data: %s\n", elf_errmsg (-1));
Packit Service 97d2fb
      exit (1);
Packit Service 97d2fb
    }
Packit Service 97d2fb
  size_t new_shstrtab_size = (shstrtab_data->d_size
Packit Service 97d2fb
			      + strlen (".extra") + 1
Packit Service 97d2fb
			      + strlen (".old_shstrtab") + 1);
Packit Service 97d2fb
  void *new_shstrtab_buf = malloc (new_shstrtab_size);
Packit Service 97d2fb
  if (new_shstrtab_buf == NULL)
Packit Service 97d2fb
    {
Packit Service 97d2fb
      printf ("couldn't allocate new shstrtab data d_buf\n");
Packit Service 97d2fb
      exit (1);
Packit Service 97d2fb
    }
Packit Service 97d2fb
  memcpy (new_shstrtab_buf, shstrtab_data->d_buf, shstrtab_data->d_size);
Packit Service 97d2fb
  size_t extra_idx = shstrtab_data->d_size;
Packit Service 97d2fb
  size_t old_shstrtab_idx = extra_idx + strlen (".extra") + 1;
Packit Service 97d2fb
  strcpy (new_shstrtab_buf + extra_idx, ".extra");
Packit Service 97d2fb
  strcpy (new_shstrtab_buf + old_shstrtab_idx, ".old_shstrtab");
Packit Service 97d2fb
Packit Service 97d2fb
  /* Change the name of the old shstrtab section, because elflint
Packit Service 97d2fb
     has a strict check on the name/type for .shstrtab.  */
Packit Service 97d2fb
  GElf_Shdr shdr_mem;
Packit Service 97d2fb
  GElf_Shdr *shdr = gelf_getshdr (shstrtab_scn, &shdr_mem);
Packit Service 97d2fb
  if (shdr == NULL)
Packit Service 97d2fb
    {
Packit Service 97d2fb
      printf ("cannot get header for old shstrtab section: %s\n",
Packit Service 97d2fb
              elf_errmsg (-1));
Packit Service 97d2fb
      exit (1);
Packit Service 97d2fb
    }
Packit Service 97d2fb
Packit Service 97d2fb
  size_t shstrtab_idx = shdr->sh_name;
Packit Service 97d2fb
  shdr->sh_name = old_shstrtab_idx;
Packit Service 97d2fb
Packit Service 97d2fb
  if (gelf_update_shdr (shstrtab_scn, shdr) == 0)
Packit Service 97d2fb
    {
Packit Service 97d2fb
      printf ("cannot update old shstrtab section header: %s\n",
Packit Service 97d2fb
	      elf_errmsg (-1));
Packit Service 97d2fb
      exit (1);
Packit Service 97d2fb
    }
Packit Service 97d2fb
Packit Service 97d2fb
  void *buf;
Packit Service 97d2fb
  size_t bufsz;
Packit Service 97d2fb
  if (sec_size == 0)
Packit Service 97d2fb
    {
Packit Service 97d2fb
      buf = strdup ("extra");
Packit Service 97d2fb
      bufsz = strlen ("extra") + 1;
Packit Service 97d2fb
    }
Packit Service 97d2fb
  else
Packit Service 97d2fb
    {
Packit Service 97d2fb
      buf = malloc (sec_size);
Packit Service 97d2fb
      if (buf == NULL)
Packit Service 97d2fb
	{
Packit Service 97d2fb
	  printf ("cannot allocate buffer data of %zd bytes\n", sec_size);
Packit Service 97d2fb
	  exit (1);
Packit Service 97d2fb
	}
Packit Service 97d2fb
      memset (buf, 0xAA, sec_size);
Packit Service 97d2fb
      bufsz = sec_size;
Packit Service 97d2fb
    }
Packit Service 97d2fb
Packit Service 97d2fb
  // Add lots of .extra sections...
Packit Service 97d2fb
  size_t cnt = 0;
Packit Service 97d2fb
  while (cnt++ < nr)
Packit Service 97d2fb
    {
Packit Service 97d2fb
      Elf_Scn *scn = elf_newscn (elf);
Packit Service 97d2fb
      if (scn == NULL)
Packit Service 97d2fb
	{
Packit Service 97d2fb
	  printf ("cannot create .extra section (%zd): %s\n", cnt,
Packit Service 97d2fb
		  elf_errmsg (-1));
Packit Service 97d2fb
	  exit (1);
Packit Service 97d2fb
	}
Packit Service 97d2fb
Packit Service 97d2fb
      Elf_Data *data = elf_newdata (scn);
Packit Service 97d2fb
      if (data == NULL)
Packit Service 97d2fb
	{
Packit Service 97d2fb
	  printf ("couldn't create new section data (%zd): %s\n", cnt,
Packit Service 97d2fb
		  elf_errmsg (-1));
Packit Service 97d2fb
	  exit (1);
Packit Service 97d2fb
	}
Packit Service 97d2fb
Packit Service 97d2fb
      data->d_size = bufsz;
Packit Service 97d2fb
      data->d_buf = buf;
Packit Service 97d2fb
      data->d_type = ELF_T_BYTE;
Packit Service 97d2fb
      data->d_align = 1;
Packit Service 97d2fb
Packit Service 97d2fb
      shdr = gelf_getshdr (scn, &shdr_mem);
Packit Service 97d2fb
      if (shdr == NULL)
Packit Service 97d2fb
	{
Packit Service 97d2fb
	  printf ("cannot get header for new section (%zd): %s\n", cnt,
Packit Service 97d2fb
		  elf_errmsg (-1));
Packit Service 97d2fb
	  exit (1);
Packit Service 97d2fb
	}
Packit Service 97d2fb
Packit Service 97d2fb
      shdr->sh_type = SHT_PROGBITS;
Packit Service 97d2fb
      shdr->sh_flags = 0;
Packit Service 97d2fb
      shdr->sh_addr = 0;
Packit Service 97d2fb
      shdr->sh_link = SHN_UNDEF;
Packit Service 97d2fb
      shdr->sh_info = SHN_UNDEF;
Packit Service 97d2fb
      shdr->sh_addralign = 1;
Packit Service 97d2fb
      shdr->sh_entsize = 0;
Packit Service 97d2fb
      shdr->sh_size = data->d_size;
Packit Service 97d2fb
      shdr->sh_name = extra_idx;
Packit Service 97d2fb
Packit Service 97d2fb
      if (gelf_update_shdr (scn, shdr) == 0)
Packit Service 97d2fb
	{
Packit Service 97d2fb
	  printf ("cannot update new section header (%zd): %s\n", cnt,
Packit Service 97d2fb
		  elf_errmsg (-1));
Packit Service 97d2fb
	  exit (1);
Packit Service 97d2fb
	}
Packit Service 97d2fb
    }
Packit Service 97d2fb
Packit Service 97d2fb
  // Create new shstrtab section.
Packit Service 97d2fb
  Elf_Scn *new_shstrtab_scn = elf_newscn (elf);
Packit Service 97d2fb
  if (new_shstrtab_scn == NULL)
Packit Service 97d2fb
    {
Packit Service 97d2fb
      printf ("cannot create new shstrtab section: %s\n", elf_errmsg (-1));
Packit Service 97d2fb
      exit (1);
Packit Service 97d2fb
    }
Packit Service 97d2fb
Packit Service 97d2fb
  Elf_Data *new_shstrtab_data = elf_newdata (new_shstrtab_scn);
Packit Service 97d2fb
  if (new_shstrtab_data == NULL)
Packit Service 97d2fb
    {
Packit Service 97d2fb
      printf ("couldn't create new shstrtab section data: %s\n",
Packit Service 97d2fb
	      elf_errmsg (-1));
Packit Service 97d2fb
      exit (1);
Packit Service 97d2fb
    }
Packit Service 97d2fb
Packit Service 97d2fb
  new_shstrtab_data->d_size = new_shstrtab_size;
Packit Service 97d2fb
  new_shstrtab_data->d_buf = new_shstrtab_buf;
Packit Service 97d2fb
  new_shstrtab_data->d_type = ELF_T_BYTE;
Packit Service 97d2fb
  new_shstrtab_data->d_align = 1;
Packit Service 97d2fb
Packit Service 97d2fb
  shdr = gelf_getshdr (new_shstrtab_scn, &shdr_mem);
Packit Service 97d2fb
  if (shdr == NULL)
Packit Service 97d2fb
    {
Packit Service 97d2fb
      printf ("cannot get header for new shstrtab section: %s\n",
Packit Service 97d2fb
	      elf_errmsg (-1));
Packit Service 97d2fb
      exit (1);
Packit Service 97d2fb
    }
Packit Service 97d2fb
Packit Service 97d2fb
  shdr->sh_type = SHT_STRTAB;
Packit Service 97d2fb
  shdr->sh_flags = 0;
Packit Service 97d2fb
  shdr->sh_addr = 0;
Packit Service 97d2fb
  shdr->sh_link = SHN_UNDEF;
Packit Service 97d2fb
  shdr->sh_info = SHN_UNDEF;
Packit Service 97d2fb
  shdr->sh_addralign = 1;
Packit Service 97d2fb
  shdr->sh_entsize = 0;
Packit Service 97d2fb
  shdr->sh_size = new_shstrtab_size;
Packit Service 97d2fb
  shdr->sh_name = shstrtab_idx;
Packit Service 97d2fb
Packit Service 97d2fb
  // Finished new shstrtab section, update the header.
Packit Service 97d2fb
  if (gelf_update_shdr (new_shstrtab_scn, shdr) == 0)
Packit Service 97d2fb
    {
Packit Service 97d2fb
      printf ("cannot update new shstrtab section header: %s\n",
Packit Service 97d2fb
	      elf_errmsg (-1));
Packit Service 97d2fb
      exit (1);
Packit Service 97d2fb
    }
Packit Service 97d2fb
Packit Service 97d2fb
  // Set it as the new shstrtab section to get the names correct.
Packit Service 97d2fb
  size_t new_shstrndx = elf_ndxscn (new_shstrtab_scn);
Packit Service 97d2fb
  if (setshstrndx (elf, new_shstrndx) < 0)
Packit Service 97d2fb
    {
Packit Service 97d2fb
      printf ("cannot set shstrndx: %s\n", elf_errmsg (-1));
Packit Service 97d2fb
      exit (1);
Packit Service 97d2fb
    }
Packit Service 97d2fb
Packit Service 97d2fb
  // Write everything to disk.
Packit Service 97d2fb
  if (elf_update (elf, ELF_C_WRITE) < 0)
Packit Service 97d2fb
    {
Packit Service 97d2fb
      printf ("failure in elf_update: %s\n", elf_errmsg (-1));
Packit Service 97d2fb
      exit (1);
Packit Service 97d2fb
    }
Packit Service 97d2fb
Packit Service 97d2fb
  if (elf_end (elf) != 0)
Packit Service 97d2fb
    {
Packit Service 97d2fb
      printf ("couldn't cleanup elf '%s': %s\n", name, elf_errmsg (-1));
Packit Service 97d2fb
      exit (1);
Packit Service 97d2fb
    }
Packit Service 97d2fb
Packit Service 97d2fb
  if (close (fd) != 0)
Packit Service 97d2fb
    {
Packit Service 97d2fb
      printf ("couldn't close '%s': %s\n", name, strerror (errno));
Packit Service 97d2fb
      exit (1);
Packit Service 97d2fb
    }
Packit Service 97d2fb
Packit Service 97d2fb
  free (buf);
Packit Service 97d2fb
  free (new_shstrtab_buf);
Packit Service 97d2fb
}
Packit Service 97d2fb
Packit Service 97d2fb
int
Packit Service 97d2fb
main (int argc, char *argv[])
Packit Service 97d2fb
{
Packit Service 97d2fb
  elf_version (EV_CURRENT);
Packit Service 97d2fb
Packit Service 97d2fb
  /* Takes the given file, and adds the given number of sections.
Packit Service 97d2fb
     Optionally using mmap and optionally using a given section size.  */
Packit Service 97d2fb
  if (argc < 3 || argc > 5)
Packit Service 97d2fb
    {
Packit Service 97d2fb
      fprintf (stderr, "addsections [--mmap] nr elf.file [sec_size]\n");
Packit Service 97d2fb
      exit (1);
Packit Service 97d2fb
    }
Packit Service 97d2fb
Packit Service 97d2fb
  int argn = 1;
Packit Service 97d2fb
  bool use_mmap = false;
Packit Service 97d2fb
  if (strcmp (argv[argn], "--mmap") == 0)
Packit Service 97d2fb
    {
Packit Service 97d2fb
      use_mmap = true;
Packit Service 97d2fb
      argn++;
Packit Service 97d2fb
    }
Packit Service 97d2fb
Packit Service 97d2fb
  size_t nr = atoi (argv[argn++]);
Packit Service 97d2fb
  const char *file = argv[argn++];
Packit Service 97d2fb
Packit Service 97d2fb
  size_t sec_size = 0;
Packit Service 97d2fb
  if (argn < argc)
Packit Service 97d2fb
    sec_size = atol (argv[argn++]);
Packit Service 97d2fb
Packit Service 97d2fb
  add_sections (file, nr, use_mmap, sec_size);
Packit Service 97d2fb
Packit Service 97d2fb
  return 0;
Packit Service 97d2fb
}