]> git.sesse.net Git - ffmpeg/blob - libavformat/bintext.c
avformat/hlsenc: fix default AES key file url with variant streams
[ffmpeg] / libavformat / bintext.c
1 /*
2  * Binary text demuxer
3  * eXtended BINary text (XBIN) demuxer
4  * Artworx Data Format demuxer
5  * iCEDraw File demuxer
6  * Copyright (c) 2010 Peter Ross <pross@xvid.org>
7  *
8  * This file is part of FFmpeg.
9  *
10  * FFmpeg is free software; you can redistribute it and/or
11  * modify it under the terms of the GNU Lesser General Public
12  * License as published by the Free Software Foundation; either
13  * version 2.1 of the License, or (at your option) any later version.
14  *
15  * FFmpeg is distributed in the hope that it will be useful,
16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
18  * Lesser General Public License for more details.
19  *
20  * You should have received a copy of the GNU Lesser General Public
21  * License along with FFmpeg; if not, write to the Free Software
22  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
23  */
24
25 /**
26  * @file
27  * Binary text demuxer
28  * eXtended BINary text (XBIN) demuxer
29  * Artworx Data Format demuxer
30  * iCEDraw File demuxer
31  */
32
33 #include "libavutil/intreadwrite.h"
34 #include "libavutil/opt.h"
35 #include "libavutil/parseutils.h"
36 #include "avformat.h"
37 #include "internal.h"
38 #include "sauce.h"
39 #include "libavcodec/bintext.h"
40
41 typedef struct {
42     const AVClass *class;
43     int chars_per_frame; /**< characters to send decoder per frame;
44                               set by private options as characters per second, and then
45                               converted to characters per frame at runtime */
46     int width, height;    /**< video size (WxH pixels) (private option) */
47     AVRational framerate; /**< frames per second (private option) */
48     uint64_t fsize;  /**< file size less metadata buffer */
49 } BinDemuxContext;
50
51 static AVStream * init_stream(AVFormatContext *s)
52 {
53     BinDemuxContext *bin = s->priv_data;
54     AVStream *st = avformat_new_stream(s, NULL);
55     if (!st)
56         return NULL;
57     st->codecpar->codec_tag   = 0;
58     st->codecpar->codec_type  = AVMEDIA_TYPE_VIDEO;
59
60     if (!bin->width) {
61         st->codecpar->width  = (80<<3);
62         st->codecpar->height = (25<<4);
63     }
64
65     avpriv_set_pts_info(st, 60, bin->framerate.den, bin->framerate.num);
66
67     /* simulate tty display speed */
68     bin->chars_per_frame = av_clip(av_q2d(st->time_base) * bin->chars_per_frame, 1, INT_MAX);
69
70     return st;
71 }
72
73 #if CONFIG_BINTEXT_DEMUXER | CONFIG_ADF_DEMUXER | CONFIG_IDF_DEMUXER
74 /**
75  * Given filesize and width, calculate height (assume font_height of 16)
76  */
77 static void calculate_height(AVCodecParameters *par, uint64_t fsize)
78 {
79     par->height = (fsize / ((par->width>>3)*2)) << 4;
80 }
81 #endif
82
83 #if CONFIG_BINTEXT_DEMUXER
84 static const uint8_t next_magic[]={
85     0x1A, 0x1B, '[', '0', ';', '3', '0', ';', '4', '0', 'm', 'N', 'E', 'X', 'T', 0x00
86 };
87
88 static int next_tag_read(AVFormatContext *avctx, uint64_t *fsize)
89 {
90     AVIOContext *pb = avctx->pb;
91     char buf[36];
92     int len;
93     uint64_t start_pos = avio_size(pb) - 256;
94
95     avio_seek(pb, start_pos, SEEK_SET);
96     if (avio_read(pb, buf, sizeof(next_magic)) != sizeof(next_magic))
97         return -1;
98     if (memcmp(buf, next_magic, sizeof(next_magic)))
99         return -1;
100     if (avio_r8(pb) != 0x01)
101         return -1;
102
103     *fsize -= 256;
104
105 #define GET_EFI2_META(name,size) \
106     len = avio_r8(pb); \
107     if (len < 1 || len > size) \
108         return -1; \
109     if (avio_read(pb, buf, size) == size && *buf) { \
110         buf[len] = 0; \
111         av_dict_set(&avctx->metadata, name, buf, 0); \
112     }
113
114     GET_EFI2_META("filename",  12)
115     GET_EFI2_META("author",    20)
116     GET_EFI2_META("publisher", 20)
117     GET_EFI2_META("title",     35)
118
119     return 0;
120 }
121
122 static void predict_width(AVCodecParameters *par, uint64_t fsize, int got_width)
123 {
124     /** attempt to guess width */
125     if (!got_width)
126         par->width = fsize > 4000 ? (160<<3) : (80<<3);
127 }
128
129 static int bin_probe(const AVProbeData *p)
130 {
131     const uint8_t *d = p->buf;
132     int magic = 0, sauce = 0;
133     int invisible = 0;
134     int i;
135
136     if (p->buf_size > 256)
137         magic = !memcmp(d + p->buf_size - 256, next_magic, sizeof(next_magic));
138     if (p->buf_size > 128)
139         sauce = !memcmp(d + p->buf_size - 128, "SAUCE00", 7);
140
141     if (magic)
142         return AVPROBE_SCORE_EXTENSION + 1;
143
144     if (av_match_ext(p->filename, "bin")) {
145         AVCodecParameters par;
146         int got_width = 0;
147         par.width = par.height = 0;
148         if (sauce)
149             return AVPROBE_SCORE_EXTENSION + 1;
150
151         predict_width(&par, p->buf_size, got_width);
152         if (par.width < 8)
153             return 0;
154         calculate_height(&par, p->buf_size);
155         if (par.height <= 0)
156             return 0;
157
158         for (i = 0; i < p->buf_size - 256;  i+=2) {
159             if ((d[i+1] & 15) == (d[i+1] >> 4) && d[i] && d[i] != 0xFF && d[i] != ' ') {
160                 invisible ++;
161             }
162         }
163
164         if (par.width * par.height * 2 / (8*16) == p->buf_size)
165             return AVPROBE_SCORE_MAX / 2;
166         return 0;
167     }
168
169     if (sauce)
170         return 1;
171
172     return 0;
173 }
174
175
176 static int bintext_read_header(AVFormatContext *s)
177 {
178     BinDemuxContext *bin = s->priv_data;
179     AVIOContext *pb = s->pb;
180     int ret;
181     AVStream *st = init_stream(s);
182     if (!st)
183         return AVERROR(ENOMEM);
184     st->codecpar->codec_id    = AV_CODEC_ID_BINTEXT;
185
186     if ((ret = ff_alloc_extradata(st->codecpar, 2)) < 0)
187         return ret;
188     st->codecpar->extradata[0] = 16;
189     st->codecpar->extradata[1] = 0;
190
191     if (pb->seekable & AVIO_SEEKABLE_NORMAL) {
192         int got_width = 0;
193         bin->fsize = avio_size(pb);
194         if (ff_sauce_read(s, &bin->fsize, &got_width, 0) < 0)
195             next_tag_read(s, &bin->fsize);
196         if (!bin->width) {
197             predict_width(st->codecpar, bin->fsize, got_width);
198             if (st->codecpar->width < 8)
199                 return AVERROR_INVALIDDATA;
200             calculate_height(st->codecpar, bin->fsize);
201         }
202         avio_seek(pb, 0, SEEK_SET);
203     }
204     return 0;
205 }
206 #endif /* CONFIG_BINTEXT_DEMUXER */
207
208 #if CONFIG_XBIN_DEMUXER
209 static int xbin_probe(const AVProbeData *p)
210 {
211     const uint8_t *d = p->buf;
212
213     if (AV_RL32(d) == MKTAG('X','B','I','N') && d[4] == 0x1A &&
214         AV_RL16(d+5) > 0 && AV_RL16(d+5) <= 160 &&
215         d[9] > 0 && d[9] <= 32)
216         return AVPROBE_SCORE_MAX;
217     return 0;
218 }
219
220 static int xbin_read_header(AVFormatContext *s)
221 {
222     BinDemuxContext *bin = s->priv_data;
223     AVIOContext *pb = s->pb;
224     char fontheight, flags;
225     int ret;
226     AVStream *st = init_stream(s);
227     if (!st)
228         return AVERROR(ENOMEM);
229
230     avio_skip(pb, 5);
231     st->codecpar->width   = avio_rl16(pb)<<3;
232     st->codecpar->height  = avio_rl16(pb);
233     fontheight         = avio_r8(pb);
234     st->codecpar->height *= fontheight;
235     flags              = avio_r8(pb);
236
237     st->codecpar->extradata_size = 2;
238     if ((flags & BINTEXT_PALETTE))
239         st->codecpar->extradata_size += 48;
240     if ((flags & BINTEXT_FONT))
241         st->codecpar->extradata_size += fontheight * (flags & 0x10 ? 512 : 256);
242     st->codecpar->codec_id    = flags & 4 ? AV_CODEC_ID_XBIN : AV_CODEC_ID_BINTEXT;
243
244     ret = ff_alloc_extradata(st->codecpar, st->codecpar->extradata_size);
245     if (ret < 0)
246         return ret;
247     st->codecpar->extradata[0] = fontheight;
248     st->codecpar->extradata[1] = flags;
249     if (avio_read(pb, st->codecpar->extradata + 2, st->codecpar->extradata_size - 2) < 0)
250         return AVERROR(EIO);
251
252     if (pb->seekable & AVIO_SEEKABLE_NORMAL) {
253         bin->fsize = avio_size(pb) - 9 - st->codecpar->extradata_size;
254         ff_sauce_read(s, &bin->fsize, NULL, 0);
255         avio_seek(pb, 9 + st->codecpar->extradata_size, SEEK_SET);
256     }
257
258     return 0;
259 }
260 #endif /* CONFIG_XBIN_DEMUXER */
261
262 #if CONFIG_ADF_DEMUXER
263 static int adf_read_header(AVFormatContext *s)
264 {
265     BinDemuxContext *bin = s->priv_data;
266     AVIOContext *pb = s->pb;
267     AVStream *st;
268     int ret;
269
270     if (avio_r8(pb) != 1)
271         return AVERROR_INVALIDDATA;
272
273     st = init_stream(s);
274     if (!st)
275         return AVERROR(ENOMEM);
276     st->codecpar->codec_id    = AV_CODEC_ID_BINTEXT;
277
278     if ((ret = ff_alloc_extradata(st->codecpar, 2 + 48 + 4096)) < 0)
279         return ret;
280     st->codecpar->extradata[0] = 16;
281     st->codecpar->extradata[1] = BINTEXT_PALETTE|BINTEXT_FONT;
282
283     if (avio_read(pb, st->codecpar->extradata + 2, 24) < 0)
284         return AVERROR(EIO);
285     avio_skip(pb, 144);
286     if (avio_read(pb, st->codecpar->extradata + 2 + 24, 24) < 0)
287         return AVERROR(EIO);
288     if (avio_read(pb, st->codecpar->extradata + 2 + 48, 4096) < 0)
289         return AVERROR(EIO);
290
291     if (pb->seekable & AVIO_SEEKABLE_NORMAL) {
292         int got_width = 0;
293         bin->fsize = avio_size(pb) - 1 - 192 - 4096;
294         st->codecpar->width = 80<<3;
295         ff_sauce_read(s, &bin->fsize, &got_width, 0);
296         if (!bin->width)
297             calculate_height(st->codecpar, bin->fsize);
298         avio_seek(pb, 1 + 192 + 4096, SEEK_SET);
299     }
300     return 0;
301 }
302 #endif /* CONFIG_ADF_DEMUXER */
303
304 #if CONFIG_IDF_DEMUXER
305 static const uint8_t idf_magic[] = {
306     0x04, 0x31, 0x2e, 0x34, 0x00, 0x00, 0x00, 0x00, 0x4f, 0x00, 0x15, 0x00
307 };
308
309 static int idf_probe(const AVProbeData *p)
310 {
311     if (p->buf_size < sizeof(idf_magic))
312         return 0;
313     if (!memcmp(p->buf, idf_magic, sizeof(idf_magic)))
314         return AVPROBE_SCORE_MAX;
315     return 0;
316 }
317
318 static int idf_read_header(AVFormatContext *s)
319 {
320     BinDemuxContext *bin = s->priv_data;
321     AVIOContext *pb = s->pb;
322     AVStream *st;
323     int got_width = 0, ret;
324
325     if (!(pb->seekable & AVIO_SEEKABLE_NORMAL))
326         return AVERROR(EIO);
327
328     st = init_stream(s);
329     if (!st)
330         return AVERROR(ENOMEM);
331     st->codecpar->codec_id    = AV_CODEC_ID_IDF;
332
333     if ((ret = ff_alloc_extradata(st->codecpar, 2 + 48 + 4096)) < 0)
334         return ret;
335     st->codecpar->extradata[0] = 16;
336     st->codecpar->extradata[1] = BINTEXT_PALETTE|BINTEXT_FONT;
337
338     avio_seek(pb, avio_size(pb) - 4096 - 48, SEEK_SET);
339
340     if (avio_read(pb, st->codecpar->extradata + 2 + 48, 4096) < 0)
341         return AVERROR(EIO);
342     if (avio_read(pb, st->codecpar->extradata + 2, 48) < 0)
343         return AVERROR(EIO);
344
345     bin->fsize = avio_size(pb) - 12 - 4096 - 48;
346     ff_sauce_read(s, &bin->fsize, &got_width, 0);
347     if (!bin->width)
348         calculate_height(st->codecpar, bin->fsize);
349     avio_seek(pb, 12, SEEK_SET);
350     return 0;
351 }
352 #endif /* CONFIG_IDF_DEMUXER */
353
354 static int read_packet(AVFormatContext *s,
355                            AVPacket *pkt)
356 {
357     BinDemuxContext *bin = s->priv_data;
358
359     if (bin->fsize > 0) {
360         if (av_get_packet(s->pb, pkt, bin->fsize) < 0)
361             return AVERROR(EIO);
362         bin->fsize = -1; /* done */
363     } else if (!bin->fsize) {
364         if (avio_feof(s->pb))
365             return AVERROR(EIO);
366         if (av_get_packet(s->pb, pkt, bin->chars_per_frame) < 0)
367             return AVERROR(EIO);
368     } else {
369         return AVERROR(EIO);
370     }
371
372     pkt->flags |= AV_PKT_FLAG_KEY;
373     return 0;
374 }
375
376 #define OFFSET(x) offsetof(BinDemuxContext, x)
377 static const AVOption options[] = {
378     { "linespeed", "set simulated line speed (bytes per second)", OFFSET(chars_per_frame), AV_OPT_TYPE_INT, {.i64 = 6000}, 1, INT_MAX, AV_OPT_FLAG_DECODING_PARAM},
379     { "video_size", "set video size, such as 640x480 or hd720.", OFFSET(width), AV_OPT_TYPE_IMAGE_SIZE, {.str = NULL}, 0, 0, AV_OPT_FLAG_DECODING_PARAM },
380     { "framerate", "set framerate (frames per second)", OFFSET(framerate), AV_OPT_TYPE_VIDEO_RATE, {.str = "25"}, 0, INT_MAX, AV_OPT_FLAG_DECODING_PARAM },
381     { NULL },
382 };
383
384 #define CLASS(name) \
385 (const AVClass[1]){{ \
386     .class_name     = name, \
387     .item_name      = av_default_item_name, \
388     .option         = options, \
389     .version        = LIBAVUTIL_VERSION_INT, \
390 }}
391
392 #if CONFIG_BINTEXT_DEMUXER
393 AVInputFormat ff_bintext_demuxer = {
394     .name           = "bin",
395     .long_name      = NULL_IF_CONFIG_SMALL("Binary text"),
396     .priv_data_size = sizeof(BinDemuxContext),
397     .read_probe     = bin_probe,
398     .read_header    = bintext_read_header,
399     .read_packet    = read_packet,
400     .priv_class     = CLASS("Binary text demuxer"),
401 };
402 #endif
403
404 #if CONFIG_XBIN_DEMUXER
405 AVInputFormat ff_xbin_demuxer = {
406     .name           = "xbin",
407     .long_name      = NULL_IF_CONFIG_SMALL("eXtended BINary text (XBIN)"),
408     .priv_data_size = sizeof(BinDemuxContext),
409     .read_probe     = xbin_probe,
410     .read_header    = xbin_read_header,
411     .read_packet    = read_packet,
412     .priv_class     = CLASS("eXtended BINary text (XBIN) demuxer"),
413 };
414 #endif
415
416 #if CONFIG_ADF_DEMUXER
417 AVInputFormat ff_adf_demuxer = {
418     .name           = "adf",
419     .long_name      = NULL_IF_CONFIG_SMALL("Artworx Data Format"),
420     .priv_data_size = sizeof(BinDemuxContext),
421     .read_header    = adf_read_header,
422     .read_packet    = read_packet,
423     .extensions     = "adf",
424     .priv_class     = CLASS("Artworx Data Format demuxer"),
425 };
426 #endif
427
428 #if CONFIG_IDF_DEMUXER
429 AVInputFormat ff_idf_demuxer = {
430     .name           = "idf",
431     .long_name      = NULL_IF_CONFIG_SMALL("iCE Draw File"),
432     .priv_data_size = sizeof(BinDemuxContext),
433     .read_probe     = idf_probe,
434     .read_header    = idf_read_header,
435     .read_packet    = read_packet,
436     .extensions     = "idf",
437     .priv_class     = CLASS("iCE Draw File demuxer"),
438 };
439 #endif