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 | 64327 | static inline int get_duration(AVIStream *ast, int len) | |
140 | { | ||
141 |
2/2✓ Branch 0 taken 9460 times.
✓ Branch 1 taken 54867 times.
|
64327 | if (ast->sample_size) |
142 | 9460 | return len; | |
143 |
2/2✓ Branch 0 taken 212 times.
✓ Branch 1 taken 54655 times.
|
54867 | else if (ast->dshow_block_align) |
144 | 212 | return (len + (int64_t)ast->dshow_block_align - 1) / ast->dshow_block_align; | |
145 | else | ||
146 | 54655 | return 1; | |
147 | } | ||
148 | |||
149 | 476 | static int get_riff(AVFormatContext *s, AVIOContext *pb) | |
150 | { | ||
151 | 476 | AVIContext *avi = s->priv_data; | |
152 | 476 | char header[8] = {0}; | |
153 | int i; | ||
154 | |||
155 | /* check RIFF header */ | ||
156 | 476 | avio_read(pb, header, 4); | |
157 | 476 | avi->riff_end = avio_rl32(pb); /* RIFF chunk size */ | |
158 | 476 | avi->riff_end += avio_tell(pb); /* RIFF chunk end */ | |
159 | 476 | avio_read(pb, header + 4, 4); | |
160 | |||
161 |
1/2✓ Branch 0 taken 484 times.
✗ Branch 1 not taken.
|
484 | for (i = 0; avi_headers[i][0]; i++) |
162 |
2/2✓ Branch 0 taken 476 times.
✓ Branch 1 taken 8 times.
|
484 | if (!memcmp(header, avi_headers[i], 8)) |
163 | 476 | break; | |
164 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 476 times.
|
476 | if (!avi_headers[i][0]) |
165 | ✗ | return AVERROR_INVALIDDATA; | |
166 | |||
167 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 476 times.
|
476 | 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 | 476 | return 0; | |
172 | } | ||
173 | |||
174 | 44 | static int read_odml_index(AVFormatContext *s, int64_t frame_num) | |
175 | { | ||
176 | 44 | AVIContext *avi = s->priv_data; | |
177 | 44 | AVIOContext *pb = s->pb; | |
178 | 44 | int longs_per_entry = avio_rl16(pb); | |
179 | 44 | int index_sub_type = avio_r8(pb); | |
180 | 44 | int index_type = avio_r8(pb); | |
181 | 44 | int entries_in_use = avio_rl32(pb); | |
182 | 44 | int chunk_id = avio_rl32(pb); | |
183 | 44 | int64_t base = avio_rl64(pb); | |
184 | 44 | int stream_id = ((chunk_id & 0xFF) - '0') * 10 + | |
185 | 44 | ((chunk_id >> 8 & 0xFF) - '0'); | |
186 | AVStream *st; | ||
187 | AVIStream *ast; | ||
188 | int i; | ||
189 | 44 | int64_t last_pos = -1; | |
190 | 44 | int64_t filesize = avi->fsize; | |
191 | |||
192 | 44 | 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 31 times.
✓ Branch 1 taken 13 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 31 times.
|
44 | if (stream_id >= s->nb_streams || stream_id < 0) |
203 | 13 | return AVERROR_INVALIDDATA; | |
204 | 31 | st = s->streams[stream_id]; | |
205 | 31 | ast = st->priv_data; | |
206 | |||
207 |
2/4✓ Branch 0 taken 31 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 31 times.
|
31 | if (index_sub_type || entries_in_use < 0) |
208 | ✗ | return AVERROR_INVALIDDATA; | |
209 | |||
210 | 31 | avio_rl32(pb); | |
211 | |||
212 |
3/4✓ Branch 0 taken 9 times.
✓ Branch 1 taken 22 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 9 times.
|
31 | if (index_type && longs_per_entry != 2) |
213 | ✗ | return AVERROR_INVALIDDATA; | |
214 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 31 times.
|
31 | if (index_type > 1) |
215 | ✗ | return AVERROR_INVALIDDATA; | |
216 | |||
217 |
2/4✓ Branch 0 taken 31 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 31 times.
|
31 | 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 5697 times.
✓ Branch 1 taken 18 times.
|
5715 | for (i = 0; i < entries_in_use; i++) { |
228 |
2/2✓ Branch 1 taken 3 times.
✓ Branch 2 taken 5694 times.
|
5697 | 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 5697 times.
|
5697 | if (avi->odml_read > avi->odml_max_pos) |
232 | ✗ | return AVERROR_INVALIDDATA; | |
233 | |||
234 |
2/2✓ Branch 0 taken 5675 times.
✓ Branch 1 taken 22 times.
|
5697 | 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 | 22 | avi->odml_read += 16; | |
259 | |||
260 | 22 | offset = avio_rl64(pb); | |
261 | 22 | avio_rl32(pb); /* size */ | |
262 | 22 | duration = avio_rl32(pb); | |
263 | |||
264 |
2/4✓ Branch 1 taken 22 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✓ Branch 4 taken 22 times.
|
22 | if (avio_feof(pb) || offset > INT64_MAX - 8) |
265 | ✗ | return AVERROR_INVALIDDATA; | |
266 | |||
267 | 22 | pos = avio_tell(pb); | |
268 | |||
269 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 22 times.
|
22 | 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 22 times.
|
22 | if (avio_seek(pb, offset + 8, SEEK_SET) < 0) |
275 | ✗ | return -1; | |
276 | 22 | avi->odml_depth++; | |
277 | 22 | ret = read_odml_index(s, frame_num); | |
278 | 22 | avi->odml_depth--; | |
279 | 22 | frame_num += duration; | |
280 | |||
281 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 22 times.
|
22 | 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 13 times.
✓ Branch 1 taken 9 times.
|
22 | if (ret < 0) |
286 | 13 | 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 | ✗ | AVExifMetadata ifd = { 0 }; | |
432 | int ret; | ||
433 | // skip 4 byte padding | ||
434 | ✗ | bytestream2_skip(&gb, 4); | |
435 | ✗ | offset = bytestream2_tell(&gb); | |
436 | |||
437 | // decode EXIF tags from IFD, AVI is always little-endian | ||
438 | ✗ | ret = av_exif_parse_buffer(s, data + offset, data_size - offset, &ifd, AV_EXIF_ASSUME_LE); | |
439 | ✗ | if (ret < 0) | |
440 | ✗ | return ret; | |
441 | ✗ | ret = av_exif_ifd_to_dict(s, &ifd, &st->metadata); | |
442 | ✗ | av_exif_free(&ifd); | |
443 | ✗ | return ret; | |
444 | } | ||
445 | ✗ | case MKTAG('C', 'A', 'S', 'I'): | |
446 | ✗ | avpriv_request_sample(s, "RIFF stream data tag type CASI (%u)", tag); | |
447 | ✗ | break; | |
448 | ✗ | case MKTAG('Z', 'o', 'r', 'a'): | |
449 | ✗ | avpriv_request_sample(s, "RIFF stream data tag type Zora (%u)", tag); | |
450 | ✗ | break; | |
451 | 6 | default: | |
452 | 6 | break; | |
453 | } | ||
454 | |||
455 | 6 | return 0; | |
456 | } | ||
457 | |||
458 | 476 | static int calculate_bitrate(AVFormatContext *s) | |
459 | { | ||
460 | 476 | AVIContext *avi = s->priv_data; | |
461 | int i, j; | ||
462 | 476 | int64_t lensum = 0; | |
463 | 476 | int64_t maxpos = 0; | |
464 | |||
465 |
2/2✓ Branch 0 taken 520 times.
✓ Branch 1 taken 476 times.
|
996 | for (i = 0; i<s->nb_streams; i++) { |
466 | 520 | int64_t len = 0; | |
467 | 520 | FFStream *const sti = ffstream(s->streams[i]); | |
468 | |||
469 |
2/2✓ Branch 0 taken 60 times.
✓ Branch 1 taken 460 times.
|
520 | if (!sti->nb_index_entries) |
470 | 60 | continue; | |
471 | |||
472 |
2/2✓ Branch 0 taken 30981 times.
✓ Branch 1 taken 460 times.
|
31441 | for (j = 0; j < sti->nb_index_entries; j++) |
473 | 30981 | len += sti->index_entries[j].size; | |
474 | 460 | maxpos = FFMAX(maxpos, sti->index_entries[j-1].pos); | |
475 | 460 | lensum += len; | |
476 | } | ||
477 |
2/2✓ Branch 0 taken 106 times.
✓ Branch 1 taken 370 times.
|
476 | if (maxpos < av_rescale(avi->io_fsize, 9, 10)) // index does not cover the whole file |
478 | 106 | return 0; | |
479 |
3/4✓ Branch 0 taken 370 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 27 times.
✓ Branch 3 taken 343 times.
|
370 | if (lensum*9/10 > maxpos || lensum < maxpos*9/10) // frame sum and filesize mismatch |
480 | 27 | return 0; | |
481 | |||
482 |
2/2✓ Branch 0 taken 363 times.
✓ Branch 1 taken 343 times.
|
706 | for (i = 0; i<s->nb_streams; i++) { |
483 | 363 | int64_t len = 0; | |
484 | 363 | AVStream *st = s->streams[i]; | |
485 | 363 | FFStream *const sti = ffstream(st); | |
486 | int64_t duration; | ||
487 | AVInteger bitrate_i, den_i, num_i; | ||
488 | |||
489 |
2/2✓ Branch 0 taken 27691 times.
✓ Branch 1 taken 363 times.
|
28054 | for (j = 0; j < sti->nb_index_entries; j++) |
490 | 27691 | len += sti->index_entries[j].size; | |
491 | |||
492 |
4/4✓ Branch 0 taken 361 times.
✓ Branch 1 taken 2 times.
✓ Branch 2 taken 21 times.
✓ Branch 3 taken 340 times.
|
363 | if (sti->nb_index_entries < 2 || st->codecpar->bit_rate > 0) |
493 | 23 | continue; | |
494 | 340 | duration = sti->index_entries[j-1].timestamp - sti->index_entries[0].timestamp; | |
495 | 340 | den_i = av_mul_i(av_int2i(duration), av_int2i(st->time_base.num)); | |
496 | 340 | num_i = av_add_i(av_mul_i(av_int2i(8*len), av_int2i(st->time_base.den)), av_shr_i(den_i, 1)); | |
497 | 340 | bitrate_i = av_div_i(num_i, den_i); | |
498 |
1/2✓ Branch 0 taken 340 times.
✗ Branch 1 not taken.
|
340 | if (av_cmp_i(bitrate_i, av_int2i(INT64_MAX)) <= 0) { |
499 | 340 | int64_t bitrate = av_i2int(bitrate_i); | |
500 |
1/2✓ Branch 0 taken 340 times.
✗ Branch 1 not taken.
|
340 | if (bitrate > 0) { |
501 | 340 | st->codecpar->bit_rate = bitrate; | |
502 | } | ||
503 | } | ||
504 | } | ||
505 | 343 | return 1; | |
506 | } | ||
507 | |||
508 | 476 | static int avi_read_header(AVFormatContext *s) | |
509 | { | ||
510 | 476 | AVIContext *avi = s->priv_data; | |
511 | 476 | AVIOContext *pb = s->pb; | |
512 | unsigned int tag, tag1, handler; | ||
513 | int codec_type, stream_index, frame_period; | ||
514 | unsigned int size; | ||
515 | int i; | ||
516 | AVStream *st; | ||
517 | 476 | AVIStream *ast = NULL; | |
518 | 476 | int avih_width = 0, avih_height = 0; | |
519 | 476 | int amv_file_format = 0; | |
520 | 476 | uint64_t list_end = 0; | |
521 | int64_t pos; | ||
522 | int ret; | ||
523 | AVDictionaryEntry *dict_entry; | ||
524 | |||
525 | 476 | avi->stream_index = -1; | |
526 | |||
527 | 476 | ret = get_riff(s, pb); | |
528 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 476 times.
|
476 | if (ret < 0) |
529 | ✗ | return ret; | |
530 | |||
531 | 476 | av_log(avi, AV_LOG_DEBUG, "use odml:%d\n", avi->use_odml); | |
532 | |||
533 | 476 | avi->io_fsize = avi->fsize = avio_size(pb); | |
534 |
3/4✓ Branch 0 taken 476 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 38 times.
✓ Branch 3 taken 438 times.
|
476 | if (avi->fsize <= 0 || avi->fsize < avi->riff_end) |
535 |
1/2✓ Branch 0 taken 38 times.
✗ Branch 1 not taken.
|
38 | avi->fsize = avi->riff_end == 8 ? INT64_MAX : avi->riff_end; |
536 | |||
537 | /* first list tag */ | ||
538 | 476 | stream_index = -1; | |
539 | 476 | codec_type = -1; | |
540 | 476 | frame_period = 0; | |
541 | 4207 | for (;;) { | |
542 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 4683 times.
|
4683 | if (avio_feof(pb)) |
543 | ✗ | return AVERROR_INVALIDDATA; | |
544 | 4683 | tag = avio_rl32(pb); | |
545 | 4683 | size = avio_rl32(pb); | |
546 | |||
547 | 4683 | print_tag(s, "tag", tag, size); | |
548 | |||
549 |
12/13✓ Branch 0 taken 1596 times.
✓ Branch 1 taken 1 times.
✓ Branch 2 taken 42 times.
✓ Branch 3 taken 2 times.
✓ Branch 4 taken 474 times.
✓ Branch 5 taken 521 times.
✓ Branch 6 taken 520 times.
✓ Branch 7 taken 11 times.
✓ Branch 8 taken 22 times.
✓ Branch 9 taken 10 times.
✓ Branch 10 taken 19 times.
✓ Branch 11 taken 1465 times.
✗ Branch 12 not taken.
|
4683 | switch (tag) { |
550 | 1596 | case MKTAG('L', 'I', 'S', 'T'): | |
551 | 1596 | list_end = avio_tell(pb) + size; | |
552 | /* Ignored, except at start of video packets. */ | ||
553 | 1596 | tag1 = avio_rl32(pb); | |
554 | |||
555 | 1596 | print_tag(s, "list", tag1, 0); | |
556 | |||
557 |
2/2✓ Branch 0 taken 476 times.
✓ Branch 1 taken 1120 times.
|
1596 | if (tag1 == MKTAG('m', 'o', 'v', 'i')) { |
558 | 476 | avi->movi_list = avio_tell(pb) - 4; | |
559 |
2/2✓ Branch 0 taken 474 times.
✓ Branch 1 taken 2 times.
|
476 | if (size) |
560 | 474 | avi->movi_end = avi->movi_list + size + (size & 1); | |
561 | else | ||
562 | 2 | avi->movi_end = avi->fsize; | |
563 | 476 | av_log(s, AV_LOG_TRACE, "movi end=%"PRIx64"\n", avi->movi_end); | |
564 | 476 | goto end_of_header; | |
565 |
2/2✓ Branch 0 taken 65 times.
✓ Branch 1 taken 1055 times.
|
1120 | } else if (tag1 == MKTAG('I', 'N', 'F', 'O')) |
566 | 65 | ff_read_riff_info(s, size - 4); | |
567 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1055 times.
|
1055 | else if (tag1 == MKTAG('n', 'c', 'd', 't')) |
568 | ✗ | avi_read_nikon(s, list_end); | |
569 | |||
570 | 4207 | break; | |
571 | 1 | case MKTAG('I', 'D', 'I', 'T'): | |
572 | { | ||
573 | 1 | unsigned char date[64] = { 0 }; | |
574 | 1 | size += (size & 1); | |
575 | 1 | size -= avio_read(pb, date, FFMIN(size, sizeof(date) - 1)); | |
576 | 1 | avio_skip(pb, size); | |
577 | 1 | avi_metadata_creation_time(&s->metadata, date); | |
578 | 1 | break; | |
579 | } | ||
580 | 42 | case MKTAG('d', 'm', 'l', 'h'): | |
581 | 42 | avi->is_odml = 1; | |
582 | 42 | avio_skip(pb, size + (size & 1)); | |
583 | 42 | break; | |
584 | 2 | case MKTAG('a', 'm', 'v', 'h'): | |
585 | 2 | amv_file_format = 1; | |
586 | 476 | case MKTAG('a', 'v', 'i', 'h'): | |
587 | /* AVI header */ | ||
588 | /* using frame_period is bad idea */ | ||
589 | 476 | frame_period = avio_rl32(pb); | |
590 | 476 | avio_rl32(pb); /* max. bytes per second */ | |
591 | 476 | avio_rl32(pb); | |
592 | 476 | avi->non_interleaved |= avio_rl32(pb) & AVIF_MUSTUSEINDEX; | |
593 | |||
594 | 476 | avio_skip(pb, 2 * 4); | |
595 | 476 | avio_rl32(pb); | |
596 | 476 | avio_rl32(pb); | |
597 | 476 | avih_width = avio_rl32(pb); | |
598 | 476 | avih_height = avio_rl32(pb); | |
599 | |||
600 | 476 | avio_skip(pb, size - 10 * 4); | |
601 | 476 | break; | |
602 | 521 | case MKTAG('s', 't', 'r', 'h'): | |
603 | /* stream header */ | ||
604 | |||
605 | 521 | tag1 = avio_rl32(pb); | |
606 | 521 | handler = avio_rl32(pb); /* codec tag */ | |
607 | |||
608 |
2/2✓ Branch 0 taken 1 times.
✓ Branch 1 taken 520 times.
|
521 | if (tag1 == MKTAG('p', 'a', 'd', 's')) { |
609 | 1 | avio_skip(pb, size - 8); | |
610 | 1 | break; | |
611 | } else { | ||
612 | 520 | stream_index++; | |
613 | 520 | st = avformat_new_stream(s, NULL); | |
614 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 520 times.
|
520 | if (!st) |
615 | ✗ | return AVERROR(ENOMEM); | |
616 | |||
617 | 520 | st->id = stream_index; | |
618 | 520 | ast = av_mallocz(sizeof(AVIStream)); | |
619 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 520 times.
|
520 | if (!ast) |
620 | ✗ | return AVERROR(ENOMEM); | |
621 | 520 | st->priv_data = ast; | |
622 | } | ||
623 |
2/2✓ Branch 0 taken 4 times.
✓ Branch 1 taken 516 times.
|
520 | if (amv_file_format) |
624 | 4 | tag1 = stream_index ? MKTAG('a', 'u', 'd', 's') | |
625 |
2/2✓ Branch 0 taken 2 times.
✓ Branch 1 taken 2 times.
|
4 | : MKTAG('v', 'i', 'd', 's'); |
626 | |||
627 | 520 | print_tag(s, "strh", tag1, -1); | |
628 | |||
629 |
2/4✓ Branch 0 taken 520 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 520 times.
|
520 | if (tag1 == MKTAG('i', 'a', 'v', 's') || |
630 | tag1 == MKTAG('i', 'v', 'a', 's')) { | ||
631 | int64_t dv_dur; | ||
632 | |||
633 | /* After some consideration -- I don't think we | ||
634 | * have to support anything but DV in type1 AVIs. */ | ||
635 | ✗ | if (s->nb_streams != 1) | |
636 | ✗ | return AVERROR_INVALIDDATA; | |
637 | |||
638 | ✗ | if (handler != MKTAG('d', 'v', 's', 'd') && | |
639 | ✗ | handler != MKTAG('d', 'v', 'h', 'd') && | |
640 | handler != MKTAG('d', 'v', 's', 'l')) | ||
641 | ✗ | return AVERROR_INVALIDDATA; | |
642 | |||
643 | if (!CONFIG_DV_DEMUXER) | ||
644 | return AVERROR_DEMUXER_NOT_FOUND; | ||
645 | |||
646 | ✗ | ast = s->streams[0]->priv_data; | |
647 | ✗ | st->priv_data = NULL; | |
648 | ✗ | ff_remove_stream(s, st); | |
649 | |||
650 | ✗ | avi->dv_demux = avpriv_dv_init_demux(s); | |
651 | ✗ | if (!avi->dv_demux) { | |
652 | ✗ | av_free(ast); | |
653 | ✗ | return AVERROR(ENOMEM); | |
654 | } | ||
655 | |||
656 | ✗ | s->streams[0]->priv_data = ast; | |
657 | ✗ | avio_skip(pb, 3 * 4); | |
658 | ✗ | ast->scale = avio_rl32(pb); | |
659 | ✗ | ast->rate = avio_rl32(pb); | |
660 | ✗ | avio_skip(pb, 4); /* start time */ | |
661 | |||
662 | ✗ | dv_dur = avio_rl32(pb); | |
663 | ✗ | if (ast->scale > 0 && ast->rate > 0 && dv_dur > 0) { | |
664 | ✗ | dv_dur *= AV_TIME_BASE; | |
665 | ✗ | s->duration = av_rescale(dv_dur, ast->scale, ast->rate); | |
666 | } | ||
667 | /* else, leave duration alone; timing estimation in utils.c | ||
668 | * will make a guess based on bitrate. */ | ||
669 | |||
670 | ✗ | stream_index = s->nb_streams - 1; | |
671 | ✗ | avio_skip(pb, size - 9 * 4); | |
672 | ✗ | break; | |
673 | } | ||
674 | |||
675 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 520 times.
|
520 | av_assert0(stream_index < s->nb_streams); |
676 | 520 | ast->handler = handler; | |
677 | |||
678 | 520 | avio_rl32(pb); /* flags */ | |
679 | 520 | avio_rl16(pb); /* priority */ | |
680 | 520 | avio_rl16(pb); /* language */ | |
681 | 520 | avio_rl32(pb); /* initial frame */ | |
682 | 520 | ast->scale = avio_rl32(pb); | |
683 | 520 | ast->rate = avio_rl32(pb); | |
684 |
3/4✓ Branch 0 taken 516 times.
✓ Branch 1 taken 4 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 516 times.
|
520 | if (!(ast->scale && ast->rate)) { |
685 | 4 | av_log(s, AV_LOG_WARNING, | |
686 | "scale/rate is %"PRIu32"/%"PRIu32" which is invalid. " | ||
687 | "(This file has been generated by broken software.)\n", | ||
688 | ast->scale, | ||
689 | ast->rate); | ||
690 |
1/2✓ Branch 0 taken 4 times.
✗ Branch 1 not taken.
|
4 | if (frame_period) { |
691 | 4 | ast->rate = 1000000; | |
692 | 4 | ast->scale = frame_period; | |
693 | } else { | ||
694 | ✗ | ast->rate = 25; | |
695 | ✗ | ast->scale = 1; | |
696 | } | ||
697 | } | ||
698 | 520 | avpriv_set_pts_info(st, 64, ast->scale, ast->rate); | |
699 | |||
700 | 520 | ast->cum_len = avio_rl32(pb); /* start */ | |
701 | 520 | st->nb_frames = avio_rl32(pb); | |
702 | |||
703 | 520 | st->start_time = 0; | |
704 | 520 | avio_rl32(pb); /* buffer size */ | |
705 | 520 | avio_rl32(pb); /* quality */ | |
706 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 520 times.
|
520 | if (ast->cum_len > 3600LL * ast->rate / ast->scale) { |
707 | ✗ | av_log(s, AV_LOG_ERROR, "crazy start time, iam scared, giving up\n"); | |
708 | ✗ | ast->cum_len = 0; | |
709 | } | ||
710 | 520 | ast->sample_size = avio_rl32(pb); | |
711 | 520 | ast->cum_len *= FFMAX(1, ast->sample_size); | |
712 | 520 | av_log(s, AV_LOG_TRACE, "%"PRIu32" %"PRIu32" %d\n", | |
713 | ast->rate, ast->scale, ast->sample_size); | ||
714 | |||
715 |
2/5✓ Branch 0 taken 473 times.
✓ Branch 1 taken 47 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
|
520 | switch (tag1) { |
716 | 473 | case MKTAG('v', 'i', 'd', 's'): | |
717 | 473 | codec_type = AVMEDIA_TYPE_VIDEO; | |
718 | |||
719 | 473 | ast->sample_size = 0; | |
720 | 473 | st->avg_frame_rate = av_inv_q(st->time_base); | |
721 | 473 | break; | |
722 | 47 | case MKTAG('a', 'u', 'd', 's'): | |
723 | 47 | codec_type = AVMEDIA_TYPE_AUDIO; | |
724 | 47 | break; | |
725 | ✗ | case MKTAG('t', 'x', 't', 's'): | |
726 | ✗ | codec_type = AVMEDIA_TYPE_SUBTITLE; | |
727 | ✗ | break; | |
728 | ✗ | case MKTAG('d', 'a', 't', 's'): | |
729 | ✗ | codec_type = AVMEDIA_TYPE_DATA; | |
730 | ✗ | break; | |
731 | ✗ | default: | |
732 | ✗ | av_log(s, AV_LOG_INFO, "unknown stream type %X\n", tag1); | |
733 | } | ||
734 | |||
735 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 520 times.
|
520 | if (ast->sample_size < 0) { |
736 | ✗ | if (s->error_recognition & AV_EF_EXPLODE) { | |
737 | ✗ | av_log(s, AV_LOG_ERROR, | |
738 | "Invalid sample_size %d at stream %d\n", | ||
739 | ast->sample_size, | ||
740 | stream_index); | ||
741 | ✗ | return AVERROR_INVALIDDATA; | |
742 | } | ||
743 | ✗ | av_log(s, AV_LOG_WARNING, | |
744 | "Invalid sample_size %d at stream %d " | ||
745 | "setting it to 0\n", | ||
746 | ast->sample_size, | ||
747 | stream_index); | ||
748 | ✗ | ast->sample_size = 0; | |
749 | } | ||
750 | |||
751 |
2/2✓ Branch 0 taken 483 times.
✓ Branch 1 taken 37 times.
|
520 | if (ast->sample_size == 0) { |
752 | 483 | st->duration = st->nb_frames; | |
753 |
5/6✓ Branch 0 taken 478 times.
✓ Branch 1 taken 5 times.
✓ Branch 2 taken 478 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 39 times.
✓ Branch 5 taken 439 times.
|
483 | if (st->duration > 0 && avi->io_fsize > 0 && avi->riff_end > avi->io_fsize) { |
754 | 39 | av_log(s, AV_LOG_DEBUG, "File is truncated adjusting duration\n"); | |
755 | 39 | st->duration = av_rescale(st->duration, avi->io_fsize, avi->riff_end); | |
756 | } | ||
757 | } | ||
758 | 520 | ast->frame_offset = ast->cum_len; | |
759 | 520 | avio_skip(pb, size - 12 * 4); | |
760 | 520 | break; | |
761 | 520 | case MKTAG('s', 't', 'r', 'f'): | |
762 | /* stream header */ | ||
763 |
1/6✗ Branch 0 not taken.
✓ Branch 1 taken 520 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
|
520 | if (!size && (codec_type == AVMEDIA_TYPE_AUDIO || |
764 | codec_type == AVMEDIA_TYPE_VIDEO)) | ||
765 | break; | ||
766 |
2/4✓ Branch 0 taken 520 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 520 times.
|
520 | if (stream_index >= (unsigned)s->nb_streams || avi->dv_demux) { |
767 | ✗ | avio_skip(pb, size); | |
768 | } else { | ||
769 | 520 | uint64_t cur_pos = avio_tell(pb); | |
770 | FFStream *sti; | ||
771 | unsigned esize; | ||
772 |
2/2✓ Branch 0 taken 516 times.
✓ Branch 1 taken 4 times.
|
520 | if (cur_pos < list_end) |
773 | 516 | size = FFMIN(size, list_end - cur_pos); | |
774 | 520 | st = s->streams[stream_index]; | |
775 | 520 | sti = ffstream(st); | |
776 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 520 times.
|
520 | if (st->codecpar->codec_type != AVMEDIA_TYPE_UNKNOWN) { |
777 | ✗ | avio_skip(pb, size); | |
778 | ✗ | break; | |
779 | } | ||
780 |
2/4✓ Branch 0 taken 473 times.
✓ Branch 1 taken 47 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
|
520 | switch (codec_type) { |
781 | 473 | case AVMEDIA_TYPE_VIDEO: | |
782 |
2/2✓ Branch 0 taken 2 times.
✓ Branch 1 taken 471 times.
|
473 | if (amv_file_format) { |
783 | 2 | st->codecpar->width = avih_width; | |
784 | 2 | st->codecpar->height = avih_height; | |
785 | 2 | st->codecpar->codec_type = AVMEDIA_TYPE_VIDEO; | |
786 | 2 | st->codecpar->codec_id = AV_CODEC_ID_AMV; | |
787 | 2 | avio_skip(pb, size); | |
788 | 520 | break; | |
789 | } | ||
790 | 471 | tag1 = ff_get_bmp_header(pb, st, &esize); | |
791 | |||
792 |
2/4✓ Branch 0 taken 471 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 471 times.
|
471 | if (tag1 == MKTAG('D', 'X', 'S', 'B') || |
793 | tag1 == MKTAG('D', 'X', 'S', 'A')) { | ||
794 | ✗ | st->codecpar->codec_type = AVMEDIA_TYPE_SUBTITLE; | |
795 | ✗ | st->codecpar->codec_tag = tag1; | |
796 | ✗ | st->codecpar->codec_id = AV_CODEC_ID_XSUB; | |
797 | ✗ | break; | |
798 | } | ||
799 | |||
800 |
4/6✓ Branch 0 taken 170 times.
✓ Branch 1 taken 301 times.
✓ Branch 2 taken 170 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 170 times.
✗ Branch 5 not taken.
|
471 | if (size > 10 * 4 && size < (1 << 30) && size < avi->fsize) { |
801 |
3/4✓ Branch 0 taken 18 times.
✓ Branch 1 taken 152 times.
✓ Branch 2 taken 18 times.
✗ Branch 3 not taken.
|
170 | if (esize == size-1 && (esize&1)) { |
802 | 18 | st->codecpar->extradata_size = esize - 10 * 4; | |
803 | } else | ||
804 | 152 | st->codecpar->extradata_size = size - 10 * 4; | |
805 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 170 times.
|
170 | if (st->codecpar->extradata) { |
806 | ✗ | av_log(s, AV_LOG_WARNING, "New extradata in strf chunk, freeing previous one.\n"); | |
807 | } | ||
808 | 170 | ret = ff_get_extradata(s, st->codecpar, pb, | |
809 | 170 | st->codecpar->extradata_size); | |
810 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 170 times.
|
170 | if (ret < 0) |
811 | ✗ | return ret; | |
812 | } | ||
813 | |||
814 | // FIXME: check if the encoder really did this correctly | ||
815 |
2/2✓ Branch 0 taken 18 times.
✓ Branch 1 taken 453 times.
|
471 | if (st->codecpar->extradata_size & 1) |
816 | 18 | avio_r8(pb); | |
817 | |||
818 | /* Extract palette from extradata if bpp <= 8. | ||
819 | * This code assumes that extradata contains only palette. | ||
820 | * This is true for all paletted codecs implemented in | ||
821 | * FFmpeg. */ | ||
822 |
2/2✓ Branch 0 taken 170 times.
✓ Branch 1 taken 301 times.
|
471 | if (st->codecpar->extradata_size && |
823 |
2/2✓ Branch 0 taken 15 times.
✓ Branch 1 taken 155 times.
|
170 | (st->codecpar->bits_per_coded_sample <= 8)) { |
824 | 15 | int pal_size = (1 << st->codecpar->bits_per_coded_sample) << 2; | |
825 | const uint8_t *pal_src; | ||
826 | |||
827 | 15 | pal_size = FFMIN(pal_size, st->codecpar->extradata_size); | |
828 | 15 | pal_src = st->codecpar->extradata + | |
829 | 15 | st->codecpar->extradata_size - pal_size; | |
830 | /* Exclude the "BottomUp" field from the palette */ | ||
831 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 15 times.
|
15 | if (pal_src - st->codecpar->extradata >= 9 && |
832 | ✗ | !memcmp(st->codecpar->extradata + st->codecpar->extradata_size - 9, "BottomUp", 9)) | |
833 | ✗ | pal_src -= 9; | |
834 |
2/2✓ Branch 0 taken 2330 times.
✓ Branch 1 taken 15 times.
|
2345 | for (i = 0; i < pal_size / 4; i++) |
835 | 2330 | ast->pal[i] = 0xFFU<<24 | AV_RL32(pal_src + 4 * i); | |
836 | 15 | ast->has_pal = 1; | |
837 | } | ||
838 | |||
839 | 471 | print_tag(s, "video", tag1, 0); | |
840 | |||
841 | 471 | st->codecpar->codec_type = AVMEDIA_TYPE_VIDEO; | |
842 | 471 | st->codecpar->codec_tag = tag1; | |
843 | 471 | st->codecpar->codec_id = ff_codec_get_id(ff_codec_bmp_tags, | |
844 | tag1); | ||
845 | /* If codec is not found yet, try with the mov tags. */ | ||
846 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 471 times.
|
471 | if (!st->codecpar->codec_id) { |
847 | ✗ | st->codecpar->codec_id = | |
848 | ✗ | ff_codec_get_id(ff_codec_movvideo_tags, tag1); | |
849 | ✗ | if (st->codecpar->codec_id) | |
850 | ✗ | av_log(s, AV_LOG_WARNING, | |
851 | "mov tag found in avi (fourcc %s)\n", | ||
852 | ✗ | av_fourcc2str(tag1)); | |
853 | } | ||
854 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 471 times.
|
471 | if (!st->codecpar->codec_id) |
855 | ✗ | st->codecpar->codec_id = ff_codec_get_id(ff_codec_bmp_tags_unofficial, tag1); | |
856 | |||
857 | /* This is needed to get the pict type which is necessary | ||
858 | * for generating correct pts. */ | ||
859 | 471 | sti->need_parsing = AVSTREAM_PARSE_HEADERS; | |
860 | |||
861 |
2/2✓ Branch 0 taken 56 times.
✓ Branch 1 taken 415 times.
|
471 | if (st->codecpar->codec_id == AV_CODEC_ID_MPEG4 && |
862 |
2/2✓ Branch 0 taken 2 times.
✓ Branch 1 taken 54 times.
|
56 | ast->handler == MKTAG('X', 'V', 'I', 'D')) |
863 | 2 | st->codecpar->codec_tag = MKTAG('X', 'V', 'I', 'D'); | |
864 | |||
865 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 471 times.
|
471 | if (st->codecpar->codec_tag == MKTAG('V', 'S', 'S', 'H')) |
866 | ✗ | sti->need_parsing = AVSTREAM_PARSE_FULL; | |
867 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 471 times.
|
471 | if (st->codecpar->codec_id == AV_CODEC_ID_RV40) |
868 | ✗ | sti->need_parsing = AVSTREAM_PARSE_NONE; | |
869 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 471 times.
|
471 | if (st->codecpar->codec_id == AV_CODEC_ID_HEVC && |
870 | ✗ | st->codecpar->codec_tag == MKTAG('H', '2', '6', '5')) | |
871 | ✗ | sti->need_parsing = AVSTREAM_PARSE_FULL; | |
872 | |||
873 |
2/2✓ Branch 0 taken 1 times.
✓ Branch 1 taken 470 times.
|
471 | if (st->codecpar->codec_id == AV_CODEC_ID_AVRN && |
874 |
1/2✓ Branch 0 taken 1 times.
✗ Branch 1 not taken.
|
1 | st->codecpar->codec_tag == MKTAG('A', 'V', 'R', 'n') && |
875 |
1/2✓ Branch 0 taken 1 times.
✗ Branch 1 not taken.
|
1 | (st->codecpar->extradata_size < 31 || |
876 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
|
1 | memcmp(&st->codecpar->extradata[28], "1:1", 3))) |
877 | ✗ | st->codecpar->codec_id = AV_CODEC_ID_MJPEG; | |
878 | |||
879 |
3/4✓ Branch 0 taken 13 times.
✓ Branch 1 taken 458 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 13 times.
|
471 | if (st->codecpar->codec_tag == 0 && st->codecpar->height > 0 && |
880 | ✗ | st->codecpar->extradata_size < 1U << 30) { | |
881 | ✗ | st->codecpar->extradata_size += 9; | |
882 | ✗ | if ((ret = av_reallocp(&st->codecpar->extradata, | |
883 | ✗ | st->codecpar->extradata_size + | |
884 | AV_INPUT_BUFFER_PADDING_SIZE)) < 0) { | ||
885 | ✗ | st->codecpar->extradata_size = 0; | |
886 | ✗ | return ret; | |
887 | } else | ||
888 | ✗ | memcpy(st->codecpar->extradata + st->codecpar->extradata_size - 9, | |
889 | "BottomUp", 9); | ||
890 | } | ||
891 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 471 times.
|
471 | if (st->codecpar->height == INT_MIN) |
892 | ✗ | return AVERROR_INVALIDDATA; | |
893 | 471 | st->codecpar->height = FFABS(st->codecpar->height); | |
894 | |||
895 | // avio_skip(pb, size - 5 * 4); | ||
896 | 471 | break; | |
897 | 47 | case AVMEDIA_TYPE_AUDIO: | |
898 | 47 | ret = ff_get_wav_header(s, pb, st->codecpar, size, 0); | |
899 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 47 times.
|
47 | if (ret < 0) |
900 | ✗ | return ret; | |
901 | 47 | ast->dshow_block_align = st->codecpar->block_align; | |
902 |
3/4✓ Branch 0 taken 37 times.
✓ Branch 1 taken 10 times.
✓ Branch 2 taken 37 times.
✗ Branch 3 not taken.
|
47 | if (ast->sample_size && st->codecpar->block_align && |
903 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 37 times.
|
37 | ast->sample_size != st->codecpar->block_align) { |
904 | ✗ | av_log(s, | |
905 | AV_LOG_WARNING, | ||
906 | "sample size (%d) != block align (%d)\n", | ||
907 | ast->sample_size, | ||
908 | ✗ | st->codecpar->block_align); | |
909 | ✗ | ast->sample_size = st->codecpar->block_align; | |
910 | } | ||
911 | /* 2-aligned | ||
912 | * (fix for Stargate SG-1 - 3x18 - Shades of Grey.avi) */ | ||
913 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 47 times.
|
47 | if (size & 1) |
914 | ✗ | avio_skip(pb, 1); | |
915 | /* Force parsing as several audio frames can be in | ||
916 | * one packet and timestamps refer to packet start. */ | ||
917 | 47 | sti->need_parsing = AVSTREAM_PARSE_TIMESTAMPS; | |
918 | /* ADTS header is in extradata, AAC without header must be | ||
919 | * stored as exact frames. Parser not needed and it will | ||
920 | * fail. */ | ||
921 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 47 times.
|
47 | if (st->codecpar->codec_id == AV_CODEC_ID_AAC && |
922 | ✗ | st->codecpar->extradata_size) | |
923 | ✗ | sti->need_parsing = AVSTREAM_PARSE_NONE; | |
924 | // The flac parser does not work with AVSTREAM_PARSE_TIMESTAMPS | ||
925 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 47 times.
|
47 | if (st->codecpar->codec_id == AV_CODEC_ID_FLAC) |
926 | ✗ | sti->need_parsing = AVSTREAM_PARSE_NONE; | |
927 | /* AVI files with Xan DPCM audio (wrongly) declare PCM | ||
928 | * audio in the header but have Axan as stream_code_tag. */ | ||
929 |
2/2✓ Branch 0 taken 2 times.
✓ Branch 1 taken 45 times.
|
47 | if (ast->handler == AV_RL32("Axan")) { |
930 | 2 | st->codecpar->codec_id = AV_CODEC_ID_XAN_DPCM; | |
931 | 2 | st->codecpar->codec_tag = 0; | |
932 | 2 | ast->dshow_block_align = 0; | |
933 | } | ||
934 |
2/2✓ Branch 0 taken 2 times.
✓ Branch 1 taken 45 times.
|
47 | if (amv_file_format) { |
935 | 2 | st->codecpar->codec_id = AV_CODEC_ID_ADPCM_IMA_AMV; | |
936 | 2 | ast->dshow_block_align = 0; | |
937 | } | ||
938 |
1/2✓ Branch 0 taken 47 times.
✗ Branch 1 not taken.
|
47 | if ((st->codecpar->codec_id == AV_CODEC_ID_AAC || |
939 |
1/2✓ Branch 0 taken 47 times.
✗ Branch 1 not taken.
|
47 | st->codecpar->codec_id == AV_CODEC_ID_FTR || |
940 |
1/2✓ Branch 0 taken 47 times.
✗ Branch 1 not taken.
|
47 | st->codecpar->codec_id == AV_CODEC_ID_FLAC || |
941 |
3/6✓ Branch 0 taken 3 times.
✓ Branch 1 taken 44 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 3 times.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
|
47 | st->codecpar->codec_id == AV_CODEC_ID_MP2 ) && ast->dshow_block_align <= 4 && ast->dshow_block_align) { |
942 | ✗ | av_log(s, AV_LOG_DEBUG, "overriding invalid dshow_block_align of %d\n", ast->dshow_block_align); | |
943 | ✗ | ast->dshow_block_align = 0; | |
944 | } | ||
945 |
1/6✗ Branch 0 not taken.
✓ Branch 1 taken 47 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
|
47 | if (st->codecpar->codec_id == AV_CODEC_ID_AAC && ast->dshow_block_align == 1024 && ast->sample_size == 1024 || |
946 |
1/6✗ Branch 0 not taken.
✓ Branch 1 taken 47 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
|
47 | st->codecpar->codec_id == AV_CODEC_ID_AAC && ast->dshow_block_align == 4096 && ast->sample_size == 4096 || |
947 |
3/6✓ Branch 0 taken 2 times.
✓ Branch 1 taken 45 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 2 times.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
|
47 | st->codecpar->codec_id == AV_CODEC_ID_MP3 && ast->dshow_block_align == 1152 && ast->sample_size == 1152) { |
948 | ✗ | av_log(s, AV_LOG_DEBUG, "overriding sample_size\n"); | |
949 | ✗ | ast->sample_size = 0; | |
950 | } | ||
951 | 47 | break; | |
952 | ✗ | case AVMEDIA_TYPE_SUBTITLE: | |
953 | ✗ | st->codecpar->codec_type = AVMEDIA_TYPE_SUBTITLE; | |
954 | ✗ | sti->request_probe = 1; | |
955 | ✗ | avio_skip(pb, size); | |
956 | ✗ | break; | |
957 | ✗ | default: | |
958 | ✗ | st->codecpar->codec_type = AVMEDIA_TYPE_DATA; | |
959 | ✗ | st->codecpar->codec_id = AV_CODEC_ID_NONE; | |
960 | ✗ | st->codecpar->codec_tag = 0; | |
961 | ✗ | avio_skip(pb, size); | |
962 | ✗ | break; | |
963 | } | ||
964 | } | ||
965 | 520 | break; | |
966 | 11 | case MKTAG('s', 't', 'r', 'd'): | |
967 |
1/2✓ Branch 0 taken 11 times.
✗ Branch 1 not taken.
|
11 | if (stream_index >= (unsigned)s->nb_streams |
968 |
2/2✓ Branch 0 taken 6 times.
✓ Branch 1 taken 5 times.
|
11 | || s->streams[stream_index]->codecpar->extradata_size |
969 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
|
6 | || s->streams[stream_index]->codecpar->codec_tag == MKTAG('H','2','6','4')) { |
970 | 5 | avio_skip(pb, size); | |
971 | } else { | ||
972 | 6 | uint64_t cur_pos = avio_tell(pb); | |
973 |
1/2✓ Branch 0 taken 6 times.
✗ Branch 1 not taken.
|
6 | if (cur_pos < list_end) |
974 | 6 | size = FFMIN(size, list_end - cur_pos); | |
975 | 6 | st = s->streams[stream_index]; | |
976 | |||
977 |
1/2✓ Branch 0 taken 6 times.
✗ Branch 1 not taken.
|
6 | if (size<(1<<30)) { |
978 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
|
6 | if (st->codecpar->extradata) { |
979 | ✗ | av_log(s, AV_LOG_WARNING, "New extradata in strd chunk, freeing previous one.\n"); | |
980 | } | ||
981 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 6 times.
|
6 | if ((ret = ff_get_extradata(s, st->codecpar, pb, size)) < 0) |
982 | ✗ | return ret; | |
983 | } | ||
984 | |||
985 |
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 |
986 | ✗ | avio_r8(pb); | |
987 | |||
988 | 6 | ret = avi_extract_stream_metadata(s, st); | |
989 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
|
6 | if (ret < 0) { |
990 | ✗ | av_log(s, AV_LOG_WARNING, "could not decoding EXIF data in stream header.\n"); | |
991 | } | ||
992 | } | ||
993 | 11 | break; | |
994 | 22 | case MKTAG('i', 'n', 'd', 'x'): | |
995 | 22 | pos = avio_tell(pb); | |
996 |
2/4✓ Branch 0 taken 22 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 22 times.
✗ Branch 3 not taken.
|
22 | if ((pb->seekable & AVIO_SEEKABLE_NORMAL) && !(s->flags & AVFMT_FLAG_IGNIDX) && |
997 |
3/4✓ Branch 0 taken 22 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 13 times.
✓ Branch 3 taken 9 times.
|
44 | avi->use_odml && |
998 | 22 | read_odml_index(s, 0) < 0 && | |
999 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 13 times.
|
13 | (s->error_recognition & AV_EF_EXPLODE)) |
1000 | ✗ | return AVERROR_INVALIDDATA; | |
1001 | 22 | avio_seek(pb, pos + size, SEEK_SET); | |
1002 | 22 | break; | |
1003 | 10 | case MKTAG('v', 'p', 'r', 'p'): | |
1004 |
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) { |
1005 | AVRational active, active_aspect; | ||
1006 | |||
1007 | 10 | st = s->streams[stream_index]; | |
1008 | 10 | avio_rl32(pb); | |
1009 | 10 | avio_rl32(pb); | |
1010 | 10 | avio_rl32(pb); | |
1011 | 10 | avio_rl32(pb); | |
1012 | 10 | avio_rl32(pb); | |
1013 | |||
1014 | 10 | active_aspect.den = avio_rl16(pb); | |
1015 | 10 | active_aspect.num = avio_rl16(pb); | |
1016 | 10 | active.num = avio_rl32(pb); | |
1017 | 10 | active.den = avio_rl32(pb); | |
1018 | 10 | avio_rl32(pb); // nbFieldsPerFrame | |
1019 | |||
1020 |
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 && |
1021 |
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) { |
1022 | 10 | st->sample_aspect_ratio = av_div_q(active_aspect, active); | |
1023 | 10 | av_log(s, AV_LOG_TRACE, "vprp %d/%d %d/%d\n", | |
1024 | active_aspect.num, active_aspect.den, | ||
1025 | active.num, active.den); | ||
1026 | } | ||
1027 | 10 | size -= 9 * 4; | |
1028 | } | ||
1029 | 10 | avio_skip(pb, size); | |
1030 | 10 | break; | |
1031 | 19 | case MKTAG('s', 't', 'r', 'n'): | |
1032 | case MKTAG('i', 's', 'b', 'j'): | ||
1033 | case MKTAG('i', 'n', 'a', 'm'): | ||
1034 | case MKTAG('i', 'a', 'r', 't'): | ||
1035 | case MKTAG('i', 'c', 'o', 'p'): | ||
1036 | case MKTAG('i', 'c', 'm', 't'): | ||
1037 | case MKTAG('i', 'g', 'n', 'r'): | ||
1038 | case MKTAG('i', 'p', 'o', 'd'): | ||
1039 | case MKTAG('i', 's', 'o', 'f'): | ||
1040 |
1/2✓ Branch 0 taken 19 times.
✗ Branch 1 not taken.
|
19 | if (s->nb_streams) { |
1041 | 19 | ret = avi_read_tag(s, s->streams[s->nb_streams - 1], tag, size); | |
1042 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 19 times.
|
19 | if (ret < 0) |
1043 | ✗ | return ret; | |
1044 | 19 | break; | |
1045 | } | ||
1046 | default: | ||
1047 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1465 times.
|
1465 | if (size > 1000000) { |
1048 | ✗ | av_log(s, AV_LOG_ERROR, | |
1049 | "Something went wrong during header parsing, " | ||
1050 | "tag %s has size %u, " | ||
1051 | "I will ignore it and try to continue anyway.\n", | ||
1052 | ✗ | av_fourcc2str(tag), size); | |
1053 | ✗ | if (s->error_recognition & AV_EF_EXPLODE) | |
1054 | ✗ | return AVERROR_INVALIDDATA; | |
1055 | ✗ | avi->movi_list = avio_tell(pb) - 4; | |
1056 | ✗ | avi->movi_end = avi->fsize; | |
1057 | ✗ | goto end_of_header; | |
1058 | } | ||
1059 | /* Do not fail for very large idx1 tags */ | ||
1060 | case MKTAG('i', 'd', 'x', '1'): | ||
1061 | /* skip tag */ | ||
1062 | 1465 | size += (size & 1); | |
1063 | 1465 | avio_skip(pb, size); | |
1064 | 1465 | break; | |
1065 | } | ||
1066 | } | ||
1067 | |||
1068 | 476 | end_of_header: | |
1069 | /* check stream number */ | ||
1070 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 476 times.
|
476 | if (stream_index != s->nb_streams - 1) |
1071 | ✗ | return AVERROR_INVALIDDATA; | |
1072 | |||
1073 |
3/4✓ Branch 0 taken 470 times.
✓ Branch 1 taken 6 times.
✓ Branch 2 taken 470 times.
✗ Branch 3 not taken.
|
476 | if (!avi->index_loaded && (pb->seekable & AVIO_SEEKABLE_NORMAL)) |
1074 | 470 | avi_load_index(s); | |
1075 | 476 | calculate_bitrate(s); | |
1076 | 476 | avi->index_loaded |= 1; | |
1077 | |||
1078 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 476 times.
|
476 | if ((ret = guess_ni_flag(s)) < 0) |
1079 | ✗ | return ret; | |
1080 | |||
1081 | 476 | avi->non_interleaved |= ret | (s->flags & AVFMT_FLAG_SORT_DTS); | |
1082 | |||
1083 | 476 | dict_entry = av_dict_get(s->metadata, "ISFT", NULL, 0); | |
1084 |
3/4✓ Branch 0 taken 64 times.
✓ Branch 1 taken 412 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 64 times.
|
476 | if (dict_entry && !strcmp(dict_entry->value, "PotEncoder")) |
1085 | ✗ | for (i = 0; i < s->nb_streams; i++) { | |
1086 | ✗ | AVStream *st = s->streams[i]; | |
1087 | ✗ | if ( st->codecpar->codec_id == AV_CODEC_ID_MPEG1VIDEO | |
1088 | ✗ | || st->codecpar->codec_id == AV_CODEC_ID_MPEG2VIDEO) | |
1089 | ✗ | ffstream(st)->need_parsing = AVSTREAM_PARSE_FULL; | |
1090 | } | ||
1091 | |||
1092 |
2/2✓ Branch 0 taken 496 times.
✓ Branch 1 taken 38 times.
|
534 | for (i = 0; i < s->nb_streams; i++) { |
1093 | 496 | AVStream *st = s->streams[i]; | |
1094 |
2/2✓ Branch 1 taken 438 times.
✓ Branch 2 taken 58 times.
|
496 | if (ffstream(st)->nb_index_entries) |
1095 | 438 | break; | |
1096 | } | ||
1097 | // DV-in-AVI cannot be non-interleaved, if set this must be | ||
1098 | // a mis-detection. | ||
1099 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 476 times.
|
476 | if (avi->dv_demux) |
1100 | ✗ | avi->non_interleaved = 0; | |
1101 |
4/4✓ Branch 0 taken 38 times.
✓ Branch 1 taken 438 times.
✓ Branch 2 taken 2 times.
✓ Branch 3 taken 36 times.
|
476 | if (i == s->nb_streams && avi->non_interleaved) { |
1102 | 2 | av_log(s, AV_LOG_WARNING, | |
1103 | "Non-interleaved AVI without index, switching to interleaved\n"); | ||
1104 | 2 | avi->non_interleaved = 0; | |
1105 | } | ||
1106 | |||
1107 |
2/2✓ Branch 0 taken 7 times.
✓ Branch 1 taken 469 times.
|
476 | if (avi->non_interleaved) { |
1108 | 7 | av_log(s, AV_LOG_INFO, "non-interleaved AVI\n"); | |
1109 | 7 | clean_index(s); | |
1110 | } | ||
1111 | |||
1112 | 476 | ff_metadata_conv_ctx(s, NULL, avi_metadata_conv); | |
1113 | 476 | ff_metadata_conv_ctx(s, NULL, ff_riff_info_conv); | |
1114 | |||
1115 | 476 | return 0; | |
1116 | } | ||
1117 | |||
1118 | ✗ | static int read_gab2_sub(AVFormatContext *s, AVStream *st, AVPacket *pkt) | |
1119 | { | ||
1120 | ✗ | if (pkt->size >= 7 && | |
1121 | ✗ | pkt->size < INT_MAX - AVPROBE_PADDING_SIZE && | |
1122 | ✗ | !strcmp(pkt->data, "GAB2") && AV_RL16(pkt->data + 5) == 2) { | |
1123 | uint8_t desc[256]; | ||
1124 | ✗ | int score = AVPROBE_SCORE_EXTENSION, ret; | |
1125 | ✗ | AVIStream *ast = st->priv_data; | |
1126 | const AVInputFormat *sub_demuxer; | ||
1127 | AVRational time_base; | ||
1128 | int size; | ||
1129 | AVProbeData pd; | ||
1130 | unsigned int desc_len; | ||
1131 | |||
1132 | ✗ | if (ast->sub_ctx) | |
1133 | ✗ | return 0; | |
1134 | |||
1135 | ✗ | AVIOContext *pb = avio_alloc_context(pkt->data + 7, | |
1136 | ✗ | pkt->size - 7, | |
1137 | 0, NULL, NULL, NULL, NULL); | ||
1138 | ✗ | if (!pb) | |
1139 | ✗ | goto error; | |
1140 | |||
1141 | ✗ | desc_len = avio_rl32(pb); | |
1142 | |||
1143 | ✗ | if (desc_len > pb->buf_end - pb->buf_ptr) | |
1144 | ✗ | goto error; | |
1145 | |||
1146 | ✗ | ret = avio_get_str16le(pb, desc_len, desc, sizeof(desc)); | |
1147 | ✗ | avio_skip(pb, desc_len - ret); | |
1148 | ✗ | if (*desc) | |
1149 | ✗ | av_dict_set(&st->metadata, "title", desc, 0); | |
1150 | |||
1151 | ✗ | avio_rl16(pb); /* flags? */ | |
1152 | ✗ | avio_rl32(pb); /* data size */ | |
1153 | |||
1154 | ✗ | size = pb->buf_end - pb->buf_ptr; | |
1155 | ✗ | pd = (AVProbeData) { .buf = av_mallocz(size + AVPROBE_PADDING_SIZE), | |
1156 | .buf_size = size }; | ||
1157 | ✗ | if (!pd.buf) | |
1158 | ✗ | goto error; | |
1159 | ✗ | memcpy(pd.buf, pb->buf_ptr, size); | |
1160 | ✗ | sub_demuxer = av_probe_input_format2(&pd, 1, &score); | |
1161 | ✗ | av_freep(&pd.buf); | |
1162 | ✗ | if (!sub_demuxer) | |
1163 | ✗ | goto error; | |
1164 | |||
1165 | ✗ | if (strcmp(sub_demuxer->name, "srt") && strcmp(sub_demuxer->name, "ass")) | |
1166 | ✗ | goto error; | |
1167 | |||
1168 | ✗ | if (!(ast->sub_pkt = av_packet_alloc())) | |
1169 | ✗ | goto error; | |
1170 | |||
1171 | ✗ | if (!(ast->sub_ctx = avformat_alloc_context())) | |
1172 | ✗ | goto error; | |
1173 | |||
1174 | ✗ | ast->sub_ctx->pb = pb; | |
1175 | |||
1176 | ✗ | if (ff_copy_whiteblacklists(ast->sub_ctx, s) < 0) | |
1177 | ✗ | goto error; | |
1178 | |||
1179 | ✗ | if (!avformat_open_input(&ast->sub_ctx, "", sub_demuxer, NULL)) { | |
1180 | ✗ | if (ast->sub_ctx->nb_streams != 1) | |
1181 | ✗ | goto error; | |
1182 | ✗ | ff_read_packet(ast->sub_ctx, ast->sub_pkt); | |
1183 | ✗ | avcodec_parameters_copy(st->codecpar, ast->sub_ctx->streams[0]->codecpar); | |
1184 | ✗ | time_base = ast->sub_ctx->streams[0]->time_base; | |
1185 | ✗ | avpriv_set_pts_info(st, 64, time_base.num, time_base.den); | |
1186 | } | ||
1187 | ✗ | ast->sub_buffer = pkt->buf; | |
1188 | ✗ | pkt->buf = NULL; | |
1189 | ✗ | av_packet_unref(pkt); | |
1190 | ✗ | return 1; | |
1191 | |||
1192 | ✗ | error: | |
1193 | ✗ | av_packet_free(&ast->sub_pkt); | |
1194 | ✗ | av_freep(&ast->sub_ctx); | |
1195 | ✗ | avio_context_free(&pb); | |
1196 | } | ||
1197 | ✗ | return 0; | |
1198 | } | ||
1199 | |||
1200 | 27447 | static AVStream *get_subtitle_pkt(AVFormatContext *s, AVStream *next_st, | |
1201 | AVPacket *pkt) | ||
1202 | { | ||
1203 | 27447 | AVIStream *ast, *next_ast = next_st->priv_data; | |
1204 | 27447 | int64_t ts, next_ts, ts_min = INT64_MAX; | |
1205 | 27447 | AVStream *st, *sub_st = NULL; | |
1206 | int i; | ||
1207 | |||
1208 | 27447 | next_ts = av_rescale_q(next_ast->frame_offset, next_st->time_base, | |
1209 | 27447 | AV_TIME_BASE_Q); | |
1210 | |||
1211 |
2/2✓ Branch 0 taken 34250 times.
✓ Branch 1 taken 27447 times.
|
61697 | for (i = 0; i < s->nb_streams; i++) { |
1212 | 34250 | st = s->streams[i]; | |
1213 | 34250 | ast = st->priv_data; | |
1214 |
4/8✓ Branch 0 taken 29642 times.
✓ Branch 1 taken 4608 times.
✓ Branch 2 taken 29642 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 29642 times.
✗ Branch 6 not taken.
✗ Branch 7 not taken.
|
34250 | if (st->discard < AVDISCARD_ALL && ast && ast->sub_pkt && ast->sub_pkt->data) { |
1215 | ✗ | ts = av_rescale_q(ast->sub_pkt->dts, st->time_base, AV_TIME_BASE_Q); | |
1216 | ✗ | if (ts <= next_ts && ts < ts_min) { | |
1217 | ✗ | ts_min = ts; | |
1218 | ✗ | sub_st = st; | |
1219 | } | ||
1220 | } | ||
1221 | } | ||
1222 | |||
1223 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 27447 times.
|
27447 | if (sub_st) { |
1224 | ✗ | ast = sub_st->priv_data; | |
1225 | ✗ | av_packet_move_ref(pkt, ast->sub_pkt); | |
1226 | ✗ | pkt->stream_index = sub_st->index; | |
1227 | |||
1228 | ✗ | if (ff_read_packet(ast->sub_ctx, ast->sub_pkt) < 0) | |
1229 | ✗ | ast->sub_pkt->data = NULL; | |
1230 | } | ||
1231 | 27447 | return sub_st; | |
1232 | } | ||
1233 | |||
1234 | 524575 | static int get_stream_idx(const unsigned *d) | |
1235 | { | ||
1236 |
4/4✓ Branch 0 taken 298660 times.
✓ Branch 1 taken 225915 times.
✓ Branch 2 taken 95709 times.
✓ Branch 3 taken 202951 times.
|
524575 | if (d[0] >= '0' && d[0] <= '9' && |
1237 |
4/4✓ Branch 0 taken 95349 times.
✓ Branch 1 taken 360 times.
✓ Branch 2 taken 64212 times.
✓ Branch 3 taken 31137 times.
|
95709 | d[1] >= '0' && d[1] <= '9') { |
1238 | 64212 | return (d[0] - '0') * 10 + (d[1] - '0'); | |
1239 | } else { | ||
1240 | 460363 | return 100; // invalid stream ID | |
1241 | } | ||
1242 | } | ||
1243 | |||
1244 | /** | ||
1245 | * | ||
1246 | * @param exit_early set to 1 to just gather packet position without making the changes needed to actually read & return the packet | ||
1247 | */ | ||
1248 | 27534 | static int avi_sync(AVFormatContext *s, int exit_early) | |
1249 | { | ||
1250 | 27534 | AVIContext *avi = s->priv_data; | |
1251 | 27534 | AVIOContext *pb = s->pb; | |
1252 | int n; | ||
1253 | unsigned int d[8]; | ||
1254 | unsigned int size; | ||
1255 | int64_t i, sync; | ||
1256 | |||
1257 | 5102 | start_sync: | |
1258 | 32636 | memset(d, -1, sizeof(d)); | |
1259 |
2/2✓ Branch 2 taken 368699 times.
✓ Branch 3 taken 487 times.
|
369186 | for (i = sync = avio_tell(pb); !avio_feof(pb); i++) { |
1260 | int j; | ||
1261 | |||
1262 |
2/2✓ Branch 0 taken 2580893 times.
✓ Branch 1 taken 368699 times.
|
2949592 | for (j = 0; j < 7; j++) |
1263 | 2580893 | d[j] = d[j + 1]; | |
1264 | 368699 | d[7] = avio_r8(pb); | |
1265 | |||
1266 | 368699 | size = d[4] + (d[5] << 8) + (d[6] << 16) + (d[7] << 24); | |
1267 | |||
1268 | 368699 | n = get_stream_idx(d + 2); | |
1269 | ff_tlog(s, "%X %X %X %X %X %X %X %X %"PRId64" %u %d\n", | ||
1270 | d[0], d[1], d[2], d[3], d[4], d[5], d[6], d[7], i, size, n); | ||
1271 |
4/4✓ Branch 0 taken 128538 times.
✓ Branch 1 taken 240161 times.
✓ Branch 2 taken 14050 times.
✓ Branch 3 taken 114488 times.
|
368699 | if (i*(avi->io_fsize>0) + (uint64_t)size > avi->fsize || d[0] > 127) |
1272 | 254211 | continue; | |
1273 | |||
1274 | // parse ix## | ||
1275 |
5/6✓ Branch 0 taken 480 times.
✓ Branch 1 taken 114008 times.
✓ Branch 2 taken 1 times.
✓ Branch 3 taken 479 times.
✗ Branch 4 not taken.
✓ Branch 5 taken 1 times.
|
114488 | if ((d[0] == 'i' && d[1] == 'x' && n < s->nb_streams) || |
1276 | // parse JUNK | ||
1277 |
6/8✓ Branch 0 taken 468 times.
✓ Branch 1 taken 114019 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.
|
114487 | (d[0] == 'J' && d[1] == 'U' && d[2] == 'N' && d[3] == 'K') || |
1278 |
6/8✓ Branch 0 taken 479 times.
✓ Branch 1 taken 113541 times.
✓ Branch 2 taken 478 times.
✓ Branch 3 taken 1 times.
✓ Branch 4 taken 478 times.
✗ Branch 5 not taken.
✗ Branch 6 not taken.
✓ Branch 7 taken 478 times.
|
114020 | (d[0] == 'i' && d[1] == 'd' && d[2] == 'x' && d[3] == '1') || |
1279 |
4/8✓ Branch 0 taken 1 times.
✓ Branch 1 taken 113541 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.
|
113542 | (d[0] == 'i' && d[1] == 'n' && d[2] == 'd' && d[3] == 'x')) { |
1280 | 946 | avio_skip(pb, size); | |
1281 | 946 | goto start_sync; | |
1282 | } | ||
1283 | |||
1284 | // parse stray LIST | ||
1285 |
6/8✓ Branch 0 taken 663 times.
✓ Branch 1 taken 112879 times.
✓ Branch 2 taken 662 times.
✓ Branch 3 taken 1 times.
✓ Branch 4 taken 662 times.
✗ Branch 5 not taken.
✓ Branch 6 taken 662 times.
✗ Branch 7 not taken.
|
113542 | if (d[0] == 'L' && d[1] == 'I' && d[2] == 'S' && d[3] == 'T') { |
1286 | 662 | avio_skip(pb, 4); | |
1287 | 662 | goto start_sync; | |
1288 | } | ||
1289 | |||
1290 | 112880 | n = get_stream_idx(d); | |
1291 | |||
1292 |
2/2✓ Branch 0 taken 42158 times.
✓ Branch 1 taken 70722 times.
|
112880 | if (!((i - avi->last_pkt_pos) & 1) && |
1293 |
2/2✓ Branch 1 taken 971 times.
✓ Branch 2 taken 41187 times.
|
42158 | get_stream_idx(d + 1) < s->nb_streams) |
1294 | 971 | continue; | |
1295 | |||
1296 | // detect ##ix chunk and skip | ||
1297 |
1/6✗ Branch 0 not taken.
✓ Branch 1 taken 111909 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
|
111909 | if (d[2] == 'i' && d[3] == 'x' && n < s->nb_streams) { |
1298 | ✗ | avio_skip(pb, size); | |
1299 | ✗ | goto start_sync; | |
1300 | } | ||
1301 | |||
1302 |
3/6✓ Branch 0 taken 6304 times.
✓ Branch 1 taken 105605 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 6304 times.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
|
111909 | if (d[2] == 'w' && d[3] == 'c' && n < s->nb_streams) { |
1303 | ✗ | avio_skip(pb, 16 * 3 + 8); | |
1304 | ✗ | goto start_sync; | |
1305 | } | ||
1306 | |||
1307 |
1/4✗ Branch 0 not taken.
✓ Branch 1 taken 111909 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
|
111909 | if (avi->dv_demux && n != 0) |
1308 | ✗ | continue; | |
1309 | |||
1310 | // parse ##dc/##wb | ||
1311 |
2/2✓ Branch 0 taken 30541 times.
✓ Branch 1 taken 81368 times.
|
111909 | if (n < s->nb_streams) { |
1312 | AVStream *st; | ||
1313 | AVIStream *ast; | ||
1314 | 30541 | st = s->streams[n]; | |
1315 | 30541 | ast = st->priv_data; | |
1316 | |||
1317 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 30541 times.
|
30541 | if (!ast) { |
1318 | ✗ | av_log(s, AV_LOG_WARNING, "Skipping foreign stream %d packet\n", n); | |
1319 | ✗ | continue; | |
1320 | } | ||
1321 | |||
1322 |
2/2✓ Branch 0 taken 9747 times.
✓ Branch 1 taken 20794 times.
|
30541 | if (s->nb_streams >= 2) { |
1323 | 9747 | AVStream *st1 = s->streams[1]; | |
1324 | 9747 | AVIStream *ast1 = st1->priv_data; | |
1325 | // workaround for broken small-file-bug402.avi | ||
1326 |
4/6✓ Branch 0 taken 9747 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 3699 times.
✓ Branch 3 taken 6048 times.
✓ Branch 4 taken 3699 times.
✗ Branch 5 not taken.
|
9747 | if (ast1 && d[2] == 'w' && d[3] == 'b' |
1327 |
2/2✓ Branch 0 taken 314 times.
✓ Branch 1 taken 3385 times.
|
3699 | && n == 0 |
1328 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 314 times.
|
314 | && st ->codecpar->codec_type == AVMEDIA_TYPE_VIDEO |
1329 | ✗ | && st1->codecpar->codec_type == AVMEDIA_TYPE_AUDIO | |
1330 | ✗ | && ast->prefix == 'd'*256+'c' | |
1331 | ✗ | && (d[2]*256+d[3] == ast1->prefix || !ast1->prefix_count) | |
1332 | ) { | ||
1333 | ✗ | n = 1; | |
1334 | ✗ | st = st1; | |
1335 | ✗ | ast = ast1; | |
1336 | ✗ | av_log(s, AV_LOG_WARNING, | |
1337 | "Invalid stream + prefix combination, assuming audio.\n"); | ||
1338 | } | ||
1339 | } | ||
1340 | |||
1341 |
4/6✓ Branch 0 taken 4 times.
✓ Branch 1 taken 30537 times.
✓ Branch 2 taken 4 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 4 times.
✗ Branch 5 not taken.
|
30541 | if (d[2] == 'p' && d[3] == 'c' && size <= 4 * 256 + 4) { |
1342 | 4 | int k = avio_r8(pb); | |
1343 | 4 | int last = (k + avio_r8(pb) - 1) & 0xFF; | |
1344 | |||
1345 | 4 | avio_rl16(pb); // flags | |
1346 | |||
1347 | // b + (g << 8) + (r << 16); | ||
1348 |
2/2✓ Branch 0 taken 759 times.
✓ Branch 1 taken 4 times.
|
763 | for (; k <= last; k++) |
1349 | 759 | ast->pal[k] = 0xFFU<<24 | avio_rb32(pb)>>8; | |
1350 | |||
1351 | 4 | ast->has_pal = 1; | |
1352 | 4 | goto start_sync; | |
1353 |
4/4✓ Branch 0 taken 27087 times.
✓ Branch 1 taken 3450 times.
✓ Branch 2 taken 27073 times.
✓ Branch 3 taken 14 times.
|
30537 | } else if (((ast->prefix_count < 5 || sync + 9 > i) && |
1354 |
2/4✓ Branch 0 taken 30523 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 30523 times.
|
30523 | d[2] < 128 && d[3] < 128) || |
1355 |
1/2✓ Branch 0 taken 14 times.
✗ Branch 1 not taken.
|
14 | d[2] * 256 + d[3] == ast->prefix /* || |
1356 | (d[2] == 'd' && d[3] == 'c') || | ||
1357 | (d[2] == 'w' && d[3] == 'b') */) { | ||
1358 |
2/2✓ Branch 0 taken 432 times.
✓ Branch 1 taken 30105 times.
|
30537 | if (exit_early) |
1359 | 432 | return 0; | |
1360 |
2/2✓ Branch 0 taken 29514 times.
✓ Branch 1 taken 591 times.
|
30105 | if (d[2] * 256 + d[3] == ast->prefix) |
1361 | 29514 | ast->prefix_count++; | |
1362 | else { | ||
1363 | 591 | ast->prefix = d[2] * 256 + d[3]; | |
1364 | 591 | ast->prefix_count = 0; | |
1365 | } | ||
1366 | |||
1367 |
1/2✓ Branch 0 taken 30105 times.
✗ Branch 1 not taken.
|
30105 | if (!avi->dv_demux && |
1368 |
4/4✓ Branch 0 taken 4925 times.
✓ Branch 1 taken 25180 times.
✓ Branch 2 taken 4123 times.
✓ Branch 3 taken 802 times.
|
30105 | ((st->discard >= AVDISCARD_DEFAULT && size == 0) /* || |
1369 | // FIXME: needs a little reordering | ||
1370 | (st->discard >= AVDISCARD_NONKEY && | ||
1371 | !(pkt->flags & AV_PKT_FLAG_KEY)) */ | ||
1372 |
2/2✓ Branch 0 taken 2688 times.
✓ Branch 1 taken 26615 times.
|
29303 | || st->discard >= AVDISCARD_ALL)) { |
1373 | |||
1374 | 3490 | ast->frame_offset += get_duration(ast, size); | |
1375 | 3490 | avio_skip(pb, size); | |
1376 | 3490 | goto start_sync; | |
1377 | } | ||
1378 | |||
1379 | 26615 | avi->stream_index = n; | |
1380 | 26615 | ast->packet_size = size + 8; | |
1381 | 26615 | ast->remaining = size; | |
1382 | |||
1383 |
2/2✓ Branch 0 taken 25168 times.
✓ Branch 1 taken 1447 times.
|
26615 | if (size) { |
1384 | 25168 | FFStream *const sti = ffstream(st); | |
1385 | 25168 | uint64_t pos = avio_tell(pb) - 8; | |
1386 |
3/4✓ Branch 0 taken 25111 times.
✓ Branch 1 taken 57 times.
✓ Branch 2 taken 25111 times.
✗ Branch 3 not taken.
|
25168 | if (!sti->index_entries || !sti->nb_index_entries || |
1387 |
2/2✓ Branch 0 taken 3160 times.
✓ Branch 1 taken 21951 times.
|
25111 | sti->index_entries[sti->nb_index_entries - 1].pos < pos) { |
1388 | 3217 | av_add_index_entry(st, pos, ast->frame_offset, size, | |
1389 | 0, AVINDEX_KEYFRAME); | ||
1390 | } | ||
1391 | } | ||
1392 | 26615 | return 0; | |
1393 | } | ||
1394 | } | ||
1395 | } | ||
1396 | |||
1397 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 487 times.
|
487 | if (pb->error) |
1398 | ✗ | return pb->error; | |
1399 | 487 | return AVERROR_EOF; | |
1400 | } | ||
1401 | |||
1402 | 766 | static int ni_prepare_read(AVFormatContext *s) | |
1403 | { | ||
1404 | 766 | AVIContext *avi = s->priv_data; | |
1405 | 766 | int best_stream_index = 0; | |
1406 | 766 | AVStream *best_st = NULL; | |
1407 | FFStream *best_sti; | ||
1408 | AVIStream *best_ast; | ||
1409 | 766 | int64_t best_ts = INT64_MAX; | |
1410 | int i; | ||
1411 | |||
1412 |
2/2✓ Branch 0 taken 1288 times.
✓ Branch 1 taken 766 times.
|
2054 | for (i = 0; i < s->nb_streams; i++) { |
1413 | 1288 | AVStream *st = s->streams[i]; | |
1414 | 1288 | FFStream *const sti = ffstream(st); | |
1415 | 1288 | AVIStream *ast = st->priv_data; | |
1416 | 1288 | int64_t ts = ast->frame_offset; | |
1417 | int64_t last_ts; | ||
1418 | |||
1419 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1288 times.
|
1288 | if (!sti->nb_index_entries) |
1420 | 16 | continue; | |
1421 | |||
1422 | 1288 | last_ts = sti->index_entries[sti->nb_index_entries - 1].timestamp; | |
1423 |
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) |
1424 | 16 | continue; | |
1425 | |||
1426 | 1272 | ts = av_rescale_q(ts, st->time_base, | |
1427 | 1272 | (AVRational) { FFMAX(1, ast->sample_size), | |
1428 | AV_TIME_BASE }); | ||
1429 | |||
1430 | 1272 | av_log(s, AV_LOG_TRACE, "%"PRId64" %d/%d %"PRId64"\n", ts, | |
1431 | st->time_base.num, st->time_base.den, ast->frame_offset); | ||
1432 |
2/2✓ Branch 0 taken 975 times.
✓ Branch 1 taken 297 times.
|
1272 | if (ts < best_ts) { |
1433 | 975 | best_ts = ts; | |
1434 | 975 | best_st = st; | |
1435 | 975 | best_stream_index = i; | |
1436 | } | ||
1437 | } | ||
1438 |
2/2✓ Branch 0 taken 5 times.
✓ Branch 1 taken 761 times.
|
766 | if (!best_st) |
1439 | 5 | return AVERROR_EOF; | |
1440 | |||
1441 | 761 | best_sti = ffstream(best_st); | |
1442 | 761 | best_ast = best_st->priv_data; | |
1443 | 761 | best_ts = best_ast->frame_offset; | |
1444 |
2/2✓ Branch 0 taken 181 times.
✓ Branch 1 taken 580 times.
|
761 | if (best_ast->remaining) { |
1445 | 181 | i = av_index_search_timestamp(best_st, | |
1446 | best_ts, | ||
1447 | AVSEEK_FLAG_ANY | | ||
1448 | AVSEEK_FLAG_BACKWARD); | ||
1449 | } else { | ||
1450 | 580 | i = av_index_search_timestamp(best_st, best_ts, AVSEEK_FLAG_ANY); | |
1451 |
1/2✓ Branch 0 taken 580 times.
✗ Branch 1 not taken.
|
580 | if (i >= 0) |
1452 | 580 | best_ast->frame_offset = best_sti->index_entries[i].timestamp; | |
1453 | } | ||
1454 | |||
1455 |
1/2✓ Branch 0 taken 761 times.
✗ Branch 1 not taken.
|
761 | if (i >= 0) { |
1456 | 761 | int64_t pos = best_sti->index_entries[i].pos; | |
1457 | 761 | pos += best_ast->packet_size - best_ast->remaining; | |
1458 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 761 times.
|
761 | if (avio_seek(s->pb, pos + 8, SEEK_SET) < 0) |
1459 | ✗ | return AVERROR_EOF; | |
1460 | |||
1461 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 761 times.
|
761 | av_assert0(best_ast->remaining <= best_ast->packet_size); |
1462 | |||
1463 | 761 | avi->stream_index = best_stream_index; | |
1464 |
2/2✓ Branch 0 taken 580 times.
✓ Branch 1 taken 181 times.
|
761 | if (!best_ast->remaining) |
1465 | 580 | best_ast->packet_size = | |
1466 | 580 | best_ast->remaining = best_sti->index_entries[i].size; | |
1467 | } | ||
1468 | else | ||
1469 | ✗ | return AVERROR_EOF; | |
1470 | |||
1471 | 761 | return 0; | |
1472 | } | ||
1473 | |||
1474 | 27938 | static int avi_read_packet(AVFormatContext *s, AVPacket *pkt) | |
1475 | { | ||
1476 | 27938 | AVIContext *avi = s->priv_data; | |
1477 | 27938 | AVIOContext *pb = s->pb; | |
1478 | int err; | ||
1479 | |||
1480 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 27938 times.
|
27938 | if (CONFIG_DV_DEMUXER && avi->dv_demux) { |
1481 | ✗ | int size = avpriv_dv_get_packet(avi->dv_demux, pkt); | |
1482 | ✗ | if (size >= 0) | |
1483 | ✗ | return size; | |
1484 | else | ||
1485 | ✗ | goto resync; | |
1486 | } | ||
1487 | |||
1488 |
2/2✓ Branch 0 taken 27172 times.
✓ Branch 1 taken 766 times.
|
27938 | if (avi->non_interleaved) { |
1489 | 766 | err = ni_prepare_read(s); | |
1490 |
2/2✓ Branch 0 taken 761 times.
✓ Branch 1 taken 5 times.
|
766 | if (err < 0) |
1491 | 5 | return err; | |
1492 | } | ||
1493 | |||
1494 | 54549 | resync: | |
1495 |
2/2✓ Branch 0 taken 27447 times.
✓ Branch 1 taken 27102 times.
|
54549 | if (avi->stream_index >= 0) { |
1496 | 27447 | AVStream *st = s->streams[avi->stream_index]; | |
1497 | 27447 | FFStream *const sti = ffstream(st); | |
1498 | 27447 | AVIStream *ast = st->priv_data; | |
1499 | 27447 | int dv_demux = CONFIG_DV_DEMUXER && avi->dv_demux; | |
1500 | int size, err; | ||
1501 | |||
1502 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 27447 times.
|
27447 | if (get_subtitle_pkt(s, st, pkt)) |
1503 | ✗ | return 0; | |
1504 | |||
1505 | // minorityreport.AVI block_align=1024 sample_size=1 IMA-ADPCM | ||
1506 |
2/2✓ Branch 0 taken 24064 times.
✓ Branch 1 taken 3383 times.
|
27447 | if (ast->sample_size <= 1) |
1507 | 24064 | size = INT_MAX; | |
1508 |
2/2✓ Branch 0 taken 239 times.
✓ Branch 1 taken 3144 times.
|
3383 | else if (ast->sample_size < 32) |
1509 | // arbitrary multiplier to avoid tiny packets for raw PCM data | ||
1510 | 239 | size = 1024 * ast->sample_size; | |
1511 | else | ||
1512 | 3144 | size = ast->sample_size; | |
1513 | |||
1514 |
2/2✓ Branch 0 taken 24076 times.
✓ Branch 1 taken 3371 times.
|
27447 | if (size > ast->remaining) |
1515 | 24076 | size = ast->remaining; | |
1516 | 27447 | avi->last_pkt_pos = avio_tell(pb); | |
1517 | 27447 | err = av_get_packet(pb, pkt, size); | |
1518 |
2/2✓ Branch 0 taken 38 times.
✓ Branch 1 taken 27409 times.
|
27447 | if (err < 0) |
1519 | 38 | return err; | |
1520 | 27409 | size = err; | |
1521 | |||
1522 |
4/6✓ Branch 0 taken 19 times.
✓ Branch 1 taken 27390 times.
✓ Branch 2 taken 19 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 19 times.
✗ Branch 5 not taken.
|
27409 | if (ast->has_pal && pkt->size < (unsigned)INT_MAX / 2 && !dv_demux) { |
1523 | uint8_t *pal; | ||
1524 | 19 | pal = av_packet_new_side_data(pkt, | |
1525 | AV_PKT_DATA_PALETTE, | ||
1526 | AVPALETTE_SIZE); | ||
1527 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 19 times.
|
19 | if (!pal) { |
1528 | ✗ | av_log(s, AV_LOG_ERROR, | |
1529 | "Failed to allocate data for palette\n"); | ||
1530 | } else { | ||
1531 | 19 | memcpy(pal, ast->pal, AVPALETTE_SIZE); | |
1532 | 19 | ast->has_pal = 0; | |
1533 | } | ||
1534 | } | ||
1535 | |||
1536 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 27409 times.
|
27409 | if (CONFIG_DV_DEMUXER && dv_demux) { |
1537 | ✗ | size = avpriv_dv_produce_packet(avi->dv_demux, pkt, | |
1538 | pkt->data, pkt->size, pkt->pos); | ||
1539 | ✗ | pkt->flags |= AV_PKT_FLAG_KEY; | |
1540 | ✗ | if (size < 0) | |
1541 | ✗ | av_packet_unref(pkt); | |
1542 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 27409 times.
|
27409 | } else if (st->codecpar->codec_type == AVMEDIA_TYPE_SUBTITLE && |
1543 | ✗ | !st->codecpar->codec_tag && read_gab2_sub(s, st, pkt)) { | |
1544 | ✗ | ast->frame_offset++; | |
1545 | ✗ | avi->stream_index = -1; | |
1546 | ✗ | ast->remaining = 0; | |
1547 | ✗ | goto resync; | |
1548 | } else { | ||
1549 | /* XXX: How to handle B-frames in AVI? */ | ||
1550 | 27409 | pkt->dts = ast->frame_offset; | |
1551 | // pkt->dts += ast->start; | ||
1552 |
2/2✓ Branch 0 taken 3752 times.
✓ Branch 1 taken 23657 times.
|
27409 | if (ast->sample_size) |
1553 | 3752 | pkt->dts /= ast->sample_size; | |
1554 | 27409 | pkt->stream_index = avi->stream_index; | |
1555 | |||
1556 |
3/4✓ Branch 0 taken 23232 times.
✓ Branch 1 taken 4177 times.
✓ Branch 2 taken 23232 times.
✗ Branch 3 not taken.
|
50641 | if (st->codecpar->codec_type == AVMEDIA_TYPE_VIDEO && sti->index_entries) { |
1557 | AVIndexEntry *e; | ||
1558 | int index; | ||
1559 | |||
1560 | 23232 | index = av_index_search_timestamp(st, ast->frame_offset, AVSEEK_FLAG_ANY); | |
1561 | 23232 | e = &sti->index_entries[index]; | |
1562 | |||
1563 |
4/4✓ Branch 0 taken 23119 times.
✓ Branch 1 taken 113 times.
✓ Branch 2 taken 21784 times.
✓ Branch 3 taken 1335 times.
|
23232 | if (index >= 0 && e->timestamp == ast->frame_offset) { |
1564 |
2/2✓ Branch 0 taken 2660 times.
✓ Branch 1 taken 19124 times.
|
21784 | if (index == sti->nb_index_entries-1) { |
1565 | 2660 | int key=1; | |
1566 | 2660 | uint32_t state=-1; | |
1567 |
2/2✓ Branch 0 taken 52 times.
✓ Branch 1 taken 2608 times.
|
2660 | if (st->codecpar->codec_id == AV_CODEC_ID_MPEG4) { |
1568 | 52 | const uint8_t *ptr = pkt->data, *end = ptr + FFMIN(size, 256); | |
1569 |
1/2✓ Branch 0 taken 109 times.
✗ Branch 1 not taken.
|
109 | while (ptr < end) { |
1570 | 109 | ptr = avpriv_find_start_code(ptr, end, &state); | |
1571 |
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) { |
1572 | 52 | key = !(*ptr & 0xC0); | |
1573 | 52 | break; | |
1574 | } | ||
1575 | } | ||
1576 | } | ||
1577 |
2/2✓ Branch 0 taken 39 times.
✓ Branch 1 taken 2621 times.
|
2660 | if (!key) |
1578 | 39 | e->flags &= ~AVINDEX_KEYFRAME; | |
1579 | } | ||
1580 |
2/2✓ Branch 0 taken 12726 times.
✓ Branch 1 taken 9058 times.
|
21784 | if (e->flags & AVINDEX_KEYFRAME) |
1581 | 12726 | pkt->flags |= AV_PKT_FLAG_KEY; | |
1582 | } | ||
1583 | } else { | ||
1584 | 4177 | pkt->flags |= AV_PKT_FLAG_KEY; | |
1585 | } | ||
1586 | 27409 | ast->frame_offset += get_duration(ast, pkt->size); | |
1587 | } | ||
1588 | 27409 | ast->remaining -= err; | |
1589 |
2/2✓ Branch 0 taken 27160 times.
✓ Branch 1 taken 249 times.
|
27409 | if (!ast->remaining) { |
1590 | 27160 | avi->stream_index = -1; | |
1591 | 27160 | ast->packet_size = 0; | |
1592 | } | ||
1593 | |||
1594 |
6/6✓ Branch 0 taken 26651 times.
✓ Branch 1 taken 758 times.
✓ Branch 2 taken 25204 times.
✓ Branch 3 taken 1447 times.
✓ Branch 4 taken 1 times.
✓ Branch 5 taken 25203 times.
|
27409 | if (!avi->non_interleaved && pkt->pos >= 0 && ast->seek_pos > pkt->pos) { |
1595 | 1 | av_packet_unref(pkt); | |
1596 | 1 | goto resync; | |
1597 | } | ||
1598 | 27408 | ast->seek_pos= 0; | |
1599 | |||
1600 |
6/6✓ Branch 0 taken 26650 times.
✓ Branch 1 taken 758 times.
✓ Branch 2 taken 26548 times.
✓ Branch 3 taken 102 times.
✓ Branch 4 taken 23494 times.
✓ Branch 5 taken 3054 times.
|
27408 | if (!avi->non_interleaved && sti->nb_index_entries > 1 && avi->index_loaded > 1) { |
1601 | 23494 | int64_t dts= av_rescale_q(pkt->dts, st->time_base, AV_TIME_BASE_Q); | |
1602 | |||
1603 |
2/2✓ Branch 0 taken 22801 times.
✓ Branch 1 taken 693 times.
|
23494 | if (avi->dts_max < dts) { |
1604 | 22801 | avi->dts_max = dts; | |
1605 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 693 times.
|
693 | } else if (avi->dts_max - (uint64_t)dts > 2*AV_TIME_BASE) { |
1606 | ✗ | avi->non_interleaved= 1; | |
1607 | ✗ | av_log(s, AV_LOG_INFO, "Switching to NI mode, due to poor interleaving\n"); | |
1608 | } | ||
1609 | } | ||
1610 | |||
1611 | 27408 | return 0; | |
1612 | } | ||
1613 | |||
1614 |
2/2✓ Branch 1 taken 487 times.
✓ Branch 2 taken 26615 times.
|
27102 | if ((err = avi_sync(s, 0)) < 0) |
1615 | 487 | return err; | |
1616 | 26615 | goto resync; | |
1617 | } | ||
1618 | |||
1619 | /* XXX: We make the implicit supposition that the positions are sorted | ||
1620 | * for each stream. */ | ||
1621 | 432 | static int avi_read_idx1(AVFormatContext *s, int size) | |
1622 | { | ||
1623 | 432 | AVIContext *avi = s->priv_data; | |
1624 | 432 | AVIOContext *pb = s->pb; | |
1625 | int nb_index_entries, i; | ||
1626 | AVStream *st; | ||
1627 | AVIStream *ast; | ||
1628 | int64_t pos; | ||
1629 | 432 | unsigned int index, tag, flags, len, first_packet = 1; | |
1630 | 432 | int64_t last_pos = -1; | |
1631 | 432 | unsigned last_idx = -1; | |
1632 | 432 | int64_t idx1_pos, first_packet_pos = 0, data_offset = 0; | |
1633 | 432 | int anykey = 0; | |
1634 | |||
1635 | 432 | nb_index_entries = size / 16; | |
1636 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 432 times.
|
432 | if (nb_index_entries <= 0) |
1637 | ✗ | return AVERROR_INVALIDDATA; | |
1638 | |||
1639 | 432 | idx1_pos = avio_tell(pb); | |
1640 | 432 | avio_seek(pb, avi->movi_list + 4, SEEK_SET); | |
1641 |
1/2✓ Branch 1 taken 432 times.
✗ Branch 2 not taken.
|
432 | if (avi_sync(s, 1) == 0) |
1642 | 432 | first_packet_pos = avio_tell(pb) - 8; | |
1643 | 432 | avi->stream_index = -1; | |
1644 | 432 | avio_seek(pb, idx1_pos, SEEK_SET); | |
1645 | |||
1646 |
3/4✓ Branch 0 taken 411 times.
✓ Branch 1 taken 21 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 411 times.
|
432 | if (s->nb_streams == 1 && s->streams[0]->codecpar->codec_tag == AV_RL32("MMES")) { |
1647 | ✗ | first_packet_pos = 0; | |
1648 | ✗ | data_offset = avi->movi_list; | |
1649 | } | ||
1650 | |||
1651 | /* Read the entries and sort them in each stream component. */ | ||
1652 |
2/2✓ Branch 0 taken 28045 times.
✓ Branch 1 taken 432 times.
|
28477 | for (i = 0; i < nb_index_entries; i++) { |
1653 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 28045 times.
|
28045 | if (avio_feof(pb)) |
1654 | ✗ | return -1; | |
1655 | |||
1656 | 28045 | tag = avio_rl32(pb); | |
1657 | 28045 | flags = avio_rl32(pb); | |
1658 | 28045 | pos = avio_rl32(pb); | |
1659 | 28045 | len = avio_rl32(pb); | |
1660 | 28045 | av_log(s, AV_LOG_TRACE, "%d: tag=0x%x flags=0x%x pos=0x%"PRIx64" len=%d/", | |
1661 | i, tag, flags, pos, len); | ||
1662 | |||
1663 | 28045 | index = ((tag & 0xff) - '0') * 10; | |
1664 | 28045 | index += (tag >> 8 & 0xff) - '0'; | |
1665 |
2/2✓ Branch 0 taken 290 times.
✓ Branch 1 taken 27755 times.
|
28045 | if (index >= s->nb_streams) |
1666 | 290 | continue; | |
1667 | 27755 | st = s->streams[index]; | |
1668 | 27755 | ast = st->priv_data; | |
1669 | |||
1670 | /* Skip 'xxpc' palette change entries in the index until a logic | ||
1671 | * to process these is properly implemented. */ | ||
1672 |
3/4✓ Branch 0 taken 2 times.
✓ Branch 1 taken 27753 times.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
|
27755 | if ((tag >> 16 & 0xff) == 'p' && (tag >> 24 & 0xff) == 'c') |
1673 | 2 | continue; | |
1674 | |||
1675 |
3/4✓ Branch 0 taken 432 times.
✓ Branch 1 taken 27321 times.
✓ Branch 2 taken 432 times.
✗ Branch 3 not taken.
|
27753 | if (first_packet && first_packet_pos) { |
1676 |
3/4✓ Branch 0 taken 1 times.
✓ Branch 1 taken 431 times.
✓ Branch 2 taken 1 times.
✗ Branch 3 not taken.
|
432 | if (avi->movi_list + 4 != pos || pos + 500 > first_packet_pos) |
1677 | 432 | data_offset = first_packet_pos - pos; | |
1678 | 432 | first_packet = 0; | |
1679 | } | ||
1680 | 27753 | pos += data_offset; | |
1681 | |||
1682 | 27753 | av_log(s, AV_LOG_TRACE, "%d cum_len=%"PRId64"\n", len, ast->cum_len); | |
1683 | |||
1684 | // even if we have only a single stream, we should | ||
1685 | // switch to non-interleaved to get correct timestamps | ||
1686 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 27753 times.
|
27753 | if (last_pos == pos) |
1687 | ✗ | avi->non_interleaved = 1; | |
1688 |
3/4✓ Branch 0 taken 27753 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 25524 times.
✓ Branch 3 taken 2229 times.
|
27753 | if (last_idx != pos && len) { |
1689 | 25524 | av_add_index_entry(st, pos, ast->cum_len, len, 0, | |
1690 | 25524 | (flags & AVIIF_INDEX) ? AVINDEX_KEYFRAME : 0); | |
1691 | 25524 | last_idx= pos; | |
1692 | } | ||
1693 | 27753 | ast->cum_len += get_duration(ast, len); | |
1694 | 27753 | last_pos = pos; | |
1695 | 27753 | anykey |= flags&AVIIF_INDEX; | |
1696 | } | ||
1697 |
2/2✓ Branch 0 taken 1 times.
✓ Branch 1 taken 431 times.
|
432 | if (!anykey) { |
1698 |
2/2✓ Branch 0 taken 1 times.
✓ Branch 1 taken 1 times.
|
2 | for (index = 0; index < s->nb_streams; index++) { |
1699 | 1 | FFStream *const sti = ffstream(s->streams[index]); | |
1700 |
1/2✓ Branch 0 taken 1 times.
✗ Branch 1 not taken.
|
1 | if (sti->nb_index_entries) |
1701 | 1 | sti->index_entries[0].flags |= AVINDEX_KEYFRAME; | |
1702 | } | ||
1703 | } | ||
1704 | 432 | return 0; | |
1705 | } | ||
1706 | |||
1707 | /* Scan the index and consider any file with streams more than | ||
1708 | * 2 seconds or 64MB apart non-interleaved. */ | ||
1709 | 471 | static int check_stream_max_drift(AVFormatContext *s) | |
1710 | { | ||
1711 | int64_t min_pos, pos; | ||
1712 | int i; | ||
1713 | 471 | int *idx = av_calloc(s->nb_streams, sizeof(*idx)); | |
1714 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 471 times.
|
471 | if (!idx) |
1715 | ✗ | return AVERROR(ENOMEM); | |
1716 |
2/2✓ Branch 0 taken 31065 times.
✓ Branch 1 taken 471 times.
|
31536 | for (min_pos = pos = 0; min_pos != INT64_MAX; pos = min_pos + 1ULL) { |
1717 | 31065 | int64_t max_dts = INT64_MIN / 2; | |
1718 | 31065 | int64_t min_dts = INT64_MAX / 2; | |
1719 | 31065 | int64_t max_buffer = 0; | |
1720 | |||
1721 | 31065 | min_pos = INT64_MAX; | |
1722 | |||
1723 |
2/2✓ Branch 0 taken 39543 times.
✓ Branch 1 taken 31065 times.
|
70608 | for (i = 0; i < s->nb_streams; i++) { |
1724 | 39543 | AVStream *st = s->streams[i]; | |
1725 | 39543 | AVIStream *ast = st->priv_data; | |
1726 | 39543 | FFStream *const sti = ffstream(st); | |
1727 | 39543 | int n = sti->nb_index_entries; | |
1728 |
4/4✓ Branch 0 taken 69383 times.
✓ Branch 1 taken 754 times.
✓ Branch 2 taken 30594 times.
✓ Branch 3 taken 38789 times.
|
70137 | while (idx[i] < n && sti->index_entries[idx[i]].pos < pos) |
1729 | 30594 | idx[i]++; | |
1730 |
2/2✓ Branch 0 taken 38789 times.
✓ Branch 1 taken 754 times.
|
39543 | if (idx[i] < n) { |
1731 | int64_t dts; | ||
1732 | 38789 | dts = av_rescale_q(sti->index_entries[idx[i]].timestamp / | |
1733 | 38789 | FFMAX(ast->sample_size, 1), | |
1734 | 38789 | st->time_base, AV_TIME_BASE_Q); | |
1735 | 38789 | min_dts = FFMIN(min_dts, dts); | |
1736 | 38789 | min_pos = FFMIN(min_pos, sti->index_entries[idx[i]].pos); | |
1737 | } | ||
1738 | } | ||
1739 |
2/2✓ Branch 0 taken 39543 times.
✓ Branch 1 taken 31065 times.
|
70608 | for (i = 0; i < s->nb_streams; i++) { |
1740 | 39543 | AVStream *st = s->streams[i]; | |
1741 | 39543 | FFStream *const sti = ffstream(st); | |
1742 | 39543 | AVIStream *ast = st->priv_data; | |
1743 | |||
1744 |
4/4✓ Branch 0 taken 38832 times.
✓ Branch 1 taken 711 times.
✓ Branch 2 taken 38382 times.
✓ Branch 3 taken 450 times.
|
39543 | if (idx[i] && min_dts != INT64_MAX / 2) { |
1745 | int64_t dts, delta_dts; | ||
1746 | 38382 | dts = av_rescale_q(sti->index_entries[idx[i] - 1].timestamp / | |
1747 | 38382 | FFMAX(ast->sample_size, 1), | |
1748 | 38382 | st->time_base, AV_TIME_BASE_Q); | |
1749 | 38382 | delta_dts = av_sat_sub64(dts, min_dts); | |
1750 | 38382 | max_dts = FFMAX(max_dts, dts); | |
1751 | 38382 | max_buffer = FFMAX(max_buffer, | |
1752 | av_rescale(delta_dts, | ||
1753 | st->codecpar->bit_rate, | ||
1754 | AV_TIME_BASE)); | ||
1755 | } | ||
1756 | } | ||
1757 |
2/4✓ Branch 1 taken 31065 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✓ Branch 4 taken 31065 times.
|
31065 | if (av_sat_sub64(max_dts, min_dts) > 2 * AV_TIME_BASE || |
1758 | max_buffer > 1024 * 1024 * 8 * 8) { | ||
1759 | ✗ | av_free(idx); | |
1760 | ✗ | return 1; | |
1761 | } | ||
1762 | } | ||
1763 | 471 | av_free(idx); | |
1764 | 471 | return 0; | |
1765 | } | ||
1766 | |||
1767 | 476 | static int guess_ni_flag(AVFormatContext *s) | |
1768 | { | ||
1769 | int i; | ||
1770 | 476 | int64_t last_start = 0; | |
1771 | 476 | int64_t first_end = INT64_MAX; | |
1772 | 476 | int64_t oldpos = avio_tell(s->pb); | |
1773 | |||
1774 |
2/2✓ Branch 0 taken 520 times.
✓ Branch 1 taken 476 times.
|
996 | for (i = 0; i < s->nb_streams; i++) { |
1775 | 520 | AVStream *st = s->streams[i]; | |
1776 | 520 | FFStream *const sti = ffstream(st); | |
1777 | 520 | int n = sti->nb_index_entries; | |
1778 | unsigned int size; | ||
1779 | |||
1780 |
2/2✓ Branch 0 taken 60 times.
✓ Branch 1 taken 460 times.
|
520 | if (n <= 0) |
1781 | 60 | continue; | |
1782 | |||
1783 |
2/2✓ Branch 0 taken 419 times.
✓ Branch 1 taken 41 times.
|
460 | if (n >= 2) { |
1784 | 419 | int64_t pos = sti->index_entries[0].pos; | |
1785 | unsigned tag[2]; | ||
1786 | 419 | avio_seek(s->pb, pos, SEEK_SET); | |
1787 | 419 | tag[0] = avio_r8(s->pb); | |
1788 | 419 | tag[1] = avio_r8(s->pb); | |
1789 | 419 | avio_rl16(s->pb); | |
1790 | 419 | size = avio_rl32(s->pb); | |
1791 |
3/4✓ Branch 1 taken 418 times.
✓ Branch 2 taken 1 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 418 times.
|
419 | if (get_stream_idx(tag) == i && pos + size > sti->index_entries[1].pos) |
1792 | ✗ | last_start = INT64_MAX; | |
1793 |
3/4✓ Branch 1 taken 418 times.
✓ Branch 2 taken 1 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 418 times.
|
419 | if (get_stream_idx(tag) == i && size == sti->index_entries[0].size + 8) |
1794 | ✗ | last_start = INT64_MAX; | |
1795 | } | ||
1796 | |||
1797 |
2/2✓ Branch 0 taken 450 times.
✓ Branch 1 taken 10 times.
|
460 | if (sti->index_entries[0].pos > last_start) |
1798 | 450 | last_start = sti->index_entries[0].pos; | |
1799 |
2/2✓ Branch 0 taken 447 times.
✓ Branch 1 taken 13 times.
|
460 | if (sti->index_entries[n - 1].pos < first_end) |
1800 | 447 | first_end = sti->index_entries[n - 1].pos; | |
1801 | } | ||
1802 | 476 | avio_seek(s->pb, oldpos, SEEK_SET); | |
1803 | |||
1804 |
2/2✓ Branch 0 taken 5 times.
✓ Branch 1 taken 471 times.
|
476 | if (last_start > first_end) |
1805 | 5 | return 1; | |
1806 | |||
1807 | 471 | return check_stream_max_drift(s); | |
1808 | } | ||
1809 | |||
1810 | 470 | static int avi_load_index(AVFormatContext *s) | |
1811 | { | ||
1812 | 470 | AVIContext *avi = s->priv_data; | |
1813 | 470 | AVIOContext *pb = s->pb; | |
1814 | uint32_t tag, size; | ||
1815 | 470 | int64_t pos = avio_tell(pb); | |
1816 | int64_t next; | ||
1817 | 470 | int ret = -1; | |
1818 | |||
1819 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 470 times.
|
470 | if (avio_seek(pb, avi->movi_end, SEEK_SET) < 0) |
1820 | ✗ | goto the_end; // maybe truncated file | |
1821 | 470 | av_log(s, AV_LOG_TRACE, "movi_end=0x%"PRIx64"\n", avi->movi_end); | |
1822 | for (;;) { | ||
1823 | 911 | tag = avio_rl32(pb); | |
1824 | 911 | size = avio_rl32(pb); | |
1825 |
2/2✓ Branch 1 taken 448 times.
✓ Branch 2 taken 463 times.
|
911 | if (avio_feof(pb)) |
1826 | 448 | break; | |
1827 | 463 | next = avio_tell(pb); | |
1828 |
2/4✓ Branch 0 taken 463 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 463 times.
|
463 | if (next < 0 || next > INT64_MAX - size - (size & 1)) |
1829 | break; | ||
1830 | 463 | next += size + (size & 1LL); | |
1831 | |||
1832 |
3/4✓ Branch 0 taken 432 times.
✓ Branch 1 taken 31 times.
✓ Branch 2 taken 432 times.
✗ Branch 3 not taken.
|
895 | if (tag == MKTAG('i', 'd', 'x', '1') && |
1833 | 432 | avi_read_idx1(s, size) >= 0) { | |
1834 | 432 | avi->index_loaded=2; | |
1835 | 432 | ret = 0; | |
1836 |
2/2✓ Branch 0 taken 9 times.
✓ Branch 1 taken 22 times.
|
31 | }else if (tag == MKTAG('L', 'I', 'S', 'T')) { |
1837 | 9 | uint32_t tag1 = avio_rl32(pb); | |
1838 | |||
1839 |
2/2✓ Branch 0 taken 5 times.
✓ Branch 1 taken 4 times.
|
9 | if (tag1 == MKTAG('I', 'N', 'F', 'O')) |
1840 | 5 | ff_read_riff_info(s, size - 4); | |
1841 |
1/2✓ Branch 0 taken 22 times.
✗ Branch 1 not taken.
|
22 | }else if (!ret) |
1842 | 22 | break; | |
1843 | |||
1844 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 441 times.
|
441 | if (avio_seek(pb, next, SEEK_SET) < 0) |
1845 | ✗ | break; // something is wrong here | |
1846 | } | ||
1847 | |||
1848 | ✗ | the_end: | |
1849 | 470 | avio_seek(pb, pos, SEEK_SET); | |
1850 | 470 | return ret; | |
1851 | } | ||
1852 | |||
1853 | ✗ | static void seek_subtitle(AVStream *st, AVStream *st2, int64_t timestamp) | |
1854 | { | ||
1855 | ✗ | AVIStream *ast2 = st2->priv_data; | |
1856 | ✗ | int64_t ts2 = av_rescale_q(timestamp, st->time_base, st2->time_base); | |
1857 | ✗ | av_packet_unref(ast2->sub_pkt); | |
1858 | ✗ | if (avformat_seek_file(ast2->sub_ctx, 0, INT64_MIN, ts2, ts2, 0) >= 0 || | |
1859 | ✗ | avformat_seek_file(ast2->sub_ctx, 0, ts2, ts2, INT64_MAX, 0) >= 0) | |
1860 | ✗ | ff_read_packet(ast2->sub_ctx, ast2->sub_pkt); | |
1861 | ✗ | } | |
1862 | |||
1863 | 728 | static int avi_read_seek(AVFormatContext *s, int stream_index, | |
1864 | int64_t timestamp, int flags) | ||
1865 | { | ||
1866 | 728 | AVIContext *avi = s->priv_data; | |
1867 | AVStream *st; | ||
1868 | FFStream *sti; | ||
1869 | int i, index; | ||
1870 | int64_t pos, pos_min; | ||
1871 | AVIStream *ast; | ||
1872 | |||
1873 | /* Does not matter which stream is requested dv in avi has the | ||
1874 | * stream information in the first video stream. | ||
1875 | */ | ||
1876 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 728 times.
|
728 | if (avi->dv_demux) |
1877 | ✗ | stream_index = 0; | |
1878 | |||
1879 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 728 times.
|
728 | if (!avi->index_loaded) { |
1880 | /* we only load the index on demand */ | ||
1881 | ✗ | avi_load_index(s); | |
1882 | ✗ | avi->index_loaded |= 1; | |
1883 | } | ||
1884 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 728 times.
|
728 | av_assert0(stream_index >= 0); |
1885 | |||
1886 | 728 | st = s->streams[stream_index]; | |
1887 | 728 | sti = ffstream(st); | |
1888 | 728 | ast = st->priv_data; | |
1889 | |||
1890 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 728 times.
|
728 | if (avi->dv_demux) { |
1891 | // index entries are in the AVI scale/rate timebase, which does | ||
1892 | // not match DV demuxer's stream timebase | ||
1893 | ✗ | timestamp = av_rescale_q(timestamp, st->time_base, | |
1894 | ✗ | (AVRational){ ast->scale, ast->rate }); | |
1895 | } else | ||
1896 | 728 | timestamp *= FFMAX(ast->sample_size, 1); | |
1897 | |||
1898 | 728 | index = av_index_search_timestamp(st, timestamp, flags); | |
1899 |
2/2✓ Branch 0 taken 198 times.
✓ Branch 1 taken 530 times.
|
728 | if (index < 0) { |
1900 |
1/2✓ Branch 0 taken 198 times.
✗ Branch 1 not taken.
|
198 | if (sti->nb_index_entries > 0) |
1901 | 198 | av_log(s, AV_LOG_DEBUG, "Failed to find timestamp %"PRId64 " in index %"PRId64 " .. %"PRId64 "\n", | |
1902 | timestamp, | ||
1903 | 198 | sti->index_entries[0].timestamp, | |
1904 | 198 | sti->index_entries[sti->nb_index_entries - 1].timestamp); | |
1905 | 198 | return AVERROR_INVALIDDATA; | |
1906 | } | ||
1907 | |||
1908 | /* find the position */ | ||
1909 | 530 | pos = sti->index_entries[index].pos; | |
1910 | 530 | timestamp = sti->index_entries[index].timestamp; | |
1911 | |||
1912 | 530 | av_log(s, AV_LOG_TRACE, "XX %"PRId64" %d %"PRId64"\n", | |
1913 | 530 | timestamp, index, sti->index_entries[index].timestamp); | |
1914 | |||
1915 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 530 times.
|
530 | if (CONFIG_DV_DEMUXER && avi->dv_demux) { |
1916 | /* One and only one real stream for DV in AVI, and it has video */ | ||
1917 | /* offsets. Calling with other stream indexes should have failed */ | ||
1918 | /* the av_index_search_timestamp call above. */ | ||
1919 | |||
1920 | ✗ | if (avio_seek(s->pb, pos, SEEK_SET) < 0) | |
1921 | ✗ | return -1; | |
1922 | |||
1923 | /* Feed the DV video stream version of the timestamp to the */ | ||
1924 | /* DV demux so it can synthesize correct timestamps. */ | ||
1925 | ✗ | ff_dv_ts_reset(avi->dv_demux, | |
1926 | ✗ | av_rescale_q(timestamp, (AVRational){ ast->scale, ast->rate }, | |
1927 | st->time_base)); | ||
1928 | |||
1929 | ✗ | avi->stream_index = -1; | |
1930 | ✗ | return 0; | |
1931 | } | ||
1932 | 530 | timestamp /= FFMAX(ast->sample_size, 1); | |
1933 | |||
1934 | 530 | pos_min = pos; | |
1935 |
2/2✓ Branch 0 taken 547 times.
✓ Branch 1 taken 530 times.
|
1077 | for (i = 0; i < s->nb_streams; i++) { |
1936 | 547 | AVStream *st2 = s->streams[i]; | |
1937 | 547 | FFStream *const sti2 = ffstream(st2); | |
1938 | 547 | AVIStream *ast2 = st2->priv_data; | |
1939 | |||
1940 | 547 | ast2->packet_size = | |
1941 | 547 | ast2->remaining = 0; | |
1942 | |||
1943 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 547 times.
|
547 | if (ast2->sub_ctx) { |
1944 | ✗ | seek_subtitle(st, st2, timestamp); | |
1945 | ✗ | continue; | |
1946 | } | ||
1947 | |||
1948 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 547 times.
|
547 | if (sti2->nb_index_entries <= 0) |
1949 | ✗ | continue; | |
1950 | |||
1951 | // av_assert1(st2->codecpar->block_align); | ||
1952 | 547 | index = av_index_search_timestamp(st2, | |
1953 | 547 | av_rescale_q(timestamp, | |
1954 | st->time_base, | ||
1955 | st2->time_base) * | ||
1956 | 547 | FFMAX(ast2->sample_size, 1), | |
1957 | 547 | flags | | |
1958 | AVSEEK_FLAG_BACKWARD | | ||
1959 |
2/2✓ Branch 0 taken 17 times.
✓ Branch 1 taken 530 times.
|
547 | (st2->codecpar->codec_type != AVMEDIA_TYPE_VIDEO ? AVSEEK_FLAG_ANY : 0)); |
1960 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 547 times.
|
547 | if (index < 0) |
1961 | ✗ | index = 0; | |
1962 | 547 | ast2->seek_pos = sti2->index_entries[index].pos; | |
1963 | 547 | pos_min = FFMIN(pos_min,ast2->seek_pos); | |
1964 | } | ||
1965 |
2/2✓ Branch 0 taken 547 times.
✓ Branch 1 taken 530 times.
|
1077 | for (i = 0; i < s->nb_streams; i++) { |
1966 | 547 | AVStream *st2 = s->streams[i]; | |
1967 | 547 | FFStream *const sti2 = ffstream(st2); | |
1968 | 547 | AVIStream *ast2 = st2->priv_data; | |
1969 | |||
1970 |
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) |
1971 | ✗ | continue; | |
1972 | |||
1973 | 547 | index = av_index_search_timestamp( | |
1974 | st2, | ||
1975 | 547 | av_rescale_q(timestamp, st->time_base, st2->time_base) * FFMAX(ast2->sample_size, 1), | |
1976 |
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)); |
1977 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 547 times.
|
547 | if (index < 0) |
1978 | ✗ | index = 0; | |
1979 |
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) |
1980 | 11 | index--; | |
1981 | 547 | ast2->frame_offset = sti2->index_entries[index].timestamp; | |
1982 | } | ||
1983 | |||
1984 | /* do the seek */ | ||
1985 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 530 times.
|
530 | if (avio_seek(s->pb, pos_min, SEEK_SET) < 0) { |
1986 | ✗ | av_log(s, AV_LOG_ERROR, "Seek failed\n"); | |
1987 | ✗ | return -1; | |
1988 | } | ||
1989 | 530 | avi->stream_index = -1; | |
1990 | 530 | avi->dts_max = INT_MIN; | |
1991 | 530 | return 0; | |
1992 | } | ||
1993 | |||
1994 | 476 | static int avi_read_close(AVFormatContext *s) | |
1995 | { | ||
1996 | int i; | ||
1997 | 476 | AVIContext *avi = s->priv_data; | |
1998 | |||
1999 |
2/2✓ Branch 0 taken 520 times.
✓ Branch 1 taken 476 times.
|
996 | for (i = 0; i < s->nb_streams; i++) { |
2000 | 520 | AVStream *st = s->streams[i]; | |
2001 | 520 | AVIStream *ast = st->priv_data; | |
2002 |
1/2✓ Branch 0 taken 520 times.
✗ Branch 1 not taken.
|
520 | if (ast) { |
2003 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 520 times.
|
520 | if (ast->sub_ctx) { |
2004 | ✗ | av_freep(&ast->sub_ctx->pb); | |
2005 | ✗ | avformat_close_input(&ast->sub_ctx); | |
2006 | } | ||
2007 | 520 | av_buffer_unref(&ast->sub_buffer); | |
2008 | 520 | av_packet_free(&ast->sub_pkt); | |
2009 | } | ||
2010 | } | ||
2011 | |||
2012 | 476 | av_freep(&avi->dv_demux); | |
2013 | |||
2014 | 476 | return 0; | |
2015 | } | ||
2016 | |||
2017 | 7260 | static int avi_probe(const AVProbeData *p) | |
2018 | { | ||
2019 | int i; | ||
2020 | |||
2021 | /* check file header */ | ||
2022 |
2/2✓ Branch 0 taken 34416 times.
✓ Branch 1 taken 6787 times.
|
41203 | for (i = 0; avi_headers[i][0]; i++) |
2023 |
2/2✓ Branch 0 taken 2483 times.
✓ Branch 1 taken 31933 times.
|
34416 | if (AV_RL32(p->buf ) == AV_RL32(avi_headers[i] ) && |
2024 |
2/2✓ Branch 0 taken 473 times.
✓ Branch 1 taken 2010 times.
|
2483 | AV_RL32(p->buf + 8) == AV_RL32(avi_headers[i] + 4)) |
2025 | 473 | return AVPROBE_SCORE_MAX; | |
2026 | |||
2027 | 6787 | return 0; | |
2028 | } | ||
2029 | |||
2030 | const FFInputFormat ff_avi_demuxer = { | ||
2031 | .p.name = "avi", | ||
2032 | .p.long_name = NULL_IF_CONFIG_SMALL("AVI (Audio Video Interleaved)"), | ||
2033 | .p.extensions = "avi", | ||
2034 | .p.priv_class = &demuxer_class, | ||
2035 | .priv_data_size = sizeof(AVIContext), | ||
2036 | .flags_internal = FF_INFMT_FLAG_INIT_CLEANUP, | ||
2037 | .read_probe = avi_probe, | ||
2038 | .read_header = avi_read_header, | ||
2039 | .read_packet = avi_read_packet, | ||
2040 | .read_close = avi_read_close, | ||
2041 | .read_seek = avi_read_seek, | ||
2042 | }; | ||
2043 |