veu->aborting = true;
 }
 
-static void sh_veu_lock(void *priv)
-{
-       struct sh_veu_dev *veu = priv;
-
-       mutex_lock(&veu->fop_lock);
-}
-
-static void sh_veu_unlock(void *priv)
-{
-       struct sh_veu_dev *veu = priv;
-
-       mutex_unlock(&veu->fop_lock);
-}
-
 static void sh_veu_process(struct sh_veu_dev *veu,
                           struct vb2_buffer *src_buf,
                           struct vb2_buffer *dst_buf)
        v4l2_m2m_buf_queue(veu->m2m_ctx, vb);
 }
 
-static void sh_veu_wait_prepare(struct vb2_queue *q)
-{
-       sh_veu_unlock(vb2_get_drv_priv(q));
-}
-
-static void sh_veu_wait_finish(struct vb2_queue *q)
-{
-       sh_veu_lock(vb2_get_drv_priv(q));
-}
-
 static const struct vb2_ops sh_veu_qops = {
        .queue_setup     = sh_veu_queue_setup,
        .buf_prepare     = sh_veu_buf_prepare,
        .buf_queue       = sh_veu_buf_queue,
-       .wait_prepare    = sh_veu_wait_prepare,
-       .wait_finish     = sh_veu_wait_finish,
+       .wait_prepare    = vb2_ops_wait_prepare,
+       .wait_finish     = vb2_ops_wait_finish,
 };
 
 static int sh_veu_queue_init(void *priv, struct vb2_queue *src_vq,
                             struct vb2_queue *dst_vq)
 {
+       struct sh_veu_dev *veu = priv;
        int ret;
 
        memset(src_vq, 0, sizeof(*src_vq));
        src_vq->type = V4L2_BUF_TYPE_VIDEO_OUTPUT;
        src_vq->io_modes = VB2_MMAP | VB2_USERPTR;
-       src_vq->drv_priv = priv;
+       src_vq->drv_priv = veu;
        src_vq->buf_struct_size = sizeof(struct v4l2_m2m_buffer);
        src_vq->ops = &sh_veu_qops;
        src_vq->mem_ops = &vb2_dma_contig_memops;
+       src_vq->lock = &veu->fop_lock;
 
        ret = vb2_queue_init(src_vq);
        if (ret < 0)
        memset(dst_vq, 0, sizeof(*dst_vq));
        dst_vq->type = V4L2_BUF_TYPE_VIDEO_CAPTURE;
        dst_vq->io_modes = VB2_MMAP | VB2_USERPTR;
-       dst_vq->drv_priv = priv;
+       dst_vq->drv_priv = veu;
        dst_vq->buf_struct_size = sizeof(struct v4l2_m2m_buffer);
        dst_vq->ops = &sh_veu_qops;
        dst_vq->mem_ops = &vb2_dma_contig_memops;
+       dst_vq->lock = &veu->fop_lock;
 
        return vb2_queue_init(dst_vq);
 }