- select实现TCP小型并发服务器的流程:
- 一、创建套接字(socket函数):
通信域
选择IPV4
网络协议、套接字类型选择流式
;
int sockfd = socket(AF_INET,SOCK_STREAM,0);
- 二、填充服务器的网络信息结构体:
- 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; serveraddr.sin_addr.s_addr = inet_addr(argv[1]); serveraddr.sin_port = htons(atoi(argv[2]));
- 三、套接字和服务器的网络信息结构体进行绑定(bind函数):
int ret = bind(sockfd,(struct sockaddr *)&serveraddr,serveraddrlen);
int ret1 = listen(sockfd, 5);
- 五、创建要监听的文件描述符集合:
- 使用select函数后,会将
没有就绪的文件描述符
在集合中去除
,所以需要创建两个文件描述符集合
,一个是母本read_fds
,类似于常量,保持不变,而另一个作为副本read_fds_t
,类似于变量,可以改变;
fd_set read_fds;FD_ZERO(&read_fds);fd_set read_fds_t;FD_ZERO(&read_fds_t);
FD_SET(sockfd,&read_fds);int fd_max = 0;fd_max = fd_max > sockfd ? fd_max : sockfd;
struct timeval tm_out;
- 八、等待文件描述符中的事件是否就绪,成功则返回就绪的文件描述符的个数(select函数):
- select函数:
#include <sys/select.h>int select(int nfds, fd_set *readfds, fd_set *writefds,fd_set *exceptfds, struct timeval *timeout);struct timeval {long tv_sec; long tv_usec; };void FD_CLR(int fd, fd_set *set); int FD_ISSET(int fd, fd_set *set); void FD_SET(int fd, fd_set *set); void FD_ZERO(fd_set *set);
if(-1 == (ret = select(fd_max + 1,&read_fds_t,NULL,NULL,&tm_out))){perror("select error");exit(-1);}else if(0 == ret){puts("timeout!!!!!");putchar('\n');continue;}
- 九、遍历文件描述符集合,判断哪些文件描述符已经准备就绪:
for(int i = 3; i < fd_max + 1 && 0 != ret; i++){...}
- 十、判断文件描述符是否还在集合中,并且接收来自客户端的数据(recv函数)和给客户端发送应答消息(send函数):
if(FD_ISSET(i,&read_fds_t)){if(i == sockfd){ if(-1 == (accept_fd = accept(sockfd,NULL,NULL))){perror("accept error");exit(-1);}printf("客户端[%d]连接到服务器\n",accept_fd);FD_SET(accept_fd,&read_fds);fd_max = fd_max > accept_fd ? fd_max : accept_fd;}else {memset(buf,0,sizeof(buf));if(-1 == (nbytes = recv(i,buf,sizeof(buf),0))){perror("recv error");exit(-1);}else if(0 == nbytes){printf("客户端[%d]已断开连接\n",i);FD_CLR(i,&read_fds);close(i);continue;}if(!strncmp(buf,"quit",4)){printf("客户端[%d]已退出\n",i);FD_CLR(i,&read_fds);close(i);continue;}printf("客户端[%d]发来信息[%s]\n",i,buf);strcat(buf,"----------k");if(-1 == send(i,buf,sizeof(buf),0)){perror("send error");exit(-1);}}ret--; }
close(sockfd);