Beesnest/Windows/DcSocketServer.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
*/


/*
 * DcSocketServer.cpp: Implementation of CdcSocketServer class.
 * for documentation see the header file.
 */

#include "DcSocketServer.h"

#define DEFAULT_PORT    80

int CdcSocketServer::nObjCounter = 0;

CdcSocketServer* CdcSocketServer::GetSocketServer (int port) throw (CdcException)
{
    /* Check for just one server socket object. */
    if ( nObjCounter > 0 )
        throw CdcException ("More than one server socket.");
    nObjCounter++;

    SOCKADDR_IN local;
    SOCKET servsock;

    /* Init socket address Struct. */
    local.sin_family = AF_INET;
    local.sin_addr.s_addr = INADDR_ANY;
    local.sin_port = htons (port);  /* Port MUST be in Network Byte Order. */

    /* Make socket object. */
    servsock = socket (AF_INET, SOCK_STREAM, 0);    /* TCP socket */
    if (servsock == INVALID_SOCKET)
        throw CdcExceptionSocket (WSAGetLastError());

    /* Bind the server socket. */
    if ( bind (servsock,(struct sockaddr*)&local, sizeof(local) )
        == SOCKET_ERROR)
        throw CdcExceptionSocket (WSAGetLastError());

    /* Init the wrapping object. */
    CdcSocketServer* psocsrv = new CdcSocketServer;
    psocsrv->Init (servsock);   /* may throw CdcException. */

    /* Return by copy the object. */
    return psocsrv;
}


CdcSocketServer* CdcSocketServer::GetSocketServer () throw (CdcException)
{
    return GetSocketServer (DEFAULT_PORT);
}


CdcSocket* CdcSocketServer::Listen () throw (CdcException)
{
    SOCKADDR_IN from;
    int fromlen = sizeof(from);

    /* Listen to the port. */
    int ret = listen (oSocket, SOMAXCONN);

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

    /* Accept connection. */
    SOCKET msgsock = accept (oSocket, (struct sockaddr*)&from, &fromlen);

    /* Wrap in new CdcSocket object. */
    CdcSocket* psock = CdcSocket::GetNewSocket (msgsock, from);

    /* Return the pointer to the object. */
    return psock;
}


CdcSocket* CdcSocketServer::ListenNonBlock (int wait) throw (CdcException)
{
    FD_SET set;
    FD_ZERO (&set);
    FD_SET (oSocket, &set);
    struct timeval swait = {0, wait * 1000};
    int ret;
    SOCKADDR_IN from;
    int fromlen = sizeof(from);

    /* Put socket in listen mode. */
    ret = listen (oSocket, 0);
    if (ret == SOCKET_ERROR)
        throw CdcExceptionSocket (WSAGetLastError());
    /* Wait for 'wait' mSec for an incoming connection. */
    ret = select (1, &set, NULL, NULL, &swait);
    /* Time out. */
    if (ret == 0) return NULL;
    /* Some error. */
    if (ret  == SOCKET_ERROR)
        throw CdcExceptionSocket (WSAGetLastError());
    /* Accept connection. */
    SOCKET msgsock = accept (oSocket, (struct sockaddr*)&from, &fromlen);
    /* Wrap in new CdcSocket object. */
    CdcSocket* psock = CdcSocket::GetNewSocket (msgsock, from);
    /* Return the pointer to the object. */
    return psock;
}