Category

Development Platform
Home > SourceCode/Document > Game Program > Game Server Simulator > View Code
DgmSocket.cpp
Package: SwordOnline.rar　[view]
Upload User: dzyhzl
Upload Date: 2019-04-29
Package Size: 56270k
Code Size: 4k
Category:

Game Server Simulator

Development Platform:

C/C++

DgmSocket.cpp：Code Content
  1. ////////////////////////////////////////////////////////////////////////////////
  2. //  
  3. //  FileName    :   DgmSocket.h
  4. //  Version     :   1.0
  5. //  Creater     :   Linsuyi
  6. //  Date        :   2002-01-16  20:13:26
  7. //  Comment     :   Tcp/ip datagram/raw base socket source file
  8. //  
  9. ////////////////////////////////////////////////////////////////////////////////
  11. #include "Stdafx.h"
  12. #include "DgmSocket.h"
  15. //////////////////////////////////////////////////////////////////////
  16. // Construction/Destruction
  17. //////////////////////////////////////////////////////////////////////
  19. CDgmSocket::CDgmSocket()
  20. {
  21.     m_nSckType = SOCK_DGRAM;
  22.     
  23.     m_nInPrivate = false;
  24.     ZeroMemory(&m_saRemoteAddr, sizeof(m_saRemoteAddr));
  25. }
  27. CDgmSocket::~CDgmSocket()
  28. {
  29. }
  31. int CDgmSocket::Attach(HSOCKET hInetSocket)
  32. {
  33.     int nResult = false;
  34.     int nRetCode = false;
  35.     
  36.     nRetCode = CIPSocket::Attach(hInetSocket);
  37.     if (!nRetCode)
  38.         goto Exit0;
  39.     
  40.     ZeroMemory(&m_saRemoteAddr, sizeof(m_saRemoteAddr));
  41.     nRetCode = GetRemoteAddress(&m_saRemoteAddr);
  42.     if (0 == nRetCode)
  43.         m_nInPrivate = true;
  44.     
  45.     nResult = true;
  46.     
  47. Exit0:
  48.     return nResult;
  49. }
  51. int CDgmSocket::GetPrivateMode()
  52. {
  53.     return m_nInPrivate;
  54. }
  56. int CDgmSocket::GetPrivateRemote(INET_ADDRESS *psaRemoteAddr)
  57. {
  58.     if (!m_nInPrivate)
  59.         return false;
  60.     
  61.     ASSERT(psaRemoteAddr != NULL);
  62.     if (NULL == psaRemoteAddr)
  63.         return false;
  64.     
  65.     *psaRemoteAddr = m_saRemoteAddr;
  66.     return true;
  67. }
  69. int CDgmSocket::SetRemoteAddress(const char *pcszRemoteAddr, int nPort /*=0*/)
  70. {
  71.     int nResult = false;
  72.     int nRetCode = 0;
  73.     INET_ADDRESS saRemoteAddr = { 0 };
  74.     
  75.     if (NULL == pcszRemoteAddr)
  76.         nResult = SetRemoteAddress(&saRemoteAddr);
  77.     else
  78.     {
  79.         nRetCode = GetSocketAddress(pcszRemoteAddr, &saRemoteAddr);
  80.         ASSERT(nRetCode);
  81.         
  82.         if (nRetCode)
  83.         {
  84.             if (nPort != 0)
  85.                 saRemoteAddr.sin_port = ::htons(nPort);
  86.             
  87.             nResult = SetRemoteAddress(&saRemoteAddr);
  88.         }
  89.     }
  90.     
  91.     return nResult;
  92. }
  94. int CDgmSocket::SetRemoteAddress(const INET_ADDRESS *psaRemoteAddr)
  95. {
  96.     int nResult = false;
  97.     int nRetCode = 0;
  98.     
  99.     ASSERT(psaRemoteAddr != NULL);
  100.     if (NULL == psaRemoteAddr)
  101.         goto Exit0;
  102.     
  103.     ASSERT(IsSocketOK());
  104.     
  105.     if (0 == psaRemoteAddr->sin_addr.s_addr)
  106.     {
  107.         if (!m_nInPrivate)
  108.             nResult = true;
  109.         else
  110.         {
  111.             nRetCode = ::connect(
  112.                 m_hSckHandle,
  113.                 (sockaddr *)psaRemoteAddr,
  114.                 sizeof(INET_ADDRESS)
  115.             );
  116.             nRetCode = CheckError(nRetCode);
  117.             if (0 == nRetCode)
  118.             {
  119.                 m_saRemoteAddr = *psaRemoteAddr;
  120.                 m_nInPrivate = false;
  121.                 
  122.                 nResult = true;
  123.             }
  124.         }
  125.     }
  126.     else
  127.     {
  128.         if (
  129.             (m_saRemoteAddr.sin_family == psaRemoteAddr->sin_family) && 
  130.             (m_saRemoteAddr.sin_addr.s_addr == psaRemoteAddr->sin_addr.s_addr)
  131.         )
  132.             nResult = true;
  133.         else
  134.         {
  135.             nResult = ::connect(
  136.                 m_hSckHandle,
  137.                 (sockaddr *)psaRemoteAddr,
  138.                 sizeof(INET_ADDRESS)
  139.             );
  140.             nRetCode = CheckError(nRetCode);
  141.             if (0 == nRetCode)
  142.             {
  143.                 m_saRemoteAddr = *psaRemoteAddr;
  144.                 m_nInPrivate = true;
  145.             
  146.                 nResult = true;
  147.             }
  148.         }
  149.     }
  150.     
  151. Exit0:
  152.     return nResult;
  153. }
  155. int CDgmSocket::SendTo(int nBufSize, const unsigned char *pbyBuffer)
  156. {
  157.     int nResult = false;
  158.     
  159.     ASSERT(m_nInPrivate);
  160.     
  161.     if (m_nInPrivate)
  162.         nResult = Send(nBufSize, pbyBuffer);
  163.     
  164.     return nResult;
  165. }
  167. int CDgmSocket::ReceiveFrom(int nBufSize, unsigned char *pbyBuffer)
  168. {
  169.     int nResult = false;
  170.     
  171.     ASSERT(m_nInPrivate);
  172.     
  173.     if (m_nInPrivate)
  174.         nResult = Receive(nBufSize, pbyBuffer);
  175.     
  176.     return nResult;
  177. }