gelf_shdr.c revision 276371 
1127376Srik/*-
2127376Srik * Copyright (c) 2006,2008 Joseph Koshy
3139823Simp * All rights reserved.
4139823Simp *
5139823Simp * Redistribution and use in source and binary forms, with or without
6127376Srik * modification, are permitted provided that the following conditions
7127376Srik * are met:
8127376Srik * 1. Redistributions of source code must retain the above copyright
9127376Srik *    notice, this list of conditions and the following disclaimer.
10127376Srik * 2. Redistributions in binary form must reproduce the above copyright
11127376Srik *    notice, this list of conditions and the following disclaimer in the
12127376Srik *    documentation and/or other materials provided with the distribution.
13127376Srik *
14127376Srik * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15127376Srik * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16127376Srik * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17127376Srik * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18127376Srik * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19127376Srik * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20127376Srik * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21127376Srik * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22127376Srik * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23127376Srik * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24127376Srik * SUCH DAMAGE.
25127376Srik */
26127376Srik
27127376Srik#include <sys/cdefs.h>
28127376Srik
29127376Srik#include <assert.h>
30127376Srik#include <gelf.h>
31127376Srik#include <libelf.h>
32127376Srik#include <limits.h>
33127376Srik#include <stdint.h>
34127376Srik
35127376Srik#include "_libelf.h"
36127376Srik
37127376SrikELFTC_VCSID("$Id: gelf_shdr.c 2268 2011-12-03 17:05:11Z jkoshy $");
38127376Srik
39127376SrikElf32_Shdr *
40elf32_getshdr(Elf_Scn *s)
41{
42	return (_libelf_getshdr(s, ELFCLASS32));
43}
44
45Elf64_Shdr *
46elf64_getshdr(Elf_Scn *s)
47{
48	return (_libelf_getshdr(s, ELFCLASS64));
49}
50
51GElf_Shdr *
52gelf_getshdr(Elf_Scn *s, GElf_Shdr *d)
53{
54	int ec;
55	void *sh;
56	Elf32_Shdr *sh32;
57	Elf64_Shdr *sh64;
58
59	if (d == NULL) {
60		LIBELF_SET_ERROR(ARGUMENT, 0);
61		return (NULL);
62	}
63
64	if ((sh = _libelf_getshdr(s, ELFCLASSNONE)) == NULL)
65		return (NULL);
66
67	ec = s->s_elf->e_class;
68	assert(ec == ELFCLASS32 || ec == ELFCLASS64);
69
70	if (ec == ELFCLASS32) {
71		sh32 = (Elf32_Shdr *) sh;
72
73		d->sh_name      = sh32->sh_name;
74		d->sh_type      = sh32->sh_type;
75		d->sh_flags     = (Elf64_Xword) sh32->sh_flags;
76		d->sh_addr      = (Elf64_Addr) sh32->sh_addr;
77		d->sh_offset    = (Elf64_Off) sh32->sh_offset;
78		d->sh_size      = (Elf64_Xword) sh32->sh_size;
79		d->sh_link      = sh32->sh_link;
80		d->sh_info      = sh32->sh_info;
81		d->sh_addralign = (Elf64_Xword) sh32->sh_addralign;
82		d->sh_entsize   = (Elf64_Xword) sh32->sh_entsize;
83	} else {
84		sh64 = (Elf64_Shdr *) sh;
85		*d = *sh64;
86	}
87
88	return (d);
89}
90
91int
92gelf_update_shdr(Elf_Scn *scn, GElf_Shdr *s)
93{
94	int ec;
95	Elf *e;
96	Elf32_Shdr *sh32;
97
98
99	if (s == NULL || scn == NULL || (e = scn->s_elf) == NULL ||
100	    e->e_kind != ELF_K_ELF ||
101	    ((ec = e->e_class) != ELFCLASS32 && ec != ELFCLASS64)) {
102		LIBELF_SET_ERROR(ARGUMENT, 0);
103		return (0);
104	}
105
106	if (e->e_cmd == ELF_C_READ) {
107		LIBELF_SET_ERROR(MODE, 0);
108		return (0);
109	}
110
111	(void) elf_flagscn(scn, ELF_C_SET, ELF_F_DIRTY);
112
113	if (ec == ELFCLASS64) {
114		scn->s_shdr.s_shdr64 = *s;
115		return (1);
116	}
117
118	sh32 = &scn->s_shdr.s_shdr32;
119
120	sh32->sh_name	 =  s->sh_name;
121	sh32->sh_type	 =  s->sh_type;
122	LIBELF_COPY_U32(sh32, s, sh_flags);
123	LIBELF_COPY_U32(sh32, s, sh_addr);
124	LIBELF_COPY_U32(sh32, s, sh_offset);
125	LIBELF_COPY_U32(sh32, s, sh_size);
126	sh32->sh_link	 =  s->sh_link;
127	sh32->sh_info	 =  s->sh_info;
128	LIBELF_COPY_U32(sh32, s, sh_addralign);
129	LIBELF_COPY_U32(sh32, s, sh_entsize);
130
131	return (1);
132}
133