多路复用的server模型

女爷i 2022-08-21 00:10 138阅读 0赞

多路复用I/O之server模型   
 主要是关于select（）这个函数：   
其原型是：int select(int n,fd\_set \*read\_fds,fd\_set \*write\_fds,fd\_set    
  
  
\*except,struct timeval \*timeout);   
  
  
功能是：监测指定集合中指定文件描述符的指定事件是否发生；   
参数一：读集合，监测登记文件描述的读事件；   
参数二：写集合，监测登记文件描述的写事件   
参数三：报错集合，监测登记文件描述的报错事件   
  
  
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); //初始化集合   
  
  
  
  
  1 \#include <stdio.h>   
  2 \#include <sys/time.h>   
  3 \#include <sys/types.h>   
  4 \#include <unistd.h>   
  5 \#include <fcntl.h>   
  6 \#include <sys/types.h>          /\* See NOTES \*/   
  7 \#include <sys/socket.h>   
  8 \#include <netinet/in.h>   
  9 \#include <arpa/inet.h>   
 10    
 11 \#include <string.h>   
 12    
 13 \#define PORT    6543   
 14    
 15 int main()   
 16 \{   
 17         /\*创建套接字，用于连接\*/   
 18         int s = socket(AF\_INET, SOCK\_STREAM, 0);   
 19         if(0 > s)\{   
 20                 perror("socket");   
 21                 return -1;   
 22         \}   
 23    
25    
 26         /\*端口重用\*/   
 27         int reuseaddr = 1;   
 28         if(0 > setsockopt(s, SOL\_SOCKET, SO\_REUSEADDR, &reuseaddr,    
  
  
sizeof(re    useaddr)))\{   
 29                 perror("setsockopt");   
 30                 return -1;   
 31         \}   
 32    
 33    
 34    
 35         /\*绑定server IP及端口号\*/   
 36         struct sockaddr\_in addr;   
 37         memset(&addr, 0, sizeof(struct sockaddr\_in));   
 38         addr.sin\_family = AF\_INET;   
 39         addr.sin\_port = htons(PORT);   
 40         addr.sin\_addr.s\_addr = htonl(INADDR\_ANY);   
 41    
 42         socklen\_t addrlen = sizeof(struct sockaddr\_in);   
 43         /\*struct sockaddr\_in addr初始化必须，socklen\_t addrlen\*/   
 44         if(0 > bind(s, (struct sockaddr \*)&addr, addrlen))\{   
 45                 perror("bind");   
 46                 return -1;   
 47         \}   
 48    
 49         /\*设置监听数\*/   
 50         listen(s, 10);   
 51    
 52         /\*设置套接字为非阻塞\*/   
 53         int val = fcntl(s, F\_GETFL, 0);   
 54         val |= O\_NONBLOCK;   
 55         fcntl(s, F\_SETFL, val);   
 56    
 57         /\*初始化读集合\*/   
 58         fd\_set oldset, newset;   
 59 /这里将oldset集合初始化   
 60         FD\_ZERO(&oldset);   
 61    
 62 将开始的套接字添加到集合oldset中   
 63         FD\_SET(s, &oldset);   
 64    
 65    
 66 ///这里用maxfd得到最大的套接字数   
 67         int maxfd = s+1;   
 68    
 69 //这里设置为select功能实现的时间，超过跳过   
 70         struct timeval tm = \{   
 71                 .tv\_sec = 1,   
 72                 .tv\_usec= 0,   
 73         \};   
 74    
 75         printf("Wait for connecting .\\n");   
 76         while(1)\{   
 77    
 78 //              printf("-------------1-----------\\n");             
 79 将oldset这个集合copy给newset（动态包含的套接字的集合）   
 80                 newset = oldset;   
 81                 //int ret = select(maxfd, &newset, NULL, NULL,    
  
  
&tm);   
 82    
 83 下面将集合newset，用读监听，而且这里没有设置功能运行时间   
 84                 int ret = select(maxfd, &newset, NULL, NULL, NULL);   
 85                 printf("------------------------\\n");   
86                 if(0 > ret)\{   
 87                         perror("select");   
 88                         return -1;   
 89                 \}else if(0 == ret)\{  //如果超时运行这个条件   
 90                         printf("Timeout.\\n");   
 91                         continue;   
 92                 \}else\{   
 93                         /select成功   
 94 //              printf("-------------1-----------\\n");             
 95                         /\*测试用于连接的套接字，看是否有新连接进来   
  
  
\*/   
 96                         if(FD\_ISSET(s, &newset))\{   
 97                                 struct sockaddr\_in newaddr;   
 98                                 memset(&newaddr, 0, sizeof(struct    
  
  
sockaddr\_i    n));   
 99                                 socklen\_t newlen = sizeof(struct    
  
  
sockaddr\_in    );   
100                                 /\*有新连接进来，得到用于读写到套接   
  
  
字\*/   
101                                 int news = accept(s, (struct    
  
  
sockaddr \*)&new    addr, &newlen);   
102                                 if(0 > news)\{   
103                                         perror("accept");   
104                                         return -1;   
105                                 \}   
106                                 /\*把套接字保存进集合\*/   
107                         //      printf("-------------1-----------   
  
  
\\n");                 
108    
109                                 FD\_SET(news, &oldset);   
110                                 if(news+1 > maxfd) \{   
111                                         maxfd = news+1;   
112                                 \}   
113                 //              maxfd++;   
114                                 printf("New connect: SOCKET\[%d\], %   
  
  
s:%d\\n", n    ews, inet\_ntoa(newaddr.sin\_addr), ntohs   
  
  
(newaddr.sin\_port));   
115                         \}   
116    
117                         for(int i = s+1; i < maxfd; i++)\{   
118                                 /\*测试用于读写到套接字，看哪个客户   
  
  
端发数据进   
    来\*/   
119                                 if(FD\_ISSET(i, &newset))\{   
120                                 \#define MAX 64   
121                                         char buf\[MAX\];   
122                                         bzero(buf, MAX);   
123                                         int num = recv(i, buf, MAX,    
  
  
0);   
124                                         if(0 >= num)\{   
125                                                 FD\_CLR(i, &oldset);   
126                                                 close(i);   
127                                                 printf("SOCKET\[%d\]    
  
  
leave.\\n"    , i);   
128                                                 continue;   
129                                         \}   
130                                         printf("SOCKET\[%d\]: %s\[%d\]   
  
  
\\n", i, bu    f, num);   
131                                 \} //if   
132                         \} //for   
133                 \} //if   
134         \} //while   
135 \}