SocketLayer.h

Go to the documentation of this file.

#ifndef __SOCKET_LAYER_H
#define __SOCKET_LAYER_H

#include "RakMemoryOverride.h"
#include "SocketIncludes.h"
#include "RakNetTypes.h"
#include "RakNetSmartPtr.h"
#include "RakNetSocket.h"
#include "Export.h"
#include "MTUSize.h"
#include "RakString.h"

//#include "ClientContextStruct.h"

class RakPeer;

class RAK_DLL_EXPORT SocketLayerOverride
{
public:
        SocketLayerOverride() {}
        virtual ~SocketLayerOverride() {}

        virtual int RakNetSendTo( SOCKET s, const char *data, int length, SystemAddress systemAddress )=0;

        // Return -1 to use RakNet's normal recvfrom, 0 to abort RakNet's normal recvfrom, and positive to return data
        virtual int RakNetRecvFrom( const SOCKET sIn, RakPeer *rakPeerIn, char dataOut[ MAXIMUM_MTU_SIZE ], SystemAddress *senderOut, bool calledFromMainThread )=0;
};


// A platform independent implementation of Berkeley sockets, with settings used by RakNet
class SocketLayer
{

public:
        
        SocketLayer();
        
        // Destructor   
        ~SocketLayer();
        
        // Get the singleton instance of the Socket Layer.
        static inline SocketLayer* Instance()
        {
                return & I;
        }

        // Connect a socket to a remote host.
        SOCKET Connect( SOCKET writeSocket, unsigned int binaryAddress, unsigned short port );
        
        SOCKET CreateBoundSocket( unsigned short port, bool blockingSocket, const char *forceHostAddress, unsigned int sleepOn10048 );
        SOCKET CreateBoundSocket_PS3Lobby( unsigned short port, bool blockingSocket, const char *forceHostAddress );

        static bool IsPortInUse(unsigned short port, const char *hostAddress=0);

        const char* DomainNameToIP( const char *domainName );
        
        bool AssociateSocketWithCompletionPortAndRead( SOCKET readSocket, unsigned int binaryAddress, unsigned short port, RakPeer* rakPeer );
        
        void Write( const SOCKET writeSocket, const char* data, const int length );
        
        int RecvFrom( const SOCKET s, RakPeer *rakPeer, int *errorCode, RakNetSmartPtr<RakNetSocket> rakNetSocket, unsigned short remotePortRakNetWasStartedOn_PS3 );
        // Newer version, for reads from a thread
        static void RecvFromBlocking( const SOCKET s, RakPeer *rakPeer, unsigned short remotePortRakNetWasStartedOn_PS3, char *dataOut, int *bytesReadOut, SystemAddress *systemAddressOut, RakNetTimeUS *timeRead );
        
        void RawRecvFromNonBlocking( const SOCKET s, unsigned short remotePortRakNetWasStartedOn_PS3, char *dataOut, int *bytesReadOut, SystemAddress *systemAddressOut, RakNetTimeUS *timeRead );


        RakNet::RakString GetSubNetForSocketAndIp(SOCKET inSock, RakNet::RakString inIpString);


        void SetNonBlocking( SOCKET listenSocket);

#if !defined(_XBOX) && !defined(_X360)



        void GetMyIP( char ipList[ MAXIMUM_NUMBER_OF_INTERNAL_IDS ][ 16 ], unsigned int binaryAddresses[MAXIMUM_NUMBER_OF_INTERNAL_IDS] );
#endif
        
        int SendTo( SOCKET s, const char *data, int length, const char ip[ 16 ], unsigned short port, unsigned short remotePortRakNetWasStartedOn_PS3 );

        int SendToTTL( SOCKET s, const char *data, int length, const char ip[ 16 ], unsigned short port, int ttl );

        int SendTo( SOCKET s, const char *data, int length, unsigned int binaryAddress, unsigned short port, unsigned short remotePortRakNetWasStartedOn_PS3 );

        static unsigned short GetLocalPort ( SOCKET s );

        static SystemAddress GetSystemAddress ( SOCKET s );

        void SetSocketLayerOverride(SocketLayerOverride *_slo);
        SocketLayerOverride* GetSocketLayerOverride(void) const {return slo;}

        int SendTo_PS3Lobby( SOCKET s, const char *data, int length, unsigned int binaryAddress, unsigned short port, unsigned short remotePortRakNetWasStartedOn_PS3 );
        int SendTo_360( SOCKET s, const char *data, int length, const char *voiceData, int voiceLength, unsigned int binaryAddress, unsigned short port );
        int SendTo_PC( SOCKET s, const char *data, int length, unsigned int binaryAddress, unsigned short port );


        static void SetDoNotFragment( SOCKET listenSocket, int opt );
private:


        static SocketLayer I;
        void SetSocketOptions( SOCKET listenSocket);
        SocketLayerOverride *slo;
};

#endif