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