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