/src/SFML/Network/SocketSelector.cpp
C++ | 205 lines | 113 code | 51 blank | 41 comment | 14 complexity | d59fef8b85bb5cb1b3902ccdbdfadea0 MD5 | raw file
- ////////////////////////////////////////////////////////////
- //
- // SFML - Simple and Fast Multimedia Library
- // Copyright (C) 2007-2019 Laurent Gomila (laurent@sfml-dev.org)
- //
- // This software is provided 'as-is', without any express or implied warranty.
- // In no event will the authors be held liable for any damages arising from the use of this software.
- //
- // Permission is granted to anyone to use this software for any purpose,
- // including commercial applications, and to alter it and redistribute it freely,
- // subject to the following restrictions:
- //
- // 1. The origin of this software must not be misrepresented;
- // you must not claim that you wrote the original software.
- // If you use this software in a product, an acknowledgment
- // in the product documentation would be appreciated but is not required.
- //
- // 2. Altered source versions must be plainly marked as such,
- // and must not be misrepresented as being the original software.
- //
- // 3. This notice may not be removed or altered from any source distribution.
- //
- ////////////////////////////////////////////////////////////
- ////////////////////////////////////////////////////////////
- // Headers
- ////////////////////////////////////////////////////////////
- #include <SFML/Network/SocketSelector.hpp>
- #include <SFML/Network/Socket.hpp>
- #include <SFML/Network/SocketImpl.hpp>
- #include <SFML/System/Err.hpp>
- #include <algorithm>
- #include <utility>
- #ifdef _MSC_VER
- #pragma warning(disable: 4127) // "conditional expression is constant" generated by the FD_SET macro
- #endif
- namespace sf
- {
- ////////////////////////////////////////////////////////////
- struct SocketSelector::SocketSelectorImpl
- {
- fd_set allSockets; //!< Set containing all the sockets handles
- fd_set socketsReady; //!< Set containing handles of the sockets that are ready
- int maxSocket; //!< Maximum socket handle
- int socketCount; //!< Number of socket handles
- };
- ////////////////////////////////////////////////////////////
- SocketSelector::SocketSelector() :
- m_impl(new SocketSelectorImpl)
- {
- clear();
- }
- ////////////////////////////////////////////////////////////
- SocketSelector::SocketSelector(const SocketSelector& copy) :
- m_impl(new SocketSelectorImpl(*copy.m_impl))
- {
- }
- ////////////////////////////////////////////////////////////
- SocketSelector::~SocketSelector()
- {
- delete m_impl;
- }
- ////////////////////////////////////////////////////////////
- void SocketSelector::add(Socket& socket)
- {
- SocketHandle handle = socket.getHandle();
- if (handle != priv::SocketImpl::invalidSocket())
- {
- #if defined(SFML_SYSTEM_WINDOWS)
- if (m_impl->socketCount >= FD_SETSIZE)
- {
- err() << "The socket can't be added to the selector because the "
- << "selector is full. This is a limitation of your operating "
- << "system's FD_SETSIZE setting.";
- return;
- }
- if (FD_ISSET(handle, &m_impl->allSockets))
- return;
- m_impl->socketCount++;
- #else
- if (handle >= FD_SETSIZE)
- {
- err() << "The socket can't be added to the selector because its "
- << "ID is too high. This is a limitation of your operating "
- << "system's FD_SETSIZE setting.";
- return;
- }
- // SocketHandle is an int in POSIX
- m_impl->maxSocket = std::max(m_impl->maxSocket, handle);
- #endif
- FD_SET(handle, &m_impl->allSockets);
- }
- }
- ////////////////////////////////////////////////////////////
- void SocketSelector::remove(Socket& socket)
- {
- SocketHandle handle = socket.getHandle();
- if (handle != priv::SocketImpl::invalidSocket())
- {
- #if defined(SFML_SYSTEM_WINDOWS)
- if (!FD_ISSET(handle, &m_impl->allSockets))
- return;
- m_impl->socketCount--;
- #else
- if (handle >= FD_SETSIZE)
- return;
- #endif
- FD_CLR(handle, &m_impl->allSockets);
- FD_CLR(handle, &m_impl->socketsReady);
- }
- }
- ////////////////////////////////////////////////////////////
- void SocketSelector::clear()
- {
- FD_ZERO(&m_impl->allSockets);
- FD_ZERO(&m_impl->socketsReady);
- m_impl->maxSocket = 0;
- m_impl->socketCount = 0;
- }
- ////////////////////////////////////////////////////////////
- bool SocketSelector::wait(Time timeout)
- {
- // Setup the timeout
- timeval time;
- time.tv_sec = static_cast<long>(timeout.asMicroseconds() / 1000000);
- time.tv_usec = static_cast<long>(timeout.asMicroseconds() % 1000000);
- // Initialize the set that will contain the sockets that are ready
- m_impl->socketsReady = m_impl->allSockets;
- // Wait until one of the sockets is ready for reading, or timeout is reached
- // The first parameter is ignored on Windows
- int count = select(m_impl->maxSocket + 1, &m_impl->socketsReady, NULL, NULL, timeout != Time::Zero ? &time : NULL);
- return count > 0;
- }
- ////////////////////////////////////////////////////////////
- bool SocketSelector::isReady(Socket& socket) const
- {
- SocketHandle handle = socket.getHandle();
- if (handle != priv::SocketImpl::invalidSocket())
- {
- #if !defined(SFML_SYSTEM_WINDOWS)
- if (handle >= FD_SETSIZE)
- return false;
- #endif
- return FD_ISSET(handle, &m_impl->socketsReady) != 0;
- }
- return false;
- }
- ////////////////////////////////////////////////////////////
- SocketSelector& SocketSelector::operator =(const SocketSelector& right)
- {
- SocketSelector temp(right);
- std::swap(m_impl, temp.m_impl);
- return *this;
- }
- } // namespace sf