1 /*-
2 * Copyright (c) 1983, 1993
3 * The Regents of the University of California. All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * 1. Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 * notice, this list of conditions and the following disclaimer in the
12 * documentation and/or other materials provided with the distribution.
13 * 4. Neither the name of the University nor the names of its contributors
14 * may be used to endorse or promote products derived from this software
15 * without specific prior written permission.
16 *
17 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
18 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
21 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
23 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
24 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
26 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
27 * SUCH DAMAGE.
28 */
29
30 #if 0
31 /* From: */
32 #ifndef lint
33 static char sccsid[] = "@(#)gprof.c 8.1 (Berkeley) 6/6/93";
34 #endif /* not lint */
35 #endif
36
37 #include <sys/cdefs.h>
38 __FBSDID("$FreeBSD$");
39
40 #include <netinet/in.h>
41
42 #include <a.out.h>
43 #include <err.h>
44 #include <string.h>
45
46 #include "gprof.h"
47
48 static void getstrtab(FILE *, const char *);
49 static void getsymtab(FILE *, const char *);
50 static void gettextspace(FILE *);
51 static bool funcsymbol(struct nlist *);
52
53 static char *strtab; /* string table in core */
54 static long ssiz; /* size of the string table */
55 static struct exec xbuf; /* exec header of a.out */
56
57 /* Things which get -E excluded by default. */
58 static char *excludes[] = { "mcount", "__mcleanup", NULL };
59
60 /*
61 * Set up string and symbol tables from a.out.
62 * and optionally the text space.
63 * On return symbol table is sorted by value.
64 *
65 * Returns 0 on success, -1 on failure.
66 */
67 int
aout_getnfile(const char * filename,char *** defaultEs)68 aout_getnfile(const char *filename, char ***defaultEs)
69 {
70 FILE *nfile;
71
72 nfile = fopen( filename ,"r");
73 if (nfile == NULL)
74 err( 1 , "%s", filename );
75 fread(&xbuf, 1, sizeof(xbuf), nfile);
76 if (N_BADMAG(xbuf)) {
77 fclose(nfile);
78 return -1;
79 }
80 getstrtab(nfile, filename);
81 getsymtab(nfile, filename);
82 gettextspace( nfile );
83 fclose(nfile);
84 # ifdef DEBUG
85 if ( debug & AOUTDEBUG ) {
86 register int j;
87
88 for (j = 0; j < nname; j++){
89 printf("[getnfile] 0X%08lx\t%s\n", nl[j].value, nl[j].name);
90 }
91 }
92 # endif /* DEBUG */
93 *defaultEs = excludes;
94 return 0;
95 }
96
97 static void
getstrtab(FILE * nfile,const char * filename)98 getstrtab(FILE *nfile, const char *filename)
99 {
100
101 fseek(nfile, (long)(N_SYMOFF(xbuf) + xbuf.a_syms), 0);
102 if (fread(&ssiz, sizeof (ssiz), 1, nfile) == 0)
103 errx( 1 , "%s: no string table (old format?)" , filename );
104 strtab = calloc(ssiz, 1);
105 if (strtab == NULL)
106 errx( 1 , "%s: no room for %ld bytes of string table", filename , ssiz);
107 if (fread(strtab+sizeof(ssiz), ssiz-sizeof(ssiz), 1, nfile) != 1)
108 errx( 1 , "%s: error reading string table" , filename );
109 }
110
111 /*
112 * Read in symbol table
113 */
114 static void
getsymtab(FILE * nfile,const char * filename)115 getsymtab(FILE *nfile, const char *filename)
116 {
117 register long i;
118 int askfor;
119 struct nlist nbuf;
120
121 /* pass1 - count symbols */
122 fseek(nfile, (long)N_SYMOFF(xbuf), 0);
123 nname = 0;
124 for (i = xbuf.a_syms; i > 0; i -= sizeof(struct nlist)) {
125 fread(&nbuf, sizeof(nbuf), 1, nfile);
126 if ( ! funcsymbol( &nbuf ) ) {
127 continue;
128 }
129 nname++;
130 }
131 if (nname == 0)
132 errx( 1 , "%s: no symbols" , filename );
133 askfor = nname + 1;
134 nl = (nltype *) calloc( askfor , sizeof(nltype) );
135 if (nl == 0)
136 errx( 1 , "no room for %zu bytes of symbol table" ,
137 askfor * sizeof(nltype) );
138
139 /* pass2 - read symbols */
140 fseek(nfile, (long)N_SYMOFF(xbuf), 0);
141 npe = nl;
142 nname = 0;
143 for (i = xbuf.a_syms; i > 0; i -= sizeof(struct nlist)) {
144 fread(&nbuf, sizeof(nbuf), 1, nfile);
145 if ( ! funcsymbol( &nbuf ) ) {
146 # ifdef DEBUG
147 if ( debug & AOUTDEBUG ) {
148 printf( "[getsymtab] rejecting: 0x%x %s\n" ,
149 nbuf.n_type , strtab + nbuf.n_un.n_strx );
150 }
151 # endif /* DEBUG */
152 continue;
153 }
154 npe->value = nbuf.n_value;
155 npe->name = strtab+nbuf.n_un.n_strx;
156 # ifdef DEBUG
157 if ( debug & AOUTDEBUG ) {
158 printf( "[getsymtab] %d %s 0x%08lx\n" ,
159 nname , npe -> name , npe -> value );
160 }
161 # endif /* DEBUG */
162 npe++;
163 nname++;
164 }
165 npe->value = -1;
166 }
167
168 /*
169 * read in the text space of an a.out file
170 */
171 static void
gettextspace(FILE * nfile)172 gettextspace(FILE *nfile)
173 {
174
175 textspace = (u_char *) malloc( xbuf.a_text );
176 if ( textspace == 0 ) {
177 warnx("no room for %u bytes of text space: can't do -c" ,
178 xbuf.a_text );
179 return;
180 }
181 (void) fseek( nfile , N_TXTOFF( xbuf ) , 0 );
182 if ( fread( textspace , 1 , xbuf.a_text , nfile ) != xbuf.a_text ) {
183 warnx("couldn't read text space: can't do -c");
184 free( textspace );
185 textspace = 0;
186 return;
187 }
188 }
189
190 static bool
funcsymbol(struct nlist * nlistp)191 funcsymbol(struct nlist *nlistp)
192 {
193 char *name, c;
194
195 /*
196 * must be a text symbol,
197 * and static text symbols don't qualify if aflag set.
198 */
199 if ( ! ( ( nlistp -> n_type == ( N_TEXT | N_EXT ) )
200 || ( ( nlistp -> n_type == N_TEXT ) && ( aflag == 0 ) ) ) ) {
201 return FALSE;
202 }
203 /*
204 * name must start with an underscore if uflag is set.
205 * can't have any `funny' characters in name,
206 * where `funny' means `.' (.o file names)
207 * need to make an exception for sparc .mul & co.
208 * perhaps we should just drop this code entirely...
209 */
210 name = strtab + nlistp -> n_un.n_strx;
211 if ( uflag && *name != '_' )
212 return FALSE;
213 #ifdef sparc
214 if ( *name == '.' ) {
215 char *p = name + 1;
216 if ( *p == 'u' )
217 p++;
218 if ( strcmp ( p, "mul" ) == 0 || strcmp ( p, "div" ) == 0 ||
219 strcmp ( p, "rem" ) == 0 )
220 return TRUE;
221 }
222 #endif
223 while ( (c = *name++) ) {
224 if ( c == '.' ) {
225 return FALSE;
226 }
227 }
228 return TRUE;
229 }
230