Commit | Line | Data |
---|---|---|
3a1bd924 TH |
1 | /************************************************************************** |
2 | * | |
3 | * Copyright 2006 Tungsten Graphics, Inc., Bismarck, ND., USA. | |
4 | * All Rights Reserved. | |
5 | * | |
6 | * Permission is hereby granted, free of charge, to any person obtaining a | |
7 | * copy of this software and associated documentation files (the | |
8 | * "Software"), to deal in the Software without restriction, including | |
9 | * without limitation the rights to use, copy, modify, merge, publish, | |
10 | * distribute, sub license, and/or sell copies of the Software, and to | |
11 | * permit persons to whom the Software is furnished to do so, subject to | |
12 | * the following conditions: | |
13 | * | |
14 | * The above copyright notice and this permission notice (including the | |
15 | * next paragraph) shall be included in all copies or substantial portions | |
16 | * of the Software. | |
17 | * | |
18 | * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR | |
19 | * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, | |
20 | * FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT. IN NO EVENT SHALL | |
21 | * THE COPYRIGHT HOLDERS, AUTHORS AND/OR ITS SUPPLIERS BE LIABLE FOR ANY CLAIM, | |
22 | * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR | |
23 | * OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE | |
24 | * USE OR OTHER DEALINGS IN THE SOFTWARE. | |
25 | * | |
26 | * | |
27 | **************************************************************************/ | |
28 | ||
29 | /* | |
30 | * Generic simple memory manager implementation. Intended to be used as a base | |
31 | * class implementation for more advanced memory managers. | |
32 | * | |
33 | * Note that the algorithm used is quite simple and there might be substantial | |
34 | * performance gains if a smarter free list is implemented. Currently it is just an | |
35 | * unordered stack of free regions. This could easily be improved if an RB-tree | |
36 | * is used instead. At least if we expect heavy fragmentation. | |
37 | * | |
38 | * Aligned allocations can also see improvement. | |
39 | * | |
40 | * Authors: | |
96de0e25 | 41 | * Thomas Hellström <thomas-at-tungstengraphics-dot-com> |
3a1bd924 TH |
42 | */ |
43 | ||
44 | #include "drmP.h" | |
249d6048 | 45 | #include "drm_mm.h" |
1d58420b | 46 | #include <linux/slab.h> |
fa8a1238 | 47 | #include <linux/seq_file.h> |
1d58420b | 48 | |
249d6048 JG |
49 | #define MM_UNUSED_TARGET 4 |
50 | ||
55910517 | 51 | unsigned long drm_mm_tail_space(struct drm_mm *mm) |
1d58420b TH |
52 | { |
53 | struct list_head *tail_node; | |
55910517 | 54 | struct drm_mm_node *entry; |
1d58420b TH |
55 | |
56 | tail_node = mm->ml_entry.prev; | |
55910517 | 57 | entry = list_entry(tail_node, struct drm_mm_node, ml_entry); |
1d58420b TH |
58 | if (!entry->free) |
59 | return 0; | |
60 | ||
61 | return entry->size; | |
62 | } | |
63 | ||
55910517 | 64 | int drm_mm_remove_space_from_tail(struct drm_mm *mm, unsigned long size) |
1d58420b TH |
65 | { |
66 | struct list_head *tail_node; | |
55910517 | 67 | struct drm_mm_node *entry; |
1d58420b TH |
68 | |
69 | tail_node = mm->ml_entry.prev; | |
55910517 | 70 | entry = list_entry(tail_node, struct drm_mm_node, ml_entry); |
1d58420b TH |
71 | if (!entry->free) |
72 | return -ENOMEM; | |
73 | ||
74 | if (entry->size <= size) | |
75 | return -ENOMEM; | |
76 | ||
77 | entry->size -= size; | |
78 | return 0; | |
79 | } | |
80 | ||
249d6048 JG |
81 | static struct drm_mm_node *drm_mm_kmalloc(struct drm_mm *mm, int atomic) |
82 | { | |
83 | struct drm_mm_node *child; | |
84 | ||
85 | if (atomic) | |
86 | child = kmalloc(sizeof(*child), GFP_ATOMIC); | |
87 | else | |
88 | child = kmalloc(sizeof(*child), GFP_KERNEL); | |
89 | ||
90 | if (unlikely(child == NULL)) { | |
91 | spin_lock(&mm->unused_lock); | |
92 | if (list_empty(&mm->unused_nodes)) | |
93 | child = NULL; | |
94 | else { | |
95 | child = | |
96 | list_entry(mm->unused_nodes.next, | |
97 | struct drm_mm_node, fl_entry); | |
98 | list_del(&child->fl_entry); | |
99 | --mm->num_unused; | |
100 | } | |
101 | spin_unlock(&mm->unused_lock); | |
102 | } | |
103 | return child; | |
104 | } | |
105 | ||
a698cf34 JG |
106 | /* drm_mm_pre_get() - pre allocate drm_mm_node structure |
107 | * drm_mm: memory manager struct we are pre-allocating for | |
108 | * | |
109 | * Returns 0 on success or -ENOMEM if allocation fails. | |
110 | */ | |
249d6048 JG |
111 | int drm_mm_pre_get(struct drm_mm *mm) |
112 | { | |
113 | struct drm_mm_node *node; | |
114 | ||
115 | spin_lock(&mm->unused_lock); | |
116 | while (mm->num_unused < MM_UNUSED_TARGET) { | |
117 | spin_unlock(&mm->unused_lock); | |
118 | node = kmalloc(sizeof(*node), GFP_KERNEL); | |
119 | spin_lock(&mm->unused_lock); | |
120 | ||
121 | if (unlikely(node == NULL)) { | |
122 | int ret = (mm->num_unused < 2) ? -ENOMEM : 0; | |
123 | spin_unlock(&mm->unused_lock); | |
124 | return ret; | |
125 | } | |
126 | ++mm->num_unused; | |
127 | list_add_tail(&node->fl_entry, &mm->unused_nodes); | |
128 | } | |
129 | spin_unlock(&mm->unused_lock); | |
130 | return 0; | |
131 | } | |
132 | EXPORT_SYMBOL(drm_mm_pre_get); | |
1d58420b | 133 | |
55910517 | 134 | static int drm_mm_create_tail_node(struct drm_mm *mm, |
249d6048 JG |
135 | unsigned long start, |
136 | unsigned long size, int atomic) | |
1d58420b | 137 | { |
55910517 | 138 | struct drm_mm_node *child; |
1d58420b | 139 | |
249d6048 JG |
140 | child = drm_mm_kmalloc(mm, atomic); |
141 | if (unlikely(child == NULL)) | |
1d58420b TH |
142 | return -ENOMEM; |
143 | ||
144 | child->free = 1; | |
145 | child->size = size; | |
146 | child->start = start; | |
147 | child->mm = mm; | |
148 | ||
149 | list_add_tail(&child->ml_entry, &mm->ml_entry); | |
150 | list_add_tail(&child->fl_entry, &mm->fl_entry); | |
151 | ||
152 | return 0; | |
153 | } | |
154 | ||
249d6048 | 155 | int drm_mm_add_space_to_tail(struct drm_mm *mm, unsigned long size, int atomic) |
1d58420b TH |
156 | { |
157 | struct list_head *tail_node; | |
55910517 | 158 | struct drm_mm_node *entry; |
1d58420b TH |
159 | |
160 | tail_node = mm->ml_entry.prev; | |
55910517 | 161 | entry = list_entry(tail_node, struct drm_mm_node, ml_entry); |
1d58420b | 162 | if (!entry->free) { |
249d6048 JG |
163 | return drm_mm_create_tail_node(mm, entry->start + entry->size, |
164 | size, atomic); | |
1d58420b TH |
165 | } |
166 | entry->size += size; | |
167 | return 0; | |
168 | } | |
169 | ||
55910517 | 170 | static struct drm_mm_node *drm_mm_split_at_start(struct drm_mm_node *parent, |
249d6048 JG |
171 | unsigned long size, |
172 | int atomic) | |
1d58420b | 173 | { |
55910517 | 174 | struct drm_mm_node *child; |
1d58420b | 175 | |
249d6048 JG |
176 | child = drm_mm_kmalloc(parent->mm, atomic); |
177 | if (unlikely(child == NULL)) | |
1d58420b TH |
178 | return NULL; |
179 | ||
180 | INIT_LIST_HEAD(&child->fl_entry); | |
181 | ||
182 | child->free = 0; | |
183 | child->size = size; | |
184 | child->start = parent->start; | |
185 | child->mm = parent->mm; | |
186 | ||
187 | list_add_tail(&child->ml_entry, &parent->ml_entry); | |
188 | INIT_LIST_HEAD(&child->fl_entry); | |
189 | ||
190 | parent->size -= size; | |
191 | parent->start += size; | |
192 | return child; | |
193 | } | |
194 | ||
195 | ||
89579f77 TH |
196 | struct drm_mm_node *drm_mm_get_block_generic(struct drm_mm_node *node, |
197 | unsigned long size, | |
198 | unsigned alignment, | |
199 | int atomic) | |
3a1bd924 TH |
200 | { |
201 | ||
55910517 | 202 | struct drm_mm_node *align_splitoff = NULL; |
1d58420b | 203 | unsigned tmp = 0; |
3a1bd924 TH |
204 | |
205 | if (alignment) | |
e6c03c5b | 206 | tmp = node->start % alignment; |
1d58420b TH |
207 | |
208 | if (tmp) { | |
249d6048 | 209 | align_splitoff = |
89579f77 | 210 | drm_mm_split_at_start(node, alignment - tmp, atomic); |
249d6048 | 211 | if (unlikely(align_splitoff == NULL)) |
1d58420b TH |
212 | return NULL; |
213 | } | |
3a1bd924 | 214 | |
e6c03c5b CW |
215 | if (node->size == size) { |
216 | list_del_init(&node->fl_entry); | |
217 | node->free = 0; | |
3a1bd924 | 218 | } else { |
89579f77 | 219 | node = drm_mm_split_at_start(node, size, atomic); |
1d58420b | 220 | } |
3a1bd924 | 221 | |
1d58420b TH |
222 | if (align_splitoff) |
223 | drm_mm_put_block(align_splitoff); | |
3a1bd924 | 224 | |
e6c03c5b | 225 | return node; |
3a1bd924 | 226 | } |
89579f77 | 227 | EXPORT_SYMBOL(drm_mm_get_block_generic); |
249d6048 | 228 | |
a2e68e92 JG |
229 | struct drm_mm_node *drm_mm_get_block_range_generic(struct drm_mm_node *node, |
230 | unsigned long size, | |
231 | unsigned alignment, | |
232 | unsigned long start, | |
233 | unsigned long end, | |
234 | int atomic) | |
235 | { | |
236 | struct drm_mm_node *align_splitoff = NULL; | |
237 | unsigned tmp = 0; | |
238 | unsigned wasted = 0; | |
239 | ||
240 | if (node->start < start) | |
241 | wasted += start - node->start; | |
242 | if (alignment) | |
243 | tmp = ((node->start + wasted) % alignment); | |
244 | ||
245 | if (tmp) | |
246 | wasted += alignment - tmp; | |
247 | if (wasted) { | |
248 | align_splitoff = drm_mm_split_at_start(node, wasted, atomic); | |
249 | if (unlikely(align_splitoff == NULL)) | |
250 | return NULL; | |
251 | } | |
252 | ||
253 | if (node->size == size) { | |
254 | list_del_init(&node->fl_entry); | |
255 | node->free = 0; | |
256 | } else { | |
257 | node = drm_mm_split_at_start(node, size, atomic); | |
258 | } | |
259 | ||
260 | if (align_splitoff) | |
261 | drm_mm_put_block(align_splitoff); | |
262 | ||
263 | return node; | |
264 | } | |
265 | EXPORT_SYMBOL(drm_mm_get_block_range_generic); | |
266 | ||
3a1bd924 TH |
267 | /* |
268 | * Put a block. Merge with the previous and / or next block if they are free. | |
269 | * Otherwise add to the free stack. | |
270 | */ | |
271 | ||
249d6048 | 272 | void drm_mm_put_block(struct drm_mm_node *cur) |
3a1bd924 TH |
273 | { |
274 | ||
55910517 | 275 | struct drm_mm *mm = cur->mm; |
3a1bd924 | 276 | struct list_head *cur_head = &cur->ml_entry; |
1d58420b | 277 | struct list_head *root_head = &mm->ml_entry; |
55910517 DA |
278 | struct drm_mm_node *prev_node = NULL; |
279 | struct drm_mm_node *next_node; | |
3a1bd924 | 280 | |
a1d0fcf5 | 281 | int merged = 0; |
3a1bd924 TH |
282 | |
283 | if (cur_head->prev != root_head) { | |
249d6048 JG |
284 | prev_node = |
285 | list_entry(cur_head->prev, struct drm_mm_node, ml_entry); | |
3a1bd924 TH |
286 | if (prev_node->free) { |
287 | prev_node->size += cur->size; | |
a1d0fcf5 | 288 | merged = 1; |
3a1bd924 TH |
289 | } |
290 | } | |
291 | if (cur_head->next != root_head) { | |
249d6048 JG |
292 | next_node = |
293 | list_entry(cur_head->next, struct drm_mm_node, ml_entry); | |
3a1bd924 TH |
294 | if (next_node->free) { |
295 | if (merged) { | |
296 | prev_node->size += next_node->size; | |
297 | list_del(&next_node->ml_entry); | |
298 | list_del(&next_node->fl_entry); | |
a698cf34 | 299 | spin_lock(&mm->unused_lock); |
249d6048 JG |
300 | if (mm->num_unused < MM_UNUSED_TARGET) { |
301 | list_add(&next_node->fl_entry, | |
302 | &mm->unused_nodes); | |
303 | ++mm->num_unused; | |
304 | } else | |
305 | kfree(next_node); | |
a698cf34 | 306 | spin_unlock(&mm->unused_lock); |
3a1bd924 TH |
307 | } else { |
308 | next_node->size += cur->size; | |
309 | next_node->start = cur->start; | |
a1d0fcf5 | 310 | merged = 1; |
3a1bd924 TH |
311 | } |
312 | } | |
313 | } | |
314 | if (!merged) { | |
a1d0fcf5 | 315 | cur->free = 1; |
1d58420b | 316 | list_add(&cur->fl_entry, &mm->fl_entry); |
3a1bd924 TH |
317 | } else { |
318 | list_del(&cur->ml_entry); | |
a698cf34 | 319 | spin_lock(&mm->unused_lock); |
249d6048 JG |
320 | if (mm->num_unused < MM_UNUSED_TARGET) { |
321 | list_add(&cur->fl_entry, &mm->unused_nodes); | |
322 | ++mm->num_unused; | |
323 | } else | |
324 | kfree(cur); | |
a698cf34 | 325 | spin_unlock(&mm->unused_lock); |
3a1bd924 TH |
326 | } |
327 | } | |
249d6048 | 328 | |
673a394b | 329 | EXPORT_SYMBOL(drm_mm_put_block); |
3a1bd924 | 330 | |
249d6048 JG |
331 | struct drm_mm_node *drm_mm_search_free(const struct drm_mm *mm, |
332 | unsigned long size, | |
333 | unsigned alignment, int best_match) | |
3a1bd924 | 334 | { |
1d58420b | 335 | const struct list_head *free_stack = &mm->fl_entry; |
55910517 DA |
336 | struct drm_mm_node *entry; |
337 | struct drm_mm_node *best; | |
3a1bd924 | 338 | unsigned long best_size; |
1d58420b | 339 | unsigned wasted; |
3a1bd924 TH |
340 | |
341 | best = NULL; | |
342 | best_size = ~0UL; | |
343 | ||
26f3751e | 344 | list_for_each_entry(entry, free_stack, fl_entry) { |
1d58420b TH |
345 | wasted = 0; |
346 | ||
347 | if (entry->size < size) | |
348 | continue; | |
349 | ||
350 | if (alignment) { | |
351 | register unsigned tmp = entry->start % alignment; | |
352 | if (tmp) | |
353 | wasted += alignment - tmp; | |
354 | } | |
355 | ||
1d58420b | 356 | if (entry->size >= size + wasted) { |
3a1bd924 TH |
357 | if (!best_match) |
358 | return entry; | |
e2108eb1 | 359 | if (entry->size < best_size) { |
3a1bd924 TH |
360 | best = entry; |
361 | best_size = entry->size; | |
362 | } | |
363 | } | |
364 | } | |
365 | ||
366 | return best; | |
367 | } | |
249d6048 | 368 | EXPORT_SYMBOL(drm_mm_search_free); |
3a1bd924 | 369 | |
a2e68e92 JG |
370 | struct drm_mm_node *drm_mm_search_free_in_range(const struct drm_mm *mm, |
371 | unsigned long size, | |
372 | unsigned alignment, | |
373 | unsigned long start, | |
374 | unsigned long end, | |
375 | int best_match) | |
376 | { | |
a2e68e92 JG |
377 | const struct list_head *free_stack = &mm->fl_entry; |
378 | struct drm_mm_node *entry; | |
379 | struct drm_mm_node *best; | |
380 | unsigned long best_size; | |
381 | unsigned wasted; | |
382 | ||
383 | best = NULL; | |
384 | best_size = ~0UL; | |
385 | ||
26f3751e | 386 | list_for_each_entry(entry, free_stack, fl_entry) { |
a2e68e92 JG |
387 | wasted = 0; |
388 | ||
389 | if (entry->size < size) | |
390 | continue; | |
391 | ||
392 | if (entry->start > end || (entry->start+entry->size) < start) | |
393 | continue; | |
394 | ||
395 | if (entry->start < start) | |
396 | wasted += start - entry->start; | |
397 | ||
398 | if (alignment) { | |
399 | register unsigned tmp = (entry->start + wasted) % alignment; | |
400 | if (tmp) | |
401 | wasted += alignment - tmp; | |
402 | } | |
403 | ||
775c6709 TH |
404 | if (entry->size >= size + wasted && |
405 | (entry->start + wasted + size) <= end) { | |
a2e68e92 JG |
406 | if (!best_match) |
407 | return entry; | |
e2108eb1 | 408 | if (entry->size < best_size) { |
a2e68e92 JG |
409 | best = entry; |
410 | best_size = entry->size; | |
411 | } | |
412 | } | |
413 | } | |
414 | ||
415 | return best; | |
416 | } | |
417 | EXPORT_SYMBOL(drm_mm_search_free_in_range); | |
418 | ||
55910517 | 419 | int drm_mm_clean(struct drm_mm * mm) |
3a1bd924 | 420 | { |
1d58420b | 421 | struct list_head *head = &mm->ml_entry; |
3a1bd924 | 422 | |
1d58420b TH |
423 | return (head->next->next == head); |
424 | } | |
249d6048 | 425 | EXPORT_SYMBOL(drm_mm_clean); |
3a1bd924 | 426 | |
55910517 | 427 | int drm_mm_init(struct drm_mm * mm, unsigned long start, unsigned long size) |
1d58420b TH |
428 | { |
429 | INIT_LIST_HEAD(&mm->ml_entry); | |
430 | INIT_LIST_HEAD(&mm->fl_entry); | |
249d6048 JG |
431 | INIT_LIST_HEAD(&mm->unused_nodes); |
432 | mm->num_unused = 0; | |
433 | spin_lock_init(&mm->unused_lock); | |
3a1bd924 | 434 | |
249d6048 | 435 | return drm_mm_create_tail_node(mm, start, size, 0); |
3a1bd924 | 436 | } |
673a394b | 437 | EXPORT_SYMBOL(drm_mm_init); |
3a1bd924 | 438 | |
55910517 | 439 | void drm_mm_takedown(struct drm_mm * mm) |
3a1bd924 | 440 | { |
1d58420b | 441 | struct list_head *bnode = mm->fl_entry.next; |
55910517 | 442 | struct drm_mm_node *entry; |
249d6048 | 443 | struct drm_mm_node *next; |
3a1bd924 | 444 | |
55910517 | 445 | entry = list_entry(bnode, struct drm_mm_node, fl_entry); |
3a1bd924 | 446 | |
1d58420b TH |
447 | if (entry->ml_entry.next != &mm->ml_entry || |
448 | entry->fl_entry.next != &mm->fl_entry) { | |
3a1bd924 TH |
449 | DRM_ERROR("Memory manager not clean. Delaying takedown\n"); |
450 | return; | |
451 | } | |
452 | ||
453 | list_del(&entry->fl_entry); | |
454 | list_del(&entry->ml_entry); | |
249d6048 JG |
455 | kfree(entry); |
456 | ||
457 | spin_lock(&mm->unused_lock); | |
458 | list_for_each_entry_safe(entry, next, &mm->unused_nodes, fl_entry) { | |
459 | list_del(&entry->fl_entry); | |
460 | kfree(entry); | |
461 | --mm->num_unused; | |
462 | } | |
463 | spin_unlock(&mm->unused_lock); | |
3a1bd924 | 464 | |
249d6048 | 465 | BUG_ON(mm->num_unused != 0); |
3a1bd924 | 466 | } |
f453ba04 | 467 | EXPORT_SYMBOL(drm_mm_takedown); |
fa8a1238 | 468 | |
99d7e48e JG |
469 | void drm_mm_debug_table(struct drm_mm *mm, const char *prefix) |
470 | { | |
471 | struct drm_mm_node *entry; | |
472 | int total_used = 0, total_free = 0, total = 0; | |
473 | ||
474 | list_for_each_entry(entry, &mm->ml_entry, ml_entry) { | |
475 | printk(KERN_DEBUG "%s 0x%08lx-0x%08lx: %8ld: %s\n", | |
476 | prefix, entry->start, entry->start + entry->size, | |
477 | entry->size, entry->free ? "free" : "used"); | |
478 | total += entry->size; | |
479 | if (entry->free) | |
480 | total_free += entry->size; | |
481 | else | |
482 | total_used += entry->size; | |
483 | } | |
484 | printk(KERN_DEBUG "%s total: %d, used %d free %d\n", prefix, total, | |
485 | total_used, total_free); | |
486 | } | |
487 | EXPORT_SYMBOL(drm_mm_debug_table); | |
488 | ||
fa8a1238 DA |
489 | #if defined(CONFIG_DEBUG_FS) |
490 | int drm_mm_dump_table(struct seq_file *m, struct drm_mm *mm) | |
491 | { | |
492 | struct drm_mm_node *entry; | |
493 | int total_used = 0, total_free = 0, total = 0; | |
494 | ||
495 | list_for_each_entry(entry, &mm->ml_entry, ml_entry) { | |
496 | seq_printf(m, "0x%08lx-0x%08lx: 0x%08lx: %s\n", entry->start, entry->start + entry->size, entry->size, entry->free ? "free" : "used"); | |
497 | total += entry->size; | |
498 | if (entry->free) | |
499 | total_free += entry->size; | |
500 | else | |
501 | total_used += entry->size; | |
502 | } | |
01d01ba9 | 503 | seq_printf(m, "total: %d, used %d free %d\n", total, total_used, total_free); |
fa8a1238 DA |
504 | return 0; |
505 | } | |
506 | EXPORT_SYMBOL(drm_mm_dump_table); | |
507 | #endif |