growfs.c revision 114067
169800Stomsoft/*
269800Stomsoft * Copyright (c) 2000 Christoph Herrmann, Thomas-Henning von Kamptz
369800Stomsoft * Copyright (c) 1980, 1989, 1993 The Regents of the University of California.
469800Stomsoft * All rights reserved.
5114067Sschweikh *
669800Stomsoft * This code is derived from software contributed to Berkeley by
769800Stomsoft * Christoph Herrmann and Thomas-Henning von Kamptz, Munich and Frankfurt.
8114067Sschweikh *
969800Stomsoft * Redistribution and use in source and binary forms, with or without
1069800Stomsoft * modification, are permitted provided that the following conditions
1169800Stomsoft * are met:
1269800Stomsoft * 1. Redistributions of source code must retain the above copyright
1369800Stomsoft *    notice, this list of conditions and the following disclaimer.
1469800Stomsoft * 2. Redistributions in binary form must reproduce the above copyright
1569800Stomsoft *    notice, this list of conditions and the following disclaimer in the
1669800Stomsoft *    documentation and/or other materials provided with the distribution.
1769800Stomsoft * 3. All advertising materials mentioning features or use of this software
1869800Stomsoft *    must display the following acknowledgment:
1969800Stomsoft *      This product includes software developed by the University of
2069800Stomsoft *      California, Berkeley and its contributors, as well as Christoph
2169800Stomsoft *      Herrmann and Thomas-Henning von Kamptz.
2269800Stomsoft * 4. Neither the name of the University nor the names of its contributors
2369800Stomsoft *    may be used to endorse or promote products derived from this software
2469800Stomsoft *    without specific prior written permission.
25114067Sschweikh *
2669800Stomsoft * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
2769800Stomsoft * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
2869800Stomsoft * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
2969800Stomsoft * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
3069800Stomsoft * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
3169800Stomsoft * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
3269800Stomsoft * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
3369800Stomsoft * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
3469800Stomsoft * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
3569800Stomsoft * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
3669800Stomsoft * SUCH DAMAGE.
3769800Stomsoft *
3869926Stomsoft * $TSHeader: src/sbin/growfs/growfs.c,v 1.5 2000/12/12 19:31:00 tomsoft Exp $
3969800Stomsoft *
4069800Stomsoft */
4169800Stomsoft
4269800Stomsoft#ifndef lint
4369800Stomsoftstatic const char copyright[] =
4469800Stomsoft"@(#) Copyright (c) 2000 Christoph Herrmann, Thomas-Henning von Kamptz\n\
4569800StomsoftCopyright (c) 1980, 1989, 1993 The Regents of the University of California.\n\
4669800StomsoftAll rights reserved.\n";
4769800Stomsoft#endif /* not lint */
4869800Stomsoft
4969800Stomsoft#ifndef lint
5069800Stomsoftstatic const char rcsid[] =
5169926Stomsoft  "$FreeBSD: head/sbin/growfs/growfs.c 114067 2003-04-26 15:22:29Z schweikh $";
5269800Stomsoft#endif /* not lint */
5369800Stomsoft
5469800Stomsoft/* ********************************************************** INCLUDES ***** */
5569800Stomsoft#include <sys/param.h>
5669800Stomsoft#include <sys/disklabel.h>
5769800Stomsoft#include <sys/ioctl.h>
5869800Stomsoft#include <sys/stat.h>
5969800Stomsoft
6069800Stomsoft#include <stdio.h>
6169800Stomsoft#include <paths.h>
6269800Stomsoft#include <ctype.h>
6369800Stomsoft#include <err.h>
6469800Stomsoft#include <fcntl.h>
65103949Smike#include <limits.h>
6669800Stomsoft#include <stdlib.h>
6769800Stomsoft#include <string.h>
6869800Stomsoft#include <unistd.h>
6969800Stomsoft#include <ufs/ufs/dinode.h>
7069800Stomsoft#include <ufs/ffs/fs.h>
7169800Stomsoft
7269800Stomsoft#include "debug.h"
7369800Stomsoft
7469800Stomsoft/* *************************************************** GLOBALS & TYPES ***** */
7569800Stomsoft#ifdef FS_DEBUG
7669800Stomsoftint	_dbg_lvl_ = (DL_INFO);	/* DL_TRC */
7769800Stomsoft#endif /* FS_DEBUG */
7869800Stomsoft
7969800Stomsoftstatic union {
8069800Stomsoft	struct fs	fs;
8198542Smckusick	char	pad[SBLOCKSIZE];
8269800Stomsoft} fsun1, fsun2;
8369800Stomsoft#define	sblock	fsun1.fs	/* the new superblock */
8469800Stomsoft#define	osblock	fsun2.fs	/* the old superblock */
8569800Stomsoft
8698542Smckusick/*
8798542Smckusick * Possible superblock locations ordered from most to least likely.
8898542Smckusick */
8998542Smckusickstatic int sblock_try[] = SBLOCKSEARCH;
9098542Smckusickstatic ufs2_daddr_t sblockloc;
9198542Smckusick
9269800Stomsoftstatic union {
9369800Stomsoft	struct cg	cg;
9469800Stomsoft	char	pad[MAXBSIZE];
9569800Stomsoft} cgun1, cgun2;
9669800Stomsoft#define	acg	cgun1.cg	/* a cylinder cgroup (new) */
9769800Stomsoft#define	aocg	cgun2.cg	/* an old cylinder group */
9869800Stomsoft
9998542Smckusickstatic char	ablk[MAXBSIZE];	/* a block */
10069800Stomsoft
10198542Smckusickstatic struct csum	*fscs;	/* cylinder summary */
10269800Stomsoft
10398542Smckusickunion dinode {
10498542Smckusick	struct ufs1_dinode dp1;
10598542Smckusick	struct ufs2_dinode dp2;
10669800Stomsoft};
10798542Smckusick#define	DIP(dp, field) \
10898542Smckusick	((sblock.fs_magic == FS_UFS1_MAGIC) ? \
10998542Smckusick	(dp)->dp1.field : (dp)->dp2.field)
11098542Smckusickstatic ufs2_daddr_t 	inoblk;			/* inode block address */
11198542Smckusickstatic char		inobuf[MAXBSIZE];	/* inode block */
11298542Smckusickstatic int		maxino;			/* last valid inode */
11369800Stomsoft
11469800Stomsoft/*
115114067Sschweikh * An array of elements of type struct gfs_bpp describes all blocks to
11669800Stomsoft * be relocated in order to free the space needed for the cylinder group
11769800Stomsoft * summary for all cylinder groups located in the first cylinder group.
11869800Stomsoft */
11969800Stomsoftstruct gfs_bpp {
12098542Smckusick	ufs2_daddr_t	old;		/* old block number */
12198542Smckusick	ufs2_daddr_t	new;		/* new block number */
12269800Stomsoft#define GFS_FL_FIRST	1
12369800Stomsoft#define GFS_FL_LAST	2
12477885Stomsoft	unsigned int	flags;	/* special handling required */
12569800Stomsoft	int	found;		/* how many references were updated */
12669800Stomsoft};
12769800Stomsoft
12869800Stomsoft/* ******************************************************** PROTOTYPES ***** */
12977885Stomsoftstatic void	growfs(int, int, unsigned int);
13098542Smckusickstatic void	rdfs(ufs2_daddr_t, size_t, void *, int);
13198542Smckusickstatic void	wtfs(ufs2_daddr_t, size_t, void *, int, unsigned int);
13298542Smckusickstatic ufs2_daddr_t alloc(void);
13369800Stomsoftstatic int	charsperline(void);
13469926Stomsoftstatic void	usage(void);
13569800Stomsoftstatic int	isblock(struct fs *, unsigned char *, int);
13669800Stomsoftstatic void	clrblock(struct fs *, unsigned char *, int);
13769800Stomsoftstatic void	setblock(struct fs *, unsigned char *, int);
13877885Stomsoftstatic void	initcg(int, time_t, int, unsigned int);
13977885Stomsoftstatic void	updjcg(int, time_t, int, int, unsigned int);
14077885Stomsoftstatic void	updcsloc(time_t, int, int, unsigned int);
14169800Stomsoftstatic struct disklabel	*get_disklabel(int);
14277885Stomsoftstatic void	return_disklabel(int, struct disklabel *, unsigned int);
14398542Smckusickstatic union dinode *ginode(ino_t, int, int);
14498542Smckusickstatic void	frag_adjust(ufs2_daddr_t, int);
14598542Smckusickstatic int	cond_bl_upd(ufs2_daddr_t *, struct gfs_bpp *, int, int,
14698542Smckusick		    unsigned int);
14769800Stomsoftstatic void	updclst(int);
14877885Stomsoftstatic void	updrefs(int, ino_t, struct gfs_bpp *, int, int, unsigned int);
14998542Smckusickstatic void	indirchk(ufs_lbn_t, ufs_lbn_t, ufs2_daddr_t, ufs_lbn_t,
15098542Smckusick		    struct gfs_bpp *, int, int, unsigned int);
15169800Stomsoft
15269800Stomsoft/* ************************************************************ growfs ***** */
15369800Stomsoft/*
154114067Sschweikh * Here we actually start growing the file system. We basically read the
155114067Sschweikh * cylinder summary from the first cylinder group as we want to update
156114067Sschweikh * this on the fly during our various operations. First we handle the
15769800Stomsoft * changes in the former last cylinder group. Afterwards we create all new
158114067Sschweikh * cylinder groups.  Now we handle the cylinder group containing the
159114067Sschweikh * cylinder summary which might result in a relocation of the whole
160114067Sschweikh * structure.  In the end we write back the updated cylinder summary, the
16169800Stomsoft * new superblock, and slightly patched versions of the super block
16269800Stomsoft * copies.
16369800Stomsoft */
16469800Stomsoftstatic void
16577885Stomsoftgrowfs(int fsi, int fso, unsigned int Nflag)
16669800Stomsoft{
16769800Stomsoft	DBG_FUNC("growfs")
16877885Stomsoft	int	i;
16977885Stomsoft	int	cylno, j;
17069800Stomsoft	time_t	utime;
17169800Stomsoft	int	width;
17269800Stomsoft	char	tmpbuf[100];
17369800Stomsoft#ifdef FSIRAND
17469800Stomsoft	static int	randinit=0;
17569800Stomsoft
17669800Stomsoft	DBG_ENTER;
17769800Stomsoft
17869800Stomsoft	if (!randinit) {
17969800Stomsoft		randinit = 1;
18069800Stomsoft		srandomdev();
18169800Stomsoft	}
18269800Stomsoft#else /* not FSIRAND */
18369800Stomsoft
18469800Stomsoft	DBG_ENTER;
18569800Stomsoft
18669800Stomsoft#endif /* FSIRAND */
18769800Stomsoft	time(&utime);
18869800Stomsoft
18969800Stomsoft	/*
19069800Stomsoft	 * Get the cylinder summary into the memory.
19169800Stomsoft	 */
19277885Stomsoft	fscs = (struct csum *)calloc((size_t)1, (size_t)sblock.fs_cssize);
19369926Stomsoft	if(fscs == NULL) {
19469926Stomsoft		errx(1, "calloc failed");
19569926Stomsoft	}
19669800Stomsoft	for (i = 0; i < osblock.fs_cssize; i += osblock.fs_bsize) {
19769800Stomsoft		rdfs(fsbtodb(&osblock, osblock.fs_csaddr +
19877885Stomsoft		    numfrags(&osblock, i)), (size_t)MIN(osblock.fs_cssize - i,
19977885Stomsoft		    osblock.fs_bsize), (void *)(((char *)fscs)+i), fsi);
20069800Stomsoft	}
20169800Stomsoft
20269800Stomsoft#ifdef FS_DEBUG
20369800Stomsoft{
20469800Stomsoft	struct csum	*dbg_csp;
20569800Stomsoft	int	dbg_csc;
20669800Stomsoft	char	dbg_line[80];
20769800Stomsoft
20869800Stomsoft	dbg_csp=fscs;
20969800Stomsoft	for(dbg_csc=0; dbg_csc<osblock.fs_ncg; dbg_csc++) {
21077885Stomsoft		snprintf(dbg_line, sizeof(dbg_line),
21177885Stomsoft		    "%d. old csum in old location", dbg_csc);
21269926Stomsoft		DBG_DUMP_CSUM(&osblock,
21369926Stomsoft		    dbg_line,
21469926Stomsoft		    dbg_csp++);
21569800Stomsoft	}
21669800Stomsoft}
21769800Stomsoft#endif /* FS_DEBUG */
21869800Stomsoft	DBG_PRINT0("fscs read\n");
21969800Stomsoft
22069800Stomsoft	/*
22169800Stomsoft	 * Do all needed changes in the former last cylinder group.
22269800Stomsoft	 */
22369800Stomsoft	updjcg(osblock.fs_ncg-1, utime, fsi, fso, Nflag);
22469800Stomsoft
22569800Stomsoft	/*
226102231Strhodes	 * Dump out summary information about file system.
22769800Stomsoft	 */
22898542Smckusick#	define B2MBFACTOR (1 / (1024.0 * 1024.0))
22998542Smckusick	printf("growfs: %.1fMB (%qd sectors) block size %d, fragment size %d\n",
23069800Stomsoft	    (float)sblock.fs_size * sblock.fs_fsize * B2MBFACTOR,
23198542Smckusick	    fsbtodb(&sblock, sblock.fs_size), sblock.fs_bsize, sblock.fs_fsize);
23298542Smckusick	printf("\tusing %d cylinder groups of %.2fMB, %d blks, %d inodes.\n",
23398542Smckusick	    sblock.fs_ncg, (float)sblock.fs_fpg * sblock.fs_fsize * B2MBFACTOR,
23498542Smckusick	    sblock.fs_fpg / sblock.fs_frag, sblock.fs_ipg);
23598542Smckusick	if (sblock.fs_flags & FS_DOSOFTDEP)
23698542Smckusick		printf("\twith soft updates\n");
23798542Smckusick#	undef B2MBFACTOR
23869800Stomsoft
23969800Stomsoft	/*
24069800Stomsoft	 * Now build the cylinders group blocks and
24169800Stomsoft	 * then print out indices of cylinder groups.
24269800Stomsoft	 */
24369800Stomsoft	printf("super-block backups (for fsck -b #) at:\n");
24469800Stomsoft	i = 0;
24569800Stomsoft	width = charsperline();
24669800Stomsoft
24769800Stomsoft	/*
24869800Stomsoft	 * Iterate for only the new cylinder groups.
24969800Stomsoft	 */
25069800Stomsoft	for (cylno = osblock.fs_ncg; cylno < sblock.fs_ncg; cylno++) {
25169800Stomsoft		initcg(cylno, utime, fso, Nflag);
25277885Stomsoft		j = sprintf(tmpbuf, " %d%s",
25377885Stomsoft		    (int)fsbtodb(&sblock, cgsblock(&sblock, cylno)),
25469800Stomsoft		    cylno < (sblock.fs_ncg-1) ? "," : "" );
25569800Stomsoft		if (i + j >= width) {
25669800Stomsoft			printf("\n");
25769800Stomsoft			i = 0;
25869800Stomsoft		}
25969800Stomsoft		i += j;
26069800Stomsoft		printf("%s", tmpbuf);
26169800Stomsoft		fflush(stdout);
26269800Stomsoft	}
26369800Stomsoft	printf("\n");
26469800Stomsoft
26569800Stomsoft	/*
26669800Stomsoft	 * Do all needed changes in the first cylinder group.
26769800Stomsoft	 * allocate blocks in new location
26869800Stomsoft	 */
26969800Stomsoft	updcsloc(utime, fsi, fso, Nflag);
27069800Stomsoft
27169800Stomsoft	/*
27269800Stomsoft	 * Now write the cylinder summary back to disk.
27369800Stomsoft	 */
27469800Stomsoft	for (i = 0; i < sblock.fs_cssize; i += sblock.fs_bsize) {
27569800Stomsoft		wtfs(fsbtodb(&sblock, sblock.fs_csaddr + numfrags(&sblock, i)),
27677885Stomsoft		    (size_t)MIN(sblock.fs_cssize - i, sblock.fs_bsize),
27777885Stomsoft		    (void *)(((char *)fscs) + i), fso, Nflag);
27869800Stomsoft	}
27969800Stomsoft	DBG_PRINT0("fscs written\n");
28069800Stomsoft
28169800Stomsoft#ifdef FS_DEBUG
28269800Stomsoft{
28369800Stomsoft	struct csum	*dbg_csp;
28469800Stomsoft	int	dbg_csc;
28569800Stomsoft	char	dbg_line[80];
28669800Stomsoft
28769800Stomsoft	dbg_csp=fscs;
28869800Stomsoft	for(dbg_csc=0; dbg_csc<sblock.fs_ncg; dbg_csc++) {
28977885Stomsoft		snprintf(dbg_line, sizeof(dbg_line),
29077885Stomsoft		    "%d. new csum in new location", dbg_csc);
29169926Stomsoft		DBG_DUMP_CSUM(&sblock,
29269926Stomsoft		    dbg_line,
29369926Stomsoft		    dbg_csp++);
29469800Stomsoft	}
29569800Stomsoft}
29669800Stomsoft#endif /* FS_DEBUG */
29769800Stomsoft
29869800Stomsoft	/*
29969800Stomsoft	 * Now write the new superblock back to disk.
30069800Stomsoft	 */
30169800Stomsoft	sblock.fs_time = utime;
30298542Smckusick	wtfs(sblockloc, (size_t)SBLOCKSIZE, (void *)&sblock, fso, Nflag);
30369800Stomsoft	DBG_PRINT0("sblock written\n");
30469926Stomsoft	DBG_DUMP_FS(&sblock,
30569926Stomsoft	    "new initial sblock");
30669800Stomsoft
30769800Stomsoft	/*
30869800Stomsoft	 * Clean up the dynamic fields in our superblock copies.
30969800Stomsoft	 */
31069800Stomsoft	sblock.fs_fmod = 0;
31169800Stomsoft	sblock.fs_clean = 1;
31269800Stomsoft	sblock.fs_ronly = 0;
31369800Stomsoft	sblock.fs_cgrotor = 0;
31469800Stomsoft	sblock.fs_state = 0;
31569800Stomsoft	memset((void *)&sblock.fs_fsmnt, 0, sizeof(sblock.fs_fsmnt));
31669800Stomsoft	sblock.fs_flags &= FS_DOSOFTDEP;
31769800Stomsoft
31869800Stomsoft	/*
31969800Stomsoft	 * XXX
320114067Sschweikh	 * The following fields are currently distributed from the superblock
32169800Stomsoft	 * to the copies:
32269800Stomsoft	 *     fs_minfree
32369800Stomsoft	 *     fs_rotdelay
32469800Stomsoft	 *     fs_maxcontig
32569800Stomsoft	 *     fs_maxbpg
32669800Stomsoft	 *     fs_minfree,
32769800Stomsoft	 *     fs_optim
32869800Stomsoft	 *     fs_flags regarding SOFTPDATES
32969800Stomsoft	 *
33069800Stomsoft	 * We probably should rather change the summary for the cylinder group
33169800Stomsoft	 * statistics here to the value of what would be in there, if the file
332114067Sschweikh	 * system were created initially with the new size. Therefor we still
33369800Stomsoft	 * need to find an easy way of calculating that.
33469800Stomsoft	 * Possibly we can try to read the first superblock copy and apply the
335114067Sschweikh	 * "diffed" stats between the old and new superblock by still copying
33669800Stomsoft	 * certain parameters onto that.
33769800Stomsoft	 */
33869800Stomsoft
33969800Stomsoft	/*
34069800Stomsoft	 * Write out the duplicate super blocks.
34169800Stomsoft	 */
34269800Stomsoft	for (cylno = 0; cylno < sblock.fs_ncg; cylno++) {
34369800Stomsoft		wtfs(fsbtodb(&sblock, cgsblock(&sblock, cylno)),
34498542Smckusick		    (size_t)SBLOCKSIZE, (void *)&sblock, fso, Nflag);
34569800Stomsoft	}
34669800Stomsoft	DBG_PRINT0("sblock copies written\n");
34769926Stomsoft	DBG_DUMP_FS(&sblock,
34869926Stomsoft	    "new other sblocks");
34969800Stomsoft
35069800Stomsoft	DBG_LEAVE;
35169800Stomsoft	return;
35269800Stomsoft}
35369800Stomsoft
35469800Stomsoft/* ************************************************************ initcg ***** */
35569800Stomsoft/*
356114067Sschweikh * This creates a new cylinder group structure, for more details please see
357114067Sschweikh * the source of newfs(8), as this function is taken over almost unchanged.
358114067Sschweikh * As this is never called for the first cylinder group, the special
35969800Stomsoft * provisions for that case are removed here.
36069800Stomsoft */
36169800Stomsoftstatic void
36277885Stomsoftinitcg(int cylno, time_t utime, int fso, unsigned int Nflag)
36369800Stomsoft{
36469800Stomsoft	DBG_FUNC("initcg")
36598542Smckusick	static caddr_t iobuf;
36698542Smckusick	long i, j, d, dlower, dupper, blkno, start;
36798542Smckusick	ufs2_daddr_t cbase, dmax;
36898542Smckusick	struct ufs1_dinode *dp1;
36998542Smckusick	struct ufs2_dinode *dp2;
37092806Sobrien	struct csum *cs;
37169800Stomsoft
37298542Smckusick	if (iobuf == NULL && (iobuf = malloc(sblock.fs_bsize)) == NULL) {
37398542Smckusick		errx(37, "panic: cannot allocate I/O buffer");
37498542Smckusick	}
37569800Stomsoft	/*
37669800Stomsoft	 * Determine block bounds for cylinder group.
37798542Smckusick	 * Allow space for super block summary information in first
37898542Smckusick	 * cylinder group.
37969800Stomsoft	 */
38069800Stomsoft	cbase = cgbase(&sblock, cylno);
38169800Stomsoft	dmax = cbase + sblock.fs_fpg;
38298542Smckusick	if (dmax > sblock.fs_size)
38369800Stomsoft		dmax = sblock.fs_size;
38469800Stomsoft	dlower = cgsblock(&sblock, cylno) - cbase;
38569800Stomsoft	dupper = cgdmin(&sblock, cylno) - cbase;
38698542Smckusick	if (cylno == 0)	/* XXX fscs may be relocated */
38769800Stomsoft		dupper += howmany(sblock.fs_cssize, sblock.fs_fsize);
38898542Smckusick	cs = &fscs[cylno];
38998542Smckusick	memset(&acg, 0, sblock.fs_cgsize);
39069800Stomsoft	acg.cg_time = utime;
39169800Stomsoft	acg.cg_magic = CG_MAGIC;
39269800Stomsoft	acg.cg_cgx = cylno;
39369800Stomsoft	acg.cg_niblk = sblock.fs_ipg;
39498542Smckusick	acg.cg_initediblk = sblock.fs_ipg;
39569800Stomsoft	acg.cg_ndblk = dmax - cbase;
39698542Smckusick	if (sblock.fs_contigsumsize > 0)
39769800Stomsoft		acg.cg_nclusterblks = acg.cg_ndblk / sblock.fs_frag;
39898542Smckusick	start = &acg.cg_space[0] - (u_char *)(&acg.cg_firstfield);
39998542Smckusick	if (sblock.fs_magic == FS_UFS2_MAGIC) {
40098542Smckusick		acg.cg_iusedoff = start;
40198542Smckusick	} else {
40298542Smckusick		acg.cg_old_ncyl = sblock.fs_old_cpg;
40398542Smckusick		acg.cg_old_time = acg.cg_time;
40498542Smckusick		acg.cg_time = 0;
40598542Smckusick		acg.cg_old_niblk = acg.cg_niblk;
40698542Smckusick		acg.cg_niblk = 0;
40798542Smckusick		acg.cg_initediblk = 0;
40898542Smckusick		acg.cg_old_btotoff = start;
40998542Smckusick		acg.cg_old_boff = acg.cg_old_btotoff +
41098542Smckusick		    sblock.fs_old_cpg * sizeof(int32_t);
41198542Smckusick		acg.cg_iusedoff = acg.cg_old_boff +
41298542Smckusick		    sblock.fs_old_cpg * sizeof(u_int16_t);
41369800Stomsoft	}
414103949Smike	acg.cg_freeoff = acg.cg_iusedoff + howmany(sblock.fs_ipg, CHAR_BIT);
415103949Smike	acg.cg_nextfreeoff = acg.cg_freeoff + howmany(sblock.fs_fpg, CHAR_BIT);
41698542Smckusick	if (sblock.fs_contigsumsize > 0) {
41769800Stomsoft		acg.cg_clustersumoff =
41898542Smckusick		    roundup(acg.cg_nextfreeoff, sizeof(u_int32_t));
41998542Smckusick		acg.cg_clustersumoff -= sizeof(u_int32_t);
42069800Stomsoft		acg.cg_clusteroff = acg.cg_clustersumoff +
42169800Stomsoft		    (sblock.fs_contigsumsize + 1) * sizeof(u_int32_t);
42298542Smckusick		acg.cg_nextfreeoff = acg.cg_clusteroff +
423103949Smike		    howmany(fragstoblks(&sblock, sblock.fs_fpg), CHAR_BIT);
42469800Stomsoft	}
42598542Smckusick	if (acg.cg_nextfreeoff > sblock.fs_cgsize) {
42669800Stomsoft		/*
42798542Smckusick		 * This should never happen as we would have had that panic
428102231Strhodes		 * already on file system creation
42969800Stomsoft		 */
43069926Stomsoft		errx(37, "panic: cylinder group too big");
43169800Stomsoft	}
43269800Stomsoft	acg.cg_cs.cs_nifree += sblock.fs_ipg;
43369800Stomsoft	if (cylno == 0)
43498542Smckusick		for (i = 0; i < ROOTINO; i++) {
43569800Stomsoft			setbit(cg_inosused(&acg), i);
43669800Stomsoft			acg.cg_cs.cs_nifree--;
43769800Stomsoft		}
43898542Smckusick	bzero(iobuf, sblock.fs_bsize);
43969800Stomsoft	for (i = 0; i < sblock.fs_ipg / INOPF(&sblock); i += sblock.fs_frag) {
44098542Smckusick		dp1 = (struct ufs1_dinode *)iobuf;
44198542Smckusick		dp2 = (struct ufs2_dinode *)iobuf;
44269800Stomsoft#ifdef FSIRAND
44398542Smckusick		for (j = 0; j < INOPB(&sblock); j++)
44498542Smckusick			if (sblock.fs_magic == FS_UFS1_MAGIC) {
44598542Smckusick				dp1->di_gen = random();
44698542Smckusick				dp1++;
44798542Smckusick			} else {
44898542Smckusick				dp2->di_gen = random();
44998542Smckusick				dp2++;
45098542Smckusick			}
45169800Stomsoft#endif
45269800Stomsoft		wtfs(fsbtodb(&sblock, cgimin(&sblock, cylno) + i),
45398542Smckusick		    sblock.fs_bsize, iobuf, fso, Nflag);
45469800Stomsoft	}
45598542Smckusick	if (cylno > 0) {
45698542Smckusick		/*
45798542Smckusick		 * In cylno 0, beginning space is reserved
45898542Smckusick		 * for boot and super blocks.
45998542Smckusick		 */
46098542Smckusick		for (d = 0; d < dlower; d += sblock.fs_frag) {
46198542Smckusick			blkno = d / sblock.fs_frag;
46298542Smckusick			setblock(&sblock, cg_blksfree(&acg), blkno);
46398542Smckusick			if (sblock.fs_contigsumsize > 0)
46498542Smckusick				setbit(cg_clustersfree(&acg), blkno);
46598542Smckusick			acg.cg_cs.cs_nbfree++;
46669800Stomsoft		}
46798542Smckusick		sblock.fs_dsize += dlower;
46869800Stomsoft	}
46969800Stomsoft	sblock.fs_dsize += acg.cg_ndblk - dupper;
47069800Stomsoft	if ((i = dupper % sblock.fs_frag)) {
47169800Stomsoft		acg.cg_frsum[sblock.fs_frag - i]++;
47269800Stomsoft		for (d = dupper + sblock.fs_frag - i; dupper < d; dupper++) {
47369800Stomsoft			setbit(cg_blksfree(&acg), dupper);
47469800Stomsoft			acg.cg_cs.cs_nffree++;
47569800Stomsoft		}
47669800Stomsoft	}
47798542Smckusick	for (d = dupper; d + sblock.fs_frag <= acg.cg_ndblk;
47898542Smckusick	     d += sblock.fs_frag) {
47969800Stomsoft		blkno = d / sblock.fs_frag;
48069800Stomsoft		setblock(&sblock, cg_blksfree(&acg), blkno);
48198542Smckusick		if (sblock.fs_contigsumsize > 0)
48269800Stomsoft			setbit(cg_clustersfree(&acg), blkno);
48369800Stomsoft		acg.cg_cs.cs_nbfree++;
48469800Stomsoft	}
48598542Smckusick	if (d < acg.cg_ndblk) {
48698542Smckusick		acg.cg_frsum[acg.cg_ndblk - d]++;
48798542Smckusick		for (; d < acg.cg_ndblk; d++) {
48869800Stomsoft			setbit(cg_blksfree(&acg), d);
48969800Stomsoft			acg.cg_cs.cs_nffree++;
49069800Stomsoft		}
49169800Stomsoft	}
49269800Stomsoft	if (sblock.fs_contigsumsize > 0) {
49398542Smckusick		int32_t *sump = cg_clustersum(&acg);
49498542Smckusick		u_char *mapp = cg_clustersfree(&acg);
49598542Smckusick		int map = *mapp++;
49698542Smckusick		int bit = 1;
49798542Smckusick		int run = 0;
49869800Stomsoft
49969800Stomsoft		for (i = 0; i < acg.cg_nclusterblks; i++) {
50098542Smckusick			if ((map & bit) != 0)
50169800Stomsoft				run++;
50298542Smckusick			else if (run != 0) {
50398542Smckusick				if (run > sblock.fs_contigsumsize)
50469800Stomsoft					run = sblock.fs_contigsumsize;
50569800Stomsoft				sump[run]++;
50669800Stomsoft				run = 0;
50769800Stomsoft			}
508103949Smike			if ((i & (CHAR_BIT - 1)) != CHAR_BIT - 1)
50969800Stomsoft				bit <<= 1;
51098542Smckusick			else {
51169800Stomsoft				map = *mapp++;
51269800Stomsoft				bit = 1;
51369800Stomsoft			}
51469800Stomsoft		}
51569800Stomsoft		if (run != 0) {
51698542Smckusick			if (run > sblock.fs_contigsumsize)
51769800Stomsoft				run = sblock.fs_contigsumsize;
51869800Stomsoft			sump[run]++;
51969800Stomsoft		}
52069800Stomsoft	}
52169800Stomsoft	sblock.fs_cstotal.cs_ndir += acg.cg_cs.cs_ndir;
52269800Stomsoft	sblock.fs_cstotal.cs_nffree += acg.cg_cs.cs_nffree;
52369800Stomsoft	sblock.fs_cstotal.cs_nbfree += acg.cg_cs.cs_nbfree;
52469800Stomsoft	sblock.fs_cstotal.cs_nifree += acg.cg_cs.cs_nifree;
52569800Stomsoft	*cs = acg.cg_cs;
52669800Stomsoft	wtfs(fsbtodb(&sblock, cgtod(&sblock, cylno)),
52798542Smckusick		sblock.fs_bsize, (char *)&acg, fso, Nflag);
52869926Stomsoft	DBG_DUMP_CG(&sblock,
52969926Stomsoft	    "new cg",
53069926Stomsoft	    &acg);
53169800Stomsoft
53269800Stomsoft	DBG_LEAVE;
53369800Stomsoft	return;
53469800Stomsoft}
53569800Stomsoft
53669800Stomsoft/* ******************************************************* frag_adjust ***** */
53769800Stomsoft/*
538114067Sschweikh * Here we add or subtract (sign +1/-1) the available fragments in a given
53969800Stomsoft * block to or from the fragment statistics. By subtracting before and adding
540114067Sschweikh * after an operation on the free frag map we can easy update the fragment
541108470Sschweikh * statistic, which seems to be otherwise a rather complex operation.
54269800Stomsoft */
54369800Stomsoftstatic void
54498542Smckusickfrag_adjust(ufs2_daddr_t frag, int sign)
54569800Stomsoft{
54669800Stomsoft	DBG_FUNC("frag_adjust")
54769800Stomsoft	int fragsize;
54869800Stomsoft	int f;
54969800Stomsoft
55069800Stomsoft	DBG_ENTER;
55169800Stomsoft
55269800Stomsoft	fragsize=0;
55369800Stomsoft	/*
55469800Stomsoft	 * Here frag only needs to point to any fragment in the block we want
55569800Stomsoft	 * to examine.
55669800Stomsoft	 */
557114067Sschweikh	for(f=rounddown(frag, sblock.fs_frag);
55869800Stomsoft	    f<roundup(frag+1, sblock.fs_frag);
55969800Stomsoft	    f++) {
56069800Stomsoft		/*
561114067Sschweikh		 * Count contiguous free fragments.
56269800Stomsoft		 */
56369800Stomsoft		if(isset(cg_blksfree(&acg), f)) {
56469800Stomsoft			fragsize++;
56569800Stomsoft		} else {
56669800Stomsoft			if(fragsize && fragsize<sblock.fs_frag) {
56769800Stomsoft				/*
56869800Stomsoft				 * We found something in between.
56969800Stomsoft				 */
57069800Stomsoft				acg.cg_frsum[fragsize]+=sign;
57169926Stomsoft				DBG_PRINT2("frag_adjust [%d]+=%d\n",
57269926Stomsoft				    fragsize,
57369926Stomsoft				    sign);
57469800Stomsoft			}
57569800Stomsoft			fragsize=0;
57669800Stomsoft		}
57769800Stomsoft	}
57869800Stomsoft	if(fragsize && fragsize<sblock.fs_frag) {
57969800Stomsoft		/*
58069800Stomsoft		 * We found something.
58169800Stomsoft		 */
58269800Stomsoft		acg.cg_frsum[fragsize]+=sign;
58369926Stomsoft		DBG_PRINT2("frag_adjust [%d]+=%d\n",
58469926Stomsoft		    fragsize,
58569926Stomsoft		    sign);
58669800Stomsoft	}
58769926Stomsoft	DBG_PRINT2("frag_adjust [[%d]]+=%d\n",
58869926Stomsoft	    fragsize,
58969926Stomsoft	    sign);
59069800Stomsoft
59169800Stomsoft	DBG_LEAVE;
59269800Stomsoft	return;
59369800Stomsoft}
59469800Stomsoft
59569800Stomsoft/* ******************************************************* cond_bl_upd ***** */
59669800Stomsoft/*
59769800Stomsoft * Here we conditionally update a pointer to a fragment. We check for all
598114067Sschweikh * relocated blocks if any of its fragments is referenced by the current
599114067Sschweikh * field, and update the pointer to the respective fragment in our new
600114067Sschweikh * block.  If we find a reference we write back the block immediately,
60169800Stomsoft * as there is no easy way for our general block reading engine to figure
60269800Stomsoft * out if a write back operation is needed.
60369800Stomsoft */
60498542Smckusickstatic int
60598542Smckusickcond_bl_upd(ufs2_daddr_t *block, struct gfs_bpp *field, int fsi, int fso,
60698542Smckusick    unsigned int Nflag)
60769800Stomsoft{
60869800Stomsoft	DBG_FUNC("cond_bl_upd")
60998542Smckusick	struct gfs_bpp *f;
61098542Smckusick	ufs2_daddr_t src, dst;
61198542Smckusick	int fragnum;
61298542Smckusick	void *ibuf;
61369800Stomsoft
61469800Stomsoft	DBG_ENTER;
61569800Stomsoft
61698542Smckusick	f = field;
61798542Smckusick	for (f = field; f->old != 0; f++) {
61898542Smckusick		src = *block;
61998542Smckusick		if (fragstoblks(&sblock, src) != f->old)
62098542Smckusick			continue;
62198542Smckusick		/*
62298542Smckusick		 * The fragment is part of the block, so update.
62398542Smckusick		 */
62498542Smckusick		dst = blkstofrags(&sblock, f->new);
62598542Smckusick		fragnum = fragnum(&sblock, src);
62698542Smckusick		*block = dst + fragnum;
62798542Smckusick		f->found++;
62898542Smckusick		DBG_PRINT3("scg (%d->%d)[%d] reference updated\n",
62998542Smckusick		    f->old,
63098542Smckusick		    f->new,
63198542Smckusick		    fragnum);
63269800Stomsoft
63398542Smckusick		/*
63498542Smckusick		 * Copy the block back immediately.
63598542Smckusick		 *
63698542Smckusick		 * XXX	If src is is from an indirect block we have
63798542Smckusick		 *	to implement copy on write here in case of
63898542Smckusick		 *	active snapshots.
63998542Smckusick		 */
64098542Smckusick		ibuf = malloc(sblock.fs_bsize);
64198542Smckusick		if (!ibuf)
64298542Smckusick			errx(1, "malloc failed");
64398542Smckusick		src -= fragnum;
64498542Smckusick		rdfs(fsbtodb(&sblock, src), (size_t)sblock.fs_bsize, ibuf, fsi);
64598542Smckusick		wtfs(dst, (size_t)sblock.fs_bsize, ibuf, fso, Nflag);
64698542Smckusick		free(ibuf);
64798542Smckusick		/*
64898542Smckusick		 * The same block can't be found again in this loop.
64998542Smckusick		 */
65098542Smckusick		return (1);
65169800Stomsoft	}
65269800Stomsoft
65369800Stomsoft	DBG_LEAVE;
65498542Smckusick	return (0);
65569800Stomsoft}
65669800Stomsoft
65769800Stomsoft/* ************************************************************ updjcg ***** */
65869800Stomsoft/*
65969800Stomsoft * Here we do all needed work for the former last cylinder group. It has to be
660114067Sschweikh * changed in any case, even if the file system ended exactly on the end of
661114067Sschweikh * this group, as there is some slightly inconsistent handling of the number
662114067Sschweikh * of cylinders in the cylinder group. We start again by reading the cylinder
66369800Stomsoft * group from disk. If the last block was not fully available, we first handle
664114067Sschweikh * the missing fragments, then we handle all new full blocks in that file
665114067Sschweikh * system and finally we handle the new last fragmented block in the file
666114067Sschweikh * system.  We again have to handle the fragment statistics rotational layout
66769800Stomsoft * tables and cluster summary during all those operations.
66869800Stomsoft */
66969800Stomsoftstatic void
67077885Stomsoftupdjcg(int cylno, time_t utime, int fsi, int fso, unsigned int Nflag)
67169800Stomsoft{
67269800Stomsoft	DBG_FUNC("updjcg")
67398542Smckusick	ufs2_daddr_t	cbase, dmax, dupper;
67469800Stomsoft	struct csum	*cs;
67569800Stomsoft	int	i,k;
67669800Stomsoft	int	j=0;
67769800Stomsoft
67869800Stomsoft	DBG_ENTER;
67969800Stomsoft
68069800Stomsoft	/*
68169800Stomsoft	 * Read the former last (joining) cylinder group from disk, and make
68269800Stomsoft	 * a copy.
68369800Stomsoft	 */
68477885Stomsoft	rdfs(fsbtodb(&osblock, cgtod(&osblock, cylno)),
68577885Stomsoft	    (size_t)osblock.fs_cgsize, (void *)&aocg, fsi);
68669800Stomsoft	DBG_PRINT0("jcg read\n");
68769926Stomsoft	DBG_DUMP_CG(&sblock,
68869926Stomsoft	    "old joining cg",
68969926Stomsoft	    &aocg);
69069800Stomsoft
69169800Stomsoft	memcpy((void *)&cgun1, (void *)&cgun2, sizeof(cgun2));
69269800Stomsoft
69369800Stomsoft	/*
694114067Sschweikh	 * If the cylinder group had already its new final size almost
69569800Stomsoft	 * nothing is to be done ... except:
69669800Stomsoft	 * For some reason the value of cg_ncyl in the last cylinder group has
697114067Sschweikh	 * to be zero instead of fs_cpg. As this is now no longer the last
69869800Stomsoft	 * cylinder group we have to change that value now to fs_cpg.
699114067Sschweikh	 */
70069800Stomsoft
70169800Stomsoft	if(cgbase(&osblock, cylno+1) == osblock.fs_size) {
70298542Smckusick		if (sblock.fs_magic == FS_UFS1_MAGIC)
70398542Smckusick			acg.cg_old_ncyl=sblock.fs_old_cpg;
70469800Stomsoft
70577885Stomsoft		wtfs(fsbtodb(&sblock, cgtod(&sblock, cylno)),
70677885Stomsoft		    (size_t)sblock.fs_cgsize, (void *)&acg, fso, Nflag);
70769800Stomsoft		DBG_PRINT0("jcg written\n");
70869926Stomsoft		DBG_DUMP_CG(&sblock,
70969926Stomsoft		    "new joining cg",
71069926Stomsoft		    &acg);
71169800Stomsoft
71269800Stomsoft		DBG_LEAVE;
71369800Stomsoft		return;
71469800Stomsoft	}
71569800Stomsoft
71669800Stomsoft	/*
71769800Stomsoft	 * Set up some variables needed later.
71869800Stomsoft	 */
71969800Stomsoft	cbase = cgbase(&sblock, cylno);
72069800Stomsoft	dmax = cbase + sblock.fs_fpg;
72169800Stomsoft	if (dmax > sblock.fs_size)
72269800Stomsoft		dmax = sblock.fs_size;
72369800Stomsoft	dupper = cgdmin(&sblock, cylno) - cbase;
72477885Stomsoft	if (cylno == 0) { /* XXX fscs may be relocated */
72569800Stomsoft		dupper += howmany(sblock.fs_cssize, sblock.fs_fsize);
72669800Stomsoft	}
72769800Stomsoft
72869800Stomsoft	/*
72969800Stomsoft	 * Set pointer to the cylinder summary for our cylinder group.
73069800Stomsoft	 */
73169800Stomsoft	cs = fscs + cylno;
73269800Stomsoft
73369800Stomsoft	/*
73469800Stomsoft	 * Touch the cylinder group, update all fields in the cylinder group as
73569800Stomsoft	 * needed, update the free space in the superblock.
73669800Stomsoft	 */
73769800Stomsoft	acg.cg_time = utime;
73869800Stomsoft	if (cylno == sblock.fs_ncg - 1) {
73969800Stomsoft		/*
74069800Stomsoft		 * This is still the last cylinder group.
74169800Stomsoft		 */
74298542Smckusick		if (sblock.fs_magic == FS_UFS1_MAGIC)
74398542Smckusick			acg.cg_old_ncyl =
74498542Smckusick			    sblock.fs_old_ncyl % sblock.fs_old_cpg;
74569800Stomsoft	} else {
74698542Smckusick		acg.cg_old_ncyl = sblock.fs_old_cpg;
74769800Stomsoft	}
74898542Smckusick	DBG_PRINT2("jcg dbg: %d %u",
74969926Stomsoft	    cylno,
75098542Smckusick	    sblock.fs_ncg);
75198542Smckusick	if (sblock.fs_magic == FS_UFS1_MAGIC)
75298542Smckusick		DBG_PRINT2("%d %u",
75398542Smckusick		    acg.cg_old_ncyl,
75498542Smckusick		    sblock.fs_old_cpg);
75598542Smckusick	DBG_PRINT0("\n");
75669800Stomsoft	acg.cg_ndblk = dmax - cbase;
75769800Stomsoft	sblock.fs_dsize += acg.cg_ndblk-aocg.cg_ndblk;
75869800Stomsoft	if (sblock.fs_contigsumsize > 0) {
75969800Stomsoft		acg.cg_nclusterblks = acg.cg_ndblk / sblock.fs_frag;
76069800Stomsoft	}
76169800Stomsoft
76269800Stomsoft	/*
763114067Sschweikh	 * Now we have to update the free fragment bitmap for our new free
764114067Sschweikh	 * space.  There again we have to handle the fragmentation and also
765114067Sschweikh	 * the rotational layout tables and the cluster summary.  This is
766114067Sschweikh	 * also done per fragment for the first new block if the old file
767114067Sschweikh	 * system end was not on a block boundary, per fragment for the new
768102231Strhodes	 * last block if the new file system end is not on a block boundary,
76969800Stomsoft	 * and per block for all space in between.
77069800Stomsoft	 *
77169800Stomsoft	 * Handle the first new block here if it was partially available
77269800Stomsoft	 * before.
77369800Stomsoft	 */
77469800Stomsoft	if(osblock.fs_size % sblock.fs_frag) {
77569800Stomsoft		if(roundup(osblock.fs_size, sblock.fs_frag)<=sblock.fs_size) {
77669800Stomsoft			/*
77769800Stomsoft			 * The new space is enough to fill at least this
77869800Stomsoft			 * block
77969800Stomsoft			 */
78069800Stomsoft			j=0;
78169800Stomsoft			for(i=roundup(osblock.fs_size-cbase, sblock.fs_frag)-1;
78269800Stomsoft			    i>=osblock.fs_size-cbase;
78369800Stomsoft			    i--) {
78469800Stomsoft				setbit(cg_blksfree(&acg), i);
78569800Stomsoft				acg.cg_cs.cs_nffree++;
78669800Stomsoft				j++;
78769800Stomsoft			}
78869800Stomsoft
78969800Stomsoft			/*
790114067Sschweikh			 * Check if the fragment just created could join an
79169800Stomsoft			 * already existing fragment at the former end of the
792102231Strhodes			 * file system.
79369800Stomsoft			 */
79469800Stomsoft			if(isblock(&sblock, cg_blksfree(&acg),
79569800Stomsoft			    ((osblock.fs_size - cgbase(&sblock, cylno))/
79669800Stomsoft			    sblock.fs_frag))) {
79769800Stomsoft				/*
798114067Sschweikh				 * The block is now completely available.
79969800Stomsoft				 */
80069800Stomsoft				DBG_PRINT0("block was\n");
80169800Stomsoft				acg.cg_frsum[osblock.fs_size%sblock.fs_frag]--;
80269800Stomsoft				acg.cg_cs.cs_nbfree++;
80369800Stomsoft				acg.cg_cs.cs_nffree-=sblock.fs_frag;
80469800Stomsoft				k=rounddown(osblock.fs_size-cbase,
80569800Stomsoft				    sblock.fs_frag);
80669800Stomsoft				updclst((osblock.fs_size-cbase)/sblock.fs_frag);
80769800Stomsoft			} else {
80869800Stomsoft				/*
80969800Stomsoft				 * Lets rejoin a possible partially growed
81069800Stomsoft				 * fragment.
81169800Stomsoft				 */
81269800Stomsoft				k=0;
81369800Stomsoft				while(isset(cg_blksfree(&acg), i) &&
81469800Stomsoft				    (i>=rounddown(osblock.fs_size-cbase,
81569800Stomsoft				    sblock.fs_frag))) {
81669800Stomsoft					i--;
81769800Stomsoft					k++;
81869800Stomsoft				}
81969800Stomsoft				if(k) {
82069800Stomsoft					acg.cg_frsum[k]--;
82169800Stomsoft				}
82269800Stomsoft				acg.cg_frsum[k+j]++;
82369800Stomsoft			}
82469800Stomsoft		} else {
82569800Stomsoft			/*
82669800Stomsoft			 * We only grow by some fragments within this last
82769800Stomsoft			 * block.
82869800Stomsoft			 */
82969800Stomsoft			for(i=sblock.fs_size-cbase-1;
83069800Stomsoft				i>=osblock.fs_size-cbase;
83169800Stomsoft				i--) {
83269800Stomsoft				setbit(cg_blksfree(&acg), i);
83369800Stomsoft				acg.cg_cs.cs_nffree++;
83469800Stomsoft				j++;
83569800Stomsoft			}
83669800Stomsoft			/*
83769800Stomsoft			 * Lets rejoin a possible partially growed fragment.
83869800Stomsoft			 */
83969800Stomsoft			k=0;
84069800Stomsoft			while(isset(cg_blksfree(&acg), i) &&
84169800Stomsoft			    (i>=rounddown(osblock.fs_size-cbase,
84269800Stomsoft			    sblock.fs_frag))) {
84369800Stomsoft				i--;
84469800Stomsoft				k++;
84569800Stomsoft			}
84669800Stomsoft			if(k) {
84769800Stomsoft				acg.cg_frsum[k]--;
84869800Stomsoft			}
84969800Stomsoft			acg.cg_frsum[k+j]++;
85069800Stomsoft		}
85169800Stomsoft	}
85269800Stomsoft
85369800Stomsoft	/*
85469800Stomsoft	 * Handle all new complete blocks here.
85569800Stomsoft	 */
85669800Stomsoft	for(i=roundup(osblock.fs_size-cbase, sblock.fs_frag);
85769800Stomsoft	    i+sblock.fs_frag<=dmax-cbase;	/* XXX <= or only < ? */
85869800Stomsoft	    i+=sblock.fs_frag) {
85969800Stomsoft		j = i / sblock.fs_frag;
86069800Stomsoft		setblock(&sblock, cg_blksfree(&acg), j);
86169800Stomsoft		updclst(j);
86269800Stomsoft		acg.cg_cs.cs_nbfree++;
86369800Stomsoft	}
86469800Stomsoft
86569800Stomsoft	/*
86669800Stomsoft	 * Handle the last new block if there are stll some new fragments left.
867114067Sschweikh	 * Here we don't have to bother about the cluster summary or the even
86869800Stomsoft	 * the rotational layout table.
86969800Stomsoft	 */
87069800Stomsoft	if (i < (dmax - cbase)) {
87169800Stomsoft		acg.cg_frsum[dmax - cbase - i]++;
87269800Stomsoft		for (; i < dmax - cbase; i++) {
87369800Stomsoft			setbit(cg_blksfree(&acg), i);
87469800Stomsoft			acg.cg_cs.cs_nffree++;
87569800Stomsoft		}
87669800Stomsoft	}
87769800Stomsoft
87869800Stomsoft	sblock.fs_cstotal.cs_nffree +=
87969800Stomsoft	    (acg.cg_cs.cs_nffree - aocg.cg_cs.cs_nffree);
88069800Stomsoft	sblock.fs_cstotal.cs_nbfree +=
88169800Stomsoft	    (acg.cg_cs.cs_nbfree - aocg.cg_cs.cs_nbfree);
88269800Stomsoft	/*
88369800Stomsoft	 * The following statistics are not changed here:
88469800Stomsoft	 *     sblock.fs_cstotal.cs_ndir
88569800Stomsoft	 *     sblock.fs_cstotal.cs_nifree
88669800Stomsoft	 * As the statistics for this cylinder group are ready, copy it to
88769800Stomsoft	 * the summary information array.
88869800Stomsoft	 */
88969800Stomsoft	*cs = acg.cg_cs;
89069800Stomsoft
89169800Stomsoft	/*
89269800Stomsoft	 * Write the updated "joining" cylinder group back to disk.
89369800Stomsoft	 */
89477885Stomsoft	wtfs(fsbtodb(&sblock, cgtod(&sblock, cylno)), (size_t)sblock.fs_cgsize,
89577885Stomsoft	    (void *)&acg, fso, Nflag);
89669800Stomsoft	DBG_PRINT0("jcg written\n");
89769926Stomsoft	DBG_DUMP_CG(&sblock,
89869926Stomsoft	    "new joining cg",
89969926Stomsoft	    &acg);
90069800Stomsoft
90169800Stomsoft	DBG_LEAVE;
90269800Stomsoft	return;
90369800Stomsoft}
90469800Stomsoft
90569800Stomsoft/* ********************************************************** updcsloc ***** */
90669800Stomsoft/*
907114067Sschweikh * Here we update the location of the cylinder summary. We have two possible
90869800Stomsoft * ways of growing the cylinder summary.
909114067Sschweikh * (1)	We can try to grow the summary in the current location, and relocate
91069800Stomsoft *	possibly used blocks within the current cylinder group.
91169800Stomsoft * (2)	Alternatively we can relocate the whole cylinder summary to the first
912114067Sschweikh *	new completely empty cylinder group. Once the cylinder summary is no
913114067Sschweikh *	longer in the beginning of the first cylinder group you should never
914114067Sschweikh *	use a version of fsck which is not aware of the possibility to have
91569800Stomsoft *	this structure in a non standard place.
916114067Sschweikh * Option (1) is considered to be less intrusive to the structure of the file-
91769800Stomsoft * system. So we try to stick to that whenever possible. If there is not enough
918114067Sschweikh * space in the cylinder group containing the cylinder summary we have to use
919114067Sschweikh * method (2). In case of active snapshots in the file system we probably can
92069800Stomsoft * completely avoid implementing copy on write if we stick to method (2) only.
92169800Stomsoft */
92269800Stomsoftstatic void
92377885Stomsoftupdcsloc(time_t utime, int fsi, int fso, unsigned int Nflag)
92469800Stomsoft{
92569800Stomsoft	DBG_FUNC("updcsloc")
92669800Stomsoft	struct csum	*cs;
92769800Stomsoft	int	ocscg, ncscg;
92869800Stomsoft	int	blocks;
92998542Smckusick	ufs2_daddr_t	cbase, dupper, odupper, d, f, g;
93069800Stomsoft	int	ind;
93169800Stomsoft	int	cylno, inc;
93269800Stomsoft	struct gfs_bpp	*bp;
93369800Stomsoft	int	i, l;
93469800Stomsoft	int	lcs=0;
93569800Stomsoft	int	block;
93669800Stomsoft
93769800Stomsoft	DBG_ENTER;
93869800Stomsoft
93969800Stomsoft	if(howmany(sblock.fs_cssize, sblock.fs_fsize) ==
94069800Stomsoft	    howmany(osblock.fs_cssize, osblock.fs_fsize)) {
94169800Stomsoft		/*
94269800Stomsoft		 * No new fragment needed.
94369800Stomsoft		 */
94469800Stomsoft		DBG_LEAVE;
94569800Stomsoft		return;
94669800Stomsoft	}
94769800Stomsoft	ocscg=dtog(&osblock, osblock.fs_csaddr);
94869800Stomsoft	cs=fscs+ocscg;
94969800Stomsoft	blocks = 1+howmany(sblock.fs_cssize, sblock.fs_bsize)-
95069800Stomsoft	    howmany(osblock.fs_cssize, osblock.fs_bsize);
95169800Stomsoft
95269800Stomsoft	/*
95369800Stomsoft	 * Read original cylinder group from disk, and make a copy.
95481311Schm	 * XXX	If Nflag is set in some very rare cases we now miss
95581311Schm	 *	some changes done in updjcg by reading the unmodified
95681311Schm	 *	block from disk.
95769800Stomsoft	 */
95877885Stomsoft	rdfs(fsbtodb(&osblock, cgtod(&osblock, ocscg)),
95977885Stomsoft	    (size_t)osblock.fs_cgsize, (void *)&aocg, fsi);
96069800Stomsoft	DBG_PRINT0("oscg read\n");
96169926Stomsoft	DBG_DUMP_CG(&sblock,
96269926Stomsoft	    "old summary cg",
96369926Stomsoft	    &aocg);
96469800Stomsoft
96569800Stomsoft	memcpy((void *)&cgun1, (void *)&cgun2, sizeof(cgun2));
96669800Stomsoft
96769800Stomsoft	/*
96869800Stomsoft	 * Touch the cylinder group, set up local variables needed later
96969800Stomsoft	 * and update the superblock.
97069800Stomsoft	 */
97169800Stomsoft	acg.cg_time = utime;
97269800Stomsoft
97369800Stomsoft	/*
97469800Stomsoft	 * XXX	In the case of having active snapshots we may need much more
975114067Sschweikh	 *	blocks for the copy on write. We need each block twice, and
976114067Sschweikh	 *	also up to 8*3 blocks for indirect blocks for all possible
97769800Stomsoft	 *	references.
97869800Stomsoft	 */
97969800Stomsoft	if(/*((int)sblock.fs_time&0x3)>0||*/ cs->cs_nbfree < blocks) {
98069800Stomsoft		/*
981114067Sschweikh		 * There is not enough space in the old cylinder group to
982114067Sschweikh		 * relocate all blocks as needed, so we relocate the whole
983114067Sschweikh		 * cylinder group summary to a new group. We try to use the
98469800Stomsoft		 * first complete new cylinder group just created. Within the
985114067Sschweikh		 * cylinder group we align the area immediately after the
986114067Sschweikh		 * cylinder group information location in order to be as
98769800Stomsoft		 * close as possible to the original implementation of ffs.
98869800Stomsoft		 *
989114067Sschweikh		 * First we have to make sure we'll find enough space in the
990114067Sschweikh		 * new cylinder group. If not, then we currently give up.
991114067Sschweikh		 * We start with freeing everything which was used by the
99269800Stomsoft		 * fragments of the old cylinder summary in the current group.
993114067Sschweikh		 * Now we write back the group meta data, read in the needed
99469800Stomsoft		 * meta data from the new cylinder group, and start allocating
995114067Sschweikh		 * within that group. Here we can assume, the group to be
99669800Stomsoft		 * completely empty. Which makes the handling of fragments and
99769800Stomsoft		 * clusters a lot easier.
99869800Stomsoft		 */
99969800Stomsoft		DBG_TRC;
100069800Stomsoft		if(sblock.fs_ncg-osblock.fs_ncg < 2) {
100169926Stomsoft			errx(2, "panic: not enough space");
100269800Stomsoft		}
100369800Stomsoft
100469800Stomsoft		/*
100569800Stomsoft		 * Point "d" to the first fragment not used by the cylinder
100669800Stomsoft		 * summary.
100769800Stomsoft		 */
100869800Stomsoft		d=osblock.fs_csaddr+(osblock.fs_cssize/osblock.fs_fsize);
100969800Stomsoft
101069800Stomsoft		/*
101169800Stomsoft		 * Set up last cluster size ("lcs") already here. Calculate
1012114067Sschweikh		 * the size for the trailing cluster just behind where "d"
101369800Stomsoft		 * points to.
101469800Stomsoft		 */
101569800Stomsoft		if(sblock.fs_contigsumsize > 0) {
101669800Stomsoft			for(block=howmany(d%sblock.fs_fpg, sblock.fs_frag),
101769800Stomsoft			    lcs=0; lcs<sblock.fs_contigsumsize;
101869800Stomsoft			    block++, lcs++) {
101969800Stomsoft				if(isclr(cg_clustersfree(&acg), block)){
102069800Stomsoft					break;
102169800Stomsoft				}
102269800Stomsoft			}
102369800Stomsoft		}
102469800Stomsoft
102569800Stomsoft		/*
102669800Stomsoft		 * Point "d" to the last frag used by the cylinder summary.
102769800Stomsoft		 */
102869800Stomsoft		d--;
102969800Stomsoft
103069926Stomsoft		DBG_PRINT1("d=%d\n",
103169926Stomsoft		    d);
103269800Stomsoft		if((d+1)%sblock.fs_frag) {
103369800Stomsoft			/*
103469800Stomsoft			 * The end of the cylinder summary is not a complete
103569800Stomsoft			 * block.
103669800Stomsoft			 */
103769800Stomsoft			DBG_TRC;
103869800Stomsoft			frag_adjust(d%sblock.fs_fpg, -1);
103969800Stomsoft			for(; (d+1)%sblock.fs_frag; d--) {
104069926Stomsoft				DBG_PRINT1("d=%d\n",
104169926Stomsoft				    d);
104269800Stomsoft				setbit(cg_blksfree(&acg), d%sblock.fs_fpg);
104369800Stomsoft				acg.cg_cs.cs_nffree++;
104469800Stomsoft				sblock.fs_cstotal.cs_nffree++;
104569800Stomsoft			}
104669800Stomsoft			/*
1047114067Sschweikh			 * Point "d" to the last fragment of the last
1048114067Sschweikh			 * (incomplete) block of the cylinder summary.
104969800Stomsoft			 */
105069800Stomsoft			d++;
105169800Stomsoft			frag_adjust(d%sblock.fs_fpg, 1);
105269800Stomsoft
105369800Stomsoft			if(isblock(&sblock, cg_blksfree(&acg),
105469800Stomsoft			    (d%sblock.fs_fpg)/sblock.fs_frag)) {
1055114067Sschweikh				DBG_PRINT1("d=%d\n", d);
105669800Stomsoft				acg.cg_cs.cs_nffree-=sblock.fs_frag;
105769800Stomsoft				acg.cg_cs.cs_nbfree++;
105869800Stomsoft				sblock.fs_cstotal.cs_nffree-=sblock.fs_frag;
105969800Stomsoft				sblock.fs_cstotal.cs_nbfree++;
106069800Stomsoft				if(sblock.fs_contigsumsize > 0) {
106169800Stomsoft					setbit(cg_clustersfree(&acg),
106269800Stomsoft					    (d%sblock.fs_fpg)/sblock.fs_frag);
106369800Stomsoft					if(lcs < sblock.fs_contigsumsize) {
106469800Stomsoft						if(lcs) {
106569800Stomsoft							cg_clustersum(&acg)
106669800Stomsoft							    [lcs]--;
106769800Stomsoft						}
106869800Stomsoft						lcs++;
106969800Stomsoft						cg_clustersum(&acg)[lcs]++;
107069800Stomsoft					}
107169800Stomsoft				}
107269800Stomsoft			}
107369800Stomsoft			/*
107469800Stomsoft			 * Point "d" to the first fragment of the block before
107569800Stomsoft			 * the last incomplete block.
107669800Stomsoft			 */
107769800Stomsoft			d--;
107869800Stomsoft		}
107969800Stomsoft
1080114067Sschweikh		DBG_PRINT1("d=%d\n", d);
108169800Stomsoft		for(d=rounddown(d, sblock.fs_frag); d >= osblock.fs_csaddr;
108269800Stomsoft		    d-=sblock.fs_frag) {
108369800Stomsoft			DBG_TRC;
1084114067Sschweikh			DBG_PRINT1("d=%d\n", d);
108569800Stomsoft			setblock(&sblock, cg_blksfree(&acg),
108669800Stomsoft			    (d%sblock.fs_fpg)/sblock.fs_frag);
108769800Stomsoft			acg.cg_cs.cs_nbfree++;
108869800Stomsoft			sblock.fs_cstotal.cs_nbfree++;
108969800Stomsoft			if(sblock.fs_contigsumsize > 0) {
109069800Stomsoft				setbit(cg_clustersfree(&acg),
109169800Stomsoft				    (d%sblock.fs_fpg)/sblock.fs_frag);
109269800Stomsoft				/*
109369800Stomsoft				 * The last cluster size is already set up.
109469800Stomsoft				 */
109569800Stomsoft				if(lcs < sblock.fs_contigsumsize) {
109669800Stomsoft					if(lcs) {
109769800Stomsoft						cg_clustersum(&acg)[lcs]--;
109869800Stomsoft					}
109969800Stomsoft					lcs++;
110069800Stomsoft					cg_clustersum(&acg)[lcs]++;
110169800Stomsoft				}
110269800Stomsoft			}
110369800Stomsoft		}
110469800Stomsoft		*cs = acg.cg_cs;
110569800Stomsoft
110669800Stomsoft		/*
110769800Stomsoft		 * Now write the former cylinder group containing the cylinder
110869800Stomsoft		 * summary back to disk.
110969800Stomsoft		 */
111077885Stomsoft		wtfs(fsbtodb(&sblock, cgtod(&sblock, ocscg)),
111177885Stomsoft		    (size_t)sblock.fs_cgsize, (void *)&acg, fso, Nflag);
111269800Stomsoft		DBG_PRINT0("oscg written\n");
111369926Stomsoft		DBG_DUMP_CG(&sblock,
111469926Stomsoft		    "old summary cg",
111569926Stomsoft		    &acg);
111669800Stomsoft
111769800Stomsoft		/*
111869800Stomsoft		 * Find the beginning of the new cylinder group containing the
111969800Stomsoft		 * cylinder summary.
112069800Stomsoft		 */
112169800Stomsoft		sblock.fs_csaddr=cgdmin(&sblock, osblock.fs_ncg);
112269800Stomsoft		ncscg=dtog(&sblock, sblock.fs_csaddr);
112369800Stomsoft		cs=fscs+ncscg;
112469800Stomsoft
112581311Schm
112669800Stomsoft		/*
112781311Schm		 * If Nflag is specified, we would now read random data instead
112881311Schm		 * of an empty cg structure from disk. So we can't simulate that
112981311Schm		 * part for now.
113081311Schm		 */
113181311Schm		if(Nflag) {
113281311Schm			DBG_PRINT0("nscg update skipped\n");
113381311Schm			DBG_LEAVE;
113481311Schm			return;
113581311Schm		}
113681311Schm
113781311Schm		/*
113869800Stomsoft		 * Read the future cylinder group containing the cylinder
113969800Stomsoft		 * summary from disk, and make a copy.
114069800Stomsoft		 */
114169800Stomsoft		rdfs(fsbtodb(&sblock, cgtod(&sblock, ncscg)),
114277885Stomsoft		    (size_t)sblock.fs_cgsize, (void *)&aocg, fsi);
114369800Stomsoft		DBG_PRINT0("nscg read\n");
114469926Stomsoft		DBG_DUMP_CG(&sblock,
114569926Stomsoft		    "new summary cg",
114669926Stomsoft		    &aocg);
114769800Stomsoft
114869800Stomsoft		memcpy((void *)&cgun1, (void *)&cgun2, sizeof(cgun2));
114969800Stomsoft
115069800Stomsoft		/*
115169800Stomsoft		 * Allocate all complete blocks used by the new cylinder
115269800Stomsoft		 * summary.
115369800Stomsoft		 */
115469800Stomsoft		for(d=sblock.fs_csaddr; d+sblock.fs_frag <=
115569800Stomsoft		    sblock.fs_csaddr+(sblock.fs_cssize/sblock.fs_fsize);
115669800Stomsoft		    d+=sblock.fs_frag) {
115769800Stomsoft			clrblock(&sblock, cg_blksfree(&acg),
115869800Stomsoft			    (d%sblock.fs_fpg)/sblock.fs_frag);
115969800Stomsoft			acg.cg_cs.cs_nbfree--;
116069800Stomsoft			sblock.fs_cstotal.cs_nbfree--;
116169800Stomsoft			if(sblock.fs_contigsumsize > 0) {
116269800Stomsoft				clrbit(cg_clustersfree(&acg),
116369800Stomsoft				    (d%sblock.fs_fpg)/sblock.fs_frag);
116469800Stomsoft			}
116569800Stomsoft		}
116669800Stomsoft
116769800Stomsoft		/*
116869800Stomsoft		 * Allocate all fragments used by the cylinder summary in the
116969800Stomsoft		 * last block.
117069800Stomsoft		 */
117169800Stomsoft		if(d<sblock.fs_csaddr+(sblock.fs_cssize/sblock.fs_fsize)) {
117269800Stomsoft			for(; d-sblock.fs_csaddr<
117369800Stomsoft			    sblock.fs_cssize/sblock.fs_fsize;
117469800Stomsoft			    d++) {
117569800Stomsoft				clrbit(cg_blksfree(&acg), d%sblock.fs_fpg);
117669800Stomsoft				acg.cg_cs.cs_nffree--;
117769800Stomsoft				sblock.fs_cstotal.cs_nffree--;
117869800Stomsoft			}
117969800Stomsoft			acg.cg_cs.cs_nbfree--;
118069800Stomsoft			acg.cg_cs.cs_nffree+=sblock.fs_frag;
118169800Stomsoft			sblock.fs_cstotal.cs_nbfree--;
118269800Stomsoft			sblock.fs_cstotal.cs_nffree+=sblock.fs_frag;
118369800Stomsoft			if(sblock.fs_contigsumsize > 0) {
118469800Stomsoft				clrbit(cg_clustersfree(&acg),
118569800Stomsoft				    (d%sblock.fs_fpg)/sblock.fs_frag);
118669800Stomsoft			}
118769800Stomsoft
118869800Stomsoft			frag_adjust(d%sblock.fs_fpg, +1);
118969800Stomsoft		}
119069800Stomsoft		/*
1191114067Sschweikh		 * XXX	Handle the cluster statistics here in the case this
119269800Stomsoft		 *	cylinder group is now almost full, and the remaining
119369800Stomsoft		 *	space is less then the maximum cluster size. This is
119469800Stomsoft		 *	probably not needed, as you would hardly find a file
119569800Stomsoft		 *	system which has only MAXCSBUFS+FS_MAXCONTIG of free
119669800Stomsoft		 *	space right behind the cylinder group information in
119769800Stomsoft		 *	any new cylinder group.
119869800Stomsoft		 */
119969800Stomsoft
120069800Stomsoft		/*
120169800Stomsoft		 * Update our statistics in the cylinder summary.
120269800Stomsoft		 */
120369800Stomsoft		*cs = acg.cg_cs;
120469800Stomsoft
120569800Stomsoft		/*
120669800Stomsoft		 * Write the new cylinder group containing the cylinder summary
120769800Stomsoft		 * back to disk.
120869800Stomsoft		 */
120977885Stomsoft		wtfs(fsbtodb(&sblock, cgtod(&sblock, ncscg)),
121077885Stomsoft		    (size_t)sblock.fs_cgsize, (void *)&acg, fso, Nflag);
121169800Stomsoft		DBG_PRINT0("nscg written\n");
121269926Stomsoft		DBG_DUMP_CG(&sblock,
121369926Stomsoft		    "new summary cg",
121469926Stomsoft		    &acg);
121569800Stomsoft
121669800Stomsoft		DBG_LEAVE;
121769800Stomsoft		return;
121869800Stomsoft	}
121969800Stomsoft	/*
122069800Stomsoft	 * We have got enough of space in the current cylinder group, so we
1221114067Sschweikh	 * can relocate just a few blocks, and let the summary information
122269800Stomsoft	 * grow in place where it is right now.
122369800Stomsoft	 */
122469800Stomsoft	DBG_TRC;
122569800Stomsoft
122669800Stomsoft	cbase = cgbase(&osblock, ocscg);	/* old and new are equal */
122769800Stomsoft	dupper = sblock.fs_csaddr - cbase +
122869800Stomsoft	    howmany(sblock.fs_cssize, sblock.fs_fsize);
122969800Stomsoft	odupper = osblock.fs_csaddr - cbase +
123069800Stomsoft	    howmany(osblock.fs_cssize, osblock.fs_fsize);
123169800Stomsoft
123269800Stomsoft	sblock.fs_dsize -= dupper-odupper;
123369800Stomsoft
123469800Stomsoft	/*
123569800Stomsoft	 * Allocate the space for the array of blocks to be relocated.
123669800Stomsoft	 */
123769800Stomsoft 	bp=(struct gfs_bpp *)malloc(((dupper-odupper)/sblock.fs_frag+2)*
123869800Stomsoft	    sizeof(struct gfs_bpp));
123969926Stomsoft	if(bp == NULL) {
124069926Stomsoft		errx(1, "malloc failed");
124169926Stomsoft	}
124277779Stomsoft	memset((char *)bp, 0, ((dupper-odupper)/sblock.fs_frag+2)*
124377779Stomsoft	    sizeof(struct gfs_bpp));
124469800Stomsoft
124569800Stomsoft	/*
1246114067Sschweikh	 * Lock all new frags needed for the cylinder group summary. This is
1247114067Sschweikh	 * done per fragment in the first and last block of the new required
124869800Stomsoft	 * area, and per block for all other blocks.
124969800Stomsoft	 *
1250114067Sschweikh	 * Handle the first new block here (but only if some fragments where
125169800Stomsoft	 * already used for the cylinder summary).
125269800Stomsoft	 */
125369800Stomsoft	ind=0;
125469800Stomsoft	frag_adjust(odupper, -1);
125569800Stomsoft	for(d=odupper; ((d<dupper)&&(d%sblock.fs_frag)); d++) {
125669926Stomsoft		DBG_PRINT1("scg first frag check loop d=%d\n",
125769926Stomsoft		    d);
125869800Stomsoft		if(isclr(cg_blksfree(&acg), d)) {
125969800Stomsoft			if (!ind) {
126069800Stomsoft				bp[ind].old=d/sblock.fs_frag;
126169800Stomsoft				bp[ind].flags|=GFS_FL_FIRST;
126269800Stomsoft				if(roundup(d, sblock.fs_frag) >= dupper) {
126369800Stomsoft					bp[ind].flags|=GFS_FL_LAST;
126469800Stomsoft				}
126569800Stomsoft				ind++;
126669800Stomsoft			}
126769800Stomsoft		} else {
126869800Stomsoft			clrbit(cg_blksfree(&acg), d);
126969800Stomsoft			acg.cg_cs.cs_nffree--;
127069800Stomsoft			sblock.fs_cstotal.cs_nffree--;
127169800Stomsoft		}
127269800Stomsoft		/*
127369800Stomsoft		 * No cluster handling is needed here, as there was at least
1274114067Sschweikh		 * one fragment in use by the cylinder summary in the old
1275102231Strhodes		 * file system.
127669800Stomsoft		 * No block-free counter handling here as this block was not
127769800Stomsoft		 * a free block.
127869800Stomsoft		 */
127969800Stomsoft	}
128069800Stomsoft	frag_adjust(odupper, 1);
128169800Stomsoft
128269800Stomsoft	/*
128369800Stomsoft	 * Handle all needed complete blocks here.
128469800Stomsoft	 */
128569800Stomsoft	for(; d+sblock.fs_frag<=dupper; d+=sblock.fs_frag) {
128669926Stomsoft		DBG_PRINT1("scg block check loop d=%d\n",
128769926Stomsoft		    d);
128869800Stomsoft		if(!isblock(&sblock, cg_blksfree(&acg), d/sblock.fs_frag)) {
128969800Stomsoft			for(f=d; f<d+sblock.fs_frag; f++) {
129069800Stomsoft				if(isset(cg_blksfree(&aocg), f)) {
129169800Stomsoft					acg.cg_cs.cs_nffree--;
129269800Stomsoft					sblock.fs_cstotal.cs_nffree--;
129369800Stomsoft				}
129469800Stomsoft			}
129569800Stomsoft			clrblock(&sblock, cg_blksfree(&acg), d/sblock.fs_frag);
129669800Stomsoft			bp[ind].old=d/sblock.fs_frag;
129769800Stomsoft			ind++;
129869800Stomsoft		} else {
129969800Stomsoft			clrblock(&sblock, cg_blksfree(&acg), d/sblock.fs_frag);
130069800Stomsoft			acg.cg_cs.cs_nbfree--;
130169800Stomsoft			sblock.fs_cstotal.cs_nbfree--;
130269800Stomsoft			if(sblock.fs_contigsumsize > 0) {
130369800Stomsoft				clrbit(cg_clustersfree(&acg), d/sblock.fs_frag);
130469800Stomsoft				for(lcs=0, l=(d/sblock.fs_frag)+1;
130569800Stomsoft				    lcs<sblock.fs_contigsumsize;
130669800Stomsoft				    l++, lcs++ ) {
130769800Stomsoft					if(isclr(cg_clustersfree(&acg),l)){
130869800Stomsoft						break;
130969800Stomsoft					}
131069800Stomsoft				}
131169800Stomsoft				if(lcs < sblock.fs_contigsumsize) {
131269800Stomsoft					cg_clustersum(&acg)[lcs+1]--;
131369800Stomsoft					if(lcs) {
131469800Stomsoft						cg_clustersum(&acg)[lcs]++;
131569800Stomsoft					}
131669800Stomsoft				}
131769800Stomsoft			}
131869800Stomsoft		}
131969800Stomsoft		/*
132069800Stomsoft		 * No fragment counter handling is needed here, as this finally
132169800Stomsoft		 * doesn't change after the relocation.
132269800Stomsoft		 */
132369800Stomsoft	}
132469800Stomsoft
132569800Stomsoft	/*
132669800Stomsoft	 * Handle all fragments needed in the last new affected block.
132769800Stomsoft	 */
132869800Stomsoft	if(d<dupper) {
132969800Stomsoft		frag_adjust(dupper-1, -1);
133069800Stomsoft
133169800Stomsoft		if(isblock(&sblock, cg_blksfree(&acg), d/sblock.fs_frag)) {
133269800Stomsoft			acg.cg_cs.cs_nbfree--;
133369800Stomsoft			sblock.fs_cstotal.cs_nbfree--;
133469800Stomsoft			acg.cg_cs.cs_nffree+=sblock.fs_frag;
133569800Stomsoft			sblock.fs_cstotal.cs_nffree+=sblock.fs_frag;
133669800Stomsoft			if(sblock.fs_contigsumsize > 0) {
133769800Stomsoft				clrbit(cg_clustersfree(&acg), d/sblock.fs_frag);
133869800Stomsoft				for(lcs=0, l=(d/sblock.fs_frag)+1;
133969800Stomsoft				    lcs<sblock.fs_contigsumsize;
134069800Stomsoft				    l++, lcs++ ) {
134169800Stomsoft					if(isclr(cg_clustersfree(&acg),l)){
134269800Stomsoft						break;
134369800Stomsoft					}
134469800Stomsoft				}
134569800Stomsoft				if(lcs < sblock.fs_contigsumsize) {
134669800Stomsoft					cg_clustersum(&acg)[lcs+1]--;
134769800Stomsoft					if(lcs) {
134869800Stomsoft						cg_clustersum(&acg)[lcs]++;
134969800Stomsoft					}
135069800Stomsoft				}
135169800Stomsoft			}
135269800Stomsoft		}
135369800Stomsoft
135469800Stomsoft		for(; d<dupper; d++) {
135569926Stomsoft			DBG_PRINT1("scg second frag check loop d=%d\n",
135669926Stomsoft			    d);
135769800Stomsoft			if(isclr(cg_blksfree(&acg), d)) {
135869800Stomsoft				bp[ind].old=d/sblock.fs_frag;
135969800Stomsoft				bp[ind].flags|=GFS_FL_LAST;
136069800Stomsoft			} else {
136169800Stomsoft				clrbit(cg_blksfree(&acg), d);
136269800Stomsoft				acg.cg_cs.cs_nffree--;
136369800Stomsoft				sblock.fs_cstotal.cs_nffree--;
136469800Stomsoft			}
136569800Stomsoft		}
136669800Stomsoft		if(bp[ind].flags & GFS_FL_LAST) { /* we have to advance here */
136769800Stomsoft			ind++;
136869800Stomsoft		}
136969800Stomsoft		frag_adjust(dupper-1, 1);
137069800Stomsoft	}
137169800Stomsoft
137269800Stomsoft	/*
137369800Stomsoft	 * If we found a block to relocate just do so.
137469800Stomsoft	 */
137569800Stomsoft	if(ind) {
137669800Stomsoft		for(i=0; i<ind; i++) {
137769800Stomsoft			if(!bp[i].old) { /* no more blocks listed */
137869800Stomsoft				/*
137969800Stomsoft				 * XXX	A relative blocknumber should not be
1380114067Sschweikh				 *	zero, which is not explicitly
138169800Stomsoft				 *	guaranteed by our code.
138269800Stomsoft				 */
138369800Stomsoft				break;
138469800Stomsoft			}
138569800Stomsoft			/*
138669800Stomsoft			 * Allocate a complete block in the same (current)
138769800Stomsoft			 * cylinder group.
138869800Stomsoft			 */
138969800Stomsoft			bp[i].new=alloc()/sblock.fs_frag;
139069800Stomsoft
139169800Stomsoft			/*
139269800Stomsoft			 * There is no frag_adjust() needed for the new block
139369800Stomsoft			 * as it will have no fragments yet :-).
139469800Stomsoft			 */
139569800Stomsoft			for(f=bp[i].old*sblock.fs_frag,
139669800Stomsoft			    g=bp[i].new*sblock.fs_frag;
139769800Stomsoft			    f<(bp[i].old+1)*sblock.fs_frag;
139869800Stomsoft			    f++, g++) {
139969800Stomsoft				if(isset(cg_blksfree(&aocg), f)) {
140069800Stomsoft					setbit(cg_blksfree(&acg), g);
140169800Stomsoft					acg.cg_cs.cs_nffree++;
140269800Stomsoft					sblock.fs_cstotal.cs_nffree++;
140369800Stomsoft				}
140469800Stomsoft			}
140569800Stomsoft
140669800Stomsoft			/*
1407114067Sschweikh			 * Special handling is required if this was the first
140869800Stomsoft			 * block. We have to consider the fragments which were
1409114067Sschweikh			 * used by the cylinder summary in the original block
1410114067Sschweikh			 * which re to be free in the copy of our block.  We
1411114067Sschweikh			 * have to be careful if this first block happens to
141269800Stomsoft			 * be also the last block to be relocated.
141369800Stomsoft			 */
141469800Stomsoft			if(bp[i].flags & GFS_FL_FIRST) {
141569800Stomsoft				for(f=bp[i].old*sblock.fs_frag,
141669800Stomsoft				    g=bp[i].new*sblock.fs_frag;
141769800Stomsoft				    f<odupper;
141869800Stomsoft				    f++, g++) {
141969800Stomsoft					setbit(cg_blksfree(&acg), g);
142069800Stomsoft					acg.cg_cs.cs_nffree++;
142169800Stomsoft					sblock.fs_cstotal.cs_nffree++;
142269800Stomsoft				}
142369800Stomsoft				if(!(bp[i].flags & GFS_FL_LAST)) {
142469800Stomsoft					frag_adjust(bp[i].new*sblock.fs_frag,1);
142569800Stomsoft				}
142669800Stomsoft			}
142769800Stomsoft
142869800Stomsoft			/*
142969800Stomsoft			 * Special handling is required if this is the last
143069800Stomsoft			 * block to be relocated.
143169800Stomsoft			 */
143269800Stomsoft			if(bp[i].flags & GFS_FL_LAST) {
143369800Stomsoft				frag_adjust(bp[i].new*sblock.fs_frag, 1);
143469800Stomsoft				frag_adjust(bp[i].old*sblock.fs_frag, -1);
143569800Stomsoft				for(f=dupper;
143669800Stomsoft				    f<roundup(dupper, sblock.fs_frag);
143769800Stomsoft				    f++) {
143869800Stomsoft					if(isclr(cg_blksfree(&acg), f)) {
143969800Stomsoft						setbit(cg_blksfree(&acg), f);
144069800Stomsoft						acg.cg_cs.cs_nffree++;
144169800Stomsoft						sblock.fs_cstotal.cs_nffree++;
144269800Stomsoft					}
144369800Stomsoft				}
144469800Stomsoft				frag_adjust(bp[i].old*sblock.fs_frag, 1);
144569800Stomsoft			}
144669800Stomsoft
144769800Stomsoft			/*
1448114067Sschweikh			 * !!! Attach the cylindergroup offset here.
144969800Stomsoft			 */
145069800Stomsoft			bp[i].old+=cbase/sblock.fs_frag;
145169800Stomsoft			bp[i].new+=cbase/sblock.fs_frag;
145269800Stomsoft
145369800Stomsoft			/*
145469800Stomsoft			 * Copy the content of the block.
145569800Stomsoft			 */
145669800Stomsoft			/*
145769800Stomsoft			 * XXX	Here we will have to implement a copy on write
145869800Stomsoft			 *	in the case we have any active snapshots.
145969800Stomsoft			 */
146069800Stomsoft			rdfs(fsbtodb(&sblock, bp[i].old*sblock.fs_frag),
146177885Stomsoft			    (size_t)sblock.fs_bsize, (void *)&ablk, fsi);
146269800Stomsoft			wtfs(fsbtodb(&sblock, bp[i].new*sblock.fs_frag),
146377885Stomsoft			    (size_t)sblock.fs_bsize, (void *)&ablk, fso, Nflag);
146469926Stomsoft			DBG_DUMP_HEX(&sblock,
146569926Stomsoft			    "copied full block",
146669926Stomsoft			    (unsigned char *)&ablk);
146769800Stomsoft
146869926Stomsoft			DBG_PRINT2("scg (%d->%d) block relocated\n",
146969926Stomsoft			    bp[i].old,
147069926Stomsoft			    bp[i].new);
147169800Stomsoft		}
147269800Stomsoft
147369800Stomsoft		/*
147469800Stomsoft		 * Now we have to update all references to any fragment which
1475114067Sschweikh		 * belongs to any block relocated. We iterate now over all
1476114067Sschweikh		 * cylinder groups, within those over all non zero length
147769800Stomsoft		 * inodes.
147869800Stomsoft		 */
147969800Stomsoft		for(cylno=0; cylno<osblock.fs_ncg; cylno++) {
148069926Stomsoft			DBG_PRINT1("scg doing cg (%d)\n",
148169926Stomsoft			    cylno);
148269800Stomsoft			for(inc=osblock.fs_ipg-1 ; inc>=0 ; inc--) {
148369800Stomsoft				updrefs(cylno, (ino_t)inc, bp, fsi, fso, Nflag);
148469800Stomsoft			}
148569800Stomsoft		}
148669800Stomsoft
148769800Stomsoft		/*
148869800Stomsoft		 * All inodes are checked, now make sure the number of
148969800Stomsoft		 * references found make sense.
149069800Stomsoft		 */
149169800Stomsoft		for(i=0; i<ind; i++) {
149269800Stomsoft			if(!bp[i].found || (bp[i].found>sblock.fs_frag)) {
149369926Stomsoft				warnx("error: %d refs found for block %d.",
149469800Stomsoft				    bp[i].found, bp[i].old);
149569800Stomsoft			}
149669800Stomsoft
149769800Stomsoft		}
149869800Stomsoft	}
149969800Stomsoft	/*
150069800Stomsoft	 * The following statistics are not changed here:
150169800Stomsoft	 *     sblock.fs_cstotal.cs_ndir
150269800Stomsoft	 *     sblock.fs_cstotal.cs_nifree
150369800Stomsoft	 * The following statistics were already updated on the fly:
150469800Stomsoft	 *     sblock.fs_cstotal.cs_nffree
150569800Stomsoft	 *     sblock.fs_cstotal.cs_nbfree
150669800Stomsoft	 * As the statistics for this cylinder group are ready, copy it to
150769800Stomsoft	 * the summary information array.
150869800Stomsoft	 */
150969800Stomsoft
151069800Stomsoft	*cs = acg.cg_cs;
151169800Stomsoft
151269800Stomsoft	/*
151369800Stomsoft	 * Write summary cylinder group back to disk.
151469800Stomsoft	 */
151577885Stomsoft	wtfs(fsbtodb(&sblock, cgtod(&sblock, ocscg)), (size_t)sblock.fs_cgsize,
151677885Stomsoft	    (void *)&acg, fso, Nflag);
151769800Stomsoft	DBG_PRINT0("scg written\n");
151869926Stomsoft	DBG_DUMP_CG(&sblock,
151969926Stomsoft	    "new summary cg",
152069926Stomsoft	    &acg);
152169800Stomsoft
152269800Stomsoft	DBG_LEAVE;
152369800Stomsoft	return;
152469800Stomsoft}
152569800Stomsoft
152669800Stomsoft/* ************************************************************** rdfs ***** */
152769800Stomsoft/*
152869800Stomsoft * Here we read some block(s) from disk.
152969800Stomsoft */
153069800Stomsoftstatic void
153198542Smckusickrdfs(ufs2_daddr_t bno, size_t size, void *bf, int fsi)
153269800Stomsoft{
153369800Stomsoft	DBG_FUNC("rdfs")
153477885Stomsoft	ssize_t	n;
153569800Stomsoft
153669800Stomsoft	DBG_ENTER;
153769800Stomsoft
153869800Stomsoft	if (lseek(fsi, (off_t)bno * DEV_BSIZE, 0) < 0) {
153969926Stomsoft		err(33, "rdfs: seek error: %ld", (long)bno);
154069800Stomsoft	}
154177885Stomsoft	n = read(fsi, bf, size);
154277885Stomsoft	if (n != (ssize_t)size) {
154369926Stomsoft		err(34, "rdfs: read error: %ld", (long)bno);
154469800Stomsoft	}
154569800Stomsoft
154669800Stomsoft	DBG_LEAVE;
154769800Stomsoft	return;
154869800Stomsoft}
154969800Stomsoft
155069800Stomsoft/* ************************************************************** wtfs ***** */
155169800Stomsoft/*
155269800Stomsoft * Here we write some block(s) to disk.
155369800Stomsoft */
155469800Stomsoftstatic void
155598542Smckusickwtfs(ufs2_daddr_t bno, size_t size, void *bf, int fso, unsigned int Nflag)
155669800Stomsoft{
155769800Stomsoft	DBG_FUNC("wtfs")
155877885Stomsoft	ssize_t	n;
155969800Stomsoft
156069800Stomsoft	DBG_ENTER;
156169800Stomsoft
156269800Stomsoft	if (Nflag) {
156369800Stomsoft		DBG_LEAVE;
156469800Stomsoft		return;
156569800Stomsoft	}
156669800Stomsoft	if (lseek(fso, (off_t)bno * DEV_BSIZE, SEEK_SET) < 0) {
156769926Stomsoft		err(35, "wtfs: seek error: %ld", (long)bno);
156869800Stomsoft	}
156977885Stomsoft	n = write(fso, bf, size);
157077885Stomsoft	if (n != (ssize_t)size) {
157169926Stomsoft		err(36, "wtfs: write error: %ld", (long)bno);
157269800Stomsoft	}
157369800Stomsoft
157469800Stomsoft	DBG_LEAVE;
157569800Stomsoft	return;
157669800Stomsoft}
157769800Stomsoft
157869800Stomsoft/* ************************************************************* alloc ***** */
157969800Stomsoft/*
158069800Stomsoft * Here we allocate a free block in the current cylinder group. It is assumed,
1581114067Sschweikh * that acg contains the current cylinder group. As we may take a block from
1582102231Strhodes * somewhere in the file system we have to handle cluster summary here.
158369800Stomsoft */
158498542Smckusickstatic ufs2_daddr_t
158569800Stomsoftalloc(void)
158669800Stomsoft{
158769800Stomsoft	DBG_FUNC("alloc")
158898542Smckusick	ufs2_daddr_t	d, blkno;
158969800Stomsoft	int	lcs1, lcs2;
159069800Stomsoft	int	l;
159169800Stomsoft	int	csmin, csmax;
159269800Stomsoft	int	dlower, dupper, dmax;
159369800Stomsoft
159469800Stomsoft	DBG_ENTER;
159569800Stomsoft
159669800Stomsoft	if (acg.cg_magic != CG_MAGIC) {
159769926Stomsoft		warnx("acg: bad magic number");
159869800Stomsoft		DBG_LEAVE;
159969800Stomsoft		return (0);
160069800Stomsoft	}
160169800Stomsoft	if (acg.cg_cs.cs_nbfree == 0) {
160269926Stomsoft		warnx("error: cylinder group ran out of space");
160369800Stomsoft		DBG_LEAVE;
160469800Stomsoft		return (0);
160569800Stomsoft	}
160669800Stomsoft	/*
160769800Stomsoft	 * We start seeking for free blocks only from the space available after
1608114067Sschweikh	 * the end of the new grown cylinder summary. Otherwise we allocate a
160969800Stomsoft	 * block here which we have to relocate a couple of seconds later again
161069800Stomsoft	 * again, and we are not prepared to to this anyway.
161169800Stomsoft	 */
161269800Stomsoft	blkno=-1;
161369800Stomsoft	dlower=cgsblock(&sblock, acg.cg_cgx)-cgbase(&sblock, acg.cg_cgx);
161469800Stomsoft	dupper=cgdmin(&sblock, acg.cg_cgx)-cgbase(&sblock, acg.cg_cgx);
161569800Stomsoft	dmax=cgbase(&sblock, acg.cg_cgx)+sblock.fs_fpg;
161669800Stomsoft	if (dmax > sblock.fs_size) {
161769800Stomsoft		dmax = sblock.fs_size;
161869800Stomsoft	}
161969800Stomsoft	dmax-=cgbase(&sblock, acg.cg_cgx); /* retransform into cg */
162069800Stomsoft	csmin=sblock.fs_csaddr-cgbase(&sblock, acg.cg_cgx);
162169800Stomsoft	csmax=csmin+howmany(sblock.fs_cssize, sblock.fs_fsize);
162269926Stomsoft	DBG_PRINT3("seek range: dl=%d, du=%d, dm=%d\n",
162369926Stomsoft	    dlower,
162469926Stomsoft	    dupper,
162569926Stomsoft	    dmax);
162669926Stomsoft	DBG_PRINT2("range cont: csmin=%d, csmax=%d\n",
162769926Stomsoft	    csmin,
162869926Stomsoft	    csmax);
162969800Stomsoft
163069800Stomsoft	for(d=0; (d<dlower && blkno==-1); d+=sblock.fs_frag) {
163169800Stomsoft		if(d>=csmin && d<=csmax) {
163269800Stomsoft			continue;
163369800Stomsoft		}
163469800Stomsoft		if(isblock(&sblock, cg_blksfree(&acg), fragstoblks(&sblock,
163569800Stomsoft		    d))) {
163669800Stomsoft			blkno = fragstoblks(&sblock, d);/* Yeah found a block */
163769800Stomsoft			break;
163869800Stomsoft		}
163969800Stomsoft	}
164069800Stomsoft	for(d=dupper; (d<dmax && blkno==-1); d+=sblock.fs_frag) {
164169800Stomsoft		if(d>=csmin && d<=csmax) {
164269800Stomsoft			continue;
164369800Stomsoft		}
164469800Stomsoft		if(isblock(&sblock, cg_blksfree(&acg), fragstoblks(&sblock,
164569800Stomsoft		    d))) {
164669800Stomsoft			blkno = fragstoblks(&sblock, d);/* Yeah found a block */
164769800Stomsoft			break;
164869800Stomsoft		}
164969800Stomsoft	}
165069800Stomsoft	if(blkno==-1) {
165169926Stomsoft		warnx("internal error: couldn't find promised block in cg");
165269800Stomsoft		DBG_LEAVE;
165369800Stomsoft		return (0);
165469800Stomsoft	}
165569800Stomsoft
165669800Stomsoft	/*
165769800Stomsoft	 * This is needed if the block was found already in the first loop.
165869800Stomsoft	 */
165969800Stomsoft	d=blkstofrags(&sblock, blkno);
166069800Stomsoft
166169800Stomsoft	clrblock(&sblock, cg_blksfree(&acg), blkno);
166269800Stomsoft	if (sblock.fs_contigsumsize > 0) {
166369800Stomsoft		/*
166469800Stomsoft		 * Handle the cluster allocation bitmap.
166569800Stomsoft		 */
166669800Stomsoft		clrbit(cg_clustersfree(&acg), blkno);
166769800Stomsoft		/*
1668114067Sschweikh		 * We possibly have split a cluster here, so we have to do
166969926Stomsoft		 * recalculate the sizes of the remaining cluster halves now,
167069800Stomsoft		 * and use them for updating the cluster summary information.
167169800Stomsoft		 *
167269800Stomsoft		 * Lets start with the blocks before our allocated block ...
167369800Stomsoft		 */
167469800Stomsoft		for(lcs1=0, l=blkno-1; lcs1<sblock.fs_contigsumsize;
167569800Stomsoft		    l--, lcs1++ ) {
167669800Stomsoft			if(isclr(cg_clustersfree(&acg),l)){
167769800Stomsoft				break;
167869800Stomsoft			}
167969800Stomsoft		}
168069800Stomsoft		/*
168169800Stomsoft		 * ... and continue with the blocks right after our allocated
168269800Stomsoft		 * block.
168369800Stomsoft		 */
168469800Stomsoft		for(lcs2=0, l=blkno+1; lcs2<sblock.fs_contigsumsize;
168569800Stomsoft		    l++, lcs2++ ) {
168669800Stomsoft			if(isclr(cg_clustersfree(&acg),l)){
168769800Stomsoft				break;
168869800Stomsoft			}
168969800Stomsoft		}
169069800Stomsoft
169169800Stomsoft		/*
169269800Stomsoft		 * Now update all counters.
169369800Stomsoft		 */
169469800Stomsoft		cg_clustersum(&acg)[MIN(lcs1+lcs2+1,sblock.fs_contigsumsize)]--;
169569800Stomsoft		if(lcs1) {
169669800Stomsoft			cg_clustersum(&acg)[lcs1]++;
169769800Stomsoft		}
169869800Stomsoft		if(lcs2) {
169969800Stomsoft			cg_clustersum(&acg)[lcs2]++;
170069800Stomsoft		}
170169800Stomsoft	}
170269800Stomsoft	/*
170369800Stomsoft	 * Update all statistics based on blocks.
170469800Stomsoft	 */
170569800Stomsoft	acg.cg_cs.cs_nbfree--;
170669800Stomsoft	sblock.fs_cstotal.cs_nbfree--;
170769800Stomsoft
170869800Stomsoft	DBG_LEAVE;
170969800Stomsoft	return (d);
171069800Stomsoft}
171169800Stomsoft
171269800Stomsoft/* *********************************************************** isblock ***** */
171369800Stomsoft/*
1714114067Sschweikh * Here we check if all frags of a block are free. For more details again
171569800Stomsoft * please see the source of newfs(8), as this function is taken over almost
171669800Stomsoft * unchanged.
171769800Stomsoft */
171869800Stomsoftstatic int
171969800Stomsoftisblock(struct fs *fs, unsigned char *cp, int h)
172069800Stomsoft{
172169800Stomsoft	DBG_FUNC("isblock")
172269800Stomsoft	unsigned char	mask;
172369800Stomsoft
172469800Stomsoft	DBG_ENTER;
172569800Stomsoft
172669800Stomsoft	switch (fs->fs_frag) {
172769800Stomsoft	case 8:
172869800Stomsoft		DBG_LEAVE;
172969800Stomsoft		return (cp[h] == 0xff);
173069800Stomsoft	case 4:
173169800Stomsoft		mask = 0x0f << ((h & 0x1) << 2);
173269800Stomsoft		DBG_LEAVE;
173369800Stomsoft		return ((cp[h >> 1] & mask) == mask);
173469800Stomsoft	case 2:
173569800Stomsoft		mask = 0x03 << ((h & 0x3) << 1);
173669800Stomsoft		DBG_LEAVE;
173769800Stomsoft		return ((cp[h >> 2] & mask) == mask);
173869800Stomsoft	case 1:
173969800Stomsoft		mask = 0x01 << (h & 0x7);
174069800Stomsoft		DBG_LEAVE;
174169800Stomsoft		return ((cp[h >> 3] & mask) == mask);
174269800Stomsoft	default:
174369800Stomsoft		fprintf(stderr, "isblock bad fs_frag %d\n", fs->fs_frag);
174469800Stomsoft		DBG_LEAVE;
174569800Stomsoft		return (0);
174669800Stomsoft	}
174769800Stomsoft}
174869800Stomsoft
174969800Stomsoft/* ********************************************************** clrblock ***** */
175069800Stomsoft/*
175169800Stomsoft * Here we allocate a complete block in the block map. For more details again
1752114067Sschweikh * please see the source of newfs(8), as this function is taken over almost
175369800Stomsoft * unchanged.
175469800Stomsoft */
175569800Stomsoftstatic void
175669800Stomsoftclrblock(struct fs *fs, unsigned char *cp, int h)
175769800Stomsoft{
175869800Stomsoft	DBG_FUNC("clrblock")
175969800Stomsoft
176069800Stomsoft	DBG_ENTER;
176169800Stomsoft
176269800Stomsoft	switch ((fs)->fs_frag) {
176369800Stomsoft	case 8:
176469800Stomsoft		cp[h] = 0;
176569800Stomsoft		break;
176669800Stomsoft	case 4:
176769800Stomsoft		cp[h >> 1] &= ~(0x0f << ((h & 0x1) << 2));
176869800Stomsoft		break;
176969800Stomsoft	case 2:
177069800Stomsoft		cp[h >> 2] &= ~(0x03 << ((h & 0x3) << 1));
177169800Stomsoft		break;
177269800Stomsoft	case 1:
177369800Stomsoft		cp[h >> 3] &= ~(0x01 << (h & 0x7));
177469800Stomsoft		break;
177569800Stomsoft	default:
177669926Stomsoft		warnx("clrblock bad fs_frag %d", fs->fs_frag);
177769800Stomsoft		break;
177869800Stomsoft	}
177969800Stomsoft
178069800Stomsoft	DBG_LEAVE;
178169800Stomsoft	return;
178269800Stomsoft}
178369800Stomsoft
178469800Stomsoft/* ********************************************************** setblock ***** */
178569800Stomsoft/*
178669800Stomsoft * Here we free a complete block in the free block map. For more details again
1787114067Sschweikh * please see the source of newfs(8), as this function is taken over almost
178869800Stomsoft * unchanged.
178969800Stomsoft */
179069800Stomsoftstatic void
179169800Stomsoftsetblock(struct fs *fs, unsigned char *cp, int h)
179269800Stomsoft{
179369800Stomsoft	DBG_FUNC("setblock")
179469800Stomsoft
179569800Stomsoft	DBG_ENTER;
179669800Stomsoft
179769800Stomsoft	switch (fs->fs_frag) {
179869800Stomsoft	case 8:
179969800Stomsoft		cp[h] = 0xff;
180069800Stomsoft		break;
180169800Stomsoft	case 4:
180269800Stomsoft		cp[h >> 1] |= (0x0f << ((h & 0x1) << 2));
180369800Stomsoft		break;
180469800Stomsoft	case 2:
180569800Stomsoft		cp[h >> 2] |= (0x03 << ((h & 0x3) << 1));
180669800Stomsoft		break;
180769800Stomsoft	case 1:
180869800Stomsoft		cp[h >> 3] |= (0x01 << (h & 0x7));
180969800Stomsoft		break;
181069800Stomsoft	default:
181169926Stomsoft		warnx("setblock bad fs_frag %d", fs->fs_frag);
181269800Stomsoft		break;
181369800Stomsoft	}
181469800Stomsoft
181569800Stomsoft	DBG_LEAVE;
181669800Stomsoft	return;
181769800Stomsoft}
181869800Stomsoft
181969800Stomsoft/* ************************************************************ ginode ***** */
182069800Stomsoft/*
182169800Stomsoft * This function provides access to an individual inode. We find out in which
1822114067Sschweikh * block the requested inode is located, read it from disk if needed, and
1823114067Sschweikh * return the pointer into that block. We maintain a cache of one block to
1824114067Sschweikh * not read the same block again and again if we iterate linearly over all
182569800Stomsoft * inodes.
182669800Stomsoft */
182798542Smckusickstatic union dinode *
182869800Stomsoftginode(ino_t inumber, int fsi, int cg)
182969800Stomsoft{
183069800Stomsoft	DBG_FUNC("ginode")
183198542Smckusick	static ino_t	startinum = 0;	/* first inode in cached block */
183269800Stomsoft
183369800Stomsoft	DBG_ENTER;
183469800Stomsoft
183598542Smckusick	inumber += (cg * sblock.fs_ipg);
183698542Smckusick	if (inumber < ROOTINO || inumber > maxino)
183798542Smckusick		errx(8, "bad inode number %d to ginode", inumber);
183898542Smckusick	if (startinum == 0 ||
183998542Smckusick	    inumber < startinum || inumber >= startinum + INOPB(&sblock)) {
184098542Smckusick		inoblk = fsbtodb(&sblock, ino_to_fsba(&sblock, inumber));
184198542Smckusick		rdfs(inoblk, (size_t)sblock.fs_bsize, inobuf, fsi);
184269800Stomsoft		startinum = (inumber / INOPB(&sblock)) * INOPB(&sblock);
184369800Stomsoft	}
184469800Stomsoft	DBG_LEAVE;
184598542Smckusick	if (sblock.fs_magic == FS_UFS1_MAGIC)
184698542Smckusick		return ((union dinode *)
184798542Smckusick		    &((struct ufs1_dinode *)inobuf)[inumber % INOPB(&sblock)]);
184898542Smckusick	return ((union dinode *)
184998542Smckusick	    &((struct ufs2_dinode *)inobuf)[inumber % INOPB(&sblock)]);
185069800Stomsoft}
185169800Stomsoft
185269800Stomsoft/* ****************************************************** charsperline ***** */
185369800Stomsoft/*
185469800Stomsoft * Figure out how many lines our current terminal has. For more details again
1855114067Sschweikh * please see the source of newfs(8), as this function is taken over almost
185669800Stomsoft * unchanged.
185769800Stomsoft */
185869800Stomsoftstatic int
185969800Stomsoftcharsperline(void)
186069800Stomsoft{
186169800Stomsoft	DBG_FUNC("charsperline")
186269800Stomsoft	int	columns;
186369800Stomsoft	char	*cp;
186469800Stomsoft	struct winsize	ws;
186569800Stomsoft
186669800Stomsoft	DBG_ENTER;
186769800Stomsoft
186869800Stomsoft	columns = 0;
186969800Stomsoft	if (ioctl(0, TIOCGWINSZ, &ws) != -1) {
187069800Stomsoft		columns = ws.ws_col;
187169800Stomsoft	}
187269800Stomsoft	if (columns == 0 && (cp = getenv("COLUMNS"))) {
187369800Stomsoft		columns = atoi(cp);
187469800Stomsoft	}
187569800Stomsoft	if (columns == 0) {
187669800Stomsoft		columns = 80;	/* last resort */
187769800Stomsoft	}
187869800Stomsoft
187969800Stomsoft	DBG_LEAVE;
188069800Stomsoft	return columns;
188169800Stomsoft}
188269800Stomsoft
188369800Stomsoft/* ************************************************************** main ***** */
188469800Stomsoft/*
1885114067Sschweikh * growfs(8)  is a utility which allows to increase the size of an existing
1886114067Sschweikh * ufs file system. Currently this can only be done on unmounted file system.
1887114067Sschweikh * It recognizes some command line options to specify the new desired size,
1888114067Sschweikh * and it does some basic checkings. The old file system size is determined
1889114067Sschweikh * and after some more checks like we can really access the new last block
189069800Stomsoft * on the disk etc. we calculate the new parameters for the superblock. After
1891114067Sschweikh * having done this we just call growfs() which will do the work.  Before
189269800Stomsoft * we finish the only thing left is to update the disklabel.
189369800Stomsoft * We still have to provide support for snapshots. Therefore we first have to
1894114067Sschweikh * understand what data structures are always replicated in the snapshot on
1895114067Sschweikh * creation, for all other blocks we touch during our procedure, we have to
189669926Stomsoft * keep the old blocks unchanged somewhere available for the snapshots. If we
1897114067Sschweikh * are lucky, then we only have to handle our blocks to be relocated in that
189869800Stomsoft * way.
1899114067Sschweikh * Also we have to consider in what order we actually update the critical
1900102231Strhodes * data structures of the file system to make sure, that in case of a disaster
190169800Stomsoft * fsck(8) is still able to restore any lost data.
1902114067Sschweikh * The foreseen last step then will be to provide for growing even mounted
1903114067Sschweikh * file systems. There we have to extend the mount() system call to provide
1904102231Strhodes * userland access to the file system locking facility.
190569800Stomsoft */
190669800Stomsoftint
190769800Stomsoftmain(int argc, char **argv)
190869800Stomsoft{
190969800Stomsoft	DBG_FUNC("main")
191069926Stomsoft	char	*device, *special, *cp;
191169800Stomsoft	char	ch;
191277885Stomsoft	unsigned int	size=0;
191369800Stomsoft	size_t	len;
191477885Stomsoft	unsigned int	Nflag=0;
191569800Stomsoft	int	ExpertFlag=0;
191669800Stomsoft	struct stat	st;
191769800Stomsoft	struct disklabel	*lp;
191869800Stomsoft	struct partition	*pp;
191998542Smckusick	int	i,fsi,fso;
192069800Stomsoft	char	reply[5];
192169800Stomsoft#ifdef FSMAXSNAP
192269800Stomsoft	int	j;
192369800Stomsoft#endif /* FSMAXSNAP */
192469800Stomsoft
192569800Stomsoft	DBG_ENTER;
192669800Stomsoft
192769800Stomsoft	while((ch=getopt(argc, argv, "Ns:vy")) != -1) {
192869800Stomsoft		switch(ch) {
192969800Stomsoft		case 'N':
193069800Stomsoft			Nflag=1;
193169800Stomsoft			break;
193269800Stomsoft		case 's':
193369800Stomsoft			size=(size_t)atol(optarg);
193469800Stomsoft			if(size<1) {
193569926Stomsoft				usage();
193669800Stomsoft			}
193769800Stomsoft			break;
193869800Stomsoft		case 'v': /* for compatibility to newfs */
193969800Stomsoft			break;
194069800Stomsoft		case 'y':
194169800Stomsoft			ExpertFlag=1;
194269800Stomsoft			break;
194369800Stomsoft		case '?':
194469800Stomsoft			/* FALLTHROUGH */
194569800Stomsoft		default:
194669926Stomsoft			usage();
194769800Stomsoft		}
194869800Stomsoft	}
194969800Stomsoft	argc -= optind;
195069800Stomsoft	argv += optind;
195169800Stomsoft
195269800Stomsoft	if(argc != 1) {
195369926Stomsoft		usage();
195469800Stomsoft	}
195569800Stomsoft	device=*argv;
195669800Stomsoft
195769800Stomsoft	/*
195869800Stomsoft	 * Now try to guess the (raw)device name.
195969800Stomsoft	 */
196069800Stomsoft	if (0 == strrchr(device, '/')) {
196169800Stomsoft		/*
196269800Stomsoft		 * No path prefix was given, so try in that order:
196369800Stomsoft		 *     /dev/r%s
196469800Stomsoft		 *     /dev/%s
196569800Stomsoft		 *     /dev/vinum/r%s
196669800Stomsoft		 *     /dev/vinum/%s.
1967114067Sschweikh		 *
1968114067Sschweikh		 * FreeBSD now doesn't distinguish between raw and block
196969800Stomsoft		 * devices any longer, but it should still work this way.
197069800Stomsoft		 */
197169800Stomsoft		len=strlen(device)+strlen(_PATH_DEV)+2+strlen("vinum/");
197269800Stomsoft		special=(char *)malloc(len);
197369926Stomsoft		if(special == NULL) {
197469926Stomsoft			errx(1, "malloc failed");
197569926Stomsoft		}
197669800Stomsoft		snprintf(special, len, "%sr%s", _PATH_DEV, device);
197769800Stomsoft		if (stat(special, &st) == -1) {
197869800Stomsoft			snprintf(special, len, "%s%s", _PATH_DEV, device);
197969800Stomsoft			if (stat(special, &st) == -1) {
198069800Stomsoft				snprintf(special, len, "%svinum/r%s",
198169800Stomsoft				    _PATH_DEV, device);
198269800Stomsoft				if (stat(special, &st) == -1) {
198369800Stomsoft					/* For now this is the 'last resort' */
198469800Stomsoft					snprintf(special, len, "%svinum/%s",
198569800Stomsoft					    _PATH_DEV, device);
198669800Stomsoft				}
198769800Stomsoft			}
198869800Stomsoft		}
198969800Stomsoft		device = special;
199069800Stomsoft	}
199169800Stomsoft
199269800Stomsoft	/*
199369800Stomsoft	 * Try to access our devices for writing ...
199469800Stomsoft	 */
199569800Stomsoft	if (Nflag) {
199669800Stomsoft		fso = -1;
199769800Stomsoft	} else {
199869800Stomsoft		fso = open(device, O_WRONLY);
199969800Stomsoft		if (fso < 0) {
200069926Stomsoft			err(1, "%s", device);
200169800Stomsoft		}
200269800Stomsoft	}
200369800Stomsoft
200469800Stomsoft	/*
200569800Stomsoft	 * ... and reading.
200669800Stomsoft	 */
200769800Stomsoft	fsi = open(device, O_RDONLY);
200869800Stomsoft	if (fsi < 0) {
200969926Stomsoft		err(1, "%s", device);
201069800Stomsoft	}
201169800Stomsoft
201269800Stomsoft	/*
2013114067Sschweikh	 * Try to read a label and guess the slice if not specified. This
2014114067Sschweikh	 * code should guess the right thing and avoid to bother the user
2015114067Sschweikh	 * with the task of specifying the option -v on vinum volumes.
201669800Stomsoft	 */
201769800Stomsoft	cp=device+strlen(device)-1;
201869800Stomsoft	lp = get_disklabel(fsi);
201969800Stomsoft	if(lp->d_type == DTYPE_VINUM) {
202069800Stomsoft		pp = &lp->d_partitions[0];
202169800Stomsoft	} else if (isdigit(*cp)) {
202269800Stomsoft		pp = &lp->d_partitions[2];
202369800Stomsoft	} else if (*cp>='a' && *cp<='h') {
202469800Stomsoft		pp = &lp->d_partitions[*cp - 'a'];
202569800Stomsoft	} else {
202669926Stomsoft		errx(1, "unknown device");
202769800Stomsoft	}
202869800Stomsoft
202969800Stomsoft	/*
2030114067Sschweikh	 * Check if that partition is suitable for growing a file system.
203169800Stomsoft	 */
203269800Stomsoft	if (pp->p_size < 1) {
203369926Stomsoft		errx(1, "partition is unavailable");
203469800Stomsoft	}
203569800Stomsoft	if (pp->p_fstype != FS_BSDFFS) {
203669926Stomsoft		errx(1, "partition not 4.2BSD");
203769800Stomsoft	}
203869800Stomsoft
203969800Stomsoft	/*
204069800Stomsoft	 * Read the current superblock, and take a backup.
204169800Stomsoft	 */
204298542Smckusick	for (i = 0; sblock_try[i] != -1; i++) {
204398542Smckusick		sblockloc = sblock_try[i] / DEV_BSIZE;
204498542Smckusick		rdfs(sblockloc, (size_t)SBLOCKSIZE, (void *)&(osblock), fsi);
204598542Smckusick		if ((osblock.fs_magic == FS_UFS1_MAGIC ||
204698542Smckusick		     (osblock.fs_magic == FS_UFS2_MAGIC &&
2047107294Smckusick		      osblock.fs_sblockloc == sblock_try[i])) &&
204898542Smckusick		    osblock.fs_bsize <= MAXBSIZE &&
204998542Smckusick		    osblock.fs_bsize >= sizeof(struct fs))
205098542Smckusick			break;
205198542Smckusick	}
205298542Smckusick	if (sblock_try[i] == -1) {
205369926Stomsoft		errx(1, "superblock not recognized");
205469800Stomsoft	}
205569800Stomsoft	memcpy((void *)&fsun1, (void *)&fsun2, sizeof(fsun2));
205698542Smckusick	maxino = sblock.fs_ncg * sblock.fs_ipg;
205769800Stomsoft
205869800Stomsoft	DBG_OPEN("/tmp/growfs.debug"); /* already here we need a superblock */
205969926Stomsoft	DBG_DUMP_FS(&sblock,
206069926Stomsoft	    "old sblock");
206169800Stomsoft
206269800Stomsoft	/*
206369800Stomsoft	 * Determine size to grow to. Default to the full size specified in
206469800Stomsoft	 * the disk label.
206569800Stomsoft	 */
206669800Stomsoft	sblock.fs_size = dbtofsb(&osblock, pp->p_size);
206769800Stomsoft	if (size != 0) {
206869800Stomsoft		if (size > pp->p_size){
206977885Stomsoft			errx(1, "There is not enough space (%d < %d)",
207069800Stomsoft			    pp->p_size, size);
207169800Stomsoft		}
2072114067Sschweikh		sblock.fs_size = dbtofsb(&osblock, size);
207369800Stomsoft	}
207469800Stomsoft
207569800Stomsoft	/*
207669800Stomsoft	 * Are we really growing ?
207769800Stomsoft	 */
207869800Stomsoft	if(osblock.fs_size >= sblock.fs_size) {
207969926Stomsoft		errx(1, "we are not growing (%d->%d)", osblock.fs_size,
208069926Stomsoft		    sblock.fs_size);
208169800Stomsoft	}
208269800Stomsoft
208369800Stomsoft
208469800Stomsoft#ifdef FSMAXSNAP
208569800Stomsoft	/*
208669800Stomsoft	 * Check if we find an active snapshot.
208769800Stomsoft	 */
208869800Stomsoft	if(ExpertFlag == 0) {
208969800Stomsoft		for(j=0; j<FSMAXSNAP; j++) {
209069800Stomsoft			if(sblock.fs_snapinum[j]) {
2091102231Strhodes				errx(1, "active snapshot found in file system\n"
209269800Stomsoft				    "	please remove all snapshots before "
209369800Stomsoft				    "using growfs\n");
209469800Stomsoft			}
209569800Stomsoft			if(!sblock.fs_snapinum[j]) { /* list is dense */
209669800Stomsoft				break;
209769800Stomsoft			}
209869800Stomsoft		}
209969800Stomsoft	}
210069800Stomsoft#endif
210169800Stomsoft
210269800Stomsoft	if (ExpertFlag == 0 && Nflag == 0) {
210369800Stomsoft		printf("We strongly recommend you to make a backup "
210469800Stomsoft		    "before growing the Filesystem\n\n"
210569800Stomsoft		    " Did you backup your data (Yes/No) ? ");
210677885Stomsoft		fgets(reply, (int)sizeof(reply), stdin);
210769800Stomsoft		if (strcmp(reply, "Yes\n")){
210869800Stomsoft			printf("\n Nothing done \n");
210969800Stomsoft			exit (0);
2110114067Sschweikh		}
211169800Stomsoft	}
211269800Stomsoft
2113102231Strhodes	printf("new file systemsize is: %d frags\n", sblock.fs_size);
211469800Stomsoft
211569800Stomsoft	/*
2116102231Strhodes	 * Try to access our new last block in the file system. Even if we
211769800Stomsoft	 * later on realize we have to abort our operation, on that block
211869800Stomsoft	 * there should be no data, so we can't destroy something yet.
211969800Stomsoft	 */
212098542Smckusick	wtfs((ufs2_daddr_t)pp->p_size-1, (size_t)DEV_BSIZE, (void *)&sblock,
212198542Smckusick	    fso, Nflag);
212269800Stomsoft
212369800Stomsoft	/*
212469800Stomsoft	 * Now calculate new superblock values and check for reasonable
2125102231Strhodes	 * bound for new file system size:
212669800Stomsoft	 *     fs_size:    is derived from label or user input
212769800Stomsoft	 *     fs_dsize:   should get updated in the routines creating or
212869800Stomsoft	 *                 updating the cylinder groups on the fly
212969800Stomsoft	 *     fs_cstotal: should get updated in the routines creating or
213069800Stomsoft	 *                 updating the cylinder groups
213169800Stomsoft	 */
213269800Stomsoft
213369800Stomsoft	/*
2134102231Strhodes	 * Update the number of cylinders and cylinder groups in the file system.
213569800Stomsoft	 */
213698542Smckusick	if (sblock.fs_magic == FS_UFS1_MAGIC) {
213798542Smckusick		sblock.fs_old_ncyl =
213898542Smckusick		    sblock.fs_size * sblock.fs_old_nspf / sblock.fs_old_spc;
213998542Smckusick		if (sblock.fs_size * sblock.fs_old_nspf >
214098542Smckusick		    sblock.fs_old_ncyl * sblock.fs_old_spc)
214198542Smckusick			sblock.fs_old_ncyl++;
214269800Stomsoft	}
214398542Smckusick	sblock.fs_ncg = howmany(sblock.fs_size, sblock.fs_fpg);
214498542Smckusick	maxino = sblock.fs_ncg * sblock.fs_ipg;
214569800Stomsoft
214698542Smckusick	if (sblock.fs_size % sblock.fs_fpg != 0 &&
214798542Smckusick	    sblock.fs_size % sblock.fs_fpg < cgdmin(&sblock, sblock.fs_ncg)) {
214869800Stomsoft		/*
214969800Stomsoft		 * The space in the new last cylinder group is too small,
215069800Stomsoft		 * so revert back.
215169800Stomsoft		 */
215269800Stomsoft		sblock.fs_ncg--;
215398542Smckusick		if (sblock.fs_magic == FS_UFS1_MAGIC)
215498542Smckusick			sblock.fs_old_ncyl = sblock.fs_ncg * sblock.fs_old_cpg;
215598542Smckusick		printf("Warning: %d sector(s) cannot be allocated.\n",
215698542Smckusick		    fsbtodb(&sblock, sblock.fs_size % sblock.fs_fpg));
215798542Smckusick		sblock.fs_size = sblock.fs_ncg * sblock.fs_fpg;
215869800Stomsoft	}
215969800Stomsoft
216069800Stomsoft	/*
216169800Stomsoft	 * Update the space for the cylinder group summary information in the
216269800Stomsoft	 * respective cylinder group data area.
216369800Stomsoft	 */
216469800Stomsoft	sblock.fs_cssize =
216569800Stomsoft	    fragroundup(&sblock, sblock.fs_ncg * sizeof(struct csum));
2166114067Sschweikh
216769800Stomsoft	if(osblock.fs_size >= sblock.fs_size) {
216869926Stomsoft		errx(1, "not enough new space");
216969800Stomsoft	}
217069800Stomsoft
217169800Stomsoft	DBG_PRINT0("sblock calculated\n");
217269800Stomsoft
217369800Stomsoft	/*
217469800Stomsoft	 * Ok, everything prepared, so now let's do the tricks.
217569800Stomsoft	 */
217669800Stomsoft	growfs(fsi, fso, Nflag);
217769800Stomsoft
217869800Stomsoft	/*
217969800Stomsoft	 * Update the disk label.
218069800Stomsoft	 */
218169800Stomsoft	pp->p_fsize = sblock.fs_fsize;
218269800Stomsoft	pp->p_frag = sblock.fs_frag;
218398542Smckusick	pp->p_cpg = sblock.fs_fpg;
218469800Stomsoft
218569800Stomsoft	return_disklabel(fso, lp, Nflag);
218669800Stomsoft	DBG_PRINT0("label rewritten\n");
218769800Stomsoft
218869800Stomsoft	close(fsi);
218969800Stomsoft	if(fso>-1) close(fso);
219069800Stomsoft
219169800Stomsoft	DBG_CLOSE;
219269800Stomsoft
219369800Stomsoft	DBG_LEAVE;
219469800Stomsoft	return 0;
219569800Stomsoft}
219669800Stomsoft
219769800Stomsoft/* ************************************************** return_disklabel ***** */
219869800Stomsoft/*
219969800Stomsoft * Write the updated disklabel back to disk.
220069800Stomsoft */
220169800Stomsoftstatic void
220277885Stomsoftreturn_disklabel(int fd, struct disklabel *lp, unsigned int Nflag)
220369800Stomsoft{
220469800Stomsoft	DBG_FUNC("return_disklabel")
220569800Stomsoft	u_short	sum;
220669800Stomsoft	u_short	*ptr;
220769800Stomsoft
220869800Stomsoft	DBG_ENTER;
220969800Stomsoft
221069800Stomsoft	if(!lp) {
221169800Stomsoft		DBG_LEAVE;
221269800Stomsoft		return;
221369800Stomsoft	}
221469800Stomsoft	if(!Nflag) {
221569800Stomsoft		lp->d_checksum=0;
221669800Stomsoft		sum = 0;
221769800Stomsoft		ptr=(u_short *)lp;
221869800Stomsoft
221969800Stomsoft		/*
222069800Stomsoft		 * recalculate checksum
222169800Stomsoft		 */
222269800Stomsoft		while(ptr < (u_short *)&lp->d_partitions[lp->d_npartitions]) {
222369800Stomsoft			sum ^= *ptr++;
222469800Stomsoft		}
222569800Stomsoft		lp->d_checksum=sum;
222669800Stomsoft
222769800Stomsoft		if (ioctl(fd, DIOCWDINFO, (char *)lp) < 0) {
222869926Stomsoft			errx(1, "DIOCWDINFO failed");
222969800Stomsoft		}
223069800Stomsoft	}
223169800Stomsoft	free(lp);
223269800Stomsoft
223369800Stomsoft	DBG_LEAVE;
223469800Stomsoft	return ;
223569800Stomsoft}
223669800Stomsoft
223769800Stomsoft/* ***************************************************** get_disklabel ***** */
223869800Stomsoft/*
223969800Stomsoft * Read the disklabel from disk.
224069800Stomsoft */
224169800Stomsoftstatic struct disklabel *
224269800Stomsoftget_disklabel(int fd)
224369800Stomsoft{
224469800Stomsoft	DBG_FUNC("get_disklabel")
224569800Stomsoft	static struct	disklabel *lab;
224669800Stomsoft
224769800Stomsoft	DBG_ENTER;
224869800Stomsoft
224969800Stomsoft	lab=(struct disklabel *)malloc(sizeof(struct disklabel));
225069800Stomsoft	if (!lab) {
225169926Stomsoft		errx(1, "malloc failed");
225269800Stomsoft	}
225369800Stomsoft	if (ioctl(fd, DIOCGDINFO, (char *)lab) < 0) {
225469926Stomsoft		errx(1, "DIOCGDINFO failed");
225569800Stomsoft	}
225669800Stomsoft
225769800Stomsoft	DBG_LEAVE;
225869800Stomsoft	return (lab);
225969800Stomsoft}
226069800Stomsoft
226169800Stomsoft
226269800Stomsoft/* ************************************************************* usage ***** */
226369800Stomsoft/*
226469800Stomsoft * Dump a line of usage.
226569800Stomsoft */
226669800Stomsoftstatic void
226769926Stomsoftusage(void)
2268114067Sschweikh{
226969800Stomsoft	DBG_FUNC("usage")
227069800Stomsoft
227169800Stomsoft	DBG_ENTER;
227269800Stomsoft
227369926Stomsoft	fprintf(stderr, "usage: growfs [-Ny] [-s size] special\n");
227469926Stomsoft
227569800Stomsoft	DBG_LEAVE;
227669926Stomsoft	exit(1);
227769800Stomsoft}
227869800Stomsoft
227969800Stomsoft/* *********************************************************** updclst ***** */
228069800Stomsoft/*
2281114067Sschweikh * This updates most parameters and the bitmap related to cluster. We have to
2282114067Sschweikh * assume that sblock, osblock, acg are set up.
228369800Stomsoft */
228469800Stomsoftstatic void
228569800Stomsoftupdclst(int block)
2286114067Sschweikh{
228769800Stomsoft	DBG_FUNC("updclst")
228869800Stomsoft	static int	lcs=0;
228969800Stomsoft
229069800Stomsoft	DBG_ENTER;
229169800Stomsoft
229269800Stomsoft	if(sblock.fs_contigsumsize < 1) { /* no clustering */
229369800Stomsoft		return;
229469800Stomsoft	}
229569800Stomsoft	/*
229669800Stomsoft	 * update cluster allocation map
229769800Stomsoft	 */
229869800Stomsoft	setbit(cg_clustersfree(&acg), block);
229969800Stomsoft
230069800Stomsoft	/*
230169800Stomsoft	 * update cluster summary table
230269800Stomsoft	 */
230369800Stomsoft	if(!lcs) {
230469800Stomsoft		/*
230569800Stomsoft		 * calculate size for the trailing cluster
230669800Stomsoft		 */
230769800Stomsoft		for(block--; lcs<sblock.fs_contigsumsize; block--, lcs++ ) {
230869800Stomsoft			if(isclr(cg_clustersfree(&acg), block)){
230969800Stomsoft				break;
231069800Stomsoft			}
231169800Stomsoft		}
2312114067Sschweikh	}
231369800Stomsoft	if(lcs < sblock.fs_contigsumsize) {
231469800Stomsoft		if(lcs) {
231569800Stomsoft			cg_clustersum(&acg)[lcs]--;
231669800Stomsoft		}
231769800Stomsoft		lcs++;
231869800Stomsoft		cg_clustersum(&acg)[lcs]++;
231969800Stomsoft	}
232069800Stomsoft
232169800Stomsoft	DBG_LEAVE;
232269800Stomsoft	return;
232369800Stomsoft}
232469800Stomsoft
232569800Stomsoft/* *********************************************************** updrefs ***** */
232669800Stomsoft/*
232769800Stomsoft * This updates all references to relocated blocks for the given inode.  The
232869800Stomsoft * inode is given as number within the cylinder group, and the number of the
232969800Stomsoft * cylinder group.
233069800Stomsoft */
233169800Stomsoftstatic void
233277885Stomsoftupdrefs(int cg, ino_t in, struct gfs_bpp *bp, int fsi, int fso, unsigned int
233377885Stomsoft    Nflag)
2334114067Sschweikh{
233569800Stomsoft	DBG_FUNC("updrefs")
233698542Smckusick	ufs_lbn_t	len, lbn, numblks;
233798542Smckusick	ufs2_daddr_t	iptr, blksperindir;
233898542Smckusick	union dinode	*ino;
233998542Smckusick	int		i, mode, remaining_blocks, inodeupdated;
234069800Stomsoft
234169800Stomsoft	DBG_ENTER;
234269800Stomsoft
234369800Stomsoft	/*
234498542Smckusick	 * XXX We should skip unused inodes even from being read from disk
234569800Stomsoft	 *     here by using the bitmap.
234669800Stomsoft	 */
234798542Smckusick	ino = ginode(in, fsi, cg);
234898542Smckusick	mode = DIP(ino, di_mode) & IFMT;
234998542Smckusick	if (mode != IFDIR && mode != IFREG && mode != IFLNK) {
235069800Stomsoft		DBG_LEAVE;
235169800Stomsoft		return; /* only check DIR, FILE, LINK */
235269800Stomsoft	}
235398542Smckusick	if (mode == IFLNK && DIP(ino, di_size) < sblock.fs_maxsymlinklen) {
235469800Stomsoft		DBG_LEAVE;
235569800Stomsoft		return;	/* skip short symlinks */
235669800Stomsoft	}
235798542Smckusick	numblks = howmany(DIP(ino, di_size), sblock.fs_bsize);
235898542Smckusick	if (numblks == 0) {
235969800Stomsoft		DBG_LEAVE;
236069800Stomsoft		return;	/* skip empty file */
236169800Stomsoft	}
236298542Smckusick	if (DIP(ino, di_blocks) == 0) {
236369800Stomsoft		DBG_LEAVE;
236469800Stomsoft		return;	/* skip empty swiss cheesy file or old fastlink */
236569800Stomsoft	}
236669926Stomsoft	DBG_PRINT2("scg checking inode (%d in %d)\n",
236769926Stomsoft	    in,
236869926Stomsoft	    cg);
236969800Stomsoft
237069800Stomsoft	/*
237198542Smckusick	 * Check all the blocks.
237269800Stomsoft	 */
237398542Smckusick	inodeupdated = 0;
237498542Smckusick	len = numblks < NDADDR ? numblks : NDADDR;
237598542Smckusick	for (i = 0; i < len; i++) {
237698542Smckusick		iptr = DIP(ino, di_db[i]);
237798542Smckusick		if (iptr == 0)
237898542Smckusick			continue;
237998542Smckusick		if (cond_bl_upd(&iptr, bp, fsi, fso, Nflag)) {
238098542Smckusick			DIP(ino, di_db[i]) = iptr;
238198542Smckusick			inodeupdated++;
238269800Stomsoft		}
238369800Stomsoft	}
238469800Stomsoft	DBG_PRINT0("~~scg direct blocks checked\n");
238569800Stomsoft
238698542Smckusick	blksperindir = 1;
238798542Smckusick	len = numblks - NDADDR;
238898542Smckusick	lbn = NDADDR;
238998542Smckusick	for (i = 0; len > 0 && i < NIADDR; i++) {
239098542Smckusick		iptr = DIP(ino, di_ib[i]);
239198542Smckusick		if (iptr == 0)
239298542Smckusick			continue;
239398542Smckusick		if (cond_bl_upd(&iptr, bp, fsi, fso, Nflag)) {
239498542Smckusick			DIP(ino, di_ib[i]) = iptr;
239598542Smckusick			inodeupdated++;
239669800Stomsoft		}
239798542Smckusick		indirchk(blksperindir, lbn, iptr, numblks, bp, fsi, fso, Nflag);
239898542Smckusick		blksperindir *= NINDIR(&sblock);
239998542Smckusick		lbn += blksperindir;
240098542Smckusick		len -= blksperindir;
240198542Smckusick		DBG_PRINT1("scg indirect_%d blocks checked\n", i + 1);
240269800Stomsoft	}
240398542Smckusick	if (inodeupdated)
240498542Smckusick		wtfs(inoblk, sblock.fs_bsize, inobuf, fso, Nflag);
240569800Stomsoft
240698542Smckusick	DBG_LEAVE;
240798542Smckusick	return;
240898542Smckusick}
240969800Stomsoft
241098542Smckusick/*
241198542Smckusick * Recursively check all the indirect blocks.
241298542Smckusick */
241398542Smckusickstatic void
241498542Smckusickindirchk(ufs_lbn_t blksperindir, ufs_lbn_t lbn, ufs2_daddr_t blkno,
241598542Smckusick    ufs_lbn_t lastlbn, struct gfs_bpp *bp, int fsi, int fso, unsigned int Nflag)
241698542Smckusick{
241798542Smckusick	DBG_FUNC("indirchk")
241898542Smckusick	void *ibuf;
241998542Smckusick	off_t offset;
242098542Smckusick	int i, last;
242198542Smckusick	ufs2_daddr_t iptr;
242298542Smckusick
242398542Smckusick	DBG_ENTER;
242498542Smckusick
242598542Smckusick	/* read in the indirect block. */
242698542Smckusick	ibuf = malloc(sblock.fs_bsize);
242798542Smckusick	if (!ibuf)
242898542Smckusick		errx(1, "malloc failed");
242998542Smckusick	rdfs(fsbtodb(&sblock, blkno), (size_t)sblock.fs_bsize, ibuf, fsi);
243098542Smckusick	last = howmany(lastlbn - lbn, blksperindir) < NINDIR(&sblock) ?
243198542Smckusick	    howmany(lastlbn - lbn, blksperindir) : NINDIR(&sblock);
243298542Smckusick	for (i = 0; i < last; i++) {
243398542Smckusick		if (sblock.fs_magic == FS_UFS1_MAGIC)
243498542Smckusick			iptr = ((ufs1_daddr_t *)ibuf)[i];
243598542Smckusick		else
243698542Smckusick			iptr = ((ufs2_daddr_t *)ibuf)[i];
243798542Smckusick		if (iptr == 0)
243898542Smckusick			continue;
243998542Smckusick		if (cond_bl_upd(&iptr, bp, fsi, fso, Nflag)) {
244098542Smckusick			if (sblock.fs_magic == FS_UFS1_MAGIC)
244198542Smckusick				((ufs1_daddr_t *)ibuf)[i] = iptr;
244298542Smckusick			else
244398542Smckusick				((ufs2_daddr_t *)ibuf)[i] = iptr;
244469800Stomsoft		}
244598542Smckusick		if (blksperindir == 1)
244698542Smckusick			continue;
244798542Smckusick		indirchk(blksperindir / NINDIR(&sblock), lbn + blksperindir * i,
244898542Smckusick		    iptr, lastlbn, bp, fsi, fso, Nflag);
244969800Stomsoft	}
245098542Smckusick	free(ibuf);
245169800Stomsoft
245269800Stomsoft	DBG_LEAVE;
245369800Stomsoft	return;
245469800Stomsoft}
2455