tests/update4.c

   1 /* Test program for elf_update function.
   2    Copyright (C) 2000, 2001, 2002, 2005 Red Hat, Inc.
   3    This file is part of elfutils.
   4    Written by Ulrich Drepper <drepper@redhat.com>, 2000.
   5
   6    This file is free software; you can redistribute it and/or modify
   7    it under the terms of the GNU General Public License as published by
   8    the Free Software Foundation; either version 3 of the License, or
   9    (at your option) any later version.
  10
  11    elfutils is distributed in the hope that it will be useful, but
  12    WITHOUT ANY WARRANTY; without even the implied warranty of
  13    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14    GNU General Public License for more details.
  15
  16    You should have received a copy of the GNU General Public License
  17    along with this program.  If not, see <http://www.gnu.org/licenses/>.  */
  18
  19 #ifdef HAVE_CONFIG_H
  20 # include <config.h>
  21 #endif
  22
  23 #include <errno.h>
  24 #include <fcntl.h>
  25 #include <libelf.h>
  26 #include <stdio.h>
  27 #include <stdlib.h>
  28 #include <string.h>
  29 #include <unistd.h>
  30
  31 #include ELFUTILS_HEADER(ebl)
  32
  33
  34 int
  35 main (int argc, char *argv[] __attribute__ ((unused)))
  36 {
  37   const char fname[] = "xxx";
  38   int fd;
  39   Elf *elf;
  40   Elf32_Ehdr *ehdr;
  41   Elf32_Phdr *phdr;
  42   Elf_Scn *scn;
  43   Elf32_Shdr *shdr;
  44   Elf_Data *data;
  45   struct Ebl_Strtab *shst;
  46   struct Ebl_Strent *firstse;
  47   struct Ebl_Strent *secondse;
  48   struct Ebl_Strent *thirdse;
  49   struct Ebl_Strent *fourthse;
  50   struct Ebl_Strent *shstrtabse;
  51   int i;
  52
  53   fd = open (fname, O_RDWR | O_CREAT | O_TRUNC, 0666);
  54   if (fd == -1)
  55     {
  56       printf ("cannot open `%s': %s\n", fname, strerror (errno));
  57       exit (1);
  58     }
  59
  60   elf_version (EV_CURRENT);
  61
  62   elf_fill (0x42);
  63
  64   elf = elf_begin (fd, ELF_C_WRITE, NULL);
  65   if (elf == NULL)
  66     {
  67       printf ("cannot create ELF descriptor: %s\n", elf_errmsg (-1));
  68       exit (1);
  69     }
  70
  71   /* Create an ELF header.  */
  72   ehdr = elf32_newehdr (elf);
  73   if (ehdr == NULL)
  74     {
  75       printf ("cannot create ELF header: %s\n", elf_errmsg (-1));
  76       exit (1);
  77     }
  78
  79   /* Print the ELF header values.  */
  80   if (argc > 1)
  81     {
  82       for (i = 0; i < EI_NIDENT; ++i)
  83         printf (" %02x", ehdr->e_ident[i]);
  84       printf ("\
  85 \ntype = %hu\nmachine = %hu\nversion = %u\nentry = %u\nphoff = %u\n"
  86               "shoff = %u\nflags = %u\nehsize = %hu\nphentsize = %hu\n"
  87               "phnum = %hu\nshentsize = %hu\nshnum = %hu\nshstrndx = %hu\n",
  88               ehdr->e_type, ehdr->e_machine, ehdr->e_version, ehdr->e_entry,
  89               ehdr->e_phoff, ehdr->e_shoff, ehdr->e_flags, ehdr->e_ehsize,
  90               ehdr->e_phentsize, ehdr->e_phnum, ehdr->e_shentsize,
  91               ehdr->e_shnum, ehdr->e_shstrndx);
  92     }
  93
  94   ehdr->e_ident[0] = 42;
  95   ehdr->e_ident[4] = 1;
  96   ehdr->e_ident[5] = 1;
  97   ehdr->e_ident[6] = 2;
  98   ehdr->e_type = ET_EXEC;
  99   ehdr->e_version = 1;
 100   ehdr->e_ehsize = 1;
 101   elf_flagehdr (elf, ELF_C_SET, ELF_F_DIRTY);
 102
 103   /* Create the program header.  */
 104   phdr = elf32_newphdr (elf, 1);
 105   if (phdr == NULL)
 106     {
 107       printf ("cannot create program header: %s\n", elf_errmsg (-1));
 108       exit (1);
 109     }
 110
 111   phdr[0].p_type = PT_PHDR;
 112   elf_flagphdr (elf, ELF_C_SET, ELF_F_DIRTY);
 113
 114   shst = ebl_strtabinit (true);
 115
 116   scn = elf_newscn (elf);
 117   if (scn == NULL)
 118     {
 119       printf ("cannot create first section: %s\n", elf_errmsg (-1));
 120       exit (1);
 121     }
 122   shdr = elf32_getshdr (scn);
 123   if (shdr == NULL)
 124     {
 125       printf ("cannot get header for first section: %s\n", elf_errmsg (-1));
 126       exit (1);
 127     }
 128
 129   firstse = ebl_strtabadd (shst, ".first", 0);
 130
 131   shdr->sh_type = SHT_PROGBITS;
 132   shdr->sh_flags = SHF_ALLOC | SHF_EXECINSTR;
 133   shdr->sh_addr = 0;
 134   shdr->sh_link = 0;
 135   shdr->sh_info = 0;
 136   shdr->sh_entsize = 1;
 137
 138   data = elf_newdata (scn);
 139   if (data == NULL)
 140     {
 141       printf ("cannot create data first section: %s\n", elf_errmsg (-1));
 142       exit (1);
 143     }
 144
 145   data->d_buf = "hello";
 146   data->d_type = ELF_T_BYTE;
 147   data->d_version = EV_CURRENT;
 148   data->d_size = 5;
 149   data->d_align = 16;
 150
 151
 152   scn = elf_newscn (elf);
 153   if (scn == NULL)
 154     {
 155       printf ("cannot create second section: %s\n", elf_errmsg (-1));
 156       exit (1);
 157     }
 158   shdr = elf32_getshdr (scn);
 159   if (shdr == NULL)
 160     {
 161       printf ("cannot get header for second section: %s\n", elf_errmsg (-1));
 162       exit (1);
 163     }
 164
 165   secondse = ebl_strtabadd (shst, ".second", 0);
 166
 167   shdr->sh_type = SHT_PROGBITS;
 168   shdr->sh_flags = SHF_ALLOC | SHF_WRITE;
 169   shdr->sh_addr = 0;
 170   shdr->sh_link = 0;
 171   shdr->sh_info = 0;
 172   shdr->sh_entsize = 1;
 173
 174   data = elf_newdata (scn);
 175   if (data == NULL)
 176     {
 177       printf ("cannot create data second section: %s\n", elf_errmsg (-1));
 178       exit (1);
 179     }
 180
 181   data->d_buf = "world";
 182   data->d_type = ELF_T_BYTE;
 183   data->d_version = EV_CURRENT;
 184   data->d_size = 5;
 185   data->d_align = 16;
 186
 187
 188   scn = elf_newscn (elf);
 189   if (scn == NULL)
 190     {
 191       printf ("cannot create third section: %s\n", elf_errmsg (-1));
 192       exit (1);
 193     }
 194   shdr = elf32_getshdr (scn);
 195   if (shdr == NULL)
 196     {
 197       printf ("cannot get header for third section: %s\n", elf_errmsg (-1));
 198       exit (1);
 199     }
 200
 201   thirdse = ebl_strtabadd (shst, ".third", 0);
 202
 203   shdr->sh_type = SHT_PROGBITS;
 204   shdr->sh_flags = SHF_ALLOC | SHF_EXECINSTR;
 205   shdr->sh_addr = 0;
 206   shdr->sh_link = 0;
 207   shdr->sh_info = 0;
 208   shdr->sh_entsize = 1;
 209
 210   data = elf_newdata (scn);
 211   if (data == NULL)
 212     {
 213       printf ("cannot create data third section: %s\n", elf_errmsg (-1));
 214       exit (1);
 215     }
 216
 217   data->d_buf = "!!!!!";
 218   data->d_type = ELF_T_BYTE;
 219   data->d_version = EV_CURRENT;
 220   data->d_size = 5;
 221   data->d_align = 16;
 222
 223
 224   scn = elf_newscn (elf);
 225   if (scn == NULL)
 226     {
 227       printf ("cannot create fourth section: %s\n", elf_errmsg (-1));
 228       exit (1);
 229     }
 230   shdr = elf32_getshdr (scn);
 231   if (shdr == NULL)
 232     {
 233       printf ("cannot get header for fourth section: %s\n", elf_errmsg (-1));
 234       exit (1);
 235     }
 236
 237   fourthse = ebl_strtabadd (shst, ".fourth", 0);
 238
 239   shdr->sh_type = SHT_NOBITS;
 240   shdr->sh_flags = SHF_ALLOC | SHF_EXECINSTR;
 241   shdr->sh_addr = 0;
 242   shdr->sh_link = 0;
 243   shdr->sh_info = 0;
 244   shdr->sh_entsize = 1;
 245   shdr->sh_size = 100;
 246
 247   data = elf_newdata (scn);
 248   if (data == NULL)
 249     {
 250       printf ("cannot create data fourth section: %s\n", elf_errmsg (-1));
 251       exit (1);
 252     }
 253
 254   data->d_buf = NULL;
 255   data->d_type = ELF_T_BYTE;
 256   data->d_version = EV_CURRENT;
 257   data->d_size = 100;
 258   data->d_align = 16;
 259
 260
 261   scn = elf_newscn (elf);
 262   if (scn == NULL)
 263     {
 264       printf ("cannot create SHSTRTAB section: %s\n", elf_errmsg (-1));
 265       exit (1);
 266     }
 267   shdr = elf32_getshdr (scn);
 268   if (shdr == NULL)
 269     {
 270       printf ("cannot get header for SHSTRTAB section: %s\n", elf_errmsg (-1));
 271       exit (1);
 272     }
 273
 274   shstrtabse = ebl_strtabadd (shst, ".shstrtab", 0);
 275
 276   shdr->sh_type = SHT_STRTAB;
 277   shdr->sh_flags = 0;
 278   shdr->sh_addr = 0;
 279   shdr->sh_link = SHN_UNDEF;
 280   shdr->sh_info = SHN_UNDEF;
 281   shdr->sh_entsize = 1;
 282
 283   /* We have to store the section index in the ELF header.  */
 284   ehdr->e_shstrndx = elf_ndxscn (scn);
 285
 286   data = elf_newdata (scn);
 287   if (data == NULL)
 288     {
 289       printf ("cannot create data SHSTRTAB section: %s\n", elf_errmsg (-1));
 290       exit (1);
 291     }
 292
 293   /* No more sections, finalize the section header string table.  */
 294   ebl_strtabfinalize (shst, data);
 295
 296   elf32_getshdr (elf_getscn (elf, 1))->sh_name = ebl_strtaboffset (firstse);
 297   elf32_getshdr (elf_getscn (elf, 2))->sh_name = ebl_strtaboffset (secondse);
 298   elf32_getshdr (elf_getscn (elf, 3))->sh_name = ebl_strtaboffset (thirdse);
 299   elf32_getshdr (elf_getscn (elf, 4))->sh_name = ebl_strtaboffset (fourthse);
 300   shdr->sh_name = ebl_strtaboffset (shstrtabse);
 301
 302   /* Let the library compute the internal structure information.  */
 303   if (elf_update (elf, ELF_C_NULL) < 0)
 304     {
 305       printf ("failure in elf_update(NULL): %s\n", elf_errmsg (-1));
 306       exit (1);
 307     }
 308
 309   ehdr = elf32_getehdr (elf);
 310
 311   phdr[0].p_offset = ehdr->e_phoff;
 312   phdr[0].p_offset = ehdr->e_phoff;
 313   phdr[0].p_vaddr = ehdr->e_phoff;
 314   phdr[0].p_paddr = ehdr->e_phoff;
 315   phdr[0].p_flags = PF_R | PF_X;
 316   phdr[0].p_filesz = ehdr->e_phnum * elf32_fsize (ELF_T_PHDR, 1, EV_CURRENT);
 317   phdr[0].p_memsz = ehdr->e_phnum * elf32_fsize (ELF_T_PHDR, 1, EV_CURRENT);
 318   phdr[0].p_align = sizeof (Elf32_Word);
 319
 320   /* Write out the file.  */
 321   if (elf_update (elf, ELF_C_WRITE) < 0)
 322     {
 323       printf ("failure in elf_update(WRITE): %s\n", elf_errmsg (-1));
 324       exit (1);
 325     }
 326
 327   /* We don't need the string table anymore.  */
 328   ebl_strtabfree (shst);
 329
 330   /* And the data allocated in the .shstrtab section.  */
 331   free (data->d_buf);
 332
 333   /* Print the ELF header values.  */
 334   if (argc > 1)
 335     {
 336       for (i = 0; i < EI_NIDENT; ++i)
 337         printf (" %02x", ehdr->e_ident[i]);
 338       printf ("\
 339 \ntype = %hu\nmachine = %hu\nversion = %u\nentry = %u\nphoff = %u\n"
 340               "shoff = %u\nflags = %u\nehsize = %hu\nphentsize = %hu\n"
 341               "phnum = %hu\nshentsize = %hu\nshnum = %hu\nshstrndx = %hu\n",
 342               ehdr->e_type, ehdr->e_machine, ehdr->e_version, ehdr->e_entry,
 343               ehdr->e_phoff, ehdr->e_shoff, ehdr->e_flags, ehdr->e_ehsize,
 344               ehdr->e_phentsize, ehdr->e_phnum, ehdr->e_shentsize,
 345               ehdr->e_shnum, ehdr->e_shstrndx);
 346     }
 347
 348   if (elf_end (elf) != 0)
 349     {
 350       printf ("failure in elf_end: %s\n", elf_errmsg (-1));
 351       exit (1);
 352     }
 353
 354   unlink (fname);
 355
 356   return 0;
 357 }