1 |
|
|
/* |
2 |
|
|
* Copyright (C) 2017 foo86 |
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 "get_bits.h" |
22 |
|
|
#include "put_bits.h" |
23 |
|
|
#include "dolby_e.h" |
24 |
|
|
|
25 |
|
|
static const uint8_t nb_programs_tab[MAX_PROG_CONF + 1] = { |
26 |
|
|
2, 3, 2, 3, 4, 5, 4, 5, 6, 7, 8, 1, 2, 3, 3, 4, 5, 6, 1, 2, 3, 4, 1, 1 |
27 |
|
|
}; |
28 |
|
|
|
29 |
|
|
static const uint8_t nb_channels_tab[MAX_PROG_CONF + 1] = { |
30 |
|
|
8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 8, 6, 6, 6, 6, 6, 6, 6, 4, 4, 4, 4, 8, 8 |
31 |
|
|
}; |
32 |
|
|
|
33 |
|
|
static const uint16_t sample_rate_tab[16] = { |
34 |
|
|
0, 42965, 43008, 44800, 53706, 53760 |
35 |
|
|
}; |
36 |
|
|
|
37 |
|
1284 |
static int skip_input(DBEContext *s, int nb_words) |
38 |
|
|
{ |
39 |
✗✓ |
1284 |
if (nb_words > s->input_size) { |
40 |
|
|
return AVERROR_INVALIDDATA; |
41 |
|
|
} |
42 |
|
|
|
43 |
|
1284 |
s->input += nb_words * s->word_bytes; |
44 |
|
1284 |
s->input_size -= nb_words; |
45 |
|
1284 |
return 0; |
46 |
|
|
} |
47 |
|
|
|
48 |
|
642 |
static int parse_key(DBEContext *s) |
49 |
|
|
{ |
50 |
✓✗ |
642 |
if (s->key_present) { |
51 |
|
642 |
const uint8_t *key = s->input; |
52 |
|
642 |
int ret = skip_input(s, 1); |
53 |
✗✓ |
642 |
if (ret < 0) |
54 |
|
|
return ret; |
55 |
|
642 |
return AV_RB24(key) >> 24 - s->word_bits; |
56 |
|
|
} |
57 |
|
|
return 0; |
58 |
|
|
} |
59 |
|
|
|
60 |
|
4380 |
int ff_dolby_e_convert_input(DBEContext *s, int nb_words, int key) |
61 |
|
|
{ |
62 |
|
4380 |
const uint8_t *src = s->input; |
63 |
|
4380 |
uint8_t *dst = s->buffer; |
64 |
|
|
PutBitContext pb; |
65 |
|
|
int i; |
66 |
|
|
|
67 |
✗✓ |
4380 |
av_assert0(nb_words <= 1024u); |
68 |
|
|
|
69 |
✗✓ |
4380 |
if (nb_words > s->input_size) { |
70 |
|
|
if (s->avctx) |
71 |
|
|
av_log(s->avctx, AV_LOG_ERROR, "Packet too short\n"); |
72 |
|
|
return AVERROR_INVALIDDATA; |
73 |
|
|
} |
74 |
|
|
|
75 |
✓✗✗✗
|
4380 |
switch (s->word_bits) { |
76 |
|
4380 |
case 16: |
77 |
✓✓ |
951144 |
for (i = 0; i < nb_words; i++, src += 2, dst += 2) |
78 |
|
946764 |
AV_WB16(dst, AV_RB16(src) ^ key); |
79 |
|
4380 |
break; |
80 |
|
|
case 20: |
81 |
|
|
init_put_bits(&pb, s->buffer, sizeof(s->buffer)); |
82 |
|
|
for (i = 0; i < nb_words; i++, src += 3) |
83 |
|
|
put_bits(&pb, 20, AV_RB24(src) >> 4 ^ key); |
84 |
|
|
flush_put_bits(&pb); |
85 |
|
|
break; |
86 |
|
|
case 24: |
87 |
|
|
for (i = 0; i < nb_words; i++, src += 3, dst += 3) |
88 |
|
|
AV_WB24(dst, AV_RB24(src) ^ key); |
89 |
|
|
break; |
90 |
|
|
default: |
91 |
|
|
av_assert0(0); |
92 |
|
|
} |
93 |
|
|
|
94 |
|
4380 |
return init_get_bits(&s->gb, s->buffer, nb_words * s->word_bits); |
95 |
|
|
} |
96 |
|
|
|
97 |
|
643 |
int ff_dolby_e_parse_header(DBEContext *s, const uint8_t *buf, int buf_size) |
98 |
|
|
{ |
99 |
|
643 |
DolbyEHeaderInfo *const header = &s->metadata; |
100 |
|
|
int hdr, ret, key, mtd_size; |
101 |
|
|
|
102 |
✓✓ |
643 |
if (buf_size < 3) |
103 |
|
1 |
return AVERROR_INVALIDDATA; |
104 |
|
|
|
105 |
|
642 |
hdr = AV_RB24(buf); |
106 |
✗✓ |
642 |
if ((hdr & 0xfffffe) == 0x7888e) { |
107 |
|
|
s->word_bits = 24; |
108 |
✗✓ |
642 |
} else if ((hdr & 0xffffe0) == 0x788e0) { |
109 |
|
|
s->word_bits = 20; |
110 |
✓✗ |
642 |
} else if ((hdr & 0xfffe00) == 0x78e00) { |
111 |
|
642 |
s->word_bits = 16; |
112 |
|
|
} else { |
113 |
|
|
if (s->avctx) |
114 |
|
|
av_log(s->avctx, AV_LOG_ERROR, "Invalid frame header\n"); |
115 |
|
|
return AVERROR_INVALIDDATA; |
116 |
|
|
} |
117 |
|
|
|
118 |
|
642 |
s->word_bytes = s->word_bits + 7 >> 3; |
119 |
|
642 |
s->input = buf + s->word_bytes; |
120 |
|
642 |
s->input_size = buf_size / s->word_bytes - 1; |
121 |
|
642 |
s->key_present = hdr >> 24 - s->word_bits & 1; |
122 |
|
|
|
123 |
✗✓ |
642 |
if ((key = parse_key(s)) < 0) |
124 |
|
|
return key; |
125 |
✗✓ |
642 |
if ((ret = ff_dolby_e_convert_input(s, 1, key)) < 0) |
126 |
|
|
return ret; |
127 |
|
|
|
128 |
|
642 |
skip_bits(&s->gb, 4); |
129 |
|
642 |
mtd_size = get_bits(&s->gb, 10); |
130 |
✗✓ |
642 |
if (!mtd_size) { |
131 |
|
|
if (s->avctx) |
132 |
|
|
av_log(s->avctx, AV_LOG_ERROR, "Invalid metadata size\n"); |
133 |
|
|
return AVERROR_INVALIDDATA; |
134 |
|
|
} |
135 |
|
|
|
136 |
✗✓ |
642 |
if ((ret = ff_dolby_e_convert_input(s, mtd_size, key)) < 0) |
137 |
|
|
return ret; |
138 |
|
|
|
139 |
|
642 |
skip_bits(&s->gb, 14); |
140 |
|
642 |
header->prog_conf = get_bits(&s->gb, 6); |
141 |
✗✓ |
642 |
if (header->prog_conf > MAX_PROG_CONF) { |
142 |
|
|
if (s->avctx) |
143 |
|
|
av_log(s->avctx, AV_LOG_ERROR, "Invalid program configuration\n"); |
144 |
|
|
return AVERROR_INVALIDDATA; |
145 |
|
|
} |
146 |
|
|
|
147 |
|
642 |
header->nb_channels = nb_channels_tab[header->prog_conf]; |
148 |
|
642 |
header->nb_programs = nb_programs_tab[header->prog_conf]; |
149 |
|
|
|
150 |
|
642 |
header->fr_code = get_bits(&s->gb, 4); |
151 |
|
642 |
header->fr_code_orig = get_bits(&s->gb, 4); |
152 |
✓✗ |
642 |
if (!(header->sample_rate = sample_rate_tab[header->fr_code]) || |
153 |
✗✓ |
642 |
!sample_rate_tab[header->fr_code_orig]) { |
154 |
|
|
if (s->avctx) |
155 |
|
|
av_log(s->avctx, AV_LOG_ERROR, "Invalid frame rate code\n"); |
156 |
|
|
return AVERROR_INVALIDDATA; |
157 |
|
|
} |
158 |
|
|
|
159 |
|
642 |
skip_bits_long(&s->gb, 88); |
160 |
✓✓ |
4494 |
for (int i = 0; i < header->nb_channels; i++) |
161 |
|
3852 |
header->ch_size[i] = get_bits(&s->gb, 10); |
162 |
|
642 |
header->mtd_ext_size = get_bits(&s->gb, 8); |
163 |
|
642 |
header->meter_size = get_bits(&s->gb, 8); |
164 |
|
|
|
165 |
|
642 |
skip_bits_long(&s->gb, 10 * header->nb_programs); |
166 |
✓✓ |
4494 |
for (int i = 0; i < header->nb_channels; i++) { |
167 |
|
3852 |
header->rev_id[i] = get_bits(&s->gb, 4); |
168 |
|
3852 |
skip_bits1(&s->gb); |
169 |
|
3852 |
header->begin_gain[i] = get_bits(&s->gb, 10); |
170 |
|
3852 |
header->end_gain[i] = get_bits(&s->gb, 10); |
171 |
|
|
} |
172 |
|
|
|
173 |
✗✓ |
642 |
if (get_bits_left(&s->gb) < 0) { |
174 |
|
|
if (s->avctx) |
175 |
|
|
av_log(s->avctx, AV_LOG_ERROR, "Read past end of metadata\n"); |
176 |
|
|
return AVERROR_INVALIDDATA; |
177 |
|
|
} |
178 |
|
|
|
179 |
|
642 |
return skip_input(s, mtd_size + 1); |
180 |
|
|
} |