simple rdma example

最新推荐文章于 2025-05-16 20:33:02 发布

mounter625

最新推荐文章于 2025-05-16 20:33:02 发布

阅读量411

点赞数 1

CC 4.0 BY-SA版权

分类专栏： Linux kernel

本文链接：https://blog.youkuaiyun.com/mounter625/article/details/104034683

Linux kernel 专栏收录该内容

427 篇文章

订阅专栏

本文介绍了一个使用InfiniBand库(libverbs)进行内存区域注册的C语言示例程序。通过创建一个结构体reg_mr来封装注册内存所需的各种参数，并在主函数中初始化该结构体并调用write函数将数据写入uverbs设备文件。示例展示了如何打开InfiniBand设备、设置注册请求参数、获取设备文件描述符以及在前后改变有效组ID的情况下执行写操作。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#include <stdio.h>
#include <infiniband/verbs.h>

struct reg_mr {
uint32_t command;
uint16_t in_words;
uint16_t out_words;
uint64_t response;
uint64_t start;
uint64_t length;
uint64_t hca_va;
uint32_t pd_handle;
uint32_t access_flags;
uint64_t driver_data[0];
};

int main() {
struct ibv_device *dev;
struct ibv_context *context;
char *dev_name="mlx5_1";
struct ibv_device **dev_list;
dev_list = ibv_get_device_list(NULL);
struct reg_mr test_buffer = {0};
int i;

for (i = 0; dev_list[i]; ++i) {
if (!strcmp(ibv_get_device_name(dev_list[i]),dev_name))
break;
}
ibv_free_device_list(dev_list);
context = ibv_open_device(dev_list[i]);

test_buffer.command = 1;
test_buffer.in_words = 12;
test_buffer.out_words = 44;
test_buffer.response = malloc(256);

int fd;
printf("pid: %d\n", getpid());
printf("run \"sudo ls -l /proc/%d/fd | grep uverbs | awk '{print $9}'\" to get fd #\n", getpid());
printf("input uverbs fd:");
scanf("%d", &fd);

printf("Before: write(%d, &test_buffer, 48) = %d\n", fd, write(fd, &test_buffer, 48));

printf("setegid(0):\n\n");
setegid(0);
printf("After: write(%d, &test_buffer, 48) = %d\n", fd, write(fd, &test_buffer, 48));

return 0;
}

Makefile:

all:

gcc -o 1 1.c -libverbs