1/*	$NetBSD: gelf_ehdr.c,v 1.5 2024/03/03 17:37:34 christos Exp $	*/
2
3/*-
4 * Copyright (c) 2006,2008 Joseph Koshy
5 * All rights reserved.
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
9 * are met:
10 * 1. Redistributions of source code must retain the above copyright
11 *    notice, this list of conditions and the following disclaimer.
12 * 2. Redistributions in binary form must reproduce the above copyright
13 *    notice, this list of conditions and the following disclaimer in the
14 *    documentation and/or other materials provided with the distribution.
15 *
16 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
17 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
18 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
19 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
20 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
21 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
22 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
23 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
24 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
25 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
26 * SUCH DAMAGE.
27 */
28
29#if HAVE_NBTOOL_CONFIG_H
30# include "nbtool_config.h"
31#endif
32
33#include <sys/cdefs.h>
34
35#include <assert.h>
36#include <gelf.h>
37#include <libelf.h>
38#include <limits.h>
39#include <stdint.h>
40#include <string.h>
41
42#include "_libelf.h"
43
44__RCSID("$NetBSD: gelf_ehdr.c,v 1.5 2024/03/03 17:37:34 christos Exp $");
45ELFTC_VCSID("Id: gelf_ehdr.c 3977 2022-05-01 06:45:34Z jkoshy");
46
47Elf32_Ehdr *
48elf32_getehdr(Elf *e)
49{
50	return (_libelf_ehdr(e, ELFCLASS32, 0));
51}
52
53Elf64_Ehdr *
54elf64_getehdr(Elf *e)
55{
56	return (_libelf_ehdr(e, ELFCLASS64, 0));
57}
58
59GElf_Ehdr *
60gelf_getehdr(Elf *e, GElf_Ehdr *d)
61{
62	int ec;
63	Elf32_Ehdr *eh32;
64	Elf64_Ehdr *eh64;
65
66	if (d == NULL || e == NULL ||
67	    ((ec = e->e_class) != ELFCLASS32 && ec != ELFCLASS64)) {
68		LIBELF_SET_ERROR(ARGUMENT, 0);
69		return (NULL);
70	}
71
72	if (ec == ELFCLASS32) {
73		if ((eh32 = _libelf_ehdr(e, ELFCLASS32, 0)) == NULL)
74			return (NULL);
75
76		(void) memcpy(d->e_ident, eh32->e_ident,
77		    sizeof(eh32->e_ident));
78		d->e_type		= eh32->e_type;
79		d->e_machine		= eh32->e_machine;
80		d->e_version		= eh32->e_version;
81		d->e_entry		= eh32->e_entry;
82		d->e_phoff		= eh32->e_phoff;
83		d->e_shoff		= eh32->e_shoff;
84		d->e_flags		= eh32->e_flags;
85		d->e_ehsize		= eh32->e_ehsize;
86		d->e_phentsize		= eh32->e_phentsize;
87		d->e_phnum		= eh32->e_phnum;
88		d->e_shentsize		= eh32->e_shentsize;
89		d->e_shnum		= eh32->e_shnum;
90		d->e_shstrndx		= eh32->e_shstrndx;
91
92		return (d);
93	}
94
95	assert(ec == ELFCLASS64);
96
97	if ((eh64 = _libelf_ehdr(e, ELFCLASS64, 0)) == NULL)
98		return (NULL);
99	*d = *eh64;
100
101	return (d);
102}
103
104Elf32_Ehdr *
105elf32_newehdr(Elf *e)
106{
107	return (_libelf_ehdr(e, ELFCLASS32, 1));
108}
109
110Elf64_Ehdr *
111elf64_newehdr(Elf *e)
112{
113	return (_libelf_ehdr(e, ELFCLASS64, 1));
114}
115
116void *
117gelf_newehdr(Elf *e, int ec)
118{
119	if (e != NULL &&
120	    (ec == ELFCLASS32 || ec == ELFCLASS64))
121		return (_libelf_ehdr(e, ec, 1));
122
123	LIBELF_SET_ERROR(ARGUMENT, 0);
124	return (NULL);
125}
126
127int
128gelf_update_ehdr(Elf *e, GElf_Ehdr *s)
129{
130	int ec;
131	void *ehdr;
132	Elf32_Ehdr *eh32;
133	Elf64_Ehdr *eh64;
134
135	if (s== NULL || e == NULL || e->e_kind != ELF_K_ELF ||
136	    ((ec = e->e_class) != ELFCLASS32 && ec != ELFCLASS64)) {
137		LIBELF_SET_ERROR(ARGUMENT, 0);
138		return (0);
139	}
140
141	if (e->e_cmd == ELF_C_READ) {
142		LIBELF_SET_ERROR(MODE, 0);
143		return (0);
144	}
145
146	if ((ehdr = _libelf_ehdr(e, ec, 0)) == NULL)
147		return (0);
148
149	(void) elf_flagehdr(e, ELF_C_SET, ELF_F_DIRTY);
150
151	if (ec == ELFCLASS64) {
152		eh64 = (Elf64_Ehdr *) ehdr;
153		*eh64 = *s;
154		return (1);
155	}
156
157	eh32 = (Elf32_Ehdr *) ehdr;
158
159	(void) memcpy(eh32->e_ident, s->e_ident, sizeof(eh32->e_ident));
160
161	eh32->e_type      = s->e_type;
162	eh32->e_machine   = s->e_machine;
163	eh32->e_version   = s->e_version;
164	LIBELF_COPY_U32(eh32, s, e_entry);
165	LIBELF_COPY_U32(eh32, s, e_phoff);
166	LIBELF_COPY_U32(eh32, s, e_shoff);
167	eh32->e_flags     = s->e_flags;
168	eh32->e_ehsize    = s->e_ehsize;
169	eh32->e_phentsize = s->e_phentsize;
170	eh32->e_phnum     = s->e_phnum;
171	eh32->e_shentsize = s->e_shentsize;
172	eh32->e_shnum     = s->e_shnum;
173	eh32->e_shstrndx  = s->e_shstrndx;
174
175	return (1);
176}
177