Add SharedArg and AsyncObject

modules/beast_asio/async/AsyncObject.h

@@ -0,0 +1,72 @@
+//------------------------------------------------------------------------------
+/*
+    This file is part of Beast: https://github.com/vinniefalco/Beast
+    Copyright 2013, Vinnie Falco <vinnie.falco@gmail.com>
+
+    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.
+*/
+//==============================================================================
+
+#ifndef BEAST_ASIO_ASYNCOBJECT_H_INCLUDED
+#define BEAST_ASIO_ASYNCOBJECT_H_INCLUDED
+
+/** Mix-in to track when all pending I/O is complete.
+    Derived classes must be callable with this signature:
+        void asyncHandlersComplete()
+*/
+template <class Derived>
+class AsyncObject
+{
+public:
+    ~AsyncObject ()
+    {
+        // Destroying the object with I/O pending? Not a clean exit!
+        bassert (m_pending.get() == 0);
+    }
+
+    /** RAII container that maintains the count of pending I/O.
+        Bind this into the argument list of every handler passed
+        to an initiating function.
+    */
+    class CompletionCounter
+    {
+    public:
+        explicit CompletionCounter (Derived* owner)
+            : m_owner (owner)
+        {
+            ++m_owner->m_pending;
+        }
+
+        CompletionCounter (CompletionCounter const& other)
+            : m_owner (other.m_owner)
+        {
+            ++m_owner->m_pending;
+        }
+
+        ~CompletionCounter ()
+        {
+            if (--m_owner->m_pending == 0)
+                m_owner->asyncHandlersComplete ();
+        }
+
+    private:
+        CompletionCounter& operator= (CompletionCounter const&);
+        Derived* m_owner;
+    };
+
+private:
+    // The number of handlers pending.
+    Atomic <int> m_pending;
+};
+
+#endif

modules/beast_asio/basics/SharedArg.h

@@ -0,0 +1,161 @@
+//------------------------------------------------------------------------------
+/*
+    This file is part of Beast: https://github.com/vinniefalco/Beast
+    Copyright 2013, Vinnie Falco <vinnie.falco@gmail.com>
+
+    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.
+*/
+//==============================================================================
+
+#ifndef BEAST_ASIO_SHAREDARG_H_INCLUDED
+#define BEAST_ASIO_SHAREDARG_H_INCLUDED
+
+/** A container that turns T into a SharedObject.
+    We use this to manage the lifetime of objects passed to handlers.
+*/
+template <typename T>
+struct SharedArg
+{
+private:
+    struct Arg : SharedObject
+    {
+        Arg ()
+        {
+        }
+
+        explicit Arg (BEAST_MOVE_ARG(T) t)
+            : value (BEAST_MOVE_CAST(T)(t))
+        {
+        }
+
+        template <class P1>
+        explicit Arg (P1 p1)
+            : value (p1)
+        {
+        }
+
+        template <class P1, class P2>
+        Arg (P1 p1, P2 p2)
+            : value (p1, p2)
+        {
+        }
+
+        template <class P1, class P2, class P3>
+        Arg (P1 p1, P2 p2, P3 p3)
+            : value (p1, p2, p3)
+        {
+        }
+
+        template <class P1, class P2, class P3, class P4>
+        Arg (P1 p1, P2 p2, P3 p3, P4 p4)
+            : value (p1, p2, p3, p4)
+        {
+        }
+
+        ~Arg ()
+        {
+        }
+
+        T value;
+    };
+
+public:
+    SharedArg ()
+    {
+    }
+
+    explicit SharedArg (BEAST_MOVE_ARG(T) t)
+        : m_arg (new Arg (BEAST_MOVE_CAST(T)(t)))
+    {
+    }
+
+    template <class P1>
+    explicit SharedArg (P1 p1)
+        : m_arg (new Arg (p1))
+    {
+    }
+
+    template <class P1, class P2>
+    SharedArg (P1 p1, P2 p2)
+        : m_arg (new Arg (p1, p2))
+    {
+    }
+
+    template <class P1, class P2, class P3>
+    SharedArg (P1 p1, P2 p2, P3 p3)
+        : m_arg (new Arg (p1, p2, p3))
+    {
+    }
+
+    template <class P1, class P2, class P3, class P4>
+    SharedArg (P1 p1, P2 p2, P3 p3, P4 p4)
+        : m_arg (new Arg (p1, p2, p3, p4))
+    {
+    }
+
+    SharedArg (SharedArg const& other)
+        : m_arg (other.m_arg)
+    {
+    }
+
+    SharedArg& operator= (SharedArg const& other)
+    {
+        m_arg = other.m_arg;
+        return *this;
+    }
+
+    T& get ()
+    {
+        return m_arg->value;
+    }
+
+    T const& get () const
+    {
+        return m_arg->value;
+    }
+
+    T& operator* ()
+    {
+        return get();
+    }
+
+    T const& operator* () const
+    {
+        return get();
+    }
+
+    T* operator-> ()
+    {
+        return &get();
+    }
+
+    T const* operator-> () const
+    {
+        return &get();
+    }
+
+    operator T& ()
+    {
+        return m_arg->value;
+    }
+
+    operator T const& () const
+    {
+        return m_arg->value;
+    }
+
+private:
+    SharedPtr <Arg> m_arg;
+};
+
+#endif

modules/beast_asio/beast_asio.h

@@ -65,6 +65,7 @@ namespace beast
 # include "async/SharedHandlerPtr.h"
 # include "async/ComposedAsyncOperation.h"
 #include "async/SharedHandlerAllocator.h"
+#include "async/AsyncObject.h"
 
 #  include "basics/BufferType.h"
 # include "basics/BuffersType.h"
@@ -72,6 +73,7 @@ namespace beast
 #include "basics/FixedInputBuffer.h"
 #include "basics/PeerRole.h"
 #include "basics/SSLContext.h"
+#include "basics/SharedArg.h"
 
 #   include "sockets/SocketBase.h"
 #  include "sockets/Socket.h"