Line data Source code
1 : /* SPDX-License-Identifier: BSD-3-Clause
2 : * Copyright (C) 2017 Intel Corporation.
3 : * All rights reserved.
4 : */
5 :
6 : #include "spdk/stdinc.h"
7 :
8 : #include "env_internal.h"
9 : #include "pci_dpdk.h"
10 :
11 : #include <rte_config.h>
12 : #include <rte_memory.h>
13 : #include <rte_eal_memconfig.h>
14 : #include <rte_dev.h>
15 : #include <rte_pci.h>
16 :
17 : #include "spdk_internal/assert.h"
18 :
19 : #include "spdk/assert.h"
20 : #include "spdk/likely.h"
21 : #include "spdk/queue.h"
22 : #include "spdk/util.h"
23 : #include "spdk/memory.h"
24 : #include "spdk/env_dpdk.h"
25 : #include "spdk/log.h"
26 :
27 : #ifdef __linux__
28 : #include <linux/version.h>
29 : #if LINUX_VERSION_CODE >= KERNEL_VERSION(3, 6, 0)
30 : #include <linux/vfio.h>
31 : #include <rte_vfio.h>
32 :
33 : struct spdk_vfio_dma_map {
34 : struct vfio_iommu_type1_dma_map map;
35 : TAILQ_ENTRY(spdk_vfio_dma_map) tailq;
36 : };
37 :
38 : struct vfio_cfg {
39 : int fd;
40 : bool enabled;
41 : bool noiommu_enabled;
42 : unsigned device_ref;
43 : TAILQ_HEAD(, spdk_vfio_dma_map) maps;
44 : pthread_mutex_t mutex;
45 : };
46 :
47 : static struct vfio_cfg g_vfio = {
48 : .fd = -1,
49 : .enabled = false,
50 : .noiommu_enabled = false,
51 : .device_ref = 0,
52 : .maps = TAILQ_HEAD_INITIALIZER(g_vfio.maps),
53 : .mutex = PTHREAD_MUTEX_INITIALIZER
54 : };
55 : #endif
56 : #endif
57 :
58 : #if DEBUG
59 : #define DEBUG_PRINT(...) SPDK_ERRLOG(__VA_ARGS__)
60 : #else
61 : #define DEBUG_PRINT(...)
62 : #endif
63 :
64 : #define FN_2MB_TO_4KB(fn) (fn << (SHIFT_2MB - SHIFT_4KB))
65 : #define FN_4KB_TO_2MB(fn) (fn >> (SHIFT_2MB - SHIFT_4KB))
66 :
67 : #define MAP_256TB_IDX(vfn_2mb) ((vfn_2mb) >> (SHIFT_1GB - SHIFT_2MB))
68 : #define MAP_1GB_IDX(vfn_2mb) ((vfn_2mb) & ((1ULL << (SHIFT_1GB - SHIFT_2MB)) - 1))
69 :
70 : /* Page is registered */
71 : #define REG_MAP_REGISTERED (1ULL << 62)
72 :
73 : /* A notification region barrier. The 2MB translation entry that's marked
74 : * with this flag must be unregistered separately. This allows contiguous
75 : * regions to be unregistered in the same chunks they were registered.
76 : */
77 : #define REG_MAP_NOTIFY_START (1ULL << 63)
78 :
79 : /* Translation of a single 2MB page. */
80 : struct map_2mb {
81 : uint64_t translation_2mb;
82 : };
83 :
84 : /* Second-level map table indexed by bits [21..29] of the virtual address.
85 : * Each entry contains the address translation or error for entries that haven't
86 : * been retrieved yet.
87 : */
88 : struct map_1gb {
89 : struct map_2mb map[1ULL << (SHIFT_1GB - SHIFT_2MB)];
90 : };
91 :
92 : /* Top-level map table indexed by bits [30..47] of the virtual address.
93 : * Each entry points to a second-level map table or NULL.
94 : */
95 : struct map_256tb {
96 : struct map_1gb *map[1ULL << (SHIFT_256TB - SHIFT_1GB)];
97 : };
98 :
99 : /* Page-granularity memory address translation */
100 : struct spdk_mem_map {
101 : struct map_256tb map_256tb;
102 : pthread_mutex_t mutex;
103 : uint64_t default_translation;
104 : struct spdk_mem_map_ops ops;
105 : void *cb_ctx;
106 : TAILQ_ENTRY(spdk_mem_map) tailq;
107 : };
108 :
109 : /* Registrations map. The 64 bit translations are bit fields with the
110 : * following layout (starting with the low bits):
111 : * 0 - 61 : reserved
112 : * 62 - 63 : flags
113 : */
114 : static struct spdk_mem_map *g_mem_reg_map;
115 : static TAILQ_HEAD(spdk_mem_map_head, spdk_mem_map) g_spdk_mem_maps =
116 : TAILQ_HEAD_INITIALIZER(g_spdk_mem_maps);
117 : static pthread_mutex_t g_spdk_mem_map_mutex = PTHREAD_MUTEX_INITIALIZER;
118 :
119 : static bool g_legacy_mem;
120 : static bool g_huge_pages = true;
121 :
122 : /*
123 : * Walk the currently registered memory via the main memory registration map
124 : * and call the new map's notify callback for each virtually contiguous region.
125 : */
126 : static int
127 0 : mem_map_notify_walk(struct spdk_mem_map *map, enum spdk_mem_map_notify_action action)
128 : {
129 0 : size_t idx_256tb;
130 0 : uint64_t idx_1gb;
131 0 : uint64_t contig_start = UINT64_MAX;
132 0 : uint64_t contig_end = UINT64_MAX;
133 0 : struct map_1gb *map_1gb;
134 0 : int rc;
135 :
136 0 : if (!g_mem_reg_map) {
137 0 : return -EINVAL;
138 : }
139 :
140 : /* Hold the memory registration map mutex so no new registrations can be added while we are looping. */
141 0 : pthread_mutex_lock(&g_mem_reg_map->mutex);
142 :
143 0 : for (idx_256tb = 0;
144 0 : idx_256tb < sizeof(g_mem_reg_map->map_256tb.map) / sizeof(g_mem_reg_map->map_256tb.map[0]);
145 0 : idx_256tb++) {
146 0 : map_1gb = g_mem_reg_map->map_256tb.map[idx_256tb];
147 :
148 0 : if (!map_1gb) {
149 0 : if (contig_start != UINT64_MAX) {
150 : /* End of of a virtually contiguous range */
151 0 : rc = map->ops.notify_cb(map->cb_ctx, map, action,
152 0 : (void *)contig_start,
153 0 : contig_end - contig_start + VALUE_2MB);
154 : /* Don't bother handling unregister failures. It can't be any worse */
155 0 : if (rc != 0 && action == SPDK_MEM_MAP_NOTIFY_REGISTER) {
156 0 : goto err_unregister;
157 : }
158 0 : }
159 0 : contig_start = UINT64_MAX;
160 0 : continue;
161 : }
162 :
163 0 : for (idx_1gb = 0; idx_1gb < sizeof(map_1gb->map) / sizeof(map_1gb->map[0]); idx_1gb++) {
164 0 : if ((map_1gb->map[idx_1gb].translation_2mb & REG_MAP_REGISTERED) &&
165 0 : (contig_start == UINT64_MAX ||
166 0 : (map_1gb->map[idx_1gb].translation_2mb & REG_MAP_NOTIFY_START) == 0)) {
167 : /* Rebuild the virtual address from the indexes */
168 0 : uint64_t vaddr = (idx_256tb << SHIFT_1GB) | (idx_1gb << SHIFT_2MB);
169 :
170 0 : if (contig_start == UINT64_MAX) {
171 0 : contig_start = vaddr;
172 0 : }
173 :
174 0 : contig_end = vaddr;
175 0 : } else {
176 0 : if (contig_start != UINT64_MAX) {
177 : /* End of of a virtually contiguous range */
178 0 : rc = map->ops.notify_cb(map->cb_ctx, map, action,
179 0 : (void *)contig_start,
180 0 : contig_end - contig_start + VALUE_2MB);
181 : /* Don't bother handling unregister failures. It can't be any worse */
182 0 : if (rc != 0 && action == SPDK_MEM_MAP_NOTIFY_REGISTER) {
183 0 : goto err_unregister;
184 : }
185 :
186 : /* This page might be a part of a neighbour region, so process
187 : * it again. The idx_1gb will be incremented immediately.
188 : */
189 0 : idx_1gb--;
190 0 : }
191 0 : contig_start = UINT64_MAX;
192 : }
193 0 : }
194 0 : }
195 :
196 0 : pthread_mutex_unlock(&g_mem_reg_map->mutex);
197 0 : return 0;
198 :
199 : err_unregister:
200 : /* Unwind to the first empty translation so we don't unregister
201 : * a region that just failed to register.
202 : */
203 0 : idx_256tb = MAP_256TB_IDX((contig_start >> SHIFT_2MB) - 1);
204 0 : idx_1gb = MAP_1GB_IDX((contig_start >> SHIFT_2MB) - 1);
205 0 : contig_start = UINT64_MAX;
206 0 : contig_end = UINT64_MAX;
207 :
208 : /* Unregister any memory we managed to register before the failure */
209 0 : for (; idx_256tb < SIZE_MAX; idx_256tb--) {
210 0 : map_1gb = g_mem_reg_map->map_256tb.map[idx_256tb];
211 :
212 0 : if (!map_1gb) {
213 0 : if (contig_end != UINT64_MAX) {
214 : /* End of of a virtually contiguous range */
215 0 : map->ops.notify_cb(map->cb_ctx, map,
216 : SPDK_MEM_MAP_NOTIFY_UNREGISTER,
217 0 : (void *)contig_start,
218 0 : contig_end - contig_start + VALUE_2MB);
219 0 : }
220 0 : contig_end = UINT64_MAX;
221 0 : continue;
222 : }
223 :
224 0 : for (; idx_1gb < UINT64_MAX; idx_1gb--) {
225 : /* Rebuild the virtual address from the indexes */
226 0 : uint64_t vaddr = (idx_256tb << SHIFT_1GB) | (idx_1gb << SHIFT_2MB);
227 0 : if ((map_1gb->map[idx_1gb].translation_2mb & REG_MAP_REGISTERED) &&
228 0 : (contig_end == UINT64_MAX || (map_1gb->map[idx_1gb].translation_2mb & REG_MAP_NOTIFY_START) == 0)) {
229 :
230 0 : if (contig_end == UINT64_MAX) {
231 0 : contig_end = vaddr;
232 0 : }
233 0 : contig_start = vaddr;
234 0 : } else {
235 0 : if (contig_end != UINT64_MAX) {
236 0 : if (map_1gb->map[idx_1gb].translation_2mb & REG_MAP_NOTIFY_START) {
237 0 : contig_start = vaddr;
238 0 : }
239 : /* End of of a virtually contiguous range */
240 0 : map->ops.notify_cb(map->cb_ctx, map,
241 : SPDK_MEM_MAP_NOTIFY_UNREGISTER,
242 0 : (void *)contig_start,
243 0 : contig_end - contig_start + VALUE_2MB);
244 0 : }
245 0 : contig_end = UINT64_MAX;
246 : }
247 0 : }
248 0 : idx_1gb = sizeof(map_1gb->map) / sizeof(map_1gb->map[0]) - 1;
249 0 : }
250 :
251 0 : pthread_mutex_unlock(&g_mem_reg_map->mutex);
252 0 : return rc;
253 0 : }
254 :
255 : struct spdk_mem_map *
256 0 : spdk_mem_map_alloc(uint64_t default_translation, const struct spdk_mem_map_ops *ops, void *cb_ctx)
257 : {
258 0 : struct spdk_mem_map *map;
259 0 : int rc;
260 0 : size_t i;
261 :
262 0 : map = calloc(1, sizeof(*map));
263 0 : if (map == NULL) {
264 0 : return NULL;
265 : }
266 :
267 0 : if (pthread_mutex_init(&map->mutex, NULL)) {
268 0 : free(map);
269 0 : return NULL;
270 : }
271 :
272 0 : map->default_translation = default_translation;
273 0 : map->cb_ctx = cb_ctx;
274 0 : if (ops) {
275 0 : map->ops = *ops;
276 0 : }
277 :
278 0 : if (ops && ops->notify_cb) {
279 0 : pthread_mutex_lock(&g_spdk_mem_map_mutex);
280 0 : rc = mem_map_notify_walk(map, SPDK_MEM_MAP_NOTIFY_REGISTER);
281 0 : if (rc != 0) {
282 0 : pthread_mutex_unlock(&g_spdk_mem_map_mutex);
283 0 : DEBUG_PRINT("Initial mem_map notify failed\n");
284 0 : pthread_mutex_destroy(&map->mutex);
285 0 : for (i = 0; i < sizeof(map->map_256tb.map) / sizeof(map->map_256tb.map[0]); i++) {
286 0 : free(map->map_256tb.map[i]);
287 0 : }
288 0 : free(map);
289 0 : return NULL;
290 : }
291 0 : TAILQ_INSERT_TAIL(&g_spdk_mem_maps, map, tailq);
292 0 : pthread_mutex_unlock(&g_spdk_mem_map_mutex);
293 0 : }
294 :
295 0 : return map;
296 0 : }
297 :
298 : void
299 0 : spdk_mem_map_free(struct spdk_mem_map **pmap)
300 : {
301 0 : struct spdk_mem_map *map;
302 0 : size_t i;
303 :
304 0 : if (!pmap) {
305 0 : return;
306 : }
307 :
308 0 : map = *pmap;
309 :
310 0 : if (!map) {
311 0 : return;
312 : }
313 :
314 0 : if (map->ops.notify_cb) {
315 0 : pthread_mutex_lock(&g_spdk_mem_map_mutex);
316 0 : mem_map_notify_walk(map, SPDK_MEM_MAP_NOTIFY_UNREGISTER);
317 0 : TAILQ_REMOVE(&g_spdk_mem_maps, map, tailq);
318 0 : pthread_mutex_unlock(&g_spdk_mem_map_mutex);
319 0 : }
320 :
321 0 : for (i = 0; i < sizeof(map->map_256tb.map) / sizeof(map->map_256tb.map[0]); i++) {
322 0 : free(map->map_256tb.map[i]);
323 0 : }
324 :
325 0 : pthread_mutex_destroy(&map->mutex);
326 :
327 0 : free(map);
328 0 : *pmap = NULL;
329 0 : }
330 :
331 : int
332 0 : spdk_mem_register(void *_vaddr, size_t len)
333 : {
334 0 : struct spdk_mem_map *map;
335 0 : int rc;
336 0 : uint64_t vaddr = (uintptr_t)_vaddr;
337 0 : uint64_t seg_vaddr;
338 0 : size_t seg_len;
339 0 : uint64_t reg;
340 :
341 0 : if ((uintptr_t)vaddr & ~MASK_256TB) {
342 0 : DEBUG_PRINT("invalid usermode virtual address %jx\n", vaddr);
343 0 : return -EINVAL;
344 : }
345 :
346 0 : if (((uintptr_t)vaddr & MASK_2MB) || (len & MASK_2MB)) {
347 0 : DEBUG_PRINT("invalid %s parameters, vaddr=%jx len=%ju\n",
348 : __func__, vaddr, len);
349 0 : return -EINVAL;
350 : }
351 :
352 0 : if (len == 0) {
353 0 : return 0;
354 : }
355 :
356 0 : pthread_mutex_lock(&g_spdk_mem_map_mutex);
357 :
358 0 : seg_vaddr = vaddr;
359 0 : seg_len = len;
360 0 : while (seg_len > 0) {
361 0 : reg = spdk_mem_map_translate(g_mem_reg_map, (uint64_t)seg_vaddr, NULL);
362 0 : if (reg & REG_MAP_REGISTERED) {
363 0 : pthread_mutex_unlock(&g_spdk_mem_map_mutex);
364 0 : return -EBUSY;
365 : }
366 0 : seg_vaddr += VALUE_2MB;
367 0 : seg_len -= VALUE_2MB;
368 : }
369 :
370 0 : seg_vaddr = vaddr;
371 0 : seg_len = 0;
372 0 : while (len > 0) {
373 0 : spdk_mem_map_set_translation(g_mem_reg_map, (uint64_t)vaddr, VALUE_2MB,
374 0 : seg_len == 0 ? REG_MAP_REGISTERED | REG_MAP_NOTIFY_START : REG_MAP_REGISTERED);
375 0 : seg_len += VALUE_2MB;
376 0 : vaddr += VALUE_2MB;
377 0 : len -= VALUE_2MB;
378 : }
379 :
380 0 : TAILQ_FOREACH(map, &g_spdk_mem_maps, tailq) {
381 0 : rc = map->ops.notify_cb(map->cb_ctx, map, SPDK_MEM_MAP_NOTIFY_REGISTER,
382 0 : (void *)seg_vaddr, seg_len);
383 0 : if (rc != 0) {
384 0 : pthread_mutex_unlock(&g_spdk_mem_map_mutex);
385 0 : return rc;
386 : }
387 0 : }
388 :
389 0 : pthread_mutex_unlock(&g_spdk_mem_map_mutex);
390 0 : return 0;
391 0 : }
392 :
393 : int
394 0 : spdk_mem_unregister(void *_vaddr, size_t len)
395 : {
396 0 : struct spdk_mem_map *map;
397 0 : int rc;
398 0 : uint64_t vaddr = (uintptr_t)_vaddr;
399 0 : uint64_t seg_vaddr;
400 0 : size_t seg_len;
401 0 : uint64_t reg, newreg;
402 :
403 0 : if ((uintptr_t)vaddr & ~MASK_256TB) {
404 0 : DEBUG_PRINT("invalid usermode virtual address %jx\n", vaddr);
405 0 : return -EINVAL;
406 : }
407 :
408 0 : if (((uintptr_t)vaddr & MASK_2MB) || (len & MASK_2MB)) {
409 0 : DEBUG_PRINT("invalid %s parameters, vaddr=%jx len=%ju\n",
410 : __func__, vaddr, len);
411 0 : return -EINVAL;
412 : }
413 :
414 0 : pthread_mutex_lock(&g_spdk_mem_map_mutex);
415 :
416 : /* The first page must be a start of a region. Also check if it's
417 : * registered to make sure we don't return -ERANGE for non-registered
418 : * regions.
419 : */
420 0 : reg = spdk_mem_map_translate(g_mem_reg_map, (uint64_t)vaddr, NULL);
421 0 : if ((reg & REG_MAP_REGISTERED) && (reg & REG_MAP_NOTIFY_START) == 0) {
422 0 : pthread_mutex_unlock(&g_spdk_mem_map_mutex);
423 0 : return -ERANGE;
424 : }
425 :
426 0 : seg_vaddr = vaddr;
427 0 : seg_len = len;
428 0 : while (seg_len > 0) {
429 0 : reg = spdk_mem_map_translate(g_mem_reg_map, (uint64_t)seg_vaddr, NULL);
430 0 : if ((reg & REG_MAP_REGISTERED) == 0) {
431 0 : pthread_mutex_unlock(&g_spdk_mem_map_mutex);
432 0 : return -EINVAL;
433 : }
434 0 : seg_vaddr += VALUE_2MB;
435 0 : seg_len -= VALUE_2MB;
436 : }
437 :
438 0 : newreg = spdk_mem_map_translate(g_mem_reg_map, (uint64_t)seg_vaddr, NULL);
439 : /* If the next page is registered, it must be a start of a region as well,
440 : * otherwise we'd be unregistering only a part of a region.
441 : */
442 0 : if ((newreg & REG_MAP_NOTIFY_START) == 0 && (newreg & REG_MAP_REGISTERED)) {
443 0 : pthread_mutex_unlock(&g_spdk_mem_map_mutex);
444 0 : return -ERANGE;
445 : }
446 0 : seg_vaddr = vaddr;
447 0 : seg_len = 0;
448 :
449 0 : while (len > 0) {
450 0 : reg = spdk_mem_map_translate(g_mem_reg_map, (uint64_t)vaddr, NULL);
451 0 : spdk_mem_map_set_translation(g_mem_reg_map, (uint64_t)vaddr, VALUE_2MB, 0);
452 :
453 0 : if (seg_len > 0 && (reg & REG_MAP_NOTIFY_START)) {
454 0 : TAILQ_FOREACH_REVERSE(map, &g_spdk_mem_maps, spdk_mem_map_head, tailq) {
455 0 : rc = map->ops.notify_cb(map->cb_ctx, map, SPDK_MEM_MAP_NOTIFY_UNREGISTER,
456 0 : (void *)seg_vaddr, seg_len);
457 0 : if (rc != 0) {
458 0 : pthread_mutex_unlock(&g_spdk_mem_map_mutex);
459 0 : return rc;
460 : }
461 0 : }
462 :
463 0 : seg_vaddr = vaddr;
464 0 : seg_len = VALUE_2MB;
465 0 : } else {
466 0 : seg_len += VALUE_2MB;
467 : }
468 :
469 0 : vaddr += VALUE_2MB;
470 0 : len -= VALUE_2MB;
471 : }
472 :
473 0 : if (seg_len > 0) {
474 0 : TAILQ_FOREACH_REVERSE(map, &g_spdk_mem_maps, spdk_mem_map_head, tailq) {
475 0 : rc = map->ops.notify_cb(map->cb_ctx, map, SPDK_MEM_MAP_NOTIFY_UNREGISTER,
476 0 : (void *)seg_vaddr, seg_len);
477 0 : if (rc != 0) {
478 0 : pthread_mutex_unlock(&g_spdk_mem_map_mutex);
479 0 : return rc;
480 : }
481 0 : }
482 0 : }
483 :
484 0 : pthread_mutex_unlock(&g_spdk_mem_map_mutex);
485 0 : return 0;
486 0 : }
487 :
488 : int
489 0 : spdk_mem_reserve(void *vaddr, size_t len)
490 : {
491 0 : struct spdk_mem_map *map;
492 0 : void *seg_vaddr;
493 0 : size_t seg_len;
494 0 : uint64_t reg;
495 :
496 0 : if ((uintptr_t)vaddr & ~MASK_256TB) {
497 0 : DEBUG_PRINT("invalid usermode virtual address %p\n", vaddr);
498 0 : return -EINVAL;
499 : }
500 :
501 0 : if (((uintptr_t)vaddr & MASK_2MB) || (len & MASK_2MB)) {
502 0 : DEBUG_PRINT("invalid %s parameters, vaddr=%p len=%ju\n",
503 : __func__, vaddr, len);
504 0 : return -EINVAL;
505 : }
506 :
507 0 : if (len == 0) {
508 0 : return 0;
509 : }
510 :
511 0 : pthread_mutex_lock(&g_spdk_mem_map_mutex);
512 :
513 : /* Check if any part of this range is already registered */
514 0 : seg_vaddr = vaddr;
515 0 : seg_len = len;
516 0 : while (seg_len > 0) {
517 0 : reg = spdk_mem_map_translate(g_mem_reg_map, (uint64_t)seg_vaddr, NULL);
518 0 : if (reg & REG_MAP_REGISTERED) {
519 0 : pthread_mutex_unlock(&g_spdk_mem_map_mutex);
520 0 : return -EBUSY;
521 : }
522 0 : seg_vaddr += VALUE_2MB;
523 0 : seg_len -= VALUE_2MB;
524 : }
525 :
526 : /* Simply set the translation to the memory map's default. This allocates the space in the
527 : * map but does not provide a valid translation. */
528 0 : spdk_mem_map_set_translation(g_mem_reg_map, (uint64_t)vaddr, len,
529 0 : g_mem_reg_map->default_translation);
530 :
531 0 : TAILQ_FOREACH(map, &g_spdk_mem_maps, tailq) {
532 0 : spdk_mem_map_set_translation(map, (uint64_t)vaddr, len, map->default_translation);
533 0 : }
534 :
535 0 : pthread_mutex_unlock(&g_spdk_mem_map_mutex);
536 0 : return 0;
537 0 : }
538 :
539 : static struct map_1gb *
540 0 : mem_map_get_map_1gb(struct spdk_mem_map *map, uint64_t vfn_2mb)
541 : {
542 0 : struct map_1gb *map_1gb;
543 0 : uint64_t idx_256tb = MAP_256TB_IDX(vfn_2mb);
544 0 : size_t i;
545 :
546 0 : if (spdk_unlikely(idx_256tb >= SPDK_COUNTOF(map->map_256tb.map))) {
547 0 : return NULL;
548 : }
549 :
550 0 : map_1gb = map->map_256tb.map[idx_256tb];
551 :
552 0 : if (!map_1gb) {
553 0 : pthread_mutex_lock(&map->mutex);
554 :
555 : /* Recheck to make sure nobody else got the mutex first. */
556 0 : map_1gb = map->map_256tb.map[idx_256tb];
557 0 : if (!map_1gb) {
558 0 : map_1gb = malloc(sizeof(struct map_1gb));
559 0 : if (map_1gb) {
560 : /* initialize all entries to default translation */
561 0 : for (i = 0; i < SPDK_COUNTOF(map_1gb->map); i++) {
562 0 : map_1gb->map[i].translation_2mb = map->default_translation;
563 0 : }
564 0 : map->map_256tb.map[idx_256tb] = map_1gb;
565 0 : }
566 0 : }
567 :
568 0 : pthread_mutex_unlock(&map->mutex);
569 :
570 0 : if (!map_1gb) {
571 0 : DEBUG_PRINT("allocation failed\n");
572 0 : return NULL;
573 : }
574 0 : }
575 :
576 0 : return map_1gb;
577 0 : }
578 :
579 : int
580 0 : spdk_mem_map_set_translation(struct spdk_mem_map *map, uint64_t vaddr, uint64_t size,
581 : uint64_t translation)
582 : {
583 0 : uint64_t vfn_2mb;
584 0 : struct map_1gb *map_1gb;
585 0 : uint64_t idx_1gb;
586 0 : struct map_2mb *map_2mb;
587 :
588 0 : if ((uintptr_t)vaddr & ~MASK_256TB) {
589 0 : DEBUG_PRINT("invalid usermode virtual address %" PRIu64 "\n", vaddr);
590 0 : return -EINVAL;
591 : }
592 :
593 : /* For now, only 2 MB-aligned registrations are supported */
594 0 : if (((uintptr_t)vaddr & MASK_2MB) || (size & MASK_2MB)) {
595 0 : DEBUG_PRINT("invalid %s parameters, vaddr=%" PRIu64 " len=%" PRIu64 "\n",
596 : __func__, vaddr, size);
597 0 : return -EINVAL;
598 : }
599 :
600 0 : vfn_2mb = vaddr >> SHIFT_2MB;
601 :
602 0 : while (size) {
603 0 : map_1gb = mem_map_get_map_1gb(map, vfn_2mb);
604 0 : if (!map_1gb) {
605 0 : DEBUG_PRINT("could not get %p map\n", (void *)vaddr);
606 0 : return -ENOMEM;
607 : }
608 :
609 0 : idx_1gb = MAP_1GB_IDX(vfn_2mb);
610 0 : map_2mb = &map_1gb->map[idx_1gb];
611 0 : map_2mb->translation_2mb = translation;
612 :
613 0 : size -= VALUE_2MB;
614 0 : vfn_2mb++;
615 : }
616 :
617 0 : return 0;
618 0 : }
619 :
620 : int
621 0 : spdk_mem_map_clear_translation(struct spdk_mem_map *map, uint64_t vaddr, uint64_t size)
622 : {
623 0 : return spdk_mem_map_set_translation(map, vaddr, size, map->default_translation);
624 : }
625 :
626 : inline uint64_t
627 0 : spdk_mem_map_translate(const struct spdk_mem_map *map, uint64_t vaddr, uint64_t *size)
628 : {
629 0 : const struct map_1gb *map_1gb;
630 0 : const struct map_2mb *map_2mb;
631 0 : uint64_t idx_256tb;
632 0 : uint64_t idx_1gb;
633 0 : uint64_t vfn_2mb;
634 0 : uint64_t cur_size;
635 0 : uint64_t prev_translation;
636 0 : uint64_t orig_translation;
637 :
638 0 : if (spdk_unlikely(vaddr & ~MASK_256TB)) {
639 0 : DEBUG_PRINT("invalid usermode virtual address %p\n", (void *)vaddr);
640 0 : return map->default_translation;
641 : }
642 :
643 0 : vfn_2mb = vaddr >> SHIFT_2MB;
644 0 : idx_256tb = MAP_256TB_IDX(vfn_2mb);
645 0 : idx_1gb = MAP_1GB_IDX(vfn_2mb);
646 :
647 0 : map_1gb = map->map_256tb.map[idx_256tb];
648 0 : if (spdk_unlikely(!map_1gb)) {
649 0 : return map->default_translation;
650 : }
651 :
652 0 : cur_size = VALUE_2MB - _2MB_OFFSET(vaddr);
653 0 : map_2mb = &map_1gb->map[idx_1gb];
654 0 : if (size == NULL || map->ops.are_contiguous == NULL ||
655 0 : map_2mb->translation_2mb == map->default_translation) {
656 0 : if (size != NULL) {
657 0 : *size = spdk_min(*size, cur_size);
658 0 : }
659 0 : return map_2mb->translation_2mb;
660 : }
661 :
662 0 : orig_translation = map_2mb->translation_2mb;
663 0 : prev_translation = orig_translation;
664 0 : while (cur_size < *size) {
665 0 : vfn_2mb++;
666 0 : idx_256tb = MAP_256TB_IDX(vfn_2mb);
667 0 : idx_1gb = MAP_1GB_IDX(vfn_2mb);
668 :
669 0 : map_1gb = map->map_256tb.map[idx_256tb];
670 0 : if (spdk_unlikely(!map_1gb)) {
671 0 : break;
672 : }
673 :
674 0 : map_2mb = &map_1gb->map[idx_1gb];
675 0 : if (!map->ops.are_contiguous(prev_translation, map_2mb->translation_2mb)) {
676 0 : break;
677 : }
678 :
679 0 : cur_size += VALUE_2MB;
680 0 : prev_translation = map_2mb->translation_2mb;
681 : }
682 :
683 0 : *size = spdk_min(*size, cur_size);
684 0 : return orig_translation;
685 0 : }
686 :
687 : static void
688 0 : memory_hotplug_cb(enum rte_mem_event event_type,
689 : const void *addr, size_t len, void *arg)
690 : {
691 0 : if (event_type == RTE_MEM_EVENT_ALLOC) {
692 0 : spdk_mem_register((void *)addr, len);
693 :
694 0 : if (!spdk_env_dpdk_external_init()) {
695 0 : return;
696 : }
697 :
698 : /* When the user initialized DPDK separately, we can't
699 : * be sure that --match-allocations RTE flag was specified.
700 : * Without this flag, DPDK can free memory in different units
701 : * than it was allocated. It doesn't work with things like RDMA MRs.
702 : *
703 : * For such cases, we mark segments so they aren't freed.
704 : */
705 0 : while (len > 0) {
706 0 : struct rte_memseg *seg;
707 :
708 0 : seg = rte_mem_virt2memseg(addr, NULL);
709 0 : assert(seg != NULL);
710 0 : seg->flags |= RTE_MEMSEG_FLAG_DO_NOT_FREE;
711 0 : addr = (void *)((uintptr_t)addr + seg->hugepage_sz);
712 0 : len -= seg->hugepage_sz;
713 0 : }
714 0 : } else if (event_type == RTE_MEM_EVENT_FREE) {
715 0 : spdk_mem_unregister((void *)addr, len);
716 0 : }
717 0 : }
718 :
719 : static int
720 0 : memory_iter_cb(const struct rte_memseg_list *msl,
721 : const struct rte_memseg *ms, size_t len, void *arg)
722 : {
723 0 : return spdk_mem_register(ms->addr, len);
724 : }
725 :
726 : int
727 0 : mem_map_init(bool legacy_mem)
728 : {
729 0 : g_legacy_mem = legacy_mem;
730 :
731 0 : g_mem_reg_map = spdk_mem_map_alloc(0, NULL, NULL);
732 0 : if (g_mem_reg_map == NULL) {
733 0 : DEBUG_PRINT("memory registration map allocation failed\n");
734 0 : return -ENOMEM;
735 : }
736 :
737 : /*
738 : * Walk all DPDK memory segments and register them
739 : * with the main memory map
740 : */
741 0 : if (g_huge_pages) {
742 0 : rte_mem_event_callback_register("spdk", memory_hotplug_cb, NULL);
743 0 : rte_memseg_contig_walk(memory_iter_cb, NULL);
744 0 : }
745 0 : return 0;
746 0 : }
747 :
748 : bool
749 0 : spdk_iommu_is_enabled(void)
750 : {
751 : #if VFIO_ENABLED
752 0 : return g_vfio.enabled && !g_vfio.noiommu_enabled;
753 : #else
754 : return false;
755 : #endif
756 : }
757 :
758 : struct spdk_vtophys_pci_device {
759 : struct rte_pci_device *pci_device;
760 : TAILQ_ENTRY(spdk_vtophys_pci_device) tailq;
761 : };
762 :
763 : static pthread_mutex_t g_vtophys_pci_devices_mutex = PTHREAD_MUTEX_INITIALIZER;
764 : static TAILQ_HEAD(, spdk_vtophys_pci_device) g_vtophys_pci_devices =
765 : TAILQ_HEAD_INITIALIZER(g_vtophys_pci_devices);
766 :
767 : static struct spdk_mem_map *g_vtophys_map;
768 : static struct spdk_mem_map *g_phys_ref_map;
769 : static struct spdk_mem_map *g_numa_map;
770 :
771 : #if VFIO_ENABLED
772 : static int
773 0 : _vfio_iommu_map_dma(uint64_t vaddr, uint64_t iova, uint64_t size)
774 : {
775 0 : struct spdk_vfio_dma_map *dma_map;
776 0 : int ret;
777 :
778 0 : dma_map = calloc(1, sizeof(*dma_map));
779 0 : if (dma_map == NULL) {
780 0 : return -ENOMEM;
781 : }
782 :
783 0 : dma_map->map.argsz = sizeof(dma_map->map);
784 0 : dma_map->map.flags = VFIO_DMA_MAP_FLAG_READ | VFIO_DMA_MAP_FLAG_WRITE;
785 0 : dma_map->map.vaddr = vaddr;
786 0 : dma_map->map.iova = iova;
787 0 : dma_map->map.size = size;
788 :
789 0 : if (g_vfio.device_ref == 0) {
790 : /* VFIO requires at least one device (IOMMU group) to be added to
791 : * a VFIO container before it is possible to perform any IOMMU
792 : * operations on that container. This memory will be mapped once
793 : * the first device (IOMMU group) is hotplugged.
794 : *
795 : * Since the vfio container is managed internally by DPDK, it is
796 : * also possible that some device is already in that container, but
797 : * it's not managed by SPDK - e.g. an NIC attached internally
798 : * inside DPDK. We could map the memory straight away in such
799 : * scenario, but there's no need to do it. DPDK devices clearly
800 : * don't need our mappings and hence we defer the mapping
801 : * unconditionally until the first SPDK-managed device is
802 : * hotplugged.
803 : */
804 0 : goto out_insert;
805 : }
806 :
807 0 : ret = ioctl(g_vfio.fd, VFIO_IOMMU_MAP_DMA, &dma_map->map);
808 0 : if (ret) {
809 : /* There are cases the vfio container doesn't have IOMMU group, it's safe for this case */
810 0 : SPDK_NOTICELOG("Cannot set up DMA mapping, error %d, ignored\n", errno);
811 0 : }
812 :
813 : out_insert:
814 0 : TAILQ_INSERT_TAIL(&g_vfio.maps, dma_map, tailq);
815 0 : return 0;
816 0 : }
817 :
818 :
819 : static int
820 0 : vtophys_iommu_map_dma(uint64_t vaddr, uint64_t iova, uint64_t size)
821 : {
822 0 : uint64_t refcount;
823 0 : int ret;
824 :
825 0 : refcount = spdk_mem_map_translate(g_phys_ref_map, iova, NULL);
826 0 : assert(refcount < UINT64_MAX);
827 0 : if (refcount > 0) {
828 0 : spdk_mem_map_set_translation(g_phys_ref_map, iova, size, refcount + 1);
829 0 : return 0;
830 : }
831 :
832 0 : pthread_mutex_lock(&g_vfio.mutex);
833 0 : ret = _vfio_iommu_map_dma(vaddr, iova, size);
834 0 : pthread_mutex_unlock(&g_vfio.mutex);
835 0 : if (ret) {
836 0 : return ret;
837 : }
838 :
839 0 : spdk_mem_map_set_translation(g_phys_ref_map, iova, size, refcount + 1);
840 0 : return 0;
841 0 : }
842 :
843 : int
844 0 : vtophys_iommu_map_dma_bar(uint64_t vaddr, uint64_t iova, uint64_t size)
845 : {
846 0 : int ret;
847 :
848 0 : pthread_mutex_lock(&g_vfio.mutex);
849 0 : ret = _vfio_iommu_map_dma(vaddr, iova, size);
850 0 : pthread_mutex_unlock(&g_vfio.mutex);
851 :
852 0 : return ret;
853 0 : }
854 :
855 : static int
856 0 : _vfio_iommu_unmap_dma(struct spdk_vfio_dma_map *dma_map)
857 : {
858 0 : struct vfio_iommu_type1_dma_unmap unmap = {};
859 0 : int ret;
860 :
861 0 : if (g_vfio.device_ref == 0) {
862 : /* Memory is not mapped anymore, just remove it's references */
863 0 : goto out_remove;
864 : }
865 :
866 0 : unmap.argsz = sizeof(unmap);
867 0 : unmap.flags = 0;
868 0 : unmap.iova = dma_map->map.iova;
869 0 : unmap.size = dma_map->map.size;
870 0 : ret = ioctl(g_vfio.fd, VFIO_IOMMU_UNMAP_DMA, &unmap);
871 0 : if (ret) {
872 0 : SPDK_NOTICELOG("Cannot clear DMA mapping, error %d, ignored\n", errno);
873 0 : }
874 :
875 : out_remove:
876 0 : TAILQ_REMOVE(&g_vfio.maps, dma_map, tailq);
877 0 : free(dma_map);
878 0 : return 0;
879 0 : }
880 :
881 : static int
882 0 : vtophys_iommu_unmap_dma(uint64_t iova, uint64_t size)
883 : {
884 0 : struct spdk_vfio_dma_map *dma_map;
885 0 : uint64_t refcount;
886 0 : int ret;
887 :
888 0 : pthread_mutex_lock(&g_vfio.mutex);
889 0 : TAILQ_FOREACH(dma_map, &g_vfio.maps, tailq) {
890 0 : if (dma_map->map.iova == iova) {
891 0 : break;
892 : }
893 0 : }
894 :
895 0 : if (dma_map == NULL) {
896 0 : DEBUG_PRINT("Cannot clear DMA mapping for IOVA %"PRIx64" - it's not mapped\n", iova);
897 0 : pthread_mutex_unlock(&g_vfio.mutex);
898 0 : return -ENXIO;
899 : }
900 :
901 0 : refcount = spdk_mem_map_translate(g_phys_ref_map, iova, NULL);
902 0 : assert(refcount < UINT64_MAX);
903 0 : if (refcount > 0) {
904 0 : spdk_mem_map_set_translation(g_phys_ref_map, iova, size, refcount - 1);
905 0 : }
906 :
907 : /* We still have outstanding references, don't clear it. */
908 0 : if (refcount > 1) {
909 0 : pthread_mutex_unlock(&g_vfio.mutex);
910 0 : return 0;
911 : }
912 :
913 : /** don't support partial or multiple-page unmap for now */
914 0 : assert(dma_map->map.size == size);
915 :
916 0 : ret = _vfio_iommu_unmap_dma(dma_map);
917 0 : pthread_mutex_unlock(&g_vfio.mutex);
918 :
919 0 : return ret;
920 0 : }
921 :
922 : int
923 0 : vtophys_iommu_unmap_dma_bar(uint64_t vaddr)
924 : {
925 0 : struct spdk_vfio_dma_map *dma_map;
926 0 : int ret;
927 :
928 0 : pthread_mutex_lock(&g_vfio.mutex);
929 0 : TAILQ_FOREACH(dma_map, &g_vfio.maps, tailq) {
930 0 : if (dma_map->map.vaddr == vaddr) {
931 0 : break;
932 : }
933 0 : }
934 :
935 0 : if (dma_map == NULL) {
936 0 : DEBUG_PRINT("Cannot clear DMA mapping for address %"PRIx64" - it's not mapped\n", vaddr);
937 0 : pthread_mutex_unlock(&g_vfio.mutex);
938 0 : return -ENXIO;
939 : }
940 :
941 0 : ret = _vfio_iommu_unmap_dma(dma_map);
942 0 : pthread_mutex_unlock(&g_vfio.mutex);
943 0 : return ret;
944 0 : }
945 : #endif
946 :
947 : static uint64_t
948 0 : vtophys_get_paddr_memseg(uint64_t vaddr)
949 : {
950 0 : uintptr_t paddr;
951 0 : struct rte_memseg *seg;
952 :
953 0 : seg = rte_mem_virt2memseg((void *)(uintptr_t)vaddr, NULL);
954 0 : if (seg != NULL) {
955 0 : paddr = seg->iova;
956 0 : if (paddr == RTE_BAD_IOVA) {
957 0 : return SPDK_VTOPHYS_ERROR;
958 : }
959 0 : paddr += (vaddr - (uintptr_t)seg->addr);
960 0 : return paddr;
961 : }
962 :
963 0 : return SPDK_VTOPHYS_ERROR;
964 0 : }
965 :
966 : /* Try to get the paddr from /proc/self/pagemap */
967 : static uint64_t
968 0 : vtophys_get_paddr_pagemap(uint64_t vaddr)
969 : {
970 0 : uintptr_t paddr;
971 :
972 : /* Silence static analyzers */
973 0 : assert(vaddr != 0);
974 0 : paddr = rte_mem_virt2iova((void *)vaddr);
975 0 : if (paddr == RTE_BAD_IOVA) {
976 : /*
977 : * The vaddr may be valid but doesn't have a backing page
978 : * assigned yet. Touch the page to ensure a backing page
979 : * gets assigned, then try to translate again.
980 : */
981 0 : rte_atomic64_read((rte_atomic64_t *)vaddr);
982 0 : paddr = rte_mem_virt2iova((void *)vaddr);
983 0 : }
984 0 : if (paddr == RTE_BAD_IOVA) {
985 : /* Unable to get to the physical address. */
986 0 : return SPDK_VTOPHYS_ERROR;
987 : }
988 :
989 0 : return paddr;
990 0 : }
991 :
992 : static uint64_t
993 0 : pci_device_vtophys(struct rte_pci_device *dev, uint64_t vaddr, size_t len)
994 : {
995 0 : struct rte_mem_resource *res;
996 0 : uint64_t paddr;
997 0 : unsigned r;
998 :
999 0 : for (r = 0; r < PCI_MAX_RESOURCE; r++) {
1000 0 : res = dpdk_pci_device_get_mem_resource(dev, r);
1001 :
1002 0 : if (res->phys_addr == 0 || vaddr < (uint64_t)res->addr ||
1003 0 : (vaddr + len) >= (uint64_t)res->addr + res->len) {
1004 0 : continue;
1005 : }
1006 :
1007 : #if VFIO_ENABLED
1008 0 : if (spdk_iommu_is_enabled() && rte_eal_iova_mode() == RTE_IOVA_VA) {
1009 : /*
1010 : * The IOMMU is on and we're using IOVA == VA. The BAR was
1011 : * automatically registered when it was mapped, so just return
1012 : * the virtual address here.
1013 : */
1014 0 : return vaddr;
1015 : }
1016 : #endif
1017 0 : paddr = res->phys_addr + (vaddr - (uint64_t)res->addr);
1018 0 : return paddr;
1019 : }
1020 :
1021 0 : return SPDK_VTOPHYS_ERROR;
1022 0 : }
1023 :
1024 : /* Try to get the paddr from pci devices */
1025 : static uint64_t
1026 0 : vtophys_get_paddr_pci(uint64_t vaddr, size_t len)
1027 : {
1028 0 : struct spdk_vtophys_pci_device *vtophys_dev;
1029 0 : uintptr_t paddr;
1030 0 : struct rte_pci_device *dev;
1031 :
1032 0 : pthread_mutex_lock(&g_vtophys_pci_devices_mutex);
1033 0 : TAILQ_FOREACH(vtophys_dev, &g_vtophys_pci_devices, tailq) {
1034 0 : dev = vtophys_dev->pci_device;
1035 0 : paddr = pci_device_vtophys(dev, vaddr, len);
1036 0 : if (paddr != SPDK_VTOPHYS_ERROR) {
1037 0 : pthread_mutex_unlock(&g_vtophys_pci_devices_mutex);
1038 0 : return paddr;
1039 : }
1040 0 : }
1041 0 : pthread_mutex_unlock(&g_vtophys_pci_devices_mutex);
1042 :
1043 0 : return SPDK_VTOPHYS_ERROR;
1044 0 : }
1045 :
1046 : static int
1047 0 : vtophys_notify(void *cb_ctx, struct spdk_mem_map *map,
1048 : enum spdk_mem_map_notify_action action,
1049 : void *vaddr, size_t len)
1050 : {
1051 0 : int rc = 0;
1052 0 : uint64_t paddr;
1053 :
1054 0 : if ((uintptr_t)vaddr & ~MASK_256TB) {
1055 0 : DEBUG_PRINT("invalid usermode virtual address %p\n", vaddr);
1056 0 : return -EINVAL;
1057 : }
1058 :
1059 0 : if (((uintptr_t)vaddr & MASK_2MB) || (len & MASK_2MB)) {
1060 0 : DEBUG_PRINT("invalid parameters, vaddr=%p len=%ju\n",
1061 : vaddr, len);
1062 0 : return -EINVAL;
1063 : }
1064 :
1065 : /* Get the physical address from the DPDK memsegs */
1066 0 : paddr = vtophys_get_paddr_memseg((uint64_t)vaddr);
1067 :
1068 0 : switch (action) {
1069 : case SPDK_MEM_MAP_NOTIFY_REGISTER:
1070 0 : if (paddr == SPDK_VTOPHYS_ERROR) {
1071 : /* This is not an address that DPDK is managing. */
1072 :
1073 : /* Check if this is a PCI BAR. They need special handling */
1074 0 : paddr = vtophys_get_paddr_pci((uint64_t)vaddr, len);
1075 0 : if (paddr != SPDK_VTOPHYS_ERROR) {
1076 : /* Get paddr for each 2MB chunk in this address range */
1077 0 : while (len > 0) {
1078 0 : paddr = vtophys_get_paddr_pci((uint64_t)vaddr, VALUE_2MB);
1079 0 : if (paddr == SPDK_VTOPHYS_ERROR) {
1080 0 : DEBUG_PRINT("could not get phys addr for %p\n", vaddr);
1081 0 : return -EFAULT;
1082 : }
1083 :
1084 0 : rc = spdk_mem_map_set_translation(map, (uint64_t)vaddr, VALUE_2MB, paddr);
1085 0 : if (rc != 0) {
1086 0 : return rc;
1087 : }
1088 :
1089 0 : vaddr += VALUE_2MB;
1090 0 : len -= VALUE_2MB;
1091 : }
1092 :
1093 0 : return 0;
1094 : }
1095 :
1096 : #if VFIO_ENABLED
1097 0 : enum rte_iova_mode iova_mode;
1098 :
1099 0 : iova_mode = rte_eal_iova_mode();
1100 :
1101 0 : if (spdk_iommu_is_enabled() && iova_mode == RTE_IOVA_VA) {
1102 : /* We'll use the virtual address as the iova to match DPDK. */
1103 0 : paddr = (uint64_t)vaddr;
1104 0 : rc = vtophys_iommu_map_dma((uint64_t)vaddr, paddr, len);
1105 0 : if (rc) {
1106 0 : return -EFAULT;
1107 : }
1108 0 : while (len > 0) {
1109 0 : rc = spdk_mem_map_set_translation(map, (uint64_t)vaddr, VALUE_2MB, paddr);
1110 0 : if (rc != 0) {
1111 0 : return rc;
1112 : }
1113 0 : vaddr += VALUE_2MB;
1114 0 : paddr += VALUE_2MB;
1115 0 : len -= VALUE_2MB;
1116 : }
1117 0 : } else
1118 : #endif
1119 : {
1120 : /* Get the physical address from /proc/self/pagemap. */
1121 0 : paddr = vtophys_get_paddr_pagemap((uint64_t)vaddr);
1122 0 : if (paddr == SPDK_VTOPHYS_ERROR) {
1123 0 : DEBUG_PRINT("could not get phys addr for %p\n", vaddr);
1124 0 : return -EFAULT;
1125 : }
1126 :
1127 : /* Get paddr for each 2MB chunk in this address range */
1128 0 : while (len > 0) {
1129 : /* Get the physical address from /proc/self/pagemap. */
1130 0 : paddr = vtophys_get_paddr_pagemap((uint64_t)vaddr);
1131 :
1132 0 : if (paddr == SPDK_VTOPHYS_ERROR) {
1133 0 : DEBUG_PRINT("could not get phys addr for %p\n", vaddr);
1134 0 : return -EFAULT;
1135 : }
1136 :
1137 0 : if (paddr & MASK_2MB) {
1138 0 : DEBUG_PRINT("invalid paddr 0x%" PRIx64 " - must be 2MB aligned\n", paddr);
1139 0 : return -EINVAL;
1140 : }
1141 : #if VFIO_ENABLED
1142 : /* If the IOMMU is on, but DPDK is using iova-mode=pa, we want to register this memory
1143 : * with the IOMMU using the physical address to match. */
1144 0 : if (spdk_iommu_is_enabled()) {
1145 0 : rc = vtophys_iommu_map_dma((uint64_t)vaddr, paddr, VALUE_2MB);
1146 0 : if (rc) {
1147 0 : DEBUG_PRINT("Unable to assign vaddr %p to paddr 0x%" PRIx64 "\n", vaddr, paddr);
1148 0 : return -EFAULT;
1149 : }
1150 0 : }
1151 : #endif
1152 :
1153 0 : rc = spdk_mem_map_set_translation(map, (uint64_t)vaddr, VALUE_2MB, paddr);
1154 0 : if (rc != 0) {
1155 0 : return rc;
1156 : }
1157 :
1158 0 : vaddr += VALUE_2MB;
1159 0 : len -= VALUE_2MB;
1160 : }
1161 : }
1162 0 : } else {
1163 : /* This is an address managed by DPDK. Just setup the translations. */
1164 0 : while (len > 0) {
1165 0 : paddr = vtophys_get_paddr_memseg((uint64_t)vaddr);
1166 0 : if (paddr == SPDK_VTOPHYS_ERROR) {
1167 0 : DEBUG_PRINT("could not get phys addr for %p\n", vaddr);
1168 0 : return -EFAULT;
1169 : }
1170 :
1171 0 : rc = spdk_mem_map_set_translation(map, (uint64_t)vaddr, VALUE_2MB, paddr);
1172 0 : if (rc != 0) {
1173 0 : return rc;
1174 : }
1175 :
1176 0 : vaddr += VALUE_2MB;
1177 0 : len -= VALUE_2MB;
1178 : }
1179 : }
1180 :
1181 0 : break;
1182 : case SPDK_MEM_MAP_NOTIFY_UNREGISTER:
1183 : #if VFIO_ENABLED
1184 0 : if (paddr == SPDK_VTOPHYS_ERROR) {
1185 : /*
1186 : * This is not an address that DPDK is managing.
1187 : */
1188 :
1189 : /* Check if this is a PCI BAR. They need special handling */
1190 0 : paddr = vtophys_get_paddr_pci((uint64_t)vaddr, len);
1191 0 : if (paddr != SPDK_VTOPHYS_ERROR) {
1192 : /* Get paddr for each 2MB chunk in this address range */
1193 0 : while (len > 0) {
1194 0 : paddr = vtophys_get_paddr_pci((uint64_t)vaddr, VALUE_2MB);
1195 0 : if (paddr == SPDK_VTOPHYS_ERROR) {
1196 0 : DEBUG_PRINT("could not get phys addr for %p\n", vaddr);
1197 0 : return -EFAULT;
1198 : }
1199 :
1200 0 : rc = spdk_mem_map_clear_translation(map, (uint64_t)vaddr, VALUE_2MB);
1201 0 : if (rc != 0) {
1202 0 : return rc;
1203 : }
1204 :
1205 0 : vaddr += VALUE_2MB;
1206 0 : len -= VALUE_2MB;
1207 : }
1208 :
1209 0 : return 0;
1210 : }
1211 :
1212 : /* If vfio is enabled,
1213 : * we need to unmap the range from the IOMMU
1214 : */
1215 0 : if (spdk_iommu_is_enabled()) {
1216 0 : uint64_t buffer_len = len;
1217 0 : uint8_t *va = vaddr;
1218 0 : enum rte_iova_mode iova_mode;
1219 :
1220 0 : iova_mode = rte_eal_iova_mode();
1221 : /*
1222 : * In virtual address mode, the region is contiguous and can be done in
1223 : * one unmap.
1224 : */
1225 0 : if (iova_mode == RTE_IOVA_VA) {
1226 0 : paddr = spdk_mem_map_translate(map, (uint64_t)va, &buffer_len);
1227 0 : if (buffer_len != len || paddr != (uintptr_t)va) {
1228 0 : DEBUG_PRINT("Unmapping %p with length %lu failed because "
1229 : "translation had address 0x%" PRIx64 " and length %lu\n",
1230 : va, len, paddr, buffer_len);
1231 0 : return -EINVAL;
1232 : }
1233 0 : rc = vtophys_iommu_unmap_dma(paddr, len);
1234 0 : if (rc) {
1235 0 : DEBUG_PRINT("Failed to iommu unmap paddr 0x%" PRIx64 "\n", paddr);
1236 0 : return -EFAULT;
1237 : }
1238 0 : } else if (iova_mode == RTE_IOVA_PA) {
1239 : /* Get paddr for each 2MB chunk in this address range */
1240 0 : while (buffer_len > 0) {
1241 0 : paddr = spdk_mem_map_translate(map, (uint64_t)va, NULL);
1242 :
1243 0 : if (paddr == SPDK_VTOPHYS_ERROR || buffer_len < VALUE_2MB) {
1244 0 : DEBUG_PRINT("could not get phys addr for %p\n", va);
1245 0 : return -EFAULT;
1246 : }
1247 :
1248 0 : rc = vtophys_iommu_unmap_dma(paddr, VALUE_2MB);
1249 0 : if (rc) {
1250 0 : DEBUG_PRINT("Failed to iommu unmap paddr 0x%" PRIx64 "\n", paddr);
1251 0 : return -EFAULT;
1252 : }
1253 :
1254 0 : va += VALUE_2MB;
1255 0 : buffer_len -= VALUE_2MB;
1256 : }
1257 0 : }
1258 0 : }
1259 0 : }
1260 : #endif
1261 0 : while (len > 0) {
1262 0 : rc = spdk_mem_map_clear_translation(map, (uint64_t)vaddr, VALUE_2MB);
1263 0 : if (rc != 0) {
1264 0 : return rc;
1265 : }
1266 :
1267 0 : vaddr += VALUE_2MB;
1268 0 : len -= VALUE_2MB;
1269 : }
1270 :
1271 0 : break;
1272 : default:
1273 0 : SPDK_UNREACHABLE();
1274 : }
1275 :
1276 0 : return rc;
1277 0 : }
1278 :
1279 : static int
1280 0 : numa_notify(void *cb_ctx, struct spdk_mem_map *map,
1281 : enum spdk_mem_map_notify_action action,
1282 : void *vaddr, size_t len)
1283 : {
1284 0 : struct rte_memseg *seg;
1285 :
1286 : /* We always return 0 from here, even if we aren't able to get a
1287 : * memseg for the address. This can happen in non-DPDK memory
1288 : * registration paths, for example vhost or vfio-user. That is OK,
1289 : * spdk_mem_get_numa_id() just returns SPDK_ENV_NUMA_ID_ANY for
1290 : * that kind of memory. If we return an error here, the
1291 : * spdk_mem_register() from vhost or vfio-user would fail which is
1292 : * not what we want.
1293 : */
1294 0 : seg = rte_mem_virt2memseg(vaddr, NULL);
1295 0 : if (seg == NULL) {
1296 0 : return 0;
1297 : }
1298 :
1299 0 : switch (action) {
1300 : case SPDK_MEM_MAP_NOTIFY_REGISTER:
1301 0 : spdk_mem_map_set_translation(map, (uint64_t)vaddr, len, seg->socket_id);
1302 0 : break;
1303 : case SPDK_MEM_MAP_NOTIFY_UNREGISTER:
1304 0 : spdk_mem_map_clear_translation(map, (uint64_t)vaddr, len);
1305 0 : break;
1306 : default:
1307 0 : break;
1308 : }
1309 :
1310 0 : return 0;
1311 0 : }
1312 :
1313 : static int
1314 0 : vtophys_check_contiguous_entries(uint64_t paddr1, uint64_t paddr2)
1315 : {
1316 : /* This function is always called with paddrs for two subsequent
1317 : * 2MB chunks in virtual address space, so those chunks will be only
1318 : * physically contiguous if the physical addresses are 2MB apart
1319 : * from each other as well.
1320 : */
1321 0 : return (paddr2 - paddr1 == VALUE_2MB);
1322 : }
1323 :
1324 : #if VFIO_ENABLED
1325 :
1326 : static bool
1327 0 : vfio_enabled(void)
1328 : {
1329 0 : return rte_vfio_is_enabled("vfio_pci");
1330 : }
1331 :
1332 : /* Check if IOMMU is enabled on the system */
1333 : static bool
1334 0 : has_iommu_groups(void)
1335 : {
1336 0 : int count = 0;
1337 0 : DIR *dir = opendir("/sys/kernel/iommu_groups");
1338 :
1339 0 : if (dir == NULL) {
1340 0 : return false;
1341 : }
1342 :
1343 0 : while (count < 3 && readdir(dir) != NULL) {
1344 0 : count++;
1345 : }
1346 :
1347 0 : closedir(dir);
1348 : /* there will always be ./ and ../ entries */
1349 0 : return count > 2;
1350 0 : }
1351 :
1352 : static bool
1353 0 : vfio_noiommu_enabled(void)
1354 : {
1355 0 : return rte_vfio_noiommu_is_enabled();
1356 : }
1357 :
1358 : static void
1359 0 : vtophys_iommu_init(void)
1360 : {
1361 0 : char proc_fd_path[PATH_MAX + 1];
1362 0 : char link_path[PATH_MAX + 1];
1363 0 : const char vfio_path[] = "/dev/vfio/vfio";
1364 0 : DIR *dir;
1365 0 : struct dirent *d;
1366 :
1367 0 : if (!vfio_enabled()) {
1368 0 : return;
1369 : }
1370 :
1371 0 : if (vfio_noiommu_enabled()) {
1372 0 : g_vfio.noiommu_enabled = true;
1373 0 : } else if (!has_iommu_groups()) {
1374 0 : return;
1375 : }
1376 :
1377 0 : dir = opendir("/proc/self/fd");
1378 0 : if (!dir) {
1379 0 : DEBUG_PRINT("Failed to open /proc/self/fd (%d)\n", errno);
1380 0 : return;
1381 : }
1382 :
1383 0 : while ((d = readdir(dir)) != NULL) {
1384 0 : if (d->d_type != DT_LNK) {
1385 0 : continue;
1386 : }
1387 :
1388 0 : snprintf(proc_fd_path, sizeof(proc_fd_path), "/proc/self/fd/%s", d->d_name);
1389 0 : if (readlink(proc_fd_path, link_path, sizeof(link_path)) != (sizeof(vfio_path) - 1)) {
1390 0 : continue;
1391 : }
1392 :
1393 0 : if (memcmp(link_path, vfio_path, sizeof(vfio_path) - 1) == 0) {
1394 0 : sscanf(d->d_name, "%d", &g_vfio.fd);
1395 0 : break;
1396 : }
1397 : }
1398 :
1399 0 : closedir(dir);
1400 :
1401 0 : if (g_vfio.fd < 0) {
1402 0 : DEBUG_PRINT("Failed to discover DPDK VFIO container fd.\n");
1403 0 : return;
1404 : }
1405 :
1406 0 : g_vfio.enabled = true;
1407 :
1408 0 : return;
1409 0 : }
1410 :
1411 : #endif
1412 :
1413 : void
1414 0 : vtophys_pci_device_added(struct rte_pci_device *pci_device)
1415 : {
1416 0 : struct spdk_vtophys_pci_device *vtophys_dev;
1417 :
1418 0 : pthread_mutex_lock(&g_vtophys_pci_devices_mutex);
1419 :
1420 0 : vtophys_dev = calloc(1, sizeof(*vtophys_dev));
1421 0 : if (vtophys_dev) {
1422 0 : vtophys_dev->pci_device = pci_device;
1423 0 : TAILQ_INSERT_TAIL(&g_vtophys_pci_devices, vtophys_dev, tailq);
1424 0 : } else {
1425 0 : DEBUG_PRINT("Memory allocation error\n");
1426 : }
1427 0 : pthread_mutex_unlock(&g_vtophys_pci_devices_mutex);
1428 :
1429 : #if VFIO_ENABLED
1430 0 : struct spdk_vfio_dma_map *dma_map;
1431 0 : int ret;
1432 :
1433 0 : if (!g_vfio.enabled) {
1434 0 : return;
1435 : }
1436 :
1437 0 : pthread_mutex_lock(&g_vfio.mutex);
1438 0 : g_vfio.device_ref++;
1439 0 : if (g_vfio.device_ref > 1) {
1440 0 : pthread_mutex_unlock(&g_vfio.mutex);
1441 0 : return;
1442 : }
1443 :
1444 : /* This is the first SPDK device using DPDK vfio. This means that the first
1445 : * IOMMU group might have been just been added to the DPDK vfio container.
1446 : * From this point it is certain that the memory can be mapped now.
1447 : */
1448 0 : TAILQ_FOREACH(dma_map, &g_vfio.maps, tailq) {
1449 0 : ret = ioctl(g_vfio.fd, VFIO_IOMMU_MAP_DMA, &dma_map->map);
1450 0 : if (ret) {
1451 0 : DEBUG_PRINT("Cannot update DMA mapping, error %d\n", errno);
1452 0 : break;
1453 : }
1454 0 : }
1455 0 : pthread_mutex_unlock(&g_vfio.mutex);
1456 : #endif
1457 0 : }
1458 :
1459 : void
1460 0 : vtophys_pci_device_removed(struct rte_pci_device *pci_device)
1461 : {
1462 0 : struct spdk_vtophys_pci_device *vtophys_dev;
1463 :
1464 0 : pthread_mutex_lock(&g_vtophys_pci_devices_mutex);
1465 0 : TAILQ_FOREACH(vtophys_dev, &g_vtophys_pci_devices, tailq) {
1466 0 : if (vtophys_dev->pci_device == pci_device) {
1467 0 : TAILQ_REMOVE(&g_vtophys_pci_devices, vtophys_dev, tailq);
1468 0 : free(vtophys_dev);
1469 0 : break;
1470 : }
1471 0 : }
1472 0 : pthread_mutex_unlock(&g_vtophys_pci_devices_mutex);
1473 :
1474 : #if VFIO_ENABLED
1475 0 : struct spdk_vfio_dma_map *dma_map;
1476 0 : int ret;
1477 :
1478 0 : if (!g_vfio.enabled) {
1479 0 : return;
1480 : }
1481 :
1482 0 : pthread_mutex_lock(&g_vfio.mutex);
1483 0 : assert(g_vfio.device_ref > 0);
1484 0 : g_vfio.device_ref--;
1485 0 : if (g_vfio.device_ref > 0) {
1486 0 : pthread_mutex_unlock(&g_vfio.mutex);
1487 0 : return;
1488 : }
1489 :
1490 : /* This is the last SPDK device using DPDK vfio. If DPDK doesn't have
1491 : * any additional devices using it's vfio container, all the mappings
1492 : * will be automatically removed by the Linux vfio driver. We unmap
1493 : * the memory manually to be able to easily re-map it later regardless
1494 : * of other, external factors.
1495 : */
1496 0 : TAILQ_FOREACH(dma_map, &g_vfio.maps, tailq) {
1497 0 : struct vfio_iommu_type1_dma_unmap unmap = {};
1498 0 : unmap.argsz = sizeof(unmap);
1499 0 : unmap.flags = 0;
1500 0 : unmap.iova = dma_map->map.iova;
1501 0 : unmap.size = dma_map->map.size;
1502 0 : ret = ioctl(g_vfio.fd, VFIO_IOMMU_UNMAP_DMA, &unmap);
1503 0 : if (ret) {
1504 0 : DEBUG_PRINT("Cannot unmap DMA memory, error %d\n", errno);
1505 0 : break;
1506 : }
1507 0 : }
1508 0 : pthread_mutex_unlock(&g_vfio.mutex);
1509 : #endif
1510 0 : }
1511 :
1512 : int
1513 0 : vtophys_init(void)
1514 : {
1515 0 : const struct spdk_mem_map_ops vtophys_map_ops = {
1516 : .notify_cb = vtophys_notify,
1517 : .are_contiguous = vtophys_check_contiguous_entries,
1518 : };
1519 :
1520 0 : const struct spdk_mem_map_ops phys_ref_map_ops = {
1521 : .notify_cb = NULL,
1522 : .are_contiguous = NULL,
1523 : };
1524 :
1525 0 : const struct spdk_mem_map_ops numa_map_ops = {
1526 : .notify_cb = numa_notify,
1527 : .are_contiguous = NULL,
1528 : };
1529 :
1530 : #if VFIO_ENABLED
1531 0 : vtophys_iommu_init();
1532 : #endif
1533 :
1534 0 : g_phys_ref_map = spdk_mem_map_alloc(0, &phys_ref_map_ops, NULL);
1535 0 : if (g_phys_ref_map == NULL) {
1536 0 : DEBUG_PRINT("phys_ref map allocation failed.\n");
1537 0 : return -ENOMEM;
1538 : }
1539 :
1540 0 : g_numa_map = spdk_mem_map_alloc(SPDK_ENV_NUMA_ID_ANY, &numa_map_ops, NULL);
1541 0 : if (g_numa_map == NULL) {
1542 0 : DEBUG_PRINT("numa map allocation failed.\n");
1543 0 : spdk_mem_map_free(&g_phys_ref_map);
1544 0 : return -ENOMEM;
1545 : }
1546 :
1547 0 : if (g_huge_pages) {
1548 0 : g_vtophys_map = spdk_mem_map_alloc(SPDK_VTOPHYS_ERROR, &vtophys_map_ops, NULL);
1549 0 : if (g_vtophys_map == NULL) {
1550 0 : DEBUG_PRINT("vtophys map allocation failed\n");
1551 0 : spdk_mem_map_free(&g_numa_map);
1552 0 : spdk_mem_map_free(&g_phys_ref_map);
1553 0 : return -ENOMEM;
1554 : }
1555 0 : }
1556 0 : return 0;
1557 0 : }
1558 :
1559 : uint64_t
1560 0 : spdk_vtophys(const void *buf, uint64_t *size)
1561 : {
1562 0 : uint64_t vaddr, paddr_2mb;
1563 :
1564 0 : if (!g_huge_pages) {
1565 0 : return SPDK_VTOPHYS_ERROR;
1566 : }
1567 :
1568 0 : vaddr = (uint64_t)buf;
1569 0 : paddr_2mb = spdk_mem_map_translate(g_vtophys_map, vaddr, size);
1570 :
1571 : /*
1572 : * SPDK_VTOPHYS_ERROR has all bits set, so if the lookup returned SPDK_VTOPHYS_ERROR,
1573 : * we will still bitwise-or it with the buf offset below, but the result will still be
1574 : * SPDK_VTOPHYS_ERROR. However now that we do + rather than | (due to PCI vtophys being
1575 : * unaligned) we must now check the return value before addition.
1576 : */
1577 : SPDK_STATIC_ASSERT(SPDK_VTOPHYS_ERROR == UINT64_C(-1), "SPDK_VTOPHYS_ERROR should be all 1s");
1578 0 : if (paddr_2mb == SPDK_VTOPHYS_ERROR) {
1579 0 : return SPDK_VTOPHYS_ERROR;
1580 : } else {
1581 0 : return paddr_2mb + (vaddr & MASK_2MB);
1582 : }
1583 0 : }
1584 :
1585 : int32_t
1586 0 : spdk_mem_get_numa_id(const void *buf, uint64_t *size)
1587 : {
1588 0 : return spdk_mem_map_translate(g_numa_map, (uint64_t)buf, size);
1589 : }
1590 :
1591 : int
1592 0 : spdk_mem_get_fd_and_offset(void *vaddr, uint64_t *offset)
1593 : {
1594 0 : struct rte_memseg *seg;
1595 0 : int ret, fd;
1596 :
1597 0 : seg = rte_mem_virt2memseg(vaddr, NULL);
1598 0 : if (!seg) {
1599 0 : SPDK_ERRLOG("memory %p doesn't exist\n", vaddr);
1600 0 : return -ENOENT;
1601 : }
1602 :
1603 0 : fd = rte_memseg_get_fd_thread_unsafe(seg);
1604 0 : if (fd < 0) {
1605 0 : return fd;
1606 : }
1607 :
1608 0 : ret = rte_memseg_get_fd_offset_thread_unsafe(seg, offset);
1609 0 : if (ret < 0) {
1610 0 : return ret;
1611 : }
1612 :
1613 0 : return fd;
1614 0 : }
1615 :
1616 : void
1617 0 : mem_disable_huge_pages(void)
1618 : {
1619 0 : g_huge_pages = false;
1620 0 : }
|
