Most files use _ instead of - as spacer, so this is more consistent
Signed-off-by: Michael Niedermayer <michaelni@gmx.at>
... | ... |
@@ -84,7 +84,7 @@ OBJS-$(CONFIG_APNG_MUXER) += apngenc.o |
84 | 84 |
OBJS-$(CONFIG_AQTITLE_DEMUXER) += aqtitledec.o subtitles.o |
85 | 85 |
OBJS-$(CONFIG_ASF_DEMUXER) += asfdec_f.o asf.o asfcrypt.o \ |
86 | 86 |
avlanguage.o |
87 |
-OBJS-$(CONFIG_ASF_O_DEMUXER) += asfdec-o.o asf.o asfcrypt.o \ |
|
87 |
+OBJS-$(CONFIG_ASF_O_DEMUXER) += asfdec_o.o asf.o asfcrypt.o \ |
|
88 | 88 |
avlanguage.o |
89 | 89 |
OBJS-$(CONFIG_ASF_MUXER) += asfenc.o asf.o |
90 | 90 |
OBJS-$(CONFIG_ASS_DEMUXER) += assdec.o subtitles.o |
91 | 91 |
deleted file mode 100644 |
... | ... |
@@ -1,1702 +0,0 @@ |
1 |
-/* |
|
2 |
- * Microsoft Advanced Streaming Format demuxer |
|
3 |
- * Copyright (c) 2014 Alexandra Hájková |
|
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 "libavutil/attributes.h" |
|
23 |
-#include "libavutil/avassert.h" |
|
24 |
-#include "libavutil/avstring.h" |
|
25 |
-#include "libavutil/bswap.h" |
|
26 |
-#include "libavutil/common.h" |
|
27 |
-#include "libavutil/dict.h" |
|
28 |
-#include "libavutil/internal.h" |
|
29 |
-#include "libavutil/mathematics.h" |
|
30 |
-#include "libavutil/opt.h" |
|
31 |
-#include "libavutil/time_internal.h" |
|
32 |
- |
|
33 |
-#include "avformat.h" |
|
34 |
-#include "avio_internal.h" |
|
35 |
-#include "avlanguage.h" |
|
36 |
-#include "id3v2.h" |
|
37 |
-#include "internal.h" |
|
38 |
-#include "riff.h" |
|
39 |
-#include "asf.h" |
|
40 |
-#include "asfcrypt.h" |
|
41 |
- |
|
42 |
-#define ASF_BOOL 0x2 |
|
43 |
-#define ASF_WORD 0x5 |
|
44 |
-#define ASF_GUID 0x6 |
|
45 |
-#define ASF_DWORD 0x3 |
|
46 |
-#define ASF_QWORD 0x4 |
|
47 |
-#define ASF_UNICODE 0x0 |
|
48 |
-#define ASF_FLAG_BROADCAST 0x1 |
|
49 |
-#define ASF_BYTE_ARRAY 0x1 |
|
50 |
-#define ASF_TYPE_AUDIO 0x2 |
|
51 |
-#define ASF_TYPE_VIDEO 0x1 |
|
52 |
-#define ASF_STREAM_NUM 0x7F |
|
53 |
-#define ASF_MAX_STREAMS 128 |
|
54 |
-#define BMP_HEADER_SIZE 40 |
|
55 |
-#define ASF_NUM_OF_PAYLOADS 0x3F |
|
56 |
-#define ASF_ERROR_CORRECTION_LENGTH_TYPE 0x60 |
|
57 |
-#define ASF_PACKET_ERROR_CORRECTION_DATA_SIZE 0x2 |
|
58 |
- |
|
59 |
-typedef struct GUIDParseTable { |
|
60 |
- const char *name; |
|
61 |
- ff_asf_guid guid; |
|
62 |
- int (*read_object)(AVFormatContext *, const struct GUIDParseTable *); |
|
63 |
- int is_subobject; |
|
64 |
-} GUIDParseTable; |
|
65 |
- |
|
66 |
-typedef struct ASFPacket { |
|
67 |
- AVPacket avpkt; |
|
68 |
- int64_t dts; |
|
69 |
- uint32_t frame_num; // ASF payloads with the same number are parts of the same frame |
|
70 |
- int flags; |
|
71 |
- int data_size; |
|
72 |
- int duration; |
|
73 |
- int size_left; |
|
74 |
- uint8_t stream_index; |
|
75 |
-} ASFPacket; |
|
76 |
- |
|
77 |
-typedef struct ASFStream { |
|
78 |
- uint8_t stream_index; // from packet header |
|
79 |
- int index; // stream index in AVFormatContext, set in asf_read_stream_properties |
|
80 |
- int type; |
|
81 |
- int indexed; // added index entries from the Simple Index Object or not |
|
82 |
- int8_t span; // for deinterleaving |
|
83 |
- uint16_t virtual_pkt_len; |
|
84 |
- uint16_t virtual_chunk_len; |
|
85 |
- int16_t lang_idx; |
|
86 |
- ASFPacket pkt; |
|
87 |
-} ASFStream; |
|
88 |
- |
|
89 |
-typedef struct ASFStreamData{ |
|
90 |
- char langs[32]; |
|
91 |
- AVDictionary *asf_met; // for storing per-stream metadata |
|
92 |
- AVRational aspect_ratio; |
|
93 |
-} ASFStreamData; |
|
94 |
- |
|
95 |
-typedef struct ASFContext { |
|
96 |
- int data_reached; |
|
97 |
- int is_simple_index; // is simple index present or not 1/0 |
|
98 |
- int is_header; |
|
99 |
- |
|
100 |
- uint64_t preroll; |
|
101 |
- uint64_t nb_packets; // ASF packets |
|
102 |
- uint32_t packet_size; |
|
103 |
- int64_t send_time; |
|
104 |
- int duration; |
|
105 |
- |
|
106 |
- uint32_t b_flags; // flags with broadcast flag |
|
107 |
- uint32_t prop_flags; // file properties object flags |
|
108 |
- |
|
109 |
- uint64_t data_size; // data object size |
|
110 |
- uint64_t unknown_size; // size of the unknown object |
|
111 |
- |
|
112 |
- int64_t offset; // offset of the current object |
|
113 |
- |
|
114 |
- int64_t data_offset; |
|
115 |
- int64_t first_packet_offset; // packet offset |
|
116 |
- int64_t unknown_offset; // for top level header objects or subobjects without specified behavior |
|
117 |
- |
|
118 |
- // ASF file must not contain more than 128 streams according to the specification |
|
119 |
- ASFStream *asf_st[ASF_MAX_STREAMS]; |
|
120 |
- ASFStreamData asf_sd[ASF_MAX_STREAMS]; |
|
121 |
- int nb_streams; |
|
122 |
- |
|
123 |
- int stream_index; // from packet header, for the subpayload case |
|
124 |
- |
|
125 |
- // packet parameteres |
|
126 |
- uint64_t sub_header_offset; // offset of subplayload header |
|
127 |
- int64_t sub_dts; |
|
128 |
- uint8_t dts_delta; // for subpayloads |
|
129 |
- uint32_t packet_size_internal; // packet size stored inside ASFPacket, can be 0 |
|
130 |
- int64_t dts; |
|
131 |
- int64_t packet_offset; // offset of the current packet inside Data Object |
|
132 |
- uint32_t pad_len; // padding after payload |
|
133 |
- uint32_t rep_data_len; |
|
134 |
- |
|
135 |
- // packet state |
|
136 |
- uint64_t sub_left; // subpayloads left or not |
|
137 |
- int nb_sub; // number of subpayloads read so far from the current ASF packet |
|
138 |
- uint16_t mult_sub_len; // total length of subpayloads array inside multiple payload |
|
139 |
- uint64_t nb_mult_left; // multiple payloads left |
|
140 |
- int return_subpayload; |
|
141 |
- enum { |
|
142 |
- PARSE_PACKET_HEADER, |
|
143 |
- READ_SINGLE, |
|
144 |
- READ_MULTI, |
|
145 |
- READ_MULTI_SUB |
|
146 |
- } state; |
|
147 |
-} ASFContext; |
|
148 |
- |
|
149 |
-static int detect_unknown_subobject(AVFormatContext *s, int64_t offset, int64_t size); |
|
150 |
-static const GUIDParseTable *find_guid(ff_asf_guid guid); |
|
151 |
- |
|
152 |
-static int asf_probe(AVProbeData *pd) |
|
153 |
-{ |
|
154 |
- /* check file header */ |
|
155 |
- if (!ff_guidcmp(pd->buf, &ff_asf_header)) |
|
156 |
- return AVPROBE_SCORE_MAX/2; |
|
157 |
- else |
|
158 |
- return 0; |
|
159 |
-} |
|
160 |
- |
|
161 |
-static void swap_guid(ff_asf_guid guid) |
|
162 |
-{ |
|
163 |
- FFSWAP(unsigned char, guid[0], guid[3]); |
|
164 |
- FFSWAP(unsigned char, guid[1], guid[2]); |
|
165 |
- FFSWAP(unsigned char, guid[4], guid[5]); |
|
166 |
- FFSWAP(unsigned char, guid[6], guid[7]); |
|
167 |
-} |
|
168 |
- |
|
169 |
-static void align_position(AVIOContext *pb, int64_t offset, uint64_t size) |
|
170 |
-{ |
|
171 |
- if (avio_tell(pb) != offset + size) |
|
172 |
- avio_seek(pb, offset + size, SEEK_SET); |
|
173 |
-} |
|
174 |
- |
|
175 |
-static int asf_read_unknown(AVFormatContext *s, const GUIDParseTable *g) |
|
176 |
-{ |
|
177 |
- ASFContext *asf = s->priv_data; |
|
178 |
- AVIOContext *pb = s->pb; |
|
179 |
- uint64_t size = avio_rl64(pb); |
|
180 |
- int ret; |
|
181 |
- |
|
182 |
- if (asf->is_header) |
|
183 |
- asf->unknown_size = size; |
|
184 |
- asf->is_header = 0; |
|
185 |
- if (!g->is_subobject) { |
|
186 |
- if (!(ret = strcmp(g->name, "Header Extension"))) |
|
187 |
- avio_skip(pb, 22); // skip reserved fields and Data Size |
|
188 |
- if ((ret = detect_unknown_subobject(s, asf->unknown_offset, |
|
189 |
- asf->unknown_size)) < 0) |
|
190 |
- return ret; |
|
191 |
- } else |
|
192 |
- avio_skip(pb, size - 24); |
|
193 |
- |
|
194 |
- return 0; |
|
195 |
-} |
|
196 |
- |
|
197 |
-static int get_asf_string(AVIOContext *pb, int maxlen, char *buf, int buflen) |
|
198 |
-{ |
|
199 |
- char *q = buf; |
|
200 |
- int ret = 0; |
|
201 |
- if (buflen <= 0) |
|
202 |
- return AVERROR(EINVAL); |
|
203 |
- while (ret + 1 < maxlen) { |
|
204 |
- uint8_t tmp; |
|
205 |
- uint32_t ch; |
|
206 |
- GET_UTF16(ch, (ret += 2) <= maxlen ? avio_rl16(pb) : 0, break;); |
|
207 |
- PUT_UTF8(ch, tmp, if (q - buf < buflen - 1) *q++ = tmp;) |
|
208 |
- } |
|
209 |
- *q = 0; |
|
210 |
- |
|
211 |
- return ret; |
|
212 |
-} |
|
213 |
- |
|
214 |
-static int asf_read_marker(AVFormatContext *s, const GUIDParseTable *g) |
|
215 |
-{ |
|
216 |
- ASFContext *asf = s->priv_data; |
|
217 |
- AVIOContext *pb = s->pb; |
|
218 |
- uint64_t size = avio_rl64(pb); |
|
219 |
- int i, nb_markers, ret; |
|
220 |
- size_t len; |
|
221 |
- char name[1024]; |
|
222 |
- |
|
223 |
- avio_skip(pb, 8); |
|
224 |
- avio_skip(pb, 8); // skip reserved GUID |
|
225 |
- nb_markers = avio_rl32(pb); |
|
226 |
- avio_skip(pb, 2); // skip reserved field |
|
227 |
- len = avio_rl16(pb); |
|
228 |
- for (i = 0; i < len; i++) |
|
229 |
- avio_skip(pb, 1); |
|
230 |
- |
|
231 |
- for (i = 0; i < nb_markers; i++) { |
|
232 |
- int64_t pts; |
|
233 |
- |
|
234 |
- avio_skip(pb, 8); |
|
235 |
- pts = avio_rl64(pb); |
|
236 |
- pts -= asf->preroll * 10000; |
|
237 |
- avio_skip(pb, 2); // entry length |
|
238 |
- avio_skip(pb, 4); // send time |
|
239 |
- avio_skip(pb, 4); // flags |
|
240 |
- len = avio_rl32(pb); |
|
241 |
- |
|
242 |
- if ((ret = avio_get_str16le(pb, len, name, |
|
243 |
- sizeof(name))) < len) |
|
244 |
- avio_skip(pb, len - ret); |
|
245 |
- avpriv_new_chapter(s, i, (AVRational) { 1, 10000000 }, pts, |
|
246 |
- AV_NOPTS_VALUE, name); |
|
247 |
- } |
|
248 |
- align_position(pb, asf->offset, size); |
|
249 |
- |
|
250 |
- return 0; |
|
251 |
-} |
|
252 |
- |
|
253 |
-static int asf_read_metadata(AVFormatContext *s, const char *title, uint16_t len, |
|
254 |
- unsigned char *ch, uint16_t buflen) |
|
255 |
-{ |
|
256 |
- AVIOContext *pb = s->pb; |
|
257 |
- |
|
258 |
- avio_get_str16le(pb, len, ch, buflen); |
|
259 |
- if (av_dict_set(&s->metadata, title, ch, 0) < 0) |
|
260 |
- av_log(s, AV_LOG_WARNING, "av_dict_set failed.\n"); |
|
261 |
- |
|
262 |
- return 0; |
|
263 |
-} |
|
264 |
- |
|
265 |
-static int asf_read_value(AVFormatContext *s, uint8_t *name, uint16_t name_len, |
|
266 |
- uint16_t val_len, int type, AVDictionary **met) |
|
267 |
-{ |
|
268 |
- int ret; |
|
269 |
- uint8_t *value; |
|
270 |
- uint16_t buflen = 2 * val_len + 1; |
|
271 |
- AVIOContext *pb = s->pb; |
|
272 |
- |
|
273 |
- value = av_malloc(buflen); |
|
274 |
- if (!value) |
|
275 |
- return AVERROR(ENOMEM); |
|
276 |
- if (type == ASF_UNICODE) { |
|
277 |
- // get_asf_string reads UTF-16 and converts it to UTF-8 which needs longer buffer |
|
278 |
- if ((ret = get_asf_string(pb, val_len, value, buflen)) < 0) |
|
279 |
- goto failed; |
|
280 |
- if (av_dict_set(met, name, value, 0) < 0) |
|
281 |
- av_log(s, AV_LOG_WARNING, "av_dict_set failed.\n"); |
|
282 |
- } else { |
|
283 |
- char buf[256]; |
|
284 |
- if (val_len > sizeof(buf)) |
|
285 |
- return AVERROR_INVALIDDATA; |
|
286 |
- if ((ret = avio_read(pb, value, val_len)) < 0) |
|
287 |
- goto failed; |
|
288 |
- if (ret < 2 * val_len) |
|
289 |
- value[ret] = '\0'; |
|
290 |
- else |
|
291 |
- value[2 * val_len - 1] = '\0'; |
|
292 |
- snprintf(buf, sizeof(buf), "%s", value); |
|
293 |
- if (av_dict_set(met, name, buf, 0) < 0) |
|
294 |
- av_log(s, AV_LOG_WARNING, "av_dict_set failed.\n"); |
|
295 |
- } |
|
296 |
- av_freep(&value); |
|
297 |
- |
|
298 |
- return 0; |
|
299 |
- |
|
300 |
-failed: |
|
301 |
- av_freep(&value); |
|
302 |
- return ret; |
|
303 |
-} |
|
304 |
- |
|
305 |
-static int asf_read_generic_value(AVFormatContext *s, uint8_t *name, uint16_t name_len, |
|
306 |
- int type, AVDictionary **met) |
|
307 |
-{ |
|
308 |
- AVIOContext *pb = s->pb; |
|
309 |
- uint64_t value; |
|
310 |
- char buf[32]; |
|
311 |
- |
|
312 |
- switch (type) { |
|
313 |
- case ASF_BOOL: |
|
314 |
- value = avio_rl32(pb); |
|
315 |
- break; |
|
316 |
- case ASF_DWORD: |
|
317 |
- value = avio_rl32(pb); |
|
318 |
- break; |
|
319 |
- case ASF_QWORD: |
|
320 |
- value = avio_rl64(pb); |
|
321 |
- break; |
|
322 |
- case ASF_WORD: |
|
323 |
- value = avio_rl16(pb); |
|
324 |
- break; |
|
325 |
- default: |
|
326 |
- av_freep(&name); |
|
327 |
- return AVERROR_INVALIDDATA; |
|
328 |
- } |
|
329 |
- snprintf(buf, sizeof(buf), "%"PRIu64, value); |
|
330 |
- if (av_dict_set(met, name, buf, 0) < 0) |
|
331 |
- av_log(s, AV_LOG_WARNING, "av_dict_set failed.\n"); |
|
332 |
- |
|
333 |
- return 0; |
|
334 |
-} |
|
335 |
- |
|
336 |
-/* MSDN claims that this should be "compatible with the ID3 frame, APIC", |
|
337 |
- * but in reality this is only loosely similar */ |
|
338 |
-static int asf_read_picture(AVFormatContext *s, int len) |
|
339 |
-{ |
|
340 |
- ASFContext *asf = s->priv_data; |
|
341 |
- AVPacket pkt = { 0 }; |
|
342 |
- const CodecMime *mime = ff_id3v2_mime_tags; |
|
343 |
- enum AVCodecID id = AV_CODEC_ID_NONE; |
|
344 |
- char mimetype[64]; |
|
345 |
- uint8_t *desc = NULL; |
|
346 |
- AVStream *st = NULL; |
|
347 |
- int ret, type, picsize, desc_len; |
|
348 |
- ASFStream *asf_st; |
|
349 |
- |
|
350 |
- /* type + picsize + mime + desc */ |
|
351 |
- if (len < 1 + 4 + 2 + 2) { |
|
352 |
- av_log(s, AV_LOG_ERROR, "Invalid attached picture size: %d.\n", len); |
|
353 |
- return AVERROR_INVALIDDATA; |
|
354 |
- } |
|
355 |
- |
|
356 |
- /* picture type */ |
|
357 |
- type = avio_r8(s->pb); |
|
358 |
- len--; |
|
359 |
- if (type >= FF_ARRAY_ELEMS(ff_id3v2_picture_types) || type < 0) { |
|
360 |
- av_log(s, AV_LOG_WARNING, "Unknown attached picture type: %d.\n", type); |
|
361 |
- type = 0; |
|
362 |
- } |
|
363 |
- |
|
364 |
- /* picture data size */ |
|
365 |
- picsize = avio_rl32(s->pb); |
|
366 |
- len -= 4; |
|
367 |
- |
|
368 |
- /* picture MIME type */ |
|
369 |
- len -= avio_get_str16le(s->pb, len, mimetype, sizeof(mimetype)); |
|
370 |
- while (mime->id != AV_CODEC_ID_NONE) { |
|
371 |
- if (!strncmp(mime->str, mimetype, sizeof(mimetype))) { |
|
372 |
- id = mime->id; |
|
373 |
- break; |
|
374 |
- } |
|
375 |
- mime++; |
|
376 |
- } |
|
377 |
- if (id == AV_CODEC_ID_NONE) { |
|
378 |
- av_log(s, AV_LOG_ERROR, "Unknown attached picture mimetype: %s.\n", |
|
379 |
- mimetype); |
|
380 |
- return 0; |
|
381 |
- } |
|
382 |
- |
|
383 |
- if (picsize >= len) { |
|
384 |
- av_log(s, AV_LOG_ERROR, "Invalid attached picture data size: %d >= %d.\n", |
|
385 |
- picsize, len); |
|
386 |
- return AVERROR_INVALIDDATA; |
|
387 |
- } |
|
388 |
- |
|
389 |
- /* picture description */ |
|
390 |
- desc_len = (len - picsize) * 2 + 1; |
|
391 |
- desc = av_malloc(desc_len); |
|
392 |
- if (!desc) |
|
393 |
- return AVERROR(ENOMEM); |
|
394 |
- len -= avio_get_str16le(s->pb, len - picsize, desc, desc_len); |
|
395 |
- |
|
396 |
- ret = av_get_packet(s->pb, &pkt, picsize); |
|
397 |
- if (ret < 0) |
|
398 |
- goto fail; |
|
399 |
- |
|
400 |
- st = avformat_new_stream(s, NULL); |
|
401 |
- if (!st) { |
|
402 |
- ret = AVERROR(ENOMEM); |
|
403 |
- goto fail; |
|
404 |
- } |
|
405 |
- asf->asf_st[asf->nb_streams] = av_mallocz(sizeof(*asf_st)); |
|
406 |
- asf_st = asf->asf_st[asf->nb_streams]; |
|
407 |
- if (!asf_st) |
|
408 |
- return AVERROR(ENOMEM); |
|
409 |
- |
|
410 |
- st->disposition |= AV_DISPOSITION_ATTACHED_PIC; |
|
411 |
- st->codec->codec_type = asf_st->type = AVMEDIA_TYPE_VIDEO; |
|
412 |
- st->codec->codec_id = id; |
|
413 |
- st->attached_pic = pkt; |
|
414 |
- st->attached_pic.stream_index = asf_st->index = st->index; |
|
415 |
- st->attached_pic.flags |= AV_PKT_FLAG_KEY; |
|
416 |
- |
|
417 |
- asf->nb_streams++; |
|
418 |
- |
|
419 |
- if (*desc) { |
|
420 |
- if (av_dict_set(&st->metadata, "title", desc, AV_DICT_DONT_STRDUP_VAL) < 0) |
|
421 |
- av_log(s, AV_LOG_WARNING, "av_dict_set failed.\n"); |
|
422 |
- } else |
|
423 |
- av_freep(&desc); |
|
424 |
- |
|
425 |
- if (av_dict_set(&st->metadata, "comment", ff_id3v2_picture_types[type], 0) < 0) |
|
426 |
- av_log(s, AV_LOG_WARNING, "av_dict_set failed.\n"); |
|
427 |
- |
|
428 |
- return 0; |
|
429 |
- |
|
430 |
-fail: |
|
431 |
- av_freep(&desc); |
|
432 |
- av_free_packet(&pkt); |
|
433 |
- return ret; |
|
434 |
-} |
|
435 |
- |
|
436 |
-static void get_id3_tag(AVFormatContext *s, int len) |
|
437 |
-{ |
|
438 |
- ID3v2ExtraMeta *id3v2_extra_meta = NULL; |
|
439 |
- |
|
440 |
- ff_id3v2_read(s, ID3v2_DEFAULT_MAGIC, &id3v2_extra_meta, len); |
|
441 |
- if (id3v2_extra_meta) |
|
442 |
- ff_id3v2_parse_apic(s, &id3v2_extra_meta); |
|
443 |
- ff_id3v2_free_extra_meta(&id3v2_extra_meta); |
|
444 |
-} |
|
445 |
- |
|
446 |
-static int process_metadata(AVFormatContext *s, uint8_t *name, uint16_t name_len, |
|
447 |
- uint16_t val_len, uint16_t type, AVDictionary **met) |
|
448 |
-{ |
|
449 |
- int ret; |
|
450 |
- ff_asf_guid guid; |
|
451 |
- |
|
452 |
- if (val_len) { |
|
453 |
- switch (type) { |
|
454 |
- case ASF_UNICODE: |
|
455 |
- asf_read_value(s, name, name_len, val_len, type, met); |
|
456 |
- break; |
|
457 |
- case ASF_BYTE_ARRAY: |
|
458 |
- if (!strcmp(name, "WM/Picture")) // handle cover art |
|
459 |
- asf_read_picture(s, val_len); |
|
460 |
- else if (!strcmp(name, "ID3")) // handle ID3 tag |
|
461 |
- get_id3_tag(s, val_len); |
|
462 |
- else |
|
463 |
- asf_read_value(s, name, name_len, val_len, type, met); |
|
464 |
- break; |
|
465 |
- case ASF_GUID: |
|
466 |
- ff_get_guid(s->pb, &guid); |
|
467 |
- break; |
|
468 |
- default: |
|
469 |
- if ((ret = asf_read_generic_value(s, name, name_len, type, met)) < 0) |
|
470 |
- return ret; |
|
471 |
- break; |
|
472 |
- } |
|
473 |
- } |
|
474 |
- av_freep(&name); |
|
475 |
- |
|
476 |
- return 0; |
|
477 |
-} |
|
478 |
- |
|
479 |
-static int asf_read_ext_content(AVFormatContext *s, const GUIDParseTable *g) |
|
480 |
-{ |
|
481 |
- ASFContext *asf = s->priv_data; |
|
482 |
- AVIOContext *pb = s->pb; |
|
483 |
- uint64_t size = avio_rl64(pb); |
|
484 |
- uint16_t nb_desc = avio_rl16(pb); |
|
485 |
- int i, ret; |
|
486 |
- |
|
487 |
- for (i = 0; i < nb_desc; i++) { |
|
488 |
- uint16_t name_len, type, val_len; |
|
489 |
- uint8_t *name = NULL; |
|
490 |
- |
|
491 |
- name_len = avio_rl16(pb); |
|
492 |
- if (!name_len) |
|
493 |
- return AVERROR_INVALIDDATA; |
|
494 |
- name = av_malloc(name_len); |
|
495 |
- if (!name) |
|
496 |
- return AVERROR(ENOMEM); |
|
497 |
- avio_get_str16le(pb, name_len, name, |
|
498 |
- name_len); |
|
499 |
- type = avio_rl16(pb); |
|
500 |
- val_len = avio_rl16(pb); |
|
501 |
- |
|
502 |
- if ((ret = process_metadata(s, name, name_len, val_len, type, &s->metadata)) < 0) |
|
503 |
- return ret; |
|
504 |
- } |
|
505 |
- |
|
506 |
- align_position(pb, asf->offset, size); |
|
507 |
- return 0; |
|
508 |
-} |
|
509 |
- |
|
510 |
-static AVStream *find_stream(AVFormatContext *s, uint16_t st_num) |
|
511 |
-{ |
|
512 |
- AVStream *st = NULL; |
|
513 |
- ASFContext *asf = s->priv_data; |
|
514 |
- int i; |
|
515 |
- |
|
516 |
- for (i = 0; i < asf->nb_streams; i++) { |
|
517 |
- if (asf->asf_st[i]->stream_index == st_num) { |
|
518 |
- st = s->streams[asf->asf_st[i]->index]; |
|
519 |
- break; |
|
520 |
- } |
|
521 |
- } |
|
522 |
- |
|
523 |
- return st; |
|
524 |
-} |
|
525 |
- |
|
526 |
-static void asf_store_aspect_ratio(AVFormatContext *s, uint8_t st_num, uint8_t *name) |
|
527 |
-{ |
|
528 |
- ASFContext *asf = s->priv_data; |
|
529 |
- AVIOContext *pb = s->pb; |
|
530 |
- uint16_t value = 0; |
|
531 |
- |
|
532 |
- value = avio_rl16(pb); |
|
533 |
- |
|
534 |
- if (st_num < ASF_MAX_STREAMS) { |
|
535 |
- if (!strcmp(name, "AspectRatioX")) |
|
536 |
- asf->asf_sd[st_num].aspect_ratio.num = value; |
|
537 |
- else |
|
538 |
- asf->asf_sd[st_num].aspect_ratio.den = value; |
|
539 |
- } |
|
540 |
-} |
|
541 |
- |
|
542 |
-static int asf_read_metadata_obj(AVFormatContext *s, const GUIDParseTable *g) |
|
543 |
-{ |
|
544 |
- ASFContext *asf = s->priv_data; |
|
545 |
- AVIOContext *pb = s->pb; |
|
546 |
- uint64_t size = avio_rl64(pb); |
|
547 |
- uint16_t nb_recs = avio_rl16(pb); // number of records in the Description Records list |
|
548 |
- int i, ret; |
|
549 |
- |
|
550 |
- for (i = 0; i < nb_recs; i++) { |
|
551 |
- uint16_t name_len, buflen, type, val_len, st_num; |
|
552 |
- uint8_t *name = NULL; |
|
553 |
- |
|
554 |
- avio_skip(pb, 2); // skip reserved field |
|
555 |
- st_num = avio_rl16(pb); |
|
556 |
- name_len = avio_rl16(pb); |
|
557 |
- buflen = 2 * name_len + 1; |
|
558 |
- if (!name_len) |
|
559 |
- break; |
|
560 |
- type = avio_rl16(pb); |
|
561 |
- val_len = avio_rl32(pb); |
|
562 |
- name = av_malloc(name_len); |
|
563 |
- if (!name) |
|
564 |
- return AVERROR(ENOMEM); |
|
565 |
- avio_get_str16le(pb, name_len, name, |
|
566 |
- buflen); |
|
567 |
- |
|
568 |
- if (!strcmp(name, "AspectRatioX") || !strcmp(name, "AspectRatioY")) { |
|
569 |
- asf_store_aspect_ratio(s, st_num, name); |
|
570 |
- } else { |
|
571 |
- if (st_num < ASF_MAX_STREAMS) { |
|
572 |
- if ((ret = process_metadata(s, name, name_len, val_len, type, |
|
573 |
- &asf->asf_sd[st_num].asf_met)) < 0) |
|
574 |
- break; |
|
575 |
- } |
|
576 |
- } |
|
577 |
- } |
|
578 |
- |
|
579 |
- align_position(pb, asf->offset, size); |
|
580 |
- return 0; |
|
581 |
-} |
|
582 |
- |
|
583 |
-static int asf_read_content_desc(AVFormatContext *s, const GUIDParseTable *g) |
|
584 |
-{ |
|
585 |
- ASFContext *asf = s->priv_data; |
|
586 |
- AVIOContext *pb = s->pb; |
|
587 |
- int i; |
|
588 |
- static const char *const titles[] = |
|
589 |
- { "Title", "Author", "Copyright", "Description", "Rate" }; |
|
590 |
- uint16_t len[5], buflen[5] = { 0 }; |
|
591 |
- uint8_t *ch; |
|
592 |
- uint64_t size = avio_rl64(pb); |
|
593 |
- |
|
594 |
- for (i = 0; i < 5; i++) { |
|
595 |
- len[i] = avio_rl16(pb); |
|
596 |
- // utf8 string should be <= 2 * utf16 string, extra byte for the terminator |
|
597 |
- buflen[i] = 2 * len[i] + 1; |
|
598 |
- } |
|
599 |
- |
|
600 |
- for (i = 0; i < 5; i++) { |
|
601 |
- ch = av_malloc(buflen[i]); |
|
602 |
- if (!ch) |
|
603 |
- return(AVERROR(ENOMEM)); |
|
604 |
- asf_read_metadata(s, titles[i], len[i], ch, buflen[i]); |
|
605 |
- av_freep(&ch); |
|
606 |
- } |
|
607 |
- align_position(pb, asf->offset, size); |
|
608 |
- |
|
609 |
- return 0; |
|
610 |
-} |
|
611 |
- |
|
612 |
-static int asf_read_properties(AVFormatContext *s, const GUIDParseTable *g) |
|
613 |
-{ |
|
614 |
- ASFContext *asf = s->priv_data; |
|
615 |
- AVIOContext *pb = s->pb; |
|
616 |
- uint64_t creation_time; |
|
617 |
- |
|
618 |
- avio_rl64(pb); // read object size |
|
619 |
- avio_skip(pb, 16); // skip File ID |
|
620 |
- avio_skip(pb, 8); // skip File size |
|
621 |
- creation_time = avio_rl64(pb); |
|
622 |
- if (!(asf->b_flags & ASF_FLAG_BROADCAST)) { |
|
623 |
- struct tm tmbuf; |
|
624 |
- struct tm *tm; |
|
625 |
- char buf[64]; |
|
626 |
- |
|
627 |
- // creation date is in 100 ns units from 1 Jan 1601, conversion to s |
|
628 |
- creation_time /= 10000000; |
|
629 |
- // there are 11644473600 seconds between 1 Jan 1601 and 1 Jan 1970 |
|
630 |
- creation_time -= 11644473600; |
|
631 |
- tm = gmtime_r(&creation_time, &tmbuf); |
|
632 |
- if (tm) { |
|
633 |
- if (!strftime(buf, sizeof(buf), "%Y-%m-%d %H:%M:%S", tm)) |
|
634 |
- buf[0] = '\0'; |
|
635 |
- } else |
|
636 |
- buf[0] = '\0'; |
|
637 |
- if (buf[0]) { |
|
638 |
- if (av_dict_set(&s->metadata, "creation_time", buf, 0) < 0) |
|
639 |
- av_log(s, AV_LOG_WARNING, "av_dict_set failed.\n"); |
|
640 |
- } |
|
641 |
- } |
|
642 |
- asf->nb_packets = avio_rl64(pb); |
|
643 |
- asf->duration = avio_rl64(pb) / 10000; // stream duration |
|
644 |
- avio_skip(pb, 8); // skip send duration |
|
645 |
- asf->preroll = avio_rl64(pb); |
|
646 |
- asf->duration -= asf->preroll; |
|
647 |
- asf->b_flags = avio_rl32(pb); |
|
648 |
- avio_skip(pb, 4); // skip minimal packet size |
|
649 |
- asf->packet_size = avio_rl32(pb); |
|
650 |
- avio_skip(pb, 4); // skip max_bitrate |
|
651 |
- |
|
652 |
- return 0; |
|
653 |
-} |
|
654 |
- |
|
655 |
-static int parse_video_info(AVIOContext *pb, AVStream *st) |
|
656 |
-{ |
|
657 |
- uint16_t size; |
|
658 |
- unsigned int tag; |
|
659 |
- |
|
660 |
- st->codec->width = avio_rl32(pb); |
|
661 |
- st->codec->height = avio_rl32(pb); |
|
662 |
- avio_skip(pb, 1); // skip reserved flags |
|
663 |
- size = avio_rl16(pb); // size of the Format Data |
|
664 |
- tag = ff_get_bmp_header(pb, st, NULL); |
|
665 |
- st->codec->codec_tag = tag; |
|
666 |
- st->codec->codec_id = ff_codec_get_id(ff_codec_bmp_tags, tag); |
|
667 |
- |
|
668 |
- if (size > BMP_HEADER_SIZE) { |
|
669 |
- int ret; |
|
670 |
- st->codec->extradata_size = size - BMP_HEADER_SIZE; |
|
671 |
- if (!(st->codec->extradata = av_malloc(st->codec->extradata_size + |
|
672 |
- FF_INPUT_BUFFER_PADDING_SIZE))) { |
|
673 |
- st->codec->extradata_size = 0; |
|
674 |
- return AVERROR(ENOMEM); |
|
675 |
- } |
|
676 |
- memset(st->codec->extradata + st->codec->extradata_size , 0, |
|
677 |
- FF_INPUT_BUFFER_PADDING_SIZE); |
|
678 |
- if ((ret = avio_read(pb, st->codec->extradata, |
|
679 |
- st->codec->extradata_size)) < 0) |
|
680 |
- return ret; |
|
681 |
- } |
|
682 |
- return 0; |
|
683 |
-} |
|
684 |
- |
|
685 |
-static int asf_read_stream_properties(AVFormatContext *s, const GUIDParseTable *g) |
|
686 |
-{ |
|
687 |
- ASFContext *asf = s->priv_data; |
|
688 |
- AVIOContext *pb = s->pb; |
|
689 |
- uint64_t size; |
|
690 |
- uint32_t err_data_len, ts_data_len; // type specific data length |
|
691 |
- uint16_t flags; |
|
692 |
- ff_asf_guid stream_type; |
|
693 |
- enum AVMediaType type; |
|
694 |
- int i, ret; |
|
695 |
- uint8_t stream_index; |
|
696 |
- AVStream *st; |
|
697 |
- ASFStream *asf_st; |
|
698 |
- |
|
699 |
- // ASF file must not contain more than 128 streams according to the specification |
|
700 |
- if (asf->nb_streams >= ASF_MAX_STREAMS) |
|
701 |
- return AVERROR_INVALIDDATA; |
|
702 |
- |
|
703 |
- size = avio_rl64(pb); |
|
704 |
- ff_get_guid(pb, &stream_type); |
|
705 |
- if (!ff_guidcmp(&stream_type, &ff_asf_audio_stream)) |
|
706 |
- type = AVMEDIA_TYPE_AUDIO; |
|
707 |
- else if (!ff_guidcmp(&stream_type, &ff_asf_video_stream)) |
|
708 |
- type = AVMEDIA_TYPE_VIDEO; |
|
709 |
- else if (!ff_guidcmp(&stream_type, &ff_asf_jfif_media)) |
|
710 |
- type = AVMEDIA_TYPE_VIDEO; |
|
711 |
- else if (!ff_guidcmp(&stream_type, &ff_asf_command_stream)) |
|
712 |
- type = AVMEDIA_TYPE_DATA; |
|
713 |
- else if (!ff_guidcmp(&stream_type, |
|
714 |
- &ff_asf_ext_stream_embed_stream_header)) |
|
715 |
- type = AVMEDIA_TYPE_UNKNOWN; |
|
716 |
- else |
|
717 |
- return AVERROR_INVALIDDATA; |
|
718 |
- |
|
719 |
- ff_get_guid(pb, &stream_type); // error correction type |
|
720 |
- avio_skip(pb, 8); // skip the time offset |
|
721 |
- ts_data_len = avio_rl32(pb); |
|
722 |
- err_data_len = avio_rl32(pb); |
|
723 |
- flags = avio_rl16(pb); // bit 15 - Encrypted Content |
|
724 |
- |
|
725 |
- stream_index = flags & ASF_STREAM_NUM; |
|
726 |
- for (i = 0; i < asf->nb_streams; i++) |
|
727 |
- if (stream_index == asf->asf_st[i]->stream_index) { |
|
728 |
- av_log(s, AV_LOG_WARNING, |
|
729 |
- "Duplicate stream found, this stream will be ignored.\n"); |
|
730 |
- align_position(pb, asf->offset, size); |
|
731 |
- return 0; |
|
732 |
- } |
|
733 |
- |
|
734 |
- st = avformat_new_stream(s, NULL); |
|
735 |
- if (!st) |
|
736 |
- return AVERROR(ENOMEM); |
|
737 |
- avpriv_set_pts_info(st, 32, 1, 1000); // pts should be dword, in milliseconds |
|
738 |
- st->codec->codec_type = type; |
|
739 |
- asf->asf_st[asf->nb_streams] = av_mallocz(sizeof(*asf_st)); |
|
740 |
- if (!asf->asf_st[asf->nb_streams]) |
|
741 |
- return AVERROR(ENOMEM); |
|
742 |
- asf_st = asf->asf_st[asf->nb_streams]; |
|
743 |
- asf_st->stream_index = stream_index; |
|
744 |
- asf_st->index = st->index; |
|
745 |
- asf_st->indexed = 0; |
|
746 |
- st->id = flags & ASF_STREAM_NUM; |
|
747 |
- av_init_packet(&asf_st->pkt.avpkt); |
|
748 |
- asf_st->pkt.data_size = 0; |
|
749 |
- avio_skip(pb, 4); // skip reserved field |
|
750 |
- if (!ts_data_len) { |
|
751 |
- av_log(s, AV_LOG_WARNING, "Suspicious data found! ASF stream #%d will be ignored.\n", |
|
752 |
- asf_st->stream_index); |
|
753 |
- align_position(pb, asf->offset, size); |
|
754 |
- return 0; |
|
755 |
- } |
|
756 |
- |
|
757 |
- switch (type) { |
|
758 |
- case AVMEDIA_TYPE_AUDIO: |
|
759 |
- asf_st->type = AVMEDIA_TYPE_AUDIO; |
|
760 |
- if ((ret = ff_get_wav_header(pb, st->codec, ts_data_len, 0)) < 0) |
|
761 |
- return ret; |
|
762 |
- break; |
|
763 |
- case AVMEDIA_TYPE_VIDEO: |
|
764 |
- asf_st->type = AVMEDIA_TYPE_VIDEO; |
|
765 |
- if ((ret = parse_video_info(pb, st)) < 0) |
|
766 |
- return ret; |
|
767 |
- break; |
|
768 |
- default: |
|
769 |
- avio_skip(pb, ts_data_len); |
|
770 |
- break; |
|
771 |
- } |
|
772 |
- |
|
773 |
- if (err_data_len) { |
|
774 |
- if (type == AVMEDIA_TYPE_AUDIO) { |
|
775 |
- uint8_t span = avio_r8(pb); |
|
776 |
- if (span > 1) { |
|
777 |
- asf_st->span = span; |
|
778 |
- asf_st->virtual_pkt_len = avio_rl16(pb); |
|
779 |
- asf_st->virtual_chunk_len = avio_rl16(pb); |
|
780 |
- avio_skip(pb, err_data_len - 5); |
|
781 |
- } else |
|
782 |
- avio_skip(pb, err_data_len - 1); |
|
783 |
- } else |
|
784 |
- avio_skip(pb, err_data_len); |
|
785 |
- } |
|
786 |
- |
|
787 |
- asf->nb_streams++; |
|
788 |
- align_position(pb, asf->offset, size); |
|
789 |
- |
|
790 |
- return 0; |
|
791 |
-} |
|
792 |
- |
|
793 |
-static void set_language(AVFormatContext *s, const char *rfc1766, AVDictionary **met) |
|
794 |
-{ |
|
795 |
- // language abbr should contain at least 2 chars |
|
796 |
- if (rfc1766 && strlen(rfc1766) > 1) { |
|
797 |
- const char primary_tag[3] = { rfc1766[0], rfc1766[1], '\0' }; // ignore country code if any |
|
798 |
- const char *iso6392 = av_convert_lang_to(primary_tag, |
|
799 |
- AV_LANG_ISO639_2_BIBL); |
|
800 |
- if (iso6392) |
|
801 |
- if (av_dict_set(met, "language", iso6392, 0) < 0) |
|
802 |
- av_log(s, AV_LOG_WARNING, "av_dict_set failed.\n"); |
|
803 |
- } |
|
804 |
-} |
|
805 |
- |
|
806 |
-static int asf_read_ext_stream_properties(AVFormatContext *s, const GUIDParseTable *g) |
|
807 |
-{ |
|
808 |
- ASFContext *asf = s->priv_data; |
|
809 |
- AVIOContext *pb = s->pb; |
|
810 |
- AVStream *st = NULL; |
|
811 |
- ff_asf_guid guid; |
|
812 |
- uint16_t nb_st_name, nb_pay_exts, st_num, lang_idx; |
|
813 |
- int i, ret; |
|
814 |
- uint32_t bitrate; |
|
815 |
- uint64_t start_time, end_time, time_per_frame; |
|
816 |
- uint64_t size = avio_rl64(pb); |
|
817 |
- |
|
818 |
- start_time = avio_rl64(pb); |
|
819 |
- end_time = avio_rl64(pb); |
|
820 |
- bitrate = avio_rl32(pb); |
|
821 |
- avio_skip(pb, 28); // skip some unused values |
|
822 |
- st_num = avio_rl16(pb); |
|
823 |
- st_num &= ASF_STREAM_NUM; |
|
824 |
- lang_idx = avio_rl16(pb); // Stream Language ID Index |
|
825 |
- for (i = 0; i < asf->nb_streams; i++) { |
|
826 |
- if (st_num == asf->asf_st[i]->stream_index) { |
|
827 |
- st = s->streams[asf->asf_st[i]->index]; |
|
828 |
- asf->asf_st[i]->lang_idx = lang_idx; |
|
829 |
- break; |
|
830 |
- } |
|
831 |
- } |
|
832 |
- time_per_frame = avio_rl64(pb); // average time per frame |
|
833 |
- if (st) { |
|
834 |
- st->start_time = start_time; |
|
835 |
- st->duration = end_time - start_time; |
|
836 |
- st->codec->bit_rate = bitrate; |
|
837 |
- st->avg_frame_rate.num = 10000000; |
|
838 |
- st->avg_frame_rate.den = time_per_frame; |
|
839 |
- } |
|
840 |
- nb_st_name = avio_rl16(pb); |
|
841 |
- nb_pay_exts = avio_rl16(pb); |
|
842 |
- for (i = 0; i < nb_st_name; i++) { |
|
843 |
- uint16_t len; |
|
844 |
- |
|
845 |
- avio_rl16(pb); // Language ID Index |
|
846 |
- len = avio_rl16(pb); |
|
847 |
- avio_skip(pb, len); |
|
848 |
- } |
|
849 |
- |
|
850 |
- for (i = 0; i < nb_pay_exts; i++) { |
|
851 |
- uint32_t len; |
|
852 |
- avio_skip(pb, 16); // Extension System ID |
|
853 |
- avio_skip(pb, 2); // Extension Data Size |
|
854 |
- len = avio_rl32(pb); |
|
855 |
- avio_skip(pb, len); |
|
856 |
- } |
|
857 |
- |
|
858 |
- if ((ret = ff_get_guid(pb, &guid)) < 0) { |
|
859 |
- align_position(pb, asf->offset, size); |
|
860 |
- |
|
861 |
- return 0; |
|
862 |
- } |
|
863 |
- |
|
864 |
- g = find_guid(guid); |
|
865 |
- if (g && !(strcmp(g->name, "Stream Properties"))) { |
|
866 |
- if ((ret = g->read_object(s, g)) < 0) |
|
867 |
- return ret; |
|
868 |
- } |
|
869 |
- |
|
870 |
- align_position(pb, asf->offset, size); |
|
871 |
- return 0; |
|
872 |
-} |
|
873 |
- |
|
874 |
-static int asf_read_language_list(AVFormatContext *s, const GUIDParseTable *g) |
|
875 |
-{ |
|
876 |
- ASFContext *asf = s->priv_data; |
|
877 |
- AVIOContext *pb = s->pb; |
|
878 |
- int i, ret; |
|
879 |
- uint64_t size = avio_rl64(pb); |
|
880 |
- uint16_t nb_langs = avio_rl16(pb); |
|
881 |
- |
|
882 |
- if (nb_langs < ASF_MAX_STREAMS) { |
|
883 |
- for (i = 0; i < nb_langs; i++) { |
|
884 |
- size_t len; |
|
885 |
- len = avio_r8(pb); |
|
886 |
- if (!len) |
|
887 |
- len = 6; |
|
888 |
- if ((ret = get_asf_string(pb, len, asf->asf_sd[i].langs, |
|
889 |
- sizeof(asf->asf_sd[i].langs))) < 0) { |
|
890 |
- return ret; |
|
891 |
- } |
|
892 |
- } |
|
893 |
- } |
|
894 |
- |
|
895 |
- align_position(pb, asf->offset, size); |
|
896 |
- return 0; |
|
897 |
-} |
|
898 |
- |
|
899 |
-// returns data object offset when reading this object for the first time |
|
900 |
-static int asf_read_data(AVFormatContext *s, const GUIDParseTable *g) |
|
901 |
-{ |
|
902 |
- ASFContext *asf = s->priv_data; |
|
903 |
- AVIOContext *pb = s->pb; |
|
904 |
- uint64_t size = asf->data_size = avio_rl64(pb); |
|
905 |
- int i; |
|
906 |
- |
|
907 |
- if (!asf->data_reached && pb->seekable) { |
|
908 |
- asf->data_reached = 1; |
|
909 |
- asf->data_offset = asf->offset; |
|
910 |
- } |
|
911 |
- |
|
912 |
- for (i = 0; i < asf->nb_streams; i++) { |
|
913 |
- if (!(asf->b_flags & ASF_FLAG_BROADCAST)) |
|
914 |
- s->streams[i]->duration = asf->duration; |
|
915 |
- } |
|
916 |
- asf->nb_mult_left = 0; |
|
917 |
- asf->sub_left = 0; |
|
918 |
- asf->state = PARSE_PACKET_HEADER; |
|
919 |
- asf->return_subpayload = 0; |
|
920 |
- asf->packet_size_internal = 0; |
|
921 |
- avio_skip(pb, 16); // skip File ID |
|
922 |
- size = avio_rl64(pb); // Total Data Packets |
|
923 |
- if (size != asf->nb_packets) |
|
924 |
- av_log(s, AV_LOG_WARNING, |
|
925 |
- "Number of Packets from File Properties Object is not equal to Total" |
|
926 |
- "Datapackets value! num of packets %"PRIu64" total num %"PRIu64".\n", |
|
927 |
- size, asf->nb_packets); |
|
928 |
- avio_skip(pb, 2); // skip reserved field |
|
929 |
- asf->first_packet_offset = avio_tell(pb); |
|
930 |
- align_position(pb, asf->offset, asf->data_size); |
|
931 |
- |
|
932 |
- return 0; |
|
933 |
-} |
|
934 |
- |
|
935 |
-static int asf_read_simple_index(AVFormatContext *s, const GUIDParseTable *g) |
|
936 |
-{ |
|
937 |
- ASFContext *asf = s->priv_data; |
|
938 |
- AVIOContext *pb = s->pb; |
|
939 |
- AVStream *st = NULL; |
|
940 |
- uint64_t interval; // index entry time interval in 100 ns units, usually it's 1s |
|
941 |
- uint32_t pkt_num, nb_entries; |
|
942 |
- int32_t prev_pkt_num = -1; |
|
943 |
- int i; |
|
944 |
- uint64_t size = avio_rl64(pb); |
|
945 |
- |
|
946 |
- // simple index objects should be ordered by stream number, this loop tries to find |
|
947 |
- // the first not indexed video stream |
|
948 |
- for (i = 0; i < asf->nb_streams; i++) { |
|
949 |
- if ((asf->asf_st[i]->type == AVMEDIA_TYPE_VIDEO) && !asf->asf_st[i]->indexed) { |
|
950 |
- asf->asf_st[i]->indexed = 1; |
|
951 |
- st = s->streams[asf->asf_st[i]->index]; |
|
952 |
- break; |
|
953 |
- } |
|
954 |
- } |
|
955 |
- if (!st) { |
|
956 |
- avio_skip(pb, size - 24); // if there's no video stream, skip index object |
|
957 |
- return 0; |
|
958 |
- } |
|
959 |
- avio_skip(pb, 16); // skip File ID |
|
960 |
- interval = avio_rl64(pb); |
|
961 |
- avio_skip(pb, 4); |
|
962 |
- nb_entries = avio_rl32(pb); |
|
963 |
- for (i = 0; i < nb_entries; i++) { |
|
964 |
- pkt_num = avio_rl32(pb); |
|
965 |
- avio_skip(pb, 2); |
|
966 |
- if (prev_pkt_num != pkt_num) { |
|
967 |
- av_add_index_entry(st, asf->first_packet_offset + asf->packet_size * |
|
968 |
- pkt_num, av_rescale(interval, i, 10000), |
|
969 |
- asf->packet_size, 0, AVINDEX_KEYFRAME); |
|
970 |
- prev_pkt_num = pkt_num; |
|
971 |
- } |
|
972 |
- } |
|
973 |
- asf->is_simple_index = 1; |
|
974 |
- align_position(pb, asf->offset, size); |
|
975 |
- |
|
976 |
- return 0; |
|
977 |
-} |
|
978 |
- |
|
979 |
-static const GUIDParseTable gdef[] = { |
|
980 |
- { "Data", { 0x75, 0xB2, 0x26, 0x36, 0x66, 0x8E, 0x11, 0xCF, 0xA6, 0xD9, 0x00, 0xAA, 0x00, 0x62, 0xCE, 0x6C }, asf_read_data, 1 }, |
|
981 |
- { "Simple Index", { 0x33, 0x00, 0x08, 0x90, 0xE5, 0xB1, 0x11, 0xCF, 0x89, 0xF4, 0x00, 0xA0, 0xC9, 0x03, 0x49, 0xCB }, asf_read_simple_index, 1 }, |
|
982 |
- { "Content Description", { 0x75, 0xB2, 0x26, 0x33, 0x66 ,0x8E, 0x11, 0xCF, 0xA6, 0xD9, 0x00, 0xAA, 0x00, 0x62, 0xCE, 0x6C }, asf_read_content_desc, 1 }, |
|
983 |
- { "Extended Content Description", { 0xD2, 0xD0, 0xA4, 0x40, 0xE3, 0x07, 0x11, 0xD2, 0x97, 0xF0, 0x00, 0xA0, 0xC9, 0x5e, 0xA8, 0x50 }, asf_read_ext_content, 1 }, |
|
984 |
- { "Stream Bitrate Properties", { 0x7B, 0xF8, 0x75, 0xCE, 0x46, 0x8D, 0x11, 0xD1, 0x8D, 0x82, 0x00, 0x60, 0x97, 0xC9, 0xA2, 0xB2 }, asf_read_unknown, 1 }, |
|
985 |
- { "File Properties", { 0x8C, 0xAB, 0xDC, 0xA1, 0xA9, 0x47, 0x11, 0xCF, 0x8E, 0xE4, 0x00, 0xC0, 0x0C, 0x20, 0x53, 0x65 }, asf_read_properties, 1 }, |
|
986 |
- { "Header Extension", { 0x5F, 0xBF, 0x03, 0xB5, 0xA9, 0x2E, 0x11, 0xCF, 0x8E, 0xE3, 0x00, 0xC0, 0x0C, 0x20, 0x53, 0x65 }, asf_read_unknown, 0 }, |
|
987 |
- { "Stream Properties", { 0xB7, 0xDC, 0x07, 0x91, 0xA9, 0xB7, 0x11, 0xCF, 0x8E, 0xE6, 0x00, 0xC0, 0x0C, 0x20, 0x53, 0x65 }, asf_read_stream_properties, 1 }, |
|
988 |
- { "Codec List", { 0x86, 0xD1, 0x52, 0x40, 0x31, 0x1D, 0x11, 0xD0, 0xA3, 0xA4, 0x00, 0xA0, 0xC9, 0x03, 0x48, 0xF6 }, asf_read_unknown, 1 }, |
|
989 |
- { "Marker", { 0xF4, 0x87, 0xCD, 0x01, 0xA9, 0x51, 0x11, 0xCF, 0x8E, 0xE6, 0x00, 0xC0, 0x0C, 0x20, 0x53, 0x65 }, asf_read_marker, 1 }, |
|
990 |
- { "Script Command", { 0x1E, 0xFB, 0x1A, 0x30, 0x0B, 0x62, 0x11, 0xD0, 0xA3, 0x9B, 0x00, 0xA0, 0xC9, 0x03, 0x48, 0xF6 }, asf_read_unknown, 1 }, |
|
991 |
- { "Language List", { 0x7C, 0x43, 0x46, 0xa9, 0xef, 0xe0, 0x4B, 0xFC, 0xB2, 0x29, 0x39, 0x3e, 0xde, 0x41, 0x5c, 0x85 }, asf_read_language_list, 1}, |
|
992 |
- { "Padding", { 0x18, 0x06, 0xD4, 0x74, 0xCA, 0xDF, 0x45, 0x09, 0xA4, 0xBA, 0x9A, 0xAB, 0xCB, 0x96, 0xAA, 0xE8 }, asf_read_unknown, 1 }, |
|
993 |
- { "DRMv1 Header", { 0x22, 0x11, 0xB3, 0xFB, 0xBD, 0x23, 0x11, 0xD2, 0xB4, 0xB7, 0x00, 0xA0, 0xC9, 0x55, 0xFC, 0x6E }, asf_read_unknown, 1 }, |
|
994 |
- { "DRMv2 Header", { 0x29, 0x8A, 0xE6, 0x14, 0x26, 0x22, 0x4C, 0x17, 0xB9, 0x35, 0xDA, 0xE0, 0x7E, 0xE9, 0x28, 0x9c }, asf_read_unknown, 1 }, |
|
995 |
- { "Index", { 0xD6, 0xE2, 0x29, 0xD3, 0x35, 0xDA, 0x11, 0xD1, 0x90, 0x34, 0x00, 0xA0, 0xC9, 0x03, 0x49, 0xBE }, asf_read_unknown, 1 }, |
|
996 |
- { "Media Object Index", { 0xFE, 0xB1, 0x03, 0xF8, 0x12, 0xAD, 0x4C, 0x64, 0x84, 0x0F, 0x2A, 0x1D, 0x2F, 0x7A, 0xD4, 0x8C }, asf_read_unknown, 1 }, |
|
997 |
- { "Timecode Index", { 0x3C, 0xB7, 0x3F, 0xD0, 0x0C, 0x4A, 0x48, 0x03, 0x95, 0x3D, 0xED, 0xF7, 0xB6, 0x22, 0x8F, 0x0C }, asf_read_unknown, 0 }, |
|
998 |
- { "Bitrate_Mutual_Exclusion", { 0xD6, 0xE2, 0x29, 0xDC, 0x35, 0xDA, 0x11, 0xD1, 0x90, 0x34, 0x00, 0xA0, 0xC9, 0x03, 0x49, 0xBE }, asf_read_unknown, 1 }, |
|
999 |
- { "Error Correction", { 0x75, 0xB2, 0x26, 0x35, 0x66, 0x8E, 0x11, 0xCF, 0xA6, 0xD9, 0x00, 0xAA, 0x00, 0x62, 0xCE, 0x6C }, asf_read_unknown, 1 }, |
|
1000 |
- { "Content Branding", { 0x22, 0x11, 0xB3, 0xFA, 0xBD, 0x23, 0x11, 0xD2, 0xB4, 0xB7, 0x00, 0xA0, 0xC9, 0x55, 0xFC, 0x6E }, asf_read_unknown, 1 }, |
|
1001 |
- { "Content Encryption", { 0x22, 0x11, 0xB3, 0xFB, 0xBD, 0x23, 0x11, 0xD2, 0xB4, 0xB7, 0x00, 0xA0, 0xC9, 0x55, 0xFC, 0x6E }, asf_read_unknown, 1 }, |
|
1002 |
- { "Extended Content Encryption", { 0x29, 0x8A, 0xE6, 0x14, 0x26, 0x22, 0x4C, 0x17, 0xB9, 0x35, 0xDA, 0xE0, 0x7E, 0xE9, 0x28, 0x9C }, asf_read_unknown, 1 }, |
|
1003 |
- { "Digital Signature", { 0x22, 0x11, 0xB3, 0xFC, 0xBD, 0x23, 0x11, 0xD2, 0xB4, 0xB7, 0x00, 0xA0, 0xC9, 0x55, 0xFC, 0x6E }, asf_read_unknown, 1 }, |
|
1004 |
- { "Extended Stream Properties", { 0x14, 0xE6, 0xA5, 0xCB, 0xC6, 0x72, 0x43, 0x32, 0x83, 0x99, 0xA9, 0x69, 0x52, 0x06, 0x5B, 0x5A }, asf_read_ext_stream_properties, 1 }, |
|
1005 |
- { "Advanced Mutual Exclusion", { 0xA0, 0x86, 0x49, 0xCF, 0x47, 0x75, 0x46, 0x70, 0x8A, 0x16, 0x6E, 0x35, 0x35, 0x75, 0x66, 0xCD }, asf_read_unknown, 1 }, |
|
1006 |
- { "Group Mutual Exclusion", { 0xD1, 0x46, 0x5A, 0x40, 0x5A, 0x79, 0x43, 0x38, 0xB7, 0x1B, 0xE3, 0x6B, 0x8F, 0xD6, 0xC2, 0x49 }, asf_read_unknown, 1}, |
|
1007 |
- { "Stream Prioritization", { 0xD4, 0xFE, 0xD1, 0x5B, 0x88, 0xD3, 0x45, 0x4F, 0x81, 0xF0, 0xED, 0x5C, 0x45, 0x99, 0x9E, 0x24 }, asf_read_unknown, 1 }, |
|
1008 |
- { "Bandwidth Sharing Object", { 0xA6, 0x96, 0x09, 0xE6, 0x51, 0x7B, 0x11, 0xD2, 0xB6, 0xAF, 0x00, 0xC0, 0x4F, 0xD9, 0x08, 0xE9 }, asf_read_unknown, 1 }, |
|
1009 |
- { "Metadata", { 0xC5, 0xF8, 0xCB, 0xEA, 0x5B, 0xAF, 0x48, 0x77, 0x84, 0x67, 0xAA, 0x8C, 0x44, 0xFA, 0x4C, 0xCA }, asf_read_metadata_obj, 1 }, |
|
1010 |
- { "Metadata Library", { 0x44, 0x23, 0x1C, 0x94, 0x94, 0x98, 0x49, 0xD1, 0xA1, 0x41, 0x1D, 0x13, 0x4E, 0x45, 0x70, 0x54 }, asf_read_metadata_obj, 1 }, |
|
1011 |
- { "Audio Spread", { 0xBF, 0xC3, 0xCD, 0x50, 0x61, 0x8F, 0x11, 0xCF, 0x8B, 0xB2, 0x00, 0xAA, 0x00, 0xB4, 0xE2, 0x20 }, asf_read_unknown, 1 }, |
|
1012 |
- { "Index Parameters", { 0xD6, 0xE2, 0x29, 0xDF, 0x35, 0xDA, 0x11, 0xD1, 0x90, 0x34, 0x00, 0xA0, 0xC9, 0x03, 0x49, 0xBE }, asf_read_unknown, 1 }, |
|
1013 |
- { "Content Encryption System Windows Media DRM Network Devices", |
|
1014 |
- { 0x7A, 0x07, 0x9B, 0xB6, 0xDA, 0XA4, 0x4e, 0x12, 0xA5, 0xCA, 0x91, 0xD3, 0x8D, 0xC1, 0x1A, 0x8D }, asf_read_unknown, 1 }, |
|
1015 |
- { "Mutex Language", { 0xD6, 0xE2, 0x2A, 0x00, 0x25, 0xDA, 0x11, 0xD1, 0x90, 0x34, 0x00, 0xA0, 0xC9, 0x03, 0x49, 0xBE }, asf_read_unknown, 1 }, |
|
1016 |
- { "Mutex Bitrate", { 0xD6, 0xE2, 0x2A, 0x01, 0x25, 0xDA, 0x11, 0xD1, 0x90, 0x34, 0x00, 0xA0, 0xC9, 0x03, 0x49, 0xBE }, asf_read_unknown, 1 }, |
|
1017 |
- { "Mutex Unknown", { 0xD6, 0xE2, 0x2A, 0x02, 0x25, 0xDA, 0x11, 0xD1, 0x90, 0x34, 0x00, 0xA0, 0xC9, 0x03, 0x49, 0xBE }, asf_read_unknown, 1 }, |
|
1018 |
- { "Bandwith Sharing Exclusive", { 0xAF, 0x60, 0x60, 0xAA, 0x51, 0x97, 0x11, 0xD2, 0xB6, 0xAF, 0x00, 0xC0, 0x4F, 0xD9, 0x08, 0xE9 }, asf_read_unknown, 1 }, |
|
1019 |
- { "Bandwith Sharing Partial", { 0xAF, 0x60, 0x60, 0xAB, 0x51, 0x97, 0x11, 0xD2, 0xB6, 0xAF, 0x00, 0xC0, 0x4F, 0xD9, 0x08, 0xE9 }, asf_read_unknown, 1 }, |
|
1020 |
- { "Payload Extension System Timecode", { 0x39, 0x95, 0x95, 0xEC, 0x86, 0x67, 0x4E, 0x2D, 0x8F, 0xDB, 0x98, 0x81, 0x4C, 0xE7, 0x6C, 0x1E }, asf_read_unknown, 1 }, |
|
1021 |
- { "Payload Extension System File Name", { 0xE1, 0x65, 0xEC, 0x0E, 0x19, 0xED, 0x45, 0xD7, 0xB4, 0xA7, 0x25, 0xCB, 0xD1, 0xE2, 0x8E, 0x9B }, asf_read_unknown, 1 }, |
|
1022 |
- { "Payload Extension System Content Type", { 0xD5, 0x90, 0xDC, 0x20, 0x07, 0xBC, 0x43, 0x6C, 0x9C, 0xF7, 0xF3, 0xBB, 0xFB, 0xF1, 0xA4, 0xDC }, asf_read_unknown, 1 }, |
|
1023 |
- { "Payload Extension System Pixel Aspect Ratio", { 0x1, 0x1E, 0xE5, 0x54, 0xF9, 0xEA, 0x4B, 0xC8, 0x82, 0x1A, 0x37, 0x6B, 0x74, 0xE4, 0xC4, 0xB8 }, asf_read_unknown, 1 }, |
|
1024 |
- { "Payload Extension System Sample Duration", { 0xC6, 0xBD, 0x94, 0x50, 0x86, 0x7F, 0x49, 0x07, 0x83, 0xA3, 0xC7, 0x79, 0x21, 0xB7, 0x33, 0xAD }, asf_read_unknown, 1 }, |
|
1025 |
- { "Payload Extension System Encryption Sample ID", { 0x66, 0x98, 0xB8, 0x4E, 0x0A, 0xFA, 0x43, 0x30, 0xAE, 0xB2, 0x1C, 0x0A, 0x98, 0xD7, 0xA4, 0x4D }, asf_read_unknown, 1 }, |
|
1026 |
- { "Payload Extension System Degradable JPEG", { 0x00, 0xE1, 0xAF, 0x06, 0x7B, 0xEC, 0x11, 0xD1, 0xA5, 0x82, 0x00, 0xC0, 0x4F, 0xC2, 0x9C, 0xFB }, asf_read_unknown, 1 }, |
|
1027 |
-}; |
|
1028 |
- |
|
1029 |
-#define READ_LEN(flag, name, len) \ |
|
1030 |
- do { \ |
|
1031 |
- if ((flag) == name ## IS_BYTE) \ |
|
1032 |
- len = avio_r8(pb); \ |
|
1033 |
- else if ((flag) == name ## IS_WORD) \ |
|
1034 |
- len = avio_rl16(pb); \ |
|
1035 |
- else if ((flag) == name ## IS_DWORD) \ |
|
1036 |
- len = avio_rl32(pb); \ |
|
1037 |
- else \ |
|
1038 |
- len = 0; \ |
|
1039 |
- } while(0) |
|
1040 |
- |
|
1041 |
-static int asf_read_subpayload(AVFormatContext *s, AVPacket *pkt, int is_header) |
|
1042 |
-{ |
|
1043 |
- ASFContext *asf = s->priv_data; |
|
1044 |
- AVIOContext *pb = s->pb; |
|
1045 |
- uint8_t sub_len; |
|
1046 |
- int ret, i; |
|
1047 |
- |
|
1048 |
- if (is_header) { |
|
1049 |
- asf->dts_delta = avio_r8(pb); |
|
1050 |
- if (asf->nb_mult_left) { |
|
1051 |
- asf->mult_sub_len = avio_rl16(pb); // total |
|
1052 |
- } |
|
1053 |
- asf->sub_header_offset = avio_tell(pb); |
|
1054 |
- asf->nb_sub = 0; |
|
1055 |
- asf->sub_left = 1; |
|
1056 |
- } |
|
1057 |
- sub_len = avio_r8(pb); |
|
1058 |
- if ((ret = av_get_packet(pb, pkt, sub_len)) < 0) // each subpayload is entire frame |
|
1059 |
- return ret; |
|
1060 |
- for (i = 0; i < asf->nb_streams; i++) { |
|
1061 |
- if (asf->stream_index == asf->asf_st[i]->stream_index) { |
|
1062 |
- pkt->stream_index = asf->asf_st[i]->index; |
|
1063 |
- break; |
|
1064 |
- } |
|
1065 |
- } |
|
1066 |
- asf->return_subpayload = 1; |
|
1067 |
- if (!sub_len) |
|
1068 |
- asf->return_subpayload = 0; |
|
1069 |
- |
|
1070 |
- if (sub_len) |
|
1071 |
- asf->nb_sub++; |
|
1072 |
- pkt->dts = asf->sub_dts + (asf->nb_sub - 1) * asf->dts_delta - asf->preroll; |
|
1073 |
- if (asf->nb_mult_left && (avio_tell(pb) >= |
|
1074 |
- (asf->sub_header_offset + asf->mult_sub_len))) { |
|
1075 |
- asf->sub_left = 0; |
|
1076 |
- asf->nb_mult_left--; |
|
1077 |
- } |
|
1078 |
- if (avio_tell(pb) >= asf->packet_offset + asf->packet_size - asf->pad_len) { |
|
1079 |
- asf->sub_left = 0; |
|
1080 |
- if (!asf->nb_mult_left) { |
|
1081 |
- avio_skip(pb, asf->pad_len); |
|
1082 |
- if (avio_tell(pb) != asf->packet_offset + asf->packet_size) { |
|
1083 |
- if (!asf->packet_size) |
|
1084 |
- return AVERROR_INVALIDDATA; |
|
1085 |
- av_log(s, AV_LOG_WARNING, |
|
1086 |
- "Position %"PRId64" wrong, should be %"PRId64"\n", |
|
1087 |
- avio_tell(pb), asf->packet_offset + asf->packet_size); |
|
1088 |
- avio_seek(pb, asf->packet_offset + asf->packet_size, SEEK_SET); |
|
1089 |
- } |
|
1090 |
- } |
|
1091 |
- } |
|
1092 |
- |
|
1093 |
- return 0; |
|
1094 |
-} |
|
1095 |
- |
|
1096 |
-static void reset_packet(ASFPacket *asf_pkt) |
|
1097 |
-{ |
|
1098 |
- asf_pkt->size_left = 0; |
|
1099 |
- asf_pkt->data_size = 0; |
|
1100 |
- asf_pkt->duration = 0; |
|
1101 |
- asf_pkt->flags = 0; |
|
1102 |
- asf_pkt->dts = 0; |
|
1103 |
- asf_pkt->duration = 0; |
|
1104 |
- av_free_packet(&asf_pkt->avpkt); |
|
1105 |
- av_init_packet(&asf_pkt->avpkt); |
|
1106 |
-} |
|
1107 |
- |
|
1108 |
-static int asf_read_multiple_payload(AVFormatContext *s, AVPacket *pkt, |
|
1109 |
- ASFPacket *asf_pkt) |
|
1110 |
-{ |
|
1111 |
- ASFContext *asf = s->priv_data; |
|
1112 |
- AVIOContext *pb = s->pb; |
|
1113 |
- uint16_t pay_len; |
|
1114 |
- unsigned char *p; |
|
1115 |
- int ret; |
|
1116 |
- int skip = 0; |
|
1117 |
- |
|
1118 |
- // if replicated lenght is 1, subpayloads are present |
|
1119 |
- if (asf->rep_data_len == 1) { |
|
1120 |
- asf->sub_left = 1; |
|
1121 |
- asf->state = READ_MULTI_SUB; |
|
1122 |
- pkt->flags = asf_pkt->flags; |
|
1123 |
- if ((ret = asf_read_subpayload(s, pkt, 1)) < 0) |
|
1124 |
- return ret; |
|
1125 |
- } else { |
|
1126 |
- if (!asf_pkt->data_size) { |
|
1127 |
- asf_pkt->data_size = asf_pkt->size_left = avio_rl32(pb); // read media object size |
|
1128 |
- if (asf_pkt->data_size <= 0) |
|
1129 |
- return AVERROR_EOF; |
|
1130 |
- if ((ret = av_new_packet(&asf_pkt->avpkt, asf_pkt->data_size)) < 0) |
|
1131 |
- return ret; |
|
1132 |
- } else |
|
1133 |
- avio_skip(pb, 4); // reading of media object size is already done |
|
1134 |
- asf_pkt->dts = avio_rl32(pb); // read presentation time |
|
1135 |
- if ((asf->rep_data_len - 8) > 0) |
|
1136 |
- avio_skip(pb, asf->rep_data_len - 8); // skip replicated data |
|
1137 |
- pay_len = avio_rl16(pb); // payload length should be WORD |
|
1138 |
- if (pay_len > asf->packet_size) { |
|
1139 |
- av_log(s, AV_LOG_ERROR, |
|
1140 |
- "Error: invalid data packet size, pay_len %"PRIu16", " |
|
1141 |
- "asf->packet_size %"PRIu32", offset %"PRId64".\n", |
|
1142 |
- pay_len, asf->packet_size, avio_tell(pb)); |
|
1143 |
- return AVERROR_INVALIDDATA; |
|
1144 |
- } |
|
1145 |
- p = asf_pkt->avpkt.data + asf_pkt->data_size - asf_pkt->size_left; |
|
1146 |
- if (pay_len > asf_pkt->size_left) { |
|
1147 |
- av_log(s, AV_LOG_ERROR, |
|
1148 |
- "Error: invalid buffer size, pay_len %d, data size left %d.\n", |
|
1149 |
- pay_len, asf_pkt->size_left); |
|
1150 |
- skip = pay_len - asf_pkt->size_left; |
|
1151 |
- pay_len = asf_pkt->size_left; |
|
1152 |
- } |
|
1153 |
- if ((ret = avio_read(pb, p, pay_len)) < 0) |
|
1154 |
- return ret; |
|
1155 |
- if (s->key && s->keylen == 20) |
|
1156 |
- ff_asfcrypt_dec(s->key, p, ret); |
|
1157 |
- avio_skip(pb, skip); |
|
1158 |
- asf_pkt->size_left -= pay_len; |
|
1159 |
- asf->nb_mult_left--; |
|
1160 |
- } |
|
1161 |
- |
|
1162 |
- return 0; |
|
1163 |
-} |
|
1164 |
- |
|
1165 |
-static int asf_read_single_payload(AVFormatContext *s, AVPacket *pkt, |
|
1166 |
- ASFPacket *asf_pkt) |
|
1167 |
-{ |
|
1168 |
- ASFContext *asf = s->priv_data; |
|
1169 |
- AVIOContext *pb = s->pb; |
|
1170 |
- int64_t offset; |
|
1171 |
- uint64_t size; |
|
1172 |
- unsigned char *p; |
|
1173 |
- int ret; |
|
1174 |
- |
|
1175 |
- if (!asf_pkt->data_size) { |
|
1176 |
- asf_pkt->data_size = asf_pkt->size_left = avio_rl32(pb); // read media object size |
|
1177 |
- if (asf_pkt->data_size <= 0) |
|
1178 |
- return AVERROR_EOF; |
|
1179 |
- if ((ret = av_new_packet(&asf_pkt->avpkt, asf_pkt->data_size)) < 0) |
|
1180 |
- return ret; |
|
1181 |
- } else |
|
1182 |
- avio_skip(pb, 4); // skip media object size |
|
1183 |
- asf_pkt->dts = avio_rl32(pb); // read presentation time |
|
1184 |
- if ((asf->rep_data_len - 8) > 0) |
|
1185 |
- avio_skip(pb, asf->rep_data_len - 8); // skip replicated data |
|
1186 |
- offset = avio_tell(pb); |
|
1187 |
- |
|
1188 |
- // size of the payload - size of the packet without header and padding |
|
1189 |
- if (asf->packet_size_internal) |
|
1190 |
- size = asf->packet_size_internal - offset + asf->packet_offset - asf->pad_len; |
|
1191 |
- else |
|
1192 |
- size = asf->packet_size - offset + asf->packet_offset - asf->pad_len; |
|
1193 |
- if (size > asf->packet_size) { |
|
1194 |
- av_log(s, AV_LOG_ERROR, |
|
1195 |
- "Error: invalid data packet size, offset %"PRId64".\n", |
|
1196 |
- avio_tell(pb)); |
|
1197 |
- return AVERROR_INVALIDDATA; |
|
1198 |
- } |
|
1199 |
- p = asf_pkt->avpkt.data + asf_pkt->data_size - asf_pkt->size_left; |
|
1200 |
- if (size > asf_pkt->size_left) |
|
1201 |
- return AVERROR_INVALIDDATA; |
|
1202 |
- if (asf_pkt->size_left > size) |
|
1203 |
- asf_pkt->size_left -= size; |
|
1204 |
- else |
|
1205 |
- asf_pkt->size_left = 0; |
|
1206 |
- if ((ret = avio_read(pb, p, size)) < 0) |
|
1207 |
- return ret; |
|
1208 |
- if (s->key && s->keylen == 20) |
|
1209 |
- ff_asfcrypt_dec(s->key, p, ret); |
|
1210 |
- if (asf->packet_size_internal) |
|
1211 |
- avio_skip(pb, asf->packet_size - asf->packet_size_internal); |
|
1212 |
- avio_skip(pb, asf->pad_len); // skip padding |
|
1213 |
- |
|
1214 |
- return 0; |
|
1215 |
-} |
|
1216 |
- |
|
1217 |
-static int asf_read_payload(AVFormatContext *s, AVPacket *pkt) |
|
1218 |
-{ |
|
1219 |
- ASFContext *asf = s->priv_data; |
|
1220 |
- AVIOContext *pb = s->pb; |
|
1221 |
- int ret, i; |
|
1222 |
- ASFPacket *asf_pkt = NULL; |
|
1223 |
- |
|
1224 |
- if (!asf->sub_left) { |
|
1225 |
- uint32_t off_len, media_len; |
|
1226 |
- uint8_t stream_num; |
|
1227 |
- |
|
1228 |
- stream_num = avio_r8(pb); |
|
1229 |
- asf->stream_index = stream_num & ASF_STREAM_NUM; |
|
1230 |
- for (i = 0; i < asf->nb_streams; i++) { |
|
1231 |
- if (asf->stream_index == asf->asf_st[i]->stream_index) { |
|
1232 |
- asf_pkt = &asf->asf_st[i]->pkt; |
|
1233 |
- asf_pkt->stream_index = asf->asf_st[i]->index; |
|
1234 |
- asf_pkt->dts = asf->dts; |
|
1235 |
- break; |
|
1236 |
- } |
|
1237 |
- } |
|
1238 |
- if (!asf_pkt) |
|
1239 |
- return AVERROR_INVALIDDATA; |
|
1240 |
- if (stream_num >> 7) |
|
1241 |
- asf_pkt->flags |= AV_PKT_FLAG_KEY; |
|
1242 |
- READ_LEN(asf->prop_flags & ASF_PL_MASK_MEDIA_OBJECT_NUMBER_LENGTH_FIELD_SIZE, |
|
1243 |
- ASF_PL_FLAG_MEDIA_OBJECT_NUMBER_LENGTH_FIELD_, media_len); |
|
1244 |
- READ_LEN(asf->prop_flags & ASF_PL_MASK_OFFSET_INTO_MEDIA_OBJECT_LENGTH_FIELD_SIZE, |
|
1245 |
- ASF_PL_FLAG_OFFSET_INTO_MEDIA_OBJECT_LENGTH_FIELD_, off_len); |
|
1246 |
- READ_LEN(asf->prop_flags & ASF_PL_MASK_REPLICATED_DATA_LENGTH_FIELD_SIZE, |
|
1247 |
- ASF_PL_FLAG_REPLICATED_DATA_LENGTH_FIELD_, asf->rep_data_len); |
|
1248 |
- if (asf_pkt->size_left && (asf_pkt->frame_num != media_len)) { |
|
1249 |
- av_log(s, AV_LOG_WARNING, "Unfinished frame will be ignored\n"); |
|
1250 |
- reset_packet(asf_pkt); |
|
1251 |
- } |
|
1252 |
- asf_pkt->frame_num = media_len; |
|
1253 |
- asf->sub_dts = off_len; |
|
1254 |
- if (asf->nb_mult_left) { |
|
1255 |
- if ((ret = asf_read_multiple_payload(s, pkt, asf_pkt)) < 0) |
|
1256 |
- return ret; |
|
1257 |
- } else if (asf->rep_data_len == 1) { |
|
1258 |
- asf->sub_left = 1; |
|
1259 |
- asf->state = READ_SINGLE; |
|
1260 |
- pkt->flags = asf_pkt->flags; |
|
1261 |
- if ((ret = asf_read_subpayload(s, pkt, 1)) < 0) |
|
1262 |
- return ret; |
|
1263 |
- } else { |
|
1264 |
- if ((ret = asf_read_single_payload(s, pkt, asf_pkt)) < 0) |
|
1265 |
- return ret; |
|
1266 |
- } |
|
1267 |
- } else { |
|
1268 |
- for (i = 0; i <= asf->nb_streams; i++) { |
|
1269 |
- if (asf->stream_index == asf->asf_st[i]->stream_index) { |
|
1270 |
- asf_pkt = &asf->asf_st[i]->pkt; |
|
1271 |
- break; |
|
1272 |
- } |
|
1273 |
- } |
|
1274 |
- if (!asf_pkt) |
|
1275 |
- return AVERROR_INVALIDDATA; |
|
1276 |
- pkt->flags = asf_pkt->flags; |
|
1277 |
- pkt->dts = asf_pkt->dts; |
|
1278 |
- pkt->stream_index = asf->asf_st[i]->index; |
|
1279 |
- if ((ret = asf_read_subpayload(s, pkt, 0)) < 0) // read subpayload without its header |
|
1280 |
- return ret; |
|
1281 |
- } |
|
1282 |
- |
|
1283 |
- return 0; |
|
1284 |
-} |
|
1285 |
- |
|
1286 |
-static int asf_read_packet_header(AVFormatContext *s) |
|
1287 |
-{ |
|
1288 |
- ASFContext *asf = s->priv_data; |
|
1289 |
- AVIOContext *pb = s->pb; |
|
1290 |
- uint64_t size; |
|
1291 |
- uint32_t av_unused seq; |
|
1292 |
- unsigned char error_flags, len_flags, pay_flags; |
|
1293 |
- |
|
1294 |
- asf->packet_offset = avio_tell(pb); |
|
1295 |
- error_flags = avio_r8(pb); // read Error Correction Flags |
|
1296 |
- if (error_flags & ASF_PACKET_FLAG_ERROR_CORRECTION_PRESENT) |
|
1297 |
- if (!(error_flags & ASF_ERROR_CORRECTION_LENGTH_TYPE)) { |
|
1298 |
- size = error_flags & ASF_PACKET_ERROR_CORRECTION_DATA_SIZE; |
|
1299 |
- avio_skip(pb, size); |
|
1300 |
- } |
|
1301 |
- len_flags = avio_r8(pb); |
|
1302 |
- asf->prop_flags = avio_r8(pb); |
|
1303 |
- READ_LEN(len_flags & ASF_PPI_MASK_PACKET_LENGTH_FIELD_SIZE, |
|
1304 |
- ASF_PPI_FLAG_PACKET_LENGTH_FIELD_, asf->packet_size_internal); |
|
1305 |
- READ_LEN(len_flags & ASF_PPI_MASK_SEQUENCE_FIELD_SIZE, |
|
1306 |
- ASF_PPI_FLAG_SEQUENCE_FIELD_, seq); |
|
1307 |
- READ_LEN(len_flags & ASF_PPI_MASK_PADDING_LENGTH_FIELD_SIZE, |
|
1308 |
- ASF_PPI_FLAG_PADDING_LENGTH_FIELD_, asf->pad_len ); |
|
1309 |
- asf->send_time = avio_rl32(pb); // send time |
|
1310 |
- avio_skip(pb, 2); // skip duration |
|
1311 |
- if (len_flags & ASF_PPI_FLAG_MULTIPLE_PAYLOADS_PRESENT) { // Multiple Payloads present |
|
1312 |
- pay_flags = avio_r8(pb); |
|
1313 |
- asf->nb_mult_left = (pay_flags & ASF_NUM_OF_PAYLOADS); |
|
1314 |
- } |
|
1315 |
- |
|
1316 |
- return 0; |
|
1317 |
-} |
|
1318 |
- |
|
1319 |
-static int asf_deinterleave(AVFormatContext *s, ASFPacket *asf_pkt, int st_num) |
|
1320 |
-{ |
|
1321 |
- ASFContext *asf = s->priv_data; |
|
1322 |
- ASFStream *asf_st = asf->asf_st[st_num]; |
|
1323 |
- unsigned char *p = asf_pkt->avpkt.data; |
|
1324 |
- uint16_t pkt_len = asf->asf_st[st_num]->virtual_pkt_len; |
|
1325 |
- uint16_t chunk_len = asf->asf_st[st_num]->virtual_chunk_len; |
|
1326 |
- int nchunks = pkt_len / chunk_len; |
|
1327 |
- AVPacket pkt; |
|
1328 |
- int pos = 0, j, l, ret; |
|
1329 |
- |
|
1330 |
- |
|
1331 |
- if ((ret = av_new_packet(&pkt, asf_pkt->data_size)) < 0) |
|
1332 |
- return ret; |
|
1333 |
- |
|
1334 |
- while (asf_pkt->data_size >= asf_st->span * pkt_len + pos) { |
|
1335 |
- if (pos >= asf_pkt->data_size) { |
|
1336 |
- break; |
|
1337 |
- } |
|
1338 |
- for (l = 0; l < pkt_len; l++) { |
|
1339 |
- if (pos >= asf_pkt->data_size) { |
|
1340 |
- break; |
|
1341 |
- } |
|
1342 |
- for (j = 0; j < asf_st->span; j++) { |
|
1343 |
- if ((pos + chunk_len) >= asf_pkt->data_size) |
|
1344 |
- break; |
|
1345 |
- memcpy(pkt.data + pos, |
|
1346 |
- p + (j * nchunks + l) * chunk_len, |
|
1347 |
- chunk_len); |
|
1348 |
- pos += chunk_len; |
|
1349 |
- } |
|
1350 |
- } |
|
1351 |
- p += asf_st->span * pkt_len; |
|
1352 |
- if (p > asf_pkt->avpkt.data + asf_pkt->data_size) |
|
1353 |
- break; |
|
1354 |
- } |
|
1355 |
- av_free_packet(&asf_pkt->avpkt); |
|
1356 |
- asf_pkt->avpkt = pkt; |
|
1357 |
- |
|
1358 |
- return 0; |
|
1359 |
-} |
|
1360 |
- |
|
1361 |
-static int asf_read_packet(AVFormatContext *s, AVPacket *pkt) |
|
1362 |
-{ |
|
1363 |
- ASFContext *asf = s->priv_data; |
|
1364 |
- AVIOContext *pb = s->pb; |
|
1365 |
- int ret, i; |
|
1366 |
- |
|
1367 |
- if ((avio_tell(pb) >= asf->data_offset + asf->data_size) && |
|
1368 |
- !(asf->b_flags & ASF_FLAG_BROADCAST)) |
|
1369 |
- return AVERROR_EOF; |
|
1370 |
- while (!pb->eof_reached) { |
|
1371 |
- if (asf->state == PARSE_PACKET_HEADER) { |
|
1372 |
- asf_read_packet_header(s); |
|
1373 |
- if (!asf->nb_mult_left) |
|
1374 |
- asf->state = READ_SINGLE; |
|
1375 |
- else |
|
1376 |
- asf->state = READ_MULTI; |
|
1377 |
- } |
|
1378 |
- if ((ret = asf_read_payload(s, pkt)) < 0) |
|
1379 |
- return ret; |
|
1380 |
- switch (asf->state) { |
|
1381 |
- case READ_SINGLE: |
|
1382 |
- if (!asf->sub_left) |
|
1383 |
- asf->state = PARSE_PACKET_HEADER; |
|
1384 |
- break; |
|
1385 |
- case READ_MULTI_SUB: |
|
1386 |
- if (!asf->sub_left && !asf->nb_mult_left) { |
|
1387 |
- asf->state = PARSE_PACKET_HEADER; |
|
1388 |
- if (!asf->return_subpayload) |
|
1389 |
- avio_skip(pb, asf->pad_len); // skip padding |
|
1390 |
- if (asf->packet_offset + asf->packet_size > avio_tell(pb)) |
|
1391 |
- avio_seek(pb, asf->packet_offset + asf->packet_size, SEEK_SET); |
|
1392 |
- } else if (!asf->sub_left) |
|
1393 |
- asf->state = READ_MULTI; |
|
1394 |
- break; |
|
1395 |
- case READ_MULTI: |
|
1396 |
- if (!asf->nb_mult_left) { |
|
1397 |
- asf->state = PARSE_PACKET_HEADER; |
|
1398 |
- if (!asf->return_subpayload) { |
|
1399 |
- avio_skip(pb, asf->pad_len); // skip padding |
|
1400 |
- } |
|
1401 |
- if (asf->packet_offset + asf->packet_size > avio_tell(pb)) |
|
1402 |
- avio_seek(pb, asf->packet_offset + asf->packet_size, SEEK_SET); |
|
1403 |
- } |
|
1404 |
- break; |
|
1405 |
- } |
|
1406 |
- if (asf->return_subpayload) { |
|
1407 |
- asf->return_subpayload = 0; |
|
1408 |
- return 0; |
|
1409 |
- } |
|
1410 |
- for (i = 0; i < s->nb_streams; i++) { |
|
1411 |
- ASFPacket *asf_pkt = &asf->asf_st[i]->pkt; |
|
1412 |
- if (asf_pkt && !asf_pkt->size_left && asf_pkt->data_size) { |
|
1413 |
- if (asf->asf_st[i]->span > 1 && |
|
1414 |
- asf->asf_st[i]->type == AVMEDIA_TYPE_AUDIO) |
|
1415 |
- if ((ret = asf_deinterleave(s, asf_pkt, i)) < 0) |
|
1416 |
- return ret; |
|
1417 |
- av_packet_move_ref(pkt, &asf_pkt->avpkt); |
|
1418 |
- pkt->stream_index = asf->asf_st[i]->index; |
|
1419 |
- pkt->flags = asf_pkt->flags; |
|
1420 |
- pkt->dts = asf_pkt->dts - asf->preroll; |
|
1421 |
- asf_pkt->data_size = 0; |
|
1422 |
- asf_pkt->frame_num = 0; |
|
1423 |
- return 0; |
|
1424 |
- } |
|
1425 |
- } |
|
1426 |
- } |
|
1427 |
- |
|
1428 |
- if (pb->eof_reached) |
|
1429 |
- return AVERROR_EOF; |
|
1430 |
- |
|
1431 |
- return 0; |
|
1432 |
-} |
|
1433 |
- |
|
1434 |
-static int asf_read_close(AVFormatContext *s) |
|
1435 |
-{ |
|
1436 |
- ASFContext *asf = s->priv_data; |
|
1437 |
- int i; |
|
1438 |
- |
|
1439 |
- for (i = 0; i < asf->nb_streams; i++) { |
|
1440 |
- av_free_packet(&asf->asf_st[i]->pkt.avpkt); |
|
1441 |
- av_freep(&asf->asf_st[i]); |
|
1442 |
- av_dict_free(&asf->asf_sd[i].asf_met); |
|
1443 |
- } |
|
1444 |
- |
|
1445 |
- return 0; |
|
1446 |
-} |
|
1447 |
- |
|
1448 |
-static void reset_packet_state(AVFormatContext *s) |
|
1449 |
-{ |
|
1450 |
- ASFContext *asf = s->priv_data; |
|
1451 |
- int i; |
|
1452 |
- |
|
1453 |
- asf->state = PARSE_PACKET_HEADER; |
|
1454 |
- asf->offset = 0; |
|
1455 |
- asf->return_subpayload = 0; |
|
1456 |
- asf->sub_left = 0; |
|
1457 |
- asf->sub_header_offset = 0; |
|
1458 |
- asf->packet_offset = asf->first_packet_offset; |
|
1459 |
- asf->pad_len = 0; |
|
1460 |
- asf->rep_data_len = 0; |
|
1461 |
- asf->dts_delta = 0; |
|
1462 |
- asf->mult_sub_len = 0; |
|
1463 |
- asf->nb_mult_left = 0; |
|
1464 |
- asf->nb_sub = 0; |
|
1465 |
- asf->prop_flags = 0; |
|
1466 |
- asf->sub_dts = 0; |
|
1467 |
- asf->dts = 0; |
|
1468 |
- for (i = 0; i < asf->nb_streams; i++) { |
|
1469 |
- ASFPacket *pkt = &asf->asf_st[i]->pkt; |
|
1470 |
- pkt->size_left = 0; |
|
1471 |
- pkt->data_size = 0; |
|
1472 |
- pkt->duration = 0; |
|
1473 |
- pkt->flags = 0; |
|
1474 |
- pkt->dts = 0; |
|
1475 |
- pkt->duration = 0; |
|
1476 |
- av_free_packet(&pkt->avpkt); |
|
1477 |
- av_init_packet(&pkt->avpkt); |
|
1478 |
- } |
|
1479 |
-} |
|
1480 |
- |
|
1481 |
-/* |
|
1482 |
- * Find a timestamp for the requested position within the payload |
|
1483 |
- * where the pos (position) is the offset inside the Data Object. |
|
1484 |
- * When position is not on the packet boundary, asf_read_timestamp tries |
|
1485 |
- * to find the closest packet offset after this position. If this packet |
|
1486 |
- * is a key frame, this packet timestamp is read and an index entry is created |
|
1487 |
- * for the packet. If this packet belongs to the requested stream, |
|
1488 |
- * asf_read_timestamp upgrades pos to the packet beginning offset and |
|
1489 |
- * returns this packet's dts. So returned dts is the dts of the first key frame with |
|
1490 |
- * matching stream number after given position. |
|
1491 |
- */ |
|
1492 |
-static int64_t asf_read_timestamp(AVFormatContext *s, int stream_index, |
|
1493 |
- int64_t *pos, int64_t pos_limit) |
|
1494 |
-{ |
|
1495 |
- ASFContext *asf = s->priv_data; |
|
1496 |
- int64_t pkt_pos = *pos, pkt_offset, dts = AV_NOPTS_VALUE, data_end; |
|
1497 |
- AVPacket pkt; |
|
1498 |
- int n; |
|
1499 |
- |
|
1500 |
- data_end = asf->data_offset + asf->data_size; |
|
1501 |
- |
|
1502 |
- n = (pkt_pos - asf->first_packet_offset + asf->packet_size - 1) / |
|
1503 |
- asf->packet_size; |
|
1504 |
- n = av_clip(n, 0, ((data_end - asf->first_packet_offset) / asf->packet_size - 1)); |
|
1505 |
- pkt_pos = asf->first_packet_offset + n * asf->packet_size; |
|
1506 |
- |
|
1507 |
- avio_seek(s->pb, pkt_pos, SEEK_SET); |
|
1508 |
- pkt_offset = pkt_pos; |
|
1509 |
- |
|
1510 |
- reset_packet_state(s); |
|
1511 |
- while (avio_tell(s->pb) < data_end) { |
|
1512 |
- |
|
1513 |
- int i, ret, st_found; |
|
1514 |
- |
|
1515 |
- av_init_packet(&pkt); |
|
1516 |
- pkt_offset = avio_tell(s->pb); |
|
1517 |
- if ((ret = asf_read_packet(s, &pkt)) < 0) { |
|
1518 |
- dts = AV_NOPTS_VALUE; |
|
1519 |
- return ret; |
|
1520 |
- } |
|
1521 |
- // ASFPacket may contain fragments of packets belonging to different streams, |
|
1522 |
- // pkt_offset is the offset of the first fragment within it. |
|
1523 |
- if ((pkt_offset >= (pkt_pos + asf->packet_size))) |
|
1524 |
- pkt_pos += asf->packet_size; |
|
1525 |
- for (i = 0; i < asf->nb_streams; i++) { |
|
1526 |
- ASFStream *st = asf->asf_st[i]; |
|
1527 |
- |
|
1528 |
- st_found = 0; |
|
1529 |
- if (pkt.flags & AV_PKT_FLAG_KEY) { |
|
1530 |
- dts = pkt.dts; |
|
1531 |
- if (dts) { |
|
1532 |
- av_add_index_entry(s->streams[pkt.stream_index], pkt_pos, |
|
1533 |
- dts, pkt.size, 0, AVINDEX_KEYFRAME); |
|
1534 |
- if (stream_index == st->index) { |
|
1535 |
- st_found = 1; |
|
1536 |
- break; |
|
1537 |
- } |
|
1538 |
- } |
|
1539 |
- } |
|
1540 |
- } |
|
1541 |
- if (st_found) |
|
1542 |
- break; |
|
1543 |
- av_free_packet(&pkt); |
|
1544 |
- } |
|
1545 |
- *pos = pkt_pos; |
|
1546 |
- |
|
1547 |
- av_free_packet(&pkt); |
|
1548 |
- return dts; |
|
1549 |
-} |
|
1550 |
- |
|
1551 |
-static int asf_read_seek(AVFormatContext *s, int stream_index, |
|
1552 |
- int64_t timestamp, int flags) |
|
1553 |
-{ |
|
1554 |
- ASFContext *asf = s->priv_data; |
|
1555 |
- int idx, ret; |
|
1556 |
- |
|
1557 |
- if (s->streams[stream_index]->nb_index_entries && asf->is_simple_index) { |
|
1558 |
- idx = av_index_search_timestamp(s->streams[stream_index], timestamp, flags); |
|
1559 |
- if (idx < 0 || idx >= s->streams[stream_index]->nb_index_entries) |
|
1560 |
- return AVERROR_INVALIDDATA; |
|
1561 |
- avio_seek(s->pb, s->streams[stream_index]->index_entries[idx].pos, SEEK_SET); |
|
1562 |
- } else { |
|
1563 |
- if ((ret = ff_seek_frame_binary(s, stream_index, timestamp, flags)) < 0) |
|
1564 |
- return ret; |
|
1565 |
- |
|
1566 |
- // asf_read_timestamp is called inside ff_seek_frame_binary and leaves state dirty, |
|
1567 |
- // so reset_packet_state have to be called after it. |
|
1568 |
- reset_packet_state(s); |
|
1569 |
- } |
|
1570 |
- |
|
1571 |
- return 0; |
|
1572 |
-} |
|
1573 |
- |
|
1574 |
-static const GUIDParseTable *find_guid(ff_asf_guid guid) |
|
1575 |
-{ |
|
1576 |
- int j, ret; |
|
1577 |
- const GUIDParseTable *g; |
|
1578 |
- |
|
1579 |
- swap_guid(guid); |
|
1580 |
- g = gdef; |
|
1581 |
- for (j = 0; j < FF_ARRAY_ELEMS(gdef); j++) { |
|
1582 |
- if (!(ret = memcmp(guid, g->guid, sizeof(g->guid)))) |
|
1583 |
- return g; |
|
1584 |
- g++; |
|
1585 |
- } |
|
1586 |
- |
|
1587 |
- return NULL; |
|
1588 |
-} |
|
1589 |
- |
|
1590 |
-static int detect_unknown_subobject(AVFormatContext *s, int64_t offset, int64_t size) |
|
1591 |
-{ |
|
1592 |
- ASFContext *asf = s->priv_data; |
|
1593 |
- AVIOContext *pb = s->pb; |
|
1594 |
- const GUIDParseTable *g = NULL; |
|
1595 |
- ff_asf_guid guid; |
|
1596 |
- int ret; |
|
1597 |
- |
|
1598 |
- while (avio_tell(pb) <= offset + size) { |
|
1599 |
- asf->offset = avio_tell(pb); |
|
1600 |
- if ((ret = ff_get_guid(pb, &guid)) < 0) |
|
1601 |
- return ret; |
|
1602 |
- g = find_guid(guid); |
|
1603 |
- if (g) { |
|
1604 |
- if ((ret = g->read_object(s, g)) < 0) |
|
1605 |
- return ret; |
|
1606 |
- } else { |
|
1607 |
- GUIDParseTable g2; |
|
1608 |
- |
|
1609 |
- g2.name = "Unknown"; |
|
1610 |
- g2.is_subobject = 1; |
|
1611 |
- asf_read_unknown(s, &g2); |
|
1612 |
- } |
|
1613 |
- } |
|
1614 |
- |
|
1615 |
- return 0; |
|
1616 |
-} |
|
1617 |
- |
|
1618 |
-static int asf_read_header(AVFormatContext *s) |
|
1619 |
-{ |
|
1620 |
- ASFContext *asf = s->priv_data; |
|
1621 |
- AVIOContext *pb = s->pb; |
|
1622 |
- const GUIDParseTable *g = NULL; |
|
1623 |
- ff_asf_guid guid; |
|
1624 |
- int i, ret; |
|
1625 |
- uint64_t size; |
|
1626 |
- |
|
1627 |
- asf->preroll = 0; |
|
1628 |
- asf->is_simple_index = 0; |
|
1629 |
- ff_get_guid(pb, &guid); |
|
1630 |
- if (ff_guidcmp(&guid, &ff_asf_header)) |
|
1631 |
- return AVERROR_INVALIDDATA; |
|
1632 |
- avio_skip(pb, 8); // skip header object size |
|
1633 |
- avio_skip(pb, 6); // skip number of header objects and 2 reserved bytes |
|
1634 |
- asf->data_reached = 0; |
|
1635 |
- |
|
1636 |
- /* 1 is here instead of pb->eof_reached because (when not streaming), Data are skipped |
|
1637 |
- * for the first time, |
|
1638 |
- * Index object is processed and got eof and then seeking back to the Data is performed. |
|
1639 |
- */ |
|
1640 |
- while (1) { |
|
1641 |
- // for the cases when object size is invalid |
|
1642 |
- if (avio_tell(pb) == asf->offset) { |
|
1643 |
- if (asf->data_reached) |
|
1644 |
- avio_seek(pb, asf->first_packet_offset, SEEK_SET); |
|
1645 |
- break; |
|
1646 |
- } |
|
1647 |
- asf->offset = avio_tell(pb); |
|
1648 |
- if ((ret = ff_get_guid(pb, &guid)) < 0) { |
|
1649 |
- if (ret == AVERROR_EOF && asf->data_reached) { |
|
1650 |
- avio_seek(pb, asf->first_packet_offset, SEEK_SET); |
|
1651 |
- break; |
|
1652 |
- } else |
|
1653 |
- return ret; |
|
1654 |
- } |
|
1655 |
- g = find_guid(guid); |
|
1656 |
- if (g) { |
|
1657 |
- asf->unknown_offset = asf->offset; |
|
1658 |
- asf->is_header = 1; |
|
1659 |
- if ((ret = g->read_object(s, g)) < 0) |
|
1660 |
- return ret; |
|
1661 |
- } else { |
|
1662 |
- size = avio_rl64(pb); |
|
1663 |
- align_position(pb, asf->offset, size); |
|
1664 |
- } |
|
1665 |
- if (asf->data_reached && !pb->seekable) |
|
1666 |
- break; |
|
1667 |
- } |
|
1668 |
- |
|
1669 |
- for (i = 0; i < asf->nb_streams; i++) { |
|
1670 |
- const char *rfc1766 = asf->asf_sd[asf->asf_st[i]->lang_idx].langs; |
|
1671 |
- AVStream *st = s->streams[asf->asf_st[i]->index]; |
|
1672 |
- set_language(s, rfc1766, &st->metadata); |
|
1673 |
- } |
|
1674 |
- |
|
1675 |
- for (i = 0; i < ASF_MAX_STREAMS; i++) { |
|
1676 |
- AVStream *st = NULL; |
|
1677 |
- |
|
1678 |
- st = find_stream(s, i); |
|
1679 |
- if (st) { |
|
1680 |
- av_dict_copy(&st->metadata, asf->asf_sd[i].asf_met, AV_DICT_IGNORE_SUFFIX); |
|
1681 |
- if (asf->asf_sd[i].aspect_ratio.num > 0 && asf->asf_sd[i].aspect_ratio.den > 0) { |
|
1682 |
- st->sample_aspect_ratio.num = asf->asf_sd[i].aspect_ratio.num; |
|
1683 |
- st->sample_aspect_ratio.den = asf->asf_sd[i].aspect_ratio.den; |
|
1684 |
- } |
|
1685 |
- } |
|
1686 |
- } |
|
1687 |
- |
|
1688 |
- return 0; |
|
1689 |
-} |
|
1690 |
- |
|
1691 |
-AVInputFormat ff_asf_o_demuxer = { |
|
1692 |
- .name = "asf_o", |
|
1693 |
- .long_name = NULL_IF_CONFIG_SMALL("ASF (Advanced / Active Streaming Format)"), |
|
1694 |
- .priv_data_size = sizeof(ASFContext), |
|
1695 |
- .read_probe = asf_probe, |
|
1696 |
- .read_header = asf_read_header, |
|
1697 |
- .read_packet = asf_read_packet, |
|
1698 |
- .read_close = asf_read_close, |
|
1699 |
- .read_timestamp = asf_read_timestamp, |
|
1700 |
- .read_seek = asf_read_seek, |
|
1701 |
- .flags = AVFMT_NOBINSEARCH | AVFMT_NOGENSEARCH, |
|
1702 |
-}; |
1703 | 1 |
new file mode 100644 |
... | ... |
@@ -0,0 +1,1702 @@ |
0 |
+/* |
|
1 |
+ * Microsoft Advanced Streaming Format demuxer |
|
2 |
+ * Copyright (c) 2014 Alexandra Hájková |
|
3 |
+ * |
|
4 |
+ * This file is part of FFmpeg. |
|
5 |
+ * |
|
6 |
+ * FFmpeg is free software; you can redistribute it and/or |
|
7 |
+ * modify it under the terms of the GNU Lesser General Public |
|
8 |
+ * License as published by the Free Software Foundation; either |
|
9 |
+ * version 2.1 of the License, or (at your option) any later version. |
|
10 |
+ * |
|
11 |
+ * FFmpeg is distributed in the hope that it will be useful, |
|
12 |
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of |
|
13 |
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
|
14 |
+ * Lesser General Public License for more details. |
|
15 |
+ * |
|
16 |
+ * You should have received a copy of the GNU Lesser General Public |
|
17 |
+ * License along with FFmpeg; if not, write to the Free Software |
|
18 |
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
|
19 |
+ */ |
|
20 |
+ |
|
21 |
+#include "libavutil/attributes.h" |
|
22 |
+#include "libavutil/avassert.h" |
|
23 |
+#include "libavutil/avstring.h" |
|
24 |
+#include "libavutil/bswap.h" |
|
25 |
+#include "libavutil/common.h" |
|
26 |
+#include "libavutil/dict.h" |
|
27 |
+#include "libavutil/internal.h" |
|
28 |
+#include "libavutil/mathematics.h" |
|
29 |
+#include "libavutil/opt.h" |
|
30 |
+#include "libavutil/time_internal.h" |
|
31 |
+ |
|
32 |
+#include "avformat.h" |
|
33 |
+#include "avio_internal.h" |
|
34 |
+#include "avlanguage.h" |
|
35 |
+#include "id3v2.h" |
|
36 |
+#include "internal.h" |
|
37 |
+#include "riff.h" |
|
38 |
+#include "asf.h" |
|
39 |
+#include "asfcrypt.h" |
|
40 |
+ |
|
41 |
+#define ASF_BOOL 0x2 |
|
42 |
+#define ASF_WORD 0x5 |
|
43 |
+#define ASF_GUID 0x6 |
|
44 |
+#define ASF_DWORD 0x3 |
|
45 |
+#define ASF_QWORD 0x4 |
|
46 |
+#define ASF_UNICODE 0x0 |
|
47 |
+#define ASF_FLAG_BROADCAST 0x1 |
|
48 |
+#define ASF_BYTE_ARRAY 0x1 |
|
49 |
+#define ASF_TYPE_AUDIO 0x2 |
|
50 |
+#define ASF_TYPE_VIDEO 0x1 |
|
51 |
+#define ASF_STREAM_NUM 0x7F |
|
52 |
+#define ASF_MAX_STREAMS 128 |
|
53 |
+#define BMP_HEADER_SIZE 40 |
|
54 |
+#define ASF_NUM_OF_PAYLOADS 0x3F |
|
55 |
+#define ASF_ERROR_CORRECTION_LENGTH_TYPE 0x60 |
|
56 |
+#define ASF_PACKET_ERROR_CORRECTION_DATA_SIZE 0x2 |
|
57 |
+ |
|
58 |
+typedef struct GUIDParseTable { |
|
59 |
+ const char *name; |
|
60 |
+ ff_asf_guid guid; |
|
61 |
+ int (*read_object)(AVFormatContext *, const struct GUIDParseTable *); |
|
62 |
+ int is_subobject; |
|
63 |
+} GUIDParseTable; |
|
64 |
+ |
|
65 |
+typedef struct ASFPacket { |
|
66 |
+ AVPacket avpkt; |
|
67 |
+ int64_t dts; |
|
68 |
+ uint32_t frame_num; // ASF payloads with the same number are parts of the same frame |
|
69 |
+ int flags; |
|
70 |
+ int data_size; |
|
71 |
+ int duration; |
|
72 |
+ int size_left; |
|
73 |
+ uint8_t stream_index; |
|
74 |
+} ASFPacket; |
|
75 |
+ |
|
76 |
+typedef struct ASFStream { |
|
77 |
+ uint8_t stream_index; // from packet header |
|
78 |
+ int index; // stream index in AVFormatContext, set in asf_read_stream_properties |
|
79 |
+ int type; |
|
80 |
+ int indexed; // added index entries from the Simple Index Object or not |
|
81 |
+ int8_t span; // for deinterleaving |
|
82 |
+ uint16_t virtual_pkt_len; |
|
83 |
+ uint16_t virtual_chunk_len; |
|
84 |
+ int16_t lang_idx; |
|
85 |
+ ASFPacket pkt; |
|
86 |
+} ASFStream; |
|
87 |
+ |
|
88 |
+typedef struct ASFStreamData{ |
|
89 |
+ char langs[32]; |
|
90 |
+ AVDictionary *asf_met; // for storing per-stream metadata |
|
91 |
+ AVRational aspect_ratio; |
|
92 |
+} ASFStreamData; |
|
93 |
+ |
|
94 |
+typedef struct ASFContext { |
|
95 |
+ int data_reached; |
|
96 |
+ int is_simple_index; // is simple index present or not 1/0 |
|
97 |
+ int is_header; |
|
98 |
+ |
|
99 |
+ uint64_t preroll; |
|
100 |
+ uint64_t nb_packets; // ASF packets |
|
101 |
+ uint32_t packet_size; |
|
102 |
+ int64_t send_time; |
|
103 |
+ int duration; |
|
104 |
+ |
|
105 |
+ uint32_t b_flags; // flags with broadcast flag |
|
106 |
+ uint32_t prop_flags; // file properties object flags |
|
107 |
+ |
|
108 |
+ uint64_t data_size; // data object size |
|
109 |
+ uint64_t unknown_size; // size of the unknown object |
|
110 |
+ |
|
111 |
+ int64_t offset; // offset of the current object |
|
112 |
+ |
|
113 |
+ int64_t data_offset; |
|
114 |
+ int64_t first_packet_offset; // packet offset |
|
115 |
+ int64_t unknown_offset; // for top level header objects or subobjects without specified behavior |
|
116 |
+ |
|
117 |
+ // ASF file must not contain more than 128 streams according to the specification |
|
118 |
+ ASFStream *asf_st[ASF_MAX_STREAMS]; |
|
119 |
+ ASFStreamData asf_sd[ASF_MAX_STREAMS]; |
|
120 |
+ int nb_streams; |
|
121 |
+ |
|
122 |
+ int stream_index; // from packet header, for the subpayload case |
|
123 |
+ |
|
124 |
+ // packet parameteres |
|
125 |
+ uint64_t sub_header_offset; // offset of subplayload header |
|
126 |
+ int64_t sub_dts; |
|
127 |
+ uint8_t dts_delta; // for subpayloads |
|
128 |
+ uint32_t packet_size_internal; // packet size stored inside ASFPacket, can be 0 |
|
129 |
+ int64_t dts; |
|
130 |
+ int64_t packet_offset; // offset of the current packet inside Data Object |
|
131 |
+ uint32_t pad_len; // padding after payload |
|
132 |
+ uint32_t rep_data_len; |
|
133 |
+ |
|
134 |
+ // packet state |
|
135 |
+ uint64_t sub_left; // subpayloads left or not |
|
136 |
+ int nb_sub; // number of subpayloads read so far from the current ASF packet |
|
137 |
+ uint16_t mult_sub_len; // total length of subpayloads array inside multiple payload |
|
138 |
+ uint64_t nb_mult_left; // multiple payloads left |
|
139 |
+ int return_subpayload; |
|
140 |
+ enum { |
|
141 |
+ PARSE_PACKET_HEADER, |
|
142 |
+ READ_SINGLE, |
|
143 |
+ READ_MULTI, |
|
144 |
+ READ_MULTI_SUB |
|
145 |
+ } state; |
|
146 |
+} ASFContext; |
|
147 |
+ |
|
148 |
+static int detect_unknown_subobject(AVFormatContext *s, int64_t offset, int64_t size); |
|
149 |
+static const GUIDParseTable *find_guid(ff_asf_guid guid); |
|
150 |
+ |
|
151 |
+static int asf_probe(AVProbeData *pd) |
|
152 |
+{ |
|
153 |
+ /* check file header */ |
|
154 |
+ if (!ff_guidcmp(pd->buf, &ff_asf_header)) |
|
155 |
+ return AVPROBE_SCORE_MAX/2; |
|
156 |
+ else |
|
157 |
+ return 0; |
|
158 |
+} |
|
159 |
+ |
|
160 |
+static void swap_guid(ff_asf_guid guid) |
|
161 |
+{ |
|
162 |
+ FFSWAP(unsigned char, guid[0], guid[3]); |
|
163 |
+ FFSWAP(unsigned char, guid[1], guid[2]); |
|
164 |
+ FFSWAP(unsigned char, guid[4], guid[5]); |
|
165 |
+ FFSWAP(unsigned char, guid[6], guid[7]); |
|
166 |
+} |
|
167 |
+ |
|
168 |
+static void align_position(AVIOContext *pb, int64_t offset, uint64_t size) |
|
169 |
+{ |
|
170 |
+ if (avio_tell(pb) != offset + size) |
|
171 |
+ avio_seek(pb, offset + size, SEEK_SET); |
|
172 |
+} |
|
173 |
+ |
|
174 |
+static int asf_read_unknown(AVFormatContext *s, const GUIDParseTable *g) |
|
175 |
+{ |
|
176 |
+ ASFContext *asf = s->priv_data; |
|
177 |
+ AVIOContext *pb = s->pb; |
|
178 |
+ uint64_t size = avio_rl64(pb); |
|
179 |
+ int ret; |
|
180 |
+ |
|
181 |
+ if (asf->is_header) |
|
182 |
+ asf->unknown_size = size; |
|
183 |
+ asf->is_header = 0; |
|
184 |
+ if (!g->is_subobject) { |
|
185 |
+ if (!(ret = strcmp(g->name, "Header Extension"))) |
|
186 |
+ avio_skip(pb, 22); // skip reserved fields and Data Size |
|
187 |
+ if ((ret = detect_unknown_subobject(s, asf->unknown_offset, |
|
188 |
+ asf->unknown_size)) < 0) |
|
189 |
+ return ret; |
|
190 |
+ } else |
|
191 |
+ avio_skip(pb, size - 24); |
|
192 |
+ |
|
193 |
+ return 0; |
|
194 |
+} |
|
195 |
+ |
|
196 |
+static int get_asf_string(AVIOContext *pb, int maxlen, char *buf, int buflen) |
|
197 |
+{ |
|
198 |
+ char *q = buf; |
|
199 |
+ int ret = 0; |
|
200 |
+ if (buflen <= 0) |
|
201 |
+ return AVERROR(EINVAL); |
|
202 |
+ while (ret + 1 < maxlen) { |
|
203 |
+ uint8_t tmp; |
|
204 |
+ uint32_t ch; |
|
205 |
+ GET_UTF16(ch, (ret += 2) <= maxlen ? avio_rl16(pb) : 0, break;); |
|
206 |
+ PUT_UTF8(ch, tmp, if (q - buf < buflen - 1) *q++ = tmp;) |
|
207 |
+ } |
|
208 |
+ *q = 0; |
|
209 |
+ |
|
210 |
+ return ret; |
|
211 |
+} |
|
212 |
+ |
|
213 |
+static int asf_read_marker(AVFormatContext *s, const GUIDParseTable *g) |
|
214 |
+{ |
|
215 |
+ ASFContext *asf = s->priv_data; |
|
216 |
+ AVIOContext *pb = s->pb; |
|
217 |
+ uint64_t size = avio_rl64(pb); |
|
218 |
+ int i, nb_markers, ret; |
|
219 |
+ size_t len; |
|
220 |
+ char name[1024]; |
|
221 |
+ |
|
222 |
+ avio_skip(pb, 8); |
|
223 |
+ avio_skip(pb, 8); // skip reserved GUID |
|
224 |
+ nb_markers = avio_rl32(pb); |
|
225 |
+ avio_skip(pb, 2); // skip reserved field |
|
226 |
+ len = avio_rl16(pb); |
|
227 |
+ for (i = 0; i < len; i++) |
|
228 |
+ avio_skip(pb, 1); |
|
229 |
+ |
|
230 |
+ for (i = 0; i < nb_markers; i++) { |
|
231 |
+ int64_t pts; |
|
232 |
+ |
|
233 |
+ avio_skip(pb, 8); |
|
234 |
+ pts = avio_rl64(pb); |
|
235 |
+ pts -= asf->preroll * 10000; |
|
236 |
+ avio_skip(pb, 2); // entry length |
|
237 |
+ avio_skip(pb, 4); // send time |
|
238 |
+ avio_skip(pb, 4); // flags |
|
239 |
+ len = avio_rl32(pb); |
|
240 |
+ |
|
241 |
+ if ((ret = avio_get_str16le(pb, len, name, |
|
242 |
+ sizeof(name))) < len) |
|
243 |
+ avio_skip(pb, len - ret); |
|
244 |
+ avpriv_new_chapter(s, i, (AVRational) { 1, 10000000 }, pts, |
|
245 |
+ AV_NOPTS_VALUE, name); |
|
246 |
+ } |
|
247 |
+ align_position(pb, asf->offset, size); |
|
248 |
+ |
|
249 |
+ return 0; |
|
250 |
+} |
|
251 |
+ |
|
252 |
+static int asf_read_metadata(AVFormatContext *s, const char *title, uint16_t len, |
|
253 |
+ unsigned char *ch, uint16_t buflen) |
|
254 |
+{ |
|
255 |
+ AVIOContext *pb = s->pb; |
|
256 |
+ |
|
257 |
+ avio_get_str16le(pb, len, ch, buflen); |
|
258 |
+ if (av_dict_set(&s->metadata, title, ch, 0) < 0) |
|
259 |
+ av_log(s, AV_LOG_WARNING, "av_dict_set failed.\n"); |
|
260 |
+ |
|
261 |
+ return 0; |
|
262 |
+} |
|
263 |
+ |
|
264 |
+static int asf_read_value(AVFormatContext *s, uint8_t *name, uint16_t name_len, |
|
265 |
+ uint16_t val_len, int type, AVDictionary **met) |
|
266 |
+{ |
|
267 |
+ int ret; |
|
268 |
+ uint8_t *value; |
|
269 |
+ uint16_t buflen = 2 * val_len + 1; |
|
270 |
+ AVIOContext *pb = s->pb; |
|
271 |
+ |
|
272 |
+ value = av_malloc(buflen); |
|
273 |
+ if (!value) |
|
274 |
+ return AVERROR(ENOMEM); |
|
275 |
+ if (type == ASF_UNICODE) { |
|
276 |
+ // get_asf_string reads UTF-16 and converts it to UTF-8 which needs longer buffer |
|
277 |
+ if ((ret = get_asf_string(pb, val_len, value, buflen)) < 0) |
|
278 |
+ goto failed; |
|
279 |
+ if (av_dict_set(met, name, value, 0) < 0) |
|
280 |
+ av_log(s, AV_LOG_WARNING, "av_dict_set failed.\n"); |
|
281 |
+ } else { |
|
282 |
+ char buf[256]; |
|
283 |
+ if (val_len > sizeof(buf)) |
|
284 |
+ return AVERROR_INVALIDDATA; |
|
285 |
+ if ((ret = avio_read(pb, value, val_len)) < 0) |
|
286 |
+ goto failed; |
|
287 |
+ if (ret < 2 * val_len) |
|
288 |
+ value[ret] = '\0'; |
|
289 |
+ else |
|
290 |
+ value[2 * val_len - 1] = '\0'; |
|
291 |
+ snprintf(buf, sizeof(buf), "%s", value); |
|
292 |
+ if (av_dict_set(met, name, buf, 0) < 0) |
|
293 |
+ av_log(s, AV_LOG_WARNING, "av_dict_set failed.\n"); |
|
294 |
+ } |
|
295 |
+ av_freep(&value); |
|
296 |
+ |
|
297 |
+ return 0; |
|
298 |
+ |
|
299 |
+failed: |
|
300 |
+ av_freep(&value); |
|
301 |
+ return ret; |
|
302 |
+} |
|
303 |
+ |
|
304 |
+static int asf_read_generic_value(AVFormatContext *s, uint8_t *name, uint16_t name_len, |
|
305 |
+ int type, AVDictionary **met) |
|
306 |
+{ |
|
307 |
+ AVIOContext *pb = s->pb; |
|
308 |
+ uint64_t value; |
|
309 |
+ char buf[32]; |
|
310 |
+ |
|
311 |
+ switch (type) { |
|
312 |
+ case ASF_BOOL: |
|
313 |
+ value = avio_rl32(pb); |
|
314 |
+ break; |
|
315 |
+ case ASF_DWORD: |
|
316 |
+ value = avio_rl32(pb); |
|
317 |
+ break; |
|
318 |
+ case ASF_QWORD: |
|
319 |
+ value = avio_rl64(pb); |
|
320 |
+ break; |
|
321 |
+ case ASF_WORD: |
|
322 |
+ value = avio_rl16(pb); |
|
323 |
+ break; |
|
324 |
+ default: |
|
325 |
+ av_freep(&name); |
|
326 |
+ return AVERROR_INVALIDDATA; |
|
327 |
+ } |
|
328 |
+ snprintf(buf, sizeof(buf), "%"PRIu64, value); |
|
329 |
+ if (av_dict_set(met, name, buf, 0) < 0) |
|
330 |
+ av_log(s, AV_LOG_WARNING, "av_dict_set failed.\n"); |
|
331 |
+ |
|
332 |
+ return 0; |
|
333 |
+} |
|
334 |
+ |
|
335 |
+/* MSDN claims that this should be "compatible with the ID3 frame, APIC", |
|
336 |
+ * but in reality this is only loosely similar */ |
|
337 |
+static int asf_read_picture(AVFormatContext *s, int len) |
|
338 |
+{ |
|
339 |
+ ASFContext *asf = s->priv_data; |
|
340 |
+ AVPacket pkt = { 0 }; |
|
341 |
+ const CodecMime *mime = ff_id3v2_mime_tags; |
|
342 |
+ enum AVCodecID id = AV_CODEC_ID_NONE; |
|
343 |
+ char mimetype[64]; |
|
344 |
+ uint8_t *desc = NULL; |
|
345 |
+ AVStream *st = NULL; |
|
346 |
+ int ret, type, picsize, desc_len; |
|
347 |
+ ASFStream *asf_st; |
|
348 |
+ |
|
349 |
+ /* type + picsize + mime + desc */ |
|
350 |
+ if (len < 1 + 4 + 2 + 2) { |
|
351 |
+ av_log(s, AV_LOG_ERROR, "Invalid attached picture size: %d.\n", len); |
|
352 |
+ return AVERROR_INVALIDDATA; |
|
353 |
+ } |
|
354 |
+ |
|
355 |
+ /* picture type */ |
|
356 |
+ type = avio_r8(s->pb); |
|
357 |
+ len--; |
|
358 |
+ if (type >= FF_ARRAY_ELEMS(ff_id3v2_picture_types) || type < 0) { |
|
359 |
+ av_log(s, AV_LOG_WARNING, "Unknown attached picture type: %d.\n", type); |
|
360 |
+ type = 0; |
|
361 |
+ } |
|
362 |
+ |
|
363 |
+ /* picture data size */ |
|
364 |
+ picsize = avio_rl32(s->pb); |
|
365 |
+ len -= 4; |
|
366 |
+ |
|
367 |
+ /* picture MIME type */ |
|
368 |
+ len -= avio_get_str16le(s->pb, len, mimetype, sizeof(mimetype)); |
|
369 |
+ while (mime->id != AV_CODEC_ID_NONE) { |
|
370 |
+ if (!strncmp(mime->str, mimetype, sizeof(mimetype))) { |
|
371 |
+ id = mime->id; |
|
372 |
+ break; |
|
373 |
+ } |
|
374 |
+ mime++; |
|
375 |
+ } |
|
376 |
+ if (id == AV_CODEC_ID_NONE) { |
|
377 |
+ av_log(s, AV_LOG_ERROR, "Unknown attached picture mimetype: %s.\n", |
|
378 |
+ mimetype); |
|
379 |
+ return 0; |
|
380 |
+ } |
|
381 |
+ |
|
382 |
+ if (picsize >= len) { |
|
383 |
+ av_log(s, AV_LOG_ERROR, "Invalid attached picture data size: %d >= %d.\n", |
|
384 |
+ picsize, len); |
|
385 |
+ return AVERROR_INVALIDDATA; |
|
386 |
+ } |
|
387 |
+ |
|
388 |
+ /* picture description */ |
|
389 |
+ desc_len = (len - picsize) * 2 + 1; |
|
390 |
+ desc = av_malloc(desc_len); |
|
391 |
+ if (!desc) |
|
392 |
+ return AVERROR(ENOMEM); |
|
393 |
+ len -= avio_get_str16le(s->pb, len - picsize, desc, desc_len); |
|
394 |
+ |
|
395 |
+ ret = av_get_packet(s->pb, &pkt, picsize); |
|
396 |
+ if (ret < 0) |
|
397 |
+ goto fail; |
|
398 |
+ |
|
399 |
+ st = avformat_new_stream(s, NULL); |
|
400 |
+ if (!st) { |
|
401 |
+ ret = AVERROR(ENOMEM); |
|
402 |
+ goto fail; |
|
403 |
+ } |
|
404 |
+ asf->asf_st[asf->nb_streams] = av_mallocz(sizeof(*asf_st)); |
|
405 |
+ asf_st = asf->asf_st[asf->nb_streams]; |
|
406 |
+ if (!asf_st) |
|
407 |
+ return AVERROR(ENOMEM); |
|
408 |
+ |
|
409 |
+ st->disposition |= AV_DISPOSITION_ATTACHED_PIC; |
|
410 |
+ st->codec->codec_type = asf_st->type = AVMEDIA_TYPE_VIDEO; |
|
411 |
+ st->codec->codec_id = id; |
|
412 |
+ st->attached_pic = pkt; |
|
413 |
+ st->attached_pic.stream_index = asf_st->index = st->index; |
|
414 |
+ st->attached_pic.flags |= AV_PKT_FLAG_KEY; |
|
415 |
+ |
|
416 |
+ asf->nb_streams++; |
|
417 |
+ |
|
418 |
+ if (*desc) { |
|
419 |
+ if (av_dict_set(&st->metadata, "title", desc, AV_DICT_DONT_STRDUP_VAL) < 0) |
|
420 |
+ av_log(s, AV_LOG_WARNING, "av_dict_set failed.\n"); |
|
421 |
+ } else |
|
422 |
+ av_freep(&desc); |
|
423 |
+ |
|
424 |
+ if (av_dict_set(&st->metadata, "comment", ff_id3v2_picture_types[type], 0) < 0) |
|
425 |
+ av_log(s, AV_LOG_WARNING, "av_dict_set failed.\n"); |
|
426 |
+ |
|
427 |
+ return 0; |
|
428 |
+ |
|
429 |
+fail: |
|
430 |
+ av_freep(&desc); |
|
431 |
+ av_free_packet(&pkt); |
|
432 |
+ return ret; |
|
433 |
+} |
|
434 |
+ |
|
435 |
+static void get_id3_tag(AVFormatContext *s, int len) |
|
436 |
+{ |
|
437 |
+ ID3v2ExtraMeta *id3v2_extra_meta = NULL; |
|
438 |
+ |
|
439 |
+ ff_id3v2_read(s, ID3v2_DEFAULT_MAGIC, &id3v2_extra_meta, len); |
|
440 |
+ if (id3v2_extra_meta) |
|
441 |
+ ff_id3v2_parse_apic(s, &id3v2_extra_meta); |
|
442 |
+ ff_id3v2_free_extra_meta(&id3v2_extra_meta); |
|
443 |
+} |
|
444 |
+ |
|
445 |
+static int process_metadata(AVFormatContext *s, uint8_t *name, uint16_t name_len, |
|
446 |
+ uint16_t val_len, uint16_t type, AVDictionary **met) |
|
447 |
+{ |
|
448 |
+ int ret; |
|
449 |
+ ff_asf_guid guid; |
|
450 |
+ |
|
451 |
+ if (val_len) { |
|
452 |
+ switch (type) { |
|
453 |
+ case ASF_UNICODE: |
|
454 |
+ asf_read_value(s, name, name_len, val_len, type, met); |
|
455 |
+ break; |
|
456 |
+ case ASF_BYTE_ARRAY: |
|
457 |
+ if (!strcmp(name, "WM/Picture")) // handle cover art |
|
458 |
+ asf_read_picture(s, val_len); |
|
459 |
+ else if (!strcmp(name, "ID3")) // handle ID3 tag |
|
460 |
+ get_id3_tag(s, val_len); |
|
461 |
+ else |
|
462 |
+ asf_read_value(s, name, name_len, val_len, type, met); |
|
463 |
+ break; |
|
464 |
+ case ASF_GUID: |
|
465 |
+ ff_get_guid(s->pb, &guid); |
|
466 |
+ break; |
|
467 |
+ default: |
|
468 |
+ if ((ret = asf_read_generic_value(s, name, name_len, type, met)) < 0) |
|
469 |
+ return ret; |
|
470 |
+ break; |
|
471 |
+ } |
|
472 |
+ } |
|
473 |
+ av_freep(&name); |
|
474 |
+ |
|
475 |
+ return 0; |
|
476 |
+} |
|
477 |
+ |
|
478 |
+static int asf_read_ext_content(AVFormatContext *s, const GUIDParseTable *g) |
|
479 |
+{ |
|
480 |
+ ASFContext *asf = s->priv_data; |
|
481 |
+ AVIOContext *pb = s->pb; |
|
482 |
+ uint64_t size = avio_rl64(pb); |
|
483 |
+ uint16_t nb_desc = avio_rl16(pb); |
|
484 |
+ int i, ret; |
|
485 |
+ |
|
486 |
+ for (i = 0; i < nb_desc; i++) { |
|
487 |
+ uint16_t name_len, type, val_len; |
|
488 |
+ uint8_t *name = NULL; |
|
489 |
+ |
|
490 |
+ name_len = avio_rl16(pb); |
|
491 |
+ if (!name_len) |
|
492 |
+ return AVERROR_INVALIDDATA; |
|
493 |
+ name = av_malloc(name_len); |
|
494 |
+ if (!name) |
|
495 |
+ return AVERROR(ENOMEM); |
|
496 |
+ avio_get_str16le(pb, name_len, name, |
|
497 |
+ name_len); |
|
498 |
+ type = avio_rl16(pb); |
|
499 |
+ val_len = avio_rl16(pb); |
|
500 |
+ |
|
501 |
+ if ((ret = process_metadata(s, name, name_len, val_len, type, &s->metadata)) < 0) |
|
502 |
+ return ret; |
|
503 |
+ } |
|
504 |
+ |
|
505 |
+ align_position(pb, asf->offset, size); |
|
506 |
+ return 0; |
|
507 |
+} |
|
508 |
+ |
|
509 |
+static AVStream *find_stream(AVFormatContext *s, uint16_t st_num) |
|
510 |
+{ |
|
511 |
+ AVStream *st = NULL; |
|
512 |
+ ASFContext *asf = s->priv_data; |
|
513 |
+ int i; |
|
514 |
+ |
|
515 |
+ for (i = 0; i < asf->nb_streams; i++) { |
|
516 |
+ if (asf->asf_st[i]->stream_index == st_num) { |
|
517 |
+ st = s->streams[asf->asf_st[i]->index]; |
|
518 |
+ break; |
|
519 |
+ } |
|
520 |
+ } |
|
521 |
+ |
|
522 |
+ return st; |
|
523 |
+} |
|
524 |
+ |
|
525 |
+static void asf_store_aspect_ratio(AVFormatContext *s, uint8_t st_num, uint8_t *name) |
|
526 |
+{ |
|
527 |
+ ASFContext *asf = s->priv_data; |
|
528 |
+ AVIOContext *pb = s->pb; |
|
529 |
+ uint16_t value = 0; |
|
530 |
+ |
|
531 |
+ value = avio_rl16(pb); |
|
532 |
+ |
|
533 |
+ if (st_num < ASF_MAX_STREAMS) { |
|
534 |
+ if (!strcmp(name, "AspectRatioX")) |
|
535 |
+ asf->asf_sd[st_num].aspect_ratio.num = value; |
|
536 |
+ else |
|
537 |
+ asf->asf_sd[st_num].aspect_ratio.den = value; |
|
538 |
+ } |
|
539 |
+} |
|
540 |
+ |
|
541 |
+static int asf_read_metadata_obj(AVFormatContext *s, const GUIDParseTable *g) |
|
542 |
+{ |
|
543 |
+ ASFContext *asf = s->priv_data; |
|
544 |
+ AVIOContext *pb = s->pb; |
|
545 |
+ uint64_t size = avio_rl64(pb); |
|
546 |
+ uint16_t nb_recs = avio_rl16(pb); // number of records in the Description Records list |
|
547 |
+ int i, ret; |
|
548 |
+ |
|
549 |
+ for (i = 0; i < nb_recs; i++) { |
|
550 |
+ uint16_t name_len, buflen, type, val_len, st_num; |
|
551 |
+ uint8_t *name = NULL; |
|
552 |
+ |
|
553 |
+ avio_skip(pb, 2); // skip reserved field |
|
554 |
+ st_num = avio_rl16(pb); |
|
555 |
+ name_len = avio_rl16(pb); |
|
556 |
+ buflen = 2 * name_len + 1; |
|
557 |
+ if (!name_len) |
|
558 |
+ break; |
|
559 |
+ type = avio_rl16(pb); |
|
560 |
+ val_len = avio_rl32(pb); |
|
561 |
+ name = av_malloc(name_len); |
|
562 |
+ if (!name) |
|
563 |
+ return AVERROR(ENOMEM); |
|
564 |
+ avio_get_str16le(pb, name_len, name, |
|
565 |
+ buflen); |
|
566 |
+ |
|
567 |
+ if (!strcmp(name, "AspectRatioX") || !strcmp(name, "AspectRatioY")) { |
|
568 |
+ asf_store_aspect_ratio(s, st_num, name); |
|
569 |
+ } else { |
|
570 |
+ if (st_num < ASF_MAX_STREAMS) { |
|
571 |
+ if ((ret = process_metadata(s, name, name_len, val_len, type, |
|
572 |
+ &asf->asf_sd[st_num].asf_met)) < 0) |
|
573 |
+ break; |
|
574 |
+ } |
|
575 |
+ } |
|
576 |
+ } |
|
577 |
+ |
|
578 |
+ align_position(pb, asf->offset, size); |
|
579 |
+ return 0; |
|
580 |
+} |
|
581 |
+ |
|
582 |
+static int asf_read_content_desc(AVFormatContext *s, const GUIDParseTable *g) |
|
583 |
+{ |
|
584 |
+ ASFContext *asf = s->priv_data; |
|
585 |
+ AVIOContext *pb = s->pb; |
|
586 |
+ int i; |
|
587 |
+ static const char *const titles[] = |
|
588 |
+ { "Title", "Author", "Copyright", "Description", "Rate" }; |
|
589 |
+ uint16_t len[5], buflen[5] = { 0 }; |
|
590 |
+ uint8_t *ch; |
|
591 |
+ uint64_t size = avio_rl64(pb); |
|
592 |
+ |
|
593 |
+ for (i = 0; i < 5; i++) { |
|
594 |
+ len[i] = avio_rl16(pb); |
|
595 |
+ // utf8 string should be <= 2 * utf16 string, extra byte for the terminator |
|
596 |
+ buflen[i] = 2 * len[i] + 1; |
|
597 |
+ } |
|
598 |
+ |
|
599 |
+ for (i = 0; i < 5; i++) { |
|
600 |
+ ch = av_malloc(buflen[i]); |
|
601 |
+ if (!ch) |
|
602 |
+ return(AVERROR(ENOMEM)); |
|
603 |
+ asf_read_metadata(s, titles[i], len[i], ch, buflen[i]); |
|
604 |
+ av_freep(&ch); |
|
605 |
+ } |
|
606 |
+ align_position(pb, asf->offset, size); |
|
607 |
+ |
|
608 |
+ return 0; |
|
609 |
+} |
|
610 |
+ |
|
611 |
+static int asf_read_properties(AVFormatContext *s, const GUIDParseTable *g) |
|
612 |
+{ |
|
613 |
+ ASFContext *asf = s->priv_data; |
|
614 |
+ AVIOContext *pb = s->pb; |
|
615 |
+ uint64_t creation_time; |
|
616 |
+ |
|
617 |
+ avio_rl64(pb); // read object size |
|
618 |
+ avio_skip(pb, 16); // skip File ID |
|
619 |
+ avio_skip(pb, 8); // skip File size |
|
620 |
+ creation_time = avio_rl64(pb); |
|
621 |
+ if (!(asf->b_flags & ASF_FLAG_BROADCAST)) { |
|
622 |
+ struct tm tmbuf; |
|
623 |
+ struct tm *tm; |
|
624 |
+ char buf[64]; |
|
625 |
+ |
|
626 |
+ // creation date is in 100 ns units from 1 Jan 1601, conversion to s |
|
627 |
+ creation_time /= 10000000; |
|
628 |
+ // there are 11644473600 seconds between 1 Jan 1601 and 1 Jan 1970 |
|
629 |
+ creation_time -= 11644473600; |
|
630 |
+ tm = gmtime_r(&creation_time, &tmbuf); |
|
631 |
+ if (tm) { |
|
632 |
+ if (!strftime(buf, sizeof(buf), "%Y-%m-%d %H:%M:%S", tm)) |
|
633 |
+ buf[0] = '\0'; |
|
634 |
+ } else |
|
635 |
+ buf[0] = '\0'; |
|
636 |
+ if (buf[0]) { |
|
637 |
+ if (av_dict_set(&s->metadata, "creation_time", buf, 0) < 0) |
|
638 |
+ av_log(s, AV_LOG_WARNING, "av_dict_set failed.\n"); |
|
639 |
+ } |
|
640 |
+ } |
|
641 |
+ asf->nb_packets = avio_rl64(pb); |
|
642 |
+ asf->duration = avio_rl64(pb) / 10000; // stream duration |
|
643 |
+ avio_skip(pb, 8); // skip send duration |
|
644 |
+ asf->preroll = avio_rl64(pb); |
|
645 |
+ asf->duration -= asf->preroll; |
|
646 |
+ asf->b_flags = avio_rl32(pb); |
|
647 |
+ avio_skip(pb, 4); // skip minimal packet size |
|
648 |
+ asf->packet_size = avio_rl32(pb); |
|
649 |
+ avio_skip(pb, 4); // skip max_bitrate |
|
650 |
+ |
|
651 |
+ return 0; |
|
652 |
+} |
|
653 |
+ |
|
654 |
+static int parse_video_info(AVIOContext *pb, AVStream *st) |
|
655 |
+{ |
|
656 |
+ uint16_t size; |
|
657 |
+ unsigned int tag; |
|
658 |
+ |
|
659 |
+ st->codec->width = avio_rl32(pb); |
|
660 |
+ st->codec->height = avio_rl32(pb); |
|
661 |
+ avio_skip(pb, 1); // skip reserved flags |
|
662 |
+ size = avio_rl16(pb); // size of the Format Data |
|
663 |
+ tag = ff_get_bmp_header(pb, st, NULL); |
|
664 |
+ st->codec->codec_tag = tag; |
|
665 |
+ st->codec->codec_id = ff_codec_get_id(ff_codec_bmp_tags, tag); |
|
666 |
+ |
|
667 |
+ if (size > BMP_HEADER_SIZE) { |
|
668 |
+ int ret; |
|
669 |
+ st->codec->extradata_size = size - BMP_HEADER_SIZE; |
|
670 |
+ if (!(st->codec->extradata = av_malloc(st->codec->extradata_size + |
|
671 |
+ FF_INPUT_BUFFER_PADDING_SIZE))) { |
|
672 |
+ st->codec->extradata_size = 0; |
|
673 |
+ return AVERROR(ENOMEM); |
|
674 |
+ } |
|
675 |
+ memset(st->codec->extradata + st->codec->extradata_size , 0, |
|
676 |
+ FF_INPUT_BUFFER_PADDING_SIZE); |
|
677 |
+ if ((ret = avio_read(pb, st->codec->extradata, |
|
678 |
+ st->codec->extradata_size)) < 0) |
|
679 |
+ return ret; |
|
680 |
+ } |
|
681 |
+ return 0; |
|
682 |
+} |
|
683 |
+ |
|
684 |
+static int asf_read_stream_properties(AVFormatContext *s, const GUIDParseTable *g) |
|
685 |
+{ |
|
686 |
+ ASFContext *asf = s->priv_data; |
|
687 |
+ AVIOContext *pb = s->pb; |
|
688 |
+ uint64_t size; |
|
689 |
+ uint32_t err_data_len, ts_data_len; // type specific data length |
|
690 |
+ uint16_t flags; |
|
691 |
+ ff_asf_guid stream_type; |
|
692 |
+ enum AVMediaType type; |
|
693 |
+ int i, ret; |
|
694 |
+ uint8_t stream_index; |
|
695 |
+ AVStream *st; |
|
696 |
+ ASFStream *asf_st; |
|
697 |
+ |
|
698 |
+ // ASF file must not contain more than 128 streams according to the specification |
|
699 |
+ if (asf->nb_streams >= ASF_MAX_STREAMS) |
|
700 |
+ return AVERROR_INVALIDDATA; |
|
701 |
+ |
|
702 |
+ size = avio_rl64(pb); |
|
703 |
+ ff_get_guid(pb, &stream_type); |
|
704 |
+ if (!ff_guidcmp(&stream_type, &ff_asf_audio_stream)) |
|
705 |
+ type = AVMEDIA_TYPE_AUDIO; |
|
706 |
+ else if (!ff_guidcmp(&stream_type, &ff_asf_video_stream)) |
|
707 |
+ type = AVMEDIA_TYPE_VIDEO; |
|
708 |
+ else if (!ff_guidcmp(&stream_type, &ff_asf_jfif_media)) |
|
709 |
+ type = AVMEDIA_TYPE_VIDEO; |
|
710 |
+ else if (!ff_guidcmp(&stream_type, &ff_asf_command_stream)) |
|
711 |
+ type = AVMEDIA_TYPE_DATA; |
|
712 |
+ else if (!ff_guidcmp(&stream_type, |
|
713 |
+ &ff_asf_ext_stream_embed_stream_header)) |
|
714 |
+ type = AVMEDIA_TYPE_UNKNOWN; |
|
715 |
+ else |
|
716 |
+ return AVERROR_INVALIDDATA; |
|
717 |
+ |
|
718 |
+ ff_get_guid(pb, &stream_type); // error correction type |
|
719 |
+ avio_skip(pb, 8); // skip the time offset |
|
720 |
+ ts_data_len = avio_rl32(pb); |
|
721 |
+ err_data_len = avio_rl32(pb); |
|
722 |
+ flags = avio_rl16(pb); // bit 15 - Encrypted Content |
|
723 |
+ |
|
724 |
+ stream_index = flags & ASF_STREAM_NUM; |
|
725 |
+ for (i = 0; i < asf->nb_streams; i++) |
|
726 |
+ if (stream_index == asf->asf_st[i]->stream_index) { |
|
727 |
+ av_log(s, AV_LOG_WARNING, |
|
728 |
+ "Duplicate stream found, this stream will be ignored.\n"); |
|
729 |
+ align_position(pb, asf->offset, size); |
|
730 |
+ return 0; |
|
731 |
+ } |
|
732 |
+ |
|
733 |
+ st = avformat_new_stream(s, NULL); |
|
734 |
+ if (!st) |
|
735 |
+ return AVERROR(ENOMEM); |
|
736 |
+ avpriv_set_pts_info(st, 32, 1, 1000); // pts should be dword, in milliseconds |
|
737 |
+ st->codec->codec_type = type; |
|
738 |
+ asf->asf_st[asf->nb_streams] = av_mallocz(sizeof(*asf_st)); |
|
739 |
+ if (!asf->asf_st[asf->nb_streams]) |
|
740 |
+ return AVERROR(ENOMEM); |
|
741 |
+ asf_st = asf->asf_st[asf->nb_streams]; |
|
742 |
+ asf_st->stream_index = stream_index; |
|
743 |
+ asf_st->index = st->index; |
|
744 |
+ asf_st->indexed = 0; |
|
745 |
+ st->id = flags & ASF_STREAM_NUM; |
|
746 |
+ av_init_packet(&asf_st->pkt.avpkt); |
|
747 |
+ asf_st->pkt.data_size = 0; |
|
748 |
+ avio_skip(pb, 4); // skip reserved field |
|
749 |
+ if (!ts_data_len) { |
|
750 |
+ av_log(s, AV_LOG_WARNING, "Suspicious data found! ASF stream #%d will be ignored.\n", |
|
751 |
+ asf_st->stream_index); |
|
752 |
+ align_position(pb, asf->offset, size); |
|
753 |
+ return 0; |
|
754 |
+ } |
|
755 |
+ |
|
756 |
+ switch (type) { |
|
757 |
+ case AVMEDIA_TYPE_AUDIO: |
|
758 |
+ asf_st->type = AVMEDIA_TYPE_AUDIO; |
|
759 |
+ if ((ret = ff_get_wav_header(pb, st->codec, ts_data_len, 0)) < 0) |
|
760 |
+ return ret; |
|
761 |
+ break; |
|
762 |
+ case AVMEDIA_TYPE_VIDEO: |
|
763 |
+ asf_st->type = AVMEDIA_TYPE_VIDEO; |
|
764 |
+ if ((ret = parse_video_info(pb, st)) < 0) |
|
765 |
+ return ret; |
|
766 |
+ break; |
|
767 |
+ default: |
|
768 |
+ avio_skip(pb, ts_data_len); |
|
769 |
+ break; |
|
770 |
+ } |
|
771 |
+ |
|
772 |
+ if (err_data_len) { |
|
773 |
+ if (type == AVMEDIA_TYPE_AUDIO) { |
|
774 |
+ uint8_t span = avio_r8(pb); |
|
775 |
+ if (span > 1) { |
|
776 |
+ asf_st->span = span; |
|
777 |
+ asf_st->virtual_pkt_len = avio_rl16(pb); |
|
778 |
+ asf_st->virtual_chunk_len = avio_rl16(pb); |
|
779 |
+ avio_skip(pb, err_data_len - 5); |
|
780 |
+ } else |
|
781 |
+ avio_skip(pb, err_data_len - 1); |
|
782 |
+ } else |
|
783 |
+ avio_skip(pb, err_data_len); |
|
784 |
+ } |
|
785 |
+ |
|
786 |
+ asf->nb_streams++; |
|
787 |
+ align_position(pb, asf->offset, size); |
|
788 |
+ |
|
789 |
+ return 0; |
|
790 |
+} |
|
791 |
+ |
|
792 |
+static void set_language(AVFormatContext *s, const char *rfc1766, AVDictionary **met) |
|
793 |
+{ |
|
794 |
+ // language abbr should contain at least 2 chars |
|
795 |
+ if (rfc1766 && strlen(rfc1766) > 1) { |
|
796 |
+ const char primary_tag[3] = { rfc1766[0], rfc1766[1], '\0' }; // ignore country code if any |
|
797 |
+ const char *iso6392 = av_convert_lang_to(primary_tag, |
|
798 |
+ AV_LANG_ISO639_2_BIBL); |
|
799 |
+ if (iso6392) |
|
800 |
+ if (av_dict_set(met, "language", iso6392, 0) < 0) |
|
801 |
+ av_log(s, AV_LOG_WARNING, "av_dict_set failed.\n"); |
|
802 |
+ } |
|
803 |
+} |
|
804 |
+ |
|
805 |
+static int asf_read_ext_stream_properties(AVFormatContext *s, const GUIDParseTable *g) |
|
806 |
+{ |
|
807 |
+ ASFContext *asf = s->priv_data; |
|
808 |
+ AVIOContext *pb = s->pb; |
|
809 |
+ AVStream *st = NULL; |
|
810 |
+ ff_asf_guid guid; |
|
811 |
+ uint16_t nb_st_name, nb_pay_exts, st_num, lang_idx; |
|
812 |
+ int i, ret; |
|
813 |
+ uint32_t bitrate; |
|
814 |
+ uint64_t start_time, end_time, time_per_frame; |
|
815 |
+ uint64_t size = avio_rl64(pb); |
|
816 |
+ |
|
817 |
+ start_time = avio_rl64(pb); |
|
818 |
+ end_time = avio_rl64(pb); |
|
819 |
+ bitrate = avio_rl32(pb); |
|
820 |
+ avio_skip(pb, 28); // skip some unused values |
|
821 |
+ st_num = avio_rl16(pb); |
|
822 |
+ st_num &= ASF_STREAM_NUM; |
|
823 |
+ lang_idx = avio_rl16(pb); // Stream Language ID Index |
|
824 |
+ for (i = 0; i < asf->nb_streams; i++) { |
|
825 |
+ if (st_num == asf->asf_st[i]->stream_index) { |
|
826 |
+ st = s->streams[asf->asf_st[i]->index]; |
|
827 |
+ asf->asf_st[i]->lang_idx = lang_idx; |
|
828 |
+ break; |
|
829 |
+ } |
|
830 |
+ } |
|
831 |
+ time_per_frame = avio_rl64(pb); // average time per frame |
|
832 |
+ if (st) { |
|
833 |
+ st->start_time = start_time; |
|
834 |
+ st->duration = end_time - start_time; |
|
835 |
+ st->codec->bit_rate = bitrate; |
|
836 |
+ st->avg_frame_rate.num = 10000000; |
|
837 |
+ st->avg_frame_rate.den = time_per_frame; |
|
838 |
+ } |
|
839 |
+ nb_st_name = avio_rl16(pb); |
|
840 |
+ nb_pay_exts = avio_rl16(pb); |
|
841 |
+ for (i = 0; i < nb_st_name; i++) { |
|
842 |
+ uint16_t len; |
|
843 |
+ |
|
844 |
+ avio_rl16(pb); // Language ID Index |
|
845 |
+ len = avio_rl16(pb); |
|
846 |
+ avio_skip(pb, len); |
|
847 |
+ } |
|
848 |
+ |
|
849 |
+ for (i = 0; i < nb_pay_exts; i++) { |
|
850 |
+ uint32_t len; |
|
851 |
+ avio_skip(pb, 16); // Extension System ID |
|
852 |
+ avio_skip(pb, 2); // Extension Data Size |
|
853 |
+ len = avio_rl32(pb); |
|
854 |
+ avio_skip(pb, len); |
|
855 |
+ } |
|
856 |
+ |
|
857 |
+ if ((ret = ff_get_guid(pb, &guid)) < 0) { |
|
858 |
+ align_position(pb, asf->offset, size); |
|
859 |
+ |
|
860 |
+ return 0; |
|
861 |
+ } |
|
862 |
+ |
|
863 |
+ g = find_guid(guid); |
|
864 |
+ if (g && !(strcmp(g->name, "Stream Properties"))) { |
|
865 |
+ if ((ret = g->read_object(s, g)) < 0) |
|
866 |
+ return ret; |
|
867 |
+ } |
|
868 |
+ |
|
869 |
+ align_position(pb, asf->offset, size); |
|
870 |
+ return 0; |
|
871 |
+} |
|
872 |
+ |
|
873 |
+static int asf_read_language_list(AVFormatContext *s, const GUIDParseTable *g) |
|
874 |
+{ |
|
875 |
+ ASFContext *asf = s->priv_data; |
|
876 |
+ AVIOContext *pb = s->pb; |
|
877 |
+ int i, ret; |
|
878 |
+ uint64_t size = avio_rl64(pb); |
|
879 |
+ uint16_t nb_langs = avio_rl16(pb); |
|
880 |
+ |
|
881 |
+ if (nb_langs < ASF_MAX_STREAMS) { |
|
882 |
+ for (i = 0; i < nb_langs; i++) { |
|
883 |
+ size_t len; |
|
884 |
+ len = avio_r8(pb); |
|
885 |
+ if (!len) |
|
886 |
+ len = 6; |
|
887 |
+ if ((ret = get_asf_string(pb, len, asf->asf_sd[i].langs, |
|
888 |
+ sizeof(asf->asf_sd[i].langs))) < 0) { |
|
889 |
+ return ret; |
|
890 |
+ } |
|
891 |
+ } |
|
892 |
+ } |
|
893 |
+ |
|
894 |
+ align_position(pb, asf->offset, size); |
|
895 |
+ return 0; |
|
896 |
+} |
|
897 |
+ |
|
898 |
+// returns data object offset when reading this object for the first time |
|
899 |
+static int asf_read_data(AVFormatContext *s, const GUIDParseTable *g) |
|
900 |
+{ |
|
901 |
+ ASFContext *asf = s->priv_data; |
|
902 |
+ AVIOContext *pb = s->pb; |
|
903 |
+ uint64_t size = asf->data_size = avio_rl64(pb); |
|
904 |
+ int i; |
|
905 |
+ |
|
906 |
+ if (!asf->data_reached && pb->seekable) { |
|
907 |
+ asf->data_reached = 1; |
|
908 |
+ asf->data_offset = asf->offset; |
|
909 |
+ } |
|
910 |
+ |
|
911 |
+ for (i = 0; i < asf->nb_streams; i++) { |
|
912 |
+ if (!(asf->b_flags & ASF_FLAG_BROADCAST)) |
|
913 |
+ s->streams[i]->duration = asf->duration; |
|
914 |
+ } |
|
915 |
+ asf->nb_mult_left = 0; |
|
916 |
+ asf->sub_left = 0; |
|
917 |
+ asf->state = PARSE_PACKET_HEADER; |
|
918 |
+ asf->return_subpayload = 0; |
|
919 |
+ asf->packet_size_internal = 0; |
|
920 |
+ avio_skip(pb, 16); // skip File ID |
|
921 |
+ size = avio_rl64(pb); // Total Data Packets |
|
922 |
+ if (size != asf->nb_packets) |
|
923 |
+ av_log(s, AV_LOG_WARNING, |
|
924 |
+ "Number of Packets from File Properties Object is not equal to Total" |
|
925 |
+ "Datapackets value! num of packets %"PRIu64" total num %"PRIu64".\n", |
|
926 |
+ size, asf->nb_packets); |
|
927 |
+ avio_skip(pb, 2); // skip reserved field |
|
928 |
+ asf->first_packet_offset = avio_tell(pb); |
|
929 |
+ align_position(pb, asf->offset, asf->data_size); |
|
930 |
+ |
|
931 |
+ return 0; |
|
932 |
+} |
|
933 |
+ |
|
934 |
+static int asf_read_simple_index(AVFormatContext *s, const GUIDParseTable *g) |
|
935 |
+{ |
|
936 |
+ ASFContext *asf = s->priv_data; |
|
937 |
+ AVIOContext *pb = s->pb; |
|
938 |
+ AVStream *st = NULL; |
|
939 |
+ uint64_t interval; // index entry time interval in 100 ns units, usually it's 1s |
|
940 |
+ uint32_t pkt_num, nb_entries; |
|
941 |
+ int32_t prev_pkt_num = -1; |
|
942 |
+ int i; |
|
943 |
+ uint64_t size = avio_rl64(pb); |
|
944 |
+ |
|
945 |
+ // simple index objects should be ordered by stream number, this loop tries to find |
|
946 |
+ // the first not indexed video stream |
|
947 |
+ for (i = 0; i < asf->nb_streams; i++) { |
|
948 |
+ if ((asf->asf_st[i]->type == AVMEDIA_TYPE_VIDEO) && !asf->asf_st[i]->indexed) { |
|
949 |
+ asf->asf_st[i]->indexed = 1; |
|
950 |
+ st = s->streams[asf->asf_st[i]->index]; |
|
951 |
+ break; |
|
952 |
+ } |
|
953 |
+ } |
|
954 |
+ if (!st) { |
|
955 |
+ avio_skip(pb, size - 24); // if there's no video stream, skip index object |
|
956 |
+ return 0; |
|
957 |
+ } |
|
958 |
+ avio_skip(pb, 16); // skip File ID |
|
959 |
+ interval = avio_rl64(pb); |
|
960 |
+ avio_skip(pb, 4); |
|
961 |
+ nb_entries = avio_rl32(pb); |
|
962 |
+ for (i = 0; i < nb_entries; i++) { |
|
963 |
+ pkt_num = avio_rl32(pb); |
|
964 |
+ avio_skip(pb, 2); |
|
965 |
+ if (prev_pkt_num != pkt_num) { |
|
966 |
+ av_add_index_entry(st, asf->first_packet_offset + asf->packet_size * |
|
967 |
+ pkt_num, av_rescale(interval, i, 10000), |
|
968 |
+ asf->packet_size, 0, AVINDEX_KEYFRAME); |
|
969 |
+ prev_pkt_num = pkt_num; |
|
970 |
+ } |
|
971 |
+ } |
|
972 |
+ asf->is_simple_index = 1; |
|
973 |
+ align_position(pb, asf->offset, size); |
|
974 |
+ |
|
975 |
+ return 0; |
|
976 |
+} |
|
977 |
+ |
|
978 |
+static const GUIDParseTable gdef[] = { |
|
979 |
+ { "Data", { 0x75, 0xB2, 0x26, 0x36, 0x66, 0x8E, 0x11, 0xCF, 0xA6, 0xD9, 0x00, 0xAA, 0x00, 0x62, 0xCE, 0x6C }, asf_read_data, 1 }, |
|
980 |
+ { "Simple Index", { 0x33, 0x00, 0x08, 0x90, 0xE5, 0xB1, 0x11, 0xCF, 0x89, 0xF4, 0x00, 0xA0, 0xC9, 0x03, 0x49, 0xCB }, asf_read_simple_index, 1 }, |
|
981 |
+ { "Content Description", { 0x75, 0xB2, 0x26, 0x33, 0x66 ,0x8E, 0x11, 0xCF, 0xA6, 0xD9, 0x00, 0xAA, 0x00, 0x62, 0xCE, 0x6C }, asf_read_content_desc, 1 }, |
|
982 |
+ { "Extended Content Description", { 0xD2, 0xD0, 0xA4, 0x40, 0xE3, 0x07, 0x11, 0xD2, 0x97, 0xF0, 0x00, 0xA0, 0xC9, 0x5e, 0xA8, 0x50 }, asf_read_ext_content, 1 }, |
|
983 |
+ { "Stream Bitrate Properties", { 0x7B, 0xF8, 0x75, 0xCE, 0x46, 0x8D, 0x11, 0xD1, 0x8D, 0x82, 0x00, 0x60, 0x97, 0xC9, 0xA2, 0xB2 }, asf_read_unknown, 1 }, |
|
984 |
+ { "File Properties", { 0x8C, 0xAB, 0xDC, 0xA1, 0xA9, 0x47, 0x11, 0xCF, 0x8E, 0xE4, 0x00, 0xC0, 0x0C, 0x20, 0x53, 0x65 }, asf_read_properties, 1 }, |
|
985 |
+ { "Header Extension", { 0x5F, 0xBF, 0x03, 0xB5, 0xA9, 0x2E, 0x11, 0xCF, 0x8E, 0xE3, 0x00, 0xC0, 0x0C, 0x20, 0x53, 0x65 }, asf_read_unknown, 0 }, |
|
986 |
+ { "Stream Properties", { 0xB7, 0xDC, 0x07, 0x91, 0xA9, 0xB7, 0x11, 0xCF, 0x8E, 0xE6, 0x00, 0xC0, 0x0C, 0x20, 0x53, 0x65 }, asf_read_stream_properties, 1 }, |
|
987 |
+ { "Codec List", { 0x86, 0xD1, 0x52, 0x40, 0x31, 0x1D, 0x11, 0xD0, 0xA3, 0xA4, 0x00, 0xA0, 0xC9, 0x03, 0x48, 0xF6 }, asf_read_unknown, 1 }, |
|
988 |
+ { "Marker", { 0xF4, 0x87, 0xCD, 0x01, 0xA9, 0x51, 0x11, 0xCF, 0x8E, 0xE6, 0x00, 0xC0, 0x0C, 0x20, 0x53, 0x65 }, asf_read_marker, 1 }, |
|
989 |
+ { "Script Command", { 0x1E, 0xFB, 0x1A, 0x30, 0x0B, 0x62, 0x11, 0xD0, 0xA3, 0x9B, 0x00, 0xA0, 0xC9, 0x03, 0x48, 0xF6 }, asf_read_unknown, 1 }, |
|
990 |
+ { "Language List", { 0x7C, 0x43, 0x46, 0xa9, 0xef, 0xe0, 0x4B, 0xFC, 0xB2, 0x29, 0x39, 0x3e, 0xde, 0x41, 0x5c, 0x85 }, asf_read_language_list, 1}, |
|
991 |
+ { "Padding", { 0x18, 0x06, 0xD4, 0x74, 0xCA, 0xDF, 0x45, 0x09, 0xA4, 0xBA, 0x9A, 0xAB, 0xCB, 0x96, 0xAA, 0xE8 }, asf_read_unknown, 1 }, |
|
992 |
+ { "DRMv1 Header", { 0x22, 0x11, 0xB3, 0xFB, 0xBD, 0x23, 0x11, 0xD2, 0xB4, 0xB7, 0x00, 0xA0, 0xC9, 0x55, 0xFC, 0x6E }, asf_read_unknown, 1 }, |
|
993 |
+ { "DRMv2 Header", { 0x29, 0x8A, 0xE6, 0x14, 0x26, 0x22, 0x4C, 0x17, 0xB9, 0x35, 0xDA, 0xE0, 0x7E, 0xE9, 0x28, 0x9c }, asf_read_unknown, 1 }, |
|
994 |
+ { "Index", { 0xD6, 0xE2, 0x29, 0xD3, 0x35, 0xDA, 0x11, 0xD1, 0x90, 0x34, 0x00, 0xA0, 0xC9, 0x03, 0x49, 0xBE }, asf_read_unknown, 1 }, |
|
995 |
+ { "Media Object Index", { 0xFE, 0xB1, 0x03, 0xF8, 0x12, 0xAD, 0x4C, 0x64, 0x84, 0x0F, 0x2A, 0x1D, 0x2F, 0x7A, 0xD4, 0x8C }, asf_read_unknown, 1 }, |
|
996 |
+ { "Timecode Index", { 0x3C, 0xB7, 0x3F, 0xD0, 0x0C, 0x4A, 0x48, 0x03, 0x95, 0x3D, 0xED, 0xF7, 0xB6, 0x22, 0x8F, 0x0C }, asf_read_unknown, 0 }, |
|
997 |
+ { "Bitrate_Mutual_Exclusion", { 0xD6, 0xE2, 0x29, 0xDC, 0x35, 0xDA, 0x11, 0xD1, 0x90, 0x34, 0x00, 0xA0, 0xC9, 0x03, 0x49, 0xBE }, asf_read_unknown, 1 }, |
|
998 |
+ { "Error Correction", { 0x75, 0xB2, 0x26, 0x35, 0x66, 0x8E, 0x11, 0xCF, 0xA6, 0xD9, 0x00, 0xAA, 0x00, 0x62, 0xCE, 0x6C }, asf_read_unknown, 1 }, |
|
999 |
+ { "Content Branding", { 0x22, 0x11, 0xB3, 0xFA, 0xBD, 0x23, 0x11, 0xD2, 0xB4, 0xB7, 0x00, 0xA0, 0xC9, 0x55, 0xFC, 0x6E }, asf_read_unknown, 1 }, |
|
1000 |
+ { "Content Encryption", { 0x22, 0x11, 0xB3, 0xFB, 0xBD, 0x23, 0x11, 0xD2, 0xB4, 0xB7, 0x00, 0xA0, 0xC9, 0x55, 0xFC, 0x6E }, asf_read_unknown, 1 }, |
|
1001 |
+ { "Extended Content Encryption", { 0x29, 0x8A, 0xE6, 0x14, 0x26, 0x22, 0x4C, 0x17, 0xB9, 0x35, 0xDA, 0xE0, 0x7E, 0xE9, 0x28, 0x9C }, asf_read_unknown, 1 }, |
|
1002 |
+ { "Digital Signature", { 0x22, 0x11, 0xB3, 0xFC, 0xBD, 0x23, 0x11, 0xD2, 0xB4, 0xB7, 0x00, 0xA0, 0xC9, 0x55, 0xFC, 0x6E }, asf_read_unknown, 1 }, |
|
1003 |
+ { "Extended Stream Properties", { 0x14, 0xE6, 0xA5, 0xCB, 0xC6, 0x72, 0x43, 0x32, 0x83, 0x99, 0xA9, 0x69, 0x52, 0x06, 0x5B, 0x5A }, asf_read_ext_stream_properties, 1 }, |
|
1004 |
+ { "Advanced Mutual Exclusion", { 0xA0, 0x86, 0x49, 0xCF, 0x47, 0x75, 0x46, 0x70, 0x8A, 0x16, 0x6E, 0x35, 0x35, 0x75, 0x66, 0xCD }, asf_read_unknown, 1 }, |
|
1005 |
+ { "Group Mutual Exclusion", { 0xD1, 0x46, 0x5A, 0x40, 0x5A, 0x79, 0x43, 0x38, 0xB7, 0x1B, 0xE3, 0x6B, 0x8F, 0xD6, 0xC2, 0x49 }, asf_read_unknown, 1}, |
|
1006 |
+ { "Stream Prioritization", { 0xD4, 0xFE, 0xD1, 0x5B, 0x88, 0xD3, 0x45, 0x4F, 0x81, 0xF0, 0xED, 0x5C, 0x45, 0x99, 0x9E, 0x24 }, asf_read_unknown, 1 }, |
|
1007 |
+ { "Bandwidth Sharing Object", { 0xA6, 0x96, 0x09, 0xE6, 0x51, 0x7B, 0x11, 0xD2, 0xB6, 0xAF, 0x00, 0xC0, 0x4F, 0xD9, 0x08, 0xE9 }, asf_read_unknown, 1 }, |
|
1008 |
+ { "Metadata", { 0xC5, 0xF8, 0xCB, 0xEA, 0x5B, 0xAF, 0x48, 0x77, 0x84, 0x67, 0xAA, 0x8C, 0x44, 0xFA, 0x4C, 0xCA }, asf_read_metadata_obj, 1 }, |
|
1009 |
+ { "Metadata Library", { 0x44, 0x23, 0x1C, 0x94, 0x94, 0x98, 0x49, 0xD1, 0xA1, 0x41, 0x1D, 0x13, 0x4E, 0x45, 0x70, 0x54 }, asf_read_metadata_obj, 1 }, |
|
1010 |
+ { "Audio Spread", { 0xBF, 0xC3, 0xCD, 0x50, 0x61, 0x8F, 0x11, 0xCF, 0x8B, 0xB2, 0x00, 0xAA, 0x00, 0xB4, 0xE2, 0x20 }, asf_read_unknown, 1 }, |
|
1011 |
+ { "Index Parameters", { 0xD6, 0xE2, 0x29, 0xDF, 0x35, 0xDA, 0x11, 0xD1, 0x90, 0x34, 0x00, 0xA0, 0xC9, 0x03, 0x49, 0xBE }, asf_read_unknown, 1 }, |
|
1012 |
+ { "Content Encryption System Windows Media DRM Network Devices", |
|
1013 |
+ { 0x7A, 0x07, 0x9B, 0xB6, 0xDA, 0XA4, 0x4e, 0x12, 0xA5, 0xCA, 0x91, 0xD3, 0x8D, 0xC1, 0x1A, 0x8D }, asf_read_unknown, 1 }, |
|
1014 |
+ { "Mutex Language", { 0xD6, 0xE2, 0x2A, 0x00, 0x25, 0xDA, 0x11, 0xD1, 0x90, 0x34, 0x00, 0xA0, 0xC9, 0x03, 0x49, 0xBE }, asf_read_unknown, 1 }, |
|
1015 |
+ { "Mutex Bitrate", { 0xD6, 0xE2, 0x2A, 0x01, 0x25, 0xDA, 0x11, 0xD1, 0x90, 0x34, 0x00, 0xA0, 0xC9, 0x03, 0x49, 0xBE }, asf_read_unknown, 1 }, |
|
1016 |
+ { "Mutex Unknown", { 0xD6, 0xE2, 0x2A, 0x02, 0x25, 0xDA, 0x11, 0xD1, 0x90, 0x34, 0x00, 0xA0, 0xC9, 0x03, 0x49, 0xBE }, asf_read_unknown, 1 }, |
|
1017 |
+ { "Bandwith Sharing Exclusive", { 0xAF, 0x60, 0x60, 0xAA, 0x51, 0x97, 0x11, 0xD2, 0xB6, 0xAF, 0x00, 0xC0, 0x4F, 0xD9, 0x08, 0xE9 }, asf_read_unknown, 1 }, |
|
1018 |
+ { "Bandwith Sharing Partial", { 0xAF, 0x60, 0x60, 0xAB, 0x51, 0x97, 0x11, 0xD2, 0xB6, 0xAF, 0x00, 0xC0, 0x4F, 0xD9, 0x08, 0xE9 }, asf_read_unknown, 1 }, |
|
1019 |
+ { "Payload Extension System Timecode", { 0x39, 0x95, 0x95, 0xEC, 0x86, 0x67, 0x4E, 0x2D, 0x8F, 0xDB, 0x98, 0x81, 0x4C, 0xE7, 0x6C, 0x1E }, asf_read_unknown, 1 }, |
|
1020 |
+ { "Payload Extension System File Name", { 0xE1, 0x65, 0xEC, 0x0E, 0x19, 0xED, 0x45, 0xD7, 0xB4, 0xA7, 0x25, 0xCB, 0xD1, 0xE2, 0x8E, 0x9B }, asf_read_unknown, 1 }, |
|
1021 |
+ { "Payload Extension System Content Type", { 0xD5, 0x90, 0xDC, 0x20, 0x07, 0xBC, 0x43, 0x6C, 0x9C, 0xF7, 0xF3, 0xBB, 0xFB, 0xF1, 0xA4, 0xDC }, asf_read_unknown, 1 }, |
|
1022 |
+ { "Payload Extension System Pixel Aspect Ratio", { 0x1, 0x1E, 0xE5, 0x54, 0xF9, 0xEA, 0x4B, 0xC8, 0x82, 0x1A, 0x37, 0x6B, 0x74, 0xE4, 0xC4, 0xB8 }, asf_read_unknown, 1 }, |
|
1023 |
+ { "Payload Extension System Sample Duration", { 0xC6, 0xBD, 0x94, 0x50, 0x86, 0x7F, 0x49, 0x07, 0x83, 0xA3, 0xC7, 0x79, 0x21, 0xB7, 0x33, 0xAD }, asf_read_unknown, 1 }, |
|
1024 |
+ { "Payload Extension System Encryption Sample ID", { 0x66, 0x98, 0xB8, 0x4E, 0x0A, 0xFA, 0x43, 0x30, 0xAE, 0xB2, 0x1C, 0x0A, 0x98, 0xD7, 0xA4, 0x4D }, asf_read_unknown, 1 }, |
|
1025 |
+ { "Payload Extension System Degradable JPEG", { 0x00, 0xE1, 0xAF, 0x06, 0x7B, 0xEC, 0x11, 0xD1, 0xA5, 0x82, 0x00, 0xC0, 0x4F, 0xC2, 0x9C, 0xFB }, asf_read_unknown, 1 }, |
|
1026 |
+}; |
|
1027 |
+ |
|
1028 |
+#define READ_LEN(flag, name, len) \ |
|
1029 |
+ do { \ |
|
1030 |
+ if ((flag) == name ## IS_BYTE) \ |
|
1031 |
+ len = avio_r8(pb); \ |
|
1032 |
+ else if ((flag) == name ## IS_WORD) \ |
|
1033 |
+ len = avio_rl16(pb); \ |
|
1034 |
+ else if ((flag) == name ## IS_DWORD) \ |
|
1035 |
+ len = avio_rl32(pb); \ |
|
1036 |
+ else \ |
|
1037 |
+ len = 0; \ |
|
1038 |
+ } while(0) |
|
1039 |
+ |
|
1040 |
+static int asf_read_subpayload(AVFormatContext *s, AVPacket *pkt, int is_header) |
|
1041 |
+{ |
|
1042 |
+ ASFContext *asf = s->priv_data; |
|
1043 |
+ AVIOContext *pb = s->pb; |
|
1044 |
+ uint8_t sub_len; |
|
1045 |
+ int ret, i; |
|
1046 |
+ |
|
1047 |
+ if (is_header) { |
|
1048 |
+ asf->dts_delta = avio_r8(pb); |
|
1049 |
+ if (asf->nb_mult_left) { |
|
1050 |
+ asf->mult_sub_len = avio_rl16(pb); // total |
|
1051 |
+ } |
|
1052 |
+ asf->sub_header_offset = avio_tell(pb); |
|
1053 |
+ asf->nb_sub = 0; |
|
1054 |
+ asf->sub_left = 1; |
|
1055 |
+ } |
|
1056 |
+ sub_len = avio_r8(pb); |
|
1057 |
+ if ((ret = av_get_packet(pb, pkt, sub_len)) < 0) // each subpayload is entire frame |
|
1058 |
+ return ret; |
|
1059 |
+ for (i = 0; i < asf->nb_streams; i++) { |
|
1060 |
+ if (asf->stream_index == asf->asf_st[i]->stream_index) { |
|
1061 |
+ pkt->stream_index = asf->asf_st[i]->index; |
|
1062 |
+ break; |
|
1063 |
+ } |
|
1064 |
+ } |
|
1065 |
+ asf->return_subpayload = 1; |
|
1066 |
+ if (!sub_len) |
|
1067 |
+ asf->return_subpayload = 0; |
|
1068 |
+ |
|
1069 |
+ if (sub_len) |
|
1070 |
+ asf->nb_sub++; |
|
1071 |
+ pkt->dts = asf->sub_dts + (asf->nb_sub - 1) * asf->dts_delta - asf->preroll; |
|
1072 |
+ if (asf->nb_mult_left && (avio_tell(pb) >= |
|
1073 |
+ (asf->sub_header_offset + asf->mult_sub_len))) { |
|
1074 |
+ asf->sub_left = 0; |
|
1075 |
+ asf->nb_mult_left--; |
|
1076 |
+ } |
|
1077 |
+ if (avio_tell(pb) >= asf->packet_offset + asf->packet_size - asf->pad_len) { |
|
1078 |
+ asf->sub_left = 0; |
|
1079 |
+ if (!asf->nb_mult_left) { |
|
1080 |
+ avio_skip(pb, asf->pad_len); |
|
1081 |
+ if (avio_tell(pb) != asf->packet_offset + asf->packet_size) { |
|
1082 |
+ if (!asf->packet_size) |
|
1083 |
+ return AVERROR_INVALIDDATA; |
|
1084 |
+ av_log(s, AV_LOG_WARNING, |
|
1085 |
+ "Position %"PRId64" wrong, should be %"PRId64"\n", |
|
1086 |
+ avio_tell(pb), asf->packet_offset + asf->packet_size); |
|
1087 |
+ avio_seek(pb, asf->packet_offset + asf->packet_size, SEEK_SET); |
|
1088 |
+ } |
|
1089 |
+ } |
|
1090 |
+ } |
|
1091 |
+ |
|
1092 |
+ return 0; |
|
1093 |
+} |
|
1094 |
+ |
|
1095 |
+static void reset_packet(ASFPacket *asf_pkt) |
|
1096 |
+{ |
|
1097 |
+ asf_pkt->size_left = 0; |
|
1098 |
+ asf_pkt->data_size = 0; |
|
1099 |
+ asf_pkt->duration = 0; |
|
1100 |
+ asf_pkt->flags = 0; |
|
1101 |
+ asf_pkt->dts = 0; |
|
1102 |
+ asf_pkt->duration = 0; |
|
1103 |
+ av_free_packet(&asf_pkt->avpkt); |
|
1104 |
+ av_init_packet(&asf_pkt->avpkt); |
|
1105 |
+} |
|
1106 |
+ |
|
1107 |
+static int asf_read_multiple_payload(AVFormatContext *s, AVPacket *pkt, |
|
1108 |
+ ASFPacket *asf_pkt) |
|
1109 |
+{ |
|
1110 |
+ ASFContext *asf = s->priv_data; |
|
1111 |
+ AVIOContext *pb = s->pb; |
|
1112 |
+ uint16_t pay_len; |
|
1113 |
+ unsigned char *p; |
|
1114 |
+ int ret; |
|
1115 |
+ int skip = 0; |
|
1116 |
+ |
|
1117 |
+ // if replicated lenght is 1, subpayloads are present |
|
1118 |
+ if (asf->rep_data_len == 1) { |
|
1119 |
+ asf->sub_left = 1; |
|
1120 |
+ asf->state = READ_MULTI_SUB; |
|
1121 |
+ pkt->flags = asf_pkt->flags; |
|
1122 |
+ if ((ret = asf_read_subpayload(s, pkt, 1)) < 0) |
|
1123 |
+ return ret; |
|
1124 |
+ } else { |
|
1125 |
+ if (!asf_pkt->data_size) { |
|
1126 |
+ asf_pkt->data_size = asf_pkt->size_left = avio_rl32(pb); // read media object size |
|
1127 |
+ if (asf_pkt->data_size <= 0) |
|
1128 |
+ return AVERROR_EOF; |
|
1129 |
+ if ((ret = av_new_packet(&asf_pkt->avpkt, asf_pkt->data_size)) < 0) |
|
1130 |
+ return ret; |
|
1131 |
+ } else |
|
1132 |
+ avio_skip(pb, 4); // reading of media object size is already done |
|
1133 |
+ asf_pkt->dts = avio_rl32(pb); // read presentation time |
|
1134 |
+ if ((asf->rep_data_len - 8) > 0) |
|
1135 |
+ avio_skip(pb, asf->rep_data_len - 8); // skip replicated data |
|
1136 |
+ pay_len = avio_rl16(pb); // payload length should be WORD |
|
1137 |
+ if (pay_len > asf->packet_size) { |
|
1138 |
+ av_log(s, AV_LOG_ERROR, |
|
1139 |
+ "Error: invalid data packet size, pay_len %"PRIu16", " |
|
1140 |
+ "asf->packet_size %"PRIu32", offset %"PRId64".\n", |
|
1141 |
+ pay_len, asf->packet_size, avio_tell(pb)); |
|
1142 |
+ return AVERROR_INVALIDDATA; |
|
1143 |
+ } |
|
1144 |
+ p = asf_pkt->avpkt.data + asf_pkt->data_size - asf_pkt->size_left; |
|
1145 |
+ if (pay_len > asf_pkt->size_left) { |
|
1146 |
+ av_log(s, AV_LOG_ERROR, |
|
1147 |
+ "Error: invalid buffer size, pay_len %d, data size left %d.\n", |
|
1148 |
+ pay_len, asf_pkt->size_left); |
|
1149 |
+ skip = pay_len - asf_pkt->size_left; |
|
1150 |
+ pay_len = asf_pkt->size_left; |
|
1151 |
+ } |
|
1152 |
+ if ((ret = avio_read(pb, p, pay_len)) < 0) |
|
1153 |
+ return ret; |
|
1154 |
+ if (s->key && s->keylen == 20) |
|
1155 |
+ ff_asfcrypt_dec(s->key, p, ret); |
|
1156 |
+ avio_skip(pb, skip); |
|
1157 |
+ asf_pkt->size_left -= pay_len; |
|
1158 |
+ asf->nb_mult_left--; |
|
1159 |
+ } |
|
1160 |
+ |
|
1161 |
+ return 0; |
|
1162 |
+} |
|
1163 |
+ |
|
1164 |
+static int asf_read_single_payload(AVFormatContext *s, AVPacket *pkt, |
|
1165 |
+ ASFPacket *asf_pkt) |
|
1166 |
+{ |
|
1167 |
+ ASFContext *asf = s->priv_data; |
|
1168 |
+ AVIOContext *pb = s->pb; |
|
1169 |
+ int64_t offset; |
|
1170 |
+ uint64_t size; |
|
1171 |
+ unsigned char *p; |
|
1172 |
+ int ret; |
|
1173 |
+ |
|
1174 |
+ if (!asf_pkt->data_size) { |
|
1175 |
+ asf_pkt->data_size = asf_pkt->size_left = avio_rl32(pb); // read media object size |
|
1176 |
+ if (asf_pkt->data_size <= 0) |
|
1177 |
+ return AVERROR_EOF; |
|
1178 |
+ if ((ret = av_new_packet(&asf_pkt->avpkt, asf_pkt->data_size)) < 0) |
|
1179 |
+ return ret; |
|
1180 |
+ } else |
|
1181 |
+ avio_skip(pb, 4); // skip media object size |
|
1182 |
+ asf_pkt->dts = avio_rl32(pb); // read presentation time |
|
1183 |
+ if ((asf->rep_data_len - 8) > 0) |
|
1184 |
+ avio_skip(pb, asf->rep_data_len - 8); // skip replicated data |
|
1185 |
+ offset = avio_tell(pb); |
|
1186 |
+ |
|
1187 |
+ // size of the payload - size of the packet without header and padding |
|
1188 |
+ if (asf->packet_size_internal) |
|
1189 |
+ size = asf->packet_size_internal - offset + asf->packet_offset - asf->pad_len; |
|
1190 |
+ else |
|
1191 |
+ size = asf->packet_size - offset + asf->packet_offset - asf->pad_len; |
|
1192 |
+ if (size > asf->packet_size) { |
|
1193 |
+ av_log(s, AV_LOG_ERROR, |
|
1194 |
+ "Error: invalid data packet size, offset %"PRId64".\n", |
|
1195 |
+ avio_tell(pb)); |
|
1196 |
+ return AVERROR_INVALIDDATA; |
|
1197 |
+ } |
|
1198 |
+ p = asf_pkt->avpkt.data + asf_pkt->data_size - asf_pkt->size_left; |
|
1199 |
+ if (size > asf_pkt->size_left) |
|
1200 |
+ return AVERROR_INVALIDDATA; |
|
1201 |
+ if (asf_pkt->size_left > size) |
|
1202 |
+ asf_pkt->size_left -= size; |
|
1203 |
+ else |
|
1204 |
+ asf_pkt->size_left = 0; |
|
1205 |
+ if ((ret = avio_read(pb, p, size)) < 0) |
|
1206 |
+ return ret; |
|
1207 |
+ if (s->key && s->keylen == 20) |
|
1208 |
+ ff_asfcrypt_dec(s->key, p, ret); |
|
1209 |
+ if (asf->packet_size_internal) |
|
1210 |
+ avio_skip(pb, asf->packet_size - asf->packet_size_internal); |
|
1211 |
+ avio_skip(pb, asf->pad_len); // skip padding |
|
1212 |
+ |
|
1213 |
+ return 0; |
|
1214 |
+} |
|
1215 |
+ |
|
1216 |
+static int asf_read_payload(AVFormatContext *s, AVPacket *pkt) |
|
1217 |
+{ |
|
1218 |
+ ASFContext *asf = s->priv_data; |
|
1219 |
+ AVIOContext *pb = s->pb; |
|
1220 |
+ int ret, i; |
|
1221 |
+ ASFPacket *asf_pkt = NULL; |
|
1222 |
+ |
|
1223 |
+ if (!asf->sub_left) { |
|
1224 |
+ uint32_t off_len, media_len; |
|
1225 |
+ uint8_t stream_num; |
|
1226 |
+ |
|
1227 |
+ stream_num = avio_r8(pb); |
|
1228 |
+ asf->stream_index = stream_num & ASF_STREAM_NUM; |
|
1229 |
+ for (i = 0; i < asf->nb_streams; i++) { |
|
1230 |
+ if (asf->stream_index == asf->asf_st[i]->stream_index) { |
|
1231 |
+ asf_pkt = &asf->asf_st[i]->pkt; |
|
1232 |
+ asf_pkt->stream_index = asf->asf_st[i]->index; |
|
1233 |
+ asf_pkt->dts = asf->dts; |
|
1234 |
+ break; |
|
1235 |
+ } |
|
1236 |
+ } |
|
1237 |
+ if (!asf_pkt) |
|
1238 |
+ return AVERROR_INVALIDDATA; |
|
1239 |
+ if (stream_num >> 7) |
|
1240 |
+ asf_pkt->flags |= AV_PKT_FLAG_KEY; |
|
1241 |
+ READ_LEN(asf->prop_flags & ASF_PL_MASK_MEDIA_OBJECT_NUMBER_LENGTH_FIELD_SIZE, |
|
1242 |
+ ASF_PL_FLAG_MEDIA_OBJECT_NUMBER_LENGTH_FIELD_, media_len); |
|
1243 |
+ READ_LEN(asf->prop_flags & ASF_PL_MASK_OFFSET_INTO_MEDIA_OBJECT_LENGTH_FIELD_SIZE, |
|
1244 |
+ ASF_PL_FLAG_OFFSET_INTO_MEDIA_OBJECT_LENGTH_FIELD_, off_len); |
|
1245 |
+ READ_LEN(asf->prop_flags & ASF_PL_MASK_REPLICATED_DATA_LENGTH_FIELD_SIZE, |
|
1246 |
+ ASF_PL_FLAG_REPLICATED_DATA_LENGTH_FIELD_, asf->rep_data_len); |
|
1247 |
+ if (asf_pkt->size_left && (asf_pkt->frame_num != media_len)) { |
|
1248 |
+ av_log(s, AV_LOG_WARNING, "Unfinished frame will be ignored\n"); |
|
1249 |
+ reset_packet(asf_pkt); |
|
1250 |
+ } |
|
1251 |
+ asf_pkt->frame_num = media_len; |
|
1252 |
+ asf->sub_dts = off_len; |
|
1253 |
+ if (asf->nb_mult_left) { |
|
1254 |
+ if ((ret = asf_read_multiple_payload(s, pkt, asf_pkt)) < 0) |
|
1255 |
+ return ret; |
|
1256 |
+ } else if (asf->rep_data_len == 1) { |
|
1257 |
+ asf->sub_left = 1; |
|
1258 |
+ asf->state = READ_SINGLE; |
|
1259 |
+ pkt->flags = asf_pkt->flags; |
|
1260 |
+ if ((ret = asf_read_subpayload(s, pkt, 1)) < 0) |
|
1261 |
+ return ret; |
|
1262 |
+ } else { |
|
1263 |
+ if ((ret = asf_read_single_payload(s, pkt, asf_pkt)) < 0) |
|
1264 |
+ return ret; |
|
1265 |
+ } |
|
1266 |
+ } else { |
|
1267 |
+ for (i = 0; i <= asf->nb_streams; i++) { |
|
1268 |
+ if (asf->stream_index == asf->asf_st[i]->stream_index) { |
|
1269 |
+ asf_pkt = &asf->asf_st[i]->pkt; |
|
1270 |
+ break; |
|
1271 |
+ } |
|
1272 |
+ } |
|
1273 |
+ if (!asf_pkt) |
|
1274 |
+ return AVERROR_INVALIDDATA; |
|
1275 |
+ pkt->flags = asf_pkt->flags; |
|
1276 |
+ pkt->dts = asf_pkt->dts; |
|
1277 |
+ pkt->stream_index = asf->asf_st[i]->index; |
|
1278 |
+ if ((ret = asf_read_subpayload(s, pkt, 0)) < 0) // read subpayload without its header |
|
1279 |
+ return ret; |
|
1280 |
+ } |
|
1281 |
+ |
|
1282 |
+ return 0; |
|
1283 |
+} |
|
1284 |
+ |
|
1285 |
+static int asf_read_packet_header(AVFormatContext *s) |
|
1286 |
+{ |
|
1287 |
+ ASFContext *asf = s->priv_data; |
|
1288 |
+ AVIOContext *pb = s->pb; |
|
1289 |
+ uint64_t size; |
|
1290 |
+ uint32_t av_unused seq; |
|
1291 |
+ unsigned char error_flags, len_flags, pay_flags; |
|
1292 |
+ |
|
1293 |
+ asf->packet_offset = avio_tell(pb); |
|
1294 |
+ error_flags = avio_r8(pb); // read Error Correction Flags |
|
1295 |
+ if (error_flags & ASF_PACKET_FLAG_ERROR_CORRECTION_PRESENT) |
|
1296 |
+ if (!(error_flags & ASF_ERROR_CORRECTION_LENGTH_TYPE)) { |
|
1297 |
+ size = error_flags & ASF_PACKET_ERROR_CORRECTION_DATA_SIZE; |
|
1298 |
+ avio_skip(pb, size); |
|
1299 |
+ } |
|
1300 |
+ len_flags = avio_r8(pb); |
|
1301 |
+ asf->prop_flags = avio_r8(pb); |
|
1302 |
+ READ_LEN(len_flags & ASF_PPI_MASK_PACKET_LENGTH_FIELD_SIZE, |
|
1303 |
+ ASF_PPI_FLAG_PACKET_LENGTH_FIELD_, asf->packet_size_internal); |
|
1304 |
+ READ_LEN(len_flags & ASF_PPI_MASK_SEQUENCE_FIELD_SIZE, |
|
1305 |
+ ASF_PPI_FLAG_SEQUENCE_FIELD_, seq); |
|
1306 |
+ READ_LEN(len_flags & ASF_PPI_MASK_PADDING_LENGTH_FIELD_SIZE, |
|
1307 |
+ ASF_PPI_FLAG_PADDING_LENGTH_FIELD_, asf->pad_len ); |
|
1308 |
+ asf->send_time = avio_rl32(pb); // send time |
|
1309 |
+ avio_skip(pb, 2); // skip duration |
|
1310 |
+ if (len_flags & ASF_PPI_FLAG_MULTIPLE_PAYLOADS_PRESENT) { // Multiple Payloads present |
|
1311 |
+ pay_flags = avio_r8(pb); |
|
1312 |
+ asf->nb_mult_left = (pay_flags & ASF_NUM_OF_PAYLOADS); |
|
1313 |
+ } |
|
1314 |
+ |
|
1315 |
+ return 0; |
|
1316 |
+} |
|
1317 |
+ |
|
1318 |
+static int asf_deinterleave(AVFormatContext *s, ASFPacket *asf_pkt, int st_num) |
|
1319 |
+{ |
|
1320 |
+ ASFContext *asf = s->priv_data; |
|
1321 |
+ ASFStream *asf_st = asf->asf_st[st_num]; |
|
1322 |
+ unsigned char *p = asf_pkt->avpkt.data; |
|
1323 |
+ uint16_t pkt_len = asf->asf_st[st_num]->virtual_pkt_len; |
|
1324 |
+ uint16_t chunk_len = asf->asf_st[st_num]->virtual_chunk_len; |
|
1325 |
+ int nchunks = pkt_len / chunk_len; |
|
1326 |
+ AVPacket pkt; |
|
1327 |
+ int pos = 0, j, l, ret; |
|
1328 |
+ |
|
1329 |
+ |
|
1330 |
+ if ((ret = av_new_packet(&pkt, asf_pkt->data_size)) < 0) |
|
1331 |
+ return ret; |
|
1332 |
+ |
|
1333 |
+ while (asf_pkt->data_size >= asf_st->span * pkt_len + pos) { |
|
1334 |
+ if (pos >= asf_pkt->data_size) { |
|
1335 |
+ break; |
|
1336 |
+ } |
|
1337 |
+ for (l = 0; l < pkt_len; l++) { |
|
1338 |
+ if (pos >= asf_pkt->data_size) { |
|
1339 |
+ break; |
|
1340 |
+ } |
|
1341 |
+ for (j = 0; j < asf_st->span; j++) { |
|
1342 |
+ if ((pos + chunk_len) >= asf_pkt->data_size) |
|
1343 |
+ break; |
|
1344 |
+ memcpy(pkt.data + pos, |
|
1345 |
+ p + (j * nchunks + l) * chunk_len, |
|
1346 |
+ chunk_len); |
|
1347 |
+ pos += chunk_len; |
|
1348 |
+ } |
|
1349 |
+ } |
|
1350 |
+ p += asf_st->span * pkt_len; |
|
1351 |
+ if (p > asf_pkt->avpkt.data + asf_pkt->data_size) |
|
1352 |
+ break; |
|
1353 |
+ } |
|
1354 |
+ av_free_packet(&asf_pkt->avpkt); |
|
1355 |
+ asf_pkt->avpkt = pkt; |
|
1356 |
+ |
|
1357 |
+ return 0; |
|
1358 |
+} |
|
1359 |
+ |
|
1360 |
+static int asf_read_packet(AVFormatContext *s, AVPacket *pkt) |
|
1361 |
+{ |
|
1362 |
+ ASFContext *asf = s->priv_data; |
|
1363 |
+ AVIOContext *pb = s->pb; |
|
1364 |
+ int ret, i; |
|
1365 |
+ |
|
1366 |
+ if ((avio_tell(pb) >= asf->data_offset + asf->data_size) && |
|
1367 |
+ !(asf->b_flags & ASF_FLAG_BROADCAST)) |
|
1368 |
+ return AVERROR_EOF; |
|
1369 |
+ while (!pb->eof_reached) { |
|
1370 |
+ if (asf->state == PARSE_PACKET_HEADER) { |
|
1371 |
+ asf_read_packet_header(s); |
|
1372 |
+ if (!asf->nb_mult_left) |
|
1373 |
+ asf->state = READ_SINGLE; |
|
1374 |
+ else |
|
1375 |
+ asf->state = READ_MULTI; |
|
1376 |
+ } |
|
1377 |
+ if ((ret = asf_read_payload(s, pkt)) < 0) |
|
1378 |
+ return ret; |
|
1379 |
+ switch (asf->state) { |
|
1380 |
+ case READ_SINGLE: |
|
1381 |
+ if (!asf->sub_left) |
|
1382 |
+ asf->state = PARSE_PACKET_HEADER; |
|
1383 |
+ break; |
|
1384 |
+ case READ_MULTI_SUB: |
|
1385 |
+ if (!asf->sub_left && !asf->nb_mult_left) { |
|
1386 |
+ asf->state = PARSE_PACKET_HEADER; |
|
1387 |
+ if (!asf->return_subpayload) |
|
1388 |
+ avio_skip(pb, asf->pad_len); // skip padding |
|
1389 |
+ if (asf->packet_offset + asf->packet_size > avio_tell(pb)) |
|
1390 |
+ avio_seek(pb, asf->packet_offset + asf->packet_size, SEEK_SET); |
|
1391 |
+ } else if (!asf->sub_left) |
|
1392 |
+ asf->state = READ_MULTI; |
|
1393 |
+ break; |
|
1394 |
+ case READ_MULTI: |
|
1395 |
+ if (!asf->nb_mult_left) { |
|
1396 |
+ asf->state = PARSE_PACKET_HEADER; |
|
1397 |
+ if (!asf->return_subpayload) { |
|
1398 |
+ avio_skip(pb, asf->pad_len); // skip padding |
|
1399 |
+ } |
|
1400 |
+ if (asf->packet_offset + asf->packet_size > avio_tell(pb)) |
|
1401 |
+ avio_seek(pb, asf->packet_offset + asf->packet_size, SEEK_SET); |
|
1402 |
+ } |
|
1403 |
+ break; |
|
1404 |
+ } |
|
1405 |
+ if (asf->return_subpayload) { |
|
1406 |
+ asf->return_subpayload = 0; |
|
1407 |
+ return 0; |
|
1408 |
+ } |
|
1409 |
+ for (i = 0; i < s->nb_streams; i++) { |
|
1410 |
+ ASFPacket *asf_pkt = &asf->asf_st[i]->pkt; |
|
1411 |
+ if (asf_pkt && !asf_pkt->size_left && asf_pkt->data_size) { |
|
1412 |
+ if (asf->asf_st[i]->span > 1 && |
|
1413 |
+ asf->asf_st[i]->type == AVMEDIA_TYPE_AUDIO) |
|
1414 |
+ if ((ret = asf_deinterleave(s, asf_pkt, i)) < 0) |
|
1415 |
+ return ret; |
|
1416 |
+ av_packet_move_ref(pkt, &asf_pkt->avpkt); |
|
1417 |
+ pkt->stream_index = asf->asf_st[i]->index; |
|
1418 |
+ pkt->flags = asf_pkt->flags; |
|
1419 |
+ pkt->dts = asf_pkt->dts - asf->preroll; |
|
1420 |
+ asf_pkt->data_size = 0; |
|
1421 |
+ asf_pkt->frame_num = 0; |
|
1422 |
+ return 0; |
|
1423 |
+ } |
|
1424 |
+ } |
|
1425 |
+ } |
|
1426 |
+ |
|
1427 |
+ if (pb->eof_reached) |
|
1428 |
+ return AVERROR_EOF; |
|
1429 |
+ |
|
1430 |
+ return 0; |
|
1431 |
+} |
|
1432 |
+ |
|
1433 |
+static int asf_read_close(AVFormatContext *s) |
|
1434 |
+{ |
|
1435 |
+ ASFContext *asf = s->priv_data; |
|
1436 |
+ int i; |
|
1437 |
+ |
|
1438 |
+ for (i = 0; i < asf->nb_streams; i++) { |
|
1439 |
+ av_free_packet(&asf->asf_st[i]->pkt.avpkt); |
|
1440 |
+ av_freep(&asf->asf_st[i]); |
|
1441 |
+ av_dict_free(&asf->asf_sd[i].asf_met); |
|
1442 |
+ } |
|
1443 |
+ |
|
1444 |
+ return 0; |
|
1445 |
+} |
|
1446 |
+ |
|
1447 |
+static void reset_packet_state(AVFormatContext *s) |
|
1448 |
+{ |
|
1449 |
+ ASFContext *asf = s->priv_data; |
|
1450 |
+ int i; |
|
1451 |
+ |
|
1452 |
+ asf->state = PARSE_PACKET_HEADER; |
|
1453 |
+ asf->offset = 0; |
|
1454 |
+ asf->return_subpayload = 0; |
|
1455 |
+ asf->sub_left = 0; |
|
1456 |
+ asf->sub_header_offset = 0; |
|
1457 |
+ asf->packet_offset = asf->first_packet_offset; |
|
1458 |
+ asf->pad_len = 0; |
|
1459 |
+ asf->rep_data_len = 0; |
|
1460 |
+ asf->dts_delta = 0; |
|
1461 |
+ asf->mult_sub_len = 0; |
|
1462 |
+ asf->nb_mult_left = 0; |
|
1463 |
+ asf->nb_sub = 0; |
|
1464 |
+ asf->prop_flags = 0; |
|
1465 |
+ asf->sub_dts = 0; |
|
1466 |
+ asf->dts = 0; |
|
1467 |
+ for (i = 0; i < asf->nb_streams; i++) { |
|
1468 |
+ ASFPacket *pkt = &asf->asf_st[i]->pkt; |
|
1469 |
+ pkt->size_left = 0; |
|
1470 |
+ pkt->data_size = 0; |
|
1471 |
+ pkt->duration = 0; |
|
1472 |
+ pkt->flags = 0; |
|
1473 |
+ pkt->dts = 0; |
|
1474 |
+ pkt->duration = 0; |
|
1475 |
+ av_free_packet(&pkt->avpkt); |
|
1476 |
+ av_init_packet(&pkt->avpkt); |
|
1477 |
+ } |
|
1478 |
+} |
|
1479 |
+ |
|
1480 |
+/* |
|
1481 |
+ * Find a timestamp for the requested position within the payload |
|
1482 |
+ * where the pos (position) is the offset inside the Data Object. |
|
1483 |
+ * When position is not on the packet boundary, asf_read_timestamp tries |
|
1484 |
+ * to find the closest packet offset after this position. If this packet |
|
1485 |
+ * is a key frame, this packet timestamp is read and an index entry is created |
|
1486 |
+ * for the packet. If this packet belongs to the requested stream, |
|
1487 |
+ * asf_read_timestamp upgrades pos to the packet beginning offset and |
|
1488 |
+ * returns this packet's dts. So returned dts is the dts of the first key frame with |
|
1489 |
+ * matching stream number after given position. |
|
1490 |
+ */ |
|
1491 |
+static int64_t asf_read_timestamp(AVFormatContext *s, int stream_index, |
|
1492 |
+ int64_t *pos, int64_t pos_limit) |
|
1493 |
+{ |
|
1494 |
+ ASFContext *asf = s->priv_data; |
|
1495 |
+ int64_t pkt_pos = *pos, pkt_offset, dts = AV_NOPTS_VALUE, data_end; |
|
1496 |
+ AVPacket pkt; |
|
1497 |
+ int n; |
|
1498 |
+ |
|
1499 |
+ data_end = asf->data_offset + asf->data_size; |
|
1500 |
+ |
|
1501 |
+ n = (pkt_pos - asf->first_packet_offset + asf->packet_size - 1) / |
|
1502 |
+ asf->packet_size; |
|
1503 |
+ n = av_clip(n, 0, ((data_end - asf->first_packet_offset) / asf->packet_size - 1)); |
|
1504 |
+ pkt_pos = asf->first_packet_offset + n * asf->packet_size; |
|
1505 |
+ |
|
1506 |
+ avio_seek(s->pb, pkt_pos, SEEK_SET); |
|
1507 |
+ pkt_offset = pkt_pos; |
|
1508 |
+ |
|
1509 |
+ reset_packet_state(s); |
|
1510 |
+ while (avio_tell(s->pb) < data_end) { |
|
1511 |
+ |
|
1512 |
+ int i, ret, st_found; |
|
1513 |
+ |
|
1514 |
+ av_init_packet(&pkt); |
|
1515 |
+ pkt_offset = avio_tell(s->pb); |
|
1516 |
+ if ((ret = asf_read_packet(s, &pkt)) < 0) { |
|
1517 |
+ dts = AV_NOPTS_VALUE; |
|
1518 |
+ return ret; |
|
1519 |
+ } |
|
1520 |
+ // ASFPacket may contain fragments of packets belonging to different streams, |
|
1521 |
+ // pkt_offset is the offset of the first fragment within it. |
|
1522 |
+ if ((pkt_offset >= (pkt_pos + asf->packet_size))) |
|
1523 |
+ pkt_pos += asf->packet_size; |
|
1524 |
+ for (i = 0; i < asf->nb_streams; i++) { |
|
1525 |
+ ASFStream *st = asf->asf_st[i]; |
|
1526 |
+ |
|
1527 |
+ st_found = 0; |
|
1528 |
+ if (pkt.flags & AV_PKT_FLAG_KEY) { |
|
1529 |
+ dts = pkt.dts; |
|
1530 |
+ if (dts) { |
|
1531 |
+ av_add_index_entry(s->streams[pkt.stream_index], pkt_pos, |
|
1532 |
+ dts, pkt.size, 0, AVINDEX_KEYFRAME); |
|
1533 |
+ if (stream_index == st->index) { |
|
1534 |
+ st_found = 1; |
|
1535 |
+ break; |
|
1536 |
+ } |
|
1537 |
+ } |
|
1538 |
+ } |
|
1539 |
+ } |
|
1540 |
+ if (st_found) |
|
1541 |
+ break; |
|
1542 |
+ av_free_packet(&pkt); |
|
1543 |
+ } |
|
1544 |
+ *pos = pkt_pos; |
|
1545 |
+ |
|
1546 |
+ av_free_packet(&pkt); |
|
1547 |
+ return dts; |
|
1548 |
+} |
|
1549 |
+ |
|
1550 |
+static int asf_read_seek(AVFormatContext *s, int stream_index, |
|
1551 |
+ int64_t timestamp, int flags) |
|
1552 |
+{ |
|
1553 |
+ ASFContext *asf = s->priv_data; |
|
1554 |
+ int idx, ret; |
|
1555 |
+ |
|
1556 |
+ if (s->streams[stream_index]->nb_index_entries && asf->is_simple_index) { |
|
1557 |
+ idx = av_index_search_timestamp(s->streams[stream_index], timestamp, flags); |
|
1558 |
+ if (idx < 0 || idx >= s->streams[stream_index]->nb_index_entries) |
|
1559 |
+ return AVERROR_INVALIDDATA; |
|
1560 |
+ avio_seek(s->pb, s->streams[stream_index]->index_entries[idx].pos, SEEK_SET); |
|
1561 |
+ } else { |
|
1562 |
+ if ((ret = ff_seek_frame_binary(s, stream_index, timestamp, flags)) < 0) |
|
1563 |
+ return ret; |
|
1564 |
+ |
|
1565 |
+ // asf_read_timestamp is called inside ff_seek_frame_binary and leaves state dirty, |
|
1566 |
+ // so reset_packet_state have to be called after it. |
|
1567 |
+ reset_packet_state(s); |
|
1568 |
+ } |
|
1569 |
+ |
|
1570 |
+ return 0; |
|
1571 |
+} |
|
1572 |
+ |
|
1573 |
+static const GUIDParseTable *find_guid(ff_asf_guid guid) |
|
1574 |
+{ |
|
1575 |
+ int j, ret; |
|
1576 |
+ const GUIDParseTable *g; |
|
1577 |
+ |
|
1578 |
+ swap_guid(guid); |
|
1579 |
+ g = gdef; |
|
1580 |
+ for (j = 0; j < FF_ARRAY_ELEMS(gdef); j++) { |
|
1581 |
+ if (!(ret = memcmp(guid, g->guid, sizeof(g->guid)))) |
|
1582 |
+ return g; |
|
1583 |
+ g++; |
|
1584 |
+ } |
|
1585 |
+ |
|
1586 |
+ return NULL; |
|
1587 |
+} |
|
1588 |
+ |
|
1589 |
+static int detect_unknown_subobject(AVFormatContext *s, int64_t offset, int64_t size) |
|
1590 |
+{ |
|
1591 |
+ ASFContext *asf = s->priv_data; |
|
1592 |
+ AVIOContext *pb = s->pb; |
|
1593 |
+ const GUIDParseTable *g = NULL; |
|
1594 |
+ ff_asf_guid guid; |
|
1595 |
+ int ret; |
|
1596 |
+ |
|
1597 |
+ while (avio_tell(pb) <= offset + size) { |
|
1598 |
+ asf->offset = avio_tell(pb); |
|
1599 |
+ if ((ret = ff_get_guid(pb, &guid)) < 0) |
|
1600 |
+ return ret; |
|
1601 |
+ g = find_guid(guid); |
|
1602 |
+ if (g) { |
|
1603 |
+ if ((ret = g->read_object(s, g)) < 0) |
|
1604 |
+ return ret; |
|
1605 |
+ } else { |
|
1606 |
+ GUIDParseTable g2; |
|
1607 |
+ |
|
1608 |
+ g2.name = "Unknown"; |
|
1609 |
+ g2.is_subobject = 1; |
|
1610 |
+ asf_read_unknown(s, &g2); |
|
1611 |
+ } |
|
1612 |
+ } |
|
1613 |
+ |
|
1614 |
+ return 0; |
|
1615 |
+} |
|
1616 |
+ |
|
1617 |
+static int asf_read_header(AVFormatContext *s) |
|
1618 |
+{ |
|
1619 |
+ ASFContext *asf = s->priv_data; |
|
1620 |
+ AVIOContext *pb = s->pb; |
|
1621 |
+ const GUIDParseTable *g = NULL; |
|
1622 |
+ ff_asf_guid guid; |
|
1623 |
+ int i, ret; |
|
1624 |
+ uint64_t size; |
|
1625 |
+ |
|
1626 |
+ asf->preroll = 0; |
|
1627 |
+ asf->is_simple_index = 0; |
|
1628 |
+ ff_get_guid(pb, &guid); |
|
1629 |
+ if (ff_guidcmp(&guid, &ff_asf_header)) |
|
1630 |
+ return AVERROR_INVALIDDATA; |
|
1631 |
+ avio_skip(pb, 8); // skip header object size |
|
1632 |
+ avio_skip(pb, 6); // skip number of header objects and 2 reserved bytes |
|
1633 |
+ asf->data_reached = 0; |
|
1634 |
+ |
|
1635 |
+ /* 1 is here instead of pb->eof_reached because (when not streaming), Data are skipped |
|
1636 |
+ * for the first time, |
|
1637 |
+ * Index object is processed and got eof and then seeking back to the Data is performed. |
|
1638 |
+ */ |
|
1639 |
+ while (1) { |
|
1640 |
+ // for the cases when object size is invalid |
|
1641 |
+ if (avio_tell(pb) == asf->offset) { |
|
1642 |
+ if (asf->data_reached) |
|
1643 |
+ avio_seek(pb, asf->first_packet_offset, SEEK_SET); |
|
1644 |
+ break; |
|
1645 |
+ } |
|
1646 |
+ asf->offset = avio_tell(pb); |
|
1647 |
+ if ((ret = ff_get_guid(pb, &guid)) < 0) { |
|
1648 |
+ if (ret == AVERROR_EOF && asf->data_reached) { |
|
1649 |
+ avio_seek(pb, asf->first_packet_offset, SEEK_SET); |
|
1650 |
+ break; |
|
1651 |
+ } else |
|
1652 |
+ return ret; |
|
1653 |
+ } |
|
1654 |
+ g = find_guid(guid); |
|
1655 |
+ if (g) { |
|
1656 |
+ asf->unknown_offset = asf->offset; |
|
1657 |
+ asf->is_header = 1; |
|
1658 |
+ if ((ret = g->read_object(s, g)) < 0) |
|
1659 |
+ return ret; |
|
1660 |
+ } else { |
|
1661 |
+ size = avio_rl64(pb); |
|
1662 |
+ align_position(pb, asf->offset, size); |
|
1663 |
+ } |
|
1664 |
+ if (asf->data_reached && !pb->seekable) |
|
1665 |
+ break; |
|
1666 |
+ } |
|
1667 |
+ |
|
1668 |
+ for (i = 0; i < asf->nb_streams; i++) { |
|
1669 |
+ const char *rfc1766 = asf->asf_sd[asf->asf_st[i]->lang_idx].langs; |
|
1670 |
+ AVStream *st = s->streams[asf->asf_st[i]->index]; |
|
1671 |
+ set_language(s, rfc1766, &st->metadata); |
|
1672 |
+ } |
|
1673 |
+ |
|
1674 |
+ for (i = 0; i < ASF_MAX_STREAMS; i++) { |
|
1675 |
+ AVStream *st = NULL; |
|
1676 |
+ |
|
1677 |
+ st = find_stream(s, i); |
|
1678 |
+ if (st) { |
|
1679 |
+ av_dict_copy(&st->metadata, asf->asf_sd[i].asf_met, AV_DICT_IGNORE_SUFFIX); |
|
1680 |
+ if (asf->asf_sd[i].aspect_ratio.num > 0 && asf->asf_sd[i].aspect_ratio.den > 0) { |
|
1681 |
+ st->sample_aspect_ratio.num = asf->asf_sd[i].aspect_ratio.num; |
|
1682 |
+ st->sample_aspect_ratio.den = asf->asf_sd[i].aspect_ratio.den; |
|
1683 |
+ } |
|
1684 |
+ } |
|
1685 |
+ } |
|
1686 |
+ |
|
1687 |
+ return 0; |
|
1688 |
+} |
|
1689 |
+ |
|
1690 |
+AVInputFormat ff_asf_o_demuxer = { |
|
1691 |
+ .name = "asf_o", |
|
1692 |
+ .long_name = NULL_IF_CONFIG_SMALL("ASF (Advanced / Active Streaming Format)"), |
|
1693 |
+ .priv_data_size = sizeof(ASFContext), |
|
1694 |
+ .read_probe = asf_probe, |
|
1695 |
+ .read_header = asf_read_header, |
|
1696 |
+ .read_packet = asf_read_packet, |
|
1697 |
+ .read_close = asf_read_close, |
|
1698 |
+ .read_timestamp = asf_read_timestamp, |
|
1699 |
+ .read_seek = asf_read_seek, |
|
1700 |
+ .flags = AVFMT_NOBINSEARCH | AVFMT_NOGENSEARCH, |
|
1701 |
+}; |