fsirand.c revision 217769 
1/*	$OpenBSD: fsirand.c,v 1.9 1997/02/28 00:46:33 millert Exp $	*/
2
3/*
4 * Copyright (c) 1997 Todd C. Miller <Todd.Miller@courtesan.com>
5 * All rights reserved.
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
9 * are met:
10 * 1. Redistributions of source code must retain the above copyright
11 *    notice, this list of conditions and the following disclaimer.
12 * 2. Redistributions in binary form must reproduce the above copyright
13 *    notice, this list of conditions and the following disclaimer in the
14 *    documentation and/or other materials provided with the distribution.
15 * 3. All advertising materials mentioning features or use of this software
16 *    must display the following acknowledgement:
17 *	This product includes software developed by Todd C. Miller.
18 * 4. The name of the author may not be used to endorse or promote products
19 *    derived from this software without specific prior written permission.
20 *
21 * THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES,
22 * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY
23 * AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL
24 * THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
25 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
26 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
27 * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
28 * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
29 * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
30 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
31 */
32
33#ifndef lint
34static const char rcsid[] =
35  "$FreeBSD: head/sbin/fsirand/fsirand.c 217769 2011-01-24 06:17:05Z mckusick $";
36#endif /* not lint */
37
38#include <sys/param.h>
39#include <sys/disklabel.h>
40#include <sys/resource.h>
41
42#include <ufs/ufs/dinode.h>
43#include <ufs/ffs/fs.h>
44
45#include <err.h>
46#include <errno.h>
47#include <fcntl.h>
48#include <stdio.h>
49#include <stdint.h>
50#include <stdlib.h>
51#include <string.h>
52#include <time.h>
53#include <unistd.h>
54
55static void usage(void) __dead2;
56int fsirand(char *);
57
58/*
59 * Possible superblock locations ordered from most to least likely.
60 */
61static int sblock_try[] = SBLOCKSEARCH;
62
63int printonly = 0, force = 0, ignorelabel = 0;
64
65int
66main(int argc, char *argv[])
67{
68	int n, ex = 0;
69	struct rlimit rl;
70
71	while ((n = getopt(argc, argv, "bfp")) != -1) {
72		switch (n) {
73		case 'b':
74			ignorelabel++;
75			break;
76		case 'p':
77			printonly++;
78			break;
79		case 'f':
80			force++;
81			break;
82		default:
83			usage();
84		}
85	}
86	if (argc - optind < 1)
87		usage();
88
89	srandomdev();
90
91	/* Increase our data size to the max */
92	if (getrlimit(RLIMIT_DATA, &rl) == 0) {
93		rl.rlim_cur = rl.rlim_max;
94		if (setrlimit(RLIMIT_DATA, &rl) < 0)
95			warn("can't get resource limit to max data size");
96	} else
97		warn("can't get resource limit for data size");
98
99	for (n = optind; n < argc; n++) {
100		if (argc - optind != 1)
101			(void)puts(argv[n]);
102		ex += fsirand(argv[n]);
103		if (n < argc - 1)
104			putchar('\n');
105	}
106
107	exit(ex);
108}
109
110int
111fsirand(char *device)
112{
113	struct ufs1_dinode *dp1;
114	struct ufs2_dinode *dp2;
115	caddr_t inodebuf;
116	ssize_t ibufsize;
117	struct fs *sblock;
118	ino_t inumber, maxino;
119	ufs2_daddr_t sblockloc, dblk;
120	char sbuf[SBLOCKSIZE], sbuftmp[SBLOCKSIZE];
121	int i, devfd, n, cg;
122	u_int32_t bsize = DEV_BSIZE;
123	struct disklabel label;
124
125	if ((devfd = open(device, printonly ? O_RDONLY : O_RDWR)) < 0) {
126		warn("can't open %s", device);
127		return (1);
128	}
129
130	/* Get block size (usually 512) from disklabel if possible */
131	if (!ignorelabel) {
132		if (ioctl(devfd, DIOCGDINFO, &label) < 0)
133			warn("can't read disklabel, using sector size of %d",
134			    bsize);
135		else
136			bsize = label.d_secsize;
137	}
138
139	dp1 = NULL;
140	dp2 = NULL;
141
142	/* Read in master superblock */
143	(void)memset(&sbuf, 0, sizeof(sbuf));
144	sblock = (struct fs *)&sbuf;
145	for (i = 0; sblock_try[i] != -1; i++) {
146		sblockloc = sblock_try[i];
147		if (lseek(devfd, sblockloc, SEEK_SET) == -1) {
148			warn("can't seek to superblock (%jd) on %s",
149			    (intmax_t)sblockloc, device);
150			return (1);
151		}
152		if ((n = read(devfd, (void *)sblock, SBLOCKSIZE))!=SBLOCKSIZE) {
153			warnx("can't read superblock on %s: %s", device,
154			    (n < SBLOCKSIZE) ? "short read" : strerror(errno));
155			return (1);
156		}
157		if ((sblock->fs_magic == FS_UFS1_MAGIC ||
158		     (sblock->fs_magic == FS_UFS2_MAGIC &&
159		      sblock->fs_sblockloc == sblock_try[i])) &&
160		    sblock->fs_bsize <= MAXBSIZE &&
161		    sblock->fs_bsize >= (ssize_t)sizeof(struct fs))
162			break;
163	}
164	if (sblock_try[i] == -1) {
165		fprintf(stderr, "Cannot find file system superblock\n");
166		return (1);
167	}
168	maxino = sblock->fs_ncg * sblock->fs_ipg;
169
170	if (sblock->fs_magic == FS_UFS1_MAGIC &&
171	    sblock->fs_old_inodefmt < FS_44INODEFMT) {
172		warnx("file system format is too old, sorry");
173		return (1);
174	}
175	if (!force && !printonly && sblock->fs_clean != 1) {
176		warnx("file system is not clean, fsck %s first", device);
177		return (1);
178	}
179
180	/* Make sure backup superblocks are sane. */
181	sblock = (struct fs *)&sbuftmp;
182	for (cg = 0; cg < (int)sblock->fs_ncg; cg++) {
183		dblk = fsbtodb(sblock, cgsblock(sblock, cg));
184		if (lseek(devfd, (off_t)dblk * bsize, SEEK_SET) < 0) {
185			warn("can't seek to %jd", (intmax_t)dblk * bsize);
186			return (1);
187		} else if ((n = write(devfd, (void *)sblock, SBLOCKSIZE)) != SBLOCKSIZE) {
188			warn("can't read backup superblock %d on %s: %s",
189			    cg + 1, device, (n < SBLOCKSIZE) ? "short write"
190			    : strerror(errno));
191			return (1);
192		}
193		if (sblock->fs_magic != FS_UFS1_MAGIC &&
194		    sblock->fs_magic != FS_UFS2_MAGIC) {
195			warnx("bad magic number in backup superblock %d on %s",
196			    cg + 1, device);
197			return (1);
198		}
199		if (sblock->fs_sbsize > SBLOCKSIZE) {
200			warnx("size of backup superblock %d on %s is preposterous",
201			    cg + 1, device);
202			return (1);
203		}
204	}
205	sblock = (struct fs *)&sbuf;
206
207	/* XXX - should really cap buffer at 512kb or so */
208	if (sblock->fs_magic == FS_UFS1_MAGIC)
209		ibufsize = sizeof(struct ufs1_dinode) * sblock->fs_ipg;
210	else
211		ibufsize = sizeof(struct ufs2_dinode) * sblock->fs_ipg;
212	if ((inodebuf = malloc(ibufsize)) == NULL)
213		errx(1, "can't allocate memory for inode buffer");
214
215	if (printonly && (sblock->fs_id[0] || sblock->fs_id[1])) {
216		if (sblock->fs_id[0])
217			(void)printf("%s was randomized on %s", device,
218			    ctime((void *)&(sblock->fs_id[0])));
219		(void)printf("fsid: %x %x\n", sblock->fs_id[0],
220			    sblock->fs_id[1]);
221	}
222
223	/* Randomize fs_id unless old 4.2BSD file system */
224	if (!printonly) {
225		/* Randomize fs_id and write out new sblock and backups */
226		sblock->fs_id[0] = (u_int32_t)time(NULL);
227		sblock->fs_id[1] = random();
228
229		if (lseek(devfd, sblockloc, SEEK_SET) == -1) {
230			warn("can't seek to superblock (%jd) on %s",
231			    (intmax_t)sblockloc, device);
232			return (1);
233		}
234		if ((n = write(devfd, (void *)sblock, SBLOCKSIZE)) !=
235		    SBLOCKSIZE) {
236			warn("can't write superblock on %s: %s", device,
237			    (n < SBLOCKSIZE) ? "short write" : strerror(errno));
238			return (1);
239		}
240	}
241
242	/* For each cylinder group, randomize inodes and update backup sblock */
243	for (cg = 0, inumber = 0; cg < (int)sblock->fs_ncg; cg++) {
244		/* Update superblock if appropriate */
245		if (!printonly) {
246			dblk = fsbtodb(sblock, cgsblock(sblock, cg));
247			if (lseek(devfd, (off_t)dblk * bsize, SEEK_SET) < 0) {
248				warn("can't seek to %jd",
249				    (intmax_t)dblk * bsize);
250				return (1);
251			} else if ((n = write(devfd, (void *)sblock,
252			    SBLOCKSIZE)) != SBLOCKSIZE) {
253			      warn("can't write backup superblock %d on %s: %s",
254				    cg + 1, device, (n < SBLOCKSIZE) ?
255				    "short write" : strerror(errno));
256				return (1);
257			}
258		}
259
260		/* Read in inodes, then print or randomize generation nums */
261		dblk = fsbtodb(sblock, ino_to_fsba(sblock, inumber));
262		if (lseek(devfd, (off_t)dblk * bsize, SEEK_SET) < 0) {
263			warn("can't seek to %jd", (intmax_t)dblk * bsize);
264			return (1);
265		} else if ((n = read(devfd, inodebuf, ibufsize)) != ibufsize) {
266			warnx("can't read inodes: %s",
267			     (n < ibufsize) ? "short read" : strerror(errno));
268			return (1);
269		}
270
271		for (n = 0; n < (int)sblock->fs_ipg; n++, inumber++) {
272			if (sblock->fs_magic == FS_UFS1_MAGIC)
273				dp1 = &((struct ufs1_dinode *)inodebuf)[n];
274			else
275				dp2 = &((struct ufs2_dinode *)inodebuf)[n];
276			if (inumber >= ROOTINO) {
277				if (printonly)
278					(void)printf("ino %d gen %08x\n",
279					    inumber,
280					    sblock->fs_magic == FS_UFS1_MAGIC ?
281					    dp1->di_gen : dp2->di_gen);
282				else if (sblock->fs_magic == FS_UFS1_MAGIC)
283					dp1->di_gen = random();
284				else
285					dp2->di_gen = random();
286			}
287		}
288
289		/* Write out modified inodes */
290		if (!printonly) {
291			if (lseek(devfd, (off_t)dblk * bsize, SEEK_SET) < 0) {
292				warn("can't seek to %jd",
293				    (intmax_t)dblk * bsize);
294				return (1);
295			} else if ((n = write(devfd, inodebuf, ibufsize)) !=
296				 ibufsize) {
297				warnx("can't write inodes: %s",
298				     (n != ibufsize) ? "short write" :
299				     strerror(errno));
300				return (1);
301			}
302		}
303	}
304	(void)close(devfd);
305
306	return(0);
307}
308
309static void
310usage(void)
311{
312	(void)fprintf(stderr,
313		"usage: fsirand [-b] [-f] [-p] special [special ...]\n");
314	exit(1);
315}
316