subr_extent.c revision 1.42 1 /* $NetBSD: subr_extent.c,v 1.42 2001/06/05 04:39:56 thorpej Exp $ */
2
3 /*-
4 * Copyright (c) 1996, 1998 The NetBSD Foundation, Inc.
5 * All rights reserved.
6 *
7 * This code is derived from software contributed to The NetBSD Foundation
8 * by Jason R. Thorpe and Matthias Drochner.
9 *
10 * Redistribution and use in source and binary forms, with or without
11 * modification, are permitted provided that the following conditions
12 * are met:
13 * 1. Redistributions of source code must retain the above copyright
14 * notice, this list of conditions and the following disclaimer.
15 * 2. Redistributions in binary form must reproduce the above copyright
16 * notice, this list of conditions and the following disclaimer in the
17 * documentation and/or other materials provided with the distribution.
18 * 3. All advertising materials mentioning features or use of this software
19 * must display the following acknowledgement:
20 * This product includes software developed by the NetBSD
21 * Foundation, Inc. and its contributors.
22 * 4. Neither the name of The NetBSD Foundation nor the names of its
23 * contributors may be used to endorse or promote products derived
24 * from this software without specific prior written permission.
25 *
26 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
27 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
28 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
29 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
30 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
31 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
32 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
33 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
34 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
35 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
36 * POSSIBILITY OF SUCH DAMAGE.
37 */
38
39 /*
40 * General purpose extent manager.
41 */
42
43 #ifdef _KERNEL
44 #include <sys/param.h>
45 #include <sys/extent.h>
46 #include <sys/malloc.h>
47 #include <sys/pool.h>
48 #include <sys/time.h>
49 #include <sys/systm.h>
50 #include <sys/proc.h>
51 #include <sys/lock.h>
52
53 #include <uvm/uvm_extern.h>
54
55 #define KMEM_IS_RUNNING (kmem_map != NULL)
56 #elif defined(_EXTENT_TESTING)
57 /*
58 * user-land definitions, so it can fit into a testing harness.
59 */
60 #include <sys/param.h>
61 #include <sys/pool.h>
62 #include <sys/extent.h>
63 #include <errno.h>
64 #include <stdlib.h>
65 #include <stdio.h>
66 #include <string.h>
67
68 /*
69 * Use multi-line #defines to avoid screwing up the kernel tags file;
70 * without this, ctags produces a tags file where panic() shows up
71 * in subr_extent.c rather than subr_prf.c.
72 */
73 #define \
74 malloc(s, t, flags) malloc(s)
75 #define \
76 free(p, t) free(p)
77 #define \
78 tsleep(chan, pri, str, timo) (EWOULDBLOCK)
79 #define \
80 ltsleep(chan,pri,str,timo,lck) (EWOULDBLOCK)
81 #define \
82 wakeup(chan) ((void)0)
83 #define \
84 pool_get(pool, flags) malloc((pool)->pr_size,0,0)
85 #define \
86 pool_put(pool, rp) free(rp,0)
87 #define \
88 panic(a) printf(a)
89 #define \
90 splhigh() (1)
91 #define \
92 splx(s) ((void)(s))
93
94 #define \
95 simple_lock_init(l) ((void)(l))
96 #define \
97 simple_lock(l) ((void)(l))
98 #define \
99 simple_unlock(l) ((void)(l))
100 #define KMEM_IS_RUNNING (1)
101 #endif
102
103 static void extent_insert_and_optimize __P((struct extent *, u_long, u_long,
104 int, struct extent_region *, struct extent_region *));
105 static struct extent_region *extent_alloc_region_descriptor
106 __P((struct extent *, int));
107 static void extent_free_region_descriptor __P((struct extent *,
108 struct extent_region *));
109
110 static struct pool expool;
111 static struct simplelock expool_init_slock = SIMPLELOCK_INITIALIZER;
112 static int expool_initialized;
113
114 /*
115 * Macro to align to an arbitrary power-of-two boundary.
116 */
117 #define EXTENT_ALIGN(_start, _align, _skew) \
118 (((((_start) - (_skew)) + ((_align) - 1)) & (-(_align))) + (_skew))
119
120 /*
121 * Create the extent_region pool.
122 * (This is deferred until one of our callers thinks we can malloc()).
123 */
124
125 static __inline void
126 expool_init(void)
127 {
128
129 simple_lock(&expool_init_slock);
130 if (expool_initialized) {
131 simple_unlock(&expool_init_slock);
132 return;
133 }
134
135 #if defined(_KERNEL)
136 pool_init(&expool, sizeof(struct extent_region), 0, 0, 0,
137 "extent", 0, 0, 0, 0);
138 #else
139 expool.pr_size = sizeof(struct extent_region);
140 #endif
141
142 expool_initialized = 1;
143 simple_unlock(&expool_init_slock);
144 }
145
146 /*
147 * Allocate and initialize an extent map.
148 */
149 struct extent *
150 extent_create(name, start, end, mtype, storage, storagesize, flags)
151 const char *name;
152 u_long start, end;
153 int mtype;
154 caddr_t storage;
155 size_t storagesize;
156 int flags;
157 {
158 struct extent *ex;
159 caddr_t cp = storage;
160 size_t sz = storagesize;
161 struct extent_region *rp;
162 int fixed_extent = (storage != NULL);
163 int s;
164
165 #ifdef DIAGNOSTIC
166 /* Check arguments. */
167 if (name == NULL)
168 panic("extent_create: name == NULL");
169 if (end < start) {
170 printf("extent_create: extent `%s', start 0x%lx, end 0x%lx\n",
171 name, start, end);
172 panic("extent_create: end < start");
173 }
174 if (fixed_extent && (storagesize < sizeof(struct extent_fixed)))
175 panic("extent_create: fixed extent, bad storagesize 0x%lx",
176 (u_long)storagesize);
177 if (fixed_extent == 0 && (storagesize != 0 || storage != NULL))
178 panic("extent_create: storage provided for non-fixed");
179 #endif
180
181 /* Allocate extent descriptor. */
182 if (fixed_extent) {
183 struct extent_fixed *fex;
184
185 memset(storage, 0, storagesize);
186
187 /*
188 * Align all descriptors on "long" boundaries.
189 */
190 fex = (struct extent_fixed *)cp;
191 ex = (struct extent *)fex;
192 cp += ALIGN(sizeof(struct extent_fixed));
193 sz -= ALIGN(sizeof(struct extent_fixed));
194 fex->fex_storage = storage;
195 fex->fex_storagesize = storagesize;
196
197 /*
198 * In a fixed extent, we have to pre-allocate region
199 * descriptors and place them in the extent's freelist.
200 */
201 LIST_INIT(&fex->fex_freelist);
202 while (sz >= ALIGN(sizeof(struct extent_region))) {
203 rp = (struct extent_region *)cp;
204 cp += ALIGN(sizeof(struct extent_region));
205 sz -= ALIGN(sizeof(struct extent_region));
206 LIST_INSERT_HEAD(&fex->fex_freelist, rp, er_link);
207 }
208 } else {
209 s = splhigh();
210 if (expool_initialized == 0)
211 expool_init();
212 splx(s);
213
214 ex = (struct extent *)malloc(sizeof(struct extent),
215 mtype, (flags & EX_WAITOK) ? M_WAITOK : M_NOWAIT);
216 if (ex == NULL)
217 return (NULL);
218 }
219
220 /* Fill in the extent descriptor and return it to the caller. */
221 simple_lock_init(&ex->ex_slock);
222 LIST_INIT(&ex->ex_regions);
223 ex->ex_name = name;
224 ex->ex_start = start;
225 ex->ex_end = end;
226 ex->ex_mtype = mtype;
227 ex->ex_flags = 0;
228 if (fixed_extent)
229 ex->ex_flags |= EXF_FIXED;
230 if (flags & EX_NOCOALESCE)
231 ex->ex_flags |= EXF_NOCOALESCE;
232 return (ex);
233 }
234
235 /*
236 * Destroy an extent map.
237 * Since we're freeing the data, there can't be any references
238 * so we don't need any locking.
239 */
240 void
241 extent_destroy(ex)
242 struct extent *ex;
243 {
244 struct extent_region *rp, *orp;
245
246 #ifdef DIAGNOSTIC
247 /* Check arguments. */
248 if (ex == NULL)
249 panic("extent_destroy: NULL extent");
250 #endif
251
252 /* Free all region descriptors in extent. */
253 for (rp = ex->ex_regions.lh_first; rp != NULL; ) {
254 orp = rp;
255 rp = rp->er_link.le_next;
256 LIST_REMOVE(orp, er_link);
257 extent_free_region_descriptor(ex, orp);
258 }
259
260 /* If we're not a fixed extent, free the extent descriptor itself. */
261 if ((ex->ex_flags & EXF_FIXED) == 0)
262 free(ex, ex->ex_mtype);
263 }
264
265 /*
266 * Insert a region descriptor into the sorted region list after the
267 * entry "after" or at the head of the list (if "after" is NULL).
268 * The region descriptor we insert is passed in "rp". We must
269 * allocate the region descriptor before calling this function!
270 * If we don't need the region descriptor, it will be freed here.
271 */
272 static void
273 extent_insert_and_optimize(ex, start, size, flags, after, rp)
274 struct extent *ex;
275 u_long start, size;
276 int flags;
277 struct extent_region *after, *rp;
278 {
279 struct extent_region *nextr;
280 int appended = 0;
281
282 if (after == NULL) {
283 /*
284 * We're the first in the region list. If there's
285 * a region after us, attempt to coalesce to save
286 * descriptor overhead.
287 */
288 if (((ex->ex_flags & EXF_NOCOALESCE) == 0) &&
289 (ex->ex_regions.lh_first != NULL) &&
290 ((start + size) == ex->ex_regions.lh_first->er_start)) {
291 /*
292 * We can coalesce. Prepend us to the first region.
293 */
294 ex->ex_regions.lh_first->er_start = start;
295 extent_free_region_descriptor(ex, rp);
296 return;
297 }
298
299 /*
300 * Can't coalesce. Fill in the region descriptor
301 * in, and insert us at the head of the region list.
302 */
303 rp->er_start = start;
304 rp->er_end = start + (size - 1);
305 LIST_INSERT_HEAD(&ex->ex_regions, rp, er_link);
306 return;
307 }
308
309 /*
310 * If EXF_NOCOALESCE is set, coalescing is disallowed.
311 */
312 if (ex->ex_flags & EXF_NOCOALESCE)
313 goto cant_coalesce;
314
315 /*
316 * Attempt to coalesce with the region before us.
317 */
318 if ((after->er_end + 1) == start) {
319 /*
320 * We can coalesce. Append ourselves and make
321 * note of it.
322 */
323 after->er_end = start + (size - 1);
324 appended = 1;
325 }
326
327 /*
328 * Attempt to coalesce with the region after us.
329 */
330 if ((after->er_link.le_next != NULL) &&
331 ((start + size) == after->er_link.le_next->er_start)) {
332 /*
333 * We can coalesce. Note that if we appended ourselves
334 * to the previous region, we exactly fit the gap, and
335 * can free the "next" region descriptor.
336 */
337 if (appended) {
338 /*
339 * Yup, we can free it up.
340 */
341 after->er_end = after->er_link.le_next->er_end;
342 nextr = after->er_link.le_next;
343 LIST_REMOVE(nextr, er_link);
344 extent_free_region_descriptor(ex, nextr);
345 } else {
346 /*
347 * Nope, just prepend us to the next region.
348 */
349 after->er_link.le_next->er_start = start;
350 }
351
352 extent_free_region_descriptor(ex, rp);
353 return;
354 }
355
356 /*
357 * We weren't able to coalesce with the next region, but
358 * we don't need to allocate a region descriptor if we
359 * appended ourselves to the previous region.
360 */
361 if (appended) {
362 extent_free_region_descriptor(ex, rp);
363 return;
364 }
365
366 cant_coalesce:
367
368 /*
369 * Fill in the region descriptor and insert ourselves
370 * into the region list.
371 */
372 rp->er_start = start;
373 rp->er_end = start + (size - 1);
374 LIST_INSERT_AFTER(after, rp, er_link);
375 }
376
377 /*
378 * Allocate a specific region in an extent map.
379 */
380 int
381 extent_alloc_region(ex, start, size, flags)
382 struct extent *ex;
383 u_long start, size;
384 int flags;
385 {
386 struct extent_region *rp, *last, *myrp;
387 u_long end = start + (size - 1);
388 int error;
389
390 #ifdef DIAGNOSTIC
391 /* Check arguments. */
392 if (ex == NULL)
393 panic("extent_alloc_region: NULL extent");
394 if (size < 1) {
395 printf("extent_alloc_region: extent `%s', size 0x%lx\n",
396 ex->ex_name, size);
397 panic("extent_alloc_region: bad size");
398 }
399 if (end < start) {
400 printf(
401 "extent_alloc_region: extent `%s', start 0x%lx, size 0x%lx\n",
402 ex->ex_name, start, size);
403 panic("extent_alloc_region: overflow");
404 }
405 #endif
406 #ifdef LOCKDEBUG
407 if (flags & EX_WAITSPACE)
408 simple_lock_only_held(NULL,
409 "extent_alloc_region(EX_WAITSPACE)");
410 #endif
411
412 /*
413 * Make sure the requested region lies within the
414 * extent.
415 *
416 * We don't lock to check the range, because those values
417 * are never modified, and if another thread deletes the
418 * extent, we're screwed anyway.
419 */
420 if ((start < ex->ex_start) || (end > ex->ex_end)) {
421 #ifdef DIAGNOSTIC
422 printf("extent_alloc_region: extent `%s' (0x%lx - 0x%lx)\n",
423 ex->ex_name, ex->ex_start, ex->ex_end);
424 printf("extent_alloc_region: start 0x%lx, end 0x%lx\n",
425 start, end);
426 panic("extent_alloc_region: region lies outside extent");
427 #else
428 return (EINVAL);
429 #endif
430 }
431
432 /*
433 * Allocate the region descriptor. It will be freed later
434 * if we can coalesce with another region. Don't lock before
435 * here! This could block.
436 */
437 myrp = extent_alloc_region_descriptor(ex, flags);
438 if (myrp == NULL) {
439 #ifdef DIAGNOSTIC
440 printf(
441 "extent_alloc_region: can't allocate region descriptor\n");
442 #endif
443 return (ENOMEM);
444 }
445
446 alloc_start:
447 simple_lock(&ex->ex_slock);
448
449 /*
450 * Attempt to place ourselves in the desired area of the
451 * extent. We save ourselves some work by keeping the list sorted.
452 * In other words, if the start of the current region is greater
453 * than the end of our region, we don't have to search any further.
454 */
455
456 /*
457 * Keep a pointer to the last region we looked at so
458 * that we don't have to traverse the list again when
459 * we insert ourselves. If "last" is NULL when we
460 * finally insert ourselves, we go at the head of the
461 * list. See extent_insert_and_optimize() for details.
462 */
463 last = NULL;
464
465 for (rp = ex->ex_regions.lh_first; rp != NULL;
466 rp = rp->er_link.le_next) {
467 if (rp->er_start > end) {
468 /*
469 * We lie before this region and don't
470 * conflict.
471 */
472 break;
473 }
474
475 /*
476 * The current region begins before we end.
477 * Check for a conflict.
478 */
479 if (rp->er_end >= start) {
480 /*
481 * We conflict. If we can (and want to) wait,
482 * do so.
483 */
484 if (flags & EX_WAITSPACE) {
485 ex->ex_flags |= EXF_WANTED;
486 error = ltsleep(ex,
487 PNORELOCK | PRIBIO | ((flags & EX_CATCH) ? PCATCH : 0),
488 "extnt", 0, &ex->ex_slock);
489 if (error)
490 return (error);
491 goto alloc_start;
492 }
493 extent_free_region_descriptor(ex, myrp);
494 simple_unlock(&ex->ex_slock);
495 return (EAGAIN);
496 }
497 /*
498 * We don't conflict, but this region lies before
499 * us. Keep a pointer to this region, and keep
500 * trying.
501 */
502 last = rp;
503 }
504
505 /*
506 * We don't conflict with any regions. "last" points
507 * to the region we fall after, or is NULL if we belong
508 * at the beginning of the region list. Insert ourselves.
509 */
510 extent_insert_and_optimize(ex, start, size, flags, last, myrp);
511 simple_unlock(&ex->ex_slock);
512 return (0);
513 }
514
515 /*
516 * Macro to check (x + y) <= z. This check is designed to fail
517 * if an overflow occurs.
518 */
519 #define LE_OV(x, y, z) ((((x) + (y)) >= (x)) && (((x) + (y)) <= (z)))
520
521 /*
522 * Allocate a region in an extent map subregion.
523 *
524 * If EX_FAST is specified, we return the first fit in the map.
525 * Otherwise, we try to minimize fragmentation by finding the
526 * smallest gap that will hold the request.
527 *
528 * The allocated region is aligned to "alignment", which must be
529 * a power of 2.
530 */
531 int
532 extent_alloc_subregion1(ex, substart, subend, size, alignment, skew, boundary,
533 flags, result)
534 struct extent *ex;
535 u_long substart, subend, size, alignment, skew, boundary;
536 int flags;
537 u_long *result;
538 {
539 struct extent_region *rp, *myrp, *last, *bestlast;
540 u_long newstart, newend, beststart, bestovh, ovh;
541 u_long dontcross;
542 int error;
543
544 #ifdef DIAGNOSTIC
545 /*
546 * Check arguments.
547 *
548 * We don't lock to check these, because these values
549 * are never modified, and if another thread deletes the
550 * extent, we're screwed anyway.
551 */
552 if (ex == NULL)
553 panic("extent_alloc_subregion: NULL extent");
554 if (result == NULL)
555 panic("extent_alloc_subregion: NULL result pointer");
556 if ((substart < ex->ex_start) || (substart > ex->ex_end) ||
557 (subend > ex->ex_end) || (subend < ex->ex_start)) {
558 printf("extent_alloc_subregion: extent `%s', ex_start 0x%lx, ex_end 0x%lx\n",
559 ex->ex_name, ex->ex_start, ex->ex_end);
560 printf("extent_alloc_subregion: substart 0x%lx, subend 0x%lx\n",
561 substart, subend);
562 panic("extent_alloc_subregion: bad subregion");
563 }
564 if ((size < 1) || ((size - 1) > (subend - substart))) {
565 printf("extent_alloc_subregion: extent `%s', size 0x%lx\n",
566 ex->ex_name, size);
567 panic("extent_alloc_subregion: bad size");
568 }
569 if (alignment == 0)
570 panic("extent_alloc_subregion: bad alignment");
571 if (boundary && (boundary < size)) {
572 printf(
573 "extent_alloc_subregion: extent `%s', size 0x%lx, "
574 "boundary 0x%lx\n", ex->ex_name, size, boundary);
575 panic("extent_alloc_subregion: bad boundary");
576 }
577 #endif
578 #ifdef LOCKDEBUG
579 if (flags & EX_WAITSPACE)
580 simple_lock_only_held(NULL,
581 "extent_alloc_subregion1(EX_WAITSPACE)");
582 #endif
583
584 /*
585 * Allocate the region descriptor. It will be freed later
586 * if we can coalesce with another region. Don't lock before
587 * here! This could block.
588 */
589 myrp = extent_alloc_region_descriptor(ex, flags);
590 if (myrp == NULL) {
591 #ifdef DIAGNOSTIC
592 printf(
593 "extent_alloc_subregion: can't allocate region descriptor\n");
594 #endif
595 return (ENOMEM);
596 }
597
598 alloc_start:
599 simple_lock(&ex->ex_slock);
600
601 /*
602 * Keep a pointer to the last region we looked at so
603 * that we don't have to traverse the list again when
604 * we insert ourselves. If "last" is NULL when we
605 * finally insert ourselves, we go at the head of the
606 * list. See extent_insert_and_optimize() for deatails.
607 */
608 last = NULL;
609
610 /*
611 * Keep track of size and location of the smallest
612 * chunk we fit in.
613 *
614 * Since the extent can be as large as the numeric range
615 * of the CPU (0 - 0xffffffff for 32-bit systems), the
616 * best overhead value can be the maximum unsigned integer.
617 * Thus, we initialize "bestovh" to 0, since we insert ourselves
618 * into the region list immediately on an exact match (which
619 * is the only case where "bestovh" would be set to 0).
620 */
621 bestovh = 0;
622 beststart = 0;
623 bestlast = NULL;
624
625 /*
626 * For N allocated regions, we must make (N + 1)
627 * checks for unallocated space. The first chunk we
628 * check is the area from the beginning of the subregion
629 * to the first allocated region after that point.
630 */
631 newstart = EXTENT_ALIGN(substart, alignment, skew);
632 if (newstart < ex->ex_start) {
633 #ifdef DIAGNOSTIC
634 printf(
635 "extent_alloc_subregion: extent `%s' (0x%lx - 0x%lx), alignment 0x%lx\n",
636 ex->ex_name, ex->ex_start, ex->ex_end, alignment);
637 simple_unlock(&ex->ex_slock);
638 panic("extent_alloc_subregion: overflow after alignment");
639 #else
640 extent_free_region_descriptor(ex, myrp);
641 simple_unlock(&ex->ex_slock);
642 return (EINVAL);
643 #endif
644 }
645
646 /*
647 * Find the first allocated region that begins on or after
648 * the subregion start, advancing the "last" pointer along
649 * the way.
650 */
651 for (rp = ex->ex_regions.lh_first; rp != NULL;
652 rp = rp->er_link.le_next) {
653 if (rp->er_start >= newstart)
654 break;
655 last = rp;
656 }
657
658 /*
659 * Relocate the start of our candidate region to the end of
660 * the last allocated region (if there was one overlapping
661 * our subrange).
662 */
663 if (last != NULL && last->er_end >= newstart)
664 newstart = EXTENT_ALIGN((last->er_end + 1), alignment, skew);
665
666 for (; rp != NULL; rp = rp->er_link.le_next) {
667 /*
668 * Check the chunk before "rp". Note that our
669 * comparison is safe from overflow conditions.
670 */
671 if (LE_OV(newstart, size, rp->er_start)) {
672 /*
673 * Do a boundary check, if necessary. Note
674 * that a region may *begin* on the boundary,
675 * but it must end before the boundary.
676 */
677 if (boundary) {
678 newend = newstart + (size - 1);
679
680 /*
681 * Calculate the next boundary after the start
682 * of this region.
683 */
684 dontcross = EXTENT_ALIGN(newstart+1, boundary,
685 (flags & EX_BOUNDZERO) ? 0 : ex->ex_start)
686 - 1;
687
688 #if 0
689 printf("newstart=%lx newend=%lx ex_start=%lx ex_end=%lx boundary=%lx dontcross=%lx\n",
690 newstart, newend, ex->ex_start, ex->ex_end,
691 boundary, dontcross);
692 #endif
693
694 /* Check for overflow */
695 if (dontcross < ex->ex_start)
696 dontcross = ex->ex_end;
697 else if (newend > dontcross) {
698 /*
699 * Candidate region crosses boundary.
700 * Throw away the leading part and see
701 * if we still fit.
702 */
703 newstart = dontcross + 1;
704 newend = newstart + (size - 1);
705 dontcross += boundary;
706 if (!LE_OV(newstart, size, rp->er_start))
707 continue;
708 }
709
710 /*
711 * If we run past the end of
712 * the extent or the boundary
713 * overflows, then the request
714 * can't fit.
715 */
716 if (newstart + size - 1 > ex->ex_end ||
717 dontcross < newstart)
718 goto fail;
719 }
720
721 /*
722 * We would fit into this space. Calculate
723 * the overhead (wasted space). If we exactly
724 * fit, or we're taking the first fit, insert
725 * ourselves into the region list.
726 */
727 ovh = rp->er_start - newstart - size;
728 if ((flags & EX_FAST) || (ovh == 0))
729 goto found;
730
731 /*
732 * Don't exactly fit, but check to see
733 * if we're better than any current choice.
734 */
735 if ((bestovh == 0) || (ovh < bestovh)) {
736 bestovh = ovh;
737 beststart = newstart;
738 bestlast = last;
739 }
740 }
741
742 /*
743 * Skip past the current region and check again.
744 */
745 newstart = EXTENT_ALIGN((rp->er_end + 1), alignment, skew);
746 if (newstart < rp->er_end) {
747 /*
748 * Overflow condition. Don't error out, since
749 * we might have a chunk of space that we can
750 * use.
751 */
752 goto fail;
753 }
754
755 last = rp;
756 }
757
758 /*
759 * The final check is from the current starting point to the
760 * end of the subregion. If there were no allocated regions,
761 * "newstart" is set to the beginning of the subregion, or
762 * just past the end of the last allocated region, adjusted
763 * for alignment in either case.
764 */
765 if (LE_OV(newstart, (size - 1), subend)) {
766 /*
767 * Do a boundary check, if necessary. Note
768 * that a region may *begin* on the boundary,
769 * but it must end before the boundary.
770 */
771 if (boundary) {
772 newend = newstart + (size - 1);
773
774 /*
775 * Calculate the next boundary after the start
776 * of this region.
777 */
778 dontcross = EXTENT_ALIGN(newstart+1, boundary,
779 (flags & EX_BOUNDZERO) ? 0 : ex->ex_start)
780 - 1;
781
782 #if 0
783 printf("newstart=%lx newend=%lx ex_start=%lx ex_end=%lx boundary=%lx dontcross=%lx\n",
784 newstart, newend, ex->ex_start, ex->ex_end,
785 boundary, dontcross);
786 #endif
787
788 /* Check for overflow */
789 if (dontcross < ex->ex_start)
790 dontcross = ex->ex_end;
791 else if (newend > dontcross) {
792 /*
793 * Candidate region crosses boundary.
794 * Throw away the leading part and see
795 * if we still fit.
796 */
797 newstart = dontcross + 1;
798 newend = newstart + (size - 1);
799 dontcross += boundary;
800 if (!LE_OV(newstart, (size - 1), subend))
801 goto fail;
802 }
803
804 /*
805 * If we run past the end of
806 * the extent or the boundary
807 * overflows, then the request
808 * can't fit.
809 */
810 if (newstart + size - 1 > ex->ex_end ||
811 dontcross < newstart)
812 goto fail;
813 }
814
815 /*
816 * We would fit into this space. Calculate
817 * the overhead (wasted space). If we exactly
818 * fit, or we're taking the first fit, insert
819 * ourselves into the region list.
820 */
821 ovh = ex->ex_end - newstart - (size - 1);
822 if ((flags & EX_FAST) || (ovh == 0))
823 goto found;
824
825 /*
826 * Don't exactly fit, but check to see
827 * if we're better than any current choice.
828 */
829 if ((bestovh == 0) || (ovh < bestovh)) {
830 bestovh = ovh;
831 beststart = newstart;
832 bestlast = last;
833 }
834 }
835
836 fail:
837 /*
838 * One of the following two conditions have
839 * occurred:
840 *
841 * There is no chunk large enough to hold the request.
842 *
843 * If EX_FAST was not specified, there is not an
844 * exact match for the request.
845 *
846 * Note that if we reach this point and EX_FAST is
847 * set, then we know there is no space in the extent for
848 * the request.
849 */
850 if (((flags & EX_FAST) == 0) && (bestovh != 0)) {
851 /*
852 * We have a match that's "good enough".
853 */
854 newstart = beststart;
855 last = bestlast;
856 goto found;
857 }
858
859 /*
860 * No space currently available. Wait for it to free up,
861 * if possible.
862 */
863 if (flags & EX_WAITSPACE) {
864 ex->ex_flags |= EXF_WANTED;
865 error = ltsleep(ex,
866 PNORELOCK | PRIBIO | ((flags & EX_CATCH) ? PCATCH : 0),
867 "extnt", 0, &ex->ex_slock);
868 if (error)
869 return (error);
870 goto alloc_start;
871 }
872
873 extent_free_region_descriptor(ex, myrp);
874 simple_unlock(&ex->ex_slock);
875 return (EAGAIN);
876
877 found:
878 /*
879 * Insert ourselves into the region list.
880 */
881 extent_insert_and_optimize(ex, newstart, size, flags, last, myrp);
882 simple_unlock(&ex->ex_slock);
883 *result = newstart;
884 return (0);
885 }
886
887 int
888 extent_free(ex, start, size, flags)
889 struct extent *ex;
890 u_long start, size;
891 int flags;
892 {
893 struct extent_region *rp, *nrp = NULL;
894 u_long end = start + (size - 1);
895 int exflags;
896
897 #ifdef DIAGNOSTIC
898 /*
899 * Check arguments.
900 *
901 * We don't lock to check these, because these values
902 * are never modified, and if another thread deletes the
903 * extent, we're screwed anyway.
904 */
905 if (ex == NULL)
906 panic("extent_free: NULL extent");
907 if ((start < ex->ex_start) || (start > ex->ex_end)) {
908 extent_print(ex);
909 printf("extent_free: extent `%s', start 0x%lx, size 0x%lx\n",
910 ex->ex_name, start, size);
911 panic("extent_free: extent `%s', region not within extent",
912 ex->ex_name);
913 }
914 /* Check for an overflow. */
915 if (end < start) {
916 extent_print(ex);
917 printf("extent_free: extent `%s', start 0x%lx, size 0x%lx\n",
918 ex->ex_name, start, size);
919 panic("extent_free: overflow");
920 }
921 #endif
922
923 /*
924 * If we're allowing coalescing, we must allocate a region
925 * descriptor now, since it might block.
926 *
927 * XXX Make a static, create-time flags word, so we don't
928 * XXX have to lock to read it!
929 */
930 simple_lock(&ex->ex_slock);
931 exflags = ex->ex_flags;
932 simple_unlock(&ex->ex_slock);
933
934 if ((exflags & EXF_NOCOALESCE) == 0) {
935 /* Allocate a region descriptor. */
936 nrp = extent_alloc_region_descriptor(ex, flags);
937 if (nrp == NULL)
938 return (ENOMEM);
939 }
940
941 simple_lock(&ex->ex_slock);
942
943 /*
944 * Find region and deallocate. Several possibilities:
945 *
946 * 1. (start == er_start) && (end == er_end):
947 * Free descriptor.
948 *
949 * 2. (start == er_start) && (end < er_end):
950 * Adjust er_start.
951 *
952 * 3. (start > er_start) && (end == er_end):
953 * Adjust er_end.
954 *
955 * 4. (start > er_start) && (end < er_end):
956 * Fragment region. Requires descriptor alloc.
957 *
958 * Cases 2, 3, and 4 require that the EXF_NOCOALESCE flag
959 * is not set.
960 */
961 for (rp = ex->ex_regions.lh_first; rp != NULL;
962 rp = rp->er_link.le_next) {
963 /*
964 * Save ourselves some comparisons; does the current
965 * region end before chunk to be freed begins? If so,
966 * then we haven't found the appropriate region descriptor.
967 */
968 if (rp->er_end < start)
969 continue;
970
971 /*
972 * Save ourselves some traversal; does the current
973 * region begin after the chunk to be freed ends? If so,
974 * then we've already passed any possible region descriptors
975 * that might have contained the chunk to be freed.
976 */
977 if (rp->er_start > end)
978 break;
979
980 /* Case 1. */
981 if ((start == rp->er_start) && (end == rp->er_end)) {
982 LIST_REMOVE(rp, er_link);
983 extent_free_region_descriptor(ex, rp);
984 goto done;
985 }
986
987 /*
988 * The following cases all require that EXF_NOCOALESCE
989 * is not set.
990 */
991 if (ex->ex_flags & EXF_NOCOALESCE)
992 continue;
993
994 /* Case 2. */
995 if ((start == rp->er_start) && (end < rp->er_end)) {
996 rp->er_start = (end + 1);
997 goto done;
998 }
999
1000 /* Case 3. */
1001 if ((start > rp->er_start) && (end == rp->er_end)) {
1002 rp->er_end = (start - 1);
1003 goto done;
1004 }
1005
1006 /* Case 4. */
1007 if ((start > rp->er_start) && (end < rp->er_end)) {
1008 /* Fill in new descriptor. */
1009 nrp->er_start = end + 1;
1010 nrp->er_end = rp->er_end;
1011
1012 /* Adjust current descriptor. */
1013 rp->er_end = start - 1;
1014
1015 /* Insert new descriptor after current. */
1016 LIST_INSERT_AFTER(rp, nrp, er_link);
1017
1018 /* We used the new descriptor, so don't free it below */
1019 nrp = NULL;
1020 goto done;
1021 }
1022 }
1023
1024 /* Region not found, or request otherwise invalid. */
1025 simple_unlock(&ex->ex_slock);
1026 extent_print(ex);
1027 printf("extent_free: start 0x%lx, end 0x%lx\n", start, end);
1028 panic("extent_free: region not found");
1029
1030 done:
1031 if (nrp != NULL)
1032 extent_free_region_descriptor(ex, nrp);
1033 if (ex->ex_flags & EXF_WANTED) {
1034 ex->ex_flags &= ~EXF_WANTED;
1035 wakeup(ex);
1036 }
1037 simple_unlock(&ex->ex_slock);
1038 return (0);
1039 }
1040
1041 /*
1042 * Allocate an extent region descriptor. EXTENT MUST NOT BE LOCKED,
1043 * AS THIS FUNCTION MAY BLOCK! We will handle any locking we may need.
1044 */
1045 static struct extent_region *
1046 extent_alloc_region_descriptor(ex, flags)
1047 struct extent *ex;
1048 int flags;
1049 {
1050 struct extent_region *rp;
1051 int exflags;
1052 int s;
1053
1054 /*
1055 * If the kernel memory allocator is not yet running, we can't
1056 * use it (obviously).
1057 */
1058 if (KMEM_IS_RUNNING == 0)
1059 flags &= ~EX_MALLOCOK;
1060
1061 /*
1062 * XXX Make a static, create-time flags word, so we don't
1063 * XXX have to lock to read it!
1064 */
1065 simple_lock(&ex->ex_slock);
1066 exflags = ex->ex_flags;
1067 simple_unlock(&ex->ex_slock);
1068
1069 if (exflags & EXF_FIXED) {
1070 struct extent_fixed *fex = (struct extent_fixed *)ex;
1071
1072 for (;;) {
1073 simple_lock(&ex->ex_slock);
1074 if ((rp = fex->fex_freelist.lh_first) != NULL) {
1075 /*
1076 * Don't muck with flags after pulling it off
1077 * the freelist; it may have been dynamically
1078 * allocated, and kindly given to us. We
1079 * need to remember that information.
1080 */
1081 LIST_REMOVE(rp, er_link);
1082 simple_unlock(&ex->ex_slock);
1083 return (rp);
1084 }
1085 if (flags & EX_MALLOCOK) {
1086 simple_unlock(&ex->ex_slock);
1087 goto alloc;
1088 }
1089 if ((flags & EX_WAITOK) == 0) {
1090 simple_unlock(&ex->ex_slock);
1091 return (NULL);
1092 }
1093 ex->ex_flags |= EXF_FLWANTED;
1094 if (ltsleep(&fex->fex_freelist,
1095 PNORELOCK| PRIBIO | ((flags & EX_CATCH) ? PCATCH : 0),
1096 "extnt", 0, &ex->ex_slock))
1097 return (NULL);
1098 }
1099 }
1100
1101 alloc:
1102 s = splhigh();
1103 if (expool_initialized == 0)
1104 expool_init();
1105 rp = pool_get(&expool, (flags & EX_WAITOK) ? PR_WAITOK : 0);
1106 splx(s);
1107
1108 if (rp != NULL)
1109 rp->er_flags = ER_ALLOC;
1110
1111 return (rp);
1112 }
1113
1114 /*
1115 * Free an extent region descriptor. EXTENT _MUST_ BE LOCKED! This
1116 * is safe as we do not block here.
1117 */
1118 static void
1119 extent_free_region_descriptor(ex, rp)
1120 struct extent *ex;
1121 struct extent_region *rp;
1122 {
1123 int s;
1124
1125 if (ex->ex_flags & EXF_FIXED) {
1126 struct extent_fixed *fex = (struct extent_fixed *)ex;
1127
1128 /*
1129 * If someone's waiting for a region descriptor,
1130 * be nice and give them this one, rather than
1131 * just free'ing it back to the system.
1132 */
1133 if (rp->er_flags & ER_ALLOC) {
1134 if (ex->ex_flags & EXF_FLWANTED) {
1135 /* Clear all but ER_ALLOC flag. */
1136 rp->er_flags = ER_ALLOC;
1137 LIST_INSERT_HEAD(&fex->fex_freelist, rp,
1138 er_link);
1139 goto wake_em_up;
1140 } else {
1141 s = splhigh();
1142 pool_put(&expool, rp);
1143 splx(s);
1144 }
1145 } else {
1146 /* Clear all flags. */
1147 rp->er_flags = 0;
1148 LIST_INSERT_HEAD(&fex->fex_freelist, rp, er_link);
1149 }
1150
1151 if (ex->ex_flags & EXF_FLWANTED) {
1152 wake_em_up:
1153 ex->ex_flags &= ~EXF_FLWANTED;
1154 wakeup(&fex->fex_freelist);
1155 }
1156 return;
1157 }
1158
1159 /*
1160 * We know it's dynamically allocated if we get here.
1161 */
1162 s = splhigh();
1163 pool_put(&expool, rp);
1164 splx(s);
1165 }
1166
1167 void
1168 extent_print(ex)
1169 struct extent *ex;
1170 {
1171 struct extent_region *rp;
1172
1173 if (ex == NULL)
1174 panic("extent_print: NULL extent");
1175
1176 simple_lock(&ex->ex_slock);
1177
1178 printf("extent `%s' (0x%lx - 0x%lx), flags = 0x%x\n", ex->ex_name,
1179 ex->ex_start, ex->ex_end, ex->ex_flags);
1180
1181 for (rp = ex->ex_regions.lh_first; rp != NULL;
1182 rp = rp->er_link.le_next)
1183 printf(" 0x%lx - 0x%lx\n", rp->er_start, rp->er_end);
1184
1185 simple_unlock(&ex->ex_slock);
1186 }
1187