回环服务器_3.select

用slect()实现IO多路复用的回环服务器

/*
 * server.cpp
 *
 *  Created on: 2021年11月13日
 *      Author: LENOVO
 */
#include <errno.h>
#include <netinet/in.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <unistd.h>

#define MAXLNE 4096

int main(int argc,char **argv)
{
	int listenfd,connfd,n;
	struct sockaddr_in servaddr;
	char buff[MAXLNE];

	if((listenfd = socket(AF_INET,SOCK_STREAM,0)) == -1)
	{
		printf("create socket error: %s(errno: %d)\n", strerror(errno), errno);
		return 0;
	}

	memset(&servaddr,0,sizeof(servaddr));
    servaddr.sin_family = AF_INET;
    servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
    servaddr.sin_port = htons(8888);

    if(bind(listenfd,(struct sockaddr *)&servaddr,sizeof(servaddr)) == -1)
    {
        printf("bind socket error: %s(errno: %d)\n", strerror(errno), errno);
        return 0;
    }
    if (listen(listenfd, 10) == -1)
    {
        printf("listen socket error: %s(errno: %d)\n", strerror(errno), errno);
        return 0;
    }

    fd_set rfds, rset, wfds, wset;

	FD_ZERO(&rfds);
	FD_SET(listenfd, &rfds);

	FD_ZERO(&wfds);

	int max_fd = listenfd;

	while (1)
	{
		rset = rfds;
		wset = wfds;

		int nready = select(max_fd+1, &rset, &wset, NULL, NULL);

		if (FD_ISSET(listenfd, &rset))
		{ //
			struct sockaddr_in client;
			socklen_t len = sizeof(client);
			if ((connfd = accept(listenfd, (struct sockaddr *)&client, &len)) == -1)
			{
				printf("accept socket error: %s(errno: %d)\n", strerror(errno), errno);
				return 0;
			}

			FD_SET(connfd, &rfds);

			if (connfd > max_fd) max_fd = connfd;

			if (--nready == 0) continue;
		}

		int i = 0;
		for (i = listenfd+1;i <= max_fd;i ++)
		{
			if (FD_ISSET(i, &rset))
			{ //
				n = recv(i, buff, MAXLNE, 0);
				if (n > 0)
				{
					buff[n] = '\0';
					printf("recv msg from client: %s\n", buff);

					FD_SET(i, &wfds);
				}
				else if (n == 0)
				{ //
					FD_CLR(i, &rfds);
					printf("client %d disconnect\n",i);
					close(i);
				}
				if (--nready == 0) break;
			}
			else if (FD_ISSET(i, &wset))
			{
				send(i, buff, n, 0);
				FD_CLR(i, &wfds);
				//FD_SET(i, &rfds);
			}
		}
	}
}