FFmpeg
 All Data Structures Files Functions Variables Typedefs Enumerations Enumerator Macros Groups Pages
wtvenc.c
Go to the documentation of this file.
1 /*
2  * Windows Television (WTV) muxer
3  * Copyright (c) 2011 Zhentan Feng <spyfeng at gmail dot com>
4  * Copyright (c) 2011 Peter Ross <pross@xvid.org>
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 Zhentan Feng <spyfeng at gmail dot com>
26  */
27 
28 #include "libavutil/intreadwrite.h"
29 #include "libavutil/avassert.h"
30 #include "avformat.h"
31 #include "internal.h"
32 #include "wtv.h"
33 #include "asf.h"
34 
35 #define WTV_BIGSECTOR_SIZE (1 << WTV_BIGSECTOR_BITS)
36 #define INDEX_BASE 0x2
37 #define MAX_NB_INDEX 10
38 
39 /* declare utf16le strings */
40 #define _ , 0,
42  {'t'_'i'_'m'_'e'_'l'_'i'_'n'_'e'_'.'_'t'_'a'_'b'_'l'_'e'_'.'_'0'_'.'_'h'_'e'_'a'_'d'_'e'_'r'_'.'_'E'_'v'_'e'_'n'_'t'_'s', 0};
44  {'t'_'a'_'b'_'l'_'e'_'.'_'0'_'.'_'h'_'e'_'a'_'d'_'e'_'r'_'.'_'l'_'e'_'g'_'a'_'c'_'y'_'_'_'a'_'t'_'t'_'r'_'i'_'b', 0};
46  {'t'_'a'_'b'_'l'_'e'_'.'_'0'_'.'_'r'_'e'_'d'_'i'_'r'_'e'_'c'_'t'_'o'_'r'_'.'_'l'_'e'_'g'_'a'_'c'_'y'_'_'_'a'_'t'_'t'_'r'_'i'_'b', 0};
47 static const uint8_t table_0_header_time[] =
48  {'t'_'a'_'b'_'l'_'e'_'.'_'0'_'.'_'h'_'e'_'a'_'d'_'e'_'r'_'.'_'t'_'i'_'m'_'e', 0};
49 static const uint8_t legacy_attrib[] =
50  {'l'_'e'_'g'_'a'_'c'_'y'_'_'_'a'_'t'_'t'_'r'_'i'_'b', 0};
51 #undef _
52 
53 static const ff_asf_guid sub_wtv_guid =
54  {0x8C,0xC3,0xD2,0xC2,0x7E,0x9A,0xDA,0x11,0x8B,0xF7,0x00,0x07,0xE9,0x5E,0xAD,0x8D};
55 
66 };
67 
68 typedef struct {
69  int64_t length;
70  const void *header;
71  int depth;
73 } WtvFile;
74 
75 typedef struct {
76  int64_t pos;
77  int64_t serial;
78  const ff_asf_guid * guid;
79  int stream_id;
81 
82 typedef struct {
83  int64_t serial;
84  int64_t value;
85 } WtvSyncEntry;
86 
87 typedef struct {
90  int64_t serial; /** chunk serial number */
91  int64_t last_chunk_pos; /** last chunk position */
92  int64_t last_timestamp_pos; /** last timestamp chunk position */
93  int64_t first_index_pos; /** first index_chunk position */
94 
96  int nb_index;
98 
99  WtvSyncEntry *st_pairs; /* (serial, timestamp) pairs */
101  WtvSyncEntry *sp_pairs; /* (serial, position) pairs */
103 
104  int64_t last_pts;
105  int64_t last_serial;
106 
108 } WtvContext;
109 
110 
111 static void add_serial_pair(WtvSyncEntry ** list, int * count, int64_t serial, int64_t value)
112 {
113  int new_count = *count + 1;
114  WtvSyncEntry *new_list = av_realloc(*list, new_count * sizeof(WtvSyncEntry));
115  if (!new_list)
116  return;
117  new_list[*count] = (WtvSyncEntry){serial, value};
118  *list = new_list;
119  *count = new_count;
120 }
121 
123 
124 typedef struct {
125  const uint8_t *header;
129 
130 static int write_pad(AVIOContext *pb, int size)
131 {
132  for (; size > 0; size--)
133  avio_w8(pb, 0);
134  return 0;
135 }
136 
137 static const ff_asf_guid *get_codec_guid(enum AVCodecID id, const AVCodecGuid *av_guid)
138 {
139  int i;
140  for (i = 0; av_guid[i].id != AV_CODEC_ID_NONE; i++) {
141  if (id == av_guid[i].id)
142  return &(av_guid[i].guid);
143  }
144  return NULL;
145 }
146 
147 /**
148  * Write chunk header. If header chunk (0x80000000 set) then add to list of header chunks
149  */
150 static void write_chunk_header(AVFormatContext *s, const ff_asf_guid *guid, int length, int stream_id)
151 {
152  WtvContext *wctx = s->priv_data;
153  AVIOContext *pb = s->pb;
154 
155  wctx->last_chunk_pos = avio_tell(pb) - wctx->timeline_start_pos;
156  ff_put_guid(pb, guid);
157  avio_wl32(pb, 32 + length);
158  avio_wl32(pb, stream_id);
159  avio_wl64(pb, wctx->serial);
160 
161  if ((stream_id & 0x80000000) && guid != &ff_index_guid) {
162  WtvChunkEntry *t = wctx->index + wctx->nb_index;
164  t->pos = wctx->last_chunk_pos;
165  t->serial = wctx->serial;
166  t->guid = guid;
167  t->stream_id = stream_id & 0x3FFFFFFF;
168  wctx->nb_index++;
169  }
170 }
171 
172 static void write_chunk_header2(AVFormatContext *s, const ff_asf_guid *guid, int stream_id)
173 {
174  WtvContext *wctx = s->priv_data;
175  AVIOContext *pb = s->pb;
176 
177  int64_t last_chunk_pos = wctx->last_chunk_pos;
178  write_chunk_header(s, guid, 0, stream_id); // length updated later
179  avio_wl64(pb, last_chunk_pos);
180 }
181 
183 {
184  WtvContext *wctx = s->priv_data;
185  AVIOContext *pb = s->pb;
186 
187  // update the chunk_len field and pad.
188  int64_t chunk_len = avio_tell(pb) - (wctx->last_chunk_pos + wctx->timeline_start_pos);
189  avio_seek(pb, -(chunk_len - 16), SEEK_CUR);
190  avio_wl32(pb, chunk_len);
191  avio_seek(pb, chunk_len - (16 + 4), SEEK_CUR);
192 
193  write_pad(pb, WTV_PAD8(chunk_len) - chunk_len);
194  wctx->serial++;
195 }
196 
198 {
199  AVIOContext *pb = s->pb;
200  WtvContext *wctx = s->priv_data;
201  int i;
202 
203  write_chunk_header2(s, &ff_index_guid, 0x80000000);
204  avio_wl32(pb, 0);
205  avio_wl32(pb, 0);
206 
207  for (i = 0; i < wctx->nb_index; i++) {
208  WtvChunkEntry *t = wctx->index + i;
209  ff_put_guid(pb, t->guid);
210  avio_wl64(pb, t->pos);
211  avio_wl32(pb, t->stream_id);
212  avio_wl32(pb, 0); // checksum?
213  avio_wl64(pb, t->serial);
214  }
215  wctx->nb_index = 0; // reset index
217 
218  if (!wctx->first_index_pos)
219  wctx->first_index_pos = wctx->last_chunk_pos;
220 }
221 
223 {
224  WtvContext *wctx = s->priv_data;
226  if (wctx->nb_index == MAX_NB_INDEX)
227  write_index(s);
228 }
229 
231 {
232  WtvContext *wctx = s->priv_data;
233  const ff_asf_guid *g, *media_type, *format_type;
234  AVIOContext *pb = s->pb;
235  int64_t hdr_pos_start;
236  int hdr_size = 0;
237 
238  if (st->codec->codec_type == AVMEDIA_TYPE_VIDEO) {
240  media_type = &ff_mediatype_video;
241  format_type = &ff_format_mpeg2_video;
242  } else if (st->codec->codec_type == AVMEDIA_TYPE_AUDIO) {
244  media_type = &ff_mediatype_audio;
245  format_type = &ff_format_waveformatex;
246  } else {
247  av_log(s, AV_LOG_ERROR, "unknown codec_type (0x%x)\n", st->codec->codec_type);
248  return -1;
249  }
250 
251  if (g == NULL) {
252  av_log(s, AV_LOG_ERROR, "can't get video codec_id (0x%x) guid.\n", st->codec->codec_id);
253  return -1;
254  }
255 
256  ff_put_guid(pb, media_type); // mediatype
258  write_pad(pb, 12);
259  ff_put_guid(pb,&ff_format_cpfilters_processed); // format type
260  avio_wl32(pb, 0); // size
261 
262  hdr_pos_start = avio_tell(pb);
263  if (st->codec->codec_type == AVMEDIA_TYPE_VIDEO) {
264  if (wctx->first_video_flag) {
265  write_pad(pb, 216); //The size is sensitive.
266  wctx->first_video_flag = 0;
267  } else {
268  write_pad(pb, 72); // aspect ratio
270  }
271  } else {
272  ff_put_wav_header(pb, st->codec);
273  }
274  hdr_size = avio_tell(pb) - hdr_pos_start;
275 
276  // seek back write hdr_size
277  avio_seek(pb, -(hdr_size + 4), SEEK_CUR);
278  avio_wl32(pb, hdr_size + 32);
279  avio_seek(pb, hdr_size, SEEK_CUR);
280  ff_put_guid(pb, g); // actual_subtype
281  ff_put_guid(pb, format_type); // actual_formattype
282 
283  return 0;
284 }
285 
287 {
288  AVIOContext *pb = s->pb;
289  int ret;
290  write_chunk_header2(s, &ff_stream1_guid, 0x80000000 | 0x01);
291 
292  avio_wl32(pb, 0x01);
293  write_pad(pb, 4);
294  write_pad(pb, 4);
295 
296  ret = write_stream_codec_info(s, st);
297  if (ret < 0) {
298  av_log(s, AV_LOG_ERROR, "write stream codec info failed codec_type(0x%x)\n", st->codec->codec_type);
299  return -1;
300  }
301 
302  finish_chunk(s);
303  return 0;
304 }
305 
307 {
308  AVIOContext *pb = s->pb;
309  WtvContext *wctx = s->priv_data;
310  int64_t last_chunk_pos = wctx->last_chunk_pos;
311 
312  write_chunk_header(s, &ff_sync_guid, 0x18, 0);
313  avio_wl64(pb, wctx->first_index_pos);
314  avio_wl64(pb, wctx->last_timestamp_pos);
315  avio_wl64(pb, 0);
316 
317  finish_chunk(s);
318  add_serial_pair(&wctx->sp_pairs, &wctx->nb_sp_pairs, wctx->serial, wctx->last_chunk_pos);
319 
320  wctx->last_chunk_pos = last_chunk_pos;
321 }
322 
324 {
325  AVIOContext *pb = s->pb;
326  int ret;
327 
329  avio_wl32(pb, 0x00000001);
330  avio_wl32(pb, st->index + INDEX_BASE); //stream_id
331  avio_wl32(pb, 0x00000001);
332  write_pad(pb, 8);
333 
334  ret = write_stream_codec_info(s, st);
335  if (ret < 0) {
336  av_log(s, AV_LOG_ERROR, "write stream codec info failed codec_type(0x%x)\n", st->codec->codec_type);
337  return -1;
338  }
339  finish_chunk(s);
340 
341  avpriv_set_pts_info(st, 64, 1, 10000000);
342 
343  return 0;
344 }
345 
347 {
348  AVIOContext *pb = s->pb;
349  WtvContext *wctx = s->priv_data;
350  int i, pad, ret;
351  AVStream *st;
352 
353  wctx->last_chunk_pos = -1;
354  wctx->last_timestamp_pos = -1;
355 
356  ff_put_guid(pb, &ff_wtv_guid);
358 
359  avio_wl32(pb, 0x01);
360  avio_wl32(pb, 0x02);
361  avio_wl32(pb, 1 << WTV_SECTOR_BITS);
362  avio_wl32(pb, 1 << WTV_BIGSECTOR_BITS);
363 
364  //write initial root fields
365  avio_wl32(pb, 0); // root_size, update later
366  write_pad(pb, 4);
367  avio_wl32(pb, 0); // root_sector, update it later.
368 
369  write_pad(pb, 32);
370  avio_wl32(pb, 0); // file ends pointer, update it later.
371 
372  pad = (1 << WTV_SECTOR_BITS) - avio_tell(pb);
373  write_pad(pb, pad);
374 
375  wctx->timeline_start_pos = avio_tell(pb);
376 
377  wctx->serial = 1;
378  wctx->last_chunk_pos = -1;
379  wctx->first_video_flag = 1;
380 
381  for (i = 0; i < s->nb_streams; i++) {
382  st = s->streams[i];
383  if (st->codec->codec_id == AV_CODEC_ID_MJPEG)
384  continue;
385  ret = write_stream_codec(s, st);
386  if (ret < 0) {
387  av_log(s, AV_LOG_ERROR, "write stream codec failed codec_type(0x%x)\n", st->codec->codec_type);
388  return -1;
389  }
390  if (!i)
391  write_sync(s);
392  }
393 
394  for (i = 0; i < s->nb_streams; i++) {
395  st = s->streams[i];
396  if (st->codec->codec_id == AV_CODEC_ID_MJPEG)
397  continue;
398  ret = write_stream_data(s, st);
399  if (ret < 0) {
400  av_log(s, AV_LOG_ERROR, "write stream data failed codec_type(0x%x)\n", st->codec->codec_type);
401  return -1;
402  }
403  }
404 
405  if (wctx->nb_index)
406  write_index(s);
407 
408  return 0;
409 }
410 
412 {
413  AVIOContext *pb = s->pb;
414  WtvContext *wctx = s->priv_data;
415  AVCodecContext *enc = s->streams[pkt->stream_index]->codec;
416 
417  write_chunk_header(s, &ff_timestamp_guid, 56, 0x40000000 | (INDEX_BASE + pkt->stream_index));
418  write_pad(pb, 8);
419  avio_wl64(pb, pkt->pts == AV_NOPTS_VALUE ? -1 : pkt->pts);
420  avio_wl64(pb, pkt->pts == AV_NOPTS_VALUE ? -1 : pkt->pts);
421  avio_wl64(pb, pkt->pts == AV_NOPTS_VALUE ? -1 : pkt->pts);
422  avio_wl64(pb, 0);
423  avio_wl64(pb, enc->codec_type == AVMEDIA_TYPE_VIDEO && (pkt->flags & AV_PKT_FLAG_KEY) ? 1 : 0);
424  avio_wl64(pb, 0);
425 
426  wctx->last_timestamp_pos = wctx->last_chunk_pos;
427 }
428 
430 {
431  AVIOContext *pb = s->pb;
432  WtvContext *wctx = s->priv_data;
433 
434  if (s->streams[pkt->stream_index]->codec->codec_id == AV_CODEC_ID_MJPEG && !wctx->thumbnail.size) {
435  av_copy_packet(&wctx->thumbnail, pkt);
436  return 0;
437  }
438 
439  /* emit sync chunk and 'timeline.table.0.entries.Event' record every 50 frames */
440  if (wctx->serial - (wctx->nb_sp_pairs ? wctx->sp_pairs[wctx->nb_sp_pairs - 1].serial : 0) >= 50)
441  write_sync(s);
442 
443  /* emit 'table.0.entries.time' record every 500ms */
444  if (pkt->pts != AV_NOPTS_VALUE && pkt->pts - (wctx->nb_st_pairs ? wctx->st_pairs[wctx->nb_st_pairs - 1].value : 0) >= 5000000)
445  add_serial_pair(&wctx->st_pairs, &wctx->nb_st_pairs, wctx->serial, pkt->pts);
446 
447  if (pkt->pts != AV_NOPTS_VALUE && pkt->pts > wctx->last_pts) {
448  wctx->last_pts = pkt->pts;
449  wctx->last_serial = wctx->serial;
450  }
451 
452  // write timestamp chunk
453  write_timestamp(s, pkt);
454 
456  avio_write(pb, pkt->data, pkt->size);
457  write_pad(pb, WTV_PAD8(pkt->size) - pkt->size);
458 
459  wctx->serial++;
460  avio_flush(pb);
461  return 0;
462 }
463 
465 {
466  avio_wl32(pb, 0x10);
467  write_pad(pb, 84);
468  avio_wl64(pb, 0x32);
469  return 96;
470 }
471 
473 {
474  int pad = 0;
475  avio_wl32(pb, 0xFFFFFFFF);
476  write_pad(pb, 12);
477  avio_write(pb, legacy_attrib, sizeof(legacy_attrib));
478  pad = WTV_PAD8(sizeof(legacy_attrib)) - sizeof(legacy_attrib);
479  write_pad(pb, pad);
480  write_pad(pb, 32);
481  return 48 + WTV_PAD8(sizeof(legacy_attrib));
482 }
483 
485 {
486  avio_wl32(pb, 0x10);
487  write_pad(pb, 76);
488  avio_wl64(pb, 0x40);
489  return 88;
490 }
491 
501 };
502 
503 static int write_root_table(AVFormatContext *s, int64_t sector_pos)
504 {
505  AVIOContext *pb = s->pb;
506  WtvContext *wctx = s->priv_data;
507  int size, pad;
508  int i;
509 
511  for (i = 0; i < sizeof(wtv_root_entry_table)/sizeof(WTVRootEntryTable); i++, h++) {
512  WtvFile *w = &wctx->file[i];
513  int filename_padding = WTV_PAD8(h->header_size) - h->header_size;
514  WTVHeaderWriteFunc *write = h->write_header;
515  int len = 0;
516  int64_t len_pos;
517 
519  len_pos = avio_tell(pb);
520  avio_wl16(pb, 40 + h->header_size + filename_padding + 8); // maybe updated later
521  write_pad(pb, 6);
522  avio_wl64(pb, write ? 0 : w->length);// maybe update later
523  avio_wl32(pb, (h->header_size + filename_padding) >> 1);
524  write_pad(pb, 4);
525 
526  avio_write(pb, h->header, h->header_size);
527  write_pad(pb, filename_padding);
528 
529  if (write) {
530  len = write(pb);
531  // update length field
532  avio_seek(pb, len_pos, SEEK_SET);
533  avio_wl64(pb, 40 + h->header_size + filename_padding + len);
534  avio_wl64(pb, len |(1ULL<<62) | (1ULL<<60));
535  avio_seek(pb, 8 + h->header_size + filename_padding + len, SEEK_CUR);
536  } else {
537  avio_wl32(pb, w->first_sector);
538  avio_wl32(pb, w->depth);
539  }
540  }
541 
542  // caculate root table size
543  size = avio_tell(pb) - sector_pos;
544  pad = WTV_SECTOR_SIZE- size;
545  write_pad(pb, pad);
546 
547  return size;
548 }
549 
550 static void write_fat(AVIOContext *pb, int start_sector, int nb_sectors, int shift)
551 {
552  int i;
553  for (i = 0; i < nb_sectors; i++) {
554  avio_wl32(pb, start_sector + (i << shift));
555  }
556  // pad left sector pointer size
557  write_pad(pb, WTV_SECTOR_SIZE - ((nb_sectors << 2) % WTV_SECTOR_SIZE));
558 }
559 
560 static int64_t write_fat_sector(AVFormatContext *s, int64_t start_pos, int nb_sectors, int sector_bits, int depth)
561 {
562  int64_t start_sector = start_pos >> WTV_SECTOR_BITS;
563  int shift = sector_bits - WTV_SECTOR_BITS;
564 
565  int64_t fat = avio_tell(s->pb);
566  write_fat(s->pb, start_sector, nb_sectors, shift);
567 
568  if (depth == 2) {
569  int64_t start_sector1 = fat >> WTV_SECTOR_BITS;
570  int nb_sectors1 = ((nb_sectors << 2) + WTV_SECTOR_SIZE - 1) / WTV_SECTOR_SIZE;
571  int64_t fat1 = avio_tell(s->pb);
572 
573  write_fat(s->pb, start_sector1, nb_sectors1, 0);
574  return fat1;
575  }
576 
577  return fat;
578 }
579 
581 {
582  AVIOContext *pb = s->pb;
583  WtvContext *wctx = s->priv_data;
584  int i;
585  for (i = 0; i < wctx->nb_sp_pairs; i++) {
586  avio_wl64(pb, wctx->sp_pairs[i].serial);
587  avio_wl64(pb, wctx->sp_pairs[i].value);
588  }
589 }
590 
592 {
593  AVIOContext *pb = s->pb;
594  WtvContext *wctx = s->priv_data;
595  int i;
596  for (i = 0; i < wctx->nb_st_pairs; i++) {
597  avio_wl64(pb, wctx->st_pairs[i].value);
598  avio_wl64(pb, wctx->st_pairs[i].serial);
599  }
600  avio_wl64(pb, wctx->last_pts);
601  avio_wl64(pb, wctx->last_serial);
602 }
603 
604 static void write_metadata_header(AVIOContext *pb, int type, const char *key, int value_size)
605 {
607  avio_wl32(pb, type);
608  avio_wl32(pb, value_size);
609  avio_put_str16le(pb, key);
610 }
611 
612 static int metadata_header_size(const char *key)
613 {
614  return 16 + 4 + 4 + strlen(key)*2 + 2;
615 }
616 
617 static void write_tag_int32(AVIOContext *pb, const char *key, int value)
618 {
619  write_metadata_header(pb, 0, key, 4);
620  avio_wl32(pb, value);
621 }
622 
623 static void write_tag(AVIOContext *pb, const char *key, const char *value)
624 {
625  write_metadata_header(pb, 1, key, strlen(value)*2 + 2);
626  avio_put_str16le(pb, value);
627 }
628 
630 {
631  return strlen("image/jpeg")*2 + 2 + 1 + (e ? strlen(e->value)*2 : 0) + 2 + 4 + pkt->size;
632 }
633 
635 {
636  WtvContext *wctx = s->priv_data;
637  AVIOContext *pb = s->pb;
638  AVDictionaryEntry *tag = 0;
639 
640  //FIXME: translate special tags (e.g. WM/Bitrate) to binary representation
642  while ((tag = av_dict_get(s->metadata, "", tag, AV_DICT_IGNORE_SUFFIX)))
643  write_tag(pb, tag->key, tag->value);
644 
645  if (wctx->thumbnail.size) {
646  AVStream *st = s->streams[wctx->thumbnail.stream_index];
647  tag = av_dict_get(st->metadata, "title", NULL, 0);
648  write_metadata_header(pb, 2, "WM/Picture", attachment_value_size(&wctx->thumbnail, tag));
649 
650  avio_put_str16le(pb, "image/jpeg");
651  avio_w8(pb, 0x10);
652  avio_put_str16le(pb, tag ? tag->value : "");
653 
654  avio_wl32(pb, wctx->thumbnail.size);
655  avio_write(pb, wctx->thumbnail.data, wctx->thumbnail.size);
656 
657  write_tag_int32(pb, "WM/MediaThumbType", 2);
658  }
659 }
660 
662 {
663  WtvContext *wctx = s->priv_data;
664  AVIOContext *pb = s->pb;
665  AVDictionaryEntry *tag = 0;
666  int64_t pos = 0;
667 
668  //FIXME: translate special tags to binary representation
669  while ((tag = av_dict_get(s->metadata, "", tag, AV_DICT_IGNORE_SUFFIX))) {
670  avio_wl64(pb, pos);
671  pos += metadata_header_size(tag->key) + strlen(tag->value)*2 + 2;
672  }
673 
674  if (wctx->thumbnail.size) {
675  AVStream *st = s->streams[wctx->thumbnail.stream_index];
676  avio_wl64(pb, pos);
677  pos += metadata_header_size("WM/Picture") + attachment_value_size(&wctx->thumbnail, av_dict_get(st->metadata, "title", NULL, 0));
678 
679  avio_wl64(pb, pos);
680  pos += metadata_header_size("WM/MediaThumbType") + 4;
681  }
682 }
683 
684 /**
685  * Pad the remainder of a file
686  * Write out fat table
687  * @return <0 on error
688  */
689 static int finish_file(AVFormatContext *s, enum WtvFileIndex index, int64_t start_pos)
690 {
691  WtvContext *wctx = s->priv_data;
692  AVIOContext *pb = s->pb;
693  WtvFile *w = &wctx->file[index];
694  int64_t end_pos = avio_tell(pb);
695  int sector_bits, nb_sectors, pad;
696 
697  av_assert0(index < WTV_FILES);
698 
699  w->length = (end_pos - start_pos);
700 
701  // determine optimal fat table depth, sector_bits, nb_sectors
702  if (w->length <= WTV_SECTOR_SIZE) {
703  w->depth = 0;
704  sector_bits = WTV_SECTOR_BITS;
705  } else if (w->length <= (WTV_SECTOR_SIZE / 4) * WTV_SECTOR_SIZE) {
706  w->depth = 1;
707  sector_bits = WTV_SECTOR_BITS;
708  } else if (w->length <= (WTV_SECTOR_SIZE / 4) * WTV_BIGSECTOR_SIZE) {
709  w->depth = 1;
710  sector_bits = WTV_BIGSECTOR_BITS;
711  } else if (w->length <= (int64_t)(WTV_SECTOR_SIZE / 4) * (WTV_SECTOR_SIZE / 4) * WTV_SECTOR_SIZE) {
712  w->depth = 2;
713  sector_bits = WTV_SECTOR_BITS;
714  } else if (w->length <= (int64_t)(WTV_SECTOR_SIZE / 4) * (WTV_SECTOR_SIZE / 4) * WTV_BIGSECTOR_SIZE) {
715  w->depth = 2;
716  sector_bits = WTV_BIGSECTOR_BITS;
717  } else {
718  av_log(s, AV_LOG_ERROR, "unsupported file allocation table depth (%"PRIi64" bytes)\n", w->length);
719  return -1;
720  }
721 
722  // determine the nb_sectors
723  nb_sectors = (int)(w->length >> sector_bits);
724 
725  // pad sector of timeline
726  pad = (1 << sector_bits) - (w->length % (1 << sector_bits));
727  if (pad) {
728  nb_sectors++;
729  write_pad(pb, pad);
730  }
731 
732  //write fat table
733  if (w->depth > 0) {
734  w->first_sector = write_fat_sector(s, start_pos, nb_sectors, sector_bits, w->depth) >> WTV_SECTOR_BITS;
735  } else {
736  w->first_sector = start_pos >> WTV_SECTOR_BITS;
737  }
738 
739  w->length |= 1ULL<<60;
740  if (sector_bits == WTV_SECTOR_BITS)
741  w->length |= 1ULL<<63;
742 
743  return 0;
744 }
745 
747 {
748  WtvContext *wctx = s->priv_data;
749  AVIOContext *pb = s->pb;
750  int root_size;
751  int64_t sector_pos;
752  int64_t start_pos, file_end_pos;
753 
754  if (finish_file(s, WTV_TIMELINE, wctx->timeline_start_pos) < 0)
755  return -1;
756 
757  start_pos = avio_tell(pb);
759  if (finish_file(s, WTV_TIMELINE_TABLE_0_ENTRIES_EVENTS, start_pos) < 0)
760  return -1;
761 
762  start_pos = avio_tell(pb);
764  if (finish_file(s, WTV_TABLE_0_ENTRIES_LEGACY_ATTRIB, start_pos) < 0)
765  return -1;
766 
767  start_pos = avio_tell(pb);
769  if (finish_file(s, WTV_TABLE_0_REDIRECTOR_LEGACY_ATTRIB, start_pos) < 0)
770  return -1;
771 
772  start_pos = avio_tell(pb);
774  if (finish_file(s, WTV_TABLE_0_ENTRIES_TIME, start_pos) < 0)
775  return -1;
776 
777  // write root table
778  sector_pos = avio_tell(pb);
779  root_size = write_root_table(s, sector_pos);
780 
781  file_end_pos = avio_tell(pb);
782  // update root value
783  avio_seek(pb, 0x30, SEEK_SET);
784  avio_wl32(pb, root_size);
785  avio_seek(pb, 4, SEEK_CUR);
786  avio_wl32(pb, sector_pos >> WTV_SECTOR_BITS);
787  avio_seek(pb, 0x5c, SEEK_SET);
788  avio_wl32(pb, file_end_pos >> WTV_SECTOR_BITS);
789 
790  avio_flush(pb);
791 
792  av_free(wctx->sp_pairs);
793  av_free(wctx->st_pairs);
794  av_free_packet(&wctx->thumbnail);
795  return 0;
796 }
797 
799  .name = "wtv",
800  .long_name = NULL_IF_CONFIG_SMALL("Windows Television (WTV)"),
801  .extensions = "wtv",
802  .priv_data_size = sizeof(WtvContext),
803  .audio_codec = AV_CODEC_ID_AC3,
804  .video_codec = AV_CODEC_ID_MPEG2VIDEO,
808  .codec_tag = (const AVCodecTag* const []){ ff_codec_bmp_tags,
809  ff_codec_wav_tags, 0 },
810 };