mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-10-04 15:40:49 +00:00
147 lines
2.7 KiB
C++
147 lines
2.7 KiB
C++
//
|
|
// ICMPClient.cpp
|
|
//
|
|
// $Id: //poco/1.4/Net/src/ICMPClient.cpp#1 $
|
|
//
|
|
// 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/ICMPSocket.h"
|
|
#include "Poco/Net/NetException.h"
|
|
#include "Poco/Channel.h"
|
|
#include "Poco/Message.h"
|
|
#include "Poco/Exception.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(IPAddress::Family family):
|
|
_family(family)
|
|
{
|
|
}
|
|
|
|
|
|
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);
|
|
SocketAddress returnAddress;
|
|
|
|
ICMPEventArgs eventArgs(address, repeat, icmpSocket.dataSize(), icmpSocket.ttl());
|
|
pingBegin.notify(this, eventArgs);
|
|
|
|
for (int i = 0; i < repeat; ++i)
|
|
{
|
|
icmpSocket.sendTo(address);
|
|
++eventArgs;
|
|
|
|
try
|
|
{
|
|
int t = icmpSocket.receiveFrom(returnAddress);
|
|
eventArgs.setReplyTime(i, t);
|
|
pingReply.notify(this, eventArgs);
|
|
}
|
|
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.what();
|
|
eventArgs.setError(i, os.str());
|
|
pingError.notify(this, eventArgs);
|
|
continue;
|
|
}
|
|
catch (Exception& ex)
|
|
{
|
|
std::ostringstream os;
|
|
os << ex.displayText();
|
|
eventArgs.setError(i, os.str());
|
|
pingError.notify(this, eventArgs);
|
|
continue;
|
|
}
|
|
}
|
|
pingEnd.notify(this, eventArgs);
|
|
return eventArgs.received();
|
|
}
|
|
|
|
|
|
int ICMPClient::pingIPv4(const std::string& address, int repeat)
|
|
{
|
|
if (repeat <= 0) return 0;
|
|
|
|
SocketAddress a(address, 0);
|
|
return ping(a, IPAddress::IPv4, repeat);
|
|
}
|
|
|
|
|
|
int ICMPClient::ping(SocketAddress& address, IPAddress::Family family, int repeat)
|
|
{
|
|
if (repeat <= 0) return 0;
|
|
|
|
ICMPSocket icmpSocket(family);
|
|
SocketAddress returnAddress;
|
|
int received = 0;
|
|
|
|
for (int i = 0; i < repeat; ++i)
|
|
{
|
|
icmpSocket.sendTo(address);
|
|
try
|
|
{
|
|
icmpSocket.receiveFrom(returnAddress);
|
|
++received;
|
|
}
|
|
catch (TimeoutException&)
|
|
{
|
|
}
|
|
catch (ICMPException&)
|
|
{
|
|
}
|
|
}
|
|
return received;
|
|
}
|
|
|
|
|
|
} } // namespace Poco::Net
|