+++ /dev/null
-/** @file\r
- UEFI RNG (Random Number Generator) Protocol test application.\r
-\r
-Copyright (c) 2013, Intel Corporation. All rights reserved.<BR>\r
-This program and the accompanying materials \r
-are licensed and made available under the terms and conditions of the BSD License \r
-which accompanies this distribution. The full text of the license may be found at \r
-http://opensource.org/licenses/bsd-license.php\r
-\r
-THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS, \r
-WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
-\r
-**/ \r
-\r
-#include <Uefi.h>\r
-#include <Library/UefiLib.h>\r
-#include <Library/UefiApplicationEntryPoint.h>\r
-#include <Library/UefiBootServicesTableLib.h>\r
-#include <Library/MemoryAllocationLib.h>\r
-#include <Library/DebugLib.h>\r
-#include <Protocol/Rng.h>\r
-\r
-/**\r
- The user Entry Point for Application. The user code starts with this function\r
- as the real entry point for the application.\r
-\r
- @param[in] ImageHandle The firmware allocated handle for the EFI image. \r
- @param[in] SystemTable A pointer to the EFI System Table.\r
- \r
- @retval EFI_SUCCESS The entry point is executed successfully.\r
- @retval other Some error occurs when executing this entry point.\r
-\r
-**/\r
-EFI_STATUS\r
-EFIAPI\r
-UefiMain (\r
- IN EFI_HANDLE ImageHandle,\r
- IN EFI_SYSTEM_TABLE *SystemTable\r
- )\r
-{\r
- EFI_STATUS Status;\r
- EFI_RNG_PROTOCOL *Rng;\r
- UINTN RngAlgListSize;\r
- EFI_RNG_ALGORITHM RngAlgList[10];\r
- EFI_RNG_ALGORITHM *PtrRngAlg;\r
- UINTN RngAlgCount;\r
- UINT8 *Rand;\r
- UINTN RandSize;\r
- UINTN Index;\r
- UINTN Index2;\r
-\r
- Status = EFI_SUCCESS;\r
- PtrRngAlg = NULL;\r
- Rand = NULL;\r
- \r
- Print (L"UEFI RNG Protocol Testing :\n");\r
- Print (L"----------------------------\n");\r
-\r
- //-----------------------------------------\r
- // Basic UEFI RNG Protocol Test\r
- //-----------------------------------------\r
- Print (L" -- Locate UEFI RNG Protocol : ");\r
- Status = gBS->LocateProtocol (&gEfiRngProtocolGuid, NULL, (VOID **)&Rng);\r
- if (EFI_ERROR (Status)) {\r
- Print (L"[Fail - Status = %r]\n", Status);\r
- goto Exit;\r
- } else {\r
- Print (L"[Pass]\n");\r
- }\r
-\r
- //-----------------------------------------\r
- // Rng->GetInfo() interface test.\r
- //-----------------------------------------\r
- \r
- Print (L" -- Call RNG->GetInfo() interface : ");\r
- RngAlgListSize = 0;\r
- Status = Rng->GetInfo (Rng, &RngAlgListSize, NULL);\r
- if (Status != EFI_BUFFER_TOO_SMALL) {\r
- Print (L"[Fail - Status = %r]\n", Status);\r
- }\r
- //\r
- // Print out the supported RNG algorithm GUIDs\r
- //\r
- RngAlgCount = RngAlgListSize / sizeof (EFI_RNG_ALGORITHM);\r
- Print (L"\n >> Supported RNG Algorithm (Count = %d) : ", RngAlgCount);\r
- Status = Rng->GetInfo (Rng, &RngAlgListSize, RngAlgList);\r
- for (Index = 0; Index < RngAlgCount; Index++) {\r
- PtrRngAlg = (EFI_RNG_ALGORITHM *)(&RngAlgList[Index]);\r
- Print (L"\n %d) ", Index);\r
- Print (L"%08x-%04x-%04x-%02x%02x-%02x%02x%02x%02x%02x%02x", PtrRngAlg->Data1,\r
- PtrRngAlg->Data2, PtrRngAlg->Data3, PtrRngAlg->Data4[0], PtrRngAlg->Data4[1],\r
- PtrRngAlg->Data4[2], PtrRngAlg->Data4[3], PtrRngAlg->Data4[4],\r
- PtrRngAlg->Data4[5], PtrRngAlg->Data4[6], PtrRngAlg->Data4[7]); \r
- }\r
-\r
- //-----------------------------------------\r
- // Rng->GetRNG() interface test.\r
- //-----------------------------------------\r
- Print (L"\n -- Call RNG->GetRNG() interface : ");\r
-\r
- //\r
- // Allocate one buffer to store random data.\r
- //\r
- RandSize = 32;\r
- Rand = AllocatePool (RandSize);\r
- if (Rand == NULL) {\r
- goto Exit;\r
- }\r
- \r
- //\r
- // RNG with default algorithm\r
- //\r
- Print (L"\n >> RNG with default algorithm : ");\r
- Status = Rng->GetRNG (Rng, NULL, RandSize, Rand);\r
- if (EFI_ERROR (Status)) {\r
- Print (L"[Fail - Status = %r]", Status);\r
- } else {\r
- Print (L"[Pass]");\r
- }\r
- \r
- //\r
- // RNG with SP800-90-HMAC-256\r
- //\r
- Print (L"\n >> RNG with SP800-90-HMAC-256 : ");\r
- Status = Rng->GetRNG (Rng, &gEfiRngAlgorithmSp80090Hmac256Guid, RandSize, Rand);\r
- if (EFI_ERROR (Status)) {\r
- Print (L"[Fail - Status = %r]", Status);\r
- } else {\r
- Print (L"[Pass]");\r
- }\r
-\r
- //\r
- // RNG with SP800-90-HASH-256\r
- //\r
- Print (L"\n >> RNG with SP800-90-Hash-256 : ");\r
- Status = Rng->GetRNG (Rng, &gEfiRngAlgorithmSp80090Hash256Guid, RandSize, Rand);\r
- if (EFI_ERROR (Status)) {\r
- Print (L"[Fail - Status = %r]", Status);\r
- } else {\r
- Print (L"[Pass]");\r
- }\r
-\r
- //\r
- // RNG with SP800-90-CTR-256\r
- //\r
- Print (L"\n >> RNG with SP800-90-CTR-256 : ");\r
- Status = Rng->GetRNG (Rng, &gEfiRngAlgorithmSp80090Ctr256Guid, RandSize, Rand);\r
- if (EFI_ERROR (Status)) {\r
- Print (L"[Fail - Status = %r]", Status);\r
- } else {\r
- Print (L"[Pass]");\r
- }\r
-\r
- //\r
- // RNG with X9.31-3DES\r
- //\r
- Print (L"\n >> RNG with X9.31-3DES : ");\r
- Status = Rng->GetRNG (Rng, &gEfiRngAlgorithmX9313DesGuid, RandSize, Rand);\r
- if (EFI_ERROR (Status)) {\r
- Print (L"[Fail - Status = %r]", Status);\r
- } else {\r
- Print (L"[Pass]");\r
- }\r
-\r
- //\r
- // RNG with X9.31-AES\r
- //\r
- Print (L"\n >> RNG with X9.31-AES : ");\r
- Status = Rng->GetRNG (Rng, &gEfiRngAlgorithmX931AesGuid, RandSize, Rand);\r
- if (EFI_ERROR (Status)) {\r
- Print (L"[Fail - Status = %r]", Status);\r
- } else {\r
- Print (L"[Pass]");\r
- }\r
-\r
- //\r
- // RNG with RAW Entropy\r
- //\r
- Print (L"\n >> RNG with RAW Entropy : ");\r
- Status = Rng->GetRNG (Rng, &gEfiRngAlgorithmRaw, RandSize, Rand);\r
- if (EFI_ERROR (Status)) {\r
- Print (L"[Fail - Status = %r]", Status);\r
- } else {\r
- Print (L"[Pass]");\r
- }\r
-\r
- //-----------------------------------------\r
- // Random Number Generator test.\r
- //-----------------------------------------\r
- Print (L"\n -- Random Number Generation Test with default RNG Algorithm (20 Rounds): ");\r
-\r
- RandSize = 1;\r
- for (Index = 0; Index < 20; Index++) {\r
- Status = Rng->GetRNG (Rng, NULL, RandSize, Rand);\r
- if (EFI_ERROR (Status)) {\r
- Print (L"[Fail - Status = %r]", Status);\r
- break;\r
- } else {\r
- Print (L"\n %02d) - ", Index + 1);\r
- for (Index2 = 0; Index2 < RandSize; Index2++) {\r
- Print (L"%02x", Rand[Index2]);\r
- }\r
- }\r
-\r
- RandSize +=1;\r
- }\r
-\r
- //-----------------------------------------\r
- // Random Number Generator test.\r
- //-----------------------------------------\r
- Print (L"\n -- RAW Entropy Generation Test (20 Rounds) : ");\r
-\r
- RandSize = 32;\r
- for (Index = 0; Index < 20; Index++) {\r
- Status = Rng->GetRNG (Rng, &gEfiRngAlgorithmRaw, RandSize, Rand);\r
- if (EFI_ERROR (Status)) {\r
- Print (L"[Fail - Status = %r]", Status);\r
- break;\r
- } else {\r
- Print (L"\n %02d) - ", Index + 1);\r
- for (Index2 = 0; Index2 < RandSize; Index2++) {\r
- Print (L"%02x", Rand[Index2]);\r
- }\r
- }\r
- }\r
-\r
- Print (L"\n -- Exit UEFI RNG Protocol Test (Status = %r).\n", Status);\r
- \r
-Exit:\r
- if (Rand != NULL) {\r
- FreePool (Rand);\r
- }\r
- return Status;\r
-}\r