1 /* SPDX-License-Identifier: GPL-2.0 OR Linux-OpenIB */
2 /*
3 * Copyright (c) 2014 Mellanox Technologies. All rights reserved.
4 */
5
6 #ifndef IB_UMEM_ODP_H
7 #define IB_UMEM_ODP_H
8
9 #include <rdma/ib_umem.h>
10 #include <rdma/ib_verbs.h>
11
12 struct ib_umem_odp {
13 struct ib_umem umem;
14 struct mmu_interval_notifier notifier;
15 struct pid *tgid;
16
17 /* An array of the pfns included in the on-demand paging umem. */
18 unsigned long *pfn_list;
19
20 /*
21 * An array with DMA addresses mapped for pfns in pfn_list.
22 * The lower two bits designate access permissions.
23 * See ODP_READ_ALLOWED_BIT and ODP_WRITE_ALLOWED_BIT.
24 */
25 dma_addr_t *dma_list;
26 /*
27 * The umem_mutex protects the page_list and dma_list fields of an ODP
28 * umem, allowing only a single thread to map/unmap pages. The mutex
29 * also protects access to the mmu notifier counters.
30 */
31 struct mutex umem_mutex;
32 void *private; /* for the HW driver to use. */
33
34 int npages;
35
36 /*
37 * An implicit odp umem cannot be DMA mapped, has 0 length, and serves
38 * only as an anchor for the driver to hold onto the per_mm. FIXME:
39 * This should be removed and drivers should work with the per_mm
40 * directly.
41 */
42 bool is_implicit_odp;
43
44 unsigned int page_shift;
45 };
46
to_ib_umem_odp(struct ib_umem * umem)47 static inline struct ib_umem_odp *to_ib_umem_odp(struct ib_umem *umem)
48 {
49 return container_of(umem, struct ib_umem_odp, umem);
50 }
51
52 /* Returns the first page of an ODP umem. */
ib_umem_start(struct ib_umem_odp * umem_odp)53 static inline unsigned long ib_umem_start(struct ib_umem_odp *umem_odp)
54 {
55 return umem_odp->notifier.interval_tree.start;
56 }
57
58 /* Returns the address of the page after the last one of an ODP umem. */
ib_umem_end(struct ib_umem_odp * umem_odp)59 static inline unsigned long ib_umem_end(struct ib_umem_odp *umem_odp)
60 {
61 return umem_odp->notifier.interval_tree.last + 1;
62 }
63
ib_umem_odp_num_pages(struct ib_umem_odp * umem_odp)64 static inline size_t ib_umem_odp_num_pages(struct ib_umem_odp *umem_odp)
65 {
66 return (ib_umem_end(umem_odp) - ib_umem_start(umem_odp)) >>
67 umem_odp->page_shift;
68 }
69
70 /*
71 * The lower 2 bits of the DMA address signal the R/W permissions for
72 * the entry. To upgrade the permissions, provide the appropriate
73 * bitmask to the map_dma_pages function.
74 *
75 * Be aware that upgrading a mapped address might result in change of
76 * the DMA address for the page.
77 */
78 #define ODP_READ_ALLOWED_BIT (1<<0ULL)
79 #define ODP_WRITE_ALLOWED_BIT (1<<1ULL)
80
81 #define ODP_DMA_ADDR_MASK (~(ODP_READ_ALLOWED_BIT | ODP_WRITE_ALLOWED_BIT))
82
83 #ifdef CONFIG_INFINIBAND_ON_DEMAND_PAGING
84
85 struct ib_umem_odp *
86 ib_umem_odp_get(struct ib_device *device, unsigned long addr, size_t size,
87 int access, const struct mmu_interval_notifier_ops *ops);
88 struct ib_umem_odp *ib_umem_odp_alloc_implicit(struct ib_device *device,
89 int access);
90 struct ib_umem_odp *
91 ib_umem_odp_alloc_child(struct ib_umem_odp *root_umem, unsigned long addr,
92 size_t size,
93 const struct mmu_interval_notifier_ops *ops);
94 void ib_umem_odp_release(struct ib_umem_odp *umem_odp);
95
96 int ib_umem_odp_map_dma_and_lock(struct ib_umem_odp *umem_odp, u64 start_offset,
97 u64 bcnt, u64 access_mask, bool fault);
98
99 void ib_umem_odp_unmap_dma_pages(struct ib_umem_odp *umem_odp, u64 start_offset,
100 u64 bound);
101
102 #else /* CONFIG_INFINIBAND_ON_DEMAND_PAGING */
103
104 static inline struct ib_umem_odp *
ib_umem_odp_get(struct ib_device * device,unsigned long addr,size_t size,int access,const struct mmu_interval_notifier_ops * ops)105 ib_umem_odp_get(struct ib_device *device, unsigned long addr, size_t size,
106 int access, const struct mmu_interval_notifier_ops *ops)
107 {
108 return ERR_PTR(-EINVAL);
109 }
110
ib_umem_odp_release(struct ib_umem_odp * umem_odp)111 static inline void ib_umem_odp_release(struct ib_umem_odp *umem_odp) {}
112
113 #endif /* CONFIG_INFINIBAND_ON_DEMAND_PAGING */
114
115 #endif /* IB_UMEM_ODP_H */
116