import os
import subprocess

import infra.basetest


class TestXfsProgs(infra.basetest.BRTest):
    # This test needs a Kernel with XFS support.
    kern_frag = \
        infra.filepath("tests/package/test_xfsprogs/linux-xfs.fragment")
    config = \
        f"""
        BR2_aarch64=y
        BR2_TOOLCHAIN_EXTERNAL=y
        BR2_TOOLCHAIN_EXTERNAL_BOOTLIN=y
        BR2_TARGET_GENERIC_GETTY_PORT="ttyAMA0"
        BR2_LINUX_KERNEL=y
        BR2_LINUX_KERNEL_CUSTOM_VERSION=y
        BR2_LINUX_KERNEL_CUSTOM_VERSION_VALUE="6.18.18"
        BR2_LINUX_KERNEL_USE_CUSTOM_CONFIG=y
        BR2_LINUX_KERNEL_CUSTOM_CONFIG_FILE="board/qemu/aarch64-virt/linux.config"
        BR2_LINUX_KERNEL_CONFIG_FRAGMENT_FILES="{kern_frag}"
        BR2_LINUX_KERNEL_NEEDS_HOST_OPENSSL=y
        BR2_PACKAGE_XFSPROGS=y
        BR2_PACKAGE_OPENSSL=y
        BR2_TARGET_ROOTFS_CPIO=y
        BR2_TARGET_ROOTFS_CPIO_GZIP=y
        # BR2_TARGET_ROOTFS_TAR is not set
        """

    def test_run(self):
        # Prepare the disk image. XFS requires at least 300MB of
        # storage to work.
        disk_file = os.path.join(self.builddir, "images", "disk.img")
        self.emulator.logfile.write(f"Creating disk image: {disk_file}")
        self.emulator.logfile.flush()
        subprocess.check_call(
            ["dd", "if=/dev/zero", f"of={disk_file}", "bs=1M", "count=512"],
            stdout=self.emulator.logfile,
            stderr=self.emulator.logfile)

        # Run the emulator with a blank drive.
        img = os.path.join(self.builddir, "images", "rootfs.cpio.gz")
        kern = os.path.join(self.builddir, "images", "Image")
        bootargs = ["console=ttyAMA0"]
        qemu_opts = ["-M", "virt", "-cpu", "cortex-a57", "-m", "256M",
                     "-initrd", img,
                     "-drive", f"file={disk_file},if=virtio,format=raw"]
        self.emulator.boot(arch="aarch64",
                           kernel=kern,
                           kernel_cmdline=bootargs,
                           options=qemu_opts)
        self.emulator.login()

        # Check a program can run and show its version.
        self.assertRunOk("mkfs.xfs -V")

        # Variables for this test.
        dev = "/dev/vda"
        label = "BR_TEST"
        mnt_pt = "/tmp/xfs"
        data_file = f"{mnt_pt}/data.bin"

        # We create the XFS filesystem on our device.
        self.assertRunOk(f"mkfs.xfs {dev}")

        # We set a label on this filesystem.
        self.assertRunOk(f"xfs_admin -L '{label}' {dev}")

        # We create a mount point.
        self.assertRunOk(f"mkdir -p {mnt_pt}")

        # We mount the XFS filesystem.
        self.assertRunOk(f"mount {dev} {mnt_pt}")

        # We create a file with random data, to use this new
        # filesystem a bit.
        self.assertRunOk(f"dd if=/dev/urandom of={data_file} bs=1M count=10")

        # We compute the sha256 hash and save it for later.
        hash_cmd = f"sha256sum {data_file}"
        out, ret = self.emulator.run(hash_cmd)
        self.assertEqual(ret, 0)
        data_sha256 = out[0]

        # We unmount the filesystem.
        self.assertRunOk(f"umount {mnt_pt}")

        # We query the fragmentation level. Since it's a new
        # filesystem, we don't expect any filesystem
        # fragmentation. This invocation is just to test the "xfs_db"
        # program can run and read the filesystem.
        self.assertRunOk(f"xfs_db -c frag -r {dev}")

        # We run a repair (this tool is the equivalent of
        # "fsck"). Again, since we cleanly unmounted the filesystem,
        # we are not expecting any repair. This is just to test the
        # program works correctly.
        self.assertRunOk(f"xfs_repair {dev}")

        # We query the label and check it is the one we set at the
        # beginning.
        out, ret = self.emulator.run(f"xfs_admin -l {dev}")
        self.assertEqual(ret, 0)
        self.assertEqual(out[0], f"label = \"{label}\"")

        # We remount our filesystem.
        self.assertRunOk(f"mount {dev} {mnt_pt}")

        # We should recompute the same sha256 hash as before.
        out, ret = self.emulator.run(hash_cmd)
        self.assertEqual(ret, 0)
        self.assertEqual(out[0], data_sha256)
