src/modules/m_spanningtree/pong.cpp

   1 /*       +------------------------------------+
   2  *       | Inspire Internet Relay Chat Daemon |
   3  *       +------------------------------------+
   4  *
   5  *  InspIRCd: (C) 2002-2010 InspIRCd Development Team
   6  * See: http://wiki.inspircd.org/Credits
   7  *
   8  * This program is free but copyrighted software; see
   9  *            the file COPYING for details.
  10  *
  11  * ---------------------------------------------------
  12  */
  13
  14 #include "inspircd.h"
  15 #include "socket.h"
  16 #include "xline.h"
  17 #include "socketengine.h"
  18
  19 #include "main.h"
  20 #include "utils.h"
  21 #include "treeserver.h"
  22 #include "treesocket.h"
  23
  24 /* $ModDep: m_spanningtree/main.h m_spanningtree/utils.h m_spanningtree/treeserver.h m_spanningtree/treesocket.h */
  25
  26 bool TreeSocket::LocalPong(const std::string &prefix, parameterlist &params)
  27 {
  28         if (params.size() < 1)
  29                 return true;
  30
  31         if (params.size() == 1)
  32         {
  33                 TreeServer* ServerSource = Utils->FindServer(prefix);
  34                 if (ServerSource)
  35                 {
  36                         ServerSource->SetPingFlag();
  37                         long ts = ServerInstance->Time() * 1000 + (ServerInstance->Time_ns() / 1000000);
  38                         ServerSource->rtt = ts - ServerSource->LastPingMsec;
  39                 }
  40         }
  41         else
  42         {
  43                 std::string forwardto = params[1];
  44                 if (forwardto == ServerInstance->Config->GetSID() || forwardto == ServerInstance->Config->ServerName)
  45                 {
  46                         /*
  47                          * this is a PONG for us
  48                          * if the prefix is a user, check theyre local, and if they are,
  49                          * dump the PONG reply back to their fd. If its a server, do nowt.
  50                          * Services might want to send these s->s, but we dont need to yet.
  51                          */
  52                         User* u = ServerInstance->FindNick(prefix);
  53                         if (u)
  54                         {
  55                                 u->WriteServ("PONG %s %s",params[0].c_str(),params[1].c_str());
  56                         }
  57
  58                         TreeServer *ServerSource = Utils->FindServer(params[0]);
  59
  60                         if (ServerSource)
  61                         {
  62                                 long ts = ServerInstance->Time() * 1000 + (ServerInstance->Time_ns() / 1000000);
  63                                 ServerSource->rtt = ts - ServerSource->LastPingMsec;
  64                                 ServerSource->SetPingFlag();
  65                         }
  66                 }
  67                 else
  68                 {
  69                         // not for us, pass it on :)
  70                         Utils->DoOneToOne(prefix,"PONG",params,forwardto);
  71                 }
  72         }
  73
  74         return true;
  75 }
  76