Home | History | Annotate | Line # | Download | only in ffs
ffs_snapshot.c revision 1.139
      1  1.139      maxv /*	$NetBSD: ffs_snapshot.c,v 1.139 2015/03/28 19:24:04 maxv Exp $	*/
      2   1.18   thorpej 
      3    1.1   hannken /*
      4    1.1   hannken  * Copyright 2000 Marshall Kirk McKusick. All Rights Reserved.
      5    1.1   hannken  *
      6    1.1   hannken  * Further information about snapshots can be obtained from:
      7    1.1   hannken  *
      8    1.1   hannken  *	Marshall Kirk McKusick		http://www.mckusick.com/softdep/
      9    1.1   hannken  *	1614 Oxford Street		mckusick (at) mckusick.com
     10    1.1   hannken  *	Berkeley, CA 94709-1608		+1-510-843-9542
     11    1.1   hannken  *	USA
     12    1.1   hannken  *
     13    1.1   hannken  * Redistribution and use in source and binary forms, with or without
     14    1.1   hannken  * modification, are permitted provided that the following conditions
     15    1.1   hannken  * are met:
     16    1.1   hannken  *
     17    1.1   hannken  * 1. Redistributions of source code must retain the above copyright
     18    1.1   hannken  *    notice, this list of conditions and the following disclaimer.
     19    1.1   hannken  * 2. Redistributions in binary form must reproduce the above copyright
     20    1.1   hannken  *    notice, this list of conditions and the following disclaimer in the
     21    1.1   hannken  *    documentation and/or other materials provided with the distribution.
     22    1.1   hannken  *
     23    1.1   hannken  * THIS SOFTWARE IS PROVIDED BY MARSHALL KIRK MCKUSICK ``AS IS'' AND ANY
     24    1.1   hannken  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
     25    1.1   hannken  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
     26    1.1   hannken  * DISCLAIMED.  IN NO EVENT SHALL MARSHALL KIRK MCKUSICK BE LIABLE FOR
     27    1.1   hannken  * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
     28    1.1   hannken  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
     29    1.1   hannken  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
     30    1.1   hannken  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
     31    1.1   hannken  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
     32    1.1   hannken  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
     33    1.1   hannken  * SUCH DAMAGE.
     34    1.1   hannken  *
     35    1.1   hannken  *	@(#)ffs_snapshot.c	8.11 (McKusick) 7/23/00
     36    1.1   hannken  *
     37    1.1   hannken  *	from FreeBSD: ffs_snapshot.c,v 1.79 2004/02/13 02:02:06 kuriyama Exp
     38    1.1   hannken  */
     39    1.1   hannken 
     40    1.1   hannken #include <sys/cdefs.h>
     41  1.139      maxv __KERNEL_RCSID(0, "$NetBSD: ffs_snapshot.c,v 1.139 2015/03/28 19:24:04 maxv Exp $");
     42    1.8   hannken 
     43    1.8   hannken #if defined(_KERNEL_OPT)
     44    1.8   hannken #include "opt_ffs.h"
     45  1.111    bouyer #include "opt_quota.h"
     46    1.8   hannken #endif
     47    1.1   hannken 
     48    1.1   hannken #include <sys/param.h>
     49    1.1   hannken #include <sys/kernel.h>
     50    1.1   hannken #include <sys/systm.h>
     51    1.1   hannken #include <sys/conf.h>
     52    1.1   hannken #include <sys/buf.h>
     53    1.1   hannken #include <sys/proc.h>
     54    1.1   hannken #include <sys/namei.h>
     55    1.1   hannken #include <sys/sched.h>
     56    1.1   hannken #include <sys/stat.h>
     57    1.1   hannken #include <sys/malloc.h>
     58    1.1   hannken #include <sys/mount.h>
     59    1.1   hannken #include <sys/resource.h>
     60    1.1   hannken #include <sys/resourcevar.h>
     61    1.1   hannken #include <sys/vnode.h>
     62   1.29      elad #include <sys/kauth.h>
     63   1.40   hannken #include <sys/fstrans.h>
     64   1.75   hannken #include <sys/wapbl.h>
     65    1.1   hannken 
     66    1.1   hannken #include <miscfs/specfs/specdev.h>
     67    1.1   hannken 
     68    1.1   hannken #include <ufs/ufs/quota.h>
     69    1.1   hannken #include <ufs/ufs/ufsmount.h>
     70    1.1   hannken #include <ufs/ufs/inode.h>
     71    1.1   hannken #include <ufs/ufs/ufs_extern.h>
     72    1.1   hannken #include <ufs/ufs/ufs_bswap.h>
     73   1.75   hannken #include <ufs/ufs/ufs_wapbl.h>
     74    1.1   hannken 
     75    1.1   hannken #include <ufs/ffs/fs.h>
     76    1.1   hannken #include <ufs/ffs/ffs_extern.h>
     77    1.1   hannken 
     78   1.72   hannken #include <uvm/uvm.h>
     79   1.72   hannken 
     80  1.137      matt TAILQ_HEAD(inodelst, inode);			/* List of active snapshots */
     81  1.137      matt 
     82   1.97   hannken struct snap_info {
     83   1.97   hannken 	kmutex_t si_lock;			/* Lock this snapinfo */
     84   1.97   hannken 	kmutex_t si_snaplock;			/* Snapshot vnode common lock */
     85  1.106   hannken 	lwp_t *si_owner;			/* Sanplock owner */
     86  1.137      matt 	struct inodelst si_snapshots;		/* List of active snapshots */
     87   1.97   hannken 	daddr_t *si_snapblklist;		/* Snapshot block hints list */
     88   1.97   hannken 	uint32_t si_gen;			/* Incremented on change */
     89   1.97   hannken };
     90   1.97   hannken 
     91   1.10   hannken #if !defined(FFS_NO_SNAPSHOT)
     92   1.76   hannken typedef int (*acctfunc_t)
     93   1.76   hannken     (struct vnode *, void *, int, int, struct fs *, daddr_t, int);
     94   1.76   hannken 
     95   1.79   hannken static int snapshot_setup(struct mount *, struct vnode *);
     96   1.79   hannken static int snapshot_copyfs(struct mount *, struct vnode *, void **);
     97   1.79   hannken static int snapshot_expunge(struct mount *, struct vnode *,
     98   1.79   hannken     struct fs *, daddr_t *, daddr_t **);
     99   1.79   hannken static int snapshot_expunge_snap(struct mount *, struct vnode *,
    100   1.79   hannken     struct fs *, daddr_t);
    101   1.79   hannken static int snapshot_writefs(struct mount *, struct vnode *, void *);
    102   1.79   hannken static int cgaccount(struct vnode *, int, int *);
    103   1.79   hannken static int cgaccount1(int, struct vnode *, void *, int);
    104   1.76   hannken static int expunge(struct vnode *, struct inode *, struct fs *,
    105   1.76   hannken     acctfunc_t, int);
    106   1.76   hannken static int indiracct(struct vnode *, struct vnode *, int, daddr_t,
    107   1.76   hannken     daddr_t, daddr_t, daddr_t, daddr_t, struct fs *, acctfunc_t, int);
    108   1.76   hannken static int fullacct(struct vnode *, void *, int, int, struct fs *,
    109   1.76   hannken     daddr_t, int);
    110   1.76   hannken static int snapacct(struct vnode *, void *, int, int, struct fs *,
    111   1.76   hannken     daddr_t, int);
    112   1.76   hannken static int mapacct(struct vnode *, void *, int, int, struct fs *,
    113   1.76   hannken     daddr_t, int);
    114   1.10   hannken #endif /* !defined(FFS_NO_SNAPSHOT) */
    115   1.10   hannken 
    116   1.55   hannken static int ffs_copyonwrite(void *, struct buf *, bool);
    117   1.72   hannken static int snapblkaddr(struct vnode *, daddr_t, daddr_t *);
    118   1.79   hannken static int rwfsblk(struct vnode *, int, void *, daddr_t);
    119   1.75   hannken static int syncsnap(struct vnode *);
    120   1.79   hannken static int wrsnapblk(struct vnode *, void *, daddr_t);
    121  1.107        he #if !defined(FFS_NO_SNAPSHOT)
    122  1.103   hannken static int blocks_in_journal(struct fs *);
    123  1.107        he #endif
    124   1.76   hannken 
    125   1.97   hannken static inline bool is_active_snapshot(struct snap_info *, struct inode *);
    126   1.79   hannken static inline daddr_t db_get(struct inode *, int);
    127   1.79   hannken static inline void db_assign(struct inode *, int, daddr_t);
    128   1.79   hannken static inline daddr_t ib_get(struct inode *, int);
    129   1.79   hannken static inline daddr_t idb_get(struct inode *, void *, int);
    130   1.79   hannken static inline void idb_assign(struct inode *, void *, int, daddr_t);
    131    1.1   hannken 
    132    1.1   hannken #ifdef DEBUG
    133    1.1   hannken static int snapdebug = 0;
    134    1.1   hannken #endif
    135   1.47   hannken 
    136   1.66   hannken int
    137   1.66   hannken ffs_snapshot_init(struct ufsmount *ump)
    138   1.47   hannken {
    139   1.66   hannken 	struct snap_info *si;
    140   1.66   hannken 
    141   1.66   hannken 	si = ump->um_snapinfo = kmem_alloc(sizeof(*si), KM_SLEEP);
    142   1.66   hannken 	if (si == NULL)
    143   1.66   hannken 		return ENOMEM;
    144   1.66   hannken 
    145   1.66   hannken 	TAILQ_INIT(&si->si_snapshots);
    146   1.66   hannken 	mutex_init(&si->si_lock, MUTEX_DEFAULT, IPL_NONE);
    147   1.74   hannken 	mutex_init(&si->si_snaplock, MUTEX_DEFAULT, IPL_NONE);
    148  1.106   hannken 	si->si_owner = NULL;
    149   1.66   hannken 	si->si_gen = 0;
    150   1.66   hannken 	si->si_snapblklist = NULL;
    151   1.47   hannken 
    152   1.66   hannken 	return 0;
    153   1.47   hannken }
    154   1.47   hannken 
    155   1.47   hannken void
    156   1.66   hannken ffs_snapshot_fini(struct ufsmount *ump)
    157   1.47   hannken {
    158   1.66   hannken 	struct snap_info *si;
    159   1.47   hannken 
    160   1.66   hannken 	si = ump->um_snapinfo;
    161   1.66   hannken 	ump->um_snapinfo = NULL;
    162   1.47   hannken 
    163   1.47   hannken 	KASSERT(TAILQ_EMPTY(&si->si_snapshots));
    164   1.49   hannken 	mutex_destroy(&si->si_lock);
    165   1.74   hannken 	mutex_destroy(&si->si_snaplock);
    166   1.47   hannken 	KASSERT(si->si_snapblklist == NULL);
    167   1.63        ad 	kmem_free(si, sizeof(*si));
    168   1.47   hannken }
    169   1.47   hannken 
    170    1.1   hannken /*
    171    1.1   hannken  * Create a snapshot file and initialize it for the filesystem.
    172    1.4   hannken  * Vnode is locked on entry and return.
    173    1.1   hannken  */
    174    1.1   hannken int
    175   1.79   hannken ffs_snapshot(struct mount *mp, struct vnode *vp, struct timespec *ctime)
    176    1.1   hannken {
    177   1.10   hannken #if defined(FFS_NO_SNAPSHOT)
    178   1.10   hannken 	return EOPNOTSUPP;
    179   1.10   hannken }
    180   1.10   hannken #else /* defined(FFS_NO_SNAPSHOT) */
    181  1.115   hannken 	bool suspended = false;
    182   1.79   hannken 	int error, redo = 0, snaploc;
    183   1.79   hannken 	void *sbbuf = NULL;
    184   1.79   hannken 	daddr_t *snaplist = NULL, snaplistsize = 0;
    185   1.79   hannken 	struct buf *bp, *nbp;
    186  1.102      matt 	struct fs *copy_fs = NULL;
    187  1.102      matt 	struct fs *fs = VFSTOUFS(mp)->um_fs;
    188   1.79   hannken 	struct inode *ip = VTOI(vp);
    189   1.79   hannken 	struct lwp *l = curlwp;
    190   1.79   hannken 	struct snap_info *si = VFSTOUFS(mp)->um_snapinfo;
    191   1.79   hannken 	struct timespec ts;
    192    1.1   hannken 	struct timeval starttime;
    193    1.1   hannken #ifdef DEBUG
    194    1.1   hannken 	struct timeval endtime;
    195    1.1   hannken #endif
    196   1.79   hannken 	struct vnode *devvp = ip->i_devvp;
    197    1.1   hannken 
    198    1.1   hannken 	/*
    199    1.1   hannken 	 * If the vnode already is a snapshot, return.
    200    1.1   hannken 	 */
    201  1.113   hannken 	if ((VTOI(vp)->i_flags & SF_SNAPSHOT)) {
    202  1.113   hannken 		if ((VTOI(vp)->i_flags & SF_SNAPINVAL))
    203  1.113   hannken 			return EINVAL;
    204    1.1   hannken 		if (ctime) {
    205    1.1   hannken 			ctime->tv_sec = DIP(VTOI(vp), mtime);
    206    1.1   hannken 			ctime->tv_nsec = DIP(VTOI(vp), mtimensec);
    207    1.1   hannken 		}
    208    1.1   hannken 		return 0;
    209    1.1   hannken 	}
    210    1.1   hannken 	/*
    211   1.79   hannken 	 * Check for free snapshot slot in the superblock.
    212   1.79   hannken 	 */
    213   1.79   hannken 	for (snaploc = 0; snaploc < FSMAXSNAP; snaploc++)
    214   1.79   hannken 		if (fs->fs_snapinum[snaploc] == 0)
    215   1.79   hannken 			break;
    216   1.79   hannken 	if (snaploc == FSMAXSNAP)
    217   1.79   hannken 		return (ENOSPC);
    218   1.79   hannken 	/*
    219   1.79   hannken 	 * Prepare the vnode to become a snapshot.
    220   1.79   hannken 	 */
    221   1.79   hannken 	error = snapshot_setup(mp, vp);
    222   1.79   hannken 	if (error)
    223   1.79   hannken 		goto out;
    224  1.111    bouyer 
    225   1.79   hannken 	/*
    226   1.79   hannken 	 * Copy all the cylinder group maps. Although the
    227   1.79   hannken 	 * filesystem is still active, we hope that only a few
    228   1.79   hannken 	 * cylinder groups will change between now and when we
    229   1.79   hannken 	 * suspend operations. Thus, we will be able to quickly
    230   1.79   hannken 	 * touch up the few cylinder groups that changed during
    231   1.79   hannken 	 * the suspension period.
    232   1.79   hannken 	 */
    233   1.79   hannken 	error = cgaccount(vp, 1, NULL);
    234   1.79   hannken 	if (error)
    235   1.79   hannken 		goto out;
    236  1.111    bouyer 
    237  1.111    bouyer 	/*
    238  1.111    bouyer 	 * snapshot is now valid
    239  1.111    bouyer 	 */
    240  1.111    bouyer 	ip->i_flags &= ~SF_SNAPINVAL;
    241  1.111    bouyer 	DIP_ASSIGN(ip, flags, ip->i_flags);
    242  1.111    bouyer 	ip->i_flag |= IN_CHANGE | IN_UPDATE;
    243  1.111    bouyer 
    244   1.79   hannken 	/*
    245   1.79   hannken 	 * Ensure that the snapshot is completely on disk.
    246   1.79   hannken 	 * Since we have marked it as a snapshot it is safe to
    247   1.79   hannken 	 * unlock it as no process will be allowed to write to it.
    248   1.79   hannken 	 */
    249   1.79   hannken 	error = VOP_FSYNC(vp, l->l_cred, FSYNC_WAIT, 0, 0);
    250   1.79   hannken 	if (error)
    251   1.79   hannken 		goto out;
    252   1.99   hannken 	VOP_UNLOCK(vp);
    253   1.79   hannken 	/*
    254   1.79   hannken 	 * All allocations are done, so we can now suspend the filesystem.
    255   1.79   hannken 	 */
    256   1.79   hannken 	error = vfs_suspend(vp->v_mount, 0);
    257   1.79   hannken 	vn_lock(vp, LK_EXCLUSIVE | LK_RETRY);
    258   1.79   hannken 	if (error)
    259   1.79   hannken 		goto out;
    260   1.79   hannken 	suspended = true;
    261   1.79   hannken 	getmicrotime(&starttime);
    262   1.79   hannken 	/*
    263   1.79   hannken 	 * First, copy all the cylinder group maps that have changed.
    264   1.79   hannken 	 */
    265   1.79   hannken 	error = cgaccount(vp, 2, &redo);
    266   1.79   hannken 	if (error)
    267   1.79   hannken 		goto out;
    268   1.79   hannken 	/*
    269   1.79   hannken 	 * Create a copy of the superblock and its summary information.
    270   1.79   hannken 	 */
    271   1.79   hannken 	error = snapshot_copyfs(mp, vp, &sbbuf);
    272  1.124  dholland 	copy_fs = (struct fs *)((char *)sbbuf + ffs_blkoff(fs, fs->fs_sblockloc));
    273   1.79   hannken 	if (error)
    274   1.79   hannken 		goto out;
    275   1.79   hannken 	/*
    276   1.79   hannken 	 * Expunge unlinked files from our view.
    277   1.79   hannken 	 */
    278   1.79   hannken 	error = snapshot_expunge(mp, vp, copy_fs, &snaplistsize, &snaplist);
    279   1.79   hannken 	if (error)
    280   1.79   hannken 		goto out;
    281   1.79   hannken 	/*
    282   1.79   hannken 	 * Record snapshot inode. Since this is the newest snapshot,
    283   1.79   hannken 	 * it must be placed at the end of the list.
    284   1.79   hannken 	 */
    285  1.110   hannken 	if (ip->i_nlink > 0)
    286  1.110   hannken 		fs->fs_snapinum[snaploc] = ip->i_number;
    287   1.79   hannken 
    288   1.79   hannken 	mutex_enter(&si->si_lock);
    289   1.97   hannken 	if (is_active_snapshot(si, ip))
    290   1.79   hannken 		panic("ffs_snapshot: %"PRIu64" already on list", ip->i_number);
    291   1.79   hannken 	TAILQ_INSERT_TAIL(&si->si_snapshots, ip, i_nextsnap);
    292   1.79   hannken 	if (TAILQ_FIRST(&si->si_snapshots) == ip) {
    293   1.79   hannken 		/*
    294   1.79   hannken 		 * If this is the first snapshot on this filesystem, put the
    295   1.79   hannken 		 * preliminary list in place and establish the cow handler.
    296   1.79   hannken 		 */
    297   1.79   hannken 		si->si_snapblklist = snaplist;
    298   1.79   hannken 		fscow_establish(mp, ffs_copyonwrite, devvp);
    299   1.79   hannken 	}
    300   1.79   hannken 	si->si_gen++;
    301   1.79   hannken 	mutex_exit(&si->si_lock);
    302   1.79   hannken 
    303   1.79   hannken 	vp->v_vflag |= VV_SYSTEM;
    304   1.79   hannken 	/*
    305   1.79   hannken 	 * Set the mtime to the time the snapshot has been taken.
    306   1.79   hannken 	 */
    307   1.79   hannken 	TIMEVAL_TO_TIMESPEC(&starttime, &ts);
    308   1.79   hannken 	if (ctime)
    309   1.79   hannken 		*ctime = ts;
    310   1.79   hannken 	DIP_ASSIGN(ip, mtime, ts.tv_sec);
    311   1.79   hannken 	DIP_ASSIGN(ip, mtimensec, ts.tv_nsec);
    312   1.79   hannken 	ip->i_flag |= IN_CHANGE | IN_UPDATE;
    313   1.79   hannken 	/*
    314   1.79   hannken 	 * Copy allocation information from all snapshots and then
    315   1.79   hannken 	 * expunge them from our view.
    316   1.79   hannken 	 */
    317   1.79   hannken 	error = snapshot_expunge_snap(mp, vp, copy_fs, snaplistsize);
    318   1.79   hannken 	if (error)
    319   1.79   hannken 		goto out;
    320   1.79   hannken 	/*
    321   1.79   hannken 	 * Write the superblock and its summary information to the snapshot.
    322   1.79   hannken 	 */
    323   1.79   hannken 	error = snapshot_writefs(mp, vp, sbbuf);
    324   1.79   hannken 	if (error)
    325   1.79   hannken 		goto out;
    326   1.79   hannken 	/*
    327   1.79   hannken 	 * We're nearly done, ensure that the snapshot is completely on disk.
    328   1.79   hannken 	 */
    329   1.79   hannken 	error = VOP_FSYNC(vp, l->l_cred, FSYNC_WAIT, 0, 0);
    330   1.79   hannken 	if (error)
    331   1.79   hannken 		goto out;
    332   1.79   hannken 	/*
    333   1.79   hannken 	 * Invalidate and free all pages on the snapshot vnode.
    334   1.79   hannken 	 * We will read and write through the buffercache.
    335   1.79   hannken 	 */
    336  1.116     rmind 	mutex_enter(vp->v_interlock);
    337   1.79   hannken 	error = VOP_PUTPAGES(vp, 0, 0,
    338   1.79   hannken 		    PGO_ALLPAGES | PGO_CLEANIT | PGO_SYNCIO | PGO_FREE);
    339   1.79   hannken 	if (error)
    340   1.79   hannken 		goto out;
    341   1.79   hannken 	/*
    342   1.79   hannken 	 * Invalidate short ( < fs_bsize ) buffers.  We will always read
    343   1.79   hannken 	 * full size buffers later.
    344   1.79   hannken 	 */
    345   1.79   hannken 	mutex_enter(&bufcache_lock);
    346   1.79   hannken 	KASSERT(LIST_FIRST(&vp->v_dirtyblkhd) == NULL);
    347   1.79   hannken 	for (bp = LIST_FIRST(&vp->v_cleanblkhd); bp; bp = nbp) {
    348   1.79   hannken 		nbp = LIST_NEXT(bp, b_vnbufs);
    349  1.122   hannken 		if (bp->b_bcount == fs->fs_bsize)
    350  1.122   hannken 			continue;
    351  1.122   hannken 		error = bbusy(bp, false, 0, NULL);
    352  1.122   hannken 		if (error != 0) {
    353  1.122   hannken 			if (error == EPASSTHROUGH) {
    354  1.122   hannken 				nbp = LIST_FIRST(&vp->v_cleanblkhd);
    355  1.122   hannken 				continue;
    356  1.122   hannken 			}
    357  1.122   hannken 			break;
    358   1.79   hannken 		}
    359  1.122   hannken 		brelsel(bp, BC_INVAL | BC_VFLUSH);
    360   1.79   hannken 	}
    361   1.79   hannken 	mutex_exit(&bufcache_lock);
    362   1.79   hannken 
    363   1.79   hannken out:
    364   1.79   hannken 	if (sbbuf != NULL) {
    365   1.79   hannken 		free(copy_fs->fs_csp, M_UFSMNT);
    366   1.79   hannken 		free(sbbuf, M_UFSMNT);
    367   1.79   hannken 	}
    368   1.79   hannken 	if (fs->fs_active != NULL) {
    369   1.79   hannken 		free(fs->fs_active, M_DEVBUF);
    370   1.79   hannken 		fs->fs_active = NULL;
    371   1.79   hannken 	}
    372   1.79   hannken 
    373   1.79   hannken 	mutex_enter(&si->si_lock);
    374   1.79   hannken 	if (snaplist != NULL) {
    375   1.79   hannken 		if (si->si_snapblklist == snaplist)
    376   1.79   hannken 			si->si_snapblklist = NULL;
    377   1.79   hannken 		free(snaplist, M_UFSMNT);
    378   1.79   hannken 	}
    379   1.79   hannken 	if (error) {
    380   1.79   hannken 		fs->fs_snapinum[snaploc] = 0;
    381   1.79   hannken 	} else {
    382   1.79   hannken 		/*
    383   1.79   hannken 		 * As this is the newest list, it is the most inclusive, so
    384   1.79   hannken 		 * should replace the previous list.
    385   1.79   hannken 		 */
    386   1.79   hannken 		si->si_snapblklist = ip->i_snapblklist;
    387   1.79   hannken 	}
    388   1.79   hannken 	si->si_gen++;
    389   1.79   hannken 	mutex_exit(&si->si_lock);
    390   1.79   hannken 
    391  1.115   hannken 	if (suspended) {
    392  1.115   hannken 		VOP_UNLOCK(vp);
    393   1.79   hannken 		vfs_resume(vp->v_mount);
    394  1.115   hannken 		vn_lock(vp, LK_EXCLUSIVE | LK_RETRY);
    395  1.115   hannken #ifdef DEBUG
    396  1.115   hannken 		getmicrotime(&endtime);
    397  1.115   hannken 		timersub(&endtime, &starttime, &endtime);
    398  1.115   hannken 		printf("%s: suspended %lld.%03d sec, redo %d of %d\n",
    399  1.115   hannken 		    mp->mnt_stat.f_mntonname, (long long)endtime.tv_sec,
    400  1.115   hannken 		    endtime.tv_usec / 1000, redo, fs->fs_ncg);
    401  1.115   hannken #endif
    402   1.79   hannken 	}
    403   1.79   hannken 	if (error) {
    404  1.136  dholland 		if (UFS_WAPBL_BEGIN(mp) == 0) {
    405   1.79   hannken 			(void) ffs_truncate(vp, (off_t)0, 0, NOCRED);
    406   1.79   hannken 			UFS_WAPBL_END(mp);
    407   1.79   hannken 		}
    408  1.110   hannken 	} else if (ip->i_nlink > 0)
    409  1.104   hannken 		vref(vp);
    410   1.79   hannken 	return (error);
    411   1.79   hannken }
    412   1.79   hannken 
    413   1.79   hannken /*
    414   1.79   hannken  * Prepare vnode to become a snapshot.
    415   1.79   hannken  */
    416   1.79   hannken static int
    417   1.79   hannken snapshot_setup(struct mount *mp, struct vnode *vp)
    418   1.79   hannken {
    419  1.112   hannken 	int error, n, len, loc, cg;
    420   1.79   hannken 	daddr_t blkno, numblks;
    421   1.79   hannken 	struct buf *ibp, *nbp;
    422   1.79   hannken 	struct fs *fs = VFSTOUFS(mp)->um_fs;
    423   1.79   hannken 	struct lwp *l = curlwp;
    424  1.103   hannken 	const int wbreak = blocks_in_journal(fs)/8;
    425  1.111    bouyer 	struct inode *ip = VTOI(vp);
    426   1.79   hannken 
    427   1.79   hannken 	/*
    428  1.135   hannken 	 * Check mount, readonly reference and owner.
    429    1.1   hannken 	 */
    430    1.4   hannken 	if (vp->v_mount != mp)
    431    1.1   hannken 		return EXDEV;
    432  1.135   hannken 	if (vp->v_writecount != 0)
    433    1.1   hannken 		return EBUSY;
    434  1.119      elad 	error = kauth_authorize_system(l->l_cred, KAUTH_SYSTEM_FS_SNAPSHOT,
    435  1.119      elad 	    0, mp, vp, NULL);
    436  1.119      elad 	if (error)
    437    1.9   hannken 		return EACCES;
    438    1.9   hannken 
    439    1.1   hannken 	if (vp->v_size != 0) {
    440   1.56     pooka 		error = ffs_truncate(vp, 0, 0, NOCRED);
    441    1.4   hannken 		if (error)
    442    1.1   hannken 			return error;
    443    1.1   hannken 	}
    444  1.111    bouyer 
    445  1.111    bouyer 	/* Change inode to snapshot type file. */
    446  1.111    bouyer 	error = UFS_WAPBL_BEGIN(mp);
    447  1.111    bouyer 	if (error)
    448  1.111    bouyer 		return error;
    449  1.111    bouyer #if defined(QUOTA) || defined(QUOTA2)
    450  1.111    bouyer 	/* shapshot inodes are not accounted in quotas */
    451  1.111    bouyer 	chkiq(ip, -1, l->l_cred, 0);
    452  1.111    bouyer #endif
    453  1.111    bouyer 	ip->i_flags |= (SF_SNAPSHOT | SF_SNAPINVAL);
    454  1.111    bouyer 	DIP_ASSIGN(ip, flags, ip->i_flags);
    455  1.111    bouyer 	ip->i_flag |= IN_CHANGE | IN_UPDATE;
    456  1.111    bouyer 	ffs_update(vp, NULL, NULL, UPDATE_WAIT);
    457  1.111    bouyer 	UFS_WAPBL_END(mp);
    458  1.111    bouyer 
    459  1.111    bouyer 	KASSERT(ip->i_flags & SF_SNAPSHOT);
    460    1.1   hannken 	/*
    461   1.16   hannken 	 * Write an empty list of preallocated blocks to the end of
    462   1.16   hannken 	 * the snapshot to set size to at least that of the filesystem.
    463    1.1   hannken 	 */
    464    1.1   hannken 	numblks = howmany(fs->fs_size, fs->fs_frag);
    465   1.16   hannken 	blkno = 1;
    466   1.79   hannken 	blkno = ufs_rw64(blkno, UFS_FSNEEDSWAP(fs));
    467    1.1   hannken 	error = vn_rdwr(UIO_WRITE, vp,
    468  1.126  dholland 	    (void *)&blkno, sizeof(blkno), ffs_lblktosize(fs, (off_t)numblks),
    469   1.31        ad 	    UIO_SYSSPACE, IO_NODELOCKED|IO_UNIT, l->l_cred, NULL, NULL);
    470    1.1   hannken 	if (error)
    471   1.79   hannken 		return error;
    472    1.1   hannken 	/*
    473    1.1   hannken 	 * Preallocate critical data structures so that we can copy
    474    1.1   hannken 	 * them in without further allocation after we suspend all
    475    1.1   hannken 	 * operations on the filesystem. We would like to just release
    476    1.1   hannken 	 * the allocated buffers without writing them since they will
    477    1.1   hannken 	 * be filled in below once we are ready to go, but this upsets
    478    1.1   hannken 	 * the soft update code, so we go ahead and write the new buffers.
    479    1.1   hannken 	 *
    480    1.1   hannken 	 * Allocate all indirect blocks and mark all of them as not
    481    1.1   hannken 	 * needing to be copied.
    482    1.1   hannken 	 */
    483   1.75   hannken 	error = UFS_WAPBL_BEGIN(mp);
    484   1.75   hannken 	if (error)
    485   1.79   hannken 		return error;
    486  1.124  dholland 	for (blkno = UFS_NDADDR, n = 0; blkno < numblks; blkno += FFS_NINDIR(fs)) {
    487  1.126  dholland 		error = ffs_balloc(vp, ffs_lblktosize(fs, (off_t)blkno),
    488   1.31        ad 		    fs->fs_bsize, l->l_cred, B_METAONLY, &ibp);
    489   1.79   hannken 		if (error)
    490    1.1   hannken 			goto out;
    491   1.92        ad 		brelse(ibp, 0);
    492  1.103   hannken 		if (wbreak > 0 && (++n % wbreak) == 0) {
    493   1.75   hannken 			UFS_WAPBL_END(mp);
    494   1.75   hannken 			error = UFS_WAPBL_BEGIN(mp);
    495   1.75   hannken 			if (error)
    496   1.79   hannken 				return error;
    497   1.75   hannken 		}
    498    1.1   hannken 	}
    499    1.1   hannken 	/*
    500    1.1   hannken 	 * Allocate copies for the superblock and its summary information.
    501    1.1   hannken 	 */
    502   1.79   hannken 	error = ffs_balloc(vp, fs->fs_sblockloc, fs->fs_sbsize, l->l_cred,
    503   1.15   hannken 	    0, &nbp);
    504   1.79   hannken 	if (error)
    505   1.15   hannken 		goto out;
    506   1.15   hannken 	bawrite(nbp);
    507  1.127  dholland 	blkno = ffs_fragstoblks(fs, fs->fs_csaddr);
    508   1.15   hannken 	len = howmany(fs->fs_cssize, fs->fs_bsize);
    509   1.15   hannken 	for (loc = 0; loc < len; loc++) {
    510  1.126  dholland 		error = ffs_balloc(vp, ffs_lblktosize(fs, (off_t)(blkno + loc)),
    511   1.79   hannken 		    fs->fs_bsize, l->l_cred, 0, &nbp);
    512   1.79   hannken 		if (error)
    513    1.1   hannken 			goto out;
    514   1.15   hannken 		bawrite(nbp);
    515  1.112   hannken 		if (wbreak > 0 && (++n % wbreak) == 0) {
    516  1.112   hannken 			UFS_WAPBL_END(mp);
    517  1.112   hannken 			error = UFS_WAPBL_BEGIN(mp);
    518  1.112   hannken 			if (error)
    519  1.112   hannken 				return error;
    520  1.112   hannken 		}
    521  1.112   hannken 	}
    522  1.112   hannken 	/*
    523  1.112   hannken 	 * Allocate all cylinder group blocks.
    524  1.112   hannken 	 */
    525  1.112   hannken 	for (cg = 0; cg < fs->fs_ncg; cg++) {
    526  1.126  dholland 		error = ffs_balloc(vp, ffs_lfragtosize(fs, cgtod(fs, cg)),
    527  1.112   hannken 		    fs->fs_bsize, l->l_cred, 0, &nbp);
    528  1.112   hannken 		if (error)
    529  1.112   hannken 			goto out;
    530  1.112   hannken 		bawrite(nbp);
    531  1.112   hannken 		if (wbreak > 0 && (++n % wbreak) == 0) {
    532  1.112   hannken 			UFS_WAPBL_END(mp);
    533  1.112   hannken 			error = UFS_WAPBL_BEGIN(mp);
    534  1.112   hannken 			if (error)
    535  1.112   hannken 				return error;
    536  1.112   hannken 		}
    537   1.15   hannken 	}
    538   1.79   hannken 
    539   1.79   hannken out:
    540   1.75   hannken 	UFS_WAPBL_END(mp);
    541   1.79   hannken 	return error;
    542   1.79   hannken }
    543   1.79   hannken 
    544   1.79   hannken /*
    545   1.79   hannken  * Create a copy of the superblock and its summary information.
    546   1.79   hannken  * It is up to the caller to free copyfs and copy_fs->fs_csp.
    547   1.79   hannken  */
    548   1.79   hannken static int
    549   1.79   hannken snapshot_copyfs(struct mount *mp, struct vnode *vp, void **sbbuf)
    550   1.79   hannken {
    551   1.79   hannken 	int error, i, len, loc, size;
    552   1.79   hannken 	void *space;
    553   1.79   hannken 	int32_t *lp;
    554   1.79   hannken 	struct buf *bp;
    555   1.79   hannken 	struct fs *copyfs, *fs = VFSTOUFS(mp)->um_fs;
    556   1.79   hannken 	struct vnode *devvp = VTOI(vp)->i_devvp;
    557   1.79   hannken 
    558    1.1   hannken 	/*
    559    1.1   hannken 	 * Grab a copy of the superblock and its summary information.
    560    1.1   hannken 	 * We delay writing it until the suspension is released below.
    561    1.1   hannken 	 */
    562   1.79   hannken 	*sbbuf = malloc(fs->fs_bsize, M_UFSMNT, M_WAITOK);
    563  1.124  dholland 	loc = ffs_blkoff(fs, fs->fs_sblockloc);
    564    1.1   hannken 	if (loc > 0)
    565   1.79   hannken 		memset(*sbbuf, 0, loc);
    566   1.79   hannken 	copyfs = (struct fs *)((char *)(*sbbuf) + loc);
    567   1.95   tsutsui 	memcpy(copyfs, fs, fs->fs_sbsize);
    568    1.1   hannken 	size = fs->fs_bsize < SBLOCKSIZE ? fs->fs_bsize : SBLOCKSIZE;
    569    1.1   hannken 	if (fs->fs_sbsize < size)
    570   1.79   hannken 		memset((char *)(*sbbuf) + loc + fs->fs_sbsize, 0,
    571   1.43  christos 		    size - fs->fs_sbsize);
    572  1.126  dholland 	size = ffs_blkroundup(fs, fs->fs_cssize);
    573    1.1   hannken 	if (fs->fs_contigsumsize > 0)
    574    1.1   hannken 		size += fs->fs_ncg * sizeof(int32_t);
    575   1.79   hannken 	space = malloc(size, M_UFSMNT, M_WAITOK);
    576   1.79   hannken 	copyfs->fs_csp = space;
    577   1.95   tsutsui 	memcpy(copyfs->fs_csp, fs->fs_csp, fs->fs_cssize);
    578   1.12      yamt 	space = (char *)space + fs->fs_cssize;
    579    1.1   hannken 	loc = howmany(fs->fs_cssize, fs->fs_fsize);
    580    1.1   hannken 	i = fs->fs_frag - loc % fs->fs_frag;
    581    1.1   hannken 	len = (i == fs->fs_frag) ? 0 : i * fs->fs_fsize;
    582    1.1   hannken 	if (len > 0) {
    583  1.125  dholland 		if ((error = bread(devvp, FFS_FSBTODB(fs, fs->fs_csaddr + loc),
    584  1.139      maxv 		    len, 0, &bp)) != 0) {
    585   1.79   hannken 			free(copyfs->fs_csp, M_UFSMNT);
    586   1.79   hannken 			free(*sbbuf, M_UFSMNT);
    587   1.79   hannken 			*sbbuf = NULL;
    588   1.79   hannken 			return error;
    589    1.1   hannken 		}
    590   1.95   tsutsui 		memcpy(space, bp->b_data, (u_int)len);
    591   1.12      yamt 		space = (char *)space + len;
    592   1.52        ad 		brelse(bp, BC_INVAL | BC_NOCACHE);
    593    1.1   hannken 	}
    594    1.1   hannken 	if (fs->fs_contigsumsize > 0) {
    595   1.79   hannken 		copyfs->fs_maxcluster = lp = space;
    596    1.1   hannken 		for (i = 0; i < fs->fs_ncg; i++)
    597    1.1   hannken 			*lp++ = fs->fs_contigsumsize;
    598    1.1   hannken 	}
    599   1.79   hannken 	if (mp->mnt_wapbl)
    600   1.79   hannken 		copyfs->fs_flags &= ~FS_DOWAPBL;
    601   1.79   hannken 	return 0;
    602   1.79   hannken }
    603   1.79   hannken 
    604  1.134  christos struct snapshot_expunge_ctx {
    605  1.134  christos 	struct vnode *logvp;
    606  1.134  christos 	struct lwp *l;
    607  1.134  christos 	struct vnode *vp;
    608  1.134  christos 	struct fs *copy_fs;
    609  1.134  christos };
    610  1.134  christos 
    611  1.134  christos static bool
    612  1.134  christos snapshot_expunge_selector(void *cl, struct vnode *xvp)
    613  1.134  christos {
    614  1.134  christos 	struct vattr vat;
    615  1.134  christos 	struct snapshot_expunge_ctx *c = cl;
    616  1.134  christos 	struct inode *xp;
    617  1.134  christos 
    618  1.134  christos 	xp = VTOI(xvp);
    619  1.134  christos 	if (xvp->v_type == VNON || VTOI(xvp) == NULL ||
    620  1.134  christos 	    (xp->i_flags & SF_SNAPSHOT))
    621  1.134  christos 		return false;
    622  1.134  christos #ifdef DEBUG
    623  1.134  christos 	if (snapdebug)
    624  1.134  christos 		vprint("ffs_snapshot: busy vnode", xvp);
    625  1.134  christos #endif
    626  1.134  christos 
    627  1.134  christos 	if (xvp == c->logvp)
    628  1.134  christos 		return true;
    629  1.134  christos 
    630  1.134  christos 	if (VOP_GETATTR(xvp, &vat, c->l->l_cred) == 0 &&
    631  1.134  christos 	    vat.va_nlink > 0)
    632  1.134  christos 		return false;
    633  1.134  christos 
    634  1.134  christos 	if (ffs_checkfreefile(c->copy_fs, c->vp, xp->i_number))
    635  1.134  christos 		return false;
    636  1.134  christos 
    637  1.134  christos 	return true;
    638  1.134  christos }
    639  1.134  christos 
    640   1.79   hannken /*
    641   1.79   hannken  * We must check for active files that have been unlinked (e.g., with a zero
    642   1.79   hannken  * link count). We have to expunge all trace of these files from the snapshot
    643   1.79   hannken  * so that they are not reclaimed prematurely by fsck or unnecessarily dumped.
    644   1.79   hannken  * Note that we skip unlinked snapshot files as they will be handled separately.
    645   1.79   hannken  * Calculate the snapshot list size and create a preliminary list.
    646   1.79   hannken  */
    647   1.79   hannken static int
    648   1.79   hannken snapshot_expunge(struct mount *mp, struct vnode *vp, struct fs *copy_fs,
    649   1.79   hannken     daddr_t *snaplistsize, daddr_t **snaplist)
    650   1.79   hannken {
    651  1.105    bouyer 	int cg, error = 0, len, loc;
    652   1.79   hannken 	daddr_t blkno, *blkp;
    653   1.79   hannken 	struct fs *fs = VFSTOUFS(mp)->um_fs;
    654   1.79   hannken 	struct inode *xp;
    655   1.79   hannken 	struct lwp *l = curlwp;
    656  1.133   hannken 	struct vnode *logvp = NULL, *xvp;
    657  1.133   hannken 	struct vnode_iterator *marker;
    658  1.134  christos 	struct snapshot_expunge_ctx ctx;
    659   1.79   hannken 
    660   1.79   hannken 	*snaplist = NULL;
    661    1.1   hannken 	/*
    662   1.79   hannken 	 * Get the log inode if any.
    663   1.79   hannken 	 */
    664   1.79   hannken 	if ((fs->fs_flags & FS_DOWAPBL) &&
    665   1.79   hannken 	    fs->fs_journal_location == UFS_WAPBL_JOURNALLOC_IN_FILESYSTEM) {
    666   1.79   hannken 		error = VFS_VGET(mp,
    667   1.79   hannken 		    fs->fs_journallocs[UFS_WAPBL_INFS_INO], &logvp);
    668   1.79   hannken 		if (error)
    669   1.79   hannken 			goto out;
    670   1.79   hannken 	}
    671   1.79   hannken 	/*
    672   1.79   hannken 	 * We also calculate the needed size for the snapshot list.
    673   1.79   hannken 	 */
    674   1.79   hannken 	*snaplistsize = fs->fs_ncg + howmany(fs->fs_cssize, fs->fs_bsize) +
    675   1.79   hannken 	    FSMAXSNAP + 1 /* superblock */ + 1 /* last block */ + 1 /* size */;
    676  1.133   hannken 
    677  1.133   hannken 	vfs_vnode_iterator_init(mp, &marker);
    678  1.134  christos 	ctx.logvp = logvp;
    679  1.134  christos 	ctx.l = l;
    680  1.134  christos 	ctx.vp = vp;
    681  1.134  christos 	ctx.copy_fs = copy_fs;
    682  1.134  christos 	while ((xvp = vfs_vnode_iterator_next(marker, snapshot_expunge_selector,
    683  1.134  christos 	    &ctx)))
    684  1.134  christos 	{
    685    1.1   hannken 		/*
    686    1.1   hannken 		 * If there is a fragment, clear it here.
    687    1.1   hannken 		 */
    688  1.134  christos 		xp = VTOI(xvp);
    689    1.1   hannken 		blkno = 0;
    690    1.1   hannken 		loc = howmany(xp->i_size, fs->fs_bsize) - 1;
    691  1.121  dholland 		if (loc < UFS_NDADDR) {
    692  1.126  dholland 			len = ffs_fragroundup(fs, ffs_blkoff(fs, xp->i_size));
    693    1.5   hannken 			if (len > 0 && len < fs->fs_bsize) {
    694  1.103   hannken 				error = UFS_WAPBL_BEGIN(mp);
    695  1.103   hannken 				if (error) {
    696  1.133   hannken 					vrele(xvp);
    697  1.133   hannken 					vfs_vnode_iterator_destroy(marker);
    698  1.103   hannken 					goto out;
    699  1.103   hannken 				}
    700   1.83     joerg 				ffs_blkfree_snap(copy_fs, vp, db_get(xp, loc),
    701    1.1   hannken 				    len, xp->i_number);
    702    1.1   hannken 				blkno = db_get(xp, loc);
    703    1.1   hannken 				db_assign(xp, loc, 0);
    704  1.103   hannken 				UFS_WAPBL_END(mp);
    705    1.1   hannken 			}
    706    1.1   hannken 		}
    707   1.79   hannken 		*snaplistsize += 1;
    708   1.76   hannken 		error = expunge(vp, xp, copy_fs, fullacct, BLK_NOCOPY);
    709    1.1   hannken 		if (blkno)
    710    1.1   hannken 			db_assign(xp, loc, blkno);
    711  1.103   hannken 		if (!error) {
    712  1.103   hannken 			error = UFS_WAPBL_BEGIN(mp);
    713  1.103   hannken 			if (!error) {
    714  1.103   hannken 				error = ffs_freefile_snap(copy_fs, vp,
    715  1.103   hannken 				    xp->i_number, xp->i_mode);
    716  1.103   hannken 				UFS_WAPBL_END(mp);
    717  1.103   hannken 			}
    718  1.103   hannken 		}
    719  1.133   hannken 		vrele(xvp);
    720    1.1   hannken 		if (error) {
    721  1.133   hannken 			vfs_vnode_iterator_destroy(marker);
    722   1.75   hannken 			goto out;
    723    1.1   hannken 		}
    724   1.65   hannken 	}
    725  1.133   hannken 	vfs_vnode_iterator_destroy(marker);
    726  1.133   hannken 
    727    1.1   hannken 	/*
    728   1.79   hannken 	 * Create a preliminary list of preallocated snapshot blocks.
    729    1.1   hannken 	 */
    730   1.79   hannken 	*snaplist = malloc(*snaplistsize * sizeof(daddr_t), M_UFSMNT, M_WAITOK);
    731   1.79   hannken 	blkp = &(*snaplist)[1];
    732  1.126  dholland 	*blkp++ = ffs_lblkno(fs, fs->fs_sblockloc);
    733  1.127  dholland 	blkno = ffs_fragstoblks(fs, fs->fs_csaddr);
    734   1.79   hannken 	for (cg = 0; cg < fs->fs_ncg; cg++) {
    735  1.127  dholland 		if (ffs_fragstoblks(fs, cgtod(fs, cg)) > blkno)
    736   1.79   hannken 			break;
    737  1.127  dholland 		*blkp++ = ffs_fragstoblks(fs, cgtod(fs, cg));
    738   1.79   hannken 	}
    739   1.79   hannken 	len = howmany(fs->fs_cssize, fs->fs_bsize);
    740   1.79   hannken 	for (loc = 0; loc < len; loc++)
    741   1.79   hannken 		*blkp++ = blkno + loc;
    742   1.79   hannken 	for (; cg < fs->fs_ncg; cg++)
    743  1.127  dholland 		*blkp++ = ffs_fragstoblks(fs, cgtod(fs, cg));
    744   1.98   hannken 	(*snaplist)[0] = blkp - &(*snaplist)[0];
    745   1.79   hannken 
    746   1.79   hannken out:
    747   1.79   hannken 	if (logvp != NULL)
    748   1.79   hannken 		vput(logvp);
    749   1.79   hannken 	if (error && *snaplist != NULL) {
    750   1.79   hannken 		free(*snaplist, M_UFSMNT);
    751   1.79   hannken 		*snaplist = NULL;
    752   1.79   hannken 	}
    753   1.79   hannken 
    754   1.79   hannken 	return error;
    755   1.79   hannken }
    756   1.79   hannken 
    757   1.79   hannken /*
    758   1.79   hannken  * Copy allocation information from all the snapshots in this snapshot and
    759   1.79   hannken  * then expunge them from its view. Also, collect the list of allocated
    760   1.79   hannken  * blocks in i_snapblklist.
    761   1.79   hannken  */
    762   1.79   hannken static int
    763   1.79   hannken snapshot_expunge_snap(struct mount *mp, struct vnode *vp,
    764   1.79   hannken     struct fs *copy_fs, daddr_t snaplistsize)
    765   1.79   hannken {
    766  1.103   hannken 	int error = 0, i;
    767   1.79   hannken 	daddr_t numblks, *snaplist = NULL;
    768   1.79   hannken 	struct fs *fs = VFSTOUFS(mp)->um_fs;
    769   1.79   hannken 	struct inode *ip = VTOI(vp), *xp;
    770   1.79   hannken 	struct lwp *l = curlwp;
    771   1.79   hannken 	struct snap_info *si = VFSTOUFS(mp)->um_snapinfo;
    772   1.79   hannken 
    773   1.47   hannken 	TAILQ_FOREACH(xp, &si->si_snapshots, i_nextsnap) {
    774  1.110   hannken 		if (xp != ip) {
    775  1.110   hannken 			error = expunge(vp, xp, fs, snapacct, BLK_SNAP);
    776  1.110   hannken 			if (error)
    777  1.110   hannken 				break;
    778  1.110   hannken 		}
    779   1.92        ad 		if (xp->i_nlink != 0)
    780   1.79   hannken 			continue;
    781  1.103   hannken 		error = UFS_WAPBL_BEGIN(mp);
    782  1.103   hannken 		if (error)
    783  1.103   hannken 			break;
    784   1.84     joerg 		error = ffs_freefile_snap(copy_fs, vp, xp->i_number, xp->i_mode);
    785  1.103   hannken 		UFS_WAPBL_END(mp);
    786   1.79   hannken 		if (error)
    787   1.79   hannken 			break;
    788    1.1   hannken 	}
    789   1.79   hannken 	if (error)
    790   1.79   hannken 		goto out;
    791    1.1   hannken 	/*
    792    1.1   hannken 	 * Allocate space for the full list of preallocated snapshot blocks.
    793    1.1   hannken 	 */
    794   1.79   hannken 	snaplist = malloc(snaplistsize * sizeof(daddr_t), M_UFSMNT, M_WAITOK);
    795   1.79   hannken 	ip->i_snapblklist = &snaplist[1];
    796    1.1   hannken 	/*
    797    1.1   hannken 	 * Expunge the blocks used by the snapshots from the set of
    798    1.1   hannken 	 * blocks marked as used in the snapshot bitmaps. Also, collect
    799    1.1   hannken 	 * the list of allocated blocks in i_snapblklist.
    800    1.1   hannken 	 */
    801   1.79   hannken 	error = expunge(vp, ip, copy_fs, mapacct, BLK_SNAP);
    802   1.79   hannken 	if (error)
    803   1.79   hannken 		goto out;
    804   1.79   hannken 	if (snaplistsize < ip->i_snapblklist - snaplist)
    805    1.1   hannken 		panic("ffs_snapshot: list too small");
    806   1.79   hannken 	snaplistsize = ip->i_snapblklist - snaplist;
    807   1.79   hannken 	snaplist[0] = snaplistsize;
    808   1.79   hannken 	ip->i_snapblklist = &snaplist[0];
    809    1.1   hannken 	/*
    810    1.1   hannken 	 * Write out the list of allocated blocks to the end of the snapshot.
    811    1.1   hannken 	 */
    812   1.79   hannken 	numblks = howmany(fs->fs_size, fs->fs_frag);
    813   1.15   hannken 	for (i = 0; i < snaplistsize; i++)
    814   1.79   hannken 		snaplist[i] = ufs_rw64(snaplist[i], UFS_FSNEEDSWAP(fs));
    815   1.79   hannken 	error = vn_rdwr(UIO_WRITE, vp, (void *)snaplist,
    816  1.126  dholland 	    snaplistsize * sizeof(daddr_t), ffs_lblktosize(fs, (off_t)numblks),
    817  1.103   hannken 	    UIO_SYSSPACE, IO_NODELOCKED | IO_UNIT, l->l_cred, NULL, NULL);
    818   1.15   hannken 	for (i = 0; i < snaplistsize; i++)
    819   1.79   hannken 		snaplist[i] = ufs_rw64(snaplist[i], UFS_FSNEEDSWAP(fs));
    820   1.79   hannken out:
    821   1.79   hannken 	if (error && snaplist != NULL) {
    822   1.79   hannken 		free(snaplist, M_UFSMNT);
    823   1.79   hannken 		ip->i_snapblklist = NULL;
    824    1.1   hannken 	}
    825   1.79   hannken 	return error;
    826   1.79   hannken }
    827   1.79   hannken 
    828   1.79   hannken /*
    829   1.79   hannken  * Write the superblock and its summary information to the snapshot.
    830  1.121  dholland  * Make sure, the first UFS_NDADDR blocks get copied to the snapshot.
    831   1.79   hannken  */
    832   1.79   hannken static int
    833   1.79   hannken snapshot_writefs(struct mount *mp, struct vnode *vp, void *sbbuf)
    834   1.79   hannken {
    835   1.79   hannken 	int error, len, loc;
    836   1.79   hannken 	void *space;
    837   1.79   hannken 	daddr_t blkno;
    838   1.79   hannken 	struct buf *bp;
    839   1.79   hannken 	struct fs *copyfs, *fs = VFSTOUFS(mp)->um_fs;
    840   1.79   hannken 	struct inode *ip = VTOI(vp);
    841   1.79   hannken 	struct lwp *l = curlwp;
    842   1.79   hannken 
    843  1.124  dholland 	copyfs = (struct fs *)((char *)sbbuf + ffs_blkoff(fs, fs->fs_sblockloc));
    844   1.79   hannken 
    845    1.1   hannken 	/*
    846    1.1   hannken 	 * Write the superblock and its summary information
    847    1.1   hannken 	 * to the snapshot.
    848    1.1   hannken 	 */
    849  1.127  dholland 	blkno = ffs_fragstoblks(fs, fs->fs_csaddr);
    850    1.1   hannken 	len = howmany(fs->fs_cssize, fs->fs_bsize);
    851   1.79   hannken 	space = copyfs->fs_csp;
    852    1.8   hannken #ifdef FFS_EI
    853   1.79   hannken 	if (UFS_FSNEEDSWAP(fs)) {
    854   1.79   hannken 		ffs_sb_swap(copyfs, copyfs);
    855    1.1   hannken 		ffs_csum_swap(space, space, fs->fs_cssize);
    856    1.1   hannken 	}
    857    1.8   hannken #endif
    858   1.75   hannken 	error = UFS_WAPBL_BEGIN(mp);
    859   1.79   hannken 	if (error)
    860   1.79   hannken 		return error;
    861    1.1   hannken 	for (loc = 0; loc < len; loc++) {
    862  1.139      maxv 		error = bread(vp, blkno + loc, fs->fs_bsize,
    863   1.79   hannken 		    B_MODIFY, &bp);
    864   1.15   hannken 		if (error) {
    865   1.79   hannken 			break;
    866    1.1   hannken 		}
    867   1.95   tsutsui 		memcpy(bp->b_data, space, fs->fs_bsize);
    868    1.1   hannken 		space = (char *)space + fs->fs_bsize;
    869   1.79   hannken 		bawrite(bp);
    870   1.79   hannken 	}
    871   1.79   hannken 	if (error)
    872   1.79   hannken 		goto out;
    873  1.126  dholland 	error = bread(vp, ffs_lblkno(fs, fs->fs_sblockloc),
    874  1.139      maxv 	    fs->fs_bsize, B_MODIFY, &bp);
    875   1.79   hannken 	if (error) {
    876   1.79   hannken 		goto out;
    877   1.79   hannken 	} else {
    878   1.95   tsutsui 		memcpy(bp->b_data, sbbuf, fs->fs_bsize);
    879   1.79   hannken 		bawrite(bp);
    880    1.1   hannken 	}
    881    1.1   hannken 	/*
    882  1.121  dholland 	 * Copy the first UFS_NDADDR blocks to the snapshot so
    883  1.121  dholland 	 * ffs_copyonwrite() and ffs_snapblkfree() will always work on
    884  1.121  dholland 	 * indirect blocks.
    885   1.75   hannken 	 */
    886  1.121  dholland 	for (loc = 0; loc < UFS_NDADDR; loc++) {
    887   1.75   hannken 		if (db_get(ip, loc) != 0)
    888   1.75   hannken 			continue;
    889  1.126  dholland 		error = ffs_balloc(vp, ffs_lblktosize(fs, (off_t)loc),
    890   1.79   hannken 		    fs->fs_bsize, l->l_cred, 0, &bp);
    891   1.75   hannken 		if (error)
    892   1.75   hannken 			break;
    893   1.79   hannken 		error = rwfsblk(vp, B_READ, bp->b_data, loc);
    894   1.75   hannken 		if (error) {
    895   1.79   hannken 			brelse(bp, 0);
    896   1.79   hannken 			break;
    897   1.75   hannken 		}
    898   1.79   hannken 		bawrite(bp);
    899   1.75   hannken 	}
    900   1.79   hannken 
    901   1.79   hannken out:
    902   1.75   hannken 	UFS_WAPBL_END(mp);
    903   1.79   hannken 	return error;
    904   1.79   hannken }
    905   1.79   hannken 
    906   1.79   hannken /*
    907   1.79   hannken  * Copy all cylinder group maps.
    908   1.79   hannken  */
    909   1.79   hannken static int
    910   1.79   hannken cgaccount(struct vnode *vp, int passno, int *redo)
    911   1.79   hannken {
    912  1.103   hannken 	int cg, error = 0;
    913   1.79   hannken 	struct buf *nbp;
    914   1.79   hannken 	struct fs *fs = VTOI(vp)->i_fs;
    915   1.79   hannken 
    916   1.79   hannken 	if (redo != NULL)
    917   1.79   hannken 		*redo = 0;
    918   1.79   hannken 	if (passno == 1)
    919   1.79   hannken 		fs->fs_active = malloc(howmany(fs->fs_ncg, NBBY),
    920   1.79   hannken 		    M_DEVBUF, M_WAITOK | M_ZERO);
    921   1.79   hannken 	for (cg = 0; cg < fs->fs_ncg; cg++) {
    922   1.79   hannken 		if (passno == 2 && ACTIVECG_ISSET(fs, cg))
    923   1.79   hannken 			continue;
    924  1.103   hannken 
    925   1.79   hannken 		if (redo != NULL)
    926   1.79   hannken 			*redo += 1;
    927  1.103   hannken 		error = UFS_WAPBL_BEGIN(vp->v_mount);
    928  1.103   hannken 		if (error)
    929  1.103   hannken 			return error;
    930  1.126  dholland 		error = ffs_balloc(vp, ffs_lfragtosize(fs, cgtod(fs, cg)),
    931   1.79   hannken 		    fs->fs_bsize, curlwp->l_cred, 0, &nbp);
    932  1.103   hannken 		if (error) {
    933  1.103   hannken 			UFS_WAPBL_END(vp->v_mount);
    934   1.80   hannken 			break;
    935  1.103   hannken 		}
    936   1.79   hannken 		error = cgaccount1(cg, vp, nbp->b_data, passno);
    937   1.79   hannken 		bawrite(nbp);
    938  1.103   hannken 		UFS_WAPBL_END(vp->v_mount);
    939   1.75   hannken 		if (error)
    940   1.80   hannken 			break;
    941   1.15   hannken 	}
    942   1.80   hannken 	return error;
    943    1.1   hannken }
    944    1.1   hannken 
    945    1.1   hannken /*
    946    1.1   hannken  * Copy a cylinder group map. All the unallocated blocks are marked
    947    1.1   hannken  * BLK_NOCOPY so that the snapshot knows that it need not copy them
    948    1.1   hannken  * if they are later written. If passno is one, then this is a first
    949    1.1   hannken  * pass, so only setting needs to be done. If passno is 2, then this
    950    1.1   hannken  * is a revision to a previous pass which must be undone as the
    951    1.1   hannken  * replacement pass is done.
    952    1.1   hannken  */
    953    1.1   hannken static int
    954   1.79   hannken cgaccount1(int cg, struct vnode *vp, void *data, int passno)
    955    1.1   hannken {
    956    1.1   hannken 	struct buf *bp, *ibp;
    957    1.1   hannken 	struct inode *ip;
    958    1.1   hannken 	struct cg *cgp;
    959    1.1   hannken 	struct fs *fs;
    960   1.79   hannken 	struct lwp *l = curlwp;
    961   1.79   hannken 	daddr_t base, numblks;
    962  1.131    martin 	int error, len, loc, ns __unused, indiroff;
    963    1.1   hannken 
    964    1.1   hannken 	ip = VTOI(vp);
    965    1.1   hannken 	fs = ip->i_fs;
    966    1.1   hannken 	ns = UFS_FSNEEDSWAP(fs);
    967  1.125  dholland 	error = bread(ip->i_devvp, FFS_FSBTODB(fs, cgtod(fs, cg)),
    968  1.139      maxv 		(int)fs->fs_cgsize, 0, &bp);
    969    1.1   hannken 	if (error) {
    970    1.1   hannken 		return (error);
    971    1.1   hannken 	}
    972    1.1   hannken 	cgp = (struct cg *)bp->b_data;
    973    1.1   hannken 	if (!cg_chkmagic(cgp, ns)) {
    974   1.52        ad 		brelse(bp, 0);
    975    1.1   hannken 		return (EIO);
    976    1.1   hannken 	}
    977    1.1   hannken 	ACTIVECG_SET(fs, cg);
    978    1.1   hannken 
    979   1.95   tsutsui 	memcpy(data, bp->b_data, fs->fs_cgsize);
    980   1.52        ad 	brelse(bp, 0);
    981    1.1   hannken 	if (fs->fs_cgsize < fs->fs_bsize)
    982   1.43  christos 		memset((char *)data + fs->fs_cgsize, 0,
    983    1.1   hannken 		    fs->fs_bsize - fs->fs_cgsize);
    984    1.1   hannken 	numblks = howmany(fs->fs_size, fs->fs_frag);
    985    1.1   hannken 	len = howmany(fs->fs_fpg, fs->fs_frag);
    986    1.1   hannken 	base = cg * fs->fs_fpg / fs->fs_frag;
    987    1.1   hannken 	if (base + len >= numblks)
    988    1.1   hannken 		len = numblks - base - 1;
    989    1.1   hannken 	loc = 0;
    990  1.121  dholland 	if (base < UFS_NDADDR) {
    991  1.121  dholland 		for ( ; loc < UFS_NDADDR; loc++) {
    992    1.1   hannken 			if (ffs_isblock(fs, cg_blksfree(cgp, ns), loc))
    993    1.1   hannken 				db_assign(ip, loc, BLK_NOCOPY);
    994    1.1   hannken 			else if (db_get(ip, loc) == BLK_NOCOPY) {
    995    1.1   hannken 				if (passno == 2)
    996    1.1   hannken 					db_assign(ip, loc, 0);
    997    1.1   hannken 				else if (passno == 1)
    998    1.1   hannken 					panic("ffs_snapshot: lost direct block");
    999    1.1   hannken 			}
   1000    1.1   hannken 		}
   1001    1.1   hannken 	}
   1002  1.126  dholland 	if ((error = ffs_balloc(vp, ffs_lblktosize(fs, (off_t)(base + loc)),
   1003   1.79   hannken 	    fs->fs_bsize, l->l_cred, B_METAONLY, &ibp)) != 0)
   1004    1.1   hannken 		return (error);
   1005  1.124  dholland 	indiroff = (base + loc - UFS_NDADDR) % FFS_NINDIR(fs);
   1006    1.1   hannken 	for ( ; loc < len; loc++, indiroff++) {
   1007  1.124  dholland 		if (indiroff >= FFS_NINDIR(fs)) {
   1008   1.15   hannken 			bawrite(ibp);
   1009   1.22      yamt 			if ((error = ffs_balloc(vp,
   1010  1.126  dholland 			    ffs_lblktosize(fs, (off_t)(base + loc)),
   1011   1.79   hannken 			    fs->fs_bsize, l->l_cred, B_METAONLY, &ibp)) != 0)
   1012    1.1   hannken 				return (error);
   1013    1.1   hannken 			indiroff = 0;
   1014    1.1   hannken 		}
   1015    1.1   hannken 		if (ffs_isblock(fs, cg_blksfree(cgp, ns), loc))
   1016    1.1   hannken 			idb_assign(ip, ibp->b_data, indiroff, BLK_NOCOPY);
   1017    1.1   hannken 		else if (idb_get(ip, ibp->b_data, indiroff) == BLK_NOCOPY) {
   1018    1.1   hannken 			if (passno == 2)
   1019    1.1   hannken 				idb_assign(ip, ibp->b_data, indiroff, 0);
   1020    1.1   hannken 			else if (passno == 1)
   1021    1.1   hannken 				panic("ffs_snapshot: lost indirect block");
   1022    1.1   hannken 		}
   1023    1.1   hannken 	}
   1024   1.15   hannken 	bdwrite(ibp);
   1025    1.1   hannken 	return (0);
   1026    1.1   hannken }
   1027    1.1   hannken 
   1028    1.1   hannken /*
   1029    1.1   hannken  * Before expunging a snapshot inode, note all the
   1030    1.1   hannken  * blocks that it claims with BLK_SNAP so that fsck will
   1031    1.1   hannken  * be able to account for those blocks properly and so
   1032    1.1   hannken  * that this snapshot knows that it need not copy them
   1033   1.76   hannken  * if the other snapshot holding them is freed.
   1034    1.1   hannken  */
   1035    1.1   hannken static int
   1036   1.76   hannken expunge(struct vnode *snapvp, struct inode *cancelip, struct fs *fs,
   1037   1.76   hannken     acctfunc_t acctfunc, int expungetype)
   1038    1.1   hannken {
   1039  1.131    martin 	int i, error, ns __unused;
   1040   1.76   hannken 	daddr_t lbn, rlbn;
   1041   1.76   hannken 	daddr_t len, blkno, numblks, blksperindir;
   1042   1.76   hannken 	struct ufs1_dinode *dip1;
   1043   1.76   hannken 	struct ufs2_dinode *dip2;
   1044   1.79   hannken 	struct lwp *l = curlwp;
   1045   1.76   hannken 	void *bap;
   1046    1.1   hannken 	struct buf *bp;
   1047  1.103   hannken 	struct mount *mp;
   1048    1.1   hannken 
   1049    1.1   hannken 	ns = UFS_FSNEEDSWAP(fs);
   1050  1.103   hannken 	mp = snapvp->v_mount;
   1051  1.103   hannken 
   1052  1.103   hannken 	error = UFS_WAPBL_BEGIN(mp);
   1053  1.103   hannken 	if (error)
   1054  1.103   hannken 		return error;
   1055    1.1   hannken 	/*
   1056    1.1   hannken 	 * Prepare to expunge the inode. If its inode block has not
   1057    1.1   hannken 	 * yet been copied, then allocate and fill the copy.
   1058    1.1   hannken 	 */
   1059  1.127  dholland 	lbn = ffs_fragstoblks(fs, ino_to_fsba(fs, cancelip->i_number));
   1060   1.72   hannken 	error = snapblkaddr(snapvp, lbn, &blkno);
   1061   1.71   hannken 	if (error)
   1062   1.71   hannken 		return error;
   1063   1.71   hannken 	if (blkno != 0) {
   1064  1.139      maxv 		error = bread(snapvp, lbn, fs->fs_bsize,
   1065   1.71   hannken 		    B_MODIFY, &bp);
   1066    1.1   hannken 	} else {
   1067  1.126  dholland 		error = ffs_balloc(snapvp, ffs_lblktosize(fs, (off_t)lbn),
   1068   1.79   hannken 		    fs->fs_bsize, l->l_cred, 0, &bp);
   1069   1.71   hannken 		if (! error)
   1070   1.75   hannken 			error = rwfsblk(snapvp, B_READ, bp->b_data, lbn);
   1071    1.1   hannken 	}
   1072  1.103   hannken 	if (error) {
   1073  1.103   hannken 		UFS_WAPBL_END(mp);
   1074    1.1   hannken 		return error;
   1075  1.103   hannken 	}
   1076    1.1   hannken 	/*
   1077    1.1   hannken 	 * Set a snapshot inode to be a zero length file, regular files
   1078   1.48   hannken 	 * or unlinked snapshots to be completely unallocated.
   1079    1.1   hannken 	 */
   1080   1.76   hannken 	if (fs->fs_magic == FS_UFS1_MAGIC) {
   1081   1.76   hannken 		dip1 = (struct ufs1_dinode *)bp->b_data +
   1082   1.76   hannken 		    ino_to_fsbo(fs, cancelip->i_number);
   1083  1.111    bouyer 		if (cancelip->i_flags & SF_SNAPSHOT) {
   1084  1.111    bouyer 			dip1->di_flags =
   1085  1.111    bouyer 			    ufs_rw32(ufs_rw32(dip1->di_flags, ns) |
   1086  1.111    bouyer 			    SF_SNAPINVAL, ns);
   1087  1.111    bouyer 		}
   1088   1.92        ad 		if (expungetype == BLK_NOCOPY || cancelip->i_nlink == 0)
   1089   1.76   hannken 			dip1->di_mode = 0;
   1090   1.76   hannken 		dip1->di_size = 0;
   1091   1.76   hannken 		dip1->di_blocks = 0;
   1092  1.121  dholland 		memset(&dip1->di_db[0], 0, (UFS_NDADDR + UFS_NIADDR) * sizeof(int32_t));
   1093   1.76   hannken 	} else {
   1094   1.76   hannken 		dip2 = (struct ufs2_dinode *)bp->b_data +
   1095   1.76   hannken 		    ino_to_fsbo(fs, cancelip->i_number);
   1096  1.111    bouyer 		if (cancelip->i_flags & SF_SNAPSHOT) {
   1097  1.111    bouyer 			dip2->di_flags =
   1098  1.111    bouyer 			    ufs_rw32(ufs_rw32(dip2->di_flags, ns) |
   1099  1.111    bouyer 			    SF_SNAPINVAL, ns);
   1100  1.111    bouyer 		}
   1101   1.92        ad 		if (expungetype == BLK_NOCOPY || cancelip->i_nlink == 0)
   1102   1.76   hannken 			dip2->di_mode = 0;
   1103   1.76   hannken 		dip2->di_size = 0;
   1104   1.76   hannken 		dip2->di_blocks = 0;
   1105  1.121  dholland 		memset(&dip2->di_db[0], 0, (UFS_NDADDR + UFS_NIADDR) * sizeof(int64_t));
   1106   1.76   hannken 	}
   1107   1.71   hannken 	bdwrite(bp);
   1108  1.103   hannken 	UFS_WAPBL_END(mp);
   1109    1.1   hannken 	/*
   1110    1.1   hannken 	 * Now go through and expunge all the blocks in the file
   1111    1.1   hannken 	 * using the function requested.
   1112    1.1   hannken 	 */
   1113    1.1   hannken 	numblks = howmany(cancelip->i_size, fs->fs_bsize);
   1114   1.76   hannken 	if (fs->fs_magic == FS_UFS1_MAGIC)
   1115   1.76   hannken 		bap = &cancelip->i_ffs1_db[0];
   1116   1.76   hannken 	else
   1117   1.76   hannken 		bap = &cancelip->i_ffs2_db[0];
   1118  1.121  dholland 	error = (*acctfunc)(snapvp, bap, 0, UFS_NDADDR, fs, 0, expungetype);
   1119  1.103   hannken 	if (error)
   1120    1.1   hannken 		return (error);
   1121   1.76   hannken 	if (fs->fs_magic == FS_UFS1_MAGIC)
   1122   1.76   hannken 		bap = &cancelip->i_ffs1_ib[0];
   1123   1.76   hannken 	else
   1124   1.76   hannken 		bap = &cancelip->i_ffs2_ib[0];
   1125  1.121  dholland 	error = (*acctfunc)(snapvp, bap, 0, UFS_NIADDR, fs, -1, expungetype);
   1126  1.103   hannken 	if (error)
   1127    1.1   hannken 		return (error);
   1128    1.1   hannken 	blksperindir = 1;
   1129  1.121  dholland 	lbn = -UFS_NDADDR;
   1130  1.121  dholland 	len = numblks - UFS_NDADDR;
   1131  1.121  dholland 	rlbn = UFS_NDADDR;
   1132  1.121  dholland 	for (i = 0; len > 0 && i < UFS_NIADDR; i++) {
   1133   1.76   hannken 		error = indiracct(snapvp, ITOV(cancelip), i,
   1134   1.76   hannken 		    ib_get(cancelip, i), lbn, rlbn, len,
   1135    1.1   hannken 		    blksperindir, fs, acctfunc, expungetype);
   1136    1.1   hannken 		if (error)
   1137    1.1   hannken 			return (error);
   1138  1.124  dholland 		blksperindir *= FFS_NINDIR(fs);
   1139    1.1   hannken 		lbn -= blksperindir + 1;
   1140    1.1   hannken 		len -= blksperindir;
   1141    1.1   hannken 		rlbn += blksperindir;
   1142    1.1   hannken 	}
   1143    1.1   hannken 	return (0);
   1144    1.1   hannken }
   1145    1.1   hannken 
   1146    1.1   hannken /*
   1147    1.1   hannken  * Descend an indirect block chain for vnode cancelvp accounting for all
   1148    1.1   hannken  * its indirect blocks in snapvp.
   1149   1.11     perry  */
   1150    1.1   hannken static int
   1151   1.76   hannken indiracct(struct vnode *snapvp, struct vnode *cancelvp, int level,
   1152   1.76   hannken     daddr_t blkno, daddr_t lbn, daddr_t rlbn, daddr_t remblks,
   1153   1.76   hannken     daddr_t blksperindir, struct fs *fs, acctfunc_t acctfunc, int expungetype)
   1154    1.1   hannken {
   1155   1.76   hannken 	int error, num, i;
   1156   1.76   hannken 	daddr_t subblksperindir;
   1157  1.121  dholland 	struct indir indirs[UFS_NIADDR + 2];
   1158   1.76   hannken 	daddr_t last;
   1159   1.76   hannken 	void *bap;
   1160    1.1   hannken 	struct buf *bp;
   1161    1.1   hannken 
   1162    1.1   hannken 	if (blkno == 0) {
   1163    1.1   hannken 		if (expungetype == BLK_NOCOPY)
   1164    1.1   hannken 			return (0);
   1165   1.76   hannken 		panic("indiracct: missing indir");
   1166    1.1   hannken 	}
   1167    1.1   hannken 	if ((error = ufs_getlbns(cancelvp, rlbn, indirs, &num)) != 0)
   1168    1.1   hannken 		return (error);
   1169    1.1   hannken 	if (lbn != indirs[num - 1 - level].in_lbn || num < 2)
   1170   1.76   hannken 		panic("indiracct: botched params");
   1171    1.1   hannken 	/*
   1172    1.1   hannken 	 * We have to expand bread here since it will deadlock looking
   1173    1.1   hannken 	 * up the block number for any blocks that are not in the cache.
   1174    1.1   hannken 	 */
   1175  1.125  dholland 	error = ffs_getblk(cancelvp, lbn, FFS_FSBTODB(fs, blkno), fs->fs_bsize,
   1176   1.69   hannken 	    false, &bp);
   1177   1.69   hannken 	if (error)
   1178   1.69   hannken 		return error;
   1179   1.72   hannken 	if ((bp->b_oflags & (BO_DONE | BO_DELWRI)) == 0 && (error =
   1180  1.127  dholland 	    rwfsblk(bp->b_vp, B_READ, bp->b_data, ffs_fragstoblks(fs, blkno)))) {
   1181   1.52        ad 		brelse(bp, 0);
   1182    1.1   hannken 		return (error);
   1183    1.1   hannken 	}
   1184    1.1   hannken 	/*
   1185    1.1   hannken 	 * Account for the block pointers in this indirect block.
   1186    1.1   hannken 	 */
   1187    1.1   hannken 	last = howmany(remblks, blksperindir);
   1188  1.124  dholland 	if (last > FFS_NINDIR(fs))
   1189  1.124  dholland 		last = FFS_NINDIR(fs);
   1190   1.88    cegger 	bap = malloc(fs->fs_bsize, M_DEVBUF, M_WAITOK | M_ZERO);
   1191   1.95   tsutsui 	memcpy((void *)bap, bp->b_data, fs->fs_bsize);
   1192   1.52        ad 	brelse(bp, 0);
   1193   1.76   hannken 	error = (*acctfunc)(snapvp, bap, 0, last,
   1194   1.76   hannken 	    fs, level == 0 ? rlbn : -1, expungetype);
   1195    1.1   hannken 	if (error || level == 0)
   1196    1.1   hannken 		goto out;
   1197    1.1   hannken 	/*
   1198    1.1   hannken 	 * Account for the block pointers in each of the indirect blocks
   1199    1.1   hannken 	 * in the levels below us.
   1200    1.1   hannken 	 */
   1201  1.124  dholland 	subblksperindir = blksperindir / FFS_NINDIR(fs);
   1202    1.1   hannken 	for (lbn++, level--, i = 0; i < last; i++) {
   1203   1.76   hannken 		error = indiracct(snapvp, cancelvp, level,
   1204   1.76   hannken 		    idb_get(VTOI(snapvp), bap, i), lbn, rlbn, remblks,
   1205   1.76   hannken 		    subblksperindir, fs, acctfunc, expungetype);
   1206    1.1   hannken 		if (error)
   1207    1.1   hannken 			goto out;
   1208    1.1   hannken 		rlbn += blksperindir;
   1209    1.1   hannken 		lbn -= blksperindir;
   1210    1.1   hannken 		remblks -= blksperindir;
   1211    1.1   hannken 	}
   1212    1.1   hannken out:
   1213   1.88    cegger 	free(bap, M_DEVBUF);
   1214    1.1   hannken 	return (error);
   1215    1.1   hannken }
   1216    1.1   hannken 
   1217    1.1   hannken /*
   1218    1.1   hannken  * Do both snap accounting and map accounting.
   1219    1.1   hannken  */
   1220    1.1   hannken static int
   1221   1.76   hannken fullacct(struct vnode *vp, void *bap, int oldblkp, int lastblkp,
   1222   1.76   hannken     struct fs *fs, daddr_t lblkno,
   1223   1.18   thorpej     int exptype /* BLK_SNAP or BLK_NOCOPY */)
   1224    1.1   hannken {
   1225    1.1   hannken 	int error;
   1226    1.1   hannken 
   1227   1.76   hannken 	if ((error = snapacct(vp, bap, oldblkp, lastblkp, fs, lblkno, exptype)))
   1228    1.1   hannken 		return (error);
   1229   1.76   hannken 	return (mapacct(vp, bap, oldblkp, lastblkp, fs, lblkno, exptype));
   1230    1.1   hannken }
   1231    1.1   hannken 
   1232    1.1   hannken /*
   1233    1.1   hannken  * Identify a set of blocks allocated in a snapshot inode.
   1234    1.1   hannken  */
   1235    1.1   hannken static int
   1236   1.76   hannken snapacct(struct vnode *vp, void *bap, int oldblkp, int lastblkp,
   1237   1.76   hannken     struct fs *fs, daddr_t lblkno,
   1238   1.18   thorpej     int expungetype /* BLK_SNAP or BLK_NOCOPY */)
   1239    1.1   hannken {
   1240    1.1   hannken 	struct inode *ip = VTOI(vp);
   1241   1.79   hannken 	struct lwp *l = curlwp;
   1242  1.103   hannken 	struct mount *mp = vp->v_mount;
   1243   1.76   hannken 	daddr_t blkno;
   1244   1.76   hannken 	daddr_t lbn;
   1245    1.1   hannken 	struct buf *ibp;
   1246  1.103   hannken 	int error, n;
   1247  1.103   hannken 	const int wbreak = blocks_in_journal(VFSTOUFS(mp)->um_fs)/8;
   1248    1.1   hannken 
   1249  1.103   hannken 	error = UFS_WAPBL_BEGIN(mp);
   1250  1.103   hannken 	if (error)
   1251  1.103   hannken 		return error;
   1252  1.103   hannken 	for ( n = 0; oldblkp < lastblkp; oldblkp++) {
   1253   1.76   hannken 		blkno = idb_get(ip, bap, oldblkp);
   1254    1.1   hannken 		if (blkno == 0 || blkno == BLK_NOCOPY || blkno == BLK_SNAP)
   1255    1.1   hannken 			continue;
   1256  1.127  dholland 		lbn = ffs_fragstoblks(fs, blkno);
   1257  1.121  dholland 		if (lbn < UFS_NDADDR) {
   1258   1.76   hannken 			blkno = db_get(ip, lbn);
   1259    1.1   hannken 			ip->i_flag |= IN_CHANGE | IN_UPDATE;
   1260    1.1   hannken 		} else {
   1261  1.126  dholland 			error = ffs_balloc(vp, ffs_lblktosize(fs, (off_t)lbn),
   1262   1.79   hannken 			    fs->fs_bsize, l->l_cred, B_METAONLY, &ibp);
   1263    1.1   hannken 			if (error)
   1264  1.103   hannken 				break;
   1265   1.76   hannken 			blkno = idb_get(ip, ibp->b_data,
   1266  1.124  dholland 			    (lbn - UFS_NDADDR) % FFS_NINDIR(fs));
   1267    1.1   hannken 		}
   1268    1.1   hannken 		/*
   1269    1.1   hannken 		 * If we are expunging a snapshot vnode and we
   1270    1.1   hannken 		 * find a block marked BLK_NOCOPY, then it is
   1271    1.1   hannken 		 * one that has been allocated to this snapshot after
   1272    1.1   hannken 		 * we took our current snapshot and can be ignored.
   1273    1.1   hannken 		 */
   1274    1.1   hannken 		if (expungetype == BLK_SNAP && blkno == BLK_NOCOPY) {
   1275  1.121  dholland 			if (lbn >= UFS_NDADDR)
   1276   1.52        ad 				brelse(ibp, 0);
   1277    1.1   hannken 		} else {
   1278    1.1   hannken 			if (blkno != 0)
   1279   1.76   hannken 				panic("snapacct: bad block");
   1280  1.121  dholland 			if (lbn < UFS_NDADDR)
   1281   1.76   hannken 				db_assign(ip, lbn, expungetype);
   1282   1.76   hannken 			else {
   1283   1.76   hannken 				idb_assign(ip, ibp->b_data,
   1284  1.124  dholland 				    (lbn - UFS_NDADDR) % FFS_NINDIR(fs), expungetype);
   1285   1.15   hannken 				bdwrite(ibp);
   1286   1.76   hannken 			}
   1287    1.1   hannken 		}
   1288  1.103   hannken 		if (wbreak > 0 && (++n % wbreak) == 0) {
   1289  1.103   hannken 			UFS_WAPBL_END(mp);
   1290  1.103   hannken 			error = UFS_WAPBL_BEGIN(mp);
   1291  1.103   hannken 			if (error)
   1292  1.103   hannken 				return error;
   1293  1.103   hannken 		}
   1294    1.1   hannken 	}
   1295  1.103   hannken 	UFS_WAPBL_END(mp);
   1296  1.103   hannken 	return error;
   1297    1.1   hannken }
   1298    1.1   hannken 
   1299    1.1   hannken /*
   1300    1.1   hannken  * Account for a set of blocks allocated in a snapshot inode.
   1301    1.1   hannken  */
   1302    1.1   hannken static int
   1303   1.76   hannken mapacct(struct vnode *vp, void *bap, int oldblkp, int lastblkp,
   1304   1.76   hannken     struct fs *fs, daddr_t lblkno, int expungetype)
   1305    1.1   hannken {
   1306   1.76   hannken 	daddr_t blkno;
   1307    1.1   hannken 	struct inode *ip;
   1308  1.103   hannken 	struct mount *mp = vp->v_mount;
   1309    1.1   hannken 	ino_t inum;
   1310  1.103   hannken 	int acctit, error, n;
   1311  1.103   hannken 	const int wbreak = blocks_in_journal(VFSTOUFS(mp)->um_fs)/8;
   1312    1.1   hannken 
   1313  1.103   hannken 	error = UFS_WAPBL_BEGIN(mp);
   1314  1.103   hannken 	if (error)
   1315  1.103   hannken 		return error;
   1316    1.1   hannken 	ip = VTOI(vp);
   1317    1.1   hannken 	inum = ip->i_number;
   1318    1.1   hannken 	if (lblkno == -1)
   1319    1.1   hannken 		acctit = 0;
   1320    1.1   hannken 	else
   1321    1.1   hannken 		acctit = 1;
   1322  1.103   hannken 	for ( n = 0; oldblkp < lastblkp; oldblkp++, lblkno++) {
   1323   1.76   hannken 		blkno = idb_get(ip, bap, oldblkp);
   1324    1.1   hannken 		if (blkno == 0 || blkno == BLK_NOCOPY)
   1325    1.1   hannken 			continue;
   1326    1.1   hannken 		if (acctit && expungetype == BLK_SNAP && blkno != BLK_SNAP)
   1327   1.15   hannken 			*ip->i_snapblklist++ = lblkno;
   1328    1.1   hannken 		if (blkno == BLK_SNAP)
   1329  1.127  dholland 			blkno = ffs_blkstofrags(fs, lblkno);
   1330   1.83     joerg 		ffs_blkfree_snap(fs, vp, blkno, fs->fs_bsize, inum);
   1331  1.103   hannken 		if (wbreak > 0 && (++n % wbreak) == 0) {
   1332  1.103   hannken 			UFS_WAPBL_END(mp);
   1333  1.103   hannken 			error = UFS_WAPBL_BEGIN(mp);
   1334  1.103   hannken 			if (error)
   1335  1.103   hannken 				return error;
   1336  1.103   hannken 		}
   1337    1.1   hannken 	}
   1338  1.103   hannken 	UFS_WAPBL_END(mp);
   1339    1.1   hannken 	return (0);
   1340    1.1   hannken }
   1341  1.107        he 
   1342  1.107        he /*
   1343  1.107        he  * Number of blocks that fit into the journal or zero if not logging.
   1344  1.107        he  */
   1345  1.107        he static int
   1346  1.107        he blocks_in_journal(struct fs *fs)
   1347  1.107        he {
   1348  1.107        he 	off_t bpj;
   1349  1.107        he 
   1350  1.107        he 	if ((fs->fs_flags & FS_DOWAPBL) == 0)
   1351  1.107        he 		return 0;
   1352  1.107        he 	bpj = 1;
   1353  1.107        he 	if (fs->fs_journal_version == UFS_WAPBL_VERSION) {
   1354  1.107        he 		switch (fs->fs_journal_location) {
   1355  1.107        he 		case UFS_WAPBL_JOURNALLOC_END_PARTITION:
   1356  1.107        he 			bpj = (off_t)fs->fs_journallocs[UFS_WAPBL_EPART_BLKSZ]*
   1357  1.107        he 			    fs->fs_journallocs[UFS_WAPBL_EPART_COUNT];
   1358  1.107        he 			break;
   1359  1.107        he 		case UFS_WAPBL_JOURNALLOC_IN_FILESYSTEM:
   1360  1.107        he 			bpj = (off_t)fs->fs_journallocs[UFS_WAPBL_INFS_BLKSZ]*
   1361  1.107        he 			    fs->fs_journallocs[UFS_WAPBL_INFS_COUNT];
   1362  1.107        he 			break;
   1363  1.107        he 		}
   1364  1.107        he 	}
   1365  1.107        he 	bpj /= fs->fs_bsize;
   1366  1.107        he 	return (bpj > 0 ? bpj : 1);
   1367  1.107        he }
   1368   1.10   hannken #endif /* defined(FFS_NO_SNAPSHOT) */
   1369    1.1   hannken 
   1370    1.1   hannken /*
   1371    1.1   hannken  * Decrement extra reference on snapshot when last name is removed.
   1372    1.1   hannken  * It will not be freed until the last open reference goes away.
   1373    1.1   hannken  */
   1374    1.1   hannken void
   1375  1.123   hannken ffs_snapgone(struct vnode *vp)
   1376    1.1   hannken {
   1377  1.123   hannken 	struct inode *xp, *ip = VTOI(vp);
   1378  1.129   hannken 	struct mount *mp = spec_node_getmountedfs(ip->i_devvp);
   1379    1.1   hannken 	struct fs *fs;
   1380   1.47   hannken 	struct snap_info *si;
   1381    1.1   hannken 	int snaploc;
   1382    1.1   hannken 
   1383   1.66   hannken 	si = VFSTOUFS(mp)->um_snapinfo;
   1384   1.66   hannken 
   1385    1.1   hannken 	/*
   1386    1.1   hannken 	 * Find snapshot in incore list.
   1387    1.1   hannken 	 */
   1388   1.49   hannken 	mutex_enter(&si->si_lock);
   1389   1.47   hannken 	TAILQ_FOREACH(xp, &si->si_snapshots, i_nextsnap)
   1390    1.1   hannken 		if (xp == ip)
   1391    1.1   hannken 			break;
   1392   1.49   hannken 	mutex_exit(&si->si_lock);
   1393  1.104   hannken 	if (xp != NULL)
   1394  1.104   hannken 		vrele(ITOV(ip));
   1395    1.1   hannken #ifdef DEBUG
   1396  1.104   hannken 	else if (snapdebug)
   1397   1.19  christos 		printf("ffs_snapgone: lost snapshot vnode %llu\n",
   1398   1.19  christos 		    (unsigned long long)ip->i_number);
   1399    1.1   hannken #endif
   1400    1.1   hannken 	/*
   1401    1.1   hannken 	 * Delete snapshot inode from superblock. Keep list dense.
   1402    1.1   hannken 	 */
   1403   1.49   hannken 	mutex_enter(&si->si_lock);
   1404    1.1   hannken 	fs = ip->i_fs;
   1405    1.1   hannken 	for (snaploc = 0; snaploc < FSMAXSNAP; snaploc++)
   1406    1.1   hannken 		if (fs->fs_snapinum[snaploc] == ip->i_number)
   1407    1.1   hannken 			break;
   1408    1.1   hannken 	if (snaploc < FSMAXSNAP) {
   1409    1.1   hannken 		for (snaploc++; snaploc < FSMAXSNAP; snaploc++) {
   1410    1.1   hannken 			if (fs->fs_snapinum[snaploc] == 0)
   1411    1.1   hannken 				break;
   1412    1.1   hannken 			fs->fs_snapinum[snaploc - 1] = fs->fs_snapinum[snaploc];
   1413    1.1   hannken 		}
   1414    1.1   hannken 		fs->fs_snapinum[snaploc - 1] = 0;
   1415    1.1   hannken 	}
   1416   1.49   hannken 	si->si_gen++;
   1417   1.49   hannken 	mutex_exit(&si->si_lock);
   1418    1.1   hannken }
   1419    1.1   hannken 
   1420    1.1   hannken /*
   1421    1.1   hannken  * Prepare a snapshot file for being removed.
   1422    1.1   hannken  */
   1423    1.1   hannken void
   1424   1.18   thorpej ffs_snapremove(struct vnode *vp)
   1425    1.1   hannken {
   1426   1.15   hannken 	struct inode *ip = VTOI(vp), *xp;
   1427    1.1   hannken 	struct vnode *devvp = ip->i_devvp;
   1428    1.1   hannken 	struct fs *fs = ip->i_fs;
   1429  1.129   hannken 	struct mount *mp = spec_node_getmountedfs(devvp);
   1430    1.1   hannken 	struct buf *ibp;
   1431   1.47   hannken 	struct snap_info *si;
   1432   1.79   hannken 	struct lwp *l = curlwp;
   1433   1.79   hannken 	daddr_t numblks, blkno, dblk;
   1434  1.106   hannken 	int error, loc, last;
   1435    1.1   hannken 
   1436   1.66   hannken 	si = VFSTOUFS(mp)->um_snapinfo;
   1437    1.1   hannken 	/*
   1438    1.1   hannken 	 * If active, delete from incore list (this snapshot may
   1439    1.1   hannken 	 * already have been in the process of being deleted, so
   1440    1.1   hannken 	 * would not have been active).
   1441    1.1   hannken 	 *
   1442    1.1   hannken 	 * Clear copy-on-write flag if last snapshot.
   1443    1.1   hannken 	 */
   1444  1.106   hannken 	mutex_enter(&si->si_snaplock);
   1445   1.97   hannken 	mutex_enter(&si->si_lock);
   1446   1.97   hannken 	if (is_active_snapshot(si, ip)) {
   1447   1.47   hannken 		TAILQ_REMOVE(&si->si_snapshots, ip, i_nextsnap);
   1448   1.47   hannken 		if (TAILQ_FIRST(&si->si_snapshots) != 0) {
   1449   1.15   hannken 			/* Roll back the list of preallocated blocks. */
   1450   1.47   hannken 			xp = TAILQ_LAST(&si->si_snapshots, inodelst);
   1451   1.47   hannken 			si->si_snapblklist = xp->i_snapblklist;
   1452   1.96   hannken 			si->si_gen++;
   1453   1.96   hannken 			mutex_exit(&si->si_lock);
   1454  1.106   hannken 			mutex_exit(&si->si_snaplock);
   1455    1.1   hannken 		} else {
   1456   1.47   hannken 			si->si_snapblklist = 0;
   1457   1.49   hannken 			si->si_gen++;
   1458   1.49   hannken 			mutex_exit(&si->si_lock);
   1459  1.106   hannken 			mutex_exit(&si->si_snaplock);
   1460   1.51   hannken 			fscow_disestablish(mp, ffs_copyonwrite, devvp);
   1461    1.1   hannken 		}
   1462   1.79   hannken 		if (ip->i_snapblklist != NULL) {
   1463   1.79   hannken 			free(ip->i_snapblklist, M_UFSMNT);
   1464   1.79   hannken 			ip->i_snapblklist = NULL;
   1465   1.79   hannken 		}
   1466  1.106   hannken 	} else {
   1467   1.97   hannken 		mutex_exit(&si->si_lock);
   1468  1.106   hannken 		mutex_exit(&si->si_snaplock);
   1469  1.106   hannken 	}
   1470    1.1   hannken 	/*
   1471    1.1   hannken 	 * Clear all BLK_NOCOPY fields. Pass any block claims to other
   1472    1.1   hannken 	 * snapshots that want them (see ffs_snapblkfree below).
   1473    1.1   hannken 	 */
   1474  1.121  dholland 	for (blkno = 1; blkno < UFS_NDADDR; blkno++) {
   1475    1.1   hannken 		dblk = db_get(ip, blkno);
   1476    1.1   hannken 		if (dblk == BLK_NOCOPY || dblk == BLK_SNAP)
   1477    1.1   hannken 			db_assign(ip, blkno, 0);
   1478  1.127  dholland 		else if ((dblk == ffs_blkstofrags(fs, blkno) &&
   1479    1.1   hannken 		     ffs_snapblkfree(fs, ip->i_devvp, dblk, fs->fs_bsize,
   1480    1.1   hannken 		     ip->i_number))) {
   1481    1.1   hannken 			DIP_ADD(ip, blocks, -btodb(fs->fs_bsize));
   1482    1.1   hannken 			db_assign(ip, blkno, 0);
   1483    1.1   hannken 		}
   1484    1.1   hannken 	}
   1485    1.1   hannken 	numblks = howmany(ip->i_size, fs->fs_bsize);
   1486  1.124  dholland 	for (blkno = UFS_NDADDR; blkno < numblks; blkno += FFS_NINDIR(fs)) {
   1487  1.126  dholland 		error = ffs_balloc(vp, ffs_lblktosize(fs, (off_t)blkno),
   1488   1.79   hannken 		    fs->fs_bsize, l->l_cred, B_METAONLY, &ibp);
   1489    1.1   hannken 		if (error)
   1490    1.1   hannken 			continue;
   1491  1.124  dholland 		if (fs->fs_size - blkno > FFS_NINDIR(fs))
   1492  1.124  dholland 			last = FFS_NINDIR(fs);
   1493    1.1   hannken 		else
   1494    1.1   hannken 			last = fs->fs_size - blkno;
   1495    1.1   hannken 		for (loc = 0; loc < last; loc++) {
   1496    1.1   hannken 			dblk = idb_get(ip, ibp->b_data, loc);
   1497    1.1   hannken 			if (dblk == BLK_NOCOPY || dblk == BLK_SNAP)
   1498    1.1   hannken 				idb_assign(ip, ibp->b_data, loc, 0);
   1499  1.127  dholland 			else if (dblk == ffs_blkstofrags(fs, blkno) &&
   1500    1.1   hannken 			    ffs_snapblkfree(fs, ip->i_devvp, dblk,
   1501    1.1   hannken 			    fs->fs_bsize, ip->i_number)) {
   1502    1.1   hannken 				DIP_ADD(ip, blocks, -btodb(fs->fs_bsize));
   1503    1.1   hannken 				idb_assign(ip, ibp->b_data, loc, 0);
   1504    1.1   hannken 			}
   1505    1.1   hannken 		}
   1506   1.15   hannken 		bawrite(ibp);
   1507  1.106   hannken 		UFS_WAPBL_END(mp);
   1508  1.106   hannken 		error = UFS_WAPBL_BEGIN(mp);
   1509  1.106   hannken 		KASSERT(error == 0);
   1510    1.1   hannken 	}
   1511    1.1   hannken 	/*
   1512    1.1   hannken 	 * Clear snapshot flag and drop reference.
   1513    1.1   hannken 	 */
   1514  1.111    bouyer 	ip->i_flags &= ~(SF_SNAPSHOT | SF_SNAPINVAL);
   1515    1.1   hannken 	DIP_ASSIGN(ip, flags, ip->i_flags);
   1516    1.1   hannken 	ip->i_flag |= IN_CHANGE | IN_UPDATE;
   1517  1.111    bouyer #if defined(QUOTA) || defined(QUOTA2)
   1518  1.111    bouyer 	chkdq(ip, DIP(ip, blocks), l->l_cred, FORCE);
   1519  1.111    bouyer 	chkiq(ip, 1, l->l_cred, FORCE);
   1520  1.111    bouyer #endif
   1521    1.1   hannken }
   1522    1.1   hannken 
   1523    1.1   hannken /*
   1524    1.1   hannken  * Notification that a block is being freed. Return zero if the free
   1525    1.1   hannken  * should be allowed to proceed. Return non-zero if the snapshot file
   1526    1.1   hannken  * wants to claim the block. The block will be claimed if it is an
   1527    1.1   hannken  * uncopied part of one of the snapshots. It will be freed if it is
   1528    1.1   hannken  * either a BLK_NOCOPY or has already been copied in all of the snapshots.
   1529    1.1   hannken  * If a fragment is being freed, then all snapshots that care about
   1530    1.1   hannken  * it must make a copy since a snapshot file can only claim full sized
   1531    1.1   hannken  * blocks. Note that if more than one snapshot file maps the block,
   1532    1.1   hannken  * we can pick one at random to claim it. Since none of the snapshots
   1533    1.1   hannken  * can change, we are assurred that they will all see the same unmodified
   1534    1.1   hannken  * image. When deleting a snapshot file (see ffs_snapremove above), we
   1535    1.1   hannken  * must push any of these claimed blocks to one of the other snapshots
   1536    1.1   hannken  * that maps it. These claimed blocks are easily identified as they will
   1537    1.1   hannken  * have a block number equal to their logical block number within the
   1538    1.1   hannken  * snapshot. A copied block can never have this property because they
   1539    1.1   hannken  * must always have been allocated from a BLK_NOCOPY location.
   1540    1.1   hannken  */
   1541    1.1   hannken int
   1542   1.79   hannken ffs_snapblkfree(struct fs *fs, struct vnode *devvp, daddr_t bno,
   1543   1.36  christos     long size, ino_t inum)
   1544    1.1   hannken {
   1545  1.129   hannken 	struct mount *mp = spec_node_getmountedfs(devvp);
   1546    1.1   hannken 	struct buf *ibp;
   1547    1.1   hannken 	struct inode *ip;
   1548   1.49   hannken 	struct vnode *vp = NULL;
   1549   1.47   hannken 	struct snap_info *si;
   1550   1.43  christos 	void *saved_data = NULL;
   1551   1.79   hannken 	daddr_t lbn;
   1552   1.79   hannken 	daddr_t blkno;
   1553   1.49   hannken 	uint32_t gen;
   1554  1.106   hannken 	int indiroff = 0, error = 0, claimedblk = 0;
   1555    1.1   hannken 
   1556   1.66   hannken 	si = VFSTOUFS(mp)->um_snapinfo;
   1557  1.127  dholland 	lbn = ffs_fragstoblks(fs, bno);
   1558  1.106   hannken 	mutex_enter(&si->si_snaplock);
   1559   1.49   hannken 	mutex_enter(&si->si_lock);
   1560  1.106   hannken 	si->si_owner = curlwp;
   1561  1.106   hannken 
   1562    1.1   hannken retry:
   1563   1.49   hannken 	gen = si->si_gen;
   1564   1.47   hannken 	TAILQ_FOREACH(ip, &si->si_snapshots, i_nextsnap) {
   1565    1.1   hannken 		vp = ITOV(ip);
   1566    1.1   hannken 		/*
   1567    1.1   hannken 		 * Lookup block being written.
   1568    1.1   hannken 		 */
   1569  1.121  dholland 		if (lbn < UFS_NDADDR) {
   1570    1.1   hannken 			blkno = db_get(ip, lbn);
   1571    1.1   hannken 		} else {
   1572   1.49   hannken 			mutex_exit(&si->si_lock);
   1573  1.126  dholland 			error = ffs_balloc(vp, ffs_lblktosize(fs, (off_t)lbn),
   1574   1.79   hannken 			    fs->fs_bsize, FSCRED, B_METAONLY, &ibp);
   1575   1.49   hannken 			if (error) {
   1576   1.49   hannken 				mutex_enter(&si->si_lock);
   1577    1.1   hannken 				break;
   1578   1.49   hannken 			}
   1579  1.124  dholland 			indiroff = (lbn - UFS_NDADDR) % FFS_NINDIR(fs);
   1580    1.1   hannken 			blkno = idb_get(ip, ibp->b_data, indiroff);
   1581   1.49   hannken 			mutex_enter(&si->si_lock);
   1582   1.49   hannken 			if (gen != si->si_gen) {
   1583   1.52        ad 				brelse(ibp, 0);
   1584   1.49   hannken 				goto retry;
   1585   1.49   hannken 			}
   1586    1.1   hannken 		}
   1587    1.1   hannken 		/*
   1588    1.1   hannken 		 * Check to see if block needs to be copied.
   1589    1.1   hannken 		 */
   1590    1.1   hannken 		if (blkno == 0) {
   1591    1.1   hannken 			/*
   1592    1.1   hannken 			 * A block that we map is being freed. If it has not
   1593    1.1   hannken 			 * been claimed yet, we will claim or copy it (below).
   1594    1.1   hannken 			 */
   1595    1.1   hannken 			claimedblk = 1;
   1596    1.1   hannken 		} else if (blkno == BLK_SNAP) {
   1597    1.1   hannken 			/*
   1598    1.1   hannken 			 * No previous snapshot claimed the block,
   1599    1.1   hannken 			 * so it will be freed and become a BLK_NOCOPY
   1600    1.1   hannken 			 * (don't care) for us.
   1601    1.1   hannken 			 */
   1602    1.1   hannken 			if (claimedblk)
   1603    1.1   hannken 				panic("snapblkfree: inconsistent block type");
   1604  1.121  dholland 			if (lbn < UFS_NDADDR) {
   1605    1.1   hannken 				db_assign(ip, lbn, BLK_NOCOPY);
   1606    1.1   hannken 				ip->i_flag |= IN_CHANGE | IN_UPDATE;
   1607    1.1   hannken 			} else {
   1608    1.1   hannken 				idb_assign(ip, ibp->b_data, indiroff,
   1609    1.1   hannken 				    BLK_NOCOPY);
   1610   1.49   hannken 				mutex_exit(&si->si_lock);
   1611   1.92        ad 				if (ip->i_nlink > 0)
   1612   1.72   hannken 					bwrite(ibp);
   1613   1.72   hannken 				else
   1614   1.72   hannken 					bdwrite(ibp);
   1615   1.49   hannken 				mutex_enter(&si->si_lock);
   1616   1.49   hannken 				if (gen != si->si_gen)
   1617   1.49   hannken 					goto retry;
   1618    1.1   hannken 			}
   1619    1.1   hannken 			continue;
   1620    1.1   hannken 		} else /* BLK_NOCOPY or default */ {
   1621    1.1   hannken 			/*
   1622    1.1   hannken 			 * If the snapshot has already copied the block
   1623    1.1   hannken 			 * (default), or does not care about the block,
   1624    1.1   hannken 			 * it is not needed.
   1625    1.1   hannken 			 */
   1626  1.121  dholland 			if (lbn >= UFS_NDADDR)
   1627   1.52        ad 				brelse(ibp, 0);
   1628    1.1   hannken 			continue;
   1629    1.1   hannken 		}
   1630    1.1   hannken 		/*
   1631    1.1   hannken 		 * If this is a full size block, we will just grab it
   1632    1.1   hannken 		 * and assign it to the snapshot inode. Otherwise we
   1633    1.1   hannken 		 * will proceed to copy it. See explanation for this
   1634    1.1   hannken 		 * routine as to why only a single snapshot needs to
   1635    1.1   hannken 		 * claim this block.
   1636    1.1   hannken 		 */
   1637    1.1   hannken 		if (size == fs->fs_bsize) {
   1638    1.1   hannken #ifdef DEBUG
   1639    1.1   hannken 			if (snapdebug)
   1640   1.19  christos 				printf("%s %llu lbn %" PRId64
   1641   1.19  christos 				    "from inum %llu\n",
   1642   1.19  christos 				    "Grabonremove: snapino",
   1643   1.19  christos 				    (unsigned long long)ip->i_number,
   1644   1.19  christos 				    lbn, (unsigned long long)inum);
   1645    1.1   hannken #endif
   1646   1.49   hannken 			mutex_exit(&si->si_lock);
   1647  1.121  dholland 			if (lbn < UFS_NDADDR) {
   1648    1.1   hannken 				db_assign(ip, lbn, bno);
   1649    1.1   hannken 			} else {
   1650    1.1   hannken 				idb_assign(ip, ibp->b_data, indiroff, bno);
   1651   1.92        ad 				if (ip->i_nlink > 0)
   1652   1.72   hannken 					bwrite(ibp);
   1653   1.72   hannken 				else
   1654   1.72   hannken 					bdwrite(ibp);
   1655    1.1   hannken 			}
   1656    1.1   hannken 			DIP_ADD(ip, blocks, btodb(size));
   1657    1.1   hannken 			ip->i_flag |= IN_CHANGE | IN_UPDATE;
   1658   1.92        ad 			if (ip->i_nlink > 0 && mp->mnt_wapbl)
   1659   1.75   hannken 				error = syncsnap(vp);
   1660   1.75   hannken 			else
   1661   1.75   hannken 				error = 0;
   1662  1.106   hannken 			mutex_enter(&si->si_lock);
   1663  1.106   hannken 			si->si_owner = NULL;
   1664  1.106   hannken 			mutex_exit(&si->si_lock);
   1665   1.74   hannken 			mutex_exit(&si->si_snaplock);
   1666   1.75   hannken 			return (error == 0);
   1667    1.1   hannken 		}
   1668  1.121  dholland 		if (lbn >= UFS_NDADDR)
   1669   1.52        ad 			brelse(ibp, 0);
   1670    1.1   hannken #ifdef DEBUG
   1671    1.1   hannken 		if (snapdebug)
   1672   1.19  christos 			printf("%s%llu lbn %" PRId64 " %s %llu size %ld\n",
   1673   1.19  christos 			    "Copyonremove: snapino ",
   1674   1.19  christos 			    (unsigned long long)ip->i_number,
   1675   1.19  christos 			    lbn, "for inum", (unsigned long long)inum, size);
   1676    1.1   hannken #endif
   1677    1.1   hannken 		/*
   1678    1.1   hannken 		 * If we have already read the old block contents, then
   1679    1.1   hannken 		 * simply copy them to the new block. Note that we need
   1680    1.1   hannken 		 * to synchronously write snapshots that have not been
   1681    1.1   hannken 		 * unlinked, and hence will be visible after a crash,
   1682    1.1   hannken 		 * to ensure their integrity.
   1683    1.1   hannken 		 */
   1684   1.49   hannken 		mutex_exit(&si->si_lock);
   1685   1.49   hannken 		if (saved_data == NULL) {
   1686   1.49   hannken 			saved_data = malloc(fs->fs_bsize, M_UFSMNT, M_WAITOK);
   1687   1.75   hannken 			error = rwfsblk(vp, B_READ, saved_data, lbn);
   1688   1.72   hannken 			if (error) {
   1689   1.49   hannken 				free(saved_data, M_UFSMNT);
   1690   1.49   hannken 				saved_data = NULL;
   1691   1.49   hannken 				mutex_enter(&si->si_lock);
   1692    1.1   hannken 				break;
   1693   1.49   hannken 			}
   1694    1.1   hannken 		}
   1695   1.72   hannken 		error = wrsnapblk(vp, saved_data, lbn);
   1696   1.92        ad 		if (error == 0 && ip->i_nlink > 0 && mp->mnt_wapbl)
   1697   1.75   hannken 			error = syncsnap(vp);
   1698   1.49   hannken 		mutex_enter(&si->si_lock);
   1699   1.49   hannken 		if (error)
   1700    1.1   hannken 			break;
   1701   1.49   hannken 		if (gen != si->si_gen)
   1702   1.49   hannken 			goto retry;
   1703    1.1   hannken 	}
   1704  1.106   hannken 	si->si_owner = NULL;
   1705   1.49   hannken 	mutex_exit(&si->si_lock);
   1706  1.106   hannken 	mutex_exit(&si->si_snaplock);
   1707   1.49   hannken 	if (saved_data)
   1708    1.1   hannken 		free(saved_data, M_UFSMNT);
   1709    1.1   hannken 	/*
   1710    1.1   hannken 	 * If we have been unable to allocate a block in which to do
   1711    1.1   hannken 	 * the copy, then return non-zero so that the fragment will
   1712    1.1   hannken 	 * not be freed. Although space will be lost, the snapshot
   1713    1.1   hannken 	 * will stay consistent.
   1714    1.1   hannken 	 */
   1715    1.1   hannken 	return (error);
   1716    1.1   hannken }
   1717    1.1   hannken 
   1718    1.1   hannken /*
   1719    1.1   hannken  * Associate snapshot files when mounting.
   1720    1.1   hannken  */
   1721    1.1   hannken void
   1722   1.18   thorpej ffs_snapshot_mount(struct mount *mp)
   1723    1.1   hannken {
   1724   1.47   hannken 	struct vnode *devvp = VFSTOUFS(mp)->um_devvp;
   1725   1.47   hannken 	struct fs *fs = VFSTOUFS(mp)->um_fs;
   1726   1.31        ad 	struct lwp *l = curlwp;
   1727    1.1   hannken 	struct vnode *vp;
   1728    1.1   hannken 	struct inode *ip, *xp;
   1729   1.47   hannken 	struct snap_info *si;
   1730   1.79   hannken 	daddr_t snaplistsize, *snapblklist;
   1731  1.130    martin 	int i, error, ns __unused, snaploc, loc;
   1732    1.1   hannken 
   1733   1.46   hannken 	/*
   1734   1.46   hannken 	 * No persistent snapshots on apple ufs file systems.
   1735   1.46   hannken 	 */
   1736   1.47   hannken 	if (UFS_MPISAPPLEUFS(VFSTOUFS(mp)))
   1737   1.46   hannken 		return;
   1738   1.46   hannken 
   1739   1.66   hannken 	si = VFSTOUFS(mp)->um_snapinfo;
   1740    1.1   hannken 	ns = UFS_FSNEEDSWAP(fs);
   1741    1.1   hannken 	/*
   1742   1.22      yamt 	 * XXX The following needs to be set before ffs_truncate or
   1743    1.1   hannken 	 * VOP_READ can be called.
   1744    1.1   hannken 	 */
   1745    1.1   hannken 	mp->mnt_stat.f_iosize = fs->fs_bsize;
   1746    1.1   hannken 	/*
   1747    1.1   hannken 	 * Process each snapshot listed in the superblock.
   1748    1.1   hannken 	 */
   1749    1.1   hannken 	vp = NULL;
   1750   1.49   hannken 	mutex_enter(&si->si_lock);
   1751    1.1   hannken 	for (snaploc = 0; snaploc < FSMAXSNAP; snaploc++) {
   1752    1.1   hannken 		if (fs->fs_snapinum[snaploc] == 0)
   1753    1.1   hannken 			break;
   1754    1.1   hannken 		if ((error = VFS_VGET(mp, fs->fs_snapinum[snaploc],
   1755    1.1   hannken 		    &vp)) != 0) {
   1756    1.1   hannken 			printf("ffs_snapshot_mount: vget failed %d\n", error);
   1757    1.1   hannken 			continue;
   1758    1.1   hannken 		}
   1759    1.1   hannken 		ip = VTOI(vp);
   1760  1.111    bouyer 		if ((ip->i_flags & (SF_SNAPSHOT | SF_SNAPINVAL)) !=
   1761  1.111    bouyer 		    SF_SNAPSHOT) {
   1762    1.1   hannken 			printf("ffs_snapshot_mount: non-snapshot inode %d\n",
   1763    1.1   hannken 			    fs->fs_snapinum[snaploc]);
   1764    1.1   hannken 			vput(vp);
   1765    1.1   hannken 			vp = NULL;
   1766    1.1   hannken 			for (loc = snaploc + 1; loc < FSMAXSNAP; loc++) {
   1767    1.1   hannken 				if (fs->fs_snapinum[loc] == 0)
   1768    1.1   hannken 					break;
   1769    1.1   hannken 				fs->fs_snapinum[loc - 1] = fs->fs_snapinum[loc];
   1770    1.1   hannken 			}
   1771    1.1   hannken 			fs->fs_snapinum[loc - 1] = 0;
   1772    1.1   hannken 			snaploc--;
   1773    1.1   hannken 			continue;
   1774    1.1   hannken 		}
   1775   1.15   hannken 
   1776   1.15   hannken 		/*
   1777   1.15   hannken 		 * Read the block hints list. Use an empty list on
   1778   1.15   hannken 		 * read errors.
   1779   1.15   hannken 		 */
   1780   1.15   hannken 		error = vn_rdwr(UIO_READ, vp,
   1781   1.43  christos 		    (void *)&snaplistsize, sizeof(snaplistsize),
   1782  1.126  dholland 		    ffs_lblktosize(fs, howmany(fs->fs_size, fs->fs_frag)),
   1783   1.87   hannken 		    UIO_SYSSPACE, IO_NODELOCKED|IO_UNIT|IO_ALTSEMANTICS,
   1784   1.31        ad 		    l->l_cred, NULL, NULL);
   1785   1.15   hannken 		if (error) {
   1786   1.15   hannken 			printf("ffs_snapshot_mount: read_1 failed %d\n", error);
   1787   1.15   hannken 			snaplistsize = 1;
   1788   1.15   hannken 		} else
   1789   1.15   hannken 			snaplistsize = ufs_rw64(snaplistsize, ns);
   1790   1.25  christos 		snapblklist = malloc(
   1791   1.79   hannken 		    snaplistsize * sizeof(daddr_t), M_UFSMNT, M_WAITOK);
   1792   1.15   hannken 		if (error)
   1793   1.15   hannken 			snapblklist[0] = 1;
   1794   1.15   hannken 		else {
   1795   1.43  christos 			error = vn_rdwr(UIO_READ, vp, (void *)snapblklist,
   1796   1.79   hannken 			    snaplistsize * sizeof(daddr_t),
   1797  1.126  dholland 			    ffs_lblktosize(fs, howmany(fs->fs_size, fs->fs_frag)),
   1798   1.87   hannken 			    UIO_SYSSPACE, IO_NODELOCKED|IO_UNIT|IO_ALTSEMANTICS,
   1799   1.31        ad 			    l->l_cred, NULL, NULL);
   1800   1.15   hannken 			for (i = 0; i < snaplistsize; i++)
   1801   1.15   hannken 				snapblklist[i] = ufs_rw64(snapblklist[i], ns);
   1802   1.15   hannken 			if (error) {
   1803   1.15   hannken 				printf("ffs_snapshot_mount: read_2 failed %d\n",
   1804   1.15   hannken 				    error);
   1805   1.15   hannken 				snapblklist[0] = 1;
   1806   1.15   hannken 			}
   1807   1.15   hannken 		}
   1808   1.15   hannken 		ip->i_snapblklist = &snapblklist[0];
   1809   1.15   hannken 
   1810    1.1   hannken 		/*
   1811    1.1   hannken 		 * Link it onto the active snapshot list.
   1812    1.1   hannken 		 */
   1813   1.97   hannken 		if (is_active_snapshot(si, ip))
   1814   1.97   hannken 			panic("ffs_snapshot_mount: %"PRIu64" already on list",
   1815   1.97   hannken 			    ip->i_number);
   1816    1.1   hannken 		else
   1817   1.47   hannken 			TAILQ_INSERT_TAIL(&si->si_snapshots, ip, i_nextsnap);
   1818   1.53        ad 		vp->v_vflag |= VV_SYSTEM;
   1819   1.99   hannken 		VOP_UNLOCK(vp);
   1820    1.1   hannken 	}
   1821    1.1   hannken 	/*
   1822    1.1   hannken 	 * No usable snapshots found.
   1823    1.1   hannken 	 */
   1824   1.49   hannken 	if (vp == NULL) {
   1825   1.49   hannken 		mutex_exit(&si->si_lock);
   1826    1.1   hannken 		return;
   1827   1.49   hannken 	}
   1828    1.1   hannken 	/*
   1829   1.15   hannken 	 * Attach the block hints list. We always want to
   1830    1.1   hannken 	 * use the list from the newest snapshot.
   1831   1.15   hannken 	*/
   1832   1.47   hannken 	xp = TAILQ_LAST(&si->si_snapshots, inodelst);
   1833   1.47   hannken 	si->si_snapblklist = xp->i_snapblklist;
   1834   1.51   hannken 	fscow_establish(mp, ffs_copyonwrite, devvp);
   1835   1.49   hannken 	si->si_gen++;
   1836   1.49   hannken 	mutex_exit(&si->si_lock);
   1837    1.1   hannken }
   1838    1.1   hannken 
   1839    1.1   hannken /*
   1840    1.1   hannken  * Disassociate snapshot files when unmounting.
   1841    1.1   hannken  */
   1842    1.1   hannken void
   1843   1.18   thorpej ffs_snapshot_unmount(struct mount *mp)
   1844    1.1   hannken {
   1845   1.47   hannken 	struct vnode *devvp = VFSTOUFS(mp)->um_devvp;
   1846  1.104   hannken 	struct inode *xp;
   1847  1.104   hannken 	struct vnode *vp = NULL;
   1848   1.47   hannken 	struct snap_info *si;
   1849    1.1   hannken 
   1850   1.66   hannken 	si = VFSTOUFS(mp)->um_snapinfo;
   1851   1.49   hannken 	mutex_enter(&si->si_lock);
   1852  1.104   hannken 	while ((xp = TAILQ_FIRST(&si->si_snapshots)) != 0) {
   1853  1.104   hannken 		vp = ITOV(xp);
   1854  1.104   hannken 		TAILQ_REMOVE(&si->si_snapshots, xp, i_nextsnap);
   1855  1.104   hannken 		if (xp->i_snapblklist == si->si_snapblklist)
   1856   1.47   hannken 			si->si_snapblklist = NULL;
   1857  1.104   hannken 		free(xp->i_snapblklist, M_UFSMNT);
   1858  1.104   hannken 		if (xp->i_nlink > 0) {
   1859  1.104   hannken 			si->si_gen++;
   1860  1.104   hannken 			mutex_exit(&si->si_lock);
   1861  1.104   hannken 			vrele(vp);
   1862  1.104   hannken 			mutex_enter(&si->si_lock);
   1863  1.104   hannken 		}
   1864    1.1   hannken 	}
   1865  1.104   hannken 	si->si_gen++;
   1866   1.96   hannken 	mutex_exit(&si->si_lock);
   1867  1.104   hannken 	if (vp)
   1868   1.51   hannken 		fscow_disestablish(mp, ffs_copyonwrite, devvp);
   1869    1.1   hannken }
   1870    1.1   hannken 
   1871    1.1   hannken /*
   1872    1.1   hannken  * Check for need to copy block that is about to be written,
   1873    1.1   hannken  * copying the block if necessary.
   1874    1.1   hannken  */
   1875    1.1   hannken static int
   1876   1.55   hannken ffs_copyonwrite(void *v, struct buf *bp, bool data_valid)
   1877    1.1   hannken {
   1878    1.1   hannken 	struct fs *fs;
   1879    1.1   hannken 	struct inode *ip;
   1880   1.49   hannken 	struct vnode *devvp = v, *vp = NULL;
   1881  1.129   hannken 	struct mount *mp = spec_node_getmountedfs(devvp);
   1882   1.47   hannken 	struct snap_info *si;
   1883   1.43  christos 	void *saved_data = NULL;
   1884   1.79   hannken 	daddr_t lbn, blkno, *snapblklist;
   1885   1.49   hannken 	uint32_t gen;
   1886   1.76   hannken 	int lower, upper, mid, snapshot_locked = 0, error = 0;
   1887    1.1   hannken 
   1888    1.1   hannken 	/*
   1889    1.1   hannken 	 * Check for valid snapshots.
   1890    1.1   hannken 	 */
   1891   1.66   hannken 	si = VFSTOUFS(mp)->um_snapinfo;
   1892   1.49   hannken 	mutex_enter(&si->si_lock);
   1893   1.47   hannken 	ip = TAILQ_FIRST(&si->si_snapshots);
   1894    1.1   hannken 	if (ip == NULL) {
   1895   1.49   hannken 		mutex_exit(&si->si_lock);
   1896   1.11     perry 		return 0;
   1897    1.1   hannken 	}
   1898    1.1   hannken 	/*
   1899  1.117   hannken 	 * First check to see if it is after the file system,
   1900  1.117   hannken 	 * in the journal or in the preallocated list.
   1901  1.117   hannken 	 * By doing these checks we avoid several potential deadlocks.
   1902    1.1   hannken 	 */
   1903    1.1   hannken 	fs = ip->i_fs;
   1904  1.127  dholland 	lbn = ffs_fragstoblks(fs, FFS_DBTOFSB(fs, bp->b_blkno));
   1905  1.125  dholland 	if (bp->b_blkno >= FFS_FSBTODB(fs, fs->fs_size)) {
   1906   1.75   hannken 		mutex_exit(&si->si_lock);
   1907   1.75   hannken 		return 0;
   1908   1.75   hannken 	}
   1909  1.117   hannken 	if ((fs->fs_flags & FS_DOWAPBL) &&
   1910  1.117   hannken 	    fs->fs_journal_location == UFS_WAPBL_JOURNALLOC_IN_FILESYSTEM) {
   1911  1.117   hannken 		off_t blk_off, log_start, log_end;
   1912  1.117   hannken 
   1913  1.117   hannken 		log_start = (off_t)fs->fs_journallocs[UFS_WAPBL_INFS_ADDR] *
   1914  1.117   hannken 		    fs->fs_journallocs[UFS_WAPBL_INFS_BLKSZ];
   1915  1.117   hannken 		log_end = log_start + fs->fs_journallocs[UFS_WAPBL_INFS_COUNT] *
   1916  1.117   hannken 		    fs->fs_journallocs[UFS_WAPBL_INFS_BLKSZ];
   1917  1.117   hannken 		blk_off = dbtob(bp->b_blkno);
   1918  1.117   hannken 		if (blk_off >= log_start && blk_off < log_end) {
   1919  1.117   hannken 			mutex_exit(&si->si_lock);
   1920  1.117   hannken 			return 0;
   1921  1.117   hannken 		}
   1922  1.117   hannken 	}
   1923   1.47   hannken 	snapblklist = si->si_snapblklist;
   1924   1.87   hannken 	upper = (snapblklist != NULL ? snapblklist[0] - 1 : 0);
   1925    1.1   hannken 	lower = 1;
   1926    1.1   hannken 	while (lower <= upper) {
   1927    1.1   hannken 		mid = (lower + upper) / 2;
   1928   1.15   hannken 		if (snapblklist[mid] == lbn)
   1929    1.1   hannken 			break;
   1930   1.15   hannken 		if (snapblklist[mid] < lbn)
   1931    1.1   hannken 			lower = mid + 1;
   1932    1.1   hannken 		else
   1933    1.1   hannken 			upper = mid - 1;
   1934    1.1   hannken 	}
   1935    1.1   hannken 	if (lower <= upper) {
   1936   1.49   hannken 		mutex_exit(&si->si_lock);
   1937    1.1   hannken 		return 0;
   1938    1.1   hannken 	}
   1939    1.1   hannken 	/*
   1940    1.1   hannken 	 * Not in the precomputed list, so check the snapshots.
   1941    1.1   hannken 	 */
   1942  1.106   hannken 	 if (si->si_owner != curlwp) {
   1943  1.106   hannken 		if (!mutex_tryenter(&si->si_snaplock)) {
   1944  1.106   hannken 			mutex_exit(&si->si_lock);
   1945  1.106   hannken 			mutex_enter(&si->si_snaplock);
   1946  1.106   hannken 			mutex_enter(&si->si_lock);
   1947  1.106   hannken 		}
   1948  1.106   hannken 		si->si_owner = curlwp;
   1949  1.106   hannken 		snapshot_locked = 1;
   1950  1.106   hannken 	 }
   1951   1.55   hannken 	 if (data_valid && bp->b_bcount == fs->fs_bsize)
   1952   1.55   hannken 		saved_data = bp->b_data;
   1953    1.1   hannken retry:
   1954   1.49   hannken 	gen = si->si_gen;
   1955   1.47   hannken 	TAILQ_FOREACH(ip, &si->si_snapshots, i_nextsnap) {
   1956    1.1   hannken 		vp = ITOV(ip);
   1957    1.1   hannken 		/*
   1958    1.1   hannken 		 * We ensure that everything of our own that needs to be
   1959    1.1   hannken 		 * copied will be done at the time that ffs_snapshot is
   1960    1.1   hannken 		 * called. Thus we can skip the check here which can
   1961   1.22      yamt 		 * deadlock in doing the lookup in ffs_balloc.
   1962    1.1   hannken 		 */
   1963    1.1   hannken 		if (bp->b_vp == vp)
   1964    1.1   hannken 			continue;
   1965    1.1   hannken 		/*
   1966   1.68   hannken 		 * Check to see if block needs to be copied.
   1967    1.1   hannken 		 */
   1968  1.121  dholland 		if (lbn < UFS_NDADDR) {
   1969    1.1   hannken 			blkno = db_get(ip, lbn);
   1970    1.1   hannken 		} else {
   1971   1.49   hannken 			mutex_exit(&si->si_lock);
   1972  1.109    dyoung 			blkno = 0; /* XXX: GCC */
   1973   1.72   hannken 			if ((error = snapblkaddr(vp, lbn, &blkno)) != 0) {
   1974   1.49   hannken 				mutex_enter(&si->si_lock);
   1975    1.1   hannken 				break;
   1976   1.49   hannken 			}
   1977   1.49   hannken 			mutex_enter(&si->si_lock);
   1978   1.49   hannken 			if (gen != si->si_gen)
   1979   1.49   hannken 				goto retry;
   1980    1.1   hannken 		}
   1981    1.1   hannken #ifdef DIAGNOSTIC
   1982    1.1   hannken 		if (blkno == BLK_SNAP && bp->b_lblkno >= 0)
   1983    1.1   hannken 			panic("ffs_copyonwrite: bad copy block");
   1984    1.1   hannken #endif
   1985    1.1   hannken 		if (blkno != 0)
   1986    1.1   hannken 			continue;
   1987   1.72   hannken 
   1988   1.74   hannken 		if (curlwp == uvm.pagedaemon_lwp) {
   1989   1.74   hannken 			error = ENOMEM;
   1990   1.74   hannken 			break;
   1991   1.74   hannken 		}
   1992  1.106   hannken 		/* Only one level of recursion allowed. */
   1993  1.106   hannken 		KASSERT(snapshot_locked);
   1994    1.1   hannken 		/*
   1995    1.1   hannken 		 * Allocate the block into which to do the copy. Since
   1996    1.1   hannken 		 * multiple processes may all try to copy the same block,
   1997    1.1   hannken 		 * we have to recheck our need to do a copy if we sleep
   1998    1.1   hannken 		 * waiting for the lock.
   1999    1.1   hannken 		 *
   2000    1.1   hannken 		 * Because all snapshots on a filesystem share a single
   2001    1.1   hannken 		 * lock, we ensure that we will never be in competition
   2002    1.1   hannken 		 * with another process to allocate a block.
   2003    1.1   hannken 		 */
   2004    1.1   hannken #ifdef DEBUG
   2005    1.1   hannken 		if (snapdebug) {
   2006   1.19  christos 			printf("Copyonwrite: snapino %llu lbn %" PRId64 " for ",
   2007   1.19  christos 			    (unsigned long long)ip->i_number, lbn);
   2008    1.1   hannken 			if (bp->b_vp == devvp)
   2009    1.1   hannken 				printf("fs metadata");
   2010    1.1   hannken 			else
   2011   1.19  christos 				printf("inum %llu", (unsigned long long)
   2012   1.19  christos 				    VTOI(bp->b_vp)->i_number);
   2013    1.4   hannken 			printf(" lblkno %" PRId64 "\n", bp->b_lblkno);
   2014    1.1   hannken 		}
   2015    1.1   hannken #endif
   2016    1.1   hannken 		/*
   2017    1.1   hannken 		 * If we have already read the old block contents, then
   2018    1.1   hannken 		 * simply copy them to the new block. Note that we need
   2019    1.1   hannken 		 * to synchronously write snapshots that have not been
   2020    1.1   hannken 		 * unlinked, and hence will be visible after a crash,
   2021    1.1   hannken 		 * to ensure their integrity.
   2022    1.1   hannken 		 */
   2023   1.49   hannken 		mutex_exit(&si->si_lock);
   2024   1.49   hannken 		if (saved_data == NULL) {
   2025   1.49   hannken 			saved_data = malloc(fs->fs_bsize, M_UFSMNT, M_WAITOK);
   2026   1.75   hannken 			error = rwfsblk(vp, B_READ, saved_data, lbn);
   2027   1.72   hannken 			if (error) {
   2028   1.49   hannken 				free(saved_data, M_UFSMNT);
   2029   1.49   hannken 				saved_data = NULL;
   2030   1.49   hannken 				mutex_enter(&si->si_lock);
   2031    1.1   hannken 				break;
   2032   1.49   hannken 			}
   2033    1.1   hannken 		}
   2034   1.72   hannken 		error = wrsnapblk(vp, saved_data, lbn);
   2035   1.92        ad 		if (error == 0 && ip->i_nlink > 0 && mp->mnt_wapbl)
   2036   1.75   hannken 			error = syncsnap(vp);
   2037   1.49   hannken 		mutex_enter(&si->si_lock);
   2038   1.49   hannken 		if (error)
   2039    1.1   hannken 			break;
   2040   1.49   hannken 		if (gen != si->si_gen)
   2041   1.49   hannken 			goto retry;
   2042    1.1   hannken 	}
   2043    1.1   hannken 	/*
   2044    1.1   hannken 	 * Note that we need to synchronously write snapshots that
   2045    1.1   hannken 	 * have not been unlinked, and hence will be visible after
   2046    1.1   hannken 	 * a crash, to ensure their integrity.
   2047    1.1   hannken 	 */
   2048  1.106   hannken 	if (snapshot_locked) {
   2049  1.106   hannken 		si->si_owner = NULL;
   2050  1.106   hannken 		mutex_exit(&si->si_lock);
   2051  1.106   hannken 		mutex_exit(&si->si_snaplock);
   2052  1.106   hannken 	} else
   2053  1.106   hannken 		mutex_exit(&si->si_lock);
   2054   1.55   hannken 	if (saved_data && saved_data != bp->b_data)
   2055    1.1   hannken 		free(saved_data, M_UFSMNT);
   2056   1.74   hannken 	return error;
   2057   1.74   hannken }
   2058   1.74   hannken 
   2059   1.74   hannken /*
   2060   1.74   hannken  * Read from a snapshot.
   2061   1.74   hannken  */
   2062   1.74   hannken int
   2063   1.74   hannken ffs_snapshot_read(struct vnode *vp, struct uio *uio, int ioflag)
   2064   1.74   hannken {
   2065   1.74   hannken 	struct inode *ip = VTOI(vp);
   2066   1.74   hannken 	struct fs *fs = ip->i_fs;
   2067   1.74   hannken 	struct snap_info *si = VFSTOUFS(vp->v_mount)->um_snapinfo;
   2068   1.74   hannken 	struct buf *bp;
   2069   1.74   hannken 	daddr_t lbn, nextlbn;
   2070   1.81   hannken 	off_t fsbytes, bytesinfile;
   2071   1.74   hannken 	long size, xfersize, blkoffset;
   2072   1.74   hannken 	int error;
   2073   1.74   hannken 
   2074   1.74   hannken 	fstrans_start(vp->v_mount, FSTRANS_SHARED);
   2075   1.74   hannken 	mutex_enter(&si->si_snaplock);
   2076   1.74   hannken 
   2077   1.87   hannken 	if (ioflag & IO_ALTSEMANTICS)
   2078   1.87   hannken 		fsbytes = ip->i_size;
   2079   1.87   hannken 	else
   2080  1.126  dholland 		fsbytes = ffs_lfragtosize(fs, fs->fs_size);
   2081   1.74   hannken 	for (error = 0, bp = NULL; uio->uio_resid > 0; bp = NULL) {
   2082   1.81   hannken 		bytesinfile = fsbytes - uio->uio_offset;
   2083   1.74   hannken 		if (bytesinfile <= 0)
   2084   1.74   hannken 			break;
   2085  1.126  dholland 		lbn = ffs_lblkno(fs, uio->uio_offset);
   2086   1.74   hannken 		nextlbn = lbn + 1;
   2087   1.82   hannken 		size = fs->fs_bsize;
   2088  1.124  dholland 		blkoffset = ffs_blkoff(fs, uio->uio_offset);
   2089   1.74   hannken 		xfersize = MIN(MIN(fs->fs_bsize - blkoffset, uio->uio_resid),
   2090   1.74   hannken 		    bytesinfile);
   2091   1.74   hannken 
   2092  1.126  dholland 		if (ffs_lblktosize(fs, nextlbn + 1) >= fsbytes) {
   2093  1.126  dholland 			if (ffs_lblktosize(fs, lbn) + size > fsbytes)
   2094  1.126  dholland 				size = ffs_fragroundup(fs,
   2095  1.126  dholland 				    fsbytes - ffs_lblktosize(fs, lbn));
   2096  1.139      maxv 			error = bread(vp, lbn, size, 0, &bp);
   2097   1.82   hannken 		} else {
   2098   1.82   hannken 			int nextsize = fs->fs_bsize;
   2099   1.74   hannken 			error = breadn(vp, lbn,
   2100  1.138      maxv 			    size, &nextlbn, &nextsize, 1, 0, &bp);
   2101   1.74   hannken 		}
   2102   1.74   hannken 		if (error)
   2103   1.74   hannken 			break;
   2104   1.74   hannken 
   2105   1.74   hannken 		/*
   2106   1.74   hannken 		 * We should only get non-zero b_resid when an I/O error
   2107   1.74   hannken 		 * has occurred, which should cause us to break above.
   2108   1.74   hannken 		 * However, if the short read did not cause an error,
   2109   1.74   hannken 		 * then we want to ensure that we do not uiomove bad
   2110   1.74   hannken 		 * or uninitialized data.
   2111   1.74   hannken 		 */
   2112   1.74   hannken 		size -= bp->b_resid;
   2113   1.82   hannken 		if (size < blkoffset + xfersize) {
   2114   1.82   hannken 			xfersize = size - blkoffset;
   2115   1.82   hannken 			if (xfersize <= 0)
   2116   1.74   hannken 				break;
   2117   1.74   hannken 		}
   2118   1.74   hannken 		error = uiomove((char *)bp->b_data + blkoffset, xfersize, uio);
   2119   1.74   hannken 		if (error)
   2120   1.74   hannken 			break;
   2121   1.75   hannken 		brelse(bp, BC_AGE);
   2122   1.74   hannken 	}
   2123   1.74   hannken 	if (bp != NULL)
   2124   1.75   hannken 		brelse(bp, BC_AGE);
   2125   1.74   hannken 
   2126   1.74   hannken 	mutex_exit(&si->si_snaplock);
   2127   1.74   hannken 	fstrans_done(vp->v_mount);
   2128    1.1   hannken 	return error;
   2129    1.1   hannken }
   2130    1.1   hannken 
   2131    1.1   hannken /*
   2132   1.79   hannken  * Lookup a snapshots data block address.
   2133   1.79   hannken  * Simpler than UFS_BALLOC() as we know all metadata is already allocated
   2134   1.79   hannken  * and safe even for the pagedaemon where we cannot bread().
   2135   1.79   hannken  */
   2136   1.79   hannken static int
   2137   1.79   hannken snapblkaddr(struct vnode *vp, daddr_t lbn, daddr_t *res)
   2138   1.79   hannken {
   2139  1.121  dholland 	struct indir indirs[UFS_NIADDR + 2];
   2140   1.79   hannken 	struct inode *ip = VTOI(vp);
   2141   1.79   hannken 	struct fs *fs = ip->i_fs;
   2142   1.79   hannken 	struct buf *bp;
   2143   1.79   hannken 	int error, num;
   2144   1.79   hannken 
   2145   1.79   hannken 	KASSERT(lbn >= 0);
   2146   1.79   hannken 
   2147  1.121  dholland 	if (lbn < UFS_NDADDR) {
   2148   1.79   hannken 		*res = db_get(ip, lbn);
   2149   1.79   hannken 		return 0;
   2150   1.79   hannken 	}
   2151   1.79   hannken 	if ((error = ufs_getlbns(vp, lbn, indirs, &num)) != 0)
   2152   1.79   hannken 		return error;
   2153   1.79   hannken 	if (curlwp == uvm.pagedaemon_lwp) {
   2154   1.79   hannken 		mutex_enter(&bufcache_lock);
   2155   1.79   hannken 		bp = incore(vp, indirs[num-1].in_lbn);
   2156   1.79   hannken 		if (bp && (bp->b_oflags & (BO_DONE | BO_DELWRI))) {
   2157   1.79   hannken 			*res = idb_get(ip, bp->b_data, indirs[num-1].in_off);
   2158   1.79   hannken 			error = 0;
   2159   1.79   hannken 		} else
   2160   1.79   hannken 			error = ENOMEM;
   2161   1.79   hannken 		mutex_exit(&bufcache_lock);
   2162   1.79   hannken 		return error;
   2163   1.79   hannken 	}
   2164  1.139      maxv 	error = bread(vp, indirs[num-1].in_lbn, fs->fs_bsize, 0, &bp);
   2165  1.120   hannken 	if (error == 0) {
   2166   1.79   hannken 		*res = idb_get(ip, bp->b_data, indirs[num-1].in_off);
   2167  1.120   hannken 		brelse(bp, 0);
   2168  1.120   hannken 	}
   2169   1.79   hannken 
   2170   1.79   hannken 	return error;
   2171   1.79   hannken }
   2172   1.79   hannken 
   2173   1.79   hannken /*
   2174   1.75   hannken  * Read or write the specified block of the filesystem vp resides on
   2175   1.75   hannken  * from or to the disk bypassing the buffer cache.
   2176    1.1   hannken  */
   2177    1.1   hannken static int
   2178   1.79   hannken rwfsblk(struct vnode *vp, int flags, void *data, daddr_t lbn)
   2179    1.1   hannken {
   2180   1.24      yamt 	int error;
   2181    1.1   hannken 	struct inode *ip = VTOI(vp);
   2182    1.1   hannken 	struct fs *fs = ip->i_fs;
   2183    1.1   hannken 	struct buf *nbp;
   2184    1.1   hannken 
   2185   1.57        ad 	nbp = getiobuf(NULL, true);
   2186   1.75   hannken 	nbp->b_flags = flags;
   2187    1.1   hannken 	nbp->b_bcount = nbp->b_bufsize = fs->fs_bsize;
   2188    1.1   hannken 	nbp->b_error = 0;
   2189    1.1   hannken 	nbp->b_data = data;
   2190  1.127  dholland 	nbp->b_blkno = nbp->b_rawblkno = FFS_FSBTODB(fs, ffs_blkstofrags(fs, lbn));
   2191    1.1   hannken 	nbp->b_proc = NULL;
   2192    1.1   hannken 	nbp->b_dev = ip->i_devvp->v_rdev;
   2193   1.70   reinoud 	SET(nbp->b_cflags, BC_BUSY);	/* mark buffer busy */
   2194    1.1   hannken 
   2195   1.44        ad 	bdev_strategy(nbp);
   2196    1.1   hannken 
   2197    1.1   hannken 	error = biowait(nbp);
   2198    1.1   hannken 
   2199   1.24      yamt 	putiobuf(nbp);
   2200    1.1   hannken 
   2201    1.1   hannken 	return error;
   2202    1.1   hannken }
   2203    1.1   hannken 
   2204    1.1   hannken /*
   2205   1.75   hannken  * Write all dirty buffers to disk and invalidate them.
   2206   1.75   hannken  */
   2207   1.75   hannken static int
   2208   1.75   hannken syncsnap(struct vnode *vp)
   2209   1.75   hannken {
   2210   1.75   hannken 	int error;
   2211   1.75   hannken 	buf_t *bp;
   2212   1.75   hannken 	struct fs *fs = VTOI(vp)->i_fs;
   2213   1.75   hannken 
   2214   1.75   hannken 	mutex_enter(&bufcache_lock);
   2215   1.75   hannken 	while ((bp = LIST_FIRST(&vp->v_dirtyblkhd))) {
   2216  1.100   hannken 		error = bbusy(bp, false, 0, NULL);
   2217  1.100   hannken 		if (error == EPASSTHROUGH)
   2218  1.100   hannken 			continue;
   2219  1.100   hannken 		else if (error != 0) {
   2220  1.100   hannken 			mutex_exit(&bufcache_lock);
   2221  1.100   hannken 			return error;
   2222  1.100   hannken 		}
   2223   1.75   hannken 		KASSERT(bp->b_bcount == fs->fs_bsize);
   2224   1.75   hannken 		mutex_exit(&bufcache_lock);
   2225   1.75   hannken 		error = rwfsblk(vp, B_WRITE, bp->b_data,
   2226  1.127  dholland 		    ffs_fragstoblks(fs, FFS_DBTOFSB(fs, bp->b_blkno)));
   2227   1.75   hannken 		brelse(bp, BC_INVAL | BC_VFLUSH);
   2228   1.75   hannken 		if (error)
   2229   1.75   hannken 			return error;
   2230   1.75   hannken 		mutex_enter(&bufcache_lock);
   2231   1.75   hannken 	}
   2232   1.75   hannken 	mutex_exit(&bufcache_lock);
   2233   1.75   hannken 
   2234   1.75   hannken 	return 0;
   2235   1.75   hannken }
   2236   1.75   hannken 
   2237   1.75   hannken /*
   2238   1.72   hannken  * Write the specified block to a snapshot.
   2239    1.1   hannken  */
   2240    1.1   hannken static int
   2241   1.79   hannken wrsnapblk(struct vnode *vp, void *data, daddr_t lbn)
   2242    1.1   hannken {
   2243    1.1   hannken 	struct inode *ip = VTOI(vp);
   2244    1.1   hannken 	struct fs *fs = ip->i_fs;
   2245   1.74   hannken 	struct buf *bp;
   2246   1.74   hannken 	int error;
   2247    1.1   hannken 
   2248  1.126  dholland 	error = ffs_balloc(vp, ffs_lblktosize(fs, (off_t)lbn), fs->fs_bsize,
   2249   1.92        ad 	    FSCRED, (ip->i_nlink > 0 ? B_SYNC : 0), &bp);
   2250   1.74   hannken 	if (error)
   2251   1.74   hannken 		return error;
   2252   1.95   tsutsui 	memcpy(bp->b_data, data, fs->fs_bsize);
   2253   1.92        ad 	if (ip->i_nlink > 0)
   2254   1.74   hannken 		error = bwrite(bp);
   2255   1.74   hannken 	else
   2256   1.74   hannken 		bawrite(bp);
   2257    1.4   hannken 
   2258   1.72   hannken 	return error;
   2259    1.4   hannken }
   2260    1.4   hannken 
   2261    1.4   hannken /*
   2262   1.97   hannken  * Check if this inode is present on the active snapshot list.
   2263   1.97   hannken  * Must be called with snapinfo locked.
   2264   1.97   hannken  */
   2265   1.97   hannken static inline bool
   2266   1.97   hannken is_active_snapshot(struct snap_info *si, struct inode *ip)
   2267   1.97   hannken {
   2268   1.97   hannken 	struct inode *xp;
   2269   1.97   hannken 
   2270   1.97   hannken 	KASSERT(mutex_owned(&si->si_lock));
   2271   1.97   hannken 
   2272   1.97   hannken 	TAILQ_FOREACH(xp, &si->si_snapshots, i_nextsnap)
   2273   1.97   hannken 		if (xp == ip)
   2274   1.97   hannken 			return true;
   2275   1.97   hannken 	return false;
   2276   1.97   hannken }
   2277   1.97   hannken 
   2278   1.97   hannken /*
   2279    1.1   hannken  * Get/Put direct block from inode or buffer containing disk addresses. Take
   2280    1.1   hannken  * care for fs type (UFS1/UFS2) and byte swapping. These functions should go
   2281    1.1   hannken  * into a global include.
   2282    1.1   hannken  */
   2283   1.79   hannken static inline daddr_t
   2284    1.1   hannken db_get(struct inode *ip, int loc)
   2285    1.1   hannken {
   2286    1.1   hannken 	if (ip->i_ump->um_fstype == UFS1)
   2287    1.2   hannken 		return ufs_rw32(ip->i_ffs1_db[loc], UFS_IPNEEDSWAP(ip));
   2288    1.1   hannken 	else
   2289    1.2   hannken 		return ufs_rw64(ip->i_ffs2_db[loc], UFS_IPNEEDSWAP(ip));
   2290    1.1   hannken }
   2291    1.1   hannken 
   2292    1.1   hannken static inline void
   2293   1.79   hannken db_assign(struct inode *ip, int loc, daddr_t val)
   2294    1.1   hannken {
   2295    1.1   hannken 	if (ip->i_ump->um_fstype == UFS1)
   2296    1.2   hannken 		ip->i_ffs1_db[loc] = ufs_rw32(val, UFS_IPNEEDSWAP(ip));
   2297    1.1   hannken 	else
   2298    1.2   hannken 		ip->i_ffs2_db[loc] = ufs_rw64(val, UFS_IPNEEDSWAP(ip));
   2299    1.1   hannken }
   2300    1.1   hannken 
   2301  1.132     joerg __unused static inline daddr_t
   2302   1.76   hannken ib_get(struct inode *ip, int loc)
   2303   1.76   hannken {
   2304   1.76   hannken 	if (ip->i_ump->um_fstype == UFS1)
   2305   1.76   hannken 		return ufs_rw32(ip->i_ffs1_ib[loc], UFS_IPNEEDSWAP(ip));
   2306   1.76   hannken 	else
   2307   1.76   hannken 		return ufs_rw64(ip->i_ffs2_ib[loc], UFS_IPNEEDSWAP(ip));
   2308   1.76   hannken }
   2309   1.76   hannken 
   2310   1.79   hannken static inline daddr_t
   2311   1.43  christos idb_get(struct inode *ip, void *bf, int loc)
   2312    1.1   hannken {
   2313    1.1   hannken 	if (ip->i_ump->um_fstype == UFS1)
   2314   1.79   hannken 		return ufs_rw32(((int32_t *)(bf))[loc], UFS_IPNEEDSWAP(ip));
   2315    1.1   hannken 	else
   2316   1.79   hannken 		return ufs_rw64(((int64_t *)(bf))[loc], UFS_IPNEEDSWAP(ip));
   2317    1.1   hannken }
   2318    1.1   hannken 
   2319    1.1   hannken static inline void
   2320   1.79   hannken idb_assign(struct inode *ip, void *bf, int loc, daddr_t val)
   2321    1.1   hannken {
   2322    1.1   hannken 	if (ip->i_ump->um_fstype == UFS1)
   2323   1.79   hannken 		((int32_t *)(bf))[loc] = ufs_rw32(val, UFS_IPNEEDSWAP(ip));
   2324    1.1   hannken 	else
   2325   1.79   hannken 		((int64_t *)(bf))[loc] = ufs_rw64(val, UFS_IPNEEDSWAP(ip));
   2326    1.1   hannken }
   2327