148 lines
2.7 KiB
C
148 lines
2.7 KiB
C
#include <u.h>
|
|
#include <libc.h>
|
|
#include <bio.h>
|
|
#include <mach.h>
|
|
|
|
#define PCRES 8
|
|
|
|
struct COUNTER
|
|
{
|
|
char *name; /* function name */
|
|
long time; /* ticks spent there */
|
|
};
|
|
|
|
void
|
|
error(int perr, char *s)
|
|
{
|
|
fprint(2, "tprof: %s", s);
|
|
if(perr){
|
|
fprint(2, ": ");
|
|
perror(0);
|
|
}else
|
|
fprint(2, "\n");
|
|
exits(s);
|
|
}
|
|
|
|
int
|
|
compar(void *va, void *vb)
|
|
{
|
|
struct COUNTER *a, *b;
|
|
|
|
a = va;
|
|
b = vb;
|
|
if(a->time < b->time)
|
|
return -1;
|
|
if(a->time == b->time)
|
|
return 0;
|
|
return 1;
|
|
}
|
|
void
|
|
main(int argc, char *argv[])
|
|
{
|
|
int fd;
|
|
long i, j, k, n;
|
|
Dir *d;
|
|
char *name;
|
|
ulong *data;
|
|
ulong tbase, sum;
|
|
long delta;
|
|
Symbol s;
|
|
Biobuf outbuf;
|
|
Fhdr f;
|
|
struct COUNTER *cp;
|
|
char filebuf[128], *file;
|
|
|
|
if(argc != 2 && argc != 3)
|
|
error(0, "usage: tprof pid [binary]");
|
|
/*
|
|
* Read symbol table
|
|
*/
|
|
if(argc == 2){
|
|
file = filebuf;
|
|
snprint(filebuf, sizeof filebuf, "/proc/%s/text", argv[1]);
|
|
}else
|
|
file = argv[2];
|
|
|
|
fd = open(file, OREAD);
|
|
if(fd < 0)
|
|
error(1, file);
|
|
|
|
if (!crackhdr(fd, &f))
|
|
error(1, "read text header");
|
|
if (f.type == FNONE)
|
|
error(0, "text file not an a.out");
|
|
machbytype(f.type);
|
|
if (syminit(fd, &f) < 0)
|
|
error(1, "syminit");
|
|
close(fd);
|
|
/*
|
|
* Read timing data
|
|
*/
|
|
file = smprint("/proc/%s/profile", argv[1]);
|
|
fd = open(file, OREAD);
|
|
if(fd < 0)
|
|
error(1, file);
|
|
free(file);
|
|
d = dirfstat(fd);
|
|
if(d == nil)
|
|
error(1, "stat");
|
|
n = d->length/sizeof(data[0]);
|
|
if(n < 2)
|
|
error(0, "data file too short");
|
|
data = malloc(d->length);
|
|
if(data == 0)
|
|
error(1, "malloc");
|
|
if(read(fd, data, d->length) < 0)
|
|
error(1, "text read");
|
|
close(fd);
|
|
|
|
for(i=0; i<n; i++)
|
|
data[i] = machdata->swal(data[i]);
|
|
|
|
delta = data[0]-data[1];
|
|
print("total: %ld\n", data[0]);
|
|
if(data[0] == 0)
|
|
exits(0);
|
|
if (!textsym(&s, 0))
|
|
error(0, "no text symbols");
|
|
tbase = s.value & ~(mach->pgsize-1); /* align down to page */
|
|
print("TEXT %.8lux\n", tbase);
|
|
/*
|
|
* Accumulate counts for each function
|
|
*/
|
|
cp = 0;
|
|
k = 0;
|
|
for (i = 0, j = (s.value-tbase)/PCRES+2; j < n; i++) {
|
|
name = s.name; /* save name */
|
|
if (!textsym(&s, i)) /* get next symbol */
|
|
break;
|
|
sum = 0;
|
|
while (j < n && j*PCRES < s.value-tbase)
|
|
sum += data[j++];
|
|
if (sum) {
|
|
cp = realloc(cp, (k+1)*sizeof(struct COUNTER));
|
|
if (cp == 0)
|
|
error(1, "realloc");
|
|
cp[k].name = name;
|
|
cp[k].time = sum;
|
|
k++;
|
|
}
|
|
}
|
|
if (!k)
|
|
error(0, "no counts");
|
|
cp[k].time = 0; /* "etext" can take no time */
|
|
/*
|
|
* Sort by time and print
|
|
*/
|
|
qsort(cp, k, sizeof(struct COUNTER), compar);
|
|
Binit(&outbuf, 1, OWRITE);
|
|
Bprint(&outbuf, " ms %% sym\n");
|
|
while(--k>=0)
|
|
Bprint(&outbuf, "%6ld\t%3lld.%lld\t%s\n",
|
|
cp[k].time,
|
|
100LL*cp[k].time/delta,
|
|
(1000LL*cp[k].time/delta)%10,
|
|
cp[k].name);
|
|
exits(0);
|
|
}
|