1 /*
2 * Bitmap Brothers JV demuxer
3 * Copyright (c) 2005, 2011 Peter Ross <pross@xvid.org>
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 /**
23 * @file
24 * Bitmap Brothers JV demuxer
25 * @author Peter Ross <pross@xvid.org>
26 */
27
28 #include "libavutil/channel_layout.h"
29 #include "libavutil/intreadwrite.h"
30
31 #include "avformat.h"
32 #include "internal.h"
33
34 #define JV_PREAMBLE_SIZE 5
35
36 typedef struct JVFrame {
37 int audio_size; /**< audio packet size (bytes) */
38 int video_size; /**< video packet size (bytes) */
39 int palette_size; /**< palette size (bytes) */
40 int video_type; /**< per-frame video compression type */
41 } JVFrame;
42
43 typedef struct JVDemuxContext {
44 JVFrame *frames;
45 enum {
46 JV_AUDIO = 0,
47 JV_VIDEO,
48 JV_PADDING
49 } state;
50 int64_t pts;
51 } JVDemuxContext;
52
53 #define MAGIC " Compression by John M Phillips Copyright (C) 1995 The Bitmap Brothers Ltd."
54
read_probe(const AVProbeData * pd)55 static int read_probe(const AVProbeData *pd)
56 {
57 if (pd->buf[0] == 'J' && pd->buf[1] == 'V' && strlen(MAGIC) + 4 <= pd->buf_size &&
58 !memcmp(pd->buf + 4, MAGIC, strlen(MAGIC)))
59 return AVPROBE_SCORE_MAX;
60 return 0;
61 }
62
read_close(AVFormatContext * s)63 static int read_close(AVFormatContext *s)
64 {
65 JVDemuxContext *jv = s->priv_data;
66
67 av_freep(&jv->frames);
68
69 return 0;
70 }
71
read_header(AVFormatContext * s)72 static int read_header(AVFormatContext *s)
73 {
74 JVDemuxContext *jv = s->priv_data;
75 AVIOContext *pb = s->pb;
76 AVStream *vst, *ast;
77 int64_t audio_pts = 0;
78 int64_t offset;
79 int i;
80
81 avio_skip(pb, 80);
82
83 ast = avformat_new_stream(s, NULL);
84 vst = avformat_new_stream(s, NULL);
85 if (!ast || !vst)
86 return AVERROR(ENOMEM);
87
88 vst->codecpar->codec_type = AVMEDIA_TYPE_VIDEO;
89 vst->codecpar->codec_id = AV_CODEC_ID_JV;
90 vst->codecpar->codec_tag = 0; /* no fourcc */
91 vst->codecpar->width = avio_rl16(pb);
92 vst->codecpar->height = avio_rl16(pb);
93 vst->duration =
94 vst->nb_frames =
95 ast->nb_index_entries = avio_rl16(pb);
96 avpriv_set_pts_info(vst, 64, avio_rl16(pb), 1000);
97
98 avio_skip(pb, 4);
99
100 ast->codecpar->codec_type = AVMEDIA_TYPE_AUDIO;
101 ast->codecpar->codec_id = AV_CODEC_ID_PCM_U8;
102 ast->codecpar->codec_tag = 0; /* no fourcc */
103 ast->codecpar->sample_rate = avio_rl16(pb);
104 ast->codecpar->channels = 1;
105 ast->codecpar->channel_layout = AV_CH_LAYOUT_MONO;
106 avpriv_set_pts_info(ast, 64, 1, ast->codecpar->sample_rate);
107
108 avio_skip(pb, 10);
109
110 ast->index_entries = av_malloc(ast->nb_index_entries *
111 sizeof(*ast->index_entries));
112 if (!ast->index_entries)
113 return AVERROR(ENOMEM);
114
115 jv->frames = av_malloc(ast->nb_index_entries * sizeof(JVFrame));
116 if (!jv->frames) {
117 av_freep(&ast->index_entries);
118 return AVERROR(ENOMEM);
119 }
120 offset = 0x68 + ast->nb_index_entries * 16;
121 for (i = 0; i < ast->nb_index_entries; i++) {
122 AVIndexEntry *e = ast->index_entries + i;
123 JVFrame *jvf = jv->frames + i;
124
125 /* total frame size including audio, video, palette data and padding */
126 e->size = avio_rl32(pb);
127 e->timestamp = i;
128 e->pos = offset;
129 offset += e->size;
130
131 jvf->audio_size = avio_rl32(pb);
132 jvf->video_size = avio_rl32(pb);
133 jvf->palette_size = avio_r8(pb) ? 768 : 0;
134
135 if ((jvf->video_size | jvf->audio_size) & ~0xFFFFFF ||
136 e->size - jvf->audio_size
137 - jvf->video_size
138 - jvf->palette_size < 0) {
139 if (s->error_recognition & AV_EF_EXPLODE) {
140 read_close(s);
141 av_freep(&jv->frames);
142 av_freep(&ast->index_entries);
143 return AVERROR_INVALIDDATA;
144 }
145 jvf->audio_size =
146 jvf->video_size =
147 jvf->palette_size = 0;
148 }
149
150 if (avio_r8(pb))
151 av_log(s, AV_LOG_WARNING, "unsupported audio codec\n");
152
153 jvf->video_type = avio_r8(pb);
154 avio_skip(pb, 1);
155
156 e->timestamp = jvf->audio_size ? audio_pts : AV_NOPTS_VALUE;
157 audio_pts += jvf->audio_size;
158
159 e->flags = jvf->video_type != 1 ? AVINDEX_KEYFRAME : 0;
160 }
161
162 jv->state = JV_AUDIO;
163 return 0;
164 }
165
read_packet(AVFormatContext * s,AVPacket * pkt)166 static int read_packet(AVFormatContext *s, AVPacket *pkt)
167 {
168 JVDemuxContext *jv = s->priv_data;
169 AVIOContext *pb = s->pb;
170 AVStream *ast = s->streams[0];
171 int ret;
172
173 while (!avio_feof(s->pb) && jv->pts < ast->nb_index_entries) {
174 const AVIndexEntry *e = ast->index_entries + jv->pts;
175 const JVFrame *jvf = jv->frames + jv->pts;
176
177 switch (jv->state) {
178 case JV_AUDIO:
179 jv->state++;
180 if (jvf->audio_size) {
181 if ((ret = av_get_packet(s->pb, pkt, jvf->audio_size)) < 0)
182 return ret;
183 pkt->stream_index = 0;
184 pkt->pts = e->timestamp;
185 pkt->flags |= AV_PKT_FLAG_KEY;
186 return 0;
187 }
188 case JV_VIDEO:
189 jv->state++;
190 if (jvf->video_size || jvf->palette_size) {
191 int size = jvf->video_size + jvf->palette_size;
192 if ((ret = av_new_packet(pkt, size + JV_PREAMBLE_SIZE)) < 0)
193 return ret;
194
195 AV_WL32(pkt->data, jvf->video_size);
196 pkt->data[4] = jvf->video_type;
197 ret = avio_read(pb, pkt->data + JV_PREAMBLE_SIZE, size);
198 if (ret < 0)
199 return ret;
200 if (ret < size) {
201 memset(pkt->data + JV_PREAMBLE_SIZE + ret, 0,
202 AV_INPUT_BUFFER_PADDING_SIZE);
203 pkt->flags |= AV_PKT_FLAG_CORRUPT;
204 }
205 pkt->size = ret + JV_PREAMBLE_SIZE;
206 pkt->stream_index = 1;
207 pkt->pts = jv->pts;
208 if (jvf->video_type != 1)
209 pkt->flags |= AV_PKT_FLAG_KEY;
210 return 0;
211 }
212 case JV_PADDING:
213 avio_skip(pb, FFMAX(e->size - jvf->audio_size - jvf->video_size
214 - jvf->palette_size, 0));
215 jv->state = JV_AUDIO;
216 jv->pts++;
217 }
218 }
219
220 if (s->pb->eof_reached)
221 return AVERROR_EOF;
222
223 return AVERROR(EIO);
224 }
225
read_seek(AVFormatContext * s,int stream_index,int64_t ts,int flags)226 static int read_seek(AVFormatContext *s, int stream_index,
227 int64_t ts, int flags)
228 {
229 JVDemuxContext *jv = s->priv_data;
230 AVStream *ast = s->streams[0];
231 int i;
232
233 if (flags & (AVSEEK_FLAG_BYTE | AVSEEK_FLAG_FRAME))
234 return AVERROR(ENOSYS);
235
236 switch (stream_index) {
237 case 0:
238 i = av_index_search_timestamp(ast, ts, flags);
239 break;
240 case 1:
241 i = ts;
242 break;
243 default:
244 return 0;
245 }
246
247 if (i < 0 || i >= ast->nb_index_entries)
248 return 0;
249 if (avio_seek(s->pb, ast->index_entries[i].pos, SEEK_SET) < 0)
250 return -1;
251
252 jv->state = JV_AUDIO;
253 jv->pts = i;
254 return 0;
255 }
256
257 AVInputFormat ff_jv_demuxer = {
258 .name = "jv",
259 .long_name = NULL_IF_CONFIG_SMALL("Bitmap Brothers JV"),
260 .priv_data_size = sizeof(JVDemuxContext),
261 .read_probe = read_probe,
262 .read_header = read_header,
263 .read_packet = read_packet,
264 .read_seek = read_seek,
265 .read_close = read_close,
266 };
267