Blame tests/asm-tst2.c

Packit Service 97d2fb
/* Copyright (C) 2002, 2005 Red Hat, Inc.
Packit Service 97d2fb
   This file is part of elfutils.
Packit Service 97d2fb
   Written by Ulrich Drepper <drepper@redhat.com>, 2002.
Packit Service 97d2fb
Packit Service 97d2fb
   This file is free software; you can redistribute it and/or modify
Packit Service 97d2fb
   it under the terms of the GNU General Public License as published by
Packit Service 97d2fb
   the Free Software Foundation; either version 3 of the License, or
Packit Service 97d2fb
   (at your option) any later version.
Packit Service 97d2fb
Packit Service 97d2fb
   elfutils is distributed in the hope that it will be useful, but
Packit Service 97d2fb
   WITHOUT ANY WARRANTY; without even the implied warranty of
Packit Service 97d2fb
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
Packit Service 97d2fb
   GNU General Public License for more details.
Packit Service 97d2fb
Packit Service 97d2fb
   You should have received a copy of the GNU General Public License
Packit Service 97d2fb
   along with this program.  If not, see <http://www.gnu.org/licenses/>.  */
Packit Service 97d2fb
Packit Service 97d2fb
#ifdef HAVE_CONFIG_H
Packit Service 97d2fb
# include <config.h>
Packit Service 97d2fb
#endif
Packit Service 97d2fb
Packit Service 97d2fb
#include <fcntl.h>
Packit Service 97d2fb
#include ELFUTILS_HEADER(asm)
Packit Service 97d2fb
#include ELFUTILS_HEADER(ebl)
Packit Service 97d2fb
#include <libelf.h>
Packit Service 97d2fb
#include <stdio.h>
Packit Service 97d2fb
#include <string.h>
Packit Service 97d2fb
#include <unistd.h>
Packit Service 97d2fb
Packit Service 97d2fb
Packit Service 97d2fb
static const char fname[] = "asm-tst2-out.o";
Packit Service 97d2fb
Packit Service 97d2fb
Packit Service 97d2fb
static const GElf_Ehdr expected_ehdr =
Packit Service 97d2fb
  {
Packit Service 97d2fb
    .e_ident = { [EI_MAG0] = ELFMAG0,
Packit Service 97d2fb
		 [EI_MAG1] = ELFMAG1,
Packit Service 97d2fb
		 [EI_MAG2] = ELFMAG2,
Packit Service 97d2fb
		 [EI_MAG3] = ELFMAG3,
Packit Service 97d2fb
		 [EI_CLASS] = ELFCLASS32,
Packit Service 97d2fb
		 [EI_DATA] = ELFDATA2LSB,
Packit Service 97d2fb
		 [EI_VERSION] = EV_CURRENT },
Packit Service 97d2fb
    .e_type = ET_REL,
Packit Service 97d2fb
    .e_machine = EM_386,
Packit Service 97d2fb
    .e_version = EV_CURRENT,
Packit Service 97d2fb
    .e_shoff = 96,
Packit Service 97d2fb
    .e_ehsize = sizeof (Elf32_Ehdr),
Packit Service 97d2fb
    .e_shentsize = sizeof (Elf32_Shdr),
Packit Service 97d2fb
    .e_shnum = 3,
Packit Service 97d2fb
    .e_shstrndx = 2
Packit Service 97d2fb
  };
Packit Service 97d2fb
Packit Service 97d2fb
Packit Service 97d2fb
static const char *scnnames[3] =
Packit Service 97d2fb
  {
Packit Service 97d2fb
    [0] = "",
Packit Service 97d2fb
    [1] = ".data",
Packit Service 97d2fb
    [2] = ".shstrtab"
Packit Service 97d2fb
  };
Packit Service 97d2fb
Packit Service 97d2fb
Packit Service 97d2fb
int
Packit Service 97d2fb
main (void)
Packit Service 97d2fb
{
Packit Service 97d2fb
  AsmCtx_t *ctx;
Packit Service 97d2fb
  AsmScn_t *scn1;
Packit Service 97d2fb
  AsmScn_t *scn2;
Packit Service 97d2fb
  int result = 0;
Packit Service 97d2fb
  int fd;
Packit Service 97d2fb
  Elf *elf;
Packit Service 97d2fb
  GElf_Ehdr ehdr_mem;
Packit Service 97d2fb
  GElf_Ehdr *ehdr;
Packit Service 97d2fb
  size_t cnt;
Packit Service 97d2fb
Packit Service 97d2fb
  elf_version (EV_CURRENT);
Packit Service 97d2fb
Packit Service 97d2fb
  Ebl *ebl = ebl_openbackend_machine (EM_386);
Packit Service 97d2fb
  if (ebl == NULL)
Packit Service 97d2fb
    {
Packit Service 97d2fb
      puts ("cannot open backend library");
Packit Service 97d2fb
      return 1;
Packit Service 97d2fb
    }
Packit Service 97d2fb
Packit Service 97d2fb
  ctx = asm_begin (fname, ebl, false);
Packit Service 97d2fb
  if (ctx == NULL)
Packit Service 97d2fb
    {
Packit Service 97d2fb
      printf ("cannot create assembler context: %s\n", asm_errmsg (-1));
Packit Service 97d2fb
      return 1;
Packit Service 97d2fb
    }
Packit Service 97d2fb
Packit Service 97d2fb
  /* Create two sections.  */
Packit Service 97d2fb
  scn1 = asm_newscn (ctx, ".data", SHT_PROGBITS, SHF_ALLOC | SHF_WRITE);
Packit Service 97d2fb
  scn2 = asm_newsubscn (scn1, 1);
Packit Service 97d2fb
  if (scn1 == NULL || scn2 == NULL)
Packit Service 97d2fb
    {
Packit Service 97d2fb
      printf ("cannot create section in output file: %s\n", asm_errmsg (-1));
Packit Service 97d2fb
      asm_abort (ctx);
Packit Service 97d2fb
      return 1;
Packit Service 97d2fb
    }
Packit Service 97d2fb
Packit Service 97d2fb
  /* Special alignment for the .text section.  */
Packit Service 97d2fb
  if (asm_align (scn1, 16) != 0)
Packit Service 97d2fb
    {
Packit Service 97d2fb
      printf ("cannot align .text section: %s\n", asm_errmsg (-1));
Packit Service 97d2fb
      result = 1;
Packit Service 97d2fb
    }
Packit Service 97d2fb
Packit Service 97d2fb
  /* Add a few strings.  */
Packit Service 97d2fb
  if (asm_addstrz (scn1, "one", 4) != 0)
Packit Service 97d2fb
    {
Packit Service 97d2fb
      printf ("cannot insert first string: %s\n", asm_errmsg (-1));
Packit Service 97d2fb
      result = 1;
Packit Service 97d2fb
    }
Packit Service 97d2fb
  if (asm_addstrz (scn2, "three", 0) != 0)
Packit Service 97d2fb
    {
Packit Service 97d2fb
      printf ("cannot insert second string: %s\n", asm_errmsg (-1));
Packit Service 97d2fb
      result = 1;
Packit Service 97d2fb
    }
Packit Service 97d2fb
  if (asm_addstrz (scn1, "two", 4) != 0)
Packit Service 97d2fb
    {
Packit Service 97d2fb
      printf ("cannot insert third string: %s\n", asm_errmsg (-1));
Packit Service 97d2fb
      result = 1;
Packit Service 97d2fb
    }
Packit Service 97d2fb
Packit Service 97d2fb
  /* Create the output file.  */
Packit Service 97d2fb
  if (asm_end (ctx) != 0)
Packit Service 97d2fb
    {
Packit Service 97d2fb
      printf ("cannot create output file: %s\n", asm_errmsg (-1));
Packit Service 97d2fb
      asm_abort (ctx);
Packit Service 97d2fb
      return 1;
Packit Service 97d2fb
    }
Packit Service 97d2fb
Packit Service 97d2fb
  /* Check the file.  */
Packit Service 97d2fb
  fd = open (fname, O_RDONLY);
Packit Service 97d2fb
  if (fd == -1)
Packit Service 97d2fb
    {
Packit Service 97d2fb
      printf ("cannot open generated file: %m\n");
Packit Service 97d2fb
      result = 1;
Packit Service 97d2fb
      goto out;
Packit Service 97d2fb
    }
Packit Service 97d2fb
Packit Service 97d2fb
  elf = elf_begin (fd, ELF_C_READ, NULL);
Packit Service 97d2fb
  if (elf == NULL)
Packit Service 97d2fb
    {
Packit Service 97d2fb
      printf ("cannot create ELF descriptor: %s\n", elf_errmsg (-1));
Packit Service 97d2fb
      result = 1;
Packit Service 97d2fb
      goto out_close;
Packit Service 97d2fb
    }
Packit Service 97d2fb
  if (elf_kind (elf) != ELF_K_ELF)
Packit Service 97d2fb
    {
Packit Service 97d2fb
      puts ("not a valid ELF file");
Packit Service 97d2fb
      result = 1;
Packit Service 97d2fb
      goto out_close2;
Packit Service 97d2fb
    }
Packit Service 97d2fb
Packit Service 97d2fb
  ehdr = gelf_getehdr (elf, &ehdr_mem);
Packit Service 97d2fb
  if (ehdr == NULL)
Packit Service 97d2fb
    {
Packit Service 97d2fb
      printf ("cannot get ELF header: %s\n", elf_errmsg (-1));
Packit Service 97d2fb
      result = 1;
Packit Service 97d2fb
      goto out_close2;
Packit Service 97d2fb
    }
Packit Service 97d2fb
Packit Service 97d2fb
  if (memcmp (ehdr, &expected_ehdr, sizeof (GElf_Ehdr)) != 0)
Packit Service 97d2fb
    {
Packit Service 97d2fb
      puts ("ELF header does not match");
Packit Service 97d2fb
      result = 1;
Packit Service 97d2fb
      goto out_close2;
Packit Service 97d2fb
    }
Packit Service 97d2fb
Packit Service 97d2fb
  for (cnt = 1; cnt < 3; ++cnt)
Packit Service 97d2fb
    {
Packit Service 97d2fb
      Elf_Scn *scn;
Packit Service 97d2fb
      GElf_Shdr shdr_mem;
Packit Service 97d2fb
      GElf_Shdr *shdr;
Packit Service 97d2fb
Packit Service 97d2fb
      scn = elf_getscn (elf, cnt);
Packit Service 97d2fb
      if (scn == NULL)
Packit Service 97d2fb
	{
Packit Service 97d2fb
	  printf ("cannot get section %zd: %s\n", cnt, elf_errmsg (-1));
Packit Service 97d2fb
	  result = 1;
Packit Service 97d2fb
	  continue;
Packit Service 97d2fb
	}
Packit Service 97d2fb
Packit Service 97d2fb
      shdr = gelf_getshdr (scn, &shdr_mem);
Packit Service 97d2fb
      if (shdr == NULL)
Packit Service 97d2fb
	{
Packit Service 97d2fb
	  printf ("cannot get section header for section %zd: %s\n",
Packit Service 97d2fb
		  cnt, elf_errmsg (-1));
Packit Service 97d2fb
	  result = 1;
Packit Service 97d2fb
	  continue;
Packit Service 97d2fb
	}
Packit Service 97d2fb
Packit Service 97d2fb
      if (strcmp (elf_strptr (elf, ehdr->e_shstrndx, shdr->sh_name),
Packit Service 97d2fb
		  scnnames[cnt]) != 0)
Packit Service 97d2fb
	{
Packit Service 97d2fb
	  printf ("section %zd's name differs: %s vs %s\n", cnt,
Packit Service 97d2fb
		  elf_strptr (elf, ehdr->e_shstrndx, shdr->sh_name),
Packit Service 97d2fb
		  scnnames[cnt]);
Packit Service 97d2fb
	  result = 1;
Packit Service 97d2fb
	}
Packit Service 97d2fb
Packit Service 97d2fb
      if (shdr->sh_type != (cnt == 2 ? SHT_STRTAB : SHT_PROGBITS))
Packit Service 97d2fb
	{
Packit Service 97d2fb
	  printf ("section %zd's type differs\n", cnt);
Packit Service 97d2fb
	  result = 1;
Packit Service 97d2fb
	}
Packit Service 97d2fb
Packit Service 97d2fb
      if ((cnt == 1 && shdr->sh_flags != (SHF_ALLOC | SHF_WRITE))
Packit Service 97d2fb
	  || (cnt == 2 && shdr->sh_flags != 0))
Packit Service 97d2fb
	{
Packit Service 97d2fb
	  printf ("section %zd's flags differs\n", cnt);
Packit Service 97d2fb
	  result = 1;
Packit Service 97d2fb
	}
Packit Service 97d2fb
Packit Service 97d2fb
      if (shdr->sh_addr != 0)
Packit Service 97d2fb
	{
Packit Service 97d2fb
	  printf ("section %zd's address differs\n", cnt);
Packit Service 97d2fb
	  result = 1;
Packit Service 97d2fb
	}
Packit Service 97d2fb
Packit Service 97d2fb
      if ((cnt == 1 && shdr->sh_offset != ((sizeof (Elf32_Ehdr) + 15) & ~15))
Packit Service 97d2fb
	  || (cnt == 2
Packit Service 97d2fb
	      && shdr->sh_offset != (((sizeof (Elf32_Ehdr) + 15) & ~15)
Packit Service 97d2fb
				     + strlen ("one") + 1
Packit Service 97d2fb
				     + strlen ("two") + 1
Packit Service 97d2fb
				     + strlen ("three") + 1)))
Packit Service 97d2fb
	{
Packit Service 97d2fb
	  printf ("section %zd's offset differs\n", cnt);
Packit Service 97d2fb
	  result = 1;
Packit Service 97d2fb
	}
Packit Service 97d2fb
Packit Service 97d2fb
      if ((cnt == 1 && shdr->sh_size != (strlen ("one") + 1
Packit Service 97d2fb
					 + strlen ("two") + 1
Packit Service 97d2fb
					 + strlen ("three") + 1))
Packit Service 97d2fb
	  || (cnt == 2 && shdr->sh_size != 17))
Packit Service 97d2fb
	{
Packit Service 97d2fb
	  printf ("section %zd's size differs\n", cnt);
Packit Service 97d2fb
	  result = 1;
Packit Service 97d2fb
	}
Packit Service 97d2fb
Packit Service 97d2fb
      if (shdr->sh_link != 0)
Packit Service 97d2fb
	{
Packit Service 97d2fb
	  printf ("section %zd's link differs\n", cnt);
Packit Service 97d2fb
	  result = 1;
Packit Service 97d2fb
	}
Packit Service 97d2fb
Packit Service 97d2fb
      if (shdr->sh_info != 0)
Packit Service 97d2fb
	{
Packit Service 97d2fb
	  printf ("section %zd's info differs\n", cnt);
Packit Service 97d2fb
	  result = 1;
Packit Service 97d2fb
	}
Packit Service 97d2fb
Packit Service 97d2fb
      if ((cnt == 1 && shdr->sh_addralign != 16)
Packit Service 97d2fb
	  || (cnt != 1 && shdr->sh_addralign != 1))
Packit Service 97d2fb
	{
Packit Service 97d2fb
	  printf ("section %zd's addralign differs\n", cnt);
Packit Service 97d2fb
	  result = 1;
Packit Service 97d2fb
	}
Packit Service 97d2fb
Packit Service 97d2fb
      if (shdr->sh_entsize != 0)
Packit Service 97d2fb
	{
Packit Service 97d2fb
	  printf ("section %zd's entsize differs\n", cnt);
Packit Service 97d2fb
	  result = 1;
Packit Service 97d2fb
	}
Packit Service 97d2fb
    }
Packit Service 97d2fb
Packit Service 97d2fb
 out_close2:
Packit Service 97d2fb
  elf_end (elf);
Packit Service 97d2fb
 out_close:
Packit Service 97d2fb
  close (fd);
Packit Service 97d2fb
 out:
Packit Service 97d2fb
  /* We don't need the file anymore.  */
Packit Service 97d2fb
  unlink (fname);
Packit Service 97d2fb
Packit Service 97d2fb
  ebl_closebackend (ebl);
Packit Service 97d2fb
Packit Service 97d2fb
  return result;
Packit Service 97d2fb
}