1*f8fb3368SJohn Marino /*-
2*f8fb3368SJohn Marino * Copyright (c) 2006,2008,2010 Joseph Koshy
3*f8fb3368SJohn Marino * All rights reserved.
4*f8fb3368SJohn Marino *
5*f8fb3368SJohn Marino * Redistribution and use in source and binary forms, with or without
6*f8fb3368SJohn Marino * modification, are permitted provided that the following conditions
7*f8fb3368SJohn Marino * are met:
8*f8fb3368SJohn Marino * 1. Redistributions of source code must retain the above copyright
9*f8fb3368SJohn Marino * notice, this list of conditions and the following disclaimer.
10*f8fb3368SJohn Marino * 2. Redistributions in binary form must reproduce the above copyright
11*f8fb3368SJohn Marino * notice, this list of conditions and the following disclaimer in the
12*f8fb3368SJohn Marino * documentation and/or other materials provided with the distribution.
13*f8fb3368SJohn Marino *
14*f8fb3368SJohn Marino * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15*f8fb3368SJohn Marino * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16*f8fb3368SJohn Marino * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17*f8fb3368SJohn Marino * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18*f8fb3368SJohn Marino * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19*f8fb3368SJohn Marino * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20*f8fb3368SJohn Marino * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21*f8fb3368SJohn Marino * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22*f8fb3368SJohn Marino * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23*f8fb3368SJohn Marino * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24*f8fb3368SJohn Marino * SUCH DAMAGE.
25*f8fb3368SJohn Marino */
26*f8fb3368SJohn Marino
27*f8fb3368SJohn Marino /*
28*f8fb3368SJohn Marino * Internal APIs
29*f8fb3368SJohn Marino */
30*f8fb3368SJohn Marino
31*f8fb3368SJohn Marino #include <assert.h>
32*f8fb3368SJohn Marino #include <errno.h>
33*f8fb3368SJohn Marino #include <libelf.h>
34*f8fb3368SJohn Marino #include <stdlib.h>
35*f8fb3368SJohn Marino #include <string.h>
36*f8fb3368SJohn Marino
37*f8fb3368SJohn Marino #include "_libelf.h"
38*f8fb3368SJohn Marino
39*f8fb3368SJohn Marino ELFTC_VCSID("$Id: libelf_allocate.c 3174 2015-03-27 17:13:41Z emaste $");
40*f8fb3368SJohn Marino
41*f8fb3368SJohn Marino Elf *
_libelf_allocate_elf(void)42*f8fb3368SJohn Marino _libelf_allocate_elf(void)
43*f8fb3368SJohn Marino {
44*f8fb3368SJohn Marino Elf *e;
45*f8fb3368SJohn Marino
46*f8fb3368SJohn Marino if ((e = malloc(sizeof(*e))) == NULL) {
47*f8fb3368SJohn Marino LIBELF_SET_ERROR(RESOURCE, errno);
48*f8fb3368SJohn Marino return NULL;
49*f8fb3368SJohn Marino }
50*f8fb3368SJohn Marino
51*f8fb3368SJohn Marino e->e_activations = 1;
52*f8fb3368SJohn Marino e->e_hdr.e_rawhdr = NULL;
53*f8fb3368SJohn Marino e->e_byteorder = ELFDATANONE;
54*f8fb3368SJohn Marino e->e_class = ELFCLASSNONE;
55*f8fb3368SJohn Marino e->e_cmd = ELF_C_NULL;
56*f8fb3368SJohn Marino e->e_fd = -1;
57*f8fb3368SJohn Marino e->e_flags = 0;
58*f8fb3368SJohn Marino e->e_kind = ELF_K_NONE;
59*f8fb3368SJohn Marino e->e_parent = NULL;
60*f8fb3368SJohn Marino e->e_rawfile = NULL;
61*f8fb3368SJohn Marino e->e_rawsize = 0;
62*f8fb3368SJohn Marino e->e_version = LIBELF_PRIVATE(version);
63*f8fb3368SJohn Marino
64*f8fb3368SJohn Marino (void) memset(&e->e_u, 0, sizeof(e->e_u));
65*f8fb3368SJohn Marino
66*f8fb3368SJohn Marino return (e);
67*f8fb3368SJohn Marino }
68*f8fb3368SJohn Marino
69*f8fb3368SJohn Marino void
_libelf_init_elf(Elf * e,Elf_Kind kind)70*f8fb3368SJohn Marino _libelf_init_elf(Elf *e, Elf_Kind kind)
71*f8fb3368SJohn Marino {
72*f8fb3368SJohn Marino assert(e != NULL);
73*f8fb3368SJohn Marino assert(e->e_kind == ELF_K_NONE);
74*f8fb3368SJohn Marino
75*f8fb3368SJohn Marino e->e_kind = kind;
76*f8fb3368SJohn Marino
77*f8fb3368SJohn Marino switch (kind) {
78*f8fb3368SJohn Marino case ELF_K_ELF:
79*f8fb3368SJohn Marino STAILQ_INIT(&e->e_u.e_elf.e_scn);
80*f8fb3368SJohn Marino break;
81*f8fb3368SJohn Marino default:
82*f8fb3368SJohn Marino break;
83*f8fb3368SJohn Marino }
84*f8fb3368SJohn Marino }
85*f8fb3368SJohn Marino
86*f8fb3368SJohn Marino #define FREE(P) do { \
87*f8fb3368SJohn Marino if (P) \
88*f8fb3368SJohn Marino free(P); \
89*f8fb3368SJohn Marino } while (0)
90*f8fb3368SJohn Marino
91*f8fb3368SJohn Marino
92*f8fb3368SJohn Marino Elf *
_libelf_release_elf(Elf * e)93*f8fb3368SJohn Marino _libelf_release_elf(Elf *e)
94*f8fb3368SJohn Marino {
95*f8fb3368SJohn Marino Elf_Arhdr *arh;
96*f8fb3368SJohn Marino
97*f8fb3368SJohn Marino switch (e->e_kind) {
98*f8fb3368SJohn Marino case ELF_K_AR:
99*f8fb3368SJohn Marino FREE(e->e_u.e_ar.e_symtab);
100*f8fb3368SJohn Marino break;
101*f8fb3368SJohn Marino
102*f8fb3368SJohn Marino case ELF_K_ELF:
103*f8fb3368SJohn Marino switch (e->e_class) {
104*f8fb3368SJohn Marino case ELFCLASS32:
105*f8fb3368SJohn Marino FREE(e->e_u.e_elf.e_ehdr.e_ehdr32);
106*f8fb3368SJohn Marino FREE(e->e_u.e_elf.e_phdr.e_phdr32);
107*f8fb3368SJohn Marino break;
108*f8fb3368SJohn Marino case ELFCLASS64:
109*f8fb3368SJohn Marino FREE(e->e_u.e_elf.e_ehdr.e_ehdr64);
110*f8fb3368SJohn Marino FREE(e->e_u.e_elf.e_phdr.e_phdr64);
111*f8fb3368SJohn Marino break;
112*f8fb3368SJohn Marino }
113*f8fb3368SJohn Marino
114*f8fb3368SJohn Marino assert(STAILQ_EMPTY(&e->e_u.e_elf.e_scn));
115*f8fb3368SJohn Marino
116*f8fb3368SJohn Marino if (e->e_flags & LIBELF_F_AR_HEADER) {
117*f8fb3368SJohn Marino arh = e->e_hdr.e_arhdr;
118*f8fb3368SJohn Marino FREE(arh->ar_name);
119*f8fb3368SJohn Marino FREE(arh->ar_rawname);
120*f8fb3368SJohn Marino free(arh);
121*f8fb3368SJohn Marino }
122*f8fb3368SJohn Marino
123*f8fb3368SJohn Marino break;
124*f8fb3368SJohn Marino
125*f8fb3368SJohn Marino default:
126*f8fb3368SJohn Marino break;
127*f8fb3368SJohn Marino }
128*f8fb3368SJohn Marino
129*f8fb3368SJohn Marino free(e);
130*f8fb3368SJohn Marino
131*f8fb3368SJohn Marino return (NULL);
132*f8fb3368SJohn Marino }
133*f8fb3368SJohn Marino
134*f8fb3368SJohn Marino struct _Libelf_Data *
_libelf_allocate_data(Elf_Scn * s)135*f8fb3368SJohn Marino _libelf_allocate_data(Elf_Scn *s)
136*f8fb3368SJohn Marino {
137*f8fb3368SJohn Marino struct _Libelf_Data *d;
138*f8fb3368SJohn Marino
139*f8fb3368SJohn Marino if ((d = calloc((size_t) 1, sizeof(*d))) == NULL) {
140*f8fb3368SJohn Marino LIBELF_SET_ERROR(RESOURCE, 0);
141*f8fb3368SJohn Marino return (NULL);
142*f8fb3368SJohn Marino }
143*f8fb3368SJohn Marino
144*f8fb3368SJohn Marino d->d_scn = s;
145*f8fb3368SJohn Marino
146*f8fb3368SJohn Marino return (d);
147*f8fb3368SJohn Marino }
148*f8fb3368SJohn Marino
149*f8fb3368SJohn Marino struct _Libelf_Data *
_libelf_release_data(struct _Libelf_Data * d)150*f8fb3368SJohn Marino _libelf_release_data(struct _Libelf_Data *d)
151*f8fb3368SJohn Marino {
152*f8fb3368SJohn Marino
153*f8fb3368SJohn Marino if (d->d_flags & LIBELF_F_DATA_MALLOCED)
154*f8fb3368SJohn Marino free(d->d_data.d_buf);
155*f8fb3368SJohn Marino
156*f8fb3368SJohn Marino free(d);
157*f8fb3368SJohn Marino
158*f8fb3368SJohn Marino return (NULL);
159*f8fb3368SJohn Marino }
160*f8fb3368SJohn Marino
161*f8fb3368SJohn Marino Elf_Scn *
_libelf_allocate_scn(Elf * e,size_t ndx)162*f8fb3368SJohn Marino _libelf_allocate_scn(Elf *e, size_t ndx)
163*f8fb3368SJohn Marino {
164*f8fb3368SJohn Marino Elf_Scn *s;
165*f8fb3368SJohn Marino
166*f8fb3368SJohn Marino if ((s = calloc((size_t) 1, sizeof(Elf_Scn))) == NULL) {
167*f8fb3368SJohn Marino LIBELF_SET_ERROR(RESOURCE, errno);
168*f8fb3368SJohn Marino return (NULL);
169*f8fb3368SJohn Marino }
170*f8fb3368SJohn Marino
171*f8fb3368SJohn Marino s->s_elf = e;
172*f8fb3368SJohn Marino s->s_ndx = ndx;
173*f8fb3368SJohn Marino
174*f8fb3368SJohn Marino STAILQ_INIT(&s->s_data);
175*f8fb3368SJohn Marino STAILQ_INIT(&s->s_rawdata);
176*f8fb3368SJohn Marino
177*f8fb3368SJohn Marino STAILQ_INSERT_TAIL(&e->e_u.e_elf.e_scn, s, s_next);
178*f8fb3368SJohn Marino
179*f8fb3368SJohn Marino return (s);
180*f8fb3368SJohn Marino }
181*f8fb3368SJohn Marino
182*f8fb3368SJohn Marino Elf_Scn *
_libelf_release_scn(Elf_Scn * s)183*f8fb3368SJohn Marino _libelf_release_scn(Elf_Scn *s)
184*f8fb3368SJohn Marino {
185*f8fb3368SJohn Marino Elf *e;
186*f8fb3368SJohn Marino struct _Libelf_Data *d, *td;
187*f8fb3368SJohn Marino
188*f8fb3368SJohn Marino assert(s != NULL);
189*f8fb3368SJohn Marino
190*f8fb3368SJohn Marino STAILQ_FOREACH_SAFE(d, &s->s_data, d_next, td) {
191*f8fb3368SJohn Marino STAILQ_REMOVE(&s->s_data, d, _Libelf_Data, d_next);
192*f8fb3368SJohn Marino d = _libelf_release_data(d);
193*f8fb3368SJohn Marino }
194*f8fb3368SJohn Marino
195*f8fb3368SJohn Marino STAILQ_FOREACH_SAFE(d, &s->s_rawdata, d_next, td) {
196*f8fb3368SJohn Marino assert((d->d_flags & LIBELF_F_DATA_MALLOCED) == 0);
197*f8fb3368SJohn Marino STAILQ_REMOVE(&s->s_rawdata, d, _Libelf_Data, d_next);
198*f8fb3368SJohn Marino d = _libelf_release_data(d);
199*f8fb3368SJohn Marino }
200*f8fb3368SJohn Marino
201*f8fb3368SJohn Marino e = s->s_elf;
202*f8fb3368SJohn Marino
203*f8fb3368SJohn Marino assert(e != NULL);
204*f8fb3368SJohn Marino
205*f8fb3368SJohn Marino STAILQ_REMOVE(&e->e_u.e_elf.e_scn, s, _Elf_Scn, s_next);
206*f8fb3368SJohn Marino
207*f8fb3368SJohn Marino free(s);
208*f8fb3368SJohn Marino
209*f8fb3368SJohn Marino return (NULL);
210*f8fb3368SJohn Marino }
211