usr.bin/gprof/aout.c

   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
  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
  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
 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
 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
 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 }