Bug 1866421 - Make Nyx::is_replay work before snapshot point. r=truber

Differential Revision: https://phabricator.services.mozilla.com/D194643
Brli · Nov 27, 2023 · 993863a · 993863a
1 parent 76269b7
commit 993863a
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/tools/fuzzing/ipc/IPCFuzzController.cpp b/tools/fuzzing/ipc/IPCFuzzController.cpp
@@ -1088,6 +1088,10 @@ void IPCFuzzController::SynchronizeOnMessageExecution(
 
 static void dumpIPCMessageToFile(const UniquePtr<IPC::Message>& aMsg,
                                  uint32_t aDumpCount, bool aUseNyx = false) {
+  if (Nyx::instance().is_replay()) {
+    return;
+  }
+
   std::stringstream dumpFilename;
   std::string msgName(IPC::StringFromIPCMessageType(aMsg->type()));
   std::replace(msgName.begin(), msgName.end(), ':', '_');

diff --git a/tools/fuzzing/nyx/Nyx.cpp b/tools/fuzzing/nyx/Nyx.cpp
@@ -19,7 +19,12 @@
 namespace mozilla {
 namespace fuzzing {
 
-Nyx::Nyx() {}
+Nyx::Nyx() {
+  char* testFilePtr = getenv("MOZ_FUZZ_TESTFILE");
+  if (testFilePtr) {
+    mReplayMode = true;
+  }
+}
 
 // static
 Nyx& Nyx::instance() {
@@ -63,8 +68,6 @@ void Nyx::start(void) {
   // Check if we are in replay mode.
   char* testFilePtr = getenv("MOZ_FUZZ_TESTFILE");
   if (testFilePtr) {
-    mReplayMode = true;
-
     MOZ_FUZZING_NYX_PRINT("[Replay Mode] Reading data file...\n");
 
     std::string testFile(testFilePtr);