David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 1 | /* |
Aurelien Jacobs | ff33c5c | 2008-08-05 00:42:43 | [diff] [blame] | 2 | * Matroska file demuxer |
Diego Biurrun | 5968d2d | 2008-08-05 08:28:57 | [diff] [blame] | 3 | * Copyright (c) 2003-2008 The FFmpeg Project |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 4 | * |
| 5 | * This file is part of FFmpeg. |
| 6 | * |
| 7 | * FFmpeg is free software; you can redistribute it and/or |
| 8 | * modify it under the terms of the GNU Lesser General Public |
| 9 | * License as published by the Free Software Foundation; either |
| 10 | * version 2.1 of the License, or (at your option) any later version. |
| 11 | * |
| 12 | * FFmpeg is distributed in the hope that it will be useful, |
| 13 | * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| 14 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
| 15 | * Lesser General Public License for more details. |
| 16 | * |
| 17 | * You should have received a copy of the GNU Lesser General Public |
| 18 | * License along with FFmpeg; if not, write to the Free Software |
| 19 | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
| 20 | */ |
| 21 | |
| 22 | /** |
Diego Biurrun | ba87f08 | 2010-04-20 14:45:34 | [diff] [blame] | 23 | * @file |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 24 | * Matroska file demuxer |
Diego Biurrun | e873c03 | 2011-10-30 18:10:50 | [diff] [blame] | 25 | * @author Ronald Bultje <[email protected]> |
| 26 | * @author with a little help from Moritz Bunkus <[email protected]> |
| 27 | * @author totally reworked by Aurelien Jacobs <[email protected]> |
| 28 | * @see specs available on the Matroska project page: http://www.matroska.org/ |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 29 | */ |
| 30 | |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 31 | #include "config.h" |
| 32 | |
Diego Biurrun | d92024f | 2014-03-10 14:35:59 | [diff] [blame] | 33 | #include <inttypes.h> |
Aurelien Jacobs | 3eb9bfb | 2008-09-04 23:26:12 | [diff] [blame] | 34 | #include <stdio.h> |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 35 | |
| 36 | #include "libavutil/avstring.h" |
Michael Niedermayer | b3d9ab1 | 2014-03-07 22:46:37 | [diff] [blame] | 37 | #include "libavutil/base64.h" |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 38 | #include "libavutil/dict.h" |
| 39 | #include "libavutil/intfloat.h" |
| 40 | #include "libavutil/intreadwrite.h" |
| 41 | #include "libavutil/lzo.h" |
Neil Birkbeck | bbda13a | 2016-03-01 00:41:04 | [diff] [blame] | 42 | #include "libavutil/mastering_display_metadata.h" |
Anton Khirnov | eb3b550 | 2014-04-29 10:03:13 | [diff] [blame] | 43 | #include "libavutil/mathematics.h" |
Vignesh Venkatasubramanian | 2171b7c | 2015-03-31 23:51:57 | [diff] [blame] | 44 | #include "libavutil/opt.h" |
Michael Niedermayer | a52cb42 | 2014-11-02 18:19:07 | [diff] [blame] | 45 | #include "libavutil/time_internal.h" |
James Almer | 445204c | 2016-12-06 17:48:45 | [diff] [blame] | 46 | #include "libavutil/spherical.h" |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 47 | |
| 48 | #include "libavcodec/bytestream.h" |
Anton Khirnov | 4efdadc | 2014-05-25 12:05:51 | [diff] [blame] | 49 | #include "libavcodec/flac.h" |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 50 | #include "libavcodec/mpeg4audio.h" |
| 51 | |
| 52 | #include "avformat.h" |
| 53 | #include "avio_internal.h" |
| 54 | #include "internal.h" |
| 55 | #include "isom.h" |
| 56 | #include "matroska.h" |
Anton Khirnov | 23f741f | 2014-05-26 10:48:56 | [diff] [blame] | 57 | #include "oggdec.h" |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 58 | /* For ff_codec_get_id(). */ |
| 59 | #include "riff.h" |
| 60 | #include "rmsipr.h" |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 61 | |
Matthew Oliver | 0167fa0 | 2014-11-27 08:00:36 | [diff] [blame] | 62 | #if CONFIG_BZLIB |
| 63 | #include <bzlib.h> |
| 64 | #endif |
| 65 | #if CONFIG_ZLIB |
| 66 | #include <zlib.h> |
| 67 | #endif |
| 68 | |
Mats Peterson | 7973603 | 2015-12-27 20:28:09 | [diff] [blame] | 69 | #include "qtpalette.h" |
| 70 | |
Steve Lhomme | a2c7702 | 2019-02-23 10:14:33 | [diff] [blame] | 71 | #define EBML_UNKNOWN_LENGTH UINT64_MAX /* EBML unknown length, in uint64_t */ |
Andreas Rheinhardt | a569a7b | 2019-06-23 23:42:30 | [diff] [blame] | 72 | #define NEEDS_CHECKING 2 /* Indicates that some error checks |
| 73 | * still need to be performed */ |
Andreas Rheinhardt | b31c9b7 | 2019-06-23 23:46:54 | [diff] [blame] | 74 | #define LEVEL_ENDED 3 /* return value of ebml_parse when the |
| 75 | * syntax level used for parsing ended. */ |
Andreas Rheinhardt | 5120305 | 2019-05-16 22:30:14 | [diff] [blame] | 76 | #define SKIP_THRESHOLD 1024 * 1024 /* In non-seekable mode, if more than SKIP_THRESHOLD |
| 77 | * of unkown, potentially damaged data is encountered, |
| 78 | * it is considered an error. */ |
| 79 | #define UNKNOWN_EQUIV 50 * 1024 /* An unknown element is considered equivalent |
| 80 | * to this many bytes of unknown data for the |
| 81 | * SKIP_THRESHOLD check. */ |
Steve Lhomme | a2c7702 | 2019-02-23 10:14:33 | [diff] [blame] | 82 | |
Aurelien Jacobs | 789ed10 | 2008-08-05 00:40:00 | [diff] [blame] | 83 | typedef enum { |
| 84 | EBML_NONE, |
| 85 | EBML_UINT, |
Andreas Rheinhardt | 60f75c9 | 2019-05-16 22:30:13 | [diff] [blame] | 86 | EBML_SINT, |
Aurelien Jacobs | 789ed10 | 2008-08-05 00:40:00 | [diff] [blame] | 87 | EBML_FLOAT, |
| 88 | EBML_STR, |
| 89 | EBML_UTF8, |
| 90 | EBML_BIN, |
| 91 | EBML_NEST, |
wm4 | cac2295 | 2015-02-09 19:39:00 | [diff] [blame] | 92 | EBML_LEVEL1, |
Aurelien Jacobs | 789ed10 | 2008-08-05 00:40:00 | [diff] [blame] | 93 | EBML_STOP, |
Reimar Döffinger | 14d735b | 2011-02-06 10:32:03 | [diff] [blame] | 94 | EBML_TYPE_COUNT |
Aurelien Jacobs | 789ed10 | 2008-08-05 00:40:00 | [diff] [blame] | 95 | } EbmlType; |
| 96 | |
| 97 | typedef const struct EbmlSyntax { |
| 98 | uint32_t id; |
| 99 | EbmlType type; |
James Almer | f34aabf | 2019-09-03 21:52:51 | [diff] [blame] | 100 | size_t list_elem_size; |
| 101 | size_t data_offset; |
Aurelien Jacobs | 789ed10 | 2008-08-05 00:40:00 | [diff] [blame] | 102 | union { |
Chris Cunningham | ac25840 | 2017-02-03 22:42:44 | [diff] [blame] | 103 | int64_t i; |
Aurelien Jacobs | 789ed10 | 2008-08-05 00:40:00 | [diff] [blame] | 104 | uint64_t u; |
| 105 | double f; |
| 106 | const char *s; |
| 107 | const struct EbmlSyntax *n; |
| 108 | } def; |
| 109 | } EbmlSyntax; |
| 110 | |
Diego Biurrun | daf8cf3 | 2014-09-22 07:19:33 | [diff] [blame] | 111 | typedef struct EbmlList { |
Aurelien Jacobs | 789ed10 | 2008-08-05 00:40:00 | [diff] [blame] | 112 | int nb_elem; |
James Almer | 3b3150c | 2019-09-03 21:45:04 | [diff] [blame] | 113 | unsigned int alloc_elem_size; |
Aurelien Jacobs | 789ed10 | 2008-08-05 00:40:00 | [diff] [blame] | 114 | void *elem; |
| 115 | } EbmlList; |
| 116 | |
Diego Biurrun | daf8cf3 | 2014-09-22 07:19:33 | [diff] [blame] | 117 | typedef struct EbmlBin { |
Aurelien Jacobs | 789ed10 | 2008-08-05 00:40:00 | [diff] [blame] | 118 | int size; |
James Almer | a618866 | 2018-04-04 17:12:54 | [diff] [blame] | 119 | AVBufferRef *buf; |
Aurelien Jacobs | 789ed10 | 2008-08-05 00:40:00 | [diff] [blame] | 120 | uint8_t *data; |
| 121 | int64_t pos; |
| 122 | } EbmlBin; |
| 123 | |
Diego Biurrun | daf8cf3 | 2014-09-22 07:19:33 | [diff] [blame] | 124 | typedef struct Ebml { |
Aurelien Jacobs | 6351132 | 2008-08-05 00:40:02 | [diff] [blame] | 125 | uint64_t version; |
| 126 | uint64_t max_size; |
| 127 | uint64_t id_length; |
| 128 | char *doctype; |
| 129 | uint64_t doctype_version; |
| 130 | } Ebml; |
| 131 | |
Diego Biurrun | daf8cf3 | 2014-09-22 07:19:33 | [diff] [blame] | 132 | typedef struct MatroskaTrackCompression { |
Aurelien Jacobs | 2cbc881 | 2008-08-05 00:40:31 | [diff] [blame] | 133 | uint64_t algo; |
| 134 | EbmlBin settings; |
| 135 | } MatroskaTrackCompression; |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 136 | |
Michael Niedermayer | a0fe1a2 | 2015-02-14 20:07:40 | [diff] [blame] | 137 | typedef struct MatroskaTrackEncryption { |
Frank Galligan | b853103 | 2013-03-07 16:11:38 | [diff] [blame] | 138 | uint64_t algo; |
| 139 | EbmlBin key_id; |
| 140 | } MatroskaTrackEncryption; |
| 141 | |
Diego Biurrun | daf8cf3 | 2014-09-22 07:19:33 | [diff] [blame] | 142 | typedef struct MatroskaTrackEncoding { |
Aurelien Jacobs | 2cbc881 | 2008-08-05 00:40:31 | [diff] [blame] | 143 | uint64_t scope; |
| 144 | uint64_t type; |
| 145 | MatroskaTrackCompression compression; |
Frank Galligan | b853103 | 2013-03-07 16:11:38 | [diff] [blame] | 146 | MatroskaTrackEncryption encryption; |
Aurelien Jacobs | 2cbc881 | 2008-08-05 00:40:31 | [diff] [blame] | 147 | } MatroskaTrackEncoding; |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 148 | |
Neil Birkbeck | bbda13a | 2016-03-01 00:41:04 | [diff] [blame] | 149 | typedef struct MatroskaMasteringMeta { |
| 150 | double r_x; |
| 151 | double r_y; |
| 152 | double g_x; |
| 153 | double g_y; |
| 154 | double b_x; |
| 155 | double b_y; |
| 156 | double white_x; |
| 157 | double white_y; |
| 158 | double max_luminance; |
| 159 | double min_luminance; |
| 160 | } MatroskaMasteringMeta; |
| 161 | |
| 162 | typedef struct MatroskaTrackVideoColor { |
| 163 | uint64_t matrix_coefficients; |
| 164 | uint64_t bits_per_channel; |
| 165 | uint64_t chroma_sub_horz; |
| 166 | uint64_t chroma_sub_vert; |
| 167 | uint64_t cb_sub_horz; |
| 168 | uint64_t cb_sub_vert; |
| 169 | uint64_t chroma_siting_horz; |
| 170 | uint64_t chroma_siting_vert; |
| 171 | uint64_t range; |
| 172 | uint64_t transfer_characteristics; |
| 173 | uint64_t primaries; |
| 174 | uint64_t max_cll; |
| 175 | uint64_t max_fall; |
| 176 | MatroskaMasteringMeta mastering_meta; |
| 177 | } MatroskaTrackVideoColor; |
| 178 | |
James Almer | 445204c | 2016-12-06 17:48:45 | [diff] [blame] | 179 | typedef struct MatroskaTrackVideoProjection { |
| 180 | uint64_t type; |
| 181 | EbmlBin private; |
| 182 | double yaw; |
| 183 | double pitch; |
| 184 | double roll; |
| 185 | } MatroskaTrackVideoProjection; |
| 186 | |
Diego Biurrun | daf8cf3 | 2014-09-22 07:19:33 | [diff] [blame] | 187 | typedef struct MatroskaTrackVideo { |
Aurelien Jacobs | 2cbc881 | 2008-08-05 00:40:31 | [diff] [blame] | 188 | double frame_rate; |
| 189 | uint64_t display_width; |
| 190 | uint64_t display_height; |
| 191 | uint64_t pixel_width; |
| 192 | uint64_t pixel_height; |
Andreas Rheinhardt | 60f75c9 | 2019-05-16 22:30:13 | [diff] [blame] | 193 | EbmlBin color_space; |
James Almer | bad8bbc | 2016-10-15 21:01:50 | [diff] [blame] | 194 | uint64_t display_unit; |
Luca Barbato | 5f02266 | 2016-03-28 18:29:54 | [diff] [blame] | 195 | uint64_t interlaced; |
| 196 | uint64_t field_order; |
Aurelien Jacobs | 4c509fe | 2011-05-23 23:09:24 | [diff] [blame] | 197 | uint64_t stereo_mode; |
Vignesh Venkatasubramanian | ce6a8e5 | 2013-02-04 23:17:52 | [diff] [blame] | 198 | uint64_t alpha_mode; |
James Almer | 4e75907 | 2016-12-05 02:22:39 | [diff] [blame] | 199 | EbmlList color; |
James Almer | 445204c | 2016-12-06 17:48:45 | [diff] [blame] | 200 | MatroskaTrackVideoProjection projection; |
Aurelien Jacobs | 2cbc881 | 2008-08-05 00:40:31 | [diff] [blame] | 201 | } MatroskaTrackVideo; |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 202 | |
Diego Biurrun | daf8cf3 | 2014-09-22 07:19:33 | [diff] [blame] | 203 | typedef struct MatroskaTrackAudio { |
Aurelien Jacobs | 2cbc881 | 2008-08-05 00:40:31 | [diff] [blame] | 204 | double samplerate; |
| 205 | double out_samplerate; |
| 206 | uint64_t bitdepth; |
| 207 | uint64_t channels; |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 208 | |
Aurelien Jacobs | 2cbc881 | 2008-08-05 00:40:31 | [diff] [blame] | 209 | /* real audio header (extracted from extradata) */ |
| 210 | int coded_framesize; |
| 211 | int sub_packet_h; |
| 212 | int frame_size; |
| 213 | int sub_packet_size; |
| 214 | int sub_packet_cnt; |
| 215 | int pkt_cnt; |
Reimar Döffinger | b09e506 | 2011-02-26 11:52:01 | [diff] [blame] | 216 | uint64_t buf_timecode; |
Aurelien Jacobs | 2cbc881 | 2008-08-05 00:40:31 | [diff] [blame] | 217 | uint8_t *buf; |
| 218 | } MatroskaTrackAudio; |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 219 | |
Michael Niedermayer | a0fe1a2 | 2015-02-14 20:07:40 | [diff] [blame] | 220 | typedef struct MatroskaTrackPlane { |
Kirill Gavrilov | e6ec921 | 2011-05-21 15:14:14 | [diff] [blame] | 221 | uint64_t uid; |
| 222 | uint64_t type; |
| 223 | } MatroskaTrackPlane; |
| 224 | |
Michael Niedermayer | a0fe1a2 | 2015-02-14 20:07:40 | [diff] [blame] | 225 | typedef struct MatroskaTrackOperation { |
Kirill Gavrilov | e6ec921 | 2011-05-21 15:14:14 | [diff] [blame] | 226 | EbmlList combine_planes; |
Kirill Gavrilov | e6ec921 | 2011-05-21 15:14:14 | [diff] [blame] | 227 | } MatroskaTrackOperation; |
| 228 | |
Diego Biurrun | daf8cf3 | 2014-09-22 07:19:33 | [diff] [blame] | 229 | typedef struct MatroskaTrack { |
Aurelien Jacobs | 2cbc881 | 2008-08-05 00:40:31 | [diff] [blame] | 230 | uint64_t num; |
Aurelien Jacobs | 325ace3 | 2009-02-15 15:34:22 | [diff] [blame] | 231 | uint64_t uid; |
Aurelien Jacobs | 2cbc881 | 2008-08-05 00:40:31 | [diff] [blame] | 232 | uint64_t type; |
Aurelien Jacobs | 38766e0 | 2009-02-15 15:29:09 | [diff] [blame] | 233 | char *name; |
Aurelien Jacobs | 2cbc881 | 2008-08-05 00:40:31 | [diff] [blame] | 234 | char *codec_id; |
| 235 | EbmlBin codec_priv; |
| 236 | char *language; |
Anton Khirnov | 7ff9708 | 2008-06-01 13:54:11 | [diff] [blame] | 237 | double time_scale; |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 238 | uint64_t default_duration; |
Aurelien Jacobs | 4eff974 | 2008-08-05 00:39:53 | [diff] [blame] | 239 | uint64_t flag_default; |
Aurelien Jacobs | 7a617a8 | 2010-07-02 16:38:44 | [diff] [blame] | 240 | uint64_t flag_forced; |
Vignesh Venkatasubramanian | d6f86d7 | 2013-10-14 17:42:08 | [diff] [blame] | 241 | uint64_t seek_preroll; |
Aurelien Jacobs | 2cbc881 | 2008-08-05 00:40:31 | [diff] [blame] | 242 | MatroskaTrackVideo video; |
| 243 | MatroskaTrackAudio audio; |
Kirill Gavrilov | e6ec921 | 2011-05-21 15:14:14 | [diff] [blame] | 244 | MatroskaTrackOperation operation; |
Aurelien Jacobs | 2cbc881 | 2008-08-05 00:40:31 | [diff] [blame] | 245 | EbmlList encodings; |
Anton Khirnov | eb3b550 | 2014-04-29 10:03:13 | [diff] [blame] | 246 | uint64_t codec_delay; |
Michael Niedermayer | b5bc436 | 2016-06-06 02:23:16 | [diff] [blame] | 247 | uint64_t codec_delay_in_track_tb; |
Aurelien Jacobs | fc4d335 | 2008-08-05 00:40:06 | [diff] [blame] | 248 | |
| 249 | AVStream *stream; |
Aurelien Jacobs | 82360e6 | 2008-09-09 12:07:10 | [diff] [blame] | 250 | int64_t end_timecode; |
Joakim Plate | 3e93c8e | 2010-03-03 21:46:43 | [diff] [blame] | 251 | int ms_compat; |
Vignesh Venkatasubramanian | 30c5c45 | 2013-02-13 21:51:48 | [diff] [blame] | 252 | uint64_t max_block_additional_id; |
Mats Peterson | 6aac43f | 2016-02-24 17:14:05 | [diff] [blame] | 253 | |
| 254 | uint32_t palette[AVPALETTE_COUNT]; |
| 255 | int has_palette; |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 256 | } MatroskaTrack; |
| 257 | |
Diego Biurrun | daf8cf3 | 2014-09-22 07:19:33 | [diff] [blame] | 258 | typedef struct MatroskaAttachment { |
Aurelien Jacobs | 325ace3 | 2009-02-15 15:34:22 | [diff] [blame] | 259 | uint64_t uid; |
Aurelien Jacobs | b414cb8 | 2008-08-05 00:40:24 | [diff] [blame] | 260 | char *filename; |
| 261 | char *mime; |
| 262 | EbmlBin bin; |
Aurelien Jacobs | 929e9de | 2009-02-15 15:53:55 | [diff] [blame] | 263 | |
| 264 | AVStream *stream; |
Diego Biurrun | f69befe | 2014-03-07 12:54:18 | [diff] [blame] | 265 | } MatroskaAttachment; |
Aurelien Jacobs | b414cb8 | 2008-08-05 00:40:24 | [diff] [blame] | 266 | |
Diego Biurrun | daf8cf3 | 2014-09-22 07:19:33 | [diff] [blame] | 267 | typedef struct MatroskaChapter { |
Aurelien Jacobs | 6bbd7c7 | 2008-08-05 00:40:21 | [diff] [blame] | 268 | uint64_t start; |
| 269 | uint64_t end; |
| 270 | uint64_t uid; |
| 271 | char *title; |
Aurelien Jacobs | 6cb6e15 | 2009-02-15 15:25:14 | [diff] [blame] | 272 | |
| 273 | AVChapter *chapter; |
Aurelien Jacobs | 6bbd7c7 | 2008-08-05 00:40:21 | [diff] [blame] | 274 | } MatroskaChapter; |
| 275 | |
Diego Biurrun | daf8cf3 | 2014-09-22 07:19:33 | [diff] [blame] | 276 | typedef struct MatroskaIndexPos { |
Aurelien Jacobs | e5929fd | 2008-08-05 00:40:15 | [diff] [blame] | 277 | uint64_t track; |
| 278 | uint64_t pos; |
| 279 | } MatroskaIndexPos; |
| 280 | |
Diego Biurrun | daf8cf3 | 2014-09-22 07:19:33 | [diff] [blame] | 281 | typedef struct MatroskaIndex { |
Aurelien Jacobs | e5929fd | 2008-08-05 00:40:15 | [diff] [blame] | 282 | uint64_t time; |
| 283 | EbmlList pos; |
| 284 | } MatroskaIndex; |
| 285 | |
Diego Biurrun | daf8cf3 | 2014-09-22 07:19:33 | [diff] [blame] | 286 | typedef struct MatroskaTag { |
Aurelien Jacobs | 44015c5 | 2008-08-08 23:50:38 | [diff] [blame] | 287 | char *name; |
| 288 | char *string; |
Aurelien Jacobs | f702df3 | 2009-02-15 16:05:37 | [diff] [blame] | 289 | char *lang; |
| 290 | uint64_t def; |
Aurelien Jacobs | 44015c5 | 2008-08-08 23:50:38 | [diff] [blame] | 291 | EbmlList sub; |
| 292 | } MatroskaTag; |
| 293 | |
Diego Biurrun | daf8cf3 | 2014-09-22 07:19:33 | [diff] [blame] | 294 | typedef struct MatroskaTagTarget { |
Aurelien Jacobs | 929e9de | 2009-02-15 15:53:55 | [diff] [blame] | 295 | char *type; |
| 296 | uint64_t typevalue; |
| 297 | uint64_t trackuid; |
| 298 | uint64_t chapteruid; |
| 299 | uint64_t attachuid; |
| 300 | } MatroskaTagTarget; |
| 301 | |
Diego Biurrun | daf8cf3 | 2014-09-22 07:19:33 | [diff] [blame] | 302 | typedef struct MatroskaTags { |
Aurelien Jacobs | 929e9de | 2009-02-15 15:53:55 | [diff] [blame] | 303 | MatroskaTagTarget target; |
| 304 | EbmlList tag; |
| 305 | } MatroskaTags; |
| 306 | |
Diego Biurrun | daf8cf3 | 2014-09-22 07:19:33 | [diff] [blame] | 307 | typedef struct MatroskaSeekhead { |
Aurelien Jacobs | 13b350a | 2008-08-05 00:40:36 | [diff] [blame] | 308 | uint64_t id; |
| 309 | uint64_t pos; |
| 310 | } MatroskaSeekhead; |
| 311 | |
Diego Biurrun | daf8cf3 | 2014-09-22 07:19:33 | [diff] [blame] | 312 | typedef struct MatroskaLevel { |
Aurelien Jacobs | 8d75b5a | 2007-06-04 22:35:16 | [diff] [blame] | 313 | uint64_t start; |
| 314 | uint64_t length; |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 315 | } MatroskaLevel; |
| 316 | |
Andreas Rheinhardt | ffa64a4 | 2019-05-16 22:29:59 | [diff] [blame] | 317 | typedef struct MatroskaBlock { |
| 318 | uint64_t duration; |
| 319 | int64_t reference; |
| 320 | uint64_t non_simple; |
| 321 | EbmlBin bin; |
| 322 | uint64_t additional_id; |
| 323 | EbmlBin additional; |
Andreas Rheinhardt | 60f75c9 | 2019-05-16 22:30:13 | [diff] [blame] | 324 | int64_t discard_padding; |
Andreas Rheinhardt | ffa64a4 | 2019-05-16 22:29:59 | [diff] [blame] | 325 | } MatroskaBlock; |
| 326 | |
Diego Biurrun | daf8cf3 | 2014-09-22 07:19:33 | [diff] [blame] | 327 | typedef struct MatroskaCluster { |
Andreas Rheinhardt | ffa64a4 | 2019-05-16 22:29:59 | [diff] [blame] | 328 | MatroskaBlock block; |
Dale Curtis | 8336eb6 | 2012-04-19 18:12:24 | [diff] [blame] | 329 | uint64_t timecode; |
Andreas Rheinhardt | ffa64a4 | 2019-05-16 22:29:59 | [diff] [blame] | 330 | int64_t pos; |
Dale Curtis | 8336eb6 | 2012-04-19 18:12:24 | [diff] [blame] | 331 | } MatroskaCluster; |
| 332 | |
Michael Niedermayer | a0fe1a2 | 2015-02-14 20:07:40 | [diff] [blame] | 333 | typedef struct MatroskaLevel1Element { |
Andreas Rheinhardt | 730ac1a | 2019-05-16 22:30:20 | [diff] [blame] | 334 | int64_t pos; |
Andreas Rheinhardt | 410a082 | 2019-05-16 22:29:47 | [diff] [blame] | 335 | uint32_t id; |
wm4 | cac2295 | 2015-02-09 19:39:00 | [diff] [blame] | 336 | int parsed; |
| 337 | } MatroskaLevel1Element; |
| 338 | |
Diego Biurrun | daf8cf3 | 2014-09-22 07:19:33 | [diff] [blame] | 339 | typedef struct MatroskaDemuxContext { |
Vignesh Venkatasubramanian | 2171b7c | 2015-03-31 23:51:57 | [diff] [blame] | 340 | const AVClass *class; |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 341 | AVFormatContext *ctx; |
| 342 | |
Diego Biurrun | 5968d2d | 2008-08-05 08:28:57 | [diff] [blame] | 343 | /* EBML stuff */ |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 344 | MatroskaLevel levels[EBML_MAX_DEPTH]; |
Andreas Rheinhardt | 60f75c9 | 2019-05-16 22:30:13 | [diff] [blame] | 345 | int num_levels; |
Aurelien Jacobs | c3ade62 | 2010-06-11 16:34:01 | [diff] [blame] | 346 | uint32_t current_id; |
Andreas Rheinhardt | a3db9f6 | 2019-05-16 22:30:05 | [diff] [blame] | 347 | int64_t resync_pos; |
Andreas Rheinhardt | 5120305 | 2019-05-16 22:30:14 | [diff] [blame] | 348 | int unknown_count; |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 349 | |
Aurelien Jacobs | 2970858 | 2008-08-05 00:40:27 | [diff] [blame] | 350 | uint64_t time_scale; |
| 351 | double duration; |
| 352 | char *title; |
James Almer | 2c759d7 | 2013-11-24 08:31:48 | [diff] [blame] | 353 | char *muxingapp; |
Andreas Rheinhardt | 60f75c9 | 2019-05-16 22:30:13 | [diff] [blame] | 354 | EbmlBin date_utc; |
Aurelien Jacobs | 2cbc881 | 2008-08-05 00:40:31 | [diff] [blame] | 355 | EbmlList tracks; |
Aurelien Jacobs | b414cb8 | 2008-08-05 00:40:24 | [diff] [blame] | 356 | EbmlList attachments; |
Aurelien Jacobs | 6bbd7c7 | 2008-08-05 00:40:21 | [diff] [blame] | 357 | EbmlList chapters; |
Aurelien Jacobs | e5929fd | 2008-08-05 00:40:15 | [diff] [blame] | 358 | EbmlList index; |
Aurelien Jacobs | 44015c5 | 2008-08-08 23:50:38 | [diff] [blame] | 359 | EbmlList tags; |
Aurelien Jacobs | 13b350a | 2008-08-05 00:40:36 | [diff] [blame] | 360 | EbmlList seekhead; |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 361 | |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 362 | /* byte position of the segment inside the stream */ |
Diego Biurrun | bc5c918 | 2008-10-03 10:16:29 | [diff] [blame] | 363 | int64_t segment_start; |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 364 | |
Diego Biurrun | 5968d2d | 2008-08-05 08:28:57 | [diff] [blame] | 365 | /* the packet queue */ |
James Almer | 78b96be | 2018-03-26 18:02:37 | [diff] [blame] | 366 | AVPacketList *queue; |
| 367 | AVPacketList *queue_end; |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 368 | |
Aurelien Jacobs | 8d75b5a | 2007-06-04 22:35:16 | [diff] [blame] | 369 | int done; |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 370 | |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 371 | /* What to skip before effectively reading a packet. */ |
| 372 | int skip_to_keyframe; |
Aurelien Jacobs | 20f7466 | 2008-09-09 12:01:51 | [diff] [blame] | 373 | uint64_t skip_to_timecode; |
Aaron Colwell | 31ad14c | 2011-07-09 05:48:43 | [diff] [blame] | 374 | |
| 375 | /* File has a CUES element, but we defer parsing until it is needed. */ |
| 376 | int cues_parsing_deferred; |
Dale Curtis | 8336eb6 | 2012-04-19 18:12:24 | [diff] [blame] | 377 | |
wm4 | cac2295 | 2015-02-09 19:39:00 | [diff] [blame] | 378 | /* Level1 elements and whether they were read yet */ |
| 379 | MatroskaLevel1Element level1_elems[64]; |
| 380 | int num_level1_elems; |
| 381 | |
Dale Curtis | 8336eb6 | 2012-04-19 18:12:24 | [diff] [blame] | 382 | MatroskaCluster current_cluster; |
| 383 | |
James Zern | 20aeee4 | 2017-04-17 17:59:31 | [diff] [blame] | 384 | /* WebM DASH Manifest live flag */ |
Vignesh Venkatasubramanian | 2171b7c | 2015-03-31 23:51:57 | [diff] [blame] | 385 | int is_live; |
Vignesh Venkatasubramanian | 62c27fd | 2017-04-12 04:33:28 | [diff] [blame] | 386 | |
| 387 | /* Bandwidth value for WebM DASH Manifest */ |
| 388 | int bandwidth; |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 389 | } MatroskaDemuxContext; |
| 390 | |
Andreas Rheinhardt | c1abd95 | 2019-05-16 22:30:06 | [diff] [blame] | 391 | #define CHILD_OF(parent) { .def = { .n = parent } } |
| 392 | |
Andreas Rheinhardt | ab4795a | 2019-07-17 03:29:40 | [diff] [blame] | 393 | // The following forward declarations need their size because |
| 394 | // a tentative definition with internal linkage must not be an |
| 395 | // incomplete type (6.7.2 in C90, 6.9.2 in C99). |
| 396 | // Removing the sizes breaks MSVC. |
Andreas Rheinhardt | 9869e21 | 2019-07-18 19:07:20 | [diff] [blame] | 397 | static EbmlSyntax ebml_syntax[3], matroska_segment[9], matroska_track_video_color[15], matroska_track_video[19], |
| 398 | matroska_track[27], matroska_track_encoding[6], matroska_track_encodings[2], |
| 399 | matroska_track_combine_planes[2], matroska_track_operation[2], matroska_tracks[2], |
| 400 | matroska_attachments[2], matroska_chapter_entry[9], matroska_chapter[6], matroska_chapters[2], |
| 401 | matroska_index_entry[3], matroska_index[2], matroska_tag[3], matroska_tags[2], matroska_seekhead[2], |
| 402 | matroska_blockadditions[2], matroska_blockgroup[8], matroska_cluster_parsing[8]; |
Andreas Rheinhardt | c1abd95 | 2019-05-16 22:30:06 | [diff] [blame] | 403 | |
Andreas Rheinhardt | 9869e21 | 2019-07-18 19:07:20 | [diff] [blame] | 404 | static EbmlSyntax ebml_header[] = { |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 405 | { EBML_ID_EBMLREADVERSION, EBML_UINT, 0, offsetof(Ebml, version), { .u = EBML_VERSION } }, |
| 406 | { EBML_ID_EBMLMAXSIZELENGTH, EBML_UINT, 0, offsetof(Ebml, max_size), { .u = 8 } }, |
| 407 | { EBML_ID_EBMLMAXIDLENGTH, EBML_UINT, 0, offsetof(Ebml, id_length), { .u = 4 } }, |
| 408 | { EBML_ID_DOCTYPE, EBML_STR, 0, offsetof(Ebml, doctype), { .s = "(none)" } }, |
| 409 | { EBML_ID_DOCTYPEREADVERSION, EBML_UINT, 0, offsetof(Ebml, doctype_version), { .u = 1 } }, |
| 410 | { EBML_ID_EBMLVERSION, EBML_NONE }, |
| 411 | { EBML_ID_DOCTYPEVERSION, EBML_NONE }, |
Andreas Rheinhardt | c1abd95 | 2019-05-16 22:30:06 | [diff] [blame] | 412 | CHILD_OF(ebml_syntax) |
Aurelien Jacobs | 6351132 | 2008-08-05 00:40:02 | [diff] [blame] | 413 | }; |
| 414 | |
Andreas Rheinhardt | 9869e21 | 2019-07-18 19:07:20 | [diff] [blame] | 415 | static EbmlSyntax ebml_syntax[] = { |
Andreas Rheinhardt | c1abd95 | 2019-05-16 22:30:06 | [diff] [blame] | 416 | { EBML_ID_HEADER, EBML_NEST, 0, 0, { .n = ebml_header } }, |
| 417 | { MATROSKA_ID_SEGMENT, EBML_STOP }, |
Aurelien Jacobs | 6351132 | 2008-08-05 00:40:02 | [diff] [blame] | 418 | { 0 } |
| 419 | }; |
| 420 | |
Andreas Rheinhardt | 9869e21 | 2019-07-18 19:07:20 | [diff] [blame] | 421 | static EbmlSyntax matroska_info[] = { |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 422 | { MATROSKA_ID_TIMECODESCALE, EBML_UINT, 0, offsetof(MatroskaDemuxContext, time_scale), { .u = 1000000 } }, |
| 423 | { MATROSKA_ID_DURATION, EBML_FLOAT, 0, offsetof(MatroskaDemuxContext, duration) }, |
| 424 | { MATROSKA_ID_TITLE, EBML_UTF8, 0, offsetof(MatroskaDemuxContext, title) }, |
| 425 | { MATROSKA_ID_WRITINGAPP, EBML_NONE }, |
Michael Niedermayer | b3d9ab1 | 2014-03-07 22:46:37 | [diff] [blame] | 426 | { MATROSKA_ID_MUXINGAPP, EBML_UTF8, 0, offsetof(MatroskaDemuxContext, muxingapp) }, |
| 427 | { MATROSKA_ID_DATEUTC, EBML_BIN, 0, offsetof(MatroskaDemuxContext, date_utc) }, |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 428 | { MATROSKA_ID_SEGMENTUID, EBML_NONE }, |
Andreas Rheinhardt | c1abd95 | 2019-05-16 22:30:06 | [diff] [blame] | 429 | CHILD_OF(matroska_segment) |
Aurelien Jacobs | 2970858 | 2008-08-05 00:40:27 | [diff] [blame] | 430 | }; |
| 431 | |
Andreas Rheinhardt | 9869e21 | 2019-07-18 19:07:20 | [diff] [blame] | 432 | static EbmlSyntax matroska_mastering_meta[] = { |
Neil Birkbeck | bbda13a | 2016-03-01 00:41:04 | [diff] [blame] | 433 | { MATROSKA_ID_VIDEOCOLOR_RX, EBML_FLOAT, 0, offsetof(MatroskaMasteringMeta, r_x), { .f=-1 } }, |
| 434 | { MATROSKA_ID_VIDEOCOLOR_RY, EBML_FLOAT, 0, offsetof(MatroskaMasteringMeta, r_y), { .f=-1 } }, |
| 435 | { MATROSKA_ID_VIDEOCOLOR_GX, EBML_FLOAT, 0, offsetof(MatroskaMasteringMeta, g_x), { .f=-1 } }, |
| 436 | { MATROSKA_ID_VIDEOCOLOR_GY, EBML_FLOAT, 0, offsetof(MatroskaMasteringMeta, g_y), { .f=-1 } }, |
| 437 | { MATROSKA_ID_VIDEOCOLOR_BX, EBML_FLOAT, 0, offsetof(MatroskaMasteringMeta, b_x), { .f=-1 } }, |
| 438 | { MATROSKA_ID_VIDEOCOLOR_BY, EBML_FLOAT, 0, offsetof(MatroskaMasteringMeta, b_y), { .f=-1 } }, |
| 439 | { MATROSKA_ID_VIDEOCOLOR_WHITEX, EBML_FLOAT, 0, offsetof(MatroskaMasteringMeta, white_x), { .f=-1 } }, |
| 440 | { MATROSKA_ID_VIDEOCOLOR_WHITEY, EBML_FLOAT, 0, offsetof(MatroskaMasteringMeta, white_y), { .f=-1 } }, |
| 441 | { MATROSKA_ID_VIDEOCOLOR_LUMINANCEMIN, EBML_FLOAT, 0, offsetof(MatroskaMasteringMeta, min_luminance), { .f=-1 } }, |
| 442 | { MATROSKA_ID_VIDEOCOLOR_LUMINANCEMAX, EBML_FLOAT, 0, offsetof(MatroskaMasteringMeta, max_luminance), { .f=-1 } }, |
Andreas Rheinhardt | c1abd95 | 2019-05-16 22:30:06 | [diff] [blame] | 443 | CHILD_OF(matroska_track_video_color) |
Neil Birkbeck | bbda13a | 2016-03-01 00:41:04 | [diff] [blame] | 444 | }; |
| 445 | |
Andreas Rheinhardt | 9869e21 | 2019-07-18 19:07:20 | [diff] [blame] | 446 | static EbmlSyntax matroska_track_video_color[] = { |
James Almer | f5cfc0c | 2016-10-19 02:40:09 | [diff] [blame] | 447 | { MATROSKA_ID_VIDEOCOLORMATRIXCOEFF, EBML_UINT, 0, offsetof(MatroskaTrackVideoColor, matrix_coefficients), { .u = AVCOL_SPC_UNSPECIFIED } }, |
James Almer | 0ad71ed | 2016-10-19 02:55:32 | [diff] [blame] | 448 | { MATROSKA_ID_VIDEOCOLORBITSPERCHANNEL, EBML_UINT, 0, offsetof(MatroskaTrackVideoColor, bits_per_channel), { .u=0 } }, |
Neil Birkbeck | bbda13a | 2016-03-01 00:41:04 | [diff] [blame] | 449 | { MATROSKA_ID_VIDEOCOLORCHROMASUBHORZ, EBML_UINT, 0, offsetof(MatroskaTrackVideoColor, chroma_sub_horz), { .u=0 } }, |
| 450 | { MATROSKA_ID_VIDEOCOLORCHROMASUBVERT, EBML_UINT, 0, offsetof(MatroskaTrackVideoColor, chroma_sub_vert), { .u=0 } }, |
| 451 | { MATROSKA_ID_VIDEOCOLORCBSUBHORZ, EBML_UINT, 0, offsetof(MatroskaTrackVideoColor, cb_sub_horz), { .u=0 } }, |
| 452 | { MATROSKA_ID_VIDEOCOLORCBSUBVERT, EBML_UINT, 0, offsetof(MatroskaTrackVideoColor, cb_sub_vert), { .u=0 } }, |
James Almer | f5cfc0c | 2016-10-19 02:40:09 | [diff] [blame] | 453 | { MATROSKA_ID_VIDEOCOLORCHROMASITINGHORZ, EBML_UINT, 0, offsetof(MatroskaTrackVideoColor, chroma_siting_horz), { .u = MATROSKA_COLOUR_CHROMASITINGHORZ_UNDETERMINED } }, |
| 454 | { MATROSKA_ID_VIDEOCOLORCHROMASITINGVERT, EBML_UINT, 0, offsetof(MatroskaTrackVideoColor, chroma_siting_vert), { .u = MATROSKA_COLOUR_CHROMASITINGVERT_UNDETERMINED } }, |
| 455 | { MATROSKA_ID_VIDEOCOLORRANGE, EBML_UINT, 0, offsetof(MatroskaTrackVideoColor, range), { .u = AVCOL_RANGE_UNSPECIFIED } }, |
| 456 | { MATROSKA_ID_VIDEOCOLORTRANSFERCHARACTERISTICS, EBML_UINT, 0, offsetof(MatroskaTrackVideoColor, transfer_characteristics), { .u = AVCOL_TRC_UNSPECIFIED } }, |
| 457 | { MATROSKA_ID_VIDEOCOLORPRIMARIES, EBML_UINT, 0, offsetof(MatroskaTrackVideoColor, primaries), { .u = AVCOL_PRI_UNSPECIFIED } }, |
Neil Birkbeck | bbda13a | 2016-03-01 00:41:04 | [diff] [blame] | 458 | { MATROSKA_ID_VIDEOCOLORMAXCLL, EBML_UINT, 0, offsetof(MatroskaTrackVideoColor, max_cll), { .u=0 } }, |
| 459 | { MATROSKA_ID_VIDEOCOLORMAXFALL, EBML_UINT, 0, offsetof(MatroskaTrackVideoColor, max_fall), { .u=0 } }, |
| 460 | { MATROSKA_ID_VIDEOCOLORMASTERINGMETA, EBML_NEST, 0, offsetof(MatroskaTrackVideoColor, mastering_meta), { .n = matroska_mastering_meta } }, |
Andreas Rheinhardt | c1abd95 | 2019-05-16 22:30:06 | [diff] [blame] | 461 | CHILD_OF(matroska_track_video) |
Neil Birkbeck | bbda13a | 2016-03-01 00:41:04 | [diff] [blame] | 462 | }; |
| 463 | |
Andreas Rheinhardt | 9869e21 | 2019-07-18 19:07:20 | [diff] [blame] | 464 | static EbmlSyntax matroska_track_video_projection[] = { |
James Almer | 445204c | 2016-12-06 17:48:45 | [diff] [blame] | 465 | { MATROSKA_ID_VIDEOPROJECTIONTYPE, EBML_UINT, 0, offsetof(MatroskaTrackVideoProjection, type), { .u = MATROSKA_VIDEO_PROJECTION_TYPE_RECTANGULAR } }, |
| 466 | { MATROSKA_ID_VIDEOPROJECTIONPRIVATE, EBML_BIN, 0, offsetof(MatroskaTrackVideoProjection, private) }, |
| 467 | { MATROSKA_ID_VIDEOPROJECTIONPOSEYAW, EBML_FLOAT, 0, offsetof(MatroskaTrackVideoProjection, yaw), { .f=0.0 } }, |
| 468 | { MATROSKA_ID_VIDEOPROJECTIONPOSEPITCH, EBML_FLOAT, 0, offsetof(MatroskaTrackVideoProjection, pitch), { .f=0.0 } }, |
| 469 | { MATROSKA_ID_VIDEOPROJECTIONPOSEROLL, EBML_FLOAT, 0, offsetof(MatroskaTrackVideoProjection, roll), { .f=0.0 } }, |
Andreas Rheinhardt | c1abd95 | 2019-05-16 22:30:06 | [diff] [blame] | 470 | CHILD_OF(matroska_track_video) |
James Almer | 445204c | 2016-12-06 17:48:45 | [diff] [blame] | 471 | }; |
| 472 | |
Andreas Rheinhardt | 9869e21 | 2019-07-18 19:07:20 | [diff] [blame] | 473 | static EbmlSyntax matroska_track_video[] = { |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 474 | { MATROSKA_ID_VIDEOFRAMERATE, EBML_FLOAT, 0, offsetof(MatroskaTrackVideo, frame_rate) }, |
Michael Niedermayer | b3d9ab1 | 2014-03-07 22:46:37 | [diff] [blame] | 475 | { MATROSKA_ID_VIDEODISPLAYWIDTH, EBML_UINT, 0, offsetof(MatroskaTrackVideo, display_width), { .u=-1 } }, |
| 476 | { MATROSKA_ID_VIDEODISPLAYHEIGHT, EBML_UINT, 0, offsetof(MatroskaTrackVideo, display_height), { .u=-1 } }, |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 477 | { MATROSKA_ID_VIDEOPIXELWIDTH, EBML_UINT, 0, offsetof(MatroskaTrackVideo, pixel_width) }, |
| 478 | { MATROSKA_ID_VIDEOPIXELHEIGHT, EBML_UINT, 0, offsetof(MatroskaTrackVideo, pixel_height) }, |
Michael Niedermayer | b3d9ab1 | 2014-03-07 22:46:37 | [diff] [blame] | 479 | { MATROSKA_ID_VIDEOCOLORSPACE, EBML_BIN, 0, offsetof(MatroskaTrackVideo, color_space) }, |
Michael Niedermayer | b3d9ab1 | 2014-03-07 22:46:37 | [diff] [blame] | 480 | { MATROSKA_ID_VIDEOALPHAMODE, EBML_UINT, 0, offsetof(MatroskaTrackVideo, alpha_mode) }, |
James Almer | 4e75907 | 2016-12-05 02:22:39 | [diff] [blame] | 481 | { MATROSKA_ID_VIDEOCOLOR, EBML_NEST, sizeof(MatroskaTrackVideoColor), offsetof(MatroskaTrackVideo, color), { .n = matroska_track_video_color } }, |
James Almer | 445204c | 2016-12-06 17:48:45 | [diff] [blame] | 482 | { MATROSKA_ID_VIDEOPROJECTION, EBML_NEST, 0, offsetof(MatroskaTrackVideo, projection), { .n = matroska_track_video_projection } }, |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 483 | { MATROSKA_ID_VIDEOPIXELCROPB, EBML_NONE }, |
| 484 | { MATROSKA_ID_VIDEOPIXELCROPT, EBML_NONE }, |
| 485 | { MATROSKA_ID_VIDEOPIXELCROPL, EBML_NONE }, |
| 486 | { MATROSKA_ID_VIDEOPIXELCROPR, EBML_NONE }, |
James Almer | bad8bbc | 2016-10-15 21:01:50 | [diff] [blame] | 487 | { MATROSKA_ID_VIDEODISPLAYUNIT, EBML_UINT, 0, offsetof(MatroskaTrackVideo, display_unit), { .u= MATROSKA_VIDEO_DISPLAYUNIT_PIXELS } }, |
Luca Barbato | 5f02266 | 2016-03-28 18:29:54 | [diff] [blame] | 488 | { MATROSKA_ID_VIDEOFLAGINTERLACED, EBML_UINT, 0, offsetof(MatroskaTrackVideo, interlaced), { .u = MATROSKA_VIDEO_INTERLACE_FLAG_UNDETERMINED } }, |
| 489 | { MATROSKA_ID_VIDEOFIELDORDER, EBML_UINT, 0, offsetof(MatroskaTrackVideo, field_order), { .u = MATROSKA_VIDEO_FIELDORDER_UNDETERMINED } }, |
Vittorio Giovara | d4ae8ac | 2014-08-12 21:28:49 | [diff] [blame] | 490 | { MATROSKA_ID_VIDEOSTEREOMODE, EBML_UINT, 0, offsetof(MatroskaTrackVideo, stereo_mode), { .u = MATROSKA_VIDEO_STEREOMODE_TYPE_NB } }, |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 491 | { MATROSKA_ID_VIDEOASPECTRATIO, EBML_NONE }, |
Andreas Rheinhardt | c1abd95 | 2019-05-16 22:30:06 | [diff] [blame] | 492 | CHILD_OF(matroska_track) |
Aurelien Jacobs | 2cbc881 | 2008-08-05 00:40:31 | [diff] [blame] | 493 | }; |
| 494 | |
Andreas Rheinhardt | 9869e21 | 2019-07-18 19:07:20 | [diff] [blame] | 495 | static EbmlSyntax matroska_track_audio[] = { |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 496 | { MATROSKA_ID_AUDIOSAMPLINGFREQ, EBML_FLOAT, 0, offsetof(MatroskaTrackAudio, samplerate), { .f = 8000.0 } }, |
| 497 | { MATROSKA_ID_AUDIOOUTSAMPLINGFREQ, EBML_FLOAT, 0, offsetof(MatroskaTrackAudio, out_samplerate) }, |
| 498 | { MATROSKA_ID_AUDIOBITDEPTH, EBML_UINT, 0, offsetof(MatroskaTrackAudio, bitdepth) }, |
| 499 | { MATROSKA_ID_AUDIOCHANNELS, EBML_UINT, 0, offsetof(MatroskaTrackAudio, channels), { .u = 1 } }, |
Andreas Rheinhardt | c1abd95 | 2019-05-16 22:30:06 | [diff] [blame] | 500 | CHILD_OF(matroska_track) |
Aurelien Jacobs | 2cbc881 | 2008-08-05 00:40:31 | [diff] [blame] | 501 | }; |
| 502 | |
Andreas Rheinhardt | 9869e21 | 2019-07-18 19:07:20 | [diff] [blame] | 503 | static EbmlSyntax matroska_track_encoding_compression[] = { |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 504 | { MATROSKA_ID_ENCODINGCOMPALGO, EBML_UINT, 0, offsetof(MatroskaTrackCompression, algo), { .u = 0 } }, |
| 505 | { MATROSKA_ID_ENCODINGCOMPSETTINGS, EBML_BIN, 0, offsetof(MatroskaTrackCompression, settings) }, |
Andreas Rheinhardt | c1abd95 | 2019-05-16 22:30:06 | [diff] [blame] | 506 | CHILD_OF(matroska_track_encoding) |
Aurelien Jacobs | 2cbc881 | 2008-08-05 00:40:31 | [diff] [blame] | 507 | }; |
| 508 | |
Andreas Rheinhardt | 9869e21 | 2019-07-18 19:07:20 | [diff] [blame] | 509 | static EbmlSyntax matroska_track_encoding_encryption[] = { |
Michael Niedermayer | b3d9ab1 | 2014-03-07 22:46:37 | [diff] [blame] | 510 | { MATROSKA_ID_ENCODINGENCALGO, EBML_UINT, 0, offsetof(MatroskaTrackEncryption,algo), {.u = 0} }, |
Frank Galligan | b853103 | 2013-03-07 16:11:38 | [diff] [blame] | 511 | { MATROSKA_ID_ENCODINGENCKEYID, EBML_BIN, 0, offsetof(MatroskaTrackEncryption,key_id) }, |
| 512 | { MATROSKA_ID_ENCODINGENCAESSETTINGS, EBML_NONE }, |
| 513 | { MATROSKA_ID_ENCODINGSIGALGO, EBML_NONE }, |
| 514 | { MATROSKA_ID_ENCODINGSIGHASHALGO, EBML_NONE }, |
| 515 | { MATROSKA_ID_ENCODINGSIGKEYID, EBML_NONE }, |
| 516 | { MATROSKA_ID_ENCODINGSIGNATURE, EBML_NONE }, |
Andreas Rheinhardt | c1abd95 | 2019-05-16 22:30:06 | [diff] [blame] | 517 | CHILD_OF(matroska_track_encoding) |
Frank Galligan | b853103 | 2013-03-07 16:11:38 | [diff] [blame] | 518 | }; |
Andreas Rheinhardt | 9869e21 | 2019-07-18 19:07:20 | [diff] [blame] | 519 | static EbmlSyntax matroska_track_encoding[] = { |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 520 | { MATROSKA_ID_ENCODINGSCOPE, EBML_UINT, 0, offsetof(MatroskaTrackEncoding, scope), { .u = 1 } }, |
| 521 | { MATROSKA_ID_ENCODINGTYPE, EBML_UINT, 0, offsetof(MatroskaTrackEncoding, type), { .u = 0 } }, |
| 522 | { MATROSKA_ID_ENCODINGCOMPRESSION, EBML_NEST, 0, offsetof(MatroskaTrackEncoding, compression), { .n = matroska_track_encoding_compression } }, |
Michael Niedermayer | b3d9ab1 | 2014-03-07 22:46:37 | [diff] [blame] | 523 | { MATROSKA_ID_ENCODINGENCRYPTION, EBML_NEST, 0, offsetof(MatroskaTrackEncoding, encryption), { .n = matroska_track_encoding_encryption } }, |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 524 | { MATROSKA_ID_ENCODINGORDER, EBML_NONE }, |
Andreas Rheinhardt | c1abd95 | 2019-05-16 22:30:06 | [diff] [blame] | 525 | CHILD_OF(matroska_track_encodings) |
Aurelien Jacobs | 2cbc881 | 2008-08-05 00:40:31 | [diff] [blame] | 526 | }; |
| 527 | |
Andreas Rheinhardt | 9869e21 | 2019-07-18 19:07:20 | [diff] [blame] | 528 | static EbmlSyntax matroska_track_encodings[] = { |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 529 | { MATROSKA_ID_TRACKCONTENTENCODING, EBML_NEST, sizeof(MatroskaTrackEncoding), offsetof(MatroskaTrack, encodings), { .n = matroska_track_encoding } }, |
Andreas Rheinhardt | c1abd95 | 2019-05-16 22:30:06 | [diff] [blame] | 530 | CHILD_OF(matroska_track) |
Aurelien Jacobs | 2cbc881 | 2008-08-05 00:40:31 | [diff] [blame] | 531 | }; |
| 532 | |
Andreas Rheinhardt | 9869e21 | 2019-07-18 19:07:20 | [diff] [blame] | 533 | static EbmlSyntax matroska_track_plane[] = { |
Kirill Gavrilov | e6ec921 | 2011-05-21 15:14:14 | [diff] [blame] | 534 | { MATROSKA_ID_TRACKPLANEUID, EBML_UINT, 0, offsetof(MatroskaTrackPlane,uid) }, |
| 535 | { MATROSKA_ID_TRACKPLANETYPE, EBML_UINT, 0, offsetof(MatroskaTrackPlane,type) }, |
Andreas Rheinhardt | c1abd95 | 2019-05-16 22:30:06 | [diff] [blame] | 536 | CHILD_OF(matroska_track_combine_planes) |
Kirill Gavrilov | e6ec921 | 2011-05-21 15:14:14 | [diff] [blame] | 537 | }; |
| 538 | |
Andreas Rheinhardt | 9869e21 | 2019-07-18 19:07:20 | [diff] [blame] | 539 | static EbmlSyntax matroska_track_combine_planes[] = { |
Michael Niedermayer | b3d9ab1 | 2014-03-07 22:46:37 | [diff] [blame] | 540 | { MATROSKA_ID_TRACKPLANE, EBML_NEST, sizeof(MatroskaTrackPlane), offsetof(MatroskaTrackOperation,combine_planes), {.n = matroska_track_plane} }, |
Andreas Rheinhardt | c1abd95 | 2019-05-16 22:30:06 | [diff] [blame] | 541 | CHILD_OF(matroska_track_operation) |
Kirill Gavrilov | e6ec921 | 2011-05-21 15:14:14 | [diff] [blame] | 542 | }; |
| 543 | |
Andreas Rheinhardt | 9869e21 | 2019-07-18 19:07:20 | [diff] [blame] | 544 | static EbmlSyntax matroska_track_operation[] = { |
Michael Niedermayer | b3d9ab1 | 2014-03-07 22:46:37 | [diff] [blame] | 545 | { MATROSKA_ID_TRACKCOMBINEPLANES, EBML_NEST, 0, 0, {.n = matroska_track_combine_planes} }, |
Andreas Rheinhardt | c1abd95 | 2019-05-16 22:30:06 | [diff] [blame] | 546 | CHILD_OF(matroska_track) |
Kirill Gavrilov | e6ec921 | 2011-05-21 15:14:14 | [diff] [blame] | 547 | }; |
| 548 | |
Andreas Rheinhardt | 9869e21 | 2019-07-18 19:07:20 | [diff] [blame] | 549 | static EbmlSyntax matroska_track[] = { |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 550 | { MATROSKA_ID_TRACKNUMBER, EBML_UINT, 0, offsetof(MatroskaTrack, num) }, |
| 551 | { MATROSKA_ID_TRACKNAME, EBML_UTF8, 0, offsetof(MatroskaTrack, name) }, |
| 552 | { MATROSKA_ID_TRACKUID, EBML_UINT, 0, offsetof(MatroskaTrack, uid) }, |
| 553 | { MATROSKA_ID_TRACKTYPE, EBML_UINT, 0, offsetof(MatroskaTrack, type) }, |
| 554 | { MATROSKA_ID_CODECID, EBML_STR, 0, offsetof(MatroskaTrack, codec_id) }, |
| 555 | { MATROSKA_ID_CODECPRIVATE, EBML_BIN, 0, offsetof(MatroskaTrack, codec_priv) }, |
Anton Khirnov | eb3b550 | 2014-04-29 10:03:13 | [diff] [blame] | 556 | { MATROSKA_ID_CODECDELAY, EBML_UINT, 0, offsetof(MatroskaTrack, codec_delay) }, |
Steve Lhomme | b5dd964 | 2020-03-22 08:59:26 | [diff] [blame] | 557 | { MATROSKA_ID_TRACKLANGUAGE, EBML_STR, 0, offsetof(MatroskaTrack, language), { .s = "eng" } }, |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 558 | { MATROSKA_ID_TRACKDEFAULTDURATION, EBML_UINT, 0, offsetof(MatroskaTrack, default_duration) }, |
Michael Niedermayer | b3d9ab1 | 2014-03-07 22:46:37 | [diff] [blame] | 559 | { MATROSKA_ID_TRACKTIMECODESCALE, EBML_FLOAT, 0, offsetof(MatroskaTrack, time_scale), { .f = 1.0 } }, |
| 560 | { MATROSKA_ID_TRACKFLAGDEFAULT, EBML_UINT, 0, offsetof(MatroskaTrack, flag_default), { .u = 1 } }, |
| 561 | { MATROSKA_ID_TRACKFLAGFORCED, EBML_UINT, 0, offsetof(MatroskaTrack, flag_forced), { .u = 0 } }, |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 562 | { MATROSKA_ID_TRACKVIDEO, EBML_NEST, 0, offsetof(MatroskaTrack, video), { .n = matroska_track_video } }, |
| 563 | { MATROSKA_ID_TRACKAUDIO, EBML_NEST, 0, offsetof(MatroskaTrack, audio), { .n = matroska_track_audio } }, |
Michael Niedermayer | b3d9ab1 | 2014-03-07 22:46:37 | [diff] [blame] | 564 | { MATROSKA_ID_TRACKOPERATION, EBML_NEST, 0, offsetof(MatroskaTrack, operation), { .n = matroska_track_operation } }, |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 565 | { MATROSKA_ID_TRACKCONTENTENCODINGS, EBML_NEST, 0, 0, { .n = matroska_track_encodings } }, |
Michael Niedermayer | b3d9ab1 | 2014-03-07 22:46:37 | [diff] [blame] | 566 | { MATROSKA_ID_TRACKMAXBLKADDID, EBML_UINT, 0, offsetof(MatroskaTrack, max_block_additional_id) }, |
Michael Niedermayer | b3d9ab1 | 2014-03-07 22:46:37 | [diff] [blame] | 567 | { MATROSKA_ID_SEEKPREROLL, EBML_UINT, 0, offsetof(MatroskaTrack, seek_preroll) }, |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 568 | { MATROSKA_ID_TRACKFLAGENABLED, EBML_NONE }, |
| 569 | { MATROSKA_ID_TRACKFLAGLACING, EBML_NONE }, |
| 570 | { MATROSKA_ID_CODECNAME, EBML_NONE }, |
| 571 | { MATROSKA_ID_CODECDECODEALL, EBML_NONE }, |
| 572 | { MATROSKA_ID_CODECINFOURL, EBML_NONE }, |
| 573 | { MATROSKA_ID_CODECDOWNLOADURL, EBML_NONE }, |
| 574 | { MATROSKA_ID_TRACKMINCACHE, EBML_NONE }, |
| 575 | { MATROSKA_ID_TRACKMAXCACHE, EBML_NONE }, |
Andreas Rheinhardt | c1abd95 | 2019-05-16 22:30:06 | [diff] [blame] | 576 | CHILD_OF(matroska_tracks) |
Aurelien Jacobs | 2cbc881 | 2008-08-05 00:40:31 | [diff] [blame] | 577 | }; |
| 578 | |
Andreas Rheinhardt | 9869e21 | 2019-07-18 19:07:20 | [diff] [blame] | 579 | static EbmlSyntax matroska_tracks[] = { |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 580 | { MATROSKA_ID_TRACKENTRY, EBML_NEST, sizeof(MatroskaTrack), offsetof(MatroskaDemuxContext, tracks), { .n = matroska_track } }, |
Andreas Rheinhardt | c1abd95 | 2019-05-16 22:30:06 | [diff] [blame] | 581 | CHILD_OF(matroska_segment) |
Aurelien Jacobs | 2cbc881 | 2008-08-05 00:40:31 | [diff] [blame] | 582 | }; |
| 583 | |
Andreas Rheinhardt | 9869e21 | 2019-07-18 19:07:20 | [diff] [blame] | 584 | static EbmlSyntax matroska_attachment[] = { |
Diego Biurrun | f69befe | 2014-03-07 12:54:18 | [diff] [blame] | 585 | { MATROSKA_ID_FILEUID, EBML_UINT, 0, offsetof(MatroskaAttachment, uid) }, |
| 586 | { MATROSKA_ID_FILENAME, EBML_UTF8, 0, offsetof(MatroskaAttachment, filename) }, |
| 587 | { MATROSKA_ID_FILEMIMETYPE, EBML_STR, 0, offsetof(MatroskaAttachment, mime) }, |
| 588 | { MATROSKA_ID_FILEDATA, EBML_BIN, 0, offsetof(MatroskaAttachment, bin) }, |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 589 | { MATROSKA_ID_FILEDESC, EBML_NONE }, |
Andreas Rheinhardt | c1abd95 | 2019-05-16 22:30:06 | [diff] [blame] | 590 | CHILD_OF(matroska_attachments) |
Aurelien Jacobs | b414cb8 | 2008-08-05 00:40:24 | [diff] [blame] | 591 | }; |
| 592 | |
Andreas Rheinhardt | 9869e21 | 2019-07-18 19:07:20 | [diff] [blame] | 593 | static EbmlSyntax matroska_attachments[] = { |
Diego Biurrun | f69befe | 2014-03-07 12:54:18 | [diff] [blame] | 594 | { MATROSKA_ID_ATTACHEDFILE, EBML_NEST, sizeof(MatroskaAttachment), offsetof(MatroskaDemuxContext, attachments), { .n = matroska_attachment } }, |
Andreas Rheinhardt | c1abd95 | 2019-05-16 22:30:06 | [diff] [blame] | 595 | CHILD_OF(matroska_segment) |
Aurelien Jacobs | b414cb8 | 2008-08-05 00:40:24 | [diff] [blame] | 596 | }; |
| 597 | |
Andreas Rheinhardt | 9869e21 | 2019-07-18 19:07:20 | [diff] [blame] | 598 | static EbmlSyntax matroska_chapter_display[] = { |
Rodger Combs | cf2719a | 2015-09-20 14:34:05 | [diff] [blame] | 599 | { MATROSKA_ID_CHAPSTRING, EBML_UTF8, 0, offsetof(MatroskaChapter, title) }, |
| 600 | { MATROSKA_ID_CHAPLANG, EBML_NONE }, |
| 601 | { MATROSKA_ID_CHAPCOUNTRY, EBML_NONE }, |
Andreas Rheinhardt | c1abd95 | 2019-05-16 22:30:06 | [diff] [blame] | 602 | CHILD_OF(matroska_chapter_entry) |
Aurelien Jacobs | 6bbd7c7 | 2008-08-05 00:40:21 | [diff] [blame] | 603 | }; |
| 604 | |
Andreas Rheinhardt | 9869e21 | 2019-07-18 19:07:20 | [diff] [blame] | 605 | static EbmlSyntax matroska_chapter_entry[] = { |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 606 | { MATROSKA_ID_CHAPTERTIMESTART, EBML_UINT, 0, offsetof(MatroskaChapter, start), { .u = AV_NOPTS_VALUE } }, |
| 607 | { MATROSKA_ID_CHAPTERTIMEEND, EBML_UINT, 0, offsetof(MatroskaChapter, end), { .u = AV_NOPTS_VALUE } }, |
| 608 | { MATROSKA_ID_CHAPTERUID, EBML_UINT, 0, offsetof(MatroskaChapter, uid) }, |
| 609 | { MATROSKA_ID_CHAPTERDISPLAY, EBML_NEST, 0, 0, { .n = matroska_chapter_display } }, |
Aurelien Jacobs | 6bbd7c7 | 2008-08-05 00:40:21 | [diff] [blame] | 610 | { MATROSKA_ID_CHAPTERFLAGHIDDEN, EBML_NONE }, |
Aurelien Jacobs | 5df3cc6 | 2008-08-13 21:15:15 | [diff] [blame] | 611 | { MATROSKA_ID_CHAPTERFLAGENABLED, EBML_NONE }, |
| 612 | { MATROSKA_ID_CHAPTERPHYSEQUIV, EBML_NONE }, |
| 613 | { MATROSKA_ID_CHAPTERATOM, EBML_NONE }, |
Andreas Rheinhardt | c1abd95 | 2019-05-16 22:30:06 | [diff] [blame] | 614 | CHILD_OF(matroska_chapter) |
Aurelien Jacobs | 6bbd7c7 | 2008-08-05 00:40:21 | [diff] [blame] | 615 | }; |
| 616 | |
Andreas Rheinhardt | 9869e21 | 2019-07-18 19:07:20 | [diff] [blame] | 617 | static EbmlSyntax matroska_chapter[] = { |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 618 | { MATROSKA_ID_CHAPTERATOM, EBML_NEST, sizeof(MatroskaChapter), offsetof(MatroskaDemuxContext, chapters), { .n = matroska_chapter_entry } }, |
Aurelien Jacobs | 6bbd7c7 | 2008-08-05 00:40:21 | [diff] [blame] | 619 | { MATROSKA_ID_EDITIONUID, EBML_NONE }, |
| 620 | { MATROSKA_ID_EDITIONFLAGHIDDEN, EBML_NONE }, |
| 621 | { MATROSKA_ID_EDITIONFLAGDEFAULT, EBML_NONE }, |
Aurelien Jacobs | 5df3cc6 | 2008-08-13 21:15:15 | [diff] [blame] | 622 | { MATROSKA_ID_EDITIONFLAGORDERED, EBML_NONE }, |
Andreas Rheinhardt | c1abd95 | 2019-05-16 22:30:06 | [diff] [blame] | 623 | CHILD_OF(matroska_chapters) |
Aurelien Jacobs | 6bbd7c7 | 2008-08-05 00:40:21 | [diff] [blame] | 624 | }; |
| 625 | |
Andreas Rheinhardt | 9869e21 | 2019-07-18 19:07:20 | [diff] [blame] | 626 | static EbmlSyntax matroska_chapters[] = { |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 627 | { MATROSKA_ID_EDITIONENTRY, EBML_NEST, 0, 0, { .n = matroska_chapter } }, |
Andreas Rheinhardt | c1abd95 | 2019-05-16 22:30:06 | [diff] [blame] | 628 | CHILD_OF(matroska_segment) |
Aurelien Jacobs | 6bbd7c7 | 2008-08-05 00:40:21 | [diff] [blame] | 629 | }; |
| 630 | |
Andreas Rheinhardt | 9869e21 | 2019-07-18 19:07:20 | [diff] [blame] | 631 | static EbmlSyntax matroska_index_pos[] = { |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 632 | { MATROSKA_ID_CUETRACK, EBML_UINT, 0, offsetof(MatroskaIndexPos, track) }, |
| 633 | { MATROSKA_ID_CUECLUSTERPOSITION, EBML_UINT, 0, offsetof(MatroskaIndexPos, pos) }, |
James Almer | 5ab7b3b | 2013-09-15 09:32:36 | [diff] [blame] | 634 | { MATROSKA_ID_CUERELATIVEPOSITION,EBML_NONE }, |
James Almer | 56f1740 | 2013-09-21 01:15:49 | [diff] [blame] | 635 | { MATROSKA_ID_CUEDURATION, EBML_NONE }, |
Aurelien Jacobs | 5df3cc6 | 2008-08-13 21:15:15 | [diff] [blame] | 636 | { MATROSKA_ID_CUEBLOCKNUMBER, EBML_NONE }, |
Andreas Rheinhardt | c1abd95 | 2019-05-16 22:30:06 | [diff] [blame] | 637 | CHILD_OF(matroska_index_entry) |
Aurelien Jacobs | e5929fd | 2008-08-05 00:40:15 | [diff] [blame] | 638 | }; |
| 639 | |
Andreas Rheinhardt | 9869e21 | 2019-07-18 19:07:20 | [diff] [blame] | 640 | static EbmlSyntax matroska_index_entry[] = { |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 641 | { MATROSKA_ID_CUETIME, EBML_UINT, 0, offsetof(MatroskaIndex, time) }, |
| 642 | { MATROSKA_ID_CUETRACKPOSITION, EBML_NEST, sizeof(MatroskaIndexPos), offsetof(MatroskaIndex, pos), { .n = matroska_index_pos } }, |
Andreas Rheinhardt | c1abd95 | 2019-05-16 22:30:06 | [diff] [blame] | 643 | CHILD_OF(matroska_index) |
Aurelien Jacobs | e5929fd | 2008-08-05 00:40:15 | [diff] [blame] | 644 | }; |
| 645 | |
Andreas Rheinhardt | 9869e21 | 2019-07-18 19:07:20 | [diff] [blame] | 646 | static EbmlSyntax matroska_index[] = { |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 647 | { MATROSKA_ID_POINTENTRY, EBML_NEST, sizeof(MatroskaIndex), offsetof(MatroskaDemuxContext, index), { .n = matroska_index_entry } }, |
Andreas Rheinhardt | c1abd95 | 2019-05-16 22:30:06 | [diff] [blame] | 648 | CHILD_OF(matroska_segment) |
Aurelien Jacobs | e5929fd | 2008-08-05 00:40:15 | [diff] [blame] | 649 | }; |
| 650 | |
Andreas Rheinhardt | 9869e21 | 2019-07-18 19:07:20 | [diff] [blame] | 651 | static EbmlSyntax matroska_simpletag[] = { |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 652 | { MATROSKA_ID_TAGNAME, EBML_UTF8, 0, offsetof(MatroskaTag, name) }, |
| 653 | { MATROSKA_ID_TAGSTRING, EBML_UTF8, 0, offsetof(MatroskaTag, string) }, |
| 654 | { MATROSKA_ID_TAGLANG, EBML_STR, 0, offsetof(MatroskaTag, lang), { .s = "und" } }, |
| 655 | { MATROSKA_ID_TAGDEFAULT, EBML_UINT, 0, offsetof(MatroskaTag, def) }, |
| 656 | { MATROSKA_ID_TAGDEFAULT_BUG, EBML_UINT, 0, offsetof(MatroskaTag, def) }, |
| 657 | { MATROSKA_ID_SIMPLETAG, EBML_NEST, sizeof(MatroskaTag), offsetof(MatroskaTag, sub), { .n = matroska_simpletag } }, |
Andreas Rheinhardt | c1abd95 | 2019-05-16 22:30:06 | [diff] [blame] | 658 | CHILD_OF(matroska_tag) |
Aurelien Jacobs | 44015c5 | 2008-08-08 23:50:38 | [diff] [blame] | 659 | }; |
| 660 | |
Andreas Rheinhardt | 9869e21 | 2019-07-18 19:07:20 | [diff] [blame] | 661 | static EbmlSyntax matroska_tagtargets[] = { |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 662 | { MATROSKA_ID_TAGTARGETS_TYPE, EBML_STR, 0, offsetof(MatroskaTagTarget, type) }, |
| 663 | { MATROSKA_ID_TAGTARGETS_TYPEVALUE, EBML_UINT, 0, offsetof(MatroskaTagTarget, typevalue), { .u = 50 } }, |
| 664 | { MATROSKA_ID_TAGTARGETS_TRACKUID, EBML_UINT, 0, offsetof(MatroskaTagTarget, trackuid) }, |
| 665 | { MATROSKA_ID_TAGTARGETS_CHAPTERUID, EBML_UINT, 0, offsetof(MatroskaTagTarget, chapteruid) }, |
| 666 | { MATROSKA_ID_TAGTARGETS_ATTACHUID, EBML_UINT, 0, offsetof(MatroskaTagTarget, attachuid) }, |
Andreas Rheinhardt | c1abd95 | 2019-05-16 22:30:06 | [diff] [blame] | 667 | CHILD_OF(matroska_tag) |
Aurelien Jacobs | 929e9de | 2009-02-15 15:53:55 | [diff] [blame] | 668 | }; |
| 669 | |
Andreas Rheinhardt | 9869e21 | 2019-07-18 19:07:20 | [diff] [blame] | 670 | static EbmlSyntax matroska_tag[] = { |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 671 | { MATROSKA_ID_SIMPLETAG, EBML_NEST, sizeof(MatroskaTag), offsetof(MatroskaTags, tag), { .n = matroska_simpletag } }, |
| 672 | { MATROSKA_ID_TAGTARGETS, EBML_NEST, 0, offsetof(MatroskaTags, target), { .n = matroska_tagtargets } }, |
Andreas Rheinhardt | c1abd95 | 2019-05-16 22:30:06 | [diff] [blame] | 673 | CHILD_OF(matroska_tags) |
Aurelien Jacobs | 44015c5 | 2008-08-08 23:50:38 | [diff] [blame] | 674 | }; |
| 675 | |
Andreas Rheinhardt | 9869e21 | 2019-07-18 19:07:20 | [diff] [blame] | 676 | static EbmlSyntax matroska_tags[] = { |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 677 | { MATROSKA_ID_TAG, EBML_NEST, sizeof(MatroskaTags), offsetof(MatroskaDemuxContext, tags), { .n = matroska_tag } }, |
Andreas Rheinhardt | c1abd95 | 2019-05-16 22:30:06 | [diff] [blame] | 678 | CHILD_OF(matroska_segment) |
Aurelien Jacobs | 434d496 | 2008-08-05 00:40:18 | [diff] [blame] | 679 | }; |
| 680 | |
Andreas Rheinhardt | 9869e21 | 2019-07-18 19:07:20 | [diff] [blame] | 681 | static EbmlSyntax matroska_seekhead_entry[] = { |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 682 | { MATROSKA_ID_SEEKID, EBML_UINT, 0, offsetof(MatroskaSeekhead, id) }, |
| 683 | { MATROSKA_ID_SEEKPOSITION, EBML_UINT, 0, offsetof(MatroskaSeekhead, pos), { .u = -1 } }, |
Andreas Rheinhardt | c1abd95 | 2019-05-16 22:30:06 | [diff] [blame] | 684 | CHILD_OF(matroska_seekhead) |
Aurelien Jacobs | 13b350a | 2008-08-05 00:40:36 | [diff] [blame] | 685 | }; |
| 686 | |
Andreas Rheinhardt | 9869e21 | 2019-07-18 19:07:20 | [diff] [blame] | 687 | static EbmlSyntax matroska_seekhead[] = { |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 688 | { MATROSKA_ID_SEEKENTRY, EBML_NEST, sizeof(MatroskaSeekhead), offsetof(MatroskaDemuxContext, seekhead), { .n = matroska_seekhead_entry } }, |
Andreas Rheinhardt | c1abd95 | 2019-05-16 22:30:06 | [diff] [blame] | 689 | CHILD_OF(matroska_segment) |
Aurelien Jacobs | 13b350a | 2008-08-05 00:40:36 | [diff] [blame] | 690 | }; |
| 691 | |
Andreas Rheinhardt | 9869e21 | 2019-07-18 19:07:20 | [diff] [blame] | 692 | static EbmlSyntax matroska_segment[] = { |
Andreas Rheinhardt | 71c9088 | 2019-05-16 22:30:18 | [diff] [blame] | 693 | { MATROSKA_ID_CLUSTER, EBML_STOP }, |
wm4 | cac2295 | 2015-02-09 19:39:00 | [diff] [blame] | 694 | { MATROSKA_ID_INFO, EBML_LEVEL1, 0, 0, { .n = matroska_info } }, |
| 695 | { MATROSKA_ID_TRACKS, EBML_LEVEL1, 0, 0, { .n = matroska_tracks } }, |
| 696 | { MATROSKA_ID_ATTACHMENTS, EBML_LEVEL1, 0, 0, { .n = matroska_attachments } }, |
| 697 | { MATROSKA_ID_CHAPTERS, EBML_LEVEL1, 0, 0, { .n = matroska_chapters } }, |
| 698 | { MATROSKA_ID_CUES, EBML_LEVEL1, 0, 0, { .n = matroska_index } }, |
| 699 | { MATROSKA_ID_TAGS, EBML_LEVEL1, 0, 0, { .n = matroska_tags } }, |
| 700 | { MATROSKA_ID_SEEKHEAD, EBML_LEVEL1, 0, 0, { .n = matroska_seekhead } }, |
Andreas Rheinhardt | c1abd95 | 2019-05-16 22:30:06 | [diff] [blame] | 701 | { 0 } /* We don't want to go back to level 0, so don't add the parent. */ |
Aurelien Jacobs | ce6f28b | 2008-08-05 00:40:58 | [diff] [blame] | 702 | }; |
| 703 | |
Andreas Rheinhardt | 9869e21 | 2019-07-18 19:07:20 | [diff] [blame] | 704 | static EbmlSyntax matroska_segments[] = { |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 705 | { MATROSKA_ID_SEGMENT, EBML_NEST, 0, 0, { .n = matroska_segment } }, |
Aurelien Jacobs | ce6f28b | 2008-08-05 00:40:58 | [diff] [blame] | 706 | { 0 } |
| 707 | }; |
| 708 | |
Andreas Rheinhardt | 9869e21 | 2019-07-18 19:07:20 | [diff] [blame] | 709 | static EbmlSyntax matroska_blockmore[] = { |
Andreas Rheinhardt | dbc50f8 | 2019-11-20 12:26:59 | [diff] [blame] | 710 | { MATROSKA_ID_BLOCKADDID, EBML_UINT, 0, offsetof(MatroskaBlock,additional_id), { .u = 1 } }, |
Vignesh Venkatasubramanian | 30c5c45 | 2013-02-13 21:51:48 | [diff] [blame] | 711 | { MATROSKA_ID_BLOCKADDITIONAL, EBML_BIN, 0, offsetof(MatroskaBlock,additional) }, |
Andreas Rheinhardt | c1abd95 | 2019-05-16 22:30:06 | [diff] [blame] | 712 | CHILD_OF(matroska_blockadditions) |
Vignesh Venkatasubramanian | 30c5c45 | 2013-02-13 21:51:48 | [diff] [blame] | 713 | }; |
| 714 | |
Andreas Rheinhardt | 9869e21 | 2019-07-18 19:07:20 | [diff] [blame] | 715 | static EbmlSyntax matroska_blockadditions[] = { |
Michael Niedermayer | b3d9ab1 | 2014-03-07 22:46:37 | [diff] [blame] | 716 | { MATROSKA_ID_BLOCKMORE, EBML_NEST, 0, 0, {.n = matroska_blockmore} }, |
Andreas Rheinhardt | c1abd95 | 2019-05-16 22:30:06 | [diff] [blame] | 717 | CHILD_OF(matroska_blockgroup) |
Vignesh Venkatasubramanian | 30c5c45 | 2013-02-13 21:51:48 | [diff] [blame] | 718 | }; |
| 719 | |
Andreas Rheinhardt | 9869e21 | 2019-07-18 19:07:20 | [diff] [blame] | 720 | static EbmlSyntax matroska_blockgroup[] = { |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 721 | { MATROSKA_ID_BLOCK, EBML_BIN, 0, offsetof(MatroskaBlock, bin) }, |
Michael Niedermayer | b3d9ab1 | 2014-03-07 22:46:37 | [diff] [blame] | 722 | { MATROSKA_ID_BLOCKADDITIONS, EBML_NEST, 0, 0, { .n = matroska_blockadditions} }, |
Michael Niedermayer | b3d9ab1 | 2014-03-07 22:46:37 | [diff] [blame] | 723 | { MATROSKA_ID_BLOCKDURATION, EBML_UINT, 0, offsetof(MatroskaBlock, duration) }, |
| 724 | { MATROSKA_ID_DISCARDPADDING, EBML_SINT, 0, offsetof(MatroskaBlock, discard_padding) }, |
Chris Cunningham | ac25840 | 2017-02-03 22:42:44 | [diff] [blame] | 725 | { MATROSKA_ID_BLOCKREFERENCE, EBML_SINT, 0, offsetof(MatroskaBlock, reference), { .i = INT64_MIN } }, |
Anton Khirnov | 564b7e0 | 2013-05-15 13:48:15 | [diff] [blame] | 726 | { MATROSKA_ID_CODECSTATE, EBML_NONE }, |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 727 | { 1, EBML_UINT, 0, offsetof(MatroskaBlock, non_simple), { .u = 1 } }, |
Andreas Rheinhardt | c1abd95 | 2019-05-16 22:30:06 | [diff] [blame] | 728 | CHILD_OF(matroska_cluster_parsing) |
Aurelien Jacobs | 209472b | 2008-08-05 00:41:05 | [diff] [blame] | 729 | }; |
| 730 | |
Andreas Rheinhardt | 38255cd | 2019-05-16 22:30:10 | [diff] [blame] | 731 | // The following array contains SimpleBlock and BlockGroup twice |
| 732 | // in order to reuse the other values for matroska_cluster_enter. |
Andreas Rheinhardt | 9869e21 | 2019-07-18 19:07:20 | [diff] [blame] | 733 | static EbmlSyntax matroska_cluster_parsing[] = { |
Andreas Rheinhardt | bc3306f | 2019-05-16 22:30:00 | [diff] [blame] | 734 | { MATROSKA_ID_SIMPLEBLOCK, EBML_BIN, 0, offsetof(MatroskaBlock, bin) }, |
Andreas Rheinhardt | 38255cd | 2019-05-16 22:30:10 | [diff] [blame] | 735 | { MATROSKA_ID_BLOCKGROUP, EBML_NEST, 0, 0, { .n = matroska_blockgroup } }, |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 736 | { MATROSKA_ID_CLUSTERTIMECODE, EBML_UINT, 0, offsetof(MatroskaCluster, timecode) }, |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 737 | { MATROSKA_ID_SIMPLEBLOCK, EBML_STOP }, |
Andreas Rheinhardt | 38255cd | 2019-05-16 22:30:10 | [diff] [blame] | 738 | { MATROSKA_ID_BLOCKGROUP, EBML_STOP }, |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 739 | { MATROSKA_ID_CLUSTERPOSITION, EBML_NONE }, |
| 740 | { MATROSKA_ID_CLUSTERPREVSIZE, EBML_NONE }, |
Andreas Rheinhardt | c1abd95 | 2019-05-16 22:30:06 | [diff] [blame] | 741 | CHILD_OF(matroska_segment) |
Dale Curtis | 8336eb6 | 2012-04-19 18:12:24 | [diff] [blame] | 742 | }; |
| 743 | |
Andreas Rheinhardt | 9869e21 | 2019-07-18 19:07:20 | [diff] [blame] | 744 | static EbmlSyntax matroska_cluster_enter[] = { |
Andreas Rheinhardt | 38255cd | 2019-05-16 22:30:10 | [diff] [blame] | 745 | { MATROSKA_ID_CLUSTER, EBML_NEST, 0, 0, { .n = &matroska_cluster_parsing[2] } }, |
Andreas Rheinhardt | 865c537 | 2019-05-16 22:30:08 | [diff] [blame] | 746 | { 0 } |
| 747 | }; |
Andreas Rheinhardt | c1abd95 | 2019-05-16 22:30:06 | [diff] [blame] | 748 | #undef CHILD_OF |
Dale Curtis | 8336eb6 | 2012-04-19 18:12:24 | [diff] [blame] | 749 | |
Andreas Rheinhardt | 67e957b | 2020-04-16 01:39:05 | [diff] [blame] | 750 | static const CodecMime mkv_image_mime_tags[] = { |
| 751 | {"image/gif" , AV_CODEC_ID_GIF}, |
| 752 | {"image/jpeg" , AV_CODEC_ID_MJPEG}, |
| 753 | {"image/png" , AV_CODEC_ID_PNG}, |
| 754 | {"image/tiff" , AV_CODEC_ID_TIFF}, |
| 755 | |
| 756 | {"" , AV_CODEC_ID_NONE} |
| 757 | }; |
| 758 | |
| 759 | static const CodecMime mkv_mime_tags[] = { |
| 760 | {"text/plain" , AV_CODEC_ID_TEXT}, |
| 761 | {"application/x-truetype-font", AV_CODEC_ID_TTF}, |
| 762 | {"application/x-font" , AV_CODEC_ID_TTF}, |
| 763 | {"application/vnd.ms-opentype", AV_CODEC_ID_OTF}, |
| 764 | {"binary" , AV_CODEC_ID_BIN_DATA}, |
| 765 | |
| 766 | {"" , AV_CODEC_ID_NONE} |
| 767 | }; |
| 768 | |
Alex Converse | b0f29db | 2012-02-20 08:42:33 | [diff] [blame] | 769 | static const char *const matroska_doctypes[] = { "matroska", "webm" }; |
James Zern | 470491f | 2010-05-22 01:41:32 | [diff] [blame] | 770 | |
Michael Niedermayer | f573acc | 2016-06-24 16:26:46 | [diff] [blame] | 771 | static int matroska_read_close(AVFormatContext *s); |
| 772 | |
Andreas Rheinhardt | 310f326 | 2019-05-16 22:30:02 | [diff] [blame] | 773 | /* |
| 774 | * This function prepares the status for parsing of level 1 elements. |
| 775 | */ |
| 776 | static int matroska_reset_status(MatroskaDemuxContext *matroska, |
| 777 | uint32_t id, int64_t position) |
| 778 | { |
| 779 | if (position >= 0) { |
Andreas Rheinhardt | c294f38 | 2019-08-17 00:27:51 | [diff] [blame] | 780 | int64_t err = avio_seek(matroska->ctx->pb, position, SEEK_SET); |
Andreas Rheinhardt | 310f326 | 2019-05-16 22:30:02 | [diff] [blame] | 781 | if (err < 0) |
| 782 | return err; |
| 783 | } |
| 784 | |
Andreas Rheinhardt | 5120305 | 2019-05-16 22:30:14 | [diff] [blame] | 785 | matroska->current_id = id; |
| 786 | matroska->num_levels = 1; |
| 787 | matroska->unknown_count = 0; |
Andreas Rheinhardt | a3db9f6 | 2019-05-16 22:30:05 | [diff] [blame] | 788 | matroska->resync_pos = avio_tell(matroska->ctx->pb); |
| 789 | if (id) |
| 790 | matroska->resync_pos -= (av_log2(id) + 7) / 8; |
Andreas Rheinhardt | 310f326 | 2019-05-16 22:30:02 | [diff] [blame] | 791 | |
| 792 | return 0; |
| 793 | } |
| 794 | |
Reimar Döffinger | d493170 | 2012-02-13 22:06:19 | [diff] [blame] | 795 | static int matroska_resync(MatroskaDemuxContext *matroska, int64_t last_pos) |
| 796 | { |
| 797 | AVIOContext *pb = matroska->ctx->pb; |
| 798 | uint32_t id; |
Reimar Döffinger | d493170 | 2012-02-13 22:06:19 | [diff] [blame] | 799 | |
Andreas Rheinhardt | 27f40b1 | 2019-05-16 22:30:01 | [diff] [blame] | 800 | /* Try to seek to the last position to resync from. If this doesn't work, |
| 801 | * we resync from the earliest position available: The start of the buffer. */ |
| 802 | if (last_pos < avio_tell(pb) && avio_seek(pb, last_pos + 1, SEEK_SET) < 0) { |
| 803 | av_log(matroska->ctx, AV_LOG_WARNING, |
| 804 | "Seek to desired resync point failed. Seeking to " |
| 805 | "earliest point available instead.\n"); |
| 806 | avio_seek(pb, FFMAX(avio_tell(pb) + (pb->buffer - pb->buf_ptr), |
| 807 | last_pos + 1), SEEK_SET); |
Sophia Wang | 8c83062 | 2016-09-27 19:00:29 | [diff] [blame] | 808 | } |
Reimar Döffinger | d493170 | 2012-02-13 22:06:19 | [diff] [blame] | 809 | |
| 810 | id = avio_rb32(pb); |
| 811 | |
| 812 | // try to find a toplevel element |
James Almer | d34ec64 | 2014-08-07 20:12:41 | [diff] [blame] | 813 | while (!avio_feof(pb)) { |
Sean McGovern | 8835c55 | 2013-05-27 22:11:50 | [diff] [blame] | 814 | if (id == MATROSKA_ID_INFO || id == MATROSKA_ID_TRACKS || |
| 815 | id == MATROSKA_ID_CUES || id == MATROSKA_ID_TAGS || |
Reimar Döffinger | d493170 | 2012-02-13 22:06:19 | [diff] [blame] | 816 | id == MATROSKA_ID_SEEKHEAD || id == MATROSKA_ID_ATTACHMENTS || |
Sean McGovern | 8835c55 | 2013-05-27 22:11:50 | [diff] [blame] | 817 | id == MATROSKA_ID_CLUSTER || id == MATROSKA_ID_CHAPTERS) { |
Andreas Rheinhardt | 8a286e7 | 2019-05-16 22:30:03 | [diff] [blame] | 818 | /* Prepare the context for parsing of a level 1 element. */ |
| 819 | matroska_reset_status(matroska, id, -1); |
Michael Niedermayer | fccc37c | 2019-08-28 17:04:42 | [diff] [blame] | 820 | /* Given that we are here means that an error has occurred, |
Andreas Rheinhardt | 8a286e7 | 2019-05-16 22:30:03 | [diff] [blame] | 821 | * so treat the segment as unknown length in order not to |
| 822 | * discard valid data that happens to be beyond the designated |
| 823 | * end of the segment. */ |
| 824 | matroska->levels[0].length = EBML_UNKNOWN_LENGTH; |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 825 | return 0; |
Reimar Döffinger | d493170 | 2012-02-13 22:06:19 | [diff] [blame] | 826 | } |
| 827 | id = (id << 8) | avio_r8(pb); |
| 828 | } |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 829 | |
Reimar Döffinger | d493170 | 2012-02-13 22:06:19 | [diff] [blame] | 830 | matroska->done = 1; |
Andreas Rheinhardt | 27f40b1 | 2019-05-16 22:30:01 | [diff] [blame] | 831 | return pb->error ? pb->error : AVERROR_EOF; |
Reimar Döffinger | d493170 | 2012-02-13 22:06:19 | [diff] [blame] | 832 | } |
| 833 | |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 834 | /* |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 835 | * Read: an "EBML number", which is defined as a variable-length |
| 836 | * array of bytes. The first byte indicates the length by giving a |
| 837 | * number of 0-bits followed by a one. The position of the first |
| 838 | * "one" bit inside the first byte indicates the length of this |
| 839 | * number. |
Diego Biurrun | 5968d2d | 2008-08-05 08:28:57 | [diff] [blame] | 840 | * Returns: number of bytes read, < 0 on error |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 841 | */ |
Anton Khirnov | 471fe57 | 2011-02-20 10:04:12 | [diff] [blame] | 842 | static int ebml_read_num(MatroskaDemuxContext *matroska, AVIOContext *pb, |
Andreas Rheinhardt | 239c736 | 2019-06-25 01:08:56 | [diff] [blame] | 843 | int max_size, uint64_t *number, int eof_forbidden) |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 844 | { |
Andreas Rheinhardt | 239c736 | 2019-06-25 01:08:56 | [diff] [blame] | 845 | int read, n = 1; |
| 846 | uint64_t total; |
| 847 | int64_t pos; |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 848 | |
Andreas Rheinhardt | 239c736 | 2019-06-25 01:08:56 | [diff] [blame] | 849 | /* The first byte tells us the length in bytes - except when it is zero. */ |
| 850 | total = avio_r8(pb); |
| 851 | if (pb->eof_reached) |
| 852 | goto err; |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 853 | |
| 854 | /* get the length of the EBML number */ |
Reimar Döffinger | ff6a5fc | 2010-09-02 19:17:46 | [diff] [blame] | 855 | read = 8 - ff_log2_tab[total]; |
Andreas Rheinhardt | 239c736 | 2019-06-25 01:08:56 | [diff] [blame] | 856 | |
| 857 | if (!total || read > max_size) { |
| 858 | pos = avio_tell(pb) - 1; |
| 859 | if (!total) { |
| 860 | av_log(matroska->ctx, AV_LOG_ERROR, |
| 861 | "0x00 at pos %"PRId64" (0x%"PRIx64") invalid as first byte " |
| 862 | "of an EBML number\n", pos, pos); |
| 863 | } else { |
| 864 | av_log(matroska->ctx, AV_LOG_ERROR, |
| 865 | "Length %d indicated by an EBML number's first byte 0x%02x " |
| 866 | "at pos %"PRId64" (0x%"PRIx64") exceeds max length %d.\n", |
| 867 | read, (uint8_t) total, pos, pos, max_size); |
| 868 | } |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 869 | return AVERROR_INVALIDDATA; |
| 870 | } |
| 871 | |
| 872 | /* read out length */ |
Reimar Döffinger | ff6a5fc | 2010-09-02 19:17:46 | [diff] [blame] | 873 | total ^= 1 << ff_log2_tab[total]; |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 874 | while (n++ < read) |
Anton Khirnov | e63a362 | 2011-02-21 15:43:01 | [diff] [blame] | 875 | total = (total << 8) | avio_r8(pb); |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 876 | |
Andreas Rheinhardt | 239c736 | 2019-06-25 01:08:56 | [diff] [blame] | 877 | if (pb->eof_reached) { |
| 878 | eof_forbidden = 1; |
| 879 | goto err; |
| 880 | } |
| 881 | |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 882 | *number = total; |
| 883 | |
| 884 | return read; |
Andreas Rheinhardt | 239c736 | 2019-06-25 01:08:56 | [diff] [blame] | 885 | |
| 886 | err: |
| 887 | pos = avio_tell(pb); |
| 888 | if (pb->error) { |
| 889 | av_log(matroska->ctx, AV_LOG_ERROR, |
| 890 | "Read error at pos. %"PRIu64" (0x%"PRIx64")\n", |
| 891 | pos, pos); |
| 892 | return pb->error; |
| 893 | } |
| 894 | if (eof_forbidden) { |
| 895 | av_log(matroska->ctx, AV_LOG_ERROR, "File ended prematurely " |
| 896 | "at pos. %"PRIu64" (0x%"PRIx64")\n", pos, pos); |
| 897 | return AVERROR(EIO); |
| 898 | } |
| 899 | return AVERROR_EOF; |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 900 | } |
| 901 | |
Reimar Döffinger | 1b4d327 | 2010-09-06 17:51:44 | [diff] [blame] | 902 | /** |
| 903 | * Read a EBML length value. |
| 904 | * This needs special handling for the "unknown length" case which has multiple |
| 905 | * encodings. |
| 906 | */ |
Anton Khirnov | 471fe57 | 2011-02-20 10:04:12 | [diff] [blame] | 907 | static int ebml_read_length(MatroskaDemuxContext *matroska, AVIOContext *pb, |
Reimar Döffinger | 1b4d327 | 2010-09-06 17:51:44 | [diff] [blame] | 908 | uint64_t *number) |
| 909 | { |
Andreas Rheinhardt | 239c736 | 2019-06-25 01:08:56 | [diff] [blame] | 910 | int res = ebml_read_num(matroska, pb, 8, number, 1); |
Reimar Döffinger | 1b4d327 | 2010-09-06 17:51:44 | [diff] [blame] | 911 | if (res > 0 && *number + 1 == 1ULL << (7 * res)) |
Steve Lhomme | a2c7702 | 2019-02-23 10:14:33 | [diff] [blame] | 912 | *number = EBML_UNKNOWN_LENGTH; |
Reimar Döffinger | 1b4d327 | 2010-09-06 17:51:44 | [diff] [blame] | 913 | return res; |
| 914 | } |
| 915 | |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 916 | /* |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 917 | * Read the next element as an unsigned int. |
Andreas Rheinhardt | a569a7b | 2019-06-23 23:42:30 | [diff] [blame] | 918 | * Returns NEEDS_CHECKING. |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 919 | */ |
Anton Khirnov | 471fe57 | 2011-02-20 10:04:12 | [diff] [blame] | 920 | static int ebml_read_uint(AVIOContext *pb, int size, uint64_t *num) |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 921 | { |
Aurelien Jacobs | c6cd2b3 | 2008-08-05 00:41:55 | [diff] [blame] | 922 | int n = 0; |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 923 | |
Diego Biurrun | 5968d2d | 2008-08-05 08:28:57 | [diff] [blame] | 924 | /* big-endian ordering; build up number */ |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 925 | *num = 0; |
| 926 | while (n++ < size) |
Anton Khirnov | e63a362 | 2011-02-21 15:43:01 | [diff] [blame] | 927 | *num = (*num << 8) | avio_r8(pb); |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 928 | |
Andreas Rheinhardt | a569a7b | 2019-06-23 23:42:30 | [diff] [blame] | 929 | return NEEDS_CHECKING; |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 930 | } |
| 931 | |
| 932 | /* |
Jan Gerber | d03eea3 | 2013-11-15 18:00:37 | [diff] [blame] | 933 | * Read the next element as a signed int. |
Andreas Rheinhardt | a569a7b | 2019-06-23 23:42:30 | [diff] [blame] | 934 | * Returns NEEDS_CHECKING. |
Jan Gerber | d03eea3 | 2013-11-15 18:00:37 | [diff] [blame] | 935 | */ |
| 936 | static int ebml_read_sint(AVIOContext *pb, int size, int64_t *num) |
| 937 | { |
| 938 | int n = 1; |
| 939 | |
Jan Gerber | d03eea3 | 2013-11-15 18:00:37 | [diff] [blame] | 940 | if (size == 0) { |
| 941 | *num = 0; |
| 942 | } else { |
Michael Niedermayer | cddd15b | 2013-11-15 20:30:30 | [diff] [blame] | 943 | *num = sign_extend(avio_r8(pb), 8); |
Jan Gerber | d03eea3 | 2013-11-15 18:00:37 | [diff] [blame] | 944 | |
| 945 | /* big-endian ordering; build up number */ |
| 946 | while (n++ < size) |
Michael Niedermayer | 2f8c816 | 2015-07-01 09:59:57 | [diff] [blame] | 947 | *num = ((uint64_t)*num << 8) | avio_r8(pb); |
Jan Gerber | d03eea3 | 2013-11-15 18:00:37 | [diff] [blame] | 948 | } |
| 949 | |
Andreas Rheinhardt | a569a7b | 2019-06-23 23:42:30 | [diff] [blame] | 950 | return NEEDS_CHECKING; |
Jan Gerber | d03eea3 | 2013-11-15 18:00:37 | [diff] [blame] | 951 | } |
| 952 | |
| 953 | /* |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 954 | * Read the next element as a float. |
Andreas Rheinhardt | a569a7b | 2019-06-23 23:42:30 | [diff] [blame] | 955 | * Returns NEEDS_CHECKING or < 0 on obvious failure. |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 956 | */ |
Anton Khirnov | 471fe57 | 2011-02-20 10:04:12 | [diff] [blame] | 957 | static int ebml_read_float(AVIOContext *pb, int size, double *num) |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 958 | { |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 959 | if (size == 0) |
Aurelien Jacobs | 4a194c8 | 2010-09-05 21:37:40 | [diff] [blame] | 960 | *num = 0; |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 961 | else if (size == 4) |
Mans Rullgard | 3383a53 | 2011-11-27 14:04:16 | [diff] [blame] | 962 | *num = av_int2float(avio_rb32(pb)); |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 963 | else if (size == 8) |
Mans Rullgard | 3383a53 | 2011-11-27 14:04:16 | [diff] [blame] | 964 | *num = av_int2double(avio_rb64(pb)); |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 965 | else |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 966 | return AVERROR_INVALIDDATA; |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 967 | |
Andreas Rheinhardt | a569a7b | 2019-06-23 23:42:30 | [diff] [blame] | 968 | return NEEDS_CHECKING; |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 969 | } |
| 970 | |
| 971 | /* |
| 972 | * Read the next element as an ASCII string. |
Andreas Rheinhardt | a569a7b | 2019-06-23 23:42:30 | [diff] [blame] | 973 | * 0 is success, < 0 or NEEDS_CHECKING is failure. |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 974 | */ |
Anton Khirnov | 471fe57 | 2011-02-20 10:04:12 | [diff] [blame] | 975 | static int ebml_read_ascii(AVIOContext *pb, int size, char **str) |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 976 | { |
Ronald S. Bultje | cd40c31 | 2012-02-25 00:12:18 | [diff] [blame] | 977 | char *res; |
Andreas Rheinhardt | a569a7b | 2019-06-23 23:42:30 | [diff] [blame] | 978 | int ret; |
Ronald S. Bultje | cd40c31 | 2012-02-25 00:12:18 | [diff] [blame] | 979 | |
Diego Biurrun | 5968d2d | 2008-08-05 08:28:57 | [diff] [blame] | 980 | /* EBML strings are usually not 0-terminated, so we allocate one |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 981 | * byte more, read the string and NULL-terminate it ourselves. */ |
Ronald S. Bultje | cd40c31 | 2012-02-25 00:12:18 | [diff] [blame] | 982 | if (!(res = av_malloc(size + 1))) |
Panagiotis Issaris | 769e10f | 2007-07-19 15:21:30 | [diff] [blame] | 983 | return AVERROR(ENOMEM); |
Andreas Rheinhardt | a569a7b | 2019-06-23 23:42:30 | [diff] [blame] | 984 | if ((ret = avio_read(pb, (uint8_t *) res, size)) != size) { |
Ronald S. Bultje | cd40c31 | 2012-02-25 00:12:18 | [diff] [blame] | 985 | av_free(res); |
Andreas Rheinhardt | a569a7b | 2019-06-23 23:42:30 | [diff] [blame] | 986 | return ret < 0 ? ret : NEEDS_CHECKING; |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 987 | } |
Ronald S. Bultje | cd40c31 | 2012-02-25 00:12:18 | [diff] [blame] | 988 | (res)[size] = '\0'; |
| 989 | av_free(*str); |
| 990 | *str = res; |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 991 | |
| 992 | return 0; |
| 993 | } |
| 994 | |
| 995 | /* |
Aurelien Jacobs | 737c40d | 2008-08-05 00:42:39 | [diff] [blame] | 996 | * Read the next element as binary data. |
Andreas Rheinhardt | a569a7b | 2019-06-23 23:42:30 | [diff] [blame] | 997 | * 0 is success, < 0 or NEEDS_CHECKING is failure. |
Aurelien Jacobs | 737c40d | 2008-08-05 00:42:39 | [diff] [blame] | 998 | */ |
Andreas Rheinhardt | 7087fc9 | 2019-05-16 22:30:12 | [diff] [blame] | 999 | static int ebml_read_binary(AVIOContext *pb, int length, |
| 1000 | int64_t pos, EbmlBin *bin) |
Aurelien Jacobs | 737c40d | 2008-08-05 00:42:39 | [diff] [blame] | 1001 | { |
James Almer | a618866 | 2018-04-04 17:12:54 | [diff] [blame] | 1002 | int ret; |
Aurelien Jacobs | 737c40d | 2008-08-05 00:42:39 | [diff] [blame] | 1003 | |
James Almer | a618866 | 2018-04-04 17:12:54 | [diff] [blame] | 1004 | ret = av_buffer_realloc(&bin->buf, length + AV_INPUT_BUFFER_PADDING_SIZE); |
| 1005 | if (ret < 0) |
| 1006 | return ret; |
| 1007 | memset(bin->buf->data + length, 0, AV_INPUT_BUFFER_PADDING_SIZE); |
| 1008 | |
| 1009 | bin->data = bin->buf->data; |
Aurelien Jacobs | 737c40d | 2008-08-05 00:42:39 | [diff] [blame] | 1010 | bin->size = length; |
Andreas Rheinhardt | 7087fc9 | 2019-05-16 22:30:12 | [diff] [blame] | 1011 | bin->pos = pos; |
Andreas Rheinhardt | a569a7b | 2019-06-23 23:42:30 | [diff] [blame] | 1012 | if ((ret = avio_read(pb, bin->data, length)) != length) { |
James Almer | a618866 | 2018-04-04 17:12:54 | [diff] [blame] | 1013 | av_buffer_unref(&bin->buf); |
| 1014 | bin->data = NULL; |
Michael Niedermayer | 5e1bacf | 2012-12-04 02:30:40 | [diff] [blame] | 1015 | bin->size = 0; |
Andreas Rheinhardt | a569a7b | 2019-06-23 23:42:30 | [diff] [blame] | 1016 | return ret < 0 ? ret : NEEDS_CHECKING; |
David Conrad | 5549aa6 | 2010-05-18 21:21:37 | [diff] [blame] | 1017 | } |
Aurelien Jacobs | 737c40d | 2008-08-05 00:42:39 | [diff] [blame] | 1018 | |
| 1019 | return 0; |
| 1020 | } |
| 1021 | |
| 1022 | /* |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 1023 | * Read the next element, but only the header. The contents |
| 1024 | * are supposed to be sub-elements which can be read separately. |
| 1025 | * 0 is success, < 0 is failure. |
| 1026 | */ |
Andreas Rheinhardt | 7087fc9 | 2019-05-16 22:30:12 | [diff] [blame] | 1027 | static int ebml_read_master(MatroskaDemuxContext *matroska, |
| 1028 | uint64_t length, int64_t pos) |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 1029 | { |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 1030 | MatroskaLevel *level; |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 1031 | |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 1032 | if (matroska->num_levels >= EBML_MAX_DEPTH) { |
| 1033 | av_log(matroska->ctx, AV_LOG_ERROR, |
| 1034 | "File moves beyond max. allowed depth (%d)\n", EBML_MAX_DEPTH); |
Panagiotis Issaris | 85565db | 2007-07-19 15:38:33 | [diff] [blame] | 1035 | return AVERROR(ENOSYS); |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 1036 | } |
| 1037 | |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 1038 | level = &matroska->levels[matroska->num_levels++]; |
Andreas Rheinhardt | 7087fc9 | 2019-05-16 22:30:12 | [diff] [blame] | 1039 | level->start = pos; |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 1040 | level->length = length; |
| 1041 | |
| 1042 | return 0; |
| 1043 | } |
| 1044 | |
| 1045 | /* |
Andreas Rheinhardt | d5274f8 | 2019-12-03 17:09:10 | [diff] [blame] | 1046 | * Read a signed "EBML number" |
Diego Biurrun | 5968d2d | 2008-08-05 08:28:57 | [diff] [blame] | 1047 | * Return: number of bytes processed, < 0 on error |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 1048 | */ |
Aurelien Jacobs | c1e0113 | 2008-08-05 00:42:52 | [diff] [blame] | 1049 | static int matroska_ebmlnum_sint(MatroskaDemuxContext *matroska, |
Andreas Rheinhardt | d5274f8 | 2019-12-03 17:09:10 | [diff] [blame] | 1050 | AVIOContext *pb, int64_t *num) |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 1051 | { |
| 1052 | uint64_t unum; |
| 1053 | int res; |
| 1054 | |
| 1055 | /* read as unsigned number first */ |
Andreas Rheinhardt | d5274f8 | 2019-12-03 17:09:10 | [diff] [blame] | 1056 | if ((res = ebml_read_num(matroska, pb, 8, &unum, 1)) < 0) |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 1057 | return res; |
| 1058 | |
| 1059 | /* make signed (weird way) */ |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 1060 | *num = unum - ((1LL << (7 * res - 1)) - 1); |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 1061 | |
| 1062 | return res; |
| 1063 | } |
| 1064 | |
Andreas Rheinhardt | 559e342 | 2019-05-16 22:30:04 | [diff] [blame] | 1065 | static int ebml_parse(MatroskaDemuxContext *matroska, |
| 1066 | EbmlSyntax *syntax, void *data); |
Aurelien Jacobs | 789ed10 | 2008-08-05 00:40:00 | [diff] [blame] | 1067 | |
Andreas Rheinhardt | 559e342 | 2019-05-16 22:30:04 | [diff] [blame] | 1068 | static EbmlSyntax *ebml_parse_id(EbmlSyntax *syntax, uint32_t id) |
Aurelien Jacobs | 789ed10 | 2008-08-05 00:40:00 | [diff] [blame] | 1069 | { |
| 1070 | int i; |
Andreas Rheinhardt | 38255cd | 2019-05-16 22:30:10 | [diff] [blame] | 1071 | |
| 1072 | // Whoever touches this should be aware of the duplication |
| 1073 | // existing in matroska_cluster_parsing. |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 1074 | for (i = 0; syntax[i].id; i++) |
Aurelien Jacobs | 789ed10 | 2008-08-05 00:40:00 | [diff] [blame] | 1075 | if (id == syntax[i].id) |
| 1076 | break; |
Aurelien Jacobs | 789ed10 | 2008-08-05 00:40:00 | [diff] [blame] | 1077 | |
Andreas Rheinhardt | 559e342 | 2019-05-16 22:30:04 | [diff] [blame] | 1078 | return &syntax[i]; |
Aurelien Jacobs | 66a37e0 | 2008-08-05 00:42:17 | [diff] [blame] | 1079 | } |
| 1080 | |
Aurelien Jacobs | 9bcb92c | 2008-08-05 00:42:13 | [diff] [blame] | 1081 | static int ebml_parse_nest(MatroskaDemuxContext *matroska, EbmlSyntax *syntax, |
Aurelien Jacobs | 6314cca | 2008-08-05 00:42:23 | [diff] [blame] | 1082 | void *data) |
Aurelien Jacobs | 789ed10 | 2008-08-05 00:40:00 | [diff] [blame] | 1083 | { |
Andreas Rheinhardt | 6854127 | 2019-05-16 22:30:19 | [diff] [blame] | 1084 | int res; |
Aurelien Jacobs | 789ed10 | 2008-08-05 00:40:00 | [diff] [blame] | 1085 | |
Andreas Rheinhardt | 71c9088 | 2019-05-16 22:30:18 | [diff] [blame] | 1086 | if (data) { |
Andreas Rheinhardt | 6854127 | 2019-05-16 22:30:19 | [diff] [blame] | 1087 | for (int i = 0; syntax[i].id; i++) |
| 1088 | switch (syntax[i].type) { |
| 1089 | case EBML_UINT: |
| 1090 | *(uint64_t *) ((char *) data + syntax[i].data_offset) = syntax[i].def.u; |
| 1091 | break; |
| 1092 | case EBML_SINT: |
| 1093 | *(int64_t *) ((char *) data + syntax[i].data_offset) = syntax[i].def.i; |
| 1094 | break; |
| 1095 | case EBML_FLOAT: |
| 1096 | *(double *) ((char *) data + syntax[i].data_offset) = syntax[i].def.f; |
| 1097 | break; |
| 1098 | case EBML_STR: |
| 1099 | case EBML_UTF8: |
| 1100 | // the default may be NULL |
| 1101 | if (syntax[i].def.s) { |
| 1102 | uint8_t **dst = (uint8_t **) ((uint8_t *) data + syntax[i].data_offset); |
| 1103 | *dst = av_strdup(syntax[i].def.s); |
| 1104 | if (!*dst) |
| 1105 | return AVERROR(ENOMEM); |
| 1106 | } |
| 1107 | break; |
Anton Khirnov | 668643b | 2013-09-04 06:55:17 | [diff] [blame] | 1108 | } |
Aurelien Jacobs | 789ed10 | 2008-08-05 00:40:00 | [diff] [blame] | 1109 | |
Andreas Rheinhardt | 6854127 | 2019-05-16 22:30:19 | [diff] [blame] | 1110 | if (!matroska->levels[matroska->num_levels - 1].length) { |
| 1111 | matroska->num_levels--; |
| 1112 | return 0; |
| 1113 | } |
Andreas Rheinhardt | 71c9088 | 2019-05-16 22:30:18 | [diff] [blame] | 1114 | } |
Aurelien Jacobs | 789ed10 | 2008-08-05 00:40:00 | [diff] [blame] | 1115 | |
Andreas Rheinhardt | b31c9b7 | 2019-06-23 23:46:54 | [diff] [blame] | 1116 | do { |
| 1117 | res = ebml_parse(matroska, syntax, data); |
| 1118 | } while (!res); |
| 1119 | |
| 1120 | return res == LEVEL_ENDED ? 0 : res; |
Aurelien Jacobs | 789ed10 | 2008-08-05 00:40:00 | [diff] [blame] | 1121 | } |
| 1122 | |
wm4 | 7e240f9 | 2015-06-12 11:11:41 | [diff] [blame] | 1123 | static int is_ebml_id_valid(uint32_t id) |
| 1124 | { |
| 1125 | // Due to endian nonsense in Matroska, the highest byte with any bits set |
| 1126 | // will contain the leading length bit. This bit in turn identifies the |
| 1127 | // total byte length of the element by its position within the byte. |
| 1128 | unsigned int bits = av_log2(id); |
| 1129 | return id && (bits + 7) / 8 == (8 - bits % 8); |
| 1130 | } |
| 1131 | |
wm4 | cac2295 | 2015-02-09 19:39:00 | [diff] [blame] | 1132 | /* |
| 1133 | * Allocate and return the entry for the level1 element with the given ID. If |
| 1134 | * an entry already exists, return the existing entry. |
| 1135 | */ |
| 1136 | static MatroskaLevel1Element *matroska_find_level1_elem(MatroskaDemuxContext *matroska, |
| 1137 | uint32_t id) |
| 1138 | { |
| 1139 | int i; |
| 1140 | MatroskaLevel1Element *elem; |
| 1141 | |
wm4 | 7e240f9 | 2015-06-12 11:11:41 | [diff] [blame] | 1142 | if (!is_ebml_id_valid(id)) |
| 1143 | return NULL; |
| 1144 | |
wm4 | cac2295 | 2015-02-09 19:39:00 | [diff] [blame] | 1145 | // Some files link to all clusters; useless. |
| 1146 | if (id == MATROSKA_ID_CLUSTER) |
| 1147 | return NULL; |
| 1148 | |
| 1149 | // There can be multiple seekheads. |
| 1150 | if (id != MATROSKA_ID_SEEKHEAD) { |
| 1151 | for (i = 0; i < matroska->num_level1_elems; i++) { |
| 1152 | if (matroska->level1_elems[i].id == id) |
| 1153 | return &matroska->level1_elems[i]; |
| 1154 | } |
| 1155 | } |
| 1156 | |
| 1157 | // Only a completely broken file would have more elements. |
| 1158 | // It also provides a low-effort way to escape from circular seekheads |
| 1159 | // (every iteration will add a level1 entry). |
| 1160 | if (matroska->num_level1_elems >= FF_ARRAY_ELEMS(matroska->level1_elems)) { |
| 1161 | av_log(matroska->ctx, AV_LOG_ERROR, "Too many level1 elements or circular seekheads.\n"); |
| 1162 | return NULL; |
| 1163 | } |
| 1164 | |
| 1165 | elem = &matroska->level1_elems[matroska->num_level1_elems++]; |
| 1166 | *elem = (MatroskaLevel1Element){.id = id}; |
| 1167 | |
| 1168 | return elem; |
| 1169 | } |
| 1170 | |
Andreas Rheinhardt | 559e342 | 2019-05-16 22:30:04 | [diff] [blame] | 1171 | static int ebml_parse(MatroskaDemuxContext *matroska, |
| 1172 | EbmlSyntax *syntax, void *data) |
Aurelien Jacobs | 737c40d | 2008-08-05 00:42:39 | [diff] [blame] | 1173 | { |
Reimar Döffinger | 14d735b | 2011-02-06 10:32:03 | [diff] [blame] | 1174 | static const uint64_t max_lengths[EBML_TYPE_COUNT] = { |
Andreas Rheinhardt | 04b62bd | 2019-05-16 22:30:15 | [diff] [blame] | 1175 | // Forbid unknown-length EBML_NONE elements. |
| 1176 | [EBML_NONE] = EBML_UNKNOWN_LENGTH - 1, |
Reimar Döffinger | 14d735b | 2011-02-06 10:32:03 | [diff] [blame] | 1177 | [EBML_UINT] = 8, |
Andreas Rheinhardt | e5ec131 | 2019-05-16 22:29:50 | [diff] [blame] | 1178 | [EBML_SINT] = 8, |
Reimar Döffinger | 14d735b | 2011-02-06 10:32:03 | [diff] [blame] | 1179 | [EBML_FLOAT] = 8, |
| 1180 | // max. 16 MB for strings |
| 1181 | [EBML_STR] = 0x1000000, |
| 1182 | [EBML_UTF8] = 0x1000000, |
| 1183 | // max. 256 MB for binary data |
| 1184 | [EBML_BIN] = 0x10000000, |
| 1185 | // no limits for anything else |
| 1186 | }; |
Anton Khirnov | 471fe57 | 2011-02-20 10:04:12 | [diff] [blame] | 1187 | AVIOContext *pb = matroska->ctx->pb; |
Andreas Rheinhardt | 559e342 | 2019-05-16 22:30:04 | [diff] [blame] | 1188 | uint32_t id; |
Aurelien Jacobs | 737c40d | 2008-08-05 00:42:39 | [diff] [blame] | 1189 | uint64_t length; |
Andreas Rheinhardt | 7087fc9 | 2019-05-16 22:30:12 | [diff] [blame] | 1190 | int64_t pos = avio_tell(pb), pos_alt; |
Andreas Rheinhardt | b31c9b7 | 2019-06-23 23:46:54 | [diff] [blame] | 1191 | int res, update_pos = 1, level_check; |
wm4 | cac2295 | 2015-02-09 19:39:00 | [diff] [blame] | 1192 | MatroskaLevel1Element *level1_elem; |
Andreas Rheinhardt | b31c9b7 | 2019-06-23 23:46:54 | [diff] [blame] | 1193 | MatroskaLevel *level = matroska->num_levels ? &matroska->levels[matroska->num_levels - 1] : NULL; |
Aurelien Jacobs | 737c40d | 2008-08-05 00:42:39 | [diff] [blame] | 1194 | |
Andreas Rheinhardt | 559e342 | 2019-05-16 22:30:04 | [diff] [blame] | 1195 | if (!matroska->current_id) { |
| 1196 | uint64_t id; |
| 1197 | res = ebml_read_num(matroska, pb, 4, &id, 0); |
| 1198 | if (res < 0) { |
Andreas Rheinhardt | b31c9b7 | 2019-06-23 23:46:54 | [diff] [blame] | 1199 | if (pb->eof_reached && res == AVERROR_EOF) { |
| 1200 | if (matroska->is_live) |
| 1201 | // in live mode, finish parsing if EOF is reached. |
| 1202 | return 1; |
Andreas Rheinhardt | 3ed2755 | 2019-05-16 22:30:11 | [diff] [blame] | 1203 | if (level && pos == avio_tell(pb)) { |
| 1204 | if (level->length == EBML_UNKNOWN_LENGTH) { |
| 1205 | // Unknown-length levels automatically end at EOF. |
| 1206 | matroska->num_levels--; |
| 1207 | return LEVEL_ENDED; |
| 1208 | } else { |
| 1209 | av_log(matroska->ctx, AV_LOG_ERROR, "File ended prematurely " |
| 1210 | "at pos. %"PRIu64" (0x%"PRIx64")\n", pos, pos); |
| 1211 | } |
Andreas Rheinhardt | b31c9b7 | 2019-06-23 23:46:54 | [diff] [blame] | 1212 | } |
| 1213 | } |
| 1214 | return res; |
Andreas Rheinhardt | 559e342 | 2019-05-16 22:30:04 | [diff] [blame] | 1215 | } |
| 1216 | matroska->current_id = id | 1 << 7 * res; |
Andreas Rheinhardt | 7087fc9 | 2019-05-16 22:30:12 | [diff] [blame] | 1217 | pos_alt = pos + res; |
| 1218 | } else { |
| 1219 | pos_alt = pos; |
| 1220 | pos -= (av_log2(matroska->current_id) + 7) / 8; |
| 1221 | } |
Andreas Rheinhardt | 559e342 | 2019-05-16 22:30:04 | [diff] [blame] | 1222 | |
| 1223 | id = matroska->current_id; |
| 1224 | |
| 1225 | syntax = ebml_parse_id(syntax, id); |
Andreas Rheinhardt | 559e342 | 2019-05-16 22:30:04 | [diff] [blame] | 1226 | if (!syntax->id && id != EBML_ID_VOID && id != EBML_ID_CRC32) { |
Andreas Rheinhardt | b31c9b7 | 2019-06-23 23:46:54 | [diff] [blame] | 1227 | if (level && level->length == EBML_UNKNOWN_LENGTH) { |
| 1228 | // Unknown-length levels end when an element from an upper level |
| 1229 | // in the hierarchy is encountered. |
| 1230 | while (syntax->def.n) { |
| 1231 | syntax = ebml_parse_id(syntax->def.n, id); |
| 1232 | if (syntax->id) { |
| 1233 | matroska->num_levels--; |
| 1234 | return LEVEL_ENDED; |
| 1235 | } |
| 1236 | }; |
| 1237 | } |
| 1238 | |
| 1239 | av_log(matroska->ctx, AV_LOG_DEBUG, "Unknown entry 0x%"PRIX32" at pos. " |
| 1240 | "%"PRId64"\n", id, pos); |
| 1241 | update_pos = 0; /* Don't update resync_pos as an error might have happened. */ |
Andreas Rheinhardt | 559e342 | 2019-05-16 22:30:04 | [diff] [blame] | 1242 | } |
| 1243 | |
Andreas Rheinhardt | 71c9088 | 2019-05-16 22:30:18 | [diff] [blame] | 1244 | if (data) { |
Andreas Rheinhardt | 6854127 | 2019-05-16 22:30:19 | [diff] [blame] | 1245 | data = (char *) data + syntax->data_offset; |
| 1246 | if (syntax->list_elem_size) { |
| 1247 | EbmlList *list = data; |
James Almer | 3b3150c | 2019-09-03 21:45:04 | [diff] [blame] | 1248 | void *newelem; |
| 1249 | |
| 1250 | if ((unsigned)list->nb_elem + 1 >= UINT_MAX / syntax->list_elem_size) |
| 1251 | return AVERROR(ENOMEM); |
| 1252 | newelem = av_fast_realloc(list->elem, |
| 1253 | &list->alloc_elem_size, |
| 1254 | (list->nb_elem + 1) * syntax->list_elem_size); |
Andreas Rheinhardt | 6854127 | 2019-05-16 22:30:19 | [diff] [blame] | 1255 | if (!newelem) |
| 1256 | return AVERROR(ENOMEM); |
| 1257 | list->elem = newelem; |
| 1258 | data = (char *) list->elem + list->nb_elem * syntax->list_elem_size; |
| 1259 | memset(data, 0, syntax->list_elem_size); |
| 1260 | list->nb_elem++; |
| 1261 | } |
Andreas Rheinhardt | 71c9088 | 2019-05-16 22:30:18 | [diff] [blame] | 1262 | } |
Aurelien Jacobs | 737c40d | 2008-08-05 00:42:39 | [diff] [blame] | 1263 | |
Andreas Rheinhardt | bc3306f | 2019-05-16 22:30:00 | [diff] [blame] | 1264 | if (syntax->type != EBML_STOP) { |
Aurelien Jacobs | c3ade62 | 2010-06-11 16:34:01 | [diff] [blame] | 1265 | matroska->current_id = 0; |
Reimar Döffinger | 1b4d327 | 2010-09-06 17:51:44 | [diff] [blame] | 1266 | if ((res = ebml_read_length(matroska, pb, &length)) < 0) |
Aurelien Jacobs | 737c40d | 2008-08-05 00:42:39 | [diff] [blame] | 1267 | return res; |
Andreas Rheinhardt | 7087fc9 | 2019-05-16 22:30:12 | [diff] [blame] | 1268 | |
| 1269 | pos_alt += res; |
| 1270 | |
Steve Lhomme | 9326117 | 2019-02-13 12:20:45 | [diff] [blame] | 1271 | if (matroska->num_levels > 0) { |
Andreas Rheinhardt via ffmpeg-devel | 18a851a | 2019-03-27 11:18:44 | [diff] [blame] | 1272 | if (length != EBML_UNKNOWN_LENGTH && |
| 1273 | level->length != EBML_UNKNOWN_LENGTH) { |
Andreas Rheinhardt | 7087fc9 | 2019-05-16 22:30:12 | [diff] [blame] | 1274 | uint64_t elem_end = pos_alt + length, |
Andreas Rheinhardt via ffmpeg-devel | 18a851a | 2019-03-27 11:18:44 | [diff] [blame] | 1275 | level_end = level->start + level->length; |
| 1276 | |
Andreas Rheinhardt | b31c9b7 | 2019-06-23 23:46:54 | [diff] [blame] | 1277 | if (elem_end < level_end) { |
| 1278 | level_check = 0; |
| 1279 | } else if (elem_end == level_end) { |
| 1280 | level_check = LEVEL_ENDED; |
| 1281 | } else { |
Andreas Rheinhardt via ffmpeg-devel | 18a851a | 2019-03-27 11:18:44 | [diff] [blame] | 1282 | av_log(matroska->ctx, AV_LOG_ERROR, |
| 1283 | "Element at 0x%"PRIx64" ending at 0x%"PRIx64" exceeds " |
| 1284 | "containing master element ending at 0x%"PRIx64"\n", |
| 1285 | pos, elem_end, level_end); |
| 1286 | return AVERROR_INVALIDDATA; |
| 1287 | } |
Andreas Rheinhardt | b31c9b7 | 2019-06-23 23:46:54 | [diff] [blame] | 1288 | } else if (length != EBML_UNKNOWN_LENGTH) { |
| 1289 | level_check = 0; |
Andreas Rheinhardt via ffmpeg-devel | 18a851a | 2019-03-27 11:18:44 | [diff] [blame] | 1290 | } else if (level->length != EBML_UNKNOWN_LENGTH) { |
| 1291 | av_log(matroska->ctx, AV_LOG_ERROR, "Unknown-sized element " |
| 1292 | "at 0x%"PRIx64" inside parent with finite size\n", pos); |
| 1293 | return AVERROR_INVALIDDATA; |
Andreas Rheinhardt | 3c70b94 | 2019-05-16 22:30:16 | [diff] [blame] | 1294 | } else { |
Andreas Rheinhardt | b31c9b7 | 2019-06-23 23:46:54 | [diff] [blame] | 1295 | level_check = 0; |
Andreas Rheinhardt | 3c70b94 | 2019-05-16 22:30:16 | [diff] [blame] | 1296 | if (id != MATROSKA_ID_CLUSTER && (syntax->type == EBML_LEVEL1 |
| 1297 | || syntax->type == EBML_NEST)) { |
| 1298 | // According to the current specifications only clusters and |
| 1299 | // segments are allowed to be unknown-length. We also accept |
| 1300 | // other unknown-length master elements. |
| 1301 | av_log(matroska->ctx, AV_LOG_WARNING, |
| 1302 | "Found unknown-length element 0x%"PRIX32" other than " |
| 1303 | "a cluster at 0x%"PRIx64". Spec-incompliant, but " |
| 1304 | "parsing will nevertheless be attempted.\n", id, pos); |
| 1305 | update_pos = -1; |
| 1306 | } |
| 1307 | } |
Andreas Rheinhardt | b31c9b7 | 2019-06-23 23:46:54 | [diff] [blame] | 1308 | } else |
| 1309 | level_check = 0; |
Andreas Rheinhardt | a3db9f6 | 2019-05-16 22:30:05 | [diff] [blame] | 1310 | |
Andreas Rheinhardt | 04b62bd | 2019-05-16 22:30:15 | [diff] [blame] | 1311 | if (max_lengths[syntax->type] && length > max_lengths[syntax->type]) { |
| 1312 | if (length != EBML_UNKNOWN_LENGTH) { |
| 1313 | av_log(matroska->ctx, AV_LOG_ERROR, |
| 1314 | "Invalid length 0x%"PRIx64" > 0x%"PRIx64" for element " |
| 1315 | "with ID 0x%"PRIX32" at 0x%"PRIx64"\n", |
| 1316 | length, max_lengths[syntax->type], id, pos); |
| 1317 | } else if (syntax->type != EBML_NONE) { |
| 1318 | av_log(matroska->ctx, AV_LOG_ERROR, |
| 1319 | "Element with ID 0x%"PRIX32" at pos. 0x%"PRIx64" has " |
| 1320 | "unknown length, yet the length of an element of its " |
| 1321 | "type must be known.\n", id, pos); |
| 1322 | } else { |
| 1323 | av_log(matroska->ctx, AV_LOG_ERROR, |
| 1324 | "Found unknown-length element with ID 0x%"PRIX32" at " |
| 1325 | "pos. 0x%"PRIx64" for which no syntax for parsing is " |
| 1326 | "available.\n", id, pos); |
| 1327 | } |
| 1328 | return AVERROR_INVALIDDATA; |
| 1329 | } |
| 1330 | |
Andreas Rheinhardt | 5120305 | 2019-05-16 22:30:14 | [diff] [blame] | 1331 | if (!(pb->seekable & AVIO_SEEKABLE_NORMAL)) { |
| 1332 | // Loosing sync will likely manifest itself as encountering unknown |
| 1333 | // elements which are not reliably distinguishable from elements |
| 1334 | // belonging to future extensions of the format. |
| 1335 | // We use a heuristic to detect such situations: If the current |
| 1336 | // element is not expected at the current syntax level and there |
| 1337 | // were only a few unknown elements in a row, then the element is |
| 1338 | // skipped or considered defective based upon the length of the |
| 1339 | // current element (i.e. how much would be skipped); if there were |
| 1340 | // more than a few skipped elements in a row and skipping the current |
| 1341 | // element would lead us more than SKIP_THRESHOLD away from the last |
Michael Niedermayer | fccc37c | 2019-08-28 17:04:42 | [diff] [blame] | 1342 | // known good position, then it is inferred that an error occurred. |
Andreas Rheinhardt | 5120305 | 2019-05-16 22:30:14 | [diff] [blame] | 1343 | // The dependency on the number of unknown elements in a row exists |
| 1344 | // because the distance to the last known good position is |
| 1345 | // automatically big if the last parsed element was big. |
| 1346 | // In both cases, each unknown element is considered equivalent to |
| 1347 | // UNKNOWN_EQUIV of skipped bytes for the check. |
| 1348 | // The whole check is only done for non-seekable output, because |
| 1349 | // in this situation skipped data can't simply be rechecked later. |
| 1350 | // This is especially important when using unkown length elements |
| 1351 | // as the check for whether a child exceeds its containing master |
| 1352 | // element is not effective in this situation. |
| 1353 | if (update_pos) { |
| 1354 | matroska->unknown_count = 0; |
| 1355 | } else { |
| 1356 | int64_t dist = length + UNKNOWN_EQUIV * matroska->unknown_count++; |
| 1357 | |
| 1358 | if (matroska->unknown_count > 3) |
| 1359 | dist += pos_alt - matroska->resync_pos; |
| 1360 | |
| 1361 | if (dist > SKIP_THRESHOLD) { |
| 1362 | av_log(matroska->ctx, AV_LOG_ERROR, |
| 1363 | "Unknown element %"PRIX32" at pos. 0x%"PRIx64" with " |
| 1364 | "length 0x%"PRIx64" considered as invalid data. Last " |
| 1365 | "known good position 0x%"PRIx64", %d unknown elements" |
| 1366 | " in a row\n", id, pos, length, matroska->resync_pos, |
| 1367 | matroska->unknown_count); |
| 1368 | return AVERROR_INVALIDDATA; |
| 1369 | } |
| 1370 | } |
| 1371 | } |
| 1372 | |
Andreas Rheinhardt | 3c70b94 | 2019-05-16 22:30:16 | [diff] [blame] | 1373 | if (update_pos > 0) { |
Andreas Rheinhardt | a3db9f6 | 2019-05-16 22:30:05 | [diff] [blame] | 1374 | // We have found an element that is allowed at this place |
| 1375 | // in the hierarchy and it passed all checks, so treat the beginning |
| 1376 | // of the element as the "last known good" position. |
| 1377 | matroska->resync_pos = pos; |
| 1378 | } |
Andreas Rheinhardt | 71c9088 | 2019-05-16 22:30:18 | [diff] [blame] | 1379 | |
| 1380 | if (!data && length != EBML_UNKNOWN_LENGTH) |
| 1381 | goto skip; |
Aurelien Jacobs | c3ade62 | 2010-06-11 16:34:01 | [diff] [blame] | 1382 | } |
Aurelien Jacobs | 737c40d | 2008-08-05 00:42:39 | [diff] [blame] | 1383 | |
| 1384 | switch (syntax->type) { |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 1385 | case EBML_UINT: |
| 1386 | res = ebml_read_uint(pb, length, data); |
| 1387 | break; |
Michael Niedermayer | b3d9ab1 | 2014-03-07 22:46:37 | [diff] [blame] | 1388 | case EBML_SINT: |
| 1389 | res = ebml_read_sint(pb, length, data); |
| 1390 | break; |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 1391 | case EBML_FLOAT: |
| 1392 | res = ebml_read_float(pb, length, data); |
| 1393 | break; |
Aurelien Jacobs | 737c40d | 2008-08-05 00:42:39 | [diff] [blame] | 1394 | case EBML_STR: |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 1395 | case EBML_UTF8: |
| 1396 | res = ebml_read_ascii(pb, length, data); |
| 1397 | break; |
| 1398 | case EBML_BIN: |
Andreas Rheinhardt | 7087fc9 | 2019-05-16 22:30:12 | [diff] [blame] | 1399 | res = ebml_read_binary(pb, length, pos_alt, data); |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 1400 | break; |
wm4 | cac2295 | 2015-02-09 19:39:00 | [diff] [blame] | 1401 | case EBML_LEVEL1: |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 1402 | case EBML_NEST: |
Andreas Rheinhardt | 7087fc9 | 2019-05-16 22:30:12 | [diff] [blame] | 1403 | if ((res = ebml_read_master(matroska, length, pos_alt)) < 0) |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 1404 | return res; |
| 1405 | if (id == MATROSKA_ID_SEGMENT) |
Andreas Rheinhardt | 7087fc9 | 2019-05-16 22:30:12 | [diff] [blame] | 1406 | matroska->segment_start = pos_alt; |
wm4 | cac2295 | 2015-02-09 19:39:00 | [diff] [blame] | 1407 | if (id == MATROSKA_ID_CUES) |
| 1408 | matroska->cues_parsing_deferred = 0; |
| 1409 | if (syntax->type == EBML_LEVEL1 && |
| 1410 | (level1_elem = matroska_find_level1_elem(matroska, syntax->id))) { |
Andreas Rheinhardt | 806ac7d | 2019-05-16 22:30:21 | [diff] [blame] | 1411 | if (!level1_elem->pos) { |
| 1412 | // Zero is not a valid position for a level 1 element. |
| 1413 | level1_elem->pos = pos; |
| 1414 | } else if (level1_elem->pos != pos) |
wm4 | cac2295 | 2015-02-09 19:39:00 | [diff] [blame] | 1415 | av_log(matroska->ctx, AV_LOG_ERROR, "Duplicate element\n"); |
| 1416 | level1_elem->parsed = 1; |
| 1417 | } |
Andreas Rheinhardt | b31c9b7 | 2019-06-23 23:46:54 | [diff] [blame] | 1418 | if (res = ebml_parse_nest(matroska, syntax->def.n, data)) |
| 1419 | return res; |
| 1420 | break; |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 1421 | case EBML_STOP: |
| 1422 | return 1; |
Andreas Rheinhardt | 71c9088 | 2019-05-16 22:30:18 | [diff] [blame] | 1423 | skip: |
Michael Niedermayer | 668c873 | 2012-08-04 00:27:51 | [diff] [blame] | 1424 | default: |
Andreas Rheinhardt | ff5ea59 | 2019-06-23 23:42:31 | [diff] [blame] | 1425 | if (length) { |
Andreas Rheinhardt | eb33be1 | 2019-07-06 16:59:22 | [diff] [blame] | 1426 | int64_t res2; |
Andreas Rheinhardt | ff5ea59 | 2019-06-23 23:42:31 | [diff] [blame] | 1427 | if (ffio_limit(pb, length) != length) { |
| 1428 | // ffio_limit emits its own error message, |
| 1429 | // so we don't have to. |
| 1430 | return AVERROR(EIO); |
| 1431 | } |
Andreas Rheinhardt | eb33be1 | 2019-07-06 16:59:22 | [diff] [blame] | 1432 | if ((res2 = avio_skip(pb, length - 1)) >= 0) { |
Andreas Rheinhardt | ff5ea59 | 2019-06-23 23:42:31 | [diff] [blame] | 1433 | // avio_skip might take us past EOF. We check for this |
| 1434 | // by skipping only length - 1 bytes, reading a byte and |
| 1435 | // checking the error flags. This is done in order to check |
| 1436 | // that the element has been properly skipped even when |
| 1437 | // no filesize (that ffio_limit relies on) is available. |
| 1438 | avio_r8(pb); |
| 1439 | res = NEEDS_CHECKING; |
Andreas Rheinhardt | eb33be1 | 2019-07-06 16:59:22 | [diff] [blame] | 1440 | } else |
| 1441 | res = res2; |
Andreas Rheinhardt | ff5ea59 | 2019-06-23 23:42:31 | [diff] [blame] | 1442 | } else |
| 1443 | res = 0; |
Aurelien Jacobs | 737c40d | 2008-08-05 00:42:39 | [diff] [blame] | 1444 | } |
Andreas Rheinhardt | a569a7b | 2019-06-23 23:42:30 | [diff] [blame] | 1445 | if (res) { |
| 1446 | if (res == NEEDS_CHECKING) { |
| 1447 | if (pb->eof_reached) { |
| 1448 | if (pb->error) |
| 1449 | res = pb->error; |
| 1450 | else |
| 1451 | res = AVERROR_EOF; |
| 1452 | } else |
Andreas Rheinhardt | b31c9b7 | 2019-06-23 23:46:54 | [diff] [blame] | 1453 | goto level_check; |
Andreas Rheinhardt | a569a7b | 2019-06-23 23:42:30 | [diff] [blame] | 1454 | } |
| 1455 | |
| 1456 | if (res == AVERROR_INVALIDDATA) |
| 1457 | av_log(matroska->ctx, AV_LOG_ERROR, "Invalid element\n"); |
| 1458 | else if (res == AVERROR(EIO)) |
| 1459 | av_log(matroska->ctx, AV_LOG_ERROR, "Read error\n"); |
| 1460 | else if (res == AVERROR_EOF) { |
| 1461 | av_log(matroska->ctx, AV_LOG_ERROR, "File ended prematurely\n"); |
| 1462 | res = AVERROR(EIO); |
| 1463 | } |
Andreas Rheinhardt | b31c9b7 | 2019-06-23 23:46:54 | [diff] [blame] | 1464 | |
| 1465 | return res; |
Andreas Rheinhardt | a569a7b | 2019-06-23 23:42:30 | [diff] [blame] | 1466 | } |
Andreas Rheinhardt | b31c9b7 | 2019-06-23 23:46:54 | [diff] [blame] | 1467 | |
| 1468 | level_check: |
| 1469 | if (level_check == LEVEL_ENDED && matroska->num_levels) { |
| 1470 | level = &matroska->levels[matroska->num_levels - 1]; |
| 1471 | pos = avio_tell(pb); |
| 1472 | |
| 1473 | // Given that pos >= level->start no check for |
| 1474 | // level->length != EBML_UNKNOWN_LENGTH is necessary. |
| 1475 | while (matroska->num_levels && pos == level->start + level->length) { |
| 1476 | matroska->num_levels--; |
| 1477 | level--; |
| 1478 | } |
| 1479 | } |
| 1480 | |
| 1481 | return level_check; |
Aurelien Jacobs | 737c40d | 2008-08-05 00:42:39 | [diff] [blame] | 1482 | } |
| 1483 | |
Aurelien Jacobs | 789ed10 | 2008-08-05 00:40:00 | [diff] [blame] | 1484 | static void ebml_free(EbmlSyntax *syntax, void *data) |
| 1485 | { |
| 1486 | int i, j; |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 1487 | for (i = 0; syntax[i].id; i++) { |
| 1488 | void *data_off = (char *) data + syntax[i].data_offset; |
Aurelien Jacobs | 789ed10 | 2008-08-05 00:40:00 | [diff] [blame] | 1489 | switch (syntax[i].type) { |
| 1490 | case EBML_STR: |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 1491 | case EBML_UTF8: |
| 1492 | av_freep(data_off); |
| 1493 | break; |
| 1494 | case EBML_BIN: |
James Almer | a618866 | 2018-04-04 17:12:54 | [diff] [blame] | 1495 | av_buffer_unref(&((EbmlBin *) data_off)->buf); |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 1496 | break; |
wm4 | cac2295 | 2015-02-09 19:39:00 | [diff] [blame] | 1497 | case EBML_LEVEL1: |
Aurelien Jacobs | 789ed10 | 2008-08-05 00:40:00 | [diff] [blame] | 1498 | case EBML_NEST: |
| 1499 | if (syntax[i].list_elem_size) { |
| 1500 | EbmlList *list = data_off; |
| 1501 | char *ptr = list->elem; |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 1502 | for (j = 0; j < list->nb_elem; |
| 1503 | j++, ptr += syntax[i].list_elem_size) |
Aurelien Jacobs | 789ed10 | 2008-08-05 00:40:00 | [diff] [blame] | 1504 | ebml_free(syntax[i].def.n, ptr); |
Michael Niedermayer | 6e70e4a | 2015-01-06 11:48:38 | [diff] [blame] | 1505 | av_freep(&list->elem); |
Michael Schenk | 18b9466 | 2016-11-25 08:36:20 | [diff] [blame] | 1506 | list->nb_elem = 0; |
James Almer | 3b3150c | 2019-09-03 21:45:04 | [diff] [blame] | 1507 | list->alloc_elem_size = 0; |
Aurelien Jacobs | 789ed10 | 2008-08-05 00:40:00 | [diff] [blame] | 1508 | } else |
| 1509 | ebml_free(syntax[i].def.n, data_off); |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 1510 | default: |
| 1511 | break; |
Aurelien Jacobs | 789ed10 | 2008-08-05 00:40:00 | [diff] [blame] | 1512 | } |
| 1513 | } |
| 1514 | } |
| 1515 | |
Aurelien Jacobs | 737c40d | 2008-08-05 00:42:39 | [diff] [blame] | 1516 | /* |
| 1517 | * Autodetecting... |
| 1518 | */ |
Carl Eugen Hoyos | 4d8875e | 2019-03-21 00:18:37 | [diff] [blame] | 1519 | static int matroska_probe(const AVProbeData *p) |
Aurelien Jacobs | 737c40d | 2008-08-05 00:42:39 | [diff] [blame] | 1520 | { |
| 1521 | uint64_t total = 0; |
James Zern | 470491f | 2010-05-22 01:41:32 | [diff] [blame] | 1522 | int len_mask = 0x80, size = 1, n = 1, i; |
Aurelien Jacobs | 737c40d | 2008-08-05 00:42:39 | [diff] [blame] | 1523 | |
Diego Biurrun | 5968d2d | 2008-08-05 08:28:57 | [diff] [blame] | 1524 | /* EBML header? */ |
Aurelien Jacobs | 737c40d | 2008-08-05 00:42:39 | [diff] [blame] | 1525 | if (AV_RB32(p->buf) != EBML_ID_HEADER) |
| 1526 | return 0; |
| 1527 | |
| 1528 | /* length of header */ |
| 1529 | total = p->buf[4]; |
| 1530 | while (size <= 8 && !(total & len_mask)) { |
| 1531 | size++; |
| 1532 | len_mask >>= 1; |
| 1533 | } |
| 1534 | if (size > 8) |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 1535 | return 0; |
Aurelien Jacobs | 737c40d | 2008-08-05 00:42:39 | [diff] [blame] | 1536 | total &= (len_mask - 1); |
| 1537 | while (n < size) |
| 1538 | total = (total << 8) | p->buf[4 + n++]; |
| 1539 | |
Andreas Rheinhardt | 9c6d14a | 2019-05-16 22:30:17 | [diff] [blame] | 1540 | if (total + 1 == 1ULL << (7 * size)){ |
| 1541 | /* Unknown-length header - simply parse the whole buffer. */ |
| 1542 | total = p->buf_size - 4 - size; |
| 1543 | } else { |
| 1544 | /* Does the probe data contain the whole header? */ |
| 1545 | if (p->buf_size < 4 + size + total) |
| 1546 | return 0; |
| 1547 | } |
Aurelien Jacobs | 737c40d | 2008-08-05 00:42:39 | [diff] [blame] | 1548 | |
James Zern | 470491f | 2010-05-22 01:41:32 | [diff] [blame] | 1549 | /* The header should contain a known document type. For now, |
Aurelien Jacobs | 737c40d | 2008-08-05 00:42:39 | [diff] [blame] | 1550 | * we don't parse the whole header but simply check for the |
| 1551 | * availability of that array of characters inside the header. |
| 1552 | * Not fully fool-proof, but good enough. */ |
James Zern | 470491f | 2010-05-22 01:41:32 | [diff] [blame] | 1553 | for (i = 0; i < FF_ARRAY_ELEMS(matroska_doctypes); i++) { |
Michael Niedermayer | 17b2630 | 2015-05-11 01:51:17 | [diff] [blame] | 1554 | size_t probelen = strlen(matroska_doctypes[i]); |
Chris Evans | 69619a1 | 2011-07-20 00:51:48 | [diff] [blame] | 1555 | if (total < probelen) |
| 1556 | continue; |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 1557 | for (n = 4 + size; n <= 4 + size + total - probelen; n++) |
| 1558 | if (!memcmp(p->buf + n, matroska_doctypes[i], probelen)) |
James Zern | 470491f | 2010-05-22 01:41:32 | [diff] [blame] | 1559 | return AVPROBE_SCORE_MAX; |
| 1560 | } |
Aurelien Jacobs | 737c40d | 2008-08-05 00:42:39 | [diff] [blame] | 1561 | |
David Conrad | c7b913c | 2010-05-22 01:41:35 | [diff] [blame] | 1562 | // probably valid EBML header but no recognized doctype |
Diego Biurrun | e0f8be6 | 2013-03-25 15:12:51 | [diff] [blame] | 1563 | return AVPROBE_SCORE_EXTENSION; |
Aurelien Jacobs | 737c40d | 2008-08-05 00:42:39 | [diff] [blame] | 1564 | } |
| 1565 | |
| 1566 | static MatroskaTrack *matroska_find_track_by_num(MatroskaDemuxContext *matroska, |
Andreas Rheinhardt | ba36a07 | 2019-12-26 03:17:24 | [diff] [blame] | 1567 | uint64_t num) |
Aurelien Jacobs | 737c40d | 2008-08-05 00:42:39 | [diff] [blame] | 1568 | { |
| 1569 | MatroskaTrack *tracks = matroska->tracks.elem; |
| 1570 | int i; |
| 1571 | |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 1572 | for (i = 0; i < matroska->tracks.nb_elem; i++) |
Aurelien Jacobs | 737c40d | 2008-08-05 00:42:39 | [diff] [blame] | 1573 | if (tracks[i].num == num) |
| 1574 | return &tracks[i]; |
| 1575 | |
Andreas Rheinhardt | ba36a07 | 2019-12-26 03:17:24 | [diff] [blame] | 1576 | av_log(matroska->ctx, AV_LOG_ERROR, "Invalid track number %"PRIu64"\n", num); |
Aurelien Jacobs | 737c40d | 2008-08-05 00:42:39 | [diff] [blame] | 1577 | return NULL; |
| 1578 | } |
| 1579 | |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 1580 | static int matroska_decode_buffer(uint8_t **buf, int *buf_size, |
Aurelien Jacobs | f7b9687 | 2008-08-05 00:42:05 | [diff] [blame] | 1581 | MatroskaTrack *track) |
Evgeniy Stepanov | 935ec5a | 2008-06-22 15:49:44 | [diff] [blame] | 1582 | { |
Aurelien Jacobs | 2cbc881 | 2008-08-05 00:40:31 | [diff] [blame] | 1583 | MatroskaTrackEncoding *encodings = track->encodings.elem; |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 1584 | uint8_t *data = *buf; |
Evgeniy Stepanov | 935ec5a | 2008-06-22 15:49:44 | [diff] [blame] | 1585 | int isize = *buf_size; |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 1586 | uint8_t *pkt_data = NULL; |
Diego Biurrun | 529506b | 2012-02-12 09:58:46 | [diff] [blame] | 1587 | uint8_t av_unused *newpktdata; |
Evgeniy Stepanov | 935ec5a | 2008-06-22 15:49:44 | [diff] [blame] | 1588 | int pkt_size = isize; |
| 1589 | int result = 0; |
| 1590 | int olen; |
| 1591 | |
Michael Niedermayer | 4b7c523 | 2012-06-14 23:29:30 | [diff] [blame] | 1592 | if (pkt_size >= 10000000U) |
Luca Barbato | c9a39ce | 2012-09-14 18:03:37 | [diff] [blame] | 1593 | return AVERROR_INVALIDDATA; |
Aurelien Jacobs | 4f90688 | 2010-08-17 14:05:23 | [diff] [blame] | 1594 | |
Aurelien Jacobs | 2cbc881 | 2008-08-05 00:40:31 | [diff] [blame] | 1595 | switch (encodings[0].compression.algo) { |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 1596 | case MATROSKA_TRACK_ENCODING_COMP_HEADERSTRIP: |
| 1597 | { |
Luca Barbato | 8d4dd55 | 2012-09-14 16:51:49 | [diff] [blame] | 1598 | int header_size = encodings[0].compression.settings.size; |
| 1599 | uint8_t *header = encodings[0].compression.settings.data; |
| 1600 | |
Michael Niedermayer | 1df2e3c | 2012-09-20 11:30:44 | [diff] [blame] | 1601 | if (header_size && !header) { |
Michael Niedermayer | 0efcf16 | 2012-11-23 17:10:02 | [diff] [blame] | 1602 | av_log(NULL, AV_LOG_ERROR, "Compression size but no data in headerstrip\n"); |
Michael Niedermayer | 2677697 | 2012-04-17 15:12:22 | [diff] [blame] | 1603 | return -1; |
| 1604 | } |
Michael Niedermayer | 1df2e3c | 2012-09-20 11:30:44 | [diff] [blame] | 1605 | |
Luca Barbato | 8d4dd55 | 2012-09-14 16:51:49 | [diff] [blame] | 1606 | if (!header_size) |
| 1607 | return 0; |
| 1608 | |
| 1609 | pkt_size = isize + header_size; |
James Almer | 9703b7d | 2018-04-04 21:12:53 | [diff] [blame] | 1610 | pkt_data = av_malloc(pkt_size + AV_INPUT_BUFFER_PADDING_SIZE); |
Luca Barbato | 8d4dd55 | 2012-09-14 16:51:49 | [diff] [blame] | 1611 | if (!pkt_data) |
| 1612 | return AVERROR(ENOMEM); |
| 1613 | |
| 1614 | memcpy(pkt_data, header, header_size); |
| 1615 | memcpy(pkt_data + header_size, data, isize); |
| 1616 | break; |
| 1617 | } |
Diego Biurrun | 2a91ada | 2012-10-18 17:48:27 | [diff] [blame] | 1618 | #if CONFIG_LZO |
Evgeniy Stepanov | 935ec5a | 2008-06-22 15:49:44 | [diff] [blame] | 1619 | case MATROSKA_TRACK_ENCODING_COMP_LZO: |
| 1620 | do { |
Andreas Rheinhardt | 2ff687c | 2019-12-28 02:40:33 | [diff] [blame] | 1621 | int insize = isize; |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 1622 | olen = pkt_size *= 3; |
James Almer | 9703b7d | 2018-04-04 21:12:53 | [diff] [blame] | 1623 | newpktdata = av_realloc(pkt_data, pkt_size + AV_LZO_OUTPUT_PADDING |
| 1624 | + AV_INPUT_BUFFER_PADDING_SIZE); |
Luca Barbato | 581281e | 2012-09-14 16:39:58 | [diff] [blame] | 1625 | if (!newpktdata) { |
Luca Barbato | c9a39ce | 2012-09-14 18:03:37 | [diff] [blame] | 1626 | result = AVERROR(ENOMEM); |
Luca Barbato | 581281e | 2012-09-14 16:39:58 | [diff] [blame] | 1627 | goto failed; |
| 1628 | } |
| 1629 | pkt_data = newpktdata; |
Andreas Rheinhardt | 2ff687c | 2019-12-28 02:40:33 | [diff] [blame] | 1630 | result = av_lzo1x_decode(pkt_data, &olen, data, &insize); |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 1631 | } while (result == AV_LZO_OUTPUT_FULL && pkt_size < 10000000); |
Luca Barbato | c9a39ce | 2012-09-14 18:03:37 | [diff] [blame] | 1632 | if (result) { |
| 1633 | result = AVERROR_INVALIDDATA; |
Evgeniy Stepanov | 935ec5a | 2008-06-22 15:49:44 | [diff] [blame] | 1634 | goto failed; |
Luca Barbato | c9a39ce | 2012-09-14 18:03:37 | [diff] [blame] | 1635 | } |
Evgeniy Stepanov | 935ec5a | 2008-06-22 15:49:44 | [diff] [blame] | 1636 | pkt_size -= olen; |
| 1637 | break; |
Diego Biurrun | 2a91ada | 2012-10-18 17:48:27 | [diff] [blame] | 1638 | #endif |
Aurelien Jacobs | b250f9c | 2009-01-13 23:44:16 | [diff] [blame] | 1639 | #if CONFIG_ZLIB |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 1640 | case MATROSKA_TRACK_ENCODING_COMP_ZLIB: |
| 1641 | { |
| 1642 | z_stream zstream = { 0 }; |
Evgeniy Stepanov | 935ec5a | 2008-06-22 15:49:44 | [diff] [blame] | 1643 | if (inflateInit(&zstream) != Z_OK) |
| 1644 | return -1; |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 1645 | zstream.next_in = data; |
Evgeniy Stepanov | 935ec5a | 2008-06-22 15:49:44 | [diff] [blame] | 1646 | zstream.avail_in = isize; |
| 1647 | do { |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 1648 | pkt_size *= 3; |
James Almer | 9703b7d | 2018-04-04 21:12:53 | [diff] [blame] | 1649 | newpktdata = av_realloc(pkt_data, pkt_size + AV_INPUT_BUFFER_PADDING_SIZE); |
Michael Niedermayer | 77d2ef1 | 2011-07-28 12:59:54 | [diff] [blame] | 1650 | if (!newpktdata) { |
| 1651 | inflateEnd(&zstream); |
Michael Niedermayer | 171af59 | 2015-05-12 16:32:12 | [diff] [blame] | 1652 | result = AVERROR(ENOMEM); |
Michael Niedermayer | 77d2ef1 | 2011-07-28 12:59:54 | [diff] [blame] | 1653 | goto failed; |
| 1654 | } |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 1655 | pkt_data = newpktdata; |
Evgeniy Stepanov | 935ec5a | 2008-06-22 15:49:44 | [diff] [blame] | 1656 | zstream.avail_out = pkt_size - zstream.total_out; |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 1657 | zstream.next_out = pkt_data + zstream.total_out; |
Michael Niedermayer | 171af59 | 2015-05-12 16:32:12 | [diff] [blame] | 1658 | result = inflate(&zstream, Z_NO_FLUSH); |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 1659 | } while (result == Z_OK && pkt_size < 10000000); |
Evgeniy Stepanov | 935ec5a | 2008-06-22 15:49:44 | [diff] [blame] | 1660 | pkt_size = zstream.total_out; |
| 1661 | inflateEnd(&zstream); |
Luca Barbato | c9a39ce | 2012-09-14 18:03:37 | [diff] [blame] | 1662 | if (result != Z_STREAM_END) { |
| 1663 | if (result == Z_MEM_ERROR) |
| 1664 | result = AVERROR(ENOMEM); |
| 1665 | else |
| 1666 | result = AVERROR_INVALIDDATA; |
Evgeniy Stepanov | 935ec5a | 2008-06-22 15:49:44 | [diff] [blame] | 1667 | goto failed; |
Luca Barbato | c9a39ce | 2012-09-14 18:03:37 | [diff] [blame] | 1668 | } |
Evgeniy Stepanov | 935ec5a | 2008-06-22 15:49:44 | [diff] [blame] | 1669 | break; |
| 1670 | } |
| 1671 | #endif |
Aurelien Jacobs | b250f9c | 2009-01-13 23:44:16 | [diff] [blame] | 1672 | #if CONFIG_BZLIB |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 1673 | case MATROSKA_TRACK_ENCODING_COMP_BZLIB: |
| 1674 | { |
| 1675 | bz_stream bzstream = { 0 }; |
Evgeniy Stepanov | 935ec5a | 2008-06-22 15:49:44 | [diff] [blame] | 1676 | if (BZ2_bzDecompressInit(&bzstream, 0, 0) != BZ_OK) |
| 1677 | return -1; |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 1678 | bzstream.next_in = data; |
Evgeniy Stepanov | 935ec5a | 2008-06-22 15:49:44 | [diff] [blame] | 1679 | bzstream.avail_in = isize; |
| 1680 | do { |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 1681 | pkt_size *= 3; |
James Almer | 9703b7d | 2018-04-04 21:12:53 | [diff] [blame] | 1682 | newpktdata = av_realloc(pkt_data, pkt_size + AV_INPUT_BUFFER_PADDING_SIZE); |
Michael Niedermayer | 77d2ef1 | 2011-07-28 12:59:54 | [diff] [blame] | 1683 | if (!newpktdata) { |
| 1684 | BZ2_bzDecompressEnd(&bzstream); |
Michael Niedermayer | 171af59 | 2015-05-12 16:32:12 | [diff] [blame] | 1685 | result = AVERROR(ENOMEM); |
Michael Niedermayer | 77d2ef1 | 2011-07-28 12:59:54 | [diff] [blame] | 1686 | goto failed; |
| 1687 | } |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 1688 | pkt_data = newpktdata; |
Evgeniy Stepanov | 935ec5a | 2008-06-22 15:49:44 | [diff] [blame] | 1689 | bzstream.avail_out = pkt_size - bzstream.total_out_lo32; |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 1690 | bzstream.next_out = pkt_data + bzstream.total_out_lo32; |
Michael Niedermayer | 171af59 | 2015-05-12 16:32:12 | [diff] [blame] | 1691 | result = BZ2_bzDecompress(&bzstream); |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 1692 | } while (result == BZ_OK && pkt_size < 10000000); |
Evgeniy Stepanov | 935ec5a | 2008-06-22 15:49:44 | [diff] [blame] | 1693 | pkt_size = bzstream.total_out_lo32; |
| 1694 | BZ2_bzDecompressEnd(&bzstream); |
Luca Barbato | c9a39ce | 2012-09-14 18:03:37 | [diff] [blame] | 1695 | if (result != BZ_STREAM_END) { |
| 1696 | if (result == BZ_MEM_ERROR) |
| 1697 | result = AVERROR(ENOMEM); |
| 1698 | else |
| 1699 | result = AVERROR_INVALIDDATA; |
Evgeniy Stepanov | 935ec5a | 2008-06-22 15:49:44 | [diff] [blame] | 1700 | goto failed; |
Luca Barbato | c9a39ce | 2012-09-14 18:03:37 | [diff] [blame] | 1701 | } |
Evgeniy Stepanov | 935ec5a | 2008-06-22 15:49:44 | [diff] [blame] | 1702 | break; |
| 1703 | } |
| 1704 | #endif |
Aurelien Jacobs | 28f27e0 | 2008-08-20 23:08:07 | [diff] [blame] | 1705 | default: |
Luca Barbato | c9a39ce | 2012-09-14 18:03:37 | [diff] [blame] | 1706 | return AVERROR_INVALIDDATA; |
Evgeniy Stepanov | 935ec5a | 2008-06-22 15:49:44 | [diff] [blame] | 1707 | } |
| 1708 | |
James Almer | 9703b7d | 2018-04-04 21:12:53 | [diff] [blame] | 1709 | memset(pkt_data + pkt_size, 0, AV_INPUT_BUFFER_PADDING_SIZE); |
| 1710 | |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 1711 | *buf = pkt_data; |
Evgeniy Stepanov | 935ec5a | 2008-06-22 15:49:44 | [diff] [blame] | 1712 | *buf_size = pkt_size; |
| 1713 | return 0; |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 1714 | |
| 1715 | failed: |
Evgeniy Stepanov | 935ec5a | 2008-06-22 15:49:44 | [diff] [blame] | 1716 | av_free(pkt_data); |
Luca Barbato | c9a39ce | 2012-09-14 18:03:37 | [diff] [blame] | 1717 | return result; |
Evgeniy Stepanov | 935ec5a | 2008-06-22 15:49:44 | [diff] [blame] | 1718 | } |
| 1719 | |
Aurelien Jacobs | 929e9de | 2009-02-15 15:53:55 | [diff] [blame] | 1720 | static void matroska_convert_tag(AVFormatContext *s, EbmlList *list, |
Anton Khirnov | d2d67e4 | 2011-05-22 10:46:29 | [diff] [blame] | 1721 | AVDictionary **metadata, char *prefix) |
Aurelien Jacobs | 44015c5 | 2008-08-08 23:50:38 | [diff] [blame] | 1722 | { |
| 1723 | MatroskaTag *tags = list->elem; |
Aurelien Jacobs | 929e9de | 2009-02-15 15:53:55 | [diff] [blame] | 1724 | char key[1024]; |
| 1725 | int i; |
Aurelien Jacobs | 44015c5 | 2008-08-08 23:50:38 | [diff] [blame] | 1726 | |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 1727 | for (i = 0; i < list->nb_elem; i++) { |
| 1728 | const char *lang = tags[i].lang && |
| 1729 | strcmp(tags[i].lang, "und") ? tags[i].lang : NULL; |
Anton Khirnov | bf800c7 | 2010-11-03 06:29:04 | [diff] [blame] | 1730 | |
| 1731 | if (!tags[i].name) { |
| 1732 | av_log(s, AV_LOG_WARNING, "Skipping invalid tag with no TagName.\n"); |
| 1733 | continue; |
| 1734 | } |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 1735 | if (prefix) |
| 1736 | snprintf(key, sizeof(key), "%s/%s", prefix, tags[i].name); |
| 1737 | else |
| 1738 | av_strlcpy(key, tags[i].name, sizeof(key)); |
Aurelien Jacobs | f702df3 | 2009-02-15 16:05:37 | [diff] [blame] | 1739 | if (tags[i].def || !lang) { |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 1740 | av_dict_set(metadata, key, tags[i].string, 0); |
| 1741 | if (tags[i].sub.nb_elem) |
| 1742 | matroska_convert_tag(s, &tags[i].sub, metadata, key); |
Aurelien Jacobs | f702df3 | 2009-02-15 16:05:37 | [diff] [blame] | 1743 | } |
| 1744 | if (lang) { |
| 1745 | av_strlcat(key, "-", sizeof(key)); |
| 1746 | av_strlcat(key, lang, sizeof(key)); |
Anton Khirnov | d2d67e4 | 2011-05-22 10:46:29 | [diff] [blame] | 1747 | av_dict_set(metadata, key, tags[i].string, 0); |
Aurelien Jacobs | f702df3 | 2009-02-15 16:05:37 | [diff] [blame] | 1748 | if (tags[i].sub.nb_elem) |
| 1749 | matroska_convert_tag(s, &tags[i].sub, metadata, key); |
| 1750 | } |
Aurelien Jacobs | 929e9de | 2009-02-15 15:53:55 | [diff] [blame] | 1751 | } |
Anton Khirnov | ad7768f | 2010-10-16 13:20:41 | [diff] [blame] | 1752 | ff_metadata_conv(metadata, NULL, ff_mkv_metadata_conv); |
Aurelien Jacobs | 929e9de | 2009-02-15 15:53:55 | [diff] [blame] | 1753 | } |
| 1754 | |
| 1755 | static void matroska_convert_tags(AVFormatContext *s) |
| 1756 | { |
| 1757 | MatroskaDemuxContext *matroska = s->priv_data; |
| 1758 | MatroskaTags *tags = matroska->tags.elem; |
| 1759 | int i, j; |
| 1760 | |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 1761 | for (i = 0; i < matroska->tags.nb_elem; i++) { |
Aurelien Jacobs | 929e9de | 2009-02-15 15:53:55 | [diff] [blame] | 1762 | if (tags[i].target.attachuid) { |
Diego Biurrun | f69befe | 2014-03-07 12:54:18 | [diff] [blame] | 1763 | MatroskaAttachment *attachment = matroska->attachments.elem; |
Luca Barbato | a0fa6d0 | 2015-10-31 18:45:27 | [diff] [blame] | 1764 | int found = 0; |
| 1765 | for (j = 0; j < matroska->attachments.nb_elem; j++) { |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 1766 | if (attachment[j].uid == tags[i].target.attachuid && |
Luca Barbato | a0fa6d0 | 2015-10-31 18:45:27 | [diff] [blame] | 1767 | attachment[j].stream) { |
Aurelien Jacobs | 929e9de | 2009-02-15 15:53:55 | [diff] [blame] | 1768 | matroska_convert_tag(s, &tags[i].tag, |
| 1769 | &attachment[j].stream->metadata, NULL); |
Luca Barbato | a0fa6d0 | 2015-10-31 18:45:27 | [diff] [blame] | 1770 | found = 1; |
| 1771 | } |
| 1772 | } |
| 1773 | if (!found) { |
| 1774 | av_log(NULL, AV_LOG_WARNING, |
| 1775 | "The tags at index %d refer to a " |
| 1776 | "non-existent attachment %"PRId64".\n", |
| 1777 | i, tags[i].target.attachuid); |
| 1778 | } |
Aurelien Jacobs | 929e9de | 2009-02-15 15:53:55 | [diff] [blame] | 1779 | } else if (tags[i].target.chapteruid) { |
| 1780 | MatroskaChapter *chapter = matroska->chapters.elem; |
Luca Barbato | a0fa6d0 | 2015-10-31 18:45:27 | [diff] [blame] | 1781 | int found = 0; |
| 1782 | for (j = 0; j < matroska->chapters.nb_elem; j++) { |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 1783 | if (chapter[j].uid == tags[i].target.chapteruid && |
Luca Barbato | a0fa6d0 | 2015-10-31 18:45:27 | [diff] [blame] | 1784 | chapter[j].chapter) { |
Aurelien Jacobs | 929e9de | 2009-02-15 15:53:55 | [diff] [blame] | 1785 | matroska_convert_tag(s, &tags[i].tag, |
| 1786 | &chapter[j].chapter->metadata, NULL); |
Luca Barbato | a0fa6d0 | 2015-10-31 18:45:27 | [diff] [blame] | 1787 | found = 1; |
| 1788 | } |
| 1789 | } |
| 1790 | if (!found) { |
| 1791 | av_log(NULL, AV_LOG_WARNING, |
| 1792 | "The tags at index %d refer to a non-existent chapter " |
| 1793 | "%"PRId64".\n", |
| 1794 | i, tags[i].target.chapteruid); |
| 1795 | } |
Aurelien Jacobs | 929e9de | 2009-02-15 15:53:55 | [diff] [blame] | 1796 | } else if (tags[i].target.trackuid) { |
| 1797 | MatroskaTrack *track = matroska->tracks.elem; |
Luca Barbato | a0fa6d0 | 2015-10-31 18:45:27 | [diff] [blame] | 1798 | int found = 0; |
| 1799 | for (j = 0; j < matroska->tracks.nb_elem; j++) { |
| 1800 | if (track[j].uid == tags[i].target.trackuid && |
| 1801 | track[j].stream) { |
Aurelien Jacobs | 929e9de | 2009-02-15 15:53:55 | [diff] [blame] | 1802 | matroska_convert_tag(s, &tags[i].tag, |
| 1803 | &track[j].stream->metadata, NULL); |
Luca Barbato | a0fa6d0 | 2015-10-31 18:45:27 | [diff] [blame] | 1804 | found = 1; |
| 1805 | } |
| 1806 | } |
| 1807 | if (!found) { |
| 1808 | av_log(NULL, AV_LOG_WARNING, |
| 1809 | "The tags at index %d refer to a non-existent track " |
| 1810 | "%"PRId64".\n", |
| 1811 | i, tags[i].target.trackuid); |
| 1812 | } |
Aurelien Jacobs | 929e9de | 2009-02-15 15:53:55 | [diff] [blame] | 1813 | } else { |
Aurelien Jacobs | 4f909c7 | 2009-06-13 22:29:38 | [diff] [blame] | 1814 | matroska_convert_tag(s, &tags[i].tag, &s->metadata, |
| 1815 | tags[i].target.type); |
Aurelien Jacobs | 929e9de | 2009-02-15 15:53:55 | [diff] [blame] | 1816 | } |
Aurelien Jacobs | 44015c5 | 2008-08-08 23:50:38 | [diff] [blame] | 1817 | } |
| 1818 | } |
| 1819 | |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 1820 | static int matroska_parse_seekhead_entry(MatroskaDemuxContext *matroska, |
Andreas Rheinhardt | 730ac1a | 2019-05-16 22:30:20 | [diff] [blame] | 1821 | int64_t pos) |
Aurelien Jacobs | 13b350a | 2008-08-05 00:40:36 | [diff] [blame] | 1822 | { |
Andreas Rheinhardt | 60f75c9 | 2019-05-16 22:30:13 | [diff] [blame] | 1823 | uint32_t saved_id = matroska->current_id; |
Anton Khirnov | 384c9c2 | 2011-03-03 19:11:45 | [diff] [blame] | 1824 | int64_t before_pos = avio_tell(matroska->ctx->pb); |
Aaron Colwell | 31ad14c | 2011-07-09 05:48:43 | [diff] [blame] | 1825 | int ret = 0; |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 1826 | |
Anton Khirnov | f47ac3c | 2011-07-09 06:11:30 | [diff] [blame] | 1827 | /* seek */ |
Andreas Rheinhardt | 730ac1a | 2019-05-16 22:30:20 | [diff] [blame] | 1828 | if (avio_seek(matroska->ctx->pb, pos, SEEK_SET) == pos) { |
Diego Biurrun | 5968d2d | 2008-08-05 08:28:57 | [diff] [blame] | 1829 | /* We don't want to lose our seekhead level, so we add |
Aurelien Jacobs | 4348571 | 2008-08-05 00:40:43 | [diff] [blame] | 1830 | * a dummy. This is a crude hack. */ |
| 1831 | if (matroska->num_levels == EBML_MAX_DEPTH) { |
| 1832 | av_log(matroska->ctx, AV_LOG_INFO, |
| 1833 | "Max EBML element depth (%d) reached, " |
| 1834 | "cannot parse further.\n", EBML_MAX_DEPTH); |
Aaron Colwell | 31ad14c | 2011-07-09 05:48:43 | [diff] [blame] | 1835 | ret = AVERROR_INVALIDDATA; |
| 1836 | } else { |
Andreas Rheinhardt | 60f75c9 | 2019-05-16 22:30:13 | [diff] [blame] | 1837 | matroska->levels[matroska->num_levels] = (MatroskaLevel) { 0, EBML_UNKNOWN_LENGTH }; |
Anton Khirnov | f47ac3c | 2011-07-09 06:11:30 | [diff] [blame] | 1838 | matroska->num_levels++; |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 1839 | matroska->current_id = 0; |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 1840 | |
Dustin Brody | 4a9628f | 2011-09-08 22:43:32 | [diff] [blame] | 1841 | ret = ebml_parse(matroska, matroska_segment, matroska); |
Andreas Rheinhardt | b31c9b7 | 2019-06-23 23:46:54 | [diff] [blame] | 1842 | if (ret == LEVEL_ENDED) { |
| 1843 | /* This can only happen if the seek brought us beyond EOF. */ |
| 1844 | ret = AVERROR_EOF; |
| 1845 | } |
Aurelien Jacobs | 4348571 | 2008-08-05 00:40:43 | [diff] [blame] | 1846 | } |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 1847 | } |
Andreas Rheinhardt | a9f0515 | 2019-05-16 22:30:09 | [diff] [blame] | 1848 | /* Seek back - notice that in all instances where this is used |
| 1849 | * it is safe to set the level to 1. */ |
Andreas Rheinhardt | 8a286e7 | 2019-05-16 22:30:03 | [diff] [blame] | 1850 | matroska_reset_status(matroska, saved_id, before_pos); |
Aaron Colwell | 31ad14c | 2011-07-09 05:48:43 | [diff] [blame] | 1851 | |
| 1852 | return ret; |
| 1853 | } |
| 1854 | |
| 1855 | static void matroska_execute_seekhead(MatroskaDemuxContext *matroska) |
| 1856 | { |
| 1857 | EbmlList *seekhead_list = &matroska->seekhead; |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 1858 | int i; |
| 1859 | |
| 1860 | // we should not do any seeking in the streaming case |
James Almer | 4de591e | 2017-03-21 20:02:30 | [diff] [blame] | 1861 | if (!(matroska->ctx->pb->seekable & AVIO_SEEKABLE_NORMAL)) |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 1862 | return; |
| 1863 | |
wm4 | cac2295 | 2015-02-09 19:39:00 | [diff] [blame] | 1864 | for (i = 0; i < seekhead_list->nb_elem; i++) { |
| 1865 | MatroskaSeekhead *seekheads = seekhead_list->elem; |
Andreas Rheinhardt | 730ac1a | 2019-05-16 22:30:20 | [diff] [blame] | 1866 | uint32_t id = seekheads[i].id; |
| 1867 | int64_t pos = seekheads[i].pos + matroska->segment_start; |
wm4 | 6551aca | 2014-12-06 15:53:30 | [diff] [blame] | 1868 | |
wm4 | cac2295 | 2015-02-09 19:39:00 | [diff] [blame] | 1869 | MatroskaLevel1Element *elem = matroska_find_level1_elem(matroska, id); |
| 1870 | if (!elem || elem->parsed) |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 1871 | continue; |
| 1872 | |
wm4 | cac2295 | 2015-02-09 19:39:00 | [diff] [blame] | 1873 | elem->pos = pos; |
| 1874 | |
Aaron Colwell | 31ad14c | 2011-07-09 05:48:43 | [diff] [blame] | 1875 | // defer cues parsing until we actually need cue data. |
wm4 | cac2295 | 2015-02-09 19:39:00 | [diff] [blame] | 1876 | if (id == MATROSKA_ID_CUES) |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 1877 | continue; |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 1878 | |
wm4 | cac2295 | 2015-02-09 19:39:00 | [diff] [blame] | 1879 | if (matroska_parse_seekhead_entry(matroska, pos) < 0) { |
Reimar Döffinger | 47e015e | 2012-02-12 13:09:03 | [diff] [blame] | 1880 | // mark index as broken |
| 1881 | matroska->cues_parsing_deferred = -1; |
Aaron Colwell | 31ad14c | 2011-07-09 05:48:43 | [diff] [blame] | 1882 | break; |
Reimar Döffinger | 47e015e | 2012-02-12 13:09:03 | [diff] [blame] | 1883 | } |
wm4 | cac2295 | 2015-02-09 19:39:00 | [diff] [blame] | 1884 | |
| 1885 | elem->parsed = 1; |
Michael Niedermayer | 72c9844 | 2014-12-07 21:45:34 | [diff] [blame] | 1886 | } |
Aaron Colwell | 31ad14c | 2011-07-09 05:48:43 | [diff] [blame] | 1887 | } |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 1888 | |
Michael Niedermayer | b3d9ab1 | 2014-03-07 22:46:37 | [diff] [blame] | 1889 | static void matroska_add_index_entries(MatroskaDemuxContext *matroska) |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 1890 | { |
Aaron Colwell | 31ad14c | 2011-07-09 05:48:43 | [diff] [blame] | 1891 | EbmlList *index_list; |
| 1892 | MatroskaIndex *index; |
Andreas Cadhalpun | eb9fb50 | 2015-05-03 21:07:20 | [diff] [blame] | 1893 | uint64_t index_scale = 1; |
Aaron Colwell | 31ad14c | 2011-07-09 05:48:43 | [diff] [blame] | 1894 | int i, j; |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 1895 | |
wm4 | 7862325 | 2015-02-09 19:39:01 | [diff] [blame] | 1896 | if (matroska->ctx->flags & AVFMT_FLAG_IGNIDX) |
| 1897 | return; |
| 1898 | |
Aaron Colwell | 31ad14c | 2011-07-09 05:48:43 | [diff] [blame] | 1899 | index_list = &matroska->index; |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 1900 | index = index_list->elem; |
Rodger Combs | 4f7d9b7 | 2015-10-08 20:34:59 | [diff] [blame] | 1901 | if (index_list->nb_elem < 2) |
| 1902 | return; |
| 1903 | if (index[1].time > 1E14 / matroska->time_scale) { |
| 1904 | av_log(matroska->ctx, AV_LOG_WARNING, "Dropping apparently-broken index.\n"); |
| 1905 | return; |
Aaron Colwell | 31ad14c | 2011-07-09 05:48:43 | [diff] [blame] | 1906 | } |
| 1907 | for (i = 0; i < index_list->nb_elem; i++) { |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 1908 | EbmlList *pos_list = &index[i].pos; |
Aaron Colwell | 31ad14c | 2011-07-09 05:48:43 | [diff] [blame] | 1909 | MatroskaIndexPos *pos = pos_list->elem; |
| 1910 | for (j = 0; j < pos_list->nb_elem; j++) { |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 1911 | MatroskaTrack *track = matroska_find_track_by_num(matroska, |
| 1912 | pos[j].track); |
Aaron Colwell | 31ad14c | 2011-07-09 05:48:43 | [diff] [blame] | 1913 | if (track && track->stream) |
| 1914 | av_add_index_entry(track->stream, |
| 1915 | pos[j].pos + matroska->segment_start, |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 1916 | index[i].time / index_scale, 0, 0, |
Aaron Colwell | 31ad14c | 2011-07-09 05:48:43 | [diff] [blame] | 1917 | AVINDEX_KEYFRAME); |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 1918 | } |
| 1919 | } |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 1920 | } |
| 1921 | |
Aaron Colwell | 6c4cc0f | 2011-09-22 14:51:00 | [diff] [blame] | 1922 | static void matroska_parse_cues(MatroskaDemuxContext *matroska) { |
Aaron Colwell | 6c4cc0f | 2011-09-22 14:51:00 | [diff] [blame] | 1923 | int i; |
| 1924 | |
wm4 | 7862325 | 2015-02-09 19:39:01 | [diff] [blame] | 1925 | if (matroska->ctx->flags & AVFMT_FLAG_IGNIDX) |
| 1926 | return; |
| 1927 | |
wm4 | cac2295 | 2015-02-09 19:39:00 | [diff] [blame] | 1928 | for (i = 0; i < matroska->num_level1_elems; i++) { |
| 1929 | MatroskaLevel1Element *elem = &matroska->level1_elems[i]; |
| 1930 | if (elem->id == MATROSKA_ID_CUES && !elem->parsed) { |
| 1931 | if (matroska_parse_seekhead_entry(matroska, elem->pos) < 0) |
| 1932 | matroska->cues_parsing_deferred = -1; |
| 1933 | elem->parsed = 1; |
Aaron Colwell | 6c4cc0f | 2011-09-22 14:51:00 | [diff] [blame] | 1934 | break; |
wm4 | cac2295 | 2015-02-09 19:39:00 | [diff] [blame] | 1935 | } |
| 1936 | } |
Aaron Colwell | 6c4cc0f | 2011-09-22 14:51:00 | [diff] [blame] | 1937 | |
Aaron Colwell | 6c4cc0f | 2011-09-22 14:51:00 | [diff] [blame] | 1938 | matroska_add_index_entries(matroska); |
| 1939 | } |
| 1940 | |
Aurelien Jacobs | f7b9687 | 2008-08-05 00:42:05 | [diff] [blame] | 1941 | static int matroska_aac_profile(char *codec_id) |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 1942 | { |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 1943 | static const char *const aac_profiles[] = { "MAIN", "LC", "SSR" }; |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 1944 | int profile; |
| 1945 | |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 1946 | for (profile = 0; profile < FF_ARRAY_ELEMS(aac_profiles); profile++) |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 1947 | if (strstr(codec_id, aac_profiles[profile])) |
| 1948 | break; |
| 1949 | return profile + 1; |
| 1950 | } |
| 1951 | |
Aurelien Jacobs | f7b9687 | 2008-08-05 00:42:05 | [diff] [blame] | 1952 | static int matroska_aac_sri(int samplerate) |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 1953 | { |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 1954 | int sri; |
| 1955 | |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 1956 | for (sri = 0; sri < FF_ARRAY_ELEMS(avpriv_mpeg4audio_sample_rates); sri++) |
Anton Khirnov | 59a9a23 | 2011-10-17 07:28:53 | [diff] [blame] | 1957 | if (avpriv_mpeg4audio_sample_rates[sri] == samplerate) |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 1958 | break; |
| 1959 | return sri; |
| 1960 | } |
| 1961 | |
Aaron Colwell | 2e06113 | 2012-03-05 18:02:48 | [diff] [blame] | 1962 | static void matroska_metadata_creation_time(AVDictionary **metadata, int64_t date_utc) |
| 1963 | { |
Aaron Colwell | 2e06113 | 2012-03-05 18:02:48 | [diff] [blame] | 1964 | /* Convert to seconds and adjust by number of seconds between 2001-01-01 and Epoch */ |
Marton Balint | 13b90ff | 2016-06-29 22:55:48 | [diff] [blame] | 1965 | avpriv_dict_set_timestamp(metadata, "creation_time", date_utc / 1000 + 978307200000000LL); |
Aaron Colwell | 2e06113 | 2012-03-05 18:02:48 | [diff] [blame] | 1966 | } |
| 1967 | |
Anton Khirnov | 4efdadc | 2014-05-25 12:05:51 | [diff] [blame] | 1968 | static int matroska_parse_flac(AVFormatContext *s, |
| 1969 | MatroskaTrack *track, |
| 1970 | int *offset) |
| 1971 | { |
Anton Khirnov | 23f741f | 2014-05-26 10:48:56 | [diff] [blame] | 1972 | AVStream *st = track->stream; |
Anton Khirnov | 4efdadc | 2014-05-25 12:05:51 | [diff] [blame] | 1973 | uint8_t *p = track->codec_priv.data; |
| 1974 | int size = track->codec_priv.size; |
| 1975 | |
| 1976 | if (size < 8 + FLAC_STREAMINFO_SIZE || p[4] & 0x7f) { |
| 1977 | av_log(s, AV_LOG_WARNING, "Invalid FLAC private data\n"); |
| 1978 | track->codec_priv.size = 0; |
| 1979 | return 0; |
| 1980 | } |
| 1981 | *offset = 8; |
| 1982 | track->codec_priv.size = 8 + FLAC_STREAMINFO_SIZE; |
| 1983 | |
Anton Khirnov | 23f741f | 2014-05-26 10:48:56 | [diff] [blame] | 1984 | p += track->codec_priv.size; |
| 1985 | size -= track->codec_priv.size; |
| 1986 | |
| 1987 | /* parse the remaining metadata blocks if present */ |
| 1988 | while (size >= 4) { |
| 1989 | int block_last, block_type, block_size; |
| 1990 | |
| 1991 | flac_parse_block_header(p, &block_last, &block_type, &block_size); |
| 1992 | |
| 1993 | p += 4; |
| 1994 | size -= 4; |
| 1995 | if (block_size > size) |
| 1996 | return 0; |
| 1997 | |
| 1998 | /* check for the channel mask */ |
| 1999 | if (block_type == FLAC_METADATA_TYPE_VORBIS_COMMENT) { |
| 2000 | AVDictionary *dict = NULL; |
| 2001 | AVDictionaryEntry *chmask; |
| 2002 | |
| 2003 | ff_vorbis_comment(s, &dict, p, block_size, 0); |
| 2004 | chmask = av_dict_get(dict, "WAVEFORMATEXTENSIBLE_CHANNEL_MASK", NULL, 0); |
| 2005 | if (chmask) { |
| 2006 | uint64_t mask = strtol(chmask->value, NULL, 0); |
| 2007 | if (!mask || mask & ~0x3ffffULL) { |
| 2008 | av_log(s, AV_LOG_WARNING, |
| 2009 | "Invalid value of WAVEFORMATEXTENSIBLE_CHANNEL_MASK\n"); |
| 2010 | } else |
Anton Khirnov | 9200514 | 2014-06-18 18:42:52 | [diff] [blame] | 2011 | st->codecpar->channel_layout = mask; |
Anton Khirnov | 23f741f | 2014-05-26 10:48:56 | [diff] [blame] | 2012 | } |
| 2013 | av_dict_free(&dict); |
| 2014 | } |
| 2015 | |
| 2016 | p += block_size; |
| 2017 | size -= block_size; |
| 2018 | } |
| 2019 | |
Anton Khirnov | 4efdadc | 2014-05-25 12:05:51 | [diff] [blame] | 2020 | return 0; |
| 2021 | } |
| 2022 | |
James Almer | 1273bc6 | 2016-10-13 17:22:07 | [diff] [blame] | 2023 | static int mkv_field_order(MatroskaDemuxContext *matroska, int64_t field_order) |
Luca Barbato | 5f02266 | 2016-03-28 18:29:54 | [diff] [blame] | 2024 | { |
James Almer | 1273bc6 | 2016-10-13 17:22:07 | [diff] [blame] | 2025 | int major, minor, micro, bttb = 0; |
| 2026 | |
| 2027 | /* workaround a bug in our Matroska muxer, introduced in version 57.36 alongside |
| 2028 | * this function, and fixed in 57.52 */ |
Andreas Cadhalpun | eb751f0 | 2016-10-16 19:47:00 | [diff] [blame] | 2029 | if (matroska->muxingapp && sscanf(matroska->muxingapp, "Lavf%d.%d.%d", &major, &minor, µ) == 3) |
James Almer | 1273bc6 | 2016-10-13 17:22:07 | [diff] [blame] | 2030 | bttb = (major == 57 && minor >= 36 && minor <= 51 && micro >= 100); |
| 2031 | |
Luca Barbato | 5f02266 | 2016-03-28 18:29:54 | [diff] [blame] | 2032 | switch (field_order) { |
| 2033 | case MATROSKA_VIDEO_FIELDORDER_PROGRESSIVE: |
| 2034 | return AV_FIELD_PROGRESSIVE; |
| 2035 | case MATROSKA_VIDEO_FIELDORDER_UNDETERMINED: |
| 2036 | return AV_FIELD_UNKNOWN; |
| 2037 | case MATROSKA_VIDEO_FIELDORDER_TT: |
| 2038 | return AV_FIELD_TT; |
| 2039 | case MATROSKA_VIDEO_FIELDORDER_BB: |
| 2040 | return AV_FIELD_BB; |
| 2041 | case MATROSKA_VIDEO_FIELDORDER_BT: |
James Almer | 1273bc6 | 2016-10-13 17:22:07 | [diff] [blame] | 2042 | return bttb ? AV_FIELD_TB : AV_FIELD_BT; |
Luca Barbato | 5f02266 | 2016-03-28 18:29:54 | [diff] [blame] | 2043 | case MATROSKA_VIDEO_FIELDORDER_TB: |
James Almer | 1273bc6 | 2016-10-13 17:22:07 | [diff] [blame] | 2044 | return bttb ? AV_FIELD_BT : AV_FIELD_TB; |
Luca Barbato | 5f02266 | 2016-03-28 18:29:54 | [diff] [blame] | 2045 | default: |
| 2046 | return AV_FIELD_UNKNOWN; |
| 2047 | } |
| 2048 | } |
| 2049 | |
Aaron Colwell | febfb49 | 2015-12-02 23:13:18 | [diff] [blame] | 2050 | static void mkv_stereo_mode_display_mul(int stereo_mode, |
| 2051 | int *h_width, int *h_height) |
Aaron Colwell | ec83efd | 2015-11-23 20:06:14 | [diff] [blame] | 2052 | { |
| 2053 | switch (stereo_mode) { |
| 2054 | case MATROSKA_VIDEO_STEREOMODE_TYPE_MONO: |
| 2055 | case MATROSKA_VIDEO_STEREOMODE_TYPE_CHECKERBOARD_RL: |
| 2056 | case MATROSKA_VIDEO_STEREOMODE_TYPE_CHECKERBOARD_LR: |
| 2057 | case MATROSKA_VIDEO_STEREOMODE_TYPE_BOTH_EYES_BLOCK_RL: |
| 2058 | case MATROSKA_VIDEO_STEREOMODE_TYPE_BOTH_EYES_BLOCK_LR: |
| 2059 | break; |
| 2060 | case MATROSKA_VIDEO_STEREOMODE_TYPE_RIGHT_LEFT: |
| 2061 | case MATROSKA_VIDEO_STEREOMODE_TYPE_LEFT_RIGHT: |
| 2062 | case MATROSKA_VIDEO_STEREOMODE_TYPE_COL_INTERLEAVED_RL: |
| 2063 | case MATROSKA_VIDEO_STEREOMODE_TYPE_COL_INTERLEAVED_LR: |
| 2064 | *h_width = 2; |
| 2065 | break; |
| 2066 | case MATROSKA_VIDEO_STEREOMODE_TYPE_BOTTOM_TOP: |
| 2067 | case MATROSKA_VIDEO_STEREOMODE_TYPE_TOP_BOTTOM: |
| 2068 | case MATROSKA_VIDEO_STEREOMODE_TYPE_ROW_INTERLEAVED_RL: |
| 2069 | case MATROSKA_VIDEO_STEREOMODE_TYPE_ROW_INTERLEAVED_LR: |
| 2070 | *h_height = 2; |
| 2071 | break; |
| 2072 | } |
| 2073 | } |
| 2074 | |
Neil Birkbeck | bbda13a | 2016-03-01 00:41:04 | [diff] [blame] | 2075 | static int mkv_parse_video_color(AVStream *st, const MatroskaTrack *track) { |
James Almer | 4e75907 | 2016-12-05 02:22:39 | [diff] [blame] | 2076 | const MatroskaTrackVideoColor *color = track->video.color.elem; |
| 2077 | const MatroskaMasteringMeta *mastering_meta; |
| 2078 | int has_mastering_primaries, has_mastering_luminance; |
| 2079 | |
| 2080 | if (!track->video.color.nb_elem) |
| 2081 | return 0; |
| 2082 | |
| 2083 | mastering_meta = &color->mastering_meta; |
Neil Birkbeck | bbda13a | 2016-03-01 00:41:04 | [diff] [blame] | 2084 | // Mastering primaries are CIE 1931 coords, and must be > 0. |
James Almer | 4e75907 | 2016-12-05 02:22:39 | [diff] [blame] | 2085 | has_mastering_primaries = |
Neil Birkbeck | bbda13a | 2016-03-01 00:41:04 | [diff] [blame] | 2086 | mastering_meta->r_x > 0 && mastering_meta->r_y > 0 && |
| 2087 | mastering_meta->g_x > 0 && mastering_meta->g_y > 0 && |
| 2088 | mastering_meta->b_x > 0 && mastering_meta->b_y > 0 && |
| 2089 | mastering_meta->white_x > 0 && mastering_meta->white_y > 0; |
James Almer | 4e75907 | 2016-12-05 02:22:39 | [diff] [blame] | 2090 | has_mastering_luminance = mastering_meta->max_luminance > 0; |
Neil Birkbeck | bbda13a | 2016-03-01 00:41:04 | [diff] [blame] | 2091 | |
James Almer | 4e75907 | 2016-12-05 02:22:39 | [diff] [blame] | 2092 | if (color->matrix_coefficients != AVCOL_SPC_RESERVED) |
| 2093 | st->codecpar->color_space = color->matrix_coefficients; |
| 2094 | if (color->primaries != AVCOL_PRI_RESERVED && |
| 2095 | color->primaries != AVCOL_PRI_RESERVED0) |
| 2096 | st->codecpar->color_primaries = color->primaries; |
| 2097 | if (color->transfer_characteristics != AVCOL_TRC_RESERVED && |
| 2098 | color->transfer_characteristics != AVCOL_TRC_RESERVED0) |
| 2099 | st->codecpar->color_trc = color->transfer_characteristics; |
| 2100 | if (color->range != AVCOL_RANGE_UNSPECIFIED && |
| 2101 | color->range <= AVCOL_RANGE_JPEG) |
| 2102 | st->codecpar->color_range = color->range; |
| 2103 | if (color->chroma_siting_horz != MATROSKA_COLOUR_CHROMASITINGHORZ_UNDETERMINED && |
| 2104 | color->chroma_siting_vert != MATROSKA_COLOUR_CHROMASITINGVERT_UNDETERMINED && |
| 2105 | color->chroma_siting_horz < MATROSKA_COLOUR_CHROMASITINGHORZ_NB && |
| 2106 | color->chroma_siting_vert < MATROSKA_COLOUR_CHROMASITINGVERT_NB) { |
James Almer | 47ee6f1 | 2016-10-18 02:50:14 | [diff] [blame] | 2107 | st->codecpar->chroma_location = |
James Almer | 4e75907 | 2016-12-05 02:22:39 | [diff] [blame] | 2108 | avcodec_chroma_pos_to_enum((color->chroma_siting_horz - 1) << 7, |
| 2109 | (color->chroma_siting_vert - 1) << 7); |
James Almer | 47ee6f1 | 2016-10-18 02:50:14 | [diff] [blame] | 2110 | } |
James Almer | 095147a | 2017-04-25 18:04:00 | [diff] [blame] | 2111 | if (color->max_cll && color->max_fall) { |
| 2112 | size_t size = 0; |
| 2113 | int ret; |
| 2114 | AVContentLightMetadata *metadata = av_content_light_metadata_alloc(&size); |
| 2115 | if (!metadata) |
| 2116 | return AVERROR(ENOMEM); |
| 2117 | ret = av_stream_add_side_data(st, AV_PKT_DATA_CONTENT_LIGHT_LEVEL, |
| 2118 | (uint8_t *)metadata, size); |
| 2119 | if (ret < 0) { |
| 2120 | av_freep(&metadata); |
| 2121 | return ret; |
| 2122 | } |
| 2123 | metadata->MaxCLL = color->max_cll; |
| 2124 | metadata->MaxFALL = color->max_fall; |
| 2125 | } |
Neil Birkbeck | bbda13a | 2016-03-01 00:41:04 | [diff] [blame] | 2126 | |
| 2127 | if (has_mastering_primaries || has_mastering_luminance) { |
Neil Birkbeck | bbda13a | 2016-03-01 00:41:04 | [diff] [blame] | 2128 | AVMasteringDisplayMetadata *metadata = |
| 2129 | (AVMasteringDisplayMetadata*) av_stream_new_side_data( |
| 2130 | st, AV_PKT_DATA_MASTERING_DISPLAY_METADATA, |
| 2131 | sizeof(AVMasteringDisplayMetadata)); |
| 2132 | if (!metadata) { |
| 2133 | return AVERROR(ENOMEM); |
| 2134 | } |
| 2135 | memset(metadata, 0, sizeof(AVMasteringDisplayMetadata)); |
| 2136 | if (has_mastering_primaries) { |
James Almer | 3b4e9a3 | 2019-10-03 23:52:20 | [diff] [blame] | 2137 | metadata->display_primaries[0][0] = av_d2q(mastering_meta->r_x, INT_MAX); |
| 2138 | metadata->display_primaries[0][1] = av_d2q(mastering_meta->r_y, INT_MAX); |
| 2139 | metadata->display_primaries[1][0] = av_d2q(mastering_meta->g_x, INT_MAX); |
| 2140 | metadata->display_primaries[1][1] = av_d2q(mastering_meta->g_y, INT_MAX); |
| 2141 | metadata->display_primaries[2][0] = av_d2q(mastering_meta->b_x, INT_MAX); |
| 2142 | metadata->display_primaries[2][1] = av_d2q(mastering_meta->b_y, INT_MAX); |
| 2143 | metadata->white_point[0] = av_d2q(mastering_meta->white_x, INT_MAX); |
| 2144 | metadata->white_point[1] = av_d2q(mastering_meta->white_y, INT_MAX); |
Neil Birkbeck | bbda13a | 2016-03-01 00:41:04 | [diff] [blame] | 2145 | metadata->has_primaries = 1; |
| 2146 | } |
| 2147 | if (has_mastering_luminance) { |
James Almer | 3b4e9a3 | 2019-10-03 23:52:20 | [diff] [blame] | 2148 | metadata->max_luminance = av_d2q(mastering_meta->max_luminance, INT_MAX); |
| 2149 | metadata->min_luminance = av_d2q(mastering_meta->min_luminance, INT_MAX); |
Neil Birkbeck | bbda13a | 2016-03-01 00:41:04 | [diff] [blame] | 2150 | metadata->has_luminance = 1; |
| 2151 | } |
| 2152 | } |
| 2153 | return 0; |
| 2154 | } |
| 2155 | |
James Almer | 445204c | 2016-12-06 17:48:45 | [diff] [blame] | 2156 | static int mkv_parse_video_projection(AVStream *st, const MatroskaTrack *track) { |
| 2157 | AVSphericalMapping *spherical; |
| 2158 | enum AVSphericalProjection projection; |
| 2159 | size_t spherical_size; |
Vittorio Giovara | f20bcec | 2017-03-16 21:20:47 | [diff] [blame] | 2160 | uint32_t l = 0, t = 0, r = 0, b = 0; |
| 2161 | uint32_t padding = 0; |
James Almer | 445204c | 2016-12-06 17:48:45 | [diff] [blame] | 2162 | int ret; |
Vittorio Giovara | bde9642 | 2017-02-10 21:02:22 | [diff] [blame] | 2163 | GetByteContext gb; |
| 2164 | |
| 2165 | bytestream2_init(&gb, track->video.projection.private.data, |
| 2166 | track->video.projection.private.size); |
| 2167 | |
| 2168 | if (bytestream2_get_byte(&gb) != 0) { |
| 2169 | av_log(NULL, AV_LOG_WARNING, "Unknown spherical metadata\n"); |
| 2170 | return 0; |
| 2171 | } |
| 2172 | |
| 2173 | bytestream2_skip(&gb, 3); // flags |
James Almer | 445204c | 2016-12-06 17:48:45 | [diff] [blame] | 2174 | |
| 2175 | switch (track->video.projection.type) { |
| 2176 | case MATROSKA_VIDEO_PROJECTION_TYPE_EQUIRECTANGULAR: |
Vittorio Giovara | 9ae3506 | 2017-03-07 22:34:32 | [diff] [blame] | 2177 | if (track->video.projection.private.size == 20) { |
Vittorio Giovara | bde9642 | 2017-02-10 21:02:22 | [diff] [blame] | 2178 | t = bytestream2_get_be32(&gb); |
| 2179 | b = bytestream2_get_be32(&gb); |
| 2180 | l = bytestream2_get_be32(&gb); |
| 2181 | r = bytestream2_get_be32(&gb); |
| 2182 | |
| 2183 | if (b >= UINT_MAX - t || r >= UINT_MAX - l) { |
| 2184 | av_log(NULL, AV_LOG_ERROR, |
| 2185 | "Invalid bounding rectangle coordinates " |
Vittorio Giovara | f20bcec | 2017-03-16 21:20:47 | [diff] [blame] | 2186 | "%"PRIu32",%"PRIu32",%"PRIu32",%"PRIu32"\n", |
Vittorio Giovara | bde9642 | 2017-02-10 21:02:22 | [diff] [blame] | 2187 | l, t, r, b); |
| 2188 | return AVERROR_INVALIDDATA; |
| 2189 | } |
Vittorio Giovara | 9ae3506 | 2017-03-07 22:34:32 | [diff] [blame] | 2190 | } else if (track->video.projection.private.size != 0) { |
Vittorio Giovara | bde9642 | 2017-02-10 21:02:22 | [diff] [blame] | 2191 | av_log(NULL, AV_LOG_ERROR, "Unknown spherical metadata\n"); |
| 2192 | return AVERROR_INVALIDDATA; |
| 2193 | } |
Vittorio Giovara | 9ae3506 | 2017-03-07 22:34:32 | [diff] [blame] | 2194 | |
| 2195 | if (l || t || r || b) |
| 2196 | projection = AV_SPHERICAL_EQUIRECTANGULAR_TILE; |
| 2197 | else |
| 2198 | projection = AV_SPHERICAL_EQUIRECTANGULAR; |
James Almer | 445204c | 2016-12-06 17:48:45 | [diff] [blame] | 2199 | break; |
| 2200 | case MATROSKA_VIDEO_PROJECTION_TYPE_CUBEMAP: |
Vittorio Giovara | bde9642 | 2017-02-10 21:02:22 | [diff] [blame] | 2201 | if (track->video.projection.private.size < 4) { |
| 2202 | av_log(NULL, AV_LOG_ERROR, "Missing projection private properties\n"); |
James Almer | 445204c | 2016-12-06 17:48:45 | [diff] [blame] | 2203 | return AVERROR_INVALIDDATA; |
Vittorio Giovara | bde9642 | 2017-02-10 21:02:22 | [diff] [blame] | 2204 | } else if (track->video.projection.private.size == 12) { |
| 2205 | uint32_t layout = bytestream2_get_be32(&gb); |
Vittorio Giovara | 9ae3506 | 2017-03-07 22:34:32 | [diff] [blame] | 2206 | if (layout) { |
Vittorio Giovara | bde9642 | 2017-02-10 21:02:22 | [diff] [blame] | 2207 | av_log(NULL, AV_LOG_WARNING, |
| 2208 | "Unknown spherical cubemap layout %"PRIu32"\n", layout); |
| 2209 | return 0; |
| 2210 | } |
Vittorio Giovara | 9ae3506 | 2017-03-07 22:34:32 | [diff] [blame] | 2211 | projection = AV_SPHERICAL_CUBEMAP; |
Vittorio Giovara | bde9642 | 2017-02-10 21:02:22 | [diff] [blame] | 2212 | padding = bytestream2_get_be32(&gb); |
| 2213 | } else { |
| 2214 | av_log(NULL, AV_LOG_ERROR, "Unknown spherical metadata\n"); |
| 2215 | return AVERROR_INVALIDDATA; |
| 2216 | } |
James Almer | 445204c | 2016-12-06 17:48:45 | [diff] [blame] | 2217 | break; |
James Almer | 55fe72a | 2017-11-02 23:42:05 | [diff] [blame] | 2218 | case MATROSKA_VIDEO_PROJECTION_TYPE_RECTANGULAR: |
| 2219 | /* No Spherical metadata */ |
| 2220 | return 0; |
James Almer | 445204c | 2016-12-06 17:48:45 | [diff] [blame] | 2221 | default: |
James Almer | 251849f | 2016-12-06 17:48:45 | [diff] [blame] | 2222 | av_log(NULL, AV_LOG_WARNING, |
| 2223 | "Unknown spherical metadata type %"PRIu64"\n", |
| 2224 | track->video.projection.type); |
James Almer | 445204c | 2016-12-06 17:48:45 | [diff] [blame] | 2225 | return 0; |
| 2226 | } |
| 2227 | |
| 2228 | spherical = av_spherical_alloc(&spherical_size); |
| 2229 | if (!spherical) |
| 2230 | return AVERROR(ENOMEM); |
James Almer | 251849f | 2016-12-06 17:48:45 | [diff] [blame] | 2231 | |
James Almer | 445204c | 2016-12-06 17:48:45 | [diff] [blame] | 2232 | spherical->projection = projection; |
| 2233 | |
James Almer | 251849f | 2016-12-06 17:48:45 | [diff] [blame] | 2234 | spherical->yaw = (int32_t) (track->video.projection.yaw * (1 << 16)); |
| 2235 | spherical->pitch = (int32_t) (track->video.projection.pitch * (1 << 16)); |
| 2236 | spherical->roll = (int32_t) (track->video.projection.roll * (1 << 16)); |
James Almer | 445204c | 2016-12-06 17:48:45 | [diff] [blame] | 2237 | |
Vittorio Giovara | bde9642 | 2017-02-10 21:02:22 | [diff] [blame] | 2238 | spherical->padding = padding; |
| 2239 | |
| 2240 | spherical->bound_left = l; |
| 2241 | spherical->bound_top = t; |
| 2242 | spherical->bound_right = r; |
| 2243 | spherical->bound_bottom = b; |
| 2244 | |
James Almer | 445204c | 2016-12-06 17:48:45 | [diff] [blame] | 2245 | ret = av_stream_add_side_data(st, AV_PKT_DATA_SPHERICAL, (uint8_t *)spherical, |
| 2246 | spherical_size); |
James Almer | 8396d54 | 2016-12-08 04:21:11 | [diff] [blame] | 2247 | if (ret < 0) { |
| 2248 | av_freep(&spherical); |
James Almer | 445204c | 2016-12-06 17:48:45 | [diff] [blame] | 2249 | return ret; |
James Almer | 8396d54 | 2016-12-08 04:21:11 | [diff] [blame] | 2250 | } |
James Almer | 445204c | 2016-12-06 17:48:45 | [diff] [blame] | 2251 | |
| 2252 | return 0; |
| 2253 | } |
| 2254 | |
Mats Peterson | 71f73ee | 2016-01-10 02:59:21 | [diff] [blame] | 2255 | static int get_qt_codec(MatroskaTrack *track, uint32_t *fourcc, enum AVCodecID *codec_id) |
| 2256 | { |
| 2257 | const AVCodecTag *codec_tags; |
| 2258 | |
| 2259 | codec_tags = track->type == MATROSKA_TRACK_TYPE_VIDEO ? |
| 2260 | ff_codec_movvideo_tags : ff_codec_movaudio_tags; |
| 2261 | |
| 2262 | /* Normalize noncompliant private data that starts with the fourcc |
| 2263 | * by expanding/shifting the data by 4 bytes and storing the data |
| 2264 | * size at the start. */ |
| 2265 | if (ff_codec_get_id(codec_tags, AV_RL32(track->codec_priv.data))) { |
James Almer | a618866 | 2018-04-04 17:12:54 | [diff] [blame] | 2266 | int ret = av_buffer_realloc(&track->codec_priv.buf, |
| 2267 | track->codec_priv.size + 4 + AV_INPUT_BUFFER_PADDING_SIZE); |
| 2268 | if (ret < 0) |
| 2269 | return ret; |
| 2270 | |
| 2271 | track->codec_priv.data = track->codec_priv.buf->data; |
| 2272 | memmove(track->codec_priv.data + 4, track->codec_priv.data, track->codec_priv.size); |
Mats Peterson | 71f73ee | 2016-01-10 02:59:21 | [diff] [blame] | 2273 | track->codec_priv.size += 4; |
| 2274 | AV_WB32(track->codec_priv.data, track->codec_priv.size); |
| 2275 | } |
| 2276 | |
| 2277 | *fourcc = AV_RL32(track->codec_priv.data + 4); |
| 2278 | *codec_id = ff_codec_get_id(codec_tags, *fourcc); |
| 2279 | |
| 2280 | return 0; |
| 2281 | } |
| 2282 | |
Anton Khirnov | 6df478b | 2014-05-25 07:07:32 | [diff] [blame] | 2283 | static int matroska_parse_tracks(AVFormatContext *s) |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 2284 | { |
| 2285 | MatroskaDemuxContext *matroska = s->priv_data; |
Anton Khirnov | 6df478b | 2014-05-25 07:07:32 | [diff] [blame] | 2286 | MatroskaTrack *tracks = matroska->tracks.elem; |
Aurelien Jacobs | 9a9a3b0 | 2008-08-05 00:40:49 | [diff] [blame] | 2287 | AVStream *st; |
Anton Khirnov | 4efdadc | 2014-05-25 12:05:51 | [diff] [blame] | 2288 | int i, j, ret; |
Michael Niedermayer | 69de229 | 2014-05-28 10:41:35 | [diff] [blame] | 2289 | int k; |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 2290 | |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 2291 | for (i = 0; i < matroska->tracks.nb_elem; i++) { |
Aurelien Jacobs | d88d806 | 2008-08-05 00:40:52 | [diff] [blame] | 2292 | MatroskaTrack *track = &tracks[i]; |
Anton Khirnov | 36ef536 | 2012-08-05 09:11:04 | [diff] [blame] | 2293 | enum AVCodecID codec_id = AV_CODEC_ID_NONE; |
Aurelien Jacobs | dc6c36c | 2011-08-17 22:21:21 | [diff] [blame] | 2294 | EbmlList *encodings_list = &track->encodings; |
Aurelien Jacobs | 9c25baf | 2008-08-05 00:40:55 | [diff] [blame] | 2295 | MatroskaTrackEncoding *encodings = encodings_list->elem; |
Aurelien Jacobs | d88d806 | 2008-08-05 00:40:52 | [diff] [blame] | 2296 | uint8_t *extradata = NULL; |
| 2297 | int extradata_size = 0; |
| 2298 | int extradata_offset = 0; |
Aurelien Jacobs | 5fec3a2 | 2011-06-13 23:58:11 | [diff] [blame] | 2299 | uint32_t fourcc = 0; |
Anton Khirnov | 471fe57 | 2011-02-20 10:04:12 | [diff] [blame] | 2300 | AVIOContext b; |
Frank Galligan | b853103 | 2013-03-07 16:11:38 | [diff] [blame] | 2301 | char* key_id_base64 = NULL; |
Carl Eugen Hoyos | 96fc290 | 2014-02-25 23:02:51 | [diff] [blame] | 2302 | int bit_depth = -1; |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 2303 | |
Aurelien Jacobs | d88d806 | 2008-08-05 00:40:52 | [diff] [blame] | 2304 | /* Apply some sanity checks. */ |
Aurelien Jacobs | 9c25baf | 2008-08-05 00:40:55 | [diff] [blame] | 2305 | if (track->type != MATROSKA_TRACK_TYPE_VIDEO && |
| 2306 | track->type != MATROSKA_TRACK_TYPE_AUDIO && |
Matthew Heaney | 818ebe9 | 2013-08-08 22:40:03 | [diff] [blame] | 2307 | track->type != MATROSKA_TRACK_TYPE_SUBTITLE && |
| 2308 | track->type != MATROSKA_TRACK_TYPE_METADATA) { |
Aurelien Jacobs | 9c25baf | 2008-08-05 00:40:55 | [diff] [blame] | 2309 | av_log(matroska->ctx, AV_LOG_INFO, |
| 2310 | "Unknown or unsupported track type %"PRIu64"\n", |
| 2311 | track->type); |
| 2312 | continue; |
| 2313 | } |
Gabriel Dume | f929ab0 | 2014-08-14 20:31:24 | [diff] [blame] | 2314 | if (!track->codec_id) |
Aurelien Jacobs | d88d806 | 2008-08-05 00:40:52 | [diff] [blame] | 2315 | continue; |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 2316 | |
Andreas Cadhalpun | 5b76c82 | 2015-06-15 18:59:22 | [diff] [blame] | 2317 | if (track->audio.samplerate < 0 || track->audio.samplerate > INT_MAX || |
| 2318 | isnan(track->audio.samplerate)) { |
| 2319 | av_log(matroska->ctx, AV_LOG_WARNING, |
| 2320 | "Invalid sample rate %f, defaulting to 8000 instead.\n", |
| 2321 | track->audio.samplerate); |
| 2322 | track->audio.samplerate = 8000; |
| 2323 | } |
| 2324 | |
Aurelien Jacobs | 9c25baf | 2008-08-05 00:40:55 | [diff] [blame] | 2325 | if (track->type == MATROSKA_TRACK_TYPE_VIDEO) { |
Nikolas Bowe | e07649e | 2018-01-18 23:21:56 | [diff] [blame] | 2326 | if (!track->default_duration && track->video.frame_rate > 0) { |
| 2327 | double default_duration = 1000000000 / track->video.frame_rate; |
| 2328 | if (default_duration > UINT64_MAX || default_duration < 0) { |
| 2329 | av_log(matroska->ctx, AV_LOG_WARNING, |
| 2330 | "Invalid frame rate %e. Cannot calculate default duration.\n", |
| 2331 | track->video.frame_rate); |
| 2332 | } else { |
| 2333 | track->default_duration = default_duration; |
| 2334 | } |
| 2335 | } |
Michael Niedermayer | f51ce34 | 2013-04-03 21:40:13 | [diff] [blame] | 2336 | if (track->video.display_width == -1) |
Aurelien Jacobs | 9c25baf | 2008-08-05 00:40:55 | [diff] [blame] | 2337 | track->video.display_width = track->video.pixel_width; |
Michael Niedermayer | f51ce34 | 2013-04-03 21:40:13 | [diff] [blame] | 2338 | if (track->video.display_height == -1) |
Aurelien Jacobs | 9c25baf | 2008-08-05 00:40:55 | [diff] [blame] | 2339 | track->video.display_height = track->video.pixel_height; |
Aurelien Jacobs | fdb5e02 | 2011-06-14 00:00:06 | [diff] [blame] | 2340 | if (track->video.color_space.size == 4) |
| 2341 | fourcc = AV_RL32(track->video.color_space.data); |
Aurelien Jacobs | 9c25baf | 2008-08-05 00:40:55 | [diff] [blame] | 2342 | } else if (track->type == MATROSKA_TRACK_TYPE_AUDIO) { |
| 2343 | if (!track->audio.out_samplerate) |
| 2344 | track->audio.out_samplerate = track->audio.samplerate; |
| 2345 | } |
| 2346 | if (encodings_list->nb_elem > 1) { |
| 2347 | av_log(matroska->ctx, AV_LOG_ERROR, |
Dustin Brody | d7d2f0e | 2011-09-15 07:34:38 | [diff] [blame] | 2348 | "Multiple combined encodings not supported"); |
Aurelien Jacobs | 9c25baf | 2008-08-05 00:40:55 | [diff] [blame] | 2349 | } else if (encodings_list->nb_elem == 1) { |
Frank Galligan | b853103 | 2013-03-07 16:11:38 | [diff] [blame] | 2350 | if (encodings[0].type) { |
| 2351 | if (encodings[0].encryption.key_id.size > 0) { |
| 2352 | /* Save the encryption key id to be stored later as a |
| 2353 | metadata tag. */ |
| 2354 | const int b64_size = AV_BASE64_SIZE(encodings[0].encryption.key_id.size); |
| 2355 | key_id_base64 = av_malloc(b64_size); |
| 2356 | if (key_id_base64 == NULL) |
| 2357 | return AVERROR(ENOMEM); |
| 2358 | |
| 2359 | av_base64_encode(key_id_base64, b64_size, |
| 2360 | encodings[0].encryption.key_id.data, |
| 2361 | encodings[0].encryption.key_id.size); |
| 2362 | } else { |
| 2363 | encodings[0].scope = 0; |
| 2364 | av_log(matroska->ctx, AV_LOG_ERROR, |
| 2365 | "Unsupported encoding type"); |
| 2366 | } |
| 2367 | } else if ( |
Aurelien Jacobs | b250f9c | 2009-01-13 23:44:16 | [diff] [blame] | 2368 | #if CONFIG_ZLIB |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 2369 | encodings[0].compression.algo != MATROSKA_TRACK_ENCODING_COMP_ZLIB && |
Aurelien Jacobs | 9c25baf | 2008-08-05 00:40:55 | [diff] [blame] | 2370 | #endif |
Aurelien Jacobs | b250f9c | 2009-01-13 23:44:16 | [diff] [blame] | 2371 | #if CONFIG_BZLIB |
Aurelien Jacobs | 9c25baf | 2008-08-05 00:40:55 | [diff] [blame] | 2372 | encodings[0].compression.algo != MATROSKA_TRACK_ENCODING_COMP_BZLIB && |
| 2373 | #endif |
Diego Biurrun | 2a91ada | 2012-10-18 17:48:27 | [diff] [blame] | 2374 | #if CONFIG_LZO |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 2375 | encodings[0].compression.algo != MATROSKA_TRACK_ENCODING_COMP_LZO && |
Diego Biurrun | 2a91ada | 2012-10-18 17:48:27 | [diff] [blame] | 2376 | #endif |
Frank Galligan | b853103 | 2013-03-07 16:11:38 | [diff] [blame] | 2377 | encodings[0].compression.algo != MATROSKA_TRACK_ENCODING_COMP_HEADERSTRIP) { |
Aurelien Jacobs | 9c25baf | 2008-08-05 00:40:55 | [diff] [blame] | 2378 | encodings[0].scope = 0; |
| 2379 | av_log(matroska->ctx, AV_LOG_ERROR, |
| 2380 | "Unsupported encoding type"); |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 2381 | } else if (track->codec_priv.size && encodings[0].scope & 2) { |
Aurelien Jacobs | 9c25baf | 2008-08-05 00:40:55 | [diff] [blame] | 2382 | uint8_t *codec_priv = track->codec_priv.data; |
Luca Barbato | 8d4dd55 | 2012-09-14 16:51:49 | [diff] [blame] | 2383 | int ret = matroska_decode_buffer(&track->codec_priv.data, |
| 2384 | &track->codec_priv.size, |
| 2385 | track); |
| 2386 | if (ret < 0) { |
Aurelien Jacobs | 9c25baf | 2008-08-05 00:40:55 | [diff] [blame] | 2387 | track->codec_priv.data = NULL; |
| 2388 | track->codec_priv.size = 0; |
| 2389 | av_log(matroska->ctx, AV_LOG_ERROR, |
| 2390 | "Failed to decode codec private data\n"); |
Aurelien Jacobs | 9c25baf | 2008-08-05 00:40:55 | [diff] [blame] | 2391 | } |
Luca Barbato | 8d4dd55 | 2012-09-14 16:51:49 | [diff] [blame] | 2392 | |
James Almer | a618866 | 2018-04-04 17:12:54 | [diff] [blame] | 2393 | if (codec_priv != track->codec_priv.data) { |
| 2394 | av_buffer_unref(&track->codec_priv.buf); |
| 2395 | if (track->codec_priv.data) { |
| 2396 | track->codec_priv.buf = av_buffer_create(track->codec_priv.data, |
| 2397 | track->codec_priv.size + AV_INPUT_BUFFER_PADDING_SIZE, |
| 2398 | NULL, NULL, 0); |
| 2399 | if (!track->codec_priv.buf) { |
| 2400 | av_freep(&track->codec_priv.data); |
| 2401 | track->codec_priv.size = 0; |
| 2402 | return AVERROR(ENOMEM); |
| 2403 | } |
| 2404 | } |
| 2405 | } |
Aurelien Jacobs | 9c25baf | 2008-08-05 00:40:55 | [diff] [blame] | 2406 | } |
| 2407 | } |
| 2408 | |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 2409 | for (j = 0; ff_mkv_codec_tags[j].id != AV_CODEC_ID_NONE; j++) { |
| 2410 | if (!strncmp(ff_mkv_codec_tags[j].str, track->codec_id, |
| 2411 | strlen(ff_mkv_codec_tags[j].str))) { |
| 2412 | codec_id = ff_mkv_codec_tags[j].id; |
Aurelien Jacobs | d88d806 | 2008-08-05 00:40:52 | [diff] [blame] | 2413 | break; |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 2414 | } |
Aurelien Jacobs | d88d806 | 2008-08-05 00:40:52 | [diff] [blame] | 2415 | } |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 2416 | |
Anton Khirnov | 3b3bbdd | 2011-06-18 09:43:24 | [diff] [blame] | 2417 | st = track->stream = avformat_new_stream(s, NULL); |
Michael Niedermayer | fb33bff | 2014-08-15 18:33:21 | [diff] [blame] | 2418 | if (!st) { |
Frank Galligan | b853103 | 2013-03-07 16:11:38 | [diff] [blame] | 2419 | av_free(key_id_base64); |
Aurelien Jacobs | d88d806 | 2008-08-05 00:40:52 | [diff] [blame] | 2420 | return AVERROR(ENOMEM); |
Frank Galligan | b853103 | 2013-03-07 16:11:38 | [diff] [blame] | 2421 | } |
| 2422 | |
| 2423 | if (key_id_base64) { |
| 2424 | /* export encryption key id as base64 metadata tag */ |
Andreas Rheinhardt | 40d9cbd | 2019-11-10 04:07:31 | [diff] [blame] | 2425 | av_dict_set(&st->metadata, "enc_key_id", key_id_base64, |
| 2426 | AV_DICT_DONT_STRDUP_VAL); |
Frank Galligan | b853103 | 2013-03-07 16:11:38 | [diff] [blame] | 2427 | } |
Aurelien Jacobs | d88d806 | 2008-08-05 00:40:52 | [diff] [blame] | 2428 | |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 2429 | if (!strcmp(track->codec_id, "V_MS/VFW/FOURCC") && |
Michael Niedermayer | b3d9ab1 | 2014-03-07 22:46:37 | [diff] [blame] | 2430 | track->codec_priv.size >= 40 && |
Gabriel Dume | 4b1f5e5 | 2014-08-14 20:31:25 | [diff] [blame] | 2431 | track->codec_priv.data) { |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 2432 | track->ms_compat = 1; |
Michael Niedermayer | b3d9ab1 | 2014-03-07 22:46:37 | [diff] [blame] | 2433 | bit_depth = AV_RL16(track->codec_priv.data + 14); |
| 2434 | fourcc = AV_RL32(track->codec_priv.data + 16); |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 2435 | codec_id = ff_codec_get_id(ff_codec_bmp_tags, |
Michael Niedermayer | b3d9ab1 | 2014-03-07 22:46:37 | [diff] [blame] | 2436 | fourcc); |
Carl Eugen Hoyos | ade5851 | 2014-04-27 07:34:13 | [diff] [blame] | 2437 | if (!codec_id) |
| 2438 | codec_id = ff_codec_get_id(ff_codec_movvideo_tags, |
| 2439 | fourcc); |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 2440 | extradata_offset = 40; |
| 2441 | } else if (!strcmp(track->codec_id, "A_MS/ACM") && |
| 2442 | track->codec_priv.size >= 14 && |
Gabriel Dume | 4b1f5e5 | 2014-08-14 20:31:25 | [diff] [blame] | 2443 | track->codec_priv.data) { |
Max Horn | ca402f3 | 2011-04-12 15:44:20 | [diff] [blame] | 2444 | int ret; |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 2445 | ffio_init_context(&b, track->codec_priv.data, |
| 2446 | track->codec_priv.size, |
Michael Niedermayer | d0b4504 | 2013-01-04 12:00:14 | [diff] [blame] | 2447 | 0, NULL, NULL, NULL, NULL); |
Derek Buitenhuis | 6f69f7a | 2016-04-10 19:58:15 | [diff] [blame] | 2448 | ret = ff_get_wav_header(s, &b, st->codecpar, track->codec_priv.size, 0); |
Max Horn | ca402f3 | 2011-04-12 15:44:20 | [diff] [blame] | 2449 | if (ret < 0) |
| 2450 | return ret; |
Anton Khirnov | 9200514 | 2014-06-18 18:42:52 | [diff] [blame] | 2451 | codec_id = st->codecpar->codec_id; |
Derek Buitenhuis | 6f69f7a | 2016-04-10 19:58:15 | [diff] [blame] | 2452 | fourcc = st->codecpar->codec_tag; |
Aurelien Jacobs | 038146e | 2009-10-01 21:14:05 | [diff] [blame] | 2453 | extradata_offset = FFMIN(track->codec_priv.size, 18); |
Michael Niedermayer | 4821858 | 2014-01-19 17:35:33 | [diff] [blame] | 2454 | } else if (!strcmp(track->codec_id, "A_QUICKTIME") |
Mats Peterson | 71f73ee | 2016-01-10 02:59:21 | [diff] [blame] | 2455 | /* Normally 36, but allow noncompliant private data */ |
| 2456 | && (track->codec_priv.size >= 32) |
Michael Niedermayer | 81a663f | 2014-08-15 19:31:59 | [diff] [blame] | 2457 | && (track->codec_priv.data)) { |
Mats Peterson | 535d09a | 2016-01-15 22:25:38 | [diff] [blame] | 2458 | uint16_t sample_size; |
Mats Peterson | 71f73ee | 2016-01-10 02:59:21 | [diff] [blame] | 2459 | int ret = get_qt_codec(track, &fourcc, &codec_id); |
| 2460 | if (ret < 0) |
| 2461 | return ret; |
Mats Peterson | 535d09a | 2016-01-15 22:25:38 | [diff] [blame] | 2462 | sample_size = AV_RB16(track->codec_priv.data + 26); |
Mats Peterson | 6f1466d | 2016-01-08 11:55:59 | [diff] [blame] | 2463 | if (fourcc == 0) { |
Mats Peterson | 535d09a | 2016-01-15 22:25:38 | [diff] [blame] | 2464 | if (sample_size == 8) { |
Mats Peterson | 6f1466d | 2016-01-08 11:55:59 | [diff] [blame] | 2465 | fourcc = MKTAG('r','a','w',' '); |
| 2466 | codec_id = ff_codec_get_id(ff_codec_movaudio_tags, fourcc); |
Mats Peterson | 535d09a | 2016-01-15 22:25:38 | [diff] [blame] | 2467 | } else if (sample_size == 16) { |
Mats Peterson | 6f1466d | 2016-01-08 11:55:59 | [diff] [blame] | 2468 | fourcc = MKTAG('t','w','o','s'); |
| 2469 | codec_id = ff_codec_get_id(ff_codec_movaudio_tags, fourcc); |
| 2470 | } |
| 2471 | } |
Mats Peterson | 535d09a | 2016-01-15 22:25:38 | [diff] [blame] | 2472 | if ((fourcc == MKTAG('t','w','o','s') || |
| 2473 | fourcc == MKTAG('s','o','w','t')) && |
| 2474 | sample_size == 8) |
| 2475 | codec_id = AV_CODEC_ID_PCM_S8; |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 2476 | } else if (!strcmp(track->codec_id, "V_QUICKTIME") && |
Michael Niedermayer | b3d9ab1 | 2014-03-07 22:46:37 | [diff] [blame] | 2477 | (track->codec_priv.size >= 21) && |
Gabriel Dume | 4b1f5e5 | 2014-08-14 20:31:25 | [diff] [blame] | 2478 | (track->codec_priv.data)) { |
Mats Peterson | 71f73ee | 2016-01-10 02:59:21 | [diff] [blame] | 2479 | int ret = get_qt_codec(track, &fourcc, &codec_id); |
| 2480 | if (ret < 0) |
| 2481 | return ret; |
Mats Peterson | 6a975cb | 2016-01-06 03:16:32 | [diff] [blame] | 2482 | if (codec_id == AV_CODEC_ID_NONE && AV_RL32(track->codec_priv.data+4) == AV_RL32("SMI ")) { |
| 2483 | fourcc = MKTAG('S','V','Q','3'); |
| 2484 | codec_id = ff_codec_get_id(ff_codec_movvideo_tags, fourcc); |
| 2485 | } |
Clément Bœsch | cd4d6cb | 2017-03-26 23:31:52 | [diff] [blame] | 2486 | if (codec_id == AV_CODEC_ID_NONE) |
Vittorio Giovara | 8fc11ab | 2015-05-11 16:23:43 | [diff] [blame] | 2487 | av_log(matroska->ctx, AV_LOG_ERROR, |
Clément Bœsch | cd4d6cb | 2017-03-26 23:31:52 | [diff] [blame] | 2488 | "mov FourCC not found %s.\n", av_fourcc2str(fourcc)); |
Mats Peterson | 7973603 | 2015-12-27 20:28:09 | [diff] [blame] | 2489 | if (track->codec_priv.size >= 86) { |
| 2490 | bit_depth = AV_RB16(track->codec_priv.data + 82); |
| 2491 | ffio_init_context(&b, track->codec_priv.data, |
| 2492 | track->codec_priv.size, |
| 2493 | 0, NULL, NULL, NULL, NULL); |
Mats Peterson | 6aac43f | 2016-02-24 17:14:05 | [diff] [blame] | 2494 | if (ff_get_qtpalette(codec_id, &b, track->palette)) { |
Mats Peterson | 7973603 | 2015-12-27 20:28:09 | [diff] [blame] | 2495 | bit_depth &= 0x1F; |
Mats Peterson | 6aac43f | 2016-02-24 17:14:05 | [diff] [blame] | 2496 | track->has_palette = 1; |
Mats Peterson | 7973603 | 2015-12-27 20:28:09 | [diff] [blame] | 2497 | } |
| 2498 | } |
Anton Khirnov | 36ef536 | 2012-08-05 09:11:04 | [diff] [blame] | 2499 | } else if (codec_id == AV_CODEC_ID_PCM_S16BE) { |
Aurelien Jacobs | eb9cf50 | 2008-08-20 00:49:45 | [diff] [blame] | 2500 | switch (track->audio.bitdepth) { |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 2501 | case 8: |
| 2502 | codec_id = AV_CODEC_ID_PCM_U8; |
| 2503 | break; |
| 2504 | case 24: |
| 2505 | codec_id = AV_CODEC_ID_PCM_S24BE; |
| 2506 | break; |
| 2507 | case 32: |
| 2508 | codec_id = AV_CODEC_ID_PCM_S32BE; |
| 2509 | break; |
Aurelien Jacobs | eb9cf50 | 2008-08-20 00:49:45 | [diff] [blame] | 2510 | } |
Anton Khirnov | 36ef536 | 2012-08-05 09:11:04 | [diff] [blame] | 2511 | } else if (codec_id == AV_CODEC_ID_PCM_S16LE) { |
Aurelien Jacobs | eb9cf50 | 2008-08-20 00:49:45 | [diff] [blame] | 2512 | switch (track->audio.bitdepth) { |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 2513 | case 8: |
| 2514 | codec_id = AV_CODEC_ID_PCM_U8; |
| 2515 | break; |
| 2516 | case 24: |
| 2517 | codec_id = AV_CODEC_ID_PCM_S24LE; |
| 2518 | break; |
| 2519 | case 32: |
| 2520 | codec_id = AV_CODEC_ID_PCM_S32LE; |
| 2521 | break; |
Aurelien Jacobs | eb9cf50 | 2008-08-20 00:49:45 | [diff] [blame] | 2522 | } |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 2523 | } else if (codec_id == AV_CODEC_ID_PCM_F32LE && |
| 2524 | track->audio.bitdepth == 64) { |
Anton Khirnov | 36ef536 | 2012-08-05 09:11:04 | [diff] [blame] | 2525 | codec_id = AV_CODEC_ID_PCM_F64LE; |
| 2526 | } else if (codec_id == AV_CODEC_ID_AAC && !track->codec_priv.size) { |
Aurelien Jacobs | d88d806 | 2008-08-05 00:40:52 | [diff] [blame] | 2527 | int profile = matroska_aac_profile(track->codec_id); |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 2528 | int sri = matroska_aac_sri(track->audio.samplerate); |
Vittorio Giovara | 059a934 | 2015-06-29 21:48:34 | [diff] [blame] | 2529 | extradata = av_mallocz(5 + AV_INPUT_BUFFER_PADDING_SIZE); |
Gabriel Dume | f929ab0 | 2014-08-14 20:31:24 | [diff] [blame] | 2530 | if (!extradata) |
Aurelien Jacobs | 28f450a | 2008-08-05 00:40:09 | [diff] [blame] | 2531 | return AVERROR(ENOMEM); |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 2532 | extradata[0] = (profile << 3) | ((sri & 0x0E) >> 1); |
| 2533 | extradata[1] = ((sri & 0x01) << 7) | (track->audio.channels << 3); |
Aurelien Jacobs | d88d806 | 2008-08-05 00:40:52 | [diff] [blame] | 2534 | if (strstr(track->codec_id, "SBR")) { |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 2535 | sri = matroska_aac_sri(track->audio.out_samplerate); |
| 2536 | extradata[2] = 0x56; |
| 2537 | extradata[3] = 0xE5; |
| 2538 | extradata[4] = 0x80 | (sri << 3); |
Aurelien Jacobs | d88d806 | 2008-08-05 00:40:52 | [diff] [blame] | 2539 | extradata_size = 5; |
Aurelien Jacobs | 16f97ab | 2008-08-05 00:41:10 | [diff] [blame] | 2540 | } else |
Aurelien Jacobs | d88d806 | 2008-08-05 00:40:52 | [diff] [blame] | 2541 | extradata_size = 2; |
Michael Niedermayer | 29d147c | 2015-07-27 20:53:16 | [diff] [blame] | 2542 | } else if (codec_id == AV_CODEC_ID_ALAC && track->codec_priv.size && track->codec_priv.size < INT_MAX - 12 - AV_INPUT_BUFFER_PADDING_SIZE) { |
Paul B Mahol | 9644fc9 | 2013-06-07 10:06:15 | [diff] [blame] | 2543 | /* Only ALAC's magic cookie is stored in Matroska's track headers. |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 2544 | * Create the "atom size", "tag", and "tag version" fields the |
| 2545 | * decoder expects manually. */ |
Paul B Mahol | 9644fc9 | 2013-06-07 10:06:15 | [diff] [blame] | 2546 | extradata_size = 12 + track->codec_priv.size; |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 2547 | extradata = av_mallocz(extradata_size + |
Vittorio Giovara | 059a934 | 2015-06-29 21:48:34 | [diff] [blame] | 2548 | AV_INPUT_BUFFER_PADDING_SIZE); |
Gabriel Dume | f929ab0 | 2014-08-14 20:31:24 | [diff] [blame] | 2549 | if (!extradata) |
Paul B Mahol | 9644fc9 | 2013-06-07 10:06:15 | [diff] [blame] | 2550 | return AVERROR(ENOMEM); |
| 2551 | AV_WB32(extradata, extradata_size); |
| 2552 | memcpy(&extradata[4], "alac", 4); |
| 2553 | AV_WB32(&extradata[8], 0); |
| 2554 | memcpy(&extradata[12], track->codec_priv.data, |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 2555 | track->codec_priv.size); |
Anton Khirnov | 36ef536 | 2012-08-05 09:11:04 | [diff] [blame] | 2556 | } else if (codec_id == AV_CODEC_ID_TTA) { |
Andreas Rheinhardt | 668490a | 2019-12-03 17:09:05 | [diff] [blame] | 2557 | uint8_t *ptr; |
Andreas Rheinhardt | f7bf59b | 2019-12-03 17:09:03 | [diff] [blame] | 2558 | if (track->audio.channels > UINT16_MAX || |
| 2559 | track->audio.bitdepth > UINT16_MAX) { |
| 2560 | av_log(matroska->ctx, AV_LOG_WARNING, |
| 2561 | "Too large audio channel number %"PRIu64 |
| 2562 | " or bitdepth %"PRIu64". Skipping track.\n", |
| 2563 | track->audio.channels, track->audio.bitdepth); |
| 2564 | if (matroska->ctx->error_recognition & AV_EF_EXPLODE) |
| 2565 | return AVERROR_INVALIDDATA; |
| 2566 | else |
| 2567 | continue; |
| 2568 | } |
| 2569 | if (track->audio.out_samplerate < 0 || track->audio.out_samplerate > INT_MAX) |
| 2570 | return AVERROR_INVALIDDATA; |
Andreas Rheinhardt | eec26b5 | 2019-12-03 17:09:04 | [diff] [blame] | 2571 | extradata_size = 22; |
Michael Niedermayer | 29d147c | 2015-07-27 20:53:16 | [diff] [blame] | 2572 | extradata = av_mallocz(extradata_size + AV_INPUT_BUFFER_PADDING_SIZE); |
Gabriel Dume | f929ab0 | 2014-08-14 20:31:24 | [diff] [blame] | 2573 | if (!extradata) |
Aurelien Jacobs | d88d806 | 2008-08-05 00:40:52 | [diff] [blame] | 2574 | return AVERROR(ENOMEM); |
Andreas Rheinhardt | 668490a | 2019-12-03 17:09:05 | [diff] [blame] | 2575 | ptr = extradata; |
| 2576 | bytestream_put_be32(&ptr, AV_RB32("TTA1")); |
| 2577 | bytestream_put_le16(&ptr, 1); |
| 2578 | bytestream_put_le16(&ptr, track->audio.channels); |
| 2579 | bytestream_put_le16(&ptr, track->audio.bitdepth); |
| 2580 | bytestream_put_le32(&ptr, track->audio.out_samplerate); |
| 2581 | bytestream_put_le32(&ptr, av_rescale(matroska->duration * matroska->time_scale, |
| 2582 | track->audio.out_samplerate, |
| 2583 | AV_TIME_BASE * 1000)); |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 2584 | } else if (codec_id == AV_CODEC_ID_RV10 || |
| 2585 | codec_id == AV_CODEC_ID_RV20 || |
| 2586 | codec_id == AV_CODEC_ID_RV30 || |
| 2587 | codec_id == AV_CODEC_ID_RV40) { |
Aurelien Jacobs | d88d806 | 2008-08-05 00:40:52 | [diff] [blame] | 2588 | extradata_offset = 26; |
Anton Khirnov | 36ef536 | 2012-08-05 09:11:04 | [diff] [blame] | 2589 | } else if (codec_id == AV_CODEC_ID_RA_144) { |
Aurelien Jacobs | d88d806 | 2008-08-05 00:40:52 | [diff] [blame] | 2590 | track->audio.out_samplerate = 8000; |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 2591 | track->audio.channels = 1; |
Michael Niedermayer | b3d9ab1 | 2014-03-07 22:46:37 | [diff] [blame] | 2592 | } else if ((codec_id == AV_CODEC_ID_RA_288 || |
| 2593 | codec_id == AV_CODEC_ID_COOK || |
| 2594 | codec_id == AV_CODEC_ID_ATRAC3 || |
| 2595 | codec_id == AV_CODEC_ID_SIPR) |
| 2596 | && track->codec_priv.data) { |
Aurelien Jacobs | 6b10228 | 2010-03-12 23:49:06 | [diff] [blame] | 2597 | int flavor; |
Michael Niedermayer | 0b27136 | 2012-04-14 17:33:24 | [diff] [blame] | 2598 | |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 2599 | ffio_init_context(&b, track->codec_priv.data, |
| 2600 | track->codec_priv.size, |
| 2601 | 0, NULL, NULL, NULL, NULL); |
Anton Khirnov | 45a8a02 | 2011-03-15 08:14:38 | [diff] [blame] | 2602 | avio_skip(&b, 22); |
Anton Khirnov | e63a362 | 2011-02-21 15:43:01 | [diff] [blame] | 2603 | flavor = avio_rb16(&b); |
| 2604 | track->audio.coded_framesize = avio_rb32(&b); |
Anton Khirnov | 45a8a02 | 2011-03-15 08:14:38 | [diff] [blame] | 2605 | avio_skip(&b, 12); |
Anton Khirnov | e63a362 | 2011-02-21 15:43:01 | [diff] [blame] | 2606 | track->audio.sub_packet_h = avio_rb16(&b); |
| 2607 | track->audio.frame_size = avio_rb16(&b); |
| 2608 | track->audio.sub_packet_size = avio_rb16(&b); |
Andreas Rheinhardt | 8287c20 | 2019-12-02 09:41:12 | [diff] [blame] | 2609 | if (track->audio.coded_framesize <= 0 || |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 2610 | track->audio.sub_packet_h <= 0 || |
Andreas Rheinhardt | c91e369 | 2020-04-20 05:47:06 | [diff] [blame] | 2611 | track->audio.frame_size <= 0) |
Martin Storsjö | 569d18a | 2013-09-16 12:36:24 | [diff] [blame] | 2612 | return AVERROR_INVALIDDATA; |
Andreas Rheinhardt | 8287c20 | 2019-12-02 09:41:12 | [diff] [blame] | 2613 | |
Anton Khirnov | 36ef536 | 2012-08-05 09:11:04 | [diff] [blame] | 2614 | if (codec_id == AV_CODEC_ID_RA_288) { |
Andreas Rheinhardt | 4f5c6c1 | 2020-04-21 22:15:54 | [diff] [blame^] | 2615 | if ((int64_t)track->audio.sub_packet_h * track->audio.coded_framesize |
| 2616 | > (2 + (track->audio.sub_packet_h & 1)) * track->audio.frame_size) |
| 2617 | return AVERROR_INVALIDDATA; |
Anton Khirnov | 9200514 | 2014-06-18 18:42:52 | [diff] [blame] | 2618 | st->codecpar->block_align = track->audio.coded_framesize; |
Aurelien Jacobs | d88d806 | 2008-08-05 00:40:52 | [diff] [blame] | 2619 | track->codec_priv.size = 0; |
| 2620 | } else { |
Andreas Rheinhardt | 8287c20 | 2019-12-02 09:41:12 | [diff] [blame] | 2621 | if (codec_id == AV_CODEC_ID_SIPR) { |
Michael Niedermayer | 62cf5c1 | 2013-08-04 19:18:49 | [diff] [blame] | 2622 | static const int sipr_bit_rate[4] = { 6504, 8496, 5000, 16000 }; |
Andreas Rheinhardt | 8287c20 | 2019-12-02 09:41:12 | [diff] [blame] | 2623 | if (flavor > 3) |
| 2624 | return AVERROR_INVALIDDATA; |
Aurelien Jacobs | 6b10228 | 2010-03-12 23:49:06 | [diff] [blame] | 2625 | track->audio.sub_packet_size = ff_sipr_subpk_size[flavor]; |
Anton Khirnov | 9200514 | 2014-06-18 18:42:52 | [diff] [blame] | 2626 | st->codecpar->bit_rate = sipr_bit_rate[flavor]; |
Andreas Rheinhardt | c91e369 | 2020-04-20 05:47:06 | [diff] [blame] | 2627 | } else if (track->audio.sub_packet_size <= 0) |
| 2628 | return AVERROR_INVALIDDATA; |
Anton Khirnov | 9200514 | 2014-06-18 18:42:52 | [diff] [blame] | 2629 | st->codecpar->block_align = track->audio.sub_packet_size; |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 2630 | extradata_offset = 78; |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 2631 | } |
Andreas Rheinhardt | 8287c20 | 2019-12-02 09:41:12 | [diff] [blame] | 2632 | track->audio.buf = av_malloc_array(track->audio.sub_packet_h, |
| 2633 | track->audio.frame_size); |
| 2634 | if (!track->audio.buf) |
| 2635 | return AVERROR(ENOMEM); |
Anton Khirnov | 4efdadc | 2014-05-25 12:05:51 | [diff] [blame] | 2636 | } else if (codec_id == AV_CODEC_ID_FLAC && track->codec_priv.size) { |
| 2637 | ret = matroska_parse_flac(s, track, &extradata_offset); |
| 2638 | if (ret < 0) |
| 2639 | return ret; |
Andreas Rheinhardt | 048bc3f | 2019-12-14 00:36:54 | [diff] [blame] | 2640 | } else if (codec_id == AV_CODEC_ID_WAVPACK && track->codec_priv.size < 2) { |
| 2641 | av_log(matroska->ctx, AV_LOG_INFO, "Assuming WavPack version 4.10 " |
| 2642 | "in absence of valid CodecPrivate.\n"); |
| 2643 | extradata_size = 2; |
| 2644 | extradata = av_mallocz(2 + AV_INPUT_BUFFER_PADDING_SIZE); |
| 2645 | if (!extradata) |
| 2646 | return AVERROR(ENOMEM); |
| 2647 | AV_WL16(extradata, 0x410); |
Carl Eugen Hoyos | 3d5c859 | 2014-04-06 18:54:48 | [diff] [blame] | 2648 | } else if (codec_id == AV_CODEC_ID_PRORES && track->codec_priv.size == 4) { |
| 2649 | fourcc = AV_RL32(track->codec_priv.data); |
James Almer | acdea9e | 2018-01-13 19:04:21 | [diff] [blame] | 2650 | } else if (codec_id == AV_CODEC_ID_VP9 && track->codec_priv.size) { |
| 2651 | /* we don't need any value stored in CodecPrivate. |
| 2652 | make sure that it's not exported as extradata. */ |
| 2653 | track->codec_priv.size = 0; |
James Almer | 8d5604a | 2018-08-16 21:01:44 | [diff] [blame] | 2654 | } else if (codec_id == AV_CODEC_ID_AV1 && track->codec_priv.size) { |
| 2655 | /* For now, propagate only the OBUs, if any. Once libavcodec is |
| 2656 | updated to handle isobmff style extradata this can be removed. */ |
| 2657 | extradata_offset = 4; |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 2658 | } |
Aurelien Jacobs | e264440 | 2009-08-24 13:40:30 | [diff] [blame] | 2659 | track->codec_priv.size -= extradata_offset; |
Aurelien Jacobs | d88d806 | 2008-08-05 00:40:52 | [diff] [blame] | 2660 | |
Anton Khirnov | 36ef536 | 2012-08-05 09:11:04 | [diff] [blame] | 2661 | if (codec_id == AV_CODEC_ID_NONE) |
Aurelien Jacobs | d88d806 | 2008-08-05 00:40:52 | [diff] [blame] | 2662 | av_log(matroska->ctx, AV_LOG_INFO, |
Anton Khirnov | 36ef536 | 2012-08-05 09:11:04 | [diff] [blame] | 2663 | "Unknown/unsupported AVCodecID %s.\n", track->codec_id); |
Aurelien Jacobs | d88d806 | 2008-08-05 00:40:52 | [diff] [blame] | 2664 | |
Aurelien Jacobs | 3fc9d7c | 2008-09-09 11:23:48 | [diff] [blame] | 2665 | if (track->time_scale < 0.01) |
| 2666 | track->time_scale = 1.0; |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 2667 | avpriv_set_pts_info(st, 64, matroska->time_scale * track->time_scale, |
| 2668 | 1000 * 1000 * 1000); /* 64 bit pts in ns */ |
Aurelien Jacobs | d88d806 | 2008-08-05 00:40:52 | [diff] [blame] | 2669 | |
Anton Khirnov | eb3b550 | 2014-04-29 10:03:13 | [diff] [blame] | 2670 | /* convert the delay from ns to the track timebase */ |
Michael Niedermayer | b5bc436 | 2016-06-06 02:23:16 | [diff] [blame] | 2671 | track->codec_delay_in_track_tb = av_rescale_q(track->codec_delay, |
Anton Khirnov | eb3b550 | 2014-04-29 10:03:13 | [diff] [blame] | 2672 | (AVRational){ 1, 1000000000 }, |
| 2673 | st->time_base); |
| 2674 | |
Anton Khirnov | 9200514 | 2014-06-18 18:42:52 | [diff] [blame] | 2675 | st->codecpar->codec_id = codec_id; |
Alex Sukhanov | 251c96a | 2013-12-23 09:41:35 | [diff] [blame] | 2676 | |
Aurelien Jacobs | d88d806 | 2008-08-05 00:40:52 | [diff] [blame] | 2677 | if (strcmp(track->language, "und")) |
Anton Khirnov | d2d67e4 | 2011-05-22 10:46:29 | [diff] [blame] | 2678 | av_dict_set(&st->metadata, "language", track->language, 0); |
| 2679 | av_dict_set(&st->metadata, "title", track->name, 0); |
Aurelien Jacobs | d88d806 | 2008-08-05 00:40:52 | [diff] [blame] | 2680 | |
| 2681 | if (track->flag_default) |
| 2682 | st->disposition |= AV_DISPOSITION_DEFAULT; |
Aurelien Jacobs | 7a617a8 | 2010-07-02 16:38:44 | [diff] [blame] | 2683 | if (track->flag_forced) |
| 2684 | st->disposition |= AV_DISPOSITION_FORCED; |
Aurelien Jacobs | d88d806 | 2008-08-05 00:40:52 | [diff] [blame] | 2685 | |
Anton Khirnov | 9200514 | 2014-06-18 18:42:52 | [diff] [blame] | 2686 | if (!st->codecpar->extradata) { |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 2687 | if (extradata) { |
Anton Khirnov | 9200514 | 2014-06-18 18:42:52 | [diff] [blame] | 2688 | st->codecpar->extradata = extradata; |
| 2689 | st->codecpar->extradata_size = extradata_size; |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 2690 | } else if (track->codec_priv.data && track->codec_priv.size > 0) { |
Derek Buitenhuis | 6f69f7a | 2016-04-10 19:58:15 | [diff] [blame] | 2691 | if (ff_alloc_extradata(st->codecpar, track->codec_priv.size)) |
Aurelien Jacobs | 553e9f7 | 2009-10-01 21:15:36 | [diff] [blame] | 2692 | return AVERROR(ENOMEM); |
Anton Khirnov | 9200514 | 2014-06-18 18:42:52 | [diff] [blame] | 2693 | memcpy(st->codecpar->extradata, |
Aurelien Jacobs | 553e9f7 | 2009-10-01 21:15:36 | [diff] [blame] | 2694 | track->codec_priv.data + extradata_offset, |
| 2695 | track->codec_priv.size); |
| 2696 | } |
Aurelien Jacobs | ff0d5a7 | 2009-10-01 21:14:46 | [diff] [blame] | 2697 | } |
Aurelien Jacobs | d88d806 | 2008-08-05 00:40:52 | [diff] [blame] | 2698 | |
| 2699 | if (track->type == MATROSKA_TRACK_TYPE_VIDEO) { |
Aurelien Jacobs | 4c509fe | 2011-05-23 23:09:24 | [diff] [blame] | 2700 | MatroskaTrackPlane *planes = track->operation.combine_planes.elem; |
Aaron Colwell | febfb49 | 2015-12-02 23:13:18 | [diff] [blame] | 2701 | int display_width_mul = 1; |
Aaron Colwell | ec83efd | 2015-11-23 20:06:14 | [diff] [blame] | 2702 | int display_height_mul = 1; |
Aurelien Jacobs | 4c509fe | 2011-05-23 23:09:24 | [diff] [blame] | 2703 | |
Anton Khirnov | 9200514 | 2014-06-18 18:42:52 | [diff] [blame] | 2704 | st->codecpar->codec_type = AVMEDIA_TYPE_VIDEO; |
Derek Buitenhuis | 6f69f7a | 2016-04-10 19:58:15 | [diff] [blame] | 2705 | st->codecpar->codec_tag = fourcc; |
Carl Eugen Hoyos | 96fc290 | 2014-02-25 23:02:51 | [diff] [blame] | 2706 | if (bit_depth >= 0) |
Derek Buitenhuis | 6f69f7a | 2016-04-10 19:58:15 | [diff] [blame] | 2707 | st->codecpar->bits_per_coded_sample = bit_depth; |
Anton Khirnov | 9200514 | 2014-06-18 18:42:52 | [diff] [blame] | 2708 | st->codecpar->width = track->video.pixel_width; |
| 2709 | st->codecpar->height = track->video.pixel_height; |
Aaron Colwell | ec83efd | 2015-11-23 20:06:14 | [diff] [blame] | 2710 | |
Luca Barbato | 5f02266 | 2016-03-28 18:29:54 | [diff] [blame] | 2711 | if (track->video.interlaced == MATROSKA_VIDEO_INTERLACE_FLAG_INTERLACED) |
James Almer | 1273bc6 | 2016-10-13 17:22:07 | [diff] [blame] | 2712 | st->codecpar->field_order = mkv_field_order(matroska, track->video.field_order); |
James Almer | 29b6c2b | 2016-09-27 17:48:33 | [diff] [blame] | 2713 | else if (track->video.interlaced == MATROSKA_VIDEO_INTERLACE_FLAG_PROGRESSIVE) |
| 2714 | st->codecpar->field_order = AV_FIELD_PROGRESSIVE; |
Luca Barbato | 5f02266 | 2016-03-28 18:29:54 | [diff] [blame] | 2715 | |
Aaron Colwell | ec83efd | 2015-11-23 20:06:14 | [diff] [blame] | 2716 | if (track->video.stereo_mode && track->video.stereo_mode < MATROSKA_VIDEO_STEREOMODE_TYPE_NB) |
| 2717 | mkv_stereo_mode_display_mul(track->video.stereo_mode, &display_width_mul, &display_height_mul); |
| 2718 | |
James Almer | 8b59ce0 | 2016-10-16 13:13:45 | [diff] [blame] | 2719 | if (track->video.display_unit < MATROSKA_VIDEO_DISPLAYUNIT_UNKNOWN) { |
James Almer | bad8bbc | 2016-10-15 21:01:50 | [diff] [blame] | 2720 | av_reduce(&st->sample_aspect_ratio.num, |
| 2721 | &st->sample_aspect_ratio.den, |
| 2722 | st->codecpar->height * track->video.display_width * display_width_mul, |
| 2723 | st->codecpar->width * track->video.display_height * display_height_mul, |
| 2724 | 255); |
| 2725 | } |
Derek Buitenhuis | 6f69f7a | 2016-04-10 19:58:15 | [diff] [blame] | 2726 | if (st->codecpar->codec_id != AV_CODEC_ID_HEVC) |
Yusuke Nakamura | 16b6839 | 2013-10-12 09:55:46 | [diff] [blame] | 2727 | st->need_parsing = AVSTREAM_PARSE_HEADERS; |
Michael Niedermayer | 0fbeeb9 | 2013-11-01 16:57:57 | [diff] [blame] | 2728 | |
Luca Barbato | ac97d47 | 2012-04-17 23:32:07 | [diff] [blame] | 2729 | if (track->default_duration) { |
Anton Khirnov | aba232c | 2012-06-26 11:10:01 | [diff] [blame] | 2730 | av_reduce(&st->avg_frame_rate.num, &st->avg_frame_rate.den, |
Luca Barbato | ac97d47 | 2012-04-17 23:32:07 | [diff] [blame] | 2731 | 1000000000, track->default_duration, 30000); |
Anton Khirnov | aba232c | 2012-06-26 11:10:01 | [diff] [blame] | 2732 | #if FF_API_R_FRAME_RATE |
Michael Niedermayer | fc3cdb0 | 2015-02-01 18:34:52 | [diff] [blame] | 2733 | if ( st->avg_frame_rate.num < st->avg_frame_rate.den * 1000LL |
| 2734 | && st->avg_frame_rate.num > st->avg_frame_rate.den * 5LL) |
Michael Niedermayer | 6853e40 | 2013-10-05 22:07:28 | [diff] [blame] | 2735 | st->r_frame_rate = st->avg_frame_rate; |
Anton Khirnov | aba232c | 2012-06-26 11:10:01 | [diff] [blame] | 2736 | #endif |
Luca Barbato | ac97d47 | 2012-04-17 23:32:07 | [diff] [blame] | 2737 | } |
Kirill Gavrilov | e6ec921 | 2011-05-21 15:14:14 | [diff] [blame] | 2738 | |
Aurelien Jacobs | 4c509fe | 2011-05-23 23:09:24 | [diff] [blame] | 2739 | /* export stereo mode flag as metadata tag */ |
Michael Niedermayer | 37520a9 | 2014-08-28 23:26:52 | [diff] [blame] | 2740 | if (track->video.stereo_mode && track->video.stereo_mode < MATROSKA_VIDEO_STEREOMODE_TYPE_NB) |
Clément Bœsch | ca81e3b | 2012-09-16 00:58:40 | [diff] [blame] | 2741 | av_dict_set(&st->metadata, "stereo_mode", ff_matroska_video_stereo_mode[track->video.stereo_mode], 0); |
Kirill Gavrilov | e6ec921 | 2011-05-21 15:14:14 | [diff] [blame] | 2742 | |
Vignesh Venkatasubramanian | ce6a8e5 | 2013-02-04 23:17:52 | [diff] [blame] | 2743 | /* export alpha mode flag as metadata tag */ |
| 2744 | if (track->video.alpha_mode) |
| 2745 | av_dict_set(&st->metadata, "alpha_mode", "1", 0); |
| 2746 | |
Aurelien Jacobs | 4c509fe | 2011-05-23 23:09:24 | [diff] [blame] | 2747 | /* if we have virtual track, mark the real tracks */ |
| 2748 | for (j=0; j < track->operation.combine_planes.nb_elem; j++) { |
| 2749 | char buf[32]; |
Aurelien Jacobs | b44bbf9 | 2011-05-24 21:26:24 | [diff] [blame] | 2750 | if (planes[j].type >= MATROSKA_VIDEO_STEREO_PLANE_COUNT) |
Aurelien Jacobs | 4c509fe | 2011-05-23 23:09:24 | [diff] [blame] | 2751 | continue; |
| 2752 | snprintf(buf, sizeof(buf), "%s_%d", |
Clément Bœsch | ca81e3b | 2012-09-16 00:58:40 | [diff] [blame] | 2753 | ff_matroska_video_stereo_plane[planes[j].type], i); |
Aurelien Jacobs | 4c509fe | 2011-05-23 23:09:24 | [diff] [blame] | 2754 | for (k=0; k < matroska->tracks.nb_elem; k++) |
Michael Niedermayer | 5d309d3 | 2015-05-04 13:47:54 | [diff] [blame] | 2755 | if (planes[j].uid == tracks[k].uid && tracks[k].stream) { |
| 2756 | av_dict_set(&tracks[k].stream->metadata, |
Aurelien Jacobs | e6ba3d4 | 2011-06-13 17:02:50 | [diff] [blame] | 2757 | "stereo_mode", buf, 0); |
Kirill Gavrilov | e6ec921 | 2011-05-21 15:14:14 | [diff] [blame] | 2758 | break; |
| 2759 | } |
Kirill Gavrilov | e6ec921 | 2011-05-21 15:14:14 | [diff] [blame] | 2760 | } |
Vittorio Giovara | d4ae8ac | 2014-08-12 21:28:49 | [diff] [blame] | 2761 | // add stream level stereo3d side data if it is a supported format |
| 2762 | if (track->video.stereo_mode < MATROSKA_VIDEO_STEREOMODE_TYPE_NB && |
| 2763 | track->video.stereo_mode != 10 && track->video.stereo_mode != 12) { |
| 2764 | int ret = ff_mkv_stereo3d_conv(st, track->video.stereo_mode); |
| 2765 | if (ret < 0) |
| 2766 | return ret; |
| 2767 | } |
Neil Birkbeck | bbda13a | 2016-03-01 00:41:04 | [diff] [blame] | 2768 | |
James Almer | edb4f5d | 2016-12-05 02:27:54 | [diff] [blame] | 2769 | ret = mkv_parse_video_color(st, track); |
| 2770 | if (ret < 0) |
| 2771 | return ret; |
James Almer | 445204c | 2016-12-06 17:48:45 | [diff] [blame] | 2772 | ret = mkv_parse_video_projection(st, track); |
| 2773 | if (ret < 0) |
| 2774 | return ret; |
Aurelien Jacobs | d88d806 | 2008-08-05 00:40:52 | [diff] [blame] | 2775 | } else if (track->type == MATROSKA_TRACK_TYPE_AUDIO) { |
Anton Khirnov | 9200514 | 2014-06-18 18:42:52 | [diff] [blame] | 2776 | st->codecpar->codec_type = AVMEDIA_TYPE_AUDIO; |
Derek Buitenhuis | 6f69f7a | 2016-04-10 19:58:15 | [diff] [blame] | 2777 | st->codecpar->codec_tag = fourcc; |
Anton Khirnov | 9200514 | 2014-06-18 18:42:52 | [diff] [blame] | 2778 | st->codecpar->sample_rate = track->audio.out_samplerate; |
| 2779 | st->codecpar->channels = track->audio.channels; |
Derek Buitenhuis | 6f69f7a | 2016-04-10 19:58:15 | [diff] [blame] | 2780 | if (!st->codecpar->bits_per_coded_sample) |
| 2781 | st->codecpar->bits_per_coded_sample = track->audio.bitdepth; |
James Almer | 63b5d04 | 2018-01-30 01:59:52 | [diff] [blame] | 2782 | if (st->codecpar->codec_id == AV_CODEC_ID_MP3 || |
| 2783 | st->codecpar->codec_id == AV_CODEC_ID_MLP || |
| 2784 | st->codecpar->codec_id == AV_CODEC_ID_TRUEHD) |
Rodger Combs | b4b2717 | 2015-08-16 08:06:04 | [diff] [blame] | 2785 | st->need_parsing = AVSTREAM_PARSE_FULL; |
Derek Buitenhuis | 6f69f7a | 2016-04-10 19:58:15 | [diff] [blame] | 2786 | else if (st->codecpar->codec_id != AV_CODEC_ID_AAC) |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 2787 | st->need_parsing = AVSTREAM_PARSE_HEADERS; |
Vignesh Venkatasubramanian | d6f86d7 | 2013-10-14 17:42:08 | [diff] [blame] | 2788 | if (track->codec_delay > 0) { |
Derek Buitenhuis | 6f69f7a | 2016-04-10 19:58:15 | [diff] [blame] | 2789 | st->codecpar->initial_padding = av_rescale_q(track->codec_delay, |
Michael Niedermayer | b5bc436 | 2016-06-06 02:23:16 | [diff] [blame] | 2790 | (AVRational){1, 1000000000}, |
James Almer | 49b0246 | 2016-06-06 04:14:11 | [diff] [blame] | 2791 | (AVRational){1, st->codecpar->codec_id == AV_CODEC_ID_OPUS ? |
| 2792 | 48000 : st->codecpar->sample_rate}); |
Vignesh Venkatasubramanian | d6f86d7 | 2013-10-14 17:42:08 | [diff] [blame] | 2793 | } |
| 2794 | if (track->seek_preroll > 0) { |
Derek Buitenhuis | 6f69f7a | 2016-04-10 19:58:15 | [diff] [blame] | 2795 | st->codecpar->seek_preroll = av_rescale_q(track->seek_preroll, |
| 2796 | (AVRational){1, 1000000000}, |
| 2797 | (AVRational){1, st->codecpar->sample_rate}); |
Vignesh Venkatasubramanian | d6f86d7 | 2013-10-14 17:42:08 | [diff] [blame] | 2798 | } |
Matthew Heaney | 818ebe9 | 2013-08-08 22:40:03 | [diff] [blame] | 2799 | } else if (codec_id == AV_CODEC_ID_WEBVTT) { |
Derek Buitenhuis | 6f69f7a | 2016-04-10 19:58:15 | [diff] [blame] | 2800 | st->codecpar->codec_type = AVMEDIA_TYPE_SUBTITLE; |
Matthew Heaney | 818ebe9 | 2013-08-08 22:40:03 | [diff] [blame] | 2801 | |
| 2802 | if (!strcmp(track->codec_id, "D_WEBVTT/CAPTIONS")) { |
| 2803 | st->disposition |= AV_DISPOSITION_CAPTIONS; |
| 2804 | } else if (!strcmp(track->codec_id, "D_WEBVTT/DESCRIPTIONS")) { |
| 2805 | st->disposition |= AV_DISPOSITION_DESCRIPTIONS; |
| 2806 | } else if (!strcmp(track->codec_id, "D_WEBVTT/METADATA")) { |
| 2807 | st->disposition |= AV_DISPOSITION_METADATA; |
| 2808 | } |
Aurelien Jacobs | d88d806 | 2008-08-05 00:40:52 | [diff] [blame] | 2809 | } else if (track->type == MATROSKA_TRACK_TYPE_SUBTITLE) { |
Anton Khirnov | 9200514 | 2014-06-18 18:42:52 | [diff] [blame] | 2810 | st->codecpar->codec_type = AVMEDIA_TYPE_SUBTITLE; |
Aurelien Jacobs | d88d806 | 2008-08-05 00:40:52 | [diff] [blame] | 2811 | } |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 2812 | } |
| 2813 | |
Anton Khirnov | 6df478b | 2014-05-25 07:07:32 | [diff] [blame] | 2814 | return 0; |
| 2815 | } |
| 2816 | |
| 2817 | static int matroska_read_header(AVFormatContext *s) |
| 2818 | { |
| 2819 | MatroskaDemuxContext *matroska = s->priv_data; |
| 2820 | EbmlList *attachments_list = &matroska->attachments; |
| 2821 | EbmlList *chapters_list = &matroska->chapters; |
| 2822 | MatroskaAttachment *attachments; |
| 2823 | MatroskaChapter *chapters; |
| 2824 | uint64_t max_start = 0; |
| 2825 | int64_t pos; |
| 2826 | Ebml ebml = { 0 }; |
| 2827 | int i, j, res; |
| 2828 | |
| 2829 | matroska->ctx = s; |
wm4 | cac2295 | 2015-02-09 19:39:00 | [diff] [blame] | 2830 | matroska->cues_parsing_deferred = 1; |
Anton Khirnov | 6df478b | 2014-05-25 07:07:32 | [diff] [blame] | 2831 | |
| 2832 | /* First read the EBML header. */ |
Thomas Guillem | b8d7f31 | 2015-04-10 17:04:51 | [diff] [blame] | 2833 | if (ebml_parse(matroska, ebml_syntax, &ebml) || !ebml.doctype) { |
| 2834 | av_log(matroska->ctx, AV_LOG_ERROR, "EBML header parsing failed\n"); |
| 2835 | ebml_free(ebml_syntax, &ebml); |
| 2836 | return AVERROR_INVALIDDATA; |
| 2837 | } |
| 2838 | if (ebml.version > EBML_VERSION || |
Anton Khirnov | 6df478b | 2014-05-25 07:07:32 | [diff] [blame] | 2839 | ebml.max_size > sizeof(uint64_t) || |
| 2840 | ebml.id_length > sizeof(uint32_t) || |
Michael Niedermayer | 0cab093 | 2015-04-19 14:45:24 | [diff] [blame] | 2841 | ebml.doctype_version > 3) { |
Diego Biurrun | 67deba8 | 2015-12-16 17:01:34 | [diff] [blame] | 2842 | avpriv_report_missing_feature(matroska->ctx, |
| 2843 | "EBML version %"PRIu64", doctype %s, doc version %"PRIu64, |
| 2844 | ebml.version, ebml.doctype, ebml.doctype_version); |
Anton Khirnov | 6df478b | 2014-05-25 07:07:32 | [diff] [blame] | 2845 | ebml_free(ebml_syntax, &ebml); |
| 2846 | return AVERROR_PATCHWELCOME; |
Michael Niedermayer | 69de229 | 2014-05-28 10:41:35 | [diff] [blame] | 2847 | } else if (ebml.doctype_version == 3) { |
| 2848 | av_log(matroska->ctx, AV_LOG_WARNING, |
| 2849 | "EBML header using unsupported features\n" |
| 2850 | "(EBML version %"PRIu64", doctype %s, doc version %"PRIu64")\n", |
| 2851 | ebml.version, ebml.doctype, ebml.doctype_version); |
Anton Khirnov | 6df478b | 2014-05-25 07:07:32 | [diff] [blame] | 2852 | } |
| 2853 | for (i = 0; i < FF_ARRAY_ELEMS(matroska_doctypes); i++) |
| 2854 | if (!strcmp(ebml.doctype, matroska_doctypes[i])) |
| 2855 | break; |
| 2856 | if (i >= FF_ARRAY_ELEMS(matroska_doctypes)) { |
| 2857 | av_log(s, AV_LOG_WARNING, "Unknown EBML doctype '%s'\n", ebml.doctype); |
| 2858 | if (matroska->ctx->error_recognition & AV_EF_EXPLODE) { |
| 2859 | ebml_free(ebml_syntax, &ebml); |
| 2860 | return AVERROR_INVALIDDATA; |
| 2861 | } |
| 2862 | } |
| 2863 | ebml_free(ebml_syntax, &ebml); |
| 2864 | |
| 2865 | /* The next thing is a segment. */ |
| 2866 | pos = avio_tell(matroska->ctx->pb); |
| 2867 | res = ebml_parse(matroska, matroska_segments, matroska); |
Andreas Rheinhardt | 60f75c9 | 2019-05-16 22:30:13 | [diff] [blame] | 2868 | // Try resyncing until we find an EBML_STOP type element. |
Anton Khirnov | 6df478b | 2014-05-25 07:07:32 | [diff] [blame] | 2869 | while (res != 1) { |
| 2870 | res = matroska_resync(matroska, pos); |
| 2871 | if (res < 0) |
Michael Niedermayer | f573acc | 2016-06-24 16:26:46 | [diff] [blame] | 2872 | goto fail; |
Anton Khirnov | 6df478b | 2014-05-25 07:07:32 | [diff] [blame] | 2873 | pos = avio_tell(matroska->ctx->pb); |
| 2874 | res = ebml_parse(matroska, matroska_segment, matroska); |
| 2875 | } |
Andreas Rheinhardt | 43c3ceb | 2019-05-16 22:29:51 | [diff] [blame] | 2876 | /* Set data_offset as it might be needed later by seek_frame_generic. */ |
| 2877 | if (matroska->current_id == MATROSKA_ID_CLUSTER) |
| 2878 | s->internal->data_offset = avio_tell(matroska->ctx->pb) - 4; |
Anton Khirnov | 6df478b | 2014-05-25 07:07:32 | [diff] [blame] | 2879 | matroska_execute_seekhead(matroska); |
| 2880 | |
| 2881 | if (!matroska->time_scale) |
| 2882 | matroska->time_scale = 1000000; |
| 2883 | if (matroska->duration) |
| 2884 | matroska->ctx->duration = matroska->duration * matroska->time_scale * |
| 2885 | 1000 / AV_TIME_BASE; |
| 2886 | av_dict_set(&s->metadata, "title", matroska->title, 0); |
Michael Niedermayer | 69de229 | 2014-05-28 10:41:35 | [diff] [blame] | 2887 | av_dict_set(&s->metadata, "encoder", matroska->muxingapp, 0); |
| 2888 | |
| 2889 | if (matroska->date_utc.size == 8) |
| 2890 | matroska_metadata_creation_time(&s->metadata, AV_RB64(matroska->date_utc.data)); |
Anton Khirnov | 6df478b | 2014-05-25 07:07:32 | [diff] [blame] | 2891 | |
| 2892 | res = matroska_parse_tracks(s); |
| 2893 | if (res < 0) |
Michael Niedermayer | f573acc | 2016-06-24 16:26:46 | [diff] [blame] | 2894 | goto fail; |
Anton Khirnov | 6df478b | 2014-05-25 07:07:32 | [diff] [blame] | 2895 | |
Diego Biurrun | f69befe | 2014-03-07 12:54:18 | [diff] [blame] | 2896 | attachments = attachments_list->elem; |
| 2897 | for (j = 0; j < attachments_list->nb_elem; j++) { |
| 2898 | if (!(attachments[j].filename && attachments[j].mime && |
| 2899 | attachments[j].bin.data && attachments[j].bin.size > 0)) { |
Aurelien Jacobs | 9c25baf | 2008-08-05 00:40:55 | [diff] [blame] | 2900 | av_log(matroska->ctx, AV_LOG_ERROR, "incomplete attachment\n"); |
| 2901 | } else { |
Anton Khirnov | 3b3bbdd | 2011-06-18 09:43:24 | [diff] [blame] | 2902 | AVStream *st = avformat_new_stream(s, NULL); |
Gabriel Dume | f929ab0 | 2014-08-14 20:31:24 | [diff] [blame] | 2903 | if (!st) |
Aurelien Jacobs | 9c25baf | 2008-08-05 00:40:55 | [diff] [blame] | 2904 | break; |
Diego Biurrun | f69befe | 2014-03-07 12:54:18 | [diff] [blame] | 2905 | av_dict_set(&st->metadata, "filename", attachments[j].filename, 0); |
| 2906 | av_dict_set(&st->metadata, "mimetype", attachments[j].mime, 0); |
Anton Khirnov | 9200514 | 2014-06-18 18:42:52 | [diff] [blame] | 2907 | st->codecpar->codec_id = AV_CODEC_ID_NONE; |
Aurelien Jacobs | 9c25baf | 2008-08-05 00:40:55 | [diff] [blame] | 2908 | |
Andreas Rheinhardt | 67e957b | 2020-04-16 01:39:05 | [diff] [blame] | 2909 | for (i = 0; mkv_image_mime_tags[i].id != AV_CODEC_ID_NONE; i++) { |
| 2910 | if (!strncmp(mkv_image_mime_tags[i].str, attachments[j].mime, |
| 2911 | strlen(mkv_image_mime_tags[i].str))) { |
| 2912 | st->codecpar->codec_id = mkv_image_mime_tags[i].id; |
Aurelien Jacobs | 9c25baf | 2008-08-05 00:40:55 | [diff] [blame] | 2913 | break; |
| 2914 | } |
| 2915 | } |
wm4 | 511585c | 2015-04-03 14:11:53 | [diff] [blame] | 2916 | |
Diego Biurrun | f69befe | 2014-03-07 12:54:18 | [diff] [blame] | 2917 | attachments[j].stream = st; |
wm4 | c4d37cd | 2015-04-03 14:11:53 | [diff] [blame] | 2918 | |
Anton Khirnov | 9200514 | 2014-06-18 18:42:52 | [diff] [blame] | 2919 | if (st->codecpar->codec_id != AV_CODEC_ID_NONE) { |
Andreas Rheinhardt | 07d4056 | 2019-05-16 22:29:52 | [diff] [blame] | 2920 | AVPacket *pkt = &st->attached_pic; |
| 2921 | |
Anton Khirnov | 9200514 | 2014-06-18 18:42:52 | [diff] [blame] | 2922 | st->disposition |= AV_DISPOSITION_ATTACHED_PIC; |
| 2923 | st->codecpar->codec_type = AVMEDIA_TYPE_VIDEO; |
wm4 | 511585c | 2015-04-03 14:11:53 | [diff] [blame] | 2924 | |
Andreas Rheinhardt | 07d4056 | 2019-05-16 22:29:52 | [diff] [blame] | 2925 | av_init_packet(pkt); |
| 2926 | pkt->buf = av_buffer_ref(attachments[j].bin.buf); |
| 2927 | if (!pkt->buf) |
| 2928 | return AVERROR(ENOMEM); |
| 2929 | pkt->data = attachments[j].bin.data; |
| 2930 | pkt->size = attachments[j].bin.size; |
| 2931 | pkt->stream_index = st->index; |
| 2932 | pkt->flags |= AV_PKT_FLAG_KEY; |
wm4 | 511585c | 2015-04-03 14:11:53 | [diff] [blame] | 2933 | } else { |
Anton Khirnov | 9200514 | 2014-06-18 18:42:52 | [diff] [blame] | 2934 | st->codecpar->codec_type = AVMEDIA_TYPE_ATTACHMENT; |
Derek Buitenhuis | 6f69f7a | 2016-04-10 19:58:15 | [diff] [blame] | 2935 | if (ff_alloc_extradata(st->codecpar, attachments[j].bin.size)) |
wm4 | 511585c | 2015-04-03 14:11:53 | [diff] [blame] | 2936 | break; |
Anton Khirnov | 9200514 | 2014-06-18 18:42:52 | [diff] [blame] | 2937 | memcpy(st->codecpar->extradata, attachments[j].bin.data, |
wm4 | c4d37cd | 2015-04-03 14:11:53 | [diff] [blame] | 2938 | attachments[j].bin.size); |
wm4 | 511585c | 2015-04-03 14:11:53 | [diff] [blame] | 2939 | |
Andreas Rheinhardt | 67e957b | 2020-04-16 01:39:05 | [diff] [blame] | 2940 | for (i = 0; mkv_mime_tags[i].id != AV_CODEC_ID_NONE; i++) { |
| 2941 | if (!strncmp(mkv_mime_tags[i].str, attachments[j].mime, |
| 2942 | strlen(mkv_mime_tags[i].str))) { |
| 2943 | st->codecpar->codec_id = mkv_mime_tags[i].id; |
wm4 | 511585c | 2015-04-03 14:11:53 | [diff] [blame] | 2944 | break; |
| 2945 | } |
| 2946 | } |
wm4 | 511585c | 2015-04-03 14:11:53 | [diff] [blame] | 2947 | } |
Aurelien Jacobs | 9c25baf | 2008-08-05 00:40:55 | [diff] [blame] | 2948 | } |
| 2949 | } |
| 2950 | |
| 2951 | chapters = chapters_list->elem; |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 2952 | for (i = 0; i < chapters_list->nb_elem; i++) |
| 2953 | if (chapters[i].start != AV_NOPTS_VALUE && chapters[i].uid && |
| 2954 | (max_start == 0 || chapters[i].start > max_start)) { |
Aurelien Jacobs | 6cb6e15 | 2009-02-15 15:25:14 | [diff] [blame] | 2955 | chapters[i].chapter = |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 2956 | avpriv_new_chapter(s, chapters[i].uid, |
| 2957 | (AVRational) { 1, 1000000000 }, |
| 2958 | chapters[i].start, chapters[i].end, |
| 2959 | chapters[i].title); |
Aurelien Jacobs | e0e4be5 | 2009-01-15 00:42:57 | [diff] [blame] | 2960 | max_start = chapters[i].start; |
| 2961 | } |
Aurelien Jacobs | 9c25baf | 2008-08-05 00:40:55 | [diff] [blame] | 2962 | |
Aaron Colwell | 6c4cc0f | 2011-09-22 14:51:00 | [diff] [blame] | 2963 | matroska_add_index_entries(matroska); |
| 2964 | |
Aurelien Jacobs | 929e9de | 2009-02-15 15:53:55 | [diff] [blame] | 2965 | matroska_convert_tags(s); |
| 2966 | |
Aurelien Jacobs | ce6f28b | 2008-08-05 00:40:58 | [diff] [blame] | 2967 | return 0; |
Michael Niedermayer | f573acc | 2016-06-24 16:26:46 | [diff] [blame] | 2968 | fail: |
| 2969 | matroska_read_close(s); |
| 2970 | return res; |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 2971 | } |
| 2972 | |
Aurelien Jacobs | 737c40d | 2008-08-05 00:42:39 | [diff] [blame] | 2973 | /* |
Aurelien Jacobs | 737c40d | 2008-08-05 00:42:39 | [diff] [blame] | 2974 | * Put one packet in an application-supplied AVPacket struct. |
| 2975 | * Returns 0 on success or -1 on failure. |
| 2976 | */ |
| 2977 | static int matroska_deliver_packet(MatroskaDemuxContext *matroska, |
| 2978 | AVPacket *pkt) |
| 2979 | { |
James Almer | 78b96be | 2018-03-26 18:02:37 | [diff] [blame] | 2980 | if (matroska->queue) { |
Mats Peterson | 6aac43f | 2016-02-24 17:14:05 | [diff] [blame] | 2981 | MatroskaTrack *tracks = matroska->tracks.elem; |
| 2982 | MatroskaTrack *track; |
James Almer | 78b96be | 2018-03-26 18:02:37 | [diff] [blame] | 2983 | |
| 2984 | ff_packet_list_get(&matroska->queue, &matroska->queue_end, pkt); |
Mats Peterson | 6aac43f | 2016-02-24 17:14:05 | [diff] [blame] | 2985 | track = &tracks[pkt->stream_index]; |
| 2986 | if (track->has_palette) { |
Mats Peterson | 7973603 | 2015-12-27 20:28:09 | [diff] [blame] | 2987 | uint8_t *pal = av_packet_new_side_data(pkt, AV_PKT_DATA_PALETTE, AVPALETTE_SIZE); |
| 2988 | if (!pal) { |
| 2989 | av_log(matroska->ctx, AV_LOG_ERROR, "Cannot append palette to packet\n"); |
| 2990 | } else { |
Mats Peterson | 6aac43f | 2016-02-24 17:14:05 | [diff] [blame] | 2991 | memcpy(pal, track->palette, AVPALETTE_SIZE); |
Mats Peterson | 7973603 | 2015-12-27 20:28:09 | [diff] [blame] | 2992 | } |
Mats Peterson | 6aac43f | 2016-02-24 17:14:05 | [diff] [blame] | 2993 | track->has_palette = 0; |
Mats Peterson | 7973603 | 2015-12-27 20:28:09 | [diff] [blame] | 2994 | } |
Aurelien Jacobs | 737c40d | 2008-08-05 00:42:39 | [diff] [blame] | 2995 | return 0; |
| 2996 | } |
| 2997 | |
| 2998 | return -1; |
| 2999 | } |
| 3000 | |
| 3001 | /* |
| 3002 | * Free all packets in our internal queue. |
| 3003 | */ |
| 3004 | static void matroska_clear_queue(MatroskaDemuxContext *matroska) |
| 3005 | { |
James Almer | 78b96be | 2018-03-26 18:02:37 | [diff] [blame] | 3006 | ff_packet_list_free(&matroska->queue, &matroska->queue_end); |
Aurelien Jacobs | 737c40d | 2008-08-05 00:42:39 | [diff] [blame] | 3007 | } |
| 3008 | |
Luca Barbato | 2d0e771 | 2012-09-16 23:58:32 | [diff] [blame] | 3009 | static int matroska_parse_laces(MatroskaDemuxContext *matroska, uint8_t **buf, |
Andreas Rheinhardt | d5274f8 | 2019-12-03 17:09:10 | [diff] [blame] | 3010 | int size, int type, AVIOContext *pb, |
Andreas Rheinhardt | 9ad1a6d | 2019-12-03 17:09:06 | [diff] [blame] | 3011 | uint32_t lace_size[256], int *laces) |
Luca Barbato | 2d0e771 | 2012-09-16 23:58:32 | [diff] [blame] | 3012 | { |
Andreas Rheinhardt | f74eaa1 | 2019-12-03 17:09:08 | [diff] [blame] | 3013 | int n; |
Luca Barbato | 2d0e771 | 2012-09-16 23:58:32 | [diff] [blame] | 3014 | uint8_t *data = *buf; |
Luca Barbato | 2d0e771 | 2012-09-16 23:58:32 | [diff] [blame] | 3015 | |
| 3016 | if (!type) { |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 3017 | *laces = 1; |
Andreas Rheinhardt | 9ad1a6d | 2019-12-03 17:09:06 | [diff] [blame] | 3018 | lace_size[0] = size; |
Luca Barbato | 2d0e771 | 2012-09-16 23:58:32 | [diff] [blame] | 3019 | return 0; |
| 3020 | } |
| 3021 | |
Michael Niedermayer | 8c51ea5 | 2012-09-20 18:37:26 | [diff] [blame] | 3022 | av_assert0(size > 0); |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 3023 | *laces = *data + 1; |
| 3024 | data += 1; |
| 3025 | size -= 1; |
Luca Barbato | 2d0e771 | 2012-09-16 23:58:32 | [diff] [blame] | 3026 | |
| 3027 | switch (type) { |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 3028 | case 0x1: /* Xiph lacing */ |
| 3029 | { |
Luca Barbato | 2d0e771 | 2012-09-16 23:58:32 | [diff] [blame] | 3030 | uint8_t temp; |
| 3031 | uint32_t total = 0; |
Andreas Rheinhardt | a69f92a | 2019-12-03 17:09:07 | [diff] [blame] | 3032 | for (n = 0; n < *laces - 1; n++) { |
Andreas Rheinhardt | 1215b3a | 2019-05-16 22:29:48 | [diff] [blame] | 3033 | lace_size[n] = 0; |
| 3034 | |
Luca Barbato | 2d0e771 | 2012-09-16 23:58:32 | [diff] [blame] | 3035 | while (1) { |
Michael Niedermayer | 115c3bc | 2013-04-04 13:17:57 | [diff] [blame] | 3036 | if (size <= total) { |
Andreas Rheinhardt | a69f92a | 2019-12-03 17:09:07 | [diff] [blame] | 3037 | return AVERROR_INVALIDDATA; |
Luca Barbato | 2d0e771 | 2012-09-16 23:58:32 | [diff] [blame] | 3038 | } |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 3039 | temp = *data; |
Michael Niedermayer | b3d9ab1 | 2014-03-07 22:46:37 | [diff] [blame] | 3040 | total += temp; |
Luca Barbato | 2d0e771 | 2012-09-16 23:58:32 | [diff] [blame] | 3041 | lace_size[n] += temp; |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 3042 | data += 1; |
| 3043 | size -= 1; |
Luca Barbato | 2d0e771 | 2012-09-16 23:58:32 | [diff] [blame] | 3044 | if (temp != 0xff) |
| 3045 | break; |
| 3046 | } |
Luca Barbato | 2d0e771 | 2012-09-16 23:58:32 | [diff] [blame] | 3047 | } |
| 3048 | if (size <= total) { |
Andreas Rheinhardt | a69f92a | 2019-12-03 17:09:07 | [diff] [blame] | 3049 | return AVERROR_INVALIDDATA; |
Luca Barbato | 2d0e771 | 2012-09-16 23:58:32 | [diff] [blame] | 3050 | } |
| 3051 | |
| 3052 | lace_size[n] = size - total; |
| 3053 | break; |
| 3054 | } |
| 3055 | |
| 3056 | case 0x2: /* fixed-size lacing */ |
Anton Khirnov | 87b017a | 2012-09-20 18:04:56 | [diff] [blame] | 3057 | if (size % (*laces)) { |
Andreas Rheinhardt | a69f92a | 2019-12-03 17:09:07 | [diff] [blame] | 3058 | return AVERROR_INVALIDDATA; |
Luca Barbato | 2d0e771 | 2012-09-16 23:58:32 | [diff] [blame] | 3059 | } |
| 3060 | for (n = 0; n < *laces; n++) |
| 3061 | lace_size[n] = size / *laces; |
| 3062 | break; |
| 3063 | |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 3064 | case 0x3: /* EBML lacing */ |
| 3065 | { |
Luca Barbato | 2d0e771 | 2012-09-16 23:58:32 | [diff] [blame] | 3066 | uint64_t num; |
Luca Barbato | 8a96df7 | 2013-03-28 10:52:52 | [diff] [blame] | 3067 | uint64_t total; |
Andreas Rheinhardt | d5274f8 | 2019-12-03 17:09:10 | [diff] [blame] | 3068 | int offset; |
| 3069 | |
| 3070 | avio_skip(pb, 4); |
| 3071 | |
| 3072 | n = ebml_read_num(matroska, pb, 8, &num, 1); |
Andreas Rheinhardt | dbe3be6 | 2019-12-03 17:09:09 | [diff] [blame] | 3073 | if (n < 0) |
| 3074 | return n; |
| 3075 | if (num > INT_MAX) |
| 3076 | return AVERROR_INVALIDDATA; |
Andreas Rheinhardt | d5274f8 | 2019-12-03 17:09:10 | [diff] [blame] | 3077 | |
Luca Barbato | 2d0e771 | 2012-09-16 23:58:32 | [diff] [blame] | 3078 | total = lace_size[0] = num; |
Andreas Rheinhardt | d5274f8 | 2019-12-03 17:09:10 | [diff] [blame] | 3079 | offset = n; |
Andreas Rheinhardt | a69f92a | 2019-12-03 17:09:07 | [diff] [blame] | 3080 | for (n = 1; n < *laces - 1; n++) { |
Luca Barbato | 2d0e771 | 2012-09-16 23:58:32 | [diff] [blame] | 3081 | int64_t snum; |
| 3082 | int r; |
Andreas Rheinhardt | d5274f8 | 2019-12-03 17:09:10 | [diff] [blame] | 3083 | r = matroska_ebmlnum_sint(matroska, pb, &snum); |
Andreas Rheinhardt | dbe3be6 | 2019-12-03 17:09:09 | [diff] [blame] | 3084 | if (r < 0) |
| 3085 | return r; |
| 3086 | if (lace_size[n - 1] + snum > (uint64_t)INT_MAX) |
| 3087 | return AVERROR_INVALIDDATA; |
Andreas Rheinhardt | d5274f8 | 2019-12-03 17:09:10 | [diff] [blame] | 3088 | |
Luca Barbato | 2d0e771 | 2012-09-16 23:58:32 | [diff] [blame] | 3089 | lace_size[n] = lace_size[n - 1] + snum; |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 3090 | total += lace_size[n]; |
Andreas Rheinhardt | d5274f8 | 2019-12-03 17:09:10 | [diff] [blame] | 3091 | offset += r; |
Luca Barbato | 2d0e771 | 2012-09-16 23:58:32 | [diff] [blame] | 3092 | } |
Andreas Rheinhardt | d5274f8 | 2019-12-03 17:09:10 | [diff] [blame] | 3093 | data += offset; |
| 3094 | size -= offset; |
Luca Barbato | 2d0e771 | 2012-09-16 23:58:32 | [diff] [blame] | 3095 | if (size <= total) { |
Andreas Rheinhardt | a69f92a | 2019-12-03 17:09:07 | [diff] [blame] | 3096 | return AVERROR_INVALIDDATA; |
Luca Barbato | 2d0e771 | 2012-09-16 23:58:32 | [diff] [blame] | 3097 | } |
| 3098 | lace_size[*laces - 1] = size - total; |
| 3099 | break; |
| 3100 | } |
| 3101 | } |
| 3102 | |
| 3103 | *buf = data; |
Luca Barbato | 2d0e771 | 2012-09-16 23:58:32 | [diff] [blame] | 3104 | |
Andreas Rheinhardt | a69f92a | 2019-12-03 17:09:07 | [diff] [blame] | 3105 | return 0; |
Luca Barbato | 2d0e771 | 2012-09-16 23:58:32 | [diff] [blame] | 3106 | } |
| 3107 | |
Luca Barbato | c831ebf | 2012-09-16 23:28:13 | [diff] [blame] | 3108 | static int matroska_parse_rm_audio(MatroskaDemuxContext *matroska, |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 3109 | MatroskaTrack *track, AVStream *st, |
| 3110 | uint8_t *data, int size, uint64_t timecode, |
Luca Barbato | c831ebf | 2012-09-16 23:28:13 | [diff] [blame] | 3111 | int64_t pos) |
| 3112 | { |
Anton Khirnov | 9200514 | 2014-06-18 18:42:52 | [diff] [blame] | 3113 | int a = st->codecpar->block_align; |
Luca Barbato | c831ebf | 2012-09-16 23:28:13 | [diff] [blame] | 3114 | int sps = track->audio.sub_packet_size; |
| 3115 | int cfs = track->audio.coded_framesize; |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 3116 | int h = track->audio.sub_packet_h; |
| 3117 | int y = track->audio.sub_packet_cnt; |
| 3118 | int w = track->audio.frame_size; |
Luca Barbato | c831ebf | 2012-09-16 23:28:13 | [diff] [blame] | 3119 | int x; |
| 3120 | |
| 3121 | if (!track->audio.pkt_cnt) { |
| 3122 | if (track->audio.sub_packet_cnt == 0) |
| 3123 | track->audio.buf_timecode = timecode; |
Anton Khirnov | 9200514 | 2014-06-18 18:42:52 | [diff] [blame] | 3124 | if (st->codecpar->codec_id == AV_CODEC_ID_RA_288) { |
Luca Barbato | c831ebf | 2012-09-16 23:28:13 | [diff] [blame] | 3125 | if (size < cfs * h / 2) { |
| 3126 | av_log(matroska->ctx, AV_LOG_ERROR, |
| 3127 | "Corrupt int4 RM-style audio packet size\n"); |
| 3128 | return AVERROR_INVALIDDATA; |
| 3129 | } |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 3130 | for (x = 0; x < h / 2; x++) |
| 3131 | memcpy(track->audio.buf + x * 2 * w + y * cfs, |
| 3132 | data + x * cfs, cfs); |
Anton Khirnov | 9200514 | 2014-06-18 18:42:52 | [diff] [blame] | 3133 | } else if (st->codecpar->codec_id == AV_CODEC_ID_SIPR) { |
Luca Barbato | c831ebf | 2012-09-16 23:28:13 | [diff] [blame] | 3134 | if (size < w) { |
| 3135 | av_log(matroska->ctx, AV_LOG_ERROR, |
| 3136 | "Corrupt sipr RM-style audio packet size\n"); |
| 3137 | return AVERROR_INVALIDDATA; |
| 3138 | } |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 3139 | memcpy(track->audio.buf + y * w, data, w); |
Luca Barbato | c831ebf | 2012-09-16 23:28:13 | [diff] [blame] | 3140 | } else { |
Michael Niedermayer | a1ed1c2 | 2014-01-10 22:10:47 | [diff] [blame] | 3141 | if (size < sps * w / sps || h<=0 || w%sps) { |
Luca Barbato | c831ebf | 2012-09-16 23:28:13 | [diff] [blame] | 3142 | av_log(matroska->ctx, AV_LOG_ERROR, |
| 3143 | "Corrupt generic RM-style audio packet size\n"); |
| 3144 | return AVERROR_INVALIDDATA; |
| 3145 | } |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 3146 | for (x = 0; x < w / sps; x++) |
| 3147 | memcpy(track->audio.buf + |
| 3148 | sps * (h * x + ((h + 1) / 2) * (y & 1) + (y >> 1)), |
| 3149 | data + x * sps, sps); |
Luca Barbato | c831ebf | 2012-09-16 23:28:13 | [diff] [blame] | 3150 | } |
| 3151 | |
| 3152 | if (++track->audio.sub_packet_cnt >= h) { |
Anton Khirnov | 9200514 | 2014-06-18 18:42:52 | [diff] [blame] | 3153 | if (st->codecpar->codec_id == AV_CODEC_ID_SIPR) |
Luca Barbato | c831ebf | 2012-09-16 23:28:13 | [diff] [blame] | 3154 | ff_rm_reorder_sipr_data(track->audio.buf, h, w); |
| 3155 | track->audio.sub_packet_cnt = 0; |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 3156 | track->audio.pkt_cnt = h * w / a; |
Luca Barbato | c831ebf | 2012-09-16 23:28:13 | [diff] [blame] | 3157 | } |
| 3158 | } |
| 3159 | |
| 3160 | while (track->audio.pkt_cnt) { |
Vittorio Giovara | e0caa1e | 2014-10-23 23:05:53 | [diff] [blame] | 3161 | int ret; |
James Almer | 4f55b94 | 2018-04-04 16:53:12 | [diff] [blame] | 3162 | AVPacket pktl, *pkt = &pktl; |
Vittorio Giovara | e0caa1e | 2014-10-23 23:05:53 | [diff] [blame] | 3163 | |
| 3164 | ret = av_new_packet(pkt, a); |
| 3165 | if (ret < 0) { |
Vittorio Giovara | e0caa1e | 2014-10-23 23:05:53 | [diff] [blame] | 3166 | return ret; |
Michael Niedermayer | 1116491 | 2012-10-20 14:55:45 | [diff] [blame] | 3167 | } |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 3168 | memcpy(pkt->data, |
| 3169 | track->audio.buf + a * (h * w / a - track->audio.pkt_cnt--), |
| 3170 | a); |
| 3171 | pkt->pts = track->audio.buf_timecode; |
Luca Barbato | c831ebf | 2012-09-16 23:28:13 | [diff] [blame] | 3172 | track->audio.buf_timecode = AV_NOPTS_VALUE; |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 3173 | pkt->pos = pos; |
| 3174 | pkt->stream_index = st->index; |
James Almer | 78b96be | 2018-03-26 18:02:37 | [diff] [blame] | 3175 | ret = ff_packet_list_put(&matroska->queue, &matroska->queue_end, pkt, 0); |
| 3176 | if (ret < 0) { |
James Almer | 4f55b94 | 2018-04-04 16:53:12 | [diff] [blame] | 3177 | av_packet_unref(pkt); |
James Almer | 78b96be | 2018-03-26 18:02:37 | [diff] [blame] | 3178 | return AVERROR(ENOMEM); |
| 3179 | } |
Luca Barbato | c831ebf | 2012-09-16 23:28:13 | [diff] [blame] | 3180 | } |
| 3181 | |
| 3182 | return 0; |
| 3183 | } |
Anton Khirnov | 9b6f47c | 2013-05-27 07:44:27 | [diff] [blame] | 3184 | |
| 3185 | /* reconstruct full wavpack blocks from mangled matroska ones */ |
| 3186 | static int matroska_parse_wavpack(MatroskaTrack *track, uint8_t *src, |
| 3187 | uint8_t **pdst, int *size) |
| 3188 | { |
| 3189 | uint8_t *dst = NULL; |
| 3190 | int dstlen = 0; |
| 3191 | int srclen = *size; |
| 3192 | uint32_t samples; |
| 3193 | uint16_t ver; |
| 3194 | int ret, offset = 0; |
| 3195 | |
Andreas Rheinhardt | 048bc3f | 2019-12-14 00:36:54 | [diff] [blame] | 3196 | if (srclen < 12) |
Anton Khirnov | 9b6f47c | 2013-05-27 07:44:27 | [diff] [blame] | 3197 | return AVERROR_INVALIDDATA; |
| 3198 | |
Andreas Rheinhardt | 048bc3f | 2019-12-14 00:36:54 | [diff] [blame] | 3199 | av_assert1(track->stream->codecpar->extradata_size >= 2); |
Anton Khirnov | 9200514 | 2014-06-18 18:42:52 | [diff] [blame] | 3200 | ver = AV_RL16(track->stream->codecpar->extradata); |
Anton Khirnov | 9b6f47c | 2013-05-27 07:44:27 | [diff] [blame] | 3201 | |
| 3202 | samples = AV_RL32(src); |
| 3203 | src += 4; |
| 3204 | srclen -= 4; |
| 3205 | |
| 3206 | while (srclen >= 8) { |
| 3207 | int multiblock; |
| 3208 | uint32_t blocksize; |
| 3209 | uint8_t *tmp; |
| 3210 | |
| 3211 | uint32_t flags = AV_RL32(src); |
| 3212 | uint32_t crc = AV_RL32(src + 4); |
| 3213 | src += 8; |
| 3214 | srclen -= 8; |
| 3215 | |
| 3216 | multiblock = (flags & 0x1800) != 0x1800; |
| 3217 | if (multiblock) { |
| 3218 | if (srclen < 4) { |
| 3219 | ret = AVERROR_INVALIDDATA; |
| 3220 | goto fail; |
| 3221 | } |
| 3222 | blocksize = AV_RL32(src); |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 3223 | src += 4; |
| 3224 | srclen -= 4; |
Anton Khirnov | 9b6f47c | 2013-05-27 07:44:27 | [diff] [blame] | 3225 | } else |
| 3226 | blocksize = srclen; |
| 3227 | |
| 3228 | if (blocksize > srclen) { |
| 3229 | ret = AVERROR_INVALIDDATA; |
| 3230 | goto fail; |
| 3231 | } |
| 3232 | |
James Almer | 9703b7d | 2018-04-04 21:12:53 | [diff] [blame] | 3233 | tmp = av_realloc(dst, dstlen + blocksize + 32 + AV_INPUT_BUFFER_PADDING_SIZE); |
Anton Khirnov | 9b6f47c | 2013-05-27 07:44:27 | [diff] [blame] | 3234 | if (!tmp) { |
| 3235 | ret = AVERROR(ENOMEM); |
| 3236 | goto fail; |
| 3237 | } |
| 3238 | dst = tmp; |
| 3239 | dstlen += blocksize + 32; |
| 3240 | |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 3241 | AV_WL32(dst + offset, MKTAG('w', 'v', 'p', 'k')); // tag |
| 3242 | AV_WL32(dst + offset + 4, blocksize + 24); // blocksize - 8 |
| 3243 | AV_WL16(dst + offset + 8, ver); // version |
| 3244 | AV_WL16(dst + offset + 10, 0); // track/index_no |
| 3245 | AV_WL32(dst + offset + 12, 0); // total samples |
| 3246 | AV_WL32(dst + offset + 16, 0); // block index |
| 3247 | AV_WL32(dst + offset + 20, samples); // number of samples |
| 3248 | AV_WL32(dst + offset + 24, flags); // flags |
| 3249 | AV_WL32(dst + offset + 28, crc); // crc |
| 3250 | memcpy(dst + offset + 32, src, blocksize); // block data |
Anton Khirnov | 9b6f47c | 2013-05-27 07:44:27 | [diff] [blame] | 3251 | |
| 3252 | src += blocksize; |
| 3253 | srclen -= blocksize; |
| 3254 | offset += blocksize + 32; |
| 3255 | } |
| 3256 | |
James Almer | 9703b7d | 2018-04-04 21:12:53 | [diff] [blame] | 3257 | memset(dst + dstlen, 0, AV_INPUT_BUFFER_PADDING_SIZE); |
| 3258 | |
Anton Khirnov | 9b6f47c | 2013-05-27 07:44:27 | [diff] [blame] | 3259 | *pdst = dst; |
| 3260 | *size = dstlen; |
| 3261 | |
| 3262 | return 0; |
| 3263 | |
| 3264 | fail: |
| 3265 | av_freep(&dst); |
| 3266 | return ret; |
| 3267 | } |
| 3268 | |
James Almer | b8e75a2 | 2018-04-04 19:07:36 | [diff] [blame] | 3269 | static int matroska_parse_prores(MatroskaTrack *track, uint8_t *src, |
| 3270 | uint8_t **pdst, int *size) |
| 3271 | { |
Andreas Rheinhardt | af50f0a | 2019-12-06 23:16:19 | [diff] [blame] | 3272 | uint8_t *dst; |
| 3273 | int dstlen = *size + 8; |
Andreas Rheinhardt | 581419e | 2019-09-28 17:54:25 | [diff] [blame] | 3274 | |
| 3275 | dst = av_malloc(dstlen + AV_INPUT_BUFFER_PADDING_SIZE); |
James Almer | b8e75a2 | 2018-04-04 19:07:36 | [diff] [blame] | 3276 | if (!dst) |
| 3277 | return AVERROR(ENOMEM); |
| 3278 | |
| 3279 | AV_WB32(dst, dstlen); |
| 3280 | AV_WB32(dst + 4, MKBETAG('i', 'c', 'p', 'f')); |
Andreas Rheinhardt | 581419e | 2019-09-28 17:54:25 | [diff] [blame] | 3281 | memcpy(dst + 8, src, dstlen - 8); |
| 3282 | memset(dst + dstlen, 0, AV_INPUT_BUFFER_PADDING_SIZE); |
James Almer | b8e75a2 | 2018-04-04 19:07:36 | [diff] [blame] | 3283 | |
| 3284 | *pdst = dst; |
| 3285 | *size = dstlen; |
| 3286 | |
| 3287 | return 0; |
| 3288 | } |
| 3289 | |
Matthew Heaney | 818ebe9 | 2013-08-08 22:40:03 | [diff] [blame] | 3290 | static int matroska_parse_webvtt(MatroskaDemuxContext *matroska, |
| 3291 | MatroskaTrack *track, |
| 3292 | AVStream *st, |
| 3293 | uint8_t *data, int data_len, |
| 3294 | uint64_t timecode, |
| 3295 | uint64_t duration, |
| 3296 | int64_t pos) |
| 3297 | { |
James Almer | 4f55b94 | 2018-04-04 16:53:12 | [diff] [blame] | 3298 | AVPacket pktl, *pkt = &pktl; |
Matthew Heaney | 818ebe9 | 2013-08-08 22:40:03 | [diff] [blame] | 3299 | uint8_t *id, *settings, *text, *buf; |
| 3300 | int id_len, settings_len, text_len; |
| 3301 | uint8_t *p, *q; |
| 3302 | int err; |
| 3303 | |
| 3304 | if (data_len <= 0) |
| 3305 | return AVERROR_INVALIDDATA; |
| 3306 | |
| 3307 | p = data; |
| 3308 | q = data + data_len; |
| 3309 | |
| 3310 | id = p; |
| 3311 | id_len = -1; |
| 3312 | while (p < q) { |
| 3313 | if (*p == '\r' || *p == '\n') { |
| 3314 | id_len = p - id; |
| 3315 | if (*p == '\r') |
| 3316 | p++; |
| 3317 | break; |
| 3318 | } |
| 3319 | p++; |
| 3320 | } |
| 3321 | |
| 3322 | if (p >= q || *p != '\n') |
| 3323 | return AVERROR_INVALIDDATA; |
| 3324 | p++; |
| 3325 | |
| 3326 | settings = p; |
| 3327 | settings_len = -1; |
| 3328 | while (p < q) { |
| 3329 | if (*p == '\r' || *p == '\n') { |
| 3330 | settings_len = p - settings; |
| 3331 | if (*p == '\r') |
| 3332 | p++; |
| 3333 | break; |
| 3334 | } |
| 3335 | p++; |
| 3336 | } |
| 3337 | |
| 3338 | if (p >= q || *p != '\n') |
| 3339 | return AVERROR_INVALIDDATA; |
| 3340 | p++; |
| 3341 | |
| 3342 | text = p; |
| 3343 | text_len = q - p; |
| 3344 | while (text_len > 0) { |
| 3345 | const int len = text_len - 1; |
| 3346 | const uint8_t c = p[len]; |
| 3347 | if (c != '\r' && c != '\n') |
| 3348 | break; |
| 3349 | text_len = len; |
| 3350 | } |
| 3351 | |
| 3352 | if (text_len <= 0) |
| 3353 | return AVERROR_INVALIDDATA; |
| 3354 | |
Matthew Heaney | 818ebe9 | 2013-08-08 22:40:03 | [diff] [blame] | 3355 | err = av_new_packet(pkt, text_len); |
| 3356 | if (err < 0) { |
James Almer | f4f3958 | 2018-02-20 13:26:21 | [diff] [blame] | 3357 | return err; |
Matthew Heaney | 818ebe9 | 2013-08-08 22:40:03 | [diff] [blame] | 3358 | } |
| 3359 | |
| 3360 | memcpy(pkt->data, text, text_len); |
| 3361 | |
| 3362 | if (id_len > 0) { |
| 3363 | buf = av_packet_new_side_data(pkt, |
| 3364 | AV_PKT_DATA_WEBVTT_IDENTIFIER, |
| 3365 | id_len); |
Michael Niedermayer | fb33bff | 2014-08-15 18:33:21 | [diff] [blame] | 3366 | if (!buf) { |
James Almer | 88eb368 | 2018-02-20 13:25:54 | [diff] [blame] | 3367 | av_packet_unref(pkt); |
Matthew Heaney | 818ebe9 | 2013-08-08 22:40:03 | [diff] [blame] | 3368 | return AVERROR(ENOMEM); |
| 3369 | } |
| 3370 | memcpy(buf, id, id_len); |
| 3371 | } |
| 3372 | |
| 3373 | if (settings_len > 0) { |
| 3374 | buf = av_packet_new_side_data(pkt, |
| 3375 | AV_PKT_DATA_WEBVTT_SETTINGS, |
| 3376 | settings_len); |
Michael Niedermayer | fb33bff | 2014-08-15 18:33:21 | [diff] [blame] | 3377 | if (!buf) { |
James Almer | 88eb368 | 2018-02-20 13:25:54 | [diff] [blame] | 3378 | av_packet_unref(pkt); |
Matthew Heaney | 818ebe9 | 2013-08-08 22:40:03 | [diff] [blame] | 3379 | return AVERROR(ENOMEM); |
| 3380 | } |
| 3381 | memcpy(buf, settings, settings_len); |
| 3382 | } |
| 3383 | |
| 3384 | // Do we need this for subtitles? |
| 3385 | // pkt->flags = AV_PKT_FLAG_KEY; |
| 3386 | |
| 3387 | pkt->stream_index = st->index; |
| 3388 | pkt->pts = timecode; |
| 3389 | |
| 3390 | // Do we need this for subtitles? |
| 3391 | // pkt->dts = timecode; |
| 3392 | |
| 3393 | pkt->duration = duration; |
| 3394 | pkt->pos = pos; |
| 3395 | |
James Almer | 78b96be | 2018-03-26 18:02:37 | [diff] [blame] | 3396 | err = ff_packet_list_put(&matroska->queue, &matroska->queue_end, pkt, 0); |
| 3397 | if (err < 0) { |
James Almer | 4f55b94 | 2018-04-04 16:53:12 | [diff] [blame] | 3398 | av_packet_unref(pkt); |
James Almer | 78b96be | 2018-03-26 18:02:37 | [diff] [blame] | 3399 | return AVERROR(ENOMEM); |
| 3400 | } |
Matthew Heaney | 818ebe9 | 2013-08-08 22:40:03 | [diff] [blame] | 3401 | |
| 3402 | return 0; |
| 3403 | } |
| 3404 | |
Luca Barbato | c831ebf | 2012-09-16 23:28:13 | [diff] [blame] | 3405 | static int matroska_parse_frame(MatroskaDemuxContext *matroska, |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 3406 | MatroskaTrack *track, AVStream *st, |
James Almer | 9703b7d | 2018-04-04 21:12:53 | [diff] [blame] | 3407 | AVBufferRef *buf, uint8_t *data, int pkt_size, |
Michael Niedermayer | 8c51ea5 | 2012-09-20 18:37:26 | [diff] [blame] | 3408 | uint64_t timecode, uint64_t lace_duration, |
Vignesh Venkatasubramanian | 30c5c45 | 2013-02-13 21:51:48 | [diff] [blame] | 3409 | int64_t pos, int is_keyframe, |
Vignesh Venkatasubramanian | 7b0a839 | 2013-09-10 18:12:21 | [diff] [blame] | 3410 | uint8_t *additional, uint64_t additional_id, int additional_size, |
Jan Gerber | f4b1ca9 | 2013-11-14 11:58:28 | [diff] [blame] | 3411 | int64_t discard_padding) |
Luca Barbato | c831ebf | 2012-09-16 23:28:13 | [diff] [blame] | 3412 | { |
| 3413 | MatroskaTrackEncoding *encodings = track->encodings.elem; |
| 3414 | uint8_t *pkt_data = data; |
James Almer | b8e75a2 | 2018-04-04 19:07:36 | [diff] [blame] | 3415 | int res; |
James Almer | 2f0e0de | 2018-04-04 13:54:14 | [diff] [blame] | 3416 | AVPacket pktl, *pkt = &pktl; |
Luca Barbato | c831ebf | 2012-09-16 23:28:13 | [diff] [blame] | 3417 | |
Frank Galligan | b853103 | 2013-03-07 16:11:38 | [diff] [blame] | 3418 | if (encodings && !encodings->type && encodings->scope & 1) { |
Luca Barbato | c831ebf | 2012-09-16 23:28:13 | [diff] [blame] | 3419 | res = matroska_decode_buffer(&pkt_data, &pkt_size, track); |
| 3420 | if (res < 0) |
| 3421 | return res; |
| 3422 | } |
| 3423 | |
Anton Khirnov | 9200514 | 2014-06-18 18:42:52 | [diff] [blame] | 3424 | if (st->codecpar->codec_id == AV_CODEC_ID_WAVPACK) { |
Anton Khirnov | 9b6f47c | 2013-05-27 07:44:27 | [diff] [blame] | 3425 | uint8_t *wv_data; |
| 3426 | res = matroska_parse_wavpack(track, pkt_data, &wv_data, &pkt_size); |
| 3427 | if (res < 0) { |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 3428 | av_log(matroska->ctx, AV_LOG_ERROR, |
| 3429 | "Error parsing a wavpack block.\n"); |
Anton Khirnov | 9b6f47c | 2013-05-27 07:44:27 | [diff] [blame] | 3430 | goto fail; |
| 3431 | } |
| 3432 | if (pkt_data != data) |
| 3433 | av_freep(&pkt_data); |
| 3434 | pkt_data = wv_data; |
| 3435 | } |
| 3436 | |
Andreas Rheinhardt | af50f0a | 2019-12-06 23:16:19 | [diff] [blame] | 3437 | if (st->codecpar->codec_id == AV_CODEC_ID_PRORES && |
| 3438 | AV_RB32(pkt_data + 4) != MKBETAG('i', 'c', 'p', 'f')) { |
James Almer | b8e75a2 | 2018-04-04 19:07:36 | [diff] [blame] | 3439 | uint8_t *pr_data; |
| 3440 | res = matroska_parse_prores(track, pkt_data, &pr_data, &pkt_size); |
| 3441 | if (res < 0) { |
| 3442 | av_log(matroska->ctx, AV_LOG_ERROR, |
| 3443 | "Error parsing a prores block.\n"); |
| 3444 | goto fail; |
| 3445 | } |
| 3446 | if (pkt_data != data) |
| 3447 | av_freep(&pkt_data); |
| 3448 | pkt_data = pr_data; |
| 3449 | } |
Luca Barbato | c831ebf | 2012-09-16 23:28:13 | [diff] [blame] | 3450 | |
James Almer | 9703b7d | 2018-04-04 21:12:53 | [diff] [blame] | 3451 | av_init_packet(pkt); |
| 3452 | if (pkt_data != data) |
| 3453 | pkt->buf = av_buffer_create(pkt_data, pkt_size + AV_INPUT_BUFFER_PADDING_SIZE, |
| 3454 | NULL, NULL, 0); |
| 3455 | else |
| 3456 | pkt->buf = av_buffer_ref(buf); |
| 3457 | |
| 3458 | if (!pkt->buf) { |
Michael Niedermayer | 2fe4b62 | 2013-06-03 14:07:06 | [diff] [blame] | 3459 | res = AVERROR(ENOMEM); |
| 3460 | goto fail; |
Luca Barbato | c831ebf | 2012-09-16 23:28:13 | [diff] [blame] | 3461 | } |
| 3462 | |
James Almer | 9703b7d | 2018-04-04 21:12:53 | [diff] [blame] | 3463 | pkt->data = pkt_data; |
| 3464 | pkt->size = pkt_size; |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 3465 | pkt->flags = is_keyframe; |
Luca Barbato | c831ebf | 2012-09-16 23:28:13 | [diff] [blame] | 3466 | pkt->stream_index = st->index; |
| 3467 | |
Vignesh Venkatasubramanian | 30c5c45 | 2013-02-13 21:51:48 | [diff] [blame] | 3468 | if (additional_size > 0) { |
| 3469 | uint8_t *side_data = av_packet_new_side_data(pkt, |
| 3470 | AV_PKT_DATA_MATROSKA_BLOCKADDITIONAL, |
Michael Niedermayer | a08ebf0 | 2013-02-13 23:55:25 | [diff] [blame] | 3471 | additional_size + 8); |
Michael Niedermayer | fb33bff | 2014-08-15 18:33:21 | [diff] [blame] | 3472 | if (!side_data) { |
Hendrik Leppkes | c2f861c | 2015-10-27 13:35:30 | [diff] [blame] | 3473 | av_packet_unref(pkt); |
Vignesh Venkatasubramanian | 30c5c45 | 2013-02-13 21:51:48 | [diff] [blame] | 3474 | return AVERROR(ENOMEM); |
| 3475 | } |
Michael Niedermayer | a08ebf0 | 2013-02-13 23:55:25 | [diff] [blame] | 3476 | AV_WB64(side_data, additional_id); |
Vignesh Venkatasubramanian | 30c5c45 | 2013-02-13 21:51:48 | [diff] [blame] | 3477 | memcpy(side_data + 8, additional, additional_size); |
| 3478 | } |
| 3479 | |
Vignesh Venkatasubramanian | 7b0a839 | 2013-09-10 18:12:21 | [diff] [blame] | 3480 | if (discard_padding) { |
| 3481 | uint8_t *side_data = av_packet_new_side_data(pkt, |
| 3482 | AV_PKT_DATA_SKIP_SAMPLES, |
| 3483 | 10); |
Michael Niedermayer | fb33bff | 2014-08-15 18:33:21 | [diff] [blame] | 3484 | if (!side_data) { |
Hendrik Leppkes | c2f861c | 2015-10-27 13:35:30 | [diff] [blame] | 3485 | av_packet_unref(pkt); |
Vignesh Venkatasubramanian | 7b0a839 | 2013-09-10 18:12:21 | [diff] [blame] | 3486 | return AVERROR(ENOMEM); |
| 3487 | } |
James Almer | 70c6a1b | 2016-11-05 20:46:52 | [diff] [blame] | 3488 | discard_padding = av_rescale_q(discard_padding, |
Vignesh Venkatasubramanian | 7b0a839 | 2013-09-10 18:12:21 | [diff] [blame] | 3489 | (AVRational){1, 1000000000}, |
James Almer | 70c6a1b | 2016-11-05 20:46:52 | [diff] [blame] | 3490 | (AVRational){1, st->codecpar->sample_rate}); |
| 3491 | if (discard_padding > 0) { |
| 3492 | AV_WL32(side_data + 4, discard_padding); |
| 3493 | } else { |
| 3494 | AV_WL32(side_data, -discard_padding); |
| 3495 | } |
Vignesh Venkatasubramanian | 7b0a839 | 2013-09-10 18:12:21 | [diff] [blame] | 3496 | } |
| 3497 | |
Luca Barbato | c831ebf | 2012-09-16 23:28:13 | [diff] [blame] | 3498 | if (track->ms_compat) |
| 3499 | pkt->dts = timecode; |
| 3500 | else |
| 3501 | pkt->pts = timecode; |
| 3502 | pkt->pos = pos; |
Hendrik Leppkes | b01891a | 2015-09-29 13:14:59 | [diff] [blame] | 3503 | pkt->duration = lace_duration; |
| 3504 | |
wm4 | 948f3c1 | 2015-09-26 16:13:55 | [diff] [blame] | 3505 | #if FF_API_CONVERGENCE_DURATION |
| 3506 | FF_DISABLE_DEPRECATION_WARNINGS |
Derek Buitenhuis | 6f69f7a | 2016-04-10 19:58:15 | [diff] [blame] | 3507 | if (st->codecpar->codec_id == AV_CODEC_ID_SUBRIP) { |
Michael Niedermayer | 8c51ea5 | 2012-09-20 18:37:26 | [diff] [blame] | 3508 | pkt->convergence_duration = lace_duration; |
| 3509 | } |
wm4 | 948f3c1 | 2015-09-26 16:13:55 | [diff] [blame] | 3510 | FF_ENABLE_DEPRECATION_WARNINGS |
| 3511 | #endif |
Luca Barbato | c831ebf | 2012-09-16 23:28:13 | [diff] [blame] | 3512 | |
James Almer | 78b96be | 2018-03-26 18:02:37 | [diff] [blame] | 3513 | res = ff_packet_list_put(&matroska->queue, &matroska->queue_end, pkt, 0); |
| 3514 | if (res < 0) { |
James Almer | 2f0e0de | 2018-04-04 13:54:14 | [diff] [blame] | 3515 | av_packet_unref(pkt); |
James Almer | 78b96be | 2018-03-26 18:02:37 | [diff] [blame] | 3516 | return AVERROR(ENOMEM); |
| 3517 | } |
Luca Barbato | c831ebf | 2012-09-16 23:28:13 | [diff] [blame] | 3518 | |
| 3519 | return 0; |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 3520 | |
Anton Khirnov | 9b6f47c | 2013-05-27 07:44:27 | [diff] [blame] | 3521 | fail: |
| 3522 | if (pkt_data != data) |
| 3523 | av_freep(&pkt_data); |
| 3524 | return res; |
Luca Barbato | c831ebf | 2012-09-16 23:28:13 | [diff] [blame] | 3525 | } |
| 3526 | |
James Almer | 9703b7d | 2018-04-04 21:12:53 | [diff] [blame] | 3527 | static int matroska_parse_block(MatroskaDemuxContext *matroska, AVBufferRef *buf, uint8_t *data, |
Aurelien Jacobs | f7b9687 | 2008-08-05 00:42:05 | [diff] [blame] | 3528 | int size, int64_t pos, uint64_t cluster_time, |
Luca Barbato | 7d84310 | 2012-09-17 00:48:02 | [diff] [blame] | 3529 | uint64_t block_duration, int is_keyframe, |
Vignesh Venkatasubramanian | 30c5c45 | 2013-02-13 21:51:48 | [diff] [blame] | 3530 | uint8_t *additional, uint64_t additional_id, int additional_size, |
Jan Gerber | f4b1ca9 | 2013-11-14 11:58:28 | [diff] [blame] | 3531 | int64_t cluster_pos, int64_t discard_padding) |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 3532 | { |
Aurelien Jacobs | f14a201 | 2008-09-09 11:54:35 | [diff] [blame] | 3533 | uint64_t timecode = AV_NOPTS_VALUE; |
Aurelien Jacobs | 009ecd5 | 2008-08-05 00:40:12 | [diff] [blame] | 3534 | MatroskaTrack *track; |
Andreas Rheinhardt | d5274f8 | 2019-12-03 17:09:10 | [diff] [blame] | 3535 | AVIOContext pb; |
Aurelien Jacobs | a3467f8 | 2008-09-06 23:44:29 | [diff] [blame] | 3536 | int res = 0; |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 3537 | AVStream *st; |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 3538 | int16_t block_time; |
Andreas Rheinhardt | 9ad1a6d | 2019-12-03 17:09:06 | [diff] [blame] | 3539 | uint32_t lace_size[256]; |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 3540 | int n, flags, laces = 0; |
| 3541 | uint64_t num; |
Michael Niedermayer | 6158a3b | 2013-07-15 15:13:45 | [diff] [blame] | 3542 | int trust_default_duration = 1; |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 3543 | |
Andreas Rheinhardt | d5274f8 | 2019-12-03 17:09:10 | [diff] [blame] | 3544 | ffio_init_context(&pb, data, size, 0, NULL, NULL, NULL, NULL); |
| 3545 | |
| 3546 | if ((n = ebml_read_num(matroska, &pb, 8, &num, 1)) < 0) |
Luca Barbato | 721af29 | 2012-04-30 00:39:31 | [diff] [blame] | 3547 | return n; |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 3548 | data += n; |
| 3549 | size -= n; |
| 3550 | |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 3551 | track = matroska_find_track_by_num(matroska, num); |
Ronald S. Bultje | d31fb1a | 2011-10-29 23:17:51 | [diff] [blame] | 3552 | if (!track || !track->stream) { |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 3553 | av_log(matroska->ctx, AV_LOG_INFO, |
Andreas Rheinhardt | f767c68 | 2019-05-16 22:29:46 | [diff] [blame] | 3554 | "Invalid stream %"PRIu64"\n", num); |
Reimar Döffinger | d493170 | 2012-02-13 22:06:19 | [diff] [blame] | 3555 | return AVERROR_INVALIDDATA; |
Ronald S. Bultje | d31fb1a | 2011-10-29 23:17:51 | [diff] [blame] | 3556 | } else if (size <= 3) |
| 3557 | return 0; |
Aurelien Jacobs | 009ecd5 | 2008-08-05 00:40:12 | [diff] [blame] | 3558 | st = track->stream; |
Aurelien Jacobs | 16f97ab | 2008-08-05 00:41:10 | [diff] [blame] | 3559 | if (st->discard >= AVDISCARD_ALL) |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 3560 | return res; |
Michael Niedermayer | 5864ce1 | 2012-09-20 19:46:35 | [diff] [blame] | 3561 | av_assert1(block_duration != AV_NOPTS_VALUE); |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 3562 | |
Michael Niedermayer | 729fa55 | 2013-05-19 21:38:01 | [diff] [blame] | 3563 | block_time = sign_extend(AV_RB16(data), 16); |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 3564 | data += 2; |
| 3565 | flags = *data++; |
| 3566 | size -= 3; |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 3567 | if (is_keyframe == -1) |
Jean-Daniel Dupas | cc947f0 | 2010-03-31 12:29:58 | [diff] [blame] | 3568 | is_keyframe = flags & 0x80 ? AV_PKT_FLAG_KEY : 0; |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 3569 | |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 3570 | if (cluster_time != (uint64_t) -1 && |
| 3571 | (block_time >= 0 || cluster_time >= -block_time)) { |
Michael Niedermayer | b5bc436 | 2016-06-06 02:23:16 | [diff] [blame] | 3572 | timecode = cluster_time + block_time - track->codec_delay_in_track_tb; |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 3573 | if (track->type == MATROSKA_TRACK_TYPE_SUBTITLE && |
| 3574 | timecode < track->end_timecode) |
Aurelien Jacobs | 82360e6 | 2008-09-09 12:07:10 | [diff] [blame] | 3575 | is_keyframe = 0; /* overlapping subtitles are not key frame */ |
Michael Niedermayer | b347ca9 | 2016-12-26 02:16:01 | [diff] [blame] | 3576 | if (is_keyframe) { |
| 3577 | ff_reduce_index(matroska->ctx, st->index); |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 3578 | av_add_index_entry(st, cluster_pos, timecode, 0, 0, |
| 3579 | AVINDEX_KEYFRAME); |
Michael Niedermayer | b347ca9 | 2016-12-26 02:16:01 | [diff] [blame] | 3580 | } |
Aurelien Jacobs | f14a201 | 2008-09-09 11:54:35 | [diff] [blame] | 3581 | } |
| 3582 | |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 3583 | if (matroska->skip_to_keyframe && |
| 3584 | track->type != MATROSKA_TRACK_TYPE_SUBTITLE) { |
Chris Cunningham | d59820f | 2016-07-21 19:01:45 | [diff] [blame] | 3585 | // Compare signed timecodes. Timecode may be negative due to codec delay |
| 3586 | // offset. We don't support timestamps greater than int64_t anyway - see |
| 3587 | // AVPacket's pts. |
Chris Cunningham | 52ec4cc | 2016-07-28 01:33:30 | [diff] [blame] | 3588 | if ((int64_t)timecode < (int64_t)matroska->skip_to_timecode) |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 3589 | return res; |
Monty Montgomery | f6622f9 | 2013-09-11 06:00:31 | [diff] [blame] | 3590 | if (is_keyframe) |
| 3591 | matroska->skip_to_keyframe = 0; |
| 3592 | else if (!st->skip_to_keyframe) { |
Reimar Döffinger | 4a95876 | 2012-04-13 22:17:30 | [diff] [blame] | 3593 | av_log(matroska->ctx, AV_LOG_ERROR, "File is broken, keyframes not correctly marked!\n"); |
| 3594 | matroska->skip_to_keyframe = 0; |
| 3595 | } |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 3596 | } |
| 3597 | |
Andreas Rheinhardt | f74eaa1 | 2019-12-03 17:09:08 | [diff] [blame] | 3598 | res = matroska_parse_laces(matroska, &data, size, (flags & 0x06) >> 1, |
Andreas Rheinhardt | d5274f8 | 2019-12-03 17:09:10 | [diff] [blame] | 3599 | &pb, lace_size, &laces); |
Andreas Rheinhardt | dbe3be6 | 2019-12-03 17:09:09 | [diff] [blame] | 3600 | if (res < 0) { |
| 3601 | av_log(matroska->ctx, AV_LOG_ERROR, "Error parsing frame sizes.\n"); |
Andreas Rheinhardt | 9ad1a6d | 2019-12-03 17:09:06 | [diff] [blame] | 3602 | return res; |
Andreas Rheinhardt | dbe3be6 | 2019-12-03 17:09:09 | [diff] [blame] | 3603 | } |
Aurelien Jacobs | eabb8ba | 2007-06-04 22:19:17 | [diff] [blame] | 3604 | |
Michael Niedermayer | 6158a3b | 2013-07-15 15:13:45 | [diff] [blame] | 3605 | if (track->audio.samplerate == 8000) { |
| 3606 | // If this is needed for more codecs, then add them here |
Derek Buitenhuis | 6f69f7a | 2016-04-10 19:58:15 | [diff] [blame] | 3607 | if (st->codecpar->codec_id == AV_CODEC_ID_AC3) { |
| 3608 | if (track->audio.samplerate != st->codecpar->sample_rate || !st->codecpar->frame_size) |
Michael Niedermayer | 6158a3b | 2013-07-15 15:13:45 | [diff] [blame] | 3609 | trust_default_duration = 0; |
| 3610 | } |
| 3611 | } |
| 3612 | |
| 3613 | if (!block_duration && trust_default_duration) |
Michael Niedermayer | 5864ce1 | 2012-09-20 19:46:35 | [diff] [blame] | 3614 | block_duration = track->default_duration * laces / matroska->time_scale; |
Michael Niedermayer | 8c51ea5 | 2012-09-20 18:37:26 | [diff] [blame] | 3615 | |
| 3616 | if (cluster_time != (uint64_t)-1 && (block_time >= 0 || cluster_time >= -block_time)) |
Luca Barbato | 7d84310 | 2012-09-17 00:48:02 | [diff] [blame] | 3617 | track->end_timecode = |
| 3618 | FFMAX(track->end_timecode, timecode + block_duration); |
Michael Niedermayer | 8c51ea5 | 2012-09-20 18:37:26 | [diff] [blame] | 3619 | |
Luca Barbato | c831ebf | 2012-09-16 23:28:13 | [diff] [blame] | 3620 | for (n = 0; n < laces; n++) { |
Michael Niedermayer | 5864ce1 | 2012-09-20 19:46:35 | [diff] [blame] | 3621 | int64_t lace_duration = block_duration*(n+1) / laces - block_duration*n / laces; |
Michael Niedermayer | 8c51ea5 | 2012-09-20 18:37:26 | [diff] [blame] | 3622 | |
Andreas Rheinhardt | c6f60b9 | 2019-12-02 10:04:55 | [diff] [blame] | 3623 | if (track->audio.buf) { |
Luca Barbato | 25a80a9 | 2013-03-29 11:51:51 | [diff] [blame] | 3624 | res = matroska_parse_rm_audio(matroska, track, st, data, |
| 3625 | lace_size[n], |
Michael Niedermayer | a6ec1e4 | 2012-09-20 21:43:20 | [diff] [blame] | 3626 | timecode, pos); |
Luca Barbato | c831ebf | 2012-09-16 23:28:13 | [diff] [blame] | 3627 | if (res) |
Andreas Rheinhardt | 9ad1a6d | 2019-12-03 17:09:06 | [diff] [blame] | 3628 | return res; |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 3629 | |
Derek Buitenhuis | 6f69f7a | 2016-04-10 19:58:15 | [diff] [blame] | 3630 | } else if (st->codecpar->codec_id == AV_CODEC_ID_WEBVTT) { |
Matthew Heaney | 818ebe9 | 2013-08-08 22:40:03 | [diff] [blame] | 3631 | res = matroska_parse_webvtt(matroska, track, st, |
| 3632 | data, lace_size[n], |
| 3633 | timecode, lace_duration, |
| 3634 | pos); |
| 3635 | if (res) |
Andreas Rheinhardt | 9ad1a6d | 2019-12-03 17:09:06 | [diff] [blame] | 3636 | return res; |
Luca Barbato | c831ebf | 2012-09-16 23:28:13 | [diff] [blame] | 3637 | } else { |
James Almer | 9703b7d | 2018-04-04 21:12:53 | [diff] [blame] | 3638 | res = matroska_parse_frame(matroska, track, st, buf, data, lace_size[n], |
Michael Niedermayer | b3d9ab1 | 2014-03-07 22:46:37 | [diff] [blame] | 3639 | timecode, lace_duration, pos, |
| 3640 | !n ? is_keyframe : 0, |
| 3641 | additional, additional_id, additional_size, |
| 3642 | discard_padding); |
Luca Barbato | c831ebf | 2012-09-16 23:28:13 | [diff] [blame] | 3643 | if (res) |
Andreas Rheinhardt | 9ad1a6d | 2019-12-03 17:09:06 | [diff] [blame] | 3644 | return res; |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 3645 | } |
Luca Barbato | c831ebf | 2012-09-16 23:28:13 | [diff] [blame] | 3646 | |
| 3647 | if (timecode != AV_NOPTS_VALUE) |
Michael Niedermayer | 8c51ea5 | 2012-09-20 18:37:26 | [diff] [blame] | 3648 | timecode = lace_duration ? timecode + lace_duration : AV_NOPTS_VALUE; |
Luca Barbato | c831ebf | 2012-09-16 23:28:13 | [diff] [blame] | 3649 | data += lace_size[n]; |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 3650 | } |
| 3651 | |
Andreas Rheinhardt | 9ad1a6d | 2019-12-03 17:09:06 | [diff] [blame] | 3652 | return 0; |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 3653 | } |
| 3654 | |
Andreas Rheinhardt | 70baf72 | 2019-05-16 22:29:58 | [diff] [blame] | 3655 | static int matroska_parse_cluster(MatroskaDemuxContext *matroska) |
Dale Curtis | 8336eb6 | 2012-04-19 18:12:24 | [diff] [blame] | 3656 | { |
Andreas Rheinhardt | ffa64a4 | 2019-05-16 22:29:59 | [diff] [blame] | 3657 | MatroskaCluster *cluster = &matroska->current_cluster; |
| 3658 | MatroskaBlock *block = &cluster->block; |
| 3659 | int res; |
Andreas Rheinhardt | 865c537 | 2019-05-16 22:30:08 | [diff] [blame] | 3660 | |
| 3661 | av_assert0(matroska->num_levels <= 2); |
| 3662 | |
| 3663 | if (matroska->num_levels == 1) { |
Andreas Rheinhardt | 71c9088 | 2019-05-16 22:30:18 | [diff] [blame] | 3664 | res = ebml_parse(matroska, matroska_segment, NULL); |
Andreas Rheinhardt | 865c537 | 2019-05-16 22:30:08 | [diff] [blame] | 3665 | |
| 3666 | if (res == 1) { |
| 3667 | /* Found a cluster: subtract the size of the ID already read. */ |
| 3668 | cluster->pos = avio_tell(matroska->ctx->pb) - 4; |
| 3669 | |
| 3670 | res = ebml_parse(matroska, matroska_cluster_enter, cluster); |
| 3671 | if (res < 0) |
| 3672 | return res; |
| 3673 | } |
Dale Curtis | 8336eb6 | 2012-04-19 18:12:24 | [diff] [blame] | 3674 | } |
| 3675 | |
Andreas Rheinhardt | 865c537 | 2019-05-16 22:30:08 | [diff] [blame] | 3676 | if (matroska->num_levels == 2) { |
Andreas Rheinhardt | 865c537 | 2019-05-16 22:30:08 | [diff] [blame] | 3677 | /* We are inside a cluster. */ |
| 3678 | res = ebml_parse(matroska, matroska_cluster_parsing, cluster); |
| 3679 | |
| 3680 | if (res >= 0 && block->bin.size > 0) { |
Andreas Rheinhardt | ffa64a4 | 2019-05-16 22:29:59 | [diff] [blame] | 3681 | int is_keyframe = block->non_simple ? block->reference == INT64_MIN : -1; |
| 3682 | uint8_t* additional = block->additional.size > 0 ? |
| 3683 | block->additional.data : NULL; |
Dale Curtis | 8336eb6 | 2012-04-19 18:12:24 | [diff] [blame] | 3684 | |
Andreas Rheinhardt | a9f0515 | 2019-05-16 22:30:09 | [diff] [blame] | 3685 | res = matroska_parse_block(matroska, block->bin.buf, block->bin.data, |
Andreas Rheinhardt | ffa64a4 | 2019-05-16 22:29:59 | [diff] [blame] | 3686 | block->bin.size, block->bin.pos, |
Andreas Rheinhardt | 865c537 | 2019-05-16 22:30:08 | [diff] [blame] | 3687 | cluster->timecode, block->duration, |
| 3688 | is_keyframe, additional, block->additional_id, |
| 3689 | block->additional.size, cluster->pos, |
Andreas Rheinhardt | ffa64a4 | 2019-05-16 22:29:59 | [diff] [blame] | 3690 | block->discard_padding); |
Andreas Rheinhardt | 865c537 | 2019-05-16 22:30:08 | [diff] [blame] | 3691 | } |
Dale Curtis | 8336eb6 | 2012-04-19 18:12:24 | [diff] [blame] | 3692 | |
Andreas Rheinhardt | 60f75c9 | 2019-05-16 22:30:13 | [diff] [blame] | 3693 | ebml_free(matroska_blockgroup, block); |
| 3694 | memset(block, 0, sizeof(*block)); |
Andreas Rheinhardt | 865c537 | 2019-05-16 22:30:08 | [diff] [blame] | 3695 | } else if (!matroska->num_levels) { |
Andreas Rheinhardt | 3ed2755 | 2019-05-16 22:30:11 | [diff] [blame] | 3696 | if (!avio_feof(matroska->ctx->pb)) { |
| 3697 | avio_r8(matroska->ctx->pb); |
| 3698 | if (!avio_feof(matroska->ctx->pb)) { |
| 3699 | av_log(matroska->ctx, AV_LOG_WARNING, "File extends beyond " |
| 3700 | "end of segment.\n"); |
| 3701 | return AVERROR_INVALIDDATA; |
| 3702 | } |
| 3703 | } |
Andreas Rheinhardt | 865c537 | 2019-05-16 22:30:08 | [diff] [blame] | 3704 | matroska->done = 1; |
| 3705 | return AVERROR_EOF; |
| 3706 | } |
| 3707 | |
Dale Curtis | 8336eb6 | 2012-04-19 18:12:24 | [diff] [blame] | 3708 | return res; |
| 3709 | } |
| 3710 | |
Aurelien Jacobs | f7b9687 | 2008-08-05 00:42:05 | [diff] [blame] | 3711 | static int matroska_read_packet(AVFormatContext *s, AVPacket *pkt) |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 3712 | { |
| 3713 | MatroskaDemuxContext *matroska = s->priv_data; |
Sophia Wang | 8c83062 | 2016-09-27 19:00:29 | [diff] [blame] | 3714 | int ret = 0; |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 3715 | |
Andreas Rheinhardt | a3db9f6 | 2019-05-16 22:30:05 | [diff] [blame] | 3716 | if (matroska->resync_pos == -1) { |
| 3717 | // This can only happen if generic seeking has been used. |
| 3718 | matroska->resync_pos = avio_tell(s->pb); |
| 3719 | } |
| 3720 | |
Hendrik Leppkes | 8689d87 | 2011-07-06 17:57:11 | [diff] [blame] | 3721 | while (matroska_deliver_packet(matroska, pkt)) { |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 3722 | if (matroska->done) |
Sophia Wang | 8c83062 | 2016-09-27 19:00:29 | [diff] [blame] | 3723 | return (ret < 0) ? ret : AVERROR_EOF; |
Andreas Rheinhardt | 3ed2755 | 2019-05-16 22:30:11 | [diff] [blame] | 3724 | if (matroska_parse_cluster(matroska) < 0 && !matroska->done) |
Andreas Rheinhardt | a3db9f6 | 2019-05-16 22:30:05 | [diff] [blame] | 3725 | ret = matroska_resync(matroska, matroska->resync_pos); |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 3726 | } |
| 3727 | |
Michael Niedermayer | d1afa72 | 2019-02-06 14:29:38 | [diff] [blame] | 3728 | return 0; |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 3729 | } |
| 3730 | |
Aurelien Jacobs | f7b9687 | 2008-08-05 00:42:05 | [diff] [blame] | 3731 | static int matroska_read_seek(AVFormatContext *s, int stream_index, |
| 3732 | int64_t timestamp, int flags) |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 3733 | { |
| 3734 | MatroskaDemuxContext *matroska = s->priv_data; |
Xiaohan Wang | 33301f0 | 2014-11-06 20:59:54 | [diff] [blame] | 3735 | MatroskaTrack *tracks = NULL; |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 3736 | AVStream *st = s->streams[stream_index]; |
Andreas Rheinhardt | c6bb825 | 2019-05-16 22:29:45 | [diff] [blame] | 3737 | int i, index; |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 3738 | |
Aaron Colwell | 31ad14c | 2011-07-09 05:48:43 | [diff] [blame] | 3739 | /* Parse the CUES now since we need the index data to seek. */ |
Reimar Döffinger | 47e015e | 2012-02-12 13:09:03 | [diff] [blame] | 3740 | if (matroska->cues_parsing_deferred > 0) { |
Aaron Colwell | 31ad14c | 2011-07-09 05:48:43 | [diff] [blame] | 3741 | matroska->cues_parsing_deferred = 0; |
Reimar Döffinger | 47e015e | 2012-02-12 13:09:03 | [diff] [blame] | 3742 | matroska_parse_cues(matroska); |
Aaron Colwell | 31ad14c | 2011-07-09 05:48:43 | [diff] [blame] | 3743 | } |
| 3744 | |
Aurelien Jacobs | a8fd7e7 | 2008-09-12 00:06:06 | [diff] [blame] | 3745 | if (!st->nb_index_entries) |
Reimar Döffinger | 47e015e | 2012-02-12 13:09:03 | [diff] [blame] | 3746 | goto err; |
Aurelien Jacobs | a8fd7e7 | 2008-09-12 00:06:06 | [diff] [blame] | 3747 | timestamp = FFMAX(timestamp, st->index_entries[0].timestamp); |
Aurelien Jacobs | dfbbbdc | 2008-08-24 23:57:29 | [diff] [blame] | 3748 | |
Michael Niedermayer | a392018 | 2014-10-22 02:21:39 | [diff] [blame] | 3749 | if ((index = av_index_search_timestamp(st, timestamp, flags)) < 0 || index == st->nb_index_entries - 1) { |
Andreas Rheinhardt | 8a286e7 | 2019-05-16 22:30:03 | [diff] [blame] | 3750 | matroska_reset_status(matroska, 0, st->index_entries[st->nb_index_entries - 1].pos); |
Michael Niedermayer | a392018 | 2014-10-22 02:21:39 | [diff] [blame] | 3751 | while ((index = av_index_search_timestamp(st, timestamp, flags)) < 0 || index == st->nb_index_entries - 1) { |
Aurelien Jacobs | 0dbddda | 2008-08-27 19:58:55 | [diff] [blame] | 3752 | matroska_clear_queue(matroska); |
| 3753 | if (matroska_parse_cluster(matroska) < 0) |
| 3754 | break; |
| 3755 | } |
Aurelien Jacobs | 6bef5f9 | 2008-08-27 19:57:42 | [diff] [blame] | 3756 | } |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 3757 | |
Aurelien Jacobs | 243cc4c | 2007-12-29 18:35:38 | [diff] [blame] | 3758 | matroska_clear_queue(matroska); |
Reimar Döffinger | 47e015e | 2012-02-12 13:09:03 | [diff] [blame] | 3759 | if (index < 0 || (matroska->cues_parsing_deferred < 0 && index == st->nb_index_entries - 1)) |
| 3760 | goto err; |
Aurelien Jacobs | 243cc4c | 2007-12-29 18:35:38 | [diff] [blame] | 3761 | |
Xiaohan Wang | 33301f0 | 2014-11-06 20:59:54 | [diff] [blame] | 3762 | tracks = matroska->tracks.elem; |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 3763 | for (i = 0; i < matroska->tracks.nb_elem; i++) { |
| 3764 | tracks[i].audio.pkt_cnt = 0; |
Reimar Döffinger | b09e506 | 2011-02-26 11:52:01 | [diff] [blame] | 3765 | tracks[i].audio.sub_packet_cnt = 0; |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 3766 | tracks[i].audio.buf_timecode = AV_NOPTS_VALUE; |
| 3767 | tracks[i].end_timecode = 0; |
Aurelien Jacobs | c165825 | 2008-09-09 12:10:25 | [diff] [blame] | 3768 | } |
| 3769 | |
Andreas Rheinhardt | 8a286e7 | 2019-05-16 22:30:03 | [diff] [blame] | 3770 | /* We seek to a level 1 element, so set the appropriate status. */ |
| 3771 | matroska_reset_status(matroska, 0, st->index_entries[index].pos); |
Michael Niedermayer | 7c00d85 | 2013-02-07 20:34:35 | [diff] [blame] | 3772 | if (flags & AVSEEK_FLAG_ANY) { |
| 3773 | st->skip_to_keyframe = 0; |
| 3774 | matroska->skip_to_timecode = timestamp; |
| 3775 | } else { |
| 3776 | st->skip_to_keyframe = 1; |
| 3777 | matroska->skip_to_timecode = st->index_entries[index].timestamp; |
| 3778 | } |
| 3779 | matroska->skip_to_keyframe = 1; |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 3780 | matroska->done = 0; |
Anton Khirnov | a2faa95 | 2011-10-16 13:03:30 | [diff] [blame] | 3781 | ff_update_cur_dts(s, st, st->index_entries[index].timestamp); |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 3782 | return 0; |
Reimar Döffinger | 47e015e | 2012-02-12 13:09:03 | [diff] [blame] | 3783 | err: |
| 3784 | // slightly hackish but allows proper fallback to |
| 3785 | // the generic seeking code. |
Andreas Rheinhardt | 8a286e7 | 2019-05-16 22:30:03 | [diff] [blame] | 3786 | matroska_reset_status(matroska, 0, -1); |
Andreas Rheinhardt | a3db9f6 | 2019-05-16 22:30:05 | [diff] [blame] | 3787 | matroska->resync_pos = -1; |
Reimar Döffinger | 47e015e | 2012-02-12 13:09:03 | [diff] [blame] | 3788 | matroska_clear_queue(matroska); |
Reimar Döffinger | 4a95876 | 2012-04-13 22:17:30 | [diff] [blame] | 3789 | st->skip_to_keyframe = |
Reimar Döffinger | 47e015e | 2012-02-12 13:09:03 | [diff] [blame] | 3790 | matroska->skip_to_keyframe = 0; |
| 3791 | matroska->done = 0; |
Reimar Döffinger | 47e015e | 2012-02-12 13:09:03 | [diff] [blame] | 3792 | return -1; |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 3793 | } |
| 3794 | |
Aurelien Jacobs | f7b9687 | 2008-08-05 00:42:05 | [diff] [blame] | 3795 | static int matroska_read_close(AVFormatContext *s) |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 3796 | { |
| 3797 | MatroskaDemuxContext *matroska = s->priv_data; |
Aurelien Jacobs | 2cbc881 | 2008-08-05 00:40:31 | [diff] [blame] | 3798 | MatroskaTrack *tracks = matroska->tracks.elem; |
Aurelien Jacobs | 70109c0 | 2008-08-05 00:41:13 | [diff] [blame] | 3799 | int n; |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 3800 | |
Aurelien Jacobs | 34c9c1b | 2007-12-29 18:32:47 | [diff] [blame] | 3801 | matroska_clear_queue(matroska); |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 3802 | |
Keiji Costantini | 84cfce9 | 2014-03-01 16:28:15 | [diff] [blame] | 3803 | for (n = 0; n < matroska->tracks.nb_elem; n++) |
Aurelien Jacobs | 2cbc881 | 2008-08-05 00:40:31 | [diff] [blame] | 3804 | if (tracks[n].type == MATROSKA_TRACK_TYPE_AUDIO) |
Michael Niedermayer | 6e70e4a | 2015-01-06 11:48:38 | [diff] [blame] | 3805 | av_freep(&tracks[n].audio.buf); |
Aurelien Jacobs | ce6f28b | 2008-08-05 00:40:58 | [diff] [blame] | 3806 | ebml_free(matroska_segment, matroska); |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 3807 | |
| 3808 | return 0; |
| 3809 | } |
| 3810 | |
Vignesh Venkatasubramanian | 5a20656 | 2014-07-07 19:52:37 | [diff] [blame] | 3811 | typedef struct { |
| 3812 | int64_t start_time_ns; |
| 3813 | int64_t end_time_ns; |
| 3814 | int64_t start_offset; |
| 3815 | int64_t end_offset; |
| 3816 | } CueDesc; |
| 3817 | |
James Zern | 20aeee4 | 2017-04-17 17:59:31 | [diff] [blame] | 3818 | /* This function searches all the Cues and returns the CueDesc corresponding to |
Vignesh Venkatasubramanian | 5a20656 | 2014-07-07 19:52:37 | [diff] [blame] | 3819 | * the timestamp ts. Returned CueDesc will be such that start_time_ns <= ts < |
| 3820 | * end_time_ns. All 4 fields will be set to -1 if ts >= file's duration. |
| 3821 | */ |
| 3822 | static CueDesc get_cue_desc(AVFormatContext *s, int64_t ts, int64_t cues_start) { |
| 3823 | MatroskaDemuxContext *matroska = s->priv_data; |
| 3824 | CueDesc cue_desc; |
| 3825 | int i; |
| 3826 | int nb_index_entries = s->streams[0]->nb_index_entries; |
| 3827 | AVIndexEntry *index_entries = s->streams[0]->index_entries; |
| 3828 | if (ts >= matroska->duration * matroska->time_scale) return (CueDesc) {-1, -1, -1, -1}; |
| 3829 | for (i = 1; i < nb_index_entries; i++) { |
| 3830 | if (index_entries[i - 1].timestamp * matroska->time_scale <= ts && |
| 3831 | index_entries[i].timestamp * matroska->time_scale > ts) { |
| 3832 | break; |
| 3833 | } |
| 3834 | } |
| 3835 | --i; |
| 3836 | cue_desc.start_time_ns = index_entries[i].timestamp * matroska->time_scale; |
| 3837 | cue_desc.start_offset = index_entries[i].pos - matroska->segment_start; |
| 3838 | if (i != nb_index_entries - 1) { |
| 3839 | cue_desc.end_time_ns = index_entries[i + 1].timestamp * matroska->time_scale; |
| 3840 | cue_desc.end_offset = index_entries[i + 1].pos - matroska->segment_start; |
| 3841 | } else { |
| 3842 | cue_desc.end_time_ns = matroska->duration * matroska->time_scale; |
| 3843 | // FIXME: this needs special handling for files where Cues appear |
| 3844 | // before Clusters. the current logic assumes Cues appear after |
| 3845 | // Clusters. |
| 3846 | cue_desc.end_offset = cues_start - matroska->segment_start; |
| 3847 | } |
| 3848 | return cue_desc; |
| 3849 | } |
| 3850 | |
| 3851 | static int webm_clusters_start_with_keyframe(AVFormatContext *s) |
| 3852 | { |
| 3853 | MatroskaDemuxContext *matroska = s->priv_data; |
Andreas Rheinhardt | 8a286e7 | 2019-05-16 22:30:03 | [diff] [blame] | 3854 | uint32_t id = matroska->current_id; |
Vignesh Venkatasubramanian | 5a20656 | 2014-07-07 19:52:37 | [diff] [blame] | 3855 | int64_t cluster_pos, before_pos; |
| 3856 | int index, rv = 1; |
| 3857 | if (s->streams[0]->nb_index_entries <= 0) return 0; |
| 3858 | // seek to the first cluster using cues. |
| 3859 | index = av_index_search_timestamp(s->streams[0], 0, 0); |
| 3860 | if (index < 0) return 0; |
| 3861 | cluster_pos = s->streams[0]->index_entries[index].pos; |
| 3862 | before_pos = avio_tell(s->pb); |
| 3863 | while (1) { |
Andreas Rheinhardt | 36aceb6 | 2019-05-16 22:29:49 | [diff] [blame] | 3864 | uint64_t cluster_id, cluster_length; |
| 3865 | int read; |
Vignesh Venkatasubramanian | 5a20656 | 2014-07-07 19:52:37 | [diff] [blame] | 3866 | AVPacket *pkt; |
| 3867 | avio_seek(s->pb, cluster_pos, SEEK_SET); |
| 3868 | // read cluster id and length |
Andreas Rheinhardt | 239c736 | 2019-06-25 01:08:56 | [diff] [blame] | 3869 | read = ebml_read_num(matroska, matroska->ctx->pb, 4, &cluster_id, 1); |
Andreas Rheinhardt | 36aceb6 | 2019-05-16 22:29:49 | [diff] [blame] | 3870 | if (read < 0 || cluster_id != 0xF43B675) // done with all clusters |
Vignesh Venkatasubramanian | 5a20656 | 2014-07-07 19:52:37 | [diff] [blame] | 3871 | break; |
Andreas Rheinhardt | 36aceb6 | 2019-05-16 22:29:49 | [diff] [blame] | 3872 | read = ebml_read_length(matroska, matroska->ctx->pb, &cluster_length); |
| 3873 | if (read < 0) |
| 3874 | break; |
Andreas Rheinhardt | 8a286e7 | 2019-05-16 22:30:03 | [diff] [blame] | 3875 | |
| 3876 | matroska_reset_status(matroska, 0, cluster_pos); |
Vignesh Venkatasubramanian | 5a20656 | 2014-07-07 19:52:37 | [diff] [blame] | 3877 | matroska_clear_queue(matroska); |
| 3878 | if (matroska_parse_cluster(matroska) < 0 || |
James Almer | 78b96be | 2018-03-26 18:02:37 | [diff] [blame] | 3879 | !matroska->queue) { |
Vignesh Venkatasubramanian | 5a20656 | 2014-07-07 19:52:37 | [diff] [blame] | 3880 | break; |
| 3881 | } |
James Almer | 78b96be | 2018-03-26 18:02:37 | [diff] [blame] | 3882 | pkt = &matroska->queue->pkt; |
Andreas Rheinhardt | 36aceb6 | 2019-05-16 22:29:49 | [diff] [blame] | 3883 | // 4 + read is the length of the cluster id and the cluster length field. |
| 3884 | cluster_pos += 4 + read + cluster_length; |
Vignesh Venkatasubramanian | 5a20656 | 2014-07-07 19:52:37 | [diff] [blame] | 3885 | if (!(pkt->flags & AV_PKT_FLAG_KEY)) { |
| 3886 | rv = 0; |
| 3887 | break; |
| 3888 | } |
| 3889 | } |
Andreas Rheinhardt | 8a286e7 | 2019-05-16 22:30:03 | [diff] [blame] | 3890 | |
| 3891 | /* Restore the status after matroska_read_header: */ |
| 3892 | matroska_reset_status(matroska, id, before_pos); |
| 3893 | |
Vignesh Venkatasubramanian | 5a20656 | 2014-07-07 19:52:37 | [diff] [blame] | 3894 | return rv; |
| 3895 | } |
| 3896 | |
| 3897 | static int buffer_size_after_time_downloaded(int64_t time_ns, double search_sec, int64_t bps, |
| 3898 | double min_buffer, double* buffer, |
| 3899 | double* sec_to_download, AVFormatContext *s, |
| 3900 | int64_t cues_start) |
| 3901 | { |
| 3902 | double nano_seconds_per_second = 1000000000.0; |
| 3903 | double time_sec = time_ns / nano_seconds_per_second; |
| 3904 | int rv = 0; |
| 3905 | int64_t time_to_search_ns = (int64_t)(search_sec * nano_seconds_per_second); |
| 3906 | int64_t end_time_ns = time_ns + time_to_search_ns; |
| 3907 | double sec_downloaded = 0.0; |
| 3908 | CueDesc desc_curr = get_cue_desc(s, time_ns, cues_start); |
| 3909 | if (desc_curr.start_time_ns == -1) |
| 3910 | return -1; |
| 3911 | *sec_to_download = 0.0; |
| 3912 | |
| 3913 | // Check for non cue start time. |
| 3914 | if (time_ns > desc_curr.start_time_ns) { |
| 3915 | int64_t cue_nano = desc_curr.end_time_ns - time_ns; |
| 3916 | double percent = (double)(cue_nano) / (desc_curr.end_time_ns - desc_curr.start_time_ns); |
| 3917 | double cueBytes = (desc_curr.end_offset - desc_curr.start_offset) * percent; |
| 3918 | double timeToDownload = (cueBytes * 8.0) / bps; |
| 3919 | |
| 3920 | sec_downloaded += (cue_nano / nano_seconds_per_second) - timeToDownload; |
| 3921 | *sec_to_download += timeToDownload; |
| 3922 | |
| 3923 | // Check if the search ends within the first cue. |
| 3924 | if (desc_curr.end_time_ns >= end_time_ns) { |
| 3925 | double desc_end_time_sec = desc_curr.end_time_ns / nano_seconds_per_second; |
| 3926 | double percent_to_sub = search_sec / (desc_end_time_sec - time_sec); |
| 3927 | sec_downloaded = percent_to_sub * sec_downloaded; |
| 3928 | *sec_to_download = percent_to_sub * *sec_to_download; |
| 3929 | } |
| 3930 | |
| 3931 | if ((sec_downloaded + *buffer) <= min_buffer) { |
| 3932 | return 1; |
| 3933 | } |
| 3934 | |
| 3935 | // Get the next Cue. |
| 3936 | desc_curr = get_cue_desc(s, desc_curr.end_time_ns, cues_start); |
| 3937 | } |
| 3938 | |
| 3939 | while (desc_curr.start_time_ns != -1) { |
| 3940 | int64_t desc_bytes = desc_curr.end_offset - desc_curr.start_offset; |
| 3941 | int64_t desc_ns = desc_curr.end_time_ns - desc_curr.start_time_ns; |
| 3942 | double desc_sec = desc_ns / nano_seconds_per_second; |
| 3943 | double bits = (desc_bytes * 8.0); |
| 3944 | double time_to_download = bits / bps; |
| 3945 | |
| 3946 | sec_downloaded += desc_sec - time_to_download; |
| 3947 | *sec_to_download += time_to_download; |
| 3948 | |
| 3949 | if (desc_curr.end_time_ns >= end_time_ns) { |
| 3950 | double desc_end_time_sec = desc_curr.end_time_ns / nano_seconds_per_second; |
| 3951 | double percent_to_sub = search_sec / (desc_end_time_sec - time_sec); |
| 3952 | sec_downloaded = percent_to_sub * sec_downloaded; |
| 3953 | *sec_to_download = percent_to_sub * *sec_to_download; |
| 3954 | |
| 3955 | if ((sec_downloaded + *buffer) <= min_buffer) |
| 3956 | rv = 1; |
| 3957 | break; |
| 3958 | } |
| 3959 | |
| 3960 | if ((sec_downloaded + *buffer) <= min_buffer) { |
| 3961 | rv = 1; |
| 3962 | break; |
| 3963 | } |
| 3964 | |
| 3965 | desc_curr = get_cue_desc(s, desc_curr.end_time_ns, cues_start); |
| 3966 | } |
| 3967 | *buffer = *buffer + sec_downloaded; |
| 3968 | return rv; |
| 3969 | } |
| 3970 | |
| 3971 | /* This function computes the bandwidth of the WebM file with the help of |
| 3972 | * buffer_size_after_time_downloaded() function. Both of these functions are |
| 3973 | * adapted from WebM Tools project and are adapted to work with FFmpeg's |
| 3974 | * Matroska parsing mechanism. |
| 3975 | * |
| 3976 | * Returns the bandwidth of the file on success; -1 on error. |
| 3977 | * */ |
| 3978 | static int64_t webm_dash_manifest_compute_bandwidth(AVFormatContext *s, int64_t cues_start) |
| 3979 | { |
| 3980 | MatroskaDemuxContext *matroska = s->priv_data; |
| 3981 | AVStream *st = s->streams[0]; |
| 3982 | double bandwidth = 0.0; |
Michael Niedermayer | e240d01 | 2014-07-15 22:06:15 | [diff] [blame] | 3983 | int i; |
| 3984 | |
| 3985 | for (i = 0; i < st->nb_index_entries; i++) { |
Vignesh Venkatasubramanian | 5a20656 | 2014-07-07 19:52:37 | [diff] [blame] | 3986 | int64_t prebuffer_ns = 1000000000; |
| 3987 | int64_t time_ns = st->index_entries[i].timestamp * matroska->time_scale; |
| 3988 | double nano_seconds_per_second = 1000000000.0; |
| 3989 | int64_t prebuffered_ns = time_ns + prebuffer_ns; |
| 3990 | double prebuffer_bytes = 0.0; |
| 3991 | int64_t temp_prebuffer_ns = prebuffer_ns; |
| 3992 | int64_t pre_bytes, pre_ns; |
| 3993 | double pre_sec, prebuffer, bits_per_second; |
| 3994 | CueDesc desc_beg = get_cue_desc(s, time_ns, cues_start); |
| 3995 | |
| 3996 | // Start with the first Cue. |
| 3997 | CueDesc desc_end = desc_beg; |
| 3998 | |
| 3999 | // Figure out how much data we have downloaded for the prebuffer. This will |
| 4000 | // be used later to adjust the bits per sample to try. |
| 4001 | while (desc_end.start_time_ns != -1 && desc_end.end_time_ns < prebuffered_ns) { |
| 4002 | // Prebuffered the entire Cue. |
| 4003 | prebuffer_bytes += desc_end.end_offset - desc_end.start_offset; |
| 4004 | temp_prebuffer_ns -= desc_end.end_time_ns - desc_end.start_time_ns; |
| 4005 | desc_end = get_cue_desc(s, desc_end.end_time_ns, cues_start); |
| 4006 | } |
| 4007 | if (desc_end.start_time_ns == -1) { |
| 4008 | // The prebuffer is larger than the duration. |
Vignesh Venkatasubramanian | b1071db | 2014-10-01 17:13:31 | [diff] [blame] | 4009 | if (matroska->duration * matroska->time_scale >= prebuffered_ns) |
| 4010 | return -1; |
| 4011 | bits_per_second = 0.0; |
| 4012 | } else { |
| 4013 | // The prebuffer ends in the last Cue. Estimate how much data was |
| 4014 | // prebuffered. |
| 4015 | pre_bytes = desc_end.end_offset - desc_end.start_offset; |
| 4016 | pre_ns = desc_end.end_time_ns - desc_end.start_time_ns; |
| 4017 | pre_sec = pre_ns / nano_seconds_per_second; |
| 4018 | prebuffer_bytes += |
| 4019 | pre_bytes * ((temp_prebuffer_ns / nano_seconds_per_second) / pre_sec); |
Vignesh Venkatasubramanian | 5a20656 | 2014-07-07 19:52:37 | [diff] [blame] | 4020 | |
Vignesh Venkatasubramanian | b1071db | 2014-10-01 17:13:31 | [diff] [blame] | 4021 | prebuffer = prebuffer_ns / nano_seconds_per_second; |
Vignesh Venkatasubramanian | 5a20656 | 2014-07-07 19:52:37 | [diff] [blame] | 4022 | |
Vignesh Venkatasubramanian | b1071db | 2014-10-01 17:13:31 | [diff] [blame] | 4023 | // Set this to 0.0 in case our prebuffer buffers the entire video. |
| 4024 | bits_per_second = 0.0; |
| 4025 | do { |
| 4026 | int64_t desc_bytes = desc_end.end_offset - desc_beg.start_offset; |
| 4027 | int64_t desc_ns = desc_end.end_time_ns - desc_beg.start_time_ns; |
| 4028 | double desc_sec = desc_ns / nano_seconds_per_second; |
| 4029 | double calc_bits_per_second = (desc_bytes * 8) / desc_sec; |
Vignesh Venkatasubramanian | 5a20656 | 2014-07-07 19:52:37 | [diff] [blame] | 4030 | |
Vignesh Venkatasubramanian | b1071db | 2014-10-01 17:13:31 | [diff] [blame] | 4031 | // Drop the bps by the percentage of bytes buffered. |
| 4032 | double percent = (desc_bytes - prebuffer_bytes) / desc_bytes; |
| 4033 | double mod_bits_per_second = calc_bits_per_second * percent; |
Vignesh Venkatasubramanian | 5a20656 | 2014-07-07 19:52:37 | [diff] [blame] | 4034 | |
Vignesh Venkatasubramanian | b1071db | 2014-10-01 17:13:31 | [diff] [blame] | 4035 | if (prebuffer < desc_sec) { |
| 4036 | double search_sec = |
| 4037 | (double)(matroska->duration * matroska->time_scale) / nano_seconds_per_second; |
Vignesh Venkatasubramanian | 5a20656 | 2014-07-07 19:52:37 | [diff] [blame] | 4038 | |
Vignesh Venkatasubramanian | b1071db | 2014-10-01 17:13:31 | [diff] [blame] | 4039 | // Add 1 so the bits per second should be a little bit greater than file |
| 4040 | // datarate. |
| 4041 | int64_t bps = (int64_t)(mod_bits_per_second) + 1; |
| 4042 | const double min_buffer = 0.0; |
| 4043 | double buffer = prebuffer; |
| 4044 | double sec_to_download = 0.0; |
Vignesh Venkatasubramanian | 5a20656 | 2014-07-07 19:52:37 | [diff] [blame] | 4045 | |
Vignesh Venkatasubramanian | b1071db | 2014-10-01 17:13:31 | [diff] [blame] | 4046 | int rv = buffer_size_after_time_downloaded(prebuffered_ns, search_sec, bps, |
| 4047 | min_buffer, &buffer, &sec_to_download, |
| 4048 | s, cues_start); |
| 4049 | if (rv < 0) { |
| 4050 | return -1; |
| 4051 | } else if (rv == 0) { |
| 4052 | bits_per_second = (double)(bps); |
| 4053 | break; |
| 4054 | } |
Vignesh Venkatasubramanian | 5a20656 | 2014-07-07 19:52:37 | [diff] [blame] | 4055 | } |
Vignesh Venkatasubramanian | 5a20656 | 2014-07-07 19:52:37 | [diff] [blame] | 4056 | |
Vignesh Venkatasubramanian | b1071db | 2014-10-01 17:13:31 | [diff] [blame] | 4057 | desc_end = get_cue_desc(s, desc_end.end_time_ns, cues_start); |
| 4058 | } while (desc_end.start_time_ns != -1); |
| 4059 | } |
Vignesh Venkatasubramanian | 5a20656 | 2014-07-07 19:52:37 | [diff] [blame] | 4060 | if (bandwidth < bits_per_second) bandwidth = bits_per_second; |
| 4061 | } |
| 4062 | return (int64_t)bandwidth; |
| 4063 | } |
| 4064 | |
Derek Buitenhuis | 6ba1c9b | 2017-04-21 15:40:35 | [diff] [blame] | 4065 | static int webm_dash_manifest_cues(AVFormatContext *s, int64_t init_range) |
Vignesh Venkatasubramanian | 5a20656 | 2014-07-07 19:52:37 | [diff] [blame] | 4066 | { |
| 4067 | MatroskaDemuxContext *matroska = s->priv_data; |
| 4068 | EbmlList *seekhead_list = &matroska->seekhead; |
| 4069 | MatroskaSeekhead *seekhead = seekhead_list->elem; |
| 4070 | char *buf; |
Vignesh Venkatasubramanian | 080acf7 | 2014-08-25 16:15:13 | [diff] [blame] | 4071 | int64_t cues_start = -1, cues_end = -1, before_pos, bandwidth; |
Vignesh Venkatasubramanian | 5a20656 | 2014-07-07 19:52:37 | [diff] [blame] | 4072 | int i; |
Derek Buitenhuis | 8e6b9ef | 2017-04-20 14:02:54 | [diff] [blame] | 4073 | int end = 0; |
Vignesh Venkatasubramanian | 5a20656 | 2014-07-07 19:52:37 | [diff] [blame] | 4074 | |
| 4075 | // determine cues start and end positions |
| 4076 | for (i = 0; i < seekhead_list->nb_elem; i++) |
| 4077 | if (seekhead[i].id == MATROSKA_ID_CUES) |
| 4078 | break; |
| 4079 | |
| 4080 | if (i >= seekhead_list->nb_elem) return -1; |
| 4081 | |
| 4082 | before_pos = avio_tell(matroska->ctx->pb); |
| 4083 | cues_start = seekhead[i].pos + matroska->segment_start; |
| 4084 | if (avio_seek(matroska->ctx->pb, cues_start, SEEK_SET) == cues_start) { |
Vignesh Venkatasubramanian | 8acb765 | 2014-10-01 17:13:30 | [diff] [blame] | 4085 | // cues_end is computed as cues_start + cues_length + length of the |
Andreas Rheinhardt | a27e539 | 2019-05-16 22:29:54 | [diff] [blame] | 4086 | // Cues element ID (i.e. 4) + EBML length of the Cues element. |
| 4087 | // cues_end is inclusive and the above sum is reduced by 1. |
| 4088 | uint64_t cues_length, cues_id; |
| 4089 | int bytes_read; |
Andreas Rheinhardt | 239c736 | 2019-06-25 01:08:56 | [diff] [blame] | 4090 | bytes_read = ebml_read_num (matroska, matroska->ctx->pb, 4, &cues_id, 1); |
Andreas Rheinhardt | a27e539 | 2019-05-16 22:29:54 | [diff] [blame] | 4091 | if (bytes_read < 0 || cues_id != (MATROSKA_ID_CUES & 0xfffffff)) |
| 4092 | return bytes_read < 0 ? bytes_read : AVERROR_INVALIDDATA; |
| 4093 | bytes_read = ebml_read_length(matroska, matroska->ctx->pb, &cues_length); |
| 4094 | if (bytes_read < 0) |
| 4095 | return bytes_read; |
| 4096 | cues_end = cues_start + 4 + bytes_read + cues_length - 1; |
Vignesh Venkatasubramanian | 5a20656 | 2014-07-07 19:52:37 | [diff] [blame] | 4097 | } |
| 4098 | avio_seek(matroska->ctx->pb, before_pos, SEEK_SET); |
Vignesh Venkatasubramanian | 080acf7 | 2014-08-25 16:15:13 | [diff] [blame] | 4099 | if (cues_start == -1 || cues_end == -1) return -1; |
Vignesh Venkatasubramanian | 5a20656 | 2014-07-07 19:52:37 | [diff] [blame] | 4100 | |
| 4101 | // parse the cues |
| 4102 | matroska_parse_cues(matroska); |
| 4103 | |
| 4104 | // cues start |
Reimar Döffinger | a0941c8 | 2014-07-29 19:10:39 | [diff] [blame] | 4105 | av_dict_set_int(&s->streams[0]->metadata, CUES_START, cues_start, 0); |
Vignesh Venkatasubramanian | 5a20656 | 2014-07-07 19:52:37 | [diff] [blame] | 4106 | |
| 4107 | // cues end |
Reimar Döffinger | a0941c8 | 2014-07-29 19:10:39 | [diff] [blame] | 4108 | av_dict_set_int(&s->streams[0]->metadata, CUES_END, cues_end, 0); |
Vignesh Venkatasubramanian | 5a20656 | 2014-07-07 19:52:37 | [diff] [blame] | 4109 | |
Andreas Rheinhardt | 60f75c9 | 2019-05-16 22:30:13 | [diff] [blame] | 4110 | // if the file has cues at the start, fix up the init range so that |
Derek Buitenhuis | 6ba1c9b | 2017-04-21 15:40:35 | [diff] [blame] | 4111 | // it does not include it |
| 4112 | if (cues_start <= init_range) |
| 4113 | av_dict_set_int(&s->streams[0]->metadata, INITIALIZATION_RANGE, cues_start - 1, 0); |
| 4114 | |
Vignesh Venkatasubramanian | 5a20656 | 2014-07-07 19:52:37 | [diff] [blame] | 4115 | // bandwidth |
| 4116 | bandwidth = webm_dash_manifest_compute_bandwidth(s, cues_start); |
| 4117 | if (bandwidth < 0) return -1; |
Reimar Döffinger | a0941c8 | 2014-07-29 19:10:39 | [diff] [blame] | 4118 | av_dict_set_int(&s->streams[0]->metadata, BANDWIDTH, bandwidth, 0); |
Vignesh Venkatasubramanian | 5a20656 | 2014-07-07 19:52:37 | [diff] [blame] | 4119 | |
| 4120 | // check if all clusters start with key frames |
Reimar Döffinger | a0941c8 | 2014-07-29 19:10:39 | [diff] [blame] | 4121 | av_dict_set_int(&s->streams[0]->metadata, CLUSTER_KEYFRAME, webm_clusters_start_with_keyframe(s), 0); |
Vignesh Venkatasubramanian | 5a20656 | 2014-07-07 19:52:37 | [diff] [blame] | 4122 | |
| 4123 | // store cue point timestamps as a comma separated list for checking subsegment alignment in |
| 4124 | // the muxer. assumes that each timestamp cannot be more than 20 characters long. |
Carl Eugen Hoyos | 6871c17 | 2018-09-03 12:18:02 | [diff] [blame] | 4125 | buf = av_malloc_array(s->streams[0]->nb_index_entries, 20); |
Vignesh Venkatasubramanian | 5a20656 | 2014-07-07 19:52:37 | [diff] [blame] | 4126 | if (!buf) return -1; |
| 4127 | strcpy(buf, ""); |
| 4128 | for (i = 0; i < s->streams[0]->nb_index_entries; i++) { |
Carl Eugen Hoyos | 6871c17 | 2018-09-03 12:18:02 | [diff] [blame] | 4129 | int ret = snprintf(buf + end, 20, |
Carl Eugen Hoyos | 73abde6 | 2018-12-10 00:55:15 | [diff] [blame] | 4130 | "%" PRId64"%s", s->streams[0]->index_entries[i].timestamp, |
| 4131 | i != s->streams[0]->nb_index_entries - 1 ? "," : ""); |
Derek Buitenhuis | 8e6b9ef | 2017-04-20 14:02:54 | [diff] [blame] | 4132 | if (ret <= 0 || (ret == 20 && i == s->streams[0]->nb_index_entries - 1)) { |
| 4133 | av_log(s, AV_LOG_ERROR, "timestamp too long.\n"); |
Steven Liu | cc25a88 | 2017-05-07 03:29:08 | [diff] [blame] | 4134 | av_free(buf); |
Derek Buitenhuis | 8e6b9ef | 2017-04-20 14:02:54 | [diff] [blame] | 4135 | return AVERROR_INVALIDDATA; |
| 4136 | } |
| 4137 | end += ret; |
Vignesh Venkatasubramanian | 5a20656 | 2014-07-07 19:52:37 | [diff] [blame] | 4138 | } |
Andreas Rheinhardt | 40d9cbd | 2019-11-10 04:07:31 | [diff] [blame] | 4139 | av_dict_set(&s->streams[0]->metadata, CUE_TIMESTAMPS, |
| 4140 | buf, AV_DICT_DONT_STRDUP_VAL); |
Vignesh Venkatasubramanian | 5a20656 | 2014-07-07 19:52:37 | [diff] [blame] | 4141 | |
| 4142 | return 0; |
| 4143 | } |
| 4144 | |
| 4145 | static int webm_dash_manifest_read_header(AVFormatContext *s) |
| 4146 | { |
| 4147 | char *buf; |
| 4148 | int ret = matroska_read_header(s); |
Derek Buitenhuis | 6ba1c9b | 2017-04-21 15:40:35 | [diff] [blame] | 4149 | int64_t init_range; |
Vignesh Venkatasubramanian | 5a20656 | 2014-07-07 19:52:37 | [diff] [blame] | 4150 | MatroskaTrack *tracks; |
| 4151 | MatroskaDemuxContext *matroska = s->priv_data; |
| 4152 | if (ret) { |
| 4153 | av_log(s, AV_LOG_ERROR, "Failed to read file headers\n"); |
| 4154 | return -1; |
| 4155 | } |
Andreas Cadhalpun | ff100c9 | 2016-11-07 23:42:23 | [diff] [blame] | 4156 | if (!s->nb_streams) { |
| 4157 | matroska_read_close(s); |
| 4158 | av_log(s, AV_LOG_ERROR, "No streams found\n"); |
| 4159 | return AVERROR_INVALIDDATA; |
| 4160 | } |
Vignesh Venkatasubramanian | 5a20656 | 2014-07-07 19:52:37 | [diff] [blame] | 4161 | |
Vignesh Venkatasubramanian | 2171b7c | 2015-03-31 23:51:57 | [diff] [blame] | 4162 | if (!matroska->is_live) { |
| 4163 | buf = av_asprintf("%g", matroska->duration); |
| 4164 | if (!buf) return AVERROR(ENOMEM); |
Andreas Rheinhardt | 40d9cbd | 2019-11-10 04:07:31 | [diff] [blame] | 4165 | av_dict_set(&s->streams[0]->metadata, DURATION, |
| 4166 | buf, AV_DICT_DONT_STRDUP_VAL); |
Vignesh Venkatasubramanian | 2171b7c | 2015-03-31 23:51:57 | [diff] [blame] | 4167 | |
| 4168 | // initialization range |
| 4169 | // 5 is the offset of Cluster ID. |
Derek Buitenhuis | 6ba1c9b | 2017-04-21 15:40:35 | [diff] [blame] | 4170 | init_range = avio_tell(s->pb) - 5; |
| 4171 | av_dict_set_int(&s->streams[0]->metadata, INITIALIZATION_RANGE, init_range, 0); |
Vignesh Venkatasubramanian | 2171b7c | 2015-03-31 23:51:57 | [diff] [blame] | 4172 | } |
Vignesh Venkatasubramanian | 5a20656 | 2014-07-07 19:52:37 | [diff] [blame] | 4173 | |
| 4174 | // basename of the file |
Marton Balint | 18ac642 | 2017-12-29 22:30:14 | [diff] [blame] | 4175 | buf = strrchr(s->url, '/'); |
| 4176 | av_dict_set(&s->streams[0]->metadata, FILENAME, buf ? ++buf : s->url, 0); |
Vignesh Venkatasubramanian | 5a20656 | 2014-07-07 19:52:37 | [diff] [blame] | 4177 | |
Vignesh Venkatasubramanian | 5a20656 | 2014-07-07 19:52:37 | [diff] [blame] | 4178 | // track number |
| 4179 | tracks = matroska->tracks.elem; |
Reimar Döffinger | a0941c8 | 2014-07-29 19:10:39 | [diff] [blame] | 4180 | av_dict_set_int(&s->streams[0]->metadata, TRACK_NUMBER, tracks[0].num, 0); |
Vignesh Venkatasubramanian | 5a20656 | 2014-07-07 19:52:37 | [diff] [blame] | 4181 | |
| 4182 | // parse the cues and populate Cue related fields |
Vignesh Venkatasubramanian | 62c27fd | 2017-04-12 04:33:28 | [diff] [blame] | 4183 | if (!matroska->is_live) { |
Derek Buitenhuis | 6ba1c9b | 2017-04-21 15:40:35 | [diff] [blame] | 4184 | ret = webm_dash_manifest_cues(s, init_range); |
Vignesh Venkatasubramanian | 62c27fd | 2017-04-12 04:33:28 | [diff] [blame] | 4185 | if (ret < 0) { |
| 4186 | av_log(s, AV_LOG_ERROR, "Error parsing Cues\n"); |
| 4187 | return ret; |
| 4188 | } |
| 4189 | } |
| 4190 | |
| 4191 | // use the bandwidth from the command line if it was provided |
| 4192 | if (matroska->bandwidth > 0) { |
| 4193 | av_dict_set_int(&s->streams[0]->metadata, BANDWIDTH, |
| 4194 | matroska->bandwidth, 0); |
| 4195 | } |
| 4196 | return 0; |
Vignesh Venkatasubramanian | 5a20656 | 2014-07-07 19:52:37 | [diff] [blame] | 4197 | } |
| 4198 | |
| 4199 | static int webm_dash_manifest_read_packet(AVFormatContext *s, AVPacket *pkt) |
| 4200 | { |
| 4201 | return AVERROR_EOF; |
| 4202 | } |
| 4203 | |
Vignesh Venkatasubramanian | 2171b7c | 2015-03-31 23:51:57 | [diff] [blame] | 4204 | #define OFFSET(x) offsetof(MatroskaDemuxContext, x) |
| 4205 | static const AVOption options[] = { |
Clément Bœsch | 43ecec0 | 2015-11-21 21:05:07 | [diff] [blame] | 4206 | { "live", "flag indicating that the input is a live file that only has the headers.", OFFSET(is_live), AV_OPT_TYPE_BOOL, {.i64 = 0}, 0, 1, AV_OPT_FLAG_DECODING_PARAM }, |
Vignesh Venkatasubramanian | 62c27fd | 2017-04-12 04:33:28 | [diff] [blame] | 4207 | { "bandwidth", "bandwidth of this stream to be specified in the DASH manifest.", OFFSET(bandwidth), AV_OPT_TYPE_INT, {.i64 = 0}, 0, INT_MAX, AV_OPT_FLAG_DECODING_PARAM }, |
Vignesh Venkatasubramanian | 2171b7c | 2015-03-31 23:51:57 | [diff] [blame] | 4208 | { NULL }, |
| 4209 | }; |
| 4210 | |
| 4211 | static const AVClass webm_dash_class = { |
| 4212 | .class_name = "WebM DASH Manifest demuxer", |
| 4213 | .item_name = av_default_item_name, |
| 4214 | .option = options, |
| 4215 | .version = LIBAVUTIL_VERSION_INT, |
| 4216 | }; |
| 4217 | |
Diego Elio Pettenò | 66355be | 2011-01-25 22:03:28 | [diff] [blame] | 4218 | AVInputFormat ff_matroska_demuxer = { |
Anton Khirnov | dfc2c4d | 2011-07-16 20:18:12 | [diff] [blame] | 4219 | .name = "matroska,webm", |
Diego Biurrun | 6774247 | 2012-07-24 21:51:41 | [diff] [blame] | 4220 | .long_name = NULL_IF_CONFIG_SMALL("Matroska / WebM"), |
Vittorio Giovara | f2583bc | 2014-07-23 08:49:24 | [diff] [blame] | 4221 | .extensions = "mkv,mk3d,mka,mks", |
Anton Khirnov | dfc2c4d | 2011-07-16 20:18:12 | [diff] [blame] | 4222 | .priv_data_size = sizeof(MatroskaDemuxContext), |
| 4223 | .read_probe = matroska_probe, |
| 4224 | .read_header = matroska_read_header, |
| 4225 | .read_packet = matroska_read_packet, |
| 4226 | .read_close = matroska_read_close, |
| 4227 | .read_seek = matroska_read_seek, |
Luca Barbato | fa38573 | 2014-03-13 21:14:43 | [diff] [blame] | 4228 | .mime_type = "audio/webm,audio/x-matroska,video/webm,video/x-matroska" |
David Conrad | b061d89 | 2007-06-04 22:10:54 | [diff] [blame] | 4229 | }; |
Vignesh Venkatasubramanian | 5a20656 | 2014-07-07 19:52:37 | [diff] [blame] | 4230 | |
| 4231 | AVInputFormat ff_webm_dash_manifest_demuxer = { |
| 4232 | .name = "webm_dash_manifest", |
| 4233 | .long_name = NULL_IF_CONFIG_SMALL("WebM DASH Manifest"), |
| 4234 | .priv_data_size = sizeof(MatroskaDemuxContext), |
| 4235 | .read_header = webm_dash_manifest_read_header, |
| 4236 | .read_packet = webm_dash_manifest_read_packet, |
| 4237 | .read_close = matroska_read_close, |
Vignesh Venkatasubramanian | 2171b7c | 2015-03-31 23:51:57 | [diff] [blame] | 4238 | .priv_class = &webm_dash_class, |
Vignesh Venkatasubramanian | 5a20656 | 2014-07-07 19:52:37 | [diff] [blame] | 4239 | }; |