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