OLD | NEW |
| (Empty) |
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 | |
3 // found in the LICENSE file. | |
4 | |
5 // An implementation of WebThread in terms of base::MessageLoop and | |
6 // base::Thread | |
7 | |
8 #include "content/child/webthread_impl.h" | |
9 | |
10 #include "base/bind.h" | |
11 #include "base/bind_helpers.h" | |
12 #include "base/pending_task.h" | |
13 #include "base/threading/platform_thread.h" | |
14 #include "third_party/WebKit/public/platform/WebTraceLocation.h" | |
15 | |
16 namespace content { | |
17 | |
18 class WebThreadBase::TaskObserverAdapter | |
19 : public base::MessageLoop::TaskObserver { | |
20 public: | |
21 TaskObserverAdapter(WebThread::TaskObserver* observer) | |
22 : observer_(observer) {} | |
23 | |
24 void WillProcessTask(const base::PendingTask& pending_task) override { | |
25 observer_->willProcessTask(); | |
26 } | |
27 | |
28 void DidProcessTask(const base::PendingTask& pending_task) override { | |
29 observer_->didProcessTask(); | |
30 } | |
31 | |
32 private: | |
33 WebThread::TaskObserver* observer_; | |
34 }; | |
35 | |
36 WebThreadBase::WebThreadBase() { | |
37 } | |
38 | |
39 WebThreadBase::~WebThreadBase() { | |
40 for (auto& observer_entry : task_observer_map_) { | |
41 delete observer_entry.second; | |
42 } | |
43 } | |
44 | |
45 void WebThreadBase::addTaskObserver(TaskObserver* observer) { | |
46 CHECK(isCurrentThread()); | |
47 std::pair<TaskObserverMap::iterator, bool> result = task_observer_map_.insert( | |
48 std::make_pair(observer, static_cast<TaskObserverAdapter*>(NULL))); | |
49 if (result.second) | |
50 result.first->second = new TaskObserverAdapter(observer); | |
51 AddTaskObserverInternal(result.first->second); | |
52 } | |
53 | |
54 void WebThreadBase::removeTaskObserver(TaskObserver* observer) { | |
55 CHECK(isCurrentThread()); | |
56 TaskObserverMap::iterator iter = task_observer_map_.find(observer); | |
57 if (iter == task_observer_map_.end()) | |
58 return; | |
59 RemoveTaskObserverInternal(iter->second); | |
60 delete iter->second; | |
61 task_observer_map_.erase(iter); | |
62 } | |
63 | |
64 void WebThreadBase::AddTaskObserverInternal( | |
65 base::MessageLoop::TaskObserver* observer) { | |
66 base::MessageLoop::current()->AddTaskObserver(observer); | |
67 } | |
68 | |
69 void WebThreadBase::RemoveTaskObserverInternal( | |
70 base::MessageLoop::TaskObserver* observer) { | |
71 base::MessageLoop::current()->RemoveTaskObserver(observer); | |
72 } | |
73 | |
74 // RunWebThreadTask takes the ownership of |task| from base::Closure and | |
75 // deletes it on the first invocation of the closure for thread-safety. | |
76 // base::Closure made from RunWebThreadTask is copyable but Closure::Run | |
77 // should be called at most only once. | |
78 // This is because WebThread::Task can contain RefPtr to a | |
79 // thread-unsafe-reference-counted object (e.g. WorkerThreadTask can contain | |
80 // RefPtr to WebKit's StringImpl), and if we don't delete |task| here, | |
81 // it causes a race condition as follows: | |
82 // [A] In task->run(), more RefPtr's to the refcounted object can be created, | |
83 // and the reference counter of the object can be modified via these | |
84 // RefPtr's (as intended) on the thread where the task is executed. | |
85 // [B] However, base::Closure still retains the ownership of WebThread::Task | |
86 // even after RunWebThreadTask is called. | |
87 // When base::Closure is deleted, WebThread::Task is deleted and the | |
88 // reference counter of the object is decreased by one, possibly from a | |
89 // different thread from [A], which is a race condition. | |
90 // Taking the ownership of |task| here by using scoped_ptr and base::Passed | |
91 // removes the reference counter modification of [B] and the race condition. | |
92 // When the closure never runs at all, the corresponding WebThread::Task is | |
93 // destructed when base::Closure is deleted (like [B]). In this case, there | |
94 // are no reference counter modification like [A] (because task->run() is not | |
95 // executed), so there are no race conditions. | |
96 // See https://crbug.com/390851 for more details. | |
97 // | |
98 // static | |
99 void WebThreadBase::RunWebThreadTask(scoped_ptr<blink::WebThread::Task> task) { | |
100 task->run(); | |
101 } | |
102 | |
103 void WebThreadBase::postTask(const blink::WebTraceLocation& location, | |
104 Task* task) { | |
105 postDelayedTask(location, task, 0); | |
106 } | |
107 | |
108 void WebThreadBase::postDelayedTask(const blink::WebTraceLocation& web_location, | |
109 Task* task, | |
110 long long delay_ms) { | |
111 tracked_objects::Location location(web_location.functionName(), | |
112 web_location.fileName(), -1, nullptr); | |
113 TaskRunner()->PostDelayedTask( | |
114 location, | |
115 base::Bind(RunWebThreadTask, base::Passed(make_scoped_ptr(task))), | |
116 base::TimeDelta::FromMilliseconds(delay_ms)); | |
117 } | |
118 | |
119 void WebThreadBase::enterRunLoop() { | |
120 CHECK(isCurrentThread()); | |
121 CHECK(MessageLoop()); | |
122 CHECK(!MessageLoop()->is_running()); // We don't support nesting. | |
123 MessageLoop()->Run(); | |
124 } | |
125 | |
126 void WebThreadBase::exitRunLoop() { | |
127 CHECK(isCurrentThread()); | |
128 CHECK(MessageLoop()); | |
129 CHECK(MessageLoop()->is_running()); | |
130 MessageLoop()->Quit(); | |
131 } | |
132 | |
133 bool WebThreadBase::isCurrentThread() const { | |
134 return TaskRunner()->BelongsToCurrentThread(); | |
135 } | |
136 | |
137 blink::PlatformThreadId WebThreadImpl::threadId() const { | |
138 return thread_->thread_id(); | |
139 } | |
140 | |
141 WebThreadImpl::WebThreadImpl(const char* name) | |
142 : thread_(new base::Thread(name)) { | |
143 thread_->Start(); | |
144 } | |
145 | |
146 WebThreadImpl::~WebThreadImpl() { | |
147 thread_->Stop(); | |
148 } | |
149 | |
150 base::MessageLoop* WebThreadImpl::MessageLoop() const { | |
151 return nullptr; | |
152 } | |
153 | |
154 base::SingleThreadTaskRunner* WebThreadImpl::TaskRunner() const { | |
155 return thread_->message_loop_proxy().get(); | |
156 } | |
157 | |
158 } // namespace content | |
OLD | NEW |