OLD | NEW |
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 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 "remoting/host/chromoting_host_context.h" | 5 #include "remoting/host/chromoting_host_context.h" |
6 | 6 |
7 #include <string> | 7 #include <string> |
8 | 8 |
9 #include "base/bind.h" | 9 #include "base/bind.h" |
| 10 #include "base/synchronization/waitable_event.h" |
10 #include "base/threading/thread.h" | 11 #include "base/threading/thread.h" |
11 #include "remoting/jingle_glue/jingle_thread.h" | 12 #include "remoting/jingle_glue/jingle_thread.h" |
12 | 13 |
13 namespace remoting { | 14 namespace remoting { |
14 | 15 |
15 ChromotingHostContext::ChromotingHostContext() | 16 namespace { |
| 17 void InitializeMessageLoopProxy( |
| 18 scoped_refptr<base::MessageLoopProxy>* proxy, |
| 19 base::WaitableEvent* done) { |
| 20 *proxy = base::MessageLoopProxy::CreateForCurrentThread(); |
| 21 done->Signal(); |
| 22 } |
| 23 } // namespace |
| 24 |
| 25 ChromotingHostContext::ChromotingHostContext( |
| 26 base::MessageLoopProxy* ui_message_loop) |
16 : main_thread_("ChromotingMainThread"), | 27 : main_thread_("ChromotingMainThread"), |
17 encode_thread_("ChromotingEncodeThread"), | 28 encode_thread_("ChromotingEncodeThread"), |
18 desktop_thread_("ChromotingDesktopThread") { | 29 desktop_thread_("ChromotingDesktopThread"), |
| 30 ui_message_loop_(ui_message_loop) { |
19 } | 31 } |
20 | 32 |
21 ChromotingHostContext::~ChromotingHostContext() { | 33 ChromotingHostContext::~ChromotingHostContext() { |
22 } | 34 } |
23 | 35 |
24 void ChromotingHostContext::Start() { | 36 void ChromotingHostContext::Start() { |
25 // Start all the threads. | 37 // Start all the threads. |
26 main_thread_.Start(); | 38 main_thread_.Start(); |
27 encode_thread_.Start(); | 39 encode_thread_.Start(); |
28 jingle_thread_.Start(); | 40 jingle_thread_.Start(); |
29 desktop_thread_.Start(); | 41 desktop_thread_.Start(); |
| 42 |
| 43 // Initialize |network_message_loop_| on the network thread. |
| 44 base::WaitableEvent proxy_event(true, false); |
| 45 jingle_thread_.message_loop()->PostTask(FROM_HERE, base::Bind( |
| 46 &InitializeMessageLoopProxy, &network_message_loop_, &proxy_event)); |
| 47 proxy_event.Wait(); |
30 } | 48 } |
31 | 49 |
32 void ChromotingHostContext::Stop() { | 50 void ChromotingHostContext::Stop() { |
33 // Stop all the threads. | 51 // Stop all the threads. |
34 jingle_thread_.Stop(); | 52 jingle_thread_.Stop(); |
35 encode_thread_.Stop(); | 53 encode_thread_.Stop(); |
36 main_thread_.Stop(); | 54 main_thread_.Stop(); |
37 desktop_thread_.Stop(); | 55 desktop_thread_.Stop(); |
38 } | 56 } |
39 | 57 |
40 JingleThread* ChromotingHostContext::jingle_thread() { | 58 JingleThread* ChromotingHostContext::jingle_thread() { |
41 return &jingle_thread_; | 59 return &jingle_thread_; |
42 } | 60 } |
43 | 61 |
| 62 base::MessageLoopProxy* ChromotingHostContext::ui_message_loop() { |
| 63 return ui_message_loop_; |
| 64 } |
| 65 |
44 MessageLoop* ChromotingHostContext::main_message_loop() { | 66 MessageLoop* ChromotingHostContext::main_message_loop() { |
45 return main_thread_.message_loop(); | 67 return main_thread_.message_loop(); |
46 } | 68 } |
47 | 69 |
48 MessageLoop* ChromotingHostContext::encode_message_loop() { | 70 MessageLoop* ChromotingHostContext::encode_message_loop() { |
49 return encode_thread_.message_loop(); | 71 return encode_thread_.message_loop(); |
50 } | 72 } |
51 | 73 |
52 base::MessageLoopProxy* ChromotingHostContext::network_message_loop() { | 74 base::MessageLoopProxy* ChromotingHostContext::network_message_loop() { |
53 return jingle_thread_.message_loop_proxy(); | 75 return network_message_loop_; |
54 } | 76 } |
55 | 77 |
56 MessageLoop* ChromotingHostContext::desktop_message_loop() { | 78 MessageLoop* ChromotingHostContext::desktop_message_loop() { |
57 return desktop_thread_.message_loop(); | 79 return desktop_thread_.message_loop(); |
58 } | 80 } |
59 | 81 |
60 void ChromotingHostContext::SetUITaskPostFunction( | |
61 const UIThreadPostTaskFunction& poster) { | |
62 ui_poster_ = poster; | |
63 ui_main_thread_id_ = base::PlatformThread::CurrentId(); | |
64 } | |
65 | |
66 void ChromotingHostContext::PostTaskToUIThread( | |
67 const tracked_objects::Location& from_here, const base::Closure& task) { | |
68 ui_poster_.Run(from_here, task); | |
69 } | |
70 | |
71 void ChromotingHostContext::PostDelayedTaskToUIThread( | |
72 const tracked_objects::Location& from_here, | |
73 const base::Closure& task, | |
74 int delay_ms) { | |
75 // Post delayed task on the main thread that will post task on UI | |
76 // thread. It is safe to use base::Unretained() here because | |
77 // ChromotingHostContext owns |main_thread_|. | |
78 main_message_loop()->PostDelayedTask(from_here, base::Bind( | |
79 &ChromotingHostContext::PostTaskToUIThread, base::Unretained(this), | |
80 from_here, task), delay_ms); | |
81 } | |
82 | |
83 bool ChromotingHostContext::IsUIThread() const { | |
84 return ui_main_thread_id_ == base::PlatformThread::CurrentId(); | |
85 } | |
86 | |
87 } // namespace remoting | 82 } // namespace remoting |
OLD | NEW |