| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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 "content/public/test/mock_render_process_host.h" | 5 #include "content/public/test/mock_render_process_host.h" |
| 6 | 6 |
| 7 #include "base/lazy_instance.h" | 7 #include "base/lazy_instance.h" |
| 8 #include "base/message_loop/message_loop.h" | 8 #include "base/message_loop/message_loop.h" |
| 9 #include "base/time/time.h" | 9 #include "base/time/time.h" |
| 10 #include "content/browser/child_process_security_policy_impl.h" | 10 #include "content/browser/child_process_security_policy_impl.h" |
| 11 #include "content/browser/renderer_host/render_process_host_impl.h" | 11 #include "content/browser/renderer_host/render_process_host_impl.h" |
| 12 #include "content/browser/renderer_host/render_view_host_impl.h" | 12 #include "content/browser/renderer_host/render_view_host_impl.h" |
| 13 #include "content/browser/renderer_host/render_widget_host_impl.h" | 13 #include "content/browser/renderer_host/render_widget_host_impl.h" |
| 14 #include "content/common/child_process_host_impl.h" | 14 #include "content/common/child_process_host_impl.h" |
| 15 #include "content/public/browser/global_request_id.h" | 15 #include "content/public/browser/global_request_id.h" |
| 16 #include "content/public/browser/render_widget_host_iterator.h" | 16 #include "content/public/browser/render_widget_host_iterator.h" |
| 17 #include "content/public/browser/storage_partition.h" | 17 #include "content/public/browser/storage_partition.h" |
| 18 | 18 |
| 19 namespace content { | 19 namespace content { |
| 20 | 20 |
| 21 MockRenderProcessHost::MockRenderProcessHost(BrowserContext* browser_context) | 21 MockRenderProcessHost::MockRenderProcessHost(BrowserContext* browser_context) |
| 22 : transport_dib_(NULL), | 22 : bad_msg_count_(0), |
| 23 bad_msg_count_(0), | |
| 24 factory_(NULL), | 23 factory_(NULL), |
| 25 id_(ChildProcessHostImpl::GenerateChildProcessUniqueId()), | 24 id_(ChildProcessHostImpl::GenerateChildProcessUniqueId()), |
| 26 browser_context_(browser_context), | 25 browser_context_(browser_context), |
| 27 prev_routing_id_(0), | 26 prev_routing_id_(0), |
| 28 fast_shutdown_started_(false), | 27 fast_shutdown_started_(false), |
| 29 deletion_callback_called_(false), | 28 deletion_callback_called_(false), |
| 30 is_guest_(false) { | 29 is_guest_(false) { |
| 31 // Child process security operations can't be unit tested unless we add | 30 // Child process security operations can't be unit tested unless we add |
| 32 // ourselves as an existing child process. | 31 // ourselves as an existing child process. |
| 33 ChildProcessSecurityPolicyImpl::GetInstance()->Add(GetID()); | 32 ChildProcessSecurityPolicyImpl::GetInstance()->Add(GetID()); |
| 34 | 33 |
| 35 RenderProcessHostImpl::RegisterHost(GetID(), this); | 34 RenderProcessHostImpl::RegisterHost(GetID(), this); |
| 36 } | 35 } |
| 37 | 36 |
| 38 MockRenderProcessHost::~MockRenderProcessHost() { | 37 MockRenderProcessHost::~MockRenderProcessHost() { |
| 39 ChildProcessSecurityPolicyImpl::GetInstance()->Remove(GetID()); | 38 ChildProcessSecurityPolicyImpl::GetInstance()->Remove(GetID()); |
| 40 delete transport_dib_; | |
| 41 if (factory_) | 39 if (factory_) |
| 42 factory_->Remove(this); | 40 factory_->Remove(this); |
| 43 | 41 |
| 44 // In unit tests, Cleanup() might not have been called. | 42 // In unit tests, Cleanup() might not have been called. |
| 45 if (!deletion_callback_called_) { | 43 if (!deletion_callback_called_) { |
| 46 FOR_EACH_OBSERVER(RenderProcessHostObserver, | 44 FOR_EACH_OBSERVER(RenderProcessHostObserver, |
| 47 observers_, | 45 observers_, |
| 48 RenderProcessHostDestroyed(this)); | 46 RenderProcessHostDestroyed(this)); |
| 49 RenderProcessHostImpl::UnregisterHost(GetID()); | 47 RenderProcessHostImpl::UnregisterHost(GetID()); |
| 50 } | 48 } |
| (...skipping 83 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 134 return base::Process::Current().handle(); | 132 return base::Process::Current().handle(); |
| 135 } | 133 } |
| 136 | 134 |
| 137 bool MockRenderProcessHost::Send(IPC::Message* msg) { | 135 bool MockRenderProcessHost::Send(IPC::Message* msg) { |
| 138 // Save the message in the sink. | 136 // Save the message in the sink. |
| 139 sink_.OnMessageReceived(*msg); | 137 sink_.OnMessageReceived(*msg); |
| 140 delete msg; | 138 delete msg; |
| 141 return true; | 139 return true; |
| 142 } | 140 } |
| 143 | 141 |
| 144 TransportDIB* MockRenderProcessHost::MapTransportDIB(TransportDIB::Id dib_id) { | |
| 145 #if defined(OS_WIN) | |
| 146 // NULL should be used here instead of INVALID_HANDLE_VALUE (or pseudo-handle) | |
| 147 // except for when dealing with the small number of Win16-derived APIs | |
| 148 // that require INVALID_HANDLE_VALUE (e.g. CreateFile and friends). | |
| 149 HANDLE duped = NULL; | |
| 150 if (!DuplicateHandle(GetCurrentProcess(), dib_id.handle, GetCurrentProcess(), | |
| 151 &duped, 0, TRUE, DUPLICATE_SAME_ACCESS)) | |
| 152 duped = NULL; | |
| 153 return TransportDIB::Map(duped); | |
| 154 #elif defined(OS_ANDROID) | |
| 155 // On Android, Handles and Ids are the same underlying type. | |
| 156 return TransportDIB::Map(dib_id); | |
| 157 #else | |
| 158 // On POSIX, TransportDIBs are always created in the browser, so we cannot map | |
| 159 // one from a dib_id. | |
| 160 return TransportDIB::Create(100 * 100 * 4, 0); | |
| 161 #endif | |
| 162 } | |
| 163 | |
| 164 TransportDIB* MockRenderProcessHost::GetTransportDIB(TransportDIB::Id dib_id) { | |
| 165 if (transport_dib_) | |
| 166 return transport_dib_; | |
| 167 | |
| 168 transport_dib_ = MapTransportDIB(dib_id); | |
| 169 return transport_dib_; | |
| 170 } | |
| 171 | |
| 172 int MockRenderProcessHost::GetID() const { | 142 int MockRenderProcessHost::GetID() const { |
| 173 return id_; | 143 return id_; |
| 174 } | 144 } |
| 175 | 145 |
| 176 bool MockRenderProcessHost::HasConnection() const { | 146 bool MockRenderProcessHost::HasConnection() const { |
| 177 return true; | 147 return true; |
| 178 } | 148 } |
| 179 | 149 |
| 180 void MockRenderProcessHost::SetIgnoreInputEvents(bool ignore_input_events) { | 150 void MockRenderProcessHost::SetIgnoreInputEvents(bool ignore_input_events) { |
| 181 } | 151 } |
| (...skipping 126 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 308 for (ScopedVector<MockRenderProcessHost>::iterator it = processes_.begin(); | 278 for (ScopedVector<MockRenderProcessHost>::iterator it = processes_.begin(); |
| 309 it != processes_.end(); ++it) { | 279 it != processes_.end(); ++it) { |
| 310 if (*it == host) { | 280 if (*it == host) { |
| 311 processes_.weak_erase(it); | 281 processes_.weak_erase(it); |
| 312 break; | 282 break; |
| 313 } | 283 } |
| 314 } | 284 } |
| 315 } | 285 } |
| 316 | 286 |
| 317 } // content | 287 } // content |
| OLD | NEW |