MdePkg/Test/BaseLib: Add SAFE_STRING_CONSTRAINT_CHECK unit test

[mirror_edk2.git] / MdePkg / Test / UnitTest / Library / BaseLib / Base64UnitTest.c

diff --git a/MdePkg/Test/UnitTest/Library/BaseLib/Base64UnitTest.c b/MdePkg/Test/UnitTest/Library/BaseLib/Base64UnitTest.c
index 8952f9da6c3a0894c75ff21e541def1872e7afe4..2c4266491ca97ffa47ae65ceda1ece11966d9cbe 100644 (file)
--- a/MdePkg/Test/UnitTest/Library/BaseLib/Base64UnitTest.c
+++ b/MdePkg/Test/UnitTest/Library/BaseLib/Base64UnitTest.c
@@ -290,6 +290,99 @@ RfcDecodeTest(
   return UNIT_TEST_PASSED;\r
 }\r
 \r
   return UNIT_TEST_PASSED;\r
 }\r
 \r

+#define SOURCE_STRING  L"Hello"\r
+\r
+STATIC\r
+UNIT_TEST_STATUS\r
+EFIAPI\r
+SafeStringContraintCheckTest (\r
+  IN UNIT_TEST_CONTEXT  Context\r
+  )\r
+{\r
+  RETURN_STATUS  Status;\r
+  CHAR16         Destination[20];\r
+  CHAR16         AllZero[20];\r
+\r
+  //\r
+  // Zero buffer used to verify Destination is not modified\r
+  //\r
+  ZeroMem (AllZero, sizeof (AllZero));\r
+\r
+  //\r
+  // Positive test case copy source unicode string to destination\r
+  //\r
+  ZeroMem (Destination, sizeof (Destination));\r
+  Status = StrCpyS (Destination, sizeof (Destination) / sizeof (CHAR16), SOURCE_STRING);\r
+  UT_ASSERT_NOT_EFI_ERROR (Status);\r
+  UT_ASSERT_MEM_EQUAL (Destination, SOURCE_STRING, sizeof (SOURCE_STRING));\r
+\r
+  //\r
+  // Positive test case with DestMax the same as Source size\r
+  //\r
+  ZeroMem (Destination, sizeof (Destination));\r
+  Status = StrCpyS (Destination, sizeof (SOURCE_STRING) / sizeof (CHAR16), SOURCE_STRING);\r
+  UT_ASSERT_NOT_EFI_ERROR (Status);\r
+  UT_ASSERT_MEM_EQUAL (Destination, SOURCE_STRING, sizeof (SOURCE_STRING));\r
+\r
+  //\r
+  // Negative test case with Destination NULL\r
+  //\r
+  ZeroMem (Destination, sizeof (Destination));\r
+  Status = StrCpyS (NULL, sizeof (Destination) / sizeof (CHAR16), SOURCE_STRING);\r
+  UT_ASSERT_STATUS_EQUAL (Status, RETURN_INVALID_PARAMETER);\r
+  UT_ASSERT_MEM_EQUAL (Destination, AllZero, sizeof (AllZero));\r
+\r
+  //\r
+  // Negative test case with Source NULL\r
+  //\r
+  ZeroMem (Destination, sizeof (Destination));\r
+  Status = StrCpyS (Destination, sizeof (Destination) / sizeof (CHAR16), NULL);\r
+  UT_ASSERT_STATUS_EQUAL (Status, RETURN_INVALID_PARAMETER);\r
+  UT_ASSERT_MEM_EQUAL (Destination, AllZero, sizeof (AllZero));\r
+\r
+  //\r
+  // Negative test case with DestMax too big\r
+  //\r
+  ZeroMem (Destination, sizeof (Destination));\r
+  Status = StrCpyS (Destination, MAX_UINTN, SOURCE_STRING);\r
+  UT_ASSERT_STATUS_EQUAL (Status, RETURN_INVALID_PARAMETER);\r
+  UT_ASSERT_MEM_EQUAL (Destination, AllZero, sizeof (AllZero));\r
+\r
+  //\r
+  // Negative test case with DestMax 0\r
+  //\r
+  ZeroMem (Destination, sizeof (Destination));\r
+  Status = StrCpyS (Destination, 0, SOURCE_STRING);\r
+  UT_ASSERT_STATUS_EQUAL (Status, RETURN_INVALID_PARAMETER);\r
+  UT_ASSERT_MEM_EQUAL (Destination, AllZero, sizeof (AllZero));\r
+\r
+  //\r
+  // Negative test case with DestMax smaller than Source size\r
+  //\r
+  ZeroMem (Destination, sizeof (Destination));\r
+  Status = StrCpyS (Destination, 1, SOURCE_STRING);\r
+  UT_ASSERT_STATUS_EQUAL (Status, RETURN_BUFFER_TOO_SMALL);\r
+  UT_ASSERT_MEM_EQUAL (Destination, AllZero, sizeof (AllZero));\r
+\r
+  //\r
+  // Negative test case with DestMax smaller than Source size by one character\r
+  //\r
+  ZeroMem (Destination, sizeof (Destination));\r
+  Status = StrCpyS (Destination, sizeof (SOURCE_STRING) / sizeof (CHAR16) - 1, SOURCE_STRING);\r
+  UT_ASSERT_STATUS_EQUAL (Status, RETURN_BUFFER_TOO_SMALL);\r
+  UT_ASSERT_MEM_EQUAL (Destination, AllZero, sizeof (AllZero));\r
+\r
+  //\r
+  // Negative test case with overlapping Destination and Source\r
+  //\r
+  ZeroMem (Destination, sizeof (Destination));\r
+  Status = StrCpyS (Destination, sizeof (Destination) / sizeof (CHAR16), Destination);\r
+  UT_ASSERT_STATUS_EQUAL (Status, RETURN_ACCESS_DENIED);\r
+  UT_ASSERT_MEM_EQUAL (Destination, AllZero, sizeof (AllZero));\r
+\r
+  return UNIT_TEST_PASSED;\r
+}\r
+\r

 /**\r
   Initialze the unit test framework, suite, and unit tests for the\r
   Base64 conversion APIs of BaseLib and run the unit tests.\r
 /**\r
   Initialze the unit test framework, suite, and unit tests for the\r
   Base64 conversion APIs of BaseLib and run the unit tests.\r


@@ -309,6 +402,7 @@ UnitTestingEntry (
   UNIT_TEST_FRAMEWORK_HANDLE  Fw;\r
   UNIT_TEST_SUITE_HANDLE      b64EncodeTests;\r
   UNIT_TEST_SUITE_HANDLE      b64DecodeTests;\r
   UNIT_TEST_FRAMEWORK_HANDLE  Fw;\r
   UNIT_TEST_SUITE_HANDLE      b64EncodeTests;\r
   UNIT_TEST_SUITE_HANDLE      b64DecodeTests;\r

+  UNIT_TEST_SUITE_HANDLE      SafeStringTests;\r

 \r
   Fw = NULL;\r
 \r
 \r
   Fw = NULL;\r
 \r


@@ -367,6 +461,19 @@ UnitTestingEntry (
   AddTestCase (b64DecodeTests, "Incorrectly placed padding character", "Error4", RfcDecodeTest, NULL, CleanUpB64TestContext, &mBasicDecodeError4);\r
   AddTestCase (b64DecodeTests, "Too small of output buffer", "Error5", RfcDecodeTest, NULL, CleanUpB64TestContext, &mBasicDecodeError5);\r
 \r
   AddTestCase (b64DecodeTests, "Incorrectly placed padding character", "Error4", RfcDecodeTest, NULL, CleanUpB64TestContext, &mBasicDecodeError4);\r
   AddTestCase (b64DecodeTests, "Too small of output buffer", "Error5", RfcDecodeTest, NULL, CleanUpB64TestContext, &mBasicDecodeError5);\r
 \r

+  //\r
+  // Populate the safe string Unit Test Suite.\r
+  //\r
+  Status = CreateUnitTestSuite (&SafeStringTests, Fw, "Safe String", "BaseLib.SafeString", NULL, NULL);\r
+  if (EFI_ERROR (Status)) {\r
+    DEBUG ((DEBUG_ERROR, "Failed in CreateUnitTestSuite for SafeStringTests\n"));\r
+    Status = EFI_OUT_OF_RESOURCES;\r
+    goto EXIT;\r
+  }\r
+\r
+  // --------------Suite-----------Description--------------Class Name----------Function--------Pre---Post-------------------Context-----------\r
+  AddTestCase (SafeStringTests, "SAFE_STRING_CONSTRAINT_CHECK", "SafeStringContraintCheckTest", SafeStringContraintCheckTest, NULL, NULL, NULL);\r
+\r

   //\r
   // Execute the tests.\r
   //\r
   //\r
   // Execute the tests.\r
   //\r