[linux.git] / net / smc / smc_wr.h

/* SPDX-License-Identifier: GPL-2.0 */
/*
 * Shared Memory Communications over RDMA (SMC-R) and RoCE
 *
 * Work Requests exploiting Infiniband API
 *
 * Copyright IBM Corp. 2016
 *
 * Author(s):  Steffen Maier <[email protected]>
 */

#ifndef SMC_WR_H
#define SMC_WR_H

#include <linux/atomic.h>
#include <rdma/ib_verbs.h>
#include <asm/div64.h>

#include "smc.h"
#include "smc_core.h"

#define SMC_WR_BUF_CNT 16	/* # of ctrl buffers per link */

#define SMC_WR_TX_WAIT_FREE_SLOT_TIME	(10 * HZ)

#define SMC_WR_TX_SIZE 44 /* actual size of wr_send data (<=SMC_WR_BUF_SIZE) */

#define SMC_WR_TX_PEND_PRIV_SIZE 32

struct smc_wr_tx_pend_priv {
	u8			priv[SMC_WR_TX_PEND_PRIV_SIZE];
};

typedef void (*smc_wr_tx_handler)(struct smc_wr_tx_pend_priv *,
				  struct smc_link *,
				  enum ib_wc_status);

typedef bool (*smc_wr_tx_filter)(struct smc_wr_tx_pend_priv *,
				 unsigned long);

typedef void (*smc_wr_tx_dismisser)(struct smc_wr_tx_pend_priv *);

struct smc_wr_rx_handler {
	struct hlist_node	list;	/* hash table collision resolution */
	void			(*handler)(struct ib_wc *, void *);
	u8			type;
};

/* Only used by RDMA write WRs.
 * All other WRs (CDC/LLC) use smc_wr_tx_send handling WR_ID implicitly
 */
static inline long smc_wr_tx_get_next_wr_id(struct smc_link *link)
{
	return atomic_long_inc_return(&link->wr_tx_id);
}

static inline void smc_wr_tx_set_wr_id(atomic_long_t *wr_tx_id, long val)
{
	atomic_long_set(wr_tx_id, val);
}

static inline bool smc_wr_tx_link_hold(struct smc_link *link)
{
	if (!smc_link_sendable(link))
		return false;
	percpu_ref_get(&link->wr_tx_refs);
	return true;
}

static inline void smc_wr_tx_link_put(struct smc_link *link)
{
	percpu_ref_put(&link->wr_tx_refs);
}

static inline void smc_wr_drain_cq(struct smc_link *lnk)
{
	wait_event(lnk->wr_rx_empty_wait, lnk->wr_rx_id_compl == lnk->wr_rx_id);
}

static inline void smc_wr_wakeup_tx_wait(struct smc_link *lnk)
{
	wake_up_all(&lnk->wr_tx_wait);
}

static inline void smc_wr_wakeup_reg_wait(struct smc_link *lnk)
{
	wake_up(&lnk->wr_reg_wait);
}

/* post a new receive work request to fill a completed old work request entry */
static inline int smc_wr_rx_post(struct smc_link *link)
{
	int rc;
	u64 wr_id, temp_wr_id;
	u32 index;

	wr_id = ++link->wr_rx_id; /* tasklet context, thus not atomic */
	temp_wr_id = wr_id;
	index = do_div(temp_wr_id, link->wr_rx_cnt);
	link->wr_rx_ibs[index].wr_id = wr_id;
	rc = ib_post_recv(link->roce_qp, &link->wr_rx_ibs[index], NULL);
	return rc;
}

int smc_wr_create_link(struct smc_link *lnk);
int smc_wr_alloc_link_mem(struct smc_link *lnk);
int smc_wr_alloc_lgr_mem(struct smc_link_group *lgr);
void smc_wr_free_link(struct smc_link *lnk);
void smc_wr_free_link_mem(struct smc_link *lnk);
void smc_wr_free_lgr_mem(struct smc_link_group *lgr);
void smc_wr_remember_qp_attr(struct smc_link *lnk);
void smc_wr_remove_dev(struct smc_ib_device *smcibdev);
void smc_wr_add_dev(struct smc_ib_device *smcibdev);

int smc_wr_tx_get_free_slot(struct smc_link *link, smc_wr_tx_handler handler,
			    struct smc_wr_buf **wr_buf,
			    struct smc_rdma_wr **wrs,
			    struct smc_wr_tx_pend_priv **wr_pend_priv);
int smc_wr_tx_get_v2_slot(struct smc_link *link,
			  smc_wr_tx_handler handler,
			  struct smc_wr_v2_buf **wr_buf,
			  struct smc_wr_tx_pend_priv **wr_pend_priv);
int smc_wr_tx_put_slot(struct smc_link *link,
		       struct smc_wr_tx_pend_priv *wr_pend_priv);
int smc_wr_tx_send(struct smc_link *link,
		   struct smc_wr_tx_pend_priv *wr_pend_priv);
int smc_wr_tx_v2_send(struct smc_link *link,
		      struct smc_wr_tx_pend_priv *priv, int len);
int smc_wr_tx_send_wait(struct smc_link *link, struct smc_wr_tx_pend_priv *priv,
			unsigned long timeout);
void smc_wr_tx_cq_handler(struct ib_cq *ib_cq, void *cq_context);
void smc_wr_tx_wait_no_pending_sends(struct smc_link *link);

int smc_wr_rx_register_handler(struct smc_wr_rx_handler *handler);
int smc_wr_rx_post_init(struct smc_link *link);
void smc_wr_rx_cq_handler(struct ib_cq *ib_cq, void *cq_context);
int smc_wr_reg_send(struct smc_link *link, struct ib_mr *mr);

#endif /* SMC_WR_H */
Commit	Line	Data
b2441318	1	/* SPDX-License-Identifier: GPL-2.0 */
f38ba179 UB	2	/*
	3	* Shared Memory Communications over RDMA (SMC-R) and RoCE
	4	*
	5	* Work Requests exploiting Infiniband API
	6	*
	7	* Copyright IBM Corp. 2016
	8	*
	9	* Author(s): Steffen Maier <[email protected]>
	10	*/
	11
	12	#ifndef SMC_WR_H
	13	#define SMC_WR_H
	14
	15	#include <linux/atomic.h>
	16	#include <rdma/ib_verbs.h>
	17	#include <asm/div64.h>
	18
	19	#include "smc.h"
	20	#include "smc_core.h"
	21
f38ba179 UB	22	#define SMC_WR_BUF_CNT 16 /* # of ctrl buffers per link */
	23
	24	#define SMC_WR_TX_WAIT_FREE_SLOT_TIME (10 * HZ)
f38ba179 UB	25
	26	#define SMC_WR_TX_SIZE 44 /* actual size of wr_send data (<=SMC_WR_BUF_SIZE) */
	27
	28	#define SMC_WR_TX_PEND_PRIV_SIZE 32
	29
	30	struct smc_wr_tx_pend_priv {
	31	u8 priv[SMC_WR_TX_PEND_PRIV_SIZE];
	32	};
	33
	34	typedef void (smc_wr_tx_handler)(struct smc_wr_tx_pend_priv ,
	35	struct smc_link *,
	36	enum ib_wc_status);
	37
5f08318f UB	38	typedef bool (smc_wr_tx_filter)(struct smc_wr_tx_pend_priv ,
	39	unsigned long);
	40
	41	typedef void (smc_wr_tx_dismisser)(struct smc_wr_tx_pend_priv );
	42
f38ba179 UB	43	struct smc_wr_rx_handler {
	44	struct hlist_node list; /* hash table collision resolution */
	45	void (handler)(struct ib_wc , void *);
	46	u8 type;
	47	};
	48
	49	/* Only used by RDMA write WRs.
	50	* All other WRs (CDC/LLC) use smc_wr_tx_send handling WR_ID implicitly
	51	*/
	52	static inline long smc_wr_tx_get_next_wr_id(struct smc_link *link)
	53	{
	54	return atomic_long_inc_return(&link->wr_tx_id);
	55	}
	56
	57	static inline void smc_wr_tx_set_wr_id(atomic_long_t *wr_tx_id, long val)
	58	{
	59	atomic_long_set(wr_tx_id, val);
95f7f3e7 KG	60	}
	61
	62	static inline bool smc_wr_tx_link_hold(struct smc_link *link)
	63	{
90cee52f	64	if (!smc_link_sendable(link))
95f7f3e7	65	return false;
79a22238	66	percpu_ref_get(&link->wr_tx_refs);
95f7f3e7 KG	67	return true;
	68	}
	69
	70	static inline void smc_wr_tx_link_put(struct smc_link *link)
	71	{
79a22238	72	percpu_ref_put(&link->wr_tx_refs);
f38ba179 UB	73	}
f38ba179 UB	74
e9b1a4f8 YL	75	static inline void smc_wr_drain_cq(struct smc_link *lnk)
	76	{
	77	wait_event(lnk->wr_rx_empty_wait, lnk->wr_rx_id_compl == lnk->wr_rx_id);
	78	}
	79
15e1b99a UB	80	static inline void smc_wr_wakeup_tx_wait(struct smc_link *lnk)
	81	{
	82	wake_up_all(&lnk->wr_tx_wait);
	83	}
	84
	85	static inline void smc_wr_wakeup_reg_wait(struct smc_link *lnk)
	86	{
	87	wake_up(&lnk->wr_reg_wait);
	88	}
	89
f38ba179 UB	90	/* post a new receive work request to fill a completed old work request entry */
	91	static inline int smc_wr_rx_post(struct smc_link *link)
	92	{
f38ba179 UB	93	int rc;
	94	u64 wr_id, temp_wr_id;
	95	u32 index;
	96
	97	wr_id = ++link->wr_rx_id; /* tasklet context, thus not atomic */
	98	temp_wr_id = wr_id;
	99	index = do_div(temp_wr_id, link->wr_rx_cnt);
	100	link->wr_rx_ibs[index].wr_id = wr_id;
2e3bbe46	101	rc = ib_post_recv(link->roce_qp, &link->wr_rx_ibs[index], NULL);
f38ba179 UB	102	return rc;
	103	}
	104
	105	int smc_wr_create_link(struct smc_link *lnk);
	106	int smc_wr_alloc_link_mem(struct smc_link *lnk);
8799e310	107	int smc_wr_alloc_lgr_mem(struct smc_link_group *lgr);
f38ba179 UB	108	void smc_wr_free_link(struct smc_link *lnk);
f38ba179 UB	109	void smc_wr_free_link_mem(struct smc_link *lnk);
8799e310	110	void smc_wr_free_lgr_mem(struct smc_link_group *lgr);
f38ba179 UB	111	void smc_wr_remember_qp_attr(struct smc_link *lnk);
	112	void smc_wr_remove_dev(struct smc_ib_device *smcibdev);
	113	void smc_wr_add_dev(struct smc_ib_device *smcibdev);
	114
	115	int smc_wr_tx_get_free_slot(struct smc_link *link, smc_wr_tx_handler handler,
	116	struct smc_wr_buf **wr_buf,
ad6f317f	117	struct smc_rdma_wr **wrs,
f38ba179	118	struct smc_wr_tx_pend_priv **wr_pend_priv);
b4ba4652 KG	119	int smc_wr_tx_get_v2_slot(struct smc_link *link,
	120	smc_wr_tx_handler handler,
	121	struct smc_wr_v2_buf **wr_buf,
	122	struct smc_wr_tx_pend_priv **wr_pend_priv);
f38ba179 UB	123	int smc_wr_tx_put_slot(struct smc_link *link,
	124	struct smc_wr_tx_pend_priv *wr_pend_priv);
	125	int smc_wr_tx_send(struct smc_link *link,
	126	struct smc_wr_tx_pend_priv *wr_pend_priv);
b4ba4652 KG	127	int smc_wr_tx_v2_send(struct smc_link *link,
b4ba4652 KG	128	struct smc_wr_tx_pend_priv *priv, int len);
09c61d24 KG	129	int smc_wr_tx_send_wait(struct smc_link link, struct smc_wr_tx_pend_priv priv,
09c61d24 KG	130	unsigned long timeout);
f38ba179	131	void smc_wr_tx_cq_handler(struct ib_cq ib_cq, void cq_context);
349d4312	132	void smc_wr_tx_wait_no_pending_sends(struct smc_link *link);
f38ba179 UB	133
	134	int smc_wr_rx_register_handler(struct smc_wr_rx_handler *handler);
	135	int smc_wr_rx_post_init(struct smc_link *link);
	136	void smc_wr_rx_cq_handler(struct ib_cq ib_cq, void cq_context);
652a1e41	137	int smc_wr_reg_send(struct smc_link link, struct ib_mr mr);
f38ba179 UB	138
f38ba179 UB	139	#endif /* SMC_WR_H */