Add PeerFinder::Checker for testing endpoints

src/ripple/peerfinder/impl/Checker.cpp

@@ -0,0 +1,183 @@
+//------------------------------------------------------------------------------
+/*
+    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.
+*/
+//==============================================================================
+
+namespace ripple {
+namespace PeerFinder {
+
+class CheckerImp
+    : public Checker
+    , private Thread
+    , private LeakChecked <CheckerImp>
+{
+private:
+    class Request;
+
+    struct State
+    {
+        List <Request> list;
+    };
+
+    typedef SharedData <State> SharedState;
+
+    SharedState m_state;
+    boost::asio::io_service m_io_service;
+    boost::optional <boost::asio::io_service::work> m_work;
+
+    //--------------------------------------------------------------------------
+
+    static boost::asio::ip::tcp::endpoint fromIPEndpoint (
+        IPEndpoint const& ipEndpoint)
+    {
+        if (ipEndpoint.isV4 ())
+        {
+            return boost::asio::ip::tcp::endpoint (
+                boost::asio::ip::address_v4 (
+                    ipEndpoint.v4().value),
+                        ipEndpoint.port ());
+        }
+        bassertfalse;
+        return boost::asio::ip::tcp::endpoint ();
+    }
+
+    //--------------------------------------------------------------------------
+
+    class Request
+        : public SharedObject
+        , public List <Request>::Node
+        , private LeakChecked <Request>
+    {
+    public:
+        typedef SharedPtr <Request>         Ptr;
+        typedef boost::asio::ip::tcp        Protocol;
+        typedef boost::system::error_code   error_code;
+        typedef Protocol::socket            socket_type;
+        typedef Protocol::endpoint          endpoint_type;
+
+        CheckerImp& m_owner;
+        boost::asio::io_service& m_io_service;
+        IPEndpoint m_address;
+        AbstractHandler <void (Result)> m_handler;
+        socket_type m_socket;
+        boost::system::error_code m_error;
+        bool m_canAccept;
+
+        Request (CheckerImp& owner, boost::asio::io_service& io_service,
+            IPEndpoint const& address, AbstractHandler <void (Result)> handler)
+            : m_owner (owner)
+            , m_io_service (io_service)
+            , m_address (address)
+            , m_handler (handler)
+            , m_socket (m_io_service)
+            , m_canAccept (false)
+        {
+            m_owner.add (*this);
+
+            m_socket.async_connect (fromIPEndpoint (m_address),
+                wrapHandler (boost::bind (&Request::handle_connect, Ptr(this),
+                    boost::asio::placeholders::error), m_handler));
+        }
+
+        ~Request ()
+        {
+            Result result;
+            result.address = m_address;
+            result.error = m_error;
+            m_io_service.wrap (m_handler) (result);
+
+            m_owner.remove (*this);
+        }
+
+        void cancel ()
+        {
+            m_socket.cancel();
+        }
+
+        void handle_connect (boost::system::error_code ec)
+        {
+            m_error = ec;
+            if (ec)
+                return;
+
+            m_canAccept = true;
+        }
+    };
+
+    //--------------------------------------------------------------------------
+
+    void add (Request& request)
+    {
+        SharedState::Access state (m_state);
+        state->list.push_back (request);
+    }
+
+    void remove (Request& request)
+    {
+        SharedState::Access state (m_state);
+        state->list.erase (state->list.iterator_to (request));
+    }
+
+    void run ()
+    {
+        m_io_service.run ();
+    }
+
+public:
+    CheckerImp ()
+        : Thread ("PeerFinder::Checker")
+        , m_work (boost::in_place (boost::ref (m_io_service)))
+    {
+        startThread ();
+    }
+
+    ~CheckerImp ()
+    {
+        // cancel pending i/o
+        cancel();
+
+        // destroy the io_service::work object
+        m_work = boost::none;
+
+        // signal and wait for the thread to exit gracefully
+        stopThread ();
+    }
+
+    void cancel ()
+    {
+        SharedState::Access state (m_state);
+        for (List <Request>::iterator iter (state->list.begin());
+            iter != state->list.end(); ++iter)
+            iter->cancel();
+    }
+
+    void async_test (IPEndpoint const& endpoint,
+        AbstractHandler <void (Result)> handler)
+    {
+        new Request (*this, m_io_service, endpoint, handler);
+    }
+};
+
+//------------------------------------------------------------------------------
+
+Checker* Checker::New ()
+{
+    return new CheckerImp;
+}
+
+}
+}