external/bsd/elftoolchain/dist/libelf/libelf_extended.c

   1 /*      $NetBSD: libelf_extended.c,v 1.2 2014/03/09 16:58:04 christos Exp $     */
   2
   3 /*-
   4  * Copyright (c) 2006,2008 Joseph Koshy
   5  * All rights reserved.
   6  *
   7  * Redistribution and use in source and binary forms, with or without
   8  * modification, are permitted provided that the following conditions
   9  * are met:
  10  * 1. Redistributions of source code must retain the above copyright
  11  *    notice, this list of conditions and the following disclaimer.
  12  * 2. Redistributions in binary form must reproduce the above copyright
  13  *    notice, this list of conditions and the following disclaimer in the
  14  *    documentation and/or other materials provided with the distribution.
  15  *
  16  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  17  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  18  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  19  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  20  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  21  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  22  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  23  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  24  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  25  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  26  * SUCH DAMAGE.
  27  */
  28
  29 #if HAVE_NBTOOL_CONFIG_H
  30 # include "nbtool_config.h"
  31 #endif
  32
  33 #include <sys/cdefs.h>
  34
  35 #include <assert.h>
  36 #include <libelf.h>
  37
  38 #include "_libelf.h"
  39
  40 __RCSID("$NetBSD: libelf_extended.c,v 1.2 2014/03/09 16:58:04 christos Exp $");
  41 ELFTC_VCSID("Id: libelf_extended.c 2225 2011-11-26 18:55:54Z jkoshy ");
  42
  43 /*
  44  * Retrieve section #0, allocating a new section if needed.
  45  */
  46 static Elf_Scn *
  47 _libelf_getscn0(Elf *e)
  48 {
  49         Elf_Scn *s;
  50
  51         if ((s = STAILQ_FIRST(&e->e_u.e_elf.e_scn)) != NULL)
  52                 return (s);
  53
  54         return (_libelf_allocate_scn(e, (size_t) SHN_UNDEF));
  55 }
  56
  57 int
  58 _libelf_setshnum(Elf *e, void *eh, int ec, size_t shnum)
  59 {
  60         Elf_Scn *scn;
  61
  62         if (shnum >= SHN_LORESERVE) {
  63                 if ((scn = _libelf_getscn0(e)) == NULL)
  64                         return (0);
  65
  66                 assert(scn->s_ndx == SHN_UNDEF);
  67
  68                 if (ec == ELFCLASS32)
  69                         scn->s_shdr.s_shdr32.sh_size = shnum;
  70                 else
  71                         scn->s_shdr.s_shdr64.sh_size = shnum;
  72
  73                 (void) elf_flagshdr(scn, ELF_C_SET, ELF_F_DIRTY);
  74
  75                 shnum = 0;
  76         }
  77
  78         if (ec == ELFCLASS32)
  79                 ((Elf32_Ehdr *) eh)->e_shnum = shnum;
  80         else
  81                 ((Elf64_Ehdr *) eh)->e_shnum = shnum;
  82
  83
  84         return (1);
  85 }
  86
  87 int
  88 _libelf_setshstrndx(Elf *e, void *eh, int ec, size_t shstrndx)
  89 {
  90         Elf_Scn *scn;
  91
  92         if (shstrndx >= SHN_LORESERVE) {
  93                 if ((scn = _libelf_getscn0(e)) == NULL)
  94                         return (0);
  95
  96                 assert(scn->s_ndx == SHN_UNDEF);
  97
  98                 if (ec == ELFCLASS32)
  99                         scn->s_shdr.s_shdr32.sh_link = shstrndx;
 100                 else
 101                         scn->s_shdr.s_shdr64.sh_link = shstrndx;
 102
 103                 (void) elf_flagshdr(scn, ELF_C_SET, ELF_F_DIRTY);
 104
 105                 shstrndx = SHN_XINDEX;
 106         }
 107
 108         if (ec == ELFCLASS32)
 109                 ((Elf32_Ehdr *) eh)->e_shstrndx = shstrndx;
 110         else
 111                 ((Elf64_Ehdr *) eh)->e_shstrndx = shstrndx;
 112
 113         return (1);
 114 }
 115
 116 int
 117 _libelf_setphnum(Elf *e, void *eh, int ec, size_t phnum)
 118 {
 119         Elf_Scn *scn;
 120
 121         if (phnum >= PN_XNUM) {
 122                 if ((scn = _libelf_getscn0(e)) == NULL)
 123                         return (0);
 124
 125                 assert(scn->s_ndx == SHN_UNDEF);
 126
 127                 if (ec == ELFCLASS32)
 128                         scn->s_shdr.s_shdr32.sh_info = phnum;
 129                 else
 130                         scn->s_shdr.s_shdr64.sh_info = phnum;
 131
 132                 (void) elf_flagshdr(scn, ELF_C_SET, ELF_F_DIRTY);
 133
 134                 phnum = PN_XNUM;
 135         }
 136
 137         if (ec == ELFCLASS32)
 138                 ((Elf32_Ehdr *) eh)->e_phnum = phnum;
 139         else
 140                 ((Elf64_Ehdr *) eh)->e_phnum = phnum;
 141
 142         return (1);
 143 }