#include <linux/slab.h>
 #include <linux/io.h>
 #include <linux/dmaengine.h>
+#include <linux/pm_runtime.h>
 
 #include <sound/core.h>
 #include <sound/pcm.h>
        struct clk *clk;
        struct clk *audio_clk;
        struct clk *sysclk;
-       int dai_fmt;
        int running_cnt;
+       u32 sp;
+       u32 ctrl;
 };
 
 static void mmp_sspa_write_reg(struct sspa_priv *sspa, u32 reg, u32 val)
 
 static void mmp_sspa_tx_enable(struct sspa_priv *sspa)
 {
-       unsigned int sspa_sp;
+       unsigned int sspa_sp = sspa->sp;
 
-       sspa_sp = mmp_sspa_read_reg(sspa, SSPA_TXSP);
+       sspa_sp &= ~SSPA_SP_MSL;
        sspa_sp |= SSPA_SP_S_EN;
        sspa_sp |= SSPA_SP_WEN;
        mmp_sspa_write_reg(sspa, SSPA_TXSP, sspa_sp);
 
 static void mmp_sspa_tx_disable(struct sspa_priv *sspa)
 {
-       unsigned int sspa_sp;
+       unsigned int sspa_sp = sspa->sp;
 
-       sspa_sp = mmp_sspa_read_reg(sspa, SSPA_TXSP);
+       sspa_sp &= ~SSPA_SP_MSL;
        sspa_sp &= ~SSPA_SP_S_EN;
        sspa_sp |= SSPA_SP_WEN;
        mmp_sspa_write_reg(sspa, SSPA_TXSP, sspa_sp);
 
 static void mmp_sspa_rx_enable(struct sspa_priv *sspa)
 {
-       unsigned int sspa_sp;
+       unsigned int sspa_sp = sspa->sp;
 
-       sspa_sp = mmp_sspa_read_reg(sspa, SSPA_RXSP);
        sspa_sp |= SSPA_SP_S_EN;
        sspa_sp |= SSPA_SP_WEN;
        mmp_sspa_write_reg(sspa, SSPA_RXSP, sspa_sp);
 
 static void mmp_sspa_rx_disable(struct sspa_priv *sspa)
 {
-       unsigned int sspa_sp;
+       unsigned int sspa_sp = sspa->sp;
 
-       sspa_sp = mmp_sspa_read_reg(sspa, SSPA_RXSP);
        sspa_sp &= ~SSPA_SP_S_EN;
        sspa_sp |= SSPA_SP_WEN;
        mmp_sspa_write_reg(sspa, SSPA_RXSP, sspa_sp);
 }
 
 /*
- * Set up the sspa dai format. The sspa port must be inactive
- * before calling this function as the physical
- * interface format is changed.
+ * Set up the sspa dai format.
  */
 static int mmp_sspa_set_dai_fmt(struct snd_soc_dai *cpu_dai,
                                 unsigned int fmt)
 {
        struct sspa_priv *sspa = snd_soc_dai_get_drvdata(cpu_dai);
-       u32 sspa_sp, sspa_ctrl;
-
-       /* check if we need to change anything at all */
-       if (sspa->dai_fmt == fmt)
-               return 0;
-
-       /* we can only change the settings if the port is not in use */
-       if ((mmp_sspa_read_reg(sspa, SSPA_TXSP) & SSPA_SP_S_EN) ||
-           (mmp_sspa_read_reg(sspa, SSPA_RXSP) & SSPA_SP_S_EN)) {
-               dev_err(cpu_dai->dev,
-                       "can't change hardware dai format: stream is in use\n");
-               return -EINVAL;
-       }
 
        /* reset port settings */
-       sspa_sp   = SSPA_SP_WEN | SSPA_SP_S_RST | SSPA_SP_FFLUSH;
-       sspa_ctrl = 0;
+       sspa->sp   = SSPA_SP_WEN | SSPA_SP_S_RST | SSPA_SP_FFLUSH;
+       sspa->ctrl = 0;
 
        switch (fmt & SND_SOC_DAIFMT_MASTER_MASK) {
        case SND_SOC_DAIFMT_CBS_CFS:
-               sspa_sp |= SSPA_SP_MSL;
+               sspa->sp |= SSPA_SP_MSL;
                break;
        case SND_SOC_DAIFMT_CBM_CFM:
                break;
 
        switch (fmt & SND_SOC_DAIFMT_INV_MASK) {
        case SND_SOC_DAIFMT_NB_NF:
-               sspa_sp |= SSPA_SP_FSP;
+               sspa->sp |= SSPA_SP_FSP;
                break;
        default:
                return -EINVAL;
 
        switch (fmt & SND_SOC_DAIFMT_FORMAT_MASK) {
        case SND_SOC_DAIFMT_I2S:
-               sspa_sp |= SSPA_TXSP_FPER(63);
-               sspa_sp |= SSPA_SP_FWID(31);
-               sspa_ctrl |= SSPA_CTL_XDATDLY(1);
+               sspa->sp |= SSPA_TXSP_FPER(63);
+               sspa->sp |= SSPA_SP_FWID(31);
+               sspa->ctrl |= SSPA_CTL_XDATDLY(1);
                break;
        default:
                return -EINVAL;
        }
 
-       mmp_sspa_write_reg(sspa, SSPA_TXSP, sspa_sp);
-       mmp_sspa_write_reg(sspa, SSPA_RXSP, sspa_sp);
-
-       sspa_sp &= ~(SSPA_SP_S_RST | SSPA_SP_FFLUSH);
-       mmp_sspa_write_reg(sspa, SSPA_TXSP, sspa_sp);
-       mmp_sspa_write_reg(sspa, SSPA_RXSP, sspa_sp);
-
-       /*
-        * FIXME: hw issue, for the tx serial port,
-        * can not config the master/slave mode;
-        * so must clean this bit.
-        * The master/slave mode has been set in the
-        * rx port.
-        */
-       sspa_sp &= ~SSPA_SP_MSL;
-       mmp_sspa_write_reg(sspa, SSPA_TXSP, sspa_sp);
-
-       mmp_sspa_write_reg(sspa, SSPA_TXCTL, sspa_ctrl);
-       mmp_sspa_write_reg(sspa, SSPA_RXCTL, sspa_ctrl);
-
        /* Since we are configuring the timings for the format by hand
         * we have to defer some things until hw_params() where we
         * know parameters like the sample size.
         */
-       sspa->dai_fmt = fmt;
        return 0;
 }
 
                               struct snd_soc_dai *dai)
 {
        struct sspa_priv *sspa = snd_soc_dai_get_drvdata(dai);
-       u32 sspa_ctrl;
-
-       if (substream->stream == SNDRV_PCM_STREAM_PLAYBACK)
-               sspa_ctrl = mmp_sspa_read_reg(sspa, SSPA_TXCTL);
-       else
-               sspa_ctrl = mmp_sspa_read_reg(sspa, SSPA_RXCTL);
+       u32 sspa_ctrl = sspa->ctrl;
 
        sspa_ctrl &= ~SSPA_CTL_XFRLEN1_MASK;
        sspa_ctrl |= SSPA_CTL_XFRLEN1(params_channels(params) - 1);
                vma->vm_end - vma->vm_start, vma->vm_page_prot);
 }
 
+static int mmp_sspa_open(struct snd_soc_component *component,
+                        struct snd_pcm_substream *substream)
+{
+       struct sspa_priv *sspa = snd_soc_component_get_drvdata(component);
+
+       pm_runtime_get_sync(component->dev);
+
+       /* we can only change the settings if the port is not in use */
+       if ((mmp_sspa_read_reg(sspa, SSPA_TXSP) & SSPA_SP_S_EN) ||
+           (mmp_sspa_read_reg(sspa, SSPA_RXSP) & SSPA_SP_S_EN)) {
+               dev_err(component->dev,
+                       "can't change hardware dai format: stream is in use\n");
+               return -EBUSY;
+       }
+
+       mmp_sspa_write_reg(sspa, SSPA_TXSP, sspa->sp);
+       mmp_sspa_write_reg(sspa, SSPA_RXSP, sspa->sp);
+
+       sspa->sp &= ~(SSPA_SP_S_RST | SSPA_SP_FFLUSH);
+       mmp_sspa_write_reg(sspa, SSPA_TXSP, sspa->sp);
+       mmp_sspa_write_reg(sspa, SSPA_RXSP, sspa->sp);
+
+       /*
+        * FIXME: hw issue, for the tx serial port,
+        * can not config the master/slave mode;
+        * so must clean this bit.
+        * The master/slave mode has been set in the
+        * rx port.
+        */
+       mmp_sspa_write_reg(sspa, SSPA_TXSP, sspa->sp & ~SSPA_SP_MSL);
+
+       mmp_sspa_write_reg(sspa, SSPA_TXCTL, sspa->ctrl);
+       mmp_sspa_write_reg(sspa, SSPA_RXCTL, sspa->ctrl);
+
+       return 0;
+}
+
+static int mmp_sspa_close(struct snd_soc_component *component,
+                         struct snd_pcm_substream *substream)
+{
+       pm_runtime_put_sync(component->dev);
+       return 0;
+}
+
 static const struct snd_soc_component_driver mmp_sspa_component = {
        .name           = "mmp-sspa",
        .mmap           = mmp_pcm_mmap,
+       .open           = mmp_sspa_open,
+       .close          = mmp_sspa_close,
 };
 
 static int asoc_mmp_sspa_probe(struct platform_device *pdev)
                clk_put(sspa->audio_clk);
                return PTR_ERR(sspa->sysclk);
        }
+       pm_runtime_enable(&pdev->dev);
        clk_prepare_enable(sspa->audio_clk);
-       sspa->dai_fmt = (unsigned int) -1;
        platform_set_drvdata(pdev, sspa);
 
        sspa->playback_dma_data.maxburst = 4;
        struct sspa_priv *sspa = platform_get_drvdata(pdev);
 
        clk_disable_unprepare(sspa->audio_clk);
+       pm_runtime_disable(&pdev->dev);
        clk_put(sspa->audio_clk);
        clk_put(sspa->sysclk);
        return 0;