1 // SPDX-License-Identifier: GPL-2.0-only
2 /*
3  * Copyright (c) 2016 HGST, a Western Digital Company.
4  */
5 #include <rdma/ib_verbs.h>
6 #include <rdma/mr_pool.h>
7 
8 struct ib_mr *ib_mr_pool_get(struct ib_qp *qp, struct list_head *list)
9 {
10 	struct ib_mr *mr;
11 	unsigned long flags;
12 
13 	spin_lock_irqsave(&qp->mr_lock, flags);
14 	mr = list_first_entry_or_null(list, struct ib_mr, qp_entry);
15 	if (mr) {
16 		list_del(&mr->qp_entry);
17 		qp->mrs_used++;
18 	}
19 	spin_unlock_irqrestore(&qp->mr_lock, flags);
20 
21 	return mr;
22 }
23 EXPORT_SYMBOL(ib_mr_pool_get);
24 
25 void ib_mr_pool_put(struct ib_qp *qp, struct list_head *list, struct ib_mr *mr)
26 {
27 	unsigned long flags;
28 
29 	spin_lock_irqsave(&qp->mr_lock, flags);
30 	list_add(&mr->qp_entry, list);
31 	qp->mrs_used--;
32 	spin_unlock_irqrestore(&qp->mr_lock, flags);
33 }
34 EXPORT_SYMBOL(ib_mr_pool_put);
35 
36 int ib_mr_pool_init(struct ib_qp *qp, struct list_head *list, int nr,
37 		enum ib_mr_type type, u32 max_num_sg, u32 max_num_meta_sg)
38 {
39 	struct ib_mr *mr;
40 	unsigned long flags;
41 	int ret, i;
42 
43 	for (i = 0; i < nr; i++) {
44 		if (type == IB_MR_TYPE_INTEGRITY)
45 			mr = ib_alloc_mr_integrity(qp->pd, max_num_sg,
46 						   max_num_meta_sg);
47 		else
48 			mr = ib_alloc_mr(qp->pd, type, max_num_sg);
49 		if (IS_ERR(mr)) {
50 			ret = PTR_ERR(mr);
51 			goto out;
52 		}
53 
54 		spin_lock_irqsave(&qp->mr_lock, flags);
55 		list_add_tail(&mr->qp_entry, list);
56 		spin_unlock_irqrestore(&qp->mr_lock, flags);
57 	}
58 
59 	return 0;
60 out:
61 	ib_mr_pool_destroy(qp, list);
62 	return ret;
63 }
64 EXPORT_SYMBOL(ib_mr_pool_init);
65 
66 void ib_mr_pool_destroy(struct ib_qp *qp, struct list_head *list)
67 {
68 	struct ib_mr *mr;
69 	unsigned long flags;
70 
71 	spin_lock_irqsave(&qp->mr_lock, flags);
72 	while (!list_empty(list)) {
73 		mr = list_first_entry(list, struct ib_mr, qp_entry);
74 		list_del(&mr->qp_entry);
75 
76 		spin_unlock_irqrestore(&qp->mr_lock, flags);
77 		ib_dereg_mr(mr);
78 		spin_lock_irqsave(&qp->mr_lock, flags);
79 	}
80 	spin_unlock_irqrestore(&qp->mr_lock, flags);
81 }
82 EXPORT_SYMBOL(ib_mr_pool_destroy);
83