| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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 #include "remoting/host/heartbeat_sender.h" | 5 #include "remoting/host/heartbeat_sender.h" |
| 6 | 6 |
| 7 #include <math.h> | 7 #include <math.h> |
| 8 | 8 |
| 9 #include "base/bind.h" | 9 #include "base/bind.h" |
| 10 #include "base/logging.h" | 10 #include "base/logging.h" |
| 11 #include "base/message_loop_proxy.h" | 11 #include "base/message_loop_proxy.h" |
| 12 #include "base/rand_util.h" | 12 #include "base/rand_util.h" |
| 13 #include "base/string_number_conversions.h" | 13 #include "base/string_number_conversions.h" |
| 14 #include "base/strings/stringize_macros.h" |
| 14 #include "base/time.h" | 15 #include "base/time.h" |
| 15 #include "remoting/base/constants.h" | 16 #include "remoting/base/constants.h" |
| 16 #include "remoting/host/server_log_entry.h" | 17 #include "remoting/host/server_log_entry.h" |
| 17 #include "remoting/jingle_glue/iq_sender.h" | 18 #include "remoting/jingle_glue/iq_sender.h" |
| 18 #include "remoting/jingle_glue/signal_strategy.h" | 19 #include "remoting/jingle_glue/signal_strategy.h" |
| 19 #include "third_party/libjingle/source/talk/xmllite/xmlelement.h" | 20 #include "third_party/libjingle/source/talk/xmllite/xmlelement.h" |
| 20 #include "third_party/libjingle/source/talk/xmpp/constants.h" | 21 #include "third_party/libjingle/source/talk/xmpp/constants.h" |
| 21 | 22 |
| 22 using buzz::QName; | 23 using buzz::QName; |
| 23 using buzz::XmlElement; | 24 using buzz::XmlElement; |
| 24 | 25 |
| 25 namespace remoting { | 26 namespace remoting { |
| 26 | 27 |
| 27 namespace { | 28 namespace { |
| 28 | 29 |
| 29 const char kHeartbeatQueryTag[] = "heartbeat"; | 30 const char kHeartbeatQueryTag[] = "heartbeat"; |
| 30 const char kHostIdAttr[] = "hostid"; | 31 const char kHostIdAttr[] = "hostid"; |
| 32 const char kHostVersionTag[] = "host-version"; |
| 31 const char kHeartbeatSignatureTag[] = "signature"; | 33 const char kHeartbeatSignatureTag[] = "signature"; |
| 32 const char kSequenceIdAttr[] = "sequence-id"; | 34 const char kSequenceIdAttr[] = "sequence-id"; |
| 33 | 35 |
| 34 const char kErrorTag[] = "error"; | 36 const char kErrorTag[] = "error"; |
| 35 const char kNotFoundTag[] = "item-not-found"; | 37 const char kNotFoundTag[] = "item-not-found"; |
| 36 | 38 |
| 37 const char kHeartbeatResultTag[] = "heartbeat-result"; | 39 const char kHeartbeatResultTag[] = "heartbeat-result"; |
| 38 const char kSetIntervalTag[] = "set-interval"; | 40 const char kSetIntervalTag[] = "set-interval"; |
| 39 const char kExpectedSequenceIdTag[] = "expected-sequence-id"; | 41 const char kExpectedSequenceIdTag[] = "expected-sequence-id"; |
| 40 | 42 |
| (...skipping 188 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 229 if (delay <= interval_ms_) { | 231 if (delay <= interval_ms_) { |
| 230 timer_resend_.Start(FROM_HERE, base::TimeDelta::FromMilliseconds(delay), | 232 timer_resend_.Start(FROM_HERE, base::TimeDelta::FromMilliseconds(delay), |
| 231 this, &HeartbeatSender::ResendStanza); | 233 this, &HeartbeatSender::ResendStanza); |
| 232 } | 234 } |
| 233 } | 235 } |
| 234 sequence_id_was_set_ = true; | 236 sequence_id_was_set_ = true; |
| 235 } | 237 } |
| 236 | 238 |
| 237 scoped_ptr<XmlElement> HeartbeatSender::CreateHeartbeatMessage() { | 239 scoped_ptr<XmlElement> HeartbeatSender::CreateHeartbeatMessage() { |
| 238 // Create heartbeat stanza. | 240 // Create heartbeat stanza. |
| 239 scoped_ptr<XmlElement> query(new XmlElement( | 241 scoped_ptr<XmlElement> heartbeat(new XmlElement( |
| 240 QName(kChromotingXmlNamespace, kHeartbeatQueryTag))); | 242 QName(kChromotingXmlNamespace, kHeartbeatQueryTag))); |
| 241 query->AddAttr(QName(kChromotingXmlNamespace, kHostIdAttr), host_id_); | 243 heartbeat->AddAttr(QName(kChromotingXmlNamespace, kHostIdAttr), host_id_); |
| 242 query->AddAttr(QName(kChromotingXmlNamespace, kSequenceIdAttr), | 244 heartbeat->AddAttr(QName(kChromotingXmlNamespace, kSequenceIdAttr), |
| 243 base::IntToString(sequence_id_)); | 245 base::IntToString(sequence_id_)); |
| 244 query->AddElement(CreateSignature().release()); | 246 heartbeat->AddElement(CreateSignature().release()); |
| 247 // Append host version. |
| 248 scoped_ptr<XmlElement> version_tag(new XmlElement( |
| 249 QName(kChromotingXmlNamespace, kHostVersionTag))); |
| 250 version_tag->AddText(STRINGIZE(VERSION)); |
| 251 heartbeat->AddElement(version_tag.release()); |
| 245 // Append log message (which isn't signed). | 252 // Append log message (which isn't signed). |
| 246 scoped_ptr<XmlElement> log(ServerLogEntry::MakeStanza()); | 253 scoped_ptr<XmlElement> log(ServerLogEntry::MakeStanza()); |
| 247 scoped_ptr<ServerLogEntry> log_entry(ServerLogEntry::MakeForHeartbeat()); | 254 scoped_ptr<ServerLogEntry> log_entry(ServerLogEntry::MakeForHeartbeat()); |
| 248 log_entry->AddHostFields(); | 255 log_entry->AddHostFields(); |
| 249 log->AddElement(log_entry->ToStanza().release()); | 256 log->AddElement(log_entry->ToStanza().release()); |
| 250 query->AddElement(log.release()); | 257 heartbeat->AddElement(log.release()); |
| 251 return query.Pass(); | 258 return heartbeat.Pass(); |
| 252 } | 259 } |
| 253 | 260 |
| 254 scoped_ptr<XmlElement> HeartbeatSender::CreateSignature() { | 261 scoped_ptr<XmlElement> HeartbeatSender::CreateSignature() { |
| 255 scoped_ptr<XmlElement> signature_tag(new XmlElement( | 262 scoped_ptr<XmlElement> signature_tag(new XmlElement( |
| 256 QName(kChromotingXmlNamespace, kHeartbeatSignatureTag))); | 263 QName(kChromotingXmlNamespace, kHeartbeatSignatureTag))); |
| 257 | 264 |
| 258 std::string message = signal_strategy_->GetLocalJid() + ' ' + | 265 std::string message = signal_strategy_->GetLocalJid() + ' ' + |
| 259 base::IntToString(sequence_id_); | 266 base::IntToString(sequence_id_); |
| 260 std::string signature(key_pair_->GetSignature(message)); | 267 std::string signature(key_pair_->GetSignature(message)); |
| 261 signature_tag->AddText(signature); | 268 signature_tag->AddText(signature); |
| 262 | 269 |
| 263 return signature_tag.Pass(); | 270 return signature_tag.Pass(); |
| 264 } | 271 } |
| 265 | 272 |
| 266 } // namespace remoting | 273 } // namespace remoting |
| OLD | NEW |