X-Git-Url: http://git.megacz.com/?a=blobdiff_plain;f=rts%2FHpc.c;h=c474d8121b35c0d64949199979c872c40fae8e79;hb=a1e3066e066f0f75da361f881b2f3198e0aada5f;hp=0eeb8dd2add3e58570820d564f72774cdbeec13b;hpb=317a95824cde3c72c6234f5a28c4c53d452f964e;p=ghc-hetmet.git diff --git a/rts/Hpc.c b/rts/Hpc.c index 0eeb8dd..c474d81 100644 --- a/rts/Hpc.c +++ b/rts/Hpc.c @@ -7,17 +7,21 @@ #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? @@ -29,7 +33,7 @@ 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, @@ -60,14 +64,13 @@ int totalTixes = 0; // total number of tix boxes. static char *tixFilename; - static void failure(char *msg) { + 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"); @@ -80,8 +83,7 @@ static int init_open(char *filename) static void expect(char c) { if (tix_ch != c) { - fprintf(stderr,"Hpc: parse failed (%c,%c)\n",tix_ch,c); - exit(-1); + failure("parse error when reading .tix file"); } tix_ch = getc(tixFile); } @@ -204,9 +206,7 @@ hs_hpc_module(char *modName,int modCount,StgWord64 *tixArr) { int i; int offset = 0; -#if DEBUG_HPC - fprintf(stderr,"hs_hpc_module(%s,%d)\n",modName,modCount); -#endif + debugTrace(DEBUG_hpc,"hs_hpc_module(%s,%d)",modName,modCount); hpc_init(); @@ -249,9 +249,8 @@ hs_hpc_module(char *modName,int modCount,StgWord64 *tixArr) { lastModule->next=tmpModule; } -#if DEBUG_HPC - fprintf(stderr,"end: hs_hpc_module\n"); -#endif + debugTrace(DEBUG_hpc,"end: hs_hpc_module"); + return offset; } @@ -261,7 +260,6 @@ static void breakPointCommand(HpcRixOp rixOp, StgThreadID rixTid); static StgThreadID previousTid = 0; static StgWord64 rixBPCounter = 0; // The global event breakpoint counter static int *tixBoxBP; -static int specialOpBP = 0; static HpcRixOp rixOpBack[WOP_SIZE]; // The actual op static HpcRixOp rixTidBack[WOP_SIZE]; // Tid's before the op @@ -281,9 +279,9 @@ hs_hpc_thread_finished_event(StgTSO *current_tso) { void hs_hpc_tick(int rixOp, StgTSO *current_tso) { -#if DEBUG_HPC - fprintf(stderr,"hs_hpc_tick(%x)\n",rixOp); -#endif + + debugTrace(DEBUG_hpc,"hs_hpc_tick(%x)",rixOp); + if (rixFile == NULL) { return; } @@ -291,7 +289,8 @@ hs_hpc_tick(int rixOp, StgTSO *current_tso) { StgThreadID tid = (current_tso == 0) ? 0 : current_tso->id; // now check to see if we have met a breakpoint condition - if (rixCounter == rixBPCounter || (specialOpBP && tid != previousTid)) { + if (rixCounter == rixBPCounter + || tid != previousTid) { breakPointCommand(rixOp,tid); } else { if (rixOp >= 0) { @@ -300,10 +299,8 @@ hs_hpc_tick(int rixOp, StgTSO *current_tso) { breakPointCommand(rixOp,tid); } } else { - // Special op - if (specialOpBP) { - breakPointCommand(rixOp,tid); - } + // record the special operation + breakPointCommand(rixOp,tid); } } // update the history information. @@ -312,48 +309,62 @@ hs_hpc_tick(int rixOp, StgTSO *current_tso) { rixTidBack[rixCounter % WOP_SIZE] = tid; rixCounter++; -#if DEBUG_HPC - fprintf(stderr,"end: hs_hpc_tick\n"); -#endif + debugTrace(DEBUG_hpc, "end: hs_hpc_tick"); } static void printEvent(FILE *out,StgWord64 rixCounter,StgThreadID rixTid,HpcRixOp rixOp) { -#if DEBUG_HPC - if (out != stderr) { - printEvent(stderr,rixCounter,rixTid,rixOp); - } -#endif - fprintf(out,"%" PRIuWORD64 " %u ",rixCounter,(unsigned int)rixTid); + char prefixMsg[128]; + char suffixMsg[128]; + + sprintf(prefixMsg, + "Event %" PRIuWORD64 " %u ", + rixCounter, + (unsigned int)rixTid); + switch(rixOp) { case RixThreadFinishedOp: - fprintf(out,"ThreadFinished\n"); + sprintf(suffixMsg,"ThreadFinished"); + break; case RixRaiseOp: - fprintf(out,"Raise\n"); + sprintf(suffixMsg,"Raise"); + break; case RixFinishedOp: - fprintf(out,"Finished\n"); + sprintf(suffixMsg,"Finished"); + break; default: - fprintf(out,"%u\n",rixOp); + 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 - * x set special bp - * o unset special bp * h history * Note that you aways end up here on the first tick - * because the specialOpBP is equal 0. + * because the rixBPCounter starts equal to 0. */ int c = getc(rixCmdFile); while(c != 10 && c != -1) { @@ -365,9 +376,8 @@ breakPointCommand(HpcRixOp rixOp, StgThreadID rixTid) { tmp64 = tmp64 * 10 + (c - '0'); c = getc(rixCmdFile); } -#if DEBUG_HPC - fprintf(stderr,"setting countBP = %" PRIuWORD64 "\n",tmp64); -#endif + debugTrace(DEBUG_hpc,"setting countBP = %" PRIuWORD64,tmp64); + rixBPCounter = tmp64; break; case 's': // s2323 -- set tick box breakpoint at 2323 @@ -377,9 +387,9 @@ breakPointCommand(HpcRixOp rixOp, StgThreadID rixTid) { tmp = tmp * 10 + (c - '0'); c = getc(rixCmdFile); } -#if DEBUG_HPC - fprintf(stderr,"seting bp for tix %d\n",tmp); -#endif + + debugTrace(DEBUG_hpc,"seting bp for tix %d",tmp); + tixBoxBP[tmp] = 1; break; case 'u': // u2323 -- unset tick box breakpoint at 2323 @@ -389,25 +399,11 @@ breakPointCommand(HpcRixOp rixOp, StgThreadID rixTid) { tmp = tmp * 10 + (c - '0'); c = getc(rixCmdFile); } -#if DEBUG_HPC - fprintf(stderr,"unseting bp for tix %d\n",tmp); -#endif + + debugTrace(DEBUG_hpc,"unseting bp for tix %d",tmp); + tixBoxBP[tmp] = 0; break; - case 'x': // x -- set special bp flag -#if DEBUG_HPC - fprintf(stderr,"seting specialOpBP = 1\n"); -#endif - specialOpBP = 1; - c = getc(rixCmdFile); - break; - case 'o': // o -- clear special bp flag -#if DEBUG_HPC - fprintf(stderr,"seting specialOpBP = 0\n"); -#endif - specialOpBP = 0; - c = getc(rixCmdFile); - break; case 'h': // h -- history of the last few (WOP_SIZE) steps if (rixCounter > WOP_SIZE) { tmp64 = rixCounter - WOP_SIZE; @@ -415,15 +411,18 @@ breakPointCommand(HpcRixOp rixOp, StgThreadID rixTid) { tmp64 = 0; } for(;tmp64 < rixCounter;tmp64++) { - printEvent(rixFile,tmp64,rixTidBack[tmp64 % WOP_SIZE],rixOpBack[tmp64 % WOP_SIZE]); + printEvent(rixFile, + tmp64, + rixTidBack[tmp64 % WOP_SIZE], + rixOpBack[tmp64 % WOP_SIZE]); } fflush(rixFile); c = getc(rixCmdFile); break; default: -#if DEBUG_HPC - fprintf(stderr,"strange command from HPCRIX (%d)\n",c); -#endif + + debugTrace(DEBUG_hpc,"strange command from HPCRIX (%d)",c); + c = getc(rixCmdFile); } while (c != 10) { // the end of the line @@ -431,9 +430,9 @@ breakPointCommand(HpcRixOp rixOp, StgThreadID rixTid) { } c = getc(rixCmdFile); // the first char on the next command } -#if DEBUG_HPC - fprintf(stderr,"re entering program\n"); -#endif + + debugTrace(DEBUG_hpc,"leaving breakPointCommand"); + } /* This is called after all the modules have registered their local tixboxes, @@ -444,10 +443,8 @@ void startupHpc(void) { Info *tmpModule; char *hpcRix; - char *hpcRixCmd; -#if DEBUG_HPC - fprintf(stderr,"startupHpc\n"); -#endif + + debugTrace(DEBUG_hpc,"startupHpc"); if (hpc_inited == 0) { return; @@ -473,10 +470,33 @@ startupHpc(void) { if (hpcRix) { int comma; Info *tmpModule; + int rixFD, rixCmdFD; + int tixCount = 0; assert(hpc_inited); - rixFile = fopen(hpcRix,"w"); + 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); @@ -491,26 +511,22 @@ startupHpc(void) { fprintf(rixFile,"(\"%s\",%u)", tmpModule->modName, tmpModule->tickCount); -#if DEBUG_HPC - fprintf(stderr,"(tracer)%s: %u (offset=%u)\n", + + tixCount += tmpModule->tickCount; + + debugTrace(DEBUG_hpc,"(tracer)%s: %u (offset=%u)\n", tmpModule->modName, tmpModule->tickCount, tmpModule->tickOffset); -#endif + } fprintf(rixFile,"]\n"); fflush(rixFile); - // Now we open the command channel. - hpcRixCmd = getenv("HPCRIXCMD"); - assert(hpcRixCmd != NULL); - rixCmdFile = fopen(hpcRixCmd,"r"); - assert(rixCmdFile != NULL); - // Allocate the tixBox breakpoint array // These are set to 1 if you want to // stop at a specific breakpoint - tixBoxBP = (int *)calloc(1,sizeof(int)); + tixBoxBP = (int *)calloc(tixCount,sizeof(int)); } } @@ -524,9 +540,7 @@ exitHpc(void) { Info *tmpModule; int i, comma; -#if DEBUG_HPC - fprintf(stderr,"exitHpc\n"); -#endif + debugTrace(DEBUG_hpc,"exitHpc"); if (hpc_inited == 0) { return; @@ -547,12 +561,10 @@ exitHpc(void) { fprintf(f,"(\"%s\",%u)", tmpModule->modName, tmpModule->tickCount); -#if DEBUG_HPC - fprintf(stderr,"%s: %u (offset=%u)\n", + debugTrace(DEBUG_hpc,"%s: %u (offset=%u)\n", tmpModule->modName, tmpModule->tickCount, tmpModule->tickOffset); -#endif } fprintf(f,"] ["); @@ -560,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++) {