diff options
Diffstat (limited to 'EDK/MiniSetup/PasswordEncode/PasswordEncode.c')
| -rw-r--r-- | EDK/MiniSetup/PasswordEncode/PasswordEncode.c | 217 |
1 files changed, 217 insertions, 0 deletions
diff --git a/EDK/MiniSetup/PasswordEncode/PasswordEncode.c b/EDK/MiniSetup/PasswordEncode/PasswordEncode.c new file mode 100644 index 0000000..8af67e9 --- /dev/null +++ b/EDK/MiniSetup/PasswordEncode/PasswordEncode.c @@ -0,0 +1,217 @@ +//*****************************************************************// +//*****************************************************************// +//*****************************************************************// +//** **// +//** (C)Copyright 2010, American Megatrends, Inc. **// +//** **// +//** All Rights Reserved. **// +//** **// +//** 5555 Oakbrook Pkwy, Building 200,Norcross, Georgia 30093 **// +//** **// +//** Phone (770)-246-8600 **// +//** **// +//*****************************************************************// +//*****************************************************************// +//*****************************************************************// +// $Archive: /Alaska/BIN/Modules/AMITSE2_0/AMITSE/PasswordEncode/PasswordEncode.c $ +// +//<AMI_FHDR_START> +//---------------------------------------------------------------------------- +// +// Name: PasswordEncode.c +// +// Description: This file contains code to handle password encoding feature +// +//---------------------------------------------------------------------------- +//<AMI_FHDR_END> +// + + +#include "PasswordEncode.h" + +INTN MemCmp(VOID* pDestination, VOID* pSource, UINTN Length); + +BOOLEAN TseEfiCompareGuid (IN EFI_GUID *Guid1,IN EFI_GUID *Guid2) +{ + return !MemCmp(Guid1,Guid2,sizeof(EFI_GUID)); +} + +VOID +TseMemCopy ( + IN VOID *Destination, + IN VOID *Source, + IN UINTN Length + ) +/*++ + +Routine Description: + + Copy Length bytes from Source to Destination. + +Arguments: + + Destination - Target of copy + + Source - Place to copy from + + Length - Number of bytes to copy + +Returns: + + None + +--*/ +{ + CHAR8 *Destination8; + CHAR8 *Source8; + + if (Source < Destination) { + Destination8 = (CHAR8 *) Destination + Length - 1; + Source8 = (CHAR8 *) Source + Length - 1; + while (Length--) { + *(Destination8--) = *(Source8--); + } + } else { + Destination8 = (CHAR8 *) Destination; + Source8 = (CHAR8 *) Source; + while (Length--) { + *(Destination8++) = *(Source8++); + } + } +} + + +//********************************************************************** +//<AMI_PHDR_START> +// +// Procedure: Hash +// +// Description: Allows creating a hash of an arbitrary message digest using one or more hash algorithms +// +// Input: +// This Pointer to the AMI_DIGITAL_SIGNATURE_PROTOCOL instance. +// HashAlgorithm Points to the EFI_GUID which identifies the algorithm to use. +// num_elem Number of blocks to be passed via next argument:addr[] +// addr[] Pointer to array of UINT8* addresses of data blocks to be hashed +// len Pointer to array of integers containing length of each block listed by addr[] +// Hash Holds the resulting hash computed from the message. +// +// Output: +// EFI_SUCCESS Hash returned successfully. +// EFI_INVALID_PARAMETER Message or Hash is NULL +// EFI_UNSUPPORTED The algorithm specified by HashAlgorithm is not supported by this +// driver. Or extend is TRUE and the algorithm doesn't support extending the hash. +// +//<AMI_PHDR_END> +//********************************************************************** +#if TSE_HASH_PASSWORD +EFI_STATUS Hash( + IN CONST EFI_GUID *HashAlgorithm, + IN UINTN num_elem, + IN CONST UINT8 *stringToHash[], + IN CONST UINTN *HashSize, + OUT UINT8 *HashOutput + ) +{ + BOOLEAN bSha1 = FALSE, bSha256 = FALSE; + UINT32 HashLen=SHA256_DIGEST_SIZE; + + // Support only SHA1 & SHA256 hashes + if(TseEfiCompareGuid((EFI_GUID*)HashAlgorithm, &gEfiHashAlgorithmSha1Guid)) + { + bSha1 = TRUE; + HashLen = SHA1_DIGEST_SIZE; + } + else + if(TseEfiCompareGuid((EFI_GUID*)HashAlgorithm, &gEfiHashAlgorithmSha256Guid)) + { + bSha256 = TRUE; + HashLen = SHA256_DIGEST_SIZE; + } + else + return EFI_UNSUPPORTED; + + MemSet(HashOutput, HashLen, 0); + + if(bSha1) + sha1_vector(num_elem, stringToHash, HashSize, HashOutput); + else + sha256_vector(num_elem, stringToHash, HashSize, HashOutput); + + return EFI_SUCCESS; +} +#endif + +//<AMI_PHDR_START> +//---------------------------------------------------------------------------- +// Procedure: PasswordEncodeLocal +// +// Description: Encodes the input string +// +// Input: Password : Password array to be encrypted. Encryped +// password is returned in the same array. +// MaxSize : Max size of Password +// +// Output: VOID +// +//---------------------------------------------------------------------------- +//<AMI_PHDR_END> +UINT8 HashOutput[20]; +VOID TsePasswordEncodeLocal( CHAR16 *Password, UINTN MaxSize); +VOID PasswordEncode( CHAR16 *Password, UINTN MaxSize) +{ +#if TSE_HASH_PASSWORD + UINTN ii; + EFI_STATUS Status; + UINTN HashSize = SHA1_DIGEST_SIZE; + + if (IsPasswordSupportNonCaseSensitive ()) + { + for ( ii = 0; ii < MaxSize/2; ii++ ) + Password[ii] = ((Password[ii]>=L'a')&&(Password[ii]<=L'z'))?(Password[ii]+L'A'-L'a'):Password[ii]; + } + + Status = Hash(&gEfiHashAlgorithmSha1Guid, TRUE, (CONST UINT8**)&Password, (CONST UINTN*)&MaxSize, (UINT8*)&HashOutput); + if (!EFI_ERROR (Status)) + { + MemSet (Password, MaxSize, 0); + TseMemCopy ((UINT8*)Password, (UINT8*)HashOutput, HashSize); + } +#else + TsePasswordEncodeLocal (Password, MaxSize); +#endif +} + +//<AMI_PHDR_START> +//---------------------------------------------------------------------------- +// Procedure: IsPasswordSupportNonCaseSensitive +// +// Description: Returns SETUP_PASSWORD_NON_CASE_SENSITIVE token value +// +// Input: void +// +// Output: BOOLEAN +// +//---------------------------------------------------------------------------- +//<AMI_PHDR_END> +BOOLEAN IsPasswordSupportNonCaseSensitive() +{ +#if SETUP_PASSWORD_NON_CASE_SENSITIVE + return TRUE; +#endif + return FALSE; +} + +//********************************************************************** +//********************************************************************** +//** ** +//** (C)Copyright 1985-2010, American Megatrends, Inc. ** +//** ** +//** All Rights Reserved. ** +//** ** +//** 5555 Oakbrook Pkwy, Building 200,Norcross, Georgia 30093 ** +//** ** +//** Phone: (770)-246-8600 ** +//** ** +//********************************************************************** +//********************************************************************** |