From e9d9e5c1ee3d5374863931094a54e052b8e3ac68 Mon Sep 17 00:00:00 2001
From: ganjing <ganjing@uniontech.com>
Date: Fri, 4 Sep 2020 16:45:33 +0800
Subject: [PATCH] freedreno/msm/: check memory allocation in
 handle_stateobj_relocs

Signed-off-by: ganjing <ganjing@uniontech.com>
---
 freedreno/msm/msm_ringbuffer.c | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/freedreno/msm/msm_ringbuffer.c b/freedreno/msm/msm_ringbuffer.c
index 7b9df4a0..8613a770 100644
--- a/freedreno/msm/msm_ringbuffer.c
+++ b/freedreno/msm/msm_ringbuffer.c
@@ -420,6 +420,11 @@ handle_stateobj_relocs(struct fd_ringbuffer *parent, struct fd_ringbuffer *state
 {
 	struct msm_ringbuffer *msm_ring = to_msm_ringbuffer(stateobj);
 	struct drm_msm_gem_submit_reloc *relocs = malloc(nr_relocs * sizeof(*relocs));
+	if (!relocs) {
+		ERROR_MSG("malloc relocs failed");
+		return NULL;
+	}
+
 	unsigned i;
 
 	for (i = 0; i < nr_relocs; i++) {