OLD | NEW |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | 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 | 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 <stdint.h> | 5 #include <stdint.h> |
6 #include <stdio.h> | 6 #include <stdio.h> |
7 #include <string.h> | 7 #include <string.h> |
8 | 8 |
9 #include <vector> | 9 #include <vector> |
10 | 10 |
(...skipping 30 matching lines...) Expand all Loading... |
41 namespace system { | 41 namespace system { |
42 namespace { | 42 namespace { |
43 | 43 |
44 class RemoteMessagePipeTest : public testing::Test { | 44 class RemoteMessagePipeTest : public testing::Test { |
45 public: | 45 public: |
46 RemoteMessagePipeTest() : io_thread_(base::TestIOThread::kAutoStart) {} | 46 RemoteMessagePipeTest() : io_thread_(base::TestIOThread::kAutoStart) {} |
47 ~RemoteMessagePipeTest() override {} | 47 ~RemoteMessagePipeTest() override {} |
48 | 48 |
49 void SetUp() override { | 49 void SetUp() override { |
50 io_thread_.PostTaskAndWait( | 50 io_thread_.PostTaskAndWait( |
51 FROM_HERE, | 51 FROM_HERE, base::Bind(&RemoteMessagePipeTest::SetUpOnIOThread, |
52 base::Bind(&RemoteMessagePipeTest::SetUpOnIOThread, | 52 base::Unretained(this))); |
53 base::Unretained(this))); | |
54 } | 53 } |
55 | 54 |
56 void TearDown() override { | 55 void TearDown() override { |
57 io_thread_.PostTaskAndWait( | 56 io_thread_.PostTaskAndWait( |
58 FROM_HERE, | 57 FROM_HERE, base::Bind(&RemoteMessagePipeTest::TearDownOnIOThread, |
59 base::Bind(&RemoteMessagePipeTest::TearDownOnIOThread, | 58 base::Unretained(this))); |
60 base::Unretained(this))); | |
61 } | 59 } |
62 | 60 |
63 protected: | 61 protected: |
64 // This connects the two given |ChannelEndpoint|s. It assumes/requires that | 62 // This connects the two given |ChannelEndpoint|s. It assumes/requires that |
65 // this is the bootstrap case (i.e., no other message pipes have ever been | 63 // this is the bootstrap case (i.e., no other message pipes have ever been |
66 // hosted on the channel). | 64 // hosted on the channel). |
67 void BootstrapChannelEndpoints(scoped_refptr<ChannelEndpoint> ep0, | 65 void BootstrapChannelEndpoints(scoped_refptr<ChannelEndpoint> ep0, |
68 scoped_refptr<ChannelEndpoint> ep1) { | 66 scoped_refptr<ChannelEndpoint> ep1) { |
69 io_thread_.PostTaskAndWait( | 67 io_thread_.PostTaskAndWait( |
70 FROM_HERE, | 68 FROM_HERE, |
71 base::Bind(&RemoteMessagePipeTest::BootstrapChannelEndpointsOnIOThread, | 69 base::Bind(&RemoteMessagePipeTest::BootstrapChannelEndpointsOnIOThread, |
72 base::Unretained(this), | 70 base::Unretained(this), ep0, ep1)); |
73 ep0, | |
74 ep1)); | |
75 } | 71 } |
76 | 72 |
77 // This bootstraps |ep| on |channels_[channel_index]|. It assumes/requires | 73 // This bootstraps |ep| on |channels_[channel_index]|. It assumes/requires |
78 // that this is the bootstrap case (i.e., no message pipes have ever been | 74 // that this is the bootstrap case (i.e., no message pipes have ever been |
79 // hosted on the channel). This returns *without* waiting. | 75 // hosted on the channel). This returns *without* waiting. |
80 void BootstrapChannelEndpointNoWait(unsigned channel_index, | 76 void BootstrapChannelEndpointNoWait(unsigned channel_index, |
81 scoped_refptr<ChannelEndpoint> ep) { | 77 scoped_refptr<ChannelEndpoint> ep) { |
82 io_thread_.PostTask( | 78 io_thread_.PostTask( |
83 FROM_HERE, | 79 FROM_HERE, |
84 base::Bind(&RemoteMessagePipeTest::BootstrapChannelEndpointOnIOThread, | 80 base::Bind(&RemoteMessagePipeTest::BootstrapChannelEndpointOnIOThread, |
85 base::Unretained(this), | 81 base::Unretained(this), channel_index, ep)); |
86 channel_index, | |
87 ep)); | |
88 } | 82 } |
89 | 83 |
90 void RestoreInitialState() { | 84 void RestoreInitialState() { |
91 io_thread_.PostTaskAndWait( | 85 io_thread_.PostTaskAndWait( |
92 FROM_HERE, | 86 FROM_HERE, |
93 base::Bind(&RemoteMessagePipeTest::RestoreInitialStateOnIOThread, | 87 base::Bind(&RemoteMessagePipeTest::RestoreInitialStateOnIOThread, |
94 base::Unretained(this))); | 88 base::Unretained(this))); |
95 } | 89 } |
96 | 90 |
97 embedder::PlatformSupport* platform_support() { return &platform_support_; } | 91 embedder::PlatformSupport* platform_support() { return &platform_support_; } |
(...skipping 93 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
191 // Write in one direction: MP 0, port 0 -> ... -> MP 1, port 1. | 185 // Write in one direction: MP 0, port 0 -> ... -> MP 1, port 1. |
192 | 186 |
193 // Prepare to wait on MP 1, port 1. (Add the waiter now. Otherwise, if we do | 187 // Prepare to wait on MP 1, port 1. (Add the waiter now. Otherwise, if we do |
194 // it later, it might already be readable.) | 188 // it later, it might already be readable.) |
195 waiter.Init(); | 189 waiter.Init(); |
196 ASSERT_EQ( | 190 ASSERT_EQ( |
197 MOJO_RESULT_OK, | 191 MOJO_RESULT_OK, |
198 mp1->AddWaiter(1, &waiter, MOJO_HANDLE_SIGNAL_READABLE, 123, nullptr)); | 192 mp1->AddWaiter(1, &waiter, MOJO_HANDLE_SIGNAL_READABLE, 123, nullptr)); |
199 | 193 |
200 // Write to MP 0, port 0. | 194 // Write to MP 0, port 0. |
201 EXPECT_EQ(MOJO_RESULT_OK, | 195 EXPECT_EQ( |
202 mp0->WriteMessage(0, | 196 MOJO_RESULT_OK, |
203 UserPointer<const void>(kHello), | 197 mp0->WriteMessage(0, UserPointer<const void>(kHello), sizeof(kHello), |
204 sizeof(kHello), | 198 nullptr, MOJO_WRITE_MESSAGE_FLAG_NONE)); |
205 nullptr, | |
206 MOJO_WRITE_MESSAGE_FLAG_NONE)); | |
207 | 199 |
208 // Wait. | 200 // Wait. |
209 EXPECT_EQ(MOJO_RESULT_OK, waiter.Wait(MOJO_DEADLINE_INDEFINITE, &context)); | 201 EXPECT_EQ(MOJO_RESULT_OK, waiter.Wait(MOJO_DEADLINE_INDEFINITE, &context)); |
210 EXPECT_EQ(123u, context); | 202 EXPECT_EQ(123u, context); |
211 hss = HandleSignalsState(); | 203 hss = HandleSignalsState(); |
212 mp1->RemoveWaiter(1, &waiter, &hss); | 204 mp1->RemoveWaiter(1, &waiter, &hss); |
213 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, | 205 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, |
214 hss.satisfied_signals); | 206 hss.satisfied_signals); |
215 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, | 207 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, |
216 hss.satisfiable_signals); | 208 hss.satisfiable_signals); |
217 | 209 |
218 // Read from MP 1, port 1. | 210 // Read from MP 1, port 1. |
219 EXPECT_EQ(MOJO_RESULT_OK, | 211 EXPECT_EQ(MOJO_RESULT_OK, |
220 mp1->ReadMessage(1, | 212 mp1->ReadMessage(1, UserPointer<void>(buffer), |
221 UserPointer<void>(buffer), | 213 MakeUserPointer(&buffer_size), nullptr, nullptr, |
222 MakeUserPointer(&buffer_size), | |
223 nullptr, | |
224 nullptr, | |
225 MOJO_READ_MESSAGE_FLAG_NONE)); | 214 MOJO_READ_MESSAGE_FLAG_NONE)); |
226 EXPECT_EQ(sizeof(kHello), static_cast<size_t>(buffer_size)); | 215 EXPECT_EQ(sizeof(kHello), static_cast<size_t>(buffer_size)); |
227 EXPECT_STREQ(kHello, buffer); | 216 EXPECT_STREQ(kHello, buffer); |
228 | 217 |
229 // Write in the other direction: MP 1, port 1 -> ... -> MP 0, port 0. | 218 // Write in the other direction: MP 1, port 1 -> ... -> MP 0, port 0. |
230 | 219 |
231 waiter.Init(); | 220 waiter.Init(); |
232 ASSERT_EQ( | 221 ASSERT_EQ( |
233 MOJO_RESULT_OK, | 222 MOJO_RESULT_OK, |
234 mp0->AddWaiter(0, &waiter, MOJO_HANDLE_SIGNAL_READABLE, 456, nullptr)); | 223 mp0->AddWaiter(0, &waiter, MOJO_HANDLE_SIGNAL_READABLE, 456, nullptr)); |
235 | 224 |
236 EXPECT_EQ(MOJO_RESULT_OK, | 225 EXPECT_EQ( |
237 mp1->WriteMessage(1, | 226 MOJO_RESULT_OK, |
238 UserPointer<const void>(kWorld), | 227 mp1->WriteMessage(1, UserPointer<const void>(kWorld), sizeof(kWorld), |
239 sizeof(kWorld), | 228 nullptr, MOJO_WRITE_MESSAGE_FLAG_NONE)); |
240 nullptr, | |
241 MOJO_WRITE_MESSAGE_FLAG_NONE)); | |
242 | 229 |
243 EXPECT_EQ(MOJO_RESULT_OK, waiter.Wait(MOJO_DEADLINE_INDEFINITE, &context)); | 230 EXPECT_EQ(MOJO_RESULT_OK, waiter.Wait(MOJO_DEADLINE_INDEFINITE, &context)); |
244 EXPECT_EQ(456u, context); | 231 EXPECT_EQ(456u, context); |
245 hss = HandleSignalsState(); | 232 hss = HandleSignalsState(); |
246 mp0->RemoveWaiter(0, &waiter, &hss); | 233 mp0->RemoveWaiter(0, &waiter, &hss); |
247 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, | 234 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, |
248 hss.satisfied_signals); | 235 hss.satisfied_signals); |
249 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, | 236 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, |
250 hss.satisfiable_signals); | 237 hss.satisfiable_signals); |
251 | 238 |
252 buffer_size = static_cast<uint32_t>(sizeof(buffer)); | 239 buffer_size = static_cast<uint32_t>(sizeof(buffer)); |
253 EXPECT_EQ(MOJO_RESULT_OK, | 240 EXPECT_EQ(MOJO_RESULT_OK, |
254 mp0->ReadMessage(0, | 241 mp0->ReadMessage(0, UserPointer<void>(buffer), |
255 UserPointer<void>(buffer), | 242 MakeUserPointer(&buffer_size), nullptr, nullptr, |
256 MakeUserPointer(&buffer_size), | |
257 nullptr, | |
258 nullptr, | |
259 MOJO_READ_MESSAGE_FLAG_NONE)); | 243 MOJO_READ_MESSAGE_FLAG_NONE)); |
260 EXPECT_EQ(sizeof(kWorld), static_cast<size_t>(buffer_size)); | 244 EXPECT_EQ(sizeof(kWorld), static_cast<size_t>(buffer_size)); |
261 EXPECT_STREQ(kWorld, buffer); | 245 EXPECT_STREQ(kWorld, buffer); |
262 | 246 |
263 // Close MP 0, port 0. | 247 // Close MP 0, port 0. |
264 mp0->Close(0); | 248 mp0->Close(0); |
265 | 249 |
266 // Try to wait for MP 1, port 1 to become readable. This will eventually fail | 250 // Try to wait for MP 1, port 1 to become readable. This will eventually fail |
267 // when it realizes that MP 0, port 0 has been closed. (It may also fail | 251 // when it realizes that MP 0, port 0 has been closed. (It may also fail |
268 // immediately.) | 252 // immediately.) |
(...skipping 44 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
313 ASSERT_TRUE(channels(0)); | 297 ASSERT_TRUE(channels(0)); |
314 ChannelEndpointId remote_id = channels(0)->AttachAndRunEndpoint(ep2, false); | 298 ChannelEndpointId remote_id = channels(0)->AttachAndRunEndpoint(ep2, false); |
315 EXPECT_TRUE(remote_id.is_remote()); | 299 EXPECT_TRUE(remote_id.is_remote()); |
316 | 300 |
317 waiter.Init(); | 301 waiter.Init(); |
318 ASSERT_EQ( | 302 ASSERT_EQ( |
319 MOJO_RESULT_OK, | 303 MOJO_RESULT_OK, |
320 mp1->AddWaiter(1, &waiter, MOJO_HANDLE_SIGNAL_READABLE, 123, nullptr)); | 304 mp1->AddWaiter(1, &waiter, MOJO_HANDLE_SIGNAL_READABLE, 123, nullptr)); |
321 | 305 |
322 EXPECT_EQ(MOJO_RESULT_OK, | 306 EXPECT_EQ(MOJO_RESULT_OK, |
323 mp0->WriteMessage(0, | 307 mp0->WriteMessage(0, UserPointer<const void>(&remote_id), |
324 UserPointer<const void>(&remote_id), | 308 sizeof(remote_id), nullptr, |
325 sizeof(remote_id), | |
326 nullptr, | |
327 MOJO_WRITE_MESSAGE_FLAG_NONE)); | 309 MOJO_WRITE_MESSAGE_FLAG_NONE)); |
328 | 310 |
329 EXPECT_EQ(MOJO_RESULT_OK, waiter.Wait(MOJO_DEADLINE_INDEFINITE, &context)); | 311 EXPECT_EQ(MOJO_RESULT_OK, waiter.Wait(MOJO_DEADLINE_INDEFINITE, &context)); |
330 EXPECT_EQ(123u, context); | 312 EXPECT_EQ(123u, context); |
331 hss = HandleSignalsState(); | 313 hss = HandleSignalsState(); |
332 mp1->RemoveWaiter(1, &waiter, &hss); | 314 mp1->RemoveWaiter(1, &waiter, &hss); |
333 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, | 315 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, |
334 hss.satisfied_signals); | 316 hss.satisfied_signals); |
335 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, | 317 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, |
336 hss.satisfiable_signals); | 318 hss.satisfiable_signals); |
337 | 319 |
338 ChannelEndpointId received_id; | 320 ChannelEndpointId received_id; |
339 buffer_size = static_cast<uint32_t>(sizeof(received_id)); | 321 buffer_size = static_cast<uint32_t>(sizeof(received_id)); |
340 EXPECT_EQ(MOJO_RESULT_OK, | 322 EXPECT_EQ(MOJO_RESULT_OK, |
341 mp1->ReadMessage(1, | 323 mp1->ReadMessage(1, UserPointer<void>(&received_id), |
342 UserPointer<void>(&received_id), | 324 MakeUserPointer(&buffer_size), nullptr, nullptr, |
343 MakeUserPointer(&buffer_size), | |
344 nullptr, | |
345 nullptr, | |
346 MOJO_READ_MESSAGE_FLAG_NONE)); | 325 MOJO_READ_MESSAGE_FLAG_NONE)); |
347 EXPECT_EQ(sizeof(received_id), static_cast<size_t>(buffer_size)); | 326 EXPECT_EQ(sizeof(received_id), static_cast<size_t>(buffer_size)); |
348 EXPECT_EQ(remote_id, received_id); | 327 EXPECT_EQ(remote_id, received_id); |
349 | 328 |
350 // Warning: The local side of mp3 is port 0, not port 1. | 329 // Warning: The local side of mp3 is port 0, not port 1. |
351 scoped_refptr<MessagePipe> mp3 = | 330 scoped_refptr<MessagePipe> mp3 = |
352 channels(1)->PassIncomingMessagePipe(received_id); | 331 channels(1)->PassIncomingMessagePipe(received_id); |
353 ASSERT_TRUE(mp3.get()); | 332 ASSERT_TRUE(mp3.get()); |
354 | 333 |
355 // Write: MP 2, port 0 -> MP 3, port 1. | 334 // Write: MP 2, port 0 -> MP 3, port 1. |
356 | 335 |
357 waiter.Init(); | 336 waiter.Init(); |
358 ASSERT_EQ( | 337 ASSERT_EQ( |
359 MOJO_RESULT_OK, | 338 MOJO_RESULT_OK, |
360 mp3->AddWaiter(0, &waiter, MOJO_HANDLE_SIGNAL_READABLE, 789, nullptr)); | 339 mp3->AddWaiter(0, &waiter, MOJO_HANDLE_SIGNAL_READABLE, 789, nullptr)); |
361 | 340 |
362 EXPECT_EQ(MOJO_RESULT_OK, | 341 EXPECT_EQ( |
363 mp2->WriteMessage(0, | 342 MOJO_RESULT_OK, |
364 UserPointer<const void>(kHello), | 343 mp2->WriteMessage(0, UserPointer<const void>(kHello), sizeof(kHello), |
365 sizeof(kHello), | 344 nullptr, MOJO_WRITE_MESSAGE_FLAG_NONE)); |
366 nullptr, | |
367 MOJO_WRITE_MESSAGE_FLAG_NONE)); | |
368 | 345 |
369 EXPECT_EQ(MOJO_RESULT_OK, waiter.Wait(MOJO_DEADLINE_INDEFINITE, &context)); | 346 EXPECT_EQ(MOJO_RESULT_OK, waiter.Wait(MOJO_DEADLINE_INDEFINITE, &context)); |
370 EXPECT_EQ(789u, context); | 347 EXPECT_EQ(789u, context); |
371 hss = HandleSignalsState(); | 348 hss = HandleSignalsState(); |
372 mp3->RemoveWaiter(0, &waiter, &hss); | 349 mp3->RemoveWaiter(0, &waiter, &hss); |
373 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, | 350 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, |
374 hss.satisfied_signals); | 351 hss.satisfied_signals); |
375 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, | 352 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, |
376 hss.satisfiable_signals); | 353 hss.satisfiable_signals); |
377 | 354 |
378 // Make sure there's nothing on MP 0, port 0 or MP 1, port 1 or MP 2, port 0. | 355 // Make sure there's nothing on MP 0, port 0 or MP 1, port 1 or MP 2, port 0. |
379 buffer_size = static_cast<uint32_t>(sizeof(buffer)); | 356 buffer_size = static_cast<uint32_t>(sizeof(buffer)); |
380 EXPECT_EQ(MOJO_RESULT_SHOULD_WAIT, | 357 EXPECT_EQ(MOJO_RESULT_SHOULD_WAIT, |
381 mp0->ReadMessage(0, | 358 mp0->ReadMessage(0, UserPointer<void>(buffer), |
382 UserPointer<void>(buffer), | 359 MakeUserPointer(&buffer_size), nullptr, nullptr, |
383 MakeUserPointer(&buffer_size), | |
384 nullptr, | |
385 nullptr, | |
386 MOJO_READ_MESSAGE_FLAG_NONE)); | 360 MOJO_READ_MESSAGE_FLAG_NONE)); |
387 buffer_size = static_cast<uint32_t>(sizeof(buffer)); | 361 buffer_size = static_cast<uint32_t>(sizeof(buffer)); |
388 EXPECT_EQ(MOJO_RESULT_SHOULD_WAIT, | 362 EXPECT_EQ(MOJO_RESULT_SHOULD_WAIT, |
389 mp1->ReadMessage(1, | 363 mp1->ReadMessage(1, UserPointer<void>(buffer), |
390 UserPointer<void>(buffer), | 364 MakeUserPointer(&buffer_size), nullptr, nullptr, |
391 MakeUserPointer(&buffer_size), | |
392 nullptr, | |
393 nullptr, | |
394 MOJO_READ_MESSAGE_FLAG_NONE)); | 365 MOJO_READ_MESSAGE_FLAG_NONE)); |
395 buffer_size = static_cast<uint32_t>(sizeof(buffer)); | 366 buffer_size = static_cast<uint32_t>(sizeof(buffer)); |
396 EXPECT_EQ(MOJO_RESULT_SHOULD_WAIT, | 367 EXPECT_EQ(MOJO_RESULT_SHOULD_WAIT, |
397 mp2->ReadMessage(0, | 368 mp2->ReadMessage(0, UserPointer<void>(buffer), |
398 UserPointer<void>(buffer), | 369 MakeUserPointer(&buffer_size), nullptr, nullptr, |
399 MakeUserPointer(&buffer_size), | |
400 nullptr, | |
401 nullptr, | |
402 MOJO_READ_MESSAGE_FLAG_NONE)); | 370 MOJO_READ_MESSAGE_FLAG_NONE)); |
403 | 371 |
404 // Read from MP 3, port 1. | 372 // Read from MP 3, port 1. |
405 buffer_size = static_cast<uint32_t>(sizeof(buffer)); | 373 buffer_size = static_cast<uint32_t>(sizeof(buffer)); |
406 EXPECT_EQ(MOJO_RESULT_OK, | 374 EXPECT_EQ(MOJO_RESULT_OK, |
407 mp3->ReadMessage(0, | 375 mp3->ReadMessage(0, UserPointer<void>(buffer), |
408 UserPointer<void>(buffer), | 376 MakeUserPointer(&buffer_size), nullptr, nullptr, |
409 MakeUserPointer(&buffer_size), | |
410 nullptr, | |
411 nullptr, | |
412 MOJO_READ_MESSAGE_FLAG_NONE)); | 377 MOJO_READ_MESSAGE_FLAG_NONE)); |
413 EXPECT_EQ(sizeof(kHello), static_cast<size_t>(buffer_size)); | 378 EXPECT_EQ(sizeof(kHello), static_cast<size_t>(buffer_size)); |
414 EXPECT_STREQ(kHello, buffer); | 379 EXPECT_STREQ(kHello, buffer); |
415 | 380 |
416 // Write: MP 0, port 0 -> MP 1, port 1 again. | 381 // Write: MP 0, port 0 -> MP 1, port 1 again. |
417 | 382 |
418 waiter.Init(); | 383 waiter.Init(); |
419 ASSERT_EQ( | 384 ASSERT_EQ( |
420 MOJO_RESULT_OK, | 385 MOJO_RESULT_OK, |
421 mp1->AddWaiter(1, &waiter, MOJO_HANDLE_SIGNAL_READABLE, 123, nullptr)); | 386 mp1->AddWaiter(1, &waiter, MOJO_HANDLE_SIGNAL_READABLE, 123, nullptr)); |
422 | 387 |
423 EXPECT_EQ(MOJO_RESULT_OK, | 388 EXPECT_EQ( |
424 mp0->WriteMessage(0, | 389 MOJO_RESULT_OK, |
425 UserPointer<const void>(kWorld), | 390 mp0->WriteMessage(0, UserPointer<const void>(kWorld), sizeof(kWorld), |
426 sizeof(kWorld), | 391 nullptr, MOJO_WRITE_MESSAGE_FLAG_NONE)); |
427 nullptr, | |
428 MOJO_WRITE_MESSAGE_FLAG_NONE)); | |
429 | 392 |
430 EXPECT_EQ(MOJO_RESULT_OK, waiter.Wait(MOJO_DEADLINE_INDEFINITE, &context)); | 393 EXPECT_EQ(MOJO_RESULT_OK, waiter.Wait(MOJO_DEADLINE_INDEFINITE, &context)); |
431 EXPECT_EQ(123u, context); | 394 EXPECT_EQ(123u, context); |
432 hss = HandleSignalsState(); | 395 hss = HandleSignalsState(); |
433 mp1->RemoveWaiter(1, &waiter, &hss); | 396 mp1->RemoveWaiter(1, &waiter, &hss); |
434 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, | 397 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, |
435 hss.satisfied_signals); | 398 hss.satisfied_signals); |
436 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, | 399 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, |
437 hss.satisfiable_signals); | 400 hss.satisfiable_signals); |
438 | 401 |
439 // Make sure there's nothing on the other ports. | 402 // Make sure there's nothing on the other ports. |
440 buffer_size = static_cast<uint32_t>(sizeof(buffer)); | 403 buffer_size = static_cast<uint32_t>(sizeof(buffer)); |
441 EXPECT_EQ(MOJO_RESULT_SHOULD_WAIT, | 404 EXPECT_EQ(MOJO_RESULT_SHOULD_WAIT, |
442 mp0->ReadMessage(0, | 405 mp0->ReadMessage(0, UserPointer<void>(buffer), |
443 UserPointer<void>(buffer), | 406 MakeUserPointer(&buffer_size), nullptr, nullptr, |
444 MakeUserPointer(&buffer_size), | |
445 nullptr, | |
446 nullptr, | |
447 MOJO_READ_MESSAGE_FLAG_NONE)); | 407 MOJO_READ_MESSAGE_FLAG_NONE)); |
448 buffer_size = static_cast<uint32_t>(sizeof(buffer)); | 408 buffer_size = static_cast<uint32_t>(sizeof(buffer)); |
449 EXPECT_EQ(MOJO_RESULT_SHOULD_WAIT, | 409 EXPECT_EQ(MOJO_RESULT_SHOULD_WAIT, |
450 mp2->ReadMessage(0, | 410 mp2->ReadMessage(0, UserPointer<void>(buffer), |
451 UserPointer<void>(buffer), | 411 MakeUserPointer(&buffer_size), nullptr, nullptr, |
452 MakeUserPointer(&buffer_size), | |
453 nullptr, | |
454 nullptr, | |
455 MOJO_READ_MESSAGE_FLAG_NONE)); | 412 MOJO_READ_MESSAGE_FLAG_NONE)); |
456 buffer_size = static_cast<uint32_t>(sizeof(buffer)); | 413 buffer_size = static_cast<uint32_t>(sizeof(buffer)); |
457 EXPECT_EQ(MOJO_RESULT_SHOULD_WAIT, | 414 EXPECT_EQ(MOJO_RESULT_SHOULD_WAIT, |
458 mp3->ReadMessage(0, | 415 mp3->ReadMessage(0, UserPointer<void>(buffer), |
459 UserPointer<void>(buffer), | 416 MakeUserPointer(&buffer_size), nullptr, nullptr, |
460 MakeUserPointer(&buffer_size), | |
461 nullptr, | |
462 nullptr, | |
463 MOJO_READ_MESSAGE_FLAG_NONE)); | 417 MOJO_READ_MESSAGE_FLAG_NONE)); |
464 | 418 |
465 buffer_size = static_cast<uint32_t>(sizeof(buffer)); | 419 buffer_size = static_cast<uint32_t>(sizeof(buffer)); |
466 EXPECT_EQ(MOJO_RESULT_OK, | 420 EXPECT_EQ(MOJO_RESULT_OK, |
467 mp1->ReadMessage(1, | 421 mp1->ReadMessage(1, UserPointer<void>(buffer), |
468 UserPointer<void>(buffer), | 422 MakeUserPointer(&buffer_size), nullptr, nullptr, |
469 MakeUserPointer(&buffer_size), | |
470 nullptr, | |
471 nullptr, | |
472 MOJO_READ_MESSAGE_FLAG_NONE)); | 423 MOJO_READ_MESSAGE_FLAG_NONE)); |
473 EXPECT_EQ(sizeof(kWorld), static_cast<size_t>(buffer_size)); | 424 EXPECT_EQ(sizeof(kWorld), static_cast<size_t>(buffer_size)); |
474 EXPECT_STREQ(kWorld, buffer); | 425 EXPECT_STREQ(kWorld, buffer); |
475 | 426 |
476 mp0->Close(0); | 427 mp0->Close(0); |
477 mp1->Close(1); | 428 mp1->Close(1); |
478 mp2->Close(0); | 429 mp2->Close(0); |
479 mp3->Close(0); | 430 mp3->Close(0); |
480 } | 431 } |
481 | 432 |
482 TEST_F(RemoteMessagePipeTest, CloseBeforeAttachAndRun) { | 433 TEST_F(RemoteMessagePipeTest, CloseBeforeAttachAndRun) { |
483 static const char kHello[] = "hello"; | 434 static const char kHello[] = "hello"; |
484 char buffer[100] = {0}; | 435 char buffer[100] = {0}; |
485 uint32_t buffer_size = static_cast<uint32_t>(sizeof(buffer)); | 436 uint32_t buffer_size = static_cast<uint32_t>(sizeof(buffer)); |
486 Waiter waiter; | 437 Waiter waiter; |
487 HandleSignalsState hss; | 438 HandleSignalsState hss; |
488 uint32_t context = 0; | 439 uint32_t context = 0; |
489 | 440 |
490 // Connect message pipes. MP 0, port 1 will be attached to channel 0 and | 441 // Connect message pipes. MP 0, port 1 will be attached to channel 0 and |
491 // connected to MP 1, port 0, which will be attached to channel 1. This leaves | 442 // connected to MP 1, port 0, which will be attached to channel 1. This leaves |
492 // MP 0, port 0 and MP 1, port 1 as the "user-facing" endpoints. | 443 // MP 0, port 0 and MP 1, port 1 as the "user-facing" endpoints. |
493 | 444 |
494 scoped_refptr<ChannelEndpoint> ep0; | 445 scoped_refptr<ChannelEndpoint> ep0; |
495 scoped_refptr<MessagePipe> mp0(MessagePipe::CreateLocalProxy(&ep0)); | 446 scoped_refptr<MessagePipe> mp0(MessagePipe::CreateLocalProxy(&ep0)); |
496 | 447 |
497 // Write to MP 0, port 0. | 448 // Write to MP 0, port 0. |
498 EXPECT_EQ(MOJO_RESULT_OK, | 449 EXPECT_EQ( |
499 mp0->WriteMessage(0, | 450 MOJO_RESULT_OK, |
500 UserPointer<const void>(kHello), | 451 mp0->WriteMessage(0, UserPointer<const void>(kHello), sizeof(kHello), |
501 sizeof(kHello), | 452 nullptr, MOJO_WRITE_MESSAGE_FLAG_NONE)); |
502 nullptr, | |
503 MOJO_WRITE_MESSAGE_FLAG_NONE)); | |
504 | 453 |
505 // Close MP 0, port 0 before it's even been attached to the channel and run. | 454 // Close MP 0, port 0 before it's even been attached to the channel and run. |
506 mp0->Close(0); | 455 mp0->Close(0); |
507 | 456 |
508 BootstrapChannelEndpointNoWait(0, ep0); | 457 BootstrapChannelEndpointNoWait(0, ep0); |
509 | 458 |
510 scoped_refptr<ChannelEndpoint> ep1; | 459 scoped_refptr<ChannelEndpoint> ep1; |
511 scoped_refptr<MessagePipe> mp1(MessagePipe::CreateProxyLocal(&ep1)); | 460 scoped_refptr<MessagePipe> mp1(MessagePipe::CreateProxyLocal(&ep1)); |
512 | 461 |
513 // Prepare to wait on MP 1, port 1. (Add the waiter now. Otherwise, if we do | 462 // Prepare to wait on MP 1, port 1. (Add the waiter now. Otherwise, if we do |
(...skipping 12 matching lines...) Expand all Loading... |
526 // Note: MP 1, port 1 should definitely should be readable, but it may or may | 475 // Note: MP 1, port 1 should definitely should be readable, but it may or may |
527 // not appear as writable (there's a race, and it may not have noticed that | 476 // not appear as writable (there's a race, and it may not have noticed that |
528 // the other side was closed yet -- e.g., inserting a sleep here would make it | 477 // the other side was closed yet -- e.g., inserting a sleep here would make it |
529 // much more likely to notice that it's no longer writable). | 478 // much more likely to notice that it's no longer writable). |
530 mp1->RemoveWaiter(1, &waiter, &hss); | 479 mp1->RemoveWaiter(1, &waiter, &hss); |
531 EXPECT_TRUE((hss.satisfied_signals & MOJO_HANDLE_SIGNAL_READABLE)); | 480 EXPECT_TRUE((hss.satisfied_signals & MOJO_HANDLE_SIGNAL_READABLE)); |
532 EXPECT_TRUE((hss.satisfiable_signals & MOJO_HANDLE_SIGNAL_READABLE)); | 481 EXPECT_TRUE((hss.satisfiable_signals & MOJO_HANDLE_SIGNAL_READABLE)); |
533 | 482 |
534 // Read from MP 1, port 1. | 483 // Read from MP 1, port 1. |
535 EXPECT_EQ(MOJO_RESULT_OK, | 484 EXPECT_EQ(MOJO_RESULT_OK, |
536 mp1->ReadMessage(1, | 485 mp1->ReadMessage(1, UserPointer<void>(buffer), |
537 UserPointer<void>(buffer), | 486 MakeUserPointer(&buffer_size), nullptr, nullptr, |
538 MakeUserPointer(&buffer_size), | |
539 nullptr, | |
540 nullptr, | |
541 MOJO_READ_MESSAGE_FLAG_NONE)); | 487 MOJO_READ_MESSAGE_FLAG_NONE)); |
542 EXPECT_EQ(sizeof(kHello), static_cast<size_t>(buffer_size)); | 488 EXPECT_EQ(sizeof(kHello), static_cast<size_t>(buffer_size)); |
543 EXPECT_STREQ(kHello, buffer); | 489 EXPECT_STREQ(kHello, buffer); |
544 | 490 |
545 // And MP 1, port 1. | 491 // And MP 1, port 1. |
546 mp1->Close(1); | 492 mp1->Close(1); |
547 } | 493 } |
548 | 494 |
549 TEST_F(RemoteMessagePipeTest, CloseBeforeConnect) { | 495 TEST_F(RemoteMessagePipeTest, CloseBeforeConnect) { |
550 static const char kHello[] = "hello"; | 496 static const char kHello[] = "hello"; |
551 char buffer[100] = {0}; | 497 char buffer[100] = {0}; |
552 uint32_t buffer_size = static_cast<uint32_t>(sizeof(buffer)); | 498 uint32_t buffer_size = static_cast<uint32_t>(sizeof(buffer)); |
553 Waiter waiter; | 499 Waiter waiter; |
554 HandleSignalsState hss; | 500 HandleSignalsState hss; |
555 uint32_t context = 0; | 501 uint32_t context = 0; |
556 | 502 |
557 // Connect message pipes. MP 0, port 1 will be attached to channel 0 and | 503 // Connect message pipes. MP 0, port 1 will be attached to channel 0 and |
558 // connected to MP 1, port 0, which will be attached to channel 1. This leaves | 504 // connected to MP 1, port 0, which will be attached to channel 1. This leaves |
559 // MP 0, port 0 and MP 1, port 1 as the "user-facing" endpoints. | 505 // MP 0, port 0 and MP 1, port 1 as the "user-facing" endpoints. |
560 | 506 |
561 scoped_refptr<ChannelEndpoint> ep0; | 507 scoped_refptr<ChannelEndpoint> ep0; |
562 scoped_refptr<MessagePipe> mp0(MessagePipe::CreateLocalProxy(&ep0)); | 508 scoped_refptr<MessagePipe> mp0(MessagePipe::CreateLocalProxy(&ep0)); |
563 | 509 |
564 // Write to MP 0, port 0. | 510 // Write to MP 0, port 0. |
565 EXPECT_EQ(MOJO_RESULT_OK, | 511 EXPECT_EQ( |
566 mp0->WriteMessage(0, | 512 MOJO_RESULT_OK, |
567 UserPointer<const void>(kHello), | 513 mp0->WriteMessage(0, UserPointer<const void>(kHello), sizeof(kHello), |
568 sizeof(kHello), | 514 nullptr, MOJO_WRITE_MESSAGE_FLAG_NONE)); |
569 nullptr, | |
570 MOJO_WRITE_MESSAGE_FLAG_NONE)); | |
571 | 515 |
572 BootstrapChannelEndpointNoWait(0, ep0); | 516 BootstrapChannelEndpointNoWait(0, ep0); |
573 | 517 |
574 // Close MP 0, port 0 before channel 1 is even connected. | 518 // Close MP 0, port 0 before channel 1 is even connected. |
575 mp0->Close(0); | 519 mp0->Close(0); |
576 | 520 |
577 scoped_refptr<ChannelEndpoint> ep1; | 521 scoped_refptr<ChannelEndpoint> ep1; |
578 scoped_refptr<MessagePipe> mp1(MessagePipe::CreateProxyLocal(&ep1)); | 522 scoped_refptr<MessagePipe> mp1(MessagePipe::CreateProxyLocal(&ep1)); |
579 | 523 |
580 // Prepare to wait on MP 1, port 1. (Add the waiter now. Otherwise, if we do | 524 // Prepare to wait on MP 1, port 1. (Add the waiter now. Otherwise, if we do |
(...skipping 12 matching lines...) Expand all Loading... |
593 // Note: MP 1, port 1 should definitely should be readable, but it may or may | 537 // Note: MP 1, port 1 should definitely should be readable, but it may or may |
594 // not appear as writable (there's a race, and it may not have noticed that | 538 // not appear as writable (there's a race, and it may not have noticed that |
595 // the other side was closed yet -- e.g., inserting a sleep here would make it | 539 // the other side was closed yet -- e.g., inserting a sleep here would make it |
596 // much more likely to notice that it's no longer writable). | 540 // much more likely to notice that it's no longer writable). |
597 mp1->RemoveWaiter(1, &waiter, &hss); | 541 mp1->RemoveWaiter(1, &waiter, &hss); |
598 EXPECT_TRUE((hss.satisfied_signals & MOJO_HANDLE_SIGNAL_READABLE)); | 542 EXPECT_TRUE((hss.satisfied_signals & MOJO_HANDLE_SIGNAL_READABLE)); |
599 EXPECT_TRUE((hss.satisfiable_signals & MOJO_HANDLE_SIGNAL_READABLE)); | 543 EXPECT_TRUE((hss.satisfiable_signals & MOJO_HANDLE_SIGNAL_READABLE)); |
600 | 544 |
601 // Read from MP 1, port 1. | 545 // Read from MP 1, port 1. |
602 EXPECT_EQ(MOJO_RESULT_OK, | 546 EXPECT_EQ(MOJO_RESULT_OK, |
603 mp1->ReadMessage(1, | 547 mp1->ReadMessage(1, UserPointer<void>(buffer), |
604 UserPointer<void>(buffer), | 548 MakeUserPointer(&buffer_size), nullptr, nullptr, |
605 MakeUserPointer(&buffer_size), | |
606 nullptr, | |
607 nullptr, | |
608 MOJO_READ_MESSAGE_FLAG_NONE)); | 549 MOJO_READ_MESSAGE_FLAG_NONE)); |
609 EXPECT_EQ(sizeof(kHello), static_cast<size_t>(buffer_size)); | 550 EXPECT_EQ(sizeof(kHello), static_cast<size_t>(buffer_size)); |
610 EXPECT_STREQ(kHello, buffer); | 551 EXPECT_STREQ(kHello, buffer); |
611 | 552 |
612 // And MP 1, port 1. | 553 // And MP 1, port 1. |
613 mp1->Close(1); | 554 mp1->Close(1); |
614 } | 555 } |
615 | 556 |
616 TEST_F(RemoteMessagePipeTest, HandlePassing) { | 557 TEST_F(RemoteMessagePipeTest, HandlePassing) { |
617 static const char kHello[] = "hello"; | 558 static const char kHello[] = "hello"; |
(...skipping 21 matching lines...) Expand all Loading... |
639 mp1->AddWaiter(1, &waiter, MOJO_HANDLE_SIGNAL_READABLE, 123, nullptr)); | 580 mp1->AddWaiter(1, &waiter, MOJO_HANDLE_SIGNAL_READABLE, 123, nullptr)); |
640 | 581 |
641 // Write to MP 0, port 0. | 582 // Write to MP 0, port 0. |
642 { | 583 { |
643 DispatcherTransport transport( | 584 DispatcherTransport transport( |
644 test::DispatcherTryStartTransport(dispatcher.get())); | 585 test::DispatcherTryStartTransport(dispatcher.get())); |
645 EXPECT_TRUE(transport.is_valid()); | 586 EXPECT_TRUE(transport.is_valid()); |
646 | 587 |
647 std::vector<DispatcherTransport> transports; | 588 std::vector<DispatcherTransport> transports; |
648 transports.push_back(transport); | 589 transports.push_back(transport); |
649 EXPECT_EQ(MOJO_RESULT_OK, | 590 EXPECT_EQ( |
650 mp0->WriteMessage(0, | 591 MOJO_RESULT_OK, |
651 UserPointer<const void>(kHello), | 592 mp0->WriteMessage(0, UserPointer<const void>(kHello), sizeof(kHello), |
652 sizeof(kHello), | 593 &transports, MOJO_WRITE_MESSAGE_FLAG_NONE)); |
653 &transports, | |
654 MOJO_WRITE_MESSAGE_FLAG_NONE)); | |
655 transport.End(); | 594 transport.End(); |
656 | 595 |
657 // |dispatcher| should have been closed. This is |DCHECK()|ed when the | 596 // |dispatcher| should have been closed. This is |DCHECK()|ed when the |
658 // |dispatcher| is destroyed. | 597 // |dispatcher| is destroyed. |
659 EXPECT_TRUE(dispatcher->HasOneRef()); | 598 EXPECT_TRUE(dispatcher->HasOneRef()); |
660 dispatcher = nullptr; | 599 dispatcher = nullptr; |
661 } | 600 } |
662 | 601 |
663 // Wait. | 602 // Wait. |
664 EXPECT_EQ(MOJO_RESULT_OK, waiter.Wait(MOJO_DEADLINE_INDEFINITE, &context)); | 603 EXPECT_EQ(MOJO_RESULT_OK, waiter.Wait(MOJO_DEADLINE_INDEFINITE, &context)); |
665 EXPECT_EQ(123u, context); | 604 EXPECT_EQ(123u, context); |
666 hss = HandleSignalsState(); | 605 hss = HandleSignalsState(); |
667 mp1->RemoveWaiter(1, &waiter, &hss); | 606 mp1->RemoveWaiter(1, &waiter, &hss); |
668 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, | 607 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, |
669 hss.satisfied_signals); | 608 hss.satisfied_signals); |
670 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, | 609 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, |
671 hss.satisfiable_signals); | 610 hss.satisfiable_signals); |
672 | 611 |
673 // Read from MP 1, port 1. | 612 // Read from MP 1, port 1. |
674 char read_buffer[100] = {0}; | 613 char read_buffer[100] = {0}; |
675 uint32_t read_buffer_size = static_cast<uint32_t>(sizeof(read_buffer)); | 614 uint32_t read_buffer_size = static_cast<uint32_t>(sizeof(read_buffer)); |
676 DispatcherVector read_dispatchers; | 615 DispatcherVector read_dispatchers; |
677 uint32_t read_num_dispatchers = 10; // Maximum to get. | 616 uint32_t read_num_dispatchers = 10; // Maximum to get. |
678 EXPECT_EQ(MOJO_RESULT_OK, | 617 EXPECT_EQ( |
679 mp1->ReadMessage(1, | 618 MOJO_RESULT_OK, |
680 UserPointer<void>(read_buffer), | 619 mp1->ReadMessage(1, UserPointer<void>(read_buffer), |
681 MakeUserPointer(&read_buffer_size), | 620 MakeUserPointer(&read_buffer_size), &read_dispatchers, |
682 &read_dispatchers, | 621 &read_num_dispatchers, MOJO_READ_MESSAGE_FLAG_NONE)); |
683 &read_num_dispatchers, | |
684 MOJO_READ_MESSAGE_FLAG_NONE)); | |
685 EXPECT_EQ(sizeof(kHello), static_cast<size_t>(read_buffer_size)); | 622 EXPECT_EQ(sizeof(kHello), static_cast<size_t>(read_buffer_size)); |
686 EXPECT_STREQ(kHello, read_buffer); | 623 EXPECT_STREQ(kHello, read_buffer); |
687 EXPECT_EQ(1u, read_dispatchers.size()); | 624 EXPECT_EQ(1u, read_dispatchers.size()); |
688 EXPECT_EQ(1u, read_num_dispatchers); | 625 EXPECT_EQ(1u, read_num_dispatchers); |
689 ASSERT_TRUE(read_dispatchers[0].get()); | 626 ASSERT_TRUE(read_dispatchers[0].get()); |
690 EXPECT_TRUE(read_dispatchers[0]->HasOneRef()); | 627 EXPECT_TRUE(read_dispatchers[0]->HasOneRef()); |
691 | 628 |
692 EXPECT_EQ(Dispatcher::kTypeMessagePipe, read_dispatchers[0]->GetType()); | 629 EXPECT_EQ(Dispatcher::kTypeMessagePipe, read_dispatchers[0]->GetType()); |
693 dispatcher = static_cast<MessagePipeDispatcher*>(read_dispatchers[0].get()); | 630 dispatcher = static_cast<MessagePipeDispatcher*>(read_dispatchers[0].get()); |
694 | 631 |
695 // Add the waiter now, before it becomes readable to avoid a race. | 632 // Add the waiter now, before it becomes readable to avoid a race. |
696 waiter.Init(); | 633 waiter.Init(); |
697 ASSERT_EQ(MOJO_RESULT_OK, | 634 ASSERT_EQ(MOJO_RESULT_OK, |
698 dispatcher->AddWaiter( | 635 dispatcher->AddWaiter(&waiter, MOJO_HANDLE_SIGNAL_READABLE, 456, |
699 &waiter, MOJO_HANDLE_SIGNAL_READABLE, 456, nullptr)); | 636 nullptr)); |
700 | 637 |
701 // Write to "local_mp", port 1. | 638 // Write to "local_mp", port 1. |
702 EXPECT_EQ(MOJO_RESULT_OK, | 639 EXPECT_EQ( |
703 local_mp->WriteMessage(1, | 640 MOJO_RESULT_OK, |
704 UserPointer<const void>(kHello), | 641 local_mp->WriteMessage(1, UserPointer<const void>(kHello), sizeof(kHello), |
705 sizeof(kHello), | 642 nullptr, MOJO_WRITE_MESSAGE_FLAG_NONE)); |
706 nullptr, | |
707 MOJO_WRITE_MESSAGE_FLAG_NONE)); | |
708 | 643 |
709 // TODO(vtl): FIXME -- We (racily) crash if I close |dispatcher| immediately | 644 // TODO(vtl): FIXME -- We (racily) crash if I close |dispatcher| immediately |
710 // here. (We don't crash if I sleep and then close.) | 645 // here. (We don't crash if I sleep and then close.) |
711 | 646 |
712 // Wait for the dispatcher to become readable. | 647 // Wait for the dispatcher to become readable. |
713 EXPECT_EQ(MOJO_RESULT_OK, waiter.Wait(MOJO_DEADLINE_INDEFINITE, &context)); | 648 EXPECT_EQ(MOJO_RESULT_OK, waiter.Wait(MOJO_DEADLINE_INDEFINITE, &context)); |
714 EXPECT_EQ(456u, context); | 649 EXPECT_EQ(456u, context); |
715 hss = HandleSignalsState(); | 650 hss = HandleSignalsState(); |
716 dispatcher->RemoveWaiter(&waiter, &hss); | 651 dispatcher->RemoveWaiter(&waiter, &hss); |
717 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, | 652 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, |
718 hss.satisfied_signals); | 653 hss.satisfied_signals); |
719 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, | 654 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, |
720 hss.satisfiable_signals); | 655 hss.satisfiable_signals); |
721 | 656 |
722 // Read from the dispatcher. | 657 // Read from the dispatcher. |
723 memset(read_buffer, 0, sizeof(read_buffer)); | 658 memset(read_buffer, 0, sizeof(read_buffer)); |
724 read_buffer_size = static_cast<uint32_t>(sizeof(read_buffer)); | 659 read_buffer_size = static_cast<uint32_t>(sizeof(read_buffer)); |
725 EXPECT_EQ(MOJO_RESULT_OK, | 660 EXPECT_EQ(MOJO_RESULT_OK, |
726 dispatcher->ReadMessage(UserPointer<void>(read_buffer), | 661 dispatcher->ReadMessage(UserPointer<void>(read_buffer), |
727 MakeUserPointer(&read_buffer_size), | 662 MakeUserPointer(&read_buffer_size), 0, |
728 0, | 663 nullptr, MOJO_READ_MESSAGE_FLAG_NONE)); |
729 nullptr, | |
730 MOJO_READ_MESSAGE_FLAG_NONE)); | |
731 EXPECT_EQ(sizeof(kHello), static_cast<size_t>(read_buffer_size)); | 664 EXPECT_EQ(sizeof(kHello), static_cast<size_t>(read_buffer_size)); |
732 EXPECT_STREQ(kHello, read_buffer); | 665 EXPECT_STREQ(kHello, read_buffer); |
733 | 666 |
734 // Prepare to wait on "local_mp", port 1. | 667 // Prepare to wait on "local_mp", port 1. |
735 waiter.Init(); | 668 waiter.Init(); |
736 ASSERT_EQ(MOJO_RESULT_OK, | 669 ASSERT_EQ(MOJO_RESULT_OK, |
737 local_mp->AddWaiter( | 670 local_mp->AddWaiter(1, &waiter, MOJO_HANDLE_SIGNAL_READABLE, 789, |
738 1, &waiter, MOJO_HANDLE_SIGNAL_READABLE, 789, nullptr)); | 671 nullptr)); |
739 | 672 |
740 // Write to the dispatcher. | 673 // Write to the dispatcher. |
741 EXPECT_EQ(MOJO_RESULT_OK, | 674 EXPECT_EQ(MOJO_RESULT_OK, dispatcher->WriteMessage( |
742 dispatcher->WriteMessage(UserPointer<const void>(kHello), | 675 UserPointer<const void>(kHello), sizeof(kHello), |
743 sizeof(kHello), | 676 nullptr, MOJO_WRITE_MESSAGE_FLAG_NONE)); |
744 nullptr, | |
745 MOJO_WRITE_MESSAGE_FLAG_NONE)); | |
746 | 677 |
747 // Wait. | 678 // Wait. |
748 EXPECT_EQ(MOJO_RESULT_OK, waiter.Wait(MOJO_DEADLINE_INDEFINITE, &context)); | 679 EXPECT_EQ(MOJO_RESULT_OK, waiter.Wait(MOJO_DEADLINE_INDEFINITE, &context)); |
749 EXPECT_EQ(789u, context); | 680 EXPECT_EQ(789u, context); |
750 hss = HandleSignalsState(); | 681 hss = HandleSignalsState(); |
751 local_mp->RemoveWaiter(1, &waiter, &hss); | 682 local_mp->RemoveWaiter(1, &waiter, &hss); |
752 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, | 683 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, |
753 hss.satisfied_signals); | 684 hss.satisfied_signals); |
754 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, | 685 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, |
755 hss.satisfiable_signals); | 686 hss.satisfiable_signals); |
756 | 687 |
757 // Read from "local_mp", port 1. | 688 // Read from "local_mp", port 1. |
758 memset(read_buffer, 0, sizeof(read_buffer)); | 689 memset(read_buffer, 0, sizeof(read_buffer)); |
759 read_buffer_size = static_cast<uint32_t>(sizeof(read_buffer)); | 690 read_buffer_size = static_cast<uint32_t>(sizeof(read_buffer)); |
760 EXPECT_EQ(MOJO_RESULT_OK, | 691 EXPECT_EQ(MOJO_RESULT_OK, |
761 local_mp->ReadMessage(1, | 692 local_mp->ReadMessage(1, UserPointer<void>(read_buffer), |
762 UserPointer<void>(read_buffer), | 693 MakeUserPointer(&read_buffer_size), nullptr, |
763 MakeUserPointer(&read_buffer_size), | 694 nullptr, MOJO_READ_MESSAGE_FLAG_NONE)); |
764 nullptr, | |
765 nullptr, | |
766 MOJO_READ_MESSAGE_FLAG_NONE)); | |
767 EXPECT_EQ(sizeof(kHello), static_cast<size_t>(read_buffer_size)); | 695 EXPECT_EQ(sizeof(kHello), static_cast<size_t>(read_buffer_size)); |
768 EXPECT_STREQ(kHello, read_buffer); | 696 EXPECT_STREQ(kHello, read_buffer); |
769 | 697 |
770 // TODO(vtl): Also test that messages queued up before the handle was sent are | 698 // TODO(vtl): Also test that messages queued up before the handle was sent are |
771 // delivered properly. | 699 // delivered properly. |
772 | 700 |
773 // Close everything that belongs to us. | 701 // Close everything that belongs to us. |
774 mp0->Close(0); | 702 mp0->Close(0); |
775 mp1->Close(1); | 703 mp1->Close(1); |
776 EXPECT_EQ(MOJO_RESULT_OK, dispatcher->Close()); | 704 EXPECT_EQ(MOJO_RESULT_OK, dispatcher->Close()); |
(...skipping 12 matching lines...) Expand all Loading... |
789 scoped_refptr<MessagePipeDispatcher> dispatcher( | 717 scoped_refptr<MessagePipeDispatcher> dispatcher( |
790 new MessagePipeDispatcher(MessagePipeDispatcher::kDefaultCreateOptions)); | 718 new MessagePipeDispatcher(MessagePipeDispatcher::kDefaultCreateOptions)); |
791 scoped_refptr<MessagePipe> local_mp(MessagePipe::CreateLocalLocal()); | 719 scoped_refptr<MessagePipe> local_mp(MessagePipe::CreateLocalLocal()); |
792 dispatcher->Init(local_mp, 0); | 720 dispatcher->Init(local_mp, 0); |
793 | 721 |
794 hss = local_mp->GetHandleSignalsState(0); | 722 hss = local_mp->GetHandleSignalsState(0); |
795 EXPECT_EQ(MOJO_HANDLE_SIGNAL_WRITABLE, hss.satisfied_signals); | 723 EXPECT_EQ(MOJO_HANDLE_SIGNAL_WRITABLE, hss.satisfied_signals); |
796 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, | 724 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, |
797 hss.satisfiable_signals); | 725 hss.satisfiable_signals); |
798 // Write to the other end (|local_mp|, port 1), and then close it. | 726 // Write to the other end (|local_mp|, port 1), and then close it. |
799 EXPECT_EQ(MOJO_RESULT_OK, | 727 EXPECT_EQ( |
800 local_mp->WriteMessage(1, | 728 MOJO_RESULT_OK, |
801 UserPointer<const void>(kHello), | 729 local_mp->WriteMessage(1, UserPointer<const void>(kHello), sizeof(kHello), |
802 sizeof(kHello), | 730 nullptr, MOJO_WRITE_MESSAGE_FLAG_NONE)); |
803 nullptr, | |
804 MOJO_WRITE_MESSAGE_FLAG_NONE)); | |
805 hss = local_mp->GetHandleSignalsState(0); | 731 hss = local_mp->GetHandleSignalsState(0); |
806 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, | 732 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, |
807 hss.satisfied_signals); | 733 hss.satisfied_signals); |
808 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, | 734 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, |
809 hss.satisfiable_signals); | 735 hss.satisfiable_signals); |
810 // Then the second message.... | 736 // Then the second message.... |
811 EXPECT_EQ(MOJO_RESULT_OK, | 737 EXPECT_EQ( |
812 local_mp->WriteMessage(1, | 738 MOJO_RESULT_OK, |
813 UserPointer<const void>(kWorld), | 739 local_mp->WriteMessage(1, UserPointer<const void>(kWorld), sizeof(kWorld), |
814 sizeof(kWorld), | 740 nullptr, MOJO_WRITE_MESSAGE_FLAG_NONE)); |
815 nullptr, | |
816 MOJO_WRITE_MESSAGE_FLAG_NONE)); | |
817 hss = local_mp->GetHandleSignalsState(0); | 741 hss = local_mp->GetHandleSignalsState(0); |
818 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, | 742 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, |
819 hss.satisfied_signals); | 743 hss.satisfied_signals); |
820 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, | 744 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, |
821 hss.satisfiable_signals); | 745 hss.satisfiable_signals); |
822 // Then close it. | 746 // Then close it. |
823 local_mp->Close(1); | 747 local_mp->Close(1); |
824 | 748 |
825 scoped_refptr<ChannelEndpoint> ep0; | 749 scoped_refptr<ChannelEndpoint> ep0; |
826 scoped_refptr<MessagePipe> mp0(MessagePipe::CreateLocalProxy(&ep0)); | 750 scoped_refptr<MessagePipe> mp0(MessagePipe::CreateLocalProxy(&ep0)); |
827 scoped_refptr<ChannelEndpoint> ep1; | 751 scoped_refptr<ChannelEndpoint> ep1; |
828 scoped_refptr<MessagePipe> mp1(MessagePipe::CreateProxyLocal(&ep1)); | 752 scoped_refptr<MessagePipe> mp1(MessagePipe::CreateProxyLocal(&ep1)); |
829 BootstrapChannelEndpoints(ep0, ep1); | 753 BootstrapChannelEndpoints(ep0, ep1); |
830 | 754 |
831 // Prepare to wait on MP 1, port 1. (Add the waiter now. Otherwise, if we do | 755 // Prepare to wait on MP 1, port 1. (Add the waiter now. Otherwise, if we do |
832 // it later, it might already be readable.) | 756 // it later, it might already be readable.) |
833 waiter.Init(); | 757 waiter.Init(); |
834 ASSERT_EQ( | 758 ASSERT_EQ( |
835 MOJO_RESULT_OK, | 759 MOJO_RESULT_OK, |
836 mp1->AddWaiter(1, &waiter, MOJO_HANDLE_SIGNAL_READABLE, 123, nullptr)); | 760 mp1->AddWaiter(1, &waiter, MOJO_HANDLE_SIGNAL_READABLE, 123, nullptr)); |
837 | 761 |
838 // Write to MP 0, port 0. | 762 // Write to MP 0, port 0. |
839 { | 763 { |
840 DispatcherTransport transport( | 764 DispatcherTransport transport( |
841 test::DispatcherTryStartTransport(dispatcher.get())); | 765 test::DispatcherTryStartTransport(dispatcher.get())); |
842 EXPECT_TRUE(transport.is_valid()); | 766 EXPECT_TRUE(transport.is_valid()); |
843 | 767 |
844 std::vector<DispatcherTransport> transports; | 768 std::vector<DispatcherTransport> transports; |
845 transports.push_back(transport); | 769 transports.push_back(transport); |
846 EXPECT_EQ(MOJO_RESULT_OK, | 770 EXPECT_EQ( |
847 mp0->WriteMessage(0, | 771 MOJO_RESULT_OK, |
848 UserPointer<const void>(kHello), | 772 mp0->WriteMessage(0, UserPointer<const void>(kHello), sizeof(kHello), |
849 sizeof(kHello), | 773 &transports, MOJO_WRITE_MESSAGE_FLAG_NONE)); |
850 &transports, | |
851 MOJO_WRITE_MESSAGE_FLAG_NONE)); | |
852 transport.End(); | 774 transport.End(); |
853 | 775 |
854 // |dispatcher| should have been closed. This is |DCHECK()|ed when the | 776 // |dispatcher| should have been closed. This is |DCHECK()|ed when the |
855 // |dispatcher| is destroyed. | 777 // |dispatcher| is destroyed. |
856 EXPECT_TRUE(dispatcher->HasOneRef()); | 778 EXPECT_TRUE(dispatcher->HasOneRef()); |
857 dispatcher = nullptr; | 779 dispatcher = nullptr; |
858 } | 780 } |
859 | 781 |
860 // Wait. | 782 // Wait. |
861 EXPECT_EQ(MOJO_RESULT_OK, waiter.Wait(MOJO_DEADLINE_INDEFINITE, &context)); | 783 EXPECT_EQ(MOJO_RESULT_OK, waiter.Wait(MOJO_DEADLINE_INDEFINITE, &context)); |
862 EXPECT_EQ(123u, context); | 784 EXPECT_EQ(123u, context); |
863 hss = HandleSignalsState(); | 785 hss = HandleSignalsState(); |
864 mp1->RemoveWaiter(1, &waiter, &hss); | 786 mp1->RemoveWaiter(1, &waiter, &hss); |
865 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, | 787 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, |
866 hss.satisfied_signals); | 788 hss.satisfied_signals); |
867 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, | 789 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, |
868 hss.satisfiable_signals); | 790 hss.satisfiable_signals); |
869 | 791 |
870 // Read from MP 1, port 1. | 792 // Read from MP 1, port 1. |
871 char read_buffer[100] = {0}; | 793 char read_buffer[100] = {0}; |
872 uint32_t read_buffer_size = static_cast<uint32_t>(sizeof(read_buffer)); | 794 uint32_t read_buffer_size = static_cast<uint32_t>(sizeof(read_buffer)); |
873 DispatcherVector read_dispatchers; | 795 DispatcherVector read_dispatchers; |
874 uint32_t read_num_dispatchers = 10; // Maximum to get. | 796 uint32_t read_num_dispatchers = 10; // Maximum to get. |
875 EXPECT_EQ(MOJO_RESULT_OK, | 797 EXPECT_EQ( |
876 mp1->ReadMessage(1, | 798 MOJO_RESULT_OK, |
877 UserPointer<void>(read_buffer), | 799 mp1->ReadMessage(1, UserPointer<void>(read_buffer), |
878 MakeUserPointer(&read_buffer_size), | 800 MakeUserPointer(&read_buffer_size), &read_dispatchers, |
879 &read_dispatchers, | 801 &read_num_dispatchers, MOJO_READ_MESSAGE_FLAG_NONE)); |
880 &read_num_dispatchers, | |
881 MOJO_READ_MESSAGE_FLAG_NONE)); | |
882 EXPECT_EQ(sizeof(kHello), static_cast<size_t>(read_buffer_size)); | 802 EXPECT_EQ(sizeof(kHello), static_cast<size_t>(read_buffer_size)); |
883 EXPECT_STREQ(kHello, read_buffer); | 803 EXPECT_STREQ(kHello, read_buffer); |
884 EXPECT_EQ(1u, read_dispatchers.size()); | 804 EXPECT_EQ(1u, read_dispatchers.size()); |
885 EXPECT_EQ(1u, read_num_dispatchers); | 805 EXPECT_EQ(1u, read_num_dispatchers); |
886 ASSERT_TRUE(read_dispatchers[0].get()); | 806 ASSERT_TRUE(read_dispatchers[0].get()); |
887 EXPECT_TRUE(read_dispatchers[0]->HasOneRef()); | 807 EXPECT_TRUE(read_dispatchers[0]->HasOneRef()); |
888 | 808 |
889 EXPECT_EQ(Dispatcher::kTypeMessagePipe, read_dispatchers[0]->GetType()); | 809 EXPECT_EQ(Dispatcher::kTypeMessagePipe, read_dispatchers[0]->GetType()); |
890 dispatcher = static_cast<MessagePipeDispatcher*>(read_dispatchers[0].get()); | 810 dispatcher = static_cast<MessagePipeDispatcher*>(read_dispatchers[0].get()); |
891 | 811 |
892 // |dispatcher| should already be readable and not writable. | 812 // |dispatcher| should already be readable and not writable. |
893 hss = dispatcher->GetHandleSignalsState(); | 813 hss = dispatcher->GetHandleSignalsState(); |
894 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE, hss.satisfied_signals); | 814 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE, hss.satisfied_signals); |
895 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE, hss.satisfiable_signals); | 815 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE, hss.satisfiable_signals); |
896 // So read from it. | 816 // So read from it. |
897 memset(read_buffer, 0, sizeof(read_buffer)); | 817 memset(read_buffer, 0, sizeof(read_buffer)); |
898 read_buffer_size = static_cast<uint32_t>(sizeof(read_buffer)); | 818 read_buffer_size = static_cast<uint32_t>(sizeof(read_buffer)); |
899 EXPECT_EQ(MOJO_RESULT_OK, | 819 EXPECT_EQ(MOJO_RESULT_OK, |
900 dispatcher->ReadMessage(UserPointer<void>(read_buffer), | 820 dispatcher->ReadMessage(UserPointer<void>(read_buffer), |
901 MakeUserPointer(&read_buffer_size), | 821 MakeUserPointer(&read_buffer_size), 0, |
902 0, | 822 nullptr, MOJO_READ_MESSAGE_FLAG_NONE)); |
903 nullptr, | |
904 MOJO_READ_MESSAGE_FLAG_NONE)); | |
905 EXPECT_EQ(sizeof(kHello), static_cast<size_t>(read_buffer_size)); | 823 EXPECT_EQ(sizeof(kHello), static_cast<size_t>(read_buffer_size)); |
906 EXPECT_STREQ(kHello, read_buffer); | 824 EXPECT_STREQ(kHello, read_buffer); |
907 // It should still be readable. | 825 // It should still be readable. |
908 hss = dispatcher->GetHandleSignalsState(); | 826 hss = dispatcher->GetHandleSignalsState(); |
909 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE, hss.satisfied_signals); | 827 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE, hss.satisfied_signals); |
910 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE, hss.satisfiable_signals); | 828 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE, hss.satisfiable_signals); |
911 // So read from it. | 829 // So read from it. |
912 memset(read_buffer, 0, sizeof(read_buffer)); | 830 memset(read_buffer, 0, sizeof(read_buffer)); |
913 read_buffer_size = static_cast<uint32_t>(sizeof(read_buffer)); | 831 read_buffer_size = static_cast<uint32_t>(sizeof(read_buffer)); |
914 EXPECT_EQ(MOJO_RESULT_OK, | 832 EXPECT_EQ(MOJO_RESULT_OK, |
915 dispatcher->ReadMessage(UserPointer<void>(read_buffer), | 833 dispatcher->ReadMessage(UserPointer<void>(read_buffer), |
916 MakeUserPointer(&read_buffer_size), | 834 MakeUserPointer(&read_buffer_size), 0, |
917 0, | 835 nullptr, MOJO_READ_MESSAGE_FLAG_NONE)); |
918 nullptr, | |
919 MOJO_READ_MESSAGE_FLAG_NONE)); | |
920 EXPECT_EQ(sizeof(kWorld), static_cast<size_t>(read_buffer_size)); | 836 EXPECT_EQ(sizeof(kWorld), static_cast<size_t>(read_buffer_size)); |
921 EXPECT_STREQ(kWorld, read_buffer); | 837 EXPECT_STREQ(kWorld, read_buffer); |
922 // Now it should no longer be readable. | 838 // Now it should no longer be readable. |
923 hss = dispatcher->GetHandleSignalsState(); | 839 hss = dispatcher->GetHandleSignalsState(); |
924 EXPECT_EQ(MOJO_HANDLE_SIGNAL_NONE, hss.satisfied_signals); | 840 EXPECT_EQ(MOJO_HANDLE_SIGNAL_NONE, hss.satisfied_signals); |
925 EXPECT_EQ(MOJO_HANDLE_SIGNAL_NONE, hss.satisfiable_signals); | 841 EXPECT_EQ(MOJO_HANDLE_SIGNAL_NONE, hss.satisfiable_signals); |
926 | 842 |
927 // Close everything that belongs to us. | 843 // Close everything that belongs to us. |
928 mp0->Close(0); | 844 mp0->Close(0); |
929 mp1->Close(1); | 845 mp1->Close(1); |
(...skipping 13 matching lines...) Expand all Loading... |
943 uint32_t context = 0; | 859 uint32_t context = 0; |
944 | 860 |
945 scoped_refptr<ChannelEndpoint> ep0; | 861 scoped_refptr<ChannelEndpoint> ep0; |
946 scoped_refptr<MessagePipe> mp0(MessagePipe::CreateLocalProxy(&ep0)); | 862 scoped_refptr<MessagePipe> mp0(MessagePipe::CreateLocalProxy(&ep0)); |
947 scoped_refptr<ChannelEndpoint> ep1; | 863 scoped_refptr<ChannelEndpoint> ep1; |
948 scoped_refptr<MessagePipe> mp1(MessagePipe::CreateProxyLocal(&ep1)); | 864 scoped_refptr<MessagePipe> mp1(MessagePipe::CreateProxyLocal(&ep1)); |
949 BootstrapChannelEndpoints(ep0, ep1); | 865 BootstrapChannelEndpoints(ep0, ep1); |
950 | 866 |
951 // We'll try to pass this dispatcher. | 867 // We'll try to pass this dispatcher. |
952 scoped_refptr<SharedBufferDispatcher> dispatcher; | 868 scoped_refptr<SharedBufferDispatcher> dispatcher; |
953 EXPECT_EQ(MOJO_RESULT_OK, | 869 EXPECT_EQ(MOJO_RESULT_OK, SharedBufferDispatcher::Create( |
954 SharedBufferDispatcher::Create( | 870 platform_support(), |
955 platform_support(), | 871 SharedBufferDispatcher::kDefaultCreateOptions, |
956 SharedBufferDispatcher::kDefaultCreateOptions, | 872 100, &dispatcher)); |
957 100, | |
958 &dispatcher)); | |
959 ASSERT_TRUE(dispatcher.get()); | 873 ASSERT_TRUE(dispatcher.get()); |
960 | 874 |
961 // Make a mapping. | 875 // Make a mapping. |
962 scoped_ptr<embedder::PlatformSharedBufferMapping> mapping0; | 876 scoped_ptr<embedder::PlatformSharedBufferMapping> mapping0; |
963 EXPECT_EQ( | 877 EXPECT_EQ(MOJO_RESULT_OK, dispatcher->MapBuffer( |
964 MOJO_RESULT_OK, | 878 0, 100, MOJO_MAP_BUFFER_FLAG_NONE, &mapping0)); |
965 dispatcher->MapBuffer(0, 100, MOJO_MAP_BUFFER_FLAG_NONE, &mapping0)); | |
966 ASSERT_TRUE(mapping0); | 879 ASSERT_TRUE(mapping0); |
967 ASSERT_TRUE(mapping0->GetBase()); | 880 ASSERT_TRUE(mapping0->GetBase()); |
968 ASSERT_EQ(100u, mapping0->GetLength()); | 881 ASSERT_EQ(100u, mapping0->GetLength()); |
969 static_cast<char*>(mapping0->GetBase())[0] = 'A'; | 882 static_cast<char*>(mapping0->GetBase())[0] = 'A'; |
970 static_cast<char*>(mapping0->GetBase())[50] = 'B'; | 883 static_cast<char*>(mapping0->GetBase())[50] = 'B'; |
971 static_cast<char*>(mapping0->GetBase())[99] = 'C'; | 884 static_cast<char*>(mapping0->GetBase())[99] = 'C'; |
972 | 885 |
973 // Prepare to wait on MP 1, port 1. (Add the waiter now. Otherwise, if we do | 886 // Prepare to wait on MP 1, port 1. (Add the waiter now. Otherwise, if we do |
974 // it later, it might already be readable.) | 887 // it later, it might already be readable.) |
975 waiter.Init(); | 888 waiter.Init(); |
976 ASSERT_EQ( | 889 ASSERT_EQ( |
977 MOJO_RESULT_OK, | 890 MOJO_RESULT_OK, |
978 mp1->AddWaiter(1, &waiter, MOJO_HANDLE_SIGNAL_READABLE, 123, nullptr)); | 891 mp1->AddWaiter(1, &waiter, MOJO_HANDLE_SIGNAL_READABLE, 123, nullptr)); |
979 | 892 |
980 // Write to MP 0, port 0. | 893 // Write to MP 0, port 0. |
981 { | 894 { |
982 DispatcherTransport transport( | 895 DispatcherTransport transport( |
983 test::DispatcherTryStartTransport(dispatcher.get())); | 896 test::DispatcherTryStartTransport(dispatcher.get())); |
984 EXPECT_TRUE(transport.is_valid()); | 897 EXPECT_TRUE(transport.is_valid()); |
985 | 898 |
986 std::vector<DispatcherTransport> transports; | 899 std::vector<DispatcherTransport> transports; |
987 transports.push_back(transport); | 900 transports.push_back(transport); |
988 EXPECT_EQ(MOJO_RESULT_OK, | 901 EXPECT_EQ( |
989 mp0->WriteMessage(0, | 902 MOJO_RESULT_OK, |
990 UserPointer<const void>(kHello), | 903 mp0->WriteMessage(0, UserPointer<const void>(kHello), sizeof(kHello), |
991 sizeof(kHello), | 904 &transports, MOJO_WRITE_MESSAGE_FLAG_NONE)); |
992 &transports, | |
993 MOJO_WRITE_MESSAGE_FLAG_NONE)); | |
994 transport.End(); | 905 transport.End(); |
995 | 906 |
996 // |dispatcher| should have been closed. This is |DCHECK()|ed when the | 907 // |dispatcher| should have been closed. This is |DCHECK()|ed when the |
997 // |dispatcher| is destroyed. | 908 // |dispatcher| is destroyed. |
998 EXPECT_TRUE(dispatcher->HasOneRef()); | 909 EXPECT_TRUE(dispatcher->HasOneRef()); |
999 dispatcher = nullptr; | 910 dispatcher = nullptr; |
1000 } | 911 } |
1001 | 912 |
1002 // Wait. | 913 // Wait. |
1003 EXPECT_EQ(MOJO_RESULT_OK, waiter.Wait(MOJO_DEADLINE_INDEFINITE, &context)); | 914 EXPECT_EQ(MOJO_RESULT_OK, waiter.Wait(MOJO_DEADLINE_INDEFINITE, &context)); |
1004 EXPECT_EQ(123u, context); | 915 EXPECT_EQ(123u, context); |
1005 hss = HandleSignalsState(); | 916 hss = HandleSignalsState(); |
1006 mp1->RemoveWaiter(1, &waiter, &hss); | 917 mp1->RemoveWaiter(1, &waiter, &hss); |
1007 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, | 918 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, |
1008 hss.satisfied_signals); | 919 hss.satisfied_signals); |
1009 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, | 920 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, |
1010 hss.satisfiable_signals); | 921 hss.satisfiable_signals); |
1011 | 922 |
1012 // Read from MP 1, port 1. | 923 // Read from MP 1, port 1. |
1013 char read_buffer[100] = {0}; | 924 char read_buffer[100] = {0}; |
1014 uint32_t read_buffer_size = static_cast<uint32_t>(sizeof(read_buffer)); | 925 uint32_t read_buffer_size = static_cast<uint32_t>(sizeof(read_buffer)); |
1015 DispatcherVector read_dispatchers; | 926 DispatcherVector read_dispatchers; |
1016 uint32_t read_num_dispatchers = 10; // Maximum to get. | 927 uint32_t read_num_dispatchers = 10; // Maximum to get. |
1017 EXPECT_EQ(MOJO_RESULT_OK, | 928 EXPECT_EQ( |
1018 mp1->ReadMessage(1, | 929 MOJO_RESULT_OK, |
1019 UserPointer<void>(read_buffer), | 930 mp1->ReadMessage(1, UserPointer<void>(read_buffer), |
1020 MakeUserPointer(&read_buffer_size), | 931 MakeUserPointer(&read_buffer_size), &read_dispatchers, |
1021 &read_dispatchers, | 932 &read_num_dispatchers, MOJO_READ_MESSAGE_FLAG_NONE)); |
1022 &read_num_dispatchers, | |
1023 MOJO_READ_MESSAGE_FLAG_NONE)); | |
1024 EXPECT_EQ(sizeof(kHello), static_cast<size_t>(read_buffer_size)); | 933 EXPECT_EQ(sizeof(kHello), static_cast<size_t>(read_buffer_size)); |
1025 EXPECT_STREQ(kHello, read_buffer); | 934 EXPECT_STREQ(kHello, read_buffer); |
1026 EXPECT_EQ(1u, read_dispatchers.size()); | 935 EXPECT_EQ(1u, read_dispatchers.size()); |
1027 EXPECT_EQ(1u, read_num_dispatchers); | 936 EXPECT_EQ(1u, read_num_dispatchers); |
1028 ASSERT_TRUE(read_dispatchers[0].get()); | 937 ASSERT_TRUE(read_dispatchers[0].get()); |
1029 EXPECT_TRUE(read_dispatchers[0]->HasOneRef()); | 938 EXPECT_TRUE(read_dispatchers[0]->HasOneRef()); |
1030 | 939 |
1031 EXPECT_EQ(Dispatcher::kTypeSharedBuffer, read_dispatchers[0]->GetType()); | 940 EXPECT_EQ(Dispatcher::kTypeSharedBuffer, read_dispatchers[0]->GetType()); |
1032 dispatcher = static_cast<SharedBufferDispatcher*>(read_dispatchers[0].get()); | 941 dispatcher = static_cast<SharedBufferDispatcher*>(read_dispatchers[0].get()); |
1033 | 942 |
1034 // Make another mapping. | 943 // Make another mapping. |
1035 scoped_ptr<embedder::PlatformSharedBufferMapping> mapping1; | 944 scoped_ptr<embedder::PlatformSharedBufferMapping> mapping1; |
1036 EXPECT_EQ( | 945 EXPECT_EQ(MOJO_RESULT_OK, dispatcher->MapBuffer( |
1037 MOJO_RESULT_OK, | 946 0, 100, MOJO_MAP_BUFFER_FLAG_NONE, &mapping1)); |
1038 dispatcher->MapBuffer(0, 100, MOJO_MAP_BUFFER_FLAG_NONE, &mapping1)); | |
1039 ASSERT_TRUE(mapping1); | 947 ASSERT_TRUE(mapping1); |
1040 ASSERT_TRUE(mapping1->GetBase()); | 948 ASSERT_TRUE(mapping1->GetBase()); |
1041 ASSERT_EQ(100u, mapping1->GetLength()); | 949 ASSERT_EQ(100u, mapping1->GetLength()); |
1042 EXPECT_NE(mapping1->GetBase(), mapping0->GetBase()); | 950 EXPECT_NE(mapping1->GetBase(), mapping0->GetBase()); |
1043 EXPECT_EQ('A', static_cast<char*>(mapping1->GetBase())[0]); | 951 EXPECT_EQ('A', static_cast<char*>(mapping1->GetBase())[0]); |
1044 EXPECT_EQ('B', static_cast<char*>(mapping1->GetBase())[50]); | 952 EXPECT_EQ('B', static_cast<char*>(mapping1->GetBase())[50]); |
1045 EXPECT_EQ('C', static_cast<char*>(mapping1->GetBase())[99]); | 953 EXPECT_EQ('C', static_cast<char*>(mapping1->GetBase())[99]); |
1046 | 954 |
1047 // Write stuff either way. | 955 // Write stuff either way. |
1048 static_cast<char*>(mapping1->GetBase())[1] = 'x'; | 956 static_cast<char*>(mapping1->GetBase())[1] = 'x'; |
(...skipping 54 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
1103 mp1->AddWaiter(1, &waiter, MOJO_HANDLE_SIGNAL_READABLE, 123, nullptr)); | 1011 mp1->AddWaiter(1, &waiter, MOJO_HANDLE_SIGNAL_READABLE, 123, nullptr)); |
1104 | 1012 |
1105 // Write to MP 0, port 0. | 1013 // Write to MP 0, port 0. |
1106 { | 1014 { |
1107 DispatcherTransport transport( | 1015 DispatcherTransport transport( |
1108 test::DispatcherTryStartTransport(dispatcher.get())); | 1016 test::DispatcherTryStartTransport(dispatcher.get())); |
1109 EXPECT_TRUE(transport.is_valid()); | 1017 EXPECT_TRUE(transport.is_valid()); |
1110 | 1018 |
1111 std::vector<DispatcherTransport> transports; | 1019 std::vector<DispatcherTransport> transports; |
1112 transports.push_back(transport); | 1020 transports.push_back(transport); |
1113 EXPECT_EQ(MOJO_RESULT_OK, | 1021 EXPECT_EQ( |
1114 mp0->WriteMessage(0, | 1022 MOJO_RESULT_OK, |
1115 UserPointer<const void>(kWorld), | 1023 mp0->WriteMessage(0, UserPointer<const void>(kWorld), sizeof(kWorld), |
1116 sizeof(kWorld), | 1024 &transports, MOJO_WRITE_MESSAGE_FLAG_NONE)); |
1117 &transports, | |
1118 MOJO_WRITE_MESSAGE_FLAG_NONE)); | |
1119 transport.End(); | 1025 transport.End(); |
1120 | 1026 |
1121 // |dispatcher| should have been closed. This is |DCHECK()|ed when the | 1027 // |dispatcher| should have been closed. This is |DCHECK()|ed when the |
1122 // |dispatcher| is destroyed. | 1028 // |dispatcher| is destroyed. |
1123 EXPECT_TRUE(dispatcher->HasOneRef()); | 1029 EXPECT_TRUE(dispatcher->HasOneRef()); |
1124 dispatcher = nullptr; | 1030 dispatcher = nullptr; |
1125 } | 1031 } |
1126 | 1032 |
1127 // Wait. | 1033 // Wait. |
1128 EXPECT_EQ(MOJO_RESULT_OK, waiter.Wait(MOJO_DEADLINE_INDEFINITE, &context)); | 1034 EXPECT_EQ(MOJO_RESULT_OK, waiter.Wait(MOJO_DEADLINE_INDEFINITE, &context)); |
1129 EXPECT_EQ(123u, context); | 1035 EXPECT_EQ(123u, context); |
1130 hss = HandleSignalsState(); | 1036 hss = HandleSignalsState(); |
1131 mp1->RemoveWaiter(1, &waiter, &hss); | 1037 mp1->RemoveWaiter(1, &waiter, &hss); |
1132 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, | 1038 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, |
1133 hss.satisfied_signals); | 1039 hss.satisfied_signals); |
1134 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, | 1040 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, |
1135 hss.satisfiable_signals); | 1041 hss.satisfiable_signals); |
1136 | 1042 |
1137 // Read from MP 1, port 1. | 1043 // Read from MP 1, port 1. |
1138 char read_buffer[100] = {0}; | 1044 char read_buffer[100] = {0}; |
1139 uint32_t read_buffer_size = static_cast<uint32_t>(sizeof(read_buffer)); | 1045 uint32_t read_buffer_size = static_cast<uint32_t>(sizeof(read_buffer)); |
1140 DispatcherVector read_dispatchers; | 1046 DispatcherVector read_dispatchers; |
1141 uint32_t read_num_dispatchers = 10; // Maximum to get. | 1047 uint32_t read_num_dispatchers = 10; // Maximum to get. |
1142 EXPECT_EQ(MOJO_RESULT_OK, | 1048 EXPECT_EQ( |
1143 mp1->ReadMessage(1, | 1049 MOJO_RESULT_OK, |
1144 UserPointer<void>(read_buffer), | 1050 mp1->ReadMessage(1, UserPointer<void>(read_buffer), |
1145 MakeUserPointer(&read_buffer_size), | 1051 MakeUserPointer(&read_buffer_size), &read_dispatchers, |
1146 &read_dispatchers, | 1052 &read_num_dispatchers, MOJO_READ_MESSAGE_FLAG_NONE)); |
1147 &read_num_dispatchers, | |
1148 MOJO_READ_MESSAGE_FLAG_NONE)); | |
1149 EXPECT_EQ(sizeof(kWorld), static_cast<size_t>(read_buffer_size)); | 1053 EXPECT_EQ(sizeof(kWorld), static_cast<size_t>(read_buffer_size)); |
1150 EXPECT_STREQ(kWorld, read_buffer); | 1054 EXPECT_STREQ(kWorld, read_buffer); |
1151 EXPECT_EQ(1u, read_dispatchers.size()); | 1055 EXPECT_EQ(1u, read_dispatchers.size()); |
1152 EXPECT_EQ(1u, read_num_dispatchers); | 1056 EXPECT_EQ(1u, read_num_dispatchers); |
1153 ASSERT_TRUE(read_dispatchers[0].get()); | 1057 ASSERT_TRUE(read_dispatchers[0].get()); |
1154 EXPECT_TRUE(read_dispatchers[0]->HasOneRef()); | 1058 EXPECT_TRUE(read_dispatchers[0]->HasOneRef()); |
1155 | 1059 |
1156 EXPECT_EQ(Dispatcher::kTypePlatformHandle, read_dispatchers[0]->GetType()); | 1060 EXPECT_EQ(Dispatcher::kTypePlatformHandle, read_dispatchers[0]->GetType()); |
1157 dispatcher = | 1061 dispatcher = |
1158 static_cast<PlatformHandleDispatcher*>(read_dispatchers[0].get()); | 1062 static_cast<PlatformHandleDispatcher*>(read_dispatchers[0].get()); |
(...skipping 86 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
1245 mp1->AddWaiter(1, &waiter, MOJO_HANDLE_SIGNAL_READABLE, 123, nullptr)); | 1149 mp1->AddWaiter(1, &waiter, MOJO_HANDLE_SIGNAL_READABLE, 123, nullptr)); |
1246 | 1150 |
1247 // Write to MP 0, port 0. | 1151 // Write to MP 0, port 0. |
1248 { | 1152 { |
1249 DispatcherTransport transport( | 1153 DispatcherTransport transport( |
1250 test::DispatcherTryStartTransport(dispatcher.get())); | 1154 test::DispatcherTryStartTransport(dispatcher.get())); |
1251 EXPECT_TRUE(transport.is_valid()); | 1155 EXPECT_TRUE(transport.is_valid()); |
1252 | 1156 |
1253 std::vector<DispatcherTransport> transports; | 1157 std::vector<DispatcherTransport> transports; |
1254 transports.push_back(transport); | 1158 transports.push_back(transport); |
1255 EXPECT_EQ(MOJO_RESULT_OK, | 1159 EXPECT_EQ( |
1256 mp0->WriteMessage(0, | 1160 MOJO_RESULT_OK, |
1257 UserPointer<const void>(kHello), | 1161 mp0->WriteMessage(0, UserPointer<const void>(kHello), sizeof(kHello), |
1258 sizeof(kHello), | 1162 &transports, MOJO_WRITE_MESSAGE_FLAG_NONE)); |
1259 &transports, | |
1260 MOJO_WRITE_MESSAGE_FLAG_NONE)); | |
1261 transport.End(); | 1163 transport.End(); |
1262 | 1164 |
1263 // |dispatcher| should have been closed. This is |DCHECK()|ed when the | 1165 // |dispatcher| should have been closed. This is |DCHECK()|ed when the |
1264 // |dispatcher| is destroyed. | 1166 // |dispatcher| is destroyed. |
1265 EXPECT_TRUE(dispatcher->HasOneRef()); | 1167 EXPECT_TRUE(dispatcher->HasOneRef()); |
1266 dispatcher = nullptr; | 1168 dispatcher = nullptr; |
1267 } | 1169 } |
1268 | 1170 |
1269 // Wait. | 1171 // Wait. |
1270 EXPECT_EQ(MOJO_RESULT_OK, waiter.Wait(MOJO_DEADLINE_INDEFINITE, &context)); | 1172 EXPECT_EQ(MOJO_RESULT_OK, waiter.Wait(MOJO_DEADLINE_INDEFINITE, &context)); |
1271 EXPECT_EQ(123u, context); | 1173 EXPECT_EQ(123u, context); |
1272 hss = HandleSignalsState(); | 1174 hss = HandleSignalsState(); |
1273 mp1->RemoveWaiter(1, &waiter, &hss); | 1175 mp1->RemoveWaiter(1, &waiter, &hss); |
1274 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, | 1176 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, |
1275 hss.satisfied_signals); | 1177 hss.satisfied_signals); |
1276 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, | 1178 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, |
1277 hss.satisfiable_signals); | 1179 hss.satisfiable_signals); |
1278 | 1180 |
1279 // Read from MP 1, port 1. | 1181 // Read from MP 1, port 1. |
1280 char read_buffer[100] = {0}; | 1182 char read_buffer[100] = {0}; |
1281 uint32_t read_buffer_size = static_cast<uint32_t>(sizeof(read_buffer)); | 1183 uint32_t read_buffer_size = static_cast<uint32_t>(sizeof(read_buffer)); |
1282 DispatcherVector read_dispatchers; | 1184 DispatcherVector read_dispatchers; |
1283 uint32_t read_num_dispatchers = 10; // Maximum to get. | 1185 uint32_t read_num_dispatchers = 10; // Maximum to get. |
1284 EXPECT_EQ(MOJO_RESULT_OK, | 1186 EXPECT_EQ( |
1285 mp1->ReadMessage(1, | 1187 MOJO_RESULT_OK, |
1286 UserPointer<void>(read_buffer), | 1188 mp1->ReadMessage(1, UserPointer<void>(read_buffer), |
1287 MakeUserPointer(&read_buffer_size), | 1189 MakeUserPointer(&read_buffer_size), &read_dispatchers, |
1288 &read_dispatchers, | 1190 &read_num_dispatchers, MOJO_READ_MESSAGE_FLAG_NONE)); |
1289 &read_num_dispatchers, | |
1290 MOJO_READ_MESSAGE_FLAG_NONE)); | |
1291 EXPECT_EQ(sizeof(kHello), static_cast<size_t>(read_buffer_size)); | 1191 EXPECT_EQ(sizeof(kHello), static_cast<size_t>(read_buffer_size)); |
1292 EXPECT_STREQ(kHello, read_buffer); | 1192 EXPECT_STREQ(kHello, read_buffer); |
1293 EXPECT_EQ(1u, read_dispatchers.size()); | 1193 EXPECT_EQ(1u, read_dispatchers.size()); |
1294 EXPECT_EQ(1u, read_num_dispatchers); | 1194 EXPECT_EQ(1u, read_num_dispatchers); |
1295 ASSERT_TRUE(read_dispatchers[0].get()); | 1195 ASSERT_TRUE(read_dispatchers[0].get()); |
1296 EXPECT_TRUE(read_dispatchers[0]->HasOneRef()); | 1196 EXPECT_TRUE(read_dispatchers[0]->HasOneRef()); |
1297 | 1197 |
1298 EXPECT_EQ(Dispatcher::kTypeMessagePipe, read_dispatchers[0]->GetType()); | 1198 EXPECT_EQ(Dispatcher::kTypeMessagePipe, read_dispatchers[0]->GetType()); |
1299 dispatcher = static_cast<MessagePipeDispatcher*>(read_dispatchers[0].get()); | 1199 dispatcher = static_cast<MessagePipeDispatcher*>(read_dispatchers[0].get()); |
1300 read_dispatchers.clear(); | 1200 read_dispatchers.clear(); |
1301 | 1201 |
1302 // Now pass it back. | 1202 // Now pass it back. |
1303 | 1203 |
1304 // Prepare to wait on MP 0, port 0. (Add the waiter now. Otherwise, if we do | 1204 // Prepare to wait on MP 0, port 0. (Add the waiter now. Otherwise, if we do |
1305 // it later, it might already be readable.) | 1205 // it later, it might already be readable.) |
1306 waiter.Init(); | 1206 waiter.Init(); |
1307 ASSERT_EQ( | 1207 ASSERT_EQ( |
1308 MOJO_RESULT_OK, | 1208 MOJO_RESULT_OK, |
1309 mp0->AddWaiter(0, &waiter, MOJO_HANDLE_SIGNAL_READABLE, 456, nullptr)); | 1209 mp0->AddWaiter(0, &waiter, MOJO_HANDLE_SIGNAL_READABLE, 456, nullptr)); |
1310 | 1210 |
1311 // Write to MP 1, port 1. | 1211 // Write to MP 1, port 1. |
1312 { | 1212 { |
1313 DispatcherTransport transport( | 1213 DispatcherTransport transport( |
1314 test::DispatcherTryStartTransport(dispatcher.get())); | 1214 test::DispatcherTryStartTransport(dispatcher.get())); |
1315 EXPECT_TRUE(transport.is_valid()); | 1215 EXPECT_TRUE(transport.is_valid()); |
1316 | 1216 |
1317 std::vector<DispatcherTransport> transports; | 1217 std::vector<DispatcherTransport> transports; |
1318 transports.push_back(transport); | 1218 transports.push_back(transport); |
1319 EXPECT_EQ(MOJO_RESULT_OK, | 1219 EXPECT_EQ( |
1320 mp1->WriteMessage(1, | 1220 MOJO_RESULT_OK, |
1321 UserPointer<const void>(kWorld), | 1221 mp1->WriteMessage(1, UserPointer<const void>(kWorld), sizeof(kWorld), |
1322 sizeof(kWorld), | 1222 &transports, MOJO_WRITE_MESSAGE_FLAG_NONE)); |
1323 &transports, | |
1324 MOJO_WRITE_MESSAGE_FLAG_NONE)); | |
1325 transport.End(); | 1223 transport.End(); |
1326 | 1224 |
1327 // |dispatcher| should have been closed. This is |DCHECK()|ed when the | 1225 // |dispatcher| should have been closed. This is |DCHECK()|ed when the |
1328 // |dispatcher| is destroyed. | 1226 // |dispatcher| is destroyed. |
1329 EXPECT_TRUE(dispatcher->HasOneRef()); | 1227 EXPECT_TRUE(dispatcher->HasOneRef()); |
1330 dispatcher = nullptr; | 1228 dispatcher = nullptr; |
1331 } | 1229 } |
1332 | 1230 |
1333 // Wait. | 1231 // Wait. |
1334 EXPECT_EQ(MOJO_RESULT_OK, waiter.Wait(MOJO_DEADLINE_INDEFINITE, &context)); | 1232 EXPECT_EQ(MOJO_RESULT_OK, waiter.Wait(MOJO_DEADLINE_INDEFINITE, &context)); |
1335 EXPECT_EQ(456u, context); | 1233 EXPECT_EQ(456u, context); |
1336 hss = HandleSignalsState(); | 1234 hss = HandleSignalsState(); |
1337 mp0->RemoveWaiter(0, &waiter, &hss); | 1235 mp0->RemoveWaiter(0, &waiter, &hss); |
1338 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, | 1236 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, |
1339 hss.satisfied_signals); | 1237 hss.satisfied_signals); |
1340 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, | 1238 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, |
1341 hss.satisfiable_signals); | 1239 hss.satisfiable_signals); |
1342 | 1240 |
1343 // Read from MP 0, port 0. | 1241 // Read from MP 0, port 0. |
1344 read_buffer_size = static_cast<uint32_t>(sizeof(read_buffer)); | 1242 read_buffer_size = static_cast<uint32_t>(sizeof(read_buffer)); |
1345 read_num_dispatchers = 10; // Maximum to get. | 1243 read_num_dispatchers = 10; // Maximum to get. |
1346 EXPECT_EQ(MOJO_RESULT_OK, | 1244 EXPECT_EQ( |
1347 mp0->ReadMessage(0, | 1245 MOJO_RESULT_OK, |
1348 UserPointer<void>(read_buffer), | 1246 mp0->ReadMessage(0, UserPointer<void>(read_buffer), |
1349 MakeUserPointer(&read_buffer_size), | 1247 MakeUserPointer(&read_buffer_size), &read_dispatchers, |
1350 &read_dispatchers, | 1248 &read_num_dispatchers, MOJO_READ_MESSAGE_FLAG_NONE)); |
1351 &read_num_dispatchers, | |
1352 MOJO_READ_MESSAGE_FLAG_NONE)); | |
1353 EXPECT_EQ(sizeof(kWorld), static_cast<size_t>(read_buffer_size)); | 1249 EXPECT_EQ(sizeof(kWorld), static_cast<size_t>(read_buffer_size)); |
1354 EXPECT_STREQ(kWorld, read_buffer); | 1250 EXPECT_STREQ(kWorld, read_buffer); |
1355 EXPECT_EQ(1u, read_dispatchers.size()); | 1251 EXPECT_EQ(1u, read_dispatchers.size()); |
1356 EXPECT_EQ(1u, read_num_dispatchers); | 1252 EXPECT_EQ(1u, read_num_dispatchers); |
1357 ASSERT_TRUE(read_dispatchers[0].get()); | 1253 ASSERT_TRUE(read_dispatchers[0].get()); |
1358 EXPECT_TRUE(read_dispatchers[0]->HasOneRef()); | 1254 EXPECT_TRUE(read_dispatchers[0]->HasOneRef()); |
1359 | 1255 |
1360 EXPECT_EQ(Dispatcher::kTypeMessagePipe, read_dispatchers[0]->GetType()); | 1256 EXPECT_EQ(Dispatcher::kTypeMessagePipe, read_dispatchers[0]->GetType()); |
1361 dispatcher = static_cast<MessagePipeDispatcher*>(read_dispatchers[0].get()); | 1257 dispatcher = static_cast<MessagePipeDispatcher*>(read_dispatchers[0].get()); |
1362 read_dispatchers.clear(); | 1258 read_dispatchers.clear(); |
1363 | 1259 |
1364 // Add the waiter now, before it becomes readable to avoid a race. | 1260 // Add the waiter now, before it becomes readable to avoid a race. |
1365 waiter.Init(); | 1261 waiter.Init(); |
1366 ASSERT_EQ(MOJO_RESULT_OK, | 1262 ASSERT_EQ(MOJO_RESULT_OK, |
1367 dispatcher->AddWaiter( | 1263 dispatcher->AddWaiter(&waiter, MOJO_HANDLE_SIGNAL_READABLE, 789, |
1368 &waiter, MOJO_HANDLE_SIGNAL_READABLE, 789, nullptr)); | 1264 nullptr)); |
1369 | 1265 |
1370 // Write to "local_mp", port 1. | 1266 // Write to "local_mp", port 1. |
1371 EXPECT_EQ(MOJO_RESULT_OK, | 1267 EXPECT_EQ( |
1372 local_mp->WriteMessage(1, | 1268 MOJO_RESULT_OK, |
1373 UserPointer<const void>(kHello), | 1269 local_mp->WriteMessage(1, UserPointer<const void>(kHello), sizeof(kHello), |
1374 sizeof(kHello), | 1270 nullptr, MOJO_WRITE_MESSAGE_FLAG_NONE)); |
1375 nullptr, | |
1376 MOJO_WRITE_MESSAGE_FLAG_NONE)); | |
1377 | 1271 |
1378 // Wait for the dispatcher to become readable. | 1272 // Wait for the dispatcher to become readable. |
1379 EXPECT_EQ(MOJO_RESULT_OK, waiter.Wait(MOJO_DEADLINE_INDEFINITE, &context)); | 1273 EXPECT_EQ(MOJO_RESULT_OK, waiter.Wait(MOJO_DEADLINE_INDEFINITE, &context)); |
1380 EXPECT_EQ(789u, context); | 1274 EXPECT_EQ(789u, context); |
1381 hss = HandleSignalsState(); | 1275 hss = HandleSignalsState(); |
1382 dispatcher->RemoveWaiter(&waiter, &hss); | 1276 dispatcher->RemoveWaiter(&waiter, &hss); |
1383 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, | 1277 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, |
1384 hss.satisfied_signals); | 1278 hss.satisfied_signals); |
1385 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, | 1279 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, |
1386 hss.satisfiable_signals); | 1280 hss.satisfiable_signals); |
1387 | 1281 |
1388 // Read from the dispatcher. | 1282 // Read from the dispatcher. |
1389 memset(read_buffer, 0, sizeof(read_buffer)); | 1283 memset(read_buffer, 0, sizeof(read_buffer)); |
1390 read_buffer_size = static_cast<uint32_t>(sizeof(read_buffer)); | 1284 read_buffer_size = static_cast<uint32_t>(sizeof(read_buffer)); |
1391 EXPECT_EQ(MOJO_RESULT_OK, | 1285 EXPECT_EQ(MOJO_RESULT_OK, |
1392 dispatcher->ReadMessage(UserPointer<void>(read_buffer), | 1286 dispatcher->ReadMessage(UserPointer<void>(read_buffer), |
1393 MakeUserPointer(&read_buffer_size), | 1287 MakeUserPointer(&read_buffer_size), 0, |
1394 0, | 1288 nullptr, MOJO_READ_MESSAGE_FLAG_NONE)); |
1395 nullptr, | |
1396 MOJO_READ_MESSAGE_FLAG_NONE)); | |
1397 EXPECT_EQ(sizeof(kHello), static_cast<size_t>(read_buffer_size)); | 1289 EXPECT_EQ(sizeof(kHello), static_cast<size_t>(read_buffer_size)); |
1398 EXPECT_STREQ(kHello, read_buffer); | 1290 EXPECT_STREQ(kHello, read_buffer); |
1399 | 1291 |
1400 // Prepare to wait on "local_mp", port 1. | 1292 // Prepare to wait on "local_mp", port 1. |
1401 waiter.Init(); | 1293 waiter.Init(); |
1402 ASSERT_EQ(MOJO_RESULT_OK, | 1294 ASSERT_EQ(MOJO_RESULT_OK, |
1403 local_mp->AddWaiter( | 1295 local_mp->AddWaiter(1, &waiter, MOJO_HANDLE_SIGNAL_READABLE, 789, |
1404 1, &waiter, MOJO_HANDLE_SIGNAL_READABLE, 789, nullptr)); | 1296 nullptr)); |
1405 | 1297 |
1406 // Write to the dispatcher. | 1298 // Write to the dispatcher. |
1407 EXPECT_EQ(MOJO_RESULT_OK, | 1299 EXPECT_EQ(MOJO_RESULT_OK, dispatcher->WriteMessage( |
1408 dispatcher->WriteMessage(UserPointer<const void>(kHello), | 1300 UserPointer<const void>(kHello), sizeof(kHello), |
1409 sizeof(kHello), | 1301 nullptr, MOJO_WRITE_MESSAGE_FLAG_NONE)); |
1410 nullptr, | |
1411 MOJO_WRITE_MESSAGE_FLAG_NONE)); | |
1412 | 1302 |
1413 // Wait. | 1303 // Wait. |
1414 EXPECT_EQ(MOJO_RESULT_OK, waiter.Wait(MOJO_DEADLINE_INDEFINITE, &context)); | 1304 EXPECT_EQ(MOJO_RESULT_OK, waiter.Wait(MOJO_DEADLINE_INDEFINITE, &context)); |
1415 EXPECT_EQ(789u, context); | 1305 EXPECT_EQ(789u, context); |
1416 hss = HandleSignalsState(); | 1306 hss = HandleSignalsState(); |
1417 local_mp->RemoveWaiter(1, &waiter, &hss); | 1307 local_mp->RemoveWaiter(1, &waiter, &hss); |
1418 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, | 1308 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, |
1419 hss.satisfied_signals); | 1309 hss.satisfied_signals); |
1420 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, | 1310 EXPECT_EQ(MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE, |
1421 hss.satisfiable_signals); | 1311 hss.satisfiable_signals); |
1422 | 1312 |
1423 // Read from "local_mp", port 1. | 1313 // Read from "local_mp", port 1. |
1424 memset(read_buffer, 0, sizeof(read_buffer)); | 1314 memset(read_buffer, 0, sizeof(read_buffer)); |
1425 read_buffer_size = static_cast<uint32_t>(sizeof(read_buffer)); | 1315 read_buffer_size = static_cast<uint32_t>(sizeof(read_buffer)); |
1426 EXPECT_EQ(MOJO_RESULT_OK, | 1316 EXPECT_EQ(MOJO_RESULT_OK, |
1427 local_mp->ReadMessage(1, | 1317 local_mp->ReadMessage(1, UserPointer<void>(read_buffer), |
1428 UserPointer<void>(read_buffer), | 1318 MakeUserPointer(&read_buffer_size), nullptr, |
1429 MakeUserPointer(&read_buffer_size), | 1319 nullptr, MOJO_READ_MESSAGE_FLAG_NONE)); |
1430 nullptr, | |
1431 nullptr, | |
1432 MOJO_READ_MESSAGE_FLAG_NONE)); | |
1433 EXPECT_EQ(sizeof(kHello), static_cast<size_t>(read_buffer_size)); | 1320 EXPECT_EQ(sizeof(kHello), static_cast<size_t>(read_buffer_size)); |
1434 EXPECT_STREQ(kHello, read_buffer); | 1321 EXPECT_STREQ(kHello, read_buffer); |
1435 | 1322 |
1436 // TODO(vtl): Also test the cases where messages are written and read (at | 1323 // TODO(vtl): Also test the cases where messages are written and read (at |
1437 // various points) on the message pipe being passed around. | 1324 // various points) on the message pipe being passed around. |
1438 | 1325 |
1439 // Close everything that belongs to us. | 1326 // Close everything that belongs to us. |
1440 mp0->Close(0); | 1327 mp0->Close(0); |
1441 mp1->Close(1); | 1328 mp1->Close(1); |
1442 EXPECT_EQ(MOJO_RESULT_OK, dispatcher->Close()); | 1329 EXPECT_EQ(MOJO_RESULT_OK, dispatcher->Close()); |
1443 // Note that |local_mp|'s port 0 belong to |dispatcher|, which was closed. | 1330 // Note that |local_mp|'s port 0 belong to |dispatcher|, which was closed. |
1444 local_mp->Close(1); | 1331 local_mp->Close(1); |
1445 } | 1332 } |
1446 | 1333 |
1447 } // namespace | 1334 } // namespace |
1448 } // namespace system | 1335 } // namespace system |
1449 } // namespace mojo | 1336 } // namespace mojo |
OLD | NEW |