19 int msi_bitmap_alloc_hwirqs(struct msi_bitmap *bmp, int num)  in msi_bitmap_alloc_hwirqs()  argument
24 	spin_lock_irqsave(&bmp->lock, flags);  in msi_bitmap_alloc_hwirqs()
26 	offset = bitmap_find_next_zero_area(bmp->bitmap, bmp->irq_count, 0,  in msi_bitmap_alloc_hwirqs()
28 	if (offset > bmp->irq_count)  in msi_bitmap_alloc_hwirqs()
31 	bitmap_set(bmp->bitmap, offset, num);  in msi_bitmap_alloc_hwirqs()
32 	spin_unlock_irqrestore(&bmp->lock, flags);  in msi_bitmap_alloc_hwirqs()
38 	spin_unlock_irqrestore(&bmp->lock, flags);  in msi_bitmap_alloc_hwirqs()
43 void msi_bitmap_free_hwirqs(struct msi_bitmap *bmp, unsigned int offset,  in msi_bitmap_free_hwirqs()  argument
51 	spin_lock_irqsave(&bmp->lock, flags);  in msi_bitmap_free_hwirqs()
52 	bitmap_clear(bmp->bitmap, offset, num);  in msi_bitmap_free_hwirqs()
53 	spin_unlock_irqrestore(&bmp->lock, flags);  in msi_bitmap_free_hwirqs()
57 void msi_bitmap_reserve_hwirq(struct msi_bitmap *bmp, unsigned int hwirq)  in msi_bitmap_reserve_hwirq()  argument
63 	spin_lock_irqsave(&bmp->lock, flags);  in msi_bitmap_reserve_hwirq()
64 	bitmap_allocate_region(bmp->bitmap, hwirq, 0);  in msi_bitmap_reserve_hwirq()
65 	spin_unlock_irqrestore(&bmp->lock, flags);  in msi_bitmap_reserve_hwirq()
79 int msi_bitmap_reserve_dt_hwirqs(struct msi_bitmap *bmp)  in msi_bitmap_reserve_dt_hwirqs()  argument
84 	if (!bmp->of_node)  in msi_bitmap_reserve_dt_hwirqs()
87 	p = of_get_property(bmp->of_node, "msi-available-ranges", &len);  in msi_bitmap_reserve_dt_hwirqs()
90 			 "found on %pOF\n", bmp->of_node);  in msi_bitmap_reserve_dt_hwirqs()
96 		       " property on %pOF\n", bmp->of_node);  in msi_bitmap_reserve_dt_hwirqs()
100 	bitmap_allocate_region(bmp->bitmap, 0, get_count_order(bmp->irq_count));  in msi_bitmap_reserve_dt_hwirqs()
102 	spin_lock(&bmp->lock);  in msi_bitmap_reserve_dt_hwirqs()
108 			bitmap_release_region(bmp->bitmap, *p + j, 0);  in msi_bitmap_reserve_dt_hwirqs()
111 	spin_unlock(&bmp->lock);  in msi_bitmap_reserve_dt_hwirqs()
116 int __ref msi_bitmap_alloc(struct msi_bitmap *bmp, unsigned int irq_count,  in msi_bitmap_alloc()  argument
127 	bmp->bitmap_from_slab = slab_is_available();  in msi_bitmap_alloc()
128 	if (bmp->bitmap_from_slab)  in msi_bitmap_alloc()
129 		bmp->bitmap = kzalloc(size, GFP_KERNEL);  in msi_bitmap_alloc()
131 		bmp->bitmap = memblock_virt_alloc(size, 0);  in msi_bitmap_alloc()
133 		kmemleak_not_leak(bmp->bitmap);  in msi_bitmap_alloc()
136 	if (!bmp->bitmap) {  in msi_bitmap_alloc()
142 	spin_lock_init(&bmp->lock);  in msi_bitmap_alloc()
143 	bmp->of_node = of_node_get(of_node);  in msi_bitmap_alloc()
144 	bmp->irq_count = irq_count;  in msi_bitmap_alloc()
149 void msi_bitmap_free(struct msi_bitmap *bmp)  in msi_bitmap_free()  argument
151 	if (bmp->bitmap_from_slab)  in msi_bitmap_free()
152 		kfree(bmp->bitmap);  in msi_bitmap_free()
153 	of_node_put(bmp->of_node);  in msi_bitmap_free()
154 	bmp->bitmap = NULL;  in msi_bitmap_free()
161 	struct msi_bitmap bmp;  in test_basics()  local
165 	WARN_ON(msi_bitmap_alloc(&bmp, 0, NULL) == 0);  in test_basics()
168 	WARN_ON(msi_bitmap_alloc(&bmp, size, NULL));  in test_basics()
171 	WARN_ON(bitmap_find_free_region(bmp.bitmap, size, get_count_order(size)));  in test_basics()
172 	bitmap_release_region(bmp.bitmap, 0, get_count_order(size));  in test_basics()
175 	WARN_ON(msi_bitmap_reserve_dt_hwirqs(&bmp) <= 0);  in test_basics()
178 	WARN_ON(bitmap_find_free_region(bmp.bitmap, size, get_count_order(size)));  in test_basics()
179 	bitmap_release_region(bmp.bitmap, 0, get_count_order(size));  in test_basics()
183 		WARN_ON(msi_bitmap_alloc_hwirqs(&bmp, 1) < 0);  in test_basics()
185 	WARN_ON(msi_bitmap_alloc_hwirqs(&bmp, 1) >= 0);  in test_basics()
188 	WARN_ON(bitmap_find_free_region(bmp.bitmap, size, 0) >= 0);  in test_basics()
191 	msi_bitmap_free_hwirqs(&bmp, size / 2, 1);  in test_basics()
192 	WARN_ON(msi_bitmap_alloc_hwirqs(&bmp, 1) != size / 2);  in test_basics()
195 	msi_bitmap_free_hwirqs(&bmp, 3, size - 3);  in test_basics()
198 	rc = msi_bitmap_alloc_hwirqs(&bmp, 2);  in test_basics()
200 	rc = msi_bitmap_alloc_hwirqs(&bmp, 4);  in test_basics()
202 	rc = msi_bitmap_alloc_hwirqs(&bmp, 8);  in test_basics()
204 	rc = msi_bitmap_alloc_hwirqs(&bmp, 9);  in test_basics()
206 	rc = msi_bitmap_alloc_hwirqs(&bmp, 3);  in test_basics()
208 	rc = msi_bitmap_alloc_hwirqs(&bmp, 7);  in test_basics()
210 	rc = msi_bitmap_alloc_hwirqs(&bmp, 121);  in test_basics()
213 	msi_bitmap_free(&bmp);  in test_basics()
216 	WARN_ON(bmp.bitmap != NULL);  in test_basics()
227 	struct msi_bitmap bmp;  in test_of_node()  local
236 	WARN_ON(msi_bitmap_alloc(&bmp, SIZE_EXPECTED, &of_node));  in test_of_node()
239 	WARN_ON(msi_bitmap_reserve_dt_hwirqs(&bmp) <= 0);  in test_of_node()
242 	WARN_ON(bitmap_find_free_region(bmp.bitmap, SIZE_EXPECTED,  in test_of_node()
244 	bitmap_release_region(bmp.bitmap, 0, get_count_order(SIZE_EXPECTED));  in test_of_node()
257 	WARN_ON(msi_bitmap_reserve_dt_hwirqs(&bmp));  in test_of_node()
261 	WARN_ON(!bitmap_equal(expected, bmp.bitmap, SIZE_EXPECTED));  in test_of_node()
263 	msi_bitmap_free(&bmp);  in test_of_node()
264 	kfree(bmp.bitmap);  in test_of_node()