view/SocketX/PeerSocket.cpp

#pragma once

#include "stdafx.h"
#include "SocketX.h"

//////////////////////////////////////////////////////////////////////////

IMPLEMENT_DYNAMIC(CPeerSocket, IBaseSocket)

CPeerSocket::CPeerSocket()
{
}

CPeerSocket::~CPeerSocket()
{
}

BOOL CPeerSocket::CreateSocket(SOCKMSG SockMsg)
{
        return doCreateSocket(SockMsg, SOCK_DGRAM, IPPROTO_UDP);
}

void CPeerSocket::OnClose(SOCKET hSocket, int nErrorCode)
{
        NETADDR NetAddr;

        if (nErrorCode != 0)
        {
                TRACE1("***** ERROR: OnClose(%d) *****\n", nErrorCode);
                return;
        }

        if (CloseSocket())
        {
                ConvertSockAddrToNetAddr(&m_PeerAddr, NetAddr);
                m_SockMsg.pCallback->OnSocketCloseMessage(m_SockMsg.dwSocketID, NetAddr);
        }       
}

void CPeerSocket::OnSend(SOCKET hSocket, int nErrorCode)
{
        if (nErrorCode != 0)
        {
                TRACE1("***** ERROR: OnSend(%d) *****\n", nErrorCode);
                return;
        }

        m_SockMsg.pCallback->OnSocketSendMessage(m_SockMsg.dwSocketID);
}

void CPeerSocket::OnReceive(SOCKET hSocket, int nErrorCode)
{
        CByteArray data;
        SOCKADDR   sAddr;
        NETADDR    NetAddr;
        int nLenSA = sizeof(SOCKADDR);

        if (nErrorCode != 0)
        {
                TRACE1("***** ERROR: OnReceive(%d) *****\n", nErrorCode);
                return;
        }

        if (!doReceiveFrom(hSocket, data, &sAddr, &nLenSA))
        {
                return;
        }

        ConvertSockAddrToNetAddr(&sAddr, NetAddr);
        m_SockMsg.pCallback->OnSocketReceiveFromMessage(m_SockMsg.dwSocketID, NetAddr, data);
}

BOOL CPeerSocket::Initialize(SOCKMSG SockMsg, DWORD dwLocalAddress, WORD wPort)
{
        SOCKADDR_IN sAddrIn;

        if (!CreateSocket(SockMsg))
        {
                return FALSE;
        }

        if (!Bind(wPort, dwLocalAddress))
        {
                CloseSocket();
                return FALSE;
        }

        if (!(SetSendBufferSize(SX_UDP_BUFSIZE) && SetReceiveBufferSize(SX_UDP_BUFSIZE)))
        {
                CloseSocket();
                return FALSE;
        }

        sAddrIn.sin_family         = AF_INET;
        sAddrIn.sin_port                   = htons(wPort);
        sAddrIn.sin_addr.s_addr = dwLocalAddress;
        CopyMemory(&m_LocalAddr, &sAddrIn, sizeof(m_LocalAddr));

        m_fInit = TRUE;

        return TRUE;
}

//////////////////////////////////////////////////////////////////////////