tests/elfshphehdr.c

   1 /* Test program for adding section and program headers and ehdr updates.
   2    Copyright (C) 2015 Red Hat, Inc.
   3    This file is part of elfutils.
   4
   5    This file is free software; you can redistribute it and/or modify
   6    it under the terms of the GNU General Public License as published by
   7    the Free Software Foundation; either version 3 of the License, or
   8    (at your option) any later version.
   9
  10    elfutils is distributed in the hope that it will be useful, but
  11    WITHOUT ANY WARRANTY; without even the implied warranty of
  12    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13    GNU General Public License for more details.
  14
  15    You should have received a copy of the GNU General Public License
  16    along with this program.  If not, see <http://www.gnu.org/licenses/>.  */
  17
  18 #include <config.h>
  19 #include <assert.h>
  20 #include ELFUTILS_HEADER(elf)
  21 #include <gelf.h>
  22
  23 #include <stdio.h>
  24 #include <stdlib.h>
  25
  26 #include <sys/types.h>
  27 #include <sys/stat.h>
  28 #include <fcntl.h>
  29 #include <unistd.h>
  30
  31 #include <stdbool.h>
  32
  33 void
  34 check (const char *msg, bool statement)
  35 {
  36   if (! statement)
  37     {
  38       fprintf (stderr, "%s FAILED\n", msg);
  39       exit (-1);
  40     }
  41   else
  42     fprintf (stderr, "%s OK\n", msg);
  43 }
  44
  45 void
  46 check_elf (const char *msg, bool statement)
  47 {
  48   if (! statement)
  49     {
  50       fprintf (stderr, "%s: %s\n", msg, elf_errmsg (-1));
  51       exit (-1);
  52     }
  53   else
  54     fprintf (stderr, "%s OK\n", msg);
  55 }
  56
  57 void
  58 test (Elf *elf, int class, bool layout)
  59 {
  60   fprintf (stderr, "testing ELF class: %d, layout: %d\n", class, layout);
  61
  62   check_elf ("gelf_newehdr", gelf_newehdr (elf, class) != 0);
  63   check_elf ("gelf_getclass", gelf_getclass (elf) == class);
  64
  65   check_elf ("elf_flagelf", elf_flagelf (elf, layout ? ELF_C_SET : ELF_C_CLR,
  66                                          ELF_F_LAYOUT) != 0);
  67
  68   GElf_Ehdr ehdr;
  69   check_elf ("gelf_getehdr", gelf_getehdr (elf, &ehdr) != NULL);
  70   check ("e_shnum == 0", ehdr.e_shnum == 0);
  71   check ("e_phnum == 0", ehdr.e_phnum == 0);
  72   check ("e_shoff == 0", ehdr.e_shoff == 0);
  73   check ("e_phoff == 0", ehdr.e_phoff == 0);
  74
  75   size_t shnum;
  76   check_elf ("elf_getshdrnum", elf_getshdrnum (elf, &shnum) == 0);
  77   check ("shnum == 0", shnum == 0);
  78
  79   size_t phnum;
  80   check_elf ("elf_getphdrnum", elf_getphdrnum (elf, &phnum) == 0);
  81   check ("phnum == 0", phnum == 0);
  82
  83   /* Lets fill in some info we are always responsible for.  */
  84   ehdr.e_ident[EI_DATA] = ELFDATANONE; /* Ask for native encoding.  */
  85   ehdr.e_type = ET_EXEC;
  86   ehdr.e_machine = EM_386;
  87   ehdr.e_version = EV_NONE; /* Ask for current version. */
  88   check_elf ("gelf_update_ehdr", gelf_update_ehdr (elf, &ehdr) != 0);
  89
  90   check_elf ("elf_update", elf_update (elf, ELF_C_NULL) > 0);
  91
  92   check_elf ("gelf_getehdr", gelf_getehdr (elf, &ehdr) != NULL);
  93   check ("EI_DATA", ehdr.e_ident[EI_DATA] != ELFDATANONE);
  94   check ("e_version", ehdr.e_version == EV_CURRENT);
  95
  96   /* The sh/ph values shouldn't have changed.  */
  97   check ("e_shnum == 0", ehdr.e_shnum == 0);
  98   check ("e_phnum == 0", ehdr.e_phnum == 0);
  99   check ("e_shoff == 0", ehdr.e_shoff == 0);
 100   check ("e_phoff == 0", ehdr.e_phoff == 0);
 101
 102   check_elf ("elf_getshdrnum", elf_getshdrnum (elf, &shnum) == 0);
 103   check ("shnum == 0", shnum == 0);
 104
 105   check_elf ("elf_getphdrnum", elf_getphdrnum (elf, &phnum) == 0);
 106   check ("phnum == 0", phnum == 0);
 107
 108   /* Lets add a header.  */
 109   check_elf ("elf_newscn", elf_newscn (elf) != NULL);
 110   check_elf ("gelf_newphdr", gelf_newphdr (elf, 1) != 0);
 111
 112   /* If we are responsible for the layout ourselves we should also
 113      tell where to put them.  */
 114   if (layout)
 115     {
 116       check_elf ("gelf_getehdr", gelf_getehdr (elf, &ehdr) != NULL);
 117       /* phdrs go right after the ehdr.  */
 118       ehdr.e_phoff = ehdr.e_ehsize;
 119       /* shdrs go right after the phdrs.  */
 120       ehdr.e_shoff = ehdr.e_phoff + ehdr.e_phnum * ehdr.e_phentsize;
 121       check_elf ("gelf_update_ehdr", gelf_update_ehdr (elf, &ehdr) != 0);
 122     }
 123
 124   check_elf ("elf_update", elf_update (elf, ELF_C_NULL) > 0);
 125
 126   check_elf ("elf_getshdrnum", elf_getshdrnum (elf, &shnum) == 0);
 127   check ("shnum == 1", shnum == 2); /* section zero is also created.  */
 128
 129   check_elf ("elf_getphdrnum", elf_getphdrnum (elf, &phnum) != 0);
 130   check ("phnum == 1", phnum == 1);
 131
 132   check_elf ("gelf_getehdr", gelf_getehdr (elf, &ehdr) != NULL);
 133
 134   check ("EI_DATA", ehdr.e_ident[EI_DATA] != ELFDATANONE);
 135   check ("e_version", ehdr.e_version == EV_CURRENT);
 136
 137   check ("e_shnum == 2", ehdr.e_shnum == 2);
 138   check ("e_phnum == 1", ehdr.e_phnum == 1);
 139   check ("e_shoff != 0", ehdr.e_shoff != 0);
 140   check ("e_phoff != 0", ehdr.e_phoff != 0);
 141
 142   size_t shentsize = (class == ELFCLASS32
 143                       ? sizeof (Elf32_Shdr) : sizeof (Elf64_Shdr));
 144   check ("e_shentsize", ehdr.e_shentsize == shentsize);
 145   size_t phentsize = (class == ELFCLASS32
 146                       ? sizeof (Elf32_Phdr) : sizeof (Elf64_Phdr));
 147   check ("e_phentsize", ehdr.e_phentsize == phentsize);
 148 }
 149
 150 int
 151 main (int argc __attribute__ ((unused)), char **argv __attribute ((unused)))
 152 {
 153   elf_version (EV_CURRENT);
 154
 155   int fd = fd = open("/dev/zero", O_WRONLY);
 156   check ("open", fd >= 0);
 157
 158   Elf *elf;
 159
 160   elf = elf_begin (fd, ELF_C_WRITE, NULL);
 161   check_elf ("elf_begin", elf != NULL);
 162   test (elf, ELFCLASS32, false);
 163   elf_end (elf);
 164
 165   elf = elf_begin (fd, ELF_C_WRITE, NULL);
 166   check_elf ("elf_begin", elf != NULL);
 167   test (elf, ELFCLASS32, true);
 168   elf_end (elf);
 169
 170   elf = elf_begin (fd, ELF_C_WRITE, NULL);
 171   check_elf ("elf_begin", elf != NULL);
 172   test (elf, ELFCLASS64, false);
 173   elf_end (elf);
 174
 175   elf = elf_begin (fd, ELF_C_WRITE, NULL);
 176   check_elf ("elf_begin", elf != NULL);
 177   test (elf, ELFCLASS64, true);
 178   elf_end (elf);
 179
 180   close (fd);
 181   return 0;
 182 }