The Design and Implementation of the FreeBSD Operating System, Second Edition
Now available: The Design and Implementation of the FreeBSD Operating System (Second Edition)


[ source navigation ] [ diff markup ] [ identifier search ] [ freetext search ] [ file search ] [ list types ] [ track identifier ]

FreeBSD/Linux Kernel Cross Reference
sys/fs/ext2fs/ext2_alloc.c

Version: -  FREEBSD  -  FREEBSD-13-STABLE  -  FREEBSD-13-0  -  FREEBSD-12-STABLE  -  FREEBSD-12-0  -  FREEBSD-11-STABLE  -  FREEBSD-11-0  -  FREEBSD-10-STABLE  -  FREEBSD-10-0  -  FREEBSD-9-STABLE  -  FREEBSD-9-0  -  FREEBSD-8-STABLE  -  FREEBSD-8-0  -  FREEBSD-7-STABLE  -  FREEBSD-7-0  -  FREEBSD-6-STABLE  -  FREEBSD-6-0  -  FREEBSD-5-STABLE  -  FREEBSD-5-0  -  FREEBSD-4-STABLE  -  FREEBSD-3-STABLE  -  FREEBSD22  -  l41  -  OPENBSD  -  linux-2.6  -  MK84  -  PLAN9  -  xnu-8792 
SearchContext: -  none  -  3  -  10 

    1 /*-
    2  *  modified for Lites 1.1
    3  *
    4  *  Aug 1995, Godmar Back (gback@cs.utah.edu)
    5  *  University of Utah, Department of Computer Science
    6  */
    7 /*-
    8  * SPDX-License-Identifier: BSD-3-Clause
    9  *
   10  * Copyright (c) 1982, 1986, 1989, 1993
   11  *      The Regents of the University of California.  All rights reserved.
   12  *
   13  * Redistribution and use in source and binary forms, with or without
   14  * modification, are permitted provided that the following conditions
   15  * are met:
   16  * 1. Redistributions of source code must retain the above copyright
   17  *    notice, this list of conditions and the following disclaimer.
   18  * 2. Redistributions in binary form must reproduce the above copyright
   19  *    notice, this list of conditions and the following disclaimer in the
   20  *    documentation and/or other materials provided with the distribution.
   21  * 3. Neither the name of the University nor the names of its contributors
   22  *    may be used to endorse or promote products derived from this software
   23  *    without specific prior written permission.
   24  *
   25  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
   26  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
   27  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
   28  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
   29  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
   30  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
   31  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
   32  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
   33  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
   34  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
   35  * SUCH DAMAGE.
   36  *
   37  *      @(#)ffs_alloc.c 8.8 (Berkeley) 2/21/94
   38  * $FreeBSD$
   39  */
   40 
   41 #include <sys/param.h>
   42 #include <sys/systm.h>
   43 #include <sys/conf.h>
   44 #include <sys/vnode.h>
   45 #include <sys/sdt.h>
   46 #include <sys/stat.h>
   47 #include <sys/mount.h>
   48 #include <sys/sysctl.h>
   49 #include <sys/syslog.h>
   50 #include <sys/buf.h>
   51 #include <sys/endian.h>
   52 
   53 #include <fs/ext2fs/fs.h>
   54 #include <fs/ext2fs/inode.h>
   55 #include <fs/ext2fs/ext2_mount.h>
   56 #include <fs/ext2fs/ext2fs.h>
   57 #include <fs/ext2fs/ext2_extern.h>
   58 
   59 SDT_PROVIDER_DEFINE(ext2fs);
   60 /*
   61  * ext2fs trace probe:
   62  * arg0: verbosity. Higher numbers give more verbose messages
   63  * arg1: Textual message
   64  */
   65 SDT_PROBE_DEFINE2(ext2fs, , alloc, trace, "int", "char*");
   66 SDT_PROBE_DEFINE3(ext2fs, , alloc, ext2_reallocblks_realloc,
   67     "ino_t", "e2fs_lbn_t", "e2fs_lbn_t");
   68 SDT_PROBE_DEFINE1(ext2fs, , alloc, ext2_reallocblks_bap, "uint32_t");
   69 SDT_PROBE_DEFINE1(ext2fs, , alloc, ext2_reallocblks_blkno, "e2fs_daddr_t");
   70 SDT_PROBE_DEFINE2(ext2fs, , alloc, ext2_b_bitmap_validate_error, "char*", "int");
   71 SDT_PROBE_DEFINE3(ext2fs, , alloc, ext2_nodealloccg_bmap_corrupted,
   72     "int", "daddr_t", "char*");
   73 SDT_PROBE_DEFINE2(ext2fs, , alloc, ext2_blkfree_bad_block, "ino_t", "e4fs_daddr_t");
   74 SDT_PROBE_DEFINE2(ext2fs, , alloc, ext2_vfree_doublefree, "char*", "ino_t");
   75 
   76 static daddr_t  ext2_alloccg(struct inode *, int, daddr_t, int);
   77 static daddr_t  ext2_clusteralloc(struct inode *, int, daddr_t, int);
   78 static u_long   ext2_dirpref(struct inode *);
   79 static e4fs_daddr_t ext2_hashalloc(struct inode *, int, long, int,
   80     daddr_t (*)(struct inode *, int, daddr_t, 
   81                                                 int));
   82 static daddr_t  ext2_nodealloccg(struct inode *, int, daddr_t, int);
   83 static daddr_t  ext2_mapsearch(struct m_ext2fs *, char *, daddr_t);
   84 
   85 /*
   86  * Allocate a block in the filesystem.
   87  *
   88  * A preference may be optionally specified. If a preference is given
   89  * the following hierarchy is used to allocate a block:
   90  *   1) allocate the requested block.
   91  *   2) allocate a rotationally optimal block in the same cylinder.
   92  *   3) allocate a block in the same cylinder group.
   93  *   4) quadratically rehash into other cylinder groups, until an
   94  *        available block is located.
   95  * If no block preference is given the following hierarchy is used
   96  * to allocate a block:
   97  *   1) allocate a block in the cylinder group that contains the
   98  *        inode for the file.
   99  *   2) quadratically rehash into other cylinder groups, until an
  100  *        available block is located.
  101  */
  102 int
  103 ext2_alloc(struct inode *ip, daddr_t lbn, e4fs_daddr_t bpref, int size,
  104     struct ucred *cred, e4fs_daddr_t *bnp)
  105 {
  106         struct m_ext2fs *fs;
  107         struct ext2mount *ump;
  108         e4fs_daddr_t bno;
  109         int cg;
  110 
  111         *bnp = 0;
  112         fs = ip->i_e2fs;
  113         ump = ip->i_ump;
  114         mtx_assert(EXT2_MTX(ump), MA_OWNED);
  115 #ifdef INVARIANTS
  116         if ((u_int)size > fs->e2fs_bsize || blkoff(fs, size) != 0) {
  117                 vn_printf(ip->i_devvp, "bsize = %lu, size = %d, fs = %s\n",
  118                     (long unsigned int)fs->e2fs_bsize, size, fs->e2fs_fsmnt);
  119                 panic("ext2_alloc: bad size");
  120         }
  121         if (cred == NOCRED)
  122                 panic("ext2_alloc: missing credential");
  123 #endif          /* INVARIANTS */
  124         if (size == fs->e2fs_bsize && fs->e2fs_fbcount == 0)
  125                 goto nospace;
  126         if (cred->cr_uid != 0 &&
  127             fs->e2fs_fbcount < fs->e2fs_rbcount)
  128                 goto nospace;
  129         if (bpref >= fs->e2fs_bcount)
  130                 bpref = 0;
  131         if (bpref == 0)
  132                 cg = ino_to_cg(fs, ip->i_number);
  133         else
  134                 cg = dtog(fs, bpref);
  135         bno = (daddr_t)ext2_hashalloc(ip, cg, bpref, fs->e2fs_bsize,
  136             ext2_alloccg);
  137         if (bno > 0) {
  138                 /* set next_alloc fields as done in block_getblk */
  139                 ip->i_next_alloc_block = lbn;
  140                 ip->i_next_alloc_goal = bno;
  141 
  142                 ip->i_blocks += btodb(fs->e2fs_bsize);
  143                 ip->i_flag |= IN_CHANGE | IN_UPDATE;
  144                 *bnp = bno;
  145                 return (0);
  146         }
  147 nospace:
  148         EXT2_UNLOCK(ump);
  149         SDT_PROBE2(ext2fs, , alloc, trace, 1, "cannot allocate data block");
  150         return (ENOSPC);
  151 }
  152 
  153 /*
  154  * Allocate EA's block for inode.
  155  */
  156 e4fs_daddr_t
  157 ext2_alloc_meta(struct inode *ip)
  158 {
  159         struct m_ext2fs *fs;
  160         daddr_t blk;
  161 
  162         fs = ip->i_e2fs;
  163 
  164         EXT2_LOCK(ip->i_ump);
  165         blk = ext2_hashalloc(ip, ino_to_cg(fs, ip->i_number), 0, fs->e2fs_bsize,
  166             ext2_alloccg);
  167         if (0 == blk) {
  168                 EXT2_UNLOCK(ip->i_ump);
  169                 SDT_PROBE2(ext2fs, , alloc, trace, 1, "cannot allocate meta block");
  170         }
  171 
  172         return (blk);
  173 }
  174 
  175 /*
  176  * Reallocate a sequence of blocks into a contiguous sequence of blocks.
  177  *
  178  * The vnode and an array of buffer pointers for a range of sequential
  179  * logical blocks to be made contiguous is given. The allocator attempts
  180  * to find a range of sequential blocks starting as close as possible to
  181  * an fs_rotdelay offset from the end of the allocation for the logical
  182  * block immediately preceding the current range. If successful, the
  183  * physical block numbers in the buffer pointers and in the inode are
  184  * changed to reflect the new allocation. If unsuccessful, the allocation
  185  * is left unchanged. The success in doing the reallocation is returned.
  186  * Note that the error return is not reflected back to the user. Rather
  187  * the previous block allocation will be used.
  188  */
  189 
  190 static SYSCTL_NODE(_vfs, OID_AUTO, ext2fs, CTLFLAG_RW | CTLFLAG_MPSAFE, 0,
  191     "EXT2FS filesystem");
  192 
  193 static int doasyncfree = 1;
  194 
  195 SYSCTL_INT(_vfs_ext2fs, OID_AUTO, doasyncfree, CTLFLAG_RW, &doasyncfree, 0,
  196     "Use asynchronous writes to update block pointers when freeing blocks");
  197 
  198 static int doreallocblks = 0;
  199 
  200 SYSCTL_INT(_vfs_ext2fs, OID_AUTO, doreallocblks, CTLFLAG_RW, &doreallocblks, 0, "");
  201 
  202 int
  203 ext2_reallocblks(struct vop_reallocblks_args *ap)
  204 {
  205         struct m_ext2fs *fs;
  206         struct inode *ip;
  207         struct vnode *vp;
  208         struct buf *sbp, *ebp;
  209         uint32_t *bap, *sbap, *ebap;
  210         struct ext2mount *ump;
  211         struct cluster_save *buflist;
  212         struct indir start_ap[EXT2_NIADDR + 1], end_ap[EXT2_NIADDR + 1], *idp;
  213         e2fs_lbn_t start_lbn, end_lbn;
  214         int soff;
  215         e2fs_daddr_t newblk, blkno;
  216         int i, len, start_lvl, end_lvl, pref, ssize;
  217 
  218         if (doreallocblks == 0)
  219                 return (ENOSPC);
  220 
  221         vp = ap->a_vp;
  222         ip = VTOI(vp);
  223         fs = ip->i_e2fs;
  224         ump = ip->i_ump;
  225 
  226         if (fs->e2fs_contigsumsize <= 0 || ip->i_flag & IN_E4EXTENTS)
  227                 return (ENOSPC);
  228 
  229         buflist = ap->a_buflist;
  230         len = buflist->bs_nchildren;
  231         start_lbn = buflist->bs_children[0]->b_lblkno;
  232         end_lbn = start_lbn + len - 1;
  233 #ifdef INVARIANTS
  234         for (i = 1; i < len; i++)
  235                 if (buflist->bs_children[i]->b_lblkno != start_lbn + i)
  236                         panic("ext2_reallocblks: non-cluster");
  237 #endif
  238         /*
  239          * If the cluster crosses the boundary for the first indirect
  240          * block, leave space for the indirect block. Indirect blocks
  241          * are initially laid out in a position after the last direct
  242          * block. Block reallocation would usually destroy locality by
  243          * moving the indirect block out of the way to make room for
  244          * data blocks if we didn't compensate here. We should also do
  245          * this for other indirect block boundaries, but it is only
  246          * important for the first one.
  247          */
  248         if (start_lbn < EXT2_NDADDR && end_lbn >= EXT2_NDADDR)
  249                 return (ENOSPC);
  250         /*
  251          * If the latest allocation is in a new cylinder group, assume that
  252          * the filesystem has decided to move and do not force it back to
  253          * the previous cylinder group.
  254          */
  255         if (dtog(fs, dbtofsb(fs, buflist->bs_children[0]->b_blkno)) !=
  256             dtog(fs, dbtofsb(fs, buflist->bs_children[len - 1]->b_blkno)))
  257                 return (ENOSPC);
  258         if (ext2_getlbns(vp, start_lbn, start_ap, &start_lvl) ||
  259             ext2_getlbns(vp, end_lbn, end_ap, &end_lvl))
  260                 return (ENOSPC);
  261         /*
  262          * Get the starting offset and block map for the first block.
  263          */
  264         if (start_lvl == 0) {
  265                 sbap = &ip->i_db[0];
  266                 soff = start_lbn;
  267         } else {
  268                 idp = &start_ap[start_lvl - 1];
  269                 if (bread(vp, idp->in_lbn, (int)fs->e2fs_bsize, NOCRED, &sbp)) {
  270                         brelse(sbp);
  271                         return (ENOSPC);
  272                 }
  273                 sbap = (u_int *)sbp->b_data;
  274                 soff = idp->in_off;
  275         }
  276         /*
  277          * If the block range spans two block maps, get the second map.
  278          */
  279         ebap = NULL;
  280         if (end_lvl == 0 || (idp = &end_ap[end_lvl - 1])->in_off + 1 >= len) {
  281                 ssize = len;
  282         } else {
  283 #ifdef INVARIANTS
  284                 if (start_ap[start_lvl - 1].in_lbn == idp->in_lbn)
  285                         panic("ext2_reallocblks: start == end");
  286 #endif
  287                 ssize = len - (idp->in_off + 1);
  288                 if (bread(vp, idp->in_lbn, (int)fs->e2fs_bsize, NOCRED, &ebp))
  289                         goto fail;
  290                 ebap = (u_int *)ebp->b_data;
  291         }
  292         /*
  293          * Find the preferred location for the cluster.
  294          */
  295         EXT2_LOCK(ump);
  296         pref = ext2_blkpref(ip, start_lbn, soff, sbap, 0);
  297         /*
  298          * Search the block map looking for an allocation of the desired size.
  299          */
  300         if ((newblk = (e2fs_daddr_t)ext2_hashalloc(ip, dtog(fs, pref), pref,
  301             len, ext2_clusteralloc)) == 0) {
  302                 EXT2_UNLOCK(ump);
  303                 goto fail;
  304         }
  305         /*
  306          * We have found a new contiguous block.
  307          *
  308          * First we have to replace the old block pointers with the new
  309          * block pointers in the inode and indirect blocks associated
  310          * with the file.
  311          */
  312         SDT_PROBE3(ext2fs, , alloc, ext2_reallocblks_realloc,
  313             ip->i_number, start_lbn, end_lbn);
  314         blkno = newblk;
  315         for (bap = &sbap[soff], i = 0; i < len; i++, blkno += fs->e2fs_fpb) {
  316                 if (i == ssize) {
  317                         bap = ebap;
  318                         soff = -i;
  319                 }
  320 #ifdef INVARIANTS
  321                 if (buflist->bs_children[i]->b_blkno != fsbtodb(fs, *bap))
  322                         panic("ext2_reallocblks: alloc mismatch");
  323 #endif
  324                 SDT_PROBE1(ext2fs, , alloc, ext2_reallocblks_bap, *bap);
  325                 *bap++ = blkno;
  326         }
  327         /*
  328          * Next we must write out the modified inode and indirect blocks.
  329          * For strict correctness, the writes should be synchronous since
  330          * the old block values may have been written to disk. In practise
  331          * they are almost never written, but if we are concerned about
  332          * strict correctness, the `doasyncfree' flag should be set to zero.
  333          *
  334          * The test on `doasyncfree' should be changed to test a flag
  335          * that shows whether the associated buffers and inodes have
  336          * been written. The flag should be set when the cluster is
  337          * started and cleared whenever the buffer or inode is flushed.
  338          * We can then check below to see if it is set, and do the
  339          * synchronous write only when it has been cleared.
  340          */
  341         if (sbap != &ip->i_db[0]) {
  342                 if (doasyncfree)
  343                         bdwrite(sbp);
  344                 else
  345                         bwrite(sbp);
  346         } else {
  347                 ip->i_flag |= IN_CHANGE | IN_UPDATE;
  348                 if (!doasyncfree)
  349                         ext2_update(vp, 1);
  350         }
  351         if (ssize < len) {
  352                 if (doasyncfree)
  353                         bdwrite(ebp);
  354                 else
  355                         bwrite(ebp);
  356         }
  357         /*
  358          * Last, free the old blocks and assign the new blocks to the buffers.
  359          */
  360         for (blkno = newblk, i = 0; i < len; i++, blkno += fs->e2fs_fpb) {
  361                 ext2_blkfree(ip, dbtofsb(fs, buflist->bs_children[i]->b_blkno),
  362                     fs->e2fs_bsize);
  363                 buflist->bs_children[i]->b_blkno = fsbtodb(fs, blkno);
  364                 SDT_PROBE1(ext2fs, , alloc, ext2_reallocblks_blkno, blkno);
  365         }
  366 
  367         return (0);
  368 
  369 fail:
  370         if (ssize < len)
  371                 brelse(ebp);
  372         if (sbap != &ip->i_db[0])
  373                 brelse(sbp);
  374         return (ENOSPC);
  375 }
  376 
  377 /*
  378  * Allocate an inode in the filesystem.
  379  *
  380  */
  381 int
  382 ext2_valloc(struct vnode *pvp, int mode, struct ucred *cred, struct vnode **vpp)
  383 {
  384         struct timespec ts;
  385         struct m_ext2fs *fs;
  386         struct ext2mount *ump;
  387         struct inode *pip;
  388         struct inode *ip;
  389         struct vnode *vp;
  390         struct thread *td;
  391         ino_t ino, ipref;
  392         int error, cg;
  393 
  394         *vpp = NULL;
  395         pip = VTOI(pvp);
  396         fs = pip->i_e2fs;
  397         ump = pip->i_ump;
  398 
  399         EXT2_LOCK(ump);
  400         if (fs->e2fs_ficount == 0)
  401                 goto noinodes;
  402         /*
  403          * If it is a directory then obtain a cylinder group based on
  404          * ext2_dirpref else obtain it using ino_to_cg. The preferred inode is
  405          * always the next inode.
  406          */
  407         if ((mode & IFMT) == IFDIR) {
  408                 cg = ext2_dirpref(pip);
  409                 if (fs->e2fs_contigdirs[cg] < 255)
  410                         fs->e2fs_contigdirs[cg]++;
  411         } else {
  412                 cg = ino_to_cg(fs, pip->i_number);
  413                 if (fs->e2fs_contigdirs[cg] > 0)
  414                         fs->e2fs_contigdirs[cg]--;
  415         }
  416         ipref = cg * fs->e2fs_ipg + 1;
  417         ino = (ino_t)ext2_hashalloc(pip, cg, (long)ipref, mode, ext2_nodealloccg);
  418         if (ino == 0)
  419                 goto noinodes;
  420 
  421         td = curthread;
  422         error = vfs_hash_get(ump->um_mountp, ino, LK_EXCLUSIVE, td, vpp, NULL, NULL);
  423         if (error || *vpp != NULL) {
  424                 return (error);
  425         }
  426 
  427         ip = malloc(sizeof(struct inode), M_EXT2NODE, M_WAITOK | M_ZERO);
  428 
  429         /* Allocate a new vnode/inode. */
  430         if ((error = getnewvnode("ext2fs", ump->um_mountp, &ext2_vnodeops, &vp)) != 0) {
  431                 free(ip, M_EXT2NODE);
  432                 return (error);
  433         }
  434 
  435         lockmgr(vp->v_vnlock, LK_EXCLUSIVE, NULL);
  436         vp->v_data = ip;
  437         ip->i_vnode = vp;
  438         ip->i_e2fs = fs = ump->um_e2fs;
  439         ip->i_ump = ump;
  440         ip->i_number = ino;
  441         ip->i_block_group = ino_to_cg(fs, ino);
  442         ip->i_next_alloc_block = 0;
  443         ip->i_next_alloc_goal = 0;
  444 
  445         error = insmntque(vp, ump->um_mountp);
  446         if (error) {
  447                 free(ip, M_EXT2NODE);
  448                 return (error);
  449         }
  450 
  451         error = vfs_hash_insert(vp, ino, LK_EXCLUSIVE, td, vpp, NULL, NULL);
  452         if (error || *vpp != NULL) {
  453                 *vpp = NULL;
  454                 free(ip, M_EXT2NODE);
  455                 return (error);
  456         }
  457 
  458         if ((error = ext2_vinit(ump->um_mountp, &ext2_fifoops, &vp)) != 0) {
  459                 vput(vp);
  460                 *vpp = NULL;
  461                 free(ip, M_EXT2NODE);
  462                 return (error);
  463         }
  464 
  465         if (EXT2_HAS_INCOMPAT_FEATURE(fs, EXT2F_INCOMPAT_EXTENTS)
  466             && (S_ISREG(mode) || S_ISDIR(mode)))
  467                 ext4_ext_tree_init(ip);
  468         else
  469                 memset(ip->i_data, 0, sizeof(ip->i_data));
  470 
  471         /*
  472          * Set up a new generation number for this inode.
  473          * Avoid zero values.
  474          */
  475         do {
  476                 ip->i_gen = arc4random();
  477         } while (ip->i_gen == 0);
  478 
  479         vfs_timestamp(&ts);
  480         ip->i_birthtime = ts.tv_sec;
  481         ip->i_birthnsec = ts.tv_nsec;
  482 
  483         vn_set_state(vp, VSTATE_CONSTRUCTED);
  484         *vpp = vp;
  485 
  486         return (0);
  487 
  488 noinodes:
  489         EXT2_UNLOCK(ump);
  490         SDT_PROBE2(ext2fs, , alloc, trace, 1, "out of inodes");
  491         return (ENOSPC);
  492 }
  493 
  494 /*
  495  * 64-bit compatible getters and setters for struct ext2_gd from ext2fs.h
  496  */
  497 uint64_t
  498 e2fs_gd_get_b_bitmap(struct ext2_gd *gd)
  499 {
  500 
  501         return (((uint64_t)(le32toh(gd->ext4bgd_b_bitmap_hi)) << 32) |
  502             le32toh(gd->ext2bgd_b_bitmap));
  503 }
  504 
  505 uint64_t
  506 e2fs_gd_get_i_bitmap(struct ext2_gd *gd)
  507 {
  508 
  509         return (((uint64_t)(le32toh(gd->ext4bgd_i_bitmap_hi)) << 32) |
  510             le32toh(gd->ext2bgd_i_bitmap));
  511 }
  512 
  513 uint64_t
  514 e2fs_gd_get_i_tables(struct ext2_gd *gd)
  515 {
  516 
  517         return (((uint64_t)(le32toh(gd->ext4bgd_i_tables_hi)) << 32) |
  518             le32toh(gd->ext2bgd_i_tables));
  519 }
  520 
  521 static uint32_t
  522 e2fs_gd_get_nbfree(struct ext2_gd *gd)
  523 {
  524 
  525         return (((uint32_t)(le16toh(gd->ext4bgd_nbfree_hi)) << 16) |
  526             le16toh(gd->ext2bgd_nbfree));
  527 }
  528 
  529 static void
  530 e2fs_gd_set_nbfree(struct ext2_gd *gd, uint32_t val)
  531 {
  532 
  533         gd->ext2bgd_nbfree = htole16(val & 0xffff);
  534         gd->ext4bgd_nbfree_hi = htole16(val >> 16);
  535 }
  536 
  537 static uint32_t
  538 e2fs_gd_get_nifree(struct ext2_gd *gd)
  539 {
  540 
  541         return (((uint32_t)(le16toh(gd->ext4bgd_nifree_hi)) << 16) |
  542             le16toh(gd->ext2bgd_nifree));
  543 }
  544 
  545 static void
  546 e2fs_gd_set_nifree(struct ext2_gd *gd, uint32_t val)
  547 {
  548 
  549         gd->ext2bgd_nifree = htole16(val & 0xffff);
  550         gd->ext4bgd_nifree_hi = htole16(val >> 16);
  551 }
  552 
  553 uint32_t
  554 e2fs_gd_get_ndirs(struct ext2_gd *gd)
  555 {
  556 
  557         return (((uint32_t)(le16toh(gd->ext4bgd_ndirs_hi)) << 16) |
  558             le16toh(gd->ext2bgd_ndirs));
  559 }
  560 
  561 static void
  562 e2fs_gd_set_ndirs(struct ext2_gd *gd, uint32_t val)
  563 {
  564 
  565         gd->ext2bgd_ndirs = htole16(val & 0xffff);
  566         gd->ext4bgd_ndirs_hi = htole16(val >> 16);
  567 }
  568 
  569 static uint32_t
  570 e2fs_gd_get_i_unused(struct ext2_gd *gd)
  571 {
  572         return ((uint32_t)(le16toh(gd->ext4bgd_i_unused_hi) << 16) |
  573             le16toh(gd->ext4bgd_i_unused));
  574 }
  575 
  576 static void
  577 e2fs_gd_set_i_unused(struct ext2_gd *gd, uint32_t val)
  578 {
  579 
  580         gd->ext4bgd_i_unused = htole16(val & 0xffff);
  581         gd->ext4bgd_i_unused_hi = htole16(val >> 16);
  582 }
  583 
  584 /*
  585  * Find a cylinder to place a directory.
  586  *
  587  * The policy implemented by this algorithm is to allocate a
  588  * directory inode in the same cylinder group as its parent
  589  * directory, but also to reserve space for its files inodes
  590  * and data. Restrict the number of directories which may be
  591  * allocated one after another in the same cylinder group
  592  * without intervening allocation of files.
  593  *
  594  * If we allocate a first level directory then force allocation
  595  * in another cylinder group.
  596  *
  597  */
  598 static u_long
  599 ext2_dirpref(struct inode *pip)
  600 {
  601         struct m_ext2fs *fs;
  602         int cg, prefcg, cgsize;
  603         uint64_t avgbfree, minbfree;
  604         u_int avgifree, avgndir, curdirsize;
  605         u_int minifree, maxndir;
  606         u_int mincg, minndir;
  607         u_int dirsize, maxcontigdirs;
  608 
  609         mtx_assert(EXT2_MTX(pip->i_ump), MA_OWNED);
  610         fs = pip->i_e2fs;
  611 
  612         avgifree = fs->e2fs_ficount / fs->e2fs_gcount;
  613         avgbfree = fs->e2fs_fbcount / fs->e2fs_gcount;
  614         avgndir = fs->e2fs_total_dir / fs->e2fs_gcount;
  615 
  616         /*
  617          * Force allocation in another cg if creating a first level dir.
  618          */
  619         ASSERT_VOP_LOCKED(ITOV(pip), "ext2fs_dirpref");
  620         if (ITOV(pip)->v_vflag & VV_ROOT) {
  621                 prefcg = arc4random() % fs->e2fs_gcount;
  622                 mincg = prefcg;
  623                 minndir = fs->e2fs_ipg;
  624                 for (cg = prefcg; cg < fs->e2fs_gcount; cg++)
  625                         if (e2fs_gd_get_ndirs(&fs->e2fs_gd[cg]) < minndir &&
  626                             e2fs_gd_get_nifree(&fs->e2fs_gd[cg]) >= avgifree &&
  627                             e2fs_gd_get_nbfree(&fs->e2fs_gd[cg]) >= avgbfree) {
  628                                 mincg = cg;
  629                                 minndir = e2fs_gd_get_ndirs(&fs->e2fs_gd[cg]);
  630                         }
  631                 for (cg = 0; cg < prefcg; cg++)
  632                         if (e2fs_gd_get_ndirs(&fs->e2fs_gd[cg]) < minndir &&
  633                             e2fs_gd_get_nifree(&fs->e2fs_gd[cg]) >= avgifree &&
  634                             e2fs_gd_get_nbfree(&fs->e2fs_gd[cg]) >= avgbfree) {
  635                                 mincg = cg;
  636                                 minndir = e2fs_gd_get_ndirs(&fs->e2fs_gd[cg]);
  637                         }
  638                 return (mincg);
  639         }
  640         /*
  641          * Count various limits which used for
  642          * optimal allocation of a directory inode.
  643          */
  644         maxndir = min(avgndir + fs->e2fs_ipg / 16, fs->e2fs_ipg);
  645         minifree = avgifree - avgifree / 4;
  646         if (minifree < 1)
  647                 minifree = 1;
  648         minbfree = avgbfree - avgbfree / 4;
  649         if (minbfree < 1)
  650                 minbfree = 1;
  651         cgsize = fs->e2fs_fsize * fs->e2fs_fpg;
  652         dirsize = AVGDIRSIZE;
  653         curdirsize = avgndir ?
  654             (cgsize - avgbfree * fs->e2fs_bsize) / avgndir : 0;
  655         if (dirsize < curdirsize)
  656                 dirsize = curdirsize;
  657         maxcontigdirs = min((avgbfree * fs->e2fs_bsize) / dirsize, 255);
  658         maxcontigdirs = min(maxcontigdirs, fs->e2fs_ipg / AFPDIR);
  659         if (maxcontigdirs == 0)
  660                 maxcontigdirs = 1;
  661 
  662         /*
  663          * Limit number of dirs in one cg and reserve space for
  664          * regular files, but only if we have no deficit in
  665          * inodes or space.
  666          */
  667         prefcg = ino_to_cg(fs, pip->i_number);
  668         for (cg = prefcg; cg < fs->e2fs_gcount; cg++)
  669                 if (e2fs_gd_get_ndirs(&fs->e2fs_gd[cg]) < maxndir &&
  670                     e2fs_gd_get_nifree(&fs->e2fs_gd[cg]) >= minifree &&
  671                     e2fs_gd_get_nbfree(&fs->e2fs_gd[cg]) >= minbfree) {
  672                         if (fs->e2fs_contigdirs[cg] < maxcontigdirs)
  673                                 return (cg);
  674                 }
  675         for (cg = 0; cg < prefcg; cg++)
  676                 if (e2fs_gd_get_ndirs(&fs->e2fs_gd[cg]) < maxndir &&
  677                     e2fs_gd_get_nifree(&fs->e2fs_gd[cg]) >= minifree &&
  678                     e2fs_gd_get_nbfree(&fs->e2fs_gd[cg]) >= minbfree) {
  679                         if (fs->e2fs_contigdirs[cg] < maxcontigdirs)
  680                                 return (cg);
  681                 }
  682         /*
  683          * This is a backstop when we have deficit in space.
  684          */
  685         for (cg = prefcg; cg < fs->e2fs_gcount; cg++)
  686                 if (e2fs_gd_get_nifree(&fs->e2fs_gd[cg]) >= avgifree)
  687                         return (cg);
  688         for (cg = 0; cg < prefcg; cg++)
  689                 if (e2fs_gd_get_nifree(&fs->e2fs_gd[cg]) >= avgifree)
  690                         break;
  691         return (cg);
  692 }
  693 
  694 /*
  695  * Select the desired position for the next block in a file.
  696  *
  697  * we try to mimic what Remy does in inode_getblk/block_getblk
  698  *
  699  * we note: blocknr == 0 means that we're about to allocate either
  700  * a direct block or a pointer block at the first level of indirection
  701  * (In other words, stuff that will go in i_db[] or i_ib[])
  702  *
  703  * blocknr != 0 means that we're allocating a block that is none
  704  * of the above. Then, blocknr tells us the number of the block
  705  * that will hold the pointer
  706  */
  707 e4fs_daddr_t
  708 ext2_blkpref(struct inode *ip, e2fs_lbn_t lbn, int indx, e2fs_daddr_t *bap,
  709     e2fs_daddr_t blocknr)
  710 {
  711         struct m_ext2fs *fs;
  712         int tmp;
  713 
  714         fs = ip->i_e2fs;
  715 
  716         mtx_assert(EXT2_MTX(ip->i_ump), MA_OWNED);
  717 
  718         /*
  719          * If the next block is actually what we thought it is, then set the
  720          * goal to what we thought it should be.
  721          */
  722         if (ip->i_next_alloc_block == lbn && ip->i_next_alloc_goal != 0)
  723                 return ip->i_next_alloc_goal;
  724 
  725         /*
  726          * Now check whether we were provided with an array that basically
  727          * tells us previous blocks to which we want to stay close.
  728          */
  729         if (bap)
  730                 for (tmp = indx - 1; tmp >= 0; tmp--)
  731                         if (bap[tmp])
  732                                 return (le32toh(bap[tmp]));
  733 
  734         /*
  735          * Else lets fall back to the blocknr or, if there is none, follow
  736          * the rule that a block should be allocated near its inode.
  737          */
  738         return (blocknr ? blocknr :
  739             (e2fs_daddr_t)(ip->i_block_group *
  740             EXT2_BLOCKS_PER_GROUP(fs)) + le32toh(fs->e2fs->e2fs_first_dblock));
  741 }
  742 
  743 /*
  744  * Implement the cylinder overflow algorithm.
  745  *
  746  * The policy implemented by this algorithm is:
  747  *   1) allocate the block in its requested cylinder group.
  748  *   2) quadratically rehash on the cylinder group number.
  749  *   3) brute force search for a free block.
  750  */
  751 static e4fs_daddr_t
  752 ext2_hashalloc(struct inode *ip, int cg, long pref, int size,
  753     daddr_t (*allocator) (struct inode *, int, daddr_t, int))
  754 {
  755         struct m_ext2fs *fs;
  756         e4fs_daddr_t result;
  757         int i, icg = cg;
  758 
  759         mtx_assert(EXT2_MTX(ip->i_ump), MA_OWNED);
  760         fs = ip->i_e2fs;
  761         /*
  762          * 1: preferred cylinder group
  763          */
  764         result = (*allocator)(ip, cg, pref, size);
  765         if (result)
  766                 return (result);
  767         /*
  768          * 2: quadratic rehash
  769          */
  770         for (i = 1; i < fs->e2fs_gcount; i *= 2) {
  771                 cg += i;
  772                 if (cg >= fs->e2fs_gcount)
  773                         cg -= fs->e2fs_gcount;
  774                 result = (*allocator)(ip, cg, 0, size);
  775                 if (result)
  776                         return (result);
  777         }
  778         /*
  779          * 3: brute force search
  780          * Note that we start at i == 2, since 0 was checked initially,
  781          * and 1 is always checked in the quadratic rehash.
  782          */
  783         cg = (icg + 2) % fs->e2fs_gcount;
  784         for (i = 2; i < fs->e2fs_gcount; i++) {
  785                 result = (*allocator)(ip, cg, 0, size);
  786                 if (result)
  787                         return (result);
  788                 cg++;
  789                 if (cg == fs->e2fs_gcount)
  790                         cg = 0;
  791         }
  792         return (0);
  793 }
  794 
  795 static uint64_t
  796 ext2_cg_number_gdb_nometa(struct m_ext2fs *fs, int cg)
  797 {
  798 
  799         if (!ext2_cg_has_sb(fs, cg))
  800                 return (0);
  801 
  802         if (EXT2_HAS_INCOMPAT_FEATURE(fs, EXT2F_INCOMPAT_META_BG))
  803                 return (le32toh(fs->e2fs->e3fs_first_meta_bg));
  804 
  805         return ((fs->e2fs_gcount + EXT2_DESCS_PER_BLOCK(fs) - 1) /
  806             EXT2_DESCS_PER_BLOCK(fs));
  807 }
  808 
  809 static uint64_t
  810 ext2_cg_number_gdb_meta(struct m_ext2fs *fs, int cg)
  811 {
  812         unsigned long metagroup;
  813         int first, last;
  814 
  815         metagroup = cg / EXT2_DESCS_PER_BLOCK(fs);
  816         first = metagroup * EXT2_DESCS_PER_BLOCK(fs);
  817         last = first + EXT2_DESCS_PER_BLOCK(fs) - 1;
  818 
  819         if (cg == first || cg == first + 1 || cg == last)
  820                 return (1);
  821 
  822         return (0);
  823 }
  824 
  825 uint64_t
  826 ext2_cg_number_gdb(struct m_ext2fs *fs, int cg)
  827 {
  828         unsigned long first_meta_bg, metagroup;
  829 
  830         first_meta_bg = le32toh(fs->e2fs->e3fs_first_meta_bg);
  831         metagroup = cg / EXT2_DESCS_PER_BLOCK(fs);
  832 
  833         if (!EXT2_HAS_INCOMPAT_FEATURE(fs, EXT2F_INCOMPAT_META_BG) ||
  834             metagroup < first_meta_bg)
  835                 return (ext2_cg_number_gdb_nometa(fs, cg));
  836 
  837         return ext2_cg_number_gdb_meta(fs, cg);
  838 }
  839 
  840 static int
  841 ext2_number_base_meta_blocks(struct m_ext2fs *fs, int cg)
  842 {
  843         int number;
  844 
  845         number = ext2_cg_has_sb(fs, cg);
  846 
  847         if (!EXT2_HAS_INCOMPAT_FEATURE(fs, EXT2F_INCOMPAT_META_BG) ||
  848             cg < le32toh(fs->e2fs->e3fs_first_meta_bg) *
  849             EXT2_DESCS_PER_BLOCK(fs)) {
  850                 if (number) {
  851                         number += ext2_cg_number_gdb(fs, cg);
  852                         number += le16toh(fs->e2fs->e2fs_reserved_ngdb);
  853                 }
  854         } else {
  855                 number += ext2_cg_number_gdb(fs, cg);
  856         }
  857 
  858         return (number);
  859 }
  860 
  861 static void
  862 ext2_mark_bitmap_end(int start_bit, int end_bit, char *bitmap)
  863 {
  864         int i;
  865 
  866         if (start_bit >= end_bit)
  867                 return;
  868 
  869         for (i = start_bit; i < ((start_bit + 7) & ~7UL); i++)
  870                 setbit(bitmap, i);
  871         if (i < end_bit)
  872                 memset(bitmap + (i >> 3), 0xff, (end_bit - i) >> 3);
  873 }
  874 
  875 static int
  876 ext2_get_group_number(struct m_ext2fs *fs, e4fs_daddr_t block)
  877 {
  878 
  879         return ((block - le32toh(fs->e2fs->e2fs_first_dblock)) /
  880             fs->e2fs_bsize);
  881 }
  882 
  883 static int
  884 ext2_block_in_group(struct m_ext2fs *fs, e4fs_daddr_t block, int cg)
  885 {
  886 
  887         return ((ext2_get_group_number(fs, block) == cg) ? 1 : 0);
  888 }
  889 
  890 static int
  891 ext2_cg_block_bitmap_init(struct m_ext2fs *fs, int cg, struct buf *bp)
  892 {
  893         int bit, bit_max, inodes_per_block;
  894         uint64_t start, tmp;
  895 
  896         if (!(le16toh(fs->e2fs_gd[cg].ext4bgd_flags) & EXT2_BG_BLOCK_UNINIT))
  897                 return (0);
  898 
  899         memset(bp->b_data, 0, fs->e2fs_bsize);
  900 
  901         bit_max = ext2_number_base_meta_blocks(fs, cg);
  902         if ((bit_max >> 3) >= fs->e2fs_bsize)
  903                 return (EINVAL);
  904 
  905         for (bit = 0; bit < bit_max; bit++)
  906                 setbit(bp->b_data, bit);
  907 
  908         start = (uint64_t)cg * fs->e2fs_bpg +
  909             le32toh(fs->e2fs->e2fs_first_dblock);
  910 
  911         /* Set bits for block and inode bitmaps, and inode table. */
  912         tmp = e2fs_gd_get_b_bitmap(&fs->e2fs_gd[cg]);
  913         if (!EXT2_HAS_INCOMPAT_FEATURE(fs, EXT2F_INCOMPAT_FLEX_BG) ||
  914             ext2_block_in_group(fs, tmp, cg))
  915                 setbit(bp->b_data, tmp - start);
  916 
  917         tmp = e2fs_gd_get_i_bitmap(&fs->e2fs_gd[cg]);
  918         if (!EXT2_HAS_INCOMPAT_FEATURE(fs, EXT2F_INCOMPAT_FLEX_BG) ||
  919             ext2_block_in_group(fs, tmp, cg))
  920                 setbit(bp->b_data, tmp - start);
  921 
  922         tmp = e2fs_gd_get_i_tables(&fs->e2fs_gd[cg]);
  923         inodes_per_block = fs->e2fs_bsize/EXT2_INODE_SIZE(fs);
  924         while( tmp < e2fs_gd_get_i_tables(&fs->e2fs_gd[cg]) +
  925             fs->e2fs_ipg / inodes_per_block ) {
  926                 if (!EXT2_HAS_INCOMPAT_FEATURE(fs, EXT2F_INCOMPAT_FLEX_BG) ||
  927                     ext2_block_in_group(fs, tmp, cg))
  928                         setbit(bp->b_data, tmp - start);
  929                 tmp++;
  930         }
  931 
  932         /*
  933          * Also if the number of blocks within the group is less than
  934          * the blocksize * 8 ( which is the size of bitmap ), set rest
  935          * of the block bitmap to 1
  936          */
  937         ext2_mark_bitmap_end(fs->e2fs_bpg, fs->e2fs_bsize * 8,
  938             bp->b_data);
  939 
  940         /* Clean the flag */
  941         fs->e2fs_gd[cg].ext4bgd_flags = htole16(le16toh(
  942             fs->e2fs_gd[cg].ext4bgd_flags) & ~EXT2_BG_BLOCK_UNINIT);
  943 
  944         return (0);
  945 }
  946 
  947 static int
  948 ext2_b_bitmap_validate(struct m_ext2fs *fs, struct buf *bp, int cg)
  949 {
  950         struct ext2_gd *gd;
  951         uint64_t group_first_block;
  952         unsigned int offset, max_bit;
  953 
  954         if (EXT2_HAS_INCOMPAT_FEATURE(fs, EXT2F_INCOMPAT_FLEX_BG)) {
  955                 /*
  956                  * It is not possible to check block bitmap in case of this
  957                  * feature, because the inode and block bitmaps and inode table
  958                  * blocks may not be in the group at all.
  959                  * So, skip check in this case.
  960                  */
  961                 return (0);
  962         }
  963 
  964         gd = &fs->e2fs_gd[cg];
  965         max_bit = fs->e2fs_fpg;
  966         group_first_block = ((uint64_t)cg) * fs->e2fs_fpg +
  967             le32toh(fs->e2fs->e2fs_first_dblock);
  968 
  969         /* Check block bitmap block number */
  970         offset = e2fs_gd_get_b_bitmap(gd) - group_first_block;
  971         if (offset >= max_bit || !isset(bp->b_data, offset)) {
  972                 SDT_PROBE2(ext2fs, , alloc, ext2_b_bitmap_validate_error,
  973                     "bad block bitmap, group", cg);
  974                 return (EINVAL);
  975         }
  976 
  977         /* Check inode bitmap block number */
  978         offset = e2fs_gd_get_i_bitmap(gd) - group_first_block;
  979         if (offset >= max_bit || !isset(bp->b_data, offset)) {
  980                 SDT_PROBE2(ext2fs, , alloc, ext2_b_bitmap_validate_error,
  981                     "bad inode bitmap", cg);
  982                 return (EINVAL);
  983         }
  984 
  985         /* Check inode table */
  986         offset = e2fs_gd_get_i_tables(gd) - group_first_block;
  987         if (offset >= max_bit || offset + fs->e2fs_itpg >= max_bit) {
  988                 SDT_PROBE2(ext2fs, , alloc, ext2_b_bitmap_validate_error,
  989                     "bad inode table, group", cg);
  990                 return (EINVAL);
  991         }
  992 
  993         return (0);
  994 }
  995 
  996 /*
  997  * Determine whether a block can be allocated.
  998  *
  999  * Check to see if a block of the appropriate size is available,
 1000  * and if it is, allocate it.
 1001  */
 1002 static daddr_t
 1003 ext2_alloccg(struct inode *ip, int cg, daddr_t bpref, int size)
 1004 {
 1005         struct m_ext2fs *fs;
 1006         struct buf *bp;
 1007         struct ext2mount *ump;
 1008         daddr_t bno, runstart, runlen;
 1009         int bit, loc, end, error, start;
 1010         char *bbp;
 1011         /* XXX ondisk32 */
 1012         fs = ip->i_e2fs;
 1013         ump = ip->i_ump;
 1014         if (e2fs_gd_get_nbfree(&fs->e2fs_gd[cg]) == 0)
 1015                 return (0);
 1016 
 1017         EXT2_UNLOCK(ump);
 1018         error = bread(ip->i_devvp, fsbtodb(fs,
 1019             e2fs_gd_get_b_bitmap(&fs->e2fs_gd[cg])),
 1020             (int)fs->e2fs_bsize, NOCRED, &bp);
 1021         if (error)
 1022                 goto fail;
 1023 
 1024         if (EXT2_HAS_RO_COMPAT_FEATURE(fs, EXT2F_ROCOMPAT_GDT_CSUM) ||
 1025             EXT2_HAS_RO_COMPAT_FEATURE(fs, EXT2F_ROCOMPAT_METADATA_CKSUM)) {
 1026                 error = ext2_cg_block_bitmap_init(fs, cg, bp);
 1027                 if (error)
 1028                         goto fail;
 1029 
 1030                 ext2_gd_b_bitmap_csum_set(fs, cg, bp);
 1031         }
 1032         error = ext2_gd_b_bitmap_csum_verify(fs, cg, bp);
 1033         if (error)
 1034                 goto fail;
 1035 
 1036         error = ext2_b_bitmap_validate(fs,bp, cg);
 1037         if (error)
 1038                 goto fail;
 1039 
 1040         /*
 1041          * Check, that another thread did not not allocate the last block in
 1042          * this group while we were waiting for the buffer.
 1043          */
 1044         if (e2fs_gd_get_nbfree(&fs->e2fs_gd[cg]) == 0)
 1045                 goto fail;
 1046 
 1047         bbp = (char *)bp->b_data;
 1048 
 1049         if (dtog(fs, bpref) != cg)
 1050                 bpref = 0;
 1051         if (bpref != 0) {
 1052                 bpref = dtogd(fs, bpref);
 1053                 /*
 1054                  * if the requested block is available, use it
 1055                  */
 1056                 if (isclr(bbp, bpref)) {
 1057                         bno = bpref;
 1058                         goto gotit;
 1059                 }
 1060         }
 1061         /*
 1062          * no blocks in the requested cylinder, so take next
 1063          * available one in this cylinder group.
 1064          * first try to get 8 contigous blocks, then fall back to a single
 1065          * block.
 1066          */
 1067         if (bpref)
 1068                 start = dtogd(fs, bpref) / NBBY;
 1069         else
 1070                 start = 0;
 1071         end = howmany(fs->e2fs_fpg, NBBY) - start;
 1072 retry:
 1073         runlen = 0;
 1074         runstart = 0;
 1075         for (loc = start; loc < end; loc++) {
 1076                 if (bbp[loc] == (char)0xff) {
 1077                         runlen = 0;
 1078                         continue;
 1079                 }
 1080 
 1081                 /* Start of a run, find the number of high clear bits. */
 1082                 if (runlen == 0) {
 1083                         bit = fls(bbp[loc]);
 1084                         runlen = NBBY - bit;
 1085                         runstart = loc * NBBY + bit;
 1086                 } else if (bbp[loc] == 0) {
 1087                         /* Continue a run. */
 1088                         runlen += NBBY;
 1089                 } else {
 1090                         /*
 1091                          * Finish the current run.  If it isn't long
 1092                          * enough, start a new one.
 1093                          */
 1094                         bit = ffs(bbp[loc]) - 1;
 1095                         runlen += bit;
 1096                         if (runlen >= 8) {
 1097                                 bno = runstart;
 1098                                 goto gotit;
 1099                         }
 1100 
 1101                         /* Run was too short, start a new one. */
 1102                         bit = fls(bbp[loc]);
 1103                         runlen = NBBY - bit;
 1104                         runstart = loc * NBBY + bit;
 1105                 }
 1106 
 1107                 /* If the current run is long enough, use it. */
 1108                 if (runlen >= 8) {
 1109                         bno = runstart;
 1110                         goto gotit;
 1111                 }
 1112         }
 1113         if (start != 0) {
 1114                 end = start;
 1115                 start = 0;
 1116                 goto retry;
 1117         }
 1118         bno = ext2_mapsearch(fs, bbp, bpref);
 1119         if (bno < 0)
 1120                 goto fail;
 1121 
 1122 gotit:
 1123 #ifdef INVARIANTS
 1124         if (isset(bbp, bno)) {
 1125                 printf("ext2fs_alloccgblk: cg=%d bno=%jd fs=%s\n",
 1126                     cg, (intmax_t)bno, fs->e2fs_fsmnt);
 1127                 panic("ext2fs_alloccg: dup alloc");
 1128         }
 1129 #endif
 1130         setbit(bbp, bno);
 1131         EXT2_LOCK(ump);
 1132         ext2_clusteracct(fs, bbp, cg, bno, -1);
 1133         fs->e2fs_fbcount--;
 1134         e2fs_gd_set_nbfree(&fs->e2fs_gd[cg],
 1135             e2fs_gd_get_nbfree(&fs->e2fs_gd[cg]) - 1);
 1136         fs->e2fs_fmod = 1;
 1137         EXT2_UNLOCK(ump);
 1138         ext2_gd_b_bitmap_csum_set(fs, cg, bp);
 1139         bdwrite(bp);
 1140         return (((uint64_t)cg) * fs->e2fs_fpg +
 1141             le32toh(fs->e2fs->e2fs_first_dblock) + bno);
 1142 
 1143 fail:
 1144         brelse(bp);
 1145         EXT2_LOCK(ump);
 1146         return (0);
 1147 }
 1148 
 1149 /*
 1150  * Determine whether a cluster can be allocated.
 1151  */
 1152 static daddr_t
 1153 ext2_clusteralloc(struct inode *ip, int cg, daddr_t bpref, int len)
 1154 {
 1155         struct m_ext2fs *fs;
 1156         struct ext2mount *ump;
 1157         struct buf *bp;
 1158         char *bbp;
 1159         int bit, error, got, i, loc, run;
 1160         int32_t *lp;
 1161         daddr_t bno;
 1162 
 1163         fs = ip->i_e2fs;
 1164         ump = ip->i_ump;
 1165 
 1166         if (fs->e2fs_maxcluster[cg] < len)
 1167                 return (0);
 1168 
 1169         EXT2_UNLOCK(ump);
 1170         error = bread(ip->i_devvp,
 1171             fsbtodb(fs, e2fs_gd_get_b_bitmap(&fs->e2fs_gd[cg])),
 1172             (int)fs->e2fs_bsize, NOCRED, &bp);
 1173         if (error)
 1174                 goto fail_lock;
 1175 
 1176         bbp = (char *)bp->b_data;
 1177         EXT2_LOCK(ump);
 1178         /*
 1179          * Check to see if a cluster of the needed size (or bigger) is
 1180          * available in this cylinder group.
 1181          */
 1182         lp = &fs->e2fs_clustersum[cg].cs_sum[len];
 1183         for (i = len; i <= fs->e2fs_contigsumsize; i++)
 1184                 if (*lp++ > 0)
 1185                         break;
 1186         if (i > fs->e2fs_contigsumsize) {
 1187                 /*
 1188                  * Update the cluster summary information to reflect
 1189                  * the true maximum-sized cluster so that future cluster
 1190                  * allocation requests can avoid reading the bitmap only
 1191                  * to find no cluster.
 1192                  */
 1193                 lp = &fs->e2fs_clustersum[cg].cs_sum[len - 1];
 1194                 for (i = len - 1; i > 0; i--)
 1195                         if (*lp-- > 0)
 1196                                 break;
 1197                 fs->e2fs_maxcluster[cg] = i;
 1198                 goto fail;
 1199         }
 1200         EXT2_UNLOCK(ump);
 1201 
 1202         /* Search the bitmap to find a big enough cluster like in FFS. */
 1203         if (dtog(fs, bpref) != cg)
 1204                 bpref = 0;
 1205         if (bpref != 0)
 1206                 bpref = dtogd(fs, bpref);
 1207         loc = bpref / NBBY;
 1208         bit = 1 << (bpref % NBBY);
 1209         for (run = 0, got = bpref; got < fs->e2fs_fpg; got++) {
 1210                 if ((bbp[loc] & bit) != 0)
 1211                         run = 0;
 1212                 else {
 1213                         run++;
 1214                         if (run == len)
 1215                                 break;
 1216                 }
 1217                 if ((got & (NBBY - 1)) != (NBBY - 1))
 1218                         bit <<= 1;
 1219                 else {
 1220                         loc++;
 1221                         bit = 1;
 1222                 }
 1223         }
 1224 
 1225         if (got >= fs->e2fs_fpg)
 1226                 goto fail_lock;
 1227 
 1228         /* Allocate the cluster that we found. */
 1229         for (i = 1; i < len; i++)
 1230                 if (!isclr(bbp, got - run + i))
 1231                         panic("ext2_clusteralloc: map mismatch");
 1232 
 1233         bno = got - run + 1;
 1234         if (bno >= fs->e2fs_fpg)
 1235                 panic("ext2_clusteralloc: allocated out of group");
 1236 
 1237         EXT2_LOCK(ump);
 1238         for (i = 0; i < len; i += fs->e2fs_fpb) {
 1239                 setbit(bbp, bno + i);
 1240                 ext2_clusteracct(fs, bbp, cg, bno + i, -1);
 1241                 fs->e2fs_fbcount--;
 1242                 e2fs_gd_set_nbfree(&fs->e2fs_gd[cg],
 1243                     e2fs_gd_get_nbfree(&fs->e2fs_gd[cg]) - 1);
 1244         }
 1245         fs->e2fs_fmod = 1;
 1246         EXT2_UNLOCK(ump);
 1247 
 1248         bdwrite(bp);
 1249         return (cg * fs->e2fs_fpg + le32toh(fs->e2fs->e2fs_first_dblock)
 1250             + bno);
 1251 
 1252 fail_lock:
 1253         EXT2_LOCK(ump);
 1254 fail:
 1255         brelse(bp);
 1256         return (0);
 1257 }
 1258 
 1259 static int
 1260 ext2_zero_inode_table(struct inode *ip, int cg)
 1261 {
 1262         struct m_ext2fs *fs;
 1263         struct buf *bp;
 1264         int i, all_blks, used_blks;
 1265 
 1266         fs = ip->i_e2fs;
 1267 
 1268         if (le16toh(fs->e2fs_gd[cg].ext4bgd_flags) & EXT2_BG_INODE_ZEROED)
 1269                 return (0);
 1270 
 1271         all_blks = le16toh(fs->e2fs->e2fs_inode_size) * fs->e2fs_ipg /
 1272             fs->e2fs_bsize;
 1273 
 1274         used_blks = howmany(fs->e2fs_ipg -
 1275             e2fs_gd_get_i_unused(&fs->e2fs_gd[cg]),
 1276             fs->e2fs_bsize / EXT2_INODE_SIZE(fs));
 1277 
 1278         for (i = 0; i < all_blks - used_blks; i++) {
 1279                 bp = getblk(ip->i_devvp, fsbtodb(fs,
 1280                     e2fs_gd_get_i_tables(&fs->e2fs_gd[cg]) + used_blks + i),
 1281                     fs->e2fs_bsize, 0, 0, 0);
 1282                 if (!bp)
 1283                         return (EIO);
 1284 
 1285                 vfs_bio_bzero_buf(bp, 0, fs->e2fs_bsize);
 1286                 bawrite(bp);
 1287         }
 1288 
 1289         fs->e2fs_gd[cg].ext4bgd_flags = htole16(le16toh(
 1290             fs->e2fs_gd[cg].ext4bgd_flags) | EXT2_BG_INODE_ZEROED);
 1291 
 1292         return (0);
 1293 }
 1294 
 1295 static void
 1296 ext2_fix_bitmap_tail(unsigned char *bitmap, int first, int last)
 1297 {
 1298         int i;
 1299 
 1300         for (i = first; i <= last; i++)
 1301                 bitmap[i] = 0xff;
 1302 }
 1303 
 1304 /*
 1305  * Determine whether an inode can be allocated.
 1306  *
 1307  * Check to see if an inode is available, and if it is,
 1308  * allocate it using tode in the specified cylinder group.
 1309  */
 1310 static daddr_t
 1311 ext2_nodealloccg(struct inode *ip, int cg, daddr_t ipref, int mode)
 1312 {
 1313         struct m_ext2fs *fs;
 1314         struct buf *bp;
 1315         struct ext2mount *ump;
 1316         int error, start, len, ifree, ibytes;
 1317         char *ibp, *loc;
 1318 
 1319         ipref--;        /* to avoid a lot of (ipref -1) */
 1320         if (ipref == -1)
 1321                 ipref = 0;
 1322         fs = ip->i_e2fs;
 1323         ump = ip->i_ump;
 1324         if (e2fs_gd_get_nifree(&fs->e2fs_gd[cg]) == 0)
 1325                 return (0);
 1326         EXT2_UNLOCK(ump);
 1327         error = bread(ip->i_devvp, fsbtodb(fs,
 1328             e2fs_gd_get_i_bitmap(&fs->e2fs_gd[cg])),
 1329             (int)fs->e2fs_bsize, NOCRED, &bp);
 1330         if (error) {
 1331                 EXT2_LOCK(ump);
 1332                 return (0);
 1333         }
 1334         if (EXT2_HAS_RO_COMPAT_FEATURE(fs, EXT2F_ROCOMPAT_GDT_CSUM) ||
 1335             EXT2_HAS_RO_COMPAT_FEATURE(fs, EXT2F_ROCOMPAT_METADATA_CKSUM)) {
 1336                 if (le16toh(fs->e2fs_gd[cg].ext4bgd_flags) &
 1337                     EXT2_BG_INODE_UNINIT) {
 1338                         ibytes = fs->e2fs_ipg / 8;
 1339                         memset(bp->b_data, 0, ibytes - 1);
 1340                         ext2_fix_bitmap_tail(bp->b_data, ibytes,
 1341                             fs->e2fs_bsize - 1);
 1342                         fs->e2fs_gd[cg].ext4bgd_flags = htole16(le16toh(
 1343                             fs->e2fs_gd[cg].ext4bgd_flags) &
 1344                             ~EXT2_BG_INODE_UNINIT);
 1345                 }
 1346                 ext2_gd_i_bitmap_csum_set(fs, cg, bp);
 1347                 error = ext2_zero_inode_table(ip, cg);
 1348                 if (error) {
 1349                         brelse(bp);
 1350                         EXT2_LOCK(ump);
 1351                         return (0);
 1352                 }
 1353         }
 1354         error = ext2_gd_i_bitmap_csum_verify(fs, cg, bp);
 1355         if (error) {
 1356                 brelse(bp);
 1357                 EXT2_LOCK(ump);
 1358                 return (0);
 1359         }
 1360         if (e2fs_gd_get_nifree(&fs->e2fs_gd[cg]) == 0) {
 1361                 /*
 1362                  * Another thread allocated the last i-node in this
 1363                  * group while we were waiting for the buffer.
 1364                  */
 1365                 brelse(bp);
 1366                 EXT2_LOCK(ump);
 1367                 return (0);
 1368         }
 1369         ibp = (char *)bp->b_data;
 1370         if (ipref) {
 1371                 ipref %= fs->e2fs_ipg;
 1372                 if (isclr(ibp, ipref))
 1373                         goto gotit;
 1374         }
 1375         start = ipref / NBBY;
 1376         len = howmany(fs->e2fs_ipg - ipref, NBBY);
 1377         loc = memcchr(&ibp[start], 0xff, len);
 1378         if (loc == NULL) {
 1379                 len = start + 1;
 1380                 start = 0;
 1381                 loc = memcchr(&ibp[start], 0xff, len);
 1382                 if (loc == NULL) {
 1383                         SDT_PROBE3(ext2fs, , alloc,
 1384                             ext2_nodealloccg_bmap_corrupted, cg, ipref,
 1385                             fs->e2fs_fsmnt);
 1386                         brelse(bp);
 1387                         EXT2_LOCK(ump);
 1388                         return (0);
 1389                 }
 1390         }
 1391         ipref = (loc - ibp) * NBBY + ffs(~*loc) - 1;
 1392 gotit:
 1393         setbit(ibp, ipref);
 1394         EXT2_LOCK(ump);
 1395         e2fs_gd_set_nifree(&fs->e2fs_gd[cg],
 1396             e2fs_gd_get_nifree(&fs->e2fs_gd[cg]) - 1);
 1397         if (EXT2_HAS_RO_COMPAT_FEATURE(fs, EXT2F_ROCOMPAT_GDT_CSUM) ||
 1398             EXT2_HAS_RO_COMPAT_FEATURE(fs, EXT2F_ROCOMPAT_METADATA_CKSUM)) {
 1399                 ifree = fs->e2fs_ipg - e2fs_gd_get_i_unused(&fs->e2fs_gd[cg]);
 1400                 if (ipref + 1 > ifree)
 1401                         e2fs_gd_set_i_unused(&fs->e2fs_gd[cg],
 1402                             fs->e2fs_ipg - (ipref + 1));
 1403         }
 1404         fs->e2fs_ficount--;
 1405         fs->e2fs_fmod = 1;
 1406         if ((mode & IFMT) == IFDIR) {
 1407                 e2fs_gd_set_ndirs(&fs->e2fs_gd[cg],
 1408                     e2fs_gd_get_ndirs(&fs->e2fs_gd[cg]) + 1);
 1409                 fs->e2fs_total_dir++;
 1410         }
 1411         EXT2_UNLOCK(ump);
 1412         ext2_gd_i_bitmap_csum_set(fs, cg, bp);
 1413         bdwrite(bp);
 1414         return ((uint64_t)cg * fs->e2fs_ipg + ipref + 1);
 1415 }
 1416 
 1417 /*
 1418  * Free a block or fragment.
 1419  *
 1420  */
 1421 void
 1422 ext2_blkfree(struct inode *ip, e4fs_daddr_t bno, long size)
 1423 {
 1424         struct m_ext2fs *fs;
 1425         struct buf *bp;
 1426         struct ext2mount *ump;
 1427         int cg, error;
 1428         char *bbp;
 1429 
 1430         fs = ip->i_e2fs;
 1431         ump = ip->i_ump;
 1432         cg = dtog(fs, bno);
 1433         if (bno >= fs->e2fs_bcount) {
 1434                 SDT_PROBE2(ext2fs, , alloc, ext2_blkfree_bad_block,
 1435                     ip->i_number, bno);
 1436                 return;
 1437         }
 1438         error = bread(ip->i_devvp,
 1439             fsbtodb(fs, e2fs_gd_get_b_bitmap(&fs->e2fs_gd[cg])),
 1440             (int)fs->e2fs_bsize, NOCRED, &bp);
 1441         if (error) {
 1442                 return;
 1443         }
 1444         bbp = (char *)bp->b_data;
 1445         bno = dtogd(fs, bno);
 1446         if (isclr(bbp, bno)) {
 1447                 panic("ext2_blkfree: freeing free block %lld, fs=%s",
 1448                     (long long)bno, fs->e2fs_fsmnt);
 1449         }
 1450         clrbit(bbp, bno);
 1451         EXT2_LOCK(ump);
 1452         ext2_clusteracct(fs, bbp, cg, bno, 1);
 1453         fs->e2fs_fbcount++;
 1454         e2fs_gd_set_nbfree(&fs->e2fs_gd[cg],
 1455             e2fs_gd_get_nbfree(&fs->e2fs_gd[cg]) + 1);
 1456         fs->e2fs_fmod = 1;
 1457         EXT2_UNLOCK(ump);
 1458         ext2_gd_b_bitmap_csum_set(fs, cg, bp);
 1459         bdwrite(bp);
 1460 }
 1461 
 1462 /*
 1463  * Free an inode.
 1464  *
 1465  */
 1466 int
 1467 ext2_vfree(struct vnode *pvp, ino_t ino, int mode)
 1468 {
 1469         struct m_ext2fs *fs;
 1470         struct inode *pip;
 1471         struct buf *bp;
 1472         struct ext2mount *ump;
 1473         int error, cg;
 1474         char *ibp;
 1475 
 1476         pip = VTOI(pvp);
 1477         fs = pip->i_e2fs;
 1478         ump = pip->i_ump;
 1479         if ((u_int)ino > fs->e2fs_ipg * fs->e2fs_gcount)
 1480                 panic("ext2_vfree: range: devvp = %p, ino = %ju, fs = %s",
 1481                     pip->i_devvp, (uintmax_t)ino, fs->e2fs_fsmnt);
 1482 
 1483         cg = ino_to_cg(fs, ino);
 1484         error = bread(pip->i_devvp,
 1485             fsbtodb(fs, e2fs_gd_get_i_bitmap(&fs->e2fs_gd[cg])),
 1486             (int)fs->e2fs_bsize, NOCRED, &bp);
 1487         if (error) {
 1488                 return (0);
 1489         }
 1490         ibp = (char *)bp->b_data;
 1491         ino = (ino - 1) % fs->e2fs_ipg;
 1492         if (isclr(ibp, ino)) {
 1493                 SDT_PROBE2(ext2fs, , alloc, ext2_vfree_doublefree,
 1494                     fs->e2fs_fsmnt, ino);
 1495                 if (fs->e2fs_ronly == 0)
 1496                         panic("ext2_vfree: freeing free inode");
 1497         }
 1498         clrbit(ibp, ino);
 1499         EXT2_LOCK(ump);
 1500         fs->e2fs_ficount++;
 1501         e2fs_gd_set_nifree(&fs->e2fs_gd[cg],
 1502             e2fs_gd_get_nifree(&fs->e2fs_gd[cg]) + 1);
 1503         if ((mode & IFMT) == IFDIR) {
 1504                 e2fs_gd_set_ndirs(&fs->e2fs_gd[cg],
 1505                     e2fs_gd_get_ndirs(&fs->e2fs_gd[cg]) - 1);
 1506                 fs->e2fs_total_dir--;
 1507         }
 1508         fs->e2fs_fmod = 1;
 1509         EXT2_UNLOCK(ump);
 1510         ext2_gd_i_bitmap_csum_set(fs, cg, bp);
 1511         bdwrite(bp);
 1512         return (0);
 1513 }
 1514 
 1515 /*
 1516  * Find a block in the specified cylinder group.
 1517  *
 1518  * It is a panic if a request is made to find a block if none are
 1519  * available.
 1520  */
 1521 static daddr_t
 1522 ext2_mapsearch(struct m_ext2fs *fs, char *bbp, daddr_t bpref)
 1523 {
 1524         char *loc;
 1525         int start, len;
 1526 
 1527         /*
 1528          * find the fragment by searching through the free block
 1529          * map for an appropriate bit pattern
 1530          */
 1531         if (bpref)
 1532                 start = dtogd(fs, bpref) / NBBY;
 1533         else
 1534                 start = 0;
 1535         len = howmany(fs->e2fs_fpg, NBBY) - start;
 1536         loc = memcchr(&bbp[start], 0xff, len);
 1537         if (loc == NULL) {
 1538                 len = start + 1;
 1539                 start = 0;
 1540                 loc = memcchr(&bbp[start], 0xff, len);
 1541                 if (loc == NULL) {
 1542                         panic("ext2_mapsearch: map corrupted: start=%d, len=%d,"
 1543                             "fs=%s", start, len, fs->e2fs_fsmnt);
 1544                         /* NOTREACHED */
 1545                 }
 1546         }
 1547         return ((loc - bbp) * NBBY + ffs(~*loc) - 1);
 1548 }
 1549 
 1550 int
 1551 ext2_cg_has_sb(struct m_ext2fs *fs, int cg)
 1552 {
 1553         int a3, a5, a7;
 1554 
 1555         if (cg == 0)
 1556                 return (1);
 1557 
 1558         if (EXT2_HAS_COMPAT_FEATURE(fs, EXT2F_COMPAT_SPARSESUPER2)) {
 1559                 if (cg == le32toh(fs->e2fs->e4fs_backup_bgs[0]) ||
 1560                     cg == le32toh(fs->e2fs->e4fs_backup_bgs[1]))
 1561                         return (1);
 1562                 return (0);
 1563         }
 1564 
 1565         if ((cg <= 1) ||
 1566             !EXT2_HAS_RO_COMPAT_FEATURE(fs, EXT2F_ROCOMPAT_SPARSESUPER))
 1567                 return (1);
 1568 
 1569         if (!(cg & 1))
 1570                 return (0);
 1571 
 1572         for (a3 = 3, a5 = 5, a7 = 7;
 1573             a3 <= cg || a5 <= cg || a7 <= cg;
 1574             a3 *= 3, a5 *= 5, a7 *= 7)
 1575                 if (cg == a3 || cg == a5 || cg == a7)
 1576                         return (1);
 1577         return (0);
 1578 }

Cache object: 91e513619a4bfba3cb8eeaf1a17aad4e


[ source navigation ] [ diff markup ] [ identifier search ] [ freetext search ] [ file search ] [ list types ] [ track identifier ]


This page is part of the FreeBSD/Linux Linux Kernel Cross-Reference, and was automatically generated using a modified version of the LXR engine.