IO多路复用select模型

IO多路复用select模型

#include <stdio.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <unistd.h>
#include <arpa/inet.h>
int main()
{	
	// 创建一个socket的文件描述符
	int lfd = socket(AF_INET, SOCK_STREAM, 0);
	
	//将socket与ip和端口号绑定
	struct sockaddr_in serv;
	serv.sin_family = AF_INET;
	serv.sin_port = htons(8888);
	serv.sin_addr.s_addr = INADDR_ANY; //监听所有本机所有ip
	bind(lfd, (struct sockaddr*)&serv, sizeof(serv));

	// 监听这个socket，就是监听这个ip和端口
	listen(lfd, 128);

	// 声明
	fd_set readfds;
	fd_set tmpfds;
	// 将readfds和tmpfds初始化，其中readfds保存的是要监听的sock，tmpfds是变化的socket也就是要处理的在下面select中会输出。
	FD_ZERO(&readfds);
	FD_ZERO(&tmpfds);
	// readfds监视lfd本身的socket，如果他有读事件说明有新的客户端连接
	FD_SET(lfd, &readfds);

	int nready, cfd, sockfd, i, n;
	int maxfd = lfd;
	char buf[1024];
	while (1) {
		// tmpfds复制成为要监听的socket
		tmpfds = readfds;
		// select里tmpfds是输出参数，tmpfds经过select处理后里面仅有有事件发生的socket
		int nready = select(maxfd + 1, &tmpfds, NULL, NULL, NULL);
		if (nready < 0) {
			printf("errrno\n");
			return -1;
		}
		// 如果服务器的socket（lfd）有事件待处理说明，有新的客户端连接
		if (FD_ISSET(lfd, &tmpfds)) {
			// 获取这个客户端socket（cfd），这里accept就肯定有要接受连接不会阻塞
			cfd = accept(lfd, NULL, NULL);
			// 监听这个cfd
			FD_SET(cfd, &readfds);
			if (maxfd < cfd) {
				maxfd = cfd;
			}
			// nready返回的是待处理事件的个数，如果为0就不用继续下面的操作
			if (--nready == 0) {
				continue;
			}
		}
		for (i = lfd + 1; i < maxfd + 1; i++) {
			sockfd = i;
			// 文件描述符从lfd服务端开始慢慢+1，依次判断他是否在tmpfds里面，如果在说明这个socket发过来信息
			if (FD_ISSET(sockfd, &tmpfds)) {
				memset(buf, 0*00, sizeof(buf));
				n = read(sockfd, buf, sizeof(buf));
				if (n <= 0) {
					// 客户端没有发送信息过来，关闭这个客户端，并且取消监听他
					printf("%dduankai\n", sockfd);
					close(sockfd);
					FD_CLR(sockfd, &readfds);
					continue;
				}
				else {
					printf("%dclient:%s\n", sockfd, buf);
					write(sockfd, buf, strlen(buf));
				}
				if (--nready == 0) {
					break;
				}
			}
			
		}
	}
	close(lfd);
	return 0;
}