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 #include "socket.h"
  22 #include "xline.h"
  23 #include "socketengine.h"
  24
  25 #include "main.h"
  26 #include "utils.h"
  27 #include "treeserver.h"
  28 #include "treesocket.h"
  29
  30 /* $ModDep: m_spanningtree/main.h m_spanningtree/utils.h m_spanningtree/treeserver.h m_spanningtree/treesocket.h */
  31
  32 bool TreeSocket::LocalPong(const std::string &prefix, parameterlist &params)
  33 {
  34         if (params.size() < 1)
  35                 return true;
  36
  37         if (params.size() == 1)
  38         {
  39                 TreeServer* ServerSource = Utils->FindServer(prefix);
  40                 if (ServerSource)
  41                 {
  42                         ServerSource->SetPingFlag();
  43                         long ts = ServerInstance->Time() * 1000 + (ServerInstance->Time_ns() / 1000000);
  44                         ServerSource->rtt = ts - ServerSource->LastPingMsec;
  45                 }
  46         }
  47         else
  48         {
  49                 std::string forwardto = params[1];
  50                 if (forwardto == ServerInstance->Config->GetSID() || forwardto == ServerInstance->Config->ServerName)
  51                 {
  52                         /*
  53                          * this is a PONG for us
  54                          * if the prefix is a user, check theyre local, and if they are,
  55                          * dump the PONG reply back to their fd. If its a server, do nowt.
  56                          * Services might want to send these s->s, but we dont need to yet.
  57                          */
  58                         User* u = ServerInstance->FindNick(prefix);
  59                         if (u)
  60                         {
  61                                 u->WriteServ("PONG %s %s",params[0].c_str(),params[1].c_str());
  62                         }
  63
  64                         TreeServer *ServerSource = Utils->FindServer(params[0]);
  65
  66                         if (ServerSource)
  67                         {
  68                                 long ts = ServerInstance->Time() * 1000 + (ServerInstance->Time_ns() / 1000000);
  69                                 ServerSource->rtt = ts - ServerSource->LastPingMsec;
  70                                 ServerSource->SetPingFlag();
  71                         }
  72                 }
  73                 else
  74                 {
  75                         // not for us, pass it on :)
  76                         Utils->DoOneToOne(prefix,"PONG",params,forwardto);
  77                 }
  78         }
  79
  80         return true;
  81 }
  82