changeset 4039:c126ba3df59c libavformat

Change header reading implementation to using get_bits() instead of directly accessing the data array. This allows to easily read optionally available header bits without causing pains. See discussion in "[PATCH] RDT/Realmedia patches #2" thread on ML.
author rbultje
date Mon, 17 Nov 2008 14:30:31 +0000
parents 69cdec2e03b2
children b22f5462903e
files rdt.c
diffstat 1 files changed, 21 insertions(+), 7 deletions(-) [+]
line wrap: on
line diff
--- a/rdt.c	Mon Nov 17 14:28:12 2008 +0000
+++ b/rdt.c	Mon Nov 17 14:30:31 2008 +0000
@@ -33,6 +33,7 @@
 #include "libavutil/md5.h"
 #include "rm.h"
 #include "internal.h"
+#include <libavcodec/bitstream.h>
 
 struct RDTDemuxContext {
     AVFormatContext *ic;
@@ -175,7 +176,9 @@
                     int *pset_id, int *pseq_no, int *pstream_id,
                     int *pis_keyframe, uint32_t *ptimestamp)
 {
-    int consumed = 10;
+    GetBitContext gb;
+    int consumed = 0, set_id, seq_no, stream_id, is_keyframe;
+    uint32_t timestamp;
 
     /* skip status packets */
     while (len >= 5 && buf[1] == 0xFF /* status packet */) {
@@ -242,13 +245,24 @@
      * [2] http://www.wireshark.org/docs/dfref/r/rdt.html and
      *     http://anonsvn.wireshark.org/viewvc/trunk/epan/dissectors/packet-rdt.c
      */
-    if (pset_id)      *pset_id      = (buf[0]>>1) & 0x1f;
-    if (pseq_no)      *pseq_no      = AV_RB16(buf+1);
-    if (ptimestamp)   *ptimestamp   = AV_RB32(buf+4);
-    if (pstream_id)   *pstream_id   = (buf[3]>>1) & 0x1f;
-    if (pis_keyframe) *pis_keyframe = !(buf[3] & 0x1);
+    init_get_bits(&gb, buf, len << 3);
+    skip_bits(&gb, 2);
+    set_id        = get_bits(&gb, 5);
+    skip_bits(&gb, 1);
+    seq_no        = get_bits(&gb, 16);
+    skip_bits(&gb, 2);
+    stream_id     = get_bits(&gb, 5);
+    is_keyframe   = !get_bits1(&gb);
+    timestamp     = get_bits_long(&gb, 32);
+    skip_bits(&gb, 16);
 
-    return consumed;
+    if (pset_id)      *pset_id      = set_id;
+    if (pseq_no)      *pseq_no      = seq_no;
+    if (pstream_id)   *pstream_id   = stream_id;
+    if (pis_keyframe) *pis_keyframe = is_keyframe;
+    if (ptimestamp)   *ptimestamp   = timestamp;
+
+    return consumed + (get_bits_count(&gb) >> 3);
 }
 
 /**< return 0 on packet, no more left, 1 on packet, 1 on partial packet... */