/* * Binary text demuxer * eXtended BINary text (XBIN) demuxer * Artworx Data Format demuxer * iCEDraw File demuxer * Copyright (c) 2010 Peter Ross <pross@xvid.org> * * This file is part of FFmpeg. * * FFmpeg is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public * License as published by the Free Software Foundation; either * version 2.1 of the License, or (at your option) any later version. * * FFmpeg is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with FFmpeg; if not, write to the Free Software * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA */ /** * @file * Binary text demuxer * eXtended BINary text (XBIN) demuxer * Artworx Data Format demuxer * iCEDraw File demuxer */ #include "config_components.h" #include "libavutil/intreadwrite.h"#include "libavutil/opt.h"#include "libavutil/parseutils.h"#include "avformat.h"#include "demux.h"#include "internal.h"#include "sauce.h"#include "libavcodec/bintext.h" typedef struct {    const AVClass *class;    int chars_per_frame; /**< characters to send decoder per frame;                              set by private options as characters per second, and then                              converted to characters per frame at runtime */    int width, height;    /**< video size (WxH pixels) (private option) */    AVRational framerate; /**< frames per second (private option) */    uint64_t fsize;  /**< file size less metadata buffer */} BinDemuxContext; static AVStream * init_stream(AVFormatContext *s){    BinDemuxContext *bin = s->priv_data;    AVStream *st = avformat_new_stream(s, NULL);    if (!st)        return NULL;    st->codecpar->codec_tag   = 0;    st->codecpar->codec_type  = AVMEDIA_TYPE_VIDEO;     if (!bin->width) {        st->codecpar->width  = (80<<3);        st->codecpar->height = (25<<4);    }     avpriv_set_pts_info(st, 60, bin->framerate.den, bin->framerate.num);     /* simulate tty display speed */    bin->chars_per_frame = av_clip(av_q2d(st->time_base) * bin->chars_per_frame, 1, INT_MAX);     return st;} #if CONFIG_BINTEXT_DEMUXER | CONFIG_ADF_DEMUXER | CONFIG_IDF_DEMUXER/** * Given filesize and width, calculate height (assume font_height of 16) */static void calculate_height(AVCodecParameters *par, uint64_t fsize){    par->height = (fsize / ((par->width>>3)*2)) << 4;}#endif #if CONFIG_BINTEXT_DEMUXERstatic const uint8_t next_magic[]={    0x1A, 0x1B, '[', '0', ';', '3', '0', ';', '4', '0', 'm', 'N', 'E', 'X', 'T', 0x00}; static int next_tag_read(AVFormatContext *avctx, uint64_t *fsize){    AVIOContext *pb = avctx->pb;    char buf[36];    int len;    int64_t start_pos = avio_size(pb);     if (start_pos < 256)        return AVERROR_INVALIDDATA;     avio_seek(pb, start_pos - 256, SEEK_SET);    if (avio_read(pb, buf, sizeof(next_magic)) != sizeof(next_magic))        return -1;    if (memcmp(buf, next_magic, sizeof(next_magic)))        return -1;    if (avio_r8(pb) != 0x01)        return -1;     *fsize -= 256; #define GET_EFI2_META(name,size) \    len = avio_r8(pb); \    if (len < 1 || len > size) \        return -1; \    if (avio_read(pb, buf, size) == size && *buf) { \        buf[len] = 0; \        av_dict_set(&avctx->metadata, name, buf, 0); \    }     GET_EFI2_META("filename",  12)    GET_EFI2_META("author",    20)    GET_EFI2_META("publisher", 20)    GET_EFI2_META("title",     35)     return 0;} static void predict_width(AVCodecParameters *par, uint64_t fsize, int got_width){    /** attempt to guess width */    if (!got_width)        par->width = fsize > 4000 ? (160<<3) : (80<<3);} static int bin_probe(const AVProbeData *p){    const uint8_t *d = p->buf;    int magic = 0, sauce = 0;     if (p->buf_size > 256)        magic = !memcmp(d + p->buf_size - 256, next_magic, sizeof(next_magic));    if (p->buf_size > 128)        sauce = !memcmp(d + p->buf_size - 128, "SAUCE00", 7);     if (magic)        return AVPROBE_SCORE_EXTENSION + 1;     if (av_match_ext(p->filename, "bin")) {        AVCodecParameters par;        int got_width = 0;        par.width = par.height = 0;        if (sauce)            return AVPROBE_SCORE_EXTENSION + 1;         predict_width(&par, p->buf_size, got_width);        if (par.width < 8)            return 0;        calculate_height(&par, p->buf_size);        if (par.height <= 0)            return 0;         if (par.width * par.height * 2 / (8*16) == p->buf_size)            return AVPROBE_SCORE_MAX / 2;        return 0;    }     if (sauce)        return 1;     return 0;}  static int bintext_read_header(AVFormatContext *s){    BinDemuxContext *bin = s->priv_data;    AVIOContext *pb = s->pb;    int ret;    AVStream *st = init_stream(s);    if (!st)        return AVERROR(ENOMEM);    st->codecpar->codec_id    = AV_CODEC_ID_BINTEXT;     if ((ret = ff_alloc_extradata(st->codecpar, 2)) < 0)        return ret;    st->codecpar->extradata[0] = 16;    st->codecpar->extradata[1] = 0;     if (pb->seekable & AVIO_SEEKABLE_NORMAL) {        int got_width = 0;        bin->fsize = avio_size(pb);        if (ff_sauce_read(s, &bin->fsize, &got_width, 0) < 0)            next_tag_read(s, &bin->fsize);        if (!bin->width) {            predict_width(st->codecpar, bin->fsize, got_width);            if (st->codecpar->width < 8)                return AVERROR_INVALIDDATA;            calculate_height(st->codecpar, bin->fsize);        }        avio_seek(pb, 0, SEEK_SET);    }    return 0;}#endif /* CONFIG_BINTEXT_DEMUXER */ #if CONFIG_XBIN_DEMUXERstatic int xbin_probe(const AVProbeData *p){    const uint8_t *d = p->buf;     if (AV_RL32(d) == MKTAG('X','B','I','N') && d[4] == 0x1A &&        AV_RL16(d+5) > 0 && AV_RL16(d+5) <= 160 &&        d[9] > 0 && d[9] <= 32)        return AVPROBE_SCORE_MAX;    return 0;} static int xbin_read_header(AVFormatContext *s){    BinDemuxContext *bin = s->priv_data;    AVIOContext *pb = s->pb;    char fontheight, flags;    int ret;    AVStream *st = init_stream(s);    if (!st)        return AVERROR(ENOMEM);     avio_skip(pb, 5);    st->codecpar->width   = avio_rl16(pb)<<3;    st->codecpar->height  = avio_rl16(pb);    fontheight         = avio_r8(pb);    st->codecpar->height *= fontheight;    flags              = avio_r8(pb);     st->codecpar->extradata_size = 2;    if ((flags & BINTEXT_PALETTE))        st->codecpar->extradata_size += 48;    if ((flags & BINTEXT_FONT))        st->codecpar->extradata_size += fontheight * (flags & 0x10 ? 512 : 256);    st->codecpar->codec_id    = flags & 4 ? AV_CODEC_ID_XBIN : AV_CODEC_ID_BINTEXT;     ret = ff_alloc_extradata(st->codecpar, st->codecpar->extradata_size);    if (ret < 0)        return ret;    st->codecpar->extradata[0] = fontheight;    st->codecpar->extradata[1] = flags;    if (avio_read(pb, st->codecpar->extradata + 2, st->codecpar->extradata_size - 2) < 0)        return AVERROR(EIO);     if (pb->seekable & AVIO_SEEKABLE_NORMAL) {        int64_t fsize =  avio_size(pb);        if (fsize < 9 + st->codecpar->extradata_size)            return 0;        bin->fsize = fsize - 9 - st->codecpar->extradata_size;        ff_sauce_read(s, &bin->fsize, NULL, 0);        avio_seek(pb, 9 + st->codecpar->extradata_size, SEEK_SET);    }     return 0;}#endif /* CONFIG_XBIN_DEMUXER */ #if CONFIG_ADF_DEMUXERstatic int adf_read_header(AVFormatContext *s){    BinDemuxContext *bin = s->priv_data;    AVIOContext *pb = s->pb;    AVStream *st;    int ret;     if (avio_r8(pb) != 1)        return AVERROR_INVALIDDATA;     st = init_stream(s);    if (!st)        return AVERROR(ENOMEM);    st->codecpar->codec_id    = AV_CODEC_ID_BINTEXT;     if ((ret = ff_alloc_extradata(st->codecpar, 2 + 48 + 4096)) < 0)        return ret;    st->codecpar->extradata[0] = 16;    st->codecpar->extradata[1] = BINTEXT_PALETTE|BINTEXT_FONT;     if (avio_read(pb, st->codecpar->extradata + 2, 24) < 0)        return AVERROR(EIO);    avio_skip(pb, 144);    if (avio_read(pb, st->codecpar->extradata + 2 + 24, 24) < 0)        return AVERROR(EIO);    if (avio_read(pb, st->codecpar->extradata + 2 + 48, 4096) < 0)        return AVERROR(EIO);     if (pb->seekable & AVIO_SEEKABLE_NORMAL) {        int got_width = 0;        int64_t fsize =  avio_size(pb);        if (fsize < 1 + 192 + 4096)            return 0;        bin->fsize = fsize - 1 - 192 - 4096;        st->codecpar->width = 80<<3;        ff_sauce_read(s, &bin->fsize, &got_width, 0);        if (st->codecpar->width < 8)            return AVERROR_INVALIDDATA;        if (!bin->width)            calculate_height(st->codecpar, bin->fsize);        avio_seek(pb, 1 + 192 + 4096, SEEK_SET);    }    return 0;}#endif /* CONFIG_ADF_DEMUXER */ #if CONFIG_IDF_DEMUXERstatic const uint8_t idf_magic[] = {    0x04, 0x31, 0x2e, 0x34, 0x00, 0x00, 0x00, 0x00, 0x4f, 0x00, 0x15, 0x00}; static int idf_probe(const AVProbeData *p){    if (p->buf_size < sizeof(idf_magic))        return 0;    if (!memcmp(p->buf, idf_magic, sizeof(idf_magic)))        return AVPROBE_SCORE_MAX;    return 0;} static int idf_read_header(AVFormatContext *s){    BinDemuxContext *bin = s->priv_data;    AVIOContext *pb = s->pb;    AVStream *st;    int got_width = 0, ret;    int64_t fsize;     if (!(pb->seekable & AVIO_SEEKABLE_NORMAL))        return AVERROR(EIO);     st = init_stream(s);    if (!st)        return AVERROR(ENOMEM);    st->codecpar->codec_id    = AV_CODEC_ID_IDF;     if ((ret = ff_alloc_extradata(st->codecpar, 2 + 48 + 4096)) < 0)        return ret;    st->codecpar->extradata[0] = 16;    st->codecpar->extradata[1] = BINTEXT_PALETTE|BINTEXT_FONT;     fsize = avio_size(pb);    if (fsize < 12 + 4096 + 48)        return AVERROR_INVALIDDATA;    bin->fsize = fsize - 12 - 4096 - 48;     avio_seek(pb, bin->fsize + 12, SEEK_SET);     if (avio_read(pb, st->codecpar->extradata + 2 + 48, 4096) < 0)        return AVERROR(EIO);    if (avio_read(pb, st->codecpar->extradata + 2, 48) < 0)        return AVERROR(EIO);     ff_sauce_read(s, &bin->fsize, &got_width, 0);    if (st->codecpar->width < 8)        return AVERROR_INVALIDDATA;    if (!bin->width)        calculate_height(st->codecpar, bin->fsize);    avio_seek(pb, 12, SEEK_SET);    return 0;}#endif /* CONFIG_IDF_DEMUXER */ static int read_packet(AVFormatContext *s,                           AVPacket *pkt){    BinDemuxContext *bin = s->priv_data;     if (bin->fsize > 0) {        if (av_get_packet(s->pb, pkt, bin->fsize) < 0)            return AVERROR(EIO);        bin->fsize = -1; /* done */    } else if (!bin->fsize) {        if (avio_feof(s->pb))            return AVERROR(EIO);        if (av_get_packet(s->pb, pkt, bin->chars_per_frame) < 0)            return AVERROR(EIO);    } else {        return AVERROR(EIO);    }     pkt->flags |= AV_PKT_FLAG_KEY;    return 0;} #define OFFSET(x) offsetof(BinDemuxContext, x)static const AVOption options[] = {    { "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},    { "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 },    { "framerate", "set framerate (frames per second)", OFFSET(framerate), AV_OPT_TYPE_VIDEO_RATE, {.str = "25"}, 0, INT_MAX, AV_OPT_FLAG_DECODING_PARAM },    { NULL },}; #define CLASS(name) \(const AVClass[1]){{ \    .class_name     = name, \    .item_name      = av_default_item_name, \    .option         = options, \    .version        = LIBAVUTIL_VERSION_INT, \}} #if CONFIG_BINTEXT_DEMUXERconst FFInputFormat ff_bintext_demuxer = {    .p.name         = "bin",    .p.long_name    = NULL_IF_CONFIG_SMALL("Binary text"),    .p.priv_class   = CLASS("Binary text demuxer"),    .priv_data_size = sizeof(BinDemuxContext),    .read_probe     = bin_probe,    .read_header    = bintext_read_header,    .read_packet    = read_packet,};#endif #if CONFIG_XBIN_DEMUXERconst FFInputFormat ff_xbin_demuxer = {    .p.name         = "xbin",    .p.long_name    = NULL_IF_CONFIG_SMALL("eXtended BINary text (XBIN)"),    .p.priv_class   = CLASS("eXtended BINary text (XBIN) demuxer"),    .priv_data_size = sizeof(BinDemuxContext),    .read_probe     = xbin_probe,    .read_header    = xbin_read_header,    .read_packet    = read_packet,};#endif #if CONFIG_ADF_DEMUXERconst FFInputFormat ff_adf_demuxer = {    .p.name         = "adf",    .p.long_name    = NULL_IF_CONFIG_SMALL("Artworx Data Format"),    .p.extensions   = "adf",    .p.priv_class   = CLASS("Artworx Data Format demuxer"),    .priv_data_size = sizeof(BinDemuxContext),    .read_header    = adf_read_header,    .read_packet    = read_packet,};#endif #if CONFIG_IDF_DEMUXERconst FFInputFormat ff_idf_demuxer = {    .p.name         = "idf",    .p.long_name    = NULL_IF_CONFIG_SMALL("iCE Draw File"),    .p.extensions   = "idf",    .p.priv_class   = CLASS("iCE Draw File demuxer"),    .priv_data_size = sizeof(BinDemuxContext),    .read_probe     = idf_probe,    .read_header    = idf_read_header,    .read_packet    = read_packet,};#endif