1 /* $NetBSD: udf_allocation.c,v 1.30 2010/12/22 12:15:02 reinoud Exp $ */ 2 3 /* 4 * Copyright (c) 2006, 2008 Reinoud Zandijk 5 * All rights reserved. 6 * 7 * Redistribution and use in source and binary forms, with or without 8 * modification, are permitted provided that the following conditions 9 * are met: 10 * 1. Redistributions of source code must retain the above copyright 11 * notice, this list of conditions and the following disclaimer. 12 * 2. Redistributions in binary form must reproduce the above copyright 13 * notice, this list of conditions and the following disclaimer in the 14 * documentation and/or other materials provided with the distribution. 15 * 16 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR 17 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 18 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 19 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, 20 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 21 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 22 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 23 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 24 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 25 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 26 * 27 */ 28 29 #include <sys/cdefs.h> 30 #ifndef lint 31 __KERNEL_RCSID(0, "$NetBSD: udf_allocation.c,v 1.30 2010/12/22 12:15:02 reinoud Exp $"); 32 #endif /* not lint */ 33 34 35 #if defined(_KERNEL_OPT) 36 #include "opt_compat_netbsd.h" 37 #endif 38 39 /* TODO strip */ 40 #include <sys/param.h> 41 #include <sys/systm.h> 42 #include <sys/sysctl.h> 43 #include <sys/namei.h> 44 #include <sys/proc.h> 45 #include <sys/kernel.h> 46 #include <sys/vnode.h> 47 #include <miscfs/genfs/genfs_node.h> 48 #include <sys/mount.h> 49 #include <sys/buf.h> 50 #include <sys/file.h> 51 #include <sys/device.h> 52 #include <sys/disklabel.h> 53 #include <sys/ioctl.h> 54 #include <sys/malloc.h> 55 #include <sys/dirent.h> 56 #include <sys/stat.h> 57 #include <sys/conf.h> 58 #include <sys/kauth.h> 59 #include <sys/kthread.h> 60 #include <dev/clock_subr.h> 61 62 #include <fs/udf/ecma167-udf.h> 63 #include <fs/udf/udf_mount.h> 64 65 #include "udf.h" 66 #include "udf_subr.h" 67 #include "udf_bswap.h" 68 69 70 #define VTOI(vnode) ((struct udf_node *) vnode->v_data) 71 72 static void udf_record_allocation_in_node(struct udf_mount *ump, 73 struct buf *buf, uint16_t vpart_num, uint64_t *mapping, 74 struct long_ad *node_ad_cpy); 75 76 static void udf_collect_free_space_for_vpart(struct udf_mount *ump, 77 uint16_t vpart_num, uint32_t num_lb); 78 79 static void udf_wipe_adslots(struct udf_node *udf_node); 80 static void udf_count_alloc_exts(struct udf_node *udf_node); 81 82 /* 83 * IDEA/BUSY: Each udf_node gets its own extentwalker state for all operations; 84 * this will hopefully/likely reduce O(nlog(n)) to O(1) for most functionality 85 * since actions are most likely sequencial and thus seeking doesn't need 86 * searching for the same or adjacent position again. 87 */ 88 89 /* --------------------------------------------------------------------- */ 90 91 #if 0 92 #if 1 93 static void 94 udf_node_dump(struct udf_node *udf_node) { 95 struct file_entry *fe; 96 struct extfile_entry *efe; 97 struct icb_tag *icbtag; 98 struct long_ad s_ad; 99 uint64_t inflen; 100 uint32_t icbflags, addr_type; 101 uint32_t len, lb_num; 102 uint32_t flags; 103 int part_num; 104 int lb_size, eof, slot; 105 106 if ((udf_verbose & UDF_DEBUG_NODEDUMP) == 0) 107 return; 108 109 lb_size = udf_rw32(udf_node->ump->logical_vol->lb_size); 110 111 fe = udf_node->fe; 112 efe = udf_node->efe; 113 if (fe) { 114 icbtag = &fe->icbtag; 115 inflen = udf_rw64(fe->inf_len); 116 } else { 117 icbtag = &efe->icbtag; 118 inflen = udf_rw64(efe->inf_len); 119 } 120 121 icbflags = udf_rw16(icbtag->flags); 122 addr_type = icbflags & UDF_ICB_TAG_FLAGS_ALLOC_MASK; 123 124 printf("udf_node_dump %p :\n", udf_node); 125 126 if (addr_type == UDF_ICB_INTERN_ALLOC) { 127 printf("\tIntern alloc, len = %"PRIu64"\n", inflen); 128 return; 129 } 130 131 printf("\tInflen = %"PRIu64"\n", inflen); 132 printf("\t\t"); 133 134 slot = 0; 135 for (;;) { 136 udf_get_adslot(udf_node, slot, &s_ad, &eof); 137 if (eof) 138 break; 139 part_num = udf_rw16(s_ad.loc.part_num); 140 lb_num = udf_rw32(s_ad.loc.lb_num); 141 len = udf_rw32(s_ad.len); 142 flags = UDF_EXT_FLAGS(len); 143 len = UDF_EXT_LEN(len); 144 145 printf("["); 146 if (part_num >= 0) 147 printf("part %d, ", part_num); 148 printf("lb_num %d, len %d", lb_num, len); 149 if (flags) 150 printf(", flags %d", flags>>30); 151 printf("] "); 152 153 if (flags == UDF_EXT_REDIRECT) { 154 printf("\n\textent END\n\tallocation extent\n\t\t"); 155 } 156 157 slot++; 158 } 159 printf("\n\tl_ad END\n\n"); 160 } 161 #else 162 #define udf_node_dump(a) 163 #endif 164 165 166 static void 167 udf_assert_allocated(struct udf_mount *ump, uint16_t vpart_num, 168 uint32_t lb_num, uint32_t num_lb) 169 { 170 struct udf_bitmap *bitmap; 171 struct part_desc *pdesc; 172 uint32_t ptov; 173 uint32_t bitval; 174 uint8_t *bpos; 175 int bit; 176 int phys_part; 177 int ok; 178 179 DPRINTF(PARANOIA, ("udf_assert_allocated: check virt lbnum %d " 180 "part %d + %d sect\n", lb_num, vpart_num, num_lb)); 181 182 /* get partition backing up this vpart_num */ 183 pdesc = ump->partitions[ump->vtop[vpart_num]]; 184 185 switch (ump->vtop_tp[vpart_num]) { 186 case UDF_VTOP_TYPE_PHYS : 187 case UDF_VTOP_TYPE_SPARABLE : 188 /* free space to freed or unallocated space bitmap */ 189 ptov = udf_rw32(pdesc->start_loc); 190 phys_part = ump->vtop[vpart_num]; 191 192 /* use unallocated bitmap */ 193 bitmap = &ump->part_unalloc_bits[phys_part]; 194 195 /* if no bitmaps are defined, bail out */ 196 if (bitmap->bits == NULL) 197 break; 198 199 /* check bits */ 200 KASSERT(bitmap->bits); 201 ok = 1; 202 bpos = bitmap->bits + lb_num/8; 203 bit = lb_num % 8; 204 while (num_lb > 0) { 205 bitval = (1 << bit); 206 DPRINTF(PARANOIA, ("XXX : check %d, %p, bit %d\n", 207 lb_num, bpos, bit)); 208 KASSERT(bitmap->bits + lb_num/8 == bpos); 209 if (*bpos & bitval) { 210 printf("\tlb_num %d is NOT marked busy\n", 211 lb_num); 212 ok = 0; 213 } 214 lb_num++; num_lb--; 215 bit = (bit + 1) % 8; 216 if (bit == 0) 217 bpos++; 218 } 219 if (!ok) { 220 /* KASSERT(0); */ 221 } 222 223 break; 224 case UDF_VTOP_TYPE_VIRT : 225 /* TODO check space */ 226 KASSERT(num_lb == 1); 227 break; 228 case UDF_VTOP_TYPE_META : 229 /* TODO check space in the metadata bitmap */ 230 default: 231 /* not implemented */ 232 break; 233 } 234 } 235 236 237 static void 238 udf_node_sanity_check(struct udf_node *udf_node, 239 uint64_t *cnt_inflen, uint64_t *cnt_logblksrec) 240 { 241 union dscrptr *dscr; 242 struct file_entry *fe; 243 struct extfile_entry *efe; 244 struct icb_tag *icbtag; 245 struct long_ad s_ad; 246 uint64_t inflen, logblksrec; 247 uint32_t icbflags, addr_type; 248 uint32_t len, lb_num, l_ea, l_ad, max_l_ad; 249 uint16_t part_num; 250 uint8_t *data_pos; 251 int dscr_size, lb_size, flags, whole_lb; 252 int i, slot, eof; 253 254 // KASSERT(mutex_owned(&udf_node->ump->allocate_mutex)); 255 256 if (1) 257 udf_node_dump(udf_node); 258 259 lb_size = udf_rw32(udf_node->ump->logical_vol->lb_size); 260 261 fe = udf_node->fe; 262 efe = udf_node->efe; 263 if (fe) { 264 dscr = (union dscrptr *) fe; 265 icbtag = &fe->icbtag; 266 inflen = udf_rw64(fe->inf_len); 267 dscr_size = sizeof(struct file_entry) -1; 268 logblksrec = udf_rw64(fe->logblks_rec); 269 l_ad = udf_rw32(fe->l_ad); 270 l_ea = udf_rw32(fe->l_ea); 271 } else { 272 dscr = (union dscrptr *) efe; 273 icbtag = &efe->icbtag; 274 inflen = udf_rw64(efe->inf_len); 275 dscr_size = sizeof(struct extfile_entry) -1; 276 logblksrec = udf_rw64(efe->logblks_rec); 277 l_ad = udf_rw32(efe->l_ad); 278 l_ea = udf_rw32(efe->l_ea); 279 } 280 data_pos = (uint8_t *) dscr + dscr_size + l_ea; 281 max_l_ad = lb_size - dscr_size - l_ea; 282 icbflags = udf_rw16(icbtag->flags); 283 addr_type = icbflags & UDF_ICB_TAG_FLAGS_ALLOC_MASK; 284 285 /* check if tail is zero */ 286 DPRINTF(PARANOIA, ("Sanity check blank tail\n")); 287 for (i = l_ad; i < max_l_ad; i++) { 288 if (data_pos[i] != 0) 289 printf( "sanity_check: violation: node byte %d " 290 "has value %d\n", i, data_pos[i]); 291 } 292 293 /* reset counters */ 294 *cnt_inflen = 0; 295 *cnt_logblksrec = 0; 296 297 if (addr_type == UDF_ICB_INTERN_ALLOC) { 298 KASSERT(l_ad <= max_l_ad); 299 KASSERT(l_ad == inflen); 300 *cnt_inflen = inflen; 301 return; 302 } 303 304 /* start counting */ 305 whole_lb = 1; 306 slot = 0; 307 for (;;) { 308 udf_get_adslot(udf_node, slot, &s_ad, &eof); 309 if (eof) 310 break; 311 KASSERT(whole_lb == 1); 312 313 part_num = udf_rw16(s_ad.loc.part_num); 314 lb_num = udf_rw32(s_ad.loc.lb_num); 315 len = udf_rw32(s_ad.len); 316 flags = UDF_EXT_FLAGS(len); 317 len = UDF_EXT_LEN(len); 318 319 if (flags != UDF_EXT_REDIRECT) { 320 *cnt_inflen += len; 321 if (flags == UDF_EXT_ALLOCATED) { 322 *cnt_logblksrec += (len + lb_size -1) / lb_size; 323 } 324 } else { 325 KASSERT(len == lb_size); 326 } 327 /* check allocation */ 328 if (flags == UDF_EXT_ALLOCATED) 329 udf_assert_allocated(udf_node->ump, part_num, lb_num, 330 (len + lb_size - 1) / lb_size); 331 332 /* check whole lb */ 333 whole_lb = ((len % lb_size) == 0); 334 335 slot++; 336 } 337 /* rest should be zero (ad_off > l_ad < max_l_ad - adlen) */ 338 339 KASSERT(*cnt_inflen == inflen); 340 KASSERT(*cnt_logblksrec == logblksrec); 341 342 // KASSERT(mutex_owned(&udf_node->ump->allocate_mutex)); 343 } 344 #else 345 static void 346 udf_node_sanity_check(struct udf_node *udf_node, 347 uint64_t *cnt_inflen, uint64_t *cnt_logblksrec) { 348 struct file_entry *fe; 349 struct extfile_entry *efe; 350 struct icb_tag *icbtag; 351 uint64_t inflen, logblksrec; 352 int dscr_size, lb_size; 353 354 lb_size = udf_rw32(udf_node->ump->logical_vol->lb_size); 355 356 fe = udf_node->fe; 357 efe = udf_node->efe; 358 if (fe) { 359 icbtag = &fe->icbtag; 360 inflen = udf_rw64(fe->inf_len); 361 dscr_size = sizeof(struct file_entry) -1; 362 logblksrec = udf_rw64(fe->logblks_rec); 363 } else { 364 icbtag = &efe->icbtag; 365 inflen = udf_rw64(efe->inf_len); 366 dscr_size = sizeof(struct extfile_entry) -1; 367 logblksrec = udf_rw64(efe->logblks_rec); 368 } 369 *cnt_logblksrec = logblksrec; 370 *cnt_inflen = inflen; 371 } 372 #endif 373 374 /* --------------------------------------------------------------------- */ 375 376 void 377 udf_calc_freespace(struct udf_mount *ump, uint64_t *sizeblks, uint64_t *freeblks) 378 { 379 struct logvol_int_desc *lvid; 380 uint32_t *pos1, *pos2; 381 int vpart, num_vpart; 382 383 lvid = ump->logvol_integrity; 384 *freeblks = *sizeblks = 0; 385 386 /* 387 * Sequentials media report free space directly (CD/DVD/BD-R), for the 388 * other media we need the logical volume integrity. 389 * 390 * We sum all free space up here regardless of type. 391 */ 392 393 KASSERT(lvid); 394 num_vpart = udf_rw32(lvid->num_part); 395 396 if (ump->discinfo.mmc_cur & MMC_CAP_SEQUENTIAL) { 397 /* use track info directly summing if there are 2 open */ 398 /* XXX assumption at most two tracks open */ 399 *freeblks = ump->data_track.free_blocks; 400 if (ump->data_track.tracknr != ump->metadata_track.tracknr) 401 *freeblks += ump->metadata_track.free_blocks; 402 *sizeblks = ump->discinfo.last_possible_lba; 403 } else { 404 /* free and used space for mountpoint based on logvol integrity */ 405 for (vpart = 0; vpart < num_vpart; vpart++) { 406 pos1 = &lvid->tables[0] + vpart; 407 pos2 = &lvid->tables[0] + num_vpart + vpart; 408 if (udf_rw32(*pos1) != (uint32_t) -1) { 409 *freeblks += udf_rw32(*pos1); 410 *sizeblks += udf_rw32(*pos2); 411 } 412 } 413 } 414 /* adjust for accounted uncommitted blocks */ 415 for (vpart = 0; vpart < num_vpart; vpart++) 416 *freeblks -= ump->uncommitted_lbs[vpart]; 417 418 if (*freeblks > UDF_DISC_SLACK) { 419 *freeblks -= UDF_DISC_SLACK; 420 } else { 421 *freeblks = 0; 422 } 423 } 424 425 426 static void 427 udf_calc_vpart_freespace(struct udf_mount *ump, uint16_t vpart_num, uint64_t *freeblks) 428 { 429 struct logvol_int_desc *lvid; 430 uint32_t *pos1; 431 432 lvid = ump->logvol_integrity; 433 *freeblks = 0; 434 435 /* 436 * Sequentials media report free space directly (CD/DVD/BD-R), for the 437 * other media we need the logical volume integrity. 438 * 439 * We sum all free space up here regardless of type. 440 */ 441 442 KASSERT(lvid); 443 if (ump->discinfo.mmc_cur & MMC_CAP_SEQUENTIAL) { 444 /* XXX assumption at most two tracks open */ 445 if (vpart_num == ump->data_part) { 446 *freeblks = ump->data_track.free_blocks; 447 } else { 448 *freeblks = ump->metadata_track.free_blocks; 449 } 450 } else { 451 /* free and used space for mountpoint based on logvol integrity */ 452 pos1 = &lvid->tables[0] + vpart_num; 453 if (udf_rw32(*pos1) != (uint32_t) -1) 454 *freeblks += udf_rw32(*pos1); 455 } 456 457 /* adjust for accounted uncommitted blocks */ 458 if (*freeblks > ump->uncommitted_lbs[vpart_num]) { 459 *freeblks -= ump->uncommitted_lbs[vpart_num]; 460 } else { 461 *freeblks = 0; 462 } 463 } 464 465 /* --------------------------------------------------------------------- */ 466 467 int 468 udf_translate_vtop(struct udf_mount *ump, struct long_ad *icb_loc, 469 uint32_t *lb_numres, uint32_t *extres) 470 { 471 struct part_desc *pdesc; 472 struct spare_map_entry *sme; 473 struct long_ad s_icb_loc; 474 uint64_t foffset, end_foffset; 475 uint32_t lb_size, len; 476 uint32_t lb_num, lb_rel, lb_packet; 477 uint32_t udf_rw32_lbmap, ext_offset; 478 uint16_t vpart; 479 int rel, part, error, eof, slot, flags; 480 481 assert(ump && icb_loc && lb_numres); 482 483 vpart = udf_rw16(icb_loc->loc.part_num); 484 lb_num = udf_rw32(icb_loc->loc.lb_num); 485 if (vpart > UDF_VTOP_RAWPART) 486 return EINVAL; 487 488 translate_again: 489 part = ump->vtop[vpart]; 490 pdesc = ump->partitions[part]; 491 492 switch (ump->vtop_tp[vpart]) { 493 case UDF_VTOP_TYPE_RAW : 494 /* 1:1 to the end of the device */ 495 *lb_numres = lb_num; 496 *extres = INT_MAX; 497 return 0; 498 case UDF_VTOP_TYPE_PHYS : 499 /* transform into its disc logical block */ 500 if (lb_num > udf_rw32(pdesc->part_len)) 501 return EINVAL; 502 *lb_numres = lb_num + udf_rw32(pdesc->start_loc); 503 504 /* extent from here to the end of the partition */ 505 *extres = udf_rw32(pdesc->part_len) - lb_num; 506 return 0; 507 case UDF_VTOP_TYPE_VIRT : 508 /* only maps one logical block, lookup in VAT */ 509 if (lb_num >= ump->vat_entries) /* XXX > or >= ? */ 510 return EINVAL; 511 512 /* lookup in virtual allocation table file */ 513 mutex_enter(&ump->allocate_mutex); 514 error = udf_vat_read(ump->vat_node, 515 (uint8_t *) &udf_rw32_lbmap, 4, 516 ump->vat_offset + lb_num * 4); 517 mutex_exit(&ump->allocate_mutex); 518 519 if (error) 520 return error; 521 522 lb_num = udf_rw32(udf_rw32_lbmap); 523 524 /* transform into its disc logical block */ 525 if (lb_num > udf_rw32(pdesc->part_len)) 526 return EINVAL; 527 *lb_numres = lb_num + udf_rw32(pdesc->start_loc); 528 529 /* just one logical block */ 530 *extres = 1; 531 return 0; 532 case UDF_VTOP_TYPE_SPARABLE : 533 /* check if the packet containing the lb_num is remapped */ 534 lb_packet = lb_num / ump->sparable_packet_size; 535 lb_rel = lb_num % ump->sparable_packet_size; 536 537 for (rel = 0; rel < udf_rw16(ump->sparing_table->rt_l); rel++) { 538 sme = &ump->sparing_table->entries[rel]; 539 if (lb_packet == udf_rw32(sme->org)) { 540 /* NOTE maps to absolute disc logical block! */ 541 *lb_numres = udf_rw32(sme->map) + lb_rel; 542 *extres = ump->sparable_packet_size - lb_rel; 543 return 0; 544 } 545 } 546 547 /* transform into its disc logical block */ 548 if (lb_num > udf_rw32(pdesc->part_len)) 549 return EINVAL; 550 *lb_numres = lb_num + udf_rw32(pdesc->start_loc); 551 552 /* rest of block */ 553 *extres = ump->sparable_packet_size - lb_rel; 554 return 0; 555 case UDF_VTOP_TYPE_META : 556 /* we have to look into the file's allocation descriptors */ 557 558 /* use metadatafile allocation mutex */ 559 lb_size = udf_rw32(ump->logical_vol->lb_size); 560 561 UDF_LOCK_NODE(ump->metadata_node, 0); 562 563 /* get first overlapping extent */ 564 foffset = 0; 565 slot = 0; 566 for (;;) { 567 udf_get_adslot(ump->metadata_node, 568 slot, &s_icb_loc, &eof); 569 DPRINTF(ADWLK, ("slot %d, eof = %d, flags = %d, " 570 "len = %d, lb_num = %d, part = %d\n", 571 slot, eof, 572 UDF_EXT_FLAGS(udf_rw32(s_icb_loc.len)), 573 UDF_EXT_LEN(udf_rw32(s_icb_loc.len)), 574 udf_rw32(s_icb_loc.loc.lb_num), 575 udf_rw16(s_icb_loc.loc.part_num))); 576 if (eof) { 577 DPRINTF(TRANSLATE, 578 ("Meta partition translation " 579 "failed: can't seek location\n")); 580 UDF_UNLOCK_NODE(ump->metadata_node, 0); 581 return EINVAL; 582 } 583 len = udf_rw32(s_icb_loc.len); 584 flags = UDF_EXT_FLAGS(len); 585 len = UDF_EXT_LEN(len); 586 587 if (flags == UDF_EXT_REDIRECT) { 588 slot++; 589 continue; 590 } 591 592 end_foffset = foffset + len; 593 594 if (end_foffset > lb_num * lb_size) 595 break; /* found */ 596 foffset = end_foffset; 597 slot++; 598 } 599 /* found overlapping slot */ 600 ext_offset = lb_num * lb_size - foffset; 601 602 /* process extent offset */ 603 lb_num = udf_rw32(s_icb_loc.loc.lb_num); 604 vpart = udf_rw16(s_icb_loc.loc.part_num); 605 lb_num += (ext_offset + lb_size -1) / lb_size; 606 ext_offset = 0; 607 608 UDF_UNLOCK_NODE(ump->metadata_node, 0); 609 if (flags != UDF_EXT_ALLOCATED) { 610 DPRINTF(TRANSLATE, ("Metadata partition translation " 611 "failed: not allocated\n")); 612 return EINVAL; 613 } 614 615 /* 616 * vpart and lb_num are updated, translate again since we 617 * might be mapped on sparable media 618 */ 619 goto translate_again; 620 default: 621 printf("UDF vtop translation scheme %d unimplemented yet\n", 622 ump->vtop_tp[vpart]); 623 } 624 625 return EINVAL; 626 } 627 628 629 /* XXX provisional primitive braindead version */ 630 /* TODO use ext_res */ 631 void 632 udf_translate_vtop_list(struct udf_mount *ump, uint32_t sectors, 633 uint16_t vpart_num, uint64_t *lmapping, uint64_t *pmapping) 634 { 635 struct long_ad loc; 636 uint32_t lb_numres, ext_res; 637 int sector; 638 639 for (sector = 0; sector < sectors; sector++) { 640 memset(&loc, 0, sizeof(struct long_ad)); 641 loc.loc.part_num = udf_rw16(vpart_num); 642 loc.loc.lb_num = udf_rw32(*lmapping); 643 udf_translate_vtop(ump, &loc, &lb_numres, &ext_res); 644 *pmapping = lb_numres; 645 lmapping++; pmapping++; 646 } 647 } 648 649 650 /* --------------------------------------------------------------------- */ 651 652 /* 653 * Translate an extent (in logical_blocks) into logical block numbers; used 654 * for read and write operations. DOESNT't check extents. 655 */ 656 657 int 658 udf_translate_file_extent(struct udf_node *udf_node, 659 uint32_t from, uint32_t num_lb, 660 uint64_t *map) 661 { 662 struct udf_mount *ump; 663 struct icb_tag *icbtag; 664 struct long_ad t_ad, s_ad; 665 uint64_t transsec; 666 uint64_t foffset, end_foffset; 667 uint32_t transsec32; 668 uint32_t lb_size; 669 uint32_t ext_offset; 670 uint32_t lb_num, len; 671 uint32_t overlap, translen; 672 uint16_t vpart_num; 673 int eof, error, flags; 674 int slot, addr_type, icbflags; 675 676 if (!udf_node) 677 return ENOENT; 678 679 KASSERT(num_lb > 0); 680 681 UDF_LOCK_NODE(udf_node, 0); 682 683 /* initialise derivative vars */ 684 ump = udf_node->ump; 685 lb_size = udf_rw32(ump->logical_vol->lb_size); 686 687 if (udf_node->fe) { 688 icbtag = &udf_node->fe->icbtag; 689 } else { 690 icbtag = &udf_node->efe->icbtag; 691 } 692 icbflags = udf_rw16(icbtag->flags); 693 addr_type = icbflags & UDF_ICB_TAG_FLAGS_ALLOC_MASK; 694 695 /* do the work */ 696 if (addr_type == UDF_ICB_INTERN_ALLOC) { 697 *map = UDF_TRANS_INTERN; 698 UDF_UNLOCK_NODE(udf_node, 0); 699 return 0; 700 } 701 702 /* find first overlapping extent */ 703 foffset = 0; 704 slot = 0; 705 for (;;) { 706 udf_get_adslot(udf_node, slot, &s_ad, &eof); 707 DPRINTF(ADWLK, ("slot %d, eof = %d, flags = %d, len = %d, " 708 "lb_num = %d, part = %d\n", slot, eof, 709 UDF_EXT_FLAGS(udf_rw32(s_ad.len)), 710 UDF_EXT_LEN(udf_rw32(s_ad.len)), 711 udf_rw32(s_ad.loc.lb_num), 712 udf_rw16(s_ad.loc.part_num))); 713 if (eof) { 714 DPRINTF(TRANSLATE, 715 ("Translate file extent " 716 "failed: can't seek location\n")); 717 UDF_UNLOCK_NODE(udf_node, 0); 718 return EINVAL; 719 } 720 len = udf_rw32(s_ad.len); 721 flags = UDF_EXT_FLAGS(len); 722 len = UDF_EXT_LEN(len); 723 lb_num = udf_rw32(s_ad.loc.lb_num); 724 725 if (flags == UDF_EXT_REDIRECT) { 726 slot++; 727 continue; 728 } 729 730 end_foffset = foffset + len; 731 732 if (end_foffset > from * lb_size) 733 break; /* found */ 734 foffset = end_foffset; 735 slot++; 736 } 737 /* found overlapping slot */ 738 ext_offset = from * lb_size - foffset; 739 740 for (;;) { 741 udf_get_adslot(udf_node, slot, &s_ad, &eof); 742 DPRINTF(ADWLK, ("slot %d, eof = %d, flags = %d, len = %d, " 743 "lb_num = %d, part = %d\n", slot, eof, 744 UDF_EXT_FLAGS(udf_rw32(s_ad.len)), 745 UDF_EXT_LEN(udf_rw32(s_ad.len)), 746 udf_rw32(s_ad.loc.lb_num), 747 udf_rw16(s_ad.loc.part_num))); 748 if (eof) { 749 DPRINTF(TRANSLATE, 750 ("Translate file extent " 751 "failed: past eof\n")); 752 UDF_UNLOCK_NODE(udf_node, 0); 753 return EINVAL; 754 } 755 756 len = udf_rw32(s_ad.len); 757 flags = UDF_EXT_FLAGS(len); 758 len = UDF_EXT_LEN(len); 759 760 lb_num = udf_rw32(s_ad.loc.lb_num); 761 vpart_num = udf_rw16(s_ad.loc.part_num); 762 763 end_foffset = foffset + len; 764 765 /* process extent, don't forget to advance on ext_offset! */ 766 lb_num += (ext_offset + lb_size -1) / lb_size; 767 overlap = (len - ext_offset + lb_size -1) / lb_size; 768 ext_offset = 0; 769 770 /* 771 * note that the while(){} is nessisary for the extent that 772 * the udf_translate_vtop() returns doens't have to span the 773 * whole extent. 774 */ 775 776 overlap = MIN(overlap, num_lb); 777 while (overlap && (flags != UDF_EXT_REDIRECT)) { 778 switch (flags) { 779 case UDF_EXT_FREE : 780 case UDF_EXT_ALLOCATED_BUT_NOT_USED : 781 transsec = UDF_TRANS_ZERO; 782 translen = overlap; 783 while (overlap && num_lb && translen) { 784 *map++ = transsec; 785 lb_num++; 786 overlap--; num_lb--; translen--; 787 } 788 break; 789 case UDF_EXT_ALLOCATED : 790 t_ad.loc.lb_num = udf_rw32(lb_num); 791 t_ad.loc.part_num = udf_rw16(vpart_num); 792 error = udf_translate_vtop(ump, 793 &t_ad, &transsec32, &translen); 794 transsec = transsec32; 795 if (error) { 796 UDF_UNLOCK_NODE(udf_node, 0); 797 return error; 798 } 799 while (overlap && num_lb && translen) { 800 *map++ = transsec; 801 lb_num++; transsec++; 802 overlap--; num_lb--; translen--; 803 } 804 break; 805 default: 806 DPRINTF(TRANSLATE, 807 ("Translate file extent " 808 "failed: bad flags %x\n", flags)); 809 UDF_UNLOCK_NODE(udf_node, 0); 810 return EINVAL; 811 } 812 } 813 if (num_lb == 0) 814 break; 815 816 if (flags != UDF_EXT_REDIRECT) 817 foffset = end_foffset; 818 slot++; 819 } 820 UDF_UNLOCK_NODE(udf_node, 0); 821 822 return 0; 823 } 824 825 /* --------------------------------------------------------------------- */ 826 827 static int 828 udf_search_free_vatloc(struct udf_mount *ump, uint32_t *lbnumres) 829 { 830 uint32_t lb_size, lb_num, lb_map, udf_rw32_lbmap; 831 uint8_t *blob; 832 int entry, chunk, found, error; 833 834 KASSERT(ump); 835 KASSERT(ump->logical_vol); 836 837 lb_size = udf_rw32(ump->logical_vol->lb_size); 838 blob = malloc(lb_size, M_UDFTEMP, M_WAITOK); 839 840 /* TODO static allocation of search chunk */ 841 842 lb_num = MIN(ump->vat_entries, ump->vat_last_free_lb); 843 found = 0; 844 error = 0; 845 entry = 0; 846 do { 847 chunk = MIN(lb_size, (ump->vat_entries - lb_num) * 4); 848 if (chunk <= 0) 849 break; 850 /* load in chunk */ 851 error = udf_vat_read(ump->vat_node, blob, chunk, 852 ump->vat_offset + lb_num * 4); 853 854 if (error) 855 break; 856 857 /* search this chunk */ 858 for (entry=0; entry < chunk /4; entry++, lb_num++) { 859 udf_rw32_lbmap = *((uint32_t *) (blob + entry * 4)); 860 lb_map = udf_rw32(udf_rw32_lbmap); 861 if (lb_map == 0xffffffff) { 862 found = 1; 863 break; 864 } 865 } 866 } while (!found); 867 if (error) { 868 printf("udf_search_free_vatloc: error reading in vat chunk " 869 "(lb %d, size %d)\n", lb_num, chunk); 870 } 871 872 if (!found) { 873 /* extend VAT */ 874 DPRINTF(WRITE, ("udf_search_free_vatloc: extending\n")); 875 lb_num = ump->vat_entries; 876 ump->vat_entries++; 877 } 878 879 /* mark entry with initialiser just in case */ 880 lb_map = udf_rw32(0xfffffffe); 881 udf_vat_write(ump->vat_node, (uint8_t *) &lb_map, 4, 882 ump->vat_offset + lb_num *4); 883 ump->vat_last_free_lb = lb_num; 884 885 free(blob, M_UDFTEMP); 886 *lbnumres = lb_num; 887 return 0; 888 } 889 890 891 static void 892 udf_bitmap_allocate(struct udf_bitmap *bitmap, int ismetadata, 893 uint32_t *num_lb, uint64_t *lmappos) 894 { 895 uint32_t offset, lb_num, bit; 896 int32_t diff; 897 uint8_t *bpos; 898 int pass; 899 900 if (!ismetadata) { 901 /* heuristic to keep the two pointers not too close */ 902 diff = bitmap->data_pos - bitmap->metadata_pos; 903 if ((diff >= 0) && (diff < 1024)) 904 bitmap->data_pos = bitmap->metadata_pos + 1024; 905 } 906 offset = ismetadata ? bitmap->metadata_pos : bitmap->data_pos; 907 offset &= ~7; 908 for (pass = 0; pass < 2; pass++) { 909 if (offset >= bitmap->max_offset) 910 offset = 0; 911 912 while (offset < bitmap->max_offset) { 913 if (*num_lb == 0) 914 break; 915 916 /* use first bit not set */ 917 bpos = bitmap->bits + offset/8; 918 bit = ffs(*bpos); /* returns 0 or 1..8 */ 919 if (bit == 0) { 920 offset += 8; 921 continue; 922 } 923 924 /* check for ffs overshoot */ 925 if (offset + bit-1 >= bitmap->max_offset) { 926 offset = bitmap->max_offset; 927 break; 928 } 929 930 DPRINTF(PARANOIA, ("XXX : allocate %d, %p, bit %d\n", 931 offset + bit -1, bpos, bit-1)); 932 *bpos &= ~(1 << (bit-1)); 933 lb_num = offset + bit-1; 934 *lmappos++ = lb_num; 935 *num_lb = *num_lb - 1; 936 // offset = (offset & ~7); 937 } 938 } 939 940 if (ismetadata) { 941 bitmap->metadata_pos = offset; 942 } else { 943 bitmap->data_pos = offset; 944 } 945 } 946 947 948 static void 949 udf_bitmap_free(struct udf_bitmap *bitmap, uint32_t lb_num, uint32_t num_lb) 950 { 951 uint32_t offset; 952 uint32_t bit, bitval; 953 uint8_t *bpos; 954 955 offset = lb_num; 956 957 /* starter bits */ 958 bpos = bitmap->bits + offset/8; 959 bit = offset % 8; 960 while ((bit != 0) && (num_lb > 0)) { 961 bitval = (1 << bit); 962 KASSERT((*bpos & bitval) == 0); 963 DPRINTF(PARANOIA, ("XXX : free %d, %p, %d\n", 964 offset, bpos, bit)); 965 *bpos |= bitval; 966 offset++; num_lb--; 967 bit = (bit + 1) % 8; 968 } 969 if (num_lb == 0) 970 return; 971 972 /* whole bytes */ 973 KASSERT(bit == 0); 974 bpos = bitmap->bits + offset / 8; 975 while (num_lb >= 8) { 976 KASSERT((*bpos == 0)); 977 DPRINTF(PARANOIA, ("XXX : free %d + 8, %p\n", offset, bpos)); 978 *bpos = 255; 979 offset += 8; num_lb -= 8; 980 bpos++; 981 } 982 983 /* stop bits */ 984 KASSERT(num_lb < 8); 985 bit = 0; 986 while (num_lb > 0) { 987 bitval = (1 << bit); 988 KASSERT((*bpos & bitval) == 0); 989 DPRINTF(PARANOIA, ("XXX : free %d, %p, %d\n", 990 offset, bpos, bit)); 991 *bpos |= bitval; 992 offset++; num_lb--; 993 bit = (bit + 1) % 8; 994 } 995 } 996 997 998 static uint32_t 999 udf_bitmap_check_trunc_free(struct udf_bitmap *bitmap, uint32_t to_trunc) 1000 { 1001 uint32_t seq_free, offset; 1002 uint8_t *bpos; 1003 uint8_t bit, bitval; 1004 1005 DPRINTF(RESERVE, ("\ttrying to trunc %d bits from bitmap\n", to_trunc)); 1006 offset = bitmap->max_offset - to_trunc; 1007 1008 /* starter bits (if any) */ 1009 bpos = bitmap->bits + offset/8; 1010 bit = offset % 8; 1011 seq_free = 0; 1012 while (to_trunc > 0) { 1013 seq_free++; 1014 bitval = (1 << bit); 1015 if (!(*bpos & bitval)) 1016 seq_free = 0; 1017 offset++; to_trunc--; 1018 bit++; 1019 if (bit == 8) { 1020 bpos++; 1021 bit = 0; 1022 } 1023 } 1024 1025 DPRINTF(RESERVE, ("\tfound %d sequential free bits in bitmap\n", seq_free)); 1026 return seq_free; 1027 } 1028 1029 /* --------------------------------------------------------------------- */ 1030 1031 /* 1032 * We check for overall disc space with a margin to prevent critical 1033 * conditions. If disc space is low we try to force a sync() to improve our 1034 * estimates. When confronted with meta-data partition size shortage we know 1035 * we have to check if it can be extended and we need to extend it when 1036 * needed. 1037 * 1038 * A 2nd strategy we could use when disc space is getting low on a disc 1039 * formatted with a meta-data partition is to see if there are sparse areas in 1040 * the meta-data partition and free blocks there for extra data. 1041 */ 1042 1043 void 1044 udf_do_reserve_space(struct udf_mount *ump, struct udf_node *udf_node, 1045 uint16_t vpart_num, uint32_t num_lb) 1046 { 1047 ump->uncommitted_lbs[vpart_num] += num_lb; 1048 if (udf_node) 1049 udf_node->uncommitted_lbs += num_lb; 1050 } 1051 1052 1053 void 1054 udf_do_unreserve_space(struct udf_mount *ump, struct udf_node *udf_node, 1055 uint16_t vpart_num, uint32_t num_lb) 1056 { 1057 ump->uncommitted_lbs[vpart_num] -= num_lb; 1058 if (ump->uncommitted_lbs[vpart_num] < 0) { 1059 DPRINTF(RESERVE, ("UDF: underflow on partition reservation, " 1060 "part %d: %d\n", vpart_num, 1061 ump->uncommitted_lbs[vpart_num])); 1062 ump->uncommitted_lbs[vpart_num] = 0; 1063 } 1064 if (udf_node) { 1065 udf_node->uncommitted_lbs -= num_lb; 1066 if (udf_node->uncommitted_lbs < 0) { 1067 DPRINTF(RESERVE, ("UDF: underflow of node " 1068 "reservation : %d\n", 1069 udf_node->uncommitted_lbs)); 1070 udf_node->uncommitted_lbs = 0; 1071 } 1072 } 1073 } 1074 1075 1076 int 1077 udf_reserve_space(struct udf_mount *ump, struct udf_node *udf_node, 1078 int udf_c_type, uint16_t vpart_num, uint32_t num_lb, int can_fail) 1079 { 1080 uint64_t freeblks; 1081 uint64_t slack; 1082 int i, error; 1083 1084 slack = 0; 1085 if (can_fail) 1086 slack = UDF_DISC_SLACK; 1087 1088 error = 0; 1089 mutex_enter(&ump->allocate_mutex); 1090 1091 /* check if there is enough space available */ 1092 for (i = 0; i < 3; i++) { /* XXX arbitrary number */ 1093 udf_calc_vpart_freespace(ump, vpart_num, &freeblks); 1094 if (num_lb + slack < freeblks) 1095 break; 1096 /* issue SYNC */ 1097 DPRINTF(RESERVE, ("udf_reserve_space: issuing sync\n")); 1098 mutex_exit(&ump->allocate_mutex); 1099 udf_do_sync(ump, FSCRED, 0); 1100 mutex_enter(&mntvnode_lock); 1101 /* 1/8 second wait */ 1102 cv_timedwait(&ump->dirtynodes_cv, &mntvnode_lock, 1103 hz/8); 1104 mutex_exit(&mntvnode_lock); 1105 mutex_enter(&ump->allocate_mutex); 1106 } 1107 1108 /* check if there is enough space available now */ 1109 udf_calc_vpart_freespace(ump, vpart_num, &freeblks); 1110 if (num_lb + slack >= freeblks) { 1111 DPRINTF(RESERVE, ("udf_reserve_space: try to redistribute " 1112 "partition space\n")); 1113 DPRINTF(RESERVE, ("\tvpart %d, type %d is full\n", 1114 vpart_num, ump->vtop_alloc[vpart_num])); 1115 /* Try to redistribute space if possible */ 1116 udf_collect_free_space_for_vpart(ump, vpart_num, num_lb + slack); 1117 } 1118 1119 /* check if there is enough space available now */ 1120 udf_calc_vpart_freespace(ump, vpart_num, &freeblks); 1121 if (num_lb + slack <= freeblks) { 1122 udf_do_reserve_space(ump, udf_node, vpart_num, num_lb); 1123 } else { 1124 DPRINTF(RESERVE, ("udf_reserve_space: out of disc space\n")); 1125 error = ENOSPC; 1126 } 1127 1128 mutex_exit(&ump->allocate_mutex); 1129 return error; 1130 } 1131 1132 1133 void 1134 udf_cleanup_reservation(struct udf_node *udf_node) 1135 { 1136 struct udf_mount *ump = udf_node->ump; 1137 int vpart_num; 1138 1139 mutex_enter(&ump->allocate_mutex); 1140 1141 /* compensate for overlapping blocks */ 1142 DPRINTF(RESERVE, ("UDF: overlapped %d blocks in count\n", udf_node->uncommitted_lbs)); 1143 1144 vpart_num = udf_get_record_vpart(ump, udf_get_c_type(udf_node)); 1145 udf_do_unreserve_space(ump, udf_node, vpart_num, udf_node->uncommitted_lbs); 1146 1147 DPRINTF(RESERVE, ("\ttotal now %d\n", ump->uncommitted_lbs[vpart_num])); 1148 1149 /* sanity */ 1150 if (ump->uncommitted_lbs[vpart_num] < 0) 1151 ump->uncommitted_lbs[vpart_num] = 0; 1152 1153 mutex_exit(&ump->allocate_mutex); 1154 } 1155 1156 /* --------------------------------------------------------------------- */ 1157 1158 /* 1159 * Allocate an extent of given length on given virt. partition. It doesn't 1160 * have to be one stretch. 1161 */ 1162 1163 int 1164 udf_allocate_space(struct udf_mount *ump, struct udf_node *udf_node, 1165 int udf_c_type, uint16_t vpart_num, uint32_t num_lb, uint64_t *lmapping) 1166 { 1167 struct mmc_trackinfo *alloc_track, *other_track; 1168 struct udf_bitmap *bitmap; 1169 struct part_desc *pdesc; 1170 struct logvol_int_desc *lvid; 1171 uint64_t *lmappos; 1172 uint32_t ptov, lb_num, *freepos, free_lbs; 1173 int lb_size, alloc_num_lb; 1174 int alloc_type, error; 1175 int is_node; 1176 1177 DPRINTF(CALL, ("udf_allocate_space(ctype %d, vpart %d, num_lb %d\n", 1178 udf_c_type, vpart_num, num_lb)); 1179 mutex_enter(&ump->allocate_mutex); 1180 1181 lb_size = udf_rw32(ump->logical_vol->lb_size); 1182 KASSERT(lb_size == ump->discinfo.sector_size); 1183 1184 alloc_type = ump->vtop_alloc[vpart_num]; 1185 is_node = (udf_c_type == UDF_C_NODE); 1186 1187 lmappos = lmapping; 1188 error = 0; 1189 switch (alloc_type) { 1190 case UDF_ALLOC_VAT : 1191 /* search empty slot in VAT file */ 1192 KASSERT(num_lb == 1); 1193 error = udf_search_free_vatloc(ump, &lb_num); 1194 if (!error) { 1195 *lmappos = lb_num; 1196 1197 /* reserve on the backing sequential partition since 1198 * that partition is credited back later */ 1199 udf_do_reserve_space(ump, udf_node, 1200 ump->vtop[vpart_num], num_lb); 1201 } 1202 break; 1203 case UDF_ALLOC_SEQUENTIAL : 1204 /* sequential allocation on recordable media */ 1205 /* get partition backing up this vpart_num_num */ 1206 pdesc = ump->partitions[ump->vtop[vpart_num]]; 1207 1208 /* calculate offset from physical base partition */ 1209 ptov = udf_rw32(pdesc->start_loc); 1210 1211 /* get our track descriptors */ 1212 if (vpart_num == ump->node_part) { 1213 alloc_track = &ump->metadata_track; 1214 other_track = &ump->data_track; 1215 } else { 1216 alloc_track = &ump->data_track; 1217 other_track = &ump->metadata_track; 1218 } 1219 1220 /* allocate */ 1221 for (lb_num = 0; lb_num < num_lb; lb_num++) { 1222 *lmappos++ = alloc_track->next_writable - ptov; 1223 alloc_track->next_writable++; 1224 alloc_track->free_blocks--; 1225 } 1226 1227 /* keep other track up-to-date */ 1228 if (alloc_track->tracknr == other_track->tracknr) 1229 memcpy(other_track, alloc_track, 1230 sizeof(struct mmc_trackinfo)); 1231 break; 1232 case UDF_ALLOC_SPACEMAP : 1233 /* try to allocate on unallocated bits */ 1234 alloc_num_lb = num_lb; 1235 bitmap = &ump->part_unalloc_bits[vpart_num]; 1236 udf_bitmap_allocate(bitmap, is_node, &alloc_num_lb, lmappos); 1237 ump->lvclose |= UDF_WRITE_PART_BITMAPS; 1238 1239 /* have we allocated all? */ 1240 if (alloc_num_lb) { 1241 /* TODO convert freed to unalloc and try again */ 1242 /* free allocated piece for now */ 1243 lmappos = lmapping; 1244 for (lb_num=0; lb_num < num_lb-alloc_num_lb; lb_num++) { 1245 udf_bitmap_free(bitmap, *lmappos++, 1); 1246 } 1247 error = ENOSPC; 1248 } 1249 if (!error) { 1250 /* adjust freecount */ 1251 lvid = ump->logvol_integrity; 1252 freepos = &lvid->tables[0] + vpart_num; 1253 free_lbs = udf_rw32(*freepos); 1254 *freepos = udf_rw32(free_lbs - num_lb); 1255 } 1256 break; 1257 case UDF_ALLOC_METABITMAP : /* UDF 2.50, 2.60 BluRay-RE */ 1258 /* allocate on metadata unallocated bits */ 1259 alloc_num_lb = num_lb; 1260 bitmap = &ump->metadata_unalloc_bits; 1261 udf_bitmap_allocate(bitmap, is_node, &alloc_num_lb, lmappos); 1262 ump->lvclose |= UDF_WRITE_PART_BITMAPS; 1263 1264 /* have we allocated all? */ 1265 if (alloc_num_lb) { 1266 /* YIKES! TODO we need to extend the metadata partition */ 1267 /* free allocated piece for now */ 1268 lmappos = lmapping; 1269 for (lb_num=0; lb_num < num_lb-alloc_num_lb; lb_num++) { 1270 udf_bitmap_free(bitmap, *lmappos++, 1); 1271 } 1272 error = ENOSPC; 1273 } 1274 if (!error) { 1275 /* adjust freecount */ 1276 lvid = ump->logvol_integrity; 1277 freepos = &lvid->tables[0] + vpart_num; 1278 free_lbs = udf_rw32(*freepos); 1279 *freepos = udf_rw32(free_lbs - num_lb); 1280 } 1281 break; 1282 case UDF_ALLOC_METASEQUENTIAL : /* UDF 2.60 BluRay-R */ 1283 case UDF_ALLOC_RELAXEDSEQUENTIAL : /* UDF 2.50/~meta BluRay-R */ 1284 printf("ALERT: udf_allocate_space : allocation %d " 1285 "not implemented yet!\n", alloc_type); 1286 /* TODO implement, doesn't have to be contiguous */ 1287 error = ENOSPC; 1288 break; 1289 } 1290 1291 if (!error) { 1292 /* credit our partition since we have committed the space */ 1293 udf_do_unreserve_space(ump, udf_node, vpart_num, num_lb); 1294 } 1295 1296 #ifdef DEBUG 1297 if (udf_verbose & UDF_DEBUG_ALLOC) { 1298 lmappos = lmapping; 1299 printf("udf_allocate_space, allocated logical lba :\n"); 1300 for (lb_num = 0; lb_num < num_lb; lb_num++) { 1301 printf("%s %"PRIu64, (lb_num > 0)?",":"", 1302 *lmappos++); 1303 } 1304 printf("\n"); 1305 } 1306 #endif 1307 mutex_exit(&ump->allocate_mutex); 1308 1309 return error; 1310 } 1311 1312 /* --------------------------------------------------------------------- */ 1313 1314 void 1315 udf_free_allocated_space(struct udf_mount *ump, uint32_t lb_num, 1316 uint16_t vpart_num, uint32_t num_lb) 1317 { 1318 struct udf_bitmap *bitmap; 1319 struct part_desc *pdesc; 1320 struct logvol_int_desc *lvid; 1321 uint32_t ptov, lb_map, udf_rw32_lbmap; 1322 uint32_t *freepos, free_lbs; 1323 int phys_part; 1324 int error; 1325 1326 DPRINTF(ALLOC, ("udf_free_allocated_space: freeing virt lbnum %d " 1327 "part %d + %d sect\n", lb_num, vpart_num, num_lb)); 1328 1329 /* no use freeing zero length */ 1330 if (num_lb == 0) 1331 return; 1332 1333 mutex_enter(&ump->allocate_mutex); 1334 1335 /* get partition backing up this vpart_num */ 1336 pdesc = ump->partitions[ump->vtop[vpart_num]]; 1337 1338 switch (ump->vtop_tp[vpart_num]) { 1339 case UDF_VTOP_TYPE_PHYS : 1340 case UDF_VTOP_TYPE_SPARABLE : 1341 /* free space to freed or unallocated space bitmap */ 1342 ptov = udf_rw32(pdesc->start_loc); 1343 phys_part = ump->vtop[vpart_num]; 1344 1345 /* first try freed space bitmap */ 1346 bitmap = &ump->part_freed_bits[phys_part]; 1347 1348 /* if not defined, use unallocated bitmap */ 1349 if (bitmap->bits == NULL) 1350 bitmap = &ump->part_unalloc_bits[phys_part]; 1351 1352 /* if no bitmaps are defined, bail out; XXX OK? */ 1353 if (bitmap->bits == NULL) 1354 break; 1355 1356 /* free bits if its defined */ 1357 KASSERT(bitmap->bits); 1358 ump->lvclose |= UDF_WRITE_PART_BITMAPS; 1359 udf_bitmap_free(bitmap, lb_num, num_lb); 1360 1361 /* adjust freecount */ 1362 lvid = ump->logvol_integrity; 1363 freepos = &lvid->tables[0] + vpart_num; 1364 free_lbs = udf_rw32(*freepos); 1365 *freepos = udf_rw32(free_lbs + num_lb); 1366 break; 1367 case UDF_VTOP_TYPE_VIRT : 1368 /* free this VAT entry */ 1369 KASSERT(num_lb == 1); 1370 1371 lb_map = 0xffffffff; 1372 udf_rw32_lbmap = udf_rw32(lb_map); 1373 error = udf_vat_write(ump->vat_node, 1374 (uint8_t *) &udf_rw32_lbmap, 4, 1375 ump->vat_offset + lb_num * 4); 1376 KASSERT(error == 0); 1377 ump->vat_last_free_lb = MIN(ump->vat_last_free_lb, lb_num); 1378 break; 1379 case UDF_VTOP_TYPE_META : 1380 /* free space in the metadata bitmap */ 1381 bitmap = &ump->metadata_unalloc_bits; 1382 KASSERT(bitmap->bits); 1383 1384 ump->lvclose |= UDF_WRITE_PART_BITMAPS; 1385 udf_bitmap_free(bitmap, lb_num, num_lb); 1386 1387 /* adjust freecount */ 1388 lvid = ump->logvol_integrity; 1389 freepos = &lvid->tables[0] + vpart_num; 1390 free_lbs = udf_rw32(*freepos); 1391 *freepos = udf_rw32(free_lbs + num_lb); 1392 break; 1393 default: 1394 printf("ALERT: udf_free_allocated_space : allocation %d " 1395 "not implemented yet!\n", ump->vtop_tp[vpart_num]); 1396 break; 1397 } 1398 1399 mutex_exit(&ump->allocate_mutex); 1400 } 1401 1402 /* --------------------------------------------------------------------- */ 1403 1404 /* 1405 * Special function to synchronise the metadatamirror file when they change on 1406 * resizing. When the metadatafile is actually duplicated, this action is a 1407 * no-op since they describe different extents on the disc. 1408 */ 1409 1410 void udf_synchronise_metadatamirror_node(struct udf_mount *ump) 1411 { 1412 struct udf_node *meta_node, *metamirror_node; 1413 struct long_ad s_ad; 1414 int slot, cpy_slot; 1415 int error, eof; 1416 1417 if (ump->metadata_flags & METADATA_DUPLICATED) 1418 return; 1419 1420 meta_node = ump->metadata_node; 1421 metamirror_node = ump->metadatamirror_node; 1422 1423 /* 1) wipe mirror node */ 1424 udf_wipe_adslots(metamirror_node); 1425 1426 /* 2) copy all node descriptors from the meta_node */ 1427 slot = 0; 1428 cpy_slot = 0; 1429 for (;;) { 1430 udf_get_adslot(meta_node, slot, &s_ad, &eof); 1431 if (eof) 1432 break; 1433 error = udf_append_adslot(metamirror_node, &cpy_slot, &s_ad); 1434 if (error) { 1435 /* WTF, this shouldn't happen, what to do now? */ 1436 panic("udf_synchronise_metadatamirror_node failed!"); 1437 } 1438 slot++; 1439 } 1440 1441 /* 3) adjust metamirror_node size */ 1442 if (meta_node->fe) { 1443 KASSERT(metamirror_node->fe); 1444 metamirror_node->fe->inf_len = meta_node->fe->inf_len; 1445 } else { 1446 KASSERT(meta_node->efe); 1447 KASSERT(metamirror_node->efe); 1448 metamirror_node->efe->inf_len = meta_node->efe->inf_len; 1449 metamirror_node->efe->obj_size = meta_node->efe->obj_size; 1450 } 1451 1452 /* for sanity */ 1453 udf_count_alloc_exts(metamirror_node); 1454 } 1455 1456 /* --------------------------------------------------------------------- */ 1457 1458 /* 1459 * When faced with an out of space but there is still space available on other 1460 * partitions, try to redistribute the space. This is only defined for media 1461 * using Metadata partitions. 1462 * 1463 * There are two formats to deal with. Either its a `normal' metadata 1464 * partition and we can move blocks between a metadata bitmap and its 1465 * companion data spacemap OR its a UDF 2.60 formatted BluRay-R disc with POW 1466 * and a metadata partition. 1467 */ 1468 1469 static uint32_t 1470 udf_trunc_metadatapart(struct udf_mount *ump, uint32_t num_lb) 1471 { 1472 struct udf_node *bitmap_node; 1473 struct udf_bitmap *bitmap; 1474 struct space_bitmap_desc *sbd, *new_sbd; 1475 struct logvol_int_desc *lvid; 1476 uint64_t inf_len; 1477 uint64_t meta_free_lbs, data_free_lbs; 1478 uint32_t *freepos, *sizepos; 1479 uint32_t unit, lb_size, to_trunc; 1480 uint16_t meta_vpart_num, data_vpart_num, num_vpart; 1481 int err; 1482 1483 unit = ump->metadata_alloc_unit_size; 1484 lb_size = udf_rw32(ump->logical_vol->lb_size); 1485 lvid = ump->logvol_integrity; 1486 1487 /* lookup vpart for metadata partition */ 1488 meta_vpart_num = ump->node_part; 1489 KASSERT(ump->vtop_alloc[meta_vpart_num] == UDF_ALLOC_METABITMAP); 1490 1491 /* lookup vpart for data partition */ 1492 data_vpart_num = ump->data_part; 1493 KASSERT(ump->vtop_alloc[data_vpart_num] == UDF_ALLOC_SPACEMAP); 1494 1495 udf_calc_vpart_freespace(ump, data_vpart_num, &data_free_lbs); 1496 udf_calc_vpart_freespace(ump, meta_vpart_num, &meta_free_lbs); 1497 1498 DPRINTF(RESERVE, ("\tfree space on data partition %"PRIu64" blks\n", data_free_lbs)); 1499 DPRINTF(RESERVE, ("\tfree space on metadata partition %"PRIu64" blks\n", meta_free_lbs)); 1500 1501 /* give away some of the free meta space, in unit block sizes */ 1502 to_trunc = meta_free_lbs/4; /* give out a quarter */ 1503 to_trunc = MAX(to_trunc, num_lb); 1504 to_trunc = unit * ((to_trunc + unit-1) / unit); /* round up */ 1505 1506 /* scale down if needed and bail out when out of space */ 1507 if (to_trunc >= meta_free_lbs) 1508 return num_lb; 1509 1510 /* check extent of bits marked free at the end of the map */ 1511 bitmap = &ump->metadata_unalloc_bits; 1512 to_trunc = udf_bitmap_check_trunc_free(bitmap, to_trunc); 1513 to_trunc = unit * (to_trunc / unit); /* round down again */ 1514 if (to_trunc == 0) 1515 return num_lb; 1516 1517 DPRINTF(RESERVE, ("\ttruncating %d lbs from the metadata bitmap\n", 1518 to_trunc)); 1519 1520 /* get length of the metadata bitmap node file */ 1521 bitmap_node = ump->metadatabitmap_node; 1522 if (bitmap_node->fe) { 1523 inf_len = udf_rw64(bitmap_node->fe->inf_len); 1524 } else { 1525 KASSERT(bitmap_node->efe); 1526 inf_len = udf_rw64(bitmap_node->efe->inf_len); 1527 } 1528 inf_len -= to_trunc/8; 1529 1530 /* as per [UDF 2.60/2.2.13.6] : */ 1531 /* 1) update the SBD in the metadata bitmap file */ 1532 sbd = (struct space_bitmap_desc *) bitmap->blob; 1533 sbd->num_bits = udf_rw32(sbd->num_bits) - to_trunc; 1534 sbd->num_bytes = udf_rw32(sbd->num_bytes) - to_trunc/8; 1535 bitmap->max_offset = udf_rw32(sbd->num_bits); 1536 1537 num_vpart = udf_rw32(lvid->num_part); 1538 freepos = &lvid->tables[0] + meta_vpart_num; 1539 sizepos = &lvid->tables[0] + num_vpart + meta_vpart_num; 1540 *freepos = udf_rw32(*freepos) - to_trunc; 1541 *sizepos = udf_rw32(*sizepos) - to_trunc; 1542 1543 /* realloc bitmap for better memory usage */ 1544 new_sbd = realloc(sbd, inf_len, M_UDFVOLD, 1545 M_CANFAIL | M_WAITOK); 1546 if (new_sbd) { 1547 /* update pointers */ 1548 ump->metadata_unalloc_dscr = new_sbd; 1549 bitmap->blob = (uint8_t *) new_sbd; 1550 } 1551 ump->lvclose |= UDF_WRITE_PART_BITMAPS; 1552 1553 /* 1554 * The truncated space is secured now and can't be allocated anymore. Release 1555 * the allocate mutex so we can shrink the nodes the normal way. 1556 */ 1557 mutex_exit(&ump->allocate_mutex); 1558 1559 /* 2) trunc the metadata bitmap information file, freeing blocks */ 1560 err = udf_shrink_node(bitmap_node, inf_len); 1561 KASSERT(err == 0); 1562 1563 /* 3) trunc the metadata file and mirror file, freeing blocks */ 1564 inf_len = udf_rw32(sbd->num_bits) * lb_size; /* [4/14.12.4] */ 1565 err = udf_shrink_node(ump->metadata_node, inf_len); 1566 KASSERT(err == 0); 1567 if (ump->metadatamirror_node && (ump->metadata_flags & METADATA_DUPLICATED)) { 1568 err = udf_shrink_node(ump->metadatamirror_node, inf_len); 1569 KASSERT(err == 0); 1570 } 1571 ump->lvclose |= UDF_WRITE_METAPART_NODES; 1572 1573 /* relock before exit */ 1574 mutex_enter(&ump->allocate_mutex); 1575 1576 if (to_trunc > num_lb) 1577 return 0; 1578 return num_lb - to_trunc; 1579 } 1580 1581 1582 static void 1583 udf_sparsify_metadatapart(struct udf_mount *ump, uint32_t num_lb) 1584 { 1585 /* NOT IMPLEMENTED, fail */ 1586 } 1587 1588 1589 static void 1590 udf_collect_free_space_for_vpart(struct udf_mount *ump, 1591 uint16_t vpart_num, uint32_t num_lb) 1592 { 1593 /* allocate mutex is helt */ 1594 1595 /* only defined for metadata partitions */ 1596 if (ump->vtop_tp[ump->node_part] != UDF_VTOP_TYPE_META) { 1597 DPRINTF(RESERVE, ("\tcan't grow/shrink; no metadata partitioning\n")); 1598 return; 1599 } 1600 1601 /* UDF 2.60 BD-R+POW? */ 1602 if (ump->vtop_alloc[ump->node_part] == UDF_ALLOC_METASEQUENTIAL) { 1603 DPRINTF(RESERVE, ("\tUDF 2.60 BD-R+POW track grow not implemented yet\n")); 1604 return; 1605 } 1606 1607 if (ump->vtop_tp[vpart_num] == UDF_VTOP_TYPE_META) { 1608 /* try to grow the meta partition */ 1609 DPRINTF(RESERVE, ("\ttrying to grow the meta partition\n")); 1610 /* as per [UDF 2.60/2.2.13.5] : extend bitmap and metadata file(s) */ 1611 DPRINTF(NOTIMPL, ("\tgrowing meta partition not implemented yet\n")); 1612 } else { 1613 /* try to shrink the metadata partition */ 1614 DPRINTF(RESERVE, ("\ttrying to shrink the meta partition\n")); 1615 /* as per [UDF 2.60/2.2.13.6] : either trunc or make sparse */ 1616 num_lb = udf_trunc_metadatapart(ump, num_lb); 1617 if (num_lb) 1618 udf_sparsify_metadatapart(ump, num_lb); 1619 } 1620 1621 /* allocate mutex should still be helt */ 1622 } 1623 1624 /* --------------------------------------------------------------------- */ 1625 1626 /* 1627 * Allocate a buf on disc for direct write out. The space doesn't have to be 1628 * contiguous as the caller takes care of this. 1629 */ 1630 1631 void 1632 udf_late_allocate_buf(struct udf_mount *ump, struct buf *buf, 1633 uint64_t *lmapping, struct long_ad *node_ad_cpy, uint16_t *vpart_nump) 1634 { 1635 struct udf_node *udf_node = VTOI(buf->b_vp); 1636 int lb_size, blks, udf_c_type; 1637 int vpart_num, num_lb; 1638 int error, s; 1639 1640 /* 1641 * for each sector in the buf, allocate a sector on disc and record 1642 * its position in the provided mapping array. 1643 * 1644 * If its userdata or FIDs, record its location in its node. 1645 */ 1646 1647 lb_size = udf_rw32(ump->logical_vol->lb_size); 1648 num_lb = (buf->b_bcount + lb_size -1) / lb_size; 1649 blks = lb_size / DEV_BSIZE; 1650 udf_c_type = buf->b_udf_c_type; 1651 1652 KASSERT(lb_size == ump->discinfo.sector_size); 1653 1654 /* select partition to record the buffer on */ 1655 vpart_num = *vpart_nump = udf_get_record_vpart(ump, udf_c_type); 1656 1657 if (udf_c_type == UDF_C_NODE) { 1658 /* if not VAT, its allready allocated */ 1659 if (ump->vtop_alloc[ump->node_part] != UDF_ALLOC_VAT) 1660 return; 1661 1662 /* allocate on its backing sequential partition */ 1663 vpart_num = ump->data_part; 1664 } 1665 1666 /* XXX can this still happen? */ 1667 /* do allocation on the selected partition */ 1668 error = udf_allocate_space(ump, udf_node, udf_c_type, 1669 vpart_num, num_lb, lmapping); 1670 if (error) { 1671 /* 1672 * ARGH! we haven't done our accounting right! it should 1673 * allways succeed. 1674 */ 1675 panic("UDF disc allocation accounting gone wrong"); 1676 } 1677 1678 /* If its userdata or FIDs, record its allocation in its node. */ 1679 if ((udf_c_type == UDF_C_USERDATA) || 1680 (udf_c_type == UDF_C_FIDS) || 1681 (udf_c_type == UDF_C_METADATA_SBM)) 1682 { 1683 udf_record_allocation_in_node(ump, buf, vpart_num, lmapping, 1684 node_ad_cpy); 1685 /* decrement our outstanding bufs counter */ 1686 s = splbio(); 1687 udf_node->outstanding_bufs--; 1688 splx(s); 1689 } 1690 } 1691 1692 /* --------------------------------------------------------------------- */ 1693 1694 /* 1695 * Try to merge a1 with the new piece a2. udf_ads_merge returns error when not 1696 * possible (anymore); a2 returns the rest piece. 1697 */ 1698 1699 static int 1700 udf_ads_merge(uint32_t lb_size, struct long_ad *a1, struct long_ad *a2) 1701 { 1702 uint32_t max_len, merge_len; 1703 uint32_t a1_len, a2_len; 1704 uint32_t a1_flags, a2_flags; 1705 uint32_t a1_lbnum, a2_lbnum; 1706 uint16_t a1_part, a2_part; 1707 1708 max_len = ((UDF_EXT_MAXLEN / lb_size) * lb_size); 1709 1710 a1_flags = UDF_EXT_FLAGS(udf_rw32(a1->len)); 1711 a1_len = UDF_EXT_LEN(udf_rw32(a1->len)); 1712 a1_lbnum = udf_rw32(a1->loc.lb_num); 1713 a1_part = udf_rw16(a1->loc.part_num); 1714 1715 a2_flags = UDF_EXT_FLAGS(udf_rw32(a2->len)); 1716 a2_len = UDF_EXT_LEN(udf_rw32(a2->len)); 1717 a2_lbnum = udf_rw32(a2->loc.lb_num); 1718 a2_part = udf_rw16(a2->loc.part_num); 1719 1720 /* defines same space */ 1721 if (a1_flags != a2_flags) 1722 return 1; 1723 1724 if (a1_flags != UDF_EXT_FREE) { 1725 /* the same partition */ 1726 if (a1_part != a2_part) 1727 return 1; 1728 1729 /* a2 is successor of a1 */ 1730 if (a1_lbnum * lb_size + a1_len != a2_lbnum * lb_size) 1731 return 1; 1732 } 1733 1734 /* merge as most from a2 if possible */ 1735 merge_len = MIN(a2_len, max_len - a1_len); 1736 a1_len += merge_len; 1737 a2_len -= merge_len; 1738 a2_lbnum += merge_len/lb_size; 1739 1740 a1->len = udf_rw32(a1_len | a1_flags); 1741 a2->len = udf_rw32(a2_len | a2_flags); 1742 a2->loc.lb_num = udf_rw32(a2_lbnum); 1743 1744 if (a2_len > 0) 1745 return 1; 1746 1747 /* there is space over to merge */ 1748 return 0; 1749 } 1750 1751 /* --------------------------------------------------------------------- */ 1752 1753 static void 1754 udf_wipe_adslots(struct udf_node *udf_node) 1755 { 1756 struct file_entry *fe; 1757 struct extfile_entry *efe; 1758 struct alloc_ext_entry *ext; 1759 uint64_t inflen, objsize; 1760 uint32_t lb_size, dscr_size, l_ea, l_ad, max_l_ad, crclen; 1761 uint8_t *data_pos; 1762 int extnr; 1763 1764 lb_size = udf_rw32(udf_node->ump->logical_vol->lb_size); 1765 1766 fe = udf_node->fe; 1767 efe = udf_node->efe; 1768 if (fe) { 1769 inflen = udf_rw64(fe->inf_len); 1770 objsize = inflen; 1771 dscr_size = sizeof(struct file_entry) -1; 1772 l_ea = udf_rw32(fe->l_ea); 1773 l_ad = udf_rw32(fe->l_ad); 1774 data_pos = (uint8_t *) fe + dscr_size + l_ea; 1775 } else { 1776 inflen = udf_rw64(efe->inf_len); 1777 objsize = udf_rw64(efe->obj_size); 1778 dscr_size = sizeof(struct extfile_entry) -1; 1779 l_ea = udf_rw32(efe->l_ea); 1780 l_ad = udf_rw32(efe->l_ad); 1781 data_pos = (uint8_t *) efe + dscr_size + l_ea; 1782 } 1783 max_l_ad = lb_size - dscr_size - l_ea; 1784 1785 /* wipe fe/efe */ 1786 memset(data_pos, 0, max_l_ad); 1787 crclen = dscr_size - UDF_DESC_TAG_LENGTH + l_ea; 1788 if (fe) { 1789 fe->l_ad = udf_rw32(0); 1790 fe->logblks_rec = udf_rw64(0); 1791 fe->tag.desc_crc_len = udf_rw16(crclen); 1792 } else { 1793 efe->l_ad = udf_rw32(0); 1794 efe->logblks_rec = udf_rw64(0); 1795 efe->tag.desc_crc_len = udf_rw16(crclen); 1796 } 1797 1798 /* wipe all allocation extent entries */ 1799 for (extnr = 0; extnr < udf_node->num_extensions; extnr++) { 1800 ext = udf_node->ext[extnr]; 1801 dscr_size = sizeof(struct alloc_ext_entry) -1; 1802 data_pos = (uint8_t *) ext->data; 1803 max_l_ad = lb_size - dscr_size; 1804 memset(data_pos, 0, max_l_ad); 1805 ext->l_ad = udf_rw32(0); 1806 1807 crclen = dscr_size - UDF_DESC_TAG_LENGTH; 1808 ext->tag.desc_crc_len = udf_rw16(crclen); 1809 } 1810 udf_node->i_flags |= IN_NODE_REBUILD; 1811 } 1812 1813 /* --------------------------------------------------------------------- */ 1814 1815 void 1816 udf_get_adslot(struct udf_node *udf_node, int slot, struct long_ad *icb, 1817 int *eof) { 1818 struct file_entry *fe; 1819 struct extfile_entry *efe; 1820 struct alloc_ext_entry *ext; 1821 struct icb_tag *icbtag; 1822 struct short_ad *short_ad; 1823 struct long_ad *long_ad, l_icb; 1824 uint32_t offset; 1825 uint32_t lb_size, dscr_size, l_ea, l_ad, flags; 1826 uint8_t *data_pos; 1827 int icbflags, addr_type, adlen, extnr; 1828 1829 /* determine what descriptor we are in */ 1830 lb_size = udf_rw32(udf_node->ump->logical_vol->lb_size); 1831 1832 fe = udf_node->fe; 1833 efe = udf_node->efe; 1834 if (fe) { 1835 icbtag = &fe->icbtag; 1836 dscr_size = sizeof(struct file_entry) -1; 1837 l_ea = udf_rw32(fe->l_ea); 1838 l_ad = udf_rw32(fe->l_ad); 1839 data_pos = (uint8_t *) fe + dscr_size + l_ea; 1840 } else { 1841 icbtag = &efe->icbtag; 1842 dscr_size = sizeof(struct extfile_entry) -1; 1843 l_ea = udf_rw32(efe->l_ea); 1844 l_ad = udf_rw32(efe->l_ad); 1845 data_pos = (uint8_t *) efe + dscr_size + l_ea; 1846 } 1847 1848 icbflags = udf_rw16(icbtag->flags); 1849 addr_type = icbflags & UDF_ICB_TAG_FLAGS_ALLOC_MASK; 1850 1851 /* just in case we're called on an intern, its EOF */ 1852 if (addr_type == UDF_ICB_INTERN_ALLOC) { 1853 memset(icb, 0, sizeof(struct long_ad)); 1854 *eof = 1; 1855 return; 1856 } 1857 1858 adlen = 0; 1859 if (addr_type == UDF_ICB_SHORT_ALLOC) { 1860 adlen = sizeof(struct short_ad); 1861 } else if (addr_type == UDF_ICB_LONG_ALLOC) { 1862 adlen = sizeof(struct long_ad); 1863 } 1864 1865 /* if offset too big, we go to the allocation extensions */ 1866 offset = slot * adlen; 1867 extnr = -1; 1868 while (offset >= l_ad) { 1869 /* check if our last entry is a redirect */ 1870 if (addr_type == UDF_ICB_SHORT_ALLOC) { 1871 short_ad = (struct short_ad *) (data_pos + l_ad-adlen); 1872 l_icb.len = short_ad->len; 1873 l_icb.loc.part_num = udf_node->loc.loc.part_num; 1874 l_icb.loc.lb_num = short_ad->lb_num; 1875 } else { 1876 KASSERT(addr_type == UDF_ICB_LONG_ALLOC); 1877 long_ad = (struct long_ad *) (data_pos + l_ad-adlen); 1878 l_icb = *long_ad; 1879 } 1880 flags = UDF_EXT_FLAGS(udf_rw32(l_icb.len)); 1881 if (flags != UDF_EXT_REDIRECT) { 1882 l_ad = 0; /* force EOF */ 1883 break; 1884 } 1885 1886 /* advance to next extent */ 1887 extnr++; 1888 if (extnr >= udf_node->num_extensions) { 1889 l_ad = 0; /* force EOF */ 1890 break; 1891 } 1892 offset = offset - l_ad; 1893 ext = udf_node->ext[extnr]; 1894 dscr_size = sizeof(struct alloc_ext_entry) -1; 1895 l_ad = udf_rw32(ext->l_ad); 1896 data_pos = (uint8_t *) ext + dscr_size; 1897 } 1898 1899 /* XXX l_ad == 0 should be enough to check */ 1900 *eof = (offset >= l_ad) || (l_ad == 0); 1901 if (*eof) { 1902 DPRINTF(PARANOIDADWLK, ("returning EOF, extnr %d, offset %d, " 1903 "l_ad %d\n", extnr, offset, l_ad)); 1904 memset(icb, 0, sizeof(struct long_ad)); 1905 return; 1906 } 1907 1908 /* get the element */ 1909 if (addr_type == UDF_ICB_SHORT_ALLOC) { 1910 short_ad = (struct short_ad *) (data_pos + offset); 1911 icb->len = short_ad->len; 1912 icb->loc.part_num = udf_node->loc.loc.part_num; 1913 icb->loc.lb_num = short_ad->lb_num; 1914 } else if (addr_type == UDF_ICB_LONG_ALLOC) { 1915 long_ad = (struct long_ad *) (data_pos + offset); 1916 *icb = *long_ad; 1917 } 1918 DPRINTF(PARANOIDADWLK, ("returning element : v %d, lb %d, len %d, " 1919 "flags %d\n", icb->loc.part_num, icb->loc.lb_num, 1920 UDF_EXT_LEN(icb->len), UDF_EXT_FLAGS(icb->len))); 1921 } 1922 1923 /* --------------------------------------------------------------------- */ 1924 1925 int 1926 udf_append_adslot(struct udf_node *udf_node, int *slot, struct long_ad *icb) { 1927 struct udf_mount *ump = udf_node->ump; 1928 union dscrptr *dscr, *extdscr; 1929 struct file_entry *fe; 1930 struct extfile_entry *efe; 1931 struct alloc_ext_entry *ext; 1932 struct icb_tag *icbtag; 1933 struct short_ad *short_ad; 1934 struct long_ad *long_ad, o_icb, l_icb; 1935 uint64_t logblks_rec, *logblks_rec_p; 1936 uint64_t lmapping; 1937 uint32_t offset, rest, len, lb_num; 1938 uint32_t lb_size, dscr_size, l_ea, l_ad, *l_ad_p, max_l_ad, crclen; 1939 uint32_t flags; 1940 uint16_t vpart_num; 1941 uint8_t *data_pos; 1942 int icbflags, addr_type, adlen, extnr; 1943 int error; 1944 1945 lb_size = udf_rw32(ump->logical_vol->lb_size); 1946 vpart_num = udf_rw16(udf_node->loc.loc.part_num); 1947 1948 /* determine what descriptor we are in */ 1949 fe = udf_node->fe; 1950 efe = udf_node->efe; 1951 if (fe) { 1952 icbtag = &fe->icbtag; 1953 dscr = (union dscrptr *) fe; 1954 dscr_size = sizeof(struct file_entry) -1; 1955 1956 l_ea = udf_rw32(fe->l_ea); 1957 l_ad_p = &fe->l_ad; 1958 logblks_rec_p = &fe->logblks_rec; 1959 } else { 1960 icbtag = &efe->icbtag; 1961 dscr = (union dscrptr *) efe; 1962 dscr_size = sizeof(struct extfile_entry) -1; 1963 1964 l_ea = udf_rw32(efe->l_ea); 1965 l_ad_p = &efe->l_ad; 1966 logblks_rec_p = &efe->logblks_rec; 1967 } 1968 data_pos = (uint8_t *) dscr + dscr_size + l_ea; 1969 max_l_ad = lb_size - dscr_size - l_ea; 1970 1971 icbflags = udf_rw16(icbtag->flags); 1972 addr_type = icbflags & UDF_ICB_TAG_FLAGS_ALLOC_MASK; 1973 1974 /* just in case we're called on an intern, its EOF */ 1975 if (addr_type == UDF_ICB_INTERN_ALLOC) { 1976 panic("udf_append_adslot on UDF_ICB_INTERN_ALLOC\n"); 1977 } 1978 1979 adlen = 0; 1980 if (addr_type == UDF_ICB_SHORT_ALLOC) { 1981 adlen = sizeof(struct short_ad); 1982 } else if (addr_type == UDF_ICB_LONG_ALLOC) { 1983 adlen = sizeof(struct long_ad); 1984 } 1985 1986 /* clean up given long_ad since it can be a synthesized one */ 1987 flags = UDF_EXT_FLAGS(udf_rw32(icb->len)); 1988 if (flags == UDF_EXT_FREE) { 1989 icb->loc.part_num = udf_rw16(0); 1990 icb->loc.lb_num = udf_rw32(0); 1991 } 1992 1993 /* if offset too big, we go to the allocation extensions */ 1994 l_ad = udf_rw32(*l_ad_p); 1995 offset = (*slot) * adlen; 1996 extnr = -1; 1997 while (offset >= l_ad) { 1998 /* check if our last entry is a redirect */ 1999 if (addr_type == UDF_ICB_SHORT_ALLOC) { 2000 short_ad = (struct short_ad *) (data_pos + l_ad-adlen); 2001 l_icb.len = short_ad->len; 2002 l_icb.loc.part_num = udf_node->loc.loc.part_num; 2003 l_icb.loc.lb_num = short_ad->lb_num; 2004 } else { 2005 KASSERT(addr_type == UDF_ICB_LONG_ALLOC); 2006 long_ad = (struct long_ad *) (data_pos + l_ad-adlen); 2007 l_icb = *long_ad; 2008 } 2009 flags = UDF_EXT_FLAGS(udf_rw32(l_icb.len)); 2010 if (flags != UDF_EXT_REDIRECT) { 2011 /* only one past the last one is adressable */ 2012 break; 2013 } 2014 2015 /* advance to next extent */ 2016 extnr++; 2017 KASSERT(extnr < udf_node->num_extensions); 2018 offset = offset - l_ad; 2019 2020 ext = udf_node->ext[extnr]; 2021 dscr = (union dscrptr *) ext; 2022 dscr_size = sizeof(struct alloc_ext_entry) -1; 2023 max_l_ad = lb_size - dscr_size; 2024 l_ad_p = &ext->l_ad; 2025 l_ad = udf_rw32(*l_ad_p); 2026 data_pos = (uint8_t *) ext + dscr_size; 2027 } 2028 DPRINTF(PARANOIDADWLK, ("append, ext %d, offset %d, l_ad %d\n", 2029 extnr, offset, udf_rw32(*l_ad_p))); 2030 KASSERT(l_ad == udf_rw32(*l_ad_p)); 2031 2032 /* offset is offset within the current (E)FE/AED */ 2033 l_ad = udf_rw32(*l_ad_p); 2034 crclen = udf_rw16(dscr->tag.desc_crc_len); 2035 logblks_rec = udf_rw64(*logblks_rec_p); 2036 2037 /* overwriting old piece? */ 2038 if (offset < l_ad) { 2039 /* overwrite entry; compensate for the old element */ 2040 if (addr_type == UDF_ICB_SHORT_ALLOC) { 2041 short_ad = (struct short_ad *) (data_pos + offset); 2042 o_icb.len = short_ad->len; 2043 o_icb.loc.part_num = udf_rw16(0); /* ignore */ 2044 o_icb.loc.lb_num = short_ad->lb_num; 2045 } else if (addr_type == UDF_ICB_LONG_ALLOC) { 2046 long_ad = (struct long_ad *) (data_pos + offset); 2047 o_icb = *long_ad; 2048 } else { 2049 panic("Invalid address type in udf_append_adslot\n"); 2050 } 2051 2052 len = udf_rw32(o_icb.len); 2053 if (UDF_EXT_FLAGS(len) == UDF_EXT_ALLOCATED) { 2054 /* adjust counts */ 2055 len = UDF_EXT_LEN(len); 2056 logblks_rec -= (len + lb_size -1) / lb_size; 2057 } 2058 } 2059 2060 /* check if we're not appending a redirection */ 2061 flags = UDF_EXT_FLAGS(udf_rw32(icb->len)); 2062 KASSERT(flags != UDF_EXT_REDIRECT); 2063 2064 /* round down available space */ 2065 rest = adlen * ((max_l_ad - offset) / adlen); 2066 if (rest <= adlen) { 2067 /* have to append aed, see if we already have a spare one */ 2068 extnr++; 2069 ext = udf_node->ext[extnr]; 2070 l_icb = udf_node->ext_loc[extnr]; 2071 if (ext == NULL) { 2072 DPRINTF(ALLOC,("adding allocation extent %d\n", extnr)); 2073 2074 error = udf_reserve_space(ump, NULL, UDF_C_NODE, 2075 vpart_num, 1, /* can fail */ false); 2076 if (error) { 2077 printf("UDF: couldn't reserve space for AED!\n"); 2078 return error; 2079 } 2080 error = udf_allocate_space(ump, NULL, UDF_C_NODE, 2081 vpart_num, 1, &lmapping); 2082 lb_num = lmapping; 2083 if (error) 2084 panic("UDF: couldn't allocate AED!\n"); 2085 2086 /* initialise pointer to location */ 2087 memset(&l_icb, 0, sizeof(struct long_ad)); 2088 l_icb.len = udf_rw32(lb_size | UDF_EXT_REDIRECT); 2089 l_icb.loc.lb_num = udf_rw32(lb_num); 2090 l_icb.loc.part_num = udf_rw16(vpart_num); 2091 2092 /* create new aed descriptor */ 2093 udf_create_logvol_dscr(ump, udf_node, &l_icb, &extdscr); 2094 ext = &extdscr->aee; 2095 2096 udf_inittag(ump, &ext->tag, TAGID_ALLOCEXTENT, lb_num); 2097 dscr_size = sizeof(struct alloc_ext_entry) -1; 2098 max_l_ad = lb_size - dscr_size; 2099 memset(ext->data, 0, max_l_ad); 2100 ext->l_ad = udf_rw32(0); 2101 ext->tag.desc_crc_len = 2102 udf_rw16(dscr_size - UDF_DESC_TAG_LENGTH); 2103 2104 /* declare aed */ 2105 udf_node->num_extensions++; 2106 udf_node->ext_loc[extnr] = l_icb; 2107 udf_node->ext[extnr] = ext; 2108 } 2109 /* add redirect and adjust l_ad and crclen for old descr */ 2110 if (addr_type == UDF_ICB_SHORT_ALLOC) { 2111 short_ad = (struct short_ad *) (data_pos + offset); 2112 short_ad->len = l_icb.len; 2113 short_ad->lb_num = l_icb.loc.lb_num; 2114 } else if (addr_type == UDF_ICB_LONG_ALLOC) { 2115 long_ad = (struct long_ad *) (data_pos + offset); 2116 *long_ad = l_icb; 2117 } 2118 l_ad += adlen; 2119 crclen += adlen; 2120 dscr->tag.desc_crc_len = udf_rw16(crclen); 2121 *l_ad_p = udf_rw32(l_ad); 2122 2123 /* advance to the new extension */ 2124 KASSERT(ext != NULL); 2125 dscr = (union dscrptr *) ext; 2126 dscr_size = sizeof(struct alloc_ext_entry) -1; 2127 max_l_ad = lb_size - dscr_size; 2128 data_pos = (uint8_t *) dscr + dscr_size; 2129 2130 l_ad_p = &ext->l_ad; 2131 l_ad = udf_rw32(*l_ad_p); 2132 crclen = udf_rw16(dscr->tag.desc_crc_len); 2133 offset = 0; 2134 2135 /* adjust callees slot count for link insert */ 2136 *slot += 1; 2137 } 2138 2139 /* write out the element */ 2140 DPRINTF(PARANOIDADWLK, ("adding element : %p : v %d, lb %d, " 2141 "len %d, flags %d\n", data_pos + offset, 2142 icb->loc.part_num, icb->loc.lb_num, 2143 UDF_EXT_LEN(icb->len), UDF_EXT_FLAGS(icb->len))); 2144 if (addr_type == UDF_ICB_SHORT_ALLOC) { 2145 short_ad = (struct short_ad *) (data_pos + offset); 2146 short_ad->len = icb->len; 2147 short_ad->lb_num = icb->loc.lb_num; 2148 } else if (addr_type == UDF_ICB_LONG_ALLOC) { 2149 long_ad = (struct long_ad *) (data_pos + offset); 2150 *long_ad = *icb; 2151 } 2152 2153 /* adjust logblks recorded count */ 2154 len = udf_rw32(icb->len); 2155 flags = UDF_EXT_FLAGS(len); 2156 if (flags == UDF_EXT_ALLOCATED) 2157 logblks_rec += (UDF_EXT_LEN(len) + lb_size -1) / lb_size; 2158 *logblks_rec_p = udf_rw64(logblks_rec); 2159 2160 /* adjust l_ad and crclen when needed */ 2161 if (offset >= l_ad) { 2162 l_ad += adlen; 2163 crclen += adlen; 2164 dscr->tag.desc_crc_len = udf_rw16(crclen); 2165 *l_ad_p = udf_rw32(l_ad); 2166 } 2167 2168 return 0; 2169 } 2170 2171 /* --------------------------------------------------------------------- */ 2172 2173 static void 2174 udf_count_alloc_exts(struct udf_node *udf_node) 2175 { 2176 struct long_ad s_ad; 2177 uint32_t lb_num, len, flags; 2178 uint16_t vpart_num; 2179 int slot, eof; 2180 int num_extents, extnr; 2181 int lb_size; 2182 2183 if (udf_node->num_extensions == 0) 2184 return; 2185 2186 lb_size = udf_rw32(udf_node->ump->logical_vol->lb_size); 2187 /* count number of allocation extents in use */ 2188 num_extents = 0; 2189 slot = 0; 2190 for (;;) { 2191 udf_get_adslot(udf_node, slot, &s_ad, &eof); 2192 if (eof) 2193 break; 2194 len = udf_rw32(s_ad.len); 2195 flags = UDF_EXT_FLAGS(len); 2196 2197 if (flags == UDF_EXT_REDIRECT) 2198 num_extents++; 2199 2200 slot++; 2201 } 2202 2203 DPRINTF(ALLOC, ("udf_count_alloc_ext counted %d live extents\n", 2204 num_extents)); 2205 2206 /* XXX choice: we could delay freeing them on node writeout */ 2207 /* free excess entries */ 2208 extnr = num_extents; 2209 for (;extnr < udf_node->num_extensions; extnr++) { 2210 DPRINTF(ALLOC, ("freeing alloc ext %d\n", extnr)); 2211 /* free dscriptor */ 2212 s_ad = udf_node->ext_loc[extnr]; 2213 udf_free_logvol_dscr(udf_node->ump, &s_ad, 2214 udf_node->ext[extnr]); 2215 udf_node->ext[extnr] = NULL; 2216 2217 /* free disc space */ 2218 lb_num = udf_rw32(s_ad.loc.lb_num); 2219 vpart_num = udf_rw16(s_ad.loc.part_num); 2220 udf_free_allocated_space(udf_node->ump, lb_num, vpart_num, 1); 2221 2222 memset(&udf_node->ext_loc[extnr], 0, sizeof(struct long_ad)); 2223 } 2224 2225 /* set our new number of allocation extents */ 2226 udf_node->num_extensions = num_extents; 2227 } 2228 2229 2230 /* --------------------------------------------------------------------- */ 2231 2232 /* 2233 * Adjust the node's allocation descriptors to reflect the new mapping; do 2234 * take note that we might glue to existing allocation descriptors. 2235 * 2236 * XXX Note there can only be one allocation being recorded/mount; maybe 2237 * explicit allocation in shedule thread? 2238 */ 2239 2240 static void 2241 udf_record_allocation_in_node(struct udf_mount *ump, struct buf *buf, 2242 uint16_t vpart_num, uint64_t *mapping, struct long_ad *node_ad_cpy) 2243 { 2244 struct vnode *vp = buf->b_vp; 2245 struct udf_node *udf_node = VTOI(vp); 2246 struct file_entry *fe; 2247 struct extfile_entry *efe; 2248 struct icb_tag *icbtag; 2249 struct long_ad s_ad, c_ad; 2250 uint64_t inflen, from, till; 2251 uint64_t foffset, end_foffset, restart_foffset; 2252 uint64_t orig_inflen, orig_lbrec, new_inflen, new_lbrec; 2253 uint32_t num_lb, len, flags, lb_num; 2254 uint32_t run_start; 2255 uint32_t slot_offset, replace_len, replace; 2256 int addr_type, icbflags; 2257 // int udf_c_type = buf->b_udf_c_type; 2258 int lb_size, run_length, eof; 2259 int slot, cpy_slot, cpy_slots, restart_slot; 2260 int error; 2261 2262 DPRINTF(ALLOC, ("udf_record_allocation_in_node\n")); 2263 2264 #if 0 2265 /* XXX disable sanity check for now */ 2266 /* sanity check ... should be panic ? */ 2267 if ((udf_c_type != UDF_C_USERDATA) && (udf_c_type != UDF_C_FIDS)) 2268 return; 2269 #endif 2270 2271 lb_size = udf_rw32(udf_node->ump->logical_vol->lb_size); 2272 2273 /* do the job */ 2274 UDF_LOCK_NODE(udf_node, 0); /* XXX can deadlock ? */ 2275 udf_node_sanity_check(udf_node, &orig_inflen, &orig_lbrec); 2276 2277 fe = udf_node->fe; 2278 efe = udf_node->efe; 2279 if (fe) { 2280 icbtag = &fe->icbtag; 2281 inflen = udf_rw64(fe->inf_len); 2282 } else { 2283 icbtag = &efe->icbtag; 2284 inflen = udf_rw64(efe->inf_len); 2285 } 2286 2287 /* do check if `till' is not past file information length */ 2288 from = buf->b_lblkno * lb_size; 2289 till = MIN(inflen, from + buf->b_resid); 2290 2291 num_lb = (till - from + lb_size -1) / lb_size; 2292 2293 DPRINTF(ALLOC, ("record allocation from %"PRIu64" + %d\n", from, buf->b_bcount)); 2294 2295 icbflags = udf_rw16(icbtag->flags); 2296 addr_type = icbflags & UDF_ICB_TAG_FLAGS_ALLOC_MASK; 2297 2298 if (addr_type == UDF_ICB_INTERN_ALLOC) { 2299 /* nothing to do */ 2300 /* XXX clean up rest of node? just in case? */ 2301 UDF_UNLOCK_NODE(udf_node, 0); 2302 return; 2303 } 2304 2305 slot = 0; 2306 cpy_slot = 0; 2307 foffset = 0; 2308 2309 /* 1) copy till first overlap piece to the rewrite buffer */ 2310 for (;;) { 2311 udf_get_adslot(udf_node, slot, &s_ad, &eof); 2312 if (eof) { 2313 DPRINTF(WRITE, 2314 ("Record allocation in node " 2315 "failed: encountered EOF\n")); 2316 UDF_UNLOCK_NODE(udf_node, 0); 2317 buf->b_error = EINVAL; 2318 return; 2319 } 2320 len = udf_rw32(s_ad.len); 2321 flags = UDF_EXT_FLAGS(len); 2322 len = UDF_EXT_LEN(len); 2323 2324 if (flags == UDF_EXT_REDIRECT) { 2325 slot++; 2326 continue; 2327 } 2328 2329 end_foffset = foffset + len; 2330 if (end_foffset > from) 2331 break; /* found */ 2332 2333 node_ad_cpy[cpy_slot++] = s_ad; 2334 2335 DPRINTF(ALLOC, ("\t1: vp %d, lb %d, len %d, flags %d " 2336 "-> stack\n", 2337 udf_rw16(s_ad.loc.part_num), 2338 udf_rw32(s_ad.loc.lb_num), 2339 UDF_EXT_LEN(udf_rw32(s_ad.len)), 2340 UDF_EXT_FLAGS(udf_rw32(s_ad.len)) >> 30)); 2341 2342 foffset = end_foffset; 2343 slot++; 2344 } 2345 restart_slot = slot; 2346 restart_foffset = foffset; 2347 2348 /* 2) trunc overlapping slot at overlap and copy it */ 2349 slot_offset = from - foffset; 2350 if (slot_offset > 0) { 2351 DPRINTF(ALLOC, ("\tslot_offset = %d, flags = %d (%d)\n", 2352 slot_offset, flags >> 30, flags)); 2353 2354 s_ad.len = udf_rw32(slot_offset | flags); 2355 node_ad_cpy[cpy_slot++] = s_ad; 2356 2357 DPRINTF(ALLOC, ("\t2: vp %d, lb %d, len %d, flags %d " 2358 "-> stack\n", 2359 udf_rw16(s_ad.loc.part_num), 2360 udf_rw32(s_ad.loc.lb_num), 2361 UDF_EXT_LEN(udf_rw32(s_ad.len)), 2362 UDF_EXT_FLAGS(udf_rw32(s_ad.len)) >> 30)); 2363 } 2364 foffset += slot_offset; 2365 2366 /* 3) insert new mappings */ 2367 memset(&s_ad, 0, sizeof(struct long_ad)); 2368 lb_num = 0; 2369 for (lb_num = 0; lb_num < num_lb; lb_num++) { 2370 run_start = mapping[lb_num]; 2371 run_length = 1; 2372 while (lb_num < num_lb-1) { 2373 if (mapping[lb_num+1] != mapping[lb_num]+1) 2374 if (mapping[lb_num+1] != mapping[lb_num]) 2375 break; 2376 run_length++; 2377 lb_num++; 2378 } 2379 /* insert slot for this mapping */ 2380 len = run_length * lb_size; 2381 2382 /* bounds checking */ 2383 if (foffset + len > till) 2384 len = till - foffset; 2385 KASSERT(foffset + len <= inflen); 2386 2387 s_ad.len = udf_rw32(len | UDF_EXT_ALLOCATED); 2388 s_ad.loc.part_num = udf_rw16(vpart_num); 2389 s_ad.loc.lb_num = udf_rw32(run_start); 2390 2391 foffset += len; 2392 2393 /* paranoia */ 2394 if (len == 0) { 2395 DPRINTF(WRITE, 2396 ("Record allocation in node " 2397 "failed: insert failed\n")); 2398 UDF_UNLOCK_NODE(udf_node, 0); 2399 buf->b_error = EINVAL; 2400 return; 2401 } 2402 node_ad_cpy[cpy_slot++] = s_ad; 2403 2404 DPRINTF(ALLOC, ("\t3: insert new mapping vp %d lb %d, len %d, " 2405 "flags %d -> stack\n", 2406 udf_rw16(s_ad.loc.part_num), udf_rw32(s_ad.loc.lb_num), 2407 UDF_EXT_LEN(udf_rw32(s_ad.len)), 2408 UDF_EXT_FLAGS(udf_rw32(s_ad.len)) >> 30)); 2409 } 2410 2411 /* 4) pop replaced length */ 2412 slot = restart_slot; 2413 foffset = restart_foffset; 2414 2415 replace_len = till - foffset; /* total amount of bytes to pop */ 2416 slot_offset = from - foffset; /* offset in first encounted slot */ 2417 KASSERT((slot_offset % lb_size) == 0); 2418 2419 for (;;) { 2420 udf_get_adslot(udf_node, slot, &s_ad, &eof); 2421 if (eof) 2422 break; 2423 2424 len = udf_rw32(s_ad.len); 2425 flags = UDF_EXT_FLAGS(len); 2426 len = UDF_EXT_LEN(len); 2427 lb_num = udf_rw32(s_ad.loc.lb_num); 2428 2429 if (flags == UDF_EXT_REDIRECT) { 2430 slot++; 2431 continue; 2432 } 2433 2434 DPRINTF(ALLOC, ("\t4i: got slot %d, slot_offset %d, " 2435 "replace_len %d, " 2436 "vp %d, lb %d, len %d, flags %d\n", 2437 slot, slot_offset, replace_len, 2438 udf_rw16(s_ad.loc.part_num), 2439 udf_rw32(s_ad.loc.lb_num), 2440 UDF_EXT_LEN(udf_rw32(s_ad.len)), 2441 UDF_EXT_FLAGS(udf_rw32(s_ad.len)) >> 30)); 2442 2443 /* adjust for slot offset */ 2444 if (slot_offset) { 2445 DPRINTF(ALLOC, ("\t4s: skipping %d\n", slot_offset)); 2446 lb_num += slot_offset / lb_size; 2447 len -= slot_offset; 2448 foffset += slot_offset; 2449 replace_len -= slot_offset; 2450 2451 /* mark adjusted */ 2452 slot_offset = 0; 2453 } 2454 2455 /* advance for (the rest of) this slot */ 2456 replace = MIN(len, replace_len); 2457 DPRINTF(ALLOC, ("\t4d: replacing %d\n", replace)); 2458 2459 /* advance for this slot */ 2460 if (replace) { 2461 /* note: dont round DOWN on num_lb since we then 2462 * forget the last partial one */ 2463 num_lb = (replace + lb_size - 1) / lb_size; 2464 if (flags != UDF_EXT_FREE) { 2465 udf_free_allocated_space(ump, lb_num, 2466 udf_rw16(s_ad.loc.part_num), num_lb); 2467 } 2468 lb_num += num_lb; 2469 len -= replace; 2470 foffset += replace; 2471 replace_len -= replace; 2472 } 2473 2474 /* do we have a slot tail ? */ 2475 if (len) { 2476 KASSERT(foffset % lb_size == 0); 2477 2478 /* we arrived at our point, push remainder */ 2479 s_ad.len = udf_rw32(len | flags); 2480 s_ad.loc.lb_num = udf_rw32(lb_num); 2481 if (flags == UDF_EXT_FREE) 2482 s_ad.loc.lb_num = udf_rw32(0); 2483 node_ad_cpy[cpy_slot++] = s_ad; 2484 foffset += len; 2485 slot++; 2486 2487 DPRINTF(ALLOC, ("\t4: vp %d, lb %d, len %d, flags %d " 2488 "-> stack\n", 2489 udf_rw16(s_ad.loc.part_num), 2490 udf_rw32(s_ad.loc.lb_num), 2491 UDF_EXT_LEN(udf_rw32(s_ad.len)), 2492 UDF_EXT_FLAGS(udf_rw32(s_ad.len)) >> 30)); 2493 break; 2494 } 2495 2496 slot++; 2497 } 2498 2499 /* 5) copy remainder */ 2500 for (;;) { 2501 udf_get_adslot(udf_node, slot, &s_ad, &eof); 2502 if (eof) 2503 break; 2504 2505 len = udf_rw32(s_ad.len); 2506 flags = UDF_EXT_FLAGS(len); 2507 len = UDF_EXT_LEN(len); 2508 2509 if (flags == UDF_EXT_REDIRECT) { 2510 slot++; 2511 continue; 2512 } 2513 2514 node_ad_cpy[cpy_slot++] = s_ad; 2515 2516 DPRINTF(ALLOC, ("\t5: insert new mapping " 2517 "vp %d lb %d, len %d, flags %d " 2518 "-> stack\n", 2519 udf_rw16(s_ad.loc.part_num), 2520 udf_rw32(s_ad.loc.lb_num), 2521 UDF_EXT_LEN(udf_rw32(s_ad.len)), 2522 UDF_EXT_FLAGS(udf_rw32(s_ad.len)) >> 30)); 2523 2524 slot++; 2525 } 2526 2527 /* 6) reset node descriptors */ 2528 udf_wipe_adslots(udf_node); 2529 2530 /* 7) copy back extents; merge when possible. Recounting on the fly */ 2531 cpy_slots = cpy_slot; 2532 2533 c_ad = node_ad_cpy[0]; 2534 slot = 0; 2535 DPRINTF(ALLOC, ("\t7s: stack -> got mapping vp %d " 2536 "lb %d, len %d, flags %d\n", 2537 udf_rw16(c_ad.loc.part_num), 2538 udf_rw32(c_ad.loc.lb_num), 2539 UDF_EXT_LEN(udf_rw32(c_ad.len)), 2540 UDF_EXT_FLAGS(udf_rw32(c_ad.len)) >> 30)); 2541 2542 for (cpy_slot = 1; cpy_slot < cpy_slots; cpy_slot++) { 2543 s_ad = node_ad_cpy[cpy_slot]; 2544 2545 DPRINTF(ALLOC, ("\t7i: stack -> got mapping vp %d " 2546 "lb %d, len %d, flags %d\n", 2547 udf_rw16(s_ad.loc.part_num), 2548 udf_rw32(s_ad.loc.lb_num), 2549 UDF_EXT_LEN(udf_rw32(s_ad.len)), 2550 UDF_EXT_FLAGS(udf_rw32(s_ad.len)) >> 30)); 2551 2552 /* see if we can merge */ 2553 if (udf_ads_merge(lb_size, &c_ad, &s_ad)) { 2554 /* not mergable (anymore) */ 2555 DPRINTF(ALLOC, ("\t7: appending vp %d lb %d, " 2556 "len %d, flags %d\n", 2557 udf_rw16(c_ad.loc.part_num), 2558 udf_rw32(c_ad.loc.lb_num), 2559 UDF_EXT_LEN(udf_rw32(c_ad.len)), 2560 UDF_EXT_FLAGS(udf_rw32(c_ad.len)) >> 30)); 2561 2562 error = udf_append_adslot(udf_node, &slot, &c_ad); 2563 if (error) { 2564 buf->b_error = error; 2565 goto out; 2566 } 2567 c_ad = s_ad; 2568 slot++; 2569 } 2570 } 2571 2572 /* 8) push rest slot (if any) */ 2573 if (UDF_EXT_LEN(c_ad.len) > 0) { 2574 DPRINTF(ALLOC, ("\t8: last append vp %d lb %d, " 2575 "len %d, flags %d\n", 2576 udf_rw16(c_ad.loc.part_num), 2577 udf_rw32(c_ad.loc.lb_num), 2578 UDF_EXT_LEN(udf_rw32(c_ad.len)), 2579 UDF_EXT_FLAGS(udf_rw32(c_ad.len)) >> 30)); 2580 2581 error = udf_append_adslot(udf_node, &slot, &c_ad); 2582 if (error) { 2583 buf->b_error = error; 2584 goto out; 2585 } 2586 } 2587 2588 out: 2589 udf_count_alloc_exts(udf_node); 2590 2591 /* the node's descriptors should now be sane */ 2592 udf_node_sanity_check(udf_node, &new_inflen, &new_lbrec); 2593 UDF_UNLOCK_NODE(udf_node, 0); 2594 2595 KASSERT(orig_inflen == new_inflen); 2596 KASSERT(new_lbrec >= orig_lbrec); 2597 2598 return; 2599 } 2600 2601 /* --------------------------------------------------------------------- */ 2602 2603 int 2604 udf_grow_node(struct udf_node *udf_node, uint64_t new_size) 2605 { 2606 union dscrptr *dscr; 2607 struct vnode *vp = udf_node->vnode; 2608 struct udf_mount *ump = udf_node->ump; 2609 struct file_entry *fe; 2610 struct extfile_entry *efe; 2611 struct icb_tag *icbtag; 2612 struct long_ad c_ad, s_ad; 2613 uint64_t size_diff, old_size, inflen, objsize, chunk, append_len; 2614 uint64_t foffset, end_foffset; 2615 uint64_t orig_inflen, orig_lbrec, new_inflen, new_lbrec; 2616 uint32_t lb_size, dscr_size, crclen, lastblock_grow; 2617 uint32_t icbflags, len, flags, max_len; 2618 uint32_t max_l_ad, l_ad, l_ea; 2619 uint16_t my_part, dst_part; 2620 uint8_t *data_pos, *evacuated_data; 2621 int addr_type; 2622 int slot, cpy_slot; 2623 int eof, error; 2624 2625 DPRINTF(ALLOC, ("udf_grow_node\n")); 2626 2627 UDF_LOCK_NODE(udf_node, 0); 2628 udf_node_sanity_check(udf_node, &orig_inflen, &orig_lbrec); 2629 2630 lb_size = udf_rw32(ump->logical_vol->lb_size); 2631 max_len = ((UDF_EXT_MAXLEN / lb_size) * lb_size); 2632 2633 fe = udf_node->fe; 2634 efe = udf_node->efe; 2635 if (fe) { 2636 dscr = (union dscrptr *) fe; 2637 icbtag = &fe->icbtag; 2638 inflen = udf_rw64(fe->inf_len); 2639 objsize = inflen; 2640 dscr_size = sizeof(struct file_entry) -1; 2641 l_ea = udf_rw32(fe->l_ea); 2642 l_ad = udf_rw32(fe->l_ad); 2643 } else { 2644 dscr = (union dscrptr *) efe; 2645 icbtag = &efe->icbtag; 2646 inflen = udf_rw64(efe->inf_len); 2647 objsize = udf_rw64(efe->obj_size); 2648 dscr_size = sizeof(struct extfile_entry) -1; 2649 l_ea = udf_rw32(efe->l_ea); 2650 l_ad = udf_rw32(efe->l_ad); 2651 } 2652 data_pos = (uint8_t *) dscr + dscr_size + l_ea; 2653 max_l_ad = lb_size - dscr_size - l_ea; 2654 2655 icbflags = udf_rw16(icbtag->flags); 2656 addr_type = icbflags & UDF_ICB_TAG_FLAGS_ALLOC_MASK; 2657 2658 old_size = inflen; 2659 size_diff = new_size - old_size; 2660 2661 DPRINTF(ALLOC, ("\tfrom %"PRIu64" to %"PRIu64"\n", old_size, new_size)); 2662 2663 evacuated_data = NULL; 2664 if (addr_type == UDF_ICB_INTERN_ALLOC) { 2665 if (l_ad + size_diff <= max_l_ad) { 2666 /* only reflect size change directly in the node */ 2667 inflen += size_diff; 2668 objsize += size_diff; 2669 l_ad += size_diff; 2670 crclen = dscr_size - UDF_DESC_TAG_LENGTH + l_ea + l_ad; 2671 if (fe) { 2672 fe->inf_len = udf_rw64(inflen); 2673 fe->l_ad = udf_rw32(l_ad); 2674 fe->tag.desc_crc_len = udf_rw16(crclen); 2675 } else { 2676 efe->inf_len = udf_rw64(inflen); 2677 efe->obj_size = udf_rw64(objsize); 2678 efe->l_ad = udf_rw32(l_ad); 2679 efe->tag.desc_crc_len = udf_rw16(crclen); 2680 } 2681 error = 0; 2682 2683 /* set new size for uvm */ 2684 uvm_vnp_setsize(vp, old_size); 2685 uvm_vnp_setwritesize(vp, new_size); 2686 2687 #if 0 2688 /* zero append space in buffer */ 2689 uvm_vnp_zerorange(vp, old_size, new_size - old_size); 2690 #endif 2691 2692 udf_node_sanity_check(udf_node, &new_inflen, &new_lbrec); 2693 2694 /* unlock */ 2695 UDF_UNLOCK_NODE(udf_node, 0); 2696 2697 KASSERT(new_inflen == orig_inflen + size_diff); 2698 KASSERT(new_lbrec == orig_lbrec); 2699 KASSERT(new_lbrec == 0); 2700 return 0; 2701 } 2702 2703 DPRINTF(ALLOC, ("\tCONVERT from internal\n")); 2704 2705 if (old_size > 0) { 2706 /* allocate some space and copy in the stuff to keep */ 2707 evacuated_data = malloc(lb_size, M_UDFTEMP, M_WAITOK); 2708 memset(evacuated_data, 0, lb_size); 2709 2710 /* node is locked, so safe to exit mutex */ 2711 UDF_UNLOCK_NODE(udf_node, 0); 2712 2713 /* read in using the `normal' vn_rdwr() */ 2714 error = vn_rdwr(UIO_READ, udf_node->vnode, 2715 evacuated_data, old_size, 0, 2716 UIO_SYSSPACE, IO_ALTSEMANTICS | IO_NODELOCKED, 2717 FSCRED, NULL, NULL); 2718 2719 /* enter again */ 2720 UDF_LOCK_NODE(udf_node, 0); 2721 } 2722 2723 /* convert to a normal alloc and select type */ 2724 my_part = udf_rw16(udf_node->loc.loc.part_num); 2725 dst_part = udf_get_record_vpart(ump, udf_get_c_type(udf_node)); 2726 addr_type = UDF_ICB_SHORT_ALLOC; 2727 if (dst_part != my_part) 2728 addr_type = UDF_ICB_LONG_ALLOC; 2729 2730 icbflags &= ~UDF_ICB_TAG_FLAGS_ALLOC_MASK; 2731 icbflags |= addr_type; 2732 icbtag->flags = udf_rw16(icbflags); 2733 2734 /* wipe old descriptor space */ 2735 udf_wipe_adslots(udf_node); 2736 2737 memset(&c_ad, 0, sizeof(struct long_ad)); 2738 c_ad.len = udf_rw32(old_size | UDF_EXT_FREE); 2739 c_ad.loc.part_num = udf_rw16(0); /* not relevant */ 2740 c_ad.loc.lb_num = udf_rw32(0); /* not relevant */ 2741 2742 slot = 0; 2743 } else { 2744 /* goto the last entry (if any) */ 2745 slot = 0; 2746 cpy_slot = 0; 2747 foffset = 0; 2748 memset(&c_ad, 0, sizeof(struct long_ad)); 2749 for (;;) { 2750 udf_get_adslot(udf_node, slot, &c_ad, &eof); 2751 if (eof) 2752 break; 2753 2754 len = udf_rw32(c_ad.len); 2755 flags = UDF_EXT_FLAGS(len); 2756 len = UDF_EXT_LEN(len); 2757 2758 end_foffset = foffset + len; 2759 if (flags != UDF_EXT_REDIRECT) 2760 foffset = end_foffset; 2761 2762 slot++; 2763 } 2764 /* at end of adslots */ 2765 2766 /* special case if the old size was zero, then there is no last slot */ 2767 if (old_size == 0) { 2768 c_ad.len = udf_rw32(0 | UDF_EXT_FREE); 2769 c_ad.loc.part_num = udf_rw16(0); /* not relevant */ 2770 c_ad.loc.lb_num = udf_rw32(0); /* not relevant */ 2771 } else { 2772 /* refetch last slot */ 2773 slot--; 2774 udf_get_adslot(udf_node, slot, &c_ad, &eof); 2775 } 2776 } 2777 2778 /* 2779 * If the length of the last slot is not a multiple of lb_size, adjust 2780 * length so that it is; don't forget to adjust `append_len'! relevant for 2781 * extending existing files 2782 */ 2783 len = udf_rw32(c_ad.len); 2784 flags = UDF_EXT_FLAGS(len); 2785 len = UDF_EXT_LEN(len); 2786 2787 lastblock_grow = 0; 2788 if (len % lb_size > 0) { 2789 lastblock_grow = lb_size - (len % lb_size); 2790 lastblock_grow = MIN(size_diff, lastblock_grow); 2791 len += lastblock_grow; 2792 c_ad.len = udf_rw32(len | flags); 2793 2794 /* TODO zero appened space in buffer! */ 2795 /* using uvm_vnp_zerorange(vp, old_size, new_size - old_size); ? */ 2796 } 2797 memset(&s_ad, 0, sizeof(struct long_ad)); 2798 2799 /* size_diff can be bigger than allowed, so grow in chunks */ 2800 append_len = size_diff - lastblock_grow; 2801 while (append_len > 0) { 2802 chunk = MIN(append_len, max_len); 2803 s_ad.len = udf_rw32(chunk | UDF_EXT_FREE); 2804 s_ad.loc.part_num = udf_rw16(0); 2805 s_ad.loc.lb_num = udf_rw32(0); 2806 2807 if (udf_ads_merge(lb_size, &c_ad, &s_ad)) { 2808 /* not mergable (anymore) */ 2809 error = udf_append_adslot(udf_node, &slot, &c_ad); 2810 if (error) 2811 goto errorout; 2812 slot++; 2813 c_ad = s_ad; 2814 memset(&s_ad, 0, sizeof(struct long_ad)); 2815 } 2816 append_len -= chunk; 2817 } 2818 2819 /* if there is a rest piece in the accumulator, append it */ 2820 if (UDF_EXT_LEN(udf_rw32(c_ad.len)) > 0) { 2821 error = udf_append_adslot(udf_node, &slot, &c_ad); 2822 if (error) 2823 goto errorout; 2824 slot++; 2825 } 2826 2827 /* if there is a rest piece that didn't fit, append it */ 2828 if (UDF_EXT_LEN(udf_rw32(s_ad.len)) > 0) { 2829 error = udf_append_adslot(udf_node, &slot, &s_ad); 2830 if (error) 2831 goto errorout; 2832 slot++; 2833 } 2834 2835 inflen += size_diff; 2836 objsize += size_diff; 2837 if (fe) { 2838 fe->inf_len = udf_rw64(inflen); 2839 } else { 2840 efe->inf_len = udf_rw64(inflen); 2841 efe->obj_size = udf_rw64(objsize); 2842 } 2843 error = 0; 2844 2845 if (evacuated_data) { 2846 /* set new write size for uvm */ 2847 uvm_vnp_setwritesize(vp, old_size); 2848 2849 /* write out evacuated data */ 2850 error = vn_rdwr(UIO_WRITE, udf_node->vnode, 2851 evacuated_data, old_size, 0, 2852 UIO_SYSSPACE, IO_ALTSEMANTICS | IO_NODELOCKED, 2853 FSCRED, NULL, NULL); 2854 uvm_vnp_setsize(vp, old_size); 2855 } 2856 2857 errorout: 2858 if (evacuated_data) 2859 free(evacuated_data, M_UDFTEMP); 2860 2861 udf_count_alloc_exts(udf_node); 2862 2863 udf_node_sanity_check(udf_node, &new_inflen, &new_lbrec); 2864 UDF_UNLOCK_NODE(udf_node, 0); 2865 2866 KASSERT(new_inflen == orig_inflen + size_diff); 2867 KASSERT(new_lbrec == orig_lbrec); 2868 2869 return error; 2870 } 2871 2872 /* --------------------------------------------------------------------- */ 2873 2874 int 2875 udf_shrink_node(struct udf_node *udf_node, uint64_t new_size) 2876 { 2877 struct vnode *vp = udf_node->vnode; 2878 struct udf_mount *ump = udf_node->ump; 2879 struct file_entry *fe; 2880 struct extfile_entry *efe; 2881 struct icb_tag *icbtag; 2882 struct long_ad c_ad, s_ad, *node_ad_cpy; 2883 uint64_t size_diff, old_size, inflen, objsize; 2884 uint64_t foffset, end_foffset; 2885 uint64_t orig_inflen, orig_lbrec, new_inflen, new_lbrec; 2886 uint32_t lb_size, dscr_size, crclen; 2887 uint32_t slot_offset, slot_offset_lb; 2888 uint32_t len, flags, max_len; 2889 uint32_t num_lb, lb_num; 2890 uint32_t max_l_ad, l_ad, l_ea; 2891 uint16_t vpart_num; 2892 uint8_t *data_pos; 2893 int icbflags, addr_type; 2894 int slot, cpy_slot, cpy_slots; 2895 int eof, error; 2896 2897 DPRINTF(ALLOC, ("udf_shrink_node\n")); 2898 2899 UDF_LOCK_NODE(udf_node, 0); 2900 udf_node_sanity_check(udf_node, &orig_inflen, &orig_lbrec); 2901 2902 lb_size = udf_rw32(ump->logical_vol->lb_size); 2903 max_len = ((UDF_EXT_MAXLEN / lb_size) * lb_size); 2904 2905 /* do the work */ 2906 fe = udf_node->fe; 2907 efe = udf_node->efe; 2908 if (fe) { 2909 icbtag = &fe->icbtag; 2910 inflen = udf_rw64(fe->inf_len); 2911 objsize = inflen; 2912 dscr_size = sizeof(struct file_entry) -1; 2913 l_ea = udf_rw32(fe->l_ea); 2914 l_ad = udf_rw32(fe->l_ad); 2915 data_pos = (uint8_t *) fe + dscr_size + l_ea; 2916 } else { 2917 icbtag = &efe->icbtag; 2918 inflen = udf_rw64(efe->inf_len); 2919 objsize = udf_rw64(efe->obj_size); 2920 dscr_size = sizeof(struct extfile_entry) -1; 2921 l_ea = udf_rw32(efe->l_ea); 2922 l_ad = udf_rw32(efe->l_ad); 2923 data_pos = (uint8_t *) efe + dscr_size + l_ea; 2924 } 2925 max_l_ad = lb_size - dscr_size - l_ea; 2926 2927 icbflags = udf_rw16(icbtag->flags); 2928 addr_type = icbflags & UDF_ICB_TAG_FLAGS_ALLOC_MASK; 2929 2930 old_size = inflen; 2931 size_diff = old_size - new_size; 2932 2933 DPRINTF(ALLOC, ("\tfrom %"PRIu64" to %"PRIu64"\n", old_size, new_size)); 2934 2935 /* shrink the node to its new size */ 2936 if (addr_type == UDF_ICB_INTERN_ALLOC) { 2937 /* only reflect size change directly in the node */ 2938 KASSERT(new_size <= max_l_ad); 2939 inflen -= size_diff; 2940 objsize -= size_diff; 2941 l_ad -= size_diff; 2942 crclen = dscr_size - UDF_DESC_TAG_LENGTH + l_ea + l_ad; 2943 if (fe) { 2944 fe->inf_len = udf_rw64(inflen); 2945 fe->l_ad = udf_rw32(l_ad); 2946 fe->tag.desc_crc_len = udf_rw16(crclen); 2947 } else { 2948 efe->inf_len = udf_rw64(inflen); 2949 efe->obj_size = udf_rw64(objsize); 2950 efe->l_ad = udf_rw32(l_ad); 2951 efe->tag.desc_crc_len = udf_rw16(crclen); 2952 } 2953 error = 0; 2954 2955 /* clear the space in the descriptor */ 2956 KASSERT(old_size > new_size); 2957 memset(data_pos + new_size, 0, old_size - new_size); 2958 2959 /* TODO zero appened space in buffer! */ 2960 /* using uvm_vnp_zerorange(vp, old_size, old_size - new_size); ? */ 2961 2962 /* set new size for uvm */ 2963 uvm_vnp_setsize(vp, new_size); 2964 2965 udf_node_sanity_check(udf_node, &new_inflen, &new_lbrec); 2966 UDF_UNLOCK_NODE(udf_node, 0); 2967 2968 KASSERT(new_inflen == orig_inflen - size_diff); 2969 KASSERT(new_lbrec == orig_lbrec); 2970 KASSERT(new_lbrec == 0); 2971 2972 return 0; 2973 } 2974 2975 /* setup node cleanup extents copy space */ 2976 node_ad_cpy = malloc(lb_size * UDF_MAX_ALLOC_EXTENTS, 2977 M_UDFMNT, M_WAITOK); 2978 memset(node_ad_cpy, 0, lb_size * UDF_MAX_ALLOC_EXTENTS); 2979 2980 /* 2981 * Shrink the node by releasing the allocations and truncate the last 2982 * allocation to the new size. If the new size fits into the 2983 * allocation descriptor itself, transform it into an 2984 * UDF_ICB_INTERN_ALLOC. 2985 */ 2986 slot = 0; 2987 cpy_slot = 0; 2988 foffset = 0; 2989 2990 /* 1) copy till first overlap piece to the rewrite buffer */ 2991 for (;;) { 2992 udf_get_adslot(udf_node, slot, &s_ad, &eof); 2993 if (eof) { 2994 DPRINTF(WRITE, 2995 ("Shrink node failed: " 2996 "encountered EOF\n")); 2997 error = EINVAL; 2998 goto errorout; /* panic? */ 2999 } 3000 len = udf_rw32(s_ad.len); 3001 flags = UDF_EXT_FLAGS(len); 3002 len = UDF_EXT_LEN(len); 3003 3004 if (flags == UDF_EXT_REDIRECT) { 3005 slot++; 3006 continue; 3007 } 3008 3009 end_foffset = foffset + len; 3010 if (end_foffset > new_size) 3011 break; /* found */ 3012 3013 node_ad_cpy[cpy_slot++] = s_ad; 3014 3015 DPRINTF(ALLOC, ("\t1: vp %d, lb %d, len %d, flags %d " 3016 "-> stack\n", 3017 udf_rw16(s_ad.loc.part_num), 3018 udf_rw32(s_ad.loc.lb_num), 3019 UDF_EXT_LEN(udf_rw32(s_ad.len)), 3020 UDF_EXT_FLAGS(udf_rw32(s_ad.len)) >> 30)); 3021 3022 foffset = end_foffset; 3023 slot++; 3024 } 3025 slot_offset = new_size - foffset; 3026 3027 /* 2) trunc overlapping slot at overlap and copy it */ 3028 if (slot_offset > 0) { 3029 lb_num = udf_rw32(s_ad.loc.lb_num); 3030 vpart_num = udf_rw16(s_ad.loc.part_num); 3031 3032 if (flags == UDF_EXT_ALLOCATED) { 3033 /* calculate extent in lb, and offset in lb */ 3034 num_lb = (len + lb_size -1) / lb_size; 3035 slot_offset_lb = (slot_offset + lb_size -1) / lb_size; 3036 3037 /* adjust our slot */ 3038 lb_num += slot_offset_lb; 3039 num_lb -= slot_offset_lb; 3040 3041 udf_free_allocated_space(ump, lb_num, vpart_num, num_lb); 3042 } 3043 3044 s_ad.len = udf_rw32(slot_offset | flags); 3045 node_ad_cpy[cpy_slot++] = s_ad; 3046 slot++; 3047 3048 DPRINTF(ALLOC, ("\t2: vp %d, lb %d, len %d, flags %d " 3049 "-> stack\n", 3050 udf_rw16(s_ad.loc.part_num), 3051 udf_rw32(s_ad.loc.lb_num), 3052 UDF_EXT_LEN(udf_rw32(s_ad.len)), 3053 UDF_EXT_FLAGS(udf_rw32(s_ad.len)) >> 30)); 3054 } 3055 3056 /* 3) delete remainder */ 3057 for (;;) { 3058 udf_get_adslot(udf_node, slot, &s_ad, &eof); 3059 if (eof) 3060 break; 3061 3062 len = udf_rw32(s_ad.len); 3063 flags = UDF_EXT_FLAGS(len); 3064 len = UDF_EXT_LEN(len); 3065 3066 if (flags == UDF_EXT_REDIRECT) { 3067 slot++; 3068 continue; 3069 } 3070 3071 DPRINTF(ALLOC, ("\t3: delete remainder " 3072 "vp %d lb %d, len %d, flags %d\n", 3073 udf_rw16(s_ad.loc.part_num), 3074 udf_rw32(s_ad.loc.lb_num), 3075 UDF_EXT_LEN(udf_rw32(s_ad.len)), 3076 UDF_EXT_FLAGS(udf_rw32(s_ad.len)) >> 30)); 3077 3078 if (flags == UDF_EXT_ALLOCATED) { 3079 lb_num = udf_rw32(s_ad.loc.lb_num); 3080 vpart_num = udf_rw16(s_ad.loc.part_num); 3081 num_lb = (len + lb_size - 1) / lb_size; 3082 3083 udf_free_allocated_space(ump, lb_num, vpart_num, 3084 num_lb); 3085 } 3086 3087 slot++; 3088 } 3089 3090 /* 4) if it will fit into the descriptor then convert */ 3091 if (new_size < max_l_ad) { 3092 /* 3093 * resque/evacuate old piece by reading it in, and convert it 3094 * to internal alloc. 3095 */ 3096 if (new_size == 0) { 3097 /* XXX/TODO only for zero sizing now */ 3098 udf_wipe_adslots(udf_node); 3099 3100 icbflags &= ~UDF_ICB_TAG_FLAGS_ALLOC_MASK; 3101 icbflags |= UDF_ICB_INTERN_ALLOC; 3102 icbtag->flags = udf_rw16(icbflags); 3103 3104 inflen -= size_diff; KASSERT(inflen == 0); 3105 objsize -= size_diff; 3106 l_ad = new_size; 3107 crclen = dscr_size - UDF_DESC_TAG_LENGTH + l_ea + l_ad; 3108 if (fe) { 3109 fe->inf_len = udf_rw64(inflen); 3110 fe->l_ad = udf_rw32(l_ad); 3111 fe->tag.desc_crc_len = udf_rw16(crclen); 3112 } else { 3113 efe->inf_len = udf_rw64(inflen); 3114 efe->obj_size = udf_rw64(objsize); 3115 efe->l_ad = udf_rw32(l_ad); 3116 efe->tag.desc_crc_len = udf_rw16(crclen); 3117 } 3118 /* eventually copy in evacuated piece */ 3119 /* set new size for uvm */ 3120 uvm_vnp_setsize(vp, new_size); 3121 3122 free(node_ad_cpy, M_UDFMNT); 3123 udf_node_sanity_check(udf_node, &new_inflen, &new_lbrec); 3124 3125 UDF_UNLOCK_NODE(udf_node, 0); 3126 3127 KASSERT(new_inflen == orig_inflen - size_diff); 3128 KASSERT(new_inflen == 0); 3129 KASSERT(new_lbrec == 0); 3130 3131 return 0; 3132 } 3133 3134 printf("UDF_SHRINK_NODE: could convert to internal alloc!\n"); 3135 } 3136 3137 /* 5) reset node descriptors */ 3138 udf_wipe_adslots(udf_node); 3139 3140 /* 6) copy back extents; merge when possible. Recounting on the fly */ 3141 cpy_slots = cpy_slot; 3142 3143 c_ad = node_ad_cpy[0]; 3144 slot = 0; 3145 for (cpy_slot = 1; cpy_slot < cpy_slots; cpy_slot++) { 3146 s_ad = node_ad_cpy[cpy_slot]; 3147 3148 DPRINTF(ALLOC, ("\t6: stack -> got mapping vp %d " 3149 "lb %d, len %d, flags %d\n", 3150 udf_rw16(s_ad.loc.part_num), 3151 udf_rw32(s_ad.loc.lb_num), 3152 UDF_EXT_LEN(udf_rw32(s_ad.len)), 3153 UDF_EXT_FLAGS(udf_rw32(s_ad.len)) >> 30)); 3154 3155 /* see if we can merge */ 3156 if (udf_ads_merge(lb_size, &c_ad, &s_ad)) { 3157 /* not mergable (anymore) */ 3158 DPRINTF(ALLOC, ("\t6: appending vp %d lb %d, " 3159 "len %d, flags %d\n", 3160 udf_rw16(c_ad.loc.part_num), 3161 udf_rw32(c_ad.loc.lb_num), 3162 UDF_EXT_LEN(udf_rw32(c_ad.len)), 3163 UDF_EXT_FLAGS(udf_rw32(c_ad.len)) >> 30)); 3164 3165 error = udf_append_adslot(udf_node, &slot, &c_ad); 3166 if (error) 3167 goto errorout; /* panic? */ 3168 c_ad = s_ad; 3169 slot++; 3170 } 3171 } 3172 3173 /* 7) push rest slot (if any) */ 3174 if (UDF_EXT_LEN(c_ad.len) > 0) { 3175 DPRINTF(ALLOC, ("\t7: last append vp %d lb %d, " 3176 "len %d, flags %d\n", 3177 udf_rw16(c_ad.loc.part_num), 3178 udf_rw32(c_ad.loc.lb_num), 3179 UDF_EXT_LEN(udf_rw32(c_ad.len)), 3180 UDF_EXT_FLAGS(udf_rw32(c_ad.len)) >> 30)); 3181 3182 error = udf_append_adslot(udf_node, &slot, &c_ad); 3183 if (error) 3184 goto errorout; /* panic? */ 3185 ; 3186 } 3187 3188 inflen -= size_diff; 3189 objsize -= size_diff; 3190 if (fe) { 3191 fe->inf_len = udf_rw64(inflen); 3192 } else { 3193 efe->inf_len = udf_rw64(inflen); 3194 efe->obj_size = udf_rw64(objsize); 3195 } 3196 error = 0; 3197 3198 /* set new size for uvm */ 3199 uvm_vnp_setsize(vp, new_size); 3200 3201 errorout: 3202 free(node_ad_cpy, M_UDFMNT); 3203 3204 udf_count_alloc_exts(udf_node); 3205 3206 udf_node_sanity_check(udf_node, &new_inflen, &new_lbrec); 3207 UDF_UNLOCK_NODE(udf_node, 0); 3208 3209 KASSERT(new_inflen == orig_inflen - size_diff); 3210 3211 return error; 3212 } 3213 3214