FFmpeg coverage

Directory: ../../../ffmpeg/
File: src/libavcodec/mpeg12dec.c
Date: 2026-02-12 15:45:49
Exec Total Coverage
Lines: 1167 1612 72.4%
Functions: 30 33 90.9%
Branches: 666 1117 59.6%
Line Branch Exec Source
1 /*
2 * MPEG-1/2 decoder
3 * Copyright (c) 2000, 2001 Fabrice Bellard
4 * Copyright (c) 2002-2013 Michael Niedermayer <michaelni@gmx.at>
5 *
6 * This file is part of FFmpeg.
7 *
8 * FFmpeg is free software; you can redistribute it and/or
9 * modify it under the terms of the GNU Lesser General Public
10 * License as published by the Free Software Foundation; either
11 * version 2.1 of the License, or (at your option) any later version.
12 *
13 * FFmpeg is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16 * Lesser General Public License for more details.
17 *
18 * You should have received a copy of the GNU Lesser General Public
19 * License along with FFmpeg; if not, write to the Free Software
20 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
21 */
22
23 /**
24 * @file
25 * MPEG-1/2 decoder
26 */
27
28 #include "config_components.h"
29
30 #define UNCHECKED_BITSTREAM_READER 1
31 #include <inttypes.h>
32 #include <stdatomic.h>
33
34 #include "libavutil/attributes.h"
35 #include "libavutil/emms.h"
36 #include "libavutil/imgutils.h"
37 #include "libavutil/internal.h"
38 #include "libavutil/mem_internal.h"
39 #include "libavutil/reverse.h"
40 #include "libavutil/stereo3d.h"
41 #include "libavutil/timecode.h"
42
43 #include "avcodec.h"
44 #include "codec_internal.h"
45 #include "decode.h"
46 #include "error_resilience.h"
47 #include "get_bits.h"
48 #include "hwaccel_internal.h"
49 #include "hwconfig.h"
50 #include "idctdsp.h"
51 #include "mpeg_er.h"
52 #include "mpeg12.h"
53 #include "mpeg12data.h"
54 #include "mpeg12dec.h"
55 #include "mpegutils.h"
56 #include "mpegvideo.h"
57 #include "mpegvideodata.h"
58 #include "mpegvideodec.h"
59 #include "profiles.h"
60 #include "startcode.h"
61
62 #define A53_MAX_CC_COUNT 2000
63
64 enum Mpeg2ClosedCaptionsFormat {
65 CC_FORMAT_AUTO,
66 CC_FORMAT_A53_PART4,
67 CC_FORMAT_SCTE20,
68 CC_FORMAT_DVD,
69 CC_FORMAT_DISH
70 };
71
72 typedef struct Mpeg12SliceContext {
73 MPVContext c;
74 GetBitContext gb;
75
76 int last_dc[3]; ///< last DC values
77
78 DECLARE_ALIGNED_32(int16_t, block)[12][64];
79 } Mpeg12SliceContext;
80
81 typedef struct Mpeg1Context {
82 Mpeg12SliceContext slice;
83 AVPanScan pan_scan; /* some temporary storage for the panscan */
84 enum AVStereo3DType stereo3d_type;
85 int has_stereo3d;
86 AVBufferRef *a53_buf_ref;
87 enum Mpeg2ClosedCaptionsFormat cc_format;
88 uint8_t afd;
89 int has_afd;
90 int slice_count;
91 unsigned aspect_ratio_info;
92 int save_progressive_seq, save_chroma_format;
93 AVRational frame_rate_ext; /* MPEG-2 specific framerate modificator */
94 unsigned frame_rate_index;
95 int sync; /* Did we reach a sync point like a GOP/SEQ/KEYFrame? */
96 int closed_gop;
97 int tmpgexs;
98 int first_slice;
99 int extradata_decoded;
100 int vbv_delay;
101 int64_t bit_rate;
102 int64_t timecode_frame_start; /*< GOP timecode frame start number, in non drop frame format */
103 } Mpeg1Context;
104
105 /* as H.263, but only 17 codes */
106 5942342 static int mpeg_decode_motion(Mpeg12SliceContext *const s, int fcode, int pred)
107 {
108 int code, sign, val, shift;
109
110 5942342 code = get_vlc2(&s->gb, ff_mv_vlc, MV_VLC_BITS, 2);
111
2/2
✓ Branch 0 taken 2410487 times.
✓ Branch 1 taken 3531855 times.
 5942342 if (code == 0)
112 2410487 return pred;
113
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 3531855 times.
 3531855 if (code < 0)
114 ✗ return 0xffff;
115
116 3531855 sign = get_bits1(&s->gb);
117 3531855 shift = fcode - 1;
118 3531855 val = code;
119
2/2
✓ Branch 0 taken 3192195 times.
✓ Branch 1 taken 339660 times.
 3531855 if (shift) {
120 3192195 val = (val - 1) << shift;
121 3192195 val |= get_bits(&s->gb, shift);
122 3192195 val++;
123 }
124
2/2
✓ Branch 0 taken 1823461 times.
✓ Branch 1 taken 1708394 times.
 3531855 if (sign)
125 1823461 val = -val;
126 3531855 val += pred;
127
128 /* modulo decoding */
129 3531855 return sign_extend(val, 5 + shift);
130 }
131
132 #define MAX_INDEX (64 - 1)
133 #define check_scantable_index(ctx, x) \
134 do { \
135 if ((x) > MAX_INDEX) { \
136 av_log(ctx->c.avctx, AV_LOG_ERROR, "ac-tex damaged at %d %d\n", \
137 ctx->c.mb_x, ctx->c.mb_y); \
138 return AVERROR_INVALIDDATA; \
139 } \
140 } while (0)
141
142 403182 static inline int mpeg1_decode_block_inter(Mpeg12SliceContext *const s,
143 int16_t *block, int n)
144 {
145 int level, i, j, run;
146 403182 const uint8_t *const scantable = s->c.intra_scantable.permutated;
147 403182 const uint16_t *quant_matrix = s->c.inter_matrix;
148 403182 const int qscale = s->c.qscale;
149
150 {
151 403182 OPEN_READER(re, &s->gb);
152 403182 i = -1;
153 // special case for first coefficient, no need to add second VLC table
154 403182 UPDATE_CACHE(re, &s->gb);
155
2/2
✓ Branch 0 taken 301257 times.
✓ Branch 1 taken 101925 times.
 403182 if (((int32_t) GET_CACHE(re, &s->gb)) < 0) {
156 101925 level = (3 * qscale * quant_matrix[0]) >> 5;
157 101925 level = (level - 1) | 1;
158
2/2
✓ Branch 0 taken 57008 times.
✓ Branch 1 taken 44917 times.
 101925 if (GET_CACHE(re, &s->gb) & 0x40000000)
159 57008 level = -level;
160 101925 block[0] = level;
161 101925 i++;
162 101925 SKIP_BITS(re, &s->gb, 2);
163
2/2
✓ Branch 0 taken 67516 times.
✓ Branch 1 taken 34409 times.
 101925 if (((int32_t) GET_CACHE(re, &s->gb)) <= (int32_t) 0xBFFFFFFF)
164 34409 goto end;
165 }
166 /* now quantify & encode AC coefficients */
167 for (;;) {
168
2/2
✓ Branch 1 taken 335978 times.
✓ Branch 2 taken 2584676 times.
 2920654 GET_RL_VLC(level, run, re, &s->gb, ff_mpeg1_rl_vlc,
169 TEX_VLC_BITS, 2, 0);
170
171
2/2
✓ Branch 0 taken 2841520 times.
✓ Branch 1 taken 79134 times.
 2920654 if (level != 0) {
172 2841520 i += run;
173
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2841520 times.
 2841520 if (i > MAX_INDEX)
174 ✗ break;
175 2841520 j = scantable[i];
176 2841520 level = ((level * 2 + 1) * qscale * quant_matrix[j]) >> 5;
177 2841520 level = (level - 1) | 1;
178 2841520 level = (level ^ SHOW_SBITS(re, &s->gb, 1)) -
179 2841520 SHOW_SBITS(re, &s->gb, 1);
180 2841520 SKIP_BITS(re, &s->gb, 1);
181 } else {
182 /* escape */
183 79134 run = SHOW_UBITS(re, &s->gb, 6) + 1;
184 79134 LAST_SKIP_BITS(re, &s->gb, 6);
185 79134 UPDATE_CACHE(re, &s->gb);
186 79134 level = SHOW_SBITS(re, &s->gb, 8);
187 79134 SKIP_BITS(re, &s->gb, 8);
188
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 79134 times.
 79134 if (level == -128) {
189 ✗ level = SHOW_UBITS(re, &s->gb, 8) - 256;
190 ✗ SKIP_BITS(re, &s->gb, 8);
191
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 79134 times.
 79134 } else if (level == 0) {
192 ✗ level = SHOW_UBITS(re, &s->gb, 8);
193 ✗ SKIP_BITS(re, &s->gb, 8);
194 }
195 79134 i += run;
196
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 79134 times.
 79134 if (i > MAX_INDEX)
197 ✗ break;
198 79134 j = scantable[i];
199
2/2
✓ Branch 0 taken 38931 times.
✓ Branch 1 taken 40203 times.
 79134 if (level < 0) {
200 38931 level = -level;
201 38931 level = ((level * 2 + 1) * qscale * quant_matrix[j]) >> 5;
202 38931 level = (level - 1) | 1;
203 38931 level = -level;
204 } else {
205 40203 level = ((level * 2 + 1) * qscale * quant_matrix[j]) >> 5;
206 40203 level = (level - 1) | 1;
207 }
208 }
209
210 2920654 block[j] = level;
211
2/2
✓ Branch 0 taken 368773 times.
✓ Branch 1 taken 2551881 times.
 2920654 if (((int32_t) GET_CACHE(re, &s->gb)) <= (int32_t) 0xBFFFFFFF)
212 368773 break;
213 2551881 UPDATE_CACHE(re, &s->gb);
214 }
215 403182 end:
216 403182 LAST_SKIP_BITS(re, &s->gb, 2);
217 403182 CLOSE_READER(re, &s->gb);
218 }
219
220
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 403182 times.
 403182 check_scantable_index(s, i);
221
222 403182 s->c.block_last_index[n] = i;
223 403182 return 0;
224 }
225
226 5498236 static inline int mpeg2_decode_block_non_intra(Mpeg12SliceContext *const s,
227 int16_t *block, int n)
228 {
229 int level, i, j, run;
230 5498236 const uint8_t *const scantable = s->c.intra_scantable.permutated;
231 const uint16_t *quant_matrix;
232 5498236 const int qscale = s->c.qscale;
233 int mismatch;
234
235 5498236 mismatch = 1;
236
237 {
238 5498236 OPEN_READER(re, &s->gb);
239 5498236 i = -1;
240
2/2
✓ Branch 0 taken 4143231 times.
✓ Branch 1 taken 1355005 times.
 5498236 if (n < 4)
241 4143231 quant_matrix = s->c.inter_matrix;
242 else
243 1355005 quant_matrix = s->c.chroma_inter_matrix;
244
245 // Special case for first coefficient, no need to add second VLC table.
246 5498236 UPDATE_CACHE(re, &s->gb);
247
2/2
✓ Branch 0 taken 3463934 times.
✓ Branch 1 taken 2034302 times.
 5498236 if (((int32_t) GET_CACHE(re, &s->gb)) < 0) {
248 2034302 level = (3 * qscale * quant_matrix[0]) >> 5;
249
2/2
✓ Branch 0 taken 1298107 times.
✓ Branch 1 taken 736195 times.
 2034302 if (GET_CACHE(re, &s->gb) & 0x40000000)
250 1298107 level = -level;
251 2034302 block[0] = level;
252 2034302 mismatch ^= level;
253 2034302 i++;
254 2034302 SKIP_BITS(re, &s->gb, 2);
255
2/2
✓ Branch 0 taken 1050418 times.
✓ Branch 1 taken 983884 times.
 2034302 if (((int32_t) GET_CACHE(re, &s->gb)) <= (int32_t) 0xBFFFFFFF)
256 983884 goto end;
257 }
258
259 /* now quantify & encode AC coefficients */
260 for (;;) {
261
2/2
✓ Branch 1 taken 1703388 times.
✓ Branch 2 taken 17316034 times.
 19019422 GET_RL_VLC(level, run, re, &s->gb, ff_mpeg1_rl_vlc,
262 TEX_VLC_BITS, 2, 0);
263
264
2/2
✓ Branch 0 taken 18734513 times.
✓ Branch 1 taken 284909 times.
 19019422 if (level != 0) {
265 18734513 i += run;
266
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 18734513 times.
 18734513 if (i > MAX_INDEX)
267 ✗ break;
268 18734513 j = scantable[i];
269 18734513 level = ((level * 2 + 1) * qscale * quant_matrix[j]) >> 5;
270 18734513 level = (level ^ SHOW_SBITS(re, &s->gb, 1)) -
271 18734513 SHOW_SBITS(re, &s->gb, 1);
272 18734513 SKIP_BITS(re, &s->gb, 1);
273 } else {
274 /* escape */
275 284909 run = SHOW_UBITS(re, &s->gb, 6) + 1;
276 284909 LAST_SKIP_BITS(re, &s->gb, 6);
277 284909 UPDATE_CACHE(re, &s->gb);
278 284909 level = SHOW_SBITS(re, &s->gb, 12);
279 284909 SKIP_BITS(re, &s->gb, 12);
280
281 284909 i += run;
282
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 284909 times.
 284909 if (i > MAX_INDEX)
283 ✗ break;
284 284909 j = scantable[i];
285
2/2
✓ Branch 0 taken 138482 times.
✓ Branch 1 taken 146427 times.
 284909 if (level < 0) {
286 138482 level = ((-level * 2 + 1) * qscale * quant_matrix[j]) >> 5;
287 138482 level = -level;
288 } else {
289 146427 level = ((level * 2 + 1) * qscale * quant_matrix[j]) >> 5;
290 }
291 }
292
293 19019422 mismatch ^= level;
294 19019422 block[j] = level;
295
2/2
✓ Branch 0 taken 4514352 times.
✓ Branch 1 taken 14505070 times.
 19019422 if (((int32_t) GET_CACHE(re, &s->gb)) <= (int32_t) 0xBFFFFFFF)
296 4514352 break;
297 14505070 UPDATE_CACHE(re, &s->gb);
298 }
299 5498236 end:
300 5498236 LAST_SKIP_BITS(re, &s->gb, 2);
301 5498236 CLOSE_READER(re, &s->gb);
302 }
303 5498236 block[63] ^= (mismatch & 1);
304
305
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 5498236 times.
 5498236 check_scantable_index(s, i);
306
307 5498236 s->c.block_last_index[n] = i;
308 5498236 return 0;
309 }
310
311 1975516 static inline int mpeg2_decode_block_intra(Mpeg12SliceContext *const s,
312 int16_t *block, int n)
313 {
314 int level, dc, diff, i, j, run;
315 int component;
316 const RL_VLC_ELEM *rl_vlc;
317 1975516 const uint8_t *const scantable = s->c.intra_scantable.permutated;
318 const uint16_t *quant_matrix;
319 1975516 const int qscale = s->c.qscale;
320 int mismatch;
321
322 /* DC coefficient */
323
2/2
✓ Branch 0 taken 1244806 times.
✓ Branch 1 taken 730710 times.
 1975516 if (n < 4) {
324 1244806 quant_matrix = s->c.intra_matrix;
325 1244806 component = 0;
326 } else {
327 730710 quant_matrix = s->c.chroma_intra_matrix;
328 730710 component = (n & 1) + 1;
329 }
330 1975516 diff = decode_dc(&s->gb, component);
331 1975516 dc = s->last_dc[component];
332 1975516 dc += diff;
333 1975516 s->last_dc[component] = dc;
334 1975516 block[0] = dc * (1 << (3 - s->c.intra_dc_precision));
335 ff_tlog(s->c.avctx, "dc=%d\n", block[0]);
336 1975516 mismatch = block[0] ^ 1;
337 1975516 i = 0;
338
2/2
✓ Branch 0 taken 1392706 times.
✓ Branch 1 taken 582810 times.
 1975516 if (s->c.intra_vlc_format)
339 1392706 rl_vlc = ff_mpeg2_rl_vlc;
340 else
341 582810 rl_vlc = ff_mpeg1_rl_vlc;
342
343 {
344 1975516 OPEN_READER(re, &s->gb);
345 /* now quantify & encode AC coefficients */
346 for (;;) {
347 27173814 UPDATE_CACHE(re, &s->gb);
348
2/2
✓ Branch 1 taken 749188 times.
✓ Branch 2 taken 13825477 times.
 14574665 GET_RL_VLC(level, run, re, &s->gb, rl_vlc,
349 TEX_VLC_BITS, 2, 0);
350
351
2/2
✓ Branch 0 taken 1975505 times.
✓ Branch 1 taken 12599160 times.
 14574665 if (level == 127) {
352 1975505 break;
353
2/2
✓ Branch 0 taken 12457016 times.
✓ Branch 1 taken 142144 times.
 12599160 } else if (level != 0) {
354 12457016 i += run;
355
2/2
✓ Branch 0 taken 11 times.
✓ Branch 1 taken 12457005 times.
 12457016 if (i > MAX_INDEX)
356 11 break;
357 12457005 j = scantable[i];
358 12457005 level = (level * qscale * quant_matrix[j]) >> 4;
359 12457005 level = (level ^ SHOW_SBITS(re, &s->gb, 1)) -
360 12457005 SHOW_SBITS(re, &s->gb, 1);
361 12457005 LAST_SKIP_BITS(re, &s->gb, 1);
362 } else {
363 /* escape */
364 142144 run = SHOW_UBITS(re, &s->gb, 6) + 1;
365 142144 SKIP_BITS(re, &s->gb, 6);
366 142144 level = SHOW_SBITS(re, &s->gb, 12);
367 142144 LAST_SKIP_BITS(re, &s->gb, 12);
368 142144 i += run;
369
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 142144 times.
 142144 if (i > MAX_INDEX)
370 ✗ break;
371 142144 j = scantable[i];
372
2/2
✓ Branch 0 taken 63814 times.
✓ Branch 1 taken 78330 times.
 142144 if (level < 0) {
373 63814 level = (-level * qscale * quant_matrix[j]) >> 4;
374 63814 level = -level;
375 } else {
376 78330 level = (level * qscale * quant_matrix[j]) >> 4;
377 }
378 }
379
380 12599149 mismatch ^= level;
381 12599149 block[j] = level;
382 }
383 1975516 CLOSE_READER(re, &s->gb);
384 }
385 1975516 block[63] ^= mismatch & 1;
386
387
2/2
✓ Branch 0 taken 11 times.
✓ Branch 1 taken 1975505 times.
 1975516 check_scantable_index(s, i);
388
389 1975505 return 0;
390 }
391
392 46436 static inline int get_dmv(Mpeg12SliceContext *const s)
393 {
394
2/2
✓ Branch 1 taken 43282 times.
✓ Branch 2 taken 3154 times.
 46436 if (get_bits1(&s->gb))
395 43282 return 1 - (get_bits1(&s->gb) << 1);
396 else
397 3154 return 0;
398 }
399
400 /* motion type (for MPEG-2) */
401 #define MT_FIELD 1
402 #define MT_FRAME 2
403 #define MT_16X8 2
404 #define MT_DMV 3
405
406 2914172 static int mpeg_decode_mb(Mpeg12SliceContext *const s, int *mb_skip_run)
407 {
408 int i, j, k, cbp, val, mb_type, motion_type;
409 2914172 const int mb_block_count = 4 + (1 << s->c.chroma_format);
410 int ret;
411
412 ff_tlog(s->c.avctx, "decode_mb: x=%d y=%d\n", s->c.mb_x, s->c.mb_y);
413
414 av_assert2(s->c.mb_skipped == 0);
415
416
2/2
✓ Branch 0 taken 266083 times.
✓ Branch 1 taken 2648089 times.
 2914172 if ((*mb_skip_run)-- != 0) {
417
2/2
✓ Branch 0 taken 80358 times.
✓ Branch 1 taken 185725 times.
 266083 if (s->c.pict_type == AV_PICTURE_TYPE_P) {
418 80358 s->c.mb_skipped = 1;
419 80358 s->c.cur_pic.mb_type[s->c.mb_x + s->c.mb_y * s->c.mb_stride] =
420 MB_TYPE_SKIP | MB_TYPE_FORWARD_MV | MB_TYPE_16x16;
421 } else {
422 int mb_type;
423
424
1/2
✓ Branch 0 taken 185725 times.
✗ Branch 1 not taken.
 185725 if (s->c.mb_x)
425 185725 mb_type = s->c.cur_pic.mb_type[s->c.mb_x + s->c.mb_y * s->c.mb_stride - 1];
426 else
427 // FIXME not sure if this is allowed in MPEG at all
428 ✗ mb_type = s->c.cur_pic.mb_type[s->c.mb_width + (s->c.mb_y - 1) * s->c.mb_stride - 1];
429
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 185725 times.
 185725 if (IS_INTRA(mb_type)) {
430 ✗ av_log(s->c.avctx, AV_LOG_ERROR, "skip with previntra\n");
431 ✗ return AVERROR_INVALIDDATA;
432 }
433 185725 s->c.cur_pic.mb_type[s->c.mb_x + s->c.mb_y * s->c.mb_stride] =
434 185725 mb_type | MB_TYPE_SKIP;
435
436
2/2
✓ Branch 0 taken 138112 times.
✓ Branch 1 taken 47613 times.
 185725 if ((s->c.mv[0][0][0] | s->c.mv[0][0][1] | s->c.mv[1][0][0] | s->c.mv[1][0][1]) == 0)
437 138112 s->c.mb_skipped = 1;
438 }
439
440 266083 return 0;
441 }
442
443
3/3
✓ Branch 0 taken 248238 times.
✓ Branch 1 taken 1440560 times.
✓ Branch 2 taken 959291 times.
 2648089 switch (s->c.pict_type) {
444 248238 default:
445 case AV_PICTURE_TYPE_I:
446
2/2
✓ Branch 1 taken 20828 times.
✓ Branch 2 taken 227410 times.
 248238 if (get_bits1(&s->gb) == 0) {
447
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 20828 times.
 20828 if (get_bits1(&s->gb) == 0) {
448 ✗ av_log(s->c.avctx, AV_LOG_ERROR,
449 "Invalid mb type in I-frame at %d %d\n",
450 s->c.mb_x, s->c.mb_y);
451 ✗ return AVERROR_INVALIDDATA;
452 }
453 20828 mb_type = MB_TYPE_QUANT | MB_TYPE_INTRA;
454 } else {
455 227410 mb_type = MB_TYPE_INTRA;
456 }
457 248238 break;
458 1440560 case AV_PICTURE_TYPE_P:
459 1440560 mb_type = get_vlc2(&s->gb, ff_mb_ptype_vlc, MB_PTYPE_VLC_BITS, 1);
460
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1440560 times.
 1440560 if (mb_type < 0) {
461 ✗ av_log(s->c.avctx, AV_LOG_ERROR,
462 "Invalid mb type in P-frame at %d %d\n", s->c.mb_x, s->c.mb_y);
463 ✗ return AVERROR_INVALIDDATA;
464 }
465 1440560 break;
466 959291 case AV_PICTURE_TYPE_B:
467 959291 mb_type = get_vlc2(&s->gb, ff_mb_btype_vlc, MB_BTYPE_VLC_BITS, 1);
468
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 959291 times.
 959291 if (mb_type < 0) {
469 ✗ av_log(s->c.avctx, AV_LOG_ERROR,
470 "Invalid mb type in B-frame at %d %d\n", s->c.mb_x, s->c.mb_y);
471 ✗ return AVERROR_INVALIDDATA;
472 }
473 959291 break;
474 }
475 ff_tlog(s->c.avctx, "mb_type=%x\n", mb_type);
476 // motion_type = 0; /* avoid warning */
477
2/2
✓ Branch 0 taken 332211 times.
✓ Branch 1 taken 2315878 times.
 2648089 if (IS_INTRA(mb_type)) {
478 332211 s->c.bdsp.clear_blocks(s->block[0]);
479
480
2/2
✓ Branch 0 taken 54159 times.
✓ Branch 1 taken 278052 times.
 332211 if (!s->c.chroma_y_shift)
481 54159 s->c.bdsp.clear_blocks(s->block[6]);
482
483 /* compute DCT type */
484 // FIXME: add an interlaced_dct coded var?
485
2/2
✓ Branch 0 taken 270635 times.
✓ Branch 1 taken 61576 times.
 332211 if (s->c.picture_structure == PICT_FRAME &&
486
2/2
✓ Branch 0 taken 127334 times.
✓ Branch 1 taken 143301 times.
 270635 !s->c.frame_pred_frame_dct)
487 127334 s->c.interlaced_dct = get_bits1(&s->gb);
488
489
2/2
✓ Branch 0 taken 29500 times.
✓ Branch 1 taken 302711 times.
 332211 if (IS_QUANT(mb_type))
490 29500 s->c.qscale = mpeg_get_qscale(&s->gb, s->c.q_scale_type);
491
492
2/2
✓ Branch 0 taken 1356 times.
✓ Branch 1 taken 330855 times.
 332211 if (s->c.concealment_motion_vectors) {
493 /* just parse them */
494
2/2
✓ Branch 0 taken 1350 times.
✓ Branch 1 taken 6 times.
 1356 if (s->c.picture_structure != PICT_FRAME)
495 1350 skip_bits1(&s->gb); /* field select */
496
497 1356 s->c.mv[0][0][0] =
498 1356 s->c.last_mv[0][0][0] =
499 1356 s->c.last_mv[0][1][0] = mpeg_decode_motion(s, s->c.mpeg_f_code[0][0],
500 s->c.last_mv[0][0][0]);
501 1356 s->c.mv[0][0][1] =
502 1356 s->c.last_mv[0][0][1] =
503 1356 s->c.last_mv[0][1][1] = mpeg_decode_motion(s, s->c.mpeg_f_code[0][1],
504 s->c.last_mv[0][0][1]);
505
506 1356 check_marker(s->c.avctx, &s->gb, "after concealment_motion_vectors");
507 } else {
508 /* reset mv prediction */
509 330855 memset(s->c.last_mv, 0, sizeof(s->c.last_mv));
510 }
511 332211 s->c.mb_intra = 1;
512
513
2/2
✓ Branch 0 taken 311207 times.
✓ Branch 1 taken 21004 times.
 332211 if (s->c.codec_id == AV_CODEC_ID_MPEG2VIDEO) {
514
2/2
✓ Branch 0 taken 1975516 times.
✓ Branch 1 taken 311196 times.
 2286712 for (i = 0; i < mb_block_count; i++)
515
2/2
✓ Branch 1 taken 11 times.
✓ Branch 2 taken 1975505 times.
 1975516 if ((ret = mpeg2_decode_block_intra(s, s->block[i], i)) < 0)
516 11 return ret;
517 } else {
518
2/2
✓ Branch 0 taken 126024 times.
✓ Branch 1 taken 21004 times.
 147028 for (i = 0; i < 6; i++) {
519 126024 ret = ff_mpeg1_decode_block_intra(&s->gb,
520 126024 s->c.intra_matrix,
521 126024 s->c.intra_scantable.permutated,
522 126024 s->last_dc, s->block[i],
523 i, s->c.qscale);
524
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 126024 times.
 126024 if (ret < 0) {
525 ✗ av_log(s->c.avctx, AV_LOG_ERROR, "ac-tex damaged at %d %d\n",
526 s->c.mb_x, s->c.mb_y);
527 ✗ return ret;
528 }
529 }
530 }
531 } else {
532
2/2
✓ Branch 0 taken 244576 times.
✓ Branch 1 taken 2071302 times.
 2315878 if (mb_type & MB_TYPE_ZERO_MV) {
533 av_assert2(mb_type & MB_TYPE_CBP);
534
535 244576 s->c.mv_dir = MV_DIR_FORWARD;
536
2/2
✓ Branch 0 taken 111840 times.
✓ Branch 1 taken 132736 times.
 244576 if (s->c.picture_structure == PICT_FRAME) {
537
1/2
✓ Branch 0 taken 111840 times.
✗ Branch 1 not taken.
 111840 if (s->c.picture_structure == PICT_FRAME
538
2/2
✓ Branch 0 taken 77272 times.
✓ Branch 1 taken 34568 times.
 111840 && !s->c.frame_pred_frame_dct)
539 77272 s->c.interlaced_dct = get_bits1(&s->gb);
540 111840 s->c.mv_type = MV_TYPE_16X16;
541 } else {
542 132736 s->c.mv_type = MV_TYPE_FIELD;
543 132736 mb_type |= MB_TYPE_INTERLACED;
544 132736 s->c.field_select[0][0] = s->c.picture_structure - 1;
545 }
546
547
2/2
✓ Branch 0 taken 63988 times.
✓ Branch 1 taken 180588 times.
 244576 if (IS_QUANT(mb_type))
548 63988 s->c.qscale = mpeg_get_qscale(&s->gb, s->c.q_scale_type);
549
550 244576 s->c.last_mv[0][0][0] = 0;
551 244576 s->c.last_mv[0][0][1] = 0;
552 244576 s->c.last_mv[0][1][0] = 0;
553 244576 s->c.last_mv[0][1][1] = 0;
554 244576 s->c.mv[0][0][0] = 0;
555 244576 s->c.mv[0][0][1] = 0;
556 } else {
557 av_assert2(mb_type & MB_TYPE_BIDIR_MV);
558 // FIXME decide if MBs in field pictures are MB_TYPE_INTERLACED
559 /* get additional motion vector type */
560
4/4
✓ Branch 0 taken 1537544 times.
✓ Branch 1 taken 533758 times.
✓ Branch 2 taken 700747 times.
✓ Branch 3 taken 836797 times.
 2071302 if (s->c.picture_structure == PICT_FRAME && s->c.frame_pred_frame_dct) {
561 700747 motion_type = MT_FRAME;
562 } else {
563 1370555 motion_type = get_bits(&s->gb, 2);
564
4/4
✓ Branch 0 taken 836797 times.
✓ Branch 1 taken 533758 times.
✓ Branch 2 taken 600736 times.
✓ Branch 3 taken 236061 times.
 1370555 if (s->c.picture_structure == PICT_FRAME && HAS_CBP(mb_type))
565 600736 s->c.interlaced_dct = get_bits1(&s->gb);
566 }
567
568
2/2
✓ Branch 0 taken 212474 times.
✓ Branch 1 taken 1858828 times.
 2071302 if (IS_QUANT(mb_type))
569 212474 s->c.qscale = mpeg_get_qscale(&s->gb, s->c.q_scale_type);
570
571 /* motion vectors */
572 2071302 s->c.mv_dir = MB_TYPE_MV_2_MV_DIR(mb_type);
573 ff_tlog(s->c.avctx, "motion_type=%d\n", motion_type);
574
3/4
✓ Branch 0 taken 1750482 times.
✓ Branch 1 taken 297602 times.
✓ Branch 2 taken 23218 times.
✗ Branch 3 not taken.
 2071302 switch (motion_type) {
575 1750482 case MT_FRAME: /* or MT_16X8 */
576
2/2
✓ Branch 0 taken 1439547 times.
✓ Branch 1 taken 310935 times.
 1750482 if (s->c.picture_structure == PICT_FRAME) {
577 1439547 mb_type |= MB_TYPE_16x16;
578 1439547 s->c.mv_type = MV_TYPE_16X16;
579
2/2
✓ Branch 0 taken 2879094 times.
✓ Branch 1 taken 1439547 times.
 4318641 for (i = 0; i < 2; i++) {
580
2/2
✓ Branch 0 taken 1876324 times.
✓ Branch 1 taken 1002770 times.
 2879094 if (HAS_MV(mb_type, i)) {
581 /* MT_FRAME */
582 1876324 s->c.mv[i][0][0] =
583 1876324 s->c.last_mv[i][0][0] =
584 1876324 s->c.last_mv[i][1][0] =
585 1876324 mpeg_decode_motion(s, s->c.mpeg_f_code[i][0],
586 s->c.last_mv[i][0][0]);
587 1876324 s->c.mv[i][0][1] =
588 1876324 s->c.last_mv[i][0][1] =
589 1876324 s->c.last_mv[i][1][1] =
590 1876324 mpeg_decode_motion(s, s->c.mpeg_f_code[i][1],
591 s->c.last_mv[i][0][1]);
592 /* full_pel: only for MPEG-1 */
593
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1876324 times.
 1876324 if (s->c.full_pel[i]) {
594 ✗ s->c.mv[i][0][0] *= 2;
595 ✗ s->c.mv[i][0][1] *= 2;
596 }
597 }
598 }
599 } else {
600 310935 mb_type |= MB_TYPE_16x8 | MB_TYPE_INTERLACED;
601 310935 s->c.mv_type = MV_TYPE_16X8;
602
2/2
✓ Branch 0 taken 621870 times.
✓ Branch 1 taken 310935 times.
 932805 for (i = 0; i < 2; i++) {
603
2/2
✓ Branch 0 taken 312297 times.
✓ Branch 1 taken 309573 times.
 621870 if (HAS_MV(mb_type, i)) {
604 /* MT_16X8 */
605
2/2
✓ Branch 0 taken 624594 times.
✓ Branch 1 taken 312297 times.
 936891 for (j = 0; j < 2; j++) {
606 624594 s->c.field_select[i][j] = get_bits1(&s->gb);
607
2/2
✓ Branch 0 taken 1249188 times.
✓ Branch 1 taken 624594 times.
 1873782 for (k = 0; k < 2; k++) {
608 1249188 val = mpeg_decode_motion(s, s->c.mpeg_f_code[i][k],
609 s->c.last_mv[i][j][k]);
610 1249188 s->c.last_mv[i][j][k] = val;
611 1249188 s->c.mv[i][j][k] = val;
612 }
613 }
614 }
615 }
616 }
617 1750482 break;
618 297602 case MT_FIELD:
619 297602 s->c.mv_type = MV_TYPE_FIELD;
620
2/2
✓ Branch 0 taken 97997 times.
✓ Branch 1 taken 199605 times.
 297602 if (s->c.picture_structure == PICT_FRAME) {
621 97997 mb_type |= MB_TYPE_16x8 | MB_TYPE_INTERLACED;
622
2/2
✓ Branch 0 taken 195994 times.
✓ Branch 1 taken 97997 times.
 293991 for (i = 0; i < 2; i++) {
623
2/2
✓ Branch 0 taken 120700 times.
✓ Branch 1 taken 75294 times.
 195994 if (HAS_MV(mb_type, i)) {
624
2/2
✓ Branch 0 taken 241400 times.
✓ Branch 1 taken 120700 times.
 362100 for (j = 0; j < 2; j++) {
625 241400 s->c.field_select[i][j] = get_bits1(&s->gb);
626 241400 val = mpeg_decode_motion(s, s->c.mpeg_f_code[i][0],
627 s->c.last_mv[i][j][0]);
628 241400 s->c.last_mv[i][j][0] = val;
629 241400 s->c.mv[i][j][0] = val;
630 ff_tlog(s->c.avctx, "fmx=%d\n", val);
631 241400 val = mpeg_decode_motion(s, s->c.mpeg_f_code[i][1],
632 241400 s->c.last_mv[i][j][1] >> 1);
633 241400 s->c.last_mv[i][j][1] = 2 * val;
634 241400 s->c.mv[i][j][1] = val;
635 ff_tlog(s->c.avctx, "fmy=%d\n", val);
636 }
637 }
638 }
639 } else {
640
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 199605 times.
 199605 av_assert0(!s->c.progressive_sequence);
641 199605 mb_type |= MB_TYPE_16x16 | MB_TYPE_INTERLACED;
642
2/2
✓ Branch 0 taken 399210 times.
✓ Branch 1 taken 199605 times.
 598815 for (i = 0; i < 2; i++) {
643
2/2
✓ Branch 0 taken 204279 times.
✓ Branch 1 taken 194931 times.
 399210 if (HAS_MV(mb_type, i)) {
644 204279 s->c.field_select[i][0] = get_bits1(&s->gb);
645
2/2
✓ Branch 0 taken 408558 times.
✓ Branch 1 taken 204279 times.
 612837 for (k = 0; k < 2; k++) {
646 408558 val = mpeg_decode_motion(s, s->c.mpeg_f_code[i][k],
647 s->c.last_mv[i][0][k]);
648 408558 s->c.last_mv[i][0][k] = val;
649 408558 s->c.last_mv[i][1][k] = val;
650 408558 s->c.mv[i][0][k] = val;
651 }
652 }
653 }
654 }
655 297602 break;
656 23218 case MT_DMV:
657
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 23218 times.
 23218 if (s->c.progressive_sequence){
658 ✗ av_log(s->c.avctx, AV_LOG_ERROR, "MT_DMV in progressive_sequence\n");
659 ✗ return AVERROR_INVALIDDATA;
660 }
661 23218 s->c.mv_type = MV_TYPE_DMV;
662
2/2
✓ Branch 0 taken 46436 times.
✓ Branch 1 taken 23218 times.
 69654 for (i = 0; i < 2; i++) {
663
2/2
✓ Branch 0 taken 23218 times.
✓ Branch 1 taken 23218 times.
 46436 if (HAS_MV(mb_type, i)) {
664 int dmx, dmy, mx, my, m;
665 23218 const int my_shift = s->c.picture_structure == PICT_FRAME;
666
667 23218 mx = mpeg_decode_motion(s, s->c.mpeg_f_code[i][0],
668 s->c.last_mv[i][0][0]);
669 23218 s->c.last_mv[i][0][0] = mx;
670 23218 s->c.last_mv[i][1][0] = mx;
671 23218 dmx = get_dmv(s);
672 23218 my = mpeg_decode_motion(s, s->c.mpeg_f_code[i][1],
673 23218 s->c.last_mv[i][0][1] >> my_shift);
674 23218 dmy = get_dmv(s);
675
676
677 23218 s->c.last_mv[i][0][1] = my * (1 << my_shift);
678 23218 s->c.last_mv[i][1][1] = my * (1 << my_shift);
679
680 23218 s->c.mv[i][0][0] = mx;
681 23218 s->c.mv[i][0][1] = my;
682 23218 s->c.mv[i][1][0] = mx; // not used
683 23218 s->c.mv[i][1][1] = my; // not used
684
685
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 23218 times.
 23218 if (s->c.picture_structure == PICT_FRAME) {
686 ✗ mb_type |= MB_TYPE_16x16 | MB_TYPE_INTERLACED;
687
688 // m = 1 + 2 * s->c.top_field_first;
689 ✗ m = s->c.top_field_first ? 1 : 3;
690
691 /* top -> top pred */
692 ✗ s->c.mv[i][2][0] = ((mx * m + (mx > 0)) >> 1) + dmx;
693 ✗ s->c.mv[i][2][1] = ((my * m + (my > 0)) >> 1) + dmy - 1;
694 ✗ m = 4 - m;
695 ✗ s->c.mv[i][3][0] = ((mx * m + (mx > 0)) >> 1) + dmx;
696 ✗ s->c.mv[i][3][1] = ((my * m + (my > 0)) >> 1) + dmy + 1;
697 } else {
698 23218 mb_type |= MB_TYPE_16x16;
699
700 23218 s->c.mv[i][2][0] = ((mx + (mx > 0)) >> 1) + dmx;
701 23218 s->c.mv[i][2][1] = ((my + (my > 0)) >> 1) + dmy;
702
2/2
✓ Branch 0 taken 19874 times.
✓ Branch 1 taken 3344 times.
 23218 if (s->c.picture_structure == PICT_TOP_FIELD)
703 19874 s->c.mv[i][2][1]--;
704 else
705 3344 s->c.mv[i][2][1]++;
706 }
707 }
708 }
709 23218 break;
710 ✗ default:
711 ✗ av_log(s->c.avctx, AV_LOG_ERROR,
712 "00 motion_type at %d %d\n", s->c.mb_x, s->c.mb_y);
713 ✗ return AVERROR_INVALIDDATA;
714 }
715 }
716
717 2315878 s->c.mb_intra = 0;
718 2315878 s->last_dc[0] = s->last_dc[1] = s->last_dc[2] = 128 << s->c.intra_dc_precision;
719
2/2
✓ Branch 0 taken 1842059 times.
✓ Branch 1 taken 473819 times.
 2315878 if (HAS_CBP(mb_type)) {
720 1842059 s->c.bdsp.clear_blocks(s->block[0]);
721
722 1842059 cbp = get_vlc2(&s->gb, ff_mb_pat_vlc, MB_PAT_VLC_BITS, 1);
723
2/2
✓ Branch 0 taken 43689 times.
✓ Branch 1 taken 1798370 times.
 1842059 if (mb_block_count > 6) {
724 43689 cbp *= 1 << mb_block_count - 6;
725 43689 cbp |= get_bits(&s->gb, mb_block_count - 6);
726 43689 s->c.bdsp.clear_blocks(s->block[6]);
727 }
728
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1842059 times.
 1842059 if (cbp <= 0) {
729 ✗ av_log(s->c.avctx, AV_LOG_ERROR,
730 "invalid cbp %d at %d %d\n", cbp, s->c.mb_x, s->c.mb_y);
731 ✗ return AVERROR_INVALIDDATA;
732 }
733
734
2/2
✓ Branch 0 taken 1730323 times.
✓ Branch 1 taken 111736 times.
 1842059 if (s->c.codec_id == AV_CODEC_ID_MPEG2VIDEO) {
735 1730323 cbp <<= 12 - mb_block_count;
736
737
2/2
✓ Branch 0 taken 10469316 times.
✓ Branch 1 taken 1730323 times.
 12199639 for (i = 0; i < mb_block_count; i++) {
738
2/2
✓ Branch 0 taken 5498236 times.
✓ Branch 1 taken 4971080 times.
 10469316 if (cbp & (1 << 11)) {
739
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 5498236 times.
 5498236 if ((ret = mpeg2_decode_block_non_intra(s, s->block[i], i)) < 0)
740 ✗ return ret;
741 } else {
742 4971080 s->c.block_last_index[i] = -1;
743 }
744 10469316 cbp += cbp;
745 }
746 } else {
747
2/2
✓ Branch 0 taken 670416 times.
✓ Branch 1 taken 111736 times.
 782152 for (i = 0; i < 6; i++) {
748
2/2
✓ Branch 0 taken 403182 times.
✓ Branch 1 taken 267234 times.
 670416 if (cbp & 32) {
749
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 403182 times.
 403182 if ((ret = mpeg1_decode_block_inter(s, s->block[i], i)) < 0)
750 ✗ return ret;
751 } else {
752 267234 s->c.block_last_index[i] = -1;
753 }
754 670416 cbp += cbp;
755 }
756 }
757 } else {
758
2/2
✓ Branch 0 taken 5685828 times.
✓ Branch 1 taken 473819 times.
 6159647 for (i = 0; i < 12; i++)
759 5685828 s->c.block_last_index[i] = -1;
760 }
761 }
762
763 2648078 s->c.cur_pic.mb_type[s->c.mb_x + s->c.mb_y * s->c.mb_stride] = mb_type;
764
765 2648078 return 0;
766 }
767
768 298 static av_cold int mpeg_decode_init(AVCodecContext *avctx)
769 {
770 298 Mpeg1Context *s = avctx->priv_data;
771 298 MPVContext *const s2 = &s->slice.c;
772 int ret;
773
774 298 s2->slice_ctx_size = sizeof(s->slice);
775 298 s2->out_format = FMT_MPEG1;
776
777
2/2
✓ Branch 0 taken 296 times.
✓ Branch 1 taken 2 times.
 298 if ( avctx->codec_tag != AV_RL32("VCR2")
778
1/2
✓ Branch 0 taken 296 times.
✗ Branch 1 not taken.
 296 && avctx->codec_tag != AV_RL32("BW10"))
779 296 avctx->coded_width = avctx->coded_height = 0; // do not trust dimensions from input
780 298 ret = ff_mpv_decode_init(s2, avctx);
781
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 298 times.
 298 if (ret < 0)
782 ✗ return ret;
783
784 298 ff_mpeg12_init_vlcs();
785
786 298 s2->chroma_format = CHROMA_420;
787 298 avctx->color_range = AVCOL_RANGE_MPEG;
788 298 return 0;
789 }
790
791 static const enum AVPixelFormat mpeg1_hwaccel_pixfmt_list_420[] = {
792 #if CONFIG_MPEG1_NVDEC_HWACCEL
793 AV_PIX_FMT_CUDA,
794 #endif
795 #if CONFIG_MPEG1_VDPAU_HWACCEL
796 AV_PIX_FMT_VDPAU,
797 #endif
798 AV_PIX_FMT_YUV420P,
799 AV_PIX_FMT_NONE
800 };
801
802 static const enum AVPixelFormat mpeg2_hwaccel_pixfmt_list_420[] = {
803 #if CONFIG_MPEG2_NVDEC_HWACCEL
804 AV_PIX_FMT_CUDA,
805 #endif
806 #if CONFIG_MPEG2_VDPAU_HWACCEL
807 AV_PIX_FMT_VDPAU,
808 #endif
809 #if CONFIG_MPEG2_DXVA2_HWACCEL
810 AV_PIX_FMT_DXVA2_VLD,
811 #endif
812 #if CONFIG_MPEG2_D3D11VA_HWACCEL
813 AV_PIX_FMT_D3D11VA_VLD,
814 AV_PIX_FMT_D3D11,
815 #endif
816 #if CONFIG_MPEG2_D3D12VA_HWACCEL
817 AV_PIX_FMT_D3D12,
818 #endif
819 #if CONFIG_MPEG2_VAAPI_HWACCEL
820 AV_PIX_FMT_VAAPI,
821 #endif
822 #if CONFIG_MPEG2_VIDEOTOOLBOX_HWACCEL
823 AV_PIX_FMT_VIDEOTOOLBOX,
824 #endif
825 AV_PIX_FMT_YUV420P,
826 AV_PIX_FMT_NONE
827 };
828
829 static const enum AVPixelFormat mpeg12_pixfmt_list_422[] = {
830 AV_PIX_FMT_YUV422P,
831 AV_PIX_FMT_NONE
832 };
833
834 static const enum AVPixelFormat mpeg12_pixfmt_list_444[] = {
835 AV_PIX_FMT_YUV444P,
836 AV_PIX_FMT_NONE
837 };
838
839 269 static enum AVPixelFormat mpeg_get_pixelformat(AVCodecContext *avctx)
840 {
841 269 Mpeg1Context *s1 = avctx->priv_data;
842 269 MPVContext *const s = &s1->slice.c;
843 const enum AVPixelFormat *pix_fmts;
844
845 if (CONFIG_GRAY && (avctx->flags & AV_CODEC_FLAG_GRAY))
846 return AV_PIX_FMT_GRAY8;
847
848
2/2
✓ Branch 0 taken 208 times.
✓ Branch 1 taken 61 times.
 269 if (s->chroma_format < CHROMA_422)
849 208 pix_fmts = avctx->codec_id == AV_CODEC_ID_MPEG1VIDEO ?
850
2/2
✓ Branch 0 taken 25 times.
✓ Branch 1 taken 183 times.
 208 mpeg1_hwaccel_pixfmt_list_420 :
851 mpeg2_hwaccel_pixfmt_list_420;
852
1/2
✓ Branch 0 taken 61 times.
✗ Branch 1 not taken.
 61 else if (s->chroma_format == CHROMA_422)
853 61 pix_fmts = mpeg12_pixfmt_list_422;
854 else
855 ✗ pix_fmts = mpeg12_pixfmt_list_444;
856
857 269 return ff_get_format(avctx, pix_fmts);
858 }
859
860 /* Call this function when we know all parameters.
861 * It may be called in different places for MPEG-1 and MPEG-2. */
862 5340 static int mpeg_decode_postinit(AVCodecContext *avctx)
863 {
864 5340 Mpeg1Context *s1 = avctx->priv_data;
865 5340 MPVContext *const s = &s1->slice.c;
866 int ret;
867
868
2/2
✓ Branch 0 taken 494 times.
✓ Branch 1 taken 4846 times.
 5340 if (avctx->codec_id == AV_CODEC_ID_MPEG1VIDEO) {
869 // MPEG-1 aspect
870 494 AVRational aspect_inv = av_d2q(ff_mpeg1_aspect[s1->aspect_ratio_info], 255);
871 494 avctx->sample_aspect_ratio = (AVRational) { aspect_inv.den, aspect_inv.num };
872 } else { // MPEG-2
873 // MPEG-2 aspect
874
2/2
✓ Branch 0 taken 2806 times.
✓ Branch 1 taken 2040 times.
 4846 if (s1->aspect_ratio_info > 1) {
875 AVRational dar =
876 2806 av_mul_q(av_div_q(ff_mpeg2_aspect[s1->aspect_ratio_info],
877 2806 (AVRational) { s1->pan_scan.width,
878 2806 s1->pan_scan.height }),
879 2806 (AVRational) { s->width, s->height });
880
881 /* We ignore the spec here and guess a bit as reality does not
882 * match the spec, see for example res_change_ffmpeg_aspect.ts
883 * and sequence-display-aspect.mpg.
884 * issue1613, 621, 562 */
885
5/6
✓ Branch 0 taken 587 times.
✓ Branch 1 taken 2219 times.
✓ Branch 2 taken 587 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 31 times.
✓ Branch 5 taken 556 times.
 3393 if ((s1->pan_scan.width == 0) || (s1->pan_scan.height == 0) ||
886
2/2
✓ Branch 1 taken 26 times.
✓ Branch 2 taken 5 times.
 618 (av_cmp_q(dar, (AVRational) { 4, 3 }) &&
887 31 av_cmp_q(dar, (AVRational) { 16, 9 }))) {
888 2245 s->avctx->sample_aspect_ratio =
889 2245 av_div_q(ff_mpeg2_aspect[s1->aspect_ratio_info],
890 2245 (AVRational) { s->width, s->height });
891 } else {
892 561 s->avctx->sample_aspect_ratio =
893 561 av_div_q(ff_mpeg2_aspect[s1->aspect_ratio_info],
894 561 (AVRational) { s1->pan_scan.width, s1->pan_scan.height });
895 // issue1613 4/3 16/9 -> 16/9
896 // res_change_ffmpeg_aspect.ts 4/3 225/44 ->4/3
897 // widescreen-issue562.mpg 4/3 16/9 -> 16/9
898 // s->avctx->sample_aspect_ratio = av_mul_q(s->avctx->sample_aspect_ratio, (AVRational) {s->width, s->height});
899 ff_dlog(avctx, "aspect A %d/%d\n",
900 ff_mpeg2_aspect[s1->aspect_ratio_info].num,
901 ff_mpeg2_aspect[s1->aspect_ratio_info].den);
902 ff_dlog(avctx, "aspect B %d/%d\n", s->avctx->sample_aspect_ratio.num,
903 s->avctx->sample_aspect_ratio.den);
904 }
905 } else {
906 2040 s->avctx->sample_aspect_ratio =
907 2040 ff_mpeg2_aspect[s1->aspect_ratio_info];
908 }
909 } // MPEG-2
910
911
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 5340 times.
 5340 if (av_image_check_sar(s->width, s->height,
912 avctx->sample_aspect_ratio) < 0) {
913 ✗ av_log(avctx, AV_LOG_WARNING, "ignoring invalid SAR: %u/%u\n",
914 avctx->sample_aspect_ratio.num,
915 avctx->sample_aspect_ratio.den);
916 ✗ avctx->sample_aspect_ratio = (AVRational){ 0, 1 };
917 }
918
919
2/2
✓ Branch 0 taken 5073 times.
✓ Branch 1 taken 267 times.
 5340 if (!s->context_initialized ||
920
1/2
✓ Branch 0 taken 5073 times.
✗ Branch 1 not taken.
 5073 avctx->coded_width != s->width ||
921
1/2
✓ Branch 0 taken 5073 times.
✗ Branch 1 not taken.
 5073 avctx->coded_height != s->height ||
922
1/2
✓ Branch 0 taken 5073 times.
✗ Branch 1 not taken.
 5073 s1->save_chroma_format != s->chroma_format ||
923
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 5073 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 5073 (s1->save_progressive_seq != s->progressive_sequence && FFALIGN(s->height, 16) != FFALIGN(s->height, 32)) ||
924 0) {
925
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 267 times.
 267 if (s->context_initialized)
926 ✗ ff_mpv_common_end(s);
927
928 267 ret = ff_set_dimensions(avctx, s->width, s->height);
929
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 267 times.
 267 if (ret < 0)
930 ✗ return ret;
931
932
3/4
✓ Branch 0 taken 242 times.
✓ Branch 1 taken 25 times.
✓ Branch 2 taken 242 times.
✗ Branch 3 not taken.
 267 if (avctx->codec_id == AV_CODEC_ID_MPEG2VIDEO && s1->bit_rate &&
933
2/2
✓ Branch 0 taken 103 times.
✓ Branch 1 taken 139 times.
 242 (s1->bit_rate != 0x3FFFF*400)) {
934 103 avctx->rc_max_rate = s1->bit_rate;
935
3/4
✓ Branch 0 taken 25 times.
✓ Branch 1 taken 139 times.
✓ Branch 2 taken 25 times.
✗ Branch 3 not taken.
 164 } else if (avctx->codec_id == AV_CODEC_ID_MPEG1VIDEO && s1->bit_rate &&
936
2/4
✓ Branch 0 taken 25 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 25 times.
✗ Branch 3 not taken.
 25 (s1->bit_rate != 0x3FFFF*400 || s1->vbv_delay != 0xFFFF)) {
937 25 avctx->bit_rate = s1->bit_rate;
938 }
939 267 s1->save_progressive_seq = s->progressive_sequence;
940 267 s1->save_chroma_format = s->chroma_format;
941
942 /* low_delay may be forced, in this case we will have B-frames
943 * that behave like P-frames. */
944 267 avctx->has_b_frames = !s->low_delay;
945
946
2/2
✓ Branch 0 taken 25 times.
✓ Branch 1 taken 242 times.
 267 if (avctx->codec_id == AV_CODEC_ID_MPEG1VIDEO) {
947 // MPEG-1 fps
948 25 avctx->framerate = ff_mpeg12_frame_rate_tab[s1->frame_rate_index];
949 25 avctx->chroma_sample_location = AVCHROMA_LOC_CENTER;
950 } else { // MPEG-2
951 // MPEG-2 fps
952 242 av_reduce(&s->avctx->framerate.num,
953 242 &s->avctx->framerate.den,
954 242 ff_mpeg12_frame_rate_tab[s1->frame_rate_index].num * s1->frame_rate_ext.num,
955 242 ff_mpeg12_frame_rate_tab[s1->frame_rate_index].den * s1->frame_rate_ext.den,
956 1 << 30);
957
958
2/3
✓ Branch 0 taken 181 times.
✓ Branch 1 taken 61 times.
✗ Branch 2 not taken.
 242 switch (s->chroma_format) {
959 181 case CHROMA_420: avctx->chroma_sample_location = AVCHROMA_LOC_LEFT; break;
960 61 case CHROMA_422:
961 61 case CHROMA_444: avctx->chroma_sample_location = AVCHROMA_LOC_TOPLEFT; break;
962 ✗ default: av_assert0(0);
963 }
964 } // MPEG-2
965
966 267 avctx->pix_fmt = mpeg_get_pixelformat(avctx);
967
968
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 267 times.
 267 if ((ret = ff_mpv_common_init(s)) < 0)
969 ✗ return ret;
970
1/2
✓ Branch 0 taken 267 times.
✗ Branch 1 not taken.
 267 if (!s->avctx->lowres)
971
2/2
✓ Branch 0 taken 315 times.
✓ Branch 1 taken 267 times.
 582 for (int i = 0; i < s->slice_context_count; i++)
972 315 ff_mpv_framesize_disable(&s->thread_context[i]->sc);
973 }
974 5340 return 0;
975 }
976
977 5340 static int mpeg1_decode_picture(AVCodecContext *avctx, const uint8_t *buf,
978 int buf_size)
979 {
980 5340 Mpeg1Context *s1 = avctx->priv_data;
981 5340 MPVContext *const s = &s1->slice.c;
982 5340 GetBitContext gb0, *const gb = &gb0;
983 int ref, f_code, vbv_delay, ret;
984
985 5340 ret = init_get_bits8(gb, buf, buf_size);
986
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 5340 times.
 5340 if (ret < 0)
987 ✗ return ret;
988
989 5340 ref = get_bits(gb, 10); /* temporal ref */
990 5340 s->pict_type = get_bits(gb, 3);
991
2/4
✓ Branch 0 taken 5340 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 5340 times.
 5340 if (s->pict_type == 0 || s->pict_type > 3)
992 ✗ return AVERROR_INVALIDDATA;
993
994 5340 vbv_delay = get_bits(gb, 16);
995 5340 s1->vbv_delay = vbv_delay;
996
2/2
✓ Branch 0 taken 2478 times.
✓ Branch 1 taken 2862 times.
 5340 if (s->pict_type == AV_PICTURE_TYPE_P ||
997
2/2
✓ Branch 0 taken 1753 times.
✓ Branch 1 taken 725 times.
 2478 s->pict_type == AV_PICTURE_TYPE_B) {
998 4615 s->full_pel[0] = get_bits1(gb);
999 4615 f_code = get_bits(gb, 3);
1000
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 4615 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 4615 if (f_code == 0 && (avctx->err_recognition & (AV_EF_BITSTREAM|AV_EF_COMPLIANT)))
1001 ✗ return AVERROR_INVALIDDATA;
1002 4615 f_code += !f_code;
1003 4615 s->mpeg_f_code[0][0] = f_code;
1004 4615 s->mpeg_f_code[0][1] = f_code;
1005 }
1006
2/2
✓ Branch 0 taken 1753 times.
✓ Branch 1 taken 3587 times.
 5340 if (s->pict_type == AV_PICTURE_TYPE_B) {
1007 1753 s->full_pel[1] = get_bits1(gb);
1008 1753 f_code = get_bits(gb, 3);
1009
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 1753 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 1753 if (f_code == 0 && (avctx->err_recognition & (AV_EF_BITSTREAM|AV_EF_COMPLIANT)))
1010 ✗ return AVERROR_INVALIDDATA;
1011 1753 f_code += !f_code;
1012 1753 s->mpeg_f_code[1][0] = f_code;
1013 1753 s->mpeg_f_code[1][1] = f_code;
1014 }
1015
1016
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 5340 times.
 5340 if (avctx->debug & FF_DEBUG_PICT_INFO)
1017 ✗ av_log(avctx, AV_LOG_DEBUG,
1018 ✗ "vbv_delay %d, ref %d type:%d\n", vbv_delay, ref, s->pict_type);
1019
1020 5340 return 0;
1021 }
1022
1023 739 static void mpeg_decode_sequence_extension(Mpeg1Context *const s1,
1024 GetBitContext *const gb)
1025 {
1026 739 MPVContext *const s = &s1->slice.c;
1027 int horiz_size_ext, vert_size_ext;
1028 int bit_rate_ext;
1029
1030 739 skip_bits(gb, 1); /* profile and level esc*/
1031 739 s->avctx->profile = get_bits(gb, 3);
1032 739 s->avctx->level = get_bits(gb, 4);
1033 739 s->progressive_sequence = get_bits1(gb); /* progressive_sequence */
1034 739 s->chroma_format = get_bits(gb, 2); /* chroma_format 1=420, 2=422, 3=444 */
1035
1036
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 739 times.
 739 if (!s->chroma_format) {
1037 ✗ s->chroma_format = CHROMA_420;
1038 ✗ av_log(s->avctx, AV_LOG_WARNING, "Chroma format invalid\n");
1039 }
1040
1041 739 horiz_size_ext = get_bits(gb, 2);
1042 739 vert_size_ext = get_bits(gb, 2);
1043 739 s->width |= (horiz_size_ext << 12);
1044 739 s->height |= (vert_size_ext << 12);
1045 739 bit_rate_ext = get_bits(gb, 12); /* XXX: handle it */
1046 739 s1->bit_rate += (bit_rate_ext << 18) * 400LL;
1047 739 check_marker(s->avctx, gb, "after bit rate extension");
1048 739 s->avctx->rc_buffer_size += get_bits(gb, 8) * 1024 * 16 << 10;
1049
1050 739 s->low_delay = get_bits1(gb);
1051
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 739 times.
 739 if (s->avctx->flags & AV_CODEC_FLAG_LOW_DELAY)
1052 ✗ s->low_delay = 1;
1053
1054 739 s1->frame_rate_ext.num = get_bits(gb, 2) + 1;
1055 739 s1->frame_rate_ext.den = get_bits(gb, 5) + 1;
1056
1057 ff_dlog(s->avctx, "sequence extension\n");
1058 739 s->codec_id = s->avctx->codec_id = AV_CODEC_ID_MPEG2VIDEO;
1059
1060
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 739 times.
 739 if (s->avctx->debug & FF_DEBUG_PICT_INFO)
1061 ✗ av_log(s->avctx, AV_LOG_DEBUG,
1062 "profile: %d, level: %d ps: %d cf:%d vbv buffer: %d, bitrate:%"PRId64"\n",
1063 ✗ s->avctx->profile, s->avctx->level, s->progressive_sequence, s->chroma_format,
1064 ✗ s->avctx->rc_buffer_size, s1->bit_rate);
1065 739 }
1066
1067 103 static void mpeg_decode_sequence_display_extension(Mpeg1Context *const s1,
1068 GetBitContext *const gb)
1069 {
1070 103 MPVContext *const s = &s1->slice.c;
1071 int color_description, w, h;
1072
1073 103 skip_bits(gb, 3); /* video format */
1074 103 color_description = get_bits1(gb);
1075
2/2
✓ Branch 0 taken 41 times.
✓ Branch 1 taken 62 times.
 103 if (color_description) {
1076 41 s->avctx->color_primaries = get_bits(gb, 8);
1077 41 s->avctx->color_trc = get_bits(gb, 8);
1078 41 s->avctx->colorspace = get_bits(gb, 8);
1079 }
1080 103 w = get_bits(gb, 14);
1081 103 skip_bits(gb, 1); // marker
1082 103 h = get_bits(gb, 14);
1083 // remaining 3 bits are zero padding
1084
1085 103 s1->pan_scan.width = 16 * w;
1086 103 s1->pan_scan.height = 16 * h;
1087
1088
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 103 times.
 103 if (s->avctx->debug & FF_DEBUG_PICT_INFO)
1089 ✗ av_log(s->avctx, AV_LOG_DEBUG, "sde w:%d, h:%d\n", w, h);
1090 103 }
1091
1092 27 static void mpeg_decode_picture_display_extension(Mpeg1Context *const s1,
1093 GetBitContext *const gb)
1094 {
1095 27 MPVContext *const s = &s1->slice.c;
1096 int i, nofco;
1097
1098 27 nofco = 1;
1099
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 27 times.
 27 if (s->progressive_sequence) {
1100 ✗ if (s->repeat_first_field) {
1101 ✗ nofco++;
1102 ✗ if (s->top_field_first)
1103 ✗ nofco++;
1104 }
1105 } else {
1106
2/2
✓ Branch 0 taken 9 times.
✓ Branch 1 taken 18 times.
 27 if (s->picture_structure == PICT_FRAME) {
1107 9 nofco++;
1108
2/2
✓ Branch 0 taken 6 times.
✓ Branch 1 taken 3 times.
 9 if (s->repeat_first_field)
1109 6 nofco++;
1110 }
1111 }
1112
2/2
✓ Branch 0 taken 42 times.
✓ Branch 1 taken 27 times.
 69 for (i = 0; i < nofco; i++) {
1113 42 s1->pan_scan.position[i][0] = get_sbits(gb, 16);
1114 42 skip_bits(gb, 1); // marker
1115 42 s1->pan_scan.position[i][1] = get_sbits(gb, 16);
1116 42 skip_bits(gb, 1); // marker
1117 }
1118
1119
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 27 times.
 27 if (s->avctx->debug & FF_DEBUG_PICT_INFO)
1120 ✗ av_log(s->avctx, AV_LOG_DEBUG,
1121 "pde (%"PRId16",%"PRId16") (%"PRId16",%"PRId16") (%"PRId16",%"PRId16")\n",
1122 ✗ s1->pan_scan.position[0][0], s1->pan_scan.position[0][1],
1123 ✗ s1->pan_scan.position[1][0], s1->pan_scan.position[1][1],
1124 ✗ s1->pan_scan.position[2][0], s1->pan_scan.position[2][1]);
1125 27 }
1126
1127 243 static int load_matrix(MPVContext *const s, GetBitContext *const gb,
1128 uint16_t matrix0[64], uint16_t matrix1[64], int intra)
1129 {
1130 int i;
1131
1132
2/2
✓ Branch 0 taken 15552 times.
✓ Branch 1 taken 243 times.
 15795 for (i = 0; i < 64; i++) {
1133 15552 int j = s->idsp.idct_permutation[ff_zigzag_direct[i]];
1134 15552 int v = get_bits(gb, 8);
1135
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 15552 times.
 15552 if (v == 0) {
1136 ✗ av_log(s->avctx, AV_LOG_ERROR, "matrix damaged\n");
1137 ✗ return AVERROR_INVALIDDATA;
1138 }
1139
5/6
✓ Branch 0 taken 4672 times.
✓ Branch 1 taken 10880 times.
✓ Branch 2 taken 73 times.
✓ Branch 3 taken 4599 times.
✗ Branch 4 not taken.
✓ Branch 5 taken 73 times.
 15552 if (intra && i == 0 && v != 8) {
1140 ✗ av_log(s->avctx, AV_LOG_DEBUG, "intra matrix specifies invalid DC quantizer %d, ignoring\n", v);
1141 ✗ v = 8; // needed by pink.mpg / issue1046
1142 }
1143 15552 matrix0[j] = v;
1144
2/2
✓ Branch 0 taken 14912 times.
✓ Branch 1 taken 640 times.
 15552 if (matrix1)
1145 14912 matrix1[j] = v;
1146 }
1147 243 return 0;
1148 }
1149
1150 1055 static void mpeg_decode_quant_matrix_extension(MPVContext *const s,
1151 GetBitContext *const gb)
1152 {
1153 ff_dlog(s->avctx, "matrix extension\n");
1154
1155
2/2
✓ Branch 1 taken 34 times.
✓ Branch 2 taken 1021 times.
 1055 if (get_bits1(gb))
1156 34 load_matrix(s, gb, s->chroma_intra_matrix, s->intra_matrix, 1);
1157
2/2
✓ Branch 1 taken 77 times.
✓ Branch 2 taken 978 times.
 1055 if (get_bits1(gb))
1158 77 load_matrix(s, gb, s->chroma_inter_matrix, s->inter_matrix, 0);
1159
2/2
✓ Branch 1 taken 9 times.
✓ Branch 2 taken 1046 times.
 1055 if (get_bits1(gb))
1160 9 load_matrix(s, gb, s->chroma_intra_matrix, NULL, 1);
1161
2/2
✓ Branch 1 taken 1 times.
✓ Branch 2 taken 1054 times.
 1055 if (get_bits1(gb))
1162 1 load_matrix(s, gb, s->chroma_inter_matrix, NULL, 0);
1163 1055 }
1164
1165 4846 static int mpeg_decode_picture_coding_extension(Mpeg1Context *const s1,
1166 GetBitContext *const gb)
1167 {
1168 4846 MPVContext *const s = &s1->slice.c;
1169
1170 4846 s->full_pel[0] = s->full_pel[1] = 0;
1171 4846 s->mpeg_f_code[0][0] = get_bits(gb, 4);
1172 4846 s->mpeg_f_code[0][1] = get_bits(gb, 4);
1173 4846 s->mpeg_f_code[1][0] = get_bits(gb, 4);
1174 4846 s->mpeg_f_code[1][1] = get_bits(gb, 4);
1175 4846 s->mpeg_f_code[0][0] += !s->mpeg_f_code[0][0];
1176 4846 s->mpeg_f_code[0][1] += !s->mpeg_f_code[0][1];
1177 4846 s->mpeg_f_code[1][0] += !s->mpeg_f_code[1][0];
1178 4846 s->mpeg_f_code[1][1] += !s->mpeg_f_code[1][1];
1179
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 4846 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 4846 if (!s->pict_type && s->context_initialized) {
1180 ✗ av_log(s->avctx, AV_LOG_ERROR, "Missing picture start code\n");
1181 ✗ if (s->avctx->err_recognition & AV_EF_EXPLODE)
1182 ✗ return AVERROR_INVALIDDATA;
1183 ✗ av_log(s->avctx, AV_LOG_WARNING, "Guessing pict_type from mpeg_f_code\n");
1184 ✗ if (s->mpeg_f_code[1][0] == 15 && s->mpeg_f_code[1][1] == 15) {
1185 ✗ if (s->mpeg_f_code[0][0] == 15 && s->mpeg_f_code[0][1] == 15)
1186 ✗ s->pict_type = AV_PICTURE_TYPE_I;
1187 else
1188 ✗ s->pict_type = AV_PICTURE_TYPE_P;
1189 } else
1190 ✗ s->pict_type = AV_PICTURE_TYPE_B;
1191 }
1192
1193 4846 s->intra_dc_precision = get_bits(gb, 2);
1194 4846 s->picture_structure = get_bits(gb, 2);
1195 4846 s->top_field_first = get_bits1(gb);
1196 4846 s->frame_pred_frame_dct = get_bits1(gb);
1197 4846 s->concealment_motion_vectors = get_bits1(gb);
1198 4846 s->q_scale_type = get_bits1(gb);
1199 4846 s->intra_vlc_format = get_bits1(gb);
1200 4846 s->alternate_scan = get_bits1(gb);
1201 4846 s->repeat_first_field = get_bits1(gb);
1202 4846 s->chroma_420_type = get_bits1(gb);
1203 4846 s->progressive_frame = get_bits1(gb);
1204
1205 // We only initialize intra_scantable.permutated, as this is all we use.
1206 4846 ff_permute_scantable(s->intra_scantable.permutated,
1207 4846 s->alternate_scan ? ff_alternate_vertical_scan : ff_zigzag_direct,
1208
2/2
✓ Branch 0 taken 1765 times.
✓ Branch 1 taken 3081 times.
 4846 s->idsp.idct_permutation);
1209
1210 /* composite display not parsed */
1211 ff_dlog(s->avctx, "intra_dc_precision=%d\n", s->intra_dc_precision);
1212 ff_dlog(s->avctx, "picture_structure=%d\n", s->picture_structure);
1213 ff_dlog(s->avctx, "top field first=%d\n", s->top_field_first);
1214 ff_dlog(s->avctx, "repeat first field=%d\n", s->repeat_first_field);
1215 ff_dlog(s->avctx, "conceal=%d\n", s->concealment_motion_vectors);
1216 ff_dlog(s->avctx, "intra_vlc_format=%d\n", s->intra_vlc_format);
1217 ff_dlog(s->avctx, "alternate_scan=%d\n", s->alternate_scan);
1218 ff_dlog(s->avctx, "frame_pred_frame_dct=%d\n", s->frame_pred_frame_dct);
1219 ff_dlog(s->avctx, "progressive_frame=%d\n", s->progressive_frame);
1220
1221 4846 return 0;
1222 }
1223
1224 4868 static int mpeg_field_start(Mpeg1Context *s1, const uint8_t *buf, int buf_size)
1225 {
1226 4868 MPVContext *const s = &s1->slice.c;
1227 4868 AVCodecContext *avctx = s->avctx;
1228 4868 int second_field = 0;
1229 int ret;
1230
1231
1/2
✓ Branch 0 taken 4868 times.
✗ Branch 1 not taken.
 4868 if (!(avctx->flags2 & AV_CODEC_FLAG2_CHUNKS)) {
1232
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 4868 times.
 4868 if (s->mb_width * s->mb_height * 11LL / (33 * 2 * 8) > buf_size)
1233 ✗ return AVERROR_INVALIDDATA;
1234 }
1235
1236 /* start frame decoding */
1237
4/4
✓ Branch 0 taken 4129 times.
✓ Branch 1 taken 739 times.
✓ Branch 2 taken 3390 times.
✓ Branch 3 taken 739 times.
 8997 if (s->first_field || s->picture_structure == PICT_FRAME) {
1238 AVFrameSideData *pan_scan;
1239
1240
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 4129 times.
 4129 if ((ret = ff_mpv_frame_start(s, avctx)) < 0)
1241 ✗ return ret;
1242
1243
2/2
✓ Branch 0 taken 739 times.
✓ Branch 1 taken 3390 times.
 4129 if (s->picture_structure != PICT_FRAME) {
1244 739 s->cur_pic.ptr->f->flags |= AV_FRAME_FLAG_TOP_FIELD_FIRST *
1245
2/2
✓ Branch 0 taken 2 times.
✓ Branch 1 taken 737 times.
 739 (s->picture_structure == PICT_TOP_FIELD);
1246
1247
2/2
✓ Branch 0 taken 2217 times.
✓ Branch 1 taken 739 times.
 2956 for (int i = 0; i < 3; i++) {
1248
2/2
✓ Branch 0 taken 2211 times.
✓ Branch 1 taken 6 times.
 2217 if (s->picture_structure == PICT_BOTTOM_FIELD) {
1249
1/2
✓ Branch 0 taken 2211 times.
✗ Branch 1 not taken.
 2211 s->cur_pic.data[i] = FF_PTR_ADD(s->cur_pic.data[i],
1250 s->cur_pic.linesize[i]);
1251 }
1252 2217 s->cur_pic.linesize[i] *= 2;
1253 }
1254 }
1255
1256 4129 ff_mpeg_er_frame_start(s);
1257
1258 /* first check if we must repeat the frame */
1259 4129 s->cur_pic.ptr->f->repeat_pict = 0;
1260
2/2
✓ Branch 0 taken 6 times.
✓ Branch 1 taken 4123 times.
 4129 if (s->repeat_first_field) {
1261
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
 6 if (s->progressive_sequence) {
1262 ✗ if (s->top_field_first)
1263 ✗ s->cur_pic.ptr->f->repeat_pict = 4;
1264 else
1265 ✗ s->cur_pic.ptr->f->repeat_pict = 2;
1266
1/2
✓ Branch 0 taken 6 times.
✗ Branch 1 not taken.
 6 } else if (s->progressive_frame) {
1267 6 s->cur_pic.ptr->f->repeat_pict = 1;
1268 }
1269 }
1270
1271 4129 ret = ff_frame_new_side_data(s->avctx, s->cur_pic.ptr->f,
1272 AV_FRAME_DATA_PANSCAN, sizeof(s1->pan_scan),
1273 &pan_scan);
1274
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 4129 times.
 4129 if (ret < 0)
1275 ✗ return ret;
1276
1/2
✓ Branch 0 taken 4129 times.
✗ Branch 1 not taken.
 4129 if (pan_scan)
1277 4129 memcpy(pan_scan->data, &s1->pan_scan, sizeof(s1->pan_scan));
1278
1279
2/2
✓ Branch 0 taken 944 times.
✓ Branch 1 taken 3185 times.
 4129 if (s1->a53_buf_ref) {
1280 944 ret = ff_frame_new_side_data_from_buf(
1281 944 s->avctx, s->cur_pic.ptr->f, AV_FRAME_DATA_A53_CC,
1282 &s1->a53_buf_ref);
1283
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 944 times.
 944 if (ret < 0)
1284 ✗ return ret;
1285 }
1286
1287
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 4129 times.
 4129 if (s1->has_stereo3d) {
1288 ✗ AVStereo3D *stereo = av_stereo3d_create_side_data(s->cur_pic.ptr->f);
1289 ✗ if (!stereo)
1290 ✗ return AVERROR(ENOMEM);
1291
1292 ✗ stereo->type = s1->stereo3d_type;
1293 ✗ s1->has_stereo3d = 0;
1294 }
1295
1296
2/2
✓ Branch 0 taken 404 times.
✓ Branch 1 taken 3725 times.
 4129 if (s1->has_afd) {
1297 AVFrameSideData *sd;
1298 404 ret = ff_frame_new_side_data(s->avctx, s->cur_pic.ptr->f,
1299 AV_FRAME_DATA_AFD, 1, &sd);
1300
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 404 times.
 404 if (ret < 0)
1301 ✗ return ret;
1302
1/2
✓ Branch 0 taken 404 times.
✗ Branch 1 not taken.
 404 if (sd)
1303 404 *sd->data = s1->afd;
1304 404 s1->has_afd = 0;
1305 }
1306 } else { // second field
1307 739 second_field = 1;
1308
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 739 times.
 739 if (!s->cur_pic.ptr) {
1309 ✗ av_log(s->avctx, AV_LOG_ERROR, "first field missing\n");
1310 ✗ return AVERROR_INVALIDDATA;
1311 }
1312
1313
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 739 times.
 739 if (s->avctx->hwaccel) {
1314 ✗ if ((ret = FF_HW_SIMPLE_CALL(s->avctx, end_frame)) < 0) {
1315 ✗ av_log(avctx, AV_LOG_ERROR,
1316 "hardware accelerator failed to decode first field\n");
1317 ✗ return ret;
1318 }
1319 }
1320 739 ret = ff_mpv_alloc_dummy_frames(s);
1321
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 739 times.
 739 if (ret < 0)
1322 ✗ return ret;
1323
1324
2/2
✓ Branch 0 taken 2217 times.
✓ Branch 1 taken 739 times.
 2956 for (int i = 0; i < 3; i++) {
1325 2217 s->cur_pic.data[i] = s->cur_pic.ptr->f->data[i];
1326
2/2
✓ Branch 0 taken 6 times.
✓ Branch 1 taken 2211 times.
 2217 if (s->picture_structure == PICT_BOTTOM_FIELD)
1327 6 s->cur_pic.data[i] +=
1328 6 s->cur_pic.ptr->f->linesize[i];
1329 }
1330 }
1331
1332
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 4868 times.
 4868 if (avctx->hwaccel) {
1333 ✗ if ((ret = FF_HW_CALL(avctx, start_frame, NULL, buf, buf_size)) < 0)
1334 ✗ return ret;
1335
2/2
✓ Branch 0 taken 157 times.
✓ Branch 1 taken 4711 times.
 4868 } else if (s->codec_tag == MKTAG('V', 'C', 'R', '2')) {
1336 // Exchange UV
1337 157 FFSWAP(uint8_t*, s->cur_pic.data[1], s->cur_pic.data[2]);
1338 157 FFSWAP(ptrdiff_t, s->cur_pic.linesize[1], s->cur_pic.linesize[2]);
1339
1/2
✓ Branch 0 taken 157 times.
✗ Branch 1 not taken.
 157 if (!second_field) {
1340 157 FFSWAP(uint8_t*, s->next_pic.data[1], s->next_pic.data[2]);
1341 157 FFSWAP(ptrdiff_t, s->next_pic.linesize[1], s->next_pic.linesize[2]);
1342 157 FFSWAP(uint8_t*, s->last_pic.data[1], s->last_pic.data[2]);
1343 157 FFSWAP(ptrdiff_t, s->last_pic.linesize[1], s->last_pic.linesize[2]);
1344 }
1345 }
1346
1347 4868 return 0;
1348 }
1349
1350 #define DECODE_SLICE_ERROR -1
1351 #define DECODE_SLICE_OK 0
1352
1353 /**
1354 * Decode a slice.
1355 * Mpeg12SliceContext.c.mb_y must be set to the MB row from the startcode.
1356 * @return DECODE_SLICE_ERROR if the slice is damaged,
1357 * DECODE_SLICE_OK if this slice is OK
1358 */
1359 134169 static int mpeg_decode_slice(Mpeg12SliceContext *const s, int mb_y,
1360 const uint8_t **buf, int buf_size)
1361 {
1362 134169 AVCodecContext *avctx = s->c.avctx;
1363 134169 const int lowres = s->c.avctx->lowres;
1364 134169 const int field_pic = s->c.picture_structure != PICT_FRAME;
1365 int ret;
1366
1367 134169 s->c.resync_mb_x =
1368 134169 s->c.resync_mb_y = -1;
1369
1370
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 134169 times.
 134169 av_assert0(mb_y < s->c.mb_height);
1371
1372 134169 ret = init_get_bits8(&s->gb, *buf, buf_size);
1373
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 134169 times.
 134169 if (ret < 0)
1374 ✗ return ret;
1375
1376
3/4
✓ Branch 0 taken 126896 times.
✓ Branch 1 taken 7273 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 126896 times.
 134169 if (s->c.codec_id != AV_CODEC_ID_MPEG1VIDEO && s->c.mb_height > 2800/16)
1377 ✗ skip_bits(&s->gb, 3);
1378
1379 134169 s->c.interlaced_dct = 0;
1380
1381 134169 s->c.qscale = mpeg_get_qscale(&s->gb, s->c.q_scale_type);
1382
1383
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 134169 times.
 134169 if (s->c.qscale == 0) {
1384 ✗ av_log(s->c.avctx, AV_LOG_ERROR, "qscale == 0\n");
1385 ✗ return AVERROR_INVALIDDATA;
1386 }
1387
1388 /* extra slice info */
1389
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 134169 times.
 134169 if (skip_1stop_8data_bits(&s->gb) < 0)
1390 ✗ return AVERROR_INVALIDDATA;
1391
1392 134169 s->c.mb_x = 0;
1393
1394
3/4
✓ Branch 0 taken 5624 times.
✓ Branch 1 taken 128545 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 5624 times.
 134169 if (mb_y == 0 && s->c.codec_tag == AV_RL32("SLIF")) {
1395 ✗ skip_bits1(&s->gb);
1396 } else {
1397
1/2
✓ Branch 1 taken 147167 times.
✗ Branch 2 not taken.
 281336 while (get_bits_left(&s->gb) > 0) {
1398 147167 int code = get_vlc2(&s->gb, ff_mbincr_vlc,
1399 MBINCR_VLC_BITS, 2);
1400
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 147167 times.
 147167 if (code < 0) {
1401 ✗ av_log(s->c.avctx, AV_LOG_ERROR, "first mb_incr damaged\n");
1402 ✗ return AVERROR_INVALIDDATA;
1403 }
1404
2/2
✓ Branch 0 taken 12998 times.
✓ Branch 1 taken 134169 times.
 147167 if (code >= 33) {
1405
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 12998 times.
 12998 if (code == 33)
1406 12998 s->c.mb_x += 33;
1407 /* otherwise, stuffing, nothing to do */
1408 } else {
1409 134169 s->c.mb_x += code;
1410 134169 break;
1411 }
1412 }
1413 }
1414
1415
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 134169 times.
 134169 if (s->c.mb_x >= (unsigned) s->c.mb_width) {
1416 ✗ av_log(s->c.avctx, AV_LOG_ERROR, "initial skip overflow\n");
1417 ✗ return AVERROR_INVALIDDATA;
1418 }
1419
1420
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 134169 times.
 134169 if (avctx->hwaccel) {
1421 ✗ const uint8_t *buf_end, *buf_start = *buf - 4; /* include start_code */
1422 ✗ int start_code = -1;
1423 ✗ buf_end = avpriv_find_start_code(buf_start + 2, *buf + buf_size, &start_code);
1424 ✗ if (buf_end < *buf + buf_size)
1425 ✗ buf_end -= 4;
1426 ✗ s->c.mb_y = mb_y;
1427 ✗ if (FF_HW_CALL(avctx, decode_slice, buf_start, buf_end - buf_start) < 0)
1428 ✗ return DECODE_SLICE_ERROR;
1429 ✗ *buf = buf_end;
1430 ✗ return DECODE_SLICE_OK;
1431 }
1432
1433 134169 s->c.resync_mb_x = s->c.mb_x;
1434 134169 s->c.resync_mb_y = s->c.mb_y = mb_y;
1435 134169 ff_init_block_index(&s->c);
1436
1437
8/8
✓ Branch 0 taken 5624 times.
✓ Branch 1 taken 128545 times.
✓ Branch 2 taken 4129 times.
✓ Branch 3 taken 1495 times.
✓ Branch 4 taken 4127 times.
✓ Branch 5 taken 2 times.
✓ Branch 6 taken 3390 times.
✓ Branch 7 taken 737 times.
 134169 if (s->c.mb_y == 0 && s->c.mb_x == 0 && (s->c.first_field || s->c.picture_structure == PICT_FRAME)) {
1438
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 3392 times.
 3392 if (s->c.avctx->debug & FF_DEBUG_PICT_INFO) {
1439 ✗ av_log(s->c.avctx, AV_LOG_DEBUG,
1440 "qp:%d fc:%2d%2d%2d%2d %c %s %s %s %s dc:%d pstruct:%d fdct:%d cmv:%d qtype:%d ivlc:%d rff:%d %s\n",
1441 s->c.qscale,
1442 s->c.mpeg_f_code[0][0], s->c.mpeg_f_code[0][1],
1443 s->c.mpeg_f_code[1][0], s->c.mpeg_f_code[1][1],
1444 ✗ s->c.pict_type == AV_PICTURE_TYPE_I ? 'I' :
1445 ✗ (s->c.pict_type == AV_PICTURE_TYPE_P ? 'P' :
1446 ✗ (s->c.pict_type == AV_PICTURE_TYPE_B ? 'B' : 'S')),
1447 ✗ s->c.progressive_sequence ? "ps" : "",
1448 ✗ s->c.progressive_frame ? "pf" : "",
1449 ✗ s->c.alternate_scan ? "alt" : "",
1450 ✗ s->c.top_field_first ? "top" : "",
1451 s->c.intra_dc_precision, s->c.picture_structure,
1452 s->c.frame_pred_frame_dct, s->c.concealment_motion_vectors,
1453 s->c.q_scale_type, s->c.intra_vlc_format,
1454 ✗ s->c.repeat_first_field, s->c.chroma_420_type ? "420" : "");
1455 }
1456 }
1457
1458 134169 s->last_dc[0] = 128 << s->c.intra_dc_precision;
1459 134169 s->last_dc[1] = s->last_dc[0];
1460 134169 s->last_dc[2] = s->last_dc[0];
1461 134169 memset(s->c.last_mv, 0, sizeof(s->c.last_mv));
1462
1463 134169 for (int mb_skip_run = 0;;) {
1464 2914172 ret = mpeg_decode_mb(s, &mb_skip_run);
1465
2/2
✓ Branch 0 taken 11 times.
✓ Branch 1 taken 2914161 times.
 2914172 if (ret < 0)
1466 11 return ret;
1467
1468 // Note motion_val is normally NULL unless we want to extract the MVs.
1469
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2914161 times.
 2914161 if (s->c.cur_pic.motion_val[0]) {
1470 ✗ const int wrap = s->c.b8_stride;
1471 ✗ int xy = s->c.mb_x * 2 + s->c.mb_y * 2 * wrap;
1472 ✗ int b8_xy = 4 * (s->c.mb_x + s->c.mb_y * s->c.mb_stride);
1473 int motion_x, motion_y, dir, i;
1474
1475 ✗ for (i = 0; i < 2; i++) {
1476 ✗ for (dir = 0; dir < 2; dir++) {
1477 ✗ if (s->c.mb_intra ||
1478 ✗ (dir == 1 && s->c.pict_type != AV_PICTURE_TYPE_B)) {
1479 ✗ motion_x = motion_y = 0;
1480 ✗ } else if (s->c.mv_type == MV_TYPE_16X16 ||
1481 ✗ (s->c.mv_type == MV_TYPE_FIELD && field_pic)) {
1482 ✗ motion_x = s->c.mv[dir][0][0];
1483 ✗ motion_y = s->c.mv[dir][0][1];
1484 } else { /* if ((s->c.mv_type == MV_TYPE_FIELD) || (s->c.mv_type == MV_TYPE_16X8)) */
1485 ✗ motion_x = s->c.mv[dir][i][0];
1486 ✗ motion_y = s->c.mv[dir][i][1];
1487 }
1488
1489 ✗ s->c.cur_pic.motion_val[dir][xy][0] = motion_x;
1490 ✗ s->c.cur_pic.motion_val[dir][xy][1] = motion_y;
1491 ✗ s->c.cur_pic.motion_val[dir][xy + 1][0] = motion_x;
1492 ✗ s->c.cur_pic.motion_val[dir][xy + 1][1] = motion_y;
1493 ✗ s->c.cur_pic.ref_index [dir][b8_xy] =
1494 ✗ s->c.cur_pic.ref_index [dir][b8_xy + 1] = s->c.field_select[dir][i];
1495 av_assert2(s->c.field_select[dir][i] == 0 ||
1496 s->c.field_select[dir][i] == 1);
1497 }
1498 ✗ xy += wrap;
1499 ✗ b8_xy += 2;
1500 }
1501 }
1502
1503 2914161 s->c.dest[0] += 16 >> lowres;
1504 2914161 s->c.dest[1] +=(16 >> lowres) >> s->c.chroma_x_shift;
1505 2914161 s->c.dest[2] +=(16 >> lowres) >> s->c.chroma_x_shift;
1506
1507 2914161 ff_mpv_reconstruct_mb(&s->c, s->block);
1508
1509
2/2
✓ Branch 0 taken 86257 times.
✓ Branch 1 taken 2827904 times.
 2914161 if (++s->c.mb_x >= s->c.mb_width) {
1510 86257 const int mb_size = 16 >> s->c.avctx->lowres;
1511 int left;
1512
1513 86257 ff_mpeg_draw_horiz_band(&s->c, mb_size * (s->c.mb_y >> field_pic), mb_size);
1514
1515 86257 s->c.mb_x = 0;
1516 86257 s->c.mb_y += 1 << field_pic;
1517
1518
2/2
✓ Branch 0 taken 4857 times.
✓ Branch 1 taken 81400 times.
 86257 if (s->c.mb_y >= s->c.mb_height) {
1519 4857 int left = get_bits_left(&s->gb);
1520 9975 int is_d10 = s->c.chroma_format == CHROMA_422 &&
1521
2/2
✓ Branch 0 taken 49 times.
✓ Branch 1 taken 212 times.
 261 s->c.pict_type == AV_PICTURE_TYPE_I &&
1522
2/4
✓ Branch 0 taken 49 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 49 times.
✗ Branch 3 not taken.
 49 avctx->profile == 0 && avctx->level == 5 &&
1523
2/2
✓ Branch 0 taken 25 times.
✓ Branch 1 taken 24 times.
 49 s->c.intra_dc_precision == 2 &&
1524
4/6
✓ Branch 0 taken 261 times.
✓ Branch 1 taken 4596 times.
✓ Branch 2 taken 25 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 25 times.
✗ Branch 5 not taken.
 5143 s->c.q_scale_type == 1 && s->c.alternate_scan == 0 &&
1525
1/2
✓ Branch 0 taken 25 times.
✗ Branch 1 not taken.
 25 s->c.progressive_frame == 0
1526 /* vbv_delay == 0xBBB || 0xE10 */;
1527
1528
4/4
✓ Branch 0 taken 1509 times.
✓ Branch 1 taken 3348 times.
✓ Branch 2 taken 1484 times.
✓ Branch 3 taken 25 times.
 4857 if (left >= 32 && !is_d10) {
1529 1484 GetBitContext gb = s->gb;
1530 1484 align_get_bits(&gb);
1531
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 1484 times.
 1484 if (show_bits(&gb, 24) == 0x060E2B) {
1532 ✗ av_log(avctx, AV_LOG_DEBUG, "Invalid MXF data found in video stream\n");
1533 ✗ is_d10 = 1;
1534 }
1535
3/4
✓ Branch 0 taken 1483 times.
✓ Branch 1 taken 1 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 1483 times.
 1484 if (left > 32 && show_bits_long(&gb, 32) == 0x201) {
1536 ✗ av_log(avctx, AV_LOG_DEBUG, "skipping m704 alpha (unsupported)\n");
1537 ✗ goto eos;
1538 }
1539 }
1540
1541
3/4
✓ Branch 0 taken 4857 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 4555 times.
✓ Branch 3 taken 302 times.
 4857 if (left < 0 ||
1542
1/4
✗ Branch 1 not taken.
✓ Branch 2 taken 4555 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
 4555 (left && show_bits(&s->gb, FFMIN(left, 23)) && !is_d10) ||
1543
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 4857 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 4857 ((avctx->err_recognition & (AV_EF_BITSTREAM | AV_EF_AGGRESSIVE)) && left > 8)) {
1544 ✗ av_log(avctx, AV_LOG_ERROR, "end mismatch left=%d %0X at %d %d\n",
1545 ✗ left, left>0 ? show_bits(&s->gb, FFMIN(left, 23)) : 0, s->c.mb_x, s->c.mb_y);
1546 ✗ return AVERROR_INVALIDDATA;
1547 } else
1548 4857 goto eos;
1549 }
1550 // There are some files out there which are missing the last slice
1551 // in cases where the slice is completely outside the visible
1552 // area, we detect this here instead of running into the end expecting
1553 // more data
1554 81400 left = get_bits_left(&s->gb);
1555
2/2
✓ Branch 0 taken 250 times.
✓ Branch 1 taken 81150 times.
 81400 if (s->c.mb_y >= ((s->c.height + 15) >> 4) &&
1556
2/4
✓ Branch 0 taken 250 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 250 times.
 250 !s->c.progressive_sequence &&
1557 ✗ left <= 25 &&
1558 ✗ left >= 0 &&
1559 ✗ mb_skip_run == -1 &&
1560 ✗ (!left || show_bits(&s->gb, left) == 0))
1561 ✗ goto eos;
1562
1563 81400 ff_init_block_index(&s->c);
1564 }
1565
1566 /* skip mb handling */
1567
2/2
✓ Branch 0 taken 2643221 times.
✓ Branch 1 taken 266083 times.
 2909304 if (mb_skip_run == -1) {
1568 /* read increment again */
1569 2643221 mb_skip_run = 0;
1570 2457 for (;;) {
1571 2645678 int code = get_vlc2(&s->gb, ff_mbincr_vlc,
1572 MBINCR_VLC_BITS, 2);
1573
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2645678 times.
 2645678 if (code < 0) {
1574 ✗ av_log(s->c.avctx, AV_LOG_ERROR, "mb incr damaged\n");
1575 ✗ return AVERROR_INVALIDDATA;
1576 }
1577
2/2
✓ Branch 0 taken 131758 times.
✓ Branch 1 taken 2513920 times.
 2645678 if (code >= 33) {
1578
2/2
✓ Branch 0 taken 2457 times.
✓ Branch 1 taken 129301 times.
 131758 if (code == 33) {
1579 2457 mb_skip_run += 33;
1580
1/2
✓ Branch 0 taken 129301 times.
✗ Branch 1 not taken.
 129301 } else if (code == 35) {
1581
2/4
✓ Branch 0 taken 129301 times.
✗ Branch 1 not taken.
✗ Branch 3 not taken.
✓ Branch 4 taken 129301 times.
 129301 if (mb_skip_run != 0 || show_bits(&s->gb, 15) != 0) {
1582 ✗ av_log(s->c.avctx, AV_LOG_ERROR, "slice mismatch\n");
1583 ✗ return AVERROR_INVALIDDATA;
1584 }
1585 129301 goto eos; /* end of slice */
1586 }
1587 /* otherwise, stuffing, nothing to do */
1588 } else {
1589 2513920 mb_skip_run += code;
1590 2513920 break;
1591 }
1592 }
1593
2/2
✓ Branch 0 taken 87812 times.
✓ Branch 1 taken 2426108 times.
 2513920 if (mb_skip_run) {
1594 int i;
1595
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 87812 times.
 87812 if (s->c.pict_type == AV_PICTURE_TYPE_I) {
1596 ✗ av_log(s->c.avctx, AV_LOG_ERROR,
1597 "skipped MB in I-frame at %d %d\n", s->c.mb_x, s->c.mb_y);
1598 ✗ return AVERROR_INVALIDDATA;
1599 }
1600
1601 /* skip mb */
1602 87812 s->c.mb_intra = 0;
1603
2/2
✓ Branch 0 taken 1053744 times.
✓ Branch 1 taken 87812 times.
 1141556 for (i = 0; i < 12; i++)
1604 1053744 s->c.block_last_index[i] = -1;
1605 87812 s->last_dc[0] = s->last_dc[1] = s->last_dc[2] = 128 << s->c.intra_dc_precision;
1606
2/2
✓ Branch 0 taken 67657 times.
✓ Branch 1 taken 20155 times.
 87812 if (s->c.picture_structure == PICT_FRAME)
1607 67657 s->c.mv_type = MV_TYPE_16X16;
1608 else
1609 20155 s->c.mv_type = MV_TYPE_FIELD;
1610
2/2
✓ Branch 0 taken 28508 times.
✓ Branch 1 taken 59304 times.
 87812 if (s->c.pict_type == AV_PICTURE_TYPE_P) {
1611 /* if P type, zero motion vector is implied */
1612 28508 s->c.mv_dir = MV_DIR_FORWARD;
1613 28508 s->c.mv[0][0][0] = s->c.mv[0][0][1] = 0;
1614 28508 s->c.last_mv[0][0][0] = s->c.last_mv[0][0][1] = 0;
1615 28508 s->c.last_mv[0][1][0] = s->c.last_mv[0][1][1] = 0;
1616 28508 s->c.field_select[0][0] = (s->c.picture_structure - 1) & 1;
1617 } else {
1618 /* if B type, reuse previous vectors and directions */
1619 59304 s->c.mv[0][0][0] = s->c.last_mv[0][0][0];
1620 59304 s->c.mv[0][0][1] = s->c.last_mv[0][0][1];
1621 59304 s->c.mv[1][0][0] = s->c.last_mv[1][0][0];
1622 59304 s->c.mv[1][0][1] = s->c.last_mv[1][0][1];
1623 59304 s->c.field_select[0][0] = (s->c.picture_structure - 1) & 1;
1624 59304 s->c.field_select[1][0] = (s->c.picture_structure - 1) & 1;
1625 }
1626 }
1627 }
1628 }
1629 134158 eos: // end of slice
1630
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 134158 times.
 134158 if (get_bits_left(&s->gb) < 0) {
1631 ✗ av_log(s->c.avctx, AV_LOG_ERROR, "overread %d\n", -get_bits_left(&s->gb));
1632 ✗ return AVERROR_INVALIDDATA;
1633 }
1634 134158 *buf += (get_bits_count(&s->gb) - 1) / 8;
1635 ff_dlog(s->c.avctx, "Slice start:%d %d end:%d %d\n", s->c.resync_mb_x, s->c.resync_mb_y, s->c.mb_x, s->c.mb_y);
1636 134158 return 0;
1637 }
1638
1639 8496 static int slice_decode_thread(AVCodecContext *c, void *arg)
1640 {
1641 8496 Mpeg12SliceContext *const s = *(void **) arg;
1642 8496 const uint8_t *buf = s->gb.buffer;
1643 8496 const uint8_t *end = buf + get_bits_bytesize(&s->gb, 0);
1644 8496 int mb_y = s->c.start_mb_y;
1645 8496 const int field_pic = s->c.picture_structure != PICT_FRAME;
1646
1647 8496 s->c.er.error_count = (3 * (s->c.end_mb_y - s->c.start_mb_y) * s->c.mb_width) >> field_pic;
1648
1649 19824 for (;;) {
1650 uint32_t start_code;
1651 int ret;
1652
1653 28320 ret = mpeg_decode_slice(s, mb_y, &buf, end - buf);
1654 28320 emms_c();
1655 ff_dlog(c, "ret:%d resync:%d/%d mb:%d/%d ts:%d/%d ec:%d\n",
1656 ret, s->c.resync_mb_x, s->c.resync_mb_y, s->c.mb_x, s->c.mb_y,
1657 s->c.start_mb_y, s->c.end_mb_y, s->c.er.error_count);
1658
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 28320 times.
 28320 if (ret < 0) {
1659 ✗ if (c->err_recognition & AV_EF_EXPLODE)
1660 8496 return ret;
1661 ✗ if (s->c.resync_mb_x >= 0 && s->c.resync_mb_y >= 0)
1662 ✗ ff_er_add_slice(&s->c.er, s->c.resync_mb_x, s->c.resync_mb_y,
1663 s->c.mb_x, s->c.mb_y,
1664 ER_AC_ERROR | ER_DC_ERROR | ER_MV_ERROR);
1665 } else {
1666 28320 ff_er_add_slice(&s->c.er, s->c.resync_mb_x, s->c.resync_mb_y,
1667 28320 s->c.mb_x - 1, s->c.mb_y,
1668 ER_AC_END | ER_DC_END | ER_MV_END);
1669 }
1670
1671
2/2
✓ Branch 0 taken 8496 times.
✓ Branch 1 taken 19824 times.
 28320 if (s->c.mb_y == s->c.end_mb_y)
1672 8496 return 0;
1673
1674 19824 start_code = -1;
1675 19824 buf = avpriv_find_start_code(buf, end, &start_code);
1676
2/4
✓ Branch 0 taken 19824 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 19824 times.
 19824 if (start_code < SLICE_MIN_START_CODE || start_code > SLICE_MAX_START_CODE)
1677 ✗ return AVERROR_INVALIDDATA;
1678 19824 mb_y = start_code - SLICE_MIN_START_CODE;
1679
2/4
✓ Branch 0 taken 19824 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 19824 times.
 19824 if (s->c.codec_id != AV_CODEC_ID_MPEG1VIDEO && s->c.mb_height > 2800/16)
1680 ✗ mb_y += (*buf&0xE0)<<2;
1681 19824 mb_y <<= field_pic;
1682
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 19824 times.
 19824 if (s->c.picture_structure == PICT_BOTTOM_FIELD)
1683 ✗ mb_y++;
1684
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 19824 times.
 19824 if (mb_y >= s->c.end_mb_y)
1685 ✗ return AVERROR_INVALIDDATA;
1686 }
1687 }
1688
1689 /**
1690 * Handle slice ends.
1691 * @return 1 if it seems to be the last slice
1692 */
1693 4426 static int slice_end(AVCodecContext *avctx, AVFrame *pict, int *got_output)
1694 {
1695 4426 Mpeg1Context *s1 = avctx->priv_data;
1696 4426 MPVContext *const s = &s1->slice.c;
1697
1698
3/4
✓ Branch 0 taken 4130 times.
✓ Branch 1 taken 296 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 4130 times.
 4426 if (!s->context_initialized || !s->cur_pic.ptr)
1699 296 return 0;
1700
1701
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 4130 times.
 4130 if (s->avctx->hwaccel) {
1702 ✗ int ret = FF_HW_SIMPLE_CALL(s->avctx, end_frame);
1703 ✗ if (ret < 0) {
1704 ✗ av_log(avctx, AV_LOG_ERROR,
1705 "hardware accelerator failed to decode picture\n");
1706 ✗ return ret;
1707 }
1708 }
1709
1710 /* end of slice reached */
1711
3/4
✓ Branch 0 taken 4129 times.
✓ Branch 1 taken 1 times.
✓ Branch 2 taken 4129 times.
✗ Branch 3 not taken.
 4130 if (/* s->mb_y << field_pic == s->mb_height && */ !s->first_field && !s1->first_slice) {
1712 /* end of image */
1713
1714 4129 ff_er_frame_end(&s->er, NULL);
1715
1716 4129 ff_mpv_frame_end(s);
1717
1718
4/4
✓ Branch 0 taken 2578 times.
✓ Branch 1 taken 1551 times.
✓ Branch 2 taken 562 times.
✓ Branch 3 taken 2016 times.
 4129 if (s->pict_type == AV_PICTURE_TYPE_B || s->low_delay) {
1719 2113 int ret = av_frame_ref(pict, s->cur_pic.ptr->f);
1720
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2113 times.
 2113 if (ret < 0)
1721 ✗ return ret;
1722 2113 ff_print_debug_info(s, s->cur_pic.ptr, pict);
1723 2113 ff_mpv_export_qp_table(s, pict, s->cur_pic.ptr, FF_MPV_QSCALE_TYPE_MPEG2);
1724 2113 *got_output = 1;
1725 } else {
1726 /* latency of 1 frame for I- and P-frames */
1727
4/4
✓ Branch 0 taken 1956 times.
✓ Branch 1 taken 60 times.
✓ Branch 2 taken 1953 times.
✓ Branch 3 taken 3 times.
 2016 if (s->last_pic.ptr && !s->last_pic.ptr->dummy) {
1728 1953 int ret = av_frame_ref(pict, s->last_pic.ptr->f);
1729
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1953 times.
 1953 if (ret < 0)
1730 ✗ return ret;
1731 1953 ff_print_debug_info(s, s->last_pic.ptr, pict);
1732 1953 ff_mpv_export_qp_table(s, pict, s->last_pic.ptr, FF_MPV_QSCALE_TYPE_MPEG2);
1733 1953 *got_output = 1;
1734 }
1735 }
1736
1737 4129 return 1;
1738 } else {
1739 1 return 0;
1740 }
1741 }
1742
1743 831 static int mpeg1_decode_sequence(AVCodecContext *avctx,
1744 const uint8_t *buf, int buf_size)
1745 {
1746 831 Mpeg1Context *s1 = avctx->priv_data;
1747 831 MPVContext *const s = &s1->slice.c;
1748 831 GetBitContext gb0, *const gb = &gb0;
1749 int width, height;
1750 int i, v, j;
1751
1752 831 int ret = init_get_bits8(gb, buf, buf_size);
1753
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 831 times.
 831 if (ret < 0)
1754 ✗ return ret;
1755
1756 831 width = get_bits(gb, 12);
1757 831 height = get_bits(gb, 12);
1758
2/4
✓ Branch 0 taken 831 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 831 times.
 831 if (width == 0 || height == 0) {
1759 ✗ av_log(avctx, AV_LOG_WARNING,
1760 "Invalid horizontal or vertical size value.\n");
1761 ✗ if (avctx->err_recognition & (AV_EF_BITSTREAM | AV_EF_COMPLIANT))
1762 ✗ return AVERROR_INVALIDDATA;
1763 }
1764 831 s1->aspect_ratio_info = get_bits(gb, 4);
1765
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 831 times.
 831 if (s1->aspect_ratio_info == 0) {
1766 ✗ av_log(avctx, AV_LOG_ERROR, "aspect ratio has forbidden 0 value\n");
1767 ✗ if (avctx->err_recognition & (AV_EF_BITSTREAM | AV_EF_COMPLIANT))
1768 ✗ return AVERROR_INVALIDDATA;
1769 }
1770 831 s1->frame_rate_index = get_bits(gb, 4);
1771
2/4
✓ Branch 0 taken 831 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 831 times.
 831 if (s1->frame_rate_index == 0 || s1->frame_rate_index > 13) {
1772 ✗ av_log(avctx, AV_LOG_WARNING,
1773 "frame_rate_index %d is invalid\n", s1->frame_rate_index);
1774 ✗ s1->frame_rate_index = 1;
1775 }
1776 831 s1->bit_rate = get_bits(gb, 18) * 400;
1777
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 831 times.
 831 if (check_marker(s->avctx, gb, "in sequence header") == 0) {
1778 ✗ return AVERROR_INVALIDDATA;
1779 }
1780
1781 831 s->avctx->rc_buffer_size = get_bits(gb, 10) * 1024 * 16;
1782 831 skip_bits(gb, 1);
1783
1784 /* get matrix */
1785
2/2
✓ Branch 1 taken 30 times.
✓ Branch 2 taken 801 times.
 831 if (get_bits1(gb)) {
1786 30 load_matrix(s, gb, s->chroma_intra_matrix, s->intra_matrix, 1);
1787 } else {
1788
2/2
✓ Branch 0 taken 51264 times.
✓ Branch 1 taken 801 times.
 52065 for (i = 0; i < 64; i++) {
1789 51264 j = s->idsp.idct_permutation[i];
1790 51264 v = ff_mpeg1_default_intra_matrix[i];
1791 51264 s->intra_matrix[j] = v;
1792 51264 s->chroma_intra_matrix[j] = v;
1793 }
1794 }
1795
2/2
✓ Branch 1 taken 92 times.
✓ Branch 2 taken 739 times.
 831 if (get_bits1(gb)) {
1796 92 load_matrix(s, gb, s->chroma_inter_matrix, s->inter_matrix, 0);
1797 } else {
1798
2/2
✓ Branch 0 taken 47296 times.
✓ Branch 1 taken 739 times.
 48035 for (i = 0; i < 64; i++) {
1799 47296 int j = s->idsp.idct_permutation[i];
1800 47296 v = ff_mpeg1_default_non_intra_matrix[i];
1801 47296 s->inter_matrix[j] = v;
1802 47296 s->chroma_inter_matrix[j] = v;
1803 }
1804 }
1805
1806
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 831 times.
 831 if (show_bits(gb, 23) != 0) {
1807 ✗ av_log(s->avctx, AV_LOG_ERROR, "sequence header damaged\n");
1808 ✗ return AVERROR_INVALIDDATA;
1809 }
1810
1811 831 s->width = width;
1812 831 s->height = height;
1813
1814 /* We set MPEG-2 parameters so that it emulates MPEG-1. */
1815 831 s->progressive_sequence = 1;
1816 831 s->progressive_frame = 1;
1817 831 s->picture_structure = PICT_FRAME;
1818 831 s->first_field = 0;
1819 831 s->frame_pred_frame_dct = 1;
1820 831 s->chroma_format = CHROMA_420;
1821 831 s->codec_id =
1822 831 s->avctx->codec_id = AV_CODEC_ID_MPEG1VIDEO;
1823
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 831 times.
 831 if (s->avctx->flags & AV_CODEC_FLAG_LOW_DELAY)
1824 ✗ s->low_delay = 1;
1825
1826
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 831 times.
 831 if (s->avctx->debug & FF_DEBUG_PICT_INFO)
1827 ✗ av_log(s->avctx, AV_LOG_DEBUG, "vbv buffer: %d, bitrate:%"PRId64", aspect_ratio_info: %d \n",
1828 ✗ s->avctx->rc_buffer_size, s1->bit_rate, s1->aspect_ratio_info);
1829
1830 831 return 0;
1831 }
1832
1833 2 static int vcr2_init_sequence(AVCodecContext *avctx)
1834 {
1835 2 Mpeg1Context *s1 = avctx->priv_data;
1836 2 MPVContext *const s = &s1->slice.c;
1837 int i, v, ret;
1838
1839 /* start new MPEG-1 context decoding */
1840
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2 times.
 2 if (s->context_initialized)
1841 ✗ ff_mpv_common_end(s);
1842
1843 2 s->width = avctx->coded_width;
1844 2 s->height = avctx->coded_height;
1845 2 avctx->has_b_frames = 0; // true?
1846 2 s->low_delay = 1;
1847
1848 2 avctx->pix_fmt = mpeg_get_pixelformat(avctx);
1849
1850
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 2 times.
 2 if ((ret = ff_mpv_common_init(s)) < 0)
1851 ✗ return ret;
1852
1/2
✓ Branch 0 taken 2 times.
✗ Branch 1 not taken.
 2 if (!s->avctx->lowres)
1853
2/2
✓ Branch 0 taken 2 times.
✓ Branch 1 taken 2 times.
 4 for (int i = 0; i < s->slice_context_count; i++)
1854 2 ff_mpv_framesize_disable(&s->thread_context[i]->sc);
1855
1856
2/2
✓ Branch 0 taken 128 times.
✓ Branch 1 taken 2 times.
 130 for (i = 0; i < 64; i++) {
1857 128 int j = s->idsp.idct_permutation[i];
1858 128 v = ff_mpeg1_default_intra_matrix[i];
1859 128 s->intra_matrix[j] = v;
1860 128 s->chroma_intra_matrix[j] = v;
1861
1862 128 v = ff_mpeg1_default_non_intra_matrix[i];
1863 128 s->inter_matrix[j] = v;
1864 128 s->chroma_inter_matrix[j] = v;
1865 }
1866
1867 2 s->progressive_sequence = 1;
1868 2 s->progressive_frame = 1;
1869 2 s->picture_structure = PICT_FRAME;
1870 2 s->first_field = 0;
1871 2 s->frame_pred_frame_dct = 1;
1872 2 s->chroma_format = CHROMA_420;
1873
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2 times.
 2 if (s->codec_tag == AV_RL32("BW10")) {
1874 ✗ s->codec_id = s->avctx->codec_id = AV_CODEC_ID_MPEG1VIDEO;
1875 } else {
1876 2 s->codec_id = s->avctx->codec_id = AV_CODEC_ID_MPEG2VIDEO;
1877 }
1878 2 s1->save_progressive_seq = s->progressive_sequence;
1879 2 s1->save_chroma_format = s->chroma_format;
1880 2 return 0;
1881 }
1882
1883 963 static void mpeg_set_cc_format(AVCodecContext *avctx, enum Mpeg2ClosedCaptionsFormat format,
1884 const char *label)
1885 {
1886 963 Mpeg1Context *s1 = avctx->priv_data;
1887
1888 av_assert2(format != CC_FORMAT_AUTO);
1889
1890
2/2
✓ Branch 0 taken 13 times.
✓ Branch 1 taken 950 times.
 963 if (!s1->cc_format) {
1891 13 s1->cc_format = format;
1892
1893 13 av_log(avctx, AV_LOG_DEBUG, "CC: first seen substream is %s format\n", label);
1894 }
1895
1896 #if FF_API_CODEC_PROPS
1897 FF_DISABLE_DEPRECATION_WARNINGS
1898 963 avctx->properties |= FF_CODEC_PROPERTY_CLOSED_CAPTIONS;
1899 FF_ENABLE_DEPRECATION_WARNINGS
1900 #endif
1901 963 }
1902
1903 1035 static int mpeg_decode_a53_cc(AVCodecContext *avctx,
1904 const uint8_t *p, int buf_size)
1905 {
1906 1035 Mpeg1Context *s1 = avctx->priv_data;
1907
1908
5/6
✓ Branch 0 taken 950 times.
✓ Branch 1 taken 85 times.
✓ Branch 2 taken 544 times.
✓ Branch 3 taken 406 times.
✓ Branch 4 taken 629 times.
✗ Branch 5 not taken.
 1035 if ((!s1->cc_format || s1->cc_format == CC_FORMAT_A53_PART4) &&
1909 629 buf_size >= 6 &&
1910
5/8
✓ Branch 0 taken 553 times.
✓ Branch 1 taken 76 times.
✓ Branch 2 taken 553 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 553 times.
✗ Branch 5 not taken.
✓ Branch 6 taken 553 times.
✗ Branch 7 not taken.
 629 p[0] == 'G' && p[1] == 'A' && p[2] == '9' && p[3] == '4' &&
1911
2/4
✓ Branch 0 taken 553 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 553 times.
✗ Branch 3 not taken.
 553 p[4] == 3 && (p[5] & 0x40)) {
1912 /* extract A53 Part 4 CC data */
1913 553 int cc_count = p[5] & 0x1f;
1914
2/4
✓ Branch 0 taken 553 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 553 times.
✗ Branch 3 not taken.
 553 if (cc_count > 0 && buf_size >= 7 + cc_count * 3) {
1915
2/2
✓ Branch 0 taken 8 times.
✓ Branch 1 taken 545 times.
 553 int old_size = s1->a53_buf_ref ? s1->a53_buf_ref->size : 0;
1916 553 const uint64_t new_size = (old_size + cc_count
1917 553 * UINT64_C(3));
1918 int ret;
1919
1920
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 553 times.
 553 if (new_size > 3*A53_MAX_CC_COUNT)
1921 ✗ return AVERROR(EINVAL);
1922
1923 553 ret = av_buffer_realloc(&s1->a53_buf_ref, new_size);
1924
1/2
✓ Branch 0 taken 553 times.
✗ Branch 1 not taken.
 553 if (ret >= 0)
1925 553 memcpy(s1->a53_buf_ref->data + old_size, p + 7, cc_count * UINT64_C(3));
1926
1927 553 mpeg_set_cc_format(avctx, CC_FORMAT_A53_PART4, "A/53 Part 4");
1928 }
1929 553 return 1;
1930
4/6
✓ Branch 0 taken 406 times.
✓ Branch 1 taken 76 times.
✓ Branch 2 taken 406 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 482 times.
✗ Branch 5 not taken.
 482 } else if ((!s1->cc_format || s1->cc_format == CC_FORMAT_SCTE20) &&
1931 482 buf_size >= 2 &&
1932
3/4
✓ Branch 0 taken 410 times.
✓ Branch 1 taken 72 times.
✓ Branch 2 taken 410 times.
✗ Branch 3 not taken.
 482 p[0] == 0x03 && (p[1]&0x7f) == 0x01) {
1933 /* extract SCTE-20 CC data */
1934 GetBitContext gb;
1935 410 int cc_count = 0;
1936 int i, ret;
1937
1938 410 ret = init_get_bits8(&gb, p + 2, buf_size - 2);
1939
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 410 times.
 410 if (ret < 0)
1940 ✗ return ret;
1941 410 cc_count = get_bits(&gb, 5);
1942
1/2
✓ Branch 0 taken 410 times.
✗ Branch 1 not taken.
 410 if (cc_count > 0) {
1943
2/2
✓ Branch 0 taken 4 times.
✓ Branch 1 taken 406 times.
 410 int old_size = s1->a53_buf_ref ? s1->a53_buf_ref->size : 0;
1944 410 const uint64_t new_size = (old_size + cc_count
1945 410 * UINT64_C(3));
1946
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 410 times.
 410 if (new_size > 3*A53_MAX_CC_COUNT)
1947 ✗ return AVERROR(EINVAL);
1948
1949 410 ret = av_buffer_realloc(&s1->a53_buf_ref, new_size);
1950
1/2
✓ Branch 0 taken 410 times.
✗ Branch 1 not taken.
 410 if (ret >= 0) {
1951 uint8_t field, cc1, cc2;
1952 410 uint8_t *cap = s1->a53_buf_ref->data + old_size;
1953
1954 410 memset(cap, 0, cc_count * 3);
1955
3/4
✓ Branch 0 taken 820 times.
✓ Branch 1 taken 410 times.
✓ Branch 3 taken 820 times.
✗ Branch 4 not taken.
 1230 for (i = 0; i < cc_count && get_bits_left(&gb) >= 26; i++) {
1956 820 skip_bits(&gb, 2); // priority
1957 820 field = get_bits(&gb, 2);
1958 820 skip_bits(&gb, 5); // line_offset
1959 820 cc1 = get_bits(&gb, 8);
1960 820 cc2 = get_bits(&gb, 8);
1961 820 skip_bits(&gb, 1); // marker
1962
1963
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 820 times.
 820 if (!field) { // forbidden
1964 ✗ cap[0] = cap[1] = cap[2] = 0x00;
1965 } else {
1966 820 field = (field == 2 ? 1 : 0);
1967
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 820 times.
 820 if (!s1->slice.c.top_field_first) field = !field;
1968 820 cap[0] = 0x04 | field;
1969 820 cap[1] = ff_reverse[cc1];
1970 820 cap[2] = ff_reverse[cc2];
1971 }
1972 820 cap += 3;
1973 }
1974 }
1975
1976 410 mpeg_set_cc_format(avctx, CC_FORMAT_SCTE20, "SCTE-20");
1977 }
1978 410 return 1;
1979
2/6
✗ Branch 0 not taken.
✓ Branch 1 taken 72 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✓ Branch 4 taken 72 times.
✗ Branch 5 not taken.
 72 } else if ((!s1->cc_format || s1->cc_format == CC_FORMAT_DVD) &&
1980 72 buf_size >= 11 &&
1981
1/8
✗ Branch 0 not taken.
✓ Branch 1 taken 72 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.
 72 p[0] == 'C' && p[1] == 'C' && p[2] == 0x01 && p[3] == 0xf8) {
1982 /* extract DVD CC data
1983 *
1984 * uint32_t user_data_start_code 0x000001B2 (big endian)
1985 * uint16_t user_identifier 0x4343 "CC"
1986 * uint8_t user_data_type_code 0x01
1987 * uint8_t caption_block_size 0xF8
1988 * uint8_t
1989 * bit 7 caption_odd_field_first 1=odd field (CC1/CC2) first 0=even field (CC3/CC4) first
1990 * bit 6 caption_filler 0
1991 * bit 5:1 caption_block_count number of caption blocks (pairs of caption words = frames). Most DVDs use 15 per start of GOP.
1992 * bit 0 caption_extra_field_added 1=one additional caption word
1993 *
1994 * struct caption_field_block {
1995 * uint8_t
1996 * bit 7:1 caption_filler 0x7F (all 1s)
1997 * bit 0 caption_field_odd 1=odd field (this is CC1/CC2) 0=even field (this is CC3/CC4)
1998 * uint8_t caption_first_byte
1999 * uint8_t caption_second_byte
2000 * } caption_block[(caption_block_count * 2) + caption_extra_field_added];
2001 *
2002 * Some DVDs encode caption data for both fields with caption_field_odd=1. The only way to decode the fields
2003 * correctly is to start on the field indicated by caption_odd_field_first and count between odd/even fields.
2004 * Don't assume that the first caption word is the odd field. There do exist MPEG files in the wild that start
2005 * on the even field. There also exist DVDs in the wild that encode an odd field count and the
2006 * caption_extra_field_added/caption_odd_field_first bits change per packet to allow that. */
2007 ✗ int cc_count = 0;
2008 int i, ret;
2009 // There is a caption count field in the data, but it is often
2010 // incorrect. So count the number of captions present.
2011 ✗ for (i = 5; i + 6 <= buf_size && ((p[i] & 0xfe) == 0xfe); i += 6)
2012 ✗ cc_count++;
2013 // Transform the DVD format into A53 Part 4 format
2014 ✗ if (cc_count > 0) {
2015 ✗ int old_size = s1->a53_buf_ref ? s1->a53_buf_ref->size : 0;
2016 ✗ const uint64_t new_size = (old_size + cc_count
2017 ✗ * UINT64_C(6));
2018 ✗ if (new_size > 3*A53_MAX_CC_COUNT)
2019 ✗ return AVERROR(EINVAL);
2020
2021 ✗ ret = av_buffer_realloc(&s1->a53_buf_ref, new_size);
2022 ✗ if (ret >= 0) {
2023 ✗ uint8_t field1 = !!(p[4] & 0x80);
2024 ✗ uint8_t *cap = s1->a53_buf_ref->data + old_size;
2025 ✗ p += 5;
2026 ✗ for (i = 0; i < cc_count; i++) {
2027 ✗ cap[0] = (p[0] == 0xff && field1) ? 0xfc : 0xfd;
2028 ✗ cap[1] = p[1];
2029 ✗ cap[2] = p[2];
2030 ✗ cap[3] = (p[3] == 0xff && !field1) ? 0xfc : 0xfd;
2031 ✗ cap[4] = p[4];
2032 ✗ cap[5] = p[5];
2033 ✗ cap += 6;
2034 ✗ p += 6;
2035 }
2036 }
2037
2038 ✗ mpeg_set_cc_format(avctx, CC_FORMAT_DVD, "DVD");
2039 }
2040 ✗ return 1;
2041
2/6
✗ Branch 0 not taken.
✓ Branch 1 taken 72 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✓ Branch 4 taken 72 times.
✗ Branch 5 not taken.
 72 } else if ((!s1->cc_format || s1->cc_format == CC_FORMAT_DISH) &&
2042 72 buf_size >= 12 &&
2043
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 72 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 72 p[0] == 0x05 && p[1] == 0x02) {
2044 /* extract Dish Network CC data */
2045 ✗ const uint8_t cc_header = 0xf8 | 0x04 /* valid */ | 0x00 /* line 21 field 1 */;
2046 ✗ uint8_t cc_data[4] = {0};
2047 ✗ int cc_count = 0;
2048 ✗ uint8_t cc_type = p[7];
2049 ✗ p += 8;
2050 ✗ buf_size -= 8;
2051
2052 ✗ if (cc_type == 0x05 && buf_size >= 7) {
2053 ✗ cc_type = p[6];
2054 ✗ p += 7;
2055 ✗ buf_size -= 7;
2056 }
2057
2058 ✗ if (cc_type == 0x02 && buf_size >= 4) { /* 2-byte caption, can be repeated */
2059 ✗ cc_count = 1;
2060 ✗ cc_data[0] = p[1];
2061 ✗ cc_data[1] = p[2];
2062 ✗ cc_type = p[3];
2063
2064 /* Only repeat characters when the next type flag
2065 * is 0x04 and the characters are repeatable (i.e., less than
2066 * 32 with the parity stripped).
2067 */
2068 ✗ if (cc_type == 0x04 && (cc_data[0] & 0x7f) < 32) {
2069 ✗ cc_count = 2;
2070 ✗ cc_data[2] = cc_data[0];
2071 ✗ cc_data[3] = cc_data[1];
2072 }
2073 ✗ } else if (cc_type == 0x04 && buf_size >= 5) { /* 4-byte caption, not repeated */
2074 ✗ cc_count = 2;
2075 ✗ cc_data[0] = p[1];
2076 ✗ cc_data[1] = p[2];
2077 ✗ cc_data[2] = p[3];
2078 ✗ cc_data[3] = p[4];
2079 }
2080
2081 ✗ if (cc_count > 0) {
2082 int ret;
2083 ✗ int old_size = s1->a53_buf_ref ? s1->a53_buf_ref->size : 0;
2084 ✗ const uint64_t new_size = (old_size + cc_count * UINT64_C(3));
2085 ✗ if (new_size > 3 * A53_MAX_CC_COUNT)
2086 ✗ return AVERROR(EINVAL);
2087
2088 ✗ ret = av_buffer_realloc(&s1->a53_buf_ref, new_size);
2089 ✗ if (ret >= 0) {
2090 ✗ uint8_t *cap = s1->a53_buf_ref->data + old_size;
2091 ✗ cap[0] = cc_header;
2092 ✗ cap[1] = cc_data[0];
2093 ✗ cap[2] = cc_data[1];
2094 ✗ if (cc_count == 2) {
2095 ✗ cap[3] = cc_header;
2096 ✗ cap[4] = cc_data[2];
2097 ✗ cap[5] = cc_data[3];
2098 }
2099 }
2100
2101 ✗ mpeg_set_cc_format(avctx, CC_FORMAT_DISH, "Dish Network");
2102 }
2103 ✗ return 1;
2104 }
2105 72 return 0;
2106 }
2107
2108 1447 static void mpeg_decode_user_data(AVCodecContext *avctx,
2109 const uint8_t *p, int buf_size)
2110 {
2111 1447 const uint8_t *buf_end = p + buf_size;
2112 1447 Mpeg1Context *s1 = avctx->priv_data;
2113
2114 #if 0
2115 int i;
2116 for(i=0; !(!p[i-2] && !p[i-1] && p[i]==1) && i<buf_size; i++){
2117 av_log(avctx, AV_LOG_ERROR, "%c", p[i]);
2118 }
2119 av_log(avctx, AV_LOG_ERROR, "\n");
2120 #endif
2121
2122
1/2
✓ Branch 0 taken 1447 times.
✗ Branch 1 not taken.
 1447 if (buf_size > 29){
2123 int i;
2124
2/2
✓ Branch 0 taken 28940 times.
✓ Branch 1 taken 1447 times.
 30387 for(i=0; i<20; i++)
2125
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 28940 times.
 28940 if (!memcmp(p+i, "\0TMPGEXS\0", 9)){
2126 ✗ s1->tmpgexs = 1;
2127 }
2128 }
2129 /* we parse the DTG active format information */
2130
1/2
✓ Branch 0 taken 1447 times.
✗ Branch 1 not taken.
 1447 if (buf_end - p >= 5 &&
2131
5/8
✓ Branch 0 taken 412 times.
✓ Branch 1 taken 1035 times.
✓ Branch 2 taken 412 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 412 times.
✗ Branch 5 not taken.
✓ Branch 6 taken 412 times.
✗ Branch 7 not taken.
 1859 p[0] == 'D' && p[1] == 'T' && p[2] == 'G' && p[3] == '1') {
2132 412 int flags = p[4];
2133 412 p += 5;
2134
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 412 times.
 412 if (flags & 0x80) {
2135 /* skip event id */
2136 ✗ p += 2;
2137 }
2138
1/2
✓ Branch 0 taken 412 times.
✗ Branch 1 not taken.
 412 if (flags & 0x40) {
2139
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 412 times.
 412 if (buf_end - p < 1)
2140 ✗ return;
2141 412 s1->has_afd = 1;
2142 412 s1->afd = p[0] & 0x0f;
2143 }
2144
1/2
✓ Branch 0 taken 1035 times.
✗ Branch 1 not taken.
 1035 } else if (buf_end - p >= 6 &&
2145
1/8
✗ Branch 0 not taken.
✓ Branch 1 taken 1035 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.
 1035 p[0] == 'J' && p[1] == 'P' && p[2] == '3' && p[3] == 'D' &&
2146 ✗ p[4] == 0x03) { // S3D_video_format_length
2147 // the 0x7F mask ignores the reserved_bit value
2148 ✗ const uint8_t S3D_video_format_type = p[5] & 0x7F;
2149
2150 ✗ if (S3D_video_format_type == 0x03 ||
2151 ✗ S3D_video_format_type == 0x04 ||
2152 ✗ S3D_video_format_type == 0x08 ||
2153 S3D_video_format_type == 0x23) {
2154
2155 ✗ s1->has_stereo3d = 1;
2156
2157 ✗ switch (S3D_video_format_type) {
2158 ✗ case 0x03:
2159 ✗ s1->stereo3d_type = AV_STEREO3D_SIDEBYSIDE;
2160 ✗ break;
2161 ✗ case 0x04:
2162 ✗ s1->stereo3d_type = AV_STEREO3D_TOPBOTTOM;
2163 ✗ break;
2164 ✗ case 0x08:
2165 ✗ s1->stereo3d_type = AV_STEREO3D_2D;
2166 ✗ break;
2167 ✗ case 0x23:
2168 ✗ s1->stereo3d_type = AV_STEREO3D_SIDEBYSIDE_QUINCUNX;
2169 ✗ break;
2170 }
2171 }
2172
2/2
✓ Branch 1 taken 963 times.
✓ Branch 2 taken 72 times.
 1035 } else if (mpeg_decode_a53_cc(avctx, p, buf_size)) {
2173 963 return;
2174 }
2175 }
2176
2177 574 static int mpeg_decode_gop(AVCodecContext *avctx,
2178 const uint8_t *buf, int buf_size)
2179 {
2180 574 Mpeg1Context *s1 = avctx->priv_data;
2181 574 MPVContext *const s = &s1->slice.c;
2182 574 GetBitContext gb0, *const gb = &gb0;
2183 int broken_link;
2184 int64_t tc;
2185
2186 574 int ret = init_get_bits8(gb, buf, buf_size);
2187
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 574 times.
 574 if (ret < 0)
2188 ✗ return ret;
2189
2190 574 tc = s1->timecode_frame_start = get_bits(gb, 25);
2191
2192 574 s1->closed_gop = get_bits1(gb);
2193 /* broken_link indicates that after editing the
2194 * reference frames of the first B-Frames after GOP I-Frame
2195 * are missing (open gop) */
2196 574 broken_link = get_bits1(gb);
2197
2198
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 574 times.
 574 if (s->avctx->debug & FF_DEBUG_PICT_INFO) {
2199 char tcbuf[AV_TIMECODE_STR_SIZE];
2200 ✗ av_timecode_make_mpeg_tc_string(tcbuf, tc);
2201 ✗ av_log(s->avctx, AV_LOG_DEBUG,
2202 "GOP (%s) closed_gop=%d broken_link=%d\n",
2203 tcbuf, s1->closed_gop, broken_link);
2204 }
2205
2206 574 return 0;
2207 }
2208
2209 4426 static void mpeg12_execute_slice_threads(AVCodecContext *avctx,
2210 Mpeg1Context *const s)
2211 {
2212
2/2
✓ Branch 0 taken 950 times.
✓ Branch 1 taken 3476 times.
 4426 if (HAVE_THREADS && (avctx->active_thread_type & FF_THREAD_SLICE) &&
2213
1/2
✓ Branch 0 taken 950 times.
✗ Branch 1 not taken.
 950 !avctx->hwaccel) {
2214 950 MPVContext *const s2 = &s->slice.c;
2215 950 int error_count = 0;
2216
2217 950 avctx->execute(avctx, slice_decode_thread,
2218 950 s2->mpeg12_contexts, NULL,
2219 s->slice_count, sizeof(s2->mpeg12_contexts[0]));
2220
2221
2/2
✓ Branch 0 taken 8496 times.
✓ Branch 1 taken 950 times.
 9446 for (int i = 0; i < s->slice_count; i++) {
2222 8496 MpegEncContext *const slice = s2->thread_context[i];
2223 8496 int slice_err = atomic_load_explicit(&slice->er.error_count,
2224 memory_order_relaxed);
2225 // error_count can get set to INT_MAX on serious errors.
2226 // So use saturated addition.
2227
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 8496 times.
 8496 if ((unsigned)slice_err > INT_MAX - error_count) {
2228 ✗ error_count = INT_MAX;
2229 ✗ break;
2230 }
2231 8496 error_count += slice_err;
2232 }
2233 950 atomic_store_explicit(&s2->er.error_count, error_count,
2234 memory_order_relaxed);
2235 }
2236 4426 }
2237
2238 4941 static int decode_chunks(AVCodecContext *avctx, AVFrame *picture,
2239 int *got_output, const uint8_t *buf, int buf_size)
2240 {
2241 4941 Mpeg1Context *s = avctx->priv_data;
2242 4941 MPVContext *const s2 = &s->slice.c;
2243 4941 const uint8_t *buf_ptr = buf;
2244 4941 const uint8_t *buf_end = buf + buf_size;
2245 int ret, input_size;
2246 4941 int last_code = 0, skip_frame = 0;
2247 4941 int picture_start_code_seen = 0;
2248
2249 232987 for (;;) {
2250 /* find next start code */
2251 237928 uint32_t start_code = -1;
2252 237928 buf_ptr = avpriv_find_start_code(buf_ptr, buf_end, &start_code);
2253
2/2
✓ Branch 0 taken 4762 times.
✓ Branch 1 taken 233166 times.
 237928 if (start_code > 0x1ff) {
2254
2/2
✓ Branch 0 taken 4426 times.
✓ Branch 1 taken 336 times.
 4762 if (!skip_frame) {
2255 4426 mpeg12_execute_slice_threads(avctx, s);
2256
2257 4426 ret = slice_end(avctx, picture, got_output);
2258
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 4426 times.
 4426 if (ret < 0)
2259 4941 return ret;
2260 }
2261 4762 s2->pict_type = 0;
2262
2263
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 4762 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 4762 if (avctx->err_recognition & AV_EF_EXPLODE && s2->er.error_count)
2264 ✗ return AVERROR_INVALIDDATA;
2265
2266 4762 return FFMAX(0, buf_ptr - buf);
2267 }
2268
2269 233166 input_size = buf_end - buf_ptr;
2270
2271
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 233166 times.
 233166 if (avctx->debug & FF_DEBUG_STARTCODE)
2272 ✗ av_log(avctx, AV_LOG_DEBUG, "%3"PRIX32" at %td left %d\n",
2273 start_code, buf_ptr - buf, input_size);
2274
2275 /* prepare data for next start code */
2276
6/6
✓ Branch 0 taken 831 times.
✓ Branch 1 taken 5519 times.
✓ Branch 2 taken 6770 times.
✓ Branch 3 taken 1447 times.
✓ Branch 4 taken 574 times.
✓ Branch 5 taken 218025 times.
 233166 switch (start_code) {
2277 831 case SEQ_START_CODE:
2278
1/2
✓ Branch 0 taken 831 times.
✗ Branch 1 not taken.
 831 if (last_code == 0) {
2279 831 mpeg1_decode_sequence(avctx, buf_ptr, input_size);
2280
2/2
✓ Branch 0 taken 565 times.
✓ Branch 1 taken 266 times.
 831 if (buf != avctx->extradata)
2281 565 s->sync = 1;
2282 } else {
2283 ✗ av_log(avctx, AV_LOG_ERROR,
2284 "ignoring SEQ_START_CODE after %X\n", last_code);
2285 ✗ if (avctx->err_recognition & AV_EF_EXPLODE)
2286 ✗ return AVERROR_INVALIDDATA;
2287 }
2288 831 break;
2289
2290 5519 case PICTURE_START_CODE:
2291
3/4
✓ Branch 0 taken 874 times.
✓ Branch 1 taken 4645 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 874 times.
 5519 if (picture_start_code_seen && s2->picture_structure == PICT_FRAME) {
2292 /* If it's a frame picture, there can't be more than one picture header.
2293 Yet, it does happen and we need to handle it. */
2294 ✗ av_log(avctx, AV_LOG_WARNING, "ignoring extra picture following a frame-picture\n");
2295 ✗ break;
2296 }
2297 5519 picture_start_code_seen = 1;
2298
2299
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 5519 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 5519 if (buf == avctx->extradata && avctx->codec_tag == AV_RL32("AVmp")) {
2300 ✗ av_log(avctx, AV_LOG_WARNING, "ignoring picture start code in AVmp extradata\n");
2301 ✗ break;
2302 }
2303
2304
3/4
✓ Branch 0 taken 5340 times.
✓ Branch 1 taken 179 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 5340 times.
 5519 if (s2->width <= 0 || s2->height <= 0) {
2305 179 av_log(avctx, AV_LOG_ERROR, "Invalid frame dimensions %dx%d.\n",
2306 s2->width, s2->height);
2307 179 return AVERROR_INVALIDDATA;
2308 }
2309
2310
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 5340 times.
 5340 if (s->tmpgexs){
2311 ✗ s2->intra_dc_precision= 3;
2312 ✗ s2->intra_matrix[0]= 1;
2313 }
2314
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 5340 times.
 5340 if (s->slice_count) {
2315 ✗ mpeg12_execute_slice_threads(avctx, s);
2316 ✗ s->slice_count = 0;
2317 }
2318
3/4
✓ Branch 0 taken 874 times.
✓ Branch 1 taken 4466 times.
✓ Branch 2 taken 874 times.
✗ Branch 3 not taken.
 5340 if (last_code == 0 || last_code == SLICE_MIN_START_CODE) {
2319 5340 ret = mpeg_decode_postinit(avctx);
2320
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 5340 times.
 5340 if (ret < 0) {
2321 ✗ av_log(avctx, AV_LOG_ERROR,
2322 "mpeg_decode_postinit() failure\n");
2323 ✗ return ret;
2324 }
2325
2326 /* We have a complete image: we try to decompress it. */
2327
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 5340 times.
 5340 if (mpeg1_decode_picture(avctx, buf_ptr, input_size) < 0)
2328 ✗ s2->pict_type = 0;
2329 5340 s->first_slice = 1;
2330 5340 last_code = PICTURE_START_CODE;
2331 } else {
2332 ✗ av_log(avctx, AV_LOG_ERROR,
2333 "ignoring pic after %X\n", last_code);
2334 ✗ if (avctx->err_recognition & AV_EF_EXPLODE)
2335 ✗ return AVERROR_INVALIDDATA;
2336 }
2337 5340 break;
2338 6770 case EXT_START_CODE: {
2339 6770 GetBitContext gb0, *const gb = &gb0;
2340
2341 6770 ret = init_get_bits8(gb, buf_ptr, input_size);
2342
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 6770 times.
 6770 if (ret < 0)
2343 ✗ return ret;
2344
2345
5/6
✓ Branch 1 taken 739 times.
✓ Branch 2 taken 103 times.
✓ Branch 3 taken 1055 times.
✓ Branch 4 taken 27 times.
✓ Branch 5 taken 4846 times.
✗ Branch 6 not taken.
 6770 switch (get_bits(gb, 4)) {
2346 739 case 0x1:
2347
1/2
✓ Branch 0 taken 739 times.
✗ Branch 1 not taken.
 739 if (last_code == 0) {
2348 739 mpeg_decode_sequence_extension(s, gb);
2349 } else {
2350 ✗ av_log(avctx, AV_LOG_ERROR,
2351 "ignoring seq ext after %X\n", last_code);
2352 ✗ if (avctx->err_recognition & AV_EF_EXPLODE)
2353 ✗ return AVERROR_INVALIDDATA;
2354 }
2355 739 break;
2356 103 case 0x2:
2357 103 mpeg_decode_sequence_display_extension(s, gb);
2358 103 break;
2359 1055 case 0x3:
2360 1055 mpeg_decode_quant_matrix_extension(s2, gb);
2361 1055 break;
2362 27 case 0x7:
2363 27 mpeg_decode_picture_display_extension(s, gb);
2364 27 break;
2365 4846 case 0x8:
2366
1/2
✓ Branch 0 taken 4846 times.
✗ Branch 1 not taken.
 4846 if (last_code == PICTURE_START_CODE) {
2367 4846 int ret = mpeg_decode_picture_coding_extension(s, gb);
2368
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 4846 times.
 4846 if (ret < 0)
2369 ✗ return ret;
2370 } else {
2371 ✗ av_log(avctx, AV_LOG_ERROR,
2372 "ignoring pic cod ext after %X\n", last_code);
2373 ✗ if (avctx->err_recognition & AV_EF_EXPLODE)
2374 ✗ return AVERROR_INVALIDDATA;
2375 }
2376 4846 break;
2377 }
2378 6770 break;
2379 }
2380 1447 case USER_START_CODE:
2381 1447 mpeg_decode_user_data(avctx, buf_ptr, input_size);
2382 1447 break;
2383 574 case GOP_START_CODE:
2384
1/2
✓ Branch 0 taken 574 times.
✗ Branch 1 not taken.
 574 if (last_code == 0) {
2385 574 s2->first_field = 0;
2386 574 ret = mpeg_decode_gop(avctx, buf_ptr, input_size);
2387
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 574 times.
 574 if (ret < 0)
2388 ✗ return ret;
2389 574 s->sync = 1;
2390 } else {
2391 ✗ av_log(avctx, AV_LOG_ERROR,
2392 "ignoring GOP_START_CODE after %X\n", last_code);
2393 ✗ if (avctx->err_recognition & AV_EF_EXPLODE)
2394 ✗ return AVERROR_INVALIDDATA;
2395 }
2396 574 break;
2397 218025 default:
2398
2/2
✓ Branch 0 taken 217101 times.
✓ Branch 1 taken 924 times.
 218025 if (start_code >= SLICE_MIN_START_CODE &&
2399
4/4
✓ Branch 0 taken 217096 times.
✓ Branch 1 taken 5 times.
✓ Branch 2 taken 5340 times.
✓ Branch 3 taken 211756 times.
 217101 start_code <= SLICE_MAX_START_CODE && last_code == PICTURE_START_CODE) {
2400
3/4
✓ Branch 0 taken 1501 times.
✓ Branch 1 taken 3839 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 1501 times.
 5340 if (s2->progressive_sequence && !s2->progressive_frame) {
2401 ✗ s2->progressive_frame = 1;
2402 ✗ av_log(s2->avctx, AV_LOG_ERROR,
2403 "interlaced frame in progressive sequence, ignoring\n");
2404 }
2405
2406
1/2
✓ Branch 0 taken 5340 times.
✗ Branch 1 not taken.
 5340 if (s2->picture_structure == 0 ||
2407
3/4
✓ Branch 0 taken 1508 times.
✓ Branch 1 taken 3832 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 1508 times.
 5340 (s2->progressive_frame && s2->picture_structure != PICT_FRAME)) {
2408 ✗ av_log(s2->avctx, AV_LOG_ERROR,
2409 "picture_structure %d invalid, ignoring\n",
2410 s2->picture_structure);
2411 ✗ s2->picture_structure = PICT_FRAME;
2412 }
2413
2414
3/4
✓ Branch 0 taken 1501 times.
✓ Branch 1 taken 3839 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 1501 times.
 5340 if (s2->progressive_sequence && !s2->frame_pred_frame_dct)
2415 ✗ av_log(s2->avctx, AV_LOG_WARNING, "invalid frame_pred_frame_dct\n");
2416
2417
2/2
✓ Branch 0 taken 3590 times.
✓ Branch 1 taken 1750 times.
 5340 if (s2->picture_structure == PICT_FRAME) {
2418 3590 s2->first_field = 0;
2419 3590 s2->v_edge_pos = 16 * s2->mb_height;
2420 } else {
2421 1750 s2->first_field ^= 1;
2422 1750 s2->v_edge_pos = 8 * s2->mb_height;
2423 1750 memset(s2->mbskip_table, 0, s2->mb_stride * s2->mb_height);
2424 }
2425 }
2426
2/2
✓ Branch 0 taken 217101 times.
✓ Branch 1 taken 924 times.
 218025 if (start_code >= SLICE_MIN_START_CODE &&
2427
4/4
✓ Branch 0 taken 217096 times.
✓ Branch 1 taken 5 times.
✓ Branch 2 taken 216916 times.
✓ Branch 3 taken 180 times.
 217101 start_code <= SLICE_MAX_START_CODE && last_code != 0) {
2428 216916 const int field_pic = s2->picture_structure != PICT_FRAME;
2429 216916 int mb_y = start_code - SLICE_MIN_START_CODE;
2430 216916 last_code = SLICE_MIN_START_CODE;
2431
3/4
✓ Branch 0 taken 209328 times.
✓ Branch 1 taken 7588 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 209328 times.
 216916 if (s2->codec_id != AV_CODEC_ID_MPEG1VIDEO && s2->mb_height > 2800/16)
2432 ✗ mb_y += (*buf_ptr&0xE0)<<2;
2433
2434 216916 mb_y <<= field_pic;
2435
2/2
✓ Branch 0 taken 14375 times.
✓ Branch 1 taken 202541 times.
 216916 if (s2->picture_structure == PICT_BOTTOM_FIELD)
2436 14375 mb_y++;
2437
2438
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 216916 times.
 216916 if (buf_end - buf_ptr < 2) {
2439 ✗ av_log(s2->avctx, AV_LOG_ERROR, "slice too small\n");
2440 ✗ return AVERROR_INVALIDDATA;
2441 }
2442
2443
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 216916 times.
 216916 if (mb_y >= s2->mb_height) {
2444 ✗ av_log(s2->avctx, AV_LOG_ERROR,
2445 "slice below image (%d >= %d)\n", mb_y, s2->mb_height);
2446 ✗ return AVERROR_INVALIDDATA;
2447 }
2448
2449
2/2
✓ Branch 0 taken 86368 times.
✓ Branch 1 taken 130548 times.
 216916 if (!s2->last_pic.ptr) {
2450 /* Skip B-frames if we do not have reference frames and
2451 * GOP is not closed. */
2452
2/2
✓ Branch 0 taken 792 times.
✓ Branch 1 taken 85576 times.
 86368 if (s2->pict_type == AV_PICTURE_TYPE_B) {
2453
1/2
✓ Branch 0 taken 792 times.
✗ Branch 1 not taken.
 792 if (!s->closed_gop) {
2454 792 skip_frame = 1;
2455 792 av_log(s2->avctx, AV_LOG_DEBUG,
2456 "Skipping B slice due to open GOP\n");
2457 792 break;
2458 }
2459 }
2460 }
2461
3/4
✓ Branch 0 taken 88103 times.
✓ Branch 1 taken 128021 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 88103 times.
 216124 if (s2->pict_type == AV_PICTURE_TYPE_I || (s2->avctx->flags2 & AV_CODEC_FLAG2_SHOW_ALL))
2462 128021 s->sync = 1;
2463
2/2
✓ Branch 0 taken 82033 times.
✓ Branch 1 taken 134091 times.
 216124 if (!s2->next_pic.ptr) {
2464 /* Skip P-frames if we do not have a reference frame or
2465 * we have an invalid header. */
2466
4/4
✓ Branch 0 taken 5286 times.
✓ Branch 1 taken 76747 times.
✓ Branch 2 taken 4356 times.
✓ Branch 3 taken 930 times.
 82033 if (s2->pict_type == AV_PICTURE_TYPE_P && !s->sync) {
2467 4356 skip_frame = 1;
2468 4356 av_log(s2->avctx, AV_LOG_DEBUG,
2469 "Skipping P slice due to !sync\n");
2470 4356 break;
2471 }
2472 }
2473
2/2
✓ Branch 0 taken 77599 times.
✓ Branch 1 taken 134169 times.
 211768 if ((avctx->skip_frame >= AVDISCARD_NONREF &&
2474
1/2
✓ Branch 0 taken 77599 times.
✗ Branch 1 not taken.
 77599 s2->pict_type == AV_PICTURE_TYPE_B) ||
2475
2/2
✓ Branch 0 taken 77599 times.
✓ Branch 1 taken 134169 times.
 211768 (avctx->skip_frame >= AVDISCARD_NONKEY &&
2476
2/2
✓ Branch 0 taken 76669 times.
✓ Branch 1 taken 930 times.
 77599 s2->pict_type != AV_PICTURE_TYPE_I) ||
2477
2/2
✓ Branch 0 taken 76669 times.
✓ Branch 1 taken 134169 times.
 210838 avctx->skip_frame >= AVDISCARD_ALL) {
2478 77599 skip_frame = 1;
2479 77599 break;
2480 }
2481
2482
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 134169 times.
 134169 if (!s2->context_initialized)
2483 ✗ break;
2484
2485
2/2
✓ Branch 0 taken 126896 times.
✓ Branch 1 taken 7273 times.
 134169 if (s2->codec_id == AV_CODEC_ID_MPEG2VIDEO) {
2486
1/2
✓ Branch 0 taken 126896 times.
✗ Branch 1 not taken.
 126896 if (mb_y < avctx->skip_top ||
2487
1/2
✓ Branch 0 taken 126896 times.
✗ Branch 1 not taken.
 126896 mb_y >= s2->mb_height - avctx->skip_bottom)
2488 break;
2489 }
2490
2491
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 134169 times.
 134169 if (!s2->pict_type) {
2492 ✗ av_log(avctx, AV_LOG_ERROR, "Missing picture start code\n");
2493 ✗ if (avctx->err_recognition & AV_EF_EXPLODE)
2494 ✗ return AVERROR_INVALIDDATA;
2495 ✗ break;
2496 }
2497
2498
2/2
✓ Branch 0 taken 4868 times.
✓ Branch 1 taken 129301 times.
 134169 if (s->first_slice) {
2499 4868 skip_frame = 0;
2500 4868 s->first_slice = 0;
2501
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 4868 times.
 4868 if ((ret = mpeg_field_start(s, buf, buf_size)) < 0)
2502 ✗ return ret;
2503 }
2504
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 134169 times.
 134169 if (!s2->cur_pic.ptr) {
2505 ✗ av_log(avctx, AV_LOG_ERROR,
2506 "current_picture not initialized\n");
2507 ✗ return AVERROR_INVALIDDATA;
2508 }
2509
2510 134169 if (HAVE_THREADS &&
2511
2/2
✓ Branch 0 taken 28320 times.
✓ Branch 1 taken 105849 times.
 134169 (avctx->active_thread_type & FF_THREAD_SLICE) &&
2512
1/2
✓ Branch 0 taken 28320 times.
✗ Branch 1 not taken.
 28320 !avctx->hwaccel) {
2513 28320 int threshold = (s2->mb_height * s->slice_count +
2514 28320 s2->slice_context_count / 2) /
2515 28320 s2->slice_context_count;
2516
2/2
✓ Branch 0 taken 8496 times.
✓ Branch 1 taken 19824 times.
 28320 if (threshold <= mb_y) {
2517 8496 Mpeg12SliceContext *const thread_context = s2->mpeg12_contexts[s->slice_count];
2518
2519 8496 thread_context->c.start_mb_y = mb_y;
2520 8496 thread_context->c.end_mb_y = s2->mb_height;
2521
2/2
✓ Branch 0 taken 7552 times.
✓ Branch 1 taken 944 times.
 8496 if (s->slice_count) {
2522 7552 s2->thread_context[s->slice_count - 1]->end_mb_y = mb_y;
2523 7552 ret = ff_update_duplicate_context(&thread_context->c, s2);
2524
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 7552 times.
 7552 if (ret < 0)
2525 ✗ return ret;
2526 }
2527 8496 ret = init_get_bits8(&thread_context->gb, buf_ptr, input_size);
2528
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 8496 times.
 8496 if (ret < 0)
2529 ✗ return ret;
2530 8496 s->slice_count++;
2531 }
2532 28320 buf_ptr += 2; // FIXME add minimum number of bytes per slice
2533 } else {
2534 105849 ret = mpeg_decode_slice(&s->slice, mb_y, &buf_ptr, input_size);
2535 105849 emms_c();
2536
2537
2/2
✓ Branch 0 taken 11 times.
✓ Branch 1 taken 105838 times.
 105849 if (ret < 0) {
2538
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 11 times.
 11 if (avctx->err_recognition & AV_EF_EXPLODE)
2539 ✗ return ret;
2540
2/4
✓ Branch 0 taken 11 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 11 times.
✗ Branch 3 not taken.
 11 if (s2->resync_mb_x >= 0 && s2->resync_mb_y >= 0)
2541 11 ff_er_add_slice(&s2->er, s2->resync_mb_x,
2542 s2->resync_mb_y, s2->mb_x, s2->mb_y,
2543 ER_AC_ERROR | ER_DC_ERROR | ER_MV_ERROR);
2544 } else {
2545 105838 ff_er_add_slice(&s2->er, s2->resync_mb_x,
2546 105838 s2->resync_mb_y, s2->mb_x - 1, s2->mb_y,
2547 ER_AC_END | ER_DC_END | ER_MV_END);
2548 }
2549 }
2550 }
2551 135278 break;
2552 }
2553 }
2554 }
2555
2556 4819 static int mpeg_decode_frame(AVCodecContext *avctx, AVFrame *picture,
2557 int *got_output, AVPacket *avpkt)
2558 {
2559 4819 const uint8_t *buf = avpkt->data;
2560 int ret;
2561 4819 int buf_size = avpkt->size;
2562 4819 Mpeg1Context *s = avctx->priv_data;
2563 4819 MPVContext *const s2 = &s->slice.c;
2564
2565
3/6
✓ Branch 0 taken 4675 times.
✓ Branch 1 taken 144 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 4675 times.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
 4819 if (buf_size == 0 || (buf_size == 4 && AV_RB32(buf) == SEQ_END_CODE)) {
2566 /* special case for last picture */
2567
4/4
✓ Branch 0 taken 131 times.
✓ Branch 1 taken 13 times.
✓ Branch 2 taken 62 times.
✓ Branch 3 taken 69 times.
 144 if (s2->low_delay == 0 && s2->next_pic.ptr) {
2568 62 int ret = av_frame_ref(picture, s2->next_pic.ptr->f);
2569
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 62 times.
 62 if (ret < 0)
2570 ✗ return ret;
2571
2572 62 ff_mpv_unref_picture(&s2->next_pic);
2573
2574 62 *got_output = 1;
2575 }
2576 144 return buf_size;
2577 }
2578
2579
2/2
✓ Branch 0 taken 478 times.
✓ Branch 1 taken 4197 times.
 4675 if (!s2->context_initialized &&
2580
3/4
✓ Branch 0 taken 476 times.
✓ Branch 1 taken 2 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 476 times.
 478 (s2->codec_tag == AV_RL32("VCR2") || s2->codec_tag == AV_RL32("BW10")))
2581 2 vcr2_init_sequence(avctx);
2582
2583 4675 s->slice_count = 0;
2584
2585
4/4
✓ Branch 0 taken 4272 times.
✓ Branch 1 taken 403 times.
✓ Branch 2 taken 266 times.
✓ Branch 3 taken 4006 times.
 4675 if (avctx->extradata && !s->extradata_decoded) {
2586 266 ret = decode_chunks(avctx, picture, got_output,
2587 266 avctx->extradata, avctx->extradata_size);
2588
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 266 times.
 266 if (*got_output) {
2589 ✗ av_log(avctx, AV_LOG_ERROR, "picture in extradata\n");
2590 ✗ av_frame_unref(picture);
2591 ✗ *got_output = 0;
2592 }
2593 266 s->extradata_decoded = 1;
2594
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 266 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 266 if (ret < 0 && (avctx->err_recognition & AV_EF_EXPLODE)) {
2595 ✗ ff_mpv_unref_picture(&s2->cur_pic);
2596 ✗ return ret;
2597 }
2598 }
2599
2600 4675 ret = decode_chunks(avctx, picture, got_output, buf, buf_size);
2601
4/4
✓ Branch 0 taken 4496 times.
✓ Branch 1 taken 179 times.
✓ Branch 2 taken 4066 times.
✓ Branch 3 taken 430 times.
 4675 if (ret<0 || *got_output) {
2602 4245 ff_mpv_unref_picture(&s2->cur_pic);
2603
2604
4/4
✓ Branch 0 taken 561 times.
✓ Branch 1 taken 3684 times.
✓ Branch 2 taken 382 times.
✓ Branch 3 taken 179 times.
 4245 if (s->timecode_frame_start != -1 && *got_output) {
2605 char tcbuf[AV_TIMECODE_STR_SIZE];
2606 382 AVFrameSideData *tcside = av_frame_new_side_data(picture,
2607 AV_FRAME_DATA_GOP_TIMECODE,
2608 sizeof(int64_t));
2609
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 382 times.
 382 if (!tcside)
2610 ✗ return AVERROR(ENOMEM);
2611 382 memcpy(tcside->data, &s->timecode_frame_start, sizeof(int64_t));
2612
2613 382 av_timecode_make_mpeg_tc_string(tcbuf, s->timecode_frame_start);
2614 382 av_dict_set(&picture->metadata, "timecode", tcbuf, 0);
2615
2616 382 s->timecode_frame_start = -1;
2617 }
2618 }
2619
2620 4675 return ret;
2621 }
2622
2623 ✗ static av_cold void flush(AVCodecContext *avctx)
2624 {
2625 ✗ Mpeg1Context *s = avctx->priv_data;
2626
2627 ✗ s->sync = 0;
2628 ✗ s->closed_gop = 0;
2629
2630 ✗ av_buffer_unref(&s->a53_buf_ref);
2631 ✗ ff_mpeg_flush(avctx);
2632 ✗ }
2633
2634 298 static av_cold int mpeg_decode_end(AVCodecContext *avctx)
2635 {
2636 298 Mpeg1Context *s = avctx->priv_data;
2637
2638 298 av_buffer_unref(&s->a53_buf_ref);
2639 298 return ff_mpv_decode_close(avctx);
2640 }
2641
2642 const FFCodec ff_mpeg1video_decoder = {
2643 .p.name = "mpeg1video",
2644 CODEC_LONG_NAME("MPEG-1 video"),
2645 .p.type = AVMEDIA_TYPE_VIDEO,
2646 .p.id = AV_CODEC_ID_MPEG1VIDEO,
2647 .priv_data_size = sizeof(Mpeg1Context),
2648 .init = mpeg_decode_init,
2649 .close = mpeg_decode_end,
2650 FF_CODEC_DECODE_CB(mpeg_decode_frame),
2651 .p.capabilities = AV_CODEC_CAP_DRAW_HORIZ_BAND | AV_CODEC_CAP_DR1 |
2652 AV_CODEC_CAP_DELAY | AV_CODEC_CAP_SLICE_THREADS,
2653 .caps_internal = FF_CODEC_CAP_SKIP_FRAME_FILL_PARAM,
2654 .flush = flush,
2655 .p.max_lowres = 3,
2656 .hw_configs = (const AVCodecHWConfigInternal *const []) {
2657 #if CONFIG_MPEG1_NVDEC_HWACCEL
2658 HWACCEL_NVDEC(mpeg1),
2659 #endif
2660 #if CONFIG_MPEG1_VDPAU_HWACCEL
2661 HWACCEL_VDPAU(mpeg1),
2662 #endif
2663 #if CONFIG_MPEG1_VIDEOTOOLBOX_HWACCEL
2664 HWACCEL_VIDEOTOOLBOX(mpeg1),
2665 #endif
2666 NULL
2667 },
2668 };
2669
2670 #define M2V_OFFSET(x) offsetof(Mpeg1Context, x)
2671 #define M2V_PARAM AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_DECODING_PARAM
2672
2673 static const AVOption mpeg2video_options[] = {
2674 { "cc_format", "extract a specific Closed Captions format",
2675 M2V_OFFSET(cc_format), AV_OPT_TYPE_INT, { .i64 = CC_FORMAT_AUTO },
2676 CC_FORMAT_AUTO, CC_FORMAT_DISH, M2V_PARAM, .unit = "cc_format" },
2677
2678 { "auto", "pick first seen CC substream", 0, AV_OPT_TYPE_CONST,
2679 { .i64 = CC_FORMAT_AUTO }, .flags = M2V_PARAM, .unit = "cc_format" },
2680 { "a53", "pick A/53 Part 4 CC substream", 0, AV_OPT_TYPE_CONST,
2681 { .i64 = CC_FORMAT_A53_PART4 }, .flags = M2V_PARAM, .unit = "cc_format" },
2682 { "scte20", "pick SCTE-20 CC substream", 0, AV_OPT_TYPE_CONST,
2683 { .i64 = CC_FORMAT_SCTE20 }, .flags = M2V_PARAM, .unit = "cc_format" },
2684 { "dvd", "pick DVD CC substream", 0, AV_OPT_TYPE_CONST,
2685 { .i64 = CC_FORMAT_DVD }, .flags = M2V_PARAM, .unit = "cc_format" },
2686 { "dish", "pick Dish Network CC substream", 0, AV_OPT_TYPE_CONST,
2687 { .i64 = CC_FORMAT_DISH }, .flags = M2V_PARAM, .unit = "cc_format" },
2688 { NULL }
2689 };
2690
2691 static const AVClass mpeg2video_class = {
2692 .class_name = "MPEG-2 video",
2693 .item_name = av_default_item_name,
2694 .option = mpeg2video_options,
2695 .version = LIBAVUTIL_VERSION_INT,
2696 .category = AV_CLASS_CATEGORY_DECODER,
2697 };
2698
2699 const FFCodec ff_mpeg2video_decoder = {
2700 .p.name = "mpeg2video",
2701 CODEC_LONG_NAME("MPEG-2 video"),
2702 .p.type = AVMEDIA_TYPE_VIDEO,
2703 .p.id = AV_CODEC_ID_MPEG2VIDEO,
2704 .p.priv_class = &mpeg2video_class,
2705 .priv_data_size = sizeof(Mpeg1Context),
2706 .init = mpeg_decode_init,
2707 .close = mpeg_decode_end,
2708 FF_CODEC_DECODE_CB(mpeg_decode_frame),
2709 .p.capabilities = AV_CODEC_CAP_DRAW_HORIZ_BAND | AV_CODEC_CAP_DR1 |
2710 AV_CODEC_CAP_DELAY | AV_CODEC_CAP_SLICE_THREADS,
2711 .caps_internal = FF_CODEC_CAP_SKIP_FRAME_FILL_PARAM,
2712 .flush = flush,
2713 .p.max_lowres = 3,
2714 .p.profiles = NULL_IF_CONFIG_SMALL(ff_mpeg2_video_profiles),
2715 .hw_configs = (const AVCodecHWConfigInternal *const []) {
2716 #if CONFIG_MPEG2_DXVA2_HWACCEL
2717 HWACCEL_DXVA2(mpeg2),
2718 #endif
2719 #if CONFIG_MPEG2_D3D11VA_HWACCEL
2720 HWACCEL_D3D11VA(mpeg2),
2721 #endif
2722 #if CONFIG_MPEG2_D3D11VA2_HWACCEL
2723 HWACCEL_D3D11VA2(mpeg2),
2724 #endif
2725 #if CONFIG_MPEG2_D3D12VA_HWACCEL
2726 HWACCEL_D3D12VA(mpeg2),
2727 #endif
2728 #if CONFIG_MPEG2_NVDEC_HWACCEL
2729 HWACCEL_NVDEC(mpeg2),
2730 #endif
2731 #if CONFIG_MPEG2_VAAPI_HWACCEL
2732 HWACCEL_VAAPI(mpeg2),
2733 #endif
2734 #if CONFIG_MPEG2_VDPAU_HWACCEL
2735 HWACCEL_VDPAU(mpeg2),
2736 #endif
2737 #if CONFIG_MPEG2_VIDEOTOOLBOX_HWACCEL
2738 HWACCEL_VIDEOTOOLBOX(mpeg2),
2739 #endif
2740 NULL
2741 },
2742 };
2743
2744 //legacy decoder
2745 const FFCodec ff_mpegvideo_decoder = {
2746 .p.name = "mpegvideo",
2747 CODEC_LONG_NAME("MPEG-1 video"),
2748 .p.type = AVMEDIA_TYPE_VIDEO,
2749 .p.id = AV_CODEC_ID_MPEG2VIDEO,
2750 .priv_data_size = sizeof(Mpeg1Context),
2751 .init = mpeg_decode_init,
2752 .close = mpeg_decode_end,
2753 FF_CODEC_DECODE_CB(mpeg_decode_frame),
2754 .p.capabilities = AV_CODEC_CAP_DRAW_HORIZ_BAND | AV_CODEC_CAP_DR1 |
2755 AV_CODEC_CAP_DELAY | AV_CODEC_CAP_SLICE_THREADS,
2756 .caps_internal = FF_CODEC_CAP_SKIP_FRAME_FILL_PARAM,
2757 .flush = flush,
2758 .p.max_lowres = 3,
2759 };
2760
2761 typedef struct IPUContext {
2762 Mpeg12SliceContext m;
2763
2764 int flags;
2765 } IPUContext;
2766
2767 ✗ static int ipu_decode_frame(AVCodecContext *avctx, AVFrame *frame,
2768 int *got_frame, AVPacket *avpkt)
2769 {
2770 ✗ IPUContext *s = avctx->priv_data;
2771 ✗ MPVContext *const m = &s->m.c;
2772 ✗ GetBitContext *const gb = &s->m.gb;
2773 ✗ int16_t (*const block)[64] = s->m.block;
2774 int ret;
2775
2776 // Check for minimal intra MB size (considering mb header, luma & chroma dc VLC, ac EOB VLC)
2777 ✗ if (avpkt->size*8LL < (avctx->width+15)/16 * ((avctx->height+15)/16) * (2LL + 3*4 + 2*2 + 2*6))
2778 ✗ return AVERROR_INVALIDDATA;
2779
2780 ✗ ret = ff_get_buffer(avctx, frame, 0);
2781 ✗ if (ret < 0)
2782 ✗ return ret;
2783
2784 ✗ ret = init_get_bits8(gb, avpkt->data, avpkt->size);
2785 ✗ if (ret < 0)
2786 ✗ return ret;
2787
2788 ✗ s->flags = get_bits(gb, 8);
2789 ✗ m->intra_dc_precision = s->flags & 3;
2790 ✗ m->q_scale_type = !!(s->flags & 0x40);
2791 ✗ m->intra_vlc_format = !!(s->flags & 0x20);
2792 ✗ m->alternate_scan = !!(s->flags & 0x10);
2793
2794 ✗ ff_permute_scantable(m->intra_scantable.permutated,
2795 ✗ s->flags & 0x10 ? ff_alternate_vertical_scan : ff_zigzag_direct,
2796 ✗ m->idsp.idct_permutation);
2797
2798 ✗ s->m.last_dc[0] = s->m.last_dc[1] = s->m.last_dc[2] = 128 << (s->flags & 3);
2799 ✗ m->qscale = 1;
2800
2801 ✗ for (int y = 0; y < avctx->height; y += 16) {
2802 int intraquant;
2803
2804 ✗ for (int x = 0; x < avctx->width; x += 16) {
2805 ✗ if (x || y) {
2806 ✗ if (!get_bits1(gb))
2807 ✗ return AVERROR_INVALIDDATA;
2808 }
2809 ✗ if (get_bits1(gb)) {
2810 ✗ intraquant = 0;
2811 } else {
2812 ✗ if (!get_bits1(gb))
2813 ✗ return AVERROR_INVALIDDATA;
2814 ✗ intraquant = 1;
2815 }
2816
2817 ✗ if (s->flags & 4)
2818 ✗ skip_bits1(gb);
2819
2820 ✗ if (intraquant)
2821 ✗ m->qscale = mpeg_get_qscale(gb, m->q_scale_type);
2822
2823 ✗ memset(block, 0, 6 * sizeof(*block));
2824
2825 ✗ for (int n = 0; n < 6; n++) {
2826 ✗ if (s->flags & 0x80) {
2827 ✗ ret = ff_mpeg1_decode_block_intra(gb,
2828 ✗ m->intra_matrix,
2829 ✗ m->intra_scantable.permutated,
2830 ✗ s->m.last_dc, block[n],
2831 n, m->qscale);
2832 } else {
2833 ✗ ret = mpeg2_decode_block_intra(&s->m, block[n], n);
2834 }
2835
2836 ✗ if (ret < 0)
2837 ✗ return ret;
2838 }
2839
2840 ✗ m->idsp.idct_put(frame->data[0] + y * frame->linesize[0] + x,
2841 ✗ frame->linesize[0], block[0]);
2842 ✗ m->idsp.idct_put(frame->data[0] + y * frame->linesize[0] + x + 8,
2843 ✗ frame->linesize[0], block[1]);
2844 ✗ m->idsp.idct_put(frame->data[0] + (y + 8) * frame->linesize[0] + x,
2845 ✗ frame->linesize[0], block[2]);
2846 ✗ m->idsp.idct_put(frame->data[0] + (y + 8) * frame->linesize[0] + x + 8,
2847 ✗ frame->linesize[0], block[3]);
2848 ✗ m->idsp.idct_put(frame->data[1] + (y >> 1) * frame->linesize[1] + (x >> 1),
2849 ✗ frame->linesize[1], block[4]);
2850 ✗ m->idsp.idct_put(frame->data[2] + (y >> 1) * frame->linesize[2] + (x >> 1),
2851 ✗ frame->linesize[2], block[5]);
2852 }
2853 }
2854
2855 ✗ align_get_bits(gb);
2856 ✗ if (get_bits_left(gb) != 32)
2857 ✗ return AVERROR_INVALIDDATA;
2858
2859 ✗ *got_frame = 1;
2860
2861 ✗ return avpkt->size;
2862 }
2863
2864 ✗ static av_cold int ipu_decode_init(AVCodecContext *avctx)
2865 {
2866 ✗ IPUContext *s = avctx->priv_data;
2867 ✗ MPVContext *const m = &s->m.c;
2868
2869 ✗ avctx->pix_fmt = AV_PIX_FMT_YUV420P;
2870 ✗ m->avctx = avctx;
2871
2872 ✗ ff_idctdsp_init(&m->idsp, avctx);
2873 ✗ ff_mpeg12_init_vlcs();
2874
2875 ✗ for (int i = 0; i < 64; i++) {
2876 ✗ int j = m->idsp.idct_permutation[i];
2877 ✗ int v = ff_mpeg1_default_intra_matrix[i];
2878 ✗ m->intra_matrix[j] = v;
2879 ✗ m->chroma_intra_matrix[j] = v;
2880 }
2881
2882 ✗ return 0;
2883 }
2884
2885 const FFCodec ff_ipu_decoder = {
2886 .p.name = "ipu",
2887 CODEC_LONG_NAME("IPU Video"),
2888 .p.type = AVMEDIA_TYPE_VIDEO,
2889 .p.id = AV_CODEC_ID_IPU,
2890 .priv_data_size = sizeof(IPUContext),
2891 .init = ipu_decode_init,
2892 FF_CODEC_DECODE_CB(ipu_decode_frame),
2893 .p.capabilities = AV_CODEC_CAP_DR1,
2894 };
2895