WSAAsyncselect 模型中，服务器与客户端消息的收发

WSAAsyncselect 模型中，服务器与客户端消息的收发

新建一个MFC对话框程序

服务器与客户端的对话框基本一致，以服务器为例：



将左右两个编辑框设置为多行显示． 编辑框属性：

CEdit 的属性进行设置：Auto HScroll 设置为 False；

MultiLine 设置为 True；Want Return 设置为 True



分别将其定义一个变量 CString m_strSend;CString m_strRecv;



将服务器与客户端的对话框内重写消息处理函数




LRESULT CWSAAsyncSelectDlgDlg::WindowProc(UINT message, WPARAM wParam, LPARAM lParam)

{

return CDialogEx::WindowProc(message, wParam, lParam);

}

服务器端

定义

#include <WinSock2.h>

#pragma comment(lib, "ws2_32.lib")



#ifdef _DEBUG

#define new DEBUG_NEW

#endif



#define WM_SOCKET (WM_USER+100)// 消息定义

#define BUF_SIZE 1024// 缓冲区大小



SOCKET socketListen;// 监听SOCKET

SOCKET socketAccept;// 连接SOCKET



typedef struct _SOCKET_INFO// SOCKET结构体

{

SOCKET socket;

WSABUF wsaBuf;

char chBuf[BUF_SIZE];

_SOCKET_INFO *pNext;

}SOCKET_INFO, *P_SOCKET_INFO;



P_SOCKET_INFO pInfoTemp;

void CreateSocketInfo(SOCKET socket);

P_SOCKET_INFO GetSocketInfo(SOCKET socket);



char chBuf[BUF_SIZE];







// 消息循环

LRESULT CWSAAsyncSelectServDlg::WindowProc(UINT message, WPARAM wParam, LPARAM lParam)

{

switch(message)

{

case WM_SYSCOMMAND:

if (SC_CLOSE == wParam)

{

closesocket(socketAccept);

closesocket(socketListen);

WSACleanup();

}

break;



case WM_SOCKET:

if (WSAGETSELECTERROR(wParam))// 错误检查

{

AfxMessageBox(_T("ERROR!"));

}

else

{

int nRecv = 0;

switch (WSAGETSELECTEVENT(lParam))// 事件判断

{

case FD_ACCEPT:// 连接通知

socketAccept = accept((SOCKET)socketListen, NULL, NULL);

if (INVALID_SOCKET != socketAccept)

{

CreateSocketInfo(socketAccept);

WSAAsyncSelect(socketAccept, m_hWnd, WM_SOCKET, FD_READ | FD_WRITE | FD_CLOSE);

}

break;



case FD_READ:// 接收通知

P_SOCKET_INFO pInfo;

pInfo = GetSocketInfo(wParam);

nRecv = recv(pInfo->socket, chBuf, BUF_SIZE, 0);

if (nRecv>0)

{

m_strRecv += chBuf;

m_strRecv += _T("\r\n");

UpdateData(FALSE);

}

break;



case FD_WRITE:// 发送通知

break;



case FD_CLOSE:

pInfo = GetSocketInfo(wParam);

{

CString strTemp;

strTemp.Format(_T("客户端 %d 已关闭"), pInfo->socket);

AfxMessageBox(strTemp);

}

break;

default:

break;

}

}

}

return CDialogEx::WindowProc(message, wParam, lParam);

}



// 消息发送按钮

void CWSAAsyncSelectServDlg::OnBnClickedBtnRecvSend()

{

UpdateData(true);



//CString 转 char[]

CStringA strSend;

strSend = m_strSend;

strncpy_s(chBuf, strSend, strSend.GetLength()+1);



int err = send(socketAccept, chBuf, strlen(chBuf), 0);

if (SOCKET_ERROR == err)

{

MessageBox(_T("send error!"));

}

m_strSend = _T("");// 编辑框清空

UpdateData(false);

}



// 获取SOCKET信息

void CreateSocketInfo(SOCKET socket)

{

P_SOCKET_INFO pInfo;

if (NULL == (pInfo = (P_SOCKET_INFO)GlobalAlloc(GPTR, sizeof(SOCKET_INFO) )))

{

return;

}

pInfo->socket = socket;

pInfo->pNext = pInfoTemp;



pInfoTemp = pInfo; // 把pInfo放到pInfoTemp的第1位

int m = 8;

}





P_SOCKET_INFO GetSocketInfo(SOCKET socket)

{

P_SOCKET_INFO pInfo = pInfoTemp;

while (pInfo)

{

if (socket == pInfo->socket)

{

return pInfo;

}

pInfo = pInfo->pNext;

}

return NULL;

}



客户端



定义：

#include <WinSock2.h>

#pragma comment(lib, "ws2_32.lib")

#define WM_SOCKET (WM_USER+101)

SOCKET socketAccept;

#define BUF_SIZE 1024

char chBuf[BUF_SIZE];







在OnInitDialog()中，将客户端与服务器进行连接



BOOL CWSAAsyncSelectClientDlg::OnInitDialog()

{

int nErrorCode = 0;

WSADATA wsaData;

nErrorCode = WSAStartup(0x0202, &wsaData);



socketAccept = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);



if (INVALID_SOCKET == socketAccept)

{

return FALSE;

}



//nErrorCode = WSAAsyncSelect(socketAccept, m_hWnd, WM_SOCKET, FD_READ|FD_WRITE|FD_CLOSE|FD_CONNECT);

nErrorCode = WSAAsyncSelect(socketAccept, m_hWnd, WM_SOCKET, FD_READ|FD_CLOSE);



// 连接

#if 1

sockaddr_in addrAccept;

addrAccept.sin_family = AF_INET;

addrAccept.sin_port = htons(8088);

addrAccept.sin_addr.S_un.S_addr = inet_addr("127.0.0.1");



while (1)
// 因为WSAAsyncSelect 将 socketAccept 置为非阻塞模式

{

nErrorCode = connect(socketAccept, (sockaddr*)&addrAccept, sizeof(sockaddr));

if (SOCKET_ERROR == nErrorCode)

{

nErrorCode = WSAGetLastError();

if (WSAEWOULDBLOCK == nErrorCode || WSAEINVAL == nErrorCode)

{

Sleep(300);

continue;

}

else if (WSAEISCONN == nErrorCode)// 连接成功

{

break;

}

else

{

AfxMessageBox(_T("连接失败"));

return -1;

}

}

}

if (SOCKET_ERROR == nErrorCode)

{

AfxMessageBox(_T("连接失败"));

return FALSE;

}

#endif



CDialogEx::OnInitDialog();



SetIcon(m_hIcon, TRUE);

SetIcon(m_hIcon, FALSE);

return TRUE;

}







// 消息发送按钮

void CWSAAsyncSelectClientDlg::OnBnClickedBtnRecvSend()

{

UpdateData(TRUE);

CStringA strInput;

strInput = m_strSend;



memset(chBuf, ' ', sizeof(char)*1024);

strncpy_s(chBuf, strInput, strInput.GetLength()+1);



int nErrorCode = 0;

nErrorCode = send(socketAccept, chBuf, 1024, 0);

if (SOCKET_ERROR == nErrorCode)

{

nErrorCode = WSAGetLastError();

return;

}

m_strSend = _T("");// 置空

UpdateData(FALSE);

GetDlgItem(IDC_EDT_SEND)->SetFocus();

}











// 消息循环

LRESULT CWSAAsyncSelectClientDlg::WindowProc(UINT message, WPARAM wParam, LPARAM lParam)

{

if (WM_SOCKET == message)

{

if (WSAGETSELECTERROR(wParam))

{

AfxMessageBox(_T("ERROR"));

}

else

{

CString strMsg;



switch (WSAGETSELECTEVENT(lParam))

{

case FD_READ:// 接收通知

recv((SOCKET)wParam, chBuf, BUF_SIZE, 0);

m_strRecv += chBuf;

m_strRecv += _T("\r\n");

UpdateData(FALSE);

break;

case FD_CLOSE:// 关闭通知

strMsg.Format(_T("服务器已关闭 %d"), socketAccept);

AfxMessageBox(strMsg);

break;

case FD_ACCEPT:

case FD_CONNECT:

case FD_WRITE:

break;

}

}

}



return CDialogEx::WindowProc(message, wParam, lParam);

}







源码下载
http://download.csdn.net/detail/yulinxx/7426575