FFmpeg
 All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Macros Groups Pages
wtvdec.c
Go to the documentation of this file.
1 /*
2  * Windows Television (WTV) demuxer
3  * Copyright (c) 2010-2011 Peter Ross <pross@xvid.org>
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 /**
23  * @file
24  * Windows Television (WTV) demuxer
25  * @author Peter Ross <pross@xvid.org>
26  */
27 
28 #include <inttypes.h>
29 
31 #include "libavutil/intreadwrite.h"
32 #include "libavutil/intfloat.h"
34 #include "avformat.h"
35 #include "internal.h"
36 #include "wtv.h"
37 #include "mpegts.h"
38 
39 /* Macros for formating GUIDs */
40 #define PRI_PRETTY_GUID \
41  "%08"PRIx32"-%04"PRIx16"-%04"PRIx16"-%02x%02x%02x%02x%02x%02x%02x%02x"
42 #define ARG_PRETTY_GUID(g) \
43  AV_RL32(g),AV_RL16(g+4),AV_RL16(g+6),g[8],g[9],g[10],g[11],g[12],g[13],g[14],g[15]
44 #define LEN_PRETTY_GUID 34
45 
46 /*
47  *
48  * File system routines
49  *
50  */
51 
52 typedef struct WtvFile {
53  AVIOContext *pb_filesystem; /**< file system (AVFormatContext->pb) */
54 
55  int sector_bits; /**< sector shift bits; used to convert sector number into pb_filesystem offset */
56  uint32_t *sectors; /**< file allocation table */
57  int nb_sectors; /**< number of sectors */
58 
59  int error;
60  int64_t position;
61  int64_t length;
62 } WtvFile;
63 
64 static int64_t seek_by_sector(AVIOContext *pb, int64_t sector, int64_t offset)
65 {
66  return avio_seek(pb, (sector << WTV_SECTOR_BITS) + offset, SEEK_SET);
67 }
68 
69 /**
70  * @return bytes read, 0 on end of file, or <0 on error
71  */
72 static int wtvfile_read_packet(void *opaque, uint8_t *buf, int buf_size)
73 {
74  WtvFile *wf = opaque;
75  AVIOContext *pb = wf->pb_filesystem;
76  int nread = 0;
77 
78  if (wf->error || pb->error)
79  return -1;
80  if (wf->position >= wf->length || avio_feof(pb))
81  return 0;
82 
83  buf_size = FFMIN(buf_size, wf->length - wf->position);
84  while(nread < buf_size) {
85  int n;
86  int remaining_in_sector = (1 << wf->sector_bits) - (wf->position & ((1 << wf->sector_bits) - 1));
87  int read_request = FFMIN(buf_size - nread, remaining_in_sector);
88 
89  n = avio_read(pb, buf, read_request);
90  if (n <= 0)
91  break;
92  nread += n;
93  buf += n;
94  wf->position += n;
95  if (n == remaining_in_sector) {
96  int i = wf->position >> wf->sector_bits;
97  if (i >= wf->nb_sectors ||
98  (wf->sectors[i] != wf->sectors[i - 1] + (1 << (wf->sector_bits - WTV_SECTOR_BITS)) &&
99  seek_by_sector(pb, wf->sectors[i], 0) < 0)) {
100  wf->error = 1;
101  break;
102  }
103  }
104  }
105  return nread;
106 }
107 
108 /**
109  * @return position (or file length)
110  */
111 static int64_t wtvfile_seek(void *opaque, int64_t offset, int whence)
112 {
113  WtvFile *wf = opaque;
114  AVIOContext *pb = wf->pb_filesystem;
115 
116  if (whence == AVSEEK_SIZE)
117  return wf->length;
118  else if (whence == SEEK_CUR)
119  offset = wf->position + offset;
120  else if (whence == SEEK_END)
121  offset = wf->length;
122 
123  wf->error = offset < 0 || offset >= wf->length ||
124  seek_by_sector(pb, wf->sectors[offset >> wf->sector_bits],
125  offset & ((1 << wf->sector_bits) - 1)) < 0;
126  wf->position = offset;
127  return offset;
128 }
129 
130 /**
131  * read non-zero integers (le32) from input stream
132  * @param pb
133  * @param[out] data destination
134  * @param count maximum number of integers to read
135  * @return total number of integers read
136  */
137 static int read_ints(AVIOContext *pb, uint32_t *data, int count)
138 {
139  int i, total = 0;
140  for (i = 0; i < count; i++) {
141  if ((data[total] = avio_rl32(pb)))
142  total++;
143  }
144  return total;
145 }
146 
147 /**
148  * Open file
149  * @param first_sector First sector
150  * @param length Length of file (bytes)
151  * @param depth File allocation table depth
152  * @return NULL on error
153  */
154 static AVIOContext * wtvfile_open_sector(int first_sector, uint64_t length, int depth, AVFormatContext *s)
155 {
156  AVIOContext *pb;
157  WtvFile *wf;
158  uint8_t *buffer;
159  int64_t size;
160 
161  if (seek_by_sector(s->pb, first_sector, 0) < 0)
162  return NULL;
163 
164  wf = av_mallocz(sizeof(WtvFile));
165  if (!wf)
166  return NULL;
167 
168  if (depth == 0) {
169  wf->sectors = av_malloc(sizeof(uint32_t));
170  if (!wf->sectors) {
171  av_free(wf);
172  return NULL;
173  }
174  wf->sectors[0] = first_sector;
175  wf->nb_sectors = 1;
176  } else if (depth == 1) {
178  if (!wf->sectors) {
179  av_free(wf);
180  return NULL;
181  }
182  wf->nb_sectors = read_ints(s->pb, wf->sectors, WTV_SECTOR_SIZE / 4);
183  } else if (depth == 2) {
184  uint32_t sectors1[WTV_SECTOR_SIZE / 4];
185  int nb_sectors1 = read_ints(s->pb, sectors1, WTV_SECTOR_SIZE / 4);
186  int i;
187 
188  wf->sectors = av_malloc_array(nb_sectors1, 1 << WTV_SECTOR_BITS);
189  if (!wf->sectors) {
190  av_free(wf);
191  return NULL;
192  }
193  wf->nb_sectors = 0;
194  for (i = 0; i < nb_sectors1; i++) {
195  if (seek_by_sector(s->pb, sectors1[i], 0) < 0)
196  break;
197  wf->nb_sectors += read_ints(s->pb, wf->sectors + i * WTV_SECTOR_SIZE / 4, WTV_SECTOR_SIZE / 4);
198  }
199  } else {
200  av_log(s, AV_LOG_ERROR, "unsupported file allocation table depth (0x%x)\n", depth);
201  av_free(wf);
202  return NULL;
203  }
204  wf->sector_bits = length & (1ULL<<63) ? WTV_SECTOR_BITS : WTV_BIGSECTOR_BITS;
205 
206  if (!wf->nb_sectors) {
207  av_freep(&wf->sectors);
208  av_freep(&wf);
209  return NULL;
210  }
211 
212  size = avio_size(s->pb);
213  if (size >= 0 && (int64_t)wf->sectors[wf->nb_sectors - 1] << WTV_SECTOR_BITS > size)
214  av_log(s, AV_LOG_WARNING, "truncated file\n");
215 
216  /* check length */
217  length &= 0xFFFFFFFFFFFF;
218  if (length > ((int64_t)wf->nb_sectors << wf->sector_bits)) {
219  av_log(s, AV_LOG_WARNING, "reported file length (0x%"PRIx64") exceeds number of available sectors (0x%"PRIx64")\n", length, (int64_t)wf->nb_sectors << wf->sector_bits);
220  length = (int64_t)wf->nb_sectors << wf->sector_bits;
221  }
222  wf->length = length;
223 
224  /* seek to initial sector */
225  wf->position = 0;
226  if (seek_by_sector(s->pb, wf->sectors[0], 0) < 0) {
227  av_freep(&wf->sectors);
228  av_freep(&wf);
229  return NULL;
230  }
231 
232  wf->pb_filesystem = s->pb;
233  buffer = av_malloc(1 << wf->sector_bits);
234  if (!buffer) {
235  av_freep(&wf->sectors);
236  av_freep(&wf);
237  return NULL;
238  }
239 
240  pb = avio_alloc_context(buffer, 1 << wf->sector_bits, 0, wf,
242  if (!pb) {
243  av_freep(&buffer);
244  av_freep(&wf->sectors);
245  av_freep(&wf);
246  }
247  return pb;
248 }
249 
250 /**
251  * Open file using filename
252  * @param[in] buf directory buffer
253  * @param buf_size directory buffer size
254  * @param[in] filename
255  * @param filename_size size of filename
256  * @return NULL on error
257  */
258 static AVIOContext * wtvfile_open2(AVFormatContext *s, const uint8_t *buf, int buf_size, const uint8_t *filename, int filename_size)
259 {
260  const uint8_t *buf_end = buf + buf_size;
261 
262  while(buf + 48 <= buf_end) {
263  int dir_length, name_size, first_sector, depth;
264  uint64_t file_length;
265  const uint8_t *name;
266  if (ff_guidcmp(buf, ff_dir_entry_guid)) {
267  av_log(s, AV_LOG_ERROR, "unknown guid "FF_PRI_GUID", expected dir_entry_guid; "
268  "remaining directory entries ignored\n", FF_ARG_GUID(buf));
269  break;
270  }
271  dir_length = AV_RL16(buf + 16);
272  file_length = AV_RL64(buf + 24);
273  name_size = 2 * AV_RL32(buf + 32);
274  if (name_size < 0) {
275  av_log(s, AV_LOG_ERROR,
276  "bad filename length, remaining directory entries ignored\n");
277  break;
278  }
279  if (48 + (int64_t)name_size > buf_end - buf) {
280  av_log(s, AV_LOG_ERROR, "filename exceeds buffer size; remaining directory entries ignored\n");
281  break;
282  }
283  first_sector = AV_RL32(buf + 40 + name_size);
284  depth = AV_RL32(buf + 44 + name_size);
285 
286  /* compare file name; test optional null terminator */
287  name = buf + 40;
288  if (name_size >= filename_size &&
289  !memcmp(name, filename, filename_size) &&
290  (name_size < filename_size + 2 || !AV_RN16(name + filename_size)))
291  return wtvfile_open_sector(first_sector, file_length, depth, s);
292 
293  buf += dir_length;
294  }
295  return 0;
296 }
297 
298 #define wtvfile_open(s, buf, buf_size, filename) \
299  wtvfile_open2(s, buf, buf_size, filename, sizeof(filename))
300 
301 /**
302  * Close file opened with wtvfile_open_sector(), or wtv_open()
303  */
304 static void wtvfile_close(AVIOContext *pb)
305 {
306  WtvFile *wf = pb->opaque;
307  av_freep(&wf->sectors);
308  av_freep(&pb->opaque);
309  av_freep(&pb->buffer);
310  av_free(pb);
311 }
312 
313 /*
314  *
315  * Main demuxer
316  *
317  */
318 
319 typedef struct WtvStream {
321 } WtvStream;
322 
323 typedef struct WtvContext {
324  AVIOContext *pb; /**< timeline file */
325  int64_t epoch;
326  int64_t pts; /**< pts for next data chunk */
327  int64_t last_valid_pts; /**< latest valid pts, used for interative seeking */
328 
329  /* maintain private seek index, as the AVIndexEntry->pos is relative to the
330  start of the 'timeline' file, not the file system (AVFormatContext->pb) */
334 } WtvContext;
335 
336 /* WTV GUIDs */
338  {0x48,0xC0,0xCE,0x5D,0xB9,0xD0,0x63,0x41,0x87,0x2C,0x4F,0x32,0x22,0x3B,0xE8,0x8A};
340  {0x6D,0x66,0x92,0xE2,0x02,0x9C,0x8D,0x44,0xAA,0x8D,0x78,0x1A,0x93,0xFD,0xC3,0x95};
342  {0x1C,0xD4,0x7B,0x10,0xDA,0xA6,0x91,0x46,0x83,0x69,0x11,0xB2,0xCD,0xAA,0x28,0x8E};
344  {0xE6,0xA2,0xB4,0x3A,0x47,0x42,0x34,0x4B,0x89,0x6C,0x30,0xAF,0xA5,0xD2,0x1C,0x24};
346  {0xD9,0x79,0xE7,0xEf,0xF0,0x97,0x86,0x47,0x80,0x0D,0x95,0xCF,0x50,0x5D,0xDC,0x66};
348  {0xC4,0xE1,0xD4,0x4B,0xA1,0x90,0x09,0x41,0x82,0x36,0x27,0xF0,0x0E,0x7D,0xCC,0x5B};
350  {0x68,0xAB,0xF1,0xCA,0x53,0xE1,0x41,0x4D,0xA6,0xB3,0xA7,0xC9,0x98,0xDB,0x75,0xEE};
352  {0x50,0xD9,0x99,0x95,0x33,0x5F,0x17,0x46,0xAF,0x7C,0x1E,0x54,0xB5,0x10,0xDA,0xA3};
354  {0xBE,0xBF,0x1C,0x50,0x49,0xB8,0xCE,0x42,0x9B,0xE9,0x3D,0xB8,0x69,0xFB,0x82,0xB3};
355 
356 /* Windows media GUIDs */
357 
358 /* Media types */
360  {0x81,0xEB,0x36,0xE4,0x4F,0x52,0xCE,0x11,0x9F,0x53,0x00,0x20,0xAF,0x0B,0xA7,0x70};
362  {0x6C,0x17,0x5F,0x45,0x06,0x4B,0xCE,0x47,0x9A,0xEF,0x8C,0xAE,0xF7,0x3D,0xF7,0xB5};
364  {0x20,0x80,0x6D,0xE0,0x46,0xDB,0xCF,0x11,0xB4,0xD1,0x00,0x80,0x5F,0x6C,0xBB,0xEA};
366  {0x89,0x8A,0x8B,0xB8,0x49,0xB0,0x80,0x4C,0xAD,0xCF,0x58,0x98,0x98,0x5E,0x22,0xC1};
367 
368 /* Media subtypes */
370  {0xC3,0xCB,0xFF,0x34,0xB3,0xD5,0x71,0x41,0x90,0x02,0xD4,0xC6,0x03,0x01,0x69,0x7F};
372  {0xE3,0x76,0x2A,0xF7,0x0A,0xEB,0xD0,0x11,0xAC,0xE4,0x00,0x00,0xC0,0xCC,0x16,0xBA};
374  {0xAA,0xDD,0x2A,0xF5,0xF0,0x36,0xF5,0x43,0x95,0xEA,0x6D,0x86,0x64,0x84,0x26,0x2A};
376  {0x79,0x85,0x9F,0x4A,0xF8,0x6B,0x92,0x43,0x8A,0x6D,0xD2,0xDD,0x09,0xFA,0x78,0x61};
377 
378 static int read_probe(AVProbeData *p)
379 {
380  return ff_guidcmp(p->buf, ff_wtv_guid) ? 0 : AVPROBE_SCORE_MAX;
381 }
382 
383 /**
384  * Convert win32 FILETIME to ISO-8601 string
385  * @return <0 on error
386  */
387 static int filetime_to_iso8601(char *buf, int buf_size, int64_t value)
388 {
389  time_t t = (value / 10000000LL) - 11644473600LL;
390  struct tm tmbuf;
391  struct tm *tm = gmtime_r(&t, &tmbuf);
392  if (!tm)
393  return -1;
394  if (!strftime(buf, buf_size, "%Y-%m-%d %H:%M:%S", tm))
395  return -1;
396  return 0;
397 }
398 
399 /**
400  * Convert crazy time (100ns since 1 Jan 0001) to ISO-8601 string
401  * @return <0 on error
402  */
403 static int crazytime_to_iso8601(char *buf, int buf_size, int64_t value)
404 {
405  time_t t = (value / 10000000LL) - 719162LL*86400LL;
406  struct tm tmbuf;
407  struct tm *tm = gmtime_r(&t, &tmbuf);
408  if (!tm)
409  return -1;
410  if (!strftime(buf, buf_size, "%Y-%m-%d %H:%M:%S", tm))
411  return -1;
412  return 0;
413 }
414 
415 /**
416  * Convert OLE DATE to ISO-8601 string
417  * @return <0 on error
418  */
419 static int oledate_to_iso8601(char *buf, int buf_size, int64_t value)
420 {
421  time_t t = (av_int2double(value) - 25569.0) * 86400;
422  struct tm tmbuf;
423  struct tm *tm= gmtime_r(&t, &tmbuf);
424  if (!tm)
425  return -1;
426  if (!strftime(buf, buf_size, "%Y-%m-%d %H:%M:%S", tm))
427  return -1;
428  return 0;
429 }
430 
432 {
433  char mime[1024];
434  char description[1024];
435  unsigned int filesize;
436  AVStream *st;
437  int ret;
438  int64_t pos = avio_tell(pb);
439 
440  avio_get_str16le(pb, INT_MAX, mime, sizeof(mime));
441  if (strcmp(mime, "image/jpeg"))
442  goto done;
443 
444  avio_r8(pb);
445  avio_get_str16le(pb, INT_MAX, description, sizeof(description));
446  filesize = avio_rl32(pb);
447  if (!filesize)
448  goto done;
449 
450  st = avformat_new_stream(s, NULL);
451  if (!st)
452  goto done;
453  av_dict_set(&st->metadata, "title", description, 0);
456  st->id = -1;
457  ret = av_get_packet(pb, &st->attached_pic, filesize);
458  if (ret < 0)
459  goto done;
460  st->attached_pic.stream_index = st->index;
463 done:
464  avio_seek(pb, pos + length, SEEK_SET);
465 }
466 
467 static void get_tag(AVFormatContext *s, AVIOContext *pb, const char *key, int type, int length)
468 {
469  int buf_size;
470  char *buf;
471 
472  if (!strcmp(key, "WM/MediaThumbType")) {
473  avio_skip(pb, length);
474  return;
475  }
476 
477  buf_size = FFMAX(2*length, LEN_PRETTY_GUID) + 1;
478  buf = av_malloc(buf_size);
479  if (!buf)
480  return;
481 
482  if (type == 0 && length == 4) {
483  snprintf(buf, buf_size, "%u", avio_rl32(pb));
484  } else if (type == 1) {
485  avio_get_str16le(pb, length, buf, buf_size);
486  if (!strlen(buf)) {
487  av_free(buf);
488  return;
489  }
490  } else if (type == 3 && length == 4) {
491  strcpy(buf, avio_rl32(pb) ? "true" : "false");
492  } else if (type == 4 && length == 8) {
493  int64_t num = avio_rl64(pb);
494  if (!strcmp(key, "WM/EncodingTime") ||
495  !strcmp(key, "WM/MediaOriginalBroadcastDateTime")) {
496  if (filetime_to_iso8601(buf, buf_size, num) < 0) {
497  av_free(buf);
498  return;
499  }
500  } else if (!strcmp(key, "WM/WMRVEncodeTime") ||
501  !strcmp(key, "WM/WMRVEndTime")) {
502  if (crazytime_to_iso8601(buf, buf_size, num) < 0) {
503  av_free(buf);
504  return;
505  }
506  } else if (!strcmp(key, "WM/WMRVExpirationDate")) {
507  if (oledate_to_iso8601(buf, buf_size, num) < 0 ) {
508  av_free(buf);
509  return;
510  }
511  } else if (!strcmp(key, "WM/WMRVBitrate"))
512  snprintf(buf, buf_size, "%f", av_int2double(num));
513  else
514  snprintf(buf, buf_size, "%"PRIi64, num);
515  } else if (type == 5 && length == 2) {
516  snprintf(buf, buf_size, "%u", avio_rl16(pb));
517  } else if (type == 6 && length == 16) {
518  ff_asf_guid guid;
519  avio_read(pb, guid, 16);
520  snprintf(buf, buf_size, PRI_PRETTY_GUID, ARG_PRETTY_GUID(guid));
521  } else if (type == 2 && !strcmp(key, "WM/Picture")) {
522  get_attachment(s, pb, length);
523  av_freep(&buf);
524  return;
525  } else {
526  av_freep(&buf);
527  av_log(s, AV_LOG_WARNING, "unsupported metadata entry; key:%s, type:%d, length:0x%x\n", key, type, length);
528  avio_skip(pb, length);
529  return;
530  }
531 
532  av_dict_set(&s->metadata, key, buf, 0);
533  av_freep(&buf);
534 }
535 
536 /**
537  * Parse metadata entries
538  */
540 {
541  ff_asf_guid guid;
542  int length, type;
543  while(!avio_feof(pb)) {
544  char key[1024];
545  ff_get_guid(pb, &guid);
546  type = avio_rl32(pb);
547  length = avio_rl32(pb);
548  if (!length)
549  break;
550  if (ff_guidcmp(&guid, ff_metadata_guid)) {
551  av_log(s, AV_LOG_WARNING, "unknown guid "FF_PRI_GUID", expected metadata_guid; "
552  "remaining metadata entries ignored\n", FF_ARG_GUID(guid));
553  break;
554  }
555  avio_get_str16le(pb, INT_MAX, key, sizeof(key));
556  get_tag(s, pb, key, type, length);
557  }
558 
560 }
561 
562 /**
563  * parse VIDEOINFOHEADER2 structure
564  * @return bytes consumed
565  */
567 {
568  WtvContext *wtv = s->priv_data;
569  AVIOContext *pb = wtv->pb;
570 
571  avio_skip(pb, 72); // picture aspect ratio is unreliable
572  st->codec->codec_tag = ff_get_bmp_header(pb, st, NULL);
573 
574  return 72 + 40;
575 }
576 
577 /**
578  * Parse MPEG1WAVEFORMATEX extradata structure
579  */
581 {
582  /* fwHeadLayer */
583  switch (AV_RL16(st->codec->extradata)) {
584  case 0x0001 : st->codec->codec_id = AV_CODEC_ID_MP1; break;
585  case 0x0002 : st->codec->codec_id = AV_CODEC_ID_MP2; break;
586  case 0x0004 : st->codec->codec_id = AV_CODEC_ID_MP3; break;
587  }
588 
589  st->codec->bit_rate = AV_RL32(st->codec->extradata + 2); /* dwHeadBitrate */
590 
591  /* dwHeadMode */
592  switch (AV_RL16(st->codec->extradata + 6)) {
593  case 1 :
594  case 2 :
595  case 4 : st->codec->channels = 2;
597  break;
598  case 8 : st->codec->channels = 1;
600  break;
601  }
602 }
603 
604 /**
605  * Initialise stream
606  * @param st Stream to initialise, or NULL to create and initialise new stream
607  * @return NULL on error
608  */
610 {
611  if (st) {
612  if (st->codec->extradata) {
613  av_freep(&st->codec->extradata);
614  st->codec->extradata_size = 0;
615  }
616  } else {
617  WtvStream *wst = av_mallocz(sizeof(WtvStream));
618  if (!wst)
619  return NULL;
620  st = avformat_new_stream(s, NULL);
621  if (!st) {
622  av_free(wst);
623  return NULL;
624  }
625  st->id = sid;
626  st->priv_data = wst;
627  }
628  st->codec->codec_type = codec_type;
630  avpriv_set_pts_info(st, 64, 1, 10000000);
631  return st;
632 }
633 
634 /**
635  * parse Media Type structure and populate stream
636  * @param st Stream, or NULL to create new stream
637  * @param mediatype Mediatype GUID
638  * @param subtype Subtype GUID
639  * @param formattype Format GUID
640  * @param size Size of format buffer
641  * @return NULL on error
642  */
644  ff_asf_guid mediatype, ff_asf_guid subtype,
645  ff_asf_guid formattype, uint64_t size)
646 {
647  WtvContext *wtv = s->priv_data;
648  AVIOContext *pb = wtv->pb;
651  ff_asf_guid actual_subtype;
652  ff_asf_guid actual_formattype;
653 
654  if (size < 32) {
655  av_log(s, AV_LOG_WARNING, "format buffer size underflow\n");
656  avio_skip(pb, size);
657  return NULL;
658  }
659 
660  avio_skip(pb, size - 32);
661  ff_get_guid(pb, &actual_subtype);
662  ff_get_guid(pb, &actual_formattype);
663  avio_seek(pb, -size, SEEK_CUR);
664 
665  st = parse_media_type(s, st, sid, mediatype, actual_subtype, actual_formattype, size - 32);
666  avio_skip(pb, 32);
667  return st;
668  } else if (!ff_guidcmp(mediatype, ff_mediatype_audio)) {
669  st = new_stream(s, st, sid, AVMEDIA_TYPE_AUDIO);
670  if (!st)
671  return NULL;
672  if (!ff_guidcmp(formattype, ff_format_waveformatex)) {
673  int ret = ff_get_wav_header(pb, st->codec, size, 0);
674  if (ret < 0)
675  return NULL;
676  } else {
677  if (ff_guidcmp(formattype, ff_format_none))
678  av_log(s, AV_LOG_WARNING, "unknown formattype:"FF_PRI_GUID"\n", FF_ARG_GUID(formattype));
679  avio_skip(pb, size);
680  }
681 
682  if (!memcmp(subtype + 4, (const uint8_t[]){FF_MEDIASUBTYPE_BASE_GUID}, 12)) {
684  } else if (!ff_guidcmp(subtype, mediasubtype_mpeg1payload)) {
685  if (st->codec->extradata && st->codec->extradata_size >= 22)
687  else
688  av_log(s, AV_LOG_WARNING, "MPEG1WAVEFORMATEX underflow\n");
689  } else {
691  if (st->codec->codec_id == AV_CODEC_ID_NONE)
692  av_log(s, AV_LOG_WARNING, "unknown subtype:"FF_PRI_GUID"\n", FF_ARG_GUID(subtype));
693  }
694  return st;
695  } else if (!ff_guidcmp(mediatype, ff_mediatype_video)) {
696  st = new_stream(s, st, sid, AVMEDIA_TYPE_VIDEO);
697  if (!st)
698  return NULL;
699  if (!ff_guidcmp(formattype, ff_format_videoinfo2)) {
700  int consumed = parse_videoinfoheader2(s, st);
701  avio_skip(pb, FFMAX(size - consumed, 0));
702  } else if (!ff_guidcmp(formattype, ff_format_mpeg2_video)) {
703  uint64_t consumed = parse_videoinfoheader2(s, st);
704  /* ignore extradata; files produced by windows media center contain meaningless mpeg1 sequence header */
705  avio_skip(pb, FFMAX(size - consumed, 0));
706  } else {
707  if (ff_guidcmp(formattype, ff_format_none))
708  av_log(s, AV_LOG_WARNING, "unknown formattype:"FF_PRI_GUID"\n", FF_ARG_GUID(formattype));
709  avio_skip(pb, size);
710  }
711 
712  if (!memcmp(subtype + 4, (const uint8_t[]){FF_MEDIASUBTYPE_BASE_GUID}, 12)) {
714  } else {
716  }
717  if (st->codec->codec_id == AV_CODEC_ID_NONE)
718  av_log(s, AV_LOG_WARNING, "unknown subtype:"FF_PRI_GUID"\n", FF_ARG_GUID(subtype));
719  return st;
720  } else if (!ff_guidcmp(mediatype, mediatype_mpeg2_pes) &&
722  st = new_stream(s, st, sid, AVMEDIA_TYPE_SUBTITLE);
723  if (!st)
724  return NULL;
725  if (ff_guidcmp(formattype, ff_format_none))
726  av_log(s, AV_LOG_WARNING, "unknown formattype:"FF_PRI_GUID"\n", FF_ARG_GUID(formattype));
727  avio_skip(pb, size);
729  return st;
730  } else if (!ff_guidcmp(mediatype, mediatype_mstvcaption) &&
732  st = new_stream(s, st, sid, AVMEDIA_TYPE_SUBTITLE);
733  if (!st)
734  return NULL;
735  if (ff_guidcmp(formattype, ff_format_none))
736  av_log(s, AV_LOG_WARNING, "unknown formattype:"FF_PRI_GUID"\n", FF_ARG_GUID(formattype));
737  avio_skip(pb, size);
739  return st;
740  } else if (!ff_guidcmp(mediatype, mediatype_mpeg2_sections) &&
742  if (ff_guidcmp(formattype, ff_format_none))
743  av_log(s, AV_LOG_WARNING, "unknown formattype:"FF_PRI_GUID"\n", FF_ARG_GUID(formattype));
744  avio_skip(pb, size);
745  return NULL;
746  }
747 
748  av_log(s, AV_LOG_WARNING, "unknown media type, mediatype:"FF_PRI_GUID
749  ", subtype:"FF_PRI_GUID", formattype:"FF_PRI_GUID"\n",
750  FF_ARG_GUID(mediatype), FF_ARG_GUID(subtype), FF_ARG_GUID(formattype));
751  avio_skip(pb, size);
752  return NULL;
753 }
754 
755 enum {
758 };
759 
760 /**
761  * Try to seek over a broken chunk
762  * @return <0 on error
763  */
764 static int recover(WtvContext *wtv, uint64_t broken_pos)
765 {
766  AVIOContext *pb = wtv->pb;
767  int i;
768  for (i = 0; i < wtv->nb_index_entries; i++) {
769  if (wtv->index_entries[i].pos > broken_pos) {
770  int64_t ret = avio_seek(pb, wtv->index_entries[i].pos, SEEK_SET);
771  if (ret < 0)
772  return ret;
773  wtv->pts = wtv->index_entries[i].timestamp;
774  return 0;
775  }
776  }
777  return AVERROR(EIO);
778 }
779 
780 /**
781  * Parse WTV chunks
782  * @param mode SEEK_TO_DATA or SEEK_TO_PTS
783  * @param seekts timestamp
784  * @param[out] len_ptr Length of data chunk
785  * @return stream index of data chunk, or <0 on error
786  */
787 static int parse_chunks(AVFormatContext *s, int mode, int64_t seekts, int *len_ptr)
788 {
789  WtvContext *wtv = s->priv_data;
790  AVIOContext *pb = wtv->pb;
791  while (!avio_feof(pb)) {
792  ff_asf_guid g;
793  int len, sid, consumed;
794 
795  ff_get_guid(pb, &g);
796  len = avio_rl32(pb);
797  if (len < 32) {
798  int ret;
799  if (avio_feof(pb))
800  return AVERROR_EOF;
801  av_log(s, AV_LOG_WARNING, "encountered broken chunk\n");
802  if ((ret = recover(wtv, avio_tell(pb) - 20)) < 0)
803  return ret;
804  continue;
805  }
806  sid = avio_rl32(pb) & 0x7FFF;
807  avio_skip(pb, 8);
808  consumed = 32;
809 
811  if (ff_find_stream_index(s, sid) < 0) {
812  ff_asf_guid mediatype, subtype, formattype;
813  int size;
814  avio_skip(pb, 28);
815  ff_get_guid(pb, &mediatype);
816  ff_get_guid(pb, &subtype);
817  avio_skip(pb, 12);
818  ff_get_guid(pb, &formattype);
819  size = avio_rl32(pb);
820  parse_media_type(s, 0, sid, mediatype, subtype, formattype, size);
821  consumed += 92 + size;
822  }
823  } else if (!ff_guidcmp(g, ff_stream2_guid)) {
824  int stream_index = ff_find_stream_index(s, sid);
825  if (stream_index >= 0 && s->streams[stream_index]->priv_data && !((WtvStream*)s->streams[stream_index]->priv_data)->seen_data) {
826  ff_asf_guid mediatype, subtype, formattype;
827  int size;
828  avio_skip(pb, 12);
829  ff_get_guid(pb, &mediatype);
830  ff_get_guid(pb, &subtype);
831  avio_skip(pb, 12);
832  ff_get_guid(pb, &formattype);
833  size = avio_rl32(pb);
834  parse_media_type(s, s->streams[stream_index], sid, mediatype, subtype, formattype, size);
835  consumed += 76 + size;
836  }
843  int stream_index = ff_find_stream_index(s, sid);
844  if (stream_index >= 0) {
845  AVStream *st = s->streams[stream_index];
846  uint8_t buf[258];
847  const uint8_t *pbuf = buf;
848  int buf_size;
849 
850  avio_skip(pb, 8);
851  consumed += 8;
854  avio_skip(pb, 6);
855  consumed += 6;
856  }
857 
858  buf_size = FFMIN(len - consumed, sizeof(buf));
859  avio_read(pb, buf, buf_size);
860  consumed += buf_size;
861  ff_parse_mpeg2_descriptor(s, st, 0, &pbuf, buf + buf_size, NULL, 0, 0, NULL);
862  }
863  } else if (!ff_guidcmp(g, EVENTID_AudioTypeSpanningEvent)) {
864  int stream_index = ff_find_stream_index(s, sid);
865  if (stream_index >= 0) {
866  AVStream *st = s->streams[stream_index];
867  int audio_type;
868  avio_skip(pb, 8);
869  audio_type = avio_r8(pb);
870  if (audio_type == 2)
872  else if (audio_type == 3)
874  consumed += 9;
875  }
877  int stream_index = ff_find_stream_index(s, sid);
878  if (stream_index >= 0) {
879  avio_skip(pb, 12);
880  if (avio_rl32(pb))
881  av_log(s, AV_LOG_WARNING, "DVB scrambled stream detected (st:%d), decoding will likely fail\n", stream_index);
882  consumed += 16;
883  }
884  } else if (!ff_guidcmp(g, EVENTID_LanguageSpanningEvent)) {
885  int stream_index = ff_find_stream_index(s, sid);
886  if (stream_index >= 0) {
887  AVStream *st = s->streams[stream_index];
888  uint8_t language[4];
889  avio_skip(pb, 12);
890  avio_read(pb, language, 3);
891  if (language[0]) {
892  language[3] = 0;
893  av_dict_set(&st->metadata, "language", language, 0);
894  if (!strcmp(language, "nar") || !strcmp(language, "NAR"))
896  }
897  consumed += 15;
898  }
899  } else if (!ff_guidcmp(g, ff_timestamp_guid)) {
900  int stream_index = ff_find_stream_index(s, sid);
901  if (stream_index >= 0) {
902  avio_skip(pb, 8);
903  wtv->pts = avio_rl64(pb);
904  consumed += 16;
905  if (wtv->pts == -1)
906  wtv->pts = AV_NOPTS_VALUE;
907  else {
908  wtv->last_valid_pts = wtv->pts;
909  if (wtv->epoch == AV_NOPTS_VALUE || wtv->pts < wtv->epoch)
910  wtv->epoch = wtv->pts;
911  if (mode == SEEK_TO_PTS && wtv->pts >= seekts) {
912  avio_skip(pb, WTV_PAD8(len) - consumed);
913  return 0;
914  }
915  }
916  }
917  } else if (!ff_guidcmp(g, ff_data_guid)) {
918  int stream_index = ff_find_stream_index(s, sid);
919  if (mode == SEEK_TO_DATA && stream_index >= 0 && len > 32 && s->streams[stream_index]->priv_data) {
920  WtvStream *wst = s->streams[stream_index]->priv_data;
921  wst->seen_data = 1;
922  if (len_ptr) {
923  *len_ptr = len;
924  }
925  return stream_index;
926  }
927  } else if (!ff_guidcmp(g, /* DSATTRIB_WMDRMProtectionInfo */ (const ff_asf_guid){0x83,0x95,0x74,0x40,0x9D,0x6B,0xEC,0x4E,0xB4,0x3C,0x67,0xA1,0x80,0x1E,0x1A,0x9B})) {
928  int stream_index = ff_find_stream_index(s, sid);
929  if (stream_index >= 0)
930  av_log(s, AV_LOG_WARNING, "encrypted stream detected (st:%d), decoding will likely fail\n", stream_index);
931  } else if (
932  !ff_guidcmp(g, /* DSATTRIB_CAPTURE_STREAMTIME */ (const ff_asf_guid){0x14,0x56,0x1A,0x0C,0xCD,0x30,0x40,0x4F,0xBC,0xBF,0xD0,0x3E,0x52,0x30,0x62,0x07}) ||
933  !ff_guidcmp(g, /* DSATTRIB_PBDATAG_ATTRIBUTE */ (const ff_asf_guid){0x79,0x66,0xB5,0xE0,0xB9,0x12,0xCC,0x43,0xB7,0xDF,0x57,0x8C,0xAA,0x5A,0x7B,0x63}) ||
934  !ff_guidcmp(g, /* DSATTRIB_PicSampleSeq */ (const ff_asf_guid){0x02,0xAE,0x5B,0x2F,0x8F,0x7B,0x60,0x4F,0x82,0xD6,0xE4,0xEA,0x2F,0x1F,0x4C,0x99}) ||
935  !ff_guidcmp(g, /* DSATTRIB_TRANSPORT_PROPERTIES */ ff_DSATTRIB_TRANSPORT_PROPERTIES) ||
936  !ff_guidcmp(g, /* dvr_ms_vid_frame_rep_data */ (const ff_asf_guid){0xCC,0x32,0x64,0xDD,0x29,0xE2,0xDB,0x40,0x80,0xF6,0xD2,0x63,0x28,0xD2,0x76,0x1F}) ||
937  !ff_guidcmp(g, /* EVENTID_ChannelChangeSpanningEvent */ (const ff_asf_guid){0xE5,0xC5,0x67,0x90,0x5C,0x4C,0x05,0x42,0x86,0xC8,0x7A,0xFE,0x20,0xFE,0x1E,0xFA}) ||
938  !ff_guidcmp(g, /* EVENTID_ChannelInfoSpanningEvent */ (const ff_asf_guid){0x80,0x6D,0xF3,0x41,0x32,0x41,0xC2,0x4C,0xB1,0x21,0x01,0xA4,0x32,0x19,0xD8,0x1B}) ||
939  !ff_guidcmp(g, /* EVENTID_ChannelTypeSpanningEvent */ (const ff_asf_guid){0x51,0x1D,0xAB,0x72,0xD2,0x87,0x9B,0x48,0xBA,0x11,0x0E,0x08,0xDC,0x21,0x02,0x43}) ||
940  !ff_guidcmp(g, /* EVENTID_PIDListSpanningEvent */ (const ff_asf_guid){0x65,0x8F,0xFC,0x47,0xBB,0xE2,0x34,0x46,0x9C,0xEF,0xFD,0xBF,0xE6,0x26,0x1D,0x5C}) ||
941  !ff_guidcmp(g, /* EVENTID_SignalAndServiceStatusSpanningEvent */ (const ff_asf_guid){0xCB,0xC5,0x68,0x80,0x04,0x3C,0x2B,0x49,0xB4,0x7D,0x03,0x08,0x82,0x0D,0xCE,0x51}) ||
942  !ff_guidcmp(g, /* EVENTID_StreamTypeSpanningEvent */ (const ff_asf_guid){0xBC,0x2E,0xAF,0x82,0xA6,0x30,0x64,0x42,0xA8,0x0B,0xAD,0x2E,0x13,0x72,0xAC,0x60}) ||
943  !ff_guidcmp(g, (const ff_asf_guid){0x1E,0xBE,0xC3,0xC5,0x43,0x92,0xDC,0x11,0x85,0xE5,0x00,0x12,0x3F,0x6F,0x73,0xB9}) ||
944  !ff_guidcmp(g, (const ff_asf_guid){0x3B,0x86,0xA2,0xB1,0xEB,0x1E,0xC3,0x44,0x8C,0x88,0x1C,0xA3,0xFF,0xE3,0xE7,0x6A}) ||
945  !ff_guidcmp(g, (const ff_asf_guid){0x4E,0x7F,0x4C,0x5B,0xC4,0xD0,0x38,0x4B,0xA8,0x3E,0x21,0x7F,0x7B,0xBF,0x52,0xE7}) ||
946  !ff_guidcmp(g, (const ff_asf_guid){0x63,0x36,0xEB,0xFE,0xA1,0x7E,0xD9,0x11,0x83,0x08,0x00,0x07,0xE9,0x5E,0xAD,0x8D}) ||
947  !ff_guidcmp(g, (const ff_asf_guid){0x70,0xE9,0xF1,0xF8,0x89,0xA4,0x4C,0x4D,0x83,0x73,0xB8,0x12,0xE0,0xD5,0xF8,0x1E}) ||
948  !ff_guidcmp(g, ff_index_guid) ||
949  !ff_guidcmp(g, ff_sync_guid) ||
951  !ff_guidcmp(g, (const ff_asf_guid){0xF7,0x10,0x02,0xB9,0xEE,0x7C,0xED,0x4E,0xBD,0x7F,0x05,0x40,0x35,0x86,0x18,0xA1})) {
952  //ignore known guids
953  } else
954  av_log(s, AV_LOG_WARNING, "unsupported chunk:"FF_PRI_GUID"\n", FF_ARG_GUID(g));
955 
956  avio_skip(pb, WTV_PAD8(len) - consumed);
957  }
958  return AVERROR_EOF;
959 }
960 
962 {
963  WtvContext *wtv = s->priv_data;
964  int root_sector, root_size;
965  uint8_t root[WTV_SECTOR_SIZE];
966  AVIOContext *pb;
967  int64_t timeline_pos;
968  int64_t ret;
969 
970  wtv->epoch =
971  wtv->pts =
973 
974  /* read root directory sector */
975  avio_skip(s->pb, 0x30);
976  root_size = avio_rl32(s->pb);
977  if (root_size > sizeof(root)) {
978  av_log(s, AV_LOG_ERROR, "root directory size exceeds sector size\n");
979  return AVERROR_INVALIDDATA;
980  }
981  avio_skip(s->pb, 4);
982  root_sector = avio_rl32(s->pb);
983 
984  ret = seek_by_sector(s->pb, root_sector, 0);
985  if (ret < 0)
986  return ret;
987  root_size = avio_read(s->pb, root, root_size);
988  if (root_size < 0)
989  return AVERROR_INVALIDDATA;
990 
991  /* parse chunks up until first data chunk */
992  wtv->pb = wtvfile_open(s, root, root_size, ff_timeline_le16);
993  if (!wtv->pb) {
994  av_log(s, AV_LOG_ERROR, "timeline data missing\n");
995  return AVERROR_INVALIDDATA;
996  }
997 
998  ret = parse_chunks(s, SEEK_TO_DATA, 0, 0);
999  if (ret < 0)
1000  return ret;
1001  avio_seek(wtv->pb, -32, SEEK_CUR);
1002 
1003  timeline_pos = avio_tell(s->pb); // save before opening another file
1004 
1005  /* read metadata */
1006  pb = wtvfile_open(s, root, root_size, ff_table_0_entries_legacy_attrib_le16);
1007  if (pb) {
1008  parse_legacy_attrib(s, pb);
1009  wtvfile_close(pb);
1010  }
1011 
1012  /* read seek index */
1013  if (s->nb_streams) {
1014  AVStream *st = s->streams[0];
1015  pb = wtvfile_open(s, root, root_size, ff_table_0_entries_time_le16);
1016  if (pb) {
1017  while(1) {
1018  uint64_t timestamp = avio_rl64(pb);
1019  uint64_t frame_nb = avio_rl64(pb);
1020  if (avio_feof(pb))
1021  break;
1023  0, timestamp, frame_nb, 0, AVINDEX_KEYFRAME);
1024  }
1025  wtvfile_close(pb);
1026 
1027  if (wtv->nb_index_entries) {
1028  pb = wtvfile_open(s, root, root_size, ff_timeline_table_0_entries_Events_le16);
1029  if (pb) {
1030  int i;
1031  while (1) {
1032  uint64_t frame_nb = avio_rl64(pb);
1033  uint64_t position = avio_rl64(pb);
1034  if (avio_feof(pb))
1035  break;
1036  for (i = wtv->nb_index_entries - 1; i >= 0; i--) {
1037  AVIndexEntry *e = wtv->index_entries + i;
1038  if (frame_nb > e->size)
1039  break;
1040  if (position > e->pos)
1041  e->pos = position;
1042  }
1043  }
1044  wtvfile_close(pb);
1045  st->duration = wtv->index_entries[wtv->nb_index_entries - 1].timestamp;
1046  }
1047  }
1048  }
1049  }
1050 
1051  avio_seek(s->pb, timeline_pos, SEEK_SET);
1052  return 0;
1053 }
1054 
1056 {
1057  WtvContext *wtv = s->priv_data;
1058  AVIOContext *pb = wtv->pb;
1059  int stream_index, len, ret;
1060 
1061  stream_index = parse_chunks(s, SEEK_TO_DATA, 0, &len);
1062  if (stream_index < 0)
1063  return stream_index;
1064 
1065  ret = av_get_packet(pb, pkt, len - 32);
1066  if (ret < 0)
1067  return ret;
1068  pkt->stream_index = stream_index;
1069  pkt->pts = wtv->pts;
1070  avio_skip(pb, WTV_PAD8(len) - len);
1071  return 0;
1072 }
1073 
1074 static int read_seek(AVFormatContext *s, int stream_index,
1075  int64_t ts, int flags)
1076 {
1077  WtvContext *wtv = s->priv_data;
1078  AVIOContext *pb = wtv->pb;
1079  AVStream *st = s->streams[0];
1080  int64_t ts_relative;
1081  int i;
1082 
1083  if ((flags & AVSEEK_FLAG_FRAME) || (flags & AVSEEK_FLAG_BYTE))
1084  return AVERROR(ENOSYS);
1085 
1086  /* timestamp adjustment is required because wtv->pts values are absolute,
1087  * whereas AVIndexEntry->timestamp values are relative to epoch. */
1088  ts_relative = ts;
1089  if (wtv->epoch != AV_NOPTS_VALUE)
1090  ts_relative -= wtv->epoch;
1091 
1092  i = ff_index_search_timestamp(wtv->index_entries, wtv->nb_index_entries, ts_relative, flags);
1093  if (i < 0) {
1094  if (wtv->last_valid_pts == AV_NOPTS_VALUE || ts < wtv->last_valid_pts) {
1095  if (avio_seek(pb, 0, SEEK_SET) < 0)
1096  return -1;
1097  } else if (st->duration != AV_NOPTS_VALUE && ts_relative > st->duration && wtv->nb_index_entries) {
1098  if (avio_seek(pb, wtv->index_entries[wtv->nb_index_entries - 1].pos, SEEK_SET) < 0)
1099  return -1;
1100  }
1101  if (parse_chunks(s, SEEK_TO_PTS, ts, 0) < 0)
1102  return AVERROR(ERANGE);
1103  return 0;
1104  }
1105  if (avio_seek(pb, wtv->index_entries[i].pos, SEEK_SET) < 0)
1106  return -1;
1107  wtv->pts = wtv->index_entries[i].timestamp;
1108  if (wtv->epoch != AV_NOPTS_VALUE)
1109  wtv->pts += wtv->epoch;
1110  wtv->last_valid_pts = wtv->pts;
1111  return 0;
1112 }
1113 
1115 {
1116  WtvContext *wtv = s->priv_data;
1117  av_freep(&wtv->index_entries);
1118  wtvfile_close(wtv->pb);
1119  return 0;
1120 }
1121 
1123  .name = "wtv",
1124  .long_name = NULL_IF_CONFIG_SMALL("Windows Television (WTV)"),
1125  .priv_data_size = sizeof(WtvContext),
1129  .read_seek = read_seek,
1131  .flags = AVFMT_SHOW_IDS,
1132 };