/*      $NetBSD: t_resize.c,v 1.2 2025/10/30 15:30:17 perseant Exp $    */

#include <sys/types.h>
#include <sys/mount.h>
#include <sys/wait.h>

#include <atf-c.h>
#include <ctype.h>
#include <errno.h>
#include <fcntl.h>
#include <limits.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>

#include <rump/rump.h>
#include <rump/rump_syscalls.h>

#include <ufs/ufs/ufsmount.h>
#include <ufs/lfs/lfs.h>
#include <ufs/lfs/lfs_extern.h>

#include "h_macros.h"
#include "util.h"

/* Debugging conditions */
/* #define FORCE_SUCCESS */ /* Don't actually revert, everything worked */
/* #define USE_DUMPLFS */ /* Dump the filesystem at certain steps */

#define UNCHANGED_CONTROL MP "/3-a-random-file"
#define BIGSIZE 15000
#define SMALLSIZE 10000
__CTASSERT(BIGSIZE > SMALLSIZE);

/* Resize filesystem */
void resize(int, size_t);

/* Actually run the test */
void test(int);

ATF_TC(resize32);
ATF_TC_HEAD(resize32, tc)
{
        atf_tc_set_md_var(tc, "descr",
                "LFS32 resize_lfs creates an inconsistent filesystem");
        atf_tc_set_md_var(tc, "timeout", "20");
}

ATF_TC(resize64);
ATF_TC_HEAD(resize64, tc)
{
        atf_tc_set_md_var(tc, "descr",
                "LFS64 resize_lfs creates an inconsistent filesystem");
        atf_tc_set_md_var(tc, "timeout", "20");
}

ATF_TC_BODY(resize32, tc)
{
        test(32);
}

ATF_TC_BODY(resize64, tc)
{
        test(64);
}

void test(int width)
{
        struct ufs_args args;
        int fd;

        setvbuf(stdout, NULL, _IONBF, 0);

        /*
         * Initialize.
         */

        /* Create image file larger than filesystem */
        create_lfs(BIGSIZE, SMALLSIZE, width, 1);

        /* Mount filesystem */
        fprintf(stderr, "* Mount fs [1]\n");
        memset(&args, 0, sizeof(args));
        args.fspec = __UNCONST(FAKEBLK);
        if (rump_sys_mount(MOUNT_LFS, MP, 0, &args, sizeof(args)) == -1)
                atf_tc_fail_errno("rump_sys_mount failed");

        /* Payload */
        fprintf(stderr, "* Initial payload\n");
        write_file(UNCHANGED_CONTROL, CHUNKSIZE, 1, 0);

        /* Unmount */
        rump_sys_unmount(MP, 0);
        if (fsck())
                atf_tc_fail_errno("fsck found errors after first unmount");

        /*
         * Remount and resize.
         */

        /* Reconfigure and mount filesystem again */
        fprintf(stderr, "* Remount fs [2, to enlarge]\n");
        if (rump_sys_mount(MOUNT_LFS, MP, 0, &args, sizeof(args)) == -1)
                atf_tc_fail_errno("rump_sys_mount failed [2]");

        /* Get a handle to the root of the file system */
        fd = rump_sys_open(MP, O_RDONLY);
        if (fd < 0)
                atf_tc_fail_errno("rump_sys_open mount point root failed");

        /* Enlarge filesystem */
        fprintf(stderr, "* Resize (enlarge)\n");
        resize(fd, BIGSIZE);

        /* Unmount fs and check */
        rump_sys_close(fd);
        rump_sys_unmount(MP, 0);
        if (fsck())
                atf_tc_fail_errno("fsck found errors after enlarge");

        /* Mount filesystem for shrink */
        fprintf(stderr, "* Mount fs [3, to shrink]\n");
        if (rump_sys_mount(MOUNT_LFS, MP, 0, &args, sizeof(args)) == -1)
                atf_tc_fail_errno("rump_sys_mount failed [3]");

        /* Get a handle to the root of the file system */
        fd = rump_sys_open(MP, O_RDONLY);
        if (fd < 0)
                atf_tc_fail_errno("rump_sys_open mount point root failed");

        /* Shrink filesystem */
        fprintf(stderr, "* Resize (shrink)\n");
        resize(fd, SMALLSIZE);

        /* Unmount and check again */
        rump_sys_close(fd);
        if (rump_sys_unmount(MP, 0) != 0)
                atf_tc_fail_errno("rump_sys_umount failed after shrink");
        fprintf(stderr, "* Fsck after shrink\n");
        if (fsck())
                atf_tc_fail("fsck found errors after shrink");

        /*
         * Check file system contents
         */

        /* Mount filesystem one last time */
        fprintf(stderr, "* Mount fs [4, to check contents]\n");
        if (rump_sys_mount(MOUNT_LFS, MP, 0, &args, sizeof(args)) == -1)
                atf_tc_fail_errno("rump_sys_mount failed [4]");

        if (check_file(UNCHANGED_CONTROL, CHUNKSIZE, 0) != 0)
                atf_tc_fail("Unchanged control file differs(!)");

        /* Umount filesystem */
        rump_sys_unmount(MP, 0);

        /* Final fsck to double check */
        fprintf(stderr, "* Fsck after final unmount\n");
        if (fsck())
                atf_tc_fail("fsck found errors after final unmount");
}

ATF_TP_ADD_TCS(tp)
{

        ATF_TP_ADD_TC(tp, resize32);
        ATF_TP_ADD_TC(tp, resize64);
        return atf_no_error();
}

void
resize(int fd, size_t size)
{
        int newnseg = (size * DEV_BSIZE) / SEGSIZE;
        
        if (rump_sys_fcntl(fd, LFCNRESIZE, &newnseg) != 0)
                atf_tc_fail_errno("LFCNRESIZE failed");
}