devbox
weixin_40725706
这个屌丝很懒，什么也没留下！
关注作者
热门标签
热门文章
当前位置:   article > 正文

【RDMA项目】如何使用rdma-core进行调用开发一个实战项目_安装rdma-core

作者：weixin_40725706 | 2024-08-07 22:56:15

赞

踩

安装rdma-core

RDMA (Remote Direct Memory Access) 是一种网络协议，可以在计算节点之间实现高效的内存数据传输，而无需CPU的干预。rdma-core 是 RDMA 的一个用户空间库，提供了一些简单易用的接口来使用 RDMA 功能。

开发了一套高级 RDMA（远程直接内存访问）连接和数据传输系统，使用 rdma-core 库（包括 rdma_cm 和 ibverbs）编写，适用于高性能计算和实时数据处理。该项目分为服务器和客户端两部分，能够在实际网络环境中实现 RDMA 连接和数据传输。通过零拷贝技术和高效的网络编程，系统显著提高了数据传输速度和减少了延迟。项目成功验证了 RDMA 在高性能和低延迟场景中的应用价值，为需要高效数据传输的应用场景提供了可靠的解决方案。

在这个项目中，rdma_cm（RDMA Connection Manager）用于管理RDMA连接的创建、绑定、监听、接受和断开。rdma_cm 提供了一套API，使得RDMA编程变得更为简便，抽象了底层复杂的连接管理过程。

在这个项目中，ibverbs（Infiniband Verbs API）用于管理低级别的Infiniband操作，如保护域、内存区域、完成队列和队列对（QP）的分配和管理。​​​​​​​

基础知识参考博文：

一文带你了解什么是RDMA

RDMA 高性能架构基本原理与设计方案

 RDMA之RoCE & Soft-RoCE_software roce

如何使用rdma-core来实现RDMA操作

教你在虚拟机上把普通网卡配置成softroce设备来运行rdma-core中的示例程序-CSDN博客

目录：

目录：

一、环境准备：

 1.1 安装依赖

1.2 安装 rdma-core

1.3 支持RDMA的机器

二、使用 RDMA（主要用于测试RDMA 环境配置是否正确）

2.1 编写 RDMA 程序

2.2 编译和运行程序

三、进一步的 RDMA-core 操作示例

3.1 服务器代码

3.1.1 初始化和准备工作

3.1.2 设置和监听

3.1.3 接受连接

3.1.4 资源分配和QP创建

3.1.5 内存注册和接收操作

3.1.6 轮询和处理完成队列

3.1.7 清理资源

3.1.8 主函数

3.2 客户端代码

3.2.1 初始化和准备工作

3.2.2 创建和解析连接

3.2.3 路由解析

3.2.4 资源分配和QP创建

3.2.5 内存注册与连接

3.2.6 数据发送

3.2.7 清理资源

3.2.8  主函数：

3.3 编译和运行

 3.5 结果

四、rdma-core 实现更复杂的 RDMA 连接和数据传输操作

4.1 RDMA 操作基础

4.2 设备发现与上下文创建

4.1.1 变量定义

4.1.2 获取设备列表 ：

4.1.3 选择设备：

4.1.4 获取设备上下文

4.1.5 打印设备名称并且清理资源

4.1.6 编译运行：

4.1.7 示例输出：

4.3 内存注册与队列对创建

4.3.1 获取InfiniBand设备列表并打开设备:（此过程就是4.2介绍的过程）

4.3.2 创建保护域（Protection Domain，PD）:

4.3.3 注册内存区域:

4.3.4 创建完成队列（Completion Queue，CQ）: 

4.3.5  设置队列对属性并创建队列对:

4.3.6 清理资源:

4.3.7 演示效果

4.4 建立 RDMA 连接

 1. 服务器端代码（与3.1代码略有不同）

2. 客户端代码（与3.2代码相同）

4.5 编译和运行

 4.6 结果

五、验证RDMA 在高性能和低延迟场景中的应用价值

5.1 性能测试方法

5.1.2 延迟测试：

5.2.3 零拷贝效应：

5.2.4 资源使用率分析：

5.2  实验设计

5.2.1  测试环境搭建：

5.2.2 基准测试：

5.2.3 压力测试：

5.3 测试指标

5.3.1 吞吐量（Throughput）：

5.3.2 延迟（Latency）：

5.3.3 资源使用率（Resource Utilization）：

5.4 实验结果分析

5.5 实验示例

一、环境准备：

 1.1 安装依赖

在安装 rdma-core 之前，确保你的系统已经安装了相关依赖库。对于基于 Debian 的系统，可以使用以下命令：

  1. sudo apt-get update
  2. sudo apt-get install build-essential cmake libnuma-dev libibverbs-dev

对于基于 Red Hat 的系统，可以使用以下命令：

  1. sudo yum groupinstall "Development Tools"
  2. sudo yum install cmake libibverbs-devel numactl-devel

1.2 安装 rdma-core

你可以从源码编译并安装 rdma-core。首先克隆 rdma-core 的 GitHub 仓库：

  1. git clone https://github.com/linux-rdma/rdma-core.git
  2. cd rdma-core

然后使用 CMake 进行编译和安装：

  1. mkdir build
  2. cd build
  3. cmake ..
  4. make
  5. sudo make install

1.3 支持RDMA的机器

 准备俩台支持RDMA的设备机器进行测试调试

二、使用 RDMA（主要用于测试RDMA 环境配置是否正确）

2.1 编写 RDMA 程序

rdma-core 提供了多个库，例如 libibverbs 和 librdmacm，用于不同的 RDMA 操作。以下是一个简单的例子，展示了如何使用 libibverbs 库进行基本的 RDMA 操作。

  1. #include <stdio.h>
  2. #include <stdlib.h>
  3. #include <string.h>
  4. #include <infiniband/verbs.h>
  5.  
  6. int main() {
  7.     struct ibv_device **dev_list;  // 用于存储设备列表的指针数组
  8.     struct ibv_device *ib_dev;     // 用于存储选择的设备指针
  9.     struct ibv_context *ctx;       // 用于存储设备上下文的指针
  10.  
  11.     // 获取设备列表
  12.     dev_list = ibv_get_device_list(NULL);
  13.     if (!dev_list) {  // 检查设备列表是否获取成功
  14.         perror("Failed to get devices list");
  15.         return EXIT_FAILURE;
  16.     }
  17.  
  18.     // 选择第一个设备
  19.     ib_dev = dev_list[0];
  20.     if (!ib_dev) {  // 检查是否找到至少一个设备
  21.         fprintf(stderr, "No IB devices found\n");
  22.         ibv_free_device_list(dev_list);  // 释放设备列表
  23.         return EXIT_FAILURE;
  24.     }
  25.  
  26.     // 获取设备的上下文
  27.     ctx = ibv_open_device(ib_dev);
  28.     if (!ctx) {  // 检查是否成功打开设备上下文
  29.         perror("Failed to open device");
  30.         ibv_free_device_list(dev_list);  // 释放设备列表
  31.         return EXIT_FAILURE;
  32.     }
  33.  
  34.     printf("Device %s opened\n", ib_dev->name);  // 输出设备名称表示成功打开
  35.  
  36.     // 清理
  37.     ibv_close_device(ctx);  // 关闭设备上下文
  38.     ibv_free_device_list(dev_list);  // 释放设备列表
  39.  
  40.     return EXIT_SUCCESS;  // 程序成功执行完毕
  41. }

2.2 编译和运行程序

将上述代码保存到一个文件中，例如 rdma_example.c，然后使用以下命令进行编译和运行：

  1. gcc -o rdma_example rdma_example.c -libverbs
  2. ./rdma_example

恭喜你成功打开了 RDMA 设备 rxe0！这说明你的 RDMA 环境已经配置正确，能够正常工作。接下来，可以尝试更多复杂的 RDMA 操作，如建立连接、数据传输等。以下是一个更复杂的示例，展示如何进行基本的 RDMA 数据传输。

三、进一步的 RDMA-core 操作示例

我们将展示一个简单的 RDMA 客户端-服务器程序，其中客户端向服务器发送数据，服务器接收数据。这只是一个简单的示例，实际应用中可能需要更多的错误处理和资源管理。

3.1 服务器代码

保存以下代码为 rdma_server.c：

实现了一种使用RDMA（远程直接数据存取）技术的简单服务器程序。RDMA允许在计算机之间进行高速数据传输，而无需操作系统干预。

  1. #include <stdio.h>
  2. #include <stdlib.h>
  3. #include <string.h>
  4. #include <rdma/rdma_cma.h>
  5. #include <infiniband/verbs.h>
  6.  
  7. #define BUFFER_SIZE 1024
  8.  
  9. void run_server() {
  10.     struct rdma_event_channel *ec = rdma_create_event_channel(); // 创建RDMA事件通道
  11.     struct rdma_cm_id *listener = NULL, *conn = NULL; // 定义监听和连接ID
  12.     struct rdma_addrinfo hints, *res; // 定义地址信息结构体
  13.     struct ibv_pd *pd = NULL; // 定义保护域
  14.     struct ibv_mr *mr = NULL; // 定义内存注册区域
  15.     struct ibv_comp_channel *comp_chan = NULL; // 定义完成通道
  16.     struct ibv_cq *cq = NULL; // 定义完成队列
  17.     struct ibv_qp_init_attr qp_attr; // 定义QP初始化属性
  18.     char buf[BUFFER_SIZE]; // 定义缓冲区
  19.     int ret; // 定义返回值变量
  20.  
  21.     memset(&hints, 0, sizeof(hints)); // 将hints结构体清零
  22.     hints.ai_flags = RAI_PASSIVE; // 设置hints为被动模式
  23.     hints.ai_port_space = RDMA_PS_TCP; // 设置地址空间为TCP
  24.  
  25.     ret = rdma_getaddrinfo(NULL, "20079", &hints, &res); // 获取地址信息
  26.     if (ret) {
  27.         perror("rdma_getaddrinfo"); // 如果失败，输出错误信息并退出
  28.         exit(EXIT_FAILURE);
  29.     }
  30.  
  31.     ret = rdma_create_id(ec, &listener, NULL, RDMA_PS_TCP); // 创建RDMA标识符
  32.     if (ret) {
  33.         perror("rdma_create_id"); // 如果失败，输出错误信息并退出
  34.         exit(EXIT_FAILURE);
  35.     }
  36.  
  37.     ret = rdma_bind_addr(listener, res->ai_src_addr); // 绑定地址
  38.     if (ret) {
  39.         perror("rdma_bind_addr"); // 如果失败，输出错误信息并退出
  40.         exit(EXIT_FAILURE);
  41.     }
  42.  
  43.     ret = rdma_listen(listener, 0); // 开始监听
  44.     if (ret) {
  45.         perror("rdma_listen"); // 如果失败，输出错误信息并退出
  46.         exit(EXIT_FAILURE);
  47.     }
  48.  
  49.     printf("Server is listening on port 20079...\n"); // 打印服务器正在监听的消息
  50.  
  51.     struct rdma_cm_event *event; // 定义RDMA事件
  52.     ret = rdma_get_cm_event(ec, &event); // 获取一个连接管理事件
  53.     if (ret) {
  54.         perror("rdma_get_cm_event"); // 如果失败，输出错误信息并退出
  55.         exit(EXIT_FAILURE);
  56.     }
  57.  
  58.     if (event->event == RDMA_CM_EVENT_CONNECT_REQUEST) { // 检查事件类型是否为连接请求
  59.         conn = event->id; // 获取连接ID
  60.         rdma_ack_cm_event(event); // 确认事件
  61.     } else {
  62.         fprintf(stderr, "Unexpected event: %d\n", event->event); // 如果事件类型不是预期的，输出错误信息并退出
  63.         rdma_ack_cm_event(event); // 确认事件
  64.         exit(EXIT_FAILURE);
  65.     }
  66.  
  67.     pd = ibv_alloc_pd(conn->verbs); // 分配保护域
  68.     if (!pd) {
  69.         perror("ibv_alloc_pd"); // 如果失败，输出错误信息并退出
  70.         exit(EXIT_FAILURE);
  71.     }
  72.  
  73.     comp_chan = ibv_create_comp_channel(conn->verbs); // 创建完成通道
  74.     if (!comp_chan) {
  75.         perror("ibv_create_comp_channel"); // 如果失败，输出错误信息并退出
  76.         exit(EXIT_FAILURE);
  77.     }
  78.  
  79.     cq = ibv_create_cq(conn->verbs, 10, NULL, comp_chan, 0); // 创建完成队列
  80.     if (!cq) {
  81.         perror("ibv_create_cq"); // 如果失败，输出错误信息并退出
  82.         exit(EXIT_FAILURE);
  83.     }
  84.  
  85.     memset(&qp_attr, 0, sizeof(qp_attr)); // 清零QP初始化属性结构体
  86.     qp_attr.cap.max_send_wr = 1; // 设置最大发送工作请求数
  87.     qp_attr.cap.max_recv_wr = 1; // 设置最大接收工作请求数
  88.     qp_attr.cap.max_send_sge = 1; // 设置最大发送SGE数
  89.     qp_attr.cap.max_recv_sge = 1; // 设置最大接收SGE数
  90.     qp_attr.send_cq = cq; // 关联发送完成队列
  91.     qp_attr.recv_cq = cq; // 关联接收完成队列
  92.     qp_attr.qp_type = IBV_QPT_RC; // 设置QP类型为RC
  93.  
  94.     ret = rdma_create_qp(conn, pd, &qp_attr); // 创建QP
  95.     if (ret) {
  96.         perror("rdma_create_qp"); // 如果失败，输出错误信息并退出
  97.         exit(EXIT_FAILURE);
  98.     }
  99.  
  100.     mr = ibv_reg_mr(pd, buf, BUFFER_SIZE, IBV_ACCESS_LOCAL_WRITE | IBV_ACCESS_REMOTE_WRITE); // 注册内存区域
  101.     if (!mr) {
  102.         perror("ibv_reg_mr"); // 如果失败，输出错误信息并退出
  103.         exit(EXIT_FAILURE);
  104.     }
  105.  
  106.     struct ibv_recv_wr wr, *bad_wr = NULL; // 定义接收工作请求和错误工作请求指针
  107.     struct ibv_sge sge; // 定义SGE
  108.  
  109.     sge.addr = (uintptr_t)buf; // 设置SGE地址为缓冲区地址
  110.     sge.length = BUFFER_SIZE; // 设置SGE长度为缓冲区大小
  111.     sge.lkey = mr->lkey; // 设置SGE的本地密钥为内存区域的密钥
  112.  
  113.     wr.wr_id = 0; // 设置工作请求ID为0
  114.     wr.next = NULL; // 设置下一个工作请求为空
  115.     wr.sg_list = &sge; // 设置工作请求的SGE列表
  116.     wr.num_sge = 1; // 设置SGE数量为1
  117.  
  118.     ret = ibv_post_recv(conn->qp, &wr, &bad_wr); // 发起接收工作请求
  119.     if (ret) {
  120.         perror("ibv_post_recv"); // 如果失败，输出错误信息并退出
  121.         exit(EXIT_FAILURE);
  122.     }
  123.  
  124.     struct rdma_conn_param cm_params; // 定义连接参数
  125.     memset(&cm_params, 0, sizeof(cm_params)); // 清零连接参数结构体
  126.     ret = rdma_accept(conn, &cm_params); // 接受连接请求
  127.     if (ret) {
  128.         perror("rdma_accept"); // 如果失败，输出错误信息并退出
  129.         exit(EXIT_FAILURE);
  130.     }
  131.  
  132.     struct ibv_wc wc; // 定义工作完成结构体
  133.     while ((ret = ibv_poll_cq(cq, 1, &wc)) == 0); // 轮询完成队列直到有工作完成
  134.     if (ret < 0) {
  135.         perror("ibv_poll_cq"); // 如果失败，输出错误信息并退出
  136.         exit(EXIT_FAILURE);
  137.     }
  138.  
  139.     if (wc.status != IBV_WC_SUCCESS) { // 检查工作完成状态是否成功
  140.         fprintf(stderr, "Failed status %s (%d) for wr_id %d\n",
  141.                 ibv_wc_status_str(wc.status), wc.status, (int) wc.wr_id); // 如果失败，输出错误信息并退出
  142.         exit(EXIT_FAILURE);
  143.     }
  144.  
  145.     printf("Received message: %s\n", buf); // 打印接收到的消息
  146.  
  147.     rdma_disconnect(conn); // 断开连接
  148.     rdma_destroy_qp(conn); // 销毁QP
  149.     ibv_dereg_mr(mr); // 解注册内存区域
  150.     ibv_destroy_cq(cq); // 销毁完成队列
  151.     ibv_destroy_comp_channel(comp_chan); // 销毁完成通道
  152.     ibv_dealloc_pd(pd); // 释放保护域
  153.     rdma_destroy_id(conn); // 销毁连接ID
  154.     rdma_destroy_id(listener); // 销毁监听ID
  155.     rdma_destroy_event_channel(ec); // 销毁事件通道
  156.     rdma_freeaddrinfo(res); // 释放地址信息
  157. }
  158.  
  159. int main() {
  160.     run_server(); // 运行服务器
  161.     return 0; // 返回0表示正常退出
  162. }

3.1.1 初始化和准备工作

1. 创建并初始化RDMA事件通道：

struct rdma_event_channel *ec = rdma_create_event_channel(); // 创建RDMA事件通道

2. 定义变量：

  1.     struct rdma_cm_id *listener = NULL, *conn = NULL; // 定义监听和连接ID
  2.     struct rdma_addrinfo hints, *res; // 定义地址信息结构体
  3.     struct ibv_pd *pd = NULL; // 定义保护域
  4.     struct ibv_mr *mr = NULL; // 定义内存注册区域
  5.     struct ibv_comp_channel *comp_chan = NULL; // 定义完成通道
  6.     struct ibv_cq *cq = NULL; // 定义完成队列
  7.     struct ibv_qp_init_attr qp_attr; // 定义QP初始化属性
  8.     char buf[BUFFER_SIZE]; // 定义缓冲区
  9.     int ret; // 定义返回值变量
  • listener和conn：用于监听和连接的RDMA标识符。
  • hints和res：用于存储地址信息。
  • pd、mr、comp_chan和cq：分别代表保护域、内存注册区、完成通道和完成队列。
  • qp_attr：用于配置QP的初始化属性。
  • buf：用于存储接收到的数据。
  • ret：用于存储各个操作的返回值。

3. 配置地址信息：

  1.     memset(&hints, 0, sizeof(hints)); // 将hints结构体清零
  2.     hints.ai_flags = RAI_PASSIVE; // 设置hints为被动模式
  3.     hints.ai_port_space = RDMA_PS_TCP; // 设置地址空间为TCP

4. 获取RDMA地址：

    ret = rdma_getaddrinfo(NULL, "20079", &hints, &res); // 获取地址信息

3.1.2 设置和监听

1. 创建RDMA标识符：

    ret = rdma_create_id(ec, &listener, NULL, RDMA_PS_TCP); // 创建RDMA标识符

2. 绑定地址并开始监听：

  1.     ret = rdma_bind_addr(listener, res->ai_src_addr);//绑定地址
  2.     ret = rdma_listen(listener, 0);//开始监听

3.1.3 接受连接

1. 等待并接受连接请求：

  1.     struct rdma_cm_event *event;//定义RDMA事件
  2.     ret = rdma_get_cm_event(ec, &event);//获取一个连接管理事件

2.. 处理连接请求：

  1.     if (event->event == RDMA_CM_EVENT_CONNECT_REQUEST) { // 检查事件类型是否为连接请求
  2.         conn = event->id; // 获取连接ID
  3.         rdma_ack_cm_event(event); // 确认事件
  4.     } else {
  5.         fprintf(stderr, "Unexpected event: %d\n", event->event); // 如果事件类型不是预期的，输出错误信息并退出
  6.         rdma_ack_cm_event(event); // 确认事件
  7.         exit(EXIT_FAILURE);
  8.     }

3.1.4 资源分配和QP创建

1. 分配保护域、创建完成通道和完成队列：

  1.     pd = ibv_alloc_pd(conn->verbs); // 分配保护域
  2.     comp_chan = ibv_create_comp_channel(conn->verbs); // 创建完成通道
  3.     cq = ibv_create_cq(conn->verbs, 10, NULL, comp_chan, 0); // 创建完成队列
  4.

2. 配置并创建QP：

  1.     memset(&qp_attr, 0, sizeof(qp_attr)); // 清零QP初始化属性结构体
  2.     qp_attr.cap.max_send_wr = 1; // 设置最大发送工作请求数
  3.     qp_attr.cap.max_recv_wr = 1; // 设置最大接收工作请求数
  4.     qp_attr.cap.max_send_sge = 1; // 设置最大发送SGE数
  5.     qp_attr.cap.max_recv_sge = 1; // 设置最大接收SGE数
  6.     qp_attr.send_cq = cq; // 关联发送完成队列
  7.     qp_attr.recv_cq = cq; // 关联接收完成队列
  8.     qp_attr.qp_type = IBV_QPT_RC; // 设置QP类型为RC
  9.  
  10.     ret = rdma_create_qp(conn, pd, &qp_attr); // 创建QP

3.1.5 内存注册和接收操作

1. 注册内存区域：

    mr = ibv_reg_mr(pd, buf, BUFFER_SIZE, IBV_ACCESS_LOCAL_WRITE | IBV_ACCESS_REMOTE_WRITE);//注册内存区域

2. 准备接收工作请求：

  1.     struct ibv_recv_wr wr, *bad_wr = NULL; // 定义接收工作请求和错误工作请求指针
  2.     struct ibv_sge sge; // 定义SGE
  3.  
  4.     sge.addr = (uintptr_t)buf; // 设置SGE地址为缓冲区地址
  5.     sge.length = BUFFER_SIZE; // 设置SGE长度为缓冲区大小
  6.     sge.lkey = mr->lkey; // 设置SGE的本地密钥为内存区域的密钥
  7.  
  8.     wr.wr_id = 0; // 设置工作请求ID为0
  9.     wr.next = NULL; // 设置下一个工作请求为空
  10.     wr.sg_list = &sge; // 设置工作请求的SGE列表
  11.     wr.num_sge = 1; // 设置SGE数量为1
  12.  
  13.     ret = ibv_post_recv(conn->qp, &wr, &bad_wr); // 发起接收工作请求

3. 接受连接：

  1.     struct rdma_conn_param cm_params; // 定义连接参数
  2.     memset(&cm_params, 0, sizeof(cm_params)); // 清零连接参数结构体
  3.     ret = rdma_accept(conn, &cm_params); // 接受连接请求

3.1.6 轮询和处理完成队列

1. 轮询完成队列，等待数据接收：

  1.     struct ibv_wc wc; // 定义工作完成结构体
  2.     while ((ret = ibv_poll_cq(cq, 1, &wc)) == 0); // 轮询完成队列直到有工作完成

2. 检查完成状态并处理数据：

  1.     if (wc.status != IBV_WC_SUCCESS) { // 检查工作完成状态是否成功
  2.         fprintf(stderr, "Failed status %s (%d) for wr_id %d\n",
  3.                 ibv_wc_status_str(wc.status), wc.status, (int) wc.wr_id); // 如果失败，输出错误信息并退出
  4.         exit(EXIT_FAILURE);
  5.     }

3.1.7 清理资源

断开连接并释放资源：

  1.     rdma_disconnect(conn); // 断开连接
  2.     rdma_destroy_qp(conn); // 销毁QP
  3.     ibv_dereg_mr(mr); // 解注册内存区域
  4.     ibv_destroy_cq(cq); // 销毁完成队列
  5.     ibv_destroy_comp_channel(comp_chan); // 销毁完成通道
  6.     ibv_dealloc_pd(pd); // 释放保护域
  7.     rdma_destroy_id(conn); // 销毁连接ID
  8.     rdma_destroy_id(listener); // 销毁监听ID
  9.     rdma_destroy_event_channel(ec); // 销毁事件通道
  10.     rdma_freeaddrinfo(res); // 释放地址信息

3.1.8 主函数

运行服务器：

  1. int main() {
  2.     run_server();
  3.     return 0;
  4. }

3.2 客户端代码

保存以下代码为 rdma_client.c：

实现了一个简单的RDMA（Remote Direct Memory Access）客户端，连接到指定的服务器并发送一条消息。RDMA允许计算机通过高效的网络通信方式直接访问彼此的内存，而无需操作系统干预，从而减少延迟并提高数据传输速度。代码使用了RDMA的通信管理API和Infiniband verbs API来实现这个功能。

  1. #include <stdio.h>  // 包含标准输入输出库
  2. #include <stdlib.h>  // 包含标准库
  3. #include <string.h>  // 包含字符串处理库
  4. #include <rdma/rdma_cma.h>  // 包含RDMA连接管理API
  5. #include <infiniband/verbs.h>  // 包含Infiniband verbs API
  6.  
  7. #define BUFFER_SIZE 1024  // 定义缓冲区大小
  8.  
  9. // 客户端主函数，接收服务器地址作为参数
  10. void run_client(const char *server) {
  11.     // 声明和初始化变量
  12.     struct rdma_event_channel *ec = rdma_create_event_channel();  // 创建RDMA事件通道
  13.     struct rdma_cm_id *conn = NULL;  // RDMA连接标识符
  14.     struct rdma_addrinfo hints, *res;  // RDMA地址信息结构和指针
  15.     struct ibv_pd *pd = NULL;  // 保护域
  16.     struct ibv_mr *mr = NULL;  // 内存注册区域
  17.     struct ibv_comp_channel *comp_chan = NULL;  // 事件通道
  18.     struct ibv_cq *cq = NULL;  // 完成队列
  19.     struct ibv_qp_init_attr qp_attr;  // 队列对初始化属性
  20.     char buf[BUFFER_SIZE];  // 数据缓冲区
  21.     int ret;  // 返回值变量
  22.  
  23.     // 初始化hints结构体
  24.     memset(&hints, 0, sizeof(hints));  // 清空结构体
  25.     hints.ai_port_space = RDMA_PS_TCP;  // 设置端口空间为TCP
  26.  
  27.     // 获取地址信息
  28.     ret = rdma_getaddrinfo(server, "20079", &hints, &res);
  29.     if (ret) {
  30.         perror("rdma_getaddrinfo");  // 打印错误信息
  31.         exit(EXIT_FAILURE);  // 退出程序
  32.     }
  33.  
  34.     // 创建RDMA标识符
  35.     ret = rdma_create_id(ec, &conn, NULL, RDMA_PS_TCP);
  36.     if (ret) {
  37.         perror("rdma_create_id");
  38.         exit(EXIT_FAILURE);
  39.     }
  40.  
  41.     // 解析地址
  42.     ret = rdma_resolve_addr(conn, NULL, res->ai_dst_addr, 2000);
  43.     if (ret) {
  44.         perror("rdma_resolve_addr");
  45.         exit(EXIT_FAILURE);
  46.     }
  47.  
  48.     // 等待地址解析事件
  49.     struct rdma_cm_event *event;
  50.     ret = rdma_get_cm_event(ec, &event);
  51.     if (ret) {
  52.         perror("rdma_get_cm_event");
  53.         exit(EXIT_FAILURE);
  54.     }
  55.  
  56.     // 检查事件类型
  57.     if (event->event == RDMA_CM_EVENT_ADDR_RESOLVED) {
  58.         rdma_ack_cm_event(event);  // 确认事件
  59.     } else {
  60.         fprintf(stderr, "Unexpected event: %d\n", event->event);
  61.         rdma_ack_cm_event(event);
  62.         exit(EXIT_FAILURE);
  63.     }
  64.  
  65.     // 解析路由
  66.     ret = rdma_resolve_route(conn, 2000);
  67.     if (ret) {
  68.         perror("rdma_resolve_route");
  69.         exit(EXIT_FAILURE);
  70.     }
  71.  
  72.     // 等待路由解析事件
  73.     ret = rdma_get_cm_event(ec, &event);
  74.     if (ret) {
  75.         perror("rdma_get_cm_event");
  76.         exit(EXIT_FAILURE);
  77.     }
  78.  
  79.     // 检查事件类型
  80.     if (event->event == RDMA_CM_EVENT_ROUTE_RESOLVED) {
  81.         rdma_ack_cm_event(event);  // 确认事件
  82.     } else {
  83.         fprintf(stderr, "Unexpected event: %d\n", event->event);
  84.         rdma_ack_cm_event(event);
  85.         exit(EXIT_FAILURE);
  86.     }
  87.  
  88.     // 分配保护域
  89.     pd = ibv_alloc_pd(conn->verbs);
  90.     if (!pd) {
  91.         perror("ibv_alloc_pd");
  92.         exit(EXIT_FAILURE);
  93.     }
  94.  
  95.     // 创建完成通道
  96.     comp_chan = ibv_create_comp_channel(conn->verbs);
  97.     if (!comp_chan) {
  98.         perror("ibv_create_comp_channel");
  99.         exit(EXIT_FAILURE);
  100.     }
  101.  
  102.     // 创建完成队列
  103.     cq = ibv_create_cq(conn->verbs, 10, NULL, comp_chan, 0);
  104.     if (!cq) {
  105.         perror("ibv_create_cq");
  106.         exit(EXIT_FAILURE);
  107.     }
  108.  
  109.     // 初始化队列对属性
  110.     memset(&qp_attr, 0, sizeof(qp_attr));
  111.     qp_attr.cap.max_send_wr = 1;  // 最大发送请求数
  112.     qp_attr.cap.max_recv_wr = 1;  // 最大接收请求数
  113.     qp_attr.cap.max_send_sge = 1;  // 最大发送SGE数
  114.     qp_attr.cap.max_recv_sge = 1;  // 最大接收SGE数
  115.     qp_attr.send_cq = cq;  // 发送完成队列
  116.     qp_attr.recv_cq = cq;  // 接收完成队列
  117.     qp_attr.qp_type = IBV_QPT_RC;  // 队列对类型
  118.  
  119.     // 创建队列对
  120.     ret = rdma_create_qp(conn, pd, &qp_attr);
  121.     if (ret) {
  122.         perror("rdma_create_qp");
  123.         exit(EXIT_FAILURE);
  124.     }
  125.  
  126.     // 注册内存区域
  127.     mr = ibv_reg_mr(pd, buf, BUFFER_SIZE, IBV_ACCESS_LOCAL_WRITE | IBV_ACCESS_REMOTE_WRITE);
  128.     if (!mr) {
  129.         perror("ibv_reg_mr");
  130.         exit(EXIT_FAILURE);
  131.     }
  132.  
  133.     // 设置连接参数
  134.     struct rdma_conn_param cm_params;
  135.     memset(&cm_params, 0, sizeof(cm_params));
  136.     
  137.     // 发起连接请求
  138.     ret = rdma_connect(conn, &cm_params);
  139.     if (ret) {
  140.         perror("rdma_connect");
  141.         exit(EXIT_FAILURE);
  142.     }
  143.  
  144.     // 等待连接建立事件
  145.     ret = rdma_get_cm_event(ec, &event);
  146.     if (ret) {
  147.         perror("rdma_get_cm_event");
  148.         exit(EXIT_FAILURE);
  149.     }
  150.  
  151.     // 检查事件类型
  152.     if (event->event == RDMA_CM_EVENT_ESTABLISHED) {
  153.         rdma_ack_cm_event(event);  // 确认事件
  154.     } else {
  155.         fprintf(stderr, "Unexpected event: %d\n", event->event);
  156.         rdma_ack_cm_event(event);
  157.         exit(EXIT_FAILURE);
  158.     }
  159.  
  160.     // 准备发送数据
  161.     strcpy(buf, "Hello, RDMA!");
  162.     struct ibv_send_wr wr, *bad_wr = NULL;  // 发送请求和错误请求指针
  163.     struct ibv_sge sge;  // 单次数据描述符
  164.     sge.addr = (uintptr_t)buf;  // 数据缓冲区地址
  165.     sge.length = BUFFER_SIZE;  // 数据长度
  166.     sge.lkey = mr->lkey;  // 本地密钥
  167.  
  168.     // 初始化发送请求
  169.     wr.wr_id = 0;
  170.     wr.next = NULL;
  171.     wr.sg_list = &sge;  // 数据描述符列表
  172.     wr.num_sge = 1;  // 数据描述符数量
  173.     wr.opcode = IBV_WR_SEND;  // 操作码为发送
  174.     wr.send_flags = IBV_SEND_SIGNALED;  // 发送标志
  175.  
  176.     // 发送数据
  177.     ret = ibv_post_send(conn->qp, &wr, &bad_wr);
  178.     if (ret) {
  179.         perror("ibv_post_send");
  180.         exit(EXIT_FAILURE);
  181.     }
  182.  
  183.     // 等待完成队列事件
  184.     struct ibv_wc wc;
  185.     while ((ret = ibv_poll_cq(cq, 1, &wc)) == 0);
  186.     if (ret < 0) {
  187.         perror("ibv_poll_cq");
  188.         exit(EXIT_FAILURE);
  189.     }
  190.  
  191.     // 检查完成状态
  192.     if (wc.status != IBV_WC_SUCCESS) {
  193.         fprintf(stderr, "Failed status %s (%d) for wr_id %d\n",
  194.                 ibv_wc_status_str(wc.status), wc.status, (int) wc.wr_id);
  195.         exit(EXIT_FAILURE);
  196.     }
  197.  
  198.     // 打印发送消息
  199.     printf("Message sent: %s\n", buf);
  200.  
  201.     // 断开连接并释放资源
  202.     rdma_disconnect(conn);  // 断开连接
  203.     rdma_destroy_qp(conn);  // 销毁队列对
  204.     ibv_dereg_mr(mr);  // 解除内存注册
  205.     ibv_destroy_cq(cq);  // 销毁完成队列
  206.     ibv_destroy_comp_channel(comp_chan);  // 销毁完成通道
  207.     ibv_dealloc_pd(pd);  // 释放保护域
  208.     rdma_destroy_id(conn);  // 销毁连接标识符
  209.     rdma_destroy_event_channel(ec);  // 销毁事件通道
  210.     rdma_freeaddrinfo(res);  // 释放地址信息
  211. }
  212.  
  213. // 主函数
  214. int main(int argc, char **argv) {
  215.     if (argc != 2) {
  216.         fprintf(stderr, "Usage: %s <server-address>\n", argv[0]);  // 打印用法信息
  217.         return EXIT_FAILURE;  // 返回失败状态
  218.     }
  219.  
  220.     run_client(argv[1]);  // 运行客户端
  221.     return 0;  // 返回成功状态
  222. }

3.2.1 初始化和准备工作

1. 创建并初始化RDMA事件通道：

struct rdma_event_channel *ec = rdma_create_event_channel();// 创建RDMA事件通道

 2. 定义变量：

  1.     struct rdma_cm_id *conn = NULL;  // RDMA连接标识符
  2.     struct rdma_addrinfo hints, *res;  // RDMA地址信息结构和指针
  3.     struct ibv_pd *pd = NULL;  // 保护域
  4.     struct ibv_mr *mr = NULL;  // 内存注册区域
  5.     struct ibv_comp_channel *comp_chan = NULL;  // 事件通道
  6.     struct ibv_cq *cq = NULL;  // 完成队列
  7.     struct ibv_qp_init_attr qp_attr;  // 队列对初始化属性
  8.     char buf[BUFFER_SIZE];  // 数据缓冲区
  9.     int ret;  // 返回值变量
  • conn：RDMA连接标识符。
  • hints和res：用于存储地址信息。
  • pd、mr、comp_chan和cq：分别代表保护域、内存注册区、完成通道和完成队列。
  • qp_attr：用于配置QP的初始化属性。
  • buf：用于存储要发送的数据。
  • ret：用于存储各个操作的返回值。

3. 配置地址信息：

  1.     // 初始化hints结构体
  2.     memset(&hints, 0, sizeof(hints));  // 清空结构体
  3.     hints.ai_port_space = RDMA_PS_TCP;  // 设置端口空间为TCP

4. 获取RDMA地址：

  1.     // 获取地址信息
  2.     ret = rdma_getaddrinfo(server, "20079", &hints, &res);
  3.     if (ret) {
  4.         perror("rdma_getaddrinfo");  // 打印错误信息
  5.         exit(EXIT_FAILURE);  // 退出程序
  6.     }

3.2.2 创建和解析连接

1. 创建RDMA标识符：

  1.     // 创建RDMA标识符
  2.     ret = rdma_create_id(ec, &conn, NULL, RDMA_PS_TCP);
  3.     if (ret) {
  4.         perror("rdma_create_id");
  5.         exit(EXIT_FAILURE);
  6.     }

2. 解析服务器地址：

  1.     // 解析地址
  2.     ret = rdma_resolve_addr(conn, NULL, res->ai_dst_addr, 2000);
  3.     if (ret) {
  4.         perror("rdma_resolve_addr");
  5.         exit(EXIT_FAILURE);
  6.     }
  7.  
  8.     // 等待地址解析事件
  9.     struct rdma_cm_event *event;
  10.     ret = rdma_get_cm_event(ec, &event);
  11.     if (ret) {
  12.         perror("rdma_get_cm_event");
  13.         exit(EXIT_FAILURE);
  14.     }
  15.  
  16.     // 检查事件类型
  17.     if (event->event == RDMA_CM_EVENT_ADDR_RESOLVED) {
  18.         rdma_ack_cm_event(event);  // 确认事件
  19.     } else {
  20.         fprintf(stderr, "Unexpected event: %d\n", event->event);
  21.         rdma_ack_cm_event(event);
  22.         exit(EXIT_FAILURE);
  23.     }

3.2.3 路由解析

  1.     // 解析路由
  2.     ret = rdma_resolve_route(conn, 2000);
  3.     if (ret) {
  4.         perror("rdma_resolve_route");
  5.         exit(EXIT_FAILURE);
  6.     }
  7.  
  8.     // 等待路由解析事件
  9.     ret = rdma_get_cm_event(ec, &event);
  10.     if (ret) {
  11.         perror("rdma_get_cm_event");
  12.         exit(EXIT_FAILURE);
  13.     }
  14.  
  15.     // 检查事件类型
  16.     if (event->event == RDMA_CM_EVENT_ROUTE_RESOLVED) {
  17.         rdma_ack_cm_event(event);  // 确认事件
  18.     } else {
  19.         fprintf(stderr, "Unexpected event: %d\n", event->event);
  20.         rdma_ack_cm_event(event);
  21.         exit(EXIT_FAILURE);
  22.     }

3.2.4 资源分配和QP创建

1. 分配保护域、创建完成通道和完成队列：

  1.     // 分配保护域
  2.     pd = ibv_alloc_pd(conn->verbs);
  3.     if (!pd) {
  4.         perror("ibv_alloc_pd");
  5.         exit(EXIT_FAILURE);
  6.     }
  7.  
  8.     // 创建完成通道
  9.     comp_chan = ibv_create_comp_channel(conn->verbs);
  10.     if (!comp_chan) {
  11.         perror("ibv_create_comp_channel");
  12.         exit(EXIT_FAILURE);
  13.     }
  14.  
  15.     // 创建完成队列
  16.     cq = ibv_create_cq(conn->verbs, 10, NULL, comp_chan, 0);
  17.     if (!cq) {
  18.         perror("ibv_create_cq");
  19.         exit(EXIT_FAILURE);
  20.     }

2. 初始化并创建QP：

  1.     // 初始化队列对属性
  2.     memset(&qp_attr, 0, sizeof(qp_attr));
  3.     qp_attr.cap.max_send_wr = 1;  // 最大发送请求数
  4.     qp_attr.cap.max_recv_wr = 1;  // 最大接收请求数
  5.     qp_attr.cap.max_send_sge = 1;  // 最大发送SGE数
  6.     qp_attr.cap.max_recv_sge = 1;  // 最大接收SGE数
  7.     qp_attr.send_cq = cq;  // 发送完成队列
  8.     qp_attr.recv_cq = cq;  // 接收完成队列
  9.     qp_attr.qp_type = IBV_QPT_RC;  // 队列对类型
  10.  
  11.     // 创建队列对
  12.     ret = rdma_create_qp(conn, pd, &qp_attr);
  13.     if (ret) {
  14.         perror("rdma_create_qp");
  15.         exit(EXIT_FAILURE);
  16.     }

3.2.5 内存注册与连接

1. 注册内存区域：

  1.     // 注册内存区域
  2.     mr = ibv_reg_mr(pd, buf, BUFFER_SIZE, IBV_ACCESS_LOCAL_WRITE | IBV_ACCESS_REMOTE_WRITE);
  3.     if (!mr) {
  4.         perror("ibv_reg_mr");
  5.         exit(EXIT_FAILURE);
  6.     }

2. 发起连接请求：

  1.     // 设置连接参数
  2.     struct rdma_conn_param cm_params;
  3.     memset(&cm_params, 0, sizeof(cm_params));
  4.     
  5.     // 发起连接请求
  6.     ret = rdma_connect(conn, &cm_params);
  7.     if (ret) {
  8.         perror("rdma_connect");
  9.         exit(EXIT_FAILURE);
  10.     }
  11.  
  12.     // 等待连接建立事件
  13.     ret = rdma_get_cm_event(ec, &event);
  14.     if (ret) {
  15.         perror("rdma_get_cm_event");
  16.         exit(EXIT_FAILURE);
  17.     }
  18.  
  19.     // 检查事件类型
  20.     if (event->event == RDMA_CM_EVENT_ESTABLISHED) {
  21.         rdma_ack_cm_event(event);  // 确认事件
  22.     } else {
  23.         fprintf(stderr, "Unexpected event: %d\n", event->event);
  24.         rdma_ack_cm_event(event);
  25.         exit(EXIT_FAILURE);
  26.     }

3.2.6 数据发送

准备并发送数据：

  1.     // 准备发送数据
  2.     strcpy(buf, "Hello, RDMA!");
  3.     struct ibv_send_wr wr, *bad_wr = NULL;  // 发送请求和错误请求指针
  4.     struct ibv_sge sge;  // 单次数据描述符
  5.     sge.addr = (uintptr_t)buf;  // 数据缓冲区地址
  6.     sge.length = BUFFER_SIZE;  // 数据长度
  7.     sge.lkey = mr->lkey;  // 本地密钥
  8.  
  9.     // 初始化发送请求
  10.     wr.wr_id = 0;
  11.     wr.next = NULL;
  12.     wr.sg_list = &sge;  // 数据描述符列表
  13.     wr.num_sge = 1;  // 数据描述符数量
  14.     wr.opcode = IBV_WR_SEND;  // 操作码为发送
  15.     wr.send_flags = IBV_SEND_SIGNALED;  // 发送标志
  16.  
  17.     // 发送数据
  18.     ret = ibv_post_send(conn->qp, &wr, &bad_wr);
  19.     if (ret) {
  20.         perror("ibv_post_send");
  21.         exit(EXIT_FAILURE);
  22.     }
  23.  
  24.     // 等待完成队列事件
  25.     struct ibv_wc wc;
  26.     while ((ret = ibv_poll_cq(cq, 1, &wc)) == 0);
  27.     if (ret < 0) {
  28.         perror("ibv_poll_cq");
  29.         exit(EXIT_FAILURE);
  30.     }
  31.  
  32.     // 检查完成状态
  33.     if (wc.status != IBV_WC_SUCCESS) {
  34.         fprintf(stderr, "Failed status %s (%d) for wr_id %d\n",
  35.                 ibv_wc_status_str(wc.status), wc.status, (int) wc.wr_id);
  36.         exit(EXIT_FAILURE);
  37.     }
  38.  
  39.     // 打印发送消息
  40.     printf("Message sent: %s\n", buf);

3.2.7 清理资源

断开连接并释放资源

  1.     // 断开连接并释放资源
  2.     rdma_disconnect(conn);  // 断开连接
  3.     rdma_destroy_qp(conn);  // 销毁队列对
  4.     ibv_dereg_mr(mr);  // 解除内存注册
  5.     ibv_destroy_cq(cq);  // 销毁完成队列
  6.     ibv_destroy_comp_channel(comp_chan);  // 销毁完成通道
  7.     ibv_dealloc_pd(pd);  // 释放保护域
  8.     rdma_destroy_id(conn);  // 销毁连接标识符
  9.     rdma_destroy_event_channel(ec);  // 销毁事件通道
  10.     rdma_freeaddrinfo(res);  // 释放地址信息

3.2.8  主函数：

运行客户端：

  1. // 主函数
  2. int main(int argc, char **argv) {
  3.     if (argc != 2) {
  4.         fprintf(stderr, "Usage: %s <server-address>\n", argv[0]);  // 打印用法信息
  5.         return EXIT_FAILURE;  // 返回失败状态
  6.     }
  7.  
  8.     run_client(argv[1]);  // 运行客户端
  9.     return 0;  // 返回成功状态
  10. }

3.3 编译和运行

确保先编译服务器和客户端代码：

  1. gcc -o rdma_server rdma_server.c -lrdmacm -libverbs
  2. gcc -o rdma_client rdma_client.c -lrdmacm -libverbs

注意：

如果遇到此错误，因为链接器找不到 ibv_reg_mr_iova2 函数的定义。这个函数可能是你所使用的库版本中不存在的，或者它属于另一个库。

解决方案：

1.确保你安装的库版本是正确的，且包含 ibv_reg_mr_iova2 函数。检查所使用的库版本以及文档，确认该函数确实存在。

2.确保链接器能够找到正确的库。确保你已经正确安装并链接了 rdma-core 包。如果库路径不在默认的搜索路径中，可能需要手动指定库路径。例如：

gcc -o rdma_client rdma_client.c -L/usr/local/lib -lrdmacm -libverbs

 3.5 结果

服务器将打印收到的消息：

客户端将打印发送的消息：

可以了解到如何使用 rdma-core 进行基本的 RDMA 连接和数据传输。

四、rdma-core 实现更复杂的 RDMA 连接和数据传输操作

为了进一步利用 rdma-core 实现更复杂的 RDMA 连接和数据传输操作，你需要掌握一些基本概念和操作步骤。

4.1 RDMA 操作基础

RDMA 编程涉及多个步骤，包括设备发现、上下文创建、连接建立、数据传输等。

4.2 设备发现与上下文创建

你需要先获取 RDMA 设备列表，并选择一个设备进行上下文创建。

  1. #include <stdio.h>
  2. #include <stdlib.h>
  3. #include <string.h>
  4. #include <infiniband/verbs.h>
  5.  
  6. // 主函数
  7. int main() {
  8.     struct ibv_device **dev_list;     // 存储设备列表的指针数组
  9.     struct ibv_device *ib_dev;        // 存储选定设备的指针
  10.     struct ibv_context *ctx;          // 存储设备上下文的指针
  11.  
  12.     // 获取设备列表
  13.     dev_list = ibv_get_device_list(NULL);
  14.     if (!dev_list) {
  15.         // 如果获取设备列表失败，打印错误信息并返回失败状态
  16.         perror("Failed to get devices list");
  17.         return EXIT_FAILURE;
  18.     }
  19.  
  20.     // 选择第一个设备
  21.     ib_dev = dev_list[0];
  22.     if (!ib_dev) {
  23.         // 如果没有找到任何IB设备，打印错误信息并返回失败状态
  24.         fprintf(stderr, "No IB devices found\n");
  25.         return EXIT_FAILURE;
  26.     }
  27.  
  28.     // 获取设备的上下文
  29.     ctx = ibv_open_device(ib_dev);
  30.     if (!ctx) {
  31.         // 如果打开设备失败，打印错误信息并返回失败状态
  32.         perror("Failed to open device");
  33.         return EXIT_FAILURE;
  34.     }
  35.  
  36.     // 打印出成功打开的设备名称
  37.     printf("Device %s opened\n", ib_dev->name);
  38.  
  39.     // 清理分配的资源
  40.     ibv_close_device(ctx);            // 关闭设备上下文
  41.     ibv_free_device_list(dev_list);   // 释放设备列表
  42.  
  43.     return EXIT_SUCCESS;              // 返回成功状态
  44. }

 展示了如何使用Infiniband Verbs API获取并打开Infiniband设备，适用于需要与Infiniband设备进行低级别交互的应用程序

4.1.1 变量定义

  1. struct ibv_device **dev_list;     // 存储设备列表的指针数组
  2. struct ibv_device *ib_dev;        // 存储选定设备的指针
  3. struct ibv_context *ctx;          // 存储设备上下文的指针

声明了三个指针变量：

  • dev_list：用于存储设备列表。
  • ib_dev：用于存储选定的设备。
  • ctx：用于存储设备的上下文信息。

4.1.2 获取设备列表 ：

  1. dev_list = ibv_get_device_list(NULL);
  2. if (!dev_list) {
  3.      perror("Failed to get devices list");
  4.      return EXIT_FAILURE;
  5. }

调用ibv_get_device_list函数获取设备列表，如果获取失败，打印错误信息并返回失败状态。 

4.1.3 选择设备：

  1. ib_dev = dev_list[0];
  2. if (!ib_dev) {
  3.     fprintf(stderr, "No IB devices found\n");
  4.     return EXIT_FAILURE;
  5. }

选择设备列表中的第一个设备。如果设备列表为空，打印错误信息并返回失败状态。 

4.1.4 获取设备上下文

  1. ctx = ibv_open_device(ib_dev);
  2. if (!ctx) {
  3.     perror("Failed to open device");
  4.     return EXIT_FAILURE;
  5. }

调用ibv_open_device函数获取选定设备的上下文。如果获取失败，打印错误信息并返回失败状态。 

4.1.5 打印设备名称并且清理资源

  1. printf("Device %s opened\n", ib_dev->name);
  2. ibv_close_device(ctx);            // 关闭设备上下文
  3. ibv_free_device_list(dev_list);   // 释放设备列表

打印出成功打开的设备名称，关闭设备上下文并释放设备列表，清理资源。

4.1.6 编译运行：

将提供的代码保存到一个文件中，例如ibv_example.c。

使用GCC编译器来编译代码，并链接Infiniband库：

gcc -o ibv_example ibv_example.c -libverbs

运行编译生成的可执行文件：

./ibv_example

4.1.7 示例输出：

假设系统中有可用的Infiniband设备，并且代码运行成功，输出可能如下：

Device rxe0 opened

如果系统中没有Infiniband设备，输出可能如下：

No IB devices found

 如果在获取设备列表或打开设备时发生错误，输出可能如下：

  1. Failed to get devices list: <error_description>
  2. Failed to open device: <error_description>

此时我们输出则显示有可用的Infiniband设备：

4.3 内存注册与队列对创建

为了传输数据，你需要注册内存区域并创建发送和接收队列。

  1. #include <stdio.h>                       // 包含标准输入输出头文件
  2. #include <stdlib.h>                      // 包含标准库头文件
  3. #include <string.h>                      // 包含字符串操作头文件
  4. #include <infiniband/verbs.h>            // 包含InfiniBand verbs库头文件
  5.  
  6. #define BUFFER_SIZE 1024                 // 定义缓冲区大小为1024字节
  7.  
  8. int main() {
  9.     struct ibv_device **dev_list;        // 定义设备列表指针，指向InfiniBand设备数组
  10.     struct ibv_device *ib_dev;           // 定义设备指针，指向单个InfiniBand设备
  11.     struct ibv_context *ctx;             // 定义上下文指针，表示设备上下文
  12.     struct ibv_pd *pd;                   // 定义保护域指针
  13.     struct ibv_mr *mr;                   // 定义内存注册区域指针
  14.     struct ibv_cq *cq;                   // 定义完成队列指针
  15.     struct ibv_qp *qp;                   // 定义队列对指针
  16.     struct ibv_qp_init_attr qp_attr;     // 定义队列对初始化属性结构
  17.     char buf[BUFFER_SIZE];               // 定义缓冲区
  18.  
  19.     // 获取设备列表并打开设备
  20.     dev_list = ibv_get_device_list(NULL);                  // 获取所有InfiniBand设备列表
  21.     ib_dev = dev_list[0];                                  // 选择第一个设备
  22.     ctx = ibv_open_device(ib_dev);                         // 打开设备并获取设备上下文
  23.  
  24.     // 创建保护域
  25.     pd = ibv_alloc_pd(ctx);                                // 为设备上下文分配保护域
  26.     if (!pd) {                                             // 检查保护域是否分配成功
  27.         perror("ibv_alloc_pd");                            // 如果失败，打印错误信息
  28.         return EXIT_FAILURE;                               // 返回失败状态
  29.     }
  30.  
  31.     // 注册内存区域
  32.     mr = ibv_reg_mr(pd, buf, BUFFER_SIZE, IBV_ACCESS_LOCAL_WRITE | IBV_ACCESS_REMOTE_WRITE);  // 注册内存区域
  33.     if (!mr) {                                             // 检查内存区域是否注册成功
  34.         perror("ibv_reg_mr");                              // 如果失败，打印错误信息
  35.         return EXIT_FAILURE;                               // 返回失败状态
  36.     }
  37.  
  38.     // 创建完成队列
  39.     cq = ibv_create_cq(ctx, 10, NULL, NULL, 0);            // 创建一个完成队列，最多包含10个完成元素
  40.     if (!cq) {                                             // 检查完成队列是否创建成功
  41.         perror("ibv_create_cq");                           // 如果失败，打印错误信息
  42.         return EXIT_FAILURE;                               // 返回失败状态
  43.     }
  44.  
  45.     // 设置队列对属性
  46.     memset(&qp_attr, 0, sizeof(qp_attr));                  // 将队列对初始化属性结构清零
  47.     qp_attr.cap.max_send_wr = 1;                           // 设置最大发送请求数为1
  48.     qp_attr.cap.max_recv_wr = 1;                           // 设置最大接收请求数为1
  49.     qp_attr.cap.max_send_sge = 1;                          // 设置单个发送请求最大散播聚合元素数为1
  50.     qp_attr.cap.max_recv_sge = 1;                          // 设置单个接收请求最大散播聚合元素数为1
  51.     qp_attr.send_cq = cq;                                  // 关联发送完成队列
  52.     qp_attr.recv_cq = cq;                                  // 关联接收完成队列
  53.     qp_attr.qp_type = IBV_QPT_RC;                          // 设置队列对类型为可靠连接
  54.  
  55.     // 创建队列对
  56.     qp = ibv_create_qp(pd, &qp_attr);                      // 创建队列对
  57.     if (!qp) {                                             // 检查队列对是否创建成功
  58.         perror("ibv_create_qp");                           // 如果失败，打印错误信息
  59.         return EXIT_FAILURE;                               // 返回失败状态
  60.     }
  61.  
  62.     printf("Queue Pair created successfully\n");           // 打印成功创建队列对的消息
  63.  
  64.     // 清理
  65.     ibv_destroy_qp(qp);                                    // 销毁队列对
  66.     ibv_destroy_cq(cq);                                    // 销毁完成队列
  67.     ibv_dereg_mr(mr);                                      // 取消注册内存区域
  68.     ibv_dealloc_pd(pd);                                    // 释放保护域
  69.     ibv_close_device(ctx);                                 // 关闭设备
  70.     ibv_free_device_list(dev_list);                        // 释放设备列表
  71.  
  72.     return EXIT_SUCCESS;                                   // 返回成功状态
  73. }

演示如何使用libibverbs库创建并初始化一个InfiniBand队列对（Queue Pair，QP）。

4.3.1 获取InfiniBand设备列表并打开设备:（此过程就是4.2介绍的过程）

  1. dev_list = ibv_get_device_list(NULL);
  2. ib_dev = dev_list[0];
  3. ctx = ibv_open_device(ib_dev);

4.3.2 创建保护域（Protection Domain，PD）:

pd = ibv_alloc_pd(ctx);

4.3.3 注册内存区域:

mr = ibv_reg_mr(pd, buf, BUFFER_SIZE, IBV_ACCESS_LOCAL_WRITE | IBV_ACCESS_REMOTE_WRITE);

4.3.4 创建完成队列（Completion Queue，CQ）: 

cq = ibv_create_cq(ctx, 10, NULL, NULL, 0);

4.3.5  设置队列对属性并创建队列对:

  1. memset(&qp_attr, 0, sizeof(qp_attr));
  2. qp_attr.cap.max_send_wr = 1;
  3. qp_attr.cap.max_recv_wr = 1;
  4. qp_attr.cap.max_send_sge = 1;
  5. qp_attr.cap.max_recv_sge = 1;
  6. qp_attr.send_cq = cq;
  7. qp_attr.recv_cq = cq;
  8. qp_attr.qp_type = IBV_QPT_RC;
  9.  
  10. qp = ibv_create_qp(pd, &qp_attr);

4.3.6 清理资源:

程序结束前清理分配的资源。 

  1. ibv_destroy_qp(qp);
  2. ibv_destroy_cq(cq);
  3. ibv_dereg_mr(mr);
  4. ibv_dealloc_pd(pd);
  5. ibv_close_device(ctx);
  6. ibv_free_device_list(dev_list);

4.3.7 演示效果

这表示队列对（Queue Pair）成功创建。程序会在执行后清理所有资源，确保没有资源泄漏。

4.4 建立 RDMA 连接

你需要使用 rdma_cm 库来建立 RDMA 连接。这包括服务器和客户端之间的连接建立。

 1. 服务器端代码（与3.1代码略有不同）

与3.1最大差异体现在资源释放更加全面，确保在出错时也能释放所有已分配的资源。

  1. #include <stdio.h>       // 包含标准输入输出库
  2. #include <stdlib.h>      // 包含标准库
  3. #include <string.h>      // 包含字符串处理库
  4. #include <rdma/rdma_cma.h>  // 包含RDMA连接管理API
  5.  
  6. #define BUFFER_SIZE 1024  // 定义缓冲区大小
  7.  
  8. void run_server() {
  9.     // 创建RDMA事件通道，用于接收RDMA事件
  10.     struct rdma_event_channel *ec = rdma_create_event_channel();
  11.     struct rdma_cm_id *listener = NULL, *conn = NULL;  // RDMA连接标识符
  12.     struct rdma_addrinfo hints, *res;  // RDMA地址信息结构和指针
  13.     struct ibv_pd *pd = NULL;  // 保护域
  14.     struct ibv_mr *mr = NULL;  // 内存注册区域
  15.     struct ibv_comp_channel *comp_chan = NULL;  // 完成事件通道
  16.     struct ibv_cq *cq = NULL;  // 完成队列
  17.     struct ibv_qp_init_attr qp_attr;  // 队列对初始化属性
  18.     char buf[BUFFER_SIZE];  // 数据缓冲区
  19.     int ret;  // 返回值变量
  20.  
  21.     // 配置地址信息
  22.     memset(&hints, 0, sizeof(hints));  // 清空结构体
  23.     hints.ai_flags = RAI_PASSIVE;  // 设置为被动模式
  24.     hints.ai_port_space = RDMA_PS_TCP;  // 设置端口空间为TCP
  25.  
  26.     // 获取地址信息
  27.     ret = rdma_getaddrinfo(NULL, "20079", &hints, &res);
  28.     if (ret) {
  29.         perror("rdma_getaddrinfo");  // 打印错误信息
  30.         exit(EXIT_FAILURE);  // 退出程序
  31.     }
  32.  
  33.     // 创建RDMA标识符
  34.     ret = rdma_create_id(ec, &listener, NULL, RDMA_PS_TCP);
  35.     if (ret) {
  36.         perror("rdma_create_id");
  37.         rdma_freeaddrinfo(res);  // 释放地址信息
  38.         exit(EXIT_FAILURE);
  39.     }
  40.  
  41.     // 绑定地址
  42.     ret = rdma_bind_addr(listener, res->ai_src_addr);
  43.     if (ret) {
  44.         perror("rdma_bind_addr");
  45.         rdma_destroy_id(listener);  // 销毁RDMA标识符
  46.         rdma_freeaddrinfo(res);  // 释放地址信息
  47.         exit(EXIT_FAILURE);
  48.     }
  49.  
  50.     // 监听连接请求
  51.     ret = rdma_listen(listener, 0);
  52.     if (ret) {
  53.         perror("rdma_listen");
  54.         rdma_destroy_id(listener);  // 销毁RDMA标识符
  55.         rdma_freeaddrinfo(res);  // 释放地址信息
  56.         exit(EXIT_FAILURE);
  57.     }
  58.  
  59.     printf("Server is listening on port 20079...\n");
  60.  
  61.     // 等待连接请求事件
  62.     struct rdma_cm_event *event;
  63.     ret = rdma_get_cm_event(ec, &event);
  64.     if (ret) {
  65.         perror("rdma_get_cm_event");
  66.         rdma_destroy_id(listener);  // 销毁RDMA标识符
  67.         rdma_freeaddrinfo(res);  // 释放地址信息
  68.         exit(EXIT_FAILURE);
  69.     }
  70.  
  71.     // 处理连接请求事件
  72.     if (event->event == RDMA_CM_EVENT_CONNECT_REQUEST) {
  73.         conn = event->id;  // 获取连接标识符
  74.         rdma_ack_cm_event(event);  // 确认事件
  75.     } else {
  76.         fprintf(stderr, "Unexpected event: %d\n", event->event);
  77.         rdma_ack_cm_event(event);  // 确认事件
  78.         rdma_destroy_id(listener);  // 销毁RDMA标识符
  79.         rdma_freeaddrinfo(res);  // 释放地址信息
  80.         exit(EXIT_FAILURE);
  81.     }
  82.  
  83.     // 分配保护域
  84.     pd = ibv_alloc_pd(conn->verbs);
  85.     if (!pd) {
  86.         perror("ibv_alloc_pd");
  87.         rdma_destroy_id(conn);  // 销毁连接标识符
  88.         rdma_destroy_id(listener);  // 销毁监听标识符
  89.         rdma_freeaddrinfo(res);  // 释放地址信息
  90.         exit(EXIT_FAILURE);
  91.     }
  92.  
  93.     // 创建完成事件通道
  94.     comp_chan = ibv_create_comp_channel(conn->verbs);
  95.     if (!comp_chan) {
  96.         perror("ibv_create_comp_channel");
  97.         ibv_dealloc_pd(pd);  // 释放保护域
  98.         rdma_destroy_id(conn);  // 销毁连接标识符
  99.         rdma_destroy_id(listener);  // 销毁监听标识符
  100.         rdma_freeaddrinfo(res);  // 释放地址信息
  101.         exit(EXIT_FAILURE);
  102.     }
  103.  
  104.     // 创建完成队列
  105.     cq = ibv_create_cq(conn->verbs, 10, NULL, comp_chan, 0);
  106.     if (!cq) {
  107.         perror("ibv_create_cq");
  108.         ibv_destroy_comp_channel(comp_chan);  // 销毁完成事件通道
  109.         ibv_dealloc_pd(pd);  // 释放保护域
  110.         rdma_destroy_id(conn);  // 销毁连接标识符
  111.         rdma_destroy_id(listener);  // 销毁监听标识符
  112.         rdma_freeaddrinfo(res);  // 释放地址信息
  113.         exit(EXIT_FAILURE);
  114.     }
  115.  
  116.     // 配置队列对属性
  117.     memset(&qp_attr, 0, sizeof(qp_attr));
  118.     qp_attr.cap.max_send_wr = 1;  // 最大发送请求数
  119.     qp_attr.cap.max_recv_wr = 1;  // 最大接收请求数
  120.     qp_attr.cap.max_send_sge = 1;  // 最大发送SGE数
  121.     qp_attr.cap.max_recv_sge = 1;  // 最大接收SGE数
  122.     qp_attr.send_cq = cq;  // 发送完成队列
  123.     qp_attr.recv_cq = cq;  // 接收完成队列
  124.     qp_attr.qp_type = IBV_QPT_RC;  // 队列对类型
  125.  
  126.     // 创建队列对
  127.     ret = rdma_create_qp(conn, pd, &qp_attr);
  128.     if (ret) {
  129.         perror("rdma_create_qp");
  130.         ibv_destroy_cq(cq);  // 销毁完成队列
  131.         ibv_destroy_comp_channel(comp_chan);  // 销毁完成事件通道
  132.         ibv_dealloc_pd(pd);  // 释放保护域
  133.         rdma_destroy_id(conn);  // 销毁连接标识符
  134.         rdma_destroy_id(listener);  // 销毁监听标识符
  135.         rdma_freeaddrinfo(res);  // 释放地址信息
  136.         exit(EXIT_FAILURE);
  137.     }
  138.  
  139.     // 注册内存区域
  140.     mr = ibv_reg_mr(pd, buf, BUFFER_SIZE, IBV_ACCESS_LOCAL_WRITE | IBV_ACCESS_REMOTE_WRITE);
  141.     if (!mr) {
  142.         perror("ibv_reg_mr");
  143.         rdma_destroy_qp(conn);  // 销毁队列对
  144.         ibv_destroy_cq(cq);  // 销毁完成队列
  145.         ibv_destroy_comp_channel(comp_chan);  // 销毁完成事件通道
  146.         ibv_dealloc_pd(pd);  // 释放保护域
  147.         rdma_destroy_id(conn);  // 销毁连接标识符
  148.         rdma_destroy_id(listener);  // 销毁监听标识符
  149.         rdma_freeaddrinfo(res);  // 释放地址信息
  150.         exit(EXIT_FAILURE);
  151.     }
  152.  
  153.     // 设置接收请求
  154.     struct ibv_recv_wr wr, *bad_wr = NULL;
  155.     struct ibv_sge sge;
  156.     sge.addr = (uintptr_t)buf;
  157.     sge.length = BUFFER_SIZE;
  158.     sge.lkey = mr->lkey;
  159.  
  160.     wr.wr_id = 0;
  161.     wr.next = NULL;
  162.     wr.sg_list = &sge;
  163.     wr.num_sge = 1;
  164.  
  165.     // 发起接收请求
  166.     ret = ibv_post_recv(conn->qp, &wr, &bad_wr);
  167.     if (ret) {
  168.         perror("ibv_post_recv");
  169.         ibv_dereg_mr(mr);  // 解除内存注册
  170.         rdma_destroy_qp(conn);  // 销毁队列对
  171.         ibv_destroy_cq(cq);  // 销毁完成队列
  172.         ibv_destroy_comp_channel(comp_chan);  // 销毁完成事件通道
  173.         ibv_dealloc_pd(pd);  // 释放保护域
  174.         rdma_destroy_id(conn);  // 销毁连接标识符
  175.         rdma_destroy_id(listener);  // 销毁监听标识符
  176.         rdma_freeaddrinfo(res);  // 释放地址信息
  177.         exit(EXIT_FAILURE);
  178.     }
  179.  
  180.     // 接受连接请求
  181.     struct rdma_conn_param cm_params;
  182.     memset(&cm_params, 0, sizeof(cm_params));
  183.     ret = rdma_accept(conn, &cm_params);
  184.     if (ret) {
  185.         perror("rdma_accept");
  186.         rdma_disconnect(conn);  // 断开连接
  187.         ibv_dereg_mr(mr);  // 解除内存注册
  188.         rdma_destroy_qp(conn);  // 销毁队列对
  189.         ibv_destroy_cq(cq);  // 销毁完成队列
  190.         ibv_destroy_comp_channel(comp_chan);  // 销毁完成事件通道
  191.         ibv_dealloc_pd(pd);  // 释放保护域
  192.         rdma_destroy_id(conn);  // 销毁连接标识符
  193.         rdma_destroy_id(listener);  // 销毁监听标识符
  194.         rdma_freeaddrinfo(res);  // 释放地址信息
  195.         exit(EXIT_FAILURE);
  196.     }
  197.  
  198.     // 轮询完成队列，等待数据接收
  199.     struct ibv_wc wc;
  200.     while ((ret = ibv_poll_cq(cq, 1, &wc)) == 0);
  201.     if (ret < 0) {
  202.         perror("ibv_poll_cq");
  203.         rdma_disconnect(conn);  // 断开连接
  204.         ibv_dereg_mr(mr);  // 解除内存注册
  205.         rdma_destroy_qp(conn);  // 销毁队列对
  206.         ibv_destroy_cq(cq);  // 销毁完成队列
  207.         ibv_destroy_comp_channel(comp_chan);  // 销毁完成事件通道
  208.         ibv_dealloc_pd(pd);  // 释放保护域
  209.         rdma_destroy_id(conn);  // 销毁连接标识符
  210.         rdma_destroy_id(listener);  // 销毁监听标识符
  211.         rdma_freeaddrinfo(res);  // 释放地址信息
  212.         exit(EXIT_FAILURE);
  213.     }
  214.  
  215.     // 检查完成状态
  216.     if (wc.status != IBV_WC_SUCCESS) {
  217.         fprintf(stderr, "Failed status %s (%d) for wr_id %d\n",
  218.                 ibv_wc_status_str(wc.status), wc.status, (int) wc.wr_id);
  219.         rdma_disconnect(conn);  // 断开连接
  220.         ibv_dereg_mr(mr);  // 解除内存注册
  221.         rdma_destroy_qp(conn);  // 销毁队列对
  222.         ibv_destroy_cq(cq);  // 销毁完成队列
  223.         ibv_destroy_comp_channel(comp_chan);  // 销毁完成事件通道
  224.         ibv_dealloc_pd(pd);  // 释放保护域
  225.         rdma_destroy_id(conn);  // 销毁连接标识符
  226.         rdma_destroy_id(listener);  // 销毁监听标识符
  227.         rdma_freeaddrinfo(res);  // 释放地址信息
  228.         exit(EXIT_FAILURE);
  229.     }
  230.  
  231.     // 打印接收到的消息
  232.     printf("Received message: %s\n", buf);
  233.  
  234.     // 清理资源
  235.     rdma_disconnect(conn);  // 断开连接
  236.     rdma_destroy_qp(conn);  // 销毁队列对
  237.     ibv_dereg_mr(mr);  // 解除内存注册
  238.     ibv_destroy_cq(cq);  // 销毁完成队列
  239.     ibv_destroy_comp_channel(comp_chan);  // 销毁完成事件通道
  240.     ibv_dealloc_pd(pd);  // 释放保护域
  241.     rdma_destroy_id(conn);  // 销毁连接标识符
  242.     rdma_destroy_id(listener);  // 销毁监听标识符
  243.     rdma_destroy_event_channel(ec);  // 销毁事件通道
  244.     rdma_freeaddrinfo(res);  // 释放地址信息
  245. }
  246.  
  247. int main() {
  248.     run_server();  // 运行服务器
  249.     return 0;
  250. }

2. 客户端代码（与3.2代码相同）

保存以下代码为 rdma_client.c：

  1. #include <stdio.h>
  2. #include <stdlib.h>
  3. #include <string.h>
  4. #include <rdma/rdma_cma.h>
  5. #include <infiniband/verbs.h>
  6.  
  7. #define BUFFER_SIZE 1024
  8.  
  9. void run_client(const char *server) {
  10.     struct rdma_event_channel *ec = rdma_create_event_channel();
  11.     struct rdma_cm_id *conn = NULL;
  12.     struct rdma_addrinfo hints, *res;
  13.     struct ibv_pd *pd = NULL;
  14.     struct ibv_mr *mr = NULL;
  15.     struct ibv_comp_channel *comp_chan = NULL;
  16.     struct ibv_cq *cq = NULL;
  17.     struct ibv_qp_init_attr qp_attr;
  18.     char buf[BUFFER_SIZE];
  19.     int ret;
  20.  
  21.     // 配置地址信息
  22.     memset(&hints, 0, sizeof(hints));
  23.     hints.ai_port_space = RDMA_PS_TCP;
  24.  
  25.     // 获取地址信息
  26.     ret = rdma_getaddrinfo(server, "20079", &hints, &res);
  27.     if (ret) {
  28.         perror("rdma_getaddrinfo");
  29.         exit(EXIT_FAILURE);
  30.     }
  31.  
  32.     // 创建 RDMA 连接标识符
  33.     ret = rdma_create_id(ec, &conn, NULL, RDMA_PS_TCP);
  34.     if (ret) {
  35.         perror("rdma_create_id");
  36.         exit(EXIT_FAILURE);
  37.     }
  38.  
  39.     // 解析地址
  40.     ret = rdma_resolve_addr(conn, NULL, res->ai_dst_addr, 2000);
  41.     if (ret) {
  42.         perror("rdma_resolve_addr");
  43.         exit(EXIT_FAILURE);
  44.     }
  45.  
  46.     struct rdma_cm_event *event;
  47.     ret = rdma_get_cm_event(ec, &event);
  48.     if (ret) {
  49.         perror("rdma_get_cm_event");
  50.         exit(EXIT_FAILURE);
  51.     }
  52.  
  53.     if (event->event == RDMA_CM_EVENT_ADDR_RESOLVED) {
  54.         rdma_ack_cm_event(event);
  55.     } else {
  56.         fprintf(stderr, "Unexpected event: %d\n", event->event);
  57.         rdma_ack_cm_event(event);
  58.         exit(EXIT_FAILURE);
  59.     }
  60.  
  61.     // 解析路由
  62.     ret = rdma_resolve_route(conn, 2000);
  63.     if (ret) {
  64.         perror("rdma_resolve_route");
  65.         exit(EXIT_FAILURE);
  66.     }
  67.  
  68.     ret = rdma_get_cm_event(ec, &event);
  69.     if (ret) {
  70.         perror("rdma_get_cm_event");
  71.         exit(EXIT_FAILURE);
  72.     }
  73.  
  74.     if (event->event == RDMA_CM_EVENT_ROUTE_RESOLVED) {
  75.         rdma_ack_cm_event(event);
  76.     } else {
  77.         fprintf(stderr, "Unexpected event: %d\n", event->event);
  78.         rdma_ack_cm_event(event);
  79.         exit(EXIT_FAILURE);
  80.     }
  81.  
  82.     // 分配保护域
  83.     pd = ibv_alloc_pd(conn->verbs);
  84.     if (!pd) {
  85.         perror("ibv_alloc_pd");
  86.         exit(EXIT_FAILURE);
  87.     }
  88.  
  89.     // 创建完成事件通道
  90.     comp_chan = ibv_create_comp_channel(conn->verbs);
  91.     if (!comp_chan) {
  92.         perror("ibv_create_comp_channel");
  93.         exit(EXIT_FAILURE);
  94.     }
  95.  
  96.     // 创建完成队列
  97.     cq = ibv_create_cq(conn->verbs, 10, NULL, comp_chan, 0);
  98.     if (!cq) {
  99.         perror("ibv_create_cq");
  100.         exit(EXIT_FAILURE);
  101.     }
  102.  
  103.     // 配置队列对属性
  104.     memset(&qp_attr, 0, sizeof(qp_attr));
  105.     qp_attr.cap.max_send_wr = 1;
  106.     qp_attr.cap.max_recv_wr = 1;
  107.     qp_attr.cap.max_send_sge = 1;
  108.     qp_attr.cap.max_recv_sge = 1;
  109.     qp_attr.send_cq = cq;
  110.     qp_attr.recv_cq = cq;
  111.     qp_attr.qp_type = IBV_QPT_RC;
  112.  
  113.     // 创建队列对
  114.     ret = rdma_create_qp(conn, pd, &qp_attr);
  115.     if (ret) {
  116.         perror("rdma_create_qp");
  117.         exit(EXIT_FAILURE);
  118.     }
  119.  
  120.     // 注册内存区域
  121.     mr = ibv_reg_mr(pd, buf, BUFFER_SIZE, IBV_ACCESS_LOCAL_WRITE | IBV_ACCESS_REMOTE_WRITE);
  122.     if (!mr) {
  123.         perror("ibv_reg_mr");
  124.         exit(EXIT_FAILURE);
  125.     }
  126.  
  127.     // 连接到服务器
  128.     struct rdma_conn_param cm_params;
  129.     memset(&cm_params, 0, sizeof(cm_params));
  130.     ret = rdma_connect(conn, &cm_params);
  131.     if (ret) {
  132.         perror("rdma_connect");
  133.         exit(EXIT_FAILURE);
  134.     }
  135.  
  136.     // 等待连接建立事件
  137.     ret = rdma_get_cm_event(ec, &event);
  138.     if (ret) {
  139.         perror("rdma_get_cm_event");
  140.         exit(EXIT_FAILURE);
  141.     }
  142.  
  143.     if (event->event == RDMA_CM_EVENT_ESTABLISHED) {
  144.         rdma_ack_cm_event(event);
  145.     } else {
  146.         fprintf(stderr, "Unexpected event: %d\n", event->event);
  147.         rdma_ack_cm_event(event);
  148.         exit(EXIT_FAILURE);
  149.     }
  150.  
  151.     // 准备发送数据
  152.     strcpy(buf, "Hello, RDMA!");
  153.     struct ibv_send_wr wr, *bad_wr = NULL;
  154.     struct ibv_sge sge;
  155.     sge.addr = (uintptr_t)buf;
  156.     sge.length = BUFFER_SIZE;
  157.     sge.lkey = mr->lkey;
  158.  
  159.     wr.wr_id = 0;
  160.     wr.next = NULL;
  161.     wr.sg_list = &sge;
  162.     wr.num_sge = 1;
  163.     wr.opcode = IBV_WR_SEND;
  164.     wr.send_flags = IBV_SEND_SIGNALED;
  165.  
  166.     // 发送数据
  167.     ret = ibv_post_send(conn->qp, &wr, &bad_wr);
  168.     if (ret) {
  169.         perror("ibv_post_send");
  170.         exit(EXIT_FAILURE);
  171.     }
  172.  
  173.     // 轮询完成队列
  174.     struct ibv_wc wc;
  175.     while ((ret = ibv_poll_cq(cq, 1, &wc)) == 0);
  176.     if (ret < 0) {
  177.         perror("ibv_poll_cq");
  178.         exit(EXIT_FAILURE);
  179.     }
  180.  
  181.     if (wc.status != IBV_WC_SUCCESS) {
  182.         fprintf(stderr, "Failed status %s (%d) for wr_id %d\n",
  183.                 ibv_wc_status_str(wc.status), wc.status, (int) wc.wr_id);
  184.         exit(EXIT_FAILURE);
  185.     }
  186.  
  187.     printf("Message sent: %s\n", buf);
  188.  
  189.     // 清理资源
  190.     rdma_disconnect(conn);
  191.     rdma_destroy_qp(conn);
  192.     ibv_dereg_mr(mr);
  193.     ibv_destroy_cq(cq);
  194.     ibv_destroy_comp_channel(comp_chan);
  195.     ibv_dealloc_pd(pd);
  196.     rdma_destroy_id(conn);
  197.     rdma_destroy_event_channel(ec);
  198.     rdma_freeaddrinfo(res);
  199. }
  200.  
  201. int main(int argc, char **argv) {
  202.     if (argc != 2) {
  203.         fprintf(stderr, "Usage: %s <server-address>\n", argv[0]);
  204.         return EXIT_FAILURE;
  205.     }
  206.  
  207.     run_client(argv[1]);
  208.     return 0;
  209. }

4.5 编译和运行

确保先编译服务器和客户端代码：

  1. gcc -o rdma_server rdma_server.c -lrdmacm -libverbs
  2. gcc -o rdma_client rdma_client.c -lrdmacm -libverbs

 注意：

如果遇到此错误，因为链接器找不到 ibv_reg_mr_iova2 函数的定义。这个函数可能是你所使用的库版本中不存在的，或者它属于另一个库。

解决方案：

1.确保你安装的库版本是正确的，且包含 ibv_reg_mr_iova2 函数。检查所使用的库版本以及文档，确认该函数确实存在。

2.确保链接器能够找到正确的库。确保你已经正确安装并链接了 rdma-core 包。如果库路径不在默认的搜索路径中，可能需要手动指定库路径。例如：

gcc -o rdma_client rdma_client.c -L/usr/local/lib -lrdmacm -libverbs

首先启动服务器：

./rdma_server

然后在另一个终端中启动客户端，替换 <server-address> 为服务器的实际 IP 地址：

./rdma_client <server-address>

 4.6 结果

服务器将打印收到的消息：

客户端将打印发送的消息：

可以了解到如何使用 rdma-core 进行基本的 RDMA 连接和数据传输。

五、验证RDMA 在高性能和低延迟场景中的应用价值

验证 RDMA（远程直接内存访问）在高性能和低延迟场景中的应用价值，可以通过一系列实验和测试来进行。下面是一些常见的方法和指标，用于验证和评估 RDMA 系统的性能和延迟。

5.1 性能测试方法

5.1.1 吞吐量测试：

目的：评估系统的数据传输速度。

方法：在一段时间内持续传输数据，统计传输的数据量（如每秒传输的字节数）。

工具：可以使用 Iperf 等网络性能测试工具，或编写自定义测试脚本。

5.1.2 延迟测试：

目的：评估系统的数据传输延迟。

方法：测量发送方发送数据包到接收方接收到数据包之间的时间。

工具：可以使用 ping、qperf 等工具，或通过编写自定义代码测量 RTT（往返时间）。

5.2.3 零拷贝效应：

目的：评估零拷贝技术对系统性能的提升。

方法：比较启用和禁用零拷贝技术情况下的吞吐量和延迟。

工具：自定义测试代码和性能分析工具。

5.2.4 资源使用率分析：

目的：评估系统的资源消耗情况，包括 CPU 使用率、内存占用等。

方法：监控系统在数据传输过程中的资源使用情况。

工具：top、htop、vmstat 等系统监控工具。

5.2  实验设计

5.2.1  测试环境搭建：

搭建一个包含服务器和客户端的测试环境，可以使用物理机或虚拟机。

配置网络，使其支持 RDMA 通信。

5.2.2 基准测试：

选择传统的 TCP/IP 网络通信作为对比基准。

运行相同的测试用例，对比 RDMA 和 TCP/IP 在吞吐量和延迟上的性能差异。

5.2.3 压力测试：

通过增加并发连接数和数据包大小，测试 RDMA 系统在高负载下的表现。

观察系统在高负载下的稳定性和性能变化。

5.3 测试指标

5.3.1 吞吐量（Throughput）：

单位：Gbps 或 MB/s。

高吞吐量表明系统能够在单位时间内传输大量数据。

5.3.2 延迟（Latency）：

单位：微秒（µs）或毫秒（ms）。

低延迟表明系统能够快速传输数据，适用于实时应用。

5.3.3 资源使用率（Resource Utilization）：

包括 CPU 使用率、内存占用、网络带宽利用率等。

低资源使用率表明系统在高效运行。

5.4 实验结果分析

通过上述方法和指标，可以得到如下结论：

1.高吞吐量：如果 RDMA 系统在吞吐量测试中表现优异，传输速率显著高于传统 TCP/IP 网络，则表明 RDMA 系统能够在单位时间内传输更多的数据。

2. 低延迟：如果 RDMA 系统在延迟测试中表现出显著低于传统网络的延迟，则表明 RDMA 系统能够较快地传输数据，适用于对时间敏感的应用场景。

3. 低资源消耗：如果 RDMA 系统在资源使用率分析中表现出较低的 CPU 和内存消耗，则表明 RDMA 系统在处理数据传输时较为高效。

5.5 实验示例

  • 环境：两台配置相同的服务器，连接在同一 RDMA 支持的网络中。
  • 基准测试：
    • 使用 TCP/IP 进行数据传输，记录吞吐量和延迟。
    • 使用 RDMA 进行相同的数据传输，记录吞吐量和延迟。
  • 结果：
    • TCP/IP 吞吐量：5 Gbps；延迟：500 微秒。
    • RDMA 吞吐量：20 Gbps；延迟：50 微秒。
  • 资源使用率：
    • TCP/IP CPU 使用率：80%；内存占用：1 GB。
    • RDMA CPU 使用率：20%；内存占用：500 MB。

通过对比，可以得出结论：RDMA 系统在吞吐量上有显著提升，延迟大幅降低，同时资源消耗也更低。这些结果验证了 RDMA 在高性能和低延迟场景中的应用价值。

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/weixin_40725706/article/detail/944939
推荐阅读
相关标签

Copyright © 2003-2013 www.wpsshop.cn 版权所有，并保留所有权利。

  

闽ICP备14008679号