socketCtrl.cpp

/*
  TCP/IPソケット管理
  Satofumi KAMIMURA
  $Id$
*/

#include "socketCtrl.h"


void SocketCtrl::bufferInit(int buffer_size) {
  recv_buffer = new RingBuffer<char>(buffer_size);
}


SocketCtrl::SocketCtrl(const char *host, Uint16 port, unsigned int buffer_size)
  : is_disconnected(false) {
  IPaddress ip;
  SDLNet_ResolveHost(&ip, host, port);
  tcp_socket = SDLNet_TCP_Open(&ip);
  bufferInit(buffer_size);
}


SocketCtrl::SocketCtrl(TCPsocket tcpsocket, unsigned int buffer_size)
  : is_disconnected(false), tcp_socket(tcpsocket) {
  bufferInit(buffer_size);
}


SocketCtrl::~SocketCtrl(void) {
  if (tcp_socket) {
    SDLNet_TCP_Close(tcp_socket);
    tcp_socket = NULL;
  }
}


void SocketCtrl::disconnect(void) {
  is_disconnected = true;
}


bool SocketCtrl::is_connect(void) {
  return ((tcp_socket != NULL) && !is_disconnected) ? true : false;
}


TCPsocket& SocketCtrl::getTCPsocket(void) {
  return tcp_socket;
}


void SocketCtrl::check(void) {
  if (is_connect() && SDLNet_SocketReady(tcp_socket)) {
    int maxlen = recv_buffer->free_size();
    maxlen = (maxlen > BUFSIZ) ? BUFSIZ : maxlen;
    char buffer[BUFSIZ];
    int n = SDLNet_TCP_Recv(tcp_socket, buffer, maxlen);
    if (n <= 0) {
      is_disconnected = true;
      return;
    }
    recv_buffer->put(buffer, n);
  }
}


int SocketCtrl::recv(void *area, unsigned int maxlen) {
  check();
  if (!is_connect()) {
    return -1;
  }

  int len = (recv_buffer->size() > maxlen) ? maxlen : recv_buffer->size();
  if (len == 0) {
    return 0;
  }
  recv_buffer->get((char *)area, maxlen);
  
  return len;
}


int SocketCtrl::send(const void *area, int len) {
  check();
  if (!is_connect()) {
    return -1;
  }
  
  return SDLNet_TCP_Send(tcp_socket, (void *)area, len);
}


unsigned long SocketCtrl::size(void) {
  check();
  return recv_buffer->size();
}