NetClient.cpp

Aller à la documentation de ce fichier.

/*! \file NetClient.cpp
    \brief Systeme client.

    Client pour le systeme serveur \a NetServeur.

    \author     aerith (http://aerith.fr)
    \version    2.0
    \date       22/05/2008
*/

#include    "NetClient.h"

NetClient::NetClient(unsigned int iCrypt)
{
    conInfo     = NULL;
    crypt       = NET_T_NORMAL;
    TRY

#if USE_SSL == 1
    sslCTX      = NULL;
    if(iCrypt == NET_T_SSL)
    {
        SSL_METHOD  *sslMethode;

        SSL_library_init();
        SSL_load_error_strings();
        sslMethode = SSLv23_server_method();

        sslCTX = SSL_CTX_new(sslMethode);
        if(!sslCTX)
            THROW(ERROR_C_SSLCTXINIT)

        crypt = NET_T_SSL;
    }
#else
    if(iCrypt != NET_T_NORMAL)
        THROW(ERROR_C_NOSSL)
#endif
    CATCH
}

NetClient::~NetClient()
{
    Stop();
}

void    NetClient::Start(const char *pIp, unsigned int iPort)
{
    struct hostent  *host;
    TRY

    if(conInfo)
        THROW(ERROR_C_ALREADY_START)

    conInfo = new NetConnect(crypt);
    conInfo->ip = pIp;
    conInfo->port = iPort;

    if(SocketLoad())
        THROW(ERROR_C_LOAD_DLL)

    conInfo->sock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
    if(conInfo->sock < 0)
        THROW(ERROR_C_INIT_CONNECT)

    host = gethostbyname(conInfo->ip.data());
    if(!host)
        THROW(ERROR_C_URL_NOTFOUND)

    memset(&conInfo->sockName, 0, sizeof(conInfo->sockName));
    conInfo->sockName.sin_port = htons(conInfo->port);
    conInfo->sockName.sin_family = (*host).h_addrtype;
    conInfo->sockName.sin_addr = *((struct in_addr *)(*host).h_addr);

    if(connect(conInfo->sock, (const struct sockaddr *)&conInfo->sockName, sizeof(conInfo->sockName)))
        THROW(ERROR_C_CONNECT)

#if USE_SSL == 1
    if(conInfo->crypt == NET_T_SSL)
    {
        conInfo->sslFD = SSL_new(sslCTX);
        if(!(conInfo->sslFD))
            THROW(ERROR_C_SSLINIT)

        if(SSL_set_fd(conInfo->sslFD, conInfo->sock) < 1)
            THROW(ERROR_C_SSLINIT)

        if(SSL_connect(conInfo->sslFD) < 1)
            THROW(ERROR_C_SSLCONNECT)
    }
#endif

    CATCH
}

void    NetClient::Stop()
{
    if(conInfo)
    {
        SocketClose(conInfo->sock);
        SocketUnLoad();

#if USE_SSL == 1
        if(conInfo->crypt == NET_T_SSL)
        {
            SSL_shutdown(conInfo->sslFD);
            SSL_free(conInfo->sslFD);
        }
#endif
        delete conInfo;
        conInfo = NULL;
    }
}

void    NetClient::Send(const char *pBuffer, ...)
{
    char    temp[NET_BUFFER];
    va_list args;

    va_start(args, pBuffer);
    vsprintf(temp, pBuffer, args);
    va_end(args);

    conInfo->BinSend(temp, strlen(temp));
};

---