292 lines
6.8 KiB
C
292 lines
6.8 KiB
C
/* $NetBSD: fsirand.c,v 1.28 2008/04/28 20:23:08 martin Exp $ */
|
|
|
|
/*-
|
|
* Copyright (c) 1997 The NetBSD Foundation, Inc.
|
|
* All rights reserved.
|
|
*
|
|
* This code is derived from software contributed to The NetBSD Foundation
|
|
* by Christos Zoulas.
|
|
*
|
|
* Redistribution and use in source and binary forms, with or without
|
|
* modification, are permitted provided that the following conditions
|
|
* are met:
|
|
* 1. Redistributions of source code must retain the above copyright
|
|
* notice, this list of conditions and the following disclaimer.
|
|
* 2. Redistributions in binary form must reproduce the above copyright
|
|
* notice, this list of conditions and the following disclaimer in the
|
|
* documentation and/or other materials provided with the distribution.
|
|
*
|
|
* THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
|
|
* ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
|
|
* TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
|
|
* PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
|
|
* BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
|
* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
|
* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
|
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
|
* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
|
* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
|
* POSSIBILITY OF SUCH DAMAGE.
|
|
*/
|
|
|
|
#include <sys/cdefs.h>
|
|
#ifndef lint
|
|
__RCSID("$NetBSD: fsirand.c,v 1.28 2008/04/28 20:23:08 martin Exp $");
|
|
#endif /* lint */
|
|
|
|
#include <sys/param.h>
|
|
#include <sys/time.h>
|
|
#include <sys/vnode.h>
|
|
#include <sys/disklabel.h>
|
|
#include <sys/ioctl.h>
|
|
|
|
#include <ctype.h>
|
|
#include <err.h>
|
|
#include <errno.h>
|
|
#include <fcntl.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <unistd.h>
|
|
#include <util.h>
|
|
#include <signal.h>
|
|
|
|
#include <ufs/ufs/ufs_bswap.h>
|
|
|
|
#include <ufs/ufs/dinode.h>
|
|
#include <ufs/ffs/fs.h>
|
|
#include <ufs/ffs/ffs_extern.h>
|
|
|
|
static void usage(void);
|
|
static void getsblock(int, const char *, struct fs *);
|
|
static void fixinodes(int, struct fs *, struct disklabel *, int, long);
|
|
static void statussig(int);
|
|
|
|
int needswap, ino, imax, is_ufs2;
|
|
time_t tstart;
|
|
|
|
static void
|
|
usage(void)
|
|
{
|
|
|
|
(void) fprintf(stderr,
|
|
"usage: %s [-F] [-p] [-x <constant>] <special>\n",
|
|
getprogname());
|
|
exit(1);
|
|
}
|
|
|
|
|
|
static const off_t sblock_try[] = SBLOCKSEARCH;
|
|
|
|
/*
|
|
* getsblock():
|
|
* Return the superblock
|
|
*/
|
|
static void
|
|
getsblock(int fd, const char *name, struct fs *fs)
|
|
{
|
|
int i;
|
|
|
|
for (i = 0; ; i++) {
|
|
if (sblock_try[i] == -1)
|
|
errx(1, "%s: can't find superblock", name);
|
|
if (pread(fd, fs, SBLOCKSIZE, sblock_try[i]) != SBLOCKSIZE)
|
|
continue;
|
|
|
|
switch(fs->fs_magic) {
|
|
case FS_UFS2_MAGIC:
|
|
is_ufs2 = 1;
|
|
/* FALLTHROUGH */
|
|
case FS_UFS1_MAGIC:
|
|
break;
|
|
case FS_UFS2_MAGIC_SWAPPED:
|
|
is_ufs2 = 1;
|
|
/* FALLTHROUGH */
|
|
case FS_UFS1_MAGIC_SWAPPED:
|
|
needswap = 1;
|
|
ffs_sb_swap(fs, fs);
|
|
break;
|
|
default:
|
|
continue;
|
|
}
|
|
|
|
if (!is_ufs2 && sblock_try[i] == SBLOCK_UFS2)
|
|
continue;
|
|
break;
|
|
}
|
|
|
|
if (fs->fs_ncg < 1)
|
|
errx(1, "%s: bad ncg in superblock", name);
|
|
|
|
if (fs->fs_sbsize > SBLOCKSIZE)
|
|
errx(1, "%s: superblock too large", name);
|
|
}
|
|
|
|
|
|
/*
|
|
* fixinodes():
|
|
* Randomize the inode generation numbers
|
|
*/
|
|
static void
|
|
fixinodes(int fd, struct fs *fs, struct disklabel *lab, int pflag, long xorval)
|
|
{
|
|
int inopb = INOPB(fs);
|
|
int size;
|
|
caddr_t buf;
|
|
struct ufs1_dinode *dp1 = NULL;
|
|
struct ufs2_dinode *dp2 = NULL;
|
|
int i;
|
|
|
|
size = is_ufs2 ? inopb * sizeof (struct ufs2_dinode) :
|
|
inopb * sizeof (struct ufs1_dinode);
|
|
|
|
if ((buf = malloc(size)) == NULL)
|
|
err(1, "Out of memory");
|
|
|
|
if (is_ufs2)
|
|
dp2 = (struct ufs2_dinode *)buf;
|
|
else
|
|
dp1 = (struct ufs1_dinode *)buf;
|
|
|
|
for (ino = 0, imax = fs->fs_ipg * fs->fs_ncg; ino < imax;) {
|
|
off_t sp;
|
|
sp = (off_t) fsbtodb(fs, ino_to_fsba(fs, ino)) *
|
|
(off_t) lab->d_secsize;
|
|
|
|
if (lseek(fd, sp, SEEK_SET) == (off_t) -1)
|
|
err(1, "Seeking to inode %d failed", ino);
|
|
|
|
if (read(fd, buf, size) != size)
|
|
err(1, "Reading inodes %d+%d failed", ino, inopb);
|
|
|
|
for (i = 0; i < inopb; i++) {
|
|
if (is_ufs2) {
|
|
if (pflag)
|
|
printf("inode %10d gen 0x%08x\n",
|
|
ino,
|
|
ufs_rw32(dp2[i].di_gen, needswap));
|
|
else
|
|
dp2[i].di_gen =
|
|
ufs_rw32((arc4random() & INT32_MAX)^ xorval,
|
|
needswap);
|
|
} else {
|
|
if (pflag)
|
|
printf("inode %10d gen 0x%08x\n",
|
|
ino,
|
|
ufs_rw32(dp1[i].di_gen, needswap));
|
|
else
|
|
dp1[i].di_gen =
|
|
ufs_rw32((arc4random() & INT32_MAX) ^ xorval,
|
|
needswap);
|
|
}
|
|
if (++ino > imax)
|
|
errx(1, "Exceeded number of inodes");
|
|
}
|
|
|
|
if (pflag)
|
|
continue;
|
|
|
|
if (lseek(fd, sp, SEEK_SET) == (off_t) -1)
|
|
err(1, "Seeking to inode %d failed", ino);
|
|
|
|
if (write(fd, buf, size) != size)
|
|
err(1, "Writing inodes %d+%d failed", ino, inopb);
|
|
}
|
|
free(buf);
|
|
}
|
|
|
|
/*
|
|
* statussig():
|
|
* display current status
|
|
*/
|
|
void
|
|
statussig(int dummy)
|
|
{
|
|
char msgbuf[256];
|
|
int len, deltat;
|
|
time_t tnow, elapsed;
|
|
|
|
(void)time(&tnow);
|
|
elapsed = tnow - tstart;
|
|
len = snprintf(msgbuf, sizeof(msgbuf),
|
|
"fsirand: completed inode %d of %d (%3.2f%%)",
|
|
ino, imax, (ino * 100.0) / imax);
|
|
if (imax - ino) {
|
|
deltat = tstart - tnow + (1.0 * (tnow - tstart)) / ino * imax;
|
|
len += snprintf(msgbuf + len, sizeof(msgbuf) - len,
|
|
", finished in %d:%02d\n", deltat / 60, deltat % 60);
|
|
} else {
|
|
len += snprintf(msgbuf + len, sizeof(msgbuf) - len, "\n");
|
|
}
|
|
write(STDERR_FILENO, msgbuf, len);
|
|
}
|
|
|
|
int
|
|
main(int argc, char *argv[])
|
|
{
|
|
const char *special;
|
|
char buf[SBLOCKSIZE], device[MAXPATHLEN];
|
|
struct fs *fs = (struct fs *) buf;
|
|
struct disklabel lab;
|
|
long xorval;
|
|
char *ep;
|
|
int fd, c, Fflag, pflag, openflags;
|
|
|
|
xorval = 0;
|
|
Fflag = pflag = 0;
|
|
|
|
while ((c = getopt(argc, argv, "Fpx:")) != -1)
|
|
switch (c) {
|
|
case 'F':
|
|
Fflag++;
|
|
break;
|
|
case 'p':
|
|
pflag++;
|
|
break;
|
|
case 'x':
|
|
errno = 0;
|
|
xorval = strtol(optarg, &ep, 0);
|
|
if ((xorval == LONG_MIN || xorval == LONG_MAX) &&
|
|
errno == ERANGE)
|
|
err(1, "Out of range constant");
|
|
if (*ep)
|
|
errx(1, "Bad constant");
|
|
break;
|
|
default:
|
|
usage();
|
|
}
|
|
|
|
argv += optind;
|
|
argc -= optind;
|
|
|
|
if (argc != 1)
|
|
usage();
|
|
|
|
special = argv[0];
|
|
openflags = pflag ? O_RDONLY : O_RDWR;
|
|
if (Fflag)
|
|
fd = open(special, openflags);
|
|
else {
|
|
fd = opendisk(special, openflags, device, sizeof(device), 0);
|
|
special = device;
|
|
}
|
|
if (fd == -1)
|
|
err(1, "Cannot open `%s'", special);
|
|
|
|
if (Fflag) {
|
|
memset(&lab, 0, sizeof(lab));
|
|
lab.d_secsize = DEV_BSIZE; /* XXX */
|
|
} else {
|
|
if (ioctl(fd, DIOCGDINFO, &lab) == -1)
|
|
err(1, "%s: cannot get disklabel information", special);
|
|
}
|
|
|
|
time(&tstart);
|
|
(void)signal(SIGINFO, statussig);
|
|
getsblock(fd, special, fs);
|
|
fixinodes(fd, fs, &lab, pflag, xorval);
|
|
|
|
(void) close(fd);
|
|
return 0;
|
|
}
|