//
 // Based on code originally by Andrey Volkov <avolkov@varma-el.com>
 
+#include <dt-bindings/firmware/imx/rsrc.h>
 #include <linux/bitfield.h>
 #include <linux/can.h>
 #include <linux/can/dev.h>
 #include <linux/can/rx-offload.h>
 #include <linux/clk.h>
 #include <linux/delay.h>
+#include <linux/firmware/imx/sci.h>
 #include <linux/interrupt.h>
 #include <linux/io.h>
 #include <linux/mfd/syscon.h>
 #define FLEXCAN_QUIRK_SUPPORT_FD BIT(9)
 /* support memory detection and correction */
 #define FLEXCAN_QUIRK_SUPPORT_ECC BIT(10)
+/* Setup stop mode with SCU firmware to support wakeup */
+#define FLEXCAN_QUIRK_SETUP_STOP_MODE_SCFW BIT(11)
 
 /* Structure of the message buffer */
 struct flexcan_mb {
        u8 mb_count;
        u8 mb_size;
        u8 clk_src;     /* clock source of CAN Protocol Engine */
+       u8 scu_idx;
 
        u64 rx_mask;
        u64 tx_mask;
        struct regulator *reg_xceiver;
        struct flexcan_stop_mode stm;
 
+       /* IPC handle when setup stop mode by System Controller firmware(scfw) */
+       struct imx_sc_ipc *sc_ipc_handle;
+
        /* Read and Write APIs */
        u32 (*read)(void __iomem *addr);
        void (*write)(u32 val, void __iomem *addr);
 static const struct flexcan_devtype_data fsl_imx8qm_devtype_data = {
        .quirks = FLEXCAN_QUIRK_DISABLE_RXFG | FLEXCAN_QUIRK_ENABLE_EACEN_RRS |
                FLEXCAN_QUIRK_USE_OFF_TIMESTAMP | FLEXCAN_QUIRK_BROKEN_PERR_STATE |
-               FLEXCAN_QUIRK_SUPPORT_FD,
+               FLEXCAN_QUIRK_SUPPORT_FD | FLEXCAN_QUIRK_SETUP_STOP_MODE_SCFW,
 };
 
 static struct flexcan_devtype_data fsl_imx8mp_devtype_data = {
        priv->write(reg_mcr, ®s->mcr);
 }
 
+static int flexcan_stop_mode_enable_scfw(struct flexcan_priv *priv, bool enabled)
+{
+       u8 idx = priv->scu_idx;
+       u32 rsrc_id, val;
+
+       rsrc_id = IMX_SC_R_CAN(idx);
+
+       if (enabled)
+               val = 1;
+       else
+               val = 0;
+
+       /* stop mode request via scu firmware */
+       return imx_sc_misc_set_control(priv->sc_ipc_handle, rsrc_id,
+                                      IMX_SC_C_IPG_STOP, val);
+}
+
 static inline int flexcan_enter_stop_mode(struct flexcan_priv *priv)
 {
        struct flexcan_regs __iomem *regs = priv->regs;
        u32 reg_mcr;
+       int ret;
 
        reg_mcr = priv->read(®s->mcr);
        reg_mcr |= FLEXCAN_MCR_SLF_WAK;
        priv->write(reg_mcr, ®s->mcr);
 
        /* enable stop request */
-       regmap_update_bits(priv->stm.gpr, priv->stm.req_gpr,
-                          1 << priv->stm.req_bit, 1 << priv->stm.req_bit);
+       if (priv->devtype_data->quirks & FLEXCAN_QUIRK_SETUP_STOP_MODE_SCFW) {
+               ret = flexcan_stop_mode_enable_scfw(priv, true);
+               if (ret < 0)
+                       return ret;
+       } else {
+               regmap_update_bits(priv->stm.gpr, priv->stm.req_gpr,
+                                  1 << priv->stm.req_bit, 1 << priv->stm.req_bit);
+       }
 
        return flexcan_low_power_enter_ack(priv);
 }
 {
        struct flexcan_regs __iomem *regs = priv->regs;
        u32 reg_mcr;
+       int ret;
 
        /* remove stop request */
-       regmap_update_bits(priv->stm.gpr, priv->stm.req_gpr,
-                          1 << priv->stm.req_bit, 0);
+       if (priv->devtype_data->quirks & FLEXCAN_QUIRK_SETUP_STOP_MODE_SCFW) {
+               ret = flexcan_stop_mode_enable_scfw(priv, false);
+               if (ret < 0)
+                       return ret;
+       } else {
+               regmap_update_bits(priv->stm.gpr, priv->stm.req_gpr,
+                                  1 << priv->stm.req_bit, 0);
+       }
 
        reg_mcr = priv->read(®s->mcr);
        reg_mcr &= ~FLEXCAN_MCR_SLF_WAK;
        unregister_candev(dev);
 }
 
-static int flexcan_setup_stop_mode(struct platform_device *pdev)
+static int flexcan_setup_stop_mode_gpr(struct platform_device *pdev)
 {
        struct net_device *dev = platform_get_drvdata(pdev);
        struct device_node *np = pdev->dev.of_node;
                "gpr %s req_gpr=0x02%x req_bit=%u\n",
                gpr_np->full_name, priv->stm.req_gpr, priv->stm.req_bit);
 
-       device_set_wakeup_capable(&pdev->dev, true);
-
-       if (of_property_read_bool(np, "wakeup-source"))
-               device_set_wakeup_enable(&pdev->dev, true);
-
        return 0;
 
 out_put_node:
        return ret;
 }
 
+static int flexcan_setup_stop_mode_scfw(struct platform_device *pdev)
+{
+       struct net_device *dev = platform_get_drvdata(pdev);
+       struct flexcan_priv *priv;
+       u8 scu_idx;
+       int ret;
+
+       ret = of_property_read_u8(pdev->dev.of_node, "fsl,scu-index", &scu_idx);
+       if (ret < 0) {
+               dev_dbg(&pdev->dev, "failed to get scu index\n");
+               return ret;
+       }
+
+       priv = netdev_priv(dev);
+       priv->scu_idx = scu_idx;
+
+       /* this function could be defered probe, return -EPROBE_DEFER */
+       return imx_scu_get_handle(&priv->sc_ipc_handle);
+}
+
+/* flexcan_setup_stop_mode - Setup stop mode for wakeup
+ *
+ * Return: = 0 setup stop mode successfully or doesn't support this feature
+ *         < 0 fail to setup stop mode (could be defered probe)
+ */
+static int flexcan_setup_stop_mode(struct platform_device *pdev)
+{
+       struct net_device *dev = platform_get_drvdata(pdev);
+       struct flexcan_priv *priv;
+       int ret;
+
+       priv = netdev_priv(dev);
+
+       if (priv->devtype_data->quirks & FLEXCAN_QUIRK_SETUP_STOP_MODE_SCFW)
+               ret = flexcan_setup_stop_mode_scfw(pdev);
+       else if (priv->devtype_data->quirks & FLEXCAN_QUIRK_SETUP_STOP_MODE_GPR)
+               ret = flexcan_setup_stop_mode_gpr(pdev);
+       else
+               /* return 0 directly if doesn't support stop mode feature */
+               return 0;
+
+       if (ret)
+               return ret;
+
+       device_set_wakeup_capable(&pdev->dev, true);
+
+       if (of_property_read_bool(pdev->dev.of_node, "wakeup-source"))
+               device_set_wakeup_enable(&pdev->dev, true);
+
+       return 0;
+}
+
 static const struct of_device_id flexcan_of_match[] = {
        { .compatible = "fsl,imx8qm-flexcan", .data = &fsl_imx8qm_devtype_data, },
        { .compatible = "fsl,imx8mp-flexcan", .data = &fsl_imx8mp_devtype_data, },
                goto failed_register;
        }
 
+       err = flexcan_setup_stop_mode(pdev);
+       if (err < 0) {
+               if (err != -EPROBE_DEFER)
+                       dev_err(&pdev->dev, "setup stop mode failed\n");
+               goto failed_setup_stop_mode;
+       }
+
        of_can_transceiver(dev);
        devm_can_led_init(dev);
 
-       if (priv->devtype_data->quirks & FLEXCAN_QUIRK_SETUP_STOP_MODE_GPR) {
-               err = flexcan_setup_stop_mode(pdev);
-               if (err)
-                       dev_dbg(&pdev->dev, "failed to setup stop-mode\n");
-       }
-
        return 0;
 
+ failed_setup_stop_mode:
+       unregister_flexcandev(dev);
  failed_register:
        pm_runtime_put_noidle(&pdev->dev);
        pm_runtime_disable(&pdev->dev);