heartbeat_sender.cc revision 5821806d5e7f356e8fa4b058a389a808ea183019
15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "remoting/host/heartbeat_sender.h" 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <math.h> 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/bind.h" 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/logging.h" 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/message_loop_proxy.h" 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/rand_util.h" 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/string_number_conversions.h" 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/time.h" 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "remoting/base/constants.h" 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "remoting/host/server_log_entry.h" 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "remoting/jingle_glue/iq_sender.h" 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "remoting/jingle_glue/signal_strategy.h" 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "third_party/libjingle/source/talk/xmllite/xmlelement.h" 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "third_party/libjingle/source/talk/xmpp/constants.h" 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using buzz::QName; 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using buzz::XmlElement; 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace remoting { 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace { 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const char kHeartbeatQueryTag[] = "heartbeat"; 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const char kHostIdAttr[] = "hostid"; 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const char kHeartbeatSignatureTag[] = "signature"; 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const char kSequenceIdAttr[] = "sequence-id"; 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const char kErrorTag[] = "error"; 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const char kNotFoundTag[] = "item-not-found"; 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const char kHeartbeatResultTag[] = "heartbeat-result"; 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const char kSetIntervalTag[] = "set-interval"; 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const char kExpectedSequenceIdTag[] = "expected-sequence-id"; 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const int64 kDefaultHeartbeatIntervalMs = 5 * 60 * 1000; // 5 minutes. 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const int64 kResendDelayMs = 10 * 1000; // 10 seconds. 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const int64 kResendDelayOnHostNotFoundMs = 10 * 1000; // 10 seconds. 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const int kMaxResendOnHostNotFoundCount = 12; // 2 minutes (12 x 10 seconds). 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)HeartbeatSender::HeartbeatSender( 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Listener* listener, 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::string& host_id, 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SignalStrategy* signal_strategy, 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) HostKeyPair* key_pair) 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) : listener_(listener), 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) host_id_(host_id), 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) signal_strategy_(signal_strategy), 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) key_pair_(key_pair), 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) interval_ms_(kDefaultHeartbeatIntervalMs), 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) sequence_id_(0), 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) sequence_id_was_set_(false), 605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) sequence_id_recent_set_num_(0), 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) heartbeat_succeeded_(false), 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) failed_startup_heartbeat_count_(0) { 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(signal_strategy_); 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(key_pair_); 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) signal_strategy_->AddListener(this); 675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Start heartbeats if the |signal_strategy_| is already connected. 695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) OnSignalStrategyStateChange(signal_strategy_->GetState()); 705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)HeartbeatSender::~HeartbeatSender() { 735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) signal_strategy_->RemoveListener(this); 745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void HeartbeatSender::OnSignalStrategyStateChange(SignalStrategy::State state) { 775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (state == SignalStrategy::CONNECTED) { 785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) iq_sender_.reset(new IqSender(signal_strategy_)); 795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SendStanza(); 805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) timer_.Start(FROM_HERE, base::TimeDelta::FromMilliseconds(interval_ms_), 815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) this, &HeartbeatSender::SendStanza); 825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } else if (state == SignalStrategy::DISCONNECTED) { 835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) request_.reset(); 845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) iq_sender_.reset(); 855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) timer_.Stop(); 865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) timer_resend_.Stop(); 875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool HeartbeatSender::OnSignalStrategyIncomingStanza( 915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const buzz::XmlElement* stanza) { 925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void HeartbeatSender::SendStanza() { 965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DoSendStanza(); 975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Make sure we don't send another heartbeat before the heartbeat interval 985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // has expired. 995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) timer_resend_.Stop(); 1005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void HeartbeatSender::ResendStanza() { 1035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DoSendStanza(); 1045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Make sure we don't send another heartbeat before the heartbeat interval 1055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // has expired. 1065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) timer_.Reset(); 1075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void HeartbeatSender::DoSendStanza() { 1105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) VLOG(1) << "Sending heartbeat stanza to " << kChromotingBotJid; 1115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) request_ = iq_sender_->SendIq( 1125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) buzz::STR_SET, kChromotingBotJid, CreateHeartbeatMessage(), 1135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Bind(&HeartbeatSender::ProcessResponse, 1145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Unretained(this))); 1155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ++sequence_id_; 1165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void HeartbeatSender::ProcessResponse(IqRequest* request, 1195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const XmlElement* response) { 1205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::string type = response->Attr(buzz::QN_TYPE); 1215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (type == buzz::STR_ERROR) { 1225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const XmlElement* error_element = 1235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) response->FirstNamed(QName(buzz::NS_CLIENT, kErrorTag)); 1245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (error_element) { 1255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (error_element->FirstNamed(QName(buzz::NS_STANZA, kNotFoundTag))) { 1265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) LOG(ERROR) << "Received error: Host ID not found"; 1275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // If the host was registered immediately before it sends a heartbeat, 1285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // then server-side latency may prevent the server recognizing the 1295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // host ID in the heartbeat. So even if all of the first few heartbeats 1305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // get a "host ID not found" error, that's not a good enough reason to 1315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // exit. 1325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) failed_startup_heartbeat_count_++; 1335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!heartbeat_succeeded_ && (failed_startup_heartbeat_count_ <= 1345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) kMaxResendOnHostNotFoundCount)) { 1355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) timer_resend_.Start(FROM_HERE, 1365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::TimeDelta::FromMilliseconds( 1375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) kResendDelayOnHostNotFoundMs), 1385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) this, 1395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) &HeartbeatSender::ResendStanza); 1405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 1415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) listener_->OnUnknownHostIdError(); 1435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 1445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) LOG(ERROR) << "Received error in response to heartbeat: " 1485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) << response->Str(); 1495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 1505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) heartbeat_succeeded_ = true; 1535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // This method must only be called for error or result stanzas. 1555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK_EQ(std::string(buzz::STR_RESULT), type); 1565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const XmlElement* result_element = 1585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) response->FirstNamed(QName(kChromotingXmlNamespace, kHeartbeatResultTag)); 1595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (result_element) { 1605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const XmlElement* set_interval_element = 1615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) result_element->FirstNamed(QName(kChromotingXmlNamespace, 1625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) kSetIntervalTag)); 1635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (set_interval_element) { 1645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::string& interval_str = set_interval_element->BodyText(); 1655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int interval; 1665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!base::StringToInt(interval_str, &interval) || interval <= 0) { 1675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) LOG(ERROR) << "Received invalid set-interval: " 1685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) << set_interval_element->Str(); 1695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } else { 1705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SetInterval(interval * base::Time::kMillisecondsPerSecond); 1715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool did_set_sequence_id = false; 1755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const XmlElement* expected_sequence_id_element = 1765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) result_element->FirstNamed(QName(kChromotingXmlNamespace, 1775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) kExpectedSequenceIdTag)); 1785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (expected_sequence_id_element) { 1795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // The sequence ID sent in the previous heartbeat was not what the server 1805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // expected, so send another heartbeat with the expected sequence ID. 1815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::string& expected_sequence_id_str = 1825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) expected_sequence_id_element->BodyText(); 1835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int expected_sequence_id; 1845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!base::StringToInt(expected_sequence_id_str, &expected_sequence_id)) { 1855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) LOG(ERROR) << "Received invalid " << kExpectedSequenceIdTag << ": " << 1865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) expected_sequence_id_element->Str(); 1875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } else { 1885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SetSequenceId(expected_sequence_id); 1895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) sequence_id_recent_set_num_++; 1905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) did_set_sequence_id = true; 1915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!did_set_sequence_id) { 1945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) sequence_id_recent_set_num_ = 0; 1955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void HeartbeatSender::SetInterval(int interval) { 2005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (interval != interval_ms_) { 2015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) interval_ms_ = interval; 2025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Restart the timer with the new interval. 2045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (timer_.IsRunning()) { 2055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) timer_.Stop(); 2065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) timer_.Start(FROM_HERE, base::TimeDelta::FromMilliseconds(interval_ms_), 2075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) this, &HeartbeatSender::SendStanza); 2085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void HeartbeatSender::SetSequenceId(int sequence_id) { 2135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) sequence_id_ = sequence_id; 2145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Setting the sequence ID may be a symptom of a temporary server-side 2155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // problem, which would affect many hosts, so don't send a new heartbeat 2165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // immediately, as many hosts doing so may overload the server. 2175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // But the server will usually set the sequence ID when it receives the first 2185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // heartbeat from a host. In that case, we can send a new heartbeat 2195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // immediately, as that only happens once per host instance. 2205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!sequence_id_was_set_) { 2215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ResendStanza(); 2225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } else { 2235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) LOG(INFO) << "The heartbeat sequence ID has been set more than once: " 2245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) << "the new value is " << sequence_id; 2255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) double delay = pow(2.0, sequence_id_recent_set_num_) * 2265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) (1 + base::RandDouble()) * kResendDelayMs; 2275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (delay <= interval_ms_) { 2285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) timer_resend_.Start(FROM_HERE, base::TimeDelta::FromMilliseconds(delay), 2295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) this, &HeartbeatSender::ResendStanza); 2305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) sequence_id_was_set_ = true; 2335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)scoped_ptr<XmlElement> HeartbeatSender::CreateHeartbeatMessage() { 2365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Create heartbeat stanza. 2375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<XmlElement> query(new XmlElement( 2385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) QName(kChromotingXmlNamespace, kHeartbeatQueryTag))); 2395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) query->AddAttr(QName(kChromotingXmlNamespace, kHostIdAttr), host_id_); 2405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) query->AddAttr(QName(kChromotingXmlNamespace, kSequenceIdAttr), 2415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::IntToString(sequence_id_)); 2425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) query->AddElement(CreateSignature().release()); 2435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Append log message (which isn't signed). 2445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<XmlElement> log(ServerLogEntry::MakeStanza()); 2455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<ServerLogEntry> log_entry(ServerLogEntry::MakeForHeartbeat()); 2465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) log_entry->AddHostFields(); 2475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) log->AddElement(log_entry->ToStanza().release()); 2485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) query->AddElement(log.release()); 2495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return query.Pass(); 2505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)scoped_ptr<XmlElement> HeartbeatSender::CreateSignature() { 2535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<XmlElement> signature_tag(new XmlElement( 2545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) QName(kChromotingXmlNamespace, kHeartbeatSignatureTag))); 2555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::string message = signal_strategy_->GetLocalJid() + ' ' + 2575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::IntToString(sequence_id_); 2585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::string signature(key_pair_->GetSignature(message)); 2595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) signature_tag->AddText(signature); 2605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return signature_tag.Pass(); 2625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace remoting 265