[PATCH] media: hantro: Use vb2_get_buffer

[Ezequiel Garcia]

Use the newly introduced vb2_get_buffer API and avoid
accessing buffers in the queue directly.

Signed-off-by: Ezequiel Garcia <ezequ...@collabora.com>
---
 drivers/staging/media/hantro/hantro_drv.c | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/drivers/staging/media/hantro/hantro_drv.c 
b/drivers/staging/media/hantro/hantro_drv.c
index 1d3af881d088..c3665f0e87a2 100644
--- a/drivers/staging/media/hantro/hantro_drv.c
+++ b/drivers/staging/media/hantro/hantro_drv.c
@@ -45,12 +45,14 @@ void *hantro_get_ctrl(struct hantro_ctx *ctx, u32 id)
 
 dma_addr_t hantro_get_ref(struct vb2_queue *q, u64 ts)
 {
+       struct vb2_buffer *buf;
        int index;
 
        index = vb2_find_timestamp(q, ts, 0);
-       if (index >= 0)
-               return vb2_dma_contig_plane_dma_addr(q->bufs[index], 0);
-       return 0;
+       if (index < 0)
+               return 0;
+       buf = vb2_get_buffer(q, index);
+       return vb2_dma_contig_plane_dma_addr(buf, 0);
 }
 
 static int
-- 
2.20.1