tests/asm-tst2.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-tst2-out.o";
  31
  32
  33 static const GElf_Ehdr expected_ehdr =
  34   {
  35     .e_ident = { [EI_MAG0] = ELFMAG0,
  36                  [EI_MAG1] = ELFMAG1,
  37                  [EI_MAG2] = ELFMAG2,
  38                  [EI_MAG3] = ELFMAG3,
  39                  [EI_CLASS] = ELFCLASS32,
  40                  [EI_DATA] = ELFDATA2LSB,
  41                  [EI_VERSION] = EV_CURRENT },
  42     .e_type = ET_REL,
  43     .e_machine = EM_386,
  44     .e_version = EV_CURRENT,
  45     .e_shoff = 96,
  46     .e_ehsize = sizeof (Elf32_Ehdr),
  47     .e_shentsize = sizeof (Elf32_Shdr),
  48     .e_shnum = 3,
  49     .e_shstrndx = 2
  50   };
  51
  52
  53 static const char *scnnames[3] =
  54   {
  55     [0] = "",
  56     [1] = ".data",
  57     [2] = ".shstrtab"
  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.  */
 108   if (asm_addstrz (scn1, "one", 4) != 0)
 109     {
 110       printf ("cannot insert first string: %s\n", asm_errmsg (-1));
 111       result = 1;
 112     }
 113   if (asm_addstrz (scn2, "three", 0) != 0)
 114     {
 115       printf ("cannot insert second string: %s\n", asm_errmsg (-1));
 116       result = 1;
 117     }
 118   if (asm_addstrz (scn1, "two", 4) != 0)
 119     {
 120       printf ("cannot insert third string: %s\n", asm_errmsg (-1));
 121       result = 1;
 122     }
 123
 124   /* Create the output file.  */
 125   if (asm_end (ctx) != 0)
 126     {
 127       printf ("cannot create output file: %s\n", asm_errmsg (-1));
 128       asm_abort (ctx);
 129       return 1;
 130     }
 131
 132   /* Check the file.  */
 133   fd = open (fname, O_RDONLY);
 134   if (fd == -1)
 135     {
 136       printf ("cannot open generated file: %m\n");
 137       result = 1;
 138       goto out;
 139     }
 140
 141   elf = elf_begin (fd, ELF_C_READ, NULL);
 142   if (elf == NULL)
 143     {
 144       printf ("cannot create ELF descriptor: %s\n", elf_errmsg (-1));
 145       result = 1;
 146       goto out_close;
 147     }
 148   if (elf_kind (elf) != ELF_K_ELF)
 149     {
 150       puts ("not a valid ELF file");
 151       result = 1;
 152       goto out_close2;
 153     }
 154
 155   ehdr = gelf_getehdr (elf, &ehdr_mem);
 156   if (ehdr == NULL)
 157     {
 158       printf ("cannot get ELF header: %s\n", elf_errmsg (-1));
 159       result = 1;
 160       goto out_close2;
 161     }
 162
 163   if (memcmp (ehdr, &expected_ehdr, sizeof (GElf_Ehdr)) != 0)
 164     {
 165       puts ("ELF header does not match");
 166       result = 1;
 167       goto out_close2;
 168     }
 169
 170   for (cnt = 1; cnt < 3; ++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 != (cnt == 2 ? SHT_STRTAB : SHT_PROGBITS))
 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 == 2 && 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 == 1 && shdr->sh_offset != ((sizeof (Elf32_Ehdr) + 15) & ~15))
 222           || (cnt == 2
 223               && shdr->sh_offset != (((sizeof (Elf32_Ehdr) + 15) & ~15)
 224                                      + strlen ("one") + 1
 225                                      + strlen ("two") + 1
 226                                      + strlen ("three") + 1)))
 227         {
 228           printf ("section %Zd's offset differs\n", cnt);
 229           result = 1;
 230         }
 231
 232       if ((cnt == 1 && shdr->sh_size != (strlen ("one") + 1
 233                                          + strlen ("two") + 1
 234                                          + strlen ("three") + 1))
 235           || (cnt == 2 && shdr->sh_size != 17))
 236         {
 237           printf ("section %Zd's size differs\n", cnt);
 238           result = 1;
 239         }
 240
 241       if (shdr->sh_link != 0)
 242         {
 243           printf ("section %Zd's link differs\n", cnt);
 244           result = 1;
 245         }
 246
 247       if (shdr->sh_info != 0)
 248         {
 249           printf ("section %Zd's info differs\n", cnt);
 250           result = 1;
 251         }
 252
 253       if ((cnt == 1 && shdr->sh_addralign != 16)
 254           || (cnt != 1 && shdr->sh_addralign != 1))
 255         {
 256           printf ("section %Zd's addralign differs\n", cnt);
 257           result = 1;
 258         }
 259
 260       if (shdr->sh_entsize != 0)
 261         {
 262           printf ("section %Zd's entsize differs\n", cnt);
 263           result = 1;
 264         }
 265     }
 266
 267  out_close2:
 268   elf_end (elf);
 269  out_close:
 270   close (fd);
 271  out:
 272   /* We don't need the file anymore.  */
 273   unlink (fname);
 274
 275   ebl_closebackend (ebl);
 276
 277   return result;
 278 }