Linux网络编程——tcp并发服务器（epoll实现）

https://blog.csdn.net/lianghe_work/article/details/46551871

通过epoll实现tcp并发回执服务器（客户端给服务器发啥，服务器就给客户端回啥）

代码如下：

#include <string.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <sys/select.h>
#include <sys/time.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <sys/epoll.h>
#include <errno.h>
#define OPEN_MAX 100

int main(int argc, char *argv[])
{
struct epoll_event event; // 告诉内核要监听什么事件
struct epoll_event wait_event; //内核监听完的结果

//1.创建tcp监听套接字
int sockfd = socket(AF_INET, SOCK_STREAM, 0);

//2.绑定sockfd
struct sockaddr_in my_addr;
bzero(&my_addr, sizeof(my_addr));
my_addr.sin_family = AF_INET;
my_addr.sin_port = htons( 8001);
my_addr.sin_addr.s_addr = htonl(INADDR_ANY);
bind(sockfd, ( struct sockaddr *)&my_addr, sizeof(my_addr));

//3.监听listen
listen(sockfd, 10);

//4.epoll相应参数准备
int fd[OPEN_MAX];
int i = 0, maxi = 0;
memset(fd, -1, sizeof(fd));
fd[ 0] = sockfd;

int epfd = epoll_create( 10); // 创建一个 epoll 的句柄，参数要大于 0， 没有太大意义
if( -1 == epfd ){
perror ( "epoll_create");
return -1;
}

event.data.fd = sockfd; //监听套接字
event.events = EPOLLIN; // 表示对应的文件描述符可以读

//5.事件注册函数，将监听套接字描述符 sockfd 加入监听事件
int ret = epoll_ctl(epfd, EPOLL_CTL_ADD, sockfd, & event);
if( -1 == ret){
perror( "epoll_ctl");
return -1;
}

//6.对已连接的客户端的数据处理
while( 1)
{
// 监视并等待多个文件（标准输入，udp套接字）描述符的属性变化（是否可读）
// 没有属性变化，这个函数会阻塞，直到有变化才往下执行，这里没有设置超时
ret = epoll_wait(epfd, &wait_event, maxi+ 1, -1);

//6.1监测sockfd(监听套接字)是否存在连接
if(( sockfd == wait_event.data.fd )
&& ( EPOLLIN == wait_event.events & EPOLLIN ) )
{
struct sockaddr_in cli_addr;
int clilen = sizeof(cli_addr);

//6.1.1 从tcp完成连接中提取客户端
int connfd = accept(sockfd, ( struct sockaddr *)&cli_addr, &clilen);

//6.1.2 将提取到的connfd放入fd数组中，以便下面轮询客户端套接字
for(i= 1; i<OPEN_MAX; i++)
{
if(fd[i] < 0)
{
fd[i] = connfd;
event.data.fd = connfd; //监听套接字
event.events = EPOLLIN; // 表示对应的文件描述符可以读

//6.1.3.事件注册函数，将监听套接字描述符 connfd 加入监听事件
ret = epoll_ctl(epfd, EPOLL_CTL_ADD, connfd, & event);
if( -1 == ret){
perror( "epoll_ctl");
return -1;
}

break;
}
}

//6.1.4 maxi更新
if(i > maxi)
maxi = i;

//6.1.5 如果没有就绪的描述符，就继续epoll监测，否则继续向下看
if(--ret <= 0)
continue;
}

//6.2继续响应就绪的描述符
for(i= 1; i<=maxi; i++)
{
if(fd[i] < 0)
continue;

if(( fd[i] == wait_event.data.fd )
&& ( EPOLLIN == wait_event.events & (EPOLLIN|EPOLLERR) ))
{
int len = 0;
char buf[ 128] = "";

//6.2.1接受客户端数据
if((len = recv(fd[i], buf, sizeof(buf), 0)) < 0)
{
if(errno == ECONNRESET) //tcp连接超时、RST
{
close(fd[i]);
fd[i] = -1;
}
else
perror( "read error:");
}
else if(len == 0) //客户端关闭连接
{
close(fd[i]);
fd[i] = -1;
}
else //正常接收到服务器的数据
send(fd[i], buf, len, 0);

//6.2.2所有的就绪描述符处理完了，就退出当前的for循环，继续poll监测
if(--ret <= 0)
break;
}
}
}
return 0;
}

运行结果：

源码下载：