1/*	$NetBSD: gelf_ehdr.c,v 1.4 2010/02/22 10:48:32 darran Exp $	*/
2
3/*-
4 * Copyright (c) 2006 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/* __FBSDID("$FreeBSD: src/lib/libelf/gelf_ehdr.c,v 1.1.10.1.2.1 2009/10/25 01:10:29 kensmith Exp $"); */
35
36#include <assert.h>
37#include <limits.h>
38#include <gelf.h>
39#include <libelf.h>
40#include <string.h>
41
42#include "_libelf.h"
43
44Elf32_Ehdr *
45elf32_getehdr(Elf *e)
46{
47	return (_libelf_ehdr(e, ELFCLASS32, 0));
48}
49
50Elf64_Ehdr *
51elf64_getehdr(Elf *e)
52{
53	return (_libelf_ehdr(e, ELFCLASS64, 0));
54}
55
56GElf_Ehdr *
57gelf_getehdr(Elf *e, GElf_Ehdr *d)
58{
59	int ec;
60	Elf32_Ehdr *eh32;
61	Elf64_Ehdr *eh64;
62
63	if (d == NULL || e == NULL ||
64	    ((ec = e->e_class) != ELFCLASS32 && ec != ELFCLASS64)) {
65		LIBELF_SET_ERROR(ARGUMENT, 0);
66		return (NULL);
67	}
68
69	if (ec == ELFCLASS32) {
70		if ((eh32 = _libelf_ehdr(e, ELFCLASS32, 0)) == NULL)
71			return (NULL);
72
73		(void) memcpy(d->e_ident, eh32->e_ident, sizeof(eh32->e_ident));
74		d->e_type		= eh32->e_type;
75		d->e_machine		= eh32->e_machine;
76		d->e_version		= eh32->e_version;
77		d->e_entry		= eh32->e_entry;
78		d->e_phoff		= eh32->e_phoff;
79		d->e_shoff		= eh32->e_shoff;
80		d->e_flags		= eh32->e_flags;
81		d->e_ehsize		= eh32->e_ehsize;
82		d->e_phentsize		= eh32->e_phentsize;
83		d->e_phnum		= eh32->e_phnum;
84		d->e_shentsize		= eh32->e_shentsize;
85		d->e_shnum		= eh32->e_shnum;
86		d->e_shstrndx		= eh32->e_shstrndx;
87
88		return (d);
89	}
90
91	assert(ec == ELFCLASS64);
92
93	if ((eh64 = _libelf_ehdr(e, ELFCLASS64, 0)) == NULL)
94		return (NULL);
95	*d = *eh64;
96
97	return (d);
98}
99
100Elf32_Ehdr *
101elf32_newehdr(Elf *e)
102{
103	return (_libelf_ehdr(e, ELFCLASS32, 1));
104}
105
106Elf64_Ehdr *
107elf64_newehdr(Elf *e)
108{
109	return (_libelf_ehdr(e, ELFCLASS64, 1));
110}
111
112void *
113gelf_newehdr(Elf *e, int ec)
114{
115	if (e != NULL &&
116	    (ec == ELFCLASS32 || ec == ELFCLASS64))
117		return (_libelf_ehdr(e, ec, 1));
118
119	LIBELF_SET_ERROR(ARGUMENT, 0);
120	return (NULL);
121}
122
123int
124gelf_update_ehdr(Elf *e, GElf_Ehdr *s)
125{
126	int ec;
127	void *ehdr;
128	Elf32_Ehdr *eh32;
129	Elf64_Ehdr *eh64;
130
131	if (s== NULL || e == NULL || e->e_kind != ELF_K_ELF ||
132	    ((ec = e->e_class) != ELFCLASS32 && ec != ELFCLASS64)) {
133		LIBELF_SET_ERROR(ARGUMENT, 0);
134		return (0);
135	}
136
137	if (e->e_cmd == ELF_C_READ) {
138		LIBELF_SET_ERROR(MODE, 0);
139		return (0);
140	}
141
142	if ((ehdr = _libelf_ehdr(e, ec, 0)) == NULL)
143		return (0);
144
145	if (ec == ELFCLASS64) {
146		eh64 = (Elf64_Ehdr *) ehdr;
147		*eh64 = *s;
148		return (1);
149	}
150
151	eh32 = (Elf32_Ehdr *) ehdr;
152
153	(void) memcpy(eh32->e_ident, s->e_ident, sizeof(eh32->e_ident));
154
155	eh32->e_type      = s->e_type;
156	eh32->e_machine   = s->e_machine;
157	eh32->e_version   = s->e_version;
158	LIBELF_COPY_U32(eh32, s, e_entry);
159	LIBELF_COPY_U32(eh32, s, e_phoff);
160	LIBELF_COPY_U32(eh32, s, e_shoff);
161	eh32->e_flags     = s->e_flags;
162	eh32->e_ehsize    = s->e_ehsize;
163	eh32->e_phentsize = s->e_phentsize;
164	eh32->e_phnum     = s->e_phnum;
165	eh32->e_shentsize = s->e_shentsize;
166	eh32->e_shnum     = s->e_shnum;
167	eh32->e_shstrndx  = s->e_shstrndx;
168
169	(void) elf_flagehdr(e, ELF_C_SET, ELF_F_DIRTY);
170
171	return (1);
172}
173