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 #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 <string> | 9 #include <string> |
10 #include <vector> | 10 #include <vector> |
(...skipping 24 matching lines...) Expand all Loading... |
35 #include "mojo/system/test_utils.h" | 35 #include "mojo/system/test_utils.h" |
36 #include "mojo/system/waiter.h" | 36 #include "mojo/system/waiter.h" |
37 #include "testing/gtest/include/gtest/gtest.h" | 37 #include "testing/gtest/include/gtest/gtest.h" |
38 | 38 |
39 namespace mojo { | 39 namespace mojo { |
40 namespace system { | 40 namespace system { |
41 namespace { | 41 namespace { |
42 | 42 |
43 class ChannelThread { | 43 class ChannelThread { |
44 public: | 44 public: |
45 ChannelThread() : test_io_thread_(test::TestIOThread::kManualStart) {} | 45 explicit ChannelThread(embedder::PlatformSupport* platform_support) |
| 46 : platform_support_(platform_support), |
| 47 test_io_thread_(test::TestIOThread::kManualStart) {} |
46 ~ChannelThread() { Stop(); } | 48 ~ChannelThread() { Stop(); } |
47 | 49 |
48 void Start(embedder::ScopedPlatformHandle platform_handle, | 50 void Start(embedder::ScopedPlatformHandle platform_handle, |
49 scoped_refptr<MessagePipe> message_pipe) { | 51 scoped_refptr<MessagePipe> message_pipe) { |
50 test_io_thread_.Start(); | 52 test_io_thread_.Start(); |
51 test_io_thread_.PostTaskAndWait( | 53 test_io_thread_.PostTaskAndWait( |
52 FROM_HERE, | 54 FROM_HERE, |
53 base::Bind(&ChannelThread::InitChannelOnIOThread, | 55 base::Bind(&ChannelThread::InitChannelOnIOThread, |
54 base::Unretained(this), | 56 base::Unretained(this), |
55 base::Passed(&platform_handle), | 57 base::Passed(&platform_handle), |
(...skipping 16 matching lines...) Expand all Loading... |
72 test_io_thread_.Stop(); | 74 test_io_thread_.Stop(); |
73 } | 75 } |
74 | 76 |
75 private: | 77 private: |
76 void InitChannelOnIOThread(embedder::ScopedPlatformHandle platform_handle, | 78 void InitChannelOnIOThread(embedder::ScopedPlatformHandle platform_handle, |
77 scoped_refptr<MessagePipe> message_pipe) { | 79 scoped_refptr<MessagePipe> message_pipe) { |
78 CHECK_EQ(base::MessageLoop::current(), test_io_thread_.message_loop()); | 80 CHECK_EQ(base::MessageLoop::current(), test_io_thread_.message_loop()); |
79 CHECK(platform_handle.is_valid()); | 81 CHECK(platform_handle.is_valid()); |
80 | 82 |
81 // Create and initialize |Channel|. | 83 // Create and initialize |Channel|. |
82 channel_ = new Channel(); | 84 channel_ = new Channel(platform_support_); |
83 CHECK(channel_->Init(RawChannel::Create(platform_handle.Pass()))); | 85 CHECK(channel_->Init(RawChannel::Create(platform_handle.Pass()))); |
84 | 86 |
85 // Attach the message pipe endpoint. | 87 // Attach the message pipe endpoint. |
86 // Note: On the "server" (parent process) side, we need not attach the | 88 // Note: On the "server" (parent process) side, we need not attach the |
87 // message pipe endpoint immediately. However, on the "client" (child | 89 // message pipe endpoint immediately. However, on the "client" (child |
88 // process) side, this *must* be done here -- otherwise, the |Channel| may | 90 // process) side, this *must* be done here -- otherwise, the |Channel| may |
89 // receive/process messages (which it can do as soon as it's hooked up to | 91 // receive/process messages (which it can do as soon as it's hooked up to |
90 // the IO thread message loop, and that message loop runs) before the | 92 // the IO thread message loop, and that message loop runs) before the |
91 // message pipe endpoint is attached. | 93 // message pipe endpoint is attached. |
92 CHECK_EQ(channel_->AttachMessagePipeEndpoint(message_pipe, 1), | 94 CHECK_EQ(channel_->AttachMessagePipeEndpoint(message_pipe, 1), |
93 Channel::kBootstrapEndpointId); | 95 Channel::kBootstrapEndpointId); |
94 CHECK(channel_->RunMessagePipeEndpoint(Channel::kBootstrapEndpointId, | 96 CHECK(channel_->RunMessagePipeEndpoint(Channel::kBootstrapEndpointId, |
95 Channel::kBootstrapEndpointId)); | 97 Channel::kBootstrapEndpointId)); |
96 } | 98 } |
97 | 99 |
98 void ShutdownChannelOnIOThread() { | 100 void ShutdownChannelOnIOThread() { |
99 CHECK(channel_); | 101 CHECK(channel_); |
100 channel_->Shutdown(); | 102 channel_->Shutdown(); |
101 channel_ = NULL; | 103 channel_ = NULL; |
102 } | 104 } |
103 | 105 |
| 106 embedder::PlatformSupport* const platform_support_; |
104 test::TestIOThread test_io_thread_; | 107 test::TestIOThread test_io_thread_; |
105 scoped_refptr<Channel> channel_; | 108 scoped_refptr<Channel> channel_; |
106 | 109 |
107 DISALLOW_COPY_AND_ASSIGN(ChannelThread); | 110 DISALLOW_COPY_AND_ASSIGN(ChannelThread); |
108 }; | 111 }; |
109 | 112 |
110 class MultiprocessMessagePipeTest : public testing::Test { | 113 class MultiprocessMessagePipeTest : public testing::Test { |
111 public: | 114 public: |
112 MultiprocessMessagePipeTest() {} | 115 MultiprocessMessagePipeTest() : channel_thread_(&platform_support_) {} |
113 virtual ~MultiprocessMessagePipeTest() {} | 116 virtual ~MultiprocessMessagePipeTest() {} |
114 | 117 |
115 protected: | 118 protected: |
116 void Init(scoped_refptr<MessagePipe> mp) { | 119 void Init(scoped_refptr<MessagePipe> mp) { |
117 channel_thread_.Start(helper_.server_platform_handle.Pass(), mp); | 120 channel_thread_.Start(helper_.server_platform_handle.Pass(), mp); |
118 } | 121 } |
119 | 122 |
| 123 embedder::PlatformSupport* platform_support() { return &platform_support_; } |
120 mojo::test::MultiprocessTestHelper* helper() { return &helper_; } | 124 mojo::test::MultiprocessTestHelper* helper() { return &helper_; } |
121 | 125 |
122 private: | 126 private: |
| 127 embedder::SimplePlatformSupport platform_support_; |
123 ChannelThread channel_thread_; | 128 ChannelThread channel_thread_; |
124 mojo::test::MultiprocessTestHelper helper_; | 129 mojo::test::MultiprocessTestHelper helper_; |
125 | 130 |
126 DISALLOW_COPY_AND_ASSIGN(MultiprocessMessagePipeTest); | 131 DISALLOW_COPY_AND_ASSIGN(MultiprocessMessagePipeTest); |
127 }; | 132 }; |
128 | 133 |
129 MojoResult WaitIfNecessary(scoped_refptr<MessagePipe> mp, | 134 MojoResult WaitIfNecessary(scoped_refptr<MessagePipe> mp, |
130 MojoHandleSignals signals, | 135 MojoHandleSignals signals, |
131 HandleSignalsState* signals_state) { | 136 HandleSignalsState* signals_state) { |
132 Waiter waiter; | 137 Waiter waiter; |
133 waiter.Init(); | 138 waiter.Init(); |
134 | 139 |
135 MojoResult add_result = mp->AddWaiter(0, &waiter, signals, 0, signals_state); | 140 MojoResult add_result = mp->AddWaiter(0, &waiter, signals, 0, signals_state); |
136 if (add_result != MOJO_RESULT_OK) { | 141 if (add_result != MOJO_RESULT_OK) { |
137 return (add_result == MOJO_RESULT_ALREADY_EXISTS) ? MOJO_RESULT_OK | 142 return (add_result == MOJO_RESULT_ALREADY_EXISTS) ? MOJO_RESULT_OK |
138 : add_result; | 143 : add_result; |
139 } | 144 } |
140 | 145 |
141 MojoResult wait_result = waiter.Wait(MOJO_DEADLINE_INDEFINITE, NULL); | 146 MojoResult wait_result = waiter.Wait(MOJO_DEADLINE_INDEFINITE, NULL); |
142 mp->RemoveWaiter(0, &waiter, signals_state); | 147 mp->RemoveWaiter(0, &waiter, signals_state); |
143 return wait_result; | 148 return wait_result; |
144 } | 149 } |
145 | 150 |
146 // For each message received, sends a reply message with the same contents | 151 // For each message received, sends a reply message with the same contents |
147 // repeated twice, until the other end is closed or it receives "quitquitquit" | 152 // repeated twice, until the other end is closed or it receives "quitquitquit" |
148 // (which it doesn't reply to). It'll return the number of messages received, | 153 // (which it doesn't reply to). It'll return the number of messages received, |
149 // not including any "quitquitquit" message, modulo 100. | 154 // not including any "quitquitquit" message, modulo 100. |
150 MOJO_MULTIPROCESS_TEST_CHILD_MAIN(EchoEcho) { | 155 MOJO_MULTIPROCESS_TEST_CHILD_MAIN(EchoEcho) { |
151 ChannelThread channel_thread; | 156 embedder::SimplePlatformSupport platform_support; |
| 157 ChannelThread channel_thread(&platform_support); |
152 embedder::ScopedPlatformHandle client_platform_handle = | 158 embedder::ScopedPlatformHandle client_platform_handle = |
153 mojo::test::MultiprocessTestHelper::client_platform_handle.Pass(); | 159 mojo::test::MultiprocessTestHelper::client_platform_handle.Pass(); |
154 CHECK(client_platform_handle.is_valid()); | 160 CHECK(client_platform_handle.is_valid()); |
155 scoped_refptr<MessagePipe> mp(new MessagePipe( | 161 scoped_refptr<MessagePipe> mp(new MessagePipe( |
156 scoped_ptr<MessagePipeEndpoint>(new LocalMessagePipeEndpoint()), | 162 scoped_ptr<MessagePipeEndpoint>(new LocalMessagePipeEndpoint()), |
157 scoped_ptr<MessagePipeEndpoint>(new ProxyMessagePipeEndpoint()))); | 163 scoped_ptr<MessagePipeEndpoint>(new ProxyMessagePipeEndpoint()))); |
158 channel_thread.Start(client_platform_handle.Pass(), mp); | 164 channel_thread.Start(client_platform_handle.Pass(), mp); |
159 | 165 |
160 const std::string quitquitquit("quitquitquit"); | 166 const std::string quitquitquit("quitquitquit"); |
161 int rv = 0; | 167 int rv = 0; |
(...skipping 148 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
310 EXPECT_EQ(0u, hss.satisfied_signals); | 316 EXPECT_EQ(0u, hss.satisfied_signals); |
311 EXPECT_EQ(0u, hss.satisfiable_signals); | 317 EXPECT_EQ(0u, hss.satisfiable_signals); |
312 | 318 |
313 mp->Close(0); | 319 mp->Close(0); |
314 | 320 |
315 EXPECT_EQ(static_cast<int>(kNumMessages % 100), | 321 EXPECT_EQ(static_cast<int>(kNumMessages % 100), |
316 helper()->WaitForChildShutdown()); | 322 helper()->WaitForChildShutdown()); |
317 } | 323 } |
318 | 324 |
319 MOJO_MULTIPROCESS_TEST_CHILD_MAIN(CheckSharedBuffer) { | 325 MOJO_MULTIPROCESS_TEST_CHILD_MAIN(CheckSharedBuffer) { |
320 ChannelThread channel_thread; | 326 embedder::SimplePlatformSupport platform_support; |
| 327 ChannelThread channel_thread(&platform_support); |
321 embedder::ScopedPlatformHandle client_platform_handle = | 328 embedder::ScopedPlatformHandle client_platform_handle = |
322 mojo::test::MultiprocessTestHelper::client_platform_handle.Pass(); | 329 mojo::test::MultiprocessTestHelper::client_platform_handle.Pass(); |
323 CHECK(client_platform_handle.is_valid()); | 330 CHECK(client_platform_handle.is_valid()); |
324 scoped_refptr<MessagePipe> mp(new MessagePipe( | 331 scoped_refptr<MessagePipe> mp(new MessagePipe( |
325 scoped_ptr<MessagePipeEndpoint>(new LocalMessagePipeEndpoint()), | 332 scoped_ptr<MessagePipeEndpoint>(new LocalMessagePipeEndpoint()), |
326 scoped_ptr<MessagePipeEndpoint>(new ProxyMessagePipeEndpoint()))); | 333 scoped_ptr<MessagePipeEndpoint>(new ProxyMessagePipeEndpoint()))); |
327 channel_thread.Start(client_platform_handle.Pass(), mp); | 334 channel_thread.Start(client_platform_handle.Pass(), mp); |
328 | 335 |
329 // Wait for the first message from our parent. | 336 // Wait for the first message from our parent. |
330 HandleSignalsState hss; | 337 HandleSignalsState hss; |
(...skipping 90 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
421 #endif | 428 #endif |
422 TEST_F(MultiprocessMessagePipeTest, MAYBE_SharedBufferPassing) { | 429 TEST_F(MultiprocessMessagePipeTest, MAYBE_SharedBufferPassing) { |
423 helper()->StartChild("CheckSharedBuffer"); | 430 helper()->StartChild("CheckSharedBuffer"); |
424 | 431 |
425 scoped_refptr<MessagePipe> mp(new MessagePipe( | 432 scoped_refptr<MessagePipe> mp(new MessagePipe( |
426 scoped_ptr<MessagePipeEndpoint>(new LocalMessagePipeEndpoint()), | 433 scoped_ptr<MessagePipeEndpoint>(new LocalMessagePipeEndpoint()), |
427 scoped_ptr<MessagePipeEndpoint>(new ProxyMessagePipeEndpoint()))); | 434 scoped_ptr<MessagePipeEndpoint>(new ProxyMessagePipeEndpoint()))); |
428 Init(mp); | 435 Init(mp); |
429 | 436 |
430 // Make a shared buffer. | 437 // Make a shared buffer. |
431 embedder::SimplePlatformSupport platform_support; | |
432 scoped_refptr<SharedBufferDispatcher> dispatcher; | 438 scoped_refptr<SharedBufferDispatcher> dispatcher; |
433 EXPECT_EQ(MOJO_RESULT_OK, | 439 EXPECT_EQ(MOJO_RESULT_OK, |
434 SharedBufferDispatcher::Create( | 440 SharedBufferDispatcher::Create( |
435 &platform_support, | 441 platform_support(), |
436 SharedBufferDispatcher::kDefaultCreateOptions, | 442 SharedBufferDispatcher::kDefaultCreateOptions, |
437 100, | 443 100, |
438 &dispatcher)); | 444 &dispatcher)); |
439 ASSERT_TRUE(dispatcher); | 445 ASSERT_TRUE(dispatcher); |
440 | 446 |
441 // Make a mapping. | 447 // Make a mapping. |
442 scoped_ptr<embedder::PlatformSharedBufferMapping> mapping; | 448 scoped_ptr<embedder::PlatformSharedBufferMapping> mapping; |
443 EXPECT_EQ(MOJO_RESULT_OK, | 449 EXPECT_EQ(MOJO_RESULT_OK, |
444 dispatcher->MapBuffer(0, 100, MOJO_MAP_BUFFER_FLAG_NONE, &mapping)); | 450 dispatcher->MapBuffer(0, 100, MOJO_MAP_BUFFER_FLAG_NONE, &mapping)); |
445 ASSERT_TRUE(mapping); | 451 ASSERT_TRUE(mapping); |
(...skipping 62 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
508 WaitIfNecessary(mp, MOJO_HANDLE_SIGNAL_READABLE, &hss)); | 514 WaitIfNecessary(mp, MOJO_HANDLE_SIGNAL_READABLE, &hss)); |
509 EXPECT_EQ(0u, hss.satisfied_signals); | 515 EXPECT_EQ(0u, hss.satisfied_signals); |
510 EXPECT_EQ(0u, hss.satisfiable_signals); | 516 EXPECT_EQ(0u, hss.satisfiable_signals); |
511 | 517 |
512 mp->Close(0); | 518 mp->Close(0); |
513 | 519 |
514 EXPECT_EQ(0, helper()->WaitForChildShutdown()); | 520 EXPECT_EQ(0, helper()->WaitForChildShutdown()); |
515 } | 521 } |
516 | 522 |
517 MOJO_MULTIPROCESS_TEST_CHILD_MAIN(CheckPlatformHandleFile) { | 523 MOJO_MULTIPROCESS_TEST_CHILD_MAIN(CheckPlatformHandleFile) { |
518 ChannelThread channel_thread; | 524 embedder::SimplePlatformSupport platform_support; |
| 525 ChannelThread channel_thread(&platform_support); |
519 embedder::ScopedPlatformHandle client_platform_handle = | 526 embedder::ScopedPlatformHandle client_platform_handle = |
520 mojo::test::MultiprocessTestHelper::client_platform_handle.Pass(); | 527 mojo::test::MultiprocessTestHelper::client_platform_handle.Pass(); |
521 CHECK(client_platform_handle.is_valid()); | 528 CHECK(client_platform_handle.is_valid()); |
522 scoped_refptr<MessagePipe> mp(new MessagePipe( | 529 scoped_refptr<MessagePipe> mp(new MessagePipe( |
523 scoped_ptr<MessagePipeEndpoint>(new LocalMessagePipeEndpoint()), | 530 scoped_ptr<MessagePipeEndpoint>(new LocalMessagePipeEndpoint()), |
524 scoped_ptr<MessagePipeEndpoint>(new ProxyMessagePipeEndpoint()))); | 531 scoped_ptr<MessagePipeEndpoint>(new ProxyMessagePipeEndpoint()))); |
525 channel_thread.Start(client_platform_handle.Pass(), mp); | 532 channel_thread.Start(client_platform_handle.Pass(), mp); |
526 | 533 |
527 HandleSignalsState hss; | 534 HandleSignalsState hss; |
528 CHECK_EQ(WaitIfNecessary(mp, MOJO_HANDLE_SIGNAL_READABLE, &hss), | 535 CHECK_EQ(WaitIfNecessary(mp, MOJO_HANDLE_SIGNAL_READABLE, &hss), |
(...skipping 94 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
623 EXPECT_EQ(0u, hss.satisfiable_signals); | 630 EXPECT_EQ(0u, hss.satisfiable_signals); |
624 | 631 |
625 mp->Close(0); | 632 mp->Close(0); |
626 | 633 |
627 EXPECT_EQ(0, helper()->WaitForChildShutdown()); | 634 EXPECT_EQ(0, helper()->WaitForChildShutdown()); |
628 } | 635 } |
629 | 636 |
630 } // namespace | 637 } // namespace |
631 } // namespace system | 638 } // namespace system |
632 } // namespace mojo | 639 } // namespace mojo |
OLD | NEW |