tests/asm-tst3.c

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