17 int msi_bitmap_alloc_hwirqs(struct msi_bitmap *bmp, int num)  in msi_bitmap_alloc_hwirqs()  argument
22 	spin_lock_irqsave(&bmp->lock, flags);  in msi_bitmap_alloc_hwirqs()
24 	offset = bitmap_find_next_zero_area(bmp->bitmap, bmp->irq_count, 0,  in msi_bitmap_alloc_hwirqs()
26 	if (offset > bmp->irq_count)  in msi_bitmap_alloc_hwirqs()
29 	bitmap_set(bmp->bitmap, offset, num);  in msi_bitmap_alloc_hwirqs()
30 	spin_unlock_irqrestore(&bmp->lock, flags);  in msi_bitmap_alloc_hwirqs()
36 	spin_unlock_irqrestore(&bmp->lock, flags);  in msi_bitmap_alloc_hwirqs()
41 void msi_bitmap_free_hwirqs(struct msi_bitmap *bmp, unsigned int offset,  in msi_bitmap_free_hwirqs()  argument
49 	spin_lock_irqsave(&bmp->lock, flags);  in msi_bitmap_free_hwirqs()
50 	bitmap_clear(bmp->bitmap, offset, num);  in msi_bitmap_free_hwirqs()
51 	spin_unlock_irqrestore(&bmp->lock, flags);  in msi_bitmap_free_hwirqs()
55 void msi_bitmap_reserve_hwirq(struct msi_bitmap *bmp, unsigned int hwirq)  in msi_bitmap_reserve_hwirq()  argument
61 	spin_lock_irqsave(&bmp->lock, flags);  in msi_bitmap_reserve_hwirq()
62 	bitmap_allocate_region(bmp->bitmap, hwirq, 0);  in msi_bitmap_reserve_hwirq()
63 	spin_unlock_irqrestore(&bmp->lock, flags);  in msi_bitmap_reserve_hwirq()
77 int msi_bitmap_reserve_dt_hwirqs(struct msi_bitmap *bmp)  in msi_bitmap_reserve_dt_hwirqs()  argument
82 	if (!bmp->of_node)  in msi_bitmap_reserve_dt_hwirqs()
85 	p = of_get_property(bmp->of_node, "msi-available-ranges", &len);  in msi_bitmap_reserve_dt_hwirqs()
88 			 "found on %s\n", bmp->of_node->full_name);  in msi_bitmap_reserve_dt_hwirqs()
94 		       " property on %s\n", bmp->of_node->full_name);  in msi_bitmap_reserve_dt_hwirqs()
98 	bitmap_allocate_region(bmp->bitmap, 0, get_count_order(bmp->irq_count));  in msi_bitmap_reserve_dt_hwirqs()
100 	spin_lock(&bmp->lock);  in msi_bitmap_reserve_dt_hwirqs()
106 			bitmap_release_region(bmp->bitmap, *p + j, 0);  in msi_bitmap_reserve_dt_hwirqs()
109 	spin_unlock(&bmp->lock);  in msi_bitmap_reserve_dt_hwirqs()
114 int msi_bitmap_alloc(struct msi_bitmap *bmp, unsigned int irq_count,  in msi_bitmap_alloc()  argument
125 	bmp->bitmap = zalloc_maybe_bootmem(size, GFP_KERNEL);  in msi_bitmap_alloc()
126 	if (!bmp->bitmap) {  in msi_bitmap_alloc()
132 	spin_lock_init(&bmp->lock);  in msi_bitmap_alloc()
133 	bmp->of_node = of_node_get(of_node);  in msi_bitmap_alloc()
134 	bmp->irq_count = irq_count;  in msi_bitmap_alloc()
139 void msi_bitmap_free(struct msi_bitmap *bmp)  in msi_bitmap_free()  argument
142 	of_node_put(bmp->of_node);  in msi_bitmap_free()
143 	bmp->bitmap = NULL;  in msi_bitmap_free()
150 	struct msi_bitmap bmp;  in test_basics()  local
154 	WARN_ON(msi_bitmap_alloc(&bmp, 0, NULL) == 0);  in test_basics()
157 	WARN_ON(msi_bitmap_alloc(&bmp, size, NULL));  in test_basics()
160 	WARN_ON(bitmap_find_free_region(bmp.bitmap, size, get_count_order(size)));  in test_basics()
161 	bitmap_release_region(bmp.bitmap, 0, get_count_order(size));  in test_basics()
164 	WARN_ON(msi_bitmap_reserve_dt_hwirqs(&bmp) <= 0);  in test_basics()
167 	WARN_ON(bitmap_find_free_region(bmp.bitmap, size, get_count_order(size)));  in test_basics()
168 	bitmap_release_region(bmp.bitmap, 0, get_count_order(size));  in test_basics()
172 		WARN_ON(msi_bitmap_alloc_hwirqs(&bmp, 1) < 0);  in test_basics()
174 	WARN_ON(msi_bitmap_alloc_hwirqs(&bmp, 1) >= 0);  in test_basics()
177 	WARN_ON(bitmap_find_free_region(bmp.bitmap, size, 0) >= 0);  in test_basics()
180 	msi_bitmap_free_hwirqs(&bmp, size / 2, 1);  in test_basics()
181 	WARN_ON(msi_bitmap_alloc_hwirqs(&bmp, 1) != size / 2);  in test_basics()
184 	msi_bitmap_free_hwirqs(&bmp, 3, size - 3);  in test_basics()
187 	rc = msi_bitmap_alloc_hwirqs(&bmp, 2);  in test_basics()
189 	rc = msi_bitmap_alloc_hwirqs(&bmp, 4);  in test_basics()
191 	rc = msi_bitmap_alloc_hwirqs(&bmp, 8);  in test_basics()
193 	rc = msi_bitmap_alloc_hwirqs(&bmp, 9);  in test_basics()
195 	rc = msi_bitmap_alloc_hwirqs(&bmp, 3);  in test_basics()
197 	rc = msi_bitmap_alloc_hwirqs(&bmp, 7);  in test_basics()
199 	rc = msi_bitmap_alloc_hwirqs(&bmp, 121);  in test_basics()
202 	msi_bitmap_free(&bmp);  in test_basics()
205 	WARN_ON(bmp.bitmap != NULL);  in test_basics()
207 	kfree(bmp.bitmap);  in test_basics()
218 	struct msi_bitmap bmp;  in test_of_node()  local
227 	WARN_ON(msi_bitmap_alloc(&bmp, size, &of_node));  in test_of_node()
230 	WARN_ON(msi_bitmap_reserve_dt_hwirqs(&bmp) <= 0);  in test_of_node()
233 	WARN_ON(bitmap_find_free_region(bmp.bitmap, size, get_count_order(size)));  in test_of_node()
234 	bitmap_release_region(bmp.bitmap, 0, get_count_order(size));  in test_of_node()
247 	WARN_ON(msi_bitmap_reserve_dt_hwirqs(&bmp));  in test_of_node()
251 	WARN_ON(!bitmap_equal(expected, bmp.bitmap, size));  in test_of_node()
253 	msi_bitmap_free(&bmp);  in test_of_node()
254 	kfree(bmp.bitmap);  in test_of_node()