Mercurial > libavcodec.hg
annotate r210dec.c @ 12494:94eaea836bf4 libavcodec
Check avctx width/height more thoroughly (e.g. all values 0 except width would
have been accepted before).
Also do not fail if they are invalid but instead override them to 0.
This allows decoding e.g. MPEG video when only the container values are corrupted.
For encoding a value of 0,0 of course makes no sense, but was allowed
through before and will be caught by an extra check in the encode function.
author | reimar |
---|---|
date | Wed, 15 Sep 2010 04:46:55 +0000 |
parents | 990f8a5fc8af |
children |
rev | line source |
---|---|
10693 | 1 /* |
2 * R210 decoder | |
3 * | |
4 * Copyright (c) 2009 Reimar Doeffinger <Reimar.Doeffinger@gmx.de> | |
5 * | |
6 * This file is part of FFmpeg. | |
7 * | |
8 * FFmpeg is free software; you can redistribute it and/or | |
9 * modify it under the terms of the GNU Lesser General Public | |
10 * License as published by the Free Software Foundation; either | |
11 * version 2.1 of the License, or (at your option) any later version. | |
12 * | |
13 * FFmpeg is distributed in the hope that it will be useful, | |
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
16 * Lesser General Public License for more details. | |
17 * | |
18 * You should have received a copy of the GNU Lesser General Public | |
19 * License along with FFmpeg; if not, write to the Free Software | |
20 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | |
21 */ | |
22 | |
23 #include "avcodec.h" | |
24 #include "libavutil/bswap.h" | |
25 | |
26 static av_cold int decode_init(AVCodecContext *avctx) | |
27 { | |
28 avctx->pix_fmt = PIX_FMT_RGB48; | |
29 avctx->bits_per_raw_sample = 10; | |
30 | |
31 avctx->coded_frame = avcodec_alloc_frame(); | |
32 | |
33 return 0; | |
34 } | |
35 | |
36 static int decode_frame(AVCodecContext *avctx, void *data, int *data_size, | |
37 AVPacket *avpkt) | |
38 { | |
39 int h, w; | |
40 AVFrame *pic = avctx->coded_frame; | |
41 const uint32_t *src = (const uint32_t *)avpkt->data; | |
42 int aligned_width = FFALIGN(avctx->width, 64); | |
43 uint8_t *dst_line; | |
44 | |
45 if (pic->data[0]) | |
46 avctx->release_buffer(avctx, pic); | |
47 | |
48 if (avpkt->size < 4 * aligned_width * avctx->height) { | |
49 av_log(avctx, AV_LOG_ERROR, "packet too small\n"); | |
50 return -1; | |
51 } | |
52 | |
53 pic->reference = 0; | |
54 if (avctx->get_buffer(avctx, pic) < 0) | |
55 return -1; | |
56 | |
57 pic->pict_type = FF_I_TYPE; | |
58 pic->key_frame = 1; | |
59 dst_line = pic->data[0]; | |
60 | |
61 for (h = 0; h < avctx->height; h++) { | |
62 uint16_t *dst = (uint16_t *)dst_line; | |
63 for (w = 0; w < avctx->width; w++) { | |
12129 | 64 uint32_t pixel = av_be2ne32(*src++); |
10693 | 65 uint16_t r, g, b; |
12490 | 66 if (avctx->codec_id==CODEC_ID_R210) { |
12491 | 67 b = pixel << 6; |
68 g = (pixel >> 4) & 0xffc0; | |
69 r = (pixel >> 14) & 0xffc0; | |
12490 | 70 } else { |
71 b = pixel << 4; | |
72 g = (pixel >> 6) & 0xffc0; | |
73 r = (pixel >> 16) & 0xffc0; | |
74 } | |
10693 | 75 *dst++ = r | (r >> 10); |
76 *dst++ = g | (g >> 10); | |
77 *dst++ = b | (b >> 10); | |
78 } | |
79 src += aligned_width - avctx->width; | |
80 dst_line += pic->linesize[0]; | |
81 } | |
82 | |
83 *data_size = sizeof(AVFrame); | |
84 *(AVFrame*)data = *avctx->coded_frame; | |
85 | |
86 return avpkt->size; | |
87 } | |
88 | |
89 static av_cold int decode_close(AVCodecContext *avctx) | |
90 { | |
91 AVFrame *pic = avctx->coded_frame; | |
92 if (pic->data[0]) | |
93 avctx->release_buffer(avctx, pic); | |
94 av_freep(&avctx->coded_frame); | |
95 | |
96 return 0; | |
97 } | |
98 | |
12490 | 99 #if CONFIG_R210_DECODER |
10693 | 100 AVCodec r210_decoder = { |
101 "r210", | |
11560
8a4984c5cacc
Define AVMediaType enum, and use it instead of enum CodecType, which
stefano
parents:
10694
diff
changeset
|
102 AVMEDIA_TYPE_VIDEO, |
10693 | 103 CODEC_ID_R210, |
104 0, | |
105 decode_init, | |
106 NULL, | |
107 decode_close, | |
108 decode_frame, | |
109 CODEC_CAP_DR1, | |
110 .long_name = NULL_IF_CONFIG_SMALL("Uncompressed RGB 10-bit"), | |
111 }; | |
12490 | 112 #endif |
113 #if CONFIG_R10K_DECODER | |
114 AVCodec r10k_decoder = { | |
115 "r10k", | |
116 AVMEDIA_TYPE_VIDEO, | |
117 CODEC_ID_R10K, | |
118 0, | |
119 decode_init, | |
120 NULL, | |
121 decode_close, | |
122 decode_frame, | |
123 CODEC_CAP_DR1, | |
124 .long_name = NULL_IF_CONFIG_SMALL("AJA Kona 10-bit RGB Codec"), | |
125 }; | |
126 #endif |