Socket WSAAsyncSelect模型值得注意的一个地方

参考Programming Windows TCP Sockets in C++ for the Beginners(见codeproject)， socket I/O模型选择WSAAsyncSelect进行编程时，参考代码有个bug：Server端第一次连接并读取Client端发送的buffer OK， 第二次就没有FD_READ事件了。

 

问题相关代码 

WSAAsyncSelect (s, hwnd, 1045, FD_READ | FD_CONNECT | FD_CLOSE | FD_ACCEPT);

 

case FD_CLOSE: //Lost connection MessageBeep(MB_ICONERROR); //Clean up if (s) closesocket(s); WSACleanup(); SendMessage(hStatus, SB_SETTEXT, 0, (LPARAM)"Connection to Remote break; case FD_READ: //Incoming data to receive char buffer[80]; memset(buffer, 0, sizeof(buffer)); //Clear the buffer recv (s, buffer, sizeof(buffer)-1, 0); //Get the text GetTextandAddLine(buffer, hwnd, ID_EDIT_DATA); //Display it break; case FD_ACCEPT: //Connection request { MessageBeep(MB_OK); SOCKET TempSock = accept(s, (struct sockaddr*)&from, &fromlen); s = TempSock; //Switch our old socket to the new one char szAcceptAddr[100]; wsprintf(szAcceptAddr, "Connection from [%s] accepted.", inet_ntoa(from.sin_addr)); SendMessage(hStatus, SB_SETTEXT, 0, (LPARAM)szAcceptAddr); } break;

 

s(套接字)为全局变量。以上代码可以看到accept客户端连接替换的套接字与WSAAsyncSelect模型事件通知的套接字同一变量。

 

看看事件顺序

第一次客户端连接并发送buffer

 FD_ACCEPT  WSAAsyncSelect模型事件通知的套接字替换为accept客户端连接替换的套接字

 FD_READ      从accept客户端连接替换的套接字读取buffer

 FD_CLOSE   关闭客户端与Server的socket连接

 

第二次客户端连接并发送buffer

FD_ACCEPT  此时WSAAsyncSelect模型事件通知的套接字已经非法，使用accept从WSAAsyncSelect模型事件通知的套接字获取客户端连接套接字显示失败。

 

所以我们需要区分客户端连接替换的套接字与WSAAsyncSelect模型事件通知的套接字。

 

解决方案如下：

1、定义两个socket全局变量

SOCKET s;  //WSAAsyncSelect模型事件通知的套接字同一变量

SOCKET c;  //accept客户端连接替换的套接字

 

 

case FD_CLOSE: //Lost connection             

if (c) closesocket(c);

break;

 

 

 

case FD_READ: //Incoming data to receive

static char buffer[80];

memset(buffer, 0, sizeof(buffer)); //Clear the buffer        

recv (c, buffer, sizeof(buffer)-1, 0); //Get the text

break;

 

 

 

case FD_ACCEPT: //Connection request

{

char szAcceptAddr[100];

SOCKET TempSock = accept(s, (struct sockaddr*)&from, &fromlen);

c= TempSock; //Switch our old socket to the new one  

break;

}