| OLD | NEW |
| 1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 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 "services/shell/public/cpp/interface_provider.h" | 5 #include "services/shell/public/cpp/interface_provider.h" |
| 6 | 6 |
| 7 #include "base/macros.h" |
| 8 #include "mojo/public/cpp/bindings/strong_binding.h" |
| 9 |
| 7 namespace shell { | 10 namespace shell { |
| 8 | 11 |
| 9 InterfaceProvider::InterfaceProvider() : weak_factory_(this) { | 12 InterfaceProvider::InterfaceProvider() : weak_factory_(this) { |
| 10 pending_request_ = GetProxy(&interface_provider_); | 13 pending_request_ = GetProxy(&interface_provider_); |
| 11 } | 14 } |
| 15 |
| 12 InterfaceProvider::~InterfaceProvider() {} | 16 InterfaceProvider::~InterfaceProvider() {} |
| 13 | 17 |
| 14 void InterfaceProvider::Bind(mojom::InterfaceProviderPtr interface_provider) { | 18 void InterfaceProvider::Bind(mojom::InterfaceProviderPtr interface_provider) { |
| 15 DCHECK(pending_request_.is_pending()); | 19 DCHECK(pending_request_.is_pending()); |
| 20 DCHECK(forward_callback_.is_null()); |
| 16 mojo::FuseInterface(std::move(pending_request_), | 21 mojo::FuseInterface(std::move(pending_request_), |
| 17 interface_provider.PassInterface()); | 22 interface_provider.PassInterface()); |
| 18 } | 23 } |
| 19 | 24 |
| 25 void InterfaceProvider::Forward(const ForwardCallback& callback) { |
| 26 DCHECK(pending_request_.is_pending()); |
| 27 DCHECK(forward_callback_.is_null()); |
| 28 interface_provider_.reset(); |
| 29 pending_request_.PassMessagePipe().reset(); |
| 30 forward_callback_ = callback; |
| 31 } |
| 32 |
| 20 void InterfaceProvider::SetConnectionLostClosure( | 33 void InterfaceProvider::SetConnectionLostClosure( |
| 21 const base::Closure& connection_lost_closure) { | 34 const base::Closure& connection_lost_closure) { |
| 22 interface_provider_.set_connection_error_handler(connection_lost_closure); | 35 interface_provider_.set_connection_error_handler(connection_lost_closure); |
| 23 } | 36 } |
| 24 | 37 |
| 25 base::WeakPtr<InterfaceProvider> InterfaceProvider::GetWeakPtr() { | 38 base::WeakPtr<InterfaceProvider> InterfaceProvider::GetWeakPtr() { |
| 26 return weak_factory_.GetWeakPtr(); | 39 return weak_factory_.GetWeakPtr(); |
| 27 } | 40 } |
| 28 | 41 |
| 29 void InterfaceProvider::GetInterface( | 42 void InterfaceProvider::GetInterface( |
| 30 const std::string& name, | 43 const std::string& name, |
| 31 mojo::ScopedMessagePipeHandle request_handle) { | 44 mojo::ScopedMessagePipeHandle request_handle) { |
| 32 // Local binders can be registered via TestApi. | 45 // Local binders can be registered via TestApi. |
| 33 auto it = binders_.find(name); | 46 auto it = binders_.find(name); |
| 34 if (it != binders_.end()) { | 47 if (it != binders_.end()) { |
| 35 it->second.Run(std::move(request_handle)); | 48 it->second.Run(std::move(request_handle)); |
| 36 return; | 49 return; |
| 37 } | 50 } |
| 38 interface_provider_->GetInterface(name, std::move(request_handle)); | 51 |
| 52 if (!forward_callback_.is_null()) { |
| 53 DCHECK(!interface_provider_.is_bound()); |
| 54 forward_callback_.Run(name, std::move(request_handle)); |
| 55 } else { |
| 56 DCHECK(interface_provider_.is_bound()); |
| 57 interface_provider_->GetInterface(name, std::move(request_handle)); |
| 58 } |
| 39 } | 59 } |
| 40 | 60 |
| 41 void InterfaceProvider::ClearBinders() { | 61 void InterfaceProvider::ClearBinders() { |
| 42 binders_.clear(); | 62 binders_.clear(); |
| 43 } | 63 } |
| 44 | 64 |
| 45 } // namespace shell | 65 } // namespace shell |
| OLD | NEW |