/*      $OpenBSD: scan_ffs.c,v 1.24 2024/05/09 08:35:40 florian Exp $   */

/*
 * Copyright (c) 1998 Niklas Hallqvist, Tobias Weingartner
 * 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.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``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.
 */

#include <sys/types.h>
#include <ufs/ffs/fs.h>

#include <err.h>
#include <fcntl.h>
#include <limits.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <time.h>
#include <unistd.h>
#include <util.h>

#define SBCOUNT 64              /* XXX - Should be configurable */

/* Flags to control ourselves... */
#define FLAG_VERBOSE            1
#define FLAG_SMART              2
#define FLAG_LABELS             4

static void usage(void);

static void
print_info(int flags, struct fs *sb, long long at, char* lastmount)
{
        if (flags & FLAG_LABELS ) {
                printf("X: %lld %lld 4.2BSD %d %d %d # %s\n",
                    ((off_t)sb->fs_ffs1_size * sb->fs_fsize / 512), at,
                    sb->fs_fsize, sb->fs_bsize, sb->fs_cpg, lastmount);
        } else {
                /* XXX 2038 */
                time_t t = sb->fs_ffs1_time;
                char *ct = ctime(&t);
                if (ct)
                        printf("ffs at %lld size %lld mount %s time %s", at,
                            (long long)(off_t) sb->fs_ffs1_size * sb->fs_fsize,
                            lastmount, ct);
                else
                        printf("ffs at %lld size %lld mount %s time %lld\n", at,
                            (long long)(off_t) sb->fs_ffs1_size * sb->fs_fsize,
                            lastmount, t);
        }
}

static int
ufsscan(int fd, daddr_t beg, daddr_t end, int flags)
{
        static char lastmount[MAXMNTLEN];
        static u_int8_t buf[SBSIZE * SBCOUNT];
        struct fs *sb;
        daddr_t blk, lastblk;
        int n;

        lastblk = -1;
        memset(lastmount, 0, MAXMNTLEN);

        for (blk = beg; blk <= ((end<0)?blk:end); blk += (SBCOUNT*SBSIZE/512)){
                memset(buf, 0, SBSIZE * SBCOUNT);
                if (lseek(fd, (off_t)blk * 512, SEEK_SET) == -1)
                    err(1, "lseek");
                if (read(fd, buf, SBSIZE * SBCOUNT) == -1)
                        err(1, "read");

                for (n = 0; n < (SBSIZE * SBCOUNT); n += 512){
                        sb = (struct fs*)(&buf[n]);
                        if (sb->fs_magic == FS_MAGIC) {
                                if (flags & FLAG_VERBOSE)
                                        printf("block %lld id %x,%x size %d\n",
                                            (long long)(blk + (n/512)),
                                            sb->fs_id[0], sb->fs_id[1],
                                            sb->fs_ffs1_size);

                                if (((blk+(n/512)) - lastblk) == (SBSIZE/512)) {
                                        print_info(flags, sb,
                                            (long long)(blk + (n/512) -
                                            (2*SBSIZE/512)),  lastmount);
                                        if (flags & FLAG_SMART) {
                                                off_t size = (off_t)sb->fs_ffs1_size *
                                                    sb->fs_fsize;

                                                if ((n + size) < (SBSIZE * SBCOUNT))
                                                        n += size;
                                                else {
                                                        blk += (size/512 -
                                                            (SBCOUNT*SBCOUNT));
                                                        break;
                                                }
                                        }
                                }

                                /* Update last potential FS SBs seen */
                                lastblk = blk + (n/512);
                                memcpy(lastmount, sb->fs_fsmnt, MAXMNTLEN);
                        }
                }
        }
        return(0);
}


static void
usage(void)
{
        extern char *__progname;

        fprintf(stderr, "usage: %s [-lsv] [-b begin] [-e end] device\n",
            __progname);
        exit(1);
}


int
main(int argc, char *argv[])
{
        int ch, fd, flags = 0;
        daddr_t beg = 0, end = -1;
        const char *errstr;

        if (pledge("stdio rpath disklabel", NULL) == -1)
                err(1, "pledge");

        while ((ch = getopt(argc, argv, "lsvb:e:")) != -1)
                switch(ch) {
                case 'b':
                        beg = strtonum(optarg, 0, LLONG_MAX, &errstr);
                        if (errstr)
                                errx(1, "%s: %s", optarg, errstr);
                        break;
                case 'e':
                        end = strtonum(optarg, 0, LLONG_MAX, &errstr);
                        if (errstr)
                                errx(1, "%s: %s", optarg, errstr);
                        break;
                case 'v':
                        flags |= FLAG_VERBOSE;
                        break;
                case 's':
                        flags |= FLAG_SMART;
                        break;
                case 'l':
                        flags |= FLAG_LABELS;
                        break;
                default:
                        usage();
                        /* NOTREACHED */
        }
        argc -= optind;
        argv += optind;

        if (argc != 1)
                usage();

        fd = opendev(argv[0], O_RDONLY, OPENDEV_PART, NULL);
        if (fd == -1)
                err(1, "%s", argv[0]);

        if (pledge("stdio", NULL) == -1)
                err(1, "pledge");

        return (ufsscan(fd, beg, end, flags));
}