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..dcbc7d79d7569fa9b6794d9b47127adf30c38a6b |
--- /dev/null |
+++ b/tools/ipc_fuzzer/dump/message_dump.cc |
@@ -0,0 +1,58 @@ |
+#include "base/files/file_path.h" |
inferno
2015/03/05 18:46:48
Missing copyright notice.
|
+#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 IntToStringType base::IntToString16 |
+#define MESSAGE_DUMP_EXPORT __declspec(dllexport) |
inferno
2015/03/05 18:46:48
Can you use IPC_EXPORT or similar ?
|
+#else |
+#define IntToStringType base::IntToString |
+#define MESSAGE_DUMP_EXPORT __attribute__((visibility("default"))) |
+#endif |
+ |
+namespace ipc_fuzzer { |
+ |
+class IPCDump : public IPC::ChannelProxy::OutgoingMessageFilter { |
+ public: |
inferno
2015/03/05 18:46:48
2 spaces indent, not 1
|
+ ~IPCDump() { |
+ base::FilePath::StringType pid_string = |
+ IntToStringType(base::Process::Current().Pid()); |
+ base::FilePath output_file_path = |
+ dump_directory_.Append(pid_string + FILE_PATH_LITERAL(".ipcdump")); |
inferno
2015/03/05 18:46:48
please use a getter for dump_directory_, like dump
|
+ |
+ MessageFile::Write(output_file_path, messages_); |
+ } |
+ |
+ IPC::Message* Rewrite(IPC::Message* message) override { |
+ messages_.push_back(new IPC::Message(*message)); |
+ return message; |
+ } |
+ |
+ void set_dump_directory(base::FilePath& dump_directory) { |
+ dump_directory_ = dump_directory; |
+ } |
+ |
+ private: |
+ MessageVector messages_; |
inferno
2015/03/05 18:46:48
indents issue.
|
+ 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(base::FilePath& dump_directory); |
+} |
+ |
+IPC::ChannelProxy::OutgoingMessageFilter* GetFilter(void) { |
+ return &ipc_fuzzer::g_ipcdump; |
+} |
+ |
+void SetDumpDirectory(base::FilePath& dump_directory) { |
+ ipc_fuzzer::g_ipcdump.set_dump_directory(dump_directory); |
+} |