cNetSocketSet.h

Go to the documentation of this file.

//
//
 
#ifndef _INC_cNetSocketSet_H
#define _INC_cNetSocketSet_H
#ifndef NO_PRAGMA_ONCE
#pragma once
#endif
 
#include "cNetSocket.h"
#include "GrayCore/include/cTimeVal.h"
 
namespace GrayLib
{
        class GRAYLIB_LINK cNetSocketSet
        {
 
        public:
                static const int k_nSocketSetSize; 
                fd_set m_fds;   
 
        public:
                cNetSocketSet()
                {
                        ClearSockets();
                }
                cNetSocketSet(SOCKET hSocket)
                {
                        ClearSockets();
                        AddSocket(hSocket);
                }
                cNetSocketSet(const cNetSocket& socket)
                {
                        ClearSockets();
                        AddSocket(socket.get_HSocket());
                }
                cNetSocketSet(const fd_set* pfds)
                {
                        ClearSockets();
                        CopySockets(pfds);
                }
                cNetSocketSet(const cNetSocketSet& nss)
                {
                        ClearSockets();
                        CopySockets(&nss.m_fds);
                }
                ~cNetSocketSet()
                {
                }
 
                void operator = (const fd_set* pfds)
                {
                        CopySockets(pfds);
                }
                void operator = (const cNetSocketSet& nss)
                {
                        CopySockets(&nss.m_fds);
                }
 
                void CopySockets(const fd_set* pfds)
                {
                        memcpy(&m_fds, pfds, sizeof(m_fds));
                }
 
                bool AddSocket(SOCKET hSocket);
                void RemoveSocket(SOCKET hSocket);
 
                bool IsSocketSet(SOCKET hSocket) const
                {
                        if (hSocket == INVALID_SOCKET)
                                return false;
                        return FD_ISSET(hSocket, &m_fds) ? true : false;
                }
                void ClearSockets() noexcept
                {
                        FD_ZERO(&m_fds);
                }
                int get_NFDS() const noexcept
                {
#ifdef _WIN32
                        return m_fds.fd_count;
#else
                        return FD_SETSIZE + 1; // __linux__ equiv?? k_nSocketSetSize
#endif
                }
                void put_NFDS(int iCount) noexcept
                {
#ifdef _WIN32
                        m_fds.fd_count = iCount;
#endif
                }
 
                operator fd_set*()
                {
                        return &m_fds;
                }
                operator const fd_set*() const
                {
                        return &m_fds;
                }
 
                static int GRAYCALL Select(fd_set* pReadSet, fd_set* pWriteSet = nullptr, fd_set* pExceptSet = nullptr, const cTimeVal* pTimeout = nullptr);
        };
};
#endif