Line data    Source code

       1             : /* Create new ELF header.
       2             :    Copyright (C) 1998, 1999, 2000, 2002, 2015 Red Hat, Inc.
       3             :    This file is part of elfutils.
       4             :    Written by Ulrich Drepper <drepper@redhat.com>, 1998.
       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 <stdlib.h>
      35             : #include <string.h>
      36             : 
      37             : #include "libelfP.h"
      38             : 
      39             : #ifndef LIBELFBITS
      40             : # define LIBELFBITS 32
      41             : #endif
      42             : 
      43             : 
      44             : ElfW2(LIBELFBITS,Ehdr) *
      45         471 : elfw2(LIBELFBITS,newehdr) (Elf *elf)
      46             : {
      47         471 :   ElfW2(LIBELFBITS,Ehdr) *result;
      48             : 
      49         471 :   if (elf == NULL)
      50             :     return NULL;
      51             : 
      52         471 :   if (unlikely (elf->kind != ELF_K_ELF))
      53             :     {
      54           0 :       __libelf_seterrno (ELF_E_INVALID_HANDLE);
      55           0 :       return NULL;
      56             :     }
      57             : 
      58         471 :   rwlock_wrlock (elf->lock);
      59             : 
      60         471 :   if (elf->class == 0)
      61         438 :     elf->class = ELFW(ELFCLASS,LIBELFBITS);
      62          33 :   else if (unlikely (elf->class != ELFW(ELFCLASS,LIBELFBITS)))
      63             :     {
      64           0 :       __libelf_seterrno (ELF_E_INVALID_CLASS);
      65           0 :       result = NULL;
      66           0 :       goto out;
      67             :     }
      68             : 
      69             :   /* Don't create an ELF header if one already exists.  */
      70         471 :   if (elf->state.ELFW(elf,LIBELFBITS).ehdr == NULL)
      71             :     {
      72             :       /* We use the memory in the ELF descriptor.  */
      73         940 :       elf->state.ELFW(elf,LIBELFBITS).ehdr =
      74         470 :         &elf->state.ELFW(elf,LIBELFBITS).ehdr_mem;
      75             : 
      76             :       /* We clear this memory.  */
      77         470 :       memset (elf->state.ELFW(elf,LIBELFBITS).ehdr, '\0',
      78             :               sizeof (ElfW2(LIBELFBITS,Ehdr)));
      79             : 
      80             :       /* Mark the ELF header has modified.  */
      81         470 :       elf->state.ELFW(elf,LIBELFBITS).ehdr_flags |= ELF_F_DIRTY;
      82             :     }
      83             : 
      84         471 :   result = elf->state.ELFW(elf,LIBELFBITS).ehdr;
      85             : 
      86             :  out:
      87             :   rwlock_unlock (elf->lock);
      88             : 
      89             :   return result;
      90             : }
      91             : INTDEF(elfw2(LIBELFBITS,newehdr))