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     int         valcmp();
  72
  73     nfile = fopen( filename ,"r");
  74     if (nfile == NULL)
  75         err( 1 , "%s", filename );
  76     fread(&xbuf, 1, sizeof(xbuf), nfile);
  77     if (N_BADMAG(xbuf)) {
  78         fclose(nfile);
  79         return -1;
  80     }
  81     getstrtab(nfile, filename);
  82     getsymtab(nfile, filename);
  83     gettextspace( nfile );
  84     fclose(nfile);
  85 #   ifdef DEBUG
  86         if ( debug & AOUTDEBUG ) {
  87             register int j;
  88
  89             for (j = 0; j < nname; j++){
  90                 printf("[getnfile] 0X%08lx\t%s\n", nl[j].value, nl[j].name);
  91             }
  92         }
  93 #   endif /* DEBUG */
  94     *defaultEs = excludes;
  95     return 0;
  96 }
  97
  98 static void
  99 getstrtab(FILE *nfile, const char *filename)
 100 {
 101
 102     fseek(nfile, (long)(N_SYMOFF(xbuf) + xbuf.a_syms), 0);
 103     if (fread(&ssiz, sizeof (ssiz), 1, nfile) == 0)
 104         errx( 1 , "%s: no string table (old format?)" , filename );
 105     strtab = calloc(ssiz, 1);
 106     if (strtab == NULL)
 107         errx( 1 , "%s: no room for %ld bytes of string table", filename , ssiz);
 108     if (fread(strtab+sizeof(ssiz), ssiz-sizeof(ssiz), 1, nfile) != 1)
 109         errx( 1 , "%s: error reading string table" , filename );
 110 }
 111
 112     /*
 113      * Read in symbol table
 114      */
 115 static void
 116 getsymtab(FILE *nfile, const char *filename)
 117 {
 118     register long       i;
 119     int                 askfor;
 120     struct nlist        nbuf;
 121
 122     /* pass1 - count symbols */
 123     fseek(nfile, (long)N_SYMOFF(xbuf), 0);
 124     nname = 0;
 125     for (i = xbuf.a_syms; i > 0; i -= sizeof(struct nlist)) {
 126         fread(&nbuf, sizeof(nbuf), 1, nfile);
 127         if ( ! funcsymbol( &nbuf ) ) {
 128             continue;
 129         }
 130         nname++;
 131     }
 132     if (nname == 0)
 133         errx( 1 , "%s: no symbols" , filename );
 134     askfor = nname + 1;
 135     nl = (nltype *) calloc( askfor , sizeof(nltype) );
 136     if (nl == 0)
 137         errx( 1 , "no room for %zu bytes of symbol table" ,
 138                 askfor * sizeof(nltype) );
 139
 140     /* pass2 - read symbols */
 141     fseek(nfile, (long)N_SYMOFF(xbuf), 0);
 142     npe = nl;
 143     nname = 0;
 144     for (i = xbuf.a_syms; i > 0; i -= sizeof(struct nlist)) {
 145         fread(&nbuf, sizeof(nbuf), 1, nfile);
 146         if ( ! funcsymbol( &nbuf ) ) {
 147 #           ifdef DEBUG
 148                 if ( debug & AOUTDEBUG ) {
 149                     printf( "[getsymtab] rejecting: 0x%x %s\n" ,
 150                             nbuf.n_type , strtab + nbuf.n_un.n_strx );
 151                 }
 152 #           endif /* DEBUG */
 153             continue;
 154         }
 155         npe->value = nbuf.n_value;
 156         npe->name = strtab+nbuf.n_un.n_strx;
 157 #       ifdef DEBUG
 158             if ( debug & AOUTDEBUG ) {
 159                 printf( "[getsymtab] %d %s 0x%08lx\n" ,
 160                         nname , npe -> name , npe -> value );
 161             }
 162 #       endif /* DEBUG */
 163         npe++;
 164         nname++;
 165     }
 166     npe->value = -1;
 167 }
 168
 169     /*
 170      *  read in the text space of an a.out file
 171      */
 172 static void
 173 gettextspace(FILE *nfile)
 174 {
 175
 176     textspace = (u_char *) malloc( xbuf.a_text );
 177     if ( textspace == 0 ) {
 178         warnx("no room for %u bytes of text space: can't do -c" ,
 179                   xbuf.a_text );
 180         return;
 181     }
 182     (void) fseek( nfile , N_TXTOFF( xbuf ) , 0 );
 183     if ( fread( textspace , 1 , xbuf.a_text , nfile ) != xbuf.a_text ) {
 184         warnx("couldn't read text space: can't do -c");
 185         free( textspace );
 186         textspace = 0;
 187         return;
 188     }
 189 }
 190
 191 static bool
 192 funcsymbol(struct nlist *nlistp)
 193 {
 194     char        *name, c;
 195
 196         /*
 197          *      must be a text symbol,
 198          *      and static text symbols don't qualify if aflag set.
 199          */
 200     if ( ! (  ( nlistp -> n_type == ( N_TEXT | N_EXT ) )
 201            || ( ( nlistp -> n_type == N_TEXT ) && ( aflag == 0 ) ) ) ) {
 202         return FALSE;
 203     }
 204         /*
 205          *      name must start with an underscore if uflag is set.
 206          *      can't have any `funny' characters in name,
 207          *      where `funny' means `.' (.o file names)
 208          *      need to make an exception for sparc .mul & co.
 209          *      perhaps we should just drop this code entirely...
 210          */
 211     name = strtab + nlistp -> n_un.n_strx;
 212     if ( uflag && *name != '_' )
 213         return FALSE;
 214 #ifdef sparc
 215     if ( *name == '.' ) {
 216         char *p = name + 1;
 217         if ( *p == 'u' )
 218             p++;
 219         if ( strcmp ( p, "mul" ) == 0 || strcmp ( p, "div" ) == 0 ||
 220              strcmp ( p, "rem" ) == 0 )
 221                 return TRUE;
 222     }
 223 #endif
 224     while ( (c = *name++) ) {
 225         if ( c == '.' ) {
 226             return FALSE;
 227         }
 228     }
 229     return TRUE;
 230 }