OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "ipc/ipc_message_utils.h" | 5 #include "ipc/ipc_message_utils.h" |
6 | 6 |
7 #include "base/files/file_path.h" | 7 #include "base/files/file_path.h" |
8 #include "ipc/ipc_message.h" | 8 #include "ipc/ipc_message.h" |
9 #include "testing/gtest/include/gtest/gtest.h" | 9 #include "testing/gtest/include/gtest/gtest.h" |
10 | 10 |
(...skipping 11 matching lines...) Expand all Loading... |
22 ParamTraits<int>::Write(&nested_msg, nested_content); | 22 ParamTraits<int>::Write(&nested_msg, nested_content); |
23 | 23 |
24 // Outer message contains the nested one as its parameter. | 24 // Outer message contains the nested one as its parameter. |
25 int32 outer_routing = 91; | 25 int32 outer_routing = 91; |
26 uint32 outer_type = 88; | 26 uint32 outer_type = 88; |
27 Message::PriorityValue outer_priority = Message::PRIORITY_NORMAL; | 27 Message::PriorityValue outer_priority = Message::PRIORITY_NORMAL; |
28 Message outer_msg(outer_routing, outer_type, outer_priority); | 28 Message outer_msg(outer_routing, outer_type, outer_priority); |
29 ParamTraits<Message>::Write(&outer_msg, nested_msg); | 29 ParamTraits<Message>::Write(&outer_msg, nested_msg); |
30 | 30 |
31 // Read back the nested message. | 31 // Read back the nested message. |
32 PickleIterator iter(outer_msg); | 32 base::PickleIterator iter(outer_msg); |
33 IPC::Message result_msg; | 33 IPC::Message result_msg; |
34 ASSERT_TRUE(ParamTraits<Message>::Read(&outer_msg, &iter, &result_msg)); | 34 ASSERT_TRUE(ParamTraits<Message>::Read(&outer_msg, &iter, &result_msg)); |
35 | 35 |
36 // Verify nested message headers. | 36 // Verify nested message headers. |
37 EXPECT_EQ(nested_msg.routing_id(), result_msg.routing_id()); | 37 EXPECT_EQ(nested_msg.routing_id(), result_msg.routing_id()); |
38 EXPECT_EQ(nested_msg.type(), result_msg.type()); | 38 EXPECT_EQ(nested_msg.type(), result_msg.type()); |
39 EXPECT_EQ(nested_msg.priority(), result_msg.priority()); | 39 EXPECT_EQ(nested_msg.priority(), result_msg.priority()); |
40 EXPECT_EQ(nested_msg.flags(), result_msg.flags()); | 40 EXPECT_EQ(nested_msg.flags(), result_msg.flags()); |
41 | 41 |
42 // Verify nested message content | 42 // Verify nested message content |
43 PickleIterator nested_iter(nested_msg); | 43 base::PickleIterator nested_iter(nested_msg); |
44 int result_content = 0; | 44 int result_content = 0; |
45 ASSERT_TRUE(ParamTraits<int>::Read(&nested_msg, &nested_iter, | 45 ASSERT_TRUE(ParamTraits<int>::Read(&nested_msg, &nested_iter, |
46 &result_content)); | 46 &result_content)); |
47 EXPECT_EQ(nested_content, result_content); | 47 EXPECT_EQ(nested_content, result_content); |
48 | 48 |
49 // Try reading past the ends for both messages and make sure it fails. | 49 // Try reading past the ends for both messages and make sure it fails. |
50 IPC::Message dummy; | 50 IPC::Message dummy; |
51 ASSERT_FALSE(ParamTraits<Message>::Read(&outer_msg, &iter, &dummy)); | 51 ASSERT_FALSE(ParamTraits<Message>::Read(&outer_msg, &iter, &dummy)); |
52 ASSERT_FALSE(ParamTraits<int>::Read(&nested_msg, &nested_iter, | 52 ASSERT_FALSE(ParamTraits<int>::Read(&nested_msg, &nested_iter, |
53 &result_content)); | 53 &result_content)); |
54 } | 54 } |
55 | 55 |
56 // Tests that detection of various bad parameters is working correctly. | 56 // Tests that detection of various bad parameters is working correctly. |
57 TEST(IPCMessageUtilsTest, ParameterValidation) { | 57 TEST(IPCMessageUtilsTest, ParameterValidation) { |
58 base::FilePath::StringType ok_string(FILE_PATH_LITERAL("hello"), 5); | 58 base::FilePath::StringType ok_string(FILE_PATH_LITERAL("hello"), 5); |
59 base::FilePath::StringType bad_string(FILE_PATH_LITERAL("hel\0o"), 5); | 59 base::FilePath::StringType bad_string(FILE_PATH_LITERAL("hel\0o"), 5); |
60 | 60 |
61 // Change this if ParamTraits<FilePath>::Write() changes. | 61 // Change this if ParamTraits<FilePath>::Write() changes. |
62 IPC::Message message; | 62 IPC::Message message; |
63 ParamTraits<base::FilePath::StringType>::Write(&message, ok_string); | 63 ParamTraits<base::FilePath::StringType>::Write(&message, ok_string); |
64 ParamTraits<base::FilePath::StringType>::Write(&message, bad_string); | 64 ParamTraits<base::FilePath::StringType>::Write(&message, bad_string); |
65 | 65 |
66 PickleIterator iter(message); | 66 base::PickleIterator iter(message); |
67 base::FilePath ok_path; | 67 base::FilePath ok_path; |
68 base::FilePath bad_path; | 68 base::FilePath bad_path; |
69 ASSERT_TRUE(ParamTraits<base::FilePath>::Read(&message, &iter, &ok_path)); | 69 ASSERT_TRUE(ParamTraits<base::FilePath>::Read(&message, &iter, &ok_path)); |
70 ASSERT_FALSE(ParamTraits<base::FilePath>::Read(&message, &iter, &bad_path)); | 70 ASSERT_FALSE(ParamTraits<base::FilePath>::Read(&message, &iter, &bad_path)); |
71 } | 71 } |
72 | 72 |
73 } // namespace | 73 } // namespace |
74 } // namespace IPC | 74 } // namespace IPC |
OLD | NEW |