1 /* -----------------------------------------------------------------------------
3 * (c) The GHC Team, 2008-2009
5 * Support for fast binary event logging.
7 * ---------------------------------------------------------------------------*/
9 #include "PosixSource.h"
15 #include "Capability.h"
24 static char *event_log_filename = NULL;
26 // File for logging events
27 FILE *event_log_file = NULL;
29 #define EVENT_LOG_SIZE 2 * (1024 * 1024) // 2MB
31 static int flushCount;
33 // Struct for record keeping of buffer to store event types and events.
34 typedef struct _EventsBuf {
39 EventCapNo capno; // which capability this buffer belongs to, or -1
42 EventsBuf *capEventBuf; // one EventsBuf for each Capability
44 EventsBuf eventBuf; // an EventsBuf not associated with any Capability
46 Mutex eventBufMutex; // protected by this mutex
50 [EVENT_CREATE_THREAD] = "Create thread",
51 [EVENT_RUN_THREAD] = "Run thread",
52 [EVENT_STOP_THREAD] = "Stop thread",
53 [EVENT_THREAD_RUNNABLE] = "Thread runnable",
54 [EVENT_MIGRATE_THREAD] = "Migrate thread",
55 [EVENT_RUN_SPARK] = "Run spark",
56 [EVENT_STEAL_SPARK] = "Steal spark",
57 [EVENT_SHUTDOWN] = "Shutdown",
58 [EVENT_THREAD_WAKEUP] = "Wakeup thread",
59 [EVENT_GC_START] = "Starting GC",
60 [EVENT_GC_END] = "Finished GC",
61 [EVENT_REQUEST_SEQ_GC] = "Request sequential GC",
62 [EVENT_REQUEST_PAR_GC] = "Request parallel GC",
63 [EVENT_CREATE_SPARK_THREAD] = "Create spark thread",
64 [EVENT_LOG_MSG] = "Log message",
65 [EVENT_USER_MSG] = "User message",
66 [EVENT_STARTUP] = "Startup",
67 [EVENT_BLOCK_MARKER] = "Block marker"
72 typedef struct _EventType {
73 EventTypeNum etNum; // Event Type number.
74 nat size; // size of the payload in bytes
75 char *desc; // Description
78 EventType eventTypes[NUM_EVENT_TAGS];
80 static void initEventsBuf(EventsBuf* eb, StgWord64 size, EventCapNo capno);
81 static void resetEventsBuf(EventsBuf* eb);
82 static void printAndClearEventBuf (EventsBuf *eventsBuf);
84 static void postEventType(EventsBuf *eb, EventType *et);
86 static void postLogMsg(EventsBuf *eb, EventTypeNum type, char *msg, va_list ap);
88 static void postBlockMarker(EventsBuf *eb);
89 static void closeBlockMarker(EventsBuf *ebuf);
91 static StgBool hasRoomForEvent(EventsBuf *eb, EventTypeNum eNum);
92 static StgBool hasRoomForVariableEvent(EventsBuf *eb, nat payload_bytes);
94 static inline void postWord8(EventsBuf *eb, StgWord8 i)
99 static inline void postWord16(EventsBuf *eb, StgWord16 i)
101 postWord8(eb, (StgWord8)(i >> 8));
102 postWord8(eb, (StgWord8)i);
105 static inline void postWord32(EventsBuf *eb, StgWord32 i)
107 postWord16(eb, (StgWord16)(i >> 16));
108 postWord16(eb, (StgWord16)i);
111 static inline void postWord64(EventsBuf *eb, StgWord64 i)
113 postWord32(eb, (StgWord32)(i >> 32));
114 postWord32(eb, (StgWord32)i);
117 static inline void postBuf(EventsBuf *eb, StgWord8 *buf, nat size)
119 memcpy(eb->pos, buf, size);
123 static inline StgWord64 time_ns(void)
124 { return stat_getElapsedTime() * (1000000000LL/TICKS_PER_SECOND); }
126 static inline void postEventTypeNum(EventsBuf *eb, EventTypeNum etNum)
127 { postWord16(eb, etNum); }
129 static inline void postTimestamp(EventsBuf *eb)
130 { postWord64(eb, time_ns()); }
132 static inline void postThreadID(EventsBuf *eb, EventThreadID id)
133 { postWord32(eb,id); }
135 static inline void postCapNo(EventsBuf *eb, EventCapNo no)
136 { postWord16(eb,no); }
138 static inline void postPayloadSize(EventsBuf *eb, EventPayloadSize size)
139 { postWord16(eb,size); }
141 static inline void postEventHeader(EventsBuf *eb, EventTypeNum type)
143 postEventTypeNum(eb, type);
147 static inline void postInt8(EventsBuf *eb, StgInt8 i)
148 { postWord8(eb, (StgWord8)i); }
150 static inline void postInt16(EventsBuf *eb, StgInt16 i)
151 { postWord16(eb, (StgWord16)i); }
153 static inline void postInt32(EventsBuf *eb, StgInt32 i)
154 { postWord32(eb, (StgWord32)i); }
156 static inline void postInt64(EventsBuf *eb, StgInt64 i)
157 { postWord64(eb, (StgWord64)i); }
161 initEventLogging(void)
166 event_log_filename = stgMallocBytes(strlen(prog_name) + 10,
169 if (sizeof(EventDesc) / sizeof(char*) != NUM_EVENT_TAGS) {
170 barf("EventDesc array has the wrong number of elements");
173 sprintf(event_log_filename, "%s.eventlog", prog_name);
175 /* Open event log file for writing. */
176 if ((event_log_file = fopen(event_log_filename, "wb")) == NULL) {
177 sysErrorBelch("initEventLogging: can't open %s", event_log_filename);
178 stg_exit(EXIT_FAILURE);
182 * Allocate buffer(s) to store events.
183 * Create buffer large enough for the header begin marker, all event
184 * types, and header end marker to prevent checking if buffer has room
185 * for each of these steps, and remove the need to flush the buffer to
186 * disk during initialization.
188 * Use a single buffer to store the header with event types, then flush
189 * the buffer so all buffers are empty for writing events.
192 // XXX n_capabilities hasn't been initislised yet
193 n_caps = RtsFlags.ParFlags.nNodes;
197 capEventBuf = stgMallocBytes(n_caps * sizeof(EventsBuf),"initEventLogging");
199 for (c = 0; c < n_caps; ++c) {
200 // Init buffer for events.
201 initEventsBuf(&capEventBuf[c], EVENT_LOG_SIZE, c);
203 initEventsBuf(&eventBuf, EVENT_LOG_SIZE, (EventCapNo)(-1));
205 // Write in buffer: the header begin marker.
206 postInt32(&eventBuf, EVENT_HEADER_BEGIN);
208 // Mark beginning of event types in the header.
209 postInt32(&eventBuf, EVENT_HET_BEGIN);
210 for (t = 0; t < NUM_EVENT_TAGS; ++t) {
212 eventTypes[t].etNum = t;
213 eventTypes[t].desc = EventDesc[t];
216 case EVENT_CREATE_THREAD: // (cap, thread)
217 case EVENT_RUN_THREAD: // (cap, thread)
218 case EVENT_THREAD_RUNNABLE: // (cap, thread)
219 case EVENT_RUN_SPARK: // (cap, thread)
220 case EVENT_CREATE_SPARK_THREAD: // (cap, spark_thread)
221 eventTypes[t].size = sizeof(EventThreadID);
224 case EVENT_MIGRATE_THREAD: // (cap, thread, new_cap)
225 case EVENT_STEAL_SPARK: // (cap, thread, victim_cap)
226 case EVENT_THREAD_WAKEUP: // (cap, thread, other_cap)
228 sizeof(EventThreadID) + sizeof(EventCapNo);
231 case EVENT_STOP_THREAD: // (cap, thread, status)
233 sizeof(EventThreadID) + sizeof(StgWord16);
236 case EVENT_SHUTDOWN: // (cap)
237 case EVENT_REQUEST_SEQ_GC: // (cap)
238 case EVENT_REQUEST_PAR_GC: // (cap)
239 case EVENT_GC_START: // (cap)
240 case EVENT_GC_END: // (cap)
242 eventTypes[t].size = 0;
245 case EVENT_LOG_MSG: // (msg)
246 case EVENT_USER_MSG: // (msg)
247 eventTypes[t].size = 0xffff;
250 case EVENT_BLOCK_MARKER:
251 eventTypes[t].size = sizeof(StgWord32) + sizeof(EventTimestamp) +
256 continue; /* ignore deprecated events */
259 // Write in buffer: the start event type.
260 postEventType(&eventBuf, &eventTypes[t]);
263 // Mark end of event types in the header.
264 postInt32(&eventBuf, EVENT_HET_END);
266 // Write in buffer: the header end marker.
267 postInt32(&eventBuf, EVENT_HEADER_END);
269 // Prepare event buffer for events (data).
270 postInt32(&eventBuf, EVENT_DATA_BEGIN);
272 // Post a STARTUP event with the number of capabilities
273 postEventHeader(&eventBuf, EVENT_STARTUP);
274 postCapNo(&eventBuf, n_caps);
276 // Flush capEventBuf with header.
278 * Flush header and data begin marker to the file, thus preparing the
279 * file to have events written to it.
281 printAndClearEventBuf(&eventBuf);
283 for (c = 0; c < n_caps; ++c) {
284 postBlockMarker(&capEventBuf[c]);
289 endEventLogging(void)
293 // Flush all events remaining in the buffers.
294 for (c = 0; c < n_capabilities; ++c) {
295 printAndClearEventBuf(&capEventBuf[c]);
297 printAndClearEventBuf(&eventBuf);
298 resetEventsBuf(&eventBuf); // we don't want the block marker
300 // Mark end of events (data).
301 postEventTypeNum(&eventBuf, EVENT_DATA_END);
303 // Flush the end of data marker.
304 printAndClearEventBuf(&eventBuf);
306 if (event_log_file != NULL) {
307 fclose(event_log_file);
312 freeEventLogging(void)
316 // Free events buffer.
317 for (c = 0; c < n_capabilities; ++c) {
318 if (capEventBuf[c].begin != NULL)
319 stgFree(capEventBuf[c].begin);
321 if (capEventBuf != NULL) {
322 stgFree(capEventBuf);
324 if (event_log_filename != NULL) {
325 stgFree(event_log_filename);
330 * Post an event message to the capability's eventlog buffer.
331 * If the buffer is full, prints out the buffer and clears it.
334 postSchedEvent (Capability *cap,
341 eb = &capEventBuf[cap->no];
343 if (!hasRoomForEvent(eb, tag)) {
344 // Flush event buffer to make room for new event.
345 printAndClearEventBuf(eb);
348 postEventHeader(eb, tag);
351 case EVENT_CREATE_THREAD: // (cap, thread)
352 case EVENT_RUN_THREAD: // (cap, thread)
353 case EVENT_THREAD_RUNNABLE: // (cap, thread)
354 case EVENT_RUN_SPARK: // (cap, thread)
356 postThreadID(eb,thread);
360 case EVENT_CREATE_SPARK_THREAD: // (cap, spark_thread)
362 postThreadID(eb,other /* spark_thread */);
366 case EVENT_MIGRATE_THREAD: // (cap, thread, new_cap)
367 case EVENT_STEAL_SPARK: // (cap, thread, victim_cap)
368 case EVENT_THREAD_WAKEUP: // (cap, thread, other_cap)
370 postThreadID(eb,thread);
371 postCapNo(eb,other /* new_cap | victim_cap | other_cap */);
375 case EVENT_STOP_THREAD: // (cap, thread, status)
377 postThreadID(eb,thread);
378 postWord16(eb,other /* status */);
382 case EVENT_SHUTDOWN: // (cap)
383 case EVENT_REQUEST_SEQ_GC: // (cap)
384 case EVENT_REQUEST_PAR_GC: // (cap)
385 case EVENT_GC_START: // (cap)
386 case EVENT_GC_END: // (cap)
392 barf("postEvent: unknown event tag %d", tag);
398 void postLogMsg(EventsBuf *eb, EventTypeNum type, char *msg, va_list ap)
403 size = vsnprintf(buf,BUF,msg,ap);
409 if (!hasRoomForVariableEvent(eb, size)) {
410 // Flush event buffer to make room for new event.
411 printAndClearEventBuf(eb);
414 postEventHeader(eb, type);
415 postPayloadSize(eb, size);
416 postBuf(eb,(StgWord8*)buf,size);
419 void postMsg(char *msg, va_list ap)
421 ACQUIRE_LOCK(&eventBufMutex);
422 postLogMsg(&eventBuf, EVENT_LOG_MSG, msg, ap);
423 RELEASE_LOCK(&eventBufMutex);
426 void postCapMsg(Capability *cap, char *msg, va_list ap)
428 postLogMsg(&capEventBuf[cap->no], EVENT_LOG_MSG, msg, ap);
431 void postUserMsg(Capability *cap, char *msg)
433 postLogMsg(&capEventBuf[cap->no], EVENT_USER_MSG, msg, NULL);
436 void closeBlockMarker (EventsBuf *ebuf)
442 // (type:16, time:64, size:32, end_time:64)
444 save_pos = ebuf->pos;
445 ebuf->pos = ebuf->marker + sizeof(EventTypeNum) +
446 sizeof(EventTimestamp);
447 postWord32(ebuf, save_pos - ebuf->marker);
449 ebuf->pos = save_pos;
455 void postBlockMarker (EventsBuf *eb)
457 if (!hasRoomForEvent(eb, EVENT_BLOCK_MARKER)) {
458 printAndClearEventBuf(eb);
461 closeBlockMarker(eb);
463 eb->marker = eb->pos;
464 postEventHeader(eb, EVENT_BLOCK_MARKER);
465 postWord32(eb,0); // these get filled in later by closeBlockMarker();
467 postCapNo(eb, eb->capno);
470 void printAndClearEventBuf (EventsBuf *ebuf)
472 StgWord64 numBytes = 0, written = 0;
474 closeBlockMarker(ebuf);
476 if (ebuf->begin != NULL && ebuf->pos != ebuf->begin)
478 numBytes = ebuf->pos - ebuf->begin;
480 written = fwrite(ebuf->begin, 1, numBytes, event_log_file);
481 if (written != numBytes) {
483 "printAndClearEventLog: fwrite() failed, written=%" FMT_Word64
484 " doesn't match numBytes=%" FMT_Word64, written, numBytes);
488 resetEventsBuf(ebuf);
491 postBlockMarker(ebuf);
495 void initEventsBuf(EventsBuf* eb, StgWord64 size, EventCapNo capno)
497 eb->begin = eb->pos = stgMallocBytes(size, "initEventsBuf");
503 void resetEventsBuf(EventsBuf* eb)
509 StgBool hasRoomForEvent(EventsBuf *eb, EventTypeNum eNum)
513 size = sizeof(EventTypeNum) + sizeof(EventTimestamp) + eventTypes[eNum].size;
515 if (eb->pos + size > eb->begin + eb->size) {
516 return 0; // Not enough space.
518 return 1; // Buf has enough space for the event.
522 StgBool hasRoomForVariableEvent(EventsBuf *eb, nat payload_bytes)
526 size = sizeof(EventTypeNum) + sizeof(EventTimestamp) +
527 sizeof(EventPayloadSize) + payload_bytes;
529 if (eb->pos + size > eb->begin + eb->size) {
530 return 0; // Not enough space.
532 return 1; // Buf has enough space for the event.
536 void postEventType(EventsBuf *eb, EventType *et)
541 postInt32(eb, EVENT_ET_BEGIN);
542 postEventTypeNum(eb, et->etNum);
543 postWord16(eb, (StgWord16)et->size);
544 desclen = strlen(et->desc);
545 postWord32(eb, desclen);
546 for (d = 0; d < desclen; ++d) {
547 postInt8(eb, (StgInt8)et->desc[d]);
549 postWord32(eb, 0); // no extensions yet
550 postInt32(eb, EVENT_ET_END);