600字范文,内容丰富有趣,生活中的好帮手!
600字范文 > linux tcp 多路复用 linux下 C++ 使用 epoll 多路复用 实现高性能的tcpserver

linux tcp 多路复用 linux下 C++ 使用 epoll 多路复用 实现高性能的tcpserver

时间:2023-12-08 12:55:35

相关推荐

linux tcp 多路复用 linux下 C++ 使用 epoll 多路复用 实现高性能的tcpserver

linux系统中,实现socket多路复用的技术有select 、poll 、epoll 等多种方式。这些不同方式个有优缺点和适用场景,这不是本文讨论的重点,又兴趣的可以自己搜索学习一下。但是在高并发场景下, epoll 性能是最高的, Nginx 都听说过吧,大名鼎鼎的Nginx 底层用的就是 epoll。

这篇文章主要是写怎么用 epoll,而不是原理分析。这篇文章不是最全的,也不是最深入的,但是绝对是一篇能让普通人看懂的,看完能自己用epoll写出一个tcpserver的文章。全废话不多说,直接开始搞

首先明确一点,epoll 是linux系统提供的系统调用,也就说,epoll 在Windows系统上是没法使用的,相应的代码也是没法编译的。如果有人知道怎么在Windows中编译,请赐教。

工具

文中使用的开发环境

系统: Debian GNU/Linux 10 (buster)

linux内核: 4.19.0-14

gcc版本: 8.3.0

准备知识

epoll是linux内核提供的功能,这个功能对外提供系统调用,在C/C++中通过三个函数对用户提供功能

epoll_create(int __size) 创建一个epoll,_size 参数在linux2.6内核之后就没有什么作用了, 但是要>0,一般直接填 1 就好了。函数返回创建的epoll的文件描述符,如果创建失败,会返回 -1。

epoll_ctl(nt __epfd, int __op, int __fd,struct epoll_event *__event) 操作已有的epoll,epfdepoll的文件描述符;op操作方式,有添加、删除、修改等等;_fd 要操作对象的描述符,如果是操作tcp连接,也会就是这个连接的描述符。_event epoll 的响应事件,当epoll管理的tcp连接有事件发生时,会通过 _event 这个对象传递出来,所以在添加连接时,要把这个连接包装成一个 epoll_event 对象

op 类型

EPOLL_CTL_ADD 添加一个描述符

EPOLL_CTL_DEL 删除一个描述符

EPOLL_CTL_MOD 修改一个描述符

epoll_wait(int __epfd, struct epoll_event *__events,int __maxevents, int __timeout) 当epoll管理的连接中有响应事件发生时,会回调这个函数。epfdepoll的文件描述符;__events 可以操作的连接数组;__maxevents 一个可以处理的最大事件数量;__timeout 超时时间(单位毫秒),如果填-1,会直到有可操作事件发生时才会返回,因为C++不支持函数多返回值,像Go可以直接返回所有事件和数量了 (╥╯^╰╥)。

events 中的常用的类型:

EPOLLIN :表示对应的文件描述符可以读(SOCKET正常关闭)

EPOLLOUT:表示对应的文件描述符可以写

EPOLLPRI:表示对应的文件描述符有紧急的数据可读(表示有带外数据到来)

EPOLLERR:表示对应的文件描述符发生错误(默认注册)

EPOLLHUP:表示对应的文件描述符被挂断(默认注册)

EPOLLET: 将EPOLL设为边缘触发(Edge Triggered)模式,这是相对于水平触发(Level Triggered)来说的

EPOLLONESHOT:只监听一次事件,当监听完这次事件之后,如果还需要继续监听这个socket的话,需要再次把这个socket加入到EPOLL队列里

是不是很惊奇,这么牛逼的epoll就三个函数,第一次看到的时候我也觉得很奇怪,三个函数就能搞定那么复杂的事情。不过想想也是,把复杂的东西简化,才能体现出大神的实力

epoll的两种模式

epoll 事件有两种模型 Level Triggered (LT) 和 Edge Triggered (ET):

LT(level triggered,水平触发模式)是默认的工作方式,并且同时支持 block 和 non-block socket。在这种做法中,内核告诉你一个文件描述符是否就绪了,然后你可以对这个就绪的fd进行IO操作。如果你不作任何操作,内核还是会继续通知你的,所以,这种模式编程出错误可能性要小一点。

ET(edge-triggered,边缘触发模式)是高速工作方式,只支持no-block socket。在这种模式下,当描述符从未就绪变为就绪时,内核通过epoll告诉你。然后它会假设你知道文件描述符已经就绪,并且不会再为那个文件描述符发送更多的就绪通知,等到下次有新的数据进来的时候才会再次出发就绪事件。

把socket设置为非阻塞模式的方法

int flags = fcntl(fd, F_GETFL, 0);

fcntl(fd, F_SETFL, flags | O_NONBLOCK);

需要的头文件 #include

epoll原理

简单通过画图解释一下epoll的工作原理

这里没有涉及太底层的东西,因为太底层的我也没研究过,不敢乱讲。知之为知之,不知为不知。

epoll可以看做是一个由操作系统提供的容器,这个容器管理了一些 epoll_event (图中我画成单向链表了,实际上用的是红黑树,因为画树太麻烦了),这个event是我们添加进去的,event中设置了要响应的事件类型,当epoll 检测到具体的 event 有对应的事件发生时,会通过epoll_wait() 通知。

简单的epoll实现

#include //控制台输出

#include //创建socket

#include //socket addr

#include //epoll

#include //close函数

#include //设置非阻塞

using namespace std;

int main() {

const int EVENTS_SIZE = 20;

//读socket的数组

char buff[1024];

//创建一个tcp socket

int socketFd = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);

//设置socket监听的地址和端口

sockaddr_in sockAddr{};

sockAddr.sin_port = htons(8088);

sockAddr.sin_family = AF_INET;

sockAddr.sin_addr.s_addr = htons(INADDR_ANY);

//将socket和地址绑定

if (bind(socketFd, (sockaddr *) &sockAddr, sizeof(sockAddr)) == -1) {

cout << "bind error" << endl;

return -1;

}

//开始监听socket,当调用listen之后,

//进程就可以调用accept来接受一个外来的请求

//第二个参数,请求队列的长度

if (listen(socketFd, 10) == -1) {

cout << "listen error" << endl;

return -1;

}

//创建一个epoll,size已经不起作用了,一般填1就好了

int eFd = epoll_create(1);

//把socket包装成一个epoll_event对象

//并添加到epoll中

epoll_event epev{};

epev.events = EPOLLIN;//可以响应的事件,这里只响应可读就可以了

epev.data.fd = socketFd;//socket的文件描述符

epoll_ctl(eFd, EPOLL_CTL_ADD, socketFd, &epev);//添加到epoll中

//回调事件的数组,当epoll中有响应事件时,通过这个数组返回

epoll_event events[EVENTS_SIZE];

//整个epoll_wait 处理都要在一个死循环中处理

while (true) {

//这个函数会阻塞,直到超时或者有响应事件发生

int eNum = epoll_wait(eFd, events, EVENTS_SIZE, -1);

if (eNum == -1) {

cout << "epoll_wait" << endl;

return -1;

}

//遍历所有的事件

for (int i = 0; i < eNum; i++) {

//判断这次是不是socket可读(是不是有新的连接)

if (events[i].data.fd == socketFd) {

if (events[i].events & EPOLLIN) {

sockaddr_in cli_addr{};

socklen_t length = sizeof(cli_addr);

//接受来自socket连接

int fd = accept(socketFd, (sockaddr *) &cli_addr, &length);

if (fd > 0) {

//设置响应事件,设置可读和边缘(ET)模式

//很多人会把可写事件(EPOLLOUT)也注册了,后面会解释

epev.events = EPOLLIN | EPOLLET;

epev.data.fd = fd;

//设置连接为非阻塞模式

int flags = fcntl(fd, F_GETFL, 0);

if (flags < 0) {

cout << "set no block error, fd:" << fd << endl;

continue;

}

if (fcntl(fd, F_SETFL, flags | O_NONBLOCK) < 0) {

cout << "set no block error, fd:" << fd << endl;

continue;

}

//将新的连接添加到epoll中

epoll_ctl(eFd, EPOLL_CTL_ADD, fd, &epev);

cout << "client on line fd:" << fd << endl;

}

}

} else {//不是socket的响应事件

//判断是不是断开和连接出错

//因为连接断开和出错时,也会响应`EPOLLIN`事件

if (events[i].events & EPOLLERR || events[i].events & EPOLLHUP) {

//出错时,从epoll中删除对应的连接

//第一个是要操作的epoll的描述符

//因为是删除,所有event参数天null就可以了

epoll_ctl(eFd, EPOLL_CTL_DEL, events[i].data.fd, nullptr);

cout << "client out fd:" << events[i].data.fd << endl;

close(events[i].data.fd);

} else if (events[i].events & EPOLLIN) {//如果是可读事件

//如果在windows中,读socket中的数据要用recv()函数

int len = read(events[i].data.fd, buff, sizeof(buff));

//如果读取数据出错,关闭并从epoll中删除连接

if (len == -1) {

epoll_ctl(eFd, EPOLL_CTL_DEL, events[i].data.fd, nullptr);

cout << "client out fd:" << events[i].data.fd << endl;

close(events[i].data.fd);

} else {

//正常读取,打印读到的数据

cout << buff << endl;

//向客户端发数据

char a[] = "123456";

//如果在windows中,向socket中写数据要用send()函数

write(events[i].data.fd, a, sizeof(a));

}

}

}

}

}

}

常见的问题和注意事项在注释中,就单解释一下新连接注册事件的问题吧,很多文章中都会把可写事件也注册进去,像这样

sockaddr_in cli_addr{};

socklen_t length = sizeof(cli_addr);

//接受来自socket连接

int fd = accept(socketFd, (sockaddr *) &cli_addr, &length);

if (fd > 0) {

epev.events = EPOLLIN | EPOLLET | EPOLLOUT;

epev.data.fd = fd;

epoll_ctl(eFd, EPOLL_CTL_ADD, fd, &epev);

cout << "client on line fd:" << fd << endl;

}

但是经过测试,不注册可写事件,直接往socket中写也是可以的

经过查资料得知:

EPOLLIN : 如果状态改变了(比如 从无到有),只要输入缓冲区可读就会触发

EPOLLOUT : 如果状态改变了(比如 从满到不满),只要输出缓冲区可写就会触

如果把可写也注册上,会频繁回调,这里会有很多无用的回调,导致性能下降。

有一种思路,当向socket写失败后(write函数返回值 == -1),注册上 EPOLLOUT 当响应了可写事件后,重新往socket中写数据,写成功后,再取消掉 EPOLLOUT。 这里就不给出示例了

客户端测试

这次关注的是服务端实现,客户端就不用C++写了,用Go写了一个client(没别的原因,只是因为Go写起了简单)

package main

import (

"fmt"

"net"

"sync"

"time"

)

const (

MAX_CONN = 10

)

func main() {

var wg sync.WaitGroup

wg.Add(1)

for i := 0; i < MAX_CONN; i++ {

go Conn("192.168.199.164:8088", i)

time.Sleep(time.Millisecond * 100)

}

wg.Wait()

}

func Conn(addr string, id int) {

conn, err := net.Dial("tcp", addr)

if err != nil {

fmt.Println(err)

return

}

fmt.Println("connect ", id)

go func() {

buf := make([]byte, 1024)

for {

n, err := conn.Read(buf)

if err != nil {

break

}

fmt.Println(id, "read: ", string(buf[:n]))

}

}()

time.Sleep(time.Second * 1)

for {

_, err := conn.Write([]byte("hello"))

if err != nil {

break

}

time.Sleep(time.Second * 10)

}

}

这只是一个测试用的,写的很粗糙,但是不影响使用

服务端打印信息

服务端

客户端打印信息

客户端

总结

epoll是socket多路复用技术的一种,还有select和poll

epoll 只能在linux使用(Windows下怎么用我没找到,如果说错了请指正)

epoll 事件有 Level Triggered (LT) 和 Edge Triggered (ET) 两种模型,LT是默认模式,ET是高性能模式

另外,我使用面向对象的方式封装了一个epoll的tcpserver 代码有点多,就不贴在这了,已经上传

github

码云

欢迎给点个star ヾ(o◕∀◕)ノヾ

本内容不代表本网观点和政治立场,如有侵犯你的权益请联系我们处理。
网友评论
网友评论仅供其表达个人看法,并不表明网站立场。