MFC 基于消息机机制的WSAAsyncSelect demo

1. 添加头文件

#include <winsock.h>

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

服务端：

1. 定义变量

SOCKET sSocket;

2. 定义消息常量

#define MSG_SOCKET (WM_USER + 1000)

添加消息映射

BEGIN_MESSAGE_MAP(CXXXDlg, CDialog)

//{{AFX_MSG_MAP(CXXXDlg)

//}}AFX_MSG_MAP

ON_MESSAGE(MSG_SOCKET, OnSockEvent)

END_MESSAGE_MAP()

3. 初始化SOCKET

bool CXXXDlg::InitWSASocket()

{

char szError[256] = {0};

WORD wVersionRequested;

WSADATA wsaData;

wVersionRequested = MAKEWORD(2, 2);

int ReturnInfo = WSAStartup(wVersionRequested, &wsaData);

if (ReturnInfo != 0)

{

return false;

}

if (LOBYTE(wsaData.wVersion) != 2 || HIBYTE(wsaData.wVersion) != 2)

{

WSACleanup();

sprintf(szError, "初始化SOCKET错误，%d", WSAGetLastError());

AfxMessageBox(szError);

return false;

}

return true;

}

3. 初始化的时候调用InitWSASocket，如果初始化成功则开始监听。

BOOL CXXXDlg::OnInitDialog()

{

CDialog::OnInitDialog();

// Set the icon for this dialog. The framework does this automatically

// when the application's main window is not a dialog

SetIcon(m_hIcon, TRUE); // Set big icon

SetIcon(m_hIcon, FALSE); // Set small icon

// TODO: Add extra initialization here

GetDlgItem(IDC_EDIT_PORT)->SetWindowText("8977");

if (!InitWSASocket())

{

return FALSE;

}

if (!StartSocketListen())

{

OnCancel();

}

return TRUE; // return TRUE unless you set the focus to a control

}

4. 监听模块，注册事件

bool CXXXDlg::StartSocketListen()

{

GetDlgItemText(IDC_EDIT_PORT, m_StrPort);

if (m_StrPort.IsEmpty())

{

return false;

}

char szError[256] = {0};

//创建套接字

m_sSocket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);

if(INVALID_SOCKET == m_sSocket)

{

sprintf(szError, "创建SOCKET错误，%d", WSAGetLastError());

AfxMessageBox(szError);

return false;

}

//服务器地址信息

SOCKADDR_IN addrSock;

memset(&addrSock, 0x0, sizeof addrSock);

addrSock.sin_addr.S_un.S_addr = inet_addr("127.0.0.1")/*htonl(INADDR_ANY)*/;

addrSock.sin_family = AF_INET;

addrSock.sin_port = htons(atoi(m_StrPort));

//成功返回0 ，否则返回SOCKET_ERROR

if (SOCKET_ERROR == ::bind(m_sSocket, (SOCKADDR*)&addrSock, sizeof(SOCKADDR)))

{

sprintf(szError, "绑定SOCKET错误，%d", WSAGetLastError());

AfxMessageBox(szError);

return false;

}

//注册网络事件

if (SOCKET_ERROR == WSAAsyncSelect(m_sSocket, m_hWnd,
MSG_SOCKET, FD_ACCEPT | FD_READ | FD_CLOSE))

{

::closesocket(m_sSocket);

sprintf(szError, "注册网络读取事件失败，%d", WSAGetLastError());

AfxMessageBox(szError);

return false;

}

//成功返回0 ，否则返回SOCKET_ERROR

if (SOCKET_ERROR == ::listen(m_sSocket, 100))

{

sprintf(szError, "监听SOCKET错误，%d", WSAGetLastError());

AfxMessageBox(szError);

return false;

}

return true;

}

5. 当有客户端来连接的时候，系统发送消息触发以下事件。

说明：当客户端connect的时候触发服务端的FD_ACCEPT。客户端send数据的时候触发服务端的FD_READ

，客户端关闭的时候触发服务端的FD_CLOSE。

void CXXXDlg::OnSockEvent(WPARAM wParam,LPARAM lParam)

{

SOCKET sSock= static_cast<SOCKET>(wParam);

if(WSAGETSELECTERROR(lParam))

{

AfxMessageBox("OnSockEvent()接收到错误消息");

::closesocket(wParam);

return;

}

switch(WSAGETSELECTEVENT(lParam))

{

case FD_ACCEPT:

{

SOCKADDR_IN addrSock;

memset(&addrSock, 0x0, sizeof addrSock);

int Len = sizeof(addrSock);

//当客户端connect的时候全局的m_sSocket和wParam传进来的SOCKET是相同的，

//当服务端调用accept去获取客户端的SOCKET传入下一次的OnSockEvent中wParam参数中

m_cSocket = ::accept(m_sSocket, (SOCKADDR*)&addrSock, &Len);

if (INVALID_SOCKET == m_cSocket)

{

char szError[256] = {0};

sprintf(szError, "接收SOCKET错误，%d", WSAGetLastError());

AfxMessageBox(szError);

}

}

break;

case FD_READ:

{

char buff[1024] = {0};

SOCKADDR_IN addrSock;

memset(&addrSock, 0x0, sizeof addrSock);

int Len = sizeof(addrSock);

//recv不等待是因为你使用的是非阻塞socket，换而你使用阻塞socket一样需要等待。

//recv的recvfrom是可以替换使用的，只是recvfrom多了两个参数，可以用来接收对端的地址信息，

//这个对于udp这种无连接的，可以很方便地进行回复。

//而换过来如果你在udp当中也使用recv，那么就不知道该回复给谁了，如果你不需要回复的话，也是可以使用的。

//另外就是对于tcp是已经知道对端的，就没必要每次接收还多收一个地址，

//没有意义，要取地址信息，在accept当中取得就可以加以记录了。

::recv(sSock, buff, 1024, 0);

// ::recvfrom(m_cSocket, buff, 1024, 0, (SOCKADDR*)&addrSock, &Len);

static_cast<CListBox *>(GetDlgItem(IDC_LIST_LOG))->AddString(buff);

::send(sSock/*m_cSocket*/, buff, 1024, 0);

}

break;

case FD_CLOSE:

{

::shutdown(sSock, 2);

::closesocket(sSock);

}

break;

}

}

5. 关闭程序的时候

CXXXDlg::~CXXXDlg()

{

WSACleanup();

::shutdown(m_sSocket, 2);

::closesocket(m_sSocket);

}

客户端：

与服务端一样。定义SOCKET变量，添加消息常量，添加消息映射。初始化SOCKET，关闭的时候清理SOCKET（WSACleanup）

void CXXXDlg::OnButtonSend()

{

char szError[256] = {0};

SOCKET m_Socket;

//创建套接字

if (m_IsTCP)

{

m_Socket = ::socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);

}

else

{

m_Socket = ::socket(AF_INET, SOCK_STREAM, 0);

}

if(INVALID_SOCKET == m_Socket)

{

sprintf(szError, "创建SOCKET错误，%d", WSAGetLastError());

AfxMessageBox(szError);

return;

}

//服务器地址信息

SOCKADDR_IN addrSock;

addrSock.sin_addr.S_un.S_addr = inet_addr(m_StrIP)/*htonl(INADDR_ANY)*/;

addrSock.sin_family = AF_INET;

addrSock.sin_port = htons(atoi(m_StrPort));

//连接服务器

//成功返回0 ，否则返回SOCKET_ERROR

if(SOCKET_ERROR == ::connect(m_Socket, (SOCKADDR*)&addrSock, sizeof(SOCKADDR)))

{

::closesocket(m_Socket);

sprintf(szError, "连接服务端错误，%d", WSAGetLastError());

AfxMessageBox(szError);

return;

}

else

{

CString SendBuff;

char RecvBuff[1024] = {0};

GetDlgItemText(IDC_EDIT_MSG, SendBuff);

::send(m_Socket, SendBuff, SendBuff.GetLength(), 0);

::recv(m_Socket, RecvBuff, 1024, 0);

static_cast<CListBox *>(GetDlgItem(IDC_LIST_TEXT))->AddString(RecvBuff);

}

closesocket(m_Socket);

}