TCP服务器接收多个客户端连接

LRESULT CNetDogSDlg::OnXAcceptSocket(WPARAM wParam,LPARAM lParam)
{
int acc_sin_len; // Accept socket address length
SOCKADDR_IN acc_sin; // Accept socket address - internet style
SOCKET ConnectSocket;
SOCKET ListenSocket;
TEMPSOCKET ts;

UpdateComStatus();

ListenSocket = m_data.GetListenSocket();

if( WSAGETSELECTERROR( lParam ) != 0 ) {
ShowNetMonitorMsg( _T("在侦听 Socket 上发生错误.") );
// Cancel any further notifications.
//WSAAsyncSelect( ListenSocket,GetSafeHwnd(),0,0 );
//closesocket( ListenSocket );
//m_data.SetListenSocket( INVALID_SOCKET );
//m_data.SetListenSocketStatus( FALSE,m_txtPortStatus,m_btnListen );
return 0;
}

switch( WSAGETSELECTEVENT( lParam ) ) {
case FD_ACCEPT:
ShowNetMonitorMsg( _T("有客户申请连接...") );

// Accept the incoming connection.
acc_sin_len = sizeof( acc_sin );
ConnectSocket = accept( ListenSocket,( struct sockaddr FAR * )&acc_sin,( int FAR * )&acc_sin_len );
if( ConnectSocket == INVALID_SOCKET ) {
ShowNetMonitorMsg( _T("创建客户连接 Socket 失败.") );
return 0;
}

// Send main window a WSA_READ when either data is pending on
// the socket (FD_READ) or the connection is closed (FD_CLOSE)
if( WSAAsyncSelect( ConnectSocket,GetSafeHwnd(),UM_CONNECT,FD_READ | FD_WRITE | FD_CLOSE ) == SOCKET_ERROR ) {
ShowNetMonitorMsg( _T("建立客户连接 Socket 异步通信消息失败.") );
closesocket( ConnectSocket );
return 0;
}

// Get User information.
// --------------------------------------------------
{
CString str,strConnectPort,strHostIP,strHostName;
str.Format( _T("客户连接成功 - 连接到: Socket( %d ) "),ConnectSocket );
strConnectPort.Format( _T("%d"),acc_sin.sin_port );
strHostIP.Format( _T("%d.%d.%d.%d"),
acc_sin.sin_addr.S_un.S_un_b.s_b1,
acc_sin.sin_addr.S_un.S_un_b.s_b2,
acc_sin.sin_addr.S_un.S_un_b.s_b3,
acc_sin.sin_addr.S_un.S_un_b.s_b4
);
HOSTENT* pHostent = gethostbyaddr( (const char *)&(acc_sin.sin_addr),sizeof(struct in_addr),AF_INET );
if( !pHostent ) {
strHostName = _T("未知");
}
else {
strHostName = pHostent->h_name;
}
str += _T("端口号: ");
str += strConnectPort;
str += _T(" 客户IP: ");
str += strHostIP;
str += _T(" 客户机器名: ");
str += strHostName;
ShowNetMonitorMsg( str );

ts.Port = acc_sin.sin_port;
ts.skt = ConnectSocket;
lstrcpy( ts.UserName,strHostName );
lstrcpy( ts.IPAddress,strHostIP );
}
if( !m_data.AddTempSocket( &ts ) ) {
closesocket( ConnectSocket );
ShowNetMonitorMsg( _T("存贮客户连接 Socket 空间不足.") );
}
break;
case FD_CLOSE:
ShowNetMonitorMsg( _T("侦听 Socket 被关闭.") );
closesocket( ListenSocket );
m_data.SetListenSocket( INVALID_SOCKET );
m_data.SetListenSocketStatus( FALSE,m_txtPortStatus,m_btnListen );
break;
case FD_READ:
ShowNetMonitorMsg( _T("<接收数据>消息 到达侦听 Socket.") );
break;
case FD_WRITE:
ShowNetMonitorMsg( _T("<发送数据>消息 到达侦听 Socket.") );
break;
default:
ShowNetMonitorMsg( _T("<未知>消息 到达侦听 Socket.") );
break;
}

return 0;
}

LRESULT CNetDogSDlg::OnXConnectSocket(WPARAM wParam,LPARAM lParam)
{
SOCKET ConnectSocket;
CString str;
int nRet;
char Buf[MAX_SOCKET_BUF];

UpdateComStatus();

ConnectSocket = ( SOCKET )wParam;
if( WSAGETSELECTERROR( lParam ) != 0 ) {
str.Format(_T("在连接 Socket( %d ) 上发生错误,断开客户连接."),ConnectSocket );
ShowNetMonitorMsg( str );
closesocket( ConnectSocket );
m_data.DeleteConnectSocket( ConnectSocket );
m_data.UpdateProductInfo( m_lstProductInfo,FALSE );
m_data.ShowUserInfo( m_lstUserInfo );
return 0;
}

switch( WSAGETSELECTEVENT( lParam ) ) {
case FD_CLOSE:
str.Format( _T("连接在 Socket( %d )上的用户断开连接."),ConnectSocket );
ShowNetMonitorMsg( str );
closesocket( ConnectSocket );
m_data.DeleteConnectSocket( ConnectSocket );
m_data.UpdateProductInfo( m_lstProductInfo,FALSE );
m_data.ShowUserInfo( m_lstUserInfo );
break;
case FD_READ:
str.Format(_T("<接收数据>消息 到达连接 Socket( %d )."),ConnectSocket );
ShowNetMonitorMsg( str );
ZeroMemory( Buf,MAX_SOCKET_BUF );
nRet = recv( ConnectSocket,Buf,MAX_SOCKET_BUF,0 );
switch( nRet ) {
case SOCKET_ERROR:
str.Format( _T("从 连接 Socket( %d ) 读取数据时发生错误."),ConnectSocket );
ShowNetMonitorMsg( str );
break;
case 0:
str.Format( _T("连接 Socket( %d ) 被关闭."),ConnectSocket );
ShowNetMonitorMsg( str );
break;
default:
str.Format( _T("从连接 Socket( %d ) 读取了 %d 字节数据."),ConnectSocket,nRet );
if( !m_data.AddXData( ConnectSocket,Buf,nRet ) ) {
str += _T("/r/n---------> 数据缓冲区已满,数据被丢弃.");
}
else {
str += _T("/r/n---------> 数据成功保存到缓冲区,等待处理.");
}
ShowNetMonitorMsg( str );
m_data.ParseData( this,m_lstProductInfo,m_lstUserInfo );
break;
}
break;
case FD_WRITE:
str.Format(_T("<发送数据>消息 到达连接 Socket( %d )."),ConnectSocket );
ShowNetMonitorMsg( str );
break;
default:
str.Format(_T("<未知>消息 到达连接 Socket( %d )."),ConnectSocket );
ShowNetMonitorMsg( str );
break;
}

return 0;
}