Allow the user to specify GID addresses (AF_IB) into riostream example.
Signed-off-by: Hal Rosenstock
---
Change since v1:
Added brief description
diff --git a/examples/riostream.c b/examples/riostream.c
index a1d3671..4c0d21f 100644
--- a/examples/riostream.c
+++ b/examples/riostream.c
@@ -1,5 +1,6 @@
/*
* Copyright (c) 2011-2012 Intel Corporation. All rights reserved.
+ * Copyright (c) 2014 Mellanox Technologies LTD. All rights reserved.
*
* This software is available to you under the OpenIB.org BSD license
* below:
@@ -75,6 +76,7 @@ static struct test_size_param test_size[] = {
static int rs, lrs;
static int use_async;
+static int use_rgai;
static int verify;
static int flags = MSG_DONTWAIT;
static int poll_timeout = 0;
@@ -92,6 +94,8 @@ static char *src_addr;
static struct timeval start, end;
static void *buf;
static volatile uint8_t *poll_byte;
+static struct rdma_addrinfo rai_hints;
+static struct addrinfo ai_hints;
static void show_perf(void)
{
@@ -355,18 +359,24 @@ static void set_options(int rs)
static int server_listen(void)
{
- struct addrinfo hints, *res;
+ struct rdma_addrinfo *rai = NULL;
+ struct addrinfo *ai;
int val, ret;
- memset(&hints, 0, sizeof hints);
- hints.ai_flags = AI_PASSIVE;
- ret = getaddrinfo(src_addr, port, &hints, &res);
+ if (use_rgai) {
+ rai_hints.ai_flags |= RAI_PASSIVE;
+ ret = rdma_getaddrinfo(src_addr, port, &rai_hints, &rai);
+ } else {
+ ai_hints.ai_flags |= AI_PASSIVE;
+ ret = getaddrinfo(src_addr, port, &ai_hints, &ai);
+ }
if (ret) {
perror("getaddrinfo");
return ret;
}
- lrs = rsocket(res->ai_family, res->ai_socktype, res->ai_protocol);
+ lrs = rai ? rsocket(rai->ai_family, SOCK_STREAM, 0) :
+ rsocket(ai->ai_family, SOCK_STREAM, 0);
if (lrs < 0) {
perror("rsocket");
ret = lrs;
@@ -380,7 +390,8 @@ static int server_listen(void)
goto close;
}
- ret = rbind(lrs, res->ai_addr, res->ai_addrlen);
+ ret = rai ? rbind(lrs, rai->ai_src_addr, rai->ai_src_len) :
+ rbind(lrs, ai->ai_addr, ai->ai_addrlen);
if (ret) {
perror("rbind");
goto close;
@@ -394,7 +405,10 @@ close:
if (ret)
rclose(lrs);
free:
- freeaddrinfo(res);
+ if (rai)
+ rdma_freeaddrinfo(rai);
+ else
+ freeaddrinfo(ai);
return ret;
}
@@ -429,18 +443,21 @@ static int server_connect(void)
static int client_connect(void)
{
- struct addrinfo *res;
+ struct rdma_addrinfo *rai = NULL;
+ struct addrinfo *ai;
struct pollfd fds;
int ret, err;
socklen_t len;
- ret = getaddrinfo(dst_addr, port, NULL, &res);
+ ret = use_rgai ? rdma_getaddrinfo(dst_addr, port, &rai_hints, &rai) :
+getaddrinfo(dst_addr, port, &ai_hints, &ai);
if (ret) {
perror("getaddrinfo");
return ret;
}
- rs = rsocket(res->ai_family, res->ai_socktype, res->ai_protocol);
+ rs = rai ? rsocket(rai->ai_family, SOCK_STREAM, 0) :
+ rsocket(ai->ai_family, SOCK_STREAM, 0);
if (rs < 0) {
perror("rsocket");
ret = rs;
@@ -450,7 +467,8 @@ static int client_connect(void)
set_options(rs);
/* TODO: bind client to src_addr */
- ret = rconnect(rs, res->ai_addr, res->ai_addrlen);
+ ret = rai ? rconnect(rs, rai->ai_dst_addr, rai->ai_dst_len) :
+ rconnect(rs, ai->ai_addr, ai->ai_addrlen);
if (ret && (errno != EINPROGRESS)) {
perror("rconnect");
goto close;
@@ -478,7 +496,10 @@ close:
if (ret)
rclose(rs);
free:
- freeaddrinfo(res);
+ if (rai)
+ rdma_freeaddrinfo(rai);
+ else
+ freeaddrinfo(ai);
return ret;
}
@@ -579,7 +600,9 @@ int main(int argc, char **argv)
{
int op, ret;
- while ((op = getopt(argc, argv, "s:b:B:I:C:S:p:T:")) != -1) {
+ ai_hints.ai_socktype = SOCK_STREAM;
+ rai_hints.ai_port_space = RDMA_PS_TCP;
+ while ((op = getopt(argc, argv, "s:b:f:B:I:C:S:p:T:")) != -1) {
switch (op) {
case 's':
dst_addr = optarg;
@@ -587,6 +610,17 @@ int main(int argc, char **argv)
case 'b':
src_addr = optarg;
break;
+ case 'f':
+ if (!strncasecmp("ip", optarg, 2)) {
+ ai_hints.ai_flags = AI_NUMERICHOST;
+ } else if (!strncasecmp("gid", optarg, 3)) {
+ rai_hints.ai_flags = RAI_NUMERICHOST |
RAI_FAMILY;
+