]> git.proxmox.com Git - mirror_edk2.git/blobdiff - MdePkg/Test/UnitTest/Library/BaseSafeIntLib/SafeIntLibUintnIntnUnitTests32.c
projects / mirror_edk2.git / blobdiff
? search:
summary | shortlog | log | commit | commitdiff | tree
raw | inline | side by side
MdePkg/Test: Add SafeIntLib and BaseLib Base64 unit tests
[mirror_edk2.git] / MdePkg / Test / UnitTest / Library / BaseSafeIntLib / SafeIntLibUintnIntnUnitTests32.c
diff --git a/MdePkg/Test/UnitTest/Library/BaseSafeIntLib/SafeIntLibUintnIntnUnitTests32.c b/MdePkg/Test/UnitTest/Library/BaseSafeIntLib/SafeIntLibUintnIntnUnitTests32.c
new file mode 100644 (file)
index 0000000..be5c0e1
--- /dev/null
+++ b/MdePkg/Test/UnitTest/Library/BaseSafeIntLib/SafeIntLibUintnIntnUnitTests32.c
@@ -0,0 +1,540 @@
+/** @file\r
+  IA32-specific functions for unit-testing INTN and UINTN functions in\r
+  SafeIntLib.\r
+\r
+  Copyright (c) Microsoft Corporation.<BR>\r
+  Copyright (c) 2019 - 2020, Intel Corporation. All rights reserved.<BR>\r
+  SPDX-License-Identifier: BSD-2-Clause-Patent\r
+\r
+**/\r
+\r
+#include "TestBaseSafeIntLib.h"\r
+\r
+UNIT_TEST_STATUS\r
+EFIAPI\r
+TestSafeInt32ToUintn (\r
+  IN UNIT_TEST_CONTEXT           Context\r
+  )\r
+{\r
+  EFI_STATUS  Status;\r
+  INT32       Operand;\r
+  UINTN       Result;\r
+\r
+  //\r
+  // If Operand is non-negative, then it's a cast\r
+  //\r
+  Operand = 0x5bababab;\r
+  Result = 0;\r
+  Status = SafeInt32ToUintn(Operand, &Result);\r
+  UT_ASSERT_NOT_EFI_ERROR(Status);\r
+  UT_ASSERT_EQUAL(0x5bababab, Result);\r
+\r
+  //\r
+  // Otherwise should result in an error status\r
+  //\r
+  Operand = (-1537977259);\r
+  Status = SafeInt32ToUintn(Operand, &Result);\r
+  UT_ASSERT_EQUAL(RETURN_BUFFER_TOO_SMALL, Status);\r
+\r
+  return UNIT_TEST_PASSED;\r
+}\r
+\r
+UNIT_TEST_STATUS\r
+EFIAPI\r
+TestSafeUint32ToIntn (\r
+  IN UNIT_TEST_CONTEXT           Context\r
+  )\r
+{\r
+  EFI_STATUS  Status;\r
+  UINT32      Operand;\r
+  INTN        Result;\r
+\r
+  //\r
+  // If Operand is <= MAX_INTN, then it's a cast\r
+  //\r
+  Operand = 0x5bababab;\r
+  Result = 0;\r
+  Status = SafeUint32ToIntn(Operand, &Result);\r
+  UT_ASSERT_NOT_EFI_ERROR(Status);\r
+  UT_ASSERT_EQUAL(0x5bababab, Result);\r
+\r
+  //\r
+  // Otherwise should result in an error status\r
+  //\r
+  Operand = (0xabababab);\r
+  Status = SafeUint32ToIntn(Operand, &Result);\r
+  UT_ASSERT_EQUAL(RETURN_BUFFER_TOO_SMALL, Status);\r
+\r
+  return UNIT_TEST_PASSED;\r
+}\r
+\r
+UNIT_TEST_STATUS\r
+EFIAPI\r
+TestSafeIntnToInt32 (\r
+  IN UNIT_TEST_CONTEXT           Context\r
+  )\r
+{\r
+  EFI_STATUS  Status;\r
+  INTN        Operand;\r
+  INT32       Result;\r
+\r
+  //\r
+  // INTN is same as INT32 in IA32, so this is just a cast\r
+  //\r
+  Operand = 0x5bababab;\r
+  Result = 0;\r
+  Status = SafeIntnToInt32(Operand, &Result);\r
+  UT_ASSERT_NOT_EFI_ERROR(Status);\r
+  UT_ASSERT_EQUAL(0x5bababab, Result);\r
+\r
+  return UNIT_TEST_PASSED;\r
+}\r
+\r
+UNIT_TEST_STATUS\r
+EFIAPI\r
+TestSafeIntnToUint32 (\r
+  IN UNIT_TEST_CONTEXT           Context\r
+  )\r
+{\r
+  EFI_STATUS  Status;\r
+  INTN        Operand;\r
+  UINT32      Result;\r
+\r
+  //\r
+  // If Operand is non-negative, then it's a cast\r
+  //\r
+  Operand = 0x5bababab;\r
+  Result = 0;\r
+  Status = SafeIntnToUint32(Operand, &Result);\r
+  UT_ASSERT_NOT_EFI_ERROR(Status);\r
+  UT_ASSERT_EQUAL(0x5bababab, Result);\r
+\r
+  //\r
+  // Otherwise should result in an error status\r
+  //\r
+  Operand = (-1537977259);\r
+  Status = SafeIntnToUint32(Operand, &Result);\r
+  UT_ASSERT_EQUAL(RETURN_BUFFER_TOO_SMALL, Status);\r
+\r
+  return UNIT_TEST_PASSED;\r
+}\r
+\r
+UNIT_TEST_STATUS\r
+EFIAPI\r
+TestSafeUintnToUint32 (\r
+  IN UNIT_TEST_CONTEXT           Context\r
+  )\r
+{\r
+  EFI_STATUS  Status;\r
+  UINTN       Operand;\r
+  UINT32      Result;\r
+\r
+  //\r
+  // UINTN is same as UINT32 in IA32, so this is just a cast\r
+  //\r
+  Operand = 0xabababab;\r
+  Result = 0;\r
+  Status = SafeUintnToUint32(Operand, &Result);\r
+  UT_ASSERT_NOT_EFI_ERROR(Status);\r
+  UT_ASSERT_EQUAL(0xabababab, Result);\r
+\r
+  return UNIT_TEST_PASSED;\r
+}\r
+\r
+UNIT_TEST_STATUS\r
+EFIAPI\r
+TestSafeUintnToIntn (\r
+  IN UNIT_TEST_CONTEXT           Context\r
+  )\r
+{\r
+  EFI_STATUS  Status;\r
+  UINTN       Operand;\r
+  INTN        Result;\r
+\r
+  //\r
+  // If Operand is <= MAX_INTN, then it's a cast\r
+  //\r
+  Operand = 0x5bababab;\r
+  Result = 0;\r
+  Status = SafeUintnToIntn(Operand, &Result);\r
+  UT_ASSERT_NOT_EFI_ERROR(Status);\r
+  UT_ASSERT_EQUAL(0x5bababab, Result);\r
+\r
+  //\r
+  // Otherwise should result in an error status\r
+  //\r
+  Operand = (0xabababab);\r
+  Status = SafeUintnToIntn(Operand, &Result);\r
+  UT_ASSERT_EQUAL(RETURN_BUFFER_TOO_SMALL, Status);\r
+\r
+  return UNIT_TEST_PASSED;\r
+}\r
+\r
+UNIT_TEST_STATUS\r
+EFIAPI\r
+TestSafeUintnToInt64 (\r
+  IN UNIT_TEST_CONTEXT           Context\r
+  )\r
+{\r
+  EFI_STATUS  Status;\r
+  UINTN       Operand;\r
+  INT64       Result;\r
+\r
+  //\r
+  // UINTN is same as UINT32 in IA32, and UINT32 is a subset of\r
+  // INT64, so this is just a cast\r
+  //\r
+  Operand = 0xabababab;\r
+  Result = 0;\r
+  Status = SafeUintnToInt64(Operand, &Result);\r
+  UT_ASSERT_NOT_EFI_ERROR(Status);\r
+  UT_ASSERT_EQUAL(0xabababab, Result);\r
+\r
+  return UNIT_TEST_PASSED;\r
+}\r
+\r
+UNIT_TEST_STATUS\r
+EFIAPI\r
+TestSafeInt64ToIntn (\r
+  IN UNIT_TEST_CONTEXT           Context\r
+  )\r
+{\r
+  EFI_STATUS  Status;\r
+  INT64       Operand;\r
+  INTN        Result;\r
+\r
+  //\r
+  // If Operand is between MIN_INTN and  MAX_INTN2 inclusive, then it's a cast\r
+  //\r
+  Operand = 0x5bababab;\r
+  Result = 0;\r
+  Status = SafeInt64ToIntn(Operand, &Result);\r
+  UT_ASSERT_NOT_EFI_ERROR(Status);\r
+  UT_ASSERT_EQUAL(0x5bababab, Result);\r
+\r
+  Operand = (-1537977259);\r
+  Status = SafeInt64ToIntn(Operand, &Result);\r
+  UT_ASSERT_NOT_EFI_ERROR(Status);\r
+  UT_ASSERT_EQUAL((-1537977259), Result);\r
+\r
+  //\r
+  // Otherwise should result in an error status\r
+  //\r
+  Operand = (0x5babababefefefef);\r
+  Status = SafeInt64ToIntn(Operand, &Result);\r
+  UT_ASSERT_EQUAL(RETURN_BUFFER_TOO_SMALL, Status);\r
+\r
+  Operand =  (-6605562033422200815);\r
+  Status = SafeInt64ToIntn(Operand, &Result);\r
+  UT_ASSERT_EQUAL(RETURN_BUFFER_TOO_SMALL, Status);\r
+\r
+  return UNIT_TEST_PASSED;\r
+}\r
+\r
+UNIT_TEST_STATUS\r
+EFIAPI\r
+TestSafeInt64ToUintn (\r
+  IN UNIT_TEST_CONTEXT           Context\r
+  )\r
+{\r
+  EFI_STATUS  Status;\r
+  INT64       Operand;\r
+  UINTN       Result;\r
+\r
+  //\r
+  // If Operand is between 0 and  MAX_UINTN inclusive, then it's a cast\r
+  //\r
+  Operand = 0xabababab;\r
+  Result = 0;\r
+  Status = SafeInt64ToUintn(Operand, &Result);\r
+  UT_ASSERT_NOT_EFI_ERROR(Status);\r
+  UT_ASSERT_EQUAL(0xabababab, Result);\r
+\r
+  //\r
+  // Otherwise should result in an error status\r
+  //\r
+  Operand = (0x5babababefefefef);\r
+  Status = SafeInt64ToUintn(Operand, &Result);\r
+  UT_ASSERT_EQUAL(RETURN_BUFFER_TOO_SMALL, Status);\r
+\r
+  Operand =  (-6605562033422200815);\r
+  Status = SafeInt64ToUintn(Operand, &Result);\r
+  UT_ASSERT_EQUAL(RETURN_BUFFER_TOO_SMALL, Status);\r
+\r
+  return UNIT_TEST_PASSED;\r
+}\r
+\r
+UNIT_TEST_STATUS\r
+EFIAPI\r
+TestSafeUint64ToIntn (\r
+  IN UNIT_TEST_CONTEXT           Context\r
+  )\r
+{\r
+  EFI_STATUS  Status;\r
+  UINT64      Operand;\r
+  INTN        Result;\r
+\r
+  //\r
+  // If Operand is <= MAX_INTN, then it's a cast\r
+  //\r
+  Operand = 0x5bababab;\r
+  Result = 0;\r
+  Status = SafeUint64ToIntn(Operand, &Result);\r
+  UT_ASSERT_NOT_EFI_ERROR(Status);\r
+  UT_ASSERT_EQUAL(0x5bababab, Result);\r
+\r
+  //\r
+  // Otherwise should result in an error status\r
+  //\r
+  Operand = (0xababababefefefef);\r
+  Status = SafeUint64ToIntn(Operand, &Result);\r
+  UT_ASSERT_EQUAL(RETURN_BUFFER_TOO_SMALL, Status);\r
+\r
+  return UNIT_TEST_PASSED;\r
+}\r
+\r
+UNIT_TEST_STATUS\r
+EFIAPI\r
+TestSafeUint64ToUintn (\r
+  IN UNIT_TEST_CONTEXT           Context\r
+  )\r
+{\r
+  EFI_STATUS  Status;\r
+  UINT64      Operand;\r
+  UINTN       Result;\r
+\r
+  //\r
+  // If Operand is <= MAX_UINTN, then it's a cast\r
+  //\r
+  Operand = 0xabababab;\r
+  Result = 0;\r
+  Status = SafeUint64ToUintn(Operand, &Result);\r
+  UT_ASSERT_NOT_EFI_ERROR(Status);\r
+  UT_ASSERT_EQUAL(0xabababab, Result);\r
+\r
+  //\r
+  // Otherwise should result in an error status\r
+  //\r
+  Operand = (0xababababefefefef);\r
+  Status = SafeUint64ToUintn(Operand, &Result);\r
+  UT_ASSERT_EQUAL(RETURN_BUFFER_TOO_SMALL, Status);\r
+\r
+  return UNIT_TEST_PASSED;\r
+}\r
+\r
+UNIT_TEST_STATUS\r
+EFIAPI\r
+TestSafeUintnAdd (\r
+  IN UNIT_TEST_CONTEXT           Context\r
+  )\r
+{\r
+  EFI_STATUS  Status;\r
+  UINTN       Augend;\r
+  UINTN       Addend;\r
+  UINTN       Result;\r
+\r
+  //\r
+  // If the result of addition doesn't overflow MAX_UINTN, then it's addition\r
+  //\r
+  Augend = 0x3a3a3a3a;\r
+  Addend = 0x3a3a3a3a;\r
+  Result = 0;\r
+  Status = SafeUintnAdd(Augend, Addend, &Result);\r
+  UT_ASSERT_NOT_EFI_ERROR(Status);\r
+  UT_ASSERT_EQUAL(0x74747474, Result);\r
+\r
+  //\r
+  // Otherwise should result in an error status\r
+  //\r
+  Augend = 0xabababab;\r
+  Addend = 0xbcbcbcbc;\r
+  Status = SafeUintnAdd(Augend, Addend, &Result);\r
+  UT_ASSERT_EQUAL(RETURN_BUFFER_TOO_SMALL, Status);\r
+\r
+  return UNIT_TEST_PASSED;\r
+}\r
+\r
+UNIT_TEST_STATUS\r
+EFIAPI\r
+TestSafeIntnAdd (\r
+  IN UNIT_TEST_CONTEXT           Context\r
+  )\r
+{\r
+  EFI_STATUS  Status;\r
+  INTN        Augend;\r
+  INTN        Addend;\r
+  INTN        Result;\r
+\r
+  //\r
+  // If the result of addition doesn't overflow MAX_INTN\r
+  // and doesn't underflow MIN_INTN, then it's addition\r
+  //\r
+  Augend = 0x3a3a3a3a;\r
+  Addend = 0x3a3a3a3a;\r
+  Result = 0;\r
+  Status = SafeIntnAdd(Augend, Addend, &Result);\r
+  UT_ASSERT_NOT_EFI_ERROR(Status);\r
+  UT_ASSERT_EQUAL(0x74747474, Result);\r
+\r
+  Augend = (-976894522);\r
+  Addend = (-976894522);\r
+  Status = SafeIntnAdd(Augend, Addend, &Result);\r
+  UT_ASSERT_NOT_EFI_ERROR(Status);\r
+  UT_ASSERT_EQUAL((-1953789044), Result);\r
+\r
+  //\r
+  // Otherwise should result in an error status\r
+  //\r
+  Augend = 0x5a5a5a5a;\r
+  Addend = 0x5a5a5a5a;\r
+  Status = SafeIntnAdd(Augend, Addend, &Result);\r
+  UT_ASSERT_EQUAL(RETURN_BUFFER_TOO_SMALL, Status);\r
+\r
+  Augend = (-1515870810);\r
+  Addend = (-1515870810);\r
+  Status = SafeIntnAdd(Augend, Addend, &Result);\r
+  UT_ASSERT_EQUAL(RETURN_BUFFER_TOO_SMALL, Status);\r
+\r
+  return UNIT_TEST_PASSED;\r
+}\r
+\r
+UNIT_TEST_STATUS\r
+EFIAPI\r
+TestSafeUintnSub (\r
+  IN UNIT_TEST_CONTEXT           Context\r
+  )\r
+{\r
+  EFI_STATUS  Status;\r
+  UINTN       Minuend;\r
+  UINTN       Subtrahend;\r
+  UINTN       Result;\r
+\r
+  //\r
+  // If Minuend >= Subtrahend, then it's subtraction\r
+  //\r
+  Minuend = 0x5a5a5a5a;\r
+  Subtrahend = 0x3b3b3b3b;\r
+  Result = 0;\r
+  Status = SafeUintnSub(Minuend, Subtrahend, &Result);\r
+  UT_ASSERT_NOT_EFI_ERROR(Status);\r
+  UT_ASSERT_EQUAL(0x1f1f1f1f, Result);\r
+\r
+  //\r
+  // Otherwise should result in an error status\r
+  //\r
+  Minuend = 0x5a5a5a5a;\r
+  Subtrahend = 0x6d6d6d6d;\r
+  Status = SafeUintnSub(Minuend, Subtrahend, &Result);\r
+  UT_ASSERT_EQUAL(RETURN_BUFFER_TOO_SMALL, Status);\r
+\r
+  return UNIT_TEST_PASSED;\r
+}\r
+\r
+UNIT_TEST_STATUS\r
+EFIAPI\r
+TestSafeIntnSub (\r
+  IN UNIT_TEST_CONTEXT           Context\r
+  )\r
+{\r
+  EFI_STATUS  Status;\r
+  INTN        Minuend;\r
+  INTN        Subtrahend;\r
+  INTN        Result;\r
+\r
+  //\r
+  // If the result of subtractions doesn't overflow MAX_INTN or\r
+  // underflow MIN_INTN, then it's subtraction\r
+  //\r
+  Minuend = 0x5a5a5a5a;\r
+  Subtrahend = 0x3a3a3a3a;\r
+  Result = 0;\r
+  Status = SafeIntnSub(Minuend, Subtrahend, &Result);\r
+  UT_ASSERT_NOT_EFI_ERROR(Status);\r
+  UT_ASSERT_EQUAL(0x20202020, Result);\r
+\r
+  Minuend = 0x3a3a3a3a;\r
+  Subtrahend = 0x5a5a5a5a;\r
+  Status = SafeIntnSub(Minuend, Subtrahend, &Result);\r
+  UT_ASSERT_NOT_EFI_ERROR(Status);\r
+  UT_ASSERT_EQUAL((-538976288), Result);\r
+\r
+  //\r
+  // Otherwise should result in an error status\r
+  //\r
+  Minuend = (-2054847098);\r
+  Subtrahend = 2054847098;\r
+  Status = SafeIntnSub(Minuend, Subtrahend, &Result);\r
+  UT_ASSERT_EQUAL(RETURN_BUFFER_TOO_SMALL, Status);\r
+\r
+  Minuend = (2054847098);\r
+  Subtrahend = (-2054847098);\r
+  Status = SafeIntnSub(Minuend, Subtrahend, &Result);\r
+  UT_ASSERT_EQUAL(RETURN_BUFFER_TOO_SMALL, Status);\r
+\r
+  return UNIT_TEST_PASSED;\r
+}\r
+\r
+UNIT_TEST_STATUS\r
+EFIAPI\r
+TestSafeUintnMult (\r
+  IN UNIT_TEST_CONTEXT           Context\r
+  )\r
+{\r
+  EFI_STATUS  Status;\r
+  UINTN       Multiplicand;\r
+  UINTN       Multiplier;\r
+  UINTN       Result;\r
+\r
+  //\r
+  // If the result of multiplication doesn't overflow MAX_UINTN, it will succeed\r
+  //\r
+  Multiplicand = 0xa122a;\r
+  Multiplier = 0xd23;\r
+  Result = 0;\r
+  Status = SafeUintnMult(Multiplicand, Multiplier, &Result);\r
+  UT_ASSERT_NOT_EFI_ERROR(Status);\r
+  UT_ASSERT_EQUAL(0x844c9dbe, Result);\r
+\r
+  //\r
+  // Otherwise should result in an error status\r
+  //\r
+  Multiplicand = 0xa122a;\r
+  Multiplier = 0xed23;\r
+  Status = SafeUintnMult(Multiplicand, Multiplier, &Result);\r
+  UT_ASSERT_EQUAL(RETURN_BUFFER_TOO_SMALL, Status);\r
+\r
+  return UNIT_TEST_PASSED;\r
+}\r
+\r
+UNIT_TEST_STATUS\r
+EFIAPI\r
+TestSafeIntnMult (\r
+  IN UNIT_TEST_CONTEXT           Context\r
+  )\r
+{\r
+  EFI_STATUS  Status;\r
+  INTN        Multiplicand;\r
+  INTN        Multiplier;\r
+  INTN        Result;\r
+\r
+  //\r
+  // If the result of multiplication doesn't overflow MAX_INTN and doesn't\r
+  // underflow MIN_UINTN, it will succeed\r
+  //\r
+  Multiplicand = 0x123456;\r
+  Multiplier = 0x678;\r
+  Result = 0;\r
+  Status = SafeIntnMult(Multiplicand, Multiplier, &Result);\r
+  UT_ASSERT_NOT_EFI_ERROR(Status);\r
+  UT_ASSERT_EQUAL(0x75c28c50, Result);\r
+\r
+  //\r
+  // Otherwise should result in an error status\r
+  //\r
+  Multiplicand = 0x123456;\r
+  Multiplier = 0xabc;\r
+  Status = SafeIntnMult(Multiplicand, Multiplier, &Result);\r
+  UT_ASSERT_EQUAL(RETURN_BUFFER_TOO_SMALL, Status);\r
+\r
+  return UNIT_TEST_PASSED;\r
+}\r
EFI Development Kit II mirror for PVE