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 <gelf.h>
32*573456a9SEd Maste #include <libelf.h>
33*573456a9SEd Maste
34*573456a9SEd Maste #include "_libelf.h"
35*573456a9SEd Maste
36*573456a9SEd Maste void *
_libelf_getchdr(Elf_Scn * s,int ec)37*573456a9SEd Maste _libelf_getchdr(Elf_Scn *s, int ec)
38*573456a9SEd Maste {
39*573456a9SEd Maste Elf *e;
40*573456a9SEd Maste void *sh;
41*573456a9SEd Maste Elf32_Shdr *sh32;
42*573456a9SEd Maste Elf64_Shdr *sh64;
43*573456a9SEd Maste
44*573456a9SEd Maste sh32 = NULL;
45*573456a9SEd Maste sh64 = NULL;
46*573456a9SEd Maste
47*573456a9SEd Maste if (s == NULL || (e = s->s_elf) == NULL || e->e_kind != ELF_K_ELF) {
48*573456a9SEd Maste LIBELF_SET_ERROR(ARGUMENT, 0);
49*573456a9SEd Maste return (NULL);
50*573456a9SEd Maste }
51*573456a9SEd Maste
52*573456a9SEd Maste if (ec == ELFCLASSNONE) {
53*573456a9SEd Maste ec = e->e_class;
54*573456a9SEd Maste } else if (ec != e->e_class) {
55*573456a9SEd Maste LIBELF_SET_ERROR(CLASS, 0);
56*573456a9SEd Maste return (NULL);
57*573456a9SEd Maste }
58*573456a9SEd Maste
59*573456a9SEd Maste if ((sh = _libelf_getshdr(s, ec)) == NULL) {
60*573456a9SEd Maste LIBELF_SET_ERROR(HEADER, 0);
61*573456a9SEd Maste return (NULL);
62*573456a9SEd Maste }
63*573456a9SEd Maste
64*573456a9SEd Maste if (ec == ELFCLASS32) {
65*573456a9SEd Maste sh32 = (Elf32_Shdr *)sh;
66*573456a9SEd Maste if ((sh32->sh_flags & SHF_ALLOC) != 0) {
67*573456a9SEd Maste LIBELF_SET_ERROR(INVALID_SECTION_FLAGS, 0);
68*573456a9SEd Maste return (NULL);
69*573456a9SEd Maste }
70*573456a9SEd Maste
71*573456a9SEd Maste if (sh32->sh_type == SHT_NULL || sh32->sh_type == SHT_NOBITS) {
72*573456a9SEd Maste LIBELF_SET_ERROR(INVALID_SECTION_TYPE, 0);
73*573456a9SEd Maste return (NULL);
74*573456a9SEd Maste }
75*573456a9SEd Maste
76*573456a9SEd Maste if ((sh32->sh_flags & SHF_COMPRESSED) == 0) {
77*573456a9SEd Maste LIBELF_SET_ERROR(NOT_COMPRESSED, 0);
78*573456a9SEd Maste return (NULL);
79*573456a9SEd Maste }
80*573456a9SEd Maste } else {
81*573456a9SEd Maste sh64 = (Elf64_Shdr *)sh;
82*573456a9SEd Maste if ((sh64->sh_flags & SHF_ALLOC) != 0) {
83*573456a9SEd Maste LIBELF_SET_ERROR(INVALID_SECTION_FLAGS, 0);
84*573456a9SEd Maste return (NULL);
85*573456a9SEd Maste }
86*573456a9SEd Maste
87*573456a9SEd Maste if (sh64->sh_type == SHT_NULL || sh64->sh_type == SHT_NOBITS) {
88*573456a9SEd Maste LIBELF_SET_ERROR(INVALID_SECTION_TYPE, 0);
89*573456a9SEd Maste return (NULL);
90*573456a9SEd Maste }
91*573456a9SEd Maste
92*573456a9SEd Maste if ((sh64->sh_flags & SHF_COMPRESSED) == 0) {
93*573456a9SEd Maste LIBELF_SET_ERROR(NOT_COMPRESSED, 0);
94*573456a9SEd Maste return (NULL);
95*573456a9SEd Maste }
96*573456a9SEd Maste }
97*573456a9SEd Maste
98*573456a9SEd Maste Elf_Data *d = elf_getdata(s, NULL);
99*573456a9SEd Maste
100*573456a9SEd Maste if (!d)
101*573456a9SEd Maste return (NULL);
102*573456a9SEd Maste
103*573456a9SEd Maste return ((void *)d->d_buf);
104*573456a9SEd Maste }
105