michael/u-boot
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

dm: crypto: Check malloc return value

Browse Source 
tmp_buffer is allocated using malloc but failure
is not handled.
This commit ensures that we do not use a NULL pointer
if malloc fails.

Signed-off-by: Francois Berder <fberder@outlook.fr>
This commit is contained in:
Francois Berder
2025-12-04 20:34:12 +01:00
committed by Tom Rini
parent 9ed99e2eea
commit 737386977b
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
drivers/crypto/aes/aes-uclass.c
View File

@@ -156,13 +156,17 @@ int dm_aes_cmac(struct udevice *dev, u8 *src, u8 *dst, u32 num_aes_blocks)
/* Process all blocks except last by calling engine several times per dma buffer size */
if (num_aes_blocks > 1) {
tmp_buffer = malloc(AES_BLOCK_LENGTH * min(num_aes_blocks - 1, TMP_BUFFER_LEN));
if (!tmp_buffer)
return -1;
while (num_aes_blocks > 1) {
u32 blocks = min(num_aes_blocks - 1, TMP_BUFFER_LEN);
/* Encrypt the current remaining set of blocks that fits in tmp buffer */
ret = dm_aes_cbc_encrypt(dev, tmp_block, src, tmp_buffer, blocks);
if (ret)
if (ret) {
free(tmp_buffer);
return -1;
}
num_aes_blocks -= blocks;
src += blocks * AES_BLOCK_LENGTH;