src/modules/m_spanningtree/pong.cpp

   1 /*
   2  * InspIRCd -- Internet Relay Chat Daemon
   3  *
   4  *   Copyright (C) 2008 Robin Burchell <robin+git@viroteck.net>
   5  *
   6  * This file is part of InspIRCd.  InspIRCd is free software: you can
   7  * redistribute it and/or modify it under the terms of the GNU General Public
   8  * License as published by the Free Software Foundation, version 2.
   9  *
  10  * This program is distributed in the hope that it will be useful, but WITHOUT
  11  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
  12  * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
  13  * details.
  14  *
  15  * You should have received a copy of the GNU General Public License
  16  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  17  */
  18
  19
  20 #include "inspircd.h"
  21
  22 #include "utils.h"
  23 #include "treeserver.h"
  24 #include "treesocket.h"
  25
  26 /* $ModDep: m_spanningtree/main.h m_spanningtree/utils.h m_spanningtree/treeserver.h m_spanningtree/treesocket.h */
  27
  28 bool TreeSocket::LocalPong(const std::string &prefix, parameterlist &params)
  29 {
  30         if (params.size() < 1)
  31                 return true;
  32
  33         if (params.size() == 1)
  34         {
  35                 TreeServer* ServerSource = Utils->FindServer(prefix);
  36                 if (ServerSource)
  37                 {
  38                         ServerSource->SetPingFlag();
  39                         long ts = ServerInstance->Time() * 1000 + (ServerInstance->Time_ns() / 1000000);
  40                         ServerSource->rtt = ts - ServerSource->LastPingMsec;
  41                 }
  42         }
  43         else
  44         {
  45                 const std::string& forwardto = params[1];
  46                 if (forwardto == ServerInstance->Config->GetSID() || forwardto == ServerInstance->Config->ServerName)
  47                 {
  48                         /*
  49                          * this is a PONG for us
  50                          * if the prefix is a user, check theyre local, and if they are,
  51                          * dump the PONG reply back to their fd. If its a server, do nowt.
  52                          * Services might want to send these s->s, but we dont need to yet.
  53                          */
  54                         User* u = ServerInstance->FindNick(prefix);
  55                         if (u)
  56                         {
  57                                 u->WriteServ("PONG %s %s",params[0].c_str(),params[1].c_str());
  58                         }
  59
  60                         TreeServer *ServerSource = Utils->FindServer(params[0]);
  61
  62                         if (ServerSource)
  63                         {
  64                                 long ts = ServerInstance->Time() * 1000 + (ServerInstance->Time_ns() / 1000000);
  65                                 ServerSource->rtt = ts - ServerSource->LastPingMsec;
  66                                 ServerSource->SetPingFlag();
  67                         }
  68                 }
  69                 else
  70                 {
  71                         // not for us, pass it on :)
  72                         Utils->DoOneToOne(prefix,"PONG",params,forwardto);
  73                 }
  74         }
  75
  76         return true;
  77 }
  78