2001-07-22 17:18:56 +03:00
/*
2006-10-23 11:57:54 +03:00
* ASF compatible demuxer
2009-01-19 17:46:40 +02:00
* Copyright ( c ) 2000 , 2001 Fabrice Bellard
2001-07-22 17:18:56 +03:00
*
2006-10-07 18:30:46 +03:00
* This file is part of FFmpeg .
*
* FFmpeg is free software ; you can redistribute it and / or
2002-05-26 01:34:32 +03:00
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation ; either
2006-10-07 18:30:46 +03:00
* version 2.1 of the License , or ( at your option ) any later version .
2001-07-22 17:18:56 +03:00
*
2006-10-07 18:30:46 +03:00
* FFmpeg is distributed in the hope that it will be useful ,
2001-07-22 17:18:56 +03:00
* but WITHOUT ANY WARRANTY ; without even the implied warranty of
2002-05-26 01:34:32 +03:00
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE . See the GNU
* Lesser General Public License for more details .
2001-07-22 17:18:56 +03:00
*
2002-05-26 01:34:32 +03:00
* You should have received a copy of the GNU Lesser General Public
2006-10-07 18:30:46 +03:00
* License along with FFmpeg ; if not , write to the Free Software
2006-01-13 00:43:26 +02:00
* Foundation , Inc . , 51 Franklin Street , Fifth Floor , Boston , MA 02110 - 1301 USA
2001-07-22 17:18:56 +03:00
*/
2008-05-09 14:56:36 +03:00
2014-03-10 17:35:59 +03:00
# include <inttypes.h>
2012-08-21 13:40:41 +03:00
# include "libavutil/attributes.h"
2013-02-08 15:54:36 +03:00
# include "libavutil/avassert.h"
2013-02-06 02:44:00 +03:00
# include "libavutil/avstring.h"
2011-05-19 14:31:05 +03:00
# include "libavutil/bswap.h"
2008-05-09 14:56:36 +03:00
# include "libavutil/common.h"
2011-05-22 13:46:29 +03:00
# include "libavutil/dict.h"
2013-03-27 20:36:51 +03:00
# include "libavutil/internal.h"
2011-06-04 14:58:23 +03:00
# include "libavutil/mathematics.h"
2012-04-06 16:54:23 +03:00
# include "libavutil/opt.h"
2001-07-22 17:18:56 +03:00
# include "avformat.h"
2011-03-14 21:38:59 +02:00
# include "avio_internal.h"
2013-02-06 02:44:00 +03:00
# include "avlanguage.h"
2012-06-21 23:17:43 +03:00
# include "id3v2.h"
2013-02-06 02:44:00 +03:00
# include "internal.h"
2006-07-12 03:09:34 +03:00
# include "riff.h"
2004-03-05 23:34:30 +02:00
# include "asf.h"
2007-10-20 17:25:02 +03:00
# include "asfcrypt.h"
2001-07-22 17:18:56 +03:00
2015-06-28 13:05:45 +02:00
typedef struct ASFPayload {
uint8_t type ;
uint16_t size ;
} ASFPayload ;
2015-06-15 14:54:45 +02:00
typedef struct ASFStream {
2015-06-28 13:05:45 +02:00
int num ;
unsigned char seq ;
/* use for reading */
AVPacket pkt ;
int frag_offset ;
int packet_obj_size ;
int timestamp ;
int64_t duration ;
int skip_to_key ;
int pkt_clean ;
2015-06-15 14:54:45 +02:00
2015-06-28 13:05:45 +02:00
int ds_span ; /* descrambling */
int ds_packet_size ;
int ds_chunk_size ;
2015-06-15 14:54:45 +02:00
2015-06-28 13:05:45 +02:00
int64_t packet_pos ;
2011-02-08 22:46:47 +02:00
2015-06-28 13:05:45 +02:00
uint16_t stream_language_index ;
2012-04-06 16:54:23 +03:00
2015-06-28 13:05:45 +02:00
int palette_changed ;
uint32_t palette [ 256 ] ;
2012-04-06 16:54:23 +03:00
2015-06-28 13:05:45 +02:00
int payload_ext_ct ;
ASFPayload payload [ 8 ] ;
} ASFStream ;
2004-03-06 00:10:50 +02:00
2014-09-22 10:19:33 +03:00
typedef struct ASFContext {
2012-04-06 16:54:23 +03:00
const AVClass * class ;
2011-02-08 22:46:47 +02:00
int asfid2avid [ 128 ] ; ///< conversion table from asf ID 2 AVStream ID
ASFStream streams [ 128 ] ; ///< it's max number and it's not that big
uint32_t stream_bitrates [ 128 ] ; ///< max number of streams, bitrate for each (for streaming)
2011-02-09 22:55:53 +02:00
AVRational dar [ 128 ] ;
2011-02-08 22:46:47 +02:00
char stream_languages [ 128 ] [ 6 ] ; ///< max number of streams, language for each (RFC1766, e.g. en-US)
/* non streamed additonnal info */
/* packet filling */
int packet_size_left ;
/* only for reading */
uint64_t data_offset ; ///< beginning of the first data packet
uint64_t data_object_offset ; ///< data object offset (excl. GUID & size)
uint64_t data_object_size ; ///< size of the data object
int index_read ;
ASFMainHeader hdr ;
int packet_flags ;
int packet_property ;
int packet_timestamp ;
int packet_segsizetype ;
int packet_segments ;
int packet_seq ;
int packet_replic_size ;
int packet_key_frame ;
int packet_padsize ;
unsigned int packet_frag_offset ;
unsigned int packet_frag_size ;
int64_t packet_frag_timestamp ;
2014-07-13 04:23:54 +03:00
int ts_is_pts ;
2011-02-08 22:46:47 +02:00
int packet_multi_size ;
int packet_time_delta ;
int packet_time_start ;
int64_t packet_pos ;
int stream_index ;
2013-02-06 02:44:00 +03:00
ASFStream * asf_st ; ///< currently decoded stream
2012-04-06 16:54:23 +03:00
int no_resync_search ;
2015-03-02 20:04:03 +02:00
int export_xmp ;
2015-07-22 01:09:14 +02:00
int uses_std_ecc ;
2011-02-08 22:46:47 +02:00
} ASFContext ;
2012-04-06 16:54:23 +03:00
static const AVOption options [ ] = {
2013-02-06 02:44:00 +03:00
{ " no_resync_search " , " Don't try to resynchronize by looking for a certain optional start code " , offsetof ( ASFContext , no_resync_search ) , AV_OPT_TYPE_INT , { . i64 = 0 } , 0 , 1 , AV_OPT_FLAG_DECODING_PARAM } ,
2015-03-02 20:04:03 +02:00
{ " export_xmp " , " Export full XMP metadata " , offsetof ( ASFContext , export_xmp ) , AV_OPT_TYPE_INT , { . i64 = 0 } , 0 , 1 , AV_OPT_FLAG_DECODING_PARAM } ,
2012-04-06 16:54:23 +03:00
{ NULL } ,
} ;
static const AVClass asf_class = {
. class_name = " asf demuxer " ,
. item_name = av_default_item_name ,
. option = options ,
. version = LIBAVUTIL_VERSION_INT ,
} ;
2004-01-14 16:45:53 +02:00
# undef NDEBUG
# include <assert.h>
2010-06-24 18:10:06 +03:00
# define ASF_MAX_STREAMS 127
2015-06-29 16:23:15 +02:00
# define FRAME_HEADER_SIZE 6
// Fix Me! FRAME_HEADER_SIZE may be different.
// (7 is known to be too large for GipsyGuitar.wmv)
2004-03-06 00:10:50 +02:00
2011-07-03 17:57:26 +03:00
# ifdef DEBUG
2009-03-18 16:03:40 +02:00
static const ff_asf_guid stream_bitrate_guid = { /* (http://get.to/sdp) */
2007-02-21 01:44:11 +02:00
0xce , 0x75 , 0xf8 , 0x7b , 0x8d , 0x46 , 0xd1 , 0x11 , 0x8d , 0x82 , 0x00 , 0x60 , 0x97 , 0xc9 , 0xa2 , 0xb2
} ;
2001-07-22 17:18:56 +03:00
2013-02-06 02:44:00 +03:00
# define PRINT_IF_GUID(g, cmp) \
if ( ! ff_guidcmp ( g , & cmp ) ) \
2015-03-16 10:57:35 +02:00
av_log ( NULL , AV_LOG_TRACE , " (GUID: %s) " , # cmp )
2003-02-25 15:12:41 +02:00
2013-02-06 16:35:25 +03:00
static void print_guid ( ff_asf_guid * g )
2001-07-22 17:18:56 +03:00
{
int i ;
2009-02-26 18:17:17 +02:00
PRINT_IF_GUID ( g , ff_asf_header ) ;
else PRINT_IF_GUID ( g , ff_asf_file_header ) ;
else PRINT_IF_GUID ( g , ff_asf_stream_header ) ;
else PRINT_IF_GUID ( g , ff_asf_audio_stream ) ;
else PRINT_IF_GUID ( g , ff_asf_audio_conceal_none ) ;
else PRINT_IF_GUID ( g , ff_asf_video_stream ) ;
else PRINT_IF_GUID ( g , ff_asf_video_conceal_none ) ;
else PRINT_IF_GUID ( g , ff_asf_command_stream ) ;
else PRINT_IF_GUID ( g , ff_asf_comment_header ) ;
else PRINT_IF_GUID ( g , ff_asf_codec_comment_header ) ;
else PRINT_IF_GUID ( g , ff_asf_codec_comment1_header ) ;
else PRINT_IF_GUID ( g , ff_asf_data_header ) ;
2012-04-03 10:44:10 +03:00
else PRINT_IF_GUID ( g , ff_asf_simple_index_header ) ;
2009-02-26 18:17:17 +02:00
else PRINT_IF_GUID ( g , ff_asf_head1_guid ) ;
else PRINT_IF_GUID ( g , ff_asf_head2_guid ) ;
else PRINT_IF_GUID ( g , ff_asf_my_guid ) ;
else PRINT_IF_GUID ( g , ff_asf_ext_stream_header ) ;
else PRINT_IF_GUID ( g , ff_asf_extended_content_header ) ;
else PRINT_IF_GUID ( g , ff_asf_ext_stream_embed_stream_header ) ;
else PRINT_IF_GUID ( g , ff_asf_ext_stream_audio_stream ) ;
else PRINT_IF_GUID ( g , ff_asf_metadata_header ) ;
2013-02-06 16:35:27 +03:00
else PRINT_IF_GUID ( g , ff_asf_metadata_library_header ) ;
2010-03-08 15:13:18 +02:00
else PRINT_IF_GUID ( g , ff_asf_marker_header ) ;
2007-02-21 01:44:11 +02:00
else PRINT_IF_GUID ( g , stream_bitrate_guid ) ;
2009-05-12 15:35:46 +03:00
else PRINT_IF_GUID ( g , ff_asf_language_guid ) ;
2003-02-25 15:12:41 +02:00
else
2015-03-16 10:57:35 +02:00
av_log ( NULL , AV_LOG_TRACE , " (GUID: unknown) " ) ;
2013-02-06 02:44:00 +03:00
for ( i = 0 ; i < 16 ; i + + )
2015-03-16 10:57:35 +02:00
av_log ( NULL , AV_LOG_TRACE , " 0x%02x, " , ( * g ) [ i ] ) ;
av_log ( NULL , AV_LOG_TRACE , " } \n " ) ;
2001-07-22 17:18:56 +03:00
}
2006-02-12 04:24:37 +02:00
# undef PRINT_IF_GUID
2008-07-13 22:41:07 +03:00
# else
2015-05-07 00:54:06 +02:00
# define print_guid(g) while(0)
2001-07-22 17:18:56 +03:00
# endif
2002-05-20 19:31:13 +03:00
static int asf_probe ( AVProbeData * pd )
{
/* check file header */
2010-12-26 03:26:29 +02:00
if ( ! ff_guidcmp ( pd - > buf , & ff_asf_header ) )
2002-05-20 19:31:13 +03:00
return AVPROBE_SCORE_MAX ;
else
return 0 ;
}
2013-02-06 16:35:26 +03:00
/* size of type 2 (BOOL) is 32bit for "Extended Content Description Object"
* but 16 bit for " Metadata Object " and " Metadata Library Object " */
static int get_value ( AVIOContext * pb , int type , int type2_size )
2013-02-06 02:44:00 +03:00
{
switch ( type ) {
case 2 :
2013-02-06 16:35:26 +03:00
return ( type2_size = = 32 ) ? avio_rl32 ( pb ) : avio_rl16 ( pb ) ;
2013-02-06 02:44:00 +03:00
case 3 :
return avio_rl32 ( pb ) ;
case 4 :
return avio_rl64 ( pb ) ;
case 5 :
return avio_rl16 ( pb ) ;
default :
return INT_MIN ;
2007-02-01 01:32:01 +02:00
}
}
2012-06-21 23:17:43 +03:00
/* MSDN claims that this should be "compatible with the ID3 frame, APIC",
* but in reality this is only loosely similar */
static int asf_read_picture ( AVFormatContext * s , int len )
{
2013-02-06 02:44:00 +03:00
AVPacket pkt = { 0 } ;
2012-06-21 23:17:43 +03:00
const CodecMime * mime = ff_id3v2_mime_tags ;
2013-02-06 02:44:00 +03:00
enum AVCodecID id = AV_CODEC_ID_NONE ;
2012-06-21 23:17:43 +03:00
char mimetype [ 64 ] ;
uint8_t * desc = NULL ;
AVStream * st = NULL ;
int ret , type , picsize , desc_len ;
/* type + picsize + mime + desc */
if ( len < 1 + 4 + 2 + 2 ) {
av_log ( s , AV_LOG_ERROR , " Invalid attached picture size: %d. \n " , len ) ;
return AVERROR_INVALIDDATA ;
}
/* picture type */
type = avio_r8 ( s - > pb ) ;
len - - ;
if ( type > = FF_ARRAY_ELEMS ( ff_id3v2_picture_types ) | | type < 0 ) {
av_log ( s , AV_LOG_WARNING , " Unknown attached picture type: %d. \n " , type ) ;
type = 0 ;
}
/* picture data size */
picsize = avio_rl32 ( s - > pb ) ;
2013-02-06 02:44:00 +03:00
len - = 4 ;
2012-06-21 23:17:43 +03:00
/* picture MIME type */
len - = avio_get_str16le ( s - > pb , len , mimetype , sizeof ( mimetype ) ) ;
2012-08-05 12:11:04 +03:00
while ( mime - > id ! = AV_CODEC_ID_NONE ) {
2012-06-21 23:17:43 +03:00
if ( ! strncmp ( mime - > str , mimetype , sizeof ( mimetype ) ) ) {
id = mime - > id ;
break ;
}
mime + + ;
}
2012-08-05 12:11:04 +03:00
if ( id = = AV_CODEC_ID_NONE ) {
2012-06-21 23:17:43 +03:00
av_log ( s , AV_LOG_ERROR , " Unknown attached picture mimetype: %s. \n " ,
mimetype ) ;
return 0 ;
}
if ( picsize > = len ) {
av_log ( s , AV_LOG_ERROR , " Invalid attached picture data size: %d >= %d. \n " ,
picsize , len ) ;
return AVERROR_INVALIDDATA ;
}
/* picture description */
desc_len = ( len - picsize ) * 2 + 1 ;
desc = av_malloc ( desc_len ) ;
if ( ! desc )
return AVERROR ( ENOMEM ) ;
len - = avio_get_str16le ( s - > pb , len - picsize , desc , desc_len ) ;
ret = av_get_packet ( s - > pb , & pkt , picsize ) ;
if ( ret < 0 )
goto fail ;
2013-02-06 02:44:00 +03:00
st = avformat_new_stream ( s , NULL ) ;
2013-02-08 15:54:36 +03:00
if ( ! st ) {
2012-06-21 23:17:43 +03:00
ret = AVERROR ( ENOMEM ) ;
goto fail ;
}
2013-02-06 02:44:00 +03:00
st - > disposition | = AV_DISPOSITION_ATTACHED_PIC ;
st - > codec - > codec_type = AVMEDIA_TYPE_VIDEO ;
st - > codec - > codec_id = id ;
st - > attached_pic = pkt ;
2012-06-21 23:17:43 +03:00
st - > attached_pic . stream_index = st - > index ;
2013-02-06 02:44:00 +03:00
st - > attached_pic . flags | = AV_PKT_FLAG_KEY ;
2012-06-21 23:17:43 +03:00
if ( * desc )
av_dict_set ( & st - > metadata , " title " , desc , AV_DICT_DONT_STRDUP_VAL ) ;
else
av_freep ( & desc ) ;
av_dict_set ( & st - > metadata , " comment " , ff_id3v2_picture_types [ type ] , 0 ) ;
return 0 ;
fail :
av_freep ( & desc ) ;
av_free_packet ( & pkt ) ;
return ret ;
}
2013-02-08 13:06:37 +03:00
static void get_id3_tag ( AVFormatContext * s , int len )
{
ID3v2ExtraMeta * id3v2_extra_meta = NULL ;
2014-04-18 07:49:40 +03:00
ff_id3v2_read ( s , ID3v2_DEFAULT_MAGIC , & id3v2_extra_meta , len ) ;
2013-02-08 13:06:37 +03:00
if ( id3v2_extra_meta )
ff_id3v2_parse_apic ( s , & id3v2_extra_meta ) ;
ff_id3v2_free_extra_meta ( & id3v2_extra_meta ) ;
}
2013-02-06 16:35:26 +03:00
static void get_tag ( AVFormatContext * s , const char * key , int type , int len , int type2_size )
2009-02-04 00:12:06 +02:00
{
2015-03-02 20:04:03 +02:00
ASFContext * asf = s - > priv_data ;
char * value = NULL ;
2011-03-03 21:11:45 +02:00
int64_t off = avio_tell ( s - > pb ) ;
2013-03-05 23:56:46 +03:00
# define LEN 22
2009-12-13 22:27:29 +02:00
2013-03-05 23:56:46 +03:00
if ( ( unsigned ) len > = ( UINT_MAX - LEN ) / 2 )
2009-12-13 22:27:29 +02:00
return ;
2015-03-02 20:04:03 +02:00
if ( ! asf - > export_xmp & & ! strncmp ( key , " xmp " , 3 ) )
goto finish ;
2013-03-05 23:56:46 +03:00
value = av_malloc ( 2 * len + LEN ) ;
2009-12-13 22:27:29 +02:00
if ( ! value )
2011-01-27 11:41:49 +02:00
goto finish ;
2009-12-13 22:27:29 +02:00
2015-07-02 00:15:13 +02:00
switch ( type ) {
case ASF_UNICODE :
2013-02-06 02:44:00 +03:00
avio_get_str16le ( s - > pb , len , value , 2 * len + 1 ) ;
2015-07-02 00:15:13 +02:00
break ;
case - 1 : // ASCI
2011-11-14 02:53:50 +03:00
avio_read ( s - > pb , value , len ) ;
2011-10-16 06:04:48 +03:00
value [ len ] = 0 ;
2015-07-02 00:15:13 +02:00
break ;
case ASF_BYTE_ARRAY :
2013-02-08 13:06:37 +03:00
if ( ! strcmp ( key , " WM/Picture " ) ) { // handle cover art
asf_read_picture ( s , len ) ;
} else if ( ! strcmp ( key , " ID3 " ) ) { // handle ID3 tag
get_id3_tag ( s , len ) ;
} else {
av_log ( s , AV_LOG_VERBOSE , " Unsupported byte array in tag %s. \n " , key ) ;
}
goto finish ;
2015-07-02 00:15:13 +02:00
case ASF_BOOL :
case ASF_DWORD :
case ASF_QWORD :
case ASF_WORD : {
2013-02-06 16:35:26 +03:00
uint64_t num = get_value ( s - > pb , type , type2_size ) ;
2013-03-05 23:56:46 +03:00
snprintf ( value , LEN , " % " PRIu64 , num ) ;
2015-07-02 00:15:13 +02:00
break ;
}
case ASF_GUID :
2013-02-06 16:35:27 +03:00
av_log ( s , AV_LOG_DEBUG , " Unsupported GUID value in tag %s. \n " , key ) ;
goto finish ;
2015-07-02 00:15:13 +02:00
default :
2013-02-06 02:44:00 +03:00
av_log ( s , AV_LOG_DEBUG ,
" Unsupported value type %d in tag %s. \n " , type , key ) ;
2011-01-27 11:41:49 +02:00
goto finish ;
2009-02-04 00:12:06 +02:00
}
2011-02-09 22:56:00 +02:00
if ( * value )
2011-05-22 13:46:29 +03:00
av_dict_set ( & s - > metadata , key , value , 0 ) ;
2013-02-06 02:44:00 +03:00
2011-01-27 11:41:49 +02:00
finish :
2010-02-24 20:19:54 +02:00
av_freep ( & value ) ;
2011-02-28 15:57:54 +02:00
avio_seek ( s - > pb , off + len , SEEK_SET ) ;
2009-02-04 00:12:06 +02:00
}
2011-02-09 22:55:55 +02:00
static int asf_read_file_properties ( AVFormatContext * s , int64_t size )
{
ASFContext * asf = s - > priv_data ;
2011-02-20 12:04:12 +02:00
AVIOContext * pb = s - > pb ;
2011-02-09 22:55:55 +02:00
ff_get_guid ( pb , & asf - > hdr . guid ) ;
2013-02-06 02:44:00 +03:00
asf - > hdr . file_size = avio_rl64 ( pb ) ;
asf - > hdr . create_time = avio_rl64 ( pb ) ;
2011-02-21 17:43:01 +02:00
avio_rl64 ( pb ) ; /* number of packets */
2013-02-06 02:44:00 +03:00
asf - > hdr . play_time = avio_rl64 ( pb ) ;
asf - > hdr . send_time = avio_rl64 ( pb ) ;
asf - > hdr . preroll = avio_rl32 ( pb ) ;
asf - > hdr . ignore = avio_rl32 ( pb ) ;
asf - > hdr . flags = avio_rl32 ( pb ) ;
asf - > hdr . min_pktsize = avio_rl32 ( pb ) ;
asf - > hdr . max_pktsize = avio_rl32 ( pb ) ;
if ( asf - > hdr . min_pktsize > = ( 1U < < 29 ) )
2012-02-17 23:21:22 +03:00
return AVERROR_INVALIDDATA ;
2013-02-06 02:44:00 +03:00
asf - > hdr . max_bitrate = avio_rl32 ( pb ) ;
s - > packet_size = asf - > hdr . max_pktsize ;
2011-02-09 22:55:55 +02:00
return 0 ;
}
2011-02-09 22:55:57 +02:00
static int asf_read_stream_properties ( AVFormatContext * s , int64_t size )
{
ASFContext * asf = s - > priv_data ;
2011-02-20 12:04:12 +02:00
AVIOContext * pb = s - > pb ;
2011-02-09 22:55:57 +02:00
AVStream * st ;
ASFStream * asf_st ;
ff_asf_guid g ;
enum AVMediaType type ;
int type_specific_size , sizeX ;
unsigned int tag1 ;
int64_t pos1 , pos2 , start_time ;
2013-02-06 02:44:00 +03:00
int test_for_ext_stream_audio , is_dvr_ms_audio = 0 ;
2011-02-09 22:55:57 +02:00
if ( s - > nb_streams = = ASF_MAX_STREAMS ) {
av_log ( s , AV_LOG_ERROR , " too many streams \n " ) ;
return AVERROR ( EINVAL ) ;
}
2011-03-03 21:11:45 +02:00
pos1 = avio_tell ( pb ) ;
2011-02-09 22:55:57 +02:00
2011-06-18 12:43:24 +03:00
st = avformat_new_stream ( s , NULL ) ;
2011-02-09 22:55:57 +02:00
if ( ! st )
return AVERROR ( ENOMEM ) ;
2011-11-29 21:28:15 +03:00
avpriv_set_pts_info ( st , 32 , 1 , 1000 ) ; /* 32 bit pts in ms */
2013-02-06 02:44:00 +03:00
start_time = asf - > hdr . preroll ;
2011-02-09 22:55:57 +02:00
2013-02-06 02:44:00 +03:00
if ( ! ( asf - > hdr . flags & 0x01 ) ) { // if we aren't streaming...
2012-10-23 17:11:38 +03:00
int64_t fsize = avio_size ( pb ) ;
2014-06-21 20:35:06 +03:00
if ( fsize < = 0 | | ( int64_t ) asf - > hdr . file_size < = 0 | |
2015-05-26 13:02:25 +02:00
20 * FFABS ( fsize - ( int64_t ) asf - > hdr . file_size ) < FFMIN ( fsize , asf - > hdr . file_size ) )
2012-10-23 17:11:38 +03:00
st - > duration = asf - > hdr . play_time /
2013-02-06 02:44:00 +03:00
( 10000000 / 1000 ) - start_time ;
2011-02-09 22:55:57 +02:00
}
ff_get_guid ( pb , & g ) ;
test_for_ext_stream_audio = 0 ;
if ( ! ff_guidcmp ( & g , & ff_asf_audio_stream ) ) {
type = AVMEDIA_TYPE_AUDIO ;
} else if ( ! ff_guidcmp ( & g , & ff_asf_video_stream ) ) {
type = AVMEDIA_TYPE_VIDEO ;
} else if ( ! ff_guidcmp ( & g , & ff_asf_jfif_media ) ) {
2013-02-06 02:44:00 +03:00
type = AVMEDIA_TYPE_VIDEO ;
2012-08-05 12:11:04 +03:00
st - > codec - > codec_id = AV_CODEC_ID_MJPEG ;
2011-02-09 22:55:57 +02:00
} else if ( ! ff_guidcmp ( & g , & ff_asf_command_stream ) ) {
type = AVMEDIA_TYPE_DATA ;
} else if ( ! ff_guidcmp ( & g , & ff_asf_ext_stream_embed_stream_header ) ) {
test_for_ext_stream_audio = 1 ;
2013-02-06 02:44:00 +03:00
type = AVMEDIA_TYPE_UNKNOWN ;
2011-02-09 22:55:57 +02:00
} else {
return - 1 ;
}
ff_get_guid ( pb , & g ) ;
2011-06-01 19:26:27 +03:00
avio_skip ( pb , 8 ) ; /* total_size */
2011-02-21 17:43:01 +02:00
type_specific_size = avio_rl32 ( pb ) ;
avio_rl32 ( pb ) ;
st - > id = avio_rl16 ( pb ) & 0x7f ; /* stream id */
2011-02-09 22:55:57 +02:00
// mapping of asf ID to AV stream ID;
asf - > asfid2avid [ st - > id ] = s - > nb_streams - 1 ;
2013-07-30 03:52:17 +03:00
asf_st = & asf - > streams [ st - > id ] ;
2011-02-09 22:55:57 +02:00
2011-02-21 17:43:01 +02:00
avio_rl32 ( pb ) ;
2011-02-09 22:55:57 +02:00
if ( test_for_ext_stream_audio ) {
ff_get_guid ( pb , & g ) ;
if ( ! ff_guidcmp ( & g , & ff_asf_ext_stream_audio_stream ) ) {
2013-02-06 02:44:00 +03:00
type = AVMEDIA_TYPE_AUDIO ;
is_dvr_ms_audio = 1 ;
2011-02-09 22:55:57 +02:00
ff_get_guid ( pb , & g ) ;
2011-02-21 17:43:01 +02:00
avio_rl32 ( pb ) ;
avio_rl32 ( pb ) ;
avio_rl32 ( pb ) ;
2011-02-09 22:55:57 +02:00
ff_get_guid ( pb , & g ) ;
2011-02-21 17:43:01 +02:00
avio_rl32 ( pb ) ;
2011-02-09 22:55:57 +02:00
}
}
st - > codec - > codec_type = type ;
if ( type = = AVMEDIA_TYPE_AUDIO ) {
2015-07-12 15:21:15 +02:00
int ret = ff_get_wav_header ( s , pb , st - > codec , type_specific_size , 0 ) ;
2011-04-12 18:44:20 +03:00
if ( ret < 0 )
return ret ;
2011-02-09 22:55:57 +02:00
if ( is_dvr_ms_audio ) {
// codec_id and codec_tag are unreliable in dvr_ms
// files. Set them later by probing stream.
2013-02-06 16:10:26 +03:00
st - > request_probe = 1 ;
2011-02-09 22:55:57 +02:00
st - > codec - > codec_tag = 0 ;
}
2013-02-06 02:44:00 +03:00
if ( st - > codec - > codec_id = = AV_CODEC_ID_AAC )
2011-02-09 22:55:57 +02:00
st - > need_parsing = AVSTREAM_PARSE_NONE ;
2013-02-06 02:44:00 +03:00
else
2011-02-09 22:55:57 +02:00
st - > need_parsing = AVSTREAM_PARSE_FULL ;
/* We have to init the frame size at some point .... */
2011-03-03 21:11:45 +02:00
pos2 = avio_tell ( pb ) ;
2011-02-09 22:55:57 +02:00
if ( size > = ( pos2 + 8 - pos1 + 24 ) ) {
2013-02-06 02:44:00 +03:00
asf_st - > ds_span = avio_r8 ( pb ) ;
2011-02-21 17:43:01 +02:00
asf_st - > ds_packet_size = avio_rl16 ( pb ) ;
2013-02-06 02:44:00 +03:00
asf_st - > ds_chunk_size = avio_rl16 ( pb ) ;
avio_rl16 ( pb ) ; // ds_data_size
avio_r8 ( pb ) ; // ds_silence_data
2011-02-09 22:55:57 +02:00
}
if ( asf_st - > ds_span > 1 ) {
2013-02-06 02:44:00 +03:00
if ( ! asf_st - > ds_chunk_size | |
( asf_st - > ds_packet_size / asf_st - > ds_chunk_size < = 1 ) | |
asf_st - > ds_packet_size % asf_st - > ds_chunk_size )
asf_st - > ds_span = 0 ; // disable descrambling
2011-02-09 22:55:57 +02:00
}
} else if ( type = = AVMEDIA_TYPE_VIDEO & &
2013-02-06 02:44:00 +03:00
size - ( avio_tell ( pb ) - pos1 + 24 ) > = 51 ) {
2011-02-21 17:43:01 +02:00
avio_rl32 ( pb ) ;
avio_rl32 ( pb ) ;
avio_r8 ( pb ) ;
avio_rl16 ( pb ) ; /* size */
2013-02-06 02:44:00 +03:00
sizeX = avio_rl32 ( pb ) ; /* size */
st - > codec - > width = avio_rl32 ( pb ) ;
2011-02-21 17:43:01 +02:00
st - > codec - > height = avio_rl32 ( pb ) ;
2011-02-09 22:55:57 +02:00
/* not available for asf */
2011-02-21 17:43:01 +02:00
avio_rl16 ( pb ) ; /* panes */
st - > codec - > bits_per_coded_sample = avio_rl16 ( pb ) ; /* depth */
2013-02-06 02:44:00 +03:00
tag1 = avio_rl32 ( pb ) ;
2011-03-15 10:14:38 +02:00
avio_skip ( pb , 20 ) ;
2011-02-09 22:55:57 +02:00
if ( sizeX > 40 ) {
2012-12-02 22:37:09 +03:00
st - > codec - > extradata_size = ffio_limit ( pb , sizeX - 40 ) ;
2013-02-06 02:44:00 +03:00
st - > codec - > extradata = av_mallocz ( st - > codec - > extradata_size +
2015-07-27 22:53:16 +02:00
AV_INPUT_BUFFER_PADDING_SIZE ) ;
2013-09-09 15:09:31 +03:00
if ( ! st - > codec - > extradata )
return AVERROR ( ENOMEM ) ;
2011-02-21 17:43:01 +02:00
avio_read ( pb , st - > codec - > extradata , st - > codec - > extradata_size ) ;
2011-02-09 22:55:57 +02:00
}
/* Extract palette from extradata if bpp <= 8 */
/* This code assumes that extradata contains only palette */
2011-10-30 20:02:42 +03:00
/* This is true for all paletted codecs implemented in libavcodec */
2011-02-09 22:55:57 +02:00
if ( st - > codec - > extradata_size & & ( st - > codec - > bits_per_coded_sample < = 8 ) ) {
# if HAVE_BIGENDIAN
2012-02-10 17:10:42 +03:00
int i ;
2013-02-06 02:44:00 +03:00
for ( i = 0 ; i < FFMIN ( st - > codec - > extradata_size , AVPALETTE_SIZE ) / 4 ; i + + )
asf_st - > palette [ i ] = av_bswap32 ( ( ( uint32_t * ) st - > codec - > extradata ) [ i ] ) ;
2011-02-09 22:55:57 +02:00
# else
2011-04-09 16:49:51 +03:00
memcpy ( asf_st - > palette , st - > codec - > extradata ,
FFMIN ( st - > codec - > extradata_size , AVPALETTE_SIZE ) ) ;
2011-02-09 22:55:57 +02:00
# endif
2011-04-09 16:49:51 +03:00
asf_st - > palette_changed = 1 ;
2011-02-09 22:55:57 +02:00
}
st - > codec - > codec_tag = tag1 ;
2013-02-06 02:44:00 +03:00
st - > codec - > codec_id = ff_codec_get_id ( ff_codec_bmp_tags , tag1 ) ;
if ( tag1 = = MKTAG ( ' D ' , ' V ' , ' R ' , ' ' ) ) {
2011-02-09 22:55:57 +02:00
st - > need_parsing = AVSTREAM_PARSE_FULL ;
2013-02-06 02:44:00 +03:00
/* issue658 contains wrong w/h and MS even puts a fake seq header
* with wrong w / h in extradata while a correct one is in the stream .
* maximum lameness */
st - > codec - > width =
2011-02-09 22:55:57 +02:00
st - > codec - > height = 0 ;
av_freep ( & st - > codec - > extradata ) ;
2013-02-06 02:44:00 +03:00
st - > codec - > extradata_size = 0 ;
2011-02-09 22:55:57 +02:00
}
2013-02-06 02:44:00 +03:00
if ( st - > codec - > codec_id = = AV_CODEC_ID_H264 )
2011-02-09 22:55:57 +02:00
st - > need_parsing = AVSTREAM_PARSE_FULL_ONCE ;
2014-06-03 15:44:46 +03:00
if ( st - > codec - > codec_id = = AV_CODEC_ID_MPEG4 )
st - > need_parsing = AVSTREAM_PARSE_FULL_ONCE ;
2011-02-09 22:55:57 +02:00
}
2011-03-03 21:11:45 +02:00
pos2 = avio_tell ( pb ) ;
2011-03-15 10:14:38 +02:00
avio_skip ( pb , size - ( pos2 - pos1 + 24 ) ) ;
2011-02-09 22:55:57 +02:00
return 0 ;
}
2011-02-09 22:55:55 +02:00
static int asf_read_ext_stream_properties ( AVFormatContext * s , int64_t size )
{
ASFContext * asf = s - > priv_data ;
2011-02-20 12:04:12 +02:00
AVIOContext * pb = s - > pb ;
2011-02-09 22:55:55 +02:00
ff_asf_guid g ;
int ext_len , payload_ext_ct , stream_ct , i ;
2011-06-01 19:26:27 +03:00
uint32_t leak_rate , stream_num ;
2011-02-09 22:55:55 +02:00
unsigned int stream_languageid_index ;
2011-02-21 17:43:01 +02:00
avio_rl64 ( pb ) ; // starttime
avio_rl64 ( pb ) ; // endtime
leak_rate = avio_rl32 ( pb ) ; // leak-datarate
avio_rl32 ( pb ) ; // bucket-datasize
avio_rl32 ( pb ) ; // init-bucket-fullness
avio_rl32 ( pb ) ; // alt-leak-datarate
avio_rl32 ( pb ) ; // alt-bucket-datasize
avio_rl32 ( pb ) ; // alt-init-bucket-fullness
avio_rl32 ( pb ) ; // max-object-size
avio_rl32 ( pb ) ; // flags (reliable,seekable,no_cleanpoints?,resend-live-cleanpoints, rest of bits reserved)
stream_num = avio_rl16 ( pb ) ; // stream-num
stream_languageid_index = avio_rl16 ( pb ) ; // stream-language-id-index
2011-02-09 22:55:55 +02:00
if ( stream_num < 128 )
asf - > streams [ stream_num ] . stream_language_index = stream_languageid_index ;
2011-02-21 17:43:01 +02:00
avio_rl64 ( pb ) ; // avg frametime in 100ns units
2013-02-06 02:44:00 +03:00
stream_ct = avio_rl16 ( pb ) ; // stream-name-count
payload_ext_ct = avio_rl16 ( pb ) ; // payload-extension-system-count
2011-02-09 22:55:55 +02:00
2012-11-03 07:35:35 +03:00
if ( stream_num < 128 ) {
2011-02-09 22:55:55 +02:00
asf - > stream_bitrates [ stream_num ] = leak_rate ;
2012-11-03 07:35:35 +03:00
asf - > streams [ stream_num ] . payload_ext_ct = 0 ;
}
2011-02-09 22:55:55 +02:00
2013-02-06 02:44:00 +03:00
for ( i = 0 ; i < stream_ct ; i + + ) {
2011-02-21 17:43:01 +02:00
avio_rl16 ( pb ) ;
ext_len = avio_rl16 ( pb ) ;
2011-03-15 10:14:38 +02:00
avio_skip ( pb , ext_len ) ;
2011-02-09 22:55:55 +02:00
}
2013-02-06 02:44:00 +03:00
for ( i = 0 ; i < payload_ext_ct ; i + + ) {
2012-11-03 07:35:35 +03:00
int size ;
2011-02-09 22:55:55 +02:00
ff_get_guid ( pb , & g ) ;
2012-11-03 07:35:35 +03:00
size = avio_rl16 ( pb ) ;
2013-02-06 02:44:00 +03:00
ext_len = avio_rl32 ( pb ) ;
2011-03-15 10:14:38 +02:00
avio_skip ( pb , ext_len ) ;
2012-11-03 07:35:35 +03:00
if ( stream_num < 128 & & i < FF_ARRAY_ELEMS ( asf - > streams [ stream_num ] . payload ) ) {
ASFPayload * p = & asf - > streams [ stream_num ] . payload [ i ] ;
p - > type = g [ 0 ] ;
p - > size = size ;
av_log ( s , AV_LOG_DEBUG , " Payload extension %x %d \n " , g [ 0 ] , p - > size ) ;
asf - > streams [ stream_num ] . payload_ext_ct + + ;
}
2011-02-09 22:55:55 +02:00
}
return 0 ;
}
static int asf_read_content_desc ( AVFormatContext * s , int64_t size )
{
2011-02-20 12:04:12 +02:00
AVIOContext * pb = s - > pb ;
2011-02-09 22:55:55 +02:00
int len1 , len2 , len3 , len4 , len5 ;
2011-02-21 17:43:01 +02:00
len1 = avio_rl16 ( pb ) ;
len2 = avio_rl16 ( pb ) ;
len3 = avio_rl16 ( pb ) ;
len4 = avio_rl16 ( pb ) ;
len5 = avio_rl16 ( pb ) ;
2013-02-06 16:35:26 +03:00
get_tag ( s , " title " , 0 , len1 , 32 ) ;
get_tag ( s , " author " , 0 , len2 , 32 ) ;
get_tag ( s , " copyright " , 0 , len3 , 32 ) ;
get_tag ( s , " comment " , 0 , len4 , 32 ) ;
2011-03-15 10:14:38 +02:00
avio_skip ( pb , len5 ) ;
2011-02-09 22:55:55 +02:00
return 0 ;
}
static int asf_read_ext_content_desc ( AVFormatContext * s , int64_t size )
{
2011-02-20 12:04:12 +02:00
AVIOContext * pb = s - > pb ;
2011-02-09 22:55:55 +02:00
ASFContext * asf = s - > priv_data ;
int desc_count , i , ret ;
2011-02-21 17:43:01 +02:00
desc_count = avio_rl16 ( pb ) ;
2013-02-06 02:44:00 +03:00
for ( i = 0 ; i < desc_count ; i + + ) {
int name_len , value_type , value_len ;
2011-02-09 22:55:55 +02:00
char name [ 1024 ] ;
2011-02-21 17:43:01 +02:00
name_len = avio_rl16 ( pb ) ;
2013-02-06 02:44:00 +03:00
if ( name_len % 2 ) // must be even, broken lavf versions wrote len-1
2011-02-09 22:55:55 +02:00
name_len + = 1 ;
if ( ( ret = avio_get_str16le ( pb , name_len , name , sizeof ( name ) ) ) < name_len )
2011-03-15 10:14:38 +02:00
avio_skip ( pb , name_len - ret ) ;
2011-02-21 17:43:01 +02:00
value_type = avio_rl16 ( pb ) ;
value_len = avio_rl16 ( pb ) ;
2013-02-06 02:44:00 +03:00
if ( ! value_type & & value_len % 2 )
2011-02-09 22:55:55 +02:00
value_len + = 1 ;
2013-02-06 02:44:00 +03:00
/* My sample has that stream set to 0 maybe that mean the container.
* ASF stream count starts at 1. I am using 0 to the container value
* since it ' s unused . */
if ( ! strcmp ( name , " AspectRatioX " ) )
2013-02-06 16:35:26 +03:00
asf - > dar [ 0 ] . num = get_value ( s - > pb , value_type , 32 ) ;
2013-02-06 02:44:00 +03:00
else if ( ! strcmp ( name , " AspectRatioY " ) )
2013-02-06 16:35:26 +03:00
asf - > dar [ 0 ] . den = get_value ( s - > pb , value_type , 32 ) ;
2013-02-06 02:44:00 +03:00
else
2013-02-06 16:35:26 +03:00
get_tag ( s , name , value_type , value_len , 32 ) ;
2011-02-09 22:55:55 +02:00
}
return 0 ;
}
static int asf_read_language_list ( AVFormatContext * s , int64_t size )
{
2011-02-20 12:04:12 +02:00
AVIOContext * pb = s - > pb ;
2011-02-09 22:55:55 +02:00
ASFContext * asf = s - > priv_data ;
int j , ret ;
2011-02-21 17:43:01 +02:00
int stream_count = avio_rl16 ( pb ) ;
2013-02-06 02:44:00 +03:00
for ( j = 0 ; j < stream_count ; j + + ) {
2011-02-09 22:55:55 +02:00
char lang [ 6 ] ;
2011-02-21 17:43:01 +02:00
unsigned int lang_len = avio_r8 ( pb ) ;
2013-02-06 02:44:00 +03:00
if ( ( ret = avio_get_str16le ( pb , lang_len , lang ,
sizeof ( lang ) ) ) < lang_len )
2011-03-15 10:14:38 +02:00
avio_skip ( pb , lang_len - ret ) ;
2011-02-09 22:55:55 +02:00
if ( j < 128 )
2013-02-06 02:44:00 +03:00
av_strlcpy ( asf - > stream_languages [ j ] , lang ,
sizeof ( * asf - > stream_languages ) ) ;
2011-02-09 22:55:55 +02:00
}
return 0 ;
}
static int asf_read_metadata ( AVFormatContext * s , int64_t size )
{
2011-02-20 12:04:12 +02:00
AVIOContext * pb = s - > pb ;
2011-02-09 22:55:55 +02:00
ASFContext * asf = s - > priv_data ;
2015-07-01 23:46:30 +02:00
int n , stream_num , name_len_utf16 , name_len_utf8 , value_len ;
2011-02-09 22:55:55 +02:00
int ret , i ;
2011-02-21 17:43:01 +02:00
n = avio_rl16 ( pb ) ;
2011-02-09 22:55:55 +02:00
2013-02-06 02:44:00 +03:00
for ( i = 0 ; i < n ; i + + ) {
2015-07-01 23:46:30 +02:00
uint8_t * name ;
2013-02-06 16:35:26 +03:00
int value_type ;
2011-02-09 22:55:55 +02:00
2013-02-06 02:44:00 +03:00
avio_rl16 ( pb ) ; // lang_list_index
stream_num = avio_rl16 ( pb ) ;
2015-07-01 23:46:30 +02:00
name_len_utf16 = avio_rl16 ( pb ) ;
2012-08-21 13:40:41 +03:00
value_type = avio_rl16 ( pb ) ; /* value_type */
2013-02-06 02:44:00 +03:00
value_len = avio_rl32 ( pb ) ;
2011-02-09 22:55:55 +02:00
2015-07-01 23:46:30 +02:00
name_len_utf8 = 2 * name_len_utf16 + 1 ;
name = av_malloc ( name_len_utf8 ) ;
if ( ! name )
return AVERROR ( ENOMEM ) ;
if ( ( ret = avio_get_str16le ( pb , name_len_utf16 , name , name_len_utf8 ) ) < name_len_utf16 )
avio_skip ( pb , name_len_utf16 - ret ) ;
2015-03-16 10:57:35 +02:00
av_log ( s , AV_LOG_TRACE , " %d stream %d name_len %2d type %d len %4d <%s> \n " ,
2015-07-01 23:46:30 +02:00
i , stream_num , name_len_utf16 , value_type , value_len , name ) ;
2013-02-06 16:35:26 +03:00
if ( ! strcmp ( name , " AspectRatioX " ) ) {
int aspect_x = get_value ( s - > pb , value_type , 16 ) ;
if ( stream_num < 128 )
asf - > dar [ stream_num ] . num = aspect_x ;
} else if ( ! strcmp ( name , " AspectRatioY " ) ) {
int aspect_y = get_value ( s - > pb , value_type , 16 ) ;
if ( stream_num < 128 )
asf - > dar [ stream_num ] . den = aspect_y ;
} else {
get_tag ( s , name , value_type , value_len , 16 ) ;
2011-02-09 22:55:55 +02:00
}
2015-07-02 09:23:35 +02:00
av_freep ( & name ) ;
2011-02-09 22:55:55 +02:00
}
return 0 ;
}
static int asf_read_marker ( AVFormatContext * s , int64_t size )
{
2011-02-20 12:04:12 +02:00
AVIOContext * pb = s - > pb ;
2013-09-19 18:50:34 +03:00
ASFContext * asf = s - > priv_data ;
2011-02-09 22:55:55 +02:00
int i , count , name_len , ret ;
char name [ 1024 ] ;
2011-02-21 17:43:01 +02:00
avio_rl64 ( pb ) ; // reserved 16 bytes
avio_rl64 ( pb ) ; // ...
count = avio_rl32 ( pb ) ; // markers count
avio_rl16 ( pb ) ; // reserved 2 bytes
name_len = avio_rl16 ( pb ) ; // name length
2013-02-06 02:44:00 +03:00
for ( i = 0 ; i < name_len ; i + + )
2011-02-21 17:43:01 +02:00
avio_r8 ( pb ) ; // skip the name
2011-02-09 22:55:55 +02:00
2013-02-06 02:44:00 +03:00
for ( i = 0 ; i < count ; i + + ) {
2011-02-09 22:55:55 +02:00
int64_t pres_time ;
int name_len ;
2011-02-21 17:43:01 +02:00
avio_rl64 ( pb ) ; // offset, 8 bytes
pres_time = avio_rl64 ( pb ) ; // presentation time
2013-09-19 18:50:34 +03:00
pres_time - = asf - > hdr . preroll * 10000 ;
2011-02-21 17:43:01 +02:00
avio_rl16 ( pb ) ; // entry length
avio_rl32 ( pb ) ; // send time
avio_rl32 ( pb ) ; // flags
name_len = avio_rl32 ( pb ) ; // name length
2013-02-06 02:44:00 +03:00
if ( ( ret = avio_get_str16le ( pb , name_len * 2 , name ,
sizeof ( name ) ) ) < name_len )
2011-03-15 10:14:38 +02:00
avio_skip ( pb , name_len - ret ) ;
2013-02-06 02:44:00 +03:00
avpriv_new_chapter ( s , i , ( AVRational ) { 1 , 10000000 } , pres_time ,
AV_NOPTS_VALUE , name ) ;
2011-02-09 22:55:55 +02:00
}
return 0 ;
}
2012-01-12 15:20:36 +03:00
static int asf_read_header ( AVFormatContext * s )
2001-07-22 17:18:56 +03:00
{
2002-05-20 19:31:13 +03:00
ASFContext * asf = s - > priv_data ;
2009-03-18 16:03:40 +02:00
ff_asf_guid g ;
2011-02-20 12:04:12 +02:00
AVIOContext * pb = s - > pb ;
2011-02-09 22:55:56 +02:00
int i ;
2003-02-11 18:35:48 +02:00
int64_t gsize ;
2001-07-22 17:18:56 +03:00
2010-12-26 03:26:29 +02:00
ff_get_guid ( pb , & g ) ;
if ( ff_guidcmp ( & g , & ff_asf_header ) )
2012-06-14 03:30:55 +03:00
return AVERROR_INVALIDDATA ;
2011-02-21 17:43:01 +02:00
avio_rl64 ( pb ) ;
avio_rl32 ( pb ) ;
avio_r8 ( pb ) ;
avio_r8 ( pb ) ;
2002-08-09 01:04:01 +03:00
memset ( & asf - > asfid2avid , - 1 , sizeof ( asf - > asfid2avid ) ) ;
2013-07-30 03:52:17 +03:00
for ( i = 0 ; i < 128 ; i + + )
asf - > streams [ i ] . stream_language_index = 128 ; // invalid stream index means no language info
2013-02-06 02:44:00 +03:00
for ( ; ; ) {
uint64_t gpos = avio_tell ( pb ) ;
2015-07-01 23:21:53 +02:00
int ret = 0 ;
2010-12-26 03:26:29 +02:00
ff_get_guid ( pb , & g ) ;
2011-02-21 17:43:01 +02:00
gsize = avio_rl64 ( pb ) ;
2001-07-22 17:18:56 +03:00
print_guid ( & g ) ;
2010-12-26 03:26:29 +02:00
if ( ! ff_guidcmp ( & g , & ff_asf_data_header ) ) {
2011-03-03 21:11:45 +02:00
asf - > data_object_offset = avio_tell ( pb ) ;
2013-02-06 02:44:00 +03:00
/* If not streaming, gsize is not unlimited (how?),
* and there is enough space in the file . . */
if ( ! ( asf - > hdr . flags & 0x01 ) & & gsize > = 100 )
2007-03-11 05:36:17 +02:00
asf - > data_object_size = gsize - 24 ;
2013-02-06 02:44:00 +03:00
else
2007-03-11 05:36:17 +02:00
asf - > data_object_size = ( uint64_t ) - 1 ;
break ;
}
2001-07-22 17:18:56 +03:00
if ( gsize < 24 )
2012-06-14 03:30:55 +03:00
return AVERROR_INVALIDDATA ;
2010-12-26 03:26:29 +02:00
if ( ! ff_guidcmp ( & g , & ff_asf_file_header ) ) {
2015-07-01 23:21:53 +02:00
ret = asf_read_file_properties ( s , gsize ) ;
2010-12-26 03:26:29 +02:00
} else if ( ! ff_guidcmp ( & g , & ff_asf_stream_header ) ) {
2015-07-01 23:21:53 +02:00
ret = asf_read_stream_properties ( s , gsize ) ;
2010-12-26 03:26:29 +02:00
} else if ( ! ff_guidcmp ( & g , & ff_asf_comment_header ) ) {
2011-02-09 22:55:55 +02:00
asf_read_content_desc ( s , gsize ) ;
2010-12-26 03:26:29 +02:00
} else if ( ! ff_guidcmp ( & g , & ff_asf_language_guid ) ) {
2011-02-09 22:55:55 +02:00
asf_read_language_list ( s , gsize ) ;
2010-12-26 03:26:29 +02:00
} else if ( ! ff_guidcmp ( & g , & ff_asf_extended_content_header ) ) {
2011-02-09 22:55:55 +02:00
asf_read_ext_content_desc ( s , gsize ) ;
2010-12-26 03:26:29 +02:00
} else if ( ! ff_guidcmp ( & g , & ff_asf_metadata_header ) ) {
2011-02-09 22:55:55 +02:00
asf_read_metadata ( s , gsize ) ;
2013-02-06 16:35:27 +03:00
} else if ( ! ff_guidcmp ( & g , & ff_asf_metadata_library_header ) ) {
asf_read_metadata ( s , gsize ) ;
2010-12-26 03:26:29 +02:00
} else if ( ! ff_guidcmp ( & g , & ff_asf_ext_stream_header ) ) {
2011-02-09 22:55:55 +02:00
asf_read_ext_stream_properties ( s , gsize ) ;
2006-01-21 01:48:16 +02:00
// there could be a optional stream properties object to follow
// if so the next iteration will pick it up
2010-06-23 05:04:55 +03:00
continue ;
2010-12-26 03:26:29 +02:00
} else if ( ! ff_guidcmp ( & g , & ff_asf_head1_guid ) ) {
ff_get_guid ( pb , & g ) ;
2011-06-01 19:26:27 +03:00
avio_skip ( pb , 6 ) ;
2010-06-23 05:04:55 +03:00
continue ;
2010-12-26 03:26:29 +02:00
} else if ( ! ff_guidcmp ( & g , & ff_asf_marker_header ) ) {
2011-02-09 22:55:55 +02:00
asf_read_marker ( s , gsize ) ;
2014-08-07 23:12:41 +03:00
} else if ( avio_feof ( pb ) ) {
2012-06-14 03:30:55 +03:00
return AVERROR_EOF ;
2001-07-22 17:18:56 +03:00
} else {
2009-10-12 02:09:33 +03:00
if ( ! s - > keylen ) {
2010-12-26 03:26:29 +02:00
if ( ! ff_guidcmp ( & g , & ff_asf_content_encryption ) ) {
2011-10-16 06:04:48 +03:00
unsigned int len ;
AVPacket pkt ;
2013-02-06 02:44:00 +03:00
av_log ( s , AV_LOG_WARNING ,
" DRM protected stream detected, decoding will likely fail! \n " ) ;
2011-10-16 06:04:48 +03:00
len = avio_rl32 ( pb ) ;
av_log ( s , AV_LOG_DEBUG , " Secret data: \n " ) ;
2015-01-17 21:39:43 +02:00
if ( ( ret = av_get_packet ( pb , & pkt , len ) ) < 0 )
return ret ;
av_hex_dump_log ( s , AV_LOG_DEBUG , pkt . data , pkt . size ) ;
av_free_packet ( & pkt ) ;
2011-10-16 06:04:48 +03:00
len = avio_rl32 ( pb ) ;
2013-02-10 14:18:50 +03:00
get_tag ( s , " ASF_Protection_Type " , - 1 , len , 32 ) ;
2011-10-16 06:04:48 +03:00
len = avio_rl32 ( pb ) ;
2013-02-10 14:18:50 +03:00
get_tag ( s , " ASF_Key_ID " , - 1 , len , 32 ) ;
2011-10-16 06:04:48 +03:00
len = avio_rl32 ( pb ) ;
2013-02-10 14:18:50 +03:00
get_tag ( s , " ASF_License_URL " , - 1 , len , 32 ) ;
2010-12-26 03:26:29 +02:00
} else if ( ! ff_guidcmp ( & g , & ff_asf_ext_content_encryption ) ) {
2013-02-06 02:44:00 +03:00
av_log ( s , AV_LOG_WARNING ,
" Ext DRM protected stream detected, decoding will likely fail! \n " ) ;
2011-10-16 05:41:11 +03:00
av_dict_set ( & s - > metadata , " encryption " , " ASF Extended Content Encryption " , 0 ) ;
2010-12-26 03:26:29 +02:00
} else if ( ! ff_guidcmp ( & g , & ff_asf_digital_signature ) ) {
2011-10-16 05:44:42 +03:00
av_log ( s , AV_LOG_INFO , " Digital signature detected! \n " ) ;
2009-10-12 02:09:33 +03:00
}
}
2001-07-22 17:18:56 +03:00
}
2015-07-01 23:21:53 +02:00
if ( ret < 0 )
return ret ;
2013-02-06 02:44:00 +03:00
if ( avio_tell ( pb ) ! = gpos + gsize )
av_log ( s , AV_LOG_DEBUG ,
" gpos mismatch our pos=% " PRIu64 " , end=% " PRId64 " \n " ,
avio_tell ( pb ) - gpos , gsize ) ;
2011-02-28 15:57:54 +02:00
avio_seek ( pb , gpos + gsize , SEEK_SET ) ;
2001-07-22 17:18:56 +03:00
}
2010-12-26 03:26:29 +02:00
ff_get_guid ( pb , & g ) ;
2011-02-21 17:43:01 +02:00
avio_rl64 ( pb ) ;
avio_r8 ( pb ) ;
avio_r8 ( pb ) ;
2014-08-07 23:12:41 +03:00
if ( avio_feof ( pb ) )
2012-06-14 03:30:55 +03:00
return AVERROR_EOF ;
2013-02-06 02:44:00 +03:00
asf - > data_offset = avio_tell ( pb ) ;
2001-07-22 17:18:56 +03:00
asf - > packet_size_left = 0 ;
2013-02-06 02:44:00 +03:00
for ( i = 0 ; i < 128 ; i + + ) {
int stream_num = asf - > asfid2avid [ i ] ;
if ( stream_num > = 0 ) {
2008-08-24 02:43:20 +03:00
AVStream * st = s - > streams [ stream_num ] ;
if ( ! st - > codec - > bit_rate )
2011-02-09 22:55:54 +02:00
st - > codec - > bit_rate = asf - > stream_bitrates [ i ] ;
2013-02-06 02:44:00 +03:00
if ( asf - > dar [ i ] . num > 0 & & asf - > dar [ i ] . den > 0 ) {
2012-11-03 07:39:19 +03:00
av_reduce ( & st - > sample_aspect_ratio . num ,
& st - > sample_aspect_ratio . den ,
2011-02-09 22:55:53 +02:00
asf - > dar [ i ] . num , asf - > dar [ i ] . den , INT_MAX ) ;
2013-02-06 02:44:00 +03:00
} else if ( ( asf - > dar [ 0 ] . num > 0 ) & & ( asf - > dar [ 0 ] . den > 0 ) & &
// Use ASF container value if the stream doesn't set AR.
( st - > codec - > codec_type = = AVMEDIA_TYPE_VIDEO ) )
2012-11-03 07:39:19 +03:00
av_reduce ( & st - > sample_aspect_ratio . num ,
& st - > sample_aspect_ratio . den ,
2011-02-09 22:55:53 +02:00
asf - > dar [ 0 ] . num , asf - > dar [ 0 ] . den , INT_MAX ) ;
2010-09-23 06:40:06 +03:00
2015-03-16 10:57:35 +02:00
av_log ( s , AV_LOG_TRACE , " i=%d, st->codec->codec_type:%d, asf->dar %d:%d sar=%d:%d \n " ,
2012-08-21 13:40:41 +03:00
i , st - > codec - > codec_type , asf - > dar [ i ] . num , asf - > dar [ i ] . den ,
st - > sample_aspect_ratio . num , st - > sample_aspect_ratio . den ) ;
2009-05-12 15:35:46 +03:00
// copy and convert language codes to the frontend
if ( asf - > streams [ i ] . stream_language_index < 128 ) {
const char * rfc1766 = asf - > stream_languages [ asf - > streams [ i ] . stream_language_index ] ;
if ( rfc1766 & & strlen ( rfc1766 ) > 1 ) {
const char primary_tag [ 3 ] = { rfc1766 [ 0 ] , rfc1766 [ 1 ] , ' \0 ' } ; // ignore country code if any
2013-02-06 02:44:00 +03:00
const char * iso6392 = av_convert_lang_to ( primary_tag ,
AV_LANG_ISO639_2_BIBL ) ;
2009-05-12 15:35:46 +03:00
if ( iso6392 )
2011-05-22 13:46:29 +03:00
av_dict_set ( & st - > metadata , " language " , iso6392 , 0 ) ;
2009-05-12 15:35:46 +03:00
}
}
2007-02-01 01:32:01 +02:00
}
}
2010-10-16 16:20:41 +03:00
ff_metadata_conv ( & s - > metadata , NULL , ff_asf_metadata_conv ) ;
2010-10-15 22:04:25 +03:00
2001-07-22 17:18:56 +03:00
return 0 ;
}
2013-02-06 02:44:00 +03:00
# define DO_2BITS(bits, var, defval) \
switch ( bits & 3 ) { \
case 3 : \
var = avio_rl32 ( pb ) ; \
rsize + = 4 ; \
break ; \
case 2 : \
var = avio_rl16 ( pb ) ; \
rsize + = 2 ; \
break ; \
case 1 : \
var = avio_r8 ( pb ) ; \
rsize + + ; \
break ; \
default : \
var = defval ; \
break ; \
2002-08-08 19:43:47 +03:00
}
2009-06-09 23:58:32 +03:00
/**
* Load a single ASF packet into the demuxer .
* @ param s demux context
* @ param pb context to read data from
2010-03-30 18:50:57 +03:00
* @ return 0 on success , < 0 on error
2009-06-09 23:58:32 +03:00
*/
2013-04-20 22:47:15 +03:00
static int asf_get_packet ( AVFormatContext * s , AVIOContext * pb )
2001-07-22 17:18:56 +03:00
{
ASFContext * asf = s - > priv_data ;
2002-08-08 19:43:47 +03:00
uint32_t packet_length , padsize ;
2007-03-30 19:20:40 +03:00
int rsize = 8 ;
2007-01-23 23:25:06 +02:00
int c , d , e , off ;
2015-07-22 13:49:31 +02:00
if ( asf - > uses_std_ecc > 0 ) {
2015-07-22 01:09:14 +02:00
// if we do not know packet size, allow skipping up to 32 kB
off = 32768 ;
if ( asf - > no_resync_search )
off = 3 ;
2015-07-22 13:49:31 +02:00
// else if (s->packet_size > 0 && !asf->uses_std_ecc)
// off = (avio_tell(pb) - s->internal->data_offset) % s->packet_size + 3;
2015-07-22 01:09:14 +02:00
c = d = e = - 1 ;
while ( off - - > 0 ) {
c = d ;
d = e ;
e = avio_r8 ( pb ) ;
if ( c = = 0x82 & & ! d & & ! e )
break ;
}
2005-12-17 20:14:38 +02:00
2015-07-22 01:09:14 +02:00
if ( c ! = 0x82 ) {
/* This code allows handling of -EAGAIN at packet boundaries (i.e.
* if the packet sync code above triggers - EAGAIN ) . This does not
* imply complete - EAGAIN handling support at random positions in
* the stream . */
if ( pb - > error = = AVERROR ( EAGAIN ) )
return AVERROR ( EAGAIN ) ;
2014-08-07 23:12:41 +03:00
if ( ! avio_feof ( pb ) )
2015-07-22 01:09:14 +02:00
av_log ( s , AV_LOG_ERROR ,
" ff asf bad header %x at:% " PRId64 " \n " , c , avio_tell ( pb ) ) ;
2005-12-22 03:10:11 +02:00
}
2015-07-22 01:09:14 +02:00
if ( ( c & 0x8f ) = = 0x82 ) {
if ( d | | e ) {
if ( ! avio_feof ( pb ) )
av_log ( s , AV_LOG_ERROR , " ff asf bad non zero \n " ) ;
return AVERROR_INVALIDDATA ;
}
c = avio_r8 ( pb ) ;
d = avio_r8 ( pb ) ;
rsize + = 3 ;
} else if ( ! avio_feof ( pb ) ) {
avio_seek ( pb , - 1 , SEEK_CUR ) ; // FIXME
}
} else {
c = avio_r8 ( pb ) ;
2015-07-22 13:49:31 +02:00
if ( c & 0x80 ) {
rsize + + ;
if ( ! ( c & 0x60 ) ) {
d = avio_r8 ( pb ) ;
e = avio_r8 ( pb ) ;
avio_seek ( pb , ( c & 0xF ) - 2 , SEEK_CUR ) ;
rsize + = c & 0xF ;
}
if ( c ! = 0x82 )
avpriv_request_sample ( s , " Invalid ECC byte \n " ) ;
if ( ! asf - > uses_std_ecc )
asf - > uses_std_ecc = ( c = = 0x82 & & ! d & & ! e ) ? 1 : - 1 ;
c = avio_r8 ( pb ) ;
} else
asf - > uses_std_ecc = - 1 ;
2015-07-22 01:09:14 +02:00
d = avio_r8 ( pb ) ;
2001-07-22 17:18:56 +03:00
}
2002-08-08 19:43:47 +03:00
2007-03-30 19:20:40 +03:00
asf - > packet_flags = c ;
asf - > packet_property = d ;
2002-08-08 19:43:47 +03:00
2009-06-25 02:04:05 +03:00
DO_2BITS ( asf - > packet_flags > > 5 , packet_length , s - > packet_size ) ;
2002-08-08 19:43:47 +03:00
DO_2BITS ( asf - > packet_flags > > 1 , padsize , 0 ) ; // sequence ignored
DO_2BITS ( asf - > packet_flags > > 3 , padsize , 0 ) ; // padding length
2013-02-06 02:44:00 +03:00
// the following checks prevent overflows and infinite loops
if ( ! packet_length | | packet_length > = ( 1U < < 29 ) ) {
av_log ( s , AV_LOG_ERROR ,
2014-03-10 17:35:59 +03:00
" invalid packet_length % " PRIu32 " at:% " PRId64 " \n " ,
2013-02-06 02:44:00 +03:00
packet_length , avio_tell ( pb ) ) ;
2012-06-14 03:30:55 +03:00
return AVERROR_INVALIDDATA ;
2006-09-28 00:19:47 +03:00
}
2013-02-06 02:44:00 +03:00
if ( padsize > = packet_length ) {
av_log ( s , AV_LOG_ERROR ,
2014-03-10 17:35:59 +03:00
" invalid padsize % " PRIu32 " at:% " PRId64 " \n " , padsize , avio_tell ( pb ) ) ;
2012-06-14 03:30:55 +03:00
return AVERROR_INVALIDDATA ;
2006-09-28 00:19:47 +03:00
}
2011-02-21 17:43:01 +02:00
asf - > packet_timestamp = avio_rl32 ( pb ) ;
avio_rl16 ( pb ) ; /* duration */
2002-08-09 01:04:01 +03:00
// rsize has at least 11 bytes which have to be present
2002-08-08 19:43:47 +03:00
if ( asf - > packet_flags & 0x01 ) {
2013-02-06 02:44:00 +03:00
asf - > packet_segsizetype = avio_r8 ( pb ) ;
rsize + + ;
2002-08-08 19:43:47 +03:00
asf - > packet_segments = asf - > packet_segsizetype & 0x3f ;
} else {
2013-02-06 02:44:00 +03:00
asf - > packet_segments = 1 ;
2002-08-08 19:43:47 +03:00
asf - > packet_segsizetype = 0x80 ;
}
2012-02-17 23:21:18 +03:00
if ( rsize > packet_length - padsize ) {
asf - > packet_size_left = 0 ;
av_log ( s , AV_LOG_ERROR ,
2014-03-10 17:35:59 +03:00
" invalid packet header length %d for pktlen % " PRIu32 " -% " PRIu32 " at % " PRId64 " \n " ,
2012-02-17 23:21:18 +03:00
rsize , packet_length , padsize , avio_tell ( pb ) ) ;
2012-06-14 03:30:55 +03:00
return AVERROR_INVALIDDATA ;
2012-02-17 23:21:18 +03:00
}
2002-08-08 19:43:47 +03:00
asf - > packet_size_left = packet_length - padsize - rsize ;
2002-11-05 19:16:41 +02:00
if ( packet_length < asf - > hdr . min_pktsize )
padsize + = asf - > hdr . min_pktsize - packet_length ;
2002-08-08 19:43:47 +03:00
asf - > packet_padsize = padsize ;
2015-03-16 10:57:35 +02:00
av_log ( s , AV_LOG_TRACE , " packet: size=%d padsize=%d left=%d \n " ,
2013-02-06 02:44:00 +03:00
s - > packet_size , asf - > packet_padsize , asf - > packet_size_left ) ;
2001-07-22 17:18:56 +03:00
return 0 ;
}
2007-01-22 14:55:23 +02:00
/**
*
* @ return < 0 if error
*/
2013-02-06 02:44:00 +03:00
static int asf_read_frame_header ( AVFormatContext * s , AVIOContext * pb )
{
2007-01-22 14:55:23 +02:00
ASFContext * asf = s - > priv_data ;
2012-11-03 07:37:18 +03:00
ASFStream * asfst ;
2013-02-06 02:44:00 +03:00
int rsize = 1 ;
int num = avio_r8 ( pb ) ;
2012-11-03 07:37:18 +03:00
int i ;
2011-11-15 15:06:36 +03:00
int64_t ts0 , ts1 av_unused ;
2007-01-22 14:55:23 +02:00
asf - > packet_segments - - ;
asf - > packet_key_frame = num > > 7 ;
2013-02-06 02:44:00 +03:00
asf - > stream_index = asf - > asfid2avid [ num & 0x7f ] ;
2013-02-06 16:10:26 +03:00
asfst = & asf - > streams [ num & 0x7f ] ;
2007-01-22 14:55:23 +02:00
// sequence should be ignored!
DO_2BITS ( asf - > packet_property > > 4 , asf - > packet_seq , 0 ) ;
DO_2BITS ( asf - > packet_property > > 2 , asf - > packet_frag_offset , 0 ) ;
DO_2BITS ( asf - > packet_property , asf - > packet_replic_size , 0 ) ;
2015-07-22 00:33:30 +02:00
av_log ( asf , AV_LOG_TRACE , " key:%d stream:%d seq:%d offset:%d replic_size:%d num:%X packet_property %X \n " ,
2012-08-21 13:40:41 +03:00
asf - > packet_key_frame , asf - > stream_index , asf - > packet_seq ,
2015-07-22 00:33:30 +02:00
asf - > packet_frag_offset , asf - > packet_replic_size , num , asf - > packet_property ) ;
2013-01-31 05:36:59 +03:00
if ( rsize + ( int64_t ) asf - > packet_replic_size > asf - > packet_size_left ) {
2011-11-01 00:42:45 +03:00
av_log ( s , AV_LOG_ERROR , " packet_replic_size %d is invalid \n " , asf - > packet_replic_size ) ;
2012-06-14 03:30:55 +03:00
return AVERROR_INVALIDDATA ;
2011-11-01 00:42:45 +03:00
}
2007-01-22 14:55:23 +02:00
if ( asf - > packet_replic_size > = 8 ) {
2012-11-03 07:37:18 +03:00
int64_t end = avio_tell ( pb ) + asf - > packet_replic_size ;
2012-11-03 07:38:12 +03:00
AVRational aspect ;
2013-07-30 04:12:28 +03:00
asfst - > packet_obj_size = avio_rl32 ( pb ) ;
2015-06-29 04:57:51 +02:00
if ( asfst - > packet_obj_size > = ( 1 < < 24 ) | | asfst - > packet_obj_size < 0 ) {
2015-06-29 04:55:06 +02:00
av_log ( s , AV_LOG_ERROR , " packet_obj_size %d invalid \n " , asfst - > packet_obj_size ) ;
2013-07-30 04:12:28 +03:00
asfst - > packet_obj_size = 0 ;
2012-06-14 03:30:55 +03:00
return AVERROR_INVALIDDATA ;
2007-01-22 14:55:23 +02:00
}
2011-02-21 17:43:01 +02:00
asf - > packet_frag_timestamp = avio_rl32 ( pb ) ; // timestamp
2012-11-03 07:37:18 +03:00
2013-02-06 16:10:26 +03:00
for ( i = 0 ; i < asfst - > payload_ext_ct ; i + + ) {
2012-11-03 07:37:18 +03:00
ASFPayload * p = & asfst - > payload [ i ] ;
int size = p - > size ;
int64_t payend ;
2013-02-06 16:10:26 +03:00
if ( size = = 0xFFFF )
2012-11-03 07:37:18 +03:00
size = avio_rl16 ( pb ) ;
payend = avio_tell ( pb ) + size ;
if ( payend > end ) {
av_log ( s , AV_LOG_ERROR , " too long payload \n " ) ;
break ;
}
2013-02-06 16:10:26 +03:00
switch ( p - > type ) {
2012-11-03 07:37:18 +03:00
case 0x50 :
// duration = avio_rl16(pb);
break ;
2012-11-03 07:38:12 +03:00
case 0x54 :
aspect . num = avio_r8 ( pb ) ;
aspect . den = avio_r8 ( pb ) ;
2012-11-03 07:58:12 +03:00
if ( aspect . num > 0 & & aspect . den > 0 & & asf - > stream_index > = 0 ) {
2012-11-03 07:38:12 +03:00
s - > streams [ asf - > stream_index ] - > sample_aspect_ratio = aspect ;
}
break ;
2012-11-03 07:37:18 +03:00
case 0x2A :
avio_skip ( pb , 8 ) ;
2013-02-06 16:10:26 +03:00
ts0 = avio_rl64 ( pb ) ;
ts1 = avio_rl64 ( pb ) ;
if ( ts0 ! = - 1 ) asf - > packet_frag_timestamp = ts0 / 10000 ;
else asf - > packet_frag_timestamp = AV_NOPTS_VALUE ;
2014-07-13 04:23:54 +03:00
asf - > ts_is_pts = 1 ;
2012-11-03 07:37:18 +03:00
break ;
case 0x5B :
case 0xB7 :
case 0xCC :
case 0xC0 :
case 0xA0 :
//unknown
break ;
}
avio_seek ( pb , payend , SEEK_SET ) ;
}
avio_seek ( pb , end , SEEK_SET ) ;
2007-01-22 14:55:23 +02:00
rsize + = asf - > packet_replic_size ; // FIXME - check validity
2013-02-06 02:44:00 +03:00
} else if ( asf - > packet_replic_size = = 1 ) {
2007-12-23 20:08:00 +02:00
// multipacket - frag_offset is beginning timestamp
2013-02-06 02:44:00 +03:00
asf - > packet_time_start = asf - > packet_frag_offset ;
asf - > packet_frag_offset = 0 ;
2007-01-22 14:55:23 +02:00
asf - > packet_frag_timestamp = asf - > packet_timestamp ;
2011-02-21 17:43:01 +02:00
asf - > packet_time_delta = avio_r8 ( pb ) ;
2007-01-22 14:55:23 +02:00
rsize + + ;
2013-02-06 02:44:00 +03:00
} else if ( asf - > packet_replic_size ! = 0 ) {
av_log ( s , AV_LOG_ERROR , " unexpected packet_replic_size of %d \n " ,
asf - > packet_replic_size ) ;
2012-06-14 03:30:55 +03:00
return AVERROR_INVALIDDATA ;
2007-01-22 14:55:23 +02:00
}
if ( asf - > packet_flags & 0x01 ) {
DO_2BITS ( asf - > packet_segsizetype > > 6 , asf - > packet_frag_size , 0 ) ; // 0 is illegal
2011-05-12 17:20:27 +03:00
if ( rsize > asf - > packet_size_left ) {
av_log ( s , AV_LOG_ERROR , " packet_replic_size is invalid \n " ) ;
2012-06-14 03:30:55 +03:00
return AVERROR_INVALIDDATA ;
2013-02-06 02:44:00 +03:00
} else if ( asf - > packet_frag_size > asf - > packet_size_left - rsize ) {
2011-04-28 01:42:16 +03:00
if ( asf - > packet_frag_size > asf - > packet_size_left - rsize + asf - > packet_padsize ) {
2015-07-22 00:33:30 +02:00
av_log ( s , AV_LOG_ERROR , " packet_frag_size is invalid (%d>%d-%d+%d) \n " ,
asf - > packet_frag_size , asf - > packet_size_left , rsize , asf - > packet_padsize ) ;
2012-06-14 03:30:55 +03:00
return AVERROR_INVALIDDATA ;
2011-04-28 01:42:16 +03:00
} else {
int diff = asf - > packet_frag_size - ( asf - > packet_size_left - rsize ) ;
asf - > packet_size_left + = diff ;
asf - > packet_padsize - = diff ;
}
2007-01-22 23:17:54 +02:00
}
2007-01-22 14:55:23 +02:00
} else {
asf - > packet_frag_size = asf - > packet_size_left - rsize ;
}
if ( asf - > packet_replic_size = = 1 ) {
asf - > packet_multi_size = asf - > packet_frag_size ;
if ( asf - > packet_multi_size > asf - > packet_size_left )
2012-06-14 03:30:55 +03:00
return AVERROR_INVALIDDATA ;
2007-01-22 14:55:23 +02:00
}
asf - > packet_size_left - = rsize ;
return 0 ;
}
2009-06-09 23:58:32 +03:00
/**
* Parse data from individual ASF packets ( which were previously loaded
* with asf_get_packet ( ) ) .
* @ param s demux context
* @ param pb context to read data from
* @ param pkt pointer to store packet data into
2010-03-30 18:50:57 +03:00
* @ return 0 if data was stored in pkt , < 0 on error or 1 if more ASF
2009-06-09 23:58:32 +03:00
* packets need to be loaded ( through asf_get_packet ( ) )
*/
2013-04-20 22:47:15 +03:00
static int asf_parse_packet ( AVFormatContext * s , AVIOContext * pb , AVPacket * pkt )
2001-07-22 17:18:56 +03:00
{
2013-02-06 02:44:00 +03:00
ASFContext * asf = s - > priv_data ;
2002-08-08 19:43:47 +03:00
ASFStream * asf_st = 0 ;
for ( ; ; ) {
2010-12-11 23:41:47 +02:00
int ret ;
2014-08-07 23:12:41 +03:00
if ( avio_feof ( pb ) )
2009-04-16 03:41:31 +03:00
return AVERROR_EOF ;
2014-07-14 02:16:08 +03:00
if ( asf - > packet_size_left < FRAME_HEADER_SIZE | |
asf - > packet_segments < 1 & & asf - > packet_time_start = = 0 ) {
2005-12-22 03:10:11 +02:00
int ret = asf - > packet_size_left + asf - > packet_padsize ;
2012-10-20 18:15:57 +03:00
2015-06-28 23:09:53 +02:00
if ( asf - > packet_size_left & & asf - > packet_size_left < FRAME_HEADER_SIZE )
av_log ( s , AV_LOG_WARNING , " Skip due to FRAME_HEADER_SIZE \n " ) ;
2012-10-20 18:15:57 +03:00
assert ( ret > = 0 ) ;
2005-12-22 03:10:11 +02:00
/* fail safe */
2011-03-15 10:14:38 +02:00
avio_skip ( pb , ret ) ;
2007-01-22 14:52:49 +02:00
2013-02-06 02:44:00 +03:00
asf - > packet_pos = avio_tell ( pb ) ;
2006-07-29 19:07:19 +03:00
if ( asf - > data_object_size ! = ( uint64_t ) - 1 & &
( asf - > packet_pos - asf - > data_object_offset > = asf - > data_object_size ) )
2013-02-06 02:44:00 +03:00
return AVERROR_EOF ; /* Do not exceed the size of the data object */
2008-12-13 19:18:11 +02:00
return 1 ;
2005-12-22 03:10:11 +02:00
}
if ( asf - > packet_time_start = = 0 ) {
2012-10-20 18:15:57 +03:00
if ( asf_read_frame_header ( s , pb ) < 0 ) {
2013-11-24 22:32:31 +03:00
asf - > packet_time_start = asf - > packet_segments = 0 ;
2007-01-22 01:03:09 +02:00
continue ;
2004-01-14 23:27:59 +02:00
}
2012-10-20 18:15:57 +03:00
if ( asf - > stream_index < 0 | |
s - > streams [ asf - > stream_index ] - > discard > = AVDISCARD_ALL | |
( ! asf - > packet_key_frame & &
2013-07-30 16:09:13 +03:00
( s - > streams [ asf - > stream_index ] - > discard > = AVDISCARD_NONKEY | | asf - > streams [ s - > streams [ asf - > stream_index ] - > id ] . skip_to_key ) ) ) {
2002-08-09 16:06:27 +03:00
asf - > packet_time_start = 0 ;
2005-12-22 03:10:11 +02:00
/* unhandled packet (should not happen) */
2011-03-15 10:14:38 +02:00
avio_skip ( pb , asf - > packet_frag_size ) ;
2005-12-22 03:10:11 +02:00
asf - > packet_size_left - = asf - > packet_frag_size ;
2012-10-20 18:15:57 +03:00
if ( asf - > stream_index < 0 )
2013-02-06 02:44:00 +03:00
av_log ( s , AV_LOG_ERROR , " ff asf skip %d (unknown stream) \n " ,
asf - > packet_frag_size ) ;
2002-08-09 16:06:27 +03:00
continue ;
2005-12-22 03:10:11 +02:00
}
2013-07-30 03:52:17 +03:00
asf - > asf_st = & asf - > streams [ s - > streams [ asf - > stream_index ] - > id ] ;
2015-07-01 16:08:05 +02:00
if ( ! asf - > packet_frag_offset )
asf - > asf_st - > skip_to_key = 0 ;
2005-12-22 03:10:11 +02:00
}
asf_st = asf - > asf_st ;
2013-02-08 15:54:36 +03:00
av_assert0 ( asf_st ) ;
2005-12-22 03:10:11 +02:00
2014-02-11 15:01:38 +03:00
if ( ! asf_st - > frag_offset & & asf - > packet_frag_offset ) {
2015-03-16 10:57:35 +02:00
av_log ( s , AV_LOG_TRACE , " skipping asf data pkt with fragment offset for "
2014-02-11 15:01:38 +03:00
" stream:%d, expected:%d but got %d from pkt) \n " ,
asf - > stream_index , asf_st - > frag_offset ,
asf - > packet_frag_offset ) ;
avio_skip ( pb , asf - > packet_frag_size ) ;
asf - > packet_size_left - = asf - > packet_frag_size ;
continue ;
}
2005-12-22 03:10:11 +02:00
if ( asf - > packet_replic_size = = 1 ) {
2007-12-23 20:08:00 +02:00
// frag_offset is here used as the beginning timestamp
2005-12-22 03:10:11 +02:00
asf - > packet_frag_timestamp = asf - > packet_time_start ;
2012-10-20 18:15:57 +03:00
asf - > packet_time_start + = asf - > packet_time_delta ;
2013-07-30 04:12:28 +03:00
asf_st - > packet_obj_size = asf - > packet_frag_size = avio_r8 ( pb ) ;
2005-12-22 03:10:11 +02:00
asf - > packet_size_left - - ;
2002-08-09 01:04:01 +03:00
asf - > packet_multi_size - - ;
2013-07-30 04:12:28 +03:00
if ( asf - > packet_multi_size < asf_st - > packet_obj_size ) {
2005-12-22 03:10:11 +02:00
asf - > packet_time_start = 0 ;
2011-03-15 10:14:38 +02:00
avio_skip ( pb , asf - > packet_multi_size ) ;
2005-12-22 03:10:11 +02:00
asf - > packet_size_left - = asf - > packet_multi_size ;
2002-08-09 01:04:01 +03:00
continue ;
2005-12-22 03:10:11 +02:00
}
2013-07-30 04:12:28 +03:00
asf - > packet_multi_size - = asf_st - > packet_obj_size ;
2005-12-22 03:10:11 +02:00
}
2007-07-27 03:29:33 +03:00
2013-07-30 04:12:28 +03:00
if ( asf_st - > pkt . size ! = asf_st - > packet_obj_size | |
2013-02-06 02:44:00 +03:00
// FIXME is this condition sufficient?
2012-10-20 18:15:57 +03:00
asf_st - > frag_offset + asf - > packet_frag_size > asf_st - > pkt . size ) {
2014-08-20 19:27:25 +03:00
int ret ;
2012-10-20 18:15:57 +03:00
if ( asf_st - > pkt . data ) {
2013-02-06 02:44:00 +03:00
av_log ( s , AV_LOG_INFO ,
" freeing incomplete packet size %d, new %d \n " ,
2013-07-30 04:12:28 +03:00
asf_st - > pkt . size , asf_st - > packet_obj_size ) ;
2007-01-22 19:01:01 +02:00
asf_st - > frag_offset = 0 ;
av_free_packet ( & asf_st - > pkt ) ;
}
2005-12-22 03:10:11 +02:00
/* new packet */
2014-08-20 19:27:25 +03:00
if ( ( ret = av_new_packet ( & asf_st - > pkt , asf_st - > packet_obj_size ) ) < 0 )
return ret ;
2012-10-20 18:15:57 +03:00
asf_st - > seq = asf - > packet_seq ;
2014-07-13 04:23:54 +03:00
if ( asf - > ts_is_pts ) {
asf_st - > pkt . pts = asf - > packet_frag_timestamp - asf - > hdr . preroll ;
} else
asf_st - > pkt . dts = asf - > packet_frag_timestamp - asf - > hdr . preroll ;
2005-12-22 03:10:11 +02:00
asf_st - > pkt . stream_index = asf - > stream_index ;
2012-10-20 18:15:57 +03:00
asf_st - > pkt . pos = asf_st - > packet_pos = asf - > packet_pos ;
2013-12-14 00:31:49 +03:00
asf_st - > pkt_clean = 0 ;
2012-10-20 18:15:57 +03:00
2011-04-09 16:49:51 +03:00
if ( asf_st - > pkt . data & & asf_st - > palette_changed ) {
uint8_t * pal ;
2011-12-20 07:16:51 +03:00
pal = av_packet_new_side_data ( & asf_st - > pkt , AV_PKT_DATA_PALETTE ,
2011-04-09 16:49:51 +03:00
AVPALETTE_SIZE ) ;
if ( ! pal ) {
av_log ( s , AV_LOG_ERROR , " Cannot append palette to packet \n " ) ;
} else {
memcpy ( pal , asf_st - > palette , AVPALETTE_SIZE ) ;
asf_st - > palette_changed = 0 ;
}
}
2015-03-16 10:57:35 +02:00
av_log ( asf , AV_LOG_TRACE , " new packet: stream:%d key:%d packet_key:%d audio:%d size:%d \n " ,
2012-08-21 13:40:41 +03:00
asf - > stream_index , asf - > packet_key_frame ,
asf_st - > pkt . flags & AV_PKT_FLAG_KEY ,
s - > streams [ asf - > stream_index ] - > codec - > codec_type = = AVMEDIA_TYPE_AUDIO ,
2013-07-30 04:12:28 +03:00
asf_st - > packet_obj_size ) ;
2010-03-31 02:30:55 +03:00
if ( s - > streams [ asf - > stream_index ] - > codec - > codec_type = = AVMEDIA_TYPE_AUDIO )
2005-12-22 03:10:11 +02:00
asf - > packet_key_frame = 1 ;
if ( asf - > packet_key_frame )
2010-03-31 15:29:58 +03:00
asf_st - > pkt . flags | = AV_PKT_FLAG_KEY ;
2005-12-22 03:10:11 +02:00
}
/* read data */
2015-03-16 10:57:35 +02:00
av_log ( asf , AV_LOG_TRACE , " READ PACKET s:%d os:%d o:%d,%d l:%d DATA:%p \n " ,
2012-08-21 13:40:41 +03:00
s - > packet_size , asf_st - > pkt . size , asf - > packet_frag_offset ,
asf_st - > frag_offset , asf - > packet_frag_size , asf_st - > pkt . data ) ;
2005-12-22 03:10:11 +02:00
asf - > packet_size_left - = asf - > packet_frag_size ;
if ( asf - > packet_size_left < 0 )
2002-08-09 01:04:01 +03:00
continue ;
2007-01-22 18:37:45 +02:00
2012-10-20 18:15:57 +03:00
if ( asf - > packet_frag_offset > = asf_st - > pkt . size | |
asf - > packet_frag_size > asf_st - > pkt . size - asf - > packet_frag_offset ) {
2013-02-06 02:44:00 +03:00
av_log ( s , AV_LOG_ERROR ,
" packet fragment position invalid %u,%u not in %u \n " ,
asf - > packet_frag_offset , asf - > packet_frag_size ,
asf_st - > pkt . size ) ;
2007-01-22 18:37:45 +02:00
continue ;
}
2013-12-14 00:31:49 +03:00
if ( asf - > packet_frag_offset ! = asf_st - > frag_offset & & ! asf_st - > pkt_clean ) {
memset ( asf_st - > pkt . data + asf_st - > frag_offset , 0 , asf_st - > pkt . size - asf_st - > frag_offset ) ;
asf_st - > pkt_clean = 1 ;
}
2011-02-21 17:43:01 +02:00
ret = avio_read ( pb , asf_st - > pkt . data + asf - > packet_frag_offset ,
2012-10-20 18:15:57 +03:00
asf - > packet_frag_size ) ;
2010-12-18 15:18:52 +02:00
if ( ret ! = asf - > packet_frag_size ) {
if ( ret < 0 | | asf - > packet_frag_offset + ret = = 0 )
return ret < 0 ? ret : AVERROR_EOF ;
2012-10-20 18:15:57 +03:00
2010-12-18 15:18:52 +02:00
if ( asf_st - > ds_span > 1 ) {
// scrambling, we can either drop it completely or fill the remainder
// TODO: should we fill the whole packet instead of just the current
// fragment?
memset ( asf_st - > pkt . data + asf - > packet_frag_offset + ret , 0 ,
asf - > packet_frag_size - ret ) ;
ret = asf - > packet_frag_size ;
2012-10-20 18:15:57 +03:00
} else {
2010-12-18 15:18:52 +02:00
// no scrambling, so we can return partial packets
av_shrink_packet ( & asf_st - > pkt , asf - > packet_frag_offset + ret ) ;
2012-10-20 18:15:57 +03:00
}
2010-12-18 15:18:52 +02:00
}
2007-10-20 17:25:02 +03:00
if ( s - > key & & s - > keylen = = 20 )
ff_asfcrypt_dec ( s - > key , asf_st - > pkt . data + asf - > packet_frag_offset ,
2010-12-18 15:18:52 +02:00
ret ) ;
asf_st - > frag_offset + = ret ;
2005-12-22 03:10:11 +02:00
/* test if whole packet is read */
if ( asf_st - > frag_offset = = asf_st - > pkt . size ) {
2013-02-06 02:44:00 +03:00
// workaround for macroshit radio DVR-MS files
2012-10-20 18:15:57 +03:00
if ( s - > streams [ asf - > stream_index ] - > codec - > codec_id = = AV_CODEC_ID_MPEG2VIDEO & &
asf_st - > pkt . size > 100 ) {
2007-09-02 18:58:43 +03:00
int i ;
2013-02-06 02:44:00 +03:00
for ( i = 0 ; i < asf_st - > pkt . size & & ! asf_st - > pkt . data [ i ] ; i + + )
;
2012-10-20 18:15:57 +03:00
if ( i = = asf_st - > pkt . size ) {
2007-09-02 18:58:43 +03:00
av_log ( s , AV_LOG_DEBUG , " discarding ms fart \n " ) ;
asf_st - > frag_offset = 0 ;
av_free_packet ( & asf_st - > pkt ) ;
continue ;
}
}
2005-12-22 03:10:11 +02:00
/* return packet */
if ( asf_st - > ds_span > 1 ) {
2013-02-06 02:44:00 +03:00
if ( asf_st - > pkt . size ! = asf_st - > ds_packet_size * asf_st - > ds_span ) {
av_log ( s , AV_LOG_ERROR ,
" pkt.size != ds_packet_size * ds_span (%d %d %d) \n " ,
asf_st - > pkt . size , asf_st - > ds_packet_size ,
asf_st - > ds_span ) ;
2012-10-20 18:15:57 +03:00
} else {
/* packet descrambling */
2012-10-31 10:53:18 +03:00
AVBufferRef * buf = av_buffer_alloc ( asf_st - > pkt . size +
2015-07-27 22:53:16 +02:00
AV_INPUT_BUFFER_PADDING_SIZE ) ;
2012-10-31 10:53:18 +03:00
if ( buf ) {
uint8_t * newdata = buf - > data ;
2012-10-20 18:15:57 +03:00
int offset = 0 ;
2013-02-06 02:44:00 +03:00
memset ( newdata + asf_st - > pkt . size , 0 ,
2015-07-27 22:53:16 +02:00
AV_INPUT_BUFFER_PADDING_SIZE ) ;
2012-10-20 18:15:57 +03:00
while ( offset < asf_st - > pkt . size ) {
int off = offset / asf_st - > ds_chunk_size ;
int row = off / asf_st - > ds_span ;
int col = off % asf_st - > ds_span ;
int idx = row + col * asf_st - > ds_packet_size / asf_st - > ds_chunk_size ;
assert ( offset + asf_st - > ds_chunk_size < = asf_st - > pkt . size ) ;
2013-02-06 02:44:00 +03:00
assert ( idx + 1 < = asf_st - > pkt . size / asf_st - > ds_chunk_size ) ;
2012-10-20 18:15:57 +03:00
memcpy ( newdata + offset ,
asf_st - > pkt . data + idx * asf_st - > ds_chunk_size ,
asf_st - > ds_chunk_size ) ;
offset + = asf_st - > ds_chunk_size ;
}
2012-10-31 10:53:18 +03:00
av_buffer_unref ( & asf_st - > pkt . buf ) ;
asf_st - > pkt . buf = buf ;
asf_st - > pkt . data = buf - > data ;
2005-12-22 03:10:11 +02:00
}
}
}
2012-10-20 18:15:57 +03:00
asf_st - > frag_offset = 0 ;
* pkt = asf_st - > pkt ;
2012-10-31 10:53:18 +03:00
asf_st - > pkt . buf = 0 ;
2012-10-20 18:15:57 +03:00
asf_st - > pkt . size = 0 ;
asf_st - > pkt . data = 0 ;
2012-03-22 02:10:37 +03:00
asf_st - > pkt . side_data_elems = 0 ;
2012-10-20 18:15:57 +03:00
asf_st - > pkt . side_data = NULL ;
2005-12-22 03:10:11 +02:00
break ; // packet completed
}
2001-07-22 17:18:56 +03:00
}
return 0 ;
}
2008-12-13 19:18:11 +02:00
static int asf_read_packet ( AVFormatContext * s , AVPacket * pkt )
{
ASFContext * asf = s - > priv_data ;
for ( ; ; ) {
int ret ;
/* parse cached packets, if any */
2013-04-20 22:47:15 +03:00
if ( ( ret = asf_parse_packet ( s , s - > pb , pkt ) ) < = 0 )
2008-12-13 19:18:11 +02:00
return ret ;
2013-04-20 22:47:15 +03:00
if ( ( ret = asf_get_packet ( s , s - > pb ) ) < 0 )
2013-02-06 02:44:00 +03:00
assert ( asf - > packet_size_left < FRAME_HEADER_SIZE | |
asf - > packet_segments < 1 ) ;
2008-12-13 19:18:11 +02:00
asf - > packet_time_start = 0 ;
}
}
2004-01-14 15:32:49 +02:00
// Added to support seeking after packets have been read
// If information is not reset, read_packet fails due to
// leftover information from previous reads
static void asf_reset_header ( AVFormatContext * s )
{
ASFContext * asf = s - > priv_data ;
2004-01-14 20:18:47 +02:00
ASFStream * asf_st ;
int i ;
2004-01-14 15:32:49 +02:00
2013-02-06 02:44:00 +03:00
asf - > packet_size_left = 0 ;
asf - > packet_flags = 0 ;
asf - > packet_property = 0 ;
asf - > packet_timestamp = 0 ;
asf - > packet_segsizetype = 0 ;
asf - > packet_segments = 0 ;
asf - > packet_seq = 0 ;
asf - > packet_replic_size = 0 ;
asf - > packet_key_frame = 0 ;
asf - > packet_padsize = 0 ;
asf - > packet_frag_offset = 0 ;
asf - > packet_frag_size = 0 ;
2004-01-14 15:32:49 +02:00
asf - > packet_frag_timestamp = 0 ;
2013-02-06 02:44:00 +03:00
asf - > packet_multi_size = 0 ;
asf - > packet_time_delta = 0 ;
asf - > packet_time_start = 0 ;
2005-12-17 20:14:38 +02:00
2013-07-30 03:52:17 +03:00
for ( i = 0 ; i < 128 ; i + + ) {
asf_st = & asf - > streams [ i ] ;
2004-01-14 20:18:47 +02:00
av_free_packet ( & asf_st - > pkt ) ;
2013-07-30 04:12:28 +03:00
asf_st - > packet_obj_size = 0 ;
2013-02-06 02:44:00 +03:00
asf_st - > frag_offset = 0 ;
asf_st - > seq = 0 ;
2004-01-14 20:18:47 +02:00
}
2013-02-06 02:44:00 +03:00
asf - > asf_st = NULL ;
2004-01-14 15:32:49 +02:00
}
2013-07-30 16:09:13 +03:00
static void skip_to_key ( AVFormatContext * s )
{
ASFContext * asf = s - > priv_data ;
int i ;
for ( i = 0 ; i < 128 ; i + + ) {
int j = asf - > asfid2avid [ i ] ;
ASFStream * asf_st = & asf - > streams [ i ] ;
if ( j < 0 | | s - > streams [ j ] - > codec - > codec_type ! = AVMEDIA_TYPE_VIDEO )
continue ;
asf_st - > skip_to_key = 1 ;
}
}
2007-10-14 20:44:38 +03:00
static int asf_read_close ( AVFormatContext * s )
{
asf_reset_header ( s ) ;
2011-04-17 20:48:27 +03:00
2007-10-14 20:44:38 +03:00
return 0 ;
}
2013-02-06 02:44:00 +03:00
static int64_t asf_read_pts ( AVFormatContext * s , int stream_index ,
int64_t * ppos , int64_t pos_limit )
2004-01-14 15:32:49 +02:00
{
2013-07-30 03:52:17 +03:00
ASFContext * asf = s - > priv_data ;
2004-01-14 15:32:49 +02:00
AVPacket pkt1 , * pkt = & pkt1 ;
2004-01-14 20:18:47 +02:00
ASFStream * asf_st ;
2004-01-14 16:45:53 +02:00
int64_t pts ;
2013-02-06 02:44:00 +03:00
int64_t pos = * ppos ;
2004-01-17 20:06:52 +02:00
int i ;
2010-06-24 18:10:06 +03:00
int64_t start_pos [ ASF_MAX_STREAMS ] ;
2005-12-17 20:14:38 +02:00
2013-02-06 02:44:00 +03:00
for ( i = 0 ; i < s - > nb_streams ; i + + )
start_pos [ i ] = pos ;
2005-12-17 20:14:38 +02:00
2009-07-03 14:18:07 +03:00
if ( s - > packet_size > 0 )
2015-02-06 15:53:40 +02:00
pos = ( pos + s - > packet_size - 1 - s - > internal - > data_offset ) /
2013-02-06 02:44:00 +03:00
s - > packet_size * s - > packet_size +
2015-02-06 15:53:40 +02:00
s - > internal - > data_offset ;
2013-02-06 02:44:00 +03:00
* ppos = pos ;
2011-09-12 05:06:55 +03:00
if ( avio_seek ( s - > pb , pos , SEEK_SET ) < 0 )
return AV_NOPTS_VALUE ;
2005-12-17 20:14:38 +02:00
2013-08-12 23:52:23 +03:00
ff_read_frame_flush ( s ) ;
2004-01-14 20:18:47 +02:00
asf_reset_header ( s ) ;
2013-02-06 02:44:00 +03:00
for ( ; ; ) {
2013-02-06 16:10:26 +03:00
if ( av_read_frame ( s , pkt ) < 0 ) {
2007-01-22 23:31:40 +02:00
av_log ( s , AV_LOG_INFO , " asf_read_pts failed \n " ) ;
2005-12-22 03:10:11 +02:00
return AV_NOPTS_VALUE ;
2004-01-14 20:18:47 +02:00
}
2005-12-17 20:14:38 +02:00
2011-11-15 05:56:37 +03:00
pts = pkt - > dts ;
2004-01-14 16:45:53 +02:00
av_free_packet ( pkt ) ;
2013-02-06 02:44:00 +03:00
if ( pkt - > flags & AV_PKT_FLAG_KEY ) {
i = pkt - > stream_index ;
2004-01-17 20:06:52 +02:00
2013-07-30 03:52:17 +03:00
asf_st = & asf - > streams [ s - > streams [ i ] - > id ] ;
2004-01-17 20:06:52 +02:00
2009-06-25 02:04:05 +03:00
// assert((asf_st->packet_pos - s->data_offset) % s->packet_size == 0);
2013-02-06 02:44:00 +03:00
pos = asf_st - > packet_pos ;
2015-07-01 16:41:26 +02:00
av_assert1 ( pkt - > pos = = asf_st - > packet_pos ) ;
2004-01-14 20:40:29 +02:00
2013-02-06 02:44:00 +03:00
av_add_index_entry ( s - > streams [ i ] , pos , pts , pkt - > size ,
pos - start_pos [ i ] + 1 , AVINDEX_KEYFRAME ) ;
start_pos [ i ] = asf_st - > packet_pos + 1 ;
2005-12-17 20:14:38 +02:00
2013-02-06 02:44:00 +03:00
if ( pkt - > stream_index = = stream_index )
break ;
2004-01-17 20:06:52 +02:00
}
}
2013-02-06 02:44:00 +03:00
* ppos = pos ;
2004-01-14 16:45:53 +02:00
return pts ;
2004-01-14 15:32:49 +02:00
}
2014-02-11 17:13:31 +03:00
static int asf_build_simple_index ( AVFormatContext * s , int stream_index )
2006-08-23 20:07:01 +03:00
{
2009-03-18 16:03:40 +02:00
ff_asf_guid g ;
2013-02-06 02:44:00 +03:00
ASFContext * asf = s - > priv_data ;
int64_t current_pos = avio_tell ( s - > pb ) ;
2015-02-20 20:29:12 +02:00
int64_t ret ;
2006-08-23 20:07:01 +03:00
2014-02-12 19:10:26 +03:00
if ( ( ret = avio_seek ( s - > pb , asf - > data_object_offset + asf - > data_object_size , SEEK_SET ) ) < 0 ) {
return ret ;
2011-09-12 22:35:32 +03:00
}
2011-09-12 05:06:55 +03:00
2014-02-11 17:13:31 +03:00
if ( ( ret = ff_get_guid ( s - > pb , & g ) ) < 0 )
goto end ;
2010-12-03 09:49:07 +02:00
/* the data object can be followed by other top-level objects,
2013-02-06 02:44:00 +03:00
* skip them until the simple index object is reached */
2012-04-03 10:44:10 +03:00
while ( ff_guidcmp ( & g , & ff_asf_simple_index_header ) ) {
2013-02-06 02:44:00 +03:00
int64_t gsize = avio_rl64 ( s - > pb ) ;
2014-08-07 23:12:41 +03:00
if ( gsize < 24 | | avio_feof ( s - > pb ) ) {
2014-02-11 17:13:31 +03:00
goto end ;
2010-12-03 09:49:07 +02:00
}
2013-02-06 02:44:00 +03:00
avio_skip ( s - > pb , gsize - 24 ) ;
2014-02-11 17:13:31 +03:00
if ( ( ret = ff_get_guid ( s - > pb , & g ) ) < 0 )
goto end ;
2010-12-03 09:49:07 +02:00
}
{
2013-02-06 02:44:00 +03:00
int64_t itime , last_pos = - 1 ;
2009-04-17 15:37:39 +03:00
int pct , ict ;
2012-01-15 22:18:22 +03:00
int i ;
2013-02-06 02:44:00 +03:00
int64_t av_unused gsize = avio_rl64 ( s - > pb ) ;
2014-02-11 17:13:31 +03:00
if ( ( ret = ff_get_guid ( s - > pb , & g ) ) < 0 )
goto end ;
2013-02-06 02:44:00 +03:00
itime = avio_rl64 ( s - > pb ) ;
pct = avio_rl32 ( s - > pb ) ;
ict = avio_rl32 ( s - > pb ) ;
av_log ( s , AV_LOG_DEBUG ,
" itime:0x% " PRIx64 " , pct:%d, ict:%d \n " , itime , pct , ict ) ;
for ( i = 0 ; i < ict ; i + + ) {
int pktnum = avio_rl32 ( s - > pb ) ;
int pktct = avio_rl16 ( s - > pb ) ;
2015-02-06 15:53:40 +02:00
int64_t pos = s - > internal - > data_offset + s - > packet_size * ( int64_t ) pktnum ;
2013-02-06 02:44:00 +03:00
int64_t index_pts = FFMAX ( av_rescale ( itime , i , 10000 ) - asf - > hdr . preroll , 0 ) ;
if ( pos ! = last_pos ) {
av_log ( s , AV_LOG_DEBUG , " pktnum:%d, pktct:%d pts: % " PRId64 " \n " ,
pktnum , pktct , index_pts ) ;
av_add_index_entry ( s - > streams [ stream_index ] , pos , index_pts ,
s - > packet_size , 0 , AVINDEX_KEYFRAME ) ;
last_pos = pos ;
2009-12-16 23:24:04 +02:00
}
2006-08-23 20:07:01 +03:00
}
2013-02-06 16:10:26 +03:00
asf - > index_read = ict > 1 ;
2006-08-23 20:07:01 +03:00
}
2014-02-11 17:13:31 +03:00
end :
2014-08-07 23:12:41 +03:00
// if (avio_feof(s->pb)) {
2014-02-12 18:00:37 +03:00
// ret = 0;
// }
2011-02-28 15:57:54 +02:00
avio_seek ( s - > pb , current_pos , SEEK_SET ) ;
2014-02-11 17:13:31 +03:00
return ret ;
2006-08-23 20:07:01 +03:00
}
2013-02-06 02:44:00 +03:00
static int asf_read_seek ( AVFormatContext * s , int stream_index ,
int64_t pts , int flags )
2002-08-08 19:43:47 +03:00
{
2003-11-10 20:49:58 +02:00
ASFContext * asf = s - > priv_data ;
2013-02-06 02:44:00 +03:00
AVStream * st = s - > streams [ stream_index ] ;
2014-02-12 18:00:37 +03:00
int ret = 0 ;
2005-12-17 20:14:38 +02:00
2009-06-25 02:04:05 +03:00
if ( s - > packet_size < = 0 )
2004-01-14 15:32:49 +02:00
return - 1 ;
2007-12-17 11:28:46 +02:00
/* Try using the protocol's read_seek if available */
2013-02-06 02:44:00 +03:00
if ( s - > pb ) {
2015-02-20 20:29:12 +02:00
int64_t ret = avio_seek_time ( s - > pb , stream_index , pts , flags ) ;
2013-02-06 02:44:00 +03:00
if ( ret > = 0 )
2007-12-17 11:28:46 +02:00
asf_reset_header ( s ) ;
2007-12-20 02:25:18 +02:00
if ( ret ! = AVERROR ( ENOSYS ) )
2007-12-20 02:26:53 +02:00
return ret ;
2007-12-17 11:28:46 +02:00
}
2014-02-03 02:05:58 +03:00
/* explicitly handle the case of seeking to 0 */
if ( ! pts ) {
asf_reset_header ( s ) ;
2015-02-06 15:53:40 +02:00
avio_seek ( s - > pb , s - > internal - > data_offset , SEEK_SET ) ;
2014-02-03 02:05:58 +03:00
return 0 ;
}
2014-02-12 18:00:37 +03:00
if ( ! asf - > index_read ) {
2014-02-11 17:13:31 +03:00
ret = asf_build_simple_index ( s , stream_index ) ;
2014-02-12 18:00:37 +03:00
if ( ret < 0 )
asf - > index_read = - 1 ;
}
2006-08-23 20:07:01 +03:00
2014-02-12 18:00:37 +03:00
if ( asf - > index_read > 0 & & st - > index_entries ) {
2013-02-06 16:10:26 +03:00
int index = av_index_search_timestamp ( st , pts , flags ) ;
2013-02-06 02:44:00 +03:00
if ( index > = 0 ) {
2011-05-12 11:26:32 +03:00
/* find the position */
2012-01-15 22:18:22 +03:00
uint64_t pos = st - > index_entries [ index ] . pos ;
2006-08-23 20:07:01 +03:00
2011-05-12 11:26:32 +03:00
/* do the seek */
av_log ( s , AV_LOG_DEBUG , " SEEKTO: % " PRId64 " \n " , pos ) ;
2011-09-12 05:06:55 +03:00
if ( avio_seek ( s - > pb , pos , SEEK_SET ) < 0 )
return - 1 ;
2011-05-12 11:26:32 +03:00
asf_reset_header ( s ) ;
2013-07-30 16:09:13 +03:00
skip_to_key ( s ) ;
2011-05-12 11:26:32 +03:00
return 0 ;
2011-05-12 11:25:54 +03:00
}
2006-08-23 20:07:01 +03:00
}
2011-05-12 11:25:54 +03:00
/* no index or seeking by index failed */
2011-10-16 16:03:30 +03:00
if ( ff_seek_frame_binary ( s , stream_index , pts , flags ) < 0 )
2011-05-12 11:25:54 +03:00
return - 1 ;
2004-01-14 15:32:49 +02:00
asf_reset_header ( s ) ;
2013-07-30 16:09:13 +03:00
skip_to_key ( s ) ;
2004-01-14 15:32:49 +02:00
return 0 ;
2002-08-08 19:43:47 +03:00
}
2011-01-26 00:03:28 +02:00
AVInputFormat ff_asf_demuxer = {
2011-07-16 23:18:12 +03:00
. name = " asf " ,
2012-07-24 04:23:48 +03:00
. long_name = NULL_IF_CONFIG_SMALL ( " ASF (Advanced / Active Streaming Format) " ) ,
2011-07-16 23:18:12 +03:00
. priv_data_size = sizeof ( ASFContext ) ,
. read_probe = asf_probe ,
. read_header = asf_read_header ,
. read_packet = asf_read_packet ,
. read_close = asf_read_close ,
. read_seek = asf_read_seek ,
. read_timestamp = asf_read_pts ,
2012-04-06 17:50:48 +03:00
. flags = AVFMT_NOBINSEARCH | AVFMT_NOGENSEARCH ,
2012-04-06 16:54:23 +03:00
. priv_class = & asf_class ,
2002-05-20 19:31:13 +03:00
} ;