pjs/tools/jprof/libmalloc.cpp

321 строка
8.2 KiB
C++

// The contents of this file are subject to the Mozilla Public License
// Version 1.1 (the "License"); you may not use this file except in
// compliance with the License. You may obtain a copy of the License
// at http://www.mozilla.org/MPL/
//
// Software distributed under the License is distributed on an "AS IS"
// basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See
// the License for the specific language governing rights and
// limitations under the License.
//
// The Initial Developer of the Original Code is Kipp E.B. Hickman.
// Portions Copyright 1999 by Jim Nance
// The linux glibc hides part of sigaction if _POSIX_SOURCE is defined
#if defined(linux)
#undef _POSIX_SOURCE
#endif
// Some versions of glibc (i.e., the one that comes with RedHat 6.0 rather
// than 6.1) seem to do things a bit differently when libpthread is involved.
// If things don't work for you, try defining this:
//#define JPROF_PTHREAD_HACK
#include <stdlib.h>
#include <signal.h>
#include <unistd.h>
#include <fcntl.h>
#include <sys/time.h>
#include <sys/types.h>
#include "libmalloc.h"
#include <stdio.h>
#include <string.h>
#include <errno.h>
#include <setjmp.h>
#include <dlfcn.h>
#ifdef NTO
#include <sys/link.h>
extern r_debug _r_debug;
#else
#include <link.h>
#endif
#ifdef NTO
#define JB_BP 0x08
#include <setjmp.h>
#endif
static int gLogFD = -1;
static void startSignalCounter(unsigned long milisec);
static void writeStrStdout(const char* str)
{
int len = strlen(str);
if(len) {
int wc, tot=0;
do {
wc = write(1, str, len);
} while(wc>0 && (tot+=wc)<len);
}
}
//----------------------------------------------------------------------
#if defined(i386) || defined(_i386)
static void CrawlStack(malloc_log_entry* me, jmp_buf jb, char* first)
{
#ifdef NTO
u_long* bp = (u_long*) (jb[0].__jmpbuf_un.__savearea[JB_BP]);
#else
u_long* bp = (u_long*) (jb[0].__jmpbuf[JB_BP]);
#endif
u_long numpcs = 0;
#ifdef JPROF_PTHREAD_HACK
int skip = 3;
#else
// This is a linux hack we have to do to figure out where the signal was
// called from
me->pcs[numpcs++] = first;
int skip = 3;
#endif
while (numpcs < MAX_STACK_CRAWL) {
u_long* nextbp = (u_long*) *bp++;
u_long pc = *bp;
#ifdef JPROF_PTHREAD_HACK
if ((pc < 0x08000000) || ((pc > 0x7fffffff) && (skip <= 0)) || (nextbp < bp)) {
#else
if ((pc < 0x08000000) || (pc > 0x7fffffff) || (nextbp < bp)) {
#endif
break;
}
if (--skip < 0) {
me->pcs[numpcs++] = (char*) pc;
}
bp = nextbp;
}
me->numpcs = numpcs;
}
#endif
//----------------------------------------------------------------------
#if defined(linux) || defined(NTO)
static void DumpAddressMap()
{
// Turn off the timer so we dont get interrupts during shutdown
startSignalCounter(0);
int mfd = open(M_MAPFILE, O_CREAT|O_WRONLY|O_TRUNC, 0666);
if (mfd >= 0) {
malloc_map_entry mme;
link_map* map = _r_debug.r_map;
while (NULL != map) {
if (0 != map->l_addr) {
mme.nameLen = strlen(map->l_name);
mme.address = map->l_addr;
write(mfd, &mme, sizeof(mme));
write(mfd, map->l_name, mme.nameLen);
#if 0
write(1, map->l_name, mme.nameLen);
write(1, "\n", 1);
#endif
}
map = map->l_next;
}
close(mfd);
}
}
#endif
static void EndProfilingHook(int signum)
{
DumpAddressMap();
writeStrStdout("Jprof: profiling paused.\n");
}
//----------------------------------------------------------------------
static void
Log(u_long aTime, char *first)
{
// Static is simply to make debugging tollerable
static malloc_log_entry me;
me.delTime = aTime;
jmp_buf jb;
setjmp(jb);
CrawlStack(&me, jb, first);
#ifndef NTO
write(gLogFD, &me, offsetof(malloc_log_entry, pcs) + me.numpcs*sizeof(char*));
#else
printf("Neutrino is missing the pcs member of malloc_log_entry!! \n");
#endif
}
static int realTime = 0;
/* Lets interrupt at 10 Hz. This is so my log files don't get too large.
* This can be changed to a faster value latter. This timer is not
* programmed to reset, even though it is capable of doing so. This is
* to keep from getting interrupts from inside of the handler.
*/
static void startSignalCounter(unsigned long milisec)
{
struct itimerval tvalue;
tvalue.it_interval.tv_sec = 0;
tvalue.it_interval.tv_usec = 0;
tvalue.it_value.tv_sec = milisec/1000;
tvalue.it_value.tv_usec = (milisec%1000)*1000;
if (realTime) {
setitimer(ITIMER_REAL, &tvalue, NULL);
} else {
setitimer(ITIMER_PROF, &tvalue, NULL);
}
}
static long timerMiliSec = 50;
static void StackHook(
int signum,
siginfo_t *info,
void *mystry)
{
static struct timeval tFirst;
static int first=1;
size_t milisec;
if(first && !(first=0)) {
writeStrStdout("Jprof: received first timer signal\n");
gettimeofday(&tFirst, 0);
milisec = 0;
} else {
struct timeval tNow;
gettimeofday(&tNow, 0);
double usec = 1e6*(tNow.tv_sec - tFirst.tv_sec);
usec += (tNow.tv_usec - tFirst.tv_usec);
milisec = static_cast<size_t>(usec*1e-3);
}
#ifdef JPROF_PTHREAD_HACK
Log(milisec, NULL);
#else
// The mystry[19] thing is a hack to figure out where we were called from.
// By playing around with the debugger it looks like [19] contains the
// information I need.
// it's really ((ucontext_t *)mystry)->uc_mcontext.gregs[14] which is
// the EIP register when the handler was called
Log(milisec, ((char**)mystry)[19]);
#endif
startSignalCounter(timerMiliSec);
}
void setupProfilingStuff(void)
{
static int gFirstTime = 1;
if(gFirstTime && !(gFirstTime=0)) {
int startTimer = 1;
int doNotStart = 1;
int firstDelay = 0;
char *tst = getenv("JPROF_FLAGS");
/* Options from JPROF_FLAGS environment variable:
* JP_DEFER -> Wait for a SIGPROF (or SIGALRM, if JP_REALTIME
* is set) from userland before starting
* to generate them internally
* JP_START -> Install the signal handler
* JP_PERIOD -> Time between prifiler ticks
* JP_FIRST -> Extra delay before starting
* JP_REALTIME -> Take stack traces in intervals of real time
* rather than time used by the process (and the
* system for the process). This is useful for
* finding time spent by the X server.
*/
if(tst) {
if(strstr(tst, "JP_DEFER"))
{
doNotStart = 0;
startTimer = 0;
}
if(strstr(tst, "JP_START")) doNotStart = 0;
if(strstr(tst, "JP_REALTIME")) realTime = 1;
char *delay = strstr(tst,"JP_PERIOD=");
if(delay) {
double tmp = strtod(delay+10, NULL);
if(tmp>1e-3) {
timerMiliSec = static_cast<unsigned long>(1000 * tmp);
}
}
char *first = strstr(tst, "JP_FIRST=");
if(first) {
firstDelay = atol(first+9);
}
}
if(!doNotStart) {
if(gLogFD<0) {
gLogFD = open(M_LOGFILE, O_CREAT|O_WRONLY|O_TRUNC, 0666);
if(gLogFD<0) {
fprintf(stderr, "Unable to create " M_LOGFILE);
perror(":");
} else {
struct sigaction action;
sigset_t mset;
char buffer[1024];
// Dump out the address map when we terminate
atexit(DumpAddressMap);
sigemptyset(&mset);
action.sa_handler = NULL;
action.sa_sigaction = StackHook;
action.sa_mask = mset;
action.sa_flags = SA_RESTART | SA_SIGINFO;
if (realTime) {
sigaction(SIGALRM, &action, NULL);
} else {
sigaction(SIGPROF, &action, NULL);
}
// make it so a SIGUSR1 will stop the profiling
// Note: It currently does not close the logfile.
// This could be configurable (so that it could
// later be reopened).
struct sigaction stop_action;
stop_action.sa_handler = EndProfilingHook;
stop_action.sa_mask = mset;
stop_action.sa_flags = SA_RESTART;
sigaction(SIGUSR1, &stop_action, NULL);
sprintf(buffer, "Jprof: Initialized signal hander and set "
"timer for %lu mili-seconds with %d second initial delay\n",
timerMiliSec, firstDelay);
writeStrStdout(buffer);
if(startTimer) {
writeStrStdout("Jprof: started timer\n");
startSignalCounter(firstDelay*1000 + timerMiliSec);
}
}
}
}
} else {
printf("setupProfilingStuff() called multiple times\n");
}
}