// SPDX-License-Identifier: GPL-2.0
/*
 * Copyright 2024, Intel Corporation
 *
 * Author: Rafael J. Wysocki <rafael.j.wysocki@intel.com>
 *
 * Thermal subsystem testing facility.
 *
 * This facility allows the thermal core functionality to be exercised in a
 * controlled way in order to verify its behavior.
 *
 * It resides in the "thermal-testing" directory under the debugfs root and
 * starts with a single file called "command" which can be written a string
 * representing a thermal testing facility command.
 *
 * The currently supported commands are listed in the tt_commands enum below.
 *
 * The "addtz" command causes a new test thermal zone template to be created,
 * for example:
 *
 * # echo addtz > /sys/kernel/debug/thermal-testing/command
 *
 * That template will be represented as a subdirectory in the "thermal-testing"
 * directory, for example
 *
 * # ls /sys/kernel/debug/thermal-testing/
 * command tz0
 *
 * The thermal zone template can be populated with trip points with the help of
 * the "tzaddtrip" command, for example:
 *
 * # echo tzaddtrip:0 > /sys/kernel/debug/thermal-testing/command
 *
 * which causes a trip point template to be added to the test thermal zone
 * template 0 (represented by the tz0 subdirectory in "thermal-testing").
 *
 * # ls /sys/kernel/debug/thermal-testing/tz0
 * init_temp temp trip_0_temp trip_0_hyst
 *
 * The temperature of a trip point template is initially THERMAL_TEMP_INVALID
 * and its hysteresis is initially 0.  They can be adjusted by writing to the
 * "trip_x_temp" and "trip_x_hyst" files correspoinding to that trip point
 * template, respectively.
 *
 * The initial temperature of a thermal zone based on a template can be set by
 * writing to the "init_temp" file in its directory under "thermal-testing", for
 * example:
 *
 * echo 50000 > /sys/kernel/debug/thermal-testing/tz0/init_temp
 *
 * When ready, "tzreg" command can be used for registering and enabling a
 * thermal zone based on a given template with the thermal core, for example
 *
 * # echo tzreg:0 > /sys/kernel/debug/thermal-testing/command
 *
 * In this case, test thermal zone template 0 is used for registering a new
 * thermal zone and the set of trip point templates associated with it is used
 * for populating the new thermal zone's trip points table.  The type of the new
 * thermal zone is "test_tz".
 *
 * The temperature and hysteresis of all of the trip points in that new thermal
 * zone are adjustable via sysfs, so they can be updated at any time.
 *
 * The current temperature of the new thermal zone can be set by writing to the
 * "temp" file in the corresponding thermal zone template's directory under
 * "thermal-testing", for example
 *
 * echo 10000 > /sys/kernel/debug/thermal-testing/tz0/temp
 *
 * which will also trigger a temperature update for this zone in the thermal
 * core, including checking its trip points, sending notifications to user space
 * if any of them have been crossed and so on.
 *
 * When it is not needed any more, a test thermal zone template can be deleted
 * with the help of the "deltz" command, for example
 *
 * # echo deltz:0 > /sys/kernel/debug/thermal-testing/command
 *
 * which will also unregister the thermal zone based on it, if present.
 */

#define pr_fmt(fmt) "thermal-testing: " fmt

#include <linux/debugfs.h>
#include <linux/module.h>

#include "thermal_testing.h"

struct dentry *d_testing;

#define TT_COMMAND_SIZE         16

enum tt_commands {
        TT_CMD_ADDTZ,
        TT_CMD_DELTZ,
        TT_CMD_TZADDTRIP,
        TT_CMD_TZREG,
        TT_CMD_TZUNREG,
};

static const char *tt_command_strings[] = {
        [TT_CMD_ADDTZ] = "addtz",
        [TT_CMD_DELTZ] = "deltz",
        [TT_CMD_TZADDTRIP] = "tzaddtrip",
        [TT_CMD_TZREG] = "tzreg",
        [TT_CMD_TZUNREG] = "tzunreg",
};

static int tt_command_exec(int index, const char *arg)
{
        int ret;

        switch (index) {
        case TT_CMD_ADDTZ:
                ret = tt_add_tz();
                break;

        case TT_CMD_DELTZ:
                ret = tt_del_tz(arg);
                break;

        case TT_CMD_TZADDTRIP:
                ret = tt_zone_add_trip(arg);
                break;

        case TT_CMD_TZREG:
                ret = tt_zone_reg(arg);
                break;

        case TT_CMD_TZUNREG:
                ret = tt_zone_unreg(arg);
                break;

        default:
                ret = -EINVAL;
                break;
        }

        return ret;
}

static ssize_t tt_command_process(char *s)
{
        char *arg;
        int i;

        strim(s);

        arg = strchr(s, ':');
        if (arg) {
                *arg = '\0';
                arg++;
        }

        for (i = 0; i < ARRAY_SIZE(tt_command_strings); i++) {
                if (!strcmp(s, tt_command_strings[i]))
                        return tt_command_exec(i, arg);
        }

        return -EINVAL;
}

static ssize_t tt_command_write(struct file *file, const char __user *user_buf,
                                size_t count, loff_t *ppos)
{
        char buf[TT_COMMAND_SIZE];
        ssize_t ret;

        if (*ppos)
                return -EINVAL;

        if (count > TT_COMMAND_SIZE - 1)
                return -E2BIG;

        if (copy_from_user(buf, user_buf, count))
                return -EFAULT;
        buf[count] = '\0';

        ret = tt_command_process(buf);
        if (ret)
                return ret;

        return count;
}

static const struct file_operations tt_command_fops = {
        .write = tt_command_write,
        .open =  simple_open,
        .llseek = default_llseek,
};

static int __init thermal_testing_init(void)
{
        d_testing = debugfs_create_dir("thermal-testing", NULL);
        if (!IS_ERR(d_testing))
                debugfs_create_file("command", 0200, d_testing, NULL,
                                    &tt_command_fops);

        return 0;
}
module_init(thermal_testing_init);

static void __exit thermal_testing_exit(void)
{
        debugfs_remove(d_testing);
        tt_zone_cleanup();
}
module_exit(thermal_testing_exit);

MODULE_DESCRIPTION("Thermal core testing facility");
MODULE_LICENSE("GPL v2");