162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0-only
262306a36Sopenharmony_ci/*
362306a36Sopenharmony_ci * Power off driver for ams AS3722 device.
462306a36Sopenharmony_ci *
562306a36Sopenharmony_ci * Copyright (c) 2013, NVIDIA CORPORATION.  All rights reserved.
662306a36Sopenharmony_ci *
762306a36Sopenharmony_ci * Author: Laxman Dewangan <ldewangan@nvidia.com>
862306a36Sopenharmony_ci */
962306a36Sopenharmony_ci
1062306a36Sopenharmony_ci#include <linux/mfd/as3722.h>
1162306a36Sopenharmony_ci#include <linux/module.h>
1262306a36Sopenharmony_ci#include <linux/of.h>
1362306a36Sopenharmony_ci#include <linux/platform_device.h>
1462306a36Sopenharmony_ci#include <linux/slab.h>
1562306a36Sopenharmony_ci
1662306a36Sopenharmony_cistruct as3722_poweroff {
1762306a36Sopenharmony_ci	struct device *dev;
1862306a36Sopenharmony_ci	struct as3722 *as3722;
1962306a36Sopenharmony_ci};
2062306a36Sopenharmony_ci
2162306a36Sopenharmony_cistatic struct as3722_poweroff *as3722_pm_poweroff;
2262306a36Sopenharmony_ci
2362306a36Sopenharmony_cistatic void as3722_pm_power_off(void)
2462306a36Sopenharmony_ci{
2562306a36Sopenharmony_ci	int ret;
2662306a36Sopenharmony_ci
2762306a36Sopenharmony_ci	if (!as3722_pm_poweroff) {
2862306a36Sopenharmony_ci		pr_err("AS3722 poweroff is not initialised\n");
2962306a36Sopenharmony_ci		return;
3062306a36Sopenharmony_ci	}
3162306a36Sopenharmony_ci
3262306a36Sopenharmony_ci	ret = as3722_update_bits(as3722_pm_poweroff->as3722,
3362306a36Sopenharmony_ci		AS3722_RESET_CONTROL_REG, AS3722_POWER_OFF, AS3722_POWER_OFF);
3462306a36Sopenharmony_ci	if (ret < 0)
3562306a36Sopenharmony_ci		dev_err(as3722_pm_poweroff->dev,
3662306a36Sopenharmony_ci			"RESET_CONTROL_REG update failed, %d\n", ret);
3762306a36Sopenharmony_ci}
3862306a36Sopenharmony_ci
3962306a36Sopenharmony_cistatic int as3722_poweroff_probe(struct platform_device *pdev)
4062306a36Sopenharmony_ci{
4162306a36Sopenharmony_ci	struct as3722_poweroff *as3722_poweroff;
4262306a36Sopenharmony_ci	struct device_node *np = pdev->dev.parent->of_node;
4362306a36Sopenharmony_ci
4462306a36Sopenharmony_ci	if (!np)
4562306a36Sopenharmony_ci		return -EINVAL;
4662306a36Sopenharmony_ci
4762306a36Sopenharmony_ci	if (!of_property_read_bool(np, "ams,system-power-controller"))
4862306a36Sopenharmony_ci		return 0;
4962306a36Sopenharmony_ci
5062306a36Sopenharmony_ci	as3722_poweroff = devm_kzalloc(&pdev->dev, sizeof(*as3722_poweroff),
5162306a36Sopenharmony_ci				GFP_KERNEL);
5262306a36Sopenharmony_ci	if (!as3722_poweroff)
5362306a36Sopenharmony_ci		return -ENOMEM;
5462306a36Sopenharmony_ci
5562306a36Sopenharmony_ci	as3722_poweroff->as3722 = dev_get_drvdata(pdev->dev.parent);
5662306a36Sopenharmony_ci	as3722_poweroff->dev = &pdev->dev;
5762306a36Sopenharmony_ci	as3722_pm_poweroff = as3722_poweroff;
5862306a36Sopenharmony_ci	if (!pm_power_off)
5962306a36Sopenharmony_ci		pm_power_off = as3722_pm_power_off;
6062306a36Sopenharmony_ci
6162306a36Sopenharmony_ci	return 0;
6262306a36Sopenharmony_ci}
6362306a36Sopenharmony_ci
6462306a36Sopenharmony_cistatic int as3722_poweroff_remove(struct platform_device *pdev)
6562306a36Sopenharmony_ci{
6662306a36Sopenharmony_ci	if (pm_power_off == as3722_pm_power_off)
6762306a36Sopenharmony_ci		pm_power_off = NULL;
6862306a36Sopenharmony_ci	as3722_pm_poweroff = NULL;
6962306a36Sopenharmony_ci
7062306a36Sopenharmony_ci	return 0;
7162306a36Sopenharmony_ci}
7262306a36Sopenharmony_ci
7362306a36Sopenharmony_cistatic struct platform_driver as3722_poweroff_driver = {
7462306a36Sopenharmony_ci	.driver = {
7562306a36Sopenharmony_ci		.name = "as3722-power-off",
7662306a36Sopenharmony_ci	},
7762306a36Sopenharmony_ci	.probe = as3722_poweroff_probe,
7862306a36Sopenharmony_ci	.remove = as3722_poweroff_remove,
7962306a36Sopenharmony_ci};
8062306a36Sopenharmony_ci
8162306a36Sopenharmony_cimodule_platform_driver(as3722_poweroff_driver);
8262306a36Sopenharmony_ci
8362306a36Sopenharmony_ciMODULE_DESCRIPTION("Power off driver for ams AS3722 PMIC Device");
8462306a36Sopenharmony_ciMODULE_ALIAS("platform:as3722-power-off");
8562306a36Sopenharmony_ciMODULE_AUTHOR("Laxman Dewangan <ldewangan@nvidia.com>");
86