]> git.sesse.net Git - vlc/blob - modules/demux/mkv/mkv.hpp
Prevent a potential integer overflow (mkv).
[vlc] / modules / demux / mkv / mkv.hpp
1 /*****************************************************************************
2  * mkv.cpp : matroska demuxer
3  *****************************************************************************
4  * Copyright (C) 2003-2004 the VideoLAN team
5  * $Id$
6  *
7  * Authors: Laurent Aimar <fenrir@via.ecp.fr>
8  *          Steve Lhomme <steve.lhomme@free.fr>
9  *
10  * This program is free software; you can redistribute it and/or modify
11  * it under the terms of the GNU General Public License as published by
12  * the Free Software Foundation; either version 2 of the License, or
13  * (at your option) any later version.
14  *
15  * This program is distributed in the hope that it will be useful,
16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
18  * GNU General Public License for more details.
19  *
20  * You should have received a copy of the GNU General Public License
21  * along with this program; if not, write to the Free Software
22  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
23  *****************************************************************************/
24
25 #ifndef _MKV_H_
26 #define _MKV_H_
27
28 /*****************************************************************************
29  * Preamble
30  *****************************************************************************/
31
32
33 /* config.h may include inttypes.h, so make sure we define that option
34  * early enough. */
35 #define __STDC_FORMAT_MACROS 1
36 #define __STDC_CONSTANT_MACROS 1
37 #define __STDC_LIMIT_MACROS 1
38
39 #ifdef HAVE_CONFIG_H
40 # include "config.h"
41 #endif
42
43 #include <inttypes.h>
44
45 #include <vlc_common.h>
46 #include <vlc_plugin.h>
47
48 #ifdef HAVE_TIME_H
49 #   include <time.h>                                               /* time() */
50 #endif
51
52
53 #include <vlc_codecs.h>               /* BITMAPINFOHEADER, WAVEFORMATEX */
54 #include <vlc_iso_lang.h>
55 #include <vlc_meta.h>
56 #include <vlc_charset.h>
57 #include <vlc_input.h>
58 #include <vlc_demux.h>
59
60 #include <iostream>
61 #include <cassert>
62 #include <typeinfo>
63 #include <string>
64 #include <vector>
65 #include <algorithm>
66
67 #ifdef HAVE_DIRENT_H
68 #   include <dirent.h>
69 #endif
70
71 /* libebml and matroska */
72 #include "ebml/EbmlHead.h"
73 #include "ebml/EbmlSubHead.h"
74 #include "ebml/EbmlStream.h"
75 #include "ebml/EbmlContexts.h"
76 #include "ebml/EbmlVoid.h"
77 #include "ebml/EbmlVersion.h"
78 #include "ebml/StdIOCallback.h"
79
80 #include "matroska/KaxAttachments.h"
81 #include "matroska/KaxAttached.h"
82 #include "matroska/KaxBlock.h"
83 #include "matroska/KaxBlockData.h"
84 #include "matroska/KaxChapters.h"
85 #include "matroska/KaxCluster.h"
86 #include "matroska/KaxClusterData.h"
87 #include "matroska/KaxContexts.h"
88 #include "matroska/KaxCues.h"
89 #include "matroska/KaxCuesData.h"
90 #include "matroska/KaxInfo.h"
91 #include "matroska/KaxInfoData.h"
92 #include "matroska/KaxSeekHead.h"
93 #include "matroska/KaxSegment.h"
94 #include "matroska/KaxTag.h"
95 #include "matroska/KaxTags.h"
96 #include "matroska/KaxTagMulti.h"
97 #include "matroska/KaxTracks.h"
98 #include "matroska/KaxTrackAudio.h"
99 #include "matroska/KaxTrackVideo.h"
100 #include "matroska/KaxTrackEntryData.h"
101 #include "matroska/KaxContentEncoding.h"
102 #include "matroska/KaxVersion.h"
103
104 #include "ebml/StdIOCallback.h"
105
106 #include <vlc_keys.h>
107
108 extern "C" {
109    #include "../mp4/libmp4.h"
110 }
111 #ifdef HAVE_ZLIB_H
112 #   include <zlib.h>
113 #endif
114
115 #define MATROSKA_COMPRESSION_NONE  -1
116 #define MATROSKA_COMPRESSION_ZLIB   0
117 #define MATROSKA_COMPRESSION_BLIB   1
118 #define MATROSKA_COMPRESSION_LZOX   2
119 #define MATROSKA_COMPRESSION_HEADER 3
120
121 #define MKVD_TIMECODESCALE 1000000
122
123 #define MKV_IS_ID( el, C ) ( EbmlId( (*el) ) == C::ClassInfos.GlobalId )
124
125
126 using namespace LIBMATROSKA_NAMESPACE;
127 using namespace std;
128
129 class attachment_c
130 {
131 public:
132     attachment_c()
133         :p_data(NULL)
134         ,i_size(0)
135     {}
136     virtual ~attachment_c()
137     {
138         free( p_data );
139     }
140
141     std::string    psz_file_name;
142     std::string    psz_mime_type;
143     void          *p_data;
144     int            i_size;
145 };
146
147 class matroska_segment_c;
148
149 class matroska_stream_c
150 {
151 public:
152     matroska_stream_c( demux_sys_t & demuxer )
153         :p_in(NULL)
154         ,p_es(NULL)
155         ,sys(demuxer)
156     {}
157
158     virtual ~matroska_stream_c()
159     {
160         delete p_in;
161         delete p_es;
162     }
163
164     IOCallback         *p_in;
165     EbmlStream         *p_es;
166
167     std::vector<matroska_segment_c*> segments;
168
169     demux_sys_t                      & sys;
170 };
171
172
173 /*****************************************************************************
174  * definitions of structures and functions used by this plugins
175  *****************************************************************************/
176 typedef struct
177 {
178 //    ~mkv_track_t();
179
180     bool         b_default;
181     bool         b_enabled;
182     unsigned int i_number;
183
184     int          i_extra_data;
185     uint8_t      *p_extra_data;
186
187     char         *psz_codec;
188     bool         b_dts_only;
189
190     uint64_t     i_default_duration;
191     float        f_timecodescale;
192     mtime_t      i_last_dts;
193
194     /* video */
195     es_format_t fmt;
196     float       f_fps;
197     es_out_id_t *p_es;
198
199     /* audio */
200     unsigned int i_original_rate;
201
202     bool            b_inited;
203     /* data to be send first */
204     int             i_data_init;
205     uint8_t         *p_data_init;
206
207     /* hack : it's for seek */
208     bool            b_search_keyframe;
209     bool            b_silent;
210
211     /* informative */
212     const char   *psz_codec_name;
213     const char   *psz_codec_settings;
214     const char   *psz_codec_info_url;
215     const char   *psz_codec_download_url;
216
217     /* encryption/compression */
218     int                    i_compression_type;
219     KaxContentCompSettings *p_compression_data;
220
221 } mkv_track_t;
222
223 typedef struct
224 {
225     int     i_track;
226     int     i_block_number;
227
228     int64_t i_position;
229     int64_t i_time;
230
231     bool       b_key;
232 } mkv_index_t;
233
234
235 #endif /* _MKV_HPP_ */