ASoC: fsl_rpmsg: Fix error handler with pm_runtime_enable

There is error message when defer probe happens:

fsl_rpmsg rpmsg_audio: Unbalanced pm_runtime_enable!

Fix the error handler with pm_runtime_enable.

Fixes: b73d9e6225e8 ("ASoC: fsl_rpmsg: Add CPU DAI driver for audio base on rpmsg")
Signed-off-by: Chancel Liu <chancel.liu@nxp.com>
Acked-by: Shengjiu Wang <shengjiu.wang@gmail.com>
Link: https://lore.kernel.org/r/20231225080608.967953-1-chancel.liu@nxp.com
Signed-off-by: Mark Brown <broonie@kernel.org>

diff --git a/sound/soc/fsl/fsl_rpmsg.c b/sound/soc/fsl/fsl_rpmsg.c
index 5c5c04ce9db7d2a91546fe36b7241ef683ef8e54..00852f174a69c59381bcec7e9ce09338dcdd6eb1 100644 (file)
--- a/sound/soc/fsl/fsl_rpmsg.c
+++ b/sound/soc/fsl/fsl_rpmsg.c
@@ -238,7 +238,7 @@ static int fsl_rpmsg_probe(struct platform_device *pdev)
        ret = devm_snd_soc_register_component(&pdev->dev, &fsl_component,
                                              &fsl_rpmsg_dai, 1);
        if (ret)
-               return ret;
+               goto err_pm_disable;
 
        rpmsg->card_pdev = platform_device_register_data(&pdev->dev,
                                                         "imx-audio-rpmsg",
@@ -248,16 +248,22 @@ static int fsl_rpmsg_probe(struct platform_device *pdev)
        if (IS_ERR(rpmsg->card_pdev)) {
                dev_err(&pdev->dev, "failed to register rpmsg card\n");
                ret = PTR_ERR(rpmsg->card_pdev);
-               return ret;
+               goto err_pm_disable;
        }
 
        return 0;
+
+err_pm_disable:
+       pm_runtime_disable(&pdev->dev);
+       return ret;
 }
 
 static void fsl_rpmsg_remove(struct platform_device *pdev)
 {
        struct fsl_rpmsg *rpmsg = platform_get_drvdata(pdev);
 
+       pm_runtime_disable(&pdev->dev);
+
        if (rpmsg->card_pdev)
                platform_device_unregister(rpmsg->card_pdev);
 }