Blame tests/update2.c

Packit Service 97d2fb
/* Test program for elf_update function.
Packit Service 97d2fb
   Copyright (C) 2000, 2002, 2005 Red Hat, Inc.
Packit Service 97d2fb
   This file is part of elfutils.
Packit Service 97d2fb
   Written by Ulrich Drepper <drepper@redhat.com>, 2000.
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
#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 <libelf.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
Packit Service 97d2fb
Packit Service 97d2fb
int
Packit Service 97d2fb
main (int argc, char *argv[] __attribute__ ((unused)))
Packit Service 97d2fb
{
Packit Service 97d2fb
  const char *fname = "xxx_update2";
Packit Service 97d2fb
  int fd;
Packit Service 97d2fb
  Elf *elf;
Packit Service 97d2fb
  Elf32_Ehdr *ehdr;
Packit Service 97d2fb
  Elf32_Phdr *phdr;
Packit Service 97d2fb
  int i;
Packit Service 97d2fb
Packit Service 97d2fb
  fd = open (fname, O_RDWR | O_CREAT | O_TRUNC, 0666);
Packit Service 97d2fb
  if (fd == -1)
Packit Service 97d2fb
    {
Packit Service 97d2fb
      printf ("cannot open `%s': %s\n", fname, strerror (errno));
Packit Service 97d2fb
      exit (1);
Packit Service 97d2fb
    }
Packit Service 97d2fb
Packit Service 97d2fb
  elf_version (EV_CURRENT);
Packit Service 97d2fb
Packit Service 97d2fb
  elf = elf_begin (fd, ELF_C_WRITE, NULL);
Packit Service 97d2fb
  if (elf == NULL)
Packit Service 97d2fb
    {
Packit Service 97d2fb
      printf ("cannot create ELF descriptor: %s\n", elf_errmsg (-1));
Packit Service 97d2fb
      exit (1);
Packit Service 97d2fb
    }
Packit Service 97d2fb
Packit Service 97d2fb
  /* Create an ELF header.  */
Packit Service 97d2fb
  ehdr = elf32_newehdr (elf);
Packit Service 97d2fb
  if (ehdr == NULL)
Packit Service 97d2fb
    {
Packit Service 97d2fb
      printf ("cannot create ELF header: %s\n", elf_errmsg (-1));
Packit Service 97d2fb
      exit (1);
Packit Service 97d2fb
    }
Packit Service 97d2fb
Packit Service 97d2fb
  /* Print the ELF header values.  */
Packit Service 97d2fb
  if (argc > 1)
Packit Service 97d2fb
    {
Packit Service 97d2fb
      for (i = 0; i < EI_NIDENT; ++i)
Packit Service 97d2fb
	printf (" %02x", ehdr->e_ident[i]);
Packit Service 97d2fb
      printf ("\
Packit Service 97d2fb
\ntype = %hu\nmachine = %hu\nversion = %u\nentry = %u\nphoff = %u\n"
Packit Service 97d2fb
	      "shoff = %u\nflags = %u\nehsize = %hu\nphentsize = %hu\n"
Packit Service 97d2fb
	      "phnum = %hu\nshentsize = %hu\nshnum = %hu\nshstrndx = %hu\n",
Packit Service 97d2fb
	      ehdr->e_type, ehdr->e_machine, ehdr->e_version, ehdr->e_entry,
Packit Service 97d2fb
	      ehdr->e_phoff, ehdr->e_shoff, ehdr->e_flags, ehdr->e_ehsize,
Packit Service 97d2fb
	      ehdr->e_phentsize, ehdr->e_phnum, ehdr->e_shentsize,
Packit Service 97d2fb
	      ehdr->e_shnum, ehdr->e_shstrndx);
Packit Service 97d2fb
    }
Packit Service 97d2fb
Packit Service 97d2fb
  ehdr->e_ident[0] = 42;
Packit Service 97d2fb
  ehdr->e_ident[4] = 1;
Packit Service 97d2fb
  ehdr->e_ident[5] = 1;
Packit Service 97d2fb
  ehdr->e_ident[6] = 2;
Packit Service 97d2fb
  ehdr->e_type = ET_EXEC;
Packit Service 97d2fb
  ehdr->e_version = 1;
Packit Service 97d2fb
  ehdr->e_ehsize = 1;
Packit Service 97d2fb
  elf_flagehdr (elf, ELF_C_SET, ELF_F_DIRTY);
Packit Service 97d2fb
Packit Service 97d2fb
  /* Create the program header.  */
Packit Service 97d2fb
  phdr = elf32_newphdr (elf, 1);
Packit Service 97d2fb
  if (phdr == NULL)
Packit Service 97d2fb
    {
Packit Service 97d2fb
      printf ("cannot create program header: %s\n", elf_errmsg (-1));
Packit Service 97d2fb
      exit (1);
Packit Service 97d2fb
    }
Packit Service 97d2fb
Packit Service 97d2fb
  phdr[0].p_type = PT_PHDR;
Packit Service 97d2fb
  elf_flagphdr (elf, ELF_C_SET, ELF_F_DIRTY);
Packit Service 97d2fb
Packit Service 97d2fb
  /* Let the library compute the internal structure information.  */
Packit Service 97d2fb
  if (elf_update (elf, ELF_C_NULL) < 0)
Packit Service 97d2fb
    {
Packit Service 97d2fb
      printf ("failure in elf_update(NULL): %s\n", elf_errmsg (-1));
Packit Service 97d2fb
      exit (1);
Packit Service 97d2fb
    }
Packit Service 97d2fb
Packit Service 97d2fb
  ehdr = elf32_getehdr (elf);
Packit Service 97d2fb
Packit Service 97d2fb
  phdr[0].p_offset = ehdr->e_phoff;
Packit Service 97d2fb
  phdr[0].p_offset = ehdr->e_phoff;
Packit Service 97d2fb
  phdr[0].p_vaddr = ehdr->e_phoff;
Packit Service 97d2fb
  phdr[0].p_paddr = ehdr->e_phoff;
Packit Service 97d2fb
  phdr[0].p_flags = PF_R | PF_X;
Packit Service 97d2fb
  phdr[0].p_filesz = ehdr->e_phnum * elf32_fsize (ELF_T_PHDR, 1, EV_CURRENT);
Packit Service 97d2fb
  phdr[0].p_memsz = ehdr->e_phnum * elf32_fsize (ELF_T_PHDR, 1, EV_CURRENT);
Packit Service 97d2fb
  phdr[0].p_align = sizeof (Elf32_Word);
Packit Service 97d2fb
Packit Service 97d2fb
  /* Write out the file.  */
Packit Service 97d2fb
  if (elf_update (elf, ELF_C_WRITE) < 0)
Packit Service 97d2fb
    {
Packit Service 97d2fb
      printf ("failure in elf_update(WRITE): %s\n", elf_errmsg (-1));
Packit Service 97d2fb
      exit (1);
Packit Service 97d2fb
    }
Packit Service 97d2fb
Packit Service 97d2fb
  /* Print the ELF header values.  */
Packit Service 97d2fb
  if (argc > 1)
Packit Service 97d2fb
    {
Packit Service 97d2fb
      for (i = 0; i < EI_NIDENT; ++i)
Packit Service 97d2fb
	printf (" %02x", ehdr->e_ident[i]);
Packit Service 97d2fb
      printf ("\
Packit Service 97d2fb
\ntype = %hu\nmachine = %hu\nversion = %u\nentry = %u\nphoff = %u\n"
Packit Service 97d2fb
	      "shoff = %u\nflags = %u\nehsize = %hu\nphentsize = %hu\n"
Packit Service 97d2fb
	      "phnum = %hu\nshentsize = %hu\nshnum = %hu\nshstrndx = %hu\n",
Packit Service 97d2fb
	      ehdr->e_type, ehdr->e_machine, ehdr->e_version, ehdr->e_entry,
Packit Service 97d2fb
	      ehdr->e_phoff, ehdr->e_shoff, ehdr->e_flags, ehdr->e_ehsize,
Packit Service 97d2fb
	      ehdr->e_phentsize, ehdr->e_phnum, ehdr->e_shentsize,
Packit Service 97d2fb
	      ehdr->e_shnum, ehdr->e_shstrndx);
Packit Service 97d2fb
    }
Packit Service 97d2fb
Packit Service 97d2fb
  if (elf_end (elf) != 0)
Packit Service 97d2fb
    {
Packit Service 97d2fb
      printf ("failure in elf_end: %s\n", elf_errmsg (-1));
Packit Service 97d2fb
      exit (1);
Packit Service 97d2fb
    }
Packit Service 97d2fb
Packit Service 97d2fb
  unlink (fname);
Packit Service 97d2fb
Packit Service 97d2fb
  return 0;
Packit Service 97d2fb
}