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 <inttypes.h>
24#include ELFUTILS_HEADER(asm)
25#include <libelf.h>
26#include <stdio.h>
27#include <unistd.h>
28
29
30static const char fname[] = "asm-tst7-out.o";
31
32
33int
34main (void)
35{
36  int result = 0;
37  size_t cnt;
38  AsmCtx_t *ctx;
39  Elf *elf;
40  int fd;
41
42  elf_version (EV_CURRENT);
43
44  Ebl *ebl = ebl_openbackend_machine (EM_386);
45  if (ebl == NULL)
46    {
47      puts ("cannot open backend library");
48      return 1;
49    }
50
51  ctx = asm_begin (fname, ebl, false);
52  if (ctx == NULL)
53    {
54      printf ("cannot create assembler context: %s\n", asm_errmsg (-1));
55      return 1;
56    }
57
58  if (asm_newcomsym (ctx, "commsym", 4, 16) == NULL)
59    {
60      printf ("cannot create common symbol: %s\n", asm_errmsg (-1));
61      asm_abort (ctx);
62      return 1;
63    }
64
65  /* Create the output file.  */
66  if (asm_end (ctx) != 0)
67    {
68      printf ("cannot create output file: %s\n", asm_errmsg (-1));
69      asm_abort (ctx);
70      return 1;
71    }
72
73  /* Check the file.  */
74  fd = open (fname, O_RDONLY);
75  if (fd == -1)
76    {
77      printf ("cannot open generated file: %m\n");
78      result = 1;
79      goto out;
80    }
81
82  elf = elf_begin (fd, ELF_C_READ, NULL);
83  if (elf == NULL)
84    {
85      printf ("cannot create ELF descriptor: %s\n", elf_errmsg (-1));
86      result = 1;
87      goto out_close;
88    }
89  if (elf_kind (elf) != ELF_K_ELF)
90    {
91      puts ("not a valid ELF file");
92      result = 1;
93      goto out_close2;
94    }
95
96  for (cnt = 1; 1; ++cnt)
97    {
98      Elf_Scn *scn;
99      GElf_Shdr shdr_mem;
100      GElf_Shdr *shdr;
101
102      scn = elf_getscn (elf, cnt);
103      if (scn == NULL)
104	{
105	  printf ("cannot get section %zd: %s\n", cnt, elf_errmsg (-1));
106	  result = 1;
107	  continue;
108	}
109
110      shdr = gelf_getshdr (scn, &shdr_mem);
111      if (shdr == NULL)
112	{
113	  printf ("cannot get section header for section %zd: %s\n",
114		  cnt, elf_errmsg (-1));
115	  result = 1;
116	  continue;
117	}
118      /* We are looking for the symbol table.  */
119      if (shdr->sh_type != SHT_SYMTAB)
120	continue;
121
122      for (cnt = 1; cnt< (shdr->sh_size
123			  / gelf_fsize (elf, ELF_T_SYM, 1, EV_CURRENT));
124	   ++cnt)
125	{
126	  GElf_Sym sym_mem;
127	  GElf_Sym *sym;
128
129	  if (cnt > 1)
130	    {
131	      puts ("too many symbol");
132	      result = 1;
133	      break;
134	    }
135
136	  sym = gelf_getsym (elf_getdata (scn, NULL), cnt, &sym_mem);
137	  if (sym == NULL)
138	    {
139	      printf ("cannot get symbol %zu: %s\n", cnt, elf_errmsg (-1));
140	      result = 1;
141	    }
142	  else
143	    {
144	      if (sym->st_shndx != SHN_COMMON)
145		{
146		  printf ("expected common symbol, got section %u\n",
147			  (unsigned int) sym->st_shndx);
148		  result = 1;
149		}
150
151	      if (sym->st_value != 16)
152		{
153		  printf ("requested alignment 16, is %" PRIuMAX "\n",
154			  (uintmax_t) sym->st_value);
155		  result = 1;
156		}
157
158	      if (sym->st_size != 4)
159		{
160		  printf ("requested size 4, is %" PRIuMAX "\n",
161			  (uintmax_t) sym->st_value);
162		  result = 1;
163		}
164	    }
165	}
166
167      break;
168    }
169
170 out_close2:
171  elf_end (elf);
172 out_close:
173  close (fd);
174 out:
175  /* We don't need the file anymore.  */
176  unlink (fname);
177
178  ebl_closebackend (ebl);
179
180  return result;
181}
182