smacker: fix integer overflow with pts_inc
[ffmpeg.git] / libavformat / pcmdec.c
1 /*
2  * RAW PCM demuxers
3  * Copyright (c) 2002 Fabrice Bellard
4  *
5  * This file is part of Libav.
6  *
7  * Libav 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  * Libav 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 Libav; if not, write to the Free Software
19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20  */
21
22 #include "avformat.h"
23 #include "internal.h"
24 #include "pcm.h"
25 #include "libavutil/log.h"
26 #include "libavutil/opt.h"
27
28 #define RAW_SAMPLES     1024
29
30 typedef struct PCMAudioDemuxerContext {
31     AVClass *class;
32     int sample_rate;
33     int channels;
34 } PCMAudioDemuxerContext;
35
36 static int pcm_read_header(AVFormatContext *s)
37 {
38     PCMAudioDemuxerContext *s1 = s->priv_data;
39     AVStream *st;
40
41     st = avformat_new_stream(s, NULL);
42     if (!st)
43         return AVERROR(ENOMEM);
44
45
46     st->codecpar->codec_type  = AVMEDIA_TYPE_AUDIO;
47     st->codecpar->codec_id    = s->iformat->raw_codec_id;
48     st->codecpar->sample_rate = s1->sample_rate;
49     st->codecpar->channels    = s1->channels;
50
51     st->codecpar->bits_per_coded_sample =
52         av_get_bits_per_sample(st->codecpar->codec_id);
53
54     assert(st->codecpar->bits_per_coded_sample > 0);
55
56     st->codecpar->block_align =
57         st->codecpar->bits_per_coded_sample * st->codecpar->channels / 8;
58
59     avpriv_set_pts_info(st, 64, 1, st->codecpar->sample_rate);
60     return 0;
61 }
62
63 static int pcm_read_packet(AVFormatContext *s, AVPacket *pkt)
64 {
65     int ret, size, bps;
66     //    AVStream *st = s->streams[0];
67
68     size= RAW_SAMPLES*s->streams[0]->codecpar->block_align;
69
70     ret= av_get_packet(s->pb, pkt, size);
71
72     pkt->stream_index = 0;
73     if (ret < 0)
74         return ret;
75
76     bps= av_get_bits_per_sample(s->streams[0]->codecpar->codec_id);
77     if (!bps) {
78         av_log(s, AV_LOG_ERROR, "Unknown number of bytes per sample.\n");
79         return AVERROR(EINVAL);
80     }
81
82     pkt->dts=
83     pkt->pts= pkt->pos*8 / (bps * s->streams[0]->codecpar->channels);
84
85     return ret;
86 }
87
88 static const AVOption pcm_options[] = {
89     { "sample_rate", "", offsetof(PCMAudioDemuxerContext, sample_rate), AV_OPT_TYPE_INT, {.i64 = 0}, 0, INT_MAX, AV_OPT_FLAG_DECODING_PARAM },
90     { "channels",    "", offsetof(PCMAudioDemuxerContext, channels),    AV_OPT_TYPE_INT, {.i64 = 1}, 0, INT_MAX, AV_OPT_FLAG_DECODING_PARAM },
91     { NULL },
92 };
93
94 #define PCMDEF(name_, long_name_, ext, codec)               \
95 static const AVClass name_ ## _demuxer_class = {            \
96     .class_name = #name_ " demuxer",                        \
97     .item_name  = av_default_item_name,                     \
98     .option     = pcm_options,                              \
99     .version    = LIBAVUTIL_VERSION_INT,                    \
100 };                                                          \
101 AVInputFormat ff_pcm_ ## name_ ## _demuxer = {              \
102     .name           = #name_,                               \
103     .long_name      = NULL_IF_CONFIG_SMALL(long_name_),     \
104     .priv_data_size = sizeof(PCMAudioDemuxerContext),       \
105     .read_header    = pcm_read_header,                      \
106     .read_packet    = pcm_read_packet,                      \
107     .read_seek      = ff_pcm_read_seek,                     \
108     .flags          = AVFMT_GENERIC_INDEX,                  \
109     .extensions     = ext,                                  \
110     .raw_codec_id   = codec,                                \
111     .priv_class     = &name_ ## _demuxer_class,             \
112 };
113
114 PCMDEF(f64be, "PCM 64-bit floating-point big-endian",
115        NULL, AV_CODEC_ID_PCM_F64BE)
116
117 PCMDEF(f64le, "PCM 64-bit floating-point little-endian",
118        NULL, AV_CODEC_ID_PCM_F64LE)
119
120 PCMDEF(f32be, "PCM 32-bit floating-point big-endian",
121        NULL, AV_CODEC_ID_PCM_F32BE)
122
123 PCMDEF(f32le, "PCM 32-bit floating-point little-endian",
124        NULL, AV_CODEC_ID_PCM_F32LE)
125
126 PCMDEF(s32be, "PCM signed 32-bit big-endian",
127        NULL, AV_CODEC_ID_PCM_S32BE)
128
129 PCMDEF(s32le, "PCM signed 32-bit little-endian",
130        NULL, AV_CODEC_ID_PCM_S32LE)
131
132 PCMDEF(s24be, "PCM signed 24-bit big-endian",
133        NULL, AV_CODEC_ID_PCM_S24BE)
134
135 PCMDEF(s24le, "PCM signed 24-bit little-endian",
136        NULL, AV_CODEC_ID_PCM_S24LE)
137
138 PCMDEF(s16be, "PCM signed 16-bit big-endian",
139        AV_NE("sw", NULL), AV_CODEC_ID_PCM_S16BE)
140
141 PCMDEF(s16le, "PCM signed 16-bit little-endian",
142        AV_NE(NULL, "sw"), AV_CODEC_ID_PCM_S16LE)
143
144 PCMDEF(s8, "PCM signed 8-bit",
145        "sb", AV_CODEC_ID_PCM_S8)
146
147 PCMDEF(u32be, "PCM unsigned 32-bit big-endian",
148        NULL, AV_CODEC_ID_PCM_U32BE)
149
150 PCMDEF(u32le, "PCM unsigned 32-bit little-endian",
151        NULL, AV_CODEC_ID_PCM_U32LE)
152
153 PCMDEF(u24be, "PCM unsigned 24-bit big-endian",
154        NULL, AV_CODEC_ID_PCM_U24BE)
155
156 PCMDEF(u24le, "PCM unsigned 24-bit little-endian",
157        NULL, AV_CODEC_ID_PCM_U24LE)
158
159 PCMDEF(u16be, "PCM unsigned 16-bit big-endian",
160        AV_NE("uw", NULL), AV_CODEC_ID_PCM_U16BE)
161
162 PCMDEF(u16le, "PCM unsigned 16-bit little-endian",
163        AV_NE(NULL, "uw"), AV_CODEC_ID_PCM_U16LE)
164
165 PCMDEF(u8, "PCM unsigned 8-bit",
166        "ub", AV_CODEC_ID_PCM_U8)
167
168 PCMDEF(alaw, "PCM A-law",
169        "al", AV_CODEC_ID_PCM_ALAW)
170
171 PCMDEF(mulaw, "PCM mu-law",
172        "ul", AV_CODEC_ID_PCM_MULAW)