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 "mojo/edk/system/message_pipe_test_utils.h" | 5 #include "mojo/edk/system/message_pipe_test_utils.h" |
6 | 6 |
7 #include "base/bind.h" | 7 #include "base/bind.h" |
8 #include "base/threading/platform_thread.h" // For |Sleep()|. | 8 #include "base/threading/platform_thread.h" // For |Sleep()|. |
9 #include "mojo/edk/system/channel.h" | 9 #include "mojo/edk/system/channel.h" |
10 #include "mojo/edk/system/channel_endpoint.h" | 10 #include "mojo/edk/system/channel_endpoint.h" |
11 #include "mojo/edk/system/message_pipe.h" | 11 #include "mojo/edk/system/message_pipe.h" |
12 #include "mojo/edk/system/waiter.h" | 12 #include "mojo/edk/system/waiter.h" |
13 | 13 |
14 namespace mojo { | 14 namespace mojo { |
15 namespace system { | 15 namespace system { |
16 namespace test { | 16 namespace test { |
17 | 17 |
18 MojoResult WaitIfNecessary(scoped_refptr<MessagePipe> mp, | 18 MojoResult WaitIfNecessary(scoped_refptr<MessagePipe> mp, |
19 MojoHandleSignals signals, | 19 MojoHandleSignals signals, |
20 HandleSignalsState* signals_state) { | 20 HandleSignalsState* signals_state) { |
21 Waiter waiter; | 21 Waiter waiter; |
22 waiter.Init(); | 22 waiter.Init(); |
23 | 23 |
24 MojoResult add_result = mp->AddWaiter(0, &waiter, signals, 0, signals_state); | 24 MojoResult add_result = |
| 25 mp->AddAwakable(0, &waiter, signals, 0, signals_state); |
25 if (add_result != MOJO_RESULT_OK) { | 26 if (add_result != MOJO_RESULT_OK) { |
26 return (add_result == MOJO_RESULT_ALREADY_EXISTS) ? MOJO_RESULT_OK | 27 return (add_result == MOJO_RESULT_ALREADY_EXISTS) ? MOJO_RESULT_OK |
27 : add_result; | 28 : add_result; |
28 } | 29 } |
29 | 30 |
30 MojoResult wait_result = waiter.Wait(MOJO_DEADLINE_INDEFINITE, nullptr); | 31 MojoResult wait_result = waiter.Wait(MOJO_DEADLINE_INDEFINITE, nullptr); |
31 mp->RemoveWaiter(0, &waiter, signals_state); | 32 mp->RemoveAwakable(0, &waiter, signals_state); |
32 return wait_result; | 33 return wait_result; |
33 } | 34 } |
34 | 35 |
35 ChannelThread::ChannelThread(embedder::PlatformSupport* platform_support) | 36 ChannelThread::ChannelThread(embedder::PlatformSupport* platform_support) |
36 : platform_support_(platform_support), | 37 : platform_support_(platform_support), |
37 test_io_thread_(base::TestIOThread::kManualStart) { | 38 test_io_thread_(base::TestIOThread::kManualStart) { |
38 } | 39 } |
39 | 40 |
40 ChannelThread::~ChannelThread() { | 41 ChannelThread::~ChannelThread() { |
41 Stop(); | 42 Stop(); |
42 } | 43 } |
43 | 44 |
44 void ChannelThread::Start(embedder::ScopedPlatformHandle platform_handle, | 45 void ChannelThread::Start(embedder::ScopedPlatformHandle platform_handle, |
45 scoped_refptr<ChannelEndpoint> channel_endpoint) { | 46 scoped_refptr<ChannelEndpoint> channel_endpoint) { |
46 test_io_thread_.Start(); | 47 test_io_thread_.Start(); |
47 test_io_thread_.PostTaskAndWait( | 48 test_io_thread_.PostTaskAndWait( |
48 FROM_HERE, | 49 FROM_HERE, |
49 base::Bind(&ChannelThread::InitChannelOnIOThread, base::Unretained(this), | 50 base::Bind(&ChannelThread::InitChannelOnIOThread, base::Unretained(this), |
50 base::Passed(&platform_handle), channel_endpoint)); | 51 base::Passed(&platform_handle), channel_endpoint)); |
51 } | 52 } |
52 | 53 |
53 void ChannelThread::Stop() { | 54 void ChannelThread::Stop() { |
54 if (channel_.get()) { | 55 if (channel_) { |
55 // Hack to flush write buffers before quitting. | 56 // Hack to flush write buffers before quitting. |
56 // TODO(vtl): Remove this once |Channel| has a | 57 // TODO(vtl): Remove this once |Channel| has a |
57 // |FlushWriteBufferAndShutdown()| (or whatever). | 58 // |FlushWriteBufferAndShutdown()| (or whatever). |
58 while (!channel_->IsWriteBufferEmpty()) | 59 while (!channel_->IsWriteBufferEmpty()) |
59 base::PlatformThread::Sleep(base::TimeDelta::FromMilliseconds(20)); | 60 base::PlatformThread::Sleep(base::TimeDelta::FromMilliseconds(20)); |
60 | 61 |
61 test_io_thread_.PostTaskAndWait( | 62 test_io_thread_.PostTaskAndWait( |
62 FROM_HERE, base::Bind(&ChannelThread::ShutdownChannelOnIOThread, | 63 FROM_HERE, base::Bind(&ChannelThread::ShutdownChannelOnIOThread, |
63 base::Unretained(this))); | 64 base::Unretained(this))); |
64 } | 65 } |
(...skipping 13 matching lines...) Expand all Loading... |
78 // Attach and run the endpoint. | 79 // Attach and run the endpoint. |
79 // Note: On the "server" (parent process) side, we need not attach/run the | 80 // Note: On the "server" (parent process) side, we need not attach/run the |
80 // endpoint immediately. However, on the "client" (child process) side, this | 81 // endpoint immediately. However, on the "client" (child process) side, this |
81 // *must* be done here -- otherwise, the |Channel| may receive/process | 82 // *must* be done here -- otherwise, the |Channel| may receive/process |
82 // messages (which it can do as soon as it's hooked up to the IO thread | 83 // messages (which it can do as soon as it's hooked up to the IO thread |
83 // message loop, and that message loop runs) before the endpoint is attached. | 84 // message loop, and that message loop runs) before the endpoint is attached. |
84 channel_->AttachAndRunEndpoint(channel_endpoint, true); | 85 channel_->AttachAndRunEndpoint(channel_endpoint, true); |
85 } | 86 } |
86 | 87 |
87 void ChannelThread::ShutdownChannelOnIOThread() { | 88 void ChannelThread::ShutdownChannelOnIOThread() { |
88 CHECK(channel_.get()); | 89 CHECK(channel_); |
89 channel_->Shutdown(); | 90 channel_->Shutdown(); |
90 channel_ = nullptr; | 91 channel_ = nullptr; |
91 } | 92 } |
92 | 93 |
93 #if !defined(OS_IOS) | 94 #if !defined(OS_IOS) |
94 MultiprocessMessagePipeTestBase::MultiprocessMessagePipeTestBase() | 95 MultiprocessMessagePipeTestBase::MultiprocessMessagePipeTestBase() |
95 : channel_thread_(&platform_support_) { | 96 : channel_thread_(&platform_support_) { |
96 } | 97 } |
97 | 98 |
98 MultiprocessMessagePipeTestBase::~MultiprocessMessagePipeTestBase() { | 99 MultiprocessMessagePipeTestBase::~MultiprocessMessagePipeTestBase() { |
99 } | 100 } |
100 | 101 |
101 void MultiprocessMessagePipeTestBase::Init(scoped_refptr<ChannelEndpoint> ep) { | 102 void MultiprocessMessagePipeTestBase::Init(scoped_refptr<ChannelEndpoint> ep) { |
102 channel_thread_.Start(helper_.server_platform_handle.Pass(), ep); | 103 channel_thread_.Start(helper_.server_platform_handle.Pass(), ep); |
103 } | 104 } |
104 #endif | 105 #endif |
105 | 106 |
106 } // namespace test | 107 } // namespace test |
107 } // namespace system | 108 } // namespace system |
108 } // namespace mojo | 109 } // namespace mojo |
OLD | NEW |