SecurityPkg: Clean up source files

[mirror_edk2.git] / SecurityPkg / RandomNumberGenerator / RngDxe / RngDxe.c

/** @file\r
  RNG Driver to produce the UEFI Random Number Generator protocol.\r
\r
  The driver will use the new RDRAND instruction to produce high-quality, high-performance\r
  entropy and random number.\r
\r
  RNG Algoritnms defined in UEFI 2.4:\r
   - EFI_RNG_ALGORITHM_SP800_90_CTR_256_GUID  - Supported\r
     (RDRAND implements a hardware NIST SP800-90 AES-CTR-256 based DRBG)\r
   - EFI_RNG_ALGORITHM_RAW                    - Supported\r
     (Structuring RDRAND invocation can be guaranteed as high-quality entropy source)\r
   - EFI_RNG_ALGORITHM_SP800_90_HMAC_256_GUID - Unsupported\r
   - EFI_RNG_ALGORITHM_SP800_90_HASH_256_GUID - Unsupported\r
   - EFI_RNG_ALGORITHM_X9_31_3DES_GUID        - Unsupported\r
   - EFI_RNG_ALGORITHM_X9_31_AES_GUID         - Unsupported\r
\r
Copyright (c) 2013 - 2018, Intel Corporation. All rights reserved.<BR>\r
(C) Copyright 2015 Hewlett Packard Enterprise Development LP<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 "RdRand.h"\r
\r
//\r
// Supported RNG Algorithms list by this driver.\r
//\r
EFI_RNG_ALGORITHM mSupportedRngAlgorithms[] = {\r
  EFI_RNG_ALGORITHM_SP800_90_CTR_256_GUID,\r
  EFI_RNG_ALGORITHM_RAW\r
};\r
\r
/**\r
  Returns information about the random number generation implementation.\r
\r
  @param[in]     This                 A pointer to the EFI_RNG_PROTOCOL instance.\r
  @param[in,out] RNGAlgorithmListSize On input, the size in bytes of RNGAlgorithmList.\r
                                      On output with a return code of EFI_SUCCESS, the size\r
                                      in bytes of the data returned in RNGAlgorithmList. On output\r
                                      with a return code of EFI_BUFFER_TOO_SMALL,\r
                                      the size of RNGAlgorithmList required to obtain the list.\r
  @param[out] RNGAlgorithmList        A caller-allocated memory buffer filled by the driver\r
                                      with one EFI_RNG_ALGORITHM element for each supported\r
                                      RNG algorithm. The list must not change across multiple\r
                                      calls to the same driver. The first algorithm in the list\r
                                      is the default algorithm for the driver.\r
\r
  @retval EFI_SUCCESS                 The RNG algorithm list was returned successfully.\r
  @retval EFI_UNSUPPORTED             The services is not supported by this driver.\r
  @retval EFI_DEVICE_ERROR            The list of algorithms could not be retrieved due to a\r
                                      hardware or firmware error.\r
  @retval EFI_INVALID_PARAMETER       One or more of the parameters are incorrect.\r
  @retval EFI_BUFFER_TOO_SMALL        The buffer RNGAlgorithmList is too small to hold the result.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
RngGetInfo (\r
  IN EFI_RNG_PROTOCOL             *This,\r
  IN OUT UINTN                    *RNGAlgorithmListSize,\r
  OUT EFI_RNG_ALGORITHM           *RNGAlgorithmList\r
  )\r
{\r
  EFI_STATUS    Status;\r
  UINTN         RequiredSize;\r
\r
  if ((This == NULL) || (RNGAlgorithmListSize == NULL)) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  RequiredSize = sizeof (mSupportedRngAlgorithms);\r
  if (*RNGAlgorithmListSize < RequiredSize) {\r
    Status = EFI_BUFFER_TOO_SMALL;\r
  } else {\r
    //\r
    // Return algorithm list supported by driver.\r
    //\r
    if (RNGAlgorithmList != NULL) {\r
      CopyMem (RNGAlgorithmList, mSupportedRngAlgorithms, RequiredSize);\r
      Status = EFI_SUCCESS;\r
    } else {\r
      Status = EFI_INVALID_PARAMETER;\r
    }\r
  }\r
  *RNGAlgorithmListSize = RequiredSize;\r
\r
  return Status;\r
}\r
\r
/**\r
  Produces and returns an RNG value using either the default or specified RNG algorithm.\r
\r
  @param[in]  This                    A pointer to the EFI_RNG_PROTOCOL instance.\r
  @param[in]  RNGAlgorithm            A pointer to the EFI_RNG_ALGORITHM that identifies the RNG\r
                                      algorithm to use. May be NULL in which case the function will\r
                                      use its default RNG algorithm.\r
  @param[in]  RNGValueLength          The length in bytes of the memory buffer pointed to by\r
                                      RNGValue. The driver shall return exactly this numbers of bytes.\r
  @param[out] RNGValue                A caller-allocated memory buffer filled by the driver with the\r
                                      resulting RNG value.\r
\r
  @retval EFI_SUCCESS                 The RNG value was returned successfully.\r
  @retval EFI_UNSUPPORTED             The algorithm specified by RNGAlgorithm is not supported by\r
                                      this driver.\r
  @retval EFI_DEVICE_ERROR            An RNG value could not be retrieved due to a hardware or\r
                                      firmware error.\r
  @retval EFI_NOT_READY               There is not enough random data available to satisfy the length\r
                                      requested by RNGValueLength.\r
  @retval EFI_INVALID_PARAMETER       RNGValue is NULL or RNGValueLength is zero.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
RngGetRNG (\r
  IN EFI_RNG_PROTOCOL            *This,\r
  IN EFI_RNG_ALGORITHM           *RNGAlgorithm, OPTIONAL\r
  IN UINTN                       RNGValueLength,\r
  OUT UINT8                      *RNGValue\r
  )\r
{\r
  EFI_STATUS    Status;\r
\r
  if ((RNGValueLength == 0) || (RNGValue == NULL)) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  Status = EFI_UNSUPPORTED;\r
  if (RNGAlgorithm == NULL) {\r
    //\r
    // Use the default RNG algorithm if RNGAlgorithm is NULL.\r
    //\r
    RNGAlgorithm = &gEfiRngAlgorithmSp80090Ctr256Guid;\r
  }\r
\r
  //\r
  // NIST SP800-90-AES-CTR-256 supported by RDRAND\r
  //\r
  if (CompareGuid (RNGAlgorithm, &gEfiRngAlgorithmSp80090Ctr256Guid)) {\r
    Status = RdRandGetBytes (RNGValueLength, RNGValue);\r
    return Status;\r
  }\r
\r
  //\r
  // The "raw" algorithm is intended to provide entropy directly\r
  //\r
  if (CompareGuid (RNGAlgorithm, &gEfiRngAlgorithmRaw)) {\r
    //\r
    // When a DRBG is used on the output of a entropy source,\r
    // its security level must be at least 256 bits according to UEFI Spec.\r
    //\r
    if (RNGValueLength < 32) {\r
      return EFI_INVALID_PARAMETER;\r
    }\r
\r
    Status = RdRandGenerateEntropy (RNGValueLength, RNGValue);\r
    return Status;\r
  }\r
\r
  //\r
  // Other algorithms were unsupported by this driver.\r
  //\r
  return Status;\r
}\r
\r
//\r
// The Random Number Generator (RNG) protocol\r
//\r
EFI_RNG_PROTOCOL mRngRdRand = {\r
  RngGetInfo,\r
  RngGetRNG\r
};\r
\r
/**\r
  The user Entry Point for the Random Number Generator (RNG) driver.\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 EFI_NOT_SUPPORTED Platform does not support RNG.\r
  @retval Other             Some error occurs when executing this entry point.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
RngDriverEntry (\r
  IN EFI_HANDLE          ImageHandle,\r
  IN EFI_SYSTEM_TABLE    *SystemTable\r
  )\r
{\r
  EFI_STATUS    Status;\r
  EFI_HANDLE    Handle;\r
\r
  //\r
  // Install UEFI RNG (Random Number Generator) Protocol\r
  //\r
  Handle = NULL;\r
  Status = gBS->InstallMultipleProtocolInterfaces (\r
                  &Handle,\r
                  &gEfiRngProtocolGuid,\r
                  &mRngRdRand,\r
                  NULL\r
                  );\r
\r
  return Status;\r
}\r