GCC Code Coverage Report
Directory: ../../../ffmpeg/ Exec Total Coverage
File: src/libavcodec/atsc_a53.c Lines: 24 54 44.4 %
Date: 2020-09-25 14:59:26 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
6
    size_t new_size, old_size = buf ? buf->size : 0;
73
    int ret, cc_count;
74
75
6
    if (size < 3)
76
        return AVERROR(EINVAL);
77
78
6
    ret = init_get_bits8(&gb, data, size);
79
6
    if (ret < 0)
80
        return ret;
81
82
6
    if (get_bits(&gb, 8) != 0x3) // user_data_type_code
83
        return 0;
84
85
6
    skip_bits(&gb, 1); // reserved
86
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
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
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
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
6
    if (ret < 0)
108
        return ret;
109
110
6
    buf = *pbuf;
111
    /* Use of av_buffer_realloc assumes buffer is writeable */
112
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
}