c++ TCP 打洞源码

CSockClient* GetNewSockClient ()
{
CSockClient *pSockClient = new CSockClient;
if ( !pSockClient )
{
printf ( "New SockClient object failed\n" );
return NULL;
}
g_CSFor_PtrAry_SockClient.Lock();
g_PtrAry_SockClient.Add ( pSockClient );
pSockClient->m_dwID = g_nSockClientID;
g_CSFor_PtrAry_SockClient.Unlock();
printf ( "Current SocketClient array count is %d\n", g_PtrAry_SockClient.GetSize() );

return pSockClient;
}

//
// 将新客户端登录信息发送给所有已登录的客户端，但不发送给自己
//
BOOL SendNewUserLoginNotifyToAll ( LPCTSTR lpszClientIP, UINT nClientPort, DWORD dwID )
{
ASSERT ( lpszClientIP && nClientPort > 0 );
g_CSFor_PtrAry_SockClient.Lock();
for ( int i=0; i<g_PtrAry_SockClient.GetSize(); i )
{
CSockClient *pSockClient = (CSockClient*)g_PtrAry_SockClient.GetAt(i);
if ( pSockClient && pSockClient->m_bMainConn && pSockClient->m_dwID > 0 && pSockClient->m_dwID != dwID )
{
if ( !pSockClient->SendNewUserLoginNotify ( lpszClientIP, nClientPort, dwID ) )
{
g_CSFor_PtrAry_SockClient.Unlock();
return FALSE;
}
}
}

g_CSFor_PtrAry_SockClient.Unlock ();
return TRUE;
}

CSockClient* FindSocketClient ( DWORD dwID )
{
g_CSFor_PtrAry_SockClient.Lock ();
for ( int i=0; i<g_PtrAry_SockClient.GetSize(); i )
{
CSockClient *pSockClient = (CSockClient*)g_PtrAry_SockClient.GetAt(i);
if ( pSockClient && pSockClient->m_dwID == dwID )
{
g_CSFor_PtrAry_SockClient.Unlock ();
return pSockClient;
}
}
printf ( "Can't find ID:%u\n", dwID );
g_CSFor_PtrAry_SockClient.Unlock ();
return NULL;
}