socketSet.cpp

/*
  TCP/IPソケットの受信イベント管理
  Satofumi KAMIMURA
  $Id$
*/

#include "socketSet.h"
#include <algorithm>


void SocketSet::destruct(void) {
  for (list<TCPsocket>::iterator it = sockets.begin();
       it != sockets.end(); ++it) {
    SDLNet_TCP_DelSocket(socket_set, *it);
  }
  if (socket_set) {
    SDLNet_FreeSocketSet(socket_set);
  }
}


void SocketSet::resize(int size) {
  destruct();
  socket_set = SDLNet_AllocSocketSet(size);
  num_set = size;

  for (list<TCPsocket>::iterator it = sockets.begin();
       it != sockets.end(); ++it) {
    SDLNet_TCP_AddSocket(socket_set, *it);
  }
}


SocketSet::SocketSet(unsigned int num) : num_set(num), socket_set(NULL) {
  resize(num);
}


SocketSet::SocketSet(void) : num_set(1), socket_set(NULL) {
  resize(1);
}


SocketSet::~SocketSet(void) {
  destruct();
  sockets.clear();
}


void SocketSet::add(SocketCtrl& socket) {
  if (sockets.size() >= num_set) {
    resize(sockets.size() * 2);
  }

  TCPsocket tcp_socket = socket.getTCPsocket();
  SDLNet_TCP_AddSocket(socket_set, tcp_socket);
  sockets.push_back(tcp_socket);
}


void SocketSet::del(SocketCtrl& socket) {
  TCPsocket tcp_socket = socket.getTCPsocket();
  SDLNet_TCP_DelSocket(socket_set, tcp_socket);

  list<TCPsocket>::iterator it
    = find(sockets.begin(), sockets.end(), tcp_socket);
  if (it != sockets.end()) {
    sockets.erase(it);
  }
}


int SocketSet::check(int timeout) {
  return SDLNet_CheckSockets(socket_set, timeout);
}