mirror of
https://github.com/Xahau/xahaud.git
synced 2025-12-06 17:27:52 +00:00
c2fd1215f5
* New tools for completion handlers: - wrap_handler provides composed io_service execution guarantees. - bind_handler rebinds arguments to handlers. - shared_handler type-erases any completion handler. - buffer_sequence type-erases templated BufferSequences - abstract_socket replaces Socket - socket_wrapper replaces SocketWrapper - beast::asio placeholders to work with std::bind * Removed obsolete classes and functions - AbstractHandler - ComposedAsyncOperation - SharedFunction - SharedHandler - SharedHandlerAllocator - SharedHandlerPtr - SharedHandlerType - SocketBase - SocketWrapperStrand - wrapHandler * Refactored classes to use new tools - abstract_socket - socket_wrapper - HandshakeDetector - HttpClientType * Miscellanous tidying - socket classes moved to beast::asio namespace - beast asio files provide their own namespace declaration. - Fix IsCallPossible conflicting template parameter name - Use <boost/get_pointer.hpp> for C++11 compatibility. - Remove extraneous include path from build environment.
314 lines
9.0 KiB
C++
314 lines
9.0 KiB
C++
//------------------------------------------------------------------------------
|
|
/*
|
|
This file is part of rippled: https://github.com/ripple/rippled
|
|
Copyright (c) 2012, 2013 Ripple Labs Inc.
|
|
|
|
Permission to use, copy, modify, and/or distribute this software for any
|
|
purpose with or without fee is hereby granted, provided that the above
|
|
copyright notice and this permission notice appear in all copies.
|
|
|
|
THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
|
|
WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
|
|
MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
|
|
ANY SPECIAL , DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
|
|
WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
|
|
ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
|
|
OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
|
|
*/
|
|
//==============================================================================
|
|
|
|
#include "../../beast/modules/beast_asio/beast_asio.h"
|
|
#include "../../beast/beast/asio/IPAddressConversion.h"
|
|
|
|
#include <atomic>
|
|
#include <deque>
|
|
|
|
#include "boost/asio.hpp"
|
|
|
|
#include "../ResolverAsio.h"
|
|
|
|
namespace ripple {
|
|
|
|
class ResolverAsioImpl
|
|
: public ResolverAsio
|
|
, public beast::asio::AsyncObject <ResolverAsioImpl>
|
|
{
|
|
public:
|
|
typedef std::pair <std::string, std::string> HostAndPort;
|
|
|
|
Journal m_journal;
|
|
|
|
boost::asio::io_service& m_io_service;
|
|
boost::asio::io_service::strand m_strand;
|
|
boost::asio::ip::tcp::resolver m_resolver;
|
|
|
|
WaitableEvent m_stop_complete;
|
|
std::atomic <bool> m_stop_called;
|
|
std::atomic <bool> m_stopped;
|
|
|
|
// Represents a unit of work for the resolver to do
|
|
struct Work
|
|
{
|
|
std::vector <std::string> names;
|
|
HandlerType handler;
|
|
|
|
template <class StringSequence>
|
|
Work (StringSequence const& names_, HandlerType const& handler_)
|
|
: handler (handler_)
|
|
{
|
|
names.reserve(names_.size ());
|
|
|
|
std::reverse_copy (names_.begin (), names_.end (),
|
|
std::back_inserter (names));
|
|
}
|
|
};
|
|
|
|
std::deque <Work> m_work;
|
|
|
|
ResolverAsioImpl (boost::asio::io_service& io_service,
|
|
Journal journal)
|
|
: m_journal (journal)
|
|
, m_io_service (io_service)
|
|
, m_strand (io_service)
|
|
, m_resolver (io_service)
|
|
, m_stop_complete (true, true)
|
|
, m_stop_called (false)
|
|
, m_stopped (true)
|
|
{
|
|
|
|
}
|
|
|
|
~ResolverAsioImpl ()
|
|
{
|
|
check_precondition (m_work.empty ());
|
|
check_precondition (m_stopped);
|
|
}
|
|
|
|
//-------------------------------------------------------------------------
|
|
// AsyncObject
|
|
void asyncHandlersComplete()
|
|
{
|
|
m_stop_complete.signal ();
|
|
}
|
|
|
|
//--------------------------------------------------------------------------
|
|
//
|
|
// Resolver
|
|
//
|
|
//--------------------------------------------------------------------------
|
|
|
|
void start ()
|
|
{
|
|
check_precondition (m_work.empty ());
|
|
check_precondition (m_stopped == true);
|
|
check_precondition (m_stop_called == false);
|
|
|
|
if (m_stopped.exchange (false) == true)
|
|
{
|
|
m_stop_complete.reset ();
|
|
addReference ();
|
|
}
|
|
}
|
|
|
|
void stop_async ()
|
|
{
|
|
if (m_stop_called.exchange (true) == false)
|
|
{
|
|
m_io_service.dispatch (m_strand.wrap (boost::bind (
|
|
&ResolverAsioImpl::do_stop,
|
|
this, CompletionCounter (this))));
|
|
|
|
m_journal.debug << "Queued a stop request";
|
|
}
|
|
}
|
|
|
|
void stop ()
|
|
{
|
|
stop_async ();
|
|
|
|
m_journal.debug << "Waiting to stop";
|
|
m_stop_complete.wait();
|
|
m_journal.debug << "Stopped";
|
|
}
|
|
|
|
void resolve (
|
|
std::vector <std::string> const& names,
|
|
HandlerType const& handler)
|
|
{
|
|
check_precondition (m_stop_called == false);
|
|
check_precondition (m_stopped == true);
|
|
check_precondition (!names.empty());
|
|
|
|
// TODO NIKB use rvalue references to construct and move
|
|
// reducing cost.
|
|
m_io_service.dispatch (m_strand.wrap (boost::bind (
|
|
&ResolverAsioImpl::do_resolve, this,
|
|
names, handler, CompletionCounter (this))));
|
|
}
|
|
|
|
//-------------------------------------------------------------------------
|
|
// Resolver
|
|
void do_stop (CompletionCounter)
|
|
{
|
|
check_precondition (m_stop_called == true);
|
|
|
|
if (m_stopped.exchange (true) == false)
|
|
{
|
|
m_work.clear ();
|
|
m_resolver.cancel ();
|
|
|
|
removeReference ();
|
|
}
|
|
}
|
|
|
|
void do_finish (
|
|
std::string name,
|
|
boost::system::error_code const& ec,
|
|
HandlerType handler,
|
|
boost::asio::ip::tcp::resolver::iterator iter,
|
|
CompletionCounter)
|
|
{
|
|
if (ec == boost::asio::error::operation_aborted)
|
|
return;
|
|
|
|
std::vector <IP::Endpoint> addresses;
|
|
|
|
// If we get an error message back, we don't return any
|
|
// results that we may have gotten.
|
|
if (ec == 0)
|
|
{
|
|
while (iter != boost::asio::ip::tcp::resolver::iterator())
|
|
{
|
|
addresses.push_back (IPAddressConversion::from_asio (*iter));
|
|
++iter;
|
|
}
|
|
}
|
|
|
|
handler (name, addresses);
|
|
|
|
m_io_service.post (m_strand.wrap (boost::bind (
|
|
&ResolverAsioImpl::do_work, this,
|
|
CompletionCounter (this))));
|
|
}
|
|
|
|
HostAndPort parseName(std::string const& str)
|
|
{
|
|
// Attempt to find the first and last non-whitespace
|
|
auto const find_whitespace = std::bind (
|
|
std::isspace <std::string::value_type>,
|
|
std::placeholders::_1,
|
|
std::locale ());
|
|
|
|
auto host_first = std::find_if_not (
|
|
str.begin (), str.end (), find_whitespace);
|
|
|
|
auto port_last = std::find_if_not (
|
|
str.rbegin (), str.rend(), find_whitespace).base();
|
|
|
|
// This should only happen for all-whitespace strings
|
|
if (host_first >= port_last)
|
|
return std::make_pair(std::string (), std::string ());
|
|
|
|
// Attempt to find the first and last valid port separators
|
|
auto const find_port_separator = [](char const c) -> bool
|
|
{
|
|
if (std::isspace (c))
|
|
return true;
|
|
|
|
if (c == ':')
|
|
return true;
|
|
|
|
return false;
|
|
};
|
|
|
|
auto host_last = std::find_if (
|
|
host_first, port_last, find_port_separator);
|
|
|
|
auto port_first = std::find_if_not (
|
|
host_last, port_last, find_port_separator);
|
|
|
|
return make_pair (
|
|
std::string (host_first, host_last),
|
|
std::string (port_first, port_last));
|
|
}
|
|
|
|
void do_work (CompletionCounter)
|
|
{
|
|
if (m_stop_called == true)
|
|
return;
|
|
|
|
// We don't have any work to do at this time
|
|
if (m_work.empty ())
|
|
return;
|
|
|
|
std::string const name (m_work.front ().names.back());
|
|
HandlerType handler (m_work.front ().handler);
|
|
|
|
m_work.front ().names.pop_back ();
|
|
|
|
if (m_work.front ().names.empty ())
|
|
m_work.pop_front();
|
|
|
|
HostAndPort const hp (parseName (name));
|
|
|
|
if (hp.first.empty ())
|
|
{
|
|
m_journal.error <<
|
|
"Unable to parse '" << name << "'";
|
|
|
|
m_io_service.post (m_strand.wrap (boost::bind (
|
|
&ResolverAsioImpl::do_work, this,
|
|
CompletionCounter (this))));
|
|
|
|
return;
|
|
}
|
|
|
|
boost::asio::ip::tcp::resolver::query query (
|
|
hp.first, hp.second);
|
|
|
|
m_resolver.async_resolve (query, boost::bind (
|
|
&ResolverAsioImpl::do_finish, this, name,
|
|
boost::asio::placeholders::error, handler,
|
|
boost::asio::placeholders::iterator,
|
|
CompletionCounter (this)));
|
|
}
|
|
|
|
void do_resolve (std::vector <std::string> const& names,
|
|
HandlerType const& handler, CompletionCounter)
|
|
{
|
|
check_precondition (! names.empty());
|
|
|
|
if (m_stop_called == false)
|
|
{
|
|
m_work.emplace_back (names, handler);
|
|
|
|
m_journal.debug <<
|
|
"Queued new job with " << names.size() <<
|
|
" tasks. " << m_work.size() << " jobs outstanding.";
|
|
|
|
if (m_work.size() == 1)
|
|
{
|
|
m_io_service.post (m_strand.wrap (boost::bind (
|
|
&ResolverAsioImpl::do_work, this,
|
|
CompletionCounter (this))));
|
|
}
|
|
}
|
|
}
|
|
};
|
|
|
|
//-----------------------------------------------------------------------------
|
|
|
|
ResolverAsio *ResolverAsio::New (
|
|
boost::asio::io_service& io_service,
|
|
beast::Journal journal)
|
|
{
|
|
return new ResolverAsioImpl (io_service, journal);
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
Resolver::~Resolver ()
|
|
{
|
|
}
|
|
|
|
}
|