select机制

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <unistd.h>

#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <netinet/ip.h>
#include <arpa/inet.h>

int client(int connfd)
{
	int ret;
	char buf[256];
	memset(buf, 0, sizeof(buf));
	ret = read(connfd, buf, sizeof(buf));
	if (ret == -1) {
		perror("server->read");
		return -1;
	} else if (ret == 0) {
		close(connfd);
		return -1;
	}

	printf("buf : %s\n", buf);
	ret = write(connfd, buf, sizeof(buf));
	if (ret == -1) {
		perror("server->write");
		return -1;
	}
	return 0;
}

int main(int argc, char *argv[])
{
	int listenfd;
	int ret;
	socklen_t addrlen;
	int connfd;
	pid_t pid;
	char buf[256];
	struct sockaddr_in srvaddr;
	struct sockaddr_in cltaddr;

	/* 1. 创建服务器（创建一socket套接字）；socket */
	listenfd = socket(AF_INET, SOCK_STREAM, 0);
	if (listenfd == -1) {
		perror("server->socket");
		return -1;
	}
	printf("create listenfd = %d success\n", listenfd);

	/* 2. 设置服务器的IP地址和端口号（将socket和服务器的IP地址和端口号进行绑定）；bind */
	memset(&srvaddr, 0, sizeof(struct sockaddr_in));
	srvaddr.sin_family = AF_INET;
	srvaddr.sin_port = htons(9999);
	srvaddr.sin_addr.s_addr = inet_addr("192.168.2.36");
	ret = bind(listenfd, (const struct sockaddr *)&srvaddr, sizeof(srvaddr));

	printf("port : %d\n", ntohs(srvaddr.sin_port));
	if (ret == -1) {
		perror("server->bind");
		return -1;
	}
	printf("bind success !\n");

	/* 3. 启动监听（启动服务器）；  listen */
	ret = listen(listenfd, 1024);
	if (ret == -1) {
		perror("server->listen");
		return -1;
	}
	printf("listen success !\n");

	int fd;
	int nfds;
	/* 创建一个集合，用来存储所要读处理的所有的文件描述符*/
	fd_set readfds;
	fd_set rfds;
	/* 添加要读处理的文件描述符 */
	FD_SET(listenfd, &readfds);
	nfds = listenfd+1;

	while(1) {
		rfds = readfds;
		/* 检测集合当中是否有继续的文件描述符 */
		ret = select(nfds, &rfds, NULL, NULL, NULL);

		/* 轮循 */
		for (fd = 0; fd < nfds; fd++) {
			/* 判断，寻找就绪的文件描述符 */
			if (FD_ISSET(fd, &rfds)) {
				/* 如果是监听套接字listenfd，则建立连接 */
				if (fd == listenfd) {
					memset(&cltaddr, 0, sizeof(cltaddr));
					addrlen = sizeof(socklen_t);
					connfd = accept(listenfd, (struct sockaddr *)&cltaddr, &addrlen);
					if (connfd == -1) {
						perror("accept");
						return -1;
					}

					printf("connfd = %d\n", connfd);
					FD_SET(connfd, &readfds);
					if (connfd >= nfds) {
						nfds = connfd+1;
					}
					//nfds = connfd >= nfds ? connfd+1 : nfds;
				} else {
					ret = client(fd);
					if (ret == -1) {
						FD_CLR(fd, &readfds);
					}
				}
			}
		}
	}
	
	close(listenfd);

	return 0;
}