| // Copyright 2018 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #ifndef BASE_FUCHSIA_SCOPED_SERVICE_BINDING_H_ |
| #define BASE_FUCHSIA_SCOPED_SERVICE_BINDING_H_ |
| |
| #include <lib/fidl/cpp/binding.h> |
| #include <lib/fidl/cpp/binding_set.h> |
| |
| #include "base/base_export.h" |
| #include "base/callback.h" |
| #include "base/fuchsia/service_directory.h" |
| |
| namespace sys { |
| class OutgoingDirectory; |
| } // namespace sys |
| |
| namespace vfs { |
| class PseudoDir; |
| } // namespace vfs |
| |
| namespace base { |
| namespace fuchsia { |
| |
| namespace internal { |
| |
| class BASE_EXPORT ScopedServiceBindingBase { |
| public: |
| explicit ScopedServiceBindingBase(sys::OutgoingDirectory* outgoing_directory); |
| explicit ScopedServiceBindingBase(vfs::PseudoDir* pseudo_dir); |
| |
| ~ScopedServiceBindingBase(); |
| |
| protected: |
| // Same type as vfs::Service::Connector, so the value can be passed directly |
| // to vfs::Service. |
| using Connector = |
| fit::function<void(zx::channel channel, async_dispatcher_t* dispatcher)>; |
| |
| void RegisterService(const char* service_name, Connector connector); |
| void UnregisterService(const char* service_name); |
| |
| private: |
| vfs::PseudoDir* const pseudo_dir_ = nullptr; |
| }; |
| |
| } // namespace internal |
| |
| template <typename Interface> |
| class ScopedServiceBinding : public internal::ScopedServiceBindingBase { |
| public: |
| // Published a public service in the specified |outgoing_directory|. |
| // |outgoing_directory| and |impl| must outlive the binding. |
| ScopedServiceBinding(sys::OutgoingDirectory* outgoing_directory, |
| Interface* impl) |
| : ScopedServiceBindingBase(outgoing_directory), impl_(impl) { |
| RegisterService(Interface::Name_, |
| fit::bind_member(this, &ScopedServiceBinding::BindClient)); |
| } |
| |
| // Publishes a service in the specified |pseudo_dir|. |pseudo_dir| and |impl| |
| // must outlive the binding. |
| ScopedServiceBinding(vfs::PseudoDir* pseudo_dir, Interface* impl) |
| : ScopedServiceBindingBase(pseudo_dir), impl_(impl) { |
| RegisterService(Interface::Name_, |
| fit::bind_member(this, &ScopedServiceBinding::BindClient)); |
| } |
| |
| // TODO(crbug.com/974072): Remove this constructor once all code has been |
| // migrated from base::fuchsia::ServiceDirectory to sys::OutgoingDirectory. |
| ScopedServiceBinding(ServiceDirectory* service_directory, Interface* impl) |
| : ScopedServiceBinding(service_directory->outgoing_directory(), impl) {} |
| |
| ~ScopedServiceBinding() { UnregisterService(Interface::Name_); } |
| |
| void SetOnLastClientCallback(base::OnceClosure on_last_client_callback) { |
| on_last_client_callback_ = std::move(on_last_client_callback); |
| bindings_.set_empty_set_handler( |
| fit::bind_member(this, &ScopedServiceBinding::OnBindingSetEmpty)); |
| } |
| |
| bool has_clients() const { return bindings_.size() != 0; } |
| |
| private: |
| void BindClient(zx::channel channel, async_dispatcher_t* dispatcher) { |
| bindings_.AddBinding(impl_, |
| fidl::InterfaceRequest<Interface>(std::move(channel)), |
| dispatcher); |
| } |
| |
| void OnBindingSetEmpty() { |
| bindings_.set_empty_set_handler(nullptr); |
| std::move(on_last_client_callback_).Run(); |
| } |
| |
| sys::OutgoingDirectory* const directory_ = nullptr; |
| vfs::PseudoDir* const pseudo_dir_ = nullptr; |
| Interface* const impl_; |
| fidl::BindingSet<Interface> bindings_; |
| base::OnceClosure on_last_client_callback_; |
| |
| DISALLOW_COPY_AND_ASSIGN(ScopedServiceBinding); |
| }; |
| |
| // Scoped service binding which allows only a single client to be connected |
| // at any time. By default a new connection will disconnect an existing client. |
| enum class ScopedServiceBindingPolicy { kPreferNew, kPreferExisting }; |
| |
| template <typename Interface, |
| ScopedServiceBindingPolicy Policy = |
| ScopedServiceBindingPolicy::kPreferNew> |
| class ScopedSingleClientServiceBinding |
| : public internal::ScopedServiceBindingBase { |
| public: |
| // |outgoing_directory| and |impl| must outlive the binding. |
| ScopedSingleClientServiceBinding(sys::OutgoingDirectory* outgoing_directory, |
| Interface* impl) |
| : ScopedServiceBindingBase(outgoing_directory), binding_(impl) { |
| RegisterService( |
| Interface::Name_, |
| fit::bind_member(this, &ScopedSingleClientServiceBinding::BindClient)); |
| } |
| |
| // TODO(crbug.com/974072): Remove this constructor once all code has been |
| // migrated from base::fuchsia::ServiceDirectory to sys::OutgoingDirectory. |
| ScopedSingleClientServiceBinding(ServiceDirectory* service_directory, |
| Interface* impl) |
| : ScopedSingleClientServiceBinding( |
| service_directory->outgoing_directory(), |
| impl) {} |
| |
| ~ScopedSingleClientServiceBinding() { UnregisterService(Interface::Name_); } |
| |
| typename Interface::EventSender_& events() { return binding_.events(); } |
| |
| void SetOnLastClientCallback(base::OnceClosure on_last_client_callback) { |
| on_last_client_callback_ = std::move(on_last_client_callback); |
| binding_.set_error_handler(fit::bind_member( |
| this, &ScopedSingleClientServiceBinding::OnBindingEmpty)); |
| } |
| |
| bool has_clients() const { return binding_.is_bound(); } |
| |
| private: |
| void BindClient(zx::channel channel, async_dispatcher_t* dispatcher) { |
| if (Policy == ScopedServiceBindingPolicy::kPreferExisting && |
| binding_.is_bound()) { |
| return; |
| } |
| binding_.Bind(fidl::InterfaceRequest<Interface>(std::move(channel)), |
| dispatcher); |
| } |
| |
| void OnBindingEmpty() { |
| binding_.set_error_handler(nullptr); |
| std::move(on_last_client_callback_).Run(); |
| } |
| |
| fidl::Binding<Interface> binding_; |
| base::OnceClosure on_last_client_callback_; |
| |
| DISALLOW_COPY_AND_ASSIGN(ScopedSingleClientServiceBinding); |
| }; |
| |
| } // namespace fuchsia |
| } // namespace base |
| |
| #endif // BASE_FUCHSIA_SCOPED_SERVICE_BINDING_H_ |