Chromium Code Reviews| OLD | NEW |
|---|---|
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 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 #ifndef CHROME_BROWSER_LOCAL_DISCOVERY_SERVICE_DISCOVERY_HOST_CLIENT_H_ | 5 #ifndef CHROME_BROWSER_LOCAL_DISCOVERY_SERVICE_DISCOVERY_HOST_CLIENT_H_ |
| 6 #define CHROME_BROWSER_LOCAL_DISCOVERY_SERVICE_DISCOVERY_HOST_CLIENT_H_ | 6 #define CHROME_BROWSER_LOCAL_DISCOVERY_SERVICE_DISCOVERY_HOST_CLIENT_H_ |
| 7 | 7 |
| 8 #include <map> | 8 #include <map> |
| 9 #include <string> | 9 #include <string> |
| 10 | 10 |
| 11 #include "base/memory/singleton.h" | 11 #include "base/memory/singleton.h" |
| 12 #include "base/threading/non_thread_safe.h" | 12 #include "base/threading/non_thread_safe.h" |
| 13 #include "chrome/common/local_discovery/service_discovery_client.h" | 13 #include "chrome/common/local_discovery/service_discovery_client.h" |
| 14 #include "content/public/browser/utility_process_host_client.h" | 14 #include "content/public/browser/utility_process_host_client.h" |
| 15 #include "net/base/network_change_notifier.h" | |
| 15 | 16 |
| 16 namespace base { | 17 namespace base { |
| 17 class TaskRunner; | 18 class TaskRunner; |
| 18 } | 19 } |
| 19 | 20 |
| 20 namespace content { | 21 namespace content { |
| 21 class UtilityProcessHost; | 22 class UtilityProcessHost; |
| 22 } | 23 } |
| 23 | 24 |
| 24 namespace local_discovery { | 25 namespace local_discovery { |
| 25 | 26 |
| 26 // Implementation of ServiceDiscoveryClient that delegates all functionality to | 27 // Implementation of ServiceDiscoveryClient that delegates all functionality to |
| 27 // utility process. | 28 // utility process. |
| 28 class ServiceDiscoveryHostClient : public base::NonThreadSafe, | 29 class ServiceDiscoveryHostClient |
| 29 public ServiceDiscoveryClient, | 30 : public base::NonThreadSafe, |
| 30 public content::UtilityProcessHostClient { | 31 public ServiceDiscoveryClient, |
| 32 public content::UtilityProcessHostClient, | |
| 33 public net::NetworkChangeNotifier::IPAddressObserver { | |
| 31 public: | 34 public: |
| 32 ServiceDiscoveryHostClient(); | 35 ServiceDiscoveryHostClient(); |
| 33 | 36 |
| 34 // Starts utility process with ServiceDiscoveryClient. | 37 // Starts utility process with ServiceDiscoveryClient. |
| 35 void Start(); | 38 void Start(); |
| 36 | 39 |
| 37 // Shutdowns utility process. | 40 // Shutdowns utility process. |
| 38 void Shutdown(); | 41 void Shutdown(); |
| 39 | 42 |
| 40 // ServiceDiscoveryClient implementation. | 43 // ServiceDiscoveryClient implementation. |
| 41 virtual scoped_ptr<ServiceWatcher> CreateServiceWatcher( | 44 virtual scoped_ptr<ServiceWatcher> CreateServiceWatcher( |
| 42 const std::string& service_type, | 45 const std::string& service_type, |
| 43 const ServiceWatcher::UpdatedCallback& callback) OVERRIDE; | 46 const ServiceWatcher::UpdatedCallback& callback) OVERRIDE; |
| 44 virtual scoped_ptr<ServiceResolver> CreateServiceResolver( | 47 virtual scoped_ptr<ServiceResolver> CreateServiceResolver( |
| 45 const std::string& service_name, | 48 const std::string& service_name, |
| 46 const ServiceResolver::ResolveCompleteCallback& callback) OVERRIDE; | 49 const ServiceResolver::ResolveCompleteCallback& callback) OVERRIDE; |
| 47 virtual scoped_ptr<LocalDomainResolver> CreateLocalDomainResolver( | 50 virtual scoped_ptr<LocalDomainResolver> CreateLocalDomainResolver( |
| 48 const std::string& domain, | 51 const std::string& domain, |
| 49 net::AddressFamily address_family, | 52 net::AddressFamily address_family, |
| 50 const LocalDomainResolver::IPAddressCallback& callback) OVERRIDE; | 53 const LocalDomainResolver::IPAddressCallback& callback) OVERRIDE; |
| 51 | 54 |
| 52 // UtilityProcessHostClient implementation. | 55 // UtilityProcessHostClient implementation. |
| 53 virtual bool OnMessageReceived(const IPC::Message& message) OVERRIDE; | 56 virtual bool OnMessageReceived(const IPC::Message& message) OVERRIDE; |
| 54 | 57 |
| 58 // net::NetworkChangeNotifier::IPAddressObserver implementation. | |
| 59 virtual void OnIPAddressChanged() OVERRIDE; | |
| 60 | |
| 55 protected: | 61 protected: |
| 56 virtual ~ServiceDiscoveryHostClient(); | 62 virtual ~ServiceDiscoveryHostClient(); |
| 57 | 63 |
| 58 private: | 64 private: |
| 59 class ServiceWatcherProxy; | 65 class ServiceWatcherProxy; |
| 60 class ServiceResolverProxy; | 66 class ServiceResolverProxy; |
| 61 class LocalDomainResolverProxy; | 67 class LocalDomainResolverProxy; |
| 62 | 68 |
| 63 typedef std::map<uint64, ServiceWatcher::UpdatedCallback> WatcherCallbacks; | 69 typedef std::map<uint64, ServiceWatcherProxy*> WatcherProxies; |
| 64 typedef std::map<uint64, ServiceResolver::ResolveCompleteCallback> | 70 typedef std::map<uint64, ServiceResolverProxy*> |
|
Vitaly Buka (NO REVIEWS)
2013/09/06 04:59:27
fits one line
| |
| 65 ResolverCallbacks; | 71 ResolverProxies; |
| 66 typedef std::map<uint64, LocalDomainResolver::IPAddressCallback> | 72 typedef std::map<uint64, LocalDomainResolverProxy*> |
|
Vitaly Buka (NO REVIEWS)
2013/09/06 04:59:27
maybe fits one line
| |
| 67 DomainResolverCallbacks; | 73 DomainResolverProxies; |
| 68 | 74 |
| 69 void StartOnIOThread(); | 75 void StartOnIOThread(); |
| 70 void ShutdownOnIOThread(); | 76 void ShutdownOnIOThread(); |
| 77 void RestartOnIOThread(); | |
| 78 | |
| 79 void StartUtilityProcess(); | |
| 80 void ShutdownUtilityProcess(); | |
| 71 | 81 |
| 72 void Send(IPC::Message* msg); | 82 void Send(IPC::Message* msg); |
| 73 void SendOnIOThread(IPC::Message* msg); | 83 void SendOnIOThread(IPC::Message* msg); |
| 74 | 84 |
| 75 uint64 RegisterWatcherCallback( | 85 uint64 RegisterWatcherProxy( |
| 76 const ServiceWatcher::UpdatedCallback& callback); | 86 ServiceWatcherProxy* proxy); |
| 77 uint64 RegisterResolverCallback( | 87 uint64 RegisterResolverProxy( |
|
Vitaly Buka (NO REVIEWS)
2013/09/06 04:59:27
same
| |
| 78 const ServiceResolver::ResolveCompleteCallback& callback); | 88 ServiceResolverProxy* proxy); |
| 79 uint64 RegisterLocalDomainResolverCallback( | 89 uint64 RegisterLocalDomainResolverProxy( |
| 80 const LocalDomainResolver::IPAddressCallback& callback); | 90 LocalDomainResolverProxy* proxy); |
| 81 | 91 |
| 82 void UnregisterWatcherCallback(uint64 id); | 92 void UnregisterWatcherProxy(uint64 id); |
| 83 void UnregisterResolverCallback(uint64 id); | 93 void UnregisterResolverProxy(uint64 id); |
| 84 void UnregisterLocalDomainResolverCallback(uint64 id); | 94 void UnregisterLocalDomainResolverProxy(uint64 id); |
| 85 | 95 |
| 86 // IPC Message handlers. | 96 // IPC Message handlers. |
| 87 void OnWatcherCallback(uint64 id, | 97 void OnWatcherCallback(uint64 id, |
| 88 ServiceWatcher::UpdateType update, | 98 ServiceWatcher::UpdateType update, |
| 89 const std::string& service_name); | 99 const std::string& service_name); |
| 90 void OnResolverCallback(uint64 id, | 100 void OnResolverCallback(uint64 id, |
| 91 ServiceResolver::RequestStatus status, | 101 ServiceResolver::RequestStatus status, |
| 92 const ServiceDescription& description); | 102 const ServiceDescription& description); |
| 93 void OnLocalDomainResolverCallback(uint64 id, | 103 void OnLocalDomainResolverCallback(uint64 id, |
| 94 bool success, | 104 bool success, |
| 95 const net::IPAddressNumber& address_ipv4, | 105 const net::IPAddressNumber& address_ipv4, |
| 96 const net::IPAddressNumber& address_ipv6); | 106 const net::IPAddressNumber& address_ipv6); |
| 97 | 107 |
| 98 | 108 |
| 99 // Runs watcher callback on owning thread. | 109 // Runs watcher callback on owning thread. |
| 100 void RunWatcherCallback(uint64 id, | 110 void RunWatcherCallback(uint64 id, |
| 101 ServiceWatcher::UpdateType update, | 111 ServiceWatcher::UpdateType update, |
| 102 const std::string& service_name); | 112 const std::string& service_name); |
| 103 // Runs resolver callback on owning thread. | 113 // Runs resolver callback on owning thread. |
| 104 void RunResolverCallback(uint64 id, | 114 void RunResolverCallback(uint64 id, |
| 105 ServiceResolver::RequestStatus status, | 115 ServiceResolver::RequestStatus status, |
| 106 const ServiceDescription& description); | 116 const ServiceDescription& description); |
| 107 // Runs local domain resolver callback on owning thread. | 117 // Runs local domain resolver callback on owning thread. |
| 108 void RunLocalDomainResolverCallback(uint64 id, | 118 void RunLocalDomainResolverCallback(uint64 id, |
| 109 bool success, | 119 bool success, |
| 110 const net::IPAddressNumber& address_ipv4, | 120 const net::IPAddressNumber& address_ipv4, |
| 111 const net::IPAddressNumber& address_ipv6); | 121 const net::IPAddressNumber& address_ipv6); |
| 112 | 122 |
| 113 | |
| 114 base::WeakPtr<content::UtilityProcessHost> utility_host_; | 123 base::WeakPtr<content::UtilityProcessHost> utility_host_; |
| 115 | 124 |
| 116 // Incrementing counter to assign ID to watchers and resolvers. | 125 // Incrementing counter to assign ID to watchers and resolvers. |
| 117 uint64 current_id_; | 126 uint64 current_id_; |
| 118 WatcherCallbacks service_watcher_callbacks_; | 127 WatcherProxies service_watcher_proxies_; |
| 119 ResolverCallbacks service_resolver_callbacks_; | 128 ResolverProxies service_resolver_proxies_; |
| 120 DomainResolverCallbacks domain_resolver_callbacks_; | 129 DomainResolverProxies domain_resolver_proxies_; |
| 121 scoped_refptr<base::TaskRunner> callback_runner_; | 130 scoped_refptr<base::TaskRunner> callback_runner_; |
| 122 | 131 |
| 123 DISALLOW_COPY_AND_ASSIGN(ServiceDiscoveryHostClient); | 132 DISALLOW_COPY_AND_ASSIGN(ServiceDiscoveryHostClient); |
| 124 }; | 133 }; |
| 125 | 134 |
| 126 class ServiceDiscoveryHostClientFactory : public base::NonThreadSafe { | 135 class ServiceDiscoveryHostClientFactory : public base::NonThreadSafe { |
| 127 public: | 136 public: |
| 128 ServiceDiscoveryHostClientFactory(); | 137 ServiceDiscoveryHostClientFactory(); |
| 129 ~ServiceDiscoveryHostClientFactory(); | 138 ~ServiceDiscoveryHostClientFactory(); |
| 130 | 139 |
| 131 static ServiceDiscoveryHostClient* GetClient(); | 140 static ServiceDiscoveryHostClient* GetClient(); |
| 132 static void ReleaseClient(); | 141 static void ReleaseClient(); |
| 133 | 142 |
| 134 private: | 143 private: |
| 135 friend class Singleton<ServiceDiscoveryHostClientFactory>; | 144 friend class Singleton<ServiceDiscoveryHostClientFactory>; |
| 136 | 145 |
| 137 static ServiceDiscoveryHostClientFactory* GetInstance(); | 146 static ServiceDiscoveryHostClientFactory* GetInstance(); |
| 138 | 147 |
| 139 ServiceDiscoveryHostClient* GetClientInternal(); | 148 ServiceDiscoveryHostClient* GetClientInternal(); |
| 140 void ReleaseClientInternal(); | 149 void ReleaseClientInternal(); |
| 141 | 150 |
| 142 scoped_refptr<ServiceDiscoveryHostClient> instance_; | 151 scoped_refptr<ServiceDiscoveryHostClient> instance_; |
| 143 int references_; | 152 int references_; |
| 144 }; | 153 }; |
| 145 | 154 |
| 146 } // namespace local_discovery | 155 } // namespace local_discovery |
| 147 | 156 |
| 148 #endif // CHROME_BROWSER_LOCAL_DISCOVERY_SERVICE_DISCOVERY_HOST_CLIENT_H_ | 157 #endif // CHROME_BROWSER_LOCAL_DISCOVERY_SERVICE_DISCOVERY_HOST_CLIENT_H_ |
| OLD | NEW |