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

h265bitwriter: Add check for data size to avoid overflow

Browse Source 
Part-of: <https://gitlab.freedesktop.org/gstreamer/gstreamer/-/merge_requests/7547>
This commit is contained in:
He Junyan 2024-09-07 10:26:22 +08:00 committed by GStreamer Marge Bot
parent 6bb294343b
commit b427f33818
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
subprojects/gst-plugins-bad/gst-libs/gst/codecparsers/gsth265bitwriter.c
View File

@ -2237,6 +2237,7 @@ gst_h265_bit_writer_convert_to_nal (guint nal_prefix_size,
GST_H265_BIT_WRITER_ERROR); GST_H265_BIT_WRITER_ERROR);
g_return_val_if_fail (raw_data != NULL, GST_H265_BIT_WRITER_ERROR); g_return_val_if_fail (raw_data != NULL, GST_H265_BIT_WRITER_ERROR);
g_return_val_if_fail (raw_size > 0, GST_H265_BIT_WRITER_ERROR); g_return_val_if_fail (raw_size > 0, GST_H265_BIT_WRITER_ERROR);
g_return_val_if_fail (raw_size / 8 <= G_MAXUINT, GST_H265_BIT_WRITER_ERROR);
g_return_val_if_fail (nal_data != NULL, GST_H265_BIT_WRITER_ERROR); g_return_val_if_fail (nal_data != NULL, GST_H265_BIT_WRITER_ERROR);
g_return_val_if_fail (nal_size != NULL, GST_H265_BIT_WRITER_ERROR); g_return_val_if_fail (nal_size != NULL, GST_H265_BIT_WRITER_ERROR);
g_return_val_if_fail (*nal_size > 0, GST_H265_BIT_WRITER_ERROR); g_return_val_if_fail (*nal_size > 0, GST_H265_BIT_WRITER_ERROR);
@ -2264,7 +2265,7 @@ gst_h265_bit_writer_convert_to_nal (guint nal_prefix_size,
nal_writer_init (&nw, nal_prefix_size, packetized); nal_writer_init (&nw, nal_prefix_size, packetized);
if (!nal_writer_put_bytes (&nw, raw_data, raw_size / 8)) if (!nal_writer_put_bytes (&nw, raw_data, (guint) (raw_size / 8)))
goto error; goto error;
if (raw_size % 8) { if (raw_size % 8) {