brettw@chromium.org | f48910a | 2012-06-29 09:05:04 +0900 | [diff] [blame] | 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 | |
tfarina@chromium.org | 97808a3 | 2013-02-16 05:05:17 +0900 | [diff] [blame] | 5 | #include "ipc/ipc_message_utils.h" |
| 6 | |
brettw@chromium.org | 59eef1f | 2013-02-24 14:40:52 +0900 | [diff] [blame] | 7 | #include "base/files/file_path.h" |
brettw@chromium.org | f48910a | 2012-06-29 09:05:04 +0900 | [diff] [blame] | 8 | #include "ipc/ipc_message.h" |
brettw@chromium.org | f48910a | 2012-06-29 09:05:04 +0900 | [diff] [blame] | 9 | #include "testing/gtest/include/gtest/gtest.h" |
| 10 | |
tfarina@chromium.org | 97808a3 | 2013-02-16 05:05:17 +0900 | [diff] [blame] | 11 | namespace IPC { |
viettrungluu@chromium.org | 7ca1913 | 2013-01-12 05:56:22 +0900 | [diff] [blame] | 12 | namespace { |
brettw@chromium.org | f48910a | 2012-06-29 09:05:04 +0900 | [diff] [blame] | 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 | |
cevans@chromium.org | 2d83ff4 | 2013-01-03 13:11:54 +0900 | [diff] [blame] | 56 | // Tests that detection of various bad parameters is working correctly. |
| 57 | TEST(IPCMessageUtilsTest, ParameterValidation) { |
brettw@chromium.org | 22b3fda | 2013-02-10 13:49:30 +0900 | [diff] [blame] | 58 | base::FilePath::StringType ok_string(FILE_PATH_LITERAL("hello"), 5); |
| 59 | base::FilePath::StringType bad_string(FILE_PATH_LITERAL("hel\0o"), 5); |
aedla@chromium.org | 51127b9 | 2013-01-28 22:47:55 +0900 | [diff] [blame] | 60 | |
| 61 | // Change this if ParamTraits<FilePath>::Write() changes. |
cevans@chromium.org | 2d83ff4 | 2013-01-03 13:11:54 +0900 | [diff] [blame] | 62 | IPC::Message message; |
brettw@chromium.org | 22b3fda | 2013-02-10 13:49:30 +0900 | [diff] [blame] | 63 | ParamTraits<base::FilePath::StringType>::Write(&message, ok_string); |
| 64 | ParamTraits<base::FilePath::StringType>::Write(&message, bad_string); |
cevans@chromium.org | 2d83ff4 | 2013-01-03 13:11:54 +0900 | [diff] [blame] | 65 | |
| 66 | PickleIterator iter(message); |
brettw@chromium.org | 22b3fda | 2013-02-10 13:49:30 +0900 | [diff] [blame] | 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)); |
cevans@chromium.org | 2d83ff4 | 2013-01-03 13:11:54 +0900 | [diff] [blame] | 71 | } |
| 72 | |
viettrungluu@chromium.org | 7ca1913 | 2013-01-12 05:56:22 +0900 | [diff] [blame] | 73 | } // namespace |
tfarina@chromium.org | 97808a3 | 2013-02-16 05:05:17 +0900 | [diff] [blame] | 74 | } // namespace IPC |