diff options
Diffstat (limited to 'PrmPkg/PrmLoaderDxe/PrmLoaderDxe.c')
| -rw-r--r-- | PrmPkg/PrmLoaderDxe/PrmLoaderDxe.c | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/PrmPkg/PrmLoaderDxe/PrmLoaderDxe.c b/PrmPkg/PrmLoaderDxe/PrmLoaderDxe.c index b43e6d6bf0..85fffdcbd9 100644 --- a/PrmPkg/PrmLoaderDxe/PrmLoaderDxe.c +++ b/PrmPkg/PrmLoaderDxe/PrmLoaderDxe.c @@ -755,7 +755,11 @@ ProcessPrmModules ( &CurrentContextBuffer
);
if (!EFI_ERROR (Status)) {
- CurrentHandlerInfoStruct->PrmContextBuffer = (UINT64) CurrentContextBuffer;
+#ifdef ALLOCATE_CONTEXT_BUFFER_IN_FW
+ CurrentHandlerInfoStruct->PrmContextBuffer = (UINT64) (UINTN) CurrentContextBuffer;
+#else
+ CurrentHandlerInfoStruct->StaticDataBuffer = (UINT64) (UINTN) CurrentContextBuffer->StaticDataBuffer;
+#endif
}
Status = GetExportEntryAddress (
|