-
Notifications
You must be signed in to change notification settings - Fork 68
/
ping.cpp
125 lines (104 loc) · 2.96 KB
/
ping.cpp
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
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
/**
* Copyright (c) 2014-2015, Arizona Board of Regents.
*
* This file is part of ndn-tools (Named Data Networking Essential Tools).
* See AUTHORS.md for complete list of ndn-tools authors and contributors.
*
* ndn-tools 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,
* either version 3 of the License, or (at your option) any later version.
*
* ndn-tools 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
* ndn-tools, e.g., in COPYING.md file. If not, see <http://www.gnu.org/licenses/>.
*
* @author: Jerald Paul Abraham <jeraldabraham@email.arizona.edu>
* @author: Eric Newberry <enewberry@email.arizona.edu>
*/
#include "ping.hpp"
namespace ndn {
namespace ping {
namespace client {
Ping::Ping(Face& face, const Options& options)
: m_options(options)
, m_nSent(0)
, m_nextSeq(options.startSeq)
, m_nOutstanding(0)
, m_face(face)
, m_scheduler(m_face.getIoService())
, m_nextPingEvent(m_scheduler)
{
if (m_options.shouldGenerateRandomSeq) {
m_nextSeq = random::generateWord64();
}
}
void
Ping::start()
{
performPing();
}
void
Ping::stop()
{
m_nextPingEvent.cancel();
}
void
Ping::performPing()
{
BOOST_ASSERT((m_options.nPings < 0) || (m_nSent < m_options.nPings));
Name pingPacketName = makePingName(m_nextSeq);
Interest interest(pingPacketName);
interest.setMustBeFresh(!m_options.shouldAllowStaleData);
interest.setInterestLifetime(m_options.timeout);
m_face.expressInterest(interest,
bind(&Ping::onData, this, _1, _2, m_nextSeq, time::steady_clock::now()),
bind(&Ping::onTimeout, this, _1, m_nextSeq));
++m_nSent;
++m_nextSeq;
++m_nOutstanding;
if ((m_options.nPings < 0) || (m_nSent < m_options.nPings)) {
m_nextPingEvent = m_scheduler.scheduleEvent(m_options.interval, bind(&Ping::performPing, this));
}
else {
finish();
}
}
void
Ping::onData(const Interest& interest, Data& data, uint64_t seq, const time::steady_clock::TimePoint& sendTime)
{
time::nanoseconds rtt = time::steady_clock::now() - sendTime;
afterResponse(seq, rtt);
finish();
}
void
Ping::onTimeout(const Interest& interest, uint64_t seq)
{
afterTimeout(seq);
finish();
}
void
Ping::finish()
{
if (--m_nOutstanding >= 0) {
return;
}
afterFinish();
}
Name
Ping::makePingName(uint64_t seq) const
{
Name name(m_options.prefix);
name.append("ping");
if (!m_options.clientIdentifier.empty()) {
name.append(m_options.clientIdentifier);
}
name.append(std::to_string(seq));
return name;
}
} // namespace client
} // namespace ping
} // namespace ndn