socket 的select函数例子

好久没有写博客了，主要最近也没有学啥新的可以写博客的东西

select函数是通过设置FD_SET来实现在一个线程中实现读写网口的，下面这段代码主要是实现绑定端口-》监听-》设置socket为非阻塞-》读取网口-》回写网口的。缺点是只能有一个socket连接，优点是代码简单，便于入门。在这里鄙视一下 《socket编程之select()》的博主，写个代码还故意整点bug，不过他写的文字还是很好的，可以配合下面的代码看看，不过代码不是他的哦。

#include <WINSOCK2.H>
#include <iostream>

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

#define PORT 8080
#define DATA_BUFSIZE 8192
char recvBuffer[DATA_BUFSIZE];
//目标是一个客户端可以连接

int main()
{
SOCKET ListenSocket; // 监听套接字
SOCKET AcceptSocket; // 与客户端进行通信的套接字
SOCKADDR_IN InternetAddr; // 服务器的地址
WSADATA wsaData; // 用于初始化套接字环境
INT Ret; // WinSock API的返回值
FD_SET WriteSet; // 获取可写性的套接字集合
FD_SET ReadSet; // 获取可读性的套接字集合
DWORD Total = 0; // 处于就绪状态的套接字数量
DWORD SendBytes; // 发送的字节数
DWORD RecvBytes; // 接收的字节数

bool bSocketInit = false;//socket是否初始化过

// 初始化WinSock环境
if ((Ret = WSAStartup(0x0202,&wsaData)) != 0)
{
printf("WSAStartup() failed with error %d\n", Ret);
WSACleanup();
return -1;
}
ListenSocket = socket(AF_INET,SOCK_STREAM,IPPROTO_TCP);
InternetAddr.sin_family = AF_INET;
InternetAddr.sin_addr.s_addr = htonl(INADDR_ANY);
InternetAddr.sin_port = htons(PORT);
//绑定
if(bind(ListenSocket, (PSOCKADDR)&InternetAddr, sizeof(InternetAddr)) == SOCKET_ERROR)
{
printf("bind() failed with error %d\n", WSAGetLastError());
return -1;
}
//监听
listen(ListenSocket,1);
//设置费阻塞
u_long noneBlock = 1 ;
ioctlsocket(ListenSocket,FIONBIO,&noneBlock);
int recvLen = 0;
while(true)
{
//清空
FD_ZERO(&ReadSet);
FD_ZERO(&WriteSet);

//指定listensocket可以读
FD_SET(ListenSocket,&ReadSet);

if(bSocketInit)
{
FD_SET(AcceptSocket,&ReadSet);//注意，就只能连接一个，因为这样代码简单
FD_SET(AcceptSocket,&WriteSet);
}

//循环调用select函数
select(1,&ReadSet,&WriteSet,NULL,NULL);//不检测错误信息，不阻塞

//判断有没有新的socket连接
if(FD_ISSET(ListenSocket,&ReadSet))
{
AcceptSocket = accept(ListenSocket,NULL,NULL);
//设置新来的socket为非阻塞
ioctlsocket(AcceptSocket,FIONBIO,&noneBlock);
bSocketInit = true;//就能连接一个
}
//判断新的socket
if(bSocketInit)
{
if(FD_ISSET(AcceptSocket,&ReadSet)){
//接受数据
int len = recv(AcceptSocket,recvBuffer,DATA_BUFSIZE,0);
if(len>0){
recvLen = len;
}else{
//客户端关闭了
bSocketInit = false;
closesocket(AcceptSocket);
len = 0;
recvLen = 0;
}
}
if(FD_ISSET(AcceptSocket,&WriteSet))
{
//
if(recvLen>0)
{
if( send(AcceptSocket,recvBuffer,recvLen,0)
!= SOCKET_ERROR)
{
recvLen = 0;
printf(recvBuffer);
}
}
}
}
}
}