src/net/dns/mdns_client.h

   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
   3 // found in the LICENSE file.
   4
   5 #ifndef NET_DNS_MDNS_CLIENT_H_
   6 #define NET_DNS_MDNS_CLIENT_H_
   7
   8 #include <string>
   9 #include <vector>
  10
  11 #include "base/callback.h"
  12 #include "net/base/ip_endpoint.h"
  13 #include "net/dns/dns_query.h"
  14 #include "net/dns/dns_response.h"
  15 #include "net/dns/record_parsed.h"
  16
  17 namespace net {
  18
  19 class DatagramServerSocket;
  20 class RecordParsed;
  21
  22 // Represents a one-time record lookup. A transaction takes one
  23 // associated callback (see |MDnsClient::CreateTransaction|) and calls it
  24 // whenever a matching record has been found, either from the cache or
  25 // by querying the network (it may choose to query either or both based on its
  26 // creation flags, see MDnsTransactionFlags). Network-based transactions will
  27 // time out after a reasonable number of seconds.
  28 class NET_EXPORT MDnsTransaction {
  29  public:
  30   // Used to signify what type of result the transaction has recieved.
  31   enum Result {
  32     // Passed whenever a record is found.
  33     RESULT_RECORD,
  34     // The transaction is done. Applies to non-single-valued transactions. Is
  35     // called when the transaction has finished (this is the last call to the
  36     // callback).
  37     RESULT_DONE,
  38     // No results have been found. Applies to single-valued transactions. Is
  39     // called when the transaction has finished without finding any results.
  40     // For transactions that use the network, this happens when a timeout
  41     // occurs, for transactions that are cache-only, this happens when no
  42     // results are in the cache.
  43     RESULT_NO_RESULTS,
  44     // Called when an NSec record is read for this transaction's
  45     // query. This means there cannot possibly be a record of the type
  46     // and name for this transaction.
  47     RESULT_NSEC
  48   };
  49
  50   // Used when creating an MDnsTransaction.
  51   enum Flags {
  52     // Transaction should return only one result, and stop listening after it.
  53     // Note that single result transactions will signal when their timeout is
  54     // reached, whereas multi-result transactions will not.
  55     SINGLE_RESULT = 1 << 0,
  56     // Query the cache or the network. May both be used. One must be present.
  57     QUERY_CACHE = 1 << 1,
  58     QUERY_NETWORK = 1 << 2,
  59     // TODO(noamsml): Add flag for flushing cache when feature is implemented
  60     // Mask of all possible flags on MDnsTransaction.
  61     FLAG_MASK = (1 << 3) - 1,
  62   };
  63
  64   typedef base::Callback<void(Result, const RecordParsed*)>
  65   ResultCallback;
  66
  67   // Destroying the transaction cancels it.
  68   virtual ~MDnsTransaction() {}
  69
  70   // Start the transaction. Return true on success. Cache-based transactions
  71   // will execute the callback synchronously.
  72   virtual bool Start() = 0;
  73
  74   // Get the host or service name for the transaction.
  75   virtual const std::string& GetName() const = 0;
  76
  77   // Get the type for this transaction (SRV, TXT, A, AAA, etc)
  78   virtual uint16 GetType() const = 0;
  79 };
  80
  81 // A listener listens for updates regarding a specific record or set of records.
  82 // Created by the MDnsClient (see |MDnsClient::CreateListener|) and used to keep
  83 // track of listeners.
  84 class NET_EXPORT MDnsListener {
  85  public:
  86   // Used in the MDnsListener delegate to signify what type of change has been
  87   // made to a record.
  88   enum UpdateType {
  89     RECORD_ADDED,
  90     RECORD_CHANGED,
  91     RECORD_REMOVED
  92   };
  93
  94   class Delegate {
  95    public:
  96     virtual ~Delegate() {}
  97
  98     // Called when a record is added, removed or updated.
  99     virtual void OnRecordUpdate(UpdateType update,
 100                                 const RecordParsed* record) = 0;
 101
 102     // Called when a record is marked nonexistent by an NSEC record.
 103     virtual void OnNsecRecord(const std::string& name, unsigned type) = 0;
 104
 105     // Called when the cache is purged (due, for example, ot the network
 106     // disconnecting).
 107     virtual void OnCachePurged() = 0;
 108   };
 109
 110   // Destroying the listener stops listening.
 111   virtual ~MDnsListener() {}
 112
 113   // Start the listener. Return true on success.
 114   virtual bool Start() = 0;
 115
 116   // Get the host or service name for this query.
 117   // Return an empty string for no name.
 118   virtual const std::string& GetName() const = 0;
 119
 120   // Get the type for this query (SRV, TXT, A, AAA, etc)
 121   virtual uint16 GetType() const = 0;
 122 };
 123
 124 // Creates bound datagram sockets ready to use by MDnsClient.
 125 class NET_EXPORT MDnsSocketFactory {
 126  public:
 127   virtual ~MDnsSocketFactory() {}
 128   virtual void CreateSockets(ScopedVector<DatagramServerSocket>* sockets) = 0;
 129
 130   static scoped_ptr<MDnsSocketFactory> CreateDefault();
 131 };
 132
 133 // Listens for Multicast DNS on the local network. You can access information
 134 // regarding multicast DNS either by creating an |MDnsListener| to be notified
 135 // of new records, or by creating an |MDnsTransaction| to look up the value of a
 136 // specific records. When all listeners and active transactions are destroyed,
 137 // the client stops listening on the network and destroys the cache.
 138 class NET_EXPORT MDnsClient {
 139  public:
 140   virtual ~MDnsClient() {}
 141
 142   // Create listener object for RRType |rrtype| and name |name|.
 143   virtual scoped_ptr<MDnsListener> CreateListener(
 144       uint16 rrtype,
 145       const std::string& name,
 146       MDnsListener::Delegate* delegate) = 0;
 147
 148   // Create a transaction that can be used to query either the MDns cache, the
 149   // network, or both for records of type |rrtype| and name |name|. |flags| is
 150   // defined by MDnsTransactionFlags.
 151   virtual scoped_ptr<MDnsTransaction> CreateTransaction(
 152       uint16 rrtype,
 153       const std::string& name,
 154       int flags,
 155       const MDnsTransaction::ResultCallback& callback) = 0;
 156
 157   virtual bool StartListening(MDnsSocketFactory* factory) = 0;
 158
 159   // Do not call this inside callbacks from related MDnsListener and
 160   // MDnsTransaction objects.
 161   virtual void StopListening() = 0;
 162   virtual bool IsListening() const = 0;
 163
 164   // Create the default MDnsClient
 165   static scoped_ptr<MDnsClient> CreateDefault();
 166 };
 167
 168 NET_EXPORT IPEndPoint GetMDnsIPEndPoint(AddressFamily address_family);
 169
 170 typedef std::vector<std::pair<uint32, AddressFamily> > InterfaceIndexFamilyList;
 171 // Returns pairs of interface and address family to bind. Current
 172 // implementation returns unique list of all available interfaces.
 173 NET_EXPORT InterfaceIndexFamilyList GetMDnsInterfacesToBind();
 174
 175 // Create sockets, binds socket to MDns endpoint, and sets multicast interface
 176 // and joins multicast group on for |interface_index|.
 177 // Returns NULL if failed.
 178 NET_EXPORT scoped_ptr<DatagramServerSocket> CreateAndBindMDnsSocket(
 179     AddressFamily address_family,
 180     uint32 interface_index);
 181
 182 }  // namespace net
 183
 184 #endif  // NET_DNS_MDNS_CLIENT_H_