2 * Vividas VIV format Demuxer
3 * Copyright (c) 2012 Krzysztof Klinikowski
4 * Copyright (c) 2010 Andrzej Szombierski
5 * based on vivparse Copyright (c) 2007 Måns Rullgård
7 * This file is part of FFmpeg.
9 * FFmpeg is free software; you can redistribute it and/or
10 * modify it under the terms of the GNU Lesser General Public
11 * License as published by the Free Software Foundation; either
12 * version 2.1 of the License, or (at your option) any later version.
14 * FFmpeg is distributed in the hope that it will be useful,
15 * but WITHOUT ANY WARRANTY; without even the implied warranty of
16 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
17 * Lesser General Public License for more details.
19 * You should have received a copy of the GNU Lesser General Public
20 * License along with FFmpeg; if not, write to the Free Software
21 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
26 * @brief Vividas VIV (.viv) file demuxer
27 * @author Andrzej Szombierski [qq at kuku eu org] (2010-07)
28 * @sa http://wiki.multimedia.cx/index.php?title=Vividas_VIV
31 #include "libavutil/intreadwrite.h"
32 #include "avio_internal.h"
36 #define MAX_AUDIO_SUBPACKETS 100
38 typedef struct VIV_SB_block {
41 int64_t packet_offset;
44 typedef struct VIV_SB_entry {
48 typedef struct VIV_AudioSubpacket {
52 typedef struct VividasDemuxContext {
54 VIV_SB_block *sb_blocks;
59 int current_sb, current_sb_entry;
63 VIV_SB_entry *sb_entries;
65 int n_audio_subpackets;
66 int current_audio_subpacket;
70 VIV_AudioSubpacket audio_subpackets[MAX_AUDIO_SUBPACKETS];
71 } VividasDemuxContext;
73 static int viv_probe(AVProbeData *p)
75 if (memcmp(p->buf, "vividas03", 9))
78 return AVPROBE_SCORE_MAX;
81 static const unsigned short keybits[32] = {
82 163, 416, 893, 82, 223, 572, 1137, 430,
83 659, 1104, 13, 626, 695, 972, 1465, 686,
84 843, 1216, 317, 1122, 1383, 92, 513, 1158,
85 1243, 48, 573, 1306, 1495, 396, 1009, 350,
88 static uint32_t decode_key(uint8_t *buf)
92 for (int i = 0; i < 32; i++) {
93 unsigned p = keybits[i];
94 key |= !!(buf[p>>3] & (1<<(p&7))) << i;
100 static void put_v(uint8_t *p, unsigned v)
103 *p++ = ((v>>28)&0x7f)|0x80;
105 *p++ = ((v>>21)&0x7f)|0x80;
107 *p++ = ((v>>14)&0x7f)|0x80;
109 *p++ = ((v>>7)&0x7f)|0x80;
112 static unsigned recover_key(unsigned char sample[4], unsigned expected_size)
114 unsigned char plaintext[8] = { 'S', 'B' };
116 put_v(plaintext+2, expected_size);
118 return (sample[0]^plaintext[0])|
119 ((sample[1]^plaintext[1])<<8)|
120 ((sample[2]^plaintext[2])<<16)|
121 ((sample[3]^plaintext[3])<<24);
124 static void xor_block(void *p1, void *p2, unsigned size, int key, int *key_ptr)
142 static void decode_block(uint8_t *src, uint8_t *dest, unsigned size,
143 uint32_t key, uint32_t *key_ptr,
154 a2 = (4 - align) & 3;
157 uint32_t tmpkey = *key_ptr - key;
158 memcpy(tmp + align, src, a2);
159 xor_block(tmp, tmp, 4, key, &tmpkey);
160 memcpy(dest, tmp + align, a2);
167 xor_block(src + a2, dest + a2, s & ~3,
174 memcpy(tmp, src + size, s);
175 xor_block(&tmp, &tmp, 4, key, key_ptr);
176 memcpy(dest + size, tmp, s);
180 static uint32_t get_v(uint8_t *p)
185 if (v >= UINT_MAX / 128 - *p)
189 } while (*p++ & 0x80);
194 static uint8_t *read_vblock(AVIOContext *src, uint32_t *size,
195 uint32_t key, uint32_t *k2, int align)
201 if (avio_read(src, tmp, 4) != 4)
204 decode_block(tmp, tmp, 4, key, k2, align);
219 if (avio_read(src, buf + 4, n) == n) {
220 decode_block(buf + 4, buf + 4, n, key, k2, align + 4);
229 static uint8_t *read_sb_block(AVIOContext *src, unsigned *size,
230 uint32_t *key, unsigned expected_size)
233 uint8_t ibuf[8], sbuf[8];
237 if (avio_read(src, ibuf, 8) < 8)
241 decode_block(ibuf, sbuf, 8, *key, &k2, 0);
245 if (sbuf[0] != 'S' || sbuf[1] != 'B' || (expected_size>0 && n != expected_size)) {
246 uint32_t tmpkey = recover_key(ibuf, expected_size);
248 decode_block(ibuf, sbuf, 8, tmpkey, &k2, 0);
250 if (sbuf[0] != 'S' || sbuf[1] != 'B' || expected_size != n)
259 memcpy(buf, sbuf, 8);
264 if (avio_read(src, buf+8, n) < n) {
269 decode_block(buf + 8, buf + 8, n, *key, &k2, 0);
274 static void track_header(VividasDemuxContext *viv, AVFormatContext *s, uint8_t *buf, int size)
279 int num_video, num_audio;
282 pb = avio_alloc_context(buf, size, 0, NULL, NULL, NULL, NULL);
286 ffio_read_varlen(pb); // track_header_len
289 val_1 = ffio_read_varlen(pb);
291 for (i=0;i<val_1;i++) {
294 avio_r8(pb); // val_3
295 avio_r8(pb); // val_4
299 avio_r8(pb); // num_streams
302 off += ffio_read_varlen(pb); // val_5
305 num_video = avio_r8(pb);
307 avio_seek(pb, off, SEEK_SET);
309 av_log(s, AV_LOG_WARNING, "number of video tracks %d is not 1\n", num_video);
311 for (i = 0; i < num_video; i++) {
312 AVStream *st = avformat_new_stream(s, NULL);
316 st->codecpar->codec_type = AVMEDIA_TYPE_VIDEO;
317 st->codecpar->codec_id = AV_CODEC_ID_VP6;
320 off += ffio_read_varlen(pb);
322 avio_r8(pb); // val_7
323 st->time_base.num = avio_rl32(pb); // frame_time
324 st->time_base.den = avio_rl32(pb); // time_base
325 st->nb_frames = avio_rl32(pb); // n frames
326 st->codecpar->width = avio_rl16(pb); // width
327 st->codecpar->height = avio_rl16(pb); // height
328 avio_r8(pb); // val_8
329 avio_rl32(pb); // val_9
331 avio_seek(pb, off, SEEK_SET);
335 off += ffio_read_varlen(pb); // val_10
337 num_audio = avio_r8(pb);
338 avio_seek(pb, off, SEEK_SET);
341 av_log(s, AV_LOG_WARNING, "number of audio tracks %d is not 1\n", num_audio);
343 for(i=0;i<num_audio;i++) {
345 AVStream *st = avformat_new_stream(s, NULL);
347 st->id = num_video + i;
349 st->codecpar->codec_type = AVMEDIA_TYPE_AUDIO;
350 st->codecpar->codec_id = AV_CODEC_ID_VORBIS;
353 off += ffio_read_varlen(pb); // length
355 avio_r8(pb); //codec_id
356 avio_rl16(pb); //codec_subid
357 st->codecpar->channels = avio_rl16(pb); // channels
358 st->codecpar->sample_rate = avio_rl32(pb); // sample_rate
359 avio_seek(pb, 10, SEEK_CUR); // data_1
361 avio_seek(pb, q, SEEK_CUR); // data_2
362 avio_r8(pb); // zeropad
364 if (avio_tell(pb) < off) {
370 ffio_read_varlen(pb); // val_13
372 ffio_read_varlen(pb); // len_3
373 num_data = avio_r8(pb);
374 for (j = 0; j < num_data; j++) {
375 data_len[j] = ffio_read_varlen(pb);
376 xd_size += data_len[j];
379 st->codecpar->extradata_size = 64 + xd_size + xd_size / 255;
380 if (ff_alloc_extradata(st->codecpar, st->codecpar->extradata_size))
383 p = st->codecpar->extradata;
386 for (j = 0; j < num_data - 1; j++)
387 offset += av_xiphlacing(&p[offset], data_len[j]);
389 for (j = 0; j < num_data; j++) {
390 avio_read(pb, &p[offset], data_len[j]);
391 offset += data_len[j];
394 st->codecpar->extradata_size = offset;
401 static void track_index(VividasDemuxContext *viv, AVFormatContext *s, uint8_t *buf, unsigned size)
409 pb = avio_alloc_context(buf, size, 0, NULL, NULL, NULL, NULL);
413 ffio_read_varlen(pb); // track_index_len
415 viv->n_sb_blocks = ffio_read_varlen(pb);
416 viv->sb_blocks = av_calloc(viv->n_sb_blocks, sizeof(VIV_SB_block));
417 if (!viv->sb_blocks) {
418 viv->n_sb_blocks = 0;
426 for (i = 0; i < viv->n_sb_blocks; i++) {
427 viv->sb_blocks[i].byte_offset = off;
428 viv->sb_blocks[i].packet_offset = poff;
430 viv->sb_blocks[i].size = ffio_read_varlen(pb);
431 viv->sb_blocks[i].n_packets = ffio_read_varlen(pb);
433 off += viv->sb_blocks[i].size;
434 poff += viv->sb_blocks[i].n_packets;
437 if (maxnp < viv->sb_blocks[i].n_packets)
438 maxnp = viv->sb_blocks[i].n_packets;
441 viv->sb_entries = av_calloc(maxnp, sizeof(VIV_SB_entry));
445 static void load_sb_block(AVFormatContext *s, VividasDemuxContext *viv, unsigned expected_size)
457 av_free(viv->sb_buf);
459 viv->sb_buf = read_sb_block(s->pb, &size, &viv->sb_key, expected_size);
464 pb = avio_alloc_context(viv->sb_buf, size, 0, NULL, NULL, NULL, NULL);
472 ffio_read_varlen(pb); // size
474 ffio_read_varlen(pb); // first packet
476 viv->n_sb_entries = viv->sb_blocks[viv->current_sb].n_packets;
478 for (i = 0; i < viv->n_sb_entries; i++) {
479 viv->sb_entries[i].size = ffio_read_varlen(pb);
480 viv->sb_entries[i].flag = avio_r8(pb);
483 ffio_read_varlen(pb);
486 viv->current_sb_entry = 0;
489 static int viv_read_header(AVFormatContext *s)
491 VividasDemuxContext *viv = s->priv_data;
492 AVIOContext *pb = s->pb;
497 uint8_t keybuffer[187];
498 uint32_t b22_size = 0;
499 uint32_t b22_key = 0;
504 header_end = avio_tell(pb);
506 header_end += ffio_read_varlen(pb);
508 num_tracks = avio_r8(pb);
510 if (num_tracks != 1) {
511 av_log(s, AV_LOG_ERROR, "number of tracks %d is not 1\n", num_tracks);
512 return AVERROR(EINVAL);
516 avio_seek(pb, v, SEEK_CUR);
518 avio_read(pb, keybuffer, 187);
519 key = decode_key(keybuffer);
525 int64_t here = avio_tell(pb);
526 int block_len, block_type;
528 if (here >= header_end)
531 block_len = ffio_read_varlen(pb);
532 block_type = avio_r8(pb);
534 if (block_type == 22) {
535 avio_read(pb, keybuffer, 187);
536 b22_key = decode_key(keybuffer);
537 b22_size = avio_rl32(pb);
540 avio_seek(pb, here + block_len, SEEK_SET);
545 buf = read_vblock(pb, &v, b22_key, &k2, 0);
553 buf = read_vblock(pb, &v, key, &k2, 0);
556 track_header(viv, s, buf, v);
559 buf = read_vblock(pb, &v, key, &k2, v);
562 track_index(viv, s, buf, v);
565 viv->sb_offset = avio_tell(pb);
566 if (viv->n_sb_blocks > 0) {
568 load_sb_block(s, viv, viv->sb_blocks[0].size);
570 viv->current_sb = -1;
576 static int viv_read_packet(AVFormatContext *s,
579 VividasDemuxContext *viv = s->priv_data;
584 if (avio_feof(viv->sb_pb))
587 if (viv->current_audio_subpacket < viv->n_audio_subpackets) {
589 int size = viv->audio_subpackets[viv->current_audio_subpacket+1].start - viv->audio_subpackets[viv->current_audio_subpacket].start;
592 ret = av_get_packet(pb, pkt, size);
595 pkt->pos += viv->sb_offset + viv->sb_blocks[viv->current_sb].byte_offset;
597 pkt->stream_index = 1;
598 astream = s->streams[pkt->stream_index];
600 pkt->pts = av_rescale(viv->audio_sample, astream->time_base.den, astream->time_base.num) / astream->codecpar->sample_rate;
601 viv->audio_sample += viv->audio_subpackets[viv->current_audio_subpacket].pcm_bytes / 2 / astream->codecpar->channels;
602 pkt->flags |= AV_PKT_FLAG_KEY;
603 viv->current_audio_subpacket++;
607 if (viv->current_sb_entry >= viv->n_sb_entries) {
608 if (viv->current_sb+1 >= viv->n_sb_blocks)
612 load_sb_block(s, viv, 0);
613 viv->current_sb_entry = 0;
618 off += viv->sb_entries[viv->current_sb_entry].size;
620 if (viv->sb_entries[viv->current_sb_entry].flag == 0) {
621 int v_size = ffio_read_varlen(pb);
623 ffio_read_varlen(pb);
624 ret = av_get_packet(pb, pkt, v_size);
627 pkt->pos += viv->sb_offset + viv->sb_blocks[viv->current_sb].byte_offset;
629 pkt->pts = viv->sb_blocks[viv->current_sb].packet_offset + viv->current_sb_entry;
630 pkt->flags |= (pkt->data[0]&0x80)?0:AV_PKT_FLAG_KEY;
631 pkt->stream_index = 0;
633 for (int i = 0; i < MAX_AUDIO_SUBPACKETS - 1; i++) {
634 int start, pcm_bytes;
635 start = ffio_read_varlen(pb);
636 pcm_bytes = ffio_read_varlen(pb);
638 if (i > 0 && start == 0)
641 viv->n_audio_subpackets = i + 1;
642 viv->audio_subpackets[i].start = start;
643 viv->audio_subpackets[i].pcm_bytes = pcm_bytes;
645 viv->audio_subpackets[viv->n_audio_subpackets].start = (int)(off - avio_tell(pb));
646 viv->current_audio_subpacket = 0;
649 int v_size = ffio_read_varlen(pb);
651 ret = av_get_packet(pb, pkt, v_size);
654 pkt->pos += viv->sb_offset + viv->sb_blocks[viv->current_sb].byte_offset;
655 pkt->pts = viv->sb_blocks[viv->current_sb].packet_offset + viv->current_sb_entry;
656 pkt->flags |= (pkt->data[0] & 0x80) ? 0 : AV_PKT_FLAG_KEY;
657 pkt->stream_index = 0;
660 viv->current_sb_entry++;
665 static int viv_read_close(AVFormatContext *s)
667 VividasDemuxContext *viv = s->priv_data;
669 av_freep(&viv->sb_pb);
670 av_freep(&viv->sb_buf);
671 av_freep(&viv->sb_blocks);
672 av_freep(&viv->sb_entries);
677 static int viv_read_seek(AVFormatContext *s, int stream_index, int64_t timestamp, int flags)
679 VividasDemuxContext *viv = s->priv_data;
682 if (stream_index == 0)
685 frame = av_rescale_q(timestamp, s->streams[0]->time_base, s->streams[stream_index]->time_base);
687 for (int i = 0; i < viv->n_sb_blocks; i++) {
688 if (frame >= viv->sb_blocks[i].packet_offset && frame < viv->sb_blocks[i].packet_offset + viv->sb_blocks[i].n_packets) {
689 // flush audio packet queue
690 viv->current_audio_subpacket = 0;
691 viv->n_audio_subpackets = 0;
693 // seek to ith sb block
694 avio_seek(s->pb, viv->sb_offset + viv->sb_blocks[i].byte_offset, SEEK_SET);
696 load_sb_block(s, viv, 0);
697 // most problematic part: guess audio offset
698 viv->audio_sample = av_rescale_q(viv->sb_blocks[i].packet_offset, av_make_q(s->streams[1]->codecpar->sample_rate, 1), av_inv_q(s->streams[0]->time_base));
699 // hand-tuned 1.s a/v offset
700 viv->audio_sample += s->streams[1]->codecpar->sample_rate;
701 viv->current_sb_entry = 0;
708 AVInputFormat ff_vividas_demuxer = {
710 .long_name = NULL_IF_CONFIG_SMALL("Vividas VIV"),
711 .priv_data_size = sizeof(VividasDemuxContext),
712 .read_probe = viv_probe,
713 .read_header = viv_read_header,
714 .read_packet = viv_read_packet,
715 .read_close = viv_read_close,
716 .read_seek = viv_read_seek,