fsutil.c revision 201700
166864Sadrian/*
266864Sadrian * Copyright (c) 1980, 1986, 1993
366864Sadrian *	The Regents of the University of California.  All rights reserved.
466864Sadrian *
566864Sadrian * Redistribution and use in source and binary forms, with or without
666864Sadrian * modification, are permitted provided that the following conditions
766864Sadrian * are met:
866864Sadrian * 1. Redistributions of source code must retain the above copyright
966864Sadrian *    notice, this list of conditions and the following disclaimer.
1066864Sadrian * 2. Redistributions in binary form must reproduce the above copyright
1166864Sadrian *    notice, this list of conditions and the following disclaimer in the
1266864Sadrian *    documentation and/or other materials provided with the distribution.
1366864Sadrian * 4. Neither the name of the University nor the names of its contributors
1466864Sadrian *    may be used to endorse or promote products derived from this software
1566864Sadrian *    without specific prior written permission.
1666864Sadrian *
1766864Sadrian * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
1866864Sadrian * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
1966864Sadrian * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
2066864Sadrian * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
2166864Sadrian * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
2266864Sadrian * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
2366864Sadrian * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
2466864Sadrian * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
2566864Sadrian * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
2666864Sadrian * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
2766864Sadrian * SUCH DAMAGE.
2866864Sadrian */
2966864Sadrian
30114589Sobrien#if 0
3166864Sadrian#ifndef lint
3266864Sadrianstatic const char sccsid[] = "@(#)utilities.c	8.6 (Berkeley) 5/19/95";
33114589Sobrien#endif /* not lint */
3466864Sadrian#endif
35114589Sobrien#include <sys/cdefs.h>
36114589Sobrien__FBSDID("$FreeBSD: head/sbin/fsck_ffs/fsutil.c 201700 2010-01-07 00:17:36Z mckusick $");
3766864Sadrian
3866864Sadrian#include <sys/param.h>
39107671Siedowse#include <sys/time.h>
4066864Sadrian#include <sys/types.h>
4186514Siedowse#include <sys/sysctl.h>
4298542Smckusick#include <sys/disklabel.h>
4366864Sadrian#include <sys/stat.h>
4496475Sphk#include <sys/disklabel.h>
4566864Sadrian
4666864Sadrian#include <ufs/ufs/dinode.h>
4766864Sadrian#include <ufs/ufs/dir.h>
4866864Sadrian#include <ufs/ffs/fs.h>
4966864Sadrian
5066864Sadrian#include <err.h>
5166864Sadrian#include <errno.h>
5266864Sadrian#include <string.h>
5366864Sadrian#include <ctype.h>
5466864Sadrian#include <fstab.h>
55101037Smux#include <stdint.h>
5666864Sadrian#include <stdio.h>
5766864Sadrian#include <stdlib.h>
5866864Sadrian#include <unistd.h>
5966864Sadrian
6066864Sadrian#include "fsck.h"
6166864Sadrian
62107671Siedowsestatic void slowio_start(void);
63107671Siedowsestatic void slowio_end(void);
64107671Siedowse
6566864Sadrianlong	diskreads, totalreads;	/* Disk cache statistics */
66107671Siedowsestruct timeval slowio_starttime;
67107671Siedowseint slowio_delay_usec = 10000;	/* Initial IO delay for background fsck */
68107671Siedowseint slowio_pollcnt;
6966864Sadrian
7066864Sadrianint
7198542Smckusickftypeok(union dinode *dp)
7266864Sadrian{
7398542Smckusick	switch (DIP(dp, di_mode) & IFMT) {
7466864Sadrian
7566864Sadrian	case IFDIR:
7666864Sadrian	case IFREG:
7766864Sadrian	case IFBLK:
7866864Sadrian	case IFCHR:
7966864Sadrian	case IFLNK:
8066864Sadrian	case IFSOCK:
8166864Sadrian	case IFIFO:
8266864Sadrian		return (1);
8366864Sadrian
8466864Sadrian	default:
8566864Sadrian		if (debug)
8698542Smckusick			printf("bad file type 0%o\n", DIP(dp, di_mode));
8766864Sadrian		return (0);
8866864Sadrian	}
8966864Sadrian}
9066864Sadrian
9166864Sadrianint
92100935Sphkreply(const char *question)
9366864Sadrian{
9466864Sadrian	int persevere;
9566864Sadrian	char c;
9666864Sadrian
9766864Sadrian	if (preen)
9866864Sadrian		pfatal("INTERNAL ERROR: GOT TO reply()");
9966864Sadrian	persevere = !strcmp(question, "CONTINUE");
10066864Sadrian	printf("\n");
10174556Smckusick	if (!persevere && (nflag || (fswritefd < 0 && bkgrdflag == 0))) {
10266864Sadrian		printf("%s? no\n\n", question);
10366864Sadrian		resolved = 0;
10466864Sadrian		return (0);
10566864Sadrian	}
10666864Sadrian	if (yflag || (persevere && nflag)) {
10766864Sadrian		printf("%s? yes\n\n", question);
10866864Sadrian		return (1);
10966864Sadrian	}
11066864Sadrian	do	{
11166864Sadrian		printf("%s? [yn] ", question);
11266864Sadrian		(void) fflush(stdout);
11366864Sadrian		c = getc(stdin);
11466864Sadrian		while (c != '\n' && getc(stdin) != '\n') {
11566864Sadrian			if (feof(stdin)) {
11666864Sadrian				resolved = 0;
11766864Sadrian				return (0);
11866864Sadrian			}
11966864Sadrian		}
12066864Sadrian	} while (c != 'y' && c != 'Y' && c != 'n' && c != 'N');
12166864Sadrian	printf("\n");
12266864Sadrian	if (c == 'y' || c == 'Y')
12366864Sadrian		return (1);
12466864Sadrian	resolved = 0;
12566864Sadrian	return (0);
12666864Sadrian}
12766864Sadrian
12866864Sadrian/*
12966864Sadrian * Look up state information for an inode.
13066864Sadrian */
13166864Sadrianstruct inostat *
13292839Simpinoinfo(ino_t inum)
13366864Sadrian{
13466864Sadrian	static struct inostat unallocated = { USTATE, 0, 0 };
13566864Sadrian	struct inostatlist *ilp;
13666864Sadrian	int iloff;
13766864Sadrian
13866864Sadrian	if (inum > maxino)
13966864Sadrian		errx(EEXIT, "inoinfo: inumber %d out of range", inum);
14066864Sadrian	ilp = &inostathead[inum / sblock.fs_ipg];
14166864Sadrian	iloff = inum % sblock.fs_ipg;
14266864Sadrian	if (iloff >= ilp->il_numalloced)
14366864Sadrian		return (&unallocated);
14466864Sadrian	return (&ilp->il_stat[iloff]);
14566864Sadrian}
14666864Sadrian
14766864Sadrian/*
14866864Sadrian * Malloc buffers and set up cache.
14966864Sadrian */
15066864Sadrianvoid
15192839Simpbufinit(void)
15266864Sadrian{
15392806Sobrien	struct bufarea *bp;
15466864Sadrian	long bufcnt, i;
15566864Sadrian	char *bufp;
15666864Sadrian
15766864Sadrian	pbp = pdirbp = (struct bufarea *)0;
15866864Sadrian	bufp = malloc((unsigned int)sblock.fs_bsize);
15966864Sadrian	if (bufp == 0)
16066864Sadrian		errx(EEXIT, "cannot allocate buffer pool");
16166864Sadrian	cgblk.b_un.b_buf = bufp;
16266864Sadrian	initbarea(&cgblk);
16366864Sadrian	bufhead.b_next = bufhead.b_prev = &bufhead;
16466864Sadrian	bufcnt = MAXBUFSPACE / sblock.fs_bsize;
16566864Sadrian	if (bufcnt < MINBUFS)
16666864Sadrian		bufcnt = MINBUFS;
16766864Sadrian	for (i = 0; i < bufcnt; i++) {
16866864Sadrian		bp = (struct bufarea *)malloc(sizeof(struct bufarea));
16966864Sadrian		bufp = malloc((unsigned int)sblock.fs_bsize);
17066864Sadrian		if (bp == NULL || bufp == NULL) {
17166864Sadrian			if (i >= MINBUFS)
17266864Sadrian				break;
17366864Sadrian			errx(EEXIT, "cannot allocate buffer pool");
17466864Sadrian		}
17566864Sadrian		bp->b_un.b_buf = bufp;
17666864Sadrian		bp->b_prev = &bufhead;
17766864Sadrian		bp->b_next = bufhead.b_next;
17866864Sadrian		bufhead.b_next->b_prev = bp;
17966864Sadrian		bufhead.b_next = bp;
18066864Sadrian		initbarea(bp);
18166864Sadrian	}
18266864Sadrian	bufhead.b_size = i;	/* save number of buffers */
18366864Sadrian}
18466864Sadrian
18566864Sadrian/*
18666864Sadrian * Manage a cache of directory blocks.
18766864Sadrian */
18866864Sadrianstruct bufarea *
18998542Smckusickgetdatablk(ufs2_daddr_t blkno, long size)
19066864Sadrian{
19192806Sobrien	struct bufarea *bp;
19266864Sadrian
19366864Sadrian	for (bp = bufhead.b_next; bp != &bufhead; bp = bp->b_next)
19466864Sadrian		if (bp->b_bno == fsbtodb(&sblock, blkno))
19566864Sadrian			goto foundit;
19666864Sadrian	for (bp = bufhead.b_prev; bp != &bufhead; bp = bp->b_prev)
19766864Sadrian		if ((bp->b_flags & B_INUSE) == 0)
19866864Sadrian			break;
19966864Sadrian	if (bp == &bufhead)
20066864Sadrian		errx(EEXIT, "deadlocked buffer pool");
20166864Sadrian	getblk(bp, blkno, size);
20266864Sadrian	/* fall through */
20366864Sadrianfoundit:
20466864Sadrian	bp->b_prev->b_next = bp->b_next;
20566864Sadrian	bp->b_next->b_prev = bp->b_prev;
20666864Sadrian	bp->b_prev = &bufhead;
20766864Sadrian	bp->b_next = bufhead.b_next;
20866864Sadrian	bufhead.b_next->b_prev = bp;
20966864Sadrian	bufhead.b_next = bp;
21066864Sadrian	bp->b_flags |= B_INUSE;
21166864Sadrian	return (bp);
21266864Sadrian}
21366864Sadrian
21466864Sadrianvoid
21598542Smckusickgetblk(struct bufarea *bp, ufs2_daddr_t blk, long size)
21666864Sadrian{
21798542Smckusick	ufs2_daddr_t dblk;
21866864Sadrian
21995897Smckusick	totalreads++;
22066864Sadrian	dblk = fsbtodb(&sblock, blk);
22166864Sadrian	if (bp->b_bno != dblk) {
22266864Sadrian		flush(fswritefd, bp);
22366864Sadrian		diskreads++;
224163845Spjd		bp->b_errs = blread(fsreadfd, bp->b_un.b_buf, dblk, size);
22566864Sadrian		bp->b_bno = dblk;
22666864Sadrian		bp->b_size = size;
22766864Sadrian	}
22866864Sadrian}
22966864Sadrian
23066864Sadrianvoid
23192839Simpflush(int fd, struct bufarea *bp)
23266864Sadrian{
23392806Sobrien	int i, j;
23466864Sadrian
23566864Sadrian	if (!bp->b_dirty)
23666864Sadrian		return;
23774556Smckusick	bp->b_dirty = 0;
23874556Smckusick	if (fswritefd < 0) {
23974556Smckusick		pfatal("WRITING IN READ_ONLY MODE.\n");
24074556Smckusick		return;
24174556Smckusick	}
24266864Sadrian	if (bp->b_errs != 0)
243100935Sphk		pfatal("WRITING %sZERO'ED BLOCK %lld TO DISK\n",
24466864Sadrian		    (bp->b_errs == bp->b_size / dev_bsize) ? "" : "PARTIALLY ",
245100935Sphk		    (long long)bp->b_bno);
24666864Sadrian	bp->b_errs = 0;
247163845Spjd	blwrite(fd, bp->b_un.b_buf, bp->b_bno, (long)bp->b_size);
24866864Sadrian	if (bp != &sblk)
24966864Sadrian		return;
25066864Sadrian	for (i = 0, j = 0; i < sblock.fs_cssize; i += sblock.fs_bsize, j++) {
251163845Spjd		blwrite(fswritefd, (char *)sblock.fs_csp + i,
25266864Sadrian		    fsbtodb(&sblock, sblock.fs_csaddr + j * sblock.fs_frag),
25366864Sadrian		    sblock.fs_cssize - i < sblock.fs_bsize ?
25466864Sadrian		    sblock.fs_cssize - i : sblock.fs_bsize);
25566864Sadrian	}
25666864Sadrian}
25766864Sadrian
25874556Smckusickvoid
259100935Sphkrwerror(const char *mesg, ufs2_daddr_t blk)
26066864Sadrian{
26166864Sadrian
26275927Smckusick	if (bkgrdcheck)
26375927Smckusick		exit(EEXIT);
26466864Sadrian	if (preen == 0)
26566864Sadrian		printf("\n");
26686514Siedowse	pfatal("CANNOT %s: %ld", mesg, (long)blk);
26766864Sadrian	if (reply("CONTINUE") == 0)
26866864Sadrian		exit(EEXIT);
26966864Sadrian}
27066864Sadrian
27166864Sadrianvoid
27292839Simpckfini(int markclean)
27366864Sadrian{
27492806Sobrien	struct bufarea *bp, *nbp;
27566864Sadrian	int ofsmodified, cnt = 0;
27666864Sadrian
27774556Smckusick	if (bkgrdflag) {
27874556Smckusick		unlink(snapname);
27974556Smckusick		if ((!(sblock.fs_flags & FS_UNCLEAN)) != markclean) {
28074556Smckusick			cmd.value = FS_UNCLEAN;
28174556Smckusick			cmd.size = markclean ? -1 : 1;
28274556Smckusick			if (sysctlbyname("vfs.ffs.setflags", 0, 0,
28374556Smckusick			    &cmd, sizeof cmd) == -1)
284118302Sru				rwerror("SET FILE SYSTEM FLAGS", FS_UNCLEAN);
28574556Smckusick			if (!preen) {
28674556Smckusick				printf("\n***** FILE SYSTEM MARKED %s *****\n",
28774556Smckusick				    markclean ? "CLEAN" : "DIRTY");
28874556Smckusick				if (!markclean)
28974556Smckusick					rerun = 1;
29074556Smckusick			}
29174556Smckusick		} else if (!preen && !markclean) {
29274556Smckusick			printf("\n***** FILE SYSTEM STILL DIRTY *****\n");
29374556Smckusick			rerun = 1;
29474556Smckusick		}
29574556Smckusick	}
29666864Sadrian	if (fswritefd < 0) {
29766864Sadrian		(void)close(fsreadfd);
29866864Sadrian		return;
29966864Sadrian	}
30066864Sadrian	flush(fswritefd, &sblk);
30198542Smckusick	if (havesb && cursnapshot == 0 && sblock.fs_magic == FS_UFS2_MAGIC &&
302107294Smckusick	    sblk.b_bno != sblock.fs_sblockloc / dev_bsize &&
30366864Sadrian	    !preen && reply("UPDATE STANDARD SUPERBLOCK")) {
304107294Smckusick		sblk.b_bno = sblock.fs_sblockloc / dev_bsize;
30566864Sadrian		sbdirty();
30666864Sadrian		flush(fswritefd, &sblk);
30766864Sadrian	}
30866864Sadrian	flush(fswritefd, &cgblk);
30966864Sadrian	free(cgblk.b_un.b_buf);
31066864Sadrian	for (bp = bufhead.b_prev; bp && bp != &bufhead; bp = nbp) {
31166864Sadrian		cnt++;
31266864Sadrian		flush(fswritefd, bp);
31366864Sadrian		nbp = bp->b_prev;
31466864Sadrian		free(bp->b_un.b_buf);
31566864Sadrian		free((char *)bp);
31666864Sadrian	}
31766864Sadrian	if (bufhead.b_size != cnt)
31866864Sadrian		errx(EEXIT, "panic: lost %d buffers", bufhead.b_size - cnt);
31966864Sadrian	pbp = pdirbp = (struct bufarea *)0;
32074556Smckusick	if (cursnapshot == 0 && sblock.fs_clean != markclean) {
321105436Smckusick		if ((sblock.fs_clean = markclean) != 0) {
32275557Smckusick			sblock.fs_flags &= ~(FS_UNCLEAN | FS_NEEDSFSCK);
323105436Smckusick			sblock.fs_pendingblocks = 0;
324105436Smckusick			sblock.fs_pendinginodes = 0;
325105436Smckusick		}
32666864Sadrian		sbdirty();
32766864Sadrian		ofsmodified = fsmodified;
32866864Sadrian		flush(fswritefd, &sblk);
32966864Sadrian		fsmodified = ofsmodified;
33066864Sadrian		if (!preen) {
33166864Sadrian			printf("\n***** FILE SYSTEM MARKED %s *****\n",
33266864Sadrian			    markclean ? "CLEAN" : "DIRTY");
33366864Sadrian			if (!markclean)
33466864Sadrian				rerun = 1;
33566864Sadrian		}
336188110Smckusick	} else if (!preen) {
337188110Smckusick		if (markclean) {
338188110Smckusick			printf("\n***** FILE SYSTEM IS CLEAN *****\n");
339188110Smckusick		} else {
340188110Smckusick			printf("\n***** FILE SYSTEM STILL DIRTY *****\n");
341188110Smckusick			rerun = 1;
342188110Smckusick		}
34366864Sadrian	}
34475557Smckusick	if (debug && totalreads > 0)
34566864Sadrian		printf("cache missed %ld of %ld (%d%%)\n", diskreads,
34666864Sadrian		    totalreads, (int)(diskreads * 100 / totalreads));
34766864Sadrian	(void)close(fsreadfd);
34866864Sadrian	(void)close(fswritefd);
34966864Sadrian}
35066864Sadrian
35166864Sadrianint
352163845Spjdblread(int fd, char *buf, ufs2_daddr_t blk, long size)
35366864Sadrian{
35466864Sadrian	char *cp;
35566864Sadrian	int i, errs;
35666864Sadrian	off_t offset;
35766864Sadrian
35866864Sadrian	offset = blk;
35966864Sadrian	offset *= dev_bsize;
360107671Siedowse	if (bkgrdflag)
361107671Siedowse		slowio_start();
36266864Sadrian	if (lseek(fd, offset, 0) < 0)
36374556Smckusick		rwerror("SEEK BLK", blk);
364107671Siedowse	else if (read(fd, buf, (int)size) == size) {
365107671Siedowse		if (bkgrdflag)
366107671Siedowse			slowio_end();
36766864Sadrian		return (0);
368107671Siedowse	}
36974556Smckusick	rwerror("READ BLK", blk);
37066864Sadrian	if (lseek(fd, offset, 0) < 0)
37174556Smckusick		rwerror("SEEK BLK", blk);
37266864Sadrian	errs = 0;
37366864Sadrian	memset(buf, 0, (size_t)size);
37466864Sadrian	printf("THE FOLLOWING DISK SECTORS COULD NOT BE READ:");
37566864Sadrian	for (cp = buf, i = 0; i < size; i += secsize, cp += secsize) {
37666864Sadrian		if (read(fd, cp, (int)secsize) != secsize) {
37766864Sadrian			(void)lseek(fd, offset + i + secsize, 0);
37866864Sadrian			if (secsize != dev_bsize && dev_bsize != 1)
379101037Smux				printf(" %jd (%jd),",
380101037Smux				    (intmax_t)(blk * dev_bsize + i) / secsize,
381101037Smux				    (intmax_t)blk + i / dev_bsize);
38266864Sadrian			else
383101037Smux				printf(" %jd,", (intmax_t)blk + i / dev_bsize);
38466864Sadrian			errs++;
38566864Sadrian		}
38666864Sadrian	}
38766864Sadrian	printf("\n");
38866864Sadrian	if (errs)
38966864Sadrian		resolved = 0;
39066864Sadrian	return (errs);
39166864Sadrian}
39266864Sadrian
39366864Sadrianvoid
394163845Spjdblwrite(int fd, char *buf, ufs2_daddr_t blk, long size)
39566864Sadrian{
39666864Sadrian	int i;
39766864Sadrian	char *cp;
39866864Sadrian	off_t offset;
39966864Sadrian
40066864Sadrian	if (fd < 0)
40166864Sadrian		return;
40266864Sadrian	offset = blk;
40366864Sadrian	offset *= dev_bsize;
40466864Sadrian	if (lseek(fd, offset, 0) < 0)
40574556Smckusick		rwerror("SEEK BLK", blk);
40666864Sadrian	else if (write(fd, buf, (int)size) == size) {
40766864Sadrian		fsmodified = 1;
40866864Sadrian		return;
40966864Sadrian	}
41066864Sadrian	resolved = 0;
41174556Smckusick	rwerror("WRITE BLK", blk);
41266864Sadrian	if (lseek(fd, offset, 0) < 0)
41374556Smckusick		rwerror("SEEK BLK", blk);
41466864Sadrian	printf("THE FOLLOWING SECTORS COULD NOT BE WRITTEN:");
41566864Sadrian	for (cp = buf, i = 0; i < size; i += dev_bsize, cp += dev_bsize)
41666864Sadrian		if (write(fd, cp, (int)dev_bsize) != dev_bsize) {
41766864Sadrian			(void)lseek(fd, offset + i + dev_bsize, 0);
418101037Smux			printf(" %jd,", (intmax_t)blk + i / dev_bsize);
41966864Sadrian		}
42066864Sadrian	printf("\n");
42166864Sadrian	return;
42266864Sadrian}
42366864Sadrian
42466864Sadrian/*
425188110Smckusick * Verify cylinder group's magic number and other parameters.  If the
426188110Smckusick * test fails, offer an option to rebuild the whole cylinder group.
427178088Sdelphij */
428188110Smckusickint
429178088Sdelphijcheck_cgmagic(int cg, struct cg *cgp)
430178088Sdelphij{
431178088Sdelphij
432188110Smckusick	/*
433188110Smckusick	 * Extended cylinder group checks.
434188110Smckusick	 */
435188110Smckusick	if (cg_chkmagic(cgp) &&
436188110Smckusick	    ((sblock.fs_magic == FS_UFS1_MAGIC &&
437188110Smckusick	      cgp->cg_old_niblk == sblock.fs_ipg &&
438188110Smckusick	      cgp->cg_ndblk <= sblock.fs_fpg &&
439201700Smckusick	      cgp->cg_old_ncyl <= sblock.fs_old_cpg) ||
440188110Smckusick	     (sblock.fs_magic == FS_UFS2_MAGIC &&
441188110Smckusick	      cgp->cg_niblk == sblock.fs_ipg &&
442188110Smckusick	      cgp->cg_ndblk <= sblock.fs_fpg &&
443188110Smckusick	      cgp->cg_initediblk <= sblock.fs_ipg))) {
444188110Smckusick		return (1);
445178088Sdelphij	}
446188110Smckusick	pfatal("CYLINDER GROUP %d: BAD MAGIC NUMBER", cg);
447188110Smckusick	if (!reply("REBUILD CYLINDER GROUP")) {
448188110Smckusick		printf("YOU WILL NEED TO RERUN FSCK.\n");
449188110Smckusick		rerun = 1;
450188110Smckusick		return (1);
451188110Smckusick	}
452188110Smckusick	/*
453188110Smckusick	 * Zero out the cylinder group and then initialize critical fields.
454188110Smckusick	 * Bit maps and summaries will be recalculated by later passes.
455188110Smckusick	 */
456188110Smckusick	memset(cgp, 0, (size_t)sblock.fs_cgsize);
457188110Smckusick	cgp->cg_magic = CG_MAGIC;
458188110Smckusick	cgp->cg_cgx = cg;
459188110Smckusick	cgp->cg_niblk = sblock.fs_ipg;
460188110Smckusick	cgp->cg_initediblk = sblock.fs_ipg < 2 * INOPB(&sblock) ?
461188110Smckusick	    sblock.fs_ipg : 2 * INOPB(&sblock);
462188110Smckusick	if (cgbase(&sblock, cg) + sblock.fs_fpg < sblock.fs_size)
463188110Smckusick		cgp->cg_ndblk = sblock.fs_fpg;
464188110Smckusick	else
465188110Smckusick		cgp->cg_ndblk = sblock.fs_size - cgbase(&sblock, cg);
466188110Smckusick	cgp->cg_iusedoff = &cgp->cg_space[0] - (u_char *)(&cgp->cg_firstfield);
467188110Smckusick	if (sblock.fs_magic == FS_UFS1_MAGIC) {
468188110Smckusick		cgp->cg_niblk = 0;
469188110Smckusick		cgp->cg_initediblk = 0;
470188110Smckusick		cgp->cg_old_ncyl = sblock.fs_old_cpg;
471188110Smckusick		cgp->cg_old_niblk = sblock.fs_ipg;
472188110Smckusick		cgp->cg_old_btotoff = cgp->cg_iusedoff;
473188110Smckusick		cgp->cg_old_boff = cgp->cg_old_btotoff +
474188110Smckusick		    sblock.fs_old_cpg * sizeof(int32_t);
475188110Smckusick		cgp->cg_iusedoff = cgp->cg_old_boff +
476188110Smckusick		    sblock.fs_old_cpg * sizeof(u_int16_t);
477188110Smckusick	}
478188110Smckusick	cgp->cg_freeoff = cgp->cg_iusedoff + howmany(sblock.fs_ipg, CHAR_BIT);
479188110Smckusick	cgp->cg_nextfreeoff = cgp->cg_freeoff + howmany(sblock.fs_fpg,CHAR_BIT);
480188110Smckusick	if (sblock.fs_contigsumsize > 0) {
481188110Smckusick		cgp->cg_nclusterblks = cgp->cg_ndblk / sblock.fs_frag;
482188110Smckusick		cgp->cg_clustersumoff =
483188110Smckusick		    roundup(cgp->cg_nextfreeoff, sizeof(u_int32_t));
484188110Smckusick		cgp->cg_clustersumoff -= sizeof(u_int32_t);
485188110Smckusick		cgp->cg_clusteroff = cgp->cg_clustersumoff +
486188110Smckusick		    (sblock.fs_contigsumsize + 1) * sizeof(u_int32_t);
487188110Smckusick		cgp->cg_nextfreeoff = cgp->cg_clusteroff +
488188110Smckusick		    howmany(fragstoblks(&sblock, sblock.fs_fpg), CHAR_BIT);
489188110Smckusick	}
490188110Smckusick	cgdirty();
491188110Smckusick	return (0);
492178088Sdelphij}
493178088Sdelphij
494178088Sdelphij/*
49566864Sadrian * allocate a data block with the specified number of fragments
49666864Sadrian */
49798542Smckusickufs2_daddr_t
49892839Simpallocblk(long frags)
49966864Sadrian{
50066864Sadrian	int i, j, k, cg, baseblk;
50166864Sadrian	struct cg *cgp = &cgrp;
50266864Sadrian
50366864Sadrian	if (frags <= 0 || frags > sblock.fs_frag)
50466864Sadrian		return (0);
50566864Sadrian	for (i = 0; i < maxfsblock - sblock.fs_frag; i += sblock.fs_frag) {
50666864Sadrian		for (j = 0; j <= sblock.fs_frag - frags; j++) {
50766864Sadrian			if (testbmap(i + j))
50866864Sadrian				continue;
50966864Sadrian			for (k = 1; k < frags; k++)
51066864Sadrian				if (testbmap(i + j + k))
51166864Sadrian					break;
51266864Sadrian			if (k < frags) {
51366864Sadrian				j += k;
51466864Sadrian				continue;
51566864Sadrian			}
51666864Sadrian			cg = dtog(&sblock, i + j);
51766864Sadrian			getblk(&cgblk, cgtod(&sblock, cg), sblock.fs_cgsize);
518188110Smckusick			if (!check_cgmagic(cg, cgp))
519188110Smckusick				return (0);
52066864Sadrian			baseblk = dtogd(&sblock, i + j);
52166864Sadrian			for (k = 0; k < frags; k++) {
52266864Sadrian				setbmap(i + j + k);
52366864Sadrian				clrbit(cg_blksfree(cgp), baseblk + k);
52466864Sadrian			}
52566864Sadrian			n_blks += frags;
52666864Sadrian			if (frags == sblock.fs_frag)
52766864Sadrian				cgp->cg_cs.cs_nbfree--;
52866864Sadrian			else
52966864Sadrian				cgp->cg_cs.cs_nffree -= frags;
53066864Sadrian			cgdirty();
53166864Sadrian			return (i + j);
53266864Sadrian		}
53366864Sadrian	}
53466864Sadrian	return (0);
53566864Sadrian}
53666864Sadrian
53766864Sadrian/*
53866864Sadrian * Free a previously allocated block
53966864Sadrian */
54066864Sadrianvoid
54198542Smckusickfreeblk(ufs2_daddr_t blkno, long frags)
54266864Sadrian{
54366864Sadrian	struct inodesc idesc;
54466864Sadrian
54566864Sadrian	idesc.id_blkno = blkno;
54666864Sadrian	idesc.id_numfrags = frags;
54766864Sadrian	(void)pass4check(&idesc);
54866864Sadrian}
54966864Sadrian
550107671Siedowse/* Slow down IO so as to leave some disk bandwidth for other processes */
551107671Siedowsevoid
552107671Siedowseslowio_start()
553107671Siedowse{
554107671Siedowse
555129401Sscottl	/* Delay one in every 8 operations */
556107671Siedowse	slowio_pollcnt = (slowio_pollcnt + 1) & 7;
557107671Siedowse	if (slowio_pollcnt == 0) {
558107671Siedowse		gettimeofday(&slowio_starttime, NULL);
559107671Siedowse	}
560107671Siedowse}
561107671Siedowse
562107671Siedowsevoid
563107671Siedowseslowio_end()
564107671Siedowse{
565107671Siedowse	struct timeval tv;
566107671Siedowse	int delay_usec;
567107671Siedowse
568107671Siedowse	if (slowio_pollcnt != 0)
569107671Siedowse		return;
570107671Siedowse
571107671Siedowse	/* Update the slowdown interval. */
572107671Siedowse	gettimeofday(&tv, NULL);
573107671Siedowse	delay_usec = (tv.tv_sec - slowio_starttime.tv_sec) * 1000000 +
574107671Siedowse	    (tv.tv_usec - slowio_starttime.tv_usec);
575107671Siedowse	if (delay_usec < 64)
576107671Siedowse		delay_usec = 64;
577129401Sscottl	if (delay_usec > 2500000)
578129401Sscottl		delay_usec = 2500000;
579107671Siedowse	slowio_delay_usec = (slowio_delay_usec * 63 + delay_usec) >> 6;
580129401Sscottl	/* delay by 8 times the average IO delay */
581129401Sscottl	if (slowio_delay_usec > 64)
582129401Sscottl		usleep(slowio_delay_usec * 8);
583107671Siedowse}
584107671Siedowse
58566864Sadrian/*
58666864Sadrian * Find a pathname
58766864Sadrian */
58866864Sadrianvoid
58992839Simpgetpathname(char *namebuf, ino_t curdir, ino_t ino)
59066864Sadrian{
59166864Sadrian	int len;
59292806Sobrien	char *cp;
59366864Sadrian	struct inodesc idesc;
59466864Sadrian	static int busy = 0;
59566864Sadrian
59666864Sadrian	if (curdir == ino && ino == ROOTINO) {
59766864Sadrian		(void)strcpy(namebuf, "/");
59866864Sadrian		return;
59966864Sadrian	}
600136281Struckman	if (busy || !INO_IS_DVALID(curdir)) {
60166864Sadrian		(void)strcpy(namebuf, "?");
60266864Sadrian		return;
60366864Sadrian	}
60466864Sadrian	busy = 1;
60566864Sadrian	memset(&idesc, 0, sizeof(struct inodesc));
60666864Sadrian	idesc.id_type = DATA;
60766864Sadrian	idesc.id_fix = IGNORE;
60866864Sadrian	cp = &namebuf[MAXPATHLEN - 1];
60966864Sadrian	*cp = '\0';
61066864Sadrian	if (curdir != ino) {
61166864Sadrian		idesc.id_parent = curdir;
61266864Sadrian		goto namelookup;
61366864Sadrian	}
61466864Sadrian	while (ino != ROOTINO) {
61566864Sadrian		idesc.id_number = ino;
61666864Sadrian		idesc.id_func = findino;
617100935Sphk		idesc.id_name = strdup("..");
61866864Sadrian		if ((ckinode(ginode(ino), &idesc) & FOUND) == 0)
61966864Sadrian			break;
62066864Sadrian	namelookup:
62166864Sadrian		idesc.id_number = idesc.id_parent;
62266864Sadrian		idesc.id_parent = ino;
62366864Sadrian		idesc.id_func = findname;
62466864Sadrian		idesc.id_name = namebuf;
62566864Sadrian		if ((ckinode(ginode(idesc.id_number), &idesc)&FOUND) == 0)
62666864Sadrian			break;
62766864Sadrian		len = strlen(namebuf);
62866864Sadrian		cp -= len;
62966864Sadrian		memmove(cp, namebuf, (size_t)len);
63066864Sadrian		*--cp = '/';
63166864Sadrian		if (cp < &namebuf[MAXNAMLEN])
63266864Sadrian			break;
63366864Sadrian		ino = idesc.id_number;
63466864Sadrian	}
63566864Sadrian	busy = 0;
63666864Sadrian	if (ino != ROOTINO)
63766864Sadrian		*--cp = '?';
63866864Sadrian	memmove(namebuf, cp, (size_t)(&namebuf[MAXPATHLEN] - cp));
63966864Sadrian}
64066864Sadrian
64166864Sadrianvoid
642100935Sphkcatch(int sig __unused)
64366864Sadrian{
64496483Sphk
64596483Sphk	ckfini(0);
64666864Sadrian	exit(12);
64766864Sadrian}
64866864Sadrian
64966864Sadrian/*
65066864Sadrian * When preening, allow a single quit to signal
651102231Strhodes * a special exit after file system checks complete
65266864Sadrian * so that reboot sequence may be interrupted.
65366864Sadrian */
65466864Sadrianvoid
655100935Sphkcatchquit(int sig __unused)
65666864Sadrian{
657102231Strhodes	printf("returning to single-user after file system check\n");
65866864Sadrian	returntosingle = 1;
65966864Sadrian	(void)signal(SIGQUIT, SIG_DFL);
66066864Sadrian}
66166864Sadrian
66266864Sadrian/*
66366864Sadrian * determine whether an inode should be fixed.
66466864Sadrian */
66566864Sadrianint
666100935Sphkdofix(struct inodesc *idesc, const char *msg)
66766864Sadrian{
66866864Sadrian
66966864Sadrian	switch (idesc->id_fix) {
67066864Sadrian
67166864Sadrian	case DONTKNOW:
67266864Sadrian		if (idesc->id_type == DATA)
67366864Sadrian			direrror(idesc->id_number, msg);
67466864Sadrian		else
67581911Skris			pwarn("%s", msg);
67666864Sadrian		if (preen) {
67766864Sadrian			printf(" (SALVAGED)\n");
67866864Sadrian			idesc->id_fix = FIX;
67966864Sadrian			return (ALTERED);
68066864Sadrian		}
68166864Sadrian		if (reply("SALVAGE") == 0) {
68266864Sadrian			idesc->id_fix = NOFIX;
68366864Sadrian			return (0);
68466864Sadrian		}
68566864Sadrian		idesc->id_fix = FIX;
68666864Sadrian		return (ALTERED);
68766864Sadrian
68866864Sadrian	case FIX:
68966864Sadrian		return (ALTERED);
69066864Sadrian
69166864Sadrian	case NOFIX:
69266864Sadrian	case IGNORE:
69366864Sadrian		return (0);
69466864Sadrian
69566864Sadrian	default:
69666864Sadrian		errx(EEXIT, "UNKNOWN INODESC FIX MODE %d", idesc->id_fix);
69766864Sadrian	}
69866864Sadrian	/* NOTREACHED */
69966864Sadrian	return (0);
70066864Sadrian}
70166864Sadrian
70266864Sadrian#include <stdarg.h>
70366864Sadrian
70466864Sadrian/*
70566864Sadrian * An unexpected inconsistency occured.
706102231Strhodes * Die if preening or file system is running with soft dependency protocol,
70766864Sadrian * otherwise just print message and continue.
70866864Sadrian */
70966864Sadrianvoid
71066864Sadrianpfatal(const char *fmt, ...)
71166864Sadrian{
71266864Sadrian	va_list ap;
71366864Sadrian	va_start(ap, fmt);
71466864Sadrian	if (!preen) {
71575927Smckusick		(void)vfprintf(stdout, fmt, ap);
71666864Sadrian		va_end(ap);
71766864Sadrian		if (usedsoftdep)
71875927Smckusick			(void)fprintf(stdout,
71966864Sadrian			    "\nUNEXPECTED SOFT UPDATE INCONSISTENCY\n");
72075557Smckusick		/*
72175557Smckusick		 * Force foreground fsck to clean up inconsistency.
72275557Smckusick		 */
72375557Smckusick		if (bkgrdflag) {
72475557Smckusick			cmd.value = FS_NEEDSFSCK;
72575557Smckusick			cmd.size = 1;
72675557Smckusick			if (sysctlbyname("vfs.ffs.setflags", 0, 0,
72775557Smckusick			    &cmd, sizeof cmd) == -1)
72875557Smckusick				pwarn("CANNOT SET FS_NEEDSFSCK FLAG\n");
72975927Smckusick			fprintf(stdout, "CANNOT RUN IN BACKGROUND\n");
73075557Smckusick			ckfini(0);
73175557Smckusick			exit(EEXIT);
73275557Smckusick		}
73366864Sadrian		return;
73466864Sadrian	}
73566864Sadrian	if (cdevname == NULL)
736100935Sphk		cdevname = strdup("fsck");
73775927Smckusick	(void)fprintf(stdout, "%s: ", cdevname);
73875927Smckusick	(void)vfprintf(stdout, fmt, ap);
73975927Smckusick	(void)fprintf(stdout,
74066864Sadrian	    "\n%s: UNEXPECTED%sINCONSISTENCY; RUN fsck MANUALLY.\n",
74166864Sadrian	    cdevname, usedsoftdep ? " SOFT UPDATE " : " ");
74275557Smckusick	/*
74375557Smckusick	 * Force foreground fsck to clean up inconsistency.
74475557Smckusick	 */
74575557Smckusick	if (bkgrdflag) {
74675557Smckusick		cmd.value = FS_NEEDSFSCK;
74775557Smckusick		cmd.size = 1;
74875557Smckusick		if (sysctlbyname("vfs.ffs.setflags", 0, 0,
74975557Smckusick		    &cmd, sizeof cmd) == -1)
75075557Smckusick			pwarn("CANNOT SET FS_NEEDSFSCK FLAG\n");
75175557Smckusick	}
75266864Sadrian	ckfini(0);
75366864Sadrian	exit(EEXIT);
75466864Sadrian}
75566864Sadrian
75666864Sadrian/*
75766864Sadrian * Pwarn just prints a message when not preening or running soft dependency
75866864Sadrian * protocol, or a warning (preceded by filename) when preening.
75966864Sadrian */
76066864Sadrianvoid
76166864Sadrianpwarn(const char *fmt, ...)
76266864Sadrian{
76366864Sadrian	va_list ap;
76466864Sadrian	va_start(ap, fmt);
76566864Sadrian	if (preen)
76675927Smckusick		(void)fprintf(stdout, "%s: ", cdevname);
76775927Smckusick	(void)vfprintf(stdout, fmt, ap);
76866864Sadrian	va_end(ap);
76966864Sadrian}
77066864Sadrian
77166864Sadrian/*
77266864Sadrian * Stub for routines from kernel.
77366864Sadrian */
77466864Sadrianvoid
77566864Sadrianpanic(const char *fmt, ...)
77666864Sadrian{
77766864Sadrian	va_list ap;
77866864Sadrian	va_start(ap, fmt);
77966864Sadrian	pfatal("INTERNAL INCONSISTENCY:");
78075927Smckusick	(void)vfprintf(stdout, fmt, ap);
78166864Sadrian	va_end(ap);
78266864Sadrian	exit(EEXIT);
78366864Sadrian}
784