diff options
| -rw-r--r-- | MdeModulePkg/Core/Dxe/Image/Image.c | 14 |
1 files changed, 11 insertions, 3 deletions
diff --git a/MdeModulePkg/Core/Dxe/Image/Image.c b/MdeModulePkg/Core/Dxe/Image/Image.c index dbb662519f..477eaa3417 100644 --- a/MdeModulePkg/Core/Dxe/Image/Image.c +++ b/MdeModulePkg/Core/Dxe/Image/Image.c @@ -17,6 +17,8 @@ WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED. // Module Globals
//
+SPIN_LOCK mUnloadImageLock;
+
LOADED_IMAGE_PRIVATE_DATA *mCurrentImage = NULL;
LOAD_PE32_IMAGE_PRIVATE_DATA mLoadPe32PrivateData = {
@@ -138,6 +140,11 @@ CoreInitializeImageServices ( mCurrentImage = Image;
//
+ // Initialize spin lock
+ //
+ InitializeSpinLock (&mUnloadImageLock);
+
+ //
// Fill in DXE globals
//
gDxeCoreImageHandle = Image->Handle;
@@ -1362,13 +1369,14 @@ CoreUnloadImage ( {
EFI_STATUS Status;
LOADED_IMAGE_PRIVATE_DATA *Image;
- EFI_TPL OldTpl;
//
// Prevent possible reentrance to this function
// for the same ImageHandle
//
- OldTpl = CoreRaiseTpl (TPL_NOTIFY);
+ if (!AcquireSpinLockOrFail (&mUnloadImageLock)) {
+ return EFI_UNSUPPORTED;
+ }
Image = CoreLoadedImageInfo (ImageHandle);
if (Image == NULL ) {
@@ -1404,7 +1412,7 @@ CoreUnloadImage ( }
Done:
- CoreRestoreTpl (OldTpl);
+ ReleaseSpinLock (&mUnloadImageLock);
return Status;
}
|