]> git.sesse.net Git - vlc/blobdiff - modules/demux/ogg.c
Parse meta data at ogg level.
[vlc] / modules / demux / ogg.c
index db22e7c8ff5f09a081a20abb37d9b1e5005d44b8..577934ccff6ba45108c991eef80150f2ee499cd5 100644 (file)
@@ -39,6 +39,8 @@
 
 #include <vlc_codecs.h>
 #include <vlc_bits.h>
+#include <vlc_charset.h>
+#include "vorbis.h"
 
 /*****************************************************************************
  * Module descriptor
 static int  Open ( vlc_object_t * );
 static void Close( vlc_object_t * );
 
-vlc_module_begin();
-    set_shortname ( "OGG" );
-    set_description( N_("OGG demuxer" ) );
-    set_category( CAT_INPUT );
-    set_subcategory( SUBCAT_INPUT_DEMUX );
-    set_capability( "demux", 50 );
-    set_callbacks( Open, Close );
-    add_shortcut( "ogg" );
-vlc_module_end();
+vlc_module_begin ()
+    set_shortname ( "OGG" )
+    set_description( N_("OGG demuxer" ) )
+    set_category( CAT_INPUT )
+    set_subcategory( SUBCAT_INPUT_DEMUX )
+    set_capability( "demux", 50 )
+    set_callbacks( Open, Close )
+    add_shortcut( "ogg" )
+vlc_module_end ()
 
 
 /*****************************************************************************
@@ -86,14 +88,14 @@ typedef struct logical_stream_s
     mtime_t          i_previous_pcr;
 
     /* Misc */
-    int b_reinit;
+    bool b_reinit;
     int i_granule_shift;
 
     /* kate streams have the number of headers in the ID header */
     int i_kate_num_headers;
 
     /* for Annodex logical bitstreams */
-    int secondary_header_packets;
+    int i_secondary_header_packets;
 
 } logical_stream_t;
 
@@ -111,10 +113,17 @@ struct demux_sys_t
     mtime_t i_pcr;
 
     /* stream state */
+    int     i_bos;
     int     i_eos;
 
     /* bitrate */
     int     i_bitrate;
+
+    /* after reading all headers, the first data page is stuffed into the relevant stream, ready to use */
+    bool    b_page_waiting;
+
+    /* */
+    vlc_meta_t *p_meta;
 };
 
 /* OggDS headers for the new header format (used in ogm files) */
@@ -184,6 +193,9 @@ static void Ogg_EndOfStream( demux_t *p_demux );
 static void Ogg_LogicalStreamDelete( demux_t *p_demux, logical_stream_t *p_stream );
 static bool Ogg_LogicalStreamResetEsFormat( demux_t *p_demux, logical_stream_t *p_stream );
 
+/* */
+static void Ogg_ExtractMeta( demux_t *p_demux, vlc_fourcc_t i_codec, const uint8_t *p_headers, int i_headers );
+
 /* Logical bitstream headers */
 static void Ogg_ReadTheoraHeader( logical_stream_t *, ogg_packet * );
 static void Ogg_ReadVorbisHeader( logical_stream_t *, ogg_packet * );
@@ -191,7 +203,7 @@ static void Ogg_ReadSpeexHeader( logical_stream_t *, ogg_packet * );
 static void Ogg_ReadKateHeader( logical_stream_t *, ogg_packet * );
 static void Ogg_ReadFlacHeader( demux_t *, logical_stream_t *, ogg_packet * );
 static void Ogg_ReadAnnodexHeader( vlc_object_t *, logical_stream_t *, ogg_packet * );
-static void Ogg_ReadDiracHeader( logical_stream_t *, ogg_packet * );
+static bool Ogg_ReadDiracHeader( logical_stream_t *, ogg_packet * );
 
 /*****************************************************************************
  * Open: initializes ogg demux structures
@@ -223,10 +235,15 @@ static int Open( vlc_object_t * p_this )
     p_sys->p_old_stream = NULL;
 
     /* Begnning of stream, tell the demux to look for elementary streams. */
+    p_sys->i_bos = 0;
     p_sys->i_eos = 0;
 
     /* Initialize the Ogg physical bitstream parser */
     ogg_sync_init( &p_sys->oy );
+    p_sys->b_page_waiting = false;
+
+    /* */
+    p_sys->p_meta = NULL;
 
     return VLC_SUCCESS;
 }
@@ -287,46 +304,66 @@ static int Demux( demux_t * p_demux )
     }
 
     /*
-     * Demux an ogg page from the stream
+     * The first data page of a physical stream is stored in the relevant logical stream
+     * in Ogg_FindLogicalStreams. Therefore, we must not read a page and only update the
+     * stream it belongs to if we haven't processed this first page yet. If we do, we
+     * will only process that first page whenever we find the second page for this stream.
+     * While this is fine for Vorbis and Theora, which are continuous codecs, which means
+     * the second page will arrive real quick, this is not fine for Kate, whose second
+     * data page will typically arrive much later.
+     * This means it is now possible to seek right at the start of a stream where the last
+     * logical stream is Kate, without having to wait for the second data page to unblock
+     * the first one, which is the one that triggers the 'no more headers to backup' code.
+     * And, as we all know, seeking without having backed up all headers is bad, since the
+     * codec will fail to initialize if it's missing its headers.
      */
-    if( Ogg_ReadPage( p_demux, &oggpage ) != VLC_SUCCESS )
+    if( !p_sys->b_page_waiting)
     {
-        return 0; /* EOF */
+        /*
+         * Demux an ogg page from the stream
+         */
+        if( Ogg_ReadPage( p_demux, &oggpage ) != VLC_SUCCESS )
+            return 0; /* EOF */
+
+        /* Test for End of Stream */
+        if( ogg_page_eos( &oggpage ) )
+            p_sys->i_eos++;
     }
 
-    /* Test for End of Stream */
-    if( ogg_page_eos( &oggpage ) ) p_sys->i_eos++;
-
 
     for( i_stream = 0; i_stream < p_sys->i_streams; i_stream++ )
     {
         logical_stream_t *p_stream = p_sys->pp_stream[i_stream];
 
-        if( ogg_stream_pagein( &p_stream->os, &oggpage ) != 0 )
-            continue;
+        /* if we've just pulled page, look for the right logical stream */
+        if( !p_sys->b_page_waiting )
+        {
+            if( ogg_stream_pagein( &p_stream->os, &oggpage ) != 0 )
+                continue;
+        }
 
         while( ogg_stream_packetout( &p_stream->os, &oggpacket ) > 0 )
         {
             /* Read info from any secondary header packets, if there are any */
-            if( p_stream->secondary_header_packets > 0 )
+            if( p_stream->i_secondary_header_packets > 0 )
             {
                 if( p_stream->fmt.i_codec == VLC_FOURCC('t','h','e','o') &&
                         oggpacket.bytes >= 7 &&
                         ! memcmp( oggpacket.packet, "\x80theora", 7 ) )
                 {
                     Ogg_ReadTheoraHeader( p_stream, &oggpacket );
-                    p_stream->secondary_header_packets = 0;
+                    p_stream->i_secondary_header_packets = 0;
                 }
                 else if( p_stream->fmt.i_codec == VLC_FOURCC('v','o','r','b') &&
                         oggpacket.bytes >= 7 &&
                         ! memcmp( oggpacket.packet, "\x01vorbis", 7 ) )
                 {
                     Ogg_ReadVorbisHeader( p_stream, &oggpacket );
-                    p_stream->secondary_header_packets = 0;
+                    p_stream->i_secondary_header_packets = 0;
                 }
-                else if ( p_stream->fmt.i_codec == VLC_FOURCC('c','m','m','l') )
+                else if( p_stream->fmt.i_codec == VLC_FOURCC('c','m','m','l') )
                 {
-                    p_stream->secondary_header_packets = 0;
+                    p_stream->i_secondary_header_packets = 0;
                 }
             }
 
@@ -338,7 +375,7 @@ static int Demux( demux_t * p_demux )
 
                 if( p_stream->i_pcr >= 0 )
                 {
-                    p_stream->b_reinit = 0;
+                    p_stream->b_reinit = false;
                 }
                 else
                 {
@@ -366,11 +403,16 @@ static int Demux( demux_t * p_demux )
 
             Ogg_DecodePacket( p_demux, p_stream, &oggpacket );
         }
-        break;
+
+        if( !p_sys->b_page_waiting )
+            break;
     }
 
-    i_stream = 0; p_sys->i_pcr = -1;
-    for( ; i_stream < p_sys->i_streams; i_stream++ )
+    /* if a page was waiting, it's now processed */
+    p_sys->b_page_waiting = false;
+
+    p_sys->i_pcr = -1;
+    for( i_stream = 0; i_stream < p_sys->i_streams; i_stream++ )
     {
         logical_stream_t *p_stream = p_sys->pp_stream[i_stream];
 
@@ -384,9 +426,7 @@ static int Demux( demux_t * p_demux )
     }
 
     if( p_sys->i_pcr >= 0 )
-    {
         es_out_Control( p_demux->out, ES_OUT_SET_PCR, p_sys->i_pcr );
-    }
 
     return 1;
 }
@@ -397,12 +437,19 @@ static int Demux( demux_t * p_demux )
 static int Control( demux_t *p_demux, int i_query, va_list args )
 {
     demux_sys_t *p_sys  = p_demux->p_sys;
+    vlc_meta_t *p_meta;
     int64_t *pi64;
     bool *pb_bool;
     int i;
 
     switch( i_query )
     {
+        case DEMUX_GET_META:
+            p_meta = (vlc_meta_t *)va_arg( args, vlc_meta_t* );
+            if( p_sys->p_meta )
+                vlc_meta_Merge( p_meta, p_sys->p_meta );
+            return VLC_SUCCESS;
+
         case DEMUX_HAS_UNSUPPORTED_META:
             pb_bool = (bool*)va_arg( args, bool* );
             *pb_bool = true;
@@ -417,17 +464,27 @@ static int Control( demux_t *p_demux, int i_query, va_list args )
             return VLC_EGENERIC;
 
         case DEMUX_SET_POSITION:
+            /* forbid seeking if we haven't initialized all logical bitstreams yet;
+               if we allowed, some headers would not get backed up and decoder init
+               would fail, making that logical stream unusable */
+            if( p_sys->i_bos > 0 )
+            {
+                return VLC_EGENERIC;
+            }
+
             for( i = 0; i < p_sys->i_streams; i++ )
             {
                 logical_stream_t *p_stream = p_sys->pp_stream[i];
 
                 /* we'll trash all the data until we find the next pcr */
-                p_stream->b_reinit = 1;
+                p_stream->b_reinit = true;
                 p_stream->i_pcr = -1;
                 p_stream->i_interpolated_pcr = -1;
                 ogg_stream_reset( &p_stream->os );
             }
             ogg_sync_reset( &p_sys->oy );
+            /* XXX The break/return is missing on purpose as
+             * demux_vaControlHelper will do the last part of the job */
 
         default:
             return demux_vaControlHelper( p_demux->s, 0, -1, p_sys->i_bitrate,
@@ -523,6 +580,7 @@ static void Ogg_DecodePacket( demux_t *p_demux,
     bool b_selected;
     int i_header_len = 0;
     mtime_t i_pts = -1, i_interpolated_pts;
+    demux_sys_t *p_ogg = p_demux->p_sys;
 
     /* Sanity check */
     if( !p_oggpacket->bytes )
@@ -623,8 +681,7 @@ static void Ogg_DecodePacket( demux_t *p_demux,
                 free( p_stream->fmt.p_extra );
 
                 p_stream->fmt.i_extra = p_stream->i_headers;
-                p_stream->fmt.p_extra =
-                    realloc( p_stream->fmt.p_extra, p_stream->i_headers );
+                p_stream->fmt.p_extra = malloc( p_stream->i_headers );
                 if( p_stream->fmt.p_extra )
                     memcpy( p_stream->fmt.p_extra, p_stream->p_headers,
                             p_stream->i_headers );
@@ -632,8 +689,15 @@ static void Ogg_DecodePacket( demux_t *p_demux,
                     p_stream->fmt.i_extra = 0;
 
                 if( Ogg_LogicalStreamResetEsFormat( p_demux, p_stream ) )
-                    es_out_Control( p_demux->out, ES_OUT_SET_FMT,
+                    es_out_Control( p_demux->out, ES_OUT_SET_ES_FMT,
                                     p_stream->p_es, &p_stream->fmt );
+
+                if( p_stream->i_headers > 0 )
+                    Ogg_ExtractMeta( p_demux, p_stream->fmt.i_codec,
+                                     p_stream->p_headers, p_stream->i_headers );
+
+                /* we're not at BOS anymore for this logical stream */
+                p_ogg->i_bos--;
             }
         }
         else
@@ -674,17 +738,21 @@ static void Ogg_DecodePacket( demux_t *p_demux,
     i_interpolated_pts = p_stream->i_interpolated_pcr;
     Ogg_UpdatePCR( p_stream, p_oggpacket );
 
-    if( p_stream->i_pcr >= 0 )
+    /* SPU streams are typically discontinuous, do not mind large gaps */
+    if( p_stream->fmt.i_cat != SPU_ES )
     {
-        /* This is for streams where the granulepos of the header packets
-         * doesn't match these of the data packets (eg. ogg web radios). */
-        if( p_stream->i_previous_pcr == 0 &&
-            p_stream->i_pcr  > 3 * DEFAULT_PTS_DELAY )
+        if( p_stream->i_pcr >= 0 )
         {
-            es_out_Control( p_demux->out, ES_OUT_RESET_PCR );
+            /* This is for streams where the granulepos of the header packets
+             * doesn't match these of the data packets (eg. ogg web radios). */
+            if( p_stream->i_previous_pcr == 0 &&
+                p_stream->i_pcr  > 3 * DEFAULT_PTS_DELAY )
+            {
+                es_out_Control( p_demux->out, ES_OUT_RESET_PCR );
 
-            /* Call the pace control */
-            es_out_Control( p_demux->out, ES_OUT_SET_PCR, p_stream->i_pcr );
+                /* Call the pace control */
+                es_out_Control( p_demux->out, ES_OUT_SET_PCR, p_stream->i_pcr );
+            }
         }
     }
 
@@ -827,7 +895,6 @@ static int Ogg_FindLogicalStreams( demux_t *p_demux )
              * We found the beginning of our first logical stream. */
             while( ogg_page_bos( &oggpage ) )
             {
-                logical_stream_t **pp_sav = p_ogg->pp_stream;
                 logical_stream_t *p_stream;
 
                 p_stream = malloc( sizeof(logical_stream_t) );
@@ -838,7 +905,7 @@ static int Ogg_FindLogicalStreams( demux_t *p_demux )
 
                 memset( p_stream, 0, sizeof(logical_stream_t) );
                 p_stream->p_headers = 0;
-                p_stream->secondary_header_packets = 0;
+                p_stream->i_secondary_header_packets = 0;
 
                 es_format_Init( &p_stream->fmt, 0, 0 );
                 es_format_Init( &p_stream->fmt_old, 0, 0 );
@@ -848,7 +915,7 @@ static int Ogg_FindLogicalStreams( demux_t *p_demux )
                 ogg_stream_init( &p_stream->os, p_stream->i_serial_no );
 
                 /* Extract the initial header from the first page and verify
-                 * the codec type of tis Ogg bitstream */
+                 * the codec type of this Ogg bitstream */
                 if( ogg_stream_pagein( &p_stream->os, &oggpage ) < 0 )
                 {
                     /* error. stream version mismatch perhaps */
@@ -924,8 +991,14 @@ static int Ogg_FindLogicalStreams( demux_t *p_demux )
                 else if( oggpacket.bytes >= 5 &&
                          ! memcmp( oggpacket.packet, "BBCD\x00", 5 ) )
                 {
-                    Ogg_ReadDiracHeader( p_stream, &oggpacket );
-                    msg_Dbg( p_demux, "found dirac header" );
+                    if( Ogg_ReadDiracHeader( p_stream, &oggpacket ) )
+                        msg_Dbg( p_demux, "found dirac header" );
+                    else
+                    {
+                        msg_Warn( p_demux, "found dirac header isn't decodable" );
+                        free( p_stream );
+                        p_ogg->i_streams--;
+                    }
                 }
                 /* Check for Tarkin header */
                 else if( oggpacket.bytes >= 7 &&
@@ -1225,6 +1298,16 @@ static int Ogg_FindLogicalStreams( demux_t *p_demux )
                     return VLC_EGENERIC;
             }
 
+            /* we'll need to get all headers for all of those streams
+               that we have to backup headers for */
+            p_ogg->i_bos = 0;
+            for( i_stream = 0; i_stream < p_ogg->i_streams; i_stream++ )
+            {
+                if( p_ogg->pp_stream[i_stream]->b_force_backup )
+                    p_ogg->i_bos++;
+            }
+
+
             /* This is the first data page, which means we are now finished
              * with the initial pages. We just need to store it in the relevant
              * bitstream. */
@@ -1233,6 +1316,7 @@ static int Ogg_FindLogicalStreams( demux_t *p_demux )
                 if( ogg_stream_pagein( &p_ogg->pp_stream[i_stream]->os,
                                        &oggpage ) == 0 )
                 {
+                    p_ogg->b_page_waiting = true;
                     break;
                 }
             }
@@ -1285,7 +1369,14 @@ static int Ogg_BeginningOfStream( demux_t *p_demux )
         }
 
         if( !p_stream->p_es )
+        {
+            /* Better be safe than sorry when possible with ogm */
+            if( p_stream->fmt.i_codec == VLC_FOURCC( 'm', 'p', 'g', 'a' ) ||
+                p_stream->fmt.i_codec == VLC_FOURCC( 'a', '5', '2', ' ' ) )
+                p_stream->fmt.b_packetized = false;
+
             p_stream->p_es = es_out_Add( p_demux->out, &p_stream->fmt );
+        }
 
         // TODO: something to do here ?
         if( p_stream->fmt.i_codec == VLC_FOURCC('c','m','m','l') )
@@ -1298,7 +1389,7 @@ static int Ogg_BeginningOfStream( demux_t *p_demux )
 
         p_stream->i_pcr = p_stream->i_previous_pcr =
             p_stream->i_interpolated_pcr = -1;
-        p_stream->b_reinit = 0;
+        p_stream->b_reinit = false;
     }
 
     if( p_ogg->p_old_stream )
@@ -1327,6 +1418,11 @@ static void Ogg_EndOfStream( demux_t *p_demux )
     p_ogg->i_bitrate = 0;
     p_ogg->i_streams = 0;
     p_ogg->pp_stream = NULL;
+
+    /* */
+    if( p_ogg->p_meta )
+        vlc_meta_Delete( p_ogg->p_meta );
+    p_ogg->p_meta = NULL;
 }
 
 /**
@@ -1394,6 +1490,66 @@ static bool Ogg_LogicalStreamResetEsFormat( demux_t *p_demux, logical_stream_t *
 
     return !b_compatible;
 }
+static void Ogg_ExtractXiphMeta( demux_t *p_demux, const uint8_t *p_headers, int i_headers, int i_skip )
+{
+    demux_sys_t *p_ogg = p_demux->p_sys;
+
+    if( i_headers <= 2 )
+        return;
+
+    /* Skip first packet */
+    const int i_tmp = GetWBE( &p_headers[0] );
+    if( i_tmp > i_headers-2 )
+        return;
+    p_headers += 2 + i_tmp;
+    i_headers -= 2 + i_tmp;
+
+    if( i_headers <= 2 )
+        return;
+
+    /* */
+    int i_comment = GetWBE( &p_headers[0] );
+    const uint8_t *p_comment = &p_headers[2];
+    if( i_comment > i_headers - 2 )
+        return;
+
+    if( i_comment <= i_skip )
+        return;
+
+    /* TODO how to handle multiple comments properly ? */
+    vorbis_ParseComment( &p_ogg->p_meta, &p_comment[i_skip], i_comment - i_skip );
+}
+static void Ogg_ExtractMeta( demux_t *p_demux, vlc_fourcc_t i_codec, const uint8_t *p_headers, int i_headers )
+{
+    demux_sys_t *p_ogg = p_demux->p_sys;
+
+    switch( i_codec )
+    {
+    /* 3 headers with the 2° one being the comments */
+    case VLC_FOURCC( 'v','o','r','b' ):
+        Ogg_ExtractXiphMeta( p_demux, p_headers, i_headers, 1+6 );
+        break;
+    case VLC_FOURCC( 't','h','e','o' ):
+        Ogg_ExtractXiphMeta( p_demux, p_headers, i_headers, 1+6 );
+        break;
+    case VLC_FOURCC( 's','p','x',' ' ):
+        Ogg_ExtractXiphMeta( p_demux, p_headers, i_headers, 0 );
+        break;
+
+    /* TODO */
+    case VLC_FOURCC( 'k','a','t','e' ):
+    case VLC_FOURCC( 'f','l','a','c' ):
+    case VLC_FOURCC( 'c','m','m','l' ):
+        msg_Warn( p_demux, "Ogg_ExtractMeta does not support %4.4s", (const char*)&i_codec );
+        break;
+    /* No meta data */
+    case VLC_FOURCC( 'd','r','a','c' ):
+    default:
+        break;
+    }
+    if( p_ogg->p_meta )
+        p_demux->info.i_update |= INPUT_UPDATE_META;
+}
 
 static void Ogg_ReadTheoraHeader( logical_stream_t *p_stream,
                                   ogg_packet *p_oggpacket )
@@ -1518,7 +1674,10 @@ static void Ogg_ReadFlacHeader( demux_t *p_demux, logical_stream_t *p_stream,
             msg_Dbg( p_demux, "FLAC header, channels: %i, rate: %i",
                      p_stream->fmt.audio.i_channels, (int)p_stream->f_rate );
         }
-        else msg_Dbg( p_demux, "FLAC STREAMINFO metadata too short" );
+        else
+        {
+            msg_Dbg( p_demux, "FLAC STREAMINFO metadata too short" );
+        }
 
         /* Fake this as the last metadata block */
         *((uint8_t*)p_oggpacket->packet) |= 0x80;
@@ -1558,27 +1717,27 @@ static void Ogg_ReadKateHeader( logical_stream_t *p_stream,
     p_stream->f_rate = (double)gnum/gden;
 
     p_stream->fmt.psz_language = malloc(16);
-    if (p_stream->fmt.psz_language)
+    if( p_stream->fmt.psz_language )
     {
-        for (n=0;n<16;++n)
+        for( n = 0; n < 16; ++n )
             p_stream->fmt.psz_language[n] = oggpack_read(&opb,8);
         p_stream->fmt.psz_language[15] = 0; /* just in case */
     }
     else
     {
-        for (n=0;n<16;++n)
+        for( n = 0; n < 16; ++n )
             oggpack_read(&opb,8);
     }
     p_stream->fmt.psz_description = malloc(16);
-    if (p_stream->fmt.psz_description)
+    if( p_stream->fmt.psz_description )
     {
-        for (n=0;n<16;++n)
+        for( n = 0; n < 16; ++n )
             p_stream->fmt.psz_description[n] = oggpack_read(&opb,8);
         p_stream->fmt.psz_description[15] = 0; /* just in case */
     }
     else
     {
-        for (n=0;n<16;++n)
+        for( n = 0; n < 16; ++n )
             oggpack_read(&opb,8);
     }
 }
@@ -1617,7 +1776,7 @@ static void Ogg_ReadAnnodexHeader( vlc_object_t *p_this,
 
         granule_rate_numerator = GetQWLE( &p_oggpacket->packet[8] );
         granule_rate_denominator = GetQWLE( &p_oggpacket->packet[16] );
-        p_stream->secondary_header_packets =
+        p_stream->i_secondary_header_packets =
             GetDWLE( &p_oggpacket->packet[24] );
 
         /* we are guaranteed that the first header field will be
@@ -1634,7 +1793,7 @@ static void Ogg_ReadAnnodexHeader( vlc_object_t *p_this,
 
         msg_Dbg( p_this, "AnxData packet info: %"PRId64" / %"PRId64", %d, ``%s''",
                  granule_rate_numerator, granule_rate_denominator,
-                 p_stream->secondary_header_packets, content_type_string );
+                 p_stream->i_secondary_header_packets, content_type_string );
 
         p_stream->f_rate = (float) granule_rate_numerator /
             (float) granule_rate_denominator;
@@ -1692,29 +1851,43 @@ static void Ogg_ReadAnnodexHeader( vlc_object_t *p_this,
 
 static uint32_t dirac_uint( bs_t *p_bs )
 {
-  uint32_t count = 0, value = 0;
-  while( !bs_read ( p_bs, 1 ) ) {
-    count++;
-    value <<= 1;
-    value |= bs_read ( p_bs, 1 );
-  }
-
-  return (1<<count) - 1 + value;
+    uint32_t u_count = 0, u_value = 0;
+
+    while( !bs_eof( p_bs ) && !bs_read( p_bs, 1 ) )
+    {
+        u_count++;
+        u_value <<= 1;
+        u_value |= bs_read( p_bs, 1 );
+    }
+
+    return (1<<u_count) - 1 + u_value;
 }
 
 static int dirac_bool( bs_t *p_bs )
 {
-    return bs_read ( p_bs, 1 );
+    return bs_read( p_bs, 1 );
 }
 
-static void Ogg_ReadDiracHeader( logical_stream_t *p_stream,
+static bool Ogg_ReadDiracHeader( logical_stream_t *p_stream,
                                  ogg_packet *p_oggpacket )
 {
+    static const struct {
+        uint32_t u_n /* numerator */, u_d /* denominator */;
+    } p_dirac_frate_tbl[] = { /* table 10.3 */
+        {1,1}, /* this first value is never used */
+        {24000,1001}, {24,1}, {25,1}, {30000,1001}, {30,1},
+        {50,1}, {60000,1001}, {60,1}, {15000,1001}, {25,2},
+    };
+    static const size_t u_dirac_frate_tbl = sizeof(p_dirac_frate_tbl)/sizeof(*p_dirac_frate_tbl);
+
+    static const uint32_t pu_dirac_vidfmt_frate[] = { /* table C.1 */
+        1, 9, 10, 9, 10, 9, 10, 4, 3, 7, 6, 4, 3, 7, 6, 2, 2, 7, 6, 7, 6,
+    };
+    static const size_t u_dirac_vidfmt_frate = sizeof(pu_dirac_vidfmt_frate)/sizeof(*pu_dirac_vidfmt_frate);
+
     bs_t bs;
 
-    p_stream->fmt.i_cat = VIDEO_ES;
-    p_stream->fmt.i_codec = VLC_FOURCC( 'd','r','a','c' );
-    p_stream->i_granule_shift = 32;
+    p_stream->i_granule_shift = 22; /* not 32 */
 
     /* Backing up stream headers is not required -- seqhdrs are repeated
      * thoughout the stream at suitable decoding start points */
@@ -1729,43 +1902,51 @@ static void Ogg_ReadDiracHeader( logical_stream_t *p_stream,
     dirac_uint( &bs ); /* level */
 
     uint32_t u_video_format = dirac_uint( &bs ); /* index */
+    if( u_video_format >= u_dirac_vidfmt_frate )
+    {
+        /* don't know how to parse this ogg dirac stream */
+        return false;
+    }
 
-    if (dirac_bool( &bs )) {
+    if( dirac_bool( &bs ) )
+    {
         dirac_uint( &bs ); /* frame_width */
         dirac_uint( &bs ); /* frame_height */
     }
 
-    if (dirac_bool( &bs )) {
+    if( dirac_bool( &bs ) )
+    {
         dirac_uint( &bs ); /* chroma_format */
     }
-    if (dirac_bool( &bs )) {
-        if (dirac_bool( &bs )) { /* interlaced */
-            dirac_bool( &bs ); /* top_field_first */
-        }
-    }
-
-    static const struct {
-        uint32_t u_n /* numerator */, u_d /* denominator */;
-    } dirac_frate_tbl[] = { /* table 10.3 */
-        {1,1}, /* this first value is never used */
-        {24000,1001}, {24,1}, {25,1}, {30000,1001}, {30,1},
-        {50,1}, {60000,1001}, {60,1}, {15000,1001}, {25,2},
-    };
 
-    static const uint32_t dirac_vidfmt_frate[] = { /* table C.1 */
-        1, 9, 10, 9, 10, 9, 10, 4, 3, 7, 6, 4, 3, 7, 6, 2, 2, 7, 6, 7, 6,
-    };
+    if( dirac_bool( &bs ) )
+    {
+        dirac_uint( &bs ); /* scan_format */
+    }
 
-    uint32_t u_n = dirac_frate_tbl[dirac_vidfmt_frate[u_video_format]].u_n;
-    uint32_t u_d = dirac_frate_tbl[dirac_vidfmt_frate[u_video_format]].u_d;
-    if (dirac_bool( &bs )) {
-        uint32_t frame_rate_index = dirac_uint( &bs );
-        u_n = dirac_frate_tbl[frame_rate_index].u_n;
-        u_d = dirac_frate_tbl[frame_rate_index].u_d;
-        if (frame_rate_index == 0) {
+    uint32_t u_n = p_dirac_frate_tbl[pu_dirac_vidfmt_frate[u_video_format]].u_n;
+    uint32_t u_d = p_dirac_frate_tbl[pu_dirac_vidfmt_frate[u_video_format]].u_d;
+    if( dirac_bool( &bs ) )
+    {
+        uint32_t u_frame_rate_index = dirac_uint( &bs );
+        if( u_frame_rate_index >= u_dirac_frate_tbl )
+        {
+            /* something is wrong with this stream */
+            return false;
+        }
+        u_n = p_dirac_frate_tbl[u_frame_rate_index].u_n;
+        u_d = p_dirac_frate_tbl[u_frame_rate_index].u_d;
+        if( u_frame_rate_index == 0 )
+        {
             u_n = dirac_uint( &bs ); /* frame_rate_numerator */
             u_d = dirac_uint( &bs ); /* frame_rate_denominator */
         }
     }
     p_stream->f_rate = (float) u_n / u_d;
+
+    /* probably is an ogg dirac es */
+    p_stream->fmt.i_cat = VIDEO_ES;
+    p_stream->fmt.i_codec = VLC_FOURCC( 'd','r','a','c' );
+
+    return true;
 }