Finished DNS-SD server. Finished Privet-specified DNS-SD server.

cloud_print/gcp20/prototype/dns_sd_server.h

Index: cloud_print/gcp20/prototype/dns_sd_server.h
diff --git a/cloud_print/gcp20/prototype/dns_sd_server.h b/cloud_print/gcp20/prototype/dns_sd_server.h
index 9c8ed96b7d33d62790d7797dc061f5b963911f2f..d5ef7931f457bf0afd2390917c758f11b41acd04 100644
--- a/cloud_print/gcp20/prototype/dns_sd_server.h
+++ b/cloud_print/gcp20/prototype/dns_sd_server.h
@@ -2,27 +2,40 @@
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
-#ifndef GCP20_PROTOTYPE_DNS_SD_H_
-#define GCP20_PROTOTYPE_DNS_SD_H_
+#ifndef CLOUD_PRINT_GCP20_PROTOTYPE_DNS_SD_SERVER_H_
+#define CLOUD_PRINT_GCP20_PROTOTYPE_DNS_SD_SERVER_H_
 
+#include <string>
 #include <vector>
 
-#include "net/dns/dns_protocol.h"
+#include "base/basictypes.h"
+#include "base/memory/scoped_ptr.h"
+#include "cloud_print/gcp20/prototype/service_parameters.h"
+#include "net/base/ip_endpoint.h"
 #include "net/udp/udp_socket.h"
 
+namespace net {
+  class IOBufferWithSize;
+}
+
+struct DnsQueryRecord;
+class DnsResponseBuilder;
+
 // Class for sending multicast announcements, receiving queries and answering on
-// them. Client should call |ProccessMessages| periodically to make server work.
+// them.
+// TODO(maksymb): Implement probing.
 class DnsSdServer {
  public:
-  // Constructs unstarted server.
+  // Constructor does not start server.
   DnsSdServer();
 
-  // Stops server.
+  // Stops the server and destroys the object.
   ~DnsSdServer();
 
   // Starts the server. Returns |true| if server works. Also sends
   // announcement.
-  bool Start();
+  bool Start(const ServiceParameters& serv_params, uint32 full_ttl,
+             const std::vector<std::string>& metadata) WARN_UNUSED_RESULT;
 
   // Sends announcement if server works.
   void Update();
@@ -30,24 +43,35 @@ class DnsSdServer {
   // Stops server with announcement.
   void Shutdown();
 
-  // Process pending queries for the server.
-  void ProcessMessages();
-
   // Returns |true| if server works.
-  bool is_online() { return is_online_; }
+  bool IsOnline() { return !!socket_; }
+
+  // Updates data for TXT respond.
+  void UpdateMetadata(const std::vector<std::string>& metadata);
 
  private:
   // Binds a socket to multicast address. Returns |true| on success.
   bool CreateSocket();
 
+  // Processes single query.
+  void ProccessQuery(uint32 current_ttl, const DnsQueryRecord& query,
+                     DnsResponseBuilder* builder) const;
+
+  // Processes DNS message.
+  void ProcessMessage(int len, net::IOBufferWithSize* buf);
+
+  // CompletionCallback for receiving data from DNS.
+  void DoLoop(int rv);
+
+  // Function to start listening to socket (delegate to DoLoop function).
+  void OnDatagramReceived();
+
   // Sends announcement.
   void SendAnnouncement(uint32 ttl);
 
-  // Returns |true| if server received some questions.
-  bool CheckPendingQueries();
-
-  // Stores |true| if server was started.
-  bool is_online_;
+  // Calculates and returns current TTL (with accordance to last send
+  // announcement time.
+  uint32 GetCurrentTLL() const;
 
   // Stores socket to multicast address.
   scoped_ptr<net::UDPSocket> socket_;
@@ -55,8 +79,26 @@ class DnsSdServer {
   // Stores multicast address end point.
   net::IPEndPoint multicast_address_;
 
+  // Stores time until last announcement is live.
+  base::Time time_until_live_;
+
+  // Stores service parameters (like service-name and service-type etc.)
+  ServiceParameters serv_params_;
+
+  // Stores the buffer for receiving messages.
+  scoped_refptr<net::IOBufferWithSize> recv_buf_;
+
+  // Stores address from where last message was sent.
+  net::IPEndPoint recv_address_;
+
+  // Stores information for TXT respond.
+  std::vector<std::string> metadata_;
+
+  // TTL for announcements
+  uint32 full_ttl_;
+
   DISALLOW_COPY_AND_ASSIGN(DnsSdServer);
 };
 
-#endif  // GCP20_PROTOTYPE_DNS_SD_H_
+#endif  // CLOUD_PRINT_GCP20_PROTOTYPE_DNS_SD_SERVER_H_