Beesnest/Windows/DcSocket.cpp

Go to the documentation of this file.

/*
Copyright 2007 Erez Bibi (erezbibi@users.sourceforge.net)
This file is part of Beesnest.

Beesnest is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 2 of the License, or
(at your option) any later version.

Beesnest is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with Beesnest; if not, write to the Free Software
Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
*/

/*
 * DcSocket.cpp: Implementation of CdcSocket class.
 * for documentation see the header file.
 */

#include "DcSocket.h"

WSADATA CdcSocket::oData;

int CdcSocket::nReqMaxLength = 0;
int CdcSocket::nWaitReqTime = 0;

/* Constructor. */
CdcSocket::CdcSocket () throw (CdcException)
{
    /* Reset all the data fields. */
    oSocket = INVALID_SOCKET;
    oClient.sin_port = 0;
    oClient.sin_addr.S_un.S_addr = 0;
    StartNewRequest ();
    aBuffer = new char [nReqMaxLength];
    if (aBuffer == NULL)
        throw CdcException ("Cannot allocate memory");
}


/* Destructor. */
CdcSocket::~CdcSocket ()
{
    /* Close socket (there is no point to handle errors). */
    closesocket (oSocket);
    /* Delete the buffer. */
    if (aBuffer != NULL)
        delete[] aBuffer;
}



CdcSocket* CdcSocket::GetNewSocket (SOCKET soc, SOCKADDR_IN from) throw (CdcException)
{
    CdcSocket* psock = new CdcSocket;
    psock->Init (soc, from);
    return psock;
}


void CdcSocket::Init (SOCKET soc, SOCKADDR_IN from) throw (CdcException)
{
    /* Init socket. */
    Init (soc);
    /* Wrap the Client object. */
    oClient = from;
}

void CdcSocket::Init (SOCKET soc) throw (CdcException)
{
    /* Some error in socket? */
    if (soc == INVALID_SOCKET)
        throw CdcExceptionSocket (WSAGetLastError());

    /* Just wrap this socket. */
    oSocket = soc;

    /* Set receive timeout. */
    int ret = setsockopt (oSocket, SOL_SOCKET, SO_RCVTIMEO, (char *)&nWaitReqTime,
                        sizeof (nWaitReqTime));

    /* Some error? */
    if (ret == SOCKET_ERROR)
        throw CdcExceptionSocket (WSAGetLastError());
}


int CdcSocket::Read (char* buff, int lng) throw (CdcException)
{
    /* Read the message from the client. */
    int ret = recv (oSocket, buff, lng, 0);
    nReqLength += ret;
    /* This is "stack overflow" attack protection (?),
        even though buff is not on the stack. */
    if (nReqLength >= nReqMaxLength)
        throw CdcExceptionSocket (WSAEMSGSIZE);
    /* Some error? */
    if (ret == SOCKET_ERROR)
        throw CdcExceptionSocket (WSAGetLastError());
    /* Client reset connection? */
    if (ret == 0)
        throw CdcExceptionSocket (WSAECONNRESET);
    return ret;
}


int CdcSocket::Read (CdcString& msg) throw (CdcException)
{
    int ret = Read (aBuffer, nReqMaxLength - 1);
    aBuffer[ret] = '\0';
    msg += aBuffer;
    return ret;
}


int CdcSocket::Write (const char* buff, int lng) throw (CdcException)
{
    /* Send the buffer. */
    int ret = send (oSocket, buff, lng, 0);

    /* Some error? */
    if (ret == SOCKET_ERROR)
        throw CdcExceptionSocket (WSAGetLastError());

    return ret;
}


const CdcString& CdcSocket::GetClientAddress ()
{
    if (sClientIP.IsEmpty () == false) return sClientIP;
    if (oClient.sin_addr.S_un.S_addr == 0) return EMPTY_STR;

    IN_ADDR& adr = oClient.sin_addr;
    /* Return the IP address of the client. */
    sClientIP = CdcString::IntToString (adr.S_un.S_un_b.s_b1) + '.' +
                CdcString::IntToString (adr.S_un.S_un_b.s_b2) + '.' +
                CdcString::IntToString (adr.S_un.S_un_b.s_b3) + '.' +
                CdcString::IntToString (adr.S_un.S_un_b.s_b4);

    return sClientIP;
}


int CdcSocket::GetClientPort () const
{
    return oClient.sin_port;
}


void CdcSocket::Startup (int max_req_len, int wait_req_time)
    throw (CdcException)
{
    nReqMaxLength = max_req_len;
    nWaitReqTime = wait_req_time;

    /* load dll. */
    int ret = WSAStartup (0x202, &oData);

    /* Some error? */
    if (ret == SOCKET_ERROR)
        throw CdcExceptionSocket (WSAGetLastError());
}


void CdcSocket::Cleanup () throw (CdcException)
{
    /* unregister dll. */
    int ret = WSACleanup();

    /* Some error? */
    if (ret == SOCKET_ERROR)
        throw CdcExceptionSocket (WSAGetLastError());
}