[mirror_ubuntu-artful-kernel.git] / drivers / infiniband / core / mr_pool.c

/*
 * Copyright (c) 2016 HGST, a Western Digital Company.
 *
 * This program is free software; you can redistribute it and/or modify it
 * under the terms and conditions of the GNU General Public License,
 * version 2, as published by the Free Software Foundation.
 *
 * This program is distributed in the hope it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
 * more details.
 */
#include <rdma/ib_verbs.h>
#include <rdma/mr_pool.h>

struct ib_mr *ib_mr_pool_get(struct ib_qp *qp, struct list_head *list)
{
	struct ib_mr *mr;
	unsigned long flags;

	spin_lock_irqsave(&qp->mr_lock, flags);
	mr = list_first_entry_or_null(list, struct ib_mr, qp_entry);
	if (mr) {
		list_del(&mr->qp_entry);
		qp->mrs_used++;
	}
	spin_unlock_irqrestore(&qp->mr_lock, flags);

	return mr;
}
EXPORT_SYMBOL(ib_mr_pool_get);

void ib_mr_pool_put(struct ib_qp *qp, struct list_head *list, struct ib_mr *mr)
{
	unsigned long flags;

	spin_lock_irqsave(&qp->mr_lock, flags);
	list_add(&mr->qp_entry, list);
	qp->mrs_used--;
	spin_unlock_irqrestore(&qp->mr_lock, flags);
}
EXPORT_SYMBOL(ib_mr_pool_put);

int ib_mr_pool_init(struct ib_qp *qp, struct list_head *list, int nr,
		enum ib_mr_type type, u32 max_num_sg)
{
	struct ib_mr *mr;
	unsigned long flags;
	int ret, i;

	for (i = 0; i < nr; i++) {
		mr = ib_alloc_mr(qp->pd, type, max_num_sg);
		if (IS_ERR(mr)) {
			ret = PTR_ERR(mr);
			goto out;
		}

		spin_lock_irqsave(&qp->mr_lock, flags);
		list_add_tail(&mr->qp_entry, list);
		spin_unlock_irqrestore(&qp->mr_lock, flags);
	}

	return 0;
out:
	ib_mr_pool_destroy(qp, list);
	return ret;
}
EXPORT_SYMBOL(ib_mr_pool_init);

void ib_mr_pool_destroy(struct ib_qp *qp, struct list_head *list)
{
	struct ib_mr *mr;
	unsigned long flags;

	spin_lock_irqsave(&qp->mr_lock, flags);
	while (!list_empty(list)) {
		mr = list_first_entry(list, struct ib_mr, qp_entry);
		list_del(&mr->qp_entry);

		spin_unlock_irqrestore(&qp->mr_lock, flags);
		ib_dereg_mr(mr);
		spin_lock_irqsave(&qp->mr_lock, flags);
	}
	spin_unlock_irqrestore(&qp->mr_lock, flags);
}
EXPORT_SYMBOL(ib_mr_pool_destroy);
Commit	Line	Data
fffb0383 CH	1	/*
	2	* Copyright (c) 2016 HGST, a Western Digital Company.
	3	*
	4	* This program is free software; you can redistribute it and/or modify it
	5	* under the terms and conditions of the GNU General Public License,
	6	* version 2, as published by the Free Software Foundation.
	7	*
	8	* This program is distributed in the hope it will be useful, but WITHOUT
	9	* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
	10	* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
	11	* more details.
	12	*/
	13	#include <rdma/ib_verbs.h>
	14	#include <rdma/mr_pool.h>
	15
	16	struct ib_mr ib_mr_pool_get(struct ib_qp qp, struct list_head *list)
	17	{
	18	struct ib_mr *mr;
	19	unsigned long flags;
	20
	21	spin_lock_irqsave(&qp->mr_lock, flags);
	22	mr = list_first_entry_or_null(list, struct ib_mr, qp_entry);
	23	if (mr) {
	24	list_del(&mr->qp_entry);
	25	qp->mrs_used++;
	26	}
	27	spin_unlock_irqrestore(&qp->mr_lock, flags);
	28
	29	return mr;
	30	}
	31	EXPORT_SYMBOL(ib_mr_pool_get);
	32
	33	void ib_mr_pool_put(struct ib_qp qp, struct list_head list, struct ib_mr *mr)
	34	{
	35	unsigned long flags;
	36
	37	spin_lock_irqsave(&qp->mr_lock, flags);
	38	list_add(&mr->qp_entry, list);
	39	qp->mrs_used--;
	40	spin_unlock_irqrestore(&qp->mr_lock, flags);
	41	}
	42	EXPORT_SYMBOL(ib_mr_pool_put);
	43
	44	int ib_mr_pool_init(struct ib_qp qp, struct list_head list, int nr,
	45	enum ib_mr_type type, u32 max_num_sg)
	46	{
	47	struct ib_mr *mr;
	48	unsigned long flags;
	49	int ret, i;
	50
	51	for (i = 0; i < nr; i++) {
	52	mr = ib_alloc_mr(qp->pd, type, max_num_sg);
	53	if (IS_ERR(mr)) {
	54	ret = PTR_ERR(mr);
	55	goto out;
	56	}
	57
	58	spin_lock_irqsave(&qp->mr_lock, flags);
	59	list_add_tail(&mr->qp_entry, list);
	60	spin_unlock_irqrestore(&qp->mr_lock, flags);
	61	}
	62
	63	return 0;
	64	out:
65	ib_mr_pool_destroy(qp, list);
66	return ret;
67	}
68	EXPORT_SYMBOL(ib_mr_pool_init);
69
70	void ib_mr_pool_destroy(struct ib_qp qp, struct list_head list)
71	{
72	struct ib_mr *mr;
73	unsigned long flags;
74
75	spin_lock_irqsave(&qp->mr_lock, flags);
76	while (!list_empty(list)) {
77	mr = list_first_entry(list, struct ib_mr, qp_entry);
78	list_del(&mr->qp_entry);
79
80	spin_unlock_irqrestore(&qp->mr_lock, flags);
81	ib_dereg_mr(mr);
82	spin_lock_irqsave(&qp->mr_lock, flags);
83	}
84	spin_unlock_irqrestore(&qp->mr_lock, flags);
85	}
86	EXPORT_SYMBOL(ib_mr_pool_destroy);