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__