Home | History | Annotate | Download | only in libelf
      1 /* Update section header.
      2    Copyright (C) 2000, 2001, 2002, 2010 Red Hat, Inc.
      3    This file is part of elfutils.
      4    Written by Ulrich Drepper <drepper (at) redhat.com>, 2000.
      5 
      6    This file is free software; you can redistribute it and/or modify
      7    it under the terms of either
      8 
      9      * the GNU Lesser General Public License as published by the Free
     10        Software Foundation; either version 3 of the License, or (at
     11        your option) any later version
     12 
     13    or
     14 
     15      * the GNU General Public License as published by the Free
     16        Software Foundation; either version 2 of the License, or (at
     17        your option) any later version
     18 
     19    or both in parallel, as here.
     20 
     21    elfutils is distributed in the hope that it will be useful, but
     22    WITHOUT ANY WARRANTY; without even the implied warranty of
     23    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
     24    General Public License for more details.
     25 
     26    You should have received copies of the GNU General Public License and
     27    the GNU Lesser General Public License along with this program.  If
     28    not, see <http://www.gnu.org/licenses/>.  */
     29 
     30 #ifdef HAVE_CONFIG_H
     31 # include <config.h>
     32 #endif
     33 
     34 #include <gelf.h>
     35 #include <string.h>
     36 
     37 #include "libelfP.h"
     38 
     39 
     40 int
     41 gelf_update_shdr (Elf_Scn *scn, GElf_Shdr *src)
     42 {
     43   int result = 0;
     44   Elf *elf;
     45 
     46   if (scn == NULL || src == NULL)
     47     return 0;
     48 
     49   elf = scn->elf;
     50   rwlock_wrlock (elf->lock);
     51 
     52   if (elf->class == ELFCLASS32)
     53     {
     54       Elf32_Shdr *shdr
     55 	= scn->shdr.e32 ?: __elf32_getshdr_wrlock (scn);
     56 
     57       if (shdr == NULL)
     58 	{
     59 	  __libelf_seterrno (ELF_E_INVALID_OPERAND);
     60 	  goto out;
     61 	}
     62 
     63       if (unlikely (src->sh_flags > 0xffffffffull)
     64 	  || unlikely (src->sh_addr > 0xffffffffull)
     65 	  || unlikely (src->sh_offset > 0xffffffffull)
     66 	  || unlikely (src->sh_size > 0xffffffffull)
     67 	  || unlikely (src->sh_addralign > 0xffffffffull)
     68 	  || unlikely (src->sh_entsize > 0xffffffffull))
     69 	{
     70 	  __libelf_seterrno (ELF_E_INVALID_DATA);
     71 	  goto out;
     72 	}
     73 
     74 #define COPY(name) \
     75       shdr->name = src->name
     76       COPY (sh_name);
     77       COPY (sh_type);
     78       COPY (sh_flags);
     79       COPY (sh_addr);
     80       COPY (sh_offset);
     81       COPY (sh_size);
     82       COPY (sh_link);
     83       COPY (sh_info);
     84       COPY (sh_addralign);
     85       COPY (sh_entsize);
     86     }
     87   else
     88     {
     89       Elf64_Shdr *shdr
     90 	= scn->shdr.e64 ?: __elf64_getshdr_wrlock (scn);
     91 
     92       if (shdr == NULL)
     93 	{
     94 	  __libelf_seterrno (ELF_E_INVALID_OPERAND);
     95 	  goto out;
     96 	}
     97 
     98       /* We only have to copy the data.  */
     99       (void) memcpy (shdr, src, sizeof (GElf_Shdr));
    100     }
    101 
    102   /* Mark the section header as modified.  */
    103   scn->shdr_flags |= ELF_F_DIRTY;
    104 
    105   result = 1;
    106 
    107  out:
    108   rwlock_unlock (elf->lock);
    109 
    110   return result;
    111 }
    112