xref: /illumos-gate/usr/src/cmd/sgs/libelf/common/getphnum.c (revision d5ace9454616652a717c9831d949dffa319381f9)
1 /*
2  * CDDL HEADER START
3  *
4  * The contents of this file are subject to the terms of the
5  * Common Development and Distribution License, Version 1.0 only
6  * (the "License").  You may not use this file except in compliance
7  * with the License.
8  *
9  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
10  * or http://www.opensolaris.org/os/licensing.
11  * See the License for the specific language governing permissions
12  * and limitations under the License.
13  *
14  * When distributing Covered Code, include this CDDL HEADER in each
15  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
16  * If applicable, add the following below this CDDL HEADER, with the
17  * fields enclosed by brackets "[]" replaced with your own identifying
18  * information: Portions Copyright [yyyy] [name of copyright owner]
19  *
20  * CDDL HEADER END
21  */
22 /*
23  * Copyright 2005 Sun Microsystems, Inc.  All rights reserved.
24  * Use is subject to license terms.
25  */
26 
27 #pragma ident	"%Z%%M%	%I%	%E% SMI"
28 
29 #include <string.h>
30 #include <gelf.h>
31 #include <decl.h>
32 #include <msg.h>
33 
34 int
35 elf_getphnum(Elf *elf, size_t *phnum)
36 {
37 	GElf_Ehdr	ehdr;
38 	Elf_Scn		*scn;
39 	GElf_Shdr	shdr0;
40 
41 	if (gelf_getehdr(elf, &ehdr) == NULL)
42 		return (0);
43 
44 	if (ehdr.e_phnum != PN_XNUM) {
45 		*phnum = ehdr.e_phnum;
46 		return (1);
47 	}
48 
49 	if ((scn = elf_getscn(elf, 0)) == NULL ||
50 	    gelf_getshdr(scn, &shdr0) == NULL)
51 		return (0);
52 
53 	if (shdr0.sh_info == 0)
54 		*phnum = ehdr.e_phnum;
55 	else
56 		*phnum = shdr0.sh_info;
57 
58 	return (1);
59 }
60