SocketSelector.h

/*
 * Gamedev Framework (gf)
 * Copyright (C) 2016-2019 Julien Bernard
 *
 * 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.
 */
#ifndef GF_SOCKET_SELECTOR_H
#define GF_SOCKET_SELECTOR_H

#include <vector>

#ifdef _WIN32
#include <winsock2.h>
#else
#include <poll.h>
#endif

#include "Portability.h"
#include "Socket.h"
#include "SocketGuard.h"
#include "Time.h"

namespace gf {
#ifndef DOXYGEN_SHOULD_SKIP_THIS
inline namespace v1 {
#endif

  enum SocketSelectorStatus {
    Event,    
    Timeout,  
    Error,    
  };

  class GF_API SocketSelector : private SocketGuard {
  public:
    SocketSelector();

    void addSocket(Socket& socket);

    void removeSocket(Socket& socket);

    void clear();

    SocketSelectorStatus wait(Time duration = seconds(-1));

    bool isReady(Socket& socket);

  private:
    std::vector<pollfd>::iterator find(Socket& socket);

    static SocketSelectorStatus nativePoll(std::vector<pollfd>& fds, Time duration);
  private:
    std::vector<pollfd> m_fds;
    bool m_sorted;
  };

#ifndef DOXYGEN_SHOULD_SKIP_THIS
}
#endif
}

#endif // GF_SOCKET_SELECTOR_H