[08/13] rtpdec_h264: Prepare h264_handle_packet_stap_a for sharing with hevc

Message ID 1424378051-6607-8-git-send-email-martin@martin.st
State Committed
Commit a3cc519d1f43bf93cc29fd53284a2b8aa6189a9a
Headers show

Commit Message

Martin Storsjö Feb. 19, 2015, 8:34 p.m.
Add a parameter for skipping a number of bytes at the start of each nal.
---
 libavformat/rtpdec_h264.c | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

Comments

Luca Barbato Feb. 21, 2015, 12:25 a.m. | #1
On 19/02/15 21:34, Martin Storsjö wrote:
> Add a parameter for skipping a number of bytes at the start of each nal.
> ---
>   libavformat/rtpdec_h264.c | 12 ++++++++----
>   1 file changed, 8 insertions(+), 4 deletions(-)
>
Probably Ok.

Patch

diff --git a/libavformat/rtpdec_h264.c b/libavformat/rtpdec_h264.c
index bc563f2..e6c8c26 100644
--- a/libavformat/rtpdec_h264.c
+++ b/libavformat/rtpdec_h264.c
@@ -177,7 +177,8 @@  static int sdp_parse_fmtp_config_h264(AVFormatContext *s,
 }
 
 static int h264_handle_packet_stap_a(AVFormatContext *ctx, AVPacket *pkt,
-                                     const uint8_t *buf, int len)
+                                     const uint8_t *buf, int len,
+                                     int start_skip)
 {
     int pass         = 0;
     int total_length = 0;
@@ -189,6 +190,9 @@  static int h264_handle_packet_stap_a(AVFormatContext *ctx, AVPacket *pkt,
         const uint8_t *src = buf;
         int src_len        = len;
 
+        src     += start_skip;
+        src_len -= start_skip;
+
         while (src_len > 2) {
             uint16_t nal_size = AV_RB16(src);
 
@@ -215,8 +219,8 @@  static int h264_handle_packet_stap_a(AVFormatContext *ctx, AVPacket *pkt,
             }
 
             // eat what we handled
-            src     += nal_size;
-            src_len -= nal_size;
+            src     += nal_size + start_skip;
+            src_len -= nal_size + start_skip;
         }
 
         if (pass == 0) {
@@ -304,7 +308,7 @@  static int h264_handle_packet(AVFormatContext *ctx, PayloadContext *data,
         // consume the STAP-A NAL
         buf++;
         len--;
-        result = h264_handle_packet_stap_a(ctx, pkt, buf, len);
+        result = h264_handle_packet_stap_a(ctx, pkt, buf, len, 0);
         break;
 
     case 25:                   // STAP-B