1 /*
2 * CDDL HEADER START
3 *
4 * The contents of this file are subject to the terms of the
5 * Common Development and Distribution License (the "License").
6 * You may not use this file except in compliance with the License.
7 *
8 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
9 * or http://www.opensolaris.org/os/licensing.
10 * See the License for the specific language governing permissions
11 * and limitations under the License.
12 *
13 * When distributing Covered Code, include this CDDL HEADER in each
14 * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
15 * If applicable, add the following below this CDDL HEADER, with the
16 * fields enclosed by brackets "[]" replaced with your own identifying
17 * information: Portions Copyright [yyyy] [name of copyright owner]
18 *
19 * CDDL HEADER END
20 */
21
22 /*
23 * Copyright 2008 Sun Microsystems, Inc. All rights reserved.
24 * Use is subject to license terms.
25 */
26
27 /* Copyright (c) 1988 AT&T */
28 /* All Rights Reserved */
29
30 #pragma ident "%Z%%M% %I% %E% SMI"
31
32 #include <stdlib.h>
33 #include <errno.h>
34 #include "decl.h"
35 #include "msg.h"
36
37 /*
38 * This module is compiled twice, the second time having
39 * -D_ELF64 defined. The following set of macros, along
40 * with machelf.h, represent the differences between the
41 * two compilations. Be careful *not* to add any class-
42 * dependent code (anything that has elf32 or elf64 in the
43 * name) to this code without hiding it behind a switch-
44 * able macro like these.
45 */
46 #if defined(_ELF64)
47
48 #define ELFCLASS ELFCLASS64
49 #define _elf_ehdr_init _elf64_ehdr_init
50 #define elf_newehdr elf64_newehdr
51 #define getehdr elf64_getehdr
52
53 #else /* else ELF32 */
54
55 #define ELFCLASS ELFCLASS32
56 #define _elf_ehdr_init _elf32_ehdr_init
57 #define elf_newehdr elf32_newehdr
58 #define getehdr elf32_getehdr
59
60 #endif /* ELF64 */
61
62
63 Ehdr *
elf_newehdr(Elf * elf)64 elf_newehdr(Elf * elf)
65 {
66 Ehdr *eh;
67
68 if (elf == 0)
69 return (0);
70
71 /*
72 * If reading file, return its hdr
73 */
74
75 ELFWLOCK(elf)
76 if (elf->ed_myflags & EDF_READ) {
77 ELFUNLOCK(elf)
78 if ((eh = (Ehdr *)getehdr(elf)) != 0) {
79 ELFWLOCK(elf)
80 elf->ed_ehflags |= ELF_F_DIRTY;
81 ELFUNLOCK(elf)
82 }
83 return (eh);
84 }
85
86 /*
87 * Writing file
88 */
89
90 if (elf->ed_class == ELFCLASSNONE)
91 elf->ed_class = ELFCLASS;
92 else if (elf->ed_class != ELFCLASS) {
93 _elf_seterr(EREQ_CLASS, 0);
94 ELFUNLOCK(elf)
95 return (0);
96 }
97 ELFUNLOCK(elf);
98 if ((eh = (Ehdr *)getehdr(elf)) != 0) { /* this cooks if necessary */
99 ELFWLOCK(elf)
100 elf->ed_ehflags |= ELF_F_DIRTY;
101 ELFUNLOCK(elf)
102 return (eh);
103 }
104 ELFWLOCK(elf)
105
106 if ((eh = (Ehdr *)malloc(sizeof (Ehdr))) == 0) {
107 _elf_seterr(EMEM_EHDR, errno);
108 ELFUNLOCK(elf)
109 return (0);
110 }
111 *eh = _elf_ehdr_init;
112 elf->ed_myflags |= EDF_EHALLOC;
113 elf->ed_ehflags |= ELF_F_DIRTY;
114 elf->ed_ehdr = eh;
115 ELFUNLOCK(elf)
116 return (eh);
117 }
118