Commit | Line | Data |
---|---|---|
e126ba97 | 1 | /* |
6cf0a15f | 2 | * Copyright (c) 2013-2015, Mellanox Technologies. All rights reserved. |
e126ba97 EC |
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 | ||
34 | #include <linux/kref.h> | |
35 | #include <linux/random.h> | |
36 | #include <linux/debugfs.h> | |
37 | #include <linux/export.h> | |
746b5583 | 38 | #include <linux/delay.h> |
e126ba97 | 39 | #include <rdma/ib_umem.h> |
b4cfe447 | 40 | #include <rdma/ib_umem_odp.h> |
968e78dd | 41 | #include <rdma/ib_verbs.h> |
e126ba97 EC |
42 | #include "mlx5_ib.h" |
43 | ||
44 | enum { | |
746b5583 | 45 | MAX_PENDING_REG_MR = 8, |
e126ba97 EC |
46 | }; |
47 | ||
832a6b06 | 48 | #define MLX5_UMR_ALIGN 2048 |
fe45f827 | 49 | |
fc6a9f86 SM |
50 | static void |
51 | create_mkey_callback(int status, struct mlx5_async_work *context); | |
52 | ||
53 | static void | |
54 | assign_mkey_variant(struct mlx5_ib_dev *dev, struct mlx5_core_mkey *mkey, | |
55 | u32 *in) | |
56 | { | |
f743ff3b | 57 | u8 key = atomic_inc_return(&dev->mkey_var); |
fc6a9f86 | 58 | void *mkc; |
fc6a9f86 SM |
59 | |
60 | mkc = MLX5_ADDR_OF(create_mkey_in, in, memory_key_mkey_entry); | |
61 | MLX5_SET(mkc, mkc, mkey_7_0, key); | |
62 | mkey->key = key; | |
63 | } | |
64 | ||
65 | static int | |
66 | mlx5_ib_create_mkey(struct mlx5_ib_dev *dev, struct mlx5_core_mkey *mkey, | |
67 | u32 *in, int inlen) | |
68 | { | |
69 | assign_mkey_variant(dev, mkey, in); | |
70 | return mlx5_core_create_mkey(dev->mdev, mkey, in, inlen); | |
71 | } | |
72 | ||
73 | static int | |
74 | mlx5_ib_create_mkey_cb(struct mlx5_ib_dev *dev, | |
75 | struct mlx5_core_mkey *mkey, | |
76 | struct mlx5_async_ctx *async_ctx, | |
77 | u32 *in, int inlen, u32 *out, int outlen, | |
78 | struct mlx5_async_work *context) | |
79 | { | |
a3cfdd39 | 80 | MLX5_SET(create_mkey_in, in, opcode, MLX5_CMD_OP_CREATE_MKEY); |
fc6a9f86 | 81 | assign_mkey_variant(dev, mkey, in); |
a3cfdd39 MG |
82 | return mlx5_cmd_exec_cb(async_ctx, in, inlen, out, outlen, |
83 | create_mkey_callback, context); | |
fc6a9f86 SM |
84 | } |
85 | ||
eeea6953 LR |
86 | static void clean_mr(struct mlx5_ib_dev *dev, struct mlx5_ib_mr *mr); |
87 | static void dereg_mr(struct mlx5_ib_dev *dev, struct mlx5_ib_mr *mr); | |
8b7ff7f3 | 88 | static int mr_cache_max_order(struct mlx5_ib_dev *dev); |
1c78a21a | 89 | static void queue_adjust_cache_locked(struct mlx5_cache_ent *ent); |
c8d75a98 MD |
90 | |
91 | static bool umr_can_use_indirect_mkey(struct mlx5_ib_dev *dev) | |
92 | { | |
93 | return !MLX5_CAP_GEN(dev->mdev, umr_indirect_mkey_disabled); | |
94 | } | |
95 | ||
b4cfe447 HE |
96 | static int destroy_mkey(struct mlx5_ib_dev *dev, struct mlx5_ib_mr *mr) |
97 | { | |
806b101b | 98 | WARN_ON(xa_load(&dev->odp_mkeys, mlx5_base_mkey(mr->mmkey.key))); |
b4cfe447 | 99 | |
806b101b | 100 | return mlx5_core_destroy_mkey(dev->mdev, &mr->mmkey); |
b4cfe447 HE |
101 | } |
102 | ||
56e11d62 NO |
103 | static bool use_umr_mtt_update(struct mlx5_ib_mr *mr, u64 start, u64 length) |
104 | { | |
105 | return ((u64)1 << mr->order) * MLX5_ADAPTER_PAGE_SIZE >= | |
106 | length + (start & (MLX5_ADAPTER_PAGE_SIZE - 1)); | |
107 | } | |
108 | ||
fc6a9f86 | 109 | static void create_mkey_callback(int status, struct mlx5_async_work *context) |
746b5583 | 110 | { |
e355477e JG |
111 | struct mlx5_ib_mr *mr = |
112 | container_of(context, struct mlx5_ib_mr, cb_work); | |
746b5583 | 113 | struct mlx5_ib_dev *dev = mr->dev; |
b91e1751 | 114 | struct mlx5_cache_ent *ent = mr->cache_ent; |
746b5583 EC |
115 | unsigned long flags; |
116 | ||
746b5583 EC |
117 | if (status) { |
118 | mlx5_ib_warn(dev, "async reg mr failed. status %d\n", status); | |
119 | kfree(mr); | |
b9358bdb JG |
120 | spin_lock_irqsave(&ent->lock, flags); |
121 | ent->pending--; | |
122 | WRITE_ONCE(dev->fill_delay, 1); | |
123 | spin_unlock_irqrestore(&ent->lock, flags); | |
746b5583 EC |
124 | mod_timer(&dev->delay_timer, jiffies + HZ); |
125 | return; | |
126 | } | |
127 | ||
aa8e08d2 | 128 | mr->mmkey.type = MLX5_MKEY_MR; |
54c62e13 SM |
129 | mr->mmkey.key |= mlx5_idx_to_mkey( |
130 | MLX5_GET(create_mkey_out, mr->out, mkey_index)); | |
746b5583 | 131 | |
b9358bdb | 132 | WRITE_ONCE(dev->cache.last_add, jiffies); |
746b5583 EC |
133 | |
134 | spin_lock_irqsave(&ent->lock, flags); | |
135 | list_add_tail(&mr->list, &ent->head); | |
7c8691a3 JG |
136 | ent->available_mrs++; |
137 | ent->total_mrs++; | |
1c78a21a JG |
138 | /* If we are doing fill_to_high_water then keep going. */ |
139 | queue_adjust_cache_locked(ent); | |
b9358bdb | 140 | ent->pending--; |
746b5583 | 141 | spin_unlock_irqrestore(&ent->lock, flags); |
aad719dc JG |
142 | } |
143 | ||
144 | static struct mlx5_ib_mr *alloc_cache_mr(struct mlx5_cache_ent *ent, void *mkc) | |
145 | { | |
146 | struct mlx5_ib_mr *mr; | |
147 | ||
148 | mr = kzalloc(sizeof(*mr), GFP_KERNEL); | |
149 | if (!mr) | |
150 | return NULL; | |
151 | mr->order = ent->order; | |
152 | mr->cache_ent = ent; | |
153 | mr->dev = ent->dev; | |
8605933a | 154 | |
aad719dc JG |
155 | MLX5_SET(mkc, mkc, free, 1); |
156 | MLX5_SET(mkc, mkc, umr_en, 1); | |
157 | MLX5_SET(mkc, mkc, access_mode_1_0, ent->access_mode & 0x3); | |
158 | MLX5_SET(mkc, mkc, access_mode_4_2, (ent->access_mode >> 2) & 0x7); | |
159 | ||
160 | MLX5_SET(mkc, mkc, qpn, 0xffffff); | |
161 | MLX5_SET(mkc, mkc, translations_octword_size, ent->xlt); | |
162 | MLX5_SET(mkc, mkc, log_page_size, ent->page); | |
163 | return mr; | |
746b5583 EC |
164 | } |
165 | ||
aad719dc | 166 | /* Asynchronously schedule new MRs to be populated in the cache. */ |
a1d8854a | 167 | static int add_keys(struct mlx5_cache_ent *ent, unsigned int num) |
e126ba97 | 168 | { |
aad719dc | 169 | size_t inlen = MLX5_ST_SZ_BYTES(create_mkey_in); |
e126ba97 | 170 | struct mlx5_ib_mr *mr; |
ec22eb53 SM |
171 | void *mkc; |
172 | u32 *in; | |
e126ba97 EC |
173 | int err = 0; |
174 | int i; | |
175 | ||
ec22eb53 | 176 | in = kzalloc(inlen, GFP_KERNEL); |
e126ba97 EC |
177 | if (!in) |
178 | return -ENOMEM; | |
179 | ||
ec22eb53 | 180 | mkc = MLX5_ADDR_OF(create_mkey_in, in, memory_key_mkey_entry); |
e126ba97 | 181 | for (i = 0; i < num; i++) { |
aad719dc | 182 | mr = alloc_cache_mr(ent, mkc); |
e126ba97 EC |
183 | if (!mr) { |
184 | err = -ENOMEM; | |
746b5583 | 185 | break; |
e126ba97 | 186 | } |
746b5583 | 187 | spin_lock_irq(&ent->lock); |
b9358bdb JG |
188 | if (ent->pending >= MAX_PENDING_REG_MR) { |
189 | err = -EAGAIN; | |
190 | spin_unlock_irq(&ent->lock); | |
191 | kfree(mr); | |
192 | break; | |
193 | } | |
746b5583 EC |
194 | ent->pending++; |
195 | spin_unlock_irq(&ent->lock); | |
b91e1751 JG |
196 | err = mlx5_ib_create_mkey_cb(ent->dev, &mr->mmkey, |
197 | &ent->dev->async_ctx, in, inlen, | |
198 | mr->out, sizeof(mr->out), | |
199 | &mr->cb_work); | |
e126ba97 | 200 | if (err) { |
d14e7110 EC |
201 | spin_lock_irq(&ent->lock); |
202 | ent->pending--; | |
203 | spin_unlock_irq(&ent->lock); | |
b91e1751 | 204 | mlx5_ib_warn(ent->dev, "create mkey failed %d\n", err); |
e126ba97 | 205 | kfree(mr); |
746b5583 | 206 | break; |
e126ba97 | 207 | } |
e126ba97 EC |
208 | } |
209 | ||
e126ba97 EC |
210 | kfree(in); |
211 | return err; | |
212 | } | |
213 | ||
aad719dc JG |
214 | /* Synchronously create a MR in the cache */ |
215 | static struct mlx5_ib_mr *create_cache_mr(struct mlx5_cache_ent *ent) | |
216 | { | |
217 | size_t inlen = MLX5_ST_SZ_BYTES(create_mkey_in); | |
218 | struct mlx5_ib_mr *mr; | |
219 | void *mkc; | |
220 | u32 *in; | |
221 | int err; | |
222 | ||
223 | in = kzalloc(inlen, GFP_KERNEL); | |
224 | if (!in) | |
225 | return ERR_PTR(-ENOMEM); | |
226 | mkc = MLX5_ADDR_OF(create_mkey_in, in, memory_key_mkey_entry); | |
227 | ||
228 | mr = alloc_cache_mr(ent, mkc); | |
229 | if (!mr) { | |
230 | err = -ENOMEM; | |
231 | goto free_in; | |
232 | } | |
233 | ||
234 | err = mlx5_core_create_mkey(ent->dev->mdev, &mr->mmkey, in, inlen); | |
235 | if (err) | |
236 | goto free_mr; | |
237 | ||
238 | mr->mmkey.type = MLX5_MKEY_MR; | |
239 | WRITE_ONCE(ent->dev->cache.last_add, jiffies); | |
240 | spin_lock_irq(&ent->lock); | |
241 | ent->total_mrs++; | |
242 | spin_unlock_irq(&ent->lock); | |
243 | kfree(in); | |
244 | return mr; | |
245 | free_mr: | |
246 | kfree(mr); | |
247 | free_in: | |
248 | kfree(in); | |
249 | return ERR_PTR(err); | |
250 | } | |
251 | ||
b9358bdb | 252 | static void remove_cache_mr_locked(struct mlx5_cache_ent *ent) |
e126ba97 | 253 | { |
e126ba97 | 254 | struct mlx5_ib_mr *mr; |
e126ba97 | 255 | |
b9358bdb JG |
256 | lockdep_assert_held(&ent->lock); |
257 | if (list_empty(&ent->head)) | |
a1d8854a | 258 | return; |
a1d8854a JG |
259 | mr = list_first_entry(&ent->head, struct mlx5_ib_mr, list); |
260 | list_del(&mr->list); | |
261 | ent->available_mrs--; | |
262 | ent->total_mrs--; | |
263 | spin_unlock_irq(&ent->lock); | |
264 | mlx5_core_destroy_mkey(ent->dev->mdev, &mr->mmkey); | |
265 | kfree(mr); | |
b9358bdb | 266 | spin_lock_irq(&ent->lock); |
a1d8854a | 267 | } |
65edd0e7 | 268 | |
a1d8854a JG |
269 | static int resize_available_mrs(struct mlx5_cache_ent *ent, unsigned int target, |
270 | bool limit_fill) | |
271 | { | |
272 | int err; | |
273 | ||
274 | lockdep_assert_held(&ent->lock); | |
275 | ||
276 | while (true) { | |
277 | if (limit_fill) | |
278 | target = ent->limit * 2; | |
279 | if (target == ent->available_mrs + ent->pending) | |
280 | return 0; | |
281 | if (target > ent->available_mrs + ent->pending) { | |
282 | u32 todo = target - (ent->available_mrs + ent->pending); | |
283 | ||
284 | spin_unlock_irq(&ent->lock); | |
285 | err = add_keys(ent, todo); | |
286 | if (err == -EAGAIN) | |
287 | usleep_range(3000, 5000); | |
288 | spin_lock_irq(&ent->lock); | |
289 | if (err) { | |
290 | if (err != -EAGAIN) | |
291 | return err; | |
292 | } else | |
293 | return 0; | |
294 | } else { | |
b9358bdb | 295 | remove_cache_mr_locked(ent); |
a1d8854a | 296 | } |
e126ba97 EC |
297 | } |
298 | } | |
299 | ||
300 | static ssize_t size_write(struct file *filp, const char __user *buf, | |
301 | size_t count, loff_t *pos) | |
302 | { | |
303 | struct mlx5_cache_ent *ent = filp->private_data; | |
a1d8854a | 304 | u32 target; |
e126ba97 | 305 | int err; |
e126ba97 | 306 | |
a1d8854a JG |
307 | err = kstrtou32_from_user(buf, count, 0, &target); |
308 | if (err) | |
309 | return err; | |
746b5583 | 310 | |
a1d8854a JG |
311 | /* |
312 | * Target is the new value of total_mrs the user requests, however we | |
313 | * cannot free MRs that are in use. Compute the target value for | |
314 | * available_mrs. | |
315 | */ | |
316 | spin_lock_irq(&ent->lock); | |
317 | if (target < ent->total_mrs - ent->available_mrs) { | |
318 | err = -EINVAL; | |
319 | goto err_unlock; | |
e126ba97 | 320 | } |
a1d8854a JG |
321 | target = target - (ent->total_mrs - ent->available_mrs); |
322 | if (target < ent->limit || target > ent->limit*2) { | |
323 | err = -EINVAL; | |
324 | goto err_unlock; | |
325 | } | |
326 | err = resize_available_mrs(ent, target, false); | |
327 | if (err) | |
328 | goto err_unlock; | |
329 | spin_unlock_irq(&ent->lock); | |
e126ba97 EC |
330 | |
331 | return count; | |
a1d8854a JG |
332 | |
333 | err_unlock: | |
334 | spin_unlock_irq(&ent->lock); | |
335 | return err; | |
e126ba97 EC |
336 | } |
337 | ||
338 | static ssize_t size_read(struct file *filp, char __user *buf, size_t count, | |
339 | loff_t *pos) | |
340 | { | |
341 | struct mlx5_cache_ent *ent = filp->private_data; | |
342 | char lbuf[20]; | |
343 | int err; | |
344 | ||
7c8691a3 | 345 | err = snprintf(lbuf, sizeof(lbuf), "%d\n", ent->total_mrs); |
e126ba97 EC |
346 | if (err < 0) |
347 | return err; | |
348 | ||
60e6627f | 349 | return simple_read_from_buffer(buf, count, pos, lbuf, err); |
e126ba97 EC |
350 | } |
351 | ||
352 | static const struct file_operations size_fops = { | |
353 | .owner = THIS_MODULE, | |
354 | .open = simple_open, | |
355 | .write = size_write, | |
356 | .read = size_read, | |
357 | }; | |
358 | ||
359 | static ssize_t limit_write(struct file *filp, const char __user *buf, | |
360 | size_t count, loff_t *pos) | |
361 | { | |
362 | struct mlx5_cache_ent *ent = filp->private_data; | |
e126ba97 EC |
363 | u32 var; |
364 | int err; | |
e126ba97 | 365 | |
a1d8854a JG |
366 | err = kstrtou32_from_user(buf, count, 0, &var); |
367 | if (err) | |
368 | return err; | |
e126ba97 | 369 | |
a1d8854a JG |
370 | /* |
371 | * Upon set we immediately fill the cache to high water mark implied by | |
372 | * the limit. | |
373 | */ | |
374 | spin_lock_irq(&ent->lock); | |
e126ba97 | 375 | ent->limit = var; |
a1d8854a JG |
376 | err = resize_available_mrs(ent, 0, true); |
377 | spin_unlock_irq(&ent->lock); | |
378 | if (err) | |
379 | return err; | |
e126ba97 EC |
380 | return count; |
381 | } | |
382 | ||
383 | static ssize_t limit_read(struct file *filp, char __user *buf, size_t count, | |
384 | loff_t *pos) | |
385 | { | |
386 | struct mlx5_cache_ent *ent = filp->private_data; | |
387 | char lbuf[20]; | |
388 | int err; | |
389 | ||
e126ba97 EC |
390 | err = snprintf(lbuf, sizeof(lbuf), "%d\n", ent->limit); |
391 | if (err < 0) | |
392 | return err; | |
393 | ||
60e6627f | 394 | return simple_read_from_buffer(buf, count, pos, lbuf, err); |
e126ba97 EC |
395 | } |
396 | ||
397 | static const struct file_operations limit_fops = { | |
398 | .owner = THIS_MODULE, | |
399 | .open = simple_open, | |
400 | .write = limit_write, | |
401 | .read = limit_read, | |
402 | }; | |
403 | ||
b9358bdb | 404 | static bool someone_adding(struct mlx5_mr_cache *cache) |
e126ba97 | 405 | { |
b9358bdb | 406 | unsigned int i; |
e126ba97 EC |
407 | |
408 | for (i = 0; i < MAX_MR_CACHE_ENTRIES; i++) { | |
b9358bdb JG |
409 | struct mlx5_cache_ent *ent = &cache->ent[i]; |
410 | bool ret; | |
e126ba97 | 411 | |
b9358bdb JG |
412 | spin_lock_irq(&ent->lock); |
413 | ret = ent->available_mrs < ent->limit; | |
414 | spin_unlock_irq(&ent->lock); | |
415 | if (ret) | |
416 | return true; | |
417 | } | |
418 | return false; | |
e126ba97 EC |
419 | } |
420 | ||
ad2d3ef4 JG |
421 | /* |
422 | * Check if the bucket is outside the high/low water mark and schedule an async | |
423 | * update. The cache refill has hysteresis, once the low water mark is hit it is | |
424 | * refilled up to the high mark. | |
425 | */ | |
426 | static void queue_adjust_cache_locked(struct mlx5_cache_ent *ent) | |
427 | { | |
428 | lockdep_assert_held(&ent->lock); | |
429 | ||
1c78a21a | 430 | if (ent->disabled || READ_ONCE(ent->dev->fill_delay)) |
b9358bdb | 431 | return; |
1c78a21a JG |
432 | if (ent->available_mrs < ent->limit) { |
433 | ent->fill_to_high_water = true; | |
434 | queue_work(ent->dev->cache.wq, &ent->work); | |
435 | } else if (ent->fill_to_high_water && | |
436 | ent->available_mrs + ent->pending < 2 * ent->limit) { | |
437 | /* | |
438 | * Once we start populating due to hitting a low water mark | |
439 | * continue until we pass the high water mark. | |
440 | */ | |
ad2d3ef4 | 441 | queue_work(ent->dev->cache.wq, &ent->work); |
1c78a21a JG |
442 | } else if (ent->available_mrs == 2 * ent->limit) { |
443 | ent->fill_to_high_water = false; | |
444 | } else if (ent->available_mrs > 2 * ent->limit) { | |
445 | /* Queue deletion of excess entries */ | |
446 | ent->fill_to_high_water = false; | |
447 | if (ent->pending) | |
448 | queue_delayed_work(ent->dev->cache.wq, &ent->dwork, | |
449 | msecs_to_jiffies(1000)); | |
450 | else | |
451 | queue_work(ent->dev->cache.wq, &ent->work); | |
452 | } | |
ad2d3ef4 JG |
453 | } |
454 | ||
e126ba97 EC |
455 | static void __cache_work_func(struct mlx5_cache_ent *ent) |
456 | { | |
457 | struct mlx5_ib_dev *dev = ent->dev; | |
458 | struct mlx5_mr_cache *cache = &dev->cache; | |
746b5583 | 459 | int err; |
e126ba97 | 460 | |
b9358bdb JG |
461 | spin_lock_irq(&ent->lock); |
462 | if (ent->disabled) | |
463 | goto out; | |
e126ba97 | 464 | |
1c78a21a JG |
465 | if (ent->fill_to_high_water && |
466 | ent->available_mrs + ent->pending < 2 * ent->limit && | |
b9358bdb JG |
467 | !READ_ONCE(dev->fill_delay)) { |
468 | spin_unlock_irq(&ent->lock); | |
b91e1751 | 469 | err = add_keys(ent, 1); |
b9358bdb JG |
470 | spin_lock_irq(&ent->lock); |
471 | if (ent->disabled) | |
472 | goto out; | |
473 | if (err) { | |
aad719dc JG |
474 | /* |
475 | * EAGAIN only happens if pending is positive, so we | |
476 | * will be rescheduled from reg_mr_callback(). The only | |
477 | * failure path here is ENOMEM. | |
478 | */ | |
479 | if (err != -EAGAIN) { | |
b9358bdb JG |
480 | mlx5_ib_warn( |
481 | dev, | |
482 | "command failed order %d, err %d\n", | |
483 | ent->order, err); | |
746b5583 EC |
484 | queue_delayed_work(cache->wq, &ent->dwork, |
485 | msecs_to_jiffies(1000)); | |
746b5583 EC |
486 | } |
487 | } | |
7c8691a3 | 488 | } else if (ent->available_mrs > 2 * ent->limit) { |
b9358bdb JG |
489 | bool need_delay; |
490 | ||
ab5cdc31 | 491 | /* |
a1d8854a JG |
492 | * The remove_cache_mr() logic is performed as garbage |
493 | * collection task. Such task is intended to be run when no | |
494 | * other active processes are running. | |
ab5cdc31 LR |
495 | * |
496 | * The need_resched() will return TRUE if there are user tasks | |
497 | * to be activated in near future. | |
498 | * | |
a1d8854a JG |
499 | * In such case, we don't execute remove_cache_mr() and postpone |
500 | * the garbage collection work to try to run in next cycle, in | |
501 | * order to free CPU resources to other tasks. | |
ab5cdc31 | 502 | */ |
b9358bdb JG |
503 | spin_unlock_irq(&ent->lock); |
504 | need_delay = need_resched() || someone_adding(cache) || | |
505 | time_after(jiffies, | |
506 | READ_ONCE(cache->last_add) + 300 * HZ); | |
507 | spin_lock_irq(&ent->lock); | |
508 | if (ent->disabled) | |
509 | goto out; | |
510 | if (need_delay) | |
746b5583 | 511 | queue_delayed_work(cache->wq, &ent->dwork, 300 * HZ); |
b9358bdb JG |
512 | remove_cache_mr_locked(ent); |
513 | queue_adjust_cache_locked(ent); | |
e126ba97 | 514 | } |
b9358bdb JG |
515 | out: |
516 | spin_unlock_irq(&ent->lock); | |
e126ba97 EC |
517 | } |
518 | ||
519 | static void delayed_cache_work_func(struct work_struct *work) | |
520 | { | |
521 | struct mlx5_cache_ent *ent; | |
522 | ||
523 | ent = container_of(work, struct mlx5_cache_ent, dwork.work); | |
524 | __cache_work_func(ent); | |
525 | } | |
526 | ||
527 | static void cache_work_func(struct work_struct *work) | |
528 | { | |
529 | struct mlx5_cache_ent *ent; | |
530 | ||
531 | ent = container_of(work, struct mlx5_cache_ent, work); | |
532 | __cache_work_func(ent); | |
533 | } | |
534 | ||
b91e1751 JG |
535 | /* Allocate a special entry from the cache */ |
536 | struct mlx5_ib_mr *mlx5_mr_cache_alloc(struct mlx5_ib_dev *dev, | |
537 | unsigned int entry) | |
49780d42 AK |
538 | { |
539 | struct mlx5_mr_cache *cache = &dev->cache; | |
540 | struct mlx5_cache_ent *ent; | |
541 | struct mlx5_ib_mr *mr; | |
49780d42 | 542 | |
b91e1751 JG |
543 | if (WARN_ON(entry <= MR_CACHE_LAST_STD_ENTRY || |
544 | entry >= ARRAY_SIZE(cache->ent))) | |
546d3009 | 545 | return ERR_PTR(-EINVAL); |
49780d42 AK |
546 | |
547 | ent = &cache->ent[entry]; | |
aad719dc JG |
548 | spin_lock_irq(&ent->lock); |
549 | if (list_empty(&ent->head)) { | |
550 | spin_unlock_irq(&ent->lock); | |
551 | mr = create_cache_mr(ent); | |
552 | if (IS_ERR(mr)) | |
49780d42 | 553 | return mr; |
aad719dc JG |
554 | } else { |
555 | mr = list_first_entry(&ent->head, struct mlx5_ib_mr, list); | |
556 | list_del(&mr->list); | |
557 | ent->available_mrs--; | |
558 | queue_adjust_cache_locked(ent); | |
559 | spin_unlock_irq(&ent->lock); | |
49780d42 | 560 | } |
aad719dc | 561 | return mr; |
49780d42 AK |
562 | } |
563 | ||
aad719dc JG |
564 | /* Return a MR already available in the cache */ |
565 | static struct mlx5_ib_mr *get_cache_mr(struct mlx5_cache_ent *req_ent) | |
e126ba97 | 566 | { |
b91e1751 | 567 | struct mlx5_ib_dev *dev = req_ent->dev; |
e126ba97 | 568 | struct mlx5_ib_mr *mr = NULL; |
b91e1751 | 569 | struct mlx5_cache_ent *ent = req_ent; |
e126ba97 | 570 | |
b91e1751 JG |
571 | /* Try larger MR pools from the cache to satisfy the allocation */ |
572 | for (; ent != &dev->cache.ent[MR_CACHE_LAST_STD_ENTRY + 1]; ent++) { | |
573 | mlx5_ib_dbg(dev, "order %u, cache index %zu\n", ent->order, | |
574 | ent - dev->cache.ent); | |
e126ba97 | 575 | |
746b5583 | 576 | spin_lock_irq(&ent->lock); |
e126ba97 EC |
577 | if (!list_empty(&ent->head)) { |
578 | mr = list_first_entry(&ent->head, struct mlx5_ib_mr, | |
579 | list); | |
580 | list_del(&mr->list); | |
7c8691a3 | 581 | ent->available_mrs--; |
ad2d3ef4 | 582 | queue_adjust_cache_locked(ent); |
746b5583 | 583 | spin_unlock_irq(&ent->lock); |
e126ba97 EC |
584 | break; |
585 | } | |
ad2d3ef4 | 586 | queue_adjust_cache_locked(ent); |
746b5583 | 587 | spin_unlock_irq(&ent->lock); |
e126ba97 EC |
588 | } |
589 | ||
590 | if (!mr) | |
b91e1751 | 591 | req_ent->miss++; |
e126ba97 EC |
592 | |
593 | return mr; | |
594 | } | |
595 | ||
1769c4c5 JG |
596 | static void detach_mr_from_cache(struct mlx5_ib_mr *mr) |
597 | { | |
598 | struct mlx5_cache_ent *ent = mr->cache_ent; | |
599 | ||
600 | mr->cache_ent = NULL; | |
601 | spin_lock_irq(&ent->lock); | |
602 | ent->total_mrs--; | |
603 | spin_unlock_irq(&ent->lock); | |
604 | } | |
605 | ||
49780d42 | 606 | void mlx5_mr_cache_free(struct mlx5_ib_dev *dev, struct mlx5_ib_mr *mr) |
e126ba97 | 607 | { |
b91e1751 | 608 | struct mlx5_cache_ent *ent = mr->cache_ent; |
e126ba97 | 609 | |
b91e1751 | 610 | if (!ent) |
dd9a4034 VF |
611 | return; |
612 | ||
09689703 | 613 | if (mlx5_mr_cache_invalidate(mr)) { |
1769c4c5 | 614 | detach_mr_from_cache(mr); |
afd14174 | 615 | destroy_mkey(dev, mr); |
e126ba97 EC |
616 | return; |
617 | } | |
49780d42 | 618 | |
746b5583 | 619 | spin_lock_irq(&ent->lock); |
e126ba97 | 620 | list_add_tail(&mr->list, &ent->head); |
7c8691a3 | 621 | ent->available_mrs++; |
ad2d3ef4 | 622 | queue_adjust_cache_locked(ent); |
746b5583 | 623 | spin_unlock_irq(&ent->lock); |
e126ba97 EC |
624 | } |
625 | ||
626 | static void clean_keys(struct mlx5_ib_dev *dev, int c) | |
627 | { | |
e126ba97 EC |
628 | struct mlx5_mr_cache *cache = &dev->cache; |
629 | struct mlx5_cache_ent *ent = &cache->ent[c]; | |
65edd0e7 | 630 | struct mlx5_ib_mr *tmp_mr; |
e126ba97 | 631 | struct mlx5_ib_mr *mr; |
65edd0e7 | 632 | LIST_HEAD(del_list); |
e126ba97 | 633 | |
3c461911 | 634 | cancel_delayed_work(&ent->dwork); |
e126ba97 | 635 | while (1) { |
746b5583 | 636 | spin_lock_irq(&ent->lock); |
e126ba97 | 637 | if (list_empty(&ent->head)) { |
746b5583 | 638 | spin_unlock_irq(&ent->lock); |
65edd0e7 | 639 | break; |
e126ba97 EC |
640 | } |
641 | mr = list_first_entry(&ent->head, struct mlx5_ib_mr, list); | |
65edd0e7 | 642 | list_move(&mr->list, &del_list); |
7c8691a3 JG |
643 | ent->available_mrs--; |
644 | ent->total_mrs--; | |
746b5583 | 645 | spin_unlock_irq(&ent->lock); |
65edd0e7 DJ |
646 | mlx5_core_destroy_mkey(dev->mdev, &mr->mmkey); |
647 | } | |
648 | ||
65edd0e7 DJ |
649 | list_for_each_entry_safe(mr, tmp_mr, &del_list, list) { |
650 | list_del(&mr->list); | |
651 | kfree(mr); | |
e126ba97 EC |
652 | } |
653 | } | |
654 | ||
12cc1a02 LR |
655 | static void mlx5_mr_cache_debugfs_cleanup(struct mlx5_ib_dev *dev) |
656 | { | |
6a4d00be | 657 | if (!mlx5_debugfs_root || dev->is_rep) |
12cc1a02 LR |
658 | return; |
659 | ||
660 | debugfs_remove_recursive(dev->cache.root); | |
661 | dev->cache.root = NULL; | |
662 | } | |
663 | ||
73eb8f03 | 664 | static void mlx5_mr_cache_debugfs_init(struct mlx5_ib_dev *dev) |
e126ba97 EC |
665 | { |
666 | struct mlx5_mr_cache *cache = &dev->cache; | |
667 | struct mlx5_cache_ent *ent; | |
73eb8f03 | 668 | struct dentry *dir; |
e126ba97 EC |
669 | int i; |
670 | ||
6a4d00be | 671 | if (!mlx5_debugfs_root || dev->is_rep) |
73eb8f03 | 672 | return; |
e126ba97 | 673 | |
9603b61d | 674 | cache->root = debugfs_create_dir("mr_cache", dev->mdev->priv.dbg_root); |
e126ba97 EC |
675 | |
676 | for (i = 0; i < MAX_MR_CACHE_ENTRIES; i++) { | |
677 | ent = &cache->ent[i]; | |
678 | sprintf(ent->name, "%d", ent->order); | |
73eb8f03 GKH |
679 | dir = debugfs_create_dir(ent->name, cache->root); |
680 | debugfs_create_file("size", 0600, dir, ent, &size_fops); | |
681 | debugfs_create_file("limit", 0600, dir, ent, &limit_fops); | |
7c8691a3 | 682 | debugfs_create_u32("cur", 0400, dir, &ent->available_mrs); |
73eb8f03 | 683 | debugfs_create_u32("miss", 0600, dir, &ent->miss); |
e126ba97 | 684 | } |
e126ba97 EC |
685 | } |
686 | ||
e99e88a9 | 687 | static void delay_time_func(struct timer_list *t) |
746b5583 | 688 | { |
e99e88a9 | 689 | struct mlx5_ib_dev *dev = from_timer(dev, t, delay_timer); |
746b5583 | 690 | |
b9358bdb | 691 | WRITE_ONCE(dev->fill_delay, 0); |
746b5583 EC |
692 | } |
693 | ||
e126ba97 EC |
694 | int mlx5_mr_cache_init(struct mlx5_ib_dev *dev) |
695 | { | |
696 | struct mlx5_mr_cache *cache = &dev->cache; | |
697 | struct mlx5_cache_ent *ent; | |
e126ba97 EC |
698 | int i; |
699 | ||
6bc1a656 | 700 | mutex_init(&dev->slow_path_mutex); |
3c856c82 | 701 | cache->wq = alloc_ordered_workqueue("mkey_cache", WQ_MEM_RECLAIM); |
e126ba97 EC |
702 | if (!cache->wq) { |
703 | mlx5_ib_warn(dev, "failed to create work queue\n"); | |
704 | return -ENOMEM; | |
705 | } | |
706 | ||
e355477e | 707 | mlx5_cmd_init_async_ctx(dev->mdev, &dev->async_ctx); |
e99e88a9 | 708 | timer_setup(&dev->delay_timer, delay_time_func, 0); |
e126ba97 | 709 | for (i = 0; i < MAX_MR_CACHE_ENTRIES; i++) { |
e126ba97 EC |
710 | ent = &cache->ent[i]; |
711 | INIT_LIST_HEAD(&ent->head); | |
712 | spin_lock_init(&ent->lock); | |
713 | ent->order = i + 2; | |
714 | ent->dev = dev; | |
49780d42 | 715 | ent->limit = 0; |
e126ba97 | 716 | |
e126ba97 EC |
717 | INIT_WORK(&ent->work, cache_work_func); |
718 | INIT_DELAYED_WORK(&ent->dwork, delayed_cache_work_func); | |
49780d42 | 719 | |
8b7ff7f3 | 720 | if (i > MR_CACHE_LAST_STD_ENTRY) { |
81713d37 | 721 | mlx5_odp_init_mr_cache_entry(ent); |
49780d42 | 722 | continue; |
81713d37 | 723 | } |
49780d42 | 724 | |
8b7ff7f3 | 725 | if (ent->order > mr_cache_max_order(dev)) |
49780d42 AK |
726 | continue; |
727 | ||
728 | ent->page = PAGE_SHIFT; | |
729 | ent->xlt = (1 << ent->order) * sizeof(struct mlx5_mtt) / | |
730 | MLX5_IB_UMR_OCTOWORD; | |
731 | ent->access_mode = MLX5_MKC_ACCESS_MODE_MTT; | |
732 | if ((dev->mdev->profile->mask & MLX5_PROF_MASK_MR_CACHE) && | |
6a4d00be | 733 | !dev->is_rep && |
49780d42 AK |
734 | mlx5_core_is_pf(dev->mdev)) |
735 | ent->limit = dev->mdev->profile->mr_cache[i].limit; | |
736 | else | |
737 | ent->limit = 0; | |
ad2d3ef4 JG |
738 | spin_lock_irq(&ent->lock); |
739 | queue_adjust_cache_locked(ent); | |
740 | spin_unlock_irq(&ent->lock); | |
e126ba97 EC |
741 | } |
742 | ||
73eb8f03 | 743 | mlx5_mr_cache_debugfs_init(dev); |
12cc1a02 | 744 | |
e126ba97 EC |
745 | return 0; |
746 | } | |
747 | ||
748 | int mlx5_mr_cache_cleanup(struct mlx5_ib_dev *dev) | |
749 | { | |
b9358bdb | 750 | unsigned int i; |
e126ba97 | 751 | |
32927e28 MB |
752 | if (!dev->cache.wq) |
753 | return 0; | |
754 | ||
b9358bdb JG |
755 | for (i = 0; i < MAX_MR_CACHE_ENTRIES; i++) { |
756 | struct mlx5_cache_ent *ent = &dev->cache.ent[i]; | |
757 | ||
758 | spin_lock_irq(&ent->lock); | |
759 | ent->disabled = true; | |
760 | spin_unlock_irq(&ent->lock); | |
761 | cancel_work_sync(&ent->work); | |
762 | cancel_delayed_work_sync(&ent->dwork); | |
763 | } | |
e126ba97 EC |
764 | |
765 | mlx5_mr_cache_debugfs_cleanup(dev); | |
e355477e | 766 | mlx5_cmd_cleanup_async_ctx(&dev->async_ctx); |
e126ba97 EC |
767 | |
768 | for (i = 0; i < MAX_MR_CACHE_ENTRIES; i++) | |
769 | clean_keys(dev, i); | |
770 | ||
3c461911 | 771 | destroy_workqueue(dev->cache.wq); |
746b5583 | 772 | del_timer_sync(&dev->delay_timer); |
3c461911 | 773 | |
e126ba97 EC |
774 | return 0; |
775 | } | |
776 | ||
03232cc4 PP |
777 | static void set_mkc_access_pd_addr_fields(void *mkc, int acc, u64 start_addr, |
778 | struct ib_pd *pd) | |
779 | { | |
d6de0bb1 MG |
780 | struct mlx5_ib_dev *dev = to_mdev(pd->device); |
781 | ||
03232cc4 PP |
782 | MLX5_SET(mkc, mkc, a, !!(acc & IB_ACCESS_REMOTE_ATOMIC)); |
783 | MLX5_SET(mkc, mkc, rw, !!(acc & IB_ACCESS_REMOTE_WRITE)); | |
784 | MLX5_SET(mkc, mkc, rr, !!(acc & IB_ACCESS_REMOTE_READ)); | |
785 | MLX5_SET(mkc, mkc, lw, !!(acc & IB_ACCESS_LOCAL_WRITE)); | |
786 | MLX5_SET(mkc, mkc, lr, 1); | |
787 | ||
d6de0bb1 MG |
788 | if (MLX5_CAP_GEN(dev->mdev, relaxed_ordering_write)) |
789 | MLX5_SET(mkc, mkc, relaxed_ordering_write, | |
790 | !!(acc & IB_ACCESS_RELAXED_ORDERING)); | |
791 | if (MLX5_CAP_GEN(dev->mdev, relaxed_ordering_read)) | |
792 | MLX5_SET(mkc, mkc, relaxed_ordering_read, | |
793 | !!(acc & IB_ACCESS_RELAXED_ORDERING)); | |
794 | ||
03232cc4 PP |
795 | MLX5_SET(mkc, mkc, pd, to_mpd(pd)->pdn); |
796 | MLX5_SET(mkc, mkc, qpn, 0xffffff); | |
797 | MLX5_SET64(mkc, mkc, start_addr, start_addr); | |
798 | } | |
799 | ||
e126ba97 EC |
800 | struct ib_mr *mlx5_ib_get_dma_mr(struct ib_pd *pd, int acc) |
801 | { | |
802 | struct mlx5_ib_dev *dev = to_mdev(pd->device); | |
ec22eb53 | 803 | int inlen = MLX5_ST_SZ_BYTES(create_mkey_in); |
e126ba97 | 804 | struct mlx5_ib_mr *mr; |
ec22eb53 SM |
805 | void *mkc; |
806 | u32 *in; | |
e126ba97 EC |
807 | int err; |
808 | ||
809 | mr = kzalloc(sizeof(*mr), GFP_KERNEL); | |
810 | if (!mr) | |
811 | return ERR_PTR(-ENOMEM); | |
812 | ||
ec22eb53 | 813 | in = kzalloc(inlen, GFP_KERNEL); |
e126ba97 EC |
814 | if (!in) { |
815 | err = -ENOMEM; | |
816 | goto err_free; | |
817 | } | |
818 | ||
ec22eb53 SM |
819 | mkc = MLX5_ADDR_OF(create_mkey_in, in, memory_key_mkey_entry); |
820 | ||
cdbd0d2b | 821 | MLX5_SET(mkc, mkc, access_mode_1_0, MLX5_MKC_ACCESS_MODE_PA); |
ec22eb53 | 822 | MLX5_SET(mkc, mkc, length64, 1); |
03232cc4 | 823 | set_mkc_access_pd_addr_fields(mkc, acc, 0, pd); |
ec22eb53 | 824 | |
fc6a9f86 | 825 | err = mlx5_ib_create_mkey(dev, &mr->mmkey, in, inlen); |
e126ba97 EC |
826 | if (err) |
827 | goto err_in; | |
828 | ||
829 | kfree(in); | |
aa8e08d2 | 830 | mr->mmkey.type = MLX5_MKEY_MR; |
a606b0f6 MB |
831 | mr->ibmr.lkey = mr->mmkey.key; |
832 | mr->ibmr.rkey = mr->mmkey.key; | |
e126ba97 EC |
833 | mr->umem = NULL; |
834 | ||
835 | return &mr->ibmr; | |
836 | ||
837 | err_in: | |
838 | kfree(in); | |
839 | ||
840 | err_free: | |
841 | kfree(mr); | |
842 | ||
843 | return ERR_PTR(err); | |
844 | } | |
845 | ||
7b4cdaae | 846 | static int get_octo_len(u64 addr, u64 len, int page_shift) |
e126ba97 | 847 | { |
7b4cdaae | 848 | u64 page_size = 1ULL << page_shift; |
e126ba97 EC |
849 | u64 offset; |
850 | int npages; | |
851 | ||
852 | offset = addr & (page_size - 1); | |
7b4cdaae | 853 | npages = ALIGN(len + offset, page_size) >> page_shift; |
e126ba97 EC |
854 | return (npages + 1) / 2; |
855 | } | |
856 | ||
8b7ff7f3 | 857 | static int mr_cache_max_order(struct mlx5_ib_dev *dev) |
e126ba97 | 858 | { |
7d0cc6ed | 859 | if (MLX5_CAP_GEN(dev->mdev, umr_extended_translation_offset)) |
8b7ff7f3 | 860 | return MR_CACHE_LAST_STD_ENTRY + 2; |
4c25b7a3 MD |
861 | return MLX5_MAX_UMR_SHIFT; |
862 | } | |
863 | ||
c320e527 MS |
864 | static int mr_umem_get(struct mlx5_ib_dev *dev, u64 start, u64 length, |
865 | int access_flags, struct ib_umem **umem, int *npages, | |
866 | int *page_shift, int *ncont, int *order) | |
395a8e4c | 867 | { |
b4bd701a | 868 | struct ib_umem *u; |
14ab8896 | 869 | |
b4bd701a LR |
870 | *umem = NULL; |
871 | ||
261dc53f JG |
872 | if (access_flags & IB_ACCESS_ON_DEMAND) { |
873 | struct ib_umem_odp *odp; | |
874 | ||
c320e527 | 875 | odp = ib_umem_odp_get(&dev->ib_dev, start, length, access_flags, |
f25a546e | 876 | &mlx5_mn_ops); |
261dc53f JG |
877 | if (IS_ERR(odp)) { |
878 | mlx5_ib_dbg(dev, "umem get failed (%ld)\n", | |
879 | PTR_ERR(odp)); | |
880 | return PTR_ERR(odp); | |
881 | } | |
882 | ||
883 | u = &odp->umem; | |
884 | ||
885 | *page_shift = odp->page_shift; | |
886 | *ncont = ib_umem_odp_num_pages(odp); | |
887 | *npages = *ncont << (*page_shift - PAGE_SHIFT); | |
888 | if (order) | |
889 | *order = ilog2(roundup_pow_of_two(*ncont)); | |
890 | } else { | |
c320e527 | 891 | u = ib_umem_get(&dev->ib_dev, start, length, access_flags); |
261dc53f JG |
892 | if (IS_ERR(u)) { |
893 | mlx5_ib_dbg(dev, "umem get failed (%ld)\n", PTR_ERR(u)); | |
894 | return PTR_ERR(u); | |
895 | } | |
896 | ||
897 | mlx5_ib_cont_pages(u, start, MLX5_MKEY_PAGE_SHIFT_MASK, npages, | |
898 | page_shift, ncont, order); | |
395a8e4c NO |
899 | } |
900 | ||
395a8e4c NO |
901 | if (!*npages) { |
902 | mlx5_ib_warn(dev, "avoid zero region\n"); | |
b4bd701a | 903 | ib_umem_release(u); |
14ab8896 | 904 | return -EINVAL; |
395a8e4c NO |
905 | } |
906 | ||
b4bd701a LR |
907 | *umem = u; |
908 | ||
395a8e4c NO |
909 | mlx5_ib_dbg(dev, "npages %d, ncont %d, order %d, page_shift %d\n", |
910 | *npages, *ncont, *order, *page_shift); | |
911 | ||
14ab8896 | 912 | return 0; |
395a8e4c NO |
913 | } |
914 | ||
add08d76 | 915 | static void mlx5_ib_umr_done(struct ib_cq *cq, struct ib_wc *wc) |
e126ba97 | 916 | { |
add08d76 CH |
917 | struct mlx5_ib_umr_context *context = |
918 | container_of(wc->wr_cqe, struct mlx5_ib_umr_context, cqe); | |
e126ba97 | 919 | |
add08d76 CH |
920 | context->status = wc->status; |
921 | complete(&context->done); | |
922 | } | |
e126ba97 | 923 | |
add08d76 CH |
924 | static inline void mlx5_ib_init_umr_context(struct mlx5_ib_umr_context *context) |
925 | { | |
926 | context->cqe.done = mlx5_ib_umr_done; | |
927 | context->status = -1; | |
928 | init_completion(&context->done); | |
e126ba97 EC |
929 | } |
930 | ||
d5ea2df9 BJ |
931 | static int mlx5_ib_post_send_wait(struct mlx5_ib_dev *dev, |
932 | struct mlx5_umr_wr *umrwr) | |
933 | { | |
934 | struct umr_common *umrc = &dev->umrc; | |
d34ac5cd | 935 | const struct ib_send_wr *bad; |
d5ea2df9 BJ |
936 | int err; |
937 | struct mlx5_ib_umr_context umr_context; | |
938 | ||
939 | mlx5_ib_init_umr_context(&umr_context); | |
940 | umrwr->wr.wr_cqe = &umr_context.cqe; | |
941 | ||
942 | down(&umrc->sem); | |
943 | err = ib_post_send(umrc->qp, &umrwr->wr, &bad); | |
944 | if (err) { | |
945 | mlx5_ib_warn(dev, "UMR post send failed, err %d\n", err); | |
946 | } else { | |
947 | wait_for_completion(&umr_context.done); | |
948 | if (umr_context.status != IB_WC_SUCCESS) { | |
949 | mlx5_ib_warn(dev, "reg umr failed (%u)\n", | |
950 | umr_context.status); | |
951 | err = -EFAULT; | |
952 | } | |
953 | } | |
954 | up(&umrc->sem); | |
955 | return err; | |
956 | } | |
957 | ||
b91e1751 JG |
958 | static struct mlx5_cache_ent *mr_cache_ent_from_order(struct mlx5_ib_dev *dev, |
959 | unsigned int order) | |
960 | { | |
961 | struct mlx5_mr_cache *cache = &dev->cache; | |
962 | ||
963 | if (order < cache->ent[0].order) | |
964 | return &cache->ent[0]; | |
965 | order = order - cache->ent[0].order; | |
966 | if (order > MR_CACHE_LAST_STD_ENTRY) | |
967 | return NULL; | |
968 | return &cache->ent[order]; | |
969 | } | |
970 | ||
971 | static struct mlx5_ib_mr * | |
972 | alloc_mr_from_cache(struct ib_pd *pd, struct ib_umem *umem, u64 virt_addr, | |
973 | u64 len, int npages, int page_shift, unsigned int order, | |
974 | int access_flags) | |
e126ba97 EC |
975 | { |
976 | struct mlx5_ib_dev *dev = to_mdev(pd->device); | |
b91e1751 | 977 | struct mlx5_cache_ent *ent = mr_cache_ent_from_order(dev, order); |
e126ba97 | 978 | struct mlx5_ib_mr *mr; |
e126ba97 | 979 | |
b91e1751 JG |
980 | if (!ent) |
981 | return ERR_PTR(-E2BIG); | |
aad719dc JG |
982 | mr = get_cache_mr(ent); |
983 | if (!mr) { | |
984 | mr = create_cache_mr(ent); | |
985 | if (IS_ERR(mr)) | |
986 | return mr; | |
e126ba97 EC |
987 | } |
988 | ||
7d0cc6ed AK |
989 | mr->ibmr.pd = pd; |
990 | mr->umem = umem; | |
991 | mr->access_flags = access_flags; | |
992 | mr->desc_size = sizeof(struct mlx5_mtt); | |
a606b0f6 MB |
993 | mr->mmkey.iova = virt_addr; |
994 | mr->mmkey.size = len; | |
995 | mr->mmkey.pd = to_mpd(pd)->pdn; | |
b475598a | 996 | |
e126ba97 | 997 | return mr; |
e126ba97 EC |
998 | } |
999 | ||
7d0cc6ed AK |
1000 | #define MLX5_MAX_UMR_CHUNK ((1 << (MLX5_MAX_UMR_SHIFT + 4)) - \ |
1001 | MLX5_UMR_MTT_ALIGNMENT) | |
1002 | #define MLX5_SPARE_UMR_CHUNK 0x10000 | |
1003 | ||
1004 | int mlx5_ib_update_xlt(struct mlx5_ib_mr *mr, u64 idx, int npages, | |
1005 | int page_shift, int flags) | |
1006 | { | |
1007 | struct mlx5_ib_dev *dev = mr->dev; | |
9b0c289e | 1008 | struct device *ddev = dev->ib_dev.dev.parent; |
832a6b06 | 1009 | int size; |
7d0cc6ed | 1010 | void *xlt; |
832a6b06 | 1011 | dma_addr_t dma; |
e622f2f4 | 1012 | struct mlx5_umr_wr wr; |
832a6b06 HE |
1013 | struct ib_sge sg; |
1014 | int err = 0; | |
81713d37 AK |
1015 | int desc_size = (flags & MLX5_IB_UPD_XLT_INDIRECT) |
1016 | ? sizeof(struct mlx5_klm) | |
1017 | : sizeof(struct mlx5_mtt); | |
7d0cc6ed AK |
1018 | const int page_align = MLX5_UMR_MTT_ALIGNMENT / desc_size; |
1019 | const int page_mask = page_align - 1; | |
832a6b06 HE |
1020 | size_t pages_mapped = 0; |
1021 | size_t pages_to_map = 0; | |
1022 | size_t pages_iter = 0; | |
cbe4b8f0 | 1023 | size_t size_to_map = 0; |
7d0cc6ed | 1024 | gfp_t gfp; |
c44ef998 | 1025 | bool use_emergency_page = false; |
832a6b06 | 1026 | |
c8d75a98 MD |
1027 | if ((flags & MLX5_IB_UPD_XLT_INDIRECT) && |
1028 | !umr_can_use_indirect_mkey(dev)) | |
1029 | return -EPERM; | |
832a6b06 HE |
1030 | |
1031 | /* UMR copies MTTs in units of MLX5_UMR_MTT_ALIGNMENT bytes, | |
7d0cc6ed AK |
1032 | * so we need to align the offset and length accordingly |
1033 | */ | |
1034 | if (idx & page_mask) { | |
1035 | npages += idx & page_mask; | |
1036 | idx &= ~page_mask; | |
832a6b06 HE |
1037 | } |
1038 | ||
7d0cc6ed AK |
1039 | gfp = flags & MLX5_IB_UPD_XLT_ATOMIC ? GFP_ATOMIC : GFP_KERNEL; |
1040 | gfp |= __GFP_ZERO | __GFP_NOWARN; | |
832a6b06 | 1041 | |
7d0cc6ed AK |
1042 | pages_to_map = ALIGN(npages, page_align); |
1043 | size = desc_size * pages_to_map; | |
1044 | size = min_t(int, size, MLX5_MAX_UMR_CHUNK); | |
832a6b06 | 1045 | |
7d0cc6ed AK |
1046 | xlt = (void *)__get_free_pages(gfp, get_order(size)); |
1047 | if (!xlt && size > MLX5_SPARE_UMR_CHUNK) { | |
1048 | mlx5_ib_dbg(dev, "Failed to allocate %d bytes of order %d. fallback to spare UMR allocation od %d bytes\n", | |
1049 | size, get_order(size), MLX5_SPARE_UMR_CHUNK); | |
1050 | ||
1051 | size = MLX5_SPARE_UMR_CHUNK; | |
1052 | xlt = (void *)__get_free_pages(gfp, get_order(size)); | |
832a6b06 | 1053 | } |
7d0cc6ed AK |
1054 | |
1055 | if (!xlt) { | |
7d0cc6ed | 1056 | mlx5_ib_warn(dev, "Using XLT emergency buffer\n"); |
c44ef998 | 1057 | xlt = (void *)mlx5_ib_get_xlt_emergency_page(); |
7d0cc6ed | 1058 | size = PAGE_SIZE; |
7d0cc6ed | 1059 | memset(xlt, 0, size); |
c44ef998 | 1060 | use_emergency_page = true; |
7d0cc6ed AK |
1061 | } |
1062 | pages_iter = size / desc_size; | |
1063 | dma = dma_map_single(ddev, xlt, size, DMA_TO_DEVICE); | |
832a6b06 | 1064 | if (dma_mapping_error(ddev, dma)) { |
7d0cc6ed | 1065 | mlx5_ib_err(dev, "unable to map DMA during XLT update.\n"); |
832a6b06 | 1066 | err = -ENOMEM; |
7d0cc6ed | 1067 | goto free_xlt; |
832a6b06 HE |
1068 | } |
1069 | ||
cbe4b8f0 AK |
1070 | if (mr->umem->is_odp) { |
1071 | if (!(flags & MLX5_IB_UPD_XLT_INDIRECT)) { | |
1072 | struct ib_umem_odp *odp = to_ib_umem_odp(mr->umem); | |
1073 | size_t max_pages = ib_umem_odp_num_pages(odp) - idx; | |
1074 | ||
1075 | pages_to_map = min_t(size_t, pages_to_map, max_pages); | |
1076 | } | |
1077 | } | |
1078 | ||
7d0cc6ed AK |
1079 | sg.addr = dma; |
1080 | sg.lkey = dev->umrc.pd->local_dma_lkey; | |
1081 | ||
1082 | memset(&wr, 0, sizeof(wr)); | |
1083 | wr.wr.send_flags = MLX5_IB_SEND_UMR_UPDATE_XLT; | |
1084 | if (!(flags & MLX5_IB_UPD_XLT_ENABLE)) | |
1085 | wr.wr.send_flags |= MLX5_IB_SEND_UMR_FAIL_IF_FREE; | |
1086 | wr.wr.sg_list = &sg; | |
1087 | wr.wr.num_sge = 1; | |
1088 | wr.wr.opcode = MLX5_IB_WR_UMR; | |
1089 | ||
1090 | wr.pd = mr->ibmr.pd; | |
1091 | wr.mkey = mr->mmkey.key; | |
1092 | wr.length = mr->mmkey.size; | |
1093 | wr.virt_addr = mr->mmkey.iova; | |
1094 | wr.access_flags = mr->access_flags; | |
1095 | wr.page_shift = page_shift; | |
1096 | ||
832a6b06 HE |
1097 | for (pages_mapped = 0; |
1098 | pages_mapped < pages_to_map && !err; | |
7d0cc6ed | 1099 | pages_mapped += pages_iter, idx += pages_iter) { |
438b228e | 1100 | npages = min_t(int, pages_iter, pages_to_map - pages_mapped); |
cbe4b8f0 | 1101 | size_to_map = npages * desc_size; |
832a6b06 | 1102 | dma_sync_single_for_cpu(ddev, dma, size, DMA_TO_DEVICE); |
cbe4b8f0 AK |
1103 | if (mr->umem->is_odp) { |
1104 | mlx5_odp_populate_xlt(xlt, idx, npages, mr, flags); | |
1105 | } else { | |
1106 | __mlx5_ib_populate_pas(dev, mr->umem, page_shift, idx, | |
1107 | npages, xlt, | |
1108 | MLX5_IB_MTT_PRESENT); | |
1109 | /* Clear padding after the pages | |
1110 | * brought from the umem. | |
1111 | */ | |
1112 | memset(xlt + size_to_map, 0, size - size_to_map); | |
1113 | } | |
832a6b06 HE |
1114 | dma_sync_single_for_device(ddev, dma, size, DMA_TO_DEVICE); |
1115 | ||
cbe4b8f0 | 1116 | sg.length = ALIGN(size_to_map, MLX5_UMR_MTT_ALIGNMENT); |
7d0cc6ed AK |
1117 | |
1118 | if (pages_mapped + pages_iter >= pages_to_map) { | |
1119 | if (flags & MLX5_IB_UPD_XLT_ENABLE) | |
1120 | wr.wr.send_flags |= | |
1121 | MLX5_IB_SEND_UMR_ENABLE_MR | | |
1122 | MLX5_IB_SEND_UMR_UPDATE_PD_ACCESS | | |
1123 | MLX5_IB_SEND_UMR_UPDATE_TRANSLATION; | |
1124 | if (flags & MLX5_IB_UPD_XLT_PD || | |
1125 | flags & MLX5_IB_UPD_XLT_ACCESS) | |
1126 | wr.wr.send_flags |= | |
1127 | MLX5_IB_SEND_UMR_UPDATE_PD_ACCESS; | |
1128 | if (flags & MLX5_IB_UPD_XLT_ADDR) | |
1129 | wr.wr.send_flags |= | |
1130 | MLX5_IB_SEND_UMR_UPDATE_TRANSLATION; | |
1131 | } | |
832a6b06 | 1132 | |
7d0cc6ed | 1133 | wr.offset = idx * desc_size; |
31616255 | 1134 | wr.xlt_size = sg.length; |
832a6b06 | 1135 | |
d5ea2df9 | 1136 | err = mlx5_ib_post_send_wait(dev, &wr); |
832a6b06 HE |
1137 | } |
1138 | dma_unmap_single(ddev, dma, size, DMA_TO_DEVICE); | |
1139 | ||
7d0cc6ed | 1140 | free_xlt: |
c44ef998 IL |
1141 | if (use_emergency_page) |
1142 | mlx5_ib_put_xlt_emergency_page(); | |
832a6b06 | 1143 | else |
7d0cc6ed | 1144 | free_pages((unsigned long)xlt, get_order(size)); |
832a6b06 HE |
1145 | |
1146 | return err; | |
1147 | } | |
832a6b06 | 1148 | |
395a8e4c NO |
1149 | /* |
1150 | * If ibmr is NULL it will be allocated by reg_create. | |
1151 | * Else, the given ibmr will be used. | |
1152 | */ | |
1153 | static struct mlx5_ib_mr *reg_create(struct ib_mr *ibmr, struct ib_pd *pd, | |
1154 | u64 virt_addr, u64 length, | |
1155 | struct ib_umem *umem, int npages, | |
ff740aef IL |
1156 | int page_shift, int access_flags, |
1157 | bool populate) | |
e126ba97 EC |
1158 | { |
1159 | struct mlx5_ib_dev *dev = to_mdev(pd->device); | |
e126ba97 | 1160 | struct mlx5_ib_mr *mr; |
ec22eb53 SM |
1161 | __be64 *pas; |
1162 | void *mkc; | |
e126ba97 | 1163 | int inlen; |
ec22eb53 | 1164 | u32 *in; |
e126ba97 | 1165 | int err; |
938fe83c | 1166 | bool pg_cap = !!(MLX5_CAP_GEN(dev->mdev, pg)); |
e126ba97 | 1167 | |
395a8e4c | 1168 | mr = ibmr ? to_mmr(ibmr) : kzalloc(sizeof(*mr), GFP_KERNEL); |
e126ba97 EC |
1169 | if (!mr) |
1170 | return ERR_PTR(-ENOMEM); | |
1171 | ||
ff740aef IL |
1172 | mr->ibmr.pd = pd; |
1173 | mr->access_flags = access_flags; | |
1174 | ||
1175 | inlen = MLX5_ST_SZ_BYTES(create_mkey_in); | |
1176 | if (populate) | |
1177 | inlen += sizeof(*pas) * roundup(npages, 2); | |
1b9a07ee | 1178 | in = kvzalloc(inlen, GFP_KERNEL); |
e126ba97 EC |
1179 | if (!in) { |
1180 | err = -ENOMEM; | |
1181 | goto err_1; | |
1182 | } | |
ec22eb53 | 1183 | pas = (__be64 *)MLX5_ADDR_OF(create_mkey_in, in, klm_pas_mtt); |
ff740aef | 1184 | if (populate && !(access_flags & IB_ACCESS_ON_DEMAND)) |
c438fde1 AK |
1185 | mlx5_ib_populate_pas(dev, umem, page_shift, pas, |
1186 | pg_cap ? MLX5_IB_MTT_PRESENT : 0); | |
e126ba97 | 1187 | |
ec22eb53 | 1188 | /* The pg_access bit allows setting the access flags |
cc149f75 | 1189 | * in the page list submitted with the command. */ |
ec22eb53 SM |
1190 | MLX5_SET(create_mkey_in, in, pg_access, !!(pg_cap)); |
1191 | ||
1192 | mkc = MLX5_ADDR_OF(create_mkey_in, in, memory_key_mkey_entry); | |
1c97ca3d | 1193 | set_mkc_access_pd_addr_fields(mkc, access_flags, virt_addr, pd); |
ff740aef | 1194 | MLX5_SET(mkc, mkc, free, !populate); |
cdbd0d2b | 1195 | MLX5_SET(mkc, mkc, access_mode_1_0, MLX5_MKC_ACCESS_MODE_MTT); |
8b7ff7f3 | 1196 | MLX5_SET(mkc, mkc, umr_en, 1); |
ec22eb53 | 1197 | |
ec22eb53 | 1198 | MLX5_SET64(mkc, mkc, len, length); |
ec22eb53 SM |
1199 | MLX5_SET(mkc, mkc, bsf_octword_size, 0); |
1200 | MLX5_SET(mkc, mkc, translations_octword_size, | |
7b4cdaae | 1201 | get_octo_len(virt_addr, length, page_shift)); |
ec22eb53 | 1202 | MLX5_SET(mkc, mkc, log_page_size, page_shift); |
ff740aef IL |
1203 | if (populate) { |
1204 | MLX5_SET(create_mkey_in, in, translations_octword_actual_size, | |
7b4cdaae | 1205 | get_octo_len(virt_addr, length, page_shift)); |
ff740aef | 1206 | } |
ec22eb53 | 1207 | |
fc6a9f86 | 1208 | err = mlx5_ib_create_mkey(dev, &mr->mmkey, in, inlen); |
e126ba97 EC |
1209 | if (err) { |
1210 | mlx5_ib_warn(dev, "create mkey failed\n"); | |
1211 | goto err_2; | |
1212 | } | |
aa8e08d2 | 1213 | mr->mmkey.type = MLX5_MKEY_MR; |
49780d42 | 1214 | mr->desc_size = sizeof(struct mlx5_mtt); |
7eae20db | 1215 | mr->dev = dev; |
479163f4 | 1216 | kvfree(in); |
e126ba97 | 1217 | |
a606b0f6 | 1218 | mlx5_ib_dbg(dev, "mkey = 0x%x\n", mr->mmkey.key); |
e126ba97 EC |
1219 | |
1220 | return mr; | |
1221 | ||
1222 | err_2: | |
479163f4 | 1223 | kvfree(in); |
e126ba97 EC |
1224 | |
1225 | err_1: | |
395a8e4c NO |
1226 | if (!ibmr) |
1227 | kfree(mr); | |
e126ba97 EC |
1228 | |
1229 | return ERR_PTR(err); | |
1230 | } | |
1231 | ||
ac2f7e62 | 1232 | static void set_mr_fields(struct mlx5_ib_dev *dev, struct mlx5_ib_mr *mr, |
395a8e4c NO |
1233 | int npages, u64 length, int access_flags) |
1234 | { | |
1235 | mr->npages = npages; | |
1236 | atomic_add(npages, &dev->mdev->priv.reg_pages); | |
a606b0f6 MB |
1237 | mr->ibmr.lkey = mr->mmkey.key; |
1238 | mr->ibmr.rkey = mr->mmkey.key; | |
395a8e4c | 1239 | mr->ibmr.length = length; |
56e11d62 | 1240 | mr->access_flags = access_flags; |
395a8e4c NO |
1241 | } |
1242 | ||
3b113a1e AL |
1243 | static struct ib_mr *mlx5_ib_get_dm_mr(struct ib_pd *pd, u64 start_addr, |
1244 | u64 length, int acc, int mode) | |
6c29f57e AL |
1245 | { |
1246 | struct mlx5_ib_dev *dev = to_mdev(pd->device); | |
1247 | int inlen = MLX5_ST_SZ_BYTES(create_mkey_in); | |
6c29f57e AL |
1248 | struct mlx5_ib_mr *mr; |
1249 | void *mkc; | |
1250 | u32 *in; | |
1251 | int err; | |
1252 | ||
1253 | mr = kzalloc(sizeof(*mr), GFP_KERNEL); | |
1254 | if (!mr) | |
1255 | return ERR_PTR(-ENOMEM); | |
1256 | ||
1257 | in = kzalloc(inlen, GFP_KERNEL); | |
1258 | if (!in) { | |
1259 | err = -ENOMEM; | |
1260 | goto err_free; | |
1261 | } | |
1262 | ||
1263 | mkc = MLX5_ADDR_OF(create_mkey_in, in, memory_key_mkey_entry); | |
1264 | ||
3b113a1e AL |
1265 | MLX5_SET(mkc, mkc, access_mode_1_0, mode & 0x3); |
1266 | MLX5_SET(mkc, mkc, access_mode_4_2, (mode >> 2) & 0x7); | |
6c29f57e | 1267 | MLX5_SET64(mkc, mkc, len, length); |
03232cc4 | 1268 | set_mkc_access_pd_addr_fields(mkc, acc, start_addr, pd); |
6c29f57e | 1269 | |
fc6a9f86 | 1270 | err = mlx5_ib_create_mkey(dev, &mr->mmkey, in, inlen); |
6c29f57e AL |
1271 | if (err) |
1272 | goto err_in; | |
1273 | ||
1274 | kfree(in); | |
1275 | ||
1276 | mr->umem = NULL; | |
ac2f7e62 | 1277 | set_mr_fields(dev, mr, 0, length, acc); |
6c29f57e AL |
1278 | |
1279 | return &mr->ibmr; | |
1280 | ||
1281 | err_in: | |
1282 | kfree(in); | |
1283 | ||
1284 | err_free: | |
1285 | kfree(mr); | |
1286 | ||
1287 | return ERR_PTR(err); | |
1288 | } | |
1289 | ||
813e90b1 MS |
1290 | int mlx5_ib_advise_mr(struct ib_pd *pd, |
1291 | enum ib_uverbs_advise_mr_advice advice, | |
1292 | u32 flags, | |
1293 | struct ib_sge *sg_list, | |
1294 | u32 num_sge, | |
1295 | struct uverbs_attr_bundle *attrs) | |
1296 | { | |
1297 | if (advice != IB_UVERBS_ADVISE_MR_ADVICE_PREFETCH && | |
1298 | advice != IB_UVERBS_ADVISE_MR_ADVICE_PREFETCH_WRITE) | |
1299 | return -EOPNOTSUPP; | |
1300 | ||
1301 | return mlx5_ib_advise_mr_prefetch(pd, advice, flags, | |
1302 | sg_list, num_sge); | |
1303 | } | |
1304 | ||
6c29f57e AL |
1305 | struct ib_mr *mlx5_ib_reg_dm_mr(struct ib_pd *pd, struct ib_dm *dm, |
1306 | struct ib_dm_mr_attr *attr, | |
1307 | struct uverbs_attr_bundle *attrs) | |
1308 | { | |
1309 | struct mlx5_ib_dm *mdm = to_mdm(dm); | |
3b113a1e AL |
1310 | struct mlx5_core_dev *dev = to_mdev(dm->device)->mdev; |
1311 | u64 start_addr = mdm->dev_addr + attr->offset; | |
1312 | int mode; | |
1313 | ||
1314 | switch (mdm->type) { | |
1315 | case MLX5_IB_UAPI_DM_TYPE_MEMIC: | |
1316 | if (attr->access_flags & ~MLX5_IB_DM_MEMIC_ALLOWED_ACCESS) | |
1317 | return ERR_PTR(-EINVAL); | |
1318 | ||
1319 | mode = MLX5_MKC_ACCESS_MODE_MEMIC; | |
1320 | start_addr -= pci_resource_start(dev->pdev, 0); | |
1321 | break; | |
25c13324 AL |
1322 | case MLX5_IB_UAPI_DM_TYPE_STEERING_SW_ICM: |
1323 | case MLX5_IB_UAPI_DM_TYPE_HEADER_MODIFY_SW_ICM: | |
1324 | if (attr->access_flags & ~MLX5_IB_DM_SW_ICM_ALLOWED_ACCESS) | |
1325 | return ERR_PTR(-EINVAL); | |
1326 | ||
1327 | mode = MLX5_MKC_ACCESS_MODE_SW_ICM; | |
1328 | break; | |
3b113a1e | 1329 | default: |
6c29f57e | 1330 | return ERR_PTR(-EINVAL); |
3b113a1e | 1331 | } |
6c29f57e | 1332 | |
3b113a1e AL |
1333 | return mlx5_ib_get_dm_mr(pd, start_addr, attr->length, |
1334 | attr->access_flags, mode); | |
6c29f57e AL |
1335 | } |
1336 | ||
e126ba97 EC |
1337 | struct ib_mr *mlx5_ib_reg_user_mr(struct ib_pd *pd, u64 start, u64 length, |
1338 | u64 virt_addr, int access_flags, | |
1339 | struct ib_udata *udata) | |
1340 | { | |
1341 | struct mlx5_ib_dev *dev = to_mdev(pd->device); | |
1342 | struct mlx5_ib_mr *mr = NULL; | |
e5366d30 | 1343 | bool use_umr; |
e126ba97 EC |
1344 | struct ib_umem *umem; |
1345 | int page_shift; | |
1346 | int npages; | |
1347 | int ncont; | |
1348 | int order; | |
1349 | int err; | |
1350 | ||
1b19b951 | 1351 | if (!IS_ENABLED(CONFIG_INFINIBAND_USER_MEM)) |
ea30f013 | 1352 | return ERR_PTR(-EOPNOTSUPP); |
1b19b951 | 1353 | |
900a6d79 EC |
1354 | mlx5_ib_dbg(dev, "start 0x%llx, virt_addr 0x%llx, length 0x%llx, access_flags 0x%x\n", |
1355 | start, virt_addr, length, access_flags); | |
81713d37 | 1356 | |
13859d5d LR |
1357 | if (IS_ENABLED(CONFIG_INFINIBAND_ON_DEMAND_PAGING) && !start && |
1358 | length == U64_MAX) { | |
8ffc3248 JG |
1359 | if (virt_addr != start) |
1360 | return ERR_PTR(-EINVAL); | |
81713d37 AK |
1361 | if (!(access_flags & IB_ACCESS_ON_DEMAND) || |
1362 | !(dev->odp_caps.general_caps & IB_ODP_SUPPORT_IMPLICIT)) | |
1363 | return ERR_PTR(-EINVAL); | |
1364 | ||
b0ea0fa5 | 1365 | mr = mlx5_ib_alloc_implicit_mr(to_mpd(pd), udata, access_flags); |
4289861d LR |
1366 | if (IS_ERR(mr)) |
1367 | return ERR_CAST(mr); | |
81713d37 AK |
1368 | return &mr->ibmr; |
1369 | } | |
81713d37 | 1370 | |
c320e527 | 1371 | err = mr_umem_get(dev, start, length, access_flags, &umem, |
b0ea0fa5 | 1372 | &npages, &page_shift, &ncont, &order); |
e126ba97 | 1373 | |
ff740aef | 1374 | if (err < 0) |
14ab8896 | 1375 | return ERR_PTR(err); |
e126ba97 | 1376 | |
d6de0bb1 | 1377 | use_umr = mlx5_ib_can_use_umr(dev, true, access_flags); |
e5366d30 GL |
1378 | |
1379 | if (order <= mr_cache_max_order(dev) && use_umr) { | |
ff740aef IL |
1380 | mr = alloc_mr_from_cache(pd, umem, virt_addr, length, ncont, |
1381 | page_shift, order, access_flags); | |
2e4e706e | 1382 | if (IS_ERR(mr)) |
e126ba97 | 1383 | mr = NULL; |
ff740aef IL |
1384 | } else if (!MLX5_CAP_GEN(dev->mdev, umr_extended_translation_offset)) { |
1385 | if (access_flags & IB_ACCESS_ON_DEMAND) { | |
1386 | err = -EINVAL; | |
d23a8baf | 1387 | pr_err("Got MR registration for ODP MR > 512MB, not supported for Connect-IB\n"); |
ff740aef IL |
1388 | goto error; |
1389 | } | |
e5366d30 | 1390 | use_umr = false; |
e126ba97 EC |
1391 | } |
1392 | ||
6bc1a656 ML |
1393 | if (!mr) { |
1394 | mutex_lock(&dev->slow_path_mutex); | |
395a8e4c | 1395 | mr = reg_create(NULL, pd, virt_addr, length, umem, ncont, |
e5366d30 | 1396 | page_shift, access_flags, !use_umr); |
6bc1a656 ML |
1397 | mutex_unlock(&dev->slow_path_mutex); |
1398 | } | |
e126ba97 EC |
1399 | |
1400 | if (IS_ERR(mr)) { | |
1401 | err = PTR_ERR(mr); | |
1402 | goto error; | |
1403 | } | |
1404 | ||
a606b0f6 | 1405 | mlx5_ib_dbg(dev, "mkey 0x%x\n", mr->mmkey.key); |
e126ba97 EC |
1406 | |
1407 | mr->umem = umem; | |
ac2f7e62 | 1408 | set_mr_fields(dev, mr, npages, length, access_flags); |
e126ba97 | 1409 | |
e5366d30 | 1410 | if (use_umr) { |
ff740aef IL |
1411 | int update_xlt_flags = MLX5_IB_UPD_XLT_ENABLE; |
1412 | ||
1413 | if (access_flags & IB_ACCESS_ON_DEMAND) | |
1414 | update_xlt_flags |= MLX5_IB_UPD_XLT_ZAP; | |
e126ba97 | 1415 | |
ff740aef IL |
1416 | err = mlx5_ib_update_xlt(mr, 0, ncont, page_shift, |
1417 | update_xlt_flags); | |
fbcd4983 | 1418 | |
ff740aef | 1419 | if (err) { |
fbcd4983 | 1420 | dereg_mr(dev, mr); |
ff740aef IL |
1421 | return ERR_PTR(err); |
1422 | } | |
1423 | } | |
1424 | ||
aa603815 JG |
1425 | if (is_odp_mr(mr)) { |
1426 | to_ib_umem_odp(mr->umem)->private = mr; | |
189277f3 | 1427 | init_waitqueue_head(&mr->q_deferred_work); |
5256edcb | 1428 | atomic_set(&mr->num_deferred_work, 0); |
806b101b JG |
1429 | err = xa_err(xa_store(&dev->odp_mkeys, |
1430 | mlx5_base_mkey(mr->mmkey.key), &mr->mmkey, | |
1431 | GFP_KERNEL)); | |
1432 | if (err) { | |
1433 | dereg_mr(dev, mr); | |
1434 | return ERR_PTR(err); | |
1435 | } | |
a6bc3875 | 1436 | } |
13859d5d | 1437 | |
ff740aef | 1438 | return &mr->ibmr; |
e126ba97 EC |
1439 | error: |
1440 | ib_umem_release(umem); | |
1441 | return ERR_PTR(err); | |
1442 | } | |
1443 | ||
09689703 JG |
1444 | /** |
1445 | * mlx5_mr_cache_invalidate - Fence all DMA on the MR | |
1446 | * @mr: The MR to fence | |
1447 | * | |
1448 | * Upon return the NIC will not be doing any DMA to the pages under the MR, | |
1449 | * and any DMA inprogress will be completed. Failure of this function | |
1450 | * indicates the HW has failed catastrophically. | |
1451 | */ | |
1452 | int mlx5_mr_cache_invalidate(struct mlx5_ib_mr *mr) | |
e126ba97 | 1453 | { |
0025b0bd | 1454 | struct mlx5_umr_wr umrwr = {}; |
e126ba97 | 1455 | |
09689703 | 1456 | if (mr->dev->mdev->state == MLX5_DEVICE_STATE_INTERNAL_ERROR) |
89ea94a7 MG |
1457 | return 0; |
1458 | ||
9ec4483a YH |
1459 | umrwr.wr.send_flags = MLX5_IB_SEND_UMR_DISABLE_MR | |
1460 | MLX5_IB_SEND_UMR_UPDATE_PD_ACCESS; | |
7d0cc6ed | 1461 | umrwr.wr.opcode = MLX5_IB_WR_UMR; |
09689703 | 1462 | umrwr.pd = mr->dev->umrc.pd; |
7d0cc6ed | 1463 | umrwr.mkey = mr->mmkey.key; |
6a053953 | 1464 | umrwr.ignore_free_state = 1; |
e126ba97 | 1465 | |
09689703 | 1466 | return mlx5_ib_post_send_wait(mr->dev, &umrwr); |
e126ba97 EC |
1467 | } |
1468 | ||
7d0cc6ed | 1469 | static int rereg_umr(struct ib_pd *pd, struct mlx5_ib_mr *mr, |
56e11d62 NO |
1470 | int access_flags, int flags) |
1471 | { | |
1472 | struct mlx5_ib_dev *dev = to_mdev(pd->device); | |
56e11d62 | 1473 | struct mlx5_umr_wr umrwr = {}; |
56e11d62 NO |
1474 | int err; |
1475 | ||
56e11d62 NO |
1476 | umrwr.wr.send_flags = MLX5_IB_SEND_UMR_FAIL_IF_FREE; |
1477 | ||
7d0cc6ed AK |
1478 | umrwr.wr.opcode = MLX5_IB_WR_UMR; |
1479 | umrwr.mkey = mr->mmkey.key; | |
56e11d62 | 1480 | |
31616255 | 1481 | if (flags & IB_MR_REREG_PD || flags & IB_MR_REREG_ACCESS) { |
56e11d62 | 1482 | umrwr.pd = pd; |
56e11d62 | 1483 | umrwr.access_flags = access_flags; |
31616255 | 1484 | umrwr.wr.send_flags |= MLX5_IB_SEND_UMR_UPDATE_PD_ACCESS; |
56e11d62 NO |
1485 | } |
1486 | ||
d5ea2df9 | 1487 | err = mlx5_ib_post_send_wait(dev, &umrwr); |
56e11d62 | 1488 | |
56e11d62 NO |
1489 | return err; |
1490 | } | |
1491 | ||
1492 | int mlx5_ib_rereg_user_mr(struct ib_mr *ib_mr, int flags, u64 start, | |
1493 | u64 length, u64 virt_addr, int new_access_flags, | |
1494 | struct ib_pd *new_pd, struct ib_udata *udata) | |
1495 | { | |
1496 | struct mlx5_ib_dev *dev = to_mdev(ib_mr->device); | |
1497 | struct mlx5_ib_mr *mr = to_mmr(ib_mr); | |
1498 | struct ib_pd *pd = (flags & IB_MR_REREG_PD) ? new_pd : ib_mr->pd; | |
1499 | int access_flags = flags & IB_MR_REREG_ACCESS ? | |
1500 | new_access_flags : | |
1501 | mr->access_flags; | |
56e11d62 | 1502 | int page_shift = 0; |
7d0cc6ed | 1503 | int upd_flags = 0; |
56e11d62 NO |
1504 | int npages = 0; |
1505 | int ncont = 0; | |
1506 | int order = 0; | |
b4bd701a | 1507 | u64 addr, len; |
56e11d62 NO |
1508 | int err; |
1509 | ||
1510 | mlx5_ib_dbg(dev, "start 0x%llx, virt_addr 0x%llx, length 0x%llx, access_flags 0x%x\n", | |
1511 | start, virt_addr, length, access_flags); | |
1512 | ||
7d0cc6ed AK |
1513 | atomic_sub(mr->npages, &dev->mdev->priv.reg_pages); |
1514 | ||
b4bd701a LR |
1515 | if (!mr->umem) |
1516 | return -EINVAL; | |
1517 | ||
880505cf JG |
1518 | if (is_odp_mr(mr)) |
1519 | return -EOPNOTSUPP; | |
1520 | ||
b4bd701a LR |
1521 | if (flags & IB_MR_REREG_TRANS) { |
1522 | addr = virt_addr; | |
1523 | len = length; | |
1524 | } else { | |
1525 | addr = mr->umem->address; | |
1526 | len = mr->umem->length; | |
1527 | } | |
1528 | ||
56e11d62 NO |
1529 | if (flags != IB_MR_REREG_PD) { |
1530 | /* | |
1531 | * Replace umem. This needs to be done whether or not UMR is | |
1532 | * used. | |
1533 | */ | |
1534 | flags |= IB_MR_REREG_TRANS; | |
1535 | ib_umem_release(mr->umem); | |
b4bd701a | 1536 | mr->umem = NULL; |
c320e527 MS |
1537 | err = mr_umem_get(dev, addr, len, access_flags, &mr->umem, |
1538 | &npages, &page_shift, &ncont, &order); | |
4638a3b2 LR |
1539 | if (err) |
1540 | goto err; | |
56e11d62 NO |
1541 | } |
1542 | ||
d6de0bb1 | 1543 | if (!mlx5_ib_can_use_umr(dev, true, access_flags) || |
25a45172 | 1544 | (flags & IB_MR_REREG_TRANS && !use_umr_mtt_update(mr, addr, len))) { |
56e11d62 NO |
1545 | /* |
1546 | * UMR can't be used - MKey needs to be replaced. | |
1547 | */ | |
b91e1751 | 1548 | if (mr->cache_ent) |
1769c4c5 JG |
1549 | detach_mr_from_cache(mr); |
1550 | err = destroy_mkey(dev, mr); | |
56e11d62 | 1551 | if (err) |
4638a3b2 | 1552 | goto err; |
56e11d62 NO |
1553 | |
1554 | mr = reg_create(ib_mr, pd, addr, len, mr->umem, ncont, | |
ff740aef | 1555 | page_shift, access_flags, true); |
56e11d62 | 1556 | |
4638a3b2 LR |
1557 | if (IS_ERR(mr)) { |
1558 | err = PTR_ERR(mr); | |
1559 | mr = to_mmr(ib_mr); | |
1560 | goto err; | |
1561 | } | |
56e11d62 NO |
1562 | } else { |
1563 | /* | |
1564 | * Send a UMR WQE | |
1565 | */ | |
7d0cc6ed AK |
1566 | mr->ibmr.pd = pd; |
1567 | mr->access_flags = access_flags; | |
1568 | mr->mmkey.iova = addr; | |
1569 | mr->mmkey.size = len; | |
1570 | mr->mmkey.pd = to_mpd(pd)->pdn; | |
1571 | ||
1572 | if (flags & IB_MR_REREG_TRANS) { | |
1573 | upd_flags = MLX5_IB_UPD_XLT_ADDR; | |
1574 | if (flags & IB_MR_REREG_PD) | |
1575 | upd_flags |= MLX5_IB_UPD_XLT_PD; | |
1576 | if (flags & IB_MR_REREG_ACCESS) | |
1577 | upd_flags |= MLX5_IB_UPD_XLT_ACCESS; | |
1578 | err = mlx5_ib_update_xlt(mr, 0, npages, page_shift, | |
1579 | upd_flags); | |
1580 | } else { | |
1581 | err = rereg_umr(pd, mr, access_flags, flags); | |
1582 | } | |
1583 | ||
4638a3b2 LR |
1584 | if (err) |
1585 | goto err; | |
56e11d62 NO |
1586 | } |
1587 | ||
ac2f7e62 | 1588 | set_mr_fields(dev, mr, npages, len, access_flags); |
56e11d62 | 1589 | |
56e11d62 | 1590 | return 0; |
4638a3b2 LR |
1591 | |
1592 | err: | |
836a0fbb LR |
1593 | ib_umem_release(mr->umem); |
1594 | mr->umem = NULL; | |
1595 | ||
4638a3b2 LR |
1596 | clean_mr(dev, mr); |
1597 | return err; | |
56e11d62 NO |
1598 | } |
1599 | ||
8a187ee5 SG |
1600 | static int |
1601 | mlx5_alloc_priv_descs(struct ib_device *device, | |
1602 | struct mlx5_ib_mr *mr, | |
1603 | int ndescs, | |
1604 | int desc_size) | |
1605 | { | |
1606 | int size = ndescs * desc_size; | |
1607 | int add_size; | |
1608 | int ret; | |
1609 | ||
1610 | add_size = max_t(int, MLX5_UMR_ALIGN - ARCH_KMALLOC_MINALIGN, 0); | |
1611 | ||
1612 | mr->descs_alloc = kzalloc(size + add_size, GFP_KERNEL); | |
1613 | if (!mr->descs_alloc) | |
1614 | return -ENOMEM; | |
1615 | ||
1616 | mr->descs = PTR_ALIGN(mr->descs_alloc, MLX5_UMR_ALIGN); | |
1617 | ||
9b0c289e | 1618 | mr->desc_map = dma_map_single(device->dev.parent, mr->descs, |
8a187ee5 | 1619 | size, DMA_TO_DEVICE); |
9b0c289e | 1620 | if (dma_mapping_error(device->dev.parent, mr->desc_map)) { |
8a187ee5 SG |
1621 | ret = -ENOMEM; |
1622 | goto err; | |
1623 | } | |
1624 | ||
1625 | return 0; | |
1626 | err: | |
1627 | kfree(mr->descs_alloc); | |
1628 | ||
1629 | return ret; | |
1630 | } | |
1631 | ||
1632 | static void | |
1633 | mlx5_free_priv_descs(struct mlx5_ib_mr *mr) | |
1634 | { | |
1635 | if (mr->descs) { | |
1636 | struct ib_device *device = mr->ibmr.device; | |
1637 | int size = mr->max_descs * mr->desc_size; | |
1638 | ||
9b0c289e | 1639 | dma_unmap_single(device->dev.parent, mr->desc_map, |
8a187ee5 SG |
1640 | size, DMA_TO_DEVICE); |
1641 | kfree(mr->descs_alloc); | |
1642 | mr->descs = NULL; | |
1643 | } | |
1644 | } | |
1645 | ||
eeea6953 | 1646 | static void clean_mr(struct mlx5_ib_dev *dev, struct mlx5_ib_mr *mr) |
e126ba97 | 1647 | { |
8b91ffc1 SG |
1648 | if (mr->sig) { |
1649 | if (mlx5_core_destroy_psv(dev->mdev, | |
1650 | mr->sig->psv_memory.psv_idx)) | |
1651 | mlx5_ib_warn(dev, "failed to destroy mem psv %d\n", | |
1652 | mr->sig->psv_memory.psv_idx); | |
1653 | if (mlx5_core_destroy_psv(dev->mdev, | |
1654 | mr->sig->psv_wire.psv_idx)) | |
1655 | mlx5_ib_warn(dev, "failed to destroy wire psv %d\n", | |
1656 | mr->sig->psv_wire.psv_idx); | |
50211ec9 | 1657 | xa_erase(&dev->sig_mrs, mlx5_base_mkey(mr->mmkey.key)); |
8b91ffc1 SG |
1658 | kfree(mr->sig); |
1659 | mr->sig = NULL; | |
1660 | } | |
1661 | ||
b91e1751 | 1662 | if (!mr->cache_ent) { |
eeea6953 | 1663 | destroy_mkey(dev, mr); |
b9332dad YH |
1664 | mlx5_free_priv_descs(mr); |
1665 | } | |
6aec21f6 HE |
1666 | } |
1667 | ||
eeea6953 | 1668 | static void dereg_mr(struct mlx5_ib_dev *dev, struct mlx5_ib_mr *mr) |
6aec21f6 | 1669 | { |
6aec21f6 HE |
1670 | int npages = mr->npages; |
1671 | struct ib_umem *umem = mr->umem; | |
1672 | ||
09689703 JG |
1673 | /* Stop all DMA */ |
1674 | if (is_odp_mr(mr)) | |
1675 | mlx5_ib_fence_odp_mr(mr); | |
1676 | else | |
1677 | clean_mr(dev, mr); | |
8b4d5bc5 | 1678 | |
b91e1751 | 1679 | if (mr->cache_ent) |
09689703 JG |
1680 | mlx5_mr_cache_free(dev, mr); |
1681 | else | |
1682 | kfree(mr); | |
6aec21f6 | 1683 | |
836a0fbb | 1684 | ib_umem_release(umem); |
09689703 | 1685 | atomic_sub(npages, &dev->mdev->priv.reg_pages); |
836a0fbb | 1686 | |
e126ba97 EC |
1687 | } |
1688 | ||
c4367a26 | 1689 | int mlx5_ib_dereg_mr(struct ib_mr *ibmr, struct ib_udata *udata) |
fbcd4983 | 1690 | { |
6c984472 MG |
1691 | struct mlx5_ib_mr *mmr = to_mmr(ibmr); |
1692 | ||
de0ae958 IR |
1693 | if (ibmr->type == IB_MR_TYPE_INTEGRITY) { |
1694 | dereg_mr(to_mdev(mmr->mtt_mr->ibmr.device), mmr->mtt_mr); | |
1695 | dereg_mr(to_mdev(mmr->klm_mr->ibmr.device), mmr->klm_mr); | |
1696 | } | |
6c984472 | 1697 | |
5256edcb JG |
1698 | if (is_odp_mr(mmr) && to_ib_umem_odp(mmr->umem)->is_implicit_odp) { |
1699 | mlx5_ib_free_implicit_mr(mmr); | |
1700 | return 0; | |
1701 | } | |
1702 | ||
6c984472 MG |
1703 | dereg_mr(to_mdev(ibmr->device), mmr); |
1704 | ||
eeea6953 | 1705 | return 0; |
fbcd4983 IL |
1706 | } |
1707 | ||
7796d2a3 MG |
1708 | static void mlx5_set_umr_free_mkey(struct ib_pd *pd, u32 *in, int ndescs, |
1709 | int access_mode, int page_shift) | |
1710 | { | |
1711 | void *mkc; | |
1712 | ||
1713 | mkc = MLX5_ADDR_OF(create_mkey_in, in, memory_key_mkey_entry); | |
1714 | ||
1715 | MLX5_SET(mkc, mkc, free, 1); | |
1716 | MLX5_SET(mkc, mkc, qpn, 0xffffff); | |
1717 | MLX5_SET(mkc, mkc, pd, to_mpd(pd)->pdn); | |
1718 | MLX5_SET(mkc, mkc, translations_octword_size, ndescs); | |
1719 | MLX5_SET(mkc, mkc, access_mode_1_0, access_mode & 0x3); | |
1720 | MLX5_SET(mkc, mkc, access_mode_4_2, (access_mode >> 2) & 0x7); | |
1721 | MLX5_SET(mkc, mkc, umr_en, 1); | |
1722 | MLX5_SET(mkc, mkc, log_page_size, page_shift); | |
1723 | } | |
1724 | ||
1725 | static int _mlx5_alloc_mkey_descs(struct ib_pd *pd, struct mlx5_ib_mr *mr, | |
1726 | int ndescs, int desc_size, int page_shift, | |
1727 | int access_mode, u32 *in, int inlen) | |
1728 | { | |
1729 | struct mlx5_ib_dev *dev = to_mdev(pd->device); | |
1730 | int err; | |
1731 | ||
1732 | mr->access_mode = access_mode; | |
1733 | mr->desc_size = desc_size; | |
1734 | mr->max_descs = ndescs; | |
1735 | ||
1736 | err = mlx5_alloc_priv_descs(pd->device, mr, ndescs, desc_size); | |
1737 | if (err) | |
1738 | return err; | |
1739 | ||
1740 | mlx5_set_umr_free_mkey(pd, in, ndescs, access_mode, page_shift); | |
1741 | ||
fc6a9f86 | 1742 | err = mlx5_ib_create_mkey(dev, &mr->mmkey, in, inlen); |
7796d2a3 MG |
1743 | if (err) |
1744 | goto err_free_descs; | |
1745 | ||
1746 | mr->mmkey.type = MLX5_MKEY_MR; | |
1747 | mr->ibmr.lkey = mr->mmkey.key; | |
1748 | mr->ibmr.rkey = mr->mmkey.key; | |
1749 | ||
1750 | return 0; | |
1751 | ||
1752 | err_free_descs: | |
1753 | mlx5_free_priv_descs(mr); | |
1754 | return err; | |
1755 | } | |
1756 | ||
6c984472 | 1757 | static struct mlx5_ib_mr *mlx5_ib_alloc_pi_mr(struct ib_pd *pd, |
de0ae958 IR |
1758 | u32 max_num_sg, u32 max_num_meta_sg, |
1759 | int desc_size, int access_mode) | |
3121e3c4 | 1760 | { |
ec22eb53 | 1761 | int inlen = MLX5_ST_SZ_BYTES(create_mkey_in); |
6c984472 | 1762 | int ndescs = ALIGN(max_num_sg + max_num_meta_sg, 4); |
7796d2a3 | 1763 | int page_shift = 0; |
ec22eb53 | 1764 | struct mlx5_ib_mr *mr; |
ec22eb53 | 1765 | u32 *in; |
b005d316 | 1766 | int err; |
3121e3c4 SG |
1767 | |
1768 | mr = kzalloc(sizeof(*mr), GFP_KERNEL); | |
1769 | if (!mr) | |
1770 | return ERR_PTR(-ENOMEM); | |
1771 | ||
7796d2a3 MG |
1772 | mr->ibmr.pd = pd; |
1773 | mr->ibmr.device = pd->device; | |
1774 | ||
ec22eb53 | 1775 | in = kzalloc(inlen, GFP_KERNEL); |
3121e3c4 SG |
1776 | if (!in) { |
1777 | err = -ENOMEM; | |
1778 | goto err_free; | |
1779 | } | |
1780 | ||
de0ae958 | 1781 | if (access_mode == MLX5_MKC_ACCESS_MODE_MTT) |
7796d2a3 | 1782 | page_shift = PAGE_SHIFT; |
3121e3c4 | 1783 | |
7796d2a3 MG |
1784 | err = _mlx5_alloc_mkey_descs(pd, mr, ndescs, desc_size, page_shift, |
1785 | access_mode, in, inlen); | |
6c984472 MG |
1786 | if (err) |
1787 | goto err_free_in; | |
6c984472 | 1788 | |
6c984472 MG |
1789 | mr->umem = NULL; |
1790 | kfree(in); | |
1791 | ||
1792 | return mr; | |
1793 | ||
6c984472 MG |
1794 | err_free_in: |
1795 | kfree(in); | |
1796 | err_free: | |
1797 | kfree(mr); | |
1798 | return ERR_PTR(err); | |
1799 | } | |
1800 | ||
7796d2a3 MG |
1801 | static int mlx5_alloc_mem_reg_descs(struct ib_pd *pd, struct mlx5_ib_mr *mr, |
1802 | int ndescs, u32 *in, int inlen) | |
1803 | { | |
1804 | return _mlx5_alloc_mkey_descs(pd, mr, ndescs, sizeof(struct mlx5_mtt), | |
1805 | PAGE_SHIFT, MLX5_MKC_ACCESS_MODE_MTT, in, | |
1806 | inlen); | |
1807 | } | |
1808 | ||
1809 | static int mlx5_alloc_sg_gaps_descs(struct ib_pd *pd, struct mlx5_ib_mr *mr, | |
1810 | int ndescs, u32 *in, int inlen) | |
1811 | { | |
1812 | return _mlx5_alloc_mkey_descs(pd, mr, ndescs, sizeof(struct mlx5_klm), | |
1813 | 0, MLX5_MKC_ACCESS_MODE_KLMS, in, inlen); | |
1814 | } | |
1815 | ||
1816 | static int mlx5_alloc_integrity_descs(struct ib_pd *pd, struct mlx5_ib_mr *mr, | |
1817 | int max_num_sg, int max_num_meta_sg, | |
1818 | u32 *in, int inlen) | |
1819 | { | |
1820 | struct mlx5_ib_dev *dev = to_mdev(pd->device); | |
1821 | u32 psv_index[2]; | |
1822 | void *mkc; | |
1823 | int err; | |
1824 | ||
1825 | mr->sig = kzalloc(sizeof(*mr->sig), GFP_KERNEL); | |
1826 | if (!mr->sig) | |
1827 | return -ENOMEM; | |
1828 | ||
1829 | /* create mem & wire PSVs */ | |
1830 | err = mlx5_core_create_psv(dev->mdev, to_mpd(pd)->pdn, 2, psv_index); | |
1831 | if (err) | |
1832 | goto err_free_sig; | |
1833 | ||
1834 | mr->sig->psv_memory.psv_idx = psv_index[0]; | |
1835 | mr->sig->psv_wire.psv_idx = psv_index[1]; | |
1836 | ||
1837 | mr->sig->sig_status_checked = true; | |
1838 | mr->sig->sig_err_exists = false; | |
1839 | /* Next UMR, Arm SIGERR */ | |
1840 | ++mr->sig->sigerr_count; | |
1841 | mr->klm_mr = mlx5_ib_alloc_pi_mr(pd, max_num_sg, max_num_meta_sg, | |
1842 | sizeof(struct mlx5_klm), | |
1843 | MLX5_MKC_ACCESS_MODE_KLMS); | |
1844 | if (IS_ERR(mr->klm_mr)) { | |
1845 | err = PTR_ERR(mr->klm_mr); | |
1846 | goto err_destroy_psv; | |
1847 | } | |
1848 | mr->mtt_mr = mlx5_ib_alloc_pi_mr(pd, max_num_sg, max_num_meta_sg, | |
1849 | sizeof(struct mlx5_mtt), | |
1850 | MLX5_MKC_ACCESS_MODE_MTT); | |
1851 | if (IS_ERR(mr->mtt_mr)) { | |
1852 | err = PTR_ERR(mr->mtt_mr); | |
1853 | goto err_free_klm_mr; | |
1854 | } | |
1855 | ||
1856 | /* Set bsf descriptors for mkey */ | |
1857 | mkc = MLX5_ADDR_OF(create_mkey_in, in, memory_key_mkey_entry); | |
1858 | MLX5_SET(mkc, mkc, bsf_en, 1); | |
1859 | MLX5_SET(mkc, mkc, bsf_octword_size, MLX5_MKEY_BSF_OCTO_SIZE); | |
1860 | ||
1861 | err = _mlx5_alloc_mkey_descs(pd, mr, 4, sizeof(struct mlx5_klm), 0, | |
1862 | MLX5_MKC_ACCESS_MODE_KLMS, in, inlen); | |
1863 | if (err) | |
1864 | goto err_free_mtt_mr; | |
1865 | ||
50211ec9 JG |
1866 | err = xa_err(xa_store(&dev->sig_mrs, mlx5_base_mkey(mr->mmkey.key), |
1867 | mr->sig, GFP_KERNEL)); | |
1868 | if (err) | |
1869 | goto err_free_descs; | |
7796d2a3 MG |
1870 | return 0; |
1871 | ||
50211ec9 JG |
1872 | err_free_descs: |
1873 | destroy_mkey(dev, mr); | |
1874 | mlx5_free_priv_descs(mr); | |
7796d2a3 MG |
1875 | err_free_mtt_mr: |
1876 | dereg_mr(to_mdev(mr->mtt_mr->ibmr.device), mr->mtt_mr); | |
1877 | mr->mtt_mr = NULL; | |
1878 | err_free_klm_mr: | |
1879 | dereg_mr(to_mdev(mr->klm_mr->ibmr.device), mr->klm_mr); | |
1880 | mr->klm_mr = NULL; | |
1881 | err_destroy_psv: | |
1882 | if (mlx5_core_destroy_psv(dev->mdev, mr->sig->psv_memory.psv_idx)) | |
1883 | mlx5_ib_warn(dev, "failed to destroy mem psv %d\n", | |
1884 | mr->sig->psv_memory.psv_idx); | |
1885 | if (mlx5_core_destroy_psv(dev->mdev, mr->sig->psv_wire.psv_idx)) | |
1886 | mlx5_ib_warn(dev, "failed to destroy wire psv %d\n", | |
1887 | mr->sig->psv_wire.psv_idx); | |
1888 | err_free_sig: | |
1889 | kfree(mr->sig); | |
1890 | ||
1891 | return err; | |
1892 | } | |
1893 | ||
6c984472 MG |
1894 | static struct ib_mr *__mlx5_ib_alloc_mr(struct ib_pd *pd, |
1895 | enum ib_mr_type mr_type, u32 max_num_sg, | |
1896 | u32 max_num_meta_sg) | |
1897 | { | |
1898 | struct mlx5_ib_dev *dev = to_mdev(pd->device); | |
1899 | int inlen = MLX5_ST_SZ_BYTES(create_mkey_in); | |
1900 | int ndescs = ALIGN(max_num_sg, 4); | |
1901 | struct mlx5_ib_mr *mr; | |
6c984472 MG |
1902 | u32 *in; |
1903 | int err; | |
1904 | ||
1905 | mr = kzalloc(sizeof(*mr), GFP_KERNEL); | |
1906 | if (!mr) | |
1907 | return ERR_PTR(-ENOMEM); | |
1908 | ||
1909 | in = kzalloc(inlen, GFP_KERNEL); | |
1910 | if (!in) { | |
1911 | err = -ENOMEM; | |
1912 | goto err_free; | |
1913 | } | |
1914 | ||
7796d2a3 MG |
1915 | mr->ibmr.device = pd->device; |
1916 | mr->umem = NULL; | |
3121e3c4 | 1917 | |
7796d2a3 MG |
1918 | switch (mr_type) { |
1919 | case IB_MR_TYPE_MEM_REG: | |
1920 | err = mlx5_alloc_mem_reg_descs(pd, mr, ndescs, in, inlen); | |
1921 | break; | |
1922 | case IB_MR_TYPE_SG_GAPS: | |
1923 | err = mlx5_alloc_sg_gaps_descs(pd, mr, ndescs, in, inlen); | |
1924 | break; | |
1925 | case IB_MR_TYPE_INTEGRITY: | |
1926 | err = mlx5_alloc_integrity_descs(pd, mr, max_num_sg, | |
1927 | max_num_meta_sg, in, inlen); | |
1928 | break; | |
1929 | default: | |
9bee178b SG |
1930 | mlx5_ib_warn(dev, "Invalid mr type %d\n", mr_type); |
1931 | err = -EINVAL; | |
3121e3c4 SG |
1932 | } |
1933 | ||
3121e3c4 | 1934 | if (err) |
7796d2a3 | 1935 | goto err_free_in; |
3121e3c4 | 1936 | |
3121e3c4 SG |
1937 | kfree(in); |
1938 | ||
1939 | return &mr->ibmr; | |
1940 | ||
3121e3c4 SG |
1941 | err_free_in: |
1942 | kfree(in); | |
1943 | err_free: | |
1944 | kfree(mr); | |
1945 | return ERR_PTR(err); | |
1946 | } | |
1947 | ||
6c984472 | 1948 | struct ib_mr *mlx5_ib_alloc_mr(struct ib_pd *pd, enum ib_mr_type mr_type, |
42a3b153 | 1949 | u32 max_num_sg) |
6c984472 MG |
1950 | { |
1951 | return __mlx5_ib_alloc_mr(pd, mr_type, max_num_sg, 0); | |
1952 | } | |
1953 | ||
1954 | struct ib_mr *mlx5_ib_alloc_mr_integrity(struct ib_pd *pd, | |
1955 | u32 max_num_sg, u32 max_num_meta_sg) | |
1956 | { | |
1957 | return __mlx5_ib_alloc_mr(pd, IB_MR_TYPE_INTEGRITY, max_num_sg, | |
1958 | max_num_meta_sg); | |
1959 | } | |
1960 | ||
d18bb3e1 | 1961 | int mlx5_ib_alloc_mw(struct ib_mw *ibmw, struct ib_udata *udata) |
d2370e0a | 1962 | { |
d18bb3e1 | 1963 | struct mlx5_ib_dev *dev = to_mdev(ibmw->device); |
ec22eb53 | 1964 | int inlen = MLX5_ST_SZ_BYTES(create_mkey_in); |
d18bb3e1 | 1965 | struct mlx5_ib_mw *mw = to_mmw(ibmw); |
ec22eb53 SM |
1966 | u32 *in = NULL; |
1967 | void *mkc; | |
d2370e0a MB |
1968 | int ndescs; |
1969 | int err; | |
1970 | struct mlx5_ib_alloc_mw req = {}; | |
1971 | struct { | |
1972 | __u32 comp_mask; | |
1973 | __u32 response_length; | |
1974 | } resp = {}; | |
1975 | ||
1976 | err = ib_copy_from_udata(&req, udata, min(udata->inlen, sizeof(req))); | |
1977 | if (err) | |
d18bb3e1 | 1978 | return err; |
d2370e0a MB |
1979 | |
1980 | if (req.comp_mask || req.reserved1 || req.reserved2) | |
d18bb3e1 | 1981 | return -EOPNOTSUPP; |
d2370e0a MB |
1982 | |
1983 | if (udata->inlen > sizeof(req) && | |
1984 | !ib_is_udata_cleared(udata, sizeof(req), | |
1985 | udata->inlen - sizeof(req))) | |
d18bb3e1 | 1986 | return -EOPNOTSUPP; |
d2370e0a MB |
1987 | |
1988 | ndescs = req.num_klms ? roundup(req.num_klms, 4) : roundup(1, 4); | |
1989 | ||
ec22eb53 | 1990 | in = kzalloc(inlen, GFP_KERNEL); |
d18bb3e1 | 1991 | if (!in) { |
d2370e0a MB |
1992 | err = -ENOMEM; |
1993 | goto free; | |
1994 | } | |
1995 | ||
ec22eb53 SM |
1996 | mkc = MLX5_ADDR_OF(create_mkey_in, in, memory_key_mkey_entry); |
1997 | ||
1998 | MLX5_SET(mkc, mkc, free, 1); | |
1999 | MLX5_SET(mkc, mkc, translations_octword_size, ndescs); | |
d18bb3e1 | 2000 | MLX5_SET(mkc, mkc, pd, to_mpd(ibmw->pd)->pdn); |
ec22eb53 SM |
2001 | MLX5_SET(mkc, mkc, umr_en, 1); |
2002 | MLX5_SET(mkc, mkc, lr, 1); | |
cdbd0d2b | 2003 | MLX5_SET(mkc, mkc, access_mode_1_0, MLX5_MKC_ACCESS_MODE_KLMS); |
d18bb3e1 | 2004 | MLX5_SET(mkc, mkc, en_rinval, !!((ibmw->type == IB_MW_TYPE_2))); |
ec22eb53 SM |
2005 | MLX5_SET(mkc, mkc, qpn, 0xffffff); |
2006 | ||
fc6a9f86 | 2007 | err = mlx5_ib_create_mkey(dev, &mw->mmkey, in, inlen); |
d2370e0a MB |
2008 | if (err) |
2009 | goto free; | |
2010 | ||
aa8e08d2 | 2011 | mw->mmkey.type = MLX5_MKEY_MW; |
d18bb3e1 | 2012 | ibmw->rkey = mw->mmkey.key; |
db570d7d | 2013 | mw->ndescs = ndescs; |
d2370e0a | 2014 | |
70c1430f LR |
2015 | resp.response_length = |
2016 | min(offsetofend(typeof(resp), response_length), udata->outlen); | |
d2370e0a MB |
2017 | if (resp.response_length) { |
2018 | err = ib_copy_to_udata(udata, &resp, resp.response_length); | |
d18bb3e1 LR |
2019 | if (err) |
2020 | goto free_mkey; | |
d2370e0a MB |
2021 | } |
2022 | ||
806b101b JG |
2023 | if (IS_ENABLED(CONFIG_INFINIBAND_ON_DEMAND_PAGING)) { |
2024 | err = xa_err(xa_store(&dev->odp_mkeys, | |
2025 | mlx5_base_mkey(mw->mmkey.key), &mw->mmkey, | |
2026 | GFP_KERNEL)); | |
2027 | if (err) | |
2028 | goto free_mkey; | |
2029 | } | |
2030 | ||
d2370e0a | 2031 | kfree(in); |
d18bb3e1 | 2032 | return 0; |
d2370e0a | 2033 | |
806b101b JG |
2034 | free_mkey: |
2035 | mlx5_core_destroy_mkey(dev->mdev, &mw->mmkey); | |
d2370e0a | 2036 | free: |
d2370e0a | 2037 | kfree(in); |
d18bb3e1 | 2038 | return err; |
d2370e0a MB |
2039 | } |
2040 | ||
2041 | int mlx5_ib_dealloc_mw(struct ib_mw *mw) | |
2042 | { | |
04177915 | 2043 | struct mlx5_ib_dev *dev = to_mdev(mw->device); |
d2370e0a | 2044 | struct mlx5_ib_mw *mmw = to_mmw(mw); |
d2370e0a | 2045 | |
04177915 | 2046 | if (IS_ENABLED(CONFIG_INFINIBAND_ON_DEMAND_PAGING)) { |
806b101b | 2047 | xa_erase(&dev->odp_mkeys, mlx5_base_mkey(mmw->mmkey.key)); |
04177915 JG |
2048 | /* |
2049 | * pagefault_single_data_segment() may be accessing mmw under | |
2050 | * SRCU if the user bound an ODP MR to this MW. | |
2051 | */ | |
806b101b | 2052 | synchronize_srcu(&dev->odp_srcu); |
04177915 JG |
2053 | } |
2054 | ||
d18bb3e1 | 2055 | return mlx5_core_destroy_mkey(dev->mdev, &mmw->mmkey); |
d2370e0a MB |
2056 | } |
2057 | ||
d5436ba0 SG |
2058 | int mlx5_ib_check_mr_status(struct ib_mr *ibmr, u32 check_mask, |
2059 | struct ib_mr_status *mr_status) | |
2060 | { | |
2061 | struct mlx5_ib_mr *mmr = to_mmr(ibmr); | |
2062 | int ret = 0; | |
2063 | ||
2064 | if (check_mask & ~IB_MR_CHECK_SIG_STATUS) { | |
2065 | pr_err("Invalid status check mask\n"); | |
2066 | ret = -EINVAL; | |
2067 | goto done; | |
2068 | } | |
2069 | ||
2070 | mr_status->fail_status = 0; | |
2071 | if (check_mask & IB_MR_CHECK_SIG_STATUS) { | |
2072 | if (!mmr->sig) { | |
2073 | ret = -EINVAL; | |
2074 | pr_err("signature status check requested on a non-signature enabled MR\n"); | |
2075 | goto done; | |
2076 | } | |
2077 | ||
2078 | mmr->sig->sig_status_checked = true; | |
2079 | if (!mmr->sig->sig_err_exists) | |
2080 | goto done; | |
2081 | ||
2082 | if (ibmr->lkey == mmr->sig->err_item.key) | |
2083 | memcpy(&mr_status->sig_err, &mmr->sig->err_item, | |
2084 | sizeof(mr_status->sig_err)); | |
2085 | else { | |
2086 | mr_status->sig_err.err_type = IB_SIG_BAD_GUARD; | |
2087 | mr_status->sig_err.sig_err_offset = 0; | |
2088 | mr_status->sig_err.key = mmr->sig->err_item.key; | |
2089 | } | |
2090 | ||
2091 | mmr->sig->sig_err_exists = false; | |
2092 | mr_status->fail_status |= IB_MR_CHECK_SIG_STATUS; | |
2093 | } | |
2094 | ||
2095 | done: | |
2096 | return ret; | |
2097 | } | |
8a187ee5 | 2098 | |
2563e2f3 MG |
2099 | static int |
2100 | mlx5_ib_map_pa_mr_sg_pi(struct ib_mr *ibmr, struct scatterlist *data_sg, | |
2101 | int data_sg_nents, unsigned int *data_sg_offset, | |
2102 | struct scatterlist *meta_sg, int meta_sg_nents, | |
2103 | unsigned int *meta_sg_offset) | |
2104 | { | |
2105 | struct mlx5_ib_mr *mr = to_mmr(ibmr); | |
2106 | unsigned int sg_offset = 0; | |
2107 | int n = 0; | |
2108 | ||
2109 | mr->meta_length = 0; | |
2110 | if (data_sg_nents == 1) { | |
2111 | n++; | |
2112 | mr->ndescs = 1; | |
2113 | if (data_sg_offset) | |
2114 | sg_offset = *data_sg_offset; | |
2115 | mr->data_length = sg_dma_len(data_sg) - sg_offset; | |
2116 | mr->data_iova = sg_dma_address(data_sg) + sg_offset; | |
2117 | if (meta_sg_nents == 1) { | |
2118 | n++; | |
2119 | mr->meta_ndescs = 1; | |
2120 | if (meta_sg_offset) | |
2121 | sg_offset = *meta_sg_offset; | |
2122 | else | |
2123 | sg_offset = 0; | |
2124 | mr->meta_length = sg_dma_len(meta_sg) - sg_offset; | |
2125 | mr->pi_iova = sg_dma_address(meta_sg) + sg_offset; | |
2126 | } | |
2127 | ibmr->length = mr->data_length + mr->meta_length; | |
2128 | } | |
2129 | ||
2130 | return n; | |
2131 | } | |
2132 | ||
b005d316 SG |
2133 | static int |
2134 | mlx5_ib_sg_to_klms(struct mlx5_ib_mr *mr, | |
2135 | struct scatterlist *sgl, | |
ff2ba993 | 2136 | unsigned short sg_nents, |
6c984472 MG |
2137 | unsigned int *sg_offset_p, |
2138 | struct scatterlist *meta_sgl, | |
2139 | unsigned short meta_sg_nents, | |
2140 | unsigned int *meta_sg_offset_p) | |
b005d316 SG |
2141 | { |
2142 | struct scatterlist *sg = sgl; | |
2143 | struct mlx5_klm *klms = mr->descs; | |
9aa8b321 | 2144 | unsigned int sg_offset = sg_offset_p ? *sg_offset_p : 0; |
b005d316 | 2145 | u32 lkey = mr->ibmr.pd->local_dma_lkey; |
6c984472 | 2146 | int i, j = 0; |
b005d316 | 2147 | |
ff2ba993 | 2148 | mr->ibmr.iova = sg_dma_address(sg) + sg_offset; |
b005d316 | 2149 | mr->ibmr.length = 0; |
b005d316 SG |
2150 | |
2151 | for_each_sg(sgl, sg, sg_nents, i) { | |
99975cd4 | 2152 | if (unlikely(i >= mr->max_descs)) |
b005d316 | 2153 | break; |
ff2ba993 CH |
2154 | klms[i].va = cpu_to_be64(sg_dma_address(sg) + sg_offset); |
2155 | klms[i].bcount = cpu_to_be32(sg_dma_len(sg) - sg_offset); | |
b005d316 | 2156 | klms[i].key = cpu_to_be32(lkey); |
0a49f2c3 | 2157 | mr->ibmr.length += sg_dma_len(sg) - sg_offset; |
ff2ba993 CH |
2158 | |
2159 | sg_offset = 0; | |
b005d316 SG |
2160 | } |
2161 | ||
9aa8b321 BVA |
2162 | if (sg_offset_p) |
2163 | *sg_offset_p = sg_offset; | |
2164 | ||
6c984472 MG |
2165 | mr->ndescs = i; |
2166 | mr->data_length = mr->ibmr.length; | |
2167 | ||
2168 | if (meta_sg_nents) { | |
2169 | sg = meta_sgl; | |
2170 | sg_offset = meta_sg_offset_p ? *meta_sg_offset_p : 0; | |
2171 | for_each_sg(meta_sgl, sg, meta_sg_nents, j) { | |
2172 | if (unlikely(i + j >= mr->max_descs)) | |
2173 | break; | |
2174 | klms[i + j].va = cpu_to_be64(sg_dma_address(sg) + | |
2175 | sg_offset); | |
2176 | klms[i + j].bcount = cpu_to_be32(sg_dma_len(sg) - | |
2177 | sg_offset); | |
2178 | klms[i + j].key = cpu_to_be32(lkey); | |
2179 | mr->ibmr.length += sg_dma_len(sg) - sg_offset; | |
2180 | ||
2181 | sg_offset = 0; | |
2182 | } | |
2183 | if (meta_sg_offset_p) | |
2184 | *meta_sg_offset_p = sg_offset; | |
2185 | ||
2186 | mr->meta_ndescs = j; | |
2187 | mr->meta_length = mr->ibmr.length - mr->data_length; | |
2188 | } | |
2189 | ||
2190 | return i + j; | |
b005d316 SG |
2191 | } |
2192 | ||
8a187ee5 SG |
2193 | static int mlx5_set_page(struct ib_mr *ibmr, u64 addr) |
2194 | { | |
2195 | struct mlx5_ib_mr *mr = to_mmr(ibmr); | |
2196 | __be64 *descs; | |
2197 | ||
2198 | if (unlikely(mr->ndescs == mr->max_descs)) | |
2199 | return -ENOMEM; | |
2200 | ||
2201 | descs = mr->descs; | |
2202 | descs[mr->ndescs++] = cpu_to_be64(addr | MLX5_EN_RD | MLX5_EN_WR); | |
2203 | ||
2204 | return 0; | |
2205 | } | |
2206 | ||
de0ae958 IR |
2207 | static int mlx5_set_page_pi(struct ib_mr *ibmr, u64 addr) |
2208 | { | |
2209 | struct mlx5_ib_mr *mr = to_mmr(ibmr); | |
2210 | __be64 *descs; | |
2211 | ||
2212 | if (unlikely(mr->ndescs + mr->meta_ndescs == mr->max_descs)) | |
2213 | return -ENOMEM; | |
2214 | ||
2215 | descs = mr->descs; | |
2216 | descs[mr->ndescs + mr->meta_ndescs++] = | |
2217 | cpu_to_be64(addr | MLX5_EN_RD | MLX5_EN_WR); | |
2218 | ||
2219 | return 0; | |
2220 | } | |
2221 | ||
2222 | static int | |
2223 | mlx5_ib_map_mtt_mr_sg_pi(struct ib_mr *ibmr, struct scatterlist *data_sg, | |
6c984472 MG |
2224 | int data_sg_nents, unsigned int *data_sg_offset, |
2225 | struct scatterlist *meta_sg, int meta_sg_nents, | |
2226 | unsigned int *meta_sg_offset) | |
2227 | { | |
2228 | struct mlx5_ib_mr *mr = to_mmr(ibmr); | |
de0ae958 | 2229 | struct mlx5_ib_mr *pi_mr = mr->mtt_mr; |
6c984472 MG |
2230 | int n; |
2231 | ||
de0ae958 IR |
2232 | pi_mr->ndescs = 0; |
2233 | pi_mr->meta_ndescs = 0; | |
2234 | pi_mr->meta_length = 0; | |
2235 | ||
2236 | ib_dma_sync_single_for_cpu(ibmr->device, pi_mr->desc_map, | |
2237 | pi_mr->desc_size * pi_mr->max_descs, | |
2238 | DMA_TO_DEVICE); | |
2239 | ||
2240 | pi_mr->ibmr.page_size = ibmr->page_size; | |
2241 | n = ib_sg_to_pages(&pi_mr->ibmr, data_sg, data_sg_nents, data_sg_offset, | |
2242 | mlx5_set_page); | |
2243 | if (n != data_sg_nents) | |
2244 | return n; | |
2245 | ||
2563e2f3 | 2246 | pi_mr->data_iova = pi_mr->ibmr.iova; |
de0ae958 IR |
2247 | pi_mr->data_length = pi_mr->ibmr.length; |
2248 | pi_mr->ibmr.length = pi_mr->data_length; | |
2249 | ibmr->length = pi_mr->data_length; | |
2250 | ||
2251 | if (meta_sg_nents) { | |
2252 | u64 page_mask = ~((u64)ibmr->page_size - 1); | |
2563e2f3 | 2253 | u64 iova = pi_mr->data_iova; |
de0ae958 IR |
2254 | |
2255 | n += ib_sg_to_pages(&pi_mr->ibmr, meta_sg, meta_sg_nents, | |
2256 | meta_sg_offset, mlx5_set_page_pi); | |
2257 | ||
2258 | pi_mr->meta_length = pi_mr->ibmr.length; | |
2259 | /* | |
2260 | * PI address for the HW is the offset of the metadata address | |
2261 | * relative to the first data page address. | |
2262 | * It equals to first data page address + size of data pages + | |
2263 | * metadata offset at the first metadata page | |
2264 | */ | |
2265 | pi_mr->pi_iova = (iova & page_mask) + | |
2266 | pi_mr->ndescs * ibmr->page_size + | |
2267 | (pi_mr->ibmr.iova & ~page_mask); | |
2268 | /* | |
2269 | * In order to use one MTT MR for data and metadata, we register | |
2270 | * also the gaps between the end of the data and the start of | |
2271 | * the metadata (the sig MR will verify that the HW will access | |
2272 | * to right addresses). This mapping is safe because we use | |
2273 | * internal mkey for the registration. | |
2274 | */ | |
2275 | pi_mr->ibmr.length = pi_mr->pi_iova + pi_mr->meta_length - iova; | |
2276 | pi_mr->ibmr.iova = iova; | |
2277 | ibmr->length += pi_mr->meta_length; | |
2278 | } | |
2279 | ||
2280 | ib_dma_sync_single_for_device(ibmr->device, pi_mr->desc_map, | |
2281 | pi_mr->desc_size * pi_mr->max_descs, | |
2282 | DMA_TO_DEVICE); | |
2283 | ||
2284 | return n; | |
2285 | } | |
2286 | ||
2287 | static int | |
2288 | mlx5_ib_map_klm_mr_sg_pi(struct ib_mr *ibmr, struct scatterlist *data_sg, | |
2289 | int data_sg_nents, unsigned int *data_sg_offset, | |
2290 | struct scatterlist *meta_sg, int meta_sg_nents, | |
2291 | unsigned int *meta_sg_offset) | |
2292 | { | |
2293 | struct mlx5_ib_mr *mr = to_mmr(ibmr); | |
2294 | struct mlx5_ib_mr *pi_mr = mr->klm_mr; | |
2295 | int n; | |
6c984472 MG |
2296 | |
2297 | pi_mr->ndescs = 0; | |
2298 | pi_mr->meta_ndescs = 0; | |
2299 | pi_mr->meta_length = 0; | |
2300 | ||
2301 | ib_dma_sync_single_for_cpu(ibmr->device, pi_mr->desc_map, | |
2302 | pi_mr->desc_size * pi_mr->max_descs, | |
2303 | DMA_TO_DEVICE); | |
2304 | ||
2305 | n = mlx5_ib_sg_to_klms(pi_mr, data_sg, data_sg_nents, data_sg_offset, | |
2306 | meta_sg, meta_sg_nents, meta_sg_offset); | |
2307 | ||
de0ae958 IR |
2308 | ib_dma_sync_single_for_device(ibmr->device, pi_mr->desc_map, |
2309 | pi_mr->desc_size * pi_mr->max_descs, | |
2310 | DMA_TO_DEVICE); | |
2311 | ||
6c984472 | 2312 | /* This is zero-based memory region */ |
2563e2f3 | 2313 | pi_mr->data_iova = 0; |
6c984472 | 2314 | pi_mr->ibmr.iova = 0; |
de0ae958 | 2315 | pi_mr->pi_iova = pi_mr->data_length; |
6c984472 | 2316 | ibmr->length = pi_mr->ibmr.length; |
6c984472 | 2317 | |
de0ae958 IR |
2318 | return n; |
2319 | } | |
6c984472 | 2320 | |
de0ae958 IR |
2321 | int mlx5_ib_map_mr_sg_pi(struct ib_mr *ibmr, struct scatterlist *data_sg, |
2322 | int data_sg_nents, unsigned int *data_sg_offset, | |
2323 | struct scatterlist *meta_sg, int meta_sg_nents, | |
2324 | unsigned int *meta_sg_offset) | |
2325 | { | |
2326 | struct mlx5_ib_mr *mr = to_mmr(ibmr); | |
2563e2f3 | 2327 | struct mlx5_ib_mr *pi_mr = NULL; |
de0ae958 IR |
2328 | int n; |
2329 | ||
2330 | WARN_ON(ibmr->type != IB_MR_TYPE_INTEGRITY); | |
2331 | ||
2563e2f3 MG |
2332 | mr->ndescs = 0; |
2333 | mr->data_length = 0; | |
2334 | mr->data_iova = 0; | |
2335 | mr->meta_ndescs = 0; | |
2336 | mr->pi_iova = 0; | |
2337 | /* | |
2338 | * As a performance optimization, if possible, there is no need to | |
2339 | * perform UMR operation to register the data/metadata buffers. | |
2340 | * First try to map the sg lists to PA descriptors with local_dma_lkey. | |
2341 | * Fallback to UMR only in case of a failure. | |
2342 | */ | |
2343 | n = mlx5_ib_map_pa_mr_sg_pi(ibmr, data_sg, data_sg_nents, | |
2344 | data_sg_offset, meta_sg, meta_sg_nents, | |
2345 | meta_sg_offset); | |
2346 | if (n == data_sg_nents + meta_sg_nents) | |
2347 | goto out; | |
de0ae958 IR |
2348 | /* |
2349 | * As a performance optimization, if possible, there is no need to map | |
2350 | * the sg lists to KLM descriptors. First try to map the sg lists to MTT | |
2351 | * descriptors and fallback to KLM only in case of a failure. | |
2352 | * It's more efficient for the HW to work with MTT descriptors | |
2353 | * (especially in high load). | |
2354 | * Use KLM (indirect access) only if it's mandatory. | |
2355 | */ | |
2563e2f3 | 2356 | pi_mr = mr->mtt_mr; |
de0ae958 IR |
2357 | n = mlx5_ib_map_mtt_mr_sg_pi(ibmr, data_sg, data_sg_nents, |
2358 | data_sg_offset, meta_sg, meta_sg_nents, | |
2359 | meta_sg_offset); | |
2360 | if (n == data_sg_nents + meta_sg_nents) | |
2361 | goto out; | |
2362 | ||
2363 | pi_mr = mr->klm_mr; | |
2364 | n = mlx5_ib_map_klm_mr_sg_pi(ibmr, data_sg, data_sg_nents, | |
2365 | data_sg_offset, meta_sg, meta_sg_nents, | |
2366 | meta_sg_offset); | |
6c984472 MG |
2367 | if (unlikely(n != data_sg_nents + meta_sg_nents)) |
2368 | return -ENOMEM; | |
2369 | ||
de0ae958 IR |
2370 | out: |
2371 | /* This is zero-based memory region */ | |
2372 | ibmr->iova = 0; | |
2373 | mr->pi_mr = pi_mr; | |
2563e2f3 MG |
2374 | if (pi_mr) |
2375 | ibmr->sig_attrs->meta_length = pi_mr->meta_length; | |
2376 | else | |
2377 | ibmr->sig_attrs->meta_length = mr->meta_length; | |
de0ae958 | 2378 | |
6c984472 MG |
2379 | return 0; |
2380 | } | |
2381 | ||
ff2ba993 | 2382 | int mlx5_ib_map_mr_sg(struct ib_mr *ibmr, struct scatterlist *sg, int sg_nents, |
9aa8b321 | 2383 | unsigned int *sg_offset) |
8a187ee5 SG |
2384 | { |
2385 | struct mlx5_ib_mr *mr = to_mmr(ibmr); | |
2386 | int n; | |
2387 | ||
2388 | mr->ndescs = 0; | |
2389 | ||
2390 | ib_dma_sync_single_for_cpu(ibmr->device, mr->desc_map, | |
2391 | mr->desc_size * mr->max_descs, | |
2392 | DMA_TO_DEVICE); | |
2393 | ||
ec22eb53 | 2394 | if (mr->access_mode == MLX5_MKC_ACCESS_MODE_KLMS) |
6c984472 MG |
2395 | n = mlx5_ib_sg_to_klms(mr, sg, sg_nents, sg_offset, NULL, 0, |
2396 | NULL); | |
b005d316 | 2397 | else |
ff2ba993 CH |
2398 | n = ib_sg_to_pages(ibmr, sg, sg_nents, sg_offset, |
2399 | mlx5_set_page); | |
8a187ee5 SG |
2400 | |
2401 | ib_dma_sync_single_for_device(ibmr->device, mr->desc_map, | |
2402 | mr->desc_size * mr->max_descs, | |
2403 | DMA_TO_DEVICE); | |
2404 | ||
2405 | return n; | |
2406 | } |