FFmpeg coverage

Directory: ../../../ffmpeg/
File: src/libavcodec/mjpegdec.c
Date: 2022-07-05 19:52:29
Exec Total Coverage
Lines: 971 1830 53.1%
Branches: 699 1665 42.0%
Line Branch Exec Source
1 /*
2 * MJPEG decoder
3 * Copyright (c) 2000, 2001 Fabrice Bellard
4 * Copyright (c) 2003 Alex Beregszaszi
5 * Copyright (c) 2003-2004 Michael Niedermayer
6 *
7 * Support for external huffman table, various fixes (AVID workaround),
8 * aspecting, new decode_frame mechanism and apple mjpeg-b support
9 * by Alex Beregszaszi
10 *
11 * This file is part of FFmpeg.
12 *
13 * FFmpeg is free software; you can redistribute it and/or
14 * modify it under the terms of the GNU Lesser General Public
15 * License as published by the Free Software Foundation; either
16 * version 2.1 of the License, or (at your option) any later version.
17 *
18 * FFmpeg is distributed in the hope that it will be useful,
19 * but WITHOUT ANY WARRANTY; without even the implied warranty of
20 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
21 * Lesser General Public License for more details.
22 *
23 * You should have received a copy of the GNU Lesser General Public
24 * License along with FFmpeg; if not, write to the Free Software
25 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
26 */
27
28 /**
29 * @file
30 * MJPEG decoder.
31 */
32
33 #include "config_components.h"
34
35 #include "libavutil/display.h"
36 #include "libavutil/imgutils.h"
37 #include "libavutil/avassert.h"
38 #include "libavutil/opt.h"
39 #include "avcodec.h"
40 #include "blockdsp.h"
41 #include "codec_internal.h"
42 #include "copy_block.h"
43 #include "decode.h"
44 #include "hwconfig.h"
45 #include "idctdsp.h"
46 #include "internal.h"
47 #include "jpegtables.h"
48 #include "mjpeg.h"
49 #include "mjpegdec.h"
50 #include "jpeglsdec.h"
51 #include "profiles.h"
52 #include "put_bits.h"
53 #include "tiff.h"
54 #include "exif.h"
55 #include "bytestream.h"
56 #include "tiff_common.h"
57
58
59 179 static int init_default_huffman_tables(MJpegDecodeContext *s)
60 {
61 static const struct {
62 int class;
63 int index;
64 const uint8_t *bits;
65 const uint8_t *values;
66 int length;
67 } ht[] = {
68 { 0, 0, ff_mjpeg_bits_dc_luminance,
69 ff_mjpeg_val_dc, 12 },
70 { 0, 1, ff_mjpeg_bits_dc_chrominance,
71 ff_mjpeg_val_dc, 12 },
72 { 1, 0, ff_mjpeg_bits_ac_luminance,
73 ff_mjpeg_val_ac_luminance, 162 },
74 { 1, 1, ff_mjpeg_bits_ac_chrominance,
75 ff_mjpeg_val_ac_chrominance, 162 },
76 { 2, 0, ff_mjpeg_bits_ac_luminance,
77 ff_mjpeg_val_ac_luminance, 162 },
78 { 2, 1, ff_mjpeg_bits_ac_chrominance,
79 ff_mjpeg_val_ac_chrominance, 162 },
80 };
81 int i, ret;
82
83
2/2
✓ Branch 0 taken 1074 times.
✓ Branch 1 taken 179 times.
 1253 for (i = 0; i < FF_ARRAY_ELEMS(ht); i++) {
84 1074 ff_free_vlc(&s->vlcs[ht[i].class][ht[i].index]);
85 1074 ret = ff_mjpeg_build_vlc(&s->vlcs[ht[i].class][ht[i].index],
86 1074 ht[i].bits, ht[i].values,
87 1074 ht[i].class == 1, s->avctx);
88
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1074 times.
 1074 if (ret < 0)
89 ✗ return ret;
90
91
2/2
✓ Branch 0 taken 716 times.
✓ Branch 1 taken 358 times.
 1074 if (ht[i].class < 2) {
92 716 memcpy(s->raw_huffman_lengths[ht[i].class][ht[i].index],
93 716 ht[i].bits + 1, 16);
94 716 memcpy(s->raw_huffman_values[ht[i].class][ht[i].index],
95 716 ht[i].values, ht[i].length);
96 }
97 }
98
99 179 return 0;
100 }
101
102 11 static void parse_avid(MJpegDecodeContext *s, uint8_t *buf, int len)
103 {
104 11 s->buggy_avid = 1;
105
2/4
✓ Branch 0 taken 11 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 11 times.
✗ Branch 3 not taken.
 11 if (len > 14 && buf[12] == 1) /* 1 - NTSC */
106 11 s->interlace_polarity = 1;
107
2/4
✓ Branch 0 taken 11 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 11 times.
 11 if (len > 14 && buf[12] == 2) /* 2 - PAL */
108 ✗ s->interlace_polarity = 0;
109
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 11 times.
 11 if (s->avctx->debug & FF_DEBUG_PICT_INFO)
110 ✗ av_log(s->avctx, AV_LOG_INFO, "AVID: len:%d %d\n", len, len > 14 ? buf[12] : -1);
111 11 }
112
113 276 static void init_idct(AVCodecContext *avctx)
114 {
115 276 MJpegDecodeContext *s = avctx->priv_data;
116
117 276 ff_idctdsp_init(&s->idsp, avctx);
118 276 ff_init_scantable(s->idsp.idct_permutation, &s->scantable,
119 ff_zigzag_direct);
120 276 }
121
122 179 av_cold int ff_mjpeg_decode_init(AVCodecContext *avctx)
123 {
124 179 MJpegDecodeContext *s = avctx->priv_data;
125 int ret;
126
127
2/2
✓ Branch 0 taken 177 times.
✓ Branch 1 taken 2 times.
 179 if (!s->picture_ptr) {
128 177 s->picture = av_frame_alloc();
129
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 177 times.
 177 if (!s->picture)
130 ✗ return AVERROR(ENOMEM);
131 177 s->picture_ptr = s->picture;
132 }
133
134 179 s->pkt = avctx->internal->in_pkt;
135
136 179 s->avctx = avctx;
137 179 ff_blockdsp_init(&s->bdsp, avctx);
138 179 ff_hpeldsp_init(&s->hdsp, avctx->flags);
139 179 init_idct(avctx);
140 179 s->buffer_size = 0;
141 179 s->buffer = NULL;
142 179 s->start_code = -1;
143 179 s->first_picture = 1;
144 179 s->got_picture = 0;
145 179 s->orig_height = avctx->coded_height;
146 179 avctx->chroma_sample_location = AVCHROMA_LOC_CENTER;
147 179 avctx->colorspace = AVCOL_SPC_BT470BG;
148 179 s->hwaccel_pix_fmt = s->hwaccel_sw_pix_fmt = AV_PIX_FMT_NONE;
149
150
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 179 times.
 179 if ((ret = init_default_huffman_tables(s)) < 0)
151 ✗ return ret;
152
153
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 179 times.
 179 if (s->extern_huff) {
154 ✗ av_log(avctx, AV_LOG_INFO, "using external huffman table\n");
155 ✗ if ((ret = init_get_bits(&s->gb, avctx->extradata, avctx->extradata_size * 8)) < 0)
156 ✗ return ret;
157 ✗ if (ff_mjpeg_decode_dht(s)) {
158 ✗ av_log(avctx, AV_LOG_ERROR,
159 "error using external huffman table, switching back to internal\n");
160 ✗ if ((ret = init_default_huffman_tables(s)) < 0)
161 ✗ return ret;
162 }
163 }
164
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 179 times.
 179 if (avctx->field_order == AV_FIELD_BB) { /* quicktime icefloe 019 */
165 ✗ s->interlace_polarity = 1; /* bottom field first */
166 ✗ av_log(avctx, AV_LOG_DEBUG, "bottom field first\n");
167
2/2
✓ Branch 0 taken 177 times.
✓ Branch 1 taken 2 times.
 179 } else if (avctx->field_order == AV_FIELD_UNKNOWN) {
168
2/2
✓ Branch 0 taken 51 times.
✓ Branch 1 taken 126 times.
 177 if (avctx->codec_tag == AV_RL32("MJPG"))
169 51 s->interlace_polarity = 1;
170 }
171
172
2/2
✓ Branch 0 taken 2 times.
✓ Branch 1 taken 177 times.
 179 if (avctx->codec_id == AV_CODEC_ID_SMVJPEG) {
173
1/2
✓ Branch 0 taken 2 times.
✗ Branch 1 not taken.
 2 if (avctx->extradata_size >= 4)
174 2 s->smv_frames_per_jpeg = AV_RL32(avctx->extradata);
175
176
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2 times.
 2 if (s->smv_frames_per_jpeg <= 0) {
177 ✗ av_log(avctx, AV_LOG_ERROR, "Invalid number of frames per jpeg.\n");
178 ✗ return AVERROR_INVALIDDATA;
179 }
180
181 2 s->smv_frame = av_frame_alloc();
182
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2 times.
 2 if (!s->smv_frame)
183 ✗ return AVERROR(ENOMEM);
184
2/2
✓ Branch 0 taken 2 times.
✓ Branch 1 taken 175 times.
 177 } else if (avctx->extradata_size > 8
185
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2 times.
 2 && AV_RL32(avctx->extradata) == 0x2C
186 ✗ && AV_RL32(avctx->extradata+4) == 0x18) {
187 ✗ parse_avid(s, avctx->extradata, avctx->extradata_size);
188 }
189
190
2/2
✓ Branch 0 taken 11 times.
✓ Branch 1 taken 168 times.
 179 if (avctx->codec->id == AV_CODEC_ID_AMV)
191 11 s->flipped = 1;
192
193 179 return 0;
194 }
195
196
197 /* quantize tables */
198 2565 int ff_mjpeg_decode_dqt(MJpegDecodeContext *s)
199 {
200 int len, index, i;
201
202 2565 len = get_bits(&s->gb, 16) - 2;
203
204
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 2565 times.
 2565 if (8*len > get_bits_left(&s->gb)) {
205 ✗ av_log(s->avctx, AV_LOG_ERROR, "dqt: len %d is too large\n", len);
206 ✗ return AVERROR_INVALIDDATA;
207 }
208
209
2/2
✓ Branch 0 taken 3060 times.
✓ Branch 1 taken 2565 times.
 5625 while (len >= 65) {
210 3060 int pr = get_bits(&s->gb, 4);
211
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 3060 times.
 3060 if (pr > 1) {
212 ✗ av_log(s->avctx, AV_LOG_ERROR, "dqt: invalid precision\n");
213 ✗ return AVERROR_INVALIDDATA;
214 }
215 3060 index = get_bits(&s->gb, 4);
216
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 3060 times.
 3060 if (index >= 4)
217 ✗ return -1;
218 3060 av_log(s->avctx, AV_LOG_DEBUG, "index=%d\n", index);
219 /* read quant table */
220
2/2
✓ Branch 0 taken 195840 times.
✓ Branch 1 taken 3060 times.
 198900 for (i = 0; i < 64; i++) {
221
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 195840 times.
 195840 s->quant_matrixes[index][i] = get_bits(&s->gb, pr ? 16 : 8);
222
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 195840 times.
 195840 if (s->quant_matrixes[index][i] == 0) {
223 ✗ int log_level = s->avctx->err_recognition & AV_EF_EXPLODE ? AV_LOG_ERROR : AV_LOG_WARNING;
224 ✗ av_log(s->avctx, log_level, "dqt: 0 quant value\n");
225 ✗ if (s->avctx->err_recognition & AV_EF_EXPLODE)
226 ✗ return AVERROR_INVALIDDATA;
227 }
228 }
229
230 // XXX FIXME fine-tune, and perhaps add dc too
231 3060 s->qscale[index] = FFMAX(s->quant_matrixes[index][1],
232 3060 s->quant_matrixes[index][8]) >> 1;
233 3060 av_log(s->avctx, AV_LOG_DEBUG, "qscale[%d]: %d\n",
234 index, s->qscale[index]);
235 3060 len -= 1 + 64 * (1+pr);
236 }
237 2565 return 0;
238 }
239
240 /* decode huffman tables and build VLC decoders */
241 2562 int ff_mjpeg_decode_dht(MJpegDecodeContext *s)
242 {
243 int len, index, i, class, n, v;
244 uint8_t bits_table[17];
245 uint8_t val_table[256];
246 2562 int ret = 0;
247
248 2562 len = get_bits(&s->gb, 16) - 2;
249
250
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 2562 times.
 2562 if (8*len > get_bits_left(&s->gb)) {
251 ✗ av_log(s->avctx, AV_LOG_ERROR, "dht: len %d is too large\n", len);
252 ✗ return AVERROR_INVALIDDATA;
253 }
254
255
2/2
✓ Branch 0 taken 9154 times.
✓ Branch 1 taken 2562 times.
 11716 while (len > 0) {
256
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 9154 times.
 9154 if (len < 17)
257 ✗ return AVERROR_INVALIDDATA;
258 9154 class = get_bits(&s->gb, 4);
259
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 9154 times.
 9154 if (class >= 2)
260 ✗ return AVERROR_INVALIDDATA;
261 9154 index = get_bits(&s->gb, 4);
262
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 9154 times.
 9154 if (index >= 4)
263 ✗ return AVERROR_INVALIDDATA;
264 9154 n = 0;
265
2/2
✓ Branch 0 taken 146464 times.
✓ Branch 1 taken 9154 times.
 155618 for (i = 1; i <= 16; i++) {
266 146464 bits_table[i] = get_bits(&s->gb, 8);
267 146464 n += bits_table[i];
268 }
269 9154 len -= 17;
270
2/4
✓ Branch 0 taken 9154 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 9154 times.
 9154 if (len < n || n > 256)
271 ✗ return AVERROR_INVALIDDATA;
272
273
2/2
✓ Branch 0 taken 450414 times.
✓ Branch 1 taken 9154 times.
 459568 for (i = 0; i < n; i++) {
274 450414 v = get_bits(&s->gb, 8);
275 450414 val_table[i] = v;
276 }
277 9154 len -= n;
278
279 /* build VLC and flush previous vlc if present */
280 9154 ff_free_vlc(&s->vlcs[class][index]);
281 9154 av_log(s->avctx, AV_LOG_DEBUG, "class=%d index=%d nb_codes=%d\n",
282 class, index, n);
283
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 9154 times.
 9154 if ((ret = ff_mjpeg_build_vlc(&s->vlcs[class][index], bits_table,
284 9154 val_table, class > 0, s->avctx)) < 0)
285 ✗ return ret;
286
287
2/2
✓ Branch 0 taken 4577 times.
✓ Branch 1 taken 4577 times.
 9154 if (class > 0) {
288 4577 ff_free_vlc(&s->vlcs[2][index]);
289
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 4577 times.
 4577 if ((ret = ff_mjpeg_build_vlc(&s->vlcs[2][index], bits_table,
290 4577 val_table, 0, s->avctx)) < 0)
291 ✗ return ret;
292 }
293
294
2/2
✓ Branch 0 taken 146464 times.
✓ Branch 1 taken 9154 times.
 155618 for (i = 0; i < 16; i++)
295 146464 s->raw_huffman_lengths[class][index][i] = bits_table[i + 1];
296
2/2
✓ Branch 0 taken 2343424 times.
✓ Branch 1 taken 9154 times.
 2352578 for (i = 0; i < 256; i++)
297 2343424 s->raw_huffman_values[class][index][i] = val_table[i];
298 }
299 2562 return 0;
300 }
301
302 2592 int ff_mjpeg_decode_sof(MJpegDecodeContext *s)
303 {
304 int len, nb_components, i, width, height, bits, ret, size_change;
305 unsigned pix_fmt_id;
306 2592 int h_count[MAX_COMPONENTS] = { 0 };
307 2592 int v_count[MAX_COMPONENTS] = { 0 };
308
309 2592 s->cur_scan = 0;
310 2592 memset(s->upscale_h, 0, sizeof(s->upscale_h));
311 2592 memset(s->upscale_v, 0, sizeof(s->upscale_v));
312
313 2592 len = get_bits(&s->gb, 16);
314 2592 bits = get_bits(&s->gb, 8);
315
316
2/4
✓ Branch 0 taken 2592 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 2592 times.
 2592 if (bits > 16 || bits < 1) {
317 ✗ av_log(s->avctx, AV_LOG_ERROR, "bits %d is invalid\n", bits);
318 ✗ return AVERROR_INVALIDDATA;
319 }
320
321
2/2
✓ Branch 0 taken 97 times.
✓ Branch 1 taken 2495 times.
 2592 if (s->avctx->bits_per_raw_sample != bits) {
322
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 97 times.
 97 av_log(s->avctx, s->avctx->bits_per_raw_sample > 0 ? AV_LOG_INFO : AV_LOG_DEBUG, "Changing bps from %d to %d\n", s->avctx->bits_per_raw_sample, bits);
323 97 s->avctx->bits_per_raw_sample = bits;
324 97 init_idct(s->avctx);
325 }
326
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2592 times.
 2592 if (s->pegasus_rct)
327 ✗ bits = 9;
328
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 2592 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 2592 if (bits == 9 && !s->pegasus_rct)
329 ✗ s->rct = 1; // FIXME ugly
330
331
3/4
✓ Branch 0 taken 426 times.
✓ Branch 1 taken 2166 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 426 times.
 2592 if(s->lossless && s->avctx->lowres){
332 ✗ av_log(s->avctx, AV_LOG_ERROR, "lowres is not possible with lossless jpeg\n");
333 ✗ return -1;
334 }
335
336 2592 height = get_bits(&s->gb, 16);
337 2592 width = get_bits(&s->gb, 16);
338
339 // HACK for odd_height.mov
340
4/6
✓ Branch 0 taken 26 times.
✓ Branch 1 taken 2566 times.
✓ Branch 2 taken 26 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 26 times.
 2592 if (s->interlaced && s->width == width && s->height == height + 1)
341 ✗ height= s->height;
342
343 2592 av_log(s->avctx, AV_LOG_DEBUG, "sof0: picture: %dx%d\n", width, height);
344
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 2592 times.
 2592 if (av_image_check_size(width, height, 0, s->avctx) < 0)
345 ✗ return AVERROR_INVALIDDATA;
346
3/4
✓ Branch 0 taken 2590 times.
✓ Branch 1 taken 2 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 2590 times.
 2592 if (s->buf_size && (width + 7) / 8 * ((height + 7) / 8) > s->buf_size * 4LL)
347 ✗ return AVERROR_INVALIDDATA;
348
349 2592 nb_components = get_bits(&s->gb, 8);
350
2/4
✓ Branch 0 taken 2592 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 2592 times.
 2592 if (nb_components <= 0 ||
351 nb_components > MAX_COMPONENTS)
352 ✗ return -1;
353
4/4
✓ Branch 0 taken 26 times.
✓ Branch 1 taken 2566 times.
✓ Branch 2 taken 15 times.
✓ Branch 3 taken 11 times.
 2592 if (s->interlaced && (s->bottom_field == !s->interlace_polarity)) {
354
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 15 times.
 15 if (nb_components != s->nb_components) {
355 ✗ av_log(s->avctx, AV_LOG_ERROR,
356 "nb_components changing in interlaced picture\n");
357 ✗ return AVERROR_INVALIDDATA;
358 }
359 }
360
3/6
✓ Branch 0 taken 221 times.
✓ Branch 1 taken 2371 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 221 times.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
 2592 if (s->ls && !(bits <= 8 || nb_components == 1)) {
361 ✗ avpriv_report_missing_feature(s->avctx,
362 "JPEG-LS that is not <= 8 "
363 "bits/component or 16-bit gray");
364 ✗ return AVERROR_PATCHWELCOME;
365 }
366
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2592 times.
 2592 if (len != 8 + 3 * nb_components) {
367 ✗ av_log(s->avctx, AV_LOG_ERROR, "decode_sof0: error, len(%d) mismatch %d components\n", len, nb_components);
368 ✗ return AVERROR_INVALIDDATA;
369 }
370
371 2592 s->nb_components = nb_components;
372 2592 s->h_max = 1;
373 2592 s->v_max = 1;
374
2/2
✓ Branch 0 taken 7740 times.
✓ Branch 1 taken 2592 times.
 10332 for (i = 0; i < nb_components; i++) {
375 /* component id */
376 7740 s->component_id[i] = get_bits(&s->gb, 8) - 1;
377 7740 h_count[i] = get_bits(&s->gb, 4);
378 7740 v_count[i] = get_bits(&s->gb, 4);
379 /* compute hmax and vmax (only used in interleaved case) */
380
2/2
✓ Branch 0 taken 2153 times.
✓ Branch 1 taken 5587 times.
 7740 if (h_count[i] > s->h_max)
381 2153 s->h_max = h_count[i];
382
2/2
✓ Branch 0 taken 2318 times.
✓ Branch 1 taken 5422 times.
 7740 if (v_count[i] > s->v_max)
383 2318 s->v_max = v_count[i];
384 7740 s->quant_index[i] = get_bits(&s->gb, 8);
385
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 7740 times.
 7740 if (s->quant_index[i] >= 4) {
386 ✗ av_log(s->avctx, AV_LOG_ERROR, "quant_index is invalid\n");
387 ✗ return AVERROR_INVALIDDATA;
388 }
389
2/4
✓ Branch 0 taken 7740 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 7740 times.
 7740 if (!h_count[i] || !v_count[i]) {
390 ✗ av_log(s->avctx, AV_LOG_ERROR,
391 "Invalid sampling factor in component %d %d:%d\n",
392 i, h_count[i], v_count[i]);
393 ✗ return AVERROR_INVALIDDATA;
394 }
395
396 7740 av_log(s->avctx, AV_LOG_DEBUG, "component %d %d:%d id: %d quant:%d\n",
397 i, h_count[i], v_count[i],
398 s->component_id[i], s->quant_index[i]);
399 }
400
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2592 times.
 2592 if ( nb_components == 4
401 ✗ && s->component_id[0] == 'C' - 1
402 ✗ && s->component_id[1] == 'M' - 1
403 ✗ && s->component_id[2] == 'Y' - 1
404 ✗ && s->component_id[3] == 'K' - 1)
405 ✗ s->adobe_transform = 0;
406
407
4/6
✓ Branch 0 taken 221 times.
✓ Branch 1 taken 2371 times.
✓ Branch 2 taken 221 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 221 times.
 2592 if (s->ls && (s->h_max > 1 || s->v_max > 1)) {
408 ✗ avpriv_report_missing_feature(s->avctx, "Subsampling in JPEG-LS");
409 ✗ return AVERROR_PATCHWELCOME;
410 }
411
412
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2592 times.
 2592 if (s->bayer) {
413 ✗ if (nb_components == 2) {
414 /* Bayer images embedded in DNGs can contain 2 interleaved components and the
415 width stored in their SOF3 markers is the width of each one. We only output
416 a single component, therefore we need to adjust the output image width. We
417 handle the deinterleaving (but not the debayering) in this file. */
418 ✗ width *= 2;
419 }
420 /* They can also contain 1 component, which is double the width and half the height
421 of the final image (rows are interleaved). We don't handle the decoding in this
422 file, but leave that to the TIFF/DNG decoder. */
423 }
424
425 /* if different size, realloc/alloc picture */
426
5/6
✓ Branch 0 taken 2236 times.
✓ Branch 1 taken 356 times.
✓ Branch 2 taken 2230 times.
✓ Branch 3 taken 6 times.
✓ Branch 4 taken 2230 times.
✗ Branch 5 not taken.
 2592 if (width != s->width || height != s->height || bits != s->bits ||
427
1/2
✓ Branch 0 taken 2230 times.
✗ Branch 1 not taken.
 2230 memcmp(s->h_count, h_count, sizeof(h_count)) ||
428
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2230 times.
 2230 memcmp(s->v_count, v_count, sizeof(v_count))) {
429 362 size_change = 1;
430
431 362 s->width = width;
432 362 s->height = height;
433 362 s->bits = bits;
434 362 memcpy(s->h_count, h_count, sizeof(h_count));
435 362 memcpy(s->v_count, v_count, sizeof(v_count));
436 362 s->interlaced = 0;
437 362 s->got_picture = 0;
438
439 /* test interlaced mode */
440
2/2
✓ Branch 0 taken 159 times.
✓ Branch 1 taken 203 times.
 362 if (s->first_picture &&
441
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 159 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 159 (s->multiscope != 2 || s->avctx->time_base.den >= 25 * s->avctx->time_base.num) &&
442
2/2
✓ Branch 0 taken 123 times.
✓ Branch 1 taken 36 times.
 159 s->orig_height != 0 &&
443
2/2
✓ Branch 0 taken 4 times.
✓ Branch 1 taken 119 times.
 123 s->height < ((s->orig_height * 3) / 4)) {
444 4 s->interlaced = 1;
445 4 s->bottom_field = s->interlace_polarity;
446 4 s->picture_ptr->interlaced_frame = 1;
447 4 s->picture_ptr->top_field_first = !s->interlace_polarity;
448 4 height *= 2;
449 }
450
451 362 ret = ff_set_dimensions(s->avctx, width, height);
452
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 362 times.
 362 if (ret < 0)
453 ✗ return ret;
454
455
1/2
✓ Branch 0 taken 362 times.
✗ Branch 1 not taken.
 362 if ((s->avctx->codec_tag == MKTAG('A', 'V', 'R', 'n') ||
456
2/2
✓ Branch 0 taken 2 times.
✓ Branch 1 taken 360 times.
 362 s->avctx->codec_tag == MKTAG('A', 'V', 'D', 'J')) &&
457
1/2
✓ Branch 0 taken 2 times.
✗ Branch 1 not taken.
 2 s->orig_height < height)
458 2 s->avctx->height = AV_CEIL_RSHIFT(s->orig_height, s->avctx->lowres);
459
460 362 s->first_picture = 0;
461 } else {
462 2230 size_change = 0;
463 }
464
465
2/2
✓ Branch 0 taken 3 times.
✓ Branch 1 taken 2589 times.
 2592 if (s->avctx->codec_id == AV_CODEC_ID_SMVJPEG) {
466 3 s->avctx->height = s->avctx->coded_height / s->smv_frames_per_jpeg;
467
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 3 times.
 3 if (s->avctx->height <= 0)
468 ✗ return AVERROR_INVALIDDATA;
469 }
470
471
5/6
✓ Branch 0 taken 23 times.
✓ Branch 1 taken 2569 times.
✓ Branch 2 taken 15 times.
✓ Branch 3 taken 8 times.
✓ Branch 4 taken 15 times.
✗ Branch 5 not taken.
 2592 if (s->got_picture && s->interlaced && (s->bottom_field == !s->interlace_polarity)) {
472
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 15 times.
 15 if (s->progressive) {
473 ✗ avpriv_request_sample(s->avctx, "progressively coded interlaced picture");
474 ✗ return AVERROR_INVALIDDATA;
475 }
476 } else {
477
9/10
✓ Branch 0 taken 259 times.
✓ Branch 1 taken 2318 times.
✓ Branch 2 taken 231 times.
✓ Branch 3 taken 28 times.
✓ Branch 4 taken 221 times.
✓ Branch 5 taken 10 times.
✓ Branch 6 taken 16 times.
✓ Branch 7 taken 205 times.
✗ Branch 8 not taken.
✓ Branch 9 taken 16 times.
 2577 if (s->v_max == 1 && s->h_max == 1 && s->lossless==1 && (nb_components==3 || nb_components==4))
478 205 s->rgb = 1;
479
2/2
✓ Branch 0 taken 2151 times.
✓ Branch 1 taken 221 times.
 2372 else if (!s->lossless)
480 2151 s->rgb = 0;
481 /* XXX: not complete test ! */
482 2577 pix_fmt_id = ((unsigned)s->h_count[0] << 28) | (s->v_count[0] << 24) |
483 2577 (s->h_count[1] << 20) | (s->v_count[1] << 16) |
484 2577 (s->h_count[2] << 12) | (s->v_count[2] << 8) |
485 2577 (s->h_count[3] << 4) | s->v_count[3];
486 2577 av_log(s->avctx, AV_LOG_DEBUG, "pix fmt id %x\n", pix_fmt_id);
487 /* NOTE we do not allocate pictures large enough for the possible
488 * padding of h/v_count being 4 */
489
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2577 times.
 2577 if (!(pix_fmt_id & 0xD0D0D0D0))
490 ✗ pix_fmt_id -= (pix_fmt_id & 0xF0F0F0F0) >> 1;
491
2/2
✓ Branch 0 taken 412 times.
✓ Branch 1 taken 2165 times.
 2577 if (!(pix_fmt_id & 0x0D0D0D0D))
492 412 pix_fmt_id -= (pix_fmt_id & 0x0F0F0F0F) >> 1;
493
494
2/2
✓ Branch 0 taken 20616 times.
✓ Branch 1 taken 2577 times.
 23193 for (i = 0; i < 8; i++) {
495 20616 int j = 6 + (i&1) - (i&6);
496 20616 int is = (pix_fmt_id >> (4*i)) & 0xF;
497 20616 int js = (pix_fmt_id >> (4*j)) & 0xF;
498
499
6/8
✓ Branch 0 taken 11346 times.
✓ Branch 1 taken 9270 times.
✓ Branch 2 taken 11346 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 11346 times.
✗ Branch 5 not taken.
✓ Branch 6 taken 1110 times.
✓ Branch 7 taken 10236 times.
 20616 if (is == 1 && js != 2 && (i < 2 || i > 5))
500 1110 js = (pix_fmt_id >> ( 8 + 4*(i&1))) & 0xF;
501
6/8
✓ Branch 0 taken 11346 times.
✓ Branch 1 taken 9270 times.
✓ Branch 2 taken 11346 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 11346 times.
✗ Branch 5 not taken.
✓ Branch 6 taken 1110 times.
✓ Branch 7 taken 10236 times.
 20616 if (is == 1 && js != 2 && (i < 2 || i > 5))
502 1110 js = (pix_fmt_id >> (16 + 4*(i&1))) & 0xF;
503
504
3/4
✓ Branch 0 taken 11346 times.
✓ Branch 1 taken 9270 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 11346 times.
 20616 if (is == 1 && js == 2) {
505 ✗ if (i & 1) s->upscale_h[j/2] = 1;
506 ✗ else s->upscale_v[j/2] = 1;
507 }
508 }
509
510
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2577 times.
 2577 if (s->bayer) {
511 ✗ if (pix_fmt_id != 0x11110000 && pix_fmt_id != 0x11000000)
512 ✗ goto unk_pixfmt;
513 }
514
515
4/13
✗ Branch 0 not taken.
✓ Branch 1 taken 421 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 18 times.
✗ Branch 6 not taken.
✓ Branch 7 taken 232 times.
✗ Branch 8 not taken.
✗ Branch 9 not taken.
✓ Branch 10 taken 1906 times.
✗ Branch 11 not taken.
✗ Branch 12 not taken.
 2577 switch (pix_fmt_id) {
516 ✗ case 0x11110000: /* for bayer-encoded huffman lossless JPEGs embedded in DNGs */
517 ✗ if (!s->bayer)
518 ✗ goto unk_pixfmt;
519 ✗ s->avctx->pix_fmt = AV_PIX_FMT_GRAY16LE;
520 ✗ break;
521 421 case 0x11111100:
522
2/2
✓ Branch 0 taken 205 times.
✓ Branch 1 taken 216 times.
 421 if (s->rgb)
523
1/2
✓ Branch 0 taken 205 times.
✗ Branch 1 not taken.
 205 s->avctx->pix_fmt = s->bits <= 9 ? AV_PIX_FMT_BGR24 : AV_PIX_FMT_BGR48;
524 else {
525
1/2
✓ Branch 0 taken 216 times.
✗ Branch 1 not taken.
 216 if ( s->adobe_transform == 0
526
1/6
✗ Branch 0 not taken.
✓ Branch 1 taken 216 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
 216 || s->component_id[0] == 'R' - 1 && s->component_id[1] == 'G' - 1 && s->component_id[2] == 'B' - 1) {
527 ✗ s->avctx->pix_fmt = s->bits <= 8 ? AV_PIX_FMT_GBRP : AV_PIX_FMT_GBRP16;
528 } else {
529
2/4
✓ Branch 0 taken 216 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 216 times.
 216 if (s->bits <= 8) s->avctx->pix_fmt = s->cs_itu601 ? AV_PIX_FMT_YUV444P : AV_PIX_FMT_YUVJ444P;
530 ✗ else s->avctx->pix_fmt = AV_PIX_FMT_YUV444P16;
531
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 216 times.
 216 s->avctx->color_range = s->cs_itu601 ? AVCOL_RANGE_MPEG : AVCOL_RANGE_JPEG;
532 }
533 }
534
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 421 times.
 421 av_assert0(s->nb_components == 3);
535 421 break;
536 ✗ case 0x11111111:
537 ✗ if (s->rgb)
538 ✗ s->avctx->pix_fmt = s->bits <= 9 ? AV_PIX_FMT_ABGR : AV_PIX_FMT_RGBA64;
539 else {
540 ✗ if (s->adobe_transform == 0 && s->bits <= 8) {
541 ✗ s->avctx->pix_fmt = AV_PIX_FMT_GBRAP;
542 } else {
543 ✗ s->avctx->pix_fmt = s->bits <= 8 ? AV_PIX_FMT_YUVA444P : AV_PIX_FMT_YUVA444P16;
544 ✗ s->avctx->color_range = s->cs_itu601 ? AVCOL_RANGE_MPEG : AVCOL_RANGE_JPEG;
545 }
546 }
547 ✗ av_assert0(s->nb_components == 4);
548 ✗ break;
549 ✗ case 0x22111122:
550 case 0x22111111:
551 ✗ if (s->adobe_transform == 0 && s->bits <= 8) {
552 ✗ s->avctx->pix_fmt = AV_PIX_FMT_GBRAP;
553 ✗ s->upscale_v[1] = s->upscale_v[2] = 1;
554 ✗ s->upscale_h[1] = s->upscale_h[2] = 1;
555 ✗ } else if (s->adobe_transform == 2 && s->bits <= 8) {
556 ✗ s->avctx->pix_fmt = AV_PIX_FMT_YUVA444P;
557 ✗ s->upscale_v[1] = s->upscale_v[2] = 1;
558 ✗ s->upscale_h[1] = s->upscale_h[2] = 1;
559 ✗ s->avctx->color_range = s->cs_itu601 ? AVCOL_RANGE_MPEG : AVCOL_RANGE_JPEG;
560 } else {
561 ✗ if (s->bits <= 8) s->avctx->pix_fmt = AV_PIX_FMT_YUVA420P;
562 ✗ else s->avctx->pix_fmt = AV_PIX_FMT_YUVA420P16;
563 ✗ s->avctx->color_range = s->cs_itu601 ? AVCOL_RANGE_MPEG : AVCOL_RANGE_JPEG;
564 }
565 ✗ av_assert0(s->nb_components == 4);
566 ✗ break;
567 ✗ case 0x12121100:
568 case 0x22122100:
569 case 0x21211100:
570 case 0x21112100:
571 case 0x22211200:
572 case 0x22221100:
573 case 0x22112200:
574 case 0x11222200:
575 ✗ if (s->bits <= 8) s->avctx->pix_fmt = s->cs_itu601 ? AV_PIX_FMT_YUV444P : AV_PIX_FMT_YUVJ444P;
576 else
577 ✗ goto unk_pixfmt;
578 ✗ s->avctx->color_range = s->cs_itu601 ? AVCOL_RANGE_MPEG : AVCOL_RANGE_JPEG;
579 ✗ break;
580 18 case 0x11000000:
581 case 0x13000000:
582 case 0x14000000:
583 case 0x31000000:
584 case 0x33000000:
585 case 0x34000000:
586 case 0x41000000:
587 case 0x43000000:
588 case 0x44000000:
589
2/2
✓ Branch 0 taken 16 times.
✓ Branch 1 taken 2 times.
 18 if(s->bits <= 8)
590
2/2
✓ Branch 0 taken 8 times.
✓ Branch 1 taken 8 times.
 16 s->avctx->pix_fmt = s->force_pal8 ? AV_PIX_FMT_PAL8 : AV_PIX_FMT_GRAY8;
591 else
592 2 s->avctx->pix_fmt = AV_PIX_FMT_GRAY16;
593 18 break;
594 ✗ case 0x12111100:
595 case 0x14121200:
596 case 0x14111100:
597 case 0x22211100:
598 case 0x22112100:
599 ✗ if (s->component_id[0] == 'Q' && s->component_id[1] == 'F' && s->component_id[2] == 'A') {
600 ✗ if (s->bits <= 8) s->avctx->pix_fmt = AV_PIX_FMT_GBRP;
601 else
602 ✗ goto unk_pixfmt;
603 ✗ s->upscale_v[0] = s->upscale_v[1] = 1;
604 } else {
605 ✗ if (pix_fmt_id == 0x14111100)
606 ✗ s->upscale_v[1] = s->upscale_v[2] = 1;
607 ✗ if (s->bits <= 8) s->avctx->pix_fmt = s->cs_itu601 ? AV_PIX_FMT_YUV440P : AV_PIX_FMT_YUVJ440P;
608 else
609 ✗ goto unk_pixfmt;
610 ✗ s->avctx->color_range = s->cs_itu601 ? AVCOL_RANGE_MPEG : AVCOL_RANGE_JPEG;
611 }
612 ✗ break;
613 232 case 0x21111100:
614
1/6
✗ Branch 0 not taken.
✓ Branch 1 taken 232 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
 232 if (s->component_id[0] == 'Q' && s->component_id[1] == 'F' && s->component_id[2] == 'A') {
615 ✗ if (s->bits <= 8) s->avctx->pix_fmt = AV_PIX_FMT_GBRP;
616 else
617 ✗ goto unk_pixfmt;
618 ✗ s->upscale_h[0] = s->upscale_h[1] = 1;
619 } else {
620
2/4
✓ Branch 0 taken 232 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 232 times.
 232 if (s->bits <= 8) s->avctx->pix_fmt = s->cs_itu601 ? AV_PIX_FMT_YUV422P : AV_PIX_FMT_YUVJ422P;
621 ✗ else s->avctx->pix_fmt = AV_PIX_FMT_YUV422P16;
622
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 232 times.
 232 s->avctx->color_range = s->cs_itu601 ? AVCOL_RANGE_MPEG : AVCOL_RANGE_JPEG;
623 }
624 232 break;
625 ✗ case 0x31111100:
626 ✗ if (s->bits > 8)
627 ✗ goto unk_pixfmt;
628 ✗ s->avctx->pix_fmt = s->cs_itu601 ? AV_PIX_FMT_YUV444P : AV_PIX_FMT_YUVJ444P;
629 ✗ s->avctx->color_range = s->cs_itu601 ? AVCOL_RANGE_MPEG : AVCOL_RANGE_JPEG;
630 ✗ s->upscale_h[1] = s->upscale_h[2] = 2;
631 ✗ break;
632 ✗ case 0x22121100:
633 case 0x22111200:
634 ✗ if (s->bits <= 8) s->avctx->pix_fmt = s->cs_itu601 ? AV_PIX_FMT_YUV422P : AV_PIX_FMT_YUVJ422P;
635 else
636 ✗ goto unk_pixfmt;
637 ✗ s->avctx->color_range = s->cs_itu601 ? AVCOL_RANGE_MPEG : AVCOL_RANGE_JPEG;
638 ✗ break;
639 1906 case 0x22111100:
640 case 0x23111100:
641 case 0x42111100:
642 case 0x24111100:
643
3/4
✓ Branch 0 taken 1906 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 205 times.
✓ Branch 3 taken 1701 times.
 1906 if (s->bits <= 8) s->avctx->pix_fmt = s->cs_itu601 ? AV_PIX_FMT_YUV420P : AV_PIX_FMT_YUVJ420P;
644 ✗ else s->avctx->pix_fmt = AV_PIX_FMT_YUV420P16;
645
2/2
✓ Branch 0 taken 205 times.
✓ Branch 1 taken 1701 times.
 1906 s->avctx->color_range = s->cs_itu601 ? AVCOL_RANGE_MPEG : AVCOL_RANGE_JPEG;
646
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1906 times.
 1906 if (pix_fmt_id == 0x42111100) {
647 ✗ if (s->bits > 8)
648 ✗ goto unk_pixfmt;
649 ✗ s->upscale_h[1] = s->upscale_h[2] = 1;
650
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1906 times.
 1906 } else if (pix_fmt_id == 0x24111100) {
651 ✗ if (s->bits > 8)
652 ✗ goto unk_pixfmt;
653 ✗ s->upscale_v[1] = s->upscale_v[2] = 1;
654
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1906 times.
 1906 } else if (pix_fmt_id == 0x23111100) {
655 ✗ if (s->bits > 8)
656 ✗ goto unk_pixfmt;
657 ✗ s->upscale_v[1] = s->upscale_v[2] = 2;
658 }
659 1906 break;
660 ✗ case 0x41111100:
661 ✗ if (s->bits <= 8) s->avctx->pix_fmt = s->cs_itu601 ? AV_PIX_FMT_YUV411P : AV_PIX_FMT_YUVJ411P;
662 else
663 ✗ goto unk_pixfmt;
664 ✗ s->avctx->color_range = s->cs_itu601 ? AVCOL_RANGE_MPEG : AVCOL_RANGE_JPEG;
665 ✗ break;
666 default:
667 ✗ unk_pixfmt:
668 ✗ avpriv_report_missing_feature(s->avctx, "Pixel format 0x%x bits:%d", pix_fmt_id, s->bits);
669 ✗ memset(s->upscale_h, 0, sizeof(s->upscale_h));
670 ✗ memset(s->upscale_v, 0, sizeof(s->upscale_v));
671 ✗ return AVERROR_PATCHWELCOME;
672 }
673
2/6
✓ Branch 0 taken 2577 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 2577 times.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
 2577 if ((AV_RB32(s->upscale_h) || AV_RB32(s->upscale_v)) && s->avctx->lowres) {
674 ✗ avpriv_report_missing_feature(s->avctx, "Lowres for weird subsampling");
675 ✗ return AVERROR_PATCHWELCOME;
676 }
677
2/8
✓ Branch 0 taken 2577 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 2577 times.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
✗ Branch 6 not taken.
✗ Branch 7 not taken.
 2577 if ((AV_RB32(s->upscale_h) || AV_RB32(s->upscale_v)) && s->progressive && s->avctx->pix_fmt == AV_PIX_FMT_GBRP) {
678 ✗ avpriv_report_missing_feature(s->avctx, "progressive for weird subsampling");
679 ✗ return AVERROR_PATCHWELCOME;
680 }
681
2/2
✓ Branch 0 taken 221 times.
✓ Branch 1 taken 2356 times.
 2577 if (s->ls) {
682 221 memset(s->upscale_h, 0, sizeof(s->upscale_h));
683 221 memset(s->upscale_v, 0, sizeof(s->upscale_v));
684
2/2
✓ Branch 0 taken 205 times.
✓ Branch 1 taken 16 times.
 221 if (s->nb_components == 3) {
685 205 s->avctx->pix_fmt = AV_PIX_FMT_RGB24;
686
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 16 times.
 16 } else if (s->nb_components != 1) {
687 ✗ av_log(s->avctx, AV_LOG_ERROR, "Unsupported number of components %d\n", s->nb_components);
688 ✗ return AVERROR_PATCHWELCOME;
689
4/6
✓ Branch 0 taken 16 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 8 times.
✓ Branch 3 taken 8 times.
✓ Branch 4 taken 8 times.
✗ Branch 5 not taken.
 16 } else if ((s->palette_index || s->force_pal8) && s->bits <= 8)
690 8 s->avctx->pix_fmt = AV_PIX_FMT_PAL8;
691
1/2
✓ Branch 0 taken 8 times.
✗ Branch 1 not taken.
 8 else if (s->bits <= 8)
692 8 s->avctx->pix_fmt = AV_PIX_FMT_GRAY8;
693 else
694 ✗ s->avctx->pix_fmt = AV_PIX_FMT_GRAY16;
695 }
696
697 2577 s->pix_desc = av_pix_fmt_desc_get(s->avctx->pix_fmt);
698
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2577 times.
 2577 if (!s->pix_desc) {
699 ✗ av_log(s->avctx, AV_LOG_ERROR, "Could not get a pixel format descriptor.\n");
700 ✗ return AVERROR_BUG;
701 }
702
703
4/4
✓ Branch 0 taken 2410 times.
✓ Branch 1 taken 167 times.
✓ Branch 2 taken 2207 times.
✓ Branch 3 taken 203 times.
 2577 if (s->avctx->pix_fmt == s->hwaccel_sw_pix_fmt && !size_change) {
704 2207 s->avctx->pix_fmt = s->hwaccel_pix_fmt;
705 } else {
706 370 enum AVPixelFormat pix_fmts[] = {
707 #if CONFIG_MJPEG_NVDEC_HWACCEL
708 AV_PIX_FMT_CUDA,
709 #endif
710 #if CONFIG_MJPEG_VAAPI_HWACCEL
711 AV_PIX_FMT_VAAPI,
712 #endif
713 370 s->avctx->pix_fmt,
714 AV_PIX_FMT_NONE,
715 };
716 370 s->hwaccel_pix_fmt = ff_get_format(s->avctx, pix_fmts);
717
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 370 times.
 370 if (s->hwaccel_pix_fmt < 0)
718 ✗ return AVERROR(EINVAL);
719
720 370 s->hwaccel_sw_pix_fmt = s->avctx->pix_fmt;
721 370 s->avctx->pix_fmt = s->hwaccel_pix_fmt;
722 }
723
724
2/2
✓ Branch 0 taken 75 times.
✓ Branch 1 taken 2502 times.
 2577 if (s->avctx->skip_frame == AVDISCARD_ALL) {
725 75 s->picture_ptr->pict_type = AV_PICTURE_TYPE_I;
726 75 s->picture_ptr->key_frame = 1;
727 75 s->got_picture = 1;
728 75 return 0;
729 }
730
731 2502 av_frame_unref(s->picture_ptr);
732
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 2502 times.
 2502 if (ff_get_buffer(s->avctx, s->picture_ptr, AV_GET_BUFFER_FLAG_REF) < 0)
733 ✗ return -1;
734 2502 s->picture_ptr->pict_type = AV_PICTURE_TYPE_I;
735 2502 s->picture_ptr->key_frame = 1;
736 2502 s->got_picture = 1;
737
738 // Lets clear the palette to avoid leaving uninitialized values in it
739
2/2
✓ Branch 0 taken 8 times.
✓ Branch 1 taken 2494 times.
 2502 if (s->avctx->pix_fmt == AV_PIX_FMT_PAL8)
740 8 memset(s->picture_ptr->data[1], 0, 1024);
741
742
2/2
✓ Branch 0 taken 10008 times.
✓ Branch 1 taken 2502 times.
 12510 for (i = 0; i < 4; i++)
743 10008 s->linesize[i] = s->picture_ptr->linesize[i] << s->interlaced;
744
745 ff_dlog(s->avctx, "%d %d %d %d %d %d\n",
746 s->width, s->height, s->linesize[0], s->linesize[1],
747 s->interlaced, s->avctx->height);
748
749 }
750
751
3/6
✓ Branch 0 taken 205 times.
✓ Branch 1 taken 2312 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 205 times.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
 2517 if ((s->rgb && !s->lossless && !s->ls) ||
752
5/6
✓ Branch 0 taken 2312 times.
✓ Branch 1 taken 205 times.
✓ Branch 2 taken 16 times.
✓ Branch 3 taken 2296 times.
✓ Branch 4 taken 16 times.
✗ Branch 5 not taken.
 2517 (!s->rgb && s->ls && s->nb_components > 1) ||
753
3/4
✓ Branch 0 taken 8 times.
✓ Branch 1 taken 2509 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 8 times.
 2517 (s->avctx->pix_fmt == AV_PIX_FMT_PAL8 && !s->ls)
754 ) {
755 ✗ av_log(s->avctx, AV_LOG_ERROR, "Unsupported coding and pixel format combination\n");
756 ✗ return AVERROR_PATCHWELCOME;
757 }
758
759 /* totally blank picture as progressive JPEG will only add details to it */
760
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2517 times.
 2517 if (s->progressive) {
761 ✗ int bw = (width + s->h_max * 8 - 1) / (s->h_max * 8);
762 ✗ int bh = (height + s->v_max * 8 - 1) / (s->v_max * 8);
763 ✗ for (i = 0; i < s->nb_components; i++) {
764 ✗ int size = bw * bh * s->h_count[i] * s->v_count[i];
765 ✗ av_freep(&s->blocks[i]);
766 ✗ av_freep(&s->last_nnz[i]);
767 ✗ s->blocks[i] = av_calloc(size, sizeof(**s->blocks));
768 ✗ s->last_nnz[i] = av_calloc(size, sizeof(**s->last_nnz));
769 ✗ if (!s->blocks[i] || !s->last_nnz[i])
770 ✗ return AVERROR(ENOMEM);
771 ✗ s->block_stride[i] = bw * s->h_count[i];
772 }
773 ✗ memset(s->coefs_finished, 0, sizeof(s->coefs_finished));
774 }
775
776
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2517 times.
 2517 if (s->avctx->hwaccel) {
777 ✗ s->hwaccel_picture_private =
778 ✗ av_mallocz(s->avctx->hwaccel->frame_priv_data_size);
779 ✗ if (!s->hwaccel_picture_private)
780 ✗ return AVERROR(ENOMEM);
781
782 ✗ ret = s->avctx->hwaccel->start_frame(s->avctx, s->raw_image_buffer,
783 ✗ s->raw_image_buffer_size);
784 ✗ if (ret < 0)
785 ✗ return ret;
786 }
787
788 2517 return 0;
789 }
790
791 26798972 static inline int mjpeg_decode_dc(MJpegDecodeContext *s, int dc_index)
792 {
793 int code;
794 26798972 code = get_vlc2(&s->gb, s->vlcs[0][dc_index].table, 9, 2);
795
2/4
✓ Branch 0 taken 26798972 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 26798972 times.
 26798972 if (code < 0 || code > 16) {
796 ✗ av_log(s->avctx, AV_LOG_WARNING,
797 "mjpeg_decode_dc: bad vlc: %d:%d (%p)\n",
798 0, dc_index, &s->vlcs[0][dc_index]);
799 ✗ return 0xfffff;
800 }
801
802
2/2
✓ Branch 0 taken 23760047 times.
✓ Branch 1 taken 3038925 times.
 26798972 if (code)
803 23760047 return get_xbits(&s->gb, code);
804 else
805 3038925 return 0;
806 }
807
808 /* decode block and dequantize */
809 3902672 static int decode_block(MJpegDecodeContext *s, int16_t *block, int component,
810 int dc_index, int ac_index, uint16_t *quant_matrix)
811 {
812 int code, i, j, level, val;
813
814 /* DC coef */
815 3902672 val = mjpeg_decode_dc(s, dc_index);
816
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 3902672 times.
 3902672 if (val == 0xfffff) {
817 ✗ av_log(s->avctx, AV_LOG_ERROR, "error dc\n");
818 ✗ return AVERROR_INVALIDDATA;
819 }
820 3902672 val = val * (unsigned)quant_matrix[0] + s->last_dc[component];
821 3902672 val = av_clip_int16(val);
822 3902672 s->last_dc[component] = val;
823 3902672 block[0] = val;
824 /* AC coefs */
825 3902672 i = 0;
826 3902672 {OPEN_READER(re, &s->gb);
827 do {
828 36156496 UPDATE_CACHE(re, &s->gb);
829
2/2
✓ Branch 1 taken 364026 times.
✓ Branch 2 taken 35792470 times.
 36156496 GET_VLC(code, re, &s->gb, s->vlcs[1][ac_index].table, 9, 2);
830
831 36156496 i += ((unsigned)code) >> 4;
832 36156496 code &= 0xf;
833
2/2
✓ Branch 0 taken 32304580 times.
✓ Branch 1 taken 3851916 times.
 36156496 if (code) {
834
2/2
✓ Branch 0 taken 485 times.
✓ Branch 1 taken 32304095 times.
 32304580 if (code > MIN_CACHE_BITS - 16)
835 485 UPDATE_CACHE(re, &s->gb);
836
837 {
838 32304580 int cache = GET_CACHE(re, &s->gb);
839 32304580 int sign = (~cache) >> 31;
840 32304580 level = (NEG_USR32(sign ^ cache,code) ^ sign) - sign;
841 }
842
843 32304580 LAST_SKIP_BITS(re, &s->gb, code);
844
845
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 32304580 times.
 32304580 if (i > 63) {
846 ✗ av_log(s->avctx, AV_LOG_ERROR, "error count: %d\n", i);
847 ✗ return AVERROR_INVALIDDATA;
848 }
849 32304580 j = s->scantable.permutated[i];
850 32304580 block[j] = level * quant_matrix[i];
851 }
852
2/2
✓ Branch 0 taken 32253824 times.
✓ Branch 1 taken 3902672 times.
 36156496 } while (i < 63);
853 3902672 CLOSE_READER(re, &s->gb);}
854
855 3902672 return 0;
856 }
857
858 ✗ static int decode_dc_progressive(MJpegDecodeContext *s, int16_t *block,
859 int component, int dc_index,
860 uint16_t *quant_matrix, int Al)
861 {
862 unsigned val;
863 ✗ s->bdsp.clear_block(block);
864 ✗ val = mjpeg_decode_dc(s, dc_index);
865 ✗ if (val == 0xfffff) {
866 ✗ av_log(s->avctx, AV_LOG_ERROR, "error dc\n");
867 ✗ return AVERROR_INVALIDDATA;
868 }
869 ✗ val = (val * (quant_matrix[0] << Al)) + s->last_dc[component];
870 ✗ s->last_dc[component] = val;
871 ✗ block[0] = val;
872 ✗ return 0;
873 }
874
875 /* decode block and dequantize - progressive JPEG version */
876 ✗ static int decode_block_progressive(MJpegDecodeContext *s, int16_t *block,
877 uint8_t *last_nnz, int ac_index,
878 uint16_t *quant_matrix,
879 int ss, int se, int Al, int *EOBRUN)
880 {
881 int code, i, j, val, run;
882 unsigned level;
883
884 ✗ if (*EOBRUN) {
885 ✗ (*EOBRUN)--;
886 ✗ return 0;
887 }
888
889 {
890 ✗ OPEN_READER(re, &s->gb);
891 ✗ for (i = ss; ; i++) {
892 ✗ UPDATE_CACHE(re, &s->gb);
893 ✗ GET_VLC(code, re, &s->gb, s->vlcs[2][ac_index].table, 9, 2);
894
895 ✗ run = ((unsigned) code) >> 4;
896 ✗ code &= 0xF;
897 ✗ if (code) {
898 ✗ i += run;
899 ✗ if (code > MIN_CACHE_BITS - 16)
900 ✗ UPDATE_CACHE(re, &s->gb);
901
902 {
903 ✗ int cache = GET_CACHE(re, &s->gb);
904 ✗ int sign = (~cache) >> 31;
905 ✗ level = (NEG_USR32(sign ^ cache,code) ^ sign) - sign;
906 }
907
908 ✗ LAST_SKIP_BITS(re, &s->gb, code);
909
910 ✗ if (i >= se) {
911 ✗ if (i == se) {
912 ✗ j = s->scantable.permutated[se];
913 ✗ block[j] = level * (quant_matrix[se] << Al);
914 ✗ break;
915 }
916 ✗ av_log(s->avctx, AV_LOG_ERROR, "error count: %d\n", i);
917 ✗ return AVERROR_INVALIDDATA;
918 }
919 ✗ j = s->scantable.permutated[i];
920 ✗ block[j] = level * (quant_matrix[i] << Al);
921 } else {
922 ✗ if (run == 0xF) {// ZRL - skip 15 coefficients
923 ✗ i += 15;
924 ✗ if (i >= se) {
925 ✗ av_log(s->avctx, AV_LOG_ERROR, "ZRL overflow: %d\n", i);
926 ✗ return AVERROR_INVALIDDATA;
927 }
928 } else {
929 ✗ val = (1 << run);
930 ✗ if (run) {
931 ✗ UPDATE_CACHE(re, &s->gb);
932 ✗ val += NEG_USR32(GET_CACHE(re, &s->gb), run);
933 ✗ LAST_SKIP_BITS(re, &s->gb, run);
934 }
935 ✗ *EOBRUN = val - 1;
936 ✗ break;
937 }
938 }
939 }
940 ✗ CLOSE_READER(re, &s->gb);
941 }
942
943 ✗ if (i > *last_nnz)
944 ✗ *last_nnz = i;
945
946 ✗ return 0;
947 }
948
949 #define REFINE_BIT(j) { \
950 UPDATE_CACHE(re, &s->gb); \
951 sign = block[j] >> 15; \
952 block[j] += SHOW_UBITS(re, &s->gb, 1) * \
953 ((quant_matrix[i] ^ sign) - sign) << Al; \
954 LAST_SKIP_BITS(re, &s->gb, 1); \
955 }
956
957 #define ZERO_RUN \
958 for (; ; i++) { \
959 if (i > last) { \
960 i += run; \
961 if (i > se) { \
962 av_log(s->avctx, AV_LOG_ERROR, "error count: %d\n", i); \
963 return -1; \
964 } \
965 break; \
966 } \
967 j = s->scantable.permutated[i]; \
968 if (block[j]) \
969 REFINE_BIT(j) \
970 else if (run-- == 0) \
971 break; \
972 }
973
974 /* decode block and dequantize - progressive JPEG refinement pass */
975 ✗ static int decode_block_refinement(MJpegDecodeContext *s, int16_t *block,
976 uint8_t *last_nnz,
977 int ac_index, uint16_t *quant_matrix,
978 int ss, int se, int Al, int *EOBRUN)
979 {
980 ✗ int code, i = ss, j, sign, val, run;
981 ✗ int last = FFMIN(se, *last_nnz);
982
983 ✗ OPEN_READER(re, &s->gb);
984 ✗ if (*EOBRUN) {
985 ✗ (*EOBRUN)--;
986 } else {
987 ✗ for (; ; i++) {
988 ✗ UPDATE_CACHE(re, &s->gb);
989 ✗ GET_VLC(code, re, &s->gb, s->vlcs[2][ac_index].table, 9, 2);
990
991 ✗ if (code & 0xF) {
992 ✗ run = ((unsigned) code) >> 4;
993 ✗ UPDATE_CACHE(re, &s->gb);
994 ✗ val = SHOW_UBITS(re, &s->gb, 1);
995 ✗ LAST_SKIP_BITS(re, &s->gb, 1);
996 ✗ ZERO_RUN;
997 ✗ j = s->scantable.permutated[i];
998 ✗ val--;
999 ✗ block[j] = ((quant_matrix[i] << Al) ^ val) - val;
1000 ✗ if (i == se) {
1001 ✗ if (i > *last_nnz)
1002 ✗ *last_nnz = i;
1003 ✗ CLOSE_READER(re, &s->gb);
1004 ✗ return 0;
1005 }
1006 } else {
1007 ✗ run = ((unsigned) code) >> 4;
1008 ✗ if (run == 0xF) {
1009 ✗ ZERO_RUN;
1010 } else {
1011 ✗ val = run;
1012 ✗ run = (1 << run);
1013 ✗ if (val) {
1014 ✗ UPDATE_CACHE(re, &s->gb);
1015 ✗ run += SHOW_UBITS(re, &s->gb, val);
1016 ✗ LAST_SKIP_BITS(re, &s->gb, val);
1017 }
1018 ✗ *EOBRUN = run - 1;
1019 ✗ break;
1020 }
1021 }
1022 }
1023
1024 ✗ if (i > *last_nnz)
1025 ✗ *last_nnz = i;
1026 }
1027
1028 ✗ for (; i <= last; i++) {
1029 ✗ j = s->scantable.permutated[i];
1030 ✗ if (block[j])
1031 ✗ REFINE_BIT(j)
1032 }
1033 ✗ CLOSE_READER(re, &s->gb);
1034
1035 ✗ return 0;
1036 }
1037 #undef REFINE_BIT
1038 #undef ZERO_RUN
1039
1040 775496 static int handle_rstn(MJpegDecodeContext *s, int nb_components)
1041 {
1042 int i;
1043 775496 int reset = 0;
1044
1045
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 775496 times.
 775496 if (s->restart_interval) {
1046 ✗ s->restart_count--;
1047 ✗ if(s->restart_count == 0 && s->avctx->codec_id == AV_CODEC_ID_THP){
1048 ✗ align_get_bits(&s->gb);
1049 ✗ for (i = 0; i < nb_components; i++) /* reset dc */
1050 ✗ s->last_dc[i] = (4 << s->bits);
1051 }
1052
1053 ✗ i = 8 + ((-get_bits_count(&s->gb)) & 7);
1054 /* skip RSTn */
1055 ✗ if (s->restart_count == 0) {
1056 ✗ if( show_bits(&s->gb, i) == (1 << i) - 1
1057 ✗ || show_bits(&s->gb, i) == 0xFF) {
1058 ✗ int pos = get_bits_count(&s->gb);
1059 ✗ align_get_bits(&s->gb);
1060 ✗ while (get_bits_left(&s->gb) >= 8 && show_bits(&s->gb, 8) == 0xFF)
1061 ✗ skip_bits(&s->gb, 8);
1062 ✗ if (get_bits_left(&s->gb) >= 8 && (get_bits(&s->gb, 8) & 0xF8) == 0xD0) {
1063 ✗ for (i = 0; i < nb_components; i++) /* reset dc */
1064 ✗ s->last_dc[i] = (4 << s->bits);
1065 ✗ reset = 1;
1066 } else
1067 ✗ skip_bits_long(&s->gb, pos - get_bits_count(&s->gb));
1068 }
1069 }
1070 }
1071 775496 return reset;
1072 }
1073
1074 /* Handles 1 to 4 components */
1075 ✗ static int ljpeg_decode_rgb_scan(MJpegDecodeContext *s, int nb_components, int predictor, int point_transform)
1076 {
1077 int i, mb_x, mb_y;
1078 unsigned width;
1079 uint16_t (*buffer)[4];
1080 int left[4], top[4], topleft[4];
1081 ✗ const int linesize = s->linesize[0];
1082 ✗ const int mask = ((1 << s->bits) - 1) << point_transform;
1083 ✗ int resync_mb_y = 0;
1084 ✗ int resync_mb_x = 0;
1085 int vpred[6];
1086
1087 ✗ if (!s->bayer && s->nb_components < 3)
1088 ✗ return AVERROR_INVALIDDATA;
1089 ✗ if (s->bayer && s->nb_components > 2)
1090 ✗ return AVERROR_INVALIDDATA;
1091 ✗ if (s->nb_components <= 0 || s->nb_components > 4)
1092 ✗ return AVERROR_INVALIDDATA;
1093 ✗ if (s->v_max != 1 || s->h_max != 1 || !s->lossless)
1094 ✗ return AVERROR_INVALIDDATA;
1095
1096
1097 ✗ s->restart_count = s->restart_interval;
1098
1099 ✗ if (s->restart_interval == 0)
1100 ✗ s->restart_interval = INT_MAX;
1101
1102 ✗ if (s->bayer)
1103 ✗ width = s->mb_width / nb_components; /* Interleaved, width stored is the total so need to divide */
1104 else
1105 ✗ width = s->mb_width;
1106
1107 ✗ av_fast_malloc(&s->ljpeg_buffer, &s->ljpeg_buffer_size, width * 4 * sizeof(s->ljpeg_buffer[0][0]));
1108 ✗ if (!s->ljpeg_buffer)
1109 ✗ return AVERROR(ENOMEM);
1110
1111 ✗ buffer = s->ljpeg_buffer;
1112
1113 ✗ for (i = 0; i < 4; i++)
1114 ✗ buffer[0][i] = 1 << (s->bits - 1);
1115
1116 ✗ for (mb_y = 0; mb_y < s->mb_height; mb_y++) {
1117 ✗ uint8_t *ptr = s->picture_ptr->data[0] + (linesize * mb_y);
1118
1119 ✗ if (s->interlaced && s->bottom_field)
1120 ✗ ptr += linesize >> 1;
1121
1122 ✗ for (i = 0; i < 4; i++)
1123 ✗ top[i] = left[i] = topleft[i] = buffer[0][i];
1124
1125 ✗ if ((mb_y * s->width) % s->restart_interval == 0) {
1126 ✗ for (i = 0; i < 6; i++)
1127 ✗ vpred[i] = 1 << (s->bits-1);
1128 }
1129
1130 ✗ for (mb_x = 0; mb_x < width; mb_x++) {
1131 ✗ int modified_predictor = predictor;
1132
1133 ✗ if (get_bits_left(&s->gb) < 1) {
1134 ✗ av_log(s->avctx, AV_LOG_ERROR, "bitstream end in rgb_scan\n");
1135 ✗ return AVERROR_INVALIDDATA;
1136 }
1137
1138 ✗ if (s->restart_interval && !s->restart_count){
1139 ✗ s->restart_count = s->restart_interval;
1140 ✗ resync_mb_x = mb_x;
1141 ✗ resync_mb_y = mb_y;
1142 ✗ for(i=0; i<4; i++)
1143 ✗ top[i] = left[i]= topleft[i]= 1 << (s->bits - 1);
1144 }
1145 ✗ if (mb_y == resync_mb_y || mb_y == resync_mb_y+1 && mb_x < resync_mb_x || !mb_x)
1146 ✗ modified_predictor = 1;
1147
1148 ✗ for (i=0;i<nb_components;i++) {
1149 int pred, dc;
1150
1151 ✗ topleft[i] = top[i];
1152 ✗ top[i] = buffer[mb_x][i];
1153
1154 ✗ dc = mjpeg_decode_dc(s, s->dc_index[i]);
1155 ✗ if(dc == 0xFFFFF)
1156 ✗ return -1;
1157
1158 ✗ if (!s->bayer || mb_x) {
1159 ✗ pred = left[i];
1160 } else { /* This path runs only for the first line in bayer images */
1161 ✗ vpred[i] += dc;
1162 ✗ pred = vpred[i] - dc;
1163 }
1164
1165 ✗ PREDICT(pred, topleft[i], top[i], pred, modified_predictor);
1166
1167 ✗ left[i] = buffer[mb_x][i] =
1168 ✗ mask & (pred + (unsigned)(dc * (1 << point_transform)));
1169 }
1170
1171 ✗ if (s->restart_interval && !--s->restart_count) {
1172 ✗ align_get_bits(&s->gb);
1173 ✗ skip_bits(&s->gb, 16); /* skip RSTn */
1174 }
1175 }
1176 ✗ if (s->rct && s->nb_components == 4) {
1177 ✗ for (mb_x = 0; mb_x < s->mb_width; mb_x++) {
1178 ✗ ptr[4*mb_x + 2] = buffer[mb_x][0] - ((buffer[mb_x][1] + buffer[mb_x][2] - 0x200) >> 2);
1179 ✗ ptr[4*mb_x + 1] = buffer[mb_x][1] + ptr[4*mb_x + 2];
1180 ✗ ptr[4*mb_x + 3] = buffer[mb_x][2] + ptr[4*mb_x + 2];
1181 ✗ ptr[4*mb_x + 0] = buffer[mb_x][3];
1182 }
1183 ✗ } else if (s->nb_components == 4) {
1184 ✗ for(i=0; i<nb_components; i++) {
1185 ✗ int c= s->comp_index[i];
1186 ✗ if (s->bits <= 8) {
1187 ✗ for(mb_x = 0; mb_x < s->mb_width; mb_x++) {
1188 ✗ ptr[4*mb_x+3-c] = buffer[mb_x][i];
1189 }
1190 ✗ } else if(s->bits == 9) {
1191 ✗ return AVERROR_PATCHWELCOME;
1192 } else {
1193 ✗ for(mb_x = 0; mb_x < s->mb_width; mb_x++) {
1194 ✗ ((uint16_t*)ptr)[4*mb_x+c] = buffer[mb_x][i];
1195 }
1196 }
1197 }
1198 ✗ } else if (s->rct) {
1199 ✗ for (mb_x = 0; mb_x < s->mb_width; mb_x++) {
1200 ✗ ptr[3*mb_x + 1] = buffer[mb_x][0] - ((buffer[mb_x][1] + buffer[mb_x][2] - 0x200) >> 2);
1201 ✗ ptr[3*mb_x + 0] = buffer[mb_x][1] + ptr[3*mb_x + 1];
1202 ✗ ptr[3*mb_x + 2] = buffer[mb_x][2] + ptr[3*mb_x + 1];
1203 }
1204 ✗ } else if (s->pegasus_rct) {
1205 ✗ for (mb_x = 0; mb_x < s->mb_width; mb_x++) {
1206 ✗ ptr[3*mb_x + 1] = buffer[mb_x][0] - ((buffer[mb_x][1] + buffer[mb_x][2]) >> 2);
1207 ✗ ptr[3*mb_x + 0] = buffer[mb_x][1] + ptr[3*mb_x + 1];
1208 ✗ ptr[3*mb_x + 2] = buffer[mb_x][2] + ptr[3*mb_x + 1];
1209 }
1210 ✗ } else if (s->bayer) {
1211 ✗ if (nb_components == 1) {
1212 /* Leave decoding to the TIFF/DNG decoder (see comment in ff_mjpeg_decode_sof) */
1213 ✗ for (mb_x = 0; mb_x < width; mb_x++)
1214 ✗ ((uint16_t*)ptr)[mb_x] = buffer[mb_x][0];
1215 ✗ } else if (nb_components == 2) {
1216 ✗ for (mb_x = 0; mb_x < width; mb_x++) {
1217 ✗ ((uint16_t*)ptr)[2*mb_x + 0] = buffer[mb_x][0];
1218 ✗ ((uint16_t*)ptr)[2*mb_x + 1] = buffer[mb_x][1];
1219 }
1220 }
1221 } else {
1222 ✗ for(i=0; i<nb_components; i++) {
1223 ✗ int c= s->comp_index[i];
1224 ✗ if (s->bits <= 8) {
1225 ✗ for(mb_x = 0; mb_x < s->mb_width; mb_x++) {
1226 ✗ ptr[3*mb_x+2-c] = buffer[mb_x][i];
1227 }
1228 ✗ } else if(s->bits == 9) {
1229 ✗ return AVERROR_PATCHWELCOME;
1230 } else {
1231 ✗ for(mb_x = 0; mb_x < s->mb_width; mb_x++) {
1232 ✗ ((uint16_t*)ptr)[3*mb_x+2-c] = buffer[mb_x][i];
1233 }
1234 }
1235 }
1236 }
1237 }
1238 ✗ return 0;
1239 }
1240
1241 200 static int ljpeg_decode_yuv_scan(MJpegDecodeContext *s, int predictor,
1242 int point_transform, int nb_components)
1243 {
1244 int i, mb_x, mb_y, mask;
1245 200 int bits= (s->bits+7)&~7;
1246 200 int resync_mb_y = 0;
1247 200 int resync_mb_x = 0;
1248
1249 200 point_transform += bits - s->bits;
1250 200 mask = ((1 << s->bits) - 1) << point_transform;
1251
1252
2/4
✓ Branch 0 taken 200 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 200 times.
 200 av_assert0(nb_components>=1 && nb_components<=4);
1253
1254
2/2
✓ Branch 0 taken 22450 times.
✓ Branch 1 taken 200 times.
 22650 for (mb_y = 0; mb_y < s->mb_height; mb_y++) {
1255
2/2
✓ Branch 0 taken 3816050 times.
✓ Branch 1 taken 22450 times.
 3838500 for (mb_x = 0; mb_x < s->mb_width; mb_x++) {
1256
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 3816050 times.
 3816050 if (get_bits_left(&s->gb) < 1) {
1257 ✗ av_log(s->avctx, AV_LOG_ERROR, "bitstream end in yuv_scan\n");
1258 ✗ return AVERROR_INVALIDDATA;
1259 }
1260
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 3816050 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 3816050 if (s->restart_interval && !s->restart_count){
1261 ✗ s->restart_count = s->restart_interval;
1262 ✗ resync_mb_x = mb_x;
1263 ✗ resync_mb_y = mb_y;
1264 }
1265
1266
8/10
✓ Branch 0 taken 3793600 times.
✓ Branch 1 taken 22450 times.
✓ Branch 2 taken 3766550 times.
✓ Branch 3 taken 27050 times.
✓ Branch 4 taken 27050 times.
✓ Branch 5 taken 3739500 times.
✓ Branch 6 taken 27050 times.
✗ Branch 7 not taken.
✗ Branch 8 not taken.
✓ Branch 9 taken 3766550 times.
 3865550 if(!mb_x || mb_y == resync_mb_y || mb_y == resync_mb_y+1 && mb_x < resync_mb_x || s->interlaced){
1267
5/6
✓ Branch 0 taken 22250 times.
✓ Branch 1 taken 27250 times.
✓ Branch 2 taken 200 times.
✓ Branch 3 taken 22050 times.
✗ Branch 4 not taken.
✓ Branch 5 taken 200 times.
 49500 int toprow = mb_y == resync_mb_y || mb_y == resync_mb_y+1 && mb_x < resync_mb_x;
1268
4/6
✓ Branch 0 taken 27050 times.
✓ Branch 1 taken 22450 times.
✓ Branch 2 taken 27050 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 27050 times.
 49500 int leftcol = !mb_x || mb_y == resync_mb_y && mb_x == resync_mb_x;
1269
2/2
✓ Branch 0 taken 148500 times.
✓ Branch 1 taken 49500 times.
 198000 for (i = 0; i < nb_components; i++) {
1270 uint8_t *ptr;
1271 uint16_t *ptr16;
1272 int n, h, v, x, y, c, j, linesize;
1273 148500 n = s->nb_blocks[i];
1274 148500 c = s->comp_index[i];
1275 148500 h = s->h_scount[i];
1276 148500 v = s->v_scount[i];
1277 148500 x = 0;
1278 148500 y = 0;
1279 148500 linesize= s->linesize[c];
1280
1281
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 148500 times.
 148500 if(bits>8) linesize /= 2;
1282
1283
2/2
✓ Branch 0 taken 297000 times.
✓ Branch 1 taken 148500 times.
 445500 for(j=0; j<n; j++) {
1284 int pred, dc;
1285
1286 297000 dc = mjpeg_decode_dc(s, s->dc_index[i]);
1287
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 297000 times.
 297000 if(dc == 0xFFFFF)
1288 ✗ return -1;
1289
1/2
✓ Branch 0 taken 297000 times.
✗ Branch 1 not taken.
 297000 if ( h * mb_x + x >= s->width
1290
1/2
✓ Branch 0 taken 297000 times.
✗ Branch 1 not taken.
 297000 || v * mb_y + y >= s->height) {
1291 // Nothing to do
1292
1/2
✓ Branch 0 taken 297000 times.
✗ Branch 1 not taken.
 297000 } else if (bits<=8) {
1293 297000 ptr = s->picture_ptr->data[c] + (linesize * (v * mb_y + y)) + (h * mb_x + x); //FIXME optimize this crap
1294
4/4
✓ Branch 0 taken 198000 times.
✓ Branch 1 taken 99000 times.
✓ Branch 2 taken 109000 times.
✓ Branch 3 taken 89000 times.
 297000 if(y==0 && toprow){
1295
4/4
✓ Branch 0 taken 81750 times.
✓ Branch 1 taken 27250 times.
✓ Branch 2 taken 600 times.
✓ Branch 3 taken 81150 times.
 109000 if(x==0 && leftcol){
1296 600 pred= 1 << (bits - 1);
1297 }else{
1298 108400 pred= ptr[-1];
1299 }
1300 }else{
1301
4/4
✓ Branch 0 taken 116250 times.
✓ Branch 1 taken 71750 times.
✓ Branch 2 taken 89200 times.
✓ Branch 3 taken 27050 times.
 188000 if(x==0 && leftcol){
1302 89200 pred= ptr[-linesize];
1303 }else{
1304
1/8
✗ Branch 0 not taken.
✓ Branch 1 taken 98800 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
✗ Branch 6 not taken.
✗ Branch 7 not taken.
 98800 PREDICT(pred, ptr[-linesize-1], ptr[-linesize], ptr[-1], predictor);
1305 }
1306 }
1307
1308
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 297000 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 297000 if (s->interlaced && s->bottom_field)
1309 ✗ ptr += linesize >> 1;
1310 297000 pred &= mask;
1311 297000 *ptr= pred + ((unsigned)dc << point_transform);
1312 }else{
1313 ✗ ptr16 = (uint16_t*)(s->picture_ptr->data[c] + 2*(linesize * (v * mb_y + y)) + 2*(h * mb_x + x)); //FIXME optimize this crap
1314 ✗ if(y==0 && toprow){
1315 ✗ if(x==0 && leftcol){
1316 ✗ pred= 1 << (bits - 1);
1317 }else{
1318 ✗ pred= ptr16[-1];
1319 }
1320 }else{
1321 ✗ if(x==0 && leftcol){
1322 ✗ pred= ptr16[-linesize];
1323 }else{
1324 ✗ PREDICT(pred, ptr16[-linesize-1], ptr16[-linesize], ptr16[-1], predictor);
1325 }
1326 }
1327
1328 ✗ if (s->interlaced && s->bottom_field)
1329 ✗ ptr16 += linesize >> 1;
1330 ✗ pred &= mask;
1331 ✗ *ptr16= pred + ((unsigned)dc << point_transform);
1332 }
1333
2/2
✓ Branch 0 taken 198000 times.
✓ Branch 1 taken 99000 times.
 297000 if (++x == h) {
1334 198000 x = 0;
1335 198000 y++;
1336 }
1337 }
1338 }
1339 } else {
1340
2/2
✓ Branch 0 taken 11299650 times.
✓ Branch 1 taken 3766550 times.
 15066200 for (i = 0; i < nb_components; i++) {
1341 uint8_t *ptr;
1342 uint16_t *ptr16;
1343 int n, h, v, x, y, c, j, linesize, dc;
1344 11299650 n = s->nb_blocks[i];
1345 11299650 c = s->comp_index[i];
1346 11299650 h = s->h_scount[i];
1347 11299650 v = s->v_scount[i];
1348 11299650 x = 0;
1349 11299650 y = 0;
1350 11299650 linesize = s->linesize[c];
1351
1352
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 11299650 times.
 11299650 if(bits>8) linesize /= 2;
1353
1354
2/2
✓ Branch 0 taken 22599300 times.
✓ Branch 1 taken 11299650 times.
 33898950 for (j = 0; j < n; j++) {
1355 int pred;
1356
1357 22599300 dc = mjpeg_decode_dc(s, s->dc_index[i]);
1358
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 22599300 times.
 22599300 if(dc == 0xFFFFF)
1359 ✗ return -1;
1360
1/2
✓ Branch 0 taken 22599300 times.
✗ Branch 1 not taken.
 22599300 if ( h * mb_x + x >= s->width
1361
1/2
✓ Branch 0 taken 22599300 times.
✗ Branch 1 not taken.
 22599300 || v * mb_y + y >= s->height) {
1362 // Nothing to do
1363
1/2
✓ Branch 0 taken 22599300 times.
✗ Branch 1 not taken.
 22599300 } else if (bits<=8) {
1364 22599300 ptr = s->picture_ptr->data[c] +
1365 22599300 (linesize * (v * mb_y + y)) +
1366 22599300 (h * mb_x + x); //FIXME optimize this crap
1367
1/8
✗ Branch 0 not taken.
✓ Branch 1 taken 22599300 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
✗ Branch 6 not taken.
✗ Branch 7 not taken.
 22599300 PREDICT(pred, ptr[-linesize-1], ptr[-linesize], ptr[-1], predictor);
1368
1369 22599300 pred &= mask;
1370 22599300 *ptr = pred + ((unsigned)dc << point_transform);
1371 }else{
1372 ✗ ptr16 = (uint16_t*)(s->picture_ptr->data[c] + 2*(linesize * (v * mb_y + y)) + 2*(h * mb_x + x)); //FIXME optimize this crap
1373 ✗ PREDICT(pred, ptr16[-linesize-1], ptr16[-linesize], ptr16[-1], predictor);
1374
1375 ✗ pred &= mask;
1376 ✗ *ptr16= pred + ((unsigned)dc << point_transform);
1377 }
1378
1379
2/2
✓ Branch 0 taken 15066200 times.
✓ Branch 1 taken 7533100 times.
 22599300 if (++x == h) {
1380 15066200 x = 0;
1381 15066200 y++;
1382 }
1383 }
1384 }
1385 }
1386
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 3816050 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 3816050 if (s->restart_interval && !--s->restart_count) {
1387 ✗ align_get_bits(&s->gb);
1388 ✗ skip_bits(&s->gb, 16); /* skip RSTn */
1389 }
1390 }
1391 }
1392 200 return 0;
1393 }
1394
1395 786684 static av_always_inline void mjpeg_copy_block(MJpegDecodeContext *s,
1396 uint8_t *dst, const uint8_t *src,
1397 int linesize, int lowres)
1398 {
1399
1/5
✓ Branch 0 taken 786684 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
 786684 switch (lowres) {
1400 786684 case 0: s->hdsp.put_pixels_tab[1][0](dst, src, linesize, 8);
1401 786684 break;
1402 ✗ case 1: copy_block4(dst, src, linesize, linesize, 4);
1403 ✗ break;
1404 ✗ case 2: copy_block2(dst, src, linesize, linesize, 2);
1405 ✗ break;
1406 ✗ case 3: *dst = *src;
1407 ✗ break;
1408 }
1409 786684 }
1410
1411 11750 static void shift_output(MJpegDecodeContext *s, uint8_t *ptr, int linesize)
1412 {
1413 int block_x, block_y;
1414 11750 int size = 8 >> s->avctx->lowres;
1415
1/2
✓ Branch 0 taken 11750 times.
✗ Branch 1 not taken.
 11750 if (s->bits > 8) {
1416
2/2
✓ Branch 0 taken 94000 times.
✓ Branch 1 taken 11750 times.
 105750 for (block_y=0; block_y<size; block_y++)
1417
2/2
✓ Branch 0 taken 752000 times.
✓ Branch 1 taken 94000 times.
 846000 for (block_x=0; block_x<size; block_x++)
1418 752000 *(uint16_t*)(ptr + 2*block_x + block_y*linesize) <<= 16 - s->bits;
1419 } else {
1420 ✗ for (block_y=0; block_y<size; block_y++)
1421 ✗ for (block_x=0; block_x<size; block_x++)
1422 ✗ *(ptr + block_x + block_y*linesize) <<= 8 - s->bits;
1423 }
1424 11750 }
1425
1426 2124 static int mjpeg_decode_scan(MJpegDecodeContext *s, int nb_components, int Ah,
1427 int Al, const uint8_t *mb_bitmask,
1428 int mb_bitmask_size,
1429 const AVFrame *reference)
1430 {
1431 int i, mb_x, mb_y, chroma_h_shift, chroma_v_shift, chroma_width, chroma_height;
1432 uint8_t *data[MAX_COMPONENTS];
1433 const uint8_t *reference_data[MAX_COMPONENTS];
1434 int linesize[MAX_COMPONENTS];
1435 2124 GetBitContext mb_bitmask_gb = {0}; // initialize to silence gcc warning
1436
2/2
✓ Branch 0 taken 1 times.
✓ Branch 1 taken 2123 times.
 2124 int bytes_per_pixel = 1 + (s->bits > 8);
1437
1438
2/2
✓ Branch 0 taken 31 times.
✓ Branch 1 taken 2093 times.
 2124 if (mb_bitmask) {
1439
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 31 times.
 31 if (mb_bitmask_size != (s->mb_width * s->mb_height + 7)>>3) {
1440 ✗ av_log(s->avctx, AV_LOG_ERROR, "mb_bitmask_size mismatches\n");
1441 ✗ return AVERROR_INVALIDDATA;
1442 }
1443 31 init_get_bits(&mb_bitmask_gb, mb_bitmask, s->mb_width * s->mb_height);
1444 }
1445
1446 2124 s->restart_count = 0;
1447
1448 2124 av_pix_fmt_get_chroma_sub_sample(s->avctx->pix_fmt, &chroma_h_shift,
1449 &chroma_v_shift);
1450 2124 chroma_width = AV_CEIL_RSHIFT(s->width, chroma_h_shift);
1451 2124 chroma_height = AV_CEIL_RSHIFT(s->height, chroma_v_shift);
1452
1453
2/2
✓ Branch 0 taken 6370 times.
✓ Branch 1 taken 2124 times.
 8494 for (i = 0; i < nb_components; i++) {
1454 6370 int c = s->comp_index[i];
1455 6370 data[c] = s->picture_ptr->data[c];
1456
2/2
✓ Branch 0 taken 93 times.
✓ Branch 1 taken 6277 times.
 6370 reference_data[c] = reference ? reference->data[c] : NULL;
1457 6370 linesize[c] = s->linesize[c];
1458 6370 s->coefs_finished[c] |= 1;
1459 }
1460
1461
2/2
✓ Branch 0 taken 28266 times.
✓ Branch 1 taken 2124 times.
 30390 for (mb_y = 0; mb_y < s->mb_height; mb_y++) {
1462
2/2
✓ Branch 0 taken 775496 times.
✓ Branch 1 taken 28266 times.
 803762 for (mb_x = 0; mb_x < s->mb_width; mb_x++) {
1463
4/4
✓ Branch 0 taken 148800 times.
✓ Branch 1 taken 626696 times.
✓ Branch 3 taken 131114 times.
✓ Branch 4 taken 17686 times.
 775496 const int copy_mb = mb_bitmask && !get_bits1(&mb_bitmask_gb);
1464
1465
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 775496 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 775496 if (s->restart_interval && !s->restart_count)
1466 ✗ s->restart_count = s->restart_interval;
1467
1468
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 775496 times.
 775496 if (get_bits_left(&s->gb) < 0) {
1469 ✗ av_log(s->avctx, AV_LOG_ERROR, "overread %d\n",
1470 ✗ -get_bits_left(&s->gb));
1471 ✗ return AVERROR_INVALIDDATA;
1472 }
1473
2/2
✓ Branch 0 taken 2302988 times.
✓ Branch 1 taken 775496 times.
 3078484 for (i = 0; i < nb_components; i++) {
1474 uint8_t *ptr;
1475 int n, h, v, x, y, c, j;
1476 int block_offset;
1477 2302988 n = s->nb_blocks[i];
1478 2302988 c = s->comp_index[i];
1479 2302988 h = s->h_scount[i];
1480 2302988 v = s->v_scount[i];
1481 2302988 x = 0;
1482 2302988 y = 0;
1483
2/2
✓ Branch 0 taken 4689356 times.
✓ Branch 1 taken 2302988 times.
 6992344 for (j = 0; j < n; j++) {
1484 4689356 block_offset = (((linesize[c] * (v * mb_y + y) * 8) +
1485 4689356 (h * mb_x + x) * 8 * bytes_per_pixel) >> s->avctx->lowres);
1486
1487
4/4
✓ Branch 0 taken 21440 times.
✓ Branch 1 taken 4667916 times.
✓ Branch 2 taken 10720 times.
✓ Branch 3 taken 10720 times.
 4689356 if (s->interlaced && s->bottom_field)
1488 10720 block_offset += linesize[c] >> 1;
1489
6/6
✓ Branch 0 taken 3745954 times.
✓ Branch 1 taken 943402 times.
✓ Branch 2 taken 2802552 times.
✓ Branch 3 taken 943402 times.
✓ Branch 4 taken 4687172 times.
✓ Branch 5 taken 2184 times.
 4689356 if ( 8*(h * mb_x + x) < ((c == 1) || (c == 2) ? chroma_width : s->width)
1490
6/6
✓ Branch 0 taken 3743770 times.
✓ Branch 1 taken 943402 times.
✓ Branch 2 taken 2800368 times.
✓ Branch 3 taken 943402 times.
✓ Branch 4 taken 4681083 times.
✓ Branch 5 taken 6089 times.
 4687172 && 8*(v * mb_y + y) < ((c == 1) || (c == 2) ? chroma_height : s->height)) {
1491 4681083 ptr = data[c] + block_offset;
1492 } else
1493 8273 ptr = NULL;
1494
1/2
✓ Branch 0 taken 4689356 times.
✗ Branch 1 not taken.
 4689356 if (!s->progressive) {
1495
2/2
✓ Branch 0 taken 786684 times.
✓ Branch 1 taken 3902672 times.
 4689356 if (copy_mb) {
1496
1/2
✓ Branch 0 taken 786684 times.
✗ Branch 1 not taken.
 786684 if (ptr)
1497 786684 mjpeg_copy_block(s, ptr, reference_data[c] + block_offset,
1498 786684 linesize[c], s->avctx->lowres);
1499
1500 } else {
1501 3902672 s->bdsp.clear_block(s->block);
1502
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 3902672 times.
 3902672 if (decode_block(s, s->block, i,
1503 s->dc_index[i], s->ac_index[i],
1504 3902672 s->quant_matrixes[s->quant_sindex[i]]) < 0) {
1505 ✗ av_log(s->avctx, AV_LOG_ERROR,
1506 "error y=%d x=%d\n", mb_y, mb_x);
1507 ✗ return AVERROR_INVALIDDATA;
1508 }
1509
2/2
✓ Branch 0 taken 3894399 times.
✓ Branch 1 taken 8273 times.
 3902672 if (ptr) {
1510 3894399 s->idsp.idct_put(ptr, linesize[c], s->block);
1511
2/2
✓ Branch 0 taken 11750 times.
✓ Branch 1 taken 3882649 times.
 3894399 if (s->bits & 7)
1512 11750 shift_output(s, ptr, linesize[c]);
1513 }
1514 }
1515 } else {
1516 ✗ int block_idx = s->block_stride[c] * (v * mb_y + y) +
1517 ✗ (h * mb_x + x);
1518 ✗ int16_t *block = s->blocks[c][block_idx];
1519 ✗ if (Ah)
1520 ✗ block[0] += get_bits1(&s->gb) *
1521 ✗ s->quant_matrixes[s->quant_sindex[i]][0] << Al;
1522 ✗ else if (decode_dc_progressive(s, block, i, s->dc_index[i],
1523 ✗ s->quant_matrixes[s->quant_sindex[i]],
1524 Al) < 0) {
1525 ✗ av_log(s->avctx, AV_LOG_ERROR,
1526 "error y=%d x=%d\n", mb_y, mb_x);
1527 ✗ return AVERROR_INVALIDDATA;
1528 }
1529 }
1530 ff_dlog(s->avctx, "mb: %d %d processed\n", mb_y, mb_x);
1531 ff_dlog(s->avctx, "%d %d %d %d %d %d %d %d \n",
1532 mb_x, mb_y, x, y, c, s->bottom_field,
1533 (v * mb_y + y) * 8, (h * mb_x + x) * 8);
1534
2/2
✓ Branch 0 taken 3413761 times.
✓ Branch 1 taken 1275595 times.
 4689356 if (++x == h) {
1535 3413761 x = 0;
1536 3413761 y++;
1537 }
1538 }
1539 }
1540
1541 775496 handle_rstn(s, nb_components);
1542 }
1543 }
1544 2124 return 0;
1545 }
1546
1547 ✗ static int mjpeg_decode_scan_progressive_ac(MJpegDecodeContext *s, int ss,
1548 int se, int Ah, int Al)
1549 {
1550 int mb_x, mb_y;
1551 ✗ int EOBRUN = 0;
1552 ✗ int c = s->comp_index[0];
1553 ✗ uint16_t *quant_matrix = s->quant_matrixes[s->quant_sindex[0]];
1554
1555 ✗ av_assert0(ss>=0 && Ah>=0 && Al>=0);
1556 ✗ if (se < ss || se > 63) {
1557 ✗ av_log(s->avctx, AV_LOG_ERROR, "SS/SE %d/%d is invalid\n", ss, se);
1558 ✗ return AVERROR_INVALIDDATA;
1559 }
1560
1561 // s->coefs_finished is a bitmask for coefficients coded
1562 // ss and se are parameters telling start and end coefficients
1563 ✗ s->coefs_finished[c] |= (2ULL << se) - (1ULL << ss);
1564
1565 ✗ s->restart_count = 0;
1566
1567 ✗ for (mb_y = 0; mb_y < s->mb_height; mb_y++) {
1568 ✗ int block_idx = mb_y * s->block_stride[c];
1569 ✗ int16_t (*block)[64] = &s->blocks[c][block_idx];
1570 ✗ uint8_t *last_nnz = &s->last_nnz[c][block_idx];
1571 ✗ if (get_bits_left(&s->gb) <= 0) {
1572 ✗ av_log(s->avctx, AV_LOG_ERROR, "bitstream truncated in mjpeg_decode_scan_progressive_ac\n");
1573 ✗ return AVERROR_INVALIDDATA;
1574 }
1575 ✗ for (mb_x = 0; mb_x < s->mb_width; mb_x++, block++, last_nnz++) {
1576 int ret;
1577 ✗ if (s->restart_interval && !s->restart_count)
1578 ✗ s->restart_count = s->restart_interval;
1579
1580 ✗ if (Ah)
1581 ✗ ret = decode_block_refinement(s, *block, last_nnz, s->ac_index[0],
1582 quant_matrix, ss, se, Al, &EOBRUN);
1583 else
1584 ✗ ret = decode_block_progressive(s, *block, last_nnz, s->ac_index[0],
1585 quant_matrix, ss, se, Al, &EOBRUN);
1586
1587 ✗ if (ret >= 0 && get_bits_left(&s->gb) < 0)
1588 ✗ ret = AVERROR_INVALIDDATA;
1589 ✗ if (ret < 0) {
1590 ✗ av_log(s->avctx, AV_LOG_ERROR,
1591 "error y=%d x=%d\n", mb_y, mb_x);
1592 ✗ return AVERROR_INVALIDDATA;
1593 }
1594
1595 ✗ if (handle_rstn(s, 0))
1596 ✗ EOBRUN = 0;
1597 }
1598 }
1599 ✗ return 0;
1600 }
1601
1602 ✗ static void mjpeg_idct_scan_progressive_ac(MJpegDecodeContext *s)
1603 {
1604 int mb_x, mb_y;
1605 int c;
1606 ✗ const int bytes_per_pixel = 1 + (s->bits > 8);
1607 ✗ const int block_size = s->lossless ? 1 : 8;
1608
1609 ✗ for (c = 0; c < s->nb_components; c++) {
1610 ✗ uint8_t *data = s->picture_ptr->data[c];
1611 ✗ int linesize = s->linesize[c];
1612 ✗ int h = s->h_max / s->h_count[c];
1613 ✗ int v = s->v_max / s->v_count[c];
1614 ✗ int mb_width = (s->width + h * block_size - 1) / (h * block_size);
1615 ✗ int mb_height = (s->height + v * block_size - 1) / (v * block_size);
1616
1617 ✗ if (~s->coefs_finished[c])
1618 ✗ av_log(s->avctx, AV_LOG_WARNING, "component %d is incomplete\n", c);
1619
1620 ✗ if (s->interlaced && s->bottom_field)
1621 ✗ data += linesize >> 1;
1622
1623 ✗ for (mb_y = 0; mb_y < mb_height; mb_y++) {
1624 ✗ uint8_t *ptr = data + (mb_y * linesize * 8 >> s->avctx->lowres);
1625 ✗ int block_idx = mb_y * s->block_stride[c];
1626 ✗ int16_t (*block)[64] = &s->blocks[c][block_idx];
1627 ✗ for (mb_x = 0; mb_x < mb_width; mb_x++, block++) {
1628 ✗ s->idsp.idct_put(ptr, linesize, *block);
1629 ✗ if (s->bits & 7)
1630 ✗ shift_output(s, ptr, linesize);
1631 ✗ ptr += bytes_per_pixel*8 >> s->avctx->lowres;
1632 }
1633 }
1634 }
1635 }
1636
1637 2537 int ff_mjpeg_decode_sos(MJpegDecodeContext *s, const uint8_t *mb_bitmask,
1638 int mb_bitmask_size, const AVFrame *reference)
1639 {
1640 int len, nb_components, i, h, v, predictor, point_transform;
1641 int index, id, ret;
1642
2/2
✓ Branch 0 taken 413 times.
✓ Branch 1 taken 2124 times.
 2537 const int block_size = s->lossless ? 1 : 8;
1643 int ilv, prev_shift;
1644
1645
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2537 times.
 2537 if (!s->got_picture) {
1646 ✗ av_log(s->avctx, AV_LOG_WARNING,
1647 "Can not process SOS before SOF, skipping\n");
1648 ✗ return -1;
1649 }
1650
1651
2/2
✓ Branch 0 taken 31 times.
✓ Branch 1 taken 2506 times.
 2537 if (reference) {
1652
1/2
✓ Branch 0 taken 31 times.
✗ Branch 1 not taken.
 31 if (reference->width != s->picture_ptr->width ||
1653
1/2
✓ Branch 0 taken 31 times.
✗ Branch 1 not taken.
 31 reference->height != s->picture_ptr->height ||
1654
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 31 times.
 31 reference->format != s->picture_ptr->format) {
1655 ✗ av_log(s->avctx, AV_LOG_ERROR, "Reference mismatching\n");
1656 ✗ return AVERROR_INVALIDDATA;
1657 }
1658 }
1659
1660 /* XXX: verify len field validity */
1661 2537 len = get_bits(&s->gb, 16);
1662 2537 nb_components = get_bits(&s->gb, 8);
1663
2/4
✓ Branch 0 taken 2537 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 2537 times.
 2537 if (nb_components == 0 || nb_components > MAX_COMPONENTS) {
1664 ✗ avpriv_report_missing_feature(s->avctx,
1665 "decode_sos: nb_components (%d)",
1666 nb_components);
1667 ✗ return AVERROR_PATCHWELCOME;
1668 }
1669
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2537 times.
 2537 if (len != 6 + 2 * nb_components) {
1670 ✗ av_log(s->avctx, AV_LOG_ERROR, "decode_sos: invalid len (%d)\n", len);
1671 ✗ return AVERROR_INVALIDDATA;
1672 }
1673
2/2
✓ Branch 0 taken 7593 times.
✓ Branch 1 taken 2537 times.
 10130 for (i = 0; i < nb_components; i++) {
1674 7593 id = get_bits(&s->gb, 8) - 1;
1675 7593 av_log(s->avctx, AV_LOG_DEBUG, "component: %d\n", id);
1676 /* find component index */
1677
1/2
✓ Branch 0 taken 15177 times.
✗ Branch 1 not taken.
 15177 for (index = 0; index < s->nb_components; index++)
1678
2/2
✓ Branch 0 taken 7593 times.
✓ Branch 1 taken 7584 times.
 15177 if (id == s->component_id[index])
1679 7593 break;
1680
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 7593 times.
 7593 if (index == s->nb_components) {
1681 ✗ av_log(s->avctx, AV_LOG_ERROR,
1682 "decode_sos: index(%d) out of components\n", index);
1683 ✗ return AVERROR_INVALIDDATA;
1684 }
1685 /* Metasoft MJPEG codec has Cb and Cr swapped */
1686
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 7593 times.
 7593 if (s->avctx->codec_tag == MKTAG('M', 'T', 'S', 'J')
1687 ✗ && nb_components == 3 && s->nb_components == 3 && i)
1688 ✗ index = 3 - i;
1689
1690 7593 s->quant_sindex[i] = s->quant_index[index];
1691 7593 s->nb_blocks[i] = s->h_count[index] * s->v_count[index];
1692 7593 s->h_scount[i] = s->h_count[index];
1693 7593 s->v_scount[i] = s->v_count[index];
1694
1695
6/8
✓ Branch 0 taken 7584 times.
✓ Branch 1 taken 9 times.
✓ Branch 2 taken 7584 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 7584 times.
✓ Branch 5 taken 9 times.
✗ Branch 6 not taken.
✓ Branch 7 taken 7584 times.
 7593 if((nb_components == 1 || nb_components == 3) && s->nb_components == 3 && s->avctx->pix_fmt == AV_PIX_FMT_GBR24P)
1696 ✗ index = (index+2)%3;
1697
1698 7593 s->comp_index[i] = index;
1699
1700 7593 s->dc_index[i] = get_bits(&s->gb, 4);
1701 7593 s->ac_index[i] = get_bits(&s->gb, 4);
1702
1703
2/4
✓ Branch 0 taken 7593 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 7593 times.
✗ Branch 3 not taken.
 7593 if (s->dc_index[i] < 0 || s->ac_index[i] < 0 ||
1704
2/4
✓ Branch 0 taken 7593 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 7593 times.
 7593 s->dc_index[i] >= 4 || s->ac_index[i] >= 4)
1705 ✗ goto out_of_range;
1706
3/8
✓ Branch 0 taken 7593 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 7593 times.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
✗ Branch 6 not taken.
✓ Branch 7 taken 7593 times.
 7593 if (!s->vlcs[0][s->dc_index[i]].table || !(s->progressive ? s->vlcs[2][s->ac_index[0]].table : s->vlcs[1][s->ac_index[i]].table))
1707 ✗ goto out_of_range;
1708 }
1709
1710 2537 predictor = get_bits(&s->gb, 8); /* JPEG Ss / lossless JPEG predictor /JPEG-LS NEAR */
1711 2537 ilv = get_bits(&s->gb, 8); /* JPEG Se / JPEG-LS ILV */
1712
1/2
✓ Branch 0 taken 2537 times.
✗ Branch 1 not taken.
 2537 if(s->avctx->codec_tag != AV_RL32("CJPG")){
1713 2537 prev_shift = get_bits(&s->gb, 4); /* Ah */
1714 2537 point_transform = get_bits(&s->gb, 4); /* Al */
1715 }else
1716 ✗ prev_shift = point_transform = 0;
1717
1718
2/2
✓ Branch 0 taken 2528 times.
✓ Branch 1 taken 9 times.
 2537 if (nb_components > 1) {
1719 /* interleaved stream */
1720 2528 s->mb_width = (s->width + s->h_max * block_size - 1) / (s->h_max * block_size);
1721 2528 s->mb_height = (s->height + s->v_max * block_size - 1) / (s->v_max * block_size);
1722
2/2
✓ Branch 0 taken 1 times.
✓ Branch 1 taken 8 times.
 9 } else if (!s->ls) { /* skip this for JPEG-LS */
1723 1 h = s->h_max / s->h_scount[0];
1724 1 v = s->v_max / s->v_scount[0];
1725 1 s->mb_width = (s->width + h * block_size - 1) / (h * block_size);
1726 1 s->mb_height = (s->height + v * block_size - 1) / (v * block_size);
1727 1 s->nb_blocks[0] = 1;
1728 1 s->h_scount[0] = 1;
1729 1 s->v_scount[0] = 1;
1730 }
1731
1732
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2537 times.
 2537 if (s->avctx->debug & FF_DEBUG_PICT_INFO)
1733 ✗ av_log(s->avctx, AV_LOG_DEBUG, "%s %s p:%d >>:%d ilv:%d bits:%d skip:%d %s comp:%d\n",
1734 ✗ s->lossless ? "lossless" : "sequential DCT", s->rgb ? "RGB" : "",
1735 predictor, point_transform, ilv, s->bits, s->mjpb_skiptosod,
1736 ✗ s->pegasus_rct ? "PRCT" : (s->rct ? "RCT" : ""), nb_components);
1737
1738
1739 /* mjpeg-b can have padding bytes between sos and image data, skip them */
1740
2/2
✓ Branch 0 taken 22 times.
✓ Branch 1 taken 2537 times.
 2559 for (i = s->mjpb_skiptosod; i > 0; i--)
1741 22 skip_bits(&s->gb, 8);
1742
1743 2537 next_field:
1744
2/2
✓ Branch 0 taken 7593 times.
✓ Branch 1 taken 2537 times.
 10130 for (i = 0; i < nb_components; i++)
1745 7593 s->last_dc[i] = (4 << s->bits);
1746
1747
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2537 times.
 2537 if (s->avctx->hwaccel) {
1748 ✗ int bytes_to_start = get_bits_count(&s->gb) / 8;
1749 ✗ av_assert0(bytes_to_start >= 0 &&
1750 s->raw_scan_buffer_size >= bytes_to_start);
1751
1752 ✗ ret = s->avctx->hwaccel->decode_slice(s->avctx,
1753 ✗ s->raw_scan_buffer + bytes_to_start,
1754 ✗ s->raw_scan_buffer_size - bytes_to_start);
1755 ✗ if (ret < 0)
1756 ✗ return ret;
1757
1758
2/2
✓ Branch 0 taken 413 times.
✓ Branch 1 taken 2124 times.
 2537 } else if (s->lossless) {
1759
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 413 times.
 413 av_assert0(s->picture_ptr == s->picture);
1760
2/2
✓ Branch 0 taken 213 times.
✓ Branch 1 taken 200 times.
 413 if (CONFIG_JPEGLS_DECODER && s->ls) {
1761 // for () {
1762 // reset_ls_coding_parameters(s, 0);
1763
1764
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 213 times.
 213 if ((ret = ff_jpegls_decode_picture(s, predictor,
1765 point_transform, ilv)) < 0)
1766 ✗ return ret;
1767 } else {
1768
2/4
✓ Branch 0 taken 200 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 200 times.
 200 if (s->rgb || s->bayer) {
1769 ✗ if ((ret = ljpeg_decode_rgb_scan(s, nb_components, predictor, point_transform)) < 0)
1770 ✗ return ret;
1771 } else {
1772
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 200 times.
 200 if ((ret = ljpeg_decode_yuv_scan(s, predictor,
1773 point_transform,
1774 nb_components)) < 0)
1775 ✗ return ret;
1776 }
1777 }
1778 } else {
1779
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 2124 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 2124 if (s->progressive && predictor) {
1780 ✗ av_assert0(s->picture_ptr == s->picture);
1781 ✗ if ((ret = mjpeg_decode_scan_progressive_ac(s, predictor,
1782 ilv, prev_shift,
1783 point_transform)) < 0)
1784 ✗ return ret;
1785 } else {
1786
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 2124 times.
 2124 if ((ret = mjpeg_decode_scan(s, nb_components,
1787 prev_shift, point_transform,
1788 mb_bitmask, mb_bitmask_size, reference)) < 0)
1789 ✗ return ret;
1790 }
1791 }
1792
1793
4/4
✓ Branch 0 taken 28 times.
✓ Branch 1 taken 2509 times.
✓ Branch 2 taken 15 times.
✓ Branch 3 taken 13 times.
 2565 if (s->interlaced &&
1794
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 15 times.
 43 get_bits_left(&s->gb) > 32 &&
1795 15 show_bits(&s->gb, 8) == 0xFF) {
1796 ✗ GetBitContext bak = s->gb;
1797 ✗ align_get_bits(&bak);
1798 ✗ if (show_bits(&bak, 16) == 0xFFD1) {
1799 ✗ av_log(s->avctx, AV_LOG_DEBUG, "AVRn interlaced picture marker found\n");
1800 ✗ s->gb = bak;
1801 ✗ skip_bits(&s->gb, 16);
1802 ✗ s->bottom_field ^= 1;
1803
1804 ✗ goto next_field;
1805 }
1806 }
1807
1808 2537 emms_c();
1809 2537 return 0;
1810 ✗ out_of_range:
1811 ✗ av_log(s->avctx, AV_LOG_ERROR, "decode_sos: ac/dc index out of range\n");
1812 ✗ return AVERROR_INVALIDDATA;
1813 }
1814
1815 16 static int mjpeg_decode_dri(MJpegDecodeContext *s)
1816 {
1817
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 16 times.
 16 if (get_bits(&s->gb, 16) != 4)
1818 ✗ return AVERROR_INVALIDDATA;
1819 16 s->restart_interval = get_bits(&s->gb, 16);
1820 16 s->restart_count = 0;
1821 16 av_log(s->avctx, AV_LOG_DEBUG, "restart interval: %d\n",
1822 s->restart_interval);
1823
1824 16 return 0;
1825 }
1826
1827 192 static int mjpeg_decode_app(MJpegDecodeContext *s)
1828 {
1829 int len, id, i;
1830
1831 192 len = get_bits(&s->gb, 16);
1832
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 192 times.
 192 if (len < 6) {
1833 ✗ if (s->bayer) {
1834 // Pentax K-1 (digital camera) JPEG images embedded in DNG images contain unknown APP0 markers
1835 ✗ av_log(s->avctx, AV_LOG_WARNING, "skipping APPx (len=%"PRId32") for bayer-encoded image\n", len);
1836 ✗ skip_bits(&s->gb, len);
1837 ✗ return 0;
1838 } else
1839 ✗ return AVERROR_INVALIDDATA;
1840 }
1841
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 192 times.
 192 if (8 * len > get_bits_left(&s->gb))
1842 ✗ return AVERROR_INVALIDDATA;
1843
1844 192 id = get_bits_long(&s->gb, 32);
1845 192 len -= 6;
1846
1847
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 192 times.
 192 if (s->avctx->debug & FF_DEBUG_STARTCODE)
1848 ✗ av_log(s->avctx, AV_LOG_DEBUG, "APPx (%s / %8X) len=%d\n",
1849 ✗ av_fourcc2str(av_bswap32(id)), id, len);
1850
1851 /* Buggy AVID, it puts EOI only at every 10th frame. */
1852 /* Also, this fourcc is used by non-avid files too, it holds some
1853 information, but it's always present in AVID-created files. */
1854
2/2
✓ Branch 0 taken 19 times.
✓ Branch 1 taken 173 times.
 192 if (id == AV_RB32("AVI1")) {
1855 /* structure:
1856 4bytes AVI1
1857 1bytes polarity
1858 1bytes always zero
1859 4bytes field_size
1860 4bytes field_size_less_padding
1861 */
1862 19 s->buggy_avid = 1;
1863 19 i = get_bits(&s->gb, 8); len--;
1864 19 av_log(s->avctx, AV_LOG_DEBUG, "polarity %d\n", i);
1865 19 goto out;
1866 }
1867
1868
2/2
✓ Branch 0 taken 132 times.
✓ Branch 1 taken 41 times.
 173 if (id == AV_RB32("JFIF")) {
1869 int t_w, t_h, v1, v2;
1870
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 132 times.
 132 if (len < 8)
1871 ✗ goto out;
1872 132 skip_bits(&s->gb, 8); /* the trailing zero-byte */
1873 132 v1 = get_bits(&s->gb, 8);
1874 132 v2 = get_bits(&s->gb, 8);
1875 132 skip_bits(&s->gb, 8);
1876
1877 132 s->avctx->sample_aspect_ratio.num = get_bits(&s->gb, 16);
1878 132 s->avctx->sample_aspect_ratio.den = get_bits(&s->gb, 16);
1879
1/2
✓ Branch 0 taken 132 times.
✗ Branch 1 not taken.
 132 if ( s->avctx->sample_aspect_ratio.num <= 0
1880
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 132 times.
 132 || s->avctx->sample_aspect_ratio.den <= 0) {
1881 ✗ s->avctx->sample_aspect_ratio.num = 0;
1882 ✗ s->avctx->sample_aspect_ratio.den = 1;
1883 }
1884
1885
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 132 times.
 132 if (s->avctx->debug & FF_DEBUG_PICT_INFO)
1886 ✗ av_log(s->avctx, AV_LOG_INFO,
1887 "mjpeg: JFIF header found (version: %x.%x) SAR=%d/%d\n",
1888 v1, v2,
1889 ✗ s->avctx->sample_aspect_ratio.num,
1890 ✗ s->avctx->sample_aspect_ratio.den);
1891
1892 132 len -= 8;
1893
2/2
✓ Branch 0 taken 130 times.
✓ Branch 1 taken 2 times.
 132 if (len >= 2) {
1894 130 t_w = get_bits(&s->gb, 8);
1895 130 t_h = get_bits(&s->gb, 8);
1896
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 130 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 130 if (t_w && t_h) {
1897 /* skip thumbnail */
1898 ✗ if (len -10 - (t_w * t_h * 3) > 0)
1899 ✗ len -= t_w * t_h * 3;
1900 }
1901 130 len -= 2;
1902 }
1903 132 goto out;
1904 }
1905
1906
2/2
✓ Branch 0 taken 5 times.
✓ Branch 1 taken 36 times.
 41 if ( id == AV_RB32("Adob")
1907
1/2
✓ Branch 0 taken 5 times.
✗ Branch 1 not taken.
 5 && len >= 7
1908
1/2
✓ Branch 1 taken 5 times.
✗ Branch 2 not taken.
 5 && show_bits(&s->gb, 8) == 'e'
1909
1/2
✓ Branch 1 taken 5 times.
✗ Branch 2 not taken.
 5 && show_bits_long(&s->gb, 32) != AV_RB32("e_CM")) {
1910 5 skip_bits(&s->gb, 8); /* 'e' */
1911 5 skip_bits(&s->gb, 16); /* version */
1912 5 skip_bits(&s->gb, 16); /* flags0 */
1913 5 skip_bits(&s->gb, 16); /* flags1 */
1914 5 s->adobe_transform = get_bits(&s->gb, 8);
1915
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 5 times.
 5 if (s->avctx->debug & FF_DEBUG_PICT_INFO)
1916 ✗ av_log(s->avctx, AV_LOG_INFO, "mjpeg: Adobe header found, transform=%d\n", s->adobe_transform);
1917 5 len -= 7;
1918 5 goto out;
1919 }
1920
1921
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 36 times.
 36 if (id == AV_RB32("LJIF")) {
1922 ✗ int rgb = s->rgb;
1923 ✗ int pegasus_rct = s->pegasus_rct;
1924 ✗ if (s->avctx->debug & FF_DEBUG_PICT_INFO)
1925 ✗ av_log(s->avctx, AV_LOG_INFO,
1926 "Pegasus lossless jpeg header found\n");
1927 ✗ skip_bits(&s->gb, 16); /* version ? */
1928 ✗ skip_bits(&s->gb, 16); /* unknown always 0? */
1929 ✗ skip_bits(&s->gb, 16); /* unknown always 0? */
1930 ✗ skip_bits(&s->gb, 16); /* unknown always 0? */
1931 ✗ switch (i=get_bits(&s->gb, 8)) {
1932 ✗ case 1:
1933 ✗ rgb = 1;
1934 ✗ pegasus_rct = 0;
1935 ✗ break;
1936 ✗ case 2:
1937 ✗ rgb = 1;
1938 ✗ pegasus_rct = 1;
1939 ✗ break;
1940 ✗ default:
1941 ✗ av_log(s->avctx, AV_LOG_ERROR, "unknown colorspace %d\n", i);
1942 }
1943
1944 ✗ len -= 9;
1945 ✗ if (s->got_picture)
1946 ✗ if (rgb != s->rgb || pegasus_rct != s->pegasus_rct) {
1947 ✗ av_log(s->avctx, AV_LOG_WARNING, "Mismatching LJIF tag\n");
1948 ✗ goto out;
1949 }
1950
1951 ✗ s->rgb = rgb;
1952 ✗ s->pegasus_rct = pegasus_rct;
1953
1954 ✗ goto out;
1955 }
1956
3/4
✓ Branch 0 taken 8 times.
✓ Branch 1 taken 28 times.
✓ Branch 2 taken 8 times.
✗ Branch 3 not taken.
 36 if (id == AV_RL32("colr") && len > 0) {
1957 8 s->colr = get_bits(&s->gb, 8);
1958
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 8 times.
 8 if (s->avctx->debug & FF_DEBUG_PICT_INFO)
1959 ✗ av_log(s->avctx, AV_LOG_INFO, "COLR %d\n", s->colr);
1960 8 len --;
1961 8 goto out;
1962 }
1963
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 28 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 28 if (id == AV_RL32("xfrm") && len > 0) {
1964 ✗ s->xfrm = get_bits(&s->gb, 8);
1965 ✗ if (s->avctx->debug & FF_DEBUG_PICT_INFO)
1966 ✗ av_log(s->avctx, AV_LOG_INFO, "XFRM %d\n", s->xfrm);
1967 ✗ len --;
1968 ✗ goto out;
1969 }
1970
1971 /* JPS extension by VRex */
1972
1/6
✗ Branch 0 not taken.
✓ Branch 1 taken 28 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
 28 if (s->start_code == APP3 && id == AV_RB32("_JPS") && len >= 10) {
1973 int flags, layout, type;
1974 ✗ if (s->avctx->debug & FF_DEBUG_PICT_INFO)
1975 ✗ av_log(s->avctx, AV_LOG_INFO, "_JPSJPS_\n");
1976
1977 ✗ skip_bits(&s->gb, 32); len -= 4; /* JPS_ */
1978 ✗ skip_bits(&s->gb, 16); len -= 2; /* block length */
1979 ✗ skip_bits(&s->gb, 8); /* reserved */
1980 ✗ flags = get_bits(&s->gb, 8);
1981 ✗ layout = get_bits(&s->gb, 8);
1982 ✗ type = get_bits(&s->gb, 8);
1983 ✗ len -= 4;
1984
1985 ✗ av_freep(&s->stereo3d);
1986 ✗ s->stereo3d = av_stereo3d_alloc();
1987 ✗ if (!s->stereo3d) {
1988 ✗ goto out;
1989 }
1990 ✗ if (type == 0) {
1991 ✗ s->stereo3d->type = AV_STEREO3D_2D;
1992 ✗ } else if (type == 1) {
1993 ✗ switch (layout) {
1994 ✗ case 0x01:
1995 ✗ s->stereo3d->type = AV_STEREO3D_LINES;
1996 ✗ break;
1997 ✗ case 0x02:
1998 ✗ s->stereo3d->type = AV_STEREO3D_SIDEBYSIDE;
1999 ✗ break;
2000 ✗ case 0x03:
2001 ✗ s->stereo3d->type = AV_STEREO3D_TOPBOTTOM;
2002 ✗ break;
2003 }
2004 ✗ if (!(flags & 0x04)) {
2005 ✗ s->stereo3d->flags = AV_STEREO3D_FLAG_INVERT;
2006 }
2007 }
2008 ✗ goto out;
2009 }
2010
2011 /* EXIF metadata */
2012
5/6
✓ Branch 0 taken 20 times.
✓ Branch 1 taken 8 times.
✓ Branch 2 taken 18 times.
✓ Branch 3 taken 2 times.
✓ Branch 4 taken 18 times.
✗ Branch 5 not taken.
 28 if (s->start_code == APP1 && id == AV_RB32("Exif") && len >= 2) {
2013 GetByteContext gbytes;
2014 int ret, le, ifd_offset, bytes_read;
2015 const uint8_t *aligned;
2016
2017 18 skip_bits(&s->gb, 16); // skip padding
2018 18 len -= 2;
2019
2020 // init byte wise reading
2021 18 aligned = align_get_bits(&s->gb);
2022 18 bytestream2_init(&gbytes, aligned, len);
2023
2024 // read TIFF header
2025 18 ret = ff_tdecode_header(&gbytes, &le, &ifd_offset);
2026
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 18 times.
 18 if (ret) {
2027 ✗ av_log(s->avctx, AV_LOG_ERROR, "mjpeg: invalid TIFF header in EXIF data\n");
2028 } else {
2029 18 bytestream2_seek(&gbytes, ifd_offset, SEEK_SET);
2030
2031 // read 0th IFD and store the metadata
2032 // (return values > 0 indicate the presence of subimage metadata)
2033 18 ret = ff_exif_decode_ifd(s->avctx, &gbytes, le, 0, &s->exif_metadata);
2034
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 18 times.
 18 if (ret < 0) {
2035 ✗ av_log(s->avctx, AV_LOG_ERROR, "mjpeg: error decoding EXIF data\n");
2036 }
2037 }
2038
2039 18 bytes_read = bytestream2_tell(&gbytes);
2040 18 skip_bits(&s->gb, bytes_read << 3);
2041 18 len -= bytes_read;
2042
2043 18 goto out;
2044 }
2045
2046 /* Apple MJPEG-A */
2047
3/4
✓ Branch 0 taken 2 times.
✓ Branch 1 taken 8 times.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
 10 if ((s->start_code == APP1) && (len > (0x28 - 8))) {
2048 2 id = get_bits_long(&s->gb, 32);
2049 2 len -= 4;
2050 /* Apple MJPEG-A */
2051
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2 times.
 2 if (id == AV_RB32("mjpg")) {
2052 /* structure:
2053 4bytes field size
2054 4bytes pad field size
2055 4bytes next off
2056 4bytes quant off
2057 4bytes huff off
2058 4bytes image off
2059 4bytes scan off
2060 4bytes data off
2061 */
2062 ✗ if (s->avctx->debug & FF_DEBUG_PICT_INFO)
2063 ✗ av_log(s->avctx, AV_LOG_INFO, "mjpeg: Apple MJPEG-A header found\n");
2064 }
2065 }
2066
2067
4/6
✓ Branch 0 taken 6 times.
✓ Branch 1 taken 4 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 4 times.
✗ Branch 4 not taken.
✓ Branch 5 taken 4 times.
 10 if (s->start_code == APP2 && id == AV_RB32("ICC_") && len >= 10) {
2068 int id2;
2069 unsigned seqno;
2070 unsigned nummarkers;
2071
2072 4 id = get_bits_long(&s->gb, 32);
2073 4 id2 = get_bits(&s->gb, 24);
2074 4 len -= 7;
2075
2/4
✓ Branch 0 taken 4 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 4 times.
 4 if (id != AV_RB32("PROF") || id2 != AV_RB24("ILE")) {
2076 ✗ av_log(s->avctx, AV_LOG_WARNING, "Invalid ICC_PROFILE header in APP2\n");
2077 ✗ goto out;
2078 }
2079
2080 4 skip_bits(&s->gb, 8);
2081 4 seqno = get_bits(&s->gb, 8);
2082 4 len -= 2;
2083
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 4 times.
 4 if (seqno == 0) {
2084 ✗ av_log(s->avctx, AV_LOG_WARNING, "Invalid sequence number in APP2\n");
2085 ✗ goto out;
2086 }
2087
2088 4 nummarkers = get_bits(&s->gb, 8);
2089 4 len -= 1;
2090
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 4 times.
 4 if (nummarkers == 0) {
2091 ✗ av_log(s->avctx, AV_LOG_WARNING, "Invalid number of markers coded in APP2\n");
2092 ✗ goto out;
2093
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 4 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 4 } else if (s->iccnum != 0 && nummarkers != s->iccnum) {
2094 ✗ av_log(s->avctx, AV_LOG_WARNING, "Mistmatch in coded number of ICC markers between markers\n");
2095 ✗ goto out;
2096
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 4 times.
 4 } else if (seqno > nummarkers) {
2097 ✗ av_log(s->avctx, AV_LOG_WARNING, "Mismatching sequence number and coded number of ICC markers\n");
2098 ✗ goto out;
2099 }
2100
2101 /* Allocate if this is the first APP2 we've seen. */
2102
1/2
✓ Branch 0 taken 4 times.
✗ Branch 1 not taken.
 4 if (s->iccnum == 0) {
2103
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 4 times.
 4 if (!FF_ALLOCZ_TYPED_ARRAY(s->iccentries, nummarkers)) {
2104 ✗ av_log(s->avctx, AV_LOG_ERROR, "Could not allocate ICC data arrays\n");
2105 ✗ return AVERROR(ENOMEM);
2106 }
2107 4 s->iccnum = nummarkers;
2108 }
2109
2110
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 4 times.
 4 if (s->iccentries[seqno - 1].data) {
2111 ✗ av_log(s->avctx, AV_LOG_WARNING, "Duplicate ICC sequence number\n");
2112 ✗ goto out;
2113 }
2114
2115 4 s->iccentries[seqno - 1].length = len;
2116 4 s->iccentries[seqno - 1].data = av_malloc(len);
2117
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 4 times.
 4 if (!s->iccentries[seqno - 1].data) {
2118 ✗ av_log(s->avctx, AV_LOG_ERROR, "Could not allocate ICC data buffer\n");
2119 ✗ return AVERROR(ENOMEM);
2120 }
2121
2122 4 memcpy(s->iccentries[seqno - 1].data, align_get_bits(&s->gb), len);
2123 4 skip_bits(&s->gb, len << 3);
2124 4 len = 0;
2125 4 s->iccread++;
2126
2127
1/2
✓ Branch 0 taken 4 times.
✗ Branch 1 not taken.
 4 if (s->iccread > s->iccnum)
2128 ✗ av_log(s->avctx, AV_LOG_WARNING, "Read more ICC markers than are supposed to be coded\n");
2129 }
2130
2131 10 out:
2132 /* slow but needed for extreme adobe jpegs */
2133
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 192 times.
 192 if (len < 0)
2134 ✗ av_log(s->avctx, AV_LOG_ERROR,
2135 "mjpeg: error, decode_app parser read over the end\n");
2136
2/2
✓ Branch 0 taken 107450 times.
✓ Branch 1 taken 192 times.
 107642 while (--len > 0)
2137 107450 skip_bits(&s->gb, 8);
2138
2139 192 return 0;
2140 }
2141
2142 231 static int mjpeg_decode_com(MJpegDecodeContext *s)
2143 {
2144 231 int len = get_bits(&s->gb, 16);
2145
2/4
✓ Branch 0 taken 231 times.
✗ Branch 1 not taken.
✓ Branch 3 taken 231 times.
✗ Branch 4 not taken.
 231 if (len >= 2 && 8 * len - 16 <= get_bits_left(&s->gb)) {
2146 int i;
2147 231 char *cbuf = av_malloc(len - 1);
2148
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 231 times.
 231 if (!cbuf)
2149 ✗ return AVERROR(ENOMEM);
2150
2151
2/2
✓ Branch 0 taken 3400 times.
✓ Branch 1 taken 231 times.
 3631 for (i = 0; i < len - 2; i++)
2152 3400 cbuf[i] = get_bits(&s->gb, 8);
2153
3/4
✓ Branch 0 taken 231 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 7 times.
✓ Branch 3 taken 224 times.
 231 if (i > 0 && cbuf[i - 1] == '\n')
2154 7 cbuf[i - 1] = 0;
2155 else
2156 224 cbuf[i] = 0;
2157
2158
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 231 times.
 231 if (s->avctx->debug & FF_DEBUG_PICT_INFO)
2159 ✗ av_log(s->avctx, AV_LOG_INFO, "comment: '%s'\n", cbuf);
2160
2161 /* buggy avid, it puts EOI only at every 10th frame */
2162
2/2
✓ Branch 0 taken 11 times.
✓ Branch 1 taken 220 times.
 231 if (!strncmp(cbuf, "AVID", 4)) {
2163 11 parse_avid(s, cbuf, len);
2164
2/2
✓ Branch 0 taken 205 times.
✓ Branch 1 taken 15 times.
 220 } else if (!strcmp(cbuf, "CS=ITU601"))
2165 205 s->cs_itu601 = 1;
2166
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 15 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 15 else if ((!strncmp(cbuf, "Intel(R) JPEG Library, version 1", 32) && s->avctx->codec_tag) ||
2167
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 15 times.
 15 (!strncmp(cbuf, "Metasoft MJPEG Codec", 20)))
2168 ✗ s->flipped = 1;
2169
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 15 times.
 15 else if (!strcmp(cbuf, "MULTISCOPE II")) {
2170 ✗ s->avctx->sample_aspect_ratio = (AVRational) { 1, 2 };
2171 ✗ s->multiscope = 2;
2172 }
2173
2174 231 av_free(cbuf);
2175 }
2176
2177 231 return 0;
2178 }
2179
2180 /* return the 8 bit start code value and update the search
2181 state. Return -1 if no start code found */
2182 16171 static int find_marker(const uint8_t **pbuf_ptr, const uint8_t *buf_end)
2183 {
2184 const uint8_t *buf_ptr;
2185 unsigned int v, v2;
2186 int val;
2187 16171 int skipped = 0;
2188
2189 16171 buf_ptr = *pbuf_ptr;
2190
2/2
✓ Branch 0 taken 279737 times.
✓ Branch 1 taken 5 times.
 279742 while (buf_end - buf_ptr > 1) {
2191 279737 v = *buf_ptr++;
2192 279737 v2 = *buf_ptr;
2193
7/8
✓ Branch 0 taken 18678 times.
✓ Branch 1 taken 261059 times.
✓ Branch 2 taken 16187 times.
✓ Branch 3 taken 2491 times.
✓ Branch 4 taken 16166 times.
✓ Branch 5 taken 21 times.
✓ Branch 6 taken 16166 times.
✗ Branch 7 not taken.
 279737 if ((v == 0xff) && (v2 >= SOF0) && (v2 <= COM) && buf_ptr < buf_end) {
2194 16166 val = *buf_ptr++;
2195 16166 goto found;
2196 }
2197 263571 skipped++;
2198 }
2199 5 buf_ptr = buf_end;
2200 5 val = -1;
2201 16171 found:
2202 ff_dlog(NULL, "find_marker skipped %d bytes\n", skipped);
2203 16171 *pbuf_ptr = buf_ptr;
2204 16171 return val;
2205 }
2206
2207 16171 int ff_mjpeg_find_marker(MJpegDecodeContext *s,
2208 const uint8_t **buf_ptr, const uint8_t *buf_end,
2209 const uint8_t **unescaped_buf_ptr,
2210 int *unescaped_buf_size)
2211 {
2212 int start_code;
2213 16171 start_code = find_marker(buf_ptr, buf_end);
2214
2215 16171 av_fast_padded_malloc(&s->buffer, &s->buffer_size, buf_end - *buf_ptr);
2216
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 16171 times.
 16171 if (!s->buffer)
2217 ✗ return AVERROR(ENOMEM);
2218
2219 /* unescape buffer of SOS, use special treatment for JPEG-LS */
2220
4/4
✓ Branch 0 taken 2591 times.
✓ Branch 1 taken 13580 times.
✓ Branch 2 taken 2378 times.
✓ Branch 3 taken 213 times.
 18549 if (start_code == SOS && !s->ls) {
2221 2378 const uint8_t *src = *buf_ptr;
2222 2378 const uint8_t *ptr = src;
2223 2378 uint8_t *dst = s->buffer;
2224
2225 #define copy_data_segment(skip) do { \
2226 ptrdiff_t length = (ptr - src) - (skip); \
2227 if (length > 0) { \
2228 memcpy(dst, src, length); \
2229 dst += length; \
2230 src = ptr; \
2231 } \
2232 } while (0)
2233
2234
2/2
✓ Branch 0 taken 74 times.
✓ Branch 1 taken 2304 times.
 2378 if (s->avctx->codec_id == AV_CODEC_ID_THP) {
2235 74 ptr = buf_end;
2236
1/2
✓ Branch 0 taken 74 times.
✗ Branch 1 not taken.
 74 copy_data_segment(0);
2237 } else {
2238
1/2
✓ Branch 0 taken 43140685 times.
✗ Branch 1 not taken.
 43140685 while (ptr < buf_end) {
2239 43140685 uint8_t x = *(ptr++);
2240
2241
2/2
✓ Branch 0 taken 185290 times.
✓ Branch 1 taken 42955395 times.
 43140685 if (x == 0xff) {
2242 185290 ptrdiff_t skip = 0;
2243
4/4
✓ Branch 0 taken 368300 times.
✓ Branch 1 taken 2294 times.
✓ Branch 2 taken 185304 times.
✓ Branch 3 taken 182996 times.
 370594 while (ptr < buf_end && x == 0xff) {
2244 185304 x = *(ptr++);
2245 185304 skip++;
2246 }
2247
2248 /* 0xFF, 0xFF, ... */
2249
2/2
✓ Branch 0 taken 7 times.
✓ Branch 1 taken 185283 times.
 185290 if (skip > 1) {
2250
1/2
✓ Branch 0 taken 7 times.
✗ Branch 1 not taken.
 7 copy_data_segment(skip);
2251
2252 /* decrement src as it is equal to ptr after the
2253 * copy_data_segment macro and we might want to
2254 * copy the current value of x later on */
2255 7 src--;
2256 }
2257
2258
4/4
✓ Branch 0 taken 2380 times.
✓ Branch 1 taken 182910 times.
✓ Branch 2 taken 2304 times.
✓ Branch 3 taken 76 times.
 185290 if (x < RST0 || x > RST7) {
2259
2/2
✓ Branch 0 taken 185207 times.
✓ Branch 1 taken 7 times.
 185214 copy_data_segment(1);
2260
2/2
✓ Branch 0 taken 2304 times.
✓ Branch 1 taken 182910 times.
 185214 if (x)
2261 2304 break;
2262 }
2263 }
2264 }
2265
2/2
✓ Branch 0 taken 7 times.
✓ Branch 1 taken 2297 times.
 2304 if (src < ptr)
2266
1/2
✓ Branch 0 taken 7 times.
✗ Branch 1 not taken.
 7 copy_data_segment(0);
2267 }
2268 #undef copy_data_segment
2269
2270 2378 *unescaped_buf_ptr = s->buffer;
2271 2378 *unescaped_buf_size = dst - s->buffer;
2272 2378 memset(s->buffer + *unescaped_buf_size, 0,
2273 AV_INPUT_BUFFER_PADDING_SIZE);
2274
2275 2378 av_log(s->avctx, AV_LOG_DEBUG, "escaping removed %"PTRDIFF_SPECIFIER" bytes\n",
2276 2378 (buf_end - *buf_ptr) - (dst - s->buffer));
2277
3/4
✓ Branch 0 taken 213 times.
✓ Branch 1 taken 13580 times.
✓ Branch 2 taken 213 times.
✗ Branch 3 not taken.
 13793 } else if (start_code == SOS && s->ls) {
2278 213 const uint8_t *src = *buf_ptr;
2279 213 uint8_t *dst = s->buffer;
2280 213 int bit_count = 0;
2281 213 int t = 0, b = 0;
2282 PutBitContext pb;
2283
2284 /* find marker */
2285
1/2
✓ Branch 0 taken 27487079 times.
✗ Branch 1 not taken.
 27487079 while (src + t < buf_end) {
2286 27487079 uint8_t x = src[t++];
2287
2/2
✓ Branch 0 taken 154051 times.
✓ Branch 1 taken 27333028 times.
 27487079 if (x == 0xff) {
2288
4/4
✓ Branch 0 taken 307889 times.
✓ Branch 1 taken 213 times.
✓ Branch 2 taken 154051 times.
✓ Branch 3 taken 153838 times.
 308102 while ((src + t < buf_end) && x == 0xff)
2289 154051 x = src[t++];
2290
2/2
✓ Branch 0 taken 213 times.
✓ Branch 1 taken 153838 times.
 154051 if (x & 0x80) {
2291 213 t -= FFMIN(2, t);
2292 213 break;
2293 }
2294 }
2295 }
2296 213 bit_count = t * 8;
2297 213 init_put_bits(&pb, dst, t);
2298
2299 /* unescape bitstream */
2300
2/2
✓ Branch 0 taken 27486866 times.
✓ Branch 1 taken 213 times.
 27487079 while (b < t) {
2301 27486866 uint8_t x = src[b++];
2302 27486866 put_bits(&pb, 8, x);
2303
3/4
✓ Branch 0 taken 153838 times.
✓ Branch 1 taken 27333028 times.
✓ Branch 2 taken 153838 times.
✗ Branch 3 not taken.
 27486866 if (x == 0xFF && b < t) {
2304 153838 x = src[b++];
2305
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 153838 times.
 153838 if (x & 0x80) {
2306 ✗ av_log(s->avctx, AV_LOG_WARNING, "Invalid escape sequence\n");
2307 ✗ x &= 0x7f;
2308 }
2309 153838 put_bits(&pb, 7, x);
2310 153838 bit_count--;
2311 }
2312 }
2313 213 flush_put_bits(&pb);
2314
2315 213 *unescaped_buf_ptr = dst;
2316 213 *unescaped_buf_size = (bit_count + 7) >> 3;
2317 213 memset(s->buffer + *unescaped_buf_size, 0,
2318 AV_INPUT_BUFFER_PADDING_SIZE);
2319 } else {
2320 13580 *unescaped_buf_ptr = *buf_ptr;
2321 13580 *unescaped_buf_size = buf_end - *buf_ptr;
2322 }
2323
2324 16171 return start_code;
2325 }
2326
2327 183 static void reset_icc_profile(MJpegDecodeContext *s)
2328 {
2329 int i;
2330
2331
2/2
✓ Branch 0 taken 4 times.
✓ Branch 1 taken 179 times.
 183 if (s->iccentries) {
2332
2/2
✓ Branch 0 taken 4 times.
✓ Branch 1 taken 4 times.
 8 for (i = 0; i < s->iccnum; i++)
2333 4 av_freep(&s->iccentries[i].data);
2334 4 av_freep(&s->iccentries);
2335 }
2336
2337 183 s->iccread = 0;
2338 183 s->iccnum = 0;
2339 183 }
2340
2341 // SMV JPEG just stacks several output frames into one JPEG picture
2342 // we handle that by setting up the cropping parameters appropriately
2343 13 static int smv_process_frame(AVCodecContext *avctx, AVFrame *frame)
2344 {
2345 13 MJpegDecodeContext *s = avctx->priv_data;
2346 int ret;
2347
2348
2/2
✓ Branch 0 taken 10 times.
✓ Branch 1 taken 3 times.
 13 if (s->smv_next_frame > 0) {
2349
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 10 times.
 10 av_assert0(s->smv_frame->buf[0]);
2350 10 av_frame_unref(frame);
2351 10 ret = av_frame_ref(frame, s->smv_frame);
2352
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 10 times.
 10 if (ret < 0)
2353 ✗ return ret;
2354 } else {
2355
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 3 times.
 3 av_assert0(frame->buf[0]);
2356 3 av_frame_unref(s->smv_frame);
2357 3 ret = av_frame_ref(s->smv_frame, frame);
2358
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 3 times.
 3 if (ret < 0)
2359 ✗ return ret;
2360 }
2361
2362
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 13 times.
 13 av_assert0((s->smv_next_frame + 1) * avctx->height <= avctx->coded_height);
2363
2364 13 frame->width = avctx->coded_width;
2365 13 frame->height = avctx->coded_height;
2366 13 frame->crop_top = FFMIN(s->smv_next_frame * avctx->height, frame->height);
2367 13 frame->crop_bottom = frame->height - (s->smv_next_frame + 1) * avctx->height;
2368
2369 13 s->smv_next_frame = (s->smv_next_frame + 1) % s->smv_frames_per_jpeg;
2370
2371
2/2
✓ Branch 0 taken 2 times.
✓ Branch 1 taken 11 times.
 13 if (s->smv_next_frame == 0)
2372 2 av_frame_unref(s->smv_frame);
2373
2374 13 return 0;
2375 }
2376
2377 4945 static int mjpeg_get_packet(AVCodecContext *avctx)
2378 {
2379 4945 MJpegDecodeContext *s = avctx->priv_data;
2380 int ret;
2381
2382 4945 av_packet_unref(s->pkt);
2383 4945 ret = ff_decode_get_packet(avctx, s->pkt);
2384
2/2
✓ Branch 0 taken 2388 times.
✓ Branch 1 taken 2557 times.
 4945 if (ret < 0)
2385 2388 return ret;
2386
2387 #if CONFIG_SP5X_DECODER || CONFIG_AMV_DECODER
2388
2/2
✓ Branch 0 taken 2546 times.
✓ Branch 1 taken 11 times.
 2557 if (avctx->codec_id == AV_CODEC_ID_SP5X ||
2389
2/2
✓ Branch 0 taken 367 times.
✓ Branch 1 taken 2179 times.
 2546 avctx->codec_id == AV_CODEC_ID_AMV) {
2390 378 ret = ff_sp5x_process_packet(avctx, s->pkt);
2391
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 378 times.
 378 if (ret < 0)
2392 ✗ return ret;
2393 }
2394 #endif
2395
2396 2557 s->buf_size = s->pkt->size;
2397
2398 2557 return 0;
2399 }
2400
2401 4955 int ff_mjpeg_receive_frame(AVCodecContext *avctx, AVFrame *frame)
2402 {
2403 4955 MJpegDecodeContext *s = avctx->priv_data;
2404 const uint8_t *buf_end, *buf_ptr;
2405 const uint8_t *unescaped_buf_ptr;
2406 int hshift, vshift;
2407 int unescaped_buf_size;
2408 int start_code;
2409 int i, index;
2410 4955 int ret = 0;
2411 int is16bit;
2412 4955 AVDictionaryEntry *e = NULL;
2413
2414 4955 s->force_pal8 = 0;
2415
2416
4/4
✓ Branch 0 taken 17 times.
✓ Branch 1 taken 4938 times.
✓ Branch 2 taken 10 times.
✓ Branch 3 taken 7 times.
 4955 if (avctx->codec_id == AV_CODEC_ID_SMVJPEG && s->smv_next_frame > 0)
2417 10 return smv_process_frame(avctx, frame);
2418
2419 4945 av_dict_free(&s->exif_metadata);
2420 4945 av_freep(&s->stereo3d);
2421 4945 s->adobe_transform = -1;
2422
2423
2/2
✓ Branch 0 taken 4 times.
✓ Branch 1 taken 4941 times.
 4945 if (s->iccnum != 0)
2424 4 reset_icc_profile(s);
2425
2426 4945 ret = mjpeg_get_packet(avctx);
2427
2/2
✓ Branch 0 taken 2388 times.
✓ Branch 1 taken 2557 times.
 4945 if (ret < 0)
2428 2388 return ret;
2429 2557 redo_for_pal8:
2430 2565 buf_ptr = s->pkt->data;
2431 2565 buf_end = s->pkt->data + s->pkt->size;
2432
1/2
✓ Branch 0 taken 15969 times.
✗ Branch 1 not taken.
 15969 while (buf_ptr < buf_end) {
2433 /* find start next marker */
2434 15969 start_code = ff_mjpeg_find_marker(s, &buf_ptr, buf_end,
2435 &unescaped_buf_ptr,
2436 &unescaped_buf_size);
2437 /* EOF */
2438
2/2
✓ Branch 0 taken 5 times.
✓ Branch 1 taken 15964 times.
 15969 if (start_code < 0) {
2439 5 break;
2440
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 15964 times.
 15964 } else if (unescaped_buf_size > INT_MAX / 8) {
2441 ✗ av_log(avctx, AV_LOG_ERROR,
2442 "MJPEG packet 0x%x too big (%d/%d), corrupt data?\n",
2443 ✗ start_code, unescaped_buf_size, s->pkt->size);
2444 ✗ return AVERROR_INVALIDDATA;
2445 }
2446 15964 av_log(avctx, AV_LOG_DEBUG, "marker=%x avail_size_in_buf=%"PTRDIFF_SPECIFIER"\n",
2447 start_code, buf_end - buf_ptr);
2448
2449 15964 ret = init_get_bits8(&s->gb, unescaped_buf_ptr, unescaped_buf_size);
2450
2451
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 15964 times.
 15964 if (ret < 0) {
2452 ✗ av_log(avctx, AV_LOG_ERROR, "invalid buffer\n");
2453 ✗ goto fail;
2454 }
2455
2456 15964 s->start_code = start_code;
2457
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 15964 times.
 15964 if (s->avctx->debug & FF_DEBUG_STARTCODE)
2458 ✗ av_log(avctx, AV_LOG_DEBUG, "startcode: %X\n", start_code);
2459
2460 /* process markers */
2461
4/4
✓ Branch 0 taken 10906 times.
✓ Branch 1 taken 5058 times.
✓ Branch 2 taken 2 times.
✓ Branch 3 taken 10904 times.
 15964 if (start_code >= RST0 && start_code <= RST7) {
2462 2 av_log(avctx, AV_LOG_DEBUG,
2463 "restart marker: %d\n", start_code & 0x0f);
2464 /* APP fields */
2465
4/4
✓ Branch 0 taken 660 times.
✓ Branch 1 taken 15302 times.
✓ Branch 2 taken 192 times.
✓ Branch 3 taken 468 times.
 15962 } else if (start_code >= APP0 && start_code <= APP15) {
2466
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 192 times.
 192 if ((ret = mjpeg_decode_app(s)) < 0)
2467 ✗ av_log(avctx, AV_LOG_ERROR, "unable to decode APP fields: %s\n",
2468 ✗ av_err2str(ret));
2469 /* Comment */
2470
2/2
✓ Branch 0 taken 231 times.
✓ Branch 1 taken 15539 times.
 15770 } else if (start_code == COM) {
2471 231 ret = mjpeg_decode_com(s);
2472
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 231 times.
 231 if (ret < 0)
2473 ✗ return ret;
2474
2/2
✓ Branch 0 taken 2539 times.
✓ Branch 1 taken 13000 times.
 15539 } else if (start_code == DQT) {
2475 2539 ret = ff_mjpeg_decode_dqt(s);
2476
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2539 times.
 2539 if (ret < 0)
2477 ✗ return ret;
2478 }
2479
2480 15964 ret = -1;
2481
2482 if (!CONFIG_JPEGLS_DECODER &&
2483 (start_code == SOF48 || start_code == LSE)) {
2484 av_log(avctx, AV_LOG_ERROR, "JPEG-LS support not enabled.\n");
2485 return AVERROR(ENOSYS);
2486 }
2487
2488
2/2
✓ Branch 0 taken 677 times.
✓ Branch 1 taken 15287 times.
 15964 if (avctx->skip_frame == AVDISCARD_ALL) {
2489
2/2
✓ Branch 0 taken 300 times.
✓ Branch 1 taken 377 times.
 677 switch(start_code) {
2490 300 case SOF0:
2491 case SOF1:
2492 case SOF2:
2493 case SOF3:
2494 case SOF48:
2495 case SOI:
2496 case SOS:
2497 case EOI:
2498 300 break;
2499 377 default:
2500 377 goto skip;
2501 }
2502 }
2503
2504
10/12
✓ Branch 0 taken 2568 times.
✓ Branch 1 taken 2532 times.
✓ Branch 2 taken 2142 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 205 times.
✓ Branch 5 taken 221 times.
✓ Branch 6 taken 16 times.
✓ Branch 7 taken 2556 times.
✓ Branch 8 taken 2560 times.
✓ Branch 9 taken 16 times.
✗ Branch 10 not taken.
✓ Branch 11 taken 2771 times.
 15587 switch (start_code) {
2505 2568 case SOI:
2506 2568 s->restart_interval = 0;
2507 2568 s->restart_count = 0;
2508 2568 s->raw_image_buffer = buf_ptr;
2509 2568 s->raw_image_buffer_size = buf_end - buf_ptr;
2510 /* nothing to do on SOI */
2511 2568 break;
2512 2532 case DHT:
2513
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 2532 times.
 2532 if ((ret = ff_mjpeg_decode_dht(s)) < 0) {
2514 ✗ av_log(avctx, AV_LOG_ERROR, "huffman table decode error\n");
2515 ✗ goto fail;
2516 }
2517 2532 break;
2518 2142 case SOF0:
2519 case SOF1:
2520
2/2
✓ Branch 0 taken 2140 times.
✓ Branch 1 taken 2 times.
 2142 if (start_code == SOF0)
2521 2140 s->avctx->profile = FF_PROFILE_MJPEG_HUFFMAN_BASELINE_DCT;
2522 else
2523 2 s->avctx->profile = FF_PROFILE_MJPEG_HUFFMAN_EXTENDED_SEQUENTIAL_DCT;
2524 2142 s->lossless = 0;
2525 2142 s->ls = 0;
2526 2142 s->progressive = 0;
2527
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 2142 times.
 2142 if ((ret = ff_mjpeg_decode_sof(s)) < 0)
2528 ✗ goto fail;
2529 2142 break;
2530 ✗ case SOF2:
2531 ✗ s->avctx->profile = FF_PROFILE_MJPEG_HUFFMAN_PROGRESSIVE_DCT;
2532 ✗ s->lossless = 0;
2533 ✗ s->ls = 0;
2534 ✗ s->progressive = 1;
2535 ✗ if ((ret = ff_mjpeg_decode_sof(s)) < 0)
2536 ✗ goto fail;
2537 ✗ break;
2538 205 case SOF3:
2539 205 s->avctx->profile = FF_PROFILE_MJPEG_HUFFMAN_LOSSLESS;
2540 205 s->avctx->properties |= FF_CODEC_PROPERTY_LOSSLESS;
2541 205 s->lossless = 1;
2542 205 s->ls = 0;
2543 205 s->progressive = 0;
2544
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 205 times.
 205 if ((ret = ff_mjpeg_decode_sof(s)) < 0)
2545 ✗ goto fail;
2546 205 break;
2547 221 case SOF48:
2548 221 s->avctx->profile = FF_PROFILE_MJPEG_JPEG_LS;
2549 221 s->avctx->properties |= FF_CODEC_PROPERTY_LOSSLESS;
2550 221 s->lossless = 1;
2551 221 s->ls = 1;
2552 221 s->progressive = 0;
2553
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 221 times.
 221 if ((ret = ff_mjpeg_decode_sof(s)) < 0)
2554 ✗ goto fail;
2555 221 break;
2556 16 case LSE:
2557
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 16 times.
 16 if (!CONFIG_JPEGLS_DECODER ||
2558 16 (ret = ff_jpegls_decode_lse(s)) < 0)
2559 ✗ goto fail;
2560
2/2
✓ Branch 0 taken 8 times.
✓ Branch 1 taken 8 times.
 16 if (ret == 1)
2561 8 goto redo_for_pal8;
2562 8 break;
2563 case EOI:
2564 2560 eoi_parser:
2565
3/4
✓ Branch 0 taken 2560 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 2484 times.
✓ Branch 3 taken 76 times.
 2560 if (!avctx->hwaccel && avctx->skip_frame != AVDISCARD_ALL &&
2566
1/6
✗ Branch 0 not taken.
✓ Branch 1 taken 2484 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
 2484 s->progressive && s->cur_scan && s->got_picture)
2567 ✗ mjpeg_idct_scan_progressive_ac(s);
2568 2560 s->cur_scan = 0;
2569
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2560 times.
 2560 if (!s->got_picture) {
2570 ✗ av_log(avctx, AV_LOG_WARNING,
2571 "Found EOI before any SOF, ignoring\n");
2572 ✗ break;
2573 }
2574
2/2
✓ Branch 0 taken 8 times.
✓ Branch 1 taken 2552 times.
 2560 if (s->interlaced) {
2575 8 s->bottom_field ^= 1;
2576 /* if not bottom field, do not output image yet */
2577
2/2
✓ Branch 0 taken 4 times.
✓ Branch 1 taken 4 times.
 8 if (s->bottom_field == !s->interlace_polarity)
2578 4 break;
2579 }
2580
2/2
✓ Branch 0 taken 75 times.
✓ Branch 1 taken 2481 times.
 2556 if (avctx->skip_frame == AVDISCARD_ALL) {
2581 75 s->got_picture = 0;
2582 75 ret = AVERROR(EAGAIN);
2583 75 goto the_end_no_picture;
2584 }
2585
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2481 times.
 2481 if (s->avctx->hwaccel) {
2586 ✗ ret = s->avctx->hwaccel->end_frame(s->avctx);
2587 ✗ if (ret < 0)
2588 ✗ return ret;
2589
2590 ✗ av_freep(&s->hwaccel_picture_private);
2591 }
2592
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 2481 times.
 2481 if ((ret = av_frame_ref(frame, s->picture_ptr)) < 0)
2593 ✗ return ret;
2594 2481 s->got_picture = 0;
2595
2596 2481 frame->pkt_dts = s->pkt->dts;
2597
2598
3/4
✓ Branch 0 taken 2068 times.
✓ Branch 1 taken 413 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 2068 times.
 2481 if (!s->lossless && avctx->debug & FF_DEBUG_QP) {
2599 ✗ int qp = FFMAX3(s->qscale[0],
2600 s->qscale[1],
2601 s->qscale[2]);
2602
2603 ✗ av_log(avctx, AV_LOG_DEBUG, "QP: %d\n", qp);
2604 }
2605
2606 2481 goto the_end;
2607 2560 case SOS:
2608 2560 s->raw_scan_buffer = buf_ptr;
2609 2560 s->raw_scan_buffer_size = buf_end - buf_ptr;
2610
2611 2560 s->cur_scan++;
2612
2/2
✓ Branch 0 taken 76 times.
✓ Branch 1 taken 2484 times.
 2560 if (avctx->skip_frame == AVDISCARD_ALL) {
2613 76 skip_bits(&s->gb, get_bits_left(&s->gb));
2614 76 break;
2615 }
2616
2617
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 2484 times.
 2484 if ((ret = ff_mjpeg_decode_sos(s, NULL, 0, NULL)) < 0 &&
2618 ✗ (avctx->err_recognition & AV_EF_EXPLODE))
2619 ✗ goto fail;
2620 2484 break;
2621 16 case DRI:
2622
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 16 times.
 16 if ((ret = mjpeg_decode_dri(s)) < 0)
2623 ✗ return ret;
2624 16 break;
2625 ✗ case SOF5:
2626 case SOF6:
2627 case SOF7:
2628 case SOF9:
2629 case SOF10:
2630 case SOF11:
2631 case SOF13:
2632 case SOF14:
2633 case SOF15:
2634 case JPG:
2635 ✗ av_log(avctx, AV_LOG_ERROR,
2636 "mjpeg: unsupported coding type (%x)\n", start_code);
2637 ✗ break;
2638 }
2639
2640 13404 skip:
2641 /* eof process start code */
2642 13404 buf_ptr += (get_bits_count(&s->gb) + 7) / 8;
2643 26808 av_log(avctx, AV_LOG_DEBUG,
2644 "marker parser used %d bytes (%d bits)\n",
2645 13404 (get_bits_count(&s->gb) + 7) / 8, get_bits_count(&s->gb));
2646 }
2647
3/4
✓ Branch 0 taken 4 times.
✓ Branch 1 taken 1 times.
✓ Branch 2 taken 4 times.
✗ Branch 3 not taken.
 5 if (s->got_picture && s->cur_scan) {
2648 4 av_log(avctx, AV_LOG_WARNING, "EOI missing, emulating\n");
2649 4 goto eoi_parser;
2650 }
2651 1 av_log(avctx, AV_LOG_FATAL, "No JPEG data found in image\n");
2652 1 return AVERROR_INVALIDDATA;
2653 ✗ fail:
2654 ✗ s->got_picture = 0;
2655 ✗ return ret;
2656 2481 the_end:
2657
2658 2481 is16bit = av_pix_fmt_desc_get(s->avctx->pix_fmt)->comp[0].step > 1;
2659
2660
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2481 times.
 2481 if (AV_RB32(s->upscale_h)) {
2661 int p;
2662 ✗ av_assert0(avctx->pix_fmt == AV_PIX_FMT_YUVJ444P ||
2663 avctx->pix_fmt == AV_PIX_FMT_YUV444P ||
2664 avctx->pix_fmt == AV_PIX_FMT_YUVJ440P ||
2665 avctx->pix_fmt == AV_PIX_FMT_YUV440P ||
2666 avctx->pix_fmt == AV_PIX_FMT_YUVA444P ||
2667 avctx->pix_fmt == AV_PIX_FMT_YUVJ420P ||
2668 avctx->pix_fmt == AV_PIX_FMT_YUV420P ||
2669 avctx->pix_fmt == AV_PIX_FMT_YUV420P16||
2670 avctx->pix_fmt == AV_PIX_FMT_YUVA420P ||
2671 avctx->pix_fmt == AV_PIX_FMT_YUVA420P16||
2672 avctx->pix_fmt == AV_PIX_FMT_GBRP ||
2673 avctx->pix_fmt == AV_PIX_FMT_GBRAP
2674 );
2675 ✗ ret = av_pix_fmt_get_chroma_sub_sample(s->avctx->pix_fmt, &hshift, &vshift);
2676 ✗ if (ret)
2677 ✗ return ret;
2678
2679 ✗ av_assert0(s->nb_components == av_pix_fmt_count_planes(s->picture_ptr->format));
2680 ✗ for (p = 0; p<s->nb_components; p++) {
2681 ✗ uint8_t *line = s->picture_ptr->data[p];
2682 ✗ int w = s->width;
2683 ✗ int h = s->height;
2684 ✗ if (!s->upscale_h[p])
2685 ✗ continue;
2686 ✗ if (p==1 || p==2) {
2687 ✗ w = AV_CEIL_RSHIFT(w, hshift);
2688 ✗ h = AV_CEIL_RSHIFT(h, vshift);
2689 }
2690 ✗ if (s->upscale_v[p] == 1)
2691 ✗ h = (h+1)>>1;
2692 ✗ av_assert0(w > 0);
2693 ✗ for (i = 0; i < h; i++) {
2694 ✗ if (s->upscale_h[p] == 1) {
2695 ✗ if (is16bit) ((uint16_t*)line)[w - 1] = ((uint16_t*)line)[(w - 1) / 2];
2696 ✗ else line[w - 1] = line[(w - 1) / 2];
2697 ✗ for (index = w - 2; index > 0; index--) {
2698 ✗ if (is16bit)
2699 ✗ ((uint16_t*)line)[index] = (((uint16_t*)line)[index / 2] + ((uint16_t*)line)[(index + 1) / 2]) >> 1;
2700 else
2701 ✗ line[index] = (line[index / 2] + line[(index + 1) / 2]) >> 1;
2702 }
2703 ✗ } else if (s->upscale_h[p] == 2) {
2704 ✗ if (is16bit) {
2705 ✗ ((uint16_t*)line)[w - 1] = ((uint16_t*)line)[(w - 1) / 3];
2706 ✗ if (w > 1)
2707 ✗ ((uint16_t*)line)[w - 2] = ((uint16_t*)line)[w - 1];
2708 } else {
2709 ✗ line[w - 1] = line[(w - 1) / 3];
2710 ✗ if (w > 1)
2711 ✗ line[w - 2] = line[w - 1];
2712 }
2713 ✗ for (index = w - 3; index > 0; index--) {
2714 ✗ line[index] = (line[index / 3] + line[(index + 1) / 3] + line[(index + 2) / 3] + 1) / 3;
2715 }
2716 }
2717 ✗ line += s->linesize[p];
2718 }
2719 }
2720 }
2721
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2481 times.
 2481 if (AV_RB32(s->upscale_v)) {
2722 int p;
2723 ✗ av_assert0(avctx->pix_fmt == AV_PIX_FMT_YUVJ444P ||
2724 avctx->pix_fmt == AV_PIX_FMT_YUV444P ||
2725 avctx->pix_fmt == AV_PIX_FMT_YUVJ422P ||
2726 avctx->pix_fmt == AV_PIX_FMT_YUV422P ||
2727 avctx->pix_fmt == AV_PIX_FMT_YUVJ420P ||
2728 avctx->pix_fmt == AV_PIX_FMT_YUV420P ||
2729 avctx->pix_fmt == AV_PIX_FMT_YUV440P ||
2730 avctx->pix_fmt == AV_PIX_FMT_YUVJ440P ||
2731 avctx->pix_fmt == AV_PIX_FMT_YUVA444P ||
2732 avctx->pix_fmt == AV_PIX_FMT_YUVA420P ||
2733 avctx->pix_fmt == AV_PIX_FMT_YUVA420P16||
2734 avctx->pix_fmt == AV_PIX_FMT_GBRP ||
2735 avctx->pix_fmt == AV_PIX_FMT_GBRAP
2736 );
2737 ✗ ret = av_pix_fmt_get_chroma_sub_sample(s->avctx->pix_fmt, &hshift, &vshift);
2738 ✗ if (ret)
2739 ✗ return ret;
2740
2741 ✗ av_assert0(s->nb_components == av_pix_fmt_count_planes(s->picture_ptr->format));
2742 ✗ for (p = 0; p < s->nb_components; p++) {
2743 uint8_t *dst;
2744 ✗ int w = s->width;
2745 ✗ int h = s->height;
2746 ✗ if (!s->upscale_v[p])
2747 ✗ continue;
2748 ✗ if (p==1 || p==2) {
2749 ✗ w = AV_CEIL_RSHIFT(w, hshift);
2750 ✗ h = AV_CEIL_RSHIFT(h, vshift);
2751 }
2752 ✗ dst = &((uint8_t *)s->picture_ptr->data[p])[(h - 1) * s->linesize[p]];
2753 ✗ for (i = h - 1; i; i--) {
2754 ✗ uint8_t *src1 = &((uint8_t *)s->picture_ptr->data[p])[i * s->upscale_v[p] / (s->upscale_v[p] + 1) * s->linesize[p]];
2755 ✗ uint8_t *src2 = &((uint8_t *)s->picture_ptr->data[p])[(i + 1) * s->upscale_v[p] / (s->upscale_v[p] + 1) * s->linesize[p]];
2756 ✗ if (s->upscale_v[p] != 2 && (src1 == src2 || i == h - 1)) {
2757 ✗ memcpy(dst, src1, w);
2758 } else {
2759 ✗ for (index = 0; index < w; index++)
2760 ✗ dst[index] = (src1[index] + src2[index]) >> 1;
2761 }
2762 ✗ dst -= s->linesize[p];
2763 }
2764 }
2765 }
2766
3/4
✓ Branch 0 taken 367 times.
✓ Branch 1 taken 2114 times.
✓ Branch 2 taken 367 times.
✗ Branch 3 not taken.
 2481 if (s->flipped && !s->rgb) {
2767 int j;
2768 367 ret = av_pix_fmt_get_chroma_sub_sample(s->avctx->pix_fmt, &hshift, &vshift);
2769
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 367 times.
 367 if (ret)
2770 ✗ return ret;
2771
2772
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 367 times.
 367 av_assert0(s->nb_components == av_pix_fmt_count_planes(s->picture_ptr->format));
2773
2/2
✓ Branch 0 taken 1101 times.
✓ Branch 1 taken 367 times.
 1468 for (index=0; index<s->nb_components; index++) {
2774 1101 uint8_t *dst = s->picture_ptr->data[index];
2775 1101 int w = s->picture_ptr->width;
2776 1101 int h = s->picture_ptr->height;
2777
3/4
✓ Branch 0 taken 734 times.
✓ Branch 1 taken 367 times.
✓ Branch 2 taken 734 times.
✗ Branch 3 not taken.
 1101 if(index && index<3){
2778 734 w = AV_CEIL_RSHIFT(w, hshift);
2779 734 h = AV_CEIL_RSHIFT(h, vshift);
2780 }
2781
1/2
✓ Branch 0 taken 1101 times.
✗ Branch 1 not taken.
 1101 if(dst){
2782 1101 uint8_t *dst2 = dst + s->picture_ptr->linesize[index]*(h-1);
2783
2/2
✓ Branch 0 taken 65307 times.
✓ Branch 1 taken 1101 times.
 66408 for (i=0; i<h/2; i++) {
2784
2/2
✓ Branch 0 taken 14023446 times.
✓ Branch 1 taken 65307 times.
 14088753 for (j=0; j<w; j++)
2785 14023446 FFSWAP(int, dst[j], dst2[j]);
2786 65307 dst += s->picture_ptr->linesize[index];
2787 65307 dst2 -= s->picture_ptr->linesize[index];
2788 }
2789 }
2790 }
2791 }
2792
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 2481 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 2481 if (s->adobe_transform == 0 && s->avctx->pix_fmt == AV_PIX_FMT_GBRAP) {
2793 ✗ int w = s->picture_ptr->width;
2794 ✗ int h = s->picture_ptr->height;
2795 ✗ av_assert0(s->nb_components == 4);
2796 ✗ for (i=0; i<h; i++) {
2797 int j;
2798 uint8_t *dst[4];
2799 ✗ for (index=0; index<4; index++) {
2800 ✗ dst[index] = s->picture_ptr->data[index]
2801 ✗ + s->picture_ptr->linesize[index]*i;
2802 }
2803 ✗ for (j=0; j<w; j++) {
2804 ✗ int k = dst[3][j];
2805 ✗ int r = dst[0][j] * k;
2806 ✗ int g = dst[1][j] * k;
2807 ✗ int b = dst[2][j] * k;
2808 ✗ dst[0][j] = g*257 >> 16;
2809 ✗ dst[1][j] = b*257 >> 16;
2810 ✗ dst[2][j] = r*257 >> 16;
2811 ✗ dst[3][j] = 255;
2812 }
2813 }
2814 }
2815
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 2481 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 2481 if (s->adobe_transform == 2 && s->avctx->pix_fmt == AV_PIX_FMT_YUVA444P) {
2816 ✗ int w = s->picture_ptr->width;
2817 ✗ int h = s->picture_ptr->height;
2818 ✗ av_assert0(s->nb_components == 4);
2819 ✗ for (i=0; i<h; i++) {
2820 int j;
2821 uint8_t *dst[4];
2822 ✗ for (index=0; index<4; index++) {
2823 ✗ dst[index] = s->picture_ptr->data[index]
2824 ✗ + s->picture_ptr->linesize[index]*i;
2825 }
2826 ✗ for (j=0; j<w; j++) {
2827 ✗ int k = dst[3][j];
2828 ✗ int r = (255 - dst[0][j]) * k;
2829 ✗ int g = (128 - dst[1][j]) * k;
2830 ✗ int b = (128 - dst[2][j]) * k;
2831 ✗ dst[0][j] = r*257 >> 16;
2832 ✗ dst[1][j] = (g*257 >> 16) + 128;
2833 ✗ dst[2][j] = (b*257 >> 16) + 128;
2834 ✗ dst[3][j] = 255;
2835 }
2836 }
2837 }
2838
2839
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2481 times.
 2481 if (s->stereo3d) {
2840 ✗ AVStereo3D *stereo = av_stereo3d_create_side_data(frame);
2841 ✗ if (stereo) {
2842 ✗ stereo->type = s->stereo3d->type;
2843 ✗ stereo->flags = s->stereo3d->flags;
2844 }
2845 ✗ av_freep(&s->stereo3d);
2846 }
2847
2848
3/4
✓ Branch 0 taken 2 times.
✓ Branch 1 taken 2479 times.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
 2481 if (s->iccnum != 0 && s->iccnum == s->iccread) {
2849 AVFrameSideData *sd;
2850 2 size_t offset = 0;
2851 2 int total_size = 0;
2852 int i;
2853
2854 /* Sum size of all parts. */
2855
2/2
✓ Branch 0 taken 2 times.
✓ Branch 1 taken 2 times.
 4 for (i = 0; i < s->iccnum; i++)
2856 2 total_size += s->iccentries[i].length;
2857
2858 2 sd = av_frame_new_side_data(frame, AV_FRAME_DATA_ICC_PROFILE, total_size);
2859
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2 times.
 2 if (!sd) {
2860 ✗ av_log(s->avctx, AV_LOG_ERROR, "Could not allocate frame side data\n");
2861 ✗ return AVERROR(ENOMEM);
2862 }
2863
2864 /* Reassemble the parts, which are now in-order. */
2865
2/2
✓ Branch 0 taken 2 times.
✓ Branch 1 taken 2 times.
 4 for (i = 0; i < s->iccnum; i++) {
2866 2 memcpy(sd->data + offset, s->iccentries[i].data, s->iccentries[i].length);
2867 2 offset += s->iccentries[i].length;
2868 }
2869 }
2870
2871
2/2
✓ Branch 1 taken 1 times.
✓ Branch 2 taken 2480 times.
 2481 if (e = av_dict_get(s->exif_metadata, "Orientation", e, AV_DICT_IGNORE_SUFFIX)) {
2872 1 char *value = e->value + strspn(e->value, " \n\t\r"), *endptr;
2873 1 int orientation = strtol(value, &endptr, 0);
2874
2875
1/2
✓ Branch 0 taken 1 times.
✗ Branch 1 not taken.
 1 if (!*endptr) {
2876 1 AVFrameSideData *sd = NULL;
2877
2878
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 1 if (orientation >= 2 && orientation <= 8) {
2879 int32_t *matrix;
2880
2881 ✗ sd = av_frame_new_side_data(frame, AV_FRAME_DATA_DISPLAYMATRIX, sizeof(int32_t) * 9);
2882 ✗ if (!sd) {
2883 ✗ av_log(s->avctx, AV_LOG_ERROR, "Could not allocate frame side data\n");
2884 ✗ return AVERROR(ENOMEM);
2885 }
2886
2887 ✗ matrix = (int32_t *)sd->data;
2888
2889 ✗ switch (orientation) {
2890 ✗ case 2:
2891 ✗ av_display_rotation_set(matrix, 0.0);
2892 ✗ av_display_matrix_flip(matrix, 1, 0);
2893 ✗ break;
2894 ✗ case 3:
2895 ✗ av_display_rotation_set(matrix, 180.0);
2896 ✗ break;
2897 ✗ case 4:
2898 ✗ av_display_rotation_set(matrix, 180.0);
2899 ✗ av_display_matrix_flip(matrix, 1, 0);
2900 ✗ break;
2901 ✗ case 5:
2902 ✗ av_display_rotation_set(matrix, 90.0);
2903 ✗ av_display_matrix_flip(matrix, 1, 0);
2904 ✗ break;
2905 ✗ case 6:
2906 ✗ av_display_rotation_set(matrix, 90.0);
2907 ✗ break;
2908 ✗ case 7:
2909 ✗ av_display_rotation_set(matrix, -90.0);
2910 ✗ av_display_matrix_flip(matrix, 1, 0);
2911 ✗ break;
2912 ✗ case 8:
2913 ✗ av_display_rotation_set(matrix, -90.0);
2914 ✗ break;
2915 ✗ default:
2916 ✗ av_assert0(0);
2917 }
2918 }
2919 }
2920 }
2921
2922 2481 av_dict_copy(&frame->metadata, s->exif_metadata, 0);
2923 2481 av_dict_free(&s->exif_metadata);
2924
2925
2/2
✓ Branch 0 taken 3 times.
✓ Branch 1 taken 2478 times.
 2481 if (avctx->codec_id == AV_CODEC_ID_SMVJPEG) {
2926 3 ret = smv_process_frame(avctx, frame);
2927
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 3 times.
 3 if (ret < 0) {
2928 ✗ av_frame_unref(frame);
2929 ✗ return ret;
2930 }
2931 }
2932
1/2
✓ Branch 0 taken 2481 times.
✗ Branch 1 not taken.
 2481 if ((avctx->codec_tag == MKTAG('A', 'V', 'R', 'n') ||
2933
2/2
✓ Branch 0 taken 10 times.
✓ Branch 1 taken 2471 times.
 2481 avctx->codec_tag == MKTAG('A', 'V', 'D', 'J')) &&
2934
1/2
✓ Branch 0 taken 10 times.
✗ Branch 1 not taken.
 10 avctx->coded_height > s->orig_height) {
2935 10 frame->height = AV_CEIL_RSHIFT(avctx->coded_height, avctx->lowres);
2936 10 frame->crop_top = frame->height - avctx->height;
2937 }
2938
2939 2481 ret = 0;
2940
2941 2556 the_end_no_picture:
2942 2556 av_log(avctx, AV_LOG_DEBUG, "decode frame unused %"PTRDIFF_SPECIFIER" bytes\n",
2943 buf_end - buf_ptr);
2944
2945 2556 return ret;
2946 }
2947
2948 /* mxpeg may call the following function (with a blank MJpegDecodeContext)
2949 * even without having called ff_mjpeg_decode_init(). */
2950 179 av_cold int ff_mjpeg_decode_end(AVCodecContext *avctx)
2951 {
2952 179 MJpegDecodeContext *s = avctx->priv_data;
2953 int i, j;
2954
2955
3/8
✓ Branch 0 taken 4 times.
✓ Branch 1 taken 175 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 4 times.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
✗ Branch 6 not taken.
✗ Branch 7 not taken.
 179 if (s->interlaced && s->bottom_field == !s->interlace_polarity && s->got_picture && !avctx->frame_number) {
2956 ✗ av_log(avctx, AV_LOG_INFO, "Single field\n");
2957 }
2958
2959
2/2
✓ Branch 0 taken 177 times.
✓ Branch 1 taken 2 times.
 179 if (s->picture) {
2960 177 av_frame_free(&s->picture);
2961 177 s->picture_ptr = NULL;
2962
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2 times.
 2 } else if (s->picture_ptr)
2963 ✗ av_frame_unref(s->picture_ptr);
2964
2965 179 av_frame_free(&s->smv_frame);
2966
2967 179 av_freep(&s->buffer);
2968 179 av_freep(&s->stereo3d);
2969 179 av_freep(&s->ljpeg_buffer);
2970 179 s->ljpeg_buffer_size = 0;
2971
2972
2/2
✓ Branch 0 taken 537 times.
✓ Branch 1 taken 179 times.
 716 for (i = 0; i < 3; i++) {
2973
2/2
✓ Branch 0 taken 2148 times.
✓ Branch 1 taken 537 times.
 2685 for (j = 0; j < 4; j++)
2974 2148 ff_free_vlc(&s->vlcs[i][j]);
2975 }
2976
2/2
✓ Branch 0 taken 716 times.
✓ Branch 1 taken 179 times.
 895 for (i = 0; i < MAX_COMPONENTS; i++) {
2977 716 av_freep(&s->blocks[i]);
2978 716 av_freep(&s->last_nnz[i]);
2979 }
2980 179 av_dict_free(&s->exif_metadata);
2981
2982 179 reset_icc_profile(s);
2983
2984 179 av_freep(&s->hwaccel_picture_private);
2985 179 av_freep(&s->jls_state);
2986
2987 179 return 0;
2988 }
2989
2990 4 static void decode_flush(AVCodecContext *avctx)
2991 {
2992 4 MJpegDecodeContext *s = avctx->priv_data;
2993 4 s->got_picture = 0;
2994
2995 4 s->smv_next_frame = 0;
2996 4 av_frame_unref(s->smv_frame);
2997 4 }
2998
2999 #if CONFIG_MJPEG_DECODER
3000 #define OFFSET(x) offsetof(MJpegDecodeContext, x)
3001 #define VD AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_DECODING_PARAM
3002 static const AVOption options[] = {
3003 { "extern_huff", "Use external huffman table.",
3004 OFFSET(extern_huff), AV_OPT_TYPE_BOOL, { .i64 = 0 }, 0, 1, VD },
3005 { NULL },
3006 };
3007
3008 static const AVClass mjpegdec_class = {
3009 .class_name = "MJPEG decoder",
3010 .item_name = av_default_item_name,
3011 .option = options,
3012 .version = LIBAVUTIL_VERSION_INT,
3013 };
3014
3015 const FFCodec ff_mjpeg_decoder = {
3016 .p.name = "mjpeg",
3017 .p.long_name = NULL_IF_CONFIG_SMALL("MJPEG (Motion JPEG)"),
3018 .p.type = AVMEDIA_TYPE_VIDEO,
3019 .p.id = AV_CODEC_ID_MJPEG,
3020 .priv_data_size = sizeof(MJpegDecodeContext),
3021 .init = ff_mjpeg_decode_init,
3022 .close = ff_mjpeg_decode_end,
3023 FF_CODEC_RECEIVE_FRAME_CB(ff_mjpeg_receive_frame),
3024 .flush = decode_flush,
3025 .p.capabilities = AV_CODEC_CAP_DR1,
3026 .p.max_lowres = 3,
3027 .p.priv_class = &mjpegdec_class,
3028 .p.profiles = NULL_IF_CONFIG_SMALL(ff_mjpeg_profiles),
3029 .caps_internal = FF_CODEC_CAP_INIT_THREADSAFE | FF_CODEC_CAP_INIT_CLEANUP |
3030 FF_CODEC_CAP_SKIP_FRAME_FILL_PARAM | FF_CODEC_CAP_SETS_PKT_DTS,
3031 .hw_configs = (const AVCodecHWConfigInternal *const []) {
3032 #if CONFIG_MJPEG_NVDEC_HWACCEL
3033 HWACCEL_NVDEC(mjpeg),
3034 #endif
3035 #if CONFIG_MJPEG_VAAPI_HWACCEL
3036 HWACCEL_VAAPI(mjpeg),
3037 #endif
3038 NULL
3039 },
3040 };
3041 #endif
3042 #if CONFIG_THP_DECODER
3043 const FFCodec ff_thp_decoder = {
3044 .p.name = "thp",
3045 .p.long_name = NULL_IF_CONFIG_SMALL("Nintendo Gamecube THP video"),
3046 .p.type = AVMEDIA_TYPE_VIDEO,
3047 .p.id = AV_CODEC_ID_THP,
3048 .priv_data_size = sizeof(MJpegDecodeContext),
3049 .init = ff_mjpeg_decode_init,
3050 .close = ff_mjpeg_decode_end,
3051 FF_CODEC_RECEIVE_FRAME_CB(ff_mjpeg_receive_frame),
3052 .flush = decode_flush,
3053 .p.capabilities = AV_CODEC_CAP_DR1,
3054 .p.max_lowres = 3,
3055 .caps_internal = FF_CODEC_CAP_INIT_THREADSAFE | FF_CODEC_CAP_INIT_CLEANUP |
3056 FF_CODEC_CAP_SETS_PKT_DTS,
3057 };
3058 #endif
3059
3060 #if CONFIG_SMVJPEG_DECODER
3061 const FFCodec ff_smvjpeg_decoder = {
3062 .p.name = "smvjpeg",
3063 .p.long_name = NULL_IF_CONFIG_SMALL("SMV JPEG"),
3064 .p.type = AVMEDIA_TYPE_VIDEO,
3065 .p.id = AV_CODEC_ID_SMVJPEG,
3066 .priv_data_size = sizeof(MJpegDecodeContext),
3067 .init = ff_mjpeg_decode_init,
3068 .close = ff_mjpeg_decode_end,
3069 FF_CODEC_RECEIVE_FRAME_CB(ff_mjpeg_receive_frame),
3070 .flush = decode_flush,
3071 .p.capabilities = AV_CODEC_CAP_DR1,
3072 .caps_internal = FF_CODEC_CAP_INIT_THREADSAFE | FF_CODEC_CAP_EXPORTS_CROPPING |
3073 FF_CODEC_CAP_SETS_PKT_DTS | FF_CODEC_CAP_INIT_CLEANUP,
3074 };
3075 #endif
3076