/*      $NetBSD: fsirand.c,v 1.33 2022/11/17 06:40:39 chs 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.33 2022/11/17 06:40:39 chs 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>

__dead 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;
static 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:
                case FS_UFS2EA_MAGIC:
                        is_ufs2 = 1;
                        /* FALLTHROUGH */
                case FS_UFS1_MAGIC:
                        break;
                case FS_UFS2_MAGIC_SWAPPED:
                case FS_UFS2EA_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 = FFS_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) FFS_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
 */
static void
statussig(int dummy)
{
        char    msgbuf[256];
        int     len, deltat;
        time_t  tnow;

        (void)time(&tnow);
        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;
}