驱动 epoll实现并发服务器

#include<string.h>	
#include<stdio.h>
#include<stdlib.h>
#include<sys/socket.h>
#include <sys/epoll.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <fcntl.h>

#define ERR_MSG(msg) do{\
	printf("line:%d\n",__LINE__);\
	perror(msg);\
	}while(0);

#define PORT 6666  //端口号的网络字节序
#define IP "192.168.250.100"   //本机IP

int main(int argc, const char *argv[])
{
	//1.创建报式套接字
	int sfd=socket(AF_INET, SOCK_STREAM, 0);
	if(sfd < 0)
	{
		ERR_MSG("socket");
		return -1;
	}

	//2.创建句柄
	int epfd = epoll_create(1);
	if(epfd < 0)
	{
		printf("epoll_create filed\n");
		exit(-1);
	}

	//3.添加事件
	struct epoll_event event;
	struct epoll_event events[10];  //存放就绪事件描述符的数组
	event.events=EPOLLIN; //读事件 边沿检测
	event.data.fd=sfd;
	if(epoll_ctl(epfd, EPOLL_CTL_ADD, sfd, &event) < 0)
	{
		printf("epoll_ctl add filed\n");
	}

    int reuse = 1;
	if(setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) <0)
	{
		perror("setsockopt");
		return -1;
	}


	//4.绑定
		//填充服务器的地址信息
	struct sockaddr_in sin;
	sin.sin_family = AF_INET;
	sin.sin_port = htons(PORT);
	sin.sin_addr.s_addr = inet_addr(IP);

	if(bind(sfd, (struct sockaddr*)&sin, sizeof(sin)) < 0)
	{
		ERR_MSG("bind");
		return -1;
	}

	//5.监听
	if(listen(sfd, 10) < 0)
	{
		ERR_MSG("listen");
		return -1;
	}

	int newfd = -1;
	struct sockaddr_in cin;  //存储客户端信息
	socklen_t len = sizeof(cin);
	char buf[128] = "";
	ssize_t res =0;
	printf("准备就绪\n");
	while(1)
	{
		//6.询问检测事件是否发生
		
		int ret = epoll_wait(epfd,events,10,-1);
		if(ret < 0)
		{
			printf("epoll_wait filed\n");
			exit(-1);
		}
		
		//7.循环遍历数组，做出事件的处理
		int i;

		for(i=0; i<10; i++)
		{
			if(sfd == events[i].data.fd)  //判断文件描述符
			{
			
			//if(EPOLLIN == events[i].events & EPOLLIN) //判断文件描述符类型
			//{
					//客户连接事件
					
					newfd = accept(events[i].data.fd, (struct sockaddr*)&cin, &len);
					if(newfd < 0)
					{
						ERR_MSG("accept");
						return -1;
					}
					printf("IP:%s 连接成功\n",inet_ntoa(cin.sin_addr));

					//将新用户连接到红黑树
					struct epoll_event event;
					struct epoll_event events[10];  //存放就绪事件描述符的数组
					event.events=EPOLLIN; //读事件 边沿检测
					event.data.fd=newfd;
					if(epoll_ctl(epfd, EPOLL_CTL_ADD, newfd, &event) < 0)
					{
						printf("epoll_ctl add filed\n");
					}
			//}	
			}
            else
			{
				//老客户信息
				//if(EPOLLIN == events[i].events & EPOLLIN) //判断文件描述符类型
				//{
					memset(buf, 0, sizeof(buf));
					res = recv(newfd, buf, sizeof(buf), 0);
					if(res < 0)
					{
						ERR_MSG("recv");
						return -1;
					}
					else if(0 == res)
					{
						//客户断开连接
						printf("%d 断开连接\n",events[i].data.fd);
						//关闭文件描述符
						close(events[i].data.fd);
						//移出红黑树
						epoll_ctl(epfd, EPOLL_CTL_DEL, events[i].data.fd, NULL);
					}
					else
					{
						//接收信息
						printf("buf:%s\n", buf);
					}
					//将接受的信息发送回去
					strcat(buf, "-yidu");
					if(send(newfd, buf, sizeof(buf), 0) <0)
					{
						ERR_MSG("send");
						return -1;
					}
				//}
			}
		}
	}

	//关闭套接字
	close(sfd);



	return 0;	
}