1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
|
// SPDX-License-Identifier: GPL-2.0 OR Linux-OpenIB
/*
* Copyright (c) 2018-2025 Advanced Micro Devices, Inc. All rights reserved.
*/
#include <infiniband/driver.h>
#include <sys/mman.h>
#include "ionic_memory.h"
#define IONIC_ANON_MFLAGS (MAP_PRIVATE | MAP_ANONYMOUS)
#define IONIC_ANON_MPROT (PROT_READ | PROT_WRITE)
#define IONIC_DEV_MFLAGS MAP_SHARED
#define IONIC_DEV_MPROT PROT_WRITE
void *ionic_map_anon(size_t size)
{
void *ptr;
int rc;
ptr = mmap(NULL, size, IONIC_ANON_MPROT, IONIC_ANON_MFLAGS, -1, 0);
if (ptr == MAP_FAILED)
return NULL;
rc = ibv_dontfork_range(ptr, size);
if (rc) {
munmap(ptr, size);
errno = rc;
return NULL;
}
return ptr;
}
void *ionic_map_device(size_t size, int fd, size_t offset)
{
void *ptr;
int rc;
ptr = mmap(NULL, size, IONIC_DEV_MPROT, IONIC_DEV_MFLAGS, fd, offset);
if (ptr == MAP_FAILED)
return NULL;
rc = ibv_dontfork_range(ptr, size);
if (rc) {
munmap(ptr, size);
errno = rc;
return NULL;
}
return ptr;
}
void ionic_unmap(void *ptr, size_t size)
{
if (ptr) {
ibv_dofork_range(ptr, size);
munmap(ptr, size);
}
}
|
