Branch data     Line data    Source code

       1                 :            : /* Create new ELF header.
       2                 :            :    Copyright (C) 1998, 1999, 2000, 2002 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                 :         64 : elfw2(LIBELFBITS,newehdr) (elf)
      46                 :            :      Elf *elf;
      47                 :            : {
      48                 :            :   ElfW2(LIBELFBITS,Ehdr) *result;
      49                 :            : 
      50         [ +  - ]:         64 :   if (elf == NULL)
      51                 :            :     return NULL;
      52                 :            : 
      53         [ -  + ]:         64 :   if (unlikely (elf->kind != ELF_K_ELF))
      54                 :            :     {
      55                 :          0 :       __libelf_seterrno (ELF_E_INVALID_HANDLE);
      56                 :          0 :       return NULL;
      57                 :            :     }
      58                 :            : 
      59                 :            :   rwlock_wrlock (elf->lock);
      60                 :            : 
      61         [ +  + ]:         64 :   if (elf->class == 0)
      62                 :         63 :     elf->class = ELFW(ELFCLASS,LIBELFBITS);
      63         [ -  + ]:          1 :   else if (unlikely (elf->class != ELFW(ELFCLASS,LIBELFBITS)))
      64                 :            :     {
      65                 :          0 :       __libelf_seterrno (ELF_E_INVALID_CLASS);
      66                 :          0 :       result = NULL;
      67                 :          0 :       goto out;
      68                 :            :     }
      69                 :            : 
      70                 :            :   /* Don't create an ELF header if one already exists.  */
      71         [ +  + ]:         64 :   if (elf->state.ELFW(elf,LIBELFBITS).ehdr == NULL)
      72                 :            :     {
      73                 :            :       /* We use the memory in the ELF descriptor.  */
      74                 :         63 :       elf->state.ELFW(elf,LIBELFBITS).ehdr =
      75                 :         63 :         &elf->state.ELFW(elf,LIBELFBITS).ehdr_mem;
      76                 :            : 
      77                 :            :       /* We clear this memory.  */
      78                 :         63 :       memset (elf->state.ELFW(elf,LIBELFBITS).ehdr, '\0',
      79                 :            :               sizeof (ElfW2(LIBELFBITS,Ehdr)));
      80                 :            : 
      81                 :            :       /* Mark the ELF header has modified.  */
      82                 :         63 :       elf->state.ELFW(elf,LIBELFBITS).ehdr_flags |= ELF_F_DIRTY;
      83                 :            :     }
      84                 :            : 
      85                 :         64 :   result = elf->state.ELFW(elf,LIBELFBITS).ehdr;
      86                 :            : 
      87                 :            :  out:
      88                 :            :   rwlock_unlock (elf->lock);
      89                 :            : 
      90                 :         64 :   return result;
      91                 :            : }
      92                 :            : INTDEF(elfw2(LIBELFBITS,newehdr))