[linux.git] / net / rds / ib_mr.h

/*
 * Copyright (c) 2016 Oracle.  All rights reserved.
 *
 * This software is available to you under a choice of one of two
 * licenses.  You may choose to be licensed under the terms of the GNU
 * General Public License (GPL) Version 2, available from the file
 * COPYING in the main directory of this source tree, or the
 * OpenIB.org BSD license below:
 *
 *     Redistribution and use in source and binary forms, with or
 *     without modification, are permitted provided that the following
 *     conditions are met:
 *
 *      - Redistributions of source code must retain the above
 *        copyright notice, this list of conditions and the following
 *        disclaimer.
 *
 *      - Redistributions in binary form must reproduce the above
 *        copyright notice, this list of conditions and the following
 *        disclaimer in the documentation and/or other materials
 *        provided with the distribution.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
 * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
 * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
 * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 */
#ifndef _RDS_IB_MR_H
#define _RDS_IB_MR_H

#include <linux/kernel.h>

#include "rds.h"
#include "ib.h"

#define RDS_MR_1M_POOL_SIZE		(8192 / 2)
#define RDS_MR_1M_MSG_SIZE		256
#define RDS_MR_8K_MSG_SIZE		2
#define RDS_MR_8K_SCALE			(256 / (RDS_MR_8K_MSG_SIZE + 1))
#define RDS_MR_8K_POOL_SIZE		(RDS_MR_8K_SCALE * (8192 / 2))

struct rds_ib_fmr {
	struct ib_fmr		*fmr;
};

enum rds_ib_fr_state {
	FRMR_IS_FREE,	/* mr invalidated & ready for use */
	FRMR_IS_INUSE,	/* mr is in use or used & can be invalidated */
	FRMR_IS_STALE,	/* Stale MR and needs to be dropped  */
};

struct rds_ib_frmr {
	struct ib_mr		*mr;
	enum rds_ib_fr_state	fr_state;
	bool			fr_inv;
	wait_queue_head_t	fr_inv_done;
	bool			fr_reg;
	wait_queue_head_t	fr_reg_done;
	struct ib_send_wr	fr_wr;
	unsigned int		dma_npages;
	unsigned int		sg_byte_len;
};

/* This is stored as mr->r_trans_private. */
struct rds_ib_mr {
	struct delayed_work		work;
	struct rds_ib_device		*device;
	struct rds_ib_mr_pool		*pool;
	struct rds_ib_connection	*ic;

	struct llist_node		llnode;

	/* unmap_list is for freeing */
	struct list_head		unmap_list;
	unsigned int			remap_count;

	struct scatterlist		*sg;
	unsigned int			sg_len;
	int				sg_dma_len;

	u8				odp:1;
	union {
		struct rds_ib_fmr	fmr;
		struct rds_ib_frmr	frmr;
		struct ib_mr		*mr;
	} u;
};

/* Our own little MR pool */
struct rds_ib_mr_pool {
	unsigned int            pool_type;
	struct mutex		flush_lock;	/* serialize fmr invalidate */
	struct delayed_work	flush_worker;	/* flush worker */

	atomic_t		item_count;	/* total # of MRs */
	atomic_t		dirty_count;	/* # dirty of MRs */

	struct llist_head	drop_list;	/* MRs not reached max_maps */
	struct llist_head	free_list;	/* unused MRs */
	struct llist_head	clean_list;	/* unused & unmapped MRs */
	wait_queue_head_t	flush_wait;
	spinlock_t		clean_lock;	/* "clean_list" concurrency */

	atomic_t		free_pinned;	/* memory pinned by free MRs */
	unsigned long		max_items;
	unsigned long		max_items_soft;
	unsigned long		max_free_pinned;
	struct ib_fmr_attr	fmr_attr;
	bool			use_fastreg;
};

extern struct workqueue_struct *rds_ib_mr_wq;
extern bool prefer_frmr;

struct rds_ib_mr_pool *rds_ib_create_mr_pool(struct rds_ib_device *rds_dev,
					     int npages);
void rds_ib_get_mr_info(struct rds_ib_device *rds_ibdev,
			struct rds_info_rdma_connection *iinfo);
void rds6_ib_get_mr_info(struct rds_ib_device *rds_ibdev,
			 struct rds6_info_rdma_connection *iinfo6);
void rds_ib_destroy_mr_pool(struct rds_ib_mr_pool *);
void *rds_ib_get_mr(struct scatterlist *sg, unsigned long nents,
		    struct rds_sock *rs, u32 *key_ret,
		    struct rds_connection *conn, u64 start, u64 length,
		    int need_odp);
void rds_ib_sync_mr(void *trans_private, int dir);
void rds_ib_free_mr(void *trans_private, int invalidate);
void rds_ib_flush_mrs(void);
int rds_ib_mr_init(void);
void rds_ib_mr_exit(void);
u32 rds_ib_get_lkey(void *trans_private);

void __rds_ib_teardown_mr(struct rds_ib_mr *);
void rds_ib_teardown_mr(struct rds_ib_mr *);
struct rds_ib_mr *rds_ib_alloc_fmr(struct rds_ib_device *, int);
struct rds_ib_mr *rds_ib_reuse_mr(struct rds_ib_mr_pool *);
int rds_ib_flush_mr_pool(struct rds_ib_mr_pool *, int, struct rds_ib_mr **);
struct rds_ib_mr *rds_ib_reg_fmr(struct rds_ib_device *, struct scatterlist *,
				 unsigned long, u32 *);
struct rds_ib_mr *rds_ib_try_reuse_ibmr(struct rds_ib_mr_pool *);
void rds_ib_unreg_fmr(struct list_head *, unsigned int *,
		      unsigned long *, unsigned int);
void rds_ib_free_fmr_list(struct rds_ib_mr *);
struct rds_ib_mr *rds_ib_reg_frmr(struct rds_ib_device *rds_ibdev,
				  struct rds_ib_connection *ic,
				  struct scatterlist *sg,
				  unsigned long nents, u32 *key);
void rds_ib_unreg_frmr(struct list_head *list, unsigned int *nfreed,
		       unsigned long *unpinned, unsigned int goal);
void rds_ib_free_frmr_list(struct rds_ib_mr *);
#endif
Commit	Line	Data
f6df683f	1	/*
	2	* Copyright (c) 2016 Oracle. All rights reserved.
	3	*
	4	* This software is available to you under a choice of one of two
	5	* licenses. You may choose to be licensed under the terms of the GNU
	6	* General Public License (GPL) Version 2, available from the file
	7	* COPYING in the main directory of this source tree, or the
	8	* OpenIB.org BSD license below:
	9	*
	10	* Redistribution and use in source and binary forms, with or
	11	* without modification, are permitted provided that the following
	12	* conditions are met:
	13	*
	14	* - Redistributions of source code must retain the above
	15	* copyright notice, this list of conditions and the following
	16	* disclaimer.
	17	*
	18	* - Redistributions in binary form must reproduce the above
	19	* copyright notice, this list of conditions and the following
	20	* disclaimer in the documentation and/or other materials
	21	* provided with the distribution.
	22	*
	23	* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
	24	* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
	25	* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
	26	* NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
	27	* BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
	28	* ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
	29	* CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
	30	* SOFTWARE.
	31	*/
	32	#ifndef _RDS_IB_MR_H
	33	#define _RDS_IB_MR_H
	34
	35	#include <linux/kernel.h>
	36
	37	#include "rds.h"
	38	#include "ib.h"
	39
	40	#define RDS_MR_1M_POOL_SIZE (8192 / 2)
	41	#define RDS_MR_1M_MSG_SIZE 256
	42	#define RDS_MR_8K_MSG_SIZE 2
	43	#define RDS_MR_8K_SCALE (256 / (RDS_MR_8K_MSG_SIZE + 1))
	44	#define RDS_MR_8K_POOL_SIZE (RDS_MR_8K_SCALE * (8192 / 2))
	45
a69365a3	46	struct rds_ib_fmr {
a69365a3	47	struct ib_fmr *fmr;
a69365a3	48	};
a69365a3	49
1659185f AR	50	enum rds_ib_fr_state {
	51	FRMR_IS_FREE, /* mr invalidated & ready for use */
	52	FRMR_IS_INUSE, /* mr is in use or used & can be invalidated */
	53	FRMR_IS_STALE, /* Stale MR and needs to be dropped */
	54	};
	55
	56	struct rds_ib_frmr {
	57	struct ib_mr *mr;
	58	enum rds_ib_fr_state fr_state;
	59	bool fr_inv;
2c7da8e6	60	wait_queue_head_t fr_inv_done;
5f33141d GR	61	bool fr_reg;
5f33141d GR	62	wait_queue_head_t fr_reg_done;
1659185f AR	63	struct ib_send_wr fr_wr;
	64	unsigned int dma_npages;
	65	unsigned int sg_byte_len;
	66	};
	67
f6df683f	68	/* This is stored as mr->r_trans_private. */
f6df683f	69	struct rds_ib_mr {
2eafa174	70	struct delayed_work work;
d4de76da	71	struct rds_ib_device *device;
	72	struct rds_ib_mr_pool *pool;
	73	struct rds_ib_connection *ic;
f6df683f	74
d4de76da	75	struct llist_node llnode;
f6df683f	76
f6df683f	77	/* unmap_list is for freeing */
d4de76da	78	struct list_head unmap_list;
d4de76da	79	unsigned int remap_count;
f6df683f	80
d4de76da	81	struct scatterlist *sg;
	82	unsigned int sg_len;
	83	int sg_dma_len;
a69365a3	84
2eafa174	85	u8 odp:1;
a69365a3	86	union {
a69365a3	87	struct rds_ib_fmr fmr;
1659185f	88	struct rds_ib_frmr frmr;
2eafa174	89	struct ib_mr *mr;
a69365a3	90	} u;
f6df683f	91	};
	92
	93	/* Our own little MR pool */
	94	struct rds_ib_mr_pool {
	95	unsigned int pool_type;
	96	struct mutex flush_lock; /* serialize fmr invalidate */
	97	struct delayed_work flush_worker; /* flush worker */
	98
	99	atomic_t item_count; /* total # of MRs */
	100	atomic_t dirty_count; /* # dirty of MRs */
	101
	102	struct llist_head drop_list; /* MRs not reached max_maps */
	103	struct llist_head free_list; /* unused MRs */
	104	struct llist_head clean_list; /* unused & unmapped MRs */
	105	wait_queue_head_t flush_wait;
c9467447	106	spinlock_t clean_lock; /* "clean_list" concurrency */
f6df683f	107
	108	atomic_t free_pinned; /* memory pinned by free MRs */
	109	unsigned long max_items;
	110	unsigned long max_items_soft;
	111	unsigned long max_free_pinned;
	112	struct ib_fmr_attr fmr_attr;
1659185f	113	bool use_fastreg;
f6df683f	114	};
	115
	116	extern struct workqueue_struct *rds_ib_mr_wq;
2cb2912d	117	extern bool prefer_frmr;
f6df683f	118
	119	struct rds_ib_mr_pool rds_ib_create_mr_pool(struct rds_ib_device rds_dev,
	120	int npages);
	121	void rds_ib_get_mr_info(struct rds_ib_device *rds_ibdev,
	122	struct rds_info_rdma_connection *iinfo);
b7ff8b10 KCP	123	void rds6_ib_get_mr_info(struct rds_ib_device *rds_ibdev,
b7ff8b10 KCP	124	struct rds6_info_rdma_connection *iinfo6);
f6df683f	125	void rds_ib_destroy_mr_pool(struct rds_ib_mr_pool *);
f6df683f	126	void rds_ib_get_mr(struct scatterlist sg, unsigned long nents,
9e630bcb	127	struct rds_sock rs, u32 key_ret,
2eafa174 HWR	128	struct rds_connection *conn, u64 start, u64 length,
2eafa174 HWR	129	int need_odp);
f6df683f	130	void rds_ib_sync_mr(void *trans_private, int dir);
	131	void rds_ib_free_mr(void *trans_private, int invalidate);
	132	void rds_ib_flush_mrs(void);
	133	int rds_ib_mr_init(void);
	134	void rds_ib_mr_exit(void);
2eafa174	135	u32 rds_ib_get_lkey(void *trans_private);
f6df683f	136
	137	void __rds_ib_teardown_mr(struct rds_ib_mr *);
	138	void rds_ib_teardown_mr(struct rds_ib_mr *);
	139	struct rds_ib_mr rds_ib_alloc_fmr(struct rds_ib_device , int);
f6df683f	140	struct rds_ib_mr rds_ib_reuse_mr(struct rds_ib_mr_pool );
f6df683f	141	int rds_ib_flush_mr_pool(struct rds_ib_mr_pool , int, struct rds_ib_mr *);
490ea596	142	struct rds_ib_mr rds_ib_reg_fmr(struct rds_ib_device , struct scatterlist *,
	143	unsigned long, u32 *);
	144	struct rds_ib_mr rds_ib_try_reuse_ibmr(struct rds_ib_mr_pool );
	145	void rds_ib_unreg_fmr(struct list_head , unsigned int ,
	146	unsigned long *, unsigned int);
	147	void rds_ib_free_fmr_list(struct rds_ib_mr *);
1659185f AR	148	struct rds_ib_mr rds_ib_reg_frmr(struct rds_ib_device rds_ibdev,
	149	struct rds_ib_connection *ic,
	150	struct scatterlist *sg,
	151	unsigned long nents, u32 *key);
	152	void rds_ib_unreg_frmr(struct list_head list, unsigned int nfreed,
	153	unsigned long *unpinned, unsigned int goal);
	154	void rds_ib_free_frmr_list(struct rds_ib_mr *);
f6df683f	155	#endif