#!/usr/bin/env bash

. $(dirname $(readlink -e ${BASH_SOURCE[0]}))/bcachefs-test-libs.sh

config-scratch-devs 4G
config-scratch-devs 4G
config-scratch-devs 16G
config-scratch-devs 16G

run_basic_tiering_test()
{
    run_basic_fio_test "$@"			\
	--block_size=4k				\
	--btree_node_size=32k			\
	--bucket=256k				\
	--label=ssd.ssd1 ${ktest_scratch_dev[0]}		\
	--label=ssd.ssd2 ${ktest_scratch_dev[1]}		\
	--label=hdd.hdd1 ${ktest_scratch_dev[2]}		\
	--label=hdd.hdd2 ${ktest_scratch_dev[3]}		\
	--foreground_target=ssd			\
	--promote_target=ssd			\
	--background_target=hdd
}

test_tiering_torture()
{
    run_quiet "" bcachefs format -f		\
	--block_size=4k				\
	--bucket=8M				\
	--label=ssd.ssd1 ${ktest_scratch_dev[0]}		\
	--label=ssd.ssd2 ${ktest_scratch_dev[1]}		\
	--label=hdd.hdd1 ${ktest_scratch_dev[2]}		\
	--label=hdd.hdd2 ${ktest_scratch_dev[3]}		\
	--foreground_target=ssd			\
	--promote_target=ssd			\
	--background_target=hdd

    mount -t bcachefs ${ktest_scratch_dev[0]}:${ktest_scratch_dev[1]}:${ktest_scratch_dev[2]}:${ktest_scratch_dev[3]} /mnt
    run_fio_randrw
    mv /mnt/fiotest /mnt/fiotest_old
    run_fio_randrw --loops=50

    mount -o remount,ro /mnt
    umount /mnt

    bcachefs fsck -ny "${ktest_scratch_dev[@]}"
    check_counters ${ktest_scratch_dev[0]}
}

test_tiering()
{
    run_basic_tiering_test
}

test_tiering_drop_alloc()
{
    run_basic_tiering_test
    # mount dropping all alloc info

    echo "dropping alloc:"
    mount -t bcachefs -o reconstruct_alloc,fsck,fix_errors "$(join_by : "${ktest_scratch_dev[@]}")" /mnt
    umount /mnt

    echo "final fsck:"

    bcachefs fsck -ny "${ktest_scratch_dev[@]}"
    check_counters ${ktest_scratch_dev[0]}
}

test_tiering_buffered()
{
    set_watchdog 6000

    run_quiet "" bcachefs format -f			\
	--block_size=4k					\
	--label=ssd.ssd1 ${ktest_scratch_dev[0]}	\
	--label=ssd.ssd2 ${ktest_scratch_dev[1]}	\
	--label=hdd.hdd1 ${ktest_scratch_dev[2]}	\
	--label=hdd.hdd2 ${ktest_scratch_dev[3]}	\
	--foreground_target=ssd				\
	--promote_target=ssd				\
	--background_target=hdd

    mount -t bcachefs "$(join_by : "${ktest_scratch_dev[@]}")" /mnt

    local loops=$((($ktest_priority + 1) * 4))

    fio --eta=always				\
	--exitall_on_error=1			\
	--ioengine=sync				\
	--numjobs=1				\
	--verify=crc32c				\
	--verify_fatal=1			\
	--buffer_compress_percentage=30		\
	--filename=/mnt/fiotest		    	\
	--filesize=3500M			\
	--loops=$loops				\
	--name=randrw				\
	--rw=randrw				\
	--bsrange=512-16k
    umount /mnt

    bcachefs fsck -ny "${ktest_scratch_dev[@]}"
    check_counters ${ktest_scratch_dev[0]}
}

test_writethrough()
{
    run_basic_fio_test "$@"								\
	--durability=0 --label=ssd ${ktest_scratch_dev[0]} ${ktest_scratch_dev[1]}	\
	--durability=1 --label=hdd ${ktest_scratch_dev[2]} ${ktest_scratch_dev[3]}	\
	--foreground_target=ssd								\
	--promote_target=ssd								\
	--background_target=hdd
}

test_promote()
{
    run_basic_fio_test "$@"			\
	--durability=0 --label=ssd ${ktest_scratch_dev[0]}	\
	--durability=1 --label=hdd ${ktest_scratch_dev[2]}	\
	--foreground_target=hdd			\
	--promote_target=ssd			\
	--background_target=hdd
}

test_setattr()
{
    setup_tracing 'bcachefs:*'

    run_quiet "" bcachefs format -f				\
	--durability=0 --label=ssd ${ktest_scratch_dev[0]}	\
	--durability=1 --label=hdd ${ktest_scratch_dev[2]}	\
	--background_target=hdd

    mount -t bcachefs ${ktest_scratch_dev[0]}:${ktest_scratch_dev[2]} /mnt

    touch /mnt/foo
    touch /mnt/bar

    bcachefs setattr --foreground_target $(basename ${ktest_scratch_dev[0]}) /mnt/
    bcachefs setattr --promote_target ssd /mnt/

    umount /mnt

    check_counters ${ktest_scratch_dev[0]}
}

test_tiering_no_checksum()
{
    run_basic_tiering_test --data_checksum=none
}

test_tiering_compression()
{
    run_basic_tiering_test --compression=lz4
}

test_tiering_crypto()
{
    run_basic_tiering_test --encrypted --no_passphrase	
}

test_tiering_crypto_lz4()
{
    run_basic_tiering_test --encrypted --no_passphrase --compression=lz4
}

test_tiering_crypto_zstd()
{
    run_basic_tiering_test --encrypted --no_passphrase --compression=zstd
}

test_tiering_replication()
{
    run_basic_tiering_test --replicas=2
}

test_tiering_variable_buckets()
{
    run_basic_fio_test					\
	--label=ssd --bucket=32k	${ktest_scratch_dev[0]}	\
	--label=ssd --bucket=64k	${ktest_scratch_dev[1]}	\
	--label=hdd --bucket=128k	${ktest_scratch_dev[3]}	\
	--label=hdd --bucket=256k	${ktest_scratch_dev[2]}	\
	--foreground_target=ssd				\
	--promote_target=ssd				\
	--background_target=hdd
}

test_tiering_variable_buckets_replicas()
{
    run_basic_fio_test					\
	--replicas=2					\
	--label=ssd --bucket=32k	${ktest_scratch_dev[0]}	\
	--label=ssd --bucket=64k	${ktest_scratch_dev[1]}	\
	--label=hdd --bucket=128k	${ktest_scratch_dev[3]}	\
	--label=hdd --bucket=256k	${ktest_scratch_dev[2]}	\
	--foreground_target=ssd				\
	--promote_target=ssd				\
	--background_target=hdd
}

test_device_add()
{
    set_watchdog 60
    run_quiet "" bcachefs format -f					\
	--label=ssd ${ktest_scratch_dev[0]}				\
	--label=hdd ${ktest_scratch_dev[2]} ${ktest_scratch_dev[3]}

    mount -t bcachefs ${ktest_scratch_dev[0]}:${ktest_scratch_dev[2]}:${ktest_scratch_dev[3]} /mnt

    run_fio_randrw|tail -n200 &

    bcachefs device add -f --label=hdd /mnt ${ktest_scratch_dev[1]}

    wait
    umount /mnt

    bcachefs fsck -ny "${ktest_scratch_dev[@]}"
    check_counters ${ktest_scratch_dev[0]}
}

test_mount_umount_torture()
{
    set_watchdog 60
    run_quiet "" bcachefs format					\
	--replicas=2							\
	--label=ssd ${ktest_scratch_dev[0]} ${ktest_scratch_dev[1]}	\
	--label=hdd ${ktest_scratch_dev[2]} ${ktest_scratch_dev[3]}	\
	--foreground_target=ssd						\
	--promote_target=ssd						\
	--background_target=hdd

    for i in {0..10}; do
	mount -t bcachefs "$(join_by : "${ktest_scratch_dev[@]}")" /mnt
	umount /mnt
    done

    bcachefs fsck -ny "${ktest_scratch_dev[@]}"
    check_counters ${ktest_scratch_dev[0]}
}

test_change_replicas()
{
    set_watchdog 180
    run_quiet "" bcachefs format -f					\
	--label=ssd ${ktest_scratch_dev[0]} ${ktest_scratch_dev[1]}	\
	--label=hdd ${ktest_scratch_dev[2]} ${ktest_scratch_dev[3]}	\
	--foreground_target=ssd						\
	--promote_target=ssd						\
	--background_target=hdd

    mount -t bcachefs "$(join_by : "${ktest_scratch_dev[@]}")" /mnt

    (
	cd /sys/fs/bcachefs/*
	while true; do
	    echo 1 > options/data_replicas
	    sleep 0.1
	    echo 2 > options/data_replicas
	    sleep 0.1
	done
    ) &

    loops=$((($ktest_priority + 1) * 4))

    fio --eta=always				\
	--numjobs=1				\
	--buffer_compress_percentage=50		\
	--filename=/mnt/fiotest		    	\
	--filesize=1500M			\
	--loops=$loops				\
	--name=randrw				\
	--rw=randrw				\
	--bsrange=4k-1M	

#    while true; do
#	dd if=/dev/zero of=/mnt/foo bs=1M count=1024 oflag=direct
#	sync
#	#rm /mnt/foo
#    done

    kill %1
    umount /mnt

    bcachefs fsck -ny "${ktest_scratch_dev[@]}"
    check_counters ${ktest_scratch_dev[0]}
}

test_concurrent_all_device_removal()
{
    set_watchdog 120

    run_quiet "" bcachefs format -f			\
       --label=ssd.ssd1 ${ktest_scratch_dev[0]}		\
       --label=ssd.ssd2 ${ktest_scratch_dev[1]}		\
       --label=hdd.hdd1 ${ktest_scratch_dev[2]}		\
       --label=hdd.hdd2 ${ktest_scratch_dev[3]}		\
       --metadata_replicas=3 \
       --data_replicas=3 \
       --promote_target=ssd

    mount -t bcachefs "$(join_by : "${ktest_scratch_dev[@]}")" /mnt

    dd if=/dev/zero of=/mnt/foo bs=128M count=40 oflag=direct

    # Fill cache
    for i in 1 2 3 4; do cat /mnt/foo > /dev/null; done

    bcachefs device remove ${ktest_scratch_dev[0]} &
    bcachefs device remove ${ktest_scratch_dev[1]} &
    bcachefs device remove ${ktest_scratch_dev[2]} &
    bcachefs device remove ${ktest_scratch_dev[3]} &
    wait

    umount /mnt
    bcachefs fsck -ny $(join_by " " "${ktest_scratch_dev[@]}")

    # Test that no incufficient devices issues appears
    mount -t bcachefs "$(join_by : "${ktest_scratch_dev[@]}")" /mnt
    umount /mnt

    check_counters ${ktest_scratch_dev[0]}
}

main $@