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