REF: https://bugzilla.tianocore.org/show_bug.cgi?id=3737 Apply uncrustify changes to .c/.h files in the MdePkg package Cc: Andrew Fish <afish@apple.com> Cc: Leif Lindholm <leif@nuviainc.com> Cc: Michael D Kinney <michael.d.kinney@intel.com> Signed-off-by: Michael Kubacki <michael.kubacki@microsoft.com> Reviewed-by: Liming Gao <gaoliming@byosoft.com.cn>
		
			
				
	
	
		
			3067 lines
		
	
	
		
			80 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			3067 lines
		
	
	
		
			80 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
/** @file
 | 
						|
  UEFI OS based application for unit testing the SafeIntLib.
 | 
						|
 | 
						|
  Copyright (c) Microsoft Corporation.<BR>
 | 
						|
  Copyright (c) 2018 - 2020, Intel Corporation. All rights reserved.<BR>
 | 
						|
  SPDX-License-Identifier: BSD-2-Clause-Patent
 | 
						|
 | 
						|
**/
 | 
						|
 | 
						|
#include "TestBaseSafeIntLib.h"
 | 
						|
 | 
						|
#define UNIT_TEST_NAME     "Int Safe Lib Unit Test Application"
 | 
						|
#define UNIT_TEST_VERSION  "0.1"
 | 
						|
 | 
						|
//
 | 
						|
// Conversion function tests:
 | 
						|
//
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeInt8ToUint8 (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  INT8        Operand;
 | 
						|
  UINT8       Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // Positive UINT8 should result in just a cast
 | 
						|
  //
 | 
						|
  Operand = 0x5b;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeInt8ToUint8 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x5b, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Negative number should result in an error status
 | 
						|
  //
 | 
						|
  Operand = (-56);
 | 
						|
  Status  = SafeInt8ToUint8 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeInt8ToUint16 (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  INT8        Operand;
 | 
						|
  UINT16      Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // Positive UINT8 should result in just a cast
 | 
						|
  //
 | 
						|
  Operand = 0x5b;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeInt8ToUint16 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x5b, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Negative number should result in an error status
 | 
						|
  //
 | 
						|
  Operand = (-56);
 | 
						|
  Status  = SafeInt8ToUint16 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeInt8ToUint32 (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  INT8        Operand;
 | 
						|
  UINT32      Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // Positive UINT8 should result in just a cast
 | 
						|
  //
 | 
						|
  Operand = 0x5b;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeInt8ToUint32 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x5b, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Negative number should result in an error status
 | 
						|
  //
 | 
						|
  Operand = (-56);
 | 
						|
  Status  = SafeInt8ToUint32 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeInt8ToUintn (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  INT8        Operand;
 | 
						|
  UINTN       Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // Positive UINT8 should result in just a cast
 | 
						|
  //
 | 
						|
  Operand = 0x5b;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeInt8ToUintn (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x5b, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Negative number should result in an error status
 | 
						|
  //
 | 
						|
  Operand = (-56);
 | 
						|
  Status  = SafeInt8ToUintn (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeInt8ToUint64 (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  INT8        Operand;
 | 
						|
  UINT64      Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // Positive UINT8 should result in just a cast
 | 
						|
  //
 | 
						|
  Operand = 0x5b;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeInt8ToUint64 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x5b, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Negative number should result in an error status
 | 
						|
  //
 | 
						|
  Operand = (-56);
 | 
						|
  Status  = SafeInt8ToUint64 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeUint8ToInt8 (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  UINT8       Operand;
 | 
						|
  INT8        Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // Operand <= 0x7F (MAX_INT8) should result in a cast
 | 
						|
  //
 | 
						|
  Operand = 0x5b;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeUint8ToInt8 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x5b, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Operand larger than 0x7f should result in an error status
 | 
						|
  //
 | 
						|
  Operand = 0xaf;
 | 
						|
  Status  = SafeUint8ToInt8 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeUint8ToChar8 (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  UINT8       Operand;
 | 
						|
  CHAR8       Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // CHAR8 is typedefed as char, which by default is signed, thus
 | 
						|
  // CHAR8 is same as INT8, so same tests as above:
 | 
						|
  //
 | 
						|
 | 
						|
  //
 | 
						|
  // Operand <= 0x7F (MAX_INT8) should result in a cast
 | 
						|
  //
 | 
						|
  Operand = 0x5b;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeUint8ToChar8 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x5b, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Operand larger than 0x7f should result in an error status
 | 
						|
  //
 | 
						|
  Operand = 0xaf;
 | 
						|
  Status  = SafeUint8ToChar8 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeInt16ToInt8 (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  INT16       Operand;
 | 
						|
  INT8        Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If Operand is between MIN_INT8 and MAX_INT8 inclusive, then it's a cast
 | 
						|
  //
 | 
						|
  Operand = 0x5b;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeInt16ToInt8 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x5b, Result);
 | 
						|
 | 
						|
  Operand = (-35);
 | 
						|
  Status  = SafeInt16ToInt8 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL ((-35), Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Operand = 0x1234;
 | 
						|
  Status  = SafeInt16ToInt8 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  Operand = (-17835);
 | 
						|
  Status  = SafeInt16ToInt8 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeInt16ToChar8 (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  INT16       Operand;
 | 
						|
  CHAR8       Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // CHAR8 is typedefed as char, which may be signed or unsigned based
 | 
						|
  // on the compiler. Thus, for compatibility CHAR8 should be between 0 and MAX_INT8.
 | 
						|
  //
 | 
						|
 | 
						|
  //
 | 
						|
  // If Operand is between 0 and MAX_INT8 inclusive, then it's a cast
 | 
						|
  //
 | 
						|
  Operand = 0x5b;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeInt16ToChar8 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x5b, Result);
 | 
						|
 | 
						|
  Operand = 0;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeInt16ToChar8 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0, Result);
 | 
						|
 | 
						|
  Operand = MAX_INT8;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeInt16ToChar8 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (MAX_INT8, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Operand = (-35);
 | 
						|
  Status  = SafeInt16ToChar8 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  Operand = 0x1234;
 | 
						|
  Status  = SafeInt16ToChar8 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  Operand = (-17835);
 | 
						|
  Status  = SafeInt16ToChar8 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeInt16ToUint8 (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  INT16       Operand;
 | 
						|
  UINT8       Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If Operand is between 0 and MAX_INT8 inclusive, then it's a cast
 | 
						|
  //
 | 
						|
  Operand = 0x5b;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeInt16ToUint8 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x5b, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Operand = 0x1234;
 | 
						|
  Status  = SafeInt16ToUint8 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  Operand = (-17835);
 | 
						|
  Status  = SafeInt16ToUint8 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeInt16ToUint16 (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  INT16       Operand = 0x5b5b;
 | 
						|
  UINT16      Result  = 0;
 | 
						|
 | 
						|
  //
 | 
						|
  // If Operand is non-negative, then it's a cast
 | 
						|
  //
 | 
						|
  Status = SafeInt16ToUint16 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x5b5b, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Operand = (-17835);
 | 
						|
  Status  = SafeInt16ToUint16 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeInt16ToUint32 (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  INT16       Operand;
 | 
						|
  UINT32      Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If Operand is non-negative, then it's a cast
 | 
						|
  //
 | 
						|
  Operand = 0x5b5b;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeInt16ToUint32 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x5b5b, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Operand = (-17835);
 | 
						|
  Status  = SafeInt16ToUint32 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeInt16ToUintn (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  INT16       Operand;
 | 
						|
  UINTN       Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If Operand is non-negative, then it's a cast
 | 
						|
  //
 | 
						|
  Operand = 0x5b5b;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeInt16ToUintn (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x5b5b, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Operand = (-17835);
 | 
						|
  Status  = SafeInt16ToUintn (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeInt16ToUint64 (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  INT16       Operand;
 | 
						|
  UINT64      Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If Operand is non-negative, then it's a cast
 | 
						|
  //
 | 
						|
  Operand = 0x5b5b;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeInt16ToUint64 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x5b5b, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Operand = (-17835);
 | 
						|
  Status  = SafeInt16ToUint64 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeUint16ToInt8 (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  UINT16      Operand;
 | 
						|
  INT8        Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If Operand is <= MAX_INT8, it's a cast
 | 
						|
  //
 | 
						|
  Operand = 0x5b;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeUint16ToInt8 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x5b, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Operand = (0x5b5b);
 | 
						|
  Status  = SafeUint16ToInt8 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeUint16ToChar8 (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  UINT16      Operand;
 | 
						|
  CHAR8       Result;
 | 
						|
 | 
						|
  // CHAR8 is typedefed as char, which by default is signed, thus
 | 
						|
  // CHAR8 is same as INT8, so same tests as above:
 | 
						|
 | 
						|
  //
 | 
						|
  // If Operand is <= MAX_INT8, it's a cast
 | 
						|
  //
 | 
						|
  Operand = 0x5b;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeUint16ToChar8 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x5b, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Operand = (0x5b5b);
 | 
						|
  Status  = SafeUint16ToChar8 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeUint16ToUint8 (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  UINT16      Operand;
 | 
						|
  UINT8       Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If Operand is <= MAX_UINT8 (0xff), it's a cast
 | 
						|
  //
 | 
						|
  Operand = 0xab;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeUint16ToUint8 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0xab, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Operand = (0x5b5b);
 | 
						|
  Status  = SafeUint16ToUint8 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeUint16ToInt16 (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  UINT16      Operand;
 | 
						|
  INT16       Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If Operand is <= MAX_INT16 (0x7fff), it's a cast
 | 
						|
  //
 | 
						|
  Operand = 0x5b5b;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeUint16ToInt16 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x5b5b, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Operand = (0xabab);
 | 
						|
  Status  = SafeUint16ToInt16 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeInt32ToInt8 (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  INT32       Operand;
 | 
						|
  INT8        Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If Operand is between MIN_INT8 and MAX_INT8 inclusive, then it's a cast
 | 
						|
  //
 | 
						|
  Operand = 0x5b;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeInt32ToInt8 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x5b, Result);
 | 
						|
 | 
						|
  Operand = (-57);
 | 
						|
  Status  = SafeInt32ToInt8 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL ((-57), Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Operand = (0x5bababab);
 | 
						|
  Status  = SafeInt32ToInt8 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  Operand = (-1537977259);
 | 
						|
  Status  = SafeInt32ToInt8 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeInt32ToChar8 (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  INT32       Operand;
 | 
						|
  CHAR8       Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // CHAR8 is typedefed as char, which may be signed or unsigned based
 | 
						|
  // on the compiler. Thus, for compatibility CHAR8 should be between 0 and MAX_INT8.
 | 
						|
  //
 | 
						|
 | 
						|
  //
 | 
						|
  // If Operand is between 0 and MAX_INT8 inclusive, then it's a cast
 | 
						|
  //
 | 
						|
  Operand = 0x5b;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeInt32ToChar8 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x5b, Result);
 | 
						|
 | 
						|
  Operand = 0;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeInt32ToChar8 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0, Result);
 | 
						|
 | 
						|
  Operand = MAX_INT8;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeInt32ToChar8 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (MAX_INT8, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Operand = (-57);
 | 
						|
  Status  = SafeInt32ToChar8 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  Operand = (0x5bababab);
 | 
						|
  Status  = SafeInt32ToChar8 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  Operand = (-1537977259);
 | 
						|
  Status  = SafeInt32ToChar8 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeInt32ToUint8 (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  INT32       Operand;
 | 
						|
  UINT8       Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If Operand is between 0 and MAX_INT8 inclusive, then it's a cast
 | 
						|
  //
 | 
						|
  Operand = 0x5b;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeInt32ToUint8 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x5b, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Operand = (-57);
 | 
						|
  Status  = SafeInt32ToUint8 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  Operand = (0x5bababab);
 | 
						|
  Status  = SafeInt32ToUint8 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  Operand = (-1537977259);
 | 
						|
  Status  = SafeInt32ToUint8 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeInt32ToInt16 (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  INT32       Operand;
 | 
						|
  INT16       Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If Operand is between MIN_INT16 and MAX_INT16 inclusive, then it's a cast
 | 
						|
  //
 | 
						|
  Operand = 0x5b5b;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeInt32ToInt16 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x5b5b, Result);
 | 
						|
 | 
						|
  Operand = (-17857);
 | 
						|
  Status  = SafeInt32ToInt16 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL ((-17857), Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Operand = (0x5bababab);
 | 
						|
  Status  = SafeInt32ToInt16 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  Operand = (-1537977259);
 | 
						|
  Status  = SafeInt32ToInt16 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeInt32ToUint16 (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  INT32       Operand;
 | 
						|
  UINT16      Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If Operand is between 0 and MAX_UINT16 inclusive, then it's a cast
 | 
						|
  //
 | 
						|
  Operand = 0xabab;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeInt32ToUint16 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0xabab, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Operand = (-17857);
 | 
						|
  Status  = SafeInt32ToUint16 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  Operand = (0x5bababab);
 | 
						|
  Status  = SafeInt32ToUint16 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  Operand = (-1537977259);
 | 
						|
  Status  = SafeInt32ToUint16 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeInt32ToUint32 (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  INT32       Operand;
 | 
						|
  UINT32      Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If Operand is non-negative, then it's a cast
 | 
						|
  //
 | 
						|
  Operand = 0x5bababab;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeInt32ToUint32 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x5bababab, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Operand = (-1537977259);
 | 
						|
  Status  = SafeInt32ToUint32 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeInt32ToUint64 (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  INT32       Operand;
 | 
						|
  UINT64      Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If Operand is non-negative, then it's a cast
 | 
						|
  //
 | 
						|
  Operand = 0x5bababab;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeInt32ToUint64 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x5bababab, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Operand = (-1537977259);
 | 
						|
  Status  = SafeInt32ToUint64 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeUint32ToInt8 (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  UINT32      Operand;
 | 
						|
  INT8        Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If Operand is <= MAX_INT8, then it's a cast
 | 
						|
  //
 | 
						|
  Operand = 0x5b;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeUint32ToInt8 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x5b, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Operand = (0x5bababab);
 | 
						|
  Status  = SafeUint32ToInt8 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeUint32ToChar8 (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  UINT32      Operand;
 | 
						|
  CHAR8       Result;
 | 
						|
 | 
						|
  // CHAR8 is typedefed as char, which by default is signed, thus
 | 
						|
  // CHAR8 is same as INT8, so same tests as above:
 | 
						|
 | 
						|
  //
 | 
						|
  // If Operand is <= MAX_INT8, then it's a cast
 | 
						|
  //
 | 
						|
  Operand = 0x5b;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeUint32ToChar8 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x5b, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Operand = (0x5bababab);
 | 
						|
  Status  = SafeUint32ToChar8 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeUint32ToUint8 (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  UINT32      Operand;
 | 
						|
  UINT8       Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If Operand is <= MAX_UINT8, then it's a cast
 | 
						|
  //
 | 
						|
  Operand = 0xab;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeUint32ToUint8 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0xab, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Operand = (0xabababab);
 | 
						|
  Status  = SafeUint32ToUint8 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeUint32ToInt16 (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  UINT32      Operand;
 | 
						|
  INT16       Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If Operand is <= MAX_INT16, then it's a cast
 | 
						|
  //
 | 
						|
  Operand = 0x5bab;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeUint32ToInt16 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x5bab, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Operand = (0xabababab);
 | 
						|
  Status  = SafeUint32ToInt16 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeUint32ToUint16 (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  UINT32      Operand;
 | 
						|
  UINT16      Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If Operand is <= MAX_UINT16, then it's a cast
 | 
						|
  //
 | 
						|
  Operand = 0xabab;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeUint32ToUint16 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0xabab, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Operand = (0xabababab);
 | 
						|
  Status  = SafeUint32ToUint16 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeUint32ToInt32 (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  UINT32      Operand;
 | 
						|
  INT32       Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If Operand is <= MAX_INT32, then it's a cast
 | 
						|
  //
 | 
						|
  Operand = 0x5bababab;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeUint32ToInt32 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x5bababab, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Operand = (0xabababab);
 | 
						|
  Status  = SafeUint32ToInt32 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeIntnToInt8 (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  INTN        Operand;
 | 
						|
  INT8        Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If Operand is between MIN_INT8 and MAX_INT8 inclusive, then it's a cast
 | 
						|
  //
 | 
						|
  Operand = 0x5b;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeIntnToInt8 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x5b, Result);
 | 
						|
 | 
						|
  Operand = (-53);
 | 
						|
  Status  = SafeIntnToInt8 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL ((-53), Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Operand = (0x5bababab);
 | 
						|
  Status  = SafeIntnToInt8 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  Operand = (-1537977259);
 | 
						|
  Status  = SafeIntnToInt8 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeIntnToChar8 (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  INTN        Operand;
 | 
						|
  CHAR8       Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // CHAR8 is typedefed as char, which may be signed or unsigned based
 | 
						|
  // on the compiler. Thus, for compatibility CHAR8 should be between 0 and MAX_INT8.
 | 
						|
  //
 | 
						|
 | 
						|
  //
 | 
						|
  // If Operand is between MIN_INT8 and MAX_INT8 inclusive, then it's a cast
 | 
						|
  //
 | 
						|
  Operand = 0x5b;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeIntnToChar8 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x5b, Result);
 | 
						|
 | 
						|
  Operand = 0;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeIntnToChar8 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0, Result);
 | 
						|
 | 
						|
  Operand = MAX_INT8;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeIntnToChar8 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (MAX_INT8, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Operand = (-53);
 | 
						|
  Status  = SafeIntnToChar8 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  Operand = (0x5bababab);
 | 
						|
  Status  = SafeIntnToChar8 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  Operand = (-1537977259);
 | 
						|
  Status  = SafeIntnToChar8 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeIntnToUint8 (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  INTN        Operand;
 | 
						|
  UINT8       Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If Operand is between 0 and MAX_UINT8 inclusive, then it's a cast
 | 
						|
  //
 | 
						|
  Operand = 0xab;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeIntnToUint8 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0xab, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Operand = (0x5bababab);
 | 
						|
  Status  = SafeIntnToUint8 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  Operand = (-1537977259);
 | 
						|
  Status  = SafeIntnToUint8 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeIntnToInt16 (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  INTN        Operand;
 | 
						|
  INT16       Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If Operand is between MIN_INT16 and MAX_INT16 inclusive, then it's a cast
 | 
						|
  //
 | 
						|
  Operand = 0x5bab;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeIntnToInt16 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x5bab, Result);
 | 
						|
 | 
						|
  Operand = (-23467);
 | 
						|
  Status  = SafeIntnToInt16 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL ((-23467), Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Operand = (0x5bababab);
 | 
						|
  Status  = SafeIntnToInt16 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  Operand = (-1537977259);
 | 
						|
  Status  = SafeIntnToInt16 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeIntnToUint16 (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  INTN        Operand;
 | 
						|
  UINT16      Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If Operand is between 0 and MAX_UINT16 inclusive, then it's a cast
 | 
						|
  //
 | 
						|
  Operand = 0xabab;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeIntnToUint16 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0xabab, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Operand = (0x5bababab);
 | 
						|
  Status  = SafeIntnToUint16 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  Operand = (-1537977259);
 | 
						|
  Status  = SafeIntnToUint16 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeIntnToUintn (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  INTN        Operand;
 | 
						|
  UINTN       Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If Operand is non-negative, then it's a cast
 | 
						|
  //
 | 
						|
  Operand = 0x5bababab;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeIntnToUintn (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x5bababab, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Operand = (-1537977259);
 | 
						|
  Status  = SafeIntnToUintn (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeIntnToUint64 (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  INTN        Operand;
 | 
						|
  UINT64      Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If Operand is non-negative, then it's a cast
 | 
						|
  //
 | 
						|
  Operand = 0x5bababab;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeIntnToUint64 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x5bababab, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Operand = (-1537977259);
 | 
						|
  Status  = SafeIntnToUint64 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeUintnToInt8 (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  UINTN       Operand;
 | 
						|
  INT8        Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If Operand is <= MAX_INT8, then it's a cast
 | 
						|
  //
 | 
						|
  Operand = 0x5b;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeUintnToInt8 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x5b, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Operand = (0xabab);
 | 
						|
  Status  = SafeUintnToInt8 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeUintnToChar8 (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  UINTN       Operand;
 | 
						|
  CHAR8       Result;
 | 
						|
 | 
						|
  // CHAR8 is typedefed as char, which by default is signed, thus
 | 
						|
  // CHAR8 is same as INT8, so same tests as above:
 | 
						|
 | 
						|
  //
 | 
						|
  // If Operand is <= MAX_INT8, then it's a cast
 | 
						|
  //
 | 
						|
  Operand = 0x5b;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeUintnToChar8 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x5b, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Operand = (0xabab);
 | 
						|
  Status  = SafeUintnToChar8 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeUintnToUint8 (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  UINTN       Operand;
 | 
						|
  UINT8       Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If Operand is <= MAX_UINT8, then it's a cast
 | 
						|
  //
 | 
						|
  Operand = 0xab;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeUintnToUint8 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0xab, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Operand = (0xabab);
 | 
						|
  Status  = SafeUintnToUint8 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeUintnToInt16 (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  UINTN       Operand;
 | 
						|
  INT16       Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If Operand is <= MAX_INT16, then it's a cast
 | 
						|
  //
 | 
						|
  Operand = 0x5bab;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeUintnToInt16 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x5bab, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Operand = (0xabab);
 | 
						|
  Status  = SafeUintnToInt16 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeUintnToUint16 (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  UINTN       Operand;
 | 
						|
  UINT16      Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If Operand is <= MAX_UINT16, then it's a cast
 | 
						|
  //
 | 
						|
  Operand = 0xabab;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeUintnToUint16 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0xabab, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Operand = (0xabababab);
 | 
						|
  Status  = SafeUintnToUint16 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeUintnToInt32 (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  UINTN       Operand;
 | 
						|
  INT32       Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If Operand is <= MAX_INT32, then it's a cast
 | 
						|
  //
 | 
						|
  Operand = 0x5bababab;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeUintnToInt32 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x5bababab, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Operand = (0xabababab);
 | 
						|
  Status  = SafeUintnToInt32 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeInt64ToInt8 (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  INT64       Operand;
 | 
						|
  INT8        Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If Operand is between MIN_INT8 and  MAX_INT8 inclusive, then it's a cast
 | 
						|
  //
 | 
						|
  Operand = 0x5b;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeInt64ToInt8 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x5b, Result);
 | 
						|
 | 
						|
  Operand = (-37);
 | 
						|
  Status  = SafeInt64ToInt8 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL ((-37), Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Operand = (0x5babababefefefef);
 | 
						|
  Status  = SafeInt64ToInt8 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  Operand =  (-6605562033422200815);
 | 
						|
  Status  = SafeInt64ToInt8 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeInt64ToChar8 (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  INT64       Operand;
 | 
						|
  CHAR8       Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // CHAR8 is typedefed as char, which may be signed or unsigned based
 | 
						|
  // on the compiler. Thus, for compatibility CHAR8 should be between 0 and MAX_INT8.
 | 
						|
  //
 | 
						|
 | 
						|
  //
 | 
						|
  // If Operand is between MIN_INT8 and  MAX_INT8 inclusive, then it's a cast
 | 
						|
  //
 | 
						|
  Operand = 0x5b;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeInt64ToChar8 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x5b, Result);
 | 
						|
 | 
						|
  Operand = 0;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeInt64ToChar8 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0, Result);
 | 
						|
 | 
						|
  Operand = MAX_INT8;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeInt64ToChar8 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (MAX_INT8, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Operand = (-37);
 | 
						|
  Status  = SafeInt64ToChar8 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  Operand = (0x5babababefefefef);
 | 
						|
  Status  = SafeInt64ToChar8 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  Operand =  (-6605562033422200815);
 | 
						|
  Status  = SafeInt64ToChar8 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeInt64ToUint8 (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  INT64       Operand;
 | 
						|
  UINT8       Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If Operand is between 0 and  MAX_UINT8 inclusive, then it's a cast
 | 
						|
  //
 | 
						|
  Operand = 0xab;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeInt64ToUint8 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0xab, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Operand = (0x5babababefefefef);
 | 
						|
  Status  = SafeInt64ToUint8 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  Operand =  (-6605562033422200815);
 | 
						|
  Status  = SafeInt64ToUint8 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeInt64ToInt16 (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  INT64       Operand;
 | 
						|
  INT16       Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If Operand is between MIN_INT16 and  MAX_INT16 inclusive, then it's a cast
 | 
						|
  //
 | 
						|
  Operand = 0x5bab;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeInt64ToInt16 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x5bab, Result);
 | 
						|
 | 
						|
  Operand = (-23467);
 | 
						|
  Status  = SafeInt64ToInt16 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL ((-23467), Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Operand = (0x5babababefefefef);
 | 
						|
  Status  = SafeInt64ToInt16 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  Operand =  (-6605562033422200815);
 | 
						|
  Status  = SafeInt64ToInt16 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeInt64ToUint16 (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  INT64       Operand;
 | 
						|
  UINT16      Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If Operand is between 0 and  MAX_UINT16 inclusive, then it's a cast
 | 
						|
  //
 | 
						|
  Operand = 0xabab;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeInt64ToUint16 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0xabab, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Operand = (0x5babababefefefef);
 | 
						|
  Status  = SafeInt64ToUint16 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  Operand =  (-6605562033422200815);
 | 
						|
  Status  = SafeInt64ToUint16 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeInt64ToInt32 (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  INT64       Operand;
 | 
						|
  INT32       Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If Operand is between MIN_INT32 and  MAX_INT32 inclusive, then it's a cast
 | 
						|
  //
 | 
						|
  Operand = 0x5bababab;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeInt64ToInt32 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x5bababab, Result);
 | 
						|
 | 
						|
  Operand = (-1537977259);
 | 
						|
  Status  = SafeInt64ToInt32 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL ((-1537977259), Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Operand = (0x5babababefefefef);
 | 
						|
  Status  = SafeInt64ToInt32 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  Operand =  (-6605562033422200815);
 | 
						|
  Status  = SafeInt64ToInt32 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeInt64ToUint32 (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  INT64       Operand;
 | 
						|
  UINT32      Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If Operand is between 0 and  MAX_UINT32 inclusive, then it's a cast
 | 
						|
  //
 | 
						|
  Operand = 0xabababab;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeInt64ToUint32 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0xabababab, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Operand = (0x5babababefefefef);
 | 
						|
  Status  = SafeInt64ToUint32 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  Operand =  (-6605562033422200815);
 | 
						|
  Status  = SafeInt64ToUint32 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeInt64ToUint64 (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  INT64       Operand;
 | 
						|
  UINT64      Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If Operand is non-negative, then it's a cast
 | 
						|
  //
 | 
						|
  Operand = 0x5babababefefefef;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeInt64ToUint64 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x5babababefefefef, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Operand =  (-6605562033422200815);
 | 
						|
  Status  = SafeInt64ToUint64 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeUint64ToInt8 (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  UINT64      Operand;
 | 
						|
  INT8        Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If Operand is <= MAX_INT8, then it's a cast
 | 
						|
  //
 | 
						|
  Operand = 0x5b;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeUint64ToInt8 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x5b, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Operand = (0xababababefefefef);
 | 
						|
  Status  = SafeUint64ToInt8 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeUint64ToChar8 (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  UINT64      Operand;
 | 
						|
  CHAR8       Result;
 | 
						|
 | 
						|
  // CHAR8 is typedefed as char, which by default is signed, thus
 | 
						|
  // CHAR8 is same as INT8, so same tests as above:
 | 
						|
 | 
						|
  //
 | 
						|
  // If Operand is <= MAX_INT8, then it's a cast
 | 
						|
  //
 | 
						|
  Operand = 0x5b;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeUint64ToChar8 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x5b, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Operand = (0xababababefefefef);
 | 
						|
  Status  = SafeUint64ToChar8 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeUint64ToUint8 (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  UINT64      Operand;
 | 
						|
  UINT8       Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If Operand is <= MAX_UINT8, then it's a cast
 | 
						|
  //
 | 
						|
  Operand = 0xab;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeUint64ToUint8 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0xab, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Operand = (0xababababefefefef);
 | 
						|
  Status  = SafeUint64ToUint8 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeUint64ToInt16 (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  UINT64      Operand;
 | 
						|
  INT16       Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If Operand is <= MAX_INT16, then it's a cast
 | 
						|
  //
 | 
						|
  Operand = 0x5bab;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeUint64ToInt16 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x5bab, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Operand = (0xababababefefefef);
 | 
						|
  Status  = SafeUint64ToInt16 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeUint64ToUint16 (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  UINT64      Operand;
 | 
						|
  UINT16      Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If Operand is <= MAX_UINT16, then it's a cast
 | 
						|
  //
 | 
						|
  Operand = 0xabab;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeUint64ToUint16 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0xabab, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Operand = (0xababababefefefef);
 | 
						|
  Status  = SafeUint64ToUint16 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeUint64ToInt32 (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  UINT64      Operand;
 | 
						|
  INT32       Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If Operand is <= MAX_INT32, then it's a cast
 | 
						|
  //
 | 
						|
  Operand = 0x5bababab;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeUint64ToInt32 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x5bababab, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Operand = (0xababababefefefef);
 | 
						|
  Status  = SafeUint64ToInt32 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeUint64ToUint32 (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  UINT64      Operand;
 | 
						|
  UINT32      Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If Operand is <= MAX_UINT32, then it's a cast
 | 
						|
  //
 | 
						|
  Operand = 0xabababab;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeUint64ToUint32 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0xabababab, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Operand = (0xababababefefefef);
 | 
						|
  Status  = SafeUint64ToUint32 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeUint64ToInt64 (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  UINT64      Operand;
 | 
						|
  INT64       Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If Operand is <= MAX_INT64, then it's a cast
 | 
						|
  //
 | 
						|
  Operand = 0x5babababefefefef;
 | 
						|
  Result  = 0;
 | 
						|
  Status  = SafeUint64ToInt64 (Operand, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x5babababefefefef, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Operand = (0xababababefefefef);
 | 
						|
  Status  = SafeUint64ToInt64 (Operand, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
//
 | 
						|
// Addition function tests:
 | 
						|
//
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeUint8Add (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  UINT8       Augend;
 | 
						|
  UINT8       Addend;
 | 
						|
  UINT8       Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If the result of addition doesn't overflow MAX_UINT8, then it's addition
 | 
						|
  //
 | 
						|
  Augend = 0x3a;
 | 
						|
  Addend = 0x3a;
 | 
						|
  Result = 0;
 | 
						|
  Status = SafeUint8Add (Augend, Addend, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x74, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Augend = 0xab;
 | 
						|
  Addend = 0xbc;
 | 
						|
  Status = SafeUint8Add (Augend, Addend, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeUint16Add (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  UINT16      Augend = 0x3a3a;
 | 
						|
  UINT16      Addend = 0x3a3a;
 | 
						|
  UINT16      Result = 0;
 | 
						|
 | 
						|
  //
 | 
						|
  // If the result of addition doesn't overflow MAX_UINT16, then it's addition
 | 
						|
  //
 | 
						|
  Status = SafeUint16Add (Augend, Addend, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x7474, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Augend = 0xabab;
 | 
						|
  Addend = 0xbcbc;
 | 
						|
  Status = SafeUint16Add (Augend, Addend, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeUint32Add (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  UINT32      Augend;
 | 
						|
  UINT32      Addend;
 | 
						|
  UINT32      Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If the result of addition doesn't overflow MAX_UINT32, then it's addition
 | 
						|
  //
 | 
						|
  Augend = 0x3a3a3a3a;
 | 
						|
  Addend = 0x3a3a3a3a;
 | 
						|
  Result = 0;
 | 
						|
  Status = SafeUint32Add (Augend, Addend, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x74747474, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Augend = 0xabababab;
 | 
						|
  Addend = 0xbcbcbcbc;
 | 
						|
  Status = SafeUint32Add (Augend, Addend, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeUint64Add (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  UINT64      Augend;
 | 
						|
  UINT64      Addend;
 | 
						|
  UINT64      Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If the result of addition doesn't overflow MAX_UINT64, then it's addition
 | 
						|
  //
 | 
						|
  Augend = 0x3a3a3a3a12121212;
 | 
						|
  Addend = 0x3a3a3a3a12121212;
 | 
						|
  Result = 0;
 | 
						|
  Status = SafeUint64Add (Augend, Addend, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x7474747424242424, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Augend = 0xababababefefefef;
 | 
						|
  Addend = 0xbcbcbcbcdededede;
 | 
						|
  Status = SafeUint64Add (Augend, Addend, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeInt8Add (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  INT8        Augend;
 | 
						|
  INT8        Addend;
 | 
						|
  INT8        Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If the result of addition doesn't overflow MAX_INT8
 | 
						|
  // and doesn't underflow MIN_INT8, then it's addition
 | 
						|
  //
 | 
						|
  Augend = 0x3a;
 | 
						|
  Addend = 0x3a;
 | 
						|
  Result = 0;
 | 
						|
  Status = SafeInt8Add (Augend, Addend, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x74, Result);
 | 
						|
 | 
						|
  Augend = (-58);
 | 
						|
  Addend = (-58);
 | 
						|
  Status = SafeInt8Add (Augend, Addend, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL ((-116), Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Augend = 0x5a;
 | 
						|
  Addend = 0x5a;
 | 
						|
  Status = SafeInt8Add (Augend, Addend, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  Augend = (-90);
 | 
						|
  Addend = (-90);
 | 
						|
  Status = SafeInt8Add (Augend, Addend, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeInt16Add (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  INT16       Augend;
 | 
						|
  INT16       Addend;
 | 
						|
  INT16       Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If the result of addition doesn't overflow MAX_INT16
 | 
						|
  // and doesn't underflow MIN_INT16, then it's addition
 | 
						|
  //
 | 
						|
  Augend = 0x3a3a;
 | 
						|
  Addend = 0x3a3a;
 | 
						|
  Result = 0;
 | 
						|
  Status = SafeInt16Add (Augend, Addend, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x7474, Result);
 | 
						|
 | 
						|
  Augend = (-14906);
 | 
						|
  Addend = (-14906);
 | 
						|
  Status = SafeInt16Add (Augend, Addend, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL ((-29812), Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Augend = 0x5a5a;
 | 
						|
  Addend = 0x5a5a;
 | 
						|
  Status = SafeInt16Add (Augend, Addend, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  Augend = (-23130);
 | 
						|
  Addend = (-23130);
 | 
						|
  Status = SafeInt16Add (Augend, Addend, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeInt32Add (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  INT32       Augend;
 | 
						|
  INT32       Addend;
 | 
						|
  INT32       Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If the result of addition doesn't overflow MAX_INT32
 | 
						|
  // and doesn't underflow MIN_INT32, then it's addition
 | 
						|
  //
 | 
						|
  Augend = 0x3a3a3a3a;
 | 
						|
  Addend = 0x3a3a3a3a;
 | 
						|
  Result = 0;
 | 
						|
  Status = SafeInt32Add (Augend, Addend, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x74747474, Result);
 | 
						|
 | 
						|
  Augend = (-976894522);
 | 
						|
  Addend = (-976894522);
 | 
						|
  Status = SafeInt32Add (Augend, Addend, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL ((-1953789044), Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Augend = 0x5a5a5a5a;
 | 
						|
  Addend = 0x5a5a5a5a;
 | 
						|
  Status = SafeInt32Add (Augend, Addend, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  Augend = (-1515870810);
 | 
						|
  Addend = (-1515870810);
 | 
						|
  Status = SafeInt32Add (Augend, Addend, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeInt64Add (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  INT64       Augend;
 | 
						|
  INT64       Addend;
 | 
						|
  INT64       Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If the result of addition doesn't overflow MAX_INT64
 | 
						|
  // and doesn't underflow MIN_INT64, then it's addition
 | 
						|
  //
 | 
						|
  Augend = 0x3a3a3a3a3a3a3a3a;
 | 
						|
  Addend = 0x3a3a3a3a3a3a3a3a;
 | 
						|
  Result = 0;
 | 
						|
  Status = SafeInt64Add (Augend, Addend, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x7474747474747474, Result);
 | 
						|
 | 
						|
  Augend = (-4195730024608447034);
 | 
						|
  Addend = (-4195730024608447034);
 | 
						|
  Status = SafeInt64Add (Augend, Addend, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL ((-8391460049216894068), Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Augend = 0x5a5a5a5a5a5a5a5a;
 | 
						|
  Addend = 0x5a5a5a5a5a5a5a5a;
 | 
						|
  Status = SafeInt64Add (Augend, Addend, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  Augend = (-6510615555426900570);
 | 
						|
  Addend = (-6510615555426900570);
 | 
						|
  Status = SafeInt64Add (Augend, Addend, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
//
 | 
						|
// Subtraction function tests:
 | 
						|
//
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeUint8Sub (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  UINT8       Minuend;
 | 
						|
  UINT8       Subtrahend;
 | 
						|
  UINT8       Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If Minuend >= Subtrahend, then it's subtraction
 | 
						|
  //
 | 
						|
  Minuend    = 0x5a;
 | 
						|
  Subtrahend = 0x3b;
 | 
						|
  Result     = 0;
 | 
						|
  Status     = SafeUint8Sub (Minuend, Subtrahend, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x1f, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Minuend    = 0x5a;
 | 
						|
  Subtrahend = 0x6d;
 | 
						|
  Status     = SafeUint8Sub (Minuend, Subtrahend, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeUint16Sub (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  UINT16      Minuend;
 | 
						|
  UINT16      Subtrahend;
 | 
						|
  UINT16      Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If Minuend >= Subtrahend, then it's subtraction
 | 
						|
  //
 | 
						|
  Minuend    = 0x5a5a;
 | 
						|
  Subtrahend = 0x3b3b;
 | 
						|
  Result     = 0;
 | 
						|
  Status     = SafeUint16Sub (Minuend, Subtrahend, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x1f1f, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Minuend    = 0x5a5a;
 | 
						|
  Subtrahend = 0x6d6d;
 | 
						|
  Status     = SafeUint16Sub (Minuend, Subtrahend, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeUint32Sub (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  UINT32      Minuend;
 | 
						|
  UINT32      Subtrahend;
 | 
						|
  UINT32      Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If Minuend >= Subtrahend, then it's subtraction
 | 
						|
  //
 | 
						|
  Minuend    = 0x5a5a5a5a;
 | 
						|
  Subtrahend = 0x3b3b3b3b;
 | 
						|
  Result     = 0;
 | 
						|
  Status     = SafeUint32Sub (Minuend, Subtrahend, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x1f1f1f1f, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Minuend    = 0x5a5a5a5a;
 | 
						|
  Subtrahend = 0x6d6d6d6d;
 | 
						|
  Status     = SafeUint32Sub (Minuend, Subtrahend, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeUint64Sub (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  UINT64      Minuend;
 | 
						|
  UINT64      Subtrahend;
 | 
						|
  UINT64      Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If Minuend >= Subtrahend, then it's subtraction
 | 
						|
  //
 | 
						|
  Minuend    = 0x5a5a5a5a5a5a5a5a;
 | 
						|
  Subtrahend = 0x3b3b3b3b3b3b3b3b;
 | 
						|
  Result     = 0;
 | 
						|
  Status     = SafeUint64Sub (Minuend, Subtrahend, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x1f1f1f1f1f1f1f1f, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Minuend    = 0x5a5a5a5a5a5a5a5a;
 | 
						|
  Subtrahend = 0x6d6d6d6d6d6d6d6d;
 | 
						|
  Status     = SafeUint64Sub (Minuend, Subtrahend, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeInt8Sub (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  INT8        Minuend;
 | 
						|
  INT8        Subtrahend;
 | 
						|
  INT8        Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If the result of subtractions doesn't overflow MAX_INT8 or
 | 
						|
  // underflow MIN_INT8, then it's subtraction
 | 
						|
  //
 | 
						|
  Minuend    = 0x5a;
 | 
						|
  Subtrahend = 0x3a;
 | 
						|
  Result     = 0;
 | 
						|
  Status     = SafeInt8Sub (Minuend, Subtrahend, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x20, Result);
 | 
						|
 | 
						|
  Minuend    = 58;
 | 
						|
  Subtrahend = 78;
 | 
						|
  Status     = SafeInt8Sub (Minuend, Subtrahend, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL ((-20), Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Minuend    = (-80);
 | 
						|
  Subtrahend = 80;
 | 
						|
  Status     = SafeInt8Sub (Minuend, Subtrahend, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  Minuend    = (80);
 | 
						|
  Subtrahend = (-80);
 | 
						|
  Status     = SafeInt8Sub (Minuend, Subtrahend, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeInt16Sub (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  INT16       Minuend;
 | 
						|
  INT16       Subtrahend;
 | 
						|
  INT16       Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If the result of subtractions doesn't overflow MAX_INT16 or
 | 
						|
  // underflow MIN_INT16, then it's subtraction
 | 
						|
  //
 | 
						|
  Minuend    = 0x5a5a;
 | 
						|
  Subtrahend = 0x3a3a;
 | 
						|
  Result     = 0;
 | 
						|
  Status     = SafeInt16Sub (Minuend, Subtrahend, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x2020, Result);
 | 
						|
 | 
						|
  Minuend    = 0x3a3a;
 | 
						|
  Subtrahend = 0x5a5a;
 | 
						|
  Status     = SafeInt16Sub (Minuend, Subtrahend, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL ((-8224), Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Minuend    = (-31354);
 | 
						|
  Subtrahend = 31354;
 | 
						|
  Status     = SafeInt16Sub (Minuend, Subtrahend, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  Minuend    = (31354);
 | 
						|
  Subtrahend = (-31354);
 | 
						|
  Status     = SafeInt16Sub (Minuend, Subtrahend, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeInt32Sub (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  INT32       Minuend;
 | 
						|
  INT32       Subtrahend;
 | 
						|
  INT32       Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If the result of subtractions doesn't overflow MAX_INT32 or
 | 
						|
  // underflow MIN_INT32, then it's subtraction
 | 
						|
  //
 | 
						|
  Minuend    = 0x5a5a5a5a;
 | 
						|
  Subtrahend = 0x3a3a3a3a;
 | 
						|
  Result     = 0;
 | 
						|
  Status     = SafeInt32Sub (Minuend, Subtrahend, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x20202020, Result);
 | 
						|
 | 
						|
  Minuend    = 0x3a3a3a3a;
 | 
						|
  Subtrahend = 0x5a5a5a5a;
 | 
						|
  Status     = SafeInt32Sub (Minuend, Subtrahend, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL ((-538976288), Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Minuend    = (-2054847098);
 | 
						|
  Subtrahend = 2054847098;
 | 
						|
  Status     = SafeInt32Sub (Minuend, Subtrahend, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  Minuend    = (2054847098);
 | 
						|
  Subtrahend = (-2054847098);
 | 
						|
  Status     = SafeInt32Sub (Minuend, Subtrahend, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeInt64Sub (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  INT64       Minuend;
 | 
						|
  INT64       Subtrahend;
 | 
						|
  INT64       Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If the result of subtractions doesn't overflow MAX_INT64 or
 | 
						|
  // underflow MIN_INT64, then it's subtraction
 | 
						|
  //
 | 
						|
  Minuend    = 0x5a5a5a5a5a5a5a5a;
 | 
						|
  Subtrahend = 0x3a3a3a3a3a3a3a3a;
 | 
						|
  Result     = 0;
 | 
						|
  Status     = SafeInt64Sub (Minuend, Subtrahend, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x2020202020202020, Result);
 | 
						|
 | 
						|
  Minuend    = 0x3a3a3a3a3a3a3a3a;
 | 
						|
  Subtrahend = 0x5a5a5a5a5a5a5a5a;
 | 
						|
  Status     = SafeInt64Sub (Minuend, Subtrahend, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL ((-2314885530818453536), Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Minuend    = (-8825501086245354106);
 | 
						|
  Subtrahend = 8825501086245354106;
 | 
						|
  Status     = SafeInt64Sub (Minuend, Subtrahend, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  Minuend    = (8825501086245354106);
 | 
						|
  Subtrahend = (-8825501086245354106);
 | 
						|
  Status     = SafeInt64Sub (Minuend, Subtrahend, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
//
 | 
						|
// Multiplication function tests:
 | 
						|
//
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeUint8Mult (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  UINT8       Multiplicand;
 | 
						|
  UINT8       Multiplier;
 | 
						|
  UINT8       Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If the result of multiplication doesn't overflow MAX_UINT8, it will succeed
 | 
						|
  //
 | 
						|
  Multiplicand = 0x12;
 | 
						|
  Multiplier   = 0xa;
 | 
						|
  Result       = 0;
 | 
						|
  Status       = SafeUint8Mult (Multiplicand, Multiplier, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0xb4, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Multiplicand = 0x12;
 | 
						|
  Multiplier   = 0x23;
 | 
						|
  Status       = SafeUint8Mult (Multiplicand, Multiplier, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeUint16Mult (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  UINT16      Multiplicand;
 | 
						|
  UINT16      Multiplier;
 | 
						|
  UINT16      Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If the result of multiplication doesn't overflow MAX_UINT16, it will succeed
 | 
						|
  //
 | 
						|
  Multiplicand = 0x212;
 | 
						|
  Multiplier   = 0x7a;
 | 
						|
  Result       = 0;
 | 
						|
  Status       = SafeUint16Mult (Multiplicand, Multiplier, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0xfc94, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Multiplicand = 0x1234;
 | 
						|
  Multiplier   = 0x213;
 | 
						|
  Status       = SafeUint16Mult (Multiplicand, Multiplier, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeUint32Mult (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  UINT32      Multiplicand;
 | 
						|
  UINT32      Multiplier;
 | 
						|
  UINT32      Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If the result of multiplication doesn't overflow MAX_UINT32, it will succeed
 | 
						|
  //
 | 
						|
  Multiplicand = 0xa122a;
 | 
						|
  Multiplier   = 0xd23;
 | 
						|
  Result       = 0;
 | 
						|
  Status       = SafeUint32Mult (Multiplicand, Multiplier, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x844c9dbe, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Multiplicand = 0xa122a;
 | 
						|
  Multiplier   = 0xed23;
 | 
						|
  Status       = SafeUint32Mult (Multiplicand, Multiplier, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeUint64Mult (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  UINT64      Multiplicand;
 | 
						|
  UINT64      Multiplier;
 | 
						|
  UINT64      Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If the result of multiplication doesn't overflow MAX_UINT64, it will succeed
 | 
						|
  //
 | 
						|
  Multiplicand = 0x123456789a;
 | 
						|
  Multiplier   = 0x1234567;
 | 
						|
  Result       = 0;
 | 
						|
  Status       = SafeUint64Mult (Multiplicand, Multiplier, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x14b66db9745a07f6, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Multiplicand = 0x123456789a;
 | 
						|
  Multiplier   = 0x12345678;
 | 
						|
  Status       = SafeUint64Mult (Multiplicand, Multiplier, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeInt8Mult (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  INT8        Multiplicand;
 | 
						|
  INT8        Multiplier;
 | 
						|
  INT8        Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If the result of multiplication doesn't overflow MAX_INT8 and doesn't
 | 
						|
  // underflow MIN_UINT8, it will succeed
 | 
						|
  //
 | 
						|
  Multiplicand = 0x12;
 | 
						|
  Multiplier   = 0x7;
 | 
						|
  Result       = 0;
 | 
						|
  Status       = SafeInt8Mult (Multiplicand, Multiplier, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x7e, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Multiplicand = 0x12;
 | 
						|
  Multiplier   = 0xa;
 | 
						|
  Status       = SafeInt8Mult (Multiplicand, Multiplier, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeInt16Mult (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  INT16       Multiplicand;
 | 
						|
  INT16       Multiplier;
 | 
						|
  INT16       Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If the result of multiplication doesn't overflow MAX_INT16 and doesn't
 | 
						|
  // underflow MIN_UINT16, it will succeed
 | 
						|
  //
 | 
						|
  Multiplicand = 0x123;
 | 
						|
  Multiplier   = 0x67;
 | 
						|
  Result       = 0;
 | 
						|
  Status       = SafeInt16Mult (Multiplicand, Multiplier, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x7515, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Multiplicand = 0x123;
 | 
						|
  Multiplier   = 0xab;
 | 
						|
  Status       = SafeInt16Mult (Multiplicand, Multiplier, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeInt32Mult (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  INT32       Multiplicand;
 | 
						|
  INT32       Multiplier;
 | 
						|
  INT32       Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If the result of multiplication doesn't overflow MAX_INT32 and doesn't
 | 
						|
  // underflow MIN_UINT32, it will succeed
 | 
						|
  //
 | 
						|
  Multiplicand = 0x123456;
 | 
						|
  Multiplier   = 0x678;
 | 
						|
  Result       = 0;
 | 
						|
  Status       = SafeInt32Mult (Multiplicand, Multiplier, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x75c28c50, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Multiplicand = 0x123456;
 | 
						|
  Multiplier   = 0xabc;
 | 
						|
  Status       = SafeInt32Mult (Multiplicand, Multiplier, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
UNIT_TEST_STATUS
 | 
						|
EFIAPI
 | 
						|
TestSafeInt64Mult (
 | 
						|
  IN UNIT_TEST_CONTEXT  Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  INT64       Multiplicand;
 | 
						|
  INT64       Multiplier;
 | 
						|
  INT64       Result;
 | 
						|
 | 
						|
  //
 | 
						|
  // If the result of multiplication doesn't overflow MAX_INT64 and doesn't
 | 
						|
  // underflow MIN_UINT64, it will succeed
 | 
						|
  //
 | 
						|
  Multiplicand = 0x123456789;
 | 
						|
  Multiplier   = 0x6789abcd;
 | 
						|
  Result       = 0;
 | 
						|
  Status       = SafeInt64Mult (Multiplicand, Multiplier, &Result);
 | 
						|
  UT_ASSERT_NOT_EFI_ERROR (Status);
 | 
						|
  UT_ASSERT_EQUAL (0x75cd9045220d6bb5, Result);
 | 
						|
 | 
						|
  //
 | 
						|
  // Otherwise should result in an error status
 | 
						|
  //
 | 
						|
  Multiplicand = 0x123456789;
 | 
						|
  Multiplier   = 0xa789abcd;
 | 
						|
  Status       = SafeInt64Mult (Multiplicand, Multiplier, &Result);
 | 
						|
  UT_ASSERT_EQUAL (RETURN_BUFFER_TOO_SMALL, Status);
 | 
						|
 | 
						|
  return UNIT_TEST_PASSED;
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 | 
						|
  Main fuction sets up the unit test environment
 | 
						|
 | 
						|
**/
 | 
						|
EFI_STATUS
 | 
						|
EFIAPI
 | 
						|
UefiTestMain (
 | 
						|
  VOID
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS                  Status;
 | 
						|
  UNIT_TEST_FRAMEWORK_HANDLE  Framework;
 | 
						|
  UNIT_TEST_SUITE_HANDLE      ConversionTestSuite;
 | 
						|
  UNIT_TEST_SUITE_HANDLE      AdditionSubtractionTestSuite;
 | 
						|
  UNIT_TEST_SUITE_HANDLE      MultiplicationTestSuite;
 | 
						|
 | 
						|
  Framework                    = NULL;
 | 
						|
  ConversionTestSuite          = NULL;
 | 
						|
  AdditionSubtractionTestSuite = NULL;
 | 
						|
  MultiplicationTestSuite      = NULL;
 | 
						|
 | 
						|
  DEBUG ((DEBUG_INFO, "%a v%a\n", UNIT_TEST_NAME, UNIT_TEST_VERSION));
 | 
						|
 | 
						|
  //
 | 
						|
  // Start setting up the test framework for running the tests.
 | 
						|
  //
 | 
						|
  Status = InitUnitTestFramework (&Framework, UNIT_TEST_NAME, gEfiCallerBaseName, UNIT_TEST_VERSION);
 | 
						|
  if (EFI_ERROR (Status)) {
 | 
						|
    DEBUG ((DEBUG_ERROR, "Failed in InitUnitTestFramework. Status = %r\n", Status));
 | 
						|
    goto EXIT;
 | 
						|
  }
 | 
						|
 | 
						|
  ///
 | 
						|
  // Test the conversion functions
 | 
						|
  //
 | 
						|
  Status = CreateUnitTestSuite (&ConversionTestSuite, Framework, "Int Safe Conversions Test Suite", "Common.SafeInt.Convert", NULL, NULL);
 | 
						|
  if (EFI_ERROR (Status)) {
 | 
						|
    DEBUG ((DEBUG_ERROR, "Failed in CreateUnitTestSuite for Conversions Test Suite\n"));
 | 
						|
    Status = EFI_OUT_OF_RESOURCES;
 | 
						|
    goto EXIT;
 | 
						|
  }
 | 
						|
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeInt8ToUint8", "TestSafeInt8ToUint8", TestSafeInt8ToUint8, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeInt8ToUint16", "TestSafeInt8ToUint16", TestSafeInt8ToUint16, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeInt8ToUint32", "TestSafeInt8ToUint32", TestSafeInt8ToUint32, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeInt8ToUintn", "TestSafeInt8ToUintn", TestSafeInt8ToUintn, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeInt8ToUint64", "TestSafeInt8ToUint64", TestSafeInt8ToUint64, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeUint8ToInt8", "TestSafeUint8ToInt8", TestSafeUint8ToInt8, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeUint8ToChar8", "TestSafeUint8ToChar8", TestSafeUint8ToChar8, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeInt16ToInt8", "TestSafeInt16ToInt8", TestSafeInt16ToInt8, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeInt16ToChar8", "TestSafeInt16ToChar8", TestSafeInt16ToChar8, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeInt16ToUint8", "TestSafeInt16ToUint8", TestSafeInt16ToUint8, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeInt16ToUint16", "TestSafeInt16ToUint16", TestSafeInt16ToUint16, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeInt16ToUint32", "TestSafeInt16ToUint32", TestSafeInt16ToUint32, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeInt16ToUintn", "TestSafeInt16ToUintn", TestSafeInt16ToUintn, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeInt16ToUint64", "TestSafeInt16ToUint64", TestSafeInt16ToUint64, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeUint16ToInt8", "TestSafeUint16ToInt8", TestSafeUint16ToInt8, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeUint16ToChar8", "TestSafeUint16ToChar8", TestSafeUint16ToChar8, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeUint16ToUint8", "TestSafeUint16ToUint8", TestSafeUint16ToUint8, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeUint16ToInt16", "TestSafeUint16ToInt16", TestSafeUint16ToInt16, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeInt32ToInt8", "TestSafeInt32ToInt8", TestSafeInt32ToInt8, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeInt32ToChar8", "TestSafeInt32ToChar8", TestSafeInt32ToChar8, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeInt32ToUint8", "TestSafeInt32ToUint8", TestSafeInt32ToUint8, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeInt32ToInt16", "TestSafeInt32ToInt16", TestSafeInt32ToInt16, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeInt32ToUint16", "TestSafeInt32ToUint16", TestSafeInt32ToUint16, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeInt32ToUint32", "TestSafeInt32ToUint32", TestSafeInt32ToUint32, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeInt32ToUintn", "TestSafeInt32ToUintn", TestSafeInt32ToUintn, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeInt32ToUint64", "TestSafeInt32ToUint64", TestSafeInt32ToUint64, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeUint32ToInt8", "TestSafeUint32ToInt8", TestSafeUint32ToInt8, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeUint32ToChar8", "TestSafeUint32ToChar8", TestSafeUint32ToChar8, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeUint32ToUint8", "TestSafeUint32ToUint8", TestSafeUint32ToUint8, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeUint32ToInt16", "TestSafeUint32ToInt16", TestSafeUint32ToInt16, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeUint32ToUint16", "TestSafeUint32ToUint16", TestSafeUint32ToUint16, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeUint32ToInt32", "TestSafeUint32ToInt32", TestSafeUint32ToInt32, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeUint32ToIntn", "TestSafeUint32ToIntn", TestSafeUint32ToIntn, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeIntnToInt8", "TestSafeIntnToInt8", TestSafeIntnToInt8, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeIntnToChar8", "TestSafeIntnToChar8", TestSafeIntnToChar8, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeIntnToUint8", "TestSafeIntnToUint8", TestSafeIntnToUint8, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeIntnToInt16", "TestSafeIntnToInt16", TestSafeIntnToInt16, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeIntnToUint16", "TestSafeIntnToUint16", TestSafeIntnToUint16, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeIntnToInt32", "TestSafeIntnToInt32", TestSafeIntnToInt32, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeIntnToUint32", "TestSafeIntnToUint32", TestSafeIntnToUint32, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeIntnToUintn", "TestSafeIntnToUintn", TestSafeIntnToUintn, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeIntnToUint64", "TestSafeIntnToUint64", TestSafeIntnToUint64, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeUintnToInt8", "TestSafeUintnToInt8", TestSafeUintnToInt8, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeUintnToChar8", "TestSafeUintnToChar8", TestSafeUintnToChar8, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeUintnToUint8", "TestSafeUintnToUint8", TestSafeUintnToUint8, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeUintnToInt16", "TestSafeUintnToInt16", TestSafeUintnToInt16, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeUintnToUint16", "TestSafeUintnToUint16", TestSafeUintnToUint16, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeUintnToInt32", "TestSafeUintnToInt32", TestSafeUintnToInt32, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeUintnToUint32", "TestSafeUintnToUint32", TestSafeUintnToUint32, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeUintnToIntn", "TestSafeUintnToIntn", TestSafeUintnToIntn, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeUintnToInt64", "TestSafeUintnToInt64", TestSafeUintnToInt64, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeInt64ToInt8", "TestSafeInt64ToInt8", TestSafeInt64ToInt8, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeInt64ToChar8", "TestSafeInt64ToChar8", TestSafeInt64ToChar8, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeInt64ToUint8", "TestSafeInt64ToUint8", TestSafeInt64ToUint8, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeInt64ToInt16", "TestSafeInt64ToInt16", TestSafeInt64ToInt16, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeInt64ToUint16", "TestSafeInt64ToUint16", TestSafeInt64ToUint16, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeInt64ToInt32", "TestSafeInt64ToInt32", TestSafeInt64ToInt32, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeInt64ToUint32", "TestSafeInt64ToUint32", TestSafeInt64ToUint32, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeInt64ToIntn", "TestSafeInt64ToIntn", TestSafeInt64ToIntn, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeInt64ToUintn", "TestSafeInt64ToUintn", TestSafeInt64ToUintn, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeInt64ToUint64", "TestSafeInt64ToUint64", TestSafeInt64ToUint64, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeUint64ToInt8", "TestSafeUint64ToInt8", TestSafeUint64ToInt8, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeUint64ToChar8", "TestSafeUint64ToChar8", TestSafeUint64ToChar8, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeUint64ToUint8", "TestSafeUint64ToUint8", TestSafeUint64ToUint8, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeUint64ToInt16", "TestSafeUint64ToInt16", TestSafeUint64ToInt16, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeUint64ToUint16", "TestSafeUint64ToUint16", TestSafeUint64ToUint16, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeUint64ToInt32", "TestSafeUint64ToInt32", TestSafeUint64ToInt32, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeUint64ToUint32", "TestSafeUint64ToUint32", TestSafeUint64ToUint32, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeUint64ToIntn", "TestSafeUint64ToIntn", TestSafeUint64ToIntn, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeUint64ToUintn", "TestSafeUint64ToUintn", TestSafeUint64ToUintn, NULL, NULL, NULL);
 | 
						|
  AddTestCase (ConversionTestSuite, "Test SafeUint64ToInt64", "TestSafeUint64ToInt64", TestSafeUint64ToInt64, NULL, NULL, NULL);
 | 
						|
 | 
						|
  //
 | 
						|
  // Test the addition and subtraction functions
 | 
						|
  //
 | 
						|
  Status = CreateUnitTestSuite (&AdditionSubtractionTestSuite, Framework, "Int Safe Add/Subtract Test Suite", "Common.SafeInt.AddSubtract", NULL, NULL);
 | 
						|
  if (EFI_ERROR (Status)) {
 | 
						|
    DEBUG ((DEBUG_ERROR, "Failed in CreateUnitTestSuite for Int Safe Add/Subtract Test Suite\n"));
 | 
						|
    Status = EFI_OUT_OF_RESOURCES;
 | 
						|
    goto EXIT;
 | 
						|
  }
 | 
						|
 | 
						|
  AddTestCase (AdditionSubtractionTestSuite, "Test SafeUint8Add", "TestSafeUint8Add", TestSafeUint8Add, NULL, NULL, NULL);
 | 
						|
  AddTestCase (AdditionSubtractionTestSuite, "Test SafeUint16Add", "TestSafeUint16Add", TestSafeUint16Add, NULL, NULL, NULL);
 | 
						|
  AddTestCase (AdditionSubtractionTestSuite, "Test SafeUint32Add", "TestSafeUint32Add", TestSafeUint32Add, NULL, NULL, NULL);
 | 
						|
  AddTestCase (AdditionSubtractionTestSuite, "Test SafeUintnAdd", "TestSafeUintnAdd", TestSafeUintnAdd, NULL, NULL, NULL);
 | 
						|
  AddTestCase (AdditionSubtractionTestSuite, "Test SafeUint64Add", "TestSafeUint64Add", TestSafeUint64Add, NULL, NULL, NULL);
 | 
						|
  AddTestCase (AdditionSubtractionTestSuite, "Test SafeInt8Add", "TestSafeInt8Add", TestSafeInt8Add, NULL, NULL, NULL);
 | 
						|
  AddTestCase (AdditionSubtractionTestSuite, "Test SafeInt16Add", "TestSafeInt16Add", TestSafeInt16Add, NULL, NULL, NULL);
 | 
						|
  AddTestCase (AdditionSubtractionTestSuite, "Test SafeInt32Add", "TestSafeInt32Add", TestSafeInt32Add, NULL, NULL, NULL);
 | 
						|
  AddTestCase (AdditionSubtractionTestSuite, "Test SafeIntnAdd", "TestSafeIntnAdd", TestSafeIntnAdd, NULL, NULL, NULL);
 | 
						|
  AddTestCase (AdditionSubtractionTestSuite, "Test SafeInt64Add", "TestSafeInt64Add", TestSafeInt64Add, NULL, NULL, NULL);
 | 
						|
  AddTestCase (AdditionSubtractionTestSuite, "Test SafeUint8Sub", "TestSafeUint8Sub", TestSafeUint8Sub, NULL, NULL, NULL);
 | 
						|
  AddTestCase (AdditionSubtractionTestSuite, "Test SafeUint16Sub", "TestSafeUint16Sub", TestSafeUint16Sub, NULL, NULL, NULL);
 | 
						|
  AddTestCase (AdditionSubtractionTestSuite, "Test SafeUint32Sub", "TestSafeUint32Sub", TestSafeUint32Sub, NULL, NULL, NULL);
 | 
						|
  AddTestCase (AdditionSubtractionTestSuite, "Test SafeUintnSub", "TestSafeUintnSub", TestSafeUintnSub, NULL, NULL, NULL);
 | 
						|
  AddTestCase (AdditionSubtractionTestSuite, "Test SafeUint64Sub", "TestSafeUint64Sub", TestSafeUint64Sub, NULL, NULL, NULL);
 | 
						|
  AddTestCase (AdditionSubtractionTestSuite, "Test SafeInt8Sub", "TestSafeInt8Sub", TestSafeInt8Sub, NULL, NULL, NULL);
 | 
						|
  AddTestCase (AdditionSubtractionTestSuite, "Test SafeInt16Sub", "TestSafeInt16Sub", TestSafeInt16Sub, NULL, NULL, NULL);
 | 
						|
  AddTestCase (AdditionSubtractionTestSuite, "Test SafeInt32Sub", "TestSafeInt32Sub", TestSafeInt32Sub, NULL, NULL, NULL);
 | 
						|
  AddTestCase (AdditionSubtractionTestSuite, "Test SafeIntnSub", "TestSafeIntnSub", TestSafeIntnSub, NULL, NULL, NULL);
 | 
						|
  AddTestCase (AdditionSubtractionTestSuite, "Test SafeInt64Sub", "TestSafeInt64Sub", TestSafeInt64Sub, NULL, NULL, NULL);
 | 
						|
 | 
						|
  //
 | 
						|
  // Test the multiplication functions
 | 
						|
  //
 | 
						|
  Status = CreateUnitTestSuite (&MultiplicationTestSuite, Framework, "Int Safe Multiply Test Suite", "Common.SafeInt.Multiply", NULL, NULL);
 | 
						|
  if (EFI_ERROR (Status)) {
 | 
						|
    DEBUG ((DEBUG_ERROR, "Failed in CreateUnitTestSuite for Int Safe Multiply Test Suite\n"));
 | 
						|
    Status = EFI_OUT_OF_RESOURCES;
 | 
						|
    goto EXIT;
 | 
						|
  }
 | 
						|
 | 
						|
  AddTestCase (MultiplicationTestSuite, "Test SafeUint8Mult", "TestSafeUint8Mult", TestSafeUint8Mult, NULL, NULL, NULL);
 | 
						|
  AddTestCase (MultiplicationTestSuite, "Test SafeUint16Mult", "TestSafeUint16Mult", TestSafeUint16Mult, NULL, NULL, NULL);
 | 
						|
  AddTestCase (MultiplicationTestSuite, "Test SafeUint32Mult", "TestSafeUint32Mult", TestSafeUint32Mult, NULL, NULL, NULL);
 | 
						|
  AddTestCase (MultiplicationTestSuite, "Test SafeUintnMult", "TestSafeUintnMult", TestSafeUintnMult, NULL, NULL, NULL);
 | 
						|
  AddTestCase (MultiplicationTestSuite, "Test SafeUint64Mult", "TestSafeUint64Mult", TestSafeUint64Mult, NULL, NULL, NULL);
 | 
						|
  AddTestCase (MultiplicationTestSuite, "Test SafeInt8Mult", "TestSafeInt8Mult", TestSafeInt8Mult, NULL, NULL, NULL);
 | 
						|
  AddTestCase (MultiplicationTestSuite, "Test SafeInt16Mult", "TestSafeInt16Mult", TestSafeInt16Mult, NULL, NULL, NULL);
 | 
						|
  AddTestCase (MultiplicationTestSuite, "Test SafeInt32Mult", "TestSafeInt32Mult", TestSafeInt32Mult, NULL, NULL, NULL);
 | 
						|
  AddTestCase (MultiplicationTestSuite, "Test SafeIntnMult", "TestSafeIntnMult", TestSafeIntnMult, NULL, NULL, NULL);
 | 
						|
  AddTestCase (MultiplicationTestSuite, "Test SafeInt64Mult", "TestSafeInt64Mult", TestSafeInt64Mult, NULL, NULL, NULL);
 | 
						|
 | 
						|
  //
 | 
						|
  // Execute the tests.
 | 
						|
  //
 | 
						|
  Status = RunAllTestSuites (Framework);
 | 
						|
 | 
						|
EXIT:
 | 
						|
  if (Framework != NULL) {
 | 
						|
    FreeUnitTestFramework (Framework);
 | 
						|
  }
 | 
						|
 | 
						|
  return Status;
 | 
						|
}
 | 
						|
 | 
						|
EFI_STATUS
 | 
						|
EFIAPI
 | 
						|
PeiEntryPoint (
 | 
						|
  IN EFI_PEI_FILE_HANDLE     FileHandle,
 | 
						|
  IN CONST EFI_PEI_SERVICES  **PeiServices
 | 
						|
  )
 | 
						|
{
 | 
						|
  return UefiTestMain ();
 | 
						|
}
 | 
						|
 | 
						|
EFI_STATUS
 | 
						|
EFIAPI
 | 
						|
DxeEntryPoint (
 | 
						|
  IN EFI_HANDLE        ImageHandle,
 | 
						|
  IN EFI_SYSTEM_TABLE  *SystemTable
 | 
						|
  )
 | 
						|
{
 | 
						|
  return UefiTestMain ();
 | 
						|
}
 | 
						|
 | 
						|
int
 | 
						|
main (
 | 
						|
  int   argc,
 | 
						|
  char  *argv[]
 | 
						|
  )
 | 
						|
{
 | 
						|
  return UefiTestMain ();
 | 
						|
}
 |