Home | History | Annotate | Download | only in xmpp 
      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 
     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 
     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.
     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 
     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