| 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 NET_DNS_MDNS_CLIENT_IMPL_H_ | 5 #ifndef NET_DNS_MDNS_CLIENT_IMPL_H_ |
| 6 #define NET_DNS_MDNS_CLIENT_IMPL_H_ | 6 #define NET_DNS_MDNS_CLIENT_IMPL_H_ |
| 7 | 7 |
| 8 #include <map> | 8 #include <map> |
| 9 #include <string> | 9 #include <string> |
| 10 #include <utility> | 10 #include <utility> |
| 11 #include <vector> | 11 #include <vector> |
| 12 | 12 |
| 13 #include "base/cancelable_callback.h" | 13 #include "base/cancelable_callback.h" |
| 14 #include "base/memory/scoped_vector.h" | 14 #include "base/memory/scoped_vector.h" |
| 15 #include "base/observer_list.h" | 15 #include "base/observer_list.h" |
| 16 #include "net/base/io_buffer.h" | 16 #include "net/base/io_buffer.h" |
| 17 #include "net/base/ip_endpoint.h" | 17 #include "net/base/ip_endpoint.h" |
| 18 #include "net/dns/mdns_cache.h" | 18 #include "net/dns/mdns_cache.h" |
| 19 #include "net/dns/mdns_client.h" | 19 #include "net/dns/mdns_client.h" |
| 20 #include "net/udp/datagram_server_socket.h" | 20 #include "net/udp/datagram_server_socket.h" |
| 21 #include "net/udp/udp_server_socket.h" | 21 #include "net/udp/udp_server_socket.h" |
| 22 #include "net/udp/udp_socket.h" | 22 #include "net/udp/udp_socket.h" |
| 23 | 23 |
| 24 namespace net { | 24 namespace net { |
| 25 | 25 |
| 26 class MDnsSocketFactoryImpl : public MDnsSocketFactory { | 26 class MDnsSocketFactoryImpl : public MDnsSocketFactory { |
| 27 public: | 27 public: |
| 28 MDnsSocketFactoryImpl() {}; | 28 MDnsSocketFactoryImpl() {}; |
| 29 virtual ~MDnsSocketFactoryImpl() {}; | 29 ~MDnsSocketFactoryImpl() override{}; |
| 30 | 30 |
| 31 virtual void CreateSockets( | 31 void CreateSockets(ScopedVector<DatagramServerSocket>* sockets) override; |
| 32 ScopedVector<DatagramServerSocket>* sockets) override; | |
| 33 | 32 |
| 34 private: | 33 private: |
| 35 DISALLOW_COPY_AND_ASSIGN(MDnsSocketFactoryImpl); | 34 DISALLOW_COPY_AND_ASSIGN(MDnsSocketFactoryImpl); |
| 36 }; | 35 }; |
| 37 | 36 |
| 38 // A connection to the network for multicast DNS clients. It reads data into | 37 // A connection to the network for multicast DNS clients. It reads data into |
| 39 // DnsResponse objects and alerts the delegate that a packet has been received. | 38 // DnsResponse objects and alerts the delegate that a packet has been received. |
| 40 class NET_EXPORT_PRIVATE MDnsConnection { | 39 class NET_EXPORT_PRIVATE MDnsConnection { |
| 41 public: | 40 public: |
| 42 class Delegate { | 41 class Delegate { |
| (...skipping 61 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 104 | 103 |
| 105 class NET_EXPORT_PRIVATE MDnsClientImpl : public MDnsClient { | 104 class NET_EXPORT_PRIVATE MDnsClientImpl : public MDnsClient { |
| 106 public: | 105 public: |
| 107 // The core object exists while the MDnsClient is listening, and is deleted | 106 // The core object exists while the MDnsClient is listening, and is deleted |
| 108 // whenever the number of listeners reaches zero. The deletion happens | 107 // whenever the number of listeners reaches zero. The deletion happens |
| 109 // asychronously, so destroying the last listener does not immediately | 108 // asychronously, so destroying the last listener does not immediately |
| 110 // invalidate the core. | 109 // invalidate the core. |
| 111 class Core : public base::SupportsWeakPtr<Core>, MDnsConnection::Delegate { | 110 class Core : public base::SupportsWeakPtr<Core>, MDnsConnection::Delegate { |
| 112 public: | 111 public: |
| 113 explicit Core(MDnsClientImpl* client); | 112 explicit Core(MDnsClientImpl* client); |
| 114 virtual ~Core(); | 113 ~Core() override; |
| 115 | 114 |
| 116 // Initialize the core. Returns true on success. | 115 // Initialize the core. Returns true on success. |
| 117 bool Init(MDnsSocketFactory* socket_factory); | 116 bool Init(MDnsSocketFactory* socket_factory); |
| 118 | 117 |
| 119 // Send a query with a specific rrtype and name. Returns true on success. | 118 // Send a query with a specific rrtype and name. Returns true on success. |
| 120 bool SendQuery(uint16 rrtype, std::string name); | 119 bool SendQuery(uint16 rrtype, std::string name); |
| 121 | 120 |
| 122 // Add/remove a listener to the list of listeners. | 121 // Add/remove a listener to the list of listeners. |
| 123 void AddListener(MDnsListenerImpl* listener); | 122 void AddListener(MDnsListenerImpl* listener); |
| 124 void RemoveListener(MDnsListenerImpl* listener); | 123 void RemoveListener(MDnsListenerImpl* listener); |
| 125 | 124 |
| 126 // Query the cache for records of a specific type and name. | 125 // Query the cache for records of a specific type and name. |
| 127 void QueryCache(uint16 rrtype, const std::string& name, | 126 void QueryCache(uint16 rrtype, const std::string& name, |
| 128 std::vector<const RecordParsed*>* records) const; | 127 std::vector<const RecordParsed*>* records) const; |
| 129 | 128 |
| 130 // Parse the response and alert relevant listeners. | 129 // Parse the response and alert relevant listeners. |
| 131 virtual void HandlePacket(DnsResponse* response, int bytes_read) override; | 130 void HandlePacket(DnsResponse* response, int bytes_read) override; |
| 132 | 131 |
| 133 virtual void OnConnectionError(int error) override; | 132 void OnConnectionError(int error) override; |
| 134 | 133 |
| 135 private: | 134 private: |
| 136 typedef std::pair<std::string, uint16> ListenerKey; | 135 typedef std::pair<std::string, uint16> ListenerKey; |
| 137 typedef std::map<ListenerKey, ObserverList<MDnsListenerImpl>* > | 136 typedef std::map<ListenerKey, ObserverList<MDnsListenerImpl>* > |
| 138 ListenerMap; | 137 ListenerMap; |
| 139 | 138 |
| 140 // Alert listeners of an update to the cache. | 139 // Alert listeners of an update to the cache. |
| 141 void AlertListeners(MDnsCache::UpdateType update_type, | 140 void AlertListeners(MDnsCache::UpdateType update_type, |
| 142 const ListenerKey& key, const RecordParsed* record); | 141 const ListenerKey& key, const RecordParsed* record); |
| 143 | 142 |
| (...skipping 19 matching lines...) Expand all Loading... |
| 163 | 162 |
| 164 base::CancelableClosure cleanup_callback_; | 163 base::CancelableClosure cleanup_callback_; |
| 165 base::Time scheduled_cleanup_; | 164 base::Time scheduled_cleanup_; |
| 166 | 165 |
| 167 scoped_ptr<MDnsConnection> connection_; | 166 scoped_ptr<MDnsConnection> connection_; |
| 168 | 167 |
| 169 DISALLOW_COPY_AND_ASSIGN(Core); | 168 DISALLOW_COPY_AND_ASSIGN(Core); |
| 170 }; | 169 }; |
| 171 | 170 |
| 172 MDnsClientImpl(); | 171 MDnsClientImpl(); |
| 173 virtual ~MDnsClientImpl(); | 172 ~MDnsClientImpl() override; |
| 174 | 173 |
| 175 // MDnsClient implementation: | 174 // MDnsClient implementation: |
| 176 virtual scoped_ptr<MDnsListener> CreateListener( | 175 scoped_ptr<MDnsListener> CreateListener( |
| 177 uint16 rrtype, | 176 uint16 rrtype, |
| 178 const std::string& name, | 177 const std::string& name, |
| 179 MDnsListener::Delegate* delegate) override; | 178 MDnsListener::Delegate* delegate) override; |
| 180 | 179 |
| 181 virtual scoped_ptr<MDnsTransaction> CreateTransaction( | 180 scoped_ptr<MDnsTransaction> CreateTransaction( |
| 182 uint16 rrtype, | 181 uint16 rrtype, |
| 183 const std::string& name, | 182 const std::string& name, |
| 184 int flags, | 183 int flags, |
| 185 const MDnsTransaction::ResultCallback& callback) override; | 184 const MDnsTransaction::ResultCallback& callback) override; |
| 186 | 185 |
| 187 virtual bool StartListening(MDnsSocketFactory* socket_factory) override; | 186 bool StartListening(MDnsSocketFactory* socket_factory) override; |
| 188 virtual void StopListening() override; | 187 void StopListening() override; |
| 189 virtual bool IsListening() const override; | 188 bool IsListening() const override; |
| 190 | 189 |
| 191 Core* core() { return core_.get(); } | 190 Core* core() { return core_.get(); } |
| 192 | 191 |
| 193 private: | 192 private: |
| 194 scoped_ptr<Core> core_; | 193 scoped_ptr<Core> core_; |
| 195 | 194 |
| 196 DISALLOW_COPY_AND_ASSIGN(MDnsClientImpl); | 195 DISALLOW_COPY_AND_ASSIGN(MDnsClientImpl); |
| 197 }; | 196 }; |
| 198 | 197 |
| 199 class MDnsListenerImpl : public MDnsListener, | 198 class MDnsListenerImpl : public MDnsListener, |
| 200 public base::SupportsWeakPtr<MDnsListenerImpl> { | 199 public base::SupportsWeakPtr<MDnsListenerImpl> { |
| 201 public: | 200 public: |
| 202 MDnsListenerImpl(uint16 rrtype, | 201 MDnsListenerImpl(uint16 rrtype, |
| 203 const std::string& name, | 202 const std::string& name, |
| 204 MDnsListener::Delegate* delegate, | 203 MDnsListener::Delegate* delegate, |
| 205 MDnsClientImpl* client); | 204 MDnsClientImpl* client); |
| 206 | 205 |
| 207 virtual ~MDnsListenerImpl(); | 206 ~MDnsListenerImpl() override; |
| 208 | 207 |
| 209 // MDnsListener implementation: | 208 // MDnsListener implementation: |
| 210 virtual bool Start() override; | 209 bool Start() override; |
| 211 | 210 |
| 212 // Actively refresh any received records. | 211 // Actively refresh any received records. |
| 213 virtual void SetActiveRefresh(bool active_refresh) override; | 212 void SetActiveRefresh(bool active_refresh) override; |
| 214 | 213 |
| 215 virtual const std::string& GetName() const override; | 214 const std::string& GetName() const override; |
| 216 | 215 |
| 217 virtual uint16 GetType() const override; | 216 uint16 GetType() const override; |
| 218 | 217 |
| 219 MDnsListener::Delegate* delegate() { return delegate_; } | 218 MDnsListener::Delegate* delegate() { return delegate_; } |
| 220 | 219 |
| 221 // Alert the delegate of a record update. | 220 // Alert the delegate of a record update. |
| 222 void HandleRecordUpdate(MDnsCache::UpdateType update_type, | 221 void HandleRecordUpdate(MDnsCache::UpdateType update_type, |
| 223 const RecordParsed* record_parsed); | 222 const RecordParsed* record_parsed); |
| 224 | 223 |
| 225 // Alert the delegate of the existence of an Nsec record. | 224 // Alert the delegate of the existence of an Nsec record. |
| 226 void AlertNsecRecord(); | 225 void AlertNsecRecord(); |
| 227 | 226 |
| (...skipping 17 matching lines...) Expand all Loading... |
| 245 | 244 |
| 246 class MDnsTransactionImpl : public base::SupportsWeakPtr<MDnsTransactionImpl>, | 245 class MDnsTransactionImpl : public base::SupportsWeakPtr<MDnsTransactionImpl>, |
| 247 public MDnsTransaction, | 246 public MDnsTransaction, |
| 248 public MDnsListener::Delegate { | 247 public MDnsListener::Delegate { |
| 249 public: | 248 public: |
| 250 MDnsTransactionImpl(uint16 rrtype, | 249 MDnsTransactionImpl(uint16 rrtype, |
| 251 const std::string& name, | 250 const std::string& name, |
| 252 int flags, | 251 int flags, |
| 253 const MDnsTransaction::ResultCallback& callback, | 252 const MDnsTransaction::ResultCallback& callback, |
| 254 MDnsClientImpl* client); | 253 MDnsClientImpl* client); |
| 255 virtual ~MDnsTransactionImpl(); | 254 ~MDnsTransactionImpl() override; |
| 256 | 255 |
| 257 // MDnsTransaction implementation: | 256 // MDnsTransaction implementation: |
| 258 virtual bool Start() override; | 257 bool Start() override; |
| 259 | 258 |
| 260 virtual const std::string& GetName() const override; | 259 const std::string& GetName() const override; |
| 261 virtual uint16 GetType() const override; | 260 uint16 GetType() const override; |
| 262 | 261 |
| 263 // MDnsListener::Delegate implementation: | 262 // MDnsListener::Delegate implementation: |
| 264 virtual void OnRecordUpdate(MDnsListener::UpdateType update, | 263 void OnRecordUpdate(MDnsListener::UpdateType update, |
| 265 const RecordParsed* record) override; | 264 const RecordParsed* record) override; |
| 266 virtual void OnNsecRecord(const std::string& name, unsigned type) override; | 265 void OnNsecRecord(const std::string& name, unsigned type) override; |
| 267 | 266 |
| 268 virtual void OnCachePurged() override; | 267 void OnCachePurged() override; |
| 269 | 268 |
| 270 private: | 269 private: |
| 271 bool is_active() { return !callback_.is_null(); } | 270 bool is_active() { return !callback_.is_null(); } |
| 272 | 271 |
| 273 void Reset(); | 272 void Reset(); |
| 274 | 273 |
| 275 // Trigger the callback and reset all related variables. | 274 // Trigger the callback and reset all related variables. |
| 276 void TriggerCallback(MDnsTransaction::Result result, | 275 void TriggerCallback(MDnsTransaction::Result result, |
| 277 const RecordParsed* record); | 276 const RecordParsed* record); |
| 278 | 277 |
| (...skipping 22 matching lines...) Expand all Loading... |
| 301 MDnsClientImpl* client_; | 300 MDnsClientImpl* client_; |
| 302 | 301 |
| 303 bool started_; | 302 bool started_; |
| 304 int flags_; | 303 int flags_; |
| 305 | 304 |
| 306 DISALLOW_COPY_AND_ASSIGN(MDnsTransactionImpl); | 305 DISALLOW_COPY_AND_ASSIGN(MDnsTransactionImpl); |
| 307 }; | 306 }; |
| 308 | 307 |
| 309 } // namespace net | 308 } // namespace net |
| 310 #endif // NET_DNS_MDNS_CLIENT_IMPL_H_ | 309 #endif // NET_DNS_MDNS_CLIENT_IMPL_H_ |
| OLD | NEW |