1a1b5ec25Sjsg /*-
2a1b5ec25Sjsg * Copyright (c) 2006,2008 Joseph Koshy
3a1b5ec25Sjsg * All rights reserved.
4a1b5ec25Sjsg *
5a1b5ec25Sjsg * Redistribution and use in source and binary forms, with or without
6a1b5ec25Sjsg * modification, are permitted provided that the following conditions
7a1b5ec25Sjsg * are met:
8a1b5ec25Sjsg * 1. Redistributions of source code must retain the above copyright
9a1b5ec25Sjsg * notice, this list of conditions and the following disclaimer.
10a1b5ec25Sjsg * 2. Redistributions in binary form must reproduce the above copyright
11a1b5ec25Sjsg * notice, this list of conditions and the following disclaimer in the
12a1b5ec25Sjsg * documentation and/or other materials provided with the distribution.
13a1b5ec25Sjsg *
14a1b5ec25Sjsg * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15a1b5ec25Sjsg * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16a1b5ec25Sjsg * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17a1b5ec25Sjsg * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18a1b5ec25Sjsg * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19a1b5ec25Sjsg * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20a1b5ec25Sjsg * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21a1b5ec25Sjsg * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22a1b5ec25Sjsg * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23a1b5ec25Sjsg * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24a1b5ec25Sjsg * SUCH DAMAGE.
25a1b5ec25Sjsg */
26a1b5ec25Sjsg
27a1b5ec25Sjsg #include <assert.h>
28a1b5ec25Sjsg #include <gelf.h>
29a1b5ec25Sjsg
30a1b5ec25Sjsg #include "_libelf.h"
31a1b5ec25Sjsg
32*63b93652Sjsg ELFTC_VCSID("$Id: gelf_symshndx.c,v 1.2 2020/05/18 06:46:23 jsg Exp $");
33a1b5ec25Sjsg
34a1b5ec25Sjsg GElf_Sym *
gelf_getsymshndx(Elf_Data * d,Elf_Data * id,int ndx,GElf_Sym * dst,Elf32_Word * shindex)35a1b5ec25Sjsg gelf_getsymshndx(Elf_Data *d, Elf_Data *id, int ndx, GElf_Sym *dst,
36a1b5ec25Sjsg Elf32_Word *shindex)
37a1b5ec25Sjsg {
38a1b5ec25Sjsg int ec;
39a1b5ec25Sjsg Elf *e;
40a1b5ec25Sjsg size_t msz;
41a1b5ec25Sjsg Elf_Scn *scn;
42a1b5ec25Sjsg uint32_t sh_type;
43a1b5ec25Sjsg struct _Libelf_Data *ld, *lid;
44a1b5ec25Sjsg
45a1b5ec25Sjsg ld = (struct _Libelf_Data *) d;
46a1b5ec25Sjsg lid = (struct _Libelf_Data *) id;
47a1b5ec25Sjsg
48a1b5ec25Sjsg if (gelf_getsym(d, ndx, dst) == 0)
49a1b5ec25Sjsg return (NULL);
50a1b5ec25Sjsg
51a1b5ec25Sjsg if (lid == NULL || (scn = lid->d_scn) == NULL ||
52a1b5ec25Sjsg (e = scn->s_elf) == NULL || (e != ld->d_scn->s_elf) ||
53a1b5ec25Sjsg shindex == NULL) {
54a1b5ec25Sjsg LIBELF_SET_ERROR(ARGUMENT, 0);
55a1b5ec25Sjsg return (NULL);
56a1b5ec25Sjsg }
57a1b5ec25Sjsg
58a1b5ec25Sjsg ec = e->e_class;
59a1b5ec25Sjsg assert(ec == ELFCLASS32 || ec == ELFCLASS64);
60a1b5ec25Sjsg
61a1b5ec25Sjsg if (ec == ELFCLASS32)
62a1b5ec25Sjsg sh_type = scn->s_shdr.s_shdr32.sh_type;
63a1b5ec25Sjsg else
64a1b5ec25Sjsg sh_type = scn->s_shdr.s_shdr64.sh_type;
65a1b5ec25Sjsg
66a1b5ec25Sjsg if (_libelf_xlate_shtype(sh_type) != ELF_T_WORD ||
67a1b5ec25Sjsg id->d_type != ELF_T_WORD) {
68a1b5ec25Sjsg LIBELF_SET_ERROR(ARGUMENT, 0);
69a1b5ec25Sjsg return (NULL);
70a1b5ec25Sjsg }
71a1b5ec25Sjsg
72*63b93652Sjsg if ((msz = _libelf_msize(ELF_T_WORD, ec, e->e_version)) == 0)
73*63b93652Sjsg return (NULL);
74a1b5ec25Sjsg
75a1b5ec25Sjsg assert(ndx >= 0);
76a1b5ec25Sjsg
77a1b5ec25Sjsg if (msz * (size_t) ndx >= id->d_size) {
78a1b5ec25Sjsg LIBELF_SET_ERROR(ARGUMENT, 0);
79a1b5ec25Sjsg return (NULL);
80a1b5ec25Sjsg }
81a1b5ec25Sjsg
82a1b5ec25Sjsg *shindex = ((Elf32_Word *) id->d_buf)[ndx];
83a1b5ec25Sjsg
84a1b5ec25Sjsg return (dst);
85a1b5ec25Sjsg }
86a1b5ec25Sjsg
87a1b5ec25Sjsg int
gelf_update_symshndx(Elf_Data * d,Elf_Data * id,int ndx,GElf_Sym * gs,Elf32_Word xindex)88a1b5ec25Sjsg gelf_update_symshndx(Elf_Data *d, Elf_Data *id, int ndx, GElf_Sym *gs,
89a1b5ec25Sjsg Elf32_Word xindex)
90a1b5ec25Sjsg {
91a1b5ec25Sjsg int ec;
92a1b5ec25Sjsg Elf *e;
93a1b5ec25Sjsg size_t msz;
94a1b5ec25Sjsg Elf_Scn *scn;
95a1b5ec25Sjsg uint32_t sh_type;
96a1b5ec25Sjsg struct _Libelf_Data *ld, *lid;
97a1b5ec25Sjsg
98a1b5ec25Sjsg ld = (struct _Libelf_Data *) d;
99a1b5ec25Sjsg lid = (struct _Libelf_Data *) id;
100a1b5ec25Sjsg
101a1b5ec25Sjsg if (gelf_update_sym(d, ndx, gs) == 0)
102a1b5ec25Sjsg return (0);
103a1b5ec25Sjsg
104a1b5ec25Sjsg if (lid == NULL || (scn = lid->d_scn) == NULL ||
105a1b5ec25Sjsg (e = scn->s_elf) == NULL || (e != ld->d_scn->s_elf)) {
106a1b5ec25Sjsg LIBELF_SET_ERROR(ARGUMENT, 0);
107a1b5ec25Sjsg return (0);
108a1b5ec25Sjsg }
109a1b5ec25Sjsg
110a1b5ec25Sjsg ec = e->e_class;
111a1b5ec25Sjsg assert(ec == ELFCLASS32 || ec == ELFCLASS64);
112a1b5ec25Sjsg
113a1b5ec25Sjsg if (ec == ELFCLASS32)
114a1b5ec25Sjsg sh_type = scn->s_shdr.s_shdr32.sh_type;
115a1b5ec25Sjsg else
116a1b5ec25Sjsg sh_type = scn->s_shdr.s_shdr64.sh_type;
117a1b5ec25Sjsg
118a1b5ec25Sjsg if (_libelf_xlate_shtype(sh_type) != ELF_T_WORD ||
119a1b5ec25Sjsg d->d_type != ELF_T_WORD) {
120a1b5ec25Sjsg LIBELF_SET_ERROR(ARGUMENT, 0);
121a1b5ec25Sjsg return (0);
122a1b5ec25Sjsg }
123a1b5ec25Sjsg
124*63b93652Sjsg if ((msz = _libelf_msize(ELF_T_WORD, ec, e->e_version)) == 0)
125*63b93652Sjsg return (0);
126a1b5ec25Sjsg
127a1b5ec25Sjsg assert(ndx >= 0);
128a1b5ec25Sjsg
129a1b5ec25Sjsg if (msz * (size_t) ndx >= id->d_size) {
130a1b5ec25Sjsg LIBELF_SET_ERROR(ARGUMENT, 0);
131a1b5ec25Sjsg return (0);
132a1b5ec25Sjsg }
133a1b5ec25Sjsg
134a1b5ec25Sjsg *(((Elf32_Word *) id->d_buf) + ndx) = xindex;
135a1b5ec25Sjsg
136a1b5ec25Sjsg return (1);
137a1b5ec25Sjsg }
138