Home | History | Annotate | Line # | Download | only in resize_ffs
resize_ffs.c revision 1.24
      1  1.24       wiz /*	$NetBSD: resize_ffs.c,v 1.24 2010/12/14 21:49:21 wiz Exp $	*/
      2   1.1       jtk /* From sources sent on February 17, 2003 */
      3   1.1       jtk /*-
      4   1.1       jtk  * As its sole author, I explicitly place this code in the public
      5   1.1       jtk  *  domain.  Anyone may use it for any purpose (though I would
      6   1.1       jtk  *  appreciate credit where it is due).
      7   1.1       jtk  *
      8   1.1       jtk  *					der Mouse
      9   1.1       jtk  *
     10   1.1       jtk  *			       mouse (at) rodents.montreal.qc.ca
     11   1.1       jtk  *		     7D C8 61 52 5D E7 2D 39  4E F1 31 3E E8 B3 27 4B
     12   1.1       jtk  */
     13   1.1       jtk /*
     14   1.3       wiz  * resize_ffs:
     15   1.1       jtk  *
     16  1.24       wiz  * Resize a file system.  Is capable of both growing and shrinking.
     17   1.1       jtk  *
     18  1.24       wiz  * Usage: resize_ffs [-s newsize] [-y] file_system
     19   1.1       jtk  *
     20  1.15       riz  * Example: resize_ffs -s 29574 /dev/rsd1e
     21   1.1       jtk  *
     22   1.1       jtk  * newsize is in DEV_BSIZE units (ie, disk sectors, usually 512 bytes
     23   1.1       jtk  *  each).
     24   1.1       jtk  *
     25   1.1       jtk  * Note: this currently requires gcc to build, since it is written
     26   1.1       jtk  *  depending on gcc-specific features, notably nested function
     27   1.1       jtk  *  definitions (which in at least a few cases depend on the lexical
     28   1.1       jtk  *  scoping gcc provides, so they can't be trivially moved outside).
     29   1.1       jtk  *
     30  1.24       wiz  * It will not do anything useful with file systems in other than
     31   1.1       jtk  *  host-native byte order.  This really should be fixed (it's largely
     32   1.1       jtk  *  a historical accident; the original version of this program is
     33   1.1       jtk  *  older than bi-endian support in FFS).
     34   1.1       jtk  *
     35   1.5      salo  * Many thanks go to John Kohl <jtk (at) NetBSD.org> for finding bugs: the
     36   1.1       jtk  *  one responsible for the "realloccgblk: can't find blk in cyl"
     37   1.1       jtk  *  problem and a more minor one which left fs_dsize wrong when
     38   1.1       jtk  *  shrinking.  (These actually indicate bugs in fsck too - it should
     39   1.1       jtk  *  have caught and fixed them.)
     40   1.1       jtk  *
     41   1.1       jtk  */
     42   1.1       jtk 
     43  1.11     perry #include <sys/cdefs.h>
     44  1.13      haad #include <sys/disk.h>
     45  1.13      haad #include <sys/disklabel.h>
     46  1.13      haad #include <sys/dkio.h>
     47  1.13      haad #include <sys/ioctl.h>
     48   1.1       jtk #include <sys/stat.h>
     49   1.1       jtk #include <sys/mman.h>
     50   1.1       jtk #include <sys/param.h>		/* MAXFRAG */
     51   1.1       jtk #include <ufs/ffs/fs.h>
     52   1.1       jtk #include <ufs/ufs/dir.h>
     53   1.1       jtk #include <ufs/ufs/dinode.h>
     54   1.1       jtk #include <ufs/ufs/ufs_bswap.h>	/* ufs_rw32 */
     55   1.1       jtk 
     56  1.15       riz #include <err.h>
     57  1.15       riz #include <errno.h>
     58  1.15       riz #include <fcntl.h>
     59  1.15       riz #include <stdio.h>
     60  1.15       riz #include <stdlib.h>
     61  1.15       riz #include <strings.h>
     62  1.15       riz #include <unistd.h>
     63  1.15       riz 
     64  1.24       wiz /* new size of file system, in sectors */
     65  1.13      haad static uint32_t newsize;
     66   1.1       jtk 
     67  1.23       riz /* fd open onto disk device or file */
     68   1.1       jtk static int fd;
     69   1.1       jtk 
     70   1.1       jtk /* must we break up big I/O operations - see checksmallio() */
     71   1.1       jtk static int smallio;
     72   1.1       jtk 
     73   1.1       jtk /* size of a cg, in bytes, rounded up to a frag boundary */
     74   1.1       jtk static int cgblksz;
     75   1.1       jtk 
     76   1.4  christos /* possible superblock localtions */
     77   1.4  christos static int search[] = SBLOCKSEARCH;
     78   1.4  christos /* location of the superblock */
     79   1.4  christos static off_t where;
     80   1.4  christos 
     81   1.1       jtk /* Superblocks. */
     82   1.1       jtk static struct fs *oldsb;	/* before we started */
     83   1.1       jtk static struct fs *newsb;	/* copy to work with */
     84   1.1       jtk /* Buffer to hold the above.  Make sure it's aligned correctly. */
     85  1.15       riz static char sbbuf[2 * SBLOCKSIZE]
     86  1.15       riz 	__attribute__((__aligned__(__alignof__(struct fs))));
     87   1.1       jtk 
     88   1.1       jtk /* a cg's worth of brand new squeaky-clean inodes */
     89   1.4  christos static struct ufs1_dinode *zinodes;
     90   1.1       jtk 
     91   1.1       jtk /* pointers to the in-core cgs, read off disk and possibly modified */
     92   1.1       jtk static struct cg **cgs;
     93   1.1       jtk 
     94   1.1       jtk /* pointer to csum array - the stuff pointed to on-disk by fs_csaddr */
     95   1.1       jtk static struct csum *csums;
     96   1.1       jtk 
     97   1.1       jtk /* per-cg flags, indexed by cg number */
     98   1.1       jtk static unsigned char *cgflags;
     99   1.1       jtk #define CGF_DIRTY   0x01	/* needs to be written to disk */
    100   1.1       jtk #define CGF_BLKMAPS 0x02	/* block bitmaps need rebuilding */
    101   1.1       jtk #define CGF_INOMAPS 0x04	/* inode bitmaps need rebuilding */
    102   1.1       jtk 
    103   1.1       jtk /* when shrinking, these two arrays record how we want blocks to move.	 */
    104   1.1       jtk /*  if blkmove[i] is j, the frag that started out as frag #i should end	 */
    105   1.1       jtk /*  up as frag #j.  inomove[i]=j means, similarly, that the inode that	 */
    106   1.1       jtk /*  started out as inode i should end up as inode j.			 */
    107   1.1       jtk static unsigned int *blkmove;
    108   1.1       jtk static unsigned int *inomove;
    109   1.1       jtk 
    110   1.1       jtk /* in-core copies of all inodes in the fs, indexed by inumber */
    111   1.4  christos static struct ufs1_dinode *inodes;
    112   1.1       jtk 
    113   1.1       jtk /* per-inode flags, indexed by inumber */
    114   1.1       jtk static unsigned char *iflags;
    115   1.1       jtk #define IF_DIRTY  0x01		/* needs to be written to disk */
    116   1.1       jtk #define IF_BDIRTY 0x02		/* like DIRTY, but is set on first inode in a
    117   1.1       jtk 				 * block of inodes, and applies to the whole
    118   1.1       jtk 				 * block. */
    119   1.1       jtk 
    120  1.15       riz /* resize_ffs works directly on dinodes, adapt blksize() */
    121   1.4  christos #define dblksize(fs, dip, lbn) \
    122   1.4  christos     (((lbn) >= NDADDR || (dip)->di_size >= lblktosize(fs, (lbn) + 1)) \
    123   1.4  christos     ? (fs)->fs_bsize \
    124   1.4  christos     : (fragroundup(fs, blkoff(fs, (dip)->di_size))))
    125   1.4  christos 
    126   1.4  christos 
    127   1.4  christos /*
    128   1.4  christos  * Number of disk sectors per block/fragment; assumes DEV_BSIZE byte
    129   1.4  christos  * sector size.
    130   1.4  christos  */
    131   1.4  christos #define NSPB(fs)	((fs)->fs_old_nspf << (fs)->fs_fragshift)
    132   1.4  christos #define NSPF(fs)	((fs)->fs_old_nspf)
    133   1.4  christos 
    134  1.23       riz /* global flags */
    135  1.23       riz int is_ufs2 = 0;
    136  1.23       riz int needswap = 0;
    137  1.23       riz 
    138  1.13      haad static void usage(void) __dead;
    139  1.13      haad 
    140   1.1       jtk /*
    141   1.1       jtk  * See if we need to break up large I/O operations.  This should never
    142   1.1       jtk  *  be needed, but under at least one <version,platform> combination,
    143   1.1       jtk  *  large enough disk transfers to the raw device hang.  So if we're
    144   1.1       jtk  *  talking to a character special device, play it safe; in this case,
    145   1.1       jtk  *  readat() and writeat() break everything up into pieces no larger
    146   1.1       jtk  *  than 8K, doing multiple syscalls for larger operations.
    147   1.1       jtk  */
    148   1.1       jtk static void
    149   1.1       jtk checksmallio(void)
    150   1.1       jtk {
    151   1.1       jtk 	struct stat stb;
    152   1.1       jtk 
    153   1.1       jtk 	fstat(fd, &stb);
    154   1.1       jtk 	smallio = ((stb.st_mode & S_IFMT) == S_IFCHR);
    155   1.1       jtk }
    156  1.19       riz 
    157  1.19       riz static int
    158  1.19       riz isplainfile(void)
    159  1.19       riz {
    160  1.19       riz 	struct stat stb;
    161  1.19       riz 
    162  1.19       riz 	fstat(fd, &stb);
    163  1.19       riz 	return S_ISREG(stb.st_mode);
    164  1.19       riz }
    165   1.1       jtk /*
    166   1.1       jtk  * Read size bytes starting at blkno into buf.  blkno is in DEV_BSIZE
    167   1.1       jtk  *  units, ie, after fsbtodb(); size is in bytes.
    168   1.1       jtk  */
    169   1.1       jtk static void
    170   1.1       jtk readat(off_t blkno, void *buf, int size)
    171   1.1       jtk {
    172   1.1       jtk 	/* Seek to the correct place. */
    173   1.4  christos 	if (lseek(fd, blkno * DEV_BSIZE, L_SET) < 0)
    174  1.15       riz 		err(EXIT_FAILURE, "lseek failed");
    175   1.4  christos 
    176   1.1       jtk 	/* See if we have to break up the transfer... */
    177   1.1       jtk 	if (smallio) {
    178   1.1       jtk 		char *bp;	/* pointer into buf */
    179   1.1       jtk 		int left;	/* bytes left to go */
    180   1.1       jtk 		int n;		/* number to do this time around */
    181   1.1       jtk 		int rv;		/* syscall return value */
    182   1.1       jtk 		bp = buf;
    183   1.1       jtk 		left = size;
    184   1.1       jtk 		while (left > 0) {
    185   1.1       jtk 			n = (left > 8192) ? 8192 : left;
    186   1.1       jtk 			rv = read(fd, bp, n);
    187   1.4  christos 			if (rv < 0)
    188  1.13      haad 				err(EXIT_FAILURE, "read failed");
    189   1.4  christos 			if (rv != n)
    190  1.15       riz 				errx(EXIT_FAILURE,
    191  1.15       riz 				    "read: wanted %d, got %d", n, rv);
    192   1.1       jtk 			bp += n;
    193   1.1       jtk 			left -= n;
    194   1.1       jtk 		}
    195   1.1       jtk 	} else {
    196   1.1       jtk 		int rv;
    197   1.1       jtk 		rv = read(fd, buf, size);
    198   1.4  christos 		if (rv < 0)
    199  1.13      haad 			err(EXIT_FAILURE, "read failed");
    200   1.4  christos 		if (rv != size)
    201  1.15       riz 			errx(EXIT_FAILURE, "read: wanted %d, got %d", size, rv);
    202   1.1       jtk 	}
    203   1.1       jtk }
    204   1.1       jtk /*
    205   1.1       jtk  * Write size bytes from buf starting at blkno.  blkno is in DEV_BSIZE
    206   1.1       jtk  *  units, ie, after fsbtodb(); size is in bytes.
    207   1.1       jtk  */
    208   1.1       jtk static void
    209   1.1       jtk writeat(off_t blkno, const void *buf, int size)
    210   1.1       jtk {
    211   1.1       jtk 	/* Seek to the correct place. */
    212   1.4  christos 	if (lseek(fd, blkno * DEV_BSIZE, L_SET) < 0)
    213  1.13      haad 		err(EXIT_FAILURE, "lseek failed");
    214   1.1       jtk 	/* See if we have to break up the transfer... */
    215   1.1       jtk 	if (smallio) {
    216   1.1       jtk 		const char *bp;	/* pointer into buf */
    217   1.1       jtk 		int left;	/* bytes left to go */
    218   1.1       jtk 		int n;		/* number to do this time around */
    219   1.1       jtk 		int rv;		/* syscall return value */
    220   1.1       jtk 		bp = buf;
    221   1.1       jtk 		left = size;
    222   1.1       jtk 		while (left > 0) {
    223   1.1       jtk 			n = (left > 8192) ? 8192 : left;
    224   1.1       jtk 			rv = write(fd, bp, n);
    225   1.4  christos 			if (rv < 0)
    226  1.13      haad 				err(EXIT_FAILURE, "write failed");
    227   1.4  christos 			if (rv != n)
    228  1.15       riz 				errx(EXIT_FAILURE,
    229  1.15       riz 				    "write: wanted %d, got %d", n, rv);
    230   1.1       jtk 			bp += n;
    231   1.1       jtk 			left -= n;
    232   1.1       jtk 		}
    233   1.1       jtk 	} else {
    234   1.1       jtk 		int rv;
    235   1.1       jtk 		rv = write(fd, buf, size);
    236   1.4  christos 		if (rv < 0)
    237  1.13      haad 			err(EXIT_FAILURE, "write failed");
    238   1.4  christos 		if (rv != size)
    239  1.15       riz 			errx(EXIT_FAILURE,
    240  1.15       riz 			    "write: wanted %d, got %d", size, rv);
    241   1.1       jtk 	}
    242   1.1       jtk }
    243   1.1       jtk /*
    244   1.1       jtk  * Never-fail versions of malloc() and realloc(), and an allocation
    245   1.1       jtk  *  routine (which also never fails) for allocating memory that will
    246   1.1       jtk  *  never be freed until exit.
    247   1.1       jtk  */
    248   1.1       jtk 
    249   1.1       jtk /*
    250   1.1       jtk  * Never-fail malloc.
    251   1.1       jtk  */
    252   1.1       jtk static void *
    253   1.1       jtk nfmalloc(size_t nb, const char *tag)
    254   1.1       jtk {
    255   1.1       jtk 	void *rv;
    256   1.1       jtk 
    257   1.1       jtk 	rv = malloc(nb);
    258   1.1       jtk 	if (rv)
    259   1.1       jtk 		return (rv);
    260  1.13      haad 	err(EXIT_FAILURE, "Can't allocate %lu bytes for %s",
    261   1.4  christos 	    (unsigned long int) nb, tag);
    262   1.1       jtk }
    263   1.1       jtk /*
    264   1.1       jtk  * Never-fail realloc.
    265   1.1       jtk  */
    266   1.1       jtk static void *
    267   1.1       jtk nfrealloc(void *blk, size_t nb, const char *tag)
    268   1.1       jtk {
    269   1.1       jtk 	void *rv;
    270   1.1       jtk 
    271   1.1       jtk 	rv = realloc(blk, nb);
    272   1.1       jtk 	if (rv)
    273   1.1       jtk 		return (rv);
    274  1.13      haad 	err(EXIT_FAILURE, "Can't re-allocate %lu bytes for %s",
    275   1.4  christos 	    (unsigned long int) nb, tag);
    276   1.1       jtk }
    277   1.1       jtk /*
    278   1.1       jtk  * Allocate memory that will never be freed or reallocated.  Arguably
    279   1.1       jtk  *  this routine should handle small allocations by chopping up pages,
    280   1.1       jtk  *  but that's not worth the bother; it's not called more than a
    281   1.1       jtk  *  handful of times per run, and if the allocations are that small the
    282   1.1       jtk  *  waste in giving each one its own page is ignorable.
    283   1.1       jtk  */
    284   1.1       jtk static void *
    285   1.1       jtk alloconce(size_t nb, const char *tag)
    286   1.1       jtk {
    287   1.1       jtk 	void *rv;
    288   1.1       jtk 
    289   1.1       jtk 	rv = mmap(0, nb, PROT_READ | PROT_WRITE, MAP_ANON | MAP_PRIVATE, -1, 0);
    290   1.1       jtk 	if (rv != MAP_FAILED)
    291   1.1       jtk 		return (rv);
    292  1.13      haad 	err(EXIT_FAILURE, "Can't map %lu bytes for %s",
    293   1.4  christos 	    (unsigned long int) nb, tag);
    294   1.1       jtk }
    295   1.1       jtk /*
    296   1.1       jtk  * Load the cgs and csums off disk.  Also allocates the space to load
    297   1.1       jtk  *  them into and initializes the per-cg flags.
    298   1.1       jtk  */
    299   1.1       jtk static void
    300   1.1       jtk loadcgs(void)
    301   1.1       jtk {
    302   1.1       jtk 	int cg;
    303   1.1       jtk 	char *cgp;
    304   1.1       jtk 
    305   1.1       jtk 	cgblksz = roundup(oldsb->fs_cgsize, oldsb->fs_fsize);
    306   1.1       jtk 	cgs = nfmalloc(oldsb->fs_ncg * sizeof(struct cg *), "cg pointers");
    307   1.1       jtk 	cgp = alloconce(oldsb->fs_ncg * cgblksz, "cgs");
    308   1.1       jtk 	cgflags = nfmalloc(oldsb->fs_ncg, "cg flags");
    309   1.1       jtk 	csums = nfmalloc(oldsb->fs_cssize, "cg summary");
    310   1.1       jtk 	for (cg = 0; cg < oldsb->fs_ncg; cg++) {
    311   1.1       jtk 		cgs[cg] = (struct cg *) cgp;
    312   1.1       jtk 		readat(fsbtodb(oldsb, cgtod(oldsb, cg)), cgp, cgblksz);
    313   1.1       jtk 		cgflags[cg] = 0;
    314   1.1       jtk 		cgp += cgblksz;
    315   1.1       jtk 	}
    316   1.1       jtk 	readat(fsbtodb(oldsb, oldsb->fs_csaddr), csums, oldsb->fs_cssize);
    317   1.1       jtk }
    318   1.1       jtk /*
    319   1.1       jtk  * Set n bits, starting with bit #base, in the bitmap pointed to by
    320   1.1       jtk  *  bitvec (which is assumed to be large enough to include bits base
    321   1.1       jtk  *  through base+n-1).
    322   1.1       jtk  */
    323   1.1       jtk static void
    324   1.1       jtk set_bits(unsigned char *bitvec, unsigned int base, unsigned int n)
    325   1.1       jtk {
    326   1.1       jtk 	if (n < 1)
    327   1.1       jtk 		return;		/* nothing to do */
    328   1.1       jtk 	if (base & 7) {		/* partial byte at beginning */
    329   1.1       jtk 		if (n <= 8 - (base & 7)) {	/* entirely within one byte */
    330   1.1       jtk 			bitvec[base >> 3] |= (~((~0U) << n)) << (base & 7);
    331   1.1       jtk 			return;
    332   1.1       jtk 		}
    333   1.1       jtk 		bitvec[base >> 3] |= (~0U) << (base & 7);
    334   1.1       jtk 		n -= 8 - (base & 7);
    335   1.1       jtk 		base = (base & ~7) + 8;
    336   1.1       jtk 	}
    337   1.1       jtk 	if (n >= 8) {		/* do full bytes */
    338   1.1       jtk 		memset(bitvec + (base >> 3), 0xff, n >> 3);
    339   1.1       jtk 		base += n & ~7;
    340   1.1       jtk 		n &= 7;
    341   1.1       jtk 	}
    342   1.1       jtk 	if (n) {		/* partial byte at end */
    343   1.1       jtk 		bitvec[base >> 3] |= ~((~0U) << n);
    344   1.1       jtk 	}
    345   1.1       jtk }
    346   1.1       jtk /*
    347   1.1       jtk  * Clear n bits, starting with bit #base, in the bitmap pointed to by
    348   1.1       jtk  *  bitvec (which is assumed to be large enough to include bits base
    349   1.1       jtk  *  through base+n-1).  Code parallels set_bits().
    350   1.1       jtk  */
    351   1.1       jtk static void
    352   1.1       jtk clr_bits(unsigned char *bitvec, int base, int n)
    353   1.1       jtk {
    354   1.1       jtk 	if (n < 1)
    355   1.1       jtk 		return;
    356   1.1       jtk 	if (base & 7) {
    357   1.1       jtk 		if (n <= 8 - (base & 7)) {
    358   1.1       jtk 			bitvec[base >> 3] &= ~((~((~0U) << n)) << (base & 7));
    359   1.1       jtk 			return;
    360   1.1       jtk 		}
    361   1.1       jtk 		bitvec[base >> 3] &= ~((~0U) << (base & 7));
    362   1.1       jtk 		n -= 8 - (base & 7);
    363   1.1       jtk 		base = (base & ~7) + 8;
    364   1.1       jtk 	}
    365   1.1       jtk 	if (n >= 8) {
    366   1.1       jtk 		bzero(bitvec + (base >> 3), n >> 3);
    367   1.1       jtk 		base += n & ~7;
    368   1.1       jtk 		n &= 7;
    369   1.1       jtk 	}
    370   1.1       jtk 	if (n) {
    371   1.1       jtk 		bitvec[base >> 3] &= (~0U) << n;
    372   1.1       jtk 	}
    373   1.1       jtk }
    374   1.1       jtk /*
    375   1.1       jtk  * Test whether bit #bit is set in the bitmap pointed to by bitvec.
    376   1.1       jtk  */
    377  1.13      haad static int
    378   1.1       jtk bit_is_set(unsigned char *bitvec, int bit)
    379   1.1       jtk {
    380   1.1       jtk 	return (bitvec[bit >> 3] & (1 << (bit & 7)));
    381   1.1       jtk }
    382   1.1       jtk /*
    383   1.1       jtk  * Test whether bit #bit is clear in the bitmap pointed to by bitvec.
    384   1.1       jtk  */
    385  1.13      haad static int
    386   1.1       jtk bit_is_clr(unsigned char *bitvec, int bit)
    387   1.1       jtk {
    388   1.1       jtk 	return (!bit_is_set(bitvec, bit));
    389   1.1       jtk }
    390   1.1       jtk /*
    391   1.1       jtk  * Test whether a whole block of bits is set in a bitmap.  This is
    392   1.1       jtk  *  designed for testing (aligned) disk blocks in a bit-per-frag
    393   1.1       jtk  *  bitmap; it has assumptions wired into it based on that, essentially
    394   1.1       jtk  *  that the entire block fits into a single byte.  This returns true
    395   1.1       jtk  *  iff _all_ the bits are set; it is not just the complement of
    396   1.1       jtk  *  blk_is_clr on the same arguments (unless blkfrags==1).
    397   1.1       jtk  */
    398  1.13      haad static int
    399   1.1       jtk blk_is_set(unsigned char *bitvec, int blkbase, int blkfrags)
    400   1.1       jtk {
    401   1.1       jtk 	unsigned int mask;
    402   1.1       jtk 
    403   1.1       jtk 	mask = (~((~0U) << blkfrags)) << (blkbase & 7);
    404   1.1       jtk 	return ((bitvec[blkbase >> 3] & mask) == mask);
    405   1.1       jtk }
    406   1.1       jtk /*
    407   1.1       jtk  * Test whether a whole block of bits is clear in a bitmap.  See
    408   1.1       jtk  *  blk_is_set (above) for assumptions.  This returns true iff _all_
    409   1.1       jtk  *  the bits are clear; it is not just the complement of blk_is_set on
    410   1.1       jtk  *  the same arguments (unless blkfrags==1).
    411   1.1       jtk  */
    412  1.13      haad static int
    413   1.1       jtk blk_is_clr(unsigned char *bitvec, int blkbase, int blkfrags)
    414   1.1       jtk {
    415   1.1       jtk 	unsigned int mask;
    416   1.1       jtk 
    417   1.1       jtk 	mask = (~((~0U) << blkfrags)) << (blkbase & 7);
    418   1.1       jtk 	return ((bitvec[blkbase >> 3] & mask) == 0);
    419   1.1       jtk }
    420   1.1       jtk /*
    421   1.1       jtk  * Initialize a new cg.  Called when growing.  Assumes memory has been
    422   1.1       jtk  *  allocated but not otherwise set up.  This code sets the fields of
    423   1.1       jtk  *  the cg, initializes the bitmaps (and cluster summaries, if
    424   1.1       jtk  *  applicable), updates both per-cylinder summary info and the global
    425   1.1       jtk  *  summary info in newsb; it also writes out new inodes for the cg.
    426   1.1       jtk  *
    427   1.1       jtk  * This code knows it can never be called for cg 0, which makes it a
    428   1.1       jtk  *  bit simpler than it would otherwise be.
    429   1.1       jtk  */
    430   1.1       jtk static void
    431   1.1       jtk initcg(int cgn)
    432   1.1       jtk {
    433   1.1       jtk 	struct cg *cg;		/* The in-core cg, of course */
    434   1.1       jtk 	int base;		/* Disk address of cg base */
    435   1.1       jtk 	int dlow;		/* Size of pre-cg data area */
    436   1.1       jtk 	int dhigh;		/* Offset of post-inode data area, from base */
    437   1.1       jtk 	int dmax;		/* Offset of end of post-inode data area */
    438   1.1       jtk 	int i;			/* Generic loop index */
    439   1.1       jtk 	int n;			/* Generic count */
    440   1.1       jtk 
    441   1.1       jtk 	cg = cgs[cgn];
    442   1.1       jtk 	/* Place the data areas */
    443   1.1       jtk 	base = cgbase(newsb, cgn);
    444   1.1       jtk 	dlow = cgsblock(newsb, cgn) - base;
    445   1.1       jtk 	dhigh = cgdmin(newsb, cgn) - base;
    446   1.1       jtk 	dmax = newsb->fs_size - base;
    447   1.1       jtk 	if (dmax > newsb->fs_fpg)
    448   1.1       jtk 		dmax = newsb->fs_fpg;
    449   1.1       jtk 	/*
    450   1.1       jtk          * Clear out the cg - assumes all-0-bytes is the correct way
    451   1.1       jtk          * to initialize fields we don't otherwise touch, which is
    452   1.1       jtk          * perhaps not the right thing to do, but it's what fsck and
    453   1.1       jtk          * mkfs do.
    454   1.1       jtk          */
    455   1.1       jtk 	bzero(cg, newsb->fs_cgsize);
    456  1.20    mhitch 	cg->cg_old_time = newsb->fs_time;
    457  1.20    mhitch 	if (newsb->fs_old_flags & FS_FLAGS_UPDATED)
    458  1.20    mhitch 		cg->cg_time = newsb->fs_time;
    459   1.1       jtk 	cg->cg_magic = CG_MAGIC;
    460   1.1       jtk 	cg->cg_cgx = cgn;
    461   1.4  christos 	cg->cg_old_ncyl = newsb->fs_old_cpg;
    462  1.14       riz 	/* Update the cg_old_ncyl value for the last cylinder. */
    463  1.20    mhitch 	if (cgn == newsb->fs_ncg - 1) {
    464  1.20    mhitch 		if ((newsb->fs_old_flags & FS_FLAGS_UPDATED) == 0)
    465  1.20    mhitch 			cg->cg_old_ncyl = newsb->fs_old_ncyl % newsb->fs_old_cpg;
    466   1.1       jtk 	}
    467  1.14       riz 	cg->cg_old_niblk = newsb->fs_ipg;
    468   1.1       jtk 	cg->cg_ndblk = dmax;
    469   1.1       jtk 	/* Set up the bitmap pointers.  We have to be careful to lay out the
    470   1.1       jtk 	 * cg _exactly_ the way mkfs and fsck do it, since fsck compares the
    471   1.1       jtk 	 * _entire_ cg against a recomputed cg, and whines if there is any
    472   1.1       jtk 	 * mismatch, including the bitmap offsets. */
    473   1.1       jtk 	/* XXX update this comment when fsck is fixed */
    474   1.4  christos 	cg->cg_old_btotoff = &cg->cg_space[0] - (unsigned char *) cg;
    475   1.4  christos 	cg->cg_old_boff = cg->cg_old_btotoff
    476   1.4  christos 	    + (newsb->fs_old_cpg * sizeof(int32_t));
    477   1.4  christos 	cg->cg_iusedoff = cg->cg_old_boff +
    478   1.4  christos 	    (newsb->fs_old_cpg * newsb->fs_old_nrpos * sizeof(int16_t));
    479   1.1       jtk 	cg->cg_freeoff = cg->cg_iusedoff + howmany(newsb->fs_ipg, NBBY);
    480   1.1       jtk 	if (newsb->fs_contigsumsize > 0) {
    481   1.1       jtk 		cg->cg_nclusterblks = cg->cg_ndblk / newsb->fs_frag;
    482   1.1       jtk 		cg->cg_clustersumoff = cg->cg_freeoff +
    483   1.4  christos 		    howmany(newsb->fs_old_cpg * newsb->fs_old_spc / NSPF(newsb),
    484   1.1       jtk 		    NBBY) - sizeof(int32_t);
    485   1.1       jtk 		cg->cg_clustersumoff =
    486   1.1       jtk 		    roundup(cg->cg_clustersumoff, sizeof(int32_t));
    487   1.1       jtk 		cg->cg_clusteroff = cg->cg_clustersumoff +
    488   1.1       jtk 		    ((newsb->fs_contigsumsize + 1) * sizeof(int32_t));
    489   1.1       jtk 		cg->cg_nextfreeoff = cg->cg_clusteroff +
    490   1.4  christos 		    howmany(newsb->fs_old_cpg * newsb->fs_old_spc / NSPB(newsb),
    491   1.1       jtk 		    NBBY);
    492   1.1       jtk 		n = dlow / newsb->fs_frag;
    493   1.1       jtk 		if (n > 0) {
    494   1.1       jtk 			set_bits(cg_clustersfree(cg, 0), 0, n);
    495   1.1       jtk 			cg_clustersum(cg, 0)[(n > newsb->fs_contigsumsize) ?
    496   1.1       jtk 			    newsb->fs_contigsumsize : n]++;
    497   1.1       jtk 		}
    498   1.1       jtk 	} else {
    499   1.1       jtk 		cg->cg_nextfreeoff = cg->cg_freeoff +
    500   1.4  christos 		    howmany(newsb->fs_old_cpg * newsb->fs_old_spc / NSPF(newsb),
    501   1.1       jtk 		    NBBY);
    502   1.1       jtk 	}
    503   1.1       jtk 	/* Mark the data areas as free; everything else is marked busy by the
    504   1.1       jtk 	 * bzero up at the top. */
    505   1.1       jtk 	set_bits(cg_blksfree(cg, 0), 0, dlow);
    506   1.1       jtk 	set_bits(cg_blksfree(cg, 0), dhigh, dmax - dhigh);
    507   1.1       jtk 	/* Initialize summary info */
    508   1.1       jtk 	cg->cg_cs.cs_ndir = 0;
    509   1.1       jtk 	cg->cg_cs.cs_nifree = newsb->fs_ipg;
    510   1.1       jtk 	cg->cg_cs.cs_nbfree = dlow / newsb->fs_frag;
    511   1.1       jtk 	cg->cg_cs.cs_nffree = 0;
    512   1.1       jtk 
    513   1.1       jtk 	/* This is the simplest way of doing this; we perhaps could compute
    514   1.1       jtk 	 * the correct cg_blktot()[] and cg_blks()[] values other ways, but it
    515   1.1       jtk 	 * would be complicated and hardly seems worth the effort.  (The
    516   1.1       jtk 	 * reason there isn't frag-at-beginning and frag-at-end code here,
    517   1.1       jtk 	 * like the code below for the post-inode data area, is that the
    518   1.1       jtk 	 * pre-sb data area always starts at 0, and thus is block-aligned, and
    519   1.1       jtk 	 * always ends at the sb, which is block-aligned.) */
    520   1.1       jtk 	for (i = 0; i < dlow; i += newsb->fs_frag) {
    521  1.15       riz 		old_cg_blktot(cg, 0)[old_cbtocylno(newsb, i)]++;
    522  1.15       riz 		old_cg_blks(newsb, cg,
    523  1.15       riz 		    old_cbtocylno(newsb, i), 0)[old_cbtorpos(newsb, i)]++;
    524   1.1       jtk 	}
    525   1.1       jtk 	/* Deal with a partial block at the beginning of the post-inode area.
    526   1.1       jtk 	 * I'm not convinced this can happen - I think the inodes are always
    527   1.1       jtk 	 * block-aligned and always an integral number of blocks - but it's
    528   1.1       jtk 	 * cheap to do the right thing just in case. */
    529   1.1       jtk 	if (dhigh % newsb->fs_frag) {
    530   1.1       jtk 		n = newsb->fs_frag - (dhigh % newsb->fs_frag);
    531   1.1       jtk 		cg->cg_frsum[n]++;
    532   1.1       jtk 		cg->cg_cs.cs_nffree += n;
    533   1.1       jtk 		dhigh += n;
    534   1.1       jtk 	}
    535   1.1       jtk 	n = (dmax - dhigh) / newsb->fs_frag;
    536   1.1       jtk 	/* We have n full-size blocks in the post-inode data area. */
    537   1.1       jtk 	if (n > 0) {
    538   1.1       jtk 		cg->cg_cs.cs_nbfree += n;
    539   1.1       jtk 		if (newsb->fs_contigsumsize > 0) {
    540   1.1       jtk 			i = dhigh / newsb->fs_frag;
    541   1.1       jtk 			set_bits(cg_clustersfree(cg, 0), i, n);
    542   1.1       jtk 			cg_clustersum(cg, 0)[(n > newsb->fs_contigsumsize) ?
    543   1.1       jtk 			    newsb->fs_contigsumsize : n]++;
    544   1.1       jtk 		}
    545   1.1       jtk 		for (i = n; i > 0; i--) {
    546  1.15       riz 			old_cg_blktot(cg, 0)[old_cbtocylno(newsb, dhigh)]++;
    547  1.15       riz 			old_cg_blks(newsb, cg,
    548  1.15       riz 			    old_cbtocylno(newsb, dhigh), 0)[old_cbtorpos(newsb,
    549   1.1       jtk 				dhigh)]++;
    550   1.1       jtk 			dhigh += newsb->fs_frag;
    551   1.1       jtk 		}
    552   1.1       jtk 	}
    553   1.1       jtk 	/* Deal with any leftover frag at the end of the cg. */
    554   1.1       jtk 	i = dmax - dhigh;
    555   1.1       jtk 	if (i) {
    556   1.1       jtk 		cg->cg_frsum[i]++;
    557   1.1       jtk 		cg->cg_cs.cs_nffree += i;
    558   1.1       jtk 	}
    559   1.1       jtk 	/* Update the csum info. */
    560   1.1       jtk 	csums[cgn] = cg->cg_cs;
    561   1.1       jtk 	newsb->fs_cstotal.cs_nffree += cg->cg_cs.cs_nffree;
    562   1.1       jtk 	newsb->fs_cstotal.cs_nbfree += cg->cg_cs.cs_nbfree;
    563   1.1       jtk 	newsb->fs_cstotal.cs_nifree += cg->cg_cs.cs_nifree;
    564   1.1       jtk 	/* Write out the cleared inodes. */
    565   1.1       jtk 	writeat(fsbtodb(newsb, cgimin(newsb, cgn)), zinodes,
    566   1.4  christos 	    newsb->fs_ipg * sizeof(struct ufs1_dinode));
    567   1.1       jtk 	/* Dirty the cg. */
    568   1.1       jtk 	cgflags[cgn] |= CGF_DIRTY;
    569   1.1       jtk }
    570   1.1       jtk /*
    571   1.1       jtk  * Find free space, at least nfrags consecutive frags of it.  Pays no
    572   1.1       jtk  *  attention to block boundaries, but refuses to straddle cg
    573   1.1       jtk  *  boundaries, even if the disk blocks involved are in fact
    574   1.1       jtk  *  consecutive.  Return value is the frag number of the first frag of
    575   1.1       jtk  *  the block, or -1 if no space was found.  Uses newsb for sb values,
    576   1.1       jtk  *  and assumes the cgs[] structures correctly describe the area to be
    577   1.1       jtk  *  searched.
    578   1.1       jtk  *
    579   1.1       jtk  * XXX is there a bug lurking in the ignoring of block boundaries by
    580   1.1       jtk  *  the routine used by fragmove() in evict_data()?  Can an end-of-file
    581   1.1       jtk  *  frag legally straddle a block boundary?  If not, this should be
    582   1.1       jtk  *  cloned and fixed to stop at block boundaries for that use.  The
    583   1.1       jtk  *  current one may still be needed for csum info motion, in case that
    584   1.1       jtk  *  takes up more than a whole block (is the csum info allowed to begin
    585   1.1       jtk  *  partway through a block and continue into the following block?).
    586   1.1       jtk  *
    587  1.24       wiz  * If we wrap off the end of the file system back to the beginning, we
    588  1.24       wiz  *  can end up searching the end of the file system twice.  I ignore
    589   1.1       jtk  *  this inefficiency, since if that happens we're going to croak with
    590   1.1       jtk  *  a no-space error anyway, so it happens at most once.
    591   1.1       jtk  */
    592   1.1       jtk static int
    593   1.1       jtk find_freespace(unsigned int nfrags)
    594   1.1       jtk {
    595   1.1       jtk 	static int hand = 0;	/* hand rotates through all frags in the fs */
    596   1.1       jtk 	int cgsize;		/* size of the cg hand currently points into */
    597   1.1       jtk 	int cgn;		/* number of cg hand currently points into */
    598   1.1       jtk 	int fwc;		/* frag-within-cg number of frag hand points
    599   1.1       jtk 				 * to */
    600   1.1       jtk 	int run;		/* length of run of free frags seen so far */
    601   1.1       jtk 	int secondpass;		/* have we wrapped from end of fs to
    602   1.1       jtk 				 * beginning? */
    603   1.1       jtk 	unsigned char *bits;	/* cg_blksfree()[] for cg hand points into */
    604   1.1       jtk 
    605   1.1       jtk 	cgn = dtog(newsb, hand);
    606   1.1       jtk 	fwc = dtogd(newsb, hand);
    607   1.1       jtk 	secondpass = (hand == 0);
    608   1.1       jtk 	run = 0;
    609   1.1       jtk 	bits = cg_blksfree(cgs[cgn], 0);
    610   1.1       jtk 	cgsize = cgs[cgn]->cg_ndblk;
    611   1.1       jtk 	while (1) {
    612   1.1       jtk 		if (bit_is_set(bits, fwc)) {
    613   1.1       jtk 			run++;
    614   1.1       jtk 			if (run >= nfrags)
    615   1.1       jtk 				return (hand + 1 - run);
    616   1.1       jtk 		} else {
    617   1.1       jtk 			run = 0;
    618   1.1       jtk 		}
    619   1.1       jtk 		hand++;
    620   1.1       jtk 		fwc++;
    621   1.1       jtk 		if (fwc >= cgsize) {
    622   1.1       jtk 			fwc = 0;
    623   1.1       jtk 			cgn++;
    624   1.1       jtk 			if (cgn >= newsb->fs_ncg) {
    625   1.1       jtk 				hand = 0;
    626   1.1       jtk 				if (secondpass)
    627   1.1       jtk 					return (-1);
    628   1.1       jtk 				secondpass = 1;
    629   1.1       jtk 				cgn = 0;
    630   1.1       jtk 			}
    631   1.1       jtk 			bits = cg_blksfree(cgs[cgn], 0);
    632   1.1       jtk 			cgsize = cgs[cgn]->cg_ndblk;
    633   1.1       jtk 			run = 0;
    634   1.1       jtk 		}
    635   1.1       jtk 	}
    636   1.1       jtk }
    637   1.1       jtk /*
    638   1.1       jtk  * Find a free block of disk space.  Finds an entire block of frags,
    639   1.1       jtk  *  all of which are free.  Return value is the frag number of the
    640   1.1       jtk  *  first frag of the block, or -1 if no space was found.  Uses newsb
    641   1.1       jtk  *  for sb values, and assumes the cgs[] structures correctly describe
    642   1.1       jtk  *  the area to be searched.
    643   1.1       jtk  *
    644   1.1       jtk  * See find_freespace(), above, for remarks about hand wrapping around.
    645   1.1       jtk  */
    646   1.1       jtk static int
    647   1.1       jtk find_freeblock(void)
    648   1.1       jtk {
    649   1.1       jtk 	static int hand = 0;	/* hand rotates through all frags in fs */
    650   1.1       jtk 	int cgn;		/* cg number of cg hand points into */
    651   1.1       jtk 	int fwc;		/* frag-within-cg number of frag hand points
    652   1.1       jtk 				 * to */
    653   1.1       jtk 	int cgsize;		/* size of cg hand points into */
    654   1.1       jtk 	int secondpass;		/* have we wrapped from end to beginning? */
    655   1.1       jtk 	unsigned char *bits;	/* cg_blksfree()[] for cg hand points into */
    656   1.1       jtk 
    657   1.1       jtk 	cgn = dtog(newsb, hand);
    658   1.1       jtk 	fwc = dtogd(newsb, hand);
    659   1.1       jtk 	secondpass = (hand == 0);
    660   1.1       jtk 	bits = cg_blksfree(cgs[cgn], 0);
    661   1.1       jtk 	cgsize = blknum(newsb, cgs[cgn]->cg_ndblk);
    662   1.1       jtk 	while (1) {
    663   1.1       jtk 		if (blk_is_set(bits, fwc, newsb->fs_frag))
    664   1.1       jtk 			return (hand);
    665   1.1       jtk 		fwc += newsb->fs_frag;
    666   1.1       jtk 		hand += newsb->fs_frag;
    667   1.1       jtk 		if (fwc >= cgsize) {
    668   1.1       jtk 			fwc = 0;
    669   1.1       jtk 			cgn++;
    670   1.1       jtk 			if (cgn >= newsb->fs_ncg) {
    671   1.1       jtk 				hand = 0;
    672   1.1       jtk 				if (secondpass)
    673   1.1       jtk 					return (-1);
    674   1.1       jtk 				secondpass = 1;
    675   1.1       jtk 				cgn = 0;
    676   1.1       jtk 			}
    677   1.1       jtk 			bits = cg_blksfree(cgs[cgn], 0);
    678   1.1       jtk 			cgsize = blknum(newsb, cgs[cgn]->cg_ndblk);
    679   1.1       jtk 		}
    680   1.1       jtk 	}
    681   1.1       jtk }
    682   1.1       jtk /*
    683   1.1       jtk  * Find a free inode, returning its inumber or -1 if none was found.
    684   1.1       jtk  *  Uses newsb for sb values, and assumes the cgs[] structures
    685   1.1       jtk  *  correctly describe the area to be searched.
    686   1.1       jtk  *
    687   1.1       jtk  * See find_freespace(), above, for remarks about hand wrapping around.
    688   1.1       jtk  */
    689   1.1       jtk static int
    690   1.1       jtk find_freeinode(void)
    691   1.1       jtk {
    692   1.1       jtk 	static int hand = 0;	/* hand rotates through all inodes in fs */
    693   1.1       jtk 	int cgn;		/* cg number of cg hand points into */
    694   1.1       jtk 	int iwc;		/* inode-within-cg number of inode hand points
    695   1.1       jtk 				 * to */
    696   1.1       jtk 	int secondpass;		/* have we wrapped from end to beginning? */
    697   1.1       jtk 	unsigned char *bits;	/* cg_inosused()[] for cg hand points into */
    698   1.1       jtk 
    699   1.1       jtk 	cgn = hand / newsb->fs_ipg;
    700   1.1       jtk 	iwc = hand % newsb->fs_ipg;
    701   1.1       jtk 	secondpass = (hand == 0);
    702   1.1       jtk 	bits = cg_inosused(cgs[cgn], 0);
    703   1.1       jtk 	while (1) {
    704   1.1       jtk 		if (bit_is_clr(bits, iwc))
    705   1.1       jtk 			return (hand);
    706   1.1       jtk 		hand++;
    707   1.1       jtk 		iwc++;
    708   1.1       jtk 		if (iwc >= newsb->fs_ipg) {
    709   1.1       jtk 			iwc = 0;
    710   1.1       jtk 			cgn++;
    711   1.1       jtk 			if (cgn >= newsb->fs_ncg) {
    712   1.1       jtk 				hand = 0;
    713   1.1       jtk 				if (secondpass)
    714   1.1       jtk 					return (-1);
    715   1.1       jtk 				secondpass = 1;
    716   1.1       jtk 				cgn = 0;
    717   1.1       jtk 			}
    718   1.1       jtk 			bits = cg_inosused(cgs[cgn], 0);
    719   1.1       jtk 		}
    720   1.1       jtk 	}
    721   1.1       jtk }
    722   1.1       jtk /*
    723   1.1       jtk  * Mark a frag as free.  Sets the frag's bit in the cg_blksfree bitmap
    724   1.1       jtk  *  for the appropriate cg, and marks the cg as dirty.
    725   1.1       jtk  */
    726   1.1       jtk static void
    727   1.1       jtk free_frag(int fno)
    728   1.1       jtk {
    729   1.1       jtk 	int cgn;
    730   1.1       jtk 
    731   1.1       jtk 	cgn = dtog(newsb, fno);
    732   1.1       jtk 	set_bits(cg_blksfree(cgs[cgn], 0), dtogd(newsb, fno), 1);
    733   1.1       jtk 	cgflags[cgn] |= CGF_DIRTY | CGF_BLKMAPS;
    734   1.1       jtk }
    735   1.1       jtk /*
    736   1.1       jtk  * Allocate a frag.  Clears the frag's bit in the cg_blksfree bitmap
    737   1.1       jtk  *  for the appropriate cg, and marks the cg as dirty.
    738   1.1       jtk  */
    739   1.1       jtk static void
    740   1.1       jtk alloc_frag(int fno)
    741   1.1       jtk {
    742   1.1       jtk 	int cgn;
    743   1.1       jtk 
    744   1.1       jtk 	cgn = dtog(newsb, fno);
    745   1.1       jtk 	clr_bits(cg_blksfree(cgs[cgn], 0), dtogd(newsb, fno), 1);
    746   1.1       jtk 	cgflags[cgn] |= CGF_DIRTY | CGF_BLKMAPS;
    747   1.1       jtk }
    748   1.1       jtk /*
    749   1.1       jtk  * Fix up the csum array.  If shrinking, this involves freeing zero or
    750   1.1       jtk  *  more frags; if growing, it involves allocating them, or if the
    751   1.1       jtk  *  frags being grown into aren't free, finding space elsewhere for the
    752   1.1       jtk  *  csum info.  (If the number of occupied frags doesn't change,
    753   1.1       jtk  *  nothing happens here.)
    754   1.1       jtk  */
    755   1.1       jtk static void
    756   1.1       jtk csum_fixup(void)
    757   1.1       jtk {
    758   1.1       jtk 	int nold;		/* # frags in old csum info */
    759   1.1       jtk 	int ntot;		/* # frags in new csum info */
    760   1.1       jtk 	int nnew;		/* ntot-nold */
    761   1.1       jtk 	int newloc;		/* new location for csum info, if necessary */
    762   1.1       jtk 	int i;			/* generic loop index */
    763   1.1       jtk 	int j;			/* generic loop index */
    764   1.1       jtk 	int f;			/* "from" frag number, if moving */
    765   1.1       jtk 	int t;			/* "to" frag number, if moving */
    766   1.1       jtk 	int cgn;		/* cg number, used when shrinking */
    767   1.1       jtk 
    768   1.1       jtk 	ntot = howmany(newsb->fs_cssize, newsb->fs_fsize);
    769   1.1       jtk 	nold = howmany(oldsb->fs_cssize, newsb->fs_fsize);
    770   1.1       jtk 	nnew = ntot - nold;
    771   1.1       jtk 	/* First, if there's no change in frag counts, it's easy. */
    772   1.1       jtk 	if (nnew == 0)
    773   1.1       jtk 		return;
    774   1.1       jtk 	/* Next, if we're shrinking, it's almost as easy.  Just free up any
    775   1.1       jtk 	 * frags in the old area we no longer need. */
    776   1.1       jtk 	if (nnew < 0) {
    777   1.1       jtk 		for ((i = newsb->fs_csaddr + ntot - 1), (j = nnew);
    778   1.1       jtk 		    j < 0;
    779   1.1       jtk 		    i--, j++) {
    780   1.1       jtk 			free_frag(i);
    781   1.1       jtk 		}
    782   1.1       jtk 		return;
    783   1.1       jtk 	}
    784   1.1       jtk 	/* We must be growing.  Check to see that the new csum area fits
    785  1.24       wiz 	 * within the file system.  I think this can never happen, since for
    786   1.1       jtk 	 * the csum area to grow, we must be adding at least one cg, so the
    787  1.24       wiz 	 * old csum area can't be this close to the end of the new file system.
    788   1.1       jtk 	 * But it's a cheap check. */
    789   1.1       jtk 	/* XXX what if csum info is at end of cg and grows into next cg, what
    790   1.1       jtk 	 * if it spills over onto the next cg's backup superblock?  Can this
    791   1.1       jtk 	 * happen? */
    792   1.1       jtk 	if (newsb->fs_csaddr + ntot <= newsb->fs_size) {
    793   1.1       jtk 		/* Okay, it fits - now,  see if the space we want is free. */
    794   1.1       jtk 		for ((i = newsb->fs_csaddr + nold), (j = nnew);
    795   1.1       jtk 		    j > 0;
    796   1.1       jtk 		    i++, j--) {
    797   1.1       jtk 			cgn = dtog(newsb, i);
    798   1.1       jtk 			if (bit_is_clr(cg_blksfree(cgs[cgn], 0),
    799   1.1       jtk 				dtogd(newsb, i)))
    800   1.1       jtk 				break;
    801   1.1       jtk 		}
    802   1.1       jtk 		if (j <= 0) {
    803   1.1       jtk 			/* Win win - all the frags we want are free. Allocate
    804   1.1       jtk 			 * 'em and we're all done.  */
    805   1.1       jtk 			for ((i = newsb->fs_csaddr + ntot - nnew), (j = nnew); j > 0; i++, j--) {
    806   1.1       jtk 				alloc_frag(i);
    807   1.1       jtk 			}
    808   1.1       jtk 			return;
    809   1.1       jtk 		}
    810   1.1       jtk 	}
    811   1.1       jtk 	/* We have to move the csum info, sigh.  Look for new space, free old
    812   1.1       jtk 	 * space, and allocate new.  Update fs_csaddr.  We don't copy anything
    813   1.1       jtk 	 * on disk at this point; the csum info will be written to the
    814   1.1       jtk 	 * then-current fs_csaddr as part of the final flush. */
    815   1.1       jtk 	newloc = find_freespace(ntot);
    816   1.1       jtk 	if (newloc < 0) {
    817   1.1       jtk 		printf("Sorry, no space available for new csums\n");
    818  1.15       riz 		exit(EXIT_FAILURE);
    819   1.1       jtk 	}
    820   1.1       jtk 	for (i = 0, f = newsb->fs_csaddr, t = newloc; i < ntot; i++, f++, t++) {
    821   1.1       jtk 		if (i < nold) {
    822   1.1       jtk 			free_frag(f);
    823   1.1       jtk 		}
    824   1.1       jtk 		alloc_frag(t);
    825   1.1       jtk 	}
    826   1.1       jtk 	newsb->fs_csaddr = newloc;
    827   1.1       jtk }
    828   1.1       jtk /*
    829   1.1       jtk  * Recompute newsb->fs_dsize.  Just scans all cgs, adding the number of
    830   1.1       jtk  *  data blocks in that cg to the total.
    831   1.1       jtk  */
    832   1.1       jtk static void
    833   1.1       jtk recompute_fs_dsize(void)
    834   1.1       jtk {
    835   1.1       jtk 	int i;
    836   1.1       jtk 
    837   1.1       jtk 	newsb->fs_dsize = 0;
    838   1.1       jtk 	for (i = 0; i < newsb->fs_ncg; i++) {
    839   1.1       jtk 		int dlow;	/* size of before-sb data area */
    840   1.1       jtk 		int dhigh;	/* offset of post-inode data area */
    841   1.1       jtk 		int dmax;	/* total size of cg */
    842   1.1       jtk 		int base;	/* base of cg, since cgsblock() etc add it in */
    843   1.1       jtk 		base = cgbase(newsb, i);
    844   1.1       jtk 		dlow = cgsblock(newsb, i) - base;
    845   1.1       jtk 		dhigh = cgdmin(newsb, i) - base;
    846   1.1       jtk 		dmax = newsb->fs_size - base;
    847   1.1       jtk 		if (dmax > newsb->fs_fpg)
    848   1.1       jtk 			dmax = newsb->fs_fpg;
    849   1.1       jtk 		newsb->fs_dsize += dlow + dmax - dhigh;
    850   1.1       jtk 	}
    851   1.1       jtk 	/* Space in cg 0 before cgsblock is boot area, not free space! */
    852   1.1       jtk 	newsb->fs_dsize -= cgsblock(newsb, 0) - cgbase(newsb, 0);
    853   1.1       jtk 	/* And of course the csum info takes up space. */
    854   1.1       jtk 	newsb->fs_dsize -= howmany(newsb->fs_cssize, newsb->fs_fsize);
    855   1.1       jtk }
    856   1.1       jtk /*
    857   1.1       jtk  * Return the current time.  We call this and assign, rather than
    858   1.1       jtk  *  calling time() directly, as insulation against OSes where fs_time
    859   1.1       jtk  *  is not a time_t.
    860   1.1       jtk  */
    861   1.1       jtk static time_t
    862   1.1       jtk timestamp(void)
    863   1.1       jtk {
    864   1.1       jtk 	time_t t;
    865   1.1       jtk 
    866   1.1       jtk 	time(&t);
    867   1.1       jtk 	return (t);
    868   1.1       jtk }
    869   1.1       jtk /*
    870  1.24       wiz  * Grow the file system.
    871   1.1       jtk  */
    872   1.1       jtk static void
    873   1.1       jtk grow(void)
    874   1.1       jtk {
    875   1.1       jtk 	int i;
    876   1.1       jtk 
    877   1.1       jtk 	/* Update the timestamp. */
    878   1.1       jtk 	newsb->fs_time = timestamp();
    879   1.1       jtk 	/* Allocate and clear the new-inode area, in case we add any cgs. */
    880   1.4  christos 	zinodes = alloconce(newsb->fs_ipg * sizeof(struct ufs1_dinode),
    881   1.1       jtk                             "zeroed inodes");
    882   1.4  christos 	bzero(zinodes, newsb->fs_ipg * sizeof(struct ufs1_dinode));
    883   1.1       jtk 	/* Update the size. */
    884   1.1       jtk 	newsb->fs_size = dbtofsb(newsb, newsize);
    885   1.1       jtk 	/* Did we actually not grow?  (This can happen if newsize is less than
    886   1.1       jtk 	 * a frag larger than the old size - unlikely, but no excuse to
    887   1.1       jtk 	 * misbehave if it happens.) */
    888  1.13      haad 	if (newsb->fs_size == oldsb->fs_size) {
    889  1.15       riz 		printf("New fs size %"PRIu64" = odl fs size %"PRIu64
    890  1.15       riz 		    ", not growing.\n", newsb->fs_size, oldsb->fs_size);
    891   1.1       jtk 		return;
    892  1.13      haad 	}
    893   1.1       jtk 	/* Check that the new last sector (frag, actually) is writable.  Since
    894   1.1       jtk 	 * it's at least one frag larger than it used to be, we know we aren't
    895   1.1       jtk 	 * overwriting anything important by this.  (The choice of sbbuf as
    896   1.1       jtk 	 * what to write is irrelevant; it's just something handy that's known
    897   1.1       jtk 	 * to be at least one frag in size.) */
    898  1.18       riz 	writeat(fsbtodb(newsb,newsb->fs_size - 1), &sbbuf, newsb->fs_fsize);
    899   1.4  christos 	/* Update fs_old_ncyl and fs_ncg. */
    900  1.17       riz 	newsb->fs_old_ncyl = howmany(newsb->fs_size * NSPF(newsb),
    901  1.17       riz 	    newsb->fs_old_spc);
    902   1.4  christos 	newsb->fs_ncg = howmany(newsb->fs_old_ncyl, newsb->fs_old_cpg);
    903   1.1       jtk 	/* Does the last cg end before the end of its inode area? There is no
    904   1.1       jtk 	 * reason why this couldn't be handled, but it would complicate a lot
    905  1.24       wiz 	 * of code (in all file system code - fsck, kernel, etc) because of the
    906   1.1       jtk 	 * potential partial inode area, and the gain in space would be
    907   1.1       jtk 	 * minimal, at most the pre-sb data area. */
    908   1.1       jtk 	if (cgdmin(newsb, newsb->fs_ncg - 1) > newsb->fs_size) {
    909   1.1       jtk 		newsb->fs_ncg--;
    910   1.4  christos 		newsb->fs_old_ncyl = newsb->fs_ncg * newsb->fs_old_cpg;
    911  1.15       riz 		newsb->fs_size = (newsb->fs_old_ncyl * newsb->fs_old_spc)
    912  1.15       riz 		    / NSPF(newsb);
    913   1.1       jtk 		printf("Warning: last cylinder group is too small;\n");
    914   1.1       jtk 		printf("    dropping it.  New size = %lu.\n",
    915   1.1       jtk 		    (unsigned long int) fsbtodb(newsb, newsb->fs_size));
    916   1.1       jtk 	}
    917   1.1       jtk 	/* Find out how big the csum area is, and realloc csums if bigger. */
    918   1.1       jtk 	newsb->fs_cssize = fragroundup(newsb,
    919   1.1       jtk 	    newsb->fs_ncg * sizeof(struct csum));
    920   1.1       jtk 	if (newsb->fs_cssize > oldsb->fs_cssize)
    921   1.1       jtk 		csums = nfrealloc(csums, newsb->fs_cssize, "new cg summary");
    922   1.1       jtk 	/* If we're adding any cgs, realloc structures and set up the new cgs. */
    923   1.1       jtk 	if (newsb->fs_ncg > oldsb->fs_ncg) {
    924   1.1       jtk 		char *cgp;
    925   1.1       jtk 		cgs = nfrealloc(cgs, newsb->fs_ncg * sizeof(struct cg *),
    926   1.1       jtk                                 "cg pointers");
    927   1.1       jtk 		cgflags = nfrealloc(cgflags, newsb->fs_ncg, "cg flags");
    928   1.1       jtk 		bzero(cgflags + oldsb->fs_ncg, newsb->fs_ncg - oldsb->fs_ncg);
    929   1.1       jtk 		cgp = alloconce((newsb->fs_ncg - oldsb->fs_ncg) * cgblksz,
    930   1.1       jtk                                 "cgs");
    931   1.1       jtk 		for (i = oldsb->fs_ncg; i < newsb->fs_ncg; i++) {
    932   1.1       jtk 			cgs[i] = (struct cg *) cgp;
    933   1.1       jtk 			initcg(i);
    934   1.1       jtk 			cgp += cgblksz;
    935   1.1       jtk 		}
    936   1.4  christos 		cgs[oldsb->fs_ncg - 1]->cg_old_ncyl = oldsb->fs_old_cpg;
    937   1.1       jtk 		cgflags[oldsb->fs_ncg - 1] |= CGF_DIRTY;
    938   1.1       jtk 	}
    939   1.1       jtk 	/* If the old fs ended partway through a cg, we have to update the old
    940   1.1       jtk 	 * last cg (though possibly not to a full cg!). */
    941   1.1       jtk 	if (oldsb->fs_size % oldsb->fs_fpg) {
    942   1.1       jtk 		struct cg *cg;
    943   1.1       jtk 		int newcgsize;
    944   1.1       jtk 		int prevcgtop;
    945   1.1       jtk 		int oldcgsize;
    946   1.1       jtk 		cg = cgs[oldsb->fs_ncg - 1];
    947   1.1       jtk 		cgflags[oldsb->fs_ncg - 1] |= CGF_DIRTY | CGF_BLKMAPS;
    948   1.1       jtk 		prevcgtop = oldsb->fs_fpg * (oldsb->fs_ncg - 1);
    949   1.1       jtk 		newcgsize = newsb->fs_size - prevcgtop;
    950   1.1       jtk 		if (newcgsize > newsb->fs_fpg)
    951   1.1       jtk 			newcgsize = newsb->fs_fpg;
    952   1.1       jtk 		oldcgsize = oldsb->fs_size % oldsb->fs_fpg;
    953   1.1       jtk 		set_bits(cg_blksfree(cg, 0), oldcgsize, newcgsize - oldcgsize);
    954  1.20    mhitch 		cg->cg_old_ncyl = oldsb->fs_old_cpg;
    955   1.1       jtk 		cg->cg_ndblk = newcgsize;
    956   1.1       jtk 	}
    957   1.1       jtk 	/* Fix up the csum info, if necessary. */
    958   1.1       jtk 	csum_fixup();
    959   1.1       jtk 	/* Make fs_dsize match the new reality. */
    960   1.1       jtk 	recompute_fs_dsize();
    961   1.1       jtk }
    962   1.1       jtk /*
    963   1.1       jtk  * Call (*fn)() for each inode, passing the inode and its inumber.  The
    964   1.1       jtk  *  number of cylinder groups is pased in, so this can be used to map
    965  1.24       wiz  *  over either the old or the new file system's set of inodes.
    966   1.1       jtk  */
    967   1.1       jtk static void
    968  1.15       riz map_inodes(void (*fn) (struct ufs1_dinode * di, unsigned int, void *arg),
    969  1.15       riz 	   int ncg, void *cbarg) {
    970   1.1       jtk 	int i;
    971   1.1       jtk 	int ni;
    972   1.1       jtk 
    973   1.1       jtk 	ni = oldsb->fs_ipg * ncg;
    974   1.1       jtk 	for (i = 0; i < ni; i++)
    975   1.1       jtk 		(*fn) (inodes + i, i, cbarg);
    976   1.1       jtk }
    977   1.1       jtk /* Values for the third argument to the map function for
    978   1.1       jtk  * map_inode_data_blocks.  MDB_DATA indicates the block is contains
    979   1.1       jtk  * file data; MDB_INDIR_PRE and MDB_INDIR_POST indicate that it's an
    980   1.1       jtk  * indirect block.  The MDB_INDIR_PRE call is made before the indirect
    981   1.1       jtk  * block pointers are followed and the pointed-to blocks scanned,
    982   1.1       jtk  * MDB_INDIR_POST after.
    983   1.1       jtk  */
    984   1.1       jtk #define MDB_DATA       1
    985   1.1       jtk #define MDB_INDIR_PRE  2
    986   1.1       jtk #define MDB_INDIR_POST 3
    987   1.1       jtk 
    988  1.15       riz typedef void (*mark_callback_t) (unsigned int blocknum, unsigned int nfrags,
    989  1.15       riz 				 unsigned int blksize, int opcode);
    990   1.1       jtk 
    991   1.1       jtk /* Helper function - handles a data block.  Calls the callback
    992   1.1       jtk  * function and returns number of bytes occupied in file (actually,
    993   1.1       jtk  * rounded up to a frag boundary).  The name is historical.  */
    994   1.1       jtk static int
    995   1.4  christos markblk(mark_callback_t fn, struct ufs1_dinode * di, int bn, off_t o)
    996   1.1       jtk {
    997   1.1       jtk 	int sz;
    998   1.1       jtk 	int nb;
    999   1.1       jtk 	if (o >= di->di_size)
   1000   1.1       jtk 		return (0);
   1001   1.1       jtk 	sz = dblksize(newsb, di, lblkno(newsb, o));
   1002   1.1       jtk 	nb = (sz > di->di_size - o) ? di->di_size - o : sz;
   1003   1.1       jtk 	if (bn)
   1004   1.1       jtk 		(*fn) (bn, numfrags(newsb, sz), nb, MDB_DATA);
   1005   1.1       jtk 	return (sz);
   1006   1.1       jtk }
   1007   1.1       jtk /* Helper function - handles an indirect block.  Makes the
   1008   1.1       jtk  * MDB_INDIR_PRE callback for the indirect block, loops over the
   1009   1.1       jtk  * pointers and recurses, and makes the MDB_INDIR_POST callback.
   1010   1.1       jtk  * Returns the number of bytes occupied in file, as does markblk().
   1011   1.1       jtk  * For the sake of update_for_data_move(), we read the indirect block
   1012   1.1       jtk  * _after_ making the _PRE callback.  The name is historical.  */
   1013   1.1       jtk static int
   1014   1.4  christos markiblk(mark_callback_t fn, struct ufs1_dinode * di, int bn, off_t o, int lev)
   1015   1.1       jtk {
   1016   1.1       jtk 	int i;
   1017   1.1       jtk 	int j;
   1018   1.1       jtk 	int tot;
   1019   1.2    martin 	static int32_t indirblk1[howmany(MAXBSIZE, sizeof(int32_t))];
   1020   1.2    martin 	static int32_t indirblk2[howmany(MAXBSIZE, sizeof(int32_t))];
   1021   1.2    martin 	static int32_t indirblk3[howmany(MAXBSIZE, sizeof(int32_t))];
   1022   1.2    martin 	static int32_t *indirblks[3] = {
   1023   1.1       jtk 		&indirblk1[0], &indirblk2[0], &indirblk3[0]
   1024   1.1       jtk 	};
   1025   1.1       jtk 	if (lev < 0)
   1026   1.1       jtk 		return (markblk(fn, di, bn, o));
   1027   1.1       jtk 	if (bn == 0) {
   1028   1.1       jtk 		for (i = newsb->fs_bsize;
   1029   1.1       jtk 		    lev >= 0;
   1030   1.1       jtk 		    i *= NINDIR(newsb), lev--);
   1031   1.1       jtk 		return (i);
   1032   1.1       jtk 	}
   1033   1.1       jtk 	(*fn) (bn, newsb->fs_frag, newsb->fs_bsize, MDB_INDIR_PRE);
   1034   1.1       jtk 	readat(fsbtodb(newsb, bn), indirblks[lev], newsb->fs_bsize);
   1035   1.1       jtk 	tot = 0;
   1036   1.1       jtk 	for (i = 0; i < NINDIR(newsb); i++) {
   1037   1.1       jtk 		j = markiblk(fn, di, indirblks[lev][i], o, lev - 1);
   1038   1.1       jtk 		if (j == 0)
   1039   1.1       jtk 			break;
   1040   1.1       jtk 		o += j;
   1041   1.1       jtk 		tot += j;
   1042   1.1       jtk 	}
   1043   1.1       jtk 	(*fn) (bn, newsb->fs_frag, newsb->fs_bsize, MDB_INDIR_POST);
   1044   1.1       jtk 	return (tot);
   1045   1.1       jtk }
   1046   1.1       jtk 
   1047   1.1       jtk 
   1048   1.1       jtk /*
   1049   1.1       jtk  * Call (*fn)() for each data block for an inode.  This routine assumes
   1050   1.1       jtk  *  the inode is known to be of a type that has data blocks (file,
   1051   1.1       jtk  *  directory, or non-fast symlink).  The called function is:
   1052   1.1       jtk  *
   1053   1.1       jtk  * (*fn)(unsigned int blkno, unsigned int nf, unsigned int nb, int op)
   1054   1.1       jtk  *
   1055   1.1       jtk  *  where blkno is the frag number, nf is the number of frags starting
   1056   1.1       jtk  *  at blkno (always <= fs_frag), nb is the number of bytes that belong
   1057   1.1       jtk  *  to the file (usually nf*fs_frag, often less for the last block/frag
   1058   1.1       jtk  *  of a file).
   1059   1.1       jtk  */
   1060   1.1       jtk static void
   1061   1.4  christos map_inode_data_blocks(struct ufs1_dinode * di, mark_callback_t fn)
   1062   1.1       jtk {
   1063   1.1       jtk 	off_t o;		/* offset within  inode */
   1064   1.1       jtk 	int inc;		/* increment for o - maybe should be off_t? */
   1065   1.1       jtk 	int b;			/* index within di_db[] and di_ib[] arrays */
   1066   1.1       jtk 
   1067   1.1       jtk 	/* Scan the direct blocks... */
   1068   1.1       jtk 	o = 0;
   1069   1.1       jtk 	for (b = 0; b < NDADDR; b++) {
   1070   1.1       jtk 		inc = markblk(fn, di, di->di_db[b], o);
   1071   1.1       jtk 		if (inc == 0)
   1072   1.1       jtk 			break;
   1073   1.1       jtk 		o += inc;
   1074   1.1       jtk 	}
   1075   1.1       jtk 	/* ...and the indirect blocks. */
   1076   1.1       jtk 	if (inc) {
   1077   1.1       jtk 		for (b = 0; b < NIADDR; b++) {
   1078   1.1       jtk 			inc = markiblk(fn, di, di->di_ib[b], o, b);
   1079   1.1       jtk 			if (inc == 0)
   1080   1.1       jtk 				return;
   1081   1.1       jtk 			o += inc;
   1082   1.1       jtk 		}
   1083   1.1       jtk 	}
   1084   1.1       jtk }
   1085   1.1       jtk 
   1086   1.1       jtk static void
   1087   1.4  christos dblk_callback(struct ufs1_dinode * di, unsigned int inum, void *arg)
   1088   1.1       jtk {
   1089   1.1       jtk 	mark_callback_t fn;
   1090   1.1       jtk 	fn = (mark_callback_t) arg;
   1091   1.1       jtk 	switch (di->di_mode & IFMT) {
   1092   1.1       jtk 	case IFLNK:
   1093   1.1       jtk 		if (di->di_size > newsb->fs_maxsymlinklen) {
   1094   1.1       jtk 	case IFDIR:
   1095   1.1       jtk 	case IFREG:
   1096   1.1       jtk 			map_inode_data_blocks(di, fn);
   1097   1.1       jtk 		}
   1098   1.1       jtk 		break;
   1099   1.1       jtk 	}
   1100   1.1       jtk }
   1101   1.1       jtk /*
   1102   1.1       jtk  * Make a callback call, a la map_inode_data_blocks, for all data
   1103   1.1       jtk  *  blocks in the entire fs.  This is used only once, in
   1104   1.1       jtk  *  update_for_data_move, but it's out at top level because the complex
   1105   1.1       jtk  *  downward-funarg nesting that would otherwise result seems to give
   1106   1.1       jtk  *  gcc gastric distress.
   1107   1.1       jtk  */
   1108   1.1       jtk static void
   1109   1.1       jtk map_data_blocks(mark_callback_t fn, int ncg)
   1110   1.1       jtk {
   1111   1.1       jtk 	map_inodes(&dblk_callback, ncg, (void *) fn);
   1112   1.1       jtk }
   1113   1.1       jtk /*
   1114   1.1       jtk  * Initialize the blkmove array.
   1115   1.1       jtk  */
   1116   1.1       jtk static void
   1117   1.1       jtk blkmove_init(void)
   1118   1.1       jtk {
   1119   1.1       jtk 	int i;
   1120   1.1       jtk 
   1121   1.1       jtk 	blkmove = alloconce(oldsb->fs_size * sizeof(*blkmove), "blkmove");
   1122   1.1       jtk 	for (i = 0; i < oldsb->fs_size; i++)
   1123   1.1       jtk 		blkmove[i] = i;
   1124   1.1       jtk }
   1125   1.1       jtk /*
   1126   1.1       jtk  * Load the inodes off disk.  Allocates the structures and initializes
   1127   1.1       jtk  *  them - the inodes from disk, the flags to zero.
   1128   1.1       jtk  */
   1129   1.1       jtk static void
   1130   1.1       jtk loadinodes(void)
   1131   1.1       jtk {
   1132   1.1       jtk 	int cg;
   1133   1.4  christos 	struct ufs1_dinode *iptr;
   1134   1.1       jtk 
   1135  1.15       riz 	inodes = alloconce(oldsb->fs_ncg * oldsb->fs_ipg *
   1136  1.15       riz 	    sizeof(struct ufs1_dinode), "inodes");
   1137   1.1       jtk 	iflags = alloconce(oldsb->fs_ncg * oldsb->fs_ipg, "inode flags");
   1138   1.1       jtk 	bzero(iflags, oldsb->fs_ncg * oldsb->fs_ipg);
   1139   1.1       jtk 	iptr = inodes;
   1140   1.1       jtk 	for (cg = 0; cg < oldsb->fs_ncg; cg++) {
   1141   1.1       jtk 		readat(fsbtodb(oldsb, cgimin(oldsb, cg)), iptr,
   1142   1.4  christos 		    oldsb->fs_ipg * sizeof(struct ufs1_dinode));
   1143   1.1       jtk 		iptr += oldsb->fs_ipg;
   1144   1.1       jtk 	}
   1145   1.1       jtk }
   1146   1.1       jtk /*
   1147  1.24       wiz  * Report a file-system-too-full problem.
   1148   1.1       jtk  */
   1149   1.1       jtk static void
   1150   1.1       jtk toofull(void)
   1151   1.1       jtk {
   1152   1.1       jtk 	printf("Sorry, would run out of data blocks\n");
   1153  1.15       riz 	exit(EXIT_FAILURE);
   1154   1.1       jtk }
   1155   1.1       jtk /*
   1156   1.1       jtk  * Record a desire to move "n" frags from "from" to "to".
   1157   1.1       jtk  */
   1158   1.1       jtk static void
   1159   1.1       jtk mark_move(unsigned int from, unsigned int to, unsigned int n)
   1160   1.1       jtk {
   1161   1.1       jtk 	for (; n > 0; n--)
   1162   1.1       jtk 		blkmove[from++] = to++;
   1163   1.1       jtk }
   1164   1.1       jtk /* Helper function - evict n frags, starting with start (cg-relative).
   1165   1.1       jtk  * The free bitmap is scanned, unallocated frags are ignored, and
   1166   1.1       jtk  * each block of consecutive allocated frags is moved as a unit.
   1167   1.1       jtk  */
   1168   1.1       jtk static void
   1169   1.1       jtk fragmove(struct cg * cg, int base, unsigned int start, unsigned int n)
   1170   1.1       jtk {
   1171   1.1       jtk 	int i;
   1172   1.1       jtk 	int run;
   1173   1.1       jtk 	run = 0;
   1174   1.1       jtk 	for (i = 0; i <= n; i++) {
   1175   1.1       jtk 		if ((i < n) && bit_is_clr(cg_blksfree(cg, 0), start + i)) {
   1176   1.1       jtk 			run++;
   1177   1.1       jtk 		} else {
   1178   1.1       jtk 			if (run > 0) {
   1179   1.1       jtk 				int off;
   1180   1.1       jtk 				off = find_freespace(run);
   1181   1.1       jtk 				if (off < 0)
   1182   1.1       jtk 					toofull();
   1183   1.1       jtk 				mark_move(base + start + i - run, off, run);
   1184   1.1       jtk 				set_bits(cg_blksfree(cg, 0), start + i - run,
   1185   1.1       jtk 				    run);
   1186   1.1       jtk 				clr_bits(cg_blksfree(cgs[dtog(oldsb, off)], 0),
   1187   1.1       jtk 				    dtogd(oldsb, off), run);
   1188   1.1       jtk 			}
   1189   1.1       jtk 			run = 0;
   1190   1.1       jtk 		}
   1191   1.1       jtk 	}
   1192   1.1       jtk }
   1193   1.1       jtk /*
   1194   1.1       jtk  * Evict all data blocks from the given cg, starting at minfrag (based
   1195   1.1       jtk  *  at the beginning of the cg), for length nfrag.  The eviction is
   1196   1.1       jtk  *  assumed to be entirely data-area; this should not be called with a
   1197   1.1       jtk  *  range overlapping the metadata structures in the cg.  It also
   1198   1.1       jtk  *  assumes minfrag points into the given cg; it will misbehave if this
   1199   1.1       jtk  *  is not true.
   1200   1.1       jtk  *
   1201   1.1       jtk  * See the comment header on find_freespace() for one possible bug
   1202   1.1       jtk  *  lurking here.
   1203   1.1       jtk  */
   1204   1.1       jtk static void
   1205   1.1       jtk evict_data(struct cg * cg, unsigned int minfrag, unsigned int nfrag)
   1206   1.1       jtk {
   1207   1.1       jtk 	int base;		/* base of cg (in frags from beginning of fs) */
   1208   1.1       jtk 
   1209   1.1       jtk 
   1210   1.1       jtk 	base = cgbase(oldsb, cg->cg_cgx);
   1211   1.1       jtk 	/* Does the boundary fall in the middle of a block?  To avoid breaking
   1212   1.1       jtk 	 * between frags allocated as consecutive, we always evict the whole
   1213   1.1       jtk 	 * block in this case, though one could argue we should check to see
   1214   1.1       jtk 	 * if the frag before or after the break is unallocated. */
   1215   1.1       jtk 	if (minfrag % oldsb->fs_frag) {
   1216   1.1       jtk 		int n;
   1217   1.1       jtk 		n = minfrag % oldsb->fs_frag;
   1218   1.1       jtk 		minfrag -= n;
   1219   1.1       jtk 		nfrag += n;
   1220   1.1       jtk 	}
   1221   1.1       jtk 	/* Do whole blocks.  If a block is wholly free, skip it; if wholly
   1222   1.1       jtk 	 * allocated, move it in toto.  If neither, call fragmove() to move
   1223   1.1       jtk 	 * the frags to new locations. */
   1224   1.1       jtk 	while (nfrag >= oldsb->fs_frag) {
   1225   1.1       jtk 		if (!blk_is_set(cg_blksfree(cg, 0), minfrag, oldsb->fs_frag)) {
   1226   1.1       jtk 			if (blk_is_clr(cg_blksfree(cg, 0), minfrag,
   1227   1.1       jtk 				oldsb->fs_frag)) {
   1228   1.1       jtk 				int off;
   1229   1.1       jtk 				off = find_freeblock();
   1230   1.1       jtk 				if (off < 0)
   1231   1.1       jtk 					toofull();
   1232   1.1       jtk 				mark_move(base + minfrag, off, oldsb->fs_frag);
   1233   1.1       jtk 				set_bits(cg_blksfree(cg, 0), minfrag,
   1234   1.1       jtk 				    oldsb->fs_frag);
   1235   1.1       jtk 				clr_bits(cg_blksfree(cgs[dtog(oldsb, off)], 0),
   1236   1.1       jtk 				    dtogd(oldsb, off), oldsb->fs_frag);
   1237   1.1       jtk 			} else {
   1238   1.1       jtk 				fragmove(cg, base, minfrag, oldsb->fs_frag);
   1239   1.1       jtk 			}
   1240   1.1       jtk 		}
   1241   1.1       jtk 		minfrag += oldsb->fs_frag;
   1242   1.1       jtk 		nfrag -= oldsb->fs_frag;
   1243   1.1       jtk 	}
   1244   1.1       jtk 	/* Clean up any sub-block amount left over. */
   1245   1.1       jtk 	if (nfrag) {
   1246   1.1       jtk 		fragmove(cg, base, minfrag, nfrag);
   1247   1.1       jtk 	}
   1248   1.1       jtk }
   1249   1.1       jtk /*
   1250   1.1       jtk  * Move all data blocks according to blkmove.  We have to be careful,
   1251   1.1       jtk  *  because we may be updating indirect blocks that will themselves be
   1252   1.2    martin  *  getting moved, or inode int32_t arrays that point to indirect
   1253   1.1       jtk  *  blocks that will be moved.  We call this before
   1254   1.1       jtk  *  update_for_data_move, and update_for_data_move does inodes first,
   1255   1.1       jtk  *  then indirect blocks in preorder, so as to make sure that the
   1256  1.24       wiz  *  file system is self-consistent at all points, for better crash
   1257   1.1       jtk  *  tolerance.  (We can get away with this only because all the writes
   1258   1.1       jtk  *  done by perform_data_move() are writing into space that's not used
   1259  1.24       wiz  *  by the old file system.)  If we crash, some things may point to the
   1260   1.1       jtk  *  old data and some to the new, but both copies are the same.  The
   1261   1.1       jtk  *  only wrong things should be csum info and free bitmaps, which fsck
   1262   1.1       jtk  *  is entirely capable of cleaning up.
   1263   1.1       jtk  *
   1264   1.1       jtk  * Since blkmove_init() initializes all blocks to move to their current
   1265   1.1       jtk  *  locations, we can have two blocks marked as wanting to move to the
   1266   1.1       jtk  *  same location, but only two and only when one of them is the one
   1267   1.1       jtk  *  that was already there.  So if blkmove[i]==i, we ignore that entry
   1268   1.1       jtk  *  entirely - for unallocated blocks, we don't want it (and may be
   1269   1.1       jtk  *  putting something else there), and for allocated blocks, we don't
   1270   1.1       jtk  *  want to copy it anywhere.
   1271   1.1       jtk  */
   1272   1.1       jtk static void
   1273   1.1       jtk perform_data_move(void)
   1274   1.1       jtk {
   1275   1.1       jtk 	int i;
   1276   1.1       jtk 	int run;
   1277   1.1       jtk 	int maxrun;
   1278   1.1       jtk 	char buf[65536];
   1279   1.1       jtk 
   1280   1.1       jtk 	maxrun = sizeof(buf) / newsb->fs_fsize;
   1281   1.1       jtk 	run = 0;
   1282   1.1       jtk 	for (i = 0; i < oldsb->fs_size; i++) {
   1283   1.1       jtk 		if ((blkmove[i] == i) ||
   1284   1.1       jtk 		    (run >= maxrun) ||
   1285   1.1       jtk 		    ((run > 0) &&
   1286   1.1       jtk 			(blkmove[i] != blkmove[i - 1] + 1))) {
   1287   1.1       jtk 			if (run > 0) {
   1288   1.1       jtk 				readat(fsbtodb(oldsb, i - run), &buf[0],
   1289   1.1       jtk 				    run << oldsb->fs_fshift);
   1290   1.1       jtk 				writeat(fsbtodb(oldsb, blkmove[i - run]),
   1291   1.1       jtk 				    &buf[0], run << oldsb->fs_fshift);
   1292   1.1       jtk 			}
   1293   1.1       jtk 			run = 0;
   1294   1.1       jtk 		}
   1295   1.1       jtk 		if (blkmove[i] != i)
   1296   1.1       jtk 			run++;
   1297   1.1       jtk 	}
   1298   1.1       jtk 	if (run > 0) {
   1299   1.1       jtk 		readat(fsbtodb(oldsb, i - run), &buf[0],
   1300   1.1       jtk 		    run << oldsb->fs_fshift);
   1301   1.1       jtk 		writeat(fsbtodb(oldsb, blkmove[i - run]), &buf[0],
   1302   1.1       jtk 		    run << oldsb->fs_fshift);
   1303   1.1       jtk 	}
   1304   1.1       jtk }
   1305   1.1       jtk /*
   1306   1.2    martin  * This modifies an array of int32_t, according to blkmove.  This is
   1307   1.1       jtk  *  used to update inode block arrays and indirect blocks to point to
   1308   1.1       jtk  *  the new locations of data blocks.
   1309   1.1       jtk  *
   1310   1.2    martin  * Return value is the number of int32_ts that needed updating; in
   1311   1.1       jtk  *  particular, the return value is zero iff nothing was modified.
   1312   1.1       jtk  */
   1313   1.1       jtk static int
   1314   1.2    martin movemap_blocks(int32_t * vec, int n)
   1315   1.1       jtk {
   1316   1.1       jtk 	int rv;
   1317   1.1       jtk 
   1318   1.1       jtk 	rv = 0;
   1319   1.1       jtk 	for (; n > 0; n--, vec++) {
   1320   1.1       jtk 		if (blkmove[*vec] != *vec) {
   1321   1.1       jtk 			*vec = blkmove[*vec];
   1322   1.1       jtk 			rv++;
   1323   1.1       jtk 		}
   1324   1.1       jtk 	}
   1325   1.1       jtk 	return (rv);
   1326   1.1       jtk }
   1327   1.1       jtk static void
   1328   1.4  christos moveblocks_callback(struct ufs1_dinode * di, unsigned int inum, void *arg)
   1329   1.1       jtk {
   1330   1.1       jtk 	switch (di->di_mode & IFMT) {
   1331   1.1       jtk 	case IFLNK:
   1332   1.1       jtk 		if (di->di_size > oldsb->fs_maxsymlinklen) {
   1333   1.1       jtk 	case IFDIR:
   1334   1.1       jtk 	case IFREG:
   1335   1.1       jtk 			/* don't || these two calls; we need their
   1336   1.1       jtk 			 * side-effects */
   1337   1.1       jtk 			if (movemap_blocks(&di->di_db[0], NDADDR)) {
   1338   1.1       jtk 				iflags[inum] |= IF_DIRTY;
   1339   1.1       jtk 			}
   1340   1.1       jtk 			if (movemap_blocks(&di->di_ib[0], NIADDR)) {
   1341   1.1       jtk 				iflags[inum] |= IF_DIRTY;
   1342   1.1       jtk 			}
   1343   1.1       jtk 		}
   1344   1.1       jtk 		break;
   1345   1.1       jtk 	}
   1346   1.1       jtk }
   1347   1.1       jtk 
   1348   1.1       jtk static void
   1349  1.15       riz moveindir_callback(unsigned int off, unsigned int nfrag, unsigned int nbytes,
   1350  1.15       riz 		   int kind)
   1351   1.1       jtk {
   1352   1.1       jtk 	if (kind == MDB_INDIR_PRE) {
   1353   1.2    martin 		int32_t blk[howmany(MAXBSIZE, sizeof(int32_t))];
   1354   1.1       jtk 		readat(fsbtodb(oldsb, off), &blk[0], oldsb->fs_bsize);
   1355   1.1       jtk 		if (movemap_blocks(&blk[0], NINDIR(oldsb))) {
   1356   1.1       jtk 			writeat(fsbtodb(oldsb, off), &blk[0], oldsb->fs_bsize);
   1357   1.1       jtk 		}
   1358   1.1       jtk 	}
   1359   1.1       jtk }
   1360   1.1       jtk /*
   1361   1.1       jtk  * Update all inode data arrays and indirect blocks to point to the new
   1362   1.1       jtk  *  locations of data blocks.  See the comment header on
   1363   1.1       jtk  *  perform_data_move for some ordering considerations.
   1364   1.1       jtk  */
   1365   1.1       jtk static void
   1366   1.1       jtk update_for_data_move(void)
   1367   1.1       jtk {
   1368   1.1       jtk 	map_inodes(&moveblocks_callback, oldsb->fs_ncg, NULL);
   1369   1.1       jtk 	map_data_blocks(&moveindir_callback, oldsb->fs_ncg);
   1370   1.1       jtk }
   1371   1.1       jtk /*
   1372   1.1       jtk  * Initialize the inomove array.
   1373   1.1       jtk  */
   1374   1.1       jtk static void
   1375   1.1       jtk inomove_init(void)
   1376   1.1       jtk {
   1377   1.1       jtk 	int i;
   1378   1.1       jtk 
   1379   1.1       jtk 	inomove = alloconce(oldsb->fs_ipg * oldsb->fs_ncg * sizeof(*inomove),
   1380   1.1       jtk                             "inomove");
   1381   1.1       jtk 	for (i = (oldsb->fs_ipg * oldsb->fs_ncg) - 1; i >= 0; i--)
   1382   1.1       jtk 		inomove[i] = i;
   1383   1.1       jtk }
   1384   1.1       jtk /*
   1385   1.1       jtk  * Flush all dirtied inodes to disk.  Scans the inode flags array; for
   1386   1.1       jtk  *  each dirty inode, it sets the BDIRTY bit on the first inode in the
   1387   1.1       jtk  *  block containing the dirty inode.  Then it scans by blocks, and for
   1388   1.1       jtk  *  each marked block, writes it.
   1389   1.1       jtk  */
   1390   1.1       jtk static void
   1391   1.1       jtk flush_inodes(void)
   1392   1.1       jtk {
   1393   1.1       jtk 	int i;
   1394   1.1       jtk 	int ni;
   1395   1.1       jtk 	int m;
   1396   1.1       jtk 
   1397   1.1       jtk 	ni = newsb->fs_ipg * newsb->fs_ncg;
   1398   1.1       jtk 	m = INOPB(newsb) - 1;
   1399   1.1       jtk 	for (i = 0; i < ni; i++) {
   1400   1.1       jtk 		if (iflags[i] & IF_DIRTY) {
   1401   1.1       jtk 			iflags[i & ~m] |= IF_BDIRTY;
   1402   1.1       jtk 		}
   1403   1.1       jtk 	}
   1404   1.1       jtk 	m++;
   1405   1.1       jtk 	for (i = 0; i < ni; i += m) {
   1406   1.1       jtk 		if (iflags[i] & IF_BDIRTY) {
   1407   1.1       jtk 			writeat(fsbtodb(newsb, ino_to_fsba(newsb, i)),
   1408   1.1       jtk 			    inodes + i, newsb->fs_bsize);
   1409   1.1       jtk 		}
   1410   1.1       jtk 	}
   1411   1.1       jtk }
   1412   1.1       jtk /*
   1413   1.1       jtk  * Evict all inodes from the specified cg.  shrink() already checked
   1414   1.1       jtk  *  that there were enough free inodes, so the no-free-inodes check is
   1415  1.24       wiz  *  a can't-happen.  If it does trip, the file system should be in good
   1416   1.1       jtk  *  enough shape for fsck to fix; see the comment on perform_data_move
   1417   1.1       jtk  *  for the considerations in question.
   1418   1.1       jtk  */
   1419   1.1       jtk static void
   1420   1.1       jtk evict_inodes(struct cg * cg)
   1421   1.1       jtk {
   1422   1.1       jtk 	int inum;
   1423   1.1       jtk 	int i;
   1424   1.1       jtk 	int fi;
   1425   1.1       jtk 
   1426   1.1       jtk 	inum = newsb->fs_ipg * cg->cg_cgx;
   1427   1.1       jtk 	for (i = 0; i < newsb->fs_ipg; i++, inum++) {
   1428   1.1       jtk 		if (inodes[inum].di_mode != 0) {
   1429   1.1       jtk 			fi = find_freeinode();
   1430   1.1       jtk 			if (fi < 0) {
   1431   1.1       jtk 				printf("Sorry, inodes evaporated - "
   1432  1.24       wiz 				    "file system probably needs fsck\n");
   1433  1.15       riz 				exit(EXIT_FAILURE);
   1434   1.1       jtk 			}
   1435   1.1       jtk 			inomove[inum] = fi;
   1436   1.1       jtk 			clr_bits(cg_inosused(cg, 0), i, 1);
   1437   1.1       jtk 			set_bits(cg_inosused(cgs[ino_to_cg(newsb, fi)], 0),
   1438   1.1       jtk 			    fi % newsb->fs_ipg, 1);
   1439   1.1       jtk 		}
   1440   1.1       jtk 	}
   1441   1.1       jtk }
   1442   1.1       jtk /*
   1443   1.1       jtk  * Move inodes from old locations to new.  Does not actually write
   1444   1.1       jtk  *  anything to disk; just copies in-core and sets dirty bits.
   1445   1.1       jtk  *
   1446   1.1       jtk  * We have to be careful here for reasons similar to those mentioned in
   1447   1.1       jtk  *  the comment header on perform_data_move, above: for the sake of
   1448   1.1       jtk  *  crash tolerance, we want to make sure everything is present at both
   1449   1.1       jtk  *  old and new locations before we update pointers.  So we call this
   1450   1.1       jtk  *  first, then flush_inodes() to get them out on disk, then update
   1451   1.1       jtk  *  directories to match.
   1452   1.1       jtk  */
   1453   1.1       jtk static void
   1454   1.1       jtk perform_inode_move(void)
   1455   1.1       jtk {
   1456   1.1       jtk 	int i;
   1457   1.1       jtk 	int ni;
   1458   1.1       jtk 
   1459   1.1       jtk 	ni = oldsb->fs_ipg * oldsb->fs_ncg;
   1460   1.1       jtk 	for (i = 0; i < ni; i++) {
   1461   1.1       jtk 		if (inomove[i] != i) {
   1462   1.1       jtk 			inodes[inomove[i]] = inodes[i];
   1463   1.1       jtk 			iflags[inomove[i]] = iflags[i] | IF_DIRTY;
   1464   1.1       jtk 		}
   1465   1.1       jtk 	}
   1466   1.1       jtk }
   1467   1.1       jtk /*
   1468   1.1       jtk  * Update the directory contained in the nb bytes at buf, to point to
   1469   1.1       jtk  *  inodes' new locations.
   1470   1.1       jtk  */
   1471   1.1       jtk static int
   1472   1.1       jtk update_dirents(char *buf, int nb)
   1473   1.1       jtk {
   1474   1.1       jtk 	int rv;
   1475   1.1       jtk #define d ((struct direct *)buf)
   1476   1.1       jtk 
   1477   1.1       jtk 	rv = 0;
   1478   1.1       jtk 	while (nb > 0) {
   1479   1.1       jtk 		if (inomove[d->d_ino] != d->d_ino) {
   1480   1.1       jtk 			rv++;
   1481   1.1       jtk 			d->d_ino = inomove[d->d_ino];
   1482   1.1       jtk 		}
   1483   1.1       jtk 		nb -= d->d_reclen;
   1484   1.1       jtk 		buf += d->d_reclen;
   1485   1.1       jtk 	}
   1486   1.1       jtk 	return (rv);
   1487   1.1       jtk #undef d
   1488   1.1       jtk }
   1489   1.1       jtk /*
   1490   1.1       jtk  * Callback function for map_inode_data_blocks, for updating a
   1491   1.1       jtk  *  directory to point to new inode locations.
   1492   1.1       jtk  */
   1493   1.1       jtk static void
   1494   1.1       jtk update_dir_data(unsigned int bn, unsigned int size, unsigned int nb, int kind)
   1495   1.1       jtk {
   1496   1.1       jtk 	if (kind == MDB_DATA) {
   1497   1.1       jtk 		union {
   1498   1.1       jtk 			struct direct d;
   1499   1.1       jtk 			char ch[MAXBSIZE];
   1500   1.1       jtk 		}     buf;
   1501   1.1       jtk 		readat(fsbtodb(oldsb, bn), &buf, size << oldsb->fs_fshift);
   1502   1.1       jtk 		if (update_dirents((char *) &buf, nb)) {
   1503   1.1       jtk 			writeat(fsbtodb(oldsb, bn), &buf,
   1504   1.1       jtk 			    size << oldsb->fs_fshift);
   1505   1.1       jtk 		}
   1506   1.1       jtk 	}
   1507   1.1       jtk }
   1508   1.1       jtk static void
   1509   1.4  christos dirmove_callback(struct ufs1_dinode * di, unsigned int inum, void *arg)
   1510   1.1       jtk {
   1511   1.1       jtk 	switch (di->di_mode & IFMT) {
   1512   1.1       jtk 	case IFDIR:
   1513   1.1       jtk 		map_inode_data_blocks(di, &update_dir_data);
   1514   1.1       jtk 		break;
   1515   1.1       jtk 	}
   1516   1.1       jtk }
   1517   1.1       jtk /*
   1518   1.1       jtk  * Update directory entries to point to new inode locations.
   1519   1.1       jtk  */
   1520   1.1       jtk static void
   1521   1.1       jtk update_for_inode_move(void)
   1522   1.1       jtk {
   1523   1.1       jtk 	map_inodes(&dirmove_callback, newsb->fs_ncg, NULL);
   1524   1.1       jtk }
   1525   1.1       jtk /*
   1526  1.24       wiz  * Shrink the file system.
   1527   1.1       jtk  */
   1528   1.1       jtk static void
   1529   1.1       jtk shrink(void)
   1530   1.1       jtk {
   1531   1.1       jtk 	int i;
   1532   1.1       jtk 
   1533   1.1       jtk 	/* Load the inodes off disk - we'll need 'em. */
   1534   1.1       jtk 	loadinodes();
   1535   1.1       jtk 	/* Update the timestamp. */
   1536   1.1       jtk 	newsb->fs_time = timestamp();
   1537   1.1       jtk 	/* Update the size figures. */
   1538   1.1       jtk 	newsb->fs_size = dbtofsb(newsb, newsize);
   1539  1.17       riz 	newsb->fs_old_ncyl = howmany(newsb->fs_size * NSPF(newsb),
   1540  1.17       riz 	    newsb->fs_old_spc);
   1541   1.4  christos 	newsb->fs_ncg = howmany(newsb->fs_old_ncyl, newsb->fs_old_cpg);
   1542   1.1       jtk 	/* Does the (new) last cg end before the end of its inode area?  See
   1543   1.1       jtk 	 * the similar code in grow() for more on this. */
   1544   1.1       jtk 	if (cgdmin(newsb, newsb->fs_ncg - 1) > newsb->fs_size) {
   1545   1.1       jtk 		newsb->fs_ncg--;
   1546   1.4  christos 		newsb->fs_old_ncyl = newsb->fs_ncg * newsb->fs_old_cpg;
   1547  1.15       riz 		newsb->fs_size = (newsb->fs_old_ncyl * newsb->fs_old_spc) /
   1548  1.15       riz 		    NSPF(newsb);
   1549   1.1       jtk 		printf("Warning: last cylinder group is too small;\n");
   1550   1.1       jtk 		printf("    dropping it.  New size = %lu.\n",
   1551   1.1       jtk 		    (unsigned long int) fsbtodb(newsb, newsb->fs_size));
   1552   1.1       jtk 	}
   1553   1.1       jtk 	/* Let's make sure we're not being shrunk into oblivion. */
   1554   1.1       jtk 	if (newsb->fs_ncg < 1) {
   1555  1.24       wiz 		printf("Size too small - file system would have no cylinders\n");
   1556  1.15       riz 		exit(EXIT_FAILURE);
   1557   1.1       jtk 	}
   1558   1.1       jtk 	/* Initialize for block motion. */
   1559   1.1       jtk 	blkmove_init();
   1560   1.1       jtk 	/* Update csum size, then fix up for the new size */
   1561   1.1       jtk 	newsb->fs_cssize = fragroundup(newsb,
   1562   1.1       jtk 	    newsb->fs_ncg * sizeof(struct csum));
   1563   1.1       jtk 	csum_fixup();
   1564   1.8       snj 	/* Evict data from any cgs being wholly eliminated */
   1565   1.1       jtk 	for (i = newsb->fs_ncg; i < oldsb->fs_ncg; i++) {
   1566   1.1       jtk 		int base;
   1567   1.1       jtk 		int dlow;
   1568   1.1       jtk 		int dhigh;
   1569   1.1       jtk 		int dmax;
   1570   1.1       jtk 		base = cgbase(oldsb, i);
   1571   1.1       jtk 		dlow = cgsblock(oldsb, i) - base;
   1572   1.1       jtk 		dhigh = cgdmin(oldsb, i) - base;
   1573   1.1       jtk 		dmax = oldsb->fs_size - base;
   1574   1.1       jtk 		if (dmax > cgs[i]->cg_ndblk)
   1575   1.1       jtk 			dmax = cgs[i]->cg_ndblk;
   1576   1.1       jtk 		evict_data(cgs[i], 0, dlow);
   1577   1.1       jtk 		evict_data(cgs[i], dhigh, dmax - dhigh);
   1578   1.1       jtk 		newsb->fs_cstotal.cs_ndir -= cgs[i]->cg_cs.cs_ndir;
   1579   1.1       jtk 		newsb->fs_cstotal.cs_nifree -= cgs[i]->cg_cs.cs_nifree;
   1580   1.1       jtk 		newsb->fs_cstotal.cs_nffree -= cgs[i]->cg_cs.cs_nffree;
   1581   1.1       jtk 		newsb->fs_cstotal.cs_nbfree -= cgs[i]->cg_cs.cs_nbfree;
   1582   1.1       jtk 	}
   1583   1.1       jtk 	/* Update the new last cg. */
   1584   1.1       jtk 	cgs[newsb->fs_ncg - 1]->cg_ndblk = newsb->fs_size -
   1585   1.1       jtk 	    ((newsb->fs_ncg - 1) * newsb->fs_fpg);
   1586   1.1       jtk 	/* Is the new last cg partial?  If so, evict any data from the part
   1587   1.1       jtk 	 * being shrunken away. */
   1588   1.1       jtk 	if (newsb->fs_size % newsb->fs_fpg) {
   1589   1.1       jtk 		struct cg *cg;
   1590   1.1       jtk 		int oldcgsize;
   1591   1.1       jtk 		int newcgsize;
   1592   1.1       jtk 		cg = cgs[newsb->fs_ncg - 1];
   1593   1.1       jtk 		newcgsize = newsb->fs_size % newsb->fs_fpg;
   1594  1.15       riz 		oldcgsize = oldsb->fs_size - ((newsb->fs_ncg - 1) &
   1595  1.15       riz 		    oldsb->fs_fpg);
   1596   1.1       jtk 		if (oldcgsize > oldsb->fs_fpg)
   1597   1.1       jtk 			oldcgsize = oldsb->fs_fpg;
   1598   1.1       jtk 		evict_data(cg, newcgsize, oldcgsize - newcgsize);
   1599   1.1       jtk 		clr_bits(cg_blksfree(cg, 0), newcgsize, oldcgsize - newcgsize);
   1600   1.1       jtk 	}
   1601   1.1       jtk 	/* Find out whether we would run out of inodes.  (Note we haven't
   1602  1.24       wiz 	 * actually done anything to the file system yet; all those evict_data
   1603   1.1       jtk 	 * calls just update blkmove.) */
   1604   1.1       jtk 	{
   1605   1.1       jtk 		int slop;
   1606   1.1       jtk 		slop = 0;
   1607   1.1       jtk 		for (i = 0; i < newsb->fs_ncg; i++)
   1608   1.1       jtk 			slop += cgs[i]->cg_cs.cs_nifree;
   1609   1.1       jtk 		for (; i < oldsb->fs_ncg; i++)
   1610   1.1       jtk 			slop -= oldsb->fs_ipg - cgs[i]->cg_cs.cs_nifree;
   1611   1.1       jtk 		if (slop < 0) {
   1612   1.1       jtk 			printf("Sorry, would run out of inodes\n");
   1613  1.15       riz 			exit(EXIT_FAILURE);
   1614   1.1       jtk 		}
   1615   1.1       jtk 	}
   1616   1.1       jtk 	/* Copy data, then update pointers to data.  See the comment header on
   1617   1.1       jtk 	 * perform_data_move for ordering considerations. */
   1618   1.1       jtk 	perform_data_move();
   1619   1.1       jtk 	update_for_data_move();
   1620   1.1       jtk 	/* Now do inodes.  Initialize, evict, move, update - see the comment
   1621   1.1       jtk 	 * header on perform_inode_move. */
   1622   1.1       jtk 	inomove_init();
   1623   1.1       jtk 	for (i = newsb->fs_ncg; i < oldsb->fs_ncg; i++)
   1624   1.1       jtk 		evict_inodes(cgs[i]);
   1625   1.1       jtk 	perform_inode_move();
   1626   1.1       jtk 	flush_inodes();
   1627   1.1       jtk 	update_for_inode_move();
   1628   1.1       jtk 	/* Recompute all the bitmaps; most of them probably need it anyway,
   1629   1.1       jtk 	 * the rest are just paranoia and not wanting to have to bother
   1630   1.1       jtk 	 * keeping track of exactly which ones require it. */
   1631   1.1       jtk 	for (i = 0; i < newsb->fs_ncg; i++)
   1632   1.1       jtk 		cgflags[i] |= CGF_DIRTY | CGF_BLKMAPS | CGF_INOMAPS;
   1633  1.14       riz 	/* Update the cg_old_ncyl value for the last cylinder. */
   1634  1.20    mhitch 	if ((newsb->fs_old_flags & FS_FLAGS_UPDATED) == 0)
   1635  1.14       riz 		cgs[newsb->fs_ncg - 1]->cg_old_ncyl =
   1636  1.20    mhitch 		    newsb->fs_old_ncyl % newsb->fs_old_cpg;
   1637   1.1       jtk 	/* Make fs_dsize match the new reality. */
   1638   1.1       jtk 	recompute_fs_dsize();
   1639   1.1       jtk }
   1640   1.1       jtk /*
   1641   1.1       jtk  * Recompute the block totals, block cluster summaries, and rotational
   1642   1.1       jtk  *  position summaries, for a given cg (specified by number), based on
   1643   1.1       jtk  *  its free-frag bitmap (cg_blksfree()[]).
   1644   1.1       jtk  */
   1645   1.1       jtk static void
   1646   1.1       jtk rescan_blkmaps(int cgn)
   1647   1.1       jtk {
   1648   1.1       jtk 	struct cg *cg;
   1649   1.1       jtk 	int f;
   1650   1.1       jtk 	int b;
   1651   1.1       jtk 	int blkfree;
   1652   1.1       jtk 	int blkrun;
   1653   1.1       jtk 	int fragrun;
   1654   1.1       jtk 	int fwb;
   1655   1.1       jtk 
   1656   1.1       jtk 	cg = cgs[cgn];
   1657   1.1       jtk 	/* Subtract off the current totals from the sb's summary info */
   1658   1.1       jtk 	newsb->fs_cstotal.cs_nffree -= cg->cg_cs.cs_nffree;
   1659   1.1       jtk 	newsb->fs_cstotal.cs_nbfree -= cg->cg_cs.cs_nbfree;
   1660   1.1       jtk 	/* Clear counters and bitmaps. */
   1661   1.1       jtk 	cg->cg_cs.cs_nffree = 0;
   1662   1.1       jtk 	cg->cg_cs.cs_nbfree = 0;
   1663   1.1       jtk 	bzero(&cg->cg_frsum[0], MAXFRAG * sizeof(cg->cg_frsum[0]));
   1664  1.15       riz 	bzero(&old_cg_blktot(cg, 0)[0],
   1665  1.15       riz 	    newsb->fs_old_cpg * sizeof(old_cg_blktot(cg, 0)[0]));
   1666  1.15       riz 	bzero(&old_cg_blks(newsb, cg, 0, 0)[0],
   1667   1.4  christos 	    newsb->fs_old_cpg * newsb->fs_old_nrpos *
   1668  1.15       riz 	    sizeof(old_cg_blks(newsb, cg, 0, 0)[0]));
   1669   1.1       jtk 	if (newsb->fs_contigsumsize > 0) {
   1670   1.1       jtk 		cg->cg_nclusterblks = cg->cg_ndblk / newsb->fs_frag;
   1671   1.1       jtk 		bzero(&cg_clustersum(cg, 0)[1],
   1672   1.1       jtk 		    newsb->fs_contigsumsize *
   1673   1.1       jtk 		    sizeof(cg_clustersum(cg, 0)[1]));
   1674   1.1       jtk 		bzero(&cg_clustersfree(cg, 0)[0],
   1675  1.15       riz 		    howmany((newsb->fs_old_cpg * newsb->fs_old_spc) /
   1676  1.15       riz 		    NSPB(newsb), NBBY));
   1677   1.1       jtk 	}
   1678   1.1       jtk 	/* Scan the free-frag bitmap.  Runs of free frags are kept track of
   1679   1.1       jtk 	 * with fragrun, and recorded into cg_frsum[] and cg_cs.cs_nffree; on
   1680   1.1       jtk 	 * each block boundary, entire free blocks are recorded as well. */
   1681   1.1       jtk 	blkfree = 1;
   1682   1.1       jtk 	blkrun = 0;
   1683   1.1       jtk 	fragrun = 0;
   1684   1.1       jtk 	f = 0;
   1685   1.1       jtk 	b = 0;
   1686   1.1       jtk 	fwb = 0;
   1687   1.1       jtk 	while (f < cg->cg_ndblk) {
   1688   1.1       jtk 		if (bit_is_set(cg_blksfree(cg, 0), f)) {
   1689   1.1       jtk 			fragrun++;
   1690   1.1       jtk 		} else {
   1691   1.1       jtk 			blkfree = 0;
   1692   1.1       jtk 			if (fragrun > 0) {
   1693   1.1       jtk 				cg->cg_frsum[fragrun]++;
   1694   1.1       jtk 				cg->cg_cs.cs_nffree += fragrun;
   1695   1.1       jtk 			}
   1696   1.1       jtk 			fragrun = 0;
   1697   1.1       jtk 		}
   1698   1.1       jtk 		f++;
   1699   1.1       jtk 		fwb++;
   1700   1.1       jtk 		if (fwb >= newsb->fs_frag) {
   1701   1.1       jtk 			if (blkfree) {
   1702   1.1       jtk 				cg->cg_cs.cs_nbfree++;
   1703   1.1       jtk 				if (newsb->fs_contigsumsize > 0)
   1704   1.1       jtk 					set_bits(cg_clustersfree(cg, 0), b, 1);
   1705  1.15       riz 				old_cg_blktot(cg, 0)[old_cbtocylno(newsb,
   1706  1.15       riz 				    f - newsb->fs_frag)]++;
   1707  1.15       riz 				old_cg_blks(newsb, cg,
   1708  1.15       riz 				    old_cbtocylno(newsb, f - newsb->fs_frag),
   1709  1.15       riz 				    0)[old_cbtorpos(newsb,
   1710  1.15       riz 				    f - newsb->fs_frag)]++;
   1711   1.1       jtk 				blkrun++;
   1712   1.1       jtk 			} else {
   1713   1.1       jtk 				if (fragrun > 0) {
   1714   1.1       jtk 					cg->cg_frsum[fragrun]++;
   1715   1.1       jtk 					cg->cg_cs.cs_nffree += fragrun;
   1716   1.1       jtk 				}
   1717   1.1       jtk 				if (newsb->fs_contigsumsize > 0) {
   1718   1.1       jtk 					if (blkrun > 0) {
   1719  1.15       riz 						cg_clustersum(cg, 0)[(blkrun
   1720  1.15       riz 						    > newsb->fs_contigsumsize)
   1721  1.15       riz 						    ? newsb->fs_contigsumsize
   1722  1.15       riz 						    : blkrun]++;
   1723   1.1       jtk 					}
   1724   1.1       jtk 				}
   1725   1.1       jtk 				blkrun = 0;
   1726   1.1       jtk 			}
   1727   1.1       jtk 			fwb = 0;
   1728   1.1       jtk 			b++;
   1729   1.1       jtk 			blkfree = 1;
   1730   1.1       jtk 			fragrun = 0;
   1731   1.1       jtk 		}
   1732   1.1       jtk 	}
   1733   1.1       jtk 	if (fragrun > 0) {
   1734   1.1       jtk 		cg->cg_frsum[fragrun]++;
   1735   1.1       jtk 		cg->cg_cs.cs_nffree += fragrun;
   1736   1.1       jtk 	}
   1737   1.1       jtk 	if ((blkrun > 0) && (newsb->fs_contigsumsize > 0)) {
   1738   1.1       jtk 		cg_clustersum(cg, 0)[(blkrun > newsb->fs_contigsumsize) ?
   1739   1.1       jtk 		    newsb->fs_contigsumsize : blkrun]++;
   1740   1.1       jtk 	}
   1741   1.1       jtk 	/*
   1742   1.1       jtk          * Put the updated summary info back into csums, and add it
   1743   1.1       jtk          * back into the sb's summary info.  Then mark the cg dirty.
   1744   1.1       jtk          */
   1745   1.1       jtk 	csums[cgn] = cg->cg_cs;
   1746   1.1       jtk 	newsb->fs_cstotal.cs_nffree += cg->cg_cs.cs_nffree;
   1747   1.1       jtk 	newsb->fs_cstotal.cs_nbfree += cg->cg_cs.cs_nbfree;
   1748   1.1       jtk 	cgflags[cgn] |= CGF_DIRTY;
   1749   1.1       jtk }
   1750   1.1       jtk /*
   1751   1.1       jtk  * Recompute the cg_inosused()[] bitmap, and the cs_nifree and cs_ndir
   1752   1.1       jtk  *  values, for a cg, based on the in-core inodes for that cg.
   1753   1.1       jtk  */
   1754   1.1       jtk static void
   1755   1.1       jtk rescan_inomaps(int cgn)
   1756   1.1       jtk {
   1757   1.1       jtk 	struct cg *cg;
   1758   1.1       jtk 	int inum;
   1759   1.1       jtk 	int iwc;
   1760   1.1       jtk 
   1761   1.1       jtk 	cg = cgs[cgn];
   1762   1.1       jtk 	newsb->fs_cstotal.cs_ndir -= cg->cg_cs.cs_ndir;
   1763   1.1       jtk 	newsb->fs_cstotal.cs_nifree -= cg->cg_cs.cs_nifree;
   1764   1.1       jtk 	cg->cg_cs.cs_ndir = 0;
   1765   1.1       jtk 	cg->cg_cs.cs_nifree = 0;
   1766   1.1       jtk 	bzero(&cg_inosused(cg, 0)[0], howmany(newsb->fs_ipg, NBBY));
   1767   1.1       jtk 	inum = cgn * newsb->fs_ipg;
   1768   1.1       jtk 	if (cgn == 0) {
   1769   1.1       jtk 		set_bits(cg_inosused(cg, 0), 0, 2);
   1770   1.1       jtk 		iwc = 2;
   1771   1.1       jtk 		inum += 2;
   1772   1.1       jtk 	} else {
   1773   1.1       jtk 		iwc = 0;
   1774   1.1       jtk 	}
   1775   1.1       jtk 	for (; iwc < newsb->fs_ipg; iwc++, inum++) {
   1776   1.1       jtk 		switch (inodes[inum].di_mode & IFMT) {
   1777   1.1       jtk 		case 0:
   1778   1.1       jtk 			cg->cg_cs.cs_nifree++;
   1779   1.1       jtk 			break;
   1780   1.1       jtk 		case IFDIR:
   1781   1.1       jtk 			cg->cg_cs.cs_ndir++;
   1782   1.1       jtk 			/* fall through */
   1783   1.1       jtk 		default:
   1784   1.1       jtk 			set_bits(cg_inosused(cg, 0), iwc, 1);
   1785   1.1       jtk 			break;
   1786   1.1       jtk 		}
   1787   1.1       jtk 	}
   1788   1.1       jtk 	csums[cgn] = cg->cg_cs;
   1789   1.1       jtk 	newsb->fs_cstotal.cs_ndir += cg->cg_cs.cs_ndir;
   1790   1.1       jtk 	newsb->fs_cstotal.cs_nifree += cg->cg_cs.cs_nifree;
   1791   1.1       jtk 	cgflags[cgn] |= CGF_DIRTY;
   1792   1.1       jtk }
   1793   1.1       jtk /*
   1794   1.1       jtk  * Flush cgs to disk, recomputing anything they're marked as needing.
   1795   1.1       jtk  */
   1796   1.1       jtk static void
   1797   1.1       jtk flush_cgs(void)
   1798   1.1       jtk {
   1799   1.1       jtk 	int i;
   1800   1.1       jtk 
   1801   1.1       jtk 	for (i = 0; i < newsb->fs_ncg; i++) {
   1802   1.1       jtk 		if (cgflags[i] & CGF_BLKMAPS) {
   1803   1.1       jtk 			rescan_blkmaps(i);
   1804   1.1       jtk 		}
   1805   1.1       jtk 		if (cgflags[i] & CGF_INOMAPS) {
   1806   1.1       jtk 			rescan_inomaps(i);
   1807   1.1       jtk 		}
   1808   1.1       jtk 		if (cgflags[i] & CGF_DIRTY) {
   1809   1.1       jtk 			cgs[i]->cg_rotor = 0;
   1810   1.1       jtk 			cgs[i]->cg_frotor = 0;
   1811   1.1       jtk 			cgs[i]->cg_irotor = 0;
   1812   1.1       jtk 			writeat(fsbtodb(newsb, cgtod(newsb, i)), cgs[i],
   1813   1.1       jtk 			    cgblksz);
   1814   1.1       jtk 		}
   1815   1.1       jtk 	}
   1816   1.1       jtk 	writeat(fsbtodb(newsb, newsb->fs_csaddr), csums, newsb->fs_cssize);
   1817   1.1       jtk }
   1818   1.1       jtk /*
   1819   1.1       jtk  * Write the superblock, both to the main superblock and to each cg's
   1820   1.1       jtk  *  alternative superblock.
   1821   1.1       jtk  */
   1822   1.1       jtk static void
   1823   1.1       jtk write_sbs(void)
   1824   1.1       jtk {
   1825   1.1       jtk 	int i;
   1826   1.1       jtk 
   1827  1.20    mhitch 	if (newsb->fs_magic == FS_UFS1_MAGIC &&
   1828  1.20    mhitch 	    (newsb->fs_old_flags & FS_FLAGS_UPDATED) == 0) {
   1829  1.20    mhitch 		newsb->fs_old_time = newsb->fs_time;
   1830  1.20    mhitch 	    	newsb->fs_old_size = newsb->fs_size;
   1831  1.20    mhitch 	    	/* we don't update fs_csaddr */
   1832  1.20    mhitch 	    	newsb->fs_old_dsize = newsb->fs_dsize;
   1833  1.20    mhitch 		newsb->fs_old_cstotal.cs_ndir = newsb->fs_cstotal.cs_ndir;
   1834  1.20    mhitch 		newsb->fs_old_cstotal.cs_nbfree = newsb->fs_cstotal.cs_nbfree;
   1835  1.20    mhitch 		newsb->fs_old_cstotal.cs_nifree = newsb->fs_cstotal.cs_nifree;
   1836  1.20    mhitch 		newsb->fs_old_cstotal.cs_nffree = newsb->fs_cstotal.cs_nffree;
   1837  1.20    mhitch 		/* fill fs_old_postbl_start with 256 bytes of 0xff? */
   1838  1.20    mhitch 	}
   1839  1.10    bouyer 	writeat(where /  DEV_BSIZE, newsb, SBLOCKSIZE);
   1840   1.1       jtk 	for (i = 0; i < newsb->fs_ncg; i++) {
   1841   1.4  christos 		writeat(fsbtodb(newsb, cgsblock(newsb, i)), newsb, SBLOCKSIZE);
   1842   1.1       jtk 	}
   1843   1.1       jtk }
   1844  1.13      haad 
   1845  1.13      haad static uint32_t
   1846  1.13      haad get_dev_size(char *dev_name)
   1847  1.13      haad {
   1848  1.13      haad 	struct dkwedge_info dkw;
   1849  1.13      haad 	struct partition *pp;
   1850  1.13      haad 	struct disklabel lp;
   1851  1.13      haad 	size_t ptn;
   1852  1.13      haad 
   1853  1.13      haad 	/* Get info about partition/wedge */
   1854  1.13      haad 	if (ioctl(fd, DIOCGWEDGEINFO, &dkw) == -1) {
   1855  1.13      haad 		if (ioctl(fd, DIOCGDINFO, &lp) == -1)
   1856  1.13      haad 			return 0;
   1857  1.13      haad 
   1858  1.13      haad 		ptn = strchr(dev_name, '\0')[-1] - 'a';
   1859  1.13      haad 		if (ptn >= lp.d_npartitions)
   1860  1.13      haad 			return 0;
   1861  1.13      haad 
   1862  1.13      haad 		pp = &lp.d_partitions[ptn];
   1863  1.13      haad 		return pp->p_size;
   1864  1.13      haad 	}
   1865  1.13      haad 
   1866  1.13      haad 	return dkw.dkw_size;
   1867  1.13      haad }
   1868  1.13      haad 
   1869   1.1       jtk /*
   1870   1.1       jtk  * main().
   1871   1.1       jtk  */
   1872   1.1       jtk int
   1873  1.13      haad main(int argc, char **argv)
   1874   1.1       jtk {
   1875  1.13      haad 	int ch;
   1876  1.13      haad 	int ExpertFlag;
   1877  1.13      haad 	int SFlag;
   1878   1.4  christos 	size_t i;
   1879  1.13      haad 
   1880  1.23       riz 	char *special;
   1881  1.13      haad 	char reply[5];
   1882  1.13      haad 
   1883  1.13      haad 	newsize = 0;
   1884  1.13      haad 	ExpertFlag = 0;
   1885  1.13      haad 	SFlag = 0;
   1886  1.13      haad 
   1887  1.13      haad 	while ((ch = getopt(argc, argv, "s:y")) != -1) {
   1888  1.13      haad 		switch (ch) {
   1889  1.13      haad 		case 's':
   1890  1.13      haad 			SFlag = 1;
   1891  1.13      haad 			newsize = (size_t)strtoul(optarg, NULL, 10);
   1892  1.13      haad 			if(newsize < 1) {
   1893  1.13      haad 				usage();
   1894  1.13      haad 			}
   1895  1.13      haad 			break;
   1896  1.13      haad 		case 'y':
   1897  1.13      haad 			ExpertFlag = 1;
   1898  1.13      haad 			break;
   1899  1.13      haad 		case '?':
   1900  1.13      haad 			/* FALLTHROUGH */
   1901  1.13      haad 		default:
   1902  1.13      haad 			usage();
   1903  1.13      haad 		}
   1904  1.13      haad 	}
   1905  1.13      haad 	argc -= optind;
   1906  1.13      haad 	argv += optind;
   1907  1.13      haad 
   1908  1.13      haad 	if (argc != 1) {
   1909  1.13      haad 		usage();
   1910  1.13      haad 	}
   1911  1.13      haad 
   1912  1.23       riz 	special = *argv;
   1913  1.13      haad 
   1914  1.13      haad 	if (ExpertFlag == 0) {
   1915  1.24       wiz 		printf("It's required to manually run fsck on file system "
   1916  1.13      haad 		    "before you can resize it\n\n"
   1917  1.13      haad 		    " Did you run fsck on your disk (Yes/No) ? ");
   1918  1.13      haad 		fgets(reply, (int)sizeof(reply), stdin);
   1919  1.13      haad 		if (strcasecmp(reply, "Yes\n")) {
   1920  1.13      haad 			printf("\n Nothing done \n");
   1921  1.13      haad 			exit(EXIT_SUCCESS);
   1922  1.13      haad 		}
   1923   1.1       jtk 	}
   1924  1.13      haad 
   1925  1.23       riz 	fd = open(special, O_RDWR, 0);
   1926   1.4  christos 	if (fd < 0)
   1927  1.23       riz 		err(EXIT_FAILURE, "Can't open `%s'", special);
   1928   1.1       jtk 	checksmallio();
   1929  1.13      haad 
   1930  1.13      haad 	if (SFlag == 0) {
   1931  1.23       riz 		newsize = get_dev_size(special);
   1932  1.13      haad 		if (newsize == 0)
   1933  1.15       riz 			err(EXIT_FAILURE,
   1934  1.24       wiz 			    "Can't resize file system, newsize not known.");
   1935  1.13      haad 	}
   1936  1.13      haad 
   1937   1.1       jtk 	oldsb = (struct fs *) & sbbuf;
   1938   1.4  christos 	newsb = (struct fs *) (SBLOCKSIZE + (char *) &sbbuf);
   1939   1.4  christos 	for (where = search[i = 0]; search[i] != -1; where = search[++i]) {
   1940   1.9    bouyer 		readat(where / DEV_BSIZE, oldsb, SBLOCKSIZE);
   1941  1.23       riz 		switch (oldsb->fs_magic) {
   1942  1.23       riz 		case FS_UFS2_MAGIC:
   1943  1.23       riz 			/* FALLTHROUGH */
   1944  1.23       riz 			is_ufs2 = 1;
   1945  1.23       riz 		case FS_UFS1_MAGIC:
   1946  1.23       riz 			needswap = 0;
   1947  1.23       riz 			break;
   1948  1.23       riz 		case FS_UFS2_MAGIC_SWAPPED:
   1949  1.23       riz  			is_ufs2 = 1;
   1950  1.23       riz 			/* FALLTHROUGH */
   1951  1.23       riz 		case FS_UFS1_MAGIC_SWAPPED:
   1952  1.23       riz 			needswap = 1;
   1953  1.23       riz 			break;
   1954  1.23       riz 		default:
   1955  1.23       riz 			continue;
   1956  1.23       riz 		}
   1957  1.23       riz 		if (!is_ufs2 && where == SBLOCK_UFS2)
   1958  1.16       riz 			continue;
   1959  1.23       riz 		break;
   1960   1.1       jtk 	}
   1961   1.4  christos 	if (where == (off_t)-1)
   1962  1.13      haad 		errx(EXIT_FAILURE, "Bad magic number");
   1963  1.20    mhitch 	if (oldsb->fs_magic == FS_UFS1_MAGIC &&
   1964  1.20    mhitch 	    (oldsb->fs_old_flags & FS_FLAGS_UPDATED) == 0) {
   1965  1.20    mhitch 		oldsb->fs_csaddr = oldsb->fs_old_csaddr;
   1966  1.20    mhitch 		oldsb->fs_size = oldsb->fs_old_size;
   1967  1.20    mhitch 		oldsb->fs_dsize = oldsb->fs_old_dsize;
   1968  1.20    mhitch 		oldsb->fs_cstotal.cs_ndir = oldsb->fs_old_cstotal.cs_ndir;
   1969  1.20    mhitch 		oldsb->fs_cstotal.cs_nbfree = oldsb->fs_old_cstotal.cs_nbfree;
   1970  1.20    mhitch 		oldsb->fs_cstotal.cs_nifree = oldsb->fs_old_cstotal.cs_nifree;
   1971  1.20    mhitch 		oldsb->fs_cstotal.cs_nffree = oldsb->fs_old_cstotal.cs_nffree;
   1972  1.20    mhitch 		/* any others? */
   1973  1.20    mhitch 		printf("Resizing with ffsv1 superblock\n");
   1974  1.20    mhitch 	}
   1975  1.23       riz 	if (is_ufs2)
   1976  1.23       riz 		errx(EXIT_FAILURE, "ffsv2 file systems currently unsupported.");
   1977  1.23       riz 	if (needswap)
   1978  1.23       riz 		errx(EXIT_FAILURE, "Swapped byte order file system detected"
   1979  1.23       riz 		    " - currently unsupported.");
   1980   1.1       jtk 	oldsb->fs_qbmask = ~(int64_t) oldsb->fs_bmask;
   1981   1.1       jtk 	oldsb->fs_qfmask = ~(int64_t) oldsb->fs_fmask;
   1982   1.1       jtk 	if (oldsb->fs_ipg % INOPB(oldsb)) {
   1983  1.15       riz 		(void)fprintf(stderr, "ipg[%d] %% INOPB[%d] != 0\n",
   1984  1.15       riz 		    (int) oldsb->fs_ipg, (int) INOPB(oldsb));
   1985  1.13      haad 		exit(EXIT_FAILURE);
   1986   1.1       jtk 	}
   1987   1.1       jtk 	/* The superblock is bigger than struct fs (there are trailing tables,
   1988   1.4  christos 	 * of non-fixed size); make sure we copy the whole thing.  SBLOCKSIZE may
   1989   1.1       jtk 	 * be an over-estimate, but we do this just once, so being generous is
   1990   1.1       jtk 	 * cheap. */
   1991   1.4  christos 	bcopy(oldsb, newsb, SBLOCKSIZE);
   1992   1.1       jtk 	loadcgs();
   1993   1.1       jtk 	if (newsize > fsbtodb(oldsb, oldsb->fs_size)) {
   1994   1.1       jtk 		grow();
   1995   1.1       jtk 	} else if (newsize < fsbtodb(oldsb, oldsb->fs_size)) {
   1996   1.1       jtk 		shrink();
   1997   1.1       jtk 	}
   1998   1.1       jtk 	flush_cgs();
   1999   1.1       jtk 	write_sbs();
   2000  1.19       riz 	if (isplainfile())
   2001  1.19       riz 		ftruncate(fd,newsize * DEV_BSIZE);
   2002  1.13      haad 	return 0;
   2003  1.13      haad }
   2004  1.13      haad 
   2005  1.13      haad static void
   2006  1.13      haad usage(void)
   2007  1.13      haad {
   2008  1.13      haad 
   2009  1.23       riz 	(void)fprintf(stderr, "usage: %s [-y] [-s size] special\n", getprogname());
   2010  1.13      haad 	exit(EXIT_FAILURE);
   2011   1.1       jtk }
   2012