Use balloc to extend Ifile.

[unix-history] / usr / src / usr.sbin / kgmon / kgmon.c

/*
* Copyright (c) 1983, 1992 The Regents of the University of California.
* All rights reserved.
*
* %sccs.include.redist.c%
*/

#ifndef lint
char copyright[] =
"@(#) Copyright (c) 1983, 1992 The Regents of the University of California.\n\
All rights reserved.\n";
#endif /* not lint */

#ifndef lint
static char sccsid[] = "@(#)kgmon.c     5.14 (Berkeley) %G%";
#endif /* not lint */

#include <sys/param.h>
#include <sys/file.h>
#include <sys/gmon.h>
#include <errno.h>
#include <kvm.h>
#include <limits.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <nlist.h>
#include <ctype.h>
#include <paths.h>

struct nlist nl[] = {
#define N_GMONPARAM     0
       { "__gmonparam" },
#define N_KCOUNT        1
       { "_kcount" },
#define N_PROFHZ        2
       { "_profhz" },
       0,
};

/*
* We should call this _gmonparam for consistency, but that would cause a 
* problem if we want to profile this program itself.
*/
struct gmonparam gmonparam;

u_short *kcount;
int profhz;

kvm_t   *kd;
int     bflag, hflag, kflag, rflag, pflag;
int     debug = 0;

#define KREAD(kd, addr, s)\
       kvm_read(kd, addr, (void *)(s), sizeof*(s)) != (sizeof*(s))

/*
* Build the gmon header and write it to a file.
*/
void
dumphdr(FILE *fp, struct gmonparam *p, int ksize)
{
       struct gmonhdr h;

       /* zero out the unused fields */
       bzero(&h, sizeof(h));

       h.lpc = p->lowpc;
       h.hpc = p->highpc;
       h.ncnt = ksize + sizeof(h);
       h.version = GMONVERSION;
       h.profrate = profhz;

       fwrite((char *)&h, sizeof(h), 1, fp);
}

/*
* Dump a range of kernel memory to a file.
*/
void
dumpbuf(FILE *fp, u_long addr, int cc)
{
       int ret, n;
       char buf[8192];

       while (cc > 0) {
               n = MIN(cc, sizeof(buf));
               if ((ret = kvm_read(kd, addr, buf, n)) != n) {
                       (void)fprintf(stderr,
                           "kgmon: read kmem: read %d, got %d: %s\n",
                           n, ret, kvm_geterr(kd));
                       exit(4);
               }
               if ((ret = fwrite(buf, n, 1, fp)) != 1) {
                       perror("kgmon: gmon.out");
                       exit(1);
               }
               addr += n;
               cc -= n;
       }
}

/*
* Enable or disable kernel profiling according to the state variable.
*/
void
setprof(int state)
{
       struct gmonparam *p = (struct gmonparam *)nl[N_GMONPARAM].n_value;

       if (kvm_write(kd, (u_long)&p->state, (void *)&state, sizeof(state)) !=
           sizeof(state))
               (void)fprintf(stderr,
                   "kgmon: warning: can't turn profiling %s\n",
                   state == GMON_PROF_OFF ? "off" : "on");
}

/*
* Build the gmon.out file.
*/
void
dumpstate(struct gmonparam *p)
{
       register FILE *fp;
       struct rawarc rawarc;
       struct tostruct *tos;
       u_long frompc, addr;
       u_short *froms;
       int i, n;
       int fromindex, endfrom, toindex;
       u_int fromssize, tossize, ksize;

       setprof(GMON_PROF_OFF);
       fp = fopen("gmon.out", "w");
       if (fp == 0) {
               perror("gmon.out");
               return;
       }
       ksize = p->textsize / HISTFRACTION;
       dumphdr(fp, p, ksize);
       dumpbuf(fp, (u_long)kcount, ksize);

       fromssize = p->textsize / HASHFRACTION;
       froms = (u_short *)malloc(fromssize);
       i = kvm_read(kd, (u_long)p->froms, (void *)froms, fromssize);
       if (i != fromssize) {
               (void)fprintf(stderr, "kgmon: read kmem: read %u, got %d: %s",
                   fromssize, i, strerror(errno));
               exit(5);
       }
       tossize = (p->textsize * ARCDENSITY / 100) * sizeof(struct tostruct);
       tos = (struct tostruct *)malloc(tossize);
       i = kvm_read(kd, (u_long)p->tos, (void *)tos, tossize);
       if (i != tossize) {
               (void)fprintf(stderr, "kgmon: read kmem: read %u, got %d: %s",
                   tossize, i, kvm_geterr(kd));
               exit(6);
       }
       if (debug)
               (void)fprintf(stderr, "lowpc 0x%x, textsize 0x%x\n",
                             p->lowpc, p->textsize);
       endfrom = fromssize / sizeof(*froms);
       for (fromindex = 0; fromindex < endfrom; ++fromindex) {
               if (froms[fromindex] == 0)
                       continue;
               frompc = (u_long)p->lowpc +
                   (fromindex * HASHFRACTION * sizeof(*froms));
               for (toindex = froms[fromindex]; toindex != 0;
                  toindex = tos[toindex].link) {
                       if (debug)
                           (void)fprintf(stderr,
                           "[mcleanup] frompc 0x%x selfpc 0x%x count %d\n" ,
                           frompc, tos[toindex].selfpc, tos[toindex].count);
                       rawarc.raw_frompc = frompc;
                       rawarc.raw_selfpc = (u_long)tos[toindex].selfpc;
                       rawarc.raw_count = tos[toindex].count;
                       fwrite((char *)&rawarc, sizeof(rawarc), 1, fp);
               }
       }
       fclose(fp);
}

/*
* Zero out a region of kernel memory.
*/
int
kzero(u_long addr, int cc)
{
       static char zbuf[MAXBSIZE];

       while (cc > 0) {
               register int n = MIN(cc, sizeof(zbuf));

               if (kvm_write(kd, addr, zbuf, n) != n)
                       return (-1);
               addr += n;
               cc -= n;
       }
       return (0);
}

/*
* Reset the kernel profiling date structures.
*/
void
reset(struct gmonparam *p)
{
       int fromssize, tossize, ksize;

       setprof(GMON_PROF_OFF);

       ksize = p->textsize / HISTFRACTION;
       if (kzero((u_long)kcount, ksize)) {
               (void)fprintf(stderr, "kgmon: sbuf write: %s\n",
                   kvm_geterr(kd));
               exit(7);
       }
       fromssize = p->textsize / HASHFRACTION;
       if (kzero((u_long)p->froms, fromssize)) {
               (void)fprintf(stderr, "kgmon: kfroms write: %s\n",
                   kvm_geterr(kd));
               exit(8);
       }
       tossize = (p->textsize * ARCDENSITY / 100) * sizeof(struct tostruct);
       if (kzero((u_long)p->tos, tossize)) {
               (void)fprintf(stderr, "kgmon: ktos write: %s\n",
                   kvm_geterr(kd));
               exit(9);
       }
}

int
main(int argc, char **argv)
{
       extern char *optarg;
       extern int optind;
       int ch, mode, disp, openmode;
       char *system, *kmemf;
       char errbuf[_POSIX2_LINE_MAX];

       kmemf = NULL;
       system = NULL;
       while ((ch = getopt(argc, argv, "M:N:bhpr")) != EOF) {
               switch((char)ch) {

               case 'M':
                       kmemf = optarg;
                       kflag = 1;
                       break;

               case 'N':
                       system = optarg;
                       break;

               case 'b':
                       bflag = 1;
                       break;

               case 'h':
                       hflag = 1;
                       break;

               case 'p':
                       pflag = 1;
                       break;

               case 'r':
                       rflag = 1;
                       break;

               default:
                       (void)fprintf(stderr,
                           "usage: kgmon [-bhrp] [-M core] [-N system]\n");
                       exit(1);
               }
       }
       argc -= optind;
       argv += optind;

#define BACKWARD_COMPATIBILITY
#ifdef  BACKWARD_COMPATIBILITY
       if (*argv) {
               system = *argv;
               if (*++argv) {
                       kmemf = *argv;
                       ++kflag;
               }
       }
#endif
       if (system == NULL)
               system = _PATH_UNIX;
       openmode = (bflag || hflag || pflag || rflag) ? O_RDWR : O_RDONLY;
       kd = kvm_openfiles(system, kmemf, NULL, openmode, errbuf);
       if (kd == NULL) {
               if (openmode == O_RDWR) {
                       openmode = O_RDONLY;
                       kd = kvm_openfiles(system, kmemf, NULL, O_RDONLY,
                           errbuf);
               }
               if (kd == NULL) {
                       (void)fprintf(stderr, "kgmon: kvm_openfiles: %s\n",
                           errbuf);
                       exit(2);
               }
               (void)fprintf(stderr, "kgmon: kernel opened read-only\n");
               if (rflag)
                       (void)fprintf(stderr, "-r supressed\n");
               if (bflag)
                       (void)fprintf(stderr, "-b supressed\n");
               if (hflag)
                       (void)fprintf(stderr, "-h supressed\n");
               rflag = bflag = hflag = 0;
       }
       if (kvm_nlist(kd, nl) < 0) {
               (void)fprintf(stderr, "kgmon: %s: no namelist\n", system);
               exit(2);
       }
       if (!nl[N_GMONPARAM].n_value) {
               (void)fprintf(stderr,
                   "kgmon: profiling not defined in kernel.\n");
               exit(10);
       }
       if (KREAD(kd, nl[N_GMONPARAM].n_value, &gmonparam))
               (void)fprintf(stderr,
                   "kgmon: read kmem: %s\n", kvm_geterr(kd));
       if (KREAD(kd, nl[N_KCOUNT].n_value, &kcount))
               (void)fprintf(stderr, "kgmon: read kmem: %s\n",
                             kvm_geterr(kd));
       if (KREAD(kd, nl[N_PROFHZ].n_value, &profhz))
               (void)fprintf(stderr, "kgmon: read kmem: %s\n",
                             kvm_geterr(kd));

       mode = gmonparam.state;
       if (hflag)
               disp = GMON_PROF_OFF;
       else if (bflag)
               disp = GMON_PROF_ON;
       else
               disp = mode;
       if (pflag) {
               if (openmode == O_RDONLY && mode == GMON_PROF_ON)
                       (void)fprintf(stderr, "data may be inconsistent\n");
               dumpstate(&gmonparam);
       }
       if (rflag)
               reset(&gmonparam);
       if (openmode == O_RDWR)
               setprof(disp);
       (void)fprintf(stdout, "kernel profiling is %s.\n",
                     disp == GMON_PROF_OFF ? "off" : "running");

       return (0);
}