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