WinSock客户端服务端实现--TCP

1.服务端代码

a)初始化winsock库，在Server.cpp中添加

WSADATA wsaData;
WORD sockVersion = MAKEWORD(2, 0);
::WSAStartup(sockVersion, &wsaData);
CserverDlg dlg;
m_pMainWnd = &dlg;
INT_PTR nResponse = dlg.DoModal();
if (nResponse == IDOK)
{
	// TODO: 在此放置处理何时用
	//  “确定”来关闭对话框的代码
}

 b)创建Server端的监听socket

void CserverDlg::OnBnClickedButton1()
{
	m_ip.ResetContent();
	m_list.ResetContent();
	// TODO: 在此添加控件通知处理程序代码
	if(m_socket == INVALID_SOCKET)  // 开启服务
	{
		// 取得端口号
		CString sPort;
		m_port.GetWindowText( sPort );
		int nPort = atoi(sPort);
		if(nPort < 1 || nPort > 65535)
		{
			m_info.AddString( "端口号错误！" );
			return;
		}

		// 创建监听套节字，使它进入监听状态
		if( !CreateAndListen(nPort) )
		{
			m_info.AddString(  "启动服务出错！" );
			return;
		}
	}
	else				// 停止服务
	{
		// 关闭所有连接
		CloseAllSocket();
		m_info.AddString( "所有连接已断开！已停止监听！" );
		// 设置相关子窗口控件状态
		m_start.SetWindowText( "开启服务" );
		m_port.EnableWindow( TRUE );
		m_msg.EnableWindow( FALSE );
		m_send.EnableWindow( FALSE );
	}
}

c）绑定地址端口等信息

BOOL CserverDlg::CreateAndListen(int nPort)
{
	if(m_socket == INVALID_SOCKET)
		::closesocket(m_socket);

	// 创建套节字
	m_socket = ::socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
	if(m_socket == INVALID_SOCKET)
		return FALSE;
	
	// 填写要关联的本地地址
	sockaddr_in sin;
	sin.sin_family = AF_INET;
	sin.sin_port = htons(nPort);
	sin.sin_addr.s_addr = INADDR_ANY;
	// 绑定端口
	if(::bind(m_socket, (sockaddr*)&sin, sizeof(sin)) == SOCKET_ERROR)
	{
		return FALSE;
	}

	// 设置socket为窗口通知消息类型
	::WSAAsyncSelect(m_socket, m_hWnd, WM_SOCKET, FD_ACCEPT|FD_CLOSE);
	// 进入监听模式
	::listen(m_socket, 5);
	m_info.AddString( "已开启监听！监听中..." );
	m_start.SetWindowText( "关闭服务" );
	m_port.EnableWindow( FALSE );
	m_msg.EnableWindow( TRUE );
	m_send.EnableWindow( TRUE );
	return TRUE;
}

d)客户端请求连接时

BOOL CserverDlg::AddClient(SOCKET s)
{
	if( m_client<MAX_SOCKET)
	{
		// 添加新的成员
		m_arClient[m_client++] = s;
		return TRUE;
	}
	return FALSE;
}

e)客户端断开连接时

void CserverDlg::RemoveClient(SOCKET s)
{
	BOOL bFind = FALSE;
	int i=0;
	for( i=0; i<m_client; i++ )
	{
		if( m_arClient[i]==s )
		{
			bFind = TRUE;
			break;
		}
	}

	// 如果找到就将此成员从列表中移除
	if(bFind)
	{
		sockaddr_in sockAddr;
		memset(&sockAddr, 0, sizeof(sockAddr));
		int nSockAddrLen = sizeof(sockAddr);
		::getpeername(m_arClient[i], (SOCKADDR*)&sockAddr, &nSockAddrLen);
		// 转化为主机字节顺序
		int nPeerPort = ::ntohs(sockAddr.sin_port);
		// 转化为字符串IP
		CString sPeerIP = ::inet_ntoa(sockAddr.sin_addr);
		m_info.AddString( "##客户端["+sPeerIP+"]已断开连接!" );
		CString sTmp;
		for( int k=0; k<m_ip.GetCount(); k++ )
		{
			m_list.GetText( k, sTmp );
			if( sTmp==sPeerIP ) m_list.DeleteString( k );
			m_ip.GetLBText( k, sTmp ) ;
			if( sTmp==sPeerIP ) 
			{
				m_ip.DeleteString( k ); 
				break;
			}
		}
		m_client--;
		// 将此成员后面的成员都向前移动一个单位
		for(int j=i; j<m_client; j++)
		{
			m_arClient[j] = m_arClient[j+1];
		}
	}
	if( m_ip.GetCount()==0 ) m_ip.SetWindowText( "" );
}

f)消息响应函数

afx_msg LRESULT CserverDlg::OnSocket(WPARAM wParam, LPARAM lParam)
{
	// 取得有事件发生的套节字句柄
	SOCKET s = wParam;
	// 查看是否出错
	if( WSAGETSELECTERROR(lParam) )
	{
		RemoveClient(s);
		::closesocket(s);
		return 0;
	}
	// 处理发生的事件
	switch( WSAGETSELECTEVENT(lParam) )
	{
		case FD_ACCEPT:		// 监听中的套接字检测到有连接进入
			{
				if( m_client<MAX_SOCKET )
				{
					// 接受连接请求，新的套节字client是新连接的套节字
					SOCKET client = ::accept(s, NULL, NULL);
					// 设置新的套节字为窗口通知消息类型
					int i = ::WSAAsyncSelect(client, 
						m_hWnd, WM_SOCKET, FD_READ|FD_WRITE|FD_CLOSE );
					AddClient(client);
					// 取得对方的IP地址和端口号（使用getpeername函数）
					// Peer对方的地址信息
					sockaddr_in sockAddr;
					memset(&sockAddr, 0, sizeof(sockAddr));
					int nSockAddrLen = sizeof(sockAddr);
					::getpeername(client, (SOCKADDR*)&sockAddr, &nSockAddrLen);
					int nPeerPort = ::ntohs(sockAddr.sin_port);
					// 转化为字符串IP
					CString sPeerIP = ::inet_ntoa(sockAddr.sin_addr);
					CString info;
					info.Format( "检测到已建立新连接，客户端IP:%s", sPeerIP );
					m_list.AddString( sPeerIP );
					m_ip.AddString( sPeerIP );
					m_info.AddString( info );
					if( m_ip.GetCount()==1 ) m_ip.SetCurSel( 0 );
				}
				else
				{
					MessageBox("连接客户太多！");
				}
			}
			break;

		case FD_CLOSE:		// 检测到套接字对应的连接被关闭。
			{
				RemoveClient(s);
				::closesocket(s);
			}
			break;

		case FD_READ:		// 套接字接受到对方发送过来的数据包
			{

				// 取得对方的IP地址和端口号（使用getpeername函数）
				// Peer对方的地址信息
				sockaddr_in sockAddr;
				memset(&sockAddr, 0, sizeof(sockAddr));
				int nSockAddrLen = sizeof(sockAddr);
				::getpeername(s, (SOCKADDR*)&sockAddr, &nSockAddrLen);
				// 转化为主机字节顺序
				int nPeerPort = ::ntohs(sockAddr.sin_port);
				// 转化为字符串IP
				CString sPeerIP = ::inet_ntoa(sockAddr.sin_addr);		 

				// 接受真正的网络数据
				char szText[1024] = { 0 };
				::recv(s, szText, 1024, 0);

				// 显示给用户
				CString strItem = "客户端["+sPeerIP+ "]: " + CString(szText);
				m_info.AddString( strItem );

			}
			break;
		case FD_WRITE:

			break;
	}
	return 0;
}

g)服务器端发送数据

void CserverDlg::OnBnClickedButton3()
{
	// TODO: 在此添加控件通知处理程序代码
	if( m_ip.GetCount()==0 ) return;
	m_CurClient = INVALID_SOCKET;
	sockaddr_in sockAddr;
	memset(&sockAddr, 0, sizeof(sockAddr));
	int nSockAddrLen = sizeof(sockAddr);
	CString sPeerIP ,sCurIP;
	m_ip.GetWindowText( sCurIP );
	for( int i=0; i<m_client; i++ )
	{
		::getpeername( m_arClient[i], (SOCKADDR*)&sockAddr, &nSockAddrLen);
		// 转化为主机字节顺序
		int nPeerPort = ::ntohs(sockAddr.sin_port);
		// 转化为字符串IP
		sPeerIP = ::inet_ntoa(sockAddr.sin_addr);
		if( sPeerIP==sCurIP )
		{
			m_CurClient = m_arClient[i];
			break;
		}
	}
	if( m_CurClient==INVALID_SOCKET ) return;
	// 取得要发送的字符串
	CString sText;
	m_msg.GetWindowText( sText );

	// 添加一个“回车换行”
	// 注意，添加它并不是必须的，但是如果使用本软件作为客户端调试网络协议，
	// 比如SMTP、FTP等，就要添加它了。因为这些协议都要求使用“回车换行”作为一个命令的结束标记
	sText += "\r\n";

	// 发送数据到服务器
	if( ::send( m_CurClient, sText, sText.GetLength(), 0)!= -1 )
	{
		m_info.AddString( "本地发送->"+sPeerIP+":"+sText );
		m_msg.SetWindowText( "" );
	}
}