1 // SPDX-License-Identifier: GPL-2.0 2 /* 3 * linux/fs/ufs/balloc.c 4 * 5 * Copyright (C) 1998 6 * Daniel Pirkl <daniel.pirkl@email.cz> 7 * Charles University, Faculty of Mathematics and Physics 8 * 9 * UFS2 write support Evgeniy Dushistov <dushistov@mail.ru>, 2007 10 */ 11 12 #include <linux/fs.h> 13 #include <linux/stat.h> 14 #include <linux/time.h> 15 #include <linux/string.h> 16 #include <linux/buffer_head.h> 17 #include <linux/capability.h> 18 #include <linux/bitops.h> 19 #include <linux/bio.h> 20 #include <asm/byteorder.h> 21 22 #include "ufs_fs.h" 23 #include "ufs.h" 24 #include "swab.h" 25 #include "util.h" 26 27 #define INVBLOCK ((u64)-1L) 28 29 static u64 ufs_add_fragments(struct inode *, u64, unsigned, unsigned); 30 static u64 ufs_alloc_fragments(struct inode *, unsigned, u64, unsigned, int *); 31 static u64 ufs_alloccg_block(struct inode *, struct ufs_cg_private_info *, u64, int *); 32 static u64 ufs_bitmap_search (struct super_block *, struct ufs_cg_private_info *, u64, unsigned); 33 static unsigned char ufs_fragtable_8fpb[], ufs_fragtable_other[]; 34 static void ufs_clusteracct(struct super_block *, struct ufs_cg_private_info *, unsigned, int); 35 36 /* 37 * Free 'count' fragments from fragment number 'fragment' 38 */ 39 void ufs_free_fragments(struct inode *inode, u64 fragment, unsigned count) 40 { 41 struct super_block * sb; 42 struct ufs_sb_private_info * uspi; 43 struct ufs_cg_private_info * ucpi; 44 struct ufs_cylinder_group * ucg; 45 unsigned cgno, bit, end_bit, bbase, blkmap, i; 46 u64 blkno; 47 48 sb = inode->i_sb; 49 uspi = UFS_SB(sb)->s_uspi; 50 51 UFSD("ENTER, fragment %llu, count %u\n", 52 (unsigned long long)fragment, count); 53 54 if (ufs_fragnum(fragment) + count > uspi->s_fpg) 55 ufs_error (sb, "ufs_free_fragments", "internal error"); 56 57 mutex_lock(&UFS_SB(sb)->s_lock); 58 59 cgno = ufs_dtog(uspi, fragment); 60 bit = ufs_dtogd(uspi, fragment); 61 if (cgno >= uspi->s_ncg) { 62 ufs_panic (sb, "ufs_free_fragments", "freeing blocks are outside device"); 63 goto failed; 64 } 65 66 ucpi = ufs_load_cylinder (sb, cgno); 67 if (!ucpi) 68 goto failed; 69 ucg = ubh_get_ucg (UCPI_UBH(ucpi)); 70 if (!ufs_cg_chkmagic(sb, ucg)) { 71 ufs_panic (sb, "ufs_free_fragments", "internal error, bad magic number on cg %u", cgno); 72 goto failed; 73 } 74 75 end_bit = bit + count; 76 bbase = ufs_blknum (bit); 77 blkmap = ubh_blkmap (UCPI_UBH(ucpi), ucpi->c_freeoff, bbase); 78 ufs_fragacct (sb, blkmap, ucg->cg_frsum, -1); 79 for (i = bit; i < end_bit; i++) { 80 if (ubh_isclr (UCPI_UBH(ucpi), ucpi->c_freeoff, i)) 81 ubh_setbit (UCPI_UBH(ucpi), ucpi->c_freeoff, i); 82 else 83 ufs_error (sb, "ufs_free_fragments", 84 "bit already cleared for fragment %u", i); 85 } 86 87 inode_sub_bytes(inode, count << uspi->s_fshift); 88 fs32_add(sb, &ucg->cg_cs.cs_nffree, count); 89 uspi->cs_total.cs_nffree += count; 90 fs32_add(sb, &UFS_SB(sb)->fs_cs(cgno).cs_nffree, count); 91 blkmap = ubh_blkmap (UCPI_UBH(ucpi), ucpi->c_freeoff, bbase); 92 ufs_fragacct(sb, blkmap, ucg->cg_frsum, 1); 93 94 /* 95 * Trying to reassemble free fragments into block 96 */ 97 blkno = ufs_fragstoblks (bbase); 98 if (ubh_isblockset(UCPI_UBH(ucpi), ucpi->c_freeoff, blkno)) { 99 fs32_sub(sb, &ucg->cg_cs.cs_nffree, uspi->s_fpb); 100 uspi->cs_total.cs_nffree -= uspi->s_fpb; 101 fs32_sub(sb, &UFS_SB(sb)->fs_cs(cgno).cs_nffree, uspi->s_fpb); 102 if ((UFS_SB(sb)->s_flags & UFS_CG_MASK) == UFS_CG_44BSD) 103 ufs_clusteracct (sb, ucpi, blkno, 1); 104 fs32_add(sb, &ucg->cg_cs.cs_nbfree, 1); 105 uspi->cs_total.cs_nbfree++; 106 fs32_add(sb, &UFS_SB(sb)->fs_cs(cgno).cs_nbfree, 1); 107 if (uspi->fs_magic != UFS2_MAGIC) { 108 unsigned cylno = ufs_cbtocylno (bbase); 109 110 fs16_add(sb, &ubh_cg_blks(ucpi, cylno, 111 ufs_cbtorpos(bbase)), 1); 112 fs32_add(sb, &ubh_cg_blktot(ucpi, cylno), 1); 113 } 114 } 115 116 ubh_mark_buffer_dirty (USPI_UBH(uspi)); 117 ubh_mark_buffer_dirty (UCPI_UBH(ucpi)); 118 if (sb->s_flags & SB_SYNCHRONOUS) 119 ubh_sync_block(UCPI_UBH(ucpi)); 120 ufs_mark_sb_dirty(sb); 121 122 mutex_unlock(&UFS_SB(sb)->s_lock); 123 UFSD("EXIT\n"); 124 return; 125 126 failed: 127 mutex_unlock(&UFS_SB(sb)->s_lock); 128 UFSD("EXIT (FAILED)\n"); 129 return; 130 } 131 132 /* 133 * Free 'count' fragments from fragment number 'fragment' (free whole blocks) 134 */ 135 void ufs_free_blocks(struct inode *inode, u64 fragment, unsigned count) 136 { 137 struct super_block * sb; 138 struct ufs_sb_private_info * uspi; 139 struct ufs_cg_private_info * ucpi; 140 struct ufs_cylinder_group * ucg; 141 unsigned overflow, cgno, bit, end_bit, i; 142 u64 blkno; 143 144 sb = inode->i_sb; 145 uspi = UFS_SB(sb)->s_uspi; 146 147 UFSD("ENTER, fragment %llu, count %u\n", 148 (unsigned long long)fragment, count); 149 150 if ((fragment & uspi->s_fpbmask) || (count & uspi->s_fpbmask)) { 151 ufs_error (sb, "ufs_free_blocks", "internal error, " 152 "fragment %llu, count %u\n", 153 (unsigned long long)fragment, count); 154 goto failed; 155 } 156 157 mutex_lock(&UFS_SB(sb)->s_lock); 158 159 do_more: 160 overflow = 0; 161 cgno = ufs_dtog(uspi, fragment); 162 bit = ufs_dtogd(uspi, fragment); 163 if (cgno >= uspi->s_ncg) { 164 ufs_panic (sb, "ufs_free_blocks", "freeing blocks are outside device"); 165 goto failed_unlock; 166 } 167 end_bit = bit + count; 168 if (end_bit > uspi->s_fpg) { 169 overflow = bit + count - uspi->s_fpg; 170 count -= overflow; 171 end_bit -= overflow; 172 } 173 174 ucpi = ufs_load_cylinder (sb, cgno); 175 if (!ucpi) 176 goto failed_unlock; 177 ucg = ubh_get_ucg (UCPI_UBH(ucpi)); 178 if (!ufs_cg_chkmagic(sb, ucg)) { 179 ufs_panic (sb, "ufs_free_blocks", "internal error, bad magic number on cg %u", cgno); 180 goto failed_unlock; 181 } 182 183 for (i = bit; i < end_bit; i += uspi->s_fpb) { 184 blkno = ufs_fragstoblks(i); 185 if (ubh_isblockset(UCPI_UBH(ucpi), ucpi->c_freeoff, blkno)) { 186 ufs_error(sb, "ufs_free_blocks", "freeing free fragment"); 187 } 188 ubh_setblock(UCPI_UBH(ucpi), ucpi->c_freeoff, blkno); 189 inode_sub_bytes(inode, uspi->s_fpb << uspi->s_fshift); 190 if ((UFS_SB(sb)->s_flags & UFS_CG_MASK) == UFS_CG_44BSD) 191 ufs_clusteracct (sb, ucpi, blkno, 1); 192 193 fs32_add(sb, &ucg->cg_cs.cs_nbfree, 1); 194 uspi->cs_total.cs_nbfree++; 195 fs32_add(sb, &UFS_SB(sb)->fs_cs(cgno).cs_nbfree, 1); 196 197 if (uspi->fs_magic != UFS2_MAGIC) { 198 unsigned cylno = ufs_cbtocylno(i); 199 200 fs16_add(sb, &ubh_cg_blks(ucpi, cylno, 201 ufs_cbtorpos(i)), 1); 202 fs32_add(sb, &ubh_cg_blktot(ucpi, cylno), 1); 203 } 204 } 205 206 ubh_mark_buffer_dirty (USPI_UBH(uspi)); 207 ubh_mark_buffer_dirty (UCPI_UBH(ucpi)); 208 if (sb->s_flags & SB_SYNCHRONOUS) 209 ubh_sync_block(UCPI_UBH(ucpi)); 210 211 if (overflow) { 212 fragment += count; 213 count = overflow; 214 goto do_more; 215 } 216 217 ufs_mark_sb_dirty(sb); 218 mutex_unlock(&UFS_SB(sb)->s_lock); 219 UFSD("EXIT\n"); 220 return; 221 222 failed_unlock: 223 mutex_unlock(&UFS_SB(sb)->s_lock); 224 failed: 225 UFSD("EXIT (FAILED)\n"); 226 return; 227 } 228 229 /* 230 * Modify inode page cache in such way: 231 * have - blocks with b_blocknr equal to oldb...oldb+count-1 232 * get - blocks with b_blocknr equal to newb...newb+count-1 233 * also we suppose that oldb...oldb+count-1 blocks 234 * situated at the end of file. 235 * 236 * We can come here from ufs_writepage or ufs_prepare_write, 237 * locked_page is argument of these functions, so we already lock it. 238 */ 239 static void ufs_change_blocknr(struct inode *inode, sector_t beg, 240 unsigned int count, sector_t oldb, 241 sector_t newb, struct page *locked_page) 242 { 243 const unsigned blks_per_page = 244 1 << (PAGE_SHIFT - inode->i_blkbits); 245 const unsigned mask = blks_per_page - 1; 246 struct address_space * const mapping = inode->i_mapping; 247 pgoff_t index, cur_index, last_index; 248 unsigned pos, j, lblock; 249 sector_t end, i; 250 struct page *page; 251 struct buffer_head *head, *bh; 252 253 UFSD("ENTER, ino %lu, count %u, oldb %llu, newb %llu\n", 254 inode->i_ino, count, 255 (unsigned long long)oldb, (unsigned long long)newb); 256 257 BUG_ON(!locked_page); 258 BUG_ON(!PageLocked(locked_page)); 259 260 cur_index = locked_page->index; 261 end = count + beg; 262 last_index = end >> (PAGE_SHIFT - inode->i_blkbits); 263 for (i = beg; i < end; i = (i | mask) + 1) { 264 index = i >> (PAGE_SHIFT - inode->i_blkbits); 265 266 if (likely(cur_index != index)) { 267 page = ufs_get_locked_page(mapping, index); 268 if (!page)/* it was truncated */ 269 continue; 270 if (IS_ERR(page)) {/* or EIO */ 271 ufs_error(inode->i_sb, __func__, 272 "read of page %llu failed\n", 273 (unsigned long long)index); 274 continue; 275 } 276 } else 277 page = locked_page; 278 279 head = page_buffers(page); 280 bh = head; 281 pos = i & mask; 282 for (j = 0; j < pos; ++j) 283 bh = bh->b_this_page; 284 285 286 if (unlikely(index == last_index)) 287 lblock = end & mask; 288 else 289 lblock = blks_per_page; 290 291 do { 292 if (j >= lblock) 293 break; 294 pos = (i - beg) + j; 295 296 if (!buffer_mapped(bh)) 297 map_bh(bh, inode->i_sb, oldb + pos); 298 if (!buffer_uptodate(bh)) { 299 ll_rw_block(REQ_OP_READ, 0, 1, &bh); 300 wait_on_buffer(bh); 301 if (!buffer_uptodate(bh)) { 302 ufs_error(inode->i_sb, __func__, 303 "read of block failed\n"); 304 break; 305 } 306 } 307 308 UFSD(" change from %llu to %llu, pos %u\n", 309 (unsigned long long)(pos + oldb), 310 (unsigned long long)(pos + newb), pos); 311 312 bh->b_blocknr = newb + pos; 313 clean_bdev_bh_alias(bh); 314 mark_buffer_dirty(bh); 315 ++j; 316 bh = bh->b_this_page; 317 } while (bh != head); 318 319 if (likely(cur_index != index)) 320 ufs_put_locked_page(page); 321 } 322 UFSD("EXIT\n"); 323 } 324 325 static void ufs_clear_frags(struct inode *inode, sector_t beg, unsigned int n, 326 int sync) 327 { 328 struct buffer_head *bh; 329 sector_t end = beg + n; 330 331 for (; beg < end; ++beg) { 332 bh = sb_getblk(inode->i_sb, beg); 333 lock_buffer(bh); 334 memset(bh->b_data, 0, inode->i_sb->s_blocksize); 335 set_buffer_uptodate(bh); 336 mark_buffer_dirty(bh); 337 unlock_buffer(bh); 338 if (IS_SYNC(inode) || sync) 339 sync_dirty_buffer(bh); 340 brelse(bh); 341 } 342 } 343 344 u64 ufs_new_fragments(struct inode *inode, void *p, u64 fragment, 345 u64 goal, unsigned count, int *err, 346 struct page *locked_page) 347 { 348 struct super_block * sb; 349 struct ufs_sb_private_info * uspi; 350 struct ufs_super_block_first * usb1; 351 unsigned cgno, oldcount, newcount; 352 u64 tmp, request, result; 353 354 UFSD("ENTER, ino %lu, fragment %llu, goal %llu, count %u\n", 355 inode->i_ino, (unsigned long long)fragment, 356 (unsigned long long)goal, count); 357 358 sb = inode->i_sb; 359 uspi = UFS_SB(sb)->s_uspi; 360 usb1 = ubh_get_usb_first(uspi); 361 *err = -ENOSPC; 362 363 mutex_lock(&UFS_SB(sb)->s_lock); 364 tmp = ufs_data_ptr_to_cpu(sb, p); 365 366 if (count + ufs_fragnum(fragment) > uspi->s_fpb) { 367 ufs_warning(sb, "ufs_new_fragments", "internal warning" 368 " fragment %llu, count %u", 369 (unsigned long long)fragment, count); 370 count = uspi->s_fpb - ufs_fragnum(fragment); 371 } 372 oldcount = ufs_fragnum (fragment); 373 newcount = oldcount + count; 374 375 /* 376 * Somebody else has just allocated our fragments 377 */ 378 if (oldcount) { 379 if (!tmp) { 380 ufs_error(sb, "ufs_new_fragments", "internal error, " 381 "fragment %llu, tmp %llu\n", 382 (unsigned long long)fragment, 383 (unsigned long long)tmp); 384 mutex_unlock(&UFS_SB(sb)->s_lock); 385 return INVBLOCK; 386 } 387 if (fragment < UFS_I(inode)->i_lastfrag) { 388 UFSD("EXIT (ALREADY ALLOCATED)\n"); 389 mutex_unlock(&UFS_SB(sb)->s_lock); 390 return 0; 391 } 392 } 393 else { 394 if (tmp) { 395 UFSD("EXIT (ALREADY ALLOCATED)\n"); 396 mutex_unlock(&UFS_SB(sb)->s_lock); 397 return 0; 398 } 399 } 400 401 /* 402 * There is not enough space for user on the device 403 */ 404 if (unlikely(ufs_freefrags(uspi) <= uspi->s_root_blocks)) { 405 if (!capable(CAP_SYS_RESOURCE)) { 406 mutex_unlock(&UFS_SB(sb)->s_lock); 407 UFSD("EXIT (FAILED)\n"); 408 return 0; 409 } 410 } 411 412 if (goal >= uspi->s_size) 413 goal = 0; 414 if (goal == 0) 415 cgno = ufs_inotocg (inode->i_ino); 416 else 417 cgno = ufs_dtog(uspi, goal); 418 419 /* 420 * allocate new fragment 421 */ 422 if (oldcount == 0) { 423 result = ufs_alloc_fragments (inode, cgno, goal, count, err); 424 if (result) { 425 ufs_clear_frags(inode, result + oldcount, 426 newcount - oldcount, locked_page != NULL); 427 *err = 0; 428 write_seqlock(&UFS_I(inode)->meta_lock); 429 ufs_cpu_to_data_ptr(sb, p, result); 430 UFS_I(inode)->i_lastfrag = 431 max(UFS_I(inode)->i_lastfrag, fragment + count); 432 write_sequnlock(&UFS_I(inode)->meta_lock); 433 } 434 mutex_unlock(&UFS_SB(sb)->s_lock); 435 UFSD("EXIT, result %llu\n", (unsigned long long)result); 436 return result; 437 } 438 439 /* 440 * resize block 441 */ 442 result = ufs_add_fragments(inode, tmp, oldcount, newcount); 443 if (result) { 444 *err = 0; 445 read_seqlock_excl(&UFS_I(inode)->meta_lock); 446 UFS_I(inode)->i_lastfrag = max(UFS_I(inode)->i_lastfrag, 447 fragment + count); 448 read_sequnlock_excl(&UFS_I(inode)->meta_lock); 449 ufs_clear_frags(inode, result + oldcount, newcount - oldcount, 450 locked_page != NULL); 451 mutex_unlock(&UFS_SB(sb)->s_lock); 452 UFSD("EXIT, result %llu\n", (unsigned long long)result); 453 return result; 454 } 455 456 /* 457 * allocate new block and move data 458 */ 459 if (fs32_to_cpu(sb, usb1->fs_optim) == UFS_OPTSPACE) { 460 request = newcount; 461 if (uspi->cs_total.cs_nffree < uspi->s_space_to_time) 462 usb1->fs_optim = cpu_to_fs32(sb, UFS_OPTTIME); 463 } else { 464 request = uspi->s_fpb; 465 if (uspi->cs_total.cs_nffree > uspi->s_time_to_space) 466 usb1->fs_optim = cpu_to_fs32(sb, UFS_OPTSPACE); 467 } 468 result = ufs_alloc_fragments (inode, cgno, goal, request, err); 469 if (result) { 470 ufs_clear_frags(inode, result + oldcount, newcount - oldcount, 471 locked_page != NULL); 472 mutex_unlock(&UFS_SB(sb)->s_lock); 473 ufs_change_blocknr(inode, fragment - oldcount, oldcount, 474 uspi->s_sbbase + tmp, 475 uspi->s_sbbase + result, locked_page); 476 *err = 0; 477 write_seqlock(&UFS_I(inode)->meta_lock); 478 ufs_cpu_to_data_ptr(sb, p, result); 479 UFS_I(inode)->i_lastfrag = max(UFS_I(inode)->i_lastfrag, 480 fragment + count); 481 write_sequnlock(&UFS_I(inode)->meta_lock); 482 if (newcount < request) 483 ufs_free_fragments (inode, result + newcount, request - newcount); 484 ufs_free_fragments (inode, tmp, oldcount); 485 UFSD("EXIT, result %llu\n", (unsigned long long)result); 486 return result; 487 } 488 489 mutex_unlock(&UFS_SB(sb)->s_lock); 490 UFSD("EXIT (FAILED)\n"); 491 return 0; 492 } 493 494 static bool try_add_frags(struct inode *inode, unsigned frags) 495 { 496 unsigned size = frags * i_blocksize(inode); 497 spin_lock(&inode->i_lock); 498 __inode_add_bytes(inode, size); 499 if (unlikely((u32)inode->i_blocks != inode->i_blocks)) { 500 __inode_sub_bytes(inode, size); 501 spin_unlock(&inode->i_lock); 502 return false; 503 } 504 spin_unlock(&inode->i_lock); 505 return true; 506 } 507 508 static u64 ufs_add_fragments(struct inode *inode, u64 fragment, 509 unsigned oldcount, unsigned newcount) 510 { 511 struct super_block * sb; 512 struct ufs_sb_private_info * uspi; 513 struct ufs_cg_private_info * ucpi; 514 struct ufs_cylinder_group * ucg; 515 unsigned cgno, fragno, fragoff, count, fragsize, i; 516 517 UFSD("ENTER, fragment %llu, oldcount %u, newcount %u\n", 518 (unsigned long long)fragment, oldcount, newcount); 519 520 sb = inode->i_sb; 521 uspi = UFS_SB(sb)->s_uspi; 522 count = newcount - oldcount; 523 524 cgno = ufs_dtog(uspi, fragment); 525 if (fs32_to_cpu(sb, UFS_SB(sb)->fs_cs(cgno).cs_nffree) < count) 526 return 0; 527 if ((ufs_fragnum (fragment) + newcount) > uspi->s_fpb) 528 return 0; 529 ucpi = ufs_load_cylinder (sb, cgno); 530 if (!ucpi) 531 return 0; 532 ucg = ubh_get_ucg (UCPI_UBH(ucpi)); 533 if (!ufs_cg_chkmagic(sb, ucg)) { 534 ufs_panic (sb, "ufs_add_fragments", 535 "internal error, bad magic number on cg %u", cgno); 536 return 0; 537 } 538 539 fragno = ufs_dtogd(uspi, fragment); 540 fragoff = ufs_fragnum (fragno); 541 for (i = oldcount; i < newcount; i++) 542 if (ubh_isclr (UCPI_UBH(ucpi), ucpi->c_freeoff, fragno + i)) 543 return 0; 544 545 if (!try_add_frags(inode, count)) 546 return 0; 547 /* 548 * Block can be extended 549 */ 550 ucg->cg_time = ufs_get_seconds(sb); 551 for (i = newcount; i < (uspi->s_fpb - fragoff); i++) 552 if (ubh_isclr (UCPI_UBH(ucpi), ucpi->c_freeoff, fragno + i)) 553 break; 554 fragsize = i - oldcount; 555 if (!fs32_to_cpu(sb, ucg->cg_frsum[fragsize])) 556 ufs_panic (sb, "ufs_add_fragments", 557 "internal error or corrupted bitmap on cg %u", cgno); 558 fs32_sub(sb, &ucg->cg_frsum[fragsize], 1); 559 if (fragsize != count) 560 fs32_add(sb, &ucg->cg_frsum[fragsize - count], 1); 561 for (i = oldcount; i < newcount; i++) 562 ubh_clrbit (UCPI_UBH(ucpi), ucpi->c_freeoff, fragno + i); 563 564 fs32_sub(sb, &ucg->cg_cs.cs_nffree, count); 565 fs32_sub(sb, &UFS_SB(sb)->fs_cs(cgno).cs_nffree, count); 566 uspi->cs_total.cs_nffree -= count; 567 568 ubh_mark_buffer_dirty (USPI_UBH(uspi)); 569 ubh_mark_buffer_dirty (UCPI_UBH(ucpi)); 570 if (sb->s_flags & SB_SYNCHRONOUS) 571 ubh_sync_block(UCPI_UBH(ucpi)); 572 ufs_mark_sb_dirty(sb); 573 574 UFSD("EXIT, fragment %llu\n", (unsigned long long)fragment); 575 576 return fragment; 577 } 578 579 #define UFS_TEST_FREE_SPACE_CG \ 580 ucg = (struct ufs_cylinder_group *) UFS_SB(sb)->s_ucg[cgno]->b_data; \ 581 if (fs32_to_cpu(sb, ucg->cg_cs.cs_nbfree)) \ 582 goto cg_found; \ 583 for (k = count; k < uspi->s_fpb; k++) \ 584 if (fs32_to_cpu(sb, ucg->cg_frsum[k])) \ 585 goto cg_found; 586 587 static u64 ufs_alloc_fragments(struct inode *inode, unsigned cgno, 588 u64 goal, unsigned count, int *err) 589 { 590 struct super_block * sb; 591 struct ufs_sb_private_info * uspi; 592 struct ufs_cg_private_info * ucpi; 593 struct ufs_cylinder_group * ucg; 594 unsigned oldcg, i, j, k, allocsize; 595 u64 result; 596 597 UFSD("ENTER, ino %lu, cgno %u, goal %llu, count %u\n", 598 inode->i_ino, cgno, (unsigned long long)goal, count); 599 600 sb = inode->i_sb; 601 uspi = UFS_SB(sb)->s_uspi; 602 oldcg = cgno; 603 604 /* 605 * 1. searching on preferred cylinder group 606 */ 607 UFS_TEST_FREE_SPACE_CG 608 609 /* 610 * 2. quadratic rehash 611 */ 612 for (j = 1; j < uspi->s_ncg; j *= 2) { 613 cgno += j; 614 if (cgno >= uspi->s_ncg) 615 cgno -= uspi->s_ncg; 616 UFS_TEST_FREE_SPACE_CG 617 } 618 619 /* 620 * 3. brute force search 621 * We start at i = 2 ( 0 is checked at 1.step, 1 at 2.step ) 622 */ 623 cgno = (oldcg + 1) % uspi->s_ncg; 624 for (j = 2; j < uspi->s_ncg; j++) { 625 cgno++; 626 if (cgno >= uspi->s_ncg) 627 cgno = 0; 628 UFS_TEST_FREE_SPACE_CG 629 } 630 631 UFSD("EXIT (FAILED)\n"); 632 return 0; 633 634 cg_found: 635 ucpi = ufs_load_cylinder (sb, cgno); 636 if (!ucpi) 637 return 0; 638 ucg = ubh_get_ucg (UCPI_UBH(ucpi)); 639 if (!ufs_cg_chkmagic(sb, ucg)) 640 ufs_panic (sb, "ufs_alloc_fragments", 641 "internal error, bad magic number on cg %u", cgno); 642 ucg->cg_time = ufs_get_seconds(sb); 643 644 if (count == uspi->s_fpb) { 645 result = ufs_alloccg_block (inode, ucpi, goal, err); 646 if (result == INVBLOCK) 647 return 0; 648 goto succed; 649 } 650 651 for (allocsize = count; allocsize < uspi->s_fpb; allocsize++) 652 if (fs32_to_cpu(sb, ucg->cg_frsum[allocsize]) != 0) 653 break; 654 655 if (allocsize == uspi->s_fpb) { 656 result = ufs_alloccg_block (inode, ucpi, goal, err); 657 if (result == INVBLOCK) 658 return 0; 659 goal = ufs_dtogd(uspi, result); 660 for (i = count; i < uspi->s_fpb; i++) 661 ubh_setbit (UCPI_UBH(ucpi), ucpi->c_freeoff, goal + i); 662 i = uspi->s_fpb - count; 663 664 inode_sub_bytes(inode, i << uspi->s_fshift); 665 fs32_add(sb, &ucg->cg_cs.cs_nffree, i); 666 uspi->cs_total.cs_nffree += i; 667 fs32_add(sb, &UFS_SB(sb)->fs_cs(cgno).cs_nffree, i); 668 fs32_add(sb, &ucg->cg_frsum[i], 1); 669 goto succed; 670 } 671 672 result = ufs_bitmap_search (sb, ucpi, goal, allocsize); 673 if (result == INVBLOCK) 674 return 0; 675 if (!try_add_frags(inode, count)) 676 return 0; 677 for (i = 0; i < count; i++) 678 ubh_clrbit (UCPI_UBH(ucpi), ucpi->c_freeoff, result + i); 679 680 fs32_sub(sb, &ucg->cg_cs.cs_nffree, count); 681 uspi->cs_total.cs_nffree -= count; 682 fs32_sub(sb, &UFS_SB(sb)->fs_cs(cgno).cs_nffree, count); 683 fs32_sub(sb, &ucg->cg_frsum[allocsize], 1); 684 685 if (count != allocsize) 686 fs32_add(sb, &ucg->cg_frsum[allocsize - count], 1); 687 688 succed: 689 ubh_mark_buffer_dirty (USPI_UBH(uspi)); 690 ubh_mark_buffer_dirty (UCPI_UBH(ucpi)); 691 if (sb->s_flags & SB_SYNCHRONOUS) 692 ubh_sync_block(UCPI_UBH(ucpi)); 693 ufs_mark_sb_dirty(sb); 694 695 result += cgno * uspi->s_fpg; 696 UFSD("EXIT3, result %llu\n", (unsigned long long)result); 697 return result; 698 } 699 700 static u64 ufs_alloccg_block(struct inode *inode, 701 struct ufs_cg_private_info *ucpi, 702 u64 goal, int *err) 703 { 704 struct super_block * sb; 705 struct ufs_sb_private_info * uspi; 706 struct ufs_cylinder_group * ucg; 707 u64 result, blkno; 708 709 UFSD("ENTER, goal %llu\n", (unsigned long long)goal); 710 711 sb = inode->i_sb; 712 uspi = UFS_SB(sb)->s_uspi; 713 ucg = ubh_get_ucg(UCPI_UBH(ucpi)); 714 715 if (goal == 0) { 716 goal = ucpi->c_rotor; 717 goto norot; 718 } 719 goal = ufs_blknum (goal); 720 goal = ufs_dtogd(uspi, goal); 721 722 /* 723 * If the requested block is available, use it. 724 */ 725 if (ubh_isblockset(UCPI_UBH(ucpi), ucpi->c_freeoff, ufs_fragstoblks(goal))) { 726 result = goal; 727 goto gotit; 728 } 729 730 norot: 731 result = ufs_bitmap_search (sb, ucpi, goal, uspi->s_fpb); 732 if (result == INVBLOCK) 733 return INVBLOCK; 734 ucpi->c_rotor = result; 735 gotit: 736 if (!try_add_frags(inode, uspi->s_fpb)) 737 return 0; 738 blkno = ufs_fragstoblks(result); 739 ubh_clrblock (UCPI_UBH(ucpi), ucpi->c_freeoff, blkno); 740 if ((UFS_SB(sb)->s_flags & UFS_CG_MASK) == UFS_CG_44BSD) 741 ufs_clusteracct (sb, ucpi, blkno, -1); 742 743 fs32_sub(sb, &ucg->cg_cs.cs_nbfree, 1); 744 uspi->cs_total.cs_nbfree--; 745 fs32_sub(sb, &UFS_SB(sb)->fs_cs(ucpi->c_cgx).cs_nbfree, 1); 746 747 if (uspi->fs_magic != UFS2_MAGIC) { 748 unsigned cylno = ufs_cbtocylno((unsigned)result); 749 750 fs16_sub(sb, &ubh_cg_blks(ucpi, cylno, 751 ufs_cbtorpos((unsigned)result)), 1); 752 fs32_sub(sb, &ubh_cg_blktot(ucpi, cylno), 1); 753 } 754 755 UFSD("EXIT, result %llu\n", (unsigned long long)result); 756 757 return result; 758 } 759 760 static unsigned ubh_scanc(struct ufs_sb_private_info *uspi, 761 struct ufs_buffer_head *ubh, 762 unsigned begin, unsigned size, 763 unsigned char *table, unsigned char mask) 764 { 765 unsigned rest, offset; 766 unsigned char *cp; 767 768 769 offset = begin & ~uspi->s_fmask; 770 begin >>= uspi->s_fshift; 771 for (;;) { 772 if ((offset + size) < uspi->s_fsize) 773 rest = size; 774 else 775 rest = uspi->s_fsize - offset; 776 size -= rest; 777 cp = ubh->bh[begin]->b_data + offset; 778 while ((table[*cp++] & mask) == 0 && --rest) 779 ; 780 if (rest || !size) 781 break; 782 begin++; 783 offset = 0; 784 } 785 return (size + rest); 786 } 787 788 /* 789 * Find a block of the specified size in the specified cylinder group. 790 * @sp: pointer to super block 791 * @ucpi: pointer to cylinder group info 792 * @goal: near which block we want find new one 793 * @count: specified size 794 */ 795 static u64 ufs_bitmap_search(struct super_block *sb, 796 struct ufs_cg_private_info *ucpi, 797 u64 goal, unsigned count) 798 { 799 /* 800 * Bit patterns for identifying fragments in the block map 801 * used as ((map & mask_arr) == want_arr) 802 */ 803 static const int mask_arr[9] = { 804 0x3, 0x7, 0xf, 0x1f, 0x3f, 0x7f, 0xff, 0x1ff, 0x3ff 805 }; 806 static const int want_arr[9] = { 807 0x0, 0x2, 0x6, 0xe, 0x1e, 0x3e, 0x7e, 0xfe, 0x1fe 808 }; 809 struct ufs_sb_private_info *uspi = UFS_SB(sb)->s_uspi; 810 unsigned start, length, loc; 811 unsigned pos, want, blockmap, mask, end; 812 u64 result; 813 814 UFSD("ENTER, cg %u, goal %llu, count %u\n", ucpi->c_cgx, 815 (unsigned long long)goal, count); 816 817 if (goal) 818 start = ufs_dtogd(uspi, goal) >> 3; 819 else 820 start = ucpi->c_frotor >> 3; 821 822 length = ((uspi->s_fpg + 7) >> 3) - start; 823 loc = ubh_scanc(uspi, UCPI_UBH(ucpi), ucpi->c_freeoff + start, length, 824 (uspi->s_fpb == 8) ? ufs_fragtable_8fpb : ufs_fragtable_other, 825 1 << (count - 1 + (uspi->s_fpb & 7))); 826 if (loc == 0) { 827 length = start + 1; 828 loc = ubh_scanc(uspi, UCPI_UBH(ucpi), ucpi->c_freeoff, length, 829 (uspi->s_fpb == 8) ? ufs_fragtable_8fpb : 830 ufs_fragtable_other, 831 1 << (count - 1 + (uspi->s_fpb & 7))); 832 if (loc == 0) { 833 ufs_error(sb, "ufs_bitmap_search", 834 "bitmap corrupted on cg %u, start %u," 835 " length %u, count %u, freeoff %u\n", 836 ucpi->c_cgx, start, length, count, 837 ucpi->c_freeoff); 838 return INVBLOCK; 839 } 840 start = 0; 841 } 842 result = (start + length - loc) << 3; 843 ucpi->c_frotor = result; 844 845 /* 846 * found the byte in the map 847 */ 848 849 for (end = result + 8; result < end; result += uspi->s_fpb) { 850 blockmap = ubh_blkmap(UCPI_UBH(ucpi), ucpi->c_freeoff, result); 851 blockmap <<= 1; 852 mask = mask_arr[count]; 853 want = want_arr[count]; 854 for (pos = 0; pos <= uspi->s_fpb - count; pos++) { 855 if ((blockmap & mask) == want) { 856 UFSD("EXIT, result %llu\n", 857 (unsigned long long)result); 858 return result + pos; 859 } 860 mask <<= 1; 861 want <<= 1; 862 } 863 } 864 865 ufs_error(sb, "ufs_bitmap_search", "block not in map on cg %u\n", 866 ucpi->c_cgx); 867 UFSD("EXIT (FAILED)\n"); 868 return INVBLOCK; 869 } 870 871 static void ufs_clusteracct(struct super_block * sb, 872 struct ufs_cg_private_info * ucpi, unsigned blkno, int cnt) 873 { 874 struct ufs_sb_private_info * uspi; 875 int i, start, end, forw, back; 876 877 uspi = UFS_SB(sb)->s_uspi; 878 if (uspi->s_contigsumsize <= 0) 879 return; 880 881 if (cnt > 0) 882 ubh_setbit(UCPI_UBH(ucpi), ucpi->c_clusteroff, blkno); 883 else 884 ubh_clrbit(UCPI_UBH(ucpi), ucpi->c_clusteroff, blkno); 885 886 /* 887 * Find the size of the cluster going forward. 888 */ 889 start = blkno + 1; 890 end = start + uspi->s_contigsumsize; 891 if ( end >= ucpi->c_nclusterblks) 892 end = ucpi->c_nclusterblks; 893 i = ubh_find_next_zero_bit (UCPI_UBH(ucpi), ucpi->c_clusteroff, end, start); 894 if (i > end) 895 i = end; 896 forw = i - start; 897 898 /* 899 * Find the size of the cluster going backward. 900 */ 901 start = blkno - 1; 902 end = start - uspi->s_contigsumsize; 903 if (end < 0 ) 904 end = -1; 905 i = ubh_find_last_zero_bit (UCPI_UBH(ucpi), ucpi->c_clusteroff, start, end); 906 if ( i < end) 907 i = end; 908 back = start - i; 909 910 /* 911 * Account for old cluster and the possibly new forward and 912 * back clusters. 913 */ 914 i = back + forw + 1; 915 if (i > uspi->s_contigsumsize) 916 i = uspi->s_contigsumsize; 917 fs32_add(sb, (__fs32*)ubh_get_addr(UCPI_UBH(ucpi), ucpi->c_clustersumoff + (i << 2)), cnt); 918 if (back > 0) 919 fs32_sub(sb, (__fs32*)ubh_get_addr(UCPI_UBH(ucpi), ucpi->c_clustersumoff + (back << 2)), cnt); 920 if (forw > 0) 921 fs32_sub(sb, (__fs32*)ubh_get_addr(UCPI_UBH(ucpi), ucpi->c_clustersumoff + (forw << 2)), cnt); 922 } 923 924 925 static unsigned char ufs_fragtable_8fpb[] = { 926 0x00, 0x01, 0x01, 0x02, 0x01, 0x01, 0x02, 0x04, 0x01, 0x01, 0x01, 0x03, 0x02, 0x03, 0x04, 0x08, 927 0x01, 0x01, 0x01, 0x03, 0x01, 0x01, 0x03, 0x05, 0x02, 0x03, 0x03, 0x02, 0x04, 0x05, 0x08, 0x10, 928 0x01, 0x01, 0x01, 0x03, 0x01, 0x01, 0x03, 0x05, 0x01, 0x01, 0x01, 0x03, 0x03, 0x03, 0x05, 0x09, 929 0x02, 0x03, 0x03, 0x02, 0x03, 0x03, 0x02, 0x06, 0x04, 0x05, 0x05, 0x06, 0x08, 0x09, 0x10, 0x20, 930 0x01, 0x01, 0x01, 0x03, 0x01, 0x01, 0x03, 0x05, 0x01, 0x01, 0x01, 0x03, 0x03, 0x03, 0x05, 0x09, 931 0x01, 0x01, 0x01, 0x03, 0x01, 0x01, 0x03, 0x05, 0x03, 0x03, 0x03, 0x03, 0x05, 0x05, 0x09, 0x11, 932 0x02, 0x03, 0x03, 0x02, 0x03, 0x03, 0x02, 0x06, 0x03, 0x03, 0x03, 0x03, 0x02, 0x03, 0x06, 0x0A, 933 0x04, 0x05, 0x05, 0x06, 0x05, 0x05, 0x06, 0x04, 0x08, 0x09, 0x09, 0x0A, 0x10, 0x11, 0x20, 0x40, 934 0x01, 0x01, 0x01, 0x03, 0x01, 0x01, 0x03, 0x05, 0x01, 0x01, 0x01, 0x03, 0x03, 0x03, 0x05, 0x09, 935 0x01, 0x01, 0x01, 0x03, 0x01, 0x01, 0x03, 0x05, 0x03, 0x03, 0x03, 0x03, 0x05, 0x05, 0x09, 0x11, 936 0x01, 0x01, 0x01, 0x03, 0x01, 0x01, 0x03, 0x05, 0x01, 0x01, 0x01, 0x03, 0x03, 0x03, 0x05, 0x09, 937 0x03, 0x03, 0x03, 0x03, 0x03, 0x03, 0x03, 0x07, 0x05, 0x05, 0x05, 0x07, 0x09, 0x09, 0x11, 0x21, 938 0x02, 0x03, 0x03, 0x02, 0x03, 0x03, 0x02, 0x06, 0x03, 0x03, 0x03, 0x03, 0x02, 0x03, 0x06, 0x0A, 939 0x03, 0x03, 0x03, 0x03, 0x03, 0x03, 0x03, 0x07, 0x02, 0x03, 0x03, 0x02, 0x06, 0x07, 0x0A, 0x12, 940 0x04, 0x05, 0x05, 0x06, 0x05, 0x05, 0x06, 0x04, 0x05, 0x05, 0x05, 0x07, 0x06, 0x07, 0x04, 0x0C, 941 0x08, 0x09, 0x09, 0x0A, 0x09, 0x09, 0x0A, 0x0C, 0x10, 0x11, 0x11, 0x12, 0x20, 0x21, 0x40, 0x80, 942 }; 943 944 static unsigned char ufs_fragtable_other[] = { 945 0x00, 0x16, 0x16, 0x2A, 0x16, 0x16, 0x26, 0x4E, 0x16, 0x16, 0x16, 0x3E, 0x2A, 0x3E, 0x4E, 0x8A, 946 0x16, 0x16, 0x16, 0x3E, 0x16, 0x16, 0x36, 0x5E, 0x16, 0x16, 0x16, 0x3E, 0x3E, 0x3E, 0x5E, 0x9E, 947 0x16, 0x16, 0x16, 0x3E, 0x16, 0x16, 0x36, 0x5E, 0x16, 0x16, 0x16, 0x3E, 0x3E, 0x3E, 0x5E, 0x9E, 948 0x2A, 0x3E, 0x3E, 0x2A, 0x3E, 0x3E, 0x2E, 0x6E, 0x3E, 0x3E, 0x3E, 0x3E, 0x2A, 0x3E, 0x6E, 0xAA, 949 0x16, 0x16, 0x16, 0x3E, 0x16, 0x16, 0x36, 0x5E, 0x16, 0x16, 0x16, 0x3E, 0x3E, 0x3E, 0x5E, 0x9E, 950 0x16, 0x16, 0x16, 0x3E, 0x16, 0x16, 0x36, 0x5E, 0x16, 0x16, 0x16, 0x3E, 0x3E, 0x3E, 0x5E, 0x9E, 951 0x26, 0x36, 0x36, 0x2E, 0x36, 0x36, 0x26, 0x6E, 0x36, 0x36, 0x36, 0x3E, 0x2E, 0x3E, 0x6E, 0xAE, 952 0x4E, 0x5E, 0x5E, 0x6E, 0x5E, 0x5E, 0x6E, 0x4E, 0x5E, 0x5E, 0x5E, 0x7E, 0x6E, 0x7E, 0x4E, 0xCE, 953 0x16, 0x16, 0x16, 0x3E, 0x16, 0x16, 0x36, 0x5E, 0x16, 0x16, 0x16, 0x3E, 0x3E, 0x3E, 0x5E, 0x9E, 954 0x16, 0x16, 0x16, 0x3E, 0x16, 0x16, 0x36, 0x5E, 0x16, 0x16, 0x16, 0x3E, 0x3E, 0x3E, 0x5E, 0x9E, 955 0x16, 0x16, 0x16, 0x3E, 0x16, 0x16, 0x36, 0x5E, 0x16, 0x16, 0x16, 0x3E, 0x3E, 0x3E, 0x5E, 0x9E, 956 0x3E, 0x3E, 0x3E, 0x3E, 0x3E, 0x3E, 0x3E, 0x7E, 0x3E, 0x3E, 0x3E, 0x3E, 0x3E, 0x3E, 0x7E, 0xBE, 957 0x2A, 0x3E, 0x3E, 0x2A, 0x3E, 0x3E, 0x2E, 0x6E, 0x3E, 0x3E, 0x3E, 0x3E, 0x2A, 0x3E, 0x6E, 0xAA, 958 0x3E, 0x3E, 0x3E, 0x3E, 0x3E, 0x3E, 0x3E, 0x7E, 0x3E, 0x3E, 0x3E, 0x3E, 0x3E, 0x3E, 0x7E, 0xBE, 959 0x4E, 0x5E, 0x5E, 0x6E, 0x5E, 0x5E, 0x6E, 0x4E, 0x5E, 0x5E, 0x5E, 0x7E, 0x6E, 0x7E, 0x4E, 0xCE, 960 0x8A, 0x9E, 0x9E, 0xAA, 0x9E, 0x9E, 0xAE, 0xCE, 0x9E, 0x9E, 0x9E, 0xBE, 0xAA, 0xBE, 0xCE, 0x8A, 961 }; 962