From 6575ebdc4570b54ad98eaa2db19703a636acdc9d Mon Sep 17 00:00:00 2001 From: Leo Liu Date: Mon, 8 Aug 2016 16:01:23 -0400 Subject: vl/rbsp: add a check for emulation prevention three byte MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit This is the case when the "00 00 03" is very close to the beginning of nal unit header v2: move the check to rbsp init Signed-off-by: Leo Liu Reviewed-by: Christian König --- src/gallium/auxiliary/vl/vl_rbsp.h | 14 ++++++++++++-- 1 file changed, 12 insertions(+), 2 deletions(-) diff --git a/src/gallium/auxiliary/vl/vl_rbsp.h b/src/gallium/auxiliary/vl/vl_rbsp.h index 7867238c49..c8bebff4de 100644 --- a/src/gallium/auxiliary/vl/vl_rbsp.h +++ b/src/gallium/auxiliary/vl/vl_rbsp.h @@ -50,7 +50,8 @@ struct vl_rbsp { */ static inline void vl_rbsp_init(struct vl_rbsp *rbsp, struct vl_vlc *nal, unsigned num_bits) { - unsigned bits_left = vl_vlc_bits_left(nal); + unsigned valid, bits_left = vl_vlc_bits_left(nal); + int i; /* copy the position */ rbsp->nal = *nal; @@ -62,10 +63,19 @@ static inline void vl_rbsp_init(struct vl_rbsp *rbsp, struct vl_vlc *nal, unsign if (vl_vlc_peekbits(nal, 24) == 0x000001 || vl_vlc_peekbits(nal, 32) == 0x00000001) { vl_vlc_limit(&rbsp->nal, bits_left - vl_vlc_bits_left(nal)); - return; + break; } vl_vlc_eatbits(nal, 8); } + + valid = vl_vlc_valid_bits(&rbsp->nal); + /* search for the emulation prevention three byte */ + for (i = 24; i <= valid; i += 8) { + if ((vl_vlc_peekbits(&rbsp->nal, i) & 0xffffff) == 0x3) { + vl_vlc_removebits(&rbsp->nal, i - 8, 8); + i += 8; + } + } } /** -- cgit v1.2.3