Test start/endEventlogging: first header must be EVENT_HEADER_BEGIN

4 files changed, 207 insertions, 0 deletions

diff --git a/testsuite/tests/rts/RestartEventLogging.hs b/testsuite/tests/rts/RestartEventLogging.hs
new file mode 100644
index 0000000000..ac72577f04
--- /dev/null
+++ b/testsuite/tests/rts/RestartEventLogging.hs
@@ -0,0 +1,36 @@
+{-# LANGUAGE ForeignFunctionInterface #-}
+
+import System.IO
+
+import Control.Concurrent
+import Control.Monad (forever, void, forM_)
+import GHC.Conc
+
+
+-- Test that the start/end/restartEventLog interface works as expected.
+main :: IO ()
+main = do
+
+  --
+  -- Start other threads to generate some event log events.
+  --
+
+  let loop f = void $ forkIO $ forever (f >> yield)
+
+  forM_ [1..10] $ \_ -> do
+    -- start lots of short lived threads
+    loop (forkIO $ yield)
+
+    -- sparks
+    loop (let x = 1 + (1 :: Int) in return (par x (sum [0,1,2,3,x])))
+
+  --
+  -- Try restarting event logging a few times.
+  --
+
+  putStrLn "Restarting eventlog..."
+  hFlush stdout
+  c_restart_eventlog
+
+foreign import ccall safe "c_restart_eventlog"
+  c_restart_eventlog :: IO ()
diff --git a/testsuite/tests/rts/RestartEventLogging.stdout b/testsuite/tests/rts/RestartEventLogging.stdout
new file mode 100644
index 0000000000..0d024b006d
--- /dev/null
+++ b/testsuite/tests/rts/RestartEventLogging.stdout
@@ -0,0 +1,89 @@
+Restarting eventlog...
+failed to start eventlog
+init
+Event log started with EVENT_HEADER_BEGIN
+stop
+init
+Event log started with EVENT_HEADER_BEGIN
+stop
+init
+Event log started with EVENT_HEADER_BEGIN
+stop
+init
+Event log started with EVENT_HEADER_BEGIN
+stop
+init
+Event log started with EVENT_HEADER_BEGIN
+stop
+init
+Event log started with EVENT_HEADER_BEGIN
+stop
+init
+Event log started with EVENT_HEADER_BEGIN
+stop
+init
+Event log started with EVENT_HEADER_BEGIN
+stop
+init
+Event log started with EVENT_HEADER_BEGIN
+stop
+init
+Event log started with EVENT_HEADER_BEGIN
+stop
+init
+Event log started with EVENT_HEADER_BEGIN
+stop
+init
+Event log started with EVENT_HEADER_BEGIN
+stop
+init
+Event log started with EVENT_HEADER_BEGIN
+stop
+init
+Event log started with EVENT_HEADER_BEGIN
+stop
+init
+Event log started with EVENT_HEADER_BEGIN
+stop
+init
+Event log started with EVENT_HEADER_BEGIN
+stop
+init
+Event log started with EVENT_HEADER_BEGIN
+stop
+init
+Event log started with EVENT_HEADER_BEGIN
+stop
+init
+Event log started with EVENT_HEADER_BEGIN
+stop
+init
+Event log started with EVENT_HEADER_BEGIN
+stop
+init
+Event log started with EVENT_HEADER_BEGIN
+stop
+init
+Event log started with EVENT_HEADER_BEGIN
+stop
+init
+Event log started with EVENT_HEADER_BEGIN
+stop
+init
+Event log started with EVENT_HEADER_BEGIN
+stop
+init
+Event log started with EVENT_HEADER_BEGIN
+stop
+init
+Event log started with EVENT_HEADER_BEGIN
+stop
+init
+Event log started with EVENT_HEADER_BEGIN
+stop
+init
+Event log started with EVENT_HEADER_BEGIN
+stop
+init
+Event log started with EVENT_HEADER_BEGIN
+stop
diff --git a/testsuite/tests/rts/RestartEventLogging_c.c b/testsuite/tests/rts/RestartEventLogging_c.c
new file mode 100644
index 0000000000..56bc4c9307
--- /dev/null
+++ b/testsuite/tests/rts/RestartEventLogging_c.c
@@ -0,0 +1,79 @@
+#include <stdio.h>
+#include <Rts.h>
+#include <rts/EventLogFormat.h>
+
+#define STOPPED       0
+#define STARTED       1
+#define WRITTEN       2
+
+static int32_t state = STOPPED;
+Mutex writeMutex;
+
+void test_init(void) {
+  if (state != STOPPED) {
+    printf("test_init was not called first or directly after test_stop\n");
+  }
+
+  state = STARTED;
+  printf("init\n");
+  fflush(stdout);
+}
+
+bool test_write(void *eventlog, size_t eventlog_size) {
+  ACQUIRE_LOCK(&writeMutex);
+  if (state == STOPPED) {
+    printf("test_init was not called\n");
+  }
+  if (state == STARTED) {
+    // Note that the encoding of the header is coppied from EventLog.c (see `postInt32()`)
+    StgWord8 * words = (StgWord8 *)eventlog;
+    StgInt32 h32 = EVENT_HEADER_BEGIN;
+    StgWord32 h = (StgWord32)h32; // Yes, the cast is correct. See `postInt32()`
+    if ((words[0] != (StgWord8)(h >> 24))
+        || (words[1] != (StgWord8)(h >> 16))
+        || (words[2] != (StgWord8)(h >> 8))
+        || (words[3] != (StgWord8)h)) {
+      printf("ERROR: event does not start with EVENT_HEADER_BEGIN\n");
+      printf("0x%x != 0x%x\n", words[0], (StgWord8)(h >> 24));
+      printf("0x%x != 0x%x\n", words[1], (StgWord8)(h >> 16));
+      printf("0x%x != 0x%x\n", words[2], (StgWord8)(h >> 8));
+      printf("0x%x != 0x%x\n", words[3], (StgWord8)h);
+    }
+    else {
+      printf("Event log started with EVENT_HEADER_BEGIN\n");
+    }
+  }
+
+  fflush(stdout);
+  state = WRITTEN;
+
+  RELEASE_LOCK(&writeMutex);
+  return true;
+}
+
+void test_flush(void) {
+}
+
+void test_stop(void) {
+  state = STOPPED;
+  printf("stop\n");
+  fflush(stdout);
+}
+
+const EventLogWriter writer = {
+  .initEventLogWriter = test_init,
+  .writeEventLog = test_write,
+  .flushEventLog = test_flush,
+  .stopEventLogWriter = test_stop
+};
+
+void c_restart_eventlog(void) {
+  initMutex(&writeMutex);
+  for (int i = 0; i < 30; i++) {
+    if (!startEventLogging(&writer)) {
+      printf("failed to start eventlog\n");
+    }
+    endEventLogging();
+  }
+}
+
diff --git a/testsuite/tests/rts/all.T b/testsuite/tests/rts/all.T
index a5a57b6a8c..e74834d2a1 100644
--- a/testsuite/tests/rts/all.T
+++ b/testsuite/tests/rts/all.T
@@ -414,6 +414,9 @@ test('T13676',
 test('InitEventLogging',
      [only_ways(['normal']), extra_run_opts('+RTS -RTS')],
      compile_and_run, ['-eventlog InitEventLogging_c.c'])
+test('RestartEventLogging',
+     [only_ways(['threaded1','threaded2']), extra_run_opts('+RTS -la -RTS')],
+     compile_and_run, ['-eventlog RestartEventLogging_c.c'])
 
 test('T17088',
      [only_ways(['normal']), extra_run_opts('+RTS -c -A256k -RTS')],