回环服务器_5.epoll

使用epoll实现

/*
 * 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>
#include <sys/poll.h>
#include <sys/epoll.h>

#define MAXLNE 4096
#define POLL_SIZE	1024

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;
    }
    
	int epfd = epoll_create(1); //int size
	
	struct epoll_event events[POLL_SIZE] = {0};
	struct epoll_event ev;

	ev.events = EPOLLIN;
	ev.data.fd = listenfd;

	epoll_ctl(epfd, EPOLL_CTL_ADD, listenfd, &ev);

	while (1)
	{
		int nready = epoll_wait(epfd, events, POLL_SIZE, 5);
		if (nready == -1)
		{
			continue;
		}

		int i = 0;
		for (i = 0;i < nready;i ++)
		{
			int clientfd =  events[i].data.fd;
			if (clientfd == listenfd)
			{
				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;
				}

				printf("accept\n");
				ev.events = EPOLLIN;
				ev.data.fd = connfd;
				epoll_ctl(epfd, EPOLL_CTL_ADD, connfd, &ev);
			}
			else if (events[i].events & EPOLLIN)
			{
				//printf("recv\n");
				n = recv(clientfd, buff, MAXLNE, 0);
				if (n > 0)
				{
					buff[n] = '\0';
					printf("recv msg from client: %s\n", buff);

					send(clientfd, buff, n, 0);
				}
				else if (n == 0)
				{ //
					ev.events = EPOLLIN;
					ev.data.fd = clientfd;

					epoll_ctl(epfd, EPOLL_CTL_DEL, clientfd, &ev);

					close(clientfd);
					printf("client %d disconnect\n",clientfd);
				}
			}
		}
	}
    close(listenfd);
    return 0;
}