| // SPDX-License-Identifier: GPL-2.0 | 
 | /* | 
 |  * Copyright (C) 2019 MediaTek Inc. | 
 |  * Author Freddy Hsin <freddy.hsin@mediatek.com> | 
 |  */ | 
 |  | 
 | #include <linux/init.h> | 
 | #include <linux/module.h> | 
 | #include <linux/kernel.h> | 
 | #include <linux/of.h> | 
 | #include <linux/platform_device.h> | 
 | #include <linux/reboot.h> | 
 | #include <linux/regmap.h> | 
 | #include <linux/of_address.h> | 
 | #include <linux/reboot-mode.h> | 
 |  | 
 | static const struct regmap_config mtk_regmap_config = { | 
 | 	.reg_bits = 32, | 
 | 	.val_bits = 32, | 
 | 	.reg_stride = 4, | 
 | }; | 
 |  | 
 | struct mtk_reboot_mode { | 
 | 	struct regmap *map; | 
 | 	struct reboot_mode_driver reboot; | 
 | 	u32 offset; | 
 | 	u32 mask; | 
 | }; | 
 |  | 
 | static int mtk_reboot_mode_write(struct reboot_mode_driver *reboot, | 
 | 				 unsigned int magic) | 
 | { | 
 | 	struct mtk_reboot_mode *mtk_rbm; | 
 | 	int ret; | 
 |  | 
 | 	mtk_rbm = container_of(reboot, struct mtk_reboot_mode, reboot); | 
 |  | 
 | 	ret = regmap_update_bits(mtk_rbm->map, mtk_rbm->offset, | 
 | 				 mtk_rbm->mask, magic); | 
 | 	if (ret < 0) | 
 | 		dev_info(reboot->dev, "update reboot mode bits failed\n"); | 
 |  | 
 | 	return ret; | 
 | } | 
 |  | 
 | static int mtk_regmap_lookup_by_phandle(struct device *dev, | 
 | 					struct mtk_reboot_mode *mtk_rbm) | 
 | { | 
 | 	struct device_node *toprgu_np; | 
 | 	struct device_node *np = dev->of_node; | 
 | 	void __iomem *base; | 
 |  | 
 | 	toprgu_np = of_parse_phandle(np, "regmap", 0); | 
 | 	if (!toprgu_np) | 
 | 		return -EINVAL; | 
 |  | 
 | 	if (!of_device_is_compatible(toprgu_np, "mediatek,toprgu")) | 
 | 		return -EINVAL; | 
 |  | 
 | 	base = of_iomap(toprgu_np, 0); | 
 | 	if (IS_ERR(base)) | 
 | 		return PTR_ERR(base); | 
 |  | 
 | 	mtk_rbm->map = devm_regmap_init_mmio(dev, base, | 
 | 					     &mtk_regmap_config); | 
 | 	return PTR_ERR_OR_ZERO(mtk_rbm->map); | 
 | } | 
 |  | 
 | static int mtk_reboot_mode_probe(struct platform_device *pdev) | 
 | { | 
 | 	int ret; | 
 | 	struct mtk_reboot_mode *mtk_rbm; | 
 |  | 
 | 	mtk_rbm = devm_kzalloc(&pdev->dev, sizeof(*mtk_rbm), GFP_KERNEL); | 
 | 	if (!mtk_rbm) | 
 | 		return -ENOMEM; | 
 |  | 
 | 	mtk_rbm->reboot.dev = &pdev->dev; | 
 | 	mtk_rbm->reboot.write = mtk_reboot_mode_write; | 
 | 	mtk_rbm->mask = 0xf; | 
 |  | 
 | 	ret = mtk_regmap_lookup_by_phandle(&pdev->dev, mtk_rbm); | 
 | 	if (ret) { | 
 | 		dev_info(&pdev->dev, "Couldn't create the toprgu regmap\n"); | 
 | 		return -EINVAL; | 
 | 	} | 
 |  | 
 | 	if (of_property_read_u32(pdev->dev.of_node, "offset", | 
 | 				 &mtk_rbm->offset)) | 
 | 		return -EINVAL; | 
 |  | 
 | 	of_property_read_u32(pdev->dev.of_node, "mask", &mtk_rbm->mask); | 
 |  | 
 | 	ret = devm_reboot_mode_register(&pdev->dev, &mtk_rbm->reboot); | 
 | 	if (ret) | 
 | 		dev_info(&pdev->dev, "can't register reboot mode\n"); | 
 |  | 
 | 	return ret; | 
 | } | 
 |  | 
 | static const struct of_device_id mtk_reboot_mode_of_match[] = { | 
 | 	{ .compatible = "toprgu-reboot-mode" }, | 
 | 	{} | 
 | }; | 
 | MODULE_DEVICE_TABLE(of, mtk_reboot_mode_of_match); | 
 |  | 
 | static struct platform_driver mtk_reboot_mode_driver = { | 
 | 	.probe = mtk_reboot_mode_probe, | 
 | 	.driver = { | 
 | 		.name = "toprgu-reboot-mode", | 
 | 		.of_match_table = mtk_reboot_mode_of_match, | 
 | 	}, | 
 | }; | 
 | module_platform_driver(mtk_reboot_mode_driver); | 
 |  | 
 | MODULE_AUTHOR("Freddy Hsin <freddy.hsin@mediatek.com>"); | 
 | MODULE_DESCRIPTION("Mediatek reboot mode driver"); | 
 | MODULE_LICENSE("GPL v2"); |