C++中I/O模型之select模型實例
更新時間:2014年10月20日 14:33:55 投稿:shichen2014
這篇文章主要介紹了C++中I/O模型的select模型,實例講述了I/O模型的用法,具有一定的參考借鑒價值,需要的朋友可以參考下
本文實例講述了C++中I/O模型的select模型用法。分享給大家供大家參考。具體實現方法如下:
復制代碼 代碼如下:
void main()
{
CInitSock initSock;
USHORT nPort = 9999; //監(jiān)聽的端口
SOCKET sListen = ::socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
if (sListen == INVALID_SOCKET)
{
printf("socket error...");
return;
}
sockaddr_in servAddr = {0};
servAddr.sin_family = AF_INET;
servAddr.sin_port = ::htons(nPort);
servAddr.sin_addr.S_un.S_addr = INADDR_ANY;
if (SOCKET_ERROR == ::bind(sListen, (sockaddr*)&servAddr, sizeof(servAddr)))
{
int nError = ::GetLastError();
printf("connect error..");
return;
}
::listen(sListen, 5);
//select模型處理過程
//1.初始化套接字集合,添加監(jiān)聽socket到這個集合
fd_set fdSocket;
FD_ZERO(&fdSocket);
FD_SET(sListen, &fdSocket);
while (TRUE)
{
//2. 將集合的一個拷貝傳遞給select函數
//當有事件發(fā)生時,select移除未決的socket,然后返回 就是說select返回時 集合的中socket就是發(fā)生事件的socket
fd_set fdRead = fdSocket;
int nRet = ::select(0, &fdRead, NULL, NULL, NULL);
if (nRet > 0)
{
//通過比較原來的fdSocket集合與經過處理的fdRead集合
for (UINT i=0;i<fdSocket.fd_count;i++)
{
if (FD_ISSET(fdSocket.fd_array[i], &fdRead)) //就是這個觸發(fā)了
{
if (fdSocket.fd_array[i] == sListen) //監(jiān)聽套接字接收到新連接 為神馬分兩種情況:因為觸發(fā)Read集合的情況有兩種:有連接來了或數據可讀了....
{
//
if (fdSocket.fd_count < FD_SETSIZE) //FD_SETSIZE=64
{
sockaddr_in addrRemote = {0};
int nAddrLen = sizeof(addrRemote);
SOCKET sNew = ::accept(sListen, (sockaddr*)&addrRemote, &nAddrLen);
FD_SET(sNew, &fdSocket);
}
else
{
printf("too many connection...error");
continue;
}
}
else //有可讀的
{
char szContent[256]={0};
int nRecv = ::recv(fdSocket.fd_array[i], szContent, sizeof(szContent), 0);
if (nRecv > 0)
{
szContent[nRecv] = '\0';
printf("recv data:%s", szContent);
}
else //沒讀到數據
{
::closesocket(fdSocket.fd_array[i]);
FD_CLR(fdSocket.fd_array[i], &fdSocket);
}
}
}
}
}
else
{
printf("nRet litter 0, error...");
return;
}
}
::closesocket(sListen); //與socket配對寫
printf("*******************************");
getchar();
}
{
CInitSock initSock;
USHORT nPort = 9999; //監(jiān)聽的端口
SOCKET sListen = ::socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
if (sListen == INVALID_SOCKET)
{
printf("socket error...");
return;
}
sockaddr_in servAddr = {0};
servAddr.sin_family = AF_INET;
servAddr.sin_port = ::htons(nPort);
servAddr.sin_addr.S_un.S_addr = INADDR_ANY;
if (SOCKET_ERROR == ::bind(sListen, (sockaddr*)&servAddr, sizeof(servAddr)))
{
int nError = ::GetLastError();
printf("connect error..");
return;
}
::listen(sListen, 5);
//select模型處理過程
//1.初始化套接字集合,添加監(jiān)聽socket到這個集合
fd_set fdSocket;
FD_ZERO(&fdSocket);
FD_SET(sListen, &fdSocket);
while (TRUE)
{
//2. 將集合的一個拷貝傳遞給select函數
//當有事件發(fā)生時,select移除未決的socket,然后返回 就是說select返回時 集合的中socket就是發(fā)生事件的socket
fd_set fdRead = fdSocket;
int nRet = ::select(0, &fdRead, NULL, NULL, NULL);
if (nRet > 0)
{
//通過比較原來的fdSocket集合與經過處理的fdRead集合
for (UINT i=0;i<fdSocket.fd_count;i++)
{
if (FD_ISSET(fdSocket.fd_array[i], &fdRead)) //就是這個觸發(fā)了
{
if (fdSocket.fd_array[i] == sListen) //監(jiān)聽套接字接收到新連接 為神馬分兩種情況:因為觸發(fā)Read集合的情況有兩種:有連接來了或數據可讀了....
{
//
if (fdSocket.fd_count < FD_SETSIZE) //FD_SETSIZE=64
{
sockaddr_in addrRemote = {0};
int nAddrLen = sizeof(addrRemote);
SOCKET sNew = ::accept(sListen, (sockaddr*)&addrRemote, &nAddrLen);
FD_SET(sNew, &fdSocket);
}
else
{
printf("too many connection...error");
continue;
}
}
else //有可讀的
{
char szContent[256]={0};
int nRecv = ::recv(fdSocket.fd_array[i], szContent, sizeof(szContent), 0);
if (nRecv > 0)
{
szContent[nRecv] = '\0';
printf("recv data:%s", szContent);
}
else //沒讀到數據
{
::closesocket(fdSocket.fd_array[i]);
FD_CLR(fdSocket.fd_array[i], &fdSocket);
}
}
}
}
}
else
{
printf("nRet litter 0, error...");
return;
}
}
::closesocket(sListen); //與socket配對寫
printf("*******************************");
getchar();
}
效果如下圖所示:
希望本文所述對大家的C++程序設計有所幫助。