LCOV - code coverage report
Current view: top level - libelf - elf_update.c (source / functions) Coverage Total Hit
Test: elfutils-0.193 Lines: 70.1 % 87 61
Test Date: 2025-08-30 14:31:09 Functions: 100.0 % 2 2
Legend: Lines:     hit not hit
Branches: + taken - not taken # not executed
Branches: 65.1 % 86 56

             Branch data     Line data    Source code
       1                 :             : /* Update data structures for changes and write them out.
       2                 :             :    Copyright (C) 1999, 2000, 2001, 2002, 2004, 2005, 2006, 2015 Red Hat, Inc.
       3                 :             :    This file is part of elfutils.
       4                 :             :    Contributed by Ulrich Drepper <drepper@redhat.com>, 1999.
       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 <libelf.h>
      35                 :             : #include <fcntl.h>
      36                 :             : #include <sys/stat.h>
      37                 :             : 
      38                 :             : #include "libelfP.h"
      39                 :             : 
      40                 :             : 
      41                 :             : static int64_t
      42                 :        1262 : write_file (Elf *elf, int64_t size, int change_bo, size_t shnum)
      43                 :             : {
      44                 :        1262 :   int class = elf->class;
      45                 :             : 
      46                 :             :   /* Check the mode bits now, before modification might change them.  */
      47                 :        1262 :   struct stat st;
      48         [ -  + ]:        1262 :   if (unlikely (fstat (elf->fildes, &st) != 0))
      49                 :             :     {
      50                 :           0 :       __libelf_seterrno (ELF_E_WRITE_ERROR);
      51                 :           0 :       return -1;
      52                 :             :     }
      53                 :             : 
      54                 :             :   /* Adjust the size in any case.  We do this even if we use `write'.
      55                 :             :      We cannot do this if this file is in an archive.  We also don't
      56                 :             :      do it *now* if we are shortening the file since this would
      57                 :             :      prevent programs to use the data of the file in generating the
      58                 :             :      new file.  We truncate the file later in this case.  */
      59         [ +  - ]:        1262 :   if (elf->parent == NULL
      60         [ +  - ]:        1262 :       && (elf->maximum_size == ~((size_t) 0)
      61         [ +  + ]:        1262 :           || (size_t) size > elf->maximum_size)
      62         [ -  + ]:        1150 :       && unlikely (ftruncate (elf->fildes, size) != 0))
      63                 :             :     {
      64                 :           0 :       __libelf_seterrno (ELF_E_WRITE_ERROR);
      65                 :           0 :       return -1;
      66                 :             :     }
      67                 :             : 
      68                 :             :   /* Try to map the file if this isn't done yet.  */
      69   [ +  +  +  + ]:        1262 :   if (elf->map_address == NULL && elf->cmd == ELF_C_WRITE_MMAP)
      70                 :             :     {
      71                 :         258 :       elf->map_address = mmap (NULL, size, PROT_READ | PROT_WRITE,
      72                 :             :                                MAP_SHARED, elf->fildes, 0);
      73         [ -  + ]:         258 :       if (unlikely (elf->map_address == MAP_FAILED))
      74                 :           0 :         elf->map_address = NULL;
      75                 :             :       else
      76                 :         258 :         elf->flags |= ELF_F_MMAPPED;
      77                 :             :     }
      78                 :             : 
      79         [ +  + ]:        1262 :   if (elf->map_address != NULL)
      80                 :             :     {
      81                 :             :       /* When using mmap we want to make sure the file content is
      82                 :             :          really there. Only using ftruncate might mean the file is
      83                 :             :          extended, but space isn't allocated yet.  This might cause a
      84                 :             :          SIGBUS once we write into the mmapped space and the disk is
      85                 :             :          full.  In glibc posix_fallocate is required to extend the
      86                 :             :          file and allocate enough space even if the underlying
      87                 :             :          filesystem would normally return EOPNOTSUPP.  But other
      88                 :             :          implementations might not work as expected.  And the glibc
      89                 :             :          fallback case might fail (with unexpected errnos) in some cases.
      90                 :             :          So we only report an error when the call fails and errno is
      91                 :             :          ENOSPC. Otherwise we ignore the error and treat it as just hint.  */
      92         [ +  - ]:         290 :       if (elf->parent == NULL
      93         [ +  - ]:         290 :           && (elf->maximum_size == ~((size_t) 0)
      94         [ +  + ]:         290 :               || (size_t) size > elf->maximum_size))
      95                 :             :         {
      96         [ -  + ]:         286 :           if (unlikely (posix_fallocate (elf->fildes, 0, size) != 0))
      97         [ #  # ]:           0 :             if (errno == ENOSPC)
      98                 :             :               {
      99                 :           0 :                 __libelf_seterrno (ELF_E_WRITE_ERROR);
     100                 :           0 :                 return -1;
     101                 :             :               }
     102                 :             : 
     103                 :             :           /* Extend the mmap address if needed.  */
     104         [ +  + ]:         286 :           if (elf->cmd == ELF_C_RDWR_MMAP
     105         [ +  - ]:          28 :               && (size_t) size > elf->maximum_size)
     106                 :             :             {
     107                 :             : #ifdef HAVE_MREMAP
     108         [ -  + ]:          28 :               if (mremap (elf->map_address, elf->maximum_size,
     109                 :             :                           size, 0) == MAP_FAILED)
     110                 :             : #endif
     111                 :             :                 {
     112                 :           0 :                   __libelf_seterrno (ELF_E_WRITE_ERROR);
     113                 :           0 :                   return -1;
     114                 :             :                 }
     115                 :          28 :               elf->maximum_size = size;
     116                 :             :             }
     117                 :             : 
     118                 :             :         }
     119                 :             : 
     120                 :             :       /* The file is mmaped.  */
     121         [ +  - ]:         290 :       if ((class == ELFCLASS32
     122                 :         138 :            ? __elf32_updatemmap (elf, change_bo, shnum)
     123         [ +  + ]:         428 :            : __elf64_updatemmap (elf, change_bo, shnum)) != 0)
     124                 :             :         /* Some problem while writing.  */
     125                 :             :         size = -1;
     126                 :             :     }
     127                 :             :   else
     128                 :             :     {
     129                 :             :       /* The file is not mmaped.  */
     130         [ -  + ]:         972 :       if ((class == ELFCLASS32
     131                 :         404 :            ? __elf32_updatefile (elf, change_bo, shnum)
     132         [ +  + ]:        1376 :            : __elf64_updatefile (elf, change_bo, shnum)) != 0)
     133                 :             :         /* Some problem while writing.  */
     134                 :             :         size = -1;
     135                 :             :     }
     136                 :             : 
     137                 :             :   /* Reduce the file size if necessary.  */
     138         [ -  + ]:        1262 :   if (size != -1
     139         [ +  - ]:        1262 :       && elf->parent == NULL
     140         [ +  - ]:        1262 :       && elf->maximum_size != ~((size_t) 0)
     141         [ +  + ]:        1262 :       && (size_t) size < elf->maximum_size
     142         [ -  + ]:          82 :       && unlikely (ftruncate (elf->fildes, size) != 0))
     143                 :             :     {
     144                 :           0 :       __libelf_seterrno (ELF_E_WRITE_ERROR);
     145                 :           0 :       size = -1;
     146                 :             :     }
     147                 :             : 
     148                 :             :   /* POSIX says that ftruncate and write may clear the S_ISUID and S_ISGID
     149                 :             :      mode bits.  So make sure we restore them afterwards if they were set.
     150                 :             :      This is not atomic if someone else chmod's the file while we operate.  */
     151         [ -  + ]:        1262 :   if (size != -1
     152         [ -  + ]:        1262 :       && unlikely (st.st_mode & (S_ISUID | S_ISGID))
     153                 :             :       /* fchmod ignores the bits we cannot change.  */
     154         [ #  # ]:           0 :       && unlikely (fchmod (elf->fildes, st.st_mode) != 0))
     155                 :             :     {
     156                 :           0 :       __libelf_seterrno (ELF_E_WRITE_ERROR);
     157                 :           0 :       size = -1;
     158                 :             :     }
     159                 :             : 
     160         [ -  + ]:        1262 :   if (size != -1 && elf->parent == NULL)
     161                 :        1262 :     elf->maximum_size = size;
     162                 :             : 
     163                 :             :   return size;
     164                 :             : }
     165                 :             : 
     166                 :             : 
     167                 :             : int64_t
     168                 :        1462 : elf_update (Elf *elf, Elf_Cmd cmd)
     169                 :             : {
     170                 :        1462 :   size_t shnum;
     171                 :        1462 :   int64_t size;
     172                 :        1462 :   int change_bo = 0;
     173                 :             : 
     174                 :        1462 :   if (cmd != ELF_C_NULL
     175         [ +  + ]:        1462 :       && cmd != ELF_C_WRITE
     176         [ -  + ]:          18 :       && unlikely (cmd != ELF_C_WRITE_MMAP))
     177                 :             :     {
     178                 :           0 :       __libelf_seterrno (ELF_E_INVALID_CMD);
     179                 :           0 :       return -1;
     180                 :             :     }
     181                 :             : 
     182         [ -  + ]:        1462 :   if (elf == NULL)
     183                 :             :     return -1;
     184                 :             : 
     185         [ -  + ]:        1462 :   if (elf->kind != ELF_K_ELF)
     186                 :             :     {
     187                 :           0 :       __libelf_seterrno (ELF_E_INVALID_HANDLE);
     188                 :           0 :       return -1;
     189                 :             :     }
     190                 :             : 
     191                 :        1462 :   rwlock_wrlock (elf->lock);
     192                 :             : 
     193                 :             :   /* Make sure we have an ELF header.  */
     194         [ -  + ]:        1462 :   if (elf->state.elf.ehdr == NULL)
     195                 :             :     {
     196                 :           0 :       __libelf_seterrno (ELF_E_WRONG_ORDER_EHDR);
     197                 :           0 :       size = -1;
     198                 :           0 :       goto out;
     199                 :             :     }
     200                 :             : 
     201                 :             :   /* Determine the number of sections.  */
     202                 :        2924 :   shnum = (elf->state.elf.scns_last->cnt == 0
     203                 :             :            ? 0
     204         [ +  + ]:        1462 :            : 1 + elf->state.elf.scns_last->data[elf->state.elf.scns_last->cnt - 1].index);
     205                 :             : 
     206                 :             :   /* Update the ELF descriptor.  First, place the program header.  It
     207                 :             :      will come right after the ELF header.  The count the size of all
     208                 :             :      sections and finally place the section table.  */
     209                 :        2924 :   size = (elf->class == ELFCLASS32
     210                 :         644 :           ? __elf32_updatenull_wrlock (elf, &change_bo, shnum)
     211         [ +  + ]:        1462 :           : __elf64_updatenull_wrlock (elf, &change_bo, shnum));
     212         [ -  + ]:        1462 :   if (likely (size != -1)
     213                 :             :       /* See whether we actually have to write out the data.  */
     214         [ +  + ]:        1462 :       && (cmd == ELF_C_WRITE || cmd == ELF_C_WRITE_MMAP))
     215                 :             :     {
     216                 :        1262 :       if (elf->cmd != ELF_C_RDWR
     217         [ +  + ]:        1262 :           && elf->cmd != ELF_C_RDWR_MMAP
     218         [ +  + ]:        1044 :           && elf->cmd != ELF_C_WRITE
     219         [ -  + ]:         258 :           && unlikely (elf->cmd != ELF_C_WRITE_MMAP))
     220                 :             :         {
     221                 :           0 :           __libelf_seterrno (ELF_E_UPDATE_RO);
     222                 :           0 :           size = -1;
     223                 :             :         }
     224         [ -  + ]:        1262 :       else if (unlikely (elf->fildes == -1))
     225                 :             :         {
     226                 :             :           /* We closed the file already.  */
     227                 :           0 :           __libelf_seterrno (ELF_E_FD_DISABLED);
     228                 :           0 :           size = -1;
     229                 :             :         }
     230                 :             :       else
     231                 :        1262 :         size = write_file (elf, size, change_bo, shnum);
     232                 :             :     }
     233                 :             : 
     234                 :         200 :  out:
     235                 :             :   rwlock_unlock (elf->lock);
     236                 :             : 
     237                 :             :   return size;
     238                 :             : }
        

Generated by: LCOV version 2.0-1