FaSTTUBe/GST-Tensordecoder-ov_ep
37
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

rtpbuffer: get_onebyte_header_end_offset: allow 0 offset

Browse Source 
There are some cases where the full extension data could be padding.
In order to make the GstRtpBuffer robust enough, this change supports
this case.
This commit is contained in:
Miguel Paris 2019-12-13 16:06:44 +01:00
parent 3f79805aa0
commit 2d4d28d662
1 changed files with 19 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
gst-libs/gst/rtp/gstrtpbuffer.c
View File

@ -1510,19 +1510,20 @@ gst_rtp_buffer_get_extension_twobytes_header (GstRTPBuffer * rtp,
return FALSE; return FALSE;
} }
static guint static gboolean
get_onebyte_header_end_offset (guint8 * pdata, guint wordlen) get_onebyte_header_end_offset (guint8 * pdata, guint wordlen, guint *offset)
{ {
guint offset = 0;
guint bytelen = wordlen * 4; guint bytelen = wordlen * 4;
guint paddingcount = 0; guint paddingcount = 0;
while (offset + 1 < bytelen) { *offset = 0;
while (*offset + 1 < bytelen) {
guint8 read_id, read_len; guint8 read_id, read_len;
read_id = GST_READ_UINT8 (pdata + offset) >> 4; read_id = GST_READ_UINT8 (pdata + *offset) >> 4;
read_len = (GST_READ_UINT8 (pdata + offset) & 0x0F) + 1; read_len = (GST_READ_UINT8 (pdata + *offset) & 0x0F) + 1;
offset += 1; *offset += 1;
/* ID 0 means its padding, skip */ /* ID 0 means its padding, skip */
if (read_id == 0) { if (read_id == 0) {
@ -1534,17 +1535,21 @@ get_onebyte_header_end_offset (guint8 * pdata, guint wordlen)
/* ID 15 is special and means we should stop parsing */ /* ID 15 is special and means we should stop parsing */
/* It also means we can't add an extra packet */ /* It also means we can't add an extra packet */
if (read_id == 15) if (read_id == 15) {
return 0; return FALSE;
/* Ignore extension headers where the size does not fit */
if (offset + read_len > bytelen)
return 0;
offset += read_len;
} }
return offset - paddingcount; /* Ignore extension headers where the size does not fit */
if (*offset + read_len > bytelen) {
return FALSE;
}
*offset += read_len;
}
*offset -= paddingcount;
return TRUE;
} }
/** /**
@ -1586,8 +1591,7 @@ gst_rtp_buffer_add_extension_onebyte_header (GstRTPBuffer * rtp, guint8 id,
if (bits != 0xBEDE) if (bits != 0xBEDE)
return FALSE; return FALSE;
offset = get_onebyte_header_end_offset (pdata, wordlen); if (!get_onebyte_header_end_offset (pdata, wordlen, &offset))
if (offset == 0)
return FALSE; return FALSE;
} }