1*573456a9SEd Maste /*-
2*573456a9SEd Maste * SPDX-License-Identifier: BSD-2-Clause
3*573456a9SEd Maste *
4*573456a9SEd Maste * Copyright (c) 2020 The FreeBSD Foundation
5*573456a9SEd Maste *
6*573456a9SEd Maste * This software was developed by Tiger Gao under sponsorship from
7*573456a9SEd Maste * the FreeBSD Foundation.
8*573456a9SEd Maste *
9*573456a9SEd Maste * Redistribution and use in source and binary forms, with or without
10*573456a9SEd Maste * modification, are permitted provided that the following conditions are
11*573456a9SEd Maste * met:
12*573456a9SEd Maste * 1. Redistributions of source code must retain the above copyright
13*573456a9SEd Maste * notice, this list of conditions and the following disclaimer.
14*573456a9SEd Maste * 2. Redistributions in binary form must reproduce the above copyright
15*573456a9SEd Maste * notice, this list of conditions and the following disclaimer in
16*573456a9SEd Maste * the documentation and/or other materials provided with the distribution.
17*573456a9SEd Maste *
18*573456a9SEd Maste * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
19*573456a9SEd Maste * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
20*573456a9SEd Maste * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
21*573456a9SEd Maste * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
22*573456a9SEd Maste * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
23*573456a9SEd Maste * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
24*573456a9SEd Maste * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
25*573456a9SEd Maste * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
26*573456a9SEd Maste * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
27*573456a9SEd Maste * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
28*573456a9SEd Maste * SUCH DAMAGE.
29*573456a9SEd Maste */
30*573456a9SEd Maste
31*573456a9SEd Maste #include <assert.h>
32*573456a9SEd Maste #include <gelf.h>
33*573456a9SEd Maste #include <libelf.h>
34*573456a9SEd Maste #include <limits.h>
35*573456a9SEd Maste #include <stdint.h>
36*573456a9SEd Maste
37*573456a9SEd Maste #include "_libelf.h"
38*573456a9SEd Maste
39*573456a9SEd Maste Elf32_Chdr *
elf32_getchdr(Elf_Scn * s)40*573456a9SEd Maste elf32_getchdr(Elf_Scn *s)
41*573456a9SEd Maste {
42*573456a9SEd Maste return (_libelf_getchdr(s, ELFCLASS32));
43*573456a9SEd Maste }
44*573456a9SEd Maste
45*573456a9SEd Maste Elf64_Chdr *
elf64_getchdr(Elf_Scn * s)46*573456a9SEd Maste elf64_getchdr(Elf_Scn *s)
47*573456a9SEd Maste {
48*573456a9SEd Maste return (_libelf_getchdr(s, ELFCLASS64));
49*573456a9SEd Maste }
50*573456a9SEd Maste
51*573456a9SEd Maste GElf_Chdr *
gelf_getchdr(Elf_Scn * s,GElf_Chdr * d)52*573456a9SEd Maste gelf_getchdr(Elf_Scn *s, GElf_Chdr *d)
53*573456a9SEd Maste {
54*573456a9SEd Maste int ec;
55*573456a9SEd Maste void *ch;
56*573456a9SEd Maste Elf32_Chdr *ch32;
57*573456a9SEd Maste Elf64_Chdr *ch64;
58*573456a9SEd Maste
59*573456a9SEd Maste if (d == NULL) {
60*573456a9SEd Maste LIBELF_SET_ERROR(ARGUMENT, 0);
61*573456a9SEd Maste return (NULL);
62*573456a9SEd Maste }
63*573456a9SEd Maste
64*573456a9SEd Maste if ((ch = _libelf_getchdr(s, ELFCLASSNONE)) == NULL)
65*573456a9SEd Maste return (NULL);
66*573456a9SEd Maste
67*573456a9SEd Maste ec = s->s_elf->e_class;
68*573456a9SEd Maste assert(ec == ELFCLASS32 || ec == ELFCLASS64);
69*573456a9SEd Maste
70*573456a9SEd Maste if (ec == ELFCLASS32) {
71*573456a9SEd Maste ch32 = (Elf32_Chdr *)ch;
72*573456a9SEd Maste
73*573456a9SEd Maste d->ch_type = (Elf64_Word)ch32->ch_type;
74*573456a9SEd Maste d->ch_size = (Elf64_Xword)ch32->ch_size;
75*573456a9SEd Maste d->ch_addralign = (Elf64_Xword)ch32->ch_addralign;
76*573456a9SEd Maste } else {
77*573456a9SEd Maste ch64 = (Elf64_Chdr *)ch;
78*573456a9SEd Maste *d = *ch64;
79*573456a9SEd Maste }
80*573456a9SEd Maste
81*573456a9SEd Maste return (d);
82*573456a9SEd Maste }
83