// SPDX-License-Identifier: GPL-2.0-only
/*
 * Copyright (c) 2022, Microsoft Corporation. All rights reserved.
 */

#include "mana_ib.h"

static int mana_ib_cfg_vport_steering(struct mana_ib_dev *dev,
				      struct net_device *ndev,
				      mana_handle_t default_rxobj,
				      mana_handle_t ind_table[],
				      u32 log_ind_tbl_size, u32 rx_hash_key_len,
				      u8 *rx_hash_key)
{
	struct mana_port_context *mpc = netdev_priv(ndev);
	struct mana_cfg_rx_steer_req_v2 *req;
	struct mana_cfg_rx_steer_resp resp = {};
	struct gdma_context *gc;
	u32 req_buf_size;
	int i, err;

	gc = mdev_to_gc(dev);

	req_buf_size = struct_size(req, indir_tab, MANA_INDIRECT_TABLE_DEF_SIZE);
	req = kzalloc(req_buf_size, GFP_KERNEL);
	if (!req)
		return -ENOMEM;

	mana_gd_init_req_hdr(&req->hdr, MANA_CONFIG_VPORT_RX, req_buf_size,
			     sizeof(resp));

	req->hdr.req.msg_version = GDMA_MESSAGE_V2;

	req->vport = mpc->port_handle;
	req->rx_enable = 1;
	req->update_default_rxobj = 1;
	req->default_rxobj = default_rxobj;
	req->hdr.dev_id = gc->mana.dev_id;

	/* If there are more than 1 entries in indirection table, enable RSS */
	if (log_ind_tbl_size)
		req->rss_enable = true;

	req->num_indir_entries = MANA_INDIRECT_TABLE_DEF_SIZE;
	req->indir_tab_offset = offsetof(struct mana_cfg_rx_steer_req_v2,
					 indir_tab);
	req->update_indir_tab = true;
	req->cqe_coalescing_enable = 1;

	/* The ind table passed to the hardware must have
	 * MANA_INDIRECT_TABLE_DEF_SIZE entries. Adjust the verb
	 * ind_table to MANA_INDIRECT_TABLE_SIZE if required
	 */
	ibdev_dbg(&dev->ib_dev, "ind table size %u\n", 1 << log_ind_tbl_size);
	for (i = 0; i < MANA_INDIRECT_TABLE_DEF_SIZE; i++) {
		req->indir_tab[i] = ind_table[i % (1 << log_ind_tbl_size)];
		ibdev_dbg(&dev->ib_dev, "index %u handle 0x%llx\n", i,
			  req->indir_tab[i]);
	}

	req->update_hashkey = true;
	if (rx_hash_key_len)
		memcpy(req->hashkey, rx_hash_key, rx_hash_key_len);
	else
		netdev_rss_key_fill(req->hashkey, MANA_HASH_KEY_SIZE);

	ibdev_dbg(&dev->ib_dev, "vport handle %llu default_rxobj 0x%llx\n",
		  req->vport, default_rxobj);

	err = mana_gd_send_request(gc, req_buf_size, req, sizeof(resp), &resp);
	if (err) {
		netdev_err(ndev, "Failed to configure vPort RX: %d\n", err);
		goto out;
	}

	if (resp.hdr.status) {
		netdev_err(ndev, "vPort RX configuration failed: 0x%x\n",
			   resp.hdr.status);
		err = -EPROTO;
		goto out;
	}

	netdev_info(ndev, "Configured steering vPort %llu log_entries %u\n",
		    mpc->port_handle, log_ind_tbl_size);

out:
	kfree(req);
	return err;
}

static int mana_ib_create_qp_rss(struct ib_qp *ibqp, struct ib_pd *pd,
				 struct ib_qp_init_attr *attr,
				 struct ib_udata *udata)
{
	struct mana_ib_qp *qp = container_of(ibqp, struct mana_ib_qp, ibqp);
	struct mana_ib_dev *mdev =
		container_of(pd->device, struct mana_ib_dev, ib_dev);
	struct ib_rwq_ind_table *ind_tbl = attr->rwq_ind_tbl;
	struct mana_ib_create_qp_rss_resp resp = {};
	struct mana_ib_create_qp_rss ucmd = {};
	mana_handle_t *mana_ind_table;
	struct mana_port_context *mpc;
	unsigned int ind_tbl_size;
	struct net_device *ndev;
	struct mana_ib_cq *cq;
	struct mana_ib_wq *wq;
	struct mana_eq *eq;
	struct ib_cq *ibcq;
	struct ib_wq *ibwq;
	int i = 0;
	u32 port;
	int ret;

	if (!udata || udata->inlen < sizeof(ucmd))
		return -EINVAL;

	ret = ib_copy_from_udata(&ucmd, udata, min(sizeof(ucmd), udata->inlen));
	if (ret) {
		ibdev_dbg(&mdev->ib_dev,
			  "Failed copy from udata for create rss-qp, err %d\n",
			  ret);
		return ret;
	}

	if (attr->cap.max_recv_wr > mdev->adapter_caps.max_qp_wr) {
		ibdev_dbg(&mdev->ib_dev,
			  "Requested max_recv_wr %d exceeding limit\n",
			  attr->cap.max_recv_wr);
		return -EINVAL;
	}

	if (attr->cap.max_recv_sge > MAX_RX_WQE_SGL_ENTRIES) {
		ibdev_dbg(&mdev->ib_dev,
			  "Requested max_recv_sge %d exceeding limit\n",
			  attr->cap.max_recv_sge);
		return -EINVAL;
	}

	ind_tbl_size = 1 << ind_tbl->log_ind_tbl_size;
	if (ind_tbl_size > MANA_INDIRECT_TABLE_DEF_SIZE) {
		ibdev_dbg(&mdev->ib_dev,
			  "Indirect table size %d exceeding limit\n",
			  ind_tbl_size);
		return -EINVAL;
	}

	if (ucmd.rx_hash_function != MANA_IB_RX_HASH_FUNC_TOEPLITZ) {
		ibdev_dbg(&mdev->ib_dev,
			  "RX Hash function is not supported, %d\n",
			  ucmd.rx_hash_function);
		return -EINVAL;
	}

	/* IB ports start with 1, MANA start with 0 */
	port = ucmd.port;
	ndev = mana_ib_get_netdev(pd->device, port);
	if (!ndev) {
		ibdev_dbg(&mdev->ib_dev, "Invalid port %u in creating qp\n",
			  port);
		return -EINVAL;
	}
	mpc = netdev_priv(ndev);

	ibdev_dbg(&mdev->ib_dev, "rx_hash_function %d port %d\n",
		  ucmd.rx_hash_function, port);

	mana_ind_table = kcalloc(ind_tbl_size, sizeof(mana_handle_t),
				 GFP_KERNEL);
	if (!mana_ind_table) {
		ret = -ENOMEM;
		goto fail;
	}

	qp->port = port;

	for (i = 0; i < ind_tbl_size; i++) {
		struct mana_obj_spec wq_spec = {};
		struct mana_obj_spec cq_spec = {};

		ibwq = ind_tbl->ind_tbl[i];
		wq = container_of(ibwq, struct mana_ib_wq, ibwq);

		ibcq = ibwq->cq;
		cq = container_of(ibcq, struct mana_ib_cq, ibcq);

		wq_spec.gdma_region = wq->queue.gdma_region;
		wq_spec.queue_size = wq->wq_buf_size;

		cq_spec.gdma_region = cq->queue.gdma_region;
		cq_spec.queue_size = cq->cqe * COMP_ENTRY_SIZE;
		cq_spec.modr_ctx_id = 0;
		eq = &mpc->ac->eqs[cq->comp_vector];
		cq_spec.attached_eq = eq->eq->id;

		ret = mana_create_wq_obj(mpc, mpc->port_handle, GDMA_RQ,
					 &wq_spec, &cq_spec, &wq->rx_object);
		if (ret) {
			/* Do cleanup starting with index i-1 */
			i--;
			goto fail;
		}

		/* The GDMA regions are now owned by the WQ object */
		wq->queue.gdma_region = GDMA_INVALID_DMA_REGION;
		cq->queue.gdma_region = GDMA_INVALID_DMA_REGION;

		wq->queue.id = wq_spec.queue_index;
		cq->queue.id = cq_spec.queue_index;

		ibdev_dbg(&mdev->ib_dev,
			  "rx_object 0x%llx wq id %llu cq id %llu\n",
			  wq->rx_object, wq->queue.id, cq->queue.id);

		resp.entries[i].cqid = cq->queue.id;
		resp.entries[i].wqid = wq->queue.id;

		mana_ind_table[i] = wq->rx_object;

		/* Create CQ table entry */
		ret = mana_ib_install_cq_cb(mdev, cq);
		if (ret)
			goto fail;
	}
	resp.num_entries = i;

	ret = mana_ib_cfg_vport_steering(mdev, ndev, wq->rx_object,
					 mana_ind_table,
					 ind_tbl->log_ind_tbl_size,
					 ucmd.rx_hash_key_len,
					 ucmd.rx_hash_key);
	if (ret)
		goto fail;

	ret = ib_copy_to_udata(udata, &resp, sizeof(resp));
	if (ret) {
		ibdev_dbg(&mdev->ib_dev,
			  "Failed to copy to udata create rss-qp, %d\n",
			  ret);
		goto fail;
	}

	kfree(mana_ind_table);

	return 0;

fail:
	while (i-- > 0) {
		ibwq = ind_tbl->ind_tbl[i];
		ibcq = ibwq->cq;
		wq = container_of(ibwq, struct mana_ib_wq, ibwq);
		cq = container_of(ibcq, struct mana_ib_cq, ibcq);

		mana_ib_remove_cq_cb(mdev, cq);
		mana_destroy_wq_obj(mpc, GDMA_RQ, wq->rx_object);
	}

	kfree(mana_ind_table);

	return ret;
}

static int mana_ib_create_qp_raw(struct ib_qp *ibqp, struct ib_pd *ibpd,
				 struct ib_qp_init_attr *attr,
				 struct ib_udata *udata)
{
	struct mana_ib_pd *pd = container_of(ibpd, struct mana_ib_pd, ibpd);
	struct mana_ib_qp *qp = container_of(ibqp, struct mana_ib_qp, ibqp);
	struct mana_ib_dev *mdev =
		container_of(ibpd->device, struct mana_ib_dev, ib_dev);
	struct mana_ib_cq *send_cq =
		container_of(attr->send_cq, struct mana_ib_cq, ibcq);
	struct mana_ib_ucontext *mana_ucontext =
		rdma_udata_to_drv_context(udata, struct mana_ib_ucontext,
					  ibucontext);
	struct mana_ib_create_qp_resp resp = {};
	struct mana_ib_create_qp ucmd = {};
	struct mana_obj_spec wq_spec = {};
	struct mana_obj_spec cq_spec = {};
	struct mana_port_context *mpc;
	struct net_device *ndev;
	struct mana_eq *eq;
	int eq_vec;
	u32 port;
	int err;

	if (!mana_ucontext || udata->inlen < sizeof(ucmd))
		return -EINVAL;

	err = ib_copy_from_udata(&ucmd, udata, min(sizeof(ucmd), udata->inlen));
	if (err) {
		ibdev_dbg(&mdev->ib_dev,
			  "Failed to copy from udata create qp-raw, %d\n", err);
		return err;
	}

	if (attr->cap.max_send_wr > mdev->adapter_caps.max_qp_wr) {
		ibdev_dbg(&mdev->ib_dev,
			  "Requested max_send_wr %d exceeding limit\n",
			  attr->cap.max_send_wr);
		return -EINVAL;
	}

	if (attr->cap.max_send_sge > MAX_TX_WQE_SGL_ENTRIES) {
		ibdev_dbg(&mdev->ib_dev,
			  "Requested max_send_sge %d exceeding limit\n",
			  attr->cap.max_send_sge);
		return -EINVAL;
	}

	port = ucmd.port;
	ndev = mana_ib_get_netdev(ibpd->device, port);
	if (!ndev) {
		ibdev_dbg(&mdev->ib_dev, "Invalid port %u in creating qp\n",
			  port);
		return -EINVAL;
	}
	mpc = netdev_priv(ndev);
	ibdev_dbg(&mdev->ib_dev, "port %u ndev %p mpc %p\n", port, ndev, mpc);

	err = mana_ib_cfg_vport(mdev, port, pd, mana_ucontext->doorbell);
	if (err)
		return -ENODEV;

	qp->port = port;

	ibdev_dbg(&mdev->ib_dev, "ucmd sq_buf_addr 0x%llx port %u\n",
		  ucmd.sq_buf_addr, ucmd.port);

	err = mana_ib_create_queue(mdev, ucmd.sq_buf_addr, ucmd.sq_buf_size, &qp->raw_sq);
	if (err) {
		ibdev_dbg(&mdev->ib_dev,
			  "Failed to create queue for create qp-raw, err %d\n", err);
		goto err_free_vport;
	}

	/* Create a WQ on the same port handle used by the Ethernet */
	wq_spec.gdma_region = qp->raw_sq.gdma_region;
	wq_spec.queue_size = ucmd.sq_buf_size;

	cq_spec.gdma_region = send_cq->queue.gdma_region;
	cq_spec.queue_size = send_cq->cqe * COMP_ENTRY_SIZE;
	cq_spec.modr_ctx_id = 0;
	eq_vec = send_cq->comp_vector;
	eq = &mpc->ac->eqs[eq_vec];
	cq_spec.attached_eq = eq->eq->id;

	err = mana_create_wq_obj(mpc, mpc->port_handle, GDMA_SQ, &wq_spec,
				 &cq_spec, &qp->qp_handle);
	if (err) {
		ibdev_dbg(&mdev->ib_dev,
			  "Failed to create wq for create raw-qp, err %d\n",
			  err);
		goto err_destroy_queue;
	}

	/* The GDMA regions are now owned by the WQ object */
	qp->raw_sq.gdma_region = GDMA_INVALID_DMA_REGION;
	send_cq->queue.gdma_region = GDMA_INVALID_DMA_REGION;

	qp->raw_sq.id = wq_spec.queue_index;
	send_cq->queue.id = cq_spec.queue_index;

	/* Create CQ table entry */
	err = mana_ib_install_cq_cb(mdev, send_cq);
	if (err)
		goto err_destroy_wq_obj;

	ibdev_dbg(&mdev->ib_dev,
		  "qp->qp_handle 0x%llx sq id %llu cq id %llu\n",
		  qp->qp_handle, qp->raw_sq.id, send_cq->queue.id);

	resp.sqid = qp->raw_sq.id;
	resp.cqid = send_cq->queue.id;
	resp.tx_vp_offset = pd->tx_vp_offset;

	err = ib_copy_to_udata(udata, &resp, sizeof(resp));
	if (err) {
		ibdev_dbg(&mdev->ib_dev,
			  "Failed copy udata for create qp-raw, %d\n",
			  err);
		goto err_remove_cq_cb;
	}

	return 0;

err_remove_cq_cb:
	mana_ib_remove_cq_cb(mdev, send_cq);

err_destroy_wq_obj:
	mana_destroy_wq_obj(mpc, GDMA_SQ, qp->qp_handle);

err_destroy_queue:
	mana_ib_destroy_queue(mdev, &qp->raw_sq);

err_free_vport:
	mana_ib_uncfg_vport(mdev, pd, port);

	return err;
}

static u32 mana_ib_wqe_size(u32 sge, u32 oob_size)
{
	u32 wqe_size = sge * sizeof(struct gdma_sge) + sizeof(struct gdma_wqe) + oob_size;

	return ALIGN(wqe_size, GDMA_WQE_BU_SIZE);
}

static u32 mana_ib_queue_size(struct ib_qp_init_attr *attr, u32 queue_type)
{
	u32 queue_size;

	switch (attr->qp_type) {
	case IB_QPT_UD:
	case IB_QPT_GSI:
		if (queue_type == MANA_UD_SEND_QUEUE)
			queue_size = attr->cap.max_send_wr *
				mana_ib_wqe_size(attr->cap.max_send_sge, INLINE_OOB_LARGE_SIZE);
		else
			queue_size = attr->cap.max_recv_wr *
				mana_ib_wqe_size(attr->cap.max_recv_sge, INLINE_OOB_SMALL_SIZE);
		break;
	default:
		return 0;
	}

	return MANA_PAGE_ALIGN(roundup_pow_of_two(queue_size));
}

static enum gdma_queue_type mana_ib_queue_type(struct ib_qp_init_attr *attr, u32 queue_type)
{
	enum gdma_queue_type type;

	switch (attr->qp_type) {
	case IB_QPT_UD:
	case IB_QPT_GSI:
		if (queue_type == MANA_UD_SEND_QUEUE)
			type = GDMA_SQ;
		else
			type = GDMA_RQ;
		break;
	default:
		type = GDMA_INVALID_QUEUE;
	}
	return type;
}

static int mana_table_store_rc_qp(struct mana_ib_dev *mdev, struct mana_ib_qp *qp)
{
	return xa_insert_irq(&mdev->qp_table_wq, qp->ibqp.qp_num, qp,
			     GFP_KERNEL);
}

static void mana_table_remove_rc_qp(struct mana_ib_dev *mdev, struct mana_ib_qp *qp)
{
	xa_erase_irq(&mdev->qp_table_wq, qp->ibqp.qp_num);
}

static int mana_table_store_ud_qp(struct mana_ib_dev *mdev, struct mana_ib_qp *qp)
{
	u32 qids = qp->ud_qp.queues[MANA_UD_SEND_QUEUE].id | MANA_SENDQ_MASK;
	u32 qidr = qp->ud_qp.queues[MANA_UD_RECV_QUEUE].id;
	int err;

	err = xa_insert_irq(&mdev->qp_table_wq, qids, qp, GFP_KERNEL);
	if (err)
		return err;

	err = xa_insert_irq(&mdev->qp_table_wq, qidr, qp, GFP_KERNEL);
	if (err)
		goto remove_sq;

	return 0;

remove_sq:
	xa_erase_irq(&mdev->qp_table_wq, qids);
	return err;
}

static void mana_table_remove_ud_qp(struct mana_ib_dev *mdev, struct mana_ib_qp *qp)
{
	u32 qids = qp->ud_qp.queues[MANA_UD_SEND_QUEUE].id | MANA_SENDQ_MASK;
	u32 qidr = qp->ud_qp.queues[MANA_UD_RECV_QUEUE].id;

	xa_erase_irq(&mdev->qp_table_wq, qids);
	xa_erase_irq(&mdev->qp_table_wq, qidr);
}

static int mana_table_store_qp(struct mana_ib_dev *mdev, struct mana_ib_qp *qp)
{
	refcount_set(&qp->refcount, 1);
	init_completion(&qp->free);

	switch (qp->ibqp.qp_type) {
	case IB_QPT_RC:
		return mana_table_store_rc_qp(mdev, qp);
	case IB_QPT_UD:
	case IB_QPT_GSI:
		return mana_table_store_ud_qp(mdev, qp);
	default:
		ibdev_dbg(&mdev->ib_dev, "Unknown QP type for storing in mana table, %d\n",
			  qp->ibqp.qp_type);
	}

	return -EINVAL;
}

static void mana_table_remove_qp(struct mana_ib_dev *mdev,
				 struct mana_ib_qp *qp)
{
	switch (qp->ibqp.qp_type) {
	case IB_QPT_RC:
		mana_table_remove_rc_qp(mdev, qp);
		break;
	case IB_QPT_UD:
	case IB_QPT_GSI:
		mana_table_remove_ud_qp(mdev, qp);
		break;
	default:
		ibdev_dbg(&mdev->ib_dev, "Unknown QP type for removing from mana table, %d\n",
			  qp->ibqp.qp_type);
		return;
	}
	mana_put_qp_ref(qp);
	wait_for_completion(&qp->free);
}

static int mana_ib_create_rc_qp(struct ib_qp *ibqp, struct ib_pd *ibpd,
				struct ib_qp_init_attr *attr, struct ib_udata *udata)
{
	struct mana_ib_dev *mdev = container_of(ibpd->device, struct mana_ib_dev, ib_dev);
	struct mana_ib_qp *qp = container_of(ibqp, struct mana_ib_qp, ibqp);
	struct mana_ib_create_rc_qp_resp resp = {};
	struct mana_ib_ucontext *mana_ucontext;
	struct mana_ib_create_rc_qp ucmd = {};
	int i, err, j;
	u64 flags = 0;
	u32 doorbell;

	if (!udata || udata->inlen < sizeof(ucmd))
		return -EINVAL;

	mana_ucontext = rdma_udata_to_drv_context(udata, struct mana_ib_ucontext, ibucontext);
	doorbell = mana_ucontext->doorbell;
	flags = MANA_RC_FLAG_NO_FMR;
	err = ib_copy_from_udata(&ucmd, udata, min(sizeof(ucmd), udata->inlen));
	if (err) {
		ibdev_dbg(&mdev->ib_dev, "Failed to copy from udata, %d\n", err);
		return err;
	}

	for (i = 0, j = 0; i < MANA_RC_QUEUE_TYPE_MAX; ++i) {
		/* skip FMR for user-level RC QPs */
		if (i == MANA_RC_SEND_QUEUE_FMR) {
			qp->rc_qp.queues[i].id = INVALID_QUEUE_ID;
			qp->rc_qp.queues[i].gdma_region = GDMA_INVALID_DMA_REGION;
			continue;
		}
		err = mana_ib_create_queue(mdev, ucmd.queue_buf[j], ucmd.queue_size[j],
					   &qp->rc_qp.queues[i]);
		if (err) {
			ibdev_err(&mdev->ib_dev, "Failed to create queue %d, err %d\n", i, err);
			goto destroy_queues;
		}
		j++;
	}

	err = mana_ib_gd_create_rc_qp(mdev, qp, attr, doorbell, flags);
	if (err) {
		ibdev_err(&mdev->ib_dev, "Failed to create rc qp  %d\n", err);
		goto destroy_queues;
	}
	qp->ibqp.qp_num = qp->rc_qp.queues[MANA_RC_RECV_QUEUE_RESPONDER].id;
	qp->port = attr->port_num;

	if (udata) {
		for (i = 0, j = 0; i < MANA_RC_QUEUE_TYPE_MAX; ++i) {
			if (i == MANA_RC_SEND_QUEUE_FMR)
				continue;
			resp.queue_id[j] = qp->rc_qp.queues[i].id;
			j++;
		}
		err = ib_copy_to_udata(udata, &resp, min(sizeof(resp), udata->outlen));
		if (err) {
			ibdev_dbg(&mdev->ib_dev, "Failed to copy to udata, %d\n", err);
			goto destroy_qp;
		}
	}

	err = mana_table_store_qp(mdev, qp);
	if (err)
		goto destroy_qp;

	return 0;

destroy_qp:
	mana_ib_gd_destroy_rc_qp(mdev, qp);
destroy_queues:
	while (i-- > 0)
		mana_ib_destroy_queue(mdev, &qp->rc_qp.queues[i]);
	return err;
}

static void mana_add_qp_to_cqs(struct mana_ib_qp *qp)
{
	struct mana_ib_cq *send_cq = container_of(qp->ibqp.send_cq, struct mana_ib_cq, ibcq);
	struct mana_ib_cq *recv_cq = container_of(qp->ibqp.recv_cq, struct mana_ib_cq, ibcq);
	unsigned long flags;

	spin_lock_irqsave(&send_cq->cq_lock, flags);
	list_add_tail(&qp->cq_send_list, &send_cq->list_send_qp);
	spin_unlock_irqrestore(&send_cq->cq_lock, flags);

	spin_lock_irqsave(&recv_cq->cq_lock, flags);
	list_add_tail(&qp->cq_recv_list, &recv_cq->list_recv_qp);
	spin_unlock_irqrestore(&recv_cq->cq_lock, flags);
}

static void mana_remove_qp_from_cqs(struct mana_ib_qp *qp)
{
	struct mana_ib_cq *send_cq = container_of(qp->ibqp.send_cq, struct mana_ib_cq, ibcq);
	struct mana_ib_cq *recv_cq = container_of(qp->ibqp.recv_cq, struct mana_ib_cq, ibcq);
	unsigned long flags;

	spin_lock_irqsave(&send_cq->cq_lock, flags);
	list_del(&qp->cq_send_list);
	spin_unlock_irqrestore(&send_cq->cq_lock, flags);

	spin_lock_irqsave(&recv_cq->cq_lock, flags);
	list_del(&qp->cq_recv_list);
	spin_unlock_irqrestore(&recv_cq->cq_lock, flags);
}

static int mana_ib_create_ud_qp(struct ib_qp *ibqp, struct ib_pd *ibpd,
				struct ib_qp_init_attr *attr, struct ib_udata *udata)
{
	struct mana_ib_dev *mdev = container_of(ibpd->device, struct mana_ib_dev, ib_dev);
	struct mana_ib_qp *qp = container_of(ibqp, struct mana_ib_qp, ibqp);
	u32 doorbell, queue_size;
	int i, err;

	if (udata) {
		ibdev_dbg(&mdev->ib_dev, "User-level UD QPs are not supported\n");
		return -EOPNOTSUPP;
	}

	for (i = 0; i < MANA_UD_QUEUE_TYPE_MAX; ++i) {
		queue_size = mana_ib_queue_size(attr, i);
		err = mana_ib_create_kernel_queue(mdev, queue_size, mana_ib_queue_type(attr, i),
						  &qp->ud_qp.queues[i]);
		if (err) {
			ibdev_err(&mdev->ib_dev, "Failed to create queue %d, err %d\n",
				  i, err);
			goto destroy_queues;
		}
	}
	doorbell = mdev->gdma_dev->doorbell;

	err = create_shadow_queue(&qp->shadow_rq, attr->cap.max_recv_wr,
				  sizeof(struct ud_rq_shadow_wqe));
	if (err) {
		ibdev_err(&mdev->ib_dev, "Failed to create shadow rq err %d\n", err);
		goto destroy_queues;
	}
	err = create_shadow_queue(&qp->shadow_sq, attr->cap.max_send_wr,
				  sizeof(struct ud_sq_shadow_wqe));
	if (err) {
		ibdev_err(&mdev->ib_dev, "Failed to create shadow sq err %d\n", err);
		goto destroy_shadow_queues;
	}

	err = mana_ib_gd_create_ud_qp(mdev, qp, attr, doorbell, attr->qp_type);
	if (err) {
		ibdev_err(&mdev->ib_dev, "Failed to create ud qp  %d\n", err);
		goto destroy_shadow_queues;
	}
	qp->ibqp.qp_num = qp->ud_qp.queues[MANA_UD_RECV_QUEUE].id;
	qp->port = attr->port_num;

	for (i = 0; i < MANA_UD_QUEUE_TYPE_MAX; ++i)
		qp->ud_qp.queues[i].kmem->id = qp->ud_qp.queues[i].id;

	err = mana_table_store_qp(mdev, qp);
	if (err)
		goto destroy_qp;

	mana_add_qp_to_cqs(qp);

	return 0;

destroy_qp:
	mana_ib_gd_destroy_ud_qp(mdev, qp);
destroy_shadow_queues:
	destroy_shadow_queue(&qp->shadow_rq);
	destroy_shadow_queue(&qp->shadow_sq);
destroy_queues:
	while (i-- > 0)
		mana_ib_destroy_queue(mdev, &qp->ud_qp.queues[i]);
	return err;
}

int mana_ib_create_qp(struct ib_qp *ibqp, struct ib_qp_init_attr *attr,
		      struct ib_udata *udata)
{
	switch (attr->qp_type) {
	case IB_QPT_RAW_PACKET:
		/* When rwq_ind_tbl is used, it's for creating WQs for RSS */
		if (attr->rwq_ind_tbl)
			return mana_ib_create_qp_rss(ibqp, ibqp->pd, attr,
						     udata);

		return mana_ib_create_qp_raw(ibqp, ibqp->pd, attr, udata);
	case IB_QPT_RC:
		return mana_ib_create_rc_qp(ibqp, ibqp->pd, attr, udata);
	case IB_QPT_UD:
	case IB_QPT_GSI:
		return mana_ib_create_ud_qp(ibqp, ibqp->pd, attr, udata);
	default:
		ibdev_dbg(ibqp->device, "Creating QP type %u not supported\n",
			  attr->qp_type);
	}

	return -EINVAL;
}

static int mana_ib_gd_modify_qp(struct ib_qp *ibqp, struct ib_qp_attr *attr,
				int attr_mask, struct ib_udata *udata)
{
	struct mana_ib_dev *mdev = container_of(ibqp->device, struct mana_ib_dev, ib_dev);
	struct mana_ib_qp *qp = container_of(ibqp, struct mana_ib_qp, ibqp);
	struct mana_rnic_set_qp_state_resp resp = {};
	struct mana_rnic_set_qp_state_req req = {};
	struct gdma_context *gc = mdev_to_gc(mdev);
	struct mana_port_context *mpc;
	struct net_device *ndev;
	int err;

	mana_gd_init_req_hdr(&req.hdr, MANA_IB_SET_QP_STATE, sizeof(req), sizeof(resp));
	req.hdr.dev_id = mdev->gdma_dev->dev_id;
	req.adapter = mdev->adapter_handle;
	req.qp_handle = qp->qp_handle;
	req.qp_state = attr->qp_state;
	req.attr_mask = attr_mask;
	req.path_mtu = attr->path_mtu;
	req.rq_psn = attr->rq_psn;
	req.sq_psn = attr->sq_psn;
	req.dest_qpn = attr->dest_qp_num;
	req.max_dest_rd_atomic = attr->max_dest_rd_atomic;
	req.retry_cnt = attr->retry_cnt;
	req.rnr_retry = attr->rnr_retry;
	req.min_rnr_timer = attr->min_rnr_timer;
	if (attr_mask & IB_QP_AV) {
		ndev = mana_ib_get_netdev(&mdev->ib_dev, ibqp->port);
		if (!ndev) {
			ibdev_dbg(&mdev->ib_dev, "Invalid port %u in QP %u\n",
				  ibqp->port, ibqp->qp_num);
			return -EINVAL;
		}
		mpc = netdev_priv(ndev);
		copy_in_reverse(req.ah_attr.src_mac, mpc->mac_addr, ETH_ALEN);
		copy_in_reverse(req.ah_attr.dest_mac, attr->ah_attr.roce.dmac, ETH_ALEN);
		copy_in_reverse(req.ah_attr.src_addr, attr->ah_attr.grh.sgid_attr->gid.raw,
				sizeof(union ib_gid));
		copy_in_reverse(req.ah_attr.dest_addr, attr->ah_attr.grh.dgid.raw,
				sizeof(union ib_gid));
		if (rdma_gid_attr_network_type(attr->ah_attr.grh.sgid_attr) == RDMA_NETWORK_IPV4) {
			req.ah_attr.src_addr_type = SGID_TYPE_IPV4;
			req.ah_attr.dest_addr_type = SGID_TYPE_IPV4;
		} else {
			req.ah_attr.src_addr_type = SGID_TYPE_IPV6;
			req.ah_attr.dest_addr_type = SGID_TYPE_IPV6;
		}
		req.ah_attr.dest_port = ROCE_V2_UDP_DPORT;
		req.ah_attr.src_port = rdma_get_udp_sport(attr->ah_attr.grh.flow_label,
							  ibqp->qp_num, attr->dest_qp_num);
		req.ah_attr.traffic_class = attr->ah_attr.grh.traffic_class >> 2;
		req.ah_attr.hop_limit = attr->ah_attr.grh.hop_limit;
	}

	err = mana_gd_send_request(gc, sizeof(req), &req, sizeof(resp), &resp);
	if (err) {
		ibdev_err(&mdev->ib_dev, "Failed modify qp err %d", err);
		return err;
	}

	return 0;
}

int mana_ib_modify_qp(struct ib_qp *ibqp, struct ib_qp_attr *attr,
		      int attr_mask, struct ib_udata *udata)
{
	switch (ibqp->qp_type) {
	case IB_QPT_RC:
	case IB_QPT_UD:
	case IB_QPT_GSI:
		return mana_ib_gd_modify_qp(ibqp, attr, attr_mask, udata);
	default:
		ibdev_dbg(ibqp->device, "Modify QP type %u not supported", ibqp->qp_type);
		return -EOPNOTSUPP;
	}
}

static int mana_ib_destroy_qp_rss(struct mana_ib_qp *qp,
				  struct ib_rwq_ind_table *ind_tbl,
				  struct ib_udata *udata)
{
	struct mana_ib_dev *mdev =
		container_of(qp->ibqp.device, struct mana_ib_dev, ib_dev);
	struct mana_port_context *mpc;
	struct net_device *ndev;
	struct mana_ib_wq *wq;
	struct ib_wq *ibwq;
	int i;

	ndev = mana_ib_get_netdev(qp->ibqp.device, qp->port);
	mpc = netdev_priv(ndev);

	for (i = 0; i < (1 << ind_tbl->log_ind_tbl_size); i++) {
		ibwq = ind_tbl->ind_tbl[i];
		wq = container_of(ibwq, struct mana_ib_wq, ibwq);
		ibdev_dbg(&mdev->ib_dev, "destroying wq->rx_object %llu\n",
			  wq->rx_object);
		mana_destroy_wq_obj(mpc, GDMA_RQ, wq->rx_object);
	}

	return 0;
}

static int mana_ib_destroy_qp_raw(struct mana_ib_qp *qp, struct ib_udata *udata)
{
	struct mana_ib_dev *mdev =
		container_of(qp->ibqp.device, struct mana_ib_dev, ib_dev);
	struct ib_pd *ibpd = qp->ibqp.pd;
	struct mana_port_context *mpc;
	struct net_device *ndev;
	struct mana_ib_pd *pd;

	ndev = mana_ib_get_netdev(qp->ibqp.device, qp->port);
	mpc = netdev_priv(ndev);
	pd = container_of(ibpd, struct mana_ib_pd, ibpd);

	mana_destroy_wq_obj(mpc, GDMA_SQ, qp->qp_handle);

	mana_ib_destroy_queue(mdev, &qp->raw_sq);

	mana_ib_uncfg_vport(mdev, pd, qp->port);

	return 0;
}

static int mana_ib_destroy_rc_qp(struct mana_ib_qp *qp, struct ib_udata *udata)
{
	struct mana_ib_dev *mdev =
		container_of(qp->ibqp.device, struct mana_ib_dev, ib_dev);
	int i;

	mana_table_remove_qp(mdev, qp);

	/* Ignore return code as there is not much we can do about it.
	 * The error message is printed inside.
	 */
	mana_ib_gd_destroy_rc_qp(mdev, qp);
	for (i = 0; i < MANA_RC_QUEUE_TYPE_MAX; ++i)
		mana_ib_destroy_queue(mdev, &qp->rc_qp.queues[i]);

	return 0;
}

static int mana_ib_destroy_ud_qp(struct mana_ib_qp *qp, struct ib_udata *udata)
{
	struct mana_ib_dev *mdev =
		container_of(qp->ibqp.device, struct mana_ib_dev, ib_dev);
	int i;

	mana_remove_qp_from_cqs(qp);
	mana_table_remove_qp(mdev, qp);

	destroy_shadow_queue(&qp->shadow_rq);
	destroy_shadow_queue(&qp->shadow_sq);

	/* Ignore return code as there is not much we can do about it.
	 * The error message is printed inside.
	 */
	mana_ib_gd_destroy_ud_qp(mdev, qp);
	for (i = 0; i < MANA_UD_QUEUE_TYPE_MAX; ++i)
		mana_ib_destroy_queue(mdev, &qp->ud_qp.queues[i]);

	return 0;
}

int mana_ib_destroy_qp(struct ib_qp *ibqp, struct ib_udata *udata)
{
	struct mana_ib_qp *qp = container_of(ibqp, struct mana_ib_qp, ibqp);

	switch (ibqp->qp_type) {
	case IB_QPT_RAW_PACKET:
		if (ibqp->rwq_ind_tbl)
			return mana_ib_destroy_qp_rss(qp, ibqp->rwq_ind_tbl,
						      udata);

		return mana_ib_destroy_qp_raw(qp, udata);
	case IB_QPT_RC:
		return mana_ib_destroy_rc_qp(qp, udata);
	case IB_QPT_UD:
	case IB_QPT_GSI:
		return mana_ib_destroy_ud_qp(qp, udata);
	default:
		ibdev_dbg(ibqp->device, "Unexpected QP type %u\n",
			  ibqp->qp_type);
	}

	return -ENOENT;
}