tools/ipc_fuzzer/message_dump/message_dump.cc

   1 // Copyright (c) 2015 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "base/files/file_path.h"
   6 #include "base/process/process.h"
   7 #include "base/strings/string_number_conversions.h"
   8 #include "ipc/ipc_channel_proxy.h"
   9 #include "tools/ipc_fuzzer/message_lib/message_file.h"
  10
  11 #if defined(OS_WIN)
  12 #define PidToStringType base::UintToString16
  13 #define MESSAGE_DUMP_EXPORT __declspec(dllexport)
  14 #else
  15 #define PidToStringType base::IntToString
  16 #define MESSAGE_DUMP_EXPORT __attribute__((visibility("default")))
  17 #endif
  18
  19 namespace ipc_fuzzer {
  20
  21 class IPCDump : public IPC::ChannelProxy::OutgoingMessageFilter {
  22  public:
  23   ~IPCDump() {
  24     base::FilePath::StringType pid_string =
  25         PidToStringType(base::Process::Current().Pid());
  26     base::FilePath output_file_path =
  27         dump_directory().Append(pid_string + FILE_PATH_LITERAL(".ipcdump"));
  28
  29     MessageFile::Write(output_file_path, messages_);
  30   }
  31
  32   IPC::Message* Rewrite(IPC::Message* message) override {
  33     messages_.push_back(new IPC::Message(*message));
  34     return message;
  35   }
  36
  37   base::FilePath dump_directory() const { return dump_directory_; }
  38
  39   void set_dump_directory(const base::FilePath& dump_directory) {
  40     dump_directory_ = dump_directory;
  41   }
  42
  43  private:
  44   MessageVector messages_;
  45   base::FilePath dump_directory_;
  46 };
  47
  48 IPCDump g_ipcdump;
  49
  50 }  // namespace ipc_fuzzer
  51
  52 // Entry point avoiding mangled names.
  53 extern "C" {
  54 MESSAGE_DUMP_EXPORT IPC::ChannelProxy::OutgoingMessageFilter* GetFilter(void);
  55 MESSAGE_DUMP_EXPORT void SetDumpDirectory(const base::FilePath& dump_directory);
  56 }
  57
  58 IPC::ChannelProxy::OutgoingMessageFilter* GetFilter(void) {
  59   return &ipc_fuzzer::g_ipcdump;
  60 }
  61
  62 void SetDumpDirectory(const base::FilePath& dump_directory) {
  63   ipc_fuzzer::g_ipcdump.set_dump_directory(dump_directory);
  64 }