]> git.proxmox.com Git - mirror_edk2.git/blobdiff - UefiCpuPkg/CpuDxe/CpuPageTable.c
UefiCpuPkg: Apply uncrustify changes
[mirror_edk2.git] / UefiCpuPkg / CpuDxe / CpuPageTable.c
index c34059f87f0bb707660197a54cfc4c1437f5a5e8..d9e65ab4b22a626084366340fbeb7da9a7deb00a 100644 (file)
 ///\r
 /// Page Table Entry\r
 ///\r
-#define IA32_PG_P                   BIT0\r
-#define IA32_PG_RW                  BIT1\r
-#define IA32_PG_U                   BIT2\r
-#define IA32_PG_WT                  BIT3\r
-#define IA32_PG_CD                  BIT4\r
-#define IA32_PG_A                   BIT5\r
-#define IA32_PG_D                   BIT6\r
-#define IA32_PG_PS                  BIT7\r
-#define IA32_PG_PAT_2M              BIT12\r
-#define IA32_PG_PAT_4K              IA32_PG_PS\r
-#define IA32_PG_PMNT                BIT62\r
-#define IA32_PG_NX                  BIT63\r
-\r
-#define PAGE_ATTRIBUTE_BITS         (IA32_PG_D | IA32_PG_A | IA32_PG_U | IA32_PG_RW | IA32_PG_P)\r
+#define IA32_PG_P       BIT0\r
+#define IA32_PG_RW      BIT1\r
+#define IA32_PG_U       BIT2\r
+#define IA32_PG_WT      BIT3\r
+#define IA32_PG_CD      BIT4\r
+#define IA32_PG_A       BIT5\r
+#define IA32_PG_D       BIT6\r
+#define IA32_PG_PS      BIT7\r
+#define IA32_PG_PAT_2M  BIT12\r
+#define IA32_PG_PAT_4K  IA32_PG_PS\r
+#define IA32_PG_PMNT    BIT62\r
+#define IA32_PG_NX      BIT63\r
+\r
+#define PAGE_ATTRIBUTE_BITS  (IA32_PG_D | IA32_PG_A | IA32_PG_U | IA32_PG_RW | IA32_PG_P)\r
 //\r
 // Bits 1, 2, 5, 6 are reserved in the IA32 PAE PDPTE\r
 // X64 PAE PDPTE does not have such restriction\r
 //\r
-#define IA32_PAE_PDPTE_ATTRIBUTE_BITS    (IA32_PG_P)\r
+#define IA32_PAE_PDPTE_ATTRIBUTE_BITS  (IA32_PG_P)\r
 \r
-#define PAGE_PROGATE_BITS           (IA32_PG_NX | PAGE_ATTRIBUTE_BITS)\r
+#define PAGE_PROGATE_BITS  (IA32_PG_NX | PAGE_ATTRIBUTE_BITS)\r
 \r
 #define PAGING_4K_MASK  0xFFF\r
 #define PAGING_2M_MASK  0x1FFFFF\r
@@ -52,9 +52,9 @@
 \r
 #define PAGING_PAE_INDEX_MASK  0x1FF\r
 \r
-#define PAGING_4K_ADDRESS_MASK_64 0x000FFFFFFFFFF000ull\r
-#define PAGING_2M_ADDRESS_MASK_64 0x000FFFFFFFE00000ull\r
-#define PAGING_1G_ADDRESS_MASK_64 0x000FFFFFC0000000ull\r
+#define PAGING_4K_ADDRESS_MASK_64  0x000FFFFFFFFFF000ull\r
+#define PAGING_2M_ADDRESS_MASK_64  0x000FFFFFFFE00000ull\r
+#define PAGING_1G_ADDRESS_MASK_64  0x000FFFFFC0000000ull\r
 \r
 #define MAX_PF_ENTRY_COUNT        10\r
 #define MAX_DEBUG_MESSAGE_LENGTH  0x100\r
@@ -68,9 +68,9 @@ typedef enum {
 } PAGE_ATTRIBUTE;\r
 \r
 typedef struct {\r
-  PAGE_ATTRIBUTE   Attribute;\r
-  UINT64           Length;\r
-  UINT64           AddressMask;\r
+  PAGE_ATTRIBUTE    Attribute;\r
+  UINT64            Length;\r
+  UINT64            AddressMask;\r
 } PAGE_ATTRIBUTE_TABLE;\r
 \r
 typedef enum {\r
@@ -79,21 +79,21 @@ typedef enum {
   PageActionClear,\r
 } PAGE_ACTION;\r
 \r
-PAGE_ATTRIBUTE_TABLE mPageAttributeTable[] = {\r
-  {Page4K,  SIZE_4KB, PAGING_4K_ADDRESS_MASK_64},\r
-  {Page2M,  SIZE_2MB, PAGING_2M_ADDRESS_MASK_64},\r
-  {Page1G,  SIZE_1GB, PAGING_1G_ADDRESS_MASK_64},\r
+PAGE_ATTRIBUTE_TABLE  mPageAttributeTable[] = {\r
+  { Page4K, SIZE_4KB, PAGING_4K_ADDRESS_MASK_64 },\r
+  { Page2M, SIZE_2MB, PAGING_2M_ADDRESS_MASK_64 },\r
+  { Page1G, SIZE_1GB, PAGING_1G_ADDRESS_MASK_64 },\r
 };\r
 \r
-PAGE_TABLE_POOL                   *mPageTablePool = NULL;\r
-BOOLEAN                           mPageTablePoolLock = FALSE;\r
-PAGE_TABLE_LIB_PAGING_CONTEXT     mPagingContext;\r
-EFI_SMM_BASE2_PROTOCOL            *mSmmBase2 = NULL;\r
+PAGE_TABLE_POOL                *mPageTablePool    = NULL;\r
+BOOLEAN                        mPageTablePoolLock = FALSE;\r
+PAGE_TABLE_LIB_PAGING_CONTEXT  mPagingContext;\r
+EFI_SMM_BASE2_PROTOCOL         *mSmmBase2 = NULL;\r
 \r
 //\r
 // Record the page fault exception count for one instruction execution.\r
 //\r
-UINTN                     *mPFEntryCount;\r
+UINTN  *mPFEntryCount;\r
 UINT64                    *(*mLastPFEntryPointer)[MAX_PF_ENTRY_COUNT];\r
 \r
 /**\r
@@ -119,7 +119,7 @@ IsInSmm (
   VOID\r
   )\r
 {\r
-  BOOLEAN                 InSmm;\r
+  BOOLEAN  InSmm;\r
 \r
   InSmm = FALSE;\r
   if (mSmmBase2 == NULL) {\r
@@ -137,7 +137,7 @@ IsInSmm (
   // load its own page table.\r
   //\r
   return (InSmm &&\r
-          mPagingContext.ContextData.X64.PageTableBase != (UINT64)AsmReadCr3());\r
+          mPagingContext.ContextData.X64.PageTableBase != (UINT64)AsmReadCr3 ());\r
 }\r
 \r
 /**\r
@@ -147,23 +147,23 @@ IsInSmm (
 **/\r
 VOID\r
 GetCurrentPagingContext (\r
-  IN OUT PAGE_TABLE_LIB_PAGING_CONTEXT     *PagingContext\r
+  IN OUT PAGE_TABLE_LIB_PAGING_CONTEXT  *PagingContext\r
   )\r
 {\r
-  UINT32                          RegEax;\r
-  CPUID_EXTENDED_CPU_SIG_EDX      RegEdx;\r
-  MSR_IA32_EFER_REGISTER          MsrEfer;\r
-  IA32_CR4                        Cr4;\r
-  IA32_CR0                        Cr0;\r
-  UINT32                          *Attributes;\r
-  UINTN                           *PageTableBase;\r
+  UINT32                      RegEax;\r
+  CPUID_EXTENDED_CPU_SIG_EDX  RegEdx;\r
+  MSR_IA32_EFER_REGISTER      MsrEfer;\r
+  IA32_CR4                    Cr4;\r
+  IA32_CR0                    Cr0;\r
+  UINT32                      *Attributes;\r
+  UINTN                       *PageTableBase;\r
 \r
   //\r
   // Don't retrieve current paging context from processor if in SMM mode.\r
   //\r
   if (!IsInSmm ()) {\r
-    ZeroMem (&mPagingContext, sizeof(mPagingContext));\r
-    if (sizeof(UINTN) == sizeof(UINT64)) {\r
+    ZeroMem (&mPagingContext, sizeof (mPagingContext));\r
+    if (sizeof (UINTN) == sizeof (UINT64)) {\r
       mPagingContext.MachineType = IMAGE_FILE_MACHINE_X64;\r
     } else {\r
       mPagingContext.MachineType = IMAGE_FILE_MACHINE_I386;\r
@@ -179,15 +179,19 @@ GetCurrentPagingContext (
     } else {\r
       *PageTableBase = 0;\r
     }\r
+\r
     if (Cr0.Bits.WP  != 0) {\r
       *Attributes |= PAGE_TABLE_LIB_PAGING_CONTEXT_IA32_X64_ATTRIBUTES_WP_ENABLE;\r
     }\r
+\r
     if (Cr4.Bits.PSE != 0) {\r
       *Attributes |= PAGE_TABLE_LIB_PAGING_CONTEXT_IA32_X64_ATTRIBUTES_PSE;\r
     }\r
+\r
     if (Cr4.Bits.PAE != 0) {\r
       *Attributes |= PAGE_TABLE_LIB_PAGING_CONTEXT_IA32_X64_ATTRIBUTES_PAE;\r
     }\r
+\r
     if (Cr4.Bits.LA57 != 0) {\r
       *Attributes |= PAGE_TABLE_LIB_PAGING_CONTEXT_IA32_X64_ATTRIBUTES_5_LEVEL;\r
     }\r
@@ -198,7 +202,7 @@ GetCurrentPagingContext (
 \r
       if (RegEdx.Bits.NX != 0) {\r
         // XD supported\r
-        MsrEfer.Uint64 = AsmReadMsr64(MSR_CORE_IA32_EFER);\r
+        MsrEfer.Uint64 = AsmReadMsr64 (MSR_CORE_IA32_EFER);\r
         if (MsrEfer.Bits.NXE != 0) {\r
           // XD activated\r
           *Attributes |= PAGE_TABLE_LIB_PAGING_CONTEXT_IA32_X64_ATTRIBUTES_XD_ACTIVATED;\r
@@ -231,11 +235,13 @@ PageAttributeToLength (
   )\r
 {\r
   UINTN  Index;\r
-  for (Index = 0; Index < sizeof(mPageAttributeTable)/sizeof(mPageAttributeTable[0]); Index++) {\r
+\r
+  for (Index = 0; Index < sizeof (mPageAttributeTable)/sizeof (mPageAttributeTable[0]); Index++) {\r
     if (PageAttribute == mPageAttributeTable[Index].Attribute) {\r
       return (UINTN)mPageAttributeTable[Index].Length;\r
     }\r
   }\r
+\r
   return 0;\r
 }\r
 \r
@@ -252,11 +258,13 @@ PageAttributeToMask (
   )\r
 {\r
   UINTN  Index;\r
-  for (Index = 0; Index < sizeof(mPageAttributeTable)/sizeof(mPageAttributeTable[0]); Index++) {\r
+\r
+  for (Index = 0; Index < sizeof (mPageAttributeTable)/sizeof (mPageAttributeTable[0]); Index++) {\r
     if (PageAttribute == mPageAttributeTable[Index].Attribute) {\r
       return (UINTN)mPageAttributeTable[Index].AddressMask;\r
     }\r
   }\r
+\r
   return 0;\r
 }\r
 \r
@@ -271,22 +279,22 @@ PageAttributeToMask (
 **/\r
 VOID *\r
 GetPageTableEntry (\r
-  IN  PAGE_TABLE_LIB_PAGING_CONTEXT     *PagingContext,\r
-  IN  PHYSICAL_ADDRESS                  Address,\r
-  OUT PAGE_ATTRIBUTE                    *PageAttribute\r
+  IN  PAGE_TABLE_LIB_PAGING_CONTEXT  *PagingContext,\r
+  IN  PHYSICAL_ADDRESS               Address,\r
+  OUT PAGE_ATTRIBUTE                 *PageAttribute\r
   )\r
 {\r
-  UINTN                 Index1;\r
-  UINTN                 Index2;\r
-  UINTN                 Index3;\r
-  UINTN                 Index4;\r
-  UINTN                 Index5;\r
-  UINT64                *L1PageTable;\r
-  UINT64                *L2PageTable;\r
-  UINT64                *L3PageTable;\r
-  UINT64                *L4PageTable;\r
-  UINT64                *L5PageTable;\r
-  UINT64                AddressEncMask;\r
+  UINTN   Index1;\r
+  UINTN   Index2;\r
+  UINTN   Index3;\r
+  UINTN   Index4;\r
+  UINTN   Index5;\r
+  UINT64  *L1PageTable;\r
+  UINT64  *L2PageTable;\r
+  UINT64  *L3PageTable;\r
+  UINT64  *L4PageTable;\r
+  UINT64  *L5PageTable;\r
+  UINT64  AddressEncMask;\r
 \r
   ASSERT (PagingContext != NULL);\r
 \r
@@ -312,6 +320,7 @@ GetPageTableEntry (
     } else {\r
       L4PageTable = (UINT64 *)(UINTN)PagingContext->ContextData.X64.PageTableBase;\r
     }\r
+\r
     if (L4PageTable[Index4] == 0) {\r
       *PageAttribute = PageNone;\r
       return NULL;\r
@@ -319,13 +328,15 @@ GetPageTableEntry (
 \r
     L3PageTable = (UINT64 *)(UINTN)(L4PageTable[Index4] & ~AddressEncMask & PAGING_4K_ADDRESS_MASK_64);\r
   } else {\r
-    ASSERT((PagingContext->ContextData.Ia32.Attributes & PAGE_TABLE_LIB_PAGING_CONTEXT_IA32_X64_ATTRIBUTES_PAE) != 0);\r
+    ASSERT ((PagingContext->ContextData.Ia32.Attributes & PAGE_TABLE_LIB_PAGING_CONTEXT_IA32_X64_ATTRIBUTES_PAE) != 0);\r
     L3PageTable = (UINT64 *)(UINTN)PagingContext->ContextData.Ia32.PageTableBase;\r
   }\r
+\r
   if (L3PageTable[Index3] == 0) {\r
     *PageAttribute = PageNone;\r
     return NULL;\r
   }\r
+\r
   if ((L3PageTable[Index3] & IA32_PG_PS) != 0) {\r
     // 1G\r
     *PageAttribute = Page1G;\r
@@ -337,6 +348,7 @@ GetPageTableEntry (
     *PageAttribute = PageNone;\r
     return NULL;\r
   }\r
+\r
   if ((L2PageTable[Index2] & IA32_PG_PS) != 0) {\r
     // 2M\r
     *PageAttribute = Page2M;\r
@@ -349,6 +361,7 @@ GetPageTableEntry (
     *PageAttribute = PageNone;\r
     return NULL;\r
   }\r
+\r
   *PageAttribute = Page4K;\r
   return &L1PageTable[Index1];\r
 }\r
@@ -362,20 +375,24 @@ GetPageTableEntry (
 **/\r
 UINT64\r
 GetAttributesFromPageEntry (\r
-  IN  UINT64                            *PageEntry\r
+  IN  UINT64  *PageEntry\r
   )\r
 {\r
   UINT64  Attributes;\r
+\r
   Attributes = 0;\r
   if ((*PageEntry & IA32_PG_P) == 0) {\r
     Attributes |= EFI_MEMORY_RP;\r
   }\r
+\r
   if ((*PageEntry & IA32_PG_RW) == 0) {\r
     Attributes |= EFI_MEMORY_RO;\r
   }\r
+\r
   if ((*PageEntry & IA32_PG_NX) != 0) {\r
     Attributes |= EFI_MEMORY_XP;\r
   }\r
+\r
   return Attributes;\r
 }\r
 \r
@@ -390,11 +407,11 @@ GetAttributesFromPageEntry (
 **/\r
 VOID\r
 ConvertPageEntryAttribute (\r
-  IN  PAGE_TABLE_LIB_PAGING_CONTEXT     *PagingContext,\r
-  IN  UINT64                            *PageEntry,\r
-  IN  UINT64                            Attributes,\r
-  IN  PAGE_ACTION                       PageAction,\r
-  OUT BOOLEAN                           *IsModified\r
+  IN  PAGE_TABLE_LIB_PAGING_CONTEXT  *PagingContext,\r
+  IN  UINT64                         *PageEntry,\r
+  IN  UINT64                         Attributes,\r
+  IN  PAGE_ACTION                    PageAction,\r
+  OUT BOOLEAN                        *IsModified\r
   )\r
 {\r
   UINT64  CurrentPageEntry;\r
@@ -402,45 +419,46 @@ ConvertPageEntryAttribute (
   UINT32  *PageAttributes;\r
 \r
   CurrentPageEntry = *PageEntry;\r
-  NewPageEntry = CurrentPageEntry;\r
+  NewPageEntry     = CurrentPageEntry;\r
   if ((Attributes & EFI_MEMORY_RP) != 0) {\r
     switch (PageAction) {\r
-    case PageActionAssign:\r
-    case PageActionSet:\r
-      NewPageEntry &= ~(UINT64)IA32_PG_P;\r
-      break;\r
-    case PageActionClear:\r
-      NewPageEntry |= IA32_PG_P;\r
-      break;\r
+      case PageActionAssign:\r
+      case PageActionSet:\r
+        NewPageEntry &= ~(UINT64)IA32_PG_P;\r
+        break;\r
+      case PageActionClear:\r
+        NewPageEntry |= IA32_PG_P;\r
+        break;\r
     }\r
   } else {\r
     switch (PageAction) {\r
-    case PageActionAssign:\r
-      NewPageEntry |= IA32_PG_P;\r
-      break;\r
-    case PageActionSet:\r
-    case PageActionClear:\r
-      break;\r
+      case PageActionAssign:\r
+        NewPageEntry |= IA32_PG_P;\r
+        break;\r
+      case PageActionSet:\r
+      case PageActionClear:\r
+        break;\r
     }\r
   }\r
+\r
   if ((Attributes & EFI_MEMORY_RO) != 0) {\r
     switch (PageAction) {\r
-    case PageActionAssign:\r
-    case PageActionSet:\r
-      NewPageEntry &= ~(UINT64)IA32_PG_RW;\r
-      break;\r
-    case PageActionClear:\r
-      NewPageEntry |= IA32_PG_RW;\r
-      break;\r
+      case PageActionAssign:\r
+      case PageActionSet:\r
+        NewPageEntry &= ~(UINT64)IA32_PG_RW;\r
+        break;\r
+      case PageActionClear:\r
+        NewPageEntry |= IA32_PG_RW;\r
+        break;\r
     }\r
   } else {\r
     switch (PageAction) {\r
-    case PageActionAssign:\r
-      NewPageEntry |= IA32_PG_RW;\r
-      break;\r
-    case PageActionSet:\r
-    case PageActionClear:\r
-      break;\r
+      case PageActionAssign:\r
+        NewPageEntry |= IA32_PG_RW;\r
+        break;\r
+      case PageActionSet:\r
+      case PageActionClear:\r
+        break;\r
     }\r
   }\r
 \r
@@ -449,25 +467,26 @@ ConvertPageEntryAttribute (
   if ((*PageAttributes & PAGE_TABLE_LIB_PAGING_CONTEXT_IA32_X64_ATTRIBUTES_XD_ACTIVATED) != 0) {\r
     if ((Attributes & EFI_MEMORY_XP) != 0) {\r
       switch (PageAction) {\r
-      case PageActionAssign:\r
-      case PageActionSet:\r
-        NewPageEntry |= IA32_PG_NX;\r
-        break;\r
-      case PageActionClear:\r
-        NewPageEntry &= ~IA32_PG_NX;\r
-        break;\r
+        case PageActionAssign:\r
+        case PageActionSet:\r
+          NewPageEntry |= IA32_PG_NX;\r
+          break;\r
+        case PageActionClear:\r
+          NewPageEntry &= ~IA32_PG_NX;\r
+          break;\r
       }\r
     } else {\r
       switch (PageAction) {\r
-      case PageActionAssign:\r
-        NewPageEntry &= ~IA32_PG_NX;\r
-        break;\r
-      case PageActionSet:\r
-      case PageActionClear:\r
-        break;\r
+        case PageActionAssign:\r
+          NewPageEntry &= ~IA32_PG_NX;\r
+          break;\r
+        case PageActionSet:\r
+        case PageActionClear:\r
+          break;\r
       }\r
     }\r
   }\r
+\r
   *PageEntry = NewPageEntry;\r
   if (CurrentPageEntry != NewPageEntry) {\r
     *IsModified = TRUE;\r
@@ -490,13 +509,13 @@ ConvertPageEntryAttribute (
 **/\r
 PAGE_ATTRIBUTE\r
 NeedSplitPage (\r
-  IN  PHYSICAL_ADDRESS                  BaseAddress,\r
-  IN  UINT64                            Length,\r
-  IN  UINT64                            *PageEntry,\r
-  IN  PAGE_ATTRIBUTE                    PageAttribute\r
+  IN  PHYSICAL_ADDRESS  BaseAddress,\r
+  IN  UINT64            Length,\r
+  IN  UINT64            *PageEntry,\r
+  IN  PAGE_ATTRIBUTE    PageAttribute\r
   )\r
 {\r
-  UINT64                PageEntryLength;\r
+  UINT64  PageEntryLength;\r
 \r
   PageEntryLength = PageAttributeToLength (PageAttribute);\r
 \r
@@ -525,16 +544,16 @@ NeedSplitPage (
 **/\r
 RETURN_STATUS\r
 SplitPage (\r
-  IN  UINT64                            *PageEntry,\r
-  IN  PAGE_ATTRIBUTE                    PageAttribute,\r
-  IN  PAGE_ATTRIBUTE                    SplitAttribute,\r
-  IN  PAGE_TABLE_LIB_ALLOCATE_PAGES     AllocatePagesFunc\r
+  IN  UINT64                         *PageEntry,\r
+  IN  PAGE_ATTRIBUTE                 PageAttribute,\r
+  IN  PAGE_ATTRIBUTE                 SplitAttribute,\r
+  IN  PAGE_TABLE_LIB_ALLOCATE_PAGES  AllocatePagesFunc\r
   )\r
 {\r
-  UINT64   BaseAddress;\r
-  UINT64   *NewPageEntry;\r
-  UINTN    Index;\r
-  UINT64   AddressEncMask;\r
+  UINT64  BaseAddress;\r
+  UINT64  *NewPageEntry;\r
+  UINTN   Index;\r
+  UINT64  AddressEncMask;\r
 \r
   ASSERT (PageAttribute == Page2M || PageAttribute == Page1G);\r
 \r
@@ -555,10 +574,12 @@ SplitPage (
       if (NewPageEntry == NULL) {\r
         return RETURN_OUT_OF_RESOURCES;\r
       }\r
+\r
       BaseAddress = *PageEntry & ~AddressEncMask & PAGING_2M_ADDRESS_MASK_64;\r
-      for (Index = 0; Index < SIZE_4KB / sizeof(UINT64); Index++) {\r
+      for (Index = 0; Index < SIZE_4KB / sizeof (UINT64); Index++) {\r
         NewPageEntry[Index] = (BaseAddress + SIZE_4KB * Index) | AddressEncMask | ((*PageEntry) & PAGE_PROGATE_BITS);\r
       }\r
+\r
       (*PageEntry) = (UINT64)(UINTN)NewPageEntry | AddressEncMask | ((*PageEntry) & PAGE_ATTRIBUTE_BITS);\r
       return RETURN_SUCCESS;\r
     } else {\r
@@ -570,16 +591,18 @@ SplitPage (
     // No need support 1G->4K directly, we should use 1G->2M, then 2M->4K to get more compact page table.\r
     //\r
     ASSERT (SplitAttribute == Page2M || SplitAttribute == Page4K);\r
-    if ((SplitAttribute == Page2M || SplitAttribute == Page4K)) {\r
+    if (((SplitAttribute == Page2M) || (SplitAttribute == Page4K))) {\r
       NewPageEntry = AllocatePagesFunc (1);\r
       DEBUG ((DEBUG_VERBOSE, "Split - 0x%x\n", NewPageEntry));\r
       if (NewPageEntry == NULL) {\r
         return RETURN_OUT_OF_RESOURCES;\r
       }\r
+\r
       BaseAddress = *PageEntry & ~AddressEncMask  & PAGING_1G_ADDRESS_MASK_64;\r
-      for (Index = 0; Index < SIZE_4KB / sizeof(UINT64); Index++) {\r
+      for (Index = 0; Index < SIZE_4KB / sizeof (UINT64); Index++) {\r
         NewPageEntry[Index] = (BaseAddress + SIZE_2MB * Index) | AddressEncMask | IA32_PG_PS | ((*PageEntry) & PAGE_PROGATE_BITS);\r
       }\r
+\r
       (*PageEntry) = (UINT64)(UINTN)NewPageEntry | AddressEncMask | ((*PageEntry) & PAGE_ATTRIBUTE_BITS);\r
       return RETURN_SUCCESS;\r
     } else {\r
@@ -603,14 +626,16 @@ IsReadOnlyPageWriteProtected (
   )\r
 {\r
   IA32_CR0  Cr0;\r
+\r
   //\r
   // To avoid unforseen consequences, don't touch paging settings in SMM mode\r
   // in this driver.\r
   //\r
   if (!IsInSmm ()) {\r
     Cr0.UintN = AsmReadCr0 ();\r
-    return (BOOLEAN) (Cr0.Bits.WP != 0);\r
+    return (BOOLEAN)(Cr0.Bits.WP != 0);\r
   }\r
+\r
   return FALSE;\r
 }\r
 \r
@@ -623,12 +648,13 @@ DisableReadOnlyPageWriteProtect (
   )\r
 {\r
   IA32_CR0  Cr0;\r
+\r
   //\r
   // To avoid unforseen consequences, don't touch paging settings in SMM mode\r
   // in this driver.\r
   //\r
   if (!IsInSmm ()) {\r
-    Cr0.UintN = AsmReadCr0 ();\r
+    Cr0.UintN   = AsmReadCr0 ();\r
     Cr0.Bits.WP = 0;\r
     AsmWriteCr0 (Cr0.UintN);\r
   }\r
@@ -643,12 +669,13 @@ EnableReadOnlyPageWriteProtect (
   )\r
 {\r
   IA32_CR0  Cr0;\r
+\r
   //\r
   // To avoid unforseen consequences, don't touch paging settings in SMM mode\r
   // in this driver.\r
   //\r
   if (!IsInSmm ()) {\r
-    Cr0.UintN = AsmReadCr0 ();\r
+    Cr0.UintN   = AsmReadCr0 ();\r
     Cr0.Bits.WP = 1;\r
     AsmWriteCr0 (Cr0.UintN);\r
   }\r
@@ -685,33 +712,35 @@ EnableReadOnlyPageWriteProtect (
 **/\r
 RETURN_STATUS\r
 ConvertMemoryPageAttributes (\r
-  IN  PAGE_TABLE_LIB_PAGING_CONTEXT     *PagingContext OPTIONAL,\r
-  IN  PHYSICAL_ADDRESS                  BaseAddress,\r
-  IN  UINT64                            Length,\r
-  IN  UINT64                            Attributes,\r
-  IN  PAGE_ACTION                       PageAction,\r
-  IN  PAGE_TABLE_LIB_ALLOCATE_PAGES     AllocatePagesFunc OPTIONAL,\r
-  OUT BOOLEAN                           *IsSplitted   OPTIONAL,\r
-  OUT BOOLEAN                           *IsModified   OPTIONAL\r
+  IN  PAGE_TABLE_LIB_PAGING_CONTEXT  *PagingContext OPTIONAL,\r
+  IN  PHYSICAL_ADDRESS               BaseAddress,\r
+  IN  UINT64                         Length,\r
+  IN  UINT64                         Attributes,\r
+  IN  PAGE_ACTION                    PageAction,\r
+  IN  PAGE_TABLE_LIB_ALLOCATE_PAGES  AllocatePagesFunc OPTIONAL,\r
+  OUT BOOLEAN                        *IsSplitted   OPTIONAL,\r
+  OUT BOOLEAN                        *IsModified   OPTIONAL\r
   )\r
 {\r
-  PAGE_TABLE_LIB_PAGING_CONTEXT     CurrentPagingContext;\r
-  UINT64                            *PageEntry;\r
-  PAGE_ATTRIBUTE                    PageAttribute;\r
-  UINTN                             PageEntryLength;\r
-  PAGE_ATTRIBUTE                    SplitAttribute;\r
-  RETURN_STATUS                     Status;\r
-  BOOLEAN                           IsEntryModified;\r
-  BOOLEAN                           IsWpEnabled;\r
+  PAGE_TABLE_LIB_PAGING_CONTEXT  CurrentPagingContext;\r
+  UINT64                         *PageEntry;\r
+  PAGE_ATTRIBUTE                 PageAttribute;\r
+  UINTN                          PageEntryLength;\r
+  PAGE_ATTRIBUTE                 SplitAttribute;\r
+  RETURN_STATUS                  Status;\r
+  BOOLEAN                        IsEntryModified;\r
+  BOOLEAN                        IsWpEnabled;\r
 \r
   if ((BaseAddress & (SIZE_4KB - 1)) != 0) {\r
     DEBUG ((DEBUG_ERROR, "BaseAddress(0x%lx) is not aligned!\n", BaseAddress));\r
     return EFI_UNSUPPORTED;\r
   }\r
+\r
   if ((Length & (SIZE_4KB - 1)) != 0) {\r
     DEBUG ((DEBUG_ERROR, "Length(0x%lx) is not aligned!\n", Length));\r
     return EFI_UNSUPPORTED;\r
   }\r
+\r
   if (Length == 0) {\r
     DEBUG ((DEBUG_ERROR, "Length is 0!\n"));\r
     return RETURN_INVALID_PARAMETER;\r
@@ -725,44 +754,50 @@ ConvertMemoryPageAttributes (
   if (PagingContext == NULL) {\r
     GetCurrentPagingContext (&CurrentPagingContext);\r
   } else {\r
-    CopyMem (&CurrentPagingContext, PagingContext, sizeof(CurrentPagingContext));\r
+    CopyMem (&CurrentPagingContext, PagingContext, sizeof (CurrentPagingContext));\r
   }\r
-  switch(CurrentPagingContext.MachineType) {\r
-  case IMAGE_FILE_MACHINE_I386:\r
-    if (CurrentPagingContext.ContextData.Ia32.PageTableBase == 0) {\r
-      if (Attributes == 0) {\r
-        return EFI_SUCCESS;\r
-      } else {\r
-        DEBUG ((DEBUG_ERROR, "PageTable is 0!\n"));\r
+\r
+  switch (CurrentPagingContext.MachineType) {\r
+    case IMAGE_FILE_MACHINE_I386:\r
+      if (CurrentPagingContext.ContextData.Ia32.PageTableBase == 0) {\r
+        if (Attributes == 0) {\r
+          return EFI_SUCCESS;\r
+        } else {\r
+          DEBUG ((DEBUG_ERROR, "PageTable is 0!\n"));\r
+          return EFI_UNSUPPORTED;\r
+        }\r
+      }\r
+\r
+      if ((CurrentPagingContext.ContextData.Ia32.Attributes & PAGE_TABLE_LIB_PAGING_CONTEXT_IA32_X64_ATTRIBUTES_PAE) == 0) {\r
+        DEBUG ((DEBUG_ERROR, "Non-PAE Paging!\n"));\r
         return EFI_UNSUPPORTED;\r
       }\r
-    }\r
-    if ((CurrentPagingContext.ContextData.Ia32.Attributes & PAGE_TABLE_LIB_PAGING_CONTEXT_IA32_X64_ATTRIBUTES_PAE) == 0) {\r
-      DEBUG ((DEBUG_ERROR, "Non-PAE Paging!\n"));\r
-      return EFI_UNSUPPORTED;\r
-    }\r
-    if ((BaseAddress + Length) > BASE_4GB) {\r
-      DEBUG ((DEBUG_ERROR, "Beyond 4GB memory in 32-bit mode!\n"));\r
+\r
+      if ((BaseAddress + Length) > BASE_4GB) {\r
+        DEBUG ((DEBUG_ERROR, "Beyond 4GB memory in 32-bit mode!\n"));\r
+        return EFI_UNSUPPORTED;\r
+      }\r
+\r
+      break;\r
+    case IMAGE_FILE_MACHINE_X64:\r
+      ASSERT (CurrentPagingContext.ContextData.X64.PageTableBase != 0);\r
+      break;\r
+    default:\r
+      ASSERT (FALSE);\r
       return EFI_UNSUPPORTED;\r
-    }\r
-    break;\r
-  case IMAGE_FILE_MACHINE_X64:\r
-    ASSERT (CurrentPagingContext.ContextData.X64.PageTableBase != 0);\r
-    break;\r
-  default:\r
-    ASSERT(FALSE);\r
-    return EFI_UNSUPPORTED;\r
-    break;\r
+      break;\r
   }\r
 \r
-//  DEBUG ((DEBUG_ERROR, "ConvertMemoryPageAttributes(%x) - %016lx, %016lx, %02lx\n", IsSet, BaseAddress, Length, Attributes));\r
+  //  DEBUG ((DEBUG_ERROR, "ConvertMemoryPageAttributes(%x) - %016lx, %016lx, %02lx\n", IsSet, BaseAddress, Length, Attributes));\r
 \r
   if (IsSplitted != NULL) {\r
     *IsSplitted = FALSE;\r
   }\r
+\r
   if (IsModified != NULL) {\r
     *IsModified = FALSE;\r
   }\r
+\r
   if (AllocatePagesFunc == NULL) {\r
     AllocatePagesFunc = AllocatePageTableMemory;\r
   }\r
@@ -785,8 +820,9 @@ ConvertMemoryPageAttributes (
       Status = RETURN_UNSUPPORTED;\r
       goto Done;\r
     }\r
+\r
     PageEntryLength = PageAttributeToLength (PageAttribute);\r
-    SplitAttribute = NeedSplitPage (BaseAddress, Length, PageEntry, PageAttribute);\r
+    SplitAttribute  = NeedSplitPage (BaseAddress, Length, PageEntry, PageAttribute);\r
     if (SplitAttribute == PageNone) {\r
       ConvertPageEntryAttribute (&CurrentPagingContext, PageEntry, Attributes, PageAction, &IsEntryModified);\r
       if (IsEntryModified) {\r
@@ -794,27 +830,32 @@ ConvertMemoryPageAttributes (
           *IsModified = TRUE;\r
         }\r
       }\r
+\r
       //\r
       // Convert success, move to next\r
       //\r
       BaseAddress += PageEntryLength;\r
-      Length -= PageEntryLength;\r
+      Length      -= PageEntryLength;\r
     } else {\r
       if (AllocatePagesFunc == NULL) {\r
         Status = RETURN_UNSUPPORTED;\r
         goto Done;\r
       }\r
+\r
       Status = SplitPage (PageEntry, PageAttribute, SplitAttribute, AllocatePagesFunc);\r
       if (RETURN_ERROR (Status)) {\r
         Status = RETURN_UNSUPPORTED;\r
         goto Done;\r
       }\r
+\r
       if (IsSplitted != NULL) {\r
         *IsSplitted = TRUE;\r
       }\r
+\r
       if (IsModified != NULL) {\r
         *IsModified = TRUE;\r
       }\r
+\r
       //\r
       // Just split current page\r
       // Convert success in next around\r
@@ -829,6 +870,7 @@ Done:
   if (IsWpEnabled) {\r
     EnableReadOnlyPageWriteProtect ();\r
   }\r
+\r
   return Status;\r
 }\r
 \r
@@ -863,20 +905,20 @@ Done:
 RETURN_STATUS\r
 EFIAPI\r
 AssignMemoryPageAttributes (\r
-  IN  PAGE_TABLE_LIB_PAGING_CONTEXT     *PagingContext OPTIONAL,\r
-  IN  PHYSICAL_ADDRESS                  BaseAddress,\r
-  IN  UINT64                            Length,\r
-  IN  UINT64                            Attributes,\r
-  IN  PAGE_TABLE_LIB_ALLOCATE_PAGES     AllocatePagesFunc OPTIONAL\r
+  IN  PAGE_TABLE_LIB_PAGING_CONTEXT  *PagingContext OPTIONAL,\r
+  IN  PHYSICAL_ADDRESS               BaseAddress,\r
+  IN  UINT64                         Length,\r
+  IN  UINT64                         Attributes,\r
+  IN  PAGE_TABLE_LIB_ALLOCATE_PAGES  AllocatePagesFunc OPTIONAL\r
   )\r
 {\r
   RETURN_STATUS  Status;\r
   BOOLEAN        IsModified;\r
   BOOLEAN        IsSplitted;\r
 \r
-//  DEBUG((DEBUG_INFO, "AssignMemoryPageAttributes: 0x%lx - 0x%lx (0x%lx)\n", BaseAddress, Length, Attributes));\r
+  //  DEBUG((DEBUG_INFO, "AssignMemoryPageAttributes: 0x%lx - 0x%lx (0x%lx)\n", BaseAddress, Length, Attributes));\r
   Status = ConvertMemoryPageAttributes (PagingContext, BaseAddress, Length, Attributes, PageActionAssign, AllocatePagesFunc, &IsSplitted, &IsModified);\r
-  if (!EFI_ERROR(Status)) {\r
+  if (!EFI_ERROR (Status)) {\r
     if ((PagingContext == NULL) && IsModified) {\r
       //\r
       // Flush TLB as last step.\r
@@ -885,7 +927,7 @@ AssignMemoryPageAttributes (
       // TLB flush in MWAIT loop mode, there's no need to flush TLB for them\r
       // here.\r
       //\r
-      CpuFlushTlb();\r
+      CpuFlushTlb ();\r
     }\r
   }\r
 \r
@@ -900,7 +942,7 @@ IsExecuteDisableEnabled (
   VOID\r
   )\r
 {\r
-  MSR_CORE_IA32_EFER_REGISTER    MsrEfer;\r
+  MSR_CORE_IA32_EFER_REGISTER  MsrEfer;\r
 \r
   MsrEfer.Uint64 = AsmReadMsr64 (MSR_IA32_EFER);\r
   return (MsrEfer.Bits.NXE == 1);\r
@@ -914,21 +956,21 @@ RefreshGcdMemoryAttributesFromPaging (
   VOID\r
   )\r
 {\r
-  EFI_STATUS                          Status;\r
-  UINTN                               NumberOfDescriptors;\r
-  EFI_GCD_MEMORY_SPACE_DESCRIPTOR     *MemorySpaceMap;\r
-  PAGE_TABLE_LIB_PAGING_CONTEXT       PagingContext;\r
-  PAGE_ATTRIBUTE                      PageAttribute;\r
-  UINT64                              *PageEntry;\r
-  UINT64                              PageLength;\r
-  UINT64                              MemorySpaceLength;\r
-  UINT64                              Length;\r
-  UINT64                              BaseAddress;\r
-  UINT64                              PageStartAddress;\r
-  UINT64                              Attributes;\r
-  UINT64                              Capabilities;\r
-  UINT64                              NewAttributes;\r
-  UINTN                               Index;\r
+  EFI_STATUS                       Status;\r
+  UINTN                            NumberOfDescriptors;\r
+  EFI_GCD_MEMORY_SPACE_DESCRIPTOR  *MemorySpaceMap;\r
+  PAGE_TABLE_LIB_PAGING_CONTEXT    PagingContext;\r
+  PAGE_ATTRIBUTE                   PageAttribute;\r
+  UINT64                           *PageEntry;\r
+  UINT64                           PageLength;\r
+  UINT64                           MemorySpaceLength;\r
+  UINT64                           Length;\r
+  UINT64                           BaseAddress;\r
+  UINT64                           PageStartAddress;\r
+  UINT64                           Attributes;\r
+  UINT64                           Capabilities;\r
+  UINT64                           NewAttributes;\r
+  UINTN                            Index;\r
 \r
   //\r
   // Assuming that memory space map returned is sorted already; otherwise sort\r
@@ -939,10 +981,10 @@ RefreshGcdMemoryAttributesFromPaging (
 \r
   GetCurrentPagingContext (&PagingContext);\r
 \r
-  Attributes      = 0;\r
-  NewAttributes   = 0;\r
-  BaseAddress     = 0;\r
-  PageLength      = 0;\r
+  Attributes    = 0;\r
+  NewAttributes = 0;\r
+  BaseAddress   = 0;\r
+  PageLength    = 0;\r
 \r
   if (IsExecuteDisableEnabled ()) {\r
     Capabilities = EFI_MEMORY_RO | EFI_MEMORY_RP | EFI_MEMORY_XP;\r
@@ -975,7 +1017,8 @@ RefreshGcdMemoryAttributesFromPaging (
       DEBUG ((\r
         DEBUG_WARN,\r
         "Failed to update capability: [%lu] %016lx - %016lx (%016lx -> %016lx)\r\n",\r
-        (UINT64)Index, MemorySpaceMap[Index].BaseAddress,\r
+        (UINT64)Index,\r
+        MemorySpaceMap[Index].BaseAddress,\r
         MemorySpaceMap[Index].BaseAddress + MemorySpaceMap[Index].Length - 1,\r
         MemorySpaceMap[Index].Capabilities,\r
         MemorySpaceMap[Index].Capabilities | Capabilities\r
@@ -1011,14 +1054,15 @@ RefreshGcdMemoryAttributesFromPaging (
         //\r
         // Note current memory space might start in the middle of a page\r
         //\r
-        PageStartAddress  = (*PageEntry) & (UINT64)PageAttributeToMask(PageAttribute);\r
-        PageLength        = PageAttributeToLength (PageAttribute) - (BaseAddress - PageStartAddress);\r
-        Attributes        = GetAttributesFromPageEntry (PageEntry);\r
+        PageStartAddress = (*PageEntry) & (UINT64)PageAttributeToMask (PageAttribute);\r
+        PageLength       = PageAttributeToLength (PageAttribute) - (BaseAddress - PageStartAddress);\r
+        Attributes       = GetAttributesFromPageEntry (PageEntry);\r
       }\r
 \r
       Length = MIN (PageLength, MemorySpaceLength);\r
       if (Attributes != (MemorySpaceMap[Index].Attributes &\r
-                         EFI_MEMORY_ATTRIBUTE_MASK)) {\r
+                         EFI_MEMORY_ATTRIBUTE_MASK))\r
+      {\r
         NewAttributes = (MemorySpaceMap[Index].Attributes &\r
                          ~EFI_MEMORY_ATTRIBUTE_MASK) | Attributes;\r
         Status = gDS->SetMemorySpaceAttributes (\r
@@ -1030,7 +1074,9 @@ RefreshGcdMemoryAttributesFromPaging (
         DEBUG ((\r
           DEBUG_VERBOSE,\r
           "Updated memory space attribute: [%lu] %016lx - %016lx (%016lx -> %016lx)\r\n",\r
-          (UINT64)Index, BaseAddress, BaseAddress + Length - 1,\r
+          (UINT64)Index,\r
+          BaseAddress,\r
+          BaseAddress + Length - 1,\r
           MemorySpaceMap[Index].Attributes,\r
           NewAttributes\r
           ));\r
@@ -1064,11 +1110,11 @@ RefreshGcdMemoryAttributesFromPaging (
 **/\r
 BOOLEAN\r
 InitializePageTablePool (\r
-  IN  UINTN                           PoolPages\r
+  IN  UINTN  PoolPages\r
   )\r
 {\r
-  VOID                      *Buffer;\r
-  BOOLEAN                   IsModified;\r
+  VOID     *Buffer;\r
+  BOOLEAN  IsModified;\r
 \r
   //\r
   // Do not allow re-entrance.\r
@@ -1078,15 +1124,15 @@ InitializePageTablePool (
   }\r
 \r
   mPageTablePoolLock = TRUE;\r
-  IsModified = FALSE;\r
+  IsModified         = FALSE;\r
 \r
   //\r
   // Always reserve at least PAGE_TABLE_POOL_UNIT_PAGES, including one page for\r
   // header.\r
   //\r
   PoolPages += 1;   // Add one page for header.\r
-  PoolPages = ((PoolPages - 1) / PAGE_TABLE_POOL_UNIT_PAGES + 1) *\r
-              PAGE_TABLE_POOL_UNIT_PAGES;\r
+  PoolPages  = ((PoolPages - 1) / PAGE_TABLE_POOL_UNIT_PAGES + 1) *\r
+               PAGE_TABLE_POOL_UNIT_PAGES;\r
   Buffer = AllocateAlignedPages (PoolPages, PAGE_TABLE_POOL_ALIGNMENT);\r
   if (Buffer == NULL) {\r
     DEBUG ((DEBUG_ERROR, "ERROR: Out of aligned pages\r\n"));\r
@@ -1103,19 +1149,19 @@ InitializePageTablePool (
   // Link all pools into a list for easier track later.\r
   //\r
   if (mPageTablePool == NULL) {\r
-    mPageTablePool = Buffer;\r
+    mPageTablePool           = Buffer;\r
     mPageTablePool->NextPool = mPageTablePool;\r
   } else {\r
     ((PAGE_TABLE_POOL *)Buffer)->NextPool = mPageTablePool->NextPool;\r
-    mPageTablePool->NextPool = Buffer;\r
-    mPageTablePool = Buffer;\r
+    mPageTablePool->NextPool              = Buffer;\r
+    mPageTablePool                        = Buffer;\r
   }\r
 \r
   //\r
   // Reserve one page for pool header.\r
   //\r
-  mPageTablePool->FreePages  = PoolPages - 1;\r
-  mPageTablePool->Offset = EFI_PAGES_TO_SIZE (1);\r
+  mPageTablePool->FreePages = PoolPages - 1;\r
+  mPageTablePool->Offset    = EFI_PAGES_TO_SIZE (1);\r
 \r
   //\r
   // Mark the whole pool pages as read-only.\r
@@ -1157,10 +1203,10 @@ Done:
 VOID *\r
 EFIAPI\r
 AllocatePageTableMemory (\r
-  IN UINTN           Pages\r
+  IN UINTN  Pages\r
   )\r
 {\r
-  VOID                            *Buffer;\r
+  VOID  *Buffer;\r
 \r
   if (Pages == 0) {\r
     return NULL;\r
@@ -1169,8 +1215,9 @@ AllocatePageTableMemory (
   //\r
   // Renew the pool if necessary.\r
   //\r
-  if (mPageTablePool == NULL ||\r
-      Pages > mPageTablePool->FreePages) {\r
+  if ((mPageTablePool == NULL) ||\r
+      (Pages > mPageTablePool->FreePages))\r
+  {\r
     if (!InitializePageTablePool (Pages)) {\r
       return NULL;\r
     }\r
@@ -1178,8 +1225,8 @@ AllocatePageTableMemory (
 \r
   Buffer = (UINT8 *)mPageTablePool + mPageTablePool->Offset;\r
 \r
-  mPageTablePool->Offset     += EFI_PAGES_TO_SIZE (Pages);\r
-  mPageTablePool->FreePages  -= Pages;\r
+  mPageTablePool->Offset    += EFI_PAGES_TO_SIZE (Pages);\r
+  mPageTablePool->FreePages -= Pages;\r
 \r
   return Buffer;\r
 }\r
@@ -1196,13 +1243,13 @@ AllocatePageTableMemory (
 VOID\r
 EFIAPI\r
 DebugExceptionHandler (\r
-  IN EFI_EXCEPTION_TYPE   ExceptionType,\r
-  IN EFI_SYSTEM_CONTEXT   SystemContext\r
+  IN EFI_EXCEPTION_TYPE  ExceptionType,\r
+  IN EFI_SYSTEM_CONTEXT  SystemContext\r
   )\r
 {\r
-  UINTN     CpuIndex;\r
-  UINTN     PFEntry;\r
-  BOOLEAN   IsWpEnabled;\r
+  UINTN    CpuIndex;\r
+  UINTN    PFEntry;\r
+  BOOLEAN  IsWpEnabled;\r
 \r
   MpInitLibWhoAmI (&CpuIndex);\r
 \r
@@ -1238,9 +1285,9 @@ DebugExceptionHandler (
   // Clear TF in EFLAGS\r
   //\r
   if (mPagingContext.MachineType == IMAGE_FILE_MACHINE_I386) {\r
-    SystemContext.SystemContextIa32->Eflags &= (UINT32)~BIT8;\r
+    SystemContext.SystemContextIa32->Eflags &= (UINT32) ~BIT8;\r
   } else {\r
-    SystemContext.SystemContextX64->Rflags &= (UINT64)~BIT8;\r
+    SystemContext.SystemContextX64->Rflags &= (UINT64) ~BIT8;\r
   }\r
 }\r
 \r
@@ -1256,20 +1303,20 @@ DebugExceptionHandler (
 VOID\r
 EFIAPI\r
 PageFaultExceptionHandler (\r
-  IN EFI_EXCEPTION_TYPE   ExceptionType,\r
-  IN EFI_SYSTEM_CONTEXT   SystemContext\r
+  IN EFI_EXCEPTION_TYPE  ExceptionType,\r
+  IN EFI_SYSTEM_CONTEXT  SystemContext\r
   )\r
 {\r
-  EFI_STATUS                      Status;\r
-  UINT64                          PFAddress;\r
-  PAGE_TABLE_LIB_PAGING_CONTEXT   PagingContext;\r
-  PAGE_ATTRIBUTE                  PageAttribute;\r
-  UINT64                          Attributes;\r
-  UINT64                          *PageEntry;\r
-  UINTN                           Index;\r
-  UINTN                           CpuIndex;\r
-  UINTN                           PageNumber;\r
-  BOOLEAN                         NonStopMode;\r
+  EFI_STATUS                     Status;\r
+  UINT64                         PFAddress;\r
+  PAGE_TABLE_LIB_PAGING_CONTEXT  PagingContext;\r
+  PAGE_ATTRIBUTE                 PageAttribute;\r
+  UINT64                         Attributes;\r
+  UINT64                         *PageEntry;\r
+  UINTN                          Index;\r
+  UINTN                          CpuIndex;\r
+  UINTN                          PageNumber;\r
+  BOOLEAN                        NonStopMode;\r
 \r
   PFAddress = AsmReadCr2 () & ~EFI_PAGE_MASK;\r
   if (PFAddress < BASE_4KB) {\r
@@ -1289,23 +1336,28 @@ PageFaultExceptionHandler (
     PageNumber = 2;\r
     while (PageNumber > 0) {\r
       PageEntry = GetPageTableEntry (&PagingContext, PFAddress, &PageAttribute);\r
-      ASSERT(PageEntry != NULL);\r
+      ASSERT (PageEntry != NULL);\r
 \r
       if (PageEntry != NULL) {\r
         Attributes = GetAttributesFromPageEntry (PageEntry);\r
         if ((Attributes & EFI_MEMORY_RP) != 0) {\r
           Attributes &= ~EFI_MEMORY_RP;\r
-          Status = AssignMemoryPageAttributes (&PagingContext, PFAddress,\r
-                                               EFI_PAGE_SIZE, Attributes, NULL);\r
-          if (!EFI_ERROR(Status)) {\r
+          Status      = AssignMemoryPageAttributes (\r
+                          &PagingContext,\r
+                          PFAddress,\r
+                          EFI_PAGE_SIZE,\r
+                          Attributes,\r
+                          NULL\r
+                          );\r
+          if (!EFI_ERROR (Status)) {\r
             Index = mPFEntryCount[CpuIndex];\r
             //\r
             // Re-retrieve page entry because above calling might update page\r
             // table due to table split.\r
             //\r
-            PageEntry = GetPageTableEntry (&PagingContext, PFAddress, &PageAttribute);\r
+            PageEntry                              = GetPageTableEntry (&PagingContext, PFAddress, &PageAttribute);\r
             mLastPFEntryPointer[CpuIndex][Index++] = PageEntry;\r
-            mPFEntryCount[CpuIndex] = Index;\r
+            mPFEntryCount[CpuIndex]                = Index;\r
           }\r
         }\r
       }\r
@@ -1345,9 +1397,9 @@ InitializePageTableLib (
   VOID\r
   )\r
 {\r
-  PAGE_TABLE_LIB_PAGING_CONTEXT     CurrentPagingContext;\r
-  UINT32                            *Attributes;\r
-  UINTN                             *PageTableBase;\r
+  PAGE_TABLE_LIB_PAGING_CONTEXT  CurrentPagingContext;\r
+  UINT32                         *Attributes;\r
+  UINTN                          *PageTableBase;\r
 \r
   GetCurrentPagingContext (&CurrentPagingContext);\r
 \r
@@ -1357,7 +1409,8 @@ InitializePageTableLib (
   // Reserve memory of page tables for future uses, if paging is enabled.\r
   //\r
   if ((*PageTableBase != 0) &&\r
-      (*Attributes & PAGE_TABLE_LIB_PAGING_CONTEXT_IA32_X64_ATTRIBUTES_PAE) != 0) {\r
+      ((*Attributes & PAGE_TABLE_LIB_PAGING_CONTEXT_IA32_X64_ATTRIBUTES_PAE) != 0))\r
+  {\r
     DisableReadOnlyPageWriteProtect ();\r
     InitializePageTablePool (1);\r
     EnableReadOnlyPageWriteProtect ();\r
@@ -1377,5 +1430,5 @@ InitializePageTableLib (
   DEBUG ((DEBUG_INFO, "  PageTableBase - 0x%Lx\n", (UINT64)*PageTableBase));\r
   DEBUG ((DEBUG_INFO, "  Attributes    - 0x%x\n", *Attributes));\r
 \r
-  return ;\r
+  return;\r
 }\r