yading@11: /* yading@11: * Matroska file demuxer yading@11: * Copyright (c) 2003-2008 The FFmpeg Project yading@11: * yading@11: * This file is part of FFmpeg. yading@11: * yading@11: * FFmpeg is free software; you can redistribute it and/or yading@11: * modify it under the terms of the GNU Lesser General Public yading@11: * License as published by the Free Software Foundation; either yading@11: * version 2.1 of the License, or (at your option) any later version. yading@11: * yading@11: * FFmpeg is distributed in the hope that it will be useful, yading@11: * but WITHOUT ANY WARRANTY; without even the implied warranty of yading@11: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU yading@11: * Lesser General Public License for more details. yading@11: * yading@11: * You should have received a copy of the GNU Lesser General Public yading@11: * License along with FFmpeg; if not, write to the Free Software yading@11: * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA yading@11: */ yading@11: yading@11: /** yading@11: * @file yading@11: * Matroska file demuxer yading@11: * @author Ronald Bultje yading@11: * @author with a little help from Moritz Bunkus yading@11: * @author totally reworked by Aurelien Jacobs yading@11: * @see specs available on the Matroska project page: http://www.matroska.org/ yading@11: */ yading@11: yading@11: #include yading@11: #include "avformat.h" yading@11: #include "internal.h" yading@11: #include "avio_internal.h" yading@11: /* For ff_codec_get_id(). */ yading@11: #include "riff.h" yading@11: #include "isom.h" yading@11: #include "rmsipr.h" yading@11: #include "matroska.h" yading@11: #include "libavcodec/bytestream.h" yading@11: #include "libavcodec/mpeg4audio.h" yading@11: #include "libavutil/base64.h" yading@11: #include "libavutil/intfloat.h" yading@11: #include "libavutil/intreadwrite.h" yading@11: #include "libavutil/avstring.h" yading@11: #include "libavutil/lzo.h" yading@11: #include "libavutil/dict.h" yading@11: #if CONFIG_ZLIB yading@11: #include yading@11: #endif yading@11: #if CONFIG_BZLIB yading@11: #include yading@11: #endif yading@11: yading@11: typedef enum { yading@11: EBML_NONE, yading@11: EBML_UINT, yading@11: EBML_FLOAT, yading@11: EBML_STR, yading@11: EBML_UTF8, yading@11: EBML_BIN, yading@11: EBML_NEST, yading@11: EBML_PASS, yading@11: EBML_STOP, yading@11: EBML_TYPE_COUNT yading@11: } EbmlType; yading@11: yading@11: typedef const struct EbmlSyntax { yading@11: uint32_t id; yading@11: EbmlType type; yading@11: int list_elem_size; yading@11: int data_offset; yading@11: union { yading@11: uint64_t u; yading@11: double f; yading@11: const char *s; yading@11: const struct EbmlSyntax *n; yading@11: } def; yading@11: } EbmlSyntax; yading@11: yading@11: typedef struct { yading@11: int nb_elem; yading@11: void *elem; yading@11: } EbmlList; yading@11: yading@11: typedef struct { yading@11: int size; yading@11: uint8_t *data; yading@11: int64_t pos; yading@11: } EbmlBin; yading@11: yading@11: typedef struct { yading@11: uint64_t version; yading@11: uint64_t max_size; yading@11: uint64_t id_length; yading@11: char *doctype; yading@11: uint64_t doctype_version; yading@11: } Ebml; yading@11: yading@11: typedef struct { yading@11: uint64_t algo; yading@11: EbmlBin settings; yading@11: } MatroskaTrackCompression; yading@11: yading@11: typedef struct { yading@11: uint64_t algo; yading@11: EbmlBin key_id; yading@11: } MatroskaTrackEncryption; yading@11: yading@11: typedef struct { yading@11: uint64_t scope; yading@11: uint64_t type; yading@11: MatroskaTrackCompression compression; yading@11: MatroskaTrackEncryption encryption; yading@11: } MatroskaTrackEncoding; yading@11: yading@11: typedef struct { yading@11: double frame_rate; yading@11: uint64_t display_width; yading@11: uint64_t display_height; yading@11: uint64_t pixel_width; yading@11: uint64_t pixel_height; yading@11: EbmlBin color_space; yading@11: uint64_t stereo_mode; yading@11: uint64_t alpha_mode; yading@11: } MatroskaTrackVideo; yading@11: yading@11: typedef struct { yading@11: double samplerate; yading@11: double out_samplerate; yading@11: uint64_t bitdepth; yading@11: uint64_t channels; yading@11: yading@11: /* real audio header (extracted from extradata) */ yading@11: int coded_framesize; yading@11: int sub_packet_h; yading@11: int frame_size; yading@11: int sub_packet_size; yading@11: int sub_packet_cnt; yading@11: int pkt_cnt; yading@11: uint64_t buf_timecode; yading@11: uint8_t *buf; yading@11: } MatroskaTrackAudio; yading@11: yading@11: typedef struct { yading@11: uint64_t uid; yading@11: uint64_t type; yading@11: } MatroskaTrackPlane; yading@11: yading@11: typedef struct { yading@11: EbmlList combine_planes; yading@11: } MatroskaTrackOperation; yading@11: yading@11: typedef struct { yading@11: uint64_t num; yading@11: uint64_t uid; yading@11: uint64_t type; yading@11: char *name; yading@11: char *codec_id; yading@11: EbmlBin codec_priv; yading@11: char *language; yading@11: double time_scale; yading@11: uint64_t default_duration; yading@11: uint64_t flag_default; yading@11: uint64_t flag_forced; yading@11: MatroskaTrackVideo video; yading@11: MatroskaTrackAudio audio; yading@11: MatroskaTrackOperation operation; yading@11: EbmlList encodings; yading@11: yading@11: AVStream *stream; yading@11: int64_t end_timecode; yading@11: int ms_compat; yading@11: uint64_t max_block_additional_id; yading@11: } MatroskaTrack; yading@11: yading@11: typedef struct { yading@11: uint64_t uid; yading@11: char *filename; yading@11: char *mime; yading@11: EbmlBin bin; yading@11: yading@11: AVStream *stream; yading@11: } MatroskaAttachement; yading@11: yading@11: typedef struct { yading@11: uint64_t start; yading@11: uint64_t end; yading@11: uint64_t uid; yading@11: char *title; yading@11: yading@11: AVChapter *chapter; yading@11: } MatroskaChapter; yading@11: yading@11: typedef struct { yading@11: uint64_t track; yading@11: uint64_t pos; yading@11: } MatroskaIndexPos; yading@11: yading@11: typedef struct { yading@11: uint64_t time; yading@11: EbmlList pos; yading@11: } MatroskaIndex; yading@11: yading@11: typedef struct { yading@11: char *name; yading@11: char *string; yading@11: char *lang; yading@11: uint64_t def; yading@11: EbmlList sub; yading@11: } MatroskaTag; yading@11: yading@11: typedef struct { yading@11: char *type; yading@11: uint64_t typevalue; yading@11: uint64_t trackuid; yading@11: uint64_t chapteruid; yading@11: uint64_t attachuid; yading@11: } MatroskaTagTarget; yading@11: yading@11: typedef struct { yading@11: MatroskaTagTarget target; yading@11: EbmlList tag; yading@11: } MatroskaTags; yading@11: yading@11: typedef struct { yading@11: uint64_t id; yading@11: uint64_t pos; yading@11: } MatroskaSeekhead; yading@11: yading@11: typedef struct { yading@11: uint64_t start; yading@11: uint64_t length; yading@11: } MatroskaLevel; yading@11: yading@11: typedef struct { yading@11: uint64_t timecode; yading@11: EbmlList blocks; yading@11: } MatroskaCluster; yading@11: yading@11: typedef struct { yading@11: AVFormatContext *ctx; yading@11: yading@11: /* EBML stuff */ yading@11: int num_levels; yading@11: MatroskaLevel levels[EBML_MAX_DEPTH]; yading@11: int level_up; yading@11: uint32_t current_id; yading@11: yading@11: uint64_t time_scale; yading@11: double duration; yading@11: char *title; yading@11: EbmlBin date_utc; yading@11: EbmlList tracks; yading@11: EbmlList attachments; yading@11: EbmlList chapters; yading@11: EbmlList index; yading@11: EbmlList tags; yading@11: EbmlList seekhead; yading@11: yading@11: /* byte position of the segment inside the stream */ yading@11: int64_t segment_start; yading@11: yading@11: /* the packet queue */ yading@11: AVPacket **packets; yading@11: int num_packets; yading@11: AVPacket *prev_pkt; yading@11: yading@11: int done; yading@11: yading@11: /* What to skip before effectively reading a packet. */ yading@11: int skip_to_keyframe; yading@11: uint64_t skip_to_timecode; yading@11: yading@11: /* File has a CUES element, but we defer parsing until it is needed. */ yading@11: int cues_parsing_deferred; yading@11: yading@11: int current_cluster_num_blocks; yading@11: int64_t current_cluster_pos; yading@11: MatroskaCluster current_cluster; yading@11: yading@11: /* File has SSA subtitles which prevent incremental cluster parsing. */ yading@11: int contains_ssa; yading@11: } MatroskaDemuxContext; yading@11: yading@11: typedef struct { yading@11: uint64_t duration; yading@11: int64_t reference; yading@11: uint64_t non_simple; yading@11: EbmlBin bin; yading@11: uint64_t additional_id; yading@11: EbmlBin additional; yading@11: } MatroskaBlock; yading@11: yading@11: static EbmlSyntax ebml_header[] = { yading@11: { EBML_ID_EBMLREADVERSION, EBML_UINT, 0, offsetof(Ebml,version), {.u=EBML_VERSION} }, yading@11: { EBML_ID_EBMLMAXSIZELENGTH, EBML_UINT, 0, offsetof(Ebml,max_size), {.u=8} }, yading@11: { EBML_ID_EBMLMAXIDLENGTH, EBML_UINT, 0, offsetof(Ebml,id_length), {.u=4} }, yading@11: { EBML_ID_DOCTYPE, EBML_STR, 0, offsetof(Ebml,doctype), {.s="(none)"} }, yading@11: { EBML_ID_DOCTYPEREADVERSION, EBML_UINT, 0, offsetof(Ebml,doctype_version), {.u=1} }, yading@11: { EBML_ID_EBMLVERSION, EBML_NONE }, yading@11: { EBML_ID_DOCTYPEVERSION, EBML_NONE }, yading@11: { 0 } yading@11: }; yading@11: yading@11: static EbmlSyntax ebml_syntax[] = { yading@11: { EBML_ID_HEADER, EBML_NEST, 0, 0, {.n=ebml_header} }, yading@11: { 0 } yading@11: }; yading@11: yading@11: static EbmlSyntax matroska_info[] = { yading@11: { MATROSKA_ID_TIMECODESCALE, EBML_UINT, 0, offsetof(MatroskaDemuxContext,time_scale), {.u=1000000} }, yading@11: { MATROSKA_ID_DURATION, EBML_FLOAT, 0, offsetof(MatroskaDemuxContext,duration) }, yading@11: { MATROSKA_ID_TITLE, EBML_UTF8, 0, offsetof(MatroskaDemuxContext,title) }, yading@11: { MATROSKA_ID_WRITINGAPP, EBML_NONE }, yading@11: { MATROSKA_ID_MUXINGAPP, EBML_NONE }, yading@11: { MATROSKA_ID_DATEUTC, EBML_BIN, 0, offsetof(MatroskaDemuxContext,date_utc) }, yading@11: { MATROSKA_ID_SEGMENTUID, EBML_NONE }, yading@11: { 0 } yading@11: }; yading@11: yading@11: static EbmlSyntax matroska_track_video[] = { yading@11: { MATROSKA_ID_VIDEOFRAMERATE, EBML_FLOAT,0, offsetof(MatroskaTrackVideo,frame_rate) }, yading@11: { MATROSKA_ID_VIDEODISPLAYWIDTH, EBML_UINT, 0, offsetof(MatroskaTrackVideo,display_width), {.u=-1} }, yading@11: { MATROSKA_ID_VIDEODISPLAYHEIGHT, EBML_UINT, 0, offsetof(MatroskaTrackVideo,display_height), {.u=-1} }, yading@11: { MATROSKA_ID_VIDEOPIXELWIDTH, EBML_UINT, 0, offsetof(MatroskaTrackVideo,pixel_width) }, yading@11: { MATROSKA_ID_VIDEOPIXELHEIGHT, EBML_UINT, 0, offsetof(MatroskaTrackVideo,pixel_height) }, yading@11: { MATROSKA_ID_VIDEOCOLORSPACE, EBML_BIN, 0, offsetof(MatroskaTrackVideo,color_space) }, yading@11: { MATROSKA_ID_VIDEOSTEREOMODE, EBML_UINT, 0, offsetof(MatroskaTrackVideo,stereo_mode) }, yading@11: { MATROSKA_ID_VIDEOALPHAMODE, EBML_UINT, 0, offsetof(MatroskaTrackVideo,alpha_mode) }, yading@11: { MATROSKA_ID_VIDEOPIXELCROPB, EBML_NONE }, yading@11: { MATROSKA_ID_VIDEOPIXELCROPT, EBML_NONE }, yading@11: { MATROSKA_ID_VIDEOPIXELCROPL, EBML_NONE }, yading@11: { MATROSKA_ID_VIDEOPIXELCROPR, EBML_NONE }, yading@11: { MATROSKA_ID_VIDEODISPLAYUNIT, EBML_NONE }, yading@11: { MATROSKA_ID_VIDEOFLAGINTERLACED,EBML_NONE }, yading@11: { MATROSKA_ID_VIDEOASPECTRATIO, EBML_NONE }, yading@11: { 0 } yading@11: }; yading@11: yading@11: static EbmlSyntax matroska_track_audio[] = { yading@11: { MATROSKA_ID_AUDIOSAMPLINGFREQ, EBML_FLOAT,0, offsetof(MatroskaTrackAudio,samplerate), {.f=8000.0} }, yading@11: { MATROSKA_ID_AUDIOOUTSAMPLINGFREQ,EBML_FLOAT,0,offsetof(MatroskaTrackAudio,out_samplerate) }, yading@11: { MATROSKA_ID_AUDIOBITDEPTH, EBML_UINT, 0, offsetof(MatroskaTrackAudio,bitdepth) }, yading@11: { MATROSKA_ID_AUDIOCHANNELS, EBML_UINT, 0, offsetof(MatroskaTrackAudio,channels), {.u=1} }, yading@11: { 0 } yading@11: }; yading@11: yading@11: static EbmlSyntax matroska_track_encoding_compression[] = { yading@11: { MATROSKA_ID_ENCODINGCOMPALGO, EBML_UINT, 0, offsetof(MatroskaTrackCompression,algo), {.u=0} }, yading@11: { MATROSKA_ID_ENCODINGCOMPSETTINGS,EBML_BIN, 0, offsetof(MatroskaTrackCompression,settings) }, yading@11: { 0 } yading@11: }; yading@11: yading@11: static EbmlSyntax matroska_track_encoding_encryption[] = { yading@11: { MATROSKA_ID_ENCODINGENCALGO, EBML_UINT, 0, offsetof(MatroskaTrackEncryption,algo), {.u=0} }, yading@11: { MATROSKA_ID_ENCODINGENCKEYID, EBML_BIN, 0, offsetof(MatroskaTrackEncryption,key_id) }, yading@11: { MATROSKA_ID_ENCODINGENCAESSETTINGS, EBML_NONE }, yading@11: { MATROSKA_ID_ENCODINGSIGALGO, EBML_NONE }, yading@11: { MATROSKA_ID_ENCODINGSIGHASHALGO, EBML_NONE }, yading@11: { MATROSKA_ID_ENCODINGSIGKEYID, EBML_NONE }, yading@11: { MATROSKA_ID_ENCODINGSIGNATURE, EBML_NONE }, yading@11: { 0 } yading@11: }; yading@11: static EbmlSyntax matroska_track_encoding[] = { yading@11: { MATROSKA_ID_ENCODINGSCOPE, EBML_UINT, 0, offsetof(MatroskaTrackEncoding,scope), {.u=1} }, yading@11: { MATROSKA_ID_ENCODINGTYPE, EBML_UINT, 0, offsetof(MatroskaTrackEncoding,type), {.u=0} }, yading@11: { MATROSKA_ID_ENCODINGCOMPRESSION,EBML_NEST, 0, offsetof(MatroskaTrackEncoding,compression), {.n=matroska_track_encoding_compression} }, yading@11: { MATROSKA_ID_ENCODINGENCRYPTION, EBML_NEST, 0, offsetof(MatroskaTrackEncoding,encryption), {.n=matroska_track_encoding_encryption} }, yading@11: { MATROSKA_ID_ENCODINGORDER, EBML_NONE }, yading@11: { 0 } yading@11: }; yading@11: yading@11: static EbmlSyntax matroska_track_encodings[] = { yading@11: { MATROSKA_ID_TRACKCONTENTENCODING, EBML_NEST, sizeof(MatroskaTrackEncoding), offsetof(MatroskaTrack,encodings), {.n=matroska_track_encoding} }, yading@11: { 0 } yading@11: }; yading@11: yading@11: static EbmlSyntax matroska_track_plane[] = { yading@11: { MATROSKA_ID_TRACKPLANEUID, EBML_UINT, 0, offsetof(MatroskaTrackPlane,uid) }, yading@11: { MATROSKA_ID_TRACKPLANETYPE, EBML_UINT, 0, offsetof(MatroskaTrackPlane,type) }, yading@11: { 0 } yading@11: }; yading@11: yading@11: static EbmlSyntax matroska_track_combine_planes[] = { yading@11: { MATROSKA_ID_TRACKPLANE, EBML_NEST, sizeof(MatroskaTrackPlane), offsetof(MatroskaTrackOperation,combine_planes), {.n=matroska_track_plane} }, yading@11: { 0 } yading@11: }; yading@11: yading@11: static EbmlSyntax matroska_track_operation[] = { yading@11: { MATROSKA_ID_TRACKCOMBINEPLANES, EBML_NEST, 0, 0, {.n=matroska_track_combine_planes} }, yading@11: { 0 } yading@11: }; yading@11: yading@11: static EbmlSyntax matroska_track[] = { yading@11: { MATROSKA_ID_TRACKNUMBER, EBML_UINT, 0, offsetof(MatroskaTrack,num) }, yading@11: { MATROSKA_ID_TRACKNAME, EBML_UTF8, 0, offsetof(MatroskaTrack,name) }, yading@11: { MATROSKA_ID_TRACKUID, EBML_UINT, 0, offsetof(MatroskaTrack,uid) }, yading@11: { MATROSKA_ID_TRACKTYPE, EBML_UINT, 0, offsetof(MatroskaTrack,type) }, yading@11: { MATROSKA_ID_CODECID, EBML_STR, 0, offsetof(MatroskaTrack,codec_id) }, yading@11: { MATROSKA_ID_CODECPRIVATE, EBML_BIN, 0, offsetof(MatroskaTrack,codec_priv) }, yading@11: { MATROSKA_ID_TRACKLANGUAGE, EBML_UTF8, 0, offsetof(MatroskaTrack,language), {.s="eng"} }, yading@11: { MATROSKA_ID_TRACKDEFAULTDURATION, EBML_UINT, 0, offsetof(MatroskaTrack,default_duration) }, yading@11: { MATROSKA_ID_TRACKTIMECODESCALE, EBML_FLOAT,0, offsetof(MatroskaTrack,time_scale), {.f=1.0} }, yading@11: { MATROSKA_ID_TRACKFLAGDEFAULT, EBML_UINT, 0, offsetof(MatroskaTrack,flag_default), {.u=1} }, yading@11: { MATROSKA_ID_TRACKFLAGFORCED, EBML_UINT, 0, offsetof(MatroskaTrack,flag_forced), {.u=0} }, yading@11: { MATROSKA_ID_TRACKVIDEO, EBML_NEST, 0, offsetof(MatroskaTrack,video), {.n=matroska_track_video} }, yading@11: { MATROSKA_ID_TRACKAUDIO, EBML_NEST, 0, offsetof(MatroskaTrack,audio), {.n=matroska_track_audio} }, yading@11: { MATROSKA_ID_TRACKOPERATION, EBML_NEST, 0, offsetof(MatroskaTrack,operation), {.n=matroska_track_operation} }, yading@11: { MATROSKA_ID_TRACKCONTENTENCODINGS,EBML_NEST, 0, 0, {.n=matroska_track_encodings} }, yading@11: { MATROSKA_ID_TRACKMAXBLKADDID, EBML_UINT, 0, offsetof(MatroskaTrack,max_block_additional_id) }, yading@11: { MATROSKA_ID_TRACKFLAGENABLED, EBML_NONE }, yading@11: { MATROSKA_ID_TRACKFLAGLACING, EBML_NONE }, yading@11: { MATROSKA_ID_CODECNAME, EBML_NONE }, yading@11: { MATROSKA_ID_CODECDECODEALL, EBML_NONE }, yading@11: { MATROSKA_ID_CODECINFOURL, EBML_NONE }, yading@11: { MATROSKA_ID_CODECDOWNLOADURL, EBML_NONE }, yading@11: { MATROSKA_ID_TRACKMINCACHE, EBML_NONE }, yading@11: { MATROSKA_ID_TRACKMAXCACHE, EBML_NONE }, yading@11: { 0 } yading@11: }; yading@11: yading@11: static EbmlSyntax matroska_tracks[] = { yading@11: { MATROSKA_ID_TRACKENTRY, EBML_NEST, sizeof(MatroskaTrack), offsetof(MatroskaDemuxContext,tracks), {.n=matroska_track} }, yading@11: { 0 } yading@11: }; yading@11: yading@11: static EbmlSyntax matroska_attachment[] = { yading@11: { MATROSKA_ID_FILEUID, EBML_UINT, 0, offsetof(MatroskaAttachement,uid) }, yading@11: { MATROSKA_ID_FILENAME, EBML_UTF8, 0, offsetof(MatroskaAttachement,filename) }, yading@11: { MATROSKA_ID_FILEMIMETYPE, EBML_STR, 0, offsetof(MatroskaAttachement,mime) }, yading@11: { MATROSKA_ID_FILEDATA, EBML_BIN, 0, offsetof(MatroskaAttachement,bin) }, yading@11: { MATROSKA_ID_FILEDESC, EBML_NONE }, yading@11: { 0 } yading@11: }; yading@11: yading@11: static EbmlSyntax matroska_attachments[] = { yading@11: { MATROSKA_ID_ATTACHEDFILE, EBML_NEST, sizeof(MatroskaAttachement), offsetof(MatroskaDemuxContext,attachments), {.n=matroska_attachment} }, yading@11: { 0 } yading@11: }; yading@11: yading@11: static EbmlSyntax matroska_chapter_display[] = { yading@11: { MATROSKA_ID_CHAPSTRING, EBML_UTF8, 0, offsetof(MatroskaChapter,title) }, yading@11: { MATROSKA_ID_CHAPLANG, EBML_NONE }, yading@11: { 0 } yading@11: }; yading@11: yading@11: static EbmlSyntax matroska_chapter_entry[] = { yading@11: { MATROSKA_ID_CHAPTERTIMESTART, EBML_UINT, 0, offsetof(MatroskaChapter,start), {.u=AV_NOPTS_VALUE} }, yading@11: { MATROSKA_ID_CHAPTERTIMEEND, EBML_UINT, 0, offsetof(MatroskaChapter,end), {.u=AV_NOPTS_VALUE} }, yading@11: { MATROSKA_ID_CHAPTERUID, EBML_UINT, 0, offsetof(MatroskaChapter,uid) }, yading@11: { MATROSKA_ID_CHAPTERDISPLAY, EBML_NEST, 0, 0, {.n=matroska_chapter_display} }, yading@11: { MATROSKA_ID_CHAPTERFLAGHIDDEN, EBML_NONE }, yading@11: { MATROSKA_ID_CHAPTERFLAGENABLED, EBML_NONE }, yading@11: { MATROSKA_ID_CHAPTERPHYSEQUIV, EBML_NONE }, yading@11: { MATROSKA_ID_CHAPTERATOM, EBML_NONE }, yading@11: { 0 } yading@11: }; yading@11: yading@11: static EbmlSyntax matroska_chapter[] = { yading@11: { MATROSKA_ID_CHAPTERATOM, EBML_NEST, sizeof(MatroskaChapter), offsetof(MatroskaDemuxContext,chapters), {.n=matroska_chapter_entry} }, yading@11: { MATROSKA_ID_EDITIONUID, EBML_NONE }, yading@11: { MATROSKA_ID_EDITIONFLAGHIDDEN, EBML_NONE }, yading@11: { MATROSKA_ID_EDITIONFLAGDEFAULT, EBML_NONE }, yading@11: { MATROSKA_ID_EDITIONFLAGORDERED, EBML_NONE }, yading@11: { 0 } yading@11: }; yading@11: yading@11: static EbmlSyntax matroska_chapters[] = { yading@11: { MATROSKA_ID_EDITIONENTRY, EBML_NEST, 0, 0, {.n=matroska_chapter} }, yading@11: { 0 } yading@11: }; yading@11: yading@11: static EbmlSyntax matroska_index_pos[] = { yading@11: { MATROSKA_ID_CUETRACK, EBML_UINT, 0, offsetof(MatroskaIndexPos,track) }, yading@11: { MATROSKA_ID_CUECLUSTERPOSITION, EBML_UINT, 0, offsetof(MatroskaIndexPos,pos) }, yading@11: { MATROSKA_ID_CUEBLOCKNUMBER, EBML_NONE }, yading@11: { 0 } yading@11: }; yading@11: yading@11: static EbmlSyntax matroska_index_entry[] = { yading@11: { MATROSKA_ID_CUETIME, EBML_UINT, 0, offsetof(MatroskaIndex,time) }, yading@11: { MATROSKA_ID_CUETRACKPOSITION, EBML_NEST, sizeof(MatroskaIndexPos), offsetof(MatroskaIndex,pos), {.n=matroska_index_pos} }, yading@11: { 0 } yading@11: }; yading@11: yading@11: static EbmlSyntax matroska_index[] = { yading@11: { MATROSKA_ID_POINTENTRY, EBML_NEST, sizeof(MatroskaIndex), offsetof(MatroskaDemuxContext,index), {.n=matroska_index_entry} }, yading@11: { 0 } yading@11: }; yading@11: yading@11: static EbmlSyntax matroska_simpletag[] = { yading@11: { MATROSKA_ID_TAGNAME, EBML_UTF8, 0, offsetof(MatroskaTag,name) }, yading@11: { MATROSKA_ID_TAGSTRING, EBML_UTF8, 0, offsetof(MatroskaTag,string) }, yading@11: { MATROSKA_ID_TAGLANG, EBML_STR, 0, offsetof(MatroskaTag,lang), {.s="und"} }, yading@11: { MATROSKA_ID_TAGDEFAULT, EBML_UINT, 0, offsetof(MatroskaTag,def) }, yading@11: { MATROSKA_ID_TAGDEFAULT_BUG, EBML_UINT, 0, offsetof(MatroskaTag,def) }, yading@11: { MATROSKA_ID_SIMPLETAG, EBML_NEST, sizeof(MatroskaTag), offsetof(MatroskaTag,sub), {.n=matroska_simpletag} }, yading@11: { 0 } yading@11: }; yading@11: yading@11: static EbmlSyntax matroska_tagtargets[] = { yading@11: { MATROSKA_ID_TAGTARGETS_TYPE, EBML_STR, 0, offsetof(MatroskaTagTarget,type) }, yading@11: { MATROSKA_ID_TAGTARGETS_TYPEVALUE, EBML_UINT, 0, offsetof(MatroskaTagTarget,typevalue), {.u=50} }, yading@11: { MATROSKA_ID_TAGTARGETS_TRACKUID, EBML_UINT, 0, offsetof(MatroskaTagTarget,trackuid) }, yading@11: { MATROSKA_ID_TAGTARGETS_CHAPTERUID,EBML_UINT, 0, offsetof(MatroskaTagTarget,chapteruid) }, yading@11: { MATROSKA_ID_TAGTARGETS_ATTACHUID, EBML_UINT, 0, offsetof(MatroskaTagTarget,attachuid) }, yading@11: { 0 } yading@11: }; yading@11: yading@11: static EbmlSyntax matroska_tag[] = { yading@11: { MATROSKA_ID_SIMPLETAG, EBML_NEST, sizeof(MatroskaTag), offsetof(MatroskaTags,tag), {.n=matroska_simpletag} }, yading@11: { MATROSKA_ID_TAGTARGETS, EBML_NEST, 0, offsetof(MatroskaTags,target), {.n=matroska_tagtargets} }, yading@11: { 0 } yading@11: }; yading@11: yading@11: static EbmlSyntax matroska_tags[] = { yading@11: { MATROSKA_ID_TAG, EBML_NEST, sizeof(MatroskaTags), offsetof(MatroskaDemuxContext,tags), {.n=matroska_tag} }, yading@11: { 0 } yading@11: }; yading@11: yading@11: static EbmlSyntax matroska_seekhead_entry[] = { yading@11: { MATROSKA_ID_SEEKID, EBML_UINT, 0, offsetof(MatroskaSeekhead,id) }, yading@11: { MATROSKA_ID_SEEKPOSITION, EBML_UINT, 0, offsetof(MatroskaSeekhead,pos), {.u=-1} }, yading@11: { 0 } yading@11: }; yading@11: yading@11: static EbmlSyntax matroska_seekhead[] = { yading@11: { MATROSKA_ID_SEEKENTRY, EBML_NEST, sizeof(MatroskaSeekhead), offsetof(MatroskaDemuxContext,seekhead), {.n=matroska_seekhead_entry} }, yading@11: { 0 } yading@11: }; yading@11: yading@11: static EbmlSyntax matroska_segment[] = { yading@11: { MATROSKA_ID_INFO, EBML_NEST, 0, 0, {.n=matroska_info } }, yading@11: { MATROSKA_ID_TRACKS, EBML_NEST, 0, 0, {.n=matroska_tracks } }, yading@11: { MATROSKA_ID_ATTACHMENTS, EBML_NEST, 0, 0, {.n=matroska_attachments} }, yading@11: { MATROSKA_ID_CHAPTERS, EBML_NEST, 0, 0, {.n=matroska_chapters } }, yading@11: { MATROSKA_ID_CUES, EBML_NEST, 0, 0, {.n=matroska_index } }, yading@11: { MATROSKA_ID_TAGS, EBML_NEST, 0, 0, {.n=matroska_tags } }, yading@11: { MATROSKA_ID_SEEKHEAD, EBML_NEST, 0, 0, {.n=matroska_seekhead } }, yading@11: { MATROSKA_ID_CLUSTER, EBML_STOP }, yading@11: { 0 } yading@11: }; yading@11: yading@11: static EbmlSyntax matroska_segments[] = { yading@11: { MATROSKA_ID_SEGMENT, EBML_NEST, 0, 0, {.n=matroska_segment } }, yading@11: { 0 } yading@11: }; yading@11: yading@11: static EbmlSyntax matroska_blockmore[] = { yading@11: { MATROSKA_ID_BLOCKADDID, EBML_UINT, 0, offsetof(MatroskaBlock,additional_id) }, yading@11: { MATROSKA_ID_BLOCKADDITIONAL, EBML_BIN, 0, offsetof(MatroskaBlock,additional) }, yading@11: { 0 } yading@11: }; yading@11: yading@11: static EbmlSyntax matroska_blockadditions[] = { yading@11: { MATROSKA_ID_BLOCKMORE, EBML_NEST, 0, 0, {.n=matroska_blockmore} }, yading@11: { 0 } yading@11: }; yading@11: yading@11: static EbmlSyntax matroska_blockgroup[] = { yading@11: { MATROSKA_ID_BLOCK, EBML_BIN, 0, offsetof(MatroskaBlock,bin) }, yading@11: { MATROSKA_ID_BLOCKADDITIONS, EBML_NEST, 0, 0, {.n=matroska_blockadditions} }, yading@11: { MATROSKA_ID_SIMPLEBLOCK, EBML_BIN, 0, offsetof(MatroskaBlock,bin) }, yading@11: { MATROSKA_ID_BLOCKDURATION, EBML_UINT, 0, offsetof(MatroskaBlock,duration) }, yading@11: { MATROSKA_ID_BLOCKREFERENCE, EBML_UINT, 0, offsetof(MatroskaBlock,reference) }, yading@11: { 1, EBML_UINT, 0, offsetof(MatroskaBlock,non_simple), {.u=1} }, yading@11: { 0 } yading@11: }; yading@11: yading@11: static EbmlSyntax matroska_cluster[] = { yading@11: { MATROSKA_ID_CLUSTERTIMECODE,EBML_UINT,0, offsetof(MatroskaCluster,timecode) }, yading@11: { MATROSKA_ID_BLOCKGROUP, EBML_NEST, sizeof(MatroskaBlock), offsetof(MatroskaCluster,blocks), {.n=matroska_blockgroup} }, yading@11: { MATROSKA_ID_SIMPLEBLOCK, EBML_PASS, sizeof(MatroskaBlock), offsetof(MatroskaCluster,blocks), {.n=matroska_blockgroup} }, yading@11: { MATROSKA_ID_CLUSTERPOSITION,EBML_NONE }, yading@11: { MATROSKA_ID_CLUSTERPREVSIZE,EBML_NONE }, yading@11: { 0 } yading@11: }; yading@11: yading@11: static EbmlSyntax matroska_clusters[] = { yading@11: { MATROSKA_ID_CLUSTER, EBML_NEST, 0, 0, {.n=matroska_cluster} }, yading@11: { MATROSKA_ID_INFO, EBML_NONE }, yading@11: { MATROSKA_ID_CUES, EBML_NONE }, yading@11: { MATROSKA_ID_TAGS, EBML_NONE }, yading@11: { MATROSKA_ID_SEEKHEAD, EBML_NONE }, yading@11: { 0 } yading@11: }; yading@11: yading@11: static EbmlSyntax matroska_cluster_incremental_parsing[] = { yading@11: { MATROSKA_ID_CLUSTERTIMECODE,EBML_UINT,0, offsetof(MatroskaCluster,timecode) }, yading@11: { MATROSKA_ID_BLOCKGROUP, EBML_NEST, sizeof(MatroskaBlock), offsetof(MatroskaCluster,blocks), {.n=matroska_blockgroup} }, yading@11: { MATROSKA_ID_SIMPLEBLOCK, EBML_PASS, sizeof(MatroskaBlock), offsetof(MatroskaCluster,blocks), {.n=matroska_blockgroup} }, yading@11: { MATROSKA_ID_CLUSTERPOSITION,EBML_NONE }, yading@11: { MATROSKA_ID_CLUSTERPREVSIZE,EBML_NONE }, yading@11: { MATROSKA_ID_INFO, EBML_NONE }, yading@11: { MATROSKA_ID_CUES, EBML_NONE }, yading@11: { MATROSKA_ID_TAGS, EBML_NONE }, yading@11: { MATROSKA_ID_SEEKHEAD, EBML_NONE }, yading@11: { MATROSKA_ID_CLUSTER, EBML_STOP }, yading@11: { 0 } yading@11: }; yading@11: yading@11: static EbmlSyntax matroska_cluster_incremental[] = { yading@11: { MATROSKA_ID_CLUSTERTIMECODE,EBML_UINT,0, offsetof(MatroskaCluster,timecode) }, yading@11: { MATROSKA_ID_BLOCKGROUP, EBML_STOP }, yading@11: { MATROSKA_ID_SIMPLEBLOCK, EBML_STOP }, yading@11: { MATROSKA_ID_CLUSTERPOSITION,EBML_NONE }, yading@11: { MATROSKA_ID_CLUSTERPREVSIZE,EBML_NONE }, yading@11: { 0 } yading@11: }; yading@11: yading@11: static EbmlSyntax matroska_clusters_incremental[] = { yading@11: { MATROSKA_ID_CLUSTER, EBML_NEST, 0, 0, {.n=matroska_cluster_incremental} }, yading@11: { MATROSKA_ID_INFO, EBML_NONE }, yading@11: { MATROSKA_ID_CUES, EBML_NONE }, yading@11: { MATROSKA_ID_TAGS, EBML_NONE }, yading@11: { MATROSKA_ID_SEEKHEAD, EBML_NONE }, yading@11: { 0 } yading@11: }; yading@11: yading@11: static const char *const matroska_doctypes[] = { "matroska", "webm" }; yading@11: yading@11: static int matroska_resync(MatroskaDemuxContext *matroska, int64_t last_pos) yading@11: { yading@11: AVIOContext *pb = matroska->ctx->pb; yading@11: uint32_t id; yading@11: matroska->current_id = 0; yading@11: matroska->num_levels = 0; yading@11: yading@11: // seek to next position to resync from yading@11: if (avio_seek(pb, last_pos + 1, SEEK_SET) < 0 || avio_tell(pb) <= last_pos) yading@11: goto eof; yading@11: yading@11: id = avio_rb32(pb); yading@11: yading@11: // try to find a toplevel element yading@11: while (!url_feof(pb)) { yading@11: if (id == MATROSKA_ID_INFO || id == MATROSKA_ID_TRACKS || yading@11: id == MATROSKA_ID_CUES || id == MATROSKA_ID_TAGS || yading@11: id == MATROSKA_ID_SEEKHEAD || id == MATROSKA_ID_ATTACHMENTS || yading@11: id == MATROSKA_ID_CLUSTER || id == MATROSKA_ID_CHAPTERS) yading@11: { yading@11: matroska->current_id = id; yading@11: return 0; yading@11: } yading@11: id = (id << 8) | avio_r8(pb); yading@11: } yading@11: eof: yading@11: matroska->done = 1; yading@11: return AVERROR_EOF; yading@11: } yading@11: yading@11: /* yading@11: * Return: Whether we reached the end of a level in the hierarchy or not. yading@11: */ yading@11: static int ebml_level_end(MatroskaDemuxContext *matroska) yading@11: { yading@11: AVIOContext *pb = matroska->ctx->pb; yading@11: int64_t pos = avio_tell(pb); yading@11: yading@11: if (matroska->num_levels > 0) { yading@11: MatroskaLevel *level = &matroska->levels[matroska->num_levels - 1]; yading@11: if (pos - level->start >= level->length || matroska->current_id) { yading@11: matroska->num_levels--; yading@11: return 1; yading@11: } yading@11: } yading@11: return 0; yading@11: } yading@11: yading@11: /* yading@11: * Read: an "EBML number", which is defined as a variable-length yading@11: * array of bytes. The first byte indicates the length by giving a yading@11: * number of 0-bits followed by a one. The position of the first yading@11: * "one" bit inside the first byte indicates the length of this yading@11: * number. yading@11: * Returns: number of bytes read, < 0 on error yading@11: */ yading@11: static int ebml_read_num(MatroskaDemuxContext *matroska, AVIOContext *pb, yading@11: int max_size, uint64_t *number) yading@11: { yading@11: int read = 1, n = 1; yading@11: uint64_t total = 0; yading@11: yading@11: /* The first byte tells us the length in bytes - avio_r8() can normally yading@11: * return 0, but since that's not a valid first ebmlID byte, we can yading@11: * use it safely here to catch EOS. */ yading@11: if (!(total = avio_r8(pb))) { yading@11: /* we might encounter EOS here */ yading@11: if (!url_feof(pb)) { yading@11: int64_t pos = avio_tell(pb); yading@11: av_log(matroska->ctx, AV_LOG_ERROR, yading@11: "Read error at pos. %"PRIu64" (0x%"PRIx64")\n", yading@11: pos, pos); yading@11: return pb->error ? pb->error : AVERROR(EIO); yading@11: } yading@11: return AVERROR_EOF; yading@11: } yading@11: yading@11: /* get the length of the EBML number */ yading@11: read = 8 - ff_log2_tab[total]; yading@11: if (read > max_size) { yading@11: int64_t pos = avio_tell(pb) - 1; yading@11: av_log(matroska->ctx, AV_LOG_ERROR, yading@11: "Invalid EBML number size tag 0x%02x at pos %"PRIu64" (0x%"PRIx64")\n", yading@11: (uint8_t) total, pos, pos); yading@11: return AVERROR_INVALIDDATA; yading@11: } yading@11: yading@11: /* read out length */ yading@11: total ^= 1 << ff_log2_tab[total]; yading@11: while (n++ < read) yading@11: total = (total << 8) | avio_r8(pb); yading@11: yading@11: *number = total; yading@11: yading@11: return read; yading@11: } yading@11: yading@11: /** yading@11: * Read a EBML length value. yading@11: * This needs special handling for the "unknown length" case which has multiple yading@11: * encodings. yading@11: */ yading@11: static int ebml_read_length(MatroskaDemuxContext *matroska, AVIOContext *pb, yading@11: uint64_t *number) yading@11: { yading@11: int res = ebml_read_num(matroska, pb, 8, number); yading@11: if (res > 0 && *number + 1 == 1ULL << (7 * res)) yading@11: *number = 0xffffffffffffffULL; yading@11: return res; yading@11: } yading@11: yading@11: /* yading@11: * Read the next element as an unsigned int. yading@11: * 0 is success, < 0 is failure. yading@11: */ yading@11: static int ebml_read_uint(AVIOContext *pb, int size, uint64_t *num) yading@11: { yading@11: int n = 0; yading@11: yading@11: if (size > 8) yading@11: return AVERROR_INVALIDDATA; yading@11: yading@11: /* big-endian ordering; build up number */ yading@11: *num = 0; yading@11: while (n++ < size) yading@11: *num = (*num << 8) | avio_r8(pb); yading@11: yading@11: return 0; yading@11: } yading@11: yading@11: /* yading@11: * Read the next element as a float. yading@11: * 0 is success, < 0 is failure. yading@11: */ yading@11: static int ebml_read_float(AVIOContext *pb, int size, double *num) yading@11: { yading@11: if (size == 0) { yading@11: *num = 0; yading@11: } else if (size == 4) { yading@11: *num = av_int2float(avio_rb32(pb)); yading@11: } else if (size == 8){ yading@11: *num = av_int2double(avio_rb64(pb)); yading@11: } else yading@11: return AVERROR_INVALIDDATA; yading@11: yading@11: return 0; yading@11: } yading@11: yading@11: /* yading@11: * Read the next element as an ASCII string. yading@11: * 0 is success, < 0 is failure. yading@11: */ yading@11: static int ebml_read_ascii(AVIOContext *pb, int size, char **str) yading@11: { yading@11: char *res; yading@11: yading@11: /* EBML strings are usually not 0-terminated, so we allocate one yading@11: * byte more, read the string and NULL-terminate it ourselves. */ yading@11: if (!(res = av_malloc(size + 1))) yading@11: return AVERROR(ENOMEM); yading@11: if (avio_read(pb, (uint8_t *) res, size) != size) { yading@11: av_free(res); yading@11: return AVERROR(EIO); yading@11: } yading@11: (res)[size] = '\0'; yading@11: av_free(*str); yading@11: *str = res; yading@11: yading@11: return 0; yading@11: } yading@11: yading@11: /* yading@11: * Read the next element as binary data. yading@11: * 0 is success, < 0 is failure. yading@11: */ yading@11: static int ebml_read_binary(AVIOContext *pb, int length, EbmlBin *bin) yading@11: { yading@11: av_fast_padded_malloc(&bin->data, &bin->size, length); yading@11: if (!bin->data) yading@11: return AVERROR(ENOMEM); yading@11: yading@11: bin->size = length; yading@11: bin->pos = avio_tell(pb); yading@11: if (avio_read(pb, bin->data, length) != length) { yading@11: av_freep(&bin->data); yading@11: bin->size = 0; yading@11: return AVERROR(EIO); yading@11: } yading@11: yading@11: return 0; yading@11: } yading@11: yading@11: /* yading@11: * Read the next element, but only the header. The contents yading@11: * are supposed to be sub-elements which can be read separately. yading@11: * 0 is success, < 0 is failure. yading@11: */ yading@11: static int ebml_read_master(MatroskaDemuxContext *matroska, uint64_t length) yading@11: { yading@11: AVIOContext *pb = matroska->ctx->pb; yading@11: MatroskaLevel *level; yading@11: yading@11: if (matroska->num_levels >= EBML_MAX_DEPTH) { yading@11: av_log(matroska->ctx, AV_LOG_ERROR, yading@11: "File moves beyond max. allowed depth (%d)\n", EBML_MAX_DEPTH); yading@11: return AVERROR(ENOSYS); yading@11: } yading@11: yading@11: level = &matroska->levels[matroska->num_levels++]; yading@11: level->start = avio_tell(pb); yading@11: level->length = length; yading@11: yading@11: return 0; yading@11: } yading@11: yading@11: /* yading@11: * Read signed/unsigned "EBML" numbers. yading@11: * Return: number of bytes processed, < 0 on error yading@11: */ yading@11: static int matroska_ebmlnum_uint(MatroskaDemuxContext *matroska, yading@11: uint8_t *data, uint32_t size, uint64_t *num) yading@11: { yading@11: AVIOContext pb; yading@11: ffio_init_context(&pb, data, size, 0, NULL, NULL, NULL, NULL); yading@11: return ebml_read_num(matroska, &pb, FFMIN(size, 8), num); yading@11: } yading@11: yading@11: /* yading@11: * Same as above, but signed. yading@11: */ yading@11: static int matroska_ebmlnum_sint(MatroskaDemuxContext *matroska, yading@11: uint8_t *data, uint32_t size, int64_t *num) yading@11: { yading@11: uint64_t unum; yading@11: int res; yading@11: yading@11: /* read as unsigned number first */ yading@11: if ((res = matroska_ebmlnum_uint(matroska, data, size, &unum)) < 0) yading@11: return res; yading@11: yading@11: /* make signed (weird way) */ yading@11: *num = unum - ((1LL << (7*res - 1)) - 1); yading@11: yading@11: return res; yading@11: } yading@11: yading@11: static int ebml_parse_elem(MatroskaDemuxContext *matroska, yading@11: EbmlSyntax *syntax, void *data); yading@11: yading@11: static int ebml_parse_id(MatroskaDemuxContext *matroska, EbmlSyntax *syntax, yading@11: uint32_t id, void *data) yading@11: { yading@11: int i; yading@11: for (i=0; syntax[i].id; i++) yading@11: if (id == syntax[i].id) yading@11: break; yading@11: if (!syntax[i].id && id == MATROSKA_ID_CLUSTER && yading@11: matroska->num_levels > 0 && yading@11: matroska->levels[matroska->num_levels-1].length == 0xffffffffffffff) yading@11: return 0; // we reached the end of an unknown size cluster yading@11: if (!syntax[i].id && id != EBML_ID_VOID && id != EBML_ID_CRC32) { yading@11: av_log(matroska->ctx, AV_LOG_INFO, "Unknown entry 0x%X\n", id); yading@11: if (matroska->ctx->error_recognition & AV_EF_EXPLODE) yading@11: return AVERROR_INVALIDDATA; yading@11: } yading@11: return ebml_parse_elem(matroska, &syntax[i], data); yading@11: } yading@11: yading@11: static int ebml_parse(MatroskaDemuxContext *matroska, EbmlSyntax *syntax, yading@11: void *data) yading@11: { yading@11: if (!matroska->current_id) { yading@11: uint64_t id; yading@11: int res = ebml_read_num(matroska, matroska->ctx->pb, 4, &id); yading@11: if (res < 0) yading@11: return res; yading@11: matroska->current_id = id | 1 << 7*res; yading@11: } yading@11: return ebml_parse_id(matroska, syntax, matroska->current_id, data); yading@11: } yading@11: yading@11: static int ebml_parse_nest(MatroskaDemuxContext *matroska, EbmlSyntax *syntax, yading@11: void *data) yading@11: { yading@11: int i, res = 0; yading@11: yading@11: for (i=0; syntax[i].id; i++) yading@11: switch (syntax[i].type) { yading@11: case EBML_UINT: yading@11: *(uint64_t *)((char *)data+syntax[i].data_offset) = syntax[i].def.u; yading@11: break; yading@11: case EBML_FLOAT: yading@11: *(double *)((char *)data+syntax[i].data_offset) = syntax[i].def.f; yading@11: break; yading@11: case EBML_STR: yading@11: case EBML_UTF8: yading@11: *(char **)((char *)data+syntax[i].data_offset) = av_strdup(syntax[i].def.s); yading@11: break; yading@11: } yading@11: yading@11: while (!res && !ebml_level_end(matroska)) yading@11: res = ebml_parse(matroska, syntax, data); yading@11: yading@11: return res; yading@11: } yading@11: yading@11: static int ebml_parse_elem(MatroskaDemuxContext *matroska, yading@11: EbmlSyntax *syntax, void *data) yading@11: { yading@11: static const uint64_t max_lengths[EBML_TYPE_COUNT] = { yading@11: [EBML_UINT] = 8, yading@11: [EBML_FLOAT] = 8, yading@11: // max. 16 MB for strings yading@11: [EBML_STR] = 0x1000000, yading@11: [EBML_UTF8] = 0x1000000, yading@11: // max. 256 MB for binary data yading@11: [EBML_BIN] = 0x10000000, yading@11: // no limits for anything else yading@11: }; yading@11: AVIOContext *pb = matroska->ctx->pb; yading@11: uint32_t id = syntax->id; yading@11: uint64_t length; yading@11: int res; yading@11: void *newelem; yading@11: yading@11: data = (char *)data + syntax->data_offset; yading@11: if (syntax->list_elem_size) { yading@11: EbmlList *list = data; yading@11: newelem = av_realloc(list->elem, (list->nb_elem+1)*syntax->list_elem_size); yading@11: if (!newelem) yading@11: return AVERROR(ENOMEM); yading@11: list->elem = newelem; yading@11: data = (char*)list->elem + list->nb_elem*syntax->list_elem_size; yading@11: memset(data, 0, syntax->list_elem_size); yading@11: list->nb_elem++; yading@11: } yading@11: yading@11: if (syntax->type != EBML_PASS && syntax->type != EBML_STOP) { yading@11: matroska->current_id = 0; yading@11: if ((res = ebml_read_length(matroska, pb, &length)) < 0) yading@11: return res; yading@11: if (max_lengths[syntax->type] && length > max_lengths[syntax->type]) { yading@11: av_log(matroska->ctx, AV_LOG_ERROR, yading@11: "Invalid length 0x%"PRIx64" > 0x%"PRIx64" for syntax element %i\n", yading@11: length, max_lengths[syntax->type], syntax->type); yading@11: return AVERROR_INVALIDDATA; yading@11: } yading@11: } yading@11: yading@11: switch (syntax->type) { yading@11: case EBML_UINT: res = ebml_read_uint (pb, length, data); break; yading@11: case EBML_FLOAT: res = ebml_read_float (pb, length, data); break; yading@11: case EBML_STR: yading@11: case EBML_UTF8: res = ebml_read_ascii (pb, length, data); break; yading@11: case EBML_BIN: res = ebml_read_binary(pb, length, data); break; yading@11: case EBML_NEST: if ((res=ebml_read_master(matroska, length)) < 0) yading@11: return res; yading@11: if (id == MATROSKA_ID_SEGMENT) yading@11: matroska->segment_start = avio_tell(matroska->ctx->pb); yading@11: return ebml_parse_nest(matroska, syntax->def.n, data); yading@11: case EBML_PASS: return ebml_parse_id(matroska, syntax->def.n, id, data); yading@11: case EBML_STOP: return 1; yading@11: default: yading@11: if(ffio_limit(pb, length) != length) yading@11: return AVERROR(EIO); yading@11: return avio_skip(pb,length)<0 ? AVERROR(EIO) : 0; yading@11: } yading@11: if (res == AVERROR_INVALIDDATA) yading@11: av_log(matroska->ctx, AV_LOG_ERROR, "Invalid element\n"); yading@11: else if (res == AVERROR(EIO)) yading@11: av_log(matroska->ctx, AV_LOG_ERROR, "Read error\n"); yading@11: return res; yading@11: } yading@11: yading@11: static void ebml_free(EbmlSyntax *syntax, void *data) yading@11: { yading@11: int i, j; yading@11: for (i=0; syntax[i].id; i++) { yading@11: void *data_off = (char *)data + syntax[i].data_offset; yading@11: switch (syntax[i].type) { yading@11: case EBML_STR: yading@11: case EBML_UTF8: av_freep(data_off); break; yading@11: case EBML_BIN: av_freep(&((EbmlBin *)data_off)->data); break; yading@11: case EBML_NEST: yading@11: if (syntax[i].list_elem_size) { yading@11: EbmlList *list = data_off; yading@11: char *ptr = list->elem; yading@11: for (j=0; jnb_elem; j++, ptr+=syntax[i].list_elem_size) yading@11: ebml_free(syntax[i].def.n, ptr); yading@11: av_free(list->elem); yading@11: } else yading@11: ebml_free(syntax[i].def.n, data_off); yading@11: default: break; yading@11: } yading@11: } yading@11: } yading@11: yading@11: yading@11: /* yading@11: * Autodetecting... yading@11: */ yading@11: static int matroska_probe(AVProbeData *p) yading@11: { yading@11: uint64_t total = 0; yading@11: int len_mask = 0x80, size = 1, n = 1, i; yading@11: yading@11: /* EBML header? */ yading@11: if (AV_RB32(p->buf) != EBML_ID_HEADER) yading@11: return 0; yading@11: yading@11: /* length of header */ yading@11: total = p->buf[4]; yading@11: while (size <= 8 && !(total & len_mask)) { yading@11: size++; yading@11: len_mask >>= 1; yading@11: } yading@11: if (size > 8) yading@11: return 0; yading@11: total &= (len_mask - 1); yading@11: while (n < size) yading@11: total = (total << 8) | p->buf[4 + n++]; yading@11: yading@11: /* Does the probe data contain the whole header? */ yading@11: if (p->buf_size < 4 + size + total) yading@11: return 0; yading@11: yading@11: /* The header should contain a known document type. For now, yading@11: * we don't parse the whole header but simply check for the yading@11: * availability of that array of characters inside the header. yading@11: * Not fully fool-proof, but good enough. */ yading@11: for (i = 0; i < FF_ARRAY_ELEMS(matroska_doctypes); i++) { yading@11: int probelen = strlen(matroska_doctypes[i]); yading@11: if (total < probelen) yading@11: continue; yading@11: for (n = 4+size; n <= 4+size+total-probelen; n++) yading@11: if (!memcmp(p->buf+n, matroska_doctypes[i], probelen)) yading@11: return AVPROBE_SCORE_MAX; yading@11: } yading@11: yading@11: // probably valid EBML header but no recognized doctype yading@11: return AVPROBE_SCORE_MAX/2; yading@11: } yading@11: yading@11: static MatroskaTrack *matroska_find_track_by_num(MatroskaDemuxContext *matroska, yading@11: int num) yading@11: { yading@11: MatroskaTrack *tracks = matroska->tracks.elem; yading@11: int i; yading@11: yading@11: for (i=0; i < matroska->tracks.nb_elem; i++) yading@11: if (tracks[i].num == num) yading@11: return &tracks[i]; yading@11: yading@11: av_log(matroska->ctx, AV_LOG_ERROR, "Invalid track number %d\n", num); yading@11: return NULL; yading@11: } yading@11: yading@11: static int matroska_decode_buffer(uint8_t** buf, int* buf_size, yading@11: MatroskaTrack *track) yading@11: { yading@11: MatroskaTrackEncoding *encodings = track->encodings.elem; yading@11: uint8_t* data = *buf; yading@11: int isize = *buf_size; yading@11: uint8_t* pkt_data = NULL; yading@11: uint8_t av_unused *newpktdata; yading@11: int pkt_size = isize; yading@11: int result = 0; yading@11: int olen; yading@11: yading@11: if (pkt_size >= 10000000U) yading@11: return AVERROR_INVALIDDATA; yading@11: yading@11: switch (encodings[0].compression.algo) { yading@11: case MATROSKA_TRACK_ENCODING_COMP_HEADERSTRIP: { yading@11: int header_size = encodings[0].compression.settings.size; yading@11: uint8_t *header = encodings[0].compression.settings.data; yading@11: yading@11: if (header_size && !header) { yading@11: av_log(NULL, AV_LOG_ERROR, "Compression size but no data in headerstrip\n"); yading@11: return -1; yading@11: } yading@11: yading@11: if (!header_size) yading@11: return 0; yading@11: yading@11: pkt_size = isize + header_size; yading@11: pkt_data = av_malloc(pkt_size); yading@11: if (!pkt_data) yading@11: return AVERROR(ENOMEM); yading@11: yading@11: memcpy(pkt_data, header, header_size); yading@11: memcpy(pkt_data + header_size, data, isize); yading@11: break; yading@11: } yading@11: #if CONFIG_LZO yading@11: case MATROSKA_TRACK_ENCODING_COMP_LZO: yading@11: do { yading@11: olen = pkt_size *= 3; yading@11: newpktdata = av_realloc(pkt_data, pkt_size + AV_LZO_OUTPUT_PADDING); yading@11: if (!newpktdata) { yading@11: result = AVERROR(ENOMEM); yading@11: goto failed; yading@11: } yading@11: pkt_data = newpktdata; yading@11: result = av_lzo1x_decode(pkt_data, &olen, data, &isize); yading@11: } while (result==AV_LZO_OUTPUT_FULL && pkt_size<10000000); yading@11: if (result) { yading@11: result = AVERROR_INVALIDDATA; yading@11: goto failed; yading@11: } yading@11: pkt_size -= olen; yading@11: break; yading@11: #endif yading@11: #if CONFIG_ZLIB yading@11: case MATROSKA_TRACK_ENCODING_COMP_ZLIB: { yading@11: z_stream zstream = {0}; yading@11: if (inflateInit(&zstream) != Z_OK) yading@11: return -1; yading@11: zstream.next_in = data; yading@11: zstream.avail_in = isize; yading@11: do { yading@11: pkt_size *= 3; yading@11: newpktdata = av_realloc(pkt_data, pkt_size); yading@11: if (!newpktdata) { yading@11: inflateEnd(&zstream); yading@11: goto failed; yading@11: } yading@11: pkt_data = newpktdata; yading@11: zstream.avail_out = pkt_size - zstream.total_out; yading@11: zstream.next_out = pkt_data + zstream.total_out; yading@11: if (pkt_data) { yading@11: result = inflate(&zstream, Z_NO_FLUSH); yading@11: } else yading@11: result = Z_MEM_ERROR; yading@11: } while (result==Z_OK && pkt_size<10000000); yading@11: pkt_size = zstream.total_out; yading@11: inflateEnd(&zstream); yading@11: if (result != Z_STREAM_END) { yading@11: if (result == Z_MEM_ERROR) yading@11: result = AVERROR(ENOMEM); yading@11: else yading@11: result = AVERROR_INVALIDDATA; yading@11: goto failed; yading@11: } yading@11: break; yading@11: } yading@11: #endif yading@11: #if CONFIG_BZLIB yading@11: case MATROSKA_TRACK_ENCODING_COMP_BZLIB: { yading@11: bz_stream bzstream = {0}; yading@11: if (BZ2_bzDecompressInit(&bzstream, 0, 0) != BZ_OK) yading@11: return -1; yading@11: bzstream.next_in = data; yading@11: bzstream.avail_in = isize; yading@11: do { yading@11: pkt_size *= 3; yading@11: newpktdata = av_realloc(pkt_data, pkt_size); yading@11: if (!newpktdata) { yading@11: BZ2_bzDecompressEnd(&bzstream); yading@11: goto failed; yading@11: } yading@11: pkt_data = newpktdata; yading@11: bzstream.avail_out = pkt_size - bzstream.total_out_lo32; yading@11: bzstream.next_out = pkt_data + bzstream.total_out_lo32; yading@11: if (pkt_data) { yading@11: result = BZ2_bzDecompress(&bzstream); yading@11: } else yading@11: result = BZ_MEM_ERROR; yading@11: } while (result==BZ_OK && pkt_size<10000000); yading@11: pkt_size = bzstream.total_out_lo32; yading@11: BZ2_bzDecompressEnd(&bzstream); yading@11: if (result != BZ_STREAM_END) { yading@11: if (result == BZ_MEM_ERROR) yading@11: result = AVERROR(ENOMEM); yading@11: else yading@11: result = AVERROR_INVALIDDATA; yading@11: goto failed; yading@11: } yading@11: break; yading@11: } yading@11: #endif yading@11: default: yading@11: return AVERROR_INVALIDDATA; yading@11: } yading@11: yading@11: *buf = pkt_data; yading@11: *buf_size = pkt_size; yading@11: return 0; yading@11: failed: yading@11: av_free(pkt_data); yading@11: return result; yading@11: } yading@11: yading@11: #if FF_API_ASS_SSA yading@11: static void matroska_fix_ass_packet(MatroskaDemuxContext *matroska, yading@11: AVPacket *pkt, uint64_t display_duration) yading@11: { yading@11: AVBufferRef *line; yading@11: char *layer, *ptr = pkt->data, *end = ptr+pkt->size; yading@11: for (; *ptr!=',' && ptrpts + display_duration; yading@11: int sc = matroska->time_scale * pkt->pts / 10000000; yading@11: int ec = matroska->time_scale * end_pts / 10000000; yading@11: int sh, sm, ss, eh, em, es, len; yading@11: sh = sc/360000; sc -= 360000*sh; yading@11: sm = sc/ 6000; sc -= 6000*sm; yading@11: ss = sc/ 100; sc -= 100*ss; yading@11: eh = ec/360000; ec -= 360000*eh; yading@11: em = ec/ 6000; ec -= 6000*em; yading@11: es = ec/ 100; ec -= 100*es; yading@11: *ptr++ = '\0'; yading@11: len = 50 + end-ptr + FF_INPUT_BUFFER_PADDING_SIZE; yading@11: if (!(line = av_buffer_alloc(len))) yading@11: return; yading@11: snprintf(line->data, len,"Dialogue: %s,%d:%02d:%02d.%02d,%d:%02d:%02d.%02d,%s\r\n", yading@11: layer, sh, sm, ss, sc, eh, em, es, ec, ptr); yading@11: av_buffer_unref(&pkt->buf); yading@11: pkt->buf = line; yading@11: pkt->data = line->data; yading@11: pkt->size = strlen(line->data); yading@11: } yading@11: } yading@11: yading@11: static int matroska_merge_packets(AVPacket *out, AVPacket *in) yading@11: { yading@11: int ret = av_grow_packet(out, in->size); yading@11: if (ret < 0) yading@11: return ret; yading@11: yading@11: memcpy(out->data + out->size - in->size, in->data, in->size); yading@11: yading@11: av_free_packet(in); yading@11: av_free(in); yading@11: return 0; yading@11: } yading@11: #endif yading@11: yading@11: static void matroska_convert_tag(AVFormatContext *s, EbmlList *list, yading@11: AVDictionary **metadata, char *prefix) yading@11: { yading@11: MatroskaTag *tags = list->elem; yading@11: char key[1024]; yading@11: int i; yading@11: yading@11: for (i=0; i < list->nb_elem; i++) { yading@11: const char *lang= (tags[i].lang && strcmp(tags[i].lang, "und")) ? tags[i].lang : NULL; yading@11: yading@11: if (!tags[i].name) { yading@11: av_log(s, AV_LOG_WARNING, "Skipping invalid tag with no TagName.\n"); yading@11: continue; yading@11: } yading@11: if (prefix) snprintf(key, sizeof(key), "%s/%s", prefix, tags[i].name); yading@11: else av_strlcpy(key, tags[i].name, sizeof(key)); yading@11: if (tags[i].def || !lang) { yading@11: av_dict_set(metadata, key, tags[i].string, 0); yading@11: if (tags[i].sub.nb_elem) yading@11: matroska_convert_tag(s, &tags[i].sub, metadata, key); yading@11: } yading@11: if (lang) { yading@11: av_strlcat(key, "-", sizeof(key)); yading@11: av_strlcat(key, lang, sizeof(key)); yading@11: av_dict_set(metadata, key, tags[i].string, 0); yading@11: if (tags[i].sub.nb_elem) yading@11: matroska_convert_tag(s, &tags[i].sub, metadata, key); yading@11: } yading@11: } yading@11: ff_metadata_conv(metadata, NULL, ff_mkv_metadata_conv); yading@11: } yading@11: yading@11: static void matroska_convert_tags(AVFormatContext *s) yading@11: { yading@11: MatroskaDemuxContext *matroska = s->priv_data; yading@11: MatroskaTags *tags = matroska->tags.elem; yading@11: int i, j; yading@11: yading@11: for (i=0; i < matroska->tags.nb_elem; i++) { yading@11: if (tags[i].target.attachuid) { yading@11: MatroskaAttachement *attachment = matroska->attachments.elem; yading@11: for (j=0; jattachments.nb_elem; j++) yading@11: if (attachment[j].uid == tags[i].target.attachuid yading@11: && attachment[j].stream) yading@11: matroska_convert_tag(s, &tags[i].tag, yading@11: &attachment[j].stream->metadata, NULL); yading@11: } else if (tags[i].target.chapteruid) { yading@11: MatroskaChapter *chapter = matroska->chapters.elem; yading@11: for (j=0; jchapters.nb_elem; j++) yading@11: if (chapter[j].uid == tags[i].target.chapteruid yading@11: && chapter[j].chapter) yading@11: matroska_convert_tag(s, &tags[i].tag, yading@11: &chapter[j].chapter->metadata, NULL); yading@11: } else if (tags[i].target.trackuid) { yading@11: MatroskaTrack *track = matroska->tracks.elem; yading@11: for (j=0; jtracks.nb_elem; j++) yading@11: if (track[j].uid == tags[i].target.trackuid && track[j].stream) yading@11: matroska_convert_tag(s, &tags[i].tag, yading@11: &track[j].stream->metadata, NULL); yading@11: } else { yading@11: matroska_convert_tag(s, &tags[i].tag, &s->metadata, yading@11: tags[i].target.type); yading@11: } yading@11: } yading@11: } yading@11: yading@11: static int matroska_parse_seekhead_entry(MatroskaDemuxContext *matroska, int idx) yading@11: { yading@11: EbmlList *seekhead_list = &matroska->seekhead; yading@11: MatroskaSeekhead *seekhead = seekhead_list->elem; yading@11: uint32_t level_up = matroska->level_up; yading@11: int64_t before_pos = avio_tell(matroska->ctx->pb); yading@11: uint32_t saved_id = matroska->current_id; yading@11: MatroskaLevel level; yading@11: int64_t offset; yading@11: int ret = 0; yading@11: yading@11: if (idx >= seekhead_list->nb_elem yading@11: || seekhead[idx].id == MATROSKA_ID_SEEKHEAD yading@11: || seekhead[idx].id == MATROSKA_ID_CLUSTER) yading@11: return 0; yading@11: yading@11: /* seek */ yading@11: offset = seekhead[idx].pos + matroska->segment_start; yading@11: if (avio_seek(matroska->ctx->pb, offset, SEEK_SET) == offset) { yading@11: /* We don't want to lose our seekhead level, so we add yading@11: * a dummy. This is a crude hack. */ yading@11: if (matroska->num_levels == EBML_MAX_DEPTH) { yading@11: av_log(matroska->ctx, AV_LOG_INFO, yading@11: "Max EBML element depth (%d) reached, " yading@11: "cannot parse further.\n", EBML_MAX_DEPTH); yading@11: ret = AVERROR_INVALIDDATA; yading@11: } else { yading@11: level.start = 0; yading@11: level.length = (uint64_t)-1; yading@11: matroska->levels[matroska->num_levels] = level; yading@11: matroska->num_levels++; yading@11: matroska->current_id = 0; yading@11: yading@11: ret = ebml_parse(matroska, matroska_segment, matroska); yading@11: yading@11: /* remove dummy level */ yading@11: while (matroska->num_levels) { yading@11: uint64_t length = matroska->levels[--matroska->num_levels].length; yading@11: if (length == (uint64_t)-1) yading@11: break; yading@11: } yading@11: } yading@11: } yading@11: /* seek back */ yading@11: avio_seek(matroska->ctx->pb, before_pos, SEEK_SET); yading@11: matroska->level_up = level_up; yading@11: matroska->current_id = saved_id; yading@11: yading@11: return ret; yading@11: } yading@11: yading@11: static void matroska_execute_seekhead(MatroskaDemuxContext *matroska) yading@11: { yading@11: EbmlList *seekhead_list = &matroska->seekhead; yading@11: int64_t before_pos = avio_tell(matroska->ctx->pb); yading@11: int i; yading@11: yading@11: // we should not do any seeking in the streaming case yading@11: if (!matroska->ctx->pb->seekable || yading@11: (matroska->ctx->flags & AVFMT_FLAG_IGNIDX)) yading@11: return; yading@11: yading@11: for (i = 0; i < seekhead_list->nb_elem; i++) { yading@11: MatroskaSeekhead *seekhead = seekhead_list->elem; yading@11: if (seekhead[i].pos <= before_pos) yading@11: continue; yading@11: yading@11: // defer cues parsing until we actually need cue data. yading@11: if (seekhead[i].id == MATROSKA_ID_CUES) { yading@11: matroska->cues_parsing_deferred = 1; yading@11: continue; yading@11: } yading@11: yading@11: if (matroska_parse_seekhead_entry(matroska, i) < 0) { yading@11: // mark index as broken yading@11: matroska->cues_parsing_deferred = -1; yading@11: break; yading@11: } yading@11: } yading@11: } yading@11: yading@11: static void matroska_add_index_entries(MatroskaDemuxContext *matroska) { yading@11: EbmlList *index_list; yading@11: MatroskaIndex *index; yading@11: int index_scale = 1; yading@11: int i, j; yading@11: yading@11: index_list = &matroska->index; yading@11: index = index_list->elem; yading@11: if (index_list->nb_elem yading@11: && index[0].time > 1E14/matroska->time_scale) { yading@11: av_log(matroska->ctx, AV_LOG_WARNING, "Working around broken index.\n"); yading@11: index_scale = matroska->time_scale; yading@11: } yading@11: for (i = 0; i < index_list->nb_elem; i++) { yading@11: EbmlList *pos_list = &index[i].pos; yading@11: MatroskaIndexPos *pos = pos_list->elem; yading@11: for (j = 0; j < pos_list->nb_elem; j++) { yading@11: MatroskaTrack *track = matroska_find_track_by_num(matroska, pos[j].track); yading@11: if (track && track->stream) yading@11: av_add_index_entry(track->stream, yading@11: pos[j].pos + matroska->segment_start, yading@11: index[i].time/index_scale, 0, 0, yading@11: AVINDEX_KEYFRAME); yading@11: } yading@11: } yading@11: } yading@11: yading@11: static void matroska_parse_cues(MatroskaDemuxContext *matroska) { yading@11: EbmlList *seekhead_list = &matroska->seekhead; yading@11: MatroskaSeekhead *seekhead = seekhead_list->elem; yading@11: int i; yading@11: yading@11: for (i = 0; i < seekhead_list->nb_elem; i++) yading@11: if (seekhead[i].id == MATROSKA_ID_CUES) yading@11: break; yading@11: av_assert1(i <= seekhead_list->nb_elem); yading@11: yading@11: if (matroska_parse_seekhead_entry(matroska, i) < 0) yading@11: matroska->cues_parsing_deferred = -1; yading@11: matroska_add_index_entries(matroska); yading@11: } yading@11: yading@11: static int matroska_aac_profile(char *codec_id) yading@11: { yading@11: static const char * const aac_profiles[] = { "MAIN", "LC", "SSR" }; yading@11: int profile; yading@11: yading@11: for (profile=0; profilepriv_data; yading@11: EbmlList *attachements_list = &matroska->attachments; yading@11: MatroskaAttachement *attachements; yading@11: EbmlList *chapters_list = &matroska->chapters; yading@11: MatroskaChapter *chapters; yading@11: MatroskaTrack *tracks; yading@11: uint64_t max_start = 0; yading@11: int64_t pos; yading@11: Ebml ebml = { 0 }; yading@11: AVStream *st; yading@11: int i, j, k, res; yading@11: yading@11: matroska->ctx = s; yading@11: yading@11: /* First read the EBML header. */ yading@11: if (ebml_parse(matroska, ebml_syntax, &ebml) yading@11: || ebml.version > EBML_VERSION || ebml.max_size > sizeof(uint64_t) yading@11: || ebml.id_length > sizeof(uint32_t) || ebml.doctype_version > 3 || !ebml.doctype) { yading@11: av_log(matroska->ctx, AV_LOG_ERROR, yading@11: "EBML header using unsupported features\n" yading@11: "(EBML version %"PRIu64", doctype %s, doc version %"PRIu64")\n", yading@11: ebml.version, ebml.doctype, ebml.doctype_version); yading@11: ebml_free(ebml_syntax, &ebml); yading@11: return AVERROR_PATCHWELCOME; yading@11: } else if (ebml.doctype_version == 3) { yading@11: av_log(matroska->ctx, AV_LOG_WARNING, yading@11: "EBML header using unsupported features\n" yading@11: "(EBML version %"PRIu64", doctype %s, doc version %"PRIu64")\n", yading@11: ebml.version, ebml.doctype, ebml.doctype_version); yading@11: } yading@11: for (i = 0; i < FF_ARRAY_ELEMS(matroska_doctypes); i++) yading@11: if (!strcmp(ebml.doctype, matroska_doctypes[i])) yading@11: break; yading@11: if (i >= FF_ARRAY_ELEMS(matroska_doctypes)) { yading@11: av_log(s, AV_LOG_WARNING, "Unknown EBML doctype '%s'\n", ebml.doctype); yading@11: if (matroska->ctx->error_recognition & AV_EF_EXPLODE) { yading@11: ebml_free(ebml_syntax, &ebml); yading@11: return AVERROR_INVALIDDATA; yading@11: } yading@11: } yading@11: ebml_free(ebml_syntax, &ebml); yading@11: yading@11: /* The next thing is a segment. */ yading@11: pos = avio_tell(matroska->ctx->pb); yading@11: res = ebml_parse(matroska, matroska_segments, matroska); yading@11: // try resyncing until we find a EBML_STOP type element. yading@11: while (res != 1) { yading@11: res = matroska_resync(matroska, pos); yading@11: if (res < 0) yading@11: return res; yading@11: pos = avio_tell(matroska->ctx->pb); yading@11: res = ebml_parse(matroska, matroska_segment, matroska); yading@11: } yading@11: matroska_execute_seekhead(matroska); yading@11: yading@11: if (!matroska->time_scale) yading@11: matroska->time_scale = 1000000; yading@11: if (matroska->duration) yading@11: matroska->ctx->duration = matroska->duration * matroska->time_scale yading@11: * 1000 / AV_TIME_BASE; yading@11: av_dict_set(&s->metadata, "title", matroska->title, 0); yading@11: yading@11: if (matroska->date_utc.size == 8) yading@11: matroska_metadata_creation_time(&s->metadata, AV_RB64(matroska->date_utc.data)); yading@11: yading@11: tracks = matroska->tracks.elem; yading@11: for (i=0; i < matroska->tracks.nb_elem; i++) { yading@11: MatroskaTrack *track = &tracks[i]; yading@11: enum AVCodecID codec_id = AV_CODEC_ID_NONE; yading@11: EbmlList *encodings_list = &track->encodings; yading@11: MatroskaTrackEncoding *encodings = encodings_list->elem; yading@11: uint8_t *extradata = NULL; yading@11: int extradata_size = 0; yading@11: int extradata_offset = 0; yading@11: uint32_t fourcc = 0; yading@11: AVIOContext b; yading@11: char* key_id_base64 = NULL; yading@11: yading@11: /* Apply some sanity checks. */ yading@11: if (track->type != MATROSKA_TRACK_TYPE_VIDEO && yading@11: track->type != MATROSKA_TRACK_TYPE_AUDIO && yading@11: track->type != MATROSKA_TRACK_TYPE_SUBTITLE) { yading@11: av_log(matroska->ctx, AV_LOG_INFO, yading@11: "Unknown or unsupported track type %"PRIu64"\n", yading@11: track->type); yading@11: continue; yading@11: } yading@11: if (track->codec_id == NULL) yading@11: continue; yading@11: yading@11: if (track->type == MATROSKA_TRACK_TYPE_VIDEO) { yading@11: if (!track->default_duration && track->video.frame_rate > 0) yading@11: track->default_duration = 1000000000/track->video.frame_rate; yading@11: if (track->video.display_width == -1) yading@11: track->video.display_width = track->video.pixel_width; yading@11: if (track->video.display_height == -1) yading@11: track->video.display_height = track->video.pixel_height; yading@11: if (track->video.color_space.size == 4) yading@11: fourcc = AV_RL32(track->video.color_space.data); yading@11: } else if (track->type == MATROSKA_TRACK_TYPE_AUDIO) { yading@11: if (!track->audio.out_samplerate) yading@11: track->audio.out_samplerate = track->audio.samplerate; yading@11: } yading@11: if (encodings_list->nb_elem > 1) { yading@11: av_log(matroska->ctx, AV_LOG_ERROR, yading@11: "Multiple combined encodings not supported"); yading@11: } else if (encodings_list->nb_elem == 1) { yading@11: if (encodings[0].type) { yading@11: if (encodings[0].encryption.key_id.size > 0) { yading@11: /* Save the encryption key id to be stored later as a yading@11: metadata tag. */ yading@11: const int b64_size = AV_BASE64_SIZE(encodings[0].encryption.key_id.size); yading@11: key_id_base64 = av_malloc(b64_size); yading@11: if (key_id_base64 == NULL) yading@11: return AVERROR(ENOMEM); yading@11: yading@11: av_base64_encode(key_id_base64, b64_size, yading@11: encodings[0].encryption.key_id.data, yading@11: encodings[0].encryption.key_id.size); yading@11: } else { yading@11: encodings[0].scope = 0; yading@11: av_log(matroska->ctx, AV_LOG_ERROR, yading@11: "Unsupported encoding type"); yading@11: } yading@11: } else if ( yading@11: #if CONFIG_ZLIB yading@11: encodings[0].compression.algo != MATROSKA_TRACK_ENCODING_COMP_ZLIB && yading@11: #endif yading@11: #if CONFIG_BZLIB yading@11: encodings[0].compression.algo != MATROSKA_TRACK_ENCODING_COMP_BZLIB && yading@11: #endif yading@11: #if CONFIG_LZO yading@11: encodings[0].compression.algo != MATROSKA_TRACK_ENCODING_COMP_LZO && yading@11: #endif yading@11: encodings[0].compression.algo != MATROSKA_TRACK_ENCODING_COMP_HEADERSTRIP) { yading@11: encodings[0].scope = 0; yading@11: av_log(matroska->ctx, AV_LOG_ERROR, yading@11: "Unsupported encoding type"); yading@11: } else if (track->codec_priv.size && encodings[0].scope&2) { yading@11: uint8_t *codec_priv = track->codec_priv.data; yading@11: int ret = matroska_decode_buffer(&track->codec_priv.data, yading@11: &track->codec_priv.size, yading@11: track); yading@11: if (ret < 0) { yading@11: track->codec_priv.data = NULL; yading@11: track->codec_priv.size = 0; yading@11: av_log(matroska->ctx, AV_LOG_ERROR, yading@11: "Failed to decode codec private data\n"); yading@11: } yading@11: yading@11: if (codec_priv != track->codec_priv.data) yading@11: av_free(codec_priv); yading@11: } yading@11: } yading@11: yading@11: for(j=0; ff_mkv_codec_tags[j].id != AV_CODEC_ID_NONE; j++){ yading@11: if(!strncmp(ff_mkv_codec_tags[j].str, track->codec_id, yading@11: strlen(ff_mkv_codec_tags[j].str))){ yading@11: codec_id= ff_mkv_codec_tags[j].id; yading@11: break; yading@11: } yading@11: } yading@11: yading@11: st = track->stream = avformat_new_stream(s, NULL); yading@11: if (st == NULL) { yading@11: av_free(key_id_base64); yading@11: return AVERROR(ENOMEM); yading@11: } yading@11: yading@11: if (key_id_base64) { yading@11: /* export encryption key id as base64 metadata tag */ yading@11: av_dict_set(&st->metadata, "enc_key_id", key_id_base64, 0); yading@11: av_freep(&key_id_base64); yading@11: } yading@11: yading@11: if (!strcmp(track->codec_id, "V_MS/VFW/FOURCC") yading@11: && track->codec_priv.size >= 40 yading@11: && track->codec_priv.data != NULL) { yading@11: track->ms_compat = 1; yading@11: fourcc = AV_RL32(track->codec_priv.data + 16); yading@11: codec_id = ff_codec_get_id(ff_codec_bmp_tags, fourcc); yading@11: extradata_offset = 40; yading@11: } else if (!strcmp(track->codec_id, "A_MS/ACM") yading@11: && track->codec_priv.size >= 14 yading@11: && track->codec_priv.data != NULL) { yading@11: int ret; yading@11: ffio_init_context(&b, track->codec_priv.data, track->codec_priv.size, yading@11: 0, NULL, NULL, NULL, NULL); yading@11: ret = ff_get_wav_header(&b, st->codec, track->codec_priv.size); yading@11: if (ret < 0) yading@11: return ret; yading@11: codec_id = st->codec->codec_id; yading@11: extradata_offset = FFMIN(track->codec_priv.size, 18); yading@11: } else if (!strcmp(track->codec_id, "V_QUICKTIME") yading@11: && (track->codec_priv.size >= 86) yading@11: && (track->codec_priv.data != NULL)) { yading@11: fourcc = AV_RL32(track->codec_priv.data); yading@11: codec_id = ff_codec_get_id(ff_codec_movvideo_tags, fourcc); yading@11: } else if (codec_id == AV_CODEC_ID_ALAC && track->codec_priv.size && track->codec_priv.size < INT_MAX - 12 - FF_INPUT_BUFFER_PADDING_SIZE) { yading@11: /* Only ALAC's magic cookie is stored in Matroska's track headers. yading@11: Create the "atom size", "tag", and "tag version" fields the yading@11: decoder expects manually. */ yading@11: extradata_size = 12 + track->codec_priv.size; yading@11: extradata = av_mallocz(extradata_size + FF_INPUT_BUFFER_PADDING_SIZE); yading@11: if (extradata == NULL) yading@11: return AVERROR(ENOMEM); yading@11: AV_WB32(extradata, extradata_size); yading@11: memcpy(&extradata[4], "alac", 4); yading@11: AV_WB32(&extradata[8], 0); yading@11: memcpy(&extradata[12], track->codec_priv.data, track->codec_priv.size); yading@11: } else if (codec_id == AV_CODEC_ID_PCM_S16BE) { yading@11: switch (track->audio.bitdepth) { yading@11: case 8: codec_id = AV_CODEC_ID_PCM_U8; break; yading@11: case 24: codec_id = AV_CODEC_ID_PCM_S24BE; break; yading@11: case 32: codec_id = AV_CODEC_ID_PCM_S32BE; break; yading@11: } yading@11: } else if (codec_id == AV_CODEC_ID_PCM_S16LE) { yading@11: switch (track->audio.bitdepth) { yading@11: case 8: codec_id = AV_CODEC_ID_PCM_U8; break; yading@11: case 24: codec_id = AV_CODEC_ID_PCM_S24LE; break; yading@11: case 32: codec_id = AV_CODEC_ID_PCM_S32LE; break; yading@11: } yading@11: } else if (codec_id==AV_CODEC_ID_PCM_F32LE && track->audio.bitdepth==64) { yading@11: codec_id = AV_CODEC_ID_PCM_F64LE; yading@11: } else if (codec_id == AV_CODEC_ID_AAC && !track->codec_priv.size) { yading@11: int profile = matroska_aac_profile(track->codec_id); yading@11: int sri = matroska_aac_sri(track->audio.samplerate); yading@11: extradata = av_mallocz(5 + FF_INPUT_BUFFER_PADDING_SIZE); yading@11: if (extradata == NULL) yading@11: return AVERROR(ENOMEM); yading@11: extradata[0] = (profile << 3) | ((sri&0x0E) >> 1); yading@11: extradata[1] = ((sri&0x01) << 7) | (track->audio.channels<<3); yading@11: if (strstr(track->codec_id, "SBR")) { yading@11: sri = matroska_aac_sri(track->audio.out_samplerate); yading@11: extradata[2] = 0x56; yading@11: extradata[3] = 0xE5; yading@11: extradata[4] = 0x80 | (sri<<3); yading@11: extradata_size = 5; yading@11: } else yading@11: extradata_size = 2; yading@11: } else if (codec_id == AV_CODEC_ID_TTA) { yading@11: extradata_size = 30; yading@11: extradata = av_mallocz(extradata_size + FF_INPUT_BUFFER_PADDING_SIZE); yading@11: if (extradata == NULL) yading@11: return AVERROR(ENOMEM); yading@11: ffio_init_context(&b, extradata, extradata_size, 1, yading@11: NULL, NULL, NULL, NULL); yading@11: avio_write(&b, "TTA1", 4); yading@11: avio_wl16(&b, 1); yading@11: avio_wl16(&b, track->audio.channels); yading@11: avio_wl16(&b, track->audio.bitdepth); yading@11: avio_wl32(&b, track->audio.out_samplerate); yading@11: avio_wl32(&b, matroska->ctx->duration * track->audio.out_samplerate); yading@11: } else if (codec_id == AV_CODEC_ID_RV10 || codec_id == AV_CODEC_ID_RV20 || yading@11: codec_id == AV_CODEC_ID_RV30 || codec_id == AV_CODEC_ID_RV40) { yading@11: extradata_offset = 26; yading@11: } else if (codec_id == AV_CODEC_ID_RA_144) { yading@11: track->audio.out_samplerate = 8000; yading@11: track->audio.channels = 1; yading@11: } else if ((codec_id == AV_CODEC_ID_RA_288 || codec_id == AV_CODEC_ID_COOK || yading@11: codec_id == AV_CODEC_ID_ATRAC3 || codec_id == AV_CODEC_ID_SIPR) yading@11: && track->codec_priv.data) { yading@11: int flavor; yading@11: yading@11: ffio_init_context(&b, track->codec_priv.data,track->codec_priv.size, yading@11: 0, NULL, NULL, NULL, NULL); yading@11: avio_skip(&b, 22); yading@11: flavor = avio_rb16(&b); yading@11: track->audio.coded_framesize = avio_rb32(&b); yading@11: avio_skip(&b, 12); yading@11: track->audio.sub_packet_h = avio_rb16(&b); yading@11: track->audio.frame_size = avio_rb16(&b); yading@11: track->audio.sub_packet_size = avio_rb16(&b); yading@11: track->audio.buf = av_malloc(track->audio.frame_size * track->audio.sub_packet_h); yading@11: if (codec_id == AV_CODEC_ID_RA_288) { yading@11: st->codec->block_align = track->audio.coded_framesize; yading@11: track->codec_priv.size = 0; yading@11: } else { yading@11: if (codec_id == AV_CODEC_ID_SIPR && flavor < 4) { yading@11: const int sipr_bit_rate[4] = { 6504, 8496, 5000, 16000 }; yading@11: track->audio.sub_packet_size = ff_sipr_subpk_size[flavor]; yading@11: st->codec->bit_rate = sipr_bit_rate[flavor]; yading@11: } yading@11: st->codec->block_align = track->audio.sub_packet_size; yading@11: extradata_offset = 78; yading@11: } yading@11: } yading@11: track->codec_priv.size -= extradata_offset; yading@11: yading@11: if (codec_id == AV_CODEC_ID_NONE) yading@11: av_log(matroska->ctx, AV_LOG_INFO, yading@11: "Unknown/unsupported AVCodecID %s.\n", track->codec_id); yading@11: yading@11: if (track->time_scale < 0.01) yading@11: track->time_scale = 1.0; yading@11: avpriv_set_pts_info(st, 64, matroska->time_scale*track->time_scale, 1000*1000*1000); /* 64 bit pts in ns */ yading@11: yading@11: st->codec->codec_id = codec_id; yading@11: st->start_time = 0; yading@11: if (strcmp(track->language, "und")) yading@11: av_dict_set(&st->metadata, "language", track->language, 0); yading@11: av_dict_set(&st->metadata, "title", track->name, 0); yading@11: yading@11: if (track->flag_default) yading@11: st->disposition |= AV_DISPOSITION_DEFAULT; yading@11: if (track->flag_forced) yading@11: st->disposition |= AV_DISPOSITION_FORCED; yading@11: yading@11: if (!st->codec->extradata) { yading@11: if(extradata){ yading@11: st->codec->extradata = extradata; yading@11: st->codec->extradata_size = extradata_size; yading@11: } else if(track->codec_priv.data && track->codec_priv.size > 0){ yading@11: st->codec->extradata = av_mallocz(track->codec_priv.size + yading@11: FF_INPUT_BUFFER_PADDING_SIZE); yading@11: if(st->codec->extradata == NULL) yading@11: return AVERROR(ENOMEM); yading@11: st->codec->extradata_size = track->codec_priv.size; yading@11: memcpy(st->codec->extradata, yading@11: track->codec_priv.data + extradata_offset, yading@11: track->codec_priv.size); yading@11: } yading@11: } yading@11: yading@11: if (track->type == MATROSKA_TRACK_TYPE_VIDEO) { yading@11: MatroskaTrackPlane *planes = track->operation.combine_planes.elem; yading@11: yading@11: st->codec->codec_type = AVMEDIA_TYPE_VIDEO; yading@11: st->codec->codec_tag = fourcc; yading@11: st->codec->width = track->video.pixel_width; yading@11: st->codec->height = track->video.pixel_height; yading@11: av_reduce(&st->sample_aspect_ratio.num, yading@11: &st->sample_aspect_ratio.den, yading@11: st->codec->height * track->video.display_width, yading@11: st->codec-> width * track->video.display_height, yading@11: 255); yading@11: st->need_parsing = AVSTREAM_PARSE_HEADERS; yading@11: if (track->default_duration) { yading@11: av_reduce(&st->avg_frame_rate.num, &st->avg_frame_rate.den, yading@11: 1000000000, track->default_duration, 30000); yading@11: #if FF_API_R_FRAME_RATE yading@11: st->r_frame_rate = st->avg_frame_rate; yading@11: #endif yading@11: } yading@11: yading@11: /* export stereo mode flag as metadata tag */ yading@11: if (track->video.stereo_mode && track->video.stereo_mode < MATROSKA_VIDEO_STEREO_MODE_COUNT) yading@11: av_dict_set(&st->metadata, "stereo_mode", ff_matroska_video_stereo_mode[track->video.stereo_mode], 0); yading@11: yading@11: /* export alpha mode flag as metadata tag */ yading@11: if (track->video.alpha_mode) yading@11: av_dict_set(&st->metadata, "alpha_mode", "1", 0); yading@11: yading@11: /* if we have virtual track, mark the real tracks */ yading@11: for (j=0; j < track->operation.combine_planes.nb_elem; j++) { yading@11: char buf[32]; yading@11: if (planes[j].type >= MATROSKA_VIDEO_STEREO_PLANE_COUNT) yading@11: continue; yading@11: snprintf(buf, sizeof(buf), "%s_%d", yading@11: ff_matroska_video_stereo_plane[planes[j].type], i); yading@11: for (k=0; k < matroska->tracks.nb_elem; k++) yading@11: if (planes[j].uid == tracks[k].uid) { yading@11: av_dict_set(&s->streams[k]->metadata, yading@11: "stereo_mode", buf, 0); yading@11: break; yading@11: } yading@11: } yading@11: } else if (track->type == MATROSKA_TRACK_TYPE_AUDIO) { yading@11: st->codec->codec_type = AVMEDIA_TYPE_AUDIO; yading@11: st->codec->sample_rate = track->audio.out_samplerate; yading@11: st->codec->channels = track->audio.channels; yading@11: st->codec->bits_per_coded_sample = track->audio.bitdepth; yading@11: if (st->codec->codec_id != AV_CODEC_ID_AAC) yading@11: st->need_parsing = AVSTREAM_PARSE_HEADERS; yading@11: } else if (track->type == MATROSKA_TRACK_TYPE_SUBTITLE) { yading@11: st->codec->codec_type = AVMEDIA_TYPE_SUBTITLE; yading@11: #if FF_API_ASS_SSA yading@11: if (st->codec->codec_id == AV_CODEC_ID_SSA || yading@11: st->codec->codec_id == AV_CODEC_ID_ASS) yading@11: #else yading@11: if (st->codec->codec_id == AV_CODEC_ID_ASS) yading@11: #endif yading@11: matroska->contains_ssa = 1; yading@11: } yading@11: } yading@11: yading@11: attachements = attachements_list->elem; yading@11: for (j=0; jnb_elem; j++) { yading@11: if (!(attachements[j].filename && attachements[j].mime && yading@11: attachements[j].bin.data && attachements[j].bin.size > 0)) { yading@11: av_log(matroska->ctx, AV_LOG_ERROR, "incomplete attachment\n"); yading@11: } else { yading@11: AVStream *st = avformat_new_stream(s, NULL); yading@11: if (st == NULL) yading@11: break; yading@11: av_dict_set(&st->metadata, "filename",attachements[j].filename, 0); yading@11: av_dict_set(&st->metadata, "mimetype", attachements[j].mime, 0); yading@11: st->codec->codec_id = AV_CODEC_ID_NONE; yading@11: st->codec->codec_type = AVMEDIA_TYPE_ATTACHMENT; yading@11: st->codec->extradata = av_malloc(attachements[j].bin.size + FF_INPUT_BUFFER_PADDING_SIZE); yading@11: if(st->codec->extradata == NULL) yading@11: break; yading@11: st->codec->extradata_size = attachements[j].bin.size; yading@11: memcpy(st->codec->extradata, attachements[j].bin.data, attachements[j].bin.size); yading@11: yading@11: for (i=0; ff_mkv_mime_tags[i].id != AV_CODEC_ID_NONE; i++) { yading@11: if (!strncmp(ff_mkv_mime_tags[i].str, attachements[j].mime, yading@11: strlen(ff_mkv_mime_tags[i].str))) { yading@11: st->codec->codec_id = ff_mkv_mime_tags[i].id; yading@11: break; yading@11: } yading@11: } yading@11: attachements[j].stream = st; yading@11: } yading@11: } yading@11: yading@11: chapters = chapters_list->elem; yading@11: for (i=0; inb_elem; i++) yading@11: if (chapters[i].start != AV_NOPTS_VALUE && chapters[i].uid yading@11: && (max_start==0 || chapters[i].start > max_start)) { yading@11: chapters[i].chapter = yading@11: avpriv_new_chapter(s, chapters[i].uid, (AVRational){1, 1000000000}, yading@11: chapters[i].start, chapters[i].end, yading@11: chapters[i].title); yading@11: av_dict_set(&chapters[i].chapter->metadata, yading@11: "title", chapters[i].title, 0); yading@11: max_start = chapters[i].start; yading@11: } yading@11: yading@11: matroska_add_index_entries(matroska); yading@11: yading@11: matroska_convert_tags(s); yading@11: yading@11: return 0; yading@11: } yading@11: yading@11: /* yading@11: * Put one packet in an application-supplied AVPacket struct. yading@11: * Returns 0 on success or -1 on failure. yading@11: */ yading@11: static int matroska_deliver_packet(MatroskaDemuxContext *matroska, yading@11: AVPacket *pkt) yading@11: { yading@11: if (matroska->num_packets > 0) { yading@11: memcpy(pkt, matroska->packets[0], sizeof(AVPacket)); yading@11: av_free(matroska->packets[0]); yading@11: if (matroska->num_packets > 1) { yading@11: void *newpackets; yading@11: memmove(&matroska->packets[0], &matroska->packets[1], yading@11: (matroska->num_packets - 1) * sizeof(AVPacket *)); yading@11: newpackets = av_realloc(matroska->packets, yading@11: (matroska->num_packets - 1) * sizeof(AVPacket *)); yading@11: if (newpackets) yading@11: matroska->packets = newpackets; yading@11: } else { yading@11: av_freep(&matroska->packets); yading@11: matroska->prev_pkt = NULL; yading@11: } yading@11: matroska->num_packets--; yading@11: return 0; yading@11: } yading@11: yading@11: return -1; yading@11: } yading@11: yading@11: /* yading@11: * Free all packets in our internal queue. yading@11: */ yading@11: static void matroska_clear_queue(MatroskaDemuxContext *matroska) yading@11: { yading@11: matroska->prev_pkt = NULL; yading@11: if (matroska->packets) { yading@11: int n; yading@11: for (n = 0; n < matroska->num_packets; n++) { yading@11: av_free_packet(matroska->packets[n]); yading@11: av_free(matroska->packets[n]); yading@11: } yading@11: av_freep(&matroska->packets); yading@11: matroska->num_packets = 0; yading@11: } yading@11: } yading@11: yading@11: static int matroska_parse_laces(MatroskaDemuxContext *matroska, uint8_t **buf, yading@11: int* buf_size, int type, yading@11: uint32_t **lace_buf, int *laces) yading@11: { yading@11: int res = 0, n, size = *buf_size; yading@11: uint8_t *data = *buf; yading@11: uint32_t *lace_size; yading@11: yading@11: if (!type) { yading@11: *laces = 1; yading@11: *lace_buf = av_mallocz(sizeof(int)); yading@11: if (!*lace_buf) yading@11: return AVERROR(ENOMEM); yading@11: yading@11: *lace_buf[0] = size; yading@11: return 0; yading@11: } yading@11: yading@11: av_assert0(size > 0); yading@11: *laces = *data + 1; yading@11: data += 1; yading@11: size -= 1; yading@11: lace_size = av_mallocz(*laces * sizeof(int)); yading@11: if (!lace_size) yading@11: return AVERROR(ENOMEM); yading@11: yading@11: switch (type) { yading@11: case 0x1: /* Xiph lacing */ { yading@11: uint8_t temp; yading@11: uint32_t total = 0; yading@11: for (n = 0; res == 0 && n < *laces - 1; n++) { yading@11: while (1) { yading@11: if (size <= total) { yading@11: res = AVERROR_INVALIDDATA; yading@11: break; yading@11: } yading@11: temp = *data; yading@11: total += temp; yading@11: lace_size[n] += temp; yading@11: data += 1; yading@11: size -= 1; yading@11: if (temp != 0xff) yading@11: break; yading@11: } yading@11: } yading@11: if (size <= total) { yading@11: res = AVERROR_INVALIDDATA; yading@11: break; yading@11: } yading@11: yading@11: lace_size[n] = size - total; yading@11: break; yading@11: } yading@11: yading@11: case 0x2: /* fixed-size lacing */ yading@11: if (size % (*laces)) { yading@11: res = AVERROR_INVALIDDATA; yading@11: break; yading@11: } yading@11: for (n = 0; n < *laces; n++) yading@11: lace_size[n] = size / *laces; yading@11: break; yading@11: yading@11: case 0x3: /* EBML lacing */ { yading@11: uint64_t num; yading@11: uint64_t total; yading@11: n = matroska_ebmlnum_uint(matroska, data, size, &num); yading@11: if (n < 0 || num > INT_MAX) { yading@11: av_log(matroska->ctx, AV_LOG_INFO, yading@11: "EBML block data error\n"); yading@11: res = n<0 ? n : AVERROR_INVALIDDATA; yading@11: break; yading@11: } yading@11: data += n; yading@11: size -= n; yading@11: total = lace_size[0] = num; yading@11: for (n = 1; res == 0 && n < *laces - 1; n++) { yading@11: int64_t snum; yading@11: int r; yading@11: r = matroska_ebmlnum_sint(matroska, data, size, &snum); yading@11: if (r < 0 || lace_size[n - 1] + snum > (uint64_t)INT_MAX) { yading@11: av_log(matroska->ctx, AV_LOG_INFO, yading@11: "EBML block data error\n"); yading@11: res = r<0 ? r : AVERROR_INVALIDDATA; yading@11: break; yading@11: } yading@11: data += r; yading@11: size -= r; yading@11: lace_size[n] = lace_size[n - 1] + snum; yading@11: total += lace_size[n]; yading@11: } yading@11: if (size <= total) { yading@11: res = AVERROR_INVALIDDATA; yading@11: break; yading@11: } yading@11: lace_size[*laces - 1] = size - total; yading@11: break; yading@11: } yading@11: } yading@11: yading@11: *buf = data; yading@11: *lace_buf = lace_size; yading@11: *buf_size = size; yading@11: yading@11: return res; yading@11: } yading@11: yading@11: static int matroska_parse_rm_audio(MatroskaDemuxContext *matroska, yading@11: MatroskaTrack *track, yading@11: AVStream *st, yading@11: uint8_t *data, int size, yading@11: uint64_t timecode, yading@11: int64_t pos) yading@11: { yading@11: int a = st->codec->block_align; yading@11: int sps = track->audio.sub_packet_size; yading@11: int cfs = track->audio.coded_framesize; yading@11: int h = track->audio.sub_packet_h; yading@11: int y = track->audio.sub_packet_cnt; yading@11: int w = track->audio.frame_size; yading@11: int x; yading@11: yading@11: if (!track->audio.pkt_cnt) { yading@11: if (track->audio.sub_packet_cnt == 0) yading@11: track->audio.buf_timecode = timecode; yading@11: if (st->codec->codec_id == AV_CODEC_ID_RA_288) { yading@11: if (size < cfs * h / 2) { yading@11: av_log(matroska->ctx, AV_LOG_ERROR, yading@11: "Corrupt int4 RM-style audio packet size\n"); yading@11: return AVERROR_INVALIDDATA; yading@11: } yading@11: for (x=0; xaudio.buf+x*2*w+y*cfs, yading@11: data+x*cfs, cfs); yading@11: } else if (st->codec->codec_id == AV_CODEC_ID_SIPR) { yading@11: if (size < w) { yading@11: av_log(matroska->ctx, AV_LOG_ERROR, yading@11: "Corrupt sipr RM-style audio packet size\n"); yading@11: return AVERROR_INVALIDDATA; yading@11: } yading@11: memcpy(track->audio.buf + y*w, data, w); yading@11: } else { yading@11: if (size < sps * w / sps || h<=0) { yading@11: av_log(matroska->ctx, AV_LOG_ERROR, yading@11: "Corrupt generic RM-style audio packet size\n"); yading@11: return AVERROR_INVALIDDATA; yading@11: } yading@11: for (x=0; xaudio.buf+sps*(h*x+((h+1)/2)*(y&1)+(y>>1)), data+x*sps, sps); yading@11: } yading@11: yading@11: if (++track->audio.sub_packet_cnt >= h) { yading@11: if (st->codec->codec_id == AV_CODEC_ID_SIPR) yading@11: ff_rm_reorder_sipr_data(track->audio.buf, h, w); yading@11: track->audio.sub_packet_cnt = 0; yading@11: track->audio.pkt_cnt = h*w / a; yading@11: } yading@11: } yading@11: yading@11: while (track->audio.pkt_cnt) { yading@11: AVPacket *pkt = NULL; yading@11: if (!(pkt = av_mallocz(sizeof(AVPacket))) || av_new_packet(pkt, a) < 0){ yading@11: av_free(pkt); yading@11: return AVERROR(ENOMEM); yading@11: } yading@11: memcpy(pkt->data, track->audio.buf yading@11: + a * (h*w / a - track->audio.pkt_cnt--), a); yading@11: pkt->pts = track->audio.buf_timecode; yading@11: track->audio.buf_timecode = AV_NOPTS_VALUE; yading@11: pkt->pos = pos; yading@11: pkt->stream_index = st->index; yading@11: dynarray_add(&matroska->packets,&matroska->num_packets,pkt); yading@11: } yading@11: yading@11: return 0; yading@11: } yading@11: static int matroska_parse_frame(MatroskaDemuxContext *matroska, yading@11: MatroskaTrack *track, yading@11: AVStream *st, yading@11: uint8_t *data, int pkt_size, yading@11: uint64_t timecode, uint64_t lace_duration, yading@11: int64_t pos, int is_keyframe, yading@11: uint8_t *additional, uint64_t additional_id, int additional_size) yading@11: { yading@11: MatroskaTrackEncoding *encodings = track->encodings.elem; yading@11: uint8_t *pkt_data = data; yading@11: int offset = 0, res; yading@11: AVPacket *pkt; yading@11: yading@11: if (encodings && !encodings->type && encodings->scope & 1) { yading@11: res = matroska_decode_buffer(&pkt_data, &pkt_size, track); yading@11: if (res < 0) yading@11: return res; yading@11: } yading@11: yading@11: if (st->codec->codec_id == AV_CODEC_ID_PRORES) yading@11: offset = 8; yading@11: yading@11: pkt = av_mallocz(sizeof(AVPacket)); yading@11: /* XXX: prevent data copy... */ yading@11: if (av_new_packet(pkt, pkt_size + offset) < 0) { yading@11: av_free(pkt); yading@11: return AVERROR(ENOMEM); yading@11: } yading@11: yading@11: if (st->codec->codec_id == AV_CODEC_ID_PRORES) { yading@11: uint8_t *buf = pkt->data; yading@11: bytestream_put_be32(&buf, pkt_size); yading@11: bytestream_put_be32(&buf, MKBETAG('i', 'c', 'p', 'f')); yading@11: } yading@11: yading@11: memcpy(pkt->data + offset, pkt_data, pkt_size); yading@11: yading@11: if (pkt_data != data) yading@11: av_free(pkt_data); yading@11: yading@11: pkt->flags = is_keyframe; yading@11: pkt->stream_index = st->index; yading@11: yading@11: if (additional_size > 0) { yading@11: uint8_t *side_data = av_packet_new_side_data(pkt, yading@11: AV_PKT_DATA_MATROSKA_BLOCKADDITIONAL, yading@11: additional_size + 8); yading@11: if(side_data == NULL) { yading@11: av_free_packet(pkt); yading@11: av_free(pkt); yading@11: return AVERROR(ENOMEM); yading@11: } yading@11: AV_WB64(side_data, additional_id); yading@11: memcpy(side_data + 8, additional, additional_size); yading@11: } yading@11: yading@11: if (track->ms_compat) yading@11: pkt->dts = timecode; yading@11: else yading@11: pkt->pts = timecode; yading@11: pkt->pos = pos; yading@11: if (st->codec->codec_id == AV_CODEC_ID_SUBRIP) { yading@11: /* yading@11: * For backward compatibility. yading@11: * Historically, we have put subtitle duration yading@11: * in convergence_duration, on the off chance yading@11: * that the time_scale is less than 1us, which yading@11: * could result in a 32bit overflow on the yading@11: * normal duration field. yading@11: */ yading@11: pkt->convergence_duration = lace_duration; yading@11: } yading@11: yading@11: if (track->type != MATROSKA_TRACK_TYPE_SUBTITLE || yading@11: lace_duration <= INT_MAX) { yading@11: /* yading@11: * For non subtitle tracks, just store the duration yading@11: * as normal. yading@11: * yading@11: * If it's a subtitle track and duration value does yading@11: * not overflow a uint32, then also store it normally. yading@11: */ yading@11: pkt->duration = lace_duration; yading@11: } yading@11: yading@11: #if FF_API_ASS_SSA yading@11: if (st->codec->codec_id == AV_CODEC_ID_SSA) yading@11: matroska_fix_ass_packet(matroska, pkt, lace_duration); yading@11: yading@11: if (matroska->prev_pkt && yading@11: timecode != AV_NOPTS_VALUE && yading@11: matroska->prev_pkt->pts == timecode && yading@11: matroska->prev_pkt->stream_index == st->index && yading@11: st->codec->codec_id == AV_CODEC_ID_SSA) yading@11: matroska_merge_packets(matroska->prev_pkt, pkt); yading@11: else { yading@11: dynarray_add(&matroska->packets,&matroska->num_packets,pkt); yading@11: matroska->prev_pkt = pkt; yading@11: } yading@11: #else yading@11: dynarray_add(&matroska->packets, &matroska->num_packets, pkt); yading@11: matroska->prev_pkt = pkt; yading@11: #endif yading@11: yading@11: return 0; yading@11: } yading@11: yading@11: static int matroska_parse_block(MatroskaDemuxContext *matroska, uint8_t *data, yading@11: int size, int64_t pos, uint64_t cluster_time, yading@11: uint64_t block_duration, int is_keyframe, yading@11: uint8_t *additional, uint64_t additional_id, int additional_size, yading@11: int64_t cluster_pos) yading@11: { yading@11: uint64_t timecode = AV_NOPTS_VALUE; yading@11: MatroskaTrack *track; yading@11: int res = 0; yading@11: AVStream *st; yading@11: int16_t block_time; yading@11: uint32_t *lace_size = NULL; yading@11: int n, flags, laces = 0; yading@11: uint64_t num; yading@11: yading@11: if ((n = matroska_ebmlnum_uint(matroska, data, size, &num)) < 0) { yading@11: av_log(matroska->ctx, AV_LOG_ERROR, "EBML block data error\n"); yading@11: return n; yading@11: } yading@11: data += n; yading@11: size -= n; yading@11: yading@11: track = matroska_find_track_by_num(matroska, num); yading@11: if (!track || !track->stream) { yading@11: av_log(matroska->ctx, AV_LOG_INFO, yading@11: "Invalid stream %"PRIu64" or size %u\n", num, size); yading@11: return AVERROR_INVALIDDATA; yading@11: } else if (size <= 3) yading@11: return 0; yading@11: st = track->stream; yading@11: if (st->discard >= AVDISCARD_ALL) yading@11: return res; yading@11: av_assert1(block_duration != AV_NOPTS_VALUE); yading@11: yading@11: block_time = AV_RB16(data); yading@11: data += 2; yading@11: flags = *data++; yading@11: size -= 3; yading@11: if (is_keyframe == -1) yading@11: is_keyframe = flags & 0x80 ? AV_PKT_FLAG_KEY : 0; yading@11: yading@11: if (cluster_time != (uint64_t)-1 yading@11: && (block_time >= 0 || cluster_time >= -block_time)) { yading@11: timecode = cluster_time + block_time; yading@11: if (track->type == MATROSKA_TRACK_TYPE_SUBTITLE yading@11: && timecode < track->end_timecode) yading@11: is_keyframe = 0; /* overlapping subtitles are not key frame */ yading@11: if (is_keyframe) yading@11: av_add_index_entry(st, cluster_pos, timecode, 0,0,AVINDEX_KEYFRAME); yading@11: } yading@11: yading@11: if (matroska->skip_to_keyframe && track->type != MATROSKA_TRACK_TYPE_SUBTITLE) { yading@11: if (timecode < matroska->skip_to_timecode) yading@11: return res; yading@11: if (!st->skip_to_keyframe) { yading@11: av_log(matroska->ctx, AV_LOG_ERROR, "File is broken, keyframes not correctly marked!\n"); yading@11: matroska->skip_to_keyframe = 0; yading@11: } yading@11: if (is_keyframe) yading@11: matroska->skip_to_keyframe = 0; yading@11: } yading@11: yading@11: res = matroska_parse_laces(matroska, &data, &size, (flags & 0x06) >> 1, yading@11: &lace_size, &laces); yading@11: yading@11: if (res) yading@11: goto end; yading@11: yading@11: if (!block_duration) yading@11: block_duration = track->default_duration * laces / matroska->time_scale; yading@11: yading@11: if (cluster_time != (uint64_t)-1 && (block_time >= 0 || cluster_time >= -block_time)) yading@11: track->end_timecode = yading@11: FFMAX(track->end_timecode, timecode + block_duration); yading@11: yading@11: for (n = 0; n < laces; n++) { yading@11: int64_t lace_duration = block_duration*(n+1) / laces - block_duration*n / laces; yading@11: yading@11: if (lace_size[n] > size) { yading@11: av_log(matroska->ctx, AV_LOG_ERROR, "Invalid packet size\n"); yading@11: break; yading@11: } yading@11: yading@11: if ((st->codec->codec_id == AV_CODEC_ID_RA_288 || yading@11: st->codec->codec_id == AV_CODEC_ID_COOK || yading@11: st->codec->codec_id == AV_CODEC_ID_SIPR || yading@11: st->codec->codec_id == AV_CODEC_ID_ATRAC3) && yading@11: st->codec->block_align && track->audio.sub_packet_size) { yading@11: yading@11: res = matroska_parse_rm_audio(matroska, track, st, data, yading@11: lace_size[n], yading@11: timecode, pos); yading@11: if (res) yading@11: goto end; yading@11: yading@11: } else { yading@11: res = matroska_parse_frame(matroska, track, st, data, lace_size[n], yading@11: timecode, lace_duration, yading@11: pos, !n? is_keyframe : 0, yading@11: additional, additional_id, additional_size); yading@11: if (res) yading@11: goto end; yading@11: } yading@11: yading@11: if (timecode != AV_NOPTS_VALUE) yading@11: timecode = lace_duration ? timecode + lace_duration : AV_NOPTS_VALUE; yading@11: data += lace_size[n]; yading@11: size -= lace_size[n]; yading@11: } yading@11: yading@11: end: yading@11: av_free(lace_size); yading@11: return res; yading@11: } yading@11: yading@11: static int matroska_parse_cluster_incremental(MatroskaDemuxContext *matroska) yading@11: { yading@11: EbmlList *blocks_list; yading@11: MatroskaBlock *blocks; yading@11: int i, res; yading@11: res = ebml_parse(matroska, yading@11: matroska_cluster_incremental_parsing, yading@11: &matroska->current_cluster); yading@11: if (res == 1) { yading@11: /* New Cluster */ yading@11: if (matroska->current_cluster_pos) yading@11: ebml_level_end(matroska); yading@11: ebml_free(matroska_cluster, &matroska->current_cluster); yading@11: memset(&matroska->current_cluster, 0, sizeof(MatroskaCluster)); yading@11: matroska->current_cluster_num_blocks = 0; yading@11: matroska->current_cluster_pos = avio_tell(matroska->ctx->pb); yading@11: matroska->prev_pkt = NULL; yading@11: /* sizeof the ID which was already read */ yading@11: if (matroska->current_id) yading@11: matroska->current_cluster_pos -= 4; yading@11: res = ebml_parse(matroska, yading@11: matroska_clusters_incremental, yading@11: &matroska->current_cluster); yading@11: /* Try parsing the block again. */ yading@11: if (res == 1) yading@11: res = ebml_parse(matroska, yading@11: matroska_cluster_incremental_parsing, yading@11: &matroska->current_cluster); yading@11: } yading@11: yading@11: if (!res && yading@11: matroska->current_cluster_num_blocks < yading@11: matroska->current_cluster.blocks.nb_elem) { yading@11: blocks_list = &matroska->current_cluster.blocks; yading@11: blocks = blocks_list->elem; yading@11: yading@11: matroska->current_cluster_num_blocks = blocks_list->nb_elem; yading@11: i = blocks_list->nb_elem - 1; yading@11: if (blocks[i].bin.size > 0 && blocks[i].bin.data) { yading@11: int is_keyframe = blocks[i].non_simple ? !blocks[i].reference : -1; yading@11: uint8_t* additional = blocks[i].additional.size > 0 ? yading@11: blocks[i].additional.data : NULL; yading@11: if (!blocks[i].non_simple) yading@11: blocks[i].duration = 0; yading@11: res = matroska_parse_block(matroska, yading@11: blocks[i].bin.data, blocks[i].bin.size, yading@11: blocks[i].bin.pos, yading@11: matroska->current_cluster.timecode, yading@11: blocks[i].duration, is_keyframe, yading@11: additional, blocks[i].additional_id, yading@11: blocks[i].additional.size, yading@11: matroska->current_cluster_pos); yading@11: } yading@11: } yading@11: yading@11: if (res < 0) matroska->done = 1; yading@11: return res; yading@11: } yading@11: yading@11: static int matroska_parse_cluster(MatroskaDemuxContext *matroska) yading@11: { yading@11: MatroskaCluster cluster = { 0 }; yading@11: EbmlList *blocks_list; yading@11: MatroskaBlock *blocks; yading@11: int i, res; yading@11: int64_t pos; yading@11: if (!matroska->contains_ssa) yading@11: return matroska_parse_cluster_incremental(matroska); yading@11: pos = avio_tell(matroska->ctx->pb); yading@11: matroska->prev_pkt = NULL; yading@11: if (matroska->current_id) yading@11: pos -= 4; /* sizeof the ID which was already read */ yading@11: res = ebml_parse(matroska, matroska_clusters, &cluster); yading@11: blocks_list = &cluster.blocks; yading@11: blocks = blocks_list->elem; yading@11: for (i=0; inb_elem; i++) yading@11: if (blocks[i].bin.size > 0 && blocks[i].bin.data) { yading@11: int is_keyframe = blocks[i].non_simple ? !blocks[i].reference : -1; yading@11: res=matroska_parse_block(matroska, yading@11: blocks[i].bin.data, blocks[i].bin.size, yading@11: blocks[i].bin.pos, cluster.timecode, yading@11: blocks[i].duration, is_keyframe, NULL, 0, 0, yading@11: pos); yading@11: } yading@11: ebml_free(matroska_cluster, &cluster); yading@11: return res; yading@11: } yading@11: yading@11: static int matroska_read_packet(AVFormatContext *s, AVPacket *pkt) yading@11: { yading@11: MatroskaDemuxContext *matroska = s->priv_data; yading@11: yading@11: while (matroska_deliver_packet(matroska, pkt)) { yading@11: int64_t pos = avio_tell(matroska->ctx->pb); yading@11: if (matroska->done) yading@11: return AVERROR_EOF; yading@11: if (matroska_parse_cluster(matroska) < 0) yading@11: matroska_resync(matroska, pos); yading@11: } yading@11: yading@11: return 0; yading@11: } yading@11: yading@11: static int matroska_read_seek(AVFormatContext *s, int stream_index, yading@11: int64_t timestamp, int flags) yading@11: { yading@11: MatroskaDemuxContext *matroska = s->priv_data; yading@11: MatroskaTrack *tracks = matroska->tracks.elem; yading@11: AVStream *st = s->streams[stream_index]; yading@11: int i, index, index_sub, index_min; yading@11: yading@11: /* Parse the CUES now since we need the index data to seek. */ yading@11: if (matroska->cues_parsing_deferred > 0) { yading@11: matroska->cues_parsing_deferred = 0; yading@11: matroska_parse_cues(matroska); yading@11: } yading@11: yading@11: if (!st->nb_index_entries) yading@11: goto err; yading@11: timestamp = FFMAX(timestamp, st->index_entries[0].timestamp); yading@11: yading@11: if ((index = av_index_search_timestamp(st, timestamp, flags)) < 0) { yading@11: avio_seek(s->pb, st->index_entries[st->nb_index_entries-1].pos, SEEK_SET); yading@11: matroska->current_id = 0; yading@11: while ((index = av_index_search_timestamp(st, timestamp, flags)) < 0) { yading@11: matroska_clear_queue(matroska); yading@11: if (matroska_parse_cluster(matroska) < 0) yading@11: break; yading@11: } yading@11: } yading@11: yading@11: matroska_clear_queue(matroska); yading@11: if (index < 0 || (matroska->cues_parsing_deferred < 0 && index == st->nb_index_entries - 1)) yading@11: goto err; yading@11: yading@11: index_min = index; yading@11: for (i=0; i < matroska->tracks.nb_elem; i++) { yading@11: tracks[i].audio.pkt_cnt = 0; yading@11: tracks[i].audio.sub_packet_cnt = 0; yading@11: tracks[i].audio.buf_timecode = AV_NOPTS_VALUE; yading@11: tracks[i].end_timecode = 0; yading@11: if (tracks[i].type == MATROSKA_TRACK_TYPE_SUBTITLE yading@11: && tracks[i].stream->discard != AVDISCARD_ALL) { yading@11: index_sub = av_index_search_timestamp(tracks[i].stream, st->index_entries[index].timestamp, AVSEEK_FLAG_BACKWARD); yading@11: if (index_sub >= 0 yading@11: && st->index_entries[index_sub].pos < st->index_entries[index_min].pos yading@11: && st->index_entries[index].timestamp - st->index_entries[index_sub].timestamp < 30000000000/matroska->time_scale) yading@11: index_min = index_sub; yading@11: } yading@11: } yading@11: yading@11: avio_seek(s->pb, st->index_entries[index_min].pos, SEEK_SET); yading@11: matroska->current_id = 0; yading@11: if (flags & AVSEEK_FLAG_ANY) { yading@11: st->skip_to_keyframe = 0; yading@11: matroska->skip_to_timecode = timestamp; yading@11: } else { yading@11: st->skip_to_keyframe = 1; yading@11: matroska->skip_to_timecode = st->index_entries[index].timestamp; yading@11: } yading@11: matroska->skip_to_keyframe = 1; yading@11: matroska->done = 0; yading@11: matroska->num_levels = 0; yading@11: ff_update_cur_dts(s, st, st->index_entries[index].timestamp); yading@11: return 0; yading@11: err: yading@11: // slightly hackish but allows proper fallback to yading@11: // the generic seeking code. yading@11: matroska_clear_queue(matroska); yading@11: matroska->current_id = 0; yading@11: st->skip_to_keyframe = yading@11: matroska->skip_to_keyframe = 0; yading@11: matroska->done = 0; yading@11: matroska->num_levels = 0; yading@11: return -1; yading@11: } yading@11: yading@11: static int matroska_read_close(AVFormatContext *s) yading@11: { yading@11: MatroskaDemuxContext *matroska = s->priv_data; yading@11: MatroskaTrack *tracks = matroska->tracks.elem; yading@11: int n; yading@11: yading@11: matroska_clear_queue(matroska); yading@11: yading@11: for (n=0; n < matroska->tracks.nb_elem; n++) yading@11: if (tracks[n].type == MATROSKA_TRACK_TYPE_AUDIO) yading@11: av_free(tracks[n].audio.buf); yading@11: ebml_free(matroska_cluster, &matroska->current_cluster); yading@11: ebml_free(matroska_segment, matroska); yading@11: yading@11: return 0; yading@11: } yading@11: yading@11: AVInputFormat ff_matroska_demuxer = { yading@11: .name = "matroska,webm", yading@11: .long_name = NULL_IF_CONFIG_SMALL("Matroska / WebM"), yading@11: .priv_data_size = sizeof(MatroskaDemuxContext), yading@11: .read_probe = matroska_probe, yading@11: .read_header = matroska_read_header, yading@11: .read_packet = matroska_read_packet, yading@11: .read_close = matroska_read_close, yading@11: .read_seek = matroska_read_seek, yading@11: };