Use the correct printf formats, so that a 32-bit compile doesn't spit out lots of warnings about %lx being incompatible with uint64_t. Suggested-by: Eric Blake <eblake@redhat.com> Signed-off-by: Yuval Shaia <yuval.shaia@oracle.com> Reviewed-by: Eric Blake <eblake@redhat.com> Tested-by: Eric Blake <eblake@redhat.com> Reviewed-by: Philippe Mathieu-Daudé <f4bug@amsat.org> Message-Id: <20180322095220.9976-4-yuval.shaia@oracle.com> Signed-off-by: Marcel Apfelbaum <marcel@redhat.com>
		
			
				
	
	
		
			832 lines
		
	
	
		
			24 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			832 lines
		
	
	
		
			24 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
/*
 | 
						|
 * QEMU paravirtual RDMA - Generic RDMA backend
 | 
						|
 *
 | 
						|
 * Copyright (C) 2018 Oracle
 | 
						|
 * Copyright (C) 2018 Red Hat Inc
 | 
						|
 *
 | 
						|
 * Authors:
 | 
						|
 *     Yuval Shaia <yuval.shaia@oracle.com>
 | 
						|
 *     Marcel Apfelbaum <marcel@redhat.com>
 | 
						|
 *
 | 
						|
 * This work is licensed under the terms of the GNU GPL, version 2 or later.
 | 
						|
 * See the COPYING file in the top-level directory.
 | 
						|
 *
 | 
						|
 */
 | 
						|
 | 
						|
#include "qemu/osdep.h"
 | 
						|
#include "qemu/error-report.h"
 | 
						|
#include "qapi/error.h"
 | 
						|
 | 
						|
#include <infiniband/verbs.h>
 | 
						|
 | 
						|
#include "trace.h"
 | 
						|
#include "rdma_utils.h"
 | 
						|
#include "rdma_rm.h"
 | 
						|
#include "rdma_backend.h"
 | 
						|
 | 
						|
/* Vendor Errors */
 | 
						|
#define VENDOR_ERR_FAIL_BACKEND     0x201
 | 
						|
#define VENDOR_ERR_TOO_MANY_SGES    0x202
 | 
						|
#define VENDOR_ERR_NOMEM            0x203
 | 
						|
#define VENDOR_ERR_QP0              0x204
 | 
						|
#define VENDOR_ERR_NO_SGE           0x205
 | 
						|
#define VENDOR_ERR_MAD_SEND         0x206
 | 
						|
#define VENDOR_ERR_INVLKEY          0x207
 | 
						|
#define VENDOR_ERR_MR_SMALL         0x208
 | 
						|
 | 
						|
#define THR_NAME_LEN 16
 | 
						|
 | 
						|
typedef struct BackendCtx {
 | 
						|
    uint64_t req_id;
 | 
						|
    void *up_ctx;
 | 
						|
    bool is_tx_req;
 | 
						|
} BackendCtx;
 | 
						|
 | 
						|
static void (*comp_handler)(int status, unsigned int vendor_err, void *ctx);
 | 
						|
 | 
						|
static void dummy_comp_handler(int status, unsigned int vendor_err, void *ctx)
 | 
						|
{
 | 
						|
    pr_err("No completion handler is registered\n");
 | 
						|
}
 | 
						|
 | 
						|
static void poll_cq(RdmaDeviceResources *rdma_dev_res, struct ibv_cq *ibcq)
 | 
						|
{
 | 
						|
    int i, ne;
 | 
						|
    BackendCtx *bctx;
 | 
						|
    struct ibv_wc wc[2];
 | 
						|
 | 
						|
    pr_dbg("Entering poll_cq loop on cq %p\n", ibcq);
 | 
						|
    do {
 | 
						|
        ne = ibv_poll_cq(ibcq, ARRAY_SIZE(wc), wc);
 | 
						|
 | 
						|
        pr_dbg("Got %d completion(s) from cq %p\n", ne, ibcq);
 | 
						|
 | 
						|
        for (i = 0; i < ne; i++) {
 | 
						|
            pr_dbg("wr_id=0x%" PRIx64 "\n", wc[i].wr_id);
 | 
						|
            pr_dbg("status=%d\n", wc[i].status);
 | 
						|
 | 
						|
            bctx = rdma_rm_get_cqe_ctx(rdma_dev_res, wc[i].wr_id);
 | 
						|
            if (unlikely(!bctx)) {
 | 
						|
                pr_dbg("Error: Failed to find ctx for req %" PRId64 "\n",
 | 
						|
                       wc[i].wr_id);
 | 
						|
                continue;
 | 
						|
            }
 | 
						|
            pr_dbg("Processing %s CQE\n", bctx->is_tx_req ? "send" : "recv");
 | 
						|
 | 
						|
            comp_handler(wc[i].status, wc[i].vendor_err, bctx->up_ctx);
 | 
						|
 | 
						|
            rdma_rm_dealloc_cqe_ctx(rdma_dev_res, wc[i].wr_id);
 | 
						|
            g_free(bctx);
 | 
						|
        }
 | 
						|
    } while (ne > 0);
 | 
						|
 | 
						|
    if (ne < 0) {
 | 
						|
        pr_dbg("Got error %d from ibv_poll_cq\n", ne);
 | 
						|
    }
 | 
						|
}
 | 
						|
 | 
						|
static void *comp_handler_thread(void *arg)
 | 
						|
{
 | 
						|
    RdmaBackendDev *backend_dev = (RdmaBackendDev *)arg;
 | 
						|
    int rc;
 | 
						|
    struct ibv_cq *ev_cq;
 | 
						|
    void *ev_ctx;
 | 
						|
 | 
						|
    pr_dbg("Starting\n");
 | 
						|
 | 
						|
    while (backend_dev->comp_thread.run) {
 | 
						|
        pr_dbg("Waiting for completion on channel %p\n", backend_dev->channel);
 | 
						|
        rc = ibv_get_cq_event(backend_dev->channel, &ev_cq, &ev_ctx);
 | 
						|
        pr_dbg("ibv_get_cq_event=%d\n", rc);
 | 
						|
        if (unlikely(rc)) {
 | 
						|
            pr_dbg("---> ibv_get_cq_event (%d)\n", rc);
 | 
						|
            continue;
 | 
						|
        }
 | 
						|
 | 
						|
        rc = ibv_req_notify_cq(ev_cq, 0);
 | 
						|
        if (unlikely(rc)) {
 | 
						|
            pr_dbg("Error %d from ibv_req_notify_cq\n", rc);
 | 
						|
        }
 | 
						|
 | 
						|
        poll_cq(backend_dev->rdma_dev_res, ev_cq);
 | 
						|
 | 
						|
        ibv_ack_cq_events(ev_cq, 1);
 | 
						|
    }
 | 
						|
 | 
						|
    pr_dbg("Going down\n");
 | 
						|
 | 
						|
    /* TODO: Post cqe for all remaining buffs that were posted */
 | 
						|
 | 
						|
    return NULL;
 | 
						|
}
 | 
						|
 | 
						|
void rdma_backend_register_comp_handler(void (*handler)(int status,
 | 
						|
                                        unsigned int vendor_err, void *ctx))
 | 
						|
{
 | 
						|
    comp_handler = handler;
 | 
						|
}
 | 
						|
 | 
						|
void rdma_backend_unregister_comp_handler(void)
 | 
						|
{
 | 
						|
    rdma_backend_register_comp_handler(dummy_comp_handler);
 | 
						|
}
 | 
						|
 | 
						|
int rdma_backend_query_port(RdmaBackendDev *backend_dev,
 | 
						|
                            struct ibv_port_attr *port_attr)
 | 
						|
{
 | 
						|
    int rc;
 | 
						|
 | 
						|
    rc = ibv_query_port(backend_dev->context, backend_dev->port_num, port_attr);
 | 
						|
    if (rc) {
 | 
						|
        pr_dbg("Error %d from ibv_query_port\n", rc);
 | 
						|
        return -EIO;
 | 
						|
    }
 | 
						|
 | 
						|
    return 0;
 | 
						|
}
 | 
						|
 | 
						|
void rdma_backend_poll_cq(RdmaDeviceResources *rdma_dev_res, RdmaBackendCQ *cq)
 | 
						|
{
 | 
						|
    poll_cq(rdma_dev_res, cq->ibcq);
 | 
						|
}
 | 
						|
 | 
						|
static GHashTable *ah_hash;
 | 
						|
 | 
						|
static struct ibv_ah *create_ah(RdmaBackendDev *backend_dev, struct ibv_pd *pd,
 | 
						|
                                uint8_t sgid_idx, union ibv_gid *dgid)
 | 
						|
{
 | 
						|
    GBytes *ah_key = g_bytes_new(dgid, sizeof(*dgid));
 | 
						|
    struct ibv_ah *ah = g_hash_table_lookup(ah_hash, ah_key);
 | 
						|
 | 
						|
    if (ah) {
 | 
						|
        trace_create_ah_cache_hit(be64_to_cpu(dgid->global.subnet_prefix),
 | 
						|
                                  be64_to_cpu(dgid->global.interface_id));
 | 
						|
        g_bytes_unref(ah_key);
 | 
						|
    } else {
 | 
						|
        struct ibv_ah_attr ah_attr = {
 | 
						|
            .is_global     = 1,
 | 
						|
            .port_num      = backend_dev->port_num,
 | 
						|
            .grh.hop_limit = 1,
 | 
						|
        };
 | 
						|
 | 
						|
        ah_attr.grh.dgid = *dgid;
 | 
						|
        ah_attr.grh.sgid_index = sgid_idx;
 | 
						|
 | 
						|
        ah = ibv_create_ah(pd, &ah_attr);
 | 
						|
        if (ah) {
 | 
						|
            g_hash_table_insert(ah_hash, ah_key, ah);
 | 
						|
        } else {
 | 
						|
            g_bytes_unref(ah_key);
 | 
						|
            pr_dbg("Fail to create AH for gid <0x%" PRIx64 ", 0x%" PRIx64 ">\n",
 | 
						|
                    be64_to_cpu(dgid->global.subnet_prefix),
 | 
						|
                    be64_to_cpu(dgid->global.interface_id));
 | 
						|
        }
 | 
						|
 | 
						|
        trace_create_ah_cache_miss(be64_to_cpu(dgid->global.subnet_prefix),
 | 
						|
                                   be64_to_cpu(dgid->global.interface_id));
 | 
						|
    }
 | 
						|
 | 
						|
    return ah;
 | 
						|
}
 | 
						|
 | 
						|
static void destroy_ah_hash_key(gpointer data)
 | 
						|
{
 | 
						|
    g_bytes_unref(data);
 | 
						|
}
 | 
						|
 | 
						|
static void destroy_ah_hast_data(gpointer data)
 | 
						|
{
 | 
						|
    struct ibv_ah *ah = data;
 | 
						|
 | 
						|
    ibv_destroy_ah(ah);
 | 
						|
}
 | 
						|
 | 
						|
static void ah_cache_init(void)
 | 
						|
{
 | 
						|
    ah_hash = g_hash_table_new_full(g_bytes_hash, g_bytes_equal,
 | 
						|
                                    destroy_ah_hash_key, destroy_ah_hast_data);
 | 
						|
}
 | 
						|
 | 
						|
static int build_host_sge_array(RdmaDeviceResources *rdma_dev_res,
 | 
						|
                                struct ibv_sge *dsge, struct ibv_sge *ssge,
 | 
						|
                                uint8_t num_sge)
 | 
						|
{
 | 
						|
    RdmaRmMR *mr;
 | 
						|
    int ssge_idx;
 | 
						|
 | 
						|
    pr_dbg("num_sge=%d\n", num_sge);
 | 
						|
 | 
						|
    for (ssge_idx = 0; ssge_idx < num_sge; ssge_idx++) {
 | 
						|
        mr = rdma_rm_get_mr(rdma_dev_res, ssge[ssge_idx].lkey);
 | 
						|
        if (unlikely(!mr)) {
 | 
						|
            pr_dbg("Invalid lkey 0x%x\n", ssge[ssge_idx].lkey);
 | 
						|
            return VENDOR_ERR_INVLKEY | ssge[ssge_idx].lkey;
 | 
						|
        }
 | 
						|
 | 
						|
        dsge->addr = (uintptr_t)mr->user_mr.host_virt + ssge[ssge_idx].addr -
 | 
						|
                     mr->user_mr.guest_start;
 | 
						|
        dsge->length = ssge[ssge_idx].length;
 | 
						|
        dsge->lkey = rdma_backend_mr_lkey(&mr->backend_mr);
 | 
						|
 | 
						|
        pr_dbg("ssge->addr=0x%" PRIx64 "\n", ssge[ssge_idx].addr);
 | 
						|
        pr_dbg("dsge->addr=0x%" PRIx64 "\n", dsge->addr);
 | 
						|
        pr_dbg("dsge->length=%d\n", dsge->length);
 | 
						|
        pr_dbg("dsge->lkey=0x%x\n", dsge->lkey);
 | 
						|
 | 
						|
        dsge++;
 | 
						|
    }
 | 
						|
 | 
						|
    return 0;
 | 
						|
}
 | 
						|
 | 
						|
void rdma_backend_post_send(RdmaBackendDev *backend_dev,
 | 
						|
                            RdmaBackendQP *qp, uint8_t qp_type,
 | 
						|
                            struct ibv_sge *sge, uint32_t num_sge,
 | 
						|
                            union ibv_gid *dgid, uint32_t dqpn,
 | 
						|
                            uint32_t dqkey, void *ctx)
 | 
						|
{
 | 
						|
    BackendCtx *bctx;
 | 
						|
    struct ibv_sge new_sge[MAX_SGE];
 | 
						|
    uint32_t bctx_id;
 | 
						|
    int rc;
 | 
						|
    struct ibv_send_wr wr = {0}, *bad_wr;
 | 
						|
 | 
						|
    if (!qp->ibqp) { /* This field does not get initialized for QP0 and QP1 */
 | 
						|
        if (qp_type == IBV_QPT_SMI) {
 | 
						|
            pr_dbg("QP0 unsupported\n");
 | 
						|
            comp_handler(IBV_WC_GENERAL_ERR, VENDOR_ERR_QP0, ctx);
 | 
						|
        } else if (qp_type == IBV_QPT_GSI) {
 | 
						|
            pr_dbg("QP1\n");
 | 
						|
            comp_handler(IBV_WC_GENERAL_ERR, VENDOR_ERR_MAD_SEND, ctx);
 | 
						|
        }
 | 
						|
        pr_dbg("qp->ibqp is NULL for qp_type %d!!!\n", qp_type);
 | 
						|
        return;
 | 
						|
    }
 | 
						|
 | 
						|
    pr_dbg("num_sge=%d\n", num_sge);
 | 
						|
    if (!num_sge) {
 | 
						|
        pr_dbg("num_sge=0\n");
 | 
						|
        comp_handler(IBV_WC_GENERAL_ERR, VENDOR_ERR_NO_SGE, ctx);
 | 
						|
        return;
 | 
						|
    }
 | 
						|
 | 
						|
    bctx = g_malloc0(sizeof(*bctx));
 | 
						|
    bctx->up_ctx = ctx;
 | 
						|
    bctx->is_tx_req = 1;
 | 
						|
 | 
						|
    rc = rdma_rm_alloc_cqe_ctx(backend_dev->rdma_dev_res, &bctx_id, bctx);
 | 
						|
    if (unlikely(rc)) {
 | 
						|
        pr_dbg("Failed to allocate cqe_ctx\n");
 | 
						|
        comp_handler(IBV_WC_GENERAL_ERR, VENDOR_ERR_NOMEM, ctx);
 | 
						|
        goto out_free_bctx;
 | 
						|
    }
 | 
						|
 | 
						|
    rc = build_host_sge_array(backend_dev->rdma_dev_res, new_sge, sge, num_sge);
 | 
						|
    if (rc) {
 | 
						|
        pr_dbg("Error: Failed to build host SGE array\n");
 | 
						|
        comp_handler(IBV_WC_GENERAL_ERR, rc, ctx);
 | 
						|
        goto out_dealloc_cqe_ctx;
 | 
						|
    }
 | 
						|
 | 
						|
    if (qp_type == IBV_QPT_UD) {
 | 
						|
        wr.wr.ud.ah = create_ah(backend_dev, qp->ibpd,
 | 
						|
                                backend_dev->backend_gid_idx, dgid);
 | 
						|
        wr.wr.ud.remote_qpn = dqpn;
 | 
						|
        wr.wr.ud.remote_qkey = dqkey;
 | 
						|
    }
 | 
						|
 | 
						|
    wr.num_sge = num_sge;
 | 
						|
    wr.opcode = IBV_WR_SEND;
 | 
						|
    wr.send_flags = IBV_SEND_SIGNALED;
 | 
						|
    wr.sg_list = new_sge;
 | 
						|
    wr.wr_id = bctx_id;
 | 
						|
 | 
						|
    rc = ibv_post_send(qp->ibqp, &wr, &bad_wr);
 | 
						|
    pr_dbg("ibv_post_send=%d\n", rc);
 | 
						|
    if (rc) {
 | 
						|
        pr_dbg("Fail (%d, %d) to post send WQE to qpn %d\n", rc, errno,
 | 
						|
                qp->ibqp->qp_num);
 | 
						|
        comp_handler(IBV_WC_GENERAL_ERR, VENDOR_ERR_FAIL_BACKEND, ctx);
 | 
						|
        goto out_dealloc_cqe_ctx;
 | 
						|
    }
 | 
						|
 | 
						|
    return;
 | 
						|
 | 
						|
out_dealloc_cqe_ctx:
 | 
						|
    rdma_rm_dealloc_cqe_ctx(backend_dev->rdma_dev_res, bctx_id);
 | 
						|
 | 
						|
out_free_bctx:
 | 
						|
    g_free(bctx);
 | 
						|
}
 | 
						|
 | 
						|
void rdma_backend_post_recv(RdmaBackendDev *backend_dev,
 | 
						|
                            RdmaDeviceResources *rdma_dev_res,
 | 
						|
                            RdmaBackendQP *qp, uint8_t qp_type,
 | 
						|
                            struct ibv_sge *sge, uint32_t num_sge, void *ctx)
 | 
						|
{
 | 
						|
    BackendCtx *bctx;
 | 
						|
    struct ibv_sge new_sge[MAX_SGE];
 | 
						|
    uint32_t bctx_id;
 | 
						|
    int rc;
 | 
						|
    struct ibv_recv_wr wr = {0}, *bad_wr;
 | 
						|
 | 
						|
    if (!qp->ibqp) { /* This field does not get initialized for QP0 and QP1 */
 | 
						|
        if (qp_type == IBV_QPT_SMI) {
 | 
						|
            pr_dbg("QP0 unsupported\n");
 | 
						|
            comp_handler(IBV_WC_GENERAL_ERR, VENDOR_ERR_QP0, ctx);
 | 
						|
        }
 | 
						|
        if (qp_type == IBV_QPT_GSI) {
 | 
						|
            pr_dbg("QP1\n");
 | 
						|
            comp_handler(IBV_WC_GENERAL_ERR, VENDOR_ERR_MAD_SEND, ctx);
 | 
						|
        }
 | 
						|
        return;
 | 
						|
    }
 | 
						|
 | 
						|
    pr_dbg("num_sge=%d\n", num_sge);
 | 
						|
    if (!num_sge) {
 | 
						|
        pr_dbg("num_sge=0\n");
 | 
						|
        comp_handler(IBV_WC_GENERAL_ERR, VENDOR_ERR_NO_SGE, ctx);
 | 
						|
        return;
 | 
						|
    }
 | 
						|
 | 
						|
    bctx = g_malloc0(sizeof(*bctx));
 | 
						|
    bctx->up_ctx = ctx;
 | 
						|
    bctx->is_tx_req = 0;
 | 
						|
 | 
						|
    rc = rdma_rm_alloc_cqe_ctx(rdma_dev_res, &bctx_id, bctx);
 | 
						|
    if (unlikely(rc)) {
 | 
						|
        pr_dbg("Failed to allocate cqe_ctx\n");
 | 
						|
        comp_handler(IBV_WC_GENERAL_ERR, VENDOR_ERR_NOMEM, ctx);
 | 
						|
        goto out_free_bctx;
 | 
						|
    }
 | 
						|
 | 
						|
    rc = build_host_sge_array(rdma_dev_res, new_sge, sge, num_sge);
 | 
						|
    if (rc) {
 | 
						|
        pr_dbg("Error: Failed to build host SGE array\n");
 | 
						|
        comp_handler(IBV_WC_GENERAL_ERR, rc, ctx);
 | 
						|
        goto out_dealloc_cqe_ctx;
 | 
						|
    }
 | 
						|
 | 
						|
    wr.num_sge = num_sge;
 | 
						|
    wr.sg_list = new_sge;
 | 
						|
    wr.wr_id = bctx_id;
 | 
						|
    rc = ibv_post_recv(qp->ibqp, &wr, &bad_wr);
 | 
						|
    pr_dbg("ibv_post_recv=%d\n", rc);
 | 
						|
    if (rc) {
 | 
						|
        pr_dbg("Fail (%d, %d) to post recv WQE to qpn %d\n", rc, errno,
 | 
						|
                qp->ibqp->qp_num);
 | 
						|
        comp_handler(IBV_WC_GENERAL_ERR, VENDOR_ERR_FAIL_BACKEND, ctx);
 | 
						|
        goto out_dealloc_cqe_ctx;
 | 
						|
    }
 | 
						|
 | 
						|
    return;
 | 
						|
 | 
						|
out_dealloc_cqe_ctx:
 | 
						|
    rdma_rm_dealloc_cqe_ctx(rdma_dev_res, bctx_id);
 | 
						|
 | 
						|
out_free_bctx:
 | 
						|
    g_free(bctx);
 | 
						|
}
 | 
						|
 | 
						|
int rdma_backend_create_pd(RdmaBackendDev *backend_dev, RdmaBackendPD *pd)
 | 
						|
{
 | 
						|
    pd->ibpd = ibv_alloc_pd(backend_dev->context);
 | 
						|
 | 
						|
    return pd->ibpd ? 0 : -EIO;
 | 
						|
}
 | 
						|
 | 
						|
void rdma_backend_destroy_pd(RdmaBackendPD *pd)
 | 
						|
{
 | 
						|
    if (pd->ibpd) {
 | 
						|
        ibv_dealloc_pd(pd->ibpd);
 | 
						|
    }
 | 
						|
}
 | 
						|
 | 
						|
int rdma_backend_create_mr(RdmaBackendMR *mr, RdmaBackendPD *pd, void *addr,
 | 
						|
                           size_t length, int access)
 | 
						|
{
 | 
						|
    pr_dbg("addr=0x%p\n", addr);
 | 
						|
    pr_dbg("len=%zu\n", length);
 | 
						|
    mr->ibmr = ibv_reg_mr(pd->ibpd, addr, length, access);
 | 
						|
    if (mr->ibmr) {
 | 
						|
        pr_dbg("lkey=0x%x\n", mr->ibmr->lkey);
 | 
						|
        pr_dbg("rkey=0x%x\n", mr->ibmr->rkey);
 | 
						|
        mr->ibpd = pd->ibpd;
 | 
						|
    }
 | 
						|
 | 
						|
    return mr->ibmr ? 0 : -EIO;
 | 
						|
}
 | 
						|
 | 
						|
void rdma_backend_destroy_mr(RdmaBackendMR *mr)
 | 
						|
{
 | 
						|
    if (mr->ibmr) {
 | 
						|
        ibv_dereg_mr(mr->ibmr);
 | 
						|
    }
 | 
						|
}
 | 
						|
 | 
						|
int rdma_backend_create_cq(RdmaBackendDev *backend_dev, RdmaBackendCQ *cq,
 | 
						|
                           int cqe)
 | 
						|
{
 | 
						|
    int rc;
 | 
						|
 | 
						|
    pr_dbg("cqe=%d\n", cqe);
 | 
						|
 | 
						|
    pr_dbg("dev->channel=%p\n", backend_dev->channel);
 | 
						|
    cq->ibcq = ibv_create_cq(backend_dev->context, cqe + 1, NULL,
 | 
						|
                             backend_dev->channel, 0);
 | 
						|
 | 
						|
    if (cq->ibcq) {
 | 
						|
        rc = ibv_req_notify_cq(cq->ibcq, 0);
 | 
						|
        if (rc) {
 | 
						|
            pr_dbg("Error %d from ibv_req_notify_cq\n", rc);
 | 
						|
        }
 | 
						|
        cq->backend_dev = backend_dev;
 | 
						|
    }
 | 
						|
 | 
						|
    return cq->ibcq ? 0 : -EIO;
 | 
						|
}
 | 
						|
 | 
						|
void rdma_backend_destroy_cq(RdmaBackendCQ *cq)
 | 
						|
{
 | 
						|
    if (cq->ibcq) {
 | 
						|
        ibv_destroy_cq(cq->ibcq);
 | 
						|
    }
 | 
						|
}
 | 
						|
 | 
						|
int rdma_backend_create_qp(RdmaBackendQP *qp, uint8_t qp_type,
 | 
						|
                           RdmaBackendPD *pd, RdmaBackendCQ *scq,
 | 
						|
                           RdmaBackendCQ *rcq, uint32_t max_send_wr,
 | 
						|
                           uint32_t max_recv_wr, uint32_t max_send_sge,
 | 
						|
                           uint32_t max_recv_sge)
 | 
						|
{
 | 
						|
    struct ibv_qp_init_attr attr = {0};
 | 
						|
 | 
						|
    qp->ibqp = 0;
 | 
						|
    pr_dbg("qp_type=%d\n", qp_type);
 | 
						|
 | 
						|
    switch (qp_type) {
 | 
						|
    case IBV_QPT_GSI:
 | 
						|
        pr_dbg("QP1 unsupported\n");
 | 
						|
        return 0;
 | 
						|
 | 
						|
    case IBV_QPT_RC:
 | 
						|
        /* fall through */
 | 
						|
    case IBV_QPT_UD:
 | 
						|
        /* do nothing */
 | 
						|
        break;
 | 
						|
 | 
						|
    default:
 | 
						|
        pr_dbg("Unsupported QP type %d\n", qp_type);
 | 
						|
        return -EIO;
 | 
						|
    }
 | 
						|
 | 
						|
    attr.qp_type = qp_type;
 | 
						|
    attr.send_cq = scq->ibcq;
 | 
						|
    attr.recv_cq = rcq->ibcq;
 | 
						|
    attr.cap.max_send_wr = max_send_wr;
 | 
						|
    attr.cap.max_recv_wr = max_recv_wr;
 | 
						|
    attr.cap.max_send_sge = max_send_sge;
 | 
						|
    attr.cap.max_recv_sge = max_recv_sge;
 | 
						|
 | 
						|
    pr_dbg("max_send_wr=%d\n", max_send_wr);
 | 
						|
    pr_dbg("max_recv_wr=%d\n", max_recv_wr);
 | 
						|
    pr_dbg("max_send_sge=%d\n", max_send_sge);
 | 
						|
    pr_dbg("max_recv_sge=%d\n", max_recv_sge);
 | 
						|
 | 
						|
    qp->ibqp = ibv_create_qp(pd->ibpd, &attr);
 | 
						|
    if (likely(!qp->ibqp)) {
 | 
						|
        pr_dbg("Error from ibv_create_qp\n");
 | 
						|
        return -EIO;
 | 
						|
    }
 | 
						|
 | 
						|
    qp->ibpd = pd->ibpd;
 | 
						|
 | 
						|
    /* TODO: Query QP to get max_inline_data and save it to be used in send */
 | 
						|
 | 
						|
    pr_dbg("qpn=0x%x\n", qp->ibqp->qp_num);
 | 
						|
 | 
						|
    return 0;
 | 
						|
}
 | 
						|
 | 
						|
int rdma_backend_qp_state_init(RdmaBackendDev *backend_dev, RdmaBackendQP *qp,
 | 
						|
                               uint8_t qp_type, uint32_t qkey)
 | 
						|
{
 | 
						|
    struct ibv_qp_attr attr = {0};
 | 
						|
    int rc, attr_mask;
 | 
						|
 | 
						|
    pr_dbg("qpn=0x%x\n", qp->ibqp->qp_num);
 | 
						|
    pr_dbg("sport_num=%d\n", backend_dev->port_num);
 | 
						|
 | 
						|
    attr_mask = IBV_QP_STATE | IBV_QP_PKEY_INDEX | IBV_QP_PORT;
 | 
						|
    attr.qp_state        = IBV_QPS_INIT;
 | 
						|
    attr.pkey_index      = 0;
 | 
						|
    attr.port_num        = backend_dev->port_num;
 | 
						|
 | 
						|
    switch (qp_type) {
 | 
						|
    case IBV_QPT_RC:
 | 
						|
        attr_mask |= IBV_QP_ACCESS_FLAGS;
 | 
						|
        break;
 | 
						|
 | 
						|
    case IBV_QPT_UD:
 | 
						|
        attr.qkey = qkey;
 | 
						|
        attr_mask |= IBV_QP_QKEY;
 | 
						|
        break;
 | 
						|
 | 
						|
    default:
 | 
						|
        pr_dbg("Unsupported QP type %d\n", qp_type);
 | 
						|
        return -EIO;
 | 
						|
    }
 | 
						|
 | 
						|
    rc = ibv_modify_qp(qp->ibqp, &attr, attr_mask);
 | 
						|
    if (rc) {
 | 
						|
        pr_dbg("Error %d from ibv_modify_qp\n", rc);
 | 
						|
        return -EIO;
 | 
						|
    }
 | 
						|
 | 
						|
    return 0;
 | 
						|
}
 | 
						|
 | 
						|
int rdma_backend_qp_state_rtr(RdmaBackendDev *backend_dev, RdmaBackendQP *qp,
 | 
						|
                              uint8_t qp_type, union ibv_gid *dgid,
 | 
						|
                              uint32_t dqpn, uint32_t rq_psn, uint32_t qkey,
 | 
						|
                              bool use_qkey)
 | 
						|
{
 | 
						|
    struct ibv_qp_attr attr = {0};
 | 
						|
    union ibv_gid ibv_gid = {
 | 
						|
        .global.interface_id = dgid->global.interface_id,
 | 
						|
        .global.subnet_prefix = dgid->global.subnet_prefix
 | 
						|
    };
 | 
						|
    int rc, attr_mask;
 | 
						|
 | 
						|
    attr.qp_state = IBV_QPS_RTR;
 | 
						|
    attr_mask = IBV_QP_STATE;
 | 
						|
 | 
						|
    switch (qp_type) {
 | 
						|
    case IBV_QPT_RC:
 | 
						|
        pr_dbg("dgid=0x%" PRIx64 ",%" PRIx64 "\n",
 | 
						|
               be64_to_cpu(ibv_gid.global.subnet_prefix),
 | 
						|
               be64_to_cpu(ibv_gid.global.interface_id));
 | 
						|
        pr_dbg("dqpn=0x%x\n", dqpn);
 | 
						|
        pr_dbg("sgid_idx=%d\n", backend_dev->backend_gid_idx);
 | 
						|
        pr_dbg("sport_num=%d\n", backend_dev->port_num);
 | 
						|
        pr_dbg("rq_psn=0x%x\n", rq_psn);
 | 
						|
 | 
						|
        attr.path_mtu               = IBV_MTU_1024;
 | 
						|
        attr.dest_qp_num            = dqpn;
 | 
						|
        attr.max_dest_rd_atomic     = 1;
 | 
						|
        attr.min_rnr_timer          = 12;
 | 
						|
        attr.ah_attr.port_num       = backend_dev->port_num;
 | 
						|
        attr.ah_attr.is_global      = 1;
 | 
						|
        attr.ah_attr.grh.hop_limit  = 1;
 | 
						|
        attr.ah_attr.grh.dgid       = ibv_gid;
 | 
						|
        attr.ah_attr.grh.sgid_index = backend_dev->backend_gid_idx;
 | 
						|
        attr.rq_psn                 = rq_psn;
 | 
						|
 | 
						|
        attr_mask |= IBV_QP_AV | IBV_QP_PATH_MTU | IBV_QP_DEST_QPN |
 | 
						|
                     IBV_QP_RQ_PSN | IBV_QP_MAX_DEST_RD_ATOMIC |
 | 
						|
                     IBV_QP_MIN_RNR_TIMER;
 | 
						|
        break;
 | 
						|
 | 
						|
    case IBV_QPT_UD:
 | 
						|
        if (use_qkey) {
 | 
						|
            pr_dbg("qkey=0x%x\n", qkey);
 | 
						|
            attr.qkey = qkey;
 | 
						|
            attr_mask |= IBV_QP_QKEY;
 | 
						|
        }
 | 
						|
        break;
 | 
						|
    }
 | 
						|
 | 
						|
    rc = ibv_modify_qp(qp->ibqp, &attr, attr_mask);
 | 
						|
    if (rc) {
 | 
						|
        pr_dbg("Error %d from ibv_modify_qp\n", rc);
 | 
						|
        return -EIO;
 | 
						|
    }
 | 
						|
 | 
						|
    return 0;
 | 
						|
}
 | 
						|
 | 
						|
int rdma_backend_qp_state_rts(RdmaBackendQP *qp, uint8_t qp_type,
 | 
						|
                              uint32_t sq_psn, uint32_t qkey, bool use_qkey)
 | 
						|
{
 | 
						|
    struct ibv_qp_attr attr = {0};
 | 
						|
    int rc, attr_mask;
 | 
						|
 | 
						|
    pr_dbg("qpn=0x%x\n", qp->ibqp->qp_num);
 | 
						|
    pr_dbg("sq_psn=0x%x\n", sq_psn);
 | 
						|
 | 
						|
    attr.qp_state = IBV_QPS_RTS;
 | 
						|
    attr.sq_psn = sq_psn;
 | 
						|
    attr_mask = IBV_QP_STATE | IBV_QP_SQ_PSN;
 | 
						|
 | 
						|
    switch (qp_type) {
 | 
						|
    case IBV_QPT_RC:
 | 
						|
        attr.timeout       = 14;
 | 
						|
        attr.retry_cnt     = 7;
 | 
						|
        attr.rnr_retry     = 7;
 | 
						|
        attr.max_rd_atomic = 1;
 | 
						|
 | 
						|
        attr_mask |= IBV_QP_TIMEOUT | IBV_QP_RETRY_CNT | IBV_QP_RNR_RETRY |
 | 
						|
                     IBV_QP_MAX_QP_RD_ATOMIC;
 | 
						|
        break;
 | 
						|
 | 
						|
    case IBV_QPT_UD:
 | 
						|
        if (use_qkey) {
 | 
						|
            pr_dbg("qkey=0x%x\n", qkey);
 | 
						|
            attr.qkey = qkey;
 | 
						|
            attr_mask |= IBV_QP_QKEY;
 | 
						|
        }
 | 
						|
        break;
 | 
						|
    }
 | 
						|
 | 
						|
    rc = ibv_modify_qp(qp->ibqp, &attr, attr_mask);
 | 
						|
    if (rc) {
 | 
						|
        pr_dbg("Error %d from ibv_modify_qp\n", rc);
 | 
						|
        return -EIO;
 | 
						|
    }
 | 
						|
 | 
						|
    return 0;
 | 
						|
}
 | 
						|
 | 
						|
int rdma_backend_query_qp(RdmaBackendQP *qp, struct ibv_qp_attr *attr,
 | 
						|
                          int attr_mask, struct ibv_qp_init_attr *init_attr)
 | 
						|
{
 | 
						|
    if (!qp->ibqp) {
 | 
						|
        pr_dbg("QP1\n");
 | 
						|
        attr->qp_state = IBV_QPS_RTS;
 | 
						|
        return 0;
 | 
						|
    }
 | 
						|
 | 
						|
    return ibv_query_qp(qp->ibqp, attr, attr_mask, init_attr);
 | 
						|
}
 | 
						|
 | 
						|
void rdma_backend_destroy_qp(RdmaBackendQP *qp)
 | 
						|
{
 | 
						|
    if (qp->ibqp) {
 | 
						|
        ibv_destroy_qp(qp->ibqp);
 | 
						|
    }
 | 
						|
}
 | 
						|
 | 
						|
#define CHK_ATTR(req, dev, member, fmt) ({ \
 | 
						|
    pr_dbg("%s="fmt","fmt"\n", #member, dev.member, req->member); \
 | 
						|
    if (req->member > dev.member) { \
 | 
						|
        warn_report("%s = "fmt" is higher than host device capability "fmt, \
 | 
						|
                    #member, req->member, dev.member); \
 | 
						|
        req->member = dev.member; \
 | 
						|
    } \
 | 
						|
    pr_dbg("%s="fmt"\n", #member, req->member); })
 | 
						|
 | 
						|
static int init_device_caps(RdmaBackendDev *backend_dev,
 | 
						|
                            struct ibv_device_attr *dev_attr)
 | 
						|
{
 | 
						|
    if (ibv_query_device(backend_dev->context, &backend_dev->dev_attr)) {
 | 
						|
        return -EIO;
 | 
						|
    }
 | 
						|
 | 
						|
    CHK_ATTR(dev_attr, backend_dev->dev_attr, max_mr_size, "%" PRId64);
 | 
						|
    CHK_ATTR(dev_attr, backend_dev->dev_attr, max_qp, "%d");
 | 
						|
    CHK_ATTR(dev_attr, backend_dev->dev_attr, max_sge, "%d");
 | 
						|
    CHK_ATTR(dev_attr, backend_dev->dev_attr, max_qp_wr, "%d");
 | 
						|
    CHK_ATTR(dev_attr, backend_dev->dev_attr, max_cq, "%d");
 | 
						|
    CHK_ATTR(dev_attr, backend_dev->dev_attr, max_cqe, "%d");
 | 
						|
    CHK_ATTR(dev_attr, backend_dev->dev_attr, max_mr, "%d");
 | 
						|
    CHK_ATTR(dev_attr, backend_dev->dev_attr, max_pd, "%d");
 | 
						|
    CHK_ATTR(dev_attr, backend_dev->dev_attr, max_qp_rd_atom, "%d");
 | 
						|
    CHK_ATTR(dev_attr, backend_dev->dev_attr, max_qp_init_rd_atom, "%d");
 | 
						|
    CHK_ATTR(dev_attr, backend_dev->dev_attr, max_ah, "%d");
 | 
						|
 | 
						|
    return 0;
 | 
						|
}
 | 
						|
 | 
						|
int rdma_backend_init(RdmaBackendDev *backend_dev,
 | 
						|
                      RdmaDeviceResources *rdma_dev_res,
 | 
						|
                      const char *backend_device_name, uint8_t port_num,
 | 
						|
                      uint8_t backend_gid_idx, struct ibv_device_attr *dev_attr,
 | 
						|
                      Error **errp)
 | 
						|
{
 | 
						|
    int i;
 | 
						|
    int ret = 0;
 | 
						|
    int num_ibv_devices;
 | 
						|
    char thread_name[THR_NAME_LEN] = {0};
 | 
						|
    struct ibv_device **dev_list;
 | 
						|
    struct ibv_port_attr port_attr;
 | 
						|
 | 
						|
    backend_dev->backend_gid_idx = backend_gid_idx;
 | 
						|
    backend_dev->port_num = port_num;
 | 
						|
    backend_dev->rdma_dev_res = rdma_dev_res;
 | 
						|
 | 
						|
    rdma_backend_register_comp_handler(dummy_comp_handler);
 | 
						|
 | 
						|
    dev_list = ibv_get_device_list(&num_ibv_devices);
 | 
						|
    if (!dev_list) {
 | 
						|
        error_setg(errp, "Failed to get IB devices list");
 | 
						|
        return -EIO;
 | 
						|
    }
 | 
						|
 | 
						|
    if (num_ibv_devices == 0) {
 | 
						|
        error_setg(errp, "No IB devices were found");
 | 
						|
        ret = -ENXIO;
 | 
						|
        goto out_free_dev_list;
 | 
						|
    }
 | 
						|
 | 
						|
    if (backend_device_name) {
 | 
						|
        for (i = 0; dev_list[i]; ++i) {
 | 
						|
            if (!strcmp(ibv_get_device_name(dev_list[i]),
 | 
						|
                        backend_device_name)) {
 | 
						|
                break;
 | 
						|
            }
 | 
						|
        }
 | 
						|
 | 
						|
        backend_dev->ib_dev = dev_list[i];
 | 
						|
        if (!backend_dev->ib_dev) {
 | 
						|
            error_setg(errp, "Failed to find IB device %s",
 | 
						|
                       backend_device_name);
 | 
						|
            ret = -EIO;
 | 
						|
            goto out_free_dev_list;
 | 
						|
        }
 | 
						|
    } else {
 | 
						|
        backend_dev->ib_dev = *dev_list;
 | 
						|
    }
 | 
						|
 | 
						|
    pr_dbg("Using backend device %s, port %d, gid_idx %d\n",
 | 
						|
           ibv_get_device_name(backend_dev->ib_dev),
 | 
						|
           backend_dev->port_num, backend_dev->backend_gid_idx);
 | 
						|
 | 
						|
    backend_dev->context = ibv_open_device(backend_dev->ib_dev);
 | 
						|
    if (!backend_dev->context) {
 | 
						|
        error_setg(errp, "Failed to open IB device");
 | 
						|
        ret = -EIO;
 | 
						|
        goto out;
 | 
						|
    }
 | 
						|
 | 
						|
    backend_dev->channel = ibv_create_comp_channel(backend_dev->context);
 | 
						|
    if (!backend_dev->channel) {
 | 
						|
        error_setg(errp, "Failed to create IB communication channel");
 | 
						|
        ret = -EIO;
 | 
						|
        goto out_close_device;
 | 
						|
    }
 | 
						|
    pr_dbg("dev->backend_dev.channel=%p\n", backend_dev->channel);
 | 
						|
 | 
						|
    ret = ibv_query_port(backend_dev->context, backend_dev->port_num,
 | 
						|
                         &port_attr);
 | 
						|
    if (ret) {
 | 
						|
        error_setg(errp, "Error %d from ibv_query_port", ret);
 | 
						|
        ret = -EIO;
 | 
						|
        goto out_destroy_comm_channel;
 | 
						|
    }
 | 
						|
 | 
						|
    if (backend_dev->backend_gid_idx > port_attr.gid_tbl_len) {
 | 
						|
        error_setg(errp, "Invalid backend_gid_idx, should be less than %d",
 | 
						|
                   port_attr.gid_tbl_len);
 | 
						|
        goto out_destroy_comm_channel;
 | 
						|
    }
 | 
						|
 | 
						|
    ret = init_device_caps(backend_dev, dev_attr);
 | 
						|
    if (ret) {
 | 
						|
        error_setg(errp, "Failed to initialize device capabilities");
 | 
						|
        ret = -EIO;
 | 
						|
        goto out_destroy_comm_channel;
 | 
						|
    }
 | 
						|
 | 
						|
    ret = ibv_query_gid(backend_dev->context, backend_dev->port_num,
 | 
						|
                         backend_dev->backend_gid_idx, &backend_dev->gid);
 | 
						|
    if (ret) {
 | 
						|
        error_setg(errp, "Failed to query gid %d",
 | 
						|
                   backend_dev->backend_gid_idx);
 | 
						|
        ret = -EIO;
 | 
						|
        goto out_destroy_comm_channel;
 | 
						|
    }
 | 
						|
    pr_dbg("subnet_prefix=0x%" PRIx64 "\n",
 | 
						|
           be64_to_cpu(backend_dev->gid.global.subnet_prefix));
 | 
						|
    pr_dbg("interface_id=0x%" PRIx64 "\n",
 | 
						|
           be64_to_cpu(backend_dev->gid.global.interface_id));
 | 
						|
 | 
						|
    snprintf(thread_name, sizeof(thread_name), "rdma_comp_%s",
 | 
						|
             ibv_get_device_name(backend_dev->ib_dev));
 | 
						|
    backend_dev->comp_thread.run = true;
 | 
						|
    qemu_thread_create(&backend_dev->comp_thread.thread, thread_name,
 | 
						|
                       comp_handler_thread, backend_dev, QEMU_THREAD_DETACHED);
 | 
						|
 | 
						|
    ah_cache_init();
 | 
						|
 | 
						|
    goto out_free_dev_list;
 | 
						|
 | 
						|
out_destroy_comm_channel:
 | 
						|
    ibv_destroy_comp_channel(backend_dev->channel);
 | 
						|
 | 
						|
out_close_device:
 | 
						|
    ibv_close_device(backend_dev->context);
 | 
						|
 | 
						|
out_free_dev_list:
 | 
						|
    ibv_free_device_list(dev_list);
 | 
						|
 | 
						|
out:
 | 
						|
    return ret;
 | 
						|
}
 | 
						|
 | 
						|
void rdma_backend_fini(RdmaBackendDev *backend_dev)
 | 
						|
{
 | 
						|
    g_hash_table_destroy(ah_hash);
 | 
						|
    ibv_destroy_comp_channel(backend_dev->channel);
 | 
						|
    ibv_close_device(backend_dev->context);
 | 
						|
}
 |