SocketHandle.h

Go to the documentation of this file.

#ifndef NET_SocketHandle_h__
#define NET_SocketHandle_h__

#include <unistd.h>
#include "TempFailure.h"

namespace NET
{
    template<class>
    class SocketHandle_Ref;

    class TCPSocket;
    class SCTPSocket;


    template<class Socket>
    class SocketHandle
    {
    public:
        friend class TCPSocket;
        friend class SCTPSocket;

        typedef Socket socket_type;

        SocketHandle() : m_sockfd(-1) {}

        SocketHandle( SocketHandle& other)
        : m_sockfd( other.release()) {}

        SocketHandle( SocketHandle_Ref<Socket> other)
        : m_sockfd( other.sockfd) {}

        operator SocketHandle_Ref<Socket>()
        { return SocketHandle_Ref<Socket>( release()); }

        SocketHandle&
        operator=( SocketHandle& other)
        {
            reset( other.release());
            return *this;
        }

        SocketHandle&
        operator=( SocketHandle_Ref<Socket> other)
        {
            reset( other.sockfd);
            return *this;
        }

        ~SocketHandle() { reset(); }


        operator bool() const { return m_sockfd >= 0; }

    private:
        explicit SocketHandle( int sockfd) : m_sockfd(sockfd) {}

        int release()
        {
            int tmp = m_sockfd;
            m_sockfd = -1;
            return tmp;
        }


        void reset( int fd = -1)
        {
            if( fd != m_sockfd)
            {
                TEMP_FAILURE_RETRY (::close(m_sockfd));
                m_sockfd = fd;
            }
        }

        int m_sockfd;
    };

    template<class Socket>
    class SocketHandle_Ref
    {
        int sockfd;

        friend class SocketHandle<Socket>;
        explicit SocketHandle_Ref( int sockfd) : sockfd(sockfd) {}
    };

} // namespace NET

#endif // NET_SocketHandle_h__