1/*
2 * LVF demuxer
3 * Copyright (c) 2012 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/intreadwrite.h"
23#include "avformat.h"
24#include "riff.h"
25
26static int lvf_probe(AVProbeData *p)
27{
28    if (AV_RL32(p->buf) != MKTAG('L', 'V', 'F', 'F'))
29        return 0;
30
31    if (!AV_RL32(p->buf + 16) || AV_RL32(p->buf + 16) > 256)
32        return AVPROBE_SCORE_MAX / 8;
33
34    return AVPROBE_SCORE_EXTENSION;
35}
36
37static int lvf_read_header(AVFormatContext *s)
38{
39    AVStream *st;
40    int64_t next_offset;
41    unsigned size, nb_streams, id;
42
43    avio_skip(s->pb, 16);
44    nb_streams = avio_rl32(s->pb);
45    if (!nb_streams)
46        return AVERROR_INVALIDDATA;
47    if (nb_streams > 2) {
48        avpriv_request_sample(s, "%d streams", nb_streams);
49        return AVERROR_PATCHWELCOME;
50    }
51
52    avio_skip(s->pb, 1012);
53
54    while (!url_feof(s->pb)) {
55        id          = avio_rl32(s->pb);
56        size        = avio_rl32(s->pb);
57        next_offset = avio_tell(s->pb) + size;
58
59        switch (id) {
60        case MKTAG('0', '0', 'f', 'm'):
61            st = avformat_new_stream(s, 0);
62            if (!st)
63                return AVERROR(ENOMEM);
64
65            st->codec->codec_type = AVMEDIA_TYPE_VIDEO;
66            avio_skip(s->pb, 4);
67            st->codec->width      = avio_rl32(s->pb);
68            st->codec->height     = avio_rl32(s->pb);
69            avio_skip(s->pb, 4);
70            st->codec->codec_tag  = avio_rl32(s->pb);
71            st->codec->codec_id   = ff_codec_get_id(ff_codec_bmp_tags,
72                                                    st->codec->codec_tag);
73            avpriv_set_pts_info(st, 32, 1, 1000);
74            break;
75        case MKTAG('0', '1', 'f', 'm'):
76            st = avformat_new_stream(s, 0);
77            if (!st)
78                return AVERROR(ENOMEM);
79
80            st->codec->codec_type  = AVMEDIA_TYPE_AUDIO;
81            st->codec->codec_tag   = avio_rl16(s->pb);
82            st->codec->channels    = avio_rl16(s->pb);
83            st->codec->sample_rate = avio_rl16(s->pb);
84            avio_skip(s->pb, 8);
85            st->codec->bits_per_coded_sample = avio_r8(s->pb);
86            st->codec->codec_id    = ff_codec_get_id(ff_codec_wav_tags,
87                                                     st->codec->codec_tag);
88            avpriv_set_pts_info(st, 32, 1, 1000);
89            break;
90        case 0:
91            avio_seek(s->pb, 2048 + 8, SEEK_SET);
92            return 0;
93        default:
94            avpriv_request_sample(s, "id %d", id);
95            return AVERROR_PATCHWELCOME;
96        }
97
98        avio_seek(s->pb, next_offset, SEEK_SET);
99    }
100
101    return AVERROR_EOF;
102}
103
104static int lvf_read_packet(AVFormatContext *s, AVPacket *pkt)
105{
106    unsigned size, flags, timestamp, id;
107    int64_t pos;
108    int ret, is_video = 0;
109
110    pos = avio_tell(s->pb);
111    while (!url_feof(s->pb)) {
112        id    = avio_rl32(s->pb);
113        size  = avio_rl32(s->pb);
114
115        if (size == 0xFFFFFFFFu)
116            return AVERROR_EOF;
117
118        switch (id) {
119        case MKTAG('0', '0', 'd', 'c'):
120            is_video = 1;
121        case MKTAG('0', '1', 'w', 'b'):
122            if (size < 8)
123                return AVERROR_INVALIDDATA;
124            timestamp = avio_rl32(s->pb);
125            flags = avio_rl32(s->pb);
126            ret = av_get_packet(s->pb, pkt, size - 8);
127            if (flags & (1 << 12))
128                pkt->flags |= AV_PKT_FLAG_KEY;
129            pkt->stream_index = is_video ? 0 : 1;
130            pkt->pts          = timestamp;
131            pkt->pos          = pos;
132            return ret;
133        default:
134            ret = avio_skip(s->pb, size);
135        }
136
137        if (ret < 0)
138            return ret;
139    }
140
141    return AVERROR_EOF;
142}
143
144AVInputFormat ff_lvf_demuxer = {
145    .name        = "lvf",
146    .long_name   = NULL_IF_CONFIG_SMALL("LVF"),
147    .read_probe  = lvf_probe,
148    .read_header = lvf_read_header,
149    .read_packet = lvf_read_packet,
150    .extensions  = "lvf",
151    .flags       = AVFMT_GENERIC_INDEX,
152};
153