1. Add more error handling code to DxeImageVerificationLib and BaseCryptLib.

[mirror_edk2.git] / CryptoPkg / Library / BaseCryptLib / Pem / CryptPem.c

/** @file\r
  PEM (Privacy Enhanced Mail) Format Handler Wrapper Implementation over OpenSSL.\r
\r
Copyright (c) 2010 - 2011, 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 "InternalCryptLib.h"\r
#include <openssl/pem.h>\r
\r
/**\r
  Callback function for password phrase conversion used for retrieving the encrypted PEM.\r
\r
  @param[out]  Buf      Pointer to the buffer to write the passphrase to.\r
  @param[in]   Size     Maximum length of the passphrase (i.e. the size of Buf).\r
  @param[in]   Flag     A flag which is set to 0 when reading and 1 when writing.\r
  @param[in]   Key      Key data to be passed to the callback routine.\r
\r
  @retval  The number of characters in the passphrase or 0 if an error occurred.\r
\r
**/\r
INTN\r
PasswordCallback (\r
  OUT  CHAR8  *Buf, \r
  IN   INTN   Size, \r
  IN   INTN   Flag, \r
  IN   VOID   *Key\r
  )\r
{\r
  INTN  KeyLength;\r
\r
  ZeroMem ((VOID *)Buf, (UINTN)Size);\r
  if (Key != NULL) {\r
    //\r
    // Duplicate key phrase directly.\r
    //\r
    KeyLength = (INTN) AsciiStrLen ((CHAR8 *)Key);\r
    KeyLength = (KeyLength > Size ) ? Size : KeyLength;\r
    CopyMem (Buf, Key, (UINTN) KeyLength);\r
    return KeyLength;\r
  } else {\r
    return 0;\r
  }\r
}\r
\r
/**\r
  Retrieve the RSA Private Key from the password-protected PEM key data.\r
\r
  @param[in]  PemData      Pointer to the PEM-encoded key data to be retrieved.\r
  @param[in]  PemSize      Size of the PEM key data in bytes.\r
  @param[in]  Password     NULL-terminated passphrase used for encrypted PEM key data.\r
  @param[out] RsaContext   Pointer to new-generated RSA context which contain the retrieved\r
                           RSA private key component. Use RsaFree() function to free the\r
                           resource.\r
\r
  If PemData is NULL, then ASSERT().\r
  If RsaContext is NULL, then ASSERT().\r
\r
  @retval  TRUE   RSA Private Key was retrieved successfully.\r
  @retval  FALSE  Invalid PEM key data or incorrect password.\r
\r
**/\r
BOOLEAN\r
EFIAPI\r
RsaGetPrivateKeyFromPem (\r
  IN   CONST UINT8  *PemData,\r
  IN   UINTN        PemSize,\r
  IN   CONST CHAR8  *Password,\r
  OUT  VOID         **RsaContext\r
  )\r
{\r
  BOOLEAN  Status;\r
  BIO      *PemBio;\r
\r
  //\r
  // ASSERT if PemData is NULL or RsaContext is NULL.\r
  //\r
  ASSERT (PemData    != NULL);\r
  ASSERT (RsaContext != NULL);\r
  ASSERT (PemSize    <= INT_MAX);\r
\r
  Status = FALSE;\r
  PemBio = NULL;\r
\r
  //\r
  // Add possible block-cipher descriptor for PEM data decryption.\r
  // NOTE: Only support most popular ciphers (3DES, AES) for the encrypted PEM.\r
  //\r
  EVP_add_cipher (EVP_des_ede3_cbc());\r
  EVP_add_cipher (EVP_aes_128_cbc());\r
  EVP_add_cipher (EVP_aes_192_cbc());\r
  EVP_add_cipher (EVP_aes_256_cbc());\r
\r
  //\r
  // Read encrypted PEM Data.\r
  //\r
  PemBio = BIO_new (BIO_s_mem ());\r
  BIO_write (PemBio, PemData, (int)PemSize);\r
  if (PemBio == NULL) {\r
    goto _Exit;\r
  }\r
\r
  //\r
  // Retrieve RSA Private Key from encrypted PEM data.\r
  //\r
  *RsaContext = PEM_read_bio_RSAPrivateKey (PemBio, NULL, (pem_password_cb *)&PasswordCallback, (void *)Password);\r
  if (*RsaContext != NULL) {\r
    Status = TRUE;\r
  }\r
\r
_Exit:\r
  //\r
  // Release Resources.\r
  //\r
  BIO_free (PemBio);\r
\r
  return Status;\r
}\r