使用RDMA的服务器上的客户端IP地址？

使用 RDMA（Remote Direct Memory Access）技术可以显著提高网络通信的性能，特别是在高性能计算（HPC）和数据中心环境中。RDMA 允许直接从一台计算机的内存访问另一台计算机的内存，而无需经过操作系统的干预，从而减少了延迟和 CPU 负载。

在使用 RDMA 的服务器上获取客户端的 IP 地址，通常涉及以下几个步骤：

配置 RDMA 环境：确保 RDMA 硬件和软件环境已经正确配置。
编写 RDMA 应用程序：使用 RDMA 编程接口（如 libibverbs 或 RDMA CM）编写应用程序。
获取客户端 IP 地址：在 RDMA 连接建立过程中，获取客户端的 IP 地址。

以下是一个使用 RDMA CM（Connection Manager）库的示例，展示了如何在服务器端获取客户端的 IP 地址。

安装 RDMA 库

首先，确保你已经安装了 RDMA 库。你可以使用以下命令在基于 Debian 的系统上安装：

sudo apt-get install rdma-core ibverbs-utils

编写 RDMA 服务器代码

以下是一个简单的 RDMA 服务器示例，展示了如何获取客户端的 IP 地址：

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <rdma/rdma_cma.h>

void die(const char *reason) {
    fprintf(stderr, "%s\n", reason);
    exit(EXIT_FAILURE);
}

int main(int argc, char **argv) {
    struct rdma_cm_id *listener = NULL;
    struct rdma_cm_id *client = NULL;
    struct rdma_event_channel *ec = NULL;
    struct rdma_cm_event *event = NULL;
    struct sockaddr_in addr;
    int ret;

    // 创建事件通道
    ec = rdma_create_event_channel();
    if (!ec) die("rdma_create_event_channel failed");

    // 创建 RDMA 监听器
    ret = rdma_create_id(ec, &listener, NULL, RDMA_PS_TCP);
    if (ret) die("rdma_create_id failed");

    // 绑定地址
    memset(&addr, 0, sizeof(addr));
    addr.sin_family = AF_INET;
    addr.sin_port = htons(20000); // 监听端口
    addr.sin_addr.s_addr = INADDR_ANY;

    ret = rdma_bind_addr(listener, (struct sockaddr *)&addr);
    if (ret) die("rdma_bind_addr failed");

    // 开始监听
    ret = rdma_listen(listener, 1);
    if (ret) die("rdma_listen failed");

    printf("Server is listening on port 20000...\n");

    // 等待连接请求
    ret = rdma_get_cm_event(ec, &event);
    if (ret) die("rdma_get_cm_event failed");

    if (event->event == RDMA_CM_EVENT_CONNECT_REQUEST) {
        client = event->id;

        // 获取客户端地址
        struct sockaddr_in *client_addr = (struct sockaddr_in *)rdma_get_peer_addr(client);
        char client_ip[INET_ADDRSTRLEN];
        inet_ntop(AF_INET, &(client_addr->sin_addr), client_ip, INET_ADDRSTRLEN);
        printf("Client connected from IP: %s\n", client_ip);

        // 接受连接
        struct rdma_conn_param conn_param;
        memset(&conn_param, 0, sizeof(conn_param));
        ret = rdma_accept(client, &conn_param);
        if (ret) die("rdma_accept failed");
    }

    // 清理
    rdma_ack_cm_event(event);
    rdma_destroy_id(listener);
    rdma_destroy_event_channel(ec);

    return 0;
}

编译和运行

编译上述代码：

gcc -o rdma_server rdma_server.c -lrdmacm -libverbs

运行服务器：

./rdma_server

解释

创建事件通道： ec = rdma_create_event_channel(); if (!ec) die("rdma_create_event_channel failed");
创建 RDMA 监听器： ret = rdma_create_id(ec, &listener, NULL, RDMA_PS_TCP); if (ret) die("rdma_create_id failed");
绑定地址： memset(&addr, 0, sizeof(addr)); addr.sin_family = AF_INET; addr.sin_port = htons(20000); // 监听端口 addr.sin_addr.s_addr = INADDR_ANY; ret = rdma_bind_addr(listener, (struct sockaddr *)&addr); if (ret) die("rdma_bind_addr failed");
开始监听： ret = rdma_listen(listener, 1); if (ret) die("rdma_listen failed");
等待连接请求： ret = rdma_get_cm_event(ec, &event); if (ret) die("rdma_get_cm_event failed");
获取客户端地址： if (event->event == RDMA_CM_EVENT_CONNECT_REQUEST) { client = event->id; struct sockaddr_in *client_addr = (struct sockaddr_in *)rdma_get_peer_addr(client); char client_ip[INET_ADDRSTRLEN]; inet_ntop(AF_INET, &(client_addr->sin_addr), client_ip, INET_ADDRSTRLEN); printf("Client connected from IP: %s\n", client_ip); struct rdma_conn_param conn_param; memset(&conn_param, 0, sizeof(conn_param)); ret = rdma_accept(client, &conn_param); if (ret) die("rdma_accept failed"); }