X-Git-Url: https://git.kernel.dk/?p=fio.git;a=blobdiff_plain;f=engines%2Frdma.c;h=10e60dc8449b7634ee1949e051289f55a606eac6;hp=a847b541d3a0927dd3b575fa5a38fb3dc5832ff4;hb=c58390119ccf711ab66e53aff6aa31507c509900;hpb=836fcc0fceb233ebcc41ee63b4ea5cae20b678a4 diff --git a/engines/rdma.c b/engines/rdma.c index a847b541..10e60dc8 100644 --- a/engines/rdma.c +++ b/engines/rdma.c @@ -5,12 +5,7 @@ * Supports both RDMA memory semantics and channel semantics * for the InfiniBand, RoCE and iWARP protocols. * - * This I/O engine is disabled by default. To enable it, execute: - * - * $ export EXTFLAGS+=" -DFIO_HAVE_RDMA " - * $ export EXTLIBS+=" -libverbs -lrdmacm " - * - * before running make. You will need the Linux RDMA software as well, either + * You will need the Linux RDMA software installed, either * from your Linux distributor or directly from openfabrics.org: * * http://www.openfabrics.org/downloads/OFED/ @@ -41,12 +36,12 @@ #include #include -#include #include #include #include "../fio.h" #include "../hash.h" +#include "../optgroup.h" #include #include @@ -61,6 +56,77 @@ enum rdma_io_mode { FIO_RDMA_CHA_RECV }; +struct rdmaio_options { + struct thread_data *td; + unsigned int port; + enum rdma_io_mode verb; +}; + +static int str_hostname_cb(void *data, const char *input) +{ + struct rdmaio_options *o = data; + + if (o->td->o.filename) + free(o->td->o.filename); + o->td->o.filename = strdup(input); + return 0; +} + +static struct fio_option options[] = { + { + .name = "hostname", + .lname = "rdma engine hostname", + .type = FIO_OPT_STR_STORE, + .cb = str_hostname_cb, + .help = "Hostname for RDMA IO engine", + .category = FIO_OPT_C_ENGINE, + .group = FIO_OPT_G_RDMA, + }, + { + .name = "port", + .lname = "rdma engine port", + .type = FIO_OPT_INT, + .off1 = offsetof(struct rdmaio_options, port), + .minval = 1, + .maxval = 65535, + .help = "Port to use for RDMA connections", + .category = FIO_OPT_C_ENGINE, + .group = FIO_OPT_G_RDMA, + }, + { + .name = "verb", + .lname = "RDMA engine verb", + .alias = "proto", + .type = FIO_OPT_STR, + .off1 = offsetof(struct rdmaio_options, verb), + .help = "RDMA engine verb", + .def = "write", + .posval = { + { .ival = "write", + .oval = FIO_RDMA_MEM_WRITE, + .help = "Memory Write", + }, + { .ival = "read", + .oval = FIO_RDMA_MEM_READ, + .help = "Memory Read", + }, + { .ival = "send", + .oval = FIO_RDMA_CHA_SEND, + .help = "Posted Send", + }, + { .ival = "recv", + .oval = FIO_RDMA_CHA_RECV, + .help = "Posted Receive", + }, + }, + .category = FIO_OPT_C_ENGINE, + .group = FIO_OPT_G_RDMA, + }, + { + .name = NULL, + }, +}; + struct remote_u { uint64_t buf; uint32_t rkey; @@ -72,6 +138,7 @@ struct rdma_info_blk { uint32_t nr; /* client: io depth server: number of records for memory semantic */ + uint32_t max_bs; /* maximum block size */ struct remote_u rmt_us[FIO_RDMA_MAX_IO_DEPTH]; }; @@ -124,13 +191,22 @@ struct rdmaio_data { static int client_recv(struct thread_data *td, struct ibv_wc *wc) { - struct rdmaio_data *rd = td->io_ops->data; + struct rdmaio_data *rd = td->io_ops_data; + unsigned int max_bs; if (wc->byte_len != sizeof(rd->recv_buf)) { log_err("Received bogus data, size %d\n", wc->byte_len); return 1; } + max_bs = max(td->o.max_bs[DDIR_READ], td->o.max_bs[DDIR_WRITE]); + if (max_bs > ntohl(rd->recv_buf.max_bs)) { + log_err("fio: Server's block size (%d) must be greater than or " + "equal to the client's block size (%d)!\n", + ntohl(rd->recv_buf.max_bs), max_bs); + return 1; + } + /* store mr info for MEMORY semantic */ if ((rd->rdma_protocol == FIO_RDMA_MEM_WRITE) || (rd->rdma_protocol == FIO_RDMA_MEM_READ)) { @@ -156,7 +232,8 @@ static int client_recv(struct thread_data *td, struct ibv_wc *wc) static int server_recv(struct thread_data *td, struct ibv_wc *wc) { - struct rdmaio_data *rd = td->io_ops->data; + struct rdmaio_data *rd = td->io_ops_data; + unsigned int max_bs; if (wc->wr_id == FIO_RDMA_MAX_IO_DEPTH) { rd->rdma_protocol = ntohl(rd->recv_buf.mode); @@ -164,6 +241,15 @@ static int server_recv(struct thread_data *td, struct ibv_wc *wc) /* CHANNEL semantic, do nothing */ if (rd->rdma_protocol == FIO_RDMA_CHA_SEND) rd->rdma_protocol = FIO_RDMA_CHA_RECV; + + max_bs = max(td->o.max_bs[DDIR_READ], td->o.max_bs[DDIR_WRITE]); + if (max_bs < ntohl(rd->recv_buf.max_bs)) { + log_err("fio: Server's block size (%d) must be greater than or " + "equal to the client's block size (%d)!\n", + ntohl(rd->recv_buf.max_bs), max_bs); + return 1; + } + } return 0; @@ -171,7 +257,7 @@ static int server_recv(struct thread_data *td, struct ibv_wc *wc) static int cq_event_handler(struct thread_data *td, enum ibv_wc_opcode opcode) { - struct rdmaio_data *rd = td->io_ops->data; + struct rdmaio_data *rd = td->io_ops_data; struct ibv_wc wc; struct rdma_io_u_data *r_io_u_d; int ret; @@ -192,9 +278,12 @@ static int cq_event_handler(struct thread_data *td, enum ibv_wc_opcode opcode) case IBV_WC_RECV: if (rd->is_client == 1) - client_recv(td, &wc); + ret = client_recv(td, &wc); else - server_recv(td, &wc); + ret = server_recv(td, &wc); + + if (ret) + return -1; if (wc.wr_id == FIO_RDMA_MAX_IO_DEPTH) break; @@ -264,6 +353,7 @@ static int cq_event_handler(struct thread_data *td, enum ibv_wc_opcode opcode) } rd->cq_event_num++; } + if (ret) { log_err("fio: poll error %d\n", ret); return 1; @@ -278,7 +368,7 @@ static int cq_event_handler(struct thread_data *td, enum ibv_wc_opcode opcode) */ static int rdma_poll_wait(struct thread_data *td, enum ibv_wc_opcode opcode) { - struct rdmaio_data *rd = td->io_ops->data; + struct rdmaio_data *rd = td->io_ops_data; struct ibv_cq *ev_cq; void *ev_ctx; int ret; @@ -303,7 +393,7 @@ again: } ret = cq_event_handler(td, opcode); - if (ret < 1) + if (ret == 0) goto again; ibv_ack_cq_events(rd->cq, ret); @@ -315,7 +405,7 @@ again: static int fio_rdmaio_setup_qp(struct thread_data *td) { - struct rdmaio_data *rd = td->io_ops->data; + struct rdmaio_data *rd = td->io_ops_data; struct ibv_qp_init_attr init_attr; int qp_depth = td->o.iodepth * 2; /* 2 times of io depth */ @@ -325,7 +415,7 @@ static int fio_rdmaio_setup_qp(struct thread_data *td) rd->pd = ibv_alloc_pd(rd->cm_id->verbs); if (rd->pd == NULL) { - log_err("fio: ibv_alloc_pd fail\n"); + log_err("fio: ibv_alloc_pd fail: %m\n"); return 1; } @@ -334,7 +424,7 @@ static int fio_rdmaio_setup_qp(struct thread_data *td) else rd->channel = ibv_create_comp_channel(rd->cm_id->verbs); if (rd->channel == NULL) { - log_err("fio: ibv_create_comp_channel fail\n"); + log_err("fio: ibv_create_comp_channel fail: %m\n"); goto err1; } @@ -348,12 +438,12 @@ static int fio_rdmaio_setup_qp(struct thread_data *td) rd->cq = ibv_create_cq(rd->cm_id->verbs, qp_depth, rd, rd->channel, 0); if (rd->cq == NULL) { - log_err("fio: ibv_create_cq failed\n"); + log_err("fio: ibv_create_cq failed: %m\n"); goto err2; } if (ibv_req_notify_cq(rd->cq, 0) != 0) { - log_err("fio: ibv_create_cq failed\n"); + log_err("fio: ibv_req_notify_cq failed: %m\n"); goto err3; } @@ -369,13 +459,13 @@ static int fio_rdmaio_setup_qp(struct thread_data *td) if (rd->is_client == 0) { if (rdma_create_qp(rd->child_cm_id, rd->pd, &init_attr) != 0) { - log_err("fio: rdma_create_qp failed\n"); + log_err("fio: rdma_create_qp failed: %m\n"); goto err3; } rd->qp = rd->child_cm_id->qp; } else { if (rdma_create_qp(rd->cm_id, rd->pd, &init_attr) != 0) { - log_err("fio: rdma_create_qp failed\n"); + log_err("fio: rdma_create_qp failed: %m\n"); goto err3; } rd->qp = rd->cm_id->qp; @@ -395,19 +485,19 @@ err1: static int fio_rdmaio_setup_control_msg_buffers(struct thread_data *td) { - struct rdmaio_data *rd = td->io_ops->data; + struct rdmaio_data *rd = td->io_ops_data; rd->recv_mr = ibv_reg_mr(rd->pd, &rd->recv_buf, sizeof(rd->recv_buf), IBV_ACCESS_LOCAL_WRITE); if (rd->recv_mr == NULL) { - log_err("fio: recv_buf reg_mr failed\n"); + log_err("fio: recv_buf reg_mr failed: %m\n"); return 1; } rd->send_mr = ibv_reg_mr(rd->pd, &rd->send_buf, sizeof(rd->send_buf), 0); if (rd->send_mr == NULL) { - log_err("fio: send_buf reg_mr failed\n"); + log_err("fio: send_buf reg_mr failed: %m\n"); ibv_dereg_mr(rd->recv_mr); return 1; } @@ -439,7 +529,7 @@ static int get_next_channel_event(struct thread_data *td, struct rdma_event_channel *channel, enum rdma_cm_event_type wait_event) { - struct rdmaio_data *rd = td->io_ops->data; + struct rdmaio_data *rd = td->io_ops_data; struct rdma_cm_event *event; int ret; @@ -471,7 +561,7 @@ static int get_next_channel_event(struct thread_data *td, static int fio_rdmaio_prep(struct thread_data *td, struct io_u *io_u) { - struct rdmaio_data *rd = td->io_ops->data; + struct rdmaio_data *rd = td->io_ops_data; struct rdma_io_u_data *r_io_u_d; r_io_u_d = io_u->engine_data; @@ -514,7 +604,7 @@ static int fio_rdmaio_prep(struct thread_data *td, struct io_u *io_u) static struct io_u *fio_rdmaio_event(struct thread_data *td, int event) { - struct rdmaio_data *rd = td->io_ops->data; + struct rdmaio_data *rd = td->io_ops_data; struct io_u *io_u; int i; @@ -530,14 +620,14 @@ static struct io_u *fio_rdmaio_event(struct thread_data *td, int event) } static int fio_rdmaio_getevents(struct thread_data *td, unsigned int min, - unsigned int max, struct timespec *t) + unsigned int max, const struct timespec *t) { - struct rdmaio_data *rd = td->io_ops->data; + struct rdmaio_data *rd = td->io_ops_data; enum ibv_wc_opcode comp_opcode; - comp_opcode = IBV_WC_RDMA_WRITE; struct ibv_cq *ev_cq; void *ev_ctx; int ret, r = 0; + comp_opcode = IBV_WC_RDMA_WRITE; switch (rd->rdma_protocol) { case FIO_RDMA_MEM_WRITE: @@ -594,7 +684,7 @@ again: static int fio_rdmaio_send(struct thread_data *td, struct io_u **io_us, unsigned int nr) { - struct rdmaio_data *rd = td->io_ops->data; + struct rdmaio_data *rd = td->io_ops_data; struct ibv_send_wr *bad_wr; #if 0 enum ibv_wc_opcode comp_opcode; @@ -641,7 +731,7 @@ static int fio_rdmaio_send(struct thread_data *td, struct io_u **io_us, } if (ibv_post_send(rd->qp, &r_io_u_d->sq_wr, &bad_wr) != 0) { - log_err("fio: ibv_post_send fail\n"); + log_err("fio: ibv_post_send fail: %m\n"); return -1; } @@ -657,7 +747,7 @@ static int fio_rdmaio_send(struct thread_data *td, struct io_u **io_us, static int fio_rdmaio_recv(struct thread_data *td, struct io_u **io_us, unsigned int nr) { - struct rdmaio_data *rd = td->io_ops->data; + struct rdmaio_data *rd = td->io_ops_data; struct ibv_recv_wr *bad_wr; struct rdma_io_u_data *r_io_u_d; int i; @@ -669,7 +759,7 @@ static int fio_rdmaio_recv(struct thread_data *td, struct io_u **io_us, r_io_u_d = io_us[i]->engine_data; if (ibv_post_recv(rd->qp, &r_io_u_d->rq_wr, &bad_wr) != 0) { - log_err("fio: ibv_post_recv fail\n"); + log_err("fio: ibv_post_recv fail: %m\n"); return 1; } } @@ -677,7 +767,7 @@ static int fio_rdmaio_recv(struct thread_data *td, struct io_u **io_us, || (rd->rdma_protocol == FIO_RDMA_MEM_WRITE)) { /* re-post the rq_wr */ if (ibv_post_recv(rd->qp, &rd->rq_wr, &bad_wr) != 0) { - log_err("fio: ibv_post_recv fail\n"); + log_err("fio: ibv_post_recv fail: %m\n"); return 1; } @@ -693,7 +783,7 @@ static int fio_rdmaio_recv(struct thread_data *td, struct io_u **io_us, static int fio_rdmaio_queue(struct thread_data *td, struct io_u *io_u) { - struct rdmaio_data *rd = td->io_ops->data; + struct rdmaio_data *rd = td->io_ops_data; fio_ro_check(td, io_u); @@ -711,7 +801,7 @@ static int fio_rdmaio_queue(struct thread_data *td, struct io_u *io_u) static void fio_rdmaio_queued(struct thread_data *td, struct io_u **io_us, unsigned int nr) { - struct rdmaio_data *rd = td->io_ops->data; + struct rdmaio_data *rd = td->io_ops_data; struct timeval now; unsigned int i; @@ -734,7 +824,7 @@ static void fio_rdmaio_queued(struct thread_data *td, struct io_u **io_us, static int fio_rdmaio_commit(struct thread_data *td) { - struct rdmaio_data *rd = td->io_ops->data; + struct rdmaio_data *rd = td->io_ops_data; struct io_u **io_us; int ret; @@ -766,7 +856,7 @@ static int fio_rdmaio_commit(struct thread_data *td) static int fio_rdmaio_connect(struct thread_data *td, struct fio_file *f) { - struct rdmaio_data *rd = td->io_ops->data; + struct rdmaio_data *rd = td->io_ops_data; struct rdma_conn_param conn_param; struct ibv_send_wr *bad_wr; @@ -776,7 +866,7 @@ static int fio_rdmaio_connect(struct thread_data *td, struct fio_file *f) conn_param.retry_count = 10; if (rdma_connect(rd->cm_id, &conn_param) != 0) { - log_err("fio: rdma_connect fail\n"); + log_err("fio: rdma_connect fail: %m\n"); return 1; } @@ -791,14 +881,16 @@ static int fio_rdmaio_connect(struct thread_data *td, struct fio_file *f) rd->send_buf.nr = htonl(td->o.iodepth); if (ibv_post_send(rd->qp, &rd->sq_wr, &bad_wr) != 0) { - log_err("fio: ibv_post_send fail"); + log_err("fio: ibv_post_send fail: %m\n"); return 1; } - rdma_poll_wait(td, IBV_WC_SEND); + if (rdma_poll_wait(td, IBV_WC_SEND) < 0) + return 1; /* wait for remote MR info from server side */ - rdma_poll_wait(td, IBV_WC_RECV); + if (rdma_poll_wait(td, IBV_WC_RECV) < 0) + return 1; /* In SEND/RECV test, it's a good practice to setup the iodepth of * of the RECV side deeper than that of the SEND side to @@ -815,9 +907,10 @@ static int fio_rdmaio_connect(struct thread_data *td, struct fio_file *f) static int fio_rdmaio_accept(struct thread_data *td, struct fio_file *f) { - struct rdmaio_data *rd = td->io_ops->data; + struct rdmaio_data *rd = td->io_ops_data; struct rdma_conn_param conn_param; struct ibv_send_wr *bad_wr; + int ret = 0; /* rdma_accept() - then wait for accept success */ memset(&conn_param, 0, sizeof(conn_param)); @@ -825,7 +918,7 @@ static int fio_rdmaio_accept(struct thread_data *td, struct fio_file *f) conn_param.initiator_depth = 1; if (rdma_accept(rd->child_cm_id, &conn_param) != 0) { - log_err("fio: rdma_accept\n"); + log_err("fio: rdma_accept: %m\n"); return 1; } @@ -836,16 +929,17 @@ static int fio_rdmaio_accept(struct thread_data *td, struct fio_file *f) } /* wait for request */ - rdma_poll_wait(td, IBV_WC_RECV); + ret = rdma_poll_wait(td, IBV_WC_RECV) < 0; if (ibv_post_send(rd->qp, &rd->sq_wr, &bad_wr) != 0) { - log_err("fio: ibv_post_send fail"); + log_err("fio: ibv_post_send fail: %m\n"); return 1; } - rdma_poll_wait(td, IBV_WC_SEND); + if (rdma_poll_wait(td, IBV_WC_SEND) < 0) + return 1; - return 0; + return ret; } static int fio_rdmaio_open_file(struct thread_data *td, struct fio_file *f) @@ -858,7 +952,7 @@ static int fio_rdmaio_open_file(struct thread_data *td, struct fio_file *f) static int fio_rdmaio_close_file(struct thread_data *td, struct fio_file *f) { - struct rdmaio_data *rd = td->io_ops->data; + struct rdmaio_data *rd = td->io_ops_data; struct ibv_send_wr *bad_wr; /* unregister rdma buffer */ @@ -871,11 +965,11 @@ static int fio_rdmaio_close_file(struct thread_data *td, struct fio_file *f) || (rd->rdma_protocol == FIO_RDMA_MEM_READ))) { if (ibv_post_send(rd->qp, &rd->sq_wr, &bad_wr) != 0) { - log_err("fio: ibv_post_send fail"); + log_err("fio: ibv_post_send fail: %m\n"); return 1; } - dprint(FD_IO, "fio: close infomation sent success\n"); + dprint(FD_IO, "fio: close information sent success\n"); rdma_poll_wait(td, IBV_WC_SEND); } @@ -914,7 +1008,7 @@ static int fio_rdmaio_close_file(struct thread_data *td, struct fio_file *f) static int fio_rdmaio_setup_connect(struct thread_data *td, const char *host, unsigned short port) { - struct rdmaio_data *rd = td->io_ops->data; + struct rdmaio_data *rd = td->io_ops_data; struct ibv_recv_wr *bad_wr; int err; @@ -978,8 +1072,11 @@ static int fio_rdmaio_setup_connect(struct thread_data *td, const char *host, static int fio_rdmaio_setup_listen(struct thread_data *td, short port) { - struct rdmaio_data *rd = td->io_ops->data; + struct rdmaio_data *rd = td->io_ops_data; struct ibv_recv_wr *bad_wr; + int state = td->runstate; + + td_set_runstate(td, TD_SETTING_UP); rd->addr.sin_family = AF_INET; rd->addr.sin_addr.s_addr = htonl(INADDR_ANY); @@ -987,15 +1084,17 @@ static int fio_rdmaio_setup_listen(struct thread_data *td, short port) /* rdma_listen */ if (rdma_bind_addr(rd->cm_id, (struct sockaddr *)&rd->addr) != 0) { - log_err("fio: rdma_bind_addr fail\n"); + log_err("fio: rdma_bind_addr fail: %m\n"); return 1; } if (rdma_listen(rd->cm_id, 3) != 0) { - log_err("fio: rdma_listen fail\n"); + log_err("fio: rdma_listen fail: %m\n"); return 1; } + log_info("fio: waiting for connection\n"); + /* wait for CONNECT_REQUEST */ if (get_next_channel_event (td, rd->cm_channel, RDMA_CM_EVENT_CONNECT_REQUEST) != 0) { @@ -1011,33 +1110,19 @@ static int fio_rdmaio_setup_listen(struct thread_data *td, short port) /* post recv buf */ if (ibv_post_recv(rd->qp, &rd->rq_wr, &bad_wr) != 0) { - log_err("fio: ibv_post_recv fail\n"); + log_err("fio: ibv_post_recv fail: %m\n"); return 1; } + td_set_runstate(td, state); return 0; } -static int fio_rdmaio_init(struct thread_data *td) +static int check_set_rlimits(struct thread_data *td) { - struct rdmaio_data *rd = td->io_ops->data; - struct flist_head *entry; - unsigned int max_bs; - unsigned int port; - char host[64], buf[128]; - char *sep, *portp, *modep; - int ret, i = 0; +#ifdef CONFIG_RLIMIT_MEMLOCK struct rlimit rl; - if (td_rw(td)) { - log_err("fio: rdma connections must be read OR write\n"); - return 1; - } - if (td_random(td)) { - log_err("fio: RDMA network IO can't be random\n"); - return 1; - } - /* check RLIMIT_MEMLOCK */ if (getrlimit(RLIMIT_MEMLOCK, &rl) != 0) { log_err("fio: getrlimit fail: %d(%s)\n", @@ -1062,57 +1147,104 @@ static int fio_rdmaio_init(struct thread_data *td) return 1; } } +#endif - strcpy(buf, td->o.filename); + return 0; +} - sep = strchr(buf, '/'); - if (!sep) - goto bad_host; +static int compat_options(struct thread_data *td) +{ + // The original RDMA engine had an ugly / seperator + // on the filename for it's options. This function + // retains backwards compatibility with it.100 - *sep = '\0'; - sep++; - strcpy(host, buf); - if (!strlen(host)) - goto bad_host; + struct rdmaio_options *o = td->eo; + char *modep, *portp; + char *filename = td->o.filename; - modep = NULL; - portp = sep; - sep = strchr(portp, '/'); - if (sep) { - *sep = '\0'; - modep = sep + 1; - } + if (!filename) + return 0; + + portp = strchr(filename, '/'); + if (portp == NULL) + return 0; - port = strtol(portp, NULL, 10); - if (!port || port > 65535) + *portp = '\0'; + portp++; + + o->port = strtol(portp, NULL, 10); + if (!o->port || o->port > 65535) goto bad_host; + modep = strchr(portp, '/'); + if (modep != NULL) { + *modep = '\0'; + modep++; + } + if (modep) { if (!strncmp("rdma_write", modep, strlen(modep)) || !strncmp("RDMA_WRITE", modep, strlen(modep))) - rd->rdma_protocol = FIO_RDMA_MEM_WRITE; + o->verb = FIO_RDMA_MEM_WRITE; else if (!strncmp("rdma_read", modep, strlen(modep)) || !strncmp("RDMA_READ", modep, strlen(modep))) - rd->rdma_protocol = FIO_RDMA_MEM_READ; + o->verb = FIO_RDMA_MEM_READ; else if (!strncmp("send", modep, strlen(modep)) || !strncmp("SEND", modep, strlen(modep))) - rd->rdma_protocol = FIO_RDMA_CHA_SEND; + o->verb = FIO_RDMA_CHA_SEND; else goto bad_host; } else - rd->rdma_protocol = FIO_RDMA_MEM_WRITE; + o->verb = FIO_RDMA_MEM_WRITE; + + + return 0; + +bad_host: + log_err("fio: bad rdma host/port/protocol: %s\n", td->o.filename); + return 1; +} + +static int fio_rdmaio_init(struct thread_data *td) +{ + struct rdmaio_data *rd = td->io_ops_data; + struct rdmaio_options *o = td->eo; + unsigned int max_bs; + int ret, i; + + if (td_rw(td)) { + log_err("fio: rdma connections must be read OR write\n"); + return 1; + } + if (td_random(td)) { + log_err("fio: RDMA network IO can't be random\n"); + return 1; + } + + if (compat_options(td)) + return 1; + if (!o->port) { + log_err("fio: no port has been specified which is required " + "for the rdma engine\n"); + return 1; + } + + if (check_set_rlimits(td)) + return 1; + + rd->rdma_protocol = o->verb; rd->cq_event_num = 0; rd->cm_channel = rdma_create_event_channel(); if (!rd->cm_channel) { - log_err("fio: rdma_create_event_channel fail\n"); + log_err("fio: rdma_create_event_channel fail: %m\n"); return 1; } ret = rdma_create_id(rd->cm_channel, &rd->cm_id, rd, RDMA_PS_TCP); if (ret) { - log_err("fio: rdma_create_id fail\n"); + log_err("fio: rdma_create_id fail: %m\n"); return 1; } @@ -1139,17 +1271,20 @@ static int fio_rdmaio_init(struct thread_data *td) if (td_read(td)) { /* READ as the server */ rd->is_client = 0; + td->flags |= TD_F_NO_PROGRESS; /* server rd->rdma_buf_len will be setup after got request */ - ret = fio_rdmaio_setup_listen(td, port); + ret = fio_rdmaio_setup_listen(td, o->port); } else { /* WRITE as the client */ rd->is_client = 1; - ret = fio_rdmaio_setup_connect(td, host, port); + ret = fio_rdmaio_setup_connect(td, td->o.filename, o->port); } max_bs = max(td->o.max_bs[DDIR_READ], td->o.max_bs[DDIR_WRITE]); + rd->send_buf.max_bs = htonl(max_bs); + /* register each io_u in the free list */ - flist_for_each(entry, &td->io_u_freelist) { - struct io_u *io_u = flist_entry(entry, struct io_u, list); + for (i = 0; i < td->io_u_freelist.nr; i++) { + struct io_u *io_u = td->io_u_freelist.io_us[i]; io_u->engine_data = malloc(sizeof(struct rdma_io_u_data)); memset(io_u->engine_data, 0, sizeof(struct rdma_io_u_data)); @@ -1160,7 +1295,7 @@ static int fio_rdmaio_init(struct thread_data *td) IBV_ACCESS_REMOTE_READ | IBV_ACCESS_REMOTE_WRITE); if (io_u->mr == NULL) { - log_err("fio: ibv_reg_mr io_u failed\n"); + log_err("fio: ibv_reg_mr io_u failed: %m\n"); return 1; } @@ -1172,20 +1307,16 @@ static int fio_rdmaio_init(struct thread_data *td) #if 0 log_info("fio: Send rkey %x addr %" PRIx64 " len %d to client\n", io_u->mr->rkey, io_u->buf, max_bs); */ #endif - i++; } rd->send_buf.nr = htonl(i); return ret; -bad_host: - log_err("fio: bad rdma host/port/protocol: %s\n", td->o.filename); - return 1; } static void fio_rdmaio_cleanup(struct thread_data *td) { - struct rdmaio_data *rd = td->io_ops->data; + struct rdmaio_data *rd = td->io_ops_data; if (rd) free(rd); @@ -1195,31 +1326,39 @@ static int fio_rdmaio_setup(struct thread_data *td) { struct rdmaio_data *rd; - if (!td->io_ops->data) { + if (!td->files_index) { + add_file(td, td->o.filename ?: "rdma", 0, 0); + td->o.nr_files = td->o.nr_files ?: 1; + td->o.open_files++; + } + + if (!td->io_ops_data) { rd = malloc(sizeof(*rd)); memset(rd, 0, sizeof(*rd)); - init_rand_seed(&rd->rand_state, (unsigned int) GOLDEN_RATIO_PRIME); - td->io_ops->data = rd; + init_rand_seed(&rd->rand_state, (unsigned int) GOLDEN_RATIO_PRIME, 0); + td->io_ops_data = rd; } return 0; } static struct ioengine_ops ioengine_rw = { - .name = "rdma", - .version = FIO_IOOPS_VERSION, - .setup = fio_rdmaio_setup, - .init = fio_rdmaio_init, - .prep = fio_rdmaio_prep, - .queue = fio_rdmaio_queue, - .commit = fio_rdmaio_commit, - .getevents = fio_rdmaio_getevents, - .event = fio_rdmaio_event, - .cleanup = fio_rdmaio_cleanup, - .open_file = fio_rdmaio_open_file, - .close_file = fio_rdmaio_close_file, - .flags = FIO_DISKLESSIO | FIO_UNIDIR | FIO_PIPEIO, + .name = "rdma", + .version = FIO_IOOPS_VERSION, + .setup = fio_rdmaio_setup, + .init = fio_rdmaio_init, + .prep = fio_rdmaio_prep, + .queue = fio_rdmaio_queue, + .commit = fio_rdmaio_commit, + .getevents = fio_rdmaio_getevents, + .event = fio_rdmaio_event, + .cleanup = fio_rdmaio_cleanup, + .open_file = fio_rdmaio_open_file, + .close_file = fio_rdmaio_close_file, + .flags = FIO_DISKLESSIO | FIO_UNIDIR | FIO_PIPEIO, + .options = options, + .option_struct_size = sizeof(struct rdmaio_options), }; static void fio_init fio_rdmaio_register(void)