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 75
