1 /*
2  * Copyright (c) 2013-2015, Mellanox Technologies. 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 
33 #include <linux/kernel.h>
34 #include <linux/module.h>
35 #include <linux/mlx5/driver.h>
36 #include <linux/mlx5/cmd.h>
37 #include "mlx5_core.h"
38 
mlx5_init_mkey_table(struct mlx5_core_dev * dev)39 void mlx5_init_mkey_table(struct mlx5_core_dev *dev)
40 {
41 	struct mlx5_mkey_table *table = &dev->priv.mkey_table;
42 
43 	memset(table, 0, sizeof(*table));
44 	rwlock_init(&table->lock);
45 	INIT_RADIX_TREE(&table->tree, GFP_ATOMIC);
46 }
47 
mlx5_cleanup_mkey_table(struct mlx5_core_dev * dev)48 void mlx5_cleanup_mkey_table(struct mlx5_core_dev *dev)
49 {
50 }
51 
mlx5_core_create_mkey_cb(struct mlx5_core_dev * dev,struct mlx5_core_mkey * mkey,u32 * in,int inlen,u32 * out,int outlen,mlx5_cmd_cbk_t callback,void * context)52 int mlx5_core_create_mkey_cb(struct mlx5_core_dev *dev,
53 			     struct mlx5_core_mkey *mkey,
54 			     u32 *in, int inlen,
55 			     u32 *out, int outlen,
56 			     mlx5_cmd_cbk_t callback, void *context)
57 {
58 	struct mlx5_mkey_table *table = &dev->priv.mkey_table;
59 	u32 lout[MLX5_ST_SZ_DW(create_mkey_out)] = {0};
60 	u32 mkey_index;
61 	void *mkc;
62 	int err;
63 	u8 key;
64 
65 	spin_lock_irq(&dev->priv.mkey_lock);
66 	key = dev->priv.mkey_key++;
67 	spin_unlock_irq(&dev->priv.mkey_lock);
68 	mkc = MLX5_ADDR_OF(create_mkey_in, in, memory_key_mkey_entry);
69 
70 	MLX5_SET(create_mkey_in, in, opcode, MLX5_CMD_OP_CREATE_MKEY);
71 	MLX5_SET(mkc, mkc, mkey_7_0, key);
72 
73 	if (callback)
74 		return mlx5_cmd_exec_cb(dev, in, inlen, out, outlen,
75 					callback, context);
76 
77 	err = mlx5_cmd_exec(dev, in, inlen, lout, sizeof(lout));
78 	if (err)
79 		return err;
80 
81 	mkey_index = MLX5_GET(create_mkey_out, lout, mkey_index);
82 	mkey->iova = MLX5_GET64(mkc, mkc, start_addr);
83 	mkey->size = MLX5_GET64(mkc, mkc, len);
84 	mkey->key = mlx5_idx_to_mkey(mkey_index) | key;
85 	mkey->pd = MLX5_GET(mkc, mkc, pd);
86 
87 	mlx5_core_dbg(dev, "out 0x%x, key 0x%x, mkey 0x%x\n",
88 		      mkey_index, key, mkey->key);
89 
90 	/* connect to mkey tree */
91 	write_lock_irq(&table->lock);
92 	err = radix_tree_insert(&table->tree, mlx5_base_mkey(mkey->key), mkey);
93 	write_unlock_irq(&table->lock);
94 	if (err) {
95 		mlx5_core_warn(dev, "failed radix tree insert of mkey 0x%x, %d\n",
96 			       mlx5_base_mkey(mkey->key), err);
97 		mlx5_core_destroy_mkey(dev, mkey);
98 	}
99 
100 	return err;
101 }
102 EXPORT_SYMBOL(mlx5_core_create_mkey_cb);
103 
mlx5_core_create_mkey(struct mlx5_core_dev * dev,struct mlx5_core_mkey * mkey,u32 * in,int inlen)104 int mlx5_core_create_mkey(struct mlx5_core_dev *dev,
105 			  struct mlx5_core_mkey *mkey,
106 			  u32 *in, int inlen)
107 {
108 	return mlx5_core_create_mkey_cb(dev, mkey, in, inlen,
109 					NULL, 0, NULL, NULL);
110 }
111 EXPORT_SYMBOL(mlx5_core_create_mkey);
112 
mlx5_core_destroy_mkey(struct mlx5_core_dev * dev,struct mlx5_core_mkey * mkey)113 int mlx5_core_destroy_mkey(struct mlx5_core_dev *dev,
114 			   struct mlx5_core_mkey *mkey)
115 {
116 	struct mlx5_mkey_table *table = &dev->priv.mkey_table;
117 	u32 out[MLX5_ST_SZ_DW(destroy_mkey_out)] = {0};
118 	u32 in[MLX5_ST_SZ_DW(destroy_mkey_in)]   = {0};
119 	struct mlx5_core_mkey *deleted_mkey;
120 	unsigned long flags;
121 
122 	write_lock_irqsave(&table->lock, flags);
123 	deleted_mkey = radix_tree_delete(&table->tree, mlx5_base_mkey(mkey->key));
124 	write_unlock_irqrestore(&table->lock, flags);
125 	if (!deleted_mkey) {
126 		mlx5_core_dbg(dev, "failed radix tree delete of mkey 0x%x\n",
127 			      mlx5_base_mkey(mkey->key));
128 		return -ENOENT;
129 	}
130 
131 	MLX5_SET(destroy_mkey_in, in, opcode, MLX5_CMD_OP_DESTROY_MKEY);
132 	MLX5_SET(destroy_mkey_in, in, mkey_index, mlx5_mkey_to_idx(mkey->key));
133 	return mlx5_cmd_exec(dev, in, sizeof(in), out, sizeof(out));
134 }
135 EXPORT_SYMBOL(mlx5_core_destroy_mkey);
136 
mlx5_core_query_mkey(struct mlx5_core_dev * dev,struct mlx5_core_mkey * mkey,u32 * out,int outlen)137 int mlx5_core_query_mkey(struct mlx5_core_dev *dev, struct mlx5_core_mkey *mkey,
138 			 u32 *out, int outlen)
139 {
140 	u32 in[MLX5_ST_SZ_DW(query_mkey_in)] = {0};
141 
142 	memset(out, 0, outlen);
143 	MLX5_SET(query_mkey_in, in, opcode, MLX5_CMD_OP_QUERY_MKEY);
144 	MLX5_SET(query_mkey_in, in, mkey_index, mlx5_mkey_to_idx(mkey->key));
145 	return mlx5_cmd_exec(dev, in, sizeof(in), out, outlen);
146 }
147 EXPORT_SYMBOL(mlx5_core_query_mkey);
148 
mlx5_get_psv(u32 * out,int psv_index)149 static inline u32 mlx5_get_psv(u32 *out, int psv_index)
150 {
151 	switch (psv_index) {
152 	case 1: return MLX5_GET(create_psv_out, out, psv1_index);
153 	case 2: return MLX5_GET(create_psv_out, out, psv2_index);
154 	case 3: return MLX5_GET(create_psv_out, out, psv3_index);
155 	default: return MLX5_GET(create_psv_out, out, psv0_index);
156 	}
157 }
158 
mlx5_core_create_psv(struct mlx5_core_dev * dev,u32 pdn,int npsvs,u32 * sig_index)159 int mlx5_core_create_psv(struct mlx5_core_dev *dev, u32 pdn,
160 			 int npsvs, u32 *sig_index)
161 {
162 	u32 out[MLX5_ST_SZ_DW(create_psv_out)] = {0};
163 	u32 in[MLX5_ST_SZ_DW(create_psv_in)]   = {0};
164 	int i, err;
165 
166 	if (npsvs > MLX5_MAX_PSVS)
167 		return -EINVAL;
168 
169 	MLX5_SET(create_psv_in, in, opcode, MLX5_CMD_OP_CREATE_PSV);
170 	MLX5_SET(create_psv_in, in, pd, pdn);
171 	MLX5_SET(create_psv_in, in, num_psv, npsvs);
172 
173 	err = mlx5_cmd_exec(dev, in, sizeof(in), out, sizeof(out));
174 	if (err)
175 		return err;
176 
177 	for (i = 0; i < npsvs; i++)
178 		sig_index[i] = mlx5_get_psv(out, i);
179 
180 	return err;
181 }
182 EXPORT_SYMBOL(mlx5_core_create_psv);
183 
mlx5_core_destroy_psv(struct mlx5_core_dev * dev,int psv_num)184 int mlx5_core_destroy_psv(struct mlx5_core_dev *dev, int psv_num)
185 {
186 	u32 out[MLX5_ST_SZ_DW(destroy_psv_out)] = {0};
187 	u32 in[MLX5_ST_SZ_DW(destroy_psv_in)]   = {0};
188 
189 	MLX5_SET(destroy_psv_in, in, opcode, MLX5_CMD_OP_DESTROY_PSV);
190 	MLX5_SET(destroy_psv_in, in, psvn, psv_num);
191 	return mlx5_cmd_exec(dev, in, sizeof(in), out, sizeof(out));
192 }
193 EXPORT_SYMBOL(mlx5_core_destroy_psv);
194