12de3b87aSKai Wang /*- 22de3b87aSKai Wang * Copyright (c) 2006,2008 Joseph Koshy 32de3b87aSKai Wang * All rights reserved. 42de3b87aSKai Wang * 52de3b87aSKai Wang * Redistribution and use in source and binary forms, with or without 62de3b87aSKai Wang * modification, are permitted provided that the following conditions 72de3b87aSKai Wang * are met: 82de3b87aSKai Wang * 1. Redistributions of source code must retain the above copyright 92de3b87aSKai Wang * notice, this list of conditions and the following disclaimer. 102de3b87aSKai Wang * 2. Redistributions in binary form must reproduce the above copyright 112de3b87aSKai Wang * notice, this list of conditions and the following disclaimer in the 122de3b87aSKai Wang * documentation and/or other materials provided with the distribution. 132de3b87aSKai Wang * 142de3b87aSKai Wang * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 152de3b87aSKai Wang * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 162de3b87aSKai Wang * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 172de3b87aSKai Wang * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 182de3b87aSKai Wang * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 192de3b87aSKai Wang * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 202de3b87aSKai Wang * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 212de3b87aSKai Wang * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 222de3b87aSKai Wang * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 232de3b87aSKai Wang * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 242de3b87aSKai Wang * SUCH DAMAGE. 252de3b87aSKai Wang */ 262de3b87aSKai Wang 272de3b87aSKai Wang #include <sys/cdefs.h> 282de3b87aSKai Wang 292de3b87aSKai Wang #include <gelf.h> 302de3b87aSKai Wang #include <libelf.h> 312de3b87aSKai Wang #include <limits.h> 32*b4e9f239SEd Maste #include <stdint.h> 332de3b87aSKai Wang 342de3b87aSKai Wang #include "_libelf.h" 352de3b87aSKai Wang 362de3b87aSKai Wang ELFTC_VCSID("$Id: gelf_phdr.c 2268 2011-12-03 17:05:11Z jkoshy $"); 372de3b87aSKai Wang 382de3b87aSKai Wang Elf32_Phdr * 392de3b87aSKai Wang elf32_getphdr(Elf *e) 402de3b87aSKai Wang { 412de3b87aSKai Wang return (_libelf_getphdr(e, ELFCLASS32)); 422de3b87aSKai Wang } 432de3b87aSKai Wang 442de3b87aSKai Wang Elf64_Phdr * 452de3b87aSKai Wang elf64_getphdr(Elf *e) 462de3b87aSKai Wang { 472de3b87aSKai Wang return (_libelf_getphdr(e, ELFCLASS64)); 482de3b87aSKai Wang } 492de3b87aSKai Wang 502de3b87aSKai Wang GElf_Phdr * 512de3b87aSKai Wang gelf_getphdr(Elf *e, int index, GElf_Phdr *d) 522de3b87aSKai Wang { 532de3b87aSKai Wang int ec; 542de3b87aSKai Wang Elf32_Ehdr *eh32; 552de3b87aSKai Wang Elf64_Ehdr *eh64; 562de3b87aSKai Wang Elf32_Phdr *ep32; 572de3b87aSKai Wang Elf64_Phdr *ep64; 582de3b87aSKai Wang 592de3b87aSKai Wang if (d == NULL || e == NULL || 602de3b87aSKai Wang ((ec = e->e_class) != ELFCLASS32 && ec != ELFCLASS64) || 612de3b87aSKai Wang (e->e_kind != ELF_K_ELF) || index < 0) { 622de3b87aSKai Wang LIBELF_SET_ERROR(ARGUMENT, 0); 632de3b87aSKai Wang return (NULL); 642de3b87aSKai Wang } 652de3b87aSKai Wang 662de3b87aSKai Wang if (ec == ELFCLASS32) { 672de3b87aSKai Wang if ((eh32 = _libelf_ehdr(e, ELFCLASS32, 0)) == NULL || 682de3b87aSKai Wang ((ep32 = _libelf_getphdr(e, ELFCLASS32)) == NULL)) 692de3b87aSKai Wang return (NULL); 702de3b87aSKai Wang 712de3b87aSKai Wang if (index >= eh32->e_phnum) { 722de3b87aSKai Wang LIBELF_SET_ERROR(ARGUMENT, 0); 732de3b87aSKai Wang return (NULL); 742de3b87aSKai Wang } 752de3b87aSKai Wang 762de3b87aSKai Wang ep32 += index; 772de3b87aSKai Wang 782de3b87aSKai Wang d->p_type = ep32->p_type; 792de3b87aSKai Wang d->p_offset = ep32->p_offset; 802de3b87aSKai Wang d->p_vaddr = (Elf64_Addr) ep32->p_vaddr; 812de3b87aSKai Wang d->p_paddr = (Elf64_Addr) ep32->p_paddr; 822de3b87aSKai Wang d->p_filesz = (Elf64_Xword) ep32->p_filesz; 832de3b87aSKai Wang d->p_memsz = (Elf64_Xword) ep32->p_memsz; 842de3b87aSKai Wang d->p_flags = ep32->p_flags; 852de3b87aSKai Wang d->p_align = (Elf64_Xword) ep32->p_align; 862de3b87aSKai Wang 872de3b87aSKai Wang } else { 882de3b87aSKai Wang if ((eh64 = _libelf_ehdr(e, ELFCLASS64, 0)) == NULL || 892de3b87aSKai Wang (ep64 = _libelf_getphdr(e, ELFCLASS64)) == NULL) 902de3b87aSKai Wang return (NULL); 912de3b87aSKai Wang 922de3b87aSKai Wang if (index >= eh64->e_phnum) { 932de3b87aSKai Wang LIBELF_SET_ERROR(ARGUMENT, 0); 942de3b87aSKai Wang return (NULL); 952de3b87aSKai Wang } 962de3b87aSKai Wang 972de3b87aSKai Wang ep64 += index; 982de3b87aSKai Wang 992de3b87aSKai Wang *d = *ep64; 1002de3b87aSKai Wang } 1012de3b87aSKai Wang 1022de3b87aSKai Wang return (d); 1032de3b87aSKai Wang } 1042de3b87aSKai Wang 1052de3b87aSKai Wang Elf32_Phdr * 1062de3b87aSKai Wang elf32_newphdr(Elf *e, size_t count) 1072de3b87aSKai Wang { 1082de3b87aSKai Wang return (_libelf_newphdr(e, ELFCLASS32, count)); 1092de3b87aSKai Wang } 1102de3b87aSKai Wang 1112de3b87aSKai Wang Elf64_Phdr * 1122de3b87aSKai Wang elf64_newphdr(Elf *e, size_t count) 1132de3b87aSKai Wang { 1142de3b87aSKai Wang return (_libelf_newphdr(e, ELFCLASS64, count)); 1152de3b87aSKai Wang } 1162de3b87aSKai Wang 1172de3b87aSKai Wang void * 1182de3b87aSKai Wang gelf_newphdr(Elf *e, size_t count) 1192de3b87aSKai Wang { 1202de3b87aSKai Wang if (e == NULL) { 1212de3b87aSKai Wang LIBELF_SET_ERROR(ARGUMENT, 0); 1222de3b87aSKai Wang return (NULL); 1232de3b87aSKai Wang } 1242de3b87aSKai Wang return (_libelf_newphdr(e, e->e_class, count)); 1252de3b87aSKai Wang } 1262de3b87aSKai Wang 1272de3b87aSKai Wang int 1282de3b87aSKai Wang gelf_update_phdr(Elf *e, int ndx, GElf_Phdr *s) 1292de3b87aSKai Wang { 1302de3b87aSKai Wang int ec, phnum; 1312de3b87aSKai Wang void *ehdr; 1322de3b87aSKai Wang Elf32_Phdr *ph32; 1332de3b87aSKai Wang Elf64_Phdr *ph64; 1342de3b87aSKai Wang 1352de3b87aSKai Wang if (s == NULL || e == NULL || e->e_kind != ELF_K_ELF || 1362de3b87aSKai Wang ((ec = e->e_class) != ELFCLASS32 && ec != ELFCLASS64)) { 1372de3b87aSKai Wang LIBELF_SET_ERROR(ARGUMENT, 0); 1382de3b87aSKai Wang return (0); 1392de3b87aSKai Wang } 1402de3b87aSKai Wang 1412de3b87aSKai Wang if (e->e_cmd == ELF_C_READ) { 1422de3b87aSKai Wang LIBELF_SET_ERROR(MODE, 0); 1432de3b87aSKai Wang return (0); 1442de3b87aSKai Wang } 1452de3b87aSKai Wang 1462de3b87aSKai Wang if ((ehdr = _libelf_ehdr(e, ec, 0)) == NULL) 1472de3b87aSKai Wang return (0); 1482de3b87aSKai Wang 1492de3b87aSKai Wang if (ec == ELFCLASS32) 1502de3b87aSKai Wang phnum = ((Elf32_Ehdr *) ehdr)->e_phnum; 1512de3b87aSKai Wang else 1522de3b87aSKai Wang phnum = ((Elf64_Ehdr *) ehdr)->e_phnum; 1532de3b87aSKai Wang 1542de3b87aSKai Wang if (ndx < 0 || ndx > phnum) { 1552de3b87aSKai Wang LIBELF_SET_ERROR(ARGUMENT, 0); 1562de3b87aSKai Wang return (0); 1572de3b87aSKai Wang } 1582de3b87aSKai Wang 1592de3b87aSKai Wang (void) elf_flagphdr(e, ELF_C_SET, ELF_F_DIRTY); 1602de3b87aSKai Wang 1612de3b87aSKai Wang if (ec == ELFCLASS64) { 1622de3b87aSKai Wang ph64 = e->e_u.e_elf.e_phdr.e_phdr64 + ndx; 1632de3b87aSKai Wang *ph64 = *s; 1642de3b87aSKai Wang return (1); 1652de3b87aSKai Wang } 1662de3b87aSKai Wang 1672de3b87aSKai Wang ph32 = e->e_u.e_elf.e_phdr.e_phdr32 + ndx; 1682de3b87aSKai Wang 1692de3b87aSKai Wang ph32->p_type = s->p_type; 1702de3b87aSKai Wang ph32->p_flags = s->p_flags; 1712de3b87aSKai Wang LIBELF_COPY_U32(ph32, s, p_offset); 1722de3b87aSKai Wang LIBELF_COPY_U32(ph32, s, p_vaddr); 1732de3b87aSKai Wang LIBELF_COPY_U32(ph32, s, p_paddr); 1742de3b87aSKai Wang LIBELF_COPY_U32(ph32, s, p_filesz); 1752de3b87aSKai Wang LIBELF_COPY_U32(ph32, s, p_memsz); 1762de3b87aSKai Wang LIBELF_COPY_U32(ph32, s, p_align); 1772de3b87aSKai Wang 1782de3b87aSKai Wang return (1); 1792de3b87aSKai Wang } 180