X-Git-Url: http://git.megacz.com/?a=blobdiff_plain;f=rts%2FHpc.c;h=c474d8121b35c0d64949199979c872c40fae8e79;hb=150cc9e2e4657cc58bd7ec4c15e5cb72f2e1c0f6;hp=50730ea70bad4359092bcf43b9619245139335a3;hpb=ae8fc4d5492af76e7938f797142a3483afa7a770;p=ghc-hetmet.git diff --git a/rts/Hpc.c b/rts/Hpc.c index 50730ea..c474d81 100644 --- a/rts/Hpc.c +++ b/rts/Hpc.c @@ -2,27 +2,45 @@ * (c)2006 Galois Connections, Inc. */ -// #include "HsFFI.h" - #include #include #include #include #include -#include "HsFFI.h" + #include "Rts.h" #include "Hpc.h" +#include "Trace.h" + +#ifdef HAVE_UNISTD_H +#include +#endif + /* This is the runtime support for the Haskell Program Coverage (hpc) toolkit, * inside GHC. * */ -#define DEBUG_HPC 0 +#define WOP_SIZE 1024 + +static int hpc_inited = 0; // Have you started this component? +static int totalTickCount = 0; // How many ticks have we got to work with +static FILE *tixFile; // file being read/written +static int tix_ch; // current char +static StgWord64 magicTixNumber; // Magic/Hash number to mark .tix files + +static FILE *rixFile = NULL; // The tracer file/pipe (to debugger) +static FILE *rixCmdFile = NULL; // The tracer file/pipe (from debugger) +static StgWord64 rixCounter = 0; // The global event counter +static int debuggee_pid; + +typedef enum { + RixThreadFinishedOp = -1, + RixRaiseOp = -2, + RixFinishedOp = -3 +} HpcRixOp; -static int hpc_inited = 0; // Have you started this component? -static FILE *tixFile; // file being read/written -static int tix_ch; // current char typedef struct _Info { char *modName; // name of module @@ -44,17 +62,15 @@ Info *nextModule = 0; StgWord64 *tixBoxes = 0; // local copy of tixBoxes array, from file. int totalTixes = 0; // total number of tix boxes. - - static char *tixFilename; static void failure(char *msg) { - printf("Hpc failure: %s\n",msg); - printf("(perhaps remove .tix file?)\n"); + debugTrace(DEBUG_hpc,"hpc failure: %s\n",msg); + fprintf(stderr,"Hpc failure: %s\n",msg); + fprintf(stderr,"(perhaps remove .tix file?)\n"); exit(-1); } - static int init_open(char *filename) { tixFile = fopen(filename,"r"); @@ -67,8 +83,7 @@ static int init_open(char *filename) static void expect(char c) { if (tix_ch != c) { - printf("Hpc: parse failed (%c,%c)\n",tix_ch,c); - exit(-1); + failure("parse error when reading .tix file"); } tix_ch = getc(tixFile); } @@ -124,7 +139,7 @@ static void hpc_init(void) { expect('i'); expect('x'); ws(); - expectWord64(); + magicTixNumber = expectWord64(); ws(); expect('['); ws(); @@ -156,7 +171,8 @@ static void hpc_init(void) { if (tix_ch == ',') { expect(','); ws(); - }} + } + } expect(']'); ws(); tixBoxes = (StgWord64 *)calloc(totalTixes,sizeof(StgWord64)); @@ -173,6 +189,9 @@ static void hpc_init(void) { expect(']'); fclose(tixFile); + } else { + // later, we will find a binary specific + magicTixNumber = (StgWord64)0; } } @@ -181,14 +200,13 @@ static void hpc_init(void) { * of the tix file, or all zeros. */ -void +int hs_hpc_module(char *modName,int modCount,StgWord64 *tixArr) { Info *tmpModule, *lastModule; int i; + int offset = 0; -#if DEBUG_HPC - printf("hs_hpc_module(%s,%d)\n",modName,modCount); -#endif + debugTrace(DEBUG_hpc,"hs_hpc_module(%s,%d)",modName,modCount); hpc_init(); @@ -206,7 +224,7 @@ hs_hpc_module(char *modName,int modCount,StgWord64 *tixArr) { for(i=0;i < modCount;i++) { tixArr[i] = tixBoxes[i + tmpModule->tickOffset]; } - return; + return tmpModule->tickOffset; } lastModule = tmpModule; } @@ -231,9 +249,190 @@ hs_hpc_module(char *modName,int modCount,StgWord64 *tixArr) { lastModule->next=tmpModule; } -#if DEBUG_HPC - printf("end: hs_hpc_module\n"); -#endif + debugTrace(DEBUG_hpc,"end: hs_hpc_module"); + + return offset; +} + +static void breakPointCommand(HpcRixOp rixOp, StgThreadID rixTid); + +// Breakpointing +static StgThreadID previousTid = 0; +static StgWord64 rixBPCounter = 0; // The global event breakpoint counter +static int *tixBoxBP; +static HpcRixOp rixOpBack[WOP_SIZE]; // The actual op +static HpcRixOp rixTidBack[WOP_SIZE]; // Tid's before the op + +void +hs_hpc_raise_event(StgTSO *current_tso) { + hs_hpc_tick(RixRaiseOp,current_tso); +} + +void +hs_hpc_thread_finished_event(StgTSO *current_tso) { + hs_hpc_tick(RixThreadFinishedOp,current_tso); +} + +/* Called on every tick, dynamically, sending to our + * external record of program execution. + */ + +void +hs_hpc_tick(int rixOp, StgTSO *current_tso) { + + debugTrace(DEBUG_hpc,"hs_hpc_tick(%x)",rixOp); + + if (rixFile == NULL) { + return; + } + assert(rixCmdFile != NULL); + StgThreadID tid = (current_tso == 0) ? 0 : current_tso->id; + + // now check to see if we have met a breakpoint condition + if (rixCounter == rixBPCounter + || tid != previousTid) { + breakPointCommand(rixOp,tid); + } else { + if (rixOp >= 0) { + // Tix op + if (tixBoxBP[rixOp] == 1) { // reached a bp tixbox + breakPointCommand(rixOp,tid); + } + } else { + // record the special operation + breakPointCommand(rixOp,tid); + } + } + // update the history information. + previousTid = tid; + rixOpBack[rixCounter % WOP_SIZE] = rixOp; + rixTidBack[rixCounter % WOP_SIZE] = tid; + rixCounter++; + + debugTrace(DEBUG_hpc, "end: hs_hpc_tick"); +} + +static void +printEvent(FILE *out,StgWord64 rixCounter,StgThreadID rixTid,HpcRixOp rixOp) { + char prefixMsg[128]; + char suffixMsg[128]; + + sprintf(prefixMsg, + "Event %" PRIuWORD64 " %u ", + rixCounter, + (unsigned int)rixTid); + + switch(rixOp) { + case RixThreadFinishedOp: + sprintf(suffixMsg,"ThreadFinished"); + break; + case RixRaiseOp: + sprintf(suffixMsg,"Raise"); + break; + case RixFinishedOp: + sprintf(suffixMsg,"Finished"); + break; + default: + sprintf(suffixMsg,"%u",rixOp); + } + + fprintf(out,"%s%s\n",prefixMsg,suffixMsg); + debugTrace(DEBUG_hpc,"sending %s%s",prefixMsg,suffixMsg); +} + +static void +breakPointCommand(HpcRixOp rixOp, StgThreadID rixTid) { + StgWord64 tmp64 = 0; + unsigned int tmp = 0; + + if (getpid() != debuggee_pid) { + // We are not the original process, to do not issue + // any events, and do not try to talk to the debugger. + return; + } + + debugTrace(DEBUG_hpc,"breakPointCommand %d %x",rixOp,(unsigned int)rixTid); + + printEvent(rixFile,rixCounter,rixTid,rixOp); + fflush(rixFile); + + /* From here, you can ask some basic questions. + * + * c set the (one) counter breakpoint + * s set the (many) tickbox breakpoint + * u unset the (many) tickbox breakpoint + * h history + + * Note that you aways end up here on the first tick + * because the rixBPCounter starts equal to 0. + */ + int c = getc(rixCmdFile); + while(c != 10 && c != -1) { + switch(c) { + case 'c': // c1234 -- set counter breakpoint at 1234 + c = getc(rixCmdFile); + tmp64 = 0; + while(isdigit(c)) { + tmp64 = tmp64 * 10 + (c - '0'); + c = getc(rixCmdFile); + } + debugTrace(DEBUG_hpc,"setting countBP = %" PRIuWORD64,tmp64); + + rixBPCounter = tmp64; + break; + case 's': // s2323 -- set tick box breakpoint at 2323 + c = getc(rixCmdFile); + tmp = 0; + while(isdigit(c)) { + tmp = tmp * 10 + (c - '0'); + c = getc(rixCmdFile); + } + + debugTrace(DEBUG_hpc,"seting bp for tix %d",tmp); + + tixBoxBP[tmp] = 1; + break; + case 'u': // u2323 -- unset tick box breakpoint at 2323 + c = getc(rixCmdFile); + tmp = 0; + while(isdigit(c)) { + tmp = tmp * 10 + (c - '0'); + c = getc(rixCmdFile); + } + + debugTrace(DEBUG_hpc,"unseting bp for tix %d",tmp); + + tixBoxBP[tmp] = 0; + break; + case 'h': // h -- history of the last few (WOP_SIZE) steps + if (rixCounter > WOP_SIZE) { + tmp64 = rixCounter - WOP_SIZE; + } else { + tmp64 = 0; + } + for(;tmp64 < rixCounter;tmp64++) { + printEvent(rixFile, + tmp64, + rixTidBack[tmp64 % WOP_SIZE], + rixOpBack[tmp64 % WOP_SIZE]); + } + fflush(rixFile); + c = getc(rixCmdFile); + break; + default: + + debugTrace(DEBUG_hpc,"strange command from HPCRIX (%d)",c); + + c = getc(rixCmdFile); + } + while (c != 10) { // the end of the line + c = getc(rixCmdFile); // to the end of the line + } + c = getc(rixCmdFile); // the first char on the next command + } + + debugTrace(DEBUG_hpc,"leaving breakPointCommand"); + } /* This is called after all the modules have registered their local tixboxes, @@ -243,9 +442,9 @@ hs_hpc_module(char *modName,int modCount,StgWord64 *tixArr) { void startupHpc(void) { Info *tmpModule; -#if DEBUG_HPC - printf("startupHpc\n"); -#endif + char *hpcRix; + + debugTrace(DEBUG_hpc,"startupHpc"); if (hpc_inited == 0) { return; @@ -255,6 +454,7 @@ startupHpc(void) { if (tixBoxes) { for(;tmpModule != 0;tmpModule = tmpModule->next) { + totalTickCount += tmpModule->tickCount; if (!tmpModule->tixArr) { fprintf(stderr,"error: module %s did not register any hpc tick data\n", tmpModule->modName); @@ -263,8 +463,75 @@ startupHpc(void) { } } } + + // HPCRIX contains the name of the file to send our dynamic runtime output to (a named pipe). + + hpcRix = getenv("HPCRIX"); + if (hpcRix) { + int comma; + Info *tmpModule; + int rixFD, rixCmdFD; + int tixCount = 0; + + assert(hpc_inited); + + if (sscanf(hpcRix,"%d:%d",&rixFD,&rixCmdFD) != 2) { + /* Bad format for HPCRIX. + */ + debugTrace(DEBUG_hpc,"Bad HPCRIX (%s)",hpcRix); + exit(0); + } + + debugTrace(DEBUG_hpc,"found HPCRIX pipes: %d:%d",rixFD,rixCmdFD); + + rixFile = fdopen(rixFD,"w"); + assert(rixFile != NULL); + + rixCmdFile = fdopen(rixCmdFD,"r"); + assert(rixCmdFile != NULL); + + // If we fork a process, then we do not want ticks inside + // the sub-process to talk to the debugger. So we remember + // our pid at startup time, so we can check if we are still + // the original process. + + debuggee_pid = getpid(); + + comma = 0; + + fprintf(rixFile,"Starting %s\n",prog_name); + fprintf(rixFile,"["); + tmpModule = modules; + for(;tmpModule != 0;tmpModule = tmpModule->next) { + if (comma) { + fprintf(rixFile,","); + } else { + comma = 1; + } + fprintf(rixFile,"(\"%s\",%u)", + tmpModule->modName, + tmpModule->tickCount); + + tixCount += tmpModule->tickCount; + + debugTrace(DEBUG_hpc,"(tracer)%s: %u (offset=%u)\n", + tmpModule->modName, + tmpModule->tickCount, + tmpModule->tickOffset); + + } + fprintf(rixFile,"]\n"); + fflush(rixFile); + + // Allocate the tixBox breakpoint array + // These are set to 1 if you want to + // stop at a specific breakpoint + tixBoxBP = (int *)calloc(tixCount,sizeof(int)); + } + } + /* Called at the end of execution, to write out the Hpc *.tix file * for this exection. Safe to call, even if coverage is not used. */ @@ -273,9 +540,7 @@ exitHpc(void) { Info *tmpModule; int i, comma; -#if DEBUG_HPC - printf("exitHpc\n"); -#endif + debugTrace(DEBUG_hpc,"exitHpc"); if (hpc_inited == 0) { return; @@ -285,7 +550,7 @@ exitHpc(void) { comma = 0; - fprintf(f,"Tix 0 ["); + fprintf(f,"Tix %" PRIuWORD64 " [", magicTixNumber); tmpModule = modules; for(;tmpModule != 0;tmpModule = tmpModule->next) { if (comma) { @@ -293,15 +558,13 @@ exitHpc(void) { } else { comma = 1; } - fprintf(f,"(\"%s\",%d)", + fprintf(f,"(\"%s\",%u)", tmpModule->modName, tmpModule->tickCount); -#if DEBUG_HPC - fprintf(stderr,"%s: %d (offset=%d)\n", + debugTrace(DEBUG_hpc,"%s: %u (offset=%u)\n", tmpModule->modName, tmpModule->tickCount, tmpModule->tickOffset); -#endif } fprintf(f,"] ["); @@ -309,8 +572,9 @@ exitHpc(void) { tmpModule = modules; for(;tmpModule != 0;tmpModule = tmpModule->next) { if (!tmpModule->tixArr) { - fprintf(stderr,"warning: module %s did not register any hpc tick data\n", - tmpModule->modName); + debugTrace(DEBUG_hpc, + "warning: module %s did not register any hpc tick data\n", + tmpModule->modName); } for(i = 0;i < tmpModule->tickCount;i++) { @@ -331,6 +595,14 @@ exitHpc(void) { fprintf(f,"]\n"); fclose(f); + + if (rixFile != NULL) { + hs_hpc_tick(RixFinishedOp,(StgThreadID)0); + fclose(rixFile); + } + if (rixCmdFile != NULL) { + fclose(rixCmdFile); + } }