1 /* Copyright (C) 2002, 2005 Red Hat, Inc. 2 This file is part of Red Hat elfutils. 3 Written by Ulrich Drepper <drepper (at) redhat.com>, 2002. 4 5 Red Hat elfutils is free software; you can redistribute it and/or modify 6 it under the terms of the GNU General Public License as published by the 7 Free Software Foundation; version 2 of the License. 8 9 Red Hat elfutils is distributed in the hope that it will be useful, but 10 WITHOUT ANY WARRANTY; without even the implied warranty of 11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 12 General Public License for more details. 13 14 You should have received a copy of the GNU General Public License along 15 with Red Hat elfutils; if not, write to the Free Software Foundation, 16 Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301 USA. 17 18 Red Hat elfutils is an included package of the Open Invention Network. 19 An included package of the Open Invention Network is a package for which 20 Open Invention Network licensees cross-license their patents. No patent 21 license is granted, either expressly or impliedly, by designation as an 22 included package. Should you wish to participate in the Open Invention 23 Network licensing program, please visit www.openinventionnetwork.com 24 <http://www.openinventionnetwork.com>. */ 25 26 #ifdef HAVE_CONFIG_H 27 # include <config.h> 28 #endif 29 30 #include <fcntl.h> 31 #include ELFUTILS_HEADER(asm) 32 #include <libelf.h> 33 #include <stdio.h> 34 #include <string.h> 35 #include <unistd.h> 36 37 38 static const char fname[] = "asm-tst3-out.o"; 39 40 41 static const char *scnnames[5] = 42 { 43 [0] = "", 44 [1] = ".data", 45 [2] = ".strtab", 46 [3] = ".symtab", 47 [4] = ".shstrtab" 48 }; 49 50 51 static unsigned int scntypes[5] = 52 { 53 [0] = SHT_NULL, 54 [1] = SHT_PROGBITS, 55 [2] = SHT_STRTAB, 56 [3] = SHT_SYMTAB, 57 [4] = SHT_STRTAB 58 }; 59 60 61 int 62 main (void) 63 { 64 AsmCtx_t *ctx; 65 AsmScn_t *scn1; 66 AsmScn_t *scn2; 67 int result = 0; 68 int fd; 69 Elf *elf; 70 GElf_Ehdr ehdr_mem; 71 GElf_Ehdr *ehdr; 72 size_t cnt; 73 74 elf_version (EV_CURRENT); 75 76 Ebl *ebl = ebl_openbackend_machine (EM_386); 77 if (ebl == NULL) 78 { 79 puts ("cannot open backend library"); 80 return 1; 81 } 82 83 ctx = asm_begin (fname, ebl, false); 84 if (ctx == NULL) 85 { 86 printf ("cannot create assembler context: %s\n", asm_errmsg (-1)); 87 return 1; 88 } 89 90 /* Create two sections. */ 91 scn1 = asm_newscn (ctx, ".data", SHT_PROGBITS, SHF_ALLOC | SHF_WRITE); 92 scn2 = asm_newsubscn (scn1, 1); 93 if (scn1 == NULL || scn2 == NULL) 94 { 95 printf ("cannot create section in output file: %s\n", asm_errmsg (-1)); 96 asm_abort (ctx); 97 return 1; 98 } 99 100 /* Special alignment for the .text section. */ 101 if (asm_align (scn1, 16) != 0) 102 { 103 printf ("cannot align .text section: %s\n", asm_errmsg (-1)); 104 result = 1; 105 } 106 107 /* Add a few strings with names. */ 108 if (asm_newsym (scn1, "one", 4, STT_OBJECT, STB_GLOBAL) == NULL) 109 { 110 printf ("cannot create first name: %s\n", asm_errmsg (-1)); 111 result = 1; 112 } 113 if (asm_addstrz (scn1, "one", 4) != 0) 114 { 115 printf ("cannot insert first string: %s\n", asm_errmsg (-1)); 116 result = 1; 117 } 118 if (asm_newsym (scn2, "three", 6, STT_OBJECT, STB_WEAK) == NULL) 119 { 120 printf ("cannot create second name: %s\n", asm_errmsg (-1)); 121 result = 1; 122 } 123 if (asm_addstrz (scn2, "three", 0) != 0) 124 { 125 printf ("cannot insert second string: %s\n", asm_errmsg (-1)); 126 result = 1; 127 } 128 if (asm_newsym (scn1, "two", 4, STT_OBJECT, STB_LOCAL) == NULL) 129 { 130 printf ("cannot create third name: %s\n", asm_errmsg (-1)); 131 result = 1; 132 } 133 if (asm_addstrz (scn1, "two", 4) != 0) 134 { 135 printf ("cannot insert third string: %s\n", asm_errmsg (-1)); 136 result = 1; 137 } 138 139 /* Create the output file. */ 140 if (asm_end (ctx) != 0) 141 { 142 printf ("cannot create output file: %s\n", asm_errmsg (-1)); 143 asm_abort (ctx); 144 return 1; 145 } 146 147 /* Check the file. */ 148 fd = open (fname, O_RDONLY); 149 if (fd == -1) 150 { 151 printf ("cannot open generated file: %m\n"); 152 result = 1; 153 goto out; 154 } 155 156 elf = elf_begin (fd, ELF_C_READ, NULL); 157 if (elf == NULL) 158 { 159 printf ("cannot create ELF descriptor: %s\n", elf_errmsg (-1)); 160 result = 1; 161 goto out_close; 162 } 163 if (elf_kind (elf) != ELF_K_ELF) 164 { 165 puts ("not a valid ELF file"); 166 result = 1; 167 goto out_close2; 168 } 169 170 ehdr = gelf_getehdr (elf, &ehdr_mem); 171 if (ehdr == NULL) 172 { 173 printf ("cannot get ELF header: %s\n", elf_errmsg (-1)); 174 result = 1; 175 goto out_close2; 176 } 177 178 for (cnt = 1; cnt < 5; ++cnt) 179 { 180 Elf_Scn *scn; 181 GElf_Shdr shdr_mem; 182 GElf_Shdr *shdr; 183 184 scn = elf_getscn (elf, cnt); 185 if (scn == NULL) 186 { 187 printf ("cannot get section %Zd: %s\n", cnt, elf_errmsg (-1)); 188 result = 1; 189 continue; 190 } 191 192 shdr = gelf_getshdr (scn, &shdr_mem); 193 if (shdr == NULL) 194 { 195 printf ("cannot get section header for section %Zd: %s\n", 196 cnt, elf_errmsg (-1)); 197 result = 1; 198 continue; 199 } 200 201 if (strcmp (elf_strptr (elf, ehdr->e_shstrndx, shdr->sh_name), 202 scnnames[cnt]) != 0) 203 { 204 printf ("section %Zd's name differs: %s vs %s\n", cnt, 205 elf_strptr (elf, ehdr->e_shstrndx, shdr->sh_name), 206 scnnames[cnt]); 207 result = 1; 208 } 209 210 if (shdr->sh_type != scntypes[cnt]) 211 { 212 printf ("section %Zd's type differs\n", cnt); 213 result = 1; 214 } 215 216 if ((cnt == 1 && shdr->sh_flags != (SHF_ALLOC | SHF_WRITE)) 217 || (cnt != 1 && shdr->sh_flags != 0)) 218 { 219 printf ("section %Zd's flags differs\n", cnt); 220 result = 1; 221 } 222 223 if (shdr->sh_addr != 0) 224 { 225 printf ("section %Zd's address differs\n", cnt); 226 result = 1; 227 } 228 229 if (cnt == 3) 230 { 231 Elf_Data *data; 232 233 if (shdr->sh_link != 2) 234 { 235 puts ("symbol table has incorrect link"); 236 result = 1; 237 } 238 239 data = elf_getdata (scn, NULL); 240 if (data == NULL) 241 { 242 puts ("cannot get data of symbol table"); 243 result = 1; 244 } 245 else 246 { 247 size_t inner; 248 249 for (inner = 1; 250 inner < (shdr->sh_size 251 / gelf_fsize (elf, ELF_T_SYM, 1, EV_CURRENT)); 252 ++inner) 253 { 254 GElf_Sym sym_mem; 255 GElf_Sym *sym; 256 257 sym = gelf_getsym (data, inner, &sym_mem); 258 if (sym == NULL) 259 { 260 printf ("cannot get symbol %zu: %s\n", 261 inner, elf_errmsg (-1)); 262 result = 1; 263 } 264 else 265 { 266 /* The order of the third and fourth entry depends 267 on how the hash table is organized. */ 268 static const char *names[4] = 269 { 270 [0] = "", 271 [1] = "two", 272 [2] = "one", 273 [3] = "three" 274 }; 275 static const int info[4] = 276 { 277 [0] = GELF_ST_INFO (STB_LOCAL, STT_NOTYPE), 278 [1] = GELF_ST_INFO (STB_LOCAL, STT_OBJECT), 279 [2] = GELF_ST_INFO (STB_GLOBAL, STT_OBJECT), 280 [3] = GELF_ST_INFO (STB_WEAK, STT_OBJECT) 281 }; 282 static const unsigned value[4] = 283 { 284 [0] = 0, 285 [1] = 4, 286 [2] = 0, 287 [3] = 8 288 }; 289 290 if (strcmp (names[inner], 291 elf_strptr (elf, shdr->sh_link, 292 sym->st_name)) != 0) 293 { 294 printf ("symbol %zu has different name\n", inner); 295 result = 1; 296 } 297 298 if (sym->st_value != value[inner]) 299 { 300 printf ("symbol %zu has wrong value\n", inner); 301 result = 1; 302 } 303 304 if (sym->st_other != 0) 305 { 306 printf ("symbol %zu has wrong other info\n", inner); 307 result = 1; 308 } 309 310 if (sym->st_shndx != 1) 311 { 312 printf ("symbol %zu has wrong section reference\n", 313 inner); 314 result = 1; 315 } 316 317 if (sym->st_info != info[inner]) 318 { 319 printf ("symbol %zu has wrong type or binding\n", 320 inner); 321 result = 1; 322 } 323 324 if ((inner != 3 && sym->st_size != 4) 325 || (inner == 3 && sym->st_size != 6)) 326 { 327 printf ("symbol %zu has wrong size\n", inner); 328 result = 1; 329 } 330 } 331 } 332 } 333 } 334 } 335 336 out_close2: 337 elf_end (elf); 338 out_close: 339 close (fd); 340 out: 341 /* We don't need the file anymore. */ 342 unlink (fname); 343 344 ebl_closebackend (ebl); 345 346 return result; 347 } 348
