D:/Zythum/DinoKod/Network/Socket.h

//---------------------------------------------------------------------------------------------
//      This file is a part of "DinoKod".
//      Copyright © 2003 Dino Productions. All Rights Reserved.
//      
//      File                    : Socket.cpp
//      Author                  : Sebastien LEIX        sebastien.leix@wanadoo.fr
//      Date                    : 09/09/2002
//      Modification    :
//
//---------------------------------------------------------------------------------------------
#ifndef __SOCKET_H__
#define __SOCKET_H__

#include "Network/NetworkDll.h"
#include <windows.h>
#include "Common/Types.h"
#include "Common/Str.h"
#include "Network/SocketDefs.h"

class KInetAddr;

//---------------------------------------------------------------------------------------------------------------------
class NETWORK_API KSocket
{
protected:
        SOCKET                  m_hSocket;
        bool                    m_bConnected;
        bool                    m_bCompress;
        bool                    m_bAttached;
                                        
        u32                             m_nBytesSent;                   // Données compressées
        u32                             m_nBytesReceived;               // Données compressées
        u32                             m_nRealBytesSent;               // Données non compressées
        u32                             m_nRealBytesReceived;   // Données non compressées

public:                         
                                        KSocket();
        virtual                 ~KSocket();
        
        bool                    Attach( SOCKET hSocket );
        bool                    Detach();
        
        bool                    Create( KNSOCKETTYPE Type );
        bool                    Close();
        bool                    Bind( KInetAddr& Addr );
        bool                    SelectRead();
        bool                    SelectWrite();
        bool                    SelectExcept();
        bool                    Connect( KInetAddr& Addr );
        bool                    Disconnect();
        bool                    Listen();
        bool                    Accept( KSocket& AcceptedSocket, KInetAddr& Addr );
        bool                    Send( void* pBuffer, s32 Size );
        bool                    Receive( void* pBuffer, s32 Size, u32* pBytesRead );
        bool                    SendTo( KInetAddr& Addr, void* pBuffer, u32 Size );
        bool                    ReceiveFrom( KInetAddr& Addr, void* pBuffer, u32 Size, u32* pBytesRead );
        bool                    AsyncSelect();

        bool                    IsConnected()                                   { return m_bConnected;                  }
        void                    SetConnected( bool bConnected ) { m_bConnected = bConnected;    }

        void                    EnableBroadcast( bool bEnable );

        virtual void    OnAccept();
        virtual void    OnRead();
        virtual void    OnWrite();
        virtual void    OnClose();
        virtual void    OnOOB();
                                        
        static bool             StartNetwork();
        static bool             StopNetwork();

        static bool                             GetHostName( char* pBuffer, s32 Size );
        static hostent*                 GetHostByName( char* pName );
        static hostent*                 GetHostByAddr( KInetAddr& Addr );
        static bool                             GetAddrByName( KStr& sHostName, KInetAddr& Addr );

        SOCKET                                  GethSocket()                                    { return m_hSocket;                                                             }

        // Stats
        u32                                             GetnBytesSent()                                 { return m_nBytesSent;                                                  }
        u32                                             GetnBytesReceived()                             { return m_nBytesReceived;                                              }
        u32                                             GetnRealBytesSent()                             { return m_nRealBytesSent;                                              }
        u32                                             GetnRealBytesReceived()                 { return m_nRealBytesReceived;                                  }
        void                                    ResetStats()                                    { m_nBytesSent = m_nBytesReceived = m_nRealBytesSent = m_nRealBytesReceived = 0;}
};

#endif  __SOCKET_H__

---