module GHC.RTS.Events.Binary
(
getHeader
, getEvent
, standardParsers
, ghc6Parsers
, ghc7Parsers
, mercuryParsers
, perfParsers
, heapProfParsers
, pre77StopParsers
, ghc782StopParser
, post782StopParser
, parRTSParsers
, putEventLog
, putHeader
, putEvent
, nEVENT_PERF_NAME
, nEVENT_PERF_COUNTER
, nEVENT_PERF_TRACEPOINT
) where
import Control.Exception (assert)
import Control.Monad
import Data.Maybe
import Prelude hiding (gcd, rem, id)
import Data.Array
import Data.Binary
import Data.Binary.Put
import qualified Data.Binary.Get as G
import qualified Data.ByteString as B
import qualified Data.Text as T
import qualified Data.Text.Encoding as TE
import qualified Data.Text.Lazy as TL
import qualified Data.Text.Lazy.Encoding as TLE
import qualified Data.Vector.Unboxed as VU
import GHC.RTS.EventTypes
import GHC.RTS.EventParserUtils
#define EVENTLOG_CONSTANTS_ONLY
#include "EventLogFormat.h"
getEventType :: Get EventType
getEventType = do
etNum <- get
size <- get :: Get EventTypeSize
let etSize = if size == 0xffff then Nothing else Just size
etDescLen <- get :: Get EventTypeDescLen
etDesc <- getEtDesc (fromIntegral etDescLen)
etExtraLen <- get :: Get Word32
G.skip (fromIntegral etExtraLen)
ete <- get :: Get Marker
when (ete /= EVENT_ET_END) $
fail "Event Type end marker not found."
return (EventType etNum etDesc etSize)
where
getEtDesc :: Int -> Get [Char]
getEtDesc s = replicateM s (get :: Get Char)
getHeader :: Get Header
getHeader = do
hdrb <- get :: Get Marker
when (hdrb /= EVENT_HEADER_BEGIN) $
fail "Header begin marker not found"
hetm <- get :: Get Marker
when (hetm /= EVENT_HET_BEGIN) $
fail "Header Event Type begin marker not found"
ets <- getEventTypes
emark <- get :: Get Marker
when (emark /= EVENT_HEADER_END) $
fail "Header end marker not found"
db <- get :: Get Marker
when (db /= EVENT_DATA_BEGIN) $
fail "My Data begin marker not found"
return $ Header ets
where
getEventTypes :: Get [EventType]
getEventTypes = do
m <- get :: Get Marker
case m of
EVENT_ET_BEGIN -> do
et <- getEventType
nextET <- getEventTypes
return (et : nextET)
EVENT_HET_END ->
return []
_ ->
fail "Malformed list of Event Types in header"
getEvent :: EventParsers -> Get (Maybe Event)
getEvent (EventParsers parsers) = do
etRef <- get :: Get EventTypeNum
if etRef == EVENT_DATA_END
then return Nothing
else do !evTime <- get
evSpec <- parsers ! fromIntegral etRef
return $ Just Event { evCap = undefined, .. }
standardParsers :: [EventParser EventInfo]
standardParsers = [
(FixedSizeParser EVENT_STARTUP sz_cap (do
c <- get :: Get CapNo
return Startup{ n_caps = fromIntegral c }
)),
(FixedSizeParser EVENT_BLOCK_MARKER (sz_block_size + sz_time + sz_cap) (do
block_size <- get :: Get BlockSize
end_time <- get :: Get Timestamp
c <- get :: Get CapNo
return EventBlock { end_time = end_time,
cap = fromIntegral c,
block_size = ((fromIntegral block_size)
(fromIntegral sz_block_event))
}
)),
(simpleEvent EVENT_SHUTDOWN Shutdown),
(simpleEvent EVENT_REQUEST_SEQ_GC RequestSeqGC),
(simpleEvent EVENT_REQUEST_PAR_GC RequestParGC),
(simpleEvent EVENT_GC_START StartGC),
(simpleEvent EVENT_GC_WORK GCWork),
(simpleEvent EVENT_GC_IDLE GCIdle),
(simpleEvent EVENT_GC_DONE GCDone),
(simpleEvent EVENT_GC_END EndGC),
(simpleEvent EVENT_GC_GLOBAL_SYNC GlobalSyncGC),
(FixedSizeParser EVENT_GC_STATS_GHC (sz_capset + 2 + 5*8 + 4) (do
heapCapset <- get
gen <- get :: Get Word16
copied <- get :: Get Word64
slop <- get :: Get Word64
frag <- get :: Get Word64
parNThreads <- get :: Get Word32
parMaxCopied <- get :: Get Word64
parTotCopied <- get :: Get Word64
return GCStatsGHC{ gen = fromIntegral gen
, parNThreads = fromIntegral parNThreads
, ..}
)),
(FixedSizeParser EVENT_HEAP_ALLOCATED (sz_capset + 8) (do
heapCapset <- get
allocBytes <- get
return HeapAllocated{..}
)),
(FixedSizeParser EVENT_HEAP_SIZE (sz_capset + 8) (do
heapCapset <- get
sizeBytes <- get
return HeapSize{..}
)),
(FixedSizeParser EVENT_HEAP_LIVE (sz_capset + 8) (do
heapCapset <- get
liveBytes <- get
return HeapLive{..}
)),
(FixedSizeParser EVENT_HEAP_INFO_GHC (sz_capset + 2 + 4*8) (do
heapCapset <- get
gens <- get :: Get Word16
maxHeapSize <- get :: Get Word64
allocAreaSize <- get :: Get Word64
mblockSize <- get :: Get Word64
blockSize <- get :: Get Word64
return HeapInfoGHC{gens = fromIntegral gens, ..}
)),
(FixedSizeParser EVENT_CAP_CREATE (sz_cap) (do
cap <- get :: Get CapNo
return CapCreate{cap = fromIntegral cap}
)),
(FixedSizeParser EVENT_CAP_DELETE (sz_cap) (do
cap <- get :: Get CapNo
return CapDelete{cap = fromIntegral cap}
)),
(FixedSizeParser EVENT_CAP_DISABLE (sz_cap) (do
cap <- get :: Get CapNo
return CapDisable{cap = fromIntegral cap}
)),
(FixedSizeParser EVENT_CAP_ENABLE (sz_cap) (do
cap <- get :: Get CapNo
return CapEnable{cap = fromIntegral cap}
)),
(FixedSizeParser EVENT_CAPSET_CREATE (sz_capset + sz_capset_type) (do
cs <- get
ct <- fmap mkCapsetType get
return CapsetCreate{capset=cs,capsetType=ct}
)),
(FixedSizeParser EVENT_CAPSET_DELETE sz_capset (do
cs <- get
return CapsetDelete{capset=cs}
)),
(FixedSizeParser EVENT_CAPSET_ASSIGN_CAP (sz_capset + sz_cap) (do
cs <- get
cp <- get :: Get CapNo
return CapsetAssignCap{capset=cs,cap=fromIntegral cp}
)),
(FixedSizeParser EVENT_CAPSET_REMOVE_CAP (sz_capset + sz_cap) (do
cs <- get
cp <- get :: Get CapNo
return CapsetRemoveCap{capset=cs,cap=fromIntegral cp}
)),
(FixedSizeParser EVENT_OSPROCESS_PID (sz_capset + sz_pid) (do
cs <- get
pd <- get
return OsProcessPid{capset=cs,pid=pd}
)),
(FixedSizeParser EVENT_OSPROCESS_PPID (sz_capset + sz_pid) (do
cs <- get
pd <- get
return OsProcessParentPid{capset=cs,ppid=pd}
)),
(FixedSizeParser EVENT_WALL_CLOCK_TIME (sz_capset + 8 + 4) (do
cs <- get
s <- get
ns <- get
return WallClockTime{capset=cs,sec=s,nsec=ns}
)),
(VariableSizeParser EVENT_LOG_MSG (do
num <- get :: Get Word16
string <- getString num
return Message{ msg = string }
)),
(VariableSizeParser EVENT_USER_MSG (do
num <- get :: Get Word16
string <- getString num
return UserMessage{ msg = string }
)),
(VariableSizeParser EVENT_USER_MARKER (do
num <- get :: Get Word16
string <- getString num
return UserMarker{ markername = string }
)),
(VariableSizeParser EVENT_PROGRAM_ARGS (do
num <- get :: Get Word16
cs <- get
string <- getString (num sz_capset)
return ProgramArgs{ capset = cs
, args = splitNull string }
)),
(VariableSizeParser EVENT_PROGRAM_ENV (do
num <- get :: Get Word16
cs <- get
string <- getString (num sz_capset)
return ProgramEnv{ capset = cs
, env = splitNull string }
)),
(VariableSizeParser EVENT_RTS_IDENTIFIER (do
num <- get :: Get Word16
cs <- get
string <- getString (num sz_capset)
return RtsIdentifier{ capset = cs
, rtsident = string }
)),
(VariableSizeParser EVENT_INTERN_STRING (do
num <- get :: Get Word16
string <- getString (num sz_string_id)
sId <- get :: Get StringId
return (InternString string sId)
)),
(VariableSizeParser EVENT_THREAD_LABEL (do
num <- get :: Get Word16
tid <- get
str <- getString (num sz_tid)
return ThreadLabel{ thread = tid
, threadlabel = str }
))
]
ghc7Parsers :: [EventParser EventInfo]
ghc7Parsers = [
(FixedSizeParser EVENT_CREATE_THREAD sz_tid (do
t <- get
return CreateThread{thread=t}
)),
(FixedSizeParser EVENT_RUN_THREAD sz_tid (do
t <- get
return RunThread{thread=t}
)),
(FixedSizeParser EVENT_THREAD_RUNNABLE sz_tid (do
t <- get
return ThreadRunnable{thread=t}
)),
(FixedSizeParser EVENT_MIGRATE_THREAD (sz_tid + sz_cap) (do
t <- get
nc <- get :: Get CapNo
return MigrateThread{thread=t,newCap=fromIntegral nc}
)),
(FixedSizeParser EVENT_RUN_SPARK sz_tid (do
_ <- get :: Get ThreadId
return SparkRun
)),
(FixedSizeParser EVENT_STEAL_SPARK (sz_tid + sz_cap) (do
_ <- get :: Get ThreadId
vc <- get :: Get CapNo
return SparkSteal{victimCap=fromIntegral vc}
)),
(FixedSizeParser EVENT_CREATE_SPARK_THREAD sz_tid (do
st <- get :: Get ThreadId
return CreateSparkThread{sparkThread=st}
)),
(FixedSizeParser EVENT_SPARK_COUNTERS (7*8) (do
crt <- get :: Get Word64
dud <- get :: Get Word64
ovf <- get :: Get Word64
cnv <- get :: Get Word64
gcd <- get :: Get Word64
fiz <- get :: Get Word64
rem <- get :: Get Word64
return SparkCounters{sparksCreated = crt, sparksDud = dud,
sparksOverflowed = ovf, sparksConverted = cnv,
sparksFizzled = fiz, sparksGCd = gcd,
sparksRemaining = rem}
)),
(simpleEvent EVENT_SPARK_CREATE SparkCreate),
(simpleEvent EVENT_SPARK_DUD SparkDud),
(simpleEvent EVENT_SPARK_OVERFLOW SparkOverflow),
(simpleEvent EVENT_SPARK_RUN SparkRun),
(FixedSizeParser EVENT_SPARK_STEAL sz_cap (do
vc <- get :: Get CapNo
return SparkSteal{victimCap=fromIntegral vc}
)),
(simpleEvent EVENT_SPARK_FIZZLE SparkFizzle),
(simpleEvent EVENT_SPARK_GC SparkGC),
(FixedSizeParser EVENT_TASK_CREATE (sz_taskid + sz_cap + sz_kernel_tid) (do
taskId <- get :: Get TaskId
cap <- get :: Get CapNo
tid <- get :: Get KernelThreadId
return TaskCreate{ taskId, cap = fromIntegral cap, tid }
)),
(FixedSizeParser EVENT_TASK_MIGRATE (sz_taskid + sz_cap*2) (do
taskId <- get :: Get TaskId
cap <- get :: Get CapNo
new_cap <- get :: Get CapNo
return TaskMigrate{ taskId, cap = fromIntegral cap
, new_cap = fromIntegral new_cap
}
)),
(FixedSizeParser EVENT_TASK_DELETE (sz_taskid) (do
taskId <- get :: Get TaskId
return TaskDelete{ taskId }
)),
(FixedSizeParser EVENT_THREAD_WAKEUP (sz_tid + sz_cap) (do
t <- get
oc <- get :: Get CapNo
return WakeupThread{thread=t,otherCap=fromIntegral oc}
))
]
ghc782StopParser :: EventParser EventInfo
ghc782StopParser =
(FixedSizeParser EVENT_STOP_THREAD (sz_tid + sz_th_stop_status + sz_tid) (do
t <- get
s <- get :: Get RawThreadStopStatus
i <- get :: Get ThreadId
return StopThread{thread = t,
status = case () of
_ | s > maxThreadStopStatus782
-> NoStatus
| s == 9
-> BlockedOnBlackHoleOwnedBy i
| otherwise
-> mkStopStatus782 s}
))
pre77StopParsers :: [EventParser EventInfo]
pre77StopParsers = [
(FixedSizeParser EVENT_STOP_THREAD (sz_tid + sz_th_stop_status) (do
t <- get
s <- get :: Get RawThreadStopStatus
return StopThread{thread=t, status = if s > maxThreadStopStatusPre77
then NoStatus
else mkStopStatus s}
)),
(FixedSizeParser EVENT_STOP_THREAD (sz_tid + sz_th_stop_status + sz_tid)
(do
t <- get
s <- get :: Get RawThreadStopStatus
i <- get :: Get ThreadId
return StopThread{thread = t,
status = case () of
_ | s > maxThreadStopStatusPre77
-> NoStatus
| s == 8
-> BlockedOnBlackHoleOwnedBy i
| otherwise
-> mkStopStatus s}
))
]
post782StopParser :: EventParser EventInfo
post782StopParser =
(FixedSizeParser EVENT_STOP_THREAD (sz_tid + sz_th_stop_status + sz_tid)
(do
t <- get
s <- get :: Get RawThreadStopStatus
i <- get :: Get ThreadId
return StopThread{thread = t,
status = case () of
_ | s > maxThreadStopStatus
-> NoStatus
| s == 8
-> BlockedOnBlackHoleOwnedBy i
| otherwise
-> mkStopStatus s}
))
ghc6Parsers :: [EventParser EventInfo]
ghc6Parsers = [
(FixedSizeParser EVENT_STARTUP 0 (do
c <- get :: Get CapNo
return Startup{ n_caps = fromIntegral c }
)),
(FixedSizeParser EVENT_CREATE_THREAD sz_old_tid (do
t <- get
return CreateThread{thread=t}
)),
(FixedSizeParser EVENT_RUN_THREAD sz_old_tid (do
t <- get
return RunThread{thread=t}
)),
(FixedSizeParser EVENT_STOP_THREAD (sz_old_tid + 2) (do
t <- get
s <- get :: Get RawThreadStopStatus
return StopThread{thread=t, status = if s > maxThreadStopStatusPre77
then NoStatus
else mkStopStatus s}
)),
(FixedSizeParser EVENT_THREAD_RUNNABLE sz_old_tid (do
t <- get
return ThreadRunnable{thread=t}
)),
(FixedSizeParser EVENT_MIGRATE_THREAD (sz_old_tid + sz_cap) (do
t <- get
nc <- get :: Get CapNo
return MigrateThread{thread=t,newCap=fromIntegral nc}
)),
(FixedSizeParser EVENT_RUN_SPARK sz_old_tid (do
_ <- get :: Get ThreadId
return SparkRun
)),
(FixedSizeParser EVENT_STEAL_SPARK (sz_old_tid + sz_cap) (do
_ <- get :: Get ThreadId
vc <- get :: Get CapNo
return SparkSteal{victimCap=fromIntegral vc}
)),
(FixedSizeParser EVENT_CREATE_SPARK_THREAD sz_old_tid (do
st <- get :: Get ThreadId
return CreateSparkThread{sparkThread=st}
)),
(FixedSizeParser EVENT_THREAD_WAKEUP (sz_old_tid + sz_cap) (do
t <- get
oc <- get :: Get CapNo
return WakeupThread{thread=t,otherCap=fromIntegral oc}
))
]
parRTSParsers :: EventTypeSize -> [EventParser EventInfo]
parRTSParsers sz_tid' = [
(VariableSizeParser EVENT_VERSION (do
num <- get :: Get Word16
string <- getString num
return Version{ version = string }
)),
(VariableSizeParser EVENT_PROGRAM_INVOCATION (do
num <- get :: Get Word16
string <- getString num
return ProgramInvocation{ commandline = string }
)),
(simpleEvent EVENT_EDEN_START_RECEIVE EdenStartReceive),
(simpleEvent EVENT_EDEN_END_RECEIVE EdenEndReceive),
(FixedSizeParser EVENT_CREATE_PROCESS sz_procid
(do p <- get
return CreateProcess{ process = p })
),
(FixedSizeParser EVENT_KILL_PROCESS sz_procid
(do p <- get
return KillProcess{ process = p })
),
(FixedSizeParser EVENT_ASSIGN_THREAD_TO_PROCESS (sz_tid' + sz_procid)
(do t <- get
p <- get
return AssignThreadToProcess { thread = t, process = p })
),
(FixedSizeParser EVENT_CREATE_MACHINE (sz_mid + sz_realtime)
(do m <- get
t <- get
return CreateMachine { machine = m, realtime = t })
),
(FixedSizeParser EVENT_KILL_MACHINE sz_mid
(do m <- get :: Get MachineId
return KillMachine { machine = m })
),
(FixedSizeParser EVENT_SEND_MESSAGE
(sz_msgtag + 2*sz_procid + 2*sz_tid' + sz_mid)
(do tag <- get :: Get RawMsgTag
sP <- get :: Get ProcessId
sT <- get :: Get ThreadId
rM <- get :: Get MachineId
rP <- get :: Get ProcessId
rIP <- get :: Get PortId
return SendMessage { mesTag = toMsgTag tag,
senderProcess = sP,
senderThread = sT,
receiverMachine = rM,
receiverProcess = rP,
receiverInport = rIP
})
),
(FixedSizeParser EVENT_RECEIVE_MESSAGE
(sz_msgtag + 2*sz_procid + 2*sz_tid' + sz_mid + sz_mes)
(do tag <- get :: Get Word8
rP <- get :: Get ProcessId
rIP <- get :: Get PortId
sM <- get :: Get MachineId
sP <- get :: Get ProcessId
sT <- get :: Get ThreadId
mS <- get :: Get MessageSize
return ReceiveMessage { mesTag = toMsgTag tag,
receiverProcess = rP,
receiverInport = rIP,
senderMachine = sM,
senderProcess = sP,
senderThread= sT,
messageSize = mS
})
),
(FixedSizeParser EVENT_SEND_RECEIVE_LOCAL_MESSAGE
(sz_msgtag + 2*sz_procid + 2*sz_tid')
(do tag <- get :: Get Word8
sP <- get :: Get ProcessId
sT <- get :: Get ThreadId
rP <- get :: Get ProcessId
rIP <- get :: Get PortId
return SendReceiveLocalMessage { mesTag = toMsgTag tag,
senderProcess = sP,
senderThread = sT,
receiverProcess = rP,
receiverInport = rIP
})
)]
mercuryParsers :: [EventParser EventInfo]
mercuryParsers = [
(FixedSizeParser EVENT_MER_START_PAR_CONJUNCTION
(sz_par_conj_dyn_id + sz_par_conj_static_id)
(do dyn_id <- get
static_id <- get
return (MerStartParConjunction dyn_id static_id))
),
(FixedSizeParser EVENT_MER_STOP_PAR_CONJUNCTION sz_par_conj_dyn_id
(do dyn_id <- get
return (MerEndParConjunction dyn_id))
),
(FixedSizeParser EVENT_MER_STOP_PAR_CONJUNCT sz_par_conj_dyn_id
(do dyn_id <- get
return (MerEndParConjunct dyn_id))
),
(FixedSizeParser EVENT_MER_CREATE_SPARK (sz_par_conj_dyn_id + sz_spark_id)
(do dyn_id <- get
spark_id <- get
return (MerCreateSpark dyn_id spark_id))
),
(FixedSizeParser EVENT_MER_FUT_CREATE (sz_future_id + sz_string_id)
(do future_id <- get
name_id <- get
return (MerFutureCreate future_id name_id))
),
(FixedSizeParser EVENT_MER_FUT_WAIT_NOSUSPEND (sz_future_id)
(do future_id <- get
return (MerFutureWaitNosuspend future_id))
),
(FixedSizeParser EVENT_MER_FUT_WAIT_SUSPENDED (sz_future_id)
(do future_id <- get
return (MerFutureWaitSuspended future_id))
),
(FixedSizeParser EVENT_MER_FUT_SIGNAL (sz_future_id)
(do future_id <- get
return (MerFutureSignal future_id))
),
(simpleEvent EVENT_MER_LOOKING_FOR_GLOBAL_CONTEXT MerLookingForGlobalThread),
(simpleEvent EVENT_MER_WORK_STEALING MerWorkStealing),
(simpleEvent EVENT_MER_LOOKING_FOR_LOCAL_SPARK MerLookingForLocalSpark),
(FixedSizeParser EVENT_MER_RELEASE_CONTEXT sz_tid
(do thread_id <- get
return (MerReleaseThread thread_id))
),
(simpleEvent EVENT_MER_ENGINE_SLEEPING MerCapSleeping),
(simpleEvent EVENT_MER_CALLING_MAIN MerCallingMain)
]
perfParsers :: [EventParser EventInfo]
perfParsers = [
(VariableSizeParser EVENT_PERF_NAME (do
num <- get :: Get Word16
perfNum <- get
name <- getString (num sz_perf_num)
return PerfName{perfNum, name}
)),
(FixedSizeParser EVENT_PERF_COUNTER (sz_perf_num + sz_kernel_tid + 8) (do
perfNum <- get
tid <- get
period <- get
return PerfCounter{perfNum, tid, period}
)),
(FixedSizeParser EVENT_PERF_TRACEPOINT (sz_perf_num + sz_kernel_tid) (do
perfNum <- get
tid <- get
return PerfTracepoint{perfNum, tid}
))
]
heapProfParsers :: [EventParser EventInfo]
heapProfParsers =
[ VariableSizeParser EVENT_HEAP_PROF_BEGIN $ do
payloadLen <- get :: Get Word16
heapProfId <- get
heapProfSamplingPeriod <- get
heapProfBreakdown <- get
heapProfModuleFilter <- getText
heapProfClosureDescrFilter <- getText
heapProfTypeDescrFilter <- getText
heapProfCostCentreFilter <- getText
heapProfCostCentreStackFilter <- getText
heapProfRetainerFilter <- getText
heapProfBiographyFilter <- getText
assert
(fromIntegral payloadLen == sum
[ 1
, 8
, 4
, textByteLen heapProfModuleFilter
, textByteLen heapProfClosureDescrFilter
, textByteLen heapProfTypeDescrFilter
, textByteLen heapProfCostCentreFilter
, textByteLen heapProfCostCentreStackFilter
, textByteLen heapProfRetainerFilter
, textByteLen heapProfBiographyFilter
])
(return ())
return $! HeapProfBegin {..}
, VariableSizeParser EVENT_HEAP_PROF_COST_CENTRE $ do
payloadLen <- get :: Get Word16
heapProfCostCentreId <- get
heapProfLabel <- getText
heapProfModule <- getText
heapProfSrcLoc <- getText
heapProfFlags <- get
assert
(fromIntegral payloadLen == sum
[ 4
, textByteLen heapProfLabel
, textByteLen heapProfModule
, textByteLen heapProfSrcLoc
, 1
])
(return ())
return $! HeapProfCostCentre {..}
, FixedSizeParser EVENT_HEAP_PROF_SAMPLE_BEGIN 8 $ do
heapProfSampleEra <- get
return $! HeapProfSampleBegin {..}
, VariableSizeParser EVENT_HEAP_PROF_SAMPLE_COST_CENTRE $ do
payloadLen <- get :: Get Word16
heapProfId <- get
heapProfResidency <- get
heapProfStackDepth <- get
heapProfStack <- VU.replicateM (fromIntegral heapProfStackDepth) get
assert
((fromIntegral payloadLen :: Int) == sum
[ 1
, 8
, 1
, fromIntegral heapProfStackDepth * 4
])
(return ())
return $! HeapProfSampleCostCentre {..}
, VariableSizeParser EVENT_HEAP_PROF_SAMPLE_STRING $ do
payloadLen <- get :: Get Word16
heapProfId <- get
heapProfResidency <- get
heapProfLabel <- getText
assert
(fromIntegral payloadLen == sum
[ 1
, 8
, textByteLen heapProfLabel
])
(return ())
return $! HeapProfSampleString {..}
]
textByteLen :: T.Text -> Int
textByteLen = (+1) . B.length . TE.encodeUtf8
putE :: Binary a => a -> PutM ()
putE = put
putType :: EventTypeNum -> PutM ()
putType = putE
putCap :: Int -> PutM ()
putCap c = putE (fromIntegral c :: CapNo)
putMarker :: Word32 -> PutM ()
putMarker = putE
putEStr :: String -> PutM ()
putEStr = mapM_ putE
putEventLog :: EventLog -> PutM ()
putEventLog (EventLog hdr es) = do
putHeader hdr
putData es
putHeader :: Header -> PutM ()
putHeader (Header ets) = do
putMarker EVENT_HEADER_BEGIN
putMarker EVENT_HET_BEGIN
mapM_ putEventType ets
putMarker EVENT_HET_END
putMarker EVENT_HEADER_END
where
putEventType (EventType n d msz) = do
putMarker EVENT_ET_BEGIN
putType n
putE $ fromMaybe 0xffff msz
putE (fromIntegral $ length d :: EventTypeDescLen)
mapM_ put d
putE (0 :: Word32)
putMarker EVENT_ET_END
putData :: Data -> PutM ()
putData (Data es) = do
putMarker EVENT_DATA_BEGIN
mapM_ putEvent es
putType EVENT_DATA_END
eventTypeNum :: EventInfo -> EventTypeNum
eventTypeNum e = case e of
CreateThread {} -> EVENT_CREATE_THREAD
RunThread {} -> EVENT_RUN_THREAD
StopThread {} -> EVENT_STOP_THREAD
ThreadRunnable {} -> EVENT_THREAD_RUNNABLE
MigrateThread {} -> EVENT_MIGRATE_THREAD
Shutdown {} -> EVENT_SHUTDOWN
WakeupThread {} -> EVENT_THREAD_WAKEUP
ThreadLabel {} -> EVENT_THREAD_LABEL
StartGC {} -> EVENT_GC_START
EndGC {} -> EVENT_GC_END
GlobalSyncGC {} -> EVENT_GC_GLOBAL_SYNC
RequestSeqGC {} -> EVENT_REQUEST_SEQ_GC
RequestParGC {} -> EVENT_REQUEST_PAR_GC
CreateSparkThread {} -> EVENT_CREATE_SPARK_THREAD
SparkCounters {} -> EVENT_SPARK_COUNTERS
SparkCreate {} -> EVENT_SPARK_CREATE
SparkDud {} -> EVENT_SPARK_DUD
SparkOverflow {} -> EVENT_SPARK_OVERFLOW
SparkRun {} -> EVENT_SPARK_RUN
SparkSteal {} -> EVENT_SPARK_STEAL
SparkFizzle {} -> EVENT_SPARK_FIZZLE
SparkGC {} -> EVENT_SPARK_GC
TaskCreate {} -> EVENT_TASK_CREATE
TaskMigrate {} -> EVENT_TASK_MIGRATE
TaskDelete {} -> EVENT_TASK_DELETE
Message {} -> EVENT_LOG_MSG
Startup {} -> EVENT_STARTUP
EventBlock {} -> EVENT_BLOCK_MARKER
UserMessage {} -> EVENT_USER_MSG
UserMarker {} -> EVENT_USER_MARKER
GCIdle {} -> EVENT_GC_IDLE
GCWork {} -> EVENT_GC_WORK
GCDone {} -> EVENT_GC_DONE
GCStatsGHC{} -> EVENT_GC_STATS_GHC
HeapAllocated{} -> EVENT_HEAP_ALLOCATED
HeapSize{} -> EVENT_HEAP_SIZE
HeapLive{} -> EVENT_HEAP_LIVE
HeapInfoGHC{} -> EVENT_HEAP_INFO_GHC
CapCreate{} -> EVENT_CAP_CREATE
CapDelete{} -> EVENT_CAP_DELETE
CapDisable{} -> EVENT_CAP_DISABLE
CapEnable{} -> EVENT_CAP_ENABLE
CapsetCreate {} -> EVENT_CAPSET_CREATE
CapsetDelete {} -> EVENT_CAPSET_DELETE
CapsetAssignCap {} -> EVENT_CAPSET_ASSIGN_CAP
CapsetRemoveCap {} -> EVENT_CAPSET_REMOVE_CAP
RtsIdentifier {} -> EVENT_RTS_IDENTIFIER
ProgramArgs {} -> EVENT_PROGRAM_ARGS
ProgramEnv {} -> EVENT_PROGRAM_ENV
OsProcessPid {} -> EVENT_OSPROCESS_PID
OsProcessParentPid{} -> EVENT_OSPROCESS_PPID
WallClockTime{} -> EVENT_WALL_CLOCK_TIME
UnknownEvent {} -> error "eventTypeNum UnknownEvent"
InternString {} -> EVENT_INTERN_STRING
Version {} -> EVENT_VERSION
ProgramInvocation {} -> EVENT_PROGRAM_INVOCATION
EdenStartReceive {} -> EVENT_EDEN_START_RECEIVE
EdenEndReceive {} -> EVENT_EDEN_END_RECEIVE
CreateProcess {} -> EVENT_CREATE_PROCESS
KillProcess {} -> EVENT_KILL_PROCESS
AssignThreadToProcess {} -> EVENT_ASSIGN_THREAD_TO_PROCESS
CreateMachine {} -> EVENT_CREATE_MACHINE
KillMachine {} -> EVENT_KILL_MACHINE
SendMessage {} -> EVENT_SEND_MESSAGE
ReceiveMessage {} -> EVENT_RECEIVE_MESSAGE
SendReceiveLocalMessage {} -> EVENT_SEND_RECEIVE_LOCAL_MESSAGE
MerStartParConjunction {} -> EVENT_MER_START_PAR_CONJUNCTION
MerEndParConjunction _ -> EVENT_MER_STOP_PAR_CONJUNCTION
MerEndParConjunct _ -> EVENT_MER_STOP_PAR_CONJUNCT
MerCreateSpark {} -> EVENT_MER_CREATE_SPARK
MerFutureCreate {} -> EVENT_MER_FUT_CREATE
MerFutureWaitNosuspend _ -> EVENT_MER_FUT_WAIT_NOSUSPEND
MerFutureWaitSuspended _ -> EVENT_MER_FUT_WAIT_SUSPENDED
MerFutureSignal _ -> EVENT_MER_FUT_SIGNAL
MerLookingForGlobalThread -> EVENT_MER_LOOKING_FOR_GLOBAL_CONTEXT
MerWorkStealing -> EVENT_MER_WORK_STEALING
MerLookingForLocalSpark -> EVENT_MER_LOOKING_FOR_LOCAL_SPARK
MerReleaseThread _ -> EVENT_MER_RELEASE_CONTEXT
MerCapSleeping -> EVENT_MER_ENGINE_SLEEPING
MerCallingMain -> EVENT_MER_CALLING_MAIN
PerfName {} -> nEVENT_PERF_NAME
PerfCounter {} -> nEVENT_PERF_COUNTER
PerfTracepoint {} -> nEVENT_PERF_TRACEPOINT
HeapProfBegin {} -> EVENT_HEAP_PROF_BEGIN
HeapProfCostCentre {} -> EVENT_HEAP_PROF_COST_CENTRE
HeapProfSampleBegin {} -> EVENT_HEAP_PROF_SAMPLE_BEGIN
HeapProfSampleCostCentre {} -> EVENT_HEAP_PROF_SAMPLE_COST_CENTRE
HeapProfSampleString {} -> EVENT_HEAP_PROF_SAMPLE_STRING
nEVENT_PERF_NAME, nEVENT_PERF_COUNTER, nEVENT_PERF_TRACEPOINT :: EventTypeNum
nEVENT_PERF_NAME = EVENT_PERF_NAME
nEVENT_PERF_COUNTER = EVENT_PERF_COUNTER
nEVENT_PERF_TRACEPOINT = EVENT_PERF_TRACEPOINT
putEvent :: Event -> PutM ()
putEvent Event {..} = do
putType (eventTypeNum evSpec)
put evTime
putEventSpec evSpec
putEventSpec :: EventInfo -> PutM ()
putEventSpec (Startup caps) = do
putCap (fromIntegral caps)
putEventSpec (EventBlock end cap sz) = do
putE (fromIntegral (sz+24) :: BlockSize)
putE end
putE (fromIntegral cap :: CapNo)
putEventSpec (CreateThread t) =
putE t
putEventSpec (RunThread t) =
putE t
putEventSpec (StopThread t s) = do
putE t
putE $ case s of
NoStatus -> 0 :: Word16
HeapOverflow -> 1
StackOverflow -> 2
ThreadYielding -> 3
ThreadBlocked -> 4
ThreadFinished -> 5
ForeignCall -> 6
BlockedOnMVar -> 7
BlockedOnMVarRead -> 20
BlockedOnBlackHole -> 8
BlockedOnBlackHoleOwnedBy _ -> 8
BlockedOnRead -> 9
BlockedOnWrite -> 10
BlockedOnDelay -> 11
BlockedOnSTM -> 12
BlockedOnDoProc -> 13
BlockedOnCCall -> 14
BlockedOnCCall_NoUnblockExc -> 15
BlockedOnMsgThrowTo -> 16
ThreadMigrating -> 17
BlockedOnMsgGlobalise -> 18
putE $ case s of
BlockedOnBlackHoleOwnedBy i -> i
_ -> 0
putEventSpec (ThreadRunnable t) =
putE t
putEventSpec (MigrateThread t c) = do
putE t
putCap c
putEventSpec (CreateSparkThread t) =
putE t
putEventSpec (SparkCounters crt dud ovf cnv fiz gcd rem) = do
putE crt
putE dud
putE ovf
putE cnv
putE gcd
putE fiz
putE rem
putEventSpec SparkCreate =
return ()
putEventSpec SparkDud =
return ()
putEventSpec SparkOverflow =
return ()
putEventSpec SparkRun =
return ()
putEventSpec (SparkSteal c) =
putCap c
putEventSpec SparkFizzle =
return ()
putEventSpec SparkGC =
return ()
putEventSpec (WakeupThread t c) = do
putE t
putCap c
putEventSpec (ThreadLabel t l) = do
putE (fromIntegral (length l) + sz_tid :: Word16)
putE t
putEStr l
putEventSpec Shutdown =
return ()
putEventSpec RequestSeqGC =
return ()
putEventSpec RequestParGC =
return ()
putEventSpec StartGC =
return ()
putEventSpec GCWork =
return ()
putEventSpec GCIdle =
return ()
putEventSpec GCDone =
return ()
putEventSpec EndGC =
return ()
putEventSpec GlobalSyncGC =
return ()
putEventSpec (TaskCreate taskId cap tid) = do
putE taskId
putCap cap
putE tid
putEventSpec (TaskMigrate taskId cap new_cap) = do
putE taskId
putCap cap
putCap new_cap
putEventSpec (TaskDelete taskId) =
putE taskId
putEventSpec GCStatsGHC{..} = do
putE heapCapset
putE (fromIntegral gen :: Word16)
putE copied
putE slop
putE frag
putE (fromIntegral parNThreads :: Word32)
putE parMaxCopied
putE parTotCopied
putEventSpec HeapAllocated{..} = do
putE heapCapset
putE allocBytes
putEventSpec HeapSize{..} = do
putE heapCapset
putE sizeBytes
putEventSpec HeapLive{..} = do
putE heapCapset
putE liveBytes
putEventSpec HeapInfoGHC{..} = do
putE heapCapset
putE (fromIntegral gens :: Word16)
putE maxHeapSize
putE allocAreaSize
putE mblockSize
putE blockSize
putEventSpec CapCreate{cap} =
putCap cap
putEventSpec CapDelete{cap} =
putCap cap
putEventSpec CapDisable{cap} =
putCap cap
putEventSpec CapEnable{cap} =
putCap cap
putEventSpec (CapsetCreate cs ct) = do
putE cs
putE $ case ct of
CapsetCustom -> 1 :: Word16
CapsetOsProcess -> 2
CapsetClockDomain -> 3
CapsetUnknown -> 0
putEventSpec (CapsetDelete cs) =
putE cs
putEventSpec (CapsetAssignCap cs cp) = do
putE cs
putCap cp
putEventSpec (CapsetRemoveCap cs cp) = do
putE cs
putCap cp
putEventSpec (RtsIdentifier cs rts) = do
putE (fromIntegral (length rts) + sz_capset :: Word16)
putE cs
putEStr rts
putEventSpec (ProgramArgs cs as) = do
let as' = unsep as
putE (fromIntegral (length as') + sz_capset :: Word16)
putE cs
mapM_ putE as'
putEventSpec (ProgramEnv cs es) = do
let es' = unsep es
putE (fromIntegral (length es') + sz_capset :: Word16)
putE cs
mapM_ putE es'
putEventSpec (OsProcessPid cs pid) = do
putE cs
putE pid
putEventSpec (OsProcessParentPid cs ppid) = do
putE cs
putE ppid
putEventSpec (WallClockTime cs sec nsec) = do
putE cs
putE sec
putE nsec
putEventSpec (Message s) = do
putE (fromIntegral (length s) :: Word16)
mapM_ putE s
putEventSpec (UserMessage s) = do
putE (fromIntegral (length s) :: Word16)
mapM_ putE s
putEventSpec (UserMarker s) = do
putE (fromIntegral (length s) :: Word16)
mapM_ putE s
putEventSpec (UnknownEvent {}) = error "putEventSpec UnknownEvent"
putEventSpec (InternString str id) = do
putE len
mapM_ putE str
putE id
where len = (fromIntegral (length str) :: Word16) + sz_string_id
putEventSpec (Version s) = do
putE (fromIntegral (length s) :: Word16)
mapM_ putE s
putEventSpec (ProgramInvocation s) = do
putE (fromIntegral (length s) :: Word16)
mapM_ putE s
putEventSpec ( EdenStartReceive ) = return ()
putEventSpec ( EdenEndReceive ) = return ()
putEventSpec ( CreateProcess process ) = do
putE process
putEventSpec ( KillProcess process ) = do
putE process
putEventSpec ( AssignThreadToProcess thread process ) = do
putE thread
putE process
putEventSpec ( CreateMachine machine realtime ) = do
putE machine
putE realtime
putEventSpec ( KillMachine machine ) = do
putE machine
putEventSpec ( SendMessage mesTag senderProcess senderThread
receiverMachine receiverProcess receiverInport ) = do
putE (fromMsgTag mesTag)
putE senderProcess
putE senderThread
putE receiverMachine
putE receiverProcess
putE receiverInport
putEventSpec ( ReceiveMessage mesTag receiverProcess receiverInport
senderMachine senderProcess senderThread messageSize ) = do
putE (fromMsgTag mesTag)
putE receiverProcess
putE receiverInport
putE senderMachine
putE senderProcess
putE senderThread
putE messageSize
putEventSpec ( SendReceiveLocalMessage mesTag senderProcess senderThread
receiverProcess receiverInport ) = do
putE (fromMsgTag mesTag)
putE senderProcess
putE senderThread
putE receiverProcess
putE receiverInport
putEventSpec (MerStartParConjunction dyn_id static_id) = do
putE dyn_id
putE static_id
putEventSpec (MerEndParConjunction dyn_id) =
putE dyn_id
putEventSpec (MerEndParConjunct dyn_id) =
putE dyn_id
putEventSpec (MerCreateSpark dyn_id spark_id) = do
putE dyn_id
putE spark_id
putEventSpec (MerFutureCreate future_id name_id) = do
putE future_id
putE name_id
putEventSpec (MerFutureWaitNosuspend future_id) =
putE future_id
putEventSpec (MerFutureWaitSuspended future_id) =
putE future_id
putEventSpec (MerFutureSignal future_id) =
putE future_id
putEventSpec MerLookingForGlobalThread = return ()
putEventSpec MerWorkStealing = return ()
putEventSpec MerLookingForLocalSpark = return ()
putEventSpec (MerReleaseThread thread_id) =
putE thread_id
putEventSpec MerCapSleeping = return ()
putEventSpec MerCallingMain = return ()
putEventSpec PerfName{..} = do
putE (fromIntegral (length name) + sz_perf_num :: Word16)
putE perfNum
mapM_ putE name
putEventSpec PerfCounter{..} = do
putE perfNum
putE tid
putE period
putEventSpec PerfTracepoint{..} = do
putE perfNum
putE tid
putEventSpec HeapProfBegin {..} = do
putE heapProfId
putE heapProfSamplingPeriod
putE heapProfBreakdown
mapM_ (putE . T.unpack)
[ heapProfModuleFilter
, heapProfClosureDescrFilter
, heapProfTypeDescrFilter
, heapProfCostCentreFilter
, heapProfCostCentreStackFilter
, heapProfRetainerFilter
, heapProfBiographyFilter
]
putEventSpec HeapProfCostCentre {..} = do
putE heapProfCostCentreId
putE $ T.unpack heapProfLabel
putE $ T.unpack heapProfModule
putE $ T.unpack heapProfSrcLoc
putE heapProfFlags
putEventSpec HeapProfSampleBegin {..} =
putE heapProfSampleEra
putEventSpec HeapProfSampleCostCentre {..} = do
putE heapProfId
putE heapProfResidency
putE heapProfStackDepth
VU.mapM_ putE heapProfStack
putEventSpec HeapProfSampleString {..} = do
putE heapProfId
putE heapProfResidency
putE $ T.unpack heapProfLabel
unsep :: [String] -> String
unsep = concatMap (++"\0")
splitNull :: String -> [String]
splitNull [] = []
splitNull xs = case span (/= '\0') xs of
(x, xs') -> x : splitNull (drop 1 xs')
getText :: Get T.Text
getText = do
chunks <- G.getLazyByteStringNul
case TLE.decodeUtf8' chunks of
Left err -> fail $ show err
Right text -> return $ TL.toStrict text