mirror of
https://github.com/pocoproject/poco.git
synced 2024-12-13 18:45:10 +01:00
c4e676d36d
* add PMTU discovery #2329 * add socket gather/scatter capabilities #2330 (win, udp) * enable WSAPoll * add FastMemoryPool * add receiveFrom() with native args * allow copying of StringTokenizer * add AtomicFlag and SpinlockMutex * update .gitignore * UDPServer and client #2343 (windows) * fix warnings * fix warnings * regenerate Net VS solutions * regenerate CppUnit projects/solutions * clang fixes * gcc fixes * try to fix travis * more travis fixes * more travis fixes * handle UDPClient exception * fix makefiles and init order warnings * add UNIX gather/scatter sendto/recvfrom implementations and tests * run travis tests as sudo * try to run tests as sudo, 2nd attempt * fix warning * use mutex in reactor * lock-order-inversion in SocketReactor #2346 * add PMTU discovery #2329 (linux) * ICMPSocket does not check reply address #1921 * remove some ignored tests * add PMTU discovery #2329 (reconcile logic with #1921) * fix native receiveFrome() * reinstate ignoring of proxy errors * add testMTU to ignore list * add include atomic * NTPClient not checking reply address #2348 * some ICMP/MTU fixes * UDPSocketReader cleanup * resolve some socket inheritance warnings * add NTP time sync to ignored tests * SocketNotifier not thread-safe #2345 * prevent x64 samples build attempt for win32 * build TestApp and Library * fix ICMP tests * regen VS projects * regen VS projects and add missing 2012 files * remove debug prints
152 lines
3.1 KiB
C++
152 lines
3.1 KiB
C++
//
|
|
// ICMPClient.cpp
|
|
//
|
|
// Library: Net
|
|
// Package: ICMP
|
|
// Module: ICMPClient
|
|
//
|
|
// Copyright (c) 2006, Applied Informatics Software Engineering GmbH.
|
|
// and Contributors.
|
|
//
|
|
// SPDX-License-Identifier: BSL-1.0
|
|
//
|
|
|
|
|
|
#include "Poco/Net/SocketAddress.h"
|
|
#include "Poco/Net/ICMPClient.h"
|
|
#include "Poco/Net/NetException.h"
|
|
#include "Poco/Channel.h"
|
|
#include "Poco/Message.h"
|
|
#include "Poco/Format.h"
|
|
#include <sstream>
|
|
|
|
|
|
using Poco::Channel;
|
|
using Poco::Message;
|
|
using Poco::InvalidArgumentException;
|
|
using Poco::NotImplementedException;
|
|
using Poco::TimeoutException;
|
|
using Poco::Exception;
|
|
|
|
|
|
namespace Poco {
|
|
namespace Net {
|
|
|
|
|
|
ICMPClient::ICMPClient(SocketAddress::Family family, int dataSize, int ttl, int timeout):
|
|
_family(family),
|
|
_dataSize(dataSize),
|
|
_ttl(ttl),
|
|
_timeout(timeout)
|
|
{
|
|
}
|
|
|
|
|
|
ICMPClient::~ICMPClient()
|
|
{
|
|
}
|
|
|
|
|
|
int ICMPClient::ping(const std::string& address, int repeat) const
|
|
{
|
|
if (repeat <= 0) return 0;
|
|
|
|
SocketAddress addr(address, 0);
|
|
return ping(addr, repeat);
|
|
}
|
|
|
|
|
|
int ICMPClient::ping(SocketAddress& address, int repeat) const
|
|
{
|
|
if (repeat <= 0) return 0;
|
|
|
|
ICMPSocket icmpSocket(_family, _dataSize, _ttl, _timeout);
|
|
|
|
ICMPEventArgs eventArgs(address, repeat, icmpSocket.dataSize(), icmpSocket.ttl());
|
|
pingBegin.notify(this, eventArgs);
|
|
|
|
for (int i = 0; i < repeat; ++i)
|
|
{
|
|
try
|
|
{
|
|
int sent = icmpSocket.sendTo(address);
|
|
if (icmpSocket.packetSize() == sent)
|
|
{
|
|
SocketAddress requestAddress(address);
|
|
++eventArgs;
|
|
int t = icmpSocket.receiveFrom(address);
|
|
poco_assert (address.host() == requestAddress.host());
|
|
eventArgs.setReplyTime(i, t);
|
|
pingReply.notify(this, eventArgs);
|
|
}
|
|
else
|
|
throw ICMPException(Poco::format("Error sending ICMP packet "
|
|
"(sent=%d, expected=%d)", sent, icmpSocket.packetSize()));
|
|
}
|
|
catch (TimeoutException&)
|
|
{
|
|
std::ostringstream os;
|
|
os << address.host().toString() << ": Request timed out.";
|
|
eventArgs.setError(i, os.str());
|
|
pingError.notify(this, eventArgs);
|
|
continue;
|
|
}
|
|
catch (ICMPException& ex)
|
|
{
|
|
std::ostringstream os;
|
|
os << address.host().toString() << ": " << ex.displayText();
|
|
eventArgs.setError(i, os.str());
|
|
pingError.notify(this, eventArgs);
|
|
continue;
|
|
}
|
|
catch (Exception& ex)
|
|
{
|
|
eventArgs.setError(i, ex.displayText());
|
|
pingError.notify(this, eventArgs);
|
|
continue;
|
|
}
|
|
}
|
|
pingEnd.notify(this, eventArgs);
|
|
return eventArgs.received();
|
|
}
|
|
|
|
|
|
int ICMPClient::pingIPv4(const std::string& address, int repeat,
|
|
int dataSize, int ttl, int timeout)
|
|
{
|
|
if (repeat <= 0) return 0;
|
|
|
|
SocketAddress a(address, 0);
|
|
return ping(a, IPAddress::IPv4, repeat, dataSize, ttl, timeout);
|
|
}
|
|
|
|
|
|
int ICMPClient::ping(SocketAddress& address,
|
|
IPAddress::Family family, int repeat,
|
|
int dataSize, int ttl, int timeout)
|
|
{
|
|
if (repeat <= 0) return 0;
|
|
|
|
ICMPSocket icmpSocket(family, dataSize, ttl, timeout);
|
|
int received = 0;
|
|
|
|
for (int i = 0; i < repeat; ++i)
|
|
{
|
|
try
|
|
{
|
|
SocketAddress requestAddress(address);
|
|
if (icmpSocket.sendTo(address) == icmpSocket.packetSize())
|
|
{
|
|
icmpSocket.receiveFrom(address);
|
|
poco_assert (address.host() == requestAddress.host());
|
|
++received;
|
|
}
|
|
}
|
|
catch (Exception&) { }
|
|
}
|
|
return received;
|
|
}
|
|
|
|
|
|
} } // namespace Poco::Net
|