OLD | NEW |
1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 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 // TODO(vtl): Enable this on non-POSIX once we have a non-POSIX implementation. | 5 // TODO(vtl): Enable this on non-POSIX once we have a non-POSIX implementation. |
6 #include "build/build_config.h" | 6 #include "build/build_config.h" |
7 #if defined(OS_POSIX) | 7 #if defined(OS_POSIX) |
8 | 8 |
9 #include <stdint.h> | 9 #include <stdint.h> |
10 | 10 |
(...skipping 104 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
115 MultiprocessMessagePipeTest() {} | 115 MultiprocessMessagePipeTest() {} |
116 virtual ~MultiprocessMessagePipeTest() {} | 116 virtual ~MultiprocessMessagePipeTest() {} |
117 | 117 |
118 virtual void TearDown() OVERRIDE { | 118 virtual void TearDown() OVERRIDE { |
119 if (io_thread_wrapper_.is_initialized()) | 119 if (io_thread_wrapper_.is_initialized()) |
120 io_thread_wrapper_.Shutdown(); | 120 io_thread_wrapper_.Shutdown(); |
121 mojo::test::MultiprocessTestBase::TearDown(); | 121 mojo::test::MultiprocessTestBase::TearDown(); |
122 } | 122 } |
123 | 123 |
124 void Init(scoped_refptr<MessagePipe> mp) { | 124 void Init(scoped_refptr<MessagePipe> mp) { |
125 io_thread_wrapper_.Init(platform_server_channel.get(), mp); | 125 io_thread_wrapper_.Init(server_platform_channel.get(), mp); |
126 } | 126 } |
127 | 127 |
128 private: | 128 private: |
129 IOThreadWrapper io_thread_wrapper_; | 129 IOThreadWrapper io_thread_wrapper_; |
130 | 130 |
131 DISALLOW_COPY_AND_ASSIGN(MultiprocessMessagePipeTest); | 131 DISALLOW_COPY_AND_ASSIGN(MultiprocessMessagePipeTest); |
132 }; | 132 }; |
133 | 133 |
134 MojoResult WaitIfNecessary(scoped_refptr<MessagePipe> mp, MojoWaitFlags flags) { | 134 MojoResult WaitIfNecessary(scoped_refptr<MessagePipe> mp, MojoWaitFlags flags) { |
135 Waiter waiter; | 135 Waiter waiter; |
136 waiter.Init(); | 136 waiter.Init(); |
137 | 137 |
138 MojoResult add_result = mp->AddWaiter(0, &waiter, flags, MOJO_RESULT_OK); | 138 MojoResult add_result = mp->AddWaiter(0, &waiter, flags, MOJO_RESULT_OK); |
139 if (add_result != MOJO_RESULT_OK) { | 139 if (add_result != MOJO_RESULT_OK) { |
140 return (add_result == MOJO_RESULT_ALREADY_EXISTS) ? MOJO_RESULT_OK : | 140 return (add_result == MOJO_RESULT_ALREADY_EXISTS) ? MOJO_RESULT_OK : |
141 add_result; | 141 add_result; |
142 } | 142 } |
143 | 143 |
144 MojoResult wait_result = waiter.Wait(MOJO_DEADLINE_INDEFINITE); | 144 MojoResult wait_result = waiter.Wait(MOJO_DEADLINE_INDEFINITE); |
145 mp->RemoveWaiter(0, &waiter); | 145 mp->RemoveWaiter(0, &waiter); |
146 return wait_result; | 146 return wait_result; |
147 } | 147 } |
148 | 148 |
149 // For each message received, sends a reply message with the same contents | 149 // For each message received, sends a reply message with the same contents |
150 // repeated twice, until the other end is closed or it receives "quitquitquit" | 150 // repeated twice, until the other end is closed or it receives "quitquitquit" |
151 // (which it doesn't reply to). It'll return the number of messages received, | 151 // (which it doesn't reply to). It'll return the number of messages received, |
152 // not including any "quitquitquit" message, modulo 100. | 152 // not including any "quitquitquit" message, modulo 100. |
153 MOJO_MULTIPROCESS_TEST_CHILD_MAIN(EchoEcho) { | 153 MOJO_MULTIPROCESS_TEST_CHILD_MAIN(EchoEcho) { |
154 IOThreadWrapper io_thread_wrapper; | 154 IOThreadWrapper io_thread_wrapper; |
155 PlatformClientChannel* const platform_client_channel = | 155 PlatformChannel* const client_platform_channel = |
156 MultiprocessMessagePipeTest::platform_client_channel.get(); | 156 MultiprocessMessagePipeTest::client_platform_channel.get(); |
157 CHECK(platform_client_channel); | 157 CHECK(client_platform_channel); |
158 CHECK(platform_client_channel->is_valid()); | 158 CHECK(client_platform_channel->is_valid()); |
159 scoped_refptr<MessagePipe> mp(new MessagePipe( | 159 scoped_refptr<MessagePipe> mp(new MessagePipe( |
160 scoped_ptr<MessagePipeEndpoint>(new LocalMessagePipeEndpoint()), | 160 scoped_ptr<MessagePipeEndpoint>(new LocalMessagePipeEndpoint()), |
161 scoped_ptr<MessagePipeEndpoint>(new ProxyMessagePipeEndpoint()))); | 161 scoped_ptr<MessagePipeEndpoint>(new ProxyMessagePipeEndpoint()))); |
162 io_thread_wrapper.Init(platform_client_channel, mp); | 162 io_thread_wrapper.Init(client_platform_channel, mp); |
163 | 163 |
164 const std::string quitquitquit("quitquitquit"); | 164 const std::string quitquitquit("quitquitquit"); |
165 int rv = 0; | 165 int rv = 0; |
166 for (;; rv = (rv + 1) % 100) { | 166 for (;; rv = (rv + 1) % 100) { |
167 // Wait for our end of the message pipe to be readable. | 167 // Wait for our end of the message pipe to be readable. |
168 MojoResult result = WaitIfNecessary(mp, MOJO_WAIT_FLAG_READABLE); | 168 MojoResult result = WaitIfNecessary(mp, MOJO_WAIT_FLAG_READABLE); |
169 if (result != MOJO_RESULT_OK) { | 169 if (result != MOJO_RESULT_OK) { |
170 // It was closed, probably. | 170 // It was closed, probably. |
171 CHECK_EQ(result, MOJO_RESULT_FAILED_PRECONDITION); | 171 CHECK_EQ(result, MOJO_RESULT_FAILED_PRECONDITION); |
172 break; | 172 break; |
(...skipping 116 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
289 mp->Close(0); | 289 mp->Close(0); |
290 | 290 |
291 EXPECT_EQ(static_cast<int>(kNumMessages % 100), WaitForChildShutdown()); | 291 EXPECT_EQ(static_cast<int>(kNumMessages % 100), WaitForChildShutdown()); |
292 } | 292 } |
293 | 293 |
294 } // namespace | 294 } // namespace |
295 } // namespace system | 295 } // namespace system |
296 } // namespace mojo | 296 } // namespace mojo |
297 | 297 |
298 #endif // defined(OS_POSIX) | 298 #endif // defined(OS_POSIX) |
OLD | NEW |