WSAStartupSingleton.cpp

Go to the documentation of this file.

#include "WSAStartupSingleton.h"

#if defined(_XBOX) || defined(X360)
                                                                                             
#elif defined(_WIN32)
#include <winsock2.h>
#include <ws2tcpip.h>
#endif
#include "RakNetDefines.h"
#include <stdio.h>

int WSAStartupSingleton::refCount=0;

WSAStartupSingleton::WSAStartupSingleton() {}
WSAStartupSingleton::~WSAStartupSingleton() {}
void WSAStartupSingleton::AddRef(void)
{
#ifdef _WIN32

        refCount++;
        
        if (refCount!=1)
                return;

#if defined(_XBOX) || defined(X360)
               
#endif

        WSADATA winsockInfo;
        if ( WSAStartup( MAKEWORD( 2, 2 ), &winsockInfo ) != 0 )
        {
#if !defined(_XBOX) && !defined(X360) && defined(_DEBUG)
                DWORD dwIOError = GetLastError();
                LPVOID messageBuffer;
                FormatMessage( FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS,
                        NULL, dwIOError, MAKELANGID( LANG_NEUTRAL, SUBLANG_DEFAULT ),  // Default language
                        ( LPTSTR ) & messageBuffer, 0, NULL );
                // something has gone wrong here...
                RAKNET_DEBUG_PRINTF( "WSAStartup failed:Error code - %d\n%s", dwIOError, messageBuffer );
                //Free the buffer.
                LocalFree( messageBuffer );
#endif
        }

#endif
}
void WSAStartupSingleton::Deref(void)
{
#ifdef _WIN32
        if (refCount==0)
                return;
                
        if (refCount>1)
        {
                refCount--;
                return;
        }
        
        WSACleanup();

#if defined(_XBOX) || defined(X360)
                
#endif

        
        refCount=0;
#endif
}