Index: tools/ipc_fuzzer/dump/message_dump.cc |
diff --git a/tools/ipc_fuzzer/dump/message_dump.cc b/tools/ipc_fuzzer/dump/message_dump.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..5fd5a0658e16c5c0e1baddcf1694585b52215d40 |
--- /dev/null |
+++ b/tools/ipc_fuzzer/dump/message_dump.cc |
@@ -0,0 +1,64 @@ |
+// Copyright (c) 2015 The Chromium Authors. All rights reserved. |
+// Use of this source code is governed by a BSD-style license that can be |
+// found in the LICENSE file. |
+ |
+#include "base/files/file_path.h" |
+#include "base/process/process.h" |
+#include "base/strings/string_number_conversions.h" |
+#include "ipc/ipc_channel_proxy.h" |
+#include "tools/ipc_fuzzer/message_lib/message_file.h" |
+ |
+#if defined(OS_WIN) |
+#define PidToStringType base::UintToString16 |
+#define MESSAGE_DUMP_EXPORT __declspec(dllexport) |
+#else |
+#define PidToStringType base::IntToString |
+#define MESSAGE_DUMP_EXPORT __attribute__((visibility("default"))) |
+#endif |
+ |
+namespace ipc_fuzzer { |
+ |
+class IPCDump : public IPC::ChannelProxy::OutgoingMessageFilter { |
+ public: |
+ ~IPCDump() { |
+ base::FilePath::StringType pid_string = |
+ PidToStringType(base::Process::Current().Pid()); |
+ base::FilePath output_file_path = |
+ dump_directory().Append(pid_string + FILE_PATH_LITERAL(".ipcdump")); |
+ |
+ MessageFile::Write(output_file_path, messages_); |
+ } |
+ |
+ IPC::Message* Rewrite(IPC::Message* message) override { |
+ messages_.push_back(new IPC::Message(*message)); |
+ return message; |
+ } |
+ |
+ base::FilePath dump_directory() const { return dump_directory_; } |
+ |
+ void set_dump_directory(const base::FilePath& dump_directory) { |
+ dump_directory_ = dump_directory; |
+ } |
+ |
+ private: |
+ MessageVector messages_; |
+ base::FilePath dump_directory_; |
+}; |
+ |
+IPCDump g_ipcdump; |
+ |
+} // namespace ipc_fuzzer |
+ |
+// Entry point avoiding mangled names. |
+extern "C" { |
+MESSAGE_DUMP_EXPORT IPC::ChannelProxy::OutgoingMessageFilter* GetFilter(void); |
+MESSAGE_DUMP_EXPORT void SetDumpDirectory(const base::FilePath& dump_directory); |
+} |
+ |
+IPC::ChannelProxy::OutgoingMessageFilter* GetFilter(void) { |
+ return &ipc_fuzzer::g_ipcdump; |
+} |
+ |
+void SetDumpDirectory(const base::FilePath& dump_directory) { |
+ ipc_fuzzer::g_ipcdump.set_dump_directory(dump_directory); |
+} |