1/*
2 * Copyright (c) 2006  Aurelien Jacobs <aurel@gnuage.org>
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#include "avformat.h"
23#include "internal.h"
24#include "voc.h"
25
26int
27ff_voc_get_packet(AVFormatContext *s, AVPacket *pkt, AVStream *st, int max_size)
28{
29    VocDecContext *voc = s->priv_data;
30    AVCodecParameters *par = st->codecpar;
31    AVIOContext *pb = s->pb;
32    VocType type;
33    int size, tmp_codec=-1;
34    int sample_rate = 0;
35    int channels = 1;
36    int64_t duration;
37    int ret;
38
39    av_add_index_entry(st,
40                       avio_tell(pb),
41                       voc->pts,
42                       voc->remaining_size,
43                       0,
44                       AVINDEX_KEYFRAME);
45
46    while (!voc->remaining_size) {
47        if (max_size < 4)
48            max_size = 0;
49        type = avio_r8(pb);
50        if (type == VOC_TYPE_EOF)
51            return AVERROR_EOF;
52        voc->remaining_size = avio_rl24(pb);
53        if (!voc->remaining_size) {
54            int64_t filesize;
55            if (!(s->pb->seekable & AVIO_SEEKABLE_NORMAL))
56                return AVERROR(EIO);
57            filesize = avio_size(pb);
58            if (filesize - avio_tell(pb) > INT_MAX)
59                return AVERROR_INVALIDDATA;
60            voc->remaining_size = filesize - avio_tell(pb);
61        }
62        max_size -= 4;
63
64        switch (type) {
65        case VOC_TYPE_VOICE_DATA:
66            if (voc->remaining_size < 2) {
67                voc->remaining_size = 0;
68                return AVERROR_INVALIDDATA;
69            }
70            if (!par->sample_rate) {
71                par->sample_rate = 1000000 / (256 - avio_r8(pb));
72                if (sample_rate)
73                    par->sample_rate = sample_rate;
74                avpriv_set_pts_info(st, 64, 1, par->sample_rate);
75                par->ch_layout.nb_channels = channels;
76                par->bits_per_coded_sample = av_get_bits_per_sample(par->codec_id);
77            } else
78                avio_skip(pb, 1);
79            tmp_codec = avio_r8(pb);
80            voc->remaining_size -= 2;
81            max_size -= 2;
82            channels = 1;
83            break;
84
85        case VOC_TYPE_VOICE_DATA_CONT:
86            break;
87
88        case VOC_TYPE_EXTENDED:
89            sample_rate = avio_rl16(pb);
90            avio_r8(pb);
91            channels = avio_r8(pb) + 1;
92            sample_rate = 256000000 / (channels * (65536 - sample_rate));
93            voc->remaining_size = 0;
94            max_size -= 4;
95            break;
96
97        case VOC_TYPE_NEW_VOICE_DATA:
98            if (voc->remaining_size < 12) {
99                voc->remaining_size = 0;
100                return AVERROR_INVALIDDATA;
101            }
102            if (!par->sample_rate) {
103                par->sample_rate = avio_rl32(pb);
104                avpriv_set_pts_info(st, 64, 1, par->sample_rate);
105                par->bits_per_coded_sample = avio_r8(pb);
106                channels = avio_r8(pb);
107                par->ch_layout.nb_channels = channels;
108            } else
109                avio_skip(pb, 6);
110            tmp_codec = avio_rl16(pb);
111            avio_skip(pb, 4);
112            voc->remaining_size -= 12;
113            max_size -= 12;
114            break;
115
116        default:
117            avio_skip(pb, voc->remaining_size);
118            max_size -= voc->remaining_size;
119            voc->remaining_size = 0;
120            break;
121        }
122    }
123
124    if (par->sample_rate <= 0) {
125        av_log(s, AV_LOG_ERROR, "Invalid sample rate %d\n", par->sample_rate);
126        return AVERROR_INVALIDDATA;
127    }
128
129    if (tmp_codec >= 0) {
130        tmp_codec = ff_codec_get_id(ff_voc_codec_tags, tmp_codec);
131        if (par->codec_id == AV_CODEC_ID_NONE)
132            par->codec_id = tmp_codec;
133        else if (par->codec_id != tmp_codec)
134            av_log(s, AV_LOG_WARNING, "Ignoring mid-stream change in audio codec\n");
135        if (par->codec_id == AV_CODEC_ID_NONE) {
136            if (s->audio_codec_id == AV_CODEC_ID_NONE) {
137                av_log(s, AV_LOG_ERROR, "unknown codec tag\n");
138                return AVERROR(EINVAL);
139            }
140            av_log(s, AV_LOG_WARNING, "unknown codec tag\n");
141        }
142    }
143
144    par->bit_rate = (int64_t)par->sample_rate * par->ch_layout.nb_channels * par->bits_per_coded_sample;
145
146    if (max_size <= 0)
147        max_size = 2048;
148    size = FFMIN(voc->remaining_size, max_size);
149    voc->remaining_size -= size;
150
151    ret = av_get_packet(pb, pkt, size);
152    pkt->dts = pkt->pts = voc->pts;
153
154    duration = av_get_audio_frame_duration2(st->codecpar, size);
155    if (duration > 0 && voc->pts != AV_NOPTS_VALUE)
156        voc->pts += duration;
157    else
158        voc->pts = AV_NOPTS_VALUE;
159
160    return ret;
161}
162