]> git.sesse.net Git - vlc/blobdiff - modules/codec/a52.c
access_pvr: really fix the potential memleak.
[vlc] / modules / codec / a52.c
index 7f6b1f1446d017b676113431e684c833ea467967..df5675265f4d153e5ef90ef9bf1a1ba89c7466b2 100644 (file)
@@ -1,12 +1,12 @@
 /*****************************************************************************
  * a52.c: parse A/52 audio sync info and packetize the stream
  *****************************************************************************
- * Copyright (C) 2001-2002 VideoLAN
- * $Id: a52.c,v 1.29 2003/11/16 21:07:30 gbazin Exp $
+ * Copyright (C) 2001-2002 the VideoLAN team
+ * $Id$
  *
- * Authors: Stéphane Borel <stef@via.ecp.fr>
+ * Authors: Stéphane Borel <stef@via.ecp.fr>
  *          Christophe Massiot <massiot@via.ecp.fr>
- *          Gildas Bazin <gbazin@netcourrier.com>
+ *          Gildas Bazin <gbazin@videolan.org>
  *
  * This program is free software; you can redistribute it and/or modify
  * it under the terms of the GNU General Public License as published by
  *
  * You should have received a copy of the GNU General Public License
  * along with this program; if not, write to the Free Software
- * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111, USA.
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
  *****************************************************************************/
 
 /*****************************************************************************
  * Preamble
  *****************************************************************************/
-#include <stdio.h>
-#include <stdlib.h>
-#include <string.h>                                              /* memcpy() */
+#ifdef HAVE_CONFIG_H
+# include "config.h"
+#endif
 
-#include <vlc/vlc.h>
-#include <vlc/decoder.h>
-#include <vlc/input.h>
+#include <vlc_common.h>
+#include <vlc_plugin.h>
+#include <vlc_codec.h>
+#include <vlc_aout.h>
+#include <vlc_block_helper.h>
+#include <vlc_bits.h>
 
-#include "vlc_block_helper.h"
+/*****************************************************************************
+ * Module descriptor
+ *****************************************************************************/
+static int  OpenDecoder   ( vlc_object_t * );
+static int  OpenPacketizer( vlc_object_t * );
+static void CloseCommon   ( vlc_object_t * );
 
-#define A52_HEADER_SIZE 7
+vlc_module_begin ()
+    set_description( N_("A/52 parser") )
+    set_capability( "decoder", 100 )
+    set_callbacks( OpenDecoder, CloseCommon )
+    set_category( CAT_INPUT )
+    set_subcategory( SUBCAT_INPUT_ACODEC )
+
+    add_submodule ()
+    set_description( N_("A/52 audio packetizer") )
+    set_capability( "packetizer", 10 )
+    set_callbacks( OpenPacketizer, CloseCommon )
+vlc_module_end ()
 
 /*****************************************************************************
  * decoder_sys_t : decoder descriptor
  *****************************************************************************/
+
+#define A52_HEADER_SIZE 7
+
 struct decoder_sys_t
 {
     /* Module mode */
-    vlc_bool_t b_packetizer;
+    bool b_packetizer;
 
     /*
      * Input properties
@@ -61,7 +83,6 @@ struct decoder_sys_t
     mtime_t i_pts;
     int i_frame_size, i_bit_rate;
     unsigned int i_rate, i_channels, i_channels_conf;
-
 };
 
 enum {
@@ -77,82 +98,76 @@ enum {
 /****************************************************************************
  * Local prototypes
  ****************************************************************************/
-static int  OpenDecoder   ( vlc_object_t * );
-static int  OpenPacketizer( vlc_object_t * );
-static void CloseDecoder  ( vlc_object_t * );
 static void *DecodeBlock  ( decoder_t *, block_t ** );
 
-static int  SyncInfo      ( const byte_t *, int *, int *, int *,int * );
+static int  SyncInfo      ( const uint8_t *, unsigned int *, unsigned int *,
+                            unsigned int *, int * );
 
 static uint8_t       *GetOutBuffer ( decoder_t *, void ** );
 static aout_buffer_t *GetAoutBuffer( decoder_t * );
 static block_t       *GetSoutBuffer( decoder_t * );
 
 /*****************************************************************************
- * Module descriptor
- *****************************************************************************/
-vlc_module_begin();
-    set_description( _("A/52 parser") );
-    set_capability( "decoder", 100 );
-    set_callbacks( OpenDecoder, CloseDecoder );
-
-    add_submodule();
-    set_description( _("A/52 audio packetizer") );
-    set_capability( "packetizer", 10 );
-    set_callbacks( OpenPacketizer, CloseDecoder );
-vlc_module_end();
-
-/*****************************************************************************
- * OpenDecoder: probe the decoder and return score
+ * OpenCommon: probe the decoder/packetizer and return score
  *****************************************************************************/
-static int OpenDecoder( vlc_object_t *p_this )
+static int OpenCommon( vlc_object_t *p_this, bool b_packetizer )
 {
     decoder_t *p_dec = (decoder_t*)p_this;
     decoder_sys_t *p_sys;
+    vlc_fourcc_t i_codec;
 
-    if( p_dec->fmt_in.i_codec != VLC_FOURCC('a','5','2',' ')
-          && p_dec->fmt_in.i_codec != VLC_FOURCC('a','5','2','b') )
+    switch( p_dec->fmt_in.i_codec )
     {
+    case VLC_FOURCC('a','5','2',' '):
+    case VLC_FOURCC('a','5','2','b'):
+        i_codec = VLC_FOURCC('a','5','2',' ');
+        break;
+    case VLC_FOURCC('e','a','c','3'):
+        /* XXX ugly hack, a52 does not support eac3 so no eac3 pass-through
+         * support */
+        if( !b_packetizer )
+            return VLC_EGENERIC;
+        i_codec = VLC_FOURCC('e','a','c','3');
+        break;
+    default:
         return VLC_EGENERIC;
     }
 
     /* Allocate the memory needed to store the decoder's structure */
     if( ( p_dec->p_sys = p_sys =
           (decoder_sys_t *)malloc(sizeof(decoder_sys_t)) ) == NULL )
-    {
-        msg_Err( p_dec, "out of memory" );
-        return VLC_EGENERIC;
-    }
+        return VLC_ENOMEM;
 
     /* Misc init */
-    p_sys->b_packetizer = VLC_FALSE;
+    p_sys->b_packetizer = b_packetizer;
     p_sys->i_state = STATE_NOSYNC;
     aout_DateSet( &p_sys->end_date, 0 );
 
-    p_sys->bytestream = block_BytestreamInit( p_dec );
+    p_sys->bytestream = block_BytestreamInit();
 
     /* Set output properties */
     p_dec->fmt_out.i_cat = AUDIO_ES;
-    p_dec->fmt_out.i_codec = VLC_FOURCC('a','5','2',' ');
+    p_dec->fmt_out.i_codec = i_codec;
+    p_dec->fmt_out.audio.i_rate = 0; /* So end_date gets initialized */
 
     /* Set callback */
-    p_dec->pf_decode_audio = (aout_buffer_t *(*)(decoder_t *, block_t **))
-        DecodeBlock;
-    p_dec->pf_packetize    = (block_t *(*)(decoder_t *, block_t **))
-        DecodeBlock;
-
+    if( b_packetizer )
+        p_dec->pf_packetize    = (block_t *(*)(decoder_t *, block_t **))
+            DecodeBlock;
+    else
+        p_dec->pf_decode_audio = (aout_buffer_t *(*)(decoder_t *, block_t **))
+            DecodeBlock;
     return VLC_SUCCESS;
 }
 
-static int OpenPacketizer( vlc_object_t *p_this )
+static int OpenDecoder( vlc_object_t *p_this )
 {
-    decoder_t *p_dec = (decoder_t*)p_this;
-
-    int i_ret = OpenDecoder( p_this );
-
-    if( i_ret == VLC_SUCCESS ) p_dec->p_sys->b_packetizer = VLC_TRUE;
+    return OpenCommon( p_this, false );
+}
 
-    return i_ret;
+static int OpenPacketizer( vlc_object_t *p_this )
+{
+    return OpenCommon( p_this, true );
 }
 
 /****************************************************************************
@@ -169,16 +184,23 @@ static void *DecodeBlock( decoder_t *p_dec, block_t **pp_block )
 
     if( !pp_block || !*pp_block ) return NULL;
 
-    if( !aout_DateGet( &p_sys->end_date ) && !(*pp_block)->i_pts )
+    if( (*pp_block)->i_flags&(BLOCK_FLAG_DISCONTINUITY|BLOCK_FLAG_CORRUPTED) )
     {
-        /* We've just started the stream, wait for the first PTS. */
+        if( (*pp_block)->i_flags&BLOCK_FLAG_CORRUPTED )
+        {
+            p_sys->i_state = STATE_NOSYNC;
+            block_BytestreamEmpty( &p_sys->bytestream );
+        }
+        aout_DateSet( &p_sys->end_date, 0 );
         block_Release( *pp_block );
         return NULL;
     }
 
-    if( (*pp_block)->b_discontinuity )
+    if( !aout_DateGet( &p_sys->end_date ) && !(*pp_block)->i_pts )
     {
-        p_sys->i_state = STATE_NOSYNC;
+        /* We've just started the stream, wait for the first PTS. */
+        block_Release( *pp_block );
+        return NULL;
     }
 
     block_BytestreamPush( &p_sys->bytestream, *pp_block );
@@ -200,7 +222,7 @@ static void *DecodeBlock( decoder_t *p_dec, block_t **pp_block )
             }
             if( p_sys->i_state != STATE_SYNC )
             {
-               block_BytestreamFlush( &p_sys->bytestream );
+                block_BytestreamFlush( &p_sys->bytestream );
 
                 /* Need more data */
                 return NULL;
@@ -253,6 +275,14 @@ static void *DecodeBlock( decoder_t *p_dec, block_t **pp_block )
                 return NULL;
             }
 
+            if( p_sys->b_packetizer &&
+                p_header[0] == 0 && p_header[1] == 0 )
+            {
+                /* A52 wav files and audio CD's use stuffing */
+                p_sys->i_state = STATE_GET_DATA;
+                break;
+            }
+
             if( p_header[0] != 0x0b || p_header[1] != 0x77 )
             {
                 msg_Dbg( p_dec, "emulated sync word "
@@ -278,7 +308,7 @@ static void *DecodeBlock( decoder_t *p_dec, block_t **pp_block )
         case STATE_SEND_DATA:
             if( !(p_buf = GetOutBuffer( p_dec, &p_out_buffer )) )
             {
-                //p_dec->b_error = VLC_TRUE;
+                //p_dec->b_error = true;
                 return NULL;
             }
 
@@ -303,9 +333,9 @@ static void *DecodeBlock( decoder_t *p_dec, block_t **pp_block )
 }
 
 /*****************************************************************************
- * CloseDecoder: clean up the decoder
+ * CloseCommon: clean up the decoder
  *****************************************************************************/
-static void CloseDecoder( vlc_object_t *p_this )
+static void CloseCommon( vlc_object_t *p_this )
 {
     decoder_t *p_dec = (decoder_t*)p_this;
     decoder_sys_t *p_sys = p_dec->p_sys;
@@ -334,7 +364,6 @@ static uint8_t *GetOutBuffer( decoder_t *p_dec, void **pp_out_buffer )
 
     p_dec->fmt_out.audio.i_rate     = p_sys->i_rate;
     p_dec->fmt_out.audio.i_channels = p_sys->i_channels;
-    p_dec->fmt_out.audio.i_bitrate  = p_sys->i_bit_rate;
     p_dec->fmt_out.audio.i_bytes_per_frame = p_sys->i_frame_size;
     p_dec->fmt_out.audio.i_frame_length = A52_FRAME_NB;
 
@@ -342,6 +371,8 @@ static uint8_t *GetOutBuffer( decoder_t *p_dec, void **pp_out_buffer )
     p_dec->fmt_out.audio.i_physical_channels =
         p_sys->i_channels_conf & AOUT_CHAN_PHYSMASK;
 
+    p_dec->fmt_out.i_bitrate = p_sys->i_bit_rate;
+
     if( p_sys->b_packetizer )
     {
         block_t *p_sout_buffer = GetSoutBuffer( p_dec );
@@ -366,7 +397,7 @@ static aout_buffer_t *GetAoutBuffer( decoder_t *p_dec )
     decoder_sys_t *p_sys = p_dec->p_sys;
     aout_buffer_t *p_buf;
 
-    p_buf = p_dec->pf_aout_buffer_new( p_dec, A52_FRAME_NB  );
+    p_buf = decoder_NewAudioBuffer( p_dec, A52_FRAME_NB  );
     if( p_buf == NULL ) return NULL;
 
     p_buf->start_date = aout_DateGet( &p_sys->end_date );
@@ -388,22 +419,42 @@ static block_t *GetSoutBuffer( decoder_t *p_dec )
 
     p_block->i_pts = p_block->i_dts = aout_DateGet( &p_sys->end_date );
 
-    p_block->i_length = aout_DateIncrement( &p_sys->end_date, A52_FRAME_NB ) -
-        p_block->i_pts;
+    p_block->i_length =
+        aout_DateIncrement( &p_sys->end_date, A52_FRAME_NB ) - p_block->i_pts;
 
     return p_block;
 }
 
-/*****************************************************************************
- * SyncInfo: parse A/52 sync info
- *****************************************************************************
+/* Tables */
+static const struct
+{
+    unsigned int i_count;
+    unsigned int i_configuration;
+} p_acmod[8] = {
+    { 2, AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT | AOUT_CHAN_DUALMONO },   /* Dual-channel 1+1 */
+    { 1, AOUT_CHAN_CENTER },                                        /* Mono 1/0 */
+    { 2, AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT },                        /* Stereo 2/0 */
+    { 3, AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT | AOUT_CHAN_CENTER },     /* 3F 3/0 */
+    { 3, AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT | AOUT_CHAN_REARCENTER }, /* 2F1R 2/1 */
+    { 4, AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT | AOUT_CHAN_CENTER |
+         AOUT_CHAN_REARCENTER },                                    /* 3F1R 3/1 */
+    { 5, AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT |
+         AOUT_CHAN_REARLEFT | AOUT_CHAN_REARRIGHT },                /* 2F2R 2/2 */
+    { 6, AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT | AOUT_CHAN_CENTER |
+         AOUT_CHAN_REARLEFT | AOUT_CHAN_REARRIGHT },                /* 3F2R 3/2 */
+};
+
+/**
+ * It parse AC3 sync info.
+ *
  * This code is borrowed from liba52 by Aaron Holtzman & Michel Lespinasse,
  * since we don't want to oblige S/PDIF people to use liba52 just to get
  * their SyncInfo...
- *****************************************************************************/
-static int SyncInfo( const byte_t * p_buf,
-                     int * pi_channels, int * pi_channels_conf,
-                     int * pi_sample_rate, int * pi_bit_rate )
+ */
+static int SyncInfoAC3( const uint8_t *p_buf,
+                        unsigned int *pi_channels,
+                        unsigned int *pi_channels_conf,
+                        unsigned int *pi_sample_rate, int *pi_bit_rate )
 {
     static const uint8_t halfrate[12] = { 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 2, 3 };
     static const int rate[] = { 32,  40,  48,  56,  64,  80,  96, 112,
@@ -416,11 +467,7 @@ static int SyncInfo( const byte_t * p_buf,
     int half;
     int acmod;
 
-    if ((p_buf[0] != 0x0b) || (p_buf[1] != 0x77))        /* syncword */
-        return 0;
-
-    if (p_buf[5] >= 0x60)                /* bsid >= 12 */
-        return 0;
+    /* */
     half = halfrate[p_buf[5] >> 3];
 
     /* acmod, dsurmod and lfeon */
@@ -432,56 +479,10 @@ static int SyncInfo( const byte_t * p_buf,
         *pi_channels_conf = AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT
                             | AOUT_CHAN_DOLBYSTEREO;
     }
-    else switch ( acmod )
+    else
     {
-    case 0x0:
-        /* Dual-mono = stereo + dual-mono */
-        *pi_channels = 2;
-        *pi_channels_conf = AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT
-                            | AOUT_CHAN_DUALMONO;
-        break;
-    case 0x1:
-        /* Mono */
-        *pi_channels = 1;
-        *pi_channels_conf = AOUT_CHAN_CENTER;
-        break;
-    case 0x2:
-        /* Stereo */
-        *pi_channels = 2;
-        *pi_channels_conf = AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT;
-        break;
-    case 0x3:
-        /* 3F */
-        *pi_channels = 3;
-        *pi_channels_conf = AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT
-                            | AOUT_CHAN_CENTER;
-        break;
-    case 0x4:
-        /* 2F1R */
-        *pi_channels = 3;
-        *pi_channels_conf = AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT
-                            | AOUT_CHAN_REARCENTER;
-        break;
-    case 0x5:
-        /* 3F1R */
-        *pi_channels = 4;
-        *pi_channels_conf = AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT | AOUT_CHAN_CENTER
-                            | AOUT_CHAN_REARCENTER;
-        break;
-    case 0x6:
-        /* 2F2R */
-        *pi_channels = 4;
-        *pi_channels_conf = AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT
-                            | AOUT_CHAN_REARLEFT | AOUT_CHAN_REARRIGHT;
-        break;
-    case 0x7:
-        /* 3F2R */
-        *pi_channels = 5;
-        *pi_channels_conf = AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT | AOUT_CHAN_CENTER
-                            | AOUT_CHAN_REARLEFT | AOUT_CHAN_REARRIGHT;
-        break;
-    default:
-        return 0;
+        *pi_channels      = p_acmod[acmod].i_count;
+        *pi_channels_conf = p_acmod[acmod].i_configuration;
     }
 
     if ( p_buf[6] & lfeon[acmod] )
@@ -510,3 +511,81 @@ static int SyncInfo( const byte_t * p_buf,
         return 0;
     }
 }
+
+/**
+ * It parse E-AC3 sync info
+ */
+static int SyncInfoEAC3( const uint8_t *p_buf,
+                         unsigned int *pi_channels,
+                         unsigned int *pi_channels_conf,
+                         unsigned int *pi_sample_rate, int *pi_bit_rate )
+{
+    static const int pi_samplerate[3] = { 48000, 44100, 32000 };
+    bs_t s;
+    int i_frame_size;
+    int i_fscod, i_fscod2;
+    int i_numblkscod;
+    int i_acmod, i_lfeon;
+    int i_bytes;
+
+
+    bs_init( &s, (void*)p_buf, A52_HEADER_SIZE );
+    bs_skip( &s, 16 +   /* start code */
+                 2 +    /* stream type */
+                 3 );   /* substream id */
+    i_frame_size = bs_read( &s, 11 );
+    if( i_frame_size < 2 )
+        return 0;
+    i_bytes = 2 * ( i_frame_size + 1 );
+
+    i_fscod = bs_read( &s, 2 );
+    if( i_fscod == 0x03 )
+    {
+        i_fscod2 = bs_read( &s, 2 );
+        if( i_fscod2 == 0X03 )
+            return 0;
+        *pi_sample_rate = pi_samplerate[i_fscod2] / 2;
+        i_numblkscod = 6;
+    }
+    else
+    {
+        static const int pi_blocks[4] = { 1, 2, 3, 6 };
+
+        *pi_sample_rate = pi_samplerate[i_fscod];
+        i_numblkscod = pi_blocks[bs_read( &s, 2 )];
+    }
+
+    i_acmod = bs_read( &s, 3 );
+    i_lfeon = bs_read1( &s );
+
+    *pi_channels      = p_acmod[i_acmod].i_count + i_lfeon;
+    *pi_channels_conf = p_acmod[i_acmod].i_configuration | ( i_lfeon ? AOUT_CHAN_LFE : 0);
+    *pi_bit_rate = 8 * i_bytes * (*pi_sample_rate) / (i_numblkscod * 256);
+
+    return i_bytes;
+}
+
+static int SyncInfo( const uint8_t *p_buf,
+                     unsigned int *pi_channels,
+                     unsigned int *pi_channels_conf,
+                     unsigned int *pi_sample_rate, int *pi_bit_rate )
+{
+    int bsid;
+
+    /* Check synword */
+    if( p_buf[0] != 0x0b || p_buf[1] != 0x77 )
+        return 0;
+
+    /* Check bsid */
+    bsid = p_buf[5] >> 3;
+    if( bsid > 16 )
+        return 0;
+
+    if( bsid <= 10 )
+        return SyncInfoAC3( p_buf, pi_channels, pi_channels_conf,
+                            pi_sample_rate, pi_bit_rate );
+    else
+        return SyncInfoEAC3( p_buf, pi_channels, pi_channels_conf,
+                             pi_sample_rate, pi_bit_rate );
+}
+