asf.c
上传用户:kjfoods
上传日期:2020-07-06
资源大小:29949k
文件大小:10k
源码类别:

midi

开发平台:

Unix_Linux

  1. /*****************************************************************************
  2.  * asf.c: MMS access plug-in
  3.  *****************************************************************************
  4.  * Copyright (C) 2001-2004 the VideoLAN team
  5.  * $Id: 0ef7c2477204b55a4b8b63ca07a362236996ea49 $
  6.  *
  7.  * Authors: Laurent Aimar <fenrir@via.ecp.fr>
  8.  *
  9.  * This program is free software; you can redistribute it and/or modify
  10.  * it under the terms of the GNU General Public License as published by
  11.  * the Free Software Foundation; either version 2 of the License, or
  12.  * (at your option) any later version.
  13.  *
  14.  * This program is distributed in the hope that it will be useful,
  15.  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  16.  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17.  * GNU General Public License for more details.
  18.  *
  19.  * You should have received a copy of the GNU General Public License
  20.  * along with this program; if not, write to the Free Software
  21.  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
  22.  *****************************************************************************/
  23. #ifdef HAVE_CONFIG_H
  24. # include "config.h"
  25. #endif
  26. #include <vlc_common.h>
  27. #include "asf.h"
  28. #include "buffer.h"
  29. static int CmpGuid( const guid_t *p_guid1, const guid_t *p_guid2 )
  30. {
  31.     return( ( p_guid1->v1 == p_guid2->v1 &&
  32.               p_guid1->v2 == p_guid2->v2 &&
  33.               p_guid1->v3 == p_guid2->v3 &&
  34.               p_guid1->v4[0] == p_guid2->v4[0] &&
  35.               p_guid1->v4[1] == p_guid2->v4[1] &&
  36.               p_guid1->v4[2] == p_guid2->v4[2] &&
  37.               p_guid1->v4[3] == p_guid2->v4[3] &&
  38.               p_guid1->v4[4] == p_guid2->v4[4] &&
  39.               p_guid1->v4[5] == p_guid2->v4[5] &&
  40.               p_guid1->v4[6] == p_guid2->v4[6] &&
  41.               p_guid1->v4[7] == p_guid2->v4[7] ) ? 1 : 0 );
  42. }
  43. void  GenerateGuid ( guid_t *p_guid )
  44. {
  45.     int i;
  46.     srand( mdate() & 0xffffffff );
  47.     /* FIXME should be generated using random data */
  48.     p_guid->v1 = 0xbabac001;
  49.     p_guid->v2 = ( (uint64_t)rand() << 16 ) / RAND_MAX;
  50.     p_guid->v3 = ( (uint64_t)rand() << 16 ) / RAND_MAX;
  51.     for( i = 0; i < 8; i++ )
  52.     {
  53.         p_guid->v4[i] = ( (uint64_t)rand() * 256 ) / RAND_MAX;
  54.     }
  55. }
  56. void  asf_HeaderParse ( asf_header_t *hdr,
  57.                             uint8_t *p_header, int i_header )
  58. {
  59.     var_buffer_t buffer;
  60.     guid_t      guid;
  61.     uint64_t    i_size;
  62.     int         i;
  63.     hdr->i_file_size = 0;
  64.     hdr->i_data_packets_count = 0;
  65.     hdr->i_min_data_packet_size = 0;
  66.     for( i = 0; i < 128; i++ )
  67.     {
  68.         hdr->stream[i].i_cat = ASF_STREAM_UNKNOWN;
  69.         hdr->stream[i].i_selected = 0;
  70.         hdr->stream[i].i_bitrate = -1;
  71.     }
  72.     var_buffer_initread( &buffer, p_header, i_header );
  73.     var_buffer_getguid( &buffer, &guid );
  74.     if( !CmpGuid( &guid, &asf_object_header_guid ) )
  75.     {
  76.         /* ERROR: */
  77.     }
  78.     var_buffer_getmemory( &buffer, NULL, 30 - 16 );
  79.     for( ;; )
  80.     {
  81.         var_buffer_getguid( &buffer, &guid );
  82.         i_size = var_buffer_get64( &buffer );
  83.         if( CmpGuid( &guid, &asf_object_file_properties_guid ) )
  84.         {
  85.             var_buffer_getmemory( &buffer, NULL, 16 );
  86.             hdr->i_file_size            = var_buffer_get64( &buffer );
  87.             var_buffer_getmemory( &buffer, NULL, 8 );
  88.             hdr->i_data_packets_count   = var_buffer_get64( &buffer );
  89.             var_buffer_getmemory( &buffer, NULL, 8+8+8+4);
  90.             hdr->i_min_data_packet_size = var_buffer_get32( &buffer );
  91.             var_buffer_getmemory( &buffer, NULL, i_size - 24 - 16 - 8 - 8 - 8 - 8-8-8-4 - 4);
  92.         }
  93.         else if( CmpGuid( &guid, &asf_object_header_extension_guid ) )
  94.         {
  95.             /* Enter it */
  96.             var_buffer_getmemory( &buffer, NULL, 46 - 24 );
  97.         }
  98.         else if( CmpGuid( &guid, &asf_object_extended_stream_properties_guid ) )
  99.         {
  100.             /* Grrrrrr */
  101.             int16_t i_count1, i_count2;
  102.             int i_subsize;
  103.             int i;
  104.             var_buffer_getmemory( &buffer, NULL, 84 - 24 );
  105.             i_count1 = var_buffer_get16( &buffer );
  106.             i_count2 = var_buffer_get16( &buffer );
  107.             i_subsize = 88;
  108.             for( i = 0; i < i_count1; i++ )
  109.             {
  110.                 int i_len;
  111.                 var_buffer_get16( &buffer );
  112.                 i_len = var_buffer_get16( &buffer );
  113.                 var_buffer_getmemory( &buffer, NULL, i_len );
  114.                 i_subsize = 4 + i_len;
  115.             }
  116.             for( i = 0; i < i_count2; i++ )
  117.             {
  118.                 int i_len;
  119.                 var_buffer_getmemory( &buffer, NULL, 16 + 2 );
  120.                 i_len = var_buffer_get32( &buffer );
  121.                 var_buffer_getmemory( &buffer, NULL, i_len );
  122.                 i_subsize += 16 + 6 + i_len;
  123.             }
  124.             if( i_size - i_subsize <= 24 )
  125.             {
  126.                 var_buffer_getmemory( &buffer, NULL, i_size - i_subsize );
  127.             }
  128.             /* It's a hack we just skip the first part of the object until
  129.              * the embed stream properties if any (ugly, but whose fault ?) */
  130.         }
  131.         else if( CmpGuid( &guid, &asf_object_stream_properties_guid ) )
  132.         {
  133.             int     i_stream_id;
  134.             guid_t  stream_type;
  135.             var_buffer_getguid( &buffer, &stream_type );
  136.             var_buffer_getmemory( &buffer, NULL, 32 );
  137.             i_stream_id = var_buffer_get8( &buffer ) & 0x7f;
  138.             var_buffer_getmemory( &buffer, NULL, i_size - 24 - 32 - 16 - 1);
  139.             if( CmpGuid( &stream_type, &asf_object_stream_type_video ) )
  140.             {
  141.                 hdr->stream[i_stream_id].i_cat = ASF_STREAM_VIDEO;
  142.             }
  143.             else if( CmpGuid( &stream_type, &asf_object_stream_type_audio ) )
  144.             {
  145.                 hdr->stream[i_stream_id].i_cat = ASF_STREAM_AUDIO;
  146.             }
  147.             else
  148.             {
  149.                 hdr->stream[i_stream_id].i_cat = ASF_STREAM_UNKNOWN;
  150.             }
  151.         }
  152.         else if ( CmpGuid( &guid, &asf_object_bitrate_properties_guid ) )
  153.         {
  154.             int     i_count;
  155.             uint8_t i_stream_id;
  156.             i_count = var_buffer_get16( &buffer );
  157.             i_size -= 2;
  158.             while( i_count > 0 )
  159.             {
  160.                 i_stream_id = var_buffer_get16( &buffer )&0x7f;
  161.                 hdr->stream[i_stream_id].i_bitrate =  var_buffer_get32( &buffer );
  162.                 i_count--;
  163.                 i_size -= 6;
  164.             }
  165.             var_buffer_getmemory( &buffer, NULL, i_size - 24 );
  166.         }
  167.         else
  168.         {
  169.             // skip unknown guid
  170.             var_buffer_getmemory( &buffer, NULL, i_size - 24 );
  171.         }
  172.         if( var_buffer_readempty( &buffer ) )
  173.             return;
  174.     }
  175. }
  176. void  asf_StreamSelect  ( asf_header_t *hdr,
  177.                               int i_bitrate_max,
  178.                               bool b_all, bool b_audio, bool b_video )
  179. {
  180.     /* XXX FIXME use mututal eclusion information */
  181.     int i;
  182.     int i_audio, i_video;
  183.     int i_bitrate_total;
  184. #if 0
  185.     char *psz_stream;
  186. #endif
  187.     i_audio = 0;
  188.     i_video = 0;
  189.     i_bitrate_total = 0;
  190.     if( b_all )
  191.     {
  192.         /* select all valid stream */
  193.         for( i = 1; i < 128; i++ )
  194.         {
  195.             if( hdr->stream[i].i_cat != ASF_STREAM_UNKNOWN )
  196.             {
  197.                 hdr->stream[i].i_selected = 1;
  198.             }
  199.         }
  200.         return;
  201.     }
  202.     else
  203.     {
  204.         for( i = 0; i < 128; i++ )
  205.         {
  206.             /* by default, not selected */
  207.             hdr->stream[i].i_selected = 0;
  208.         }
  209.     }
  210.     /* big test:
  211.      * select a stream if
  212.      *    - no audio nor video stream
  213.      *    - or:
  214.      *         - if i_bitrate_max not set keep the highest bitrate
  215.      *         - if i_bitrate_max is set, keep stream that make we used best
  216.      *           quality regarding i_bitrate_max
  217.      *
  218.      * XXX: little buggy:
  219.      *        - it doesn't use mutual exclusion info..
  220.      *        - when selecting a better stream we could select
  221.      *        something that make i_bitrate_total> i_bitrate_max
  222.      */
  223.     for( i = 1; i < 128; i++ )
  224.     {
  225.         if( hdr->stream[i].i_cat == ASF_STREAM_UNKNOWN )
  226.         {
  227.             continue;
  228.         }
  229.         else if( hdr->stream[i].i_cat == ASF_STREAM_AUDIO && b_audio &&
  230.                  ( i_audio <= 0 ||
  231.                     ( ( ( hdr->stream[i].i_bitrate > hdr->stream[i_audio].i_bitrate &&
  232.                           ( i_bitrate_total + hdr->stream[i].i_bitrate - hdr->stream[i_audio].i_bitrate
  233.                                             < i_bitrate_max || !i_bitrate_max) ) ||
  234.                         ( hdr->stream[i].i_bitrate < hdr->stream[i_audio].i_bitrate &&
  235.                               i_bitrate_max != 0 && i_bitrate_total > i_bitrate_max )
  236.                       ) )  ) )
  237.         {
  238.             /* unselect old stream */
  239.             if( i_audio > 0 )
  240.             {
  241.                 hdr->stream[i_audio].i_selected = 0;
  242.                 if( hdr->stream[i_audio].i_bitrate> 0 )
  243.                 {
  244.                     i_bitrate_total -= hdr->stream[i_audio].i_bitrate;
  245.                 }
  246.             }
  247.             hdr->stream[i].i_selected = 1;
  248.             if( hdr->stream[i].i_bitrate> 0 )
  249.             {
  250.                 i_bitrate_total += hdr->stream[i].i_bitrate;
  251.             }
  252.             i_audio = i;
  253.         }
  254.         else if( hdr->stream[i].i_cat == ASF_STREAM_VIDEO && b_video &&
  255.                  ( i_video <= 0 ||
  256.                     (
  257.                         ( ( hdr->stream[i].i_bitrate > hdr->stream[i_video].i_bitrate &&
  258.                             ( i_bitrate_total + hdr->stream[i].i_bitrate - hdr->stream[i_video].i_bitrate
  259.                                             < i_bitrate_max || !i_bitrate_max) ) ||
  260.                           ( hdr->stream[i].i_bitrate < hdr->stream[i_video].i_bitrate &&
  261.                             i_bitrate_max != 0 && i_bitrate_total > i_bitrate_max )
  262.                         ) ) )  )
  263.         {
  264.             /* unselect old stream */
  265.             if( i_video > 0 )
  266.             {
  267.                 hdr->stream[i_video].i_selected = 0;
  268.                 if( hdr->stream[i_video].i_bitrate> 0 )
  269.                 {
  270.                     i_bitrate_total -= hdr->stream[i_video].i_bitrate;
  271.                 }
  272.             }
  273.             hdr->stream[i].i_selected = 1;
  274.             if( hdr->stream[i].i_bitrate> 0 )
  275.             {
  276.                 i_bitrate_total += hdr->stream[i].i_bitrate;
  277.             }
  278.             i_video = i;
  279.         }
  280.     }
  281. }