1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
|
// Copyright (c) 2012 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "ipc/ipc_message_utils.h"
#include "base/files/file_path.h"
#include "ipc/ipc_message.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace IPC {
namespace {
// Tests nesting of messages as parameters to other messages.
TEST(IPCMessageUtilsTest, NestedMessages) {
int32 nested_routing = 12;
uint32 nested_type = 78;
int nested_content = 456789;
Message::PriorityValue nested_priority = Message::PRIORITY_HIGH;
Message nested_msg(nested_routing, nested_type, nested_priority);
nested_msg.set_sync();
ParamTraits<int>::Write(&nested_msg, nested_content);
// Outer message contains the nested one as its parameter.
int32 outer_routing = 91;
uint32 outer_type = 88;
Message::PriorityValue outer_priority = Message::PRIORITY_NORMAL;
Message outer_msg(outer_routing, outer_type, outer_priority);
ParamTraits<Message>::Write(&outer_msg, nested_msg);
// Read back the nested message.
base::PickleIterator iter(outer_msg);
IPC::Message result_msg;
ASSERT_TRUE(ParamTraits<Message>::Read(&outer_msg, &iter, &result_msg));
// Verify nested message headers.
EXPECT_EQ(nested_msg.routing_id(), result_msg.routing_id());
EXPECT_EQ(nested_msg.type(), result_msg.type());
EXPECT_EQ(nested_msg.priority(), result_msg.priority());
EXPECT_EQ(nested_msg.flags(), result_msg.flags());
// Verify nested message content
base::PickleIterator nested_iter(nested_msg);
int result_content = 0;
ASSERT_TRUE(ParamTraits<int>::Read(&nested_msg, &nested_iter,
&result_content));
EXPECT_EQ(nested_content, result_content);
// Try reading past the ends for both messages and make sure it fails.
IPC::Message dummy;
ASSERT_FALSE(ParamTraits<Message>::Read(&outer_msg, &iter, &dummy));
ASSERT_FALSE(ParamTraits<int>::Read(&nested_msg, &nested_iter,
&result_content));
}
// Tests that detection of various bad parameters is working correctly.
TEST(IPCMessageUtilsTest, ParameterValidation) {
base::FilePath::StringType ok_string(FILE_PATH_LITERAL("hello"), 5);
base::FilePath::StringType bad_string(FILE_PATH_LITERAL("hel\0o"), 5);
// Change this if ParamTraits<FilePath>::Write() changes.
IPC::Message message;
ParamTraits<base::FilePath::StringType>::Write(&message, ok_string);
ParamTraits<base::FilePath::StringType>::Write(&message, bad_string);
base::PickleIterator iter(message);
base::FilePath ok_path;
base::FilePath bad_path;
ASSERT_TRUE(ParamTraits<base::FilePath>::Read(&message, &iter, &ok_path));
ASSERT_FALSE(ParamTraits<base::FilePath>::Read(&message, &iter, &bad_path));
}
TEST(IPCMessageUtilsTest, StackVector) {
static const size_t stack_capacity = 5;
base::StackVector<double, stack_capacity> stack_vector;
for (size_t i = 0; i < 2 * stack_capacity; i++)
stack_vector->push_back(i * 2.0);
IPC::Message msg(1, 2, IPC::Message::PRIORITY_NORMAL);
IPC::WriteParam(&msg, stack_vector);
base::StackVector<double, stack_capacity> output;
base::PickleIterator iter(msg);
EXPECT_TRUE(IPC::ReadParam(&msg, &iter, &output));
for (size_t i = 0; i < 2 * stack_capacity; i++)
EXPECT_EQ(stack_vector[i], output[i]);
}
} // namespace
} // namespace IPC
|
