162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0-only
262306a36Sopenharmony_ci//
362306a36Sopenharmony_ci// Copyright 2020 Google LLC.
462306a36Sopenharmony_ci
562306a36Sopenharmony_ci#include <linux/module.h>
662306a36Sopenharmony_ci#include <linux/of.h>
762306a36Sopenharmony_ci#include <linux/platform_data/cros_ec_proto.h>
862306a36Sopenharmony_ci#include <linux/platform_device.h>
962306a36Sopenharmony_ci#include <linux/regulator/driver.h>
1062306a36Sopenharmony_ci#include <linux/regulator/machine.h>
1162306a36Sopenharmony_ci#include <linux/regulator/of_regulator.h>
1262306a36Sopenharmony_ci#include <linux/slab.h>
1362306a36Sopenharmony_ci
1462306a36Sopenharmony_cistruct cros_ec_regulator_data {
1562306a36Sopenharmony_ci	struct regulator_desc desc;
1662306a36Sopenharmony_ci	struct regulator_dev *dev;
1762306a36Sopenharmony_ci	struct cros_ec_device *ec_dev;
1862306a36Sopenharmony_ci
1962306a36Sopenharmony_ci	u32 index;
2062306a36Sopenharmony_ci
2162306a36Sopenharmony_ci	u16 *voltages_mV;
2262306a36Sopenharmony_ci	u16 num_voltages;
2362306a36Sopenharmony_ci};
2462306a36Sopenharmony_ci
2562306a36Sopenharmony_cistatic int cros_ec_regulator_enable(struct regulator_dev *dev)
2662306a36Sopenharmony_ci{
2762306a36Sopenharmony_ci	struct cros_ec_regulator_data *data = rdev_get_drvdata(dev);
2862306a36Sopenharmony_ci	struct ec_params_regulator_enable cmd = {
2962306a36Sopenharmony_ci		.index = data->index,
3062306a36Sopenharmony_ci		.enable = 1,
3162306a36Sopenharmony_ci	};
3262306a36Sopenharmony_ci
3362306a36Sopenharmony_ci	return cros_ec_cmd(data->ec_dev, 0, EC_CMD_REGULATOR_ENABLE, &cmd,
3462306a36Sopenharmony_ci			   sizeof(cmd), NULL, 0);
3562306a36Sopenharmony_ci}
3662306a36Sopenharmony_ci
3762306a36Sopenharmony_cistatic int cros_ec_regulator_disable(struct regulator_dev *dev)
3862306a36Sopenharmony_ci{
3962306a36Sopenharmony_ci	struct cros_ec_regulator_data *data = rdev_get_drvdata(dev);
4062306a36Sopenharmony_ci	struct ec_params_regulator_enable cmd = {
4162306a36Sopenharmony_ci		.index = data->index,
4262306a36Sopenharmony_ci		.enable = 0,
4362306a36Sopenharmony_ci	};
4462306a36Sopenharmony_ci
4562306a36Sopenharmony_ci	return cros_ec_cmd(data->ec_dev, 0, EC_CMD_REGULATOR_ENABLE, &cmd,
4662306a36Sopenharmony_ci			   sizeof(cmd), NULL, 0);
4762306a36Sopenharmony_ci}
4862306a36Sopenharmony_ci
4962306a36Sopenharmony_cistatic int cros_ec_regulator_is_enabled(struct regulator_dev *dev)
5062306a36Sopenharmony_ci{
5162306a36Sopenharmony_ci	struct cros_ec_regulator_data *data = rdev_get_drvdata(dev);
5262306a36Sopenharmony_ci	struct ec_params_regulator_is_enabled cmd = {
5362306a36Sopenharmony_ci		.index = data->index,
5462306a36Sopenharmony_ci	};
5562306a36Sopenharmony_ci	struct ec_response_regulator_is_enabled resp;
5662306a36Sopenharmony_ci	int ret;
5762306a36Sopenharmony_ci
5862306a36Sopenharmony_ci	ret = cros_ec_cmd(data->ec_dev, 0, EC_CMD_REGULATOR_IS_ENABLED, &cmd,
5962306a36Sopenharmony_ci			  sizeof(cmd), &resp, sizeof(resp));
6062306a36Sopenharmony_ci	if (ret < 0)
6162306a36Sopenharmony_ci		return ret;
6262306a36Sopenharmony_ci	return resp.enabled;
6362306a36Sopenharmony_ci}
6462306a36Sopenharmony_ci
6562306a36Sopenharmony_cistatic int cros_ec_regulator_list_voltage(struct regulator_dev *dev,
6662306a36Sopenharmony_ci					  unsigned int selector)
6762306a36Sopenharmony_ci{
6862306a36Sopenharmony_ci	struct cros_ec_regulator_data *data = rdev_get_drvdata(dev);
6962306a36Sopenharmony_ci
7062306a36Sopenharmony_ci	if (selector >= data->num_voltages)
7162306a36Sopenharmony_ci		return -EINVAL;
7262306a36Sopenharmony_ci
7362306a36Sopenharmony_ci	return data->voltages_mV[selector] * 1000;
7462306a36Sopenharmony_ci}
7562306a36Sopenharmony_ci
7662306a36Sopenharmony_cistatic int cros_ec_regulator_get_voltage(struct regulator_dev *dev)
7762306a36Sopenharmony_ci{
7862306a36Sopenharmony_ci	struct cros_ec_regulator_data *data = rdev_get_drvdata(dev);
7962306a36Sopenharmony_ci	struct ec_params_regulator_get_voltage cmd = {
8062306a36Sopenharmony_ci		.index = data->index,
8162306a36Sopenharmony_ci	};
8262306a36Sopenharmony_ci	struct ec_response_regulator_get_voltage resp;
8362306a36Sopenharmony_ci	int ret;
8462306a36Sopenharmony_ci
8562306a36Sopenharmony_ci	ret = cros_ec_cmd(data->ec_dev, 0, EC_CMD_REGULATOR_GET_VOLTAGE, &cmd,
8662306a36Sopenharmony_ci			  sizeof(cmd), &resp, sizeof(resp));
8762306a36Sopenharmony_ci	if (ret < 0)
8862306a36Sopenharmony_ci		return ret;
8962306a36Sopenharmony_ci	return resp.voltage_mv * 1000;
9062306a36Sopenharmony_ci}
9162306a36Sopenharmony_ci
9262306a36Sopenharmony_cistatic int cros_ec_regulator_set_voltage(struct regulator_dev *dev, int min_uV,
9362306a36Sopenharmony_ci					 int max_uV, unsigned int *selector)
9462306a36Sopenharmony_ci{
9562306a36Sopenharmony_ci	struct cros_ec_regulator_data *data = rdev_get_drvdata(dev);
9662306a36Sopenharmony_ci	int min_mV = DIV_ROUND_UP(min_uV, 1000);
9762306a36Sopenharmony_ci	int max_mV = max_uV / 1000;
9862306a36Sopenharmony_ci	struct ec_params_regulator_set_voltage cmd = {
9962306a36Sopenharmony_ci		.index = data->index,
10062306a36Sopenharmony_ci		.min_mv = min_mV,
10162306a36Sopenharmony_ci		.max_mv = max_mV,
10262306a36Sopenharmony_ci	};
10362306a36Sopenharmony_ci
10462306a36Sopenharmony_ci	/*
10562306a36Sopenharmony_ci	 * This can happen when the given range [min_uV, max_uV] doesn't
10662306a36Sopenharmony_ci	 * contain any voltage that can be represented exactly in mV.
10762306a36Sopenharmony_ci	 */
10862306a36Sopenharmony_ci	if (min_mV > max_mV)
10962306a36Sopenharmony_ci		return -EINVAL;
11062306a36Sopenharmony_ci
11162306a36Sopenharmony_ci	return cros_ec_cmd(data->ec_dev, 0, EC_CMD_REGULATOR_SET_VOLTAGE, &cmd,
11262306a36Sopenharmony_ci			   sizeof(cmd), NULL, 0);
11362306a36Sopenharmony_ci}
11462306a36Sopenharmony_ci
11562306a36Sopenharmony_cistatic const struct regulator_ops cros_ec_regulator_voltage_ops = {
11662306a36Sopenharmony_ci	.enable = cros_ec_regulator_enable,
11762306a36Sopenharmony_ci	.disable = cros_ec_regulator_disable,
11862306a36Sopenharmony_ci	.is_enabled = cros_ec_regulator_is_enabled,
11962306a36Sopenharmony_ci	.list_voltage = cros_ec_regulator_list_voltage,
12062306a36Sopenharmony_ci	.get_voltage = cros_ec_regulator_get_voltage,
12162306a36Sopenharmony_ci	.set_voltage = cros_ec_regulator_set_voltage,
12262306a36Sopenharmony_ci};
12362306a36Sopenharmony_ci
12462306a36Sopenharmony_cistatic int cros_ec_regulator_init_info(struct device *dev,
12562306a36Sopenharmony_ci				       struct cros_ec_regulator_data *data)
12662306a36Sopenharmony_ci{
12762306a36Sopenharmony_ci	struct ec_params_regulator_get_info cmd = {
12862306a36Sopenharmony_ci		.index = data->index,
12962306a36Sopenharmony_ci	};
13062306a36Sopenharmony_ci	struct ec_response_regulator_get_info resp;
13162306a36Sopenharmony_ci	int ret;
13262306a36Sopenharmony_ci
13362306a36Sopenharmony_ci	ret = cros_ec_cmd(data->ec_dev, 0, EC_CMD_REGULATOR_GET_INFO, &cmd,
13462306a36Sopenharmony_ci			  sizeof(cmd), &resp, sizeof(resp));
13562306a36Sopenharmony_ci	if (ret < 0)
13662306a36Sopenharmony_ci		return ret;
13762306a36Sopenharmony_ci
13862306a36Sopenharmony_ci	data->num_voltages =
13962306a36Sopenharmony_ci		min_t(u16, ARRAY_SIZE(resp.voltages_mv), resp.num_voltages);
14062306a36Sopenharmony_ci	data->voltages_mV =
14162306a36Sopenharmony_ci		devm_kmemdup(dev, resp.voltages_mv,
14262306a36Sopenharmony_ci			     sizeof(u16) * data->num_voltages, GFP_KERNEL);
14362306a36Sopenharmony_ci	if (!data->voltages_mV)
14462306a36Sopenharmony_ci		return -ENOMEM;
14562306a36Sopenharmony_ci
14662306a36Sopenharmony_ci	data->desc.n_voltages = data->num_voltages;
14762306a36Sopenharmony_ci
14862306a36Sopenharmony_ci	/* Make sure the returned name is always a valid string */
14962306a36Sopenharmony_ci	resp.name[ARRAY_SIZE(resp.name) - 1] = '\0';
15062306a36Sopenharmony_ci	data->desc.name = devm_kstrdup(dev, resp.name, GFP_KERNEL);
15162306a36Sopenharmony_ci	if (!data->desc.name)
15262306a36Sopenharmony_ci		return -ENOMEM;
15362306a36Sopenharmony_ci
15462306a36Sopenharmony_ci	return 0;
15562306a36Sopenharmony_ci}
15662306a36Sopenharmony_ci
15762306a36Sopenharmony_cistatic int cros_ec_regulator_probe(struct platform_device *pdev)
15862306a36Sopenharmony_ci{
15962306a36Sopenharmony_ci	struct device *dev = &pdev->dev;
16062306a36Sopenharmony_ci	struct device_node *np = dev->of_node;
16162306a36Sopenharmony_ci	struct cros_ec_regulator_data *drvdata;
16262306a36Sopenharmony_ci	struct regulator_init_data *init_data;
16362306a36Sopenharmony_ci	struct regulator_config cfg = {};
16462306a36Sopenharmony_ci	struct regulator_desc *desc;
16562306a36Sopenharmony_ci	int ret;
16662306a36Sopenharmony_ci
16762306a36Sopenharmony_ci	drvdata = devm_kzalloc(
16862306a36Sopenharmony_ci		&pdev->dev, sizeof(struct cros_ec_regulator_data), GFP_KERNEL);
16962306a36Sopenharmony_ci	if (!drvdata)
17062306a36Sopenharmony_ci		return -ENOMEM;
17162306a36Sopenharmony_ci
17262306a36Sopenharmony_ci	drvdata->ec_dev = dev_get_drvdata(dev->parent);
17362306a36Sopenharmony_ci	desc = &drvdata->desc;
17462306a36Sopenharmony_ci
17562306a36Sopenharmony_ci	init_data = of_get_regulator_init_data(dev, np, desc);
17662306a36Sopenharmony_ci	if (!init_data)
17762306a36Sopenharmony_ci		return -EINVAL;
17862306a36Sopenharmony_ci
17962306a36Sopenharmony_ci	ret = of_property_read_u32(np, "reg", &drvdata->index);
18062306a36Sopenharmony_ci	if (ret < 0)
18162306a36Sopenharmony_ci		return ret;
18262306a36Sopenharmony_ci
18362306a36Sopenharmony_ci	desc->owner = THIS_MODULE;
18462306a36Sopenharmony_ci	desc->type = REGULATOR_VOLTAGE;
18562306a36Sopenharmony_ci	desc->ops = &cros_ec_regulator_voltage_ops;
18662306a36Sopenharmony_ci
18762306a36Sopenharmony_ci	ret = cros_ec_regulator_init_info(dev, drvdata);
18862306a36Sopenharmony_ci	if (ret < 0)
18962306a36Sopenharmony_ci		return ret;
19062306a36Sopenharmony_ci
19162306a36Sopenharmony_ci	cfg.dev = &pdev->dev;
19262306a36Sopenharmony_ci	cfg.init_data = init_data;
19362306a36Sopenharmony_ci	cfg.driver_data = drvdata;
19462306a36Sopenharmony_ci	cfg.of_node = np;
19562306a36Sopenharmony_ci
19662306a36Sopenharmony_ci	drvdata->dev = devm_regulator_register(dev, &drvdata->desc, &cfg);
19762306a36Sopenharmony_ci	if (IS_ERR(drvdata->dev)) {
19862306a36Sopenharmony_ci		ret = PTR_ERR(drvdata->dev);
19962306a36Sopenharmony_ci		dev_err(&pdev->dev, "Failed to register regulator: %d\n", ret);
20062306a36Sopenharmony_ci		return ret;
20162306a36Sopenharmony_ci	}
20262306a36Sopenharmony_ci
20362306a36Sopenharmony_ci	platform_set_drvdata(pdev, drvdata);
20462306a36Sopenharmony_ci
20562306a36Sopenharmony_ci	return 0;
20662306a36Sopenharmony_ci}
20762306a36Sopenharmony_ci
20862306a36Sopenharmony_cistatic const struct of_device_id regulator_cros_ec_of_match[] = {
20962306a36Sopenharmony_ci	{ .compatible = "google,cros-ec-regulator", },
21062306a36Sopenharmony_ci	{}
21162306a36Sopenharmony_ci};
21262306a36Sopenharmony_ciMODULE_DEVICE_TABLE(of, regulator_cros_ec_of_match);
21362306a36Sopenharmony_ci
21462306a36Sopenharmony_cistatic struct platform_driver cros_ec_regulator_driver = {
21562306a36Sopenharmony_ci	.probe		= cros_ec_regulator_probe,
21662306a36Sopenharmony_ci	.driver		= {
21762306a36Sopenharmony_ci		.name		= "cros-ec-regulator",
21862306a36Sopenharmony_ci		.probe_type	= PROBE_PREFER_ASYNCHRONOUS,
21962306a36Sopenharmony_ci		.of_match_table = regulator_cros_ec_of_match,
22062306a36Sopenharmony_ci	},
22162306a36Sopenharmony_ci};
22262306a36Sopenharmony_ci
22362306a36Sopenharmony_cimodule_platform_driver(cros_ec_regulator_driver);
22462306a36Sopenharmony_ci
22562306a36Sopenharmony_ciMODULE_LICENSE("GPL v2");
22662306a36Sopenharmony_ciMODULE_DESCRIPTION("ChromeOS EC controlled regulator");
22762306a36Sopenharmony_ciMODULE_AUTHOR("Pi-Hsun Shih <pihsun@chromium.org>");
228