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