From mboxrd@z Thu Jan 1 00:00:00 1970 Return-Path: X-Spam-Checker-Version: SpamAssassin 3.4.0 (2014-02-07) on aws-us-west-2-korg-lkml-1.web.codeaurora.org Received: from gabe.freedesktop.org (gabe.freedesktop.org [131.252.210.177]) (using TLSv1.2 with cipher ECDHE-RSA-AES256-GCM-SHA384 (256/256 bits)) (No client certificate requested) by smtp.lore.kernel.org (Postfix) with ESMTPS id BE79DCCA471 for ; Fri, 3 Oct 2025 07:03:47 +0000 (UTC) Received: from gabe.freedesktop.org (localhost [127.0.0.1]) by gabe.freedesktop.org (Postfix) with ESMTP id 26E7210E8D4; Fri, 3 Oct 2025 07:03:36 +0000 (UTC) Authentication-Results: gabe.freedesktop.org; dkim=pass (2048-bit key; unprotected) header.d=gmail.com header.i=@gmail.com header.b="co5RWL4Y"; dkim-atps=neutral Received: from mail-pf1-f176.google.com (mail-pf1-f176.google.com [209.85.210.176]) by gabe.freedesktop.org (Postfix) with ESMTPS id 605AD10E155 for ; Thu, 2 Oct 2025 02:23:03 +0000 (UTC) Received: by mail-pf1-f176.google.com with SMTP id d2e1a72fcca58-789fb76b466so597147b3a.0 for ; Wed, 01 Oct 2025 19:23:03 -0700 (PDT) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=gmail.com; s=20230601; t=1759371783; x=1759976583; darn=lists.freedesktop.org; h=content-transfer-encoding:mime-version:message-id:date:subject:cc :to:from:from:to:cc:subject:date:message-id:reply-to; bh=mF3LfHpef/3mN9CeEleeS0pHSg8cp/5+RdXD9b+jUXQ=; b=co5RWL4Y/jvvEyf5oE01LaASHguRWrX/rxO9RuS8A2v7hGRs2hr6aD0Dps2K8u/09f zf0/BfS2xFp0PU63nj2Wo2bAF+e8xXb7+dCB40iwCN1AiMOJs6uBuh8gqY5nF/YBDRUU U66Y9cJtpzMKeKVqLN8hee5k70VklIRnF/qSW9TT1JVlt88ifMZU8lSMu9xNwkfnLqJo quI8bXEMC2LQx1ujFb5b0IX7L9QMIy5qNTSF3oA70kGb8Rg0Saq5NYxlhephdKCZHIrN Fzg+dgN3QmmNQC05Y0uV3eAUNpvIpoef+NSQgNgQby1lJ1HArKiWmV+WDYVmnI4QZi0V hnTw== X-Google-DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=1e100.net; s=20230601; t=1759371783; x=1759976583; h=content-transfer-encoding:mime-version:message-id:date:subject:cc :to:from:x-gm-message-state:from:to:cc:subject:date:message-id :reply-to; bh=mF3LfHpef/3mN9CeEleeS0pHSg8cp/5+RdXD9b+jUXQ=; b=HH1gFkRH8Unung6CUMS5v/64tDdRIX7f6n1FU2BEZrFstxVbSs1vnXsW6U7npRNgMs Xmu7GHJK/W8lIQvk06gpLm+zEOZOoACZI60ADACYnFCQdyA5U0Vq+IPAgmKA3aBUxWHO d92b9PrgmzOZf4P0+5NgZY7GmHSeJrfHDgtWaqPLYLPS/P/JiYatF8tD4a03f2suy/Xc C0s9bo4K2rhM7on7chIwQX3FZ1YM652JG3hxYpv24MZwkmdgJ1WDzb58+Gmtc7Qk3u0V CLk+OtAUjpxmR9/2FjV6IBUqjjOv97BN6oSWoCvASIDDYKN6H0nwIGXkXYtU+KTPQdh5 7JBQ== X-Gm-Message-State: AOJu0YwF6oH6ud5lAyqCgNfwJDCuhkTheRgtwPpApyxyr8flrPMfzzBT tfniZgAVSqTsTnWhMUBCr3sXXgJC3Xk2faIoOWC8tIwHQXwfyv3Fl2Wp X-Gm-Gg: ASbGncs0I7uyBenptvP63F8JImsen0rPfSw7VoGsJBnJfY1T+REMRxWxuINZBNwaLAf Gzid5VTrdZoI/7eiZhD9Nyj/kZ1T3xQWl19jX5Gkz+AECqtRvw2S38QKBPT+hpceFQbCzvujPx2 HgPh4H3LBy+jaf9L8GIaB3mX1Yh6MO1cP3J2xe0dUOJnw7CfsMgN8SLYVXOOxCxZKbqx7pAuM1B qz/Gz8+13dG0tP0s26M4AO1F7tPxuWqTYxo00O/ByOsIy4nr5wiyqqltlxo0WwQrp0I5ibJo29E 33033sZ2D2DNAFZ29kr9QnPz08zA/lFfesWqxGDYKArJ674PqBtwiKoH5mGyFargBANd0kGIwga eoMqZaRj//kLKFkbXnz0lzJXYCTZcdHLU0zLLjw7bh6UMVC+BdzWKeqN9At+yaGdYdaRzCYolGJ ImF13DoO/jHw== X-Google-Smtp-Source: AGHT+IHxLShFCm76xin7rHm8qNL6ADQW6sxwH14YEVEejbEnDXp8CfZFyeDbyzek8/KKOZq8n0qs+g== X-Received: by 2002:a05:6a00:17a6:b0:781:556:f33 with SMTP id d2e1a72fcca58-78af3fe8bcbmr6233829b3a.5.1759371782828; Wed, 01 Oct 2025 19:23:02 -0700 (PDT) Received: from ti-am64x-sdk.. ([157.50.91.136]) by smtp.gmail.com with ESMTPSA id d2e1a72fcca58-78b01f99acesm1046229b3a.5.2025.10.01.19.22.54 (version=TLS1_3 cipher=TLS_AES_256_GCM_SHA384 bits=256/256); Wed, 01 Oct 2025 19:23:02 -0700 (PDT) From: Bhanu Seshu Kumar Valluri To: Alex Deucher , =?UTF-8?q?Christian=20K=C3=B6nig?= , David Airlie , Simona Vetter , Harry Wentland , Leo Li , Rodrigo Siqueira , Tao Zhou , Hawking Zhang , ganglxie , Lijo Lazar , Candice Li , Victor Skvortsov , Roman Li , Alvin Lee , Karthi Kandasamy , David Rosca , =?UTF-8?q?Marek=20Ol=C5=A1=C3=A1k?= , Jocelyn Falempe , =?UTF-8?q?Andr=C3=A9=20Almeida?= , Mario Limonciello Cc: amd-gfx@lists.freedesktop.org, dri-devel@lists.freedesktop.org, linux-kernel@vger.kernel.org, khalid@kernel.org, linux-kernel-mentees@lists.linuxfoundation.org, skhan@linuxfoundation.org, david.hunter.linux@gmail.com, bhanuseshukumar@gmail.com Subject: [PATCH] drm: amd: Use kmalloc_array to prevent overflow of dynamic size calculation Date: Thu, 2 Oct 2025 07:52:41 +0530 Message-Id: <20251002022241.77823-1-bhanuseshukumar@gmail.com> X-Mailer: git-send-email 2.34.1 MIME-Version: 1.0 Content-Transfer-Encoding: 8bit X-Mailman-Approved-At: Fri, 03 Oct 2025 07:03:33 +0000 X-BeenThere: amd-gfx@lists.freedesktop.org X-Mailman-Version: 2.1.29 Precedence: list List-Id: Discussion list for AMD gfx List-Unsubscribe: , List-Archive: List-Post: List-Help: List-Subscribe: , Errors-To: amd-gfx-bounces@lists.freedesktop.org Sender: "amd-gfx" Use kmalloc_array to avoid potential overflow during dynamic size calculation inside kmalloc. Signed-off-by: Bhanu Seshu Kumar Valluri --- Note: Patch is verified for compilation. drivers/gpu/drm/amd/amdgpu/amdgpu_ras.c | 4 ++-- drivers/gpu/drm/amd/display/amdgpu_dm/amdgpu_dm_plane.c | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/drivers/gpu/drm/amd/amdgpu/amdgpu_ras.c b/drivers/gpu/drm/amd/amdgpu/amdgpu_ras.c index 540817e296da..642addf70466 100644 --- a/drivers/gpu/drm/amd/amdgpu/amdgpu_ras.c +++ b/drivers/gpu/drm/amd/amdgpu/amdgpu_ras.c @@ -2566,7 +2566,7 @@ static int amdgpu_ras_badpages_read(struct amdgpu_device *adev, goto out; } - *bps = kmalloc(sizeof(struct ras_badpage) * data->count, GFP_KERNEL); + *bps = kmalloc_array(sizeof(struct ras_badpage), data->count, GFP_KERNEL); if (!*bps) { ret = -ENOMEM; goto out; @@ -2722,7 +2722,7 @@ static int amdgpu_ras_realloc_eh_data_space(struct amdgpu_device *adev, unsigned int old_space = data->count + data->space_left; unsigned int new_space = old_space + pages; unsigned int align_space = ALIGN(new_space, 512); - void *bps = kmalloc(align_space * sizeof(*data->bps), GFP_KERNEL); + void *bps = kmalloc_array(align_space, sizeof(*data->bps), GFP_KERNEL); if (!bps) { return -ENOMEM; diff --git a/drivers/gpu/drm/amd/display/amdgpu_dm/amdgpu_dm_plane.c b/drivers/gpu/drm/amd/display/amdgpu_dm/amdgpu_dm_plane.c index 3d2f8eedeef2..e027798ece03 100644 --- a/drivers/gpu/drm/amd/display/amdgpu_dm/amdgpu_dm_plane.c +++ b/drivers/gpu/drm/amd/display/amdgpu_dm/amdgpu_dm_plane.c @@ -146,7 +146,7 @@ static void amdgpu_dm_plane_add_modifier(uint64_t **mods, uint64_t *size, uint64 if (*cap - *size < 1) { uint64_t new_cap = *cap * 2; - uint64_t *new_mods = kmalloc(new_cap * sizeof(uint64_t), GFP_KERNEL); + uint64_t *new_mods = kmalloc_array(new_cap, sizeof(uint64_t), GFP_KERNEL); if (!new_mods) { kfree(*mods); @@ -732,7 +732,7 @@ static int amdgpu_dm_plane_get_plane_modifiers(struct amdgpu_device *adev, unsig if (adev->family < AMDGPU_FAMILY_AI) return 0; - *mods = kmalloc(capacity * sizeof(uint64_t), GFP_KERNEL); + *mods = kmalloc_array(capacity, sizeof(uint64_t), GFP_KERNEL); if (plane_type == DRM_PLANE_TYPE_CURSOR) { amdgpu_dm_plane_add_modifier(mods, &size, &capacity, DRM_FORMAT_MOD_LINEAR); -- 2.34.1