From mboxrd@z Thu Jan 1 00:00:00 1970 Received: from mail-lf1-f46.google.com (mail-lf1-f46.google.com [209.85.167.46]) by mx.groups.io with SMTP id smtpd.web11.5560.1670602276434237032 for ; Fri, 09 Dec 2022 08:11:16 -0800 Authentication-Results: mx.groups.io; dkim=pass header.i=@gmail.com header.s=20210112 header.b=LvLYyk0n; spf=pass (domain: gmail.com, ip: 209.85.167.46, mailfrom: savvamtr@gmail.com) Received: by mail-lf1-f46.google.com with SMTP id cf42so7801066lfb.1 for ; Fri, 09 Dec 2022 08:11:16 -0800 (PST) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=gmail.com; s=20210112; h=content-transfer-encoding:mime-version:references:in-reply-to :message-id:date:subject:cc:to:from:from:to:cc:subject:date :message-id:reply-to; bh=5lWEkmTDpvrpGMsy3oJ7LfqfIl/ROHoLeBudPT6tSHw=; b=LvLYyk0nNPIXkCdS0cbJg90bfaUlPZOtlYsHdYQ/hltfhOHJ1eDKgMTrZ6wwjdmRcd ws8StHLFiliCggIg0Of2ONAdoFoe+ANqqzZUacR1qZhCzOJKJWw0b2tka94mAoG5kK1f BFRhHLZBAmxfWUFCwaQ0m02F880a/mYcdp3rsIuiHKw/Q2U6EwQCYbhl6wuVJvF1chG5 7I6+2VOHU+6L9K11sVhgj+ykGt5EOYuNu/NW8uwc81oU5nMwH7DZVdsrKcGLD/MbLlEm A+VkYt32AiVJAIu9V/Wt3syg2qhnGAFthXXEccQGuLpWPawxwVHTD1Smm9y6urAkha4s 0JXw== X-Google-DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=1e100.net; s=20210112; h=content-transfer-encoding:mime-version:references:in-reply-to :message-id:date:subject:cc:to:from:x-gm-message-state:from:to:cc :subject:date:message-id:reply-to; bh=5lWEkmTDpvrpGMsy3oJ7LfqfIl/ROHoLeBudPT6tSHw=; b=nlwO3wy3sKGLdh9MvnAUrdIsLXwTl0WhpImzMh221EUiS7/CjKu6DopUt44ohYDWPD BwYyCyzCPDJEuxICFKioMe6ROSRuGNoXiCcL8Bg4hC2Ugj4iqHCPotSOEt+u91QBH7KL 8vvGyLXa50qizL0Iq9tJKFlc2c3GJQYrXT4Yo6i4IGXczSUtL0djj8sI5eSK1UIIVYRz UREVCA8L4t+JzeDkr27BpcLWh8LGqXY3L0JvqAIvdIVshOgaLqKePI7zG22q2wroCiEE ZdT98ZYfl36MhOcw/OFZiwUNE33KYDJM7SzjSdG8EHvVYZwW6xnLA78aHDfrSm0lYfac 8PEA== X-Gm-Message-State: ANoB5plKAvZ+/09S5ZCsBFhd8HHKC1S5DMRqsY8qVKzp1onq6DP//r5W h0+zWhFghoqkYStKdOBXKogSquK3EDtqrr4Jgqk= X-Google-Smtp-Source: AA0mqf4/aBZkUcKeyoHiwajYn6js8HwW8wpXG+n5m8iE86LOZkb8gwigB9j5EgLHxenlAiimS07mbw== X-Received: by 2002:ac2:5e63:0:b0:4b5:8054:1ded with SMTP id a3-20020ac25e63000000b004b580541dedmr1604509lfr.9.1670602274492; Fri, 09 Dec 2022 08:11:14 -0800 (PST) Return-Path: Received: from localhost.localdomain ([109.194.121.139]) by smtp.gmail.com with ESMTPSA id v5-20020a05651203a500b00492c663bba2sm318430lfp.124.2022.12.09.08.11.13 (version=TLS1_3 cipher=TLS_AES_256_GCM_SHA384 bits=256/256); Fri, 09 Dec 2022 08:11:14 -0800 (PST) From: "Savva Mitrofanov" To: devel@edk2.groups.io Cc: =?UTF-8?q?Marvin=20H=C3=A4user?= , Pedro Falcato , Vitaly Cheptsov Subject: [edk2-platforms][PATCH v1 01/12] Ext4Pkg: Fix memory leak in Ext4RetrieveDirent Date: Fri, 9 Dec 2022 22:10:53 +0600 Message-Id: <20221209161104.70220-2-savvamtr@gmail.com> X-Mailer: git-send-email 2.38.1 In-Reply-To: <20221209161104.70220-1-savvamtr@gmail.com> References: <20221209161104.70220-1-savvamtr@gmail.com> MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: quoted-printable We need to free buffer on return if BlockRemainder !=3D 0. Also changed return logic from function to use use common exit to prevent code duplication. Cc: Marvin H=C3=A4user Cc: Pedro Falcato Cc: Vitaly Cheptsov Signed-off-by: Savva Mitrofanov --- Features/Ext4Pkg/Ext4Dxe/Directory.c | 30 +++++++++++--------- 1 file changed, 16 insertions(+), 14 deletions(-) diff --git a/Features/Ext4Pkg/Ext4Dxe/Directory.c b/Features/Ext4Pkg/Ext4Dx= e/Directory.c index 4441e6d192b6..8b8fce568e43 100644 --- a/Features/Ext4Pkg/Ext4Dxe/Directory.c +++ b/Features/Ext4Pkg/Ext4Dxe/Directory.c @@ -112,8 +112,7 @@ Ext4RetrieveDirent ( UINTN ToCopy;=0D UINTN BlockOffset;=0D =0D - Status =3D EFI_NOT_FOUND;=0D - Buf =3D AllocatePool (Partition->BlockSize);=0D + Buf =3D AllocatePool (Partition->BlockSize);=0D =0D if (Buf =3D=3D NULL) {=0D return EFI_OUT_OF_RESOURCES;=0D @@ -127,7 +126,8 @@ Ext4RetrieveDirent ( DivU64x32Remainder (DirInoSize, Partition->BlockSize, &BlockRemainder);= =0D if (BlockRemainder !=3D 0) {=0D // Directory inodes need to have block aligned sizes=0D - return EFI_VOLUME_CORRUPTED;=0D + Status =3D EFI_VOLUME_CORRUPTED;=0D + goto Out;=0D }=0D =0D while (Off < DirInoSize) {=0D @@ -136,8 +136,7 @@ Ext4RetrieveDirent ( Status =3D Ext4Read (Partition, Directory, Buf, Off, &Length);=0D =0D if (Status !=3D EFI_SUCCESS) {=0D - FreePool (Buf);=0D - return Status;=0D + goto Out;=0D }=0D =0D for (BlockOffset =3D 0; BlockOffset < Partition->BlockSize; ) {=0D @@ -145,19 +144,19 @@ Ext4RetrieveDirent ( RemainingBlock =3D Partition->BlockSize - BlockOffset;=0D // Check if the minimum directory entry fits inside [BlockOffset, En= dOfBlock]=0D if (RemainingBlock < EXT4_MIN_DIR_ENTRY_LEN) {=0D - FreePool (Buf);=0D - return EFI_VOLUME_CORRUPTED;=0D + Status =3D EFI_VOLUME_CORRUPTED;=0D + goto Out;=0D }=0D =0D if (!Ext4ValidDirent (Entry)) {=0D - FreePool (Buf);=0D - return EFI_VOLUME_CORRUPTED;=0D + Status =3D EFI_VOLUME_CORRUPTED;=0D + goto Out;=0D }=0D =0D if ((Entry->name_len > RemainingBlock) || (Entry->rec_len > Remainin= gBlock)) {=0D // Corrupted filesystem=0D - FreePool (Buf);=0D - return EFI_VOLUME_CORRUPTED;=0D + Status =3D EFI_VOLUME_CORRUPTED;=0D + goto Out;=0D }=0D =0D // Unused entry=0D @@ -186,8 +185,8 @@ Ext4RetrieveDirent ( ToCopy =3D MIN (Entry->rec_len, sizeof (EXT4_DIR_ENTRY));=0D =0D CopyMem (Result, Entry, ToCopy);=0D - FreePool (Buf);=0D - return EFI_SUCCESS;=0D + Status =3D EFI_SUCCESS;=0D + goto Out;=0D }=0D =0D BlockOffset +=3D Entry->rec_len;=0D @@ -196,8 +195,11 @@ Ext4RetrieveDirent ( Off +=3D Partition->BlockSize;=0D }=0D =0D + Status =3D EFI_NOT_FOUND;=0D +=0D +Out:=0D FreePool (Buf);=0D - return EFI_NOT_FOUND;=0D + return Status;=0D }=0D =0D /**=0D --=20 2.38.1