FFmpeg coverage

Directory: ../../../ffmpeg/
File: src/libavcodec/cbs.c
Date: 2022-01-18 10:05:52
Exec Total Coverage
Lines: 297 520 57.1%
Branches: 145 360 40.3%
Line Branch Exec Source
1 /*
2 * This file is part of FFmpeg.
3 *
4 * FFmpeg is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU Lesser General Public
6 * License as published by the Free Software Foundation; either
7 * version 2.1 of the License, or (at your option) any later version.
8 *
9 * FFmpeg is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 * Lesser General Public License for more details.
13 *
14 * You should have received a copy of the GNU Lesser General Public
15 * License along with FFmpeg; if not, write to the Free Software
16 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17 */
18
19 #include <string.h>
20
21 #include "config.h"
22
23 #include "libavutil/avassert.h"
24 #include "libavutil/buffer.h"
25 #include "libavutil/common.h"
26 #include "libavutil/opt.h"
27
28 #include "avcodec.h"
29 #include "cbs.h"
30 #include "cbs_internal.h"
31
32
33 static const CodedBitstreamType *const cbs_type_table[] = {
34 #if CONFIG_CBS_AV1
35 &ff_cbs_type_av1,
36 #endif
37 #if CONFIG_CBS_H264
38 &ff_cbs_type_h264,
39 #endif
40 #if CONFIG_CBS_H265
41 &ff_cbs_type_h265,
42 #endif
43 #if CONFIG_CBS_JPEG
44 &ff_cbs_type_jpeg,
45 #endif
46 #if CONFIG_CBS_MPEG2
47 &ff_cbs_type_mpeg2,
48 #endif
49 #if CONFIG_CBS_VP9
50 &ff_cbs_type_vp9,
51 #endif
52 };
53
54 const enum AVCodecID ff_cbs_all_codec_ids[] = {
55 #if CONFIG_CBS_AV1
56 AV_CODEC_ID_AV1,
57 #endif
58 #if CONFIG_CBS_H264
59 AV_CODEC_ID_H264,
60 #endif
61 #if CONFIG_CBS_H265
62 AV_CODEC_ID_H265,
63 #endif
64 #if CONFIG_CBS_JPEG
65 AV_CODEC_ID_MJPEG,
66 #endif
67 #if CONFIG_CBS_MPEG2
68 AV_CODEC_ID_MPEG2VIDEO,
69 #endif
70 #if CONFIG_CBS_VP9
71 AV_CODEC_ID_VP9,
72 #endif
73 AV_CODEC_ID_NONE
74 };
75
76 204 int ff_cbs_init(CodedBitstreamContext **ctx_ptr,
77 enum AVCodecID codec_id, void *log_ctx)
78 {
79 CodedBitstreamContext *ctx;
80 const CodedBitstreamType *type;
81 int i;
82
83 204 type = NULL;
84
1/2
✓ Branch 0 taken 480 times.
✗ Branch 1 not taken.
 480 for (i = 0; i < FF_ARRAY_ELEMS(cbs_type_table); i++) {
85
2/2
✓ Branch 0 taken 204 times.
✓ Branch 1 taken 276 times.
 480 if (cbs_type_table[i]->codec_id == codec_id) {
86 204 type = cbs_type_table[i];
87 204 break;
88 }
89 }
90
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 204 times.
 204 if (!type)
91 ✗ return AVERROR(EINVAL);
92
93 204 ctx = av_mallocz(sizeof(*ctx));
94
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 204 times.
 204 if (!ctx)
95 ✗ return AVERROR(ENOMEM);
96
97 204 ctx->log_ctx = log_ctx;
98 204 ctx->codec = type; /* Must be before any error */
99
100
1/2
✓ Branch 0 taken 204 times.
✗ Branch 1 not taken.
 204 if (type->priv_data_size) {
101 204 ctx->priv_data = av_mallocz(ctx->codec->priv_data_size);
102
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 204 times.
 204 if (!ctx->priv_data) {
103 ✗ av_freep(&ctx);
104 ✗ return AVERROR(ENOMEM);
105 }
106
2/2
✓ Branch 0 taken 98 times.
✓ Branch 1 taken 106 times.
 204 if (type->priv_class) {
107 98 *(const AVClass **)ctx->priv_data = type->priv_class;
108 98 av_opt_set_defaults(ctx->priv_data);
109 }
110 }
111
112 204 ctx->decompose_unit_types = NULL;
113
114 204 ctx->trace_enable = 0;
115 204 ctx->trace_level = AV_LOG_TRACE;
116
117 204 *ctx_ptr = ctx;
118 204 return 0;
119 }
120
121 ✗ void ff_cbs_flush(CodedBitstreamContext *ctx)
122 {
123 ✗ if (ctx->codec->flush)
124 ✗ ctx->codec->flush(ctx);
125 }
126
127 204 void ff_cbs_close(CodedBitstreamContext **ctx_ptr)
128 {
129 204 CodedBitstreamContext *ctx = *ctx_ptr;
130
131
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 204 times.
 204 if (!ctx)
132 ✗ return;
133
134
2/2
✓ Branch 0 taken 170 times.
✓ Branch 1 taken 34 times.
 204 if (ctx->codec->close)
135 170 ctx->codec->close(ctx);
136
137 204 av_freep(&ctx->write_buffer);
138
139
3/4
✓ Branch 0 taken 98 times.
✓ Branch 1 taken 106 times.
✓ Branch 2 taken 98 times.
✗ Branch 3 not taken.
 204 if (ctx->codec->priv_class && ctx->priv_data)
140 98 av_opt_free(ctx->priv_data);
141
142 204 av_freep(&ctx->priv_data);
143 204 av_freep(ctx_ptr);
144 }
145
146 27287 static void cbs_unit_uninit(CodedBitstreamUnit *unit)
147 {
148 27287 av_buffer_unref(&unit->content_ref);
149 27287 unit->content = NULL;
150
151 27287 av_buffer_unref(&unit->data_ref);
152 27287 unit->data = NULL;
153 27287 unit->data_size = 0;
154 27287 unit->data_bit_padding = 0;
155 27287 }
156
157 6257 void ff_cbs_fragment_reset(CodedBitstreamFragment *frag)
158 {
159 int i;
160
161
2/2
✓ Branch 0 taken 27215 times.
✓ Branch 1 taken 6257 times.
 33472 for (i = 0; i < frag->nb_units; i++)
162 27215 cbs_unit_uninit(&frag->units[i]);
163 6257 frag->nb_units = 0;
164
165 6257 av_buffer_unref(&frag->data_ref);
166 6257 frag->data = NULL;
167 6257 frag->data_size = 0;
168 6257 frag->data_bit_padding = 0;
169 6257 }
170
171 135 void ff_cbs_fragment_free(CodedBitstreamFragment *frag)
172 {
173 135 ff_cbs_fragment_reset(frag);
174
175 135 av_freep(&frag->units);
176 135 frag->nb_units_allocated = 0;
177 135 }
178
179 6107 static int cbs_read_fragment_content(CodedBitstreamContext *ctx,
180 CodedBitstreamFragment *frag)
181 {
182 int err, i, j;
183
184
2/2
✓ Branch 0 taken 27274 times.
✓ Branch 1 taken 6106 times.
 33380 for (i = 0; i < frag->nb_units; i++) {
185 27274 CodedBitstreamUnit *unit = &frag->units[i];
186
187
2/2
✓ Branch 0 taken 1822 times.
✓ Branch 1 taken 25452 times.
 27274 if (ctx->decompose_unit_types) {
188
2/2
✓ Branch 0 taken 5477 times.
✓ Branch 1 taken 20 times.
 5497 for (j = 0; j < ctx->nb_decompose_unit_types; j++) {
189
2/2
✓ Branch 0 taken 1802 times.
✓ Branch 1 taken 3675 times.
 5477 if (ctx->decompose_unit_types[j] == unit->type)
190 1802 break;
191 }
192
2/2
✓ Branch 0 taken 20 times.
✓ Branch 1 taken 1802 times.
 1822 if (j >= ctx->nb_decompose_unit_types)
193 20 continue;
194 }
195
196 27254 av_buffer_unref(&unit->content_ref);
197 27254 unit->content = NULL;
198
199
2/4
✓ Branch 0 taken 27254 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 27254 times.
 27254 av_assert0(unit->data && unit->data_ref);
200
201 27254 err = ctx->codec->read_unit(ctx, unit);
202
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 27254 times.
 27254 if (err == AVERROR(ENOSYS)) {
203 ✗ av_log(ctx->log_ctx, AV_LOG_VERBOSE,
204 "Decomposition unimplemented for unit %d "
205 "(type %"PRIu32").\n", i, unit->type);
206
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 27254 times.
 27254 } else if (err == AVERROR(EAGAIN)) {
207 ✗ av_log(ctx->log_ctx, AV_LOG_VERBOSE,
208 "Skipping decomposition of unit %d "
209 "(type %"PRIu32").\n", i, unit->type);
210 ✗ av_buffer_unref(&unit->content_ref);
211 ✗ unit->content = NULL;
212
2/2
✓ Branch 0 taken 1 times.
✓ Branch 1 taken 27253 times.
 27254 } else if (err < 0) {
213 1 av_log(ctx->log_ctx, AV_LOG_ERROR, "Failed to read unit %d "
214 "(type %"PRIu32").\n", i, unit->type);
215 1 return err;
216 }
217 }
218
219 6106 return 0;
220 }
221
222 456 static int cbs_fill_fragment_data(CodedBitstreamFragment *frag,
223 const uint8_t *data, size_t size)
224 {
225
2/4
✓ Branch 0 taken 456 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 456 times.
 456 av_assert0(!frag->data && !frag->data_ref);
226
227 456 frag->data_ref =
228 456 av_buffer_alloc(size + AV_INPUT_BUFFER_PADDING_SIZE);
229
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 456 times.
 456 if (!frag->data_ref)
230 ✗ return AVERROR(ENOMEM);
231
232 456 frag->data = frag->data_ref->data;
233 456 frag->data_size = size;
234
235 456 memcpy(frag->data, data, size);
236 456 memset(frag->data + size, 0,
237 AV_INPUT_BUFFER_PADDING_SIZE);
238
239 456 return 0;
240 }
241
242 6107 static int cbs_read_data(CodedBitstreamContext *ctx,
243 CodedBitstreamFragment *frag,
244 AVBufferRef *buf,
245 const uint8_t *data, size_t size,
246 int header)
247 {
248 int err;
249
250
2/2
✓ Branch 0 taken 5651 times.
✓ Branch 1 taken 456 times.
 6107 if (buf) {
251 5651 frag->data_ref = av_buffer_ref(buf);
252
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 5651 times.
 5651 if (!frag->data_ref)
253 ✗ return AVERROR(ENOMEM);
254
255 5651 frag->data = (uint8_t *)data;
256 5651 frag->data_size = size;
257
258 } else {
259 456 err = cbs_fill_fragment_data(frag, data, size);
260
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 456 times.
 456 if (err < 0)
261 ✗ return err;
262 }
263
264 6107 err = ctx->codec->split_fragment(ctx, frag, header);
265
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 6107 times.
 6107 if (err < 0)
266 ✗ return err;
267
268 6107 return cbs_read_fragment_content(ctx, frag);
269 }
270
271 57 int ff_cbs_read_extradata(CodedBitstreamContext *ctx,
272 CodedBitstreamFragment *frag,
273 const AVCodecParameters *par)
274 {
275 114 return cbs_read_data(ctx, frag, NULL,
276 57 par->extradata,
277 57 par->extradata_size, 1);
278 }
279
280 24 int ff_cbs_read_extradata_from_codec(CodedBitstreamContext *ctx,
281 CodedBitstreamFragment *frag,
282 const AVCodecContext *avctx)
283 {
284 48 return cbs_read_data(ctx, frag, NULL,
285 24 avctx->extradata,
286 24 avctx->extradata_size, 1);
287 }
288
289 5651 int ff_cbs_read_packet(CodedBitstreamContext *ctx,
290 CodedBitstreamFragment *frag,
291 const AVPacket *pkt)
292 {
293 11302 return cbs_read_data(ctx, frag, pkt->buf,
294 5651 pkt->data, pkt->size, 0);
295 }
296
297 ✗ int ff_cbs_read_packet_side_data(CodedBitstreamContext *ctx,
298 CodedBitstreamFragment *frag,
299 const AVPacket *pkt)
300 {
301 size_t side_data_size;
302 const uint8_t *side_data =
303 ✗ av_packet_get_side_data(pkt, AV_PKT_DATA_NEW_EXTRADATA,
304 &side_data_size);
305
306 ✗ return cbs_read_data(ctx, frag, NULL,
307 side_data, side_data_size, 1);
308 }
309
310 375 int ff_cbs_read(CodedBitstreamContext *ctx,
311 CodedBitstreamFragment *frag,
312 const uint8_t *data, size_t size)
313 {
314 375 return cbs_read_data(ctx, frag, NULL,
315 data, size, 0);
316 }
317
318 /**
319 * Allocate a new internal data buffer of the given size in the unit.
320 *
321 * The data buffer will have input padding.
322 */
323 24609 static int cbs_alloc_unit_data(CodedBitstreamUnit *unit,
324 size_t size)
325 {
326
2/4
✓ Branch 0 taken 24609 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 24609 times.
 24609 av_assert0(!unit->data && !unit->data_ref);
327
328 24609 unit->data_ref = av_buffer_alloc(size + AV_INPUT_BUFFER_PADDING_SIZE);
329
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 24609 times.
 24609 if (!unit->data_ref)
330 ✗ return AVERROR(ENOMEM);
331
332 24609 unit->data = unit->data_ref->data;
333 24609 unit->data_size = size;
334
335 24609 memset(unit->data + size, 0, AV_INPUT_BUFFER_PADDING_SIZE);
336
337 24609 return 0;
338 }
339
340 24609 static int cbs_write_unit_data(CodedBitstreamContext *ctx,
341 CodedBitstreamUnit *unit)
342 {
343 PutBitContext pbc;
344 int ret;
345
346
2/2
✓ Branch 0 taken 70 times.
✓ Branch 1 taken 24539 times.
 24609 if (!ctx->write_buffer) {
347 // Initial write buffer size is 1MB.
348 70 ctx->write_buffer_size = 1024 * 1024;
349
350 70 reallocate_and_try_again:
351 70 ret = av_reallocp(&ctx->write_buffer, ctx->write_buffer_size);
352
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 70 times.
 70 if (ret < 0) {
353 ✗ av_log(ctx->log_ctx, AV_LOG_ERROR, "Unable to allocate a "
354 "sufficiently large write buffer (last attempt "
355 "%"SIZE_SPECIFIER" bytes).\n", ctx->write_buffer_size);
356 ✗ return ret;
357 }
358 }
359
360 24609 init_put_bits(&pbc, ctx->write_buffer, ctx->write_buffer_size);
361
362 24609 ret = ctx->codec->write_unit(ctx, unit, &pbc);
363
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 24609 times.
 24609 if (ret < 0) {
364 ✗ if (ret == AVERROR(ENOSPC)) {
365 // Overflow.
366 ✗ if (ctx->write_buffer_size == INT_MAX / 8)
367 ✗ return AVERROR(ENOMEM);
368 ✗ ctx->write_buffer_size = FFMIN(2 * ctx->write_buffer_size, INT_MAX / 8);
369 ✗ goto reallocate_and_try_again;
370 }
371 // Write failed for some other reason.
372 ✗ return ret;
373 }
374
375 // Overflow but we didn't notice.
376
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 24609 times.
 24609 av_assert0(put_bits_count(&pbc) <= 8 * ctx->write_buffer_size);
377
378
2/2
✓ Branch 1 taken 52 times.
✓ Branch 2 taken 24557 times.
 24609 if (put_bits_count(&pbc) % 8)
379 52 unit->data_bit_padding = 8 - put_bits_count(&pbc) % 8;
380 else
381 24557 unit->data_bit_padding = 0;
382
383 24609 flush_put_bits(&pbc);
384
385 24609 ret = cbs_alloc_unit_data(unit, put_bytes_output(&pbc));
386
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 24609 times.
 24609 if (ret < 0)
387 ✗ return ret;
388
389 24609 memcpy(unit->data, ctx->write_buffer, unit->data_size);
390
391 24609 return 0;
392 }
393
394 5096 int ff_cbs_write_fragment_data(CodedBitstreamContext *ctx,
395 CodedBitstreamFragment *frag)
396 {
397 int err, i;
398
399
2/2
✓ Branch 0 taken 24609 times.
✓ Branch 1 taken 5096 times.
 29705 for (i = 0; i < frag->nb_units; i++) {
400 24609 CodedBitstreamUnit *unit = &frag->units[i];
401
402
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 24609 times.
 24609 if (!unit->content)
403 ✗ continue;
404
405 24609 av_buffer_unref(&unit->data_ref);
406 24609 unit->data = NULL;
407
408 24609 err = cbs_write_unit_data(ctx, unit);
409
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 24609 times.
 24609 if (err < 0) {
410 ✗ av_log(ctx->log_ctx, AV_LOG_ERROR, "Failed to write unit %d "
411 "(type %"PRIu32").\n", i, unit->type);
412 ✗ return err;
413 }
414
2/4
✓ Branch 0 taken 24609 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 24609 times.
 24609 av_assert0(unit->data && unit->data_ref);
415 }
416
417 5096 av_buffer_unref(&frag->data_ref);
418 5096 frag->data = NULL;
419
420 5096 err = ctx->codec->assemble_fragment(ctx, frag);
421
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 5096 times.
 5096 if (err < 0) {
422 ✗ av_log(ctx->log_ctx, AV_LOG_ERROR, "Failed to assemble fragment.\n");
423 ✗ return err;
424 }
425
2/4
✓ Branch 0 taken 5096 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 5096 times.
 5096 av_assert0(frag->data && frag->data_ref);
426
427 5096 return 0;
428 }
429
430 55 int ff_cbs_write_extradata(CodedBitstreamContext *ctx,
431 AVCodecParameters *par,
432 CodedBitstreamFragment *frag)
433 {
434 int err;
435
436 55 err = ff_cbs_write_fragment_data(ctx, frag);
437
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 55 times.
 55 if (err < 0)
438 ✗ return err;
439
440 55 av_freep(&par->extradata);
441
442 55 par->extradata = av_malloc(frag->data_size +
443 AV_INPUT_BUFFER_PADDING_SIZE);
444
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 55 times.
 55 if (!par->extradata)
445 ✗ return AVERROR(ENOMEM);
446
447 55 memcpy(par->extradata, frag->data, frag->data_size);
448 55 memset(par->extradata + frag->data_size, 0,
449 AV_INPUT_BUFFER_PADDING_SIZE);
450 55 par->extradata_size = frag->data_size;
451
452 55 return 0;
453 }
454
455 5041 int ff_cbs_write_packet(CodedBitstreamContext *ctx,
456 AVPacket *pkt,
457 CodedBitstreamFragment *frag)
458 {
459 AVBufferRef *buf;
460 int err;
461
462 5041 err = ff_cbs_write_fragment_data(ctx, frag);
463
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 5041 times.
 5041 if (err < 0)
464 ✗ return err;
465
466 5041 buf = av_buffer_ref(frag->data_ref);
467
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 5041 times.
 5041 if (!buf)
468 ✗ return AVERROR(ENOMEM);
469
470 5041 av_buffer_unref(&pkt->buf);
471
472 5041 pkt->buf = buf;
473 5041 pkt->data = frag->data;
474 5041 pkt->size = frag->data_size;
475
476 5041 return 0;
477 }
478
479
480 67437 void ff_cbs_trace_header(CodedBitstreamContext *ctx,
481 const char *name)
482 {
483
1/2
✓ Branch 0 taken 67437 times.
✗ Branch 1 not taken.
 67437 if (!ctx->trace_enable)
484 67437 return;
485
486 ✗ av_log(ctx->log_ctx, ctx->trace_level, "%s\n", name);
487 }
488
489 ✗ void ff_cbs_trace_syntax_element(CodedBitstreamContext *ctx, int position,
490 const char *str, const int *subscripts,
491 const char *bits, int64_t value)
492 {
493 char name[256];
494 size_t name_len, bits_len;
495 int pad, subs, i, j, k, n;
496
497 ✗ if (!ctx->trace_enable)
498 ✗ return;
499
500 ✗ av_assert0(value >= INT_MIN && value <= UINT32_MAX);
501
502 ✗ subs = subscripts ? subscripts[0] : 0;
503 ✗ n = 0;
504 ✗ for (i = j = 0; str[i];) {
505 ✗ if (str[i] == '[') {
506 ✗ if (n < subs) {
507 ✗ ++n;
508 ✗ k = snprintf(name + j, sizeof(name) - j, "[%d", subscripts[n]);
509 ✗ av_assert0(k > 0 && j + k < sizeof(name));
510 ✗ j += k;
511 ✗ for (++i; str[i] && str[i] != ']'; i++);
512 ✗ av_assert0(str[i] == ']');
513 } else {
514 ✗ while (str[i] && str[i] != ']')
515 ✗ name[j++] = str[i++];
516 ✗ av_assert0(str[i] == ']');
517 }
518 } else {
519 ✗ av_assert0(j + 1 < sizeof(name));
520 ✗ name[j++] = str[i++];
521 }
522 }
523 ✗ av_assert0(j + 1 < sizeof(name));
524 ✗ name[j] = 0;
525 ✗ av_assert0(n == subs);
526
527 ✗ name_len = strlen(name);
528 ✗ bits_len = strlen(bits);
529
530 ✗ if (name_len + bits_len > 60)
531 ✗ pad = bits_len + 2;
532 else
533 ✗ pad = 61 - name_len;
534
535 ✗ av_log(ctx->log_ctx, ctx->trace_level, "%-10d %s%*s = %"PRId64"\n",
536 position, name, pad, bits, value);
537 }
538
539 1475559 int ff_cbs_read_unsigned(CodedBitstreamContext *ctx, GetBitContext *gbc,
540 int width, const char *name,
541 const int *subscripts, uint32_t *write_to,
542 uint32_t range_min, uint32_t range_max)
543 {
544 uint32_t value;
545 int position;
546
547
2/4
✓ Branch 0 taken 1475559 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 1475559 times.
 1475559 av_assert0(width > 0 && width <= 32);
548
549
2/2
✓ Branch 1 taken 1 times.
✓ Branch 2 taken 1475558 times.
 1475559 if (get_bits_left(gbc) < width) {
550 1 av_log(ctx->log_ctx, AV_LOG_ERROR, "Invalid value at "
551 "%s: bitstream ended.\n", name);
552 1 return AVERROR_INVALIDDATA;
553 }
554
555
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1475558 times.
 1475558 if (ctx->trace_enable)
556 ✗ position = get_bits_count(gbc);
557
558 1475558 value = get_bits_long(gbc, width);
559
560
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1475558 times.
 1475558 if (ctx->trace_enable) {
561 char bits[33];
562 int i;
563 ✗ for (i = 0; i < width; i++)
564 ✗ bits[i] = value >> (width - i - 1) & 1 ? '1' : '0';
565 ✗ bits[i] = 0;
566
567 ✗ ff_cbs_trace_syntax_element(ctx, position, name, subscripts,
568 bits, value);
569 }
570
571
2/4
✓ Branch 0 taken 1475558 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 1475558 times.
 1475558 if (value < range_min || value > range_max) {
572 ✗ av_log(ctx->log_ctx, AV_LOG_ERROR, "%s out of range: "
573 "%"PRIu32", but must be in [%"PRIu32",%"PRIu32"].\n",
574 name, value, range_min, range_max);
575 ✗ return AVERROR_INVALIDDATA;
576 }
577
578 1475558 *write_to = value;
579 1475558 return 0;
580 }
581
582 478008 int ff_cbs_write_unsigned(CodedBitstreamContext *ctx, PutBitContext *pbc,
583 int width, const char *name,
584 const int *subscripts, uint32_t value,
585 uint32_t range_min, uint32_t range_max)
586 {
587
2/4
✓ Branch 0 taken 478008 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 478008 times.
 478008 av_assert0(width > 0 && width <= 32);
588
589
2/4
✓ Branch 0 taken 478008 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 478008 times.
 478008 if (value < range_min || value > range_max) {
590 ✗ av_log(ctx->log_ctx, AV_LOG_ERROR, "%s out of range: "
591 "%"PRIu32", but must be in [%"PRIu32",%"PRIu32"].\n",
592 name, value, range_min, range_max);
593 ✗ return AVERROR_INVALIDDATA;
594 }
595
596
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 478008 times.
 478008 if (put_bits_left(pbc) < width)
597 ✗ return AVERROR(ENOSPC);
598
599
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 478008 times.
 478008 if (ctx->trace_enable) {
600 char bits[33];
601 int i;
602 ✗ for (i = 0; i < width; i++)
603 ✗ bits[i] = value >> (width - i - 1) & 1 ? '1' : '0';
604 ✗ bits[i] = 0;
605
606 ✗ ff_cbs_trace_syntax_element(ctx, put_bits_count(pbc),
607 name, subscripts, bits, value);
608 }
609
610
2/2
✓ Branch 0 taken 477050 times.
✓ Branch 1 taken 958 times.
 478008 if (width < 32)
611 477050 put_bits(pbc, width, value);
612 else
613 958 put_bits32(pbc, value);
614
615 478008 return 0;
616 }
617
618 50 int ff_cbs_read_signed(CodedBitstreamContext *ctx, GetBitContext *gbc,
619 int width, const char *name,
620 const int *subscripts, int32_t *write_to,
621 int32_t range_min, int32_t range_max)
622 {
623 int32_t value;
624 int position;
625
626
2/4
✓ Branch 0 taken 50 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 50 times.
 50 av_assert0(width > 0 && width <= 32);
627
628
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 50 times.
 50 if (get_bits_left(gbc) < width) {
629 ✗ av_log(ctx->log_ctx, AV_LOG_ERROR, "Invalid value at "
630 "%s: bitstream ended.\n", name);
631 ✗ return AVERROR_INVALIDDATA;
632 }
633
634
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 50 times.
 50 if (ctx->trace_enable)
635 ✗ position = get_bits_count(gbc);
636
637 50 value = get_sbits_long(gbc, width);
638
639
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 50 times.
 50 if (ctx->trace_enable) {
640 char bits[33];
641 int i;
642 ✗ for (i = 0; i < width; i++)
643 ✗ bits[i] = value & (1U << (width - i - 1)) ? '1' : '0';
644 ✗ bits[i] = 0;
645
646 ✗ ff_cbs_trace_syntax_element(ctx, position, name, subscripts,
647 bits, value);
648 }
649
650
2/4
✓ Branch 0 taken 50 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 50 times.
 50 if (value < range_min || value > range_max) {
651 ✗ av_log(ctx->log_ctx, AV_LOG_ERROR, "%s out of range: "
652 "%"PRId32", but must be in [%"PRId32",%"PRId32"].\n",
653 name, value, range_min, range_max);
654 ✗ return AVERROR_INVALIDDATA;
655 }
656
657 50 *write_to = value;
658 50 return 0;
659 }
660
661 50 int ff_cbs_write_signed(CodedBitstreamContext *ctx, PutBitContext *pbc,
662 int width, const char *name,
663 const int *subscripts, int32_t value,
664 int32_t range_min, int32_t range_max)
665 {
666
2/4
✓ Branch 0 taken 50 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 50 times.
 50 av_assert0(width > 0 && width <= 32);
667
668
2/4
✓ Branch 0 taken 50 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 50 times.
 50 if (value < range_min || value > range_max) {
669 ✗ av_log(ctx->log_ctx, AV_LOG_ERROR, "%s out of range: "
670 "%"PRId32", but must be in [%"PRId32",%"PRId32"].\n",
671 name, value, range_min, range_max);
672 ✗ return AVERROR_INVALIDDATA;
673 }
674
675
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 50 times.
 50 if (put_bits_left(pbc) < width)
676 ✗ return AVERROR(ENOSPC);
677
678
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 50 times.
 50 if (ctx->trace_enable) {
679 char bits[33];
680 int i;
681 ✗ for (i = 0; i < width; i++)
682 ✗ bits[i] = value & (1U << (width - i - 1)) ? '1' : '0';
683 ✗ bits[i] = 0;
684
685 ✗ ff_cbs_trace_syntax_element(ctx, put_bits_count(pbc),
686 name, subscripts, bits, value);
687 }
688
689
1/2
✓ Branch 0 taken 50 times.
✗ Branch 1 not taken.
 50 if (width < 32)
690 50 put_sbits(pbc, width, value);
691 else
692 ✗ put_bits32(pbc, value);
693
694 50 return 0;
695 }
696
697
698 ✗ int ff_cbs_alloc_unit_content(CodedBitstreamUnit *unit,
699 size_t size,
700 void (*free)(void *opaque, uint8_t *data))
701 {
702 ✗ av_assert0(!unit->content && !unit->content_ref);
703
704 ✗ unit->content = av_mallocz(size);
705 ✗ if (!unit->content)
706 ✗ return AVERROR(ENOMEM);
707
708 ✗ unit->content_ref = av_buffer_create(unit->content, size,
709 free, NULL, 0);
710 ✗ if (!unit->content_ref) {
711 ✗ av_freep(&unit->content);
712 ✗ return AVERROR(ENOMEM);
713 }
714
715 ✗ return 0;
716 }
717
718 27287 static int cbs_insert_unit(CodedBitstreamFragment *frag,
719 int position)
720 {
721 CodedBitstreamUnit *units;
722
723
2/2
✓ Branch 0 taken 26910 times.
✓ Branch 1 taken 377 times.
 27287 if (frag->nb_units < frag->nb_units_allocated) {
724 26910 units = frag->units;
725
726
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 26910 times.
 26910 if (position < frag->nb_units)
727 ✗ memmove(units + position + 1, units + position,
728 ✗ (frag->nb_units - position) * sizeof(*units));
729 } else {
730 377 units = av_malloc_array(frag->nb_units*2 + 1, sizeof(*units));
731
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 377 times.
 377 if (!units)
732 ✗ return AVERROR(ENOMEM);
733
734 377 frag->nb_units_allocated = 2*frag->nb_units_allocated + 1;
735
736
2/2
✓ Branch 0 taken 242 times.
✓ Branch 1 taken 135 times.
 377 if (position > 0)
737 242 memcpy(units, frag->units, position * sizeof(*units));
738
739
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 377 times.
 377 if (position < frag->nb_units)
740 ✗ memcpy(units + position + 1, frag->units + position,
741 ✗ (frag->nb_units - position) * sizeof(*units));
742 }
743
744 27287 memset(units + position, 0, sizeof(*units));
745
746
2/2
✓ Branch 0 taken 377 times.
✓ Branch 1 taken 26910 times.
 27287 if (units != frag->units) {
747 377 av_free(frag->units);
748 377 frag->units = units;
749 }
750
751 27287 ++frag->nb_units;
752
753 27287 return 0;
754 }
755
756 13 int ff_cbs_insert_unit_content(CodedBitstreamFragment *frag,
757 int position,
758 CodedBitstreamUnitType type,
759 void *content,
760 AVBufferRef *content_buf)
761 {
762 CodedBitstreamUnit *unit;
763 AVBufferRef *content_ref;
764 int err;
765
766
1/2
✓ Branch 0 taken 13 times.
✗ Branch 1 not taken.
 13 if (position == -1)
767 13 position = frag->nb_units;
768
2/4
✓ Branch 0 taken 13 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 13 times.
 13 av_assert0(position >= 0 && position <= frag->nb_units);
769
770
1/2
✓ Branch 0 taken 13 times.
✗ Branch 1 not taken.
 13 if (content_buf) {
771 13 content_ref = av_buffer_ref(content_buf);
772
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 13 times.
 13 if (!content_ref)
773 ✗ return AVERROR(ENOMEM);
774 } else {
775 ✗ content_ref = NULL;
776 }
777
778 13 err = cbs_insert_unit(frag, position);
779
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 13 times.
 13 if (err < 0) {
780 ✗ av_buffer_unref(&content_ref);
781 ✗ return err;
782 }
783
784 13 unit = &frag->units[position];
785 13 unit->type = type;
786 13 unit->content = content;
787 13 unit->content_ref = content_ref;
788
789 13 return 0;
790 }
791
792 27274 int ff_cbs_insert_unit_data(CodedBitstreamFragment *frag,
793 int position,
794 CodedBitstreamUnitType type,
795 uint8_t *data, size_t data_size,
796 AVBufferRef *data_buf)
797 {
798 CodedBitstreamUnit *unit;
799 AVBufferRef *data_ref;
800 int err;
801
802
2/2
✓ Branch 0 taken 18431 times.
✓ Branch 1 taken 8843 times.
 27274 if (position == -1)
803 18431 position = frag->nb_units;
804
2/4
✓ Branch 0 taken 27274 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 27274 times.
 27274 av_assert0(position >= 0 && position <= frag->nb_units);
805
806
1/2
✓ Branch 0 taken 27274 times.
✗ Branch 1 not taken.
 27274 if (data_buf)
807 27274 data_ref = av_buffer_ref(data_buf);
808 else
809 ✗ data_ref = av_buffer_create(data, data_size, NULL, NULL, 0);
810
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 27274 times.
 27274 if (!data_ref) {
811 ✗ if (!data_buf)
812 ✗ av_free(data);
813 ✗ return AVERROR(ENOMEM);
814 }
815
816 27274 err = cbs_insert_unit(frag, position);
817
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 27274 times.
 27274 if (err < 0) {
818 ✗ av_buffer_unref(&data_ref);
819 ✗ return err;
820 }
821
822 27274 unit = &frag->units[position];
823 27274 unit->type = type;
824 27274 unit->data = data;
825 27274 unit->data_size = data_size;
826 27274 unit->data_ref = data_ref;
827
828 27274 return 0;
829 }
830
831 72 void ff_cbs_delete_unit(CodedBitstreamFragment *frag,
832 int position)
833 {
834
2/4
✓ Branch 0 taken 72 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 72 times.
 72 av_assert0(0 <= position && position < frag->nb_units
835 && "Unit to be deleted not in fragment.");
836
837 72 cbs_unit_uninit(&frag->units[position]);
838
839 72 --frag->nb_units;
840
841
1/2
✓ Branch 0 taken 72 times.
✗ Branch 1 not taken.
 72 if (frag->nb_units > 0)
842 72 memmove(frag->units + position,
843 72 frag->units + position + 1,
844 72 (frag->nb_units - position) * sizeof(*frag->units));
845 72 }
846
847 25354 static void cbs_default_free_unit_content(void *opaque, uint8_t *data)
848 {
849 25354 const CodedBitstreamUnitTypeDescriptor *desc = opaque;
850
2/2
✓ Branch 0 taken 22934 times.
✓ Branch 1 taken 2420 times.
 25354 if (desc->content_type == CBS_CONTENT_TYPE_INTERNAL_REFS) {
851 int i;
852
2/2
✓ Branch 0 taken 31680 times.
✓ Branch 1 taken 22934 times.
 54614 for (i = 0; i < desc->nb_ref_offsets; i++) {
853 31680 void **ptr = (void**)(data + desc->ref_offsets[i]);
854 31680 av_buffer_unref((AVBufferRef**)(ptr + 1));
855 }
856 }
857 25354 av_free(data);
858 25354 }
859
860 static const CodedBitstreamUnitTypeDescriptor
861 27254 *cbs_find_unit_type_desc(CodedBitstreamContext *ctx,
862 CodedBitstreamUnit *unit)
863 {
864 const CodedBitstreamUnitTypeDescriptor *desc;
865 int i, j;
866
867
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 27254 times.
 27254 if (!ctx->codec->unit_types)
868 ✗ return NULL;
869
870 100481 for (i = 0;; i++) {
871 100481 desc = &ctx->codec->unit_types[i];
872
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 100481 times.
 100481 if (desc->nb_unit_types == 0)
873 ✗ break;
874
2/2
✓ Branch 0 taken 17887 times.
✓ Branch 1 taken 82594 times.
 100481 if (desc->nb_unit_types == CBS_UNIT_TYPE_RANGE) {
875
1/2
✓ Branch 0 taken 17887 times.
✗ Branch 1 not taken.
 17887 if (unit->type >= desc->unit_type_range_start &&
876
2/2
✓ Branch 0 taken 14610 times.
✓ Branch 1 taken 3277 times.
 17887 unit->type <= desc->unit_type_range_end)
877 14610 return desc;
878 } else {
879
2/2
✓ Branch 0 taken 90089 times.
✓ Branch 1 taken 69950 times.
 160039 for (j = 0; j < desc->nb_unit_types; j++) {
880
2/2
✓ Branch 0 taken 12644 times.
✓ Branch 1 taken 77445 times.
 90089 if (desc->unit_types[j] == unit->type)
881 12644 return desc;
882 }
883 }
884 }
885 ✗ return NULL;
886 }
887
888 27254 int ff_cbs_alloc_unit_content2(CodedBitstreamContext *ctx,
889 CodedBitstreamUnit *unit)
890 {
891 const CodedBitstreamUnitTypeDescriptor *desc;
892
893
2/4
✓ Branch 0 taken 27254 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 27254 times.
 27254 av_assert0(!unit->content && !unit->content_ref);
894
895 27254 desc = cbs_find_unit_type_desc(ctx, unit);
896
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 27254 times.
 27254 if (!desc)
897 ✗ return AVERROR(ENOSYS);
898
899 27254 unit->content = av_mallocz(desc->content_size);
900
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 27254 times.
 27254 if (!unit->content)
901 ✗ return AVERROR(ENOMEM);
902
903 27254 unit->content_ref =
904 27254 av_buffer_create(unit->content, desc->content_size,
905
2/2
✓ Branch 0 taken 1900 times.
✓ Branch 1 taken 25354 times.
 27254 desc->content_free ? desc->content_free
906 : cbs_default_free_unit_content,
907 (void*)desc, 0);
908
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 27254 times.
 27254 if (!unit->content_ref) {
909 ✗ av_freep(&unit->content);
910 ✗ return AVERROR(ENOMEM);
911 }
912
913 27254 return 0;
914 }
915
916 ✗ static int cbs_clone_unit_content(AVBufferRef **clone_ref,
917 CodedBitstreamUnit *unit,
918 const CodedBitstreamUnitTypeDescriptor *desc)
919 {
920 uint8_t *src, *copy;
921 uint8_t **src_ptr, **copy_ptr;
922 AVBufferRef **src_buf, **copy_buf;
923 int err, i;
924
925 ✗ av_assert0(unit->content);
926 ✗ src = unit->content;
927
928 ✗ copy = av_memdup(src, desc->content_size);
929 ✗ if (!copy)
930 ✗ return AVERROR(ENOMEM);
931
932 ✗ for (i = 0; i < desc->nb_ref_offsets; i++) {
933 ✗ src_ptr = (uint8_t**)(src + desc->ref_offsets[i]);
934 ✗ src_buf = (AVBufferRef**)(src_ptr + 1);
935 ✗ copy_ptr = (uint8_t**)(copy + desc->ref_offsets[i]);
936 ✗ copy_buf = (AVBufferRef**)(copy_ptr + 1);
937
938 ✗ if (!*src_ptr) {
939 ✗ av_assert0(!*src_buf);
940 ✗ continue;
941 }
942 ✗ if (!*src_buf) {
943 // We can't handle a non-refcounted pointer here - we don't
944 // have enough information to handle whatever structure lies
945 // at the other end of it.
946 ✗ err = AVERROR(EINVAL);
947 ✗ goto fail;
948 }
949
950 // src_ptr is required to point somewhere inside src_buf. If it
951 // doesn't, there is a bug somewhere.
952 ✗ av_assert0(*src_ptr >= (*src_buf)->data &&
953 *src_ptr < (*src_buf)->data + (*src_buf)->size);
954
955 ✗ *copy_buf = av_buffer_ref(*src_buf);
956 ✗ if (!*copy_buf) {
957 ✗ err = AVERROR(ENOMEM);
958 ✗ goto fail;
959 }
960 ✗ *copy_ptr = (*copy_buf)->data + (*src_ptr - (*src_buf)->data);
961 }
962
963 ✗ *clone_ref = av_buffer_create(copy, desc->content_size,
964 ✗ desc->content_free ? desc->content_free :
965 cbs_default_free_unit_content,
966 (void*)desc, 0);
967 ✗ if (!*clone_ref) {
968 ✗ err = AVERROR(ENOMEM);
969 ✗ goto fail;
970 }
971
972 ✗ return 0;
973
974 ✗ fail:
975 ✗ for (--i; i >= 0; i--)
976 ✗ av_buffer_unref((AVBufferRef**)(copy + desc->ref_offsets[i]));
977 ✗ av_freep(&copy);
978 ✗ *clone_ref = NULL;
979 ✗ return err;
980 }
981
982 3341 int ff_cbs_make_unit_refcounted(CodedBitstreamContext *ctx,
983 CodedBitstreamUnit *unit)
984 {
985 const CodedBitstreamUnitTypeDescriptor *desc;
986 AVBufferRef *ref;
987 int err;
988
989
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 3341 times.
 3341 av_assert0(unit->content);
990
1/2
✓ Branch 0 taken 3341 times.
✗ Branch 1 not taken.
 3341 if (unit->content_ref) {
991 // Already refcounted, nothing to do.
992 3341 return 0;
993 }
994
995 ✗ desc = cbs_find_unit_type_desc(ctx, unit);
996 ✗ if (!desc)
997 ✗ return AVERROR(ENOSYS);
998
999 ✗ switch (desc->content_type) {
1000 ✗ case CBS_CONTENT_TYPE_POD:
1001 ✗ ref = av_buffer_alloc(desc->content_size);
1002 ✗ if (!ref)
1003 ✗ return AVERROR(ENOMEM);
1004 ✗ memcpy(ref->data, unit->content, desc->content_size);
1005 ✗ err = 0;
1006 ✗ break;
1007
1008 ✗ case CBS_CONTENT_TYPE_INTERNAL_REFS:
1009 ✗ err = cbs_clone_unit_content(&ref, unit, desc);
1010 ✗ break;
1011
1012 ✗ case CBS_CONTENT_TYPE_COMPLEX:
1013 ✗ if (!desc->content_clone)
1014 ✗ return AVERROR_PATCHWELCOME;
1015 ✗ err = desc->content_clone(&ref, unit);
1016 ✗ break;
1017
1018 ✗ default:
1019 ✗ av_assert0(0 && "Invalid content type.");
1020 }
1021
1022 ✗ if (err < 0)
1023 ✗ return err;
1024
1025 ✗ unit->content_ref = ref;
1026 ✗ unit->content = ref->data;
1027 ✗ return 0;
1028 }
1029
1030 ✗ int ff_cbs_make_unit_writable(CodedBitstreamContext *ctx,
1031 CodedBitstreamUnit *unit)
1032 {
1033 const CodedBitstreamUnitTypeDescriptor *desc;
1034 AVBufferRef *ref;
1035 int err;
1036
1037 // This can only be applied to refcounted units.
1038 ✗ err = ff_cbs_make_unit_refcounted(ctx, unit);
1039 ✗ if (err < 0)
1040 ✗ return err;
1041 ✗ av_assert0(unit->content && unit->content_ref);
1042
1043 ✗ if (av_buffer_is_writable(unit->content_ref))
1044 ✗ return 0;
1045
1046 ✗ desc = cbs_find_unit_type_desc(ctx, unit);
1047 ✗ if (!desc)
1048 ✗ return AVERROR(ENOSYS);
1049
1050 ✗ switch (desc->content_type) {
1051 ✗ case CBS_CONTENT_TYPE_POD:
1052 ✗ err = av_buffer_make_writable(&unit->content_ref);
1053 ✗ break;
1054
1055 ✗ case CBS_CONTENT_TYPE_INTERNAL_REFS:
1056 ✗ err = cbs_clone_unit_content(&ref, unit, desc);
1057 ✗ break;
1058
1059 ✗ case CBS_CONTENT_TYPE_COMPLEX:
1060 ✗ if (!desc->content_clone)
1061 ✗ return AVERROR_PATCHWELCOME;
1062 ✗ err = desc->content_clone(&ref, unit);
1063 ✗ break;
1064
1065 ✗ default:
1066 ✗ av_assert0(0 && "Invalid content type.");
1067 }
1068 ✗ if (err < 0)
1069 ✗ return err;
1070
1071 ✗ if (desc->content_type != CBS_CONTENT_TYPE_POD) {
1072 ✗ av_buffer_unref(&unit->content_ref);
1073 ✗ unit->content_ref = ref;
1074 }
1075 ✗ unit->content = unit->content_ref->data;
1076 ✗ return 0;
1077 }
1078