ipc/ipc_message_utils_unittest.cc

   1 // Copyright (c) 2012 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 "ipc/ipc_message_utils.h"
   6
   7 #include "base/files/file_path.h"
   8 #include "ipc/ipc_message.h"
   9 #include "testing/gtest/include/gtest/gtest.h"
  10
  11 namespace IPC {
  12 namespace {
  13
  14 // Tests nesting of messages as parameters to other messages.
  15 TEST(IPCMessageUtilsTest, NestedMessages) {
  16   int32 nested_routing = 12;
  17   uint32 nested_type = 78;
  18   int nested_content = 456789;
  19   Message::PriorityValue nested_priority = Message::PRIORITY_HIGH;
  20   Message nested_msg(nested_routing, nested_type, nested_priority);
  21   nested_msg.set_sync();
  22   ParamTraits<int>::Write(&nested_msg, nested_content);
  23
  24   // Outer message contains the nested one as its parameter.
  25   int32 outer_routing = 91;
  26   uint32 outer_type = 88;
  27   Message::PriorityValue outer_priority = Message::PRIORITY_NORMAL;
  28   Message outer_msg(outer_routing, outer_type, outer_priority);
  29   ParamTraits<Message>::Write(&outer_msg, nested_msg);
  30
  31   // Read back the nested message.
  32   PickleIterator iter(outer_msg);
  33   IPC::Message result_msg;
  34   ASSERT_TRUE(ParamTraits<Message>::Read(&outer_msg, &iter, &result_msg));
  35
  36   // Verify nested message headers.
  37   EXPECT_EQ(nested_msg.routing_id(), result_msg.routing_id());
  38   EXPECT_EQ(nested_msg.type(), result_msg.type());
  39   EXPECT_EQ(nested_msg.priority(), result_msg.priority());
  40   EXPECT_EQ(nested_msg.flags(), result_msg.flags());
  41
  42   // Verify nested message content
  43   PickleIterator nested_iter(nested_msg);
  44   int result_content = 0;
  45   ASSERT_TRUE(ParamTraits<int>::Read(&nested_msg, &nested_iter,
  46                                      &result_content));
  47   EXPECT_EQ(nested_content, result_content);
  48
  49   // Try reading past the ends for both messages and make sure it fails.
  50   IPC::Message dummy;
  51   ASSERT_FALSE(ParamTraits<Message>::Read(&outer_msg, &iter, &dummy));
  52   ASSERT_FALSE(ParamTraits<int>::Read(&nested_msg, &nested_iter,
  53                                       &result_content));
  54 }
  55
  56 // Tests that detection of various bad parameters is working correctly.
  57 TEST(IPCMessageUtilsTest, ParameterValidation) {
  58   base::FilePath::StringType ok_string(FILE_PATH_LITERAL("hello"), 5);
  59   base::FilePath::StringType bad_string(FILE_PATH_LITERAL("hel\0o"), 5);
  60
  61   // Change this if ParamTraits<FilePath>::Write() changes.
  62   IPC::Message message;
  63   ParamTraits<base::FilePath::StringType>::Write(&message, ok_string);
  64   ParamTraits<base::FilePath::StringType>::Write(&message, bad_string);
  65
  66   PickleIterator iter(message);
  67   base::FilePath ok_path;
  68   base::FilePath bad_path;
  69   ASSERT_TRUE(ParamTraits<base::FilePath>::Read(&message, &iter, &ok_path));
  70   ASSERT_FALSE(ParamTraits<base::FilePath>::Read(&message, &iter, &bad_path));
  71 }
  72
  73 }  // namespace
  74 }  // namespace IPC