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