Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(102)

Side by Side Diff: mojo/edk/system/message_pipe_perftest.cc

Issue 623883002: Revert "Move mojo edk into mojo/edk" (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: Created 6 years, 2 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
« no previous file with comments | « mojo/edk/system/message_pipe_endpoint.cc ('k') | mojo/edk/system/message_pipe_test_utils.h » ('j') | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
(Empty)
1 // Copyright 2014 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 <stdint.h>
6 #include <stdio.h>
7 #include <string.h>
8
9 #include <string>
10 #include <vector>
11
12 #include "base/bind.h"
13 #include "base/location.h"
14 #include "base/logging.h"
15 #include "base/macros.h"
16 #include "base/pickle.h"
17 #include "base/strings/stringprintf.h"
18 #include "base/test/perf_time_logger.h"
19 #include "base/time/time.h"
20 #include "mojo/edk/embedder/scoped_platform_handle.h"
21 #include "mojo/edk/system/channel.h"
22 #include "mojo/edk/system/local_message_pipe_endpoint.h"
23 #include "mojo/edk/system/message_pipe.h"
24 #include "mojo/edk/system/message_pipe_test_utils.h"
25 #include "mojo/edk/system/proxy_message_pipe_endpoint.h"
26 #include "mojo/edk/system/raw_channel.h"
27 #include "mojo/edk/system/test_utils.h"
28 #include "mojo/edk/test/test_utils.h"
29 #include "testing/gtest/include/gtest/gtest.h"
30
31 namespace mojo {
32 namespace system {
33 namespace {
34
35 class MultiprocessMessagePipePerfTest
36 : public test::MultiprocessMessagePipeTestBase {
37 public:
38 MultiprocessMessagePipePerfTest() : message_count_(0), message_size_(0) {}
39
40 void SetUpMeasurement(int message_count, size_t message_size) {
41 message_count_ = message_count;
42 message_size_ = message_size;
43 payload_ = Pickle();
44 payload_.WriteString(std::string(message_size, '*'));
45 read_buffer_.resize(message_size * 2);
46 }
47
48 protected:
49 void WriteWaitThenRead(scoped_refptr<MessagePipe> mp) {
50 CHECK_EQ(mp->WriteMessage(0,
51 UserPointer<const void>(payload_.data()),
52 static_cast<uint32_t>(payload_.size()),
53 nullptr,
54 MOJO_WRITE_MESSAGE_FLAG_NONE),
55 MOJO_RESULT_OK);
56 HandleSignalsState hss;
57 CHECK_EQ(test::WaitIfNecessary(mp, MOJO_HANDLE_SIGNAL_READABLE, &hss),
58 MOJO_RESULT_OK);
59 uint32_t read_buffer_size = static_cast<uint32_t>(read_buffer_.size());
60 CHECK_EQ(mp->ReadMessage(0,
61 UserPointer<void>(&read_buffer_[0]),
62 MakeUserPointer(&read_buffer_size),
63 nullptr,
64 nullptr,
65 MOJO_READ_MESSAGE_FLAG_NONE),
66 MOJO_RESULT_OK);
67 CHECK_EQ(read_buffer_size, static_cast<uint32_t>(payload_.size()));
68 }
69
70 void SendQuitMessage(scoped_refptr<MessagePipe> mp) {
71 CHECK_EQ(mp->WriteMessage(0,
72 UserPointer<const void>(""),
73 0,
74 nullptr,
75 MOJO_WRITE_MESSAGE_FLAG_NONE),
76 MOJO_RESULT_OK);
77 }
78
79 void Measure(scoped_refptr<MessagePipe> mp) {
80 // Have one ping-pong to ensure channel being established.
81 WriteWaitThenRead(mp);
82
83 std::string test_name =
84 base::StringPrintf("IPC_Perf_%dx_%u",
85 message_count_,
86 static_cast<unsigned>(message_size_));
87 base::PerfTimeLogger logger(test_name.c_str());
88
89 for (int i = 0; i < message_count_; ++i)
90 WriteWaitThenRead(mp);
91
92 logger.Done();
93 }
94
95 private:
96 int message_count_;
97 size_t message_size_;
98 Pickle payload_;
99 std::string read_buffer_;
100 scoped_ptr<base::PerfTimeLogger> perf_logger_;
101 };
102
103 // For each message received, sends a reply message with the same contents
104 // repeated twice, until the other end is closed or it receives "quitquitquit"
105 // (which it doesn't reply to). It'll return the number of messages received,
106 // not including any "quitquitquit" message, modulo 100.
107 MOJO_MULTIPROCESS_TEST_CHILD_MAIN(PingPongClient) {
108 embedder::SimplePlatformSupport platform_support;
109 test::ChannelThread channel_thread(&platform_support);
110 embedder::ScopedPlatformHandle client_platform_handle =
111 mojo::test::MultiprocessTestHelper::client_platform_handle.Pass();
112 CHECK(client_platform_handle.is_valid());
113 scoped_refptr<ChannelEndpoint> ep;
114 scoped_refptr<MessagePipe> mp(MessagePipe::CreateLocalProxy(&ep));
115 channel_thread.Start(client_platform_handle.Pass(), ep);
116
117 std::string buffer(1000000, '\0');
118 int rv = 0;
119 while (true) {
120 // Wait for our end of the message pipe to be readable.
121 HandleSignalsState hss;
122 MojoResult result =
123 test::WaitIfNecessary(mp, MOJO_HANDLE_SIGNAL_READABLE, &hss);
124 if (result != MOJO_RESULT_OK) {
125 rv = result;
126 break;
127 }
128
129 uint32_t read_size = static_cast<uint32_t>(buffer.size());
130 CHECK_EQ(mp->ReadMessage(0,
131 UserPointer<void>(&buffer[0]),
132 MakeUserPointer(&read_size),
133 nullptr,
134 nullptr,
135 MOJO_READ_MESSAGE_FLAG_NONE),
136 MOJO_RESULT_OK);
137
138 // Empty message indicates quitting
139 if (0 == read_size)
140 break;
141
142 CHECK_EQ(mp->WriteMessage(0,
143 UserPointer<const void>(&buffer[0]),
144 static_cast<uint32_t>(read_size),
145 nullptr,
146 MOJO_WRITE_MESSAGE_FLAG_NONE),
147 MOJO_RESULT_OK);
148 }
149
150 mp->Close(0);
151 return rv;
152 }
153
154 // Repeatedly sends messages as previous one got replied by the child.
155 // Waits for the child to close its end before quitting once specified
156 // number of messages has been sent.
157 TEST_F(MultiprocessMessagePipePerfTest, PingPong) {
158 helper()->StartChild("PingPongClient");
159
160 scoped_refptr<ChannelEndpoint> ep;
161 scoped_refptr<MessagePipe> mp(MessagePipe::CreateLocalProxy(&ep));
162 Init(ep);
163
164 // This values are set to align with one at ipc_pertests.cc for comparison.
165 const size_t kMsgSize[5] = {12, 144, 1728, 20736, 248832};
166 const int kMessageCount[5] = {50000, 50000, 50000, 12000, 1000};
167
168 for (size_t i = 0; i < 5; i++) {
169 SetUpMeasurement(kMessageCount[i], kMsgSize[i]);
170 Measure(mp);
171 }
172
173 SendQuitMessage(mp);
174 mp->Close(0);
175 EXPECT_EQ(0, helper()->WaitForChildShutdown());
176 }
177
178 } // namespace
179 } // namespace system
180 } // namespace mojo
OLDNEW
« no previous file with comments | « mojo/edk/system/message_pipe_endpoint.cc ('k') | mojo/edk/system/message_pipe_test_utils.h » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698