Bug 1863729 - IPC Fuzzing: Add MOZ_FUZZ_DUMP_FILTER. r=truber

Differential Revision: https://phabricator.services.mozilla.com/D193063
Brli · Nov 13, 2023 · 367939a · 367939a
1 parent b9069ea
commit 367939a
Showing 1 changed file with 20 additions and 2 deletions.
diff --git a/tools/fuzzing/ipc/IPCFuzzController.cpp b/tools/fuzzing/ipc/IPCFuzzController.cpp
@@ -1024,12 +1024,30 @@ UniquePtr<IPC::Message> IPCFuzzController::replaceIPCMessage(
     return aMsg;
   }
 
+  static bool dumpFilterInitialized = false;
+  static std::string dumpFilter;
+  if (!dumpFilterInitialized) {
+    const char* dumpFilterStr = getenv("MOZ_FUZZ_DUMP_FILTER");
+    if (dumpFilterStr) {
+      dumpFilter = std::string(dumpFilterStr);
+    }
+    dumpFilterInitialized = true;
+  }
+
   if (aMsg->type() != mIPCTriggerMsg) {
     if ((mIPCDumpMsg && aMsg->type() == mIPCDumpMsg.value()) ||
         (mIPCDumpAllMsgsSize.isSome() &&
          aMsg->Buffers().Size() >= mIPCDumpAllMsgsSize.value())) {
-      dumpIPCMessageToFile(aMsg, mIPCDumpCount);
-      mIPCDumpCount++;
+      if (!dumpFilter.empty()) {
+        std::string msgName(IPC::StringFromIPCMessageType(aMsg->type()));
+        if (msgName.find(dumpFilter) != std::string::npos) {
+          dumpIPCMessageToFile(aMsg, mIPCDumpCount);
+          mIPCDumpCount++;
+        }
+      } else {
+        dumpIPCMessageToFile(aMsg, mIPCDumpCount);
+        mIPCDumpCount++;
+      }
     }
 
     // Not the trigger message. Output additional information here for