1 /*
2 * AAX demuxer
3 * Copyright (c) 2020 Paul B Mahol
4 *
5 * This file is part of FFmpeg.
6 *
7 * FFmpeg is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
11 *
12 * FFmpeg is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
16 *
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with FFmpeg; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20 */
21
22 #include "libavutil/avassert.h"
23 #include "libavutil/intreadwrite.h"
24 #include "avformat.h"
25 #include "internal.h"
26
27 typedef struct AAXColumn {
28 uint8_t flag;
29 uint8_t type;
30 const char *name;
31 uint32_t offset;
32 int size;
33 } AAXColumn;
34
35 typedef struct AAXSegment {
36 int64_t start;
37 int64_t end;
38 } AAXSegment;
39
40 typedef struct AAXContext {
41 int64_t table_size;
42 uint16_t version;
43 int64_t rows_offset;
44 int64_t strings_offset;
45 int64_t data_offset;
46 int64_t name_offset;
47 uint16_t columns;
48 uint16_t row_width;
49 uint32_t nb_segments;
50 int64_t schema_offset;
51 int64_t strings_size;
52 char *string_table;
53
54 uint32_t current_segment;
55
56 AAXColumn *xcolumns;
57 AAXSegment *segments;
58 } AAXContext;
59
aax_probe(const AVProbeData * p)60 static int aax_probe(const AVProbeData *p)
61 {
62 if (AV_RB32(p->buf) != MKBETAG('@','U','T','F'))
63 return 0;
64 if (AV_RB32(p->buf + 4) == 0)
65 return 0;
66 if (AV_RB16(p->buf + 8) > 1)
67 return 0;
68 if (AV_RB32(p->buf + 28) < 1)
69 return 0;
70
71 return AVPROBE_SCORE_MAX;
72 }
73
74 enum ColumnFlag {
75 COLUMN_FLAG_NAME = 0x1,
76 COLUMN_FLAG_DEFAULT = 0x2,
77 COLUMN_FLAG_ROW = 0x4,
78 COLUMN_FLAG_UNDEFINED = 0x8 /* shouldn't exist */
79 };
80
81 enum ColumnType {
82 COLUMN_TYPE_UINT8 = 0x00,
83 COLUMN_TYPE_SINT8 = 0x01,
84 COLUMN_TYPE_UINT16 = 0x02,
85 COLUMN_TYPE_SINT16 = 0x03,
86 COLUMN_TYPE_UINT32 = 0x04,
87 COLUMN_TYPE_SINT32 = 0x05,
88 COLUMN_TYPE_UINT64 = 0x06,
89 COLUMN_TYPE_SINT64 = 0x07,
90 COLUMN_TYPE_FLOAT = 0x08,
91 COLUMN_TYPE_DOUBLE = 0x09,
92 COLUMN_TYPE_STRING = 0x0a,
93 COLUMN_TYPE_VLDATA = 0x0b,
94 COLUMN_TYPE_UINT128 = 0x0c, /* for GUIDs */
95 COLUMN_TYPE_UNDEFINED = -1
96 };
97
get_pts(AVFormatContext * s,int64_t pos,int size)98 static int64_t get_pts(AVFormatContext *s, int64_t pos, int size)
99 {
100 AAXContext *a = s->priv_data;
101 int64_t pts = 0;
102
103 for (int seg = 0; seg < a->current_segment; seg++)
104 pts += (a->segments[seg].end - a->segments[seg].start) / size;
105
106 pts += ((pos - a->segments[a->current_segment].start) / size);
107
108 return pts;
109 }
110
aax_read_header(AVFormatContext * s)111 static int aax_read_header(AVFormatContext *s)
112 {
113 AAXContext *a = s->priv_data;
114 AVIOContext *pb = s->pb;
115 AVCodecParameters *par;
116 AVStream *st;
117 int64_t column_offset = 0;
118 int ret, extradata_size;
119 char *codec;
120 int64_t ret64;
121
122 avio_skip(pb, 4);
123 a->table_size = avio_rb32(pb) + 8LL;
124 a->version = avio_rb16(pb);
125 a->rows_offset = avio_rb16(pb) + 8LL;
126 a->strings_offset = avio_rb32(pb) + 8LL;
127 a->data_offset = avio_rb32(pb) + 8LL;
128 a->name_offset = avio_rb32(pb);
129 a->columns = avio_rb16(pb);
130 a->row_width = avio_rb16(pb);
131 a->nb_segments = avio_rb32(pb);
132
133 if (a->nb_segments < 1)
134 return AVERROR_INVALIDDATA;
135
136 a->schema_offset = 0x20;
137 a->strings_size = a->data_offset - a->strings_offset;
138
139 if (a->rows_offset > a->table_size ||
140 a->strings_offset > a->table_size ||
141 a->data_offset > a->table_size)
142 return AVERROR_INVALIDDATA;
143 if (a->strings_size <= 0 || a->name_offset >= a->strings_size ||
144 a->strings_size > UINT16_MAX)
145 return AVERROR_INVALIDDATA;
146 if (a->columns <= 0)
147 return AVERROR_INVALIDDATA;
148
149 a->segments = av_calloc(a->nb_segments, sizeof(*a->segments));
150 if (!a->segments)
151 return AVERROR(ENOMEM);
152
153 a->xcolumns = av_calloc(a->columns, sizeof(*a->xcolumns));
154 if (!a->xcolumns) {
155 ret = AVERROR(ENOMEM);
156 goto fail;
157 }
158
159 a->string_table = av_calloc(a->strings_size + 1, sizeof(*a->string_table));
160 if (!a->string_table) {
161 ret = AVERROR(ENOMEM);
162 goto fail;
163 }
164
165 for (int c = 0; c < a->columns; c++) {
166 uint8_t info = avio_r8(pb);
167 uint32_t offset = avio_rb32(pb);
168 int value_size;
169
170 if (offset >= a->strings_size) {
171 ret = AVERROR_INVALIDDATA;
172 goto fail;
173 }
174
175 a->xcolumns[c].flag = info >> 4;
176 a->xcolumns[c].type = info & 0x0F;
177
178 switch (a->xcolumns[c].type) {
179 case COLUMN_TYPE_UINT8:
180 case COLUMN_TYPE_SINT8:
181 value_size = 0x01;
182 break;
183 case COLUMN_TYPE_UINT16:
184 case COLUMN_TYPE_SINT16:
185 value_size = 0x02;
186 break;
187 case COLUMN_TYPE_UINT32:
188 case COLUMN_TYPE_SINT32:
189 case COLUMN_TYPE_FLOAT:
190 case COLUMN_TYPE_STRING:
191 value_size = 0x04;
192 break;
193 case COLUMN_TYPE_VLDATA:
194 value_size = 0x08;
195 break;
196 case COLUMN_TYPE_UINT128:
197 value_size = 0x10;
198 break;
199 default:
200 ret = AVERROR_INVALIDDATA;
201 goto fail;
202 }
203
204 a->xcolumns[c].size = value_size;
205
206 if (a->xcolumns[c].flag & COLUMN_FLAG_NAME)
207 a->xcolumns[c].name = a->string_table + offset;
208
209 if (a->xcolumns[c].flag & COLUMN_FLAG_DEFAULT) {
210 /* data is found relative to columns start */
211 a->xcolumns[c].offset = avio_tell(pb) - a->schema_offset;
212 avio_skip(pb, value_size);
213 }
214
215 if (a->xcolumns[c].flag & COLUMN_FLAG_ROW) {
216 /* data is found relative to row start */
217 a->xcolumns[c].offset = column_offset;
218 column_offset += value_size;
219 }
220 }
221
222 ret = ret64 = avio_seek(pb, a->strings_offset, SEEK_SET);
223 if (ret64 < 0)
224 goto fail;
225
226 ret = avio_read(pb, a->string_table, a->strings_size);
227 if (ret != a->strings_size) {
228 if (ret < 0)
229 goto fail;
230 ret = AVERROR(EIO);
231 goto fail;
232 }
233
234 for (int c = 0; c < a->columns; c++) {
235 int64_t data_offset = 0;
236 int64_t col_offset;
237 int flag, type;
238
239 if (!a->xcolumns[c].name || strcmp(a->xcolumns[c].name, "data"))
240 continue;
241
242 type = a->xcolumns[c].type;
243 flag = a->xcolumns[c].flag;
244 col_offset = a->xcolumns[c].offset;
245
246 for (uint64_t r = 0; r < a->nb_segments; r++) {
247 if (flag & COLUMN_FLAG_DEFAULT) {
248 data_offset = a->schema_offset + col_offset;
249 } else if (flag & COLUMN_FLAG_ROW) {
250 data_offset = a->rows_offset + r * a->row_width + col_offset;
251 } else {
252 ret = AVERROR_INVALIDDATA;
253 goto fail;
254 }
255
256 ret = ret64 = avio_seek(pb, data_offset, SEEK_SET);
257 if (ret64 < 0)
258 goto fail;
259
260 if (type == COLUMN_TYPE_VLDATA) {
261 int64_t start, size;
262
263 start = avio_rb32(pb);
264 size = avio_rb32(pb);
265 a->segments[r].start = start + a->data_offset;
266 a->segments[r].end = a->segments[r].start + size;
267 } else {
268 ret = AVERROR_INVALIDDATA;
269 goto fail;
270 }
271 }
272 }
273
274 if (!a->segments[0].end) {
275 ret = AVERROR_INVALIDDATA;
276 goto fail;
277 }
278
279 st = avformat_new_stream(s, NULL);
280 if (!st) {
281 ret = AVERROR(ENOMEM);
282 goto fail;
283 }
284 st->start_time = 0;
285 par = s->streams[0]->codecpar;
286 par->codec_type = AVMEDIA_TYPE_AUDIO;
287
288 codec = a->string_table + a->name_offset;
289 if (!strcmp(codec, "AAX")) {
290 par->codec_id = AV_CODEC_ID_ADPCM_ADX;
291 ret64 = avio_seek(pb, a->segments[0].start, SEEK_SET);
292 if (ret64 < 0 || avio_rb16(pb) != 0x8000) {
293 ret = AVERROR_INVALIDDATA;
294 goto fail;
295 }
296 extradata_size = avio_rb16(pb) + 4;
297 if (extradata_size < 12) {
298 ret = AVERROR_INVALIDDATA;
299 goto fail;
300 }
301 avio_seek(pb, -4, SEEK_CUR);
302 ret = ff_get_extradata(s, par, pb, extradata_size);
303 if (ret < 0) {
304 goto fail;
305 }
306 par->channels = AV_RB8 (par->extradata + 7);
307 par->sample_rate = AV_RB32(par->extradata + 8);
308 if (!par->channels || !par->sample_rate) {
309 ret = AVERROR_INVALIDDATA;
310 goto fail;
311 }
312
313 avpriv_set_pts_info(st, 64, 32, par->sample_rate);
314 /*} else if (!strcmp(codec, "HCA") ){
315 par->codec_id = AV_CODEC_ID_HCA;*/
316 } else {
317 ret = AVERROR_INVALIDDATA;
318 goto fail;
319 }
320
321 return 0;
322 fail:
323 av_freep(&a->string_table);
324 av_freep(&a->xcolumns);
325 av_freep(&a->segments);
326
327 return ret;
328 }
329
aax_read_packet(AVFormatContext * s,AVPacket * pkt)330 static int aax_read_packet(AVFormatContext *s, AVPacket *pkt)
331 {
332 AAXContext *a = s->priv_data;
333 AVCodecParameters *par = s->streams[0]->codecpar;
334 AVIOContext *pb = s->pb;
335 const int size = 18 * par->channels;
336 int ret, extradata_size = 0;
337 uint8_t *extradata = NULL;
338 int skip = 0;
339
340 if (avio_feof(pb))
341 return AVERROR_EOF;
342
343 pkt->pos = avio_tell(pb);
344
345 for (uint32_t seg = 0; seg < a->nb_segments; seg++) {
346 int64_t start = a->segments[seg].start;
347 int64_t end = a->segments[seg].end;
348
349 if (pkt->pos >= start && pkt->pos <= end) {
350 a->current_segment = seg;
351 if (par->codec_id == AV_CODEC_ID_ADPCM_ADX)
352 skip = (end - start) - ((end - start) / size) * size;
353 break;
354 }
355 }
356
357 if (pkt->pos >= a->segments[a->current_segment].end - skip) {
358 if (a->current_segment + 1 == a->nb_segments)
359 return AVERROR_EOF;
360 a->current_segment++;
361 avio_seek(pb, a->segments[a->current_segment].start, SEEK_SET);
362
363 if (par->codec_id == AV_CODEC_ID_ADPCM_ADX) {
364 if (avio_rb16(pb) != 0x8000)
365 return AVERROR_INVALIDDATA;
366 extradata_size = avio_rb16(pb) + 4;
367 avio_seek(pb, -4, SEEK_CUR);
368 if (extradata_size < 12)
369 return AVERROR_INVALIDDATA;
370 extradata = av_malloc(extradata_size + AV_INPUT_BUFFER_PADDING_SIZE);
371 if (!extradata)
372 return AVERROR(ENOMEM);
373 if (avio_read(pb, extradata, extradata_size) != extradata_size) {
374 av_free(extradata);
375 return AVERROR(EIO);
376 }
377 memset(extradata + extradata_size, 0, AV_INPUT_BUFFER_PADDING_SIZE);
378 }
379 }
380
381 ret = av_get_packet(pb, pkt, size);
382 if (ret != size) {
383 av_free(extradata);
384 return ret < 0 ? ret : AVERROR(EIO);
385 }
386 pkt->duration = 1;
387 pkt->stream_index = 0;
388 pkt->pts = get_pts(s, pkt->pos, size);
389
390 if (extradata) {
391 ret = av_packet_add_side_data(pkt, AV_PKT_DATA_NEW_EXTRADATA, extradata, extradata_size);
392 if (ret < 0) {
393 av_free(extradata);
394 return ret;
395 }
396 }
397
398 return ret;
399 }
400
aax_read_close(AVFormatContext * s)401 static int aax_read_close(AVFormatContext *s)
402 {
403 AAXContext *a = s->priv_data;
404
405 av_freep(&a->segments);
406 av_freep(&a->xcolumns);
407 av_freep(&a->string_table);
408
409 return 0;
410 }
411
412 AVInputFormat ff_aax_demuxer = {
413 .name = "aax",
414 .long_name = NULL_IF_CONFIG_SMALL("CRI AAX"),
415 .priv_data_size = sizeof(AAXContext),
416 .read_probe = aax_probe,
417 .read_header = aax_read_header,
418 .read_packet = aax_read_packet,
419 .read_close = aax_read_close,
420 .extensions = "aax",
421 .flags = AVFMT_GENERIC_INDEX,
422 };
423