Add heartbeat timeout.

remoting/host/heartbeat_sender.h

 // Copyright (c) 2012 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 REMOTING_HOST_HEARTBEAT_SENDER_H_
 #define REMOTING_HOST_HEARTBEAT_SENDER_H_
 
 #include <memory>
 #include <string>
 
@@ skipping 113 matching lines @@
   // SignalStrategy::Listener interface.
   void OnSignalStrategyStateChange(SignalStrategy::State state) override;
   bool OnSignalStrategyIncomingStanza(const buzz::XmlElement* stanza) override;
 
  private:
   FRIEND_TEST_ALL_PREFIXES(HeartbeatSenderTest,
                            DoSendStanzaWithExpectedSequenceId);
   FRIEND_TEST_ALL_PREFIXES(HeartbeatSenderTest, ProcessResponseSetInterval);
   FRIEND_TEST_ALL_PREFIXES(HeartbeatSenderTest,
                            ProcessResponseExpectedSequenceId);
+  FRIEND_TEST_ALL_PREFIXES(HeartbeatSenderTest, ResponseTimeout);
   friend class HeartbeatSenderTest;
 
   void SendStanza();
   void ResendStanza();
   void DoSendStanza();
   void ProcessResponse(bool is_offline_heartbeat_response,
                        IqRequest* request,
                        const buzz::XmlElement* response);
-  void SetInterval(int interval);
+  void SetInterval(base::TimeDelta interval);
   void SetSequenceId(int sequence_id);
 
   // Handlers for host-offline-reason completion and timeout.
   void OnHostOfflineReasonTimeout();
   void OnHostOfflineReasonAck();
 
   // Helper methods used by DoSendStanza() to generate heartbeat stanzas.
   std::unique_ptr<buzz::XmlElement> CreateHeartbeatMessage();
   std::unique_ptr<buzz::XmlElement> CreateSignature();
 
   base::Closure on_heartbeat_successful_callback_;
   base::Closure on_unknown_host_id_error_;
   std::string host_id_;
   SignalStrategy* signal_strategy_;
   scoped_refptr<const RsaKeyPair> host_key_pair_;
   std::string directory_bot_jid_;
   std::unique_ptr<IqSender> iq_sender_;
   std::unique_ptr<IqRequest> request_;
-  int interval_ms_;
+  base::TimeDelta interval_;
   base::RepeatingTimer timer_;
   base::OneShotTimer timer_resend_;
-  int sequence_id_;
-  bool sequence_id_was_set_;
-  int sequence_id_recent_set_num_;
-  bool heartbeat_succeeded_;
-  int failed_startup_heartbeat_count_;
+  int sequence_id_ = 0;
+  bool sequence_id_was_set_ = false;
+  int sequence_id_recent_set_num_ = 0;
+  bool heartbeat_succeeded_ = false;
+  int failed_startup_heartbeat_count_ = 0;
+  int timed_out_heartbeats_count_ = 0;
 
   // Fields to send and indicate completion of sending host-offline-reason.
   std::string host_offline_reason_;
   base::Callback<void(bool success)> host_offline_reason_ack_callback_;
   base::OneShotTimer host_offline_reason_timeout_timer_;
 
   base::ThreadChecker thread_checker_;
 
   DISALLOW_COPY_AND_ASSIGN(HeartbeatSender);
 };
 
 }  // namespace remoting
 
 #endif  // REMOTING_HOST_HEARTBEAT_SENDER_H_