/*
* 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:
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;
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)
{
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;
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;
if (ret)
rclose(lrs);
free:
- freeaddrinfo(res);
+ if (rai)
+ rdma_freeaddrinfo(rai);
+ else
+ freeaddrinfo(ai);
return ret;
}
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;
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;
if (ret)
rclose(rs);
free:
- freeaddrinfo(res);
+ if (rai)
+ rdma_freeaddrinfo(rai);
+ else
+ freeaddrinfo(ai);
return ret;
}
{
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;
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;
+ rai_hints.ai_family = AF_IB;
+ use_rgai = 1;
+ } else {
+ fprintf(stderr, "Warning: unknown address format\n");
+ }
+ break;
case 'B':
buffer_size = atoi(optarg);
break;
printf("usage: %s\n", argv[0]);
printf("\t[-s server_address]\n");
printf("\t[-b bind_address]\n");
+ printf("\t[-f address_format]\n");
+ printf("\t name, ip, ipv6, or gid\n");
printf("\t[-B buffer_size]\n");
printf("\t[-I iterations]\n");
printf("\t[-C transfer_count]\n");