1 /* Test program for elf_update function. 2 Copyright (C) 2000, 2002, 2005 Red Hat, Inc. 3 This file is part of Red Hat elfutils. 4 Written by Ulrich Drepper <drepper (at) redhat.com>, 2000. 5 6 Red Hat elfutils is free software; you can redistribute it and/or modify 7 it under the terms of the GNU General Public License as published by the 8 Free Software Foundation; version 2 of the License. 9 10 Red Hat 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 GNU 13 General Public License for more details. 14 15 You should have received a copy of the GNU General Public License along 16 with Red Hat elfutils; if not, write to the Free Software Foundation, 17 Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301 USA. 18 19 Red Hat elfutils is an included package of the Open Invention Network. 20 An included package of the Open Invention Network is a package for which 21 Open Invention Network licensees cross-license their patents. No patent 22 license is granted, either expressly or impliedly, by designation as an 23 included package. Should you wish to participate in the Open Invention 24 Network licensing program, please visit www.openinventionnetwork.com 25 <http://www.openinventionnetwork.com>. */ 26 27 #ifdef HAVE_CONFIG_H 28 # include <config.h> 29 #endif 30 31 #include <errno.h> 32 #include <fcntl.h> 33 #include <libelf.h> 34 #include <stdio.h> 35 #include <stdlib.h> 36 #include <string.h> 37 38 #include ELFUTILS_HEADER(ebl) 39 40 41 int 42 main (int argc, char *argv[] __attribute__ ((unused))) 43 { 44 const char *fname = "xxx"; 45 int fd; 46 Elf *elf; 47 Elf32_Ehdr *ehdr; 48 Elf32_Phdr *phdr; 49 Elf_Scn *scn; 50 Elf32_Shdr *shdr; 51 Elf_Data *data; 52 struct Ebl_Strtab *shst; 53 struct Ebl_Strent *shstrtabse; 54 int i; 55 56 fd = open (fname, O_RDWR | O_CREAT | O_TRUNC, 0666); 57 if (fd == -1) 58 { 59 printf ("cannot open `%s': %s\n", fname, strerror (errno)); 60 exit (1); 61 } 62 63 elf_version (EV_CURRENT); 64 65 elf_fill (0x42); 66 67 elf = elf_begin (fd, ELF_C_WRITE, NULL); 68 if (elf == NULL) 69 { 70 printf ("cannot create ELF descriptor: %s\n", elf_errmsg (-1)); 71 exit (1); 72 } 73 74 /* Create an ELF header. */ 75 ehdr = elf32_newehdr (elf); 76 if (ehdr == NULL) 77 { 78 printf ("cannot create ELF header: %s\n", elf_errmsg (-1)); 79 exit (1); 80 } 81 82 /* Print the ELF header values. */ 83 if (argc > 1) 84 { 85 for (i = 0; i < EI_NIDENT; ++i) 86 printf (" %02x", ehdr->e_ident[i]); 87 printf ("\ 88 \ntype = %hu\nmachine = %hu\nversion = %u\nentry = %u\nphoff = %u\n" 89 "shoff = %u\nflags = %u\nehsize = %hu\nphentsize = %hu\n" 90 "phnum = %hu\nshentsize = %hu\nshnum = %hu\nshstrndx = %hu\n", 91 ehdr->e_type, ehdr->e_machine, ehdr->e_version, ehdr->e_entry, 92 ehdr->e_phoff, ehdr->e_shoff, ehdr->e_flags, ehdr->e_ehsize, 93 ehdr->e_phentsize, ehdr->e_phnum, ehdr->e_shentsize, 94 ehdr->e_shnum, ehdr->e_shstrndx); 95 } 96 97 ehdr->e_ident[0] = 42; 98 ehdr->e_ident[4] = 1; 99 ehdr->e_ident[5] = 1; 100 ehdr->e_ident[6] = 2; 101 ehdr->e_type = ET_EXEC; 102 ehdr->e_version = 1; 103 ehdr->e_ehsize = 1; 104 elf_flagehdr (elf, ELF_C_SET, ELF_F_DIRTY); 105 106 /* Create the program header. */ 107 phdr = elf32_newphdr (elf, 1); 108 if (phdr == NULL) 109 { 110 printf ("cannot create program header: %s\n", elf_errmsg (-1)); 111 exit (1); 112 } 113 114 phdr[0].p_type = PT_PHDR; 115 elf_flagphdr (elf, ELF_C_SET, ELF_F_DIRTY); 116 117 shst = ebl_strtabinit (true); 118 119 scn = elf_newscn (elf); 120 if (scn == NULL) 121 { 122 printf ("cannot create SHSTRTAB section: %s\n", elf_errmsg (-1)); 123 exit (1); 124 } 125 shdr = elf32_getshdr (scn); 126 if (shdr == NULL) 127 { 128 printf ("cannot get header for SHSTRTAB section: %s\n", elf_errmsg (-1)); 129 exit (1); 130 } 131 132 shstrtabse = ebl_strtabadd (shst, ".shstrtab", 0); 133 134 shdr->sh_type = SHT_STRTAB; 135 shdr->sh_flags = 0; 136 shdr->sh_addr = 0; 137 shdr->sh_link = SHN_UNDEF; 138 shdr->sh_info = SHN_UNDEF; 139 shdr->sh_addralign = 1; 140 shdr->sh_entsize = 0; 141 142 /* We have to store the section index in the ELF header. */ 143 ehdr->e_shstrndx = elf_ndxscn (scn); 144 145 data = elf_newdata (scn); 146 if (data == NULL) 147 { 148 printf ("cannot create data SHSTRTAB section: %s\n", elf_errmsg (-1)); 149 exit (1); 150 } 151 152 /* No more sections, finalize the section header string table. */ 153 ebl_strtabfinalize (shst, data); 154 155 shdr->sh_name = ebl_strtaboffset (shstrtabse); 156 157 /* Let the library compute the internal structure information. */ 158 if (elf_update (elf, ELF_C_NULL) < 0) 159 { 160 printf ("failure in elf_update(NULL): %s\n", elf_errmsg (-1)); 161 exit (1); 162 } 163 164 ehdr = elf32_getehdr (elf); 165 166 phdr[0].p_offset = ehdr->e_phoff; 167 phdr[0].p_offset = ehdr->e_phoff; 168 phdr[0].p_vaddr = ehdr->e_phoff; 169 phdr[0].p_paddr = ehdr->e_phoff; 170 phdr[0].p_flags = PF_R | PF_X; 171 phdr[0].p_filesz = ehdr->e_phnum * elf32_fsize (ELF_T_PHDR, 1, EV_CURRENT); 172 phdr[0].p_memsz = ehdr->e_phnum * elf32_fsize (ELF_T_PHDR, 1, EV_CURRENT); 173 phdr[0].p_align = sizeof (Elf32_Word); 174 175 /* Write out the file. */ 176 if (elf_update (elf, ELF_C_WRITE) < 0) 177 { 178 printf ("failure in elf_update(WRITE): %s\n", elf_errmsg (-1)); 179 exit (1); 180 } 181 182 /* We don't need the string table anymore. */ 183 ebl_strtabfree (shst); 184 185 /* And the data allocated in the .shstrtab section. */ 186 free (data->d_buf); 187 188 /* Print the ELF header values. */ 189 if (argc > 1) 190 { 191 for (i = 0; i < EI_NIDENT; ++i) 192 printf (" %02x", ehdr->e_ident[i]); 193 printf ("\ 194 \ntype = %hu\nmachine = %hu\nversion = %u\nentry = %u\nphoff = %u\n" 195 "shoff = %u\nflags = %u\nehsize = %hu\nphentsize = %hu\n" 196 "phnum = %hu\nshentsize = %hu\nshnum = %hu\nshstrndx = %hu\n", 197 ehdr->e_type, ehdr->e_machine, ehdr->e_version, ehdr->e_entry, 198 ehdr->e_phoff, ehdr->e_shoff, ehdr->e_flags, ehdr->e_ehsize, 199 ehdr->e_phentsize, ehdr->e_phnum, ehdr->e_shentsize, 200 ehdr->e_shnum, ehdr->e_shstrndx); 201 } 202 203 if (elf_end (elf) != 0) 204 { 205 printf ("failure in elf_end: %s\n", elf_errmsg (-1)); 206 exit (1); 207 } 208 209 return 0; 210 } 211
