mtcp 是一个用户态的 tcp 协议栈,结合 dpdk 可以实现高性能的收发包。mtcp 不支持 udp 协议,想要在 bind 里利用 mtcp 进行加速,需要改动源码以提供支持。
mtcp 项目地址:https://github.com/mtcp-stack/mtcp
1、修改配置文件支持UDP
1.1、修改数据结果struct mtcp_config,增加对UDP协议支持标记变量。
1.2、在core.c文件中的mtcp_init函数中,会调用config.c文件中的LoadConfiguration函数来读取配置文件,在配置文件中也加入协议标记。
1.3、修改struct mtcp_manager,增加协议区分。
2、在tcp_in.h、tcp_in.c相同路径下,建立udp_in.h、udp_in.cpp
2.1、在udp_in.c文件中,实现ProcessUDPPacket函数。
2.2、在ip_in.c文件中的case IPPROTO_TCP:下面,增加case IPPROTO_UDP:并且调用ProcessUDPPacket。
2.3、因为UDP协议没有像TCP协议的11种状态和三次握手,所以实现起来比较简单。
2.4、调用过程为RunMainLoop->ProcessPacket->ProcessIPv4Packet->ProcessUDPPacket;
RunMainLoop->recv_pkts->dpdk_recv_pkts->rte_eth_rx_burst
3、在tcp_out.h、tcp_out.c相同路径下,建立udp_out.h、udp_out.c
3.1、在udp_out.c中,实现WriteUDPDataList函数,这个函数应该在core.c的WritePacketsToChunks函数中调用。WritePacketsToChunks函数默认都是TCP协议,应该增加对协议的判断。
3.2、在udp_out.c中,实现SendUDPPacket函数,这个函数被WriteUDPDataList调用。
3.3、在ip_out.c中,修改IPOutput函数,需要增加对UDP协议包组包的支持。
3.4、调用过程为RunMainLoop->WritePacketsToChunks->WriteUDPDataList->SendUDPPacket;
RunMainLoop->send_pkts->dpdk_send_pkts->rte_eth_tx_burst
4、在tcp_util.c文件中,增加UDPCalcChecksum函数,分别在tcp_in.c的ProcessUDPPacket函数和tcp_out.c的SendUDPPacket中被调用。
具体增加的函数比较简单,在这里就不写了。按照上述修改,可以支持单独UDP,也可以同时支持TCP和UDP。