xref: /onnv-gate/usr/src/cmd/sgs/libelf/common/newphdr.c (revision 0:68f95e015346)
1*0Sstevel@tonic-gate /*
2*0Sstevel@tonic-gate  * CDDL HEADER START
3*0Sstevel@tonic-gate  *
4*0Sstevel@tonic-gate  * The contents of this file are subject to the terms of the
5*0Sstevel@tonic-gate  * Common Development and Distribution License, Version 1.0 only
6*0Sstevel@tonic-gate  * (the "License").  You may not use this file except in compliance
7*0Sstevel@tonic-gate  * with the License.
8*0Sstevel@tonic-gate  *
9*0Sstevel@tonic-gate  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
10*0Sstevel@tonic-gate  * or http://www.opensolaris.org/os/licensing.
11*0Sstevel@tonic-gate  * See the License for the specific language governing permissions
12*0Sstevel@tonic-gate  * and limitations under the License.
13*0Sstevel@tonic-gate  *
14*0Sstevel@tonic-gate  * When distributing Covered Code, include this CDDL HEADER in each
15*0Sstevel@tonic-gate  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
16*0Sstevel@tonic-gate  * If applicable, add the following below this CDDL HEADER, with the
17*0Sstevel@tonic-gate  * fields enclosed by brackets "[]" replaced with your own identifying
18*0Sstevel@tonic-gate  * information: Portions Copyright [yyyy] [name of copyright owner]
19*0Sstevel@tonic-gate  *
20*0Sstevel@tonic-gate  * CDDL HEADER END
21*0Sstevel@tonic-gate  */
22*0Sstevel@tonic-gate /*	Copyright (c) 1988 AT&T	*/
23*0Sstevel@tonic-gate /*	  All Rights Reserved  	*/
24*0Sstevel@tonic-gate 
25*0Sstevel@tonic-gate 
26*0Sstevel@tonic-gate /*
27*0Sstevel@tonic-gate  * Copyright 2004 Sun Microsystems, Inc.  All rights reserved.
28*0Sstevel@tonic-gate  * Use is subject to license terms.
29*0Sstevel@tonic-gate  */
30*0Sstevel@tonic-gate 
31*0Sstevel@tonic-gate #pragma ident	"%Z%%M%	%I%	%E% SMI"
32*0Sstevel@tonic-gate 
33*0Sstevel@tonic-gate #if !defined(_ELF64)
34*0Sstevel@tonic-gate #pragma weak elf32_newphdr = _elf32_newphdr
35*0Sstevel@tonic-gate #endif
36*0Sstevel@tonic-gate 
37*0Sstevel@tonic-gate #include "syn.h"
38*0Sstevel@tonic-gate #include <stdlib.h>
39*0Sstevel@tonic-gate #include <memory.h>
40*0Sstevel@tonic-gate #include <errno.h>
41*0Sstevel@tonic-gate #include "decl.h"
42*0Sstevel@tonic-gate #include "msg.h"
43*0Sstevel@tonic-gate 
44*0Sstevel@tonic-gate /*
45*0Sstevel@tonic-gate  * This module is compiled twice, the second time having
46*0Sstevel@tonic-gate  * -D_ELF64 defined.  The following set of macros, along
47*0Sstevel@tonic-gate  * with machelf.h, represent the differences between the
48*0Sstevel@tonic-gate  * two compilations.  Be careful *not* to add any class-
49*0Sstevel@tonic-gate  * dependent code (anything that has elf32 or elf64 in the
50*0Sstevel@tonic-gate  * name) to this code without hiding it behind a switch-
51*0Sstevel@tonic-gate  * able macro like these.
52*0Sstevel@tonic-gate  */
53*0Sstevel@tonic-gate #if	defined(_ELF64)
54*0Sstevel@tonic-gate 
55*0Sstevel@tonic-gate #define	ELFCLASS	ELFCLASS64
56*0Sstevel@tonic-gate #define	elf_newphdr	elf64_newphdr
57*0Sstevel@tonic-gate #define	elf_getehdr	elf64_getehdr
58*0Sstevel@tonic-gate #define	_elf_msize	_elf64_msize
59*0Sstevel@tonic-gate #define	elf_fsize	elf64_fsize
60*0Sstevel@tonic-gate 
61*0Sstevel@tonic-gate #else	/* else ELF32 */
62*0Sstevel@tonic-gate 
63*0Sstevel@tonic-gate #define	ELFCLASS	ELFCLASS32
64*0Sstevel@tonic-gate #define	elf_newphdr	elf32_newphdr
65*0Sstevel@tonic-gate #define	elf_getehdr	elf32_getehdr
66*0Sstevel@tonic-gate #define	_elf_msize	_elf32_msize
67*0Sstevel@tonic-gate #define	elf_fsize	elf32_fsize
68*0Sstevel@tonic-gate 
69*0Sstevel@tonic-gate #endif /* ELF64 */
70*0Sstevel@tonic-gate 
71*0Sstevel@tonic-gate 
72*0Sstevel@tonic-gate Phdr *
73*0Sstevel@tonic-gate elf_newphdr(Elf * elf, size_t count)
74*0Sstevel@tonic-gate {
75*0Sstevel@tonic-gate 	Elf_Void *	ph;
76*0Sstevel@tonic-gate 	size_t		sz;
77*0Sstevel@tonic-gate 	Phdr *		rc;
78*0Sstevel@tonic-gate 	unsigned	work;
79*0Sstevel@tonic-gate 
80*0Sstevel@tonic-gate 	if (elf == 0)
81*0Sstevel@tonic-gate 		return (0);
82*0Sstevel@tonic-gate 	ELFRLOCK(elf)
83*0Sstevel@tonic-gate 	if (elf->ed_class != ELFCLASS) {
84*0Sstevel@tonic-gate 		_elf_seterr(EREQ_CLASS, 0);
85*0Sstevel@tonic-gate 		ELFUNLOCK(elf)
86*0Sstevel@tonic-gate 		return (0);
87*0Sstevel@tonic-gate 	}
88*0Sstevel@tonic-gate 	ELFUNLOCK(elf)
89*0Sstevel@tonic-gate 	if (elf_getehdr(elf) == 0) {		/* this cooks if necessary */
90*0Sstevel@tonic-gate 		_elf_seterr(ESEQ_EHDR, 0);
91*0Sstevel@tonic-gate 		return (0);
92*0Sstevel@tonic-gate 	}
93*0Sstevel@tonic-gate 
94*0Sstevel@tonic-gate 	/*
95*0Sstevel@tonic-gate 	 * Free the existing header if appropriate.  This could reuse
96*0Sstevel@tonic-gate 	 * existing space if big enough, but that's unlikely, benefit
97*0Sstevel@tonic-gate 	 * would be negligible, and code would be more complicated.
98*0Sstevel@tonic-gate 	 */
99*0Sstevel@tonic-gate 
100*0Sstevel@tonic-gate 	ELFWLOCK(elf)
101*0Sstevel@tonic-gate 	if (elf->ed_myflags & EDF_PHALLOC) {
102*0Sstevel@tonic-gate 		elf->ed_myflags &= ~EDF_PHALLOC;
103*0Sstevel@tonic-gate 		rc = elf->ed_phdr;
104*0Sstevel@tonic-gate 		free(rc);
105*0Sstevel@tonic-gate 	}
106*0Sstevel@tonic-gate 
107*0Sstevel@tonic-gate 	/*
108*0Sstevel@tonic-gate 	 * Delete the header if count is zero.
109*0Sstevel@tonic-gate 	 */
110*0Sstevel@tonic-gate 
111*0Sstevel@tonic-gate 	ELFACCESSDATA(work, _elf_work)
112*0Sstevel@tonic-gate 	if ((sz = count * _elf_msize(ELF_T_PHDR, work)) == 0) {
113*0Sstevel@tonic-gate 		elf->ed_phflags &= ~ELF_F_DIRTY;
114*0Sstevel@tonic-gate 		elf->ed_phdr = 0;
115*0Sstevel@tonic-gate 		((Ehdr*)elf->ed_ehdr)->e_phnum = 0;
116*0Sstevel@tonic-gate 		((Ehdr*)elf->ed_ehdr)->e_phentsize = 0;
117*0Sstevel@tonic-gate 		elf->ed_phdrsz = 0;
118*0Sstevel@tonic-gate 		ELFUNLOCK(elf)
119*0Sstevel@tonic-gate 		return (0);
120*0Sstevel@tonic-gate 	}
121*0Sstevel@tonic-gate 
122*0Sstevel@tonic-gate 	if ((ph = malloc(sz)) == 0) {
123*0Sstevel@tonic-gate 		_elf_seterr(EMEM_PHDR, errno);
124*0Sstevel@tonic-gate 		elf->ed_phflags &= ~ELF_F_DIRTY;
125*0Sstevel@tonic-gate 		elf->ed_phdr = 0;
126*0Sstevel@tonic-gate 		((Ehdr*)elf->ed_ehdr)->e_phnum = 0;
127*0Sstevel@tonic-gate 		((Ehdr*)elf->ed_ehdr)->e_phentsize = 0;
128*0Sstevel@tonic-gate 		elf->ed_phdrsz = 0;
129*0Sstevel@tonic-gate 		ELFUNLOCK(elf)
130*0Sstevel@tonic-gate 		return (0);
131*0Sstevel@tonic-gate 	}
132*0Sstevel@tonic-gate 
133*0Sstevel@tonic-gate 	elf->ed_myflags |= EDF_PHALLOC;
134*0Sstevel@tonic-gate 	(void) memset(ph, 0, sz);
135*0Sstevel@tonic-gate 	elf->ed_phflags |= ELF_F_DIRTY;
136*0Sstevel@tonic-gate 	/* LINTED */
137*0Sstevel@tonic-gate 	((Ehdr*)elf->ed_ehdr)->e_phnum = (Half)count;
138*0Sstevel@tonic-gate 	((Ehdr*)elf->ed_ehdr)->e_phentsize
139*0Sstevel@tonic-gate 	    /* LINTED */
140*0Sstevel@tonic-gate 	    = (Half)elf_fsize(ELF_T_PHDR, 1, work);
141*0Sstevel@tonic-gate 	elf->ed_phdrsz = sz;
142*0Sstevel@tonic-gate 	elf->ed_phdr = rc = ph;
143*0Sstevel@tonic-gate 
144*0Sstevel@tonic-gate 	ELFUNLOCK(elf)
145*0Sstevel@tonic-gate 	return (rc);
146*0Sstevel@tonic-gate }
147