1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
|
/*
* InspIRCd -- Internet Relay Chat Daemon
*
* Copyright (C) 2015 Attila Molnar <attilamolnar@hush.com>
*
* This file is part of InspIRCd. InspIRCd is free software: you can
* redistribute it and/or modify it under the terms of the GNU General Public
* License as published by the Free Software Foundation, version 2.
*
* This program is distributed in the hope that it will be useful, but WITHOUT
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
* FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
* details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#include "inspircd.h"
#include "pingtimer.h"
#include "treeserver.h"
#include "commandbuilder.h"
PingTimer::PingTimer(TreeServer* ts)
: Timer(Utils->PingFreq)
, server(ts)
, state(PS_SENDPING)
{
}
PingTimer::State PingTimer::TickInternal()
{
// Timer expired, take next action based on what happened last time
if (state == PS_SENDPING)
{
// Last ping was answered, send next ping
server->GetSocket()->WriteLine(CmdBuilder("PING").push(server->GetId()));
LastPingMsec = ServerInstance->Time() * 1000 + (ServerInstance->Time_ns() / 1000000);
// Warn next unless warnings are disabled. If they are, jump straight to timeout.
if (Utils->PingWarnTime)
return PS_WARN;
else
return PS_TIMEOUT;
}
else if (state == PS_WARN)
{
// No pong arrived in PingWarnTime seconds, send a warning to opers
ServerInstance->SNO->WriteToSnoMask('l', "Server \002%s\002 has not responded to PING for %d seconds, high latency.", server->GetName().c_str(), GetInterval());
return PS_TIMEOUT;
}
else // PS_TIMEOUT
{
// They didn't answer the last ping, if they are locally connected, get rid of them
if (server->IsLocal())
{
TreeSocket* sock = server->GetSocket();
sock->SendError("Ping timeout");
sock->Close();
}
// If the server is non-locally connected, don't do anything until we get a PONG.
// This is to avoid pinging the server and warning opers more than once.
// If they do answer eventually, we will move to the PS_SENDPING state and ping them again.
return PS_IDLE;
}
}
void PingTimer::SetState(State newstate)
{
state = newstate;
// Set when should the next Tick() happen based on the state
if (state == PS_SENDPING)
SetInterval(Utils->PingFreq);
else if (state == PS_WARN)
SetInterval(Utils->PingWarnTime);
else if (state == PS_TIMEOUT)
SetInterval(Utils->PingFreq - Utils->PingWarnTime);
// If state == PS_IDLE, do not set the timer, see above why
}
bool PingTimer::Tick(time_t currtime)
{
if (server->IsDead())
return false;
SetState(TickInternal());
return false;
}
void PingTimer::OnPong()
{
// Calculate RTT
long ts = ServerInstance->Time() * 1000 + (ServerInstance->Time_ns() / 1000000);
server->rtt = ts - LastPingMsec;
// Change state to send ping next, also reschedules the timer appropriately
SetState(PS_SENDPING);
}
|
