#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);
}