1 // SPDX-License-Identifier: GPL-2.0-only
2 /* Copyright (C) 2020 Red Hat, Inc.
3 * Author: Jason Wang <jasowang@redhat.com>
5 * IOTLB implementation for vhost.
7 #include <linux/slab.h>
8 #include <linux/vhost_iotlb.h>
9 #include <linux/module.h>
11 #define MOD_VERSION "0.1"
12 #define MOD_DESC "VHOST IOTLB"
13 #define MOD_AUTHOR "Jason Wang <jasowang@redhat.com>"
14 #define MOD_LICENSE "GPL v2"
16 #define START(map) ((map)->start)
17 #define LAST(map) ((map)->last)
19 INTERVAL_TREE_DEFINE(struct vhost_iotlb_map
,
20 rb
, __u64
, __subtree_last
,
21 START
, LAST
, static inline, vhost_iotlb_itree
);
24 * vhost_iotlb_map_free - remove a map node and free it
26 * @map: the map that want to be remove and freed
28 void vhost_iotlb_map_free(struct vhost_iotlb
*iotlb
,
29 struct vhost_iotlb_map
*map
)
31 vhost_iotlb_itree_remove(map
, &iotlb
->root
);
36 EXPORT_SYMBOL_GPL(vhost_iotlb_map_free
);
39 * vhost_iotlb_add_range_ctx - add a new range to vhost IOTLB
41 * @start: start of the IOVA range
42 * @last: last of IOVA range
43 * @addr: the address that is mapped to @start
44 * @perm: access permission of this range
45 * @opaque: the opaque pointer for the new mapping
47 * Returns an error last is smaller than start or memory allocation
50 int vhost_iotlb_add_range_ctx(struct vhost_iotlb
*iotlb
,
52 u64 addr
, unsigned int perm
,
55 struct vhost_iotlb_map
*map
;
61 iotlb
->nmaps
== iotlb
->limit
&&
62 iotlb
->flags
& VHOST_IOTLB_FLAG_RETIRE
) {
63 map
= list_first_entry(&iotlb
->list
, typeof(*map
), link
);
64 vhost_iotlb_map_free(iotlb
, map
);
67 map
= kmalloc(sizeof(*map
), GFP_ATOMIC
);
72 map
->size
= last
- start
+ 1;
79 vhost_iotlb_itree_insert(map
, &iotlb
->root
);
81 INIT_LIST_HEAD(&map
->link
);
82 list_add_tail(&map
->link
, &iotlb
->list
);
86 EXPORT_SYMBOL_GPL(vhost_iotlb_add_range_ctx
);
88 int vhost_iotlb_add_range(struct vhost_iotlb
*iotlb
,
90 u64 addr
, unsigned int perm
)
92 return vhost_iotlb_add_range_ctx(iotlb
, start
, last
,
95 EXPORT_SYMBOL_GPL(vhost_iotlb_add_range
);
98 * vhost_iotlb_del_range - delete overlapped ranges from vhost IOTLB
100 * @start: start of the IOVA range
101 * @last: last of IOVA range
103 void vhost_iotlb_del_range(struct vhost_iotlb
*iotlb
, u64 start
, u64 last
)
105 struct vhost_iotlb_map
*map
;
107 while ((map
= vhost_iotlb_itree_iter_first(&iotlb
->root
,
109 vhost_iotlb_map_free(iotlb
, map
);
111 EXPORT_SYMBOL_GPL(vhost_iotlb_del_range
);
114 * vhost_iotlb_alloc - add a new vhost IOTLB
115 * @limit: maximum number of IOTLB entries
116 * @flags: VHOST_IOTLB_FLAG_XXX
118 * Returns an error is memory allocation fails
120 struct vhost_iotlb
*vhost_iotlb_alloc(unsigned int limit
, unsigned int flags
)
122 struct vhost_iotlb
*iotlb
= kzalloc(sizeof(*iotlb
), GFP_KERNEL
);
127 iotlb
->root
= RB_ROOT_CACHED
;
128 iotlb
->limit
= limit
;
130 iotlb
->flags
= flags
;
131 INIT_LIST_HEAD(&iotlb
->list
);
135 EXPORT_SYMBOL_GPL(vhost_iotlb_alloc
);
138 * vhost_iotlb_reset - reset vhost IOTLB (free all IOTLB entries)
139 * @iotlb: the IOTLB to be reset
141 void vhost_iotlb_reset(struct vhost_iotlb
*iotlb
)
143 vhost_iotlb_del_range(iotlb
, 0ULL, 0ULL - 1);
145 EXPORT_SYMBOL_GPL(vhost_iotlb_reset
);
148 * vhost_iotlb_free - reset and free vhost IOTLB
149 * @iotlb: the IOTLB to be freed
151 void vhost_iotlb_free(struct vhost_iotlb
*iotlb
)
154 vhost_iotlb_reset(iotlb
);
158 EXPORT_SYMBOL_GPL(vhost_iotlb_free
);
161 * vhost_iotlb_itree_first - return the first overlapped range
163 * @start: start of IOVA range
164 * @last: last byte in IOVA range
166 struct vhost_iotlb_map
*
167 vhost_iotlb_itree_first(struct vhost_iotlb
*iotlb
, u64 start
, u64 last
)
169 return vhost_iotlb_itree_iter_first(&iotlb
->root
, start
, last
);
171 EXPORT_SYMBOL_GPL(vhost_iotlb_itree_first
);
174 * vhost_iotlb_itree_next - return the next overlapped range
175 * @map: the starting map node
176 * @start: start of IOVA range
177 * @last: last byte IOVA range
179 struct vhost_iotlb_map
*
180 vhost_iotlb_itree_next(struct vhost_iotlb_map
*map
, u64 start
, u64 last
)
182 return vhost_iotlb_itree_iter_next(map
, start
, last
);
184 EXPORT_SYMBOL_GPL(vhost_iotlb_itree_next
);
186 MODULE_VERSION(MOD_VERSION
);
187 MODULE_DESCRIPTION(MOD_DESC
);
188 MODULE_AUTHOR(MOD_AUTHOR
);
189 MODULE_LICENSE(MOD_LICENSE
);