#!/bin/bash
#
# Copyright(c) 2012-2019 Intel Corporation
# SPDX-License-Identifier: BSD-3-Clause-Clear
#

MODULE_FILE=test_mod.c
OBJ_MOD=test_mod.o
KERN_VER=`uname -r`
PWD=`pwd`
NPROC=`nproc`
DEFINE_FILE=modules/generated_defines.h


add_define() {
   echo -n "#define " >> $DEFINE_FILE
   for arg in "$@"; do
       echo -e "$arg" >> $DEFINE_FILE
   done
}

add_function() {
	for arg in "$@"; do
		echo -e "$arg" >> $DEFINE_FILE
	done
}

compile_module(){
	if [ $# -gt 1 ]
	then
		INCLUDE="#include <$2>"
	else
		INCLUDE=""
	fi

############# TEST MODULE #############
cat > $MODULE_FILE << EOF
#include <linux/module.h>
#include <linux/kernel.h>
$INCLUDE

int init_module(void) {
	$1;
	return 0;
}
void cleanup_module(void) {};
EOF
#######################################

	make -C /lib/modules/$KERN_VER/build M=$PWD modules\
		obj-m=$OBJ_MOD -j$NPROC &> /dev/null

	local ret=$?
	if [ $ret -eq 0 ]; then
		make -j$NPROC -C /lib/modules/$KERN_VER/build M=$PWD clean\
			obj-m=test.o &> /dev/null
	fi

	return $ret
}

kernel_not_supp_fail() {
	echo "Current kernel is not supported!"
	rm $DEFINE_FILE
	exit 1
}

rm -f $DEFINE_FILE

compile_module "part_round_stats(1, 1)" "linux/genhd.h"
if [ $? -eq 0 ]; then
	add_define "CAS_PART_ROUND_STATS(q, cpu, part) part_round_stats(cpu, part)"
else
	compile_module "part_round_stats(NULL, 1, 1)" "linux/genhd.h"
	if [ $? -eq 0 ]; then
		add_define "CAS_PART_ROUND_STATS(q, cpu, part)\\"\
			"\tpart_round_stats(q, cpu, part)"
	fi
fi


compile_module "part_inc_in_flight(1, 1)" "linux/genhd.h"
if [ $? -eq 0 ]; then
	add_define "CAS_PART_INC_IN_FLIGHT(q, cpu, part)\\"\
		"\tpart_inc_in_flight(cpu, part)"
else
	compile_module "part_inc_in_flight(NULL, 1, 1)" "linux/genhd.h"
	if [ $? -eq 0 ]; then
		add_define "CAS_PART_INC_IN_FLIGHT(q, cpu, part)\\"\
			"\tpart_inc_in_flight(q, cpu, part)"
	fi
fi


compile_module "part_dec_in_flight(1, 1)" "linux/genhd.h"
if [ $? -eq 0 ]; then
	add_define "CAS_PART_DEC_IN_FLIGHT(q, cpu, part)\\"\
		"\tpart_dec_in_flight(cpu, part)"
else
	compile_module "part_dec_in_flight(NULL, 1, 1)" "linux/genhd.h"
	if [ $? -eq 0 ]; then
		add_define "CAS_PART_DEC_IN_FLIGHT(q, cpu, part)\\"\
			"\tpart_dec_in_flight(q, cpu, part)"
	fi
fi