libavformat/flvdec.c

   1 /*
   2  * FLV demuxer
   3  * Copyright (c) 2003 The FFmpeg Project.
   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  * This demuxer will generate a 1 byte extradata for VP6F content.
  23  * It is composed of:
  24  *  - upper 4bits: difference between encoded width and visible width
  25  *  - lower 4bits: difference between encoded height and visible height
  26  */
  27 #include "avformat.h"
  28 #include "flv.h"
  29
  30 static int flv_probe(AVProbeData *p)
  31 {
  32     const uint8_t *d;
  33
  34     if (p->buf_size < 6)
  35         return 0;
  36     d = p->buf;
  37     if (d[0] == 'F' && d[1] == 'L' && d[2] == 'V') {
  38         return 50;
  39     }
  40     return 0;
  41 }
  42
  43 static int amf_get_string(ByteIOContext *ioc, char *buffer, int buffsize) {
  44     int length = get_be16(ioc);
  45     if(length >= buffsize) {
  46         url_fskip(ioc, length);
  47         return -1;
  48     }
  49
  50     get_buffer(ioc, buffer, length);
  51
  52     buffer[length] = '\0';
  53
  54     return length;
  55 }
  56
  57 static int flv_read_header(AVFormatContext *s,
  58                            AVFormatParameters *ap)
  59 {
  60     int offset, flags;
  61     AVStream *st;
  62
  63     url_fskip(&s->pb, 4);
  64     flags = get_byte(&s->pb);
  65
  66     if(flags & FLV_HEADER_FLAG_HASVIDEO){
  67         st = av_new_stream(s, 0);
  68         if (!st)
  69             return AVERROR_NOMEM;
  70         st->codec->codec_type = CODEC_TYPE_VIDEO;
  71         av_set_pts_info(st, 24, 1, 1000); /* 24 bit pts in ms */
  72     }
  73     if(flags & FLV_HEADER_FLAG_HASAUDIO){
  74         st = av_new_stream(s, 1);
  75         if (!st)
  76             return AVERROR_NOMEM;
  77         st->codec->codec_type = CODEC_TYPE_AUDIO;
  78         av_set_pts_info(st, 24, 1, 1000); /* 24 bit pts in ms */
  79     }
  80
  81     offset = get_be32(&s->pb);
  82     url_fseek(&s->pb, offset, SEEK_SET);
  83
  84     s->start_time = 0;
  85
  86     return 0;
  87 }
  88
  89 static int flv_read_packet(AVFormatContext *s, AVPacket *pkt)
  90 {
  91     int ret, i, type, size, pts, flags, is_audio, next, pos;
  92     AVStream *st = NULL;
  93
  94  for(;;){
  95     pos = url_ftell(&s->pb);
  96     url_fskip(&s->pb, 4); /* size of previous packet */
  97     type = get_byte(&s->pb);
  98     size = get_be24(&s->pb);
  99     pts = get_be24(&s->pb);
 100 //    av_log(s, AV_LOG_DEBUG, "type:%d, size:%d, pts:%d\n", type, size, pts);
 101     if (url_feof(&s->pb))
 102         return AVERROR_IO;
 103     url_fskip(&s->pb, 4); /* reserved */
 104     flags = 0;
 105
 106     if(size == 0)
 107         continue;
 108
 109     next= size + url_ftell(&s->pb);
 110
 111     if (type == FLV_TAG_TYPE_AUDIO) {
 112         is_audio=1;
 113         flags = get_byte(&s->pb);
 114     } else if (type == FLV_TAG_TYPE_VIDEO) {
 115         is_audio=0;
 116         flags = get_byte(&s->pb);
 117     } else if (type == FLV_TAG_TYPE_META && size > 13+1+4) {
 118         url_fskip(&s->pb, 13); //onMetaData blah
 119         if(get_byte(&s->pb) == 8){
 120             url_fskip(&s->pb, 4);
 121         }
 122         while(url_ftell(&s->pb) + 5 < next){
 123             char tmp[128];
 124             int type, len;
 125             double d= 0;
 126
 127             if(amf_get_string(&s->pb, tmp, sizeof(tmp))<0)
 128                 break;
 129
 130             type= get_byte(&s->pb);
 131             if(type == AMF_DATA_TYPE_NUMBER){
 132                 d= av_int2dbl(get_be64(&s->pb));
 133             }else if(type == AMF_DATA_TYPE_STRING){
 134                 amf_get_string(&s->pb, NULL, 0);
 135             }else if(type == AMF_DATA_TYPE_MIXEDARRAY){
 136                 //array
 137                 break;
 138             }else if(type == AMF_DATA_TYPE_DATE){
 139                 d= av_int2dbl(get_be64(&s->pb));
 140                 get_be16(&s->pb);
 141             }
 142
 143             if(!strcmp(tmp, "duration")){
 144                 s->duration = d*AV_TIME_BASE;
 145             }else if(!strcmp(tmp, "videodatarate")){
 146             }else if(!strcmp(tmp, "audiodatarate")){
 147             }
 148         }
 149         url_fseek(&s->pb, next, SEEK_SET);
 150         continue;
 151     } else {
 152         /* skip packet */
 153         av_log(s, AV_LOG_ERROR, "skipping flv packet: type %d, size %d, flags %d\n", type, size, flags);
 154         url_fseek(&s->pb, next, SEEK_SET);
 155         continue;
 156     }
 157
 158     /* now find stream */
 159     for(i=0;i<s->nb_streams;i++) {
 160         st = s->streams[i];
 161         if (st->id == is_audio)
 162             break;
 163     }
 164     if(i == s->nb_streams){
 165         av_log(NULL, AV_LOG_ERROR, "invalid stream\n");
 166         url_fseek(&s->pb, next, SEEK_SET);
 167         continue;
 168     }
 169 //    av_log(NULL, AV_LOG_DEBUG, "%d %X %d \n", is_audio, flags, st->discard);
 170     if(  (st->discard >= AVDISCARD_NONKEY && !((flags & FLV_VIDEO_FRAMETYPE_MASK) == FLV_FRAME_KEY ||         is_audio))
 171        ||(st->discard >= AVDISCARD_BIDIR  &&  ((flags & FLV_VIDEO_FRAMETYPE_MASK) == FLV_FRAME_DISP_INTER && !is_audio))
 172        || st->discard >= AVDISCARD_ALL
 173        ){
 174         url_fseek(&s->pb, next, SEEK_SET);
 175         continue;
 176     }
 177     if ((flags & FLV_VIDEO_FRAMETYPE_MASK) == FLV_FRAME_KEY)
 178         av_add_index_entry(st, pos, pts, size, 0, AVINDEX_KEYFRAME);
 179     break;
 180  }
 181
 182     // if not streamed and no duration from metadata then seek to end to find the duration from the timestamps
 183     if(!url_is_streamed(&s->pb) && s->duration==AV_NOPTS_VALUE){
 184         int size;
 185         const int pos= url_ftell(&s->pb);
 186         const int fsize= url_fsize(&s->pb);
 187         url_fseek(&s->pb, fsize-4, SEEK_SET);
 188         size= get_be32(&s->pb);
 189         url_fseek(&s->pb, fsize-3-size, SEEK_SET);
 190         if(size == get_be24(&s->pb) + 11){
 191             s->duration= get_be24(&s->pb) * (int64_t)AV_TIME_BASE / 1000;
 192         }
 193         url_fseek(&s->pb, pos, SEEK_SET);
 194     }
 195
 196     if(is_audio){
 197         if(st->codec->sample_rate == 0){
 198             st->codec->channels = (flags & FLV_AUDIO_CHANNEL_MASK) == FLV_STEREO ? 2 : 1;
 199             if((flags & FLV_AUDIO_CODECID_MASK) == FLV_CODECID_NELLYMOSER_8HZ_MONO)
 200                 st->codec->sample_rate= 8000;
 201             else
 202                 st->codec->sample_rate = (44100 << ((flags & FLV_AUDIO_SAMPLERATE_MASK) >> FLV_AUDIO_SAMPLERATE_OFFSET) >> 3);
 203             switch(flags & FLV_AUDIO_CODECID_MASK) {
 204             case FLV_CODECID_PCM_BE: if (flags & FLV_AUDIO_SAMPLESIZE_MASK) st->codec->codec_id = CODEC_ID_PCM_S16BE;
 205                     else st->codec->codec_id = CODEC_ID_PCM_S8; break;
 206             case FLV_CODECID_ADPCM : st->codec->codec_id = CODEC_ID_ADPCM_SWF;                       break;
 207             case FLV_CODECID_MP3   : st->codec->codec_id = CODEC_ID_MP3      ; st->need_parsing = 1; break;
 208             // this is not listed at FLV but at SWF, strange...
 209             case FLV_CODECID_PCM_LE: if (flags & FLV_AUDIO_SAMPLESIZE_MASK) st->codec->codec_id = CODEC_ID_PCM_S16LE;
 210                     else st->codec->codec_id = CODEC_ID_PCM_S8; break;
 211             default:
 212                     av_log(s, AV_LOG_INFO, "Unsupported audio codec (%x)\n", flags >> 4);
 213                 st->codec->codec_tag= (flags >> 4);
 214             }
 215             st->codec->bits_per_sample = (flags & FLV_AUDIO_SAMPLESIZE_MASK) ? 16 : 8;
 216         }
 217     }else{
 218             switch(flags & FLV_VIDEO_CODECID_MASK){
 219             case FLV_CODECID_H263  : st->codec->codec_id = CODEC_ID_FLV1   ; break;
 220             case FLV_CODECID_SCREEN: st->codec->codec_id = CODEC_ID_FLASHSV; break;
 221             case FLV_CODECID_VP6   :
 222                 st->codec->codec_id = CODEC_ID_VP6F;
 223                 if (st->codec->extradata_size != 1) {
 224                     st->codec->extradata_size = 1;
 225                     st->codec->extradata = av_malloc(1);
 226                 }
 227                 /* width and height adjustment */
 228                 st->codec->extradata[0] = get_byte(&s->pb);
 229                 size--;
 230                 break;
 231             default:
 232                 av_log(s, AV_LOG_INFO, "Unsupported video codec (%x)\n", flags & FLV_VIDEO_CODECID_MASK);
 233                 st->codec->codec_tag = flags & FLV_VIDEO_CODECID_MASK;
 234             }
 235     }
 236
 237     ret= av_get_packet(&s->pb, pkt, size - 1);
 238     if (ret <= 0) {
 239         return AVERROR_IO;
 240     }
 241     /* note: we need to modify the packet size here to handle the last
 242        packet */
 243     pkt->size = ret;
 244     pkt->pts = pts;
 245     pkt->stream_index = st->index;
 246
 247     if (is_audio || ((flags & FLV_VIDEO_FRAMETYPE_MASK) == FLV_FRAME_KEY))
 248         pkt->flags |= PKT_FLAG_KEY;
 249
 250     return ret;
 251 }
 252
 253 static int flv_read_close(AVFormatContext *s)
 254 {
 255     return 0;
 256 }
 257
 258 static int flv_read_seek(AVFormatContext *s, int stream_index, int64_t timestamp, int flags)
 259 {
 260     AVStream *st = s->streams[stream_index];
 261     int index = av_index_search_timestamp(st, timestamp, flags);
 262     if (index < 0)
 263         return -1;
 264     url_fseek(&s->pb, st->index_entries[index].pos, SEEK_SET);
 265
 266     return 0;
 267 }
 268
 269 AVInputFormat flv_demuxer = {
 270     "flv",
 271     "flv format",
 272     0,
 273     flv_probe,
 274     flv_read_header,
 275     flv_read_packet,
 276     flv_read_close,
 277     flv_read_seek,
 278     .extensions = "flv",
 279     .value = CODEC_ID_FLV1,
 280 };