多路复用的server模型

多路复用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 }