From 6560fa390e9a296276faaa22779936f187f05a96 Mon Sep 17 00:00:00 2001 From: Michael Niedermayer Date: Sat, 7 Apr 2012 16:04:09 +0200 Subject: get_bits: check the number of bits parameter through av_assert2() Signed-off-by: Michael Niedermayer --- libavcodec/get_bits.h | 4 ++++ 1 file changed, 4 insertions(+) (limited to 'libavcodec/get_bits.h') diff --git a/libavcodec/get_bits.h b/libavcodec/get_bits.h index 09b846eb42..2123972fb0 100644 --- a/libavcodec/get_bits.h +++ b/libavcodec/get_bits.h @@ -30,6 +30,7 @@ #include "libavutil/common.h" #include "libavutil/intreadwrite.h" #include "libavutil/log.h" +#include "libavutil/avassert.h" #include "mathops.h" /* @@ -222,6 +223,7 @@ static inline int get_sbits(GetBitContext *s, int n) { register int tmp; OPEN_READER(re, s); + av_assert2(n>0 && n<=25); UPDATE_CACHE(re, s); tmp = SHOW_SBITS(re, s, n); LAST_SKIP_BITS(re, s, n); @@ -236,6 +238,7 @@ static inline unsigned int get_bits(GetBitContext *s, int n) { register int tmp; OPEN_READER(re, s); + av_assert2(n>0 && n<=25); UPDATE_CACHE(re, s); tmp = SHOW_UBITS(re, s, n); LAST_SKIP_BITS(re, s, n); @@ -250,6 +253,7 @@ static inline unsigned int show_bits(GetBitContext *s, int n) { register int tmp; OPEN_READER(re, s); + av_assert2(n>0 && n<=25); UPDATE_CACHE(re, s); tmp = SHOW_UBITS(re, s, n); return tmp; -- cgit v1.2.3