v4l2_m2m_buf_queue(ctx->fh.m2m_ctx, to_vb2_v4l2_buffer(vb));
 }
 
-static void *mtk_jpeg_buf_remove(struct mtk_jpeg_ctx *ctx,
+static struct vb2_v4l2_buffer *mtk_jpeg_buf_remove(struct mtk_jpeg_ctx *ctx,
                                 enum v4l2_buf_type type)
 {
        if (V4L2_TYPE_IS_OUTPUT(type))
 static int mtk_jpeg_start_streaming(struct vb2_queue *q, unsigned int count)
 {
        struct mtk_jpeg_ctx *ctx = vb2_get_drv_priv(q);
-       struct vb2_buffer *vb;
+       struct vb2_v4l2_buffer *vb;
        int ret = 0;
 
        ret = pm_runtime_get_sync(ctx->jpeg->dev);
        return 0;
 err:
        while ((vb = mtk_jpeg_buf_remove(ctx, q->type)))
-               v4l2_m2m_buf_done(to_vb2_v4l2_buffer(vb), VB2_BUF_STATE_QUEUED);
+               v4l2_m2m_buf_done(vb, VB2_BUF_STATE_QUEUED);
        return ret;
 }
 
 static void mtk_jpeg_stop_streaming(struct vb2_queue *q)
 {
        struct mtk_jpeg_ctx *ctx = vb2_get_drv_priv(q);
-       struct vb2_buffer *vb;
+       struct vb2_v4l2_buffer *vb;
 
        /*
         * STREAMOFF is an acknowledgment for source change event.
                struct mtk_jpeg_src_buf *src_buf;
 
                vb = v4l2_m2m_next_src_buf(ctx->fh.m2m_ctx);
-               src_buf = mtk_jpeg_vb2_to_srcbuf(vb);
+               src_buf = mtk_jpeg_vb2_to_srcbuf(&vb->vb2_buf);
                mtk_jpeg_set_queue_data(ctx, &src_buf->dec_param);
                ctx->state = MTK_JPEG_RUNNING;
        } else if (V4L2_TYPE_IS_OUTPUT(q->type)) {
        }
 
        while ((vb = mtk_jpeg_buf_remove(ctx, q->type)))
-               v4l2_m2m_buf_done(to_vb2_v4l2_buffer(vb), VB2_BUF_STATE_ERROR);
+               v4l2_m2m_buf_done(vb, VB2_BUF_STATE_ERROR);
 
        pm_runtime_put_sync(ctx->jpeg->dev);
 }
 {
        struct mtk_jpeg_ctx *ctx = priv;
        struct mtk_jpeg_dev *jpeg = ctx->jpeg;
-       struct vb2_buffer *src_buf, *dst_buf;
+       struct vb2_v4l2_buffer *src_buf, *dst_buf;
        enum vb2_buffer_state buf_state = VB2_BUF_STATE_ERROR;
        unsigned long flags;
        struct mtk_jpeg_src_buf *jpeg_src_buf;
 
        src_buf = v4l2_m2m_next_src_buf(ctx->fh.m2m_ctx);
        dst_buf = v4l2_m2m_next_dst_buf(ctx->fh.m2m_ctx);
-       jpeg_src_buf = mtk_jpeg_vb2_to_srcbuf(src_buf);
+       jpeg_src_buf = mtk_jpeg_vb2_to_srcbuf(&src_buf->vb2_buf);
 
        if (jpeg_src_buf->flags & MTK_JPEG_BUF_FLAGS_LAST_FRAME) {
-               for (i = 0; i < dst_buf->num_planes; i++)
-                       vb2_set_plane_payload(dst_buf, i, 0);
+               for (i = 0; i < dst_buf->vb2_buf.num_planes; i++)
+                       vb2_set_plane_payload(&dst_buf->vb2_buf, i, 0);
                buf_state = VB2_BUF_STATE_DONE;
                goto dec_end;
        }
                return;
        }
 
-       mtk_jpeg_set_dec_src(ctx, src_buf, &bs);
-       if (mtk_jpeg_set_dec_dst(ctx, &jpeg_src_buf->dec_param, dst_buf, &fb))
+       mtk_jpeg_set_dec_src(ctx, &src_buf->vb2_buf, &bs);
+       if (mtk_jpeg_set_dec_dst(ctx, &jpeg_src_buf->dec_param, &dst_buf->vb2_buf, &fb))
                goto dec_end;
 
        spin_lock_irqsave(&jpeg->hw_lock, flags);
 dec_end:
        v4l2_m2m_src_buf_remove(ctx->fh.m2m_ctx);
        v4l2_m2m_dst_buf_remove(ctx->fh.m2m_ctx);
-       v4l2_m2m_buf_done(to_vb2_v4l2_buffer(src_buf), buf_state);
-       v4l2_m2m_buf_done(to_vb2_v4l2_buffer(dst_buf), buf_state);
+       v4l2_m2m_buf_done(src_buf, buf_state);
+       v4l2_m2m_buf_done(dst_buf, buf_state);
        v4l2_m2m_job_finish(jpeg->m2m_dev, ctx->fh.m2m_ctx);
 }
 
 {
        struct mtk_jpeg_dev *jpeg = priv;
        struct mtk_jpeg_ctx *ctx;
-       struct vb2_buffer *src_buf, *dst_buf;
+       struct vb2_v4l2_buffer *src_buf, *dst_buf;
        struct mtk_jpeg_src_buf *jpeg_src_buf;
        enum vb2_buffer_state buf_state = VB2_BUF_STATE_ERROR;
        u32     dec_irq_ret;
 
        src_buf = v4l2_m2m_src_buf_remove(ctx->fh.m2m_ctx);
        dst_buf = v4l2_m2m_dst_buf_remove(ctx->fh.m2m_ctx);
-       jpeg_src_buf = mtk_jpeg_vb2_to_srcbuf(src_buf);
+       jpeg_src_buf = mtk_jpeg_vb2_to_srcbuf(&src_buf->vb2_buf);
 
        if (dec_irq_ret >= MTK_JPEG_DEC_RESULT_UNDERFLOW)
                mtk_jpeg_dec_reset(jpeg->dec_reg_base);
                goto dec_end;
        }
 
-       for (i = 0; i < dst_buf->num_planes; i++)
-               vb2_set_plane_payload(dst_buf, i,
+       for (i = 0; i < dst_buf->vb2_buf.num_planes; i++)
+               vb2_set_plane_payload(&dst_buf->vb2_buf, i,
                                      jpeg_src_buf->dec_param.comp_size[i]);
 
        buf_state = VB2_BUF_STATE_DONE;
 
 dec_end:
-       v4l2_m2m_buf_done(to_vb2_v4l2_buffer(src_buf), buf_state);
-       v4l2_m2m_buf_done(to_vb2_v4l2_buffer(dst_buf), buf_state);
+       v4l2_m2m_buf_done(src_buf, buf_state);
+       v4l2_m2m_buf_done(dst_buf, buf_state);
        v4l2_m2m_job_finish(jpeg->m2m_dev, ctx->fh.m2m_ctx);
        return IRQ_HANDLED;
 }