Subversion Repositories Kolibri OS

Rev

Blame | Last modification | View Log | RSS feed

  1. /*
  2.  * ASF muxer
  3.  * Copyright (c) 2000, 2001 Fabrice Bellard
  4.  *
  5.  * This file is part of FFmpeg.
  6.  *
  7.  * FFmpeg is free software; you can redistribute it and/or
  8.  * modify it under the terms of the GNU Lesser General Public
  9.  * License as published by the Free Software Foundation; either
  10.  * version 2.1 of the License, or (at your option) any later version.
  11.  *
  12.  * FFmpeg is distributed in the hope that it will be useful,
  13.  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14.  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15.  * Lesser General Public License for more details.
  16.  *
  17.  * You should have received a copy of the GNU Lesser General Public
  18.  * License along with FFmpeg; if not, write to the Free Software
  19.  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  20.  */
  21.  
  22. #include "libavutil/avassert.h"
  23. #include "libavutil/dict.h"
  24. #include "libavutil/mathematics.h"
  25. #include "avformat.h"
  26. #include "avio_internal.h"
  27. #include "internal.h"
  28. #include "riff.h"
  29. #include "asf.h"
  30.  
  31. #define ASF_INDEXED_INTERVAL    10000000
  32. #define ASF_INDEX_BLOCK         (1<<9)
  33. #define ASF_PAYLOADS_PER_PACKET 63
  34.  
  35. #define ASF_PACKET_ERROR_CORRECTION_DATA_SIZE 0x2
  36. #define ASF_PACKET_ERROR_CORRECTION_FLAGS          \
  37.     (ASF_PACKET_FLAG_ERROR_CORRECTION_PRESENT |    \
  38.      ASF_PACKET_ERROR_CORRECTION_DATA_SIZE)
  39.  
  40. #if (ASF_PACKET_ERROR_CORRECTION_FLAGS != 0)
  41. #   define ASF_PACKET_ERROR_CORRECTION_FLAGS_FIELD_SIZE 1
  42. #else
  43. #   define ASF_PACKET_ERROR_CORRECTION_FLAGS_FIELD_SIZE 0
  44. #endif
  45.  
  46. #define ASF_PPI_PROPERTY_FLAGS                                       \
  47.     (ASF_PL_FLAG_REPLICATED_DATA_LENGTH_FIELD_IS_BYTE           |    \
  48.      ASF_PL_FLAG_OFFSET_INTO_MEDIA_OBJECT_LENGTH_FIELD_IS_DWORD |    \
  49.      ASF_PL_FLAG_MEDIA_OBJECT_NUMBER_LENGTH_FIELD_IS_BYTE       |    \
  50.      ASF_PL_FLAG_STREAM_NUMBER_LENGTH_FIELD_IS_BYTE)
  51.  
  52. #define ASF_PPI_LENGTH_TYPE_FLAGS 0
  53.  
  54. #define ASF_PAYLOAD_FLAGS ASF_PL_FLAG_PAYLOAD_LENGTH_FIELD_IS_WORD
  55.  
  56. #if (ASF_PPI_FLAG_SEQUENCE_FIELD_IS_BYTE == (ASF_PPI_LENGTH_TYPE_FLAGS & ASF_PPI_MASK_SEQUENCE_FIELD_SIZE))
  57. #   define ASF_PPI_SEQUENCE_FIELD_SIZE 1
  58. #endif
  59. #if (ASF_PPI_FLAG_SEQUENCE_FIELD_IS_WORD == (ASF_PPI_LENGTH_TYPE_FLAGS & ASF_PPI_MASK_SEQUENCE_FIELD_SIZE))
  60. #   define ASF_PPI_SEQUENCE_FIELD_SIZE 2
  61. #endif
  62. #if (ASF_PPI_FLAG_SEQUENCE_FIELD_IS_DWORD == (ASF_PPI_LENGTH_TYPE_FLAGS & ASF_PPI_MASK_SEQUENCE_FIELD_SIZE))
  63. #   define ASF_PPI_SEQUENCE_FIELD_SIZE 4
  64. #endif
  65. #ifndef ASF_PPI_SEQUENCE_FIELD_SIZE
  66. #   define ASF_PPI_SEQUENCE_FIELD_SIZE 0
  67. #endif
  68.  
  69. #if (ASF_PPI_FLAG_PACKET_LENGTH_FIELD_IS_BYTE == (ASF_PPI_LENGTH_TYPE_FLAGS & ASF_PPI_MASK_PACKET_LENGTH_FIELD_SIZE))
  70. #   define ASF_PPI_PACKET_LENGTH_FIELD_SIZE 1
  71. #endif
  72. #if (ASF_PPI_FLAG_PACKET_LENGTH_FIELD_IS_WORD == (ASF_PPI_LENGTH_TYPE_FLAGS & ASF_PPI_MASK_PACKET_LENGTH_FIELD_SIZE))
  73. #   define ASF_PPI_PACKET_LENGTH_FIELD_SIZE 2
  74. #endif
  75. #if (ASF_PPI_FLAG_PACKET_LENGTH_FIELD_IS_DWORD == (ASF_PPI_LENGTH_TYPE_FLAGS & ASF_PPI_MASK_PACKET_LENGTH_FIELD_SIZE))
  76. #   define ASF_PPI_PACKET_LENGTH_FIELD_SIZE 4
  77. #endif
  78. #ifndef ASF_PPI_PACKET_LENGTH_FIELD_SIZE
  79. #   define ASF_PPI_PACKET_LENGTH_FIELD_SIZE 0
  80. #endif
  81.  
  82. #if (ASF_PPI_FLAG_PADDING_LENGTH_FIELD_IS_BYTE == (ASF_PPI_LENGTH_TYPE_FLAGS & ASF_PPI_MASK_PADDING_LENGTH_FIELD_SIZE))
  83. #   define ASF_PPI_PADDING_LENGTH_FIELD_SIZE 1
  84. #endif
  85. #if (ASF_PPI_FLAG_PADDING_LENGTH_FIELD_IS_WORD == (ASF_PPI_LENGTH_TYPE_FLAGS & ASF_PPI_MASK_PADDING_LENGTH_FIELD_SIZE))
  86. #   define ASF_PPI_PADDING_LENGTH_FIELD_SIZE 2
  87. #endif
  88. #if (ASF_PPI_FLAG_PADDING_LENGTH_FIELD_IS_DWORD == (ASF_PPI_LENGTH_TYPE_FLAGS & ASF_PPI_MASK_PADDING_LENGTH_FIELD_SIZE))
  89. #   define ASF_PPI_PADDING_LENGTH_FIELD_SIZE 4
  90. #endif
  91. #ifndef ASF_PPI_PADDING_LENGTH_FIELD_SIZE
  92. #   define ASF_PPI_PADDING_LENGTH_FIELD_SIZE 0
  93. #endif
  94.  
  95. #if (ASF_PL_FLAG_REPLICATED_DATA_LENGTH_FIELD_IS_BYTE == (ASF_PPI_PROPERTY_FLAGS & ASF_PL_MASK_REPLICATED_DATA_LENGTH_FIELD_SIZE))
  96. #   define ASF_PAYLOAD_REPLICATED_DATA_LENGTH_FIELD_SIZE 1
  97. #endif
  98. #if (ASF_PL_FLAG_REPLICATED_DATA_LENGTH_FIELD_IS_WORD == (ASF_PPI_PROPERTY_FLAGS & ASF_PL_MASK_REPLICATED_DATA_LENGTH_FIELD_SIZE))
  99. #   define ASF_PAYLOAD_REPLICATED_DATA_LENGTH_FIELD_SIZE 2
  100. #endif
  101. #if (ASF_PL_FLAG_REPLICATED_DATA_LENGTH_FIELD_IS_DWORD == (ASF_PPI_PROPERTY_FLAGS & ASF_PL_MASK_REPLICATED_DATA_LENGTH_FIELD_SIZE))
  102. #   define ASF_PAYLOAD_REPLICATED_DATA_LENGTH_FIELD_SIZE 4
  103. #endif
  104. #ifndef ASF_PAYLOAD_REPLICATED_DATA_LENGTH_FIELD_SIZE
  105. #   define ASF_PAYLOAD_REPLICATED_DATA_LENGTH_FIELD_SIZE 0
  106. #endif
  107.  
  108. #if (ASF_PL_FLAG_OFFSET_INTO_MEDIA_OBJECT_LENGTH_FIELD_IS_BYTE == (ASF_PPI_PROPERTY_FLAGS & ASF_PL_MASK_OFFSET_INTO_MEDIA_OBJECT_LENGTH_FIELD_SIZE))
  109. #   define ASF_PAYLOAD_OFFSET_INTO_MEDIA_OBJECT_FIELD_SIZE 1
  110. #endif
  111. #if (ASF_PL_FLAG_OFFSET_INTO_MEDIA_OBJECT_LENGTH_FIELD_IS_WORD == (ASF_PPI_PROPERTY_FLAGS & ASF_PL_MASK_OFFSET_INTO_MEDIA_OBJECT_LENGTH_FIELD_SIZE))
  112. #   define ASF_PAYLOAD_OFFSET_INTO_MEDIA_OBJECT_FIELD_SIZE 2
  113. #endif
  114. #if (ASF_PL_FLAG_OFFSET_INTO_MEDIA_OBJECT_LENGTH_FIELD_IS_DWORD == (ASF_PPI_PROPERTY_FLAGS & ASF_PL_MASK_OFFSET_INTO_MEDIA_OBJECT_LENGTH_FIELD_SIZE))
  115. #   define ASF_PAYLOAD_OFFSET_INTO_MEDIA_OBJECT_FIELD_SIZE 4
  116. #endif
  117. #ifndef ASF_PAYLOAD_OFFSET_INTO_MEDIA_OBJECT_FIELD_SIZE
  118. #   define ASF_PAYLOAD_OFFSET_INTO_MEDIA_OBJECT_FIELD_SIZE 0
  119. #endif
  120.  
  121. #if (ASF_PL_FLAG_MEDIA_OBJECT_NUMBER_LENGTH_FIELD_IS_BYTE == (ASF_PPI_PROPERTY_FLAGS & ASF_PL_MASK_MEDIA_OBJECT_NUMBER_LENGTH_FIELD_SIZE))
  122. #   define ASF_PAYLOAD_MEDIA_OBJECT_NUMBER_FIELD_SIZE 1
  123. #endif
  124. #if (ASF_PL_FLAG_MEDIA_OBJECT_NUMBER_LENGTH_FIELD_IS_WORD == (ASF_PPI_PROPERTY_FLAGS & ASF_PL_MASK_MEDIA_OBJECT_NUMBER_LENGTH_FIELD_SIZE))
  125. #   define ASF_PAYLOAD_MEDIA_OBJECT_NUMBER_FIELD_SIZE 2
  126. #endif
  127. #if (ASF_PL_FLAG_MEDIA_OBJECT_NUMBER_LENGTH_FIELD_IS_DWORD == (ASF_PPI_PROPERTY_FLAGS & ASF_PL_MASK_MEDIA_OBJECT_NUMBER_LENGTH_FIELD_SIZE))
  128. #   define ASF_PAYLOAD_MEDIA_OBJECT_NUMBER_FIELD_SIZE 4
  129. #endif
  130. #ifndef ASF_PAYLOAD_MEDIA_OBJECT_NUMBER_FIELD_SIZE
  131. #   define ASF_PAYLOAD_MEDIA_OBJECT_NUMBER_FIELD_SIZE 0
  132. #endif
  133.  
  134. #if (ASF_PL_FLAG_PAYLOAD_LENGTH_FIELD_IS_BYTE == (ASF_PAYLOAD_FLAGS & ASF_PL_MASK_PAYLOAD_LENGTH_FIELD_SIZE))
  135. #   define ASF_PAYLOAD_LENGTH_FIELD_SIZE 1
  136. #endif
  137. #if (ASF_PL_FLAG_PAYLOAD_LENGTH_FIELD_IS_WORD == (ASF_PAYLOAD_FLAGS & ASF_PL_MASK_PAYLOAD_LENGTH_FIELD_SIZE))
  138. #   define ASF_PAYLOAD_LENGTH_FIELD_SIZE 2
  139. #endif
  140. #ifndef ASF_PAYLOAD_LENGTH_FIELD_SIZE
  141. #   define ASF_PAYLOAD_LENGTH_FIELD_SIZE 0
  142. #endif
  143.  
  144. #define PACKET_HEADER_MIN_SIZE \
  145.     (ASF_PACKET_ERROR_CORRECTION_FLAGS_FIELD_SIZE +       \
  146.      ASF_PACKET_ERROR_CORRECTION_DATA_SIZE +              \
  147.      1 +        /* Length Type Flags */                   \
  148.      1 +        /* Property Flags */                      \
  149.      ASF_PPI_PACKET_LENGTH_FIELD_SIZE +                   \
  150.      ASF_PPI_SEQUENCE_FIELD_SIZE +                        \
  151.      ASF_PPI_PADDING_LENGTH_FIELD_SIZE +                  \
  152.      4 +        /* Send Time Field */                     \
  153.      2)         /* Duration Field */
  154.  
  155. // Replicated Data shall be at least 8 bytes long.
  156. #define ASF_PAYLOAD_REPLICATED_DATA_LENGTH 0x08
  157.  
  158. #define PAYLOAD_HEADER_SIZE_SINGLE_PAYLOAD                \
  159.     (1 +     /* Stream Number */                          \
  160.      ASF_PAYLOAD_MEDIA_OBJECT_NUMBER_FIELD_SIZE +         \
  161.      ASF_PAYLOAD_OFFSET_INTO_MEDIA_OBJECT_FIELD_SIZE +    \
  162.      ASF_PAYLOAD_REPLICATED_DATA_LENGTH_FIELD_SIZE +      \
  163.      ASF_PAYLOAD_REPLICATED_DATA_LENGTH)
  164.  
  165. #define PAYLOAD_HEADER_SIZE_MULTIPLE_PAYLOADS             \
  166.     (1 +        /* Stream Number */                       \
  167.      ASF_PAYLOAD_MEDIA_OBJECT_NUMBER_FIELD_SIZE +         \
  168.      ASF_PAYLOAD_OFFSET_INTO_MEDIA_OBJECT_FIELD_SIZE +    \
  169.      ASF_PAYLOAD_REPLICATED_DATA_LENGTH_FIELD_SIZE +      \
  170.      ASF_PAYLOAD_REPLICATED_DATA_LENGTH +                 \
  171.      ASF_PAYLOAD_LENGTH_FIELD_SIZE)
  172.  
  173. #define SINGLE_PAYLOAD_DATA_LENGTH                        \
  174.     (PACKET_SIZE -                                        \
  175.      PACKET_HEADER_MIN_SIZE -                             \
  176.      PAYLOAD_HEADER_SIZE_SINGLE_PAYLOAD)
  177.  
  178. #define MULTI_PAYLOAD_CONSTANT                            \
  179.     (PACKET_SIZE -                                        \
  180.      PACKET_HEADER_MIN_SIZE -                             \
  181.      1 -         /* Payload Flags */                      \
  182.      2 * PAYLOAD_HEADER_SIZE_MULTIPLE_PAYLOADS)
  183.  
  184. #define DATA_HEADER_SIZE 50
  185.  
  186. typedef struct ASFPayload {
  187.     uint8_t type;
  188.     uint16_t size;
  189. } ASFPayload;
  190.  
  191. typedef struct ASFStream {
  192.     int num;
  193.     unsigned char seq;
  194.     /* use for reading */
  195.     AVPacket pkt;
  196.     int frag_offset;
  197.     int packet_obj_size;
  198.     int timestamp;
  199.     int64_t duration;
  200.     int skip_to_key;
  201.     int pkt_clean;
  202.  
  203.     int ds_span;                /* descrambling  */
  204.     int ds_packet_size;
  205.     int ds_chunk_size;
  206.  
  207.     int64_t packet_pos;
  208.  
  209.     uint16_t stream_language_index;
  210.  
  211.     int      palette_changed;
  212.     uint32_t palette[256];
  213.  
  214.     int payload_ext_ct;
  215.     ASFPayload payload[8];
  216. } ASFStream;
  217.  
  218. typedef struct ASFContext {
  219.     uint32_t seqno;
  220.     int is_streamed;
  221.     ASFStream streams[128];              ///< it's max number and it's not that big
  222.     /* non streamed additonnal info */
  223.     uint64_t nb_packets;                 ///< how many packets are there in the file, invalid if broadcasting
  224.     int64_t duration;                    ///< in 100ns units
  225.     /* packet filling */
  226.     unsigned char multi_payloads_present;
  227.     int packet_size_left;
  228.     int64_t packet_timestamp_start;
  229.     int64_t packet_timestamp_end;
  230.     unsigned int packet_nb_payloads;
  231.     uint8_t packet_buf[PACKET_SIZE];
  232.     AVIOContext pb;
  233.     /* only for reading */
  234.     uint64_t data_offset;                ///< beginning of the first data packet
  235.  
  236.     ASFIndex *index_ptr;
  237.     uint32_t nb_index_memory_alloc;
  238.     uint16_t maximum_packet;
  239.     uint32_t next_packet_number;
  240.     uint16_t next_packet_count;
  241.     uint64_t next_packet_offset;
  242.     int      next_start_sec;
  243.     int      end_sec;
  244. } ASFContext;
  245.  
  246. static const AVCodecTag codec_asf_bmp_tags[] = {
  247.     { AV_CODEC_ID_MPEG4,     MKTAG('M', '4', 'S', '2') },
  248.     { AV_CODEC_ID_MPEG4,     MKTAG('M', 'P', '4', 'S') },
  249.     { AV_CODEC_ID_MSMPEG4V3, MKTAG('M', 'P', '4', '3') },
  250.     { AV_CODEC_ID_NONE,      0 },
  251. };
  252.  
  253. #define PREROLL_TIME 3100
  254.  
  255. static void put_str16(AVIOContext *s, const char *tag)
  256. {
  257.     int len;
  258.     uint8_t *pb;
  259.     AVIOContext *dyn_buf;
  260.     if (avio_open_dyn_buf(&dyn_buf) < 0)
  261.         return;
  262.  
  263.     avio_put_str16le(dyn_buf, tag);
  264.     len = avio_close_dyn_buf(dyn_buf, &pb);
  265.     avio_wl16(s, len);
  266.     avio_write(s, pb, len);
  267.     av_freep(&pb);
  268. }
  269.  
  270. static int64_t put_header(AVIOContext *pb, const ff_asf_guid *g)
  271. {
  272.     int64_t pos;
  273.  
  274.     pos = avio_tell(pb);
  275.     ff_put_guid(pb, g);
  276.     avio_wl64(pb, 24);
  277.     return pos;
  278. }
  279.  
  280. /* update header size */
  281. static void end_header(AVIOContext *pb, int64_t pos)
  282. {
  283.     int64_t pos1;
  284.  
  285.     pos1 = avio_tell(pb);
  286.     avio_seek(pb, pos + 16, SEEK_SET);
  287.     avio_wl64(pb, pos1 - pos);
  288.     avio_seek(pb, pos1, SEEK_SET);
  289. }
  290.  
  291. /* write an asf chunk (only used in streaming case) */
  292. static void put_chunk(AVFormatContext *s, int type,
  293.                       int payload_length, int flags)
  294. {
  295.     ASFContext *asf = s->priv_data;
  296.     AVIOContext *pb = s->pb;
  297.     int length;
  298.  
  299.     length = payload_length + 8;
  300.     avio_wl16(pb, type);
  301.     avio_wl16(pb, length);      // size
  302.     avio_wl32(pb, asf->seqno);  // sequence number
  303.     avio_wl16(pb, flags);       // unknown bytes
  304.     avio_wl16(pb, length);      // size_confirm
  305.     asf->seqno++;
  306. }
  307.  
  308. /* convert from unix to windows time */
  309. static int64_t unix_to_file_time(int ti)
  310. {
  311.     int64_t t;
  312.  
  313.     t  = ti * INT64_C(10000000);
  314.     t += INT64_C(116444736000000000);
  315.     return t;
  316. }
  317.  
  318. static int32_t get_send_time(ASFContext *asf, int64_t pres_time, uint64_t *offset)
  319. {
  320.     int i;
  321.     int32_t send_time = 0;
  322.     *offset = asf->data_offset + DATA_HEADER_SIZE;
  323.     for (i = 0; i < asf->next_start_sec; i++) {
  324.         if (pres_time <= asf->index_ptr[i].send_time)
  325.             break;
  326.         send_time = asf->index_ptr[i].send_time;
  327.         *offset   = asf->index_ptr[i].offset;
  328.     }
  329.  
  330.     return send_time / 10000;
  331. }
  332.  
  333. static int asf_write_markers(AVFormatContext *s)
  334. {
  335.     ASFContext *asf = s->priv_data;
  336.     AVIOContext *pb = s->pb;
  337.     int i;
  338.     AVRational scale = {1, 10000000};
  339.     int64_t hpos = put_header(pb, &ff_asf_marker_header);
  340.  
  341.     ff_put_guid(pb, &ff_asf_reserved_4);// ASF spec mandates this reserved value
  342.     avio_wl32(pb, s->nb_chapters);     // markers count
  343.     avio_wl16(pb, 0);                  // ASF spec mandates 0 for this
  344.     avio_wl16(pb, 0);                  // name length 0, no name given
  345.  
  346.     for (i = 0; i < s->nb_chapters; i++) {
  347.         AVChapter *c = s->chapters[i];
  348.         AVDictionaryEntry *t = av_dict_get(c->metadata, "title", NULL, 0);
  349.         int64_t pres_time = av_rescale_q(c->start, c->time_base, scale);
  350.         uint64_t offset;
  351.         int32_t send_time = get_send_time(asf, pres_time, &offset);
  352.         int len = 0;
  353.         uint8_t *buf;
  354.         AVIOContext *dyn_buf;
  355.         if (t) {
  356.             if (avio_open_dyn_buf(&dyn_buf) < 0)
  357.                 return AVERROR(ENOMEM);
  358.             avio_put_str16le(dyn_buf, t->value);
  359.             len = avio_close_dyn_buf(dyn_buf, &buf);
  360.         }
  361.         avio_wl64(pb, offset);            // offset of the packet with send_time
  362.         avio_wl64(pb, pres_time + PREROLL_TIME * 10000); // presentation time
  363.         avio_wl16(pb, 12 + len);          // entry length
  364.         avio_wl32(pb, send_time);         // send time
  365.         avio_wl32(pb, 0);                 // flags, should be 0
  366.         avio_wl32(pb, len / 2);           // marker desc length in WCHARS!
  367.         if (t) {
  368.             avio_write(pb, buf, len);     // marker desc
  369.             av_freep(&buf);
  370.         }
  371.     }
  372.     end_header(pb, hpos);
  373.     return 0;
  374. }
  375.  
  376. /* write the header (used two times if non streamed) */
  377. static int asf_write_header1(AVFormatContext *s, int64_t file_size,
  378.                              int64_t data_chunk_size)
  379. {
  380.     ASFContext *asf = s->priv_data;
  381.     AVIOContext *pb = s->pb;
  382.     AVDictionaryEntry *tags[5];
  383.     int header_size, n, extra_size, extra_size2, wav_extra_size, file_time;
  384.     int has_title, has_aspect_ratio = 0;
  385.     int metadata_count;
  386.     AVCodecContext *enc;
  387.     int64_t header_offset, cur_pos, hpos;
  388.     int bit_rate;
  389.     int64_t duration;
  390.  
  391.     ff_metadata_conv(&s->metadata, ff_asf_metadata_conv, NULL);
  392.  
  393.     tags[0] = av_dict_get(s->metadata, "title", NULL, 0);
  394.     tags[1] = av_dict_get(s->metadata, "author", NULL, 0);
  395.     tags[2] = av_dict_get(s->metadata, "copyright", NULL, 0);
  396.     tags[3] = av_dict_get(s->metadata, "comment", NULL, 0);
  397.     tags[4] = av_dict_get(s->metadata, "rating", NULL, 0);
  398.  
  399.     duration       = asf->duration + PREROLL_TIME * 10000;
  400.     has_title      = tags[0] || tags[1] || tags[2] || tags[3] || tags[4];
  401.     metadata_count = av_dict_count(s->metadata);
  402.  
  403.     bit_rate = 0;
  404.     for (n = 0; n < s->nb_streams; n++) {
  405.         enc = s->streams[n]->codec;
  406.  
  407.         avpriv_set_pts_info(s->streams[n], 32, 1, 1000); /* 32 bit pts in ms */
  408.  
  409.         bit_rate += enc->bit_rate;
  410.         if (   enc->codec_type == AVMEDIA_TYPE_VIDEO
  411.             && enc->sample_aspect_ratio.num > 0
  412.             && enc->sample_aspect_ratio.den > 0)
  413.             has_aspect_ratio++;
  414.     }
  415.  
  416.     if (asf->is_streamed) {
  417.         put_chunk(s, 0x4824, 0, 0xc00); /* start of stream (length will be patched later) */
  418.     }
  419.  
  420.     ff_put_guid(pb, &ff_asf_header);
  421.     avio_wl64(pb, -1); /* header length, will be patched after */
  422.     avio_wl32(pb, 3 + has_title + !!metadata_count + s->nb_streams); /* number of chunks in header */
  423.     avio_w8(pb, 1); /* ??? */
  424.     avio_w8(pb, 2); /* ??? */
  425.  
  426.     /* file header */
  427.     header_offset = avio_tell(pb);
  428.     hpos          = put_header(pb, &ff_asf_file_header);
  429.     ff_put_guid(pb, &ff_asf_my_guid);
  430.     avio_wl64(pb, file_size);
  431.     file_time = 0;
  432.     avio_wl64(pb, unix_to_file_time(file_time));
  433.     avio_wl64(pb, asf->nb_packets); /* number of packets */
  434.     avio_wl64(pb, duration); /* end time stamp (in 100ns units) */
  435.     avio_wl64(pb, asf->duration); /* duration (in 100ns units) */
  436.     avio_wl64(pb, PREROLL_TIME); /* start time stamp */
  437.     avio_wl32(pb, (asf->is_streamed || !pb->seekable) ? 3 : 2);  /* ??? */
  438.     avio_wl32(pb, s->packet_size); /* packet size */
  439.     avio_wl32(pb, s->packet_size); /* packet size */
  440.     avio_wl32(pb, bit_rate ? bit_rate : -1); /* Maximum data rate in bps */
  441.     end_header(pb, hpos);
  442.  
  443.     /* unknown headers */
  444.     hpos = put_header(pb, &ff_asf_head1_guid);
  445.     ff_put_guid(pb, &ff_asf_head2_guid);
  446.     avio_wl16(pb, 6);
  447.     if (has_aspect_ratio) {
  448.         int64_t hpos2;
  449.         avio_wl32(pb, 26 + has_aspect_ratio * 84);
  450.         hpos2 = put_header(pb, &ff_asf_metadata_header);
  451.         avio_wl16(pb, 2 * has_aspect_ratio);
  452.         for (n = 0; n < s->nb_streams; n++) {
  453.             enc = s->streams[n]->codec;
  454.             if (   enc->codec_type == AVMEDIA_TYPE_VIDEO
  455.                 && enc->sample_aspect_ratio.num > 0
  456.                 && enc->sample_aspect_ratio.den > 0) {
  457.                 AVRational sar = enc->sample_aspect_ratio;
  458.                 avio_wl16(pb, 0);
  459.                 // the stream number is set like this below
  460.                 avio_wl16(pb, n + 1);
  461.                 avio_wl16(pb, 26); // name_len
  462.                 avio_wl16(pb,  3); // value_type
  463.                 avio_wl32(pb,  4); // value_len
  464.                 avio_put_str16le(pb, "AspectRatioX");
  465.                 avio_wl32(pb, sar.num);
  466.                 avio_wl16(pb, 0);
  467.                 // the stream number is set like this below
  468.                 avio_wl16(pb, n + 1);
  469.                 avio_wl16(pb, 26); // name_len
  470.                 avio_wl16(pb,  3); // value_type
  471.                 avio_wl32(pb,  4); // value_len
  472.                 avio_put_str16le(pb, "AspectRatioY");
  473.                 avio_wl32(pb, sar.den);
  474.             }
  475.         }
  476.         end_header(pb, hpos2);
  477.     } else {
  478.         avio_wl32(pb, 0);
  479.     }
  480.     end_header(pb, hpos);
  481.  
  482.     /* title and other infos */
  483.     if (has_title) {
  484.         int len;
  485.         uint8_t *buf;
  486.         AVIOContext *dyn_buf;
  487.  
  488.         if (avio_open_dyn_buf(&dyn_buf) < 0)
  489.             return AVERROR(ENOMEM);
  490.  
  491.         hpos = put_header(pb, &ff_asf_comment_header);
  492.  
  493.         for (n = 0; n < FF_ARRAY_ELEMS(tags); n++) {
  494.             len = tags[n] ? avio_put_str16le(dyn_buf, tags[n]->value) : 0;
  495.             avio_wl16(pb, len);
  496.         }
  497.         len = avio_close_dyn_buf(dyn_buf, &buf);
  498.         avio_write(pb, buf, len);
  499.         av_freep(&buf);
  500.         end_header(pb, hpos);
  501.     }
  502.     if (metadata_count) {
  503.         AVDictionaryEntry *tag = NULL;
  504.         hpos = put_header(pb, &ff_asf_extended_content_header);
  505.         avio_wl16(pb, metadata_count);
  506.         while ((tag = av_dict_get(s->metadata, "", tag, AV_DICT_IGNORE_SUFFIX))) {
  507.             put_str16(pb, tag->key);
  508.             avio_wl16(pb, 0);
  509.             put_str16(pb, tag->value);
  510.         }
  511.         end_header(pb, hpos);
  512.     }
  513.     /* chapters using ASF markers */
  514.     if (!asf->is_streamed && s->nb_chapters) {
  515.         int ret;
  516.         if ((ret = asf_write_markers(s)) < 0)
  517.             return ret;
  518.     }
  519.     /* stream headers */
  520.     for (n = 0; n < s->nb_streams; n++) {
  521.         int64_t es_pos;
  522.         //        ASFStream *stream = &asf->streams[n];
  523.  
  524.         enc                 = s->streams[n]->codec;
  525.         asf->streams[n].num = n + 1;
  526.         asf->streams[n].seq = 1;
  527.  
  528.         switch (enc->codec_type) {
  529.         case AVMEDIA_TYPE_AUDIO:
  530.             wav_extra_size = 0;
  531.             extra_size     = 18 + wav_extra_size;
  532.             extra_size2    = 8;
  533.             break;
  534.         default:
  535.         case AVMEDIA_TYPE_VIDEO:
  536.             wav_extra_size = enc->extradata_size;
  537.             extra_size     = 0x33 + wav_extra_size;
  538.             extra_size2    = 0;
  539.             break;
  540.         }
  541.  
  542.         hpos = put_header(pb, &ff_asf_stream_header);
  543.         if (enc->codec_type == AVMEDIA_TYPE_AUDIO) {
  544.             ff_put_guid(pb, &ff_asf_audio_stream);
  545.             ff_put_guid(pb, &ff_asf_audio_conceal_spread);
  546.         } else {
  547.             ff_put_guid(pb, &ff_asf_video_stream);
  548.             ff_put_guid(pb, &ff_asf_video_conceal_none);
  549.         }
  550.         avio_wl64(pb, 0); /* ??? */
  551.         es_pos = avio_tell(pb);
  552.         avio_wl32(pb, extra_size); /* wav header len */
  553.         avio_wl32(pb, extra_size2); /* additional data len */
  554.         avio_wl16(pb, n + 1); /* stream number */
  555.         avio_wl32(pb, 0); /* ??? */
  556.  
  557.         if (enc->codec_type == AVMEDIA_TYPE_AUDIO) {
  558.             /* WAVEFORMATEX header */
  559.             int wavsize = ff_put_wav_header(pb, enc, FF_PUT_WAV_HEADER_FORCE_WAVEFORMATEX);
  560.  
  561.             if (wavsize < 0)
  562.                 return -1;
  563.             if (wavsize != extra_size) {
  564.                 cur_pos = avio_tell(pb);
  565.                 avio_seek(pb, es_pos, SEEK_SET);
  566.                 avio_wl32(pb, wavsize); /* wav header len */
  567.                 avio_seek(pb, cur_pos, SEEK_SET);
  568.             }
  569.             /* ERROR Correction */
  570.             avio_w8(pb, 0x01);
  571.             if (enc->codec_id == AV_CODEC_ID_ADPCM_G726 || !enc->block_align) {
  572.                 avio_wl16(pb, 0x0190);
  573.                 avio_wl16(pb, 0x0190);
  574.             } else {
  575.                 avio_wl16(pb, enc->block_align);
  576.                 avio_wl16(pb, enc->block_align);
  577.             }
  578.             avio_wl16(pb, 0x01);
  579.             avio_w8(pb, 0x00);
  580.         } else {
  581.             avio_wl32(pb, enc->width);
  582.             avio_wl32(pb, enc->height);
  583.             avio_w8(pb, 2); /* ??? */
  584.             avio_wl16(pb, 40 + enc->extradata_size); /* size */
  585.  
  586.             /* BITMAPINFOHEADER header */
  587.             ff_put_bmp_header(pb, enc, ff_codec_bmp_tags, 1, 0);
  588.         }
  589.         end_header(pb, hpos);
  590.     }
  591.  
  592.     /* media comments */
  593.  
  594.     hpos = put_header(pb, &ff_asf_codec_comment_header);
  595.     ff_put_guid(pb, &ff_asf_codec_comment1_header);
  596.     avio_wl32(pb, s->nb_streams);
  597.     for (n = 0; n < s->nb_streams; n++) {
  598.         const AVCodecDescriptor *codec_desc;
  599.         const char *desc;
  600.  
  601.         enc  = s->streams[n]->codec;
  602.         codec_desc = avcodec_descriptor_get(enc->codec_id);
  603.  
  604.         if (enc->codec_type == AVMEDIA_TYPE_AUDIO)
  605.             avio_wl16(pb, 2);
  606.         else if (enc->codec_type == AVMEDIA_TYPE_VIDEO)
  607.             avio_wl16(pb, 1);
  608.         else
  609.             avio_wl16(pb, -1);
  610.  
  611.         if (enc->codec_id == AV_CODEC_ID_WMAV2)
  612.             desc = "Windows Media Audio V8";
  613.         else
  614.             desc = codec_desc ? codec_desc->name : NULL;
  615.  
  616.         if (desc) {
  617.             AVIOContext *dyn_buf;
  618.             uint8_t *buf;
  619.             int len;
  620.  
  621.             if (avio_open_dyn_buf(&dyn_buf) < 0)
  622.                 return AVERROR(ENOMEM);
  623.  
  624.             avio_put_str16le(dyn_buf, desc);
  625.             len = avio_close_dyn_buf(dyn_buf, &buf);
  626.             avio_wl16(pb, len / 2); // "number of characters" = length in bytes / 2
  627.  
  628.             avio_write(pb, buf, len);
  629.             av_freep(&buf);
  630.         } else
  631.             avio_wl16(pb, 0);
  632.  
  633.         avio_wl16(pb, 0); /* no parameters */
  634.  
  635.         /* id */
  636.         if (enc->codec_type == AVMEDIA_TYPE_AUDIO) {
  637.             avio_wl16(pb, 2);
  638.             avio_wl16(pb, enc->codec_tag);
  639.         } else {
  640.             avio_wl16(pb, 4);
  641.             avio_wl32(pb, enc->codec_tag);
  642.         }
  643.         if (!enc->codec_tag)
  644.             return -1;
  645.     }
  646.     end_header(pb, hpos);
  647.  
  648.     /* patch the header size fields */
  649.  
  650.     cur_pos     = avio_tell(pb);
  651.     header_size = cur_pos - header_offset;
  652.     if (asf->is_streamed) {
  653.         header_size += 8 + 30 + DATA_HEADER_SIZE;
  654.  
  655.         avio_seek(pb, header_offset - 10 - 30, SEEK_SET);
  656.         avio_wl16(pb, header_size);
  657.         avio_seek(pb, header_offset - 2 - 30, SEEK_SET);
  658.         avio_wl16(pb, header_size);
  659.  
  660.         header_size -= 8 + 30 + DATA_HEADER_SIZE;
  661.     }
  662.     header_size += 24 + 6;
  663.     avio_seek(pb, header_offset - 14, SEEK_SET);
  664.     avio_wl64(pb, header_size);
  665.     avio_seek(pb, cur_pos, SEEK_SET);
  666.  
  667.     /* movie chunk, followed by packets of packet_size */
  668.     asf->data_offset = cur_pos;
  669.     ff_put_guid(pb, &ff_asf_data_header);
  670.     avio_wl64(pb, data_chunk_size);
  671.     ff_put_guid(pb, &ff_asf_my_guid);
  672.     avio_wl64(pb, asf->nb_packets); /* nb packets */
  673.     avio_w8(pb, 1); /* ??? */
  674.     avio_w8(pb, 1); /* ??? */
  675.     return 0;
  676. }
  677.  
  678. static int asf_write_header(AVFormatContext *s)
  679. {
  680.     ASFContext *asf = s->priv_data;
  681.  
  682.     s->packet_size  = PACKET_SIZE;
  683.     s->max_interleave_delta = 0;
  684.     asf->nb_packets = 0;
  685.  
  686.     asf->index_ptr             = av_malloc(sizeof(ASFIndex) * ASF_INDEX_BLOCK);
  687.     if (!asf->index_ptr)
  688.         return AVERROR(ENOMEM);
  689.     asf->nb_index_memory_alloc = ASF_INDEX_BLOCK;
  690.     asf->maximum_packet        = 0;
  691.  
  692.     /* the data-chunk-size has to be 50 (DATA_HEADER_SIZE), which is
  693.      * data_size - asf->data_offset at the moment this function is done.
  694.      * It is needed to use asf as a streamable format. */
  695.     if (asf_write_header1(s, 0, DATA_HEADER_SIZE) < 0) {
  696.         //av_free(asf);
  697.         av_freep(&asf->index_ptr);
  698.         return -1;
  699.     }
  700.  
  701.     avio_flush(s->pb);
  702.  
  703.     asf->packet_nb_payloads     = 0;
  704.     asf->packet_timestamp_start = -1;
  705.     asf->packet_timestamp_end   = -1;
  706.     ffio_init_context(&asf->pb, asf->packet_buf, s->packet_size, 1,
  707.                       NULL, NULL, NULL, NULL);
  708.  
  709.     if (s->avoid_negative_ts < 0)
  710.         s->avoid_negative_ts = 1;
  711.  
  712.     return 0;
  713. }
  714.  
  715. static int asf_write_stream_header(AVFormatContext *s)
  716. {
  717.     ASFContext *asf = s->priv_data;
  718.  
  719.     asf->is_streamed = 1;
  720.  
  721.     return asf_write_header(s);
  722. }
  723.  
  724. static int put_payload_parsing_info(AVFormatContext *s,
  725.                                     unsigned sendtime, unsigned duration,
  726.                                     int nb_payloads, int padsize)
  727. {
  728.     ASFContext *asf = s->priv_data;
  729.     AVIOContext *pb = s->pb;
  730.     int ppi_size, i;
  731.     int64_t start = avio_tell(pb);
  732.  
  733.     int iLengthTypeFlags = ASF_PPI_LENGTH_TYPE_FLAGS;
  734.  
  735.     padsize -= PACKET_HEADER_MIN_SIZE;
  736.     if (asf->multi_payloads_present)
  737.         padsize--;
  738.     av_assert0(padsize >= 0);
  739.  
  740.     avio_w8(pb, ASF_PACKET_ERROR_CORRECTION_FLAGS);
  741.     for (i = 0; i < ASF_PACKET_ERROR_CORRECTION_DATA_SIZE; i++)
  742.         avio_w8(pb, 0x0);
  743.  
  744.     if (asf->multi_payloads_present)
  745.         iLengthTypeFlags |= ASF_PPI_FLAG_MULTIPLE_PAYLOADS_PRESENT;
  746.  
  747.     if (padsize > 0) {
  748.         if (padsize < 256)
  749.             iLengthTypeFlags |= ASF_PPI_FLAG_PADDING_LENGTH_FIELD_IS_BYTE;
  750.         else
  751.             iLengthTypeFlags |= ASF_PPI_FLAG_PADDING_LENGTH_FIELD_IS_WORD;
  752.     }
  753.     avio_w8(pb, iLengthTypeFlags);
  754.  
  755.     avio_w8(pb, ASF_PPI_PROPERTY_FLAGS);
  756.  
  757.     if (iLengthTypeFlags & ASF_PPI_FLAG_PADDING_LENGTH_FIELD_IS_WORD)
  758.         avio_wl16(pb, padsize - 2);
  759.     if (iLengthTypeFlags & ASF_PPI_FLAG_PADDING_LENGTH_FIELD_IS_BYTE)
  760.         avio_w8(pb, padsize - 1);
  761.  
  762.     avio_wl32(pb, sendtime);
  763.     avio_wl16(pb, duration);
  764.     if (asf->multi_payloads_present)
  765.         avio_w8(pb, nb_payloads | ASF_PAYLOAD_FLAGS);
  766.  
  767.     ppi_size = avio_tell(pb) - start;
  768.  
  769.     return ppi_size;
  770. }
  771.  
  772. static void flush_packet(AVFormatContext *s)
  773. {
  774.     ASFContext *asf = s->priv_data;
  775.     int packet_hdr_size, packet_filled_size;
  776.  
  777.     av_assert0(asf->packet_timestamp_end >= asf->packet_timestamp_start);
  778.  
  779.     if (asf->is_streamed)
  780.         put_chunk(s, 0x4424, s->packet_size, 0);
  781.  
  782.     packet_hdr_size = put_payload_parsing_info(s,
  783.                                                asf->packet_timestamp_start,
  784.                                                asf->packet_timestamp_end - asf->packet_timestamp_start,
  785.                                                asf->packet_nb_payloads,
  786.                                                asf->packet_size_left);
  787.  
  788.     packet_filled_size = PACKET_SIZE - asf->packet_size_left;
  789.     av_assert0(packet_hdr_size <= asf->packet_size_left);
  790.     memset(asf->packet_buf + packet_filled_size, 0, asf->packet_size_left);
  791.  
  792.     avio_write(s->pb, asf->packet_buf, s->packet_size - packet_hdr_size);
  793.  
  794.     avio_flush(s->pb);
  795.     asf->nb_packets++;
  796.     asf->packet_nb_payloads     = 0;
  797.     asf->packet_timestamp_start = -1;
  798.     asf->packet_timestamp_end   = -1;
  799.     ffio_init_context(&asf->pb, asf->packet_buf, s->packet_size, 1,
  800.                       NULL, NULL, NULL, NULL);
  801. }
  802.  
  803. static void put_payload_header(AVFormatContext *s, ASFStream *stream,
  804.                                int64_t presentation_time, int m_obj_size,
  805.                                int m_obj_offset, int payload_len, int flags)
  806. {
  807.     ASFContext *asf = s->priv_data;
  808.     AVIOContext *pb = &asf->pb;
  809.     int val;
  810.  
  811.     val = stream->num;
  812.     if (flags & AV_PKT_FLAG_KEY)
  813.         val |= ASF_PL_FLAG_KEY_FRAME;
  814.     avio_w8(pb, val);
  815.  
  816.     avio_w8(pb, stream->seq);     // Media object number
  817.     avio_wl32(pb, m_obj_offset);  // Offset Into Media Object
  818.  
  819.     // Replicated Data shall be at least 8 bytes long.
  820.     // The first 4 bytes of data shall contain the
  821.     // Size of the Media Object that the payload belongs to.
  822.     // The next 4 bytes of data shall contain the
  823.     // Presentation Time for the media object that the payload belongs to.
  824.     avio_w8(pb, ASF_PAYLOAD_REPLICATED_DATA_LENGTH);
  825.  
  826.     avio_wl32(pb, m_obj_size);        // Replicated Data - Media Object Size
  827.     avio_wl32(pb, (uint32_t) presentation_time); // Replicated Data - Presentation Time
  828.  
  829.     if (asf->multi_payloads_present) {
  830.         avio_wl16(pb, payload_len);   // payload length
  831.     }
  832. }
  833.  
  834. static void put_frame(AVFormatContext *s, ASFStream *stream, AVStream *avst,
  835.                       int64_t timestamp, const uint8_t *buf,
  836.                       int m_obj_size, int flags)
  837. {
  838.     ASFContext *asf = s->priv_data;
  839.     int m_obj_offset, payload_len, frag_len1;
  840.  
  841.     m_obj_offset = 0;
  842.     while (m_obj_offset < m_obj_size) {
  843.         payload_len = m_obj_size - m_obj_offset;
  844.         if (asf->packet_timestamp_start == -1) {
  845.             asf->multi_payloads_present = (payload_len < MULTI_PAYLOAD_CONSTANT);
  846.  
  847.             asf->packet_size_left = PACKET_SIZE;
  848.             if (asf->multi_payloads_present) {
  849.                 frag_len1 = MULTI_PAYLOAD_CONSTANT - 1;
  850.             } else {
  851.                 frag_len1 = SINGLE_PAYLOAD_DATA_LENGTH;
  852.             }
  853.             asf->packet_timestamp_start = timestamp;
  854.         } else {
  855.             // multi payloads
  856.             frag_len1 = asf->packet_size_left -
  857.                         PAYLOAD_HEADER_SIZE_MULTIPLE_PAYLOADS -
  858.                         PACKET_HEADER_MIN_SIZE - 1;
  859.  
  860.             if (frag_len1 < payload_len &&
  861.                 avst->codec->codec_type == AVMEDIA_TYPE_AUDIO) {
  862.                 flush_packet(s);
  863.                 continue;
  864.             }
  865.         }
  866.         if (frag_len1 > 0) {
  867.             if (payload_len > frag_len1)
  868.                 payload_len = frag_len1;
  869.             else if (payload_len == (frag_len1 - 1))
  870.                 payload_len = frag_len1 - 2;  // additional byte need to put padding length
  871.  
  872.             put_payload_header(s, stream, timestamp + PREROLL_TIME,
  873.                                m_obj_size, m_obj_offset, payload_len, flags);
  874.             avio_write(&asf->pb, buf, payload_len);
  875.  
  876.             if (asf->multi_payloads_present)
  877.                 asf->packet_size_left -= (payload_len + PAYLOAD_HEADER_SIZE_MULTIPLE_PAYLOADS);
  878.             else
  879.                 asf->packet_size_left -= (payload_len + PAYLOAD_HEADER_SIZE_SINGLE_PAYLOAD);
  880.             asf->packet_timestamp_end = timestamp;
  881.  
  882.             asf->packet_nb_payloads++;
  883.         } else {
  884.             payload_len = 0;
  885.         }
  886.         m_obj_offset += payload_len;
  887.         buf          += payload_len;
  888.  
  889.         if (!asf->multi_payloads_present)
  890.             flush_packet(s);
  891.         else if (asf->packet_size_left <= (PAYLOAD_HEADER_SIZE_MULTIPLE_PAYLOADS + PACKET_HEADER_MIN_SIZE + 1))
  892.             flush_packet(s);
  893.         else if (asf->packet_nb_payloads == ASF_PAYLOADS_PER_PACKET)
  894.             flush_packet(s);
  895.     }
  896.     stream->seq++;
  897. }
  898.  
  899. static int update_index(AVFormatContext *s, int start_sec,
  900.                          uint32_t packet_number, uint16_t packet_count,
  901.                          uint64_t packet_offset)
  902. {
  903.     ASFContext *asf = s->priv_data;
  904.  
  905.     if (start_sec > asf->next_start_sec) {
  906.         int i;
  907.  
  908.         if (!asf->next_start_sec) {
  909.             asf->next_packet_number = packet_number;
  910.             asf->next_packet_count  = packet_count;
  911.             asf->next_packet_offset = packet_offset;
  912.         }
  913.  
  914.         if (start_sec > asf->nb_index_memory_alloc) {
  915.             int err;
  916.             asf->nb_index_memory_alloc = (start_sec + ASF_INDEX_BLOCK) & ~(ASF_INDEX_BLOCK - 1);
  917.             if ((err = av_reallocp_array(&asf->index_ptr,
  918.                                          asf->nb_index_memory_alloc,
  919.                                          sizeof(*asf->index_ptr))) < 0) {
  920.                 asf->nb_index_memory_alloc = 0;
  921.                 return err;
  922.             }
  923.         }
  924.         for (i = asf->next_start_sec; i < start_sec; i++) {
  925.             asf->index_ptr[i].packet_number = asf->next_packet_number;
  926.             asf->index_ptr[i].packet_count  = asf->next_packet_count;
  927.             asf->index_ptr[i].send_time     = asf->next_start_sec * INT64_C(10000000);
  928.             asf->index_ptr[i].offset        = asf->next_packet_offset;
  929.  
  930.         }
  931.     }
  932.     asf->maximum_packet     = FFMAX(asf->maximum_packet, packet_count);
  933.     asf->next_packet_number = packet_number;
  934.     asf->next_packet_count  = packet_count;
  935.     asf->next_packet_offset = packet_offset;
  936.     asf->next_start_sec     = start_sec;
  937.  
  938.     return 0;
  939. }
  940.  
  941. static int asf_write_packet(AVFormatContext *s, AVPacket *pkt)
  942. {
  943.     ASFContext *asf = s->priv_data;
  944.     AVIOContext *pb = s->pb;
  945.     ASFStream *stream;
  946.     AVCodecContext *codec;
  947.     uint32_t packet_number;
  948.     int64_t pts;
  949.     int start_sec;
  950.     int flags = pkt->flags;
  951.     int ret;
  952.     uint64_t offset = avio_tell(pb);
  953.  
  954.     codec  = s->streams[pkt->stream_index]->codec;
  955.     stream = &asf->streams[pkt->stream_index];
  956.  
  957.     if (codec->codec_type == AVMEDIA_TYPE_AUDIO)
  958.         flags &= ~AV_PKT_FLAG_KEY;
  959.  
  960.     pts = (pkt->pts != AV_NOPTS_VALUE) ? pkt->pts : pkt->dts;
  961.     av_assert0(pts != AV_NOPTS_VALUE);
  962.     if (   pts < - PREROLL_TIME
  963.         || pts > (INT_MAX-3)/10000LL * ASF_INDEXED_INTERVAL - PREROLL_TIME) {
  964.         av_log(s, AV_LOG_ERROR, "input pts %"PRId64" is invalid\n", pts);
  965.         return AVERROR(EINVAL);
  966.     }
  967.     pts *= 10000;
  968.     asf->duration = FFMAX(asf->duration, pts + pkt->duration * 10000);
  969.  
  970.     packet_number = asf->nb_packets;
  971.     put_frame(s, stream, s->streams[pkt->stream_index],
  972.               pkt->dts, pkt->data, pkt->size, flags);
  973.  
  974.     start_sec = (int)((PREROLL_TIME * 10000 + pts + ASF_INDEXED_INTERVAL - 1)
  975.               / ASF_INDEXED_INTERVAL);
  976.  
  977.     /* check index */
  978.     if ((!asf->is_streamed) && (flags & AV_PKT_FLAG_KEY)) {
  979.         uint16_t packet_count = asf->nb_packets - packet_number;
  980.         ret = update_index(s, start_sec, packet_number, packet_count, offset);
  981.         if (ret < 0)
  982.             return ret;
  983.     }
  984.     asf->end_sec = start_sec;
  985.  
  986.     return 0;
  987. }
  988.  
  989. static int asf_write_index(AVFormatContext *s, const ASFIndex *index,
  990.                            uint16_t max, uint32_t count)
  991. {
  992.     AVIOContext *pb = s->pb;
  993.     int i;
  994.  
  995.     ff_put_guid(pb, &ff_asf_simple_index_header);
  996.     avio_wl64(pb, 24 + 16 + 8 + 4 + 4 + (4 + 2) * count);
  997.     ff_put_guid(pb, &ff_asf_my_guid);
  998.     avio_wl64(pb, ASF_INDEXED_INTERVAL);
  999.     avio_wl32(pb, max);
  1000.     avio_wl32(pb, count);
  1001.     for (i = 0; i < count; i++) {
  1002.         avio_wl32(pb, index[i].packet_number);
  1003.         avio_wl16(pb, index[i].packet_count);
  1004.     }
  1005.  
  1006.     return 0;
  1007. }
  1008.  
  1009. static int asf_write_trailer(AVFormatContext *s)
  1010. {
  1011.     ASFContext *asf = s->priv_data;
  1012.     int64_t file_size, data_size;
  1013.     int ret;
  1014.  
  1015.     /* flush the current packet */
  1016.     if (asf->pb.buf_ptr > asf->pb.buffer)
  1017.         flush_packet(s);
  1018.  
  1019.     /* write index */
  1020.     data_size = avio_tell(s->pb);
  1021.     if (!asf->is_streamed && asf->next_start_sec) {
  1022.         if ((ret = update_index(s, asf->end_sec + 1, 0, 0, 0)) < 0)
  1023.             return ret;
  1024.         asf_write_index(s, asf->index_ptr, asf->maximum_packet, asf->next_start_sec);
  1025.     }
  1026.     avio_flush(s->pb);
  1027.  
  1028.     if (asf->is_streamed || !s->pb->seekable) {
  1029.         put_chunk(s, 0x4524, 0, 0); /* end of stream */
  1030.     } else {
  1031.         /* rewrite an updated header */
  1032.         file_size = avio_tell(s->pb);
  1033.         avio_seek(s->pb, 0, SEEK_SET);
  1034.         asf_write_header1(s, file_size, data_size - asf->data_offset);
  1035.     }
  1036.  
  1037.     av_freep(&asf->index_ptr);
  1038.     return 0;
  1039. }
  1040.  
  1041. #if CONFIG_ASF_MUXER
  1042. AVOutputFormat ff_asf_muxer = {
  1043.     .name           = "asf",
  1044.     .long_name      = NULL_IF_CONFIG_SMALL("ASF (Advanced / Active Streaming Format)"),
  1045.     .mime_type      = "video/x-ms-asf",
  1046.     .extensions     = "asf,wmv,wma",
  1047.     .priv_data_size = sizeof(ASFContext),
  1048.     .audio_codec    = AV_CODEC_ID_WMAV2,
  1049.     .video_codec    = AV_CODEC_ID_MSMPEG4V3,
  1050.     .write_header   = asf_write_header,
  1051.     .write_packet   = asf_write_packet,
  1052.     .write_trailer  = asf_write_trailer,
  1053.     .flags          = AVFMT_GLOBALHEADER,
  1054.     .codec_tag      = (const AVCodecTag * const []) {
  1055.         codec_asf_bmp_tags, ff_codec_bmp_tags, ff_codec_wav_tags, 0
  1056.     },
  1057. };
  1058. #endif /* CONFIG_ASF_MUXER */
  1059.  
  1060. #if CONFIG_ASF_STREAM_MUXER
  1061. AVOutputFormat ff_asf_stream_muxer = {
  1062.     .name           = "asf_stream",
  1063.     .long_name      = NULL_IF_CONFIG_SMALL("ASF (Advanced / Active Streaming Format)"),
  1064.     .mime_type      = "video/x-ms-asf",
  1065.     .extensions     = "asf,wmv,wma",
  1066.     .priv_data_size = sizeof(ASFContext),
  1067.     .audio_codec    = AV_CODEC_ID_WMAV2,
  1068.     .video_codec    = AV_CODEC_ID_MSMPEG4V3,
  1069.     .write_header   = asf_write_stream_header,
  1070.     .write_packet   = asf_write_packet,
  1071.     .write_trailer  = asf_write_trailer,
  1072.     .flags          = AVFMT_GLOBALHEADER,
  1073.     .codec_tag      = (const AVCodecTag * const []) {
  1074.         codec_asf_bmp_tags, ff_codec_bmp_tags, ff_codec_wav_tags, 0
  1075.     },
  1076. };
  1077. #endif /* CONFIG_ASF_STREAM_MUXER */
  1078.