#!/bin/bash
#
# Copyright(c) 2012-2022 Intel Corporation
# Copyright(c) 2025 Huawei Technologies
# SPDX-License-Identifier: BSD-3-Clause
#

. $(dirname $3)/conf_framework.sh


check() {
	cur_name=$(basename $2)
	config_file_path=$1
	if compile_module $cur_name "blk_cleanup_disk(NULL);" "linux/blk-mq.h"
	then
		echo $cur_name 1 >> $config_file_path
	else
		echo $cur_name 2 >> $config_file_path
	fi
}

apply() {
    case "$1" in
    "1")
        add_function "
        static inline void _cas_cleanup_disk(struct gendisk *gd)
        {
		blk_cleanup_disk(gd);
	}"
	;;

    "2")

	add_function "
	static inline void _cas_cleanup_disk(struct gendisk *gd)
	{
		put_disk(gd);
	}"
	;;

    *)
        exit 1
    esac
}

conf_run $@