C/S架构学习之使用poll实现TCP中型并发服务器
- poll实现TCP中型并发服务器的流程：
- 一、创建套接字（socket函数）：
通信域选择IPV4网络协议、套接字类型选择流式；
int sockfd = socket(AF_INET,SOCK_STREAM,0); //通信域选择IPV4、套接字类型选择流式
- 二、填充服务器的网络信息结构体：
- 1.定义网络信息结构体变量；
- 2.求出网络信息结构体变量的内存空间大小，以作备用；
- 3.网络信息
结构体清0； - 4.使用
IPV4网络协议AF_INET； - 5.在终端输入的服务器端主机的
IP地址：inet_addr(argv[1])； - 6.在终端输入的服务器端网络字节序的
端口号：htons(atoi(argv[2]))；
struct sockaddr_in serveraddr; //定义网络信息结构体变量
socklen_t serveraddrlen = sizeof(serveraddr);//求出结构体变量的内存空间大小
memset(&serveraddr,0,serveraddrlen); //结构体清零
serveraddr.sin_family = AF_INET; //使用IPV4网络协议
serveraddr.sin_addr.s_addr = inet_addr(argv[1]); //IP地址
serveraddr.sin_port = htons(atoi(argv[2]));//网络字节序的端口号
- 三、套接字和服务器的网络信息结构体进行绑定（bind函数）：
int ret = bind(sockfd,(struct sockaddr *)&serveraddr,serveraddrlen);
- 四、套接字设置成被动监听（listen函数）：
int ret1 = listen(sockfd, 5);
- 五、创建要监听的文件描述符集合并清空文件描述符集合：
//创建要监听的文件描述符集合
struct pollfd new_fds[2048] = {0};
//清空文件描述符集合
for(int i = 0; i < 2048; ++i)
{
new_fds[i].fd = -1;
}
- 六、把创建的套接字添加到要监视的集合中：
FD_SET(sockfd,&read_fds);
int fd_max = 0;
fd_max = fd_max > sockfd ? fd_max : sockfd;
- 七、套接字添加到要监视的集合中，并且设置要监听的事件：
//套接字添加到要监视的集合中
new_fds[0].fd = sockfd;
//设置要监听的事件
new_fds[0].events |= POLLIN;
- 八、记录文件描述符集合中最大的文件描述符，并且设置超时的时间：
//记录文件描述符集合中最大的文件描述符
int fd_max = 0;
fd_max = fd_max > sockfd ? fd_max : sockfd;
//设置超时的时间
int tm_out = 10000;
- 九、等待文件描述符中的事件是否就绪，成功则返回就绪的文件描述符的个数（poll函数）：
poll实现TCP中型并发服务器与select实现TCP小型并发服务器的区别就是无需每次重置集合,并且可以设置要监视的最大文件描述符的个数，而select至多监视1024个文件描述符；- poll函数：
#include if(-1 == (ret = poll(new_fds,fd_max,tm_out)))
{
perror("poll error");
exit(-1);
}
else if(0 == ret)
{
puts("timeout!!!!!");
putchar('\n');
continue;
}
- 十、遍历文件描述符集合，判断哪些文件描述符已经准备就绪：
for(k = 0; k <= fd_max && ret != 0; ++k)
{
...
}
- 十一、找到实际就绪的事件的文件描述符，并且接收来自客户端的数据（recv函数）和给客户端发送应答消息（send函数）：
//找到实际就绪的事件的文件描述符
if(0 != (new_fds[k].revents & POLLIN))
{
//说明有新的客户端连接服务器
if(new_fds[k].fd == sockfd)
{
if(-1 == (accept_fd = accept(sockfd,NULL,NULL)))
{
perror("accept error");
exit(-1);
}
printf("客户端[%d]连接到服务器\n",accept_fd);
//将新连接的客户端的套接字添加到要监视的集合中
//遍历文件描述符集合，给新的文件描述符找一个位置
for(j = 0; j < 2048; j++)
{
if(-1 == new_fds[j].fd)
{
new_fds[j].fd = accept_fd;
new_fds[j].events |= POLLIN;
fd_max = fd_max > accept_fd ? fd_max : accept_fd;
break;
}
}
if(2048 == j)
{
//此时集合容量满了
close(accept_fd);
}
}
else //之前连接的客户端在向服务器发送信息
{
memset(buf,0,sizeof(buf));
if(-1 == (nbytes = recv(new_fds[k].fd,buf,sizeof(buf),0)))
{
perror("recv error");
exit(-1);
}
else if(0 == nbytes)
{
printf("客户端[%d]已断开连接\n",new_fds[k].fd);
//将已断开连接客户端的套接字在文件描述符集合中剔除
close(new_fds[k].fd);
new_fds[k].fd = -1;
continue;
}
if(!strncmp(buf,"quit",4))
{
printf("客户端[%d]已退出\n",new_fds[k].fd);
//将已断开连接客户端的套接字在文件描述符集合中剔除
close(new_fds[k].fd);
new_fds[k].fd = -1;
continue;
}
printf("客户端[%d]发来信息[%s]\n",new_fds[k].fd,buf);
//组装应答消息
strcat(buf,"----------k");
//给客户端发送应答消息
if(-1 == send(new_fds[k].fd,buf,sizeof(buf),0))
{
perror("send error");
exit(-1);
}
}
ret--; //减少遍历次数
}
- 十二、关闭套接字（close函数）：
close(sockfd);