/*      $OpenBSD: fsirand.c,v 1.9 1997/02/28 00:46:33 millert Exp $     */

/*
 * Copyright (c) 1997 Todd C. Miller <Todd.Miller@courtesan.com>
 * All rights reserved.
 *
 * 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.
 * 3. All advertising materials mentioning features or use of this software
 *    must display the following acknowledgement:
 *      This product includes software developed by Todd C. Miller.
 * 4. The name of the author may not be used to endorse or promote products
 *    derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED ``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 AUTHOR 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.
 *
 * $FreeBSD: src/sbin/fsirand/fsirand.c,v 1.7.2.1 2000/07/01 06:23:36 ps Exp $
 */

#include <sys/diskslice.h>
#include <sys/param.h>
#include <sys/time.h>
#include <sys/resource.h>

#include <vfs/ufs/dinode.h>
#include <vfs/ufs/fs.h>

#include <err.h>
#include <errno.h>
#include <fcntl.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>

static void usage(void);
int fsirand(char *);

int printonly = 0, force = 0, ignorelabel = 0;

int
main(int argc, char **argv)
{
        int n, ex = 0;
        struct rlimit rl;

        while ((n = getopt(argc, argv, "bfp")) != -1) {
                switch (n) {
                case 'b':
                        ignorelabel++;
                        break;
                case 'p':
                        printonly++;
                        break;
                case 'f':
                        force++;
                        break;
                default:
                        usage();
                }
        }
        if (argc - optind < 1)
                usage();

        srandomdev();

        /* Increase our data size to the max */
        if (getrlimit(RLIMIT_DATA, &rl) == 0) {
                rl.rlim_cur = rl.rlim_max;
                if (setrlimit(RLIMIT_DATA, &rl) < 0)
                        warn("can't get resource limit to max data size");
        } else
                warn("can't get resource limit for data size");

        for (n = optind; n < argc; n++) {
                if (argc - optind != 1)
                        puts(argv[n]);
                ex += fsirand(argv[n]);
                if (n < argc - 1)
                        putchar('\n');
        }

        exit(ex);
}

int
fsirand(char *device)
{
        static struct ufs1_dinode *inodebuf;
        static ssize_t oldibufsize = 0;
        ssize_t ibufsize;
        struct fs *sblock;
        ino_t inumber;
        daddr_t dblk;
        char sbuf[SBSIZE], sbuftmp[SBSIZE];
        int devfd, n, cg;
        u_int32_t bsize = DEV_BSIZE;

        if ((devfd = open(device, printonly ? O_RDONLY : O_RDWR)) < 0) {
                warn("can't open %s", device);
                return (1);
        }

        /* Get block size (usually 512) from partinfo if possible */
        if (!ignorelabel) {
                struct partinfo pinfo;

                if (ioctl(devfd, DIOCGPART, &pinfo) < 0) {
                        fprintf(stderr,
                             "can't read partition info, assuming sector "
                             "size of %d\n", bsize);
                } else {
                        bsize = pinfo.media_blksize;
                }
        }

        /* Read in master superblock */
        memset(&sbuf, 0, sizeof(sbuf));
        sblock = (struct fs *)&sbuf;
        if (lseek(devfd, SBOFF, SEEK_SET) == -1) {
                warn("can't seek to superblock (%jd) on %s",
                    (intmax_t)SBOFF, device);
                return (1);
        }
        if ((n = read(devfd, (void *)sblock, SBSIZE)) != SBSIZE) {
                warnx("can't read superblock on %s: %s", device,
                    (n < SBSIZE) ? "short read" : strerror(errno));
                return (1);
        }

        /* Simple sanity checks on the superblock */
        if (sblock->fs_magic != FS_MAGIC) {
                warnx("bad magic number in superblock");
                return (1);
        }
        if (sblock->fs_sbsize > SBSIZE) {
                warnx("superblock size is preposterous");
                return (1);
        }
        if (sblock->fs_postblformat == FS_42POSTBLFMT) {
                warnx("filesystem format is too old, sorry");
                return (1);
        }
        if (!force && !printonly && sblock->fs_clean != 1) {
                warnx("filesystem is not clean, fsck %s first", device);
                return (1);
        }

        /* Make sure backup superblocks are sane. */
        sblock = (struct fs *)&sbuftmp;
        for (cg = 0; cg < sblock->fs_ncg; cg++) {
                dblk = fsbtodb(sblock, cgsblock(sblock, cg));
                if (lseek(devfd, (off_t)dblk * bsize, SEEK_SET) < 0) {
                        warn("can't seek to %jd", (intmax_t)dblk * bsize);
                        return (1);
                } else if ((n = write(devfd, (void *)sblock, SBSIZE)) != SBSIZE) {
                        warn("can't read backup superblock %d on %s: %s",
                            cg + 1, device, (n < SBSIZE) ? "short write"
                            : strerror(errno));
                        return (1);
                }
                if (sblock->fs_magic != FS_MAGIC) {
                        warnx("bad magic number in backup superblock %d on %s",
                            cg + 1, device);
                        return (1);
                }
                if (sblock->fs_sbsize > SBSIZE) {
                        warnx("size of backup superblock %d on %s is preposterous",
                            cg + 1, device);
                        return (1);
                }
        }
        sblock = (struct fs *)&sbuf;

        /* XXX - should really cap buffer at 512kb or so */
        ibufsize = sizeof(struct ufs1_dinode) * sblock->fs_ipg;
        if (oldibufsize < ibufsize) {
                if ((inodebuf = realloc(inodebuf, ibufsize)) == NULL)
                        errx(1, "can't allocate memory for inode buffer");
                oldibufsize = ibufsize;
        }

        if (printonly && (sblock->fs_id[0] || sblock->fs_id[1])) {
                if (sblock->fs_inodefmt >= FS_44INODEFMT && sblock->fs_id[0])
                        printf("%s was randomized on %s", device,
                            ctime((const time_t *)&(sblock->fs_id[0])));
                printf("fsid: %x %x\n", sblock->fs_id[0],
                    sblock->fs_id[1]);
        }

        /* Randomize fs_id unless old 4.2BSD filesystem */
        if ((sblock->fs_inodefmt >= FS_44INODEFMT) && !printonly) {
                /* Randomize fs_id and write out new sblock and backups */
                sblock->fs_id[0] = (u_int32_t)time(NULL);
                sblock->fs_id[1] = random();

                if (lseek(devfd, SBOFF, SEEK_SET) == -1) {
                        warn("can't seek to superblock (%jd) on %s",
                             (intmax_t)SBOFF, device);
                        return (1);
                }
                if ((n = write(devfd, (void *)sblock, SBSIZE)) != SBSIZE) {
                        warn("can't read superblock on %s: %s", device,
                            (n < SBSIZE) ? "short write" : strerror(errno));
                        return (1);
                }
        }

        /* For each cylinder group, randomize inodes and update backup sblock */
        for (cg = 0, inumber = 0; cg < sblock->fs_ncg; cg++) {
                /* Update superblock if appropriate */
                if ((sblock->fs_inodefmt >= FS_44INODEFMT) && !printonly) {
                        dblk = fsbtodb(sblock, cgsblock(sblock, cg));
                        if (lseek(devfd, (off_t)dblk * bsize, SEEK_SET) < 0) {
                                warn("can't seek to %jd",
                                     (intmax_t)dblk * bsize);
                                return (1);
                        } else if ((n = write(devfd, (void *)sblock, SBSIZE)) != SBSIZE) {
                                warn("can't read backup superblock %d on %s: %s",
                                    cg + 1, device, (n < SBSIZE) ? "short write"
                                    : strerror(errno));
                                return (1);
                        }
                }

                /* Read in inodes, then print or randomize generation nums */
                dblk = fsbtodb(sblock, ino_to_fsba(sblock, inumber));
                if (lseek(devfd, (off_t)dblk * bsize, SEEK_SET) < 0) {
                        warn("can't seek to %jd", (intmax_t)dblk * bsize);
                        return (1);
                } else if ((n = read(devfd, inodebuf, ibufsize)) != ibufsize) {
                        warnx("can't read inodes: %s",
                             (n < ibufsize) ? "short read" : strerror(errno));
                        return (1);
                }

                for (n = 0; n < sblock->fs_ipg; n++, inumber++) {
                        if (inumber >= UFS_ROOTINO) {
                                if (printonly) {
                                        printf("ino %jd gen %x\n",
                                               (intmax_t)inumber,
                                               inodebuf[n].di_gen);
                                } else {
                                        inodebuf[n].di_gen = random();
                                }
                        }
                }

                /* Write out modified inodes */
                if (!printonly) {
                        if (lseek(devfd, (off_t)dblk * bsize, SEEK_SET) < 0) {
                                warn("can't seek to %jd",
                                    (intmax_t)dblk * bsize);
                                return (1);
                        } else if ((n = write(devfd, inodebuf, ibufsize)) !=
                                 ibufsize) {
                                warnx("can't write inodes: %s",
                                     (n != ibufsize) ? "short write" :
                                     strerror(errno));
                                return (1);
                        }
                }
        }
        close(devfd);

        return(0);
}

static void
usage(void)
{
        fprintf(stderr,
            "usage: fsirand [-b] [-f] [-p] special [special ...]\n");
        exit(1);
}