tests/asm-tst3.c

   1 /* Copyright (C) 2002, 2005 Red Hat, Inc.
   2    This file is part of Red Hat elfutils.
   3    Written by Ulrich Drepper <drepper@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 }