162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0-only
262306a36Sopenharmony_ci/*
362306a36Sopenharmony_ci * Copyright (c) 2011-2014, The Linux Foundation. All rights reserved.
462306a36Sopenharmony_ci * Copyright (c) 2014,2015, Linaro Ltd.
562306a36Sopenharmony_ci *
662306a36Sopenharmony_ci * SAW power controller driver
762306a36Sopenharmony_ci */
862306a36Sopenharmony_ci
962306a36Sopenharmony_ci#include <linux/kernel.h>
1062306a36Sopenharmony_ci#include <linux/init.h>
1162306a36Sopenharmony_ci#include <linux/io.h>
1262306a36Sopenharmony_ci#include <linux/slab.h>
1362306a36Sopenharmony_ci#include <linux/of.h>
1462306a36Sopenharmony_ci#include <linux/of_platform.h>
1562306a36Sopenharmony_ci#include <linux/err.h>
1662306a36Sopenharmony_ci#include <linux/platform_device.h>
1762306a36Sopenharmony_ci#include <linux/cpuidle.h>
1862306a36Sopenharmony_ci#include <linux/cpu_pm.h>
1962306a36Sopenharmony_ci#include <linux/firmware/qcom/qcom_scm.h>
2062306a36Sopenharmony_ci#include <soc/qcom/spm.h>
2162306a36Sopenharmony_ci
2262306a36Sopenharmony_ci#include <asm/proc-fns.h>
2362306a36Sopenharmony_ci#include <asm/suspend.h>
2462306a36Sopenharmony_ci
2562306a36Sopenharmony_ci#include "dt_idle_states.h"
2662306a36Sopenharmony_ci
2762306a36Sopenharmony_cistruct cpuidle_qcom_spm_data {
2862306a36Sopenharmony_ci	struct cpuidle_driver cpuidle_driver;
2962306a36Sopenharmony_ci	struct spm_driver_data *spm;
3062306a36Sopenharmony_ci};
3162306a36Sopenharmony_ci
3262306a36Sopenharmony_cistatic int qcom_pm_collapse(unsigned long int unused)
3362306a36Sopenharmony_ci{
3462306a36Sopenharmony_ci	qcom_scm_cpu_power_down(QCOM_SCM_CPU_PWR_DOWN_L2_ON);
3562306a36Sopenharmony_ci
3662306a36Sopenharmony_ci	/*
3762306a36Sopenharmony_ci	 * Returns here only if there was a pending interrupt and we did not
3862306a36Sopenharmony_ci	 * power down as a result.
3962306a36Sopenharmony_ci	 */
4062306a36Sopenharmony_ci	return -1;
4162306a36Sopenharmony_ci}
4262306a36Sopenharmony_ci
4362306a36Sopenharmony_cistatic int qcom_cpu_spc(struct spm_driver_data *drv)
4462306a36Sopenharmony_ci{
4562306a36Sopenharmony_ci	int ret;
4662306a36Sopenharmony_ci
4762306a36Sopenharmony_ci	spm_set_low_power_mode(drv, PM_SLEEP_MODE_SPC);
4862306a36Sopenharmony_ci	ret = cpu_suspend(0, qcom_pm_collapse);
4962306a36Sopenharmony_ci	/*
5062306a36Sopenharmony_ci	 * ARM common code executes WFI without calling into our driver and
5162306a36Sopenharmony_ci	 * if the SPM mode is not reset, then we may accidently power down the
5262306a36Sopenharmony_ci	 * cpu when we intended only to gate the cpu clock.
5362306a36Sopenharmony_ci	 * Ensure the state is set to standby before returning.
5462306a36Sopenharmony_ci	 */
5562306a36Sopenharmony_ci	spm_set_low_power_mode(drv, PM_SLEEP_MODE_STBY);
5662306a36Sopenharmony_ci
5762306a36Sopenharmony_ci	return ret;
5862306a36Sopenharmony_ci}
5962306a36Sopenharmony_ci
6062306a36Sopenharmony_cistatic __cpuidle int spm_enter_idle_state(struct cpuidle_device *dev,
6162306a36Sopenharmony_ci					  struct cpuidle_driver *drv, int idx)
6262306a36Sopenharmony_ci{
6362306a36Sopenharmony_ci	struct cpuidle_qcom_spm_data *data = container_of(drv, struct cpuidle_qcom_spm_data,
6462306a36Sopenharmony_ci							  cpuidle_driver);
6562306a36Sopenharmony_ci
6662306a36Sopenharmony_ci	return CPU_PM_CPU_IDLE_ENTER_PARAM(qcom_cpu_spc, idx, data->spm);
6762306a36Sopenharmony_ci}
6862306a36Sopenharmony_ci
6962306a36Sopenharmony_cistatic struct cpuidle_driver qcom_spm_idle_driver = {
7062306a36Sopenharmony_ci	.name = "qcom_spm",
7162306a36Sopenharmony_ci	.owner = THIS_MODULE,
7262306a36Sopenharmony_ci	.states[0] = {
7362306a36Sopenharmony_ci		.enter			= spm_enter_idle_state,
7462306a36Sopenharmony_ci		.exit_latency		= 1,
7562306a36Sopenharmony_ci		.target_residency	= 1,
7662306a36Sopenharmony_ci		.power_usage		= UINT_MAX,
7762306a36Sopenharmony_ci		.name			= "WFI",
7862306a36Sopenharmony_ci		.desc			= "ARM WFI",
7962306a36Sopenharmony_ci	}
8062306a36Sopenharmony_ci};
8162306a36Sopenharmony_ci
8262306a36Sopenharmony_cistatic const struct of_device_id qcom_idle_state_match[] = {
8362306a36Sopenharmony_ci	{ .compatible = "qcom,idle-state-spc", .data = spm_enter_idle_state },
8462306a36Sopenharmony_ci	{ },
8562306a36Sopenharmony_ci};
8662306a36Sopenharmony_ci
8762306a36Sopenharmony_cistatic int spm_cpuidle_register(struct device *cpuidle_dev, int cpu)
8862306a36Sopenharmony_ci{
8962306a36Sopenharmony_ci	struct platform_device *pdev = NULL;
9062306a36Sopenharmony_ci	struct device_node *cpu_node, *saw_node;
9162306a36Sopenharmony_ci	struct cpuidle_qcom_spm_data *data = NULL;
9262306a36Sopenharmony_ci	int ret;
9362306a36Sopenharmony_ci
9462306a36Sopenharmony_ci	cpu_node = of_cpu_device_node_get(cpu);
9562306a36Sopenharmony_ci	if (!cpu_node)
9662306a36Sopenharmony_ci		return -ENODEV;
9762306a36Sopenharmony_ci
9862306a36Sopenharmony_ci	saw_node = of_parse_phandle(cpu_node, "qcom,saw", 0);
9962306a36Sopenharmony_ci	if (!saw_node)
10062306a36Sopenharmony_ci		return -ENODEV;
10162306a36Sopenharmony_ci
10262306a36Sopenharmony_ci	pdev = of_find_device_by_node(saw_node);
10362306a36Sopenharmony_ci	of_node_put(saw_node);
10462306a36Sopenharmony_ci	of_node_put(cpu_node);
10562306a36Sopenharmony_ci	if (!pdev)
10662306a36Sopenharmony_ci		return -ENODEV;
10762306a36Sopenharmony_ci
10862306a36Sopenharmony_ci	data = devm_kzalloc(cpuidle_dev, sizeof(*data), GFP_KERNEL);
10962306a36Sopenharmony_ci	if (!data)
11062306a36Sopenharmony_ci		return -ENOMEM;
11162306a36Sopenharmony_ci
11262306a36Sopenharmony_ci	data->spm = dev_get_drvdata(&pdev->dev);
11362306a36Sopenharmony_ci	if (!data->spm)
11462306a36Sopenharmony_ci		return -EINVAL;
11562306a36Sopenharmony_ci
11662306a36Sopenharmony_ci	data->cpuidle_driver = qcom_spm_idle_driver;
11762306a36Sopenharmony_ci	data->cpuidle_driver.cpumask = (struct cpumask *)cpumask_of(cpu);
11862306a36Sopenharmony_ci
11962306a36Sopenharmony_ci	ret = dt_init_idle_driver(&data->cpuidle_driver,
12062306a36Sopenharmony_ci				  qcom_idle_state_match, 1);
12162306a36Sopenharmony_ci	if (ret <= 0)
12262306a36Sopenharmony_ci		return ret ? : -ENODEV;
12362306a36Sopenharmony_ci
12462306a36Sopenharmony_ci	return cpuidle_register(&data->cpuidle_driver, NULL);
12562306a36Sopenharmony_ci}
12662306a36Sopenharmony_ci
12762306a36Sopenharmony_cistatic int spm_cpuidle_drv_probe(struct platform_device *pdev)
12862306a36Sopenharmony_ci{
12962306a36Sopenharmony_ci	int cpu, ret;
13062306a36Sopenharmony_ci
13162306a36Sopenharmony_ci	if (!qcom_scm_is_available())
13262306a36Sopenharmony_ci		return -EPROBE_DEFER;
13362306a36Sopenharmony_ci
13462306a36Sopenharmony_ci	ret = qcom_scm_set_warm_boot_addr(cpu_resume_arm);
13562306a36Sopenharmony_ci	if (ret)
13662306a36Sopenharmony_ci		return dev_err_probe(&pdev->dev, ret, "set warm boot addr failed");
13762306a36Sopenharmony_ci
13862306a36Sopenharmony_ci	for_each_possible_cpu(cpu) {
13962306a36Sopenharmony_ci		ret = spm_cpuidle_register(&pdev->dev, cpu);
14062306a36Sopenharmony_ci		if (ret && ret != -ENODEV) {
14162306a36Sopenharmony_ci			dev_err(&pdev->dev,
14262306a36Sopenharmony_ci				"Cannot register for CPU%d: %d\n", cpu, ret);
14362306a36Sopenharmony_ci		}
14462306a36Sopenharmony_ci	}
14562306a36Sopenharmony_ci
14662306a36Sopenharmony_ci	return 0;
14762306a36Sopenharmony_ci}
14862306a36Sopenharmony_ci
14962306a36Sopenharmony_cistatic struct platform_driver spm_cpuidle_driver = {
15062306a36Sopenharmony_ci	.probe = spm_cpuidle_drv_probe,
15162306a36Sopenharmony_ci	.driver = {
15262306a36Sopenharmony_ci		.name = "qcom-spm-cpuidle",
15362306a36Sopenharmony_ci		.suppress_bind_attrs = true,
15462306a36Sopenharmony_ci	},
15562306a36Sopenharmony_ci};
15662306a36Sopenharmony_ci
15762306a36Sopenharmony_cistatic bool __init qcom_spm_find_any_cpu(void)
15862306a36Sopenharmony_ci{
15962306a36Sopenharmony_ci	struct device_node *cpu_node, *saw_node;
16062306a36Sopenharmony_ci
16162306a36Sopenharmony_ci	for_each_of_cpu_node(cpu_node) {
16262306a36Sopenharmony_ci		saw_node = of_parse_phandle(cpu_node, "qcom,saw", 0);
16362306a36Sopenharmony_ci		if (of_device_is_available(saw_node)) {
16462306a36Sopenharmony_ci			of_node_put(saw_node);
16562306a36Sopenharmony_ci			of_node_put(cpu_node);
16662306a36Sopenharmony_ci			return true;
16762306a36Sopenharmony_ci		}
16862306a36Sopenharmony_ci		of_node_put(saw_node);
16962306a36Sopenharmony_ci	}
17062306a36Sopenharmony_ci	return false;
17162306a36Sopenharmony_ci}
17262306a36Sopenharmony_ci
17362306a36Sopenharmony_cistatic int __init qcom_spm_cpuidle_init(void)
17462306a36Sopenharmony_ci{
17562306a36Sopenharmony_ci	struct platform_device *pdev;
17662306a36Sopenharmony_ci	int ret;
17762306a36Sopenharmony_ci
17862306a36Sopenharmony_ci	ret = platform_driver_register(&spm_cpuidle_driver);
17962306a36Sopenharmony_ci	if (ret)
18062306a36Sopenharmony_ci		return ret;
18162306a36Sopenharmony_ci
18262306a36Sopenharmony_ci	/* Make sure there is actually any CPU managed by the SPM */
18362306a36Sopenharmony_ci	if (!qcom_spm_find_any_cpu())
18462306a36Sopenharmony_ci		return 0;
18562306a36Sopenharmony_ci
18662306a36Sopenharmony_ci	pdev = platform_device_register_simple("qcom-spm-cpuidle",
18762306a36Sopenharmony_ci					       -1, NULL, 0);
18862306a36Sopenharmony_ci	if (IS_ERR(pdev)) {
18962306a36Sopenharmony_ci		platform_driver_unregister(&spm_cpuidle_driver);
19062306a36Sopenharmony_ci		return PTR_ERR(pdev);
19162306a36Sopenharmony_ci	}
19262306a36Sopenharmony_ci
19362306a36Sopenharmony_ci	return 0;
19462306a36Sopenharmony_ci}
19562306a36Sopenharmony_cidevice_initcall(qcom_spm_cpuidle_init);
196