SecurityPkg/RngDxe: Check before advertising Cpu Rng algo

[mirror_edk2.git] / SecurityPkg / RandomNumberGenerator / RngDxe / RngDxeInternals.h

diff --git a/SecurityPkg/RandomNumberGenerator/RngDxe/RngDxeInternals.h b/SecurityPkg/RandomNumberGenerator/RngDxe/RngDxeInternals.h
index 7ecab140483def8505b34dd2d8998441b8ade430..f75140260820ab16e29e96ab5a6fe4f1f3643112 100644 (file)
--- a/SecurityPkg/RandomNumberGenerator/RngDxe/RngDxeInternals.h
+++ b/SecurityPkg/RandomNumberGenerator/RngDxe/RngDxeInternals.h
@@ -12,6 +12,33 @@
 \r
 #include <Protocol/Rng.h>\r
 \r
+//\r
+// Array containing the validated Rng algorithm.\r
+// The entry with the lowest index will be the default algorithm.\r
+//\r
+extern UINTN              mAvailableAlgoArrayCount;\r
+extern EFI_RNG_ALGORITHM  *mAvailableAlgoArray;\r
+\r
+/** Allocate and initialize mAvailableAlgoArray with the available\r
+    Rng algorithms. Also update mAvailableAlgoArrayCount.\r
+\r
+  @retval EFI_SUCCESS             The function completed successfully.\r
+  @retval EFI_OUT_OF_RESOURCES    Could not allocate memory.\r
+**/\r
+EFI_STATUS\r
+EFIAPI\r
+GetAvailableAlgorithms (\r
+  VOID\r
+  );\r
+\r
+/** Free mAvailableAlgoArray.\r
+**/\r
+VOID\r
+EFIAPI\r
+FreeAvailableAlgorithms (\r
+  VOID\r
+  );\r
+\r
 /**\r
   Returns information about the random number generation implementation.\r
 \r