xref: /csrg-svn/sbin/fsck/setup.c (revision 30408)
1 /*
2  * Copyright (c) 1980 Regents of the University of California.
3  * All rights reserved.  The Berkeley software License Agreement
4  * specifies the terms and conditions for redistribution.
5  */
6 
7 #ifndef lint
8 static char sccsid[] = "@(#)setup.c	5.6 (Berkeley) 01/17/87";
9 #endif not lint
10 
11 #define DKTYPENAMES
12 #include <sys/param.h>
13 #include <sys/inode.h>
14 #include <sys/fs.h>
15 #include <sys/stat.h>
16 #include <sys/ioctl.h>
17 #include <sys/disklabel.h>
18 #include <sys/file.h>
19 #include <ctype.h>
20 #include "fsck.h"
21 
22 #define POWEROF2(num)	(((num) & ((num) - 1)) == 0)
23 
24 char	*calloc();
25 char	*index();
26 
27 setup(dev)
28 	char *dev;
29 {
30 	dev_t rootdev;
31 	long cg, ncg, size, i, j;
32 	struct stat statb;
33 	struct fs proto;
34 
35 	if (stat("/", &statb) < 0)
36 		errexit("Can't stat root\n");
37 	rootdev = statb.st_dev;
38 	if (stat(dev, &statb) < 0) {
39 		printf("Can't stat %s\n", dev);
40 		return (0);
41 	}
42 	rawflg = 0;
43 	if ((statb.st_mode & S_IFMT) == S_IFBLK)
44 		;
45 	else if ((statb.st_mode & S_IFMT) == S_IFCHR)
46 		rawflg++;
47 	else {
48 		if (reply("file is not a block or character device; OK") == 0)
49 			return (0);
50 	}
51 	if (rootdev == statb.st_rdev)
52 		hotroot++;
53 	if ((dfile.rfdes = open(dev, O_RDONLY)) < 0) {
54 		printf("Can't open %s\n", dev);
55 		return (0);
56 	}
57 	if (preen == 0)
58 		printf("** %s", dev);
59 	if (nflag || (dfile.wfdes = open(dev, O_WRONLY)) < 0) {
60 		dfile.wfdes = -1;
61 		if (preen)
62 			pfatal("NO WRITE ACCESS");
63 		printf(" (NO WRITE)");
64 	}
65 	if (preen == 0)
66 		printf("\n");
67 	dfile.mod = 0;
68 	lfdir = 0;
69 	initbarea(&sblk);
70 	initbarea(&fileblk);
71 	initbarea(&inoblk);
72 	initbarea(&cgblk);
73 	/*
74 	 * Read in the superblock, looking for alternates if necessary
75 	 */
76 	if (readsb(1) == 0) {
77 		if (bflag || calcsb(dev, dfile.rfdes, &proto) == 0 || preen)
78 			return(0);
79 		if (reply("LOOK FOR ALTERNATE SUPERBLOCKS") == 0)
80 			return (0);
81 		for (cg = 0; cg < proto.fs_ncg; cg++) {
82 			bflag = fsbtodb(&proto, cgsblock(&proto, cg));
83 			if (readsb(0) != 0)
84 				break;
85 		}
86 		if (cg >= proto.fs_ncg) {
87 			printf("%s %s\n%s %s\n%s %s\n",
88 				"SEARCH FOR ALTERNATE SUPER-BLOCK",
89 				"FAILED. YOU MUST USE THE",
90 				"-b OPTION TO FSCK TO SPECIFY THE",
91 				"LOCATION OF AN ALTERNATE",
92 				"SUPER-BLOCK TO SUPPLY NEEDED",
93 				"INFORMATION; SEE fsck(8).");
94 			return(0);
95 		}
96 		pwarn("USING ALTERNATE SUPERBLOCK AT %d\n", bflag);
97 	}
98 	fmax = sblock.fs_size;
99 	imax = sblock.fs_ncg * sblock.fs_ipg;
100 	/*
101 	 * Check and potentially fix certain fields in the super block.
102 	 */
103 	if (sblock.fs_optim != FS_OPTTIME && sblock.fs_optim != FS_OPTSPACE) {
104 		pfatal("UNDEFINED OPTIMIZATION IN SUPERBLOCK");
105 		if (reply("SET TO DEFAULT") == 1) {
106 			sblock.fs_optim = FS_OPTTIME;
107 			sbdirty();
108 		}
109 	}
110 	if ((sblock.fs_minfree < 0 || sblock.fs_minfree > 99)) {
111 		pfatal("IMPOSSIBLE MINFREE=%d IN SUPERBLOCK",
112 			sblock.fs_minfree);
113 		if (reply("SET TO DEFAULT") == 1) {
114 			sblock.fs_minfree = 10;
115 			sbdirty();
116 		}
117 	}
118 	if (sblock.fs_interleave < 1) {
119 		pfatal("IMPOSSIBLE INTERLEAVE=%d IN SUPERBLOCK",
120 			sblock.fs_interleave);
121 		if (reply("SET TO DEFAULT") == 1) {
122 			sblock.fs_interleave = 1;
123 			sbdirty();
124 		}
125 	}
126 	if (sblock.fs_npsect < sblock.fs_nsect) {
127 		pfatal("IMPOSSIBLE NPSECT=%d IN SUPERBLOCK",
128 			sblock.fs_npsect);
129 		if (reply("SET TO DEFAULT") == 1) {
130 			sblock.fs_npsect = sblock.fs_nsect;
131 			sbdirty();
132 		}
133 	}
134 	/*
135 	 * read in the summary info.
136 	 */
137 	for (i = 0, j = 0; i < sblock.fs_cssize; i += sblock.fs_bsize, j++) {
138 		size = sblock.fs_cssize - i < sblock.fs_bsize ?
139 		    sblock.fs_cssize - i : sblock.fs_bsize;
140 		sblock.fs_csp[j] = (struct csum *)calloc(1, (unsigned)size);
141 		if (bread(&dfile, (char *)sblock.fs_csp[j],
142 		    fsbtodb(&sblock, sblock.fs_csaddr + j * sblock.fs_frag),
143 		    size) != 0)
144 			return (0);
145 	}
146 	/*
147 	 * allocate and initialize the necessary maps
148 	 */
149 	bmapsz = roundup(howmany(fmax, NBBY), sizeof(short));
150 	blockmap = calloc((unsigned)bmapsz, sizeof (char));
151 	if (blockmap == NULL) {
152 		printf("cannot alloc %d bytes for blockmap\n", bmapsz);
153 		goto badsb;
154 	}
155 	statemap = calloc((unsigned)(imax + 1), sizeof(char));
156 	if (statemap == NULL) {
157 		printf("cannot alloc %d bytes for statemap\n", imax + 1);
158 		goto badsb;
159 	}
160 	lncntp = (short *)calloc((unsigned)(imax + 1), sizeof(short));
161 	if (lncntp == NULL) {
162 		printf("cannot alloc %d bytes for lncntp\n",
163 		    (imax + 1) * sizeof(short));
164 		goto badsb;
165 	}
166 
167 	return (1);
168 
169 badsb:
170 	ckfini();
171 	return (0);
172 }
173 
174 /*
175  * Read in the super block and its summary info.
176  */
177 readsb(listerr)
178 	int listerr;
179 {
180 	BUFAREA asblk;
181 #	define altsblock asblk.b_un.b_fs
182 	daddr_t super = bflag ? bflag : SBLOCK;
183 
184 	initbarea(&asblk);
185 	if (bread(&dfile, (char *)&sblock, super, (long)SBSIZE) != 0)
186 		return (0);
187 	sblk.b_bno = super;
188 	sblk.b_size = SBSIZE;
189 	/*
190 	 * run a few consistency checks of the super block
191 	 */
192 	if (sblock.fs_magic != FS_MAGIC)
193 		{ badsb(listerr, "MAGIC NUMBER WRONG"); return (0); }
194 	if (sblock.fs_ncg < 1)
195 		{ badsb(listerr, "NCG OUT OF RANGE"); return (0); }
196 	if (sblock.fs_cpg < 1 || sblock.fs_cpg > MAXCPG)
197 		{ badsb(listerr, "CPG OUT OF RANGE"); return (0); }
198 	if (sblock.fs_ncg * sblock.fs_cpg < sblock.fs_ncyl ||
199 	    (sblock.fs_ncg - 1) * sblock.fs_cpg >= sblock.fs_ncyl)
200 		{ badsb(listerr, "NCYL LESS THAN NCG*CPG"); return (0); }
201 	if (sblock.fs_sbsize > SBSIZE)
202 		{ badsb(listerr, "SIZE PREPOSTEROUSLY LARGE"); return (0); }
203 	/*
204 	 * Set all possible fields that could differ, then do check
205 	 * of whole super block against an alternate super block.
206 	 * When an alternate super-block is specified this check is skipped.
207 	 */
208 	if (bflag)
209 		return (1);
210 	getblk(&asblk, cgsblock(&sblock, sblock.fs_ncg - 1), sblock.fs_sbsize);
211 	if (asblk.b_errs != NULL)
212 		return (0);
213 	altsblock.fs_link = sblock.fs_link;
214 	altsblock.fs_rlink = sblock.fs_rlink;
215 	altsblock.fs_time = sblock.fs_time;
216 	altsblock.fs_cstotal = sblock.fs_cstotal;
217 	altsblock.fs_cgrotor = sblock.fs_cgrotor;
218 	altsblock.fs_fmod = sblock.fs_fmod;
219 	altsblock.fs_clean = sblock.fs_clean;
220 	altsblock.fs_ronly = sblock.fs_ronly;
221 	altsblock.fs_flags = sblock.fs_flags;
222 	altsblock.fs_maxcontig = sblock.fs_maxcontig;
223 	altsblock.fs_minfree = sblock.fs_minfree;
224 	altsblock.fs_optim = sblock.fs_optim;
225 	altsblock.fs_rotdelay = sblock.fs_rotdelay;
226 	altsblock.fs_maxbpg = sblock.fs_maxbpg;
227 	altsblock.fs_npsect = sblock.fs_npsect;
228 	altsblock.fs_interleave = sblock.fs_interleave;
229 	bcopy((char *)sblock.fs_csp, (char *)altsblock.fs_csp,
230 		sizeof sblock.fs_csp);
231 	bcopy((char *)sblock.fs_fsmnt, (char *)altsblock.fs_fsmnt,
232 		sizeof sblock.fs_fsmnt);
233 	if (bcmp((char *)&sblock, (char *)&altsblock, (int)sblock.fs_sbsize))
234 		{ badsb(listerr, "TRASHED VALUES IN SUPER BLOCK"); return (0); }
235 	return (1);
236 #	undef altsblock
237 }
238 
239 badsb(listerr, s)
240 	int listerr;
241 	char *s;
242 {
243 
244 	if (!listerr)
245 		return;
246 	if (preen)
247 		printf("%s: ", devname);
248 	pfatal("BAD SUPER BLOCK: %s\n", s);
249 }
250 
251 /*
252  * Calculate a prototype superblock based on information in the disk label.
253  * When done the cgsblock macro can be calculated and the fs_ncg field
254  * can be used. Do NOT attempt to use other macros without verifying that
255  * their needed information is available!
256  */
257 calcsb(dev, devfd, fs)
258 	char *dev;
259 	int devfd;
260 	register struct fs *fs;
261 {
262 	register struct disklabel *lp;
263 	register struct partition *pp;
264 	register char *cp;
265 	struct disklabel *getdisklabel();
266 	int i;
267 
268 	cp = index(dev, '\0') - 1;
269 	if (cp == (char *)-1 || (*cp < 'a' || *cp > 'h') && !isdigit(*cp)) {
270 		pfatal("%s: CANNOT FIGURE OUT FILE SYSTEM PARTITION\n", dev);
271 		return (0);
272 	}
273 	lp = getdisklabel(dev, devfd);
274 	if (isdigit(*cp))
275 		pp = &lp->d_partitions[0];
276 	else
277 		pp = &lp->d_partitions[*cp - 'a'];
278 	if (pp->p_fstype != FS_BSDFFS) {
279 		pfatal("%s: NOT FORMATTED AS A BSD FILE SYSTEM (%s)\n",
280 			dev, pp->p_fstype < FSMAXTYPES ?
281 			fstypenames[pp->p_fstype] : "unknown");
282 		return (0);
283 	}
284 	bzero(fs, sizeof(struct fs));
285 	fs->fs_fsize = pp->p_fsize;
286 	fs->fs_frag = pp->p_frag;
287 	fs->fs_cpg = pp->p_cpg;
288 	fs->fs_size = pp->p_size;
289 	fs->fs_ntrak = lp->d_ntracks;
290 	fs->fs_nsect = lp->d_nsectors;
291 	fs->fs_spc = lp->d_secpercyl;
292 	fs->fs_nspf = fs->fs_fsize / lp->d_secsize;
293 	fs->fs_sblkno = roundup(
294 		howmany(lp->d_bbsize + lp->d_sbsize, fs->fs_fsize),
295 		fs->fs_frag);
296 	fs->fs_cgmask = 0xffffffff;
297 	for (i = fs->fs_ntrak; i > 1; i >>= 1)
298 		fs->fs_cgmask <<= 1;
299 	if (!POWEROF2(fs->fs_ntrak))
300 		fs->fs_cgmask <<= 1;
301 	fs->fs_cgoffset = roundup(
302 		howmany(fs->fs_nsect, NSPF(fs)), fs->fs_frag);
303 	fs->fs_fpg = (fs->fs_cpg * fs->fs_spc) / NSPF(fs);
304 	fs->fs_ncg = howmany(fs->fs_size / fs->fs_spc, fs->fs_cpg);
305 	for (fs->fs_fsbtodb = 0, i = NSPF(fs); i > 1; i >>= 1)
306 		fs->fs_fsbtodb++;
307 	return (1);
308 }
309 
310 #ifdef byioctl
311 struct disklabel *
312 getdisklabel(s, fd)
313 	char *s;
314 	int	fd;
315 {
316 	static struct disklabel lab;
317 
318 	if (ioctl(fd, DIOCGDINFO, (char *)&lab) < 0) {
319 		perror("ioctl (GDINFO)");
320 		fatal("%s: can't read disk label", s);
321 	}
322 	return (&lab);
323 }
324 #else byioctl
325 char specname[64];
326 char boot[BBSIZE];
327 
328 struct disklabel *
329 getdisklabel(s, fd)
330 	char *s;
331 	int	fd;
332 {
333 	char *cp;
334 	u_long magic = htonl(DISKMAGIC);
335 	register struct disklabel *lp;
336 	int cfd;
337 
338 	/*
339 	 * Make name for 'c' partition.
340 	 */
341 	strcpy(specname, s);
342 	cp = specname + strlen(specname) - 1;
343 	if (!isdigit(*cp))
344 		*cp = 'c';
345 	cfd = open(specname, O_RDONLY);
346 	if (cfd < 0) {
347 		perror(specname);
348 		exit(2);
349 	}
350 
351 	if (read(cfd, boot, BBSIZE) < BBSIZE) {
352 		perror(specname);
353 		exit(2);
354 	}
355 	close(cfd);
356 	for (lp = (struct disklabel *)(boot + LABELOFFSET);
357 	    lp <= (struct disklabel *)(boot + BBSIZE -
358 	    sizeof(struct disklabel));
359 	    lp = (struct disklabel *)((char *)lp + 128))
360 		if (lp->d_magic == magic && lp->d_magic2 == magic)
361 			break;
362 	if (lp > (struct disklabel *)(boot + BBSIZE -
363 	    sizeof(struct disklabel)) ||
364 	    lp->d_magic != magic || lp->d_magic2 != magic ||
365 	    dkcksum(lp) != 0) {
366 		printf("Bad pack magic number %s\n",
367 			"(label is damaged, or pack is unlabeled)");
368 		exit(1);
369 	}
370 #if ENDIAN != BIG
371 	swablabel(lp);
372 #endif
373 	return (lp);
374 }
375 #endif byioctl
376