v4l2_m2m_buf_queue(ctx->fh.m2m_ctx, vbuf);
 }
 
+static void rga_buf_return_buffers(struct vb2_queue *q,
+                                  enum vb2_buffer_state state)
+{
+       struct rga_ctx *ctx = vb2_get_drv_priv(q);
+       struct vb2_v4l2_buffer *vbuf;
+
+       for (;;) {
+               if (V4L2_TYPE_IS_OUTPUT(q->type))
+                       vbuf = v4l2_m2m_src_buf_remove(ctx->fh.m2m_ctx);
+               else
+                       vbuf = v4l2_m2m_dst_buf_remove(ctx->fh.m2m_ctx);
+               if (!vbuf)
+                       break;
+               v4l2_m2m_buf_done(vbuf, state);
+       }
+}
+
 static int rga_buf_start_streaming(struct vb2_queue *q, unsigned int count)
 {
        struct rga_ctx *ctx = vb2_get_drv_priv(q);
        struct rockchip_rga *rga = ctx->rga;
-       int ret, i;
+       int ret;
 
        ret = pm_runtime_get_sync(rga->dev);
-
-       if (!ret)
-               return 0;
-
-       for (i = 0; i < q->num_buffers; ++i) {
-               if (q->bufs[i]->state == VB2_BUF_STATE_ACTIVE) {
-                       v4l2_m2m_buf_done(to_vb2_v4l2_buffer(q->bufs[i]),
-                                         VB2_BUF_STATE_QUEUED);
-               }
+       if (ret < 0) {
+               rga_buf_return_buffers(q, VB2_BUF_STATE_QUEUED);
+               return ret;
        }
 
-       return ret;
+       return 0;
 }
 
 static void rga_buf_stop_streaming(struct vb2_queue *q)
 {
        struct rga_ctx *ctx = vb2_get_drv_priv(q);
        struct rockchip_rga *rga = ctx->rga;
-       struct vb2_v4l2_buffer *vbuf;
-
-       for (;;) {
-               if (V4L2_TYPE_IS_OUTPUT(q->type))
-                       vbuf = v4l2_m2m_src_buf_remove(ctx->fh.m2m_ctx);
-               else
-                       vbuf = v4l2_m2m_dst_buf_remove(ctx->fh.m2m_ctx);
-               if (!vbuf)
-                       break;
-               v4l2_m2m_buf_done(vbuf, VB2_BUF_STATE_ERROR);
-       }
 
+       rga_buf_return_buffers(q, VB2_BUF_STATE_ERROR);
        pm_runtime_put(rga->dev);
 }