From 0c868aa94a218b99e13f31bf0e83a43de01e0da5 Mon Sep 17 00:00:00 2001
From: Pierre-Eric Pelloux-Prayer <pierre-eric.pelloux-prayer@amd.com>
Date: Fri, 19 Jul 2024 08:42:01 +0200
Subject: [PATCH] amd: use a valid size for ac_pm4_state allocation
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

If max_dw is smaller than the pm4 array the allocation size would be
smaller than sizeof(ac_pm4_state).

Fixes: 428601095c3 ("ac,radeonsi import PM4 state from RadeonSI")
Reviewed-by: Samuel Pitoiset <samuel.pitoiset@gmail.com>
Reviewed-by: Marek Olšák <marek.olsak@amd.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/30257>
---
 src/amd/common/ac_pm4.c | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/amd/common/ac_pm4.c b/src/amd/common/ac_pm4.c
index bbac8cccfac..df9a6d9448c 100644
--- a/src/amd/common/ac_pm4.c
+++ b/src/amd/common/ac_pm4.c
@@ -409,7 +409,11 @@ ac_pm4_create_sized(const struct radeon_info *info, bool debug_sqtt,
                     unsigned max_dw, bool is_compute_queue)
 {
    struct ac_pm4_state *pm4;
-   unsigned size = sizeof(*pm4) + 4 * (max_dw - ARRAY_SIZE(pm4->pm4));
+   unsigned size;
+
+   max_dw = MAX2(max_dw, ARRAY_SIZE(pm4->pm4));
+
+   size = sizeof(*pm4) + 4 * (max_dw - ARRAY_SIZE(pm4->pm4));
 
    pm4 = (struct ac_pm4_state *)calloc(1, size);
    if (pm4) {