[PATCH] v4l2-core: v4l2-h264: check after increasing memory allocation

From: Li zeming
Date: Sun Dec 18 2022 - 22:37:26 EST


The pointer *out_str adds judgment and should help with program robustness.

Signed-off-by: Li zeming <zeming@xxxxxxxxxxxx>
---
drivers/media/v4l2-core/v4l2-h264.c | 2 ++
1 file changed, 2 insertions(+)

diff --git a/drivers/media/v4l2-core/v4l2-h264.c b/drivers/media/v4l2-core/v4l2-h264.c
index 72bd64f65198..934aba539332 100644
--- a/drivers/media/v4l2-core/v4l2-h264.c
+++ b/drivers/media/v4l2-core/v4l2-h264.c
@@ -305,6 +305,8 @@ static const char *format_ref_list_p(const struct v4l2_h264_reflist_builder *bui
int n = 0, i;

*out_str = kmalloc(tmp_str_size, GFP_KERNEL);
+ if (!*out_str)
+ return NULL;

n += snprintf(*out_str + n, tmp_str_size - n, "|");

--
2.18.2