You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

1125 lines
42KB

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