网站建设推广途径,电商学习网站,大的网站建设公司好,个人怎样申请网站RDMA编程实践
本文描述了RDMA编程过程中的SEND-RECEIVE双边原语的代码实现。包含多个版本#xff0c;1、client向server发送消息#xff0c;server回复client收到消息(ACK)#xff0c;然后两边断开连接。2、server端循环等待客户端建立连接#xff0c;client发送一次消息后…RDMA编程实践
本文描述了RDMA编程过程中的SEND-RECEIVE双边原语的代码实现。包含多个版本1、client向server发送消息server回复client收到消息(ACK)然后两边断开连接。2、server端循环等待客户端建立连接client发送一次消息后双方断开连接。3、server端循环等待客户端建立连接一旦建立client端可以一直向server端发送消息直到发送消息为disconnectserver和client断开链接但是server此时仍然可以等待别的client发送消息。 代码基于代码基于send-receive样例实现。关于代码注释可以参考代码解释 Makefile文件、会编译当前目录下的所有.c文件
.PHONY: all cleanCC : gcc
CFLAGS : -Wall -g
LDLIBS : -lrdmacm -libverbs -lpthread -gSRCS : $(wildcard *.c)
APPS : $(SRCS:.c)all: $(APPS)%: %.c$(CC) $(CFLAGS) $ -o $ $(LDLIBS)clean:rm -f $(APPS)version1 客户端-服务端消息一次传递
在这个阶段我们希望能实现下面这样一个场景。client与server端相连接client端能够发送一条消息给serverserver收到该条消息之后恢复一条消息给client端表示我已经确认收到。之后两者断开连接。
代码
// client1.c
#include stdio.h
#include stdlib.h
#include string.h
#include netdb.h
#include errno.h
#include getopt.h
#include rdma/rdma_cma.h
#include rdma/rdma_verbs.hstatic const char *server 10.10.10.1;
static const char *port 7471;static struct rdma_cm_id *id;
static struct ibv_mr *mr, *send_mr;
static int send_flags;
static uint8_t send_msg[16];
static uint8_t recv_msg[16];static int run(void)
{struct rdma_addrinfo hints, *res;struct ibv_qp_init_attr attr;struct ibv_wc wc;int ret;memset(hints, 0, sizeof hints);hints.ai_port_space RDMA_PS_TCP;ret rdma_getaddrinfo(server, port, hints, res);if (ret) {printf(rdma_getaddrinfo: %s\n, gai_strerror(ret));goto out;}memset(attr, 0, sizeof attr);attr.cap.max_send_wr attr.cap.max_recv_wr 1;attr.cap.max_send_sge attr.cap.max_recv_sge 1;attr.cap.max_inline_data 16;attr.qp_context id;attr.sq_sig_all 1;ret rdma_create_ep(id, res, NULL, attr);// Check to see if we got inline data allowed or notif (attr.cap.max_inline_data 16)send_flags IBV_SEND_INLINE;elseprintf(rdma_client: device doesnt support IBV_SEND_INLINE, using sge sends\n);if (ret) {perror(rdma_create_ep);goto out_free_addrinfo;}mr rdma_reg_msgs(id, recv_msg, 16);if (!mr) {perror(rdma_reg_msgs for recv_msg);ret -1;goto out_destroy_ep;}if ((send_flags IBV_SEND_INLINE) 0) {send_mr rdma_reg_msgs(id, send_msg, 16);if (!send_mr) {perror(rdma_reg_msgs for send_msg);ret -1;goto out_dereg_recv;}}ret rdma_post_recv(id, NULL, recv_msg, 16, mr);if (ret) {perror(rdma_post_recv);goto out_dereg_send;}ret rdma_connect(id, NULL);if (ret) {perror(rdma_connect);goto out_dereg_send;}printf(client send: %s\n, (char *)send_msg);ret rdma_post_send(id, NULL, send_msg, 16, send_mr, send_flags);if (ret) {perror(rdma_post_send);goto out_disconnect;}while ((ret rdma_get_send_comp(id, wc)) 0);if (ret 0) {perror(rdma_get_send_comp);goto out_disconnect;}while ((ret rdma_get_recv_comp(id, wc)) 0);if (ret 0)perror(rdma_get_recv_comp);elseret 0;printf(client received: %s\n, (char *) recv_msg);out_disconnect:rdma_disconnect(id);
out_dereg_send:if ((send_flags IBV_SEND_INLINE) 0)rdma_dereg_mr(send_mr);
out_dereg_recv:rdma_dereg_mr(mr);
out_destroy_ep:rdma_destroy_ep(id);
out_free_addrinfo:rdma_freeaddrinfo(res);
out:return ret;
}int main(int argc, char **argv)
{int ret;char *s hello world;// printf(client send: %s\n, s);memcpy(send_msg, s , strlen(s));printf(rdma_client: start\n);ret run();printf(rdma_client: end %d\n, ret);return ret;
}server端代码
// server1.c
/** Copyright (c) 2005-2009 Intel Corporation. All rights reserved.** This software is available to you under the OpenIB.org BSD license* below:** Redistribution and use in source and binary forms, with or* without modification, are permitted provided that the following* conditions are met:** - Redistributions of source code must retain the above* copyright notice, this list of conditions and the following* disclaimer.** - Redistributions in binary form must reproduce the above* copyright notice, this list of conditions and the following* disclaimer in the documentation and/or other materials* provided with the distribution.** THE SOFTWARE IS PROVIDED AS IS, WITHOUT WARRANTY OF ANY KIND,* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AWV* NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS* BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN* ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN* CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE* SOFTWARE.*/#include stdio.h
#include stdlib.h
#include string.h
#include errno.h
#include getopt.h
#include netdb.h
#include rdma/rdma_cma.h
#include rdma/rdma_verbs.hstatic const char *server 0.0.0.0;
static const char *port 7471;static struct rdma_cm_id *listen_id, *id;
static struct ibv_mr *mr, *send_mr;
static int send_flags;
static uint8_t send_msg[16];
static uint8_t recv_msg[16];static int run(void)
{struct rdma_addrinfo hints, *res;struct ibv_qp_init_attr init_attr;struct ibv_qp_attr qp_attr;struct ibv_wc wc;int ret;memset(hints, 0, sizeof hints);hints.ai_flags RAI_PASSIVE;hints.ai_port_space RDMA_PS_TCP;ret rdma_getaddrinfo(server, port, hints, res);if (ret) {printf(rdma_getaddrinfo: %s\n, gai_strerror(ret));return ret;}memset(init_attr, 0, sizeof init_attr);init_attr.cap.max_send_wr init_attr.cap.max_recv_wr 1;init_attr.cap.max_send_sge init_attr.cap.max_recv_sge 1;init_attr.cap.max_inline_data 16;init_attr.sq_sig_all 1;ret rdma_create_ep(listen_id, res, NULL, init_attr);if (ret) {perror(rdma_create_ep);goto out_free_addrinfo;}ret rdma_listen(listen_id, 0);if (ret) {perror(rdma_listen);goto out_destroy_listen_ep;}ret rdma_get_request(listen_id, id);if (ret) {perror(rdma_get_request);goto out_destroy_listen_ep;}memset(qp_attr, 0, sizeof qp_attr);memset(init_attr, 0, sizeof init_attr);ret ibv_query_qp(id-qp, qp_attr, IBV_QP_CAP,init_attr);if (ret) {perror(ibv_query_qp);goto out_destroy_accept_ep;}if (init_attr.cap.max_inline_data 16)send_flags IBV_SEND_INLINE;elseprintf(rdma_server: device doesnt support IBV_SEND_INLINE, using sge sends\n);mr rdma_reg_msgs(id, recv_msg, 16);if (!mr) {ret -1;perror(rdma_reg_msgs for recv_msg);goto out_destroy_accept_ep;}if ((send_flags IBV_SEND_INLINE) 0) {send_mr rdma_reg_msgs(id, send_msg, 16);if (!send_mr) {ret -1;perror(rdma_reg_msgs for send_msg);goto out_dereg_recv;}}ret rdma_post_recv(id, NULL, recv_msg, 16, mr);if (ret) {perror(rdma_post_recv);goto out_dereg_send;}ret rdma_accept(id, NULL);if (ret) {perror(rdma_accept);goto out_dereg_send;}while ((ret rdma_get_recv_comp(id, wc)) 0);if (ret 0) {perror(rdma_get_recv_comp);goto out_disconnect;}printf(server received: %s\n , (char *)recv_msg);char *s ACK;memcpy(send_msg, s, strlen(s));printf(server send: %s\n, (char *)send_msg);ret rdma_post_send(id, NULL, send_msg, 16, send_mr, send_flags);if (ret) {perror(rdma_post_send);goto out_disconnect;}while ((ret rdma_get_send_comp(id, wc)) 0);if (ret 0)perror(rdma_get_send_comp);elseret 0;out_disconnect:rdma_disconnect(id);
out_dereg_send:if ((send_flags IBV_SEND_INLINE) 0)rdma_dereg_mr(send_mr);
out_dereg_recv:rdma_dereg_mr(mr);
out_destroy_accept_ep:rdma_destroy_ep(id);
out_destroy_listen_ep:rdma_destroy_ep(listen_id);
out_free_addrinfo:rdma_freeaddrinfo(res);return ret;
}int main(int argc, char **argv)
{int ret;printf(rdma_server: start\n);ret run();printf(rdma_server: end %d\n, ret);return ret;
}首先make编译完之后在server端执行 ./server1然后在客户端执行./client1 运行结果 可以看到 client向server发送了hello worldserver收到之后打印出来并回复给client端ACK消息client收到之后并打印。最后双方断开连接完成
version2-客户端发送一次服务端循环等待
client2的代码跟上面一样server2代码不一样。 server2的逻辑在run函数进来之后记录一个connect点当远程客户端发送完信息后释放连接的资源跳转到connect阶段准备让下一个client连接。
/** Copyright (c) 2005-2009 Intel Corporation. All rights reserved.** This software is available to you under the OpenIB.org BSD license* below:** Redistribution and use in source and binary forms, with or* without modification, are permitted provided that the following* conditions are met:** - Redistributions of source code must retain the above* copyright notice, this list of conditions and the following* disclaimer.** - Redistributions in binary form must reproduce the above* copyright notice, this list of conditions and the following* disclaimer in the documentation and/or other materials* provided with the distribution.** THE SOFTWARE IS PROVIDED AS IS, WITHOUT WARRANTY OF ANY KIND,* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AWV* NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS* BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN* ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN* CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE* SOFTWARE.*/#include stdio.h
#include stdlib.h
#include string.h
#include errno.h
#include getopt.h
#include netdb.h
#include rdma/rdma_cma.h
#include rdma/rdma_verbs.h#define N 100
#define MAX_CAP 32static const char *server 0.0.0.0;
static const char *port 7471;static struct rdma_cm_id *listen_id, *id;
static struct ibv_mr *mr, *send_mr;
static int send_flags;
static uint8_t send_msg[MAX_CAP];
static uint8_t recv_msg[MAX_CAP];static int run(void)
{struct rdma_addrinfo hints, *res;struct ibv_qp_init_attr init_attr;struct ibv_qp_attr qp_attr;struct ibv_wc wc;int ret;while(1){memset(hints, 0, sizeof hints);hints.ai_flags RAI_PASSIVE;hints.ai_port_space RDMA_PS_TCP;ret rdma_getaddrinfo(server, port, hints, res);if (ret) {printf(rdma_getaddrinfo: %s\n, gai_strerror(ret));return ret;}memset(init_attr, 0, sizeof init_attr);init_attr.cap.max_send_wr init_attr.cap.max_recv_wr N;init_attr.cap.max_send_sge init_attr.cap.max_recv_sge 1;init_attr.cap.max_inline_data MAX_CAP;init_attr.sq_sig_all 1;ret rdma_create_ep(listen_id, res, NULL, init_attr);if (ret) {perror(rdma_create_ep);goto out_free_addrinfo;}ret rdma_listen(listen_id, 0);if (ret) {perror(rdma_listen);goto out_destroy_listen_ep;}ret rdma_get_request(listen_id, id);if (ret) {perror(rdma_get_request);goto out_destroy_listen_ep;}memset(qp_attr, 0, sizeof qp_attr);memset(init_attr, 0, sizeof init_attr);ret ibv_query_qp(id-qp, qp_attr, IBV_QP_CAP,init_attr);if (ret) {perror(ibv_query_qp);goto out_destroy_accept_ep;}if (init_attr.cap.max_inline_data MAX_CAP)send_flags IBV_SEND_INLINE;elseprintf(rdma_server: device doesnt support IBV_SEND_INLINE, using sge sends\n);mr rdma_reg_msgs(id, recv_msg, N);if (!mr) {ret -1;perror(rdma_reg_msgs for recv_msg);goto out_destroy_accept_ep;}if ((send_flags IBV_SEND_INLINE) 0) {send_mr rdma_reg_msgs(id, send_msg, MAX_CAP);if (!send_mr) {ret -1;perror(rdma_reg_msgs for send_msg);goto out_dereg_recv;}} ret rdma_accept(id, NULL);if (ret) {perror(rdma_accept);goto out_dereg_send;}memset(recv_msg, 0 , sizeof recv_msg);memset(send_msg, 0 , sizeof send_msg);ret rdma_post_recv(id, NULL, recv_msg, MAX_CAP, mr);if (ret) {perror(rdma_post_recv);goto out_dereg_send;}while ((ret rdma_get_recv_comp(id, wc)) 0);if (ret 0) {perror(rdma_get_recv_comp);goto out_disconnect;}printf(server received: %s\n, (char *)recv_msg);memcpy(send_msg, recv_msg, sizeof(recv_msg));ret rdma_post_send(id, NULL, send_msg, MAX_CAP, send_mr, send_flags);if (ret) {perror(rdma_post_send);goto out_disconnect;}while ((ret rdma_get_send_comp(id, wc)) 0); // 确认对方已经收到 对方会发送ackif (ret 0)perror(rdma_get_send_comp);elseret 0;rdma_disconnect(id);if ((send_flags IBV_SEND_INLINE) 0)rdma_dereg_mr(send_mr);rdma_dereg_mr(mr);rdma_destroy_ep(id);rdma_destroy_ep(listen_id);rdma_freeaddrinfo(res); }out_disconnect:rdma_disconnect(id);
out_dereg_send:if ((send_flags IBV_SEND_INLINE) 0)rdma_dereg_mr(send_mr);
out_dereg_recv:rdma_dereg_mr(mr);
out_destroy_accept_ep:rdma_destroy_ep(id);
out_destroy_listen_ep:rdma_destroy_ep(listen_id);
out_free_addrinfo:rdma_freeaddrinfo(res); return ret;
}int main(int argc, char **argv)
{int ret;printf(rdma_server: start\n);ret run();printf(rdma_server: end %d\n, ret);return ret;
}运行结果 可以看到客户端发送一次消息之后便结束了服务端却一直等待连接直到按下ctrlc。
version3-客户端循环发送服务端循环等待一次连接
和上述版本2不同的时候这里client和server只连接一次然后可以多次发送消息。直到client发送的消息为disconnect
// client3.c
/** Copyright (c) 2010 Intel Corporation. All rights reserved.** This software is available to you under the OpenIB.org BSD license* below:** Redistribution and use in source and binary forms, with or* without modification, are permitted provided that the following* conditions are met:** - Redistributions of source code must retain the above* copyright notice, this list of conditions and the following* disclaimer.** - Redistributions in binary form must reproduce the above* copyright notice, this list of conditions and the following* disclaimer in the documentation and/or other materials* provided with the distribution.** THE SOFTWARE IS PROVIDED AS IS, WITHOUT WARRANTY OF ANY KIND,* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AWV* NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS* BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN* ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN* CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE* SOFTWARE.*/#include stdio.h
#include stdlib.h
#include string.h
#include netdb.h
#include errno.h
#include getopt.h
#include rdma/rdma_cma.h
#include rdma/rdma_verbs.h#define N 100
#define MAX_CAP 32static const char *server 10.10.10.1;
static const char *port 7471;static struct rdma_cm_id *id;
static struct ibv_mr *mr, *send_mr;
static int send_flags;
static uint8_t send_msg[MAX_CAP];
static uint8_t recv_msg[MAX_CAP];static int run(void)
{struct rdma_addrinfo hints, *res;struct ibv_qp_init_attr attr;struct ibv_wc wc;int ret;memset(hints, 0, sizeof hints);hints.ai_port_space RDMA_PS_TCP;ret rdma_getaddrinfo(server, port, hints, res);if (ret) {printf(rdma_getaddrinfo: %s\n, gai_strerror(ret));goto out;}memset(attr, 0, sizeof attr);attr.cap.max_send_wr attr.cap.max_recv_wr 5;attr.cap.max_send_sge attr.cap.max_recv_sge 1;attr.cap.max_inline_data MAX_CAP;attr.qp_context id;attr.sq_sig_all 1;ret rdma_create_ep(id, res, NULL, attr);// Check to see if we got inline data allowed or notif (attr.cap.max_inline_data MAX_CAP)send_flags IBV_SEND_INLINE;elseprintf(rdma_client: device doesnt support IBV_SEND_INLINE, using sge sends\n);if (ret) {perror(rdma_create_ep);goto out_free_addrinfo;}mr rdma_reg_msgs(id, recv_msg, MAX_CAP);if (!mr) {perror(rdma_reg_msgs for recv_msg);ret -1;goto out_destroy_ep;}if ((send_flags IBV_SEND_INLINE) 0) {send_mr rdma_reg_msgs(id, send_msg, MAX_CAP);if (!send_mr) {perror(rdma_reg_msgs for send_msg);ret -1;goto out_dereg_recv;}}// ret rdma_post_recv(id, NULL, recv_msg, 16, mr);// if (ret) {// perror(rdma_post_recv);// goto out_dereg_send;// }// printf(123\n);ret rdma_connect(id, NULL);if (ret) {perror(rdma_connect);goto out_dereg_send;}while(1){// sleep(5);memset(recv_msg, 0 , sizeof recv_msg);memset(send_msg, 0 , sizeof send_msg);printf(input send message: );scanf(%s, send_msg);getchar();ret rdma_post_recv(id, NULL, recv_msg, MAX_CAP, mr);if (ret) {perror(rdma_post_recv);goto out_dereg_send;}ret rdma_post_send(id, NULL, send_msg, MAX_CAP, send_mr, send_flags);if (ret) {perror(rdma_post_send);goto out_disconnect;}while ((ret rdma_get_send_comp(id, wc)) 0);if (ret 0) {perror(rdma_get_send_comp);goto out_disconnect;}while ((ret rdma_get_recv_comp(id, wc)) 0);if (ret 0)perror(rdma_get_recv_comp);elseret 0;if(strcmp((char*)send_msg,disconnect) 0){printf(disconnect\n);goto out_disconnect;}else{printf(%s\n, recv_msg);}}out_disconnect:rdma_disconnect(id);
out_dereg_send:if ((send_flags IBV_SEND_INLINE) 0)rdma_dereg_mr(send_mr);
out_dereg_recv:rdma_dereg_mr(mr);
out_destroy_ep:rdma_destroy_ep(id);
out_free_addrinfo:rdma_freeaddrinfo(res);
out:return ret;}int main(int argc, char **argv)
{int ret;//memcpy(send_msg, argv[1], 50);// while ((op getopt(argc, argv, s:p:)) ! -1) {// switch (op) {// case s:// server optarg;// break;// case p:// port optarg;// break;// default:// printf(usage: %s\n, argv[0]);// printf(\t[-s server_address]\n);// printf(\t[-p port_number]\n);// exit(1);// }// }printf(rdma_client: start\n);ret run();printf(rdma_client: end %d\n, ret);return ret;
}// server3.c
/** Copyright (c) 2005-2009 Intel Corporation. All rights reserved.** This software is available to you under the OpenIB.org BSD license* below:** Redistribution and use in source and binary forms, with or* without modification, are permitted provided that the following* conditions are met:** - Redistributions of source code must retain the above* copyright notice, this list of conditions and the following* disclaimer.** - Redistributions in binary form must reproduce the above* copyright notice, this list of conditions and the following* disclaimer in the documentation and/or other materials* provided with the distribution.** THE SOFTWARE IS PROVIDED AS IS, WITHOUT WARRANTY OF ANY KIND,* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AWV* NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS* BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN* ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN* CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE* SOFTWARE.*/#include stdio.h
#include stdlib.h
#include string.h
#include errno.h
#include getopt.h
#include netdb.h
#include rdma/rdma_cma.h
#include rdma/rdma_verbs.h#define N 100
#define MAX_CAP 32static const char *server 0.0.0.0;
static const char *port 7471;static struct rdma_cm_id *listen_id, *id;
static struct ibv_mr *mr, *send_mr;
static int send_flags;
static uint8_t send_msg[MAX_CAP];
static uint8_t recv_msg[MAX_CAP];static int run(void)
{struct rdma_addrinfo hints, *res;struct ibv_qp_init_attr init_attr;struct ibv_qp_attr qp_attr;struct ibv_wc wc;int ret;connect:memset(hints, 0, sizeof hints);hints.ai_flags RAI_PASSIVE;hints.ai_port_space RDMA_PS_TCP;ret rdma_getaddrinfo(server, port, hints, res);if (ret) {printf(rdma_getaddrinfo: %s\n, gai_strerror(ret));return ret;}memset(init_attr, 0, sizeof init_attr);init_attr.cap.max_send_wr init_attr.cap.max_recv_wr N;init_attr.cap.max_send_sge init_attr.cap.max_recv_sge 1;init_attr.cap.max_inline_data MAX_CAP;init_attr.sq_sig_all 1;ret rdma_create_ep(listen_id, res, NULL, init_attr);if (ret) {perror(rdma_create_ep);goto out_free_addrinfo;}ret rdma_listen(listen_id, 0);if (ret) {perror(rdma_listen);goto out_destroy_listen_ep;}ret rdma_get_request(listen_id, id);if (ret) {perror(rdma_get_request);goto out_destroy_listen_ep;}memset(qp_attr, 0, sizeof qp_attr);memset(init_attr, 0, sizeof init_attr);ret ibv_query_qp(id-qp, qp_attr, IBV_QP_CAP,init_attr);if (ret) {perror(ibv_query_qp);goto out_destroy_accept_ep;}if (init_attr.cap.max_inline_data MAX_CAP)send_flags IBV_SEND_INLINE;elseprintf(rdma_server: device doesnt support IBV_SEND_INLINE, using sge sends\n);mr rdma_reg_msgs(id, recv_msg, N);if (!mr) {ret -1;perror(rdma_reg_msgs for recv_msg);goto out_destroy_accept_ep;}if ((send_flags IBV_SEND_INLINE) 0) {send_mr rdma_reg_msgs(id, send_msg, MAX_CAP);if (!send_mr) {ret -1;perror(rdma_reg_msgs for send_msg);goto out_dereg_recv;}} ret rdma_accept(id, NULL);if (ret) {perror(rdma_accept);goto out_dereg_send;}while (1) {memset(recv_msg, 0 , sizeof recv_msg);memset(send_msg, 0 , sizeof send_msg);ret rdma_post_recv(id, NULL, recv_msg, MAX_CAP, mr);if (ret) {perror(rdma_post_recv);goto out_dereg_send;}while ((ret rdma_get_recv_comp(id, wc)) 0);if (ret 0) {perror(rdma_get_recv_comp);goto out_disconnect;}char *s (char *)recv_msg;int total_length strlen(server get ) strlen(s); // 加1是为了存储字符串结束符\0char *recv_str (char *)malloc(total_length); // 分配足够的空间strcpy(recv_str, server get );strcat(recv_str, s);//printf(%s\n, recv_str);memcpy(send_msg, recv_str, strlen(recv_str));ret rdma_post_send(id, NULL, send_msg, MAX_CAP, send_mr, send_flags);if (ret) {perror(rdma_post_send);goto out_disconnect;}if(strcmp((char*)recv_msg,disconnect) 0){//printf(%s\n,recv_msg);printf(client disconnect\n);rdma_disconnect(id);if ((send_flags IBV_SEND_INLINE) 0)rdma_dereg_mr(send_mr);rdma_dereg_mr(mr);rdma_destroy_ep(id);rdma_destroy_ep(listen_id);rdma_freeaddrinfo(res); //goto out_disconnect;goto connect;}else{printf(%s\n, recv_msg);}// while ((ret rdma_get_send_comp(id, wc)) 0); // 确认对方已经收到 对方发送ack// printf(after send\n);// if (ret 0)// perror(rdma_get_send_comp);// else// ret 0;}out_disconnect:rdma_disconnect(id);
out_dereg_send:if ((send_flags IBV_SEND_INLINE) 0)rdma_dereg_mr(send_mr);
out_dereg_recv:rdma_dereg_mr(mr);
out_destroy_accept_ep:rdma_destroy_ep(id);
out_destroy_listen_ep:rdma_destroy_ep(listen_id);
out_free_addrinfo:rdma_freeaddrinfo(res); return ret;
}int main(int argc, char **argv)
{int op, ret;while ((op getopt(argc, argv, s:p:)) ! -1) {switch (op) {case s:server optarg;break;case p:port optarg;break;default:printf(usage: %s\n, argv[0]);printf(\t[-s server_address]\n);printf(\t[-p port_number]\n);exit(1);}}printf(rdma_server: start\n);ret run();printf(rdma_server: end %d\n, ret);return ret;
}运行结果 总结
本文实现了rdma中send-receive双边原语的三种需求版本从单次发送到两者都能多次发送。理解其中的代码逻辑想要发送消息之前对端得创建一个recv队列用来接收消息。发送完了有一个发送完成队列接收完了也有一个接收完成队列。最后双方断开连接需要一起断开不能某一方执行disconnect另一方不执行。本次实验有一个关键点
while ((ret rdma_get_send_comp(id, wc)) 0)这一行代码是等待发送成功发送成功之后对方会给一个隐式信息表示我已经收到。这里耗费得时间比较长一点在版本3中如果不注释掉在server端的receive队列还没有建立好这就导致client发送了消息server还没有收到双方就陷入了死循环中。
