FFmpeg coverage


Directory: ../../../ffmpeg/
File: src/libavcodec/atsc_a53.c
Date: 2021-09-16 08:47:15
Exec Total Coverage
Lines: 24 54 44.4%
Branches: 11 26 42.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 <stddef.h>
20 #include <stdint.h>
21
22 #include "atsc_a53.h"
23 #include "get_bits.h"
24
25 int ff_alloc_a53_sei(const AVFrame *frame, size_t prefix_len,
26 void **data, size_t *sei_size)
27 {
28 AVFrameSideData *side_data = NULL;
29 uint8_t *sei_data;
30
31 if (frame)
32 side_data = av_frame_get_side_data(frame, AV_FRAME_DATA_A53_CC);
33
34 if (!side_data) {
35 *data = NULL;
36 return 0;
37 }
38
39 *sei_size = side_data->size + 11;
40 *data = av_mallocz(*sei_size + prefix_len);
41 if (!*data)
42 return AVERROR(ENOMEM);
43 sei_data = (uint8_t*)*data + prefix_len;
44
45 // country code
46 sei_data[0] = 181;
47 sei_data[1] = 0;
48 sei_data[2] = 49;
49
50 /**
51 * 'GA94' is standard in North America for ATSC, but hard coding
52 * this style may not be the right thing to do -- other formats
53 * do exist. This information is not available in the side_data
54 * so we are going with this right now.
55 */
56 AV_WL32(sei_data + 3, MKTAG('G', 'A', '9', '4'));
57 sei_data[7] = 3;
58 sei_data[8] = ((side_data->size/3) & 0x1f) | 0x40;
59 sei_data[9] = 0;
60
61 memcpy(sei_data + 10, side_data->data, side_data->size);
62
63 sei_data[side_data->size+10] = 255;
64
65 return 0;
66 }
67
68 6 int ff_parse_a53_cc(AVBufferRef **pbuf, const uint8_t *data, int size)
69 {
70 6 AVBufferRef *buf = *pbuf;
71 GetBitContext gb;
72
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
6 size_t new_size, old_size = buf ? buf->size : 0;
73 int ret, cc_count;
74
75
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
6 if (size < 3)
76 return AVERROR(EINVAL);
77
78 6 ret = init_get_bits8(&gb, data, size);
79
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
6 if (ret < 0)
80 return ret;
81
82
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 6 times.
6 if (get_bits(&gb, 8) != 0x3) // user_data_type_code
83 return 0;
84
85 6 skip_bits(&gb, 1); // reserved
86
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 6 times.
6 if (!get_bits(&gb, 1)) // process_cc_data_flag
87 return 0;
88
89 6 skip_bits(&gb, 1); // zero bit
90 6 cc_count = get_bits(&gb, 5);
91
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
6 if (!cc_count)
92 return 0;
93
94 6 skip_bits(&gb, 8); // reserved
95
96 /* 3 bytes per CC plus one byte marker_bits at the end */
97
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 6 times.
6 if (cc_count * 3 >= (get_bits_left(&gb) >> 3))
98 return AVERROR(EINVAL);
99
100 6 new_size = (old_size + cc_count * 3);
101
102
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
6 if (new_size > INT_MAX)
103 return AVERROR(EINVAL);
104
105 /* Allow merging of the cc data from two fields. */
106 6 ret = av_buffer_realloc(pbuf, new_size);
107
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
6 if (ret < 0)
108 return ret;
109
110 6 buf = *pbuf;
111 /* Use of av_buffer_realloc assumes buffer is writeable */
112
2/2
✓ Branch 0 taken 13 times.
✓ Branch 1 taken 6 times.
19 for (int i = 0; i < cc_count; i++) {
113 13 buf->data[old_size++] = get_bits(&gb, 8);
114 13 buf->data[old_size++] = get_bits(&gb, 8);
115 13 buf->data[old_size++] = get_bits(&gb, 8);
116 }
117
118 6 return cc_count;
119 }
120