1 /*
2 * Copyright 2011 The WebRTC Project Authors. All rights reserved.
3 *
4 * Use of this source code is governed by a BSD-style license
5 * that can be found in the LICENSE file in the root of the source
6 * tree. An additional intellectual property rights grant can be found
7 * in the file PATENTS. All contributing project authors may
8 * be found in the AUTHORS file in the root of the source tree.
9 */
10
11 #include "webrtc/libjingle/xmpp/pingtask.h"
12
13 #include "webrtc/libjingle/xmpp/constants.h"
14 #include "webrtc/base/logging.h"
15 #include "webrtc/base/scoped_ptr.h"
16
17 namespace buzz {
18
PingTask(buzz::XmppTaskParentInterface * parent,rtc::MessageQueue * message_queue,uint32_t ping_period_millis,uint32_t ping_timeout_millis)19 PingTask::PingTask(buzz::XmppTaskParentInterface* parent,
20 rtc::MessageQueue* message_queue,
21 uint32_t ping_period_millis,
22 uint32_t ping_timeout_millis)
23 : buzz::XmppTask(parent, buzz::XmppEngine::HL_SINGLE),
24 message_queue_(message_queue),
25 ping_period_millis_(ping_period_millis),
26 ping_timeout_millis_(ping_timeout_millis),
27 next_ping_time_(0),
28 ping_response_deadline_(0) {
29 ASSERT(ping_period_millis >= ping_timeout_millis);
30 }
31
HandleStanza(const buzz::XmlElement * stanza)32 bool PingTask::HandleStanza(const buzz::XmlElement* stanza) {
33 if (!MatchResponseIq(stanza, Jid(STR_EMPTY), task_id())) {
34 return false;
35 }
36
37 if (stanza->Attr(buzz::QN_TYPE) != buzz::STR_RESULT &&
38 stanza->Attr(buzz::QN_TYPE) != buzz::STR_ERROR) {
39 return false;
40 }
41
42 QueueStanza(stanza);
43 return true;
44 }
45
46 // This task runs indefinitely and remains in either the start or blocked
47 // states.
ProcessStart()48 int PingTask::ProcessStart() {
49 if (ping_period_millis_ < ping_timeout_millis_) {
50 LOG(LS_ERROR) << "ping_period_millis should be >= ping_timeout_millis";
51 return STATE_ERROR;
52 }
53 const buzz::XmlElement* stanza = NextStanza();
54 if (stanza != NULL) {
55 // Received a ping response of some sort (don't care what it is).
56 ping_response_deadline_ = 0;
57 }
58
59 uint32_t now = rtc::Time();
60
61 // If the ping timed out, signal.
62 if (ping_response_deadline_ != 0 && now >= ping_response_deadline_) {
63 SignalTimeout();
64 return STATE_ERROR;
65 }
66
67 // Send a ping if it's time.
68 if (now >= next_ping_time_) {
69 rtc::scoped_ptr<buzz::XmlElement> stanza(
70 MakeIq(buzz::STR_GET, Jid(STR_EMPTY), task_id()));
71 stanza->AddElement(new buzz::XmlElement(QN_PING));
72 SendStanza(stanza.get());
73
74 ping_response_deadline_ = now + ping_timeout_millis_;
75 next_ping_time_ = now + ping_period_millis_;
76
77 // Wake ourselves up when it's time to send another ping or when the ping
78 // times out (so we can fire a signal).
79 message_queue_->PostDelayed(ping_timeout_millis_, this);
80 message_queue_->PostDelayed(ping_period_millis_, this);
81 }
82
83 return STATE_BLOCKED;
84 }
85
OnMessage(rtc::Message * msg)86 void PingTask::OnMessage(rtc::Message* msg) {
87 // Get the task manager to run this task so we can send a ping or signal or
88 // process a ping response.
89 Wake();
90 }
91
92 } // namespace buzz
93