X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=libavcodec%2Fcinepak.c;h=2b069925cb48907f7b894412dec9600a3d187059;hb=eb239a577fcf1f19cdaa7aeb6e6586a172ee59c1;hp=3b00a1605d34c2b4453fc729766bf1a467247703;hpb=472ea1284e925297b08921b7189f57f6789c898c;p=ffmpeg diff --git a/libavcodec/cinepak.c b/libavcodec/cinepak.c index 3b00a1605d3..2b069925cb4 100644 --- a/libavcodec/cinepak.c +++ b/libavcodec/cinepak.c @@ -2,44 +2,47 @@ * Cinepak Video Decoder * Copyright (C) 2003 the ffmpeg project * - * This library is free software; you can redistribute it and/or + * This file is part of Libav. + * + * Libav is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public * License as published by the Free Software Foundation; either - * version 2 of the License, or (at your option) any later version. + * version 2.1 of the License, or (at your option) any later version. * - * This library is distributed in the hope that it will be useful, + * Libav is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public - * License along with this library; if not, write to the Free Software - * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA - * + * License along with Libav; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA */ /** - * @file cinepak.c + * @file * Cinepak video decoder - * by Ewald Snel - * For more information on the Cinepak algorithm, visit: + * @author Ewald Snel + * + * @see For more information on the Cinepak algorithm, visit: * http://www.csse.monash.edu.au/~timf/ + * @see For more information on the quirky data inside Sega FILM/CPK files, visit: + * http://wiki.multimedia.cx/index.php?title=Sega_FILM */ #include #include #include -#include -#include "common.h" +#include "libavutil/common.h" +#include "libavutil/intreadwrite.h" #include "avcodec.h" -#include "dsputil.h" typedef struct { uint8_t y0, y1, y2, y3; uint8_t u, v; -} cvid_codebook_t; +} cvid_codebook; #define MAX_STRIPS 32 @@ -47,49 +50,51 @@ typedef struct { uint16_t id; uint16_t x1, y1; uint16_t x2, y2; - cvid_codebook_t v4_codebook[256]; - cvid_codebook_t v1_codebook[256]; -} cvid_strip_t; + cvid_codebook v4_codebook[256]; + cvid_codebook v1_codebook[256]; +} cvid_strip; typedef struct CinepakContext { AVCodecContext *avctx; - DSPContext dsp; AVFrame frame; - unsigned char *data; + const unsigned char *data; int size; int width, height; int palette_video; - cvid_strip_t strips[MAX_STRIPS]; + cvid_strip strips[MAX_STRIPS]; + + int sega_film_skip_bytes; + uint32_t pal[256]; } CinepakContext; -static void cinepak_decode_codebook (cvid_codebook_t *codebook, - int chunk_id, int size, uint8_t *data) +static void cinepak_decode_codebook (cvid_codebook *codebook, + int chunk_id, int size, const uint8_t *data) { - uint8_t *eod = (data + size); + const uint8_t *eod = (data + size); uint32_t flag, mask; int i, n; /* check if this chunk contains 4- or 6-element vectors */ - n = (chunk_id & 0x0400) ? 4 : 6; + n = (chunk_id & 0x04) ? 4 : 6; flag = 0; mask = 0; for (i=0; i < 256; i++) { - if ((chunk_id & 0x0100) && !(mask >>= 1)) { + if ((chunk_id & 0x01) && !(mask >>= 1)) { if ((data + 4) > eod) break; - flag = BE_32 (data); + flag = AV_RB32 (data); data += 4; mask = 0x80000000; } - if (!(chunk_id & 0x0100) || (flag & mask)) { + if (!(chunk_id & 0x01) || (flag & mask)) { if ((data + n) > eod) break; @@ -101,7 +106,7 @@ static void cinepak_decode_codebook (cvid_codebook_t *codebook, codebook[i].u = 128 + *data++; codebook[i].v = 128 + *data++; } else { - /* this codebook type indicates either greyscale or + /* this codebook type indicates either greyscale or * palettized video; if palettized, U & V components will * not be used so it is safe to set them to 128 for the * benefit of greyscale rendering in YUV420P */ @@ -116,12 +121,12 @@ static void cinepak_decode_codebook (cvid_codebook_t *codebook, } } -static int cinepak_decode_vectors (CinepakContext *s, cvid_strip_t *strip, - int chunk_id, int size, uint8_t *data) +static int cinepak_decode_vectors (CinepakContext *s, cvid_strip *strip, + int chunk_id, int size, const uint8_t *data) { - uint8_t *eod = (data + size); + const uint8_t *eod = (data + size); uint32_t flag, mask; - cvid_codebook_t *codebook; + cvid_codebook *codebook; unsigned int x, y; uint32_t iy[4]; uint32_t iu[2]; @@ -142,28 +147,28 @@ static int cinepak_decode_vectors (CinepakContext *s, cvid_strip_t *strip, iv[1] = iv[0] + s->frame.linesize[2]; for (x=strip->x1; x < strip->x2; x+=4) { - if ((chunk_id & 0x0100) && !(mask >>= 1)) { + if ((chunk_id & 0x01) && !(mask >>= 1)) { if ((data + 4) > eod) - return -1; + return AVERROR_INVALIDDATA; - flag = BE_32 (data); + flag = AV_RB32 (data); data += 4; mask = 0x80000000; } - if (!(chunk_id & 0x0100) || (flag & mask)) { - if (!(chunk_id & 0x0200) && !(mask >>= 1)) { + if (!(chunk_id & 0x01) || (flag & mask)) { + if (!(chunk_id & 0x02) && !(mask >>= 1)) { if ((data + 4) > eod) - return -1; + return AVERROR_INVALIDDATA; - flag = BE_32 (data); + flag = AV_RB32 (data); data += 4; mask = 0x80000000; } - if ((chunk_id & 0x0200) || (~flag & mask)) { + if ((chunk_id & 0x02) || (~flag & mask)) { if (data >= eod) - return -1; + return AVERROR_INVALIDDATA; codebook = &strip->v1_codebook[*data++]; s->frame.data[0][iy[0] + 0] = codebook->y0; @@ -204,7 +209,7 @@ static int cinepak_decode_vectors (CinepakContext *s, cvid_strip_t *strip, } else if (flag & mask) { if ((data + 4) > eod) - return -1; + return AVERROR_INVALIDDATA; codebook = &strip->v4_codebook[*data++]; s->frame.data[0][iy[0] + 0] = codebook->y0; @@ -260,84 +265,115 @@ static int cinepak_decode_vectors (CinepakContext *s, cvid_strip_t *strip, } static int cinepak_decode_strip (CinepakContext *s, - cvid_strip_t *strip, uint8_t *data, int size) + cvid_strip *strip, const uint8_t *data, int size) { - uint8_t *eod = (data + size); + const uint8_t *eod = (data + size); int chunk_id, chunk_size; /* coordinate sanity checks */ - if (strip->x1 >= s->width || strip->x2 > s->width || - strip->y1 >= s->height || strip->y2 > s->height || + if (strip->x2 > s->width || + strip->y2 > s->height || strip->x1 >= strip->x2 || strip->y1 >= strip->y2) - return -1; + return AVERROR_INVALIDDATA; while ((data + 4) <= eod) { - chunk_id = BE_16 (&data[0]); - chunk_size = BE_16 (&data[2]) - 4; + chunk_id = data[0]; + chunk_size = AV_RB24 (&data[1]) - 4; if(chunk_size < 0) - return -1; + return AVERROR_INVALIDDATA; data += 4; chunk_size = ((data + chunk_size) > eod) ? (eod - data) : chunk_size; switch (chunk_id) { - case 0x2000: - case 0x2100: - case 0x2400: - case 0x2500: - cinepak_decode_codebook (strip->v4_codebook, chunk_id, + case 0x20: + case 0x21: + case 0x24: + case 0x25: + cinepak_decode_codebook (strip->v4_codebook, chunk_id, chunk_size, data); break; - case 0x2200: - case 0x2300: - case 0x2600: - case 0x2700: - cinepak_decode_codebook (strip->v1_codebook, chunk_id, + case 0x22: + case 0x23: + case 0x26: + case 0x27: + cinepak_decode_codebook (strip->v1_codebook, chunk_id, chunk_size, data); break; - case 0x3000: - case 0x3100: - case 0x3200: - return cinepak_decode_vectors (s, strip, chunk_id, + case 0x30: + case 0x31: + case 0x32: + return cinepak_decode_vectors (s, strip, chunk_id, chunk_size, data); } data += chunk_size; } - return -1; + return AVERROR_INVALIDDATA; } static int cinepak_decode (CinepakContext *s) { - uint8_t *eod = (s->data + s->size); + const uint8_t *eod = (s->data + s->size); int i, result, strip_size, frame_flags, num_strips; int y0 = 0; + int encoded_buf_size; if (s->size < 10) - return -1; + return AVERROR_INVALIDDATA; frame_flags = s->data[0]; - num_strips = BE_16 (&s->data[8]); - s->data += 10; + num_strips = AV_RB16 (&s->data[8]); + encoded_buf_size = AV_RB24(&s->data[1]); + + /* if this is the first frame, check for deviant Sega FILM data */ + if (s->sega_film_skip_bytes == -1) { + if (!encoded_buf_size) { + av_log_ask_for_sample(s->avctx, "encoded_buf_size is 0"); + return AVERROR_INVALIDDATA; + } + if (encoded_buf_size != s->size && (s->size % encoded_buf_size) != 0) { + /* If the encoded frame size differs from the frame size as indicated + * by the container file, this data likely comes from a Sega FILM/CPK file. + * If the frame header is followed by the bytes FE 00 00 06 00 00 then + * this is probably one of the two known files that have 6 extra bytes + * after the frame header. Else, assume 2 extra bytes. The container + * size also cannot be a multiple of the encoded size. */ + if (s->size >= 16 && + (s->data[10] == 0xFE) && + (s->data[11] == 0x00) && + (s->data[12] == 0x00) && + (s->data[13] == 0x06) && + (s->data[14] == 0x00) && + (s->data[15] == 0x00)) + s->sega_film_skip_bytes = 6; + else + s->sega_film_skip_bytes = 2; + } else + s->sega_film_skip_bytes = 0; + } + + s->data += 10 + s->sega_film_skip_bytes; - if (num_strips > MAX_STRIPS) - num_strips = MAX_STRIPS; + num_strips = FFMIN(num_strips, MAX_STRIPS); for (i=0; i < num_strips; i++) { if ((s->data + 12) > eod) - return -1; + return AVERROR_INVALIDDATA; - s->strips[i].id = BE_16 (s->data); + s->strips[i].id = s->data[0]; s->strips[i].y1 = y0; s->strips[i].x1 = 0; - s->strips[i].y2 = y0 + BE_16 (&s->data[8]); + s->strips[i].y2 = y0 + AV_RB16 (&s->data[8]); s->strips[i].x2 = s->avctx->width; - strip_size = BE_16 (&s->data[2]) - 12; + strip_size = AV_RB24 (&s->data[1]) - 12; + if (strip_size < 0) + return AVERROR_INVALIDDATA; s->data += 12; strip_size = ((s->data + strip_size) > eod) ? (eod - s->data) : strip_size; @@ -359,16 +395,17 @@ static int cinepak_decode (CinepakContext *s) return 0; } -static int cinepak_decode_init(AVCodecContext *avctx) +static av_cold int cinepak_decode_init(AVCodecContext *avctx) { - CinepakContext *s = (CinepakContext *)avctx->priv_data; + CinepakContext *s = avctx->priv_data; s->avctx = avctx; s->width = (avctx->width + 3) & ~3; s->height = (avctx->height + 3) & ~3; + s->sega_film_skip_bytes = -1; /* uninitialized state */ // check for paletted data - if ((avctx->palctrl == NULL) || (avctx->bits_per_sample == 40)) { + if (avctx->bits_per_coded_sample != 8) { s->palette_video = 0; avctx->pix_fmt = PIX_FMT_YUV420P; } else { @@ -376,9 +413,6 @@ static int cinepak_decode_init(AVCodecContext *avctx) avctx->pix_fmt = PIX_FMT_PAL8; } - avctx->has_b_frames = 0; - dsputil_init(&s->dsp, avctx); - s->frame.data[0] = NULL; return 0; @@ -386,9 +420,11 @@ static int cinepak_decode_init(AVCodecContext *avctx) static int cinepak_decode_frame(AVCodecContext *avctx, void *data, int *data_size, - uint8_t *buf, int buf_size) + AVPacket *avpkt) { - CinepakContext *s = (CinepakContext *)avctx->priv_data; + const uint8_t *buf = avpkt->data; + int ret = 0, buf_size = avpkt->size; + CinepakContext *s = avctx->priv_data; s->data = buf; s->size = buf_size; @@ -396,22 +432,24 @@ static int cinepak_decode_frame(AVCodecContext *avctx, s->frame.reference = 1; s->frame.buffer_hints = FF_BUFFER_HINTS_VALID | FF_BUFFER_HINTS_PRESERVE | FF_BUFFER_HINTS_REUSABLE; - if (avctx->reget_buffer(avctx, &s->frame)) { + if ((ret = avctx->reget_buffer(avctx, &s->frame))) { av_log(avctx, AV_LOG_ERROR, "reget_buffer() failed\n"); - return -1; + return ret; } - cinepak_decode(s); - if (s->palette_video) { - memcpy (s->frame.data[1], avctx->palctrl->palette, AVPALETTE_SIZE); - if (avctx->palctrl->palette_changed) { + const uint8_t *pal = av_packet_get_side_data(avpkt, AV_PKT_DATA_PALETTE, NULL); + if (pal) { s->frame.palette_has_changed = 1; - avctx->palctrl->palette_changed = 0; - } else - s->frame.palette_has_changed = 0; + memcpy(s->pal, pal, AVPALETTE_SIZE); + } } + cinepak_decode(s); + + if (s->palette_video) + memcpy (s->frame.data[1], s->pal, AVPALETTE_SIZE); + *data_size = sizeof(AVFrame); *(AVFrame*)data = s->frame; @@ -419,9 +457,9 @@ static int cinepak_decode_frame(AVCodecContext *avctx, return buf_size; } -static int cinepak_decode_end(AVCodecContext *avctx) +static av_cold int cinepak_decode_end(AVCodecContext *avctx) { - CinepakContext *s = (CinepakContext *)avctx->priv_data; + CinepakContext *s = avctx->priv_data; if (s->frame.data[0]) avctx->release_buffer(avctx, &s->frame); @@ -429,14 +467,14 @@ static int cinepak_decode_end(AVCodecContext *avctx) return 0; } -AVCodec cinepak_decoder = { - "cinepak", - CODEC_TYPE_VIDEO, - CODEC_ID_CINEPAK, - sizeof(CinepakContext), - cinepak_decode_init, - NULL, - cinepak_decode_end, - cinepak_decode_frame, - CODEC_CAP_DR1, +AVCodec ff_cinepak_decoder = { + .name = "cinepak", + .type = AVMEDIA_TYPE_VIDEO, + .id = AV_CODEC_ID_CINEPAK, + .priv_data_size = sizeof(CinepakContext), + .init = cinepak_decode_init, + .close = cinepak_decode_end, + .decode = cinepak_decode_frame, + .capabilities = CODEC_CAP_DR1, + .long_name = NULL_IF_CONFIG_SMALL("Cinepak"), };