blob: f7bef11043b688260239e8303e7abf1a705941a5 [file] [log] [blame]
/*
* Copyright 2011 The WebRTC Project Authors. All rights reserved.
*
* Use of this source code is governed by a BSD-style license
* that can be found in the LICENSE file in the root of the source
* tree. An additional intellectual property rights grant can be found
* in the file PATENTS. All contributing project authors may
* be found in the AUTHORS file in the root of the source tree.
*/
#include "webrtc/libjingle/xmpp/pingtask.h"
#include <memory>
#include "webrtc/libjingle/xmpp/constants.h"
#include "webrtc/base/logging.h"
namespace buzz {
PingTask::PingTask(buzz::XmppTaskParentInterface* parent,
rtc::MessageQueue* message_queue,
uint32_t ping_period_millis,
uint32_t ping_timeout_millis)
: buzz::XmppTask(parent, buzz::XmppEngine::HL_SINGLE),
message_queue_(message_queue),
ping_period_millis_(ping_period_millis),
ping_timeout_millis_(ping_timeout_millis),
next_ping_time_(0),
ping_response_deadline_(0) {
ASSERT(ping_period_millis >= ping_timeout_millis);
}
bool PingTask::HandleStanza(const buzz::XmlElement* stanza) {
if (!MatchResponseIq(stanza, Jid(STR_EMPTY), task_id())) {
return false;
}
if (stanza->Attr(buzz::QN_TYPE) != buzz::STR_RESULT &&
stanza->Attr(buzz::QN_TYPE) != buzz::STR_ERROR) {
return false;
}
QueueStanza(stanza);
return true;
}
// This task runs indefinitely and remains in either the start or blocked
// states.
int PingTask::ProcessStart() {
if (ping_period_millis_ < ping_timeout_millis_) {
LOG(LS_ERROR) << "ping_period_millis should be >= ping_timeout_millis";
return STATE_ERROR;
}
const buzz::XmlElement* stanza = NextStanza();
if (stanza != NULL) {
// Received a ping response of some sort (don't care what it is).
ping_response_deadline_ = 0;
}
int64_t now = rtc::TimeMillis();
// If the ping timed out, signal.
if (ping_response_deadline_ != 0 && now >= ping_response_deadline_) {
SignalTimeout();
return STATE_ERROR;
}
// Send a ping if it's time.
if (now >= next_ping_time_) {
std::unique_ptr<buzz::XmlElement> stanza(
MakeIq(buzz::STR_GET, Jid(STR_EMPTY), task_id()));
stanza->AddElement(new buzz::XmlElement(QN_PING));
SendStanza(stanza.get());
ping_response_deadline_ = now + ping_timeout_millis_;
next_ping_time_ = now + ping_period_millis_;
// Wake ourselves up when it's time to send another ping or when the ping
// times out (so we can fire a signal).
message_queue_->PostDelayed(RTC_FROM_HERE, ping_timeout_millis_, this);
message_queue_->PostDelayed(RTC_FROM_HERE, ping_period_millis_, this);
}
return STATE_BLOCKED;
}
void PingTask::OnMessage(rtc::Message* msg) {
// Get the task manager to run this task so we can send a ping or signal or
// process a ping response.
Wake();
}
} // namespace buzz