linux select编程示例

小灰灰 2022-07-14 10:57 153阅读 0赞

## select编程示例 ##

*代码中出现的errwrap.h和errwrap.c请参考我的上一篇[文章][Link 1]*

### 服务器 ###

#include <stdio.h>
    #include <stdlib.h>
    #include <string.h>
    #include <netinet/in.h>
    #include <arpa/inet.h>
    #include "errwrap.h"
    
    #define MAXLINE 80
    #define SERVERPORT 8000
    
    int main(int argc, char *argv[])
    {
    		int i, maxi, maxfd, listenfd, confd, sockfd;
    		int nret, client[FD_SETSIZE]; // FD_SETSIZE 默认为1024 
    		ssize_t n;
    		fd_set rset, allset;
    		char buf[MAXLINE];
    		char str[INET_ADDRSTRLEN]; //#define INET_ADDRSTRLEN 16 
    		socklen_t clientaddr_len;
    		struct sockaddr_in clientaddr, serveraddr;
    		listenfd = Socket(AF_INET, SOCK_STREAM, 0);//获得一个监听fd
    		
    		bzero(&serveraddr, sizeof(serveraddr));
    		serveraddr.sin_family = AF_INET;
    		serveraddr.sin_addr.s_addr = htonl(INADDR_ANY);
    		serveraddr.sin_port = htons(SERVERPORT);
    		Bind(listenfd, (struct sockaddr *)&serveraddr, sizeof(serveraddr));
    		Listen(listenfd, 128); 
    
    		maxfd = listenfd; //maxfd为现在监听的文件描述符的最大值
    		maxi = -1; /* client[]的下标*/
    		for (i = 0; i < FD_SETSIZE; i++)
    				client[i] = -1; /* 用-1初始化client[] */
    
    		FD_ZERO(&allset);
    		FD_SET(listenfd, &allset); // 设置allset（所有监听文件描述符集合最新状态）
    		for ( ; ; ) 
    		{
    				rset = allset; //当前文件描述符集合
    				nret = select(maxfd+1, &rset, NULL, NULL, NULL);//检测“读”
    				if (nret < 0)
    						perr_exit("select error");
    						if (FD_ISSET(listenfd, &rset)) //如果是新的连接到来
    						{ 
    								clientaddr_len = sizeof(clientaddr);
    								confd = Accept(listenfd, (struct sockaddr *)&clientaddr, &clientaddr_len);
    								printf("received from %s at PORT %d\n",
    												inet_ntop(AF_INET, &clientaddr.sin_addr, str, sizeof(str)),
    												ntohs(clientaddr.sin_port));
    								for (i = 0; i < FD_SETSIZE; i++)
    								{
    										if (client[i] < 0) 
    										{
    												client[i] = confd; //将新的文件描述符添加到client里
    												break;
    										}
    								}
    								// 达到select能监控的文件个数上限1024 
    								if (i == FD_SETSIZE) 
    								{
    										fputs("too many clients\n", stderr);
    										exit(1);
    								}
    								FD_SET(confd, &allset); //将新的新的文件描述符添加到监控信号集里
    								if (confd > maxfd)
    										maxfd = confd; //更新最大的文件描述符
    								if (i > maxi)
    										maxi = i; // 更新client最大下标
    								if (--nret == 0)
    										continue; /* 如果没有更多的就绪文件描述符继续回到上面select阻塞监听,负责处理未
    													 处理完的就绪文件描述符*/
    						}
    				for (i = 0; i <= maxi; i++) 
    				{ // 检测哪个clients 有数据就绪
    						if ( (sockfd = client[i]) < 0)
    								continue;
    						if (FD_ISSET(sockfd, &rset)) //对连接到服务器的每个客户机（文件描述符），检测是否有数据到达
    						{
    								if ( (n = Read(sockfd, buf, MAXLINE)) == 0) 
    								{
    										// 当client关闭链接时，服务器端也关闭对应链接
    										Close(sockfd);
    										FD_CLR(sockfd, &allset); // 解除select监控此文件描述符
    										client[i] = -1;
    								} else 
    								{//处理每个有数据到达的fd
    										int j;
    										for (j = 0; j < n; j++)
    												buf[j] = toupper(buf[j]);
    												Write(sockfd, buf, n);
    								}
    								if (--nret == 0)//处理的那个fd不是最后一个
    										break;
    						}
    				}
    		}
    		close(listenfd);
    		return 0;
    }

### 客户端 ###

#include <stdio.h>
    #include <string.h>
    #include <unistd.h>
    #include <netinet/in.h>
    #include "errwrap.h"
    
    #define MAXLINE 80
    #define SERV_PORT 8000
    
    int main(int argc, char *argv[])
    {
    		if(argc<2)
    		{
    			err_exit("参数过少");
    		}
    		struct sockaddr_in servaddr;
    		char buf[MAXLINE];
    		int sockfd, n;
    		sockfd = Socket(AF_INET, SOCK_STREAM, 0);
    		bzero(&servaddr, sizeof(servaddr));
    		servaddr.sin_family = AF_INET;
    		inet_pton(AF_INET, argv[1], &servaddr.sin_addr);
    		servaddr.sin_port = htons(SERV_PORT);
    		Connect(sockfd, (struct sockaddr *)&servaddr, sizeof(servaddr));
    		while (fgets(buf, MAXLINE, stdin) != NULL) 
    		{
    				Write(sockfd, buf, strlen(buf));
    				n = Read(sockfd, buf, MAXLINE);
    				if (n == 0)
    						printf("the other side has been closed.\n");
    				else
    						Write(STDOUT_FILENO, buf, n);
    		}
    		Close(sockfd);
    		return 0;
    }

[Link 1]: http://blog.csdn.net/u010155023/article/details/53507656