From nobody Thu May 2 09:05:17 2024 Delivered-To: importer@patchew.org Received-SPF: none (zoho.com: 198.145.21.10 is neither permitted nor denied by domain of lists.01.org) client-ip=198.145.21.10; envelope-from=edk2-devel-bounces@lists.01.org; helo=ml01.01.org; Authentication-Results: mx.zoho.com; spf=none (zoho.com: 198.145.21.10 is neither permitted nor denied by domain of lists.01.org) smtp.mailfrom=edk2-devel-bounces@lists.01.org; Return-Path: Received: from ml01.01.org (ml01.01.org [198.145.21.10]) by mx.zohomail.com with SMTPS id 1492409469812896.5670065180241; Sun, 16 Apr 2017 23:11:09 -0700 (PDT) Received: from [127.0.0.1] (localhost [IPv6:::1]) by ml01.01.org (Postfix) with ESMTP id 5E38420D7A3AC; Sun, 16 Apr 2017 23:11:07 -0700 (PDT) Received: from mga11.intel.com (mga11.intel.com [192.55.52.93]) (using TLSv1.2 with cipher ECDHE-RSA-AES256-GCM-SHA384 (256/256 bits)) (No client certificate requested) by ml01.01.org (Postfix) with ESMTPS id C023020D7A3AA for ; Sun, 16 Apr 2017 23:11:05 -0700 (PDT) Received: from fmsmga006.fm.intel.com ([10.253.24.20]) by fmsmga102.fm.intel.com with ESMTP/TLS/DHE-RSA-AES256-GCM-SHA384; 16 Apr 2017 23:11:05 -0700 Received: from jiaxinwu-mobl2.ccr.corp.intel.com ([10.239.196.63]) by fmsmga006.fm.intel.com with ESMTP; 16 Apr 2017 23:11:04 -0700 X-Original-To: edk2-devel@lists.01.org X-ExtLoop1: 1 X-IronPort-AV: E=Sophos;i="5.37,213,1488873600"; d="scan'208";a="90698934" From: Jiaxin Wu To: edk2-devel@lists.01.org Date: Mon, 17 Apr 2017 14:11:02 +0800 Message-Id: <1492409462-3224-1-git-send-email-jiaxin.wu@intel.com> X-Mailer: git-send-email 1.9.5.msysgit.1 Subject: [edk2] [PATCH v2] NetworkPkg/TlsAuthConfigDxe: Close and free the file related resource X-BeenThere: edk2-devel@lists.01.org X-Mailman-Version: 2.1.22 Precedence: list List-Id: EDK II Development List-Unsubscribe: , List-Archive: List-Post: List-Help: List-Subscribe: , Cc: Ye Ting , Fu Siyuan , Wu Jiaxin , Zhang Chao B MIME-Version: 1.0 Content-Transfer-Encoding: quoted-printable Errors-To: edk2-devel-bounces@lists.01.org Sender: "edk2-devel" X-ZohoMail: RSF_4 Z_629925259 SPT_0 Content-Type: text/plain; charset="utf-8" v2: * Define one new internal function to clean the file content. TlsAuthConfigDxe open file by FileExplorerLib. It need to close file handler and free file related resource in some cases. * User enrolls Cert by escape the Config page. * The Cert is not X509 type. * User chooses another file after he selected a file. Cc: Zhang Chao B Cc: Ye Ting Cc: Fu Siyuan Contributed-under: TianoCore Contribution Agreement 1.0 Signed-off-by: Wu Jiaxin Reviewed-by: Chao Zhang --- NetworkPkg/TlsAuthConfigDxe/TlsAuthConfigImpl.c | 49 +++++++++++++--------= ---- 1 file changed, 25 insertions(+), 24 deletions(-) diff --git a/NetworkPkg/TlsAuthConfigDxe/TlsAuthConfigImpl.c b/NetworkPkg/T= lsAuthConfigDxe/TlsAuthConfigImpl.c index 81f7e7d..faefc72 100644 --- a/NetworkPkg/TlsAuthConfigDxe/TlsAuthConfigImpl.c +++ b/NetworkPkg/TlsAuthConfigDxe/TlsAuthConfigImpl.c @@ -294,11 +294,11 @@ ON_EXIT: } =20 /** Delete one entry from cert database. =20 - @param[in] PrivateData Module's private data. + @param[in] Private Module's private data. @param[in] VariableName The variable name of the database. @param[in] VendorGuid A unique identifier for the vendor. @param[in] LabelNumber Label number to insert opcodes. @param[in] FormId Form ID of current page. @param[in] QuestionIdBase Base question id of the cert list. @@ -475,22 +475,27 @@ ON_EXIT: ); } =20 =20 /** - Close an open file handle. + Clean the file related resource. =20 - @param[in] FileHandle The file handle to close. + @param[in] Private Module's private data. =20 **/ VOID -CloseFile ( - IN EFI_FILE_HANDLE FileHandle +CleanFileContext ( + IN TLS_AUTH_CONFIG_PRIVATE_DATA *Private ) { - if (FileHandle !=3D NULL) { - FileHandle->Close (FileHandle); + if (Private->FileContext->FHandle !=3D NULL) { + Private->FileContext->FHandle->Close (Private->FileContext->FHandle); + Private->FileContext->FHandle =3D NULL; + if (Private->FileContext->FileName!=3D NULL){ + FreePool(Private->FileContext->FileName); + Private->FileContext->FileName =3D NULL; + } } } =20 /** Read file content into BufferPtr, the size of the allocate buffer @@ -871,18 +876,11 @@ EnrollX509toVariable ( if (EFI_ERROR (Status)) { goto ON_EXIT; } =20 ON_EXIT: - - CloseFile (Private->FileContext->FHandle); - if (Private->FileContext->FileName !=3D NULL) { - FreePool(Private->FileContext->FileName); - Private->FileContext->FileName =3D NULL; - } - - Private->FileContext->FHandle =3D NULL; + CleanFileContext (Private); =20 if (Private->CertGuid !=3D NULL) { FreePool (Private->CertGuid); Private->CertGuid =3D NULL; } @@ -1559,11 +1557,12 @@ TlsAuthConfigAccessCallback ( } =20 HiiGetBrowserData (&gTlsAuthConfigGuid, mTlsAuthConfigStorageName, Buffe= rSize, (UINT8 *) IfrNvData); =20 if ((Action !=3D EFI_BROWSER_ACTION_CHANGED) && - (Action !=3D EFI_BROWSER_ACTION_CHANGING)) { + (Action !=3D EFI_BROWSER_ACTION_CHANGING) &&=20 + (Action !=3D EFI_BROWSER_ACTION_FORM_CLOSE)) { Status =3D EFI_UNSUPPORTED; goto EXIT; } =20 if (Action =3D=3D EFI_BROWSER_ACTION_CHANGING) { @@ -1590,34 +1589,34 @@ TlsAuthConfigAccessCallback ( // Refresh selected file. // CleanUpPage (LabelId, Private); break; case KEY_TLS_AUTH_CONFIG_ENROLL_CERT_FROM_FILE: + // + // If the file is already opened, clean the file related resource fi= rst.=20 + // + CleanFileContext (Private); + =20 ChooseFile( NULL, NULL, UpdateCAFromFile, &File); break; =20 case KEY_TLS_AUTH_CONFIG_VALUE_SAVE_AND_EXIT: Status =3D EnrollCertDatabase (Private, EFI_TLS_CA_CERTIFICATE_VARIA= BLE); if (EFI_ERROR (Status)) { + CleanFileContext (Private); + CreatePopUp ( EFI_LIGHTGRAY | EFI_BACKGROUND_BLUE, &Key, L"ERROR: Enroll Cert Failure!", NULL ); } break; =20 case KEY_TLS_AUTH_CONFIG_VALUE_NO_SAVE_AND_EXIT: - if (Private->FileContext->FHandle !=3D NULL) { - CloseFile (Private->FileContext->FHandle); - Private->FileContext->FHandle =3D NULL; - if (Private->FileContext->FileName!=3D NULL){ - FreePool(Private->FileContext->FileName); - Private->FileContext->FileName =3D NULL; - } - } + CleanFileContext (Private); =20 if (Private->CertGuid!=3D NULL) { FreePool (Private->CertGuid); Private->CertGuid =3D NULL; } @@ -1665,10 +1664,12 @@ TlsAuthConfigAccessCallback ( *ActionRequest =3D EFI_BROWSER_ACTION_REQUEST_FORM_APPLY; break; default: break; } + } else if (Action =3D=3D EFI_BROWSER_ACTION_FORM_CLOSE) { + CleanFileContext (Private); } =20 EXIT: =20 if (!EFI_ERROR (Status)) { --=20 1.9.5.msysgit.1 _______________________________________________ edk2-devel mailing list edk2-devel@lists.01.org https://lists.01.org/mailman/listinfo/edk2-devel