brintos

brintos / linux-shallow public Read only

0
0
Text · 20.0 KiB · 8b33853 Raw
775 lines · c
1// SPDX-License-Identifier: GPL-2.0 OR Linux-OpenIB2/*3 * Copyright (c) 2013-2018, Mellanox Technologies inc.  All rights reserved.4 */5 6#include <linux/kernel.h>7#include <linux/mlx5/driver.h>8#include "mlx5_ib.h"9#include "srq.h"10#include "qp.h"11 12static int get_pas_size(struct mlx5_srq_attr *in)13{14	u32 log_page_size = in->log_page_size + 12;15	u32 log_srq_size  = in->log_size;16	u32 log_rq_stride = in->wqe_shift;17	u32 page_offset   = in->page_offset;18	u32 po_quanta	  = 1 << (log_page_size - 6);19	u32 rq_sz	  = 1 << (log_srq_size + 4 + log_rq_stride);20	u32 page_size	  = 1 << log_page_size;21	u32 rq_sz_po      = rq_sz + (page_offset * po_quanta);22	u32 rq_num_pas    = DIV_ROUND_UP(rq_sz_po, page_size);23 24	return rq_num_pas * sizeof(u64);25}26 27static void set_wq(void *wq, struct mlx5_srq_attr *in)28{29	MLX5_SET(wq,   wq, wq_signature,  !!(in->flags30		 & MLX5_SRQ_FLAG_WQ_SIG));31	MLX5_SET(wq,   wq, log_wq_pg_sz,  in->log_page_size);32	MLX5_SET(wq,   wq, log_wq_stride, in->wqe_shift + 4);33	MLX5_SET(wq,   wq, log_wq_sz,     in->log_size);34	MLX5_SET(wq,   wq, page_offset,   in->page_offset);35	MLX5_SET(wq,   wq, lwm,		  in->lwm);36	MLX5_SET(wq,   wq, pd,		  in->pd);37	MLX5_SET64(wq, wq, dbr_addr,	  in->db_record);38}39 40static void set_srqc(void *srqc, struct mlx5_srq_attr *in)41{42	MLX5_SET(srqc,   srqc, wq_signature,  !!(in->flags43		 & MLX5_SRQ_FLAG_WQ_SIG));44	MLX5_SET(srqc,   srqc, log_page_size, in->log_page_size);45	MLX5_SET(srqc,   srqc, log_rq_stride, in->wqe_shift);46	MLX5_SET(srqc,   srqc, log_srq_size,  in->log_size);47	MLX5_SET(srqc,   srqc, page_offset,   in->page_offset);48	MLX5_SET(srqc,	 srqc, lwm,	      in->lwm);49	MLX5_SET(srqc,	 srqc, pd,	      in->pd);50	MLX5_SET64(srqc, srqc, dbr_addr,      in->db_record);51	MLX5_SET(srqc,	 srqc, xrcd,	      in->xrcd);52	MLX5_SET(srqc,	 srqc, cqn,	      in->cqn);53}54 55static void get_wq(void *wq, struct mlx5_srq_attr *in)56{57	if (MLX5_GET(wq, wq, wq_signature))58		in->flags &= MLX5_SRQ_FLAG_WQ_SIG;59	in->log_page_size = MLX5_GET(wq,   wq, log_wq_pg_sz);60	in->wqe_shift	  = MLX5_GET(wq,   wq, log_wq_stride) - 4;61	in->log_size	  = MLX5_GET(wq,   wq, log_wq_sz);62	in->page_offset   = MLX5_GET(wq,   wq, page_offset);63	in->lwm		  = MLX5_GET(wq,   wq, lwm);64	in->pd		  = MLX5_GET(wq,   wq, pd);65	in->db_record	  = MLX5_GET64(wq, wq, dbr_addr);66}67 68static void get_srqc(void *srqc, struct mlx5_srq_attr *in)69{70	if (MLX5_GET(srqc, srqc, wq_signature))71		in->flags &= MLX5_SRQ_FLAG_WQ_SIG;72	in->log_page_size = MLX5_GET(srqc,   srqc, log_page_size);73	in->wqe_shift	  = MLX5_GET(srqc,   srqc, log_rq_stride);74	in->log_size	  = MLX5_GET(srqc,   srqc, log_srq_size);75	in->page_offset   = MLX5_GET(srqc,   srqc, page_offset);76	in->lwm		  = MLX5_GET(srqc,   srqc, lwm);77	in->pd		  = MLX5_GET(srqc,   srqc, pd);78	in->db_record	  = MLX5_GET64(srqc, srqc, dbr_addr);79}80 81struct mlx5_core_srq *mlx5_cmd_get_srq(struct mlx5_ib_dev *dev, u32 srqn)82{83	struct mlx5_srq_table *table = &dev->srq_table;84	struct mlx5_core_srq *srq;85 86	xa_lock_irq(&table->array);87	srq = xa_load(&table->array, srqn);88	if (srq)89		refcount_inc(&srq->common.refcount);90	xa_unlock_irq(&table->array);91 92	return srq;93}94 95static int __set_srq_page_size(struct mlx5_srq_attr *in,96			       unsigned long page_size)97{98	if (!page_size)99		return -EINVAL;100	in->log_page_size = order_base_2(page_size) - MLX5_ADAPTER_PAGE_SHIFT;101 102	if (WARN_ON(get_pas_size(in) !=103		    ib_umem_num_dma_blocks(in->umem, page_size) * sizeof(u64)))104		return -EINVAL;105	return 0;106}107 108#define set_srq_page_size(in, typ, log_pgsz_fld)                               \109	__set_srq_page_size(in, mlx5_umem_find_best_quantized_pgoff(           \110					(in)->umem, typ, log_pgsz_fld,         \111					MLX5_ADAPTER_PAGE_SHIFT, page_offset,  \112					64, &(in)->page_offset))113 114static int create_srq_cmd(struct mlx5_ib_dev *dev, struct mlx5_core_srq *srq,115			  struct mlx5_srq_attr *in)116{117	u32 create_out[MLX5_ST_SZ_DW(create_srq_out)] = {0};118	void *create_in;119	void *srqc;120	void *pas;121	int pas_size;122	int inlen;123	int err;124 125	if (in->umem) {126		err = set_srq_page_size(in, srqc, log_page_size);127		if (err)128			return err;129	}130 131	pas_size  = get_pas_size(in);132	inlen	  = MLX5_ST_SZ_BYTES(create_srq_in) + pas_size;133	create_in = kvzalloc(inlen, GFP_KERNEL);134	if (!create_in)135		return -ENOMEM;136 137	MLX5_SET(create_srq_in, create_in, uid, in->uid);138	srqc = MLX5_ADDR_OF(create_srq_in, create_in, srq_context_entry);139	pas = MLX5_ADDR_OF(create_srq_in, create_in, pas);140 141	set_srqc(srqc, in);142	if (in->umem)143		mlx5_ib_populate_pas(144			in->umem,145			1UL << (in->log_page_size + MLX5_ADAPTER_PAGE_SHIFT),146			pas, 0);147	else148		memcpy(pas, in->pas, pas_size);149 150	MLX5_SET(create_srq_in, create_in, opcode,151		 MLX5_CMD_OP_CREATE_SRQ);152 153	err = mlx5_cmd_exec(dev->mdev, create_in, inlen, create_out,154			    sizeof(create_out));155	kvfree(create_in);156	if (!err) {157		srq->srqn = MLX5_GET(create_srq_out, create_out, srqn);158		srq->uid = in->uid;159	}160 161	return err;162}163 164static int destroy_srq_cmd(struct mlx5_ib_dev *dev, struct mlx5_core_srq *srq)165{166	u32 in[MLX5_ST_SZ_DW(destroy_srq_in)] = {};167 168	MLX5_SET(destroy_srq_in, in, opcode, MLX5_CMD_OP_DESTROY_SRQ);169	MLX5_SET(destroy_srq_in, in, srqn, srq->srqn);170	MLX5_SET(destroy_srq_in, in, uid, srq->uid);171 172	return mlx5_cmd_exec_in(dev->mdev, destroy_srq, in);173}174 175static int arm_srq_cmd(struct mlx5_ib_dev *dev, struct mlx5_core_srq *srq,176		       u16 lwm, int is_srq)177{178	u32 in[MLX5_ST_SZ_DW(arm_rq_in)] = {};179 180	MLX5_SET(arm_rq_in, in, opcode, MLX5_CMD_OP_ARM_RQ);181	MLX5_SET(arm_rq_in, in, op_mod, MLX5_ARM_RQ_IN_OP_MOD_SRQ);182	MLX5_SET(arm_rq_in, in, srq_number, srq->srqn);183	MLX5_SET(arm_rq_in, in, lwm, lwm);184	MLX5_SET(arm_rq_in, in, uid, srq->uid);185 186	return mlx5_cmd_exec_in(dev->mdev, arm_rq, in);187}188 189static int query_srq_cmd(struct mlx5_ib_dev *dev, struct mlx5_core_srq *srq,190			 struct mlx5_srq_attr *out)191{192	u32 in[MLX5_ST_SZ_DW(query_srq_in)] = {};193	u32 *srq_out;194	void *srqc;195	int err;196 197	srq_out = kvzalloc(MLX5_ST_SZ_BYTES(query_srq_out), GFP_KERNEL);198	if (!srq_out)199		return -ENOMEM;200 201	MLX5_SET(query_srq_in, in, opcode, MLX5_CMD_OP_QUERY_SRQ);202	MLX5_SET(query_srq_in, in, srqn, srq->srqn);203	err = mlx5_cmd_exec_inout(dev->mdev, query_srq, in, srq_out);204	if (err)205		goto out;206 207	srqc = MLX5_ADDR_OF(query_srq_out, srq_out, srq_context_entry);208	get_srqc(srqc, out);209	if (MLX5_GET(srqc, srqc, state) != MLX5_SRQC_STATE_GOOD)210		out->flags |= MLX5_SRQ_FLAG_ERR;211out:212	kvfree(srq_out);213	return err;214}215 216static int create_xrc_srq_cmd(struct mlx5_ib_dev *dev,217			      struct mlx5_core_srq *srq,218			      struct mlx5_srq_attr *in)219{220	u32 create_out[MLX5_ST_SZ_DW(create_xrc_srq_out)];221	void *create_in;222	void *xrc_srqc;223	void *pas;224	int pas_size;225	int inlen;226	int err;227 228	if (in->umem) {229		err = set_srq_page_size(in, xrc_srqc, log_page_size);230		if (err)231			return err;232	}233 234	pas_size  = get_pas_size(in);235	inlen	  = MLX5_ST_SZ_BYTES(create_xrc_srq_in) + pas_size;236	create_in = kvzalloc(inlen, GFP_KERNEL);237	if (!create_in)238		return -ENOMEM;239 240	MLX5_SET(create_xrc_srq_in, create_in, uid, in->uid);241	xrc_srqc = MLX5_ADDR_OF(create_xrc_srq_in, create_in,242				xrc_srq_context_entry);243	pas	 = MLX5_ADDR_OF(create_xrc_srq_in, create_in, pas);244 245	set_srqc(xrc_srqc, in);246	MLX5_SET(xrc_srqc, xrc_srqc, user_index, in->user_index);247	if (in->umem)248		mlx5_ib_populate_pas(249			in->umem,250			1UL << (in->log_page_size + MLX5_ADAPTER_PAGE_SHIFT),251			pas, 0);252	else253		memcpy(pas, in->pas, pas_size);254	MLX5_SET(create_xrc_srq_in, create_in, opcode,255		 MLX5_CMD_OP_CREATE_XRC_SRQ);256 257	memset(create_out, 0, sizeof(create_out));258	err = mlx5_cmd_exec(dev->mdev, create_in, inlen, create_out,259			    sizeof(create_out));260	if (err)261		goto out;262 263	srq->srqn = MLX5_GET(create_xrc_srq_out, create_out, xrc_srqn);264	srq->uid = in->uid;265out:266	kvfree(create_in);267	return err;268}269 270static int destroy_xrc_srq_cmd(struct mlx5_ib_dev *dev,271			       struct mlx5_core_srq *srq)272{273	u32 in[MLX5_ST_SZ_DW(destroy_xrc_srq_in)] = {};274 275	MLX5_SET(destroy_xrc_srq_in, in, opcode, MLX5_CMD_OP_DESTROY_XRC_SRQ);276	MLX5_SET(destroy_xrc_srq_in, in, xrc_srqn, srq->srqn);277	MLX5_SET(destroy_xrc_srq_in, in, uid, srq->uid);278 279	return mlx5_cmd_exec_in(dev->mdev, destroy_xrc_srq, in);280}281 282static int arm_xrc_srq_cmd(struct mlx5_ib_dev *dev, struct mlx5_core_srq *srq,283			   u16 lwm)284{285	u32 in[MLX5_ST_SZ_DW(arm_xrc_srq_in)] = {};286 287	MLX5_SET(arm_xrc_srq_in, in, opcode, MLX5_CMD_OP_ARM_XRC_SRQ);288	MLX5_SET(arm_xrc_srq_in, in, op_mod,289		 MLX5_ARM_XRC_SRQ_IN_OP_MOD_XRC_SRQ);290	MLX5_SET(arm_xrc_srq_in, in, xrc_srqn, srq->srqn);291	MLX5_SET(arm_xrc_srq_in, in, lwm, lwm);292	MLX5_SET(arm_xrc_srq_in, in, uid, srq->uid);293 294	return  mlx5_cmd_exec_in(dev->mdev, arm_xrc_srq, in);295}296 297static int query_xrc_srq_cmd(struct mlx5_ib_dev *dev,298			     struct mlx5_core_srq *srq,299			     struct mlx5_srq_attr *out)300{301	u32 in[MLX5_ST_SZ_DW(query_xrc_srq_in)] = {};302	u32 *xrcsrq_out;303	void *xrc_srqc;304	int err;305 306	xrcsrq_out = kvzalloc(MLX5_ST_SZ_BYTES(query_xrc_srq_out), GFP_KERNEL);307	if (!xrcsrq_out)308		return -ENOMEM;309 310	MLX5_SET(query_xrc_srq_in, in, opcode, MLX5_CMD_OP_QUERY_XRC_SRQ);311	MLX5_SET(query_xrc_srq_in, in, xrc_srqn, srq->srqn);312 313	err = mlx5_cmd_exec_inout(dev->mdev, query_xrc_srq, in, xrcsrq_out);314	if (err)315		goto out;316 317	xrc_srqc = MLX5_ADDR_OF(query_xrc_srq_out, xrcsrq_out,318				xrc_srq_context_entry);319	get_srqc(xrc_srqc, out);320	if (MLX5_GET(xrc_srqc, xrc_srqc, state) != MLX5_XRC_SRQC_STATE_GOOD)321		out->flags |= MLX5_SRQ_FLAG_ERR;322 323out:324	kvfree(xrcsrq_out);325	return err;326}327 328static int create_rmp_cmd(struct mlx5_ib_dev *dev, struct mlx5_core_srq *srq,329			  struct mlx5_srq_attr *in)330{331	void *create_out = NULL;332	void *create_in = NULL;333	void *rmpc;334	void *wq;335	void *pas;336	int pas_size;337	int outlen;338	int inlen;339	int err;340 341	if (in->umem) {342		err = set_srq_page_size(in, wq, log_wq_pg_sz);343		if (err)344			return err;345	}346 347	pas_size = get_pas_size(in);348	inlen = MLX5_ST_SZ_BYTES(create_rmp_in) + pas_size;349	outlen = MLX5_ST_SZ_BYTES(create_rmp_out);350	create_in = kvzalloc(inlen, GFP_KERNEL);351	create_out = kvzalloc(outlen, GFP_KERNEL);352	if (!create_in || !create_out) {353		err = -ENOMEM;354		goto out;355	}356 357	rmpc = MLX5_ADDR_OF(create_rmp_in, create_in, ctx);358	wq = MLX5_ADDR_OF(rmpc, rmpc, wq);359 360	MLX5_SET(rmpc, rmpc, state, MLX5_RMPC_STATE_RDY);361	MLX5_SET(create_rmp_in, create_in, uid, in->uid);362	pas = MLX5_ADDR_OF(rmpc, rmpc, wq.pas);363 364	set_wq(wq, in);365	if (in->umem)366		mlx5_ib_populate_pas(367			in->umem,368			1UL << (in->log_page_size + MLX5_ADAPTER_PAGE_SHIFT),369			pas, 0);370	else371		memcpy(pas, in->pas, pas_size);372 373	MLX5_SET(create_rmp_in, create_in, opcode, MLX5_CMD_OP_CREATE_RMP);374	err = mlx5_cmd_exec(dev->mdev, create_in, inlen, create_out, outlen);375	if (!err) {376		srq->srqn = MLX5_GET(create_rmp_out, create_out, rmpn);377		srq->uid = in->uid;378	}379 380out:381	kvfree(create_in);382	kvfree(create_out);383	return err;384}385 386static int destroy_rmp_cmd(struct mlx5_ib_dev *dev, struct mlx5_core_srq *srq)387{388	u32 in[MLX5_ST_SZ_DW(destroy_rmp_in)] = {};389 390	MLX5_SET(destroy_rmp_in, in, opcode, MLX5_CMD_OP_DESTROY_RMP);391	MLX5_SET(destroy_rmp_in, in, rmpn, srq->srqn);392	MLX5_SET(destroy_rmp_in, in, uid, srq->uid);393	return mlx5_cmd_exec_in(dev->mdev, destroy_rmp, in);394}395 396static int arm_rmp_cmd(struct mlx5_ib_dev *dev, struct mlx5_core_srq *srq,397		       u16 lwm)398{399	void *out = NULL;400	void *in = NULL;401	void *rmpc;402	void *wq;403	void *bitmask;404	int outlen;405	int inlen;406	int err;407 408	inlen = MLX5_ST_SZ_BYTES(modify_rmp_in);409	outlen = MLX5_ST_SZ_BYTES(modify_rmp_out);410 411	in = kvzalloc(inlen, GFP_KERNEL);412	out = kvzalloc(outlen, GFP_KERNEL);413	if (!in || !out) {414		err = -ENOMEM;415		goto out;416	}417 418	rmpc =	  MLX5_ADDR_OF(modify_rmp_in,   in,   ctx);419	bitmask = MLX5_ADDR_OF(modify_rmp_in,   in,   bitmask);420	wq   =	  MLX5_ADDR_OF(rmpc,	        rmpc, wq);421 422	MLX5_SET(modify_rmp_in, in,	 rmp_state, MLX5_RMPC_STATE_RDY);423	MLX5_SET(modify_rmp_in, in,	 rmpn,      srq->srqn);424	MLX5_SET(modify_rmp_in, in, uid, srq->uid);425	MLX5_SET(wq,		wq,	 lwm,	    lwm);426	MLX5_SET(rmp_bitmask,	bitmask, lwm,	    1);427	MLX5_SET(rmpc, rmpc, state, MLX5_RMPC_STATE_RDY);428	MLX5_SET(modify_rmp_in, in, opcode, MLX5_CMD_OP_MODIFY_RMP);429 430	err = mlx5_cmd_exec_inout(dev->mdev, modify_rmp, in, out);431 432out:433	kvfree(in);434	kvfree(out);435	return err;436}437 438static int query_rmp_cmd(struct mlx5_ib_dev *dev, struct mlx5_core_srq *srq,439			 struct mlx5_srq_attr *out)440{441	u32 *rmp_out = NULL;442	u32 *rmp_in = NULL;443	void *rmpc;444	int outlen;445	int inlen;446	int err;447 448	outlen = MLX5_ST_SZ_BYTES(query_rmp_out);449	inlen = MLX5_ST_SZ_BYTES(query_rmp_in);450 451	rmp_out = kvzalloc(outlen, GFP_KERNEL);452	rmp_in = kvzalloc(inlen, GFP_KERNEL);453	if (!rmp_out || !rmp_in) {454		err = -ENOMEM;455		goto out;456	}457 458	MLX5_SET(query_rmp_in, rmp_in, opcode, MLX5_CMD_OP_QUERY_RMP);459	MLX5_SET(query_rmp_in, rmp_in, rmpn,   srq->srqn);460	err = mlx5_cmd_exec_inout(dev->mdev, query_rmp, rmp_in, rmp_out);461	if (err)462		goto out;463 464	rmpc = MLX5_ADDR_OF(query_rmp_out, rmp_out, rmp_context);465	get_wq(MLX5_ADDR_OF(rmpc, rmpc, wq), out);466	if (MLX5_GET(rmpc, rmpc, state) != MLX5_RMPC_STATE_RDY)467		out->flags |= MLX5_SRQ_FLAG_ERR;468 469out:470	kvfree(rmp_out);471	kvfree(rmp_in);472	return err;473}474 475static int create_xrq_cmd(struct mlx5_ib_dev *dev, struct mlx5_core_srq *srq,476			  struct mlx5_srq_attr *in)477{478	u32 create_out[MLX5_ST_SZ_DW(create_xrq_out)] = {0};479	void *create_in;480	void *xrqc;481	void *wq;482	void *pas;483	int pas_size;484	int inlen;485	int err;486 487	if (in->umem) {488		err = set_srq_page_size(in, wq, log_wq_pg_sz);489		if (err)490			return err;491	}492 493	pas_size = get_pas_size(in);494	inlen = MLX5_ST_SZ_BYTES(create_xrq_in) + pas_size;495	create_in = kvzalloc(inlen, GFP_KERNEL);496	if (!create_in)497		return -ENOMEM;498 499	xrqc = MLX5_ADDR_OF(create_xrq_in, create_in, xrq_context);500	wq = MLX5_ADDR_OF(xrqc, xrqc, wq);501	pas = MLX5_ADDR_OF(xrqc, xrqc, wq.pas);502 503	set_wq(wq, in);504	if (in->umem)505		mlx5_ib_populate_pas(506			in->umem,507			1UL << (in->log_page_size + MLX5_ADAPTER_PAGE_SHIFT),508			pas, 0);509	else510		memcpy(pas, in->pas, pas_size);511 512	if (in->type == IB_SRQT_TM) {513		MLX5_SET(xrqc, xrqc, topology, MLX5_XRQC_TOPOLOGY_TAG_MATCHING);514		if (in->flags & MLX5_SRQ_FLAG_RNDV)515			MLX5_SET(xrqc, xrqc, offload, MLX5_XRQC_OFFLOAD_RNDV);516		MLX5_SET(xrqc, xrqc,517			 tag_matching_topology_context.log_matching_list_sz,518			 in->tm_log_list_size);519	}520	MLX5_SET(xrqc, xrqc, user_index, in->user_index);521	MLX5_SET(xrqc, xrqc, cqn, in->cqn);522	MLX5_SET(create_xrq_in, create_in, opcode, MLX5_CMD_OP_CREATE_XRQ);523	MLX5_SET(create_xrq_in, create_in, uid, in->uid);524	err = mlx5_cmd_exec(dev->mdev, create_in, inlen, create_out,525			    sizeof(create_out));526	kvfree(create_in);527	if (!err) {528		srq->srqn = MLX5_GET(create_xrq_out, create_out, xrqn);529		srq->uid = in->uid;530	}531 532	return err;533}534 535static int destroy_xrq_cmd(struct mlx5_ib_dev *dev, struct mlx5_core_srq *srq)536{537	u32 in[MLX5_ST_SZ_DW(destroy_xrq_in)] = {};538 539	MLX5_SET(destroy_xrq_in, in, opcode, MLX5_CMD_OP_DESTROY_XRQ);540	MLX5_SET(destroy_xrq_in, in, xrqn, srq->srqn);541	MLX5_SET(destroy_xrq_in, in, uid, srq->uid);542 543	return mlx5_cmd_exec_in(dev->mdev, destroy_xrq, in);544}545 546static int arm_xrq_cmd(struct mlx5_ib_dev *dev,547		       struct mlx5_core_srq *srq,548		       u16 lwm)549{550	u32 in[MLX5_ST_SZ_DW(arm_rq_in)] = {};551 552	MLX5_SET(arm_rq_in, in, opcode, MLX5_CMD_OP_ARM_RQ);553	MLX5_SET(arm_rq_in, in, op_mod, MLX5_ARM_RQ_IN_OP_MOD_XRQ);554	MLX5_SET(arm_rq_in, in, srq_number, srq->srqn);555	MLX5_SET(arm_rq_in, in, lwm, lwm);556	MLX5_SET(arm_rq_in, in, uid, srq->uid);557 558	return mlx5_cmd_exec_in(dev->mdev, arm_rq, in);559}560 561static int query_xrq_cmd(struct mlx5_ib_dev *dev, struct mlx5_core_srq *srq,562			 struct mlx5_srq_attr *out)563{564	u32 in[MLX5_ST_SZ_DW(query_xrq_in)] = {};565	u32 *xrq_out;566	int outlen = MLX5_ST_SZ_BYTES(query_xrq_out);567	void *xrqc;568	int err;569 570	xrq_out = kvzalloc(outlen, GFP_KERNEL);571	if (!xrq_out)572		return -ENOMEM;573 574	MLX5_SET(query_xrq_in, in, opcode, MLX5_CMD_OP_QUERY_XRQ);575	MLX5_SET(query_xrq_in, in, xrqn, srq->srqn);576 577	err = mlx5_cmd_exec_inout(dev->mdev, query_xrq, in, xrq_out);578	if (err)579		goto out;580 581	xrqc = MLX5_ADDR_OF(query_xrq_out, xrq_out, xrq_context);582	get_wq(MLX5_ADDR_OF(xrqc, xrqc, wq), out);583	if (MLX5_GET(xrqc, xrqc, state) != MLX5_XRQC_STATE_GOOD)584		out->flags |= MLX5_SRQ_FLAG_ERR;585	out->tm_next_tag =586		MLX5_GET(xrqc, xrqc,587			 tag_matching_topology_context.append_next_index);588	out->tm_hw_phase_cnt =589		MLX5_GET(xrqc, xrqc,590			 tag_matching_topology_context.hw_phase_cnt);591	out->tm_sw_phase_cnt =592		MLX5_GET(xrqc, xrqc,593			 tag_matching_topology_context.sw_phase_cnt);594 595out:596	kvfree(xrq_out);597	return err;598}599 600static int create_srq_split(struct mlx5_ib_dev *dev, struct mlx5_core_srq *srq,601			    struct mlx5_srq_attr *in)602{603	if (!dev->mdev->issi)604		return create_srq_cmd(dev, srq, in);605	switch (srq->common.res) {606	case MLX5_RES_XSRQ:607		return create_xrc_srq_cmd(dev, srq, in);608	case MLX5_RES_XRQ:609		return create_xrq_cmd(dev, srq, in);610	default:611		return create_rmp_cmd(dev, srq, in);612	}613}614 615static int destroy_srq_split(struct mlx5_ib_dev *dev, struct mlx5_core_srq *srq)616{617	if (!dev->mdev->issi)618		return destroy_srq_cmd(dev, srq);619	switch (srq->common.res) {620	case MLX5_RES_XSRQ:621		return destroy_xrc_srq_cmd(dev, srq);622	case MLX5_RES_XRQ:623		return destroy_xrq_cmd(dev, srq);624	default:625		return destroy_rmp_cmd(dev, srq);626	}627}628 629int mlx5_cmd_create_srq(struct mlx5_ib_dev *dev, struct mlx5_core_srq *srq,630			struct mlx5_srq_attr *in)631{632	struct mlx5_srq_table *table = &dev->srq_table;633	int err;634 635	switch (in->type) {636	case IB_SRQT_XRC:637		srq->common.res = MLX5_RES_XSRQ;638		break;639	case IB_SRQT_TM:640		srq->common.res = MLX5_RES_XRQ;641		break;642	default:643		srq->common.res = MLX5_RES_SRQ;644	}645 646	err = create_srq_split(dev, srq, in);647	if (err)648		return err;649 650	refcount_set(&srq->common.refcount, 1);651	init_completion(&srq->common.free);652 653	err = xa_err(xa_store_irq(&table->array, srq->srqn, srq, GFP_KERNEL));654	if (err)655		goto err_destroy_srq_split;656 657	return 0;658 659err_destroy_srq_split:660	destroy_srq_split(dev, srq);661 662	return err;663}664 665int mlx5_cmd_destroy_srq(struct mlx5_ib_dev *dev, struct mlx5_core_srq *srq)666{667	struct mlx5_srq_table *table = &dev->srq_table;668	struct mlx5_core_srq *tmp;669	int err;670 671	/* Delete entry, but leave index occupied */672	tmp = xa_cmpxchg_irq(&table->array, srq->srqn, srq, XA_ZERO_ENTRY, 0);673	if (WARN_ON(tmp != srq))674		return xa_err(tmp) ?: -EINVAL;675 676	err = destroy_srq_split(dev, srq);677	if (err) {678		/*679		 * We don't need to check returned result for an error,680		 * because  we are storing in pre-allocated space xarray681		 * entry and it can't fail at this stage.682		 */683		xa_cmpxchg_irq(&table->array, srq->srqn, XA_ZERO_ENTRY, srq, 0);684		return err;685	}686	xa_erase_irq(&table->array, srq->srqn);687 688	mlx5_core_res_put(&srq->common);689	wait_for_completion(&srq->common.free);690	return 0;691}692 693int mlx5_cmd_query_srq(struct mlx5_ib_dev *dev, struct mlx5_core_srq *srq,694		       struct mlx5_srq_attr *out)695{696	if (!dev->mdev->issi)697		return query_srq_cmd(dev, srq, out);698	switch (srq->common.res) {699	case MLX5_RES_XSRQ:700		return query_xrc_srq_cmd(dev, srq, out);701	case MLX5_RES_XRQ:702		return query_xrq_cmd(dev, srq, out);703	default:704		return query_rmp_cmd(dev, srq, out);705	}706}707 708int mlx5_cmd_arm_srq(struct mlx5_ib_dev *dev, struct mlx5_core_srq *srq,709		     u16 lwm, int is_srq)710{711	if (!dev->mdev->issi)712		return arm_srq_cmd(dev, srq, lwm, is_srq);713	switch (srq->common.res) {714	case MLX5_RES_XSRQ:715		return arm_xrc_srq_cmd(dev, srq, lwm);716	case MLX5_RES_XRQ:717		return arm_xrq_cmd(dev, srq, lwm);718	default:719		return arm_rmp_cmd(dev, srq, lwm);720	}721}722 723static int srq_event_notifier(struct notifier_block *nb,724			      unsigned long type, void *data)725{726	struct mlx5_srq_table *table;727	struct mlx5_core_srq *srq;728	struct mlx5_eqe *eqe;729	u32 srqn;730 731	if (type != MLX5_EVENT_TYPE_SRQ_CATAS_ERROR &&732	    type != MLX5_EVENT_TYPE_SRQ_RQ_LIMIT)733		return NOTIFY_DONE;734 735	table = container_of(nb, struct mlx5_srq_table, nb);736 737	eqe = data;738	srqn = be32_to_cpu(eqe->data.qp_srq.qp_srq_n) & 0xffffff;739 740	xa_lock(&table->array);741	srq = xa_load(&table->array, srqn);742	if (srq)743		refcount_inc(&srq->common.refcount);744	xa_unlock(&table->array);745 746	if (!srq)747		return NOTIFY_OK;748 749	srq->event(srq, eqe->type);750 751	mlx5_core_res_put(&srq->common);752 753	return NOTIFY_OK;754}755 756int mlx5_init_srq_table(struct mlx5_ib_dev *dev)757{758	struct mlx5_srq_table *table = &dev->srq_table;759 760	memset(table, 0, sizeof(*table));761	xa_init_flags(&table->array, XA_FLAGS_LOCK_IRQ);762 763	table->nb.notifier_call = srq_event_notifier;764	mlx5_notifier_register(dev->mdev, &table->nb);765 766	return 0;767}768 769void mlx5_cleanup_srq_table(struct mlx5_ib_dev *dev)770{771	struct mlx5_srq_table *table = &dev->srq_table;772 773	mlx5_notifier_unregister(dev->mdev, &table->nb);774}775