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 #include <stdint.h> 33 34 #include "_libelf.h" 35 36 ELFTC_VCSID("$Id: gelf_phdr.c 2268 2011-12-03 17:05:11Z jkoshy $"); 37 38 Elf32_Phdr * 39 elf32_getphdr(Elf *e) 40 { 41 return (_libelf_getphdr(e, ELFCLASS32)); 42 } 43 44 Elf64_Phdr * 45 elf64_getphdr(Elf *e) 46 { 47 return (_libelf_getphdr(e, ELFCLASS64)); 48 } 49 50 GElf_Phdr * 51 gelf_getphdr(Elf *e, int index, GElf_Phdr *d) 52 { 53 int ec; 54 Elf32_Ehdr *eh32; 55 Elf64_Ehdr *eh64; 56 Elf32_Phdr *ep32; 57 Elf64_Phdr *ep64; 58 59 if (d == NULL || e == NULL || 60 ((ec = e->e_class) != ELFCLASS32 && ec != ELFCLASS64) || 61 (e->e_kind != ELF_K_ELF) || index < 0) { 62 LIBELF_SET_ERROR(ARGUMENT, 0); 63 return (NULL); 64 } 65 66 if (ec == ELFCLASS32) { 67 if ((eh32 = _libelf_ehdr(e, ELFCLASS32, 0)) == NULL || 68 ((ep32 = _libelf_getphdr(e, ELFCLASS32)) == NULL)) 69 return (NULL); 70 71 if (index >= eh32->e_phnum) { 72 LIBELF_SET_ERROR(ARGUMENT, 0); 73 return (NULL); 74 } 75 76 ep32 += index; 77 78 d->p_type = ep32->p_type; 79 d->p_offset = ep32->p_offset; 80 d->p_vaddr = (Elf64_Addr) ep32->p_vaddr; 81 d->p_paddr = (Elf64_Addr) ep32->p_paddr; 82 d->p_filesz = (Elf64_Xword) ep32->p_filesz; 83 d->p_memsz = (Elf64_Xword) ep32->p_memsz; 84 d->p_flags = ep32->p_flags; 85 d->p_align = (Elf64_Xword) ep32->p_align; 86 87 } else { 88 if ((eh64 = _libelf_ehdr(e, ELFCLASS64, 0)) == NULL || 89 (ep64 = _libelf_getphdr(e, ELFCLASS64)) == NULL) 90 return (NULL); 91 92 if (index >= eh64->e_phnum) { 93 LIBELF_SET_ERROR(ARGUMENT, 0); 94 return (NULL); 95 } 96 97 ep64 += index; 98 99 *d = *ep64; 100 } 101 102 return (d); 103 } 104 105 Elf32_Phdr * 106 elf32_newphdr(Elf *e, size_t count) 107 { 108 return (_libelf_newphdr(e, ELFCLASS32, count)); 109 } 110 111 Elf64_Phdr * 112 elf64_newphdr(Elf *e, size_t count) 113 { 114 return (_libelf_newphdr(e, ELFCLASS64, count)); 115 } 116 117 void * 118 gelf_newphdr(Elf *e, size_t count) 119 { 120 if (e == NULL) { 121 LIBELF_SET_ERROR(ARGUMENT, 0); 122 return (NULL); 123 } 124 return (_libelf_newphdr(e, e->e_class, count)); 125 } 126 127 int 128 gelf_update_phdr(Elf *e, int ndx, GElf_Phdr *s) 129 { 130 int ec, phnum; 131 void *ehdr; 132 Elf32_Phdr *ph32; 133 Elf64_Phdr *ph64; 134 135 if (s == NULL || e == NULL || e->e_kind != ELF_K_ELF || 136 ((ec = e->e_class) != ELFCLASS32 && ec != ELFCLASS64)) { 137 LIBELF_SET_ERROR(ARGUMENT, 0); 138 return (0); 139 } 140 141 if (e->e_cmd == ELF_C_READ) { 142 LIBELF_SET_ERROR(MODE, 0); 143 return (0); 144 } 145 146 if ((ehdr = _libelf_ehdr(e, ec, 0)) == NULL) 147 return (0); 148 149 if (ec == ELFCLASS32) 150 phnum = ((Elf32_Ehdr *) ehdr)->e_phnum; 151 else 152 phnum = ((Elf64_Ehdr *) ehdr)->e_phnum; 153 154 if (ndx < 0 || ndx > phnum) { 155 LIBELF_SET_ERROR(ARGUMENT, 0); 156 return (0); 157 } 158 159 (void) elf_flagphdr(e, ELF_C_SET, ELF_F_DIRTY); 160 161 if (ec == ELFCLASS64) { 162 ph64 = e->e_u.e_elf.e_phdr.e_phdr64 + ndx; 163 *ph64 = *s; 164 return (1); 165 } 166 167 ph32 = e->e_u.e_elf.e_phdr.e_phdr32 + ndx; 168 169 ph32->p_type = s->p_type; 170 ph32->p_flags = s->p_flags; 171 LIBELF_COPY_U32(ph32, s, p_offset); 172 LIBELF_COPY_U32(ph32, s, p_vaddr); 173 LIBELF_COPY_U32(ph32, s, p_paddr); 174 LIBELF_COPY_U32(ph32, s, p_filesz); 175 LIBELF_COPY_U32(ph32, s, p_memsz); 176 LIBELF_COPY_U32(ph32, s, p_align); 177 178 return (1); 179 } 180