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