---

socket.cpp

Go to the documentation of this file.

#include "sysconfig.h"
#include "socket.h"

Socket::Socket(SOCKET sock, int size) : mSock(sock), mSize(size) {
  mpInBuffer = new char[mSize];
  mpEod = mpRead = mpInBuffer;
  mpOutBuffer = new char[mSize];
  mpWrite = mpStartWrite = mpOutBuffer;
}

Socket::Socket(const Socket& r_sock)
  : mSock(r_sock.mSock), mSize(r_sock.mSize) {
  // Set up new input buffer and pointers exactly
  mpInBuffer = new char[mSize];
  memcpy(mpInBuffer, r_sock.mpInBuffer, mSize);
  mpEod = mpInBuffer + (r_sock.mpEod - r_sock.mpInBuffer);
  mpRead = mpInBuffer + (r_sock.mpRead - r_sock.mpInBuffer);
  // Set up new output buffer and pointers exactly
  mpOutBuffer = new char[mSize];
  memcpy(mpOutBuffer, r_sock.mpOutBuffer, mSize);
  mpWrite = mpOutBuffer + (r_sock.mpWrite - r_sock.mpOutBuffer);
  mpStartWrite = mpOutBuffer + (r_sock.mpStartWrite - r_sock.mpOutBuffer);
}

Socket::~Socket() {
  delete[] mpInBuffer;
  delete[] mpOutBuffer;
}

char Socket::Read() {
  if (mpRead == mpEod) {              // Nothing in bufffer. Fetch more data.
    mpEod = mpRead = mpInBuffer;      // reset buffer for a full load.
    int bytes = recv (mSock, mpInBuffer, mSize, 0);
    if (bytes == 0) {
      throw ShutdownException();      // Connection was shutdown.
    } else if (bytes == SOCKET_ERROR) {
      int err = WSAGetLastError();
      if (err == WSAEWOULDBLOCK) {
        throw BlockingException(err); // No more input available.
      } else {
        throw IOErrorException(err);  // We have a serious error.
      }
    }
    mpEod += bytes;
  }
  char c = *mpRead;
  mpRead++;
  return c;
}

void Socket::Write(const char c) {
  if (mpWrite == mpOutBuffer + mSize) {
    // There is no room in buffer - attempt to flush it.
    Flush();
  }
  *mpWrite = c;
  mpWrite++;
  return;
}

void Socket::Write(const string& r_str) {
  string::const_iterator si = r_str.begin();
  while(si != r_str.end()) {
    Write(*si);
    si++;
  }
}

void Socket::Write(const char* p_char) {
  while(p_char) {
    Write(*p_char);
    p_char++;
  }
}

void Socket::Write(const char* p_char, int len) {
  while(p_char && len) {
    Write(*p_char);
    p_char++;
    len--;
  }
}

void Socket::Flush() {
  int wlen = mpWrite - mpStartWrite;
  if (wlen == 0) {  // Nothing to write
    if (mpStartWrite == mpOutBuffer + mSize) { // Oh we're at end of buffer
      mpWrite = mpStartWrite = mpOutBuffer; // reinitialize buffer
    }
    return;
  }
  int bytes = send (mSock, mpStartWrite, wlen, 0);
  if (bytes == SOCKET_ERROR) {
    int err = WSAGetLastError ();
    if (err == WSAEWOULDBLOCK) {
      throw BlockingException(); // will block
    } else {
      throw IOErrorException(err); // severe error
    }
  } else if (bytes < wlen) {  // We haven't written it all
    mpStartWrite += bytes;
    return;
  }
  // Okay everything was written - reinitalize buffer
  mpWrite = mpStartWrite = mpOutBuffer; // reinitialize buffer
  return;
}

SOCKET Socket::GetSocket() {
  return mSock;
}

void Socket::Close() {
  shutdown(mSock, SD_SEND); // Inform the client so they don't hang
  try {
    // Attempt to read all the data left in the socket's buffer.  This
    // should throw an exception which we will catch gracefully.
    while (true) Read();
  }
  catch (exception &ex) {
    closesocket(mSock);
  }
}

---