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
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
|
// SPDX-License-Identifier: GPL-2.0
/*
* GCIP IOMMU domain allocator.
*
* Copyright (C) 2022 Google LLC
*/
#include <linux/device.h>
#include <linux/idr.h>
#include <linux/iommu.h>
#include <linux/vmalloc.h>
#include <gcip/gcip-domain-pool.h>
int gcip_domain_pool_init(struct device *dev, struct gcip_domain_pool *pool, unsigned int size)
{
unsigned int i;
struct iommu_domain *domain;
pool->size = size;
pool->dev = dev;
if (!size)
return 0;
dev_dbg(pool->dev, "Initializing domain pool with %u domains\n", size);
ida_init(&pool->idp);
pool->array = vzalloc(sizeof(*pool->array) * size);
if (!pool->array) {
ida_destroy(&pool->idp);
return -ENOMEM;
}
for (i = 0; i < size; i++) {
domain = iommu_domain_alloc(dev->bus);
if (!domain) {
dev_err(pool->dev, "Failed to allocate iommu domain %d of %u\n", i + 1,
size);
gcip_domain_pool_destroy(pool);
return -ENOMEM;
}
pool->array[i] = domain;
}
return 0;
}
struct iommu_domain *gcip_domain_pool_alloc(struct gcip_domain_pool *pool)
{
int id;
if (!pool->size)
return iommu_domain_alloc(pool->dev->bus);
id = ida_alloc_max(&pool->idp, pool->size - 1, GFP_KERNEL);
if (id < 0) {
dev_err(pool->dev, "No more domains available from pool of size %u\n", pool->size);
return NULL;
}
dev_dbg(pool->dev, "Allocated domain from pool with id = %d\n", id);
return pool->array[id];
}
void gcip_domain_pool_free(struct gcip_domain_pool *pool, struct iommu_domain *domain)
{
int id;
if (!pool->size) {
iommu_domain_free(domain);
return;
}
for (id = 0; id < pool->size; id++) {
if (pool->array[id] == domain) {
dev_dbg(pool->dev, "Released domain from pool with id = %d\n", id);
ida_free(&pool->idp, id);
return;
}
}
dev_err(pool->dev, "Domain not found in pool\n");
}
void gcip_domain_pool_destroy(struct gcip_domain_pool *pool)
{
int i;
if (!pool->size)
return;
dev_dbg(pool->dev, "Destroying domain pool with %u domains\n", pool->size);
for (i = 0; i < pool->size; i++) {
if (pool->array[i])
iommu_domain_free(pool->array[i]);
}
ida_destroy(&pool->idp);
vfree(pool->array);
}
|