| OLD | NEW |
| (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 | |
| OLD | NEW |