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