WIP: Add clocks and resets to the Rockchip UFS driver

Signed-off-by: Alexey Charkov <alchark@gmail.com>

Author: alchark

drivers/ufs/ufs-rockchip.c

@@ -5,6 +5,7 @@
  * Copyright (C) 2025 Rockchip Electronics Co.Ltd.
  */
 
+#include <asm/gpio.h>
 #include <asm/io.h>
 #include <clk.h>
 #include <dm.h>
@@ -19,6 +20,13 @@
 #include "unipro.h"
 #include "ufs-rockchip.h"
 
+static void ufs_rockchip_controller_reset(struct ufs_rockchip_host *host)
+{
+        reset_assert_bulk(&host->rsts);
+        udelay(1);
+        reset_deassert_bulk(&host->rsts);
+}
+
 static int ufs_rockchip_hce_enable_notify(struct ufs_hba *hba,
 					  enum ufs_notify_change_status status)
 {
@@ -153,11 +161,53 @@ static int ufs_rockchip_common_init(struct ufs_hba *hba)
 		return err;
 	}
 
+	ufs_rockchip_controller_reset(host);
+
+	err = clk_get_by_name(dev, "ref_out", &host->ref_out_clk);
+	if (err) {
+		dev_err(dev, "cannot get ref_out_clk: %d\n", err);
+		return err;
+	}
+
+	err = gpio_request_by_name(dev, "reset-gpios", 0, &host->device_reset,
+				   GPIOD_IS_OUT | GPIOD_ACTIVE_LOW);
+	if (err) {
+		dev_err(dev, "Warning: cannot get reset GPIO\n");
+	}
+
+	err = clk_get_bulk(dev, &host->clks);
+	if (err) {
+		dev_err(dev, "cannot get clocks: %d\n", err);
+		return err;
+	}
+
+	err = clk_enable_bulk(&host->clks);
+	if (err) {
+		dev_err(dev, "cannot enable clocks: %d\n", err);
+		return err;
+	}
+
 	host->hba = hba;
 
 	return 0;
 }
 
+static int ufs_rockchip_device_reset(struct ufs_hba *hba)
+{
+	struct ufs_rockchip_host *host = dev_get_priv(hba->dev);
+
+        if (!dm_gpio_is_valid(&host->device_reset))
+                return 0;
+
+        dm_gpio_set_value(&host->device_reset, true);
+        udelay(20);
+
+        dm_gpio_set_value(&host->device_reset, false);
+        udelay(20);
+
+        return 0;
+}
+
 static int ufs_rockchip_rk3576_init(struct ufs_hba *hba)
 {
 	int ret = 0;
@@ -175,6 +225,7 @@ static struct ufs_hba_ops ufs_hba_rk3576_vops = {
 	.init = ufs_rockchip_rk3576_init,
 	.phy_initialization = ufs_rockchip_rk3576_phy_init,
 	.hce_enable_notify = ufs_rockchip_hce_enable_notify,
+	.device_reset = ufs_rockchip_device_reset,
 };
 
 static const struct udevice_id ufs_rockchip_of_match[] = {

drivers/ufs/ufs-rockchip.h

@@ -72,7 +72,9 @@ struct ufs_rockchip_host {
 	void __iomem *ufs_sys_ctrl;
 	void __iomem *mphy_base;
 	struct reset_ctl_bulk rsts;
+	struct gpio_desc device_reset;
 	struct clk ref_out_clk;
+	struct clk_bulk clks;
 	uint64_t caps;
 	uint32_t phy_config_mode;
 };