162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0
262306a36Sopenharmony_ci/*
362306a36Sopenharmony_ci * LinkStation power off restart driver
462306a36Sopenharmony_ci * Copyright (C) 2020 Daniel González Cabanelas <dgcbueu@gmail.com>
562306a36Sopenharmony_ci */
662306a36Sopenharmony_ci
762306a36Sopenharmony_ci#include <linux/module.h>
862306a36Sopenharmony_ci#include <linux/notifier.h>
962306a36Sopenharmony_ci#include <linux/of.h>
1062306a36Sopenharmony_ci#include <linux/of_mdio.h>
1162306a36Sopenharmony_ci#include <linux/of_platform.h>
1262306a36Sopenharmony_ci#include <linux/reboot.h>
1362306a36Sopenharmony_ci#include <linux/phy.h>
1462306a36Sopenharmony_ci
1562306a36Sopenharmony_ci/* Defines from the eth phy Marvell driver */
1662306a36Sopenharmony_ci#define MII_MARVELL_COPPER_PAGE		0
1762306a36Sopenharmony_ci#define MII_MARVELL_LED_PAGE		3
1862306a36Sopenharmony_ci#define MII_MARVELL_WOL_PAGE		17
1962306a36Sopenharmony_ci#define MII_MARVELL_PHY_PAGE		22
2062306a36Sopenharmony_ci
2162306a36Sopenharmony_ci#define MII_PHY_LED_CTRL		16
2262306a36Sopenharmony_ci#define MII_PHY_LED_POL_CTRL		17
2362306a36Sopenharmony_ci#define MII_88E1318S_PHY_LED_TCR	18
2462306a36Sopenharmony_ci#define MII_88E1318S_PHY_WOL_CTRL	16
2562306a36Sopenharmony_ci#define MII_M1011_IEVENT		19
2662306a36Sopenharmony_ci
2762306a36Sopenharmony_ci#define MII_88E1318S_PHY_LED_TCR_INTn_ENABLE		BIT(7)
2862306a36Sopenharmony_ci#define MII_88E1318S_PHY_LED_TCR_FORCE_INT		BIT(15)
2962306a36Sopenharmony_ci#define MII_88E1318S_PHY_WOL_CTRL_CLEAR_WOL_STATUS	BIT(12)
3062306a36Sopenharmony_ci#define LED2_FORCE_ON					(0x8 << 8)
3162306a36Sopenharmony_ci#define LEDMASK						GENMASK(11,8)
3262306a36Sopenharmony_ci
3362306a36Sopenharmony_ci#define MII_88E1318S_PHY_LED_POL_LED2		BIT(4)
3462306a36Sopenharmony_ci
3562306a36Sopenharmony_cistruct power_off_cfg {
3662306a36Sopenharmony_ci	char *mdio_node_name;
3762306a36Sopenharmony_ci	void (*phy_set_reg)(bool restart);
3862306a36Sopenharmony_ci};
3962306a36Sopenharmony_ci
4062306a36Sopenharmony_cistatic struct phy_device *phydev;
4162306a36Sopenharmony_cistatic const struct power_off_cfg *cfg;
4262306a36Sopenharmony_ci
4362306a36Sopenharmony_cistatic void linkstation_mvphy_reg_intn(bool restart)
4462306a36Sopenharmony_ci{
4562306a36Sopenharmony_ci	int rc = 0, saved_page;
4662306a36Sopenharmony_ci	u16 data = 0;
4762306a36Sopenharmony_ci
4862306a36Sopenharmony_ci	if (restart)
4962306a36Sopenharmony_ci		data = MII_88E1318S_PHY_LED_TCR_FORCE_INT;
5062306a36Sopenharmony_ci
5162306a36Sopenharmony_ci	saved_page = phy_select_page(phydev, MII_MARVELL_LED_PAGE);
5262306a36Sopenharmony_ci	if (saved_page < 0)
5362306a36Sopenharmony_ci		goto err;
5462306a36Sopenharmony_ci
5562306a36Sopenharmony_ci	/* Force manual LED2 control to let INTn work */
5662306a36Sopenharmony_ci	__phy_modify(phydev, MII_PHY_LED_CTRL, LEDMASK, LED2_FORCE_ON);
5762306a36Sopenharmony_ci
5862306a36Sopenharmony_ci	/* Set the LED[2]/INTn pin to the required state */
5962306a36Sopenharmony_ci	__phy_modify(phydev, MII_88E1318S_PHY_LED_TCR,
6062306a36Sopenharmony_ci		     MII_88E1318S_PHY_LED_TCR_FORCE_INT,
6162306a36Sopenharmony_ci		     MII_88E1318S_PHY_LED_TCR_INTn_ENABLE | data);
6262306a36Sopenharmony_ci
6362306a36Sopenharmony_ci	if (!data) {
6462306a36Sopenharmony_ci		/* Clear interrupts to ensure INTn won't be holded in high state */
6562306a36Sopenharmony_ci		__phy_write(phydev, MII_MARVELL_PHY_PAGE, MII_MARVELL_COPPER_PAGE);
6662306a36Sopenharmony_ci		__phy_read(phydev, MII_M1011_IEVENT);
6762306a36Sopenharmony_ci
6862306a36Sopenharmony_ci		/* If WOL was enabled and a magic packet was received before powering
6962306a36Sopenharmony_ci		 * off, we won't be able to wake up by sending another magic packet.
7062306a36Sopenharmony_ci		 * Clear WOL status.
7162306a36Sopenharmony_ci		 */
7262306a36Sopenharmony_ci		__phy_write(phydev, MII_MARVELL_PHY_PAGE, MII_MARVELL_WOL_PAGE);
7362306a36Sopenharmony_ci		__phy_set_bits(phydev, MII_88E1318S_PHY_WOL_CTRL,
7462306a36Sopenharmony_ci			       MII_88E1318S_PHY_WOL_CTRL_CLEAR_WOL_STATUS);
7562306a36Sopenharmony_ci	}
7662306a36Sopenharmony_cierr:
7762306a36Sopenharmony_ci	rc = phy_restore_page(phydev, saved_page, rc);
7862306a36Sopenharmony_ci	if (rc < 0)
7962306a36Sopenharmony_ci		dev_err(&phydev->mdio.dev, "Write register failed, %d\n", rc);
8062306a36Sopenharmony_ci}
8162306a36Sopenharmony_ci
8262306a36Sopenharmony_cistatic void readynas_mvphy_set_reg(bool restart)
8362306a36Sopenharmony_ci{
8462306a36Sopenharmony_ci	int rc = 0, saved_page;
8562306a36Sopenharmony_ci	u16 data = 0;
8662306a36Sopenharmony_ci
8762306a36Sopenharmony_ci	if (restart)
8862306a36Sopenharmony_ci		data = MII_88E1318S_PHY_LED_POL_LED2;
8962306a36Sopenharmony_ci
9062306a36Sopenharmony_ci	saved_page = phy_select_page(phydev, MII_MARVELL_LED_PAGE);
9162306a36Sopenharmony_ci	if (saved_page < 0)
9262306a36Sopenharmony_ci		goto err;
9362306a36Sopenharmony_ci
9462306a36Sopenharmony_ci	/* Set the LED[2].0 Polarity bit to the required state */
9562306a36Sopenharmony_ci	__phy_modify(phydev, MII_PHY_LED_POL_CTRL,
9662306a36Sopenharmony_ci		     MII_88E1318S_PHY_LED_POL_LED2, data);
9762306a36Sopenharmony_ci
9862306a36Sopenharmony_ci	if (!data) {
9962306a36Sopenharmony_ci		/* If WOL was enabled and a magic packet was received before powering
10062306a36Sopenharmony_ci		 * off, we won't be able to wake up by sending another magic packet.
10162306a36Sopenharmony_ci		 * Clear WOL status.
10262306a36Sopenharmony_ci		 */
10362306a36Sopenharmony_ci		__phy_write(phydev, MII_MARVELL_PHY_PAGE, MII_MARVELL_WOL_PAGE);
10462306a36Sopenharmony_ci		__phy_set_bits(phydev, MII_88E1318S_PHY_WOL_CTRL,
10562306a36Sopenharmony_ci			       MII_88E1318S_PHY_WOL_CTRL_CLEAR_WOL_STATUS);
10662306a36Sopenharmony_ci	}
10762306a36Sopenharmony_cierr:
10862306a36Sopenharmony_ci	rc = phy_restore_page(phydev, saved_page, rc);
10962306a36Sopenharmony_ci	if (rc < 0)
11062306a36Sopenharmony_ci		dev_err(&phydev->mdio.dev, "Write register failed, %d\n", rc);
11162306a36Sopenharmony_ci}
11262306a36Sopenharmony_ci
11362306a36Sopenharmony_cistatic const struct power_off_cfg linkstation_power_off_cfg = {
11462306a36Sopenharmony_ci	.mdio_node_name = "mdio",
11562306a36Sopenharmony_ci	.phy_set_reg = linkstation_mvphy_reg_intn,
11662306a36Sopenharmony_ci};
11762306a36Sopenharmony_ci
11862306a36Sopenharmony_cistatic const struct power_off_cfg readynas_power_off_cfg = {
11962306a36Sopenharmony_ci	.mdio_node_name = "mdio-bus",
12062306a36Sopenharmony_ci	.phy_set_reg = readynas_mvphy_set_reg,
12162306a36Sopenharmony_ci};
12262306a36Sopenharmony_ci
12362306a36Sopenharmony_cistatic int linkstation_reboot_notifier(struct notifier_block *nb,
12462306a36Sopenharmony_ci				       unsigned long action, void *unused)
12562306a36Sopenharmony_ci{
12662306a36Sopenharmony_ci	if (action == SYS_RESTART)
12762306a36Sopenharmony_ci		cfg->phy_set_reg(true);
12862306a36Sopenharmony_ci
12962306a36Sopenharmony_ci	return NOTIFY_DONE;
13062306a36Sopenharmony_ci}
13162306a36Sopenharmony_ci
13262306a36Sopenharmony_cistatic struct notifier_block linkstation_reboot_nb = {
13362306a36Sopenharmony_ci	.notifier_call = linkstation_reboot_notifier,
13462306a36Sopenharmony_ci};
13562306a36Sopenharmony_ci
13662306a36Sopenharmony_cistatic void linkstation_poweroff(void)
13762306a36Sopenharmony_ci{
13862306a36Sopenharmony_ci	unregister_reboot_notifier(&linkstation_reboot_nb);
13962306a36Sopenharmony_ci	cfg->phy_set_reg(false);
14062306a36Sopenharmony_ci
14162306a36Sopenharmony_ci	kernel_restart("Power off");
14262306a36Sopenharmony_ci}
14362306a36Sopenharmony_ci
14462306a36Sopenharmony_cistatic const struct of_device_id ls_poweroff_of_match[] = {
14562306a36Sopenharmony_ci	{ .compatible = "buffalo,ls421d",
14662306a36Sopenharmony_ci	  .data = &linkstation_power_off_cfg,
14762306a36Sopenharmony_ci	},
14862306a36Sopenharmony_ci	{ .compatible = "buffalo,ls421de",
14962306a36Sopenharmony_ci	  .data = &linkstation_power_off_cfg,
15062306a36Sopenharmony_ci	},
15162306a36Sopenharmony_ci	{ .compatible = "netgear,readynas-duo-v2",
15262306a36Sopenharmony_ci	  .data = &readynas_power_off_cfg,
15362306a36Sopenharmony_ci	},
15462306a36Sopenharmony_ci	{ },
15562306a36Sopenharmony_ci};
15662306a36Sopenharmony_ci
15762306a36Sopenharmony_cistatic int __init linkstation_poweroff_init(void)
15862306a36Sopenharmony_ci{
15962306a36Sopenharmony_ci	struct mii_bus *bus;
16062306a36Sopenharmony_ci	struct device_node *dn;
16162306a36Sopenharmony_ci	const struct of_device_id *match;
16262306a36Sopenharmony_ci
16362306a36Sopenharmony_ci	dn = of_find_matching_node(NULL, ls_poweroff_of_match);
16462306a36Sopenharmony_ci	if (!dn)
16562306a36Sopenharmony_ci		return -ENODEV;
16662306a36Sopenharmony_ci	of_node_put(dn);
16762306a36Sopenharmony_ci
16862306a36Sopenharmony_ci	match = of_match_node(ls_poweroff_of_match, dn);
16962306a36Sopenharmony_ci	cfg = match->data;
17062306a36Sopenharmony_ci
17162306a36Sopenharmony_ci	dn = of_find_node_by_name(NULL, cfg->mdio_node_name);
17262306a36Sopenharmony_ci	if (!dn)
17362306a36Sopenharmony_ci		return -ENODEV;
17462306a36Sopenharmony_ci
17562306a36Sopenharmony_ci	bus = of_mdio_find_bus(dn);
17662306a36Sopenharmony_ci	of_node_put(dn);
17762306a36Sopenharmony_ci	if (!bus)
17862306a36Sopenharmony_ci		return -EPROBE_DEFER;
17962306a36Sopenharmony_ci
18062306a36Sopenharmony_ci	phydev = phy_find_first(bus);
18162306a36Sopenharmony_ci	put_device(&bus->dev);
18262306a36Sopenharmony_ci	if (!phydev)
18362306a36Sopenharmony_ci		return -EPROBE_DEFER;
18462306a36Sopenharmony_ci
18562306a36Sopenharmony_ci	register_reboot_notifier(&linkstation_reboot_nb);
18662306a36Sopenharmony_ci	pm_power_off = linkstation_poweroff;
18762306a36Sopenharmony_ci
18862306a36Sopenharmony_ci	return 0;
18962306a36Sopenharmony_ci}
19062306a36Sopenharmony_ci
19162306a36Sopenharmony_cistatic void __exit linkstation_poweroff_exit(void)
19262306a36Sopenharmony_ci{
19362306a36Sopenharmony_ci	pm_power_off = NULL;
19462306a36Sopenharmony_ci	unregister_reboot_notifier(&linkstation_reboot_nb);
19562306a36Sopenharmony_ci}
19662306a36Sopenharmony_ci
19762306a36Sopenharmony_cimodule_init(linkstation_poweroff_init);
19862306a36Sopenharmony_cimodule_exit(linkstation_poweroff_exit);
19962306a36Sopenharmony_ci
20062306a36Sopenharmony_ciMODULE_AUTHOR("Daniel González Cabanelas <dgcbueu@gmail.com>");
20162306a36Sopenharmony_ciMODULE_DESCRIPTION("LinkStation power off driver");
20262306a36Sopenharmony_ciMODULE_LICENSE("GPL v2");
203