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/embedder/channel_init.h" | 5 #include "mojo/edk/embedder/channel_init.h" |
6 | 6 |
7 #include "base/bind.h" | 7 #include "base/bind.h" |
8 #include "base/message_loop/message_loop.h" | 8 #include "base/message_loop/message_loop.h" |
9 #include "mojo/edk/embedder/embedder.h" | 9 #include "mojo/edk/embedder/embedder.h" |
10 | 10 |
11 namespace mojo { | 11 namespace mojo { |
12 namespace embedder { | 12 namespace embedder { |
13 | 13 |
14 ChannelInit::ChannelInit() : channel_info_(nullptr), weak_factory_(this) { | 14 ChannelInit::ChannelInit() : channel_info_(nullptr), weak_factory_(this) { |
15 } | 15 } |
16 | 16 |
17 ChannelInit::~ChannelInit() { | 17 ChannelInit::~ChannelInit() { |
| 18 // TODO(vtl): This is likely leaky in common scenarios (we're on the main |
| 19 // thread, which outlives the I/O thread, and we're destroyed after the I/O |
| 20 // thread is destroyed. |
18 if (channel_info_) | 21 if (channel_info_) |
19 DestroyChannel(channel_info_); | 22 DestroyChannel(channel_info_); |
20 } | 23 } |
21 | 24 |
22 ScopedMessagePipeHandle ChannelInit::Init( | 25 ScopedMessagePipeHandle ChannelInit::Init( |
23 base::PlatformFile file, | 26 base::PlatformFile file, |
24 scoped_refptr<base::TaskRunner> io_thread_task_runner) { | 27 scoped_refptr<base::TaskRunner> io_thread_task_runner) { |
25 DCHECK(!io_thread_task_runner_); // Should only init once. | |
26 io_thread_task_runner_ = io_thread_task_runner; | |
27 ScopedMessagePipeHandle message_pipe = | 28 ScopedMessagePipeHandle message_pipe = |
28 CreateChannel( | 29 CreateChannel(ScopedPlatformHandle(PlatformHandle(file)), |
29 ScopedPlatformHandle(PlatformHandle(file)), io_thread_task_runner, | 30 io_thread_task_runner, |
30 base::Bind(&ChannelInit::OnCreatedChannel, weak_factory_.GetWeakPtr(), | 31 base::Bind(&ChannelInit::OnCreatedChannel, |
31 io_thread_task_runner), | 32 weak_factory_.GetWeakPtr()), |
32 base::MessageLoop::current()->message_loop_proxy()).Pass(); | 33 base::MessageLoop::current()->task_runner()).Pass(); |
33 return message_pipe.Pass(); | 34 return message_pipe.Pass(); |
34 } | 35 } |
35 | 36 |
36 void ChannelInit::WillDestroySoon() { | 37 void ChannelInit::WillDestroySoon() { |
37 if (channel_info_) | 38 if (channel_info_) |
38 WillDestroyChannelSoon(channel_info_); | 39 WillDestroyChannelSoon(channel_info_); |
39 } | 40 } |
40 | 41 |
41 // static | 42 // static |
42 void ChannelInit::OnCreatedChannel(base::WeakPtr<ChannelInit> self, | 43 void ChannelInit::OnCreatedChannel(base::WeakPtr<ChannelInit> self, |
43 scoped_refptr<base::TaskRunner> io_thread, | |
44 ChannelInfo* channel) { | 44 ChannelInfo* channel) { |
45 // If |self| was already destroyed, shut the channel down. | 45 // If |self| was already destroyed, shut the channel down. |
46 if (!self) { | 46 if (!self) { |
47 DestroyChannel(channel); | 47 DestroyChannel(channel); |
48 return; | 48 return; |
49 } | 49 } |
50 | 50 |
| 51 DCHECK(!self->channel_info_); |
51 self->channel_info_ = channel; | 52 self->channel_info_ = channel; |
52 } | 53 } |
53 | 54 |
54 } // namespace embedder | 55 } // namespace embedder |
55 } // namespace mojo | 56 } // namespace mojo |
OLD | NEW |