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