FFmpeg coverage


Directory: ../../../ffmpeg/
File: src/libavcodec/cbs.c
Date: 2021-09-16 08:47:15
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 183 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 183 type = NULL;
84
1/2
✓ Branch 0 taken 459 times.
✗ Branch 1 not taken.
459 for (i = 0; i < FF_ARRAY_ELEMS(cbs_type_table); i++) {
85
2/2
✓ Branch 0 taken 183 times.
✓ Branch 1 taken 276 times.
459 if (cbs_type_table[i]->codec_id == codec_id) {
86 183 type = cbs_type_table[i];
87 183 break;
88 }
89 }
90
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 183 times.
183 if (!type)
91 return AVERROR(EINVAL);
92
93 183 ctx = av_mallocz(sizeof(*ctx));
94
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 183 times.
183 if (!ctx)
95 return AVERROR(ENOMEM);
96
97 183 ctx->log_ctx = log_ctx;
98 183 ctx->codec = type; /* Must be before any error */
99
100
1/2
✓ Branch 0 taken 183 times.
✗ Branch 1 not taken.
183 if (type->priv_data_size) {
101 183 ctx->priv_data = av_mallocz(ctx->codec->priv_data_size);
102
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 183 times.
183 if (!ctx->priv_data) {
103 av_freep(&ctx);
104 return AVERROR(ENOMEM);
105 }
106
2/2
✓ Branch 0 taken 77 times.
✓ Branch 1 taken 106 times.
183 if (type->priv_class) {
107 77 *(const AVClass **)ctx->priv_data = type->priv_class;
108 77 av_opt_set_defaults(ctx->priv_data);
109 }
110 }
111
112 183 ctx->decompose_unit_types = NULL;
113
114 183 ctx->trace_enable = 0;
115 183 ctx->trace_level = AV_LOG_TRACE;
116
117 183 *ctx_ptr = ctx;
118 183 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 183 void ff_cbs_close(CodedBitstreamContext **ctx_ptr)
128 {
129 183 CodedBitstreamContext *ctx = *ctx_ptr;
130
131
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 183 times.
183 if (!ctx)
132 return;
133
134
2/2
✓ Branch 0 taken 149 times.
✓ Branch 1 taken 34 times.
183 if (ctx->codec->close)
135 149 ctx->codec->close(ctx);
136
137 183 av_freep(&ctx->write_buffer);
138
139
3/4
✓ Branch 0 taken 77 times.
✓ Branch 1 taken 106 times.
✓ Branch 2 taken 77 times.
✗ Branch 3 not taken.
183 if (ctx->codec->priv_class && ctx->priv_data)
140 77 av_opt_free(ctx->priv_data);
141
142 183 av_freep(&ctx->priv_data);
143 183 av_freep(ctx_ptr);
144 }
145
146 26716 static void cbs_unit_uninit(CodedBitstreamUnit *unit)
147 {
148 26716 av_buffer_unref(&unit->content_ref);
149 26716 unit->content = NULL;
150
151 26716 av_buffer_unref(&unit->data_ref);
152 26716 unit->data = NULL;
153 26716 unit->data_size = 0;
154 26716 unit->data_bit_padding = 0;
155 26716 }
156
157 5967 void ff_cbs_fragment_reset(CodedBitstreamFragment *frag)
158 {
159 int i;
160
161
2/2
✓ Branch 0 taken 26644 times.
✓ Branch 1 taken 5967 times.
32611 for (i = 0; i < frag->nb_units; i++)
162 26644 cbs_unit_uninit(&frag->units[i]);
163 5967 frag->nb_units = 0;
164
165 5967 av_buffer_unref(&frag->data_ref);
166 5967 frag->data = NULL;
167 5967 frag->data_size = 0;
168 5967 frag->data_bit_padding = 0;
169 5967 }
170
171 114 void ff_cbs_fragment_free(CodedBitstreamFragment *frag)
172 {
173 114 ff_cbs_fragment_reset(frag);
174
175 114 av_freep(&frag->units);
176 114 frag->nb_units_allocated = 0;
177 114 }
178
179 5838 static int cbs_read_fragment_content(CodedBitstreamContext *ctx,
180 CodedBitstreamFragment *frag)
181 {
182 int err, i, j;
183
184
2/2
✓ Branch 0 taken 26703 times.
✓ Branch 1 taken 5837 times.
32540 for (i = 0; i < frag->nb_units; i++) {
185 26703 CodedBitstreamUnit *unit = &frag->units[i];
186
187
2/2
✓ Branch 0 taken 1055 times.
✓ Branch 1 taken 25648 times.
26703 if (ctx->decompose_unit_types) {
188
2/2
✓ Branch 0 taken 3150 times.
✓ Branch 1 taken 12 times.
3162 for (j = 0; j < ctx->nb_decompose_unit_types; j++) {
189
2/2
✓ Branch 0 taken 1043 times.
✓ Branch 1 taken 2107 times.
3150 if (ctx->decompose_unit_types[j] == unit->type)
190 1043 break;
191 }
192
2/2
✓ Branch 0 taken 12 times.
✓ Branch 1 taken 1043 times.
1055 if (j >= ctx->nb_decompose_unit_types)
193 12 continue;
194 }
195
196 26691 av_buffer_unref(&unit->content_ref);
197 26691 unit->content = NULL;
198
199
2/4
✓ Branch 0 taken 26691 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 26691 times.
26691 av_assert0(unit->data && unit->data_ref);
200
201 26691 err = ctx->codec->read_unit(ctx, unit);
202
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 26691 times.
26691 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 26691 times.
26691 } 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 26690 times.
26691 } 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 5837 return 0;
220 }
221
222 454 static int cbs_fill_fragment_data(CodedBitstreamFragment *frag,
223 const uint8_t *data, size_t size)
224 {
225
2/4
✓ Branch 0 taken 454 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 454 times.
454 av_assert0(!frag->data && !frag->data_ref);
226
227 454 frag->data_ref =
228 454 av_buffer_alloc(size + AV_INPUT_BUFFER_PADDING_SIZE);
229
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 454 times.
454 if (!frag->data_ref)
230 return AVERROR(ENOMEM);
231
232 454 frag->data = frag->data_ref->data;
233 454 frag->data_size = size;
234
235 454 memcpy(frag->data, data, size);
236 454 memset(frag->data + size, 0,
237 AV_INPUT_BUFFER_PADDING_SIZE);
238
239 454 return 0;
240 }
241
242 5838 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 5384 times.
✓ Branch 1 taken 454 times.
5838 if (buf) {
251 5384 frag->data_ref = av_buffer_ref(buf);
252
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 5384 times.
5384 if (!frag->data_ref)
253 return AVERROR(ENOMEM);
254
255 5384 frag->data = (uint8_t *)data;
256 5384 frag->data_size = size;
257
258 } else {
259 454 err = cbs_fill_fragment_data(frag, data, size);
260
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 454 times.
454 if (err < 0)
261 return err;
262 }
263
264 5838 err = ctx->codec->split_fragment(ctx, frag, header);
265
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 5838 times.
5838 if (err < 0)
266 return err;
267
268 5838 return cbs_read_fragment_content(ctx, frag);
269 }
270
271 55 int ff_cbs_read_extradata(CodedBitstreamContext *ctx,
272 CodedBitstreamFragment *frag,
273 const AVCodecParameters *par)
274 {
275 110 return cbs_read_data(ctx, frag, NULL,
276 55 par->extradata,
277 55 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 5384 int ff_cbs_read_packet(CodedBitstreamContext *ctx,
290 CodedBitstreamFragment *frag,
291 const AVPacket *pkt)
292 {
293 10768 return cbs_read_data(ctx, frag, pkt->buf,
294 5384 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 65503 void ff_cbs_trace_header(CodedBitstreamContext *ctx,
481 const char *name)
482 {
483
1/2
✓ Branch 0 taken 65503 times.
✗ Branch 1 not taken.
65503 if (!ctx->trace_enable)
484 65503 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 1443884 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 1443884 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 1443884 times.
1443884 av_assert0(width > 0 && width <= 32);
548
549
2/2
✓ Branch 1 taken 1 times.
✓ Branch 2 taken 1443883 times.
1443884 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 1443883 times.
1443883 if (ctx->trace_enable)
556 position = get_bits_count(gbc);
557
558 1443883 value = get_bits_long(gbc, width);
559
560
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1443883 times.
1443883 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 1443883 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 1443883 times.
1443883 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 1443883 *write_to = value;
579 1443883 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 26716 static int cbs_insert_unit(CodedBitstreamFragment *frag,
719 int position)
720 {
721 CodedBitstreamUnit *units;
722
723
2/2
✓ Branch 0 taken 26386 times.
✓ Branch 1 taken 330 times.
26716 if (frag->nb_units < frag->nb_units_allocated) {
724 26386 units = frag->units;
725
726
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 26386 times.
26386 if (position < frag->nb_units)
727 memmove(units + position + 1, units + position,
728 (frag->nb_units - position) * sizeof(*units));
729 } else {
730 330 units = av_malloc_array(frag->nb_units*2 + 1, sizeof(*units));
731
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 330 times.
330 if (!units)
732 return AVERROR(ENOMEM);
733
734 330 frag->nb_units_allocated = 2*frag->nb_units_allocated + 1;
735
736
2/2
✓ Branch 0 taken 216 times.
✓ Branch 1 taken 114 times.
330 if (position > 0)
737 216 memcpy(units, frag->units, position * sizeof(*units));
738
739
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 330 times.
330 if (position < frag->nb_units)
740 memcpy(units + position + 1, frag->units + position,
741 (frag->nb_units - position) * sizeof(*units));
742 }
743
744 26716 memset(units + position, 0, sizeof(*units));
745
746
2/2
✓ Branch 0 taken 330 times.
✓ Branch 1 taken 26386 times.
26716 if (units != frag->units) {
747 330 av_free(frag->units);
748 330 frag->units = units;
749 }
750
751 26716 ++frag->nb_units;
752
753 26716 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 26703 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 17860 times.
✓ Branch 1 taken 8843 times.
26703 if (position == -1)
803 17860 position = frag->nb_units;
804
2/4
✓ Branch 0 taken 26703 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 26703 times.
26703 av_assert0(position >= 0 && position <= frag->nb_units);
805
806
1/2
✓ Branch 0 taken 26703 times.
✗ Branch 1 not taken.
26703 if (data_buf)
807 26703 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 26703 times.
26703 if (!data_ref) {
811 if (!data_buf)
812 av_free(data);
813 return AVERROR(ENOMEM);
814 }
815
816 26703 err = cbs_insert_unit(frag, position);
817
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 26703 times.
26703 if (err < 0) {
818 av_buffer_unref(&data_ref);
819 return err;
820 }
821
822 26703 unit = &frag->units[position];
823 26703 unit->type = type;
824 26703 unit->data = data;
825 26703 unit->data_size = data_size;
826 26703 unit->data_ref = data_ref;
827
828 26703 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 24791 static void cbs_default_free_unit_content(void *opaque, uint8_t *data)
848 {
849 24791 const CodedBitstreamUnitTypeDescriptor *desc = opaque;
850
2/2
✓ Branch 0 taken 22673 times.
✓ Branch 1 taken 2118 times.
24791 if (desc->content_type == CBS_CONTENT_TYPE_INTERNAL_REFS) {
851 int i;
852
2/2
✓ Branch 0 taken 31419 times.
✓ Branch 1 taken 22673 times.
54092 for (i = 0; i < desc->nb_ref_offsets; i++) {
853 31419 void **ptr = (void**)(data + desc->ref_offsets[i]);
854 31419 av_buffer_unref((AVBufferRef**)(ptr + 1));
855 }
856 }
857 24791 av_free(data);
858 24791 }
859
860 static const CodedBitstreamUnitTypeDescriptor
861 26691 *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 26691 times.
26691 if (!ctx->codec->unit_types)
868 return NULL;
869
870 98419 for (i = 0;; i++) {
871 98419 desc = &ctx->codec->unit_types[i];
872
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 98419 times.
98419 if (desc->nb_unit_types == 0)
873 break;
874
2/2
✓ Branch 0 taken 17887 times.
✓ Branch 1 taken 80532 times.
98419 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 88027 times.
✓ Branch 1 taken 68451 times.
156478 for (j = 0; j < desc->nb_unit_types; j++) {
880
2/2
✓ Branch 0 taken 12081 times.
✓ Branch 1 taken 75946 times.
88027 if (desc->unit_types[j] == unit->type)
881 12081 return desc;
882 }
883 }
884 }
885 return NULL;
886 }
887
888 26691 int ff_cbs_alloc_unit_content2(CodedBitstreamContext *ctx,
889 CodedBitstreamUnit *unit)
890 {
891 const CodedBitstreamUnitTypeDescriptor *desc;
892
893
2/4
✓ Branch 0 taken 26691 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 26691 times.
26691 av_assert0(!unit->content && !unit->content_ref);
894
895 26691 desc = cbs_find_unit_type_desc(ctx, unit);
896
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 26691 times.
26691 if (!desc)
897 return AVERROR(ENOSYS);
898
899 26691 unit->content = av_mallocz(desc->content_size);
900
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 26691 times.
26691 if (!unit->content)
901 return AVERROR(ENOMEM);
902
903 26691 unit->content_ref =
904 26691 av_buffer_create(unit->content, desc->content_size,
905
2/2
✓ Branch 0 taken 1900 times.
✓ Branch 1 taken 24791 times.
26691 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 26691 times.
26691 if (!unit->content_ref) {
909 av_freep(&unit->content);
910 return AVERROR(ENOMEM);
911 }
912
913 26691 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