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
main(void)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 }
279