From mboxrd@z Thu Jan 1 00:00:00 1970 Return-Path: Received: from mail02.groups.io (mail02.groups.io [66.175.222.108]) by spool.mail.gandi.net (Postfix) with ESMTPS id 79BB7940EF8 for ; Fri, 1 Mar 2024 03:05:45 +0000 (UTC) DKIM-Signature: a=rsa-sha256; bh=+GZN0/CviUU+lWTRouoPpvYb/kXyO1JxIbnoH6SJCPo=; c=relaxed/simple; d=groups.io; h=From:To:Cc:Subject:Date:Message-Id:In-Reply-To:References:MIME-Version:Precedence:List-Subscribe:List-Help:Sender:List-Id:Mailing-List:Delivered-To:Reply-To:List-Unsubscribe-Post:List-Unsubscribe:Content-Transfer-Encoding; s=20140610; t=1709262344; v=1; b=sQa3cH1IuF8xjzxsARWepf50GPZH3kuVx/oK3CU0fXT62PqLVdPobEyeLT1Qz4SgoPaoCLBc VB15tIXuqbMrjvRYbum0nCJ0qc8pdxJzvdkzRk/02/ofM+AXjeBwzB7jO3GSVgk0uIwTHCq1DL5 YoWW9hoz6X0iveyIKTK2SMaY= X-Received: by 127.0.0.2 with SMTP id 4hkNYY7687511x2MqqpQEkBZ; Thu, 29 Feb 2024 19:05:44 -0800 X-Received: from mgamail.intel.com (mgamail.intel.com [198.175.65.17]) by mx.groups.io with SMTP id smtpd.web10.14105.1709262340273339436 for ; Thu, 29 Feb 2024 19:05:43 -0800 X-IronPort-AV: E=McAfee;i="6600,9927,10999"; a="3955930" X-IronPort-AV: E=Sophos;i="6.06,194,1705392000"; d="scan'208";a="3955930" X-Received: from fmviesa008.fm.intel.com ([10.60.135.148]) by orvoesa109.jf.intel.com with ESMTP/TLS/ECDHE-RSA-AES256-GCM-SHA384; 29 Feb 2024 19:05:43 -0800 X-ExtLoop1: 1 X-IronPort-AV: E=Sophos;i="6.06,194,1705392000"; d="scan'208";a="8181893" X-Received: from shwdesfp01.ccr.corp.intel.com ([10.239.158.151]) by fmviesa008-auth.fm.intel.com with ESMTP/TLS/ECDHE-RSA-AES256-GCM-SHA384; 29 Feb 2024 19:05:41 -0800 From: "Zhiguang Liu" To: devel@edk2.groups.io Cc: Zhiguang Liu , Liming Gao , Jiaxin Wu , Ray Ni , Laszlo Ersek Subject: [edk2-devel] [PATCH v3 2/4] MdeModulePkg/SMM: Disallow unregister SMI handler in other SMI handler Date: Fri, 1 Mar 2024 11:01:31 +0800 Message-Id: <20240301030133.628-3-zhiguang.liu@intel.com> In-Reply-To: <20240301030133.628-1-zhiguang.liu@intel.com> References: <20240301030133.628-1-zhiguang.liu@intel.com> MIME-Version: 1.0 Precedence: Bulk List-Subscribe: List-Help: Sender: devel@edk2.groups.io List-Id: Mailing-List: list devel@edk2.groups.io; contact devel+owner@edk2.groups.io Reply-To: devel@edk2.groups.io,zhiguang.liu@intel.com List-Unsubscribe-Post: List-Unsubscribe=One-Click List-Unsubscribe: X-Gm-Message-State: pRnW9B6m2yUwEoO52OSpuTw8x7686176AA= Content-Transfer-Encoding: 8bit X-GND-Status: LEGIT Authentication-Results: spool.mail.gandi.net; dkim=pass header.d=groups.io header.s=20140610 header.b=sQa3cH1I; dmarc=fail reason="SPF not aligned (relaxed), DKIM not aligned (relaxed)" header.from=intel.com (policy=none); spf=pass (spool.mail.gandi.net: domain of bounce@groups.io designates 66.175.222.108 as permitted sender) smtp.mailfrom=bounce@groups.io In last patch, we add code support to unregister SMI handler inside itself. However, the code doesn't support unregister SMI handler insider other SMI handler. While this is not a must-have usage. So add check to disallow unregister SMI handler in other SMI handler. Cc: Liming Gao Cc: Jiaxin Wu Cc: Ray Ni Cc: Laszlo Ersek Signed-off-by: Zhiguang Liu --- MdeModulePkg/Core/PiSmmCore/Smi.c | 29 +++++++++++++++++++++-------- 1 file changed, 21 insertions(+), 8 deletions(-) diff --git a/MdeModulePkg/Core/PiSmmCore/Smi.c b/MdeModulePkg/Core/PiSmmCore/Smi.c index 3489c130fd..b3a81ac877 100644 --- a/MdeModulePkg/Core/PiSmmCore/Smi.c +++ b/MdeModulePkg/Core/PiSmmCore/Smi.c @@ -8,7 +8,8 @@ #include "PiSmmCore.h" -LIST_ENTRY mSmiEntryList = INITIALIZE_LIST_HEAD_VARIABLE (mSmiEntryList); +SMI_HANDLER *mCurrentSmiHandler = NULL; +LIST_ENTRY mSmiEntryList = INITIALIZE_LIST_HEAD_VARIABLE (mSmiEntryList); SMI_ENTRY mRootSmiEntry = { SMI_ENTRY_SIGNATURE, @@ -142,13 +143,18 @@ SmiManage ( // Link points to may be freed if unregister SMI handler. // Link = Link->ForwardLink; - - Status = SmiHandler->Handler ( - (EFI_HANDLE)SmiHandler, - Context, - CommBuffer, - CommBufferSize - ); + // + // Assign gCurrentSmiHandle before calling the SMI handler and + // set to NULL when it returns. + // + mCurrentSmiHandler = SmiHandler; + Status = SmiHandler->Handler ( + (EFI_HANDLE)SmiHandler, + Context, + CommBuffer, + CommBufferSize + ); + mCurrentSmiHandler = NULL; switch (Status) { case EFI_INTERRUPT_PENDING: @@ -328,6 +334,13 @@ SmiHandlerUnRegister ( return EFI_INVALID_PARAMETER; } + // + // Do not allow to unregister SMI Handler inside other SMI Handler + // + if ((mCurrentSmiHandler != NULL) && (mCurrentSmiHandler != SmiHandler)) { + return EFI_INVALID_PARAMETER; + } + SmiEntry = SmiHandler->SmiEntry; RemoveEntryList (&SmiHandler->Link); -- 2.31.1.windows.1 -=-=-=-=-=-=-=-=-=-=-=- Groups.io Links: You receive all messages sent to this group. View/Reply Online (#116204): https://edk2.groups.io/g/devel/message/116204 Mute This Topic: https://groups.io/mt/104657667/7686176 Group Owner: devel+owner@edk2.groups.io Unsubscribe: https://edk2.groups.io/g/devel/unsub [rebecca@openfw.io] -=-=-=-=-=-=-=-=-=-=-=-