summaryrefslogtreecommitdiff
path: root/libavcodec/eamad.c
diff options
context:
space:
mode:
authorMichael Niedermayer <michaelni@gmx.at>2014-07-27 12:23:53 +0200
committerMichael Niedermayer <michaelni@gmx.at>2014-07-27 12:27:01 +0200
commitc2ef844aa74c56b72a135fb389e346845ed7be50 (patch)
tree1ae29afaba4d1a9e8cf1b63dbd7670ebabfc9648 /libavcodec/eamad.c
parentc437765be7628c9dacb64aa1c1c46c0c39df1ca9 (diff)
parent541427ab4d5b4b6f5a90a687a06decdb78e7bc3c (diff)
Merge commit '541427ab4d5b4b6f5a90a687a06decdb78e7bc3c'
* commit '541427ab4d5b4b6f5a90a687a06decdb78e7bc3c': eamad: use the bytestream2 API instead of AV_RL Conflicts: libavcodec/eamad.c Merged-by: Michael Niedermayer <michaelni@gmx.at>
Diffstat (limited to 'libavcodec/eamad.c')
-rw-r--r--libavcodec/eamad.c41
1 files changed, 22 insertions, 19 deletions
diff --git a/libavcodec/eamad.c b/libavcodec/eamad.c
index 6457e2a134..711363ca29 100644
--- a/libavcodec/eamad.c
+++ b/libavcodec/eamad.c
@@ -29,6 +29,7 @@
*/
#include "avcodec.h"
+#include "bytestream.h"
#include "bswapdsp.h"
#include "get_bits.h"
#include "aandcttab.h"
@@ -243,30 +244,32 @@ static int decode_frame(AVCodecContext *avctx,
{
const uint8_t *buf = avpkt->data;
int buf_size = avpkt->size;
- const uint8_t *buf_end = buf+buf_size;
MadContext *s = avctx->priv_data;
AVFrame *frame = data;
+ GetByteContext gb;
int width, height;
int chunk_type;
int inter, ret;
- if (buf_size < 26) {
- av_log(avctx, AV_LOG_ERROR, "Input buffer too small\n");
- *got_frame = 0;
- return AVERROR_INVALIDDATA;
- }
+ bytestream2_init(&gb, buf, buf_size);
- chunk_type = AV_RL32(&buf[0]);
+ chunk_type = bytestream2_get_le32(&gb);
inter = (chunk_type == MADm_TAG || chunk_type == MADe_TAG);
- buf += 8;
+ bytestream2_skip(&gb, 10);
av_reduce(&avctx->time_base.num, &avctx->time_base.den,
- AV_RL16(&buf[6]), 1000, 1<<30);
+ bytestream2_get_le16(&gb), 1000, 1<<30);
+
+ width = bytestream2_get_le16(&gb);
+ height = bytestream2_get_le16(&gb);
+ bytestream2_skip(&gb, 1);
+ calc_quant_matrix(s, bytestream2_get_byte(&gb));
+ bytestream2_skip(&gb, 2);
- width = AV_RL16(&buf[8]);
- height = AV_RL16(&buf[10]);
- calc_quant_matrix(s, buf[13]);
- buf += 16;
+ if (bytestream2_get_bytes_left(&gb) < 2) {
+ av_log(avctx, AV_LOG_ERROR, "Input data too small\n");
+ return AVERROR_INVALIDDATA;
+ }
if (width < 16 || height < 16) {
av_log(avctx, AV_LOG_ERROR, "Dimensions too small\n");
@@ -275,7 +278,7 @@ static int decode_frame(AVCodecContext *avctx,
if (avctx->width != width || avctx->height != height) {
av_frame_unref(s->last_frame);
- if((width * height)/2048*7 > buf_end-buf)
+ if((width * height)/2048*7 > bytestream2_get_bytes_left(&gb))
return AVERROR_INVALIDDATA;
if ((ret = ff_set_dimensions(avctx, width, height)) < 0)
return ret;
@@ -298,13 +301,13 @@ static int decode_frame(AVCodecContext *avctx,
}
av_fast_padded_malloc(&s->bitstream_buf, &s->bitstream_buf_size,
- buf_end - buf);
+ bytestream2_get_bytes_left(&gb));
if (!s->bitstream_buf)
return AVERROR(ENOMEM);
- s->bbdsp.bswap16_buf(s->bitstream_buf, (const uint16_t *) buf,
- (buf_end - buf) / 2);
- memset((uint8_t*)s->bitstream_buf + (buf_end-buf), 0, FF_INPUT_BUFFER_PADDING_SIZE);
- init_get_bits(&s->gb, s->bitstream_buf, 8*(buf_end-buf));
+ s->bbdsp.bswap16_buf(s->bitstream_buf, (const uint16_t *)(buf + bytestream2_tell(&gb)),
+ bytestream2_get_bytes_left(&gb) / 2);
+ memset((uint8_t*)s->bitstream_buf + bytestream2_get_bytes_left(&gb), 0, FF_INPUT_BUFFER_PADDING_SIZE);
+ init_get_bits(&s->gb, s->bitstream_buf, 8*(bytestream2_get_bytes_left(&gb)));
for (s->mb_y=0; s->mb_y < (avctx->height+15)/16; s->mb_y++)
for (s->mb_x=0; s->mb_x < (avctx->width +15)/16; s->mb_x++)