mirror of
https://github.com/Xahau/xahaud.git
synced 2025-12-06 17:27:52 +00:00
Refactor the structure of source files:
* New src/ripple/crypto and src/ripple/protocol directories * Merged src/ripple/common into src/ripple/basics * Move resource/api files up a level * Add headers for "include what you use" * Normalized include guards * Renamed to JsonFields.h * Remove obsolete files * Remove net.h unity header * Remove resource.h unity header * Removed some deprecated unity includes
This commit is contained in:
313
src/ripple/basics/impl/ResolverAsio.cpp
Normal file
313
src/ripple/basics/impl/ResolverAsio.cpp
Normal file
@@ -0,0 +1,313 @@
|
||||
//------------------------------------------------------------------------------
|
||||
/*
|
||||
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 <ripple/basics/ResolverAsio.h>
|
||||
#include <beast/asio/IPAddressConversion.h>
|
||||
#include <beast/asio/placeholders.h>
|
||||
#include <beast/module/asio/AsyncObject.h>
|
||||
#include <beast/threads/WaitableEvent.h>
|
||||
#include <boost/asio.hpp>
|
||||
#include <atomic>
|
||||
#include <cassert>
|
||||
#include <deque>
|
||||
#include <locale>
|
||||
|
||||
namespace ripple {
|
||||
|
||||
class ResolverAsioImpl
|
||||
: public ResolverAsio
|
||||
, public beast::asio::AsyncObject <ResolverAsioImpl>
|
||||
{
|
||||
public:
|
||||
typedef std::pair <std::string, std::string> HostAndPort;
|
||||
|
||||
beast::Journal m_journal;
|
||||
|
||||
boost::asio::io_service& m_io_service;
|
||||
boost::asio::io_service::strand m_strand;
|
||||
boost::asio::ip::tcp::resolver m_resolver;
|
||||
|
||||
beast::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,
|
||||
beast::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 ()
|
||||
{
|
||||
assert (m_work.empty ());
|
||||
assert (m_stopped);
|
||||
}
|
||||
|
||||
//-------------------------------------------------------------------------
|
||||
// AsyncObject
|
||||
void asyncHandlersComplete()
|
||||
{
|
||||
m_stop_complete.signal ();
|
||||
}
|
||||
|
||||
//--------------------------------------------------------------------------
|
||||
//
|
||||
// Resolver
|
||||
//
|
||||
//--------------------------------------------------------------------------
|
||||
|
||||
void start ()
|
||||
{
|
||||
assert (m_stopped == true);
|
||||
assert (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 (std::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)
|
||||
{
|
||||
assert (m_stop_called == false);
|
||||
assert (m_stopped == true);
|
||||
assert (!names.empty());
|
||||
|
||||
// TODO NIKB use rvalue references to construct and move
|
||||
// reducing cost.
|
||||
m_io_service.dispatch (m_strand.wrap (std::bind (
|
||||
&ResolverAsioImpl::do_resolve, this,
|
||||
names, handler, CompletionCounter (this))));
|
||||
}
|
||||
|
||||
//-------------------------------------------------------------------------
|
||||
// Resolver
|
||||
void do_stop (CompletionCounter)
|
||||
{
|
||||
assert (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 <beast::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 (beast::IPAddressConversion::from_asio (*iter));
|
||||
++iter;
|
||||
}
|
||||
}
|
||||
|
||||
handler (name, addresses);
|
||||
|
||||
m_io_service.post (m_strand.wrap (std::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 (std::bind (
|
||||
&ResolverAsioImpl::do_work, this,
|
||||
CompletionCounter (this))));
|
||||
|
||||
return;
|
||||
}
|
||||
|
||||
boost::asio::ip::tcp::resolver::query query (
|
||||
hp.first, hp.second);
|
||||
|
||||
m_resolver.async_resolve (query, std::bind (
|
||||
&ResolverAsioImpl::do_finish, this, name,
|
||||
beast::asio::placeholders::error, handler,
|
||||
beast::asio::placeholders::iterator,
|
||||
CompletionCounter (this)));
|
||||
}
|
||||
|
||||
void do_resolve (std::vector <std::string> const& names,
|
||||
HandlerType const& handler, CompletionCounter)
|
||||
{
|
||||
assert (! 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 (std::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 ()
|
||||
{
|
||||
}
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user