Line | Branch | Exec | Source |
---|---|---|---|
1 | /* | ||
2 | * AVI demuxer | ||
3 | * Copyright (c) 2001 Fabrice Bellard | ||
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 | #include "config_components.h" | ||
23 | |||
24 | #include <inttypes.h> | ||
25 | |||
26 | #include "libavutil/avassert.h" | ||
27 | #include "libavutil/avstring.h" | ||
28 | #include "libavutil/mem.h" | ||
29 | #include "libavutil/opt.h" | ||
30 | #include "libavutil/dict.h" | ||
31 | #include "libavutil/integer.h" | ||
32 | #include "libavutil/internal.h" | ||
33 | #include "libavutil/intreadwrite.h" | ||
34 | #include "libavutil/mathematics.h" | ||
35 | #include "avformat.h" | ||
36 | #include "avi.h" | ||
37 | #include "demux.h" | ||
38 | #include "dv.h" | ||
39 | #include "internal.h" | ||
40 | #include "isom.h" | ||
41 | #include "riff.h" | ||
42 | #include "libavcodec/bytestream.h" | ||
43 | #include "libavcodec/exif.h" | ||
44 | #include "libavcodec/startcode.h" | ||
45 | |||
46 | typedef struct AVIStream { | ||
47 | int64_t frame_offset; /* current frame (video) or byte (audio) counter | ||
48 | * (used to compute the pts) */ | ||
49 | int remaining; | ||
50 | int packet_size; | ||
51 | |||
52 | uint32_t handler; | ||
53 | uint32_t scale; | ||
54 | uint32_t rate; | ||
55 | int sample_size; /* size of one sample (or packet) | ||
56 | * (in the rate/scale sense) in bytes */ | ||
57 | |||
58 | int64_t cum_len; /* temporary storage (used during seek) */ | ||
59 | int prefix; /* normally 'd'<<8 + 'c' or 'w'<<8 + 'b' */ | ||
60 | int prefix_count; | ||
61 | uint32_t pal[256]; | ||
62 | int has_pal; | ||
63 | int dshow_block_align; /* block align variable used to emulate bugs in | ||
64 | * the MS dshow demuxer */ | ||
65 | |||
66 | AVFormatContext *sub_ctx; | ||
67 | AVPacket *sub_pkt; | ||
68 | AVBufferRef *sub_buffer; | ||
69 | |||
70 | int64_t seek_pos; | ||
71 | } AVIStream; | ||
72 | |||
73 | typedef struct AVIContext { | ||
74 | const AVClass *class; | ||
75 | int64_t riff_end; | ||
76 | int64_t movi_end; | ||
77 | int64_t fsize; | ||
78 | int64_t io_fsize; | ||
79 | int64_t movi_list; | ||
80 | int64_t last_pkt_pos; | ||
81 | int index_loaded; | ||
82 | int is_odml; | ||
83 | int non_interleaved; | ||
84 | int stream_index; | ||
85 | DVDemuxContext *dv_demux; | ||
86 | int odml_depth; | ||
87 | int64_t odml_read; | ||
88 | int64_t odml_max_pos; | ||
89 | int use_odml; | ||
90 | #define MAX_ODML_DEPTH 1000 | ||
91 | int64_t dts_max; | ||
92 | } AVIContext; | ||
93 | |||
94 | |||
95 | static const AVOption options[] = { | ||
96 | { "use_odml", "use odml index", offsetof(AVIContext, use_odml), AV_OPT_TYPE_BOOL, {.i64 = 1}, -1, 1, AV_OPT_FLAG_DECODING_PARAM}, | ||
97 | { NULL }, | ||
98 | }; | ||
99 | |||
100 | static const AVClass demuxer_class = { | ||
101 | .class_name = "avi", | ||
102 | .item_name = av_default_item_name, | ||
103 | .option = options, | ||
104 | .version = LIBAVUTIL_VERSION_INT, | ||
105 | .category = AV_CLASS_CATEGORY_DEMUXER, | ||
106 | }; | ||
107 | |||
108 | |||
109 | static const char avi_headers[][8] = { | ||
110 | { 'R', 'I', 'F', 'F', 'A', 'V', 'I', ' ' }, | ||
111 | { 'R', 'I', 'F', 'F', 'A', 'V', 'I', 'X' }, | ||
112 | { 'R', 'I', 'F', 'F', 'A', 'V', 'I', 0x19 }, | ||
113 | { 'O', 'N', '2', ' ', 'O', 'N', '2', 'f' }, | ||
114 | { 'R', 'I', 'F', 'F', 'A', 'M', 'V', ' ' }, | ||
115 | { 0 } | ||
116 | }; | ||
117 | |||
118 | static const AVMetadataConv avi_metadata_conv[] = { | ||
119 | { "strn", "title" }, | ||
120 | { "isbj", "subject" }, | ||
121 | { "inam", "title" }, | ||
122 | { "iart", "artist" }, | ||
123 | { "icop", "copyright" }, | ||
124 | { "icmt", "comment" }, | ||
125 | { "ignr", "genre" }, | ||
126 | { "iprd", "product" }, | ||
127 | { "isft", "software" }, | ||
128 | |||
129 | { 0 }, | ||
130 | }; | ||
131 | |||
132 | static int avi_load_index(AVFormatContext *s); | ||
133 | static int guess_ni_flag(AVFormatContext *s); | ||
134 | |||
135 | #define print_tag(s, str, tag, size) \ | ||
136 | av_log(s, AV_LOG_TRACE, "pos:%"PRIX64" %s: tag=%s size=0x%x\n", \ | ||
137 | avio_tell(pb), str, av_fourcc2str(tag), size) \ | ||
138 | |||
139 | 63661 | static inline int get_duration(AVIStream *ast, int len) | |
140 | { | ||
141 |
2/2✓ Branch 0 taken 9247 times.
✓ Branch 1 taken 54414 times.
|
63661 | if (ast->sample_size) |
142 | 9247 | return len; | |
143 |
2/2✓ Branch 0 taken 212 times.
✓ Branch 1 taken 54202 times.
|
54414 | else if (ast->dshow_block_align) |
144 | 212 | return (len + (int64_t)ast->dshow_block_align - 1) / ast->dshow_block_align; | |
145 | else | ||
146 | 54202 | return 1; | |
147 | } | ||
148 | |||
149 | 468 | static int get_riff(AVFormatContext *s, AVIOContext *pb) | |
150 | { | ||
151 | 468 | AVIContext *avi = s->priv_data; | |
152 | 468 | char header[8] = {0}; | |
153 | int i; | ||
154 | |||
155 | /* check RIFF header */ | ||
156 | 468 | avio_read(pb, header, 4); | |
157 | 468 | avi->riff_end = avio_rl32(pb); /* RIFF chunk size */ | |
158 | 468 | avi->riff_end += avio_tell(pb); /* RIFF chunk end */ | |
159 | 468 | avio_read(pb, header + 4, 4); | |
160 | |||
161 |
1/2✓ Branch 0 taken 476 times.
✗ Branch 1 not taken.
|
476 | for (i = 0; avi_headers[i][0]; i++) |
162 |
2/2✓ Branch 0 taken 468 times.
✓ Branch 1 taken 8 times.
|
476 | if (!memcmp(header, avi_headers[i], 8)) |
163 | 468 | break; | |
164 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 468 times.
|
468 | if (!avi_headers[i][0]) |
165 | ✗ | return AVERROR_INVALIDDATA; | |
166 | |||
167 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 468 times.
|
468 | if (header[7] == 0x19) |
168 | ✗ | av_log(s, AV_LOG_INFO, | |
169 | "This file has been generated by a totally broken muxer.\n"); | ||
170 | |||
171 | 468 | return 0; | |
172 | } | ||
173 | |||
174 | 40 | static int read_odml_index(AVFormatContext *s, int64_t frame_num) | |
175 | { | ||
176 | 40 | AVIContext *avi = s->priv_data; | |
177 | 40 | AVIOContext *pb = s->pb; | |
178 | 40 | int longs_per_entry = avio_rl16(pb); | |
179 | 40 | int index_sub_type = avio_r8(pb); | |
180 | 40 | int index_type = avio_r8(pb); | |
181 | 40 | int entries_in_use = avio_rl32(pb); | |
182 | 40 | int chunk_id = avio_rl32(pb); | |
183 | 40 | int64_t base = avio_rl64(pb); | |
184 | 40 | int stream_id = ((chunk_id & 0xFF) - '0') * 10 + | |
185 | 40 | ((chunk_id >> 8 & 0xFF) - '0'); | |
186 | AVStream *st; | ||
187 | AVIStream *ast; | ||
188 | int i; | ||
189 | 40 | int64_t last_pos = -1; | |
190 | 40 | int64_t filesize = avi->fsize; | |
191 | |||
192 | 40 | av_log(s, AV_LOG_TRACE, | |
193 | "longs_per_entry:%d index_type:%d entries_in_use:%d " | ||
194 | "chunk_id:%X base:%16"PRIX64" frame_num:%"PRId64"\n", | ||
195 | longs_per_entry, | ||
196 | index_type, | ||
197 | entries_in_use, | ||
198 | chunk_id, | ||
199 | base, | ||
200 | frame_num); | ||
201 | |||
202 |
3/4✓ Branch 0 taken 29 times.
✓ Branch 1 taken 11 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 29 times.
|
40 | if (stream_id >= s->nb_streams || stream_id < 0) |
203 | 11 | return AVERROR_INVALIDDATA; | |
204 | 29 | st = s->streams[stream_id]; | |
205 | 29 | ast = st->priv_data; | |
206 | |||
207 |
2/4✓ Branch 0 taken 29 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 29 times.
|
29 | if (index_sub_type || entries_in_use < 0) |
208 | ✗ | return AVERROR_INVALIDDATA; | |
209 | |||
210 | 29 | avio_rl32(pb); | |
211 | |||
212 |
3/4✓ Branch 0 taken 9 times.
✓ Branch 1 taken 20 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 9 times.
|
29 | if (index_type && longs_per_entry != 2) |
213 | ✗ | return AVERROR_INVALIDDATA; | |
214 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 29 times.
|
29 | if (index_type > 1) |
215 | ✗ | return AVERROR_INVALIDDATA; | |
216 | |||
217 |
2/4✓ Branch 0 taken 29 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 29 times.
|
29 | if (filesize > 0 && base >= filesize) { |
218 | ✗ | av_log(s, AV_LOG_ERROR, "ODML index invalid\n"); | |
219 | ✗ | if (base >> 32 == (base & 0xFFFFFFFF) && | |
220 | ✗ | (base & 0xFFFFFFFF) < filesize && | |
221 | filesize <= 0xFFFFFFFF) | ||
222 | ✗ | base &= 0xFFFFFFFF; | |
223 | else | ||
224 | ✗ | return AVERROR_INVALIDDATA; | |
225 | } | ||
226 | |||
227 |
2/2✓ Branch 0 taken 5695 times.
✓ Branch 1 taken 18 times.
|
5713 | for (i = 0; i < entries_in_use; i++) { |
228 |
2/2✓ Branch 1 taken 3 times.
✓ Branch 2 taken 5692 times.
|
5695 | avi->odml_max_pos = FFMAX(avi->odml_max_pos, avio_tell(pb)); |
229 | |||
230 | // If we read more than there are bytes then we must have been reading something twice | ||
231 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 5695 times.
|
5695 | if (avi->odml_read > avi->odml_max_pos) |
232 | ✗ | return AVERROR_INVALIDDATA; | |
233 | |||
234 |
2/2✓ Branch 0 taken 5675 times.
✓ Branch 1 taken 20 times.
|
5695 | if (index_type) { |
235 | 5675 | int64_t pos = avio_rl32(pb) + base - 8; | |
236 | 5675 | int len = avio_rl32(pb); | |
237 | 5675 | int key = len >= 0; | |
238 | 5675 | len &= 0x7FFFFFFF; | |
239 | 5675 | avi->odml_read += 8; | |
240 | |||
241 | 5675 | av_log(s, AV_LOG_TRACE, "pos:%"PRId64", len:%X\n", pos, len); | |
242 | |||
243 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 5675 times.
|
5675 | if (avio_feof(pb)) |
244 | ✗ | return AVERROR_INVALIDDATA; | |
245 | |||
246 |
3/4✓ Branch 0 taken 5675 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 2 times.
✓ Branch 3 taken 5673 times.
|
5675 | if (last_pos == pos || pos == base - 8) |
247 | 2 | avi->non_interleaved = 1; | |
248 |
3/4✓ Branch 0 taken 5675 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 5457 times.
✓ Branch 3 taken 218 times.
|
5675 | if (last_pos != pos && len) |
249 | 5457 | av_add_index_entry(st, pos, ast->cum_len, len, 0, | |
250 | key ? AVINDEX_KEYFRAME : 0); | ||
251 | |||
252 | 5675 | ast->cum_len += get_duration(ast, len); | |
253 | 5675 | last_pos = pos; | |
254 | } else { | ||
255 | int64_t offset, pos; | ||
256 | int duration; | ||
257 | int ret; | ||
258 | 20 | avi->odml_read += 16; | |
259 | |||
260 | 20 | offset = avio_rl64(pb); | |
261 | 20 | avio_rl32(pb); /* size */ | |
262 | 20 | duration = avio_rl32(pb); | |
263 | |||
264 |
2/4✓ Branch 1 taken 20 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✓ Branch 4 taken 20 times.
|
20 | if (avio_feof(pb) || offset > INT64_MAX - 8) |
265 | ✗ | return AVERROR_INVALIDDATA; | |
266 | |||
267 | 20 | pos = avio_tell(pb); | |
268 | |||
269 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 20 times.
|
20 | if (avi->odml_depth > MAX_ODML_DEPTH) { |
270 | ✗ | av_log(s, AV_LOG_ERROR, "Too deeply nested ODML indexes\n"); | |
271 | ✗ | return AVERROR_INVALIDDATA; | |
272 | } | ||
273 | |||
274 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 20 times.
|
20 | if (avio_seek(pb, offset + 8, SEEK_SET) < 0) |
275 | ✗ | return -1; | |
276 | 20 | avi->odml_depth++; | |
277 | 20 | ret = read_odml_index(s, frame_num); | |
278 | 20 | avi->odml_depth--; | |
279 | 20 | frame_num += duration; | |
280 | |||
281 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 20 times.
|
20 | if (avio_seek(pb, pos, SEEK_SET) < 0) { |
282 | ✗ | av_log(s, AV_LOG_ERROR, "Failed to restore position after reading index\n"); | |
283 | ✗ | return -1; | |
284 | } | ||
285 |
2/2✓ Branch 0 taken 11 times.
✓ Branch 1 taken 9 times.
|
20 | if (ret < 0) |
286 | 11 | return ret; | |
287 | } | ||
288 | } | ||
289 | 18 | avi->index_loaded = 2; | |
290 | 18 | return 0; | |
291 | } | ||
292 | |||
293 | 7 | static void clean_index(AVFormatContext *s) | |
294 | { | ||
295 | int i; | ||
296 | int64_t j; | ||
297 | |||
298 |
2/2✓ Branch 0 taken 12 times.
✓ Branch 1 taken 7 times.
|
19 | for (i = 0; i < s->nb_streams; i++) { |
299 | 12 | AVStream *st = s->streams[i]; | |
300 | 12 | FFStream *const sti = ffstream(st); | |
301 | 12 | AVIStream *ast = st->priv_data; | |
302 | 12 | int n = sti->nb_index_entries; | |
303 | 12 | int max = ast->sample_size; | |
304 | int64_t pos, size, ts; | ||
305 | |||
306 |
4/4✓ Branch 0 taken 3 times.
✓ Branch 1 taken 9 times.
✓ Branch 2 taken 1 times.
✓ Branch 3 taken 2 times.
|
12 | if (n != 1 || ast->sample_size == 0) |
307 | 10 | continue; | |
308 | |||
309 |
2/2✓ Branch 0 taken 18 times.
✓ Branch 1 taken 2 times.
|
20 | while (max < 1024) |
310 | 18 | max += max; | |
311 | |||
312 | 2 | pos = sti->index_entries[0].pos; | |
313 | 2 | size = sti->index_entries[0].size; | |
314 | 2 | ts = sti->index_entries[0].timestamp; | |
315 | |||
316 |
2/2✓ Branch 0 taken 13 times.
✓ Branch 1 taken 2 times.
|
15 | for (j = 0; j < size; j += max) |
317 | 13 | av_add_index_entry(st, pos + j, ts + j, FFMIN(max, size - j), 0, | |
318 | AVINDEX_KEYFRAME); | ||
319 | } | ||
320 | 7 | } | |
321 | |||
322 | 19 | static int avi_read_tag(AVFormatContext *s, AVStream *st, uint32_t tag, | |
323 | uint32_t size) | ||
324 | { | ||
325 | 19 | AVIOContext *pb = s->pb; | |
326 | 19 | char key[5] = { 0 }; | |
327 | char *value; | ||
328 | |||
329 | 19 | size += (size & 1); | |
330 | |||
331 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 19 times.
|
19 | if (size == UINT_MAX) |
332 | ✗ | return AVERROR(EINVAL); | |
333 | 19 | value = av_malloc(size + 1); | |
334 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 19 times.
|
19 | if (!value) |
335 | ✗ | return AVERROR(ENOMEM); | |
336 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 19 times.
|
19 | if (avio_read(pb, value, size) != size) { |
337 | ✗ | av_freep(&value); | |
338 | ✗ | return AVERROR_INVALIDDATA; | |
339 | } | ||
340 | 19 | value[size] = 0; | |
341 | |||
342 | 19 | AV_WL32(key, tag); | |
343 | |||
344 |
1/2✓ Branch 0 taken 19 times.
✗ Branch 1 not taken.
|
19 | return av_dict_set(st ? &st->metadata : &s->metadata, key, value, |
345 | AV_DICT_DONT_STRDUP_VAL); | ||
346 | } | ||
347 | |||
348 | static const char months[12][4] = { "Jan", "Feb", "Mar", "Apr", "May", "Jun", | ||
349 | "Jul", "Aug", "Sep", "Oct", "Nov", "Dec" }; | ||
350 | |||
351 | 1 | static void avi_metadata_creation_time(AVDictionary **metadata, char *date) | |
352 | { | ||
353 | char month[4], time[9], buffer[64]; | ||
354 | int i, day, year; | ||
355 | /* parse standard AVI date format (ie. "Mon Mar 10 15:04:43 2003") */ | ||
356 |
1/2✓ Branch 0 taken 1 times.
✗ Branch 1 not taken.
|
1 | if (sscanf(date, "%*3s%*[ ]%3s%*[ ]%2d%*[ ]%8s%*[ ]%4d", |
357 | month, &day, time, &year) == 4) { | ||
358 |
2/2✓ Branch 0 taken 12 times.
✓ Branch 1 taken 1 times.
|
13 | for (i = 0; i < 12; i++) |
359 |
2/2✓ Branch 1 taken 1 times.
✓ Branch 2 taken 11 times.
|
12 | if (!av_strcasecmp(month, months[i])) { |
360 | 1 | snprintf(buffer, sizeof(buffer), "%.4d-%.2d-%.2d %s", | |
361 | year, i + 1, day, time); | ||
362 | 1 | av_dict_set(metadata, "creation_time", buffer, 0); | |
363 | } | ||
364 | ✗ | } else if (date[4] == '/' && date[7] == '/') { | |
365 | ✗ | date[4] = date[7] = '-'; | |
366 | ✗ | av_dict_set(metadata, "creation_time", date, 0); | |
367 | } | ||
368 | 1 | } | |
369 | |||
370 | ✗ | static void avi_read_nikon(AVFormatContext *s, uint64_t end) | |
371 | { | ||
372 | ✗ | while (avio_tell(s->pb) < end && !avio_feof(s->pb)) { | |
373 | ✗ | uint32_t tag = avio_rl32(s->pb); | |
374 | ✗ | uint32_t size = avio_rl32(s->pb); | |
375 | ✗ | switch (tag) { | |
376 | ✗ | case MKTAG('n', 'c', 't', 'g'): /* Nikon Tags */ | |
377 | { | ||
378 | ✗ | uint64_t tag_end = avio_tell(s->pb) + size; | |
379 | ✗ | while (avio_tell(s->pb) < tag_end && !avio_feof(s->pb)) { | |
380 | ✗ | uint16_t tag = avio_rl16(s->pb); | |
381 | ✗ | uint16_t size = avio_rl16(s->pb); | |
382 | ✗ | const char *name = NULL; | |
383 | ✗ | char buffer[64] = { 0 }; | |
384 | ✗ | uint64_t remaining = tag_end - avio_tell(s->pb); | |
385 | ✗ | size = FFMIN(size, remaining); | |
386 | ✗ | size -= avio_read(s->pb, buffer, | |
387 | ✗ | FFMIN(size, sizeof(buffer) - 1)); | |
388 | ✗ | switch (tag) { | |
389 | ✗ | case 0x03: | |
390 | ✗ | name = "maker"; | |
391 | ✗ | break; | |
392 | ✗ | case 0x04: | |
393 | ✗ | name = "model"; | |
394 | ✗ | break; | |
395 | ✗ | case 0x13: | |
396 | ✗ | name = "creation_time"; | |
397 | ✗ | if (buffer[4] == ':' && buffer[7] == ':') | |
398 | ✗ | buffer[4] = buffer[7] = '-'; | |
399 | ✗ | break; | |
400 | } | ||
401 | ✗ | if (name) | |
402 | ✗ | av_dict_set(&s->metadata, name, buffer, 0); | |
403 | ✗ | avio_skip(s->pb, size); | |
404 | } | ||
405 | ✗ | break; | |
406 | } | ||
407 | ✗ | default: | |
408 | ✗ | avio_skip(s->pb, size); | |
409 | ✗ | break; | |
410 | } | ||
411 | } | ||
412 | ✗ | } | |
413 | |||
414 | 6 | static int avi_extract_stream_metadata(AVFormatContext *s, AVStream *st) | |
415 | { | ||
416 | GetByteContext gb; | ||
417 | 6 | uint8_t *data = st->codecpar->extradata; | |
418 | 6 | int data_size = st->codecpar->extradata_size; | |
419 | int tag, offset; | ||
420 | |||
421 |
2/4✓ Branch 0 taken 6 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 6 times.
|
6 | if (!data || data_size < 8) { |
422 | ✗ | return AVERROR_INVALIDDATA; | |
423 | } | ||
424 | |||
425 | 6 | bytestream2_init(&gb, data, data_size); | |
426 | |||
427 | 6 | tag = bytestream2_get_le32(&gb); | |
428 | |||
429 |
1/4✗ Branch 0 not taken.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 6 times.
|
6 | switch (tag) { |
430 | ✗ | case MKTAG('A', 'V', 'I', 'F'): | |
431 | // skip 4 byte padding | ||
432 | ✗ | bytestream2_skip(&gb, 4); | |
433 | ✗ | offset = bytestream2_tell(&gb); | |
434 | |||
435 | // decode EXIF tags from IFD, AVI is always little-endian | ||
436 | ✗ | return avpriv_exif_decode_ifd(s, data + offset, data_size - offset, | |
437 | 1, 0, &st->metadata); | ||
438 | break; | ||
439 | ✗ | case MKTAG('C', 'A', 'S', 'I'): | |
440 | ✗ | avpriv_request_sample(s, "RIFF stream data tag type CASI (%u)", tag); | |
441 | ✗ | break; | |
442 | ✗ | case MKTAG('Z', 'o', 'r', 'a'): | |
443 | ✗ | avpriv_request_sample(s, "RIFF stream data tag type Zora (%u)", tag); | |
444 | ✗ | break; | |
445 | 6 | default: | |
446 | 6 | break; | |
447 | } | ||
448 | |||
449 | 6 | return 0; | |
450 | } | ||
451 | |||
452 | 468 | static int calculate_bitrate(AVFormatContext *s) | |
453 | { | ||
454 | 468 | AVIContext *avi = s->priv_data; | |
455 | int i, j; | ||
456 | 468 | int64_t lensum = 0; | |
457 | 468 | int64_t maxpos = 0; | |
458 | |||
459 |
2/2✓ Branch 0 taken 511 times.
✓ Branch 1 taken 468 times.
|
979 | for (i = 0; i<s->nb_streams; i++) { |
460 | 511 | int64_t len = 0; | |
461 | 511 | FFStream *const sti = ffstream(s->streams[i]); | |
462 | |||
463 |
2/2✓ Branch 0 taken 58 times.
✓ Branch 1 taken 453 times.
|
511 | if (!sti->nb_index_entries) |
464 | 58 | continue; | |
465 | |||
466 |
2/2✓ Branch 0 taken 30827 times.
✓ Branch 1 taken 453 times.
|
31280 | for (j = 0; j < sti->nb_index_entries; j++) |
467 | 30827 | len += sti->index_entries[j].size; | |
468 | 453 | maxpos = FFMAX(maxpos, sti->index_entries[j-1].pos); | |
469 | 453 | lensum += len; | |
470 | } | ||
471 |
2/2✓ Branch 0 taken 101 times.
✓ Branch 1 taken 367 times.
|
468 | if (maxpos < av_rescale(avi->io_fsize, 9, 10)) // index does not cover the whole file |
472 | 101 | return 0; | |
473 |
3/4✓ Branch 0 taken 367 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 29 times.
✓ Branch 3 taken 338 times.
|
367 | if (lensum*9/10 > maxpos || lensum < maxpos*9/10) // frame sum and filesize mismatch |
474 | 29 | return 0; | |
475 | |||
476 |
2/2✓ Branch 0 taken 358 times.
✓ Branch 1 taken 338 times.
|
696 | for (i = 0; i<s->nb_streams; i++) { |
477 | 358 | int64_t len = 0; | |
478 | 358 | AVStream *st = s->streams[i]; | |
479 | 358 | FFStream *const sti = ffstream(st); | |
480 | int64_t duration; | ||
481 | AVInteger bitrate_i, den_i, num_i; | ||
482 | |||
483 |
2/2✓ Branch 0 taken 27441 times.
✓ Branch 1 taken 358 times.
|
27799 | for (j = 0; j < sti->nb_index_entries; j++) |
484 | 27441 | len += sti->index_entries[j].size; | |
485 | |||
486 |
4/4✓ Branch 0 taken 356 times.
✓ Branch 1 taken 2 times.
✓ Branch 2 taken 21 times.
✓ Branch 3 taken 335 times.
|
358 | if (sti->nb_index_entries < 2 || st->codecpar->bit_rate > 0) |
487 | 23 | continue; | |
488 | 335 | duration = sti->index_entries[j-1].timestamp - sti->index_entries[0].timestamp; | |
489 | 335 | den_i = av_mul_i(av_int2i(duration), av_int2i(st->time_base.num)); | |
490 | 335 | num_i = av_add_i(av_mul_i(av_int2i(8*len), av_int2i(st->time_base.den)), av_shr_i(den_i, 1)); | |
491 | 335 | bitrate_i = av_div_i(num_i, den_i); | |
492 |
1/2✓ Branch 0 taken 335 times.
✗ Branch 1 not taken.
|
335 | if (av_cmp_i(bitrate_i, av_int2i(INT64_MAX)) <= 0) { |
493 | 335 | int64_t bitrate = av_i2int(bitrate_i); | |
494 |
1/2✓ Branch 0 taken 335 times.
✗ Branch 1 not taken.
|
335 | if (bitrate > 0) { |
495 | 335 | st->codecpar->bit_rate = bitrate; | |
496 | } | ||
497 | } | ||
498 | } | ||
499 | 338 | return 1; | |
500 | } | ||
501 | |||
502 | 468 | static int avi_read_header(AVFormatContext *s) | |
503 | { | ||
504 | 468 | AVIContext *avi = s->priv_data; | |
505 | 468 | AVIOContext *pb = s->pb; | |
506 | unsigned int tag, tag1, handler; | ||
507 | int codec_type, stream_index, frame_period; | ||
508 | unsigned int size; | ||
509 | int i; | ||
510 | AVStream *st; | ||
511 | 468 | AVIStream *ast = NULL; | |
512 | 468 | int avih_width = 0, avih_height = 0; | |
513 | 468 | int amv_file_format = 0; | |
514 | 468 | uint64_t list_end = 0; | |
515 | int64_t pos; | ||
516 | int ret; | ||
517 | AVDictionaryEntry *dict_entry; | ||
518 | |||
519 | 468 | avi->stream_index = -1; | |
520 | |||
521 | 468 | ret = get_riff(s, pb); | |
522 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 468 times.
|
468 | if (ret < 0) |
523 | ✗ | return ret; | |
524 | |||
525 | 468 | av_log(avi, AV_LOG_DEBUG, "use odml:%d\n", avi->use_odml); | |
526 | |||
527 | 468 | avi->io_fsize = avi->fsize = avio_size(pb); | |
528 |
3/4✓ Branch 0 taken 468 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 37 times.
✓ Branch 3 taken 431 times.
|
468 | if (avi->fsize <= 0 || avi->fsize < avi->riff_end) |
529 |
1/2✓ Branch 0 taken 37 times.
✗ Branch 1 not taken.
|
37 | avi->fsize = avi->riff_end == 8 ? INT64_MAX : avi->riff_end; |
530 | |||
531 | /* first list tag */ | ||
532 | 468 | stream_index = -1; | |
533 | 468 | codec_type = -1; | |
534 | 468 | frame_period = 0; | |
535 | 4134 | for (;;) { | |
536 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 4602 times.
|
4602 | if (avio_feof(pb)) |
537 | ✗ | return AVERROR_INVALIDDATA; | |
538 | 4602 | tag = avio_rl32(pb); | |
539 | 4602 | size = avio_rl32(pb); | |
540 | |||
541 | 4602 | print_tag(s, "tag", tag, size); | |
542 | |||
543 |
12/13✓ Branch 0 taken 1566 times.
✓ Branch 1 taken 1 times.
✓ Branch 2 taken 41 times.
✓ Branch 3 taken 2 times.
✓ Branch 4 taken 466 times.
✓ Branch 5 taken 512 times.
✓ Branch 6 taken 511 times.
✓ Branch 7 taken 11 times.
✓ Branch 8 taken 20 times.
✓ Branch 9 taken 10 times.
✓ Branch 10 taken 19 times.
✓ Branch 11 taken 1443 times.
✗ Branch 12 not taken.
|
4602 | switch (tag) { |
544 | 1566 | case MKTAG('L', 'I', 'S', 'T'): | |
545 | 1566 | list_end = avio_tell(pb) + size; | |
546 | /* Ignored, except at start of video packets. */ | ||
547 | 1566 | tag1 = avio_rl32(pb); | |
548 | |||
549 | 1566 | print_tag(s, "list", tag1, 0); | |
550 | |||
551 |
2/2✓ Branch 0 taken 468 times.
✓ Branch 1 taken 1098 times.
|
1566 | if (tag1 == MKTAG('m', 'o', 'v', 'i')) { |
552 | 468 | avi->movi_list = avio_tell(pb) - 4; | |
553 |
2/2✓ Branch 0 taken 466 times.
✓ Branch 1 taken 2 times.
|
468 | if (size) |
554 | 466 | avi->movi_end = avi->movi_list + size + (size & 1); | |
555 | else | ||
556 | 2 | avi->movi_end = avi->fsize; | |
557 | 468 | av_log(s, AV_LOG_TRACE, "movi end=%"PRIx64"\n", avi->movi_end); | |
558 | 468 | goto end_of_header; | |
559 |
2/2✓ Branch 0 taken 61 times.
✓ Branch 1 taken 1037 times.
|
1098 | } else if (tag1 == MKTAG('I', 'N', 'F', 'O')) |
560 | 61 | ff_read_riff_info(s, size - 4); | |
561 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1037 times.
|
1037 | else if (tag1 == MKTAG('n', 'c', 'd', 't')) |
562 | ✗ | avi_read_nikon(s, list_end); | |
563 | |||
564 | 4134 | break; | |
565 | 1 | case MKTAG('I', 'D', 'I', 'T'): | |
566 | { | ||
567 | 1 | unsigned char date[64] = { 0 }; | |
568 | 1 | size += (size & 1); | |
569 | 1 | size -= avio_read(pb, date, FFMIN(size, sizeof(date) - 1)); | |
570 | 1 | avio_skip(pb, size); | |
571 | 1 | avi_metadata_creation_time(&s->metadata, date); | |
572 | 1 | break; | |
573 | } | ||
574 | 41 | case MKTAG('d', 'm', 'l', 'h'): | |
575 | 41 | avi->is_odml = 1; | |
576 | 41 | avio_skip(pb, size + (size & 1)); | |
577 | 41 | break; | |
578 | 2 | case MKTAG('a', 'm', 'v', 'h'): | |
579 | 2 | amv_file_format = 1; | |
580 | 468 | case MKTAG('a', 'v', 'i', 'h'): | |
581 | /* AVI header */ | ||
582 | /* using frame_period is bad idea */ | ||
583 | 468 | frame_period = avio_rl32(pb); | |
584 | 468 | avio_rl32(pb); /* max. bytes per second */ | |
585 | 468 | avio_rl32(pb); | |
586 | 468 | avi->non_interleaved |= avio_rl32(pb) & AVIF_MUSTUSEINDEX; | |
587 | |||
588 | 468 | avio_skip(pb, 2 * 4); | |
589 | 468 | avio_rl32(pb); | |
590 | 468 | avio_rl32(pb); | |
591 | 468 | avih_width = avio_rl32(pb); | |
592 | 468 | avih_height = avio_rl32(pb); | |
593 | |||
594 | 468 | avio_skip(pb, size - 10 * 4); | |
595 | 468 | break; | |
596 | 512 | case MKTAG('s', 't', 'r', 'h'): | |
597 | /* stream header */ | ||
598 | |||
599 | 512 | tag1 = avio_rl32(pb); | |
600 | 512 | handler = avio_rl32(pb); /* codec tag */ | |
601 | |||
602 |
2/2✓ Branch 0 taken 1 times.
✓ Branch 1 taken 511 times.
|
512 | if (tag1 == MKTAG('p', 'a', 'd', 's')) { |
603 | 1 | avio_skip(pb, size - 8); | |
604 | 1 | break; | |
605 | } else { | ||
606 | 511 | stream_index++; | |
607 | 511 | st = avformat_new_stream(s, NULL); | |
608 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 511 times.
|
511 | if (!st) |
609 | ✗ | return AVERROR(ENOMEM); | |
610 | |||
611 | 511 | st->id = stream_index; | |
612 | 511 | ast = av_mallocz(sizeof(AVIStream)); | |
613 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 511 times.
|
511 | if (!ast) |
614 | ✗ | return AVERROR(ENOMEM); | |
615 | 511 | st->priv_data = ast; | |
616 | } | ||
617 |
2/2✓ Branch 0 taken 4 times.
✓ Branch 1 taken 507 times.
|
511 | if (amv_file_format) |
618 | 4 | tag1 = stream_index ? MKTAG('a', 'u', 'd', 's') | |
619 |
2/2✓ Branch 0 taken 2 times.
✓ Branch 1 taken 2 times.
|
4 | : MKTAG('v', 'i', 'd', 's'); |
620 | |||
621 | 511 | print_tag(s, "strh", tag1, -1); | |
622 | |||
623 |
2/4✓ Branch 0 taken 511 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 511 times.
|
511 | if (tag1 == MKTAG('i', 'a', 'v', 's') || |
624 | tag1 == MKTAG('i', 'v', 'a', 's')) { | ||
625 | int64_t dv_dur; | ||
626 | |||
627 | /* After some consideration -- I don't think we | ||
628 | * have to support anything but DV in type1 AVIs. */ | ||
629 | ✗ | if (s->nb_streams != 1) | |
630 | ✗ | return AVERROR_INVALIDDATA; | |
631 | |||
632 | ✗ | if (handler != MKTAG('d', 'v', 's', 'd') && | |
633 | ✗ | handler != MKTAG('d', 'v', 'h', 'd') && | |
634 | handler != MKTAG('d', 'v', 's', 'l')) | ||
635 | ✗ | return AVERROR_INVALIDDATA; | |
636 | |||
637 | if (!CONFIG_DV_DEMUXER) | ||
638 | return AVERROR_DEMUXER_NOT_FOUND; | ||
639 | |||
640 | ✗ | ast = s->streams[0]->priv_data; | |
641 | ✗ | st->priv_data = NULL; | |
642 | ✗ | ff_remove_stream(s, st); | |
643 | |||
644 | ✗ | avi->dv_demux = avpriv_dv_init_demux(s); | |
645 | ✗ | if (!avi->dv_demux) { | |
646 | ✗ | av_free(ast); | |
647 | ✗ | return AVERROR(ENOMEM); | |
648 | } | ||
649 | |||
650 | ✗ | s->streams[0]->priv_data = ast; | |
651 | ✗ | avio_skip(pb, 3 * 4); | |
652 | ✗ | ast->scale = avio_rl32(pb); | |
653 | ✗ | ast->rate = avio_rl32(pb); | |
654 | ✗ | avio_skip(pb, 4); /* start time */ | |
655 | |||
656 | ✗ | dv_dur = avio_rl32(pb); | |
657 | ✗ | if (ast->scale > 0 && ast->rate > 0 && dv_dur > 0) { | |
658 | ✗ | dv_dur *= AV_TIME_BASE; | |
659 | ✗ | s->duration = av_rescale(dv_dur, ast->scale, ast->rate); | |
660 | } | ||
661 | /* else, leave duration alone; timing estimation in utils.c | ||
662 | * will make a guess based on bitrate. */ | ||
663 | |||
664 | ✗ | stream_index = s->nb_streams - 1; | |
665 | ✗ | avio_skip(pb, size - 9 * 4); | |
666 | ✗ | break; | |
667 | } | ||
668 | |||
669 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 511 times.
|
511 | av_assert0(stream_index < s->nb_streams); |
670 | 511 | ast->handler = handler; | |
671 | |||
672 | 511 | avio_rl32(pb); /* flags */ | |
673 | 511 | avio_rl16(pb); /* priority */ | |
674 | 511 | avio_rl16(pb); /* language */ | |
675 | 511 | avio_rl32(pb); /* initial frame */ | |
676 | 511 | ast->scale = avio_rl32(pb); | |
677 | 511 | ast->rate = avio_rl32(pb); | |
678 |
3/4✓ Branch 0 taken 507 times.
✓ Branch 1 taken 4 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 507 times.
|
511 | if (!(ast->scale && ast->rate)) { |
679 | 4 | av_log(s, AV_LOG_WARNING, | |
680 | "scale/rate is %"PRIu32"/%"PRIu32" which is invalid. " | ||
681 | "(This file has been generated by broken software.)\n", | ||
682 | ast->scale, | ||
683 | ast->rate); | ||
684 |
1/2✓ Branch 0 taken 4 times.
✗ Branch 1 not taken.
|
4 | if (frame_period) { |
685 | 4 | ast->rate = 1000000; | |
686 | 4 | ast->scale = frame_period; | |
687 | } else { | ||
688 | ✗ | ast->rate = 25; | |
689 | ✗ | ast->scale = 1; | |
690 | } | ||
691 | } | ||
692 | 511 | avpriv_set_pts_info(st, 64, ast->scale, ast->rate); | |
693 | |||
694 | 511 | ast->cum_len = avio_rl32(pb); /* start */ | |
695 | 511 | st->nb_frames = avio_rl32(pb); | |
696 | |||
697 | 511 | st->start_time = 0; | |
698 | 511 | avio_rl32(pb); /* buffer size */ | |
699 | 511 | avio_rl32(pb); /* quality */ | |
700 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 511 times.
|
511 | if (ast->cum_len > 3600LL * ast->rate / ast->scale) { |
701 | ✗ | av_log(s, AV_LOG_ERROR, "crazy start time, iam scared, giving up\n"); | |
702 | ✗ | ast->cum_len = 0; | |
703 | } | ||
704 | 511 | ast->sample_size = avio_rl32(pb); | |
705 | 511 | ast->cum_len *= FFMAX(1, ast->sample_size); | |
706 | 511 | av_log(s, AV_LOG_TRACE, "%"PRIu32" %"PRIu32" %d\n", | |
707 | ast->rate, ast->scale, ast->sample_size); | ||
708 | |||
709 |
2/5✓ Branch 0 taken 465 times.
✓ Branch 1 taken 46 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
|
511 | switch (tag1) { |
710 | 465 | case MKTAG('v', 'i', 'd', 's'): | |
711 | 465 | codec_type = AVMEDIA_TYPE_VIDEO; | |
712 | |||
713 | 465 | ast->sample_size = 0; | |
714 | 465 | st->avg_frame_rate = av_inv_q(st->time_base); | |
715 | 465 | break; | |
716 | 46 | case MKTAG('a', 'u', 'd', 's'): | |
717 | 46 | codec_type = AVMEDIA_TYPE_AUDIO; | |
718 | 46 | break; | |
719 | ✗ | case MKTAG('t', 'x', 't', 's'): | |
720 | ✗ | codec_type = AVMEDIA_TYPE_SUBTITLE; | |
721 | ✗ | break; | |
722 | ✗ | case MKTAG('d', 'a', 't', 's'): | |
723 | ✗ | codec_type = AVMEDIA_TYPE_DATA; | |
724 | ✗ | break; | |
725 | ✗ | default: | |
726 | ✗ | av_log(s, AV_LOG_INFO, "unknown stream type %X\n", tag1); | |
727 | } | ||
728 | |||
729 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 511 times.
|
511 | if (ast->sample_size < 0) { |
730 | ✗ | if (s->error_recognition & AV_EF_EXPLODE) { | |
731 | ✗ | av_log(s, AV_LOG_ERROR, | |
732 | "Invalid sample_size %d at stream %d\n", | ||
733 | ast->sample_size, | ||
734 | stream_index); | ||
735 | ✗ | return AVERROR_INVALIDDATA; | |
736 | } | ||
737 | ✗ | av_log(s, AV_LOG_WARNING, | |
738 | "Invalid sample_size %d at stream %d " | ||
739 | "setting it to 0\n", | ||
740 | ast->sample_size, | ||
741 | stream_index); | ||
742 | ✗ | ast->sample_size = 0; | |
743 | } | ||
744 | |||
745 |
2/2✓ Branch 0 taken 475 times.
✓ Branch 1 taken 36 times.
|
511 | if (ast->sample_size == 0) { |
746 | 475 | st->duration = st->nb_frames; | |
747 |
5/6✓ Branch 0 taken 470 times.
✓ Branch 1 taken 5 times.
✓ Branch 2 taken 470 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 38 times.
✓ Branch 5 taken 432 times.
|
475 | if (st->duration > 0 && avi->io_fsize > 0 && avi->riff_end > avi->io_fsize) { |
748 | 38 | av_log(s, AV_LOG_DEBUG, "File is truncated adjusting duration\n"); | |
749 | 38 | st->duration = av_rescale(st->duration, avi->io_fsize, avi->riff_end); | |
750 | } | ||
751 | } | ||
752 | 511 | ast->frame_offset = ast->cum_len; | |
753 | 511 | avio_skip(pb, size - 12 * 4); | |
754 | 511 | break; | |
755 | 511 | case MKTAG('s', 't', 'r', 'f'): | |
756 | /* stream header */ | ||
757 |
1/6✗ Branch 0 not taken.
✓ Branch 1 taken 511 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
|
511 | if (!size && (codec_type == AVMEDIA_TYPE_AUDIO || |
758 | codec_type == AVMEDIA_TYPE_VIDEO)) | ||
759 | break; | ||
760 |
2/4✓ Branch 0 taken 511 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 511 times.
|
511 | if (stream_index >= (unsigned)s->nb_streams || avi->dv_demux) { |
761 | ✗ | avio_skip(pb, size); | |
762 | } else { | ||
763 | 511 | uint64_t cur_pos = avio_tell(pb); | |
764 | FFStream *sti; | ||
765 | unsigned esize; | ||
766 |
2/2✓ Branch 0 taken 507 times.
✓ Branch 1 taken 4 times.
|
511 | if (cur_pos < list_end) |
767 | 507 | size = FFMIN(size, list_end - cur_pos); | |
768 | 511 | st = s->streams[stream_index]; | |
769 | 511 | sti = ffstream(st); | |
770 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 511 times.
|
511 | if (st->codecpar->codec_type != AVMEDIA_TYPE_UNKNOWN) { |
771 | ✗ | avio_skip(pb, size); | |
772 | ✗ | break; | |
773 | } | ||
774 |
2/4✓ Branch 0 taken 465 times.
✓ Branch 1 taken 46 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
|
511 | switch (codec_type) { |
775 | 465 | case AVMEDIA_TYPE_VIDEO: | |
776 |
2/2✓ Branch 0 taken 2 times.
✓ Branch 1 taken 463 times.
|
465 | if (amv_file_format) { |
777 | 2 | st->codecpar->width = avih_width; | |
778 | 2 | st->codecpar->height = avih_height; | |
779 | 2 | st->codecpar->codec_type = AVMEDIA_TYPE_VIDEO; | |
780 | 2 | st->codecpar->codec_id = AV_CODEC_ID_AMV; | |
781 | 2 | avio_skip(pb, size); | |
782 | 511 | break; | |
783 | } | ||
784 | 463 | tag1 = ff_get_bmp_header(pb, st, &esize); | |
785 | |||
786 |
2/4✓ Branch 0 taken 463 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 463 times.
|
463 | if (tag1 == MKTAG('D', 'X', 'S', 'B') || |
787 | tag1 == MKTAG('D', 'X', 'S', 'A')) { | ||
788 | ✗ | st->codecpar->codec_type = AVMEDIA_TYPE_SUBTITLE; | |
789 | ✗ | st->codecpar->codec_tag = tag1; | |
790 | ✗ | st->codecpar->codec_id = AV_CODEC_ID_XSUB; | |
791 | ✗ | break; | |
792 | } | ||
793 | |||
794 |
4/6✓ Branch 0 taken 162 times.
✓ Branch 1 taken 301 times.
✓ Branch 2 taken 162 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 162 times.
✗ Branch 5 not taken.
|
463 | if (size > 10 * 4 && size < (1 << 30) && size < avi->fsize) { |
795 |
3/4✓ Branch 0 taken 18 times.
✓ Branch 1 taken 144 times.
✓ Branch 2 taken 18 times.
✗ Branch 3 not taken.
|
162 | if (esize == size-1 && (esize&1)) { |
796 | 18 | st->codecpar->extradata_size = esize - 10 * 4; | |
797 | } else | ||
798 | 144 | st->codecpar->extradata_size = size - 10 * 4; | |
799 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 162 times.
|
162 | if (st->codecpar->extradata) { |
800 | ✗ | av_log(s, AV_LOG_WARNING, "New extradata in strf chunk, freeing previous one.\n"); | |
801 | } | ||
802 | 162 | ret = ff_get_extradata(s, st->codecpar, pb, | |
803 | 162 | st->codecpar->extradata_size); | |
804 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 162 times.
|
162 | if (ret < 0) |
805 | ✗ | return ret; | |
806 | } | ||
807 | |||
808 | // FIXME: check if the encoder really did this correctly | ||
809 |
2/2✓ Branch 0 taken 18 times.
✓ Branch 1 taken 445 times.
|
463 | if (st->codecpar->extradata_size & 1) |
810 | 18 | avio_r8(pb); | |
811 | |||
812 | /* Extract palette from extradata if bpp <= 8. | ||
813 | * This code assumes that extradata contains only palette. | ||
814 | * This is true for all paletted codecs implemented in | ||
815 | * FFmpeg. */ | ||
816 |
2/2✓ Branch 0 taken 162 times.
✓ Branch 1 taken 301 times.
|
463 | if (st->codecpar->extradata_size && |
817 |
2/2✓ Branch 0 taken 15 times.
✓ Branch 1 taken 147 times.
|
162 | (st->codecpar->bits_per_coded_sample <= 8)) { |
818 | 15 | int pal_size = (1 << st->codecpar->bits_per_coded_sample) << 2; | |
819 | const uint8_t *pal_src; | ||
820 | |||
821 | 15 | pal_size = FFMIN(pal_size, st->codecpar->extradata_size); | |
822 | 15 | pal_src = st->codecpar->extradata + | |
823 | 15 | st->codecpar->extradata_size - pal_size; | |
824 | /* Exclude the "BottomUp" field from the palette */ | ||
825 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 15 times.
|
15 | if (pal_src - st->codecpar->extradata >= 9 && |
826 | ✗ | !memcmp(st->codecpar->extradata + st->codecpar->extradata_size - 9, "BottomUp", 9)) | |
827 | ✗ | pal_src -= 9; | |
828 |
2/2✓ Branch 0 taken 2330 times.
✓ Branch 1 taken 15 times.
|
2345 | for (i = 0; i < pal_size / 4; i++) |
829 | 2330 | ast->pal[i] = 0xFFU<<24 | AV_RL32(pal_src + 4 * i); | |
830 | 15 | ast->has_pal = 1; | |
831 | } | ||
832 | |||
833 | 463 | print_tag(s, "video", tag1, 0); | |
834 | |||
835 | 463 | st->codecpar->codec_type = AVMEDIA_TYPE_VIDEO; | |
836 | 463 | st->codecpar->codec_tag = tag1; | |
837 | 463 | st->codecpar->codec_id = ff_codec_get_id(ff_codec_bmp_tags, | |
838 | tag1); | ||
839 | /* If codec is not found yet, try with the mov tags. */ | ||
840 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 463 times.
|
463 | if (!st->codecpar->codec_id) { |
841 | ✗ | st->codecpar->codec_id = | |
842 | ✗ | ff_codec_get_id(ff_codec_movvideo_tags, tag1); | |
843 | ✗ | if (st->codecpar->codec_id) | |
844 | ✗ | av_log(s, AV_LOG_WARNING, | |
845 | "mov tag found in avi (fourcc %s)\n", | ||
846 | ✗ | av_fourcc2str(tag1)); | |
847 | } | ||
848 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 463 times.
|
463 | if (!st->codecpar->codec_id) |
849 | ✗ | st->codecpar->codec_id = ff_codec_get_id(ff_codec_bmp_tags_unofficial, tag1); | |
850 | |||
851 | /* This is needed to get the pict type which is necessary | ||
852 | * for generating correct pts. */ | ||
853 | 463 | sti->need_parsing = AVSTREAM_PARSE_HEADERS; | |
854 | |||
855 |
2/2✓ Branch 0 taken 57 times.
✓ Branch 1 taken 406 times.
|
463 | if (st->codecpar->codec_id == AV_CODEC_ID_MPEG4 && |
856 |
2/2✓ Branch 0 taken 2 times.
✓ Branch 1 taken 55 times.
|
57 | ast->handler == MKTAG('X', 'V', 'I', 'D')) |
857 | 2 | st->codecpar->codec_tag = MKTAG('X', 'V', 'I', 'D'); | |
858 | |||
859 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 463 times.
|
463 | if (st->codecpar->codec_tag == MKTAG('V', 'S', 'S', 'H')) |
860 | ✗ | sti->need_parsing = AVSTREAM_PARSE_FULL; | |
861 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 463 times.
|
463 | if (st->codecpar->codec_id == AV_CODEC_ID_RV40) |
862 | ✗ | sti->need_parsing = AVSTREAM_PARSE_NONE; | |
863 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 463 times.
|
463 | if (st->codecpar->codec_id == AV_CODEC_ID_HEVC && |
864 | ✗ | st->codecpar->codec_tag == MKTAG('H', '2', '6', '5')) | |
865 | ✗ | sti->need_parsing = AVSTREAM_PARSE_FULL; | |
866 | |||
867 |
2/2✓ Branch 0 taken 1 times.
✓ Branch 1 taken 462 times.
|
463 | if (st->codecpar->codec_id == AV_CODEC_ID_AVRN && |
868 |
1/2✓ Branch 0 taken 1 times.
✗ Branch 1 not taken.
|
1 | st->codecpar->codec_tag == MKTAG('A', 'V', 'R', 'n') && |
869 |
1/2✓ Branch 0 taken 1 times.
✗ Branch 1 not taken.
|
1 | (st->codecpar->extradata_size < 31 || |
870 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
|
1 | memcmp(&st->codecpar->extradata[28], "1:1", 3))) |
871 | ✗ | st->codecpar->codec_id = AV_CODEC_ID_MJPEG; | |
872 | |||
873 |
3/4✓ Branch 0 taken 13 times.
✓ Branch 1 taken 450 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 13 times.
|
463 | if (st->codecpar->codec_tag == 0 && st->codecpar->height > 0 && |
874 | ✗ | st->codecpar->extradata_size < 1U << 30) { | |
875 | ✗ | st->codecpar->extradata_size += 9; | |
876 | ✗ | if ((ret = av_reallocp(&st->codecpar->extradata, | |
877 | ✗ | st->codecpar->extradata_size + | |
878 | AV_INPUT_BUFFER_PADDING_SIZE)) < 0) { | ||
879 | ✗ | st->codecpar->extradata_size = 0; | |
880 | ✗ | return ret; | |
881 | } else | ||
882 | ✗ | memcpy(st->codecpar->extradata + st->codecpar->extradata_size - 9, | |
883 | "BottomUp", 9); | ||
884 | } | ||
885 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 463 times.
|
463 | if (st->codecpar->height == INT_MIN) |
886 | ✗ | return AVERROR_INVALIDDATA; | |
887 | 463 | st->codecpar->height = FFABS(st->codecpar->height); | |
888 | |||
889 | // avio_skip(pb, size - 5 * 4); | ||
890 | 463 | break; | |
891 | 46 | case AVMEDIA_TYPE_AUDIO: | |
892 | 46 | ret = ff_get_wav_header(s, pb, st->codecpar, size, 0); | |
893 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 46 times.
|
46 | if (ret < 0) |
894 | ✗ | return ret; | |
895 | 46 | ast->dshow_block_align = st->codecpar->block_align; | |
896 |
3/4✓ Branch 0 taken 36 times.
✓ Branch 1 taken 10 times.
✓ Branch 2 taken 36 times.
✗ Branch 3 not taken.
|
46 | if (ast->sample_size && st->codecpar->block_align && |
897 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 36 times.
|
36 | ast->sample_size != st->codecpar->block_align) { |
898 | ✗ | av_log(s, | |
899 | AV_LOG_WARNING, | ||
900 | "sample size (%d) != block align (%d)\n", | ||
901 | ast->sample_size, | ||
902 | ✗ | st->codecpar->block_align); | |
903 | ✗ | ast->sample_size = st->codecpar->block_align; | |
904 | } | ||
905 | /* 2-aligned | ||
906 | * (fix for Stargate SG-1 - 3x18 - Shades of Grey.avi) */ | ||
907 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 46 times.
|
46 | if (size & 1) |
908 | ✗ | avio_skip(pb, 1); | |
909 | /* Force parsing as several audio frames can be in | ||
910 | * one packet and timestamps refer to packet start. */ | ||
911 | 46 | sti->need_parsing = AVSTREAM_PARSE_TIMESTAMPS; | |
912 | /* ADTS header is in extradata, AAC without header must be | ||
913 | * stored as exact frames. Parser not needed and it will | ||
914 | * fail. */ | ||
915 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 46 times.
|
46 | if (st->codecpar->codec_id == AV_CODEC_ID_AAC && |
916 | ✗ | st->codecpar->extradata_size) | |
917 | ✗ | sti->need_parsing = AVSTREAM_PARSE_NONE; | |
918 | // The flac parser does not work with AVSTREAM_PARSE_TIMESTAMPS | ||
919 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 46 times.
|
46 | if (st->codecpar->codec_id == AV_CODEC_ID_FLAC) |
920 | ✗ | sti->need_parsing = AVSTREAM_PARSE_NONE; | |
921 | /* AVI files with Xan DPCM audio (wrongly) declare PCM | ||
922 | * audio in the header but have Axan as stream_code_tag. */ | ||
923 |
2/2✓ Branch 0 taken 2 times.
✓ Branch 1 taken 44 times.
|
46 | if (ast->handler == AV_RL32("Axan")) { |
924 | 2 | st->codecpar->codec_id = AV_CODEC_ID_XAN_DPCM; | |
925 | 2 | st->codecpar->codec_tag = 0; | |
926 | 2 | ast->dshow_block_align = 0; | |
927 | } | ||
928 |
2/2✓ Branch 0 taken 2 times.
✓ Branch 1 taken 44 times.
|
46 | if (amv_file_format) { |
929 | 2 | st->codecpar->codec_id = AV_CODEC_ID_ADPCM_IMA_AMV; | |
930 | 2 | ast->dshow_block_align = 0; | |
931 | } | ||
932 |
1/2✓ Branch 0 taken 46 times.
✗ Branch 1 not taken.
|
46 | if ((st->codecpar->codec_id == AV_CODEC_ID_AAC || |
933 |
1/2✓ Branch 0 taken 46 times.
✗ Branch 1 not taken.
|
46 | st->codecpar->codec_id == AV_CODEC_ID_FTR || |
934 |
1/2✓ Branch 0 taken 46 times.
✗ Branch 1 not taken.
|
46 | st->codecpar->codec_id == AV_CODEC_ID_FLAC || |
935 |
3/6✓ Branch 0 taken 3 times.
✓ Branch 1 taken 43 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 3 times.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
|
46 | st->codecpar->codec_id == AV_CODEC_ID_MP2 ) && ast->dshow_block_align <= 4 && ast->dshow_block_align) { |
936 | ✗ | av_log(s, AV_LOG_DEBUG, "overriding invalid dshow_block_align of %d\n", ast->dshow_block_align); | |
937 | ✗ | ast->dshow_block_align = 0; | |
938 | } | ||
939 |
1/6✗ Branch 0 not taken.
✓ Branch 1 taken 46 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
|
46 | if (st->codecpar->codec_id == AV_CODEC_ID_AAC && ast->dshow_block_align == 1024 && ast->sample_size == 1024 || |
940 |
1/6✗ Branch 0 not taken.
✓ Branch 1 taken 46 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
|
46 | st->codecpar->codec_id == AV_CODEC_ID_AAC && ast->dshow_block_align == 4096 && ast->sample_size == 4096 || |
941 |
3/6✓ Branch 0 taken 2 times.
✓ Branch 1 taken 44 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 2 times.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
|
46 | st->codecpar->codec_id == AV_CODEC_ID_MP3 && ast->dshow_block_align == 1152 && ast->sample_size == 1152) { |
942 | ✗ | av_log(s, AV_LOG_DEBUG, "overriding sample_size\n"); | |
943 | ✗ | ast->sample_size = 0; | |
944 | } | ||
945 | 46 | break; | |
946 | ✗ | case AVMEDIA_TYPE_SUBTITLE: | |
947 | ✗ | st->codecpar->codec_type = AVMEDIA_TYPE_SUBTITLE; | |
948 | ✗ | sti->request_probe = 1; | |
949 | ✗ | avio_skip(pb, size); | |
950 | ✗ | break; | |
951 | ✗ | default: | |
952 | ✗ | st->codecpar->codec_type = AVMEDIA_TYPE_DATA; | |
953 | ✗ | st->codecpar->codec_id = AV_CODEC_ID_NONE; | |
954 | ✗ | st->codecpar->codec_tag = 0; | |
955 | ✗ | avio_skip(pb, size); | |
956 | ✗ | break; | |
957 | } | ||
958 | } | ||
959 | 511 | break; | |
960 | 11 | case MKTAG('s', 't', 'r', 'd'): | |
961 |
1/2✓ Branch 0 taken 11 times.
✗ Branch 1 not taken.
|
11 | if (stream_index >= (unsigned)s->nb_streams |
962 |
2/2✓ Branch 0 taken 6 times.
✓ Branch 1 taken 5 times.
|
11 | || s->streams[stream_index]->codecpar->extradata_size |
963 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
|
6 | || s->streams[stream_index]->codecpar->codec_tag == MKTAG('H','2','6','4')) { |
964 | 5 | avio_skip(pb, size); | |
965 | } else { | ||
966 | 6 | uint64_t cur_pos = avio_tell(pb); | |
967 |
1/2✓ Branch 0 taken 6 times.
✗ Branch 1 not taken.
|
6 | if (cur_pos < list_end) |
968 | 6 | size = FFMIN(size, list_end - cur_pos); | |
969 | 6 | st = s->streams[stream_index]; | |
970 | |||
971 |
1/2✓ Branch 0 taken 6 times.
✗ Branch 1 not taken.
|
6 | if (size<(1<<30)) { |
972 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
|
6 | if (st->codecpar->extradata) { |
973 | ✗ | av_log(s, AV_LOG_WARNING, "New extradata in strd chunk, freeing previous one.\n"); | |
974 | } | ||
975 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 6 times.
|
6 | if ((ret = ff_get_extradata(s, st->codecpar, pb, size)) < 0) |
976 | ✗ | return ret; | |
977 | } | ||
978 | |||
979 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
|
6 | if (st->codecpar->extradata_size & 1) //FIXME check if the encoder really did this correctly |
980 | ✗ | avio_r8(pb); | |
981 | |||
982 | 6 | ret = avi_extract_stream_metadata(s, st); | |
983 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
|
6 | if (ret < 0) { |
984 | ✗ | av_log(s, AV_LOG_WARNING, "could not decoding EXIF data in stream header.\n"); | |
985 | } | ||
986 | } | ||
987 | 11 | break; | |
988 | 20 | case MKTAG('i', 'n', 'd', 'x'): | |
989 | 20 | pos = avio_tell(pb); | |
990 |
2/4✓ Branch 0 taken 20 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 20 times.
✗ Branch 3 not taken.
|
20 | if ((pb->seekable & AVIO_SEEKABLE_NORMAL) && !(s->flags & AVFMT_FLAG_IGNIDX) && |
991 |
3/4✓ Branch 0 taken 20 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 11 times.
✓ Branch 3 taken 9 times.
|
40 | avi->use_odml && |
992 | 20 | read_odml_index(s, 0) < 0 && | |
993 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 11 times.
|
11 | (s->error_recognition & AV_EF_EXPLODE)) |
994 | ✗ | return AVERROR_INVALIDDATA; | |
995 | 20 | avio_seek(pb, pos + size, SEEK_SET); | |
996 | 20 | break; | |
997 | 10 | case MKTAG('v', 'p', 'r', 'p'): | |
998 |
2/4✓ Branch 0 taken 10 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 10 times.
✗ Branch 3 not taken.
|
10 | if (stream_index < (unsigned)s->nb_streams && size > 9 * 4) { |
999 | AVRational active, active_aspect; | ||
1000 | |||
1001 | 10 | st = s->streams[stream_index]; | |
1002 | 10 | avio_rl32(pb); | |
1003 | 10 | avio_rl32(pb); | |
1004 | 10 | avio_rl32(pb); | |
1005 | 10 | avio_rl32(pb); | |
1006 | 10 | avio_rl32(pb); | |
1007 | |||
1008 | 10 | active_aspect.den = avio_rl16(pb); | |
1009 | 10 | active_aspect.num = avio_rl16(pb); | |
1010 | 10 | active.num = avio_rl32(pb); | |
1011 | 10 | active.den = avio_rl32(pb); | |
1012 | 10 | avio_rl32(pb); // nbFieldsPerFrame | |
1013 | |||
1014 |
2/4✓ Branch 0 taken 10 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 10 times.
✗ Branch 3 not taken.
|
10 | if (active_aspect.num && active_aspect.den && |
1015 |
2/4✓ Branch 0 taken 10 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 10 times.
✗ Branch 3 not taken.
|
10 | active.num && active.den) { |
1016 | 10 | st->sample_aspect_ratio = av_div_q(active_aspect, active); | |
1017 | 10 | av_log(s, AV_LOG_TRACE, "vprp %d/%d %d/%d\n", | |
1018 | active_aspect.num, active_aspect.den, | ||
1019 | active.num, active.den); | ||
1020 | } | ||
1021 | 10 | size -= 9 * 4; | |
1022 | } | ||
1023 | 10 | avio_skip(pb, size); | |
1024 | 10 | break; | |
1025 | 19 | case MKTAG('s', 't', 'r', 'n'): | |
1026 | case MKTAG('i', 's', 'b', 'j'): | ||
1027 | case MKTAG('i', 'n', 'a', 'm'): | ||
1028 | case MKTAG('i', 'a', 'r', 't'): | ||
1029 | case MKTAG('i', 'c', 'o', 'p'): | ||
1030 | case MKTAG('i', 'c', 'm', 't'): | ||
1031 | case MKTAG('i', 'g', 'n', 'r'): | ||
1032 | case MKTAG('i', 'p', 'o', 'd'): | ||
1033 | case MKTAG('i', 's', 'o', 'f'): | ||
1034 |
1/2✓ Branch 0 taken 19 times.
✗ Branch 1 not taken.
|
19 | if (s->nb_streams) { |
1035 | 19 | ret = avi_read_tag(s, s->streams[s->nb_streams - 1], tag, size); | |
1036 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 19 times.
|
19 | if (ret < 0) |
1037 | ✗ | return ret; | |
1038 | 19 | break; | |
1039 | } | ||
1040 | default: | ||
1041 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1443 times.
|
1443 | if (size > 1000000) { |
1042 | ✗ | av_log(s, AV_LOG_ERROR, | |
1043 | "Something went wrong during header parsing, " | ||
1044 | "tag %s has size %u, " | ||
1045 | "I will ignore it and try to continue anyway.\n", | ||
1046 | ✗ | av_fourcc2str(tag), size); | |
1047 | ✗ | if (s->error_recognition & AV_EF_EXPLODE) | |
1048 | ✗ | return AVERROR_INVALIDDATA; | |
1049 | ✗ | avi->movi_list = avio_tell(pb) - 4; | |
1050 | ✗ | avi->movi_end = avi->fsize; | |
1051 | ✗ | goto end_of_header; | |
1052 | } | ||
1053 | /* Do not fail for very large idx1 tags */ | ||
1054 | case MKTAG('i', 'd', 'x', '1'): | ||
1055 | /* skip tag */ | ||
1056 | 1443 | size += (size & 1); | |
1057 | 1443 | avio_skip(pb, size); | |
1058 | 1443 | break; | |
1059 | } | ||
1060 | } | ||
1061 | |||
1062 | 468 | end_of_header: | |
1063 | /* check stream number */ | ||
1064 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 468 times.
|
468 | if (stream_index != s->nb_streams - 1) |
1065 | ✗ | return AVERROR_INVALIDDATA; | |
1066 | |||
1067 |
3/4✓ Branch 0 taken 462 times.
✓ Branch 1 taken 6 times.
✓ Branch 2 taken 462 times.
✗ Branch 3 not taken.
|
468 | if (!avi->index_loaded && (pb->seekable & AVIO_SEEKABLE_NORMAL)) |
1068 | 462 | avi_load_index(s); | |
1069 | 468 | calculate_bitrate(s); | |
1070 | 468 | avi->index_loaded |= 1; | |
1071 | |||
1072 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 468 times.
|
468 | if ((ret = guess_ni_flag(s)) < 0) |
1073 | ✗ | return ret; | |
1074 | |||
1075 | 468 | avi->non_interleaved |= ret | (s->flags & AVFMT_FLAG_SORT_DTS); | |
1076 | |||
1077 | 468 | dict_entry = av_dict_get(s->metadata, "ISFT", NULL, 0); | |
1078 |
3/4✓ Branch 0 taken 60 times.
✓ Branch 1 taken 408 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 60 times.
|
468 | if (dict_entry && !strcmp(dict_entry->value, "PotEncoder")) |
1079 | ✗ | for (i = 0; i < s->nb_streams; i++) { | |
1080 | ✗ | AVStream *st = s->streams[i]; | |
1081 | ✗ | if ( st->codecpar->codec_id == AV_CODEC_ID_MPEG1VIDEO | |
1082 | ✗ | || st->codecpar->codec_id == AV_CODEC_ID_MPEG2VIDEO) | |
1083 | ✗ | ffstream(st)->need_parsing = AVSTREAM_PARSE_FULL; | |
1084 | } | ||
1085 | |||
1086 |
2/2✓ Branch 0 taken 487 times.
✓ Branch 1 taken 37 times.
|
524 | for (i = 0; i < s->nb_streams; i++) { |
1087 | 487 | AVStream *st = s->streams[i]; | |
1088 |
2/2✓ Branch 1 taken 431 times.
✓ Branch 2 taken 56 times.
|
487 | if (ffstream(st)->nb_index_entries) |
1089 | 431 | break; | |
1090 | } | ||
1091 | // DV-in-AVI cannot be non-interleaved, if set this must be | ||
1092 | // a mis-detection. | ||
1093 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 468 times.
|
468 | if (avi->dv_demux) |
1094 | ✗ | avi->non_interleaved = 0; | |
1095 |
4/4✓ Branch 0 taken 37 times.
✓ Branch 1 taken 431 times.
✓ Branch 2 taken 1 times.
✓ Branch 3 taken 36 times.
|
468 | if (i == s->nb_streams && avi->non_interleaved) { |
1096 | 1 | av_log(s, AV_LOG_WARNING, | |
1097 | "Non-interleaved AVI without index, switching to interleaved\n"); | ||
1098 | 1 | avi->non_interleaved = 0; | |
1099 | } | ||
1100 | |||
1101 |
2/2✓ Branch 0 taken 7 times.
✓ Branch 1 taken 461 times.
|
468 | if (avi->non_interleaved) { |
1102 | 7 | av_log(s, AV_LOG_INFO, "non-interleaved AVI\n"); | |
1103 | 7 | clean_index(s); | |
1104 | } | ||
1105 | |||
1106 | 468 | ff_metadata_conv_ctx(s, NULL, avi_metadata_conv); | |
1107 | 468 | ff_metadata_conv_ctx(s, NULL, ff_riff_info_conv); | |
1108 | |||
1109 | 468 | return 0; | |
1110 | } | ||
1111 | |||
1112 | ✗ | static int read_gab2_sub(AVFormatContext *s, AVStream *st, AVPacket *pkt) | |
1113 | { | ||
1114 | ✗ | if (pkt->size >= 7 && | |
1115 | ✗ | pkt->size < INT_MAX - AVPROBE_PADDING_SIZE && | |
1116 | ✗ | !strcmp(pkt->data, "GAB2") && AV_RL16(pkt->data + 5) == 2) { | |
1117 | uint8_t desc[256]; | ||
1118 | ✗ | int score = AVPROBE_SCORE_EXTENSION, ret; | |
1119 | ✗ | AVIStream *ast = st->priv_data; | |
1120 | const AVInputFormat *sub_demuxer; | ||
1121 | AVRational time_base; | ||
1122 | int size; | ||
1123 | AVProbeData pd; | ||
1124 | unsigned int desc_len; | ||
1125 | ✗ | AVIOContext *pb = avio_alloc_context(pkt->data + 7, | |
1126 | ✗ | pkt->size - 7, | |
1127 | 0, NULL, NULL, NULL, NULL); | ||
1128 | ✗ | if (!pb) | |
1129 | ✗ | goto error; | |
1130 | |||
1131 | ✗ | desc_len = avio_rl32(pb); | |
1132 | |||
1133 | ✗ | if (desc_len > pb->buf_end - pb->buf_ptr) | |
1134 | ✗ | goto error; | |
1135 | |||
1136 | ✗ | ret = avio_get_str16le(pb, desc_len, desc, sizeof(desc)); | |
1137 | ✗ | avio_skip(pb, desc_len - ret); | |
1138 | ✗ | if (*desc) | |
1139 | ✗ | av_dict_set(&st->metadata, "title", desc, 0); | |
1140 | |||
1141 | ✗ | avio_rl16(pb); /* flags? */ | |
1142 | ✗ | avio_rl32(pb); /* data size */ | |
1143 | |||
1144 | ✗ | size = pb->buf_end - pb->buf_ptr; | |
1145 | ✗ | pd = (AVProbeData) { .buf = av_mallocz(size + AVPROBE_PADDING_SIZE), | |
1146 | .buf_size = size }; | ||
1147 | ✗ | if (!pd.buf) | |
1148 | ✗ | goto error; | |
1149 | ✗ | memcpy(pd.buf, pb->buf_ptr, size); | |
1150 | ✗ | sub_demuxer = av_probe_input_format2(&pd, 1, &score); | |
1151 | ✗ | av_freep(&pd.buf); | |
1152 | ✗ | if (!sub_demuxer) | |
1153 | ✗ | goto error; | |
1154 | |||
1155 | ✗ | if (strcmp(sub_demuxer->name, "srt") && strcmp(sub_demuxer->name, "ass")) | |
1156 | ✗ | goto error; | |
1157 | |||
1158 | ✗ | if (!(ast->sub_pkt = av_packet_alloc())) | |
1159 | ✗ | goto error; | |
1160 | |||
1161 | ✗ | if (!(ast->sub_ctx = avformat_alloc_context())) | |
1162 | ✗ | goto error; | |
1163 | |||
1164 | ✗ | ast->sub_ctx->pb = pb; | |
1165 | |||
1166 | ✗ | if (ff_copy_whiteblacklists(ast->sub_ctx, s) < 0) | |
1167 | ✗ | goto error; | |
1168 | |||
1169 | ✗ | if (!avformat_open_input(&ast->sub_ctx, "", sub_demuxer, NULL)) { | |
1170 | ✗ | if (ast->sub_ctx->nb_streams != 1) | |
1171 | ✗ | goto error; | |
1172 | ✗ | ff_read_packet(ast->sub_ctx, ast->sub_pkt); | |
1173 | ✗ | avcodec_parameters_copy(st->codecpar, ast->sub_ctx->streams[0]->codecpar); | |
1174 | ✗ | time_base = ast->sub_ctx->streams[0]->time_base; | |
1175 | ✗ | avpriv_set_pts_info(st, 64, time_base.num, time_base.den); | |
1176 | } | ||
1177 | ✗ | ast->sub_buffer = pkt->buf; | |
1178 | ✗ | pkt->buf = NULL; | |
1179 | ✗ | av_packet_unref(pkt); | |
1180 | ✗ | return 1; | |
1181 | |||
1182 | ✗ | error: | |
1183 | ✗ | av_packet_free(&ast->sub_pkt); | |
1184 | ✗ | av_freep(&ast->sub_ctx); | |
1185 | ✗ | avio_context_free(&pb); | |
1186 | } | ||
1187 | ✗ | return 0; | |
1188 | } | ||
1189 | |||
1190 | 27148 | static AVStream *get_subtitle_pkt(AVFormatContext *s, AVStream *next_st, | |
1191 | AVPacket *pkt) | ||
1192 | { | ||
1193 | 27148 | AVIStream *ast, *next_ast = next_st->priv_data; | |
1194 | 27148 | int64_t ts, next_ts, ts_min = INT64_MAX; | |
1195 | 27148 | AVStream *st, *sub_st = NULL; | |
1196 | int i; | ||
1197 | |||
1198 | 27148 | next_ts = av_rescale_q(next_ast->frame_offset, next_st->time_base, | |
1199 | 27148 | AV_TIME_BASE_Q); | |
1200 | |||
1201 |
2/2✓ Branch 0 taken 33833 times.
✓ Branch 1 taken 27148 times.
|
60981 | for (i = 0; i < s->nb_streams; i++) { |
1202 | 33833 | st = s->streams[i]; | |
1203 | 33833 | ast = st->priv_data; | |
1204 |
4/8✓ Branch 0 taken 29352 times.
✓ Branch 1 taken 4481 times.
✓ Branch 2 taken 29352 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 29352 times.
✗ Branch 6 not taken.
✗ Branch 7 not taken.
|
33833 | if (st->discard < AVDISCARD_ALL && ast && ast->sub_pkt && ast->sub_pkt->data) { |
1205 | ✗ | ts = av_rescale_q(ast->sub_pkt->dts, st->time_base, AV_TIME_BASE_Q); | |
1206 | ✗ | if (ts <= next_ts && ts < ts_min) { | |
1207 | ✗ | ts_min = ts; | |
1208 | ✗ | sub_st = st; | |
1209 | } | ||
1210 | } | ||
1211 | } | ||
1212 | |||
1213 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 27148 times.
|
27148 | if (sub_st) { |
1214 | ✗ | ast = sub_st->priv_data; | |
1215 | ✗ | av_packet_move_ref(pkt, ast->sub_pkt); | |
1216 | ✗ | pkt->stream_index = sub_st->index; | |
1217 | |||
1218 | ✗ | if (ff_read_packet(ast->sub_ctx, ast->sub_pkt) < 0) | |
1219 | ✗ | ast->sub_pkt->data = NULL; | |
1220 | } | ||
1221 | 27148 | return sub_st; | |
1222 | } | ||
1223 | |||
1224 | 517500 | static int get_stream_idx(const unsigned *d) | |
1225 | { | ||
1226 |
4/4✓ Branch 0 taken 293780 times.
✓ Branch 1 taken 223720 times.
✓ Branch 2 taken 94121 times.
✓ Branch 3 taken 199659 times.
|
517500 | if (d[0] >= '0' && d[0] <= '9' && |
1227 |
4/4✓ Branch 0 taken 93761 times.
✓ Branch 1 taken 360 times.
✓ Branch 2 taken 63151 times.
✓ Branch 3 taken 30610 times.
|
94121 | d[1] >= '0' && d[1] <= '9') { |
1228 | 63151 | return (d[0] - '0') * 10 + (d[1] - '0'); | |
1229 | } else { | ||
1230 | 454349 | return 100; // invalid stream ID | |
1231 | } | ||
1232 | } | ||
1233 | |||
1234 | /** | ||
1235 | * | ||
1236 | * @param exit_early set to 1 to just gather packet position without making the changes needed to actually read & return the packet | ||
1237 | */ | ||
1238 | 27221 | static int avi_sync(AVFormatContext *s, int exit_early) | |
1239 | { | ||
1240 | 27221 | AVIContext *avi = s->priv_data; | |
1241 | 27221 | AVIOContext *pb = s->pb; | |
1242 | int n; | ||
1243 | unsigned int d[8]; | ||
1244 | unsigned int size; | ||
1245 | int64_t i, sync; | ||
1246 | |||
1247 | 4873 | start_sync: | |
1248 | 32094 | memset(d, -1, sizeof(d)); | |
1249 |
2/2✓ Branch 2 taken 363452 times.
✓ Branch 3 taken 479 times.
|
363931 | for (i = sync = avio_tell(pb); !avio_feof(pb); i++) { |
1250 | int j; | ||
1251 | |||
1252 |
2/2✓ Branch 0 taken 2544164 times.
✓ Branch 1 taken 363452 times.
|
2907616 | for (j = 0; j < 7; j++) |
1253 | 2544164 | d[j] = d[j + 1]; | |
1254 | 363452 | d[7] = avio_r8(pb); | |
1255 | |||
1256 | 363452 | size = d[4] + (d[5] << 8) + (d[6] << 16) + (d[7] << 24); | |
1257 | |||
1258 | 363452 | n = get_stream_idx(d + 2); | |
1259 | ff_tlog(s, "%X %X %X %X %X %X %X %X %"PRId64" %u %d\n", | ||
1260 | d[0], d[1], d[2], d[3], d[4], d[5], d[6], d[7], i, size, n); | ||
1261 |
4/4✓ Branch 0 taken 126806 times.
✓ Branch 1 taken 236646 times.
✓ Branch 2 taken 13725 times.
✓ Branch 3 taken 113081 times.
|
363452 | if (i*(avi->io_fsize>0) + (uint64_t)size > avi->fsize || d[0] > 127) |
1262 | 250371 | continue; | |
1263 | |||
1264 | // parse ix## | ||
1265 |
5/6✓ Branch 0 taken 472 times.
✓ Branch 1 taken 112609 times.
✓ Branch 2 taken 1 times.
✓ Branch 3 taken 471 times.
✗ Branch 4 not taken.
✓ Branch 5 taken 1 times.
|
113081 | if ((d[0] == 'i' && d[1] == 'x' && n < s->nb_streams) || |
1266 | // parse JUNK | ||
1267 |
6/8✓ Branch 0 taken 468 times.
✓ Branch 1 taken 112612 times.
✓ Branch 2 taken 467 times.
✓ Branch 3 taken 1 times.
✓ Branch 4 taken 467 times.
✗ Branch 5 not taken.
✗ Branch 6 not taken.
✓ Branch 7 taken 467 times.
|
113080 | (d[0] == 'J' && d[1] == 'U' && d[2] == 'N' && d[3] == 'K') || |
1268 |
6/8✓ Branch 0 taken 471 times.
✓ Branch 1 taken 112142 times.
✓ Branch 2 taken 470 times.
✓ Branch 3 taken 1 times.
✓ Branch 4 taken 470 times.
✗ Branch 5 not taken.
✗ Branch 6 not taken.
✓ Branch 7 taken 470 times.
|
112613 | (d[0] == 'i' && d[1] == 'd' && d[2] == 'x' && d[3] == '1') || |
1269 |
4/8✓ Branch 0 taken 1 times.
✓ Branch 1 taken 112142 times.
✓ Branch 2 taken 1 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 1 times.
✗ Branch 6 not taken.
✗ Branch 7 not taken.
|
112143 | (d[0] == 'i' && d[1] == 'n' && d[2] == 'd' && d[3] == 'x')) { |
1270 | 938 | avio_skip(pb, size); | |
1271 | 938 | goto start_sync; | |
1272 | } | ||
1273 | |||
1274 | // parse stray LIST | ||
1275 |
6/8✓ Branch 0 taken 656 times.
✓ Branch 1 taken 111487 times.
✓ Branch 2 taken 655 times.
✓ Branch 3 taken 1 times.
✓ Branch 4 taken 655 times.
✗ Branch 5 not taken.
✓ Branch 6 taken 655 times.
✗ Branch 7 not taken.
|
112143 | if (d[0] == 'L' && d[1] == 'I' && d[2] == 'S' && d[3] == 'T') { |
1276 | 655 | avio_skip(pb, 4); | |
1277 | 655 | goto start_sync; | |
1278 | } | ||
1279 | |||
1280 | 111488 | n = get_stream_idx(d); | |
1281 | |||
1282 |
2/2✓ Branch 0 taken 41728 times.
✓ Branch 1 taken 69760 times.
|
111488 | if (!((i - avi->last_pkt_pos) & 1) && |
1283 |
2/2✓ Branch 1 taken 968 times.
✓ Branch 2 taken 40760 times.
|
41728 | get_stream_idx(d + 1) < s->nb_streams) |
1284 | 968 | continue; | |
1285 | |||
1286 | // detect ##ix chunk and skip | ||
1287 |
1/6✗ Branch 0 not taken.
✓ Branch 1 taken 110520 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
|
110520 | if (d[2] == 'i' && d[3] == 'x' && n < s->nb_streams) { |
1288 | ✗ | avio_skip(pb, size); | |
1289 | ✗ | goto start_sync; | |
1290 | } | ||
1291 | |||
1292 |
3/6✓ Branch 0 taken 6090 times.
✓ Branch 1 taken 104430 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 6090 times.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
|
110520 | if (d[2] == 'w' && d[3] == 'c' && n < s->nb_streams) { |
1293 | ✗ | avio_skip(pb, 16 * 3 + 8); | |
1294 | ✗ | goto start_sync; | |
1295 | } | ||
1296 | |||
1297 |
1/4✗ Branch 0 not taken.
✓ Branch 1 taken 110520 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
|
110520 | if (avi->dv_demux && n != 0) |
1298 | ✗ | continue; | |
1299 | |||
1300 | // parse ##dc/##wb | ||
1301 |
2/2✓ Branch 0 taken 30022 times.
✓ Branch 1 taken 80498 times.
|
110520 | if (n < s->nb_streams) { |
1302 | AVStream *st; | ||
1303 | AVIStream *ast; | ||
1304 | 30022 | st = s->streams[n]; | |
1305 | 30022 | ast = st->priv_data; | |
1306 | |||
1307 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 30022 times.
|
30022 | if (!ast) { |
1308 | ✗ | av_log(s, AV_LOG_WARNING, "Skipping foreign stream %d packet\n", n); | |
1309 | ✗ | continue; | |
1310 | } | ||
1311 | |||
1312 |
2/2✓ Branch 0 taken 9416 times.
✓ Branch 1 taken 20606 times.
|
30022 | if (s->nb_streams >= 2) { |
1313 | 9416 | AVStream *st1 = s->streams[1]; | |
1314 | 9416 | AVIStream *ast1 = st1->priv_data; | |
1315 | // workaround for broken small-file-bug402.avi | ||
1316 |
4/6✓ Branch 0 taken 9416 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 3485 times.
✓ Branch 3 taken 5931 times.
✓ Branch 4 taken 3485 times.
✗ Branch 5 not taken.
|
9416 | if (ast1 && d[2] == 'w' && d[3] == 'b' |
1317 |
2/2✓ Branch 0 taken 314 times.
✓ Branch 1 taken 3171 times.
|
3485 | && n == 0 |
1318 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 314 times.
|
314 | && st ->codecpar->codec_type == AVMEDIA_TYPE_VIDEO |
1319 | ✗ | && st1->codecpar->codec_type == AVMEDIA_TYPE_AUDIO | |
1320 | ✗ | && ast->prefix == 'd'*256+'c' | |
1321 | ✗ | && (d[2]*256+d[3] == ast1->prefix || !ast1->prefix_count) | |
1322 | ) { | ||
1323 | ✗ | n = 1; | |
1324 | ✗ | st = st1; | |
1325 | ✗ | ast = ast1; | |
1326 | ✗ | av_log(s, AV_LOG_WARNING, | |
1327 | "Invalid stream + prefix combination, assuming audio.\n"); | ||
1328 | } | ||
1329 | } | ||
1330 | |||
1331 |
4/6✓ Branch 0 taken 4 times.
✓ Branch 1 taken 30018 times.
✓ Branch 2 taken 4 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 4 times.
✗ Branch 5 not taken.
|
30022 | if (d[2] == 'p' && d[3] == 'c' && size <= 4 * 256 + 4) { |
1332 | 4 | int k = avio_r8(pb); | |
1333 | 4 | int last = (k + avio_r8(pb) - 1) & 0xFF; | |
1334 | |||
1335 | 4 | avio_rl16(pb); // flags | |
1336 | |||
1337 | // b + (g << 8) + (r << 16); | ||
1338 |
2/2✓ Branch 0 taken 759 times.
✓ Branch 1 taken 4 times.
|
763 | for (; k <= last; k++) |
1339 | 759 | ast->pal[k] = 0xFFU<<24 | avio_rb32(pb)>>8; | |
1340 | |||
1341 | 4 | ast->has_pal = 1; | |
1342 | 4 | goto start_sync; | |
1343 |
4/4✓ Branch 0 taken 26609 times.
✓ Branch 1 taken 3409 times.
✓ Branch 2 taken 26595 times.
✓ Branch 3 taken 14 times.
|
30018 | } else if (((ast->prefix_count < 5 || sync + 9 > i) && |
1344 |
2/4✓ Branch 0 taken 30004 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 30004 times.
|
30004 | d[2] < 128 && d[3] < 128) || |
1345 |
1/2✓ Branch 0 taken 14 times.
✗ Branch 1 not taken.
|
14 | d[2] * 256 + d[3] == ast->prefix /* || |
1346 | (d[2] == 'd' && d[3] == 'c') || | ||
1347 | (d[2] == 'w' && d[3] == 'b') */) { | ||
1348 |
2/2✓ Branch 0 taken 425 times.
✓ Branch 1 taken 29593 times.
|
30018 | if (exit_early) |
1349 | 425 | return 0; | |
1350 |
2/2✓ Branch 0 taken 29011 times.
✓ Branch 1 taken 582 times.
|
29593 | if (d[2] * 256 + d[3] == ast->prefix) |
1351 | 29011 | ast->prefix_count++; | |
1352 | else { | ||
1353 | 582 | ast->prefix = d[2] * 256 + d[3]; | |
1354 | 582 | ast->prefix_count = 0; | |
1355 | } | ||
1356 | |||
1357 |
1/2✓ Branch 0 taken 29593 times.
✗ Branch 1 not taken.
|
29593 | if (!avi->dv_demux && |
1358 |
4/4✓ Branch 0 taken 4701 times.
✓ Branch 1 taken 24892 times.
✓ Branch 2 taken 3899 times.
✓ Branch 3 taken 802 times.
|
29593 | ((st->discard >= AVDISCARD_DEFAULT && size == 0) /* || |
1359 | // FIXME: needs a little reordering | ||
1360 | (st->discard >= AVDISCARD_NONKEY && | ||
1361 | !(pkt->flags & AV_PKT_FLAG_KEY)) */ | ||
1362 |
2/2✓ Branch 0 taken 2474 times.
✓ Branch 1 taken 26317 times.
|
28791 | || st->discard >= AVDISCARD_ALL)) { |
1363 | |||
1364 | 3276 | ast->frame_offset += get_duration(ast, size); | |
1365 | 3276 | avio_skip(pb, size); | |
1366 | 3276 | goto start_sync; | |
1367 | } | ||
1368 | |||
1369 | 26317 | avi->stream_index = n; | |
1370 | 26317 | ast->packet_size = size + 8; | |
1371 | 26317 | ast->remaining = size; | |
1372 | |||
1373 |
2/2✓ Branch 0 taken 24870 times.
✓ Branch 1 taken 1447 times.
|
26317 | if (size) { |
1374 | 24870 | FFStream *const sti = ffstream(st); | |
1375 | 24870 | uint64_t pos = avio_tell(pb) - 8; | |
1376 |
3/4✓ Branch 0 taken 24815 times.
✓ Branch 1 taken 55 times.
✓ Branch 2 taken 24815 times.
✗ Branch 3 not taken.
|
24870 | if (!sti->index_entries || !sti->nb_index_entries || |
1377 |
2/2✓ Branch 0 taken 3036 times.
✓ Branch 1 taken 21779 times.
|
24815 | sti->index_entries[sti->nb_index_entries - 1].pos < pos) { |
1378 | 3091 | av_add_index_entry(st, pos, ast->frame_offset, size, | |
1379 | 0, AVINDEX_KEYFRAME); | ||
1380 | } | ||
1381 | } | ||
1382 | 26317 | return 0; | |
1383 | } | ||
1384 | } | ||
1385 | } | ||
1386 | |||
1387 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 479 times.
|
479 | if (pb->error) |
1388 | ✗ | return pb->error; | |
1389 | 479 | return AVERROR_EOF; | |
1390 | } | ||
1391 | |||
1392 | 766 | static int ni_prepare_read(AVFormatContext *s) | |
1393 | { | ||
1394 | 766 | AVIContext *avi = s->priv_data; | |
1395 | 766 | int best_stream_index = 0; | |
1396 | 766 | AVStream *best_st = NULL; | |
1397 | FFStream *best_sti; | ||
1398 | AVIStream *best_ast; | ||
1399 | 766 | int64_t best_ts = INT64_MAX; | |
1400 | int i; | ||
1401 | |||
1402 |
2/2✓ Branch 0 taken 1288 times.
✓ Branch 1 taken 766 times.
|
2054 | for (i = 0; i < s->nb_streams; i++) { |
1403 | 1288 | AVStream *st = s->streams[i]; | |
1404 | 1288 | FFStream *const sti = ffstream(st); | |
1405 | 1288 | AVIStream *ast = st->priv_data; | |
1406 | 1288 | int64_t ts = ast->frame_offset; | |
1407 | int64_t last_ts; | ||
1408 | |||
1409 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1288 times.
|
1288 | if (!sti->nb_index_entries) |
1410 | 16 | continue; | |
1411 | |||
1412 | 1288 | last_ts = sti->index_entries[sti->nb_index_entries - 1].timestamp; | |
1413 |
4/4✓ Branch 0 taken 858 times.
✓ Branch 1 taken 430 times.
✓ Branch 2 taken 16 times.
✓ Branch 3 taken 842 times.
|
1288 | if (!ast->remaining && ts > last_ts) |
1414 | 16 | continue; | |
1415 | |||
1416 | 1272 | ts = av_rescale_q(ts, st->time_base, | |
1417 | 1272 | (AVRational) { FFMAX(1, ast->sample_size), | |
1418 | AV_TIME_BASE }); | ||
1419 | |||
1420 | 1272 | av_log(s, AV_LOG_TRACE, "%"PRId64" %d/%d %"PRId64"\n", ts, | |
1421 | st->time_base.num, st->time_base.den, ast->frame_offset); | ||
1422 |
2/2✓ Branch 0 taken 975 times.
✓ Branch 1 taken 297 times.
|
1272 | if (ts < best_ts) { |
1423 | 975 | best_ts = ts; | |
1424 | 975 | best_st = st; | |
1425 | 975 | best_stream_index = i; | |
1426 | } | ||
1427 | } | ||
1428 |
2/2✓ Branch 0 taken 5 times.
✓ Branch 1 taken 761 times.
|
766 | if (!best_st) |
1429 | 5 | return AVERROR_EOF; | |
1430 | |||
1431 | 761 | best_sti = ffstream(best_st); | |
1432 | 761 | best_ast = best_st->priv_data; | |
1433 | 761 | best_ts = best_ast->frame_offset; | |
1434 |
2/2✓ Branch 0 taken 181 times.
✓ Branch 1 taken 580 times.
|
761 | if (best_ast->remaining) { |
1435 | 181 | i = av_index_search_timestamp(best_st, | |
1436 | best_ts, | ||
1437 | AVSEEK_FLAG_ANY | | ||
1438 | AVSEEK_FLAG_BACKWARD); | ||
1439 | } else { | ||
1440 | 580 | i = av_index_search_timestamp(best_st, best_ts, AVSEEK_FLAG_ANY); | |
1441 |
1/2✓ Branch 0 taken 580 times.
✗ Branch 1 not taken.
|
580 | if (i >= 0) |
1442 | 580 | best_ast->frame_offset = best_sti->index_entries[i].timestamp; | |
1443 | } | ||
1444 | |||
1445 |
1/2✓ Branch 0 taken 761 times.
✗ Branch 1 not taken.
|
761 | if (i >= 0) { |
1446 | 761 | int64_t pos = best_sti->index_entries[i].pos; | |
1447 | 761 | pos += best_ast->packet_size - best_ast->remaining; | |
1448 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 761 times.
|
761 | if (avio_seek(s->pb, pos + 8, SEEK_SET) < 0) |
1449 | ✗ | return AVERROR_EOF; | |
1450 | |||
1451 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 761 times.
|
761 | av_assert0(best_ast->remaining <= best_ast->packet_size); |
1452 | |||
1453 | 761 | avi->stream_index = best_stream_index; | |
1454 |
2/2✓ Branch 0 taken 580 times.
✓ Branch 1 taken 181 times.
|
761 | if (!best_ast->remaining) |
1455 | 580 | best_ast->packet_size = | |
1456 | 580 | best_ast->remaining = best_sti->index_entries[i].size; | |
1457 | } | ||
1458 | else | ||
1459 | ✗ | return AVERROR_EOF; | |
1460 | |||
1461 | 761 | return 0; | |
1462 | } | ||
1463 | |||
1464 | 27631 | static int avi_read_packet(AVFormatContext *s, AVPacket *pkt) | |
1465 | { | ||
1466 | 27631 | AVIContext *avi = s->priv_data; | |
1467 | 27631 | AVIOContext *pb = s->pb; | |
1468 | int err; | ||
1469 | |||
1470 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 27631 times.
|
27631 | if (CONFIG_DV_DEMUXER && avi->dv_demux) { |
1471 | ✗ | int size = avpriv_dv_get_packet(avi->dv_demux, pkt); | |
1472 | ✗ | if (size >= 0) | |
1473 | ✗ | return size; | |
1474 | else | ||
1475 | ✗ | goto resync; | |
1476 | } | ||
1477 | |||
1478 |
2/2✓ Branch 0 taken 26865 times.
✓ Branch 1 taken 766 times.
|
27631 | if (avi->non_interleaved) { |
1479 | 766 | err = ni_prepare_read(s); | |
1480 |
2/2✓ Branch 0 taken 761 times.
✓ Branch 1 taken 5 times.
|
766 | if (err < 0) |
1481 | 5 | return err; | |
1482 | } | ||
1483 | |||
1484 | 53944 | resync: | |
1485 |
2/2✓ Branch 0 taken 27148 times.
✓ Branch 1 taken 26796 times.
|
53944 | if (avi->stream_index >= 0) { |
1486 | 27148 | AVStream *st = s->streams[avi->stream_index]; | |
1487 | 27148 | FFStream *const sti = ffstream(st); | |
1488 | 27148 | AVIStream *ast = st->priv_data; | |
1489 | 27148 | int dv_demux = CONFIG_DV_DEMUXER && avi->dv_demux; | |
1490 | int size, err; | ||
1491 | |||
1492 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 27148 times.
|
27148 | if (get_subtitle_pkt(s, st, pkt)) |
1493 | ✗ | return 0; | |
1494 | |||
1495 | // minorityreport.AVI block_align=1024 sample_size=1 IMA-ADPCM | ||
1496 |
2/2✓ Branch 0 taken 23766 times.
✓ Branch 1 taken 3382 times.
|
27148 | if (ast->sample_size <= 1) |
1497 | 23766 | size = INT_MAX; | |
1498 |
2/2✓ Branch 0 taken 238 times.
✓ Branch 1 taken 3144 times.
|
3382 | else if (ast->sample_size < 32) |
1499 | // arbitrary multiplier to avoid tiny packets for raw PCM data | ||
1500 | 238 | size = 1024 * ast->sample_size; | |
1501 | else | ||
1502 | 3144 | size = ast->sample_size; | |
1503 | |||
1504 |
2/2✓ Branch 0 taken 23777 times.
✓ Branch 1 taken 3371 times.
|
27148 | if (size > ast->remaining) |
1505 | 23777 | size = ast->remaining; | |
1506 | 27148 | avi->last_pkt_pos = avio_tell(pb); | |
1507 | 27148 | err = av_get_packet(pb, pkt, size); | |
1508 |
2/2✓ Branch 0 taken 37 times.
✓ Branch 1 taken 27111 times.
|
27148 | if (err < 0) |
1509 | 37 | return err; | |
1510 | 27111 | size = err; | |
1511 | |||
1512 |
4/6✓ Branch 0 taken 19 times.
✓ Branch 1 taken 27092 times.
✓ Branch 2 taken 19 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 19 times.
✗ Branch 5 not taken.
|
27111 | if (ast->has_pal && pkt->size < (unsigned)INT_MAX / 2 && !dv_demux) { |
1513 | uint8_t *pal; | ||
1514 | 19 | pal = av_packet_new_side_data(pkt, | |
1515 | AV_PKT_DATA_PALETTE, | ||
1516 | AVPALETTE_SIZE); | ||
1517 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 19 times.
|
19 | if (!pal) { |
1518 | ✗ | av_log(s, AV_LOG_ERROR, | |
1519 | "Failed to allocate data for palette\n"); | ||
1520 | } else { | ||
1521 | 19 | memcpy(pal, ast->pal, AVPALETTE_SIZE); | |
1522 | 19 | ast->has_pal = 0; | |
1523 | } | ||
1524 | } | ||
1525 | |||
1526 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 27111 times.
|
27111 | if (CONFIG_DV_DEMUXER && dv_demux) { |
1527 | ✗ | size = avpriv_dv_produce_packet(avi->dv_demux, pkt, | |
1528 | pkt->data, pkt->size, pkt->pos); | ||
1529 | ✗ | pkt->flags |= AV_PKT_FLAG_KEY; | |
1530 | ✗ | if (size < 0) | |
1531 | ✗ | av_packet_unref(pkt); | |
1532 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 27111 times.
|
27111 | } else if (st->codecpar->codec_type == AVMEDIA_TYPE_SUBTITLE && |
1533 | ✗ | !st->codecpar->codec_tag && read_gab2_sub(s, st, pkt)) { | |
1534 | ✗ | ast->frame_offset++; | |
1535 | ✗ | avi->stream_index = -1; | |
1536 | ✗ | ast->remaining = 0; | |
1537 | ✗ | goto resync; | |
1538 | } else { | ||
1539 | /* XXX: How to handle B-frames in AVI? */ | ||
1540 | 27111 | pkt->dts = ast->frame_offset; | |
1541 | // pkt->dts += ast->start; | ||
1542 |
2/2✓ Branch 0 taken 3752 times.
✓ Branch 1 taken 23359 times.
|
27111 | if (ast->sample_size) |
1543 | 3752 | pkt->dts /= ast->sample_size; | |
1544 | 27111 | pkt->stream_index = avi->stream_index; | |
1545 | |||
1546 |
3/4✓ Branch 0 taken 22933 times.
✓ Branch 1 taken 4178 times.
✓ Branch 2 taken 22933 times.
✗ Branch 3 not taken.
|
50044 | if (st->codecpar->codec_type == AVMEDIA_TYPE_VIDEO && sti->index_entries) { |
1547 | AVIndexEntry *e; | ||
1548 | int index; | ||
1549 | |||
1550 | 22933 | index = av_index_search_timestamp(st, ast->frame_offset, AVSEEK_FLAG_ANY); | |
1551 | 22933 | e = &sti->index_entries[index]; | |
1552 | |||
1553 |
4/4✓ Branch 0 taken 22820 times.
✓ Branch 1 taken 113 times.
✓ Branch 2 taken 21485 times.
✓ Branch 3 taken 1335 times.
|
22933 | if (index >= 0 && e->timestamp == ast->frame_offset) { |
1554 |
2/2✓ Branch 0 taken 2524 times.
✓ Branch 1 taken 18961 times.
|
21485 | if (index == sti->nb_index_entries-1) { |
1555 | 2524 | int key=1; | |
1556 | 2524 | uint32_t state=-1; | |
1557 |
2/2✓ Branch 0 taken 52 times.
✓ Branch 1 taken 2472 times.
|
2524 | if (st->codecpar->codec_id == AV_CODEC_ID_MPEG4) { |
1558 | 52 | const uint8_t *ptr = pkt->data, *end = ptr + FFMIN(size, 256); | |
1559 |
1/2✓ Branch 0 taken 109 times.
✗ Branch 1 not taken.
|
109 | while (ptr < end) { |
1560 | 109 | ptr = avpriv_find_start_code(ptr, end, &state); | |
1561 |
3/4✓ Branch 0 taken 52 times.
✓ Branch 1 taken 57 times.
✓ Branch 2 taken 52 times.
✗ Branch 3 not taken.
|
109 | if (state == 0x1B6 && ptr < end) { |
1562 | 52 | key = !(*ptr & 0xC0); | |
1563 | 52 | break; | |
1564 | } | ||
1565 | } | ||
1566 | } | ||
1567 |
2/2✓ Branch 0 taken 39 times.
✓ Branch 1 taken 2485 times.
|
2524 | if (!key) |
1568 | 39 | e->flags &= ~AVINDEX_KEYFRAME; | |
1569 | } | ||
1570 |
2/2✓ Branch 0 taken 12396 times.
✓ Branch 1 taken 9089 times.
|
21485 | if (e->flags & AVINDEX_KEYFRAME) |
1571 | 12396 | pkt->flags |= AV_PKT_FLAG_KEY; | |
1572 | } | ||
1573 | } else { | ||
1574 | 4178 | pkt->flags |= AV_PKT_FLAG_KEY; | |
1575 | } | ||
1576 | 27111 | ast->frame_offset += get_duration(ast, pkt->size); | |
1577 | } | ||
1578 | 27111 | ast->remaining -= err; | |
1579 |
2/2✓ Branch 0 taken 26863 times.
✓ Branch 1 taken 248 times.
|
27111 | if (!ast->remaining) { |
1580 | 26863 | avi->stream_index = -1; | |
1581 | 26863 | ast->packet_size = 0; | |
1582 | } | ||
1583 | |||
1584 |
6/6✓ Branch 0 taken 26353 times.
✓ Branch 1 taken 758 times.
✓ Branch 2 taken 24906 times.
✓ Branch 3 taken 1447 times.
✓ Branch 4 taken 1 times.
✓ Branch 5 taken 24905 times.
|
27111 | if (!avi->non_interleaved && pkt->pos >= 0 && ast->seek_pos > pkt->pos) { |
1585 | 1 | av_packet_unref(pkt); | |
1586 | 1 | goto resync; | |
1587 | } | ||
1588 | 27110 | ast->seek_pos= 0; | |
1589 | |||
1590 |
6/6✓ Branch 0 taken 26352 times.
✓ Branch 1 taken 758 times.
✓ Branch 2 taken 26256 times.
✓ Branch 3 taken 96 times.
✓ Branch 4 taken 23326 times.
✓ Branch 5 taken 2930 times.
|
27110 | if (!avi->non_interleaved && sti->nb_index_entries > 1 && avi->index_loaded > 1) { |
1591 | 23326 | int64_t dts= av_rescale_q(pkt->dts, st->time_base, AV_TIME_BASE_Q); | |
1592 | |||
1593 |
2/2✓ Branch 0 taken 22636 times.
✓ Branch 1 taken 690 times.
|
23326 | if (avi->dts_max < dts) { |
1594 | 22636 | avi->dts_max = dts; | |
1595 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 690 times.
|
690 | } else if (avi->dts_max - (uint64_t)dts > 2*AV_TIME_BASE) { |
1596 | ✗ | avi->non_interleaved= 1; | |
1597 | ✗ | av_log(s, AV_LOG_INFO, "Switching to NI mode, due to poor interleaving\n"); | |
1598 | } | ||
1599 | } | ||
1600 | |||
1601 | 27110 | return 0; | |
1602 | } | ||
1603 | |||
1604 |
2/2✓ Branch 1 taken 479 times.
✓ Branch 2 taken 26317 times.
|
26796 | if ((err = avi_sync(s, 0)) < 0) |
1605 | 479 | return err; | |
1606 | 26317 | goto resync; | |
1607 | } | ||
1608 | |||
1609 | /* XXX: We make the implicit supposition that the positions are sorted | ||
1610 | * for each stream. */ | ||
1611 | 425 | static int avi_read_idx1(AVFormatContext *s, int size) | |
1612 | { | ||
1613 | 425 | AVIContext *avi = s->priv_data; | |
1614 | 425 | AVIOContext *pb = s->pb; | |
1615 | int nb_index_entries, i; | ||
1616 | AVStream *st; | ||
1617 | AVIStream *ast; | ||
1618 | int64_t pos; | ||
1619 | 425 | unsigned int index, tag, flags, len, first_packet = 1; | |
1620 | 425 | int64_t last_pos = -1; | |
1621 | 425 | unsigned last_idx = -1; | |
1622 | 425 | int64_t idx1_pos, first_packet_pos = 0, data_offset = 0; | |
1623 | 425 | int anykey = 0; | |
1624 | |||
1625 | 425 | nb_index_entries = size / 16; | |
1626 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 425 times.
|
425 | if (nb_index_entries <= 0) |
1627 | ✗ | return AVERROR_INVALIDDATA; | |
1628 | |||
1629 | 425 | idx1_pos = avio_tell(pb); | |
1630 | 425 | avio_seek(pb, avi->movi_list + 4, SEEK_SET); | |
1631 |
1/2✓ Branch 1 taken 425 times.
✗ Branch 2 not taken.
|
425 | if (avi_sync(s, 1) == 0) |
1632 | 425 | first_packet_pos = avio_tell(pb) - 8; | |
1633 | 425 | avi->stream_index = -1; | |
1634 | 425 | avio_seek(pb, idx1_pos, SEEK_SET); | |
1635 | |||
1636 |
3/4✓ Branch 0 taken 404 times.
✓ Branch 1 taken 21 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 404 times.
|
425 | if (s->nb_streams == 1 && s->streams[0]->codecpar->codec_tag == AV_RL32("MMES")) { |
1637 | ✗ | first_packet_pos = 0; | |
1638 | ✗ | data_offset = avi->movi_list; | |
1639 | } | ||
1640 | |||
1641 | /* Read the entries and sort them in each stream component. */ | ||
1642 |
2/2✓ Branch 0 taken 27891 times.
✓ Branch 1 taken 425 times.
|
28316 | for (i = 0; i < nb_index_entries; i++) { |
1643 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 27891 times.
|
27891 | if (avio_feof(pb)) |
1644 | ✗ | return -1; | |
1645 | |||
1646 | 27891 | tag = avio_rl32(pb); | |
1647 | 27891 | flags = avio_rl32(pb); | |
1648 | 27891 | pos = avio_rl32(pb); | |
1649 | 27891 | len = avio_rl32(pb); | |
1650 | 27891 | av_log(s, AV_LOG_TRACE, "%d: tag=0x%x flags=0x%x pos=0x%"PRIx64" len=%d/", | |
1651 | i, tag, flags, pos, len); | ||
1652 | |||
1653 | 27891 | index = ((tag & 0xff) - '0') * 10; | |
1654 | 27891 | index += (tag >> 8 & 0xff) - '0'; | |
1655 |
2/2✓ Branch 0 taken 290 times.
✓ Branch 1 taken 27601 times.
|
27891 | if (index >= s->nb_streams) |
1656 | 290 | continue; | |
1657 | 27601 | st = s->streams[index]; | |
1658 | 27601 | ast = st->priv_data; | |
1659 | |||
1660 | /* Skip 'xxpc' palette change entries in the index until a logic | ||
1661 | * to process these is properly implemented. */ | ||
1662 |
3/4✓ Branch 0 taken 2 times.
✓ Branch 1 taken 27599 times.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
|
27601 | if ((tag >> 16 & 0xff) == 'p' && (tag >> 24 & 0xff) == 'c') |
1663 | 2 | continue; | |
1664 | |||
1665 |
3/4✓ Branch 0 taken 425 times.
✓ Branch 1 taken 27174 times.
✓ Branch 2 taken 425 times.
✗ Branch 3 not taken.
|
27599 | if (first_packet && first_packet_pos) { |
1666 |
3/4✓ Branch 0 taken 1 times.
✓ Branch 1 taken 424 times.
✓ Branch 2 taken 1 times.
✗ Branch 3 not taken.
|
425 | if (avi->movi_list + 4 != pos || pos + 500 > first_packet_pos) |
1667 | 425 | data_offset = first_packet_pos - pos; | |
1668 | 425 | first_packet = 0; | |
1669 | } | ||
1670 | 27599 | pos += data_offset; | |
1671 | |||
1672 | 27599 | av_log(s, AV_LOG_TRACE, "%d cum_len=%"PRId64"\n", len, ast->cum_len); | |
1673 | |||
1674 | // even if we have only a single stream, we should | ||
1675 | // switch to non-interleaved to get correct timestamps | ||
1676 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 27599 times.
|
27599 | if (last_pos == pos) |
1677 | ✗ | avi->non_interleaved = 1; | |
1678 |
3/4✓ Branch 0 taken 27599 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 25370 times.
✓ Branch 3 taken 2229 times.
|
27599 | if (last_idx != pos && len) { |
1679 | 25370 | av_add_index_entry(st, pos, ast->cum_len, len, 0, | |
1680 | 25370 | (flags & AVIIF_INDEX) ? AVINDEX_KEYFRAME : 0); | |
1681 | 25370 | last_idx= pos; | |
1682 | } | ||
1683 | 27599 | ast->cum_len += get_duration(ast, len); | |
1684 | 27599 | last_pos = pos; | |
1685 | 27599 | anykey |= flags&AVIIF_INDEX; | |
1686 | } | ||
1687 |
2/2✓ Branch 0 taken 1 times.
✓ Branch 1 taken 424 times.
|
425 | if (!anykey) { |
1688 |
2/2✓ Branch 0 taken 1 times.
✓ Branch 1 taken 1 times.
|
2 | for (index = 0; index < s->nb_streams; index++) { |
1689 | 1 | FFStream *const sti = ffstream(s->streams[index]); | |
1690 |
1/2✓ Branch 0 taken 1 times.
✗ Branch 1 not taken.
|
1 | if (sti->nb_index_entries) |
1691 | 1 | sti->index_entries[0].flags |= AVINDEX_KEYFRAME; | |
1692 | } | ||
1693 | } | ||
1694 | 425 | return 0; | |
1695 | } | ||
1696 | |||
1697 | /* Scan the index and consider any file with streams more than | ||
1698 | * 2 seconds or 64MB apart non-interleaved. */ | ||
1699 | 463 | static int check_stream_max_drift(AVFormatContext *s) | |
1700 | { | ||
1701 | int64_t min_pos, pos; | ||
1702 | int i; | ||
1703 | 463 | int *idx = av_calloc(s->nb_streams, sizeof(*idx)); | |
1704 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 463 times.
|
463 | if (!idx) |
1705 | ✗ | return AVERROR(ENOMEM); | |
1706 |
2/2✓ Branch 0 taken 30903 times.
✓ Branch 1 taken 463 times.
|
31366 | for (min_pos = pos = 0; min_pos != INT64_MAX; pos = min_pos + 1ULL) { |
1707 | 30903 | int64_t max_dts = INT64_MIN / 2; | |
1708 | 30903 | int64_t min_dts = INT64_MAX / 2; | |
1709 | 30903 | int64_t max_buffer = 0; | |
1710 | |||
1711 | 30903 | min_pos = INT64_MAX; | |
1712 | |||
1713 |
2/2✓ Branch 0 taken 39380 times.
✓ Branch 1 taken 30903 times.
|
70283 | for (i = 0; i < s->nb_streams; i++) { |
1714 | 39380 | AVStream *st = s->streams[i]; | |
1715 | 39380 | AVIStream *ast = st->priv_data; | |
1716 | 39380 | FFStream *const sti = ffstream(st); | |
1717 | 39380 | int n = sti->nb_index_entries; | |
1718 |
4/4✓ Branch 0 taken 69075 times.
✓ Branch 1 taken 745 times.
✓ Branch 2 taken 30440 times.
✓ Branch 3 taken 38635 times.
|
69820 | while (idx[i] < n && sti->index_entries[idx[i]].pos < pos) |
1719 | 30440 | idx[i]++; | |
1720 |
2/2✓ Branch 0 taken 38635 times.
✓ Branch 1 taken 745 times.
|
39380 | if (idx[i] < n) { |
1721 | int64_t dts; | ||
1722 | 38635 | dts = av_rescale_q(sti->index_entries[idx[i]].timestamp / | |
1723 | 38635 | FFMAX(ast->sample_size, 1), | |
1724 | 38635 | st->time_base, AV_TIME_BASE_Q); | |
1725 | 38635 | min_dts = FFMIN(min_dts, dts); | |
1726 | 38635 | min_pos = FFMIN(min_pos, sti->index_entries[idx[i]].pos); | |
1727 | } | ||
1728 | } | ||
1729 |
2/2✓ Branch 0 taken 39380 times.
✓ Branch 1 taken 30903 times.
|
70283 | for (i = 0; i < s->nb_streams; i++) { |
1730 | 39380 | AVStream *st = s->streams[i]; | |
1731 | 39380 | FFStream *const sti = ffstream(st); | |
1732 | 39380 | AVIStream *ast = st->priv_data; | |
1733 | |||
1734 |
4/4✓ Branch 0 taken 38678 times.
✓ Branch 1 taken 702 times.
✓ Branch 2 taken 38235 times.
✓ Branch 3 taken 443 times.
|
39380 | if (idx[i] && min_dts != INT64_MAX / 2) { |
1735 | int64_t dts, delta_dts; | ||
1736 | 38235 | dts = av_rescale_q(sti->index_entries[idx[i] - 1].timestamp / | |
1737 | 38235 | FFMAX(ast->sample_size, 1), | |
1738 | 38235 | st->time_base, AV_TIME_BASE_Q); | |
1739 | 38235 | delta_dts = av_sat_sub64(dts, min_dts); | |
1740 | 38235 | max_dts = FFMAX(max_dts, dts); | |
1741 | 38235 | max_buffer = FFMAX(max_buffer, | |
1742 | av_rescale(delta_dts, | ||
1743 | st->codecpar->bit_rate, | ||
1744 | AV_TIME_BASE)); | ||
1745 | } | ||
1746 | } | ||
1747 |
2/4✓ Branch 1 taken 30903 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✓ Branch 4 taken 30903 times.
|
30903 | if (av_sat_sub64(max_dts, min_dts) > 2 * AV_TIME_BASE || |
1748 | max_buffer > 1024 * 1024 * 8 * 8) { | ||
1749 | ✗ | av_free(idx); | |
1750 | ✗ | return 1; | |
1751 | } | ||
1752 | } | ||
1753 | 463 | av_free(idx); | |
1754 | 463 | return 0; | |
1755 | } | ||
1756 | |||
1757 | 468 | static int guess_ni_flag(AVFormatContext *s) | |
1758 | { | ||
1759 | int i; | ||
1760 | 468 | int64_t last_start = 0; | |
1761 | 468 | int64_t first_end = INT64_MAX; | |
1762 | 468 | int64_t oldpos = avio_tell(s->pb); | |
1763 | |||
1764 |
2/2✓ Branch 0 taken 511 times.
✓ Branch 1 taken 468 times.
|
979 | for (i = 0; i < s->nb_streams; i++) { |
1765 | 511 | AVStream *st = s->streams[i]; | |
1766 | 511 | FFStream *const sti = ffstream(st); | |
1767 | 511 | int n = sti->nb_index_entries; | |
1768 | unsigned int size; | ||
1769 | |||
1770 |
2/2✓ Branch 0 taken 58 times.
✓ Branch 1 taken 453 times.
|
511 | if (n <= 0) |
1771 | 58 | continue; | |
1772 | |||
1773 |
2/2✓ Branch 0 taken 416 times.
✓ Branch 1 taken 37 times.
|
453 | if (n >= 2) { |
1774 | 416 | int64_t pos = sti->index_entries[0].pos; | |
1775 | unsigned tag[2]; | ||
1776 | 416 | avio_seek(s->pb, pos, SEEK_SET); | |
1777 | 416 | tag[0] = avio_r8(s->pb); | |
1778 | 416 | tag[1] = avio_r8(s->pb); | |
1779 | 416 | avio_rl16(s->pb); | |
1780 | 416 | size = avio_rl32(s->pb); | |
1781 |
3/4✓ Branch 1 taken 415 times.
✓ Branch 2 taken 1 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 415 times.
|
416 | if (get_stream_idx(tag) == i && pos + size > sti->index_entries[1].pos) |
1782 | ✗ | last_start = INT64_MAX; | |
1783 |
3/4✓ Branch 1 taken 415 times.
✓ Branch 2 taken 1 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 415 times.
|
416 | if (get_stream_idx(tag) == i && size == sti->index_entries[0].size + 8) |
1784 | ✗ | last_start = INT64_MAX; | |
1785 | } | ||
1786 | |||
1787 |
2/2✓ Branch 0 taken 443 times.
✓ Branch 1 taken 10 times.
|
453 | if (sti->index_entries[0].pos > last_start) |
1788 | 443 | last_start = sti->index_entries[0].pos; | |
1789 |
2/2✓ Branch 0 taken 440 times.
✓ Branch 1 taken 13 times.
|
453 | if (sti->index_entries[n - 1].pos < first_end) |
1790 | 440 | first_end = sti->index_entries[n - 1].pos; | |
1791 | } | ||
1792 | 468 | avio_seek(s->pb, oldpos, SEEK_SET); | |
1793 | |||
1794 |
2/2✓ Branch 0 taken 5 times.
✓ Branch 1 taken 463 times.
|
468 | if (last_start > first_end) |
1795 | 5 | return 1; | |
1796 | |||
1797 | 463 | return check_stream_max_drift(s); | |
1798 | } | ||
1799 | |||
1800 | 462 | static int avi_load_index(AVFormatContext *s) | |
1801 | { | ||
1802 | 462 | AVIContext *avi = s->priv_data; | |
1803 | 462 | AVIOContext *pb = s->pb; | |
1804 | uint32_t tag, size; | ||
1805 | 462 | int64_t pos = avio_tell(pb); | |
1806 | int64_t next; | ||
1807 | 462 | int ret = -1; | |
1808 | |||
1809 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 462 times.
|
462 | if (avio_seek(pb, avi->movi_end, SEEK_SET) < 0) |
1810 | ✗ | goto the_end; // maybe truncated file | |
1811 | 462 | av_log(s, AV_LOG_TRACE, "movi_end=0x%"PRIx64"\n", avi->movi_end); | |
1812 | for (;;) { | ||
1813 | 896 | tag = avio_rl32(pb); | |
1814 | 896 | size = avio_rl32(pb); | |
1815 |
2/2✓ Branch 1 taken 440 times.
✓ Branch 2 taken 456 times.
|
896 | if (avio_feof(pb)) |
1816 | 440 | break; | |
1817 | 456 | next = avio_tell(pb); | |
1818 |
2/4✓ Branch 0 taken 456 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 456 times.
|
456 | if (next < 0 || next > INT64_MAX - size - (size & 1)) |
1819 | break; | ||
1820 | 456 | next += size + (size & 1LL); | |
1821 | |||
1822 |
3/4✓ Branch 0 taken 425 times.
✓ Branch 1 taken 31 times.
✓ Branch 2 taken 425 times.
✗ Branch 3 not taken.
|
881 | if (tag == MKTAG('i', 'd', 'x', '1') && |
1823 | 425 | avi_read_idx1(s, size) >= 0) { | |
1824 | 425 | avi->index_loaded=2; | |
1825 | 425 | ret = 0; | |
1826 |
2/2✓ Branch 0 taken 9 times.
✓ Branch 1 taken 22 times.
|
31 | }else if (tag == MKTAG('L', 'I', 'S', 'T')) { |
1827 | 9 | uint32_t tag1 = avio_rl32(pb); | |
1828 | |||
1829 |
2/2✓ Branch 0 taken 5 times.
✓ Branch 1 taken 4 times.
|
9 | if (tag1 == MKTAG('I', 'N', 'F', 'O')) |
1830 | 5 | ff_read_riff_info(s, size - 4); | |
1831 |
1/2✓ Branch 0 taken 22 times.
✗ Branch 1 not taken.
|
22 | }else if (!ret) |
1832 | 22 | break; | |
1833 | |||
1834 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 434 times.
|
434 | if (avio_seek(pb, next, SEEK_SET) < 0) |
1835 | ✗ | break; // something is wrong here | |
1836 | } | ||
1837 | |||
1838 | ✗ | the_end: | |
1839 | 462 | avio_seek(pb, pos, SEEK_SET); | |
1840 | 462 | return ret; | |
1841 | } | ||
1842 | |||
1843 | ✗ | static void seek_subtitle(AVStream *st, AVStream *st2, int64_t timestamp) | |
1844 | { | ||
1845 | ✗ | AVIStream *ast2 = st2->priv_data; | |
1846 | ✗ | int64_t ts2 = av_rescale_q(timestamp, st->time_base, st2->time_base); | |
1847 | ✗ | av_packet_unref(ast2->sub_pkt); | |
1848 | ✗ | if (avformat_seek_file(ast2->sub_ctx, 0, INT64_MIN, ts2, ts2, 0) >= 0 || | |
1849 | ✗ | avformat_seek_file(ast2->sub_ctx, 0, ts2, ts2, INT64_MAX, 0) >= 0) | |
1850 | ✗ | ff_read_packet(ast2->sub_ctx, ast2->sub_pkt); | |
1851 | ✗ | } | |
1852 | |||
1853 | 728 | static int avi_read_seek(AVFormatContext *s, int stream_index, | |
1854 | int64_t timestamp, int flags) | ||
1855 | { | ||
1856 | 728 | AVIContext *avi = s->priv_data; | |
1857 | AVStream *st; | ||
1858 | FFStream *sti; | ||
1859 | int i, index; | ||
1860 | int64_t pos, pos_min; | ||
1861 | AVIStream *ast; | ||
1862 | |||
1863 | /* Does not matter which stream is requested dv in avi has the | ||
1864 | * stream information in the first video stream. | ||
1865 | */ | ||
1866 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 728 times.
|
728 | if (avi->dv_demux) |
1867 | ✗ | stream_index = 0; | |
1868 | |||
1869 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 728 times.
|
728 | if (!avi->index_loaded) { |
1870 | /* we only load the index on demand */ | ||
1871 | ✗ | avi_load_index(s); | |
1872 | ✗ | avi->index_loaded |= 1; | |
1873 | } | ||
1874 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 728 times.
|
728 | av_assert0(stream_index >= 0); |
1875 | |||
1876 | 728 | st = s->streams[stream_index]; | |
1877 | 728 | sti = ffstream(st); | |
1878 | 728 | ast = st->priv_data; | |
1879 | |||
1880 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 728 times.
|
728 | if (avi->dv_demux) { |
1881 | // index entries are in the AVI scale/rate timebase, which does | ||
1882 | // not match DV demuxer's stream timebase | ||
1883 | ✗ | timestamp = av_rescale_q(timestamp, st->time_base, | |
1884 | ✗ | (AVRational){ ast->scale, ast->rate }); | |
1885 | } else | ||
1886 | 728 | timestamp *= FFMAX(ast->sample_size, 1); | |
1887 | |||
1888 | 728 | index = av_index_search_timestamp(st, timestamp, flags); | |
1889 |
2/2✓ Branch 0 taken 198 times.
✓ Branch 1 taken 530 times.
|
728 | if (index < 0) { |
1890 |
1/2✓ Branch 0 taken 198 times.
✗ Branch 1 not taken.
|
198 | if (sti->nb_index_entries > 0) |
1891 | 198 | av_log(s, AV_LOG_DEBUG, "Failed to find timestamp %"PRId64 " in index %"PRId64 " .. %"PRId64 "\n", | |
1892 | timestamp, | ||
1893 | 198 | sti->index_entries[0].timestamp, | |
1894 | 198 | sti->index_entries[sti->nb_index_entries - 1].timestamp); | |
1895 | 198 | return AVERROR_INVALIDDATA; | |
1896 | } | ||
1897 | |||
1898 | /* find the position */ | ||
1899 | 530 | pos = sti->index_entries[index].pos; | |
1900 | 530 | timestamp = sti->index_entries[index].timestamp; | |
1901 | |||
1902 | 530 | av_log(s, AV_LOG_TRACE, "XX %"PRId64" %d %"PRId64"\n", | |
1903 | 530 | timestamp, index, sti->index_entries[index].timestamp); | |
1904 | |||
1905 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 530 times.
|
530 | if (CONFIG_DV_DEMUXER && avi->dv_demux) { |
1906 | /* One and only one real stream for DV in AVI, and it has video */ | ||
1907 | /* offsets. Calling with other stream indexes should have failed */ | ||
1908 | /* the av_index_search_timestamp call above. */ | ||
1909 | |||
1910 | ✗ | if (avio_seek(s->pb, pos, SEEK_SET) < 0) | |
1911 | ✗ | return -1; | |
1912 | |||
1913 | /* Feed the DV video stream version of the timestamp to the */ | ||
1914 | /* DV demux so it can synthesize correct timestamps. */ | ||
1915 | ✗ | ff_dv_ts_reset(avi->dv_demux, | |
1916 | ✗ | av_rescale_q(timestamp, (AVRational){ ast->scale, ast->rate }, | |
1917 | st->time_base)); | ||
1918 | |||
1919 | ✗ | avi->stream_index = -1; | |
1920 | ✗ | return 0; | |
1921 | } | ||
1922 | 530 | timestamp /= FFMAX(ast->sample_size, 1); | |
1923 | |||
1924 | 530 | pos_min = pos; | |
1925 |
2/2✓ Branch 0 taken 547 times.
✓ Branch 1 taken 530 times.
|
1077 | for (i = 0; i < s->nb_streams; i++) { |
1926 | 547 | AVStream *st2 = s->streams[i]; | |
1927 | 547 | FFStream *const sti2 = ffstream(st2); | |
1928 | 547 | AVIStream *ast2 = st2->priv_data; | |
1929 | |||
1930 | 547 | ast2->packet_size = | |
1931 | 547 | ast2->remaining = 0; | |
1932 | |||
1933 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 547 times.
|
547 | if (ast2->sub_ctx) { |
1934 | ✗ | seek_subtitle(st, st2, timestamp); | |
1935 | ✗ | continue; | |
1936 | } | ||
1937 | |||
1938 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 547 times.
|
547 | if (sti2->nb_index_entries <= 0) |
1939 | ✗ | continue; | |
1940 | |||
1941 | // av_assert1(st2->codecpar->block_align); | ||
1942 | 547 | index = av_index_search_timestamp(st2, | |
1943 | 547 | av_rescale_q(timestamp, | |
1944 | st->time_base, | ||
1945 | st2->time_base) * | ||
1946 | 547 | FFMAX(ast2->sample_size, 1), | |
1947 | 547 | flags | | |
1948 | AVSEEK_FLAG_BACKWARD | | ||
1949 |
2/2✓ Branch 0 taken 17 times.
✓ Branch 1 taken 530 times.
|
547 | (st2->codecpar->codec_type != AVMEDIA_TYPE_VIDEO ? AVSEEK_FLAG_ANY : 0)); |
1950 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 547 times.
|
547 | if (index < 0) |
1951 | ✗ | index = 0; | |
1952 | 547 | ast2->seek_pos = sti2->index_entries[index].pos; | |
1953 | 547 | pos_min = FFMIN(pos_min,ast2->seek_pos); | |
1954 | } | ||
1955 |
2/2✓ Branch 0 taken 547 times.
✓ Branch 1 taken 530 times.
|
1077 | for (i = 0; i < s->nb_streams; i++) { |
1956 | 547 | AVStream *st2 = s->streams[i]; | |
1957 | 547 | FFStream *const sti2 = ffstream(st2); | |
1958 | 547 | AVIStream *ast2 = st2->priv_data; | |
1959 | |||
1960 |
2/4✓ Branch 0 taken 547 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 547 times.
|
547 | if (ast2->sub_ctx || sti2->nb_index_entries <= 0) |
1961 | ✗ | continue; | |
1962 | |||
1963 | 547 | index = av_index_search_timestamp( | |
1964 | st2, | ||
1965 | 547 | av_rescale_q(timestamp, st->time_base, st2->time_base) * FFMAX(ast2->sample_size, 1), | |
1966 |
2/2✓ Branch 0 taken 17 times.
✓ Branch 1 taken 530 times.
|
547 | flags | AVSEEK_FLAG_BACKWARD | (st2->codecpar->codec_type != AVMEDIA_TYPE_VIDEO ? AVSEEK_FLAG_ANY : 0)); |
1967 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 547 times.
|
547 | if (index < 0) |
1968 | ✗ | index = 0; | |
1969 |
5/6✓ Branch 0 taken 558 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 421 times.
✓ Branch 3 taken 137 times.
✓ Branch 4 taken 11 times.
✓ Branch 5 taken 410 times.
|
558 | while (!avi->non_interleaved && index > 0 && sti2->index_entries[index-1].pos >= pos_min) |
1970 | 11 | index--; | |
1971 | 547 | ast2->frame_offset = sti2->index_entries[index].timestamp; | |
1972 | } | ||
1973 | |||
1974 | /* do the seek */ | ||
1975 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 530 times.
|
530 | if (avio_seek(s->pb, pos_min, SEEK_SET) < 0) { |
1976 | ✗ | av_log(s, AV_LOG_ERROR, "Seek failed\n"); | |
1977 | ✗ | return -1; | |
1978 | } | ||
1979 | 530 | avi->stream_index = -1; | |
1980 | 530 | avi->dts_max = INT_MIN; | |
1981 | 530 | return 0; | |
1982 | } | ||
1983 | |||
1984 | 468 | static int avi_read_close(AVFormatContext *s) | |
1985 | { | ||
1986 | int i; | ||
1987 | 468 | AVIContext *avi = s->priv_data; | |
1988 | |||
1989 |
2/2✓ Branch 0 taken 511 times.
✓ Branch 1 taken 468 times.
|
979 | for (i = 0; i < s->nb_streams; i++) { |
1990 | 511 | AVStream *st = s->streams[i]; | |
1991 | 511 | AVIStream *ast = st->priv_data; | |
1992 |
1/2✓ Branch 0 taken 511 times.
✗ Branch 1 not taken.
|
511 | if (ast) { |
1993 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 511 times.
|
511 | if (ast->sub_ctx) { |
1994 | ✗ | av_freep(&ast->sub_ctx->pb); | |
1995 | ✗ | avformat_close_input(&ast->sub_ctx); | |
1996 | } | ||
1997 | 511 | av_buffer_unref(&ast->sub_buffer); | |
1998 | 511 | av_packet_free(&ast->sub_pkt); | |
1999 | } | ||
2000 | } | ||
2001 | |||
2002 | 468 | av_freep(&avi->dv_demux); | |
2003 | |||
2004 | 468 | return 0; | |
2005 | } | ||
2006 | |||
2007 | 7186 | static int avi_probe(const AVProbeData *p) | |
2008 | { | ||
2009 | int i; | ||
2010 | |||
2011 | /* check file header */ | ||
2012 |
2/2✓ Branch 0 taken 34078 times.
✓ Branch 1 taken 6721 times.
|
40799 | for (i = 0; avi_headers[i][0]; i++) |
2013 |
2/2✓ Branch 0 taken 2451 times.
✓ Branch 1 taken 31627 times.
|
34078 | if (AV_RL32(p->buf ) == AV_RL32(avi_headers[i] ) && |
2014 |
2/2✓ Branch 0 taken 465 times.
✓ Branch 1 taken 1986 times.
|
2451 | AV_RL32(p->buf + 8) == AV_RL32(avi_headers[i] + 4)) |
2015 | 465 | return AVPROBE_SCORE_MAX; | |
2016 | |||
2017 | 6721 | return 0; | |
2018 | } | ||
2019 | |||
2020 | const FFInputFormat ff_avi_demuxer = { | ||
2021 | .p.name = "avi", | ||
2022 | .p.long_name = NULL_IF_CONFIG_SMALL("AVI (Audio Video Interleaved)"), | ||
2023 | .p.extensions = "avi", | ||
2024 | .p.priv_class = &demuxer_class, | ||
2025 | .priv_data_size = sizeof(AVIContext), | ||
2026 | .flags_internal = FF_INFMT_FLAG_INIT_CLEANUP, | ||
2027 | .read_probe = avi_probe, | ||
2028 | .read_header = avi_read_header, | ||
2029 | .read_packet = avi_read_packet, | ||
2030 | .read_close = avi_read_close, | ||
2031 | .read_seek = avi_read_seek, | ||
2032 | }; | ||
2033 |