GCC Code Coverage Report
Directory: ../../../ffmpeg/ Exec Total Coverage
File: src/libavcodec/hcom.c Lines: 0 60 0.0 %
Date: 2019-11-22 03:34:36 Branches: 0 36 0.0 %

Line Branch Exec Source
1
/*
2
 * HCOM audio decoder
3
 *
4
 * This file is part of FFmpeg.
5
 *
6
 * FFmpeg is free software; you can redistribute it and/or
7
 * modify it under the terms of the GNU Lesser General Public
8
 * License as published by the Free Software Foundation; either
9
 * version 2.1 of the License, or (at your option) any later version.
10
 *
11
 * FFmpeg is distributed in the hope that it will be useful,
12
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14
 * Lesser General Public License for more details.
15
 *
16
 * You should have received a copy of the GNU Lesser General Public
17
 * License along with FFmpeg; if not, write to the Free Software
18
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19
 */
20
21
#include "libavutil/intreadwrite.h"
22
23
#include "avcodec.h"
24
#include "get_bits.h"
25
#include "internal.h"
26
27
typedef struct HEntry {
28
    int16_t l, r;
29
} HEntry;
30
31
typedef struct HCOMContext {
32
    AVCodecContext *avctx;
33
34
    uint8_t first_sample;
35
    uint8_t sample;
36
    int dict_entries;
37
    int dict_entry;
38
    int delta_compression;
39
40
    HEntry *dict;
41
} HCOMContext;
42
43
static av_cold int hcom_init(AVCodecContext *avctx)
44
{
45
    HCOMContext *s = avctx->priv_data;
46
47
    if (avctx->channels != 1) {
48
        av_log(avctx, AV_LOG_ERROR, "invalid number of channels\n");
49
        return AVERROR_INVALIDDATA;
50
    }
51
52
    if (avctx->extradata_size <= 7)
53
        return AVERROR_INVALIDDATA;
54
    s->dict_entries = AV_RB16(avctx->extradata);
55
    if (avctx->extradata_size < s->dict_entries * 4 + 7 ||
56
        s->dict_entries == 0)
57
        return AVERROR_INVALIDDATA;
58
    s->delta_compression = AV_RB32(avctx->extradata + 2);
59
    s->sample = s->first_sample = avctx->extradata[avctx->extradata_size - 1];
60
61
    s->dict = av_calloc(s->dict_entries, sizeof(*s->dict));
62
    if (!s->dict)
63
        return AVERROR(ENOMEM);
64
    for (int i = 0; i < s->dict_entries; i++) {
65
        s->dict[i].l = AV_RB16(avctx->extradata + 6 + 4 * i);
66
        s->dict[i].r = AV_RB16(avctx->extradata + 6 + 4 * i + 2);
67
        if (s->dict[i].l >= 0 &&
68
            (s->dict[i].l >= s->dict_entries ||
69
             s->dict[i].r >= s->dict_entries ||
70
             s->dict[i].r < 0 )) {
71
            av_freep(&s->dict);
72
            return AVERROR_INVALIDDATA;
73
        }
74
    }
75
    if (s->dict[0].l < 0) {
76
        av_freep(&s->dict);
77
        return AVERROR_INVALIDDATA;
78
    }
79
80
    avctx->sample_fmt = AV_SAMPLE_FMT_U8;
81
    s->dict_entry = 0;
82
83
    return 0;
84
}
85
86
static int hcom_decode(AVCodecContext *avctx, void *data,
87
                       int *got_frame, AVPacket *pkt)
88
{
89
    HCOMContext *s = avctx->priv_data;
90
    AVFrame *frame = data;
91
    GetBitContext gb;
92
    int ret, n = 0;
93
94
    if (pkt->size > INT16_MAX)
95
        return AVERROR_INVALIDDATA;
96
97
    frame->nb_samples = pkt->size * 8;
98
    if ((ret = ff_get_buffer(avctx, frame, 0)) < 0)
99
        return ret;
100
101
    if ((ret = init_get_bits8(&gb, pkt->data, pkt->size)) < 0)
102
        return ret;
103
104
    while (get_bits_left(&gb) > 0) {
105
        if (get_bits1(&gb))
106
            s->dict_entry = s->dict[s->dict_entry].r;
107
        else
108
            s->dict_entry = s->dict[s->dict_entry].l;
109
110
        if (s->dict[s->dict_entry].l < 0) {
111
            int16_t datum;
112
113
            datum = s->dict[s->dict_entry].r;
114
115
            if (!s->delta_compression)
116
                s->sample = 0;
117
            s->sample = (s->sample + datum) & 0xFF;
118
119
            frame->data[0][n++] = s->sample;
120
121
            s->dict_entry = 0;
122
        }
123
    }
124
125
    frame->nb_samples = n;
126
127
    *got_frame = 1;
128
129
    return pkt->size;
130
}
131
132
static av_cold int hcom_close(AVCodecContext *avctx)
133
{
134
    HCOMContext *s = avctx->priv_data;
135
136
    av_freep(&s->dict);
137
138
    return 0;
139
}
140
141
AVCodec ff_hcom_decoder = {
142
    .name           = "hcom",
143
    .long_name      = NULL_IF_CONFIG_SMALL("HCOM Audio"),
144
    .type           = AVMEDIA_TYPE_AUDIO,
145
    .id             = AV_CODEC_ID_HCOM,
146
    .priv_data_size = sizeof(HCOMContext),
147
    .init           = hcom_init,
148
    .close          = hcom_close,
149
    .decode         = hcom_decode,
150
    .capabilities   = AV_CODEC_CAP_DR1,
151
};