return (ssi->playback.dma_ch && (ssi->dma_rt || ssi->capture.dma_ch));
 }
 
+static void rz_ssi_set_substream(struct rz_ssi_stream *strm,
+                                struct snd_pcm_substream *substream)
+{
+       struct rz_ssi_priv *ssi = strm->priv;
+       unsigned long flags;
+
+       spin_lock_irqsave(&ssi->lock, flags);
+       strm->substream = substream;
+       spin_unlock_irqrestore(&ssi->lock, flags);
+}
+
 static bool rz_ssi_stream_is_valid(struct rz_ssi_priv *ssi,
                                   struct rz_ssi_stream *strm)
 {
 {
        struct snd_pcm_runtime *runtime = substream->runtime;
 
-       strm->substream = substream;
+       rz_ssi_set_substream(strm, substream);
        strm->sample_width = samples_to_bytes(runtime, 1);
        strm->dma_buffer_pos = 0;
        strm->period_counter = 0;
                               struct rz_ssi_stream *strm)
 {
        struct snd_soc_dai *dai = rz_ssi_get_dai(strm->substream);
-       unsigned long flags;
 
-       spin_lock_irqsave(&ssi->lock, flags);
-       strm->substream = NULL;
-       spin_unlock_irqrestore(&ssi->lock, flags);
+       rz_ssi_set_substream(strm, NULL);
 
        if (strm->oerr_num > 0)
                dev_info(dai->dev, "overrun = %d\n", strm->oerr_num);