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