#!/bin/bash

kpatch_set_ccache_max_size()

{

	local ccache_max_size=${1:-10G}

	ccache --max-size="${ccache_max_size}"

}

kpatch_fedora_dependencies()

{

	local kernel_version

	kernel_version=$(uname -r)

	sudo dnf install -y gcc "kernel-devel-${kernel_version%.*}" elfutils elfutils-devel

	sudo dnf install -y pesign yum-utils openssl wget numactl-devel

	sudo dnf builddep -y "kernel-${kernel_version%.*}"

	sudo dnf debuginfo-install -y "kernel-${kernel_version%.*}"

	sudo dnf install -y ccache

}

kpatch_ubuntu_dependencies()

{

	sudo sed -i 's/# deb-src/deb-src/' /etc/apt/sources.list

	sudo apt-get update

	sudo apt-get install -y make gcc libelf-dev elfutils

	sudo apt-get install -y dpkg-dev devscripts

	sudo apt-get build-dep -y linux

	sudo apt-get install -y ccache

	# Add ddebs repository

	if ! grep -q 'ddebs.ubuntu.com' /etc/apt/sources.list.d/ddebs.list; then

		local codename

		codename=$(lsb_release -sc)

		sudo tee /etc/apt/sources.list.d/ddebs.list <<-EOF

			deb http://ddebs.ubuntu.com/ ${codename} main restricted universe multiverse

			deb http://ddebs.ubuntu.com/ ${codename}-security main restricted universe multiverse

			deb http://ddebs.ubuntu.com/ ${codename}-updates main restricted universe multiverse

			deb http://ddebs.ubuntu.com/ ${codename}-proposed main restricted universe multiverse

		EOF

		# add APT key

		wget -Nq http://ddebs.ubuntu.com/dbgsym-release-key.asc -O- | sudo apt-key add -

		sudo apt-get update

	fi

	sudo apt-get install -y "linux-image-$(uname -r)-dbgsym"

}

kpatch_rhel_dependencies()

{

	local kernel_version

	local arch

	local rhel_major

	local py_version

	kernel_version=$(uname -r)

	arch=$(uname -m)

	rhel_major=${VERSION_ID%%.*}

	if [ "${rhel_major}" -ge 8 ]; then

		py_version="3"

	else

		py_version="2"

	fi

	sudo yum install -y git gcc gcc-c++ "kernel-devel-${kernel_version%.*}" elfutils elfutils-devel

	sudo yum install -y yum-utils zlib-devel binutils-devel newt-devel \

		python${py_version}-devel perl-ExtUtils-Embed audit-libs-devel numactl-devel \

		pciutils-devel bison ncurses-devel rpm-build java-devel

	sudo yum-builddep -y "kernel-${kernel_version%.*}"

	sudo debuginfo-install -y "kernel-${kernel_version%.*}"

	case "${arch}" in

		"x86_64")

			sudo yum install -y pesign

			;;

		"ppc64le")

			sudo yum install -y gcc-plugin-devel

			if [ "${rhel_major}" -ge 8 ]; then

				# yum-builddep doesn't provide everything we need :(

				sudo yum install -y flex openssl-devel

			fi

			;;

		*)

			;;

	esac

	sudo yum install -y "https://dl.fedoraproject.org/pub/epel/epel-release-latest-7.noarch.rpm"

	sudo yum install -y ccache

	sudo yum remove -y epel-release

}

kpatch_centos_dependencies()

{

	local kernel_version

	local arch

	kernel_version=$(uname -r)

	arch=$(uname -m)

	sudo yum install -y gcc gcc-c++ "kernel-devel-${kernel_version%.*}" elfutils elfutils-devel

	sudo yum install -y yum-utils zlib-devel binutils-devel newt-devel \

		python-devel perl-ExtUtils-Embed audit-libs-devel numactl-devel \

		pciutils-devel bison ncurses-devel rpm-build java-devel pesign

	sudo yum-config-manager --enable debug

	sudo yum-builddep -y "kernel-${kernel_version%.*}"

	sudo debuginfo-install -y "kernel-${kernel_version%.*}"

	sudo yum install -y "https://dl.fedoraproject.org/pub/epel/epel-release-latest-7.noarch.rpm"

	sudo yum install -y ccache

	sudo yum remove -y epel-release

}

kpatch_dependencies()

{

	# shellcheck disable=SC1091

	source /etc/os-release

	eval "kpatch_${ID}_dependencies" || { echo "Unsupported distro: ${ID}"; exit 1; }

}

kpatch_separate_partition_cache()

{

	local partition=${1}

	local mountpoint=${2}

	local reformat=${3}

	local owner=${USER}

	if [[ "${reformat}" == "y" ]]; then

		sudo mkfs.xfs -f "${partition}"

	fi

	sudo mkdir -p "${mountpoint}"

	sudo mount "${partition}" "${mountpoint}"

	sudo chown "${owner}":"${owner}" "${mountpoint}"

	rm -rf "${mountpoint}/.ccache"

	rm -rf "${mountpoint}/.kpatch"

	mkdir "${mountpoint}/.ccache"

	mkdir "${mountpoint}/.kpatch"

	rm -rf "${HOME}/.ccache"

	rm -rf "${HOME}/.kpatch"

	ln -sv "${mountpoint}/.ccache" "${HOME}/.ccache"

	ln -sv "${mountpoint}/.kpatch" "${HOME}/.kpatch"

}

kpatch_separate_disk_cache()

{

	local device=${1}

	local mountpoint=${2}

	local partition="${device}1"

	echo -e "o\\nn\\np\\n1\\n\\n\\nw\\n" | sudo fdisk "${device}"

	kpatch_separate_partition_cache "${partition}" "${mountpoint}" y

}

kpatch_install_vagrant_centos()

{

	local image_path=${1}

	sudo yum group install -y "Development Tools"

	sudo yum -y install qemu-kvm libvirt virt-install bridge-utils libvirt-devel libxslt-devel libxml2-devel libvirt-devel libguestfs-tools-c libvirt-client

	echo "net.ipv4.ip_forward = 1" | sudo tee /etc/sysctl.d/99-ipforward.conf

	sudo sysctl -p /etc/sysctl.d/99-ipforward.conf

	sudo systemctl enable libvirtd

	sudo systemctl start libvirtd || exit 1

	if [[ -n "${image_path}" ]]; then

		mkdir -p "${image_path}/libvirt/images"

		virsh pool-define-as --target "${image_path}/libvirt/images" default dir || exit 1

		virsh pool-start default || exit 1

	fi

	sudo yum install -y https://releases.hashicorp.com/vagrant/2.1.2/vagrant_2.1.2_x86_64.rpm || exit 1

	vagrant plugin install vagrant-libvirt

}

kpatch_install_vagrant_rhel()

{

	local image_path=${1}

	kpatch_install_vagrant_centos "${image_path}"

	sudo systemctl enable nfs

	sudo systemctl start nfs || exit 1

}

kpatch_install_vagrant_fedora()

{

	local image_path=${1}

	echo "net.ipv4.ip_forward = 1" | sudo tee /etc/sysctl.d/99-ipforward.conf

	sudo sysctl -p /etc/sysctl.d/99-ipforward.conf

	sudo dnf install -y libvirt virt-install libvirt-client nfs-utils vagrant vagrant-libvirt

	echo "[nfsd]" | sudo tee -a /etc/nfs.conf

	echo "udp=y" | sudo tee -a /etc/nfs.conf

	echo "vers3=y" | sudo tee -a /etc/nfs.conf

	sudo systemctl restart nfs

	sudo systemctl enable libvirtd

	sudo systemctl start libvirtd || exit 1

	if [[ -n "${image_path}" ]]; then

		mkdir -p "${image_path}/libvirt/images"

		virsh pool-define-as --target "${image_path}/libvirt/images" default dir || exit 1

		virsh pool-start default || exit 1

	fi

}

kpatch_install_vagrant()

{

	local image_path=${1}

	# shellcheck disable=SC1091

	source /etc/os-release

	eval "kpatch_install_vagrant_${ID} ${image_path}" || { echo "Unsupported distro: ${ID}"; exit 1; }

}

kpatch_check_install_vagrant()

{

	local image_path=${1}

	# shellcheck disable=SC2230

	[ "$(which vagrant)" == "" ] && kpatch_install_vagrant "${image_path}"

	return 0

}

kpatch_write_vagrantfile_template()

{

	local target_distro=${1}

	local box_prefix="kpatch"

	cat >Vagrantfile <

Vagrant.configure("2") do |config|

	config.vm.provider :libvirt do |libvirt|

		libvirt.storage :file, :size => '40G'

		libvirt.cpus = $(getconf _NPROCESSORS_ONLN)

		libvirt.memory = $(awk '/MemTotal/ {printf("%d\n", ($2*0.8)/1024)}' /proc/meminfo)

		libvirt.graphics_type = "none"

		libvirt.disk_bus = 'virtio'

		libvirt.disk_device = 'vda'

	end

	config.vm.box = "${box_prefix}/${target_distro}"

	config.vm.synced_folder ".", "/vagrant", type: "nfs"

EOF

}

kpatch_write_vagrantfile_centos_provision()

{

	cat >>Vagrantfile <

	config.vm.provision "shell", inline: "yum install -y git"

EOF

}

kpatch_write_vagrantfile()

{

	local target_distro=${1}

	kpatch_write_vagrantfile_template "${target_distro}"

	if echo "${target_distro}" | grep -qE "^centos"; then

		kpatch_write_vagrantfile_centos_provision

	fi

	echo 'end' >>Vagrantfile

}

kpatch_integration_tests_vagrant_distro()

{

	local target_distro=${1}

	local test_script=${2}

	local slowtest=${3}

	local testdir

	local workdir

	local logdir

	testdir="$(pwd)"

	workdir="${target_distro}.vagrant"

	rm -rf "${workdir}"

	mkdir -p "${workdir}"

	cd "${workdir}" || exit 1

	kpatch_write_vagrantfile "${target_distro}"

	vagrant up || { vagrant destroy -f; exit 1; }

	local test_cmd="KPATCH_GIT=${KPATCH_GIT} KPATCH_REV=${KPATCH_REV} bash /vagrant/runtest.sh"

	if [ "${slowtest}" == "1" ]; then

		test_cmd="${test_cmd} --slow"

	fi

	cp "${test_script}" ./runtest.sh

	vagrant ssh -c "${test_cmd}"

	local rc=$?

	if [ $rc -eq 0 ]; then

		echo "${target_distro} PASS"

	else

		echo "${target_distro} FAIL"

	fi

	logdir="${testdir}/${target_distro}_log"

	rm -rf "${logdir}"

	mkdir -p "${logdir}"

	cp logs/* "${logdir}"

	vagrant destroy -f

	cd "${testdir}" || exit 1

	if [ $rc -eq 0 ]; then

		rm -rf "${workdir}"

	fi

	return "${rc}"

}