git.sesse.net Git - ffmpeg/blob - libavformat/ast.c

   1 /*
   2  * AST demuxer
   3  * Copyright (c) 2012 Paul B Mahol
   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 #include "libavutil/channel_layout.h"
  23 #include "libavutil/intreadwrite.h"
  24 #include "avformat.h"
  25 #include "internal.h"
  26
  27 static int ast_probe(AVProbeData *p)
  28 {
  29     if (AV_RL32(p->buf) == MKTAG('S','T','R','M') &&
  30         AV_RB16(p->buf + 10) &&
  31         AV_RB16(p->buf + 12) &&
  32         AV_RB32(p->buf + 16))
  33         return AVPROBE_SCORE_MAX / 3 * 2;
  34     return 0;
  35 }
  36
  37 static int ast_read_header(AVFormatContext *s)
  38 {
  39     int codec, depth;
  40     AVStream *st;
  41
  42     st = avformat_new_stream(s, NULL);
  43     if (!st)
  44         return AVERROR(ENOMEM);
  45
  46     avio_skip(s->pb, 8);
  47     codec = avio_rb16(s->pb);
  48     switch (codec) {
  49     case 0:
  50         st->codec->codec_id = AV_CODEC_ID_ADPCM_AFC;
  51         break;
  52     case 1:
  53         st->codec->codec_id = AV_CODEC_ID_PCM_S16BE_PLANAR;
  54         break;
  55     default:
  56         av_log(s, AV_LOG_ERROR, "unsupported codec %d\n", codec);
  57     }
  58
  59     depth = avio_rb16(s->pb);
  60     if (depth != 16) {
  61         av_log_ask_for_sample(s, "unsupported depth %d\n", depth);
  62         return AVERROR_INVALIDDATA;
  63     }
  64
  65     st->codec->codec_type = AVMEDIA_TYPE_AUDIO;
  66     st->codec->channels = avio_rb16(s->pb);
  67     if (!st->codec->channels)
  68         return AVERROR_INVALIDDATA;
  69
  70     if (st->codec->channels == 2)
  71         st->codec->channel_layout = AV_CH_LAYOUT_STEREO;
  72     else if (st->codec->channels == 4)
  73         st->codec->channel_layout = AV_CH_LAYOUT_4POINT0;
  74
  75     avio_skip(s->pb, 2);
  76     st->codec->sample_rate = avio_rb32(s->pb);
  77     if (st->codec->sample_rate <= 0)
  78         return AVERROR_INVALIDDATA;
  79     st->start_time         = 0;
  80     st->duration           = avio_rb32(s->pb);
  81     avio_skip(s->pb, 40);
  82     avpriv_set_pts_info(st, 64, 1, st->codec->sample_rate);
  83
  84     return 0;
  85 }
  86
  87 static int ast_read_packet(AVFormatContext *s, AVPacket *pkt)
  88 {
  89     uint32_t type, size;
  90     int64_t pos;
  91     int ret;
  92
  93     if (url_feof(s->pb))
  94         return AVERROR_EOF;
  95
  96     pos  = avio_tell(s->pb);
  97     type = avio_rl32(s->pb);
  98     size = avio_rb32(s->pb);
  99     if (size > INT_MAX / s->streams[0]->codec->channels)
 100         return AVERROR_INVALIDDATA;
 101
 102     size *= s->streams[0]->codec->channels;
 103     if ((ret = avio_skip(s->pb, 24)) < 0) // padding
 104         return ret;
 105
 106     if (type == MKTAG('B','L','C','K')) {
 107         ret = av_get_packet(s->pb, pkt, size);
 108         pkt->stream_index = 0;
 109         pkt->pos = pos;
 110     } else {
 111         av_log(s, AV_LOG_ERROR, "unknown chunk %x\n", type);
 112         avio_skip(s->pb, size);
 113         ret = AVERROR_INVALIDDATA;
 114     }
 115
 116     return ret;
 117 }
 118
 119 AVInputFormat ff_ast_demuxer = {
 120     .name           = "ast",
 121     .long_name      = NULL_IF_CONFIG_SMALL("AST (Audio Stream)"),
 122     .read_probe     = ast_probe,
 123     .read_header    = ast_read_header,
 124     .read_packet    = ast_read_packet,
 125     .extensions     = "ast",
 126     .flags          = AVFMT_GENERIC_INDEX,
 127 };