14484Sbinkertn@umich.edu/*- 24484Sbinkertn@umich.edu * Copyright (c) 2006 Joseph Koshy 34484Sbinkertn@umich.edu * All rights reserved. 44484Sbinkertn@umich.edu * 54484Sbinkertn@umich.edu * Redistribution and use in source and binary forms, with or without 64484Sbinkertn@umich.edu * modification, are permitted provided that the following conditions 74484Sbinkertn@umich.edu * are met: 84484Sbinkertn@umich.edu * 1. Redistributions of source code must retain the above copyright 94484Sbinkertn@umich.edu * notice, this list of conditions and the following disclaimer. 104484Sbinkertn@umich.edu * 2. Redistributions in binary form must reproduce the above copyright 114484Sbinkertn@umich.edu * notice, this list of conditions and the following disclaimer in the 124484Sbinkertn@umich.edu * documentation and/or other materials provided with the distribution. 134484Sbinkertn@umich.edu * 144484Sbinkertn@umich.edu * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 154484Sbinkertn@umich.edu * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 164484Sbinkertn@umich.edu * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 174484Sbinkertn@umich.edu * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 184484Sbinkertn@umich.edu * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 194484Sbinkertn@umich.edu * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 204484Sbinkertn@umich.edu * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 214484Sbinkertn@umich.edu * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 224484Sbinkertn@umich.edu * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 234484Sbinkertn@umich.edu * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 244484Sbinkertn@umich.edu * SUCH DAMAGE. 254484Sbinkertn@umich.edu */ 264484Sbinkertn@umich.edu 274484Sbinkertn@umich.edu#include <assert.h> 284484Sbinkertn@umich.edu#include "libelf.h" 294484Sbinkertn@umich.edu 304484Sbinkertn@umich.edu#include "_libelf.h" 314484Sbinkertn@umich.edu 324484Sbinkertn@umich.edu/* 334484Sbinkertn@umich.edu * Retrieve section #0, allocating a new section if needed. 344484Sbinkertn@umich.edu */ 354484Sbinkertn@umich.edustatic Elf_Scn * 364484Sbinkertn@umich.edu_libelf_getscn0(Elf *e) 374484Sbinkertn@umich.edu{ 384484Sbinkertn@umich.edu Elf_Scn *s; 394484Sbinkertn@umich.edu 404484Sbinkertn@umich.edu if ((s = STAILQ_FIRST(&e->e_u.e_elf.e_scn)) != NULL) 414484Sbinkertn@umich.edu return (s); 424484Sbinkertn@umich.edu 434484Sbinkertn@umich.edu return (_libelf_allocate_scn(e, (size_t) SHN_UNDEF)); 444484Sbinkertn@umich.edu} 454484Sbinkertn@umich.edu 464484Sbinkertn@umich.eduint 474484Sbinkertn@umich.edu_libelf_setshnum(Elf *e, void *eh, int ec, size_t shnum) 484484Sbinkertn@umich.edu{ 494484Sbinkertn@umich.edu Elf_Scn *scn; 504484Sbinkertn@umich.edu 514484Sbinkertn@umich.edu if (shnum >= SHN_LORESERVE) { 524484Sbinkertn@umich.edu if ((scn = _libelf_getscn0(e)) == NULL) 534484Sbinkertn@umich.edu return (0); 544484Sbinkertn@umich.edu 554484Sbinkertn@umich.edu assert(scn->s_ndx == SHN_UNDEF); 564484Sbinkertn@umich.edu 574484Sbinkertn@umich.edu if (ec == ELFCLASS32) 584484Sbinkertn@umich.edu scn->s_shdr.s_shdr32.sh_size = shnum; 594484Sbinkertn@umich.edu else 604484Sbinkertn@umich.edu scn->s_shdr.s_shdr64.sh_size = shnum; 614484Sbinkertn@umich.edu 624484Sbinkertn@umich.edu (void) elf_flagshdr(scn, ELF_C_SET, ELF_F_DIRTY); 634484Sbinkertn@umich.edu 644484Sbinkertn@umich.edu shnum = 0; 654484Sbinkertn@umich.edu } 664484Sbinkertn@umich.edu 674484Sbinkertn@umich.edu if (ec == ELFCLASS32) 684484Sbinkertn@umich.edu ((Elf32_Ehdr *) eh)->e_shnum = shnum; 694484Sbinkertn@umich.edu else 704484Sbinkertn@umich.edu ((Elf64_Ehdr *) eh)->e_shnum = shnum; 714484Sbinkertn@umich.edu 724484Sbinkertn@umich.edu 734484Sbinkertn@umich.edu return (1); 744484Sbinkertn@umich.edu} 754484Sbinkertn@umich.edu 764484Sbinkertn@umich.eduint 774484Sbinkertn@umich.edu_libelf_setshstrndx(Elf *e, void *eh, int ec, size_t shstrndx) 784484Sbinkertn@umich.edu{ 794484Sbinkertn@umich.edu Elf_Scn *scn; 804484Sbinkertn@umich.edu 814484Sbinkertn@umich.edu if (shstrndx >= SHN_LORESERVE) { 824484Sbinkertn@umich.edu if ((scn = _libelf_getscn0(e)) == NULL) 834484Sbinkertn@umich.edu return (0); 844484Sbinkertn@umich.edu 854484Sbinkertn@umich.edu assert(scn->s_ndx == SHN_UNDEF); 864484Sbinkertn@umich.edu 874484Sbinkertn@umich.edu if (ec == ELFCLASS32) 884484Sbinkertn@umich.edu scn->s_shdr.s_shdr32.sh_link = shstrndx; 894484Sbinkertn@umich.edu else 904484Sbinkertn@umich.edu scn->s_shdr.s_shdr64.sh_link = shstrndx; 914484Sbinkertn@umich.edu 924484Sbinkertn@umich.edu (void) elf_flagshdr(scn, ELF_C_SET, ELF_F_DIRTY); 934484Sbinkertn@umich.edu 944484Sbinkertn@umich.edu shstrndx = SHN_XINDEX; 954484Sbinkertn@umich.edu } 964484Sbinkertn@umich.edu 974484Sbinkertn@umich.edu if (ec == ELFCLASS32) 984484Sbinkertn@umich.edu ((Elf32_Ehdr *) eh)->e_shstrndx = shstrndx; 994484Sbinkertn@umich.edu else 1004484Sbinkertn@umich.edu ((Elf64_Ehdr *) eh)->e_shstrndx = shstrndx; 1014484Sbinkertn@umich.edu 1024484Sbinkertn@umich.edu return (1); 1034484Sbinkertn@umich.edu} 1044484Sbinkertn@umich.edu 1054484Sbinkertn@umich.eduint 1064484Sbinkertn@umich.edu_libelf_setphnum(Elf *e, void *eh, int ec, size_t phnum) 1074484Sbinkertn@umich.edu{ 1084484Sbinkertn@umich.edu Elf_Scn *scn; 1094484Sbinkertn@umich.edu 1104484Sbinkertn@umich.edu if (phnum >= PN_XNUM) { 1114484Sbinkertn@umich.edu if ((scn = _libelf_getscn0(e)) == NULL) 1124484Sbinkertn@umich.edu return (0); 1134484Sbinkertn@umich.edu 1144484Sbinkertn@umich.edu assert(scn->s_ndx == SHN_UNDEF); 1154484Sbinkertn@umich.edu 1164484Sbinkertn@umich.edu if (ec == ELFCLASS32) 1174484Sbinkertn@umich.edu scn->s_shdr.s_shdr32.sh_info = phnum; 1184484Sbinkertn@umich.edu else 1194484Sbinkertn@umich.edu scn->s_shdr.s_shdr64.sh_info = phnum; 1204484Sbinkertn@umich.edu 1214484Sbinkertn@umich.edu (void) elf_flagshdr(scn, ELF_C_SET, ELF_F_DIRTY); 1224484Sbinkertn@umich.edu 1234484Sbinkertn@umich.edu phnum = PN_XNUM; 1244484Sbinkertn@umich.edu } 1254484Sbinkertn@umich.edu 1264484Sbinkertn@umich.edu if (ec == ELFCLASS32) 1274484Sbinkertn@umich.edu ((Elf32_Ehdr *) eh)->e_phnum = phnum; 1284484Sbinkertn@umich.edu else 1294484Sbinkertn@umich.edu ((Elf64_Ehdr *) eh)->e_phnum = phnum; 1304484Sbinkertn@umich.edu 1314484Sbinkertn@umich.edu return (1); 1324484Sbinkertn@umich.edu} 1334484Sbinkertn@umich.edu 134