1 /* Update program header program header table entry.
2 Copyright (C) 2000, 2001, 2002 Red Hat, Inc.
3 Written by Ulrich Drepper <drepper@redhat.com>, 2000.
4
5 This program is free software; you can redistribute it and/or modify
6 it under the terms of the GNU General Public License as published by
7 the Free Software Foundation, version 2.
8
9 This program is distributed in the hope that it will be useful,
10 but WITHOUT ANY WARRANTY; without even the implied warranty of
11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 GNU General Public License for more details.
13
14 You should have received a copy of the GNU General Public License
15 along with this program; if not, write to the Free Software Foundation,
16 Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. */
17
18 #ifdef HAVE_CONFIG_H
19 # include <config.h>
20 #endif
21
22 #include <gelf.h>
23 #include <string.h>
24
25 #include "libelfP.h"
26
27
28 int
gelf_update_phdr(Elf * elf,int ndx,GElf_Phdr * src)29 gelf_update_phdr (Elf *elf, int ndx, GElf_Phdr *src)
30 {
31 int result = 0;
32
33 if (elf == NULL)
34 return 0;
35
36 if (unlikely (elf->kind != ELF_K_ELF))
37 {
38 __libelf_seterrno (ELF_E_INVALID_HANDLE);
39 return 0;
40 }
41
42 rwlock_wrlock (elf->lock);
43
44 if (elf->class == ELFCLASS32)
45 {
46 Elf32_Phdr *phdr = elf->state.elf32.phdr;
47
48 /* We have to convert the data to the 32 bit format. This might
49 overflow some fields so we have to test for this case before
50 copying. */
51 if (unlikely (src->p_offset > 0xffffffffull)
52 || unlikely (src->p_vaddr > 0xffffffffull)
53 || unlikely (src->p_paddr > 0xffffffffull)
54 || unlikely (src->p_filesz > 0xffffffffull)
55 || unlikely (src->p_memsz > 0xffffffffull)
56 || unlikely (src->p_align > 0xffffffffull))
57 {
58 __libelf_seterrno (ELF_E_INVALID_DATA);
59 goto out;
60 }
61
62 if (phdr == NULL)
63 {
64 phdr = INTUSE(elf32_getphdr) (elf);
65 if (phdr == NULL)
66 /* The error number is already set. */
67 goto out;
68 }
69
70 /* Test whether the index is ok. */
71 if (unlikely (ndx >= elf->state.elf32.ehdr->e_phnum))
72 {
73 __libelf_seterrno (ELF_E_INVALID_INDEX);
74 goto out;
75 }
76
77 /* Now correct the pointer to point to the correct element. */
78 phdr += ndx;
79
80 #define COPY(name) \
81 phdr->name = src->name
82 COPY (p_type);
83 COPY (p_offset);
84 COPY (p_vaddr);
85 COPY (p_paddr);
86 COPY (p_filesz);
87 COPY (p_memsz);
88 COPY (p_flags);
89 COPY (p_align);
90 }
91 else
92 {
93 Elf64_Phdr *phdr = elf->state.elf64.phdr;
94
95 if (phdr == NULL)
96 {
97 phdr = INTUSE(elf64_getphdr) (elf);
98 if (phdr == NULL)
99 /* The error number is already set. */
100 goto out;
101 }
102
103 /* Test whether the index is ok. */
104 if (unlikely (ndx >= elf->state.elf64.ehdr->e_phnum))
105 {
106 __libelf_seterrno (ELF_E_INVALID_INDEX);
107 goto out;
108 }
109
110 /* Just copy the data. */
111 memcpy (phdr + ndx, src, sizeof (Elf64_Phdr));
112 }
113
114 result = 1;
115
116 out:
117 rwlock_unlock (elf->lock);
118
119 return result;
120 }
121