elf_flag.c revision 4484
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 284484Sbinkertn@umich.edu#include "libelf.h" 294484Sbinkertn@umich.edu 304484Sbinkertn@umich.edu#include "_libelf.h" 314484Sbinkertn@umich.edu 324484Sbinkertn@umich.eduunsigned int 334484Sbinkertn@umich.eduelf_flagdata(Elf_Data *d, Elf_Cmd c, unsigned int flags) 344484Sbinkertn@umich.edu{ 354484Sbinkertn@umich.edu Elf *e; 364484Sbinkertn@umich.edu Elf_Scn *scn; 374484Sbinkertn@umich.edu unsigned int r; 384484Sbinkertn@umich.edu 394484Sbinkertn@umich.edu if (d == NULL) 404484Sbinkertn@umich.edu return (0); 414484Sbinkertn@umich.edu 424484Sbinkertn@umich.edu if ((c != ELF_C_SET && c != ELF_C_CLR) || (scn = d->d_scn) == NULL || 434484Sbinkertn@umich.edu (e = scn->s_elf) == NULL || e->e_kind != ELF_K_ELF || 444484Sbinkertn@umich.edu (flags & ~ELF_F_DIRTY) != 0) { 454484Sbinkertn@umich.edu LIBELF_SET_ERROR(ARGUMENT, 0); 464484Sbinkertn@umich.edu return (0); 474484Sbinkertn@umich.edu } 484484Sbinkertn@umich.edu 494484Sbinkertn@umich.edu if (c == ELF_C_SET) 504484Sbinkertn@umich.edu r = scn->s_flags |= flags; 514484Sbinkertn@umich.edu else 524484Sbinkertn@umich.edu r = scn->s_flags &= ~flags; 534484Sbinkertn@umich.edu 544484Sbinkertn@umich.edu return (r); 554484Sbinkertn@umich.edu} 564484Sbinkertn@umich.edu 574484Sbinkertn@umich.eduunsigned int 584484Sbinkertn@umich.eduelf_flagehdr(Elf *e, Elf_Cmd c, unsigned int flags) 594484Sbinkertn@umich.edu{ 604484Sbinkertn@umich.edu int ec; 614484Sbinkertn@umich.edu void *ehdr; 624484Sbinkertn@umich.edu 634484Sbinkertn@umich.edu if (e == NULL) 644484Sbinkertn@umich.edu return (0); 654484Sbinkertn@umich.edu 664484Sbinkertn@umich.edu if ((c != ELF_C_SET && c != ELF_C_CLR) || 674484Sbinkertn@umich.edu (e->e_kind != ELF_K_ELF) || (flags & ~ELF_F_DIRTY) != 0 || 684484Sbinkertn@umich.edu ((ec = e->e_class) != ELFCLASS32 && ec != ELFCLASS64)) { 694484Sbinkertn@umich.edu LIBELF_SET_ERROR(ARGUMENT, 0); 704484Sbinkertn@umich.edu return (0); 714484Sbinkertn@umich.edu } 724484Sbinkertn@umich.edu 734484Sbinkertn@umich.edu if (ec == ELFCLASS32) 744484Sbinkertn@umich.edu ehdr = e->e_u.e_elf.e_ehdr.e_ehdr32; 754484Sbinkertn@umich.edu else 764484Sbinkertn@umich.edu ehdr = e->e_u.e_elf.e_ehdr.e_ehdr64; 774484Sbinkertn@umich.edu 784484Sbinkertn@umich.edu if (ehdr == NULL) { 794484Sbinkertn@umich.edu LIBELF_SET_ERROR(SEQUENCE, 0); 804484Sbinkertn@umich.edu return (0); 814484Sbinkertn@umich.edu } 824484Sbinkertn@umich.edu 834484Sbinkertn@umich.edu return (elf_flagelf(e, c, flags)); 844484Sbinkertn@umich.edu} 854484Sbinkertn@umich.edu 864484Sbinkertn@umich.eduunsigned int 874484Sbinkertn@umich.eduelf_flagelf(Elf *e, Elf_Cmd c, unsigned int flags) 884484Sbinkertn@umich.edu{ 894484Sbinkertn@umich.edu int r; 904484Sbinkertn@umich.edu 914484Sbinkertn@umich.edu if (e == NULL) 924484Sbinkertn@umich.edu return (0); 934484Sbinkertn@umich.edu 944484Sbinkertn@umich.edu if ((c != ELF_C_SET && c != ELF_C_CLR) || 954484Sbinkertn@umich.edu (e->e_kind != ELF_K_ELF) || 964484Sbinkertn@umich.edu (flags & ~(ELF_F_DIRTY|ELF_F_LAYOUT)) != 0) { 974484Sbinkertn@umich.edu LIBELF_SET_ERROR(ARGUMENT, 0); 984484Sbinkertn@umich.edu return (0); 994484Sbinkertn@umich.edu } 1004484Sbinkertn@umich.edu 1014484Sbinkertn@umich.edu if (c == ELF_C_SET) 1024484Sbinkertn@umich.edu r = e->e_flags |= flags; 1034484Sbinkertn@umich.edu else 1044484Sbinkertn@umich.edu r = e->e_flags &= ~flags; 1054484Sbinkertn@umich.edu return (r); 1064484Sbinkertn@umich.edu} 1074484Sbinkertn@umich.edu 1084484Sbinkertn@umich.eduunsigned int 1094484Sbinkertn@umich.eduelf_flagphdr(Elf *e, Elf_Cmd c, unsigned int flags) 1104484Sbinkertn@umich.edu{ 1114484Sbinkertn@umich.edu int ec; 1124484Sbinkertn@umich.edu void *phdr; 1134484Sbinkertn@umich.edu 1144484Sbinkertn@umich.edu if (e == NULL) 1154484Sbinkertn@umich.edu return (0); 1164484Sbinkertn@umich.edu 1174484Sbinkertn@umich.edu if ((c != ELF_C_SET && c != ELF_C_CLR) || 1184484Sbinkertn@umich.edu (e->e_kind != ELF_K_ELF) || (flags & ~ELF_F_DIRTY) != 0 || 1194484Sbinkertn@umich.edu ((ec = e->e_class) != ELFCLASS32 && ec != ELFCLASS64)) { 1204484Sbinkertn@umich.edu LIBELF_SET_ERROR(ARGUMENT, 0); 1214484Sbinkertn@umich.edu return (0); 1224484Sbinkertn@umich.edu } 1234484Sbinkertn@umich.edu 1244484Sbinkertn@umich.edu if (ec == ELFCLASS32) 1254484Sbinkertn@umich.edu phdr = e->e_u.e_elf.e_phdr.e_phdr32; 1264484Sbinkertn@umich.edu else 1274484Sbinkertn@umich.edu phdr = e->e_u.e_elf.e_phdr.e_phdr64; 1284484Sbinkertn@umich.edu 1294484Sbinkertn@umich.edu if (phdr == NULL) { 1304484Sbinkertn@umich.edu LIBELF_SET_ERROR(SEQUENCE, 0); 1314484Sbinkertn@umich.edu return (0); 1324484Sbinkertn@umich.edu } 1334484Sbinkertn@umich.edu 1344484Sbinkertn@umich.edu return (elf_flagelf(e, c, flags)); 1354484Sbinkertn@umich.edu} 1364484Sbinkertn@umich.edu 1374484Sbinkertn@umich.eduunsigned int 1384484Sbinkertn@umich.eduelf_flagscn(Elf_Scn *s, Elf_Cmd c, unsigned int flags) 1394484Sbinkertn@umich.edu{ 1404484Sbinkertn@umich.edu int r; 1414484Sbinkertn@umich.edu 1424484Sbinkertn@umich.edu if (s == NULL) 1434484Sbinkertn@umich.edu return (0); 1444484Sbinkertn@umich.edu 1454484Sbinkertn@umich.edu if ((c != ELF_C_SET && c != ELF_C_CLR) || 1464484Sbinkertn@umich.edu (flags & ~ELF_F_DIRTY) != 0) { 1474484Sbinkertn@umich.edu LIBELF_SET_ERROR(ARGUMENT, 0); 1484484Sbinkertn@umich.edu return (0); 1494484Sbinkertn@umich.edu } 1504484Sbinkertn@umich.edu 1514484Sbinkertn@umich.edu if (c == ELF_C_SET) 1524484Sbinkertn@umich.edu r = s->s_flags |= flags; 1534484Sbinkertn@umich.edu else 1544484Sbinkertn@umich.edu r = s->s_flags &= ~flags; 1554484Sbinkertn@umich.edu return (r); 1564484Sbinkertn@umich.edu} 1574484Sbinkertn@umich.edu 1584484Sbinkertn@umich.eduunsigned int 1594484Sbinkertn@umich.eduelf_flagshdr(Elf_Scn *s, Elf_Cmd c, unsigned int flags) 1604484Sbinkertn@umich.edu{ 1614484Sbinkertn@umich.edu return (elf_flagscn(s, c, flags)); 1624484Sbinkertn@umich.edu} 163