TnbSocketXp.h

[詳解]

#pragma once
#ifndef _TnbWINSOCK2_ENABLE
 #error
#endif
 
#include "TnbSocket.h"
 
 
 
//TNB Library
namespace TNB
{
 
 
 
class CSocketXp
{
public:
 
    CSocketXp(void) : m_isInit(true), m_lpfnTransmitPackets(NULL)
    {
    }
 
    bool Send(SOCKET so, size_t dataSize, LPCVOID lpcvData)
    {
        if ( m_Init(so) )
        {
            TRANSMIT_PACKETS_ELEMENT packets;
            packets.dwElFlags = TP_ELEMENT_MEMORY;
            packets.cLength = dataSize;
            packets.pBuffer = const_cast<void* const>(lpcvData);
            return !! m_lpfnTransmitPackets(so, &packets, 1, 0, NULL, 0);
        }
        int r =  ::send(so, static_cast<const char*>(lpcvData), dataSize, 0);
        TRACE2("::send(%u) result = %d\n", dataSize, r);
        return r == ToInt(dataSize);
    }
 
private:
    bool                    m_isInit;               
    LPFN_TRANSMITPACKETS    m_lpfnTransmitPackets;  
    bool m_Init(SOCKET so)
    {
        if ( ! m_isInit )
        {
            m_isInit = true;
            DWORD dwBytes;
            GUID guidTransmitPackets = WSAID_TRANSMITPACKETS;
            ::WSAIoctl(so, SIO_GET_EXTENSION_FUNCTION_POINTER, &guidTransmitPackets, sizeof(GUID), &m_lpfnTransmitPackets, sizeof(LPVOID), &dwBytes, NULL, NULL);
        }
        return m_lpfnTransmitPackets != NULL;
    }
};
 
 
 
class CSocketClientXp : public CSocketClient
{
    DEFSUPER(CSocketClient);
protected:
    
    virtual bool m_Send(SOCKET so, size_t dataSize, LPCVOID lpcvData)
    {
        _super::m_OnSend(dataSize, lpcvData);
        if ( so == INVALID_SOCKET )
        {
            return false;
        }
        return m_xp.Send(so, dataSize, lpcvData);
    }
private:
    CSocketXp   m_xp;
};
 
 
 
class CSocketServerXp : public CSocketServer
{
    DEFSUPER(CSocketServer);
protected:
    
    virtual bool m_Send(SOCKET so, size_t dataSize, LPCVOID lpcvData)
    {
        _super::m_OnSend(dataSize, lpcvData);
        if ( so == INVALID_SOCKET )
        {
            return false;
        }
        return m_xp.Send(so, dataSize, lpcvData);
    }
private:
    CSocketXp   m_xp;
};
 
 
 
#ifndef _TnbDOXYGEN //Document作成用シンボル
 #define CSocketClient CSocketClientXp
 #define CSocketServer CSocketServerXp
#endif
 
 
 
};