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
26 int
ff_voc_get_packet(AVFormatContext * s,AVPacket * pkt,AVStream * st,int max_size)27 ff_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->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 par->channels = avio_r8(pb);
107 } else
108 avio_skip(pb, 6);
109 tmp_codec = avio_rl16(pb);
110 avio_skip(pb, 4);
111 voc->remaining_size -= 12;
112 max_size -= 12;
113 break;
114
115 default:
116 avio_skip(pb, voc->remaining_size);
117 max_size -= voc->remaining_size;
118 voc->remaining_size = 0;
119 break;
120 }
121 }
122
123 if (par->sample_rate <= 0) {
124 av_log(s, AV_LOG_ERROR, "Invalid sample rate %d\n", par->sample_rate);
125 return AVERROR_INVALIDDATA;
126 }
127
128 if (tmp_codec >= 0) {
129 tmp_codec = ff_codec_get_id(ff_voc_codec_tags, tmp_codec);
130 if (par->codec_id == AV_CODEC_ID_NONE)
131 par->codec_id = tmp_codec;
132 else if (par->codec_id != tmp_codec)
133 av_log(s, AV_LOG_WARNING, "Ignoring mid-stream change in audio codec\n");
134 if (par->codec_id == AV_CODEC_ID_NONE) {
135 if (s->audio_codec_id == AV_CODEC_ID_NONE) {
136 av_log(s, AV_LOG_ERROR, "unknown codec tag\n");
137 return AVERROR(EINVAL);
138 }
139 av_log(s, AV_LOG_WARNING, "unknown codec tag\n");
140 }
141 }
142
143 par->bit_rate = (int64_t)par->sample_rate * par->channels * par->bits_per_coded_sample;
144
145 if (max_size <= 0)
146 max_size = 2048;
147 size = FFMIN(voc->remaining_size, max_size);
148 voc->remaining_size -= size;
149
150 ret = av_get_packet(pb, pkt, size);
151 pkt->dts = pkt->pts = voc->pts;
152
153 duration = av_get_audio_frame_duration2(st->codecpar, size);
154 if (duration > 0 && voc->pts != AV_NOPTS_VALUE)
155 voc->pts += duration;
156 else
157 voc->pts = AV_NOPTS_VALUE;
158
159 return ret;
160 }
161